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