diff --git a/distribution/pull_v2.go b/distribution/pull_v2.go index 39bf782495..d1d5f5c8b8 100644 --- a/distribution/pull_v2.go +++ b/distribution/pull_v2.go @@ -708,29 +708,20 @@ func (p *v2Puller) pullManifestList(ctx context.Context, ref reference.Named, mf } logrus.Debugf("%s resolved to a manifestList object with %d entries; looking for a os/arch match", ref, len(mfstList.Manifests)) - var manifestDigest digest.Digest - // TODO @jhowardmsft LCOW Support: Need to remove the hard coding in LCOW mode. - lookingForOS := runtime.GOOS - if system.LCOWSupported() { - lookingForOS = "linux" - } - for _, manifestDescriptor := range mfstList.Manifests { - // TODO(aaronl): The manifest list spec supports optional - // "features" and "variant" fields. These are not yet used. - // Once they are, their values should be interpreted here. - if manifestDescriptor.Platform.Architecture == runtime.GOARCH && manifestDescriptor.Platform.OS == lookingForOS { - manifestDigest = manifestDescriptor.Digest - logrus.Debugf("found match for %s/%s with media type %s, digest %s", runtime.GOOS, runtime.GOARCH, manifestDescriptor.MediaType, manifestDigest.String()) - break - } - } - if manifestDigest == "" { + manifestMatches := filterManifests(mfstList.Manifests) + + if len(manifestMatches) == 0 { errMsg := fmt.Sprintf("no matching manifest for %s/%s in the manifest list entries", runtime.GOOS, runtime.GOARCH) logrus.Debugf(errMsg) return "", "", errors.New(errMsg) } + if len(manifestMatches) > 1 { + logrus.Debugf("found multiple matches in manifest list, choosing best match %s", manifestMatches[0].Digest.String()) + } + manifestDigest := manifestMatches[0].Digest + manSvc, err := p.repo.Manifests(ctx) if err != nil { return "", "", err diff --git a/distribution/pull_v2_unix.go b/distribution/pull_v2_unix.go index 45a7a0c150..60d8605fb8 100644 --- a/distribution/pull_v2_unix.go +++ b/distribution/pull_v2_unix.go @@ -3,11 +3,27 @@ package distribution import ( + "runtime" + "github.com/docker/distribution" "github.com/docker/distribution/context" + "github.com/docker/distribution/manifest/manifestlist" + "github.com/sirupsen/logrus" ) func (ld *v2LayerDescriptor) open(ctx context.Context) (distribution.ReadSeekCloser, error) { blobs := ld.repo.Blobs(ctx) return blobs.Open(ctx, ld.digest) } + +func filterManifests(manifests []manifestlist.ManifestDescriptor) []manifestlist.ManifestDescriptor { + var matches []manifestlist.ManifestDescriptor + for _, manifestDescriptor := range manifests { + if manifestDescriptor.Platform.Architecture == runtime.GOARCH && manifestDescriptor.Platform.OS == runtime.GOOS { + matches = append(matches, manifestDescriptor) + + logrus.Debugf("found match for %s/%s with media type %s, digest %s", runtime.GOOS, runtime.GOARCH, manifestDescriptor.MediaType, manifestDescriptor.Digest.String()) + } + } + return matches +} diff --git a/distribution/pull_v2_windows.go b/distribution/pull_v2_windows.go index e10070d53a..e94109484d 100644 --- a/distribution/pull_v2_windows.go +++ b/distribution/pull_v2_windows.go @@ -3,13 +3,19 @@ package distribution import ( + "fmt" "net/http" "os" + "runtime" + "sort" + "strings" "github.com/docker/distribution" "github.com/docker/distribution/context" + "github.com/docker/distribution/manifest/manifestlist" "github.com/docker/distribution/manifest/schema2" "github.com/docker/distribution/registry/client/transport" + "github.com/docker/docker/pkg/system" "github.com/sirupsen/logrus" ) @@ -55,3 +61,57 @@ func (ld *v2LayerDescriptor) open(ctx context.Context) (distribution.ReadSeekClo } return rsc, err } + +func filterManifests(manifests []manifestlist.ManifestDescriptor) []manifestlist.ManifestDescriptor { + version := system.GetOSVersion() + + // TODO @jhowardmsft LCOW Support: Need to remove the hard coding in LCOW mode. + lookingForOS := runtime.GOOS + osVersion := fmt.Sprintf("%d.%d.%d", version.MajorVersion, version.MinorVersion, version.Build) + if system.LCOWSupported() { + lookingForOS = "linux" + osVersion = "" + } + + var matches []manifestlist.ManifestDescriptor + for _, manifestDescriptor := range manifests { + if manifestDescriptor.Platform.Architecture == runtime.GOARCH && manifestDescriptor.Platform.OS == lookingForOS { + if !versionMatch(manifestDescriptor.Platform.OSVersion, osVersion) { + continue + } + matches = append(matches, manifestDescriptor) + + logrus.Debugf("found match for %s/%s with media type %s, digest %s", runtime.GOOS, runtime.GOARCH, manifestDescriptor.MediaType, manifestDescriptor.Digest.String()) + } + } + sort.Stable(manifestsByVersion(matches)) + return matches +} + +func versionMatch(actual, expected string) bool { + // Check whether actual and expected are equivalent, or whether + // expected is a version prefix of actual. + return actual == "" || expected == "" || actual == expected || strings.HasPrefix(actual, expected+".") +} + +type manifestsByVersion []manifestlist.ManifestDescriptor + +func (mbv manifestsByVersion) Less(i, j int) bool { + if mbv[i].Platform.OSVersion == "" { + return false + } + if mbv[j].Platform.OSVersion == "" { + return true + } + // TODO: Split version by parts and compare + // TODO: Prefer versions which have a greater version number + return false +} + +func (mbv manifestsByVersion) Len() int { + return len(mbv) +} + +func (mbv manifestsByVersion) Swap(i, j int) { + mbv[i], mbv[j] = mbv[j], mbv[i] +}