github.com/morganxf/moby@v1.13.1/integration-cli/fixtures_linux_daemon.go (about)

     1  package main
     2  
     3  import (
     4  	"fmt"
     5  	"io/ioutil"
     6  	"os"
     7  	"os/exec"
     8  	"path/filepath"
     9  	"runtime"
    10  	"strings"
    11  	"sync"
    12  	"testing"
    13  
    14  	"github.com/docker/docker/integration-cli/fixtures/load"
    15  	"github.com/docker/docker/pkg/integration/checker"
    16  	"github.com/go-check/check"
    17  )
    18  
    19  func ensureFrozenImagesLinux(t *testing.T) {
    20  	images := []string{"busybox:latest", "hello-world:frozen", "debian:jessie"}
    21  	err := load.FrozenImagesLinux(dockerBinary, images...)
    22  	if err != nil {
    23  		t.Log(dockerCmdWithError("images"))
    24  		t.Fatalf("%+v", err)
    25  	}
    26  	for _, img := range images {
    27  		protectedImages[img] = struct{}{}
    28  	}
    29  }
    30  
    31  var ensureSyscallTestOnce sync.Once
    32  
    33  func ensureSyscallTest(c *check.C) {
    34  	var doIt bool
    35  	ensureSyscallTestOnce.Do(func() {
    36  		doIt = true
    37  	})
    38  	if !doIt {
    39  		return
    40  	}
    41  	protectedImages["syscall-test:latest"] = struct{}{}
    42  
    43  	// if no match, must build in docker, which is significantly slower
    44  	// (slower mostly because of the vfs graphdriver)
    45  	if daemonPlatform != runtime.GOOS {
    46  		ensureSyscallTestBuild(c)
    47  		return
    48  	}
    49  
    50  	tmp, err := ioutil.TempDir("", "syscall-test-build")
    51  	c.Assert(err, checker.IsNil, check.Commentf("couldn't create temp dir"))
    52  	defer os.RemoveAll(tmp)
    53  
    54  	gcc, err := exec.LookPath("gcc")
    55  	c.Assert(err, checker.IsNil, check.Commentf("could not find gcc"))
    56  
    57  	tests := []string{"userns", "ns", "acct", "setuid", "setgid", "socket", "raw"}
    58  	for _, test := range tests {
    59  		out, err := exec.Command(gcc, "-g", "-Wall", "-static", fmt.Sprintf("../contrib/syscall-test/%s.c", test), "-o", fmt.Sprintf("%s/%s-test", tmp, test)).CombinedOutput()
    60  		c.Assert(err, checker.IsNil, check.Commentf(string(out)))
    61  	}
    62  
    63  	if runtime.GOOS == "linux" && runtime.GOARCH == "amd64" {
    64  		out, err := exec.Command(gcc, "-s", "-m32", "-nostdlib", "../contrib/syscall-test/exit32.s", "-o", tmp+"/"+"exit32-test").CombinedOutput()
    65  		c.Assert(err, checker.IsNil, check.Commentf(string(out)))
    66  	}
    67  
    68  	dockerFile := filepath.Join(tmp, "Dockerfile")
    69  	content := []byte(`
    70  	FROM debian:jessie
    71  	COPY . /usr/bin/
    72  	`)
    73  	err = ioutil.WriteFile(dockerFile, content, 600)
    74  	c.Assert(err, checker.IsNil)
    75  
    76  	var buildArgs []string
    77  	if arg := os.Getenv("DOCKER_BUILD_ARGS"); strings.TrimSpace(arg) != "" {
    78  		buildArgs = strings.Split(arg, " ")
    79  	}
    80  	buildArgs = append(buildArgs, []string{"-q", "-t", "syscall-test", tmp}...)
    81  	buildArgs = append([]string{"build"}, buildArgs...)
    82  	dockerCmd(c, buildArgs...)
    83  }
    84  
    85  func ensureSyscallTestBuild(c *check.C) {
    86  	err := load.FrozenImagesLinux(dockerBinary, "buildpack-deps:jessie")
    87  	c.Assert(err, checker.IsNil)
    88  
    89  	var buildArgs []string
    90  	if arg := os.Getenv("DOCKER_BUILD_ARGS"); strings.TrimSpace(arg) != "" {
    91  		buildArgs = strings.Split(arg, " ")
    92  	}
    93  	buildArgs = append(buildArgs, []string{"-q", "-t", "syscall-test", "../contrib/syscall-test"}...)
    94  	buildArgs = append([]string{"build"}, buildArgs...)
    95  	dockerCmd(c, buildArgs...)
    96  }
    97  
    98  func ensureNNPTest(c *check.C) {
    99  	protectedImages["nnp-test:latest"] = struct{}{}
   100  	if daemonPlatform != runtime.GOOS {
   101  		ensureNNPTestBuild(c)
   102  		return
   103  	}
   104  
   105  	tmp, err := ioutil.TempDir("", "docker-nnp-test")
   106  	c.Assert(err, checker.IsNil)
   107  
   108  	gcc, err := exec.LookPath("gcc")
   109  	c.Assert(err, checker.IsNil, check.Commentf("could not find gcc"))
   110  
   111  	out, err := exec.Command(gcc, "-g", "-Wall", "-static", "../contrib/nnp-test/nnp-test.c", "-o", filepath.Join(tmp, "nnp-test")).CombinedOutput()
   112  	c.Assert(err, checker.IsNil, check.Commentf(string(out)))
   113  
   114  	dockerfile := filepath.Join(tmp, "Dockerfile")
   115  	content := `
   116  	FROM debian:jessie
   117  	COPY . /usr/bin
   118  	RUN chmod +s /usr/bin/nnp-test
   119  	`
   120  	err = ioutil.WriteFile(dockerfile, []byte(content), 600)
   121  	c.Assert(err, checker.IsNil, check.Commentf("could not write Dockerfile for nnp-test image"))
   122  
   123  	var buildArgs []string
   124  	if arg := os.Getenv("DOCKER_BUILD_ARGS"); strings.TrimSpace(arg) != "" {
   125  		buildArgs = strings.Split(arg, " ")
   126  	}
   127  	buildArgs = append(buildArgs, []string{"-q", "-t", "nnp-test", tmp}...)
   128  	buildArgs = append([]string{"build"}, buildArgs...)
   129  	dockerCmd(c, buildArgs...)
   130  }
   131  
   132  func ensureNNPTestBuild(c *check.C) {
   133  	err := load.FrozenImagesLinux(dockerBinary, "buildpack-deps:jessie")
   134  	c.Assert(err, checker.IsNil)
   135  
   136  	var buildArgs []string
   137  	if arg := os.Getenv("DOCKER_BUILD_ARGS"); strings.TrimSpace(arg) != "" {
   138  		buildArgs = strings.Split(arg, " ")
   139  	}
   140  	buildArgs = append(buildArgs, []string{"-q", "-t", "npp-test", "../contrib/nnp-test"}...)
   141  	buildArgs = append([]string{"build"}, buildArgs...)
   142  	dockerCmd(c, buildArgs...)
   143  }