Annotation of /trunk/mkinitrd-magellan/busybox/scripts/bloat-o-meter

#!/usr/bin/python
#
# Copyright 2004 Matt Mackall <mpm@selenic.com>
#
# inspired by perl Bloat-O-Meter (c) 1997 by Andi Kleen
#
# This software may be used and distributed according to the terms
# of the GNU General Public License, incorporated herein by reference.

import sys, os, re

def usage():
    sys.stderr.write("usage: %s file1 file2\n" % sys.argv[0])
    sys.exit(-1)

if len(sys.argv) < 3:
    usage()

for f in sys.argv[1:3]:
    if not os.path.exists(f):
        sys.stderr.write("Error: file '%s' does not exist\n" % f)
        usage()

nm_args = " ".join(sys.argv[3:])
def getsizes(file):
    sym = {}
    for l in os.popen("nm --size-sort %s %s" % (nm_args, file)).readlines():
        l = l.strip()
        # Skip empty lines
        if not len(l): continue
        # Skip archive members
        if len(l.split()) == 1 and l.endswith(':'):
          continue
        size, type, name = l.split()
        if type in "tTdDbBrR":
            if "." in name: name = "static." + name.split(".")[0]
            sym[name] = sym.get(name, 0) + int(size, 16)
    for l in os.popen("readelf -S " + file).readlines():
        x = l.split()
        if len(x)<6 or x[1] != ".rodata": continue
        sym[".rodata"] = int(x[5], 16)
    return sym

old = getsizes(sys.argv[1])
new = getsizes(sys.argv[2])
grow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0
delta, common = [], {}

for a in old:
    if a in new:
        common[a] = 1

for name in old:
    if name not in common:
        remove += 1
        down += old[name]
        delta.append((-old[name], name))

for name in new:
    if name not in common:
        add += 1
        up += new[name]
        delta.append((new[name], name))

for name in common:
        d = new.get(name, 0) - old.get(name, 0)
        if d>0: grow, up = grow+1, up+d
        if d<0: shrink, down = shrink+1, down-d
        delta.append((d, name))

delta.sort()
delta.reverse()

print "%-48s %7s %7s %+7s" % ("function", "old", "new", "delta")
for d, n in delta:
    if d: print "%-48s %7s %7s %+7d" % (n, old.get(n,"-"), new.get(n,"-"), d)
print "-"*78
total="(add/remove: %s/%s grow/shrink: %s/%s up/down: %s/%s)%%sTotal: %s bytes"\
    % (add, remove, grow, shrink, up, -down, up-down)
print total % (" "*(80-len(total)))
1	niro	532	#!/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	niro	816	def usage():
13	niro	532	sys.stderr.write("usage: %s file1 file2\n" % sys.argv[0])
14			sys.exit(-1)
15
16	niro	816	if len(sys.argv) < 3:
17			usage()
18
19	niro	984	for f in sys.argv[1:3]:
20	niro	816	if not os.path.exists(f):
21			sys.stderr.write("Error: file '%s' does not exist\n" % f)
22			usage()
23
24	niro	984	nm_args = " ".join(sys.argv[3:])
25	niro	532	def getsizes(file):
26			sym = {}
27	niro	816	for l in os.popen("nm --size-sort %s %s" % (nm_args, file)).readlines():
28	niro	984	l = l.strip()
29			# Skip empty lines
30	niro	816	if not len(l): continue
31	niro	984	# Skip archive members
32	niro	816	if len(l.split()) == 1 and l.endswith(':'):
33			continue
34			size, type, name = l.split()
35	niro	532	if type in "tTdDbBrR":
36			if "." in name: name = "static." + name.split(".")[0]
37			sym[name] = sym.get(name, 0) + int(size, 16)
38			for l in os.popen("readelf -S " + file).readlines():
39			x = l.split()
40			if len(x)<6 or x[1] != ".rodata": continue
41			sym[".rodata"] = int(x[5], 16)
42			return sym
43
44			old = getsizes(sys.argv[1])
45			new = getsizes(sys.argv[2])
46			grow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0
47			delta, common = [], {}
48
49			for a in old:
50			if a in new:
51			common[a] = 1
52
53			for name in old:
54			if name not in common:
55			remove += 1
56			down += old[name]
57			delta.append((-old[name], name))
58
59			for name in new:
60			if name not in common:
61			add += 1
62			up += new[name]
63			delta.append((new[name], name))
64
65			for name in common:
66			d = new.get(name, 0) - old.get(name, 0)
67			if d>0: grow, up = grow+1, up+d
68			if d<0: shrink, down = shrink+1, down-d
69			delta.append((d, name))
70
71			delta.sort()
72			delta.reverse()
73
74			print "%-48s %7s %7s %+7s" % ("function", "old", "new", "delta")
75			for d, n in delta:
76			if d: print "%-48s %7s %7s %+7d" % (n, old.get(n,"-"), new.get(n,"-"), d)
77			print "-"*78
78			total="(add/remove: %s/%s grow/shrink: %s/%s up/down: %s/%s)%%sTotal: %s bytes"\
79			% (add, remove, grow, shrink, up, -down, up-down)
80			print total % (" "*(80-len(total)))