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
  11 from signal import signal, SIGPIPE, SIG_DFL
  12
  13 signal(SIGPIPE, SIG_DFL)
  14
  15 if len(sys.argv) < 3:
  16     sys.stderr.write("usage: %s [option] file1 file2\n" % sys.argv[0])
  17     sys.stderr.write("The options are:\n")
  18     sys.stderr.write("-c        categorize output based on symbol type\n")
  19     sys.stderr.write("-d        Show delta of Data Section\n")
  20     sys.stderr.write("-t        Show delta of text Section\n")
  21     sys.exit(-1)
  22
  23 re_NUMBER = re.compile(r'\.[0-9]+')
  24
  25 def getsizes(file, format):
  26     sym = {}
  27     with os.popen("nm --size-sort " + file) as f:
  28         for line in f:
  29             if line.startswith("\n") or ":" in line:
  30                 continue
  31             size, type, name = line.split()
  32             if type in format:
  33                 # strip generated symbols
  34                 if name.startswith("__mod_"): continue
  35                 if name.startswith("__se_sys"): continue
  36                 if name.startswith("__se_compat_sys"): continue
  37                 if name.startswith("__addressable_"): continue
  38                 if name == "linux_banner": continue
  39                 if name == "vermagic": continue
  40                 # statics and some other optimizations adds random .NUMBER
  41                 name = re_NUMBER.sub('', name)
  42                 sym[name] = sym.get(name, 0) + int(size, 16)
  43     return sym
  44
  45 def calc(oldfile, newfile, format):
  46     old = getsizes(oldfile, format)
  47     new = getsizes(newfile, format)
  48     grow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0
  49     delta, common = [], {}
  50     otot, ntot = 0, 0
  51
  52     for a in old:
  53         if a in new:
  54             common[a] = 1
  55
  56     for name in old:
  57         otot += old[name]
  58         if name not in common:
  59             remove += 1
  60             down += old[name]
  61             delta.append((-old[name], name))
  62
  63     for name in new:
  64         ntot += new[name]
  65         if name not in common:
  66             add += 1
  67             up += new[name]
  68             delta.append((new[name], name))
  69
  70     for name in common:
  71         d = new.get(name, 0) - old.get(name, 0)
  72         if d>0: grow, up = grow+1, up+d
  73         if d<0: shrink, down = shrink+1, down-d
  74         delta.append((d, name))
  75
  76     delta.sort()
  77     delta.reverse()
  78     return grow, shrink, add, remove, up, down, delta, old, new, otot, ntot
  79
  80 def print_result(symboltype, symbolformat, argc):
  81     grow, shrink, add, remove, up, down, delta, old, new, otot, ntot = \
  82     calc(sys.argv[argc - 1], sys.argv[argc], symbolformat)
  83
  84     print("add/remove: %s/%s grow/shrink: %s/%s up/down: %s/%s (%s)" % \
  85           (add, remove, grow, shrink, up, -down, up-down))
  86     print("%-40s %7s %7s %+7s" % (symboltype, "old", "new", "delta"))
  87     for d, n in delta:
  88         if d: print("%-40s %7s %7s %+7d" % (n, old.get(n,"-"), new.get(n,"-"), d))
  89
  90     if otot:
  91         percent = (ntot - otot) * 100.0 / otot
  92     else:
  93         percent = 0
  94     print("Total: Before=%d, After=%d, chg %+.2f%%" % (otot, ntot, percent))
  95
  96 if sys.argv[1] == "-c":
  97     print_result("Function", "tT", 3)
  98     print_result("Data", "dDbB", 3)
  99     print_result("RO Data", "rR", 3)
 100 elif sys.argv[1] == "-d":
 101     print_result("Data", "dDbBrR", 3)
 102 elif sys.argv[1] == "-t":
 103     print_result("Function", "tT", 3)
 104 else:
 105     print_result("Function", "tTdDbBrR", 2)