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