Skip to content

Commit 32dc6bb

Browse files
patrick.rissmann@l3montree.compatrick.rissmann@l3montree.com
authored andcommitted
Added a vuln event for removing and adding a scanner to an existing vulnerability
1 parent 01f5f08 commit 32dc6bb

2 files changed

Lines changed: 24 additions & 1 deletion

File tree

internal/core/assetversion/asset_version_service.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -336,7 +336,7 @@ func buildBomRefMap(bom normalize.SBOM) map[string]cdx.Component {
336336

337337
func (s *service) UpdateSBOM(assetVersion models.AssetVersion, scannerID string, sbom normalize.SBOM) error {
338338
// load the asset components
339-
339+
scannerID = "Other-Scanner"
340340
assetComponents, err := s.componentRepository.LoadComponents(nil, assetVersion.Name, assetVersion.AssetID, "")
341341
if err != nil {
342342
return errors.Wrap(err, "could not load asset components")

internal/database/models/vulnevent_model.go

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@ const (
2626
EventTypeRawRiskAssessmentUpdated VulnEventType = "rawRiskAssessmentUpdated"
2727

2828
EventTypeComment VulnEventType = "comment"
29+
30+
EventTypeAddedScanner VulnEventType = "addedScanner"
31+
EventTypeRemovedScanner VulnEventType = "removedScanner"
2932
)
3033

3134
type VulnEvent struct {
@@ -211,6 +214,22 @@ func NewRawRiskAssessmentUpdatedEvent(vulnID string, userID string, justificatio
211214
return event
212215
}
213216

217+
func NewAddedScannerEvent(vulnID string, userID string) VulnEvent {
218+
return VulnEvent{
219+
Type: EventTypeAddedScanner,
220+
VulnID: vulnID,
221+
UserID: userID,
222+
}
223+
}
224+
225+
func NewRemovedScannerEvent(vulnID string, userID string) VulnEvent {
226+
return VulnEvent{
227+
Type: EventTypeRemovedScanner,
228+
VulnID: vulnID,
229+
UserID: userID,
230+
}
231+
}
232+
214233
func CheckStatusType(statusType string) error {
215234
switch statusType {
216235
case "fixed":
@@ -229,6 +248,10 @@ func CheckStatusType(statusType string) error {
229248
return nil
230249
case "markedForTransfer":
231250
return nil
251+
case "addedScanner":
252+
return nil
253+
case "removedScanner":
254+
return nil
232255
default:
233256
return fmt.Errorf("invalid status type")
234257
}

0 commit comments

Comments
 (0)