| package integration |
| |
| import ( |
| "bytes" |
| "io/ioutil" |
| "os" |
| "path/filepath" |
| "strconv" |
| "strings" |
| "syscall" |
| "testing" |
| |
| "github.com/opencontainers/runc/libcontainer" |
| "github.com/opencontainers/runc/libcontainer/cgroups/systemd" |
| "github.com/opencontainers/runc/libcontainer/configs" |
| ) |
| |
| func TestExecPS(t *testing.T) { |
| testExecPS(t, false) |
| } |
| |
| func TestUsernsExecPS(t *testing.T) { |
| if _, err := os.Stat("/proc/self/ns/user"); os.IsNotExist(err) { |
| t.Skip("userns is unsupported") |
| } |
| testExecPS(t, true) |
| } |
| |
| func testExecPS(t *testing.T, userns bool) { |
| if testing.Short() { |
| return |
| } |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| config := newTemplateConfig(rootfs) |
| if userns { |
| config.UidMappings = []configs.IDMap{{0, 0, 1000}} |
| config.GidMappings = []configs.IDMap{{0, 0, 1000}} |
| config.Namespaces = append(config.Namespaces, configs.Namespace{Type: configs.NEWUSER}) |
| } |
| |
| buffers, exitCode, err := runContainer(config, "", "ps") |
| if err != nil { |
| t.Fatalf("%s: %s", buffers, err) |
| } |
| if exitCode != 0 { |
| t.Fatalf("exit code not 0. code %d stderr %q", exitCode, buffers.Stderr) |
| } |
| lines := strings.Split(buffers.Stdout.String(), "\n") |
| if len(lines) < 2 { |
| t.Fatalf("more than one process running for output %q", buffers.Stdout.String()) |
| } |
| expected := `1 root ps` |
| actual := strings.Trim(lines[1], "\n ") |
| if actual != expected { |
| t.Fatalf("expected output %q but received %q", expected, actual) |
| } |
| } |
| |
| func TestIPCPrivate(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| l, err := os.Readlink("/proc/1/ns/ipc") |
| ok(t, err) |
| |
| config := newTemplateConfig(rootfs) |
| buffers, exitCode, err := runContainer(config, "", "readlink", "/proc/self/ns/ipc") |
| ok(t, err) |
| |
| if exitCode != 0 { |
| t.Fatalf("exit code not 0. code %d stderr %q", exitCode, buffers.Stderr) |
| } |
| |
| if actual := strings.Trim(buffers.Stdout.String(), "\n"); actual == l { |
| t.Fatalf("ipc link should be private to the container but equals host %q %q", actual, l) |
| } |
| } |
| |
| func TestIPCHost(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| l, err := os.Readlink("/proc/1/ns/ipc") |
| ok(t, err) |
| |
| config := newTemplateConfig(rootfs) |
| config.Namespaces.Remove(configs.NEWIPC) |
| buffers, exitCode, err := runContainer(config, "", "readlink", "/proc/self/ns/ipc") |
| ok(t, err) |
| |
| if exitCode != 0 { |
| t.Fatalf("exit code not 0. code %d stderr %q", exitCode, buffers.Stderr) |
| } |
| |
| if actual := strings.Trim(buffers.Stdout.String(), "\n"); actual != l { |
| t.Fatalf("ipc link not equal to host link %q %q", actual, l) |
| } |
| } |
| |
| func TestIPCJoinPath(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| l, err := os.Readlink("/proc/1/ns/ipc") |
| ok(t, err) |
| |
| config := newTemplateConfig(rootfs) |
| config.Namespaces.Add(configs.NEWIPC, "/proc/1/ns/ipc") |
| |
| buffers, exitCode, err := runContainer(config, "", "readlink", "/proc/self/ns/ipc") |
| ok(t, err) |
| |
| if exitCode != 0 { |
| t.Fatalf("exit code not 0. code %d stderr %q", exitCode, buffers.Stderr) |
| } |
| |
| if actual := strings.Trim(buffers.Stdout.String(), "\n"); actual != l { |
| t.Fatalf("ipc link not equal to host link %q %q", actual, l) |
| } |
| } |
| |
| func TestIPCBadPath(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| config.Namespaces.Add(configs.NEWIPC, "/proc/1/ns/ipcc") |
| |
| _, _, err = runContainer(config, "", "true") |
| if err == nil { |
| t.Fatal("container succeeded with bad ipc path") |
| } |
| } |
| |
| func TestRlimit(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| out, _, err := runContainer(config, "", "/bin/sh", "-c", "ulimit -n") |
| ok(t, err) |
| if limit := strings.TrimSpace(out.Stdout.String()); limit != "1025" { |
| t.Fatalf("expected rlimit to be 1025, got %s", limit) |
| } |
| } |
| |
| func newTestRoot() (string, error) { |
| dir, err := ioutil.TempDir("", "libcontainer") |
| if err != nil { |
| return "", err |
| } |
| if err := os.MkdirAll(dir, 0700); err != nil { |
| return "", err |
| } |
| return dir, nil |
| } |
| |
| func TestEnter(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| root, err := newTestRoot() |
| ok(t, err) |
| defer os.RemoveAll(root) |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| |
| container, err := factory.Create("test", config) |
| ok(t, err) |
| defer container.Destroy() |
| |
| // Execute a first process in the container |
| stdinR, stdinW, err := os.Pipe() |
| ok(t, err) |
| |
| var stdout, stdout2 bytes.Buffer |
| |
| pconfig := libcontainer.Process{ |
| Args: []string{"sh", "-c", "cat && readlink /proc/self/ns/pid"}, |
| Env: standardEnvironment, |
| Stdin: stdinR, |
| Stdout: &stdout, |
| } |
| err = container.Start(&pconfig) |
| stdinR.Close() |
| defer stdinW.Close() |
| ok(t, err) |
| pid, err := pconfig.Pid() |
| ok(t, err) |
| |
| // Execute another process in the container |
| stdinR2, stdinW2, err := os.Pipe() |
| ok(t, err) |
| pconfig2 := libcontainer.Process{ |
| Env: standardEnvironment, |
| } |
| pconfig2.Args = []string{"sh", "-c", "cat && readlink /proc/self/ns/pid"} |
| pconfig2.Stdin = stdinR2 |
| pconfig2.Stdout = &stdout2 |
| |
| err = container.Start(&pconfig2) |
| stdinR2.Close() |
| defer stdinW2.Close() |
| ok(t, err) |
| |
| pid2, err := pconfig2.Pid() |
| ok(t, err) |
| |
| processes, err := container.Processes() |
| ok(t, err) |
| |
| n := 0 |
| for i := range processes { |
| if processes[i] == pid || processes[i] == pid2 { |
| n++ |
| } |
| } |
| if n != 2 { |
| t.Fatal("unexpected number of processes", processes, pid, pid2) |
| } |
| |
| // Wait processes |
| stdinW2.Close() |
| waitProcess(&pconfig2, t) |
| |
| stdinW.Close() |
| waitProcess(&pconfig, t) |
| |
| // Check that both processes live in the same pidns |
| pidns := string(stdout.Bytes()) |
| ok(t, err) |
| |
| pidns2 := string(stdout2.Bytes()) |
| ok(t, err) |
| |
| if pidns != pidns2 { |
| t.Fatal("The second process isn't in the required pid namespace", pidns, pidns2) |
| } |
| } |
| |
| func TestProcessEnv(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| root, err := newTestRoot() |
| ok(t, err) |
| defer os.RemoveAll(root) |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| |
| container, err := factory.Create("test", config) |
| ok(t, err) |
| defer container.Destroy() |
| |
| var stdout bytes.Buffer |
| pconfig := libcontainer.Process{ |
| Args: []string{"sh", "-c", "env"}, |
| Env: []string{ |
| "PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin", |
| "HOSTNAME=integration", |
| "TERM=xterm", |
| "FOO=BAR", |
| }, |
| Stdin: nil, |
| Stdout: &stdout, |
| } |
| err = container.Start(&pconfig) |
| ok(t, err) |
| |
| // Wait for process |
| waitProcess(&pconfig, t) |
| |
| outputEnv := string(stdout.Bytes()) |
| |
| // Check that the environment has the key/value pair we added |
| if !strings.Contains(outputEnv, "FOO=BAR") { |
| t.Fatal("Environment doesn't have the expected FOO=BAR key/value pair: ", outputEnv) |
| } |
| |
| // Make sure that HOME is set |
| if !strings.Contains(outputEnv, "HOME=/root") { |
| t.Fatal("Environment doesn't have HOME set: ", outputEnv) |
| } |
| } |
| |
| func TestProcessCaps(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| root, err := newTestRoot() |
| ok(t, err) |
| defer os.RemoveAll(root) |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| |
| container, err := factory.Create("test", config) |
| ok(t, err) |
| defer container.Destroy() |
| |
| processCaps := append(config.Capabilities, "NET_ADMIN") |
| |
| var stdout bytes.Buffer |
| pconfig := libcontainer.Process{ |
| Args: []string{"sh", "-c", "cat /proc/self/status"}, |
| Env: standardEnvironment, |
| Capabilities: processCaps, |
| Stdin: nil, |
| Stdout: &stdout, |
| } |
| err = container.Start(&pconfig) |
| ok(t, err) |
| |
| // Wait for process |
| waitProcess(&pconfig, t) |
| |
| outputStatus := string(stdout.Bytes()) |
| |
| lines := strings.Split(outputStatus, "\n") |
| |
| effectiveCapsLine := "" |
| for _, l := range lines { |
| line := strings.TrimSpace(l) |
| if strings.Contains(line, "CapEff:") { |
| effectiveCapsLine = line |
| break |
| } |
| } |
| |
| if effectiveCapsLine == "" { |
| t.Fatal("Couldn't find effective caps: ", outputStatus) |
| } |
| |
| parts := strings.Split(effectiveCapsLine, ":") |
| effectiveCapsStr := strings.TrimSpace(parts[1]) |
| |
| effectiveCaps, err := strconv.ParseUint(effectiveCapsStr, 16, 64) |
| if err != nil { |
| t.Fatal("Could not parse effective caps", err) |
| } |
| |
| var netAdminMask uint64 |
| var netAdminBit uint |
| netAdminBit = 12 // from capability.h |
| netAdminMask = 1 << netAdminBit |
| if effectiveCaps&netAdminMask != netAdminMask { |
| t.Fatal("CAP_NET_ADMIN is not set as expected") |
| } |
| } |
| |
| func TestAdditionalGroups(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| root, err := newTestRoot() |
| ok(t, err) |
| defer os.RemoveAll(root) |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| config.AdditionalGroups = []string{"plugdev", "audio"} |
| |
| factory, err := libcontainer.New(root, libcontainer.Cgroupfs) |
| ok(t, err) |
| |
| container, err := factory.Create("test", config) |
| ok(t, err) |
| defer container.Destroy() |
| |
| var stdout bytes.Buffer |
| pconfig := libcontainer.Process{ |
| Args: []string{"sh", "-c", "id", "-Gn"}, |
| Env: standardEnvironment, |
| Stdin: nil, |
| Stdout: &stdout, |
| } |
| err = container.Start(&pconfig) |
| ok(t, err) |
| |
| // Wait for process |
| waitProcess(&pconfig, t) |
| |
| outputGroups := string(stdout.Bytes()) |
| |
| // Check that the groups output has the groups that we specified |
| if !strings.Contains(outputGroups, "audio") { |
| t.Fatalf("Listed groups do not contain the audio group as expected: %v", outputGroups) |
| } |
| |
| if !strings.Contains(outputGroups, "plugdev") { |
| t.Fatalf("Listed groups do not contain the plugdev group as expected: %v", outputGroups) |
| } |
| } |
| |
| func TestFreeze(t *testing.T) { |
| testFreeze(t, false) |
| } |
| |
| func TestSystemdFreeze(t *testing.T) { |
| if !systemd.UseSystemd() { |
| t.Skip("Systemd is unsupported") |
| } |
| testFreeze(t, true) |
| } |
| |
| func testFreeze(t *testing.T, systemd bool) { |
| if testing.Short() { |
| return |
| } |
| root, err := newTestRoot() |
| ok(t, err) |
| defer os.RemoveAll(root) |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| f := factory |
| if systemd { |
| f = systemdFactory |
| } |
| |
| container, err := f.Create("test", config) |
| ok(t, err) |
| defer container.Destroy() |
| |
| stdinR, stdinW, err := os.Pipe() |
| ok(t, err) |
| |
| pconfig := &libcontainer.Process{ |
| Args: []string{"cat"}, |
| Env: standardEnvironment, |
| Stdin: stdinR, |
| } |
| err = container.Start(pconfig) |
| stdinR.Close() |
| defer stdinW.Close() |
| ok(t, err) |
| |
| err = container.Pause() |
| ok(t, err) |
| state, err := container.Status() |
| ok(t, err) |
| err = container.Resume() |
| ok(t, err) |
| if state != libcontainer.Paused { |
| t.Fatal("Unexpected state: ", state) |
| } |
| |
| stdinW.Close() |
| waitProcess(pconfig, t) |
| } |
| |
| func TestCpuShares(t *testing.T) { |
| testCpuShares(t, false) |
| } |
| |
| func TestCpuSharesSystemd(t *testing.T) { |
| if !systemd.UseSystemd() { |
| t.Skip("Systemd is unsupported") |
| } |
| testCpuShares(t, true) |
| } |
| |
| func testCpuShares(t *testing.T, systemd bool) { |
| if testing.Short() { |
| return |
| } |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| if systemd { |
| config.Cgroups.Slice = "system.slice" |
| } |
| config.Cgroups.CpuShares = 1 |
| |
| _, _, err = runContainer(config, "", "ps") |
| if err == nil { |
| t.Fatalf("runContainer should failed with invalid CpuShares") |
| } |
| } |
| |
| func TestRunWithKernelMemory(t *testing.T) { |
| testRunWithKernelMemory(t, false) |
| } |
| |
| func TestRunWithKernelMemorySystemd(t *testing.T) { |
| if !systemd.UseSystemd() { |
| t.Skip("Systemd is unsupported") |
| } |
| testRunWithKernelMemory(t, true) |
| } |
| |
| func testRunWithKernelMemory(t *testing.T, systemd bool) { |
| if testing.Short() { |
| return |
| } |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| if systemd { |
| config.Cgroups.Slice = "system.slice" |
| } |
| config.Cgroups.KernelMemory = 52428800 |
| |
| _, _, err = runContainer(config, "", "ps") |
| if err != nil { |
| t.Fatalf("runContainer failed with kernel memory limit: %v", err) |
| } |
| } |
| |
| func TestContainerState(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| root, err := newTestRoot() |
| if err != nil { |
| t.Fatal(err) |
| } |
| defer os.RemoveAll(root) |
| |
| rootfs, err := newRootfs() |
| if err != nil { |
| t.Fatal(err) |
| } |
| defer remove(rootfs) |
| |
| l, err := os.Readlink("/proc/1/ns/ipc") |
| if err != nil { |
| t.Fatal(err) |
| } |
| |
| config := newTemplateConfig(rootfs) |
| config.Namespaces = configs.Namespaces([]configs.Namespace{ |
| {Type: configs.NEWNS}, |
| {Type: configs.NEWUTS}, |
| // host for IPC |
| //{Type: configs.NEWIPC}, |
| {Type: configs.NEWPID}, |
| {Type: configs.NEWNET}, |
| }) |
| |
| container, err := factory.Create("test", config) |
| if err != nil { |
| t.Fatal(err) |
| } |
| defer container.Destroy() |
| |
| stdinR, stdinW, err := os.Pipe() |
| if err != nil { |
| t.Fatal(err) |
| } |
| p := &libcontainer.Process{ |
| Args: []string{"cat"}, |
| Env: standardEnvironment, |
| Stdin: stdinR, |
| } |
| err = container.Start(p) |
| if err != nil { |
| t.Fatal(err) |
| } |
| stdinR.Close() |
| defer stdinW.Close() |
| |
| st, err := container.State() |
| if err != nil { |
| t.Fatal(err) |
| } |
| |
| l1, err := os.Readlink(st.NamespacePaths[configs.NEWIPC]) |
| if err != nil { |
| t.Fatal(err) |
| } |
| if l1 != l { |
| t.Fatal("Container using non-host ipc namespace") |
| } |
| stdinW.Close() |
| waitProcess(p, t) |
| } |
| |
| func TestPassExtraFiles(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| |
| rootfs, err := newRootfs() |
| if err != nil { |
| t.Fatal(err) |
| } |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| |
| container, err := factory.Create("test", config) |
| if err != nil { |
| t.Fatal(err) |
| } |
| defer container.Destroy() |
| |
| var stdout bytes.Buffer |
| pipeout1, pipein1, err := os.Pipe() |
| pipeout2, pipein2, err := os.Pipe() |
| process := libcontainer.Process{ |
| Args: []string{"sh", "-c", "cd /proc/$$/fd; echo -n *; echo -n 1 >3; echo -n 2 >4"}, |
| Env: []string{"PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"}, |
| ExtraFiles: []*os.File{pipein1, pipein2}, |
| Stdin: nil, |
| Stdout: &stdout, |
| } |
| err = container.Start(&process) |
| if err != nil { |
| t.Fatal(err) |
| } |
| |
| waitProcess(&process, t) |
| |
| out := string(stdout.Bytes()) |
| // fd 5 is the directory handle for /proc/$$/fd |
| if out != "0 1 2 3 4 5" { |
| t.Fatalf("expected to have the file descriptors '0 1 2 3 4 5' passed to init, got '%s'", out) |
| } |
| var buf = []byte{0} |
| _, err = pipeout1.Read(buf) |
| if err != nil { |
| t.Fatal(err) |
| } |
| out1 := string(buf) |
| if out1 != "1" { |
| t.Fatalf("expected first pipe to receive '1', got '%s'", out1) |
| } |
| |
| _, err = pipeout2.Read(buf) |
| if err != nil { |
| t.Fatal(err) |
| } |
| out2 := string(buf) |
| if out2 != "2" { |
| t.Fatalf("expected second pipe to receive '2', got '%s'", out2) |
| } |
| } |
| |
| func TestMountCmds(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| root, err := newTestRoot() |
| if err != nil { |
| t.Fatal(err) |
| } |
| defer os.RemoveAll(root) |
| |
| rootfs, err := newRootfs() |
| if err != nil { |
| t.Fatal(err) |
| } |
| defer remove(rootfs) |
| |
| tmpDir, err := ioutil.TempDir("", "tmpdir") |
| if err != nil { |
| t.Fatal(err) |
| } |
| defer os.RemoveAll(tmpDir) |
| |
| config := newTemplateConfig(rootfs) |
| config.Mounts = append(config.Mounts, &configs.Mount{ |
| Source: tmpDir, |
| Destination: "/tmp", |
| Device: "bind", |
| Flags: syscall.MS_BIND | syscall.MS_REC, |
| PremountCmds: []configs.Command{ |
| {Path: "touch", Args: []string{filepath.Join(tmpDir, "hello")}}, |
| {Path: "touch", Args: []string{filepath.Join(tmpDir, "world")}}, |
| }, |
| PostmountCmds: []configs.Command{ |
| {Path: "cp", Args: []string{filepath.Join(rootfs, "tmp", "hello"), filepath.Join(rootfs, "tmp", "hello-backup")}}, |
| {Path: "cp", Args: []string{filepath.Join(rootfs, "tmp", "world"), filepath.Join(rootfs, "tmp", "world-backup")}}, |
| }, |
| }) |
| |
| container, err := factory.Create("test", config) |
| if err != nil { |
| t.Fatal(err) |
| } |
| defer container.Destroy() |
| |
| pconfig := libcontainer.Process{ |
| Args: []string{"sh", "-c", "env"}, |
| Env: standardEnvironment, |
| } |
| err = container.Start(&pconfig) |
| if err != nil { |
| t.Fatal(err) |
| } |
| |
| // Wait for process |
| waitProcess(&pconfig, t) |
| |
| entries, err := ioutil.ReadDir(tmpDir) |
| if err != nil { |
| t.Fatal(err) |
| } |
| expected := []string{"hello", "hello-backup", "world", "world-backup"} |
| for i, e := range entries { |
| if e.Name() != expected[i] { |
| t.Errorf("Got(%s), expect %s", e.Name(), expected[i]) |
| } |
| } |
| } |
| |
| func TestSysctl(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| root, err := newTestRoot() |
| ok(t, err) |
| defer os.RemoveAll(root) |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| config.Sysctl = map[string]string{ |
| "kernel.shmmni": "8192", |
| } |
| |
| container, err := factory.Create("test", config) |
| ok(t, err) |
| defer container.Destroy() |
| |
| var stdout bytes.Buffer |
| pconfig := libcontainer.Process{ |
| Args: []string{"sh", "-c", "cat /proc/sys/kernel/shmmni"}, |
| Env: standardEnvironment, |
| Stdin: nil, |
| Stdout: &stdout, |
| } |
| err = container.Start(&pconfig) |
| ok(t, err) |
| |
| // Wait for process |
| waitProcess(&pconfig, t) |
| |
| shmmniOutput := strings.TrimSpace(string(stdout.Bytes())) |
| if shmmniOutput != "8192" { |
| t.Fatalf("kernel.shmmni property expected to be 8192, but is %s", shmmniOutput) |
| } |
| } |
| |
| func TestMountCgroupRO(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| config := newTemplateConfig(rootfs) |
| |
| config.Mounts = append(config.Mounts, &configs.Mount{ |
| Destination: "/sys/fs/cgroup", |
| Device: "cgroup", |
| Flags: defaultMountFlags | syscall.MS_RDONLY, |
| }) |
| |
| buffers, exitCode, err := runContainer(config, "", "mount") |
| if err != nil { |
| t.Fatalf("%s: %s", buffers, err) |
| } |
| if exitCode != 0 { |
| t.Fatalf("exit code not 0. code %d stderr %q", exitCode, buffers.Stderr) |
| } |
| mountInfo := buffers.Stdout.String() |
| lines := strings.Split(mountInfo, "\n") |
| for _, l := range lines { |
| if strings.HasPrefix(l, "tmpfs on /sys/fs/cgroup") { |
| if !strings.Contains(l, "ro") || |
| !strings.Contains(l, "nosuid") || |
| !strings.Contains(l, "nodev") || |
| !strings.Contains(l, "noexec") { |
| t.Fatalf("Mode expected to contain 'ro,nosuid,nodev,noexec': %s", l) |
| } |
| if !strings.Contains(l, "mode=755") { |
| t.Fatalf("Mode expected to contain 'mode=755': %s", l) |
| } |
| continue |
| } |
| if !strings.HasPrefix(l, "cgroup") { |
| continue |
| } |
| if !strings.Contains(l, "ro") || |
| !strings.Contains(l, "nosuid") || |
| !strings.Contains(l, "nodev") || |
| !strings.Contains(l, "noexec") { |
| t.Fatalf("Mode expected to contain 'ro,nosuid,nodev,noexec': %s", l) |
| } |
| } |
| } |
| |
| func TestMountCgroupRW(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| config := newTemplateConfig(rootfs) |
| |
| config.Mounts = append(config.Mounts, &configs.Mount{ |
| Destination: "/sys/fs/cgroup", |
| Device: "cgroup", |
| Flags: defaultMountFlags, |
| }) |
| |
| buffers, exitCode, err := runContainer(config, "", "mount") |
| if err != nil { |
| t.Fatalf("%s: %s", buffers, err) |
| } |
| if exitCode != 0 { |
| t.Fatalf("exit code not 0. code %d stderr %q", exitCode, buffers.Stderr) |
| } |
| mountInfo := buffers.Stdout.String() |
| lines := strings.Split(mountInfo, "\n") |
| for _, l := range lines { |
| if strings.HasPrefix(l, "tmpfs on /sys/fs/cgroup") { |
| if !strings.Contains(l, "rw") || |
| !strings.Contains(l, "nosuid") || |
| !strings.Contains(l, "nodev") || |
| !strings.Contains(l, "noexec") { |
| t.Fatalf("Mode expected to contain 'rw,nosuid,nodev,noexec': %s", l) |
| } |
| if !strings.Contains(l, "mode=755") { |
| t.Fatalf("Mode expected to contain 'mode=755': %s", l) |
| } |
| continue |
| } |
| if !strings.HasPrefix(l, "cgroup") { |
| continue |
| } |
| if !strings.Contains(l, "rw") || |
| !strings.Contains(l, "nosuid") || |
| !strings.Contains(l, "nodev") || |
| !strings.Contains(l, "noexec") { |
| t.Fatalf("Mode expected to contain 'rw,nosuid,nodev,noexec': %s", l) |
| } |
| } |
| } |
| |
| func TestOomScoreAdj(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| root, err := newTestRoot() |
| ok(t, err) |
| defer os.RemoveAll(root) |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| config.OomScoreAdj = 200 |
| |
| factory, err := libcontainer.New(root, libcontainer.Cgroupfs) |
| ok(t, err) |
| |
| container, err := factory.Create("test", config) |
| ok(t, err) |
| defer container.Destroy() |
| |
| var stdout bytes.Buffer |
| pconfig := libcontainer.Process{ |
| Args: []string{"sh", "-c", "cat /proc/self/oom_score_adj"}, |
| Env: standardEnvironment, |
| Stdin: nil, |
| Stdout: &stdout, |
| } |
| err = container.Start(&pconfig) |
| ok(t, err) |
| |
| // Wait for process |
| waitProcess(&pconfig, t) |
| outputOomScoreAdj := strings.TrimSpace(string(stdout.Bytes())) |
| |
| // Check that the oom_score_adj matches the value that was set as part of config. |
| if outputOomScoreAdj != strconv.Itoa(config.OomScoreAdj) { |
| t.Fatalf("Expected oom_score_adj %d; got %q", config.OomScoreAdj, outputOomScoreAdj) |
| } |
| } |
| |
| func TestHook(t *testing.T) { |
| if testing.Short() { |
| return |
| } |
| root, err := newTestRoot() |
| ok(t, err) |
| defer os.RemoveAll(root) |
| |
| rootfs, err := newRootfs() |
| ok(t, err) |
| defer remove(rootfs) |
| |
| config := newTemplateConfig(rootfs) |
| config.Hooks = &configs.Hooks{ |
| Prestart: []configs.Hook{ |
| configs.NewFunctionHook(func(s configs.HookState) error { |
| f, err := os.Create(filepath.Join(s.Root, "test")) |
| if err != nil { |
| return err |
| } |
| return f.Close() |
| }), |
| }, |
| Poststop: []configs.Hook{ |
| configs.NewFunctionHook(func(s configs.HookState) error { |
| return os.RemoveAll(filepath.Join(s.Root, "test")) |
| }), |
| }, |
| } |
| container, err := factory.Create("test", config) |
| ok(t, err) |
| |
| var stdout bytes.Buffer |
| pconfig := libcontainer.Process{ |
| Args: []string{"sh", "-c", "ls /test"}, |
| Env: standardEnvironment, |
| Stdin: nil, |
| Stdout: &stdout, |
| } |
| err = container.Start(&pconfig) |
| ok(t, err) |
| |
| // Wait for process |
| waitProcess(&pconfig, t) |
| |
| outputLs := string(stdout.Bytes()) |
| |
| // Check that the ls output has the expected file touched by the prestart hook |
| if !strings.Contains(outputLs, "/test") { |
| container.Destroy() |
| t.Fatalf("ls output doesn't have the expected file: %s", outputLs) |
| } |
| |
| if err := container.Destroy(); err != nil { |
| t.Fatalf("container destory %s", err) |
| } |
| fi, err := os.Stat(filepath.Join(rootfs, "test")) |
| if err == nil || !os.IsNotExist(err) { |
| t.Fatalf("expected file to not exist, got %s", fi.Name()) |
| } |
| } |