From 757281f21aee9389f94eaf4f283db56d5d041995 Mon Sep 17 00:00:00 2001
From: akumars1 <abhishek@deepfence.io>
Date: Wed, 20 Mar 2024 16:39:28 +0530
Subject: [PATCH] Added Unknown Severity for Standalone run

---
 output/output.go | 3 +++
 output/utils.go  | 1 +
 run-once.go      | 3 ++-
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/output/output.go b/output/output.go
index 456c471..7dce0cb 100644
--- a/output/output.go
+++ b/output/output.go
@@ -363,7 +363,10 @@ func CountBySeverity(report *[]scanner.VulnerabilityScanReport) *VulnerabilitySc
 			detail.Severity.Medium += 1
 		case utils.LOW:
 			detail.Severity.Low += 1
+		case utils.UNKNOWN:
+			detail.Severity.Unknown += 1
 		}
+
 	}
 
 	detail.CveScore = math.Min((cveScore*10.0)/500.0, 10.0)
diff --git a/output/utils.go b/output/utils.go
index 9f89267..522302f 100644
--- a/output/utils.go
+++ b/output/utils.go
@@ -17,6 +17,7 @@ type VulnerabilityScanDetail struct {
 		High     int `json:"high,omitempty"`
 		Medium   int `json:"medium,omitempty"`
 		Low      int `json:"low,omitempty"`
+		Unknown  int `json:"unknown,omitempty"`
 	} `json:"severity,omitempty"`
 	TimeStamp time.Time `json:"time_stamp,omitempty"`
 	Total     int       `json:"total,omitempty"`
diff --git a/run-once.go b/run-once.go
index 1c2d6ca..bce2f0c 100644
--- a/run-once.go
+++ b/run-once.go
@@ -161,7 +161,8 @@ func RunOnce(config utils.Config) {
 			utils.CRITICAL, details.Severity.Critical,
 			utils.HIGH, details.Severity.High,
 			utils.MEDIUM, details.Severity.Medium,
-			utils.LOW, details.Severity.Low)
+			utils.LOW, details.Severity.Low,
+			utils.UNKNOWN, details.Severity.Unknown)
 		if len(exploitable) > 0 {
 			fmt.Println("\nMost Exploitable Vulnerabilities:")
 			_ = out.TableOutput(&exploitable)