syscall_windows.go 47 KB

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