forked from DrKLO/Telegram
-
Notifications
You must be signed in to change notification settings - Fork 0
/
apkfrombundle.py
71 lines (57 loc) · 2.35 KB
/
apkfrombundle.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
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!")