U d@srdZddlZddlZddlZGdddeZddZddZd d Zd d Z d dZ e dkrne ej dddS)aA tool to inspect the binary size of a built binary file. This script prints out a tree of symbols and their corresponding sizes, using Linux's nm functionality. Usage: python binary_size.py -- --target=/path/to/your/target/binary [--nm_command=/path/to/your/custom/nm] [--max_depth=10] [--min_size=1024] [--color] To assist visualization, pass in '--color' to make the symbols color coded to green, assuming that you have a xterm connection that supports color. Nc@seZdZdZddZdS)Triez&A simple class that represents a Trie.cCs||_d|_i|_dS)zInitializes a Trie object.rN)namesize dictionary)selfrrRsz GetSymbolTrie..) subprocessPopenPIPESTDOUT communicate returncodeprintsysexitsplitlenappendintjoinrrrrsumvalues)target nm_command max_depthprocZnm_out_outdatalinecontent symbol_trierrcurrcrrr GetSymbolTrie%sN         &    r7cCs|rd|S|SdS)zEWrap the input string to the xterm green color, if color is set. z {0}N)format)scolorrrr MaybeAddColorVs r;cCs8dD](}t|dkr$d||fS|d}qd|fS)zGet a human-readable size.)BZKBZMBZGBg@z%3.2f%sz%.1f TB)abs)numunitrrr ReadableSize_s   r@cstj|ksjsDj|krtd|tj|tjntjdkrrt j d||||nj|krtd|tj|tjfddjD}|j ddd|d d d D]"\}}t j||d |||qd S) z1Prints the symbol trie in a readable manner. z {0}{1} {2}rrcsg|]}|j|jfqSr)rr)rktrierr szPrintTrie..cSs|dS)Nrr)xrrrzPrintTrie..)keyNz |) r%rrkeysrr!r8r;r@ PrintTrier*sort)rCprefixr-min_sizer:Zkeys_with_sizesrAr/rrBrrKjsB      rKcCstjdstdtjdd}|jdtddd|jd td d d|jd td dd|jdddd|jdtdd| |}|j stdt |j |j |j }t|d|j |j|jdS)Nlinuxz(Currently this tool only supports Linux.zTool to inspect binary size.) descriptionz --max_depth z+The maximum depth to print the symbol tree.)typedefaulthelpz --min_sizeiz!The mininum symbol size to print.z --nm_commandnmz/The path to the nm command that the tool needs.z--color store_truez#If set, use ascii color for output.)actionrTz--targetzThe binary target to inspect.)rRrTz%You must specify a target to inspect.r)r"platform startswith RuntimeErrorargparseArgumentParser add_argumentr'str parse_argsr+r7r,r-rKrNr:)argvparserargsr4rrrmainsT  rc__main__r) r r[rr"objectrr7r;r@rKrcr r`rrrrs 1