2 changed files with 100 additions and 0 deletions
@ -0,0 +1,78 @@
|
||||
#! /usr/bin/env python3 |
||||
# from https://github.com/signalapp/Signal-Android/blob/2029ea378f249a70983c1fc3d55b9a63588bc06c/reproducible-builds/apkdiff/apkdiff.py |
||||
|
||||
import sys |
||||
from zipfile import ZipFile |
||||
|
||||
class ApkDiff: |
||||
IGNORE_FILES = ["META-INF/MANIFEST.MF", "META-INF/CERT.RSA", "META-INF/CERT.SF"] |
||||
|
||||
def compare(self, sourceApk, destinationApk): |
||||
sourceZip = ZipFile(sourceApk, 'r') |
||||
destinationZip = ZipFile(destinationApk, 'r') |
||||
|
||||
if self.compareManifests(sourceZip, destinationZip) and self.compareEntries(sourceZip, destinationZip) == True: |
||||
print("APKs match!") |
||||
else: |
||||
print("APKs don't match!") |
||||
|
||||
def compareManifests(self, sourceZip, destinationZip): |
||||
sourceEntrySortedList = sorted(sourceZip.namelist()) |
||||
destinationEntrySortedList = sorted(destinationZip.namelist()) |
||||
|
||||
for ignoreFile in self.IGNORE_FILES: |
||||
while ignoreFile in sourceEntrySortedList: sourceEntrySortedList.remove(ignoreFile) |
||||
while ignoreFile in destinationEntrySortedList: destinationEntrySortedList.remove(ignoreFile) |
||||
|
||||
if len(sourceEntrySortedList) != len(destinationEntrySortedList): |
||||
print("Manifest lengths differ!") |
||||
|
||||
for (sourceEntryName, destinationEntryName) in zip(sourceEntrySortedList, destinationEntrySortedList): |
||||
if sourceEntryName != destinationEntryName: |
||||
print("Sorted manifests don't match, %s vs %s" % (sourceEntryName, destinationEntryName)) |
||||
return False |
||||
|
||||
return True |
||||
|
||||
def compareEntries(self, sourceZip, destinationZip): |
||||
sourceInfoList = list(filter(lambda sourceInfo: sourceInfo.filename not in self.IGNORE_FILES, sourceZip.infolist())) |
||||
destinationInfoList = list(filter(lambda destinationInfo: destinationInfo.filename not in self.IGNORE_FILES, destinationZip.infolist())) |
||||
|
||||
if len(sourceInfoList) != len(destinationInfoList): |
||||
print("APK info lists of different length!") |
||||
return False |
||||
|
||||
for sourceEntryInfo in sourceInfoList: |
||||
for destinationEntryInfo in list(destinationInfoList): |
||||
if sourceEntryInfo.filename == destinationEntryInfo.filename: |
||||
sourceEntry = sourceZip.open(sourceEntryInfo, 'r') |
||||
destinationEntry = destinationZip.open(destinationEntryInfo, 'r') |
||||
|
||||
if self.compareFiles(sourceEntry, destinationEntry) != True: |
||||
print("APK entry %s does not match %s!" % (sourceEntryInfo.filename, destinationEntryInfo.filename)) |
||||
return False |
||||
|
||||
destinationInfoList.remove(destinationEntryInfo) |
||||
break |
||||
|
||||
return True |
||||
|
||||
def compareFiles(self, sourceFile, destinationFile): |
||||
sourceChunk = sourceFile.read(1024) |
||||
destinationChunk = destinationFile.read(1024) |
||||
|
||||
while sourceChunk != b"" or destinationChunk != b"": |
||||
if sourceChunk != destinationChunk: |
||||
return False |
||||
|
||||
sourceChunk = sourceFile.read(1024) |
||||
destinationChunk = destinationFile.read(1024) |
||||
|
||||
return True |
||||
|
||||
if __name__ == '__main__': |
||||
if len(sys.argv) != 3: |
||||
print("Usage: apkdiff <pathToFirstApk> <pathToSecondApk>") |
||||
sys.exit(1) |
||||
|
||||
ApkDiff().compare(sys.argv[1], sys.argv[2]) |
||||
Loading…
Reference in new issue