package main import ( "fmt" "io/ioutil" "net/http" "os" "regexp" "github.com/docker/docker/integration-cli/registry" "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")) } // registerUserAgentHandler registers a handler for the `/v2/*` endpoint. // Note that a 404 is returned to prevent the client to proceed. // We are only checking if the client sent a valid User Agent string along // with the request. func registerUserAgentHandler(reg *registry.Mock, result *string) { reg.RegisterHandler("/v2/", func(w http.ResponseWriter, r *http.Request) { w.WriteHeader(404) w.Write([]byte(`{"errors":[{"code": "UNSUPPORTED","message": "this is a mock registry"}]}`)) 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] UpstreamClientSTREAM-CLIENT([client UA]) func (s *DockerRegistrySuite) TestUserAgentPassThrough(c *check.C) { var ua string reg, err := registry.NewMock(c) defer reg.Close() c.Assert(err, check.IsNil) registerUserAgentHandler(reg, &ua) repoName := fmt.Sprintf("%s/busybox", reg.URL()) s.d.StartWithBusybox(c, "--insecure-registry", reg.URL()) tmp, err := ioutil.TempDir("", "integration-cli-") c.Assert(err, check.IsNil) defer os.RemoveAll(tmp) dockerfile, err := makefile(tmp, fmt.Sprintf("FROM %s", repoName)) c.Assert(err, check.IsNil, check.Commentf("Unable to create test dockerfile")) s.d.Cmd("build", "--file", dockerfile, tmp) regexpCheckUA(c, ua) s.d.Cmd("login", "-u", "richard", "-p", "testtest", reg.URL()) regexpCheckUA(c, ua) s.d.Cmd("pull", repoName) regexpCheckUA(c, ua) s.d.Cmd("tag", "busybox", repoName) s.d.Cmd("push", repoName) regexpCheckUA(c, ua) }