Fix apk classification (#72)

* Fix apk classification
CERT-Polska · Oct 2, 2023 · ab285aa · ab285aa
1 parent 62faee9
commit ab285aa
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/karton/classifier/classifier.py b/karton/classifier/classifier.py
@@ -265,9 +265,13 @@ def zip_has_file(path: str) -> bool:
             except Exception:
                 return False
 
-        if magic.startswith("Zip archive data") or magic.startswith(
-            "Java archive data (JAR)"
-        ):
+        JAVA_ARCHIVES = [
+            "Zip archive data",
+            "Java archive data (JAR)",
+            "Android package (APK)",
+        ]
+
+        if any(magic.startswith(x) for x in JAVA_ARCHIVES):
             if extension == "apk" or zip_has_file("AndroidManifest.xml"):
                 sample_class.update(
                     {"kind": "runnable", "platform": "android", "extension": "apk"}