Bugfix: Fix issue with results that do not have a malware_family or c…

…lassification property
OALabs · Nov 20, 2023 · 99742d1 · 99742d1
1 parent f2fcc9b
commit 99742d1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,8 @@
 # Changelog
 
+## 1.1.1 - 2023-11-19
+ - Bugfix: Fix issue with results that do not have a malware_family or classification property 
+
 ## 1.1.0 - 2023-10-09
 
  - Added support for String Searching

diff --git a/src/unpacme_search.py b/src/unpacme_search.py
@@ -453,6 +453,8 @@ def PopulateForm(self):
                     if entry['malware_family'].lower() not in family_lower:
                         family_lower.append(entry['malware_family'].lower())
                         malware_family.append(entry['malware_family'])
+                except KeyError:
+                    logger.debug("No malware family")
                 except AttributeError:
                     logger.debug("No malware family")
 
@@ -653,11 +655,11 @@ def build_result(self, result: Dict[str, Any], search_string: str):
             family = ""
             for entry in r['malwareid']:
 
-                label = entry['malware_family']
+                label = entry.get('malware_family', None)
                 if not family:
                     family = label
 
-                classification = entry['classification_type']
+                classification = entry.get('classification_type', None)
                 if classification:
                     classification_type = classification
 
@@ -1088,7 +1090,7 @@ class UnpacMeByteSearchPlugin(ida_idaapi.plugin_t):
     wanted_name = "UnpacMe Byte Search"
     wanted_hotkey = ""
 
-    _version = "1.1.0"
+    _version = "1.1.1"
 
     def _banner(self):
         return f"""