xref: /openbmc/linux/scripts/bloat-o-meter (revision 2bc7d3e0) 
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                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
45def 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
80def 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
96if sys.argv[1] == "-c":
97    print_result("Function", "tT", 3)
98    print_result("Data", "dDbB", 3)
99    print_result("RO Data", "rR", 3)
100elif sys.argv[1] == "-d":
101    print_result("Data", "dDbBrR", 3)
102elif sys.argv[1] == "-t":
103    print_result("Function", "tT", 3)
104else:
105    print_result("Function", "tTdDbBrR", 2)
106