82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestInheritedListener(t *testing.T) {
|
|
if mode := os.Getenv("SCERT_LISTENER_TEST_CHILD"); mode != "" {
|
|
os.Setenv("LISTEN_PID", strconv.Itoa(os.Getpid()))
|
|
os.Setenv("LISTEN_FDS", "1")
|
|
switch mode {
|
|
case "pid":
|
|
os.Setenv("LISTEN_PID", "0")
|
|
case "fds":
|
|
os.Setenv("LISTEN_FDS", "2")
|
|
}
|
|
listener, err := inheritedListener(os.Getenv("SCERT_LISTENER_TEST_PATH"))
|
|
if mode == "valid" {
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := listener.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := os.Lstat(os.Getenv("SCERT_LISTENER_TEST_PATH")); err != nil {
|
|
t.Fatal("继承监听器关闭不应删除 systemd socket", err)
|
|
}
|
|
} else if err == nil {
|
|
listener.Close()
|
|
t.Fatal("无效继承监听器被接受")
|
|
}
|
|
fmt.Println("INHERITED_LISTENER_VERIFIED")
|
|
return
|
|
}
|
|
for _, mode := range []string{"valid", "pid", "fds", "path", "tcp"} {
|
|
t.Run(mode, func(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "a.sock")
|
|
var listener net.Listener
|
|
var file *os.File
|
|
var err error
|
|
if mode == "tcp" {
|
|
tcp, listenErr := net.ListenTCP("tcp4", &net.TCPAddr{IP: net.IPv4(127, 0, 0, 1)})
|
|
if listenErr != nil {
|
|
t.Fatal(listenErr)
|
|
}
|
|
listener = tcp
|
|
file, err = tcp.File()
|
|
} else {
|
|
unix, listenErr := net.ListenUnix("unix", &net.UnixAddr{Name: path, Net: "unix"})
|
|
if listenErr != nil {
|
|
t.Fatal(listenErr)
|
|
}
|
|
unix.SetUnlinkOnClose(false)
|
|
listener = unix
|
|
file, err = unix.File()
|
|
}
|
|
defer listener.Close()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer file.Close()
|
|
if mode == "path" {
|
|
path += ".wrong"
|
|
}
|
|
command := exec.Command(os.Args[0], "-test.run=^TestInheritedListener$")
|
|
command.Env = append(os.Environ(), "SCERT_LISTENER_TEST_CHILD="+mode, "SCERT_LISTENER_TEST_PATH="+path)
|
|
command.ExtraFiles = []*os.File{file}
|
|
output, err := command.CombinedOutput()
|
|
if err != nil || !strings.Contains(string(output), "INHERITED_LISTENER_VERIFIED") {
|
|
t.Fatalf("继承监听器子进程失败：%s, %v", output, err)
|
|
}
|
|
})
|
|
}
|
|
}
|