| | 1 | #!/usr/bin/env python |
| | 2 | from subprocess import Popen, PIPE, call |
| | 3 | import re |
| | 4 | import sys |
| | 5 | import os |
| | 6 | |
| | 7 | SLIDE = 0 |
| | 8 | |
| | 9 | NM_FORMAT = "([0-9a-f]+) ([UuAaTtDdBbCcSsIi]) (.*)" |
| | 10 | |
| | 11 | nm_re = re.compile(NM_FORMAT) |
| | 12 | |
| | 13 | def parse_nm_output(str): |
| | 14 | "returns (start, type, name)" |
| | 15 | m = nm_re.match(str) |
| | 16 | if m: |
| | 17 | start = int(m.group(1), 16) |
| | 18 | return (start, m.group(2), m.group(3)) |
| | 19 | else: |
| | 20 | return None |
| | 21 | |
| | 22 | def nm(file): |
| | 23 | cmd = "nm %s" % file |
| | 24 | p = Popen(cmd, shell=True, stdout=PIPE) |
| | 25 | return p.stdout |
| | 26 | |
| | 27 | class SymbolLookup: |
| | 28 | def __init__(self, file, min_width=16): |
| | 29 | self.min_width = min_width |
| | 30 | self.symbols = [parse_nm_output(l) for l in nm(file)] |
| | 31 | self.symbols.sort(key=lambda x: x[0]) |
| | 32 | |
| | 33 | def padded(self, str): |
| | 34 | return ("%%%ds" % self.min_width) % str |
| | 35 | |
| | 36 | def __call__(self, saddr): |
| | 37 | addr = int(saddr.group(0), 16) |
| | 38 | last = (0, ' ', '<start of file>') |
| | 39 | if( addr > SLIDE ): |
| | 40 | addr -= SLIDE |
| | 41 | # stupid linear search... feel free to improve |
| | 42 | for s in self.symbols: |
| | 43 | if s[0] == addr: |
| | 44 | return self.padded(s[2]) |
| | 45 | elif s[0] > addr: |
| | 46 | if last[2] == "_last_kernel_symbol": |
| | 47 | return saddr.group(0) |
| | 48 | return self.padded("<%s>+%x" % (last[2], addr - last[0])) |
| | 49 | else: |
| | 50 | last = s |
| | 51 | if last[2] == "_last_kernel_symbol": |
| | 52 | return saddr.group(0) |
| | 53 | return self.padded("<%s>+%x" % (last[2], addr - last[0])) |
| | 54 | |
| | 55 | def symbolify(objfile, input, *args, **kargs): |
| | 56 | replacer = SymbolLookup(objfile, *args, **kargs) |
| | 57 | for l in input: |
| | 58 | print re.sub("(0x)?[0-9a-f]{6,16}", replacer, l), |
| | 59 | |
| | 60 | |
| | 61 | def usage(): |
| | 62 | |
| | 63 | print "usage: %s [filename] [slide]" % sys.argv[0] |
| | 64 | print "\tor speficy a filename in your SYMBOLIFY_KERNEL environment variable" |
| | 65 | |
| | 66 | # die now |
| | 67 | sys.exit(1) |
| | 68 | |
| | 69 | KERNEL_FILE = None |
| | 70 | |
| | 71 | if( len(sys.argv) > 3 ): |
| | 72 | usage() |
| | 73 | |
| | 74 | if( len(sys.argv) == 3 ): |
| | 75 | SLIDE = int(sys.argv[2], 16) |
| | 76 | |
| | 77 | if( len(sys.argv) >= 2 ): |
| | 78 | KERNEL_FILE = sys.argv[1] |
| | 79 | |
| | 80 | if( KERNEL_FILE is None ): |
| | 81 | KERNEL_FILE = os.environ.get("SYMBOLIFY_KERNEL") |
| | 82 | |
| | 83 | if( KERNEL_FILE is None ): |
| | 84 | usage() |
| | 85 | |
| | 86 | print "using kernel file '%s', slide 0x%x" % (KERNEL_FILE, SLIDE) |
| | 87 | |
| | 88 | symbolify(KERNEL_FILE, sys.stdin, min_width=40) |
| | 89 | |
projects / apple / xnu.git / blame_incremental