-
Notifications
You must be signed in to change notification settings - Fork 2.3k
/
vmdk.go
36 lines (28 loc) · 869 Bytes
/
vmdk.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
package disk
import (
"errors"
"io"
"github.com/masahiro331/go-vmdk-parser/pkg/virtualization/vmdk"
"golang.org/x/xerrors"
"github.com/aquasecurity/trivy/pkg/fanal/vm"
)
type VMDK struct{}
func (VMDK) NewReader(rs io.ReadSeeker, cache vm.Cache[string, []byte]) (*io.SectionReader, error) {
if _, err := rs.Seek(0, io.SeekStart); err != nil {
return nil, xerrors.Errorf("seek error: %w", err)
}
if _, err := vmdk.Check(rs); err != nil {
return nil, vm.ErrInvalidSignature
}
if _, err := rs.Seek(0, io.SeekStart); err != nil {
return nil, xerrors.Errorf("seek error: %w", err)
}
reader, err := vmdk.Open(rs, cache)
if err != nil {
if errors.Is(err, vmdk.ErrUnSupportedType) {
return nil, xerrors.Errorf("%s: %w", err.Error(), vm.ErrUnsupportedType)
}
return nil, xerrors.Errorf("failed to open vmdk: %w", err)
}
return reader, nil
}