func TestAttach(t *testing.T) {
var err error
flag.Parse()
ufs := new(ufs.Ufs)
ufs.Dotu = false
ufs.Id = "ufs"
ufs.Debuglevel = *debug
ufs.Start(ufs)
t.Log("ufs starting\n")
// determined by build tags
//extraFuncs()
l, err := net.Listen("unix", "")
if err != nil {
t.Fatalf("Can not start listener: %v", err)
}
srvAddr := l.Addr().String()
t.Logf("Server is at %v", srvAddr)
go func() {
if err = ufs.StartListener(l); err != nil {
t.Fatalf("Can not start listener: %v", err)
}
}()
var conn net.Conn
if conn, err = net.Dial("unix", srvAddr); err != nil {
t.Fatalf("%v", err)
} else {
t.Logf("Got a conn, %v\n", conn)
}
user := p.OsUsers.Uid2User(os.Geteuid())
clnt := NewClnt(conn, 8192, false)
// run enough attaches to maybe let the race detector trip.
// The default, 1024, is lower than I'd like, but some environments don't
// let you do a huge number, as they throttle the accept rate.
for i := 0; i < *numAttach; i++ {
_, err := clnt.Attach(nil, user, "/tmp")
if err != nil {
t.Fatalf("Connect failed: %v\n", err)
}
defer clnt.Unmount()
}
}
func TestAttachOpenReaddir(t *testing.T) {
var err error
flag.Parse()
ufs := new(ufs.Ufs)
ufs.Dotu = false
ufs.Id = "ufs"
ufs.Debuglevel = *debug
ufs.Start(ufs)
tmpDir, err := ioutil.TempDir("", "go9")
if err != nil {
t.Fatal("Can't create temp directory")
}
//defer os.RemoveAll(tmpDir)
ufs.Root = tmpDir
t.Logf("ufs starting in %v\n", tmpDir)
// determined by build tags
//extraFuncs()
l, err := net.Listen("unix", "")
if err != nil {
t.Fatalf("Can not start listener: %v", err)
}
srvAddr := l.Addr().String()
t.Logf("Server is at %v", srvAddr)
go func() {
if err = ufs.StartListener(l); err != nil {
t.Fatalf("Can not start listener: %v", err)
}
}()
var conn net.Conn
if conn, err = net.Dial("unix", srvAddr); err != nil {
t.Fatalf("%v", err)
} else {
t.Logf("Got a conn, %v\n", conn)
}
clnt := NewClnt(conn, 8192, false)
// packet debugging on clients is broken.
clnt.Debuglevel = 0 // *debug
user := p.OsUsers.Uid2User(os.Geteuid())
rootfid, err := clnt.Attach(nil, user, "/")
if err != nil {
t.Fatalf("%v", err)
}
t.Logf("attached, rootfid %v\n", rootfid)
dirfid := clnt.FidAlloc()
if _, err = clnt.Walk(rootfid, dirfid, []string{"."}); err != nil {
t.Fatalf("%v", err)
}
// Now create a whole bunch of files to test readdir
for i := 0; i < *numDir; i++ {
f := fmt.Sprintf(path.Join(tmpDir, fmt.Sprintf("%d", i)))
if err := ioutil.WriteFile(f, []byte(f), 0600); err != nil {
t.Fatalf("Create %v: got %v, want nil", f, err)
}
}
if err = clnt.Open(dirfid, 0); err != nil {
t.Fatalf("%v", err)
}
var b []byte
if b, err = clnt.Read(dirfid, 0, 64*1024); err != nil {
t.Fatalf("%v", err)
}
var i, amt int
var offset uint64
err = nil
found := make([]int, *numDir)
fail := false
for err == nil {
if b, err = clnt.Read(dirfid, offset, 64*1024); err != nil {
t.Fatalf("%v", err)
}
t.Logf("clnt.Read returns [%v,%v]", len(b), err)
if len(b) == 0 {
break
}
for b != nil && len(b) > 0 {
var d *p.Dir
if d, b, amt, err = p.UnpackDir(b, ufs.Dotu); err != nil {
t.Errorf("UnpackDir returns %v", err)
break
} else {
if *debug > 128 {
t.Logf("Entry %d: %v", i, d)
}
i++
offset += uint64(amt)
ix, err := strconv.Atoi(d.Name)
if err != nil {
t.Errorf("File name %v is wrong; %v (dirent %v)", d.Name, err, d)
continue
}
if found[ix] > 0 {
t.Errorf("Element %d already returned %d times", ix, found[ix])
}
found[ix]++
}
}
}
if i != *numDir {
t.Fatalf("Reading %v: got %d entries, wanted %d, err %v", tmpDir, i, *numDir, err)
}
if fail {
t.Fatalf("I give up")
}
t.Logf("-----------------------------> Alternate form, using readdir and File")
// Alternate form, using readdir and File
dirfile, err := clnt.FOpen(".", p.OREAD)
if err != nil {
t.Fatalf("%v", err)
}
i, amt, offset = 0, 0, 0
err = nil
passes := 0
found = make([]int, *numDir)
fail = false
for err == nil {
d, err := dirfile.Readdir(*numDir)
if err != nil && err != io.EOF {
t.Errorf("%v", err)
}
t.Logf("d is %v", d)
if len(d) == 0 {
break
}
for _, v := range d {
ix, err := strconv.Atoi(v.Name)
if err != nil {
t.Errorf("File name %v is wrong; %v (dirent %v)", v.Name, err, v)
continue
}
if found[ix] > 0 {
t.Errorf("Element %d already returned %d times", ix, found[ix])
}
found[ix]++
}
i += len(d)
if i >= *numDir {
break
}
if passes > *numDir {
t.Fatalf("%d iterations, %d read: no progress", passes, i)
}
passes++
}
if i != *numDir {
t.Fatalf("Readdir %v: got %d entries, wanted %d", tmpDir, i, *numDir)
}
}
func TestRename(t *testing.T) {
var err error
flag.Parse()
ufs := new(ufs.Ufs)
ufs.Dotu = false
ufs.Id = "ufs"
ufs.Debuglevel = *debug
ufs.Msize = 8192
ufs.Start(ufs)
tmpDir, err := ioutil.TempDir("", "go9")
if err != nil {
t.Fatal("Can't create temp directory")
}
defer os.RemoveAll(tmpDir)
ufs.Root = tmpDir
t.Logf("ufs starting in %v", tmpDir)
// determined by build tags
//extraFuncs()
l, err := net.Listen("unix", "")
if err != nil {
t.Fatalf("Can not start listener: %v", err)
}
srvAddr := l.Addr().String()
t.Logf("Server is at %v", srvAddr)
go func() {
if err = ufs.StartListener(l); err != nil {
t.Fatalf("Can not start listener: %v", err)
}
}()
var conn net.Conn
if conn, err = net.Dial("unix", srvAddr); err != nil {
t.Fatalf("%v", err)
} else {
t.Logf("Got a conn, %v\n", conn)
}
clnt := NewClnt(conn, 8192, false)
user := p.OsUsers.Uid2User(os.Geteuid())
rootfid, err := clnt.Attach(nil, user, "/")
if err != nil {
t.Fatalf("%v", err)
}
t.Logf("attached to %v, rootfid %v\n", tmpDir, rootfid)
// OK, create a file behind go9ps back and then rename it.
b := make([]byte, 0)
from := path.Join(tmpDir, "a")
to := path.Join(tmpDir, "b")
if err = ioutil.WriteFile(from, b, 0666); err != nil {
t.Fatalf("%v", err)
}
f := clnt.FidAlloc()
if _, err = clnt.Walk(rootfid, f, []string{"a"}); err != nil {
t.Fatalf("%v", err)
}
t.Logf("Walked to a")
if _, err := clnt.Stat(f); err != nil {
t.Fatalf("%v", err)
}
if err := clnt.FSync(f); err != nil {
t.Fatalf("%v", err)
}
if err = clnt.Rename(f, "b"); err != nil {
t.Errorf("%v", err)
}
// the old one should be gone, and the new one should be there.
if _, err = ioutil.ReadFile(from); err == nil {
t.Errorf("ReadFile(%v): got nil, want err", from)
}
if _, err = ioutil.ReadFile(to); err != nil {
t.Errorf("ReadFile(%v): got %v, want nil", from, err)
}
// now try with an absolute path, which is supported on ufs servers.
// It's not guaranteed to work on all servers, but it is hugely useful
// on those that can do it -- which is almost all of them, save Plan 9
// of course.
from = to
if err = clnt.Rename(f, "c"); err != nil {
t.Errorf("%v", err)
}
// the old one should be gone, and the new one should be there.
if _, err = ioutil.ReadFile(from); err == nil {
t.Errorf("ReadFile(%v): got nil, want err", from)
}
to = path.Join(tmpDir, "c")
if _, err = ioutil.ReadFile(to); err != nil {
t.Errorf("ReadFile(%v): got %v, want nil", to, err)
}
// Make sure they can't walk out of the root.
from = to
if err = clnt.Rename(f, "../../../../d"); err != nil {
t.Errorf("%v", err)
}
// the old one should be gone, and the new one should be there.
if _, err = ioutil.ReadFile(from); err == nil {
t.Errorf("ReadFile(%v): got nil, want err", from)
}
to = path.Join(tmpDir, "d")
if _, err = ioutil.ReadFile(to); err != nil {
t.Errorf("ReadFile(%v): got %v, want nil", to, err)
}
}
