1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
|
#!/usr/bin/env python3
# Copyright 2017 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Collect, archive, and analyze Chrome's binary size."""
import argparse
import atexit
import logging
import pathlib
import platform
import resource
import sys
import archive
import console
import diff
import dex_disassembly
import file_format
import models
import native_disassembly
import os
def _LogPeakRamUsage():
peak_ram_usage = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
peak_ram_usage += resource.getrusage(resource.RUSAGE_CHILDREN).ru_maxrss
logging.info('Peak RAM usage was %d MB.', peak_ram_usage // 1024)
def _AddCommonArguments(parser):
parser.add_argument('-v',
'--verbose',
default=0,
action='count',
help='Verbose level (multiple times for more)')
class _PathResolver:
def __init__(self, parent_path):
self._parent_path = pathlib.Path(parent_path)
def __call__(self, subpath):
# Use dict to de-dupe while keeping order.
candidates = list({
self._parent_path / subpath: 0,
self._parent_path / pathlib.PosixPath(subpath).name: 0,
})
for p in candidates:
if p.exists():
return p
raise Exception('Paths do not exist: ' +
', '.join(str(t) for t in candidates))
class _DiffAction:
@staticmethod
def AddArguments(parser):
parser.add_argument('before', help='Before-patch .size file.')
parser.add_argument('after', help='After-patch .size file.')
parser.add_argument('--all', action='store_true', help='Verbose diff')
@staticmethod
def Run(args, on_config_error):
args.output_directory = None
args.inputs = [args.before, args.after]
args.query = '\n'.join([
'd = Diff()',
'sis = canned_queries.StaticInitializers(d.symbols)',
'count = sis.CountsByDiffStatus()[models.DIFF_STATUS_ADDED]',
'count += sis.CountsByDiffStatus()[models.DIFF_STATUS_REMOVED]',
'if count > 0:',
' print("Static Initializers Diff:")',
' Print(sis, summarize=False)',
' print()',
' print("Full diff:")',
'Print(d, verbose=%s)' % bool(args.all),
])
console.Run(args, on_config_error)
class _SaveDiffAction:
@staticmethod
def AddArguments(parser):
parser.add_argument('before', help='Before-patch .size file.')
parser.add_argument('after', help='After-patch .size file.')
parser.add_argument(
'output_file',
help='Write generated data to the specified .sizediff file.')
parser.add_argument('--title',
help='Value for the "title" build_config entry.')
parser.add_argument('--url', help='Value for the "url" build_config entry.')
parser.add_argument(
'--save-disassembly',
help='Adds the disassembly for the top 10 changed symbols.',
action='store_true')
parser.add_argument(
'--before-directory',
help='Defaults to directory containing before-patch .size file.')
parser.add_argument(
'--after-directory',
help='Defaults to directory containing after-patch .size file.')
@staticmethod
def Run(args, on_config_error):
if not args.before.endswith('.size'):
on_config_error('Before input must end with ".size"')
if not args.after.endswith('.size'):
on_config_error('After input must end with ".size"')
if not args.output_file.endswith('.sizediff'):
on_config_error('Output must end with ".sizediff"')
if args.save_disassembly:
if not args.before_directory:
args.before_directory = os.path.dirname(args.before)
if not args.after_directory:
args.after_directory = os.path.dirname(args.after)
before_size_info = archive.LoadAndPostProcessSizeInfo(args.before)
after_size_info = archive.LoadAndPostProcessSizeInfo(args.after)
# If a URL or title exists, we only want to add it to the build config of
# the after size file.
if args.title:
after_size_info.build_config[models.BUILD_CONFIG_TITLE] = args.title
if args.url:
after_size_info.build_config[models.BUILD_CONFIG_URL] = args.url
delta_size_info = diff.Diff(before_size_info, after_size_info)
if args.save_disassembly:
before_path_resolver = _PathResolver(args.before_directory)
after_path_resolver = _PathResolver(args.after_directory)
dex_disassembly.AddDisassembly(delta_size_info, before_path_resolver,
after_path_resolver)
native_disassembly.AddDisassembly(delta_size_info, before_path_resolver,
after_path_resolver)
file_format.SaveDeltaSizeInfo(delta_size_info, args.output_file)
def main():
parser = argparse.ArgumentParser(prog='supersize', description=__doc__)
sub_parsers = parser.add_subparsers()
actions = {}
actions['archive'] = (archive, 'Create a .size file')
actions['console'] = (
console,
'Starts an interactive Python console for analyzing .size files.')
actions['diff'] = (
_DiffAction(),
'Shorthand for console --query "Print(Diff())" (plus highlights static '
'initializers in diff)')
actions['save_diff'] = (
_SaveDiffAction(),
'Create a stand-alone .sizediff diff report from two .size files.')
for name, tup in actions.items():
sub_parser = sub_parsers.add_parser(name, help=tup[1])
_AddCommonArguments(sub_parser)
tup[0].AddArguments(sub_parser)
sub_parser.set_defaults(func=tup[0].Run)
# Show help if the command or a subcommand is called with no arguments
if len(sys.argv) == 1:
parser.print_help()
sys.exit(1)
elif len(sys.argv) == 2 and sys.argv[1] in actions:
parser.parse_args(sys.argv[1:] + ['-h'])
sys.exit(1)
args = parser.parse_args()
logging.basicConfig(level=logging.WARNING - args.verbose * 10,
format='%(levelname).1s %(relativeCreated)6d %(message)s')
if logging.getLogger().isEnabledFor(logging.DEBUG):
atexit.register(_LogPeakRamUsage)
def on_config_error(*args):
parser.error(*args)
args.func(args, on_config_error)
if __name__ == '__main__':
main()
|
