supersize: Track symbol aliases and shared symbols

tools/binary_size/libsupersize/file_format.py

 # 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.
 
 """Deals with loading & saving .size files."""
 
 import cStringIO
 import calendar
 import collections
 import datetime
@@ skipping 62 matching lines @@
 
   write_numeric(lambda s: s.address, delta=True)
   _LogSize(file_obj, 'addresses')  # For libchrome, adds 300kb.
   # Do not write padding, it will be recalcualted from addresses on load.
   write_numeric(lambda s: s.size_without_padding)
   _LogSize(file_obj, 'sizes')  # For libchrome, adds 300kb
   write_numeric(lambda s: path_tuples[(s.object_path, s.source_path)],
                 delta=True)
   _LogSize(file_obj, 'path indices')  # For libchrome: adds 125kb.
 
+  prev_aliases = None
   for group in by_section:
     for symbol in group:
       # Do not write name when full_name exists. It will be derived on load.
       file_obj.write(symbol.full_name or symbol.name)
+      if symbol.aliases and symbol.aliases is not prev_aliases:
+        file_obj.write('\t0%x' % symbol.num_aliases)
+      prev_aliases = symbol.aliases
       if symbol.flags:
         file_obj.write('\t%x' % symbol.flags)
       file_obj.write('\n')
   _LogSize(file_obj, 'names (final)')  # For libchrome: adds 3.5mb.
 
 
 def _LoadSizeInfoFromFile(file_obj):
   """Loads a size_info from the given file."""
   lines = iter(file_obj)
   next(lines)  # Comment line.
@@ skipping 28 matching lines @@
       ret.append(fields)
     return ret
 
   addresses = read_numeric(delta=True)
   sizes = read_numeric(delta=False)
   path_indices = read_numeric(delta=True)
 
   raw_symbols = [None] * sum(section_counts)
   symbol_idx = 0
   for section_index, cur_section_name in enumerate(section_names):
+    alias_counter = 0
     for i in xrange(section_counts[section_index]):
-      line = next(lines)[:-1]
-      name = line
-      flags = 0
-      last_tab_idx = line.find('\t', -3)  # Allows for two digits of flags.
-      if last_tab_idx != -1:
-        flags = int(line[last_tab_idx + 1:], 16)
-        name = line[:last_tab_idx]
+      parts = next(lines)[:-1].split('\t')
+      flags_part = None
+      aliases_part = None
+
+      if len(parts) == 3:
+        aliases_part = parts[1]
+        flags_part = parts[2]
+      elif len(parts) == 2:
+        if parts[1][0] == '0':
+          aliases_part = parts[1]
+        else:
+          flags_part = parts[1]
+
+      name = parts[0]
+      flags = int(flags_part, 16) if flags_part else 0
+      num_aliases = int(aliases_part, 16) if aliases_part else 0
 
       new_sym = models.Symbol.__new__(models.Symbol)
       new_sym.section_name = cur_section_name
       new_sym.address = addresses[section_index][i]
       new_sym.size = sizes[section_index][i]
       new_sym.name = name
       paths = path_tuples[path_indices[section_index][i]]
       new_sym.object_path = paths[0]
       new_sym.source_path = paths[1]
       new_sym.flags = flags
       new_sym.padding = 0  # Derived
       new_sym.full_name = None  # Derived
+
+      if num_aliases:
+        assert alias_counter == 0
+        new_sym.aliases = [new_sym]
+        alias_counter = num_aliases - 1
+      elif alias_counter > 0:
+        new_sym.aliases = raw_symbols[symbol_idx - 1].aliases
+        new_sym.aliases.append(new_sym)
+        alias_counter -= 1
+      else:
+        new_sym.aliases = None
+
       raw_symbols[symbol_idx] = new_sym
       symbol_idx += 1
 
   return models.SizeInfo(section_sizes, raw_symbols, metadata=metadata)
 
 
 def SaveSizeInfo(size_info, path):
   """Saves |size_info| to |path}."""
-  if os.environ.get('MEASURE_GZIP') == '1':
+  if os.environ.get('SUPERSIZE_MEASURE_GZIP') == '1':
     with gzip.open(path, 'wb') as f:
       _SaveSizeInfoToFile(size_info, f)
   else:
     # It is seconds faster to do gzip in a separate step. 6s -> 3.5s.
     stringio = cStringIO.StringIO()
     _SaveSizeInfoToFile(size_info, stringio)
 
     logging.debug('Serialization complete. Gzipping...')
     stringio.seek(0)
     with gzip.open(path, 'wb') as f:
       shutil.copyfileobj(stringio, f)
 
 
 def LoadSizeInfo(path):
   """Returns a SizeInfo loaded from |path|."""
   with gzip.open(path) as f:
     return _LoadSizeInfoFromFile(f)