1 #!/usr/bin/env python3 1 #!/usr/bin/env python3
2 # 2 #
3 # Copyright 2004 Matt Mackall <mpm@selenic.com> 3 # Copyright 2004 Matt Mackall <mpm@selenic.com>
4 # 4 #
5 # inspired by perl Bloat-O-Meter (c) 1997 by A 5 # inspired by perl Bloat-O-Meter (c) 1997 by Andi Kleen
6 # 6 #
7 # This software may be used and distributed ac 7 # This software may be used and distributed according to the terms
8 # of the GNU General Public License, incorpora 8 # of the GNU General Public License, incorporated herein by reference.
9 9
10 import sys, os, re, argparse !! 10 import sys, os, re
11 from signal import signal, SIGPIPE, SIG_DFL 11 from signal import signal, SIGPIPE, SIG_DFL
12 12
13 signal(SIGPIPE, SIG_DFL) 13 signal(SIGPIPE, SIG_DFL)
14 14
15 parser = argparse.ArgumentParser(description=" !! 15 if len(sys.argv) != 3:
16 group = parser.add_mutually_exclusive_group() !! 16 sys.stderr.write("usage: %s file1 file2\n" % sys.argv[0])
17 group.add_argument('-c', help='categorize outp !! 17 sys.exit(-1)
18 group.add_argument('-d', help='Show delta of D <<
19 group.add_argument('-t', help='Show delta of t <<
20 parser.add_argument('-p', dest='prefix', help= <<
21 parser.add_argument('file1', help='First file <<
22 parser.add_argument('file2', help='Second file <<
23 18
24 args = parser.parse_args() !! 19 def getsizes(file):
25 <<
26 re_NUMBER = re.compile(r'\.[0-9]+') <<
27 <<
28 def getsizes(file, format): <<
29 sym = {} 20 sym = {}
30 nm = "nm" !! 21 for l in os.popen("nm --size-sort " + file).readlines():
31 if args.prefix: !! 22 size, type, name = l[:-1].split()
32 nm = "{}nm".format(args.prefix) !! 23 if type in "tTdDbBrR":
33 !! 24 # strip generated symbols
34 with os.popen("{} --size-sort {}".format(n !! 25 if name.startswith("__mod_"): continue
35 for line in f: !! 26 if name.startswith("SyS_"): continue
36 if line.startswith("\n") or ":" in !! 27 if name.startswith("compat_SyS_"): continue
37 continue !! 28 if name == "linux_banner": continue
38 size, type, name = line.split() !! 29 # statics and some other optimizations adds random .NUMBER
39 if type in format: !! 30 name = re.sub(r'\.[0-9]+', '', name)
40 # strip generated symbols !! 31 sym[name] = sym.get(name, 0) + int(size, 16)
41 if name.startswith("__mod_"): <<
42 if name.startswith("__se_sys") <<
43 if name.startswith("__se_compa <<
44 if name.startswith("__addressa <<
45 if name == "linux_banner": con <<
46 if name == "vermagic": continu <<
47 # statics and some other optim <<
48 name = re_NUMBER.sub('', name) <<
49 sym[name] = sym.get(name, 0) + <<
50 return sym 32 return sym
51 33
52 def calc(oldfile, newfile, format): !! 34 old = getsizes(sys.argv[1])
53 old = getsizes(oldfile, format) !! 35 new = getsizes(sys.argv[2])
54 new = getsizes(newfile, format) !! 36 grow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0
55 grow, shrink, add, remove, up, down = 0, 0 !! 37 delta, common = [], {}
56 delta, common = [], {} !! 38 otot, ntot = 0, 0
57 otot, ntot = 0, 0 !! 39
58 !! 40 for a in old:
59 for a in old: !! 41 if a in new:
60 if a in new: !! 42 common[a] = 1
61 common[a] = 1 !! 43
62 !! 44 for name in old:
63 for name in old: !! 45 otot += old[name]
64 otot += old[name] !! 46 if name not in common:
65 if name not in common: !! 47 remove += 1
66 remove += 1 !! 48 down += old[name]
67 down += old[name] !! 49 delta.append((-old[name], name))
68 delta.append((-old[name], name)) !! 50
69 !! 51 for name in new:
70 for name in new: !! 52 ntot += new[name]
71 ntot += new[name] !! 53 if name not in common:
72 if name not in common: !! 54 add += 1
73 add += 1 !! 55 up += new[name]
74 up += new[name] !! 56 delta.append((new[name], name))
75 delta.append((new[name], name)) <<
76 57
77 for name in common: !! 58 for name in common:
78 d = new.get(name, 0) - old.get(name, 0 59 d = new.get(name, 0) - old.get(name, 0)
79 if d>0: grow, up = grow+1, up+d 60 if d>0: grow, up = grow+1, up+d
80 if d<0: shrink, down = shrink+1, down- 61 if d<0: shrink, down = shrink+1, down-d
81 delta.append((d, name)) 62 delta.append((d, name))
82 63
83 delta.sort(reverse=True) !! 64 delta.sort()
84 return grow, shrink, add, remove, up, down !! 65 delta.reverse()
>> 66
>> 67 print("add/remove: %s/%s grow/shrink: %s/%s up/down: %s/%s (%s)" % \
>> 68 (add, remove, grow, shrink, up, -down, up-down))
>> 69 print("%-40s %7s %7s %+7s" % ("function", "old", "new", "delta"))
>> 70 for d, n in delta:
>> 71 if d: print("%-40s %7s %7s %+7d" % (n, old.get(n,"-"), new.get(n,"-"), d))
85 72
86 def print_result(symboltype, symbolformat): !! 73 print("Total: Before=%d, After=%d, chg %+.2f%%" % \
87 grow, shrink, add, remove, up, down, delta !! 74 (otot, ntot, (ntot - otot)*100.0/otot))
88 calc(args.file1, args.file2, symbolformat) <<
89 <<
90 print("add/remove: %s/%s grow/shrink: %s/% <<
91 (add, remove, grow, shrink, up, -dow <<
92 print("%-40s %7s %7s %+7s" % (symboltype, <<
93 for d, n in delta: <<
94 if d: print("%-40s %7s %7s %+7d" % (n, <<
95 <<
96 if otot: <<
97 percent = (ntot - otot) * 100.0 / otot <<
98 else: <<
99 percent = 0 <<
100 print("Total: Before=%d, After=%d, chg %+. <<
101 <<
102 if args.c: <<
103 print_result("Function", "tTwW") <<
104 print_result("Data", "dDbBvV") <<
105 print_result("RO Data", "rR") <<
106 elif args.d: <<
107 print_result("Data", "dDbBrRvV") <<
108 elif args.t: <<
109 print_result("Function", "tTwW") <<
110 else: <<
111 print_result("Function", "tTdDbBrRvVwW") <<
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.