import binascii
import os
import hashlib
import zlib
import zipfile
import shutil
from xml.dom import minidom

unshellpath = r"D:\CodeAS\shell1\app\build\outputs\apk\debug\app-debug.apk"
srcapkpath = r"D:\CodeAS\srcapk\app\build\outputs\apk\debug\app-debug.apk"


def fixCheckSum(shell):
    shell.seek(0x0C)
    data = shell.read()
    checksum = zlib.adler32(data)
    strchecksum = str(hex(checksum))
    strchecksum = strchecksum.replace("0x", "")
    if not len(strchecksum) % 2 == 0:
        strchecksum = strchecksum.zfill(len(strchecksum)+1)
    shell.seek(0x08)
    shell.write(bytearray.fromhex(strchecksum)[::-1])
    
def fixSHA1(shell):
    shell.seek(0x20)
    signBytes = shell.read()
    sha1 = hashlib.sha1()
    sha1.update(signBytes)
    sign = sha1.hexdigest()
    b = bytes.fromhex(sign)
    shell.seek(0x0C)
    shell.write(b)


def fixFileSize(shell, num):
    b = bytearray()
    for i in range(4):
        number = int(num % 256)
        b.append(number)
        num = num >> 8
    shell.seek(0x20)
    shell.write(b)

def IntToHex(num):
    b = bytearray()
    for i in range(4):
        number = int(num % 256)
        b.append(number)
        num = num >> 8
    b.reverse()
    return b

def fixXML():
    doc = minidom.parse(srcapkpath.replace('.apk', r'\AndroidManifest.xml'))
    root = doc.documentElement
    app_node = root.getElementsByTagName('application')[0]
    app_name = app_node.getAttribute('android:name')
    app_node.setAttribute('android:name', 'app.simba.shell1.ShellApplication')
    meta = doc.createElement('meta-data')
    meta.setAttribute('android:name', 'ApplicationName')
    meta.setAttribute('android:value', app_name)
    app_node.appendChild(meta)

    doc.writexml(open(srcapkpath.replace(
        '.apk', r'\AndroidManifest.xml'), 'w'), encoding='utf-8')


def main():
    
    if os.path.exists(unshellpath.replace('.apk','')):
        shutil.rmtree(unshellpath.replace('.apk',''))

    apk = zipfile.ZipFile(unshellpath)
    for n in apk.namelist():
        apk.extract(n,unshellpath.replace('.apk',''))

    for root, dirs, files in os.walk(unshellpath.replace('.apk', '')):
        for f in files:
            if os.path.join(root, f).endswith('.dex'):
                unshelldexpath = os.path.join(root, f)
    print('[*] 成功反编译shellAPK')

    os.system('apktool.bat -f -s d ' + srcapkpath +
              ' -o ' + srcapkpath.replace('.apk', ''))
    srcjarpath = srcapkpath.replace('.apk','.jar')
    os.system('dex2jar.bat -f '+srcapkpath+' -o '+srcjarpath)
    os.system('java -jar dx.jar --dex --output='+srcjarpath+' '+srcjarpath)
    for root, dirs, files in os.walk(srcapkpath.replace('.apk', '')):
        for f in files:
            if os.path.join(root, f).endswith('.dex'):
                os.remove(os.path.join(root, f))
    print('[*] 成功反编译待加壳APK')

    unshelldex = open(unshelldexpath, 'rb+')
    tmpshell = unshelldex.read()
    unshellArray = bytearray(tmpshell)
    unshelldex.close()
    print('[*] 成功读取脱壳DEX文件')

    tmp = open(srcjarpath,'rb+')
    srcjarArray = bytearray(tmp.read())
    tmp.close()

    tmpdex = unshellArray +srcjarArray+IntToHex(len(srcjarArray))

    f = open(srcapkpath.replace('.apk','/classes.dex'),'wb+')
    f.write(tmpdex)
    f.close()

    shell = open(srcapkpath.replace('.apk','/classes.dex'),'rb+')
    fixFileSize(shell, len(tmpdex))
    fixSHA1(shell)
    fixCheckSum(shell)
    shell.close()

    print('[*] DEX文件修复完毕')

    fixXML()
    print('[*] AndroidManifest文件修改完毕')

    outputpath = os.path.dirname(
        srcapkpath)+r'\shell_'+os.path.basename(srcapkpath)
    os.system('apktool.bat  b '+srcapkpath.replace('.apk', '')+' -o ' +
              outputpath)
    print('[*] APK重打包完毕')

    os.system('jarsigner -keystore '+r'D:\CodeAS\mykey.jks'+' -signedjar ' +
              outputpath.replace('.apk', '_signed.apk')+' '+outputpath+' mykey '+'-storepass 123456')


if __name__ == "__main__":
    main()