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
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            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                # statics and some other optimizations adds random .NUMBER
40                name = re_NUMBER.sub('', name)
41                sym[name] = sym.get(name, 0) + int(size, 16)
42    return sym
43
44def calc(oldfile, newfile, format):
45    old = getsizes(oldfile, format)
46    new = getsizes(newfile, format)
47    grow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0
48    delta, common = [], {}
49    otot, ntot = 0, 0
50
51    for a in old:
52        if a in new:
53            common[a] = 1
54
55    for name in old:
56        otot += old[name]
57        if name not in common:
58            remove += 1
59            down += old[name]
60            delta.append((-old[name], name))
61
62    for name in new:
63        ntot += new[name]
64        if name not in common:
65            add += 1
66            up += new[name]
67            delta.append((new[name], name))
68
69    for name in common:
70        d = new.get(name, 0) - old.get(name, 0)
71        if d>0: grow, up = grow+1, up+d
72        if d<0: shrink, down = shrink+1, down-d
73        delta.append((d, name))
74
75    delta.sort()
76    delta.reverse()
77    return grow, shrink, add, remove, up, down, delta, old, new, otot, ntot
78
79def print_result(symboltype, symbolformat, argc):
80    grow, shrink, add, remove, up, down, delta, old, new, otot, ntot = \
81    calc(sys.argv[argc - 1], sys.argv[argc], symbolformat)
82
83    print("add/remove: %s/%s grow/shrink: %s/%s up/down: %s/%s (%s)" % \
84          (add, remove, grow, shrink, up, -down, up-down))
85    print("%-40s %7s %7s %+7s" % (symboltype, "old", "new", "delta"))
86    for d, n in delta:
87        if d: print("%-40s %7s %7s %+7d" % (n, old.get(n,"-"), new.get(n,"-"), d))
88
89    if otot:
90        percent = (ntot - otot) * 100.0 / otot
91    else:
92        percent = 0
93    print("Total: Before=%d, After=%d, chg %+.2f%%" % (otot, ntot, percent))
94
95if sys.argv[1] == "-c":
96    print_result("Function", "tT", 3)
97    print_result("Data", "dDbB", 3)
98    print_result("RO Data", "rR", 3)
99elif sys.argv[1] == "-d":
100    print_result("Data", "dDbBrR", 3)
101elif sys.argv[1] == "-t":
102    print_result("Function", "tT", 3)
103else:
104    print_result("Function", "tTdDbBrR", 2)
105