cpython/Tools/scripts/objgraph.py

211 lines
5.8 KiB
Python
Raw Normal View History

#! /usr/bin/env python3
1991-06-04 17:36:54 -03:00
# objgraph
#
# Read "nm -o" input of a set of libraries or modules and print various
# interesting listings, such as:
1991-06-04 17:36:54 -03:00
#
# - which names are used but not defined in the set (and used where),
# - which names are defined in the set (and where),
# - which modules use which other modules,
# - which modules are used by which other modules.
#
# Usage: objgraph [-cdu] [file] ...
# -c: print callers per objectfile
# -d: print callees per objectfile
# -u: print usage of undefined symbols
# If none of -cdu is specified, all are assumed.
# Use "nm -o" to generate the input
1991-06-04 17:36:54 -03:00
# e.g.: nm -o /lib/libc.a | objgraph
import sys
1992-12-09 20:00:58 -04:00
import os
1991-06-04 17:36:54 -03:00
import getopt
import re
1991-06-04 17:36:54 -03:00
# Types of symbols.
#
definitions = 'TRGDSBAEC'
externals = 'UV'
ignore = 'Nntrgdsbavuc'
# Regular expression to parse "nm -o" output.
#
matcher = re.compile('(.*):\t?........ (.) (.*)$')
1991-06-04 17:36:54 -03:00
# Store "item" in "dict" under "key".
# The dictionary maps keys to lists of items.
# If there is no list for the key yet, it is created.
#
def store(dict, key, item):
2008-05-16 12:23:30 -03:00
if key in dict:
2001-01-17 04:48:39 -04:00
dict[key].append(item)
else:
dict[key] = [item]
1991-06-04 17:36:54 -03:00
# Return a flattened version of a list of strings: the concatenation
# of its elements with intervening spaces.
#
def flat(list):
2001-01-17 04:48:39 -04:00
s = ''
for item in list:
s = s + ' ' + item
return s[1:]
1991-06-04 17:36:54 -03:00
# Global variables mapping defined/undefined names to files and back.
#
file2undef = {}
def2file = {}
file2def = {}
undef2file = {}
# Read one input file and merge the data into the tables.
# Argument is an open file.
#
def readinput(fp):
2001-01-17 04:48:39 -04:00
while 1:
s = fp.readline()
2001-01-17 04:48:39 -04:00
if not s:
break
# If you get any output from this line,
# it is probably caused by an unexpected input line:
if matcher.search(s) < 0: s; continue # Shouldn't happen
(ra, rb), (r1a, r1b), (r2a, r2b), (r3a, r3b) = matcher.regs[:4]
fn, name, type = s[r1a:r1b], s[r3a:r3b], s[r2a:r2b]
if type in definitions:
store(def2file, name, fn)
store(file2def, fn, name)
elif type in externals:
store(file2undef, fn, name)
store(undef2file, name, fn)
elif not type in ignore:
print(fn + ':' + name + ': unknown type ' + type)
1991-06-04 17:36:54 -03:00
# Print all names that were undefined in some module and where they are
# defined.
#
def printcallee():
2008-05-16 12:23:30 -03:00
flist = sorted(file2undef.keys())
for filename in flist:
print(filename + ':')
elist = file2undef[filename]
2001-01-17 04:48:39 -04:00
elist.sort()
for ext in elist:
if len(ext) >= 8:
tabs = '\t'
else:
tabs = '\t\t'
2008-05-16 12:23:30 -03:00
if ext not in def2file:
print('\t' + ext + tabs + ' *undefined')
2001-01-17 04:48:39 -04:00
else:
print('\t' + ext + tabs + flat(def2file[ext]))
1991-06-04 17:36:54 -03:00
# Print for each module the names of the other modules that use it.
#
def printcaller():
2008-05-16 12:23:30 -03:00
files = sorted(file2def.keys())
for filename in files:
2001-01-17 04:48:39 -04:00
callers = []
for label in file2def[filename]:
2008-05-16 12:23:30 -03:00
if label in undef2file:
2001-01-17 04:48:39 -04:00
callers = callers + undef2file[label]
if callers:
callers.sort()
print(filename + ':')
2001-01-17 04:48:39 -04:00
lastfn = ''
for fn in callers:
2008-05-16 12:23:30 -03:00
if fn != lastfn:
print('\t' + fn)
2001-01-17 04:48:39 -04:00
lastfn = fn
else:
print(filename + ': unused')
1991-06-04 17:36:54 -03:00
# Print undefined names and where they are used.
1991-06-04 17:36:54 -03:00
#
def printundef():
2001-01-17 04:48:39 -04:00
undefs = {}
2008-05-16 12:23:30 -03:00
for filename in list(file2undef.keys()):
for ext in file2undef[filename]:
2008-05-16 12:23:30 -03:00
if ext not in def2file:
store(undefs, ext, filename)
2008-05-16 12:23:30 -03:00
elist = sorted(undefs.keys())
2001-01-17 04:48:39 -04:00
for ext in elist:
print(ext + ':')
2008-05-16 12:23:30 -03:00
flist = sorted(undefs[ext])
for filename in flist:
print('\t' + filename)
1991-06-04 17:36:54 -03:00
# Print warning messages about names defined in more than one file.
#
def warndups():
2001-01-17 04:48:39 -04:00
savestdout = sys.stdout
sys.stdout = sys.stderr
2008-05-16 12:23:30 -03:00
names = sorted(def2file.keys())
2001-01-17 04:48:39 -04:00
for name in names:
if len(def2file[name]) > 1:
print('warning:', name, 'multiply defined:', end=' ')
print(flat(def2file[name]))
2001-01-17 04:48:39 -04:00
sys.stdout = savestdout
1991-06-04 17:36:54 -03:00
# Main program
#
def main():
2001-01-17 04:48:39 -04:00
try:
optlist, args = getopt.getopt(sys.argv[1:], 'cdu')
except getopt.error:
sys.stdout = sys.stderr
print('Usage:', os.path.basename(sys.argv[0]), end=' ')
print('[-cdu] [file] ...')
print('-c: print callers per objectfile')
print('-d: print callees per objectfile')
print('-u: print usage of undefined symbols')
print('If none of -cdu is specified, all are assumed.')
print('Use "nm -o" to generate the input')
print('e.g.: nm -o /lib/libc.a | objgraph')
2001-01-17 04:48:39 -04:00
return 1
optu = optc = optd = 0
for opt, void in optlist:
if opt == '-u':
optu = 1
elif opt == '-c':
optc = 1
elif opt == '-d':
optd = 1
if optu == optc == optd == 0:
optu = optc = optd = 1
if not args:
args = ['-']
for filename in args:
if filename == '-':
2001-01-17 04:48:39 -04:00
readinput(sys.stdin)
else:
readinput(open(filename, 'r'))
2001-01-17 04:48:39 -04:00
#
warndups()
#
more = (optu + optc + optd > 1)
if optd:
if more:
print('---------------All callees------------------')
2001-01-17 04:48:39 -04:00
printcallee()
if optu:
if more:
print('---------------Undefined callees------------')
2001-01-17 04:48:39 -04:00
printundef()
if optc:
if more:
print('---------------All Callers------------------')
2001-01-17 04:48:39 -04:00
printcaller()
return 0
1991-06-04 17:36:54 -03:00
# Call the main program.
# Use its return value as exit status.
# Catch interrupts to avoid stack trace.
#
if __name__ == '__main__':
try:
sys.exit(main())
except KeyboardInterrupt:
sys.exit(1)