Refacotring

This commit is contained in:
Kota Kanbe
2017-05-05 09:50:59 +09:00
committed by kota kanbe
parent 17a4e532c1
commit a2c364f9eb
8 changed files with 175 additions and 130 deletions

View File

@@ -31,7 +31,6 @@ import (
c "github.com/future-architect/vuls/config"
"github.com/future-architect/vuls/cveapi"
"github.com/future-architect/vuls/models"
"github.com/future-architect/vuls/oval"
"github.com/future-architect/vuls/report"
"github.com/future-architect/vuls/util"
)
@@ -161,39 +160,6 @@ func loadScanResults(jsonDir string) (results models.ScanResults, err error) {
return
}
func fillCveInfoFromCveDB(r models.ScanResult) (*models.ScanResult, error) {
var err error
var vs []models.VulnInfo
sInfo := c.Conf.Servers[r.ServerName]
vs, err = scanVulnByCpeNames(sInfo.CpeNames, r.ScannedCves)
if err != nil {
return nil, err
}
r.ScannedCves = vs
return r.FillCveDetail()
}
func fillCveInfoFromOvalDB(r *models.ScanResult) (*models.ScanResult, error) {
var ovalClient oval.Client
switch r.Family {
case "ubuntu", "debian":
ovalClient = oval.NewDebian()
case "rhel", "centos":
ovalClient = oval.NewRedhat()
case "amazon", "oraclelinux", "Raspbian", "FreeBSD":
//TODO implement OracleLinux
return r, nil
default:
return nil, fmt.Errorf("Oval %s is not implemented yet", r.Family)
}
result, err := ovalClient.FillCveInfoFromOvalDB(r)
if err != nil {
return nil, err
}
return result, nil
}
func loadPrevious(current models.ScanResults) (previous models.ScanResults, err error) {
var dirs jsonDirs
if dirs, err = lsValidJSONDirs(); err != nil {