Repo created

apkfrombundle.py

@@ -0,0 +1,71 @@
+import sys
+from zipfile import ZipFile
+
+def compareFiles(first, second):
+    while True:
+        firstBytes = first.read(4096)
+        secondBytes = second.read(4096)
+        if firstBytes != secondBytes:
+            return False
+
+        if firstBytes == b"" and secondBytes == b"":
+            break
+
+    return True
+
+def remove_prefix(text, prefix):
+    if text.startswith(prefix):
+        return text[len(prefix):]
+    return text
+
+def compareApkFromBundle(bundle, apk):
+    FILES_TO_IGNORE = ["resources.arsc", "stamp-cert-sha256"]
+
+    apkZip = ZipFile(apk, 'r')
+    bundleZip = ZipFile(bundle, 'r')
+
+    firstList = list(filter(lambda info: info.filename not in FILES_TO_IGNORE, apkZip.infolist()))
+    secondList = list(filter(lambda secondInfo: secondInfo.filename not in FILES_TO_IGNORE, bundleZip.infolist()))
+
+    for apkInfo in firstList:
+        if (apkInfo.filename.startswith("META-INF/")):
+            continue
+        if (apkInfo.filename.startswith("res/")):
+            continue
+        if (apkInfo.filename.startswith("AndroidManifest.xml")):
+            continue
+
+        found = False
+        for bundleInfo in secondList:
+            fileName = bundleInfo.filename
+            fileName = remove_prefix(fileName, "base/root/")
+            fileName = remove_prefix(fileName, "base/dex/")
+            fileName = remove_prefix(fileName, "base/manifest/")
+            fileName = remove_prefix(fileName, "base/")
+            if (fileName.startswith("BUNDLE-METADATA")):
+                fileName = "META-INF" + remove_prefix(fileName, "BUNDLE-METADATA/")
+            if fileName == apkInfo.filename:
+                found = True
+                firstFile = apkZip.open(apkInfo, 'r')
+                secondFile = bundleZip.open(bundleInfo, 'r')
+                if compareFiles(firstFile, secondFile) != True:
+                    print("APK file %s does not match" % apkInfo.filename)
+                    return False
+                break
+
+        if found == False:
+            print("file %s not found in APK" % apkInfo.filename)
+            return False
+
+    return True
+
+if __name__ == '__main__':
+    if len(sys.argv) != 3:
+        print("Usage: apkfrombundle <pathToBundle> <pathToApk>")
+        sys.exit(1)
+
+
+    if sys.argv[1] == sys.argv[2] or compareApkFromBundle(sys.argv[1], sys.argv[2]) == True:
+        print("APK from bundle!")
+    else:
+        print("APK has difference!")