2010-01-07 21:31:53 +03:00
|
|
|
#!/usr/bin/python
|
|
|
|
# vim:sw=4:ts=4:et:
|
|
|
|
# ***** BEGIN LICENSE BLOCK *****
|
|
|
|
# Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
|
|
#
|
|
|
|
# The contents of this file are subject to the Mozilla Public License Version
|
|
|
|
# 1.1 (the "License"); you may not use this file except in compliance with
|
|
|
|
# the License. You may obtain a copy of the License at
|
|
|
|
# http://www.mozilla.org/MPL/
|
|
|
|
#
|
|
|
|
# Software distributed under the License is distributed on an "AS IS" basis,
|
|
|
|
# WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
|
|
# for the specific language governing rights and limitations under the
|
|
|
|
# License.
|
|
|
|
#
|
|
|
|
# The Original Code is fix-linux-stack.pl.
|
|
|
|
#
|
|
|
|
# The Initial Developer of the Original Code is L. David Baron.
|
|
|
|
# Portions created by the Initial Developer are Copyright (C) 2003
|
|
|
|
# the Initial Developer. All Rights Reserved.
|
|
|
|
#
|
|
|
|
# Contributor(s):
|
|
|
|
# L. David Baron <dbaron@dbaron.org> (original author)
|
|
|
|
#
|
|
|
|
# Alternatively, the contents of this file may be used under the terms of
|
|
|
|
# either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
|
|
# the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
|
|
# in which case the provisions of the GPL or the LGPL are applicable instead
|
|
|
|
# of those above. If you wish to allow use of your version of this file only
|
|
|
|
# under the terms of either the GPL or the LGPL, and not to allow others to
|
|
|
|
# use your version of this file under the terms of the MPL, indicate your
|
|
|
|
# decision by deleting the provisions above and replace them with the notice
|
|
|
|
# and other provisions required by the GPL or the LGPL. If you do not delete
|
|
|
|
# the provisions above, a recipient may use your version of this file under
|
|
|
|
# the terms of any one of the MPL, the GPL or the LGPL.
|
|
|
|
#
|
|
|
|
# ***** END LICENSE BLOCK *****
|
|
|
|
|
|
|
|
# This script uses atos to process the output of nsTraceRefcnt's Mac OS
|
|
|
|
# X stack walking code. This is useful for two things:
|
|
|
|
# (1) Getting line number information out of
|
|
|
|
# |nsTraceRefcntImpl::WalkTheStack|'s output in debug builds.
|
|
|
|
# (2) Getting function names out of |nsTraceRefcntImpl::WalkTheStack|'s
|
|
|
|
# output on all builds (where it mostly prints UNKNOWN because only
|
|
|
|
# a handful of symbols are exported from component libraries).
|
|
|
|
#
|
|
|
|
# Use the script by piping output containing stacks (such as raw stacks
|
|
|
|
# or make-tree.pl balance trees) through this script.
|
|
|
|
|
|
|
|
import subprocess
|
|
|
|
import sys
|
|
|
|
import re
|
2010-01-13 22:43:13 +03:00
|
|
|
import os
|
|
|
|
import pty
|
|
|
|
import termios
|
|
|
|
|
|
|
|
class unbufferedLineConverter:
|
|
|
|
"""
|
|
|
|
Wrap a child process that responds to each line of input with one line of
|
|
|
|
output. Uses pty to trick the child into providing unbuffered output.
|
|
|
|
"""
|
|
|
|
def __init__(self, command, args = []):
|
|
|
|
pid, fd = pty.fork()
|
|
|
|
if pid == 0:
|
|
|
|
# We're the child. Transfer control to command.
|
|
|
|
os.execvp(command, [command] + args)
|
|
|
|
else:
|
|
|
|
# Disable echoing.
|
|
|
|
attr = termios.tcgetattr(fd)
|
|
|
|
attr[3] = attr[3] & ~termios.ECHO
|
|
|
|
termios.tcsetattr(fd, termios.TCSANOW, attr)
|
|
|
|
# Set up a file()-like interface to the child process
|
|
|
|
self.r = os.fdopen(fd, "r", 1)
|
|
|
|
self.w = os.fdopen(os.dup(fd), "w", 1)
|
|
|
|
def convert(self, line):
|
|
|
|
self.w.write(line + "\n")
|
|
|
|
return self.r.readline().rstrip("\r\n")
|
|
|
|
@staticmethod
|
|
|
|
def test():
|
|
|
|
assert unbufferedLineConverter("rev").convert("123") == "321"
|
|
|
|
assert unbufferedLineConverter("cut", ["-c3"]).convert("abcde") == "c"
|
|
|
|
print "Pass"
|
2010-01-07 21:31:53 +03:00
|
|
|
|
|
|
|
def separate_debug_file_for(file):
|
|
|
|
return None
|
|
|
|
|
|
|
|
address_adjustments = {}
|
|
|
|
def address_adjustment(file):
|
|
|
|
if not file in address_adjustments:
|
|
|
|
result = None
|
|
|
|
otool = subprocess.Popen(["otool", "-l", file], stdout=subprocess.PIPE)
|
|
|
|
while True:
|
|
|
|
line = otool.stdout.readline()
|
|
|
|
if line == "":
|
|
|
|
break
|
|
|
|
if line == " segname __TEXT\n":
|
|
|
|
line = otool.stdout.readline()
|
|
|
|
if not line.startswith(" vmaddr "):
|
|
|
|
raise StandardError("unexpected otool output")
|
|
|
|
result = int(line[10:], 16)
|
|
|
|
break
|
|
|
|
otool.stdout.close()
|
|
|
|
|
|
|
|
if result is None:
|
|
|
|
raise StandardError("unexpected otool output")
|
|
|
|
|
|
|
|
address_adjustments[file] = result
|
|
|
|
|
|
|
|
return address_adjustments[file]
|
|
|
|
|
|
|
|
atoses = {}
|
2010-01-13 22:43:13 +03:00
|
|
|
def addressToSymbol(file, address):
|
|
|
|
converter = None
|
2010-01-07 21:31:53 +03:00
|
|
|
if not file in atoses:
|
|
|
|
debug_file = separate_debug_file_for(file) or file
|
2010-01-13 22:43:13 +03:00
|
|
|
converter = unbufferedLineConverter('/usr/bin/atos', ['-o', debug_file])
|
|
|
|
atoses[file] = converter
|
2010-01-07 21:31:53 +03:00
|
|
|
else:
|
2010-01-13 22:43:13 +03:00
|
|
|
converter = atoses[file]
|
|
|
|
return converter.convert("0x%X" % address)
|
2010-01-07 21:31:53 +03:00
|
|
|
|
|
|
|
cxxfilt_proc = None
|
|
|
|
def cxxfilt(sym):
|
|
|
|
if cxxfilt_proc is None:
|
|
|
|
globals()["cxxfilt_proc"] = subprocess.Popen(['c++filt',
|
|
|
|
'--no-strip-underscores',
|
|
|
|
'--format', 'gnu-v3'],
|
|
|
|
stdin=subprocess.PIPE,
|
|
|
|
stdout=subprocess.PIPE)
|
2010-01-07 21:56:09 +03:00
|
|
|
# strip underscores ourselves (works better than c++filt's
|
|
|
|
# --strip-underscores)
|
2010-01-07 21:52:51 +03:00
|
|
|
cxxfilt_proc.stdin.write(sym[1:] + "\n")
|
2010-01-07 21:31:53 +03:00
|
|
|
return cxxfilt_proc.stdout.readline().rstrip("\n")
|
|
|
|
|
2010-03-21 09:08:47 +03:00
|
|
|
line_re = re.compile("^(.*) ?\[([^ ]*) \+(0x[0-9A-F]{1,8})\](.*)$")
|
|
|
|
balance_tree_re = re.compile("^([ \|0-9-]*)")
|
2010-01-07 21:31:53 +03:00
|
|
|
atos_sym_re = re.compile("^(\S+) \(in ([^)]+)\) \((.+)\)$")
|
2010-02-22 00:03:20 +03:00
|
|
|
|
|
|
|
def fixSymbols(line):
|
2010-01-07 21:31:53 +03:00
|
|
|
result = line_re.match(line)
|
|
|
|
if result is not None:
|
|
|
|
# before allows preservation of balance trees
|
|
|
|
# after allows preservation of counts
|
2010-03-21 09:08:47 +03:00
|
|
|
(before, file, address, after) = result.groups()
|
2010-01-07 21:31:53 +03:00
|
|
|
address = int(address, 16)
|
|
|
|
|
|
|
|
if os.path.exists(file) and os.path.isfile(file):
|
|
|
|
address += address_adjustment(file)
|
2010-06-09 07:55:43 +04:00
|
|
|
info = addressToSymbol(file, address)
|
2010-01-07 21:31:53 +03:00
|
|
|
|
|
|
|
# atos output seems to have three forms:
|
|
|
|
# address
|
|
|
|
# address (in foo.dylib)
|
|
|
|
# symbol (in foo.dylib) (file:line)
|
|
|
|
symresult = atos_sym_re.match(info)
|
|
|
|
if symresult is not None:
|
|
|
|
# Print the first two forms as-is, and transform the third
|
|
|
|
(symbol, library, fileline) = symresult.groups()
|
|
|
|
symbol = cxxfilt(symbol)
|
|
|
|
info = "%s (%s, in %s)" % (symbol, fileline, library)
|
|
|
|
|
2010-03-21 09:08:47 +03:00
|
|
|
# throw away the bad symbol, but keep balance tree structure
|
|
|
|
before = balance_tree_re.match(before).groups()[0]
|
|
|
|
|
2010-02-22 00:03:20 +03:00
|
|
|
return before + info + after + "\n"
|
2010-01-07 21:31:53 +03:00
|
|
|
else:
|
|
|
|
sys.stderr.write("Warning: File \"" + file + "\" does not exist.\n")
|
2010-02-22 00:03:20 +03:00
|
|
|
return line
|
2010-01-07 21:31:53 +03:00
|
|
|
else:
|
2010-02-22 00:03:20 +03:00
|
|
|
return line
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
for line in sys.stdin:
|
|
|
|
sys.stdout.write(fixSymbols(line))
|