blob: cb861b8e9250cc4aeca26823cc51826b1b4aca75
1 | #!/usr/bin/env 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 |
11 | |
12 | def usage(): |
13 | sys.stderr.write("usage: %s [-t] file1 file2 [-- <readelf options>]\n" |
14 | % sys.argv[0]) |
15 | sys.stderr.write("\t-t\tShow time spent on parsing/processing\n") |
16 | sys.stderr.write("\t--\tPass additional parameters to readelf\n") |
17 | sys.exit(1) |
18 | |
19 | f1, f2 = (None, None) |
20 | flag_timing, dashes = (False, False) |
21 | |
22 | for f in sys.argv[1:]: |
23 | if f.startswith("-"): |
24 | if f == "--": # sym_args |
25 | dashes = True |
26 | break |
27 | if f == "-t": # timings |
28 | flag_timing = True |
29 | else: |
30 | if not os.path.exists(f): |
31 | sys.stderr.write("Error: file '%s' does not exist\n" % f) |
32 | usage() |
33 | if f1 is None: |
34 | f1 = f |
35 | elif f2 is None: |
36 | f2 = f |
37 | else: |
38 | usage() |
39 | if flag_timing: |
40 | import time |
41 | if f1 is None or f2 is None: |
42 | usage() |
43 | |
44 | sym_args = " ".join(sys.argv[3 + flag_timing + dashes:]) |
45 | def getsizes(file): |
46 | sym, alias, lut = {}, {}, {} |
47 | for l in os.popen("readelf -W -s %s %s" % (sym_args, file)).readlines(): |
48 | l = l.strip() |
49 | if not (len(l) and l[0].isdigit() and len(l.split()) == 8): |
50 | continue |
51 | num, value, size, typ, bind, vis, ndx, name = l.split() |
52 | if ndx == "UND": continue # skip undefined |
53 | if typ in ["SECTION", "FILES"]: continue # skip sections and files |
54 | if "." in name: name = "static." + name.split(".")[0] |
55 | value = int(value, 16) |
56 | size = int(size, 16) if size.startswith('0x') else int(size) |
57 | if vis != "DEFAULT" and bind != "GLOBAL": # see if it is an alias |
58 | alias[(value, size)] = {"name" : name} |
59 | else: |
60 | sym[name] = {"addr" : value, "size": size} |
61 | lut[(value, size)] = 0 |
62 | for addr, sz in iter(alias.keys()): |
63 | # If the non-GLOBAL sym has an implementation elsewhere then |
64 | # it's an alias, disregard it. |
65 | if not (addr, sz) in lut: |
66 | # If this non-GLOBAL sym does not have an implementation at |
67 | # another address, then treat it as a normal symbol. |
68 | sym[alias[(addr, sz)]["name"]] = {"addr" : addr, "size": sz} |
69 | for l in os.popen("readelf -W -S " + file).readlines(): |
70 | x = l.split() |
71 | if len(x)<6: continue |
72 | # Should take these into account too! |
73 | #if x[1] not in [".text", ".rodata", ".symtab", ".strtab"]: continue |
74 | if x[1] not in [".rodata"]: continue |
75 | sym[x[1]] = {"addr" : int(x[3], 16), "size" : int(x[5], 16)} |
76 | return sym |
77 | |
78 | if flag_timing: |
79 | start_t1 = int(time.time() * 1e9) |
80 | old = getsizes(f1) |
81 | if flag_timing: |
82 | end_t1 = int(time.time() * 1e9) |
83 | start_t2 = int(time.time() * 1e9) |
84 | new = getsizes(f2) |
85 | if flag_timing: |
86 | end_t2 = int(time.time() * 1e9) |
87 | start_t3 = int(time.time() * 1e9) |
88 | grow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0 |
89 | delta, common = [], {} |
90 | |
91 | for name in iter(old.keys()): |
92 | if name in new: |
93 | common[name] = 1 |
94 | |
95 | for name in old: |
96 | if name not in common: |
97 | remove += 1 |
98 | sz = old[name]["size"] |
99 | down += sz |
100 | delta.append((-sz, name)) |
101 | |
102 | for name in new: |
103 | if name not in common: |
104 | add += 1 |
105 | sz = new[name]["size"] |
106 | up += sz |
107 | delta.append((sz, name)) |
108 | |
109 | for name in common: |
110 | d = new[name].get("size", 0) - old[name].get("size", 0) |
111 | if d>0: grow, up = grow+1, up+d |
112 | elif d<0: shrink, down = shrink+1, down-d |
113 | else: |
114 | continue |
115 | delta.append((d, name)) |
116 | |
117 | delta.sort() |
118 | delta.reverse() |
119 | if flag_timing: |
120 | end_t3 = int(time.time() * 1e9) |
121 | |
122 | print("%-48s %7s %7s %+7s" % ("function", "old", "new", "delta")) |
123 | for d, n in delta: |
124 | if d: |
125 | old_sz = old.get(n, {}).get("size", "-") |
126 | new_sz = new.get(n, {}).get("size", "-") |
127 | print("%-48s %7s %7s %+7d" % (n, old_sz, new_sz, d)) |
128 | print("-"*78) |
129 | total="(add/remove: %s/%s grow/shrink: %s/%s up/down: %s/%s)%%sTotal: %s bytes"\ |
130 | % (add, remove, grow, shrink, up, -down, up-down) |
131 | print(total % (" "*(80-len(total)))) |
132 | if flag_timing: |
133 | print("\n%d/%d; %d Parse origin/new; processing nsecs" % |
134 | (end_t1-start_t1, end_t2-start_t2, end_t3-start_t3)) |
135 | print("total nsecs: %d" % (end_t3-start_t1)) |
136 |