100 lines
1.7 KiB
Go
100 lines
1.7 KiB
Go
|
package nsenter
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
"os/signal"
|
||
|
|
"strings"
|
||
|
|
"syscall"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
type pid struct {
|
||
|
|
Pid int `json:"Pid"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNsenterAlivePid(t *testing.T) {
|
||
|
|
args := []string{"nsenter-exec"}
|
||
|
|
r, w, err := os.Pipe()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to create pipe %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
cmd := &exec.Cmd{
|
||
|
|
Path: os.Args[0],
|
||
|
|
Args: args,
|
||
|
|
ExtraFiles: []*os.File{w},
|
||
|
|
Env: []string{fmt.Sprintf("_LIBCONTAINER_INITPID=%d", os.Getpid())},
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := cmd.Start(); err != nil {
|
||
|
|
t.Fatalf("nsenter failed to start %v", err)
|
||
|
|
}
|
||
|
|
w.Close()
|
||
|
|
|
||
|
|
decoder := json.NewDecoder(r)
|
||
|
|
var pid *pid
|
||
|
|
|
||
|
|
if err := decoder.Decode(&pid); err != nil {
|
||
|
|
t.Fatalf("%v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := cmd.Wait(); err != nil {
|
||
|
|
t.Fatalf("nsenter exits with a non-zero exit status")
|
||
|
|
}
|
||
|
|
p, err := os.FindProcess(pid.Pid)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("%v", err)
|
||
|
|
}
|
||
|
|
p.Wait()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNsenterInvalidPid(t *testing.T) {
|
||
|
|
args := []string{"nsenter-exec"}
|
||
|
|
|
||
|
|
cmd := &exec.Cmd{
|
||
|
|
Path: os.Args[0],
|
||
|
|
Args: args,
|
||
|
|
Env: []string{"_LIBCONTAINER_INITPID=-1"},
|
||
|
|
}
|
||
|
|
|
||
|
|
err := cmd.Run()
|
||
|
|
if err == nil {
|
||
|
|
t.Fatal("nsenter exits with a zero exit status")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNsenterDeadPid(t *testing.T) {
|
||
|
|
|
||
|
|
c := make(chan os.Signal)
|
||
|
|
signal.Notify(c, syscall.SIGCHLD)
|
||
|
|
dead_cmd := exec.Command("true")
|
||
|
|
if err := dead_cmd.Start(); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer dead_cmd.Wait()
|
||
|
|
<-c // dead_cmd is zombie
|
||
|
|
|
||
|
|
args := []string{"nsenter-exec"}
|
||
|
|
|
||
|
|
cmd := &exec.Cmd{
|
||
|
|
Path: os.Args[0],
|
||
|
|
Args: args,
|
||
|
|
Env: []string{fmt.Sprintf("_LIBCONTAINER_INITPID=%d", dead_cmd.Process.Pid)},
|
||
|
|
}
|
||
|
|
|
||
|
|
err := cmd.Run()
|
||
|
|
if err == nil {
|
||
|
|
t.Fatal("nsenter exits with a zero exit status")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
if strings.HasPrefix(os.Args[0], "nsenter-") {
|
||
|
|
os.Exit(0)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|