mirror of
https://github.com/moby/moby.git
synced 2026-01-11 18:51:37 +00:00
d3e45f8743
This package was originally internal, but was moved out when BuildKit used it for its integration tests. That's no longer the case, so we can make it internal again. Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
106 lines
3.1 KiB
Go
106 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/moby/moby/v2/internal/testutil"
|
|
"github.com/moby/moby/v2/internal/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(t *testing.T, ua string) {
|
|
re := regexp.MustCompile("(?P<dockerUA>.+) UpstreamClient(?P<upstreamUA>.+)")
|
|
substrArr := re.FindStringSubmatch(ua)
|
|
|
|
assert.Equal(t, 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(t, 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(t, 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(http.StatusNotFound)
|
|
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)
|
|
}
|