syscall_windows.go 44 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155
  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. "unicode/utf16"
  11. "unsafe"
  12. )
  13. type Handle uintptr
  14. const (
  15. InvalidHandle = ^Handle(0)
  16. // Flags for DefineDosDevice.
  17. DDD_EXACT_MATCH_ON_REMOVE = 0x00000004
  18. DDD_NO_BROADCAST_SYSTEM = 0x00000008
  19. DDD_RAW_TARGET_PATH = 0x00000001
  20. DDD_REMOVE_DEFINITION = 0x00000002
  21. // Return values for GetDriveType.
  22. DRIVE_UNKNOWN = 0
  23. DRIVE_NO_ROOT_DIR = 1
  24. DRIVE_REMOVABLE = 2
  25. DRIVE_FIXED = 3
  26. DRIVE_REMOTE = 4
  27. DRIVE_CDROM = 5
  28. DRIVE_RAMDISK = 6
  29. // File system flags from GetVolumeInformation and GetVolumeInformationByHandle.
  30. FILE_CASE_SENSITIVE_SEARCH = 0x00000001
  31. FILE_CASE_PRESERVED_NAMES = 0x00000002
  32. FILE_FILE_COMPRESSION = 0x00000010
  33. FILE_DAX_VOLUME = 0x20000000
  34. FILE_NAMED_STREAMS = 0x00040000
  35. FILE_PERSISTENT_ACLS = 0x00000008
  36. FILE_READ_ONLY_VOLUME = 0x00080000
  37. FILE_SEQUENTIAL_WRITE_ONCE = 0x00100000
  38. FILE_SUPPORTS_ENCRYPTION = 0x00020000
  39. FILE_SUPPORTS_EXTENDED_ATTRIBUTES = 0x00800000
  40. FILE_SUPPORTS_HARD_LINKS = 0x00400000
  41. FILE_SUPPORTS_OBJECT_IDS = 0x00010000
  42. FILE_SUPPORTS_OPEN_BY_FILE_ID = 0x01000000
  43. FILE_SUPPORTS_REPARSE_POINTS = 0x00000080
  44. FILE_SUPPORTS_SPARSE_FILES = 0x00000040
  45. FILE_SUPPORTS_TRANSACTIONS = 0x00200000
  46. FILE_SUPPORTS_USN_JOURNAL = 0x02000000
  47. FILE_UNICODE_ON_DISK = 0x00000004
  48. FILE_VOLUME_IS_COMPRESSED = 0x00008000
  49. FILE_VOLUME_QUOTAS = 0x00000020
  50. )
  51. // StringToUTF16 is deprecated. Use UTF16FromString instead.
  52. // If s contains a NUL byte this function panics instead of
  53. // returning an error.
  54. func StringToUTF16(s string) []uint16 {
  55. a, err := UTF16FromString(s)
  56. if err != nil {
  57. panic("windows: string with NUL passed to StringToUTF16")
  58. }
  59. return a
  60. }
  61. // UTF16FromString returns the UTF-16 encoding of the UTF-8 string
  62. // s, with a terminating NUL added. If s contains a NUL byte at any
  63. // location, it returns (nil, syscall.EINVAL).
  64. func UTF16FromString(s string) ([]uint16, error) {
  65. for i := 0; i < len(s); i++ {
  66. if s[i] == 0 {
  67. return nil, syscall.EINVAL
  68. }
  69. }
  70. return utf16.Encode([]rune(s + "\x00")), nil
  71. }
  72. // UTF16ToString returns the UTF-8 encoding of the UTF-16 sequence s,
  73. // with a terminating NUL removed.
  74. func UTF16ToString(s []uint16) string {
  75. for i, v := range s {
  76. if v == 0 {
  77. s = s[0:i]
  78. break
  79. }
  80. }
  81. return string(utf16.Decode(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. func Getpagesize() int { return 4096 }
  98. // NewCallback converts a Go function to a function pointer conforming to the stdcall calling convention.
  99. // This is useful when interoperating with Windows code requiring callbacks.
  100. // 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.
  101. func NewCallback(fn interface{}) uintptr {
  102. return syscall.NewCallback(fn)
  103. }
  104. // NewCallbackCDecl converts a Go function to a function pointer conforming to the cdecl calling convention.
  105. // This is useful when interoperating with Windows code requiring callbacks.
  106. // 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.
  107. func NewCallbackCDecl(fn interface{}) uintptr {
  108. return syscall.NewCallbackCDecl(fn)
  109. }
  110. // windows api calls
  111. //sys GetLastError() (lasterr error)
  112. //sys LoadLibrary(libname string) (handle Handle, err error) = LoadLibraryW
  113. //sys LoadLibraryEx(libname string, zero Handle, flags uintptr) (handle Handle, err error) = LoadLibraryExW
  114. //sys FreeLibrary(handle Handle) (err error)
  115. //sys GetProcAddress(module Handle, procname string) (proc uintptr, err error)
  116. //sys GetVersion() (ver uint32, err error)
  117. //sys FormatMessage(flags uint32, msgsrc uintptr, msgid uint32, langid uint32, buf []uint16, args *byte) (n uint32, err error) = FormatMessageW
  118. //sys ExitProcess(exitcode uint32)
  119. //sys CreateFile(name *uint16, access uint32, mode uint32, sa *SecurityAttributes, createmode uint32, attrs uint32, templatefile int32) (handle Handle, err error) [failretval==InvalidHandle] = CreateFileW
  120. //sys ReadFile(handle Handle, buf []byte, done *uint32, overlapped *Overlapped) (err error)
  121. //sys WriteFile(handle Handle, buf []byte, done *uint32, overlapped *Overlapped) (err error)
  122. //sys SetFilePointer(handle Handle, lowoffset int32, highoffsetptr *int32, whence uint32) (newlowoffset uint32, err error) [failretval==0xffffffff]
  123. //sys CloseHandle(handle Handle) (err error)
  124. //sys GetStdHandle(stdhandle uint32) (handle Handle, err error) [failretval==InvalidHandle]
  125. //sys SetStdHandle(stdhandle uint32, handle Handle) (err error)
  126. //sys findFirstFile1(name *uint16, data *win32finddata1) (handle Handle, err error) [failretval==InvalidHandle] = FindFirstFileW
  127. //sys findNextFile1(handle Handle, data *win32finddata1) (err error) = FindNextFileW
  128. //sys FindClose(handle Handle) (err error)
  129. //sys GetFileInformationByHandle(handle Handle, data *ByHandleFileInformation) (err error)
  130. //sys GetCurrentDirectory(buflen uint32, buf *uint16) (n uint32, err error) = GetCurrentDirectoryW
  131. //sys SetCurrentDirectory(path *uint16) (err error) = SetCurrentDirectoryW
  132. //sys CreateDirectory(path *uint16, sa *SecurityAttributes) (err error) = CreateDirectoryW
  133. //sys RemoveDirectory(path *uint16) (err error) = RemoveDirectoryW
  134. //sys DeleteFile(path *uint16) (err error) = DeleteFileW
  135. //sys MoveFile(from *uint16, to *uint16) (err error) = MoveFileW
  136. //sys MoveFileEx(from *uint16, to *uint16, flags uint32) (err error) = MoveFileExW
  137. //sys GetComputerName(buf *uint16, n *uint32) (err error) = GetComputerNameW
  138. //sys GetComputerNameEx(nametype uint32, buf *uint16, n *uint32) (err error) = GetComputerNameExW
  139. //sys SetEndOfFile(handle Handle) (err error)
  140. //sys GetSystemTimeAsFileTime(time *Filetime)
  141. //sys GetSystemTimePreciseAsFileTime(time *Filetime)
  142. //sys GetTimeZoneInformation(tzi *Timezoneinformation) (rc uint32, err error) [failretval==0xffffffff]
  143. //sys CreateIoCompletionPort(filehandle Handle, cphandle Handle, key uint32, threadcnt uint32) (handle Handle, err error)
  144. //sys GetQueuedCompletionStatus(cphandle Handle, qty *uint32, key *uint32, overlapped **Overlapped, timeout uint32) (err error)
  145. //sys PostQueuedCompletionStatus(cphandle Handle, qty uint32, key uint32, overlapped *Overlapped) (err error)
  146. //sys CancelIo(s Handle) (err error)
  147. //sys CancelIoEx(s Handle, o *Overlapped) (err error)
  148. //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
  149. //sys OpenProcess(da uint32, inheritHandle bool, pid uint32) (handle Handle, err error)
  150. //sys TerminateProcess(handle Handle, exitcode uint32) (err error)
  151. //sys GetExitCodeProcess(handle Handle, exitcode *uint32) (err error)
  152. //sys GetStartupInfo(startupInfo *StartupInfo) (err error) = GetStartupInfoW
  153. //sys GetCurrentProcess() (pseudoHandle Handle, err error)
  154. //sys GetProcessTimes(handle Handle, creationTime *Filetime, exitTime *Filetime, kernelTime *Filetime, userTime *Filetime) (err error)
  155. //sys DuplicateHandle(hSourceProcessHandle Handle, hSourceHandle Handle, hTargetProcessHandle Handle, lpTargetHandle *Handle, dwDesiredAccess uint32, bInheritHandle bool, dwOptions uint32) (err error)
  156. //sys WaitForSingleObject(handle Handle, waitMilliseconds uint32) (event uint32, err error) [failretval==0xffffffff]
  157. //sys GetTempPath(buflen uint32, buf *uint16) (n uint32, err error) = GetTempPathW
  158. //sys CreatePipe(readhandle *Handle, writehandle *Handle, sa *SecurityAttributes, size uint32) (err error)
  159. //sys GetFileType(filehandle Handle) (n uint32, err error)
  160. //sys CryptAcquireContext(provhandle *Handle, container *uint16, provider *uint16, provtype uint32, flags uint32) (err error) = advapi32.CryptAcquireContextW
  161. //sys CryptReleaseContext(provhandle Handle, flags uint32) (err error) = advapi32.CryptReleaseContext
  162. //sys CryptGenRandom(provhandle Handle, buflen uint32, buf *byte) (err error) = advapi32.CryptGenRandom
  163. //sys GetEnvironmentStrings() (envs *uint16, err error) [failretval==nil] = kernel32.GetEnvironmentStringsW
  164. //sys FreeEnvironmentStrings(envs *uint16) (err error) = kernel32.FreeEnvironmentStringsW
  165. //sys GetEnvironmentVariable(name *uint16, buffer *uint16, size uint32) (n uint32, err error) = kernel32.GetEnvironmentVariableW
  166. //sys SetEnvironmentVariable(name *uint16, value *uint16) (err error) = kernel32.SetEnvironmentVariableW
  167. //sys SetFileTime(handle Handle, ctime *Filetime, atime *Filetime, wtime *Filetime) (err error)
  168. //sys GetFileAttributes(name *uint16) (attrs uint32, err error) [failretval==INVALID_FILE_ATTRIBUTES] = kernel32.GetFileAttributesW
  169. //sys SetFileAttributes(name *uint16, attrs uint32) (err error) = kernel32.SetFileAttributesW
  170. //sys GetFileAttributesEx(name *uint16, level uint32, info *byte) (err error) = kernel32.GetFileAttributesExW
  171. //sys GetCommandLine() (cmd *uint16) = kernel32.GetCommandLineW
  172. //sys CommandLineToArgv(cmd *uint16, argc *int32) (argv *[8192]*[8192]uint16, err error) [failretval==nil] = shell32.CommandLineToArgvW
  173. //sys LocalFree(hmem Handle) (handle Handle, err error) [failretval!=0]
  174. //sys SetHandleInformation(handle Handle, mask uint32, flags uint32) (err error)
  175. //sys FlushFileBuffers(handle Handle) (err error)
  176. //sys GetFullPathName(path *uint16, buflen uint32, buf *uint16, fname **uint16) (n uint32, err error) = kernel32.GetFullPathNameW
  177. //sys GetLongPathName(path *uint16, buf *uint16, buflen uint32) (n uint32, err error) = kernel32.GetLongPathNameW
  178. //sys GetShortPathName(longpath *uint16, shortpath *uint16, buflen uint32) (n uint32, err error) = kernel32.GetShortPathNameW
  179. //sys CreateFileMapping(fhandle Handle, sa *SecurityAttributes, prot uint32, maxSizeHigh uint32, maxSizeLow uint32, name *uint16) (handle Handle, err error) = kernel32.CreateFileMappingW
  180. //sys MapViewOfFile(handle Handle, access uint32, offsetHigh uint32, offsetLow uint32, length uintptr) (addr uintptr, err error)
  181. //sys UnmapViewOfFile(addr uintptr) (err error)
  182. //sys FlushViewOfFile(addr uintptr, length uintptr) (err error)
  183. //sys VirtualLock(addr uintptr, length uintptr) (err error)
  184. //sys VirtualUnlock(addr uintptr, length uintptr) (err error)
  185. //sys VirtualAlloc(address uintptr, size uintptr, alloctype uint32, protect uint32) (value uintptr, err error) = kernel32.VirtualAlloc
  186. //sys VirtualFree(address uintptr, size uintptr, freetype uint32) (err error) = kernel32.VirtualFree
  187. //sys VirtualProtect(address uintptr, size uintptr, newprotect uint32, oldprotect *uint32) (err error) = kernel32.VirtualProtect
  188. //sys TransmitFile(s Handle, handle Handle, bytesToWrite uint32, bytsPerSend uint32, overlapped *Overlapped, transmitFileBuf *TransmitFileBuffers, flags uint32) (err error) = mswsock.TransmitFile
  189. //sys ReadDirectoryChanges(handle Handle, buf *byte, buflen uint32, watchSubTree bool, mask uint32, retlen *uint32, overlapped *Overlapped, completionRoutine uintptr) (err error) = kernel32.ReadDirectoryChangesW
  190. //sys CertOpenSystemStore(hprov Handle, name *uint16) (store Handle, err error) = crypt32.CertOpenSystemStoreW
  191. //sys CertOpenStore(storeProvider uintptr, msgAndCertEncodingType uint32, cryptProv uintptr, flags uint32, para uintptr) (handle Handle, err error) [failretval==InvalidHandle] = crypt32.CertOpenStore
  192. //sys CertEnumCertificatesInStore(store Handle, prevContext *CertContext) (context *CertContext, err error) [failretval==nil] = crypt32.CertEnumCertificatesInStore
  193. //sys CertAddCertificateContextToStore(store Handle, certContext *CertContext, addDisposition uint32, storeContext **CertContext) (err error) = crypt32.CertAddCertificateContextToStore
  194. //sys CertCloseStore(store Handle, flags uint32) (err error) = crypt32.CertCloseStore
  195. //sys CertGetCertificateChain(engine Handle, leaf *CertContext, time *Filetime, additionalStore Handle, para *CertChainPara, flags uint32, reserved uintptr, chainCtx **CertChainContext) (err error) = crypt32.CertGetCertificateChain
  196. //sys CertFreeCertificateChain(ctx *CertChainContext) = crypt32.CertFreeCertificateChain
  197. //sys CertCreateCertificateContext(certEncodingType uint32, certEncoded *byte, encodedLen uint32) (context *CertContext, err error) [failretval==nil] = crypt32.CertCreateCertificateContext
  198. //sys CertFreeCertificateContext(ctx *CertContext) (err error) = crypt32.CertFreeCertificateContext
  199. //sys CertVerifyCertificateChainPolicy(policyOID uintptr, chain *CertChainContext, para *CertChainPolicyPara, status *CertChainPolicyStatus) (err error) = crypt32.CertVerifyCertificateChainPolicy
  200. //sys RegOpenKeyEx(key Handle, subkey *uint16, options uint32, desiredAccess uint32, result *Handle) (regerrno error) = advapi32.RegOpenKeyExW
  201. //sys RegCloseKey(key Handle) (regerrno error) = advapi32.RegCloseKey
  202. //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
  203. //sys RegEnumKeyEx(key Handle, index uint32, name *uint16, nameLen *uint32, reserved *uint32, class *uint16, classLen *uint32, lastWriteTime *Filetime) (regerrno error) = advapi32.RegEnumKeyExW
  204. //sys RegQueryValueEx(key Handle, name *uint16, reserved *uint32, valtype *uint32, buf *byte, buflen *uint32) (regerrno error) = advapi32.RegQueryValueExW
  205. //sys getCurrentProcessId() (pid uint32) = kernel32.GetCurrentProcessId
  206. //sys GetConsoleMode(console Handle, mode *uint32) (err error) = kernel32.GetConsoleMode
  207. //sys SetConsoleMode(console Handle, mode uint32) (err error) = kernel32.SetConsoleMode
  208. //sys GetConsoleScreenBufferInfo(console Handle, info *ConsoleScreenBufferInfo) (err error) = kernel32.GetConsoleScreenBufferInfo
  209. //sys WriteConsole(console Handle, buf *uint16, towrite uint32, written *uint32, reserved *byte) (err error) = kernel32.WriteConsoleW
  210. //sys ReadConsole(console Handle, buf *uint16, toread uint32, read *uint32, inputControl *byte) (err error) = kernel32.ReadConsoleW
  211. //sys CreateToolhelp32Snapshot(flags uint32, processId uint32) (handle Handle, err error) [failretval==InvalidHandle] = kernel32.CreateToolhelp32Snapshot
  212. //sys Process32First(snapshot Handle, procEntry *ProcessEntry32) (err error) = kernel32.Process32FirstW
  213. //sys Process32Next(snapshot Handle, procEntry *ProcessEntry32) (err error) = kernel32.Process32NextW
  214. //sys DeviceIoControl(handle Handle, ioControlCode uint32, inBuffer *byte, inBufferSize uint32, outBuffer *byte, outBufferSize uint32, bytesReturned *uint32, overlapped *Overlapped) (err error)
  215. // This function returns 1 byte BOOLEAN rather than the 4 byte BOOL.
  216. //sys CreateSymbolicLink(symlinkfilename *uint16, targetfilename *uint16, flags uint32) (err error) [failretval&0xff==0] = CreateSymbolicLinkW
  217. //sys CreateHardLink(filename *uint16, existingfilename *uint16, reserved uintptr) (err error) [failretval&0xff==0] = CreateHardLinkW
  218. //sys GetCurrentThreadId() (id uint32)
  219. //sys CreateEvent(eventAttrs *SecurityAttributes, manualReset uint32, initialState uint32, name *uint16) (handle Handle, err error) = kernel32.CreateEventW
  220. //sys CreateEventEx(eventAttrs *SecurityAttributes, name *uint16, flags uint32, desiredAccess uint32) (handle Handle, err error) = kernel32.CreateEventExW
  221. //sys OpenEvent(desiredAccess uint32, inheritHandle bool, name *uint16) (handle Handle, err error) = kernel32.OpenEventW
  222. //sys SetEvent(event Handle) (err error) = kernel32.SetEvent
  223. //sys ResetEvent(event Handle) (err error) = kernel32.ResetEvent
  224. //sys PulseEvent(event Handle) (err error) = kernel32.PulseEvent
  225. // Volume Management Functions
  226. //sys DefineDosDevice(flags uint32, deviceName *uint16, targetPath *uint16) (err error) = DefineDosDeviceW
  227. //sys DeleteVolumeMountPoint(volumeMountPoint *uint16) (err error) = DeleteVolumeMountPointW
  228. //sys FindFirstVolume(volumeName *uint16, bufferLength uint32) (handle Handle, err error) [failretval==InvalidHandle] = FindFirstVolumeW
  229. //sys FindFirstVolumeMountPoint(rootPathName *uint16, volumeMountPoint *uint16, bufferLength uint32) (handle Handle, err error) [failretval==InvalidHandle] = FindFirstVolumeMountPointW
  230. //sys FindNextVolume(findVolume Handle, volumeName *uint16, bufferLength uint32) (err error) = FindNextVolumeW
  231. //sys FindNextVolumeMountPoint(findVolumeMountPoint Handle, volumeMountPoint *uint16, bufferLength uint32) (err error) = FindNextVolumeMountPointW
  232. //sys FindVolumeClose(findVolume Handle) (err error)
  233. //sys FindVolumeMountPointClose(findVolumeMountPoint Handle) (err error)
  234. //sys GetDriveType(rootPathName *uint16) (driveType uint32) = GetDriveTypeW
  235. //sys GetLogicalDrives() (drivesBitMask uint32, err error) [failretval==0]
  236. //sys GetLogicalDriveStrings(bufferLength uint32, buffer *uint16) (n uint32, err error) [failretval==0] = GetLogicalDriveStringsW
  237. //sys GetVolumeInformation(rootPathName *uint16, volumeNameBuffer *uint16, volumeNameSize uint32, volumeNameSerialNumber *uint32, maximumComponentLength *uint32, fileSystemFlags *uint32, fileSystemNameBuffer *uint16, fileSystemNameSize uint32) (err error) = GetVolumeInformationW
  238. //sys GetVolumeInformationByHandle(file Handle, volumeNameBuffer *uint16, volumeNameSize uint32, volumeNameSerialNumber *uint32, maximumComponentLength *uint32, fileSystemFlags *uint32, fileSystemNameBuffer *uint16, fileSystemNameSize uint32) (err error) = GetVolumeInformationByHandleW
  239. //sys GetVolumeNameForVolumeMountPoint(volumeMountPoint *uint16, volumeName *uint16, bufferlength uint32) (err error) = GetVolumeNameForVolumeMountPointW
  240. //sys GetVolumePathName(fileName *uint16, volumePathName *uint16, bufferLength uint32) (err error) = GetVolumePathNameW
  241. //sys GetVolumePathNamesForVolumeName(volumeName *uint16, volumePathNames *uint16, bufferLength uint32, returnLength *uint32) (err error) = GetVolumePathNamesForVolumeNameW
  242. //sys QueryDosDevice(deviceName *uint16, targetPath *uint16, max uint32) (n uint32, err error) [failretval==0] = QueryDosDeviceW
  243. //sys SetVolumeLabel(rootPathName *uint16, volumeName *uint16) (err error) = SetVolumeLabelW
  244. //sys SetVolumeMountPoint(volumeMountPoint *uint16, volumeName *uint16) (err error) = SetVolumeMountPointW
  245. // syscall interface implementation for other packages
  246. // GetProcAddressByOrdinal retrieves the address of the exported
  247. // function from module by ordinal.
  248. func GetProcAddressByOrdinal(module Handle, ordinal uintptr) (proc uintptr, err error) {
  249. r0, _, e1 := syscall.Syscall(procGetProcAddress.Addr(), 2, uintptr(module), ordinal, 0)
  250. proc = uintptr(r0)
  251. if proc == 0 {
  252. if e1 != 0 {
  253. err = errnoErr(e1)
  254. } else {
  255. err = syscall.EINVAL
  256. }
  257. }
  258. return
  259. }
  260. func Exit(code int) { ExitProcess(uint32(code)) }
  261. func makeInheritSa() *SecurityAttributes {
  262. var sa SecurityAttributes
  263. sa.Length = uint32(unsafe.Sizeof(sa))
  264. sa.InheritHandle = 1
  265. return &sa
  266. }
  267. func Open(path string, mode int, perm uint32) (fd Handle, err error) {
  268. if len(path) == 0 {
  269. return InvalidHandle, ERROR_FILE_NOT_FOUND
  270. }
  271. pathp, err := UTF16PtrFromString(path)
  272. if err != nil {
  273. return InvalidHandle, err
  274. }
  275. var access uint32
  276. switch mode & (O_RDONLY | O_WRONLY | O_RDWR) {
  277. case O_RDONLY:
  278. access = GENERIC_READ
  279. case O_WRONLY:
  280. access = GENERIC_WRITE
  281. case O_RDWR:
  282. access = GENERIC_READ | GENERIC_WRITE
  283. }
  284. if mode&O_CREAT != 0 {
  285. access |= GENERIC_WRITE
  286. }
  287. if mode&O_APPEND != 0 {
  288. access &^= GENERIC_WRITE
  289. access |= FILE_APPEND_DATA
  290. }
  291. sharemode := uint32(FILE_SHARE_READ | FILE_SHARE_WRITE)
  292. var sa *SecurityAttributes
  293. if mode&O_CLOEXEC == 0 {
  294. sa = makeInheritSa()
  295. }
  296. var createmode uint32
  297. switch {
  298. case mode&(O_CREAT|O_EXCL) == (O_CREAT | O_EXCL):
  299. createmode = CREATE_NEW
  300. case mode&(O_CREAT|O_TRUNC) == (O_CREAT | O_TRUNC):
  301. createmode = CREATE_ALWAYS
  302. case mode&O_CREAT == O_CREAT:
  303. createmode = OPEN_ALWAYS
  304. case mode&O_TRUNC == O_TRUNC:
  305. createmode = TRUNCATE_EXISTING
  306. default:
  307. createmode = OPEN_EXISTING
  308. }
  309. h, e := CreateFile(pathp, access, sharemode, sa, createmode, FILE_ATTRIBUTE_NORMAL, 0)
  310. return h, e
  311. }
  312. func Read(fd Handle, p []byte) (n int, err error) {
  313. var done uint32
  314. e := ReadFile(fd, p, &done, nil)
  315. if e != nil {
  316. if e == ERROR_BROKEN_PIPE {
  317. // NOTE(brainman): work around ERROR_BROKEN_PIPE is returned on reading EOF from stdin
  318. return 0, nil
  319. }
  320. return 0, e
  321. }
  322. if raceenabled {
  323. if done > 0 {
  324. raceWriteRange(unsafe.Pointer(&p[0]), int(done))
  325. }
  326. raceAcquire(unsafe.Pointer(&ioSync))
  327. }
  328. return int(done), nil
  329. }
  330. func Write(fd Handle, p []byte) (n int, err error) {
  331. if raceenabled {
  332. raceReleaseMerge(unsafe.Pointer(&ioSync))
  333. }
  334. var done uint32
  335. e := WriteFile(fd, p, &done, nil)
  336. if e != nil {
  337. return 0, e
  338. }
  339. if raceenabled && done > 0 {
  340. raceReadRange(unsafe.Pointer(&p[0]), int(done))
  341. }
  342. return int(done), nil
  343. }
  344. var ioSync int64
  345. func Seek(fd Handle, offset int64, whence int) (newoffset int64, err error) {
  346. var w uint32
  347. switch whence {
  348. case 0:
  349. w = FILE_BEGIN
  350. case 1:
  351. w = FILE_CURRENT
  352. case 2:
  353. w = FILE_END
  354. }
  355. hi := int32(offset >> 32)
  356. lo := int32(offset)
  357. // use GetFileType to check pipe, pipe can't do seek
  358. ft, _ := GetFileType(fd)
  359. if ft == FILE_TYPE_PIPE {
  360. return 0, syscall.EPIPE
  361. }
  362. rlo, e := SetFilePointer(fd, lo, &hi, w)
  363. if e != nil {
  364. return 0, e
  365. }
  366. return int64(hi)<<32 + int64(rlo), nil
  367. }
  368. func Close(fd Handle) (err error) {
  369. return CloseHandle(fd)
  370. }
  371. var (
  372. Stdin = getStdHandle(STD_INPUT_HANDLE)
  373. Stdout = getStdHandle(STD_OUTPUT_HANDLE)
  374. Stderr = getStdHandle(STD_ERROR_HANDLE)
  375. )
  376. func getStdHandle(stdhandle uint32) (fd Handle) {
  377. r, _ := GetStdHandle(stdhandle)
  378. CloseOnExec(r)
  379. return r
  380. }
  381. const ImplementsGetwd = true
  382. func Getwd() (wd string, err error) {
  383. b := make([]uint16, 300)
  384. n, e := GetCurrentDirectory(uint32(len(b)), &b[0])
  385. if e != nil {
  386. return "", e
  387. }
  388. return string(utf16.Decode(b[0:n])), nil
  389. }
  390. func Chdir(path string) (err error) {
  391. pathp, err := UTF16PtrFromString(path)
  392. if err != nil {
  393. return err
  394. }
  395. return SetCurrentDirectory(pathp)
  396. }
  397. func Mkdir(path string, mode uint32) (err error) {
  398. pathp, err := UTF16PtrFromString(path)
  399. if err != nil {
  400. return err
  401. }
  402. return CreateDirectory(pathp, nil)
  403. }
  404. func Rmdir(path string) (err error) {
  405. pathp, err := UTF16PtrFromString(path)
  406. if err != nil {
  407. return err
  408. }
  409. return RemoveDirectory(pathp)
  410. }
  411. func Unlink(path string) (err error) {
  412. pathp, err := UTF16PtrFromString(path)
  413. if err != nil {
  414. return err
  415. }
  416. return DeleteFile(pathp)
  417. }
  418. func Rename(oldpath, newpath string) (err error) {
  419. from, err := UTF16PtrFromString(oldpath)
  420. if err != nil {
  421. return err
  422. }
  423. to, err := UTF16PtrFromString(newpath)
  424. if err != nil {
  425. return err
  426. }
  427. return MoveFileEx(from, to, MOVEFILE_REPLACE_EXISTING)
  428. }
  429. func ComputerName() (name string, err error) {
  430. var n uint32 = MAX_COMPUTERNAME_LENGTH + 1
  431. b := make([]uint16, n)
  432. e := GetComputerName(&b[0], &n)
  433. if e != nil {
  434. return "", e
  435. }
  436. return string(utf16.Decode(b[0:n])), nil
  437. }
  438. func Ftruncate(fd Handle, length int64) (err error) {
  439. curoffset, e := Seek(fd, 0, 1)
  440. if e != nil {
  441. return e
  442. }
  443. defer Seek(fd, curoffset, 0)
  444. _, e = Seek(fd, length, 0)
  445. if e != nil {
  446. return e
  447. }
  448. e = SetEndOfFile(fd)
  449. if e != nil {
  450. return e
  451. }
  452. return nil
  453. }
  454. func Gettimeofday(tv *Timeval) (err error) {
  455. var ft Filetime
  456. GetSystemTimeAsFileTime(&ft)
  457. *tv = NsecToTimeval(ft.Nanoseconds())
  458. return nil
  459. }
  460. func Pipe(p []Handle) (err error) {
  461. if len(p) != 2 {
  462. return syscall.EINVAL
  463. }
  464. var r, w Handle
  465. e := CreatePipe(&r, &w, makeInheritSa(), 0)
  466. if e != nil {
  467. return e
  468. }
  469. p[0] = r
  470. p[1] = w
  471. return nil
  472. }
  473. func Utimes(path string, tv []Timeval) (err error) {
  474. if len(tv) != 2 {
  475. return syscall.EINVAL
  476. }
  477. pathp, e := UTF16PtrFromString(path)
  478. if e != nil {
  479. return e
  480. }
  481. h, e := CreateFile(pathp,
  482. FILE_WRITE_ATTRIBUTES, FILE_SHARE_WRITE, nil,
  483. OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, 0)
  484. if e != nil {
  485. return e
  486. }
  487. defer Close(h)
  488. a := NsecToFiletime(tv[0].Nanoseconds())
  489. w := NsecToFiletime(tv[1].Nanoseconds())
  490. return SetFileTime(h, nil, &a, &w)
  491. }
  492. func UtimesNano(path string, ts []Timespec) (err error) {
  493. if len(ts) != 2 {
  494. return syscall.EINVAL
  495. }
  496. pathp, e := UTF16PtrFromString(path)
  497. if e != nil {
  498. return e
  499. }
  500. h, e := CreateFile(pathp,
  501. FILE_WRITE_ATTRIBUTES, FILE_SHARE_WRITE, nil,
  502. OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, 0)
  503. if e != nil {
  504. return e
  505. }
  506. defer Close(h)
  507. a := NsecToFiletime(TimespecToNsec(ts[0]))
  508. w := NsecToFiletime(TimespecToNsec(ts[1]))
  509. return SetFileTime(h, nil, &a, &w)
  510. }
  511. func Fsync(fd Handle) (err error) {
  512. return FlushFileBuffers(fd)
  513. }
  514. func Chmod(path string, mode uint32) (err error) {
  515. if mode == 0 {
  516. return syscall.EINVAL
  517. }
  518. p, e := UTF16PtrFromString(path)
  519. if e != nil {
  520. return e
  521. }
  522. attrs, e := GetFileAttributes(p)
  523. if e != nil {
  524. return e
  525. }
  526. if mode&S_IWRITE != 0 {
  527. attrs &^= FILE_ATTRIBUTE_READONLY
  528. } else {
  529. attrs |= FILE_ATTRIBUTE_READONLY
  530. }
  531. return SetFileAttributes(p, attrs)
  532. }
  533. func LoadGetSystemTimePreciseAsFileTime() error {
  534. return procGetSystemTimePreciseAsFileTime.Find()
  535. }
  536. func LoadCancelIoEx() error {
  537. return procCancelIoEx.Find()
  538. }
  539. func LoadSetFileCompletionNotificationModes() error {
  540. return procSetFileCompletionNotificationModes.Find()
  541. }
  542. // net api calls
  543. const socket_error = uintptr(^uint32(0))
  544. //sys WSAStartup(verreq uint32, data *WSAData) (sockerr error) = ws2_32.WSAStartup
  545. //sys WSACleanup() (err error) [failretval==socket_error] = ws2_32.WSACleanup
  546. //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
  547. //sys socket(af int32, typ int32, protocol int32) (handle Handle, err error) [failretval==InvalidHandle] = ws2_32.socket
  548. //sys Setsockopt(s Handle, level int32, optname int32, optval *byte, optlen int32) (err error) [failretval==socket_error] = ws2_32.setsockopt
  549. //sys Getsockopt(s Handle, level int32, optname int32, optval *byte, optlen *int32) (err error) [failretval==socket_error] = ws2_32.getsockopt
  550. //sys bind(s Handle, name unsafe.Pointer, namelen int32) (err error) [failretval==socket_error] = ws2_32.bind
  551. //sys connect(s Handle, name unsafe.Pointer, namelen int32) (err error) [failretval==socket_error] = ws2_32.connect
  552. //sys getsockname(s Handle, rsa *RawSockaddrAny, addrlen *int32) (err error) [failretval==socket_error] = ws2_32.getsockname
  553. //sys getpeername(s Handle, rsa *RawSockaddrAny, addrlen *int32) (err error) [failretval==socket_error] = ws2_32.getpeername
  554. //sys listen(s Handle, backlog int32) (err error) [failretval==socket_error] = ws2_32.listen
  555. //sys shutdown(s Handle, how int32) (err error) [failretval==socket_error] = ws2_32.shutdown
  556. //sys Closesocket(s Handle) (err error) [failretval==socket_error] = ws2_32.closesocket
  557. //sys AcceptEx(ls Handle, as Handle, buf *byte, rxdatalen uint32, laddrlen uint32, raddrlen uint32, recvd *uint32, overlapped *Overlapped) (err error) = mswsock.AcceptEx
  558. //sys GetAcceptExSockaddrs(buf *byte, rxdatalen uint32, laddrlen uint32, raddrlen uint32, lrsa **RawSockaddrAny, lrsalen *int32, rrsa **RawSockaddrAny, rrsalen *int32) = mswsock.GetAcceptExSockaddrs
  559. //sys WSARecv(s Handle, bufs *WSABuf, bufcnt uint32, recvd *uint32, flags *uint32, overlapped *Overlapped, croutine *byte) (err error) [failretval==socket_error] = ws2_32.WSARecv
  560. //sys WSASend(s Handle, bufs *WSABuf, bufcnt uint32, sent *uint32, flags uint32, overlapped *Overlapped, croutine *byte) (err error) [failretval==socket_error] = ws2_32.WSASend
  561. //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
  562. //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
  563. //sys GetHostByName(name string) (h *Hostent, err error) [failretval==nil] = ws2_32.gethostbyname
  564. //sys GetServByName(name string, proto string) (s *Servent, err error) [failretval==nil] = ws2_32.getservbyname
  565. //sys Ntohs(netshort uint16) (u uint16) = ws2_32.ntohs
  566. //sys GetProtoByName(name string) (p *Protoent, err error) [failretval==nil] = ws2_32.getprotobyname
  567. //sys DnsQuery(name string, qtype uint16, options uint32, extra *byte, qrs **DNSRecord, pr *byte) (status error) = dnsapi.DnsQuery_W
  568. //sys DnsRecordListFree(rl *DNSRecord, freetype uint32) = dnsapi.DnsRecordListFree
  569. //sys DnsNameCompare(name1 *uint16, name2 *uint16) (same bool) = dnsapi.DnsNameCompare_W
  570. //sys GetAddrInfoW(nodename *uint16, servicename *uint16, hints *AddrinfoW, result **AddrinfoW) (sockerr error) = ws2_32.GetAddrInfoW
  571. //sys FreeAddrInfoW(addrinfo *AddrinfoW) = ws2_32.FreeAddrInfoW
  572. //sys GetIfEntry(pIfRow *MibIfRow) (errcode error) = iphlpapi.GetIfEntry
  573. //sys GetAdaptersInfo(ai *IpAdapterInfo, ol *uint32) (errcode error) = iphlpapi.GetAdaptersInfo
  574. //sys SetFileCompletionNotificationModes(handle Handle, flags uint8) (err error) = kernel32.SetFileCompletionNotificationModes
  575. //sys WSAEnumProtocols(protocols *int32, protocolBuffer *WSAProtocolInfo, bufferLength *uint32) (n int32, err error) [failretval==-1] = ws2_32.WSAEnumProtocolsW
  576. //sys GetAdaptersAddresses(family uint32, flags uint32, reserved uintptr, adapterAddresses *IpAdapterAddresses, sizePointer *uint32) (errcode error) = iphlpapi.GetAdaptersAddresses
  577. //sys GetACP() (acp uint32) = kernel32.GetACP
  578. //sys MultiByteToWideChar(codePage uint32, dwFlags uint32, str *byte, nstr int32, wchar *uint16, nwchar int32) (nwrite int32, err error) = kernel32.MultiByteToWideChar
  579. // For testing: clients can set this flag to force
  580. // creation of IPv6 sockets to return EAFNOSUPPORT.
  581. var SocketDisableIPv6 bool
  582. type RawSockaddrInet4 struct {
  583. Family uint16
  584. Port uint16
  585. Addr [4]byte /* in_addr */
  586. Zero [8]uint8
  587. }
  588. type RawSockaddrInet6 struct {
  589. Family uint16
  590. Port uint16
  591. Flowinfo uint32
  592. Addr [16]byte /* in6_addr */
  593. Scope_id uint32
  594. }
  595. type RawSockaddr struct {
  596. Family uint16
  597. Data [14]int8
  598. }
  599. type RawSockaddrAny struct {
  600. Addr RawSockaddr
  601. Pad [96]int8
  602. }
  603. type Sockaddr interface {
  604. sockaddr() (ptr unsafe.Pointer, len int32, err error) // lowercase; only we can define Sockaddrs
  605. }
  606. type SockaddrInet4 struct {
  607. Port int
  608. Addr [4]byte
  609. raw RawSockaddrInet4
  610. }
  611. func (sa *SockaddrInet4) sockaddr() (unsafe.Pointer, int32, error) {
  612. if sa.Port < 0 || sa.Port > 0xFFFF {
  613. return nil, 0, syscall.EINVAL
  614. }
  615. sa.raw.Family = AF_INET
  616. p := (*[2]byte)(unsafe.Pointer(&sa.raw.Port))
  617. p[0] = byte(sa.Port >> 8)
  618. p[1] = byte(sa.Port)
  619. for i := 0; i < len(sa.Addr); i++ {
  620. sa.raw.Addr[i] = sa.Addr[i]
  621. }
  622. return unsafe.Pointer(&sa.raw), int32(unsafe.Sizeof(sa.raw)), nil
  623. }
  624. type SockaddrInet6 struct {
  625. Port int
  626. ZoneId uint32
  627. Addr [16]byte
  628. raw RawSockaddrInet6
  629. }
  630. func (sa *SockaddrInet6) sockaddr() (unsafe.Pointer, int32, error) {
  631. if sa.Port < 0 || sa.Port > 0xFFFF {
  632. return nil, 0, syscall.EINVAL
  633. }
  634. sa.raw.Family = AF_INET6
  635. p := (*[2]byte)(unsafe.Pointer(&sa.raw.Port))
  636. p[0] = byte(sa.Port >> 8)
  637. p[1] = byte(sa.Port)
  638. sa.raw.Scope_id = sa.ZoneId
  639. for i := 0; i < len(sa.Addr); i++ {
  640. sa.raw.Addr[i] = sa.Addr[i]
  641. }
  642. return unsafe.Pointer(&sa.raw), int32(unsafe.Sizeof(sa.raw)), nil
  643. }
  644. type SockaddrUnix struct {
  645. Name string
  646. }
  647. func (sa *SockaddrUnix) sockaddr() (unsafe.Pointer, int32, error) {
  648. // TODO(brainman): implement SockaddrUnix.sockaddr()
  649. return nil, 0, syscall.EWINDOWS
  650. }
  651. func (rsa *RawSockaddrAny) Sockaddr() (Sockaddr, error) {
  652. switch rsa.Addr.Family {
  653. case AF_UNIX:
  654. return nil, syscall.EWINDOWS
  655. case AF_INET:
  656. pp := (*RawSockaddrInet4)(unsafe.Pointer(rsa))
  657. sa := new(SockaddrInet4)
  658. p := (*[2]byte)(unsafe.Pointer(&pp.Port))
  659. sa.Port = int(p[0])<<8 + int(p[1])
  660. for i := 0; i < len(sa.Addr); i++ {
  661. sa.Addr[i] = pp.Addr[i]
  662. }
  663. return sa, nil
  664. case AF_INET6:
  665. pp := (*RawSockaddrInet6)(unsafe.Pointer(rsa))
  666. sa := new(SockaddrInet6)
  667. p := (*[2]byte)(unsafe.Pointer(&pp.Port))
  668. sa.Port = int(p[0])<<8 + int(p[1])
  669. sa.ZoneId = pp.Scope_id
  670. for i := 0; i < len(sa.Addr); i++ {
  671. sa.Addr[i] = pp.Addr[i]
  672. }
  673. return sa, nil
  674. }
  675. return nil, syscall.EAFNOSUPPORT
  676. }
  677. func Socket(domain, typ, proto int) (fd Handle, err error) {
  678. if domain == AF_INET6 && SocketDisableIPv6 {
  679. return InvalidHandle, syscall.EAFNOSUPPORT
  680. }
  681. return socket(int32(domain), int32(typ), int32(proto))
  682. }
  683. func SetsockoptInt(fd Handle, level, opt int, value int) (err error) {
  684. v := int32(value)
  685. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(&v)), int32(unsafe.Sizeof(v)))
  686. }
  687. func Bind(fd Handle, sa Sockaddr) (err error) {
  688. ptr, n, err := sa.sockaddr()
  689. if err != nil {
  690. return err
  691. }
  692. return bind(fd, ptr, n)
  693. }
  694. func Connect(fd Handle, sa Sockaddr) (err error) {
  695. ptr, n, err := sa.sockaddr()
  696. if err != nil {
  697. return err
  698. }
  699. return connect(fd, ptr, n)
  700. }
  701. func Getsockname(fd Handle) (sa Sockaddr, err error) {
  702. var rsa RawSockaddrAny
  703. l := int32(unsafe.Sizeof(rsa))
  704. if err = getsockname(fd, &rsa, &l); err != nil {
  705. return
  706. }
  707. return rsa.Sockaddr()
  708. }
  709. func Getpeername(fd Handle) (sa Sockaddr, err error) {
  710. var rsa RawSockaddrAny
  711. l := int32(unsafe.Sizeof(rsa))
  712. if err = getpeername(fd, &rsa, &l); err != nil {
  713. return
  714. }
  715. return rsa.Sockaddr()
  716. }
  717. func Listen(s Handle, n int) (err error) {
  718. return listen(s, int32(n))
  719. }
  720. func Shutdown(fd Handle, how int) (err error) {
  721. return shutdown(fd, int32(how))
  722. }
  723. func WSASendto(s Handle, bufs *WSABuf, bufcnt uint32, sent *uint32, flags uint32, to Sockaddr, overlapped *Overlapped, croutine *byte) (err error) {
  724. rsa, l, err := to.sockaddr()
  725. if err != nil {
  726. return err
  727. }
  728. return WSASendTo(s, bufs, bufcnt, sent, flags, (*RawSockaddrAny)(unsafe.Pointer(rsa)), l, overlapped, croutine)
  729. }
  730. func LoadGetAddrInfo() error {
  731. return procGetAddrInfoW.Find()
  732. }
  733. var connectExFunc struct {
  734. once sync.Once
  735. addr uintptr
  736. err error
  737. }
  738. func LoadConnectEx() error {
  739. connectExFunc.once.Do(func() {
  740. var s Handle
  741. s, connectExFunc.err = Socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)
  742. if connectExFunc.err != nil {
  743. return
  744. }
  745. defer CloseHandle(s)
  746. var n uint32
  747. connectExFunc.err = WSAIoctl(s,
  748. SIO_GET_EXTENSION_FUNCTION_POINTER,
  749. (*byte)(unsafe.Pointer(&WSAID_CONNECTEX)),
  750. uint32(unsafe.Sizeof(WSAID_CONNECTEX)),
  751. (*byte)(unsafe.Pointer(&connectExFunc.addr)),
  752. uint32(unsafe.Sizeof(connectExFunc.addr)),
  753. &n, nil, 0)
  754. })
  755. return connectExFunc.err
  756. }
  757. func connectEx(s Handle, name unsafe.Pointer, namelen int32, sendBuf *byte, sendDataLen uint32, bytesSent *uint32, overlapped *Overlapped) (err error) {
  758. 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)
  759. if r1 == 0 {
  760. if e1 != 0 {
  761. err = error(e1)
  762. } else {
  763. err = syscall.EINVAL
  764. }
  765. }
  766. return
  767. }
  768. func ConnectEx(fd Handle, sa Sockaddr, sendBuf *byte, sendDataLen uint32, bytesSent *uint32, overlapped *Overlapped) error {
  769. err := LoadConnectEx()
  770. if err != nil {
  771. return errorspkg.New("failed to find ConnectEx: " + err.Error())
  772. }
  773. ptr, n, err := sa.sockaddr()
  774. if err != nil {
  775. return err
  776. }
  777. return connectEx(fd, ptr, n, sendBuf, sendDataLen, bytesSent, overlapped)
  778. }
  779. var sendRecvMsgFunc struct {
  780. once sync.Once
  781. sendAddr uintptr
  782. recvAddr uintptr
  783. err error
  784. }
  785. func loadWSASendRecvMsg() error {
  786. sendRecvMsgFunc.once.Do(func() {
  787. var s Handle
  788. s, sendRecvMsgFunc.err = Socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)
  789. if sendRecvMsgFunc.err != nil {
  790. return
  791. }
  792. defer CloseHandle(s)
  793. var n uint32
  794. sendRecvMsgFunc.err = WSAIoctl(s,
  795. SIO_GET_EXTENSION_FUNCTION_POINTER,
  796. (*byte)(unsafe.Pointer(&WSAID_WSARECVMSG)),
  797. uint32(unsafe.Sizeof(WSAID_WSARECVMSG)),
  798. (*byte)(unsafe.Pointer(&sendRecvMsgFunc.recvAddr)),
  799. uint32(unsafe.Sizeof(sendRecvMsgFunc.recvAddr)),
  800. &n, nil, 0)
  801. if sendRecvMsgFunc.err != nil {
  802. return
  803. }
  804. sendRecvMsgFunc.err = WSAIoctl(s,
  805. SIO_GET_EXTENSION_FUNCTION_POINTER,
  806. (*byte)(unsafe.Pointer(&WSAID_WSASENDMSG)),
  807. uint32(unsafe.Sizeof(WSAID_WSASENDMSG)),
  808. (*byte)(unsafe.Pointer(&sendRecvMsgFunc.sendAddr)),
  809. uint32(unsafe.Sizeof(sendRecvMsgFunc.sendAddr)),
  810. &n, nil, 0)
  811. })
  812. return sendRecvMsgFunc.err
  813. }
  814. func WSASendMsg(fd Handle, msg *WSAMsg, flags uint32, bytesSent *uint32, overlapped *Overlapped, croutine *byte) error {
  815. err := loadWSASendRecvMsg()
  816. if err != nil {
  817. return err
  818. }
  819. 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)))
  820. if r1 == socket_error {
  821. if e1 != 0 {
  822. err = errnoErr(e1)
  823. } else {
  824. err = syscall.EINVAL
  825. }
  826. }
  827. return err
  828. }
  829. func WSARecvMsg(fd Handle, msg *WSAMsg, bytesReceived *uint32, overlapped *Overlapped, croutine *byte) error {
  830. err := loadWSASendRecvMsg()
  831. if err != nil {
  832. return err
  833. }
  834. 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)
  835. if r1 == socket_error {
  836. if e1 != 0 {
  837. err = errnoErr(e1)
  838. } else {
  839. err = syscall.EINVAL
  840. }
  841. }
  842. return err
  843. }
  844. // Invented structures to support what package os expects.
  845. type Rusage struct {
  846. CreationTime Filetime
  847. ExitTime Filetime
  848. KernelTime Filetime
  849. UserTime Filetime
  850. }
  851. type WaitStatus struct {
  852. ExitCode uint32
  853. }
  854. func (w WaitStatus) Exited() bool { return true }
  855. func (w WaitStatus) ExitStatus() int { return int(w.ExitCode) }
  856. func (w WaitStatus) Signal() Signal { return -1 }
  857. func (w WaitStatus) CoreDump() bool { return false }
  858. func (w WaitStatus) Stopped() bool { return false }
  859. func (w WaitStatus) Continued() bool { return false }
  860. func (w WaitStatus) StopSignal() Signal { return -1 }
  861. func (w WaitStatus) Signaled() bool { return false }
  862. func (w WaitStatus) TrapCause() int { return -1 }
  863. // Timespec is an invented structure on Windows, but here for
  864. // consistency with the corresponding package for other operating systems.
  865. type Timespec struct {
  866. Sec int64
  867. Nsec int64
  868. }
  869. func TimespecToNsec(ts Timespec) int64 { return int64(ts.Sec)*1e9 + int64(ts.Nsec) }
  870. func NsecToTimespec(nsec int64) (ts Timespec) {
  871. ts.Sec = nsec / 1e9
  872. ts.Nsec = nsec % 1e9
  873. return
  874. }
  875. // TODO(brainman): fix all needed for net
  876. func Accept(fd Handle) (nfd Handle, sa Sockaddr, err error) { return 0, nil, syscall.EWINDOWS }
  877. func Recvfrom(fd Handle, p []byte, flags int) (n int, from Sockaddr, err error) {
  878. return 0, nil, syscall.EWINDOWS
  879. }
  880. func Sendto(fd Handle, p []byte, flags int, to Sockaddr) (err error) { return syscall.EWINDOWS }
  881. func SetsockoptTimeval(fd Handle, level, opt int, tv *Timeval) (err error) { return syscall.EWINDOWS }
  882. // The Linger struct is wrong but we only noticed after Go 1.
  883. // sysLinger is the real system call structure.
  884. // BUG(brainman): The definition of Linger is not appropriate for direct use
  885. // with Setsockopt and Getsockopt.
  886. // Use SetsockoptLinger instead.
  887. type Linger struct {
  888. Onoff int32
  889. Linger int32
  890. }
  891. type sysLinger struct {
  892. Onoff uint16
  893. Linger uint16
  894. }
  895. type IPMreq struct {
  896. Multiaddr [4]byte /* in_addr */
  897. Interface [4]byte /* in_addr */
  898. }
  899. type IPv6Mreq struct {
  900. Multiaddr [16]byte /* in6_addr */
  901. Interface uint32
  902. }
  903. func GetsockoptInt(fd Handle, level, opt int) (int, error) { return -1, syscall.EWINDOWS }
  904. func SetsockoptLinger(fd Handle, level, opt int, l *Linger) (err error) {
  905. sys := sysLinger{Onoff: uint16(l.Onoff), Linger: uint16(l.Linger)}
  906. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(&sys)), int32(unsafe.Sizeof(sys)))
  907. }
  908. func SetsockoptInet4Addr(fd Handle, level, opt int, value [4]byte) (err error) {
  909. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(&value[0])), 4)
  910. }
  911. func SetsockoptIPMreq(fd Handle, level, opt int, mreq *IPMreq) (err error) {
  912. return Setsockopt(fd, int32(level), int32(opt), (*byte)(unsafe.Pointer(mreq)), int32(unsafe.Sizeof(*mreq)))
  913. }
  914. func SetsockoptIPv6Mreq(fd Handle, level, opt int, mreq *IPv6Mreq) (err error) {
  915. return syscall.EWINDOWS
  916. }
  917. func Getpid() (pid int) { return int(getCurrentProcessId()) }
  918. func FindFirstFile(name *uint16, data *Win32finddata) (handle Handle, err error) {
  919. // NOTE(rsc): The Win32finddata struct is wrong for the system call:
  920. // the two paths are each one uint16 short. Use the correct struct,
  921. // a win32finddata1, and then copy the results out.
  922. // There is no loss of expressivity here, because the final
  923. // uint16, if it is used, is supposed to be a NUL, and Go doesn't need that.
  924. // For Go 1.1, we might avoid the allocation of win32finddata1 here
  925. // by adding a final Bug [2]uint16 field to the struct and then
  926. // adjusting the fields in the result directly.
  927. var data1 win32finddata1
  928. handle, err = findFirstFile1(name, &data1)
  929. if err == nil {
  930. copyFindData(data, &data1)
  931. }
  932. return
  933. }
  934. func FindNextFile(handle Handle, data *Win32finddata) (err error) {
  935. var data1 win32finddata1
  936. err = findNextFile1(handle, &data1)
  937. if err == nil {
  938. copyFindData(data, &data1)
  939. }
  940. return
  941. }
  942. func getProcessEntry(pid int) (*ProcessEntry32, error) {
  943. snapshot, err := CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0)
  944. if err != nil {
  945. return nil, err
  946. }
  947. defer CloseHandle(snapshot)
  948. var procEntry ProcessEntry32
  949. procEntry.Size = uint32(unsafe.Sizeof(procEntry))
  950. if err = Process32First(snapshot, &procEntry); err != nil {
  951. return nil, err
  952. }
  953. for {
  954. if procEntry.ProcessID == uint32(pid) {
  955. return &procEntry, nil
  956. }
  957. err = Process32Next(snapshot, &procEntry)
  958. if err != nil {
  959. return nil, err
  960. }
  961. }
  962. }
  963. func Getppid() (ppid int) {
  964. pe, err := getProcessEntry(Getpid())
  965. if err != nil {
  966. return -1
  967. }
  968. return int(pe.ParentProcessID)
  969. }
  970. // TODO(brainman): fix all needed for os
  971. func Fchdir(fd Handle) (err error) { return syscall.EWINDOWS }
  972. func Link(oldpath, newpath string) (err error) { return syscall.EWINDOWS }
  973. func Symlink(path, link string) (err error) { return syscall.EWINDOWS }
  974. func Fchmod(fd Handle, mode uint32) (err error) { return syscall.EWINDOWS }
  975. func Chown(path string, uid int, gid int) (err error) { return syscall.EWINDOWS }
  976. func Lchown(path string, uid int, gid int) (err error) { return syscall.EWINDOWS }
  977. func Fchown(fd Handle, uid int, gid int) (err error) { return syscall.EWINDOWS }
  978. func Getuid() (uid int) { return -1 }
  979. func Geteuid() (euid int) { return -1 }
  980. func Getgid() (gid int) { return -1 }
  981. func Getegid() (egid int) { return -1 }
  982. func Getgroups() (gids []int, err error) { return nil, syscall.EWINDOWS }
  983. type Signal int
  984. func (s Signal) Signal() {}
  985. func (s Signal) String() string {
  986. if 0 <= s && int(s) < len(signals) {
  987. str := signals[s]
  988. if str != "" {
  989. return str
  990. }
  991. }
  992. return "signal " + itoa(int(s))
  993. }
  994. func LoadCreateSymbolicLink() error {
  995. return procCreateSymbolicLinkW.Find()
  996. }
  997. // Readlink returns the destination of the named symbolic link.
  998. func Readlink(path string, buf []byte) (n int, err error) {
  999. fd, err := CreateFile(StringToUTF16Ptr(path), GENERIC_READ, 0, nil, OPEN_EXISTING,
  1000. FILE_FLAG_OPEN_REPARSE_POINT|FILE_FLAG_BACKUP_SEMANTICS, 0)
  1001. if err != nil {
  1002. return -1, err
  1003. }
  1004. defer CloseHandle(fd)
  1005. rdbbuf := make([]byte, MAXIMUM_REPARSE_DATA_BUFFER_SIZE)
  1006. var bytesReturned uint32
  1007. err = DeviceIoControl(fd, FSCTL_GET_REPARSE_POINT, nil, 0, &rdbbuf[0], uint32(len(rdbbuf)), &bytesReturned, nil)
  1008. if err != nil {
  1009. return -1, err
  1010. }
  1011. rdb := (*reparseDataBuffer)(unsafe.Pointer(&rdbbuf[0]))
  1012. var s string
  1013. switch rdb.ReparseTag {
  1014. case IO_REPARSE_TAG_SYMLINK:
  1015. data := (*symbolicLinkReparseBuffer)(unsafe.Pointer(&rdb.reparseBuffer))
  1016. p := (*[0xffff]uint16)(unsafe.Pointer(&data.PathBuffer[0]))
  1017. s = UTF16ToString(p[data.PrintNameOffset/2 : (data.PrintNameLength-data.PrintNameOffset)/2])
  1018. case IO_REPARSE_TAG_MOUNT_POINT:
  1019. data := (*mountPointReparseBuffer)(unsafe.Pointer(&rdb.reparseBuffer))
  1020. p := (*[0xffff]uint16)(unsafe.Pointer(&data.PathBuffer[0]))
  1021. s = UTF16ToString(p[data.PrintNameOffset/2 : (data.PrintNameLength-data.PrintNameOffset)/2])
  1022. default:
  1023. // the path is not a symlink or junction but another type of reparse
  1024. // point
  1025. return -1, syscall.ENOENT
  1026. }
  1027. n = copy(buf, []byte(s))
  1028. return n, nil
  1029. }