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