Emit partial results if binary_size is interrupted.

tools/binary_size/run_binary_size_analysis.py

 #!/usr/bin/env python
 # Copyright 2014 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.
 
 """Generate a spatial analysis against an arbitrary library.
 
 To use, build the 'binary_size_tool' target. Then run this tool, passing
 in the location of the library to be analyzed along with any other options
 you desire.
@@ skipping 437 matching lines @@
         else:
           speed = 0
         progress_percent = (100.0 * (progress.count + progress.skip_count) /
                             nm_output_lines_len)
         print('%.1f%%: Looked up %d symbols (%d collisions) - %.1f lookups/s.' %
               (progress_percent, progress.count, progress.collisions, speed))
 
   symbolizer = elf_symbolizer.ELFSymbolizer(library, addr2line_binary,
                                             map_address_symbol,
                                             max_concurrent_jobs=jobs)
-  for line in nm_output_lines:
-    match = sNmPattern.match(line)
-    if match:
-      location = match.group(5)
-      if not location:
-        addr = int(match.group(1), 16)
-        size = int(match.group(2), 16)
-        if addr in address_symbol:  # Already looked up, shortcut ELFSymbolizer.
-          map_address_symbol(address_symbol[addr], addr)
-          continue
-        elif size == 0:
-          # Save time by not looking up empty symbols (do they even exist?)
-          print('Empty symbol: ' + line)
-        else:
-          symbolizer.SymbolizeAsync(addr, addr)
-          continue
+  user_interrupted = False
+  try:
+    for line in nm_output_lines:
+      match = sNmPattern.match(line)
+      if match:
+        location = match.group(5)
+        if not location:
+          addr = int(match.group(1), 16)
+          size = int(match.group(2), 16)
+          if addr in address_symbol:  # Already looked up, shortcut
+                                      # ELFSymbolizer.
+            map_address_symbol(address_symbol[addr], addr)
+            continue
+          elif size == 0:
+            # Save time by not looking up empty symbols (do they even exist?)
+            print('Empty symbol: ' + line)
+          else:
+            symbolizer.SymbolizeAsync(addr, addr)
+            continue
 
-    progress.skip_count += 1
+      progress.skip_count += 1
+  except KeyboardInterrupt:
+    user_interrupted = True
 
-  symbolizer.Join()
+  if user_interrupted:
+    print('Interrupting - killing subprocesses.')

[Primiano Tucci (use gerrit), 2014/05/27 09:31:46]

Can you move this (and indent) to line 481, in the execpt block, and avoid
checking the condition here (like you did below on 490)?

+
+  try:
+    symbolizer.Join()
+  except KeyboardInterrupt:
+    # Don't want to abort here since we will be finished in a few seconds.
+    user_interrupted = True
+    print('Patience you must have my young padawan.')

[Primiano Tucci (use gerrit), 2014/05/27 09:31:46]

:-D

+
+  if user_interrupted:
+    print('Skipping the rest of the file mapping. '
+          'Output will not be fully classified.')
 
   with open(outfile, 'w') as out:
     for line in nm_output_lines:
       match = sNmPattern.match(line)
       if match:
         location = match.group(5)
         if not location:
           addr = int(match.group(1), 16)
-          symbol = address_symbol[addr]
-          path = '??'
-          if symbol.source_path is not None:
-            path = symbol.source_path
-          line_number = 0
-          if symbol.source_line is not None:
-            line_number = symbol.source_line
-          out.write('%s\t%s:%d\n' % (line, path, line_number))
-          continue
+          symbol = address_symbol.get(addr)
+          if symbol is not None:
+            path = '??'
+            if symbol.source_path is not None:
+              path = symbol.source_path
+            line_number = 0
+            if symbol.source_line is not None:
+              line_number = symbol.source_line
+            out.write('%s\t%s:%d\n' % (line, path, line_number))
+            continue
 
       out.write('%s\n' % line)
 
   print('%d symbols in the results.' % len(address_symbol))
 
 
 def RunNm(binary, nm_binary):
   print('Starting nm')
-  cmd = [nm_binary, '-C', '--print-size', binary]
+  cmd = [nm_binary, '-C', '--print-size', '--size-sort', '--reverse-sort',
+         binary]
   nm_process = subprocess.Popen(cmd,
                                 stdout=subprocess.PIPE,
                                 stderr=subprocess.PIPE)
   (process_output, err_output) = nm_process.communicate()
 
   if nm_process.returncode != 0:
     if err_output:
       raise Exception, err_output
     else:
       raise Exception, process_output
@@ skipping 157 matching lines @@
     if not os.path.exists(d3_out):
       os.makedirs(d3_out, 0755)
     d3_src = os.path.join(os.path.dirname(__file__),
                           '..',
                           '..',
                           'third_party', 'd3', 'src')
     template_src = os.path.join(os.path.dirname(__file__),
                                 'template')
     shutil.copy(os.path.join(d3_src, 'LICENSE'), d3_out)
     shutil.copy(os.path.join(d3_src, 'd3.js'), d3_out)
-    print('Copying index.html')
     shutil.copy(os.path.join(template_src, 'index.html'), opts.destdir)
     shutil.copy(os.path.join(template_src, 'D3SymbolTreeMap.js'), opts.destdir)
 
-  if opts.verbose:
-    print 'Report saved to ' + opts.destdir + '/index.html'
+  print 'Report saved to ' + opts.destdir + '/index.html'
 
 
 if __name__ == '__main__':
   sys.exit(main())