0
Fork 0
mirror of https://github.com/project-zot/zot.git synced 2024-12-16 21:56:37 -05:00

fix GetReferrers function to be able to retrieve referrers of any specified artifactType

Signed-off-by: Alex Stan <alexandrustan96@yahoo.ro>
This commit is contained in:
Alex Stan 2022-07-14 15:30:24 +03:00 committed by Ramkumar Chinchani
parent 317064ffc9
commit a31869f270
2 changed files with 56 additions and 5 deletions

View file

@ -19,7 +19,6 @@ import (
guuid "github.com/gofrs/uuid" guuid "github.com/gofrs/uuid"
"github.com/minio/sha256-simd" "github.com/minio/sha256-simd"
"github.com/notaryproject/notation-go-lib" "github.com/notaryproject/notation-go-lib"
notreg "github.com/notaryproject/notation/pkg/registry"
godigest "github.com/opencontainers/go-digest" godigest "github.com/opencontainers/go-digest"
ispec "github.com/opencontainers/image-spec/specs-go/v1" ispec "github.com/opencontainers/image-spec/specs-go/v1"
"github.com/opencontainers/umoci" "github.com/opencontainers/umoci"
@ -1443,7 +1442,7 @@ func (is *ImageStoreLocal) DeleteBlob(repo, digest string) error {
return nil return nil
} }
func (is *ImageStoreLocal) GetReferrers(repo, digest, mediaType string) ([]artifactspec.Descriptor, error) { func (is *ImageStoreLocal) GetReferrers(repo, digest, artifactType string) ([]artifactspec.Descriptor, error) {
var lockLatency time.Time var lockLatency time.Time
dir := path.Join(is.rootDir, repo) dir := path.Join(is.rootDir, repo)
@ -1509,14 +1508,16 @@ func (is *ImageStoreLocal) GetReferrers(repo, digest, mediaType string) ([]artif
return nil, err return nil, err
} }
if mediaType != artManifest.ArtifactType || gdigest != artManifest.Subject.Digest { if artifactType != artManifest.ArtifactType || gdigest != artManifest.Subject.Digest {
continue continue
} }
result = append(result, artifactspec.Descriptor{ result = append(result, artifactspec.Descriptor{
MediaType: manifest.MediaType, MediaType: manifest.MediaType,
ArtifactType: notreg.ArtifactTypeNotation, ArtifactType: artManifest.ArtifactType,
Digest: manifest.Digest, Size: manifest.Size, Annotations: manifest.Annotations, Digest: manifest.Digest,
Size: manifest.Size,
Annotations: manifest.Annotations,
}) })
found = true found = true

View file

@ -17,6 +17,7 @@ import (
godigest "github.com/opencontainers/go-digest" godigest "github.com/opencontainers/go-digest"
ispec "github.com/opencontainers/image-spec/specs-go/v1" ispec "github.com/opencontainers/image-spec/specs-go/v1"
artifactspec "github.com/oras-project/artifacts-spec/specs-go/v1"
"github.com/rs/zerolog" "github.com/rs/zerolog"
. "github.com/smartystreets/goconvey/convey" . "github.com/smartystreets/goconvey/convey"
"zotregistry.io/zot/errors" "zotregistry.io/zot/errors"
@ -163,6 +164,55 @@ func TestStorageFSAPIs(t *testing.T) {
}) })
} }
func TestGetReferrers(t *testing.T) {
dir := t.TempDir()
log := log.Logger{Logger: zerolog.New(os.Stdout)}
metrics := monitoring.NewMetricsServer(false, log)
imgStore := storage.NewImageStore(dir, true, storage.DefaultGCDelay, true, true, log, metrics)
Convey("Get referrers", t, func(c C) {
err := test.CopyFiles("../../test/data/zot-test", path.Join(dir, "zot-test"))
So(err, ShouldBeNil)
body := []byte("this is a blob")
digest := godigest.FromBytes(body)
buf := bytes.NewBuffer(body)
buflen := buf.Len()
err = ioutil.WriteFile(path.Join(imgStore.RootDir(), //nolint: gosec
"zot-test", "blobs", digest.Algorithm().String(), digest.Encoded()),
buf.Bytes(), 0o644)
So(err, ShouldBeNil)
_, n, err := imgStore.FullBlobUpload("zot-test", buf, digest.String())
So(err, ShouldBeNil)
So(n, ShouldEqual, buflen)
artifactManifest := artifactspec.Manifest{}
artifactManifest.ArtifactType = "signature-example"
artifactManifest.Subject = artifactspec.Descriptor{
MediaType: ispec.MediaTypeImageManifest,
Digest: digest,
Size: int64(buflen),
}
artifactManifest.Blobs = []artifactspec.Descriptor{}
manBuf, err := json.Marshal(artifactManifest)
manBufLen := len(manBuf)
So(err, ShouldBeNil)
manDigest := godigest.FromBytes(manBuf)
_, err = imgStore.PutImageManifest("zot-test", manDigest.Encoded(), artifactspec.MediaTypeArtifactManifest, manBuf)
So(err, ShouldBeNil)
So(err, ShouldBeNil)
descriptors, err := imgStore.GetReferrers("zot-test", digest.String(), "signature-example")
So(err, ShouldBeNil)
So(descriptors, ShouldNotBeEmpty)
So(descriptors[0].ArtifactType, ShouldEqual, "signature-example")
So(descriptors[0].MediaType, ShouldEqual, artifactspec.MediaTypeArtifactManifest)
So(descriptors[0].Size, ShouldEqual, manBufLen)
So(descriptors[0].Digest, ShouldEqual, manDigest)
})
}
func TestDedupeLinks(t *testing.T) { func TestDedupeLinks(t *testing.T) {
dir := t.TempDir() dir := t.TempDir()