diff options
Diffstat (limited to '')
-rw-r--r-- | dependencies/pkg/mod/golang.org/x/sys@v0.1.0/unix/syscall_zos_test.go | 863 |
1 files changed, 863 insertions, 0 deletions
diff --git a/dependencies/pkg/mod/golang.org/x/sys@v0.1.0/unix/syscall_zos_test.go b/dependencies/pkg/mod/golang.org/x/sys@v0.1.0/unix/syscall_zos_test.go new file mode 100644 index 0000000..42ae5ba --- /dev/null +++ b/dependencies/pkg/mod/golang.org/x/sys@v0.1.0/unix/syscall_zos_test.go @@ -0,0 +1,863 @@ +// Copyright 2020 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. + +//go:build zos && s390x +// +build zos,s390x + +package unix_test + +import ( + "flag" + "fmt" + "io/ioutil" + "net" + "os" + "os/exec" + "path/filepath" + "runtime" + "strconv" + "syscall" + "testing" + "time" + "unsafe" + + "golang.org/x/sys/unix" +) + +// Tests that below functions, structures and constants are consistent +// on all Unix-like systems. +func _() { + // program scheduling priority functions and constants + var ( + _ func(int, int, int) error = unix.Setpriority + _ func(int, int) (int, error) = unix.Getpriority + ) + const ( + _ int = unix.PRIO_USER + _ int = unix.PRIO_PROCESS + _ int = unix.PRIO_PGRP + ) + + // termios constants + const ( + _ int = unix.TCIFLUSH + _ int = unix.TCIOFLUSH + _ int = unix.TCOFLUSH + ) + + // fcntl file locking structure and constants + var ( + _ = unix.Flock_t{ + Type: int16(0), + Whence: int16(0), + Start: int64(0), + Len: int64(0), + Pid: int32(0), + } + ) + const ( + _ = unix.F_GETLK + _ = unix.F_SETLK + _ = unix.F_SETLKW + ) +} + +func TestErrnoSignalName(t *testing.T) { + testErrors := []struct { + num syscall.Errno + name string + }{ + {syscall.EPERM, "EDC5139I"}, + {syscall.EINVAL, "EDC5121I"}, + {syscall.ENOENT, "EDC5129I"}, + } + + for _, te := range testErrors { + t.Run(fmt.Sprintf("%d/%s", te.num, te.name), func(t *testing.T) { + e := unix.ErrnoName(te.num) + if e != te.name { + t.Errorf("ErrnoName(%d) returned %s, want %s", te.num, e, te.name) + } + }) + } + + testSignals := []struct { + num syscall.Signal + name string + }{ + {syscall.SIGHUP, "SIGHUP"}, + {syscall.SIGPIPE, "SIGPIPE"}, + {syscall.SIGSEGV, "SIGSEGV"}, + } + + for _, ts := range testSignals { + t.Run(fmt.Sprintf("%d/%s", ts.num, ts.name), func(t *testing.T) { + s := unix.SignalName(ts.num) + if s != ts.name { + t.Errorf("SignalName(%d) returned %s, want %s", ts.num, s, ts.name) + } + }) + } +} + +func TestSignalNum(t *testing.T) { + testSignals := []struct { + name string + want syscall.Signal + }{ + {"SIGHUP", syscall.SIGHUP}, + {"SIGPIPE", syscall.SIGPIPE}, + {"SIGSEGV", syscall.SIGSEGV}, + {"NONEXISTS", 0}, + } + for _, ts := range testSignals { + t.Run(fmt.Sprintf("%s/%d", ts.name, ts.want), func(t *testing.T) { + got := unix.SignalNum(ts.name) + if got != ts.want { + t.Errorf("SignalNum(%s) returned %d, want %d", ts.name, got, ts.want) + } + }) + + } +} + +func TestFcntlInt(t *testing.T) { + t.Parallel() + file, err := ioutil.TempFile("", "TestFnctlInt") + if err != nil { + t.Fatal(err) + } + defer os.Remove(file.Name()) + defer file.Close() + f := file.Fd() + flags, err := unix.FcntlInt(f, unix.F_GETFD, 0) + if err != nil { + t.Fatal(err) + } + if flags&unix.FD_CLOEXEC == 0 { + t.Errorf("flags %#x do not include FD_CLOEXEC", flags) + } +} + +// TestFcntlFlock tests whether the file locking structure matches +// the calling convention of each kernel. +func TestFcntlFlock(t *testing.T) { + name := filepath.Join(os.TempDir(), "TestFcntlFlock") + fd, err := unix.Open(name, unix.O_CREAT|unix.O_RDWR|unix.O_CLOEXEC, 0) + if err != nil { + t.Fatalf("Open failed: %v", err) + } + defer unix.Unlink(name) + defer unix.Close(fd) + flock := unix.Flock_t{ + Type: unix.F_RDLCK, + Start: 0, Len: 0, Whence: 1, + } + if err := unix.FcntlFlock(uintptr(fd), unix.F_GETLK, &flock); err != nil { + t.Fatalf("FcntlFlock failed: %v", err) + } +} + +// TestPassFD tests passing a file descriptor over a Unix socket. +// +// This test involved both a parent and child process. The parent +// process is invoked as a normal test, with "go test", which then +// runs the child process by running the current test binary with args +// "-test.run=^TestPassFD$" and an environment variable used to signal +// that the test should become the child process instead. +func TestPassFD(t *testing.T) { + if runtime.GOOS == "darwin" && runtime.GOARCH == "arm64" { + t.Skip("cannot exec subprocess on iOS, skipping test") + } + + if os.Getenv("GO_WANT_HELPER_PROCESS") == "1" { + passFDChild() + return + } + + if runtime.GOOS == "aix" { + // Unix network isn't properly working on AIX + // 7.2 with Technical Level < 2 + out, err := exec.Command("oslevel", "-s").Output() + if err != nil { + t.Skipf("skipping on AIX because oslevel -s failed: %v", err) + } + + if len(out) < len("7200-XX-ZZ-YYMM") { // AIX 7.2, Tech Level XX, Service Pack ZZ, date YYMM + t.Skip("skipping on AIX because oslevel -s hasn't the right length") + } + aixVer := string(out[:4]) + tl, err := strconv.Atoi(string(out[5:7])) + if err != nil { + t.Skipf("skipping on AIX because oslevel -s output cannot be parsed: %v", err) + } + if aixVer < "7200" || (aixVer == "7200" && tl < 2) { + t.Skip("skipped on AIX versions previous to 7.2 TL 2") + } + } + + tempDir, err := ioutil.TempDir("", "TestPassFD") + if err != nil { + t.Fatal(err) + } + defer os.RemoveAll(tempDir) + + fds, err := unix.Socketpair(unix.AF_LOCAL, unix.SOCK_STREAM, 0) + if err != nil { + t.Fatalf("Socketpair: %v", err) + } + writeFile := os.NewFile(uintptr(fds[0]), "child-writes") + readFile := os.NewFile(uintptr(fds[1]), "parent-reads") + defer writeFile.Close() + defer readFile.Close() + + cmd := exec.Command(os.Args[0], "-test.run=^TestPassFD$", "--", tempDir) + cmd.Env = []string{"GO_WANT_HELPER_PROCESS=1"} + if lp := os.Getenv("LD_LIBRARY_PATH"); lp != "" { + cmd.Env = append(cmd.Env, "LD_LIBRARY_PATH="+lp) + } + cmd.ExtraFiles = []*os.File{writeFile} + + out, err := cmd.CombinedOutput() + if len(out) > 0 || err != nil { + t.Fatalf("child process: %q, %v", out, err) + } + + c, err := net.FileConn(readFile) + if err != nil { + t.Fatalf("FileConn: %v", err) + } + defer c.Close() + + uc, ok := c.(*net.UnixConn) + if !ok { + t.Fatalf("unexpected FileConn type; expected UnixConn, got %T", c) + } + + buf := make([]byte, 32) // expect 1 byte + oob := make([]byte, 32) // expect 24 bytes + closeUnix := time.AfterFunc(5*time.Second, func() { + t.Logf("timeout reading from unix socket") + uc.Close() + }) + _, oobn, _, _, err := uc.ReadMsgUnix(buf, oob) + if err != nil { + t.Fatalf("ReadMsgUnix: %v", err) + } + closeUnix.Stop() + + scms, err := unix.ParseSocketControlMessage(oob[:oobn]) + if err != nil { + t.Fatalf("ParseSocketControlMessage: %v", err) + } + if len(scms) != 1 { + t.Fatalf("expected 1 SocketControlMessage; got scms = %#v", scms) + } + scm := scms[0] + gotFds, err := unix.ParseUnixRights(&scm) + if err != nil { + t.Fatalf("unix.ParseUnixRights: %v", err) + } + if len(gotFds) != 1 { + t.Fatalf("wanted 1 fd; got %#v", gotFds) + } + + f := os.NewFile(uintptr(gotFds[0]), "fd-from-child") + defer f.Close() + + got, err := ioutil.ReadAll(f) + want := "Hello from child process!\n" + if string(got) != want { + t.Errorf("child process ReadAll: %q, %v; want %q", got, err, want) + } +} + +// passFDChild is the child process used by TestPassFD. +func passFDChild() { + defer os.Exit(0) + + // Look for our fd. It should be fd 3, but we work around an fd leak + // bug here (http://golang.org/issue/2603) to let it be elsewhere. + var uc *net.UnixConn + for fd := uintptr(3); fd <= 10; fd++ { + f := os.NewFile(fd, "unix-conn") + var ok bool + netc, _ := net.FileConn(f) + uc, ok = netc.(*net.UnixConn) + if ok { + break + } + } + if uc == nil { + fmt.Println("failed to find unix fd") + return + } + + // Make a file f to send to our parent process on uc. + // We make it in tempDir, which our parent will clean up. + flag.Parse() + tempDir := flag.Arg(0) + f, err := ioutil.TempFile(tempDir, "") + if err != nil { + fmt.Printf("TempFile: %v", err) + return + } + + f.Write([]byte("Hello from child process!\n")) + f.Seek(0, 0) + + rights := unix.UnixRights(int(f.Fd())) + dummyByte := []byte("x") + n, oobn, err := uc.WriteMsgUnix(dummyByte, rights, nil) + if err != nil { + fmt.Printf("WriteMsgUnix: %v", err) + return + } + if n != 1 || oobn != len(rights) { + fmt.Printf("WriteMsgUnix = %d, %d; want 1, %d", n, oobn, len(rights)) + return + } +} + +// TestUnixRightsRoundtrip tests that UnixRights, ParseSocketControlMessage, +// and ParseUnixRights are able to successfully round-trip lists of file descriptors. +func TestUnixRightsRoundtrip(t *testing.T) { + testCases := [...][][]int{ + {{42}}, + {{1, 2}}, + {{3, 4, 5}}, + {{}}, + {{1, 2}, {3, 4, 5}, {}, {7}}, + } + for _, testCase := range testCases { + b := []byte{} + var n int + for _, fds := range testCase { + // Last assignment to n wins + n = len(b) + unix.CmsgLen(4*len(fds)) + b = append(b, unix.UnixRights(fds...)...) + } + // Truncate b + b = b[:n] + + scms, err := unix.ParseSocketControlMessage(b) + if err != nil { + t.Fatalf("ParseSocketControlMessage: %v", err) + } + if len(scms) != len(testCase) { + t.Fatalf("expected %v SocketControlMessage; got scms = %#v", len(testCase), scms) + } + for i, scm := range scms { + gotFds, err := unix.ParseUnixRights(&scm) + if err != nil { + t.Fatalf("ParseUnixRights: %v", err) + } + wantFds := testCase[i] + if len(gotFds) != len(wantFds) { + t.Fatalf("expected %v fds, got %#v", len(wantFds), gotFds) + } + for j, fd := range gotFds { + if fd != wantFds[j] { + t.Fatalf("expected fd %v, got %v", wantFds[j], fd) + } + } + } + } +} + +func TestRlimit(t *testing.T) { + var rlimit, zero unix.Rlimit + err := unix.Getrlimit(unix.RLIMIT_NOFILE, &rlimit) + if err != nil { + t.Fatalf("Getrlimit: save failed: %v", err) + } + if zero == rlimit { + t.Fatalf("Getrlimit: save failed: got zero value %#v", rlimit) + } + set := rlimit + set.Cur = set.Max - 1 + if runtime.GOOS == "darwin" && set.Cur > 10240 { + // The max file limit is 10240, even though + // the max returned by Getrlimit is 1<<63-1. + // This is OPEN_MAX in sys/syslimits.h. + set.Cur = 10240 + } + err = unix.Setrlimit(unix.RLIMIT_NOFILE, &set) + if err != nil { + t.Fatalf("Setrlimit: set failed: %#v %v", set, err) + } + var get unix.Rlimit + err = unix.Getrlimit(unix.RLIMIT_NOFILE, &get) + if err != nil { + t.Fatalf("Getrlimit: get failed: %v", err) + } + set = rlimit + set.Cur = set.Max - 1 + if set != get { + // Seems like Darwin requires some privilege to + // increase the soft limit of rlimit sandbox, though + // Setrlimit never reports an error. + switch runtime.GOOS { + case "darwin": + default: + t.Fatalf("Rlimit: change failed: wanted %#v got %#v", set, get) + } + } + err = unix.Setrlimit(unix.RLIMIT_NOFILE, &rlimit) + if err != nil { + t.Fatalf("Setrlimit: restore failed: %#v %v", rlimit, err) + } +} + +func TestSeekFailure(t *testing.T) { + _, err := unix.Seek(-1, 0, 0) + if err == nil { + t.Fatalf("Seek(-1, 0, 0) did not fail") + } + str := err.Error() // used to crash on Linux + t.Logf("Seek: %v", str) + if str == "" { + t.Fatalf("Seek(-1, 0, 0) return error with empty message") + } +} + +func TestSetsockoptString(t *testing.T) { + // should not panic on empty string, see issue #31277 + err := unix.SetsockoptString(-1, 0, 0, "") + if err == nil { + t.Fatalf("SetsockoptString: did not fail") + } +} + +func TestDup(t *testing.T) { + file, err := ioutil.TempFile("", "TestDup") + if err != nil { + t.Fatalf("Tempfile failed: %v", err) + } + defer os.Remove(file.Name()) + defer file.Close() + f := int(file.Fd()) + + newFd, err := unix.Dup(f) + if err != nil { + t.Fatalf("Dup: %v", err) + } + + // Create and reserve a file descriptor. + // Dup2 automatically closes it before reusing it. + nullFile, err := os.Open("/dev/null") + if err != nil { + t.Fatal(err) + } + dupFd := int(file.Fd()) + err = unix.Dup2(newFd, dupFd) + if err != nil { + t.Fatalf("Dup2: %v", err) + } + // Keep the dummy file open long enough to not be closed in + // its finalizer. + runtime.KeepAlive(nullFile) + + b1 := []byte("Test123") + b2 := make([]byte, 7) + _, err = unix.Write(dupFd, b1) + if err != nil { + t.Fatalf("Write to dup2 fd failed: %v", err) + } + _, err = unix.Seek(f, 0, 0) + if err != nil { + t.Fatalf("Seek failed: %v", err) + } + _, err = unix.Read(f, b2) + if err != nil { + t.Fatalf("Read back failed: %v", err) + } + if string(b1) != string(b2) { + t.Errorf("Dup: stdout write not in file, expected %v, got %v", string(b1), string(b2)) + } +} + +func TestGetwd(t *testing.T) { + fd, err := os.Open(".") + if err != nil { + t.Fatalf("Open .: %s", err) + } + defer fd.Close() + // Directory list for test. Do not worry if any are symlinks or do not + // exist on some common unix desktop environments. That will be checked. + dirs := []string{"/", "/usr/bin", "/etc", "/var", "/opt"} + switch runtime.GOOS { + case "android": + dirs = []string{"/", "/system/bin"} + case "darwin": + switch runtime.GOARCH { + case "arm64": + d1, err := ioutil.TempDir("", "d1") + if err != nil { + t.Fatalf("TempDir: %v", err) + } + d2, err := ioutil.TempDir("", "d2") + if err != nil { + t.Fatalf("TempDir: %v", err) + } + dirs = []string{d1, d2} + } + } + oldwd := os.Getenv("PWD") + for _, d := range dirs { + // Check whether d exists, is a dir and that d's path does not contain a symlink + fi, err := os.Stat(d) + if err != nil || !fi.IsDir() { + t.Logf("Test dir %s stat error (%v) or not a directory, skipping", d, err) + continue + } + check, err := filepath.EvalSymlinks(d) + if err != nil || check != d { + t.Logf("Test dir %s (%s) is symlink or other error (%v), skipping", d, check, err) + continue + } + err = os.Chdir(d) + if err != nil { + t.Fatalf("Chdir: %v", err) + } + pwd, err := unix.Getwd() + if err != nil { + t.Fatalf("Getwd in %s: %s", d, err) + } + os.Setenv("PWD", oldwd) + err = fd.Chdir() + if err != nil { + // We changed the current directory and cannot go back. + // Don't let the tests continue; they'll scribble + // all over some other directory. + fmt.Fprintf(os.Stderr, "fchdir back to dot failed: %s\n", err) + os.Exit(1) + } + if pwd != d { + t.Fatalf("Getwd returned %q want %q", pwd, d) + } + } +} + +func TestMkdev(t *testing.T) { + major := uint32(42) + minor := uint32(7) + dev := unix.Mkdev(major, minor) + + if unix.Major(dev) != major { + t.Errorf("Major(%#x) == %d, want %d", dev, unix.Major(dev), major) + } + if unix.Minor(dev) != minor { + t.Errorf("Minor(%#x) == %d, want %d", dev, unix.Minor(dev), minor) + } +} + +// mktmpfifo creates a temporary FIFO and provides a cleanup function. +func mktmpfifo(t *testing.T) (*os.File, func()) { + err := unix.Mkfifo("fifo", 0666) + if err != nil { + t.Fatalf("mktmpfifo: failed to create FIFO: %v", err) + } + + f, err := os.OpenFile("fifo", os.O_RDWR, 0666) + if err != nil { + os.Remove("fifo") + t.Fatalf("mktmpfifo: failed to open FIFO: %v", err) + } + + return f, func() { + f.Close() + os.Remove("fifo") + } +} + +// utilities taken from os/os_test.go + +func touch(t *testing.T, name string) { + f, err := os.Create(name) + if err != nil { + t.Fatal(err) + } + if err := f.Close(); err != nil { + t.Fatal(err) + } +} + +// chtmpdir changes the working directory to a new temporary directory and +// provides a cleanup function. Used when PWD is read-only. +func chtmpdir(t *testing.T) func() { + oldwd, err := os.Getwd() + if err != nil { + t.Fatalf("chtmpdir: %v", err) + } + d, err := ioutil.TempDir("", "test") + if err != nil { + t.Fatalf("chtmpdir: %v", err) + } + if err := os.Chdir(d); err != nil { + t.Fatalf("chtmpdir: %v", err) + } + return func() { + if err := os.Chdir(oldwd); err != nil { + t.Fatalf("chtmpdir: %v", err) + } + os.RemoveAll(d) + } +} + +func TestMountUnmount(t *testing.T) { + // use an available fs + var buffer struct { + header unix.W_Mnth + fsinfo [64]unix.W_Mntent + } + fsCount, err := unix.W_Getmntent_A((*byte)(unsafe.Pointer(&buffer)), int(unsafe.Sizeof(buffer))) + if err != nil { + t.Fatalf("W_Getmntent_A returns with error: %s", err.Error()) + } else if fsCount == 0 { + t.Fatalf("W_Getmntent_A returns no entries") + } + var fs string + var fstype string + var mountpoint string + var available bool = false + for i := 0; i < fsCount; i++ { + err = unix.Unmount(unix.ByteSliceToString(buffer.fsinfo[i].Mountpoint[:]), unix.MTM_RDWR) + if err != nil { + // Unmount and Mount require elevated privilege + // If test is run without such permission, skip test + if err == unix.EPERM { + t.Logf("Permission denied for Unmount. Skipping test (Errno2: %X)", unix.Errno2()) + return + } else if err == unix.EBUSY { + continue + } else { + t.Fatalf("Unmount returns with error: %s", err.Error()) + } + } else { + available = true + fs = unix.ByteSliceToString(buffer.fsinfo[i].Fsname[:]) + fstype = unix.ByteSliceToString(buffer.fsinfo[i].Fstname[:]) + mountpoint = unix.ByteSliceToString(buffer.fsinfo[i].Mountpoint[:]) + t.Logf("using file system = %s; fstype = %s and mountpoint = %s\n", fs, fstype, mountpoint) + break + } + } + if !available { + t.Fatalf("No filesystem available") + } + // test unmount + buffer.header = unix.W_Mnth{} + fsCount, err = unix.W_Getmntent_A((*byte)(unsafe.Pointer(&buffer)), int(unsafe.Sizeof(buffer))) + if err != nil { + t.Fatalf("W_Getmntent_A returns with error: %s", err.Error()) + } + for i := 0; i < fsCount; i++ { + if unix.ByteSliceToString(buffer.fsinfo[i].Fsname[:]) == fs { + t.Fatalf("File system found after unmount") + } + } + // test mount + err = unix.Mount(fs, mountpoint, fstype, unix.MTM_RDWR, "") + if err != nil { + t.Fatalf("Mount returns with error: %s", err.Error()) + } + buffer.header = unix.W_Mnth{} + fsCount, err = unix.W_Getmntent_A((*byte)(unsafe.Pointer(&buffer)), int(unsafe.Sizeof(buffer))) + if err != nil { + t.Fatalf("W_Getmntent_A returns with error: %s", err.Error()) + } + fsMounted := false + for i := 0; i < fsCount; i++ { + if unix.ByteSliceToString(buffer.fsinfo[i].Fsname[:]) == fs && + unix.ByteSliceToString(buffer.fsinfo[i].Mountpoint[:]) == mountpoint { + fsMounted = true + } + } + if !fsMounted { + t.Fatalf("%s not mounted after Mount()", fs) + } +} + +func TestChroot(t *testing.T) { + // create temp dir and tempfile 1 + tempDir, err := ioutil.TempDir("", "TestChroot") + if err != nil { + t.Fatalf("TempDir: %s", err.Error()) + } + defer os.RemoveAll(tempDir) + f, err := ioutil.TempFile(tempDir, "chroot_test_file") + if err != nil { + t.Fatalf("TempFile: %s", err.Error()) + } + // chroot temp dir + err = unix.Chroot(tempDir) + // Chroot requires elevated privilege + // If test is run without such permission, skip test + if err == unix.EPERM { + t.Logf("Denied permission for Chroot. Skipping test (Errno2: %X)", unix.Errno2()) + return + } else if err != nil { + t.Fatalf("Chroot: %s", err.Error()) + } + // check if tempDir contains test file + files, err := ioutil.ReadDir("/") + if err != nil { + t.Fatalf("ReadDir: %s", err.Error()) + } + found := false + for _, file := range files { + if file.Name() == filepath.Base(f.Name()) { + found = true + break + } + } + if !found { + t.Fatalf("Temp file not found in temp dir") + } +} + +func TestFlock(t *testing.T) { + if os.Getenv("GO_WANT_HELPER_PROCESS") == "1" { + defer os.Exit(0) + if len(os.Args) != 3 { + fmt.Printf("bad argument") + return + } + fn := os.Args[2] + f, err := os.OpenFile(fn, os.O_RDWR, 0755) + if err != nil { + fmt.Printf("%s", err.Error()) + return + } + err = unix.Flock(int(f.Fd()), unix.LOCK_EX|unix.LOCK_NB) + // if the lock we are trying should be locked, ignore EAGAIN error + // otherwise, report all errors + if err != nil && err != unix.EAGAIN { + fmt.Printf("%s", err.Error()) + } + } else { + // create temp dir and tempfile 1 + tempDir, err := ioutil.TempDir("", "TestFlock") + if err != nil { + t.Fatalf("TempDir: %s", err.Error()) + } + defer os.RemoveAll(tempDir) + f, err := ioutil.TempFile(tempDir, "flock_test_file") + if err != nil { + t.Fatalf("TempFile: %s", err.Error()) + } + fd := int(f.Fd()) + + /* Test Case 1 + * Try acquiring an occupied lock from another process + */ + err = unix.Flock(fd, unix.LOCK_EX) + if err != nil { + t.Fatalf("Flock: %s", err.Error()) + } + cmd := exec.Command(os.Args[0], "-test.run=TestFlock", f.Name()) + cmd.Env = append(os.Environ(), "GO_WANT_HELPER_PROCESS=1") + out, err := cmd.CombinedOutput() + if len(out) > 0 || err != nil { + t.Fatalf("child process: %q, %v", out, err) + } + err = unix.Flock(fd, unix.LOCK_UN) + if err != nil { + t.Fatalf("Flock: %s", err.Error()) + } + + /* Test Case 2 + * Try locking with Flock and FcntlFlock for same file + */ + err = unix.Flock(fd, unix.LOCK_EX) + if err != nil { + t.Fatalf("Flock: %s", err.Error()) + } + flock := unix.Flock_t{ + Type: int16(unix.F_WRLCK), + Whence: int16(0), + Start: int64(0), + Len: int64(0), + Pid: int32(unix.Getppid()), + } + err = unix.FcntlFlock(f.Fd(), unix.F_SETLK, &flock) + if err != nil { + t.Fatalf("FcntlFlock: %s", err.Error()) + } + } +} + +func TestSelect(t *testing.T) { + for { + n, err := unix.Select(0, nil, nil, nil, &unix.Timeval{Sec: 0, Usec: 0}) + if err == unix.EINTR { + t.Logf("Select interrupted") + continue + } else if err != nil { + t.Fatalf("Select: %v", err) + } + if n != 0 { + t.Fatalf("Select: got %v ready file descriptors, expected 0", n) + } + break + } + + dur := 250 * time.Millisecond + var took time.Duration + for { + // On some platforms (e.g. Linux), the passed-in timeval is + // updated by select(2). Make sure to reset to the full duration + // in case of an EINTR. + tv := unix.NsecToTimeval(int64(dur)) + start := time.Now() + n, err := unix.Select(0, nil, nil, nil, &tv) + took = time.Since(start) + if err == unix.EINTR { + t.Logf("Select interrupted after %v", took) + continue + } else if err != nil { + t.Fatalf("Select: %v", err) + } + if n != 0 { + t.Fatalf("Select: got %v ready file descriptors, expected 0", n) + } + break + } + + // On some BSDs the actual timeout might also be slightly less than the requested. + // Add an acceptable margin to avoid flaky tests. + if took < dur*2/3 { + t.Errorf("Select: got %v timeout, expected at least %v", took, dur) + } + + rr, ww, err := os.Pipe() + if err != nil { + t.Fatal(err) + } + defer rr.Close() + defer ww.Close() + + if _, err := ww.Write([]byte("HELLO GOPHER")); err != nil { + t.Fatal(err) + } + + rFdSet := &unix.FdSet{} + fd := int(rr.Fd()) + rFdSet.Set(fd) + + for { + n, err := unix.Select(fd+1, rFdSet, nil, nil, nil) + if err == unix.EINTR { + t.Log("Select interrupted") + continue + } else if err != nil { + t.Fatalf("Select: %v", err) + } + if n != 1 { + t.Fatalf("Select: got %v ready file descriptors, expected 1", n) + } + break + } +} |