64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package detect
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/future-architect/vuls/pkg/cmd/version"
|
|
"github.com/future-architect/vuls/pkg/detect/cpe"
|
|
"github.com/future-architect/vuls/pkg/detect/debian"
|
|
detectTypes "github.com/future-architect/vuls/pkg/detect/types"
|
|
"github.com/future-architect/vuls/pkg/detect/ubuntu"
|
|
"github.com/future-architect/vuls/pkg/detect/windows"
|
|
"github.com/future-architect/vuls/pkg/types"
|
|
)
|
|
|
|
func Detect(ctx context.Context, host *types.Host) error {
|
|
if host.ScanError != "" {
|
|
return nil
|
|
}
|
|
|
|
var detectors []detectTypes.Detector
|
|
if len(host.Packages.OSPkg) > 0 {
|
|
switch host.Family {
|
|
case "debian":
|
|
detectors = append(detectors, debian.Detector{})
|
|
case "ubuntu":
|
|
detectors = append(detectors, ubuntu.Detector{})
|
|
default:
|
|
return errors.New("not implemented")
|
|
}
|
|
}
|
|
if len(host.Packages.KB) > 0 {
|
|
detectors = append(detectors, windows.Detector{})
|
|
}
|
|
if len(host.Packages.CPE) > 0 {
|
|
detectors = append(detectors, cpe.Detector{})
|
|
}
|
|
|
|
var err error
|
|
for {
|
|
if len(detectors) == 0 {
|
|
break
|
|
}
|
|
d := detectors[0]
|
|
if err = d.Detect(ctx, host); err != nil {
|
|
break
|
|
}
|
|
detectors = detectors[1:]
|
|
}
|
|
|
|
t := time.Now()
|
|
host.DetecteddAt = &t
|
|
host.DetectedVersion = version.Version
|
|
host.DetectedRevision = version.Revision
|
|
|
|
if err != nil {
|
|
return errors.Wrapf(err, "detect %s", host.Name)
|
|
}
|
|
|
|
return nil
|
|
}
|