Files
trivy/pkg/scanner/scan.go
2021-07-27 17:08:57 +03:00

123 lines
3.4 KiB
Go

package scanner
import (
"context"
"github.com/google/wire"
"golang.org/x/xerrors"
"github.com/aquasecurity/fanal/artifact"
aimage "github.com/aquasecurity/fanal/artifact/image"
flocal "github.com/aquasecurity/fanal/artifact/local"
"github.com/aquasecurity/fanal/artifact/remote"
"github.com/aquasecurity/fanal/image"
ftypes "github.com/aquasecurity/fanal/types"
"github.com/aquasecurity/trivy/pkg/log"
"github.com/aquasecurity/trivy/pkg/report"
"github.com/aquasecurity/trivy/pkg/rpc/client"
"github.com/aquasecurity/trivy/pkg/scanner/local"
"github.com/aquasecurity/trivy/pkg/types"
)
// StandaloneSuperSet is used in the standalone mode
var StandaloneSuperSet = wire.NewSet(
local.SuperSet,
wire.Bind(new(Driver), new(local.Scanner)),
NewScanner,
)
// StandaloneDockerSet binds docker dependencies
var StandaloneDockerSet = wire.NewSet(
types.GetDockerOption,
image.NewDockerImage,
aimage.NewArtifact,
StandaloneSuperSet,
)
// StandaloneArchiveSet binds archive scan dependencies
var StandaloneArchiveSet = wire.NewSet(
image.NewArchiveImage,
aimage.NewArtifact,
StandaloneSuperSet,
)
// StandaloneFilesystemSet binds filesystem dependencies
var StandaloneFilesystemSet = wire.NewSet(
flocal.NewArtifact,
StandaloneSuperSet,
)
// StandaloneRepositorySet binds repository dependencies
var StandaloneRepositorySet = wire.NewSet(
remote.NewArtifact,
StandaloneSuperSet,
)
// RemoteSuperSet is used in the client mode
var RemoteSuperSet = wire.NewSet(
aimage.NewArtifact,
client.SuperSet,
wire.Bind(new(Driver), new(client.Scanner)),
NewScanner,
)
// RemoteDockerSet binds remote docker dependencies
var RemoteDockerSet = wire.NewSet(
types.GetDockerOption,
image.NewDockerImage,
RemoteSuperSet,
)
// RemoteArchiveSet binds remote archive dependencies
var RemoteArchiveSet = wire.NewSet(
image.NewArchiveImage,
RemoteSuperSet,
)
// Scanner implements the Artifact and Driver operations
type Scanner struct {
driver Driver
artifact artifact.Artifact
}
// Driver defines operations of scanner
type Driver interface {
Scan(target string, artifactKey string, blobKeys []string, options types.ScanOptions) (
results report.Results, osFound *ftypes.OS, err error)
}
// NewScanner is the factory method of Scanner
func NewScanner(driver Driver, ar artifact.Artifact) Scanner {
return Scanner{driver: driver, artifact: ar}
}
// ScanArtifact scans the artifacts and returns results
func (s Scanner) ScanArtifact(ctx context.Context, options types.ScanOptions) (report.Report, error) {
artifactInfo, err := s.artifact.Inspect(ctx)
if err != nil {
return report.Report{}, xerrors.Errorf("failed analysis: %w", err)
}
results, osFound, err := s.driver.Scan(artifactInfo.Name, artifactInfo.ID, artifactInfo.BlobIDs, options)
if err != nil {
return report.Report{}, xerrors.Errorf("scan failed: %w", err)
}
if osFound != nil && osFound.Eosl {
log.Logger.Warnf("This OS version is no longer supported by the distribution: %s %s", osFound.Family, osFound.Name)
log.Logger.Warnf("The vulnerability detection may be insufficient because security updates are not provided")
}
return report.Report{
SchemaVersion: report.SchemaVersion,
ArtifactName: artifactInfo.Name,
ArtifactType: artifactInfo.Type,
Metadata: report.Metadata{
OS: osFound,
RepoTags: artifactInfo.RepoTags,
RepoDigests: artifactInfo.RepoDigests,
},
Results: results,
}, nil
}