93 lines
2.8 KiB
Go
93 lines
2.8 KiB
Go
//go:build windows
|
|
|
|
package authconfig
|
|
|
|
import (
|
|
"errors"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/aritmolab/thothii/tools/tht/internal/safeio"
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
func TestLoadRejectsPermissiveWindowsDirectoryAuthAndUsersDACLs(t *testing.T) {
|
|
for name, makePermissive := range map[string]func(t *testing.T, directory string){
|
|
"directory": func(t *testing.T, directory string) {
|
|
setPermissiveAuthDACL(t, directory)
|
|
},
|
|
"auth file": func(t *testing.T, directory string) {
|
|
setPermissiveAuthDACL(t, filepath.Join(directory, "auth.yaml"))
|
|
},
|
|
"users file": func(t *testing.T, directory string) {
|
|
setPermissiveAuthDACL(t, filepath.Join(directory, "users.yaml"))
|
|
},
|
|
} {
|
|
t.Run(name, func(t *testing.T) {
|
|
directory := writePrivateWindowsAuthFiles(t)
|
|
makePermissive(t, directory)
|
|
if _, _, err := Load(directory); !errors.Is(err, safeio.ErrUnsafeFile) {
|
|
t.Fatalf("Load() error = %v, want ErrUnsafeFile", err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestMutateUsersCreatesAndRejectsPermissiveWindowsLockDACL(t *testing.T) {
|
|
directory := writePrivateWindowsAuthFiles(t)
|
|
if err := MutateUsers(directory, func(*Registry) error { return nil }); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
lockPath := filepath.Join(directory, lockFileName)
|
|
if err := safeio.ValidatePrivateRegular(lockPath); err != nil {
|
|
t.Fatalf("lock privacy error = %v", err)
|
|
}
|
|
setPermissiveAuthDACL(t, lockPath)
|
|
if err := MutateUsers(directory, func(*Registry) error { return nil }); !errors.Is(err, safeio.ErrUnsafeFile) {
|
|
t.Fatalf("MutateUsers() error = %v, want ErrUnsafeFile", err)
|
|
}
|
|
}
|
|
|
|
func writePrivateWindowsAuthFiles(t *testing.T) string {
|
|
t.Helper()
|
|
directory := writeAuthFiles(t, defaultAuthYAML, registryYAML(adminUserYAML("admin", "Admin", true, "admin")))
|
|
if err := safeio.ProtectPrivateDirectory(directory); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, name := range []string{"auth.yaml", "users.yaml"} {
|
|
if err := safeio.ProtectPrivateRegular(filepath.Join(directory, name)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
return directory
|
|
}
|
|
|
|
func setPermissiveAuthDACL(t *testing.T, path string) {
|
|
t.Helper()
|
|
world, err := windows.StringToSid("S-1-1-0")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var pinner runtime.Pinner
|
|
pinner.Pin(world)
|
|
defer pinner.Unpin()
|
|
acl, err := windows.ACLFromEntries([]windows.EXPLICIT_ACCESS{{
|
|
AccessPermissions: windows.GENERIC_READ | windows.GENERIC_WRITE,
|
|
AccessMode: windows.GRANT_ACCESS,
|
|
Trustee: windows.TRUSTEE{
|
|
TrusteeForm: windows.TRUSTEE_IS_SID,
|
|
TrusteeType: windows.TRUSTEE_IS_GROUP,
|
|
TrusteeValue: windows.TrusteeValueFromSID(world),
|
|
},
|
|
}}, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := windows.SetNamedSecurityInfo(path, windows.SE_FILE_OBJECT,
|
|
windows.DACL_SECURITY_INFORMATION|windows.PROTECTED_DACL_SECURITY_INFORMATION,
|
|
nil, nil, acl, nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|