92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
// +build !scanner
|
|
|
|
package gost
|
|
|
|
import (
|
|
"github.com/future-architect/vuls/config"
|
|
"github.com/future-architect/vuls/logging"
|
|
"github.com/future-architect/vuls/models"
|
|
"github.com/knqyf263/gost/db"
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/future-architect/vuls/constant"
|
|
)
|
|
|
|
// DBDriver is a DB Driver
|
|
type DBDriver struct {
|
|
DB db.DB
|
|
Cnf config.VulnDictInterface
|
|
}
|
|
|
|
// Client is the interface of OVAL client.
|
|
type Client interface {
|
|
DetectUnfixed(*models.ScanResult, bool) (int, error)
|
|
CloseDB() error
|
|
}
|
|
|
|
// Base is a base struct
|
|
type Base struct {
|
|
DBDriver DBDriver
|
|
}
|
|
|
|
func (b Base) CloseDB() error {
|
|
return b.DBDriver.DB.CloseDB()
|
|
}
|
|
|
|
// FillCVEsWithRedHat fills CVE detailed with Red Hat Security
|
|
func FillCVEsWithRedHat(r *models.ScanResult, cnf config.GostConf) error {
|
|
db, locked, err := newGostDB(cnf)
|
|
if locked {
|
|
return xerrors.Errorf("SQLite3 is locked: %s", cnf.GetSQLite3Path())
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
defer func() {
|
|
if err := db.CloseDB(); err != nil {
|
|
logging.Log.Errorf("Failed to close DB. err: %+v", err)
|
|
}
|
|
}()
|
|
return RedHat{Base{DBDriver{DB: db, Cnf: &cnf}}}.fillCvesWithRedHatAPI(r)
|
|
}
|
|
|
|
// NewClient make Client by family
|
|
func NewClient(cnf config.GostConf, family string) (Client, error) {
|
|
db, locked, err := newGostDB(cnf)
|
|
if locked {
|
|
return nil, xerrors.Errorf("SQLite3 is locked: %s", cnf.GetSQLite3Path())
|
|
} else if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
driver := DBDriver{DB: db, Cnf: &cnf}
|
|
|
|
switch family {
|
|
case constant.RedHat, constant.CentOS:
|
|
return RedHat{Base{DBDriver: driver}}, nil
|
|
case constant.Debian, constant.Raspbian:
|
|
return Debian{Base{DBDriver: driver}}, nil
|
|
case constant.Windows:
|
|
return Microsoft{Base{DBDriver: driver}}, nil
|
|
default:
|
|
return Pseudo{}, nil
|
|
}
|
|
}
|
|
|
|
// NewGostDB returns db client for Gost
|
|
func newGostDB(cnf config.GostConf) (driver db.DB, locked bool, err error) {
|
|
if cnf.IsFetchViaHTTP() {
|
|
return nil, false, nil
|
|
}
|
|
path := cnf.GetURL()
|
|
if cnf.GetType() == "sqlite3" {
|
|
path = cnf.GetSQLite3Path()
|
|
}
|
|
if driver, locked, err = db.NewDB(cnf.GetType(), path, cnf.GetDebugSQL()); err != nil {
|
|
if locked {
|
|
return nil, true, xerrors.Errorf("gostDB is locked. err: %w", err)
|
|
}
|
|
return nil, false, err
|
|
}
|
|
return driver, false, nil
|
|
}
|