| Index: tools/binary_size/symbols.py
|
| diff --git a/tools/binary_size/symbols.py b/tools/binary_size/symbols.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..e92610abd3c1aaea500cb8e34fda652fc1e977c7
|
| --- /dev/null
|
| +++ b/tools/binary_size/symbols.py
|
| @@ -0,0 +1,207 @@
|
| +# Copyright 2017 The Chromium Authors. All rights reserved.
|
| +# Use of this source code is governed by a BSD-style license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +import collections
|
| +import re
|
| +
|
| +
|
| +SECTION_TO_SECTION_NAME = {
|
| + 'b': '.bss',
|
| + 'd': '.data',
|
| + 'r': '.rodata',
|
| + 't': '.text',
|
| +}
|
| +
|
| +
|
| +class BaseSymbol(object):
|
| + """Base class for Symbol and SymbolGroup."""
|
| + __slots__ = ()
|
| +
|
| + def __repr__(self):
|
| + return '%s@%x(size=%d,padding=%d,name=%s,path=%s)' % (
|
| + self.section_name, self.address, self.size_without_padding,
|
| + self.size - self.size_without_padding, self.name, self.path)
|
| +
|
| + @property
|
| + def section(self):
|
| + return self.section_name[1]
|
| +
|
| + @property
|
| + def size_without_padding(self):
|
| + return self.size - self.padding
|
| +
|
| + @property
|
| + def end_address(self):
|
| + return self.address + self.size_without_padding
|
| +
|
| + def IsGroup(self):
|
| + return False
|
| +
|
| + def IsGenerated(self):
|
| + return self.name and self.name.endswith(']') and (
|
| + not self.name.endswith('[]'))
|
| +
|
| +
|
| +class Symbol(BaseSymbol):
|
| + """Represents a single symbol within a binary."""
|
| +
|
| + __slots__ = (
|
| + 'section_name',
|
| + 'address',
|
| + 'size',
|
| + 'padding',
|
| + 'name',
|
| + 'function_signature',
|
| + 'path',
|
| + )
|
| +
|
| + def __init__(self, section_name, address, size_without_padding, name, path):
|
| + self.section_name = intern(section_name)
|
| + self.address = address
|
| + self.size = size_without_padding
|
| + self.padding = 0
|
| + self.name = name
|
| + self.function_signature = None
|
| + self.path = path
|
| +
|
| +
|
| +class SymbolGroup(BaseSymbol):
|
| + """Represents a group of symbols using the same interface as Symbol."""
|
| +
|
| + __slots__ = (
|
| + 'symbols',
|
| + 'filtered_symbols',
|
| + 'name',
|
| + 'section_name',
|
| + )
|
| +
|
| + def __init__(self, symbols, filtered_symbols=None, name=None,
|
| + section_name='**'):
|
| + self.symbols = symbols
|
| + self.filtered_symbols = filtered_symbols or []
|
| + self.name = name
|
| + self.section_name = section_name
|
| +
|
| + @property
|
| + def address(self):
|
| + return 0
|
| +
|
| + @property
|
| + def function_signature(self):
|
| + return None
|
| +
|
| + @property
|
| + def path(self):
|
| + return None
|
| +
|
| + @property
|
| + def size(self):
|
| + return sum(s.size for s in self)
|
| +
|
| + @property
|
| + def padding(self):
|
| + return sum(s.padding for s in self)
|
| +
|
| + def __iter__(self):
|
| + return iter(self.symbols)
|
| +
|
| + def __len__(self):
|
| + return len(self.symbols)
|
| +
|
| + def IsGroup(self):
|
| + return True
|
| +
|
| + def Sorted(self, cmp_func=None, key=None, reverse=False):
|
| + # Default to sorting by size then name.
|
| + if cmp_func is None and key is None:
|
| + cmp_func = lambda a, b: cmp((b.size, a.name), (a.size, b.name))
|
| +
|
| + new_symbols = sorted(self.symbols, cmp_func, key, reverse)
|
| + return self.__class__(new_symbols, filtered_symbols=self.filtered_symbols,
|
| + section_name=self.section_name)
|
| +
|
| + def Filter(self, func, include_filtered=False):
|
| + filtered_and_kept = ([], [])
|
| + for symbol in self:
|
| + filtered_and_kept[int(bool(func(symbol)))].append(symbol)
|
| +
|
| + if include_filtered:
|
| + filtered_and_kept[0].extend(self.filtered_symbols)
|
| + return self.__class__(filtered_and_kept[1],
|
| + filtered_symbols=filtered_and_kept[0],
|
| + section_name=self.section_name)
|
| +
|
| + def WhereBiggerThan(self, min_size, include_filtered=False):
|
| + return self.Filter(lambda s: s.size >= min_size,
|
| + include_filtered=include_filtered)
|
| +
|
| + def WhereInSection(self, section, include_filtered=False):
|
| + ret = self.Filter(lambda s: s.section == section,
|
| + include_filtered=include_filtered)
|
| + ret.section_name = SECTION_TO_SECTION_NAME[section]
|
| + return ret
|
| +
|
| + def WhereNameMatches(self, pattern, include_filtered=False):
|
| + regex = re.compile(pattern)
|
| + return self.Filter(lambda s: s.name and regex.search(s.name),
|
| + include_filtered=include_filtered)
|
| +
|
| + def WherePathMatches(self, pattern, include_filtered=False):
|
| + regex = re.compile(pattern)
|
| + return self.Filter(lambda s: s.path and regex.search(s.path),
|
| + include_filtered=include_filtered)
|
| +
|
| + def WhereAddressInRange(self, start, end, include_filtered=False):
|
| + return self.Filter(lambda s: s.address >= start and s.address <= end,
|
| + include_filtered=include_filtered)
|
| +
|
| + def WhereHasAnyAttribution(self, include_filtered=False):
|
| + return self.Filter(lambda s: s.name or s.path,
|
| + include_filtered=include_filtered)
|
| +
|
| + def Inverted(self):
|
| + return self.__class__(self.filtered_symbols, filtered_symbols=self.symbols)
|
| +
|
| + def GroupBy(self, func, include_filtered=False):
|
| + new_syms = []
|
| + filtered_symbols = []
|
| + symbols_by_token = collections.defaultdict(list)
|
| + for symbol in self:
|
| + token = func(symbol)
|
| + if not token:
|
| + filtered_symbols.append(symbol)
|
| + continue
|
| + symbols_by_token[token].append(symbol)
|
| + for token, symbols in symbols_by_token.iteritems():
|
| + new_syms.append(self.__class__(symbols, name=token,
|
| + section_name=self.section_name))
|
| + if include_filtered:
|
| + filtered_symbols.extend(self.filtered_symbols)
|
| + return self.__class__(new_syms, filtered_symbols=filtered_symbols,
|
| + section_name=self.section_name)
|
| +
|
| + def GroupByNamespace(self, depth=1, include_filtered=False):
|
| + def extract_namespace(symbol):
|
| + # Does not distinguish between classes and namespaces.
|
| + if symbol.name:
|
| + idx = -2
|
| + for _ in xrange(depth):
|
| + idx = symbol.name.find('::', idx + 2)
|
| + if idx != -1:
|
| + ret = symbol.name[:idx]
|
| + if '<' not in ret:
|
| + return ret
|
| + return '{global}'
|
| + return self.GroupBy(extract_namespace, include_filtered=include_filtered)
|
| +
|
| + def GroupByPath(self, depth=1, include_filtered=False):
|
| + def extract_path(symbol):
|
| + if symbol.path:
|
| + idx = -1
|
| + for _ in xrange(depth):
|
| + idx = symbol.path.find('/', idx + 1)
|
| + if idx != -1:
|
| + return symbol.path[:idx]
|
| + return None
|
| + return self.GroupBy(extract_path, include_filtered=include_filtered)
|
|
|