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.

syscall_windows.go 58KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489
  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. "sync"
  9. "syscall"
  10. "time"
  11. "unicode/utf16"
  12. "unsafe"
  13. "golang.org/x/sys/internal/unsafeheader"
  14. )
  15. type Handle uintptr
  16. const (
  17. InvalidHandle = ^Handle(0)
  18. // Flags for DefineDosDevice.
  19. DDD_EXACT_MATCH_ON_REMOVE = 0x00000004
  20. DDD_NO_BROADCAST_SYSTEM = 0x00000008
  21. DDD_RAW_TARGET_PATH = 0x00000001
  22. DDD_REMOVE_DEFINITION = 0x00000002
  23. // Return values for GetDriveType.
  24. DRIVE_UNKNOWN = 0
  25. DRIVE_NO_ROOT_DIR = 1
  26. DRIVE_REMOVABLE = 2
  27. DRIVE_FIXED = 3
  28. DRIVE_REMOTE = 4
  29. DRIVE_CDROM = 5
  30. DRIVE_RAMDISK = 6
  31. // File system flags from GetVolumeInformation and GetVolumeInformationByHandle.
  32. FILE_CASE_SENSITIVE_SEARCH = 0x00000001
  33. FILE_CASE_PRESERVED_NAMES = 0x00000002
  34. FILE_FILE_COMPRESSION = 0x00000010
  35. FILE_DAX_VOLUME = 0x20000000
  36. FILE_NAMED_STREAMS = 0x00040000
  37. FILE_PERSISTENT_ACLS = 0x00000008
  38. FILE_READ_ONLY_VOLUME = 0x00080000
  39. FILE_SEQUENTIAL_WRITE_ONCE = 0x00100000
  40. FILE_SUPPORTS_ENCRYPTION = 0x00020000
  41. FILE_SUPPORTS_EXTENDED_ATTRIBUTES = 0x00800000
  42. FILE_SUPPORTS_HARD_LINKS = 0x00400000
  43. FILE_SUPPORTS_OBJECT_IDS = 0x00010000
  44. FILE_SUPPORTS_OPEN_BY_FILE_ID = 0x01000000
  45. FILE_SUPPORTS_REPARSE_POINTS = 0x00000080
  46. FILE_SUPPORTS_SPARSE_FILES = 0x00000040
  47. FILE_SUPPORTS_TRANSACTIONS = 0x00200000
  48. FILE_SUPPORTS_USN_JOURNAL = 0x02000000
  49. FILE_UNICODE_ON_DISK = 0x00000004
  50. FILE_VOLUME_IS_COMPRESSED = 0x00008000
  51. FILE_VOLUME_QUOTAS = 0x00000020
  52. // Flags for LockFileEx.
  53. LOCKFILE_FAIL_IMMEDIATELY = 0x00000001
  54. LOCKFILE_EXCLUSIVE_LOCK = 0x00000002
  55. // Return values of SleepEx and other APC functions
  56. STATUS_USER_APC = 0x000000C0
  57. WAIT_IO_COMPLETION = STATUS_USER_APC
  58. )
  59. // StringToUTF16 is deprecated. Use UTF16FromString instead.
  60. // If s contains a NUL byte this function panics instead of
  61. // returning an error.
  62. func StringToUTF16(s string) []uint16 {
  63. a, err := UTF16FromString(s)
  64. if err != nil {
  65. panic("windows: string with NUL passed to StringToUTF16")
  66. }
  67. return a
  68. }
  69. // UTF16FromString returns the UTF-16 encoding of the UTF-8 string
  70. // s, with a terminating NUL added. If s contains a NUL byte at any
  71. // location, it returns (nil, syscall.EINVAL).
  72. func UTF16FromString(s string) ([]uint16, error) {
  73. for i := 0; i < len(s); i++ {
  74. if s[i] == 0 {
  75. return nil, syscall.EINVAL
  76. }
  77. }
  78. return utf16.Encode([]rune(s + "\x00")), nil
  79. }
  80. // UTF16ToString returns the UTF-8 encoding of the UTF-16 sequence s,
  81. // with a terminating NUL removed.
  82. func UTF16ToString(s []uint16) string {
  83. for i, v := range s {
  84. if v == 0 {
  85. s = s[0:i]
  86. break
  87. }
  88. }
  89. return string(utf16.Decode(s))
  90. }
  91. // StringToUTF16Ptr is deprecated. Use UTF16PtrFromString instead.
  92. // If s contains a NUL byte this function panics instead of
  93. // returning an error.
  94. func StringToUTF16Ptr(s string) *uint16 { return &StringToUTF16(s)[0] }
  95. // UTF16PtrFromString returns pointer to the UTF-16 encoding of
  96. // the UTF-8 string s, with a terminating NUL added. If s
  97. // contains a NUL byte at any location, it returns (nil, syscall.EINVAL).
  98. func UTF16PtrFromString(s string) (*uint16, error) {
  99. a, err := UTF16FromString(s)
  100. if err != nil {
  101. return nil, err
  102. }
  103. return &a[0], nil
  104. }
  105. // UTF16PtrToString takes a pointer to a UTF-16 sequence and returns the corresponding UTF-8 encoded string.
  106. // If the pointer is nil, this returns the empty string. This assumes that the UTF-16 sequence is terminated
  107. // at a zero word; if the zero word is not present, the program may crash.
  108. func UTF16PtrToString(p *uint16) string {
  109. if p == nil {
  110. return ""
  111. }
  112. if *p == 0 {
  113. return ""
  114. }
  115. // Find NUL terminator.
  116. n := 0
  117. for ptr := unsafe.Pointer(p); *(*uint16)(ptr) != 0; n++ {
  118. ptr = unsafe.Pointer(uintptr(ptr) + unsafe.Sizeof(*p))
  119. }
  120. var s []uint16
  121. h := (*unsafeheader.Slice)(unsafe.Pointer(&s))
  122. h.Data = unsafe.Pointer(p)
  123. h.Len = n
  124. h.Cap = n
  125. return string(utf16.Decode(s))
  126. }
  127. func Getpagesize() int { return 4096 }
  128. // NewCallback converts a Go function to a function pointer conforming to the stdcall calling convention.
  129. // This is useful when interoperating with Windows code requiring callbacks.
  130. // 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.
  131. func NewCallback(fn interface{}) uintptr {
  132. return syscall.NewCallback(fn)
  133. }
  134. // NewCallbackCDecl converts a Go function to a function pointer conforming to the cdecl calling convention.
  135. // This is useful when interoperating with Windows code requiring callbacks.
  136. // 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.
  137. func NewCallbackCDecl(fn interface{}) uintptr {
  138. return syscall.NewCallbackCDecl(fn)
  139. }
  140. // windows api calls
  141. //sys GetLastError() (lasterr error)
  142. //sys LoadLibrary(libname string) (handle Handle, err error) = LoadLibraryW
  143. //sys LoadLibraryEx(libname string, zero Handle, flags uintptr) (handle Handle, err error) = LoadLibraryExW
  144. //sys FreeLibrary(handle Handle) (err error)
  145. //sys GetProcAddress(module Handle, procname string) (proc uintptr, err error)
  146. //sys GetModuleFileName(module Handle, filename *uint16, size uint32) (n uint32, err error) = kernel32.GetModuleFileNameW
  147. //sys GetModuleHandleEx(flags uint32, moduleName *uint16, module *Handle) (err error) = kernel32.GetModuleHandleExW
  148. //sys GetVersion() (ver uint32, err error)
  149. //sys FormatMessage(flags uint32, msgsrc uintptr, msgid uint32, langid uint32, buf []uint16, args *byte) (n uint32, err error) = FormatMessageW
  150. //sys ExitProcess(exitcode uint32)
  151. //sys IsWow64Process(handle Handle, isWow64 *bool) (err error) = IsWow64Process
  152. //sys CreateFile(name *uint16, access uint32, mode uint32, sa *SecurityAttributes, createmode uint32, attrs uint32, templatefile Handle) (handle Handle, err error) [failretval==InvalidHandle] = CreateFileW
  153. //sys ReadFile(handle Handle, buf []byte, done *uint32, overlapped *Overlapped) (err error)
  154. //sys WriteFile(handle Handle, buf []byte, done *uint32, overlapped *Overlapped) (err error)
  155. //sys GetOverlappedResult(handle Handle, overlapped *Overlapped, done *uint32, wait bool) (err error)
  156. //sys SetFilePointer(handle Handle, lowoffset int32, highoffsetptr *int32, whence uint32) (newlowoffset uint32, err error) [failretval==0xffffffff]
  157. //sys CloseHandle(handle Handle) (err error)
  158. //sys GetStdHandle(stdhandle uint32) (handle Handle, err error) [failretval==InvalidHandle]
  159. //sys SetStdHandle(stdhandle uint32, handle Handle) (err error)
  160. //sys findFirstFile1(name *uint16, data *win32finddata1) (handle Handle, err error) [failretval==InvalidHandle] = FindFirstFileW
  161. //sys findNextFile1(handle Handle, data *win32finddata1) (err error) = FindNextFileW
  162. //sys FindClose(handle Handle) (err error)
  163. //sys GetFileInformationByHandle(handle Handle, data *ByHandleFileInformation) (err error)
  164. //sys GetFileInformationByHandleEx(handle Handle, class uint32, outBuffer *byte, outBufferLen uint32) (err error)
  165. //sys GetCurrentDirectory(buflen uint32, buf *uint16) (n uint32, err error) = GetCurrentDirectoryW
  166. //sys SetCurrentDirectory(path *uint16) (err error) = SetCurrentDirectoryW
  167. //sys CreateDirectory(path *uint16, sa *SecurityAttributes) (err error) = CreateDirectoryW
  168. //sys RemoveDirectory(path *uint16) (err error) = RemoveDirectoryW
  169. //sys DeleteFile(path *uint16) (err error) = DeleteFileW
  170. //sys MoveFile(from *uint16, to *uint16) (err error) = MoveFileW
  171. //sys MoveFileEx(from *uint16, to *uint16, flags uint32) (err error) = MoveFileExW
  172. //sys LockFileEx(file Handle, flags uint32, reserved uint32, bytesLow uint32, bytesHigh uint32, overlapped *Overlapped) (err error)
  173. //sys UnlockFileEx(file Handle, reserved uint32, bytesLow uint32, bytesHigh uint32, overlapped *Overlapped) (err error)
  174. //sys GetComputerName(buf *uint16, n *uint32) (err error) = GetComputerNameW
  175. //sys GetComputerNameEx(nametype uint32, buf *uint16, n *uint32) (err error) = GetComputerNameExW
  176. //sys SetEndOfFile(handle Handle) (err error)
  177. //sys GetSystemTimeAsFileTime(time *Filetime)
  178. //sys GetSystemTimePreciseAsFileTime(time *Filetime)
  179. //sys GetTimeZoneInformation(tzi *Timezoneinformation) (rc uint32, err error) [failretval==0xffffffff]
  180. //sys CreateIoCompletionPort(filehandle Handle, cphandle Handle, key uint32, threadcnt uint32) (handle Handle, err error)
  181. //sys GetQueuedCompletionStatus(cphandle Handle, qty *uint32, key *uint32, overlapped **Overlapped, timeout uint32) (err error)
  182. //sys PostQueuedCompletionStatus(cphandle Handle, qty uint32, key uint32, overlapped *Overlapped) (err error)
  183. //sys CancelIo(s Handle) (err error)
  184. //sys CancelIoEx(s Handle, o *Overlapped) (err error)
  185. //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
  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 shGetKnownFolderPath(id *KNOWNFOLDERID, flags uint32, token Token, path **uint16) (ret error) = shell32.SHGetKnownFolderPath
  189. //sys TerminateProcess(handle Handle, exitcode uint32) (err error)
  190. //sys GetExitCodeProcess(handle Handle, exitcode *uint32) (err error)
  191. //sys GetStartupInfo(startupInfo *StartupInfo) (err error) = GetStartupInfoW
  192. //sys GetProcessTimes(handle Handle, creationTime *Filetime, exitTime *Filetime, kernelTime *Filetime, userTime *Filetime) (err error)
  193. //sys DuplicateHandle(hSourceProcessHandle Handle, hSourceHandle Handle, hTargetProcessHandle Handle, lpTargetHandle *Handle, dwDesiredAccess uint32, bInheritHandle bool, dwOptions uint32) (err error)
  194. //sys WaitForSingleObject(handle Handle, waitMilliseconds uint32) (event uint32, err error) [failretval==0xffffffff]
  195. //sys waitForMultipleObjects(count uint32, handles uintptr, waitAll bool, waitMilliseconds uint32) (event uint32, err error) [failretval==0xffffffff] = WaitForMultipleObjects
  196. //sys GetTempPath(buflen uint32, buf *uint16) (n uint32, err error) = GetTempPathW
  197. //sys CreatePipe(readhandle *Handle, writehandle *Handle, sa *SecurityAttributes, size uint32) (err error)
  198. //sys GetFileType(filehandle Handle) (n uint32, err error)
  199. //sys CryptAcquireContext(provhandle *Handle, container *uint16, provider *uint16, provtype uint32, flags uint32) (err error) = advapi32.CryptAcquireContextW
  200. //sys CryptReleaseContext(provhandle Handle, flags uint32) (err error) = advapi32.CryptReleaseContext
  201. //sys CryptGenRandom(provhandle Handle, buflen uint32, buf *byte) (err error) = advapi32.CryptGenRandom
  202. //sys GetEnvironmentStrings() (envs *uint16, err error) [failretval==nil] = kernel32.GetEnvironmentStringsW
  203. //sys FreeEnvironmentStrings(envs *uint16) (err error) = kernel32.FreeEnvironmentStringsW
  204. //sys GetEnvironmentVariable(name *uint16, buffer *uint16, size uint32) (n uint32, err error) = kernel32.GetEnvironmentVariableW
  205. //sys SetEnvironmentVariable(name *uint16, value *uint16) (err error) = kernel32.SetEnvironmentVariableW
  206. //sys CreateEnvironmentBlock(block **uint16, token Token, inheritExisting bool) (err error) = userenv.CreateEnvironmentBlock
  207. //sys DestroyEnvironmentBlock(block *uint16) (err error) = userenv.DestroyEnvironmentBlock
  208. //sys getTickCount64() (ms uint64) = kernel32.GetTickCount64
  209. //sys SetFileTime(handle Handle, ctime *Filetime, atime *Filetime, wtime *Filetime) (err error)
  210. //sys GetFileAttributes(name *uint16) (attrs uint32, err error) [failretval==INVALID_FILE_ATTRIBUTES] = kernel32.GetFileAttributesW
  211. //sys SetFileAttributes(name *uint16, attrs uint32) (err error) = kernel32.SetFileAttributesW
  212. //sys GetFileAttributesEx(name *uint16, level uint32, info *byte) (err error) = kernel32.GetFileAttributesExW
  213. //sys GetCommandLine() (cmd *uint16) = kernel32.GetCommandLineW
  214. //sys CommandLineToArgv(cmd *uint16, argc *int32) (argv *[8192]*[8192]uint16, err error) [failretval==nil] = shell32.CommandLineToArgvW
  215. //sys LocalFree(hmem Handle) (handle Handle, err error) [failretval!=0]
  216. //sys SetHandleInformation(handle Handle, mask uint32, flags uint32) (err error)
  217. //sys FlushFileBuffers(handle Handle) (err error)
  218. //sys GetFullPathName(path *uint16, buflen uint32, buf *uint16, fname **uint16) (n uint32, err error) = kernel32.GetFullPathNameW
  219. //sys GetLongPathName(path *uint16, buf *uint16, buflen uint32) (n uint32, err error) = kernel32.GetLongPathNameW
  220. //sys GetShortPathName(longpath *uint16, shortpath *uint16, buflen uint32) (n uint32, err error) = kernel32.GetShortPathNameW
  221. //sys CreateFileMapping(fhandle Handle, sa *SecurityAttributes, prot uint32, maxSizeHigh uint32, maxSizeLow uint32, name *uint16) (handle Handle, err error) = kernel32.CreateFileMappingW
  222. //sys MapViewOfFile(handle Handle, access uint32, offsetHigh uint32, offsetLow uint32, length uintptr) (addr uintptr, err error)
  223. //sys UnmapViewOfFile(addr uintptr) (err error)
  224. //sys FlushViewOfFile(addr uintptr, length uintptr) (err error)
  225. //sys VirtualLock(addr uintptr, length uintptr) (err error)
  226. //sys VirtualUnlock(addr uintptr, length uintptr) (err error)
  227. //sys VirtualAlloc(address uintptr, size uintptr, alloctype uint32, protect uint32) (value uintptr, err error) = kernel32.VirtualAlloc
  228. //sys VirtualFree(address uintptr, size uintptr, freetype uint32) (err error) = kernel32.VirtualFree
  229. //sys VirtualProtect(address uintptr, size uintptr, newprotect uint32, oldprotect *uint32) (err error) = kernel32.VirtualProtect
  230. //sys TransmitFile(s Handle, handle Handle, bytesToWrite uint32, bytsPerSend uint32, overlapped *Overlapped, transmitFileBuf *TransmitFileBuffers, flags uint32) (err error) = mswsock.TransmitFile
  231. //sys ReadDirectoryChanges(handle Handle, buf *byte, buflen uint32, watchSubTree bool, mask uint32, retlen *uint32, overlapped *Overlapped, completionRoutine uintptr) (err error) = kernel32.ReadDirectoryChangesW
  232. //sys CertOpenSystemStore(hprov Handle, name *uint16) (store Handle, err error) = crypt32.CertOpenSystemStoreW
  233. //sys CertOpenStore(storeProvider uintptr, msgAndCertEncodingType uint32, cryptProv uintptr, flags uint32, para uintptr) (handle Handle, err error) [failretval==InvalidHandle] = crypt32.CertOpenStore
  234. //sys CertEnumCertificatesInStore(store Handle, prevContext *CertContext) (context *CertContext, err error) [failretval==nil] = crypt32.CertEnumCertificatesInStore
  235. //sys CertAddCertificateContextToStore(store Handle, certContext *CertContext, addDisposition uint32, storeContext **CertContext) (err error) = crypt32.CertAddCertificateContextToStore
  236. //sys CertCloseStore(store Handle, flags uint32) (err error) = crypt32.CertCloseStore
  237. //sys CertGetCertificateChain(engine Handle, leaf *CertContext, time *Filetime, additionalStore Handle, para *CertChainPara, flags uint32, reserved uintptr, chainCtx **CertChainContext) (err error) = crypt32.CertGetCertificateChain
  238. //sys CertFreeCertificateChain(ctx *CertChainContext) = crypt32.CertFreeCertificateChain
  239. //sys CertCreateCertificateContext(certEncodingType uint32, certEncoded *byte, encodedLen uint32) (context *CertContext, err error) [failretval==nil] = crypt32.CertCreateCertificateContext
  240. //sys CertFreeCertificateContext(ctx *CertContext) (err error) = crypt32.CertFreeCertificateContext
  241. //sys CertVerifyCertificateChainPolicy(policyOID uintptr, chain *CertChainContext, para *CertChainPolicyPara, status *CertChainPolicyStatus) (err error) = crypt32.CertVerifyCertificateChainPolicy
  242. //sys RegOpenKeyEx(key Handle, subkey *uint16, options uint32, desiredAccess uint32, result *Handle) (regerrno error) = advapi32.RegOpenKeyExW
  243. //sys RegCloseKey(key Handle) (regerrno error) = advapi32.RegCloseKey
  244. //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
  245. //sys RegEnumKeyEx(key Handle, index uint32, name *uint16, nameLen *uint32, reserved *uint32, class *uint16, classLen *uint32, lastWriteTime *Filetime) (regerrno error) = advapi32.RegEnumKeyExW
  246. //sys RegQueryValueEx(key Handle, name *uint16, reserved *uint32, valtype *uint32, buf *byte, buflen *uint32) (regerrno error) = advapi32.RegQueryValueExW
  247. //sys GetCurrentProcessId() (pid uint32) = kernel32.GetCurrentProcessId
  248. //sys GetConsoleMode(console Handle, mode *uint32) (err error) = kernel32.GetConsoleMode
  249. //sys SetConsoleMode(console Handle, mode uint32) (err error) = kernel32.SetConsoleMode
  250. //sys GetConsoleScreenBufferInfo(console Handle, info *ConsoleScreenBufferInfo) (err error) = kernel32.GetConsoleScreenBufferInfo
  251. //sys WriteConsole(console Handle, buf *uint16, towrite uint32, written *uint32, reserved *byte) (err error) = kernel32.WriteConsoleW
  252. //sys ReadConsole(console Handle, buf *uint16, toread uint32, read *uint32, inputControl *byte) (err error) = kernel32.ReadConsoleW
  253. //sys CreateToolhelp32Snapshot(flags uint32, processId uint32) (handle Handle, err error) [failretval==InvalidHandle] = kernel32.CreateToolhelp32Snapshot
  254. //sys Process32First(snapshot Handle, procEntry *ProcessEntry32) (err error) = kernel32.Process32FirstW
  255. //sys Process32Next(snapshot Handle, procEntry *ProcessEntry32) (err error) = kernel32.Process32NextW
  256. //sys Thread32First(snapshot Handle, threadEntry *ThreadEntry32) (err error)
  257. //sys Thread32Next(snapshot Handle, threadEntry *ThreadEntry32) (err error)
  258. //sys DeviceIoControl(handle Handle, ioControlCode uint32, inBuffer *byte, inBufferSize uint32, outBuffer *byte, outBufferSize uint32, bytesReturned *uint32, overlapped *Overlapped) (err error)
  259. // This function returns 1 byte BOOLEAN rather than the 4 byte BOOL.
  260. //sys CreateSymbolicLink(symlinkfilename *uint16, targetfilename *uint16, flags uint32) (err error) [failretval&0xff==0] = CreateSymbolicLinkW
  261. //sys CreateHardLink(filename *uint16, existingfilename *uint16, reserved uintptr) (err error) [failretval&0xff==0] = CreateHardLinkW
  262. //sys GetCurrentThreadId() (id uint32)
  263. //sys CreateEvent(eventAttrs *SecurityAttributes, manualReset uint32, initialState uint32, name *uint16) (handle Handle, err error) = kernel32.CreateEventW
  264. //sys CreateEventEx(eventAttrs *SecurityAttributes, name *uint16, flags uint32, desiredAccess uint32) (handle Handle, err error) = kernel32.CreateEventExW
  265. //sys OpenEvent(desiredAccess uint32, inheritHandle bool, name *uint16) (handle Handle, err error) = kernel32.OpenEventW
  266. //sys SetEvent(event Handle) (err error) = kernel32.SetEvent
  267. //sys ResetEvent(event Handle) (err error) = kernel32.ResetEvent
  268. //sys PulseEvent(event Handle) (err error) = kernel32.PulseEvent
  269. //sys CreateMutex(mutexAttrs *SecurityAttributes, initialOwner bool, name *uint16) (handle Handle, err error) = kernel32.CreateMutexW
  270. //sys CreateMutexEx(mutexAttrs *SecurityAttributes, name *uint16, flags uint32, desiredAccess uint32) (handle Handle, err error) = kernel32.CreateMutexExW
  271. //sys OpenMutex(desiredAccess uint32, inheritHandle bool, name *uint16) (handle Handle, err error) = kernel32.OpenMutexW
  272. //sys ReleaseMutex(mutex Handle) (err error) = kernel32.ReleaseMutex
  273. //sys SleepEx(milliseconds uint32, alertable bool) (ret uint32) = kernel32.SleepEx
  274. //sys CreateJobObject(jobAttr *SecurityAttributes, name *uint16) (handle Handle, err error) = kernel32.CreateJobObjectW
  275. //sys AssignProcessToJobObject(job Handle, process Handle) (err error) = kernel32.AssignProcessToJobObject
  276. //sys TerminateJobObject(job Handle, exitCode uint32) (err error) = kernel32.TerminateJobObject
  277. //sys SetErrorMode(mode uint32) (ret uint32) = kernel32.SetErrorMode
  278. //sys ResumeThread(thread Handle) (ret uint32, err error) [failretval==0xffffffff] = kernel32.ResumeThread
  279. //sys SetPriorityClass(process Handle, priorityClass uint32) (err error) = kernel32.SetPriorityClass
  280. //sys GetPriorityClass(process Handle) (ret uint32, err error) = kernel32.GetPriorityClass
  281. //sys SetInformationJobObject(job Handle, JobObjectInformationClass uint32, JobObjectInformation uintptr, JobObjectInformationLength uint32) (ret int, err error)
  282. //sys GenerateConsoleCtrlEvent(ctrlEvent uint32, processGroupID uint32) (err error)
  283. //sys GetProcessId(process Handle) (id uint32, err error)
  284. //sys OpenThread(desiredAccess uint32, inheritHandle bool, threadId uint32) (handle Handle, err error)
  285. //sys SetProcessPriorityBoost(process Handle, disable bool) (err error) = kernel32.SetProcessPriorityBoost
  286. //sys GetProcessWorkingSetSizeEx(hProcess Handle, lpMinimumWorkingSetSize *uintptr, lpMaximumWorkingSetSize *uintptr, flags *uint32)
  287. //sys SetProcessWorkingSetSizeEx(hProcess Handle, dwMinimumWorkingSetSize uintptr, dwMaximumWorkingSetSize uintptr, flags uint32) (err error)
  288. // Volume Management Functions
  289. //sys DefineDosDevice(flags uint32, deviceName *uint16, targetPath *uint16) (err error) = DefineDosDeviceW
  290. //sys DeleteVolumeMountPoint(volumeMountPoint *uint16) (err error) = DeleteVolumeMountPointW
  291. //sys FindFirstVolume(volumeName *uint16, bufferLength uint32) (handle Handle, err error) [failretval==InvalidHandle] = FindFirstVolumeW
  292. //sys FindFirstVolumeMountPoint(rootPathName *uint16, volumeMountPoint *uint16, bufferLength uint32) (handle Handle, err error) [failretval==InvalidHandle] = FindFirstVolumeMountPointW
  293. //sys FindNextVolume(findVolume Handle, volumeName *uint16, bufferLength uint32) (err error) = FindNextVolumeW
  294. //sys FindNextVolumeMountPoint(findVolumeMountPoint Handle, volumeMountPoint *uint16, bufferLength uint32) (err error) = FindNextVolumeMountPointW
  295. //sys FindVolumeClose(findVolume Handle) (err error)
  296. //sys FindVolumeMountPointClose(findVolumeMountPoint Handle) (err error)
  297. //sys GetDiskFreeSpaceEx(directoryName *uint16, freeBytesAvailableToCaller *uint64, totalNumberOfBytes *uint64, totalNumberOfFreeBytes *uint64) (err error) = GetDiskFreeSpaceExW
  298. //sys GetDriveType(rootPathName *uint16) (driveType uint32) = GetDriveTypeW
  299. //sys GetLogicalDrives() (drivesBitMask uint32, err error) [failretval==0]
  300. //sys GetLogicalDriveStrings(bufferLength uint32, buffer *uint16) (n uint32, err error) [failretval==0] = GetLogicalDriveStringsW
  301. //sys GetVolumeInformation(rootPathName *uint16, volumeNameBuffer *uint16, volumeNameSize uint32, volumeNameSerialNumber *uint32, maximumComponentLength *uint32, fileSystemFlags *uint32, fileSystemNameBuffer *uint16, fileSystemNameSize uint32) (err error) = GetVolumeInformationW
  302. //sys GetVolumeInformationByHandle(file Handle, volumeNameBuffer *uint16, volumeNameSize uint32, volumeNameSerialNumber *uint32, maximumComponentLength *uint32, fileSystemFlags *uint32, fileSystemNameBuffer *uint16, fileSystemNameSize uint32) (err error) = GetVolumeInformationByHandleW
  303. //sys GetVolumeNameForVolumeMountPoint(volumeMountPoint *uint16, volumeName *uint16, bufferlength uint32) (err error) = GetVolumeNameForVolumeMountPointW
  304. //sys GetVolumePathName(fileName *uint16, volumePathName *uint16, bufferLength uint32) (err error) = GetVolumePathNameW
  305. //sys GetVolumePathNamesForVolumeName(volumeName *uint16, volumePathNames *uint16, bufferLength uint32, returnLength *uint32) (err error) = GetVolumePathNamesForVolumeNameW
  306. //sys QueryDosDevice(deviceName *uint16, targetPath *uint16, max uint32) (n uint32, err error) [failretval==0] = QueryDosDeviceW
  307. //sys SetVolumeLabel(rootPathName *uint16, volumeName *uint16) (err error) = SetVolumeLabelW
  308. //sys SetVolumeMountPoint(volumeMountPoint *uint16, volumeName *uint16) (err error) = SetVolumeMountPointW
  309. //sys MessageBox(hwnd Handle, text *uint16, caption *uint16, boxtype uint32) (ret int32, err error) [failretval==0] = user32.MessageBoxW
  310. //sys ExitWindowsEx(flags uint32, reason uint32) (err error) = user32.ExitWindowsEx
  311. //sys InitiateSystemShutdownEx(machineName *uint16, message *uint16, timeout uint32, forceAppsClosed bool, rebootAfterShutdown bool, reason uint32) (err error) = advapi32.InitiateSystemShutdownExW
  312. //sys SetProcessShutdownParameters(level uint32, flags uint32) (err error) = kernel32.SetProcessShutdownParameters
  313. //sys GetProcessShutdownParameters(level *uint32, flags *uint32) (err error) = kernel32.GetProcessShutdownParameters
  314. //sys clsidFromString(lpsz *uint16, pclsid *GUID) (ret error) = ole32.CLSIDFromString
  315. //sys stringFromGUID2(rguid *GUID, lpsz *uint16, cchMax int32) (chars int32) = ole32.StringFromGUID2
  316. //sys coCreateGuid(pguid *GUID) (ret error) = ole32.CoCreateGuid
  317. //sys CoTaskMemFree(address unsafe.Pointer) = ole32.CoTaskMemFree
  318. //sys rtlGetVersion(info *OsVersionInfoEx) (ret error) = ntdll.RtlGetVersion
  319. //sys rtlGetNtVersionNumbers(majorVersion *uint32, minorVersion *uint32, buildNumber *uint32) = ntdll.RtlGetNtVersionNumbers
  320. //sys getProcessPreferredUILanguages(flags uint32, numLanguages *uint32, buf *uint16, bufSize *uint32) (err error) = kernel32.GetProcessPreferredUILanguages
  321. //sys getThreadPreferredUILanguages(flags uint32, numLanguages *uint32, buf *uint16, bufSize *uint32) (err error) = kernel32.GetThreadPreferredUILanguages
  322. //sys getUserPreferredUILanguages(flags uint32, numLanguages *uint32, buf *uint16, bufSize *uint32) (err error) = kernel32.GetUserPreferredUILanguages
  323. //sys getSystemPreferredUILanguages(flags uint32, numLanguages *uint32, buf *uint16, bufSize *uint32) (err error) = kernel32.GetSystemPreferredUILanguages
  324. // Process Status API (PSAPI)
  325. //sys EnumProcesses(processIds []uint32, bytesReturned *uint32) (err error) = psapi.EnumProcesses
  326. // syscall interface implementation for other packages
  327. // GetCurrentProcess returns the handle for the current process.
  328. // It is a pseudo handle that does not need to be closed.
  329. // The returned error is always nil.
  330. //
  331. // Deprecated: use CurrentProcess for the same Handle without the nil
  332. // error.
  333. func GetCurrentProcess() (Handle, error) {
  334. return CurrentProcess(), nil
  335. }
  336. // CurrentProcess returns the handle for the current process.
  337. // It is a pseudo handle that does not need to be closed.
  338. func CurrentProcess() Handle { return Handle(^uintptr(1 - 1)) }
  339. // GetCurrentThread returns the handle for the current thread.
  340. // It is a pseudo handle that does not need to be closed.
  341. // The returned error is always nil.
  342. //
  343. // Deprecated: use CurrentThread for the same Handle without the nil
  344. // error.
  345. func GetCurrentThread() (Handle, error) {
  346. return CurrentThread(), nil
  347. }
  348. // CurrentThread returns the handle for the current thread.
  349. // It is a pseudo handle that does not need to be closed.
  350. func CurrentThread() Handle { return Handle(^uintptr(2 - 1)) }
  351. // GetProcAddressByOrdinal retrieves the address of the exported
  352. // function from module by ordinal.
  353. func GetProcAddressByOrdinal(module Handle, ordinal uintptr) (proc uintptr, err error) {
  354. r0, _, e1 := syscall.Syscall(procGetProcAddress.Addr(), 2, uintptr(module), ordinal, 0)
  355. proc = uintptr(r0)
  356. if proc == 0 {
  357. if e1 != 0 {
  358. err = errnoErr(e1)
  359. } else {
  360. err = syscall.EINVAL
  361. }
  362. }
  363. return
  364. }
  365. func Exit(code int) { ExitProcess(uint32(code)) }
  366. func makeInheritSa() *SecurityAttributes {
  367. var sa SecurityAttributes
  368. sa.Length = uint32(unsafe.Sizeof(sa))
  369. sa.InheritHandle = 1
  370. return &sa
  371. }
  372. func Open(path string, mode int, perm uint32) (fd Handle, err error) {
  373. if len(path) == 0 {
  374. return InvalidHandle, ERROR_FILE_NOT_FOUND
  375. }
  376. pathp, err := UTF16PtrFromString(path)
  377. if err != nil {
  378. return InvalidHandle, err
  379. }
  380. var access uint32
  381. switch mode & (O_RDONLY | O_WRONLY | O_RDWR) {
  382. case O_RDONLY:
  383. access = GENERIC_READ
  384. case O_WRONLY:
  385. access = GENERIC_WRITE
  386. case O_RDWR:
  387. access = GENERIC_READ | GENERIC_WRITE
  388. }
  389. if mode&O_CREAT != 0 {
  390. access |= GENERIC_WRITE
  391. }
  392. if mode&O_APPEND != 0 {
  393. access &^= GENERIC_WRITE
  394. access |= FILE_APPEND_DATA
  395. }
  396. sharemode := uint32(FILE_SHARE_READ | FILE_SHARE_WRITE)
  397. var sa *SecurityAttributes
  398. if mode&O_CLOEXEC == 0 {
  399. sa = makeInheritSa()
  400. }
  401. var createmode uint32
  402. switch {
  403. case mode&(O_CREAT|O_EXCL) == (O_CREAT | O_EXCL):
  404. createmode = CREATE_NEW
  405. case mode&(O_CREAT|O_TRUNC) == (O_CREAT | O_TRUNC):
  406. createmode = CREATE_ALWAYS
  407. case mode&O_CREAT == O_CREAT:
  408. createmode = OPEN_ALWAYS
  409. case mode&O_TRUNC == O_TRUNC:
  410. createmode = TRUNCATE_EXISTING
  411. default:
  412. createmode = OPEN_EXISTING
  413. }
  414. var attrs uint32 = FILE_ATTRIBUTE_NORMAL
  415. if perm&S_IWRITE == 0 {
  416. attrs = FILE_ATTRIBUTE_READONLY
  417. }
  418. h, e := CreateFile(pathp, access, sharemode, sa, createmode, attrs, 0)
  419. return h, e
  420. }
  421. func Read(fd Handle, p []byte) (n int, err error) {
  422. var done uint32
  423. e := ReadFile(fd, p, &done, nil)
  424. if e != nil {
  425. if e == ERROR_BROKEN_PIPE {
  426. // NOTE(brainman): work around ERROR_BROKEN_PIPE is returned on reading EOF from stdin
  427. return 0, nil
  428. }
  429. return 0, e
  430. }
  431. if raceenabled {
  432. if done > 0 {
  433. raceWriteRange(unsafe.Pointer(&p[0]), int(done))
  434. }
  435. raceAcquire(unsafe.Pointer(&ioSync))
  436. }
  437. return int(done), nil
  438. }
  439. func Write(fd Handle, p []byte) (n int, err error) {
  440. if raceenabled {
  441. raceReleaseMerge(unsafe.Pointer(&ioSync))
  442. }
  443. var done uint32
  444. e := WriteFile(fd, p, &done, nil)
  445. if e != nil {
  446. return 0, e
  447. }
  448. if raceenabled && done > 0 {
  449. raceReadRange(unsafe.Pointer(&p[0]), int(done))
  450. }
  451. return int(done), nil
  452. }
  453. var ioSync int64
  454. func Seek(fd Handle, offset int64, whence int) (newoffset int64, err error) {
  455. var w uint32
  456. switch whence {
  457. case 0:
  458. w = FILE_BEGIN
  459. case 1:
  460. w = FILE_CURRENT
  461. case 2:
  462. w = FILE_END
  463. }
  464. hi := int32(offset >> 32)
  465. lo := int32(offset)
  466. // use GetFileType to check pipe, pipe can't do seek
  467. ft, _ := GetFileType(fd)
  468. if ft == FILE_TYPE_PIPE {
  469. return 0, syscall.EPIPE
  470. }
  471. rlo, e := SetFilePointer(fd, lo, &hi, w)
  472. if e != nil {
  473. return 0, e
  474. }
  475. return int64(hi)<<32 + int64(rlo), nil
  476. }
  477. func Close(fd Handle) (err error) {
  478. return CloseHandle(fd)
  479. }
  480. var (
  481. Stdin = getStdHandle(STD_INPUT_HANDLE)
  482. Stdout = getStdHandle(STD_OUTPUT_HANDLE)
  483. Stderr = getStdHandle(STD_ERROR_HANDLE)
  484. )
  485. func getStdHandle(stdhandle uint32) (fd Handle) {
  486. r, _ := GetStdHandle(stdhandle)
  487. CloseOnExec(r)
  488. return r
  489. }
  490. const ImplementsGetwd = true
  491. func Getwd() (wd string, err error) {
  492. b := make([]uint16, 300)
  493. n, e := GetCurrentDirectory(uint32(len(b)), &b[0])
  494. if e != nil {
  495. return "", e
  496. }
  497. return string(utf16.Decode(b[0:n])), nil
  498. }
  499. func Chdir(path string) (err error) {
  500. pathp, err := UTF16PtrFromString(path)
  501. if err != nil {
  502. return err
  503. }
  504. return SetCurrentDirectory(pathp)
  505. }
  506. func Mkdir(path string, mode uint32) (err error) {
  507. pathp, err := UTF16PtrFromString(path)
  508. if err != nil {
  509. return err
  510. }
  511. return CreateDirectory(pathp, nil)
  512. }
  513. func Rmdir(path string) (err error) {
  514. pathp, err := UTF16PtrFromString(path)
  515. if err != nil {
  516. return err
  517. }
  518. return RemoveDirectory(pathp)
  519. }
  520. func Unlink(path string) (err error) {
  521. pathp, err := UTF16PtrFromString(path)
  522. if err != nil {
  523. return err
  524. }
  525. return DeleteFile(pathp)
  526. }
  527. func Rename(oldpath, newpath string) (err error) {
  528. from, err := UTF16PtrFromString(oldpath)
  529. if err != nil {
  530. return err
  531. }
  532. to, err := UTF16PtrFromString(newpath)
  533. if err != nil {
  534. return err
  535. }
  536. return MoveFileEx(from, to, MOVEFILE_REPLACE_EXISTING)
  537. }
  538. func ComputerName() (name string, err error) {
  539. var n uint32 = MAX_COMPUTERNAME_LENGTH + 1
  540. b := make([]uint16, n)
  541. e := GetComputerName(&b[0], &n)
  542. if e != nil {
  543. return "", e
  544. }
  545. return string(utf16.Decode(b[0:n])), nil
  546. }
  547. func DurationSinceBoot() time.Duration {
  548. return time.Duration(getTickCount64()) * time.Millisecond
  549. }
  550. func Ftruncate(fd Handle, length int64) (err error) {
  551. curoffset, e := Seek(fd, 0, 1)
  552. if e != nil {
  553. return e
  554. }
  555. defer Seek(fd, curoffset, 0)
  556. _, e = Seek(fd, length, 0)
  557. if e != nil {
  558. return e
  559. }
  560. e = SetEndOfFile(fd)
  561. if e != nil {
  562. return e
  563. }
  564. return nil
  565. }
  566. func Gettimeofday(tv *Timeval) (err error) {
  567. var ft Filetime
  568. GetSystemTimeAsFileTime(&ft)
  569. *tv = NsecToTimeval(ft.Nanoseconds())
  570. return nil
  571. }
  572. func Pipe(p []Handle) (err error) {
  573. if len(p) != 2 {
  574. return syscall.EINVAL
  575. }
  576. var r, w Handle
  577. e := CreatePipe(&r, &w, makeInheritSa(), 0)
  578. if e != nil {
  579. return e
  580. }
  581. p[0] = r
  582. p[1] = w
  583. return nil
  584. }
  585. func Utimes(path string, tv []Timeval) (err error) {
  586. if len(tv) != 2 {
  587. return syscall.EINVAL
  588. }
  589. pathp, e := UTF16PtrFromString(path)
  590. if e != nil {
  591. return e
  592. }
  593. h, e := CreateFile(pathp,
  594. FILE_WRITE_ATTRIBUTES, FILE_SHARE_WRITE, nil,
  595. OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, 0)
  596. if e != nil {
  597. return e
  598. }
  599. defer Close(h)
  600. a := NsecToFiletime(tv[0].Nanoseconds())
  601. w := NsecToFiletime(tv[1].Nanoseconds())
  602. return SetFileTime(h, nil, &a, &w)
  603. }
  604. func UtimesNano(path string, ts []Timespec) (err error) {
  605. if len(ts) != 2 {
  606. return syscall.EINVAL
  607. }
  608. pathp, e := UTF16PtrFromString(path)
  609. if e != nil {
  610. return e
  611. }
  612. h, e := CreateFile(pathp,
  613. FILE_WRITE_ATTRIBUTES, FILE_SHARE_WRITE, nil,
  614. OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, 0)
  615. if e != nil {
  616. return e
  617. }
  618. defer Close(h)
  619. a := NsecToFiletime(TimespecToNsec(ts[0]))
  620. w := NsecToFiletime(TimespecToNsec(ts[1]))
  621. return SetFileTime(h, nil, &a, &w)
  622. }
  623. func Fsync(fd Handle) (err error) {
  624. return FlushFileBuffers(fd)
  625. }
  626. func Chmod(path string, mode uint32) (err error) {
  627. p, e := UTF16PtrFromString(path)
  628. if e != nil {
  629. return e
  630. }
  631. attrs, e := GetFileAttributes(p)
  632. if e != nil {
  633. return e
  634. }
  635. if mode&S_IWRITE != 0 {
  636. attrs &^= FILE_ATTRIBUTE_READONLY
  637. } else {
  638. attrs |= FILE_ATTRIBUTE_READONLY
  639. }
  640. return SetFileAttributes(p, attrs)
  641. }
  642. func LoadGetSystemTimePreciseAsFileTime() error {
  643. return procGetSystemTimePreciseAsFileTime.Find()
  644. }
  645. func LoadCancelIoEx() error {
  646. return procCancelIoEx.Find()
  647. }
  648. func LoadSetFileCompletionNotificationModes() error {
  649. return procSetFileCompletionNotificationModes.Find()
  650. }
  651. func WaitForMultipleObjects(handles []Handle, waitAll bool, waitMilliseconds uint32) (event uint32, err error) {
  652. // Every other win32 array API takes arguments as "pointer, count", except for this function. So we
  653. // can't declare it as a usual [] type, because mksyscall will use the opposite order. We therefore
  654. // trivially stub this ourselves.
  655. var handlePtr *Handle
  656. if len(handles) > 0 {
  657. handlePtr = &handles[0]
  658. }
  659. return waitForMultipleObjects(uint32(len(handles)), uintptr(unsafe.Pointer(handlePtr)), waitAll, waitMilliseconds)
  660. }
  661. // net api calls
  662. const socket_error = uintptr(^uint32(0))
  663. //sys WSAStartup(verreq uint32, data *WSAData) (sockerr error) = ws2_32.WSAStartup
  664. //sys WSACleanup() (err error) [failretval==socket_error] = ws2_32.WSACleanup
  665. //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
  666. //sys socket(af int32, typ int32, protocol int32) (handle Handle, err error) [failretval==InvalidHandle] = ws2_32.socket
  667. //sys sendto(s Handle, buf []byte, flags int32, to unsafe.Pointer, tolen int32) (err error) [failretval==socket_error] = ws2_32.sendto
  668. //sys recvfrom(s Handle, buf []byte, flags int32, from *RawSockaddrAny, fromlen *int32) (n int32, err error) [failretval==-1] = ws2_32.recvfrom
  669. //sys Setsockopt(s Handle, level int32, optname int32, optval *byte, optlen int32) (err error) [failretval==socket_error] = ws2_32.setsockopt
  670. //sys Getsockopt(s Handle, level int32, optname int32, optval *byte, optlen *int32) (err error) [failretval==socket_error] = ws2_32.getsockopt
  671. //sys bind(s Handle, name unsafe.Pointer, namelen int32) (err error) [failretval==socket_error] = ws2_32.bind
  672. //sys connect(s Handle, name unsafe.Pointer, namelen int32) (err error) [failretval==socket_error] = ws2_32.connect
  673. //sys getsockname(s Handle, rsa *RawSockaddrAny, addrlen *int32) (err error) [failretval==socket_error] = ws2_32.getsockname
  674. //sys getpeername(s Handle, rsa *RawSockaddrAny, addrlen *int32) (err error) [failretval==socket_error] = ws2_32.getpeername
  675. //sys listen(s Handle, backlog int32) (err error) [failretval==socket_error] = ws2_32.listen
  676. //sys shutdown(s Handle, how int32) (err error) [failretval==socket_error] = ws2_32.shutdown
  677. //sys Closesocket(s Handle) (err error) [failretval==socket_error] = ws2_32.closesocket
  678. //sys AcceptEx(ls Handle, as Handle, buf *byte, rxdatalen uint32, laddrlen uint32, raddrlen uint32, recvd *uint32, overlapped *Overlapped) (err error) = mswsock.AcceptEx
  679. //sys GetAcceptExSockaddrs(buf *byte, rxdatalen uint32, laddrlen uint32, raddrlen uint32, lrsa **RawSockaddrAny, lrsalen *int32, rrsa **RawSockaddrAny, rrsalen *int32) = mswsock.GetAcceptExSockaddrs
  680. //sys WSARecv(s Handle, bufs *WSABuf, bufcnt uint32, recvd *uint32, flags *uint32, overlapped *Overlapped, croutine *byte) (err error) [failretval==socket_error] = ws2_32.WSARecv
  681. //sys WSASend(s Handle, bufs *WSABuf, bufcnt uint32, sent *uint32, flags uint32, overlapped *Overlapped, croutine *byte) (err error) [failretval==socket_error] = ws2_32.WSASend
  682. //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
  683. //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
  684. //sys GetHostByName(name string) (h *Hostent, err error) [failretval==nil] = ws2_32.gethostbyname
  685. //sys GetServByName(name string, proto string) (s *Servent, err error) [failretval==nil] = ws2_32.getservbyname
  686. //sys Ntohs(netshort uint16) (u uint16) = ws2_32.ntohs
  687. //sys GetProtoByName(name string) (p *Protoent, err error) [failretval==nil] = ws2_32.getprotobyname
  688. //sys DnsQuery(name string, qtype uint16, options uint32, extra *byte, qrs **DNSRecord, pr *byte) (status error) = dnsapi.DnsQuery_W
  689. //sys DnsRecordListFree(rl *DNSRecord, freetype uint32) = dnsapi.DnsRecordListFree
  690. //sys DnsNameCompare(name1 *uint16, name2 *uint16) (same bool) = dnsapi.DnsNameCompare_W
  691. //sys GetAddrInfoW(nodename *uint16, servicename *uint16, hints *AddrinfoW, result **AddrinfoW) (sockerr error) = ws2_32.GetAddrInfoW
  692. //sys FreeAddrInfoW(addrinfo *AddrinfoW) = ws2_32.FreeAddrInfoW
  693. //sys GetIfEntry(pIfRow *MibIfRow) (errcode error) = iphlpapi.GetIfEntry
  694. //sys GetAdaptersInfo(ai *IpAdapterInfo, ol *uint32) (errcode error) = iphlpapi.GetAdaptersInfo
  695. //sys SetFileCompletionNotificationModes(handle Handle, flags uint8) (err error) = kernel32.SetFileCompletionNotificationModes
  696. //sys WSAEnumProtocols(protocols *int32, protocolBuffer *WSAProtocolInfo, bufferLength *uint32) (n int32, err error) [failretval==-1] = ws2_32.WSAEnumProtocolsW
  697. //sys GetAdaptersAddresses(family uint32, flags uint32, reserved uintptr, adapterAddresses *IpAdapterAddresses, sizePointer *uint32) (errcode error) = iphlpapi.GetAdaptersAddresses
  698. //sys GetACP() (acp uint32) = kernel32.GetACP
  699. //sys MultiByteToWideChar(codePage uint32, dwFlags uint32, str *byte, nstr int32, wchar *uint16, nwchar int32) (nwrite int32, err error) = kernel32.MultiByteToWideChar
  700. // For testing: clients can set this flag to force
  701. // creation of IPv6 sockets to return EAFNOSUPPORT.
  702. var SocketDisableIPv6 bool
  703. type RawSockaddrInet4 struct {
  704. Family uint16
  705. Port uint16
  706. Addr [4]byte /* in_addr */
  707. Zero [8]uint8
  708. }
  709. type RawSockaddrInet6 struct {
  710. Family uint16
  711. Port uint16
  712. Flowinfo uint32
  713. Addr [16]byte /* in6_addr */
  714. Scope_id uint32
  715. }
  716. type RawSockaddr struct {
  717. Family uint16
  718. Data [14]int8
  719. }
  720. type RawSockaddrAny struct {
  721. Addr RawSockaddr
  722. Pad [100]int8
  723. }
  724. type Sockaddr interface {
  725. sockaddr() (ptr unsafe.Pointer, len int32, err error) // lowercase; only we can define Sockaddrs
  726. }
  727. type SockaddrInet4 struct {
  728. Port int
  729. Addr [4]byte
  730. raw RawSockaddrInet4
  731. }
  732. func (sa *SockaddrInet4) sockaddr() (unsafe.Pointer, int32, error) {
  733. if sa.Port < 0 || sa.Port > 0xFFFF {
  734. return nil, 0, syscall.EINVAL
  735. }
  736. sa.raw.Family = AF_INET
  737. p := (*[2]byte)(unsafe.Pointer(&sa.raw.Port))
  738. p[0] = byte(sa.Port >> 8)
  739. p[1] = byte(sa.Port)
  740. for i := 0; i < len(sa.Addr); i++ {
  741. sa.raw.Addr[i] = sa.Addr[i]
  742. }
  743. return unsafe.Pointer(&sa.raw), int32(unsafe.Sizeof(sa.raw)), nil
  744. }
  745. type SockaddrInet6 struct {
  746. Port int
  747. ZoneId uint32
  748. Addr [16]byte
  749. raw RawSockaddrInet6
  750. }
  751. func (sa *SockaddrInet6) sockaddr() (unsafe.Pointer, int32, error) {
  752. if sa.Port < 0 || sa.Port > 0xFFFF {
  753. return nil, 0, syscall.EINVAL
  754. }
  755. sa.raw.Family = AF_INET6
  756. p := (*[2]byte)(unsafe.Pointer(&sa.raw.Port))
  757. p[0] = byte(sa.Port >> 8)
  758. p[1] = byte(sa.Port)
  759. sa.raw.Scope_id = sa.ZoneId
  760. for i := 0; i < len(sa.Addr); i++ {
  761. sa.raw.Addr[i] = sa.Addr[i]
  762. }
  763. return unsafe.Pointer(&sa.raw), int32(unsafe.Sizeof(sa.raw)), nil
  764. }
  765. type RawSockaddrUnix struct {
  766. Family uint16
  767. Path [UNIX_PATH_MAX]int8
  768. }
  769. type SockaddrUnix struct {
  770. Name string
  771. raw RawSockaddrUnix
  772. }
  773. func (sa *SockaddrUnix) sockaddr() (unsafe.Pointer, int32, error) {
  774. name := sa.Name
  775. n := len(name)
  776. if n > len(sa.raw.Path) {
  777. return nil, 0, syscall.EINVAL
  778. }
  779. if n == len(sa.raw.Path) && name[0] != '@' {
  780. return nil, 0, syscall.EINVAL
  781. }
  782. sa.raw.Family = AF_UNIX
  783. for i := 0; i < n; i++ {
  784. sa.raw.Path[i] = int8(name[i])
  785. }
  786. // length is family (uint16), name, NUL.
  787. sl := int32(2)
  788. if n > 0 {
  789. sl += int32(n) + 1
  790. }
  791. if sa.raw.Path[0] == '@' {
  792. sa.raw.Path[0] = 0
  793. // Don't count trailing NUL for abstract address.
  794. sl--
  795. }
  796. return unsafe.Pointer(&sa.raw), sl, nil
  797. }
  798. func (rsa *RawSockaddrAny) Sockaddr() (Sockaddr, error) {
  799. switch rsa.Addr.Family {
  800. case AF_UNIX:
  801. pp := (*RawSockaddrUnix)(unsafe.Pointer(rsa))
  802. sa := new(SockaddrUnix)
  803. if pp.Path[0] == 0 {
  804. // "Abstract" Unix domain socket.
  805. // Rewrite leading NUL as @ for textual display.
  806. // (This is the standard convention.)
  807. // Not friendly to overwrite in place,
  808. // but the callers below don't care.
  809. pp.Path[0] = '@'
  810. }
  811. // Assume path ends at NUL.
  812. // This is not technically the Linux semantics for
  813. // abstract Unix domain sockets--they are supposed
  814. // to be uninterpreted fixed-size binary blobs--but
  815. // everyone uses this convention.
  816. n := 0
  817. for n < len(pp.Path) && pp.Path[n] != 0 {
  818. n++
  819. }
  820. bytes := (*[len(pp.Path)]byte)(unsafe.Pointer(&pp.Path[0]))[0:n]
  821. sa.Name = string(bytes)
  822. return sa, nil
  823. case AF_INET:
  824. pp := (*RawSockaddrInet4)(unsafe.Pointer(rsa))
  825. sa := new(SockaddrInet4)
  826. p := (*[2]byte)(unsafe.Pointer(&pp.Port))
  827. sa.Port = int(p[0])<<8 + int(p[1])
  828. for i := 0; i < len(sa.Addr); i++ {
  829. sa.Addr[i] = pp.Addr[i]
  830. }
  831. return sa, nil
  832. case AF_INET6:
  833. pp := (*RawSockaddrInet6)(unsafe.Pointer(rsa))
  834. sa := new(SockaddrInet6)
  835. p := (*[2]byte)(unsafe.Pointer(&pp.Port))
  836. sa.Port = int(p[0])<<8 + int(p[1])
  837. sa.ZoneId = pp.Scope_id
  838. for i := 0; i < len(sa.Addr); i++ {
  839. sa.Addr[i] = pp.Addr[i]
  840. }
  841. return sa, nil
  842. }
  843. return nil, syscall.EAFNOSUPPORT
  844. }
  845. func Socket(domain, typ, proto int) (fd Handle, err error) {
  846. if domain == AF_INET6 && SocketDisableIPv6 {
  847. return InvalidHandle, syscall.EAFNOSUPPORT
  848. }
  849. return socket(int32(domain), int32(typ), int32(proto))
  850. }
  851. func SetsockoptInt(fd Handle, level, opt int, value int) (err error) {
  852. v := int32(value)
  853. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(&v)), int32(unsafe.Sizeof(v)))
  854. }
  855. func Bind(fd Handle, sa Sockaddr) (err error) {
  856. ptr, n, err := sa.sockaddr()
  857. if err != nil {
  858. return err
  859. }
  860. return bind(fd, ptr, n)
  861. }
  862. func Connect(fd Handle, sa Sockaddr) (err error) {
  863. ptr, n, err := sa.sockaddr()
  864. if err != nil {
  865. return err
  866. }
  867. return connect(fd, ptr, n)
  868. }
  869. func Getsockname(fd Handle) (sa Sockaddr, err error) {
  870. var rsa RawSockaddrAny
  871. l := int32(unsafe.Sizeof(rsa))
  872. if err = getsockname(fd, &rsa, &l); err != nil {
  873. return
  874. }
  875. return rsa.Sockaddr()
  876. }
  877. func Getpeername(fd Handle) (sa Sockaddr, err error) {
  878. var rsa RawSockaddrAny
  879. l := int32(unsafe.Sizeof(rsa))
  880. if err = getpeername(fd, &rsa, &l); err != nil {
  881. return
  882. }
  883. return rsa.Sockaddr()
  884. }
  885. func Listen(s Handle, n int) (err error) {
  886. return listen(s, int32(n))
  887. }
  888. func Shutdown(fd Handle, how int) (err error) {
  889. return shutdown(fd, int32(how))
  890. }
  891. func WSASendto(s Handle, bufs *WSABuf, bufcnt uint32, sent *uint32, flags uint32, to Sockaddr, overlapped *Overlapped, croutine *byte) (err error) {
  892. rsa, l, err := to.sockaddr()
  893. if err != nil {
  894. return err
  895. }
  896. return WSASendTo(s, bufs, bufcnt, sent, flags, (*RawSockaddrAny)(unsafe.Pointer(rsa)), l, overlapped, croutine)
  897. }
  898. func LoadGetAddrInfo() error {
  899. return procGetAddrInfoW.Find()
  900. }
  901. var connectExFunc struct {
  902. once sync.Once
  903. addr uintptr
  904. err error
  905. }
  906. func LoadConnectEx() error {
  907. connectExFunc.once.Do(func() {
  908. var s Handle
  909. s, connectExFunc.err = Socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)
  910. if connectExFunc.err != nil {
  911. return
  912. }
  913. defer CloseHandle(s)
  914. var n uint32
  915. connectExFunc.err = WSAIoctl(s,
  916. SIO_GET_EXTENSION_FUNCTION_POINTER,
  917. (*byte)(unsafe.Pointer(&WSAID_CONNECTEX)),
  918. uint32(unsafe.Sizeof(WSAID_CONNECTEX)),
  919. (*byte)(unsafe.Pointer(&connectExFunc.addr)),
  920. uint32(unsafe.Sizeof(connectExFunc.addr)),
  921. &n, nil, 0)
  922. })
  923. return connectExFunc.err
  924. }
  925. func connectEx(s Handle, name unsafe.Pointer, namelen int32, sendBuf *byte, sendDataLen uint32, bytesSent *uint32, overlapped *Overlapped) (err error) {
  926. 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)
  927. if r1 == 0 {
  928. if e1 != 0 {
  929. err = error(e1)
  930. } else {
  931. err = syscall.EINVAL
  932. }
  933. }
  934. return
  935. }
  936. func ConnectEx(fd Handle, sa Sockaddr, sendBuf *byte, sendDataLen uint32, bytesSent *uint32, overlapped *Overlapped) error {
  937. err := LoadConnectEx()
  938. if err != nil {
  939. return errorspkg.New("failed to find ConnectEx: " + err.Error())
  940. }
  941. ptr, n, err := sa.sockaddr()
  942. if err != nil {
  943. return err
  944. }
  945. return connectEx(fd, ptr, n, sendBuf, sendDataLen, bytesSent, overlapped)
  946. }
  947. var sendRecvMsgFunc struct {
  948. once sync.Once
  949. sendAddr uintptr
  950. recvAddr uintptr
  951. err error
  952. }
  953. func loadWSASendRecvMsg() error {
  954. sendRecvMsgFunc.once.Do(func() {
  955. var s Handle
  956. s, sendRecvMsgFunc.err = Socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)
  957. if sendRecvMsgFunc.err != nil {
  958. return
  959. }
  960. defer CloseHandle(s)
  961. var n uint32
  962. sendRecvMsgFunc.err = WSAIoctl(s,
  963. SIO_GET_EXTENSION_FUNCTION_POINTER,
  964. (*byte)(unsafe.Pointer(&WSAID_WSARECVMSG)),
  965. uint32(unsafe.Sizeof(WSAID_WSARECVMSG)),
  966. (*byte)(unsafe.Pointer(&sendRecvMsgFunc.recvAddr)),
  967. uint32(unsafe.Sizeof(sendRecvMsgFunc.recvAddr)),
  968. &n, nil, 0)
  969. if sendRecvMsgFunc.err != nil {
  970. return
  971. }
  972. sendRecvMsgFunc.err = WSAIoctl(s,
  973. SIO_GET_EXTENSION_FUNCTION_POINTER,
  974. (*byte)(unsafe.Pointer(&WSAID_WSASENDMSG)),
  975. uint32(unsafe.Sizeof(WSAID_WSASENDMSG)),
  976. (*byte)(unsafe.Pointer(&sendRecvMsgFunc.sendAddr)),
  977. uint32(unsafe.Sizeof(sendRecvMsgFunc.sendAddr)),
  978. &n, nil, 0)
  979. })
  980. return sendRecvMsgFunc.err
  981. }
  982. func WSASendMsg(fd Handle, msg *WSAMsg, flags uint32, bytesSent *uint32, overlapped *Overlapped, croutine *byte) error {
  983. err := loadWSASendRecvMsg()
  984. if err != nil {
  985. return err
  986. }
  987. 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)))
  988. if r1 == socket_error {
  989. if e1 != 0 {
  990. err = errnoErr(e1)
  991. } else {
  992. err = syscall.EINVAL
  993. }
  994. }
  995. return err
  996. }
  997. func WSARecvMsg(fd Handle, msg *WSAMsg, bytesReceived *uint32, overlapped *Overlapped, croutine *byte) error {
  998. err := loadWSASendRecvMsg()
  999. if err != nil {
  1000. return err
  1001. }
  1002. 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)
  1003. if r1 == socket_error {
  1004. if e1 != 0 {
  1005. err = errnoErr(e1)
  1006. } else {
  1007. err = syscall.EINVAL
  1008. }
  1009. }
  1010. return err
  1011. }
  1012. // Invented structures to support what package os expects.
  1013. type Rusage struct {
  1014. CreationTime Filetime
  1015. ExitTime Filetime
  1016. KernelTime Filetime
  1017. UserTime Filetime
  1018. }
  1019. type WaitStatus struct {
  1020. ExitCode uint32
  1021. }
  1022. func (w WaitStatus) Exited() bool { return true }
  1023. func (w WaitStatus) ExitStatus() int { return int(w.ExitCode) }
  1024. func (w WaitStatus) Signal() Signal { return -1 }
  1025. func (w WaitStatus) CoreDump() bool { return false }
  1026. func (w WaitStatus) Stopped() bool { return false }
  1027. func (w WaitStatus) Continued() bool { return false }
  1028. func (w WaitStatus) StopSignal() Signal { return -1 }
  1029. func (w WaitStatus) Signaled() bool { return false }
  1030. func (w WaitStatus) TrapCause() int { return -1 }
  1031. // Timespec is an invented structure on Windows, but here for
  1032. // consistency with the corresponding package for other operating systems.
  1033. type Timespec struct {
  1034. Sec int64
  1035. Nsec int64
  1036. }
  1037. func TimespecToNsec(ts Timespec) int64 { return int64(ts.Sec)*1e9 + int64(ts.Nsec) }
  1038. func NsecToTimespec(nsec int64) (ts Timespec) {
  1039. ts.Sec = nsec / 1e9
  1040. ts.Nsec = nsec % 1e9
  1041. return
  1042. }
  1043. // TODO(brainman): fix all needed for net
  1044. func Accept(fd Handle) (nfd Handle, sa Sockaddr, err error) { return 0, nil, syscall.EWINDOWS }
  1045. func Recvfrom(fd Handle, p []byte, flags int) (n int, from Sockaddr, err error) {
  1046. var rsa RawSockaddrAny
  1047. l := int32(unsafe.Sizeof(rsa))
  1048. n32, err := recvfrom(fd, p, int32(flags), &rsa, &l)
  1049. n = int(n32)
  1050. if err != nil {
  1051. return
  1052. }
  1053. from, err = rsa.Sockaddr()
  1054. return
  1055. }
  1056. func Sendto(fd Handle, p []byte, flags int, to Sockaddr) (err error) {
  1057. ptr, l, err := to.sockaddr()
  1058. if err != nil {
  1059. return err
  1060. }
  1061. return sendto(fd, p, int32(flags), ptr, l)
  1062. }
  1063. func SetsockoptTimeval(fd Handle, level, opt int, tv *Timeval) (err error) { return syscall.EWINDOWS }
  1064. // The Linger struct is wrong but we only noticed after Go 1.
  1065. // sysLinger is the real system call structure.
  1066. // BUG(brainman): The definition of Linger is not appropriate for direct use
  1067. // with Setsockopt and Getsockopt.
  1068. // Use SetsockoptLinger instead.
  1069. type Linger struct {
  1070. Onoff int32
  1071. Linger int32
  1072. }
  1073. type sysLinger struct {
  1074. Onoff uint16
  1075. Linger uint16
  1076. }
  1077. type IPMreq struct {
  1078. Multiaddr [4]byte /* in_addr */
  1079. Interface [4]byte /* in_addr */
  1080. }
  1081. type IPv6Mreq struct {
  1082. Multiaddr [16]byte /* in6_addr */
  1083. Interface uint32
  1084. }
  1085. func GetsockoptInt(fd Handle, level, opt int) (int, error) {
  1086. v := int32(0)
  1087. l := int32(unsafe.Sizeof(v))
  1088. err := Getsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(&v)), &l)
  1089. return int(v), err
  1090. }
  1091. func SetsockoptLinger(fd Handle, level, opt int, l *Linger) (err error) {
  1092. sys := sysLinger{Onoff: uint16(l.Onoff), Linger: uint16(l.Linger)}
  1093. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(&sys)), int32(unsafe.Sizeof(sys)))
  1094. }
  1095. func SetsockoptInet4Addr(fd Handle, level, opt int, value [4]byte) (err error) {
  1096. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(&value[0])), 4)
  1097. }
  1098. func SetsockoptIPMreq(fd Handle, level, opt int, mreq *IPMreq) (err error) {
  1099. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(mreq)), int32(unsafe.Sizeof(*mreq)))
  1100. }
  1101. func SetsockoptIPv6Mreq(fd Handle, level, opt int, mreq *IPv6Mreq) (err error) {
  1102. return syscall.EWINDOWS
  1103. }
  1104. func Getpid() (pid int) { return int(GetCurrentProcessId()) }
  1105. func FindFirstFile(name *uint16, data *Win32finddata) (handle Handle, err error) {
  1106. // NOTE(rsc): The Win32finddata struct is wrong for the system call:
  1107. // the two paths are each one uint16 short. Use the correct struct,
  1108. // a win32finddata1, and then copy the results out.
  1109. // There is no loss of expressivity here, because the final
  1110. // uint16, if it is used, is supposed to be a NUL, and Go doesn't need that.
  1111. // For Go 1.1, we might avoid the allocation of win32finddata1 here
  1112. // by adding a final Bug [2]uint16 field to the struct and then
  1113. // adjusting the fields in the result directly.
  1114. var data1 win32finddata1
  1115. handle, err = findFirstFile1(name, &data1)
  1116. if err == nil {
  1117. copyFindData(data, &data1)
  1118. }
  1119. return
  1120. }
  1121. func FindNextFile(handle Handle, data *Win32finddata) (err error) {
  1122. var data1 win32finddata1
  1123. err = findNextFile1(handle, &data1)
  1124. if err == nil {
  1125. copyFindData(data, &data1)
  1126. }
  1127. return
  1128. }
  1129. func getProcessEntry(pid int) (*ProcessEntry32, error) {
  1130. snapshot, err := CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0)
  1131. if err != nil {
  1132. return nil, err
  1133. }
  1134. defer CloseHandle(snapshot)
  1135. var procEntry ProcessEntry32
  1136. procEntry.Size = uint32(unsafe.Sizeof(procEntry))
  1137. if err = Process32First(snapshot, &procEntry); err != nil {
  1138. return nil, err
  1139. }
  1140. for {
  1141. if procEntry.ProcessID == uint32(pid) {
  1142. return &procEntry, nil
  1143. }
  1144. err = Process32Next(snapshot, &procEntry)
  1145. if err != nil {
  1146. return nil, err
  1147. }
  1148. }
  1149. }
  1150. func Getppid() (ppid int) {
  1151. pe, err := getProcessEntry(Getpid())
  1152. if err != nil {
  1153. return -1
  1154. }
  1155. return int(pe.ParentProcessID)
  1156. }
  1157. // TODO(brainman): fix all needed for os
  1158. func Fchdir(fd Handle) (err error) { return syscall.EWINDOWS }
  1159. func Link(oldpath, newpath string) (err error) { return syscall.EWINDOWS }
  1160. func Symlink(path, link string) (err error) { return syscall.EWINDOWS }
  1161. func Fchmod(fd Handle, mode uint32) (err error) { return syscall.EWINDOWS }
  1162. func Chown(path string, uid int, gid int) (err error) { return syscall.EWINDOWS }
  1163. func Lchown(path string, uid int, gid int) (err error) { return syscall.EWINDOWS }
  1164. func Fchown(fd Handle, uid int, gid int) (err error) { return syscall.EWINDOWS }
  1165. func Getuid() (uid int) { return -1 }
  1166. func Geteuid() (euid int) { return -1 }
  1167. func Getgid() (gid int) { return -1 }
  1168. func Getegid() (egid int) { return -1 }
  1169. func Getgroups() (gids []int, err error) { return nil, syscall.EWINDOWS }
  1170. type Signal int
  1171. func (s Signal) Signal() {}
  1172. func (s Signal) String() string {
  1173. if 0 <= s && int(s) < len(signals) {
  1174. str := signals[s]
  1175. if str != "" {
  1176. return str
  1177. }
  1178. }
  1179. return "signal " + itoa(int(s))
  1180. }
  1181. func LoadCreateSymbolicLink() error {
  1182. return procCreateSymbolicLinkW.Find()
  1183. }
  1184. // Readlink returns the destination of the named symbolic link.
  1185. func Readlink(path string, buf []byte) (n int, err error) {
  1186. fd, err := CreateFile(StringToUTF16Ptr(path), GENERIC_READ, 0, nil, OPEN_EXISTING,
  1187. FILE_FLAG_OPEN_REPARSE_POINT|FILE_FLAG_BACKUP_SEMANTICS, 0)
  1188. if err != nil {
  1189. return -1, err
  1190. }
  1191. defer CloseHandle(fd)
  1192. rdbbuf := make([]byte, MAXIMUM_REPARSE_DATA_BUFFER_SIZE)
  1193. var bytesReturned uint32
  1194. err = DeviceIoControl(fd, FSCTL_GET_REPARSE_POINT, nil, 0, &rdbbuf[0], uint32(len(rdbbuf)), &bytesReturned, nil)
  1195. if err != nil {
  1196. return -1, err
  1197. }
  1198. rdb := (*reparseDataBuffer)(unsafe.Pointer(&rdbbuf[0]))
  1199. var s string
  1200. switch rdb.ReparseTag {
  1201. case IO_REPARSE_TAG_SYMLINK:
  1202. data := (*symbolicLinkReparseBuffer)(unsafe.Pointer(&rdb.reparseBuffer))
  1203. p := (*[0xffff]uint16)(unsafe.Pointer(&data.PathBuffer[0]))
  1204. s = UTF16ToString(p[data.PrintNameOffset/2 : (data.PrintNameLength-data.PrintNameOffset)/2])
  1205. case IO_REPARSE_TAG_MOUNT_POINT:
  1206. data := (*mountPointReparseBuffer)(unsafe.Pointer(&rdb.reparseBuffer))
  1207. p := (*[0xffff]uint16)(unsafe.Pointer(&data.PathBuffer[0]))
  1208. s = UTF16ToString(p[data.PrintNameOffset/2 : (data.PrintNameLength-data.PrintNameOffset)/2])
  1209. default:
  1210. // the path is not a symlink or junction but another type of reparse
  1211. // point
  1212. return -1, syscall.ENOENT
  1213. }
  1214. n = copy(buf, []byte(s))
  1215. return n, nil
  1216. }
  1217. // GUIDFromString parses a string in the form of
  1218. // "{XXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX}" into a GUID.
  1219. func GUIDFromString(str string) (GUID, error) {
  1220. guid := GUID{}
  1221. str16, err := syscall.UTF16PtrFromString(str)
  1222. if err != nil {
  1223. return guid, err
  1224. }
  1225. err = clsidFromString(str16, &guid)
  1226. if err != nil {
  1227. return guid, err
  1228. }
  1229. return guid, nil
  1230. }
  1231. // GenerateGUID creates a new random GUID.
  1232. func GenerateGUID() (GUID, error) {
  1233. guid := GUID{}
  1234. err := coCreateGuid(&guid)
  1235. if err != nil {
  1236. return guid, err
  1237. }
  1238. return guid, nil
  1239. }
  1240. // String returns the canonical string form of the GUID,
  1241. // in the form of "{XXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX}".
  1242. func (guid GUID) String() string {
  1243. var str [100]uint16
  1244. chars := stringFromGUID2(&guid, &str[0], int32(len(str)))
  1245. if chars <= 1 {
  1246. return ""
  1247. }
  1248. return string(utf16.Decode(str[:chars-1]))
  1249. }
  1250. // KnownFolderPath returns a well-known folder path for the current user, specified by one of
  1251. // the FOLDERID_ constants, and chosen and optionally created based on a KF_ flag.
  1252. func KnownFolderPath(folderID *KNOWNFOLDERID, flags uint32) (string, error) {
  1253. return Token(0).KnownFolderPath(folderID, flags)
  1254. }
  1255. // KnownFolderPath returns a well-known folder path for the user token, specified by one of
  1256. // the FOLDERID_ constants, and chosen and optionally created based on a KF_ flag.
  1257. func (t Token) KnownFolderPath(folderID *KNOWNFOLDERID, flags uint32) (string, error) {
  1258. var p *uint16
  1259. err := shGetKnownFolderPath(folderID, flags, t, &p)
  1260. if err != nil {
  1261. return "", err
  1262. }
  1263. defer CoTaskMemFree(unsafe.Pointer(p))
  1264. return UTF16PtrToString(p), nil
  1265. }
  1266. // RtlGetVersion returns the version of the underlying operating system, ignoring
  1267. // manifest semantics but is affected by the application compatibility layer.
  1268. func RtlGetVersion() *OsVersionInfoEx {
  1269. info := &OsVersionInfoEx{}
  1270. info.osVersionInfoSize = uint32(unsafe.Sizeof(*info))
  1271. // According to documentation, this function always succeeds.
  1272. // The function doesn't even check the validity of the
  1273. // osVersionInfoSize member. Disassembling ntdll.dll indicates
  1274. // that the documentation is indeed correct about that.
  1275. _ = rtlGetVersion(info)
  1276. return info
  1277. }
  1278. // RtlGetNtVersionNumbers returns the version of the underlying operating system,
  1279. // ignoring manifest semantics and the application compatibility layer.
  1280. func RtlGetNtVersionNumbers() (majorVersion, minorVersion, buildNumber uint32) {
  1281. rtlGetNtVersionNumbers(&majorVersion, &minorVersion, &buildNumber)
  1282. buildNumber &= 0xffff
  1283. return
  1284. }
  1285. // GetProcessPreferredUILanguages retrieves the process preferred UI languages.
  1286. func GetProcessPreferredUILanguages(flags uint32) ([]string, error) {
  1287. return getUILanguages(flags, getProcessPreferredUILanguages)
  1288. }
  1289. // GetThreadPreferredUILanguages retrieves the thread preferred UI languages for the current thread.
  1290. func GetThreadPreferredUILanguages(flags uint32) ([]string, error) {
  1291. return getUILanguages(flags, getThreadPreferredUILanguages)
  1292. }
  1293. // GetUserPreferredUILanguages retrieves information about the user preferred UI languages.
  1294. func GetUserPreferredUILanguages(flags uint32) ([]string, error) {
  1295. return getUILanguages(flags, getUserPreferredUILanguages)
  1296. }
  1297. // GetSystemPreferredUILanguages retrieves the system preferred UI languages.
  1298. func GetSystemPreferredUILanguages(flags uint32) ([]string, error) {
  1299. return getUILanguages(flags, getSystemPreferredUILanguages)
  1300. }
  1301. func getUILanguages(flags uint32, f func(flags uint32, numLanguages *uint32, buf *uint16, bufSize *uint32) error) ([]string, error) {
  1302. size := uint32(128)
  1303. for {
  1304. var numLanguages uint32
  1305. buf := make([]uint16, size)
  1306. err := f(flags, &numLanguages, &buf[0], &size)
  1307. if err == ERROR_INSUFFICIENT_BUFFER {
  1308. continue
  1309. }
  1310. if err != nil {
  1311. return nil, err
  1312. }
  1313. buf = buf[:size]
  1314. if numLanguages == 0 || len(buf) == 0 { // GetProcessPreferredUILanguages may return numLanguages==0 with "\0\0"
  1315. return []string{}, nil
  1316. }
  1317. if buf[len(buf)-1] == 0 {
  1318. buf = buf[:len(buf)-1] // remove terminating null
  1319. }
  1320. languages := make([]string, 0, numLanguages)
  1321. from := 0
  1322. for i, c := range buf {
  1323. if c == 0 {
  1324. languages = append(languages, string(utf16.Decode(buf[from:i])))
  1325. from = i + 1
  1326. }
  1327. }
  1328. return languages, nil
  1329. }
  1330. }