Add cvss_v4_vector to vulnerability entry

Ostorlab · Aug 15, 2024 · fd014b4 · fd014b4
1 parent fb66adb
commit fd014b4
Show file tree

Hide file tree

Showing 4 changed files with 603 additions and 24 deletions.
diff --git a/src/ostorlab/agent/kb/kb.py b/src/ostorlab/agent/kb/kb.py
@@ -29,6 +29,7 @@ class Entry:
     targeted_by_ransomware: bool = False
     targeted_by_nation_state: bool = False
     cvss_v3_vector: str = ""
+    cvss_v4_vector: str = ""
     category_groups: Optional[list[dict[str, Union[str, list[str]]]]] = None
 
 
@@ -68,6 +69,7 @@ def __getattr__(cls, item: str) -> Entry:
                 targeted_by_ransomware=meta.get("targeted_by_ransomware", False),
                 targeted_by_nation_state=meta.get("targeted_by_nation_state", False),
                 cvss_v3_vector=meta.get("cvss_v3_vector", ""),
+                cvss_v4_vector=meta.get("cvss_v4_vector", ""),
                 category_groups=category_groups,
             )
 

diff --git a/src/ostorlab/agent/message/proto/v3/report/vulnerability/vulnerability.proto b/src/ostorlab/agent/message/proto/v3/report/vulnerability/vulnerability.proto
@@ -77,4 +77,5 @@ message Message{
   optional string cvss_v3_vector = 15;
   optional VulnerabilityLocation vulnerability_location = 16;
   repeated CategoryGroup category_groups = 17;
+  optional string cvss_v4_vector = 18;
 }