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