89d7b77022
Also renaming vars that collided with package-level vars and using consts for fixed values. Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
105 lines
3.1 KiB
Go
105 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/docker/docker/testutil"
|
|
"github.com/docker/docker/testutil/registry"
|
|
"gotest.tools/v3/assert"
|
|
)
|
|
|
|
// unescapeBackslashSemicolonParens unescapes \;()
|
|
func unescapeBackslashSemicolonParens(s string) string {
|
|
re := regexp.MustCompile(`\\;`)
|
|
ret := re.ReplaceAll([]byte(s), []byte(";"))
|
|
|
|
re = regexp.MustCompile(`\\\(`)
|
|
ret = re.ReplaceAll(ret, []byte("("))
|
|
|
|
re = regexp.MustCompile(`\\\)`)
|
|
ret = re.ReplaceAll(ret, []byte(")"))
|
|
|
|
re = regexp.MustCompile(`\\\\`)
|
|
ret = re.ReplaceAll(ret, []byte(`\`))
|
|
|
|
return string(ret)
|
|
}
|
|
|
|
func regexpCheckUA(c *testing.T, ua string) {
|
|
re := regexp.MustCompile("(?P<dockerUA>.+) UpstreamClient(?P<upstreamUA>.+)")
|
|
substrArr := re.FindStringSubmatch(ua)
|
|
|
|
assert.Equal(c, len(substrArr), 3, "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)
|
|
assert.Assert(c, bMatchDockerUA, "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)
|
|
assert.Assert(c, bMatchUpstreamUA, "(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 *testing.T) {
|
|
ctx := testutil.GetContext(c)
|
|
var ua string
|
|
|
|
reg, err := registry.NewMock(c)
|
|
assert.NilError(c, err)
|
|
defer reg.Close()
|
|
|
|
registerUserAgentHandler(reg, &ua)
|
|
imgRepo := reg.URL() + "/busybox"
|
|
|
|
s.d.StartWithBusybox(ctx, c, "--insecure-registry", reg.URL())
|
|
|
|
tmp, err := os.MkdirTemp("", "integration-cli-")
|
|
assert.NilError(c, err)
|
|
defer os.RemoveAll(tmp)
|
|
|
|
dockerfile, err := makefile(tmp, "FROM "+imgRepo)
|
|
assert.NilError(c, err, "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", imgRepo)
|
|
regexpCheckUA(c, ua)
|
|
|
|
s.d.Cmd("tag", "busybox", imgRepo)
|
|
s.d.Cmd("push", imgRepo)
|
|
regexpCheckUA(c, ua)
|
|
}
|