世界
63 lines
1.6 KiB
Go
63 lines
1.6 KiB
Go
//go:build linux
|
|
|
|
package main
|
|
|
|
import (
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
func TestLinuxPeerAuthentication(t *testing.T) {
|
|
socketPath := filepath.Join(t.TempDir(), "daemon.sock")
|
|
listener, err := net.Listen("unix", socketPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer listener.Close()
|
|
clientConnection, err := net.Dial("unix", socketPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer clientConnection.Close()
|
|
serverConnection, err := listener.Accept()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
daemon := &Daemon{}
|
|
authenticatedConnection, authenticationInformation, err := (&linuxTransportCredentials{daemon: daemon}).ServerHandshake(serverConnection)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
authentication, loaded := authenticationInformation.(*peerAuthInfo)
|
|
if !loaded {
|
|
t.Fatal("missing peer authentication information")
|
|
}
|
|
identity := authentication.identity
|
|
if identity.UserID != strconv.Itoa(os.Getuid()) {
|
|
t.Fatalf("unexpected peer user ID: %s", identity.UserID)
|
|
}
|
|
if identity.ProcessID != uint32(os.Getpid()) {
|
|
t.Fatalf("unexpected peer process ID: %d", identity.ProcessID)
|
|
}
|
|
expectedStartTime, err := linuxProcessStartTime(identity.ProcessID)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if identity.ProcessStartTime != expectedStartTime {
|
|
t.Fatalf("unexpected peer process start time: %d", identity.ProcessStartTime)
|
|
}
|
|
if len(daemon.peerConnections) != 1 {
|
|
t.Fatalf("unexpected authenticated connection count: %d", len(daemon.peerConnections))
|
|
}
|
|
err = authenticatedConnection.Close()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(daemon.peerConnections) != 0 {
|
|
t.Fatalf("authenticated connection was not removed: %d", len(daemon.peerConnections))
|
|
}
|
|
}
|