+++ /dev/null
-"""Disassembler of Python byte code into mnemonics."""\r
-\r
-import sys\r
-import types\r
-\r
-from opcode import *\r
-from opcode import __all__ as _opcodes_all\r
-\r
-__all__ = ["dis", "disassemble", "distb", "disco",\r
- "findlinestarts", "findlabels"] + _opcodes_all\r
-del _opcodes_all\r
-\r
-_have_code = (types.MethodType, types.FunctionType, types.CodeType,\r
- types.ClassType, type)\r
-\r
-def dis(x=None):\r
- """Disassemble classes, methods, functions, or code.\r
-\r
- With no argument, disassemble the last traceback.\r
-\r
- """\r
- if x is None:\r
- distb()\r
- return\r
- if isinstance(x, types.InstanceType):\r
- x = x.__class__\r
- if hasattr(x, 'im_func'):\r
- x = x.im_func\r
- if hasattr(x, 'func_code'):\r
- x = x.func_code\r
- if hasattr(x, '__dict__'):\r
- items = x.__dict__.items()\r
- items.sort()\r
- for name, x1 in items:\r
- if isinstance(x1, _have_code):\r
- print "Disassembly of %s:" % name\r
- try:\r
- dis(x1)\r
- except TypeError, msg:\r
- print "Sorry:", msg\r
- print\r
- elif hasattr(x, 'co_code'):\r
- disassemble(x)\r
- elif isinstance(x, str):\r
- disassemble_string(x)\r
- else:\r
- raise TypeError, \\r
- "don't know how to disassemble %s objects" % \\r
- type(x).__name__\r
-\r
-def distb(tb=None):\r
- """Disassemble a traceback (default: last traceback)."""\r
- if tb is None:\r
- try:\r
- tb = sys.last_traceback\r
- except AttributeError:\r
- raise RuntimeError, "no last traceback to disassemble"\r
- while tb.tb_next: tb = tb.tb_next\r
- disassemble(tb.tb_frame.f_code, tb.tb_lasti)\r
-\r
-def disassemble(co, lasti=-1):\r
- """Disassemble a code object."""\r
- code = co.co_code\r
- labels = findlabels(code)\r
- linestarts = dict(findlinestarts(co))\r
- n = len(code)\r
- i = 0\r
- extended_arg = 0\r
- free = None\r
- while i < n:\r
- c = code[i]\r
- op = ord(c)\r
- if i in linestarts:\r
- if i > 0:\r
- print\r
- print "%3d" % linestarts[i],\r
- else:\r
- print ' ',\r
-\r
- if i == lasti: print '-->',\r
- else: print ' ',\r
- if i in labels: print '>>',\r
- else: print ' ',\r
- print repr(i).rjust(4),\r
- print opname[op].ljust(20),\r
- i = i+1\r
- if op >= HAVE_ARGUMENT:\r
- oparg = ord(code[i]) + ord(code[i+1])*256 + extended_arg\r
- extended_arg = 0\r
- i = i+2\r
- if op == EXTENDED_ARG:\r
- extended_arg = oparg*65536L\r
- print repr(oparg).rjust(5),\r
- if op in hasconst:\r
- print '(' + repr(co.co_consts[oparg]) + ')',\r
- elif op in hasname:\r
- print '(' + co.co_names[oparg] + ')',\r
- elif op in hasjrel:\r
- print '(to ' + repr(i + oparg) + ')',\r
- elif op in haslocal:\r
- print '(' + co.co_varnames[oparg] + ')',\r
- elif op in hascompare:\r
- print '(' + cmp_op[oparg] + ')',\r
- elif op in hasfree:\r
- if free is None:\r
- free = co.co_cellvars + co.co_freevars\r
- print '(' + free[oparg] + ')',\r
- print\r
-\r
-def disassemble_string(code, lasti=-1, varnames=None, names=None,\r
- constants=None):\r
- labels = findlabels(code)\r
- n = len(code)\r
- i = 0\r
- while i < n:\r
- c = code[i]\r
- op = ord(c)\r
- if i == lasti: print '-->',\r
- else: print ' ',\r
- if i in labels: print '>>',\r
- else: print ' ',\r
- print repr(i).rjust(4),\r
- print opname[op].ljust(15),\r
- i = i+1\r
- if op >= HAVE_ARGUMENT:\r
- oparg = ord(code[i]) + ord(code[i+1])*256\r
- i = i+2\r
- print repr(oparg).rjust(5),\r
- if op in hasconst:\r
- if constants:\r
- print '(' + repr(constants[oparg]) + ')',\r
- else:\r
- print '(%d)'%oparg,\r
- elif op in hasname:\r
- if names is not None:\r
- print '(' + names[oparg] + ')',\r
- else:\r
- print '(%d)'%oparg,\r
- elif op in hasjrel:\r
- print '(to ' + repr(i + oparg) + ')',\r
- elif op in haslocal:\r
- if varnames:\r
- print '(' + varnames[oparg] + ')',\r
- else:\r
- print '(%d)' % oparg,\r
- elif op in hascompare:\r
- print '(' + cmp_op[oparg] + ')',\r
- print\r
-\r
-disco = disassemble # XXX For backwards compatibility\r
-\r
-def findlabels(code):\r
- """Detect all offsets in a byte code which are jump targets.\r
-\r
- Return the list of offsets.\r
-\r
- """\r
- labels = []\r
- n = len(code)\r
- i = 0\r
- while i < n:\r
- c = code[i]\r
- op = ord(c)\r
- i = i+1\r
- if op >= HAVE_ARGUMENT:\r
- oparg = ord(code[i]) + ord(code[i+1])*256\r
- i = i+2\r
- label = -1\r
- if op in hasjrel:\r
- label = i+oparg\r
- elif op in hasjabs:\r
- label = oparg\r
- if label >= 0:\r
- if label not in labels:\r
- labels.append(label)\r
- return labels\r
-\r
-def findlinestarts(code):\r
- """Find the offsets in a byte code which are start of lines in the source.\r
-\r
- Generate pairs (offset, lineno) as described in Python/compile.c.\r
-\r
- """\r
- byte_increments = [ord(c) for c in code.co_lnotab[0::2]]\r
- line_increments = [ord(c) for c in code.co_lnotab[1::2]]\r
-\r
- lastlineno = None\r
- lineno = code.co_firstlineno\r
- addr = 0\r
- for byte_incr, line_incr in zip(byte_increments, line_increments):\r
- if byte_incr:\r
- if lineno != lastlineno:\r
- yield (addr, lineno)\r
- lastlineno = lineno\r
- addr += byte_incr\r
- lineno += line_incr\r
- if lineno != lastlineno:\r
- yield (addr, lineno)\r
-\r
-def _test():\r
- """Simple test program to disassemble a file."""\r
- if sys.argv[1:]:\r
- if sys.argv[2:]:\r
- sys.stderr.write("usage: python dis.py [-|file]\n")\r
- sys.exit(2)\r
- fn = sys.argv[1]\r
- if not fn or fn == "-":\r
- fn = None\r
- else:\r
- fn = None\r
- if fn is None:\r
- f = sys.stdin\r
- else:\r
- f = open(fn)\r
- source = f.read()\r
- if fn is not None:\r
- f.close()\r
- else:\r
- fn = "<stdin>"\r
- code = compile(source, fn, "exec")\r
- dis(code)\r
-\r
-if __name__ == "__main__":\r
- _test()\r