trivy/pkg/module/memfs.go

package module

import (
	"io"
	"io/fs"
	"path/filepath"

	"golang.org/x/xerrors"

	dio "github.com/aquasecurity/go-dep-parser/pkg/io"
	"github.com/aquasecurity/trivy/pkg/mapfs"
)

// memFS is a wrapper of mapfs.FS and can change its underlying file system
// at runtime. This implements fs.FS.
type memFS struct {
	current *mapfs.FS
}

// Open implements fs.FS.
func (m *memFS) Open(name string) (fs.File, error) {
	if m.current == nil {
		return nil, fs.ErrNotExist
	}
	return m.current.Open(name)
}

// initialize changes the underlying memory file system with the given file path and contents.
//
// Note: it is always to safe swap the underlying FS with this API since this is called only at the beginning of
// Analyze interface call, which is not concurrently called per module instance.
func (m *memFS) initialize(filePath string, content dio.ReadSeekerAt) error {
	mfs := mapfs.New()
	if err := mfs.MkdirAll(filepath.Dir(filePath), fs.ModePerm); err != nil {
		return xerrors.Errorf("mapfs mkdir error: %w", err)
	}
	b, err := io.ReadAll(content)
	if err != nil {
		return xerrors.Errorf("read error: %w", err)
	}
	err = mfs.WriteVirtualFile(filePath, b, fs.ModePerm)
	if err != nil {
		return xerrors.Errorf("mapfs write error: %w", err)
	}

	m.current = mfs
	return nil
}