mirror of
https://github.com/aquasecurity/trivy.git
synced 2025-12-22 07:10:41 -08:00
44 lines
1.2 KiB
Go
44 lines
1.2 KiB
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/memoryfs"
|
|
)
|
|
|
|
// memFS is a wrapper of memoryfs.FS and can change its underlying file system
|
|
// at runtime. This implements fs.FS.
|
|
type memFS struct {
|
|
current *memoryfs.FS
|
|
}
|
|
|
|
// Open implements fs.FS.
|
|
func (m *memFS) Open(name string) (fs.File, error) {
|
|
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) (err error) {
|
|
memfs := memoryfs.New()
|
|
if err = memfs.MkdirAll(filepath.Dir(filePath), fs.ModePerm); err != nil {
|
|
return xerrors.Errorf("memory fs mkdir error: %w", err)
|
|
}
|
|
err = memfs.WriteLazyFile(filePath, func() (io.Reader, error) {
|
|
return content, nil
|
|
}, fs.ModePerm)
|
|
if err != nil {
|
|
return xerrors.Errorf("memory fs write error: %w", err)
|
|
}
|
|
|
|
m.current = memfs
|
|
return
|
|
}
|