OLD | NEW |
(Empty) | |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. |
| 4 |
| 5 import os |
| 6 import re |
| 7 import tempfile |
| 8 |
| 9 from pylib import constants |
| 10 from pylib import cmd_helper |
| 11 |
| 12 |
| 13 _PROGUARD_CLASS_RE = re.compile(r'\s*?- Program class:\s*([\S]+)$') |
| 14 _PROGUARD_SUPERCLASS_RE = re.compile(r'\s*? Superclass:\s*([\S]+)$') |
| 15 _PROGUARD_SECTION_RE = re.compile( |
| 16 r'^(?:Interfaces|Constant Pool|Fields|Methods|Class file attributes) ' |
| 17 r'\(count = \d+\):$') |
| 18 _PROGUARD_METHOD_RE = re.compile(r'\s*?- Method:\s*(\S*)[(].*$') |
| 19 _PROGUARD_ANNOTATION_RE = re.compile(r'\s*?- Annotation \[L(\S*);\]:$') |
| 20 _PROGUARD_ANNOTATION_CONST_RE = ( |
| 21 re.compile(r'\s*?- Constant element value.*$')) |
| 22 _PROGUARD_ANNOTATION_VALUE_RE = re.compile(r'\s*?- \S+? \[(.*)\]$') |
| 23 |
| 24 _PROGUARD_PATH_SDK = os.path.join( |
| 25 constants.ANDROID_SDK_ROOT, 'tools', 'proguard', 'lib', 'proguard.jar') |
| 26 _PROGUARD_PATH_BUILT = ( |
| 27 os.path.join(os.environ['ANDROID_BUILD_TOP'], 'external', 'proguard', |
| 28 'lib', 'proguard.jar') |
| 29 if 'ANDROID_BUILD_TOP' in os.environ else None) |
| 30 _PROGUARD_PATH = ( |
| 31 _PROGUARD_PATH_SDK if os.path.exists(_PROGUARD_PATH_SDK) |
| 32 else _PROGUARD_PATH_BUILT) |
| 33 |
| 34 |
| 35 def Dump(jar_path): |
| 36 """Dumps class and method information from a JAR into a dict via proguard. |
| 37 |
| 38 Args: |
| 39 jar_path: An absolute path to the JAR file to dump. |
| 40 Returns: |
| 41 A dict in the following format: |
| 42 { |
| 43 'classes': [ |
| 44 { |
| 45 'class': '', |
| 46 'superclass': '', |
| 47 'annotations': {}, |
| 48 'methods': [ |
| 49 { |
| 50 'method': '', |
| 51 'annotations': {}, |
| 52 }, |
| 53 ... |
| 54 ], |
| 55 }, |
| 56 ... |
| 57 ], |
| 58 } |
| 59 """ |
| 60 |
| 61 with tempfile.NamedTemporaryFile() as proguard_output: |
| 62 cmd_helper.RunCmd(['java', '-jar', |
| 63 _PROGUARD_PATH, |
| 64 '-injars', jar_path, |
| 65 '-dontshrink', |
| 66 '-dontoptimize', |
| 67 '-dontobfuscate', |
| 68 '-dontpreverify', |
| 69 '-dump', proguard_output.name]) |
| 70 |
| 71 |
| 72 results = { |
| 73 'classes': [], |
| 74 } |
| 75 |
| 76 annotation = None |
| 77 annotation_has_value = False |
| 78 class_result = None |
| 79 method_result = None |
| 80 |
| 81 for line in proguard_output: |
| 82 line = line.strip('\r\n') |
| 83 |
| 84 m = _PROGUARD_CLASS_RE.match(line) |
| 85 if m: |
| 86 class_result = { |
| 87 'class': m.group(1).replace('/', '.'), |
| 88 'superclass': '', |
| 89 'annotations': {}, |
| 90 'methods': [], |
| 91 } |
| 92 results['classes'].append(class_result) |
| 93 annotation = None |
| 94 annotation_has_value = False |
| 95 method_result = None |
| 96 continue |
| 97 |
| 98 if not class_result: |
| 99 continue |
| 100 |
| 101 m = _PROGUARD_SUPERCLASS_RE.match(line) |
| 102 if m: |
| 103 class_result['superclass'] = m.group(1).replace('/', '.') |
| 104 continue |
| 105 |
| 106 m = _PROGUARD_SECTION_RE.match(line) |
| 107 if m: |
| 108 annotation = None |
| 109 annotation_has_value = False |
| 110 method_result = None |
| 111 continue |
| 112 |
| 113 m = _PROGUARD_METHOD_RE.match(line) |
| 114 if m: |
| 115 method_result = { |
| 116 'method': m.group(1), |
| 117 'annotations': {}, |
| 118 } |
| 119 class_result['methods'].append(method_result) |
| 120 annotation = None |
| 121 annotation_has_value = False |
| 122 continue |
| 123 |
| 124 m = _PROGUARD_ANNOTATION_RE.match(line) |
| 125 if m: |
| 126 # Ignore the annotation package. |
| 127 annotation = m.group(1).split('/')[-1] |
| 128 if method_result: |
| 129 method_result['annotations'][annotation] = None |
| 130 else: |
| 131 class_result['annotations'][annotation] = None |
| 132 continue |
| 133 |
| 134 if annotation: |
| 135 if not annotation_has_value: |
| 136 m = _PROGUARD_ANNOTATION_CONST_RE.match(line) |
| 137 annotation_has_value = bool(m) |
| 138 else: |
| 139 m = _PROGUARD_ANNOTATION_VALUE_RE.match(line) |
| 140 if m: |
| 141 if method_result: |
| 142 method_result['annotations'][annotation] = m.group(1) |
| 143 else: |
| 144 class_result['annotations'][annotation] = m.group(1) |
| 145 annotation_has_value = None |
| 146 |
| 147 return results |
| 148 |
OLD | NEW |