Commit | Line | Data |
---|---|---|
51839e29 | 1 | #!/usr/bin/env python3 |
d960600d MM |
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 | ||
b62eb273 | 10 | import sys, os, re, argparse |
eef06b82 AD |
11 | from signal import signal, SIGPIPE, SIG_DFL |
12 | ||
13 | signal(SIGPIPE, SIG_DFL) | |
d960600d | 14 | |
b62eb273 NB |
15 | parser = argparse.ArgumentParser(description="Simple script used to compare the symbol sizes of 2 object files") |
16 | group = parser.add_mutually_exclusive_group() | |
17 | group.add_argument('-c', help='categorize output based on symbol type', action='store_true') | |
18 | group.add_argument('-d', help='Show delta of Data Section', action='store_true') | |
19 | group.add_argument('-t', help='Show delta of text Section', action='store_true') | |
8b5db667 | 20 | parser.add_argument('-p', dest='prefix', help='Arch prefix for the tool being used. Useful in cross build scenarios') |
b62eb273 NB |
21 | parser.add_argument('file1', help='First file to compare') |
22 | parser.add_argument('file2', help='Second file to compare') | |
23 | ||
24 | args = parser.parse_args() | |
d960600d | 25 | |
0d7bbb43 AD |
26 | re_NUMBER = re.compile(r'\.[0-9]+') |
27 | ||
192efb7a | 28 | def getsizes(file, format): |
d960600d | 29 | sym = {} |
8b5db667 NB |
30 | nm = "nm" |
31 | if args.prefix: | |
32 | nm = "{}nm".format(args.prefix) | |
33 | ||
34 | with os.popen("{} --size-sort {}".format(nm, file)) as f: | |
3af06fd9 | 35 | for line in f: |
1d35b605 NB |
36 | if line.startswith("\n") or ":" in line: |
37 | continue | |
3af06fd9 | 38 | size, type, name = line.split() |
192efb7a | 39 | if type in format: |
3af06fd9 AD |
40 | # strip generated symbols |
41 | if name.startswith("__mod_"): continue | |
e145242e | 42 | if name.startswith("__se_sys"): continue |
5ac9efa3 | 43 | if name.startswith("__se_compat_sys"): continue |
e0b2475a | 44 | if name.startswith("__addressable_"): continue |
3af06fd9 | 45 | if name == "linux_banner": continue |
dec81a53 | 46 | if name == "vermagic": continue |
3af06fd9 | 47 | # statics and some other optimizations adds random .NUMBER |
0d7bbb43 | 48 | name = re_NUMBER.sub('', name) |
3af06fd9 | 49 | sym[name] = sym.get(name, 0) + int(size, 16) |
d960600d MM |
50 | return sym |
51 | ||
192efb7a MS |
52 | def calc(oldfile, newfile, format): |
53 | old = getsizes(oldfile, format) | |
54 | new = getsizes(newfile, format) | |
55 | grow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0 | |
56 | delta, common = [], {} | |
57 | otot, ntot = 0, 0 | |
d960600d | 58 | |
192efb7a MS |
59 | for a in old: |
60 | if a in new: | |
61 | common[a] = 1 | |
d960600d | 62 | |
192efb7a MS |
63 | for name in old: |
64 | otot += old[name] | |
65 | if name not in common: | |
66 | remove += 1 | |
67 | down += old[name] | |
68 | delta.append((-old[name], name)) | |
d960600d | 69 | |
192efb7a MS |
70 | for name in new: |
71 | ntot += new[name] | |
72 | if name not in common: | |
73 | add += 1 | |
74 | up += new[name] | |
75 | delta.append((new[name], name)) | |
d960600d | 76 | |
192efb7a | 77 | for name in common: |
d960600d MM |
78 | d = new.get(name, 0) - old.get(name, 0) |
79 | if d>0: grow, up = grow+1, up+d | |
80 | if d<0: shrink, down = shrink+1, down-d | |
81 | delta.append((d, name)) | |
82 | ||
0371ba6f | 83 | delta.sort(reverse=True) |
192efb7a | 84 | return grow, shrink, add, remove, up, down, delta, old, new, otot, ntot |
d960600d | 85 | |
b62eb273 | 86 | def print_result(symboltype, symbolformat): |
192efb7a | 87 | grow, shrink, add, remove, up, down, delta, old, new, otot, ntot = \ |
b62eb273 | 88 | calc(args.file1, args.file2, symbolformat) |
b21e91c3 | 89 | |
192efb7a MS |
90 | print("add/remove: %s/%s grow/shrink: %s/%s up/down: %s/%s (%s)" % \ |
91 | (add, remove, grow, shrink, up, -down, up-down)) | |
92 | print("%-40s %7s %7s %+7s" % (symboltype, "old", "new", "delta")) | |
93 | for d, n in delta: | |
94 | if d: print("%-40s %7s %7s %+7d" % (n, old.get(n,"-"), new.get(n,"-"), d)) | |
95 | ||
edbddb83 AS |
96 | if otot: |
97 | percent = (ntot - otot) * 100.0 / otot | |
98 | else: | |
99 | percent = 0 | |
100 | print("Total: Before=%d, After=%d, chg %+.2f%%" % (otot, ntot, percent)) | |
192efb7a | 101 | |
b62eb273 | 102 | if args.c: |
198430f7 GU |
103 | print_result("Function", "tTwW") |
104 | print_result("Data", "dDbBvV") | |
b62eb273 NB |
105 | print_result("RO Data", "rR") |
106 | elif args.d: | |
198430f7 | 107 | print_result("Data", "dDbBrRvV") |
b62eb273 | 108 | elif args.t: |
198430f7 | 109 | print_result("Function", "tTwW") |
192efb7a | 110 | else: |
198430f7 | 111 | print_result("Function", "tTdDbBrRvVwW") |