mirror of
				https://gitea.com/Lydanne/buildx.git
				synced 2025-11-04 10:03:42 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			203 lines
		
	
	
		
			7.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			203 lines
		
	
	
		
			7.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
//go:build windows
 | 
						|
 | 
						|
package fs
 | 
						|
 | 
						|
import (
 | 
						|
	"golang.org/x/sys/windows"
 | 
						|
 | 
						|
	"github.com/Microsoft/go-winio/internal/stringbuffer"
 | 
						|
)
 | 
						|
 | 
						|
//go:generate go run github.com/Microsoft/go-winio/tools/mkwinsyscall -output zsyscall_windows.go fs.go
 | 
						|
 | 
						|
// https://learn.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-createfilew
 | 
						|
//sys CreateFile(name string, access AccessMask, mode FileShareMode, sa *syscall.SecurityAttributes, createmode FileCreationDisposition, attrs FileFlagOrAttribute, templatefile windows.Handle) (handle windows.Handle, err error) [failretval==windows.InvalidHandle] = CreateFileW
 | 
						|
 | 
						|
const NullHandle windows.Handle = 0
 | 
						|
 | 
						|
// AccessMask defines standard, specific, and generic rights.
 | 
						|
//
 | 
						|
//	Bitmask:
 | 
						|
//	 3 3 2 2 2 2 2 2 2 2 2 2 1 1 1 1 1 1 1 1 1 1
 | 
						|
//	 1 0 9 8 7 6 5 4 3 2 1 0 9 8 7 6 5 4 3 2 1 0 9 8 7 6 5 4 3 2 1 0
 | 
						|
//	+---------------+---------------+-------------------------------+
 | 
						|
//	|G|G|G|G|Resvd|A| StandardRights|         SpecificRights        |
 | 
						|
//	|R|W|E|A|     |S|               |                               |
 | 
						|
//	+-+-------------+---------------+-------------------------------+
 | 
						|
//
 | 
						|
//	GR     Generic Read
 | 
						|
//	GW     Generic Write
 | 
						|
//	GE     Generic Exectue
 | 
						|
//	GA     Generic All
 | 
						|
//	Resvd  Reserved
 | 
						|
//	AS     Access Security System
 | 
						|
//
 | 
						|
// https://learn.microsoft.com/en-us/windows/win32/secauthz/access-mask
 | 
						|
//
 | 
						|
// https://learn.microsoft.com/en-us/windows/win32/secauthz/generic-access-rights
 | 
						|
//
 | 
						|
// https://learn.microsoft.com/en-us/windows/win32/fileio/file-access-rights-constants
 | 
						|
type AccessMask = windows.ACCESS_MASK
 | 
						|
 | 
						|
//nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
 | 
						|
const (
 | 
						|
	// Not actually any.
 | 
						|
	//
 | 
						|
	// For CreateFile: "query certain metadata such as file, directory, or device attributes without accessing that file or device"
 | 
						|
	// https://learn.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-createfilew#parameters
 | 
						|
	FILE_ANY_ACCESS AccessMask = 0
 | 
						|
 | 
						|
	// Specific Object Access
 | 
						|
	// from ntioapi.h
 | 
						|
 | 
						|
	FILE_READ_DATA      AccessMask = (0x0001) // file & pipe
 | 
						|
	FILE_LIST_DIRECTORY AccessMask = (0x0001) // directory
 | 
						|
 | 
						|
	FILE_WRITE_DATA AccessMask = (0x0002) // file & pipe
 | 
						|
	FILE_ADD_FILE   AccessMask = (0x0002) // directory
 | 
						|
 | 
						|
	FILE_APPEND_DATA          AccessMask = (0x0004) // file
 | 
						|
	FILE_ADD_SUBDIRECTORY     AccessMask = (0x0004) // directory
 | 
						|
	FILE_CREATE_PIPE_INSTANCE AccessMask = (0x0004) // named pipe
 | 
						|
 | 
						|
	FILE_READ_EA         AccessMask = (0x0008) // file & directory
 | 
						|
	FILE_READ_PROPERTIES AccessMask = FILE_READ_EA
 | 
						|
 | 
						|
	FILE_WRITE_EA         AccessMask = (0x0010) // file & directory
 | 
						|
	FILE_WRITE_PROPERTIES AccessMask = FILE_WRITE_EA
 | 
						|
 | 
						|
	FILE_EXECUTE  AccessMask = (0x0020) // file
 | 
						|
	FILE_TRAVERSE AccessMask = (0x0020) // directory
 | 
						|
 | 
						|
	FILE_DELETE_CHILD AccessMask = (0x0040) // directory
 | 
						|
 | 
						|
	FILE_READ_ATTRIBUTES AccessMask = (0x0080) // all
 | 
						|
 | 
						|
	FILE_WRITE_ATTRIBUTES AccessMask = (0x0100) // all
 | 
						|
 | 
						|
	FILE_ALL_ACCESS      AccessMask = (STANDARD_RIGHTS_REQUIRED | SYNCHRONIZE | 0x1FF)
 | 
						|
	FILE_GENERIC_READ    AccessMask = (STANDARD_RIGHTS_READ | FILE_READ_DATA | FILE_READ_ATTRIBUTES | FILE_READ_EA | SYNCHRONIZE)
 | 
						|
	FILE_GENERIC_WRITE   AccessMask = (STANDARD_RIGHTS_WRITE | FILE_WRITE_DATA | FILE_WRITE_ATTRIBUTES | FILE_WRITE_EA | FILE_APPEND_DATA | SYNCHRONIZE)
 | 
						|
	FILE_GENERIC_EXECUTE AccessMask = (STANDARD_RIGHTS_EXECUTE | FILE_READ_ATTRIBUTES | FILE_EXECUTE | SYNCHRONIZE)
 | 
						|
 | 
						|
	SPECIFIC_RIGHTS_ALL AccessMask = 0x0000FFFF
 | 
						|
 | 
						|
	// Standard Access
 | 
						|
	// from ntseapi.h
 | 
						|
 | 
						|
	DELETE       AccessMask = 0x0001_0000
 | 
						|
	READ_CONTROL AccessMask = 0x0002_0000
 | 
						|
	WRITE_DAC    AccessMask = 0x0004_0000
 | 
						|
	WRITE_OWNER  AccessMask = 0x0008_0000
 | 
						|
	SYNCHRONIZE  AccessMask = 0x0010_0000
 | 
						|
 | 
						|
	STANDARD_RIGHTS_REQUIRED AccessMask = 0x000F_0000
 | 
						|
 | 
						|
	STANDARD_RIGHTS_READ    AccessMask = READ_CONTROL
 | 
						|
	STANDARD_RIGHTS_WRITE   AccessMask = READ_CONTROL
 | 
						|
	STANDARD_RIGHTS_EXECUTE AccessMask = READ_CONTROL
 | 
						|
 | 
						|
	STANDARD_RIGHTS_ALL AccessMask = 0x001F_0000
 | 
						|
)
 | 
						|
 | 
						|
type FileShareMode uint32
 | 
						|
 | 
						|
//nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
 | 
						|
const (
 | 
						|
	FILE_SHARE_NONE        FileShareMode = 0x00
 | 
						|
	FILE_SHARE_READ        FileShareMode = 0x01
 | 
						|
	FILE_SHARE_WRITE       FileShareMode = 0x02
 | 
						|
	FILE_SHARE_DELETE      FileShareMode = 0x04
 | 
						|
	FILE_SHARE_VALID_FLAGS FileShareMode = 0x07
 | 
						|
)
 | 
						|
 | 
						|
type FileCreationDisposition uint32
 | 
						|
 | 
						|
//nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
 | 
						|
const (
 | 
						|
	// from winbase.h
 | 
						|
 | 
						|
	CREATE_NEW        FileCreationDisposition = 0x01
 | 
						|
	CREATE_ALWAYS     FileCreationDisposition = 0x02
 | 
						|
	OPEN_EXISTING     FileCreationDisposition = 0x03
 | 
						|
	OPEN_ALWAYS       FileCreationDisposition = 0x04
 | 
						|
	TRUNCATE_EXISTING FileCreationDisposition = 0x05
 | 
						|
)
 | 
						|
 | 
						|
// CreateFile and co. take flags or attributes together as one parameter.
 | 
						|
// Define alias until we can use generics to allow both
 | 
						|
 | 
						|
// https://learn.microsoft.com/en-us/windows/win32/fileio/file-attribute-constants
 | 
						|
type FileFlagOrAttribute uint32
 | 
						|
 | 
						|
//nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
 | 
						|
const ( // from winnt.h
 | 
						|
	FILE_FLAG_WRITE_THROUGH       FileFlagOrAttribute = 0x8000_0000
 | 
						|
	FILE_FLAG_OVERLAPPED          FileFlagOrAttribute = 0x4000_0000
 | 
						|
	FILE_FLAG_NO_BUFFERING        FileFlagOrAttribute = 0x2000_0000
 | 
						|
	FILE_FLAG_RANDOM_ACCESS       FileFlagOrAttribute = 0x1000_0000
 | 
						|
	FILE_FLAG_SEQUENTIAL_SCAN     FileFlagOrAttribute = 0x0800_0000
 | 
						|
	FILE_FLAG_DELETE_ON_CLOSE     FileFlagOrAttribute = 0x0400_0000
 | 
						|
	FILE_FLAG_BACKUP_SEMANTICS    FileFlagOrAttribute = 0x0200_0000
 | 
						|
	FILE_FLAG_POSIX_SEMANTICS     FileFlagOrAttribute = 0x0100_0000
 | 
						|
	FILE_FLAG_OPEN_REPARSE_POINT  FileFlagOrAttribute = 0x0020_0000
 | 
						|
	FILE_FLAG_OPEN_NO_RECALL      FileFlagOrAttribute = 0x0010_0000
 | 
						|
	FILE_FLAG_FIRST_PIPE_INSTANCE FileFlagOrAttribute = 0x0008_0000
 | 
						|
)
 | 
						|
 | 
						|
type FileSQSFlag = FileFlagOrAttribute
 | 
						|
 | 
						|
//nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
 | 
						|
const ( // from winbase.h
 | 
						|
	SECURITY_ANONYMOUS      FileSQSFlag = FileSQSFlag(SecurityAnonymous << 16)
 | 
						|
	SECURITY_IDENTIFICATION FileSQSFlag = FileSQSFlag(SecurityIdentification << 16)
 | 
						|
	SECURITY_IMPERSONATION  FileSQSFlag = FileSQSFlag(SecurityImpersonation << 16)
 | 
						|
	SECURITY_DELEGATION     FileSQSFlag = FileSQSFlag(SecurityDelegation << 16)
 | 
						|
 | 
						|
	SECURITY_SQOS_PRESENT     FileSQSFlag = 0x00100000
 | 
						|
	SECURITY_VALID_SQOS_FLAGS FileSQSFlag = 0x001F0000
 | 
						|
)
 | 
						|
 | 
						|
// GetFinalPathNameByHandle flags
 | 
						|
//
 | 
						|
// https://learn.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-getfinalpathnamebyhandlew#parameters
 | 
						|
type GetFinalPathFlag uint32
 | 
						|
 | 
						|
//nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
 | 
						|
const (
 | 
						|
	GetFinalPathDefaultFlag GetFinalPathFlag = 0x0
 | 
						|
 | 
						|
	FILE_NAME_NORMALIZED GetFinalPathFlag = 0x0
 | 
						|
	FILE_NAME_OPENED     GetFinalPathFlag = 0x8
 | 
						|
 | 
						|
	VOLUME_NAME_DOS  GetFinalPathFlag = 0x0
 | 
						|
	VOLUME_NAME_GUID GetFinalPathFlag = 0x1
 | 
						|
	VOLUME_NAME_NT   GetFinalPathFlag = 0x2
 | 
						|
	VOLUME_NAME_NONE GetFinalPathFlag = 0x4
 | 
						|
)
 | 
						|
 | 
						|
// getFinalPathNameByHandle facilitates calling the Windows API GetFinalPathNameByHandle
 | 
						|
// with the given handle and flags. It transparently takes care of creating a buffer of the
 | 
						|
// correct size for the call.
 | 
						|
//
 | 
						|
// https://learn.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-getfinalpathnamebyhandlew
 | 
						|
func GetFinalPathNameByHandle(h windows.Handle, flags GetFinalPathFlag) (string, error) {
 | 
						|
	b := stringbuffer.NewWString()
 | 
						|
	//TODO: can loop infinitely if Win32 keeps returning the same (or a larger) n?
 | 
						|
	for {
 | 
						|
		n, err := windows.GetFinalPathNameByHandle(h, b.Pointer(), b.Cap(), uint32(flags))
 | 
						|
		if err != nil {
 | 
						|
			return "", err
 | 
						|
		}
 | 
						|
		// If the buffer wasn't large enough, n will be the total size needed (including null terminator).
 | 
						|
		// Resize and try again.
 | 
						|
		if n > b.Cap() {
 | 
						|
			b.ResizeTo(n)
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		// If the buffer is large enough, n will be the size not including the null terminator.
 | 
						|
		// Convert to a Go string and return.
 | 
						|
		return b.String(), nil
 | 
						|
	}
 | 
						|
}
 |