apkdiff.py

#!/usr/bin/python3

from pathlib import Path
import os
import zipfile
from filecmp import *
from subprocess import call, STDOUT
import shutil
import re
import argparse
import difflib

at = "at"
ignore = ".*(align|apktool.yml|pak|MF|RSA|SF|bin|so)"
count = 0
args = None
d = difflib.Differ()

class bcolors:
    OKBLUE = '\033[94m'
    OKGREEN = '\033[92m'
    FAIL = '\033[91m'

def format(string, col):
    return col + string + '\033[0m'

def main():

    print("")
    print("\t\t\t apktool")
    print("")

    parser = argparse.ArgumentParser(description='Diff two versions of an APK file.')
    parser.add_argument('apk1', metavar='apk1', help='Location of the first APK.')
    parser.add_argument('apk2', metavar='apk2', help='Location of the second APK.')
    parser.add_argument('-c', '--cleanup', action='store_true', help='Remove all extracted files after computation.')
    parser.add_argument('-m', '--meld', action='store_true', help='Open meld to compare directories after.')
    parser.add_argument('-o', '--output', default="/tmp/apkdiff/", help='The location to output the extracted files to.')
    
    global args
    args = parser.parse_args()

    # Make sure the APKs exist.
    exists(args.apk1)
    exists(args.apk2)
    
    # Check the temporary folder exists & clear it.
    folderExists(args.output, True)

    # Individual folders for each APK.
    temp1 = Path(args.output) / "1"
    temp2 = Path(args.output) / "2"

    # Extracted code + resources from Apktool.
    at1 = temp1 / at
    at2 = temp2 / at

    folderExists(temp1, True)
    folderExists(temp2, True)

    extract(args.apk1, temp1)
    extract(args.apk2, temp2)

    apktoolit(args.apk1, at1)
    apktoolit(args.apk2, at2)

    compare(at1, at2)

    # Remove all the stuff we have created.
    if args.cleanup and os.path.exists(temp1):
        shutil.rmtree(temp1)

    if args.cleanup and os.path.exists(temp2):
        shutil.rmtree(temp2)

    if args.meld:
        call(["meld", at1, at2])

def apktoolit(file, dir):
    print("Running apktool against '" + format(file, bcolors.OKBLUE) + "'")
    call(["apktool", "d", "--no-debug-info", "--match-original", "-o", dir, file], stdout=open(os.devnull, 'w'), stderr=STDOUT)
    print("[" + format("OK", bcolors.OKGREEN) + "]")

def compare(folder1, folder2):
    print("")
    compared = dircmp(folder1, folder2)
    report_full_closure(compared)
    print("\n\t" + format(str(count), bcolors.OKBLUE) + " files are different.\n")

def report_full_closure(self):
    for name in self.diff_files:
        if not re.match(ignore, name):
            content1 = reader(Path(self.left) / name).splitlines(1)
            content2 = reader(Path(self.right) / name).splitlines(1)
            diff = difflib.unified_diff(content1, content2)
            tidied = tidy(list(diff))
            if tidied.replace(" \n","") != f"{format('---', bcolors.FAIL)}{format('+++', bcolors.OKGREEN)}":
                print("[" + format(name, bcolors.OKGREEN)  + "] " + 
                    format(self.left.replace(args.output + "1",""), bcolors.OKBLUE))
                print(tidied)
                global count
                count += 1

    for sd in self.subdirs.values():
        report_full_closure(sd)

def tidy(lines):
    sorted = ""
    i = 0
    while i < len(lines):
        line = ""
        report = True
        #Check if they differ because of apktools naming
        if lines[i][:1] == "-" and lines[i][1:2] != "-" :
            if len(lines[i]) == len(lines[i+1]):
                for c in range(1,len(lines[i])):
                    if lines[i][c] != lines[i+1][c]:
                        if not lines[i][c-1].isalnum() and not lines[i][c+1].isalnum():
                            report = False
        # If the diff is caused by apktool naming don't report the diff
        if report:
            if lines[i][:1] == "+":
                line = format(lines[i], bcolors.OKGREEN)
            elif lines[i][:1] == "-":
                line = format(lines[i], bcolors.FAIL)
            sorted += line
            i+=1
        else:
            i+=2
    return sorted

def reader(file):
    f = open(file, 'r', encoding='utf8', errors='ignore')
    data = f.read()
    f.close()
    return data

def folderExists(path, clean=False):
    if clean and os.path.exists(path):
        shutil.rmtree(path)
    os.makedirs(path)

def exists(file):
    if not os.path.isfile(file):
        print(format("'{}' does not exist.".format(file), bcolors.FAIL))
        exit(0)

def extract(apk, dir):
    zipfi = zipfile.ZipFile(apk, 'r')
    zipfi.extract("classes.dex", dir)
    zipfi.close()
    print("Extracted '" + format("classes.dex", bcolors.OKBLUE) + "' from '" + format(apk, bcolors.OKBLUE) + "'.")

if __name__ == '__main__':
    main()