2012-06-03 03:27:17 -06:00
|
|
|
// Copyright 2012 The Go Authors. All rights reserved.
|
|
|
|
// Use of this source code is governed by a BSD-style
|
|
|
|
// license that can be found in the LICENSE file.
|
|
|
|
|
|
|
|
package syscall_test
|
|
|
|
|
|
|
|
import (
|
|
|
|
"io/ioutil"
|
|
|
|
"os"
|
|
|
|
"path/filepath"
|
|
|
|
"syscall"
|
|
|
|
"testing"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestWin32finddata(t *testing.T) {
|
|
|
|
dir, err := ioutil.TempDir("", "go-build")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("failed to create temp directory: %v", err)
|
|
|
|
}
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
|
|
|
|
path := filepath.Join(dir, "long_name.and_extension")
|
|
|
|
f, err := os.Create(path)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("failed to create %v: %v", path, err)
|
|
|
|
}
|
|
|
|
f.Close()
|
|
|
|
|
|
|
|
type X struct {
|
2012-06-08 12:04:44 -06:00
|
|
|
fd syscall.Win32finddata
|
2012-06-03 03:27:17 -06:00
|
|
|
got byte
|
|
|
|
pad [10]byte // to protect ourselves
|
|
|
|
|
|
|
|
}
|
|
|
|
var want byte = 2 // it is unlikely to have this character in the filename
|
|
|
|
x := X{got: want}
|
|
|
|
|
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 15:24:32 -06:00
|
|
|
pathp, _ := syscall.UTF16PtrFromString(path)
|
|
|
|
h, err := syscall.FindFirstFile(pathp, &(x.fd))
|
2012-06-03 03:27:17 -06:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("FindFirstFile failed: %v", err)
|
|
|
|
}
|
|
|
|
err = syscall.FindClose(h)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("FindClose failed: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if x.got != want {
|
|
|
|
t.Fatalf("memory corruption: want=%d got=%d", want, x.got)
|
|
|
|
}
|
|
|
|
}
|
2012-08-29 07:44:46 -06:00
|
|
|
|
|
|
|
func abort(funcname string, err error) {
|
|
|
|
panic(funcname + " failed: " + err.Error())
|
|
|
|
}
|
|
|
|
|
|
|
|
func ExampleLoadLibrary() {
|
|
|
|
h, err := syscall.LoadLibrary("kernel32.dll")
|
|
|
|
if err != nil {
|
|
|
|
abort("LoadLibrary", err)
|
|
|
|
}
|
|
|
|
defer syscall.FreeLibrary(h)
|
|
|
|
proc, err := syscall.GetProcAddress(h, "GetVersion")
|
|
|
|
if err != nil {
|
|
|
|
abort("GetProcAddress", err)
|
|
|
|
}
|
|
|
|
r, _, _ := syscall.Syscall(uintptr(proc), 0, 0, 0, 0)
|
|
|
|
major := byte(r)
|
|
|
|
minor := uint8(r >> 8)
|
|
|
|
build := uint16(r >> 16)
|
|
|
|
print("windows version ", major, ".", minor, " (Build ", build, ")\n")
|
|
|
|
}
|