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 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] 175//sys waitForMultipleObjects(count uint32, handles uintptr, waitAll bool, waitMilliseconds uint32) (event uint32, err error) [failretval==0xffffffff] = WaitForMultipleObjects 176//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 593func 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 605// 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} 1219