2016-03-01 22:57:46 +00:00
// Copyright 2009 The Go Authors. All rights reserved.
2010-03-09 15:09:09 -08:00
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// Windows system calls.
package syscall
2010-03-16 23:10:07 -07:00
import (
2013-01-11 12:42:09 +11:00
errorspkg "errors"
2019-02-19 13:03:55 -08:00
"internal/oserror"
2015-09-23 10:03:54 +02:00
"internal/race"
2018-08-08 15:00:56 -07:00
"runtime"
2013-01-11 12:42:09 +11:00
"sync"
2011-11-08 15:41:54 -08:00
"unicode/utf16"
2010-03-16 23:10:07 -07:00
"unsafe"
)
2010-03-09 15:09:09 -08:00
2011-07-01 10:18:07 -04:00
type Handle uintptr
const InvalidHandle = ^ Handle ( 0 )
2015-05-18 15:50:00 -04:00
// StringToUTF16 returns the UTF-16 encoding of the UTF-8 string s,
// with a terminating NUL added. If s contains a NUL byte this
// function panics instead of returning an error.
//
// Deprecated: Use UTF16FromString instead.
syscall: return EINVAL when string arguments have NUL characters
Since NUL usually terminates strings in underlying syscalls, allowing
it when converting string arguments is a security risk, especially
when dealing with filenames. For example, a program might reason that
filename like "/root/..\x00/" is a subdirectory or "/root/" and allow
access to it, while underlying syscall will treat "\x00" as an end of
that string and the actual filename will be "/root/..", which might
be unexpected. Returning EINVAL when string arguments have NUL in
them makes sure this attack vector is unusable.
R=golang-dev, r, bradfitz, fullung, rsc, minux.ma
CC=golang-dev
https://golang.org/cl/6458050
2012-08-05 17:24:32 -04:00
func StringToUTF16 ( s string ) [ ] uint16 {
a , err := UTF16FromString ( s )
if err != nil {
panic ( "syscall: string with NUL passed to StringToUTF16" )
}
return a
}
// UTF16FromString returns the UTF-16 encoding of the UTF-8 string
// s, with a terminating NUL added. If s contains a NUL byte at any
// location, it returns (nil, EINVAL).
func UTF16FromString ( s string ) ( [ ] uint16 , error ) {
for i := 0 ; i < len ( s ) ; i ++ {
if s [ i ] == 0 {
return nil , EINVAL
}
}
return utf16 . Encode ( [ ] rune ( s + "\x00" ) ) , nil
}
2010-03-09 15:09:09 -08:00
2010-03-16 23:10:07 -07:00
// UTF16ToString returns the UTF-8 encoding of the UTF-16 sequence s,
// with a terminating NUL removed.
func UTF16ToString ( s [ ] uint16 ) string {
2010-04-13 16:30:11 -07:00
for i , v := range s {
if v == 0 {
s = s [ 0 : i ]
break
}
2010-03-09 15:09:09 -08:00
}
2010-03-16 23:10:07 -07:00
return string ( utf16 . Decode ( s ) )
2010-03-09 15:09:09 -08:00
}
2015-05-18 15:50:00 -04:00
// StringToUTF16Ptr returns pointer to the UTF-16 encoding of
// the UTF-8 string s, with a terminating NUL added. If s
2017-04-27 23:53:58 -05:00
// contains a NUL byte this function panics instead of
syscall: return EINVAL when string arguments have NUL characters
Since NUL usually terminates strings in underlying syscalls, allowing
it when converting string arguments is a security risk, especially
when dealing with filenames. For example, a program might reason that
filename like "/root/..\x00/" is a subdirectory or "/root/" and allow
access to it, while underlying syscall will treat "\x00" as an end of
that string and the actual filename will be "/root/..", which might
be unexpected. Returning EINVAL when string arguments have NUL in
them makes sure this attack vector is unusable.
R=golang-dev, r, bradfitz, fullung, rsc, minux.ma
CC=golang-dev
https://golang.org/cl/6458050
2012-08-05 17:24:32 -04:00
// returning an error.
2015-05-18 15:50:00 -04:00
//
// Deprecated: Use UTF16PtrFromString instead.
2010-03-16 23:10:07 -07:00
func StringToUTF16Ptr ( s string ) * uint16 { return & StringToUTF16 ( s ) [ 0 ] }
2010-03-09 15:09:09 -08:00
syscall: return EINVAL when string arguments have NUL characters
Since NUL usually terminates strings in underlying syscalls, allowing
it when converting string arguments is a security risk, especially
when dealing with filenames. For example, a program might reason that
filename like "/root/..\x00/" is a subdirectory or "/root/" and allow
access to it, while underlying syscall will treat "\x00" as an end of
that string and the actual filename will be "/root/..", which might
be unexpected. Returning EINVAL when string arguments have NUL in
them makes sure this attack vector is unusable.
R=golang-dev, r, bradfitz, fullung, rsc, minux.ma
CC=golang-dev
https://golang.org/cl/6458050
2012-08-05 17:24:32 -04:00
// UTF16PtrFromString returns pointer to the UTF-16 encoding of
// the UTF-8 string s, with a terminating NUL added. If s
// contains a NUL byte at any location, it returns (nil, EINVAL).
func UTF16PtrFromString ( s string ) ( * uint16 , error ) {
a , err := UTF16FromString ( s )
if err != nil {
return nil , err
}
return & a [ 0 ] , nil
}
2011-12-08 12:07:21 +11:00
// Errno is the Windows error number.
2019-08-22 12:40:52 -04:00
//
2019-08-30 19:27:55 +00:00
// Errno values can be tested against error values from the os package
2019-08-22 12:40:52 -04:00
// using errors.Is. For example:
//
// _, _, err := syscall.Syscall(...)
// if errors.Is(err, os.ErrNotExist) ...
2011-12-08 12:07:21 +11:00
type Errno uintptr
2012-02-20 09:51:25 +11:00
func langid ( pri , sub uint16 ) uint32 { return uint32 ( sub ) << 10 | uint32 ( pri ) }
2015-07-13 10:55:49 +10:00
// FormatMessage is deprecated (msgsrc should be uintptr, not uint32, but can
// not be changed due to the Go 1 compatibility guarantee).
//
// Deprecated: Use FormatMessage from golang.org/x/sys/windows instead.
func FormatMessage ( flags uint32 , msgsrc uint32 , msgid uint32 , langid uint32 , buf [ ] uint16 , args * byte ) ( n uint32 , err error ) {
return formatMessage ( flags , uintptr ( msgsrc ) , msgid , langid , buf , args )
}
2011-12-08 12:07:21 +11:00
func ( e Errno ) Error ( ) string {
// deal with special go errors
idx := int ( e - APPLICATION_ERROR )
if 0 <= idx && idx < len ( errors ) {
return errors [ idx ]
}
// ask windows for the remaining errors
var flags uint32 = FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ARGUMENT_ARRAY | FORMAT_MESSAGE_IGNORE_INSERTS
b := make ( [ ] uint16 , 300 )
2015-07-13 10:55:49 +10:00
n , err := formatMessage ( flags , 0 , uint32 ( e ) , langid ( LANG_ENGLISH , SUBLANG_ENGLISH_US ) , b , nil )
2011-12-08 12:07:21 +11:00
if err != nil {
2015-07-13 10:55:49 +10:00
n , err = formatMessage ( flags , 0 , uint32 ( e ) , 0 , b , nil )
2012-09-18 13:02:37 +10:00
if err != nil {
return "winapi error #" + itoa ( int ( e ) )
}
2011-12-08 12:07:21 +11:00
}
// trim terminating \r and \n
for ; n > 0 && ( b [ n - 1 ] == '\n' || b [ n - 1 ] == '\r' ) ; n -- {
}
return string ( utf16 . Decode ( b [ : n ] ) )
}
2019-02-19 13:03:55 -08:00
const _ERROR_BAD_NETPATH = Errno ( 53 )
func ( e Errno ) Is ( target error ) bool {
switch target {
case oserror . ErrPermission :
return e == ERROR_ACCESS_DENIED
case oserror . ErrExist :
return e == ERROR_ALREADY_EXISTS ||
e == ERROR_DIR_NOT_EMPTY ||
e == ERROR_FILE_EXISTS
case oserror . ErrNotExist :
return e == ERROR_FILE_NOT_FOUND ||
e == _ERROR_BAD_NETPATH ||
e == ERROR_PATH_NOT_FOUND
}
return false
}
2011-12-08 12:07:21 +11:00
func ( e Errno ) Temporary ( ) bool {
2018-04-30 20:23:37 -07:00
return e == EINTR || e == EMFILE || e . Timeout ( )
2011-12-08 12:07:21 +11:00
}
func ( e Errno ) Timeout ( ) bool {
return e == EAGAIN || e == EWOULDBLOCK || e == ETIMEDOUT
}
2015-06-14 19:48:45 -07:00
// Implemented in runtime/syscall_windows.go.
2014-08-25 15:59:13 +10:00
func compileCallback ( fn interface { } , cleanstack bool ) uintptr
2018-07-27 15:52:36 +00:00
// NewCallback converts a Go function to a function pointer conforming to the stdcall calling convention.
// This is useful when interoperating with Windows code requiring callbacks.
2018-10-06 06:10:25 +00:00
// The argument is expected to be a function with one uintptr-sized result. The function must not have arguments with size larger than the size of uintptr.
2014-08-25 15:59:13 +10:00
func NewCallback ( fn interface { } ) uintptr {
return compileCallback ( fn , true )
}
2018-07-27 15:52:36 +00:00
// NewCallbackCDecl converts a Go function to a function pointer conforming to the cdecl calling convention.
// This is useful when interoperating with Windows code requiring callbacks.
2018-10-06 06:10:25 +00:00
// The argument is expected to be a function with one uintptr-sized result. The function must not have arguments with size larger than the size of uintptr.
2014-08-25 15:59:13 +10:00
func NewCallbackCDecl ( fn interface { } ) uintptr {
return compileCallback ( fn , false )
}
2011-01-22 13:55:53 +11:00
2010-03-16 23:10:07 -07:00
// windows api calls
2011-12-08 12:07:21 +11:00
//sys GetLastError() (lasterr error)
2011-11-13 22:44:52 -05:00
//sys LoadLibrary(libname string) (handle Handle, err error) = LoadLibraryW
//sys FreeLibrary(handle Handle) (err error)
//sys GetProcAddress(module Handle, procname string) (proc uintptr, err error)
//sys GetVersion() (ver uint32, err error)
2015-07-13 10:55:49 +10:00
//sys formatMessage(flags uint32, msgsrc uintptr, msgid uint32, langid uint32, buf []uint16, args *byte) (n uint32, err error) = FormatMessageW
2010-04-02 01:11:17 -07:00
//sys ExitProcess(exitcode uint32)
2011-11-13 22:44:52 -05:00
//sys CreateFile(name *uint16, access uint32, mode uint32, sa *SecurityAttributes, createmode uint32, attrs uint32, templatefile int32) (handle Handle, err error) [failretval==InvalidHandle] = CreateFileW
//sys ReadFile(handle Handle, buf []byte, done *uint32, overlapped *Overlapped) (err error)
//sys WriteFile(handle Handle, buf []byte, done *uint32, overlapped *Overlapped) (err error)
//sys SetFilePointer(handle Handle, lowoffset int32, highoffsetptr *int32, whence uint32) (newlowoffset uint32, err error) [failretval==0xffffffff]
//sys CloseHandle(handle Handle) (err error)
//sys GetStdHandle(stdhandle int) (handle Handle, err error) [failretval==InvalidHandle]
2012-06-08 13:54:48 -04:00
//sys findFirstFile1(name *uint16, data *win32finddata1) (handle Handle, err error) [failretval==InvalidHandle] = FindFirstFileW
//sys findNextFile1(handle Handle, data *win32finddata1) (err error) = FindNextFileW
2011-11-13 22:44:52 -05:00
//sys FindClose(handle Handle) (err error)
//sys GetFileInformationByHandle(handle Handle, data *ByHandleFileInformation) (err error)
//sys GetCurrentDirectory(buflen uint32, buf *uint16) (n uint32, err error) = GetCurrentDirectoryW
//sys SetCurrentDirectory(path *uint16) (err error) = SetCurrentDirectoryW
//sys CreateDirectory(path *uint16, sa *SecurityAttributes) (err error) = CreateDirectoryW
//sys RemoveDirectory(path *uint16) (err error) = RemoveDirectoryW
//sys DeleteFile(path *uint16) (err error) = DeleteFileW
//sys MoveFile(from *uint16, to *uint16) (err error) = MoveFileW
//sys GetComputerName(buf *uint16, n *uint32) (err error) = GetComputerNameW
//sys SetEndOfFile(handle Handle) (err error)
2010-04-26 23:17:14 -07:00
//sys GetSystemTimeAsFileTime(time *Filetime)
2011-11-13 22:44:52 -05:00
//sys GetTimeZoneInformation(tzi *Timezoneinformation) (rc uint32, err error) [failretval==0xffffffff]
//sys CreateIoCompletionPort(filehandle Handle, cphandle Handle, key uint32, threadcnt uint32) (handle Handle, err error)
//sys GetQueuedCompletionStatus(cphandle Handle, qty *uint32, key *uint32, overlapped **Overlapped, timeout uint32) (err error)
//sys PostQueuedCompletionStatus(cphandle Handle, qty uint32, key uint32, overlapped *Overlapped) (err error)
//sys CancelIo(s Handle) (err error)
net: fix connection resets when closed on windows
It is common to close network connection while another goroutine is
blocked reading on another goroutine. This sequence corresponds to
windows calls to WSARecv to start io, followed by GetQueuedCompletionStatus
that blocks until io completes, and, finally, closesocket called from
another thread. We were expecting that closesocket would unblock
GetQueuedCompletionStatus, and it does, but not always
(http://code.google.com/p/go/issues/detail?id=4170#c5). Also that sequence
results in connection is being reset.
This CL inserts CancelIo between GetQueuedCompletionStatus and closesocket,
and waits for both WSARecv and GetQueuedCompletionStatus to complete before
proceeding to closesocket. This seems to fix both connection resets and
issue 4170. It also makes windows code behave similar to unix version.
Unfortunately, CancelIo needs to be called on the same thread as WSARecv.
So we have to employ strategy we use for connections with deadlines to
every connection now. It means, there are 2 unavoidable thread switches
for every io. Some newer versions of windows have new CancelIoEx api that
doesn't have these drawbacks, and this CL uses this capability when available.
As time goes by, we should have less of CancelIo and more of CancelIoEx
systems. Computers with CancelIoEx are also not affected by issue 4195 anymore.
Fixes #3710
Fixes #3746
Fixes #4170
Partial fix for issue 4195
R=golang-dev, mikioh.mikioh, bradfitz, rsc
CC=golang-dev
https://golang.org/cl/6604072
2012-10-31 10:24:37 +11:00
//sys CancelIoEx(s Handle, o *Overlapped) (err error)
2011-11-13 22:44:52 -05:00
//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
2017-11-07 12:09:59 +11:00
//sys CreateProcessAsUser(token Token, appName *uint16, commandLine *uint16, procSecurity *SecurityAttributes, threadSecurity *SecurityAttributes, inheritHandles bool, creationFlags uint32, env *uint16, currentDir *uint16, startupInfo *StartupInfo, outProcInfo *ProcessInformation) (err error) = advapi32.CreateProcessAsUserW
2011-11-13 22:44:52 -05:00
//sys OpenProcess(da uint32, inheritHandle bool, pid uint32) (handle Handle, err error)
//sys TerminateProcess(handle Handle, exitcode uint32) (err error)
//sys GetExitCodeProcess(handle Handle, exitcode *uint32) (err error)
//sys GetStartupInfo(startupInfo *StartupInfo) (err error) = GetStartupInfoW
//sys GetCurrentProcess() (pseudoHandle Handle, err error)
2012-03-02 14:47:40 +11:00
//sys GetProcessTimes(handle Handle, creationTime *Filetime, exitTime *Filetime, kernelTime *Filetime, userTime *Filetime) (err error)
2011-11-13 22:44:52 -05:00
//sys DuplicateHandle(hSourceProcessHandle Handle, hSourceHandle Handle, hTargetProcessHandle Handle, lpTargetHandle *Handle, dwDesiredAccess uint32, bInheritHandle bool, dwOptions uint32) (err error)
//sys WaitForSingleObject(handle Handle, waitMilliseconds uint32) (event uint32, err error) [failretval==0xffffffff]
//sys GetTempPath(buflen uint32, buf *uint16) (n uint32, err error) = GetTempPathW
//sys CreatePipe(readhandle *Handle, writehandle *Handle, sa *SecurityAttributes, size uint32) (err error)
//sys GetFileType(filehandle Handle) (n uint32, err error)
//sys CryptAcquireContext(provhandle *Handle, container *uint16, provider *uint16, provtype uint32, flags uint32) (err error) = advapi32.CryptAcquireContextW
//sys CryptReleaseContext(provhandle Handle, flags uint32) (err error) = advapi32.CryptReleaseContext
//sys CryptGenRandom(provhandle Handle, buflen uint32, buf *byte) (err error) = advapi32.CryptGenRandom
//sys GetEnvironmentStrings() (envs *uint16, err error) [failretval==nil] = kernel32.GetEnvironmentStringsW
//sys FreeEnvironmentStrings(envs *uint16) (err error) = kernel32.FreeEnvironmentStringsW
//sys GetEnvironmentVariable(name *uint16, buffer *uint16, size uint32) (n uint32, err error) = kernel32.GetEnvironmentVariableW
//sys SetEnvironmentVariable(name *uint16, value *uint16) (err error) = kernel32.SetEnvironmentVariableW
//sys SetFileTime(handle Handle, ctime *Filetime, atime *Filetime, wtime *Filetime) (err error)
//sys GetFileAttributes(name *uint16) (attrs uint32, err error) [failretval==INVALID_FILE_ATTRIBUTES] = kernel32.GetFileAttributesW
//sys SetFileAttributes(name *uint16, attrs uint32) (err error) = kernel32.SetFileAttributesW
//sys GetFileAttributesEx(name *uint16, level uint32, info *byte) (err error) = kernel32.GetFileAttributesExW
2010-12-16 12:18:18 +11:00
//sys GetCommandLine() (cmd *uint16) = kernel32.GetCommandLineW
2011-11-13 22:44:52 -05:00
//sys CommandLineToArgv(cmd *uint16, argc *int32) (argv *[8192]*[8192]uint16, err error) [failretval==nil] = shell32.CommandLineToArgvW
//sys LocalFree(hmem Handle) (handle Handle, err error) [failretval!=0]
//sys SetHandleInformation(handle Handle, mask uint32, flags uint32) (err error)
//sys FlushFileBuffers(handle Handle) (err error)
//sys GetFullPathName(path *uint16, buflen uint32, buf *uint16, fname **uint16) (n uint32, err error) = kernel32.GetFullPathNameW
2012-03-08 10:00:25 +11:00
//sys GetLongPathName(path *uint16, buf *uint16, buflen uint32) (n uint32, err error) = kernel32.GetLongPathNameW
2012-03-27 15:53:08 +11:00
//sys GetShortPathName(longpath *uint16, shortpath *uint16, buflen uint32) (n uint32, err error) = kernel32.GetShortPathNameW
2011-11-13 22:44:52 -05:00
//sys CreateFileMapping(fhandle Handle, sa *SecurityAttributes, prot uint32, maxSizeHigh uint32, maxSizeLow uint32, name *uint16) (handle Handle, err error) = kernel32.CreateFileMappingW
//sys MapViewOfFile(handle Handle, access uint32, offsetHigh uint32, offsetLow uint32, length uintptr) (addr uintptr, err error)
//sys UnmapViewOfFile(addr uintptr) (err error)
//sys FlushViewOfFile(addr uintptr, length uintptr) (err error)
//sys VirtualLock(addr uintptr, length uintptr) (err error)
//sys VirtualUnlock(addr uintptr, length uintptr) (err error)
//sys TransmitFile(s Handle, handle Handle, bytesToWrite uint32, bytsPerSend uint32, overlapped *Overlapped, transmitFileBuf *TransmitFileBuffers, flags uint32) (err error) = mswsock.TransmitFile
//sys ReadDirectoryChanges(handle Handle, buf *byte, buflen uint32, watchSubTree bool, mask uint32, retlen *uint32, overlapped *Overlapped, completionRoutine uintptr) (err error) = kernel32.ReadDirectoryChangesW
//sys CertOpenSystemStore(hprov Handle, name *uint16) (store Handle, err error) = crypt32.CertOpenSystemStoreW
2012-03-07 13:12:35 -05:00
//sys CertOpenStore(storeProvider uintptr, msgAndCertEncodingType uint32, cryptProv uintptr, flags uint32, para uintptr) (handle Handle, err error) [failretval==InvalidHandle] = crypt32.CertOpenStore
2011-12-01 12:38:00 -05:00
//sys CertEnumCertificatesInStore(store Handle, prevContext *CertContext) (context *CertContext, err error) [failretval==nil] = crypt32.CertEnumCertificatesInStore
2012-03-07 13:12:35 -05:00
//sys CertAddCertificateContextToStore(store Handle, certContext *CertContext, addDisposition uint32, storeContext **CertContext) (err error) = crypt32.CertAddCertificateContextToStore
2011-11-13 22:44:52 -05:00
//sys CertCloseStore(store Handle, flags uint32) (err error) = crypt32.CertCloseStore
2012-03-07 13:12:35 -05:00
//sys CertGetCertificateChain(engine Handle, leaf *CertContext, time *Filetime, additionalStore Handle, para *CertChainPara, flags uint32, reserved uintptr, chainCtx **CertChainContext) (err error) = crypt32.CertGetCertificateChain
//sys CertFreeCertificateChain(ctx *CertChainContext) = crypt32.CertFreeCertificateChain
//sys CertCreateCertificateContext(certEncodingType uint32, certEncoded *byte, encodedLen uint32) (context *CertContext, err error) [failretval==nil] = crypt32.CertCreateCertificateContext
//sys CertFreeCertificateContext(ctx *CertContext) (err error) = crypt32.CertFreeCertificateContext
//sys CertVerifyCertificateChainPolicy(policyOID uintptr, chain *CertChainContext, para *CertChainPolicyPara, status *CertChainPolicyStatus) (err error) = crypt32.CertVerifyCertificateChainPolicy
2011-12-08 12:07:21 +11:00
//sys RegOpenKeyEx(key Handle, subkey *uint16, options uint32, desiredAccess uint32, result *Handle) (regerrno error) = advapi32.RegOpenKeyExW
//sys RegCloseKey(key Handle) (regerrno error) = advapi32.RegCloseKey
//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
//sys RegEnumKeyEx(key Handle, index uint32, name *uint16, nameLen *uint32, reserved *uint32, class *uint16, classLen *uint32, lastWriteTime *Filetime) (regerrno error) = advapi32.RegEnumKeyExW
//sys RegQueryValueEx(key Handle, name *uint16, reserved *uint32, valtype *uint32, buf *byte, buflen *uint32) (regerrno error) = advapi32.RegQueryValueExW
2012-06-08 14:28:29 -04:00
//sys getCurrentProcessId() (pid uint32) = kernel32.GetCurrentProcessId
2012-09-12 12:04:45 +10:00
//sys GetConsoleMode(console Handle, mode *uint32) (err error) = kernel32.GetConsoleMode
//sys WriteConsole(console Handle, buf *uint16, towrite uint32, written *uint32, reserved *byte) (err error) = kernel32.WriteConsoleW
2013-02-26 14:18:48 +11:00
//sys ReadConsole(console Handle, buf *uint16, toread uint32, read *uint32, inputControl *byte) (err error) = kernel32.ReadConsoleW
2014-06-14 15:51:00 +10:00
//sys CreateToolhelp32Snapshot(flags uint32, processId uint32) (handle Handle, err error) [failretval==InvalidHandle] = kernel32.CreateToolhelp32Snapshot
//sys Process32First(snapshot Handle, procEntry *ProcessEntry32) (err error) = kernel32.Process32FirstW
//sys Process32Next(snapshot Handle, procEntry *ProcessEntry32) (err error) = kernel32.Process32NextW
2014-07-17 17:02:46 +10:00
//sys DeviceIoControl(handle Handle, ioControlCode uint32, inBuffer *byte, inBufferSize uint32, outBuffer *byte, outBufferSize uint32, bytesReturned *uint32, overlapped *Overlapped) (err error)
// This function returns 1 byte BOOLEAN rather than the 4 byte BOOL.
//sys CreateSymbolicLink(symlinkfilename *uint16, targetfilename *uint16, flags uint32) (err error) [failretval&0xff==0] = CreateSymbolicLinkW
//sys CreateHardLink(filename *uint16, existingfilename *uint16, reserved uintptr) (err error) [failretval&0xff==0] = CreateHardLinkW
2010-04-02 01:11:17 -07:00
// syscall interface implementation for other packages
2010-03-16 23:10:07 -07:00
2011-02-03 12:50:41 +11:00
func makeInheritSa ( ) * SecurityAttributes {
var sa SecurityAttributes
sa . Length = uint32 ( unsafe . Sizeof ( sa ) )
sa . InheritHandle = 1
return & sa
}
2011-11-13 22:44:52 -05:00
func Open ( path string , mode int , perm uint32 ) ( fd Handle , err error ) {
2010-04-02 01:11:17 -07:00
if len ( path ) == 0 {
2011-07-01 10:18:07 -04:00
return InvalidHandle , ERROR_FILE_NOT_FOUND
2010-04-02 01:11:17 -07:00
}
syscall: return EINVAL when string arguments have NUL characters
Since NUL usually terminates strings in underlying syscalls, allowing
it when converting string arguments is a security risk, especially
when dealing with filenames. For example, a program might reason that
filename like "/root/..\x00/" is a subdirectory or "/root/" and allow
access to it, while underlying syscall will treat "\x00" as an end of
that string and the actual filename will be "/root/..", which might
be unexpected. Returning EINVAL when string arguments have NUL in
them makes sure this attack vector is unusable.
R=golang-dev, r, bradfitz, fullung, rsc, minux.ma
CC=golang-dev
https://golang.org/cl/6458050
2012-08-05 17:24:32 -04:00
pathp , err := UTF16PtrFromString ( path )
if err != nil {
return InvalidHandle , err
}
2010-04-13 22:30:41 -07:00
var access uint32
switch mode & ( O_RDONLY | O_WRONLY | O_RDWR ) {
case O_RDONLY :
2010-04-02 01:11:17 -07:00
access = GENERIC_READ
2010-04-13 22:30:41 -07:00
case O_WRONLY :
2010-04-02 01:11:17 -07:00
access = GENERIC_WRITE
2010-04-13 22:30:41 -07:00
case O_RDWR :
2010-04-02 01:11:17 -07:00
access = GENERIC_READ | GENERIC_WRITE
}
2010-04-13 22:30:41 -07:00
if mode & O_CREAT != 0 {
access |= GENERIC_WRITE
}
2010-09-22 13:12:25 +10:00
if mode & O_APPEND != 0 {
access &^= GENERIC_WRITE
access |= FILE_APPEND_DATA
}
2010-04-13 22:30:41 -07:00
sharemode := uint32 ( FILE_SHARE_READ | FILE_SHARE_WRITE )
2011-02-03 12:50:41 +11:00
var sa * SecurityAttributes
if mode & O_CLOEXEC == 0 {
sa = makeInheritSa ( )
}
2010-04-02 01:11:17 -07:00
var createmode uint32
switch {
2011-05-27 17:02:24 +10:00
case mode & ( O_CREAT | O_EXCL ) == ( O_CREAT | O_EXCL ) :
createmode = CREATE_NEW
case mode & ( O_CREAT | O_TRUNC ) == ( O_CREAT | O_TRUNC ) :
createmode = CREATE_ALWAYS
case mode & O_CREAT == O_CREAT :
createmode = OPEN_ALWAYS
case mode & O_TRUNC == O_TRUNC :
2010-04-02 01:11:17 -07:00
createmode = TRUNCATE_EXISTING
default :
createmode = OPEN_EXISTING
}
2019-10-21 16:12:22 +02:00
var attrs uint32 = FILE_ATTRIBUTE_NORMAL
if perm & S_IWRITE == 0 {
attrs = FILE_ATTRIBUTE_READONLY
}
h , e := CreateFile ( pathp , access , sharemode , sa , createmode , attrs , 0 )
2011-11-13 22:44:52 -05:00
return h , e
2010-04-02 01:11:17 -07:00
}
2011-11-13 22:44:52 -05:00
func Read ( fd Handle , p [ ] byte ) ( n int , err error ) {
2010-04-02 01:11:17 -07:00
var done uint32
2011-07-01 10:18:07 -04:00
e := ReadFile ( fd , p , & done , nil )
2011-11-13 22:44:52 -05:00
if e != nil {
2010-04-30 12:46:46 -07:00
if e == ERROR_BROKEN_PIPE {
2010-08-25 01:26:35 -04:00
// NOTE(brainman): work around ERROR_BROKEN_PIPE is returned on reading EOF from stdin
2011-11-13 22:44:52 -05:00
return 0 , nil
2010-04-30 12:46:46 -07:00
}
2010-04-02 01:11:17 -07:00
return 0 , e
}
2015-09-23 10:03:54 +02:00
if race . Enabled {
2013-06-10 22:40:35 +04:00
if done > 0 {
2015-09-23 10:03:54 +02:00
race . WriteRange ( unsafe . Pointer ( & p [ 0 ] ) , int ( done ) )
2013-06-10 22:40:35 +04:00
}
2015-09-23 10:03:54 +02:00
race . Acquire ( unsafe . Pointer ( & ioSync ) )
2012-10-09 20:51:58 +04:00
}
2015-10-21 11:04:42 -07:00
if msanenabled && done > 0 {
msanWrite ( unsafe . Pointer ( & p [ 0 ] ) , int ( done ) )
}
2011-11-13 22:44:52 -05:00
return int ( done ) , nil
2010-04-02 01:11:17 -07:00
}
2011-11-13 22:44:52 -05:00
func Write ( fd Handle , p [ ] byte ) ( n int , err error ) {
2015-09-23 10:03:54 +02:00
if race . Enabled {
race . ReleaseMerge ( unsafe . Pointer ( & ioSync ) )
2012-10-09 20:51:58 +04:00
}
2010-04-02 01:11:17 -07:00
var done uint32
2011-07-01 10:18:07 -04:00
e := WriteFile ( fd , p , & done , nil )
2011-11-13 22:44:52 -05:00
if e != nil {
2010-04-02 01:11:17 -07:00
return 0 , e
}
2015-09-23 10:03:54 +02:00
if race . Enabled && done > 0 {
race . ReadRange ( unsafe . Pointer ( & p [ 0 ] ) , int ( done ) )
2013-06-10 22:40:35 +04:00
}
2015-10-21 11:04:42 -07:00
if msanenabled && done > 0 {
msanRead ( unsafe . Pointer ( & p [ 0 ] ) , int ( done ) )
}
2011-11-13 22:44:52 -05:00
return int ( done ) , nil
2010-04-02 01:11:17 -07:00
}
2012-10-09 20:51:58 +04:00
var ioSync int64
2017-12-06 18:20:26 +00:00
var procSetFilePointerEx = modkernel32 . NewProc ( "SetFilePointerEx" )
const ptrSize = unsafe . Sizeof ( uintptr ( 0 ) )
// setFilePointerEx calls SetFilePointerEx.
// See https://msdn.microsoft.com/en-us/library/windows/desktop/aa365542(v=vs.85).aspx
func setFilePointerEx ( handle Handle , distToMove int64 , newFilePointer * int64 , whence uint32 ) error {
var e1 Errno
2018-08-08 15:00:56 -07:00
switch runtime . GOARCH {
default :
panic ( "unsupported architecture" )
case "amd64" :
2017-12-06 18:20:26 +00:00
_ , _ , e1 = Syscall6 ( procSetFilePointerEx . Addr ( ) , 4 , uintptr ( handle ) , uintptr ( distToMove ) , uintptr ( unsafe . Pointer ( newFilePointer ) ) , uintptr ( whence ) , 0 , 0 )
2018-08-08 15:00:56 -07:00
case "386" :
2017-12-06 18:20:26 +00:00
// distToMove is a LARGE_INTEGER:
// https://msdn.microsoft.com/en-us/library/windows/desktop/aa383713(v=vs.85).aspx
_ , _ , e1 = Syscall6 ( procSetFilePointerEx . Addr ( ) , 5 , uintptr ( handle ) , uintptr ( distToMove ) , uintptr ( distToMove >> 32 ) , uintptr ( unsafe . Pointer ( newFilePointer ) ) , uintptr ( whence ) , 0 )
2018-08-08 15:00:56 -07:00
case "arm" :
// distToMove must be 8-byte aligned per ARM calling convention
// https://msdn.microsoft.com/en-us/library/dn736986.aspx#Anchor_7
_ , _ , e1 = Syscall6 ( procSetFilePointerEx . Addr ( ) , 6 , uintptr ( handle ) , 0 , uintptr ( distToMove ) , uintptr ( distToMove >> 32 ) , uintptr ( unsafe . Pointer ( newFilePointer ) ) , uintptr ( whence ) )
2017-12-06 18:20:26 +00:00
}
if e1 != 0 {
return errnoErr ( e1 )
}
return nil
}
2011-11-13 22:44:52 -05:00
func Seek ( fd Handle , offset int64 , whence int ) ( newoffset int64 , err error ) {
2010-04-02 01:11:17 -07:00
var w uint32
switch whence {
case 0 :
w = FILE_BEGIN
case 1 :
w = FILE_CURRENT
case 2 :
w = FILE_END
}
2010-07-26 15:55:01 +10:00
// use GetFileType to check pipe, pipe can't do seek
2011-07-01 10:18:07 -04:00
ft , _ := GetFileType ( fd )
2010-07-26 15:55:01 +10:00
if ft == FILE_TYPE_PIPE {
2017-04-21 17:10:58 +09:00
return 0 , ESPIPE
2010-07-26 15:55:01 +10:00
}
2017-12-06 18:20:26 +00:00
err = setFilePointerEx ( fd , offset , & newoffset , w )
return
2010-04-02 01:11:17 -07:00
}
2011-11-13 22:44:52 -05:00
func Close ( fd Handle ) ( err error ) {
2011-07-01 10:18:07 -04:00
return CloseHandle ( fd )
2010-04-02 01:11:17 -07:00
}
var (
Stdin = getStdHandle ( STD_INPUT_HANDLE )
Stdout = getStdHandle ( STD_OUTPUT_HANDLE )
Stderr = getStdHandle ( STD_ERROR_HANDLE )
)
2011-07-01 10:18:07 -04:00
func getStdHandle ( h int ) ( fd Handle ) {
2010-04-02 01:11:17 -07:00
r , _ := GetStdHandle ( h )
2011-09-23 18:16:13 +10:00
CloseOnExec ( r )
2011-07-01 10:18:07 -04:00
return r
2010-04-02 01:11:17 -07:00
}
2010-04-26 23:17:14 -07:00
const ImplementsGetwd = true
2011-11-13 22:44:52 -05:00
func Getwd ( ) ( wd string , err error ) {
2010-04-26 23:17:14 -07:00
b := make ( [ ] uint16 , 300 )
n , e := GetCurrentDirectory ( uint32 ( len ( b ) ) , & b [ 0 ] )
2011-11-13 22:44:52 -05:00
if e != nil {
2010-04-26 23:17:14 -07:00
return "" , e
}
2011-11-13 22:44:52 -05:00
return string ( utf16 . Decode ( b [ 0 : n ] ) ) , nil
2010-04-26 23:17:14 -07:00
}
2011-11-13 22:44:52 -05:00
func Chdir ( path string ) ( err error ) {
syscall: return EINVAL when string arguments have NUL characters
Since NUL usually terminates strings in underlying syscalls, allowing
it when converting string arguments is a security risk, especially
when dealing with filenames. For example, a program might reason that
filename like "/root/..\x00/" is a subdirectory or "/root/" and allow
access to it, while underlying syscall will treat "\x00" as an end of
that string and the actual filename will be "/root/..", which might
be unexpected. Returning EINVAL when string arguments have NUL in
them makes sure this attack vector is unusable.
R=golang-dev, r, bradfitz, fullung, rsc, minux.ma
CC=golang-dev
https://golang.org/cl/6458050
2012-08-05 17:24:32 -04:00
pathp , err := UTF16PtrFromString ( path )
if err != nil {
return err
}
return SetCurrentDirectory ( pathp )
2010-04-26 23:17:14 -07:00
}
2011-11-13 22:44:52 -05:00
func Mkdir ( path string , mode uint32 ) ( err error ) {
syscall: return EINVAL when string arguments have NUL characters
Since NUL usually terminates strings in underlying syscalls, allowing
it when converting string arguments is a security risk, especially
when dealing with filenames. For example, a program might reason that
filename like "/root/..\x00/" is a subdirectory or "/root/" and allow
access to it, while underlying syscall will treat "\x00" as an end of
that string and the actual filename will be "/root/..", which might
be unexpected. Returning EINVAL when string arguments have NUL in
them makes sure this attack vector is unusable.
R=golang-dev, r, bradfitz, fullung, rsc, minux.ma
CC=golang-dev
https://golang.org/cl/6458050
2012-08-05 17:24:32 -04:00
pathp , err := UTF16PtrFromString ( path )
if err != nil {
return err
}
return CreateDirectory ( pathp , nil )
2010-04-26 23:17:14 -07:00
}
2011-11-13 22:44:52 -05:00
func Rmdir ( path string ) ( err error ) {
syscall: return EINVAL when string arguments have NUL characters
Since NUL usually terminates strings in underlying syscalls, allowing
it when converting string arguments is a security risk, especially
when dealing with filenames. For example, a program might reason that
filename like "/root/..\x00/" is a subdirectory or "/root/" and allow
access to it, while underlying syscall will treat "\x00" as an end of
that string and the actual filename will be "/root/..", which might
be unexpected. Returning EINVAL when string arguments have NUL in
them makes sure this attack vector is unusable.
R=golang-dev, r, bradfitz, fullung, rsc, minux.ma
CC=golang-dev
https://golang.org/cl/6458050
2012-08-05 17:24:32 -04:00
pathp , err := UTF16PtrFromString ( path )
if err != nil {
return err
}
return RemoveDirectory ( pathp )
2010-04-26 23:17:14 -07:00
}
2011-11-13 22:44:52 -05:00
func Unlink ( path string ) ( err error ) {
syscall: return EINVAL when string arguments have NUL characters
Since NUL usually terminates strings in underlying syscalls, allowing
it when converting string arguments is a security risk, especially
when dealing with filenames. For example, a program might reason that
filename like "/root/..\x00/" is a subdirectory or "/root/" and allow
access to it, while underlying syscall will treat "\x00" as an end of
that string and the actual filename will be "/root/..", which might
be unexpected. Returning EINVAL when string arguments have NUL in
them makes sure this attack vector is unusable.
R=golang-dev, r, bradfitz, fullung, rsc, minux.ma
CC=golang-dev
https://golang.org/cl/6458050
2012-08-05 17:24:32 -04:00
pathp , err := UTF16PtrFromString ( path )
if err != nil {
return err
}
return DeleteFile ( pathp )
2010-04-26 23:17:14 -07:00
}
2011-11-13 22:44:52 -05:00
func Rename ( oldpath , newpath string ) ( err error ) {
syscall: return EINVAL when string arguments have NUL characters
Since NUL usually terminates strings in underlying syscalls, allowing
it when converting string arguments is a security risk, especially
when dealing with filenames. For example, a program might reason that
filename like "/root/..\x00/" is a subdirectory or "/root/" and allow
access to it, while underlying syscall will treat "\x00" as an end of
that string and the actual filename will be "/root/..", which might
be unexpected. Returning EINVAL when string arguments have NUL in
them makes sure this attack vector is unusable.
R=golang-dev, r, bradfitz, fullung, rsc, minux.ma
CC=golang-dev
https://golang.org/cl/6458050
2012-08-05 17:24:32 -04:00
from , err := UTF16PtrFromString ( oldpath )
if err != nil {
return err
}
to , err := UTF16PtrFromString ( newpath )
if err != nil {
return err
}
2011-02-11 10:15:51 +11:00
return MoveFile ( from , to )
2010-04-26 23:17:14 -07:00
}
2011-11-13 22:44:52 -05:00
func ComputerName ( ) ( name string , err error ) {
2010-04-26 23:17:14 -07:00
var n uint32 = MAX_COMPUTERNAME_LENGTH + 1
b := make ( [ ] uint16 , n )
2011-02-11 10:15:51 +11:00
e := GetComputerName ( & b [ 0 ] , & n )
2011-11-13 22:44:52 -05:00
if e != nil {
2010-04-26 23:17:14 -07:00
return "" , e
}
2011-11-13 22:44:52 -05:00
return string ( utf16 . Decode ( b [ 0 : n ] ) ) , nil
2010-04-26 23:17:14 -07:00
}
2011-11-13 22:44:52 -05:00
func Ftruncate ( fd Handle , length int64 ) ( err error ) {
2010-04-26 23:17:14 -07:00
curoffset , e := Seek ( fd , 0 , 1 )
2011-11-13 22:44:52 -05:00
if e != nil {
2010-04-26 23:17:14 -07:00
return e
}
defer Seek ( fd , curoffset , 0 )
2011-02-11 10:15:51 +11:00
_ , e = Seek ( fd , length , 0 )
2011-11-13 22:44:52 -05:00
if e != nil {
2010-04-26 23:17:14 -07:00
return e
}
2011-07-01 10:18:07 -04:00
e = SetEndOfFile ( fd )
2011-11-13 22:44:52 -05:00
if e != nil {
2010-04-26 23:17:14 -07:00
return e
}
2011-11-13 22:44:52 -05:00
return nil
2010-04-26 23:17:14 -07:00
}
2011-11-13 22:44:52 -05:00
func Gettimeofday ( tv * Timeval ) ( err error ) {
2010-04-26 23:17:14 -07:00
var ft Filetime
GetSystemTimeAsFileTime ( & ft )
2010-09-23 12:36:52 +10:00
* tv = NsecToTimeval ( ft . Nanoseconds ( ) )
2011-11-13 22:44:52 -05:00
return nil
2010-04-26 23:17:14 -07:00
}
2011-11-13 22:44:52 -05:00
func Pipe ( p [ ] Handle ) ( err error ) {
2010-07-26 15:55:01 +10:00
if len ( p ) != 2 {
return EINVAL
}
2011-07-01 10:18:07 -04:00
var r , w Handle
2011-02-11 10:15:51 +11:00
e := CreatePipe ( & r , & w , makeInheritSa ( ) , 0 )
2011-11-13 22:44:52 -05:00
if e != nil {
2011-02-11 10:15:51 +11:00
return e
2010-07-26 15:55:01 +10:00
}
2011-07-01 10:18:07 -04:00
p [ 0 ] = r
p [ 1 ] = w
2011-11-13 22:44:52 -05:00
return nil
2010-07-26 15:55:01 +10:00
}
2011-11-13 22:44:52 -05:00
func Utimes ( path string , tv [ ] Timeval ) ( err error ) {
2010-09-23 12:36:52 +10:00
if len ( tv ) != 2 {
return EINVAL
}
syscall: return EINVAL when string arguments have NUL characters
Since NUL usually terminates strings in underlying syscalls, allowing
it when converting string arguments is a security risk, especially
when dealing with filenames. For example, a program might reason that
filename like "/root/..\x00/" is a subdirectory or "/root/" and allow
access to it, while underlying syscall will treat "\x00" as an end of
that string and the actual filename will be "/root/..", which might
be unexpected. Returning EINVAL when string arguments have NUL in
them makes sure this attack vector is unusable.
R=golang-dev, r, bradfitz, fullung, rsc, minux.ma
CC=golang-dev
https://golang.org/cl/6458050
2012-08-05 17:24:32 -04:00
pathp , e := UTF16PtrFromString ( path )
if e != nil {
return e
}
h , e := CreateFile ( pathp ,
2010-09-23 12:36:52 +10:00
FILE_WRITE_ATTRIBUTES , FILE_SHARE_WRITE , nil ,
2014-10-06 19:22:48 -04:00
OPEN_EXISTING , FILE_FLAG_BACKUP_SEMANTICS , 0 )
2011-11-13 22:44:52 -05:00
if e != nil {
2010-09-23 12:36:52 +10:00
return e
}
2011-07-01 10:18:07 -04:00
defer Close ( h )
2010-09-23 12:36:52 +10:00
a := NsecToFiletime ( tv [ 0 ] . Nanoseconds ( ) )
w := NsecToFiletime ( tv [ 1 ] . Nanoseconds ( ) )
2011-02-11 10:15:51 +11:00
return SetFileTime ( h , nil , & a , & w )
2010-05-24 11:48:14 -07:00
}
2012-12-13 13:02:39 -08:00
func UtimesNano ( path string , ts [ ] Timespec ) ( err error ) {
if len ( ts ) != 2 {
return EINVAL
}
pathp , e := UTF16PtrFromString ( path )
if e != nil {
return e
}
h , e := CreateFile ( pathp ,
FILE_WRITE_ATTRIBUTES , FILE_SHARE_WRITE , nil ,
2014-10-06 19:22:48 -04:00
OPEN_EXISTING , FILE_FLAG_BACKUP_SEMANTICS , 0 )
2012-12-13 13:02:39 -08:00
if e != nil {
return e
}
defer Close ( h )
a := NsecToFiletime ( TimespecToNsec ( ts [ 0 ] ) )
w := NsecToFiletime ( TimespecToNsec ( ts [ 1 ] ) )
return SetFileTime ( h , nil , & a , & w )
}
2011-11-13 22:44:52 -05:00
func Fsync ( fd Handle ) ( err error ) {
2011-07-01 10:18:07 -04:00
return FlushFileBuffers ( fd )
2011-02-09 14:54:54 +11:00
}
2011-11-13 22:44:52 -05:00
func Chmod ( path string , mode uint32 ) ( err error ) {
syscall: return EINVAL when string arguments have NUL characters
Since NUL usually terminates strings in underlying syscalls, allowing
it when converting string arguments is a security risk, especially
when dealing with filenames. For example, a program might reason that
filename like "/root/..\x00/" is a subdirectory or "/root/" and allow
access to it, while underlying syscall will treat "\x00" as an end of
that string and the actual filename will be "/root/..", which might
be unexpected. Returning EINVAL when string arguments have NUL in
them makes sure this attack vector is unusable.
R=golang-dev, r, bradfitz, fullung, rsc, minux.ma
CC=golang-dev
https://golang.org/cl/6458050
2012-08-05 17:24:32 -04:00
p , e := UTF16PtrFromString ( path )
if e != nil {
return e
}
2011-02-17 12:09:12 +11:00
attrs , e := GetFileAttributes ( p )
2011-11-13 22:44:52 -05:00
if e != nil {
2011-02-17 12:09:12 +11:00
return e
}
if mode & S_IWRITE != 0 {
attrs &^= FILE_ATTRIBUTE_READONLY
} else {
attrs |= FILE_ATTRIBUTE_READONLY
}
return SetFileAttributes ( p , attrs )
}
net: fix connection resets when closed on windows
It is common to close network connection while another goroutine is
blocked reading on another goroutine. This sequence corresponds to
windows calls to WSARecv to start io, followed by GetQueuedCompletionStatus
that blocks until io completes, and, finally, closesocket called from
another thread. We were expecting that closesocket would unblock
GetQueuedCompletionStatus, and it does, but not always
(http://code.google.com/p/go/issues/detail?id=4170#c5). Also that sequence
results in connection is being reset.
This CL inserts CancelIo between GetQueuedCompletionStatus and closesocket,
and waits for both WSARecv and GetQueuedCompletionStatus to complete before
proceeding to closesocket. This seems to fix both connection resets and
issue 4170. It also makes windows code behave similar to unix version.
Unfortunately, CancelIo needs to be called on the same thread as WSARecv.
So we have to employ strategy we use for connections with deadlines to
every connection now. It means, there are 2 unavoidable thread switches
for every io. Some newer versions of windows have new CancelIoEx api that
doesn't have these drawbacks, and this CL uses this capability when available.
As time goes by, we should have less of CancelIo and more of CancelIoEx
systems. Computers with CancelIoEx are also not affected by issue 4195 anymore.
Fixes #3710
Fixes #3746
Fixes #4170
Partial fix for issue 4195
R=golang-dev, mikioh.mikioh, bradfitz, rsc
CC=golang-dev
https://golang.org/cl/6604072
2012-10-31 10:24:37 +11:00
func LoadCancelIoEx ( ) error {
return procCancelIoEx . Find ( )
}
2013-08-08 17:36:43 +04:00
func LoadSetFileCompletionNotificationModes ( ) error {
return procSetFileCompletionNotificationModes . Find ( )
}
2010-06-29 20:23:39 -07:00
// net api calls
2012-09-25 17:06:39 +10:00
const socket_error = uintptr ( ^ uint32 ( 0 ) )
2011-12-08 12:07:21 +11:00
//sys WSAStartup(verreq uint32, data *WSAData) (sockerr error) = ws2_32.WSAStartup
2012-09-25 17:06:39 +10:00
//sys WSACleanup() (err error) [failretval==socket_error] = ws2_32.WSACleanup
//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
2011-11-13 22:44:52 -05:00
//sys socket(af int32, typ int32, protocol int32) (handle Handle, err error) [failretval==InvalidHandle] = ws2_32.socket
2012-09-25 17:06:39 +10:00
//sys Setsockopt(s Handle, level int32, optname int32, optval *byte, optlen int32) (err error) [failretval==socket_error] = ws2_32.setsockopt
//sys Getsockopt(s Handle, level int32, optname int32, optval *byte, optlen *int32) (err error) [failretval==socket_error] = ws2_32.getsockopt
2014-04-06 12:18:01 +10:00
//sys bind(s Handle, name unsafe.Pointer, namelen int32) (err error) [failretval==socket_error] = ws2_32.bind
//sys connect(s Handle, name unsafe.Pointer, namelen int32) (err error) [failretval==socket_error] = ws2_32.connect
2012-09-25 17:06:39 +10:00
//sys getsockname(s Handle, rsa *RawSockaddrAny, addrlen *int32) (err error) [failretval==socket_error] = ws2_32.getsockname
//sys getpeername(s Handle, rsa *RawSockaddrAny, addrlen *int32) (err error) [failretval==socket_error] = ws2_32.getpeername
//sys listen(s Handle, backlog int32) (err error) [failretval==socket_error] = ws2_32.listen
//sys shutdown(s Handle, how int32) (err error) [failretval==socket_error] = ws2_32.shutdown
//sys Closesocket(s Handle) (err error) [failretval==socket_error] = ws2_32.closesocket
2011-11-13 22:44:52 -05:00
//sys AcceptEx(ls Handle, as Handle, buf *byte, rxdatalen uint32, laddrlen uint32, raddrlen uint32, recvd *uint32, overlapped *Overlapped) (err error) = mswsock.AcceptEx
2011-09-08 16:32:40 +10:00
//sys GetAcceptExSockaddrs(buf *byte, rxdatalen uint32, laddrlen uint32, raddrlen uint32, lrsa **RawSockaddrAny, lrsalen *int32, rrsa **RawSockaddrAny, rrsalen *int32) = mswsock.GetAcceptExSockaddrs
2012-09-25 17:06:39 +10:00
//sys WSARecv(s Handle, bufs *WSABuf, bufcnt uint32, recvd *uint32, flags *uint32, overlapped *Overlapped, croutine *byte) (err error) [failretval==socket_error] = ws2_32.WSARecv
//sys WSASend(s Handle, bufs *WSABuf, bufcnt uint32, sent *uint32, flags uint32, overlapped *Overlapped, croutine *byte) (err error) [failretval==socket_error] = ws2_32.WSASend
//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
//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
2011-11-13 22:44:52 -05:00
//sys GetHostByName(name string) (h *Hostent, err error) [failretval==nil] = ws2_32.gethostbyname
//sys GetServByName(name string, proto string) (s *Servent, err error) [failretval==nil] = ws2_32.getservbyname
2010-07-29 14:58:28 +10:00
//sys Ntohs(netshort uint16) (u uint16) = ws2_32.ntohs
2011-11-13 22:44:52 -05:00
//sys GetProtoByName(name string) (p *Protoent, err error) [failretval==nil] = ws2_32.getprotobyname
2011-12-08 12:07:21 +11:00
//sys DnsQuery(name string, qtype uint16, options uint32, extra *byte, qrs **DNSRecord, pr *byte) (status error) = dnsapi.DnsQuery_W
2010-07-29 14:58:28 +10:00
//sys DnsRecordListFree(rl *DNSRecord, freetype uint32) = dnsapi.DnsRecordListFree
2014-08-15 16:37:19 +10:00
//sys DnsNameCompare(name1 *uint16, name2 *uint16) (same bool) = dnsapi.DnsNameCompare_W
2012-11-07 16:58:20 +11:00
//sys GetAddrInfoW(nodename *uint16, servicename *uint16, hints *AddrinfoW, result **AddrinfoW) (sockerr error) = ws2_32.GetAddrInfoW
//sys FreeAddrInfoW(addrinfo *AddrinfoW) = ws2_32.FreeAddrInfoW
2011-12-08 12:07:21 +11:00
//sys GetIfEntry(pIfRow *MibIfRow) (errcode error) = iphlpapi.GetIfEntry
//sys GetAdaptersInfo(ai *IpAdapterInfo, ol *uint32) (errcode error) = iphlpapi.GetAdaptersInfo
2013-08-08 17:36:43 +04:00
//sys SetFileCompletionNotificationModes(handle Handle, flags uint8) (err error) = kernel32.SetFileCompletionNotificationModes
//sys WSAEnumProtocols(protocols *int32, protocolBuffer *WSAProtocolInfo, bufferLength *uint32) (n int32, err error) [failretval==-1] = ws2_32.WSAEnumProtocolsW
2010-07-29 14:58:28 +10:00
2010-09-12 12:02:29 +10:00
// For testing: clients can set this flag to force
// creation of IPv6 sockets to return EAFNOSUPPORT.
var SocketDisableIPv6 bool
2010-06-29 20:23:39 -07:00
type RawSockaddrInet4 struct {
Family uint16
Port uint16
Addr [ 4 ] byte /* in_addr */
Zero [ 8 ] uint8
}
2012-11-07 16:58:20 +11:00
type RawSockaddrInet6 struct {
Family uint16
Port uint16
Flowinfo uint32
Addr [ 16 ] byte /* in6_addr */
Scope_id uint32
}
2010-06-29 20:23:39 -07:00
type RawSockaddr struct {
Family uint16
Data [ 14 ] int8
}
type RawSockaddrAny struct {
Addr RawSockaddr
2018-07-23 01:47:11 +09:00
Pad [ 100 ] int8
2010-06-29 20:23:39 -07:00
}
type Sockaddr interface {
2014-04-06 12:18:01 +10:00
sockaddr ( ) ( ptr unsafe . Pointer , len int32 , err error ) // lowercase; only we can define Sockaddrs
2010-06-29 20:23:39 -07:00
}
type SockaddrInet4 struct {
Port int
Addr [ 4 ] byte
raw RawSockaddrInet4
}
2014-04-06 12:18:01 +10:00
func ( sa * SockaddrInet4 ) sockaddr ( ) ( unsafe . Pointer , int32 , error ) {
2010-06-29 20:23:39 -07:00
if sa . Port < 0 || sa . Port > 0xFFFF {
2014-04-06 12:18:01 +10:00
return nil , 0 , EINVAL
2010-06-29 20:23:39 -07:00
}
sa . raw . Family = AF_INET
p := ( * [ 2 ] byte ) ( unsafe . Pointer ( & sa . raw . Port ) )
p [ 0 ] = byte ( sa . Port >> 8 )
p [ 1 ] = byte ( sa . Port )
for i := 0 ; i < len ( sa . Addr ) ; i ++ {
sa . raw . Addr [ i ] = sa . Addr [ i ]
}
2014-04-06 12:18:01 +10:00
return unsafe . Pointer ( & sa . raw ) , int32 ( unsafe . Sizeof ( sa . raw ) ) , nil
2010-06-29 20:23:39 -07:00
}
type SockaddrInet6 struct {
2011-05-22 10:09:07 -07:00
Port int
ZoneId uint32
Addr [ 16 ] byte
2012-11-07 16:58:20 +11:00
raw RawSockaddrInet6
2010-06-29 20:23:39 -07:00
}
2014-04-06 12:18:01 +10:00
func ( sa * SockaddrInet6 ) sockaddr ( ) ( unsafe . Pointer , int32 , error ) {
2012-11-07 16:58:20 +11:00
if sa . Port < 0 || sa . Port > 0xFFFF {
2014-04-06 12:18:01 +10:00
return nil , 0 , EINVAL
2012-11-07 16:58:20 +11:00
}
sa . raw . Family = AF_INET6
p := ( * [ 2 ] byte ) ( unsafe . Pointer ( & sa . raw . Port ) )
p [ 0 ] = byte ( sa . Port >> 8 )
p [ 1 ] = byte ( sa . Port )
sa . raw . Scope_id = sa . ZoneId
for i := 0 ; i < len ( sa . Addr ) ; i ++ {
sa . raw . Addr [ i ] = sa . Addr [ i ]
}
2014-04-06 12:18:01 +10:00
return unsafe . Pointer ( & sa . raw ) , int32 ( unsafe . Sizeof ( sa . raw ) ) , nil
2010-06-29 20:23:39 -07:00
}
2018-07-23 01:47:11 +09:00
type RawSockaddrUnix struct {
Family uint16
Path [ UNIX_PATH_MAX ] int8
}
2010-06-29 20:23:39 -07:00
type SockaddrUnix struct {
Name string
2018-07-23 01:47:11 +09:00
raw RawSockaddrUnix
2010-06-29 20:23:39 -07:00
}
2014-04-06 12:18:01 +10:00
func ( sa * SockaddrUnix ) sockaddr ( ) ( unsafe . Pointer , int32 , error ) {
2018-07-23 01:47:11 +09:00
name := sa . Name
n := len ( name )
if n > len ( sa . raw . Path ) {
return nil , 0 , EINVAL
}
if n == len ( sa . raw . Path ) && name [ 0 ] != '@' {
return nil , 0 , EINVAL
}
sa . raw . Family = AF_UNIX
for i := 0 ; i < n ; i ++ {
sa . raw . Path [ i ] = int8 ( name [ i ] )
}
// length is family (uint16), name, NUL.
sl := int32 ( 2 )
if n > 0 {
sl += int32 ( n ) + 1
}
if sa . raw . Path [ 0 ] == '@' {
sa . raw . Path [ 0 ] = 0
// Don't count trailing NUL for abstract address.
sl --
}
return unsafe . Pointer ( & sa . raw ) , sl , nil
2010-06-29 20:23:39 -07:00
}
2011-11-13 22:44:52 -05:00
func ( rsa * RawSockaddrAny ) Sockaddr ( ) ( Sockaddr , error ) {
2010-06-29 20:23:39 -07:00
switch rsa . Addr . Family {
case AF_UNIX :
2018-07-23 01:47:11 +09:00
pp := ( * RawSockaddrUnix ) ( unsafe . Pointer ( rsa ) )
sa := new ( SockaddrUnix )
if pp . Path [ 0 ] == 0 {
// "Abstract" Unix domain socket.
// Rewrite leading NUL as @ for textual display.
// (This is the standard convention.)
// Not friendly to overwrite in place,
// but the callers below don't care.
pp . Path [ 0 ] = '@'
}
// Assume path ends at NUL.
// This is not technically the Linux semantics for
// abstract Unix domain sockets--they are supposed
// to be uninterpreted fixed-size binary blobs--but
// everyone uses this convention.
n := 0
for n < len ( pp . Path ) && pp . Path [ n ] != 0 {
n ++
}
2019-10-18 10:14:32 +07:00
bytes := ( * [ len ( pp . Path ) ] byte ) ( unsafe . Pointer ( & pp . Path [ 0 ] ) ) [ 0 : n ]
2018-07-23 01:47:11 +09:00
sa . Name = string ( bytes )
return sa , nil
2010-06-29 20:23:39 -07:00
case AF_INET :
pp := ( * RawSockaddrInet4 ) ( unsafe . Pointer ( rsa ) )
sa := new ( SockaddrInet4 )
p := ( * [ 2 ] byte ) ( unsafe . Pointer ( & pp . Port ) )
sa . Port = int ( p [ 0 ] ) << 8 + int ( p [ 1 ] )
for i := 0 ; i < len ( sa . Addr ) ; i ++ {
sa . Addr [ i ] = pp . Addr [ i ]
}
2011-11-13 22:44:52 -05:00
return sa , nil
2010-06-29 20:23:39 -07:00
case AF_INET6 :
2012-11-07 16:58:20 +11:00
pp := ( * RawSockaddrInet6 ) ( unsafe . Pointer ( rsa ) )
sa := new ( SockaddrInet6 )
p := ( * [ 2 ] byte ) ( unsafe . Pointer ( & pp . Port ) )
sa . Port = int ( p [ 0 ] ) << 8 + int ( p [ 1 ] )
sa . ZoneId = pp . Scope_id
for i := 0 ; i < len ( sa . Addr ) ; i ++ {
sa . Addr [ i ] = pp . Addr [ i ]
}
return sa , nil
2010-06-29 20:23:39 -07:00
}
return nil , EAFNOSUPPORT
}
2011-11-13 22:44:52 -05:00
func Socket ( domain , typ , proto int ) ( fd Handle , err error ) {
2010-09-12 12:02:29 +10:00
if domain == AF_INET6 && SocketDisableIPv6 {
2011-07-01 10:18:07 -04:00
return InvalidHandle , EAFNOSUPPORT
2010-09-12 12:02:29 +10:00
}
2011-11-13 22:44:52 -05:00
return socket ( int32 ( domain ) , int32 ( typ ) , int32 ( proto ) )
2010-06-29 20:23:39 -07:00
}
2011-11-13 22:44:52 -05:00
func SetsockoptInt ( fd Handle , level , opt int , value int ) ( err error ) {
2010-06-29 20:23:39 -07:00
v := int32 ( value )
2011-11-13 22:44:52 -05:00
return Setsockopt ( fd , int32 ( level ) , int32 ( opt ) , ( * byte ) ( unsafe . Pointer ( & v ) ) , int32 ( unsafe . Sizeof ( v ) ) )
2010-06-29 20:23:39 -07:00
}
2011-11-13 22:44:52 -05:00
func Bind ( fd Handle , sa Sockaddr ) ( err error ) {
2010-06-29 20:23:39 -07:00
ptr , n , err := sa . sockaddr ( )
2011-11-13 22:44:52 -05:00
if err != nil {
2010-06-29 20:23:39 -07:00
return err
}
2011-07-01 10:18:07 -04:00
return bind ( fd , ptr , n )
2010-06-29 20:23:39 -07:00
}
2011-11-13 22:44:52 -05:00
func Connect ( fd Handle , sa Sockaddr ) ( err error ) {
2010-06-29 20:23:39 -07:00
ptr , n , err := sa . sockaddr ( )
2011-11-13 22:44:52 -05:00
if err != nil {
2010-06-29 20:23:39 -07:00
return err
}
2011-07-01 10:18:07 -04:00
return connect ( fd , ptr , n )
2010-06-29 20:23:39 -07:00
}
2011-11-13 22:44:52 -05:00
func Getsockname ( fd Handle ) ( sa Sockaddr , err error ) {
2010-06-29 20:23:39 -07:00
var rsa RawSockaddrAny
l := int32 ( unsafe . Sizeof ( rsa ) )
2011-11-13 22:44:52 -05:00
if err = getsockname ( fd , & rsa , & l ) ; err != nil {
2010-06-29 20:23:39 -07:00
return
}
return rsa . Sockaddr ( )
}
2011-11-13 22:44:52 -05:00
func Getpeername ( fd Handle ) ( sa Sockaddr , err error ) {
2010-06-29 20:23:39 -07:00
var rsa RawSockaddrAny
l := int32 ( unsafe . Sizeof ( rsa ) )
2011-11-13 22:44:52 -05:00
if err = getpeername ( fd , & rsa , & l ) ; err != nil {
2010-06-29 20:23:39 -07:00
return
}
return rsa . Sockaddr ( )
}
2011-11-13 22:44:52 -05:00
func Listen ( s Handle , n int ) ( err error ) {
return listen ( s , int32 ( n ) )
2010-06-29 20:23:39 -07:00
}
2011-11-13 22:44:52 -05:00
func Shutdown ( fd Handle , how int ) ( err error ) {
return shutdown ( fd , int32 ( how ) )
2010-06-29 20:23:39 -07:00
}
2011-11-13 22:44:52 -05:00
func WSASendto ( s Handle , bufs * WSABuf , bufcnt uint32 , sent * uint32 , flags uint32 , to Sockaddr , overlapped * Overlapped , croutine * byte ) ( err error ) {
2010-11-22 11:01:30 -05:00
rsa , l , err := to . sockaddr ( )
2011-11-13 22:44:52 -05:00
if err != nil {
2010-11-22 11:01:30 -05:00
return err
}
2011-11-13 22:44:52 -05:00
return WSASendTo ( s , bufs , bufcnt , sent , flags , ( * RawSockaddrAny ) ( unsafe . Pointer ( rsa ) ) , l , overlapped , croutine )
2010-11-22 11:01:30 -05:00
}
2012-11-07 16:58:20 +11:00
func LoadGetAddrInfo ( ) error {
return procGetAddrInfoW . Find ( )
}
2013-01-11 12:42:09 +11:00
var connectExFunc struct {
once sync . Once
addr uintptr
err error
}
func LoadConnectEx ( ) error {
connectExFunc . once . Do ( func ( ) {
var s Handle
s , connectExFunc . err = Socket ( AF_INET , SOCK_STREAM , IPPROTO_TCP )
if connectExFunc . err != nil {
return
}
defer CloseHandle ( s )
var n uint32
connectExFunc . err = WSAIoctl ( s ,
SIO_GET_EXTENSION_FUNCTION_POINTER ,
( * byte ) ( unsafe . Pointer ( & WSAID_CONNECTEX ) ) ,
uint32 ( unsafe . Sizeof ( WSAID_CONNECTEX ) ) ,
( * byte ) ( unsafe . Pointer ( & connectExFunc . addr ) ) ,
uint32 ( unsafe . Sizeof ( connectExFunc . addr ) ) ,
& n , nil , 0 )
} )
return connectExFunc . err
}
2014-04-06 12:18:01 +10:00
func connectEx ( s Handle , name unsafe . Pointer , namelen int32 , sendBuf * byte , sendDataLen uint32 , bytesSent * uint32 , overlapped * Overlapped ) ( err error ) {
2013-01-11 12:42:09 +11:00
r1 , _ , e1 := 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 )
if r1 == 0 {
if e1 != 0 {
err = error ( e1 )
} else {
err = EINVAL
}
}
return
}
func ConnectEx ( fd Handle , sa Sockaddr , sendBuf * byte , sendDataLen uint32 , bytesSent * uint32 , overlapped * Overlapped ) error {
err := LoadConnectEx ( )
if err != nil {
return errorspkg . New ( "failed to find ConnectEx: " + err . Error ( ) )
}
ptr , n , err := sa . sockaddr ( )
if err != nil {
return err
}
return connectEx ( fd , ptr , n , sendBuf , sendDataLen , bytesSent , overlapped )
}
2011-02-04 14:41:26 +11:00
// Invented structures to support what package os expects.
2012-03-02 14:47:40 +11:00
type Rusage struct {
CreationTime Filetime
ExitTime Filetime
KernelTime Filetime
UserTime Filetime
}
2011-02-04 14:41:26 +11:00
type WaitStatus struct {
ExitCode uint32
}
func ( w WaitStatus ) Exited ( ) bool { return true }
func ( w WaitStatus ) ExitStatus ( ) int { return int ( w . ExitCode ) }
2012-02-14 13:51:38 +11:00
func ( w WaitStatus ) Signal ( ) Signal { return - 1 }
2011-02-04 14:41:26 +11:00
func ( w WaitStatus ) CoreDump ( ) bool { return false }
func ( w WaitStatus ) Stopped ( ) bool { return false }
func ( w WaitStatus ) Continued ( ) bool { return false }
2012-02-14 13:51:38 +11:00
func ( w WaitStatus ) StopSignal ( ) Signal { return - 1 }
2011-02-04 14:41:26 +11:00
2011-04-08 10:27:47 +10:00
func ( w WaitStatus ) Signaled ( ) bool { return false }
2011-02-04 14:41:26 +11:00
func ( w WaitStatus ) TrapCause ( ) int { return - 1 }
2012-01-18 19:05:44 -08:00
// Timespec is an invented structure on Windows, but here for
// consistency with the syscall package for other operating systems.
type Timespec struct {
Sec int64
Nsec int64
}
2012-12-13 13:02:39 -08:00
func TimespecToNsec ( ts Timespec ) int64 { return int64 ( ts . Sec ) * 1e9 + int64 ( ts . Nsec ) }
func NsecToTimespec ( nsec int64 ) ( ts Timespec ) {
ts . Sec = nsec / 1e9
ts . Nsec = nsec % 1e9
return
}
2010-06-29 20:23:39 -07:00
// TODO(brainman): fix all needed for net
2011-11-13 22:44:52 -05:00
func Accept ( fd Handle ) ( nfd Handle , sa Sockaddr , err error ) { return 0 , nil , EWINDOWS }
func Recvfrom ( fd Handle , p [ ] byte , flags int ) ( n int , from Sockaddr , err error ) {
2011-07-01 10:18:07 -04:00
return 0 , nil , EWINDOWS
}
2011-11-13 22:44:52 -05:00
func Sendto ( fd Handle , p [ ] byte , flags int , to Sockaddr ) ( err error ) { return EWINDOWS }
func SetsockoptTimeval ( fd Handle , level , opt int , tv * Timeval ) ( err error ) { return EWINDOWS }
2010-06-29 20:23:39 -07:00
2012-05-23 13:05:05 +10:00
// The Linger struct is wrong but we only noticed after Go 1.
// sysLinger is the real system call structure.
// BUG(brainman): The definition of Linger is not appropriate for direct use
// with Setsockopt and Getsockopt.
// Use SetsockoptLinger instead.
2010-06-29 20:23:39 -07:00
type Linger struct {
Onoff int32
Linger int32
}
2012-05-23 13:05:05 +10:00
type sysLinger struct {
Onoff uint16
Linger uint16
}
2011-06-02 10:10:17 -04:00
type IPMreq struct {
2011-02-17 12:00:02 +11:00
Multiaddr [ 4 ] byte /* in_addr */
Interface [ 4 ] byte /* in_addr */
}
2011-06-02 10:10:17 -04:00
type IPv6Mreq struct {
Multiaddr [ 16 ] byte /* in6_addr */
Interface uint32
}
2012-05-23 13:05:05 +10:00
func GetsockoptInt ( fd Handle , level , opt int ) ( int , error ) { return - 1 , EWINDOWS }
func SetsockoptLinger ( fd Handle , level , opt int , l * Linger ) ( err error ) {
sys := sysLinger { Onoff : uint16 ( l . Onoff ) , Linger : uint16 ( l . Linger ) }
return Setsockopt ( fd , int32 ( level ) , int32 ( opt ) , ( * byte ) ( unsafe . Pointer ( & sys ) ) , int32 ( unsafe . Sizeof ( sys ) ) )
}
2012-02-01 14:14:04 +09:00
func SetsockoptInet4Addr ( fd Handle , level , opt int , value [ 4 ] byte ) ( err error ) {
return Setsockopt ( fd , int32 ( level ) , int32 ( opt ) , ( * byte ) ( unsafe . Pointer ( & value [ 0 ] ) ) , 4 )
}
2011-11-13 22:44:52 -05:00
func SetsockoptIPMreq ( fd Handle , level , opt int , mreq * IPMreq ) ( err error ) {
2011-09-08 16:32:40 +10:00
return Setsockopt ( fd , int32 ( level ) , int32 ( opt ) , ( * byte ) ( unsafe . Pointer ( mreq ) ) , int32 ( unsafe . Sizeof ( * mreq ) ) )
}
2011-11-13 22:44:52 -05:00
func SetsockoptIPv6Mreq ( fd Handle , level , opt int , mreq * IPv6Mreq ) ( err error ) { return EWINDOWS }
2010-06-29 20:23:39 -07:00
2012-06-08 14:28:29 -04:00
func Getpid ( ) ( pid int ) { return int ( getCurrentProcessId ( ) ) }
2010-04-02 01:11:17 -07:00
2012-06-03 19:27:17 +10:00
func FindFirstFile ( name * uint16 , data * Win32finddata ) ( handle Handle , err error ) {
2012-06-08 13:54:48 -04:00
// NOTE(rsc): The Win32finddata struct is wrong for the system call:
// the two paths are each one uint16 short. Use the correct struct,
// a win32finddata1, and then copy the results out.
// There is no loss of expressivity here, because the final
// uint16, if it is used, is supposed to be a NUL, and Go doesn't need that.
// For Go 1.1, we might avoid the allocation of win32finddata1 here
// by adding a final Bug [2]uint16 field to the struct and then
// adjusting the fields in the result directly.
var data1 win32finddata1
handle , err = findFirstFile1 ( name , & data1 )
if err == nil {
copyFindData ( data , & data1 )
}
return
2012-06-03 19:27:17 +10:00
}
func FindNextFile ( handle Handle , data * Win32finddata ) ( err error ) {
2012-06-08 13:54:48 -04:00
var data1 win32finddata1
err = findNextFile1 ( handle , & data1 )
if err == nil {
copyFindData ( data , & data1 )
}
return
2012-06-03 19:27:17 +10:00
}
2014-06-14 15:51:00 +10:00
func getProcessEntry ( pid int ) ( * ProcessEntry32 , error ) {
snapshot , err := CreateToolhelp32Snapshot ( TH32CS_SNAPPROCESS , 0 )
if err != nil {
return nil , err
}
defer CloseHandle ( snapshot )
var procEntry ProcessEntry32
procEntry . Size = uint32 ( unsafe . Sizeof ( procEntry ) )
if err = Process32First ( snapshot , & procEntry ) ; err != nil {
return nil , err
}
for {
if procEntry . ProcessID == uint32 ( pid ) {
return & procEntry , nil
}
err = Process32Next ( snapshot , & procEntry )
if err != nil {
return nil , err
}
}
}
func Getppid ( ) ( ppid int ) {
pe , err := getProcessEntry ( Getpid ( ) )
if err != nil {
return - 1
}
return int ( pe . ParentProcessID )
}
2010-04-02 01:11:17 -07:00
2014-06-14 15:51:00 +10:00
// TODO(brainman): fix all needed for os
2014-07-17 17:02:46 +10:00
func Fchdir ( fd Handle ) ( err error ) { return EWINDOWS }
func Link ( oldpath , newpath string ) ( err error ) { return EWINDOWS }
func Symlink ( path , link string ) ( err error ) { return EWINDOWS }
2011-02-16 16:24:59 +11:00
2011-11-13 22:44:52 -05:00
func Fchmod ( fd Handle , mode uint32 ) ( err error ) { return EWINDOWS }
func Chown ( path string , uid int , gid int ) ( err error ) { return EWINDOWS }
func Lchown ( path string , uid int , gid int ) ( err error ) { return EWINDOWS }
func Fchown ( fd Handle , uid int , gid int ) ( err error ) { return EWINDOWS }
2010-04-02 01:11:17 -07:00
2010-04-26 23:17:14 -07:00
func Getuid ( ) ( uid int ) { return - 1 }
func Geteuid ( ) ( euid int ) { return - 1 }
func Getgid ( ) ( gid int ) { return - 1 }
func Getegid ( ) ( egid int ) { return - 1 }
2011-11-13 22:44:52 -05:00
func Getgroups ( ) ( gids [ ] int , err error ) { return nil , EWINDOWS }
2012-02-14 13:51:38 +11:00
type Signal int
func ( s Signal ) Signal ( ) { }
func ( s Signal ) String ( ) string {
if 0 <= s && int ( s ) < len ( signals ) {
str := signals [ s ]
if str != "" {
return str
}
}
return "signal " + itoa ( int ( s ) )
}
2014-07-17 17:02:46 +10:00
func LoadCreateSymbolicLink ( ) error {
return procCreateSymbolicLinkW . Find ( )
}
// Readlink returns the destination of the named symbolic link.
func Readlink ( path string , buf [ ] byte ) ( n int , err error ) {
fd , err := CreateFile ( StringToUTF16Ptr ( path ) , GENERIC_READ , 0 , nil , OPEN_EXISTING ,
FILE_FLAG_OPEN_REPARSE_POINT | FILE_FLAG_BACKUP_SEMANTICS , 0 )
if err != nil {
return - 1 , err
}
defer CloseHandle ( fd )
rdbbuf := make ( [ ] byte , MAXIMUM_REPARSE_DATA_BUFFER_SIZE )
var bytesReturned uint32
err = DeviceIoControl ( fd , FSCTL_GET_REPARSE_POINT , nil , 0 , & rdbbuf [ 0 ] , uint32 ( len ( rdbbuf ) ) , & bytesReturned , nil )
if err != nil {
return - 1 , err
}
rdb := ( * reparseDataBuffer ) ( unsafe . Pointer ( & rdbbuf [ 0 ] ) )
2015-01-06 09:47:37 +09:00
var s string
switch rdb . ReparseTag {
case IO_REPARSE_TAG_SYMLINK :
data := ( * symbolicLinkReparseBuffer ) ( unsafe . Pointer ( & rdb . reparseBuffer ) )
p := ( * [ 0xffff ] uint16 ) ( unsafe . Pointer ( & data . PathBuffer [ 0 ] ) )
2016-10-16 13:25:53 +09:00
s = UTF16ToString ( p [ data . SubstituteNameOffset / 2 : ( data . SubstituteNameOffset + data . SubstituteNameLength ) / 2 ] )
if data . Flags & _SYMLINK_FLAG_RELATIVE == 0 {
if len ( s ) >= 4 && s [ : 4 ] == ` \??\ ` {
s = s [ 4 : ]
switch {
case len ( s ) >= 2 && s [ 1 ] == ':' : // \??\C:\foo\bar
// do nothing
case len ( s ) >= 4 && s [ : 4 ] == ` UNC\ ` : // \??\UNC\foo\bar
s = ` \\ ` + s [ 4 : ]
default :
// unexpected; do nothing
}
} else {
// unexpected; do nothing
}
}
2015-01-06 09:47:37 +09:00
case _IO_REPARSE_TAG_MOUNT_POINT :
data := ( * mountPointReparseBuffer ) ( unsafe . Pointer ( & rdb . reparseBuffer ) )
p := ( * [ 0xffff ] uint16 ) ( unsafe . Pointer ( & data . PathBuffer [ 0 ] ) )
2016-10-16 13:25:53 +09:00
s = UTF16ToString ( p [ data . SubstituteNameOffset / 2 : ( data . SubstituteNameOffset + data . SubstituteNameLength ) / 2 ] )
if len ( s ) >= 4 && s [ : 4 ] == ` \??\ ` { // \??\C:\foo\bar
s = s [ 4 : ]
} else {
// unexpected; do nothing
}
2015-01-06 09:47:37 +09:00
default :
// the path is not a symlink or junction but another type of reparse
// point
2014-07-17 17:02:46 +10:00
return - 1 , ENOENT
}
n = copy ( buf , [ ] byte ( s ) )
2015-01-06 09:47:37 +09:00
2014-07-17 17:02:46 +10:00
return n , nil
}