xref: /openbmc/linux/scripts/bloat-o-meter (revision 0d7bbb43641c35390378e951785f2351bc36650a)
1d960600dSMatt Mackall#!/usr/bin/python
2d960600dSMatt Mackall#
3d960600dSMatt Mackall# Copyright 2004 Matt Mackall <mpm@selenic.com>
4d960600dSMatt Mackall#
5d960600dSMatt Mackall# inspired by perl Bloat-O-Meter (c) 1997 by Andi Kleen
6d960600dSMatt Mackall#
7d960600dSMatt Mackall# This software may be used and distributed according to the terms
8d960600dSMatt Mackall# of the GNU General Public License, incorporated herein by reference.
9d960600dSMatt Mackall
10d960600dSMatt Mackallimport sys, os, re
11eef06b82SAlexey Dobriyanfrom signal import signal, SIGPIPE, SIG_DFL
12eef06b82SAlexey Dobriyan
13eef06b82SAlexey Dobriyansignal(SIGPIPE, SIG_DFL)
14d960600dSMatt Mackall
15d960600dSMatt Mackallif len(sys.argv) != 3:
16d960600dSMatt Mackall    sys.stderr.write("usage: %s file1 file2\n" % sys.argv[0])
17d960600dSMatt Mackall    sys.exit(-1)
18d960600dSMatt Mackall
19*0d7bbb43SAlexey Dobriyanre_NUMBER = re.compile(r'\.[0-9]+')
20*0d7bbb43SAlexey Dobriyan
21d960600dSMatt Mackalldef getsizes(file):
22d960600dSMatt Mackall    sym = {}
233af06fd9SAlexey Dobriyan    with os.popen("nm --size-sort " + file) as f:
243af06fd9SAlexey Dobriyan        for line in f:
253af06fd9SAlexey Dobriyan            size, type, name = line.split()
26c50e3f51SJean Delvare            if type in "tTdDbBrR":
27c50e3f51SJean Delvare                # strip generated symbols
28c2e182faSJosh Triplett                if name.startswith("__mod_"): continue
29b25c2ff5SJosh Triplett                if name.startswith("SyS_"): continue
30b25c2ff5SJosh Triplett                if name.startswith("compat_SyS_"): continue
315a7b2d27SJosh Triplett                if name == "linux_banner": continue
3221cf6e58SAndi Kleen                # statics and some other optimizations adds random .NUMBER
33*0d7bbb43SAlexey Dobriyan                name = re_NUMBER.sub('', name)
3451849738SRob Landley                sym[name] = sym.get(name, 0) + int(size, 16)
35d960600dSMatt Mackall    return sym
36d960600dSMatt Mackall
37d960600dSMatt Mackallold = getsizes(sys.argv[1])
38d960600dSMatt Mackallnew = getsizes(sys.argv[2])
39d960600dSMatt Mackallgrow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0
40d960600dSMatt Mackalldelta, common = [], {}
41b21e91c3SVineet Guptaotot, ntot = 0, 0
42d960600dSMatt Mackall
43d960600dSMatt Mackallfor a in old:
44d960600dSMatt Mackall    if a in new:
45d960600dSMatt Mackall        common[a] = 1
46d960600dSMatt Mackall
47d960600dSMatt Mackallfor name in old:
48b21e91c3SVineet Gupta    otot += old[name]
49d960600dSMatt Mackall    if name not in common:
50d960600dSMatt Mackall        remove += 1
51d960600dSMatt Mackall        down += old[name]
52d960600dSMatt Mackall        delta.append((-old[name], name))
53d960600dSMatt Mackall
54d960600dSMatt Mackallfor name in new:
55b21e91c3SVineet Gupta    ntot += new[name]
56d960600dSMatt Mackall    if name not in common:
57d960600dSMatt Mackall        add += 1
58d960600dSMatt Mackall        up += new[name]
59d960600dSMatt Mackall        delta.append((new[name], name))
60d960600dSMatt Mackall
61d960600dSMatt Mackallfor name in common:
62d960600dSMatt Mackall        d = new.get(name, 0) - old.get(name, 0)
63d960600dSMatt Mackall        if d>0: grow, up = grow+1, up+d
64d960600dSMatt Mackall        if d<0: shrink, down = shrink+1, down-d
65d960600dSMatt Mackall        delta.append((d, name))
66d960600dSMatt Mackall
67d960600dSMatt Mackalldelta.sort()
68d960600dSMatt Mackalldelta.reverse()
69d960600dSMatt Mackall
7072214a24SSergey Senozhatskyprint("add/remove: %s/%s grow/shrink: %s/%s up/down: %s/%s (%s)" % \
7172214a24SSergey Senozhatsky      (add, remove, grow, shrink, up, -down, up-down))
7272214a24SSergey Senozhatskyprint("%-40s %7s %7s %+7s" % ("function", "old", "new", "delta"))
73d960600dSMatt Mackallfor d, n in delta:
7472214a24SSergey Senozhatsky    if d: print("%-40s %7s %7s %+7d" % (n, old.get(n,"-"), new.get(n,"-"), d))
75b21e91c3SVineet Gupta
768cde0dafSRiku Voipioprint("Total: Before=%d, After=%d, chg %+.2f%%" % \
778cde0dafSRiku Voipio    (otot, ntot, (ntot - otot)*100.0/otot))
78