57 lines
2.6 KiB
Python
57 lines
2.6 KiB
Python
import sys, os, subprocess, ConfigParser
|
|
from optparse import OptionParser
|
|
|
|
class2jar = dict()
|
|
jars_to_process = set()
|
|
|
|
opt_parser = OptionParser()
|
|
opt_parser.add_option("-v", "--verbose", action="store_true", dest="verbose", default=False, help="set verbosity")
|
|
opt_parser.add_option("-c", "--config", dest="config", default=os.path.dirname(sys.argv[0])+"/android-jar.conf", help="configuration file path")
|
|
(options, args) = opt_parser.parse_args()
|
|
if not os.path.isfile(options.config):
|
|
sys.exit('Error! No configuration file at: '+options.config)
|
|
config_parser = ConfigParser.SafeConfigParser()
|
|
config_parser.read(options.config)
|
|
ajars_dir = config_parser.get('android-paths', 'aosp-out-dir')
|
|
stubs_jar = config_parser.get('android-paths', 'android-stubs-jar')
|
|
jarinfer = config_parser.get('jar-infer-paths', 'jar-infer-path')
|
|
wrk_dir = config_parser.get('android-model', 'wrk-dir')
|
|
astubx_file = config_parser.get('android-model', 'astubx-path')
|
|
|
|
### Finding android libraries ###
|
|
print "> Finding android libraries..."
|
|
# Since SDK 31, the directory structure mixes classes.jar and classes-header.jar files, we use
|
|
# only the former, as classes-header.jar files are missing methods' bytecode.
|
|
cmd = "find " + ajars_dir + " -name \"*.jar\" | grep -v classes-header.jar"
|
|
ajars = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True).stdout.read().splitlines()
|
|
for ajar in ajars:
|
|
cmd = "jar tvf " + ajar + " | grep \"\.class$\" | awk '{print $8}'"
|
|
ajar_classes = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True).stdout.read().splitlines()
|
|
for ajar_class in ajar_classes:
|
|
if ajar_class in class2jar:
|
|
if options.verbose:
|
|
print "[Warn] Same class in multiple jars : " + ajar_class + " in " + class2jar[ajar_class] + " , " + ajar
|
|
else:
|
|
class2jar[ajar_class] = ajar
|
|
cmd = "jar tvf " + stubs_jar + " | grep \"\.class$\" | awk '{print $8}'"
|
|
ajar_classes = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True).stdout.read().splitlines()
|
|
found = 0
|
|
for ajar_class in ajar_classes:
|
|
if ajar_class in class2jar:
|
|
found += 1
|
|
jars_to_process.add(class2jar[ajar_class])
|
|
else:
|
|
if options.verbose:
|
|
print "[Warn] Class not found in any jar : " + ajar_class
|
|
print ",".join(list(jars_to_process))
|
|
print "Found " + str(found) + " / " + str(len(ajar_classes)) + " in " + str(len(jars_to_process)) + " jars"
|
|
|
|
### Running jarinfer ###
|
|
print "> Running jarinfer on android jars..."
|
|
cmd = "java -jar " + jarinfer + " -i " + ",".join(list(jars_to_process)) + " -o " + wrk_dir + "/" + astubx_file
|
|
if options.verbose:
|
|
cmd += " -dv"
|
|
print cmd
|
|
returncode = subprocess.call(cmd, shell=True)
|
|
sys.exit(returncode)
|