mirror of
https://github.com/aquasecurity/trivy.git
synced 2025-12-23 15:37:50 -08:00
* refactor: embed config * refactor: replace image and layer with artifact and blob * feat(config): add ArtifactConfig * fix(scanner): use Artifact * test(scanner): update mocks * feat: add repo and fs subcommands * chore(mod): update * refactor: fix warn message * feat(cli): add --no-progress to repo and fs * mod: Update fanal dependency Signed-off-by: Simarpreet Singh <simar@linux.com> Co-authored-by: Simarpreet Singh <simar@linux.com>
97 lines
3.7 KiB
Go
97 lines
3.7 KiB
Go
// Code generated by Wire. DO NOT EDIT.
|
|
|
|
//go:generate wire
|
|
//+build !wireinject
|
|
|
|
package artifact
|
|
|
|
import (
|
|
"context"
|
|
"github.com/aquasecurity/fanal/applier"
|
|
image2 "github.com/aquasecurity/fanal/artifact/image"
|
|
local2 "github.com/aquasecurity/fanal/artifact/local"
|
|
"github.com/aquasecurity/fanal/artifact/remote"
|
|
"github.com/aquasecurity/fanal/cache"
|
|
"github.com/aquasecurity/fanal/image"
|
|
"github.com/aquasecurity/trivy-db/pkg/db"
|
|
"github.com/aquasecurity/trivy/pkg/detector/library"
|
|
"github.com/aquasecurity/trivy/pkg/detector/ospkg"
|
|
"github.com/aquasecurity/trivy/pkg/scanner"
|
|
"github.com/aquasecurity/trivy/pkg/scanner/local"
|
|
"github.com/aquasecurity/trivy/pkg/types"
|
|
"github.com/aquasecurity/trivy/pkg/vulnerability"
|
|
"time"
|
|
)
|
|
|
|
// Injectors from inject.go:
|
|
|
|
func initializeDockerScanner(ctx context.Context, imageName string, artifactCache cache.ArtifactCache, localArtifactCache cache.LocalArtifactCache, timeout time.Duration) (scanner.Scanner, func(), error) {
|
|
applierApplier := applier.NewApplier(localArtifactCache)
|
|
detector := ospkg.Detector{}
|
|
driverFactory := library.DriverFactory{}
|
|
libraryDetector := library.NewDetector(driverFactory)
|
|
localScanner := local.NewScanner(applierApplier, detector, libraryDetector)
|
|
dockerOption, err := types.GetDockerOption(timeout)
|
|
if err != nil {
|
|
return scanner.Scanner{}, nil, err
|
|
}
|
|
imageImage, cleanup, err := image.NewDockerImage(ctx, imageName, dockerOption)
|
|
if err != nil {
|
|
return scanner.Scanner{}, nil, err
|
|
}
|
|
artifact := image2.NewArtifact(imageImage, artifactCache)
|
|
scannerScanner := scanner.NewScanner(localScanner, artifact)
|
|
return scannerScanner, func() {
|
|
cleanup()
|
|
}, nil
|
|
}
|
|
|
|
func initializeArchiveScanner(ctx context.Context, filePath string, artifactCache cache.ArtifactCache, localArtifactCache cache.LocalArtifactCache, timeout time.Duration) (scanner.Scanner, error) {
|
|
applierApplier := applier.NewApplier(localArtifactCache)
|
|
detector := ospkg.Detector{}
|
|
driverFactory := library.DriverFactory{}
|
|
libraryDetector := library.NewDetector(driverFactory)
|
|
localScanner := local.NewScanner(applierApplier, detector, libraryDetector)
|
|
imageImage, err := image.NewArchiveImage(filePath)
|
|
if err != nil {
|
|
return scanner.Scanner{}, err
|
|
}
|
|
artifact := image2.NewArtifact(imageImage, artifactCache)
|
|
scannerScanner := scanner.NewScanner(localScanner, artifact)
|
|
return scannerScanner, nil
|
|
}
|
|
|
|
func initializeFilesystemScanner(ctx context.Context, dir string, artifactCache cache.ArtifactCache, localArtifactCache cache.LocalArtifactCache) (scanner.Scanner, func(), error) {
|
|
applierApplier := applier.NewApplier(localArtifactCache)
|
|
detector := ospkg.Detector{}
|
|
driverFactory := library.DriverFactory{}
|
|
libraryDetector := library.NewDetector(driverFactory)
|
|
localScanner := local.NewScanner(applierApplier, detector, libraryDetector)
|
|
artifact := local2.NewArtifact(dir, artifactCache)
|
|
scannerScanner := scanner.NewScanner(localScanner, artifact)
|
|
return scannerScanner, func() {
|
|
}, nil
|
|
}
|
|
|
|
func initializeRepositoryScanner(ctx context.Context, url string, artifactCache cache.ArtifactCache, localArtifactCache cache.LocalArtifactCache) (scanner.Scanner, func(), error) {
|
|
applierApplier := applier.NewApplier(localArtifactCache)
|
|
detector := ospkg.Detector{}
|
|
driverFactory := library.DriverFactory{}
|
|
libraryDetector := library.NewDetector(driverFactory)
|
|
localScanner := local.NewScanner(applierApplier, detector, libraryDetector)
|
|
artifact, cleanup, err := remote.NewArtifact(url, artifactCache)
|
|
if err != nil {
|
|
return scanner.Scanner{}, nil, err
|
|
}
|
|
scannerScanner := scanner.NewScanner(localScanner, artifact)
|
|
return scannerScanner, func() {
|
|
cleanup()
|
|
}, nil
|
|
}
|
|
|
|
func initializeVulnerabilityClient() vulnerability.Client {
|
|
config := db.Config{}
|
|
client := vulnerability.NewClient(config)
|
|
return client
|
|
}
|