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