-
Notifications
You must be signed in to change notification settings - Fork 45
/
rpm.go
134 lines (114 loc) · 3.14 KB
/
rpm.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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
package analyzer
import (
"database/sql"
"fmt"
"strings"
"github.com/yaklang/yaklang/common/sca/dxtypes"
licenses "github.com/yaklang/yaklang/common/sca/license"
rpmdb "github.com/knqyf263/go-rpmdb/pkg"
"github.com/mattn/go-sqlite3"
"github.com/yaklang/yaklang/common/utils"
)
const (
TypRPM TypAnalyzer = "rpm-pkg"
statusRPM int = 1
)
func init() {
RegisterAnalyzer(TypRPM, NewRPMAnalyzer())
sql.Register("sqlite", &sqlite3.SQLiteDriver{})
}
var rpmRequiredFiles = []string{
// Berkeley DB
"usr/lib/sysimage/rpm/Packages",
"var/lib/rpm/Packages",
// NDB
"usr/lib/sysimage/rpm/Packages.db",
"var/lib/rpm/Packages.db",
// SQLite3
"usr/lib/sysimage/rpm/rpmdb.sqlite",
"var/lib/rpm/rpmdb.sqlite",
}
type rpmAnalyzer struct{}
func NewRPMAnalyzer() *rpmAnalyzer {
return &rpmAnalyzer{}
}
func (a rpmAnalyzer) createPackage(pkgInfo *rpmdb.PackageInfo, provides map[string]*dxtypes.Package) *dxtypes.Package {
pkg := &dxtypes.Package{
Name: pkgInfo.Name,
Version: pkgInfo.Version,
Verification: fmt.Sprintf("md5:%s", pkgInfo.SigMD5),
License: []string{licenses.Normalize(pkgInfo.License)},
}
for _, provide := range pkgInfo.Provides {
// handler libc.so.6(GLIBC_2.2.5)(64bit) => libc.so.6
if strings.Contains(provide, "(") {
provide = provide[:strings.Index(provide, "(")]
}
// handler /usr/bin/pkg-config => pkg-config
if strings.Contains(provide, "/") {
provide = provide[strings.LastIndex(provide, "/")+1:]
}
provides[provide] = pkg
}
pkg.DependsOn.And = make(map[string]string)
for _, dep := range pkgInfo.Requires {
// pass rpm package manage
// all package depend on rpm because these installed by rpm
if strings.HasPrefix(dep, "rpmlib") {
continue
}
// handler libc.so.6(GLIBC_2.2.5)(64bit) => libc.so.6
if strings.Contains(dep, "(") {
dep = dep[:strings.Index(dep, "(")]
}
// handler /usr/bin/pkg-config => pkg-config
if strings.Contains(dep, "/") {
dep = dep[strings.LastIndex(dep, "/")+1:]
}
// remove depende that provide by self
if p, ok := provides[dep]; ok {
if p == pkg {
continue
}
pkg.DependsOn.And[dep] = p.Version
} else {
pkg.DependsOn.And[dep] = "*"
}
}
return pkg
}
func (a rpmAnalyzer) Analyze(afi AnalyzeFileInfo) ([]*dxtypes.Package, error) {
fi := afi.Self
switch fi.MatchStatus {
case statusRPM:
provides := make(map[string]*dxtypes.Package)
db, err := rpmdb.Open(fi.LazyFile.Name())
if err != nil {
return nil, utils.Errorf("failed to open RPM DB: %v", err)
}
defer db.Close()
pkgList, err := db.ListPackages()
if err != nil {
return nil, utils.Errorf("failed to list packages: %v", err)
}
pkgs := make([]*dxtypes.Package, len(pkgList))
for i, pkgInfo := range pkgList {
pkgs[i] = a.createPackage(pkgInfo, provides)
}
handleDependsOn(pkgs, provides)
// lo.ForEach(pkgs, func(pkg *dxtypes.Package, _ int) {
// fmt.Printf(`
// name: %s
// depends: %v
// `, pkg.Name, pkg.DependsOn)
// })
return makePotentialPkgs(pkgs), nil
}
return nil, nil
}
func (a rpmAnalyzer) Match(info MatchInfo) int {
if utils.StringSliceContainsAll(rpmRequiredFiles, info.path) {
return statusRPM
}
return 0
}