Update cape_extracted.py

ctxis · Sep 5, 2019 · 66fbdff · 66fbdff
1 parent c675f0d
commit 66fbdff
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/modules/signatures/cape_extracted.py b/modules/signatures/cape_extracted.py
@@ -27,14 +27,14 @@ class CAPEExtractedContent(Signature):
     def run(self):
         ret = False
         if "CAPE" in self.results:
-            for cape in self.results["CAPE"]:
-                capetype = cape["cape_type"]
-                yara = cape["cape_yara"]
-                process = cape["process_name"]
+            for cape in self.results.get("CAPE", []):
+                capetype = cape.get("cape_type", "")
+                yara = cape.get("cape_yara", "")
+                process = cape.get("process_name", "")
                 if capetype and process:
-                    self.data.append({process : capetype})
+                    self.data.append({process: capetype})
                     ret = True
                     if yara:
-                        self.data.append({process : yara})
+                        self.data.append({process: yara})
 
             return ret