Skip to content

Commit

Permalink
fix(containerd): Use img platform in exporter instead of strict host …
Browse files Browse the repository at this point in the history
…platform (aquasecurity#4477)

* match with img platform instead of host platform

* client matching pull spec

* use default platform

* pull with platforms default strict

* use withplatform to pull and add debug log

* looks like we are trying to scan a i386 image

* revert changes on test, use the right platform match

* try with Config.Platform

* use spect.platform

* fix function usage

* try another way to retrieve the platform

* fix compilation

* read platforms from config manifest

* use platform from RegistryOptions if available, otherwise get the actual platform

* goimport

* put platform in containerd client

* fix panic

* use DefaultStrict as default
  • Loading branch information
AliDatadog committed Jul 19, 2023
1 parent ce77bb4 commit 3e2416d
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 7 deletions.
4 changes: 2 additions & 2 deletions pkg/fanal/image/daemon.go
Original file line number Diff line number Diff line change
Expand Up @@ -32,8 +32,8 @@ func tryPodmanDaemon(_ context.Context, imageName string, _ name.Reference, _ ty
}, cleanup, nil
}

func tryContainerdDaemon(ctx context.Context, imageName string, _ name.Reference, _ types.ImageOptions) (types.Image, func(), error) {
img, cleanup, err := daemon.ContainerdImage(ctx, imageName)
func tryContainerdDaemon(ctx context.Context, imageName string, _ name.Reference, opts types.ImageOptions) (types.Image, func(), error) {
img, cleanup, err := daemon.ContainerdImage(ctx, imageName, opts)
if err != nil {
return nil, cleanup, err
}
Expand Down
29 changes: 24 additions & 5 deletions pkg/fanal/image/daemon/containerd.go
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,8 @@ import (
ocispec "github.com/opencontainers/image-spec/specs-go/v1"
"github.com/samber/lo"
"golang.org/x/xerrors"

"github.com/aquasecurity/trivy/pkg/fanal/types"
)

const (
Expand All @@ -50,14 +52,21 @@ func (n familiarNamed) String() string {
return string(n)
}

func imageWriter(client *containerd.Client, img containerd.Image) imageSave {
func imageWriter(client *containerd.Client, img containerd.Image, platform types.Platform) imageSave {
return func(ctx context.Context, ref []string) (io.ReadCloser, error) {
if len(ref) < 1 {
return nil, xerrors.New("no image reference")
}
imgOpts := archive.WithImage(client.ImageService(), ref[0])
manifestOpts := archive.WithManifest(img.Target())
platOpts := archive.WithPlatform(platforms.DefaultStrict())

var platformMatchComparer platforms.MatchComparer
if platform.Platform == nil {
platformMatchComparer = platforms.DefaultStrict()
} else {
platformMatchComparer = img.Platform()
}
platOpts := archive.WithPlatform(platformMatchComparer)
pr, pw := io.Pipe()
go func() {
pw.CloseWithError(archive.Export(ctx, client.ContentStore(), pw, imgOpts, manifestOpts, platOpts))
Expand All @@ -67,7 +76,7 @@ func imageWriter(client *containerd.Client, img containerd.Image) imageSave {
}

// ContainerdImage implements v1.Image
func ContainerdImage(ctx context.Context, imageName string) (Image, func(), error) {
func ContainerdImage(ctx context.Context, imageName string, opts types.ImageOptions) (Image, func(), error) {
cleanup := func() {}

addr := os.Getenv("CONTAINERD_ADDRESS")
Expand All @@ -85,7 +94,17 @@ func ContainerdImage(ctx context.Context, imageName string) (Image, func(), erro
return nil, cleanup, err
}

client, err := containerd.New(addr)
options := []containerd.ClientOpt{}
if opts.RegistryOptions.Platform.Platform != nil {
ociPlatform, err := platforms.Parse(opts.RegistryOptions.Platform.String())
if err != nil {
return nil, cleanup, err
}

options = append(options, containerd.WithDefaultPlatform(platforms.OnlyStrict(ociPlatform)))
}

client, err := containerd.New(addr, options...)
if err != nil {
return nil, cleanup, xerrors.Errorf("failed to initialize a containerd client: %w", err)
}
Expand Down Expand Up @@ -125,7 +144,7 @@ func ContainerdImage(ctx context.Context, imageName string) (Image, func(), erro
}

return &image{
opener: imageOpener(ctx, ref.String(), f, imageWriter(client, img)),
opener: imageOpener(ctx, ref.String(), f, imageWriter(client, img, opts.RegistryOptions.Platform)),
inspect: insp,
history: history,
}, cleanup, nil
Expand Down

0 comments on commit 3e2416d

Please sign in to comment.