build/android/pylib/utils/proguard.py

   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