Update packages.py

kevoreilly · Jul 18, 2023 · ae2577d · ae2577d
1 parent 529c61d
commit ae2577d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/analyzer/windows/lib/core/packages.py b/analyzer/windows/lib/core/packages.py
@@ -125,7 +125,11 @@ def choose_package(file_type, file_name, exports, target):
         return "pdf"
     elif re.search(b'<script\\s+language="(J|VB|Perl)Script"', file_content, re.I):
         return "wsf"
-    elif file_name.endswith((".vbs", ".vbe")) or re.findall(rb"\s?Dim\s", file_content, re.I) or re.findall(b"\s?\x00D\x00i\x00m\x00\s", file_content, re.I):
+    elif (
+        file_name.endswith((".vbs", ".vbe"))
+        or re.findall(rb"\s?Dim\s", file_content, re.I)
+        or re.findall(b"\s?\x00D\x00i\x00m\x00\s", file_content, re.I)
+    ):
         return "vbs"
     elif b"Set-StrictMode" in file_content[:100]:
         return "ps1"