mirror of
https://github.com/aquasecurity/trivy.git
synced 2025-12-21 23:00:42 -08:00
7b86f81e29
* feat(config): add --cache-backend * feat(operation): embed cache.Cache into operation.Cache * feat(cache): support redis:// * test(integration): add redis test * chore(README): add --cache-backend * chore(mod): update * chore: add disclaimer
113 lines
3.1 KiB
Go
113 lines
3.1 KiB
Go
package operation
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/go-redis/redis/v8"
|
|
"github.com/google/wire"
|
|
"github.com/spf13/afero"
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/aquasecurity/fanal/cache"
|
|
"github.com/aquasecurity/trivy/pkg/db"
|
|
"github.com/aquasecurity/trivy/pkg/log"
|
|
"github.com/aquasecurity/trivy/pkg/utils"
|
|
)
|
|
|
|
// SuperSet binds cache dependencies
|
|
var SuperSet = wire.NewSet(
|
|
cache.NewFSCache,
|
|
wire.Bind(new(cache.LocalArtifactCache), new(cache.FSCache)),
|
|
NewCache,
|
|
)
|
|
|
|
// Cache implements the local cache
|
|
type Cache struct {
|
|
cache.Cache
|
|
}
|
|
|
|
// NewCache is the factory method for Cache
|
|
func NewCache(backend string) (Cache, error) {
|
|
if strings.HasPrefix(backend, "redis://") {
|
|
log.Logger.Info("Redis cache: %s", backend)
|
|
redisCache := cache.NewRedisCache(&redis.Options{
|
|
Addr: strings.TrimPrefix(backend, "redis://"),
|
|
})
|
|
return Cache{Cache: redisCache}, nil
|
|
}
|
|
fsCache, err := cache.NewFSCache(utils.CacheDir())
|
|
if err != nil {
|
|
return Cache{}, xerrors.Errorf("unable to initialize fs cache: %w", err)
|
|
}
|
|
return Cache{Cache: fsCache}, nil
|
|
}
|
|
|
|
// Reset resets the cache
|
|
func (c Cache) Reset() (err error) {
|
|
if err := c.ClearDB(); err != nil {
|
|
return xerrors.Errorf("failed to clear the database: %w", err)
|
|
}
|
|
if err := c.ClearImages(); err != nil {
|
|
return xerrors.Errorf("failed to clear the image cache: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// ClearDB clears the DB cache
|
|
func (c Cache) ClearDB() (err error) {
|
|
log.Logger.Info("Removing DB file...")
|
|
if err = os.RemoveAll(utils.CacheDir()); err != nil {
|
|
return xerrors.Errorf("failed to remove the directory (%s) : %w", utils.CacheDir(), err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// ClearImages clears the cache images
|
|
func (c Cache) ClearImages() error {
|
|
log.Logger.Info("Removing image caches...")
|
|
if err := c.Clear(); err != nil {
|
|
return xerrors.Errorf("failed to remove the cache: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// DownloadDB downloads the DB
|
|
func DownloadDB(appVersion, cacheDir string, quiet, light, skipUpdate bool) error {
|
|
client := initializeDBClient(cacheDir, quiet)
|
|
ctx := context.Background()
|
|
needsUpdate, err := client.NeedsUpdate(appVersion, light, skipUpdate)
|
|
if err != nil {
|
|
return xerrors.Errorf("database error: %w", err)
|
|
}
|
|
|
|
if needsUpdate {
|
|
log.Logger.Info("Need to update DB")
|
|
log.Logger.Info("Downloading DB...")
|
|
if err = client.Download(ctx, cacheDir, light); err != nil {
|
|
return xerrors.Errorf("failed to download vulnerability DB: %w", err)
|
|
}
|
|
if err = client.UpdateMetadata(cacheDir); err != nil {
|
|
return xerrors.Errorf("unable to update database metadata: %w", err)
|
|
}
|
|
}
|
|
|
|
// for debug
|
|
if err := showDBInfo(cacheDir); err != nil {
|
|
return xerrors.Errorf("failed to show database info: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func showDBInfo(cacheDir string) error {
|
|
m := db.NewMetadata(afero.NewOsFs(), cacheDir)
|
|
metadata, err := m.Get()
|
|
if err != nil {
|
|
return xerrors.Errorf("something wrong with DB: %w", err)
|
|
}
|
|
log.Logger.Debugf("DB Schema: %d, Type: %d, UpdatedAt: %s, NextUpdate: %s, DownloadedAt: %s",
|
|
metadata.Version, metadata.Type, metadata.UpdatedAt, metadata.NextUpdate, metadata.DownloadedAt)
|
|
return nil
|
|
}
|