1#!/usr/bin/env python 2# Copyright 2019 The Chromium Authors. All rights reserved. 3# Use of this source code is governed by a BSD-style license that can be 4# found in the LICENSE file. 5 6import argparse 7import os 8import re 9import sys 10import zipfile 11 12sys.path.append(os.path.join(os.path.dirname(__file__), os.pardir)) 13from pylib.dex import dex_parser 14from util import build_utils 15 16_FLAGS_PATH = ( 17 '//chrome/android/java/static_library_dex_reference_workarounds.flags') 18 19 20def _FindIllegalStaticLibraryReferences(static_lib_dex_files, 21 main_apk_dex_files): 22 main_apk_defined_types = set() 23 for dex_file in main_apk_dex_files: 24 for class_def_item in dex_file.class_def_item_list: 25 main_apk_defined_types.add( 26 dex_file.GetTypeString(class_def_item.class_idx)) 27 28 static_lib_referenced_types = set() 29 for dex_file in static_lib_dex_files: 30 for type_item in dex_file.type_item_list: 31 static_lib_referenced_types.add( 32 dex_file.GetString(type_item.descriptor_idx)) 33 34 return main_apk_defined_types.intersection(static_lib_referenced_types) 35 36 37def _DexFilesFromPath(path): 38 if zipfile.is_zipfile(path): 39 with zipfile.ZipFile(path) as z: 40 return [ 41 dex_parser.DexFile(bytearray(z.read(name))) for name in z.namelist() 42 if re.match(r'.*classes[0-9]*\.dex$', name) 43 ] 44 else: 45 with open(path) as f: 46 return dex_parser.DexFile(bytearray(f.read())) 47 48 49def main(args): 50 args = build_utils.ExpandFileArgs(args) 51 parser = argparse.ArgumentParser() 52 parser.add_argument( 53 '--depfile', required=True, help='Path to output depfile.') 54 parser.add_argument( 55 '--stamp', required=True, help='Path to file to touch upon success.') 56 parser.add_argument( 57 '--static-library-dex', 58 required=True, 59 help='classes.dex or classes.zip for the static library APK that was ' 60 'proguarded with other dependent APKs') 61 parser.add_argument( 62 '--static-library-dependent-dex', 63 required=True, 64 action='append', 65 dest='static_library_dependent_dexes', 66 help='classes.dex or classes.zip for the APKs that use the static ' 67 'library APK') 68 args = parser.parse_args(args) 69 70 static_library_dexfiles = _DexFilesFromPath(args.static_library_dex) 71 for path in args.static_library_dependent_dexes: 72 dependent_dexfiles = _DexFilesFromPath(path) 73 illegal_references = _FindIllegalStaticLibraryReferences( 74 static_library_dexfiles, dependent_dexfiles) 75 76 if illegal_references: 77 msg = 'Found illegal references from {} to {}\n'.format( 78 args.static_library_dex, path) 79 msg += 'Add a -keep rule to avoid this. ' 80 msg += 'See {} for an example and why this is necessary.\n'.format( 81 _FLAGS_PATH) 82 msg += 'The illegal references are:\n' 83 msg += '\n'.join(illegal_references) 84 sys.stderr.write(msg) 85 sys.exit(1) 86 87 input_paths = [args.static_library_dex] + args.static_library_dependent_dexes 88 build_utils.Touch(args.stamp) 89 build_utils.WriteDepfile(args.depfile, args.stamp, inputs=input_paths) 90 91 92if __name__ == '__main__': 93 sys.exit(main(sys.argv[1:])) 94