diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 13:14:23 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 13:14:23 +0000 |
commit | 73df946d56c74384511a194dd01dbe099584fd1a (patch) | |
tree | fd0bcea490dd81327ddfbb31e215439672c9a068 /src/syscall/getdirentries_test.go | |
parent | Initial commit. (diff) | |
download | golang-1.16-73df946d56c74384511a194dd01dbe099584fd1a.tar.xz golang-1.16-73df946d56c74384511a194dd01dbe099584fd1a.zip |
Adding upstream version 1.16.10.upstream/1.16.10upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/syscall/getdirentries_test.go')
-rw-r--r-- | src/syscall/getdirentries_test.go | 88 |
1 files changed, 88 insertions, 0 deletions
diff --git a/src/syscall/getdirentries_test.go b/src/syscall/getdirentries_test.go new file mode 100644 index 0000000..66bb8ac --- /dev/null +++ b/src/syscall/getdirentries_test.go @@ -0,0 +1,88 @@ +// Copyright 2019 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 darwin freebsd netbsd openbsd + +package syscall_test + +import ( + "fmt" + "os" + "path/filepath" + "sort" + "strings" + "syscall" + "testing" + "unsafe" +) + +func TestGetdirentries(t *testing.T) { + for _, count := range []int{10, 1000} { + t.Run(fmt.Sprintf("n=%d", count), func(t *testing.T) { + testGetdirentries(t, count) + }) + } +} +func testGetdirentries(t *testing.T, count int) { + if count > 100 && testing.Short() && os.Getenv("GO_BUILDER_NAME") == "" { + t.Skip("skipping in -short mode") + } + d, err := os.MkdirTemp("", "getdirentries-test") + if err != nil { + t.Fatalf("Tempdir: %v", err) + } + defer os.RemoveAll(d) + var names []string + for i := 0; i < count; i++ { + names = append(names, fmt.Sprintf("file%03d", i)) + } + + // Make files in the temp directory + for _, name := range names { + err := os.WriteFile(filepath.Join(d, name), []byte("data"), 0) + if err != nil { + t.Fatalf("WriteFile: %v", err) + } + } + + // Read files using Getdirentries + var names2 []string + fd, err := syscall.Open(d, syscall.O_RDONLY, 0) + if err != nil { + t.Fatalf("Open: %v", err) + } + defer syscall.Close(fd) + var base uintptr + var buf [2048]byte + for { + n, err := syscall.Getdirentries(fd, buf[:], &base) + if err != nil { + t.Fatalf("Getdirentries: %v", err) + } + if n == 0 { + break + } + data := buf[:n] + for len(data) > 0 { + // If multiple Dirents are written into buf, sometimes when we reach the final one, + // we have cap(buf) < Sizeof(Dirent). So use an appropriate slice to copy from data. + var dirent syscall.Dirent + copy((*[unsafe.Sizeof(dirent)]byte)(unsafe.Pointer(&dirent))[:], data) + + data = data[dirent.Reclen:] + name := make([]byte, dirent.Namlen) + for i := 0; i < int(dirent.Namlen); i++ { + name[i] = byte(dirent.Name[i]) + } + names2 = append(names2, string(name)) + } + } + + names = append(names, ".", "..") // Getdirentries returns these also + sort.Strings(names) + sort.Strings(names2) + if strings.Join(names, ":") != strings.Join(names2, ":") { + t.Errorf("names don't match\n names: %q\nnames2: %q", names, names2) + } +} |