scripts/bloat-o-meter

   1 #!/usr/bin/env python3
   2 #
   3 # Copyright 2004 Matt Mackall <mpm@selenic.com>
   4 #
   5 # inspired by perl Bloat-O-Meter (c) 1997 by Andi Kleen
   6 #
   7 # This software may be used and distributed according to the terms
   8 # of the GNU General Public License, incorporated herein by reference.
   9
  10 import sys, os, re, argparse
  11 from signal import signal, SIGPIPE, SIG_DFL
  12
  13 signal(SIGPIPE, SIG_DFL)
  14
  15 parser = argparse.ArgumentParser(description="Simple script used to compare the symbol sizes of 2 object files")
  16 group = parser.add_mutually_exclusive_group()
  17 group.add_argument('-c', help='categorize output based on symbol type', action='store_true')
  18 group.add_argument('-d', help='Show delta of Data Section', action='store_true')
  19 group.add_argument('-t', help='Show delta of text Section', action='store_true')
  20 parser.add_argument('-p', dest='prefix', help='Arch prefix for the tool being used. Useful in cross build scenarios')
  21 parser.add_argument('file1', help='First file to compare')
  22 parser.add_argument('file2', help='Second file to compare')
  23
  24 args = parser.parse_args()
  25
  26 re_NUMBER = re.compile(r'\.[0-9]+')
  27
  28 def getsizes(file, format):
  29     sym = {}
  30     nm = "nm"
  31     if args.prefix:
  32         nm = "{}nm".format(args.prefix)
  33
  34     with os.popen("{} --size-sort {}".format(nm, file)) as f:
  35         for line in f:
  36             if line.startswith("\n") or ":" in line:
  37                 continue
  38             size, type, name = line.split()
  39             if type in format:
  40                 # strip generated symbols
  41                 if name.startswith("__mod_"): continue
  42                 if name.startswith("__se_sys"): continue
  43                 if name.startswith("__se_compat_sys"): continue
  44                 if name.startswith("__addressable_"): continue
  45                 if name == "linux_banner": continue
  46                 if name == "vermagic": continue
  47                 # statics and some other optimizations adds random .NUMBER
  48                 name = re_NUMBER.sub('', name)
  49                 sym[name] = sym.get(name, 0) + int(size, 16)
  50     return sym
  51
  52 def calc(oldfile, newfile, format):
  53     old = getsizes(oldfile, format)
  54     new = getsizes(newfile, format)
  55     grow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0
  56     delta, common = [], {}
  57     otot, ntot = 0, 0
  58
  59     for a in old:
  60         if a in new:
  61             common[a] = 1
  62
  63     for name in old:
  64         otot += old[name]
  65         if name not in common:
  66             remove += 1
  67             down += old[name]
  68             delta.append((-old[name], name))
  69
  70     for name in new:
  71         ntot += new[name]
  72         if name not in common:
  73             add += 1
  74             up += new[name]
  75             delta.append((new[name], name))
  76
  77     for name in common:
  78         d = new.get(name, 0) - old.get(name, 0)
  79         if d>0: grow, up = grow+1, up+d
  80         if d<0: shrink, down = shrink+1, down-d
  81         delta.append((d, name))
  82
  83     delta.sort(reverse=True)
  84     return grow, shrink, add, remove, up, down, delta, old, new, otot, ntot
  85
  86 def print_result(symboltype, symbolformat):
  87     grow, shrink, add, remove, up, down, delta, old, new, otot, ntot = \
  88     calc(args.file1, args.file2, symbolformat)
  89
  90     print("add/remove: %s/%s grow/shrink: %s/%s up/down: %s/%s (%s)" % \
  91           (add, remove, grow, shrink, up, -down, up-down))
  92     print("%-40s %7s %7s %+7s" % (symboltype, "old", "new", "delta"))
  93     for d, n in delta:
  94         if d: print("%-40s %7s %7s %+7d" % (n, old.get(n,"-"), new.get(n,"-"), d))
  95
  96     if otot:
  97         percent = (ntot - otot) * 100.0 / otot
  98     else:
  99         percent = 0
 100     print("Total: Before=%d, After=%d, chg %+.2f%%" % (otot, ntot, percent))
 101
 102 if args.c:
 103     print_result("Function", "tT")
 104     print_result("Data", "dDbB")
 105     print_result("RO Data", "rR")
 106 elif args.d:
 107     print_result("Data", "dDbBrR")
 108 elif args.t:
 109     print_result("Function", "tT")
 110 else:
 111     print_result("Function", "tTdDbBrR")