scripts/bloat-o-meter

   1 #!/usr/bin/python
   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
  12 def usage():
  13     sys.stderr.write("usage: %s [-t] file1 file2\n" % sys.argv[0])
  14     sys.exit(-1)
  15
  16 f1, f2 = (None, None)
  17 flag_timing, dashes = (False, False)
  18
  19 for f in sys.argv[1:]:
  20     if f.startswith("-"):
  21         if f == "--": # sym_args
  22             dashes = True
  23             break
  24         if f == "-t": # timings
  25             flag_timing = True
  26     else:
  27         if not os.path.exists(f):
  28             sys.stderr.write("Error: file '%s' does not exist\n" % f)
  29             usage()
  30         if f1 is None:
  31             f1 = f
  32         elif f2 is None:
  33             f2 = f
  34 if flag_timing:
  35     import time
  36 if f1 is None or f2 is None:
  37     usage()
  38
  39 sym_args = " ".join(sys.argv[3 + flag_timing + dashes:])
  40 def getsizes(file):
  41     sym, alias, lut = {}, {}, {}
  42     #dynsym_filter = re.compile("^\d+:\s+[\dA-Fa-f]+\s+\d+\s+\w+\s+\w+\s+\w+\s+\w+\s+\w+$")
  43     for l in os.popen("readelf -W -s %s %s" % (sym_args, file)).readlines():
  44         if True:
  45             l = l.strip()
  46             if not (len(l) and l[0].isdigit() and len(l.split()) == 8):
  47                 continue
  48             num, value, size, typ, bind, vis, ndx, name = l.split()
  49             if ndx == "UND": continue # skip undefined
  50             if typ in ["SECTION", "FILES"]: continue # skip sections and files
  51         #else:
  52         #    l = l.strip()
  53         #    match = dynsym_filter.match(l)
  54         #    if not match: continue
  55         #    x, value, size, typ, bind, x, ndx, name = l.split()
  56         #    if ndx == "UND": continue # skip undefined
  57         #    if typ in ["SECTION", "FILES"]: continue # skip sections and files
  58         if "." in name: name = "static." + name.split(".")[0]
  59         value = int(value, 16)
  60         size = int(size)
  61         if vis != "DEFAULT" and bind != "GLOBAL": # see if it is an alias
  62             alias[(value, size)] = {"name" : name}
  63         else:
  64             sym[name] = {"addr" : value, "size":  size}
  65             lut[(value, size)] = 0
  66     for addr, sz in alias.iterkeys():
  67         # If the non-GLOBAL sym has an implementation elsewhere then
  68         # it's an alias, disregard it.
  69         if not (addr, sz) in lut:
  70             # If this non-GLOBAL sym does not have an implementation at
  71             # another address, then treat it as a normal symbol.
  72             sym[alias[(addr, sz)]["name"]] = {"addr" : addr, "size": sz}
  73     for l in os.popen("readelf -W -S " + file).readlines():
  74         x = l.split()
  75         if len(x)<6: continue
  76         # Should take these into account too!
  77         #if x[1] not in [".text", ".rodata", ".symtab", ".strtab"]: continue
  78         if x[1] not in [".rodata"]: continue
  79         sym[x[1]] = {"addr" : int(x[3], 16), "size" : int(x[5], 16)}
  80     return sym
  81
  82 if flag_timing:
  83     start_t1 = int(time.time() * 1e9)
  84 old = getsizes(f1)
  85 if flag_timing:
  86     end_t1 = int(time.time() * 1e9)
  87     start_t2 = int(time.time() * 1e9)
  88 new = getsizes(f2)
  89 if flag_timing:
  90     end_t2 = int(time.time() * 1e9)
  91     start_t3 = int(time.time() * 1e9)
  92 grow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0
  93 delta, common = [], {}
  94
  95 for name in old.iterkeys():
  96     if name in new:
  97         common[name] = 1
  98
  99 for name in old:
 100     if name not in common:
 101         remove += 1
 102         sz = old[name]["size"]
 103         down += sz
 104         delta.append((-sz, name))
 105
 106 for name in new:
 107     if name not in common:
 108         add += 1
 109         sz = new[name]["size"]
 110         up += sz
 111         delta.append((sz, name))
 112
 113 for name in common:
 114         d = new[name].get("size", 0) - old[name].get("size", 0)
 115         if d>0: grow, up = grow+1, up+d
 116         elif d<0: shrink, down = shrink+1, down-d
 117         else:
 118             continue
 119         delta.append((d, name))
 120
 121 delta.sort()
 122 delta.reverse()
 123 if flag_timing:
 124     end_t3 = int(time.time() * 1e9)
 125
 126 print("%-48s %7s %7s %+7s" % ("function", "old", "new", "delta"))
 127 for d, n in delta:
 128     if d:
 129         old_sz = old.get(n, {}).get("size", "-")
 130         new_sz = new.get(n, {}).get("size", "-")
 131         print("%-48s %7s %7s %+7d" % (n, old_sz, new_sz, d))
 132 print("-"*78)
 133 total="(add/remove: %s/%s grow/shrink: %s/%s up/down: %s/%s)%%sTotal: %s bytes"\
 134     % (add, remove, grow, shrink, up, -down, up-down)
 135 print(total % (" "*(80-len(total))))
 136 if flag_timing:
 137     print("\n%d/%d; %d Parse origin/new; processing nsecs" %
 138         (end_t1-start_t1, end_t2-start_t2, end_t3-start_t3))
 139     print("total nsecs: %d" % (end_t3-start_t1))