patchrom_tools/apkcerts.py at marshmallow · MiCode/patchrom_tools · GitHub

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
#!/usr/bin/env python

import os
import sys
import string
import xml
import xml.dom
from xml.dom import minidom

def getName(pkg, updatedPackages):
    codePath = pkg.attributes["codePath"].value
    if codePath.startswith("/data/app"):
        codePath = ""
        pkgName = pkg.attributes["name"].value
        for updatedPkg in updatedPackages:
            updatedPkgName = updatedPkg.attributes["name"].value
            if updatedPkgName == pkgName:
                codePath=updatedPkg.attributes["codePath"].value
                break
    return os.path.basename(codePath)

def usage():
    print "Usage: python ./apkcerts.py path-to-packages.xml path-to-apkcerts.txt"

def main():
    if len(sys.argv) != 3:
        usage()
        sys.exit(1)

    try:
        xmldoc = minidom.parse(sys.argv[1])
    except :
        print "Error: %s doesn't exist or isn't a vaild xml file" %(sys.argv[1])
        sys.exit(1)

    updatedPackages = xmldoc.getElementsByTagName("updated-package")
    packages = xmldoc.getElementsByTagName("package")
    sigpkgs = {}
    for pkg in packages:
        name = getName(pkg, updatedPackages)
        cert= pkg.getElementsByTagName("cert")
        if not cert or not name: continue
        index = cert[0].getAttribute("index")
        sigpkgs[index] = sigpkgs.get(index, []) + [name]

    sharedUsers = xmldoc.getElementsByTagName("shared-user")
    sigmap = {}
    for user in sharedUsers:
        name = user.getAttribute("name")
        cert = user.getElementsByTagName("cert")
        if not cert: continue
        index = cert[0].getAttribute("index")
        if name == "android.uid.system":
            sigmap[index] = "platform"
        elif name == "android.media":
            sigmap[index] = "media"
        elif name == "android.uid.shared":
            sigmap[index] = "shared"

    with open(sys.argv[2], "w") as f:
        for keyindex, pkgnames in sigpkgs.items():
            sig = sigmap.get(keyindex, None)
            for pkgname in pkgnames:
                if sig:
                    line = 'name="{0}.apk" certificate="build/target/product/security/{1}.x509.pem" private_key="build/target/product/security/{1}.pk8"\n'.format(pkgname, sig)
                else:
                    line = 'name="{0}.apk" certificate="PRESIGNED" private_key=""\n'.format(pkgname)
                f.write(line)

if "__main__" == __name__:
    main()