package main import ( "fmt" "net/http" "regexp" "github.com/go-check/check" ) // unescapeBackslashSemicolonParens unescapes \;() func unescapeBackslashSemicolonParens(s string) string { re := regexp.MustCompile(`\\;`) ret := re.ReplaceAll([]byte(s), []byte(";")) re = regexp.MustCompile(`\\\(`) ret = re.ReplaceAll([]byte(ret), []byte("(")) re = regexp.MustCompile(`\\\)`) ret = re.ReplaceAll([]byte(ret), []byte(")")) re = regexp.MustCompile(`\\\\`) ret = re.ReplaceAll([]byte(ret), []byte(`\`)) return string(ret) } func regexpCheckUA(c *check.C, ua string) { re := regexp.MustCompile("(?P.+) UpstreamClient(?P.+)") substrArr := re.FindStringSubmatch(ua) c.Assert(substrArr, check.HasLen, 3, check.Commentf("Expected 'UpstreamClient()' with upstream client UA")) dockerUA := substrArr[1] upstreamUAEscaped := substrArr[2] // check dockerUA looks correct reDockerUA := regexp.MustCompile("^docker/[0-9A-Za-z+]") bMatchDockerUA := reDockerUA.MatchString(dockerUA) c.Assert(bMatchDockerUA, check.Equals, true, check.Commentf("Docker Engine User-Agent malformed")) // check upstreamUA looks correct // Expecting something like: Docker-Client/1.11.0-dev (linux) upstreamUA := unescapeBackslashSemicolonParens(upstreamUAEscaped) reUpstreamUA := regexp.MustCompile("^\\(Docker-Client/[0-9A-Za-z+]") bMatchUpstreamUA := reUpstreamUA.MatchString(upstreamUA) c.Assert(bMatchUpstreamUA, check.Equals, true, check.Commentf("(Upstream) Docker Client User-Agent malformed")) } func registerUserAgentHandler(reg *testRegistry, result *string) { reg.registerHandler("/v2/", func(w http.ResponseWriter, r *http.Request) { w.WriteHeader(404) var ua string for k, v := range r.Header { if k == "User-Agent" { ua = v[0] } } *result = ua }) } // TestUserAgentPassThrough verifies that when an image is pulled from // a registry, the registry should see a User-Agent string of the form // [docker engine UA] UptreamClientSTREAM-CLIENT([client UA]) func (s *DockerRegistrySuite) TestUserAgentPassThrough(c *check.C) { var ( buildUA string pullUA string pushUA string loginUA string ) buildReg, err := newTestRegistry(c) c.Assert(err, check.IsNil) registerUserAgentHandler(buildReg, &buildUA) buildRepoName := fmt.Sprintf("%s/busybox", buildReg.hostport) pullReg, err := newTestRegistry(c) c.Assert(err, check.IsNil) registerUserAgentHandler(pullReg, &pullUA) pullRepoName := fmt.Sprintf("%s/busybox", pullReg.hostport) pushReg, err := newTestRegistry(c) c.Assert(err, check.IsNil) registerUserAgentHandler(pushReg, &pushUA) pushRepoName := fmt.Sprintf("%s/busybox", pushReg.hostport) loginReg, err := newTestRegistry(c) c.Assert(err, check.IsNil) registerUserAgentHandler(loginReg, &loginUA) err = s.d.Start( "--insecure-registry", buildReg.hostport, "--insecure-registry", pullReg.hostport, "--insecure-registry", pushReg.hostport, "--insecure-registry", loginReg.hostport, "--disable-legacy-registry=true") c.Assert(err, check.IsNil) dockerfileName, cleanup1, err := makefile(fmt.Sprintf("FROM %s", buildRepoName)) c.Assert(err, check.IsNil, check.Commentf("Unable to create test dockerfile")) defer cleanup1() s.d.Cmd("build", "--file", dockerfileName, ".") regexpCheckUA(c, buildUA) s.d.Cmd("login", "-u", "richard", "-p", "testtest", loginReg.hostport) regexpCheckUA(c, loginUA) s.d.Cmd("pull", pullRepoName) regexpCheckUA(c, pullUA) dockerfileName, cleanup2, err := makefile(`FROM scratch ENV foo bar`) c.Assert(err, check.IsNil, check.Commentf("Unable to create test dockerfile")) defer cleanup2() s.d.Cmd("build", "-t", pushRepoName, "--file", dockerfileName, ".") s.d.Cmd("push", pushRepoName) regexpCheckUA(c, pushUA) }