mirror of
https://github.com/NekoX-Dev/NekoX.git
synced 2024-11-26 18:59:19 +01:00
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
|
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"" or 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 = ["META-INF/MANIFEST.MF", "META-INF/CERT.RSA", "META-INF/CERT.SF", "resources.arsc"]
|
||
|
|
|
||
|
|
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("APKs are from bundle!")
|
||
|
|
else:
|
||
|
|
print("APKs are different!")
|