Add reference link in malware result (#2383)

deepfence · Dec 4, 2024 · 3d7e50e · 3d7e50e
1 parent ad22512
commit 3d7e50e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/deepfence_server/model/scans.go b/deepfence_server/model/scans.go
@@ -453,6 +453,7 @@ type Malware struct {
 	Filetype         string        `json:"filetype"`
 	Info             string        `json:"info"`
 	Version          string        `json:"version"`
+	Reference        string        `json:"reference"`
 	SeverityScore    int           `json:"severity_score"`
 	StringsToMatch   []interface{} `json:"strings_to_match"`
 	Summary          string        `json:"summary"`
@@ -485,7 +486,6 @@ type MalwareRule struct {
 	Filetype     string `json:"filetype"`
 	Info         string `json:"info"`
 	Version      string `json:"version"`
-	Reference    string `json:"reference"`
 	FileSeverity string `json:"file_severity"`
 	Masked       bool   `json:"masked" required:"true"`
 	Payload      string `json:"payload" required:"true"`

diff --git a/deepfence_worker/ingesters/malware.go b/deepfence_worker/ingesters/malware.go
@@ -81,6 +81,7 @@ func malwareToMaps(data []ingestersUtil.Malware) ([]map[string]map[string]interf
 
 		malware["node_id"] = generateHashFromString(fmt.Sprintf("%v:%v",
 			i.MetaRules.RuleID, i.CompleteFilename))
+		malware["reference"] = i.MetaRules.Reference
 
 		malwares = append(malwares, map[string]map[string]interface{}{
 			"Rule":    utils.ToMap(i.MetaRules),