You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

1809 lines
79 KiB

  1. // Copyright 2009 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. // Windows system calls.
  5. package windows
  6. import (
  7. errorspkg "errors"
  8. "fmt"
  9. "runtime"
  10. "sync"
  11. "syscall"
  12. "time"
  13. "unicode/utf16"
  14. "unsafe"
  15. "golang.org/x/sys/internal/unsafeheader"
  16. )
  17. type Handle uintptr
  18. type HWND uintptr
  19. const (
  20. InvalidHandle = ^Handle(0)
  21. InvalidHWND = ^HWND(0)
  22. // Flags for DefineDosDevice.
  23. DDD_EXACT_MATCH_ON_REMOVE = 0x00000004
  24. DDD_NO_BROADCAST_SYSTEM = 0x00000008
  25. DDD_RAW_TARGET_PATH = 0x00000001
  26. DDD_REMOVE_DEFINITION = 0x00000002
  27. // Return values for GetDriveType.
  28. DRIVE_UNKNOWN = 0
  29. DRIVE_NO_ROOT_DIR = 1
  30. DRIVE_REMOVABLE = 2
  31. DRIVE_FIXED = 3
  32. DRIVE_REMOTE = 4
  33. DRIVE_CDROM = 5
  34. DRIVE_RAMDISK = 6
  35. // File system flags from GetVolumeInformation and GetVolumeInformationByHandle.
  36. FILE_CASE_SENSITIVE_SEARCH = 0x00000001
  37. FILE_CASE_PRESERVED_NAMES = 0x00000002
  38. FILE_FILE_COMPRESSION = 0x00000010
  39. FILE_DAX_VOLUME = 0x20000000
  40. FILE_NAMED_STREAMS = 0x00040000
  41. FILE_PERSISTENT_ACLS = 0x00000008
  42. FILE_READ_ONLY_VOLUME = 0x00080000
  43. FILE_SEQUENTIAL_WRITE_ONCE = 0x00100000
  44. FILE_SUPPORTS_ENCRYPTION = 0x00020000
  45. FILE_SUPPORTS_EXTENDED_ATTRIBUTES = 0x00800000
  46. FILE_SUPPORTS_HARD_LINKS = 0x00400000
  47. FILE_SUPPORTS_OBJECT_IDS = 0x00010000
  48. FILE_SUPPORTS_OPEN_BY_FILE_ID = 0x01000000
  49. FILE_SUPPORTS_REPARSE_POINTS = 0x00000080
  50. FILE_SUPPORTS_SPARSE_FILES = 0x00000040
  51. FILE_SUPPORTS_TRANSACTIONS = 0x00200000
  52. FILE_SUPPORTS_USN_JOURNAL = 0x02000000
  53. FILE_UNICODE_ON_DISK = 0x00000004
  54. FILE_VOLUME_IS_COMPRESSED = 0x00008000
  55. FILE_VOLUME_QUOTAS = 0x00000020
  56. // Flags for LockFileEx.
  57. LOCKFILE_FAIL_IMMEDIATELY = 0x00000001
  58. LOCKFILE_EXCLUSIVE_LOCK = 0x00000002
  59. // Return value of SleepEx and other APC functions
  60. WAIT_IO_COMPLETION = 0x000000C0
  61. )
  62. // StringToUTF16 is deprecated. Use UTF16FromString instead.
  63. // If s contains a NUL byte this function panics instead of
  64. // returning an error.
  65. func StringToUTF16(s string) []uint16 {
  66. a, err := UTF16FromString(s)
  67. if err != nil {
  68. panic("windows: string with NUL passed to StringToUTF16")
  69. }
  70. return a
  71. }
  72. // UTF16FromString returns the UTF-16 encoding of the UTF-8 string
  73. // s, with a terminating NUL added. If s contains a NUL byte at any
  74. // location, it returns (nil, syscall.EINVAL).
  75. func UTF16FromString(s string) ([]uint16, error) {
  76. return syscall.UTF16FromString(s)
  77. }
  78. // UTF16ToString returns the UTF-8 encoding of the UTF-16 sequence s,
  79. // with a terminating NUL and any bytes after the NUL removed.
  80. func UTF16ToString(s []uint16) string {
  81. return syscall.UTF16ToString(s)
  82. }
  83. // StringToUTF16Ptr is deprecated. Use UTF16PtrFromString instead.
  84. // If s contains a NUL byte this function panics instead of
  85. // returning an error.
  86. func StringToUTF16Ptr(s string) *uint16 { return &StringToUTF16(s)[0] }
  87. // UTF16PtrFromString returns pointer to the UTF-16 encoding of
  88. // the UTF-8 string s, with a terminating NUL added. If s
  89. // contains a NUL byte at any location, it returns (nil, syscall.EINVAL).
  90. func UTF16PtrFromString(s string) (*uint16, error) {
  91. a, err := UTF16FromString(s)
  92. if err != nil {
  93. return nil, err
  94. }
  95. return &a[0], nil
  96. }
  97. // UTF16PtrToString takes a pointer to a UTF-16 sequence and returns the corresponding UTF-8 encoded string.
  98. // If the pointer is nil, it returns the empty string. It assumes that the UTF-16 sequence is terminated
  99. // at a zero word; if the zero word is not present, the program may crash.
  100. func UTF16PtrToString(p *uint16) string {
  101. if p == nil {
  102. return ""
  103. }
  104. if *p == 0 {
  105. return ""
  106. }
  107. // Find NUL terminator.
  108. n := 0
  109. for ptr := unsafe.Pointer(p); *(*uint16)(ptr) != 0; n++ {
  110. ptr = unsafe.Pointer(uintptr(ptr) + unsafe.Sizeof(*p))
  111. }
  112. return string(utf16.Decode(unsafe.Slice(p, n)))
  113. }
  114. func Getpagesize() int { return 4096 }
  115. // NewCallback converts a Go function to a function pointer conforming to the stdcall calling convention.
  116. // This is useful when interoperating with Windows code requiring callbacks.
  117. // The argument is expected to be a function with with one uintptr-sized result. The function must not have arguments with size larger than the size of uintptr.
  118. func NewCallback(fn interface{}) uintptr {
  119. return syscall.NewCallback(fn)
  120. }
  121. // NewCallbackCDecl converts a Go function to a function pointer conforming to the cdecl calling convention.
  122. // This is useful when interoperating with Windows code requiring callbacks.
  123. // The argument is expected to be a function with with one uintptr-sized result. The function must not have arguments with size larger than the size of uintptr.
  124. func NewCallbackCDecl(fn interface{}) uintptr {
  125. return syscall.NewCallbackCDecl(fn)
  126. }
  127. // windows api calls
  128. //sys GetLastError() (lasterr error)
  129. //sys LoadLibrary(libname string) (handle Handle, err error) = LoadLibraryW
  130. //sys LoadLibraryEx(libname string, zero Handle, flags uintptr) (handle Handle, err error) = LoadLibraryExW
  131. //sys FreeLibrary(handle Handle) (err error)
  132. //sys GetProcAddress(module Handle, procname string) (proc uintptr, err error)
  133. //sys GetModuleFileName(module Handle, filename *uint16, size uint32) (n uint32, err error) = kernel32.GetModuleFileNameW
  134. //sys GetModuleHandleEx(flags uint32, moduleName *uint16, module *Handle) (err error) = kernel32.GetModuleHandleExW
  135. //sys SetDefaultDllDirectories(directoryFlags uint32) (err error)
  136. //sys SetDllDirectory(path string) (err error) = kernel32.SetDllDirectoryW
  137. //sys GetVersion() (ver uint32, err error)
  138. //sys FormatMessage(flags uint32, msgsrc uintptr, msgid uint32, langid uint32, buf []uint16, args *byte) (n uint32, err error) = FormatMessageW
  139. //sys ExitProcess(exitcode uint32)
  140. //sys IsWow64Process(handle Handle, isWow64 *bool) (err error) = IsWow64Process
  141. //sys IsWow64Process2(handle Handle, processMachine *uint16, nativeMachine *uint16) (err error) = IsWow64Process2?
  142. //sys CreateFile(name *uint16, access uint32, mode uint32, sa *SecurityAttributes, createmode uint32, attrs uint32, templatefile Handle) (handle Handle, err error) [failretval==InvalidHandle] = CreateFileW
  143. //sys CreateNamedPipe(name *uint16, flags uint32, pipeMode uint32, maxInstances uint32, outSize uint32, inSize uint32, defaultTimeout uint32, sa *SecurityAttributes) (handle Handle, err error) [failretval==InvalidHandle] = CreateNamedPipeW
  144. //sys ConnectNamedPipe(pipe Handle, overlapped *Overlapped) (err error)
  145. //sys GetNamedPipeInfo(pipe Handle, flags *uint32, outSize *uint32, inSize *uint32, maxInstances *uint32) (err error)
  146. //sys GetNamedPipeHandleState(pipe Handle, state *uint32, curInstances *uint32, maxCollectionCount *uint32, collectDataTimeout *uint32, userName *uint16, maxUserNameSize uint32) (err error) = GetNamedPipeHandleStateW
  147. //sys SetNamedPipeHandleState(pipe Handle, state *uint32, maxCollectionCount *uint32, collectDataTimeout *uint32) (err error) = SetNamedPipeHandleState
  148. //sys readFile(handle Handle, buf []byte, done *uint32, overlapped *Overlapped) (err error) = ReadFile
  149. //sys writeFile(handle Handle, buf []byte, done *uint32, overlapped *Overlapped) (err error) = WriteFile
  150. //sys GetOverlappedResult(handle Handle, overlapped *Overlapped, done *uint32, wait bool) (err error)
  151. //sys SetFilePointer(handle Handle, lowoffset int32, highoffsetptr *int32, whence uint32) (newlowoffset uint32, err error) [failretval==0xffffffff]
  152. //sys CloseHandle(handle Handle) (err error)
  153. //sys GetStdHandle(stdhandle uint32) (handle Handle, err error) [failretval==InvalidHandle]
  154. //sys SetStdHandle(stdhandle uint32, handle Handle) (err error)
  155. //sys findFirstFile1(name *uint16, data *win32finddata1) (handle Handle, err error) [failretval==InvalidHandle] = FindFirstFileW
  156. //sys findNextFile1(handle Handle, data *win32finddata1) (err error) = FindNextFileW
  157. //sys FindClose(handle Handle) (err error)
  158. //sys GetFileInformationByHandle(handle Handle, data *ByHandleFileInformation) (err error)
  159. //sys GetFileInformationByHandleEx(handle Handle, class uint32, outBuffer *byte, outBufferLen uint32) (err error)
  160. //sys SetFileInformationByHandle(handle Handle, class uint32, inBuffer *byte, inBufferLen uint32) (err error)
  161. //sys GetCurrentDirectory(buflen uint32, buf *uint16) (n uint32, err error) = GetCurrentDirectoryW
  162. //sys SetCurrentDirectory(path *uint16) (err error) = SetCurrentDirectoryW
  163. //sys CreateDirectory(path *uint16, sa *SecurityAttributes) (err error) = CreateDirectoryW
  164. //sys RemoveDirectory(path *uint16) (err error) = RemoveDirectoryW
  165. //sys DeleteFile(path *uint16) (err error) = DeleteFileW
  166. //sys MoveFile(from *uint16, to *uint16) (err error) = MoveFileW
  167. //sys MoveFileEx(from *uint16, to *uint16, flags uint32) (err error) = MoveFileExW
  168. //sys LockFileEx(file Handle, flags uint32, reserved uint32, bytesLow uint32, bytesHigh uint32, overlapped *Overlapped) (err error)
  169. //sys UnlockFileEx(file Handle, reserved uint32, bytesLow uint32, bytesHigh uint32, overlapped *Overlapped) (err error)
  170. //sys GetComputerName(buf *uint16, n *uint32) (err error) = GetComputerNameW
  171. //sys GetComputerNameEx(nametype uint32, buf *uint16, n *uint32) (err error) = GetComputerNameExW
  172. //sys SetEndOfFile(handle Handle) (err error)
  173. //sys GetSystemTimeAsFileTime(time *Filetime)
  174. //sys GetSystemTimePreciseAsFileTime(time *Filetime)
  175. //sys GetTimeZoneInformation(tzi *Timezoneinformation) (rc uint32, err error) [failretval==0xffffffff]
  176. //sys CreateIoCompletionPort(filehandle Handle, cphandle Handle, key uintptr, threadcnt uint32) (handle Handle, err error)
  177. //sys GetQueuedCompletionStatus(cphandle Handle, qty *uint32, key *uintptr, overlapped **Overlapped, timeout uint32) (err error)
  178. //sys PostQueuedCompletionStatus(cphandle Handle, qty uint32, key uintptr, overlapped *Overlapped) (err error)
  179. //sys CancelIo(s Handle) (err error)
  180. //sys CancelIoEx(s Handle, o *Overlapped) (err error)
  181. //sys CreateProcess(appName *uint16, commandLine *uint16, procSecurity *SecurityAttributes, threadSecurity *SecurityAttributes, inheritHandles bool, creationFlags uint32, env *uint16, currentDir *uint16, startupInfo *StartupInfo, outProcInfo *ProcessInformation) (err error) = CreateProcessW
  182. //sys CreateProcessAsUser(token Token, appName *uint16, commandLine *uint16, procSecurity *SecurityAttributes, threadSecurity *SecurityAttributes, inheritHandles bool, creationFlags uint32, env *uint16, currentDir *uint16, startupInfo *StartupInfo, outProcInfo *ProcessInformation) (err error) = advapi32.CreateProcessAsUserW
  183. //sys initializeProcThreadAttributeList(attrlist *ProcThreadAttributeList, attrcount uint32, flags uint32, size *uintptr) (err error) = InitializeProcThreadAttributeList
  184. //sys deleteProcThreadAttributeList(attrlist *ProcThreadAttributeList) = DeleteProcThreadAttributeList
  185. //sys updateProcThreadAttribute(attrlist *ProcThreadAttributeList, flags uint32, attr uintptr, value unsafe.Pointer, size uintptr, prevvalue unsafe.Pointer, returnedsize *uintptr) (err error) = UpdateProcThreadAttribute
  186. //sys OpenProcess(desiredAccess uint32, inheritHandle bool, processId uint32) (handle Handle, err error)
  187. //sys ShellExecute(hwnd Handle, verb *uint16, file *uint16, args *uint16, cwd *uint16, showCmd int32) (err error) [failretval<=32] = shell32.ShellExecuteW
  188. //sys GetWindowThreadProcessId(hwnd HWND, pid *uint32) (tid uint32, err error) = user32.GetWindowThreadProcessId
  189. //sys GetShellWindow() (shellWindow HWND) = user32.GetShellWindow
  190. //sys MessageBox(hwnd HWND, text *uint16, caption *uint16, boxtype uint32) (ret int32, err error) [failretval==0] = user32.MessageBoxW
  191. //sys ExitWindowsEx(flags uint32, reason uint32) (err error) = user32.ExitWindowsEx
  192. //sys shGetKnownFolderPath(id *KNOWNFOLDERID, flags uint32, token Token, path **uint16) (ret error) = shell32.SHGetKnownFolderPath
  193. //sys TerminateProcess(handle Handle, exitcode uint32) (err error)
  194. //sys GetExitCodeProcess(handle Handle, exitcode *uint32) (err error)
  195. //sys GetStartupInfo(startupInfo *StartupInfo) (err error) = GetStartupInfoW
  196. //sys GetProcessTimes(handle Handle, creationTime *Filetime, exitTime *Filetime, kernelTime *Filetime, userTime *Filetime) (err error)
  197. //sys DuplicateHandle(hSourceProcessHandle Handle, hSourceHandle Handle, hTargetProcessHandle Handle, lpTargetHandle *Handle, dwDesiredAccess uint32, bInheritHandle bool, dwOptions uint32) (err error)
  198. //sys WaitForSingleObject(handle Handle, waitMilliseconds uint32) (event uint32, err error) [failretval==0xffffffff]
  199. //sys waitForMultipleObjects(count uint32, handles uintptr, waitAll bool, waitMilliseconds uint32) (event uint32, err error) [failretval==0xffffffff] = WaitForMultipleObjects
  200. //sys GetTempPath(buflen uint32, buf *uint16) (n uint32, err error) = GetTempPathW
  201. //sys CreatePipe(readhandle *Handle, writehandle *Handle, sa *SecurityAttributes, size uint32) (err error)
  202. //sys GetFileType(filehandle Handle) (n uint32, err error)
  203. //sys CryptAcquireContext(provhandle *Handle, container *uint16, provider *uint16, provtype uint32, flags uint32) (err error) = advapi32.CryptAcquireContextW
  204. //sys CryptReleaseContext(provhandle Handle, flags uint32) (err error) = advapi32.CryptReleaseContext
  205. //sys CryptGenRandom(provhandle Handle, buflen uint32, buf *byte) (err error) = advapi32.CryptGenRandom
  206. //sys GetEnvironmentStrings() (envs *uint16, err error) [failretval==nil] = kernel32.GetEnvironmentStringsW
  207. //sys FreeEnvironmentStrings(envs *uint16) (err error) = kernel32.FreeEnvironmentStringsW
  208. //sys GetEnvironmentVariable(name *uint16, buffer *uint16, size uint32) (n uint32, err error) = kernel32.GetEnvironmentVariableW
  209. //sys SetEnvironmentVariable(name *uint16, value *uint16) (err error) = kernel32.SetEnvironmentVariableW
  210. //sys ExpandEnvironmentStrings(src *uint16, dst *uint16, size uint32) (n uint32, err error) = kernel32.ExpandEnvironmentStringsW
  211. //sys CreateEnvironmentBlock(block **uint16, token Token, inheritExisting bool) (err error) = userenv.CreateEnvironmentBlock
  212. //sys DestroyEnvironmentBlock(block *uint16) (err error) = userenv.DestroyEnvironmentBlock
  213. //sys getTickCount64() (ms uint64) = kernel32.GetTickCount64
  214. //sys SetFileTime(handle Handle, ctime *Filetime, atime *Filetime, wtime *Filetime) (err error)
  215. //sys GetFileAttributes(name *uint16) (attrs uint32, err error) [failretval==INVALID_FILE_ATTRIBUTES] = kernel32.GetFileAttributesW
  216. //sys SetFileAttributes(name *uint16, attrs uint32) (err error) = kernel32.SetFileAttributesW
  217. //sys GetFileAttributesEx(name *uint16, level uint32, info *byte) (err error) = kernel32.GetFileAttributesExW
  218. //sys GetCommandLine() (cmd *uint16) = kernel32.GetCommandLineW
  219. //sys CommandLineToArgv(cmd *uint16, argc *int32) (argv *[8192]*[8192]uint16, err error) [failretval==nil] = shell32.CommandLineToArgvW
  220. //sys LocalFree(hmem Handle) (handle Handle, err error) [failretval!=0]
  221. //sys LocalAlloc(flags uint32, length uint32) (ptr uintptr, err error)
  222. //sys SetHandleInformation(handle Handle, mask uint32, flags uint32) (err error)
  223. //sys FlushFileBuffers(handle Handle) (err error)
  224. //sys GetFullPathName(path *uint16, buflen uint32, buf *uint16, fname **uint16) (n uint32, err error) = kernel32.GetFullPathNameW
  225. //sys GetLongPathName(path *uint16, buf *uint16, buflen uint32) (n uint32, err error) = kernel32.GetLongPathNameW
  226. //sys GetShortPathName(longpath *uint16, shortpath *uint16, buflen uint32) (n uint32, err error) = kernel32.GetShortPathNameW
  227. //sys GetFinalPathNameByHandle(file Handle, filePath *uint16, filePathSize uint32, flags uint32) (n uint32, err error) = kernel32.GetFinalPathNameByHandleW
  228. //sys CreateFileMapping(fhandle Handle, sa *SecurityAttributes, prot uint32, maxSizeHigh uint32, maxSizeLow uint32, name *uint16) (handle Handle, err error) [failretval == 0 || e1 == ERROR_ALREADY_EXISTS] = kernel32.CreateFileMappingW
  229. //sys MapViewOfFile(handle Handle, access uint32, offsetHigh uint32, offsetLow uint32, length uintptr) (addr uintptr, err error)
  230. //sys UnmapViewOfFile(addr uintptr) (err error)
  231. //sys FlushViewOfFile(addr uintptr, length uintptr) (err error)
  232. //sys VirtualLock(addr uintptr, length uintptr) (err error)
  233. //sys VirtualUnlock(addr uintptr, length uintptr) (err error)
  234. //sys VirtualAlloc(address uintptr, size uintptr, alloctype uint32, protect uint32) (value uintptr, err error) = kernel32.VirtualAlloc
  235. //sys VirtualFree(address uintptr, size uintptr, freetype uint32) (err error) = kernel32.VirtualFree
  236. //sys VirtualProtect(address uintptr, size uintptr, newprotect uint32, oldprotect *uint32) (err error) = kernel32.VirtualProtect
  237. //sys VirtualProtectEx(process Handle, address uintptr, size uintptr, newProtect uint32, oldProtect *uint32) (err error) = kernel32.VirtualProtectEx
  238. //sys VirtualQuery(address uintptr, buffer *MemoryBasicInformation, length uintptr) (err error) = kernel32.VirtualQuery
  239. //sys VirtualQueryEx(process Handle, address uintptr, buffer *MemoryBasicInformation, length uintptr) (err error) = kernel32.VirtualQueryEx
  240. //sys ReadProcessMemory(process Handle, baseAddress uintptr, buffer *byte, size uintptr, numberOfBytesRead *uintptr) (err error) = kernel32.ReadProcessMemory
  241. //sys WriteProcessMemory(process Handle, baseAddress uintptr, buffer *byte, size uintptr, numberOfBytesWritten *uintptr) (err error) = kernel32.WriteProcessMemory
  242. //sys TransmitFile(s Handle, handle Handle, bytesToWrite uint32, bytsPerSend uint32, overlapped *Overlapped, transmitFileBuf *TransmitFileBuffers, flags uint32) (err error) = mswsock.TransmitFile
  243. //sys ReadDirectoryChanges(handle Handle, buf *byte, buflen uint32, watchSubTree bool, mask uint32, retlen *uint32, overlapped *Overlapped, completionRoutine uintptr) (err error) = kernel32.ReadDirectoryChangesW
  244. //sys FindFirstChangeNotification(path string, watchSubtree bool, notifyFilter uint32) (handle Handle, err error) [failretval==InvalidHandle] = kernel32.FindFirstChangeNotificationW
  245. //sys FindNextChangeNotification(handle Handle) (err error)
  246. //sys FindCloseChangeNotification(handle Handle) (err error)
  247. //sys CertOpenSystemStore(hprov Handle, name *uint16) (store Handle, err error) = crypt32.CertOpenSystemStoreW
  248. //sys CertOpenStore(storeProvider uintptr, msgAndCertEncodingType uint32, cryptProv uintptr, flags uint32, para uintptr) (handle Handle, err error) = crypt32.CertOpenStore
  249. //sys CertEnumCertificatesInStore(store Handle, prevContext *CertContext) (context *CertContext, err error) [failretval==nil] = crypt32.CertEnumCertificatesInStore
  250. //sys CertAddCertificateContextToStore(store Handle, certContext *CertContext, addDisposition uint32, storeContext **CertContext) (err error) = crypt32.CertAddCertificateContextToStore
  251. //sys CertCloseStore(store Handle, flags uint32) (err error) = crypt32.CertCloseStore
  252. //sys CertDeleteCertificateFromStore(certContext *CertContext) (err error) = crypt32.CertDeleteCertificateFromStore
  253. //sys CertDuplicateCertificateContext(certContext *CertContext) (dupContext *CertContext) = crypt32.CertDuplicateCertificateContext
  254. //sys PFXImportCertStore(pfx *CryptDataBlob, password *uint16, flags uint32) (store Handle, err error) = crypt32.PFXImportCertStore
  255. //sys CertGetCertificateChain(engine Handle, leaf *CertContext, time *Filetime, additionalStore Handle, para *CertChainPara, flags uint32, reserved uintptr, chainCtx **CertChainContext) (err error) = crypt32.CertGetCertificateChain
  256. //sys CertFreeCertificateChain(ctx *CertChainContext) = crypt32.CertFreeCertificateChain
  257. //sys CertCreateCertificateContext(certEncodingType uint32, certEncoded *byte, encodedLen uint32) (context *CertContext, err error) [failretval==nil] = crypt32.CertCreateCertificateContext
  258. //sys CertFreeCertificateContext(ctx *CertContext) (err error) = crypt32.CertFreeCertificateContext
  259. //sys CertVerifyCertificateChainPolicy(policyOID uintptr, chain *CertChainContext, para *CertChainPolicyPara, status *CertChainPolicyStatus) (err error) = crypt32.CertVerifyCertificateChainPolicy
  260. //sys CertGetNameString(certContext *CertContext, nameType uint32, flags uint32, typePara unsafe.Pointer, name *uint16, size uint32) (chars uint32) = crypt32.CertGetNameStringW
  261. //sys CertFindExtension(objId *byte, countExtensions uint32, extensions *CertExtension) (ret *CertExtension) = crypt32.CertFindExtension
  262. //sys CertFindCertificateInStore(store Handle, certEncodingType uint32, findFlags uint32, findType uint32, findPara unsafe.Pointer, prevCertContext *CertContext) (cert *CertContext, err error) [failretval==nil] = crypt32.CertFindCertificateInStore
  263. //sys CertFindChainInStore(store Handle, certEncodingType uint32, findFlags uint32, findType uint32, findPara unsafe.Pointer, prevChainContext *CertChainContext) (certchain *CertChainContext, err error) [failretval==nil] = crypt32.CertFindChainInStore
  264. //sys CryptAcquireCertificatePrivateKey(cert *CertContext, flags uint32, parameters unsafe.Pointer, cryptProvOrNCryptKey *Handle, keySpec *uint32, callerFreeProvOrNCryptKey *bool) (err error) = crypt32.CryptAcquireCertificatePrivateKey
  265. //sys CryptQueryObject(objectType uint32, object unsafe.Pointer, expectedContentTypeFlags uint32, expectedFormatTypeFlags uint32, flags uint32, msgAndCertEncodingType *uint32, contentType *uint32, formatType *uint32, certStore *Handle, msg *Handle, context *unsafe.Pointer) (err error) = crypt32.CryptQueryObject
  266. //sys CryptDecodeObject(encodingType uint32, structType *byte, encodedBytes *byte, lenEncodedBytes uint32, flags uint32, decoded unsafe.Pointer, decodedLen *uint32) (err error) = crypt32.CryptDecodeObject
  267. //sys CryptProtectData(dataIn *DataBlob, name *uint16, optionalEntropy *DataBlob, reserved uintptr, promptStruct *CryptProtectPromptStruct, flags uint32, dataOut *DataBlob) (err error) = crypt32.CryptProtectData
  268. //sys CryptUnprotectData(dataIn *DataBlob, name **uint16, optionalEntropy *DataBlob, reserved uintptr, promptStruct *CryptProtectPromptStruct, flags uint32, dataOut *DataBlob) (err error) = crypt32.CryptUnprotectData
  269. //sys WinVerifyTrustEx(hwnd HWND, actionId *GUID, data *WinTrustData) (ret error) = wintrust.WinVerifyTrustEx
  270. //sys RegOpenKeyEx(key Handle, subkey *uint16, options uint32, desiredAccess uint32, result *Handle) (regerrno error) = advapi32.RegOpenKeyExW
  271. //sys RegCloseKey(key Handle) (regerrno error) = advapi32.RegCloseKey
  272. //sys RegQueryInfoKey(key Handle, class *uint16, classLen *uint32, reserved *uint32, subkeysLen *uint32, maxSubkeyLen *uint32, maxClassLen *uint32, valuesLen *uint32, maxValueNameLen *uint32, maxValueLen *uint32, saLen *uint32, lastWriteTime *Filetime) (regerrno error) = advapi32.RegQueryInfoKeyW
  273. //sys RegEnumKeyEx(key Handle, index uint32, name *uint16, nameLen *uint32, reserved *uint32, class *uint16, classLen *uint32, lastWriteTime *Filetime) (regerrno error) = advapi32.RegEnumKeyExW
  274. //sys RegQueryValueEx(key Handle, name *uint16, reserved *uint32, valtype *uint32, buf *byte, buflen *uint32) (regerrno error) = advapi32.RegQueryValueExW
  275. //sys RegNotifyChangeKeyValue(key Handle, watchSubtree bool, notifyFilter uint32, event Handle, asynchronous bool) (regerrno error) = advapi32.RegNotifyChangeKeyValue
  276. //sys GetCurrentProcessId() (pid uint32) = kernel32.GetCurrentProcessId
  277. //sys ProcessIdToSessionId(pid uint32, sessionid *uint32) (err error) = kernel32.ProcessIdToSessionId
  278. //sys GetConsoleMode(console Handle, mode *uint32) (err error) = kernel32.GetConsoleMode
  279. //sys SetConsoleMode(console Handle, mode uint32) (err error) = kernel32.SetConsoleMode
  280. //sys GetConsoleScreenBufferInfo(console Handle, info *ConsoleScreenBufferInfo) (err error) = kernel32.GetConsoleScreenBufferInfo
  281. //sys setConsoleCursorPosition(console Handle, position uint32) (err error) = kernel32.SetConsoleCursorPosition
  282. //sys WriteConsole(console Handle, buf *uint16, towrite uint32, written *uint32, reserved *byte) (err error) = kernel32.WriteConsoleW
  283. //sys ReadConsole(console Handle, buf *uint16, toread uint32, read *uint32, inputControl *byte) (err error) = kernel32.ReadConsoleW
  284. //sys CreateToolhelp32Snapshot(flags uint32, processId uint32) (handle Handle, err error) [failretval==InvalidHandle] = kernel32.CreateToolhelp32Snapshot
  285. //sys Module32First(snapshot Handle, moduleEntry *ModuleEntry32) (err error) = kernel32.Module32FirstW
  286. //sys Module32Next(snapshot Handle, moduleEntry *ModuleEntry32) (err error) = kernel32.Module32NextW
  287. //sys Process32First(snapshot Handle, procEntry *ProcessEntry32) (err error) = kernel32.Process32FirstW
  288. //sys Process32Next(snapshot Handle, procEntry *ProcessEntry32) (err error) = kernel32.Process32NextW
  289. //sys Thread32First(snapshot Handle, threadEntry *ThreadEntry32) (err error)
  290. //sys Thread32Next(snapshot Handle, threadEntry *ThreadEntry32) (err error)
  291. //sys DeviceIoControl(handle Handle, ioControlCode uint32, inBuffer *byte, inBufferSize uint32, outBuffer *byte, outBufferSize uint32, bytesReturned *uint32, overlapped *Overlapped) (err error)
  292. // This function returns 1 byte BOOLEAN rather than the 4 byte BOOL.
  293. //sys CreateSymbolicLink(symlinkfilename *uint16, targetfilename *uint16, flags uint32) (err error) [failretval&0xff==0] = CreateSymbolicLinkW
  294. //sys CreateHardLink(filename *uint16, existingfilename *uint16, reserved uintptr) (err error) [failretval&0xff==0] = CreateHardLinkW
  295. //sys GetCurrentThreadId() (id uint32)
  296. //sys CreateEvent(eventAttrs *SecurityAttributes, manualReset uint32, initialState uint32, name *uint16) (handle Handle, err error) [failretval == 0 || e1 == ERROR_ALREADY_EXISTS] = kernel32.CreateEventW
  297. //sys CreateEventEx(eventAttrs *SecurityAttributes, name *uint16, flags uint32, desiredAccess uint32) (handle Handle, err error) [failretval == 0 || e1 == ERROR_ALREADY_EXISTS] = kernel32.CreateEventExW
  298. //sys OpenEvent(desiredAccess uint32, inheritHandle bool, name *uint16) (handle Handle, err error) = kernel32.OpenEventW
  299. //sys SetEvent(event Handle) (err error) = kernel32.SetEvent
  300. //sys ResetEvent(event Handle) (err error) = kernel32.ResetEvent
  301. //sys PulseEvent(event Handle) (err error) = kernel32.PulseEvent
  302. //sys CreateMutex(mutexAttrs *SecurityAttributes, initialOwner bool, name *uint16) (handle Handle, err error) [failretval == 0 || e1 == ERROR_ALREADY_EXISTS] = kernel32.CreateMutexW
  303. //sys CreateMutexEx(mutexAttrs *SecurityAttributes, name *uint16, flags uint32, desiredAccess uint32) (handle Handle, err error) [failretval == 0 || e1 == ERROR_ALREADY_EXISTS] = kernel32.CreateMutexExW
  304. //sys OpenMutex(desiredAccess uint32, inheritHandle bool, name *uint16) (handle Handle, err error) = kernel32.OpenMutexW
  305. //sys ReleaseMutex(mutex Handle) (err error) = kernel32.ReleaseMutex
  306. //sys SleepEx(milliseconds uint32, alertable bool) (ret uint32) = kernel32.SleepEx
  307. //sys CreateJobObject(jobAttr *SecurityAttributes, name *uint16) (handle Handle, err error) = kernel32.CreateJobObjectW
  308. //sys AssignProcessToJobObject(job Handle, process Handle) (err error) = kernel32.AssignProcessToJobObject
  309. //sys TerminateJobObject(job Handle, exitCode uint32) (err error) = kernel32.TerminateJobObject
  310. //sys SetErrorMode(mode uint32) (ret uint32) = kernel32.SetErrorMode
  311. //sys ResumeThread(thread Handle) (ret uint32, err error) [failretval==0xffffffff] = kernel32.ResumeThread
  312. //sys SetPriorityClass(process Handle, priorityClass uint32) (err error) = kernel32.SetPriorityClass
  313. //sys GetPriorityClass(process Handle) (ret uint32, err error) = kernel32.GetPriorityClass
  314. //sys QueryInformationJobObject(job Handle, JobObjectInformationClass int32, JobObjectInformation uintptr, JobObjectInformationLength uint32, retlen *uint32) (err error) = kernel32.QueryInformationJobObject
  315. //sys SetInformationJobObject(job Handle, JobObjectInformationClass uint32, JobObjectInformation uintptr, JobObjectInformationLength uint32) (ret int, err error)
  316. //sys GenerateConsoleCtrlEvent(ctrlEvent uint32, processGroupID uint32) (err error)
  317. //sys GetProcessId(process Handle) (id uint32, err error)
  318. //sys QueryFullProcessImageName(proc Handle, flags uint32, exeName *uint16, size *uint32) (err error) = kernel32.QueryFullProcessImageNameW
  319. //sys OpenThread(desiredAccess uint32, inheritHandle bool, threadId uint32) (handle Handle, err error)
  320. //sys SetProcessPriorityBoost(process Handle, disable bool) (err error) = kernel32.SetProcessPriorityBoost
  321. //sys GetProcessWorkingSetSizeEx(hProcess Handle, lpMinimumWorkingSetSize *uintptr, lpMaximumWorkingSetSize *uintptr, flags *uint32)
  322. //sys SetProcessWorkingSetSizeEx(hProcess Handle, dwMinimumWorkingSetSize uintptr, dwMaximumWorkingSetSize uintptr, flags uint32) (err error)
  323. //sys GetCommTimeouts(handle Handle, timeouts *CommTimeouts) (err error)
  324. //sys SetCommTimeouts(handle Handle, timeouts *CommTimeouts) (err error)
  325. //sys GetActiveProcessorCount(groupNumber uint16) (ret uint32)
  326. //sys GetMaximumProcessorCount(groupNumber uint16) (ret uint32)
  327. //sys EnumWindows(enumFunc uintptr, param unsafe.Pointer) (err error) = user32.EnumWindows
  328. //sys EnumChildWindows(hwnd HWND, enumFunc uintptr, param unsafe.Pointer) = user32.EnumChildWindows
  329. //sys GetClassName(hwnd HWND, className *uint16, maxCount int32) (copied int32, err error) = user32.GetClassNameW
  330. //sys GetDesktopWindow() (hwnd HWND) = user32.GetDesktopWindow
  331. //sys GetForegroundWindow() (hwnd HWND) = user32.GetForegroundWindow
  332. //sys IsWindow(hwnd HWND) (isWindow bool) = user32.IsWindow
  333. //sys IsWindowUnicode(hwnd HWND) (isUnicode bool) = user32.IsWindowUnicode
  334. //sys IsWindowVisible(hwnd HWND) (isVisible bool) = user32.IsWindowVisible
  335. //sys GetGUIThreadInfo(thread uint32, info *GUIThreadInfo) (err error) = user32.GetGUIThreadInfo
  336. //sys GetLargePageMinimum() (size uintptr)
  337. // Volume Management Functions
  338. //sys DefineDosDevice(flags uint32, deviceName *uint16, targetPath *uint16) (err error) = DefineDosDeviceW
  339. //sys DeleteVolumeMountPoint(volumeMountPoint *uint16) (err error) = DeleteVolumeMountPointW
  340. //sys FindFirstVolume(volumeName *uint16, bufferLength uint32) (handle Handle, err error) [failretval==InvalidHandle] = FindFirstVolumeW
  341. //sys FindFirstVolumeMountPoint(rootPathName *uint16, volumeMountPoint *uint16, bufferLength uint32) (handle Handle, err error) [failretval==InvalidHandle] = FindFirstVolumeMountPointW
  342. //sys FindNextVolume(findVolume Handle, volumeName *uint16, bufferLength uint32) (err error) = FindNextVolumeW
  343. //sys FindNextVolumeMountPoint(findVolumeMountPoint Handle, volumeMountPoint *uint16, bufferLength uint32) (err error) = FindNextVolumeMountPointW
  344. //sys FindVolumeClose(findVolume Handle) (err error)
  345. //sys FindVolumeMountPointClose(findVolumeMountPoint Handle) (err error)
  346. //sys GetDiskFreeSpaceEx(directoryName *uint16, freeBytesAvailableToCaller *uint64, totalNumberOfBytes *uint64, totalNumberOfFreeBytes *uint64) (err error) = GetDiskFreeSpaceExW
  347. //sys GetDriveType(rootPathName *uint16) (driveType uint32) = GetDriveTypeW
  348. //sys GetLogicalDrives() (drivesBitMask uint32, err error) [failretval==0]
  349. //sys GetLogicalDriveStrings(bufferLength uint32, buffer *uint16) (n uint32, err error) [failretval==0] = GetLogicalDriveStringsW
  350. //sys GetVolumeInformation(rootPathName *uint16, volumeNameBuffer *uint16, volumeNameSize uint32, volumeNameSerialNumber *uint32, maximumComponentLength *uint32, fileSystemFlags *uint32, fileSystemNameBuffer *uint16, fileSystemNameSize uint32) (err error) = GetVolumeInformationW
  351. //sys GetVolumeInformationByHandle(file Handle, volumeNameBuffer *uint16, volumeNameSize uint32, volumeNameSerialNumber *uint32, maximumComponentLength *uint32, fileSystemFlags *uint32, fileSystemNameBuffer *uint16, fileSystemNameSize uint32) (err error) = GetVolumeInformationByHandleW
  352. //sys GetVolumeNameForVolumeMountPoint(volumeMountPoint *uint16, volumeName *uint16, bufferlength uint32) (err error) = GetVolumeNameForVolumeMountPointW
  353. //sys GetVolumePathName(fileName *uint16, volumePathName *uint16, bufferLength uint32) (err error) = GetVolumePathNameW
  354. //sys GetVolumePathNamesForVolumeName(volumeName *uint16, volumePathNames *uint16, bufferLength uint32, returnLength *uint32) (err error) = GetVolumePathNamesForVolumeNameW
  355. //sys QueryDosDevice(deviceName *uint16, targetPath *uint16, max uint32) (n uint32, err error) [failretval==0] = QueryDosDeviceW
  356. //sys SetVolumeLabel(rootPathName *uint16, volumeName *uint16) (err error) = SetVolumeLabelW
  357. //sys SetVolumeMountPoint(volumeMountPoint *uint16, volumeName *uint16) (err error) = SetVolumeMountPointW
  358. //sys InitiateSystemShutdownEx(machineName *uint16, message *uint16, timeout uint32, forceAppsClosed bool, rebootAfterShutdown bool, reason uint32) (err error) = advapi32.InitiateSystemShutdownExW
  359. //sys SetProcessShutdownParameters(level uint32, flags uint32) (err error) = kernel32.SetProcessShutdownParameters
  360. //sys GetProcessShutdownParameters(level *uint32, flags *uint32) (err error) = kernel32.GetProcessShutdownParameters
  361. //sys clsidFromString(lpsz *uint16, pclsid *GUID) (ret error) = ole32.CLSIDFromString
  362. //sys stringFromGUID2(rguid *GUID, lpsz *uint16, cchMax int32) (chars int32) = ole32.StringFromGUID2
  363. //sys coCreateGuid(pguid *GUID) (ret error) = ole32.CoCreateGuid
  364. //sys CoTaskMemFree(address unsafe.Pointer) = ole32.CoTaskMemFree
  365. //sys CoInitializeEx(reserved uintptr, coInit uint32) (ret error) = ole32.CoInitializeEx
  366. //sys CoUninitialize() = ole32.CoUninitialize
  367. //sys CoGetObject(name *uint16, bindOpts *BIND_OPTS3, guid *GUID, functionTable **uintptr) (ret error) = ole32.CoGetObject
  368. //sys getProcessPreferredUILanguages(flags uint32, numLanguages *uint32, buf *uint16, bufSize *uint32) (err error) = kernel32.GetProcessPreferredUILanguages
  369. //sys getThreadPreferredUILanguages(flags uint32, numLanguages *uint32, buf *uint16, bufSize *uint32) (err error) = kernel32.GetThreadPreferredUILanguages
  370. //sys getUserPreferredUILanguages(flags uint32, numLanguages *uint32, buf *uint16, bufSize *uint32) (err error) = kernel32.GetUserPreferredUILanguages
  371. //sys getSystemPreferredUILanguages(flags uint32, numLanguages *uint32, buf *uint16, bufSize *uint32) (err error) = kernel32.GetSystemPreferredUILanguages
  372. //sys findResource(module Handle, name uintptr, resType uintptr) (resInfo Handle, err error) = kernel32.FindResourceW
  373. //sys SizeofResource(module Handle, resInfo Handle) (size uint32, err error) = kernel32.SizeofResource
  374. //sys LoadResource(module Handle, resInfo Handle) (resData Handle, err error) = kernel32.LoadResource
  375. //sys LockResource(resData Handle) (addr uintptr, err error) = kernel32.LockResource
  376. // Version APIs
  377. //sys GetFileVersionInfoSize(filename string, zeroHandle *Handle) (bufSize uint32, err error) = version.GetFileVersionInfoSizeW
  378. //sys GetFileVersionInfo(filename string, handle uint32, bufSize uint32, buffer unsafe.Pointer) (err error) = version.GetFileVersionInfoW
  379. //sys VerQueryValue(block unsafe.Pointer, subBlock string, pointerToBufferPointer unsafe.Pointer, bufSize *uint32) (err error) = version.VerQueryValueW
  380. // Process Status API (PSAPI)
  381. //sys EnumProcesses(processIds []uint32, bytesReturned *uint32) (err error) = psapi.EnumProcesses
  382. //sys EnumProcessModules(process Handle, module *Handle, cb uint32, cbNeeded *uint32) (err error) = psapi.EnumProcessModules
  383. //sys EnumProcessModulesEx(process Handle, module *Handle, cb uint32, cbNeeded *uint32, filterFlag uint32) (err error) = psapi.EnumProcessModulesEx
  384. //sys GetModuleInformation(process Handle, module Handle, modinfo *ModuleInfo, cb uint32) (err error) = psapi.GetModuleInformation
  385. //sys GetModuleFileNameEx(process Handle, module Handle, filename *uint16, size uint32) (err error) = psapi.GetModuleFileNameExW
  386. //sys GetModuleBaseName(process Handle, module Handle, baseName *uint16, size uint32) (err error) = psapi.GetModuleBaseNameW
  387. //sys QueryWorkingSetEx(process Handle, pv uintptr, cb uint32) (err error) = psapi.QueryWorkingSetEx
  388. // NT Native APIs
  389. //sys rtlNtStatusToDosErrorNoTeb(ntstatus NTStatus) (ret syscall.Errno) = ntdll.RtlNtStatusToDosErrorNoTeb
  390. //sys rtlGetVersion(info *OsVersionInfoEx) (ntstatus error) = ntdll.RtlGetVersion
  391. //sys rtlGetNtVersionNumbers(majorVersion *uint32, minorVersion *uint32, buildNumber *uint32) = ntdll.RtlGetNtVersionNumbers
  392. //sys RtlGetCurrentPeb() (peb *PEB) = ntdll.RtlGetCurrentPeb
  393. //sys RtlInitUnicodeString(destinationString *NTUnicodeString, sourceString *uint16) = ntdll.RtlInitUnicodeString
  394. //sys RtlInitString(destinationString *NTString, sourceString *byte) = ntdll.RtlInitString
  395. //sys NtCreateFile(handle *Handle, access uint32, oa *OBJECT_ATTRIBUTES, iosb *IO_STATUS_BLOCK, allocationSize *int64, attributes uint32, share uint32, disposition uint32, options uint32, eabuffer uintptr, ealength uint32) (ntstatus error) = ntdll.NtCreateFile
  396. //sys NtCreateNamedPipeFile(pipe *Handle, access uint32, oa *OBJECT_ATTRIBUTES, iosb *IO_STATUS_BLOCK, share uint32, disposition uint32, options uint32, typ uint32, readMode uint32, completionMode uint32, maxInstances uint32, inboundQuota uint32, outputQuota uint32, timeout *int64) (ntstatus error) = ntdll.NtCreateNamedPipeFile
  397. //sys NtSetInformationFile(handle Handle, iosb *IO_STATUS_BLOCK, inBuffer *byte, inBufferLen uint32, class uint32) (ntstatus error) = ntdll.NtSetInformationFile
  398. //sys RtlDosPathNameToNtPathName(dosName *uint16, ntName *NTUnicodeString, ntFileNamePart *uint16, relativeName *RTL_RELATIVE_NAME) (ntstatus error) = ntdll.RtlDosPathNameToNtPathName_U_WithStatus
  399. //sys RtlDosPathNameToRelativeNtPathName(dosName *uint16, ntName *NTUnicodeString, ntFileNamePart *uint16, relativeName *RTL_RELATIVE_NAME) (ntstatus error) = ntdll.RtlDosPathNameToRelativeNtPathName_U_WithStatus
  400. //sys RtlDefaultNpAcl(acl **ACL) (ntstatus error) = ntdll.RtlDefaultNpAcl
  401. //sys NtQueryInformationProcess(proc Handle, procInfoClass int32, procInfo unsafe.Pointer, procInfoLen uint32, retLen *uint32) (ntstatus error) = ntdll.NtQueryInformationProcess
  402. //sys NtSetInformationProcess(proc Handle, procInfoClass int32, procInfo unsafe.Pointer, procInfoLen uint32) (ntstatus error) = ntdll.NtSetInformationProcess
  403. //sys NtQuerySystemInformation(sysInfoClass int32, sysInfo unsafe.Pointer, sysInfoLen uint32, retLen *uint32) (ntstatus error) = ntdll.NtQuerySystemInformation
  404. //sys NtSetSystemInformation(sysInfoClass int32, sysInfo unsafe.Pointer, sysInfoLen uint32) (ntstatus error) = ntdll.NtSetSystemInformation
  405. //sys RtlAddFunctionTable(functionTable *RUNTIME_FUNCTION, entryCount uint32, baseAddress uintptr) (ret bool) = ntdll.RtlAddFunctionTable
  406. //sys RtlDeleteFunctionTable(functionTable *RUNTIME_FUNCTION) (ret bool) = ntdll.RtlDeleteFunctionTable
  407. // Desktop Window Manager API (Dwmapi)
  408. //sys DwmGetWindowAttribute(hwnd HWND, attribute uint32, value unsafe.Pointer, size uint32) (ret error) = dwmapi.DwmGetWindowAttribute
  409. //sys DwmSetWindowAttribute(hwnd HWND, attribute uint32, value unsafe.Pointer, size uint32) (ret error) = dwmapi.DwmSetWindowAttribute
  410. // syscall interface implementation for other packages
  411. // GetCurrentProcess returns the handle for the current process.
  412. // It is a pseudo handle that does not need to be closed.
  413. // The returned error is always nil.
  414. //
  415. // Deprecated: use CurrentProcess for the same Handle without the nil
  416. // error.
  417. func GetCurrentProcess() (Handle, error) {
  418. return CurrentProcess(), nil
  419. }
  420. // CurrentProcess returns the handle for the current process.
  421. // It is a pseudo handle that does not need to be closed.
  422. func CurrentProcess() Handle { return Handle(^uintptr(1 - 1)) }
  423. // GetCurrentThread returns the handle for the current thread.
  424. // It is a pseudo handle that does not need to be closed.
  425. // The returned error is always nil.
  426. //
  427. // Deprecated: use CurrentThread for the same Handle without the nil
  428. // error.
  429. func GetCurrentThread() (Handle, error) {
  430. return CurrentThread(), nil
  431. }
  432. // CurrentThread returns the handle for the current thread.
  433. // It is a pseudo handle that does not need to be closed.
  434. func CurrentThread() Handle { return Handle(^uintptr(2 - 1)) }
  435. // GetProcAddressByOrdinal retrieves the address of the exported
  436. // function from module by ordinal.
  437. func GetProcAddressByOrdinal(module Handle, ordinal uintptr) (proc uintptr, err error) {
  438. r0, _, e1 := syscall.Syscall(procGetProcAddress.Addr(), 2, uintptr(module), ordinal, 0)
  439. proc = uintptr(r0)
  440. if proc == 0 {
  441. err = errnoErr(e1)
  442. }
  443. return
  444. }
  445. func Exit(code int) { ExitProcess(uint32(code)) }
  446. func makeInheritSa() *SecurityAttributes {
  447. var sa SecurityAttributes
  448. sa.Length = uint32(unsafe.Sizeof(sa))
  449. sa.InheritHandle = 1
  450. return &sa
  451. }
  452. func Open(path string, mode int, perm uint32) (fd Handle, err error) {
  453. if len(path) == 0 {
  454. return InvalidHandle, ERROR_FILE_NOT_FOUND
  455. }
  456. pathp, err := UTF16PtrFromString(path)
  457. if err != nil {
  458. return InvalidHandle, err
  459. }
  460. var access uint32
  461. switch mode & (O_RDONLY | O_WRONLY | O_RDWR) {
  462. case O_RDONLY:
  463. access = GENERIC_READ
  464. case O_WRONLY:
  465. access = GENERIC_WRITE
  466. case O_RDWR:
  467. access = GENERIC_READ | GENERIC_WRITE
  468. }
  469. if mode&O_CREAT != 0 {
  470. access |= GENERIC_WRITE
  471. }
  472. if mode&O_APPEND != 0 {
  473. access &^= GENERIC_WRITE
  474. access |= FILE_APPEND_DATA
  475. }
  476. sharemode := uint32(FILE_SHARE_READ | FILE_SHARE_WRITE)
  477. var sa *SecurityAttributes
  478. if mode&O_CLOEXEC == 0 {
  479. sa = makeInheritSa()
  480. }
  481. var createmode uint32
  482. switch {
  483. case mode&(O_CREAT|O_EXCL) == (O_CREAT | O_EXCL):
  484. createmode = CREATE_NEW
  485. case mode&(O_CREAT|O_TRUNC) == (O_CREAT | O_TRUNC):
  486. createmode = CREATE_ALWAYS
  487. case mode&O_CREAT == O_CREAT:
  488. createmode = OPEN_ALWAYS
  489. case mode&O_TRUNC == O_TRUNC:
  490. createmode = TRUNCATE_EXISTING
  491. default:
  492. createmode = OPEN_EXISTING
  493. }
  494. var attrs uint32 = FILE_ATTRIBUTE_NORMAL
  495. if perm&S_IWRITE == 0 {
  496. attrs = FILE_ATTRIBUTE_READONLY
  497. }
  498. h, e := CreateFile(pathp, access, sharemode, sa, createmode, attrs, 0)
  499. return h, e
  500. }
  501. func Read(fd Handle, p []byte) (n int, err error) {
  502. var done uint32
  503. e := ReadFile(fd, p, &done, nil)
  504. if e != nil {
  505. if e == ERROR_BROKEN_PIPE {
  506. // NOTE(brainman): work around ERROR_BROKEN_PIPE is returned on reading EOF from stdin
  507. return 0, nil
  508. }
  509. return 0, e
  510. }
  511. return int(done), nil
  512. }
  513. func Write(fd Handle, p []byte) (n int, err error) {
  514. if raceenabled {
  515. raceReleaseMerge(unsafe.Pointer(&ioSync))
  516. }
  517. var done uint32
  518. e := WriteFile(fd, p, &done, nil)
  519. if e != nil {
  520. return 0, e
  521. }
  522. return int(done), nil
  523. }
  524. func ReadFile(fd Handle, p []byte, done *uint32, overlapped *Overlapped) error {
  525. err := readFile(fd, p, done, overlapped)
  526. if raceenabled {
  527. if *done > 0 {
  528. raceWriteRange(unsafe.Pointer(&p[0]), int(*done))
  529. }
  530. raceAcquire(unsafe.Pointer(&ioSync))
  531. }
  532. return err
  533. }
  534. func WriteFile(fd Handle, p []byte, done *uint32, overlapped *Overlapped) error {
  535. if raceenabled {
  536. raceReleaseMerge(unsafe.Pointer(&ioSync))
  537. }
  538. err := writeFile(fd, p, done, overlapped)
  539. if raceenabled && *done > 0 {
  540. raceReadRange(unsafe.Pointer(&p[0]), int(*done))
  541. }
  542. return err
  543. }
  544. var ioSync int64
  545. func Seek(fd Handle, offset int64, whence int) (newoffset int64, err error) {
  546. var w uint32
  547. switch whence {
  548. case 0:
  549. w = FILE_BEGIN
  550. case 1:
  551. w = FILE_CURRENT
  552. case 2:
  553. w = FILE_END
  554. }
  555. hi := int32(offset >> 32)
  556. lo := int32(offset)
  557. // use GetFileType to check pipe, pipe can't do seek
  558. ft, _ := GetFileType(fd)
  559. if ft == FILE_TYPE_PIPE {
  560. return 0, syscall.EPIPE
  561. }
  562. rlo, e := SetFilePointer(fd, lo, &hi, w)
  563. if e != nil {
  564. return 0, e
  565. }
  566. return int64(hi)<<32 + int64(rlo), nil
  567. }
  568. func Close(fd Handle) (err error) {
  569. return CloseHandle(fd)
  570. }
  571. var (
  572. Stdin = getStdHandle(STD_INPUT_HANDLE)
  573. Stdout = getStdHandle(STD_OUTPUT_HANDLE)
  574. Stderr = getStdHandle(STD_ERROR_HANDLE)
  575. )
  576. func getStdHandle(stdhandle uint32) (fd Handle) {
  577. r, _ := GetStdHandle(stdhandle)
  578. return r
  579. }
  580. const ImplementsGetwd = true
  581. func Getwd() (wd string, err error) {
  582. b := make([]uint16, 300)
  583. n, e := GetCurrentDirectory(uint32(len(b)), &b[0])
  584. if e != nil {
  585. return "", e
  586. }
  587. return string(utf16.Decode(b[0:n])), nil
  588. }
  589. func Chdir(path string) (err error) {
  590. pathp, err := UTF16PtrFromString(path)
  591. if err != nil {
  592. return err
  593. }
  594. return SetCurrentDirectory(pathp)
  595. }
  596. func Mkdir(path string, mode uint32) (err error) {
  597. pathp, err := UTF16PtrFromString(path)
  598. if err != nil {
  599. return err
  600. }
  601. return CreateDirectory(pathp, nil)
  602. }
  603. func Rmdir(path string) (err error) {
  604. pathp, err := UTF16PtrFromString(path)
  605. if err != nil {
  606. return err
  607. }
  608. return RemoveDirectory(pathp)
  609. }
  610. func Unlink(path string) (err error) {
  611. pathp, err := UTF16PtrFromString(path)
  612. if err != nil {
  613. return err
  614. }
  615. return DeleteFile(pathp)
  616. }
  617. func Rename(oldpath, newpath string) (err error) {
  618. from, err := UTF16PtrFromString(oldpath)
  619. if err != nil {
  620. return err
  621. }
  622. to, err := UTF16PtrFromString(newpath)
  623. if err != nil {
  624. return err
  625. }
  626. return MoveFileEx(from, to, MOVEFILE_REPLACE_EXISTING)
  627. }
  628. func ComputerName() (name string, err error) {
  629. var n uint32 = MAX_COMPUTERNAME_LENGTH + 1
  630. b := make([]uint16, n)
  631. e := GetComputerName(&b[0], &n)
  632. if e != nil {
  633. return "", e
  634. }
  635. return string(utf16.Decode(b[0:n])), nil
  636. }
  637. func DurationSinceBoot() time.Duration {
  638. return time.Duration(getTickCount64()) * time.Millisecond
  639. }
  640. func Ftruncate(fd Handle, length int64) (err error) {
  641. curoffset, e := Seek(fd, 0, 1)
  642. if e != nil {
  643. return e
  644. }
  645. defer Seek(fd, curoffset, 0)
  646. _, e = Seek(fd, length, 0)
  647. if e != nil {
  648. return e
  649. }
  650. e = SetEndOfFile(fd)
  651. if e != nil {
  652. return e
  653. }
  654. return nil
  655. }
  656. func Gettimeofday(tv *Timeval) (err error) {
  657. var ft Filetime
  658. GetSystemTimeAsFileTime(&ft)
  659. *tv = NsecToTimeval(ft.Nanoseconds())
  660. return nil
  661. }
  662. func Pipe(p []Handle) (err error) {
  663. if len(p) != 2 {
  664. return syscall.EINVAL
  665. }
  666. var r, w Handle
  667. e := CreatePipe(&r, &w, makeInheritSa(), 0)
  668. if e != nil {
  669. return e
  670. }
  671. p[0] = r
  672. p[1] = w
  673. return nil
  674. }
  675. func Utimes(path string, tv []Timeval) (err error) {
  676. if len(tv) != 2 {
  677. return syscall.EINVAL
  678. }
  679. pathp, e := UTF16PtrFromString(path)
  680. if e != nil {
  681. return e
  682. }
  683. h, e := CreateFile(pathp,
  684. FILE_WRITE_ATTRIBUTES, FILE_SHARE_WRITE, nil,
  685. OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, 0)
  686. if e != nil {
  687. return e
  688. }
  689. defer CloseHandle(h)
  690. a := NsecToFiletime(tv[0].Nanoseconds())
  691. w := NsecToFiletime(tv[1].Nanoseconds())
  692. return SetFileTime(h, nil, &a, &w)
  693. }
  694. func UtimesNano(path string, ts []Timespec) (err error) {
  695. if len(ts) != 2 {
  696. return syscall.EINVAL
  697. }
  698. pathp, e := UTF16PtrFromString(path)
  699. if e != nil {
  700. return e
  701. }
  702. h, e := CreateFile(pathp,
  703. FILE_WRITE_ATTRIBUTES, FILE_SHARE_WRITE, nil,
  704. OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, 0)
  705. if e != nil {
  706. return e
  707. }
  708. defer CloseHandle(h)
  709. a := NsecToFiletime(TimespecToNsec(ts[0]))
  710. w := NsecToFiletime(TimespecToNsec(ts[1]))
  711. return SetFileTime(h, nil, &a, &w)
  712. }
  713. func Fsync(fd Handle) (err error) {
  714. return FlushFileBuffers(fd)
  715. }
  716. func Chmod(path string, mode uint32) (err error) {
  717. p, e := UTF16PtrFromString(path)
  718. if e != nil {
  719. return e
  720. }
  721. attrs, e := GetFileAttributes(p)
  722. if e != nil {
  723. return e
  724. }
  725. if mode&S_IWRITE != 0 {
  726. attrs &^= FILE_ATTRIBUTE_READONLY
  727. } else {
  728. attrs |= FILE_ATTRIBUTE_READONLY
  729. }
  730. return SetFileAttributes(p, attrs)
  731. }
  732. func LoadGetSystemTimePreciseAsFileTime() error {
  733. return procGetSystemTimePreciseAsFileTime.Find()
  734. }
  735. func LoadCancelIoEx() error {
  736. return procCancelIoEx.Find()
  737. }
  738. func LoadSetFileCompletionNotificationModes() error {
  739. return procSetFileCompletionNotificationModes.Find()
  740. }
  741. func WaitForMultipleObjects(handles []Handle, waitAll bool, waitMilliseconds uint32) (event uint32, err error) {
  742. // Every other win32 array API takes arguments as "pointer, count", except for this function. So we
  743. // can't declare it as a usual [] type, because mksyscall will use the opposite order. We therefore
  744. // trivially stub this ourselves.
  745. var handlePtr *Handle
  746. if len(handles) > 0 {
  747. handlePtr = &handles[0]
  748. }
  749. return waitForMultipleObjects(uint32(len(handles)), uintptr(unsafe.Pointer(handlePtr)), waitAll, waitMilliseconds)
  750. }
  751. // net api calls
  752. const socket_error = uintptr(^uint32(0))
  753. //sys WSAStartup(verreq uint32, data *WSAData) (sockerr error) = ws2_32.WSAStartup
  754. //sys WSACleanup() (err error) [failretval==socket_error] = ws2_32.WSACleanup
  755. //sys WSAIoctl(s Handle, iocc uint32, inbuf *byte, cbif uint32, outbuf *byte, cbob uint32, cbbr *uint32, overlapped *Overlapped, completionRoutine uintptr) (err error) [failretval==socket_error] = ws2_32.WSAIoctl
  756. //sys WSALookupServiceBegin(querySet *WSAQUERYSET, flags uint32, handle *Handle) (err error) [failretval==socket_error] = ws2_32.WSALookupServiceBeginW
  757. //sys WSALookupServiceNext(handle Handle, flags uint32, size *int32, querySet *WSAQUERYSET) (err error) [failretval==socket_error] = ws2_32.WSALookupServiceNextW
  758. //sys WSALookupServiceEnd(handle Handle) (err error) [failretval==socket_error] = ws2_32.WSALookupServiceEnd
  759. //sys socket(af int32, typ int32, protocol int32) (handle Handle, err error) [failretval==InvalidHandle] = ws2_32.socket
  760. //sys sendto(s Handle, buf []byte, flags int32, to unsafe.Pointer, tolen int32) (err error) [failretval==socket_error] = ws2_32.sendto
  761. //sys recvfrom(s Handle, buf []byte, flags int32, from *RawSockaddrAny, fromlen *int32) (n int32, err error) [failretval==-1] = ws2_32.recvfrom
  762. //sys Setsockopt(s Handle, level int32, optname int32, optval *byte, optlen int32) (err error) [failretval==socket_error] = ws2_32.setsockopt
  763. //sys Getsockopt(s Handle, level int32, optname int32, optval *byte, optlen *int32) (err error) [failretval==socket_error] = ws2_32.getsockopt
  764. //sys bind(s Handle, name unsafe.Pointer, namelen int32) (err error) [failretval==socket_error] = ws2_32.bind
  765. //sys connect(s Handle, name unsafe.Pointer, namelen int32) (err error) [failretval==socket_error] = ws2_32.connect
  766. //sys getsockname(s Handle, rsa *RawSockaddrAny, addrlen *int32) (err error) [failretval==socket_error] = ws2_32.getsockname
  767. //sys getpeername(s Handle, rsa *RawSockaddrAny, addrlen *int32) (err error) [failretval==socket_error] = ws2_32.getpeername
  768. //sys listen(s Handle, backlog int32) (err error) [failretval==socket_error] = ws2_32.listen
  769. //sys shutdown(s Handle, how int32) (err error) [failretval==socket_error] = ws2_32.shutdown
  770. //sys Closesocket(s Handle) (err error) [failretval==socket_error] = ws2_32.closesocket
  771. //sys AcceptEx(ls Handle, as Handle, buf *byte, rxdatalen uint32, laddrlen uint32, raddrlen uint32, recvd *uint32, overlapped *Overlapped) (err error) = mswsock.AcceptEx
  772. //sys GetAcceptExSockaddrs(buf *byte, rxdatalen uint32, laddrlen uint32, raddrlen uint32, lrsa **RawSockaddrAny, lrsalen *int32, rrsa **RawSockaddrAny, rrsalen *int32) = mswsock.GetAcceptExSockaddrs
  773. //sys WSARecv(s Handle, bufs *WSABuf, bufcnt uint32, recvd *uint32, flags *uint32, overlapped *Overlapped, croutine *byte) (err error) [failretval==socket_error] = ws2_32.WSARecv
  774. //sys WSASend(s Handle, bufs *WSABuf, bufcnt uint32, sent *uint32, flags uint32, overlapped *Overlapped, croutine *byte) (err error) [failretval==socket_error] = ws2_32.WSASend
  775. //sys WSARecvFrom(s Handle, bufs *WSABuf, bufcnt uint32, recvd *uint32, flags *uint32, from *RawSockaddrAny, fromlen *int32, overlapped *Overlapped, croutine *byte) (err error) [failretval==socket_error] = ws2_32.WSARecvFrom
  776. //sys WSASendTo(s Handle, bufs *WSABuf, bufcnt uint32, sent *uint32, flags uint32, to *RawSockaddrAny, tolen int32, overlapped *Overlapped, croutine *byte) (err error) [failretval==socket_error] = ws2_32.WSASendTo
  777. //sys WSASocket(af int32, typ int32, protocol int32, protoInfo *WSAProtocolInfo, group uint32, flags uint32) (handle Handle, err error) [failretval==InvalidHandle] = ws2_32.WSASocketW
  778. //sys GetHostByName(name string) (h *Hostent, err error) [failretval==nil] = ws2_32.gethostbyname
  779. //sys GetServByName(name string, proto string) (s *Servent, err error) [failretval==nil] = ws2_32.getservbyname
  780. //sys Ntohs(netshort uint16) (u uint16) = ws2_32.ntohs
  781. //sys GetProtoByName(name string) (p *Protoent, err error) [failretval==nil] = ws2_32.getprotobyname
  782. //sys DnsQuery(name string, qtype uint16, options uint32, extra *byte, qrs **DNSRecord, pr *byte) (status error) = dnsapi.DnsQuery_W
  783. //sys DnsRecordListFree(rl *DNSRecord, freetype uint32) = dnsapi.DnsRecordListFree
  784. //sys DnsNameCompare(name1 *uint16, name2 *uint16) (same bool) = dnsapi.DnsNameCompare_W
  785. //sys GetAddrInfoW(nodename *uint16, servicename *uint16, hints *AddrinfoW, result **AddrinfoW) (sockerr error) = ws2_32.GetAddrInfoW
  786. //sys FreeAddrInfoW(addrinfo *AddrinfoW) = ws2_32.FreeAddrInfoW
  787. //sys GetIfEntry(pIfRow *MibIfRow) (errcode error) = iphlpapi.GetIfEntry
  788. //sys GetAdaptersInfo(ai *IpAdapterInfo, ol *uint32) (errcode error) = iphlpapi.GetAdaptersInfo
  789. //sys SetFileCompletionNotificationModes(handle Handle, flags uint8) (err error) = kernel32.SetFileCompletionNotificationModes
  790. //sys WSAEnumProtocols(protocols *int32, protocolBuffer *WSAProtocolInfo, bufferLength *uint32) (n int32, err error) [failretval==-1] = ws2_32.WSAEnumProtocolsW
  791. //sys WSAGetOverlappedResult(h Handle, o *Overlapped, bytes *uint32, wait bool, flags *uint32) (err error) = ws2_32.WSAGetOverlappedResult
  792. //sys GetAdaptersAddresses(family uint32, flags uint32, reserved uintptr, adapterAddresses *IpAdapterAddresses, sizePointer *uint32) (errcode error) = iphlpapi.GetAdaptersAddresses
  793. //sys GetACP() (acp uint32) = kernel32.GetACP
  794. //sys MultiByteToWideChar(codePage uint32, dwFlags uint32, str *byte, nstr int32, wchar *uint16, nwchar int32) (nwrite int32, err error) = kernel32.MultiByteToWideChar
  795. //sys getBestInterfaceEx(sockaddr unsafe.Pointer, pdwBestIfIndex *uint32) (errcode error) = iphlpapi.GetBestInterfaceEx
  796. // For testing: clients can set this flag to force
  797. // creation of IPv6 sockets to return EAFNOSUPPORT.
  798. var SocketDisableIPv6 bool
  799. type RawSockaddrInet4 struct {
  800. Family uint16
  801. Port uint16
  802. Addr [4]byte /* in_addr */
  803. Zero [8]uint8
  804. }
  805. type RawSockaddrInet6 struct {
  806. Family uint16
  807. Port uint16
  808. Flowinfo uint32
  809. Addr [16]byte /* in6_addr */
  810. Scope_id uint32
  811. }
  812. type RawSockaddr struct {
  813. Family uint16
  814. Data [14]int8
  815. }
  816. type RawSockaddrAny struct {
  817. Addr RawSockaddr
  818. Pad [100]int8
  819. }
  820. type Sockaddr interface {
  821. sockaddr() (ptr unsafe.Pointer, len int32, err error) // lowercase; only we can define Sockaddrs
  822. }
  823. type SockaddrInet4 struct {
  824. Port int
  825. Addr [4]byte
  826. raw RawSockaddrInet4
  827. }
  828. func (sa *SockaddrInet4) sockaddr() (unsafe.Pointer, int32, error) {
  829. if sa.Port < 0 || sa.Port > 0xFFFF {
  830. return nil, 0, syscall.EINVAL
  831. }
  832. sa.raw.Family = AF_INET
  833. p := (*[2]byte)(unsafe.Pointer(&sa.raw.Port))
  834. p[0] = byte(sa.Port >> 8)
  835. p[1] = byte(sa.Port)
  836. sa.raw.Addr = sa.Addr
  837. return unsafe.Pointer(&sa.raw), int32(unsafe.Sizeof(sa.raw)), nil
  838. }
  839. type SockaddrInet6 struct {
  840. Port int
  841. ZoneId uint32
  842. Addr [16]byte
  843. raw RawSockaddrInet6
  844. }
  845. func (sa *SockaddrInet6) sockaddr() (unsafe.Pointer, int32, error) {
  846. if sa.Port < 0 || sa.Port > 0xFFFF {
  847. return nil, 0, syscall.EINVAL
  848. }
  849. sa.raw.Family = AF_INET6
  850. p := (*[2]byte)(unsafe.Pointer(&sa.raw.Port))
  851. p[0] = byte(sa.Port >> 8)
  852. p[1] = byte(sa.Port)
  853. sa.raw.Scope_id = sa.ZoneId
  854. sa.raw.Addr = sa.Addr
  855. return unsafe.Pointer(&sa.raw), int32(unsafe.Sizeof(sa.raw)), nil
  856. }
  857. type RawSockaddrUnix struct {
  858. Family uint16
  859. Path [UNIX_PATH_MAX]int8
  860. }
  861. type SockaddrUnix struct {
  862. Name string
  863. raw RawSockaddrUnix
  864. }
  865. func (sa *SockaddrUnix) sockaddr() (unsafe.Pointer, int32, error) {
  866. name := sa.Name
  867. n := len(name)
  868. if n > len(sa.raw.Path) {
  869. return nil, 0, syscall.EINVAL
  870. }
  871. if n == len(sa.raw.Path) && name[0] != '@' {
  872. return nil, 0, syscall.EINVAL
  873. }
  874. sa.raw.Family = AF_UNIX
  875. for i := 0; i < n; i++ {
  876. sa.raw.Path[i] = int8(name[i])
  877. }
  878. // length is family (uint16), name, NUL.
  879. sl := int32(2)
  880. if n > 0 {
  881. sl += int32(n) + 1
  882. }
  883. if sa.raw.Path[0] == '@' {
  884. sa.raw.Path[0] = 0
  885. // Don't count trailing NUL for abstract address.
  886. sl--
  887. }
  888. return unsafe.Pointer(&sa.raw), sl, nil
  889. }
  890. type RawSockaddrBth struct {
  891. AddressFamily [2]byte
  892. BtAddr [8]byte
  893. ServiceClassId [16]byte
  894. Port [4]byte
  895. }
  896. type SockaddrBth struct {
  897. BtAddr uint64
  898. ServiceClassId GUID
  899. Port uint32
  900. raw RawSockaddrBth
  901. }
  902. func (sa *SockaddrBth) sockaddr() (unsafe.Pointer, int32, error) {
  903. family := AF_BTH
  904. sa.raw = RawSockaddrBth{
  905. AddressFamily: *(*[2]byte)(unsafe.Pointer(&family)),
  906. BtAddr: *(*[8]byte)(unsafe.Pointer(&sa.BtAddr)),
  907. Port: *(*[4]byte)(unsafe.Pointer(&sa.Port)),
  908. ServiceClassId: *(*[16]byte)(unsafe.Pointer(&sa.ServiceClassId)),
  909. }
  910. return unsafe.Pointer(&sa.raw), int32(unsafe.Sizeof(sa.raw)), nil
  911. }
  912. func (rsa *RawSockaddrAny) Sockaddr() (Sockaddr, error) {
  913. switch rsa.Addr.Family {
  914. case AF_UNIX:
  915. pp := (*RawSockaddrUnix)(unsafe.Pointer(rsa))
  916. sa := new(SockaddrUnix)
  917. if pp.Path[0] == 0 {
  918. // "Abstract" Unix domain socket.
  919. // Rewrite leading NUL as @ for textual display.
  920. // (This is the standard convention.)
  921. // Not friendly to overwrite in place,
  922. // but the callers below don't care.
  923. pp.Path[0] = '@'
  924. }
  925. // Assume path ends at NUL.
  926. // This is not technically the Linux semantics for
  927. // abstract Unix domain sockets--they are supposed
  928. // to be uninterpreted fixed-size binary blobs--but
  929. // everyone uses this convention.
  930. n := 0
  931. for n < len(pp.Path) && pp.Path[n] != 0 {
  932. n++
  933. }
  934. sa.Name = string(unsafe.Slice((*byte)(unsafe.Pointer(&pp.Path[0])), n))
  935. return sa, nil
  936. case AF_INET:
  937. pp := (*RawSockaddrInet4)(unsafe.Pointer(rsa))
  938. sa := new(SockaddrInet4)
  939. p := (*[2]byte)(unsafe.Pointer(&pp.Port))
  940. sa.Port = int(p[0])<<8 + int(p[1])
  941. sa.Addr = pp.Addr
  942. return sa, nil
  943. case AF_INET6:
  944. pp := (*RawSockaddrInet6)(unsafe.Pointer(rsa))
  945. sa := new(SockaddrInet6)
  946. p := (*[2]byte)(unsafe.Pointer(&pp.Port))
  947. sa.Port = int(p[0])<<8 + int(p[1])
  948. sa.ZoneId = pp.Scope_id
  949. sa.Addr = pp.Addr
  950. return sa, nil
  951. }
  952. return nil, syscall.EAFNOSUPPORT
  953. }
  954. func Socket(domain, typ, proto int) (fd Handle, err error) {
  955. if domain == AF_INET6 && SocketDisableIPv6 {
  956. return InvalidHandle, syscall.EAFNOSUPPORT
  957. }
  958. return socket(int32(domain), int32(typ), int32(proto))
  959. }
  960. func SetsockoptInt(fd Handle, level, opt int, value int) (err error) {
  961. v := int32(value)
  962. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(&v)), int32(unsafe.Sizeof(v)))
  963. }
  964. func Bind(fd Handle, sa Sockaddr) (err error) {
  965. ptr, n, err := sa.sockaddr()
  966. if err != nil {
  967. return err
  968. }
  969. return bind(fd, ptr, n)
  970. }
  971. func Connect(fd Handle, sa Sockaddr) (err error) {
  972. ptr, n, err := sa.sockaddr()
  973. if err != nil {
  974. return err
  975. }
  976. return connect(fd, ptr, n)
  977. }
  978. func GetBestInterfaceEx(sa Sockaddr, pdwBestIfIndex *uint32) (err error) {
  979. ptr, _, err := sa.sockaddr()
  980. if err != nil {
  981. return err
  982. }
  983. return getBestInterfaceEx(ptr, pdwBestIfIndex)
  984. }
  985. func Getsockname(fd Handle) (sa Sockaddr, err error) {
  986. var rsa RawSockaddrAny
  987. l := int32(unsafe.Sizeof(rsa))
  988. if err = getsockname(fd, &rsa, &l); err != nil {
  989. return
  990. }
  991. return rsa.Sockaddr()
  992. }
  993. func Getpeername(fd Handle) (sa Sockaddr, err error) {
  994. var rsa RawSockaddrAny
  995. l := int32(unsafe.Sizeof(rsa))
  996. if err = getpeername(fd, &rsa, &l); err != nil {
  997. return
  998. }
  999. return rsa.Sockaddr()
  1000. }
  1001. func Listen(s Handle, n int) (err error) {
  1002. return listen(s, int32(n))
  1003. }
  1004. func Shutdown(fd Handle, how int) (err error) {
  1005. return shutdown(fd, int32(how))
  1006. }
  1007. func WSASendto(s Handle, bufs *WSABuf, bufcnt uint32, sent *uint32, flags uint32, to Sockaddr, overlapped *Overlapped, croutine *byte) (err error) {
  1008. var rsa unsafe.Pointer
  1009. var l int32
  1010. if to != nil {
  1011. rsa, l, err = to.sockaddr()
  1012. if err != nil {
  1013. return err
  1014. }
  1015. }
  1016. return WSASendTo(s, bufs, bufcnt, sent, flags, (*RawSockaddrAny)(unsafe.Pointer(rsa)), l, overlapped, croutine)
  1017. }
  1018. func LoadGetAddrInfo() error {
  1019. return procGetAddrInfoW.Find()
  1020. }
  1021. var connectExFunc struct {
  1022. once sync.Once
  1023. addr uintptr
  1024. err error
  1025. }
  1026. func LoadConnectEx() error {
  1027. connectExFunc.once.Do(func() {
  1028. var s Handle
  1029. s, connectExFunc.err = Socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)
  1030. if connectExFunc.err != nil {
  1031. return
  1032. }
  1033. defer CloseHandle(s)
  1034. var n uint32
  1035. connectExFunc.err = WSAIoctl(s,
  1036. SIO_GET_EXTENSION_FUNCTION_POINTER,
  1037. (*byte)(unsafe.Pointer(&WSAID_CONNECTEX)),
  1038. uint32(unsafe.Sizeof(WSAID_CONNECTEX)),
  1039. (*byte)(unsafe.Pointer(&connectExFunc.addr)),
  1040. uint32(unsafe.Sizeof(connectExFunc.addr)),
  1041. &n, nil, 0)
  1042. })
  1043. return connectExFunc.err
  1044. }
  1045. func connectEx(s Handle, name unsafe.Pointer, namelen int32, sendBuf *byte, sendDataLen uint32, bytesSent *uint32, overlapped *Overlapped) (err error) {
  1046. r1, _, e1 := syscall.Syscall9(connectExFunc.addr, 7, uintptr(s), uintptr(name), uintptr(namelen), uintptr(unsafe.Pointer(sendBuf)), uintptr(sendDataLen), uintptr(unsafe.Pointer(bytesSent)), uintptr(unsafe.Pointer(overlapped)), 0, 0)
  1047. if r1 == 0 {
  1048. if e1 != 0 {
  1049. err = error(e1)
  1050. } else {
  1051. err = syscall.EINVAL
  1052. }
  1053. }
  1054. return
  1055. }
  1056. func ConnectEx(fd Handle, sa Sockaddr, sendBuf *byte, sendDataLen uint32, bytesSent *uint32, overlapped *Overlapped) error {
  1057. err := LoadConnectEx()
  1058. if err != nil {
  1059. return errorspkg.New("failed to find ConnectEx: " + err.Error())
  1060. }
  1061. ptr, n, err := sa.sockaddr()
  1062. if err != nil {
  1063. return err
  1064. }
  1065. return connectEx(fd, ptr, n, sendBuf, sendDataLen, bytesSent, overlapped)
  1066. }
  1067. var sendRecvMsgFunc struct {
  1068. once sync.Once
  1069. sendAddr uintptr
  1070. recvAddr uintptr
  1071. err error
  1072. }
  1073. func loadWSASendRecvMsg() error {
  1074. sendRecvMsgFunc.once.Do(func() {
  1075. var s Handle
  1076. s, sendRecvMsgFunc.err = Socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)
  1077. if sendRecvMsgFunc.err != nil {
  1078. return
  1079. }
  1080. defer CloseHandle(s)
  1081. var n uint32
  1082. sendRecvMsgFunc.err = WSAIoctl(s,
  1083. SIO_GET_EXTENSION_FUNCTION_POINTER,
  1084. (*byte)(unsafe.Pointer(&WSAID_WSARECVMSG)),
  1085. uint32(unsafe.Sizeof(WSAID_WSARECVMSG)),
  1086. (*byte)(unsafe.Pointer(&sendRecvMsgFunc.recvAddr)),
  1087. uint32(unsafe.Sizeof(sendRecvMsgFunc.recvAddr)),
  1088. &n, nil, 0)
  1089. if sendRecvMsgFunc.err != nil {
  1090. return
  1091. }
  1092. sendRecvMsgFunc.err = WSAIoctl(s,
  1093. SIO_GET_EXTENSION_FUNCTION_POINTER,
  1094. (*byte)(unsafe.Pointer(&WSAID_WSASENDMSG)),
  1095. uint32(unsafe.Sizeof(WSAID_WSASENDMSG)),
  1096. (*byte)(unsafe.Pointer(&sendRecvMsgFunc.sendAddr)),
  1097. uint32(unsafe.Sizeof(sendRecvMsgFunc.sendAddr)),
  1098. &n, nil, 0)
  1099. })
  1100. return sendRecvMsgFunc.err
  1101. }
  1102. func WSASendMsg(fd Handle, msg *WSAMsg, flags uint32, bytesSent *uint32, overlapped *Overlapped, croutine *byte) error {
  1103. err := loadWSASendRecvMsg()
  1104. if err != nil {
  1105. return err
  1106. }
  1107. r1, _, e1 := syscall.Syscall6(sendRecvMsgFunc.sendAddr, 6, uintptr(fd), uintptr(unsafe.Pointer(msg)), uintptr(flags), uintptr(unsafe.Pointer(bytesSent)), uintptr(unsafe.Pointer(overlapped)), uintptr(unsafe.Pointer(croutine)))
  1108. if r1 == socket_error {
  1109. err = errnoErr(e1)
  1110. }
  1111. return err
  1112. }
  1113. func WSARecvMsg(fd Handle, msg *WSAMsg, bytesReceived *uint32, overlapped *Overlapped, croutine *byte) error {
  1114. err := loadWSASendRecvMsg()
  1115. if err != nil {
  1116. return err
  1117. }
  1118. r1, _, e1 := syscall.Syscall6(sendRecvMsgFunc.recvAddr, 5, uintptr(fd), uintptr(unsafe.Pointer(msg)), uintptr(unsafe.Pointer(bytesReceived)), uintptr(unsafe.Pointer(overlapped)), uintptr(unsafe.Pointer(croutine)), 0)
  1119. if r1 == socket_error {
  1120. err = errnoErr(e1)
  1121. }
  1122. return err
  1123. }
  1124. // Invented structures to support what package os expects.
  1125. type Rusage struct {
  1126. CreationTime Filetime
  1127. ExitTime Filetime
  1128. KernelTime Filetime
  1129. UserTime Filetime
  1130. }
  1131. type WaitStatus struct {
  1132. ExitCode uint32
  1133. }
  1134. func (w WaitStatus) Exited() bool { return true }
  1135. func (w WaitStatus) ExitStatus() int { return int(w.ExitCode) }
  1136. func (w WaitStatus) Signal() Signal { return -1 }
  1137. func (w WaitStatus) CoreDump() bool { return false }
  1138. func (w WaitStatus) Stopped() bool { return false }
  1139. func (w WaitStatus) Continued() bool { return false }
  1140. func (w WaitStatus) StopSignal() Signal { return -1 }
  1141. func (w WaitStatus) Signaled() bool { return false }
  1142. func (w WaitStatus) TrapCause() int { return -1 }
  1143. // Timespec is an invented structure on Windows, but here for
  1144. // consistency with the corresponding package for other operating systems.
  1145. type Timespec struct {
  1146. Sec int64
  1147. Nsec int64
  1148. }
  1149. func TimespecToNsec(ts Timespec) int64 { return int64(ts.Sec)*1e9 + int64(ts.Nsec) }
  1150. func NsecToTimespec(nsec int64) (ts Timespec) {
  1151. ts.Sec = nsec / 1e9
  1152. ts.Nsec = nsec % 1e9
  1153. return
  1154. }
  1155. // TODO(brainman): fix all needed for net
  1156. func Accept(fd Handle) (nfd Handle, sa Sockaddr, err error) { return 0, nil, syscall.EWINDOWS }
  1157. func Recvfrom(fd Handle, p []byte, flags int) (n int, from Sockaddr, err error) {
  1158. var rsa RawSockaddrAny
  1159. l := int32(unsafe.Sizeof(rsa))
  1160. n32, err := recvfrom(fd, p, int32(flags), &rsa, &l)
  1161. n = int(n32)
  1162. if err != nil {
  1163. return
  1164. }
  1165. from, err = rsa.Sockaddr()
  1166. return
  1167. }
  1168. func Sendto(fd Handle, p []byte, flags int, to Sockaddr) (err error) {
  1169. ptr, l, err := to.sockaddr()
  1170. if err != nil {
  1171. return err
  1172. }
  1173. return sendto(fd, p, int32(flags), ptr, l)
  1174. }
  1175. func SetsockoptTimeval(fd Handle, level, opt int, tv *Timeval) (err error) { return syscall.EWINDOWS }
  1176. // The Linger struct is wrong but we only noticed after Go 1.
  1177. // sysLinger is the real system call structure.
  1178. // BUG(brainman): The definition of Linger is not appropriate for direct use
  1179. // with Setsockopt and Getsockopt.
  1180. // Use SetsockoptLinger instead.
  1181. type Linger struct {
  1182. Onoff int32
  1183. Linger int32
  1184. }
  1185. type sysLinger struct {
  1186. Onoff uint16
  1187. Linger uint16
  1188. }
  1189. type IPMreq struct {
  1190. Multiaddr [4]byte /* in_addr */
  1191. Interface [4]byte /* in_addr */
  1192. }
  1193. type IPv6Mreq struct {
  1194. Multiaddr [16]byte /* in6_addr */
  1195. Interface uint32
  1196. }
  1197. func GetsockoptInt(fd Handle, level, opt int) (int, error) {
  1198. v := int32(0)
  1199. l := int32(unsafe.Sizeof(v))
  1200. err := Getsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(&v)), &l)
  1201. return int(v), err
  1202. }
  1203. func SetsockoptLinger(fd Handle, level, opt int, l *Linger) (err error) {
  1204. sys := sysLinger{Onoff: uint16(l.Onoff), Linger: uint16(l.Linger)}
  1205. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(&sys)), int32(unsafe.Sizeof(sys)))
  1206. }
  1207. func SetsockoptInet4Addr(fd Handle, level, opt int, value [4]byte) (err error) {
  1208. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(&value[0])), 4)
  1209. }
  1210. func SetsockoptIPMreq(fd Handle, level, opt int, mreq *IPMreq) (err error) {
  1211. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(mreq)), int32(unsafe.Sizeof(*mreq)))
  1212. }
  1213. func SetsockoptIPv6Mreq(fd Handle, level, opt int, mreq *IPv6Mreq) (err error) {
  1214. return syscall.EWINDOWS
  1215. }
  1216. func Getpid() (pid int) { return int(GetCurrentProcessId()) }
  1217. func FindFirstFile(name *uint16, data *Win32finddata) (handle Handle, err error) {
  1218. // NOTE(rsc): The Win32finddata struct is wrong for the system call:
  1219. // the two paths are each one uint16 short. Use the correct struct,
  1220. // a win32finddata1, and then copy the results out.
  1221. // There is no loss of expressivity here, because the final
  1222. // uint16, if it is used, is supposed to be a NUL, and Go doesn't need that.
  1223. // For Go 1.1, we might avoid the allocation of win32finddata1 here
  1224. // by adding a final Bug [2]uint16 field to the struct and then
  1225. // adjusting the fields in the result directly.
  1226. var data1 win32finddata1
  1227. handle, err = findFirstFile1(name, &data1)
  1228. if err == nil {
  1229. copyFindData(data, &data1)
  1230. }
  1231. return
  1232. }
  1233. func FindNextFile(handle Handle, data *Win32finddata) (err error) {
  1234. var data1 win32finddata1
  1235. err = findNextFile1(handle, &data1)
  1236. if err == nil {
  1237. copyFindData(data, &data1)
  1238. }
  1239. return
  1240. }
  1241. func getProcessEntry(pid int) (*ProcessEntry32, error) {
  1242. snapshot, err := CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0)
  1243. if err != nil {
  1244. return nil, err
  1245. }
  1246. defer CloseHandle(snapshot)
  1247. var procEntry ProcessEntry32
  1248. procEntry.Size = uint32(unsafe.Sizeof(procEntry))
  1249. if err = Process32First(snapshot, &procEntry); err != nil {
  1250. return nil, err
  1251. }
  1252. for {
  1253. if procEntry.ProcessID == uint32(pid) {
  1254. return &procEntry, nil
  1255. }
  1256. err = Process32Next(snapshot, &procEntry)
  1257. if err != nil {
  1258. return nil, err
  1259. }
  1260. }
  1261. }
  1262. func Getppid() (ppid int) {
  1263. pe, err := getProcessEntry(Getpid())
  1264. if err != nil {
  1265. return -1
  1266. }
  1267. return int(pe.ParentProcessID)
  1268. }
  1269. // TODO(brainman): fix all needed for os
  1270. func Fchdir(fd Handle) (err error) { return syscall.EWINDOWS }
  1271. func Link(oldpath, newpath string) (err error) { return syscall.EWINDOWS }
  1272. func Symlink(path, link string) (err error) { return syscall.EWINDOWS }
  1273. func Fchmod(fd Handle, mode uint32) (err error) { return syscall.EWINDOWS }
  1274. func Chown(path string, uid int, gid int) (err error) { return syscall.EWINDOWS }
  1275. func Lchown(path string, uid int, gid int) (err error) { return syscall.EWINDOWS }
  1276. func Fchown(fd Handle, uid int, gid int) (err error) { return syscall.EWINDOWS }
  1277. func Getuid() (uid int) { return -1 }
  1278. func Geteuid() (euid int) { return -1 }
  1279. func Getgid() (gid int) { return -1 }
  1280. func Getegid() (egid int) { return -1 }
  1281. func Getgroups() (gids []int, err error) { return nil, syscall.EWINDOWS }
  1282. type Signal int
  1283. func (s Signal) Signal() {}
  1284. func (s Signal) String() string {
  1285. if 0 <= s && int(s) < len(signals) {
  1286. str := signals[s]
  1287. if str != "" {
  1288. return str
  1289. }
  1290. }
  1291. return "signal " + itoa(int(s))
  1292. }
  1293. func LoadCreateSymbolicLink() error {
  1294. return procCreateSymbolicLinkW.Find()
  1295. }
  1296. // Readlink returns the destination of the named symbolic link.
  1297. func Readlink(path string, buf []byte) (n int, err error) {
  1298. fd, err := CreateFile(StringToUTF16Ptr(path), GENERIC_READ, 0, nil, OPEN_EXISTING,
  1299. FILE_FLAG_OPEN_REPARSE_POINT|FILE_FLAG_BACKUP_SEMANTICS, 0)
  1300. if err != nil {
  1301. return -1, err
  1302. }
  1303. defer CloseHandle(fd)
  1304. rdbbuf := make([]byte, MAXIMUM_REPARSE_DATA_BUFFER_SIZE)
  1305. var bytesReturned uint32
  1306. err = DeviceIoControl(fd, FSCTL_GET_REPARSE_POINT, nil, 0, &rdbbuf[0], uint32(len(rdbbuf)), &bytesReturned, nil)
  1307. if err != nil {
  1308. return -1, err
  1309. }
  1310. rdb := (*reparseDataBuffer)(unsafe.Pointer(&rdbbuf[0]))
  1311. var s string
  1312. switch rdb.ReparseTag {
  1313. case IO_REPARSE_TAG_SYMLINK:
  1314. data := (*symbolicLinkReparseBuffer)(unsafe.Pointer(&rdb.reparseBuffer))
  1315. p := (*[0xffff]uint16)(unsafe.Pointer(&data.PathBuffer[0]))
  1316. s = UTF16ToString(p[data.PrintNameOffset/2 : (data.PrintNameLength-data.PrintNameOffset)/2])
  1317. case IO_REPARSE_TAG_MOUNT_POINT:
  1318. data := (*mountPointReparseBuffer)(unsafe.Pointer(&rdb.reparseBuffer))
  1319. p := (*[0xffff]uint16)(unsafe.Pointer(&data.PathBuffer[0]))
  1320. s = UTF16ToString(p[data.PrintNameOffset/2 : (data.PrintNameLength-data.PrintNameOffset)/2])
  1321. default:
  1322. // the path is not a symlink or junction but another type of reparse
  1323. // point
  1324. return -1, syscall.ENOENT
  1325. }
  1326. n = copy(buf, []byte(s))
  1327. return n, nil
  1328. }
  1329. // GUIDFromString parses a string in the form of
  1330. // "{XXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX}" into a GUID.
  1331. func GUIDFromString(str string) (GUID, error) {
  1332. guid := GUID{}
  1333. str16, err := syscall.UTF16PtrFromString(str)
  1334. if err != nil {
  1335. return guid, err
  1336. }
  1337. err = clsidFromString(str16, &guid)
  1338. if err != nil {
  1339. return guid, err
  1340. }
  1341. return guid, nil
  1342. }
  1343. // GenerateGUID creates a new random GUID.
  1344. func GenerateGUID() (GUID, error) {
  1345. guid := GUID{}
  1346. err := coCreateGuid(&guid)
  1347. if err != nil {
  1348. return guid, err
  1349. }
  1350. return guid, nil
  1351. }
  1352. // String returns the canonical string form of the GUID,
  1353. // in the form of "{XXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX}".
  1354. func (guid GUID) String() string {
  1355. var str [100]uint16
  1356. chars := stringFromGUID2(&guid, &str[0], int32(len(str)))
  1357. if chars <= 1 {
  1358. return ""
  1359. }
  1360. return string(utf16.Decode(str[:chars-1]))
  1361. }
  1362. // KnownFolderPath returns a well-known folder path for the current user, specified by one of
  1363. // the FOLDERID_ constants, and chosen and optionally created based on a KF_ flag.
  1364. func KnownFolderPath(folderID *KNOWNFOLDERID, flags uint32) (string, error) {
  1365. return Token(0).KnownFolderPath(folderID, flags)
  1366. }
  1367. // KnownFolderPath returns a well-known folder path for the user token, specified by one of
  1368. // the FOLDERID_ constants, and chosen and optionally created based on a KF_ flag.
  1369. func (t Token) KnownFolderPath(folderID *KNOWNFOLDERID, flags uint32) (string, error) {
  1370. var p *uint16
  1371. err := shGetKnownFolderPath(folderID, flags, t, &p)
  1372. if err != nil {
  1373. return "", err
  1374. }
  1375. defer CoTaskMemFree(unsafe.Pointer(p))
  1376. return UTF16PtrToString(p), nil
  1377. }
  1378. // RtlGetVersion returns the version of the underlying operating system, ignoring
  1379. // manifest semantics but is affected by the application compatibility layer.
  1380. func RtlGetVersion() *OsVersionInfoEx {
  1381. info := &OsVersionInfoEx{}
  1382. info.osVersionInfoSize = uint32(unsafe.Sizeof(*info))
  1383. // According to documentation, this function always succeeds.
  1384. // The function doesn't even check the validity of the
  1385. // osVersionInfoSize member. Disassembling ntdll.dll indicates
  1386. // that the documentation is indeed correct about that.
  1387. _ = rtlGetVersion(info)
  1388. return info
  1389. }
  1390. // RtlGetNtVersionNumbers returns the version of the underlying operating system,
  1391. // ignoring manifest semantics and the application compatibility layer.
  1392. func RtlGetNtVersionNumbers() (majorVersion, minorVersion, buildNumber uint32) {
  1393. rtlGetNtVersionNumbers(&majorVersion, &minorVersion, &buildNumber)
  1394. buildNumber &= 0xffff
  1395. return
  1396. }
  1397. // GetProcessPreferredUILanguages retrieves the process preferred UI languages.
  1398. func GetProcessPreferredUILanguages(flags uint32) ([]string, error) {
  1399. return getUILanguages(flags, getProcessPreferredUILanguages)
  1400. }
  1401. // GetThreadPreferredUILanguages retrieves the thread preferred UI languages for the current thread.
  1402. func GetThreadPreferredUILanguages(flags uint32) ([]string, error) {
  1403. return getUILanguages(flags, getThreadPreferredUILanguages)
  1404. }
  1405. // GetUserPreferredUILanguages retrieves information about the user preferred UI languages.
  1406. func GetUserPreferredUILanguages(flags uint32) ([]string, error) {
  1407. return getUILanguages(flags, getUserPreferredUILanguages)
  1408. }
  1409. // GetSystemPreferredUILanguages retrieves the system preferred UI languages.
  1410. func GetSystemPreferredUILanguages(flags uint32) ([]string, error) {
  1411. return getUILanguages(flags, getSystemPreferredUILanguages)
  1412. }
  1413. func getUILanguages(flags uint32, f func(flags uint32, numLanguages *uint32, buf *uint16, bufSize *uint32) error) ([]string, error) {
  1414. size := uint32(128)
  1415. for {
  1416. var numLanguages uint32
  1417. buf := make([]uint16, size)
  1418. err := f(flags, &numLanguages, &buf[0], &size)
  1419. if err == ERROR_INSUFFICIENT_BUFFER {
  1420. continue
  1421. }
  1422. if err != nil {
  1423. return nil, err
  1424. }
  1425. buf = buf[:size]
  1426. if numLanguages == 0 || len(buf) == 0 { // GetProcessPreferredUILanguages may return numLanguages==0 with "\0\0"
  1427. return []string{}, nil
  1428. }
  1429. if buf[len(buf)-1] == 0 {
  1430. buf = buf[:len(buf)-1] // remove terminating null
  1431. }
  1432. languages := make([]string, 0, numLanguages)
  1433. from := 0
  1434. for i, c := range buf {
  1435. if c == 0 {
  1436. languages = append(languages, string(utf16.Decode(buf[from:i])))
  1437. from = i + 1
  1438. }
  1439. }
  1440. return languages, nil
  1441. }
  1442. }
  1443. func SetConsoleCursorPosition(console Handle, position Coord) error {
  1444. return setConsoleCursorPosition(console, *((*uint32)(unsafe.Pointer(&position))))
  1445. }
  1446. func (s NTStatus) Errno() syscall.Errno {
  1447. return rtlNtStatusToDosErrorNoTeb(s)
  1448. }
  1449. func langID(pri, sub uint16) uint32 { return uint32(sub)<<10 | uint32(pri) }
  1450. func (s NTStatus) Error() string {
  1451. b := make([]uint16, 300)
  1452. n, err := FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_FROM_HMODULE|FORMAT_MESSAGE_ARGUMENT_ARRAY, modntdll.Handle(), uint32(s), langID(LANG_ENGLISH, SUBLANG_ENGLISH_US), b, nil)
  1453. if err != nil {
  1454. return fmt.Sprintf("NTSTATUS 0x%08x", uint32(s))
  1455. }
  1456. // trim terminating \r and \n
  1457. for ; n > 0 && (b[n-1] == '\n' || b[n-1] == '\r'); n-- {
  1458. }
  1459. return string(utf16.Decode(b[:n]))
  1460. }
  1461. // NewNTUnicodeString returns a new NTUnicodeString structure for use with native
  1462. // NT APIs that work over the NTUnicodeString type. Note that most Windows APIs
  1463. // do not use NTUnicodeString, and instead UTF16PtrFromString should be used for
  1464. // the more common *uint16 string type.
  1465. func NewNTUnicodeString(s string) (*NTUnicodeString, error) {
  1466. var u NTUnicodeString
  1467. s16, err := UTF16PtrFromString(s)
  1468. if err != nil {
  1469. return nil, err
  1470. }
  1471. RtlInitUnicodeString(&u, s16)
  1472. return &u, nil
  1473. }
  1474. // Slice returns a uint16 slice that aliases the data in the NTUnicodeString.
  1475. func (s *NTUnicodeString) Slice() []uint16 {
  1476. var slice []uint16
  1477. hdr := (*unsafeheader.Slice)(unsafe.Pointer(&slice))
  1478. hdr.Data = unsafe.Pointer(s.Buffer)
  1479. hdr.Len = int(s.Length)
  1480. hdr.Cap = int(s.MaximumLength)
  1481. return slice
  1482. }
  1483. func (s *NTUnicodeString) String() string {
  1484. return UTF16ToString(s.Slice())
  1485. }
  1486. // NewNTString returns a new NTString structure for use with native
  1487. // NT APIs that work over the NTString type. Note that most Windows APIs
  1488. // do not use NTString, and instead UTF16PtrFromString should be used for
  1489. // the more common *uint16 string type.
  1490. func NewNTString(s string) (*NTString, error) {
  1491. var nts NTString
  1492. s8, err := BytePtrFromString(s)
  1493. if err != nil {
  1494. return nil, err
  1495. }
  1496. RtlInitString(&nts, s8)
  1497. return &nts, nil
  1498. }
  1499. // Slice returns a byte slice that aliases the data in the NTString.
  1500. func (s *NTString) Slice() []byte {
  1501. var slice []byte
  1502. hdr := (*unsafeheader.Slice)(unsafe.Pointer(&slice))
  1503. hdr.Data = unsafe.Pointer(s.Buffer)
  1504. hdr.Len = int(s.Length)
  1505. hdr.Cap = int(s.MaximumLength)
  1506. return slice
  1507. }
  1508. func (s *NTString) String() string {
  1509. return ByteSliceToString(s.Slice())
  1510. }
  1511. // FindResource resolves a resource of the given name and resource type.
  1512. func FindResource(module Handle, name, resType ResourceIDOrString) (Handle, error) {
  1513. var namePtr, resTypePtr uintptr
  1514. var name16, resType16 *uint16
  1515. var err error
  1516. resolvePtr := func(i interface{}, keep **uint16) (uintptr, error) {
  1517. switch v := i.(type) {
  1518. case string:
  1519. *keep, err = UTF16PtrFromString(v)
  1520. if err != nil {
  1521. return 0, err
  1522. }
  1523. return uintptr(unsafe.Pointer(*keep)), nil
  1524. case ResourceID:
  1525. return uintptr(v), nil
  1526. }
  1527. return 0, errorspkg.New("parameter must be a ResourceID or a string")
  1528. }
  1529. namePtr, err = resolvePtr(name, &name16)
  1530. if err != nil {
  1531. return 0, err
  1532. }
  1533. resTypePtr, err = resolvePtr(resType, &resType16)
  1534. if err != nil {
  1535. return 0, err
  1536. }
  1537. resInfo, err := findResource(module, namePtr, resTypePtr)
  1538. runtime.KeepAlive(name16)
  1539. runtime.KeepAlive(resType16)
  1540. return resInfo, err
  1541. }
  1542. func LoadResourceData(module, resInfo Handle) (data []byte, err error) {
  1543. size, err := SizeofResource(module, resInfo)
  1544. if err != nil {
  1545. return
  1546. }
  1547. resData, err := LoadResource(module, resInfo)
  1548. if err != nil {
  1549. return
  1550. }
  1551. ptr, err := LockResource(resData)
  1552. if err != nil {
  1553. return
  1554. }
  1555. h := (*unsafeheader.Slice)(unsafe.Pointer(&data))
  1556. h.Data = unsafe.Pointer(ptr)
  1557. h.Len = int(size)
  1558. h.Cap = int(size)
  1559. return
  1560. }
  1561. // PSAPI_WORKING_SET_EX_BLOCK contains extended working set information for a page.
  1562. type PSAPI_WORKING_SET_EX_BLOCK uint64
  1563. // Valid returns the validity of this page.
  1564. // If this bit is 1, the subsequent members are valid; otherwise they should be ignored.
  1565. func (b PSAPI_WORKING_SET_EX_BLOCK) Valid() bool {
  1566. return (b & 1) == 1
  1567. }
  1568. // ShareCount is the number of processes that share this page. The maximum value of this member is 7.
  1569. func (b PSAPI_WORKING_SET_EX_BLOCK) ShareCount() uint64 {
  1570. return b.intField(1, 3)
  1571. }
  1572. // Win32Protection is the memory protection attributes of the page. For a list of values, see
  1573. // https://docs.microsoft.com/en-us/windows/win32/memory/memory-protection-constants
  1574. func (b PSAPI_WORKING_SET_EX_BLOCK) Win32Protection() uint64 {
  1575. return b.intField(4, 11)
  1576. }
  1577. // Shared returns the shared status of this page.
  1578. // If this bit is 1, the page can be shared.
  1579. func (b PSAPI_WORKING_SET_EX_BLOCK) Shared() bool {
  1580. return (b & (1 << 15)) == 1
  1581. }
  1582. // Node is the NUMA node. The maximum value of this member is 63.
  1583. func (b PSAPI_WORKING_SET_EX_BLOCK) Node() uint64 {
  1584. return b.intField(16, 6)
  1585. }
  1586. // Locked returns the locked status of this page.
  1587. // If this bit is 1, the virtual page is locked in physical memory.
  1588. func (b PSAPI_WORKING_SET_EX_BLOCK) Locked() bool {
  1589. return (b & (1 << 22)) == 1
  1590. }
  1591. // LargePage returns the large page status of this page.
  1592. // If this bit is 1, the page is a large page.
  1593. func (b PSAPI_WORKING_SET_EX_BLOCK) LargePage() bool {
  1594. return (b & (1 << 23)) == 1
  1595. }
  1596. // Bad returns the bad status of this page.
  1597. // If this bit is 1, the page is has been reported as bad.
  1598. func (b PSAPI_WORKING_SET_EX_BLOCK) Bad() bool {
  1599. return (b & (1 << 31)) == 1
  1600. }
  1601. // intField extracts an integer field in the PSAPI_WORKING_SET_EX_BLOCK union.
  1602. func (b PSAPI_WORKING_SET_EX_BLOCK) intField(start, length int) uint64 {
  1603. var mask PSAPI_WORKING_SET_EX_BLOCK
  1604. for pos := start; pos < start+length; pos++ {
  1605. mask |= (1 << pos)
  1606. }
  1607. masked := b & mask
  1608. return uint64(masked >> start)
  1609. }
  1610. // PSAPI_WORKING_SET_EX_INFORMATION contains extended working set information for a process.
  1611. type PSAPI_WORKING_SET_EX_INFORMATION struct {
  1612. // The virtual address.
  1613. VirtualAddress Pointer
  1614. // A PSAPI_WORKING_SET_EX_BLOCK union that indicates the attributes of the page at VirtualAddress.
  1615. VirtualAttributes PSAPI_WORKING_SET_EX_BLOCK
  1616. }