From ba48fc0d738e05a3c1f4182c6cd9a0752ea18880 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C5=82=20Praszmo?= <michalpr@cert.pl>
Date: Mon, 2 Oct 2023 15:30:00 +0200
Subject: [PATCH] Fix apk classification (#72)

* Fix apk classification
---
 karton/classifier/classifier.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/karton/classifier/classifier.py b/karton/classifier/classifier.py
index 930c643..46c6590 100644
--- a/karton/classifier/classifier.py
+++ b/karton/classifier/classifier.py
@@ -284,9 +284,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"}