// Copyright 2016 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. // +build linux package unix_test import ( "bufio" "bytes" "errors" "fmt" "io/ioutil" "os" "path/filepath" "runtime" "runtime/debug" "strconv" "strings" "testing" "time" "unsafe" "golang.org/x/sys/unix" ) func TestIoctlGetInt(t *testing.T) { f, err := os.Open("/dev/random") if err != nil { t.Fatalf("failed to open device: %v", err) } defer f.Close() v, err := unix.IoctlGetInt(int(f.Fd()), unix.RNDGETENTCNT) if err != nil { t.Fatalf("failed to perform ioctl: %v", err) } t.Logf("%d bits of entropy available", v) } func TestIoctlRetInt(t *testing.T) { f, err := os.Open("/proc/self/ns/mnt") if err != nil { t.Skipf("skipping test, %v", err) } defer f.Close() v, err := unix.IoctlRetInt(int(f.Fd()), unix.NS_GET_NSTYPE) if err != nil { if err == unix.ENOTTY { t.Skipf("old kernel? (need Linux >= 4.11)") } t.Fatalf("failed to perform ioctl: %v", err) } if v != unix.CLONE_NEWNS { t.Fatalf("unexpected return from ioctl; expected %v, got %v", v, unix.CLONE_NEWNS) } } func TestIoctlGetRTCTime(t *testing.T) { f, err := os.Open("/dev/rtc0") if err != nil { t.Skipf("skipping test, %v", err) } defer f.Close() v, err := unix.IoctlGetRTCTime(int(f.Fd())) if err != nil { t.Fatalf("failed to perform ioctl: %v", err) } t.Logf("RTC time: %04d-%02d-%02d %02d:%02d:%02d", v.Year+1900, v.Mon+1, v.Mday, v.Hour, v.Min, v.Sec) } func TestIoctlGetRTCWkAlrm(t *testing.T) { f, err := os.Open("/dev/rtc0") if err != nil { t.Skipf("skipping test, %v", err) } defer f.Close() v, err := unix.IoctlGetRTCWkAlrm(int(f.Fd())) if err != nil { t.Fatalf("failed to perform ioctl: %v", err) } t.Logf("RTC wake alarm enabled '%d'; time: %04d-%02d-%02d %02d:%02d:%02d", v.Enabled, v.Time.Year+1900, v.Time.Mon+1, v.Time.Mday, v.Time.Hour, v.Time.Min, v.Time.Sec) } func TestPpoll(t *testing.T) { if runtime.GOOS == "android" { t.Skip("mkfifo syscall is not available on android, skipping test") } defer chtmpdir(t)() f, cleanup := mktmpfifo(t) defer cleanup() const timeout = 100 * time.Millisecond ok := make(chan bool, 1) go func() { select { case <-time.After(10 * timeout): t.Errorf("Ppoll: failed to timeout after %d", 10*timeout) case <-ok: } }() fds := []unix.PollFd{{Fd: int32(f.Fd()), Events: unix.POLLIN}} timeoutTs := unix.NsecToTimespec(int64(timeout)) n, err := unix.Ppoll(fds, &timeoutTs, nil) ok <- true if err != nil { t.Errorf("Ppoll: unexpected error: %v", err) return } if n != 0 { t.Errorf("Ppoll: wrong number of events: got %v, expected %v", n, 0) return } } func TestTime(t *testing.T) { var ut unix.Time_t ut2, err := unix.Time(&ut) if err != nil { t.Fatalf("Time: %v", err) } if ut != ut2 { t.Errorf("Time: return value %v should be equal to argument %v", ut2, ut) } var now time.Time for i := 0; i < 10; i++ { ut, err = unix.Time(nil) if err != nil { t.Fatalf("Time: %v", err) } now = time.Now() diff := int64(ut) - now.Unix() if -1 <= diff && diff <= 1 { return } } t.Errorf("Time: return value %v should be nearly equal to time.Now().Unix() %v±1", ut, now.Unix()) } func TestUtime(t *testing.T) { defer chtmpdir(t)() touch(t, "file1") buf := &unix.Utimbuf{ Modtime: 12345, } err := unix.Utime("file1", buf) if err != nil { t.Fatalf("Utime: %v", err) } fi, err := os.Stat("file1") if err != nil { t.Fatal(err) } if fi.ModTime().Unix() != 12345 { t.Errorf("Utime: failed to change modtime: expected %v, got %v", 12345, fi.ModTime().Unix()) } } func TestRlimitAs(t *testing.T) { // disable GC during to avoid flaky test defer debug.SetGCPercent(debug.SetGCPercent(-1)) var rlim unix.Rlimit err := unix.Getrlimit(unix.RLIMIT_AS, &rlim) if err != nil { t.Fatalf("Getrlimit: %v", err) } var zero unix.Rlimit if zero == rlim { t.Fatalf("Getrlimit: got zero value %#v", rlim) } set := rlim set.Cur = uint64(unix.Getpagesize()) err = unix.Setrlimit(unix.RLIMIT_AS, &set) if err != nil { t.Fatalf("Setrlimit: set failed: %#v %v", set, err) } // RLIMIT_AS was set to the page size, so mmap()'ing twice the page size // should fail. See 'man 2 getrlimit'. _, err = unix.Mmap(-1, 0, 2*unix.Getpagesize(), unix.PROT_NONE, unix.MAP_ANON|unix.MAP_PRIVATE) if err == nil { t.Fatal("Mmap: unexpectedly succeeded after setting RLIMIT_AS") } err = unix.Setrlimit(unix.RLIMIT_AS, &rlim) if err != nil { t.Fatalf("Setrlimit: restore failed: %#v %v", rlim, err) } b, err := unix.Mmap(-1, 0, 2*unix.Getpagesize(), unix.PROT_NONE, unix.MAP_ANON|unix.MAP_PRIVATE) if err != nil { t.Fatalf("Mmap: %v", err) } err = unix.Munmap(b) if err != nil { t.Fatalf("Munmap: %v", err) } } func TestPselect(t *testing.T) { for { n, err := unix.Pselect(0, nil, nil, nil, &unix.Timespec{Sec: 0, Nsec: 0}, nil) if err == unix.EINTR { t.Logf("Pselect interrupted") continue } else if err != nil { t.Fatalf("Pselect: %v", err) } if n != 0 { t.Fatalf("Pselect: got %v ready file descriptors, expected 0", n) } break } dur := 2500 * time.Microsecond var took time.Duration for { // On some platforms (e.g. Linux), the passed-in timespec is // updated by pselect(2). Make sure to reset to the full // duration in case of an EINTR. ts := unix.NsecToTimespec(int64(dur)) start := time.Now() n, err := unix.Pselect(0, nil, nil, nil, &ts, nil) took = time.Since(start) if err == unix.EINTR { t.Logf("Pselect interrupted after %v", took) continue } else if err != nil { t.Fatalf("Pselect: %v", err) } if n != 0 { t.Fatalf("Pselect: got %v ready file descriptors, expected 0", n) } break } // On some builder 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("Pselect: got %v timeout, expected at least %v", took, dur) } } func TestSchedSetaffinity(t *testing.T) { var newMask unix.CPUSet newMask.Zero() if newMask.Count() != 0 { t.Errorf("CpuZero: didn't zero CPU set: %v", newMask) } cpu := 1 newMask.Set(cpu) if newMask.Count() != 1 || !newMask.IsSet(cpu) { t.Errorf("CpuSet: didn't set CPU %d in set: %v", cpu, newMask) } cpu = 5 newMask.Set(cpu) if newMask.Count() != 2 || !newMask.IsSet(cpu) { t.Errorf("CpuSet: didn't set CPU %d in set: %v", cpu, newMask) } newMask.Clear(cpu) if newMask.Count() != 1 || newMask.IsSet(cpu) { t.Errorf("CpuClr: didn't clear CPU %d in set: %v", cpu, newMask) } runtime.LockOSThread() defer runtime.UnlockOSThread() var oldMask unix.CPUSet err := unix.SchedGetaffinity(0, &oldMask) if err != nil { t.Fatalf("SchedGetaffinity: %v", err) } if runtime.NumCPU() < 2 { t.Skip("skipping setaffinity tests on single CPU system") } if runtime.GOOS == "android" { t.Skip("skipping setaffinity tests on android") } // On a system like ppc64x where some cores can be disabled using ppc64_cpu, // setaffinity should only be called with enabled cores. The valid cores // are found from the oldMask, but if none are found then the setaffinity // tests are skipped. Issue #27875. cpu = 1 if !oldMask.IsSet(cpu) { newMask.Zero() for i := 0; i < len(oldMask); i++ { if oldMask.IsSet(i) { newMask.Set(i) break } } if newMask.Count() == 0 { t.Skip("skipping setaffinity tests if CPU not available") } } err = unix.SchedSetaffinity(0, &newMask) if err != nil { t.Fatalf("SchedSetaffinity: %v", err) } var gotMask unix.CPUSet err = unix.SchedGetaffinity(0, &gotMask) if err != nil { t.Fatalf("SchedGetaffinity: %v", err) } if gotMask != newMask { t.Errorf("SchedSetaffinity: returned affinity mask does not match set affinity mask") } // Restore old mask so it doesn't affect successive tests err = unix.SchedSetaffinity(0, &oldMask) if err != nil { t.Fatalf("SchedSetaffinity: %v", err) } } func TestStatx(t *testing.T) { var stx unix.Statx_t err := unix.Statx(unix.AT_FDCWD, ".", 0, 0, &stx) if err == unix.ENOSYS || err == unix.EPERM { t.Skip("statx syscall is not available, skipping test") } else if err != nil { t.Fatalf("Statx: %v", err) } defer chtmpdir(t)() touch(t, "file1") var st unix.Stat_t err = unix.Stat("file1", &st) if err != nil { t.Fatalf("Stat: %v", err) } flags := unix.AT_STATX_SYNC_AS_STAT err = unix.Statx(unix.AT_FDCWD, "file1", flags, unix.STATX_ALL, &stx) if err != nil { t.Fatalf("Statx: %v", err) } if uint32(stx.Mode) != st.Mode { t.Errorf("Statx: returned stat mode does not match Stat") } ctime := unix.StatxTimestamp{Sec: int64(st.Ctim.Sec), Nsec: uint32(st.Ctim.Nsec)} mtime := unix.StatxTimestamp{Sec: int64(st.Mtim.Sec), Nsec: uint32(st.Mtim.Nsec)} if stx.Ctime != ctime { t.Errorf("Statx: returned stat ctime does not match Stat") } if stx.Mtime != mtime { t.Errorf("Statx: returned stat mtime does not match Stat") } err = os.Symlink("file1", "symlink1") if err != nil { t.Fatal(err) } err = unix.Lstat("symlink1", &st) if err != nil { t.Fatalf("Lstat: %v", err) } err = unix.Statx(unix.AT_FDCWD, "symlink1", flags, unix.STATX_BASIC_STATS, &stx) if err != nil { t.Fatalf("Statx: %v", err) } // follow symlink, expect a regulat file if stx.Mode&unix.S_IFREG == 0 { t.Errorf("Statx: didn't follow symlink") } err = unix.Statx(unix.AT_FDCWD, "symlink1", flags|unix.AT_SYMLINK_NOFOLLOW, unix.STATX_ALL, &stx) if err != nil { t.Fatalf("Statx: %v", err) } // follow symlink, expect a symlink if stx.Mode&unix.S_IFLNK == 0 { t.Errorf("Statx: unexpectedly followed symlink") } if uint32(stx.Mode) != st.Mode { t.Errorf("Statx: returned stat mode does not match Lstat") } ctime = unix.StatxTimestamp{Sec: int64(st.Ctim.Sec), Nsec: uint32(st.Ctim.Nsec)} mtime = unix.StatxTimestamp{Sec: int64(st.Mtim.Sec), Nsec: uint32(st.Mtim.Nsec)} if stx.Ctime != ctime { t.Errorf("Statx: returned stat ctime does not match Lstat") } if stx.Mtime != mtime { t.Errorf("Statx: returned stat mtime does not match Lstat") } } // stringsFromByteSlice converts a sequence of attributes to a []string. // On Linux, each entry is a NULL-terminated string. func stringsFromByteSlice(buf []byte) []string { var result []string off := 0 for i, b := range buf { if b == 0 { result = append(result, string(buf[off:i])) off = i + 1 } } return result } func TestFaccessat(t *testing.T) { defer chtmpdir(t)() touch(t, "file1") err := unix.Faccessat(unix.AT_FDCWD, "file1", unix.R_OK, 0) if err != nil { t.Errorf("Faccessat: unexpected error: %v", err) } err = unix.Faccessat(unix.AT_FDCWD, "file1", unix.R_OK, 2) if err != unix.EINVAL { t.Errorf("Faccessat: unexpected error: %v, want EINVAL", err) } err = unix.Faccessat(unix.AT_FDCWD, "file1", unix.R_OK, unix.AT_EACCESS) if err != nil { t.Errorf("Faccessat: unexpected error: %v", err) } err = os.Symlink("file1", "symlink1") if err != nil { t.Fatal(err) } err = unix.Faccessat(unix.AT_FDCWD, "symlink1", unix.R_OK, unix.AT_SYMLINK_NOFOLLOW) if err != nil { t.Errorf("Faccessat SYMLINK_NOFOLLOW: unexpected error %v", err) } // We can't really test AT_SYMLINK_NOFOLLOW, because there // doesn't seem to be any way to change the mode of a symlink. // We don't test AT_EACCESS because such tests are only // meaningful if run as root. err = unix.Fchmodat(unix.AT_FDCWD, "file1", 0, 0) if err != nil { t.Errorf("Fchmodat: unexpected error %v", err) } err = unix.Faccessat(unix.AT_FDCWD, "file1", unix.F_OK, unix.AT_SYMLINK_NOFOLLOW) if err != nil { t.Errorf("Faccessat: unexpected error: %v", err) } err = unix.Faccessat(unix.AT_FDCWD, "file1", unix.R_OK, unix.AT_SYMLINK_NOFOLLOW) if err != unix.EACCES { if unix.Getuid() != 0 { t.Errorf("Faccessat: unexpected error: %v, want EACCES", err) } } } func TestSyncFileRange(t *testing.T) { file, err := ioutil.TempFile("", "TestSyncFileRange") if err != nil { t.Fatal(err) } defer os.Remove(file.Name()) defer file.Close() err = unix.SyncFileRange(int(file.Fd()), 0, 0, 0) if err == unix.ENOSYS || err == unix.EPERM { t.Skip("sync_file_range syscall is not available, skipping test") } else if err != nil { t.Fatalf("SyncFileRange: %v", err) } // invalid flags flags := 0xf00 err = unix.SyncFileRange(int(file.Fd()), 0, 0, flags) if err != unix.EINVAL { t.Fatalf("SyncFileRange: unexpected error: %v, want EINVAL", err) } } func TestClockNanosleep(t *testing.T) { delay := 50 * time.Millisecond // Relative timespec. start := time.Now() rel := unix.NsecToTimespec(delay.Nanoseconds()) remain := unix.Timespec{} for { err := unix.ClockNanosleep(unix.CLOCK_MONOTONIC, 0, &rel, &remain) if err == unix.ENOSYS || err == unix.EPERM { t.Skip("clock_nanosleep syscall is not available, skipping test") } else if err == unix.EINTR { t.Logf("ClockNanosleep interrupted after %v", time.Since(start)) rel = remain continue } else if err != nil { t.Errorf("ClockNanosleep(CLOCK_MONOTONIC, 0, %#v, nil) = %v", &rel, err) } else if slept := time.Since(start); slept < delay { t.Errorf("ClockNanosleep(CLOCK_MONOTONIC, 0, %#v, nil) slept only %v", &rel, slept) } break } // Absolute timespec. for { start = time.Now() until := start.Add(delay) abs := unix.NsecToTimespec(until.UnixNano()) err := unix.ClockNanosleep(unix.CLOCK_REALTIME, unix.TIMER_ABSTIME, &abs, nil) if err == unix.EINTR { t.Logf("ClockNanosleep interrupted after %v", time.Since(start)) continue } else if err != nil { t.Errorf("ClockNanosleep(CLOCK_REALTIME, TIMER_ABSTIME, %#v (=%v), nil) = %v", &abs, until, err) } else if slept := time.Since(start); slept < delay { t.Errorf("ClockNanosleep(CLOCK_REALTIME, TIMER_ABSTIME, %#v (=%v), nil) slept only %v", &abs, until, slept) } break } // Invalid clock. clock_nanosleep(2) says EINVAL, but it’s actually EOPNOTSUPP. err := unix.ClockNanosleep(unix.CLOCK_THREAD_CPUTIME_ID, 0, &rel, nil) if err != unix.EINVAL && err != unix.EOPNOTSUPP { t.Errorf("ClockNanosleep(CLOCK_THREAD_CPUTIME_ID, 0, %#v, nil) = %v, want EINVAL or EOPNOTSUPP", &rel, err) } } func TestOpenByHandleAt(t *testing.T) { skipIfNotSupported := func(t *testing.T, name string, err error) { if err == unix.EPERM { t.Skipf("skipping %s test without CAP_DAC_READ_SEARCH", name) } if err == unix.ENOSYS { t.Skipf("%s system call not available", name) } if err == unix.EOPNOTSUPP { t.Skipf("%s not supported on this filesystem", name) } } h, mountID, err := unix.NameToHandleAt(unix.AT_FDCWD, "syscall_linux_test.go", 0) if err != nil { skipIfNotSupported(t, "name_to_handle_at", err) t.Fatalf("NameToHandleAt: %v", err) } t.Logf("mountID: %v, handle: size=%d, type=%d, bytes=%q", mountID, h.Size(), h.Type(), h.Bytes()) mount, err := openMountByID(mountID) if err != nil { t.Fatalf("openMountByID: %v", err) } defer mount.Close() for _, clone := range []bool{false, true} { t.Run("clone="+strconv.FormatBool(clone), func(t *testing.T) { if clone { h = unix.NewFileHandle(h.Type(), h.Bytes()) } fd, err := unix.OpenByHandleAt(int(mount.Fd()), h, unix.O_RDONLY) skipIfNotSupported(t, "open_by_handle_at", err) if err != nil { t.Fatalf("OpenByHandleAt: %v", err) } defer unix.Close(fd) t.Logf("opened fd %v", fd) f := os.NewFile(uintptr(fd), "") slurp, err := ioutil.ReadAll(f) if err != nil { t.Fatal(err) } const substr = "Some substring for a test." if !strings.Contains(string(slurp), substr) { t.Errorf("didn't find substring %q in opened file; read %d bytes", substr, len(slurp)) } }) } } func openMountByID(mountID int) (f *os.File, err error) { mi, err := os.Open("/proc/self/mountinfo") if err != nil { return nil, err } defer mi.Close() bs := bufio.NewScanner(mi) wantPrefix := []byte(fmt.Sprintf("%v ", mountID)) for bs.Scan() { if !bytes.HasPrefix(bs.Bytes(), wantPrefix) { continue } fields := strings.Fields(bs.Text()) dev := fields[4] return os.Open(dev) } if err := bs.Err(); err != nil { return nil, err } return nil, errors.New("mountID not found") } func TestEpoll(t *testing.T) { efd, err := unix.EpollCreate1(unix.EPOLL_CLOEXEC) if err != nil { t.Fatalf("EpollCreate1: %v", err) } defer unix.Close(efd) r, w, err := os.Pipe() if err != nil { t.Fatal(err) } defer r.Close() defer w.Close() fd := int(r.Fd()) ev := unix.EpollEvent{Events: unix.EPOLLIN, Fd: int32(fd)} err = unix.EpollCtl(efd, unix.EPOLL_CTL_ADD, fd, &ev) if err != nil { t.Fatalf("EpollCtl: %v", err) } if _, err := w.Write([]byte("HELLO GOPHER")); err != nil { t.Fatal(err) } events := make([]unix.EpollEvent, 128) n, err := unix.EpollWait(efd, events, 1) if err != nil { t.Fatalf("EpollWait: %v", err) } if n != 1 { t.Errorf("EpollWait: wrong number of events: got %v, expected 1", n) } got := int(events[0].Fd) if got != fd { t.Errorf("EpollWait: wrong Fd in event: got %v, expected %v", got, fd) } } func TestPrctlRetInt(t *testing.T) { err := unix.Prctl(unix.PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0) if err != nil { t.Skipf("Prctl: %v, skipping test", err) } v, err := unix.PrctlRetInt(unix.PR_GET_NO_NEW_PRIVS, 0, 0, 0, 0) if err != nil { t.Fatalf("failed to perform prctl: %v", err) } if v != 1 { t.Fatalf("unexpected return from prctl; got %v, expected %v", v, 1) } } func TestTimerfd(t *testing.T) { var now unix.Timespec if err := unix.ClockGettime(unix.CLOCK_REALTIME, &now); err != nil { t.Fatalf("ClockGettime: %v", err) } tfd, err := unix.TimerfdCreate(unix.CLOCK_REALTIME, 0) if err == unix.ENOSYS { t.Skip("timerfd_create system call not implemented") } else if err != nil { t.Fatalf("TimerfdCreate: %v", err) } defer unix.Close(tfd) var timeSpec unix.ItimerSpec if err := unix.TimerfdGettime(tfd, &timeSpec); err != nil { t.Fatalf("TimerfdGettime: %v", err) } if timeSpec.Value.Nsec != 0 || timeSpec.Value.Sec != 0 { t.Fatalf("TimerfdGettime: timer is already set, but shouldn't be") } timeSpec = unix.ItimerSpec{ Interval: unix.NsecToTimespec(int64(time.Millisecond)), Value: now, } if err := unix.TimerfdSettime(tfd, unix.TFD_TIMER_ABSTIME, &timeSpec, nil); err != nil { t.Fatalf("TimerfdSettime: %v", err) } const totalTicks = 10 const bufferLength = 8 buffer := make([]byte, bufferLength) var count uint64 = 0 for count < totalTicks { n, err := unix.Read(tfd, buffer) if err != nil { t.Fatalf("Timerfd: %v", err) } else if n != bufferLength { t.Fatalf("Timerfd: got %d bytes from timerfd, expected %d bytes", n, bufferLength) } count += *(*uint64)(unsafe.Pointer(&buffer)) } } func TestOpenat2(t *testing.T) { how := &unix.OpenHow{ Flags: unix.O_RDONLY, } fd, err := unix.Openat2(unix.AT_FDCWD, ".", how) if err != nil { if err == unix.ENOSYS || err == unix.EPERM { t.Skipf("openat2: %v (old kernel? need Linux >= 5.6)", err) } t.Fatalf("openat2: %v", err) } if err := unix.Close(fd); err != nil { t.Fatalf("close: %v", err) } // prepare tempDir, err := ioutil.TempDir("", t.Name()) if err != nil { t.Fatal(err) } defer os.RemoveAll(tempDir) subdir := filepath.Join(tempDir, "dir") if err := os.Mkdir(subdir, 0755); err != nil { t.Fatal(err) } symlink := filepath.Join(subdir, "symlink") if err := os.Symlink("../", symlink); err != nil { t.Fatal(err) } dirfd, err := unix.Open(subdir, unix.O_RDONLY, 0) if err != nil { t.Fatalf("open(%q): %v", subdir, err) } defer unix.Close(dirfd) // openat2 with no extra flags -- should succeed fd, err = unix.Openat2(dirfd, "symlink", how) if err != nil { t.Errorf("Openat2 should succeed, got %v", err) } if err := unix.Close(fd); err != nil { t.Fatalf("close: %v", err) } // open with RESOLVE_BENEATH, should result in EXDEV how.Resolve = unix.RESOLVE_BENEATH fd, err = unix.Openat2(dirfd, "symlink", how) if err == nil { if err := unix.Close(fd); err != nil { t.Fatalf("close: %v", err) } } if err != unix.EXDEV { t.Errorf("Openat2 should fail with EXDEV, got %v", err) } }