88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
package gost
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
|
|
cnf "github.com/future-architect/vuls/config"
|
|
"github.com/future-architect/vuls/models"
|
|
"github.com/knqyf263/gost/db"
|
|
"github.com/parnurzeal/gorequest"
|
|
"golang.org/x/xerrors"
|
|
)
|
|
|
|
// Client is the interface of OVAL client.
|
|
type Client interface {
|
|
FillWithGost(db.DB, *models.ScanResult, bool) (int, error)
|
|
|
|
//TODO implement
|
|
// CheckHTTPHealth() error
|
|
// CheckIfGostFetched checks if Gost entries are fetched
|
|
// CheckIfGostFetched(db.DB, string, string) (bool, error)
|
|
// CheckIfGostFresh(db.DB, string, string) (bool, error)
|
|
}
|
|
|
|
// NewClient make Client by family
|
|
func NewClient(family string) Client {
|
|
switch family {
|
|
case cnf.RedHat, cnf.CentOS:
|
|
return RedHat{}
|
|
case cnf.Debian:
|
|
return Debian{}
|
|
case cnf.Windows:
|
|
return Microsoft{}
|
|
default:
|
|
return Pseudo{}
|
|
}
|
|
}
|
|
|
|
// Base is a base struct
|
|
type Base struct {
|
|
family string
|
|
}
|
|
|
|
// CheckHTTPHealth do health check
|
|
func (b Base) CheckHTTPHealth() error {
|
|
if !cnf.Conf.Gost.IsFetchViaHTTP() {
|
|
return nil
|
|
}
|
|
|
|
url := fmt.Sprintf("%s/health", cnf.Conf.Gost.URL)
|
|
var errs []error
|
|
var resp *http.Response
|
|
resp, _, errs = gorequest.New().Get(url).End()
|
|
// resp, _, errs = gorequest.New().SetDebug(config.Conf.Debug).Get(url).End()
|
|
// resp, _, errs = gorequest.New().Proxy(api.httpProxy).Get(url).End()
|
|
if 0 < len(errs) || resp == nil || resp.StatusCode != 200 {
|
|
return xerrors.Errorf("Failed to connect to gost server. url: %s, errs: %w", url, errs)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// CheckIfGostFetched checks if oval entries are in DB by family, release.
|
|
func (b Base) CheckIfGostFetched(driver db.DB, osFamily string) (fetched bool, err error) {
|
|
//TODO
|
|
return true, nil
|
|
}
|
|
|
|
// CheckIfGostFresh checks if oval entries are fresh enough
|
|
func (b Base) CheckIfGostFresh(driver db.DB, osFamily string) (ok bool, err error) {
|
|
//TODO
|
|
return true, nil
|
|
}
|
|
|
|
// Pseudo is Gost client except for RedHat family and Debian
|
|
type Pseudo struct {
|
|
Base
|
|
}
|
|
|
|
// FillWithGost fills cve information that has in Gost
|
|
func (pse Pseudo) FillWithGost(driver db.DB, r *models.ScanResult, _ bool) (int, error) {
|
|
return 0, nil
|
|
}
|
|
|
|
func major(osVer string) (majorVersion string) {
|
|
return strings.Split(osVer, ".")[0]
|
|
}
|