fix(Triage) adding checks for dict

SEKOIA-IO · Jan 9, 2025 · c99e4f6 · c99e4f6
1 parent b82da32
commit c99e4f6
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/Triage/triage_modules/trigger_triage.py b/Triage/triage_modules/trigger_triage.py
@@ -142,14 +142,14 @@ def check_suspicious_analysis(self, sample_id: str, data: dict) -> bool:
         try:
             tags: list[str] = list()
             tasks: list[dict] = list()
-            tags = data.get("analysis")["tags"]
-            if "linux" in tags:
+            tags = data.get("analysis", {}).get("tags")
+            if len(tags) == 0 or "linux" in tags:
                 return False
-            tasks = data.get("tasks")
+            tasks = data.get("tasks", [])
             scores = []
             for t in tasks:
-                if t.get("kind") == "behavioral":
-                    scores.append(t.get("score"))
+                if t.get("kind") == "behavioral" and "score" in t:
+                    scores.append(t["score"])
             if len(scores) < 2:
                 self.log(f"PE in {sample_id} report does not have at least two dynamic analysis")
                 return True