Merge pull request #97 from CybercentreCanada/cccs-kevin-patch-1

Cccs kevin patch 1 [dev]
CybercentreCanada · Jun 14, 2024 · 744d1f8 · 744d1f8
2 parents f9b11fd + 11d9e5d
commit 744d1f8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/intezer.py b/intezer.py
@@ -554,7 +554,7 @@ def execute(self, request: ServiceRequest) -> None:
             return
 
         # Do we want to generate our own verdicts based our own gene metric?
-        if not self.config.get("use_black_box_verdicts", True) and verdict in Verdicts.MALICIOUS_VERDICTS.value:
+        if not self.config.get("use_black_box_verdicts", True) and verdict in Verdicts.MALICIOUS_VERDICTS.value + Verdicts.SUSPICIOUS_VERDICTS.value:
             verdict = Verdicts.TESTING.value
 
         analysis_id = main_api_result["analysis_id"]
@@ -767,7 +767,7 @@ def _process_iocs(
         if file_iocs:
             for file in file_iocs:
                 # Do we want to generate our own verdicts based our own gene metric?
-                if not self.config.get("use_black_box_verdicts", True) and file["verdict"] in Verdicts.MALICIOUS_VERDICTS.value:
+                if not self.config.get("use_black_box_verdicts", True) and file["verdict"] in Verdicts.MALICIOUS_VERDICTS.value + Verdicts.SUSPICIOUS_VERDICTS.value:
                     file_verdict_map[file["sha256"]] = Verdicts.TESTING.value
                 else:
                     file_verdict_map[file["sha256"]] = file["verdict"]