Andy Shevchenko | 51839e29 | 2020-12-09 13:50:17 +0200 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
Matt Mackall | d960600 | 2006-01-08 01:05:19 -0800 | [diff] [blame] | 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 | |
Nikolay Borisov | b62eb27 | 2022-07-01 14:35:12 +0300 | [diff] [blame] | 10 | import sys, os, re, argparse |
Alexey Dobriyan | eef06b82 | 2016-11-10 10:46:13 -0800 | [diff] [blame] | 11 | from signal import signal, SIGPIPE, SIG_DFL |
| 12 | |
| 13 | signal(SIGPIPE, SIG_DFL) |
Matt Mackall | d960600 | 2006-01-08 01:05:19 -0800 | [diff] [blame] | 14 | |
Nikolay Borisov | b62eb27 | 2022-07-01 14:35:12 +0300 | [diff] [blame] | 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') |
Nikolay Borisov | 8b5db66 | 2022-07-01 14:35:13 +0300 | [diff] [blame] | 20 | parser.add_argument('-p', dest='prefix', help='Arch prefix for the tool being used. Useful in cross build scenarios') |
Nikolay Borisov | b62eb27 | 2022-07-01 14:35:12 +0300 | [diff] [blame] | 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() |
Matt Mackall | d960600 | 2006-01-08 01:05:19 -0800 | [diff] [blame] | 25 | |
Alexey Dobriyan | 0d7bbb4 | 2016-12-12 16:40:48 -0800 | [diff] [blame] | 26 | re_NUMBER = re.compile(r'\.[0-9]+') |
| 27 | |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 28 | def getsizes(file, format): |
Matt Mackall | d960600 | 2006-01-08 01:05:19 -0800 | [diff] [blame] | 29 | sym = {} |
Nikolay Borisov | 8b5db66 | 2022-07-01 14:35:13 +0300 | [diff] [blame] | 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: |
Alexey Dobriyan | 3af06fd | 2016-12-12 16:40:45 -0800 | [diff] [blame] | 35 | for line in f: |
Nikolay Borisov | 1d35b60 | 2020-08-06 23:17:32 -0700 | [diff] [blame] | 36 | if line.startswith("\n") or ":" in line: |
| 37 | continue |
Alexey Dobriyan | 3af06fd | 2016-12-12 16:40:45 -0800 | [diff] [blame] | 38 | size, type, name = line.split() |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 39 | if type in format: |
Alexey Dobriyan | 3af06fd | 2016-12-12 16:40:45 -0800 | [diff] [blame] | 40 | # strip generated symbols |
| 41 | if name.startswith("__mod_"): continue |
Dominik Brodowski | e145242e | 2018-04-09 12:51:42 +0200 | [diff] [blame] | 42 | if name.startswith("__se_sys"): continue |
Dominik Brodowski | 5ac9efa | 2018-04-09 12:51:43 +0200 | [diff] [blame] | 43 | if name.startswith("__se_compat_sys"): continue |
Rasmus Villemoes | e0b2475 | 2018-12-28 00:31:18 -0800 | [diff] [blame] | 44 | if name.startswith("__addressable_"): continue |
Alexey Dobriyan | 3af06fd | 2016-12-12 16:40:45 -0800 | [diff] [blame] | 45 | if name == "linux_banner": continue |
Paul Gortmaker | dec81a5 | 2022-04-29 14:37:57 -0700 | [diff] [blame] | 46 | if name == "vermagic": continue |
Alexey Dobriyan | 3af06fd | 2016-12-12 16:40:45 -0800 | [diff] [blame] | 47 | # statics and some other optimizations adds random .NUMBER |
Alexey Dobriyan | 0d7bbb4 | 2016-12-12 16:40:48 -0800 | [diff] [blame] | 48 | name = re_NUMBER.sub('', name) |
Alexey Dobriyan | 3af06fd | 2016-12-12 16:40:45 -0800 | [diff] [blame] | 49 | sym[name] = sym.get(name, 0) + int(size, 16) |
Matt Mackall | d960600 | 2006-01-08 01:05:19 -0800 | [diff] [blame] | 50 | return sym |
| 51 | |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 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 |
Matt Mackall | d960600 | 2006-01-08 01:05:19 -0800 | [diff] [blame] | 58 | |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 59 | for a in old: |
| 60 | if a in new: |
| 61 | common[a] = 1 |
Matt Mackall | d960600 | 2006-01-08 01:05:19 -0800 | [diff] [blame] | 62 | |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 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)) |
Matt Mackall | d960600 | 2006-01-08 01:05:19 -0800 | [diff] [blame] | 69 | |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 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)) |
Matt Mackall | d960600 | 2006-01-08 01:05:19 -0800 | [diff] [blame] | 76 | |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 77 | for name in common: |
Matt Mackall | d960600 | 2006-01-08 01:05:19 -0800 | [diff] [blame] | 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 | |
Alexander Pantyukhin | 0371ba6 | 2023-01-06 14:13:19 +0500 | [diff] [blame] | 83 | delta.sort(reverse=True) |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 84 | return grow, shrink, add, remove, up, down, delta, old, new, otot, ntot |
Matt Mackall | d960600 | 2006-01-08 01:05:19 -0800 | [diff] [blame] | 85 | |
Nikolay Borisov | b62eb27 | 2022-07-01 14:35:12 +0300 | [diff] [blame] | 86 | def print_result(symboltype, symbolformat): |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 87 | grow, shrink, add, remove, up, down, delta, old, new, otot, ntot = \ |
Nikolay Borisov | b62eb27 | 2022-07-01 14:35:12 +0300 | [diff] [blame] | 88 | calc(args.file1, args.file2, symbolformat) |
Vineet Gupta | b21e91c | 2016-05-19 17:09:17 -0700 | [diff] [blame] | 89 | |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 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 | |
Andy Shevchenko | edbddb8 | 2017-11-29 16:11:05 -0800 | [diff] [blame] | 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)) |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 101 | |
Nikolay Borisov | b62eb27 | 2022-07-01 14:35:12 +0300 | [diff] [blame] | 102 | if args.c: |
| 103 | print_result("Function", "tT") |
| 104 | print_result("Data", "dDbB") |
| 105 | print_result("RO Data", "rR") |
| 106 | elif args.d: |
| 107 | print_result("Data", "dDbBrR") |
| 108 | elif args.t: |
| 109 | print_result("Function", "tT") |
Maninder Singh | 192efb7 | 2017-11-15 17:31:14 -0800 | [diff] [blame] | 110 | else: |
Nikolay Borisov | b62eb27 | 2022-07-01 14:35:12 +0300 | [diff] [blame] | 111 | print_result("Function", "tTdDbBrR") |