Merge pull request #477 from CybercentreCanada/cccs-mog-patch-1

Cccs mog patch 1[dev]
CybercentreCanada · Nov 7, 2024 · 11617a8 · 11617a8
2 parents a11b2dd + 4b2e2df
commit 11617a8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/cape/cape_main.py b/cape/cape_main.py
@@ -229,7 +229,7 @@ def __init__(self, config: Optional[Dict] = None) -> None:
         self.identify = get_identify(use_cache=os.environ.get("PRIVILEGED", "false").lower() == "true")
         self.retry_on_no_machine = False
         self.uwsgi_with_recycle = False
-        self.classification = forge.get_classification()
+        self.classification = get_classification()
 
         # Properies pertaining to using YARA rules with CAPE
         self.yara_sigs = None

diff --git a/cape/cape_result.py b/cape/cape_result.py
@@ -14,6 +14,7 @@
 import lief
 from peutils import is_valid
 
+from assemblyline.common import forge
 from assemblyline.common import log as al_log
 from assemblyline.common.attack_map import revoke_map
 from assemblyline.common.identify import CUSTOM_BATCH_ID, CUSTOM_PS1_ID
@@ -84,7 +85,7 @@
     7: 1000,
     8: 1000,
 }  # dead_host signature
-
+Classification = forge.get_classification()
 # Signature Processing Constants
 SKIPPED_FAMILIES = ["generic"]
 
@@ -706,7 +707,7 @@ def process_signatures(
             name=sig_name,
             type="CUCKOO",
             score=translated_score,
-            classification= Classification.UNRESTRICTED,
+            classification = Classification.UNRESTRICTED,
         )
         sig_res = _create_signature_result_section(
             sig_name,