76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
|
// Copyright 2009 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 linux openbsd
|
||
|
|
||
|
package os_test
|
||
|
|
||
|
import (
|
||
|
. "os"
|
||
|
"syscall"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
func checkUidGid(t *testing.T, path string, uid, gid int) {
|
||
|
dir, err := Stat(path)
|
||
|
if err != nil {
|
||
|
t.Fatalf("Stat %q (looking for uid/gid %d/%d): %s", path, uid, gid, err)
|
||
|
}
|
||
|
sys := dir.(*FileStat).Sys.(*syscall.Stat_t)
|
||
|
if int(sys.Uid) != uid {
|
||
|
t.Errorf("Stat %q: uid %d want %d", path, sys.Uid, uid)
|
||
|
}
|
||
|
if int(sys.Gid) != gid {
|
||
|
t.Errorf("Stat %q: gid %d want %d", path, sys.Gid, gid)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestChown(t *testing.T) {
|
||
|
// Chown is not supported under windows or Plan 9.
|
||
|
// Plan9 provides a native ChownPlan9 version instead.
|
||
|
if syscall.OS == "windows" || syscall.OS == "plan9" {
|
||
|
return
|
||
|
}
|
||
|
// Use TempDir() to make sure we're on a local file system,
|
||
|
// so that the group ids returned by Getgroups will be allowed
|
||
|
// on the file. On NFS, the Getgroups groups are
|
||
|
// basically useless.
|
||
|
f := newFile("TestChown", t)
|
||
|
defer Remove(f.Name())
|
||
|
defer f.Close()
|
||
|
dir, err := f.Stat()
|
||
|
if err != nil {
|
||
|
t.Fatalf("stat %s: %s", f.Name(), err)
|
||
|
}
|
||
|
|
||
|
// Can't change uid unless root, but can try
|
||
|
// changing the group id. First try our current group.
|
||
|
gid := Getgid()
|
||
|
t.Log("gid:", gid)
|
||
|
if err = Chown(f.Name(), -1, gid); err != nil {
|
||
|
t.Fatalf("chown %s -1 %d: %s", f.Name(), gid, err)
|
||
|
}
|
||
|
sys := dir.(*FileStat).Sys.(*syscall.Stat_t)
|
||
|
checkUidGid(t, f.Name(), int(sys.Uid), gid)
|
||
|
|
||
|
// Then try all the auxiliary groups.
|
||
|
groups, err := Getgroups()
|
||
|
if err != nil {
|
||
|
t.Fatalf("getgroups: %s", err)
|
||
|
}
|
||
|
t.Log("groups: ", groups)
|
||
|
for _, g := range groups {
|
||
|
if err = Chown(f.Name(), -1, g); err != nil {
|
||
|
t.Fatalf("chown %s -1 %d: %s", f.Name(), g, err)
|
||
|
}
|
||
|
checkUidGid(t, f.Name(), int(sys.Uid), g)
|
||
|
|
||
|
// change back to gid to test fd.Chown
|
||
|
if err = f.Chown(-1, gid); err != nil {
|
||
|
t.Fatalf("fchown %s -1 %d: %s", f.Name(), gid, err)
|
||
|
}
|
||
|
checkUidGid(t, f.Name(), int(sys.Uid), gid)
|
||
|
}
|
||
|
}
|