fs.go 7.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202
  1. //go:build windows
  2. package fs
  3. import (
  4. "golang.org/x/sys/windows"
  5. "github.com/Microsoft/go-winio/internal/stringbuffer"
  6. )
  7. //go:generate go run github.com/Microsoft/go-winio/tools/mkwinsyscall -output zsyscall_windows.go fs.go
  8. // https://learn.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-createfilew
  9. //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
  10. const NullHandle windows.Handle = 0
  11. // AccessMask defines standard, specific, and generic rights.
  12. //
  13. // Bitmask:
  14. // 3 3 2 2 2 2 2 2 2 2 2 2 1 1 1 1 1 1 1 1 1 1
  15. // 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
  16. // +---------------+---------------+-------------------------------+
  17. // |G|G|G|G|Resvd|A| StandardRights| SpecificRights |
  18. // |R|W|E|A| |S| | |
  19. // +-+-------------+---------------+-------------------------------+
  20. //
  21. // GR Generic Read
  22. // GW Generic Write
  23. // GE Generic Exectue
  24. // GA Generic All
  25. // Resvd Reserved
  26. // AS Access Security System
  27. //
  28. // https://learn.microsoft.com/en-us/windows/win32/secauthz/access-mask
  29. //
  30. // https://learn.microsoft.com/en-us/windows/win32/secauthz/generic-access-rights
  31. //
  32. // https://learn.microsoft.com/en-us/windows/win32/fileio/file-access-rights-constants
  33. type AccessMask = windows.ACCESS_MASK
  34. //nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
  35. const (
  36. // Not actually any.
  37. //
  38. // For CreateFile: "query certain metadata such as file, directory, or device attributes without accessing that file or device"
  39. // https://learn.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-createfilew#parameters
  40. FILE_ANY_ACCESS AccessMask = 0
  41. // Specific Object Access
  42. // from ntioapi.h
  43. FILE_READ_DATA AccessMask = (0x0001) // file & pipe
  44. FILE_LIST_DIRECTORY AccessMask = (0x0001) // directory
  45. FILE_WRITE_DATA AccessMask = (0x0002) // file & pipe
  46. FILE_ADD_FILE AccessMask = (0x0002) // directory
  47. FILE_APPEND_DATA AccessMask = (0x0004) // file
  48. FILE_ADD_SUBDIRECTORY AccessMask = (0x0004) // directory
  49. FILE_CREATE_PIPE_INSTANCE AccessMask = (0x0004) // named pipe
  50. FILE_READ_EA AccessMask = (0x0008) // file & directory
  51. FILE_READ_PROPERTIES AccessMask = FILE_READ_EA
  52. FILE_WRITE_EA AccessMask = (0x0010) // file & directory
  53. FILE_WRITE_PROPERTIES AccessMask = FILE_WRITE_EA
  54. FILE_EXECUTE AccessMask = (0x0020) // file
  55. FILE_TRAVERSE AccessMask = (0x0020) // directory
  56. FILE_DELETE_CHILD AccessMask = (0x0040) // directory
  57. FILE_READ_ATTRIBUTES AccessMask = (0x0080) // all
  58. FILE_WRITE_ATTRIBUTES AccessMask = (0x0100) // all
  59. FILE_ALL_ACCESS AccessMask = (STANDARD_RIGHTS_REQUIRED | SYNCHRONIZE | 0x1FF)
  60. FILE_GENERIC_READ AccessMask = (STANDARD_RIGHTS_READ | FILE_READ_DATA | FILE_READ_ATTRIBUTES | FILE_READ_EA | SYNCHRONIZE)
  61. FILE_GENERIC_WRITE AccessMask = (STANDARD_RIGHTS_WRITE | FILE_WRITE_DATA | FILE_WRITE_ATTRIBUTES | FILE_WRITE_EA | FILE_APPEND_DATA | SYNCHRONIZE)
  62. FILE_GENERIC_EXECUTE AccessMask = (STANDARD_RIGHTS_EXECUTE | FILE_READ_ATTRIBUTES | FILE_EXECUTE | SYNCHRONIZE)
  63. SPECIFIC_RIGHTS_ALL AccessMask = 0x0000FFFF
  64. // Standard Access
  65. // from ntseapi.h
  66. DELETE AccessMask = 0x0001_0000
  67. READ_CONTROL AccessMask = 0x0002_0000
  68. WRITE_DAC AccessMask = 0x0004_0000
  69. WRITE_OWNER AccessMask = 0x0008_0000
  70. SYNCHRONIZE AccessMask = 0x0010_0000
  71. STANDARD_RIGHTS_REQUIRED AccessMask = 0x000F_0000
  72. STANDARD_RIGHTS_READ AccessMask = READ_CONTROL
  73. STANDARD_RIGHTS_WRITE AccessMask = READ_CONTROL
  74. STANDARD_RIGHTS_EXECUTE AccessMask = READ_CONTROL
  75. STANDARD_RIGHTS_ALL AccessMask = 0x001F_0000
  76. )
  77. type FileShareMode uint32
  78. //nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
  79. const (
  80. FILE_SHARE_NONE FileShareMode = 0x00
  81. FILE_SHARE_READ FileShareMode = 0x01
  82. FILE_SHARE_WRITE FileShareMode = 0x02
  83. FILE_SHARE_DELETE FileShareMode = 0x04
  84. FILE_SHARE_VALID_FLAGS FileShareMode = 0x07
  85. )
  86. type FileCreationDisposition uint32
  87. //nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
  88. const (
  89. // from winbase.h
  90. CREATE_NEW FileCreationDisposition = 0x01
  91. CREATE_ALWAYS FileCreationDisposition = 0x02
  92. OPEN_EXISTING FileCreationDisposition = 0x03
  93. OPEN_ALWAYS FileCreationDisposition = 0x04
  94. TRUNCATE_EXISTING FileCreationDisposition = 0x05
  95. )
  96. // CreateFile and co. take flags or attributes together as one parameter.
  97. // Define alias until we can use generics to allow both
  98. // https://learn.microsoft.com/en-us/windows/win32/fileio/file-attribute-constants
  99. type FileFlagOrAttribute uint32
  100. //nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
  101. const ( // from winnt.h
  102. FILE_FLAG_WRITE_THROUGH FileFlagOrAttribute = 0x8000_0000
  103. FILE_FLAG_OVERLAPPED FileFlagOrAttribute = 0x4000_0000
  104. FILE_FLAG_NO_BUFFERING FileFlagOrAttribute = 0x2000_0000
  105. FILE_FLAG_RANDOM_ACCESS FileFlagOrAttribute = 0x1000_0000
  106. FILE_FLAG_SEQUENTIAL_SCAN FileFlagOrAttribute = 0x0800_0000
  107. FILE_FLAG_DELETE_ON_CLOSE FileFlagOrAttribute = 0x0400_0000
  108. FILE_FLAG_BACKUP_SEMANTICS FileFlagOrAttribute = 0x0200_0000
  109. FILE_FLAG_POSIX_SEMANTICS FileFlagOrAttribute = 0x0100_0000
  110. FILE_FLAG_OPEN_REPARSE_POINT FileFlagOrAttribute = 0x0020_0000
  111. FILE_FLAG_OPEN_NO_RECALL FileFlagOrAttribute = 0x0010_0000
  112. FILE_FLAG_FIRST_PIPE_INSTANCE FileFlagOrAttribute = 0x0008_0000
  113. )
  114. type FileSQSFlag = FileFlagOrAttribute
  115. //nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
  116. const ( // from winbase.h
  117. SECURITY_ANONYMOUS FileSQSFlag = FileSQSFlag(SecurityAnonymous << 16)
  118. SECURITY_IDENTIFICATION FileSQSFlag = FileSQSFlag(SecurityIdentification << 16)
  119. SECURITY_IMPERSONATION FileSQSFlag = FileSQSFlag(SecurityImpersonation << 16)
  120. SECURITY_DELEGATION FileSQSFlag = FileSQSFlag(SecurityDelegation << 16)
  121. SECURITY_SQOS_PRESENT FileSQSFlag = 0x00100000
  122. SECURITY_VALID_SQOS_FLAGS FileSQSFlag = 0x001F0000
  123. )
  124. // GetFinalPathNameByHandle flags
  125. //
  126. // https://learn.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-getfinalpathnamebyhandlew#parameters
  127. type GetFinalPathFlag uint32
  128. //nolint:revive // SNAKE_CASE is not idiomatic in Go, but aligned with Win32 API.
  129. const (
  130. GetFinalPathDefaultFlag GetFinalPathFlag = 0x0
  131. FILE_NAME_NORMALIZED GetFinalPathFlag = 0x0
  132. FILE_NAME_OPENED GetFinalPathFlag = 0x8
  133. VOLUME_NAME_DOS GetFinalPathFlag = 0x0
  134. VOLUME_NAME_GUID GetFinalPathFlag = 0x1
  135. VOLUME_NAME_NT GetFinalPathFlag = 0x2
  136. VOLUME_NAME_NONE GetFinalPathFlag = 0x4
  137. )
  138. // getFinalPathNameByHandle facilitates calling the Windows API GetFinalPathNameByHandle
  139. // with the given handle and flags. It transparently takes care of creating a buffer of the
  140. // correct size for the call.
  141. //
  142. // https://learn.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-getfinalpathnamebyhandlew
  143. func GetFinalPathNameByHandle(h windows.Handle, flags GetFinalPathFlag) (string, error) {
  144. b := stringbuffer.NewWString()
  145. //TODO: can loop infinitely if Win32 keeps returning the same (or a larger) n?
  146. for {
  147. n, err := windows.GetFinalPathNameByHandle(h, b.Pointer(), b.Cap(), uint32(flags))
  148. if err != nil {
  149. return "", err
  150. }
  151. // If the buffer wasn't large enough, n will be the total size needed (including null terminator).
  152. // Resize and try again.
  153. if n > b.Cap() {
  154. b.ResizeTo(n)
  155. continue
  156. }
  157. // If the buffer is large enough, n will be the size not including the null terminator.
  158. // Convert to a Go string and return.
  159. return b.String(), nil
  160. }
  161. }