annotate Lib/dis.py @ 79675:c0a2294de787 2.7

Issue #16199: Backport 2to3 typo fix from 3.2.
author Chris Jerdonek <chris.jerdonek@gmail.com>
date Thu, 11 Oct 2012 16:15:46 -0700
parents b36cb4602e21
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
6592
14025a4f793e Added docstrings (contributed by Martin von Loewis).
Guido van Rossum <guido@python.org>
parents: 5475
diff changeset
1 """Disassembler of Python byte code into mnemonics."""
103
1229ba124e0b Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
2
1229ba124e0b Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
3 import sys
5189
bab9196a8f30 Support disassembly of a variety of objects through dis.dis().
Guido van Rossum <guido@python.org>
parents: 4989
diff changeset
4 import types
103
1229ba124e0b Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
5
28092
12bab87a3dd2 * separate opcode definitions into opcode.py
Skip Montanaro <skip@pobox.com>
parents: 25078
diff changeset
6 from opcode import *
12bab87a3dd2 * separate opcode definitions into opcode.py
Skip Montanaro <skip@pobox.com>
parents: 25078
diff changeset
7 from opcode import __all__ as _opcodes_all
12bab87a3dd2 * separate opcode definitions into opcode.py
Skip Montanaro <skip@pobox.com>
parents: 25078
diff changeset
8
50409
20edb7e5159f #4222: document dis.findlabels() and dis.findlinestarts() and
Georg Brandl <georg@python.org>
parents: 31515
diff changeset
9 __all__ = ["dis", "disassemble", "distb", "disco",
20edb7e5159f #4222: document dis.findlabels() and dis.findlinestarts() and
Georg Brandl <georg@python.org>
parents: 31515
diff changeset
10 "findlinestarts", "findlabels"] + _opcodes_all
28092
12bab87a3dd2 * separate opcode definitions into opcode.py
Skip Montanaro <skip@pobox.com>
parents: 25078
diff changeset
11 del _opcodes_all
16070
b5345e99596b added __all__ lists to a number of Python modules
Skip Montanaro <skip@pobox.com>
parents: 16012
diff changeset
12
60332
33a02e4ac4c2 Merged revisions 79769 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 31515
diff changeset
13 _have_code = (types.MethodType, types.FunctionType, types.CodeType,
33a02e4ac4c2 Merged revisions 79769 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 31515
diff changeset
14 types.ClassType, type)
33a02e4ac4c2 Merged revisions 79769 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 31515
diff changeset
15
4988
df385cc0d2e3 More user friedly interface:
Guido van Rossum <guido@python.org>
parents: 4974
diff changeset
16 def dis(x=None):
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
17 """Disassemble classes, methods, functions, or code.
6592
14025a4f793e Added docstrings (contributed by Martin von Loewis).
Guido van Rossum <guido@python.org>
parents: 5475
diff changeset
18
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
19 With no argument, disassemble the last traceback.
6592
14025a4f793e Added docstrings (contributed by Martin von Loewis).
Guido van Rossum <guido@python.org>
parents: 5475
diff changeset
20
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
21 """
23611
21b1d742e4a7 Replaced boolean test with 'is None'
Raymond Hettinger <python@rcn.com>
parents: 21924
diff changeset
22 if x is None:
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
23 distb()
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
24 return
53988
630e681100e0 use isinstance
Benjamin Peterson <benjamin@python.org>
parents: 50409
diff changeset
25 if isinstance(x, types.InstanceType):
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
26 x = x.__class__
16012
8e4dfb681689 Checking in a slight variation of Barry's patch 103303.
Guido van Rossum <guido@python.org>
parents: 15867
diff changeset
27 if hasattr(x, 'im_func'):
8e4dfb681689 Checking in a slight variation of Barry's patch 103303.
Guido van Rossum <guido@python.org>
parents: 15867
diff changeset
28 x = x.im_func
8e4dfb681689 Checking in a slight variation of Barry's patch 103303.
Guido van Rossum <guido@python.org>
parents: 15867
diff changeset
29 if hasattr(x, 'func_code'):
8e4dfb681689 Checking in a slight variation of Barry's patch 103303.
Guido van Rossum <guido@python.org>
parents: 15867
diff changeset
30 x = x.func_code
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
31 if hasattr(x, '__dict__'):
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
32 items = x.__dict__.items()
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
33 items.sort()
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
34 for name, x1 in items:
60332
33a02e4ac4c2 Merged revisions 79769 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 31515
diff changeset
35 if isinstance(x1, _have_code):
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
36 print "Disassembly of %s:" % name
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
37 try:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
38 dis(x1)
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
39 except TypeError, msg:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
40 print "Sorry:", msg
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
41 print
16012
8e4dfb681689 Checking in a slight variation of Barry's patch 103303.
Guido van Rossum <guido@python.org>
parents: 15867
diff changeset
42 elif hasattr(x, 'co_code'):
8e4dfb681689 Checking in a slight variation of Barry's patch 103303.
Guido van Rossum <guido@python.org>
parents: 15867
diff changeset
43 disassemble(x)
28092
12bab87a3dd2 * separate opcode definitions into opcode.py
Skip Montanaro <skip@pobox.com>
parents: 25078
diff changeset
44 elif isinstance(x, str):
12bab87a3dd2 * separate opcode definitions into opcode.py
Skip Montanaro <skip@pobox.com>
parents: 25078
diff changeset
45 disassemble_string(x)
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
46 else:
16012
8e4dfb681689 Checking in a slight variation of Barry's patch 103303.
Guido van Rossum <guido@python.org>
parents: 15867
diff changeset
47 raise TypeError, \
8e4dfb681689 Checking in a slight variation of Barry's patch 103303.
Guido van Rossum <guido@python.org>
parents: 15867
diff changeset
48 "don't know how to disassemble %s objects" % \
8e4dfb681689 Checking in a slight variation of Barry's patch 103303.
Guido van Rossum <guido@python.org>
parents: 15867
diff changeset
49 type(x).__name__
103
1229ba124e0b Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
50
4988
df385cc0d2e3 More user friedly interface:
Guido van Rossum <guido@python.org>
parents: 4974
diff changeset
51 def distb(tb=None):
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
52 """Disassemble a traceback (default: last traceback)."""
23611
21b1d742e4a7 Replaced boolean test with 'is None'
Raymond Hettinger <python@rcn.com>
parents: 21924
diff changeset
53 if tb is None:
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
54 try:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
55 tb = sys.last_traceback
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
56 except AttributeError:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
57 raise RuntimeError, "no last traceback to disassemble"
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
58 while tb.tb_next: tb = tb.tb_next
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
59 disassemble(tb.tb_frame.f_code, tb.tb_lasti)
103
1229ba124e0b Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
60
4988
df385cc0d2e3 More user friedly interface:
Guido van Rossum <guido@python.org>
parents: 4974
diff changeset
61 def disassemble(co, lasti=-1):
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
62 """Disassemble a code object."""
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
63 code = co.co_code
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
64 labels = findlabels(code)
30803
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
65 linestarts = dict(findlinestarts(co))
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
66 n = len(code)
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
67 i = 0
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
68 extended_arg = 0
16260
20ef849ec877 PEP 227 implementation
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 16070
diff changeset
69 free = None
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
70 while i < n:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
71 c = code[i]
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
72 op = ord(c)
30803
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
73 if i in linestarts:
24844
cf5928fab108 This is my patch
Michael W. Hudson <mwh@python.net>
parents: 23829
diff changeset
74 if i > 0:
cf5928fab108 This is my patch
Michael W. Hudson <mwh@python.net>
parents: 23829
diff changeset
75 print
30803
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
76 print "%3d" % linestarts[i],
24844
cf5928fab108 This is my patch
Michael W. Hudson <mwh@python.net>
parents: 23829
diff changeset
77 else:
cf5928fab108 This is my patch
Michael W. Hudson <mwh@python.net>
parents: 23829
diff changeset
78 print ' ',
cf5928fab108 This is my patch
Michael W. Hudson <mwh@python.net>
parents: 23829
diff changeset
79
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
80 if i == lasti: print '-->',
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
81 else: print ' ',
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
82 if i in labels: print '>>',
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
83 else: print ' ',
31515
71a294bd7950 Replace backticks with repr() or "%r"
Walter Dörwald <walter@livinglogic.de>
parents: 30803
diff changeset
84 print repr(i).rjust(4),
16492
fc4ace0242cb String method conversion.
Eric S. Raymond <esr@thyrsus.com>
parents: 16398
diff changeset
85 print opname[op].ljust(20),
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
86 i = i+1
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
87 if op >= HAVE_ARGUMENT:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
88 oparg = ord(code[i]) + ord(code[i+1])*256 + extended_arg
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
89 extended_arg = 0
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
90 i = i+2
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
91 if op == EXTENDED_ARG:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
92 extended_arg = oparg*65536L
31515
71a294bd7950 Replace backticks with repr() or "%r"
Walter Dörwald <walter@livinglogic.de>
parents: 30803
diff changeset
93 print repr(oparg).rjust(5),
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
94 if op in hasconst:
31515
71a294bd7950 Replace backticks with repr() or "%r"
Walter Dörwald <walter@livinglogic.de>
parents: 30803
diff changeset
95 print '(' + repr(co.co_consts[oparg]) + ')',
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
96 elif op in hasname:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
97 print '(' + co.co_names[oparg] + ')',
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
98 elif op in hasjrel:
31515
71a294bd7950 Replace backticks with repr() or "%r"
Walter Dörwald <walter@livinglogic.de>
parents: 30803
diff changeset
99 print '(to ' + repr(i + oparg) + ')',
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
100 elif op in haslocal:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
101 print '(' + co.co_varnames[oparg] + ')',
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
102 elif op in hascompare:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
103 print '(' + cmp_op[oparg] + ')',
16260
20ef849ec877 PEP 227 implementation
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 16070
diff changeset
104 elif op in hasfree:
20ef849ec877 PEP 227 implementation
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 16070
diff changeset
105 if free is None:
20ef849ec877 PEP 227 implementation
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 16070
diff changeset
106 free = co.co_cellvars + co.co_freevars
20ef849ec877 PEP 227 implementation
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 16070
diff changeset
107 print '(' + free[oparg] + ')',
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
108 print
103
1229ba124e0b Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
109
28092
12bab87a3dd2 * separate opcode definitions into opcode.py
Skip Montanaro <skip@pobox.com>
parents: 25078
diff changeset
110 def disassemble_string(code, lasti=-1, varnames=None, names=None,
12bab87a3dd2 * separate opcode definitions into opcode.py
Skip Montanaro <skip@pobox.com>
parents: 25078
diff changeset
111 constants=None):
28231
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
112 labels = findlabels(code)
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
113 n = len(code)
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
114 i = 0
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
115 while i < n:
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
116 c = code[i]
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
117 op = ord(c)
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
118 if i == lasti: print '-->',
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
119 else: print ' ',
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
120 if i in labels: print '>>',
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
121 else: print ' ',
31515
71a294bd7950 Replace backticks with repr() or "%r"
Walter Dörwald <walter@livinglogic.de>
parents: 30803
diff changeset
122 print repr(i).rjust(4),
28231
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
123 print opname[op].ljust(15),
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
124 i = i+1
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
125 if op >= HAVE_ARGUMENT:
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
126 oparg = ord(code[i]) + ord(code[i+1])*256
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
127 i = i+2
31515
71a294bd7950 Replace backticks with repr() or "%r"
Walter Dörwald <walter@livinglogic.de>
parents: 30803
diff changeset
128 print repr(oparg).rjust(5),
28231
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
129 if op in hasconst:
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
130 if constants:
31515
71a294bd7950 Replace backticks with repr() or "%r"
Walter Dörwald <walter@livinglogic.de>
parents: 30803
diff changeset
131 print '(' + repr(constants[oparg]) + ')',
28231
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
132 else:
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
133 print '(%d)'%oparg,
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
134 elif op in hasname:
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
135 if names is not None:
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
136 print '(' + names[oparg] + ')',
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
137 else:
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
138 print '(%d)'%oparg,
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
139 elif op in hasjrel:
31515
71a294bd7950 Replace backticks with repr() or "%r"
Walter Dörwald <walter@livinglogic.de>
parents: 30803
diff changeset
140 print '(to ' + repr(i + oparg) + ')',
28231
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
141 elif op in haslocal:
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
142 if varnames:
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
143 print '(' + varnames[oparg] + ')',
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
144 else:
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
145 print '(%d)' % oparg,
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
146 elif op in hascompare:
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
147 print '(' + cmp_op[oparg] + ')',
4cafa5141012 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28092
diff changeset
148 print
28092
12bab87a3dd2 * separate opcode definitions into opcode.py
Skip Montanaro <skip@pobox.com>
parents: 25078
diff changeset
149
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
150 disco = disassemble # XXX For backwards compatibility
4988
df385cc0d2e3 More user friedly interface:
Guido van Rossum <guido@python.org>
parents: 4974
diff changeset
151
103
1229ba124e0b Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
152 def findlabels(code):
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
153 """Detect all offsets in a byte code which are jump targets.
6592
14025a4f793e Added docstrings (contributed by Martin von Loewis).
Guido van Rossum <guido@python.org>
parents: 5475
diff changeset
154
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
155 Return the list of offsets.
6592
14025a4f793e Added docstrings (contributed by Martin von Loewis).
Guido van Rossum <guido@python.org>
parents: 5475
diff changeset
156
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
157 """
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
158 labels = []
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
159 n = len(code)
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
160 i = 0
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
161 while i < n:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
162 c = code[i]
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
163 op = ord(c)
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
164 i = i+1
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
165 if op >= HAVE_ARGUMENT:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
166 oparg = ord(code[i]) + ord(code[i+1])*256
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
167 i = i+2
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
168 label = -1
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
169 if op in hasjrel:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
170 label = i+oparg
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
171 elif op in hasjabs:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
172 label = oparg
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
173 if label >= 0:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
174 if label not in labels:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
175 labels.append(label)
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
176 return labels
103
1229ba124e0b Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
177
30803
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
178 def findlinestarts(code):
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
179 """Find the offsets in a byte code which are start of lines in the source.
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
180
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
181 Generate pairs (offset, lineno) as described in Python/compile.c.
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
182
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
183 """
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
184 byte_increments = [ord(c) for c in code.co_lnotab[0::2]]
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
185 line_increments = [ord(c) for c in code.co_lnotab[1::2]]
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
186
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
187 lastlineno = None
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
188 lineno = code.co_firstlineno
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
189 addr = 0
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
190 for byte_incr, line_incr in zip(byte_increments, line_increments):
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
191 if byte_incr:
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
192 if lineno != lastlineno:
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
193 yield (addr, lineno)
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
194 lastlineno = lineno
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
195 addr += byte_incr
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
196 lineno += line_incr
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
197 if lineno != lastlineno:
aecbba0294bb Fixed dis.disassemble_string().
Armin Rigo <arigo@tunes.org>
parents: 28231
diff changeset
198 yield (addr, lineno)
11709
5d625a365e76 Added a simple test program to disassemble a file, invoked as __main__.
Guido van Rossum <guido@python.org>
parents: 10822
diff changeset
199
5d625a365e76 Added a simple test program to disassemble a file, invoked as __main__.
Guido van Rossum <guido@python.org>
parents: 10822
diff changeset
200 def _test():
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
201 """Simple test program to disassemble a file."""
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
202 if sys.argv[1:]:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
203 if sys.argv[2:]:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
204 sys.stderr.write("usage: python dis.py [-|file]\n")
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
205 sys.exit(2)
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
206 fn = sys.argv[1]
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
207 if not fn or fn == "-":
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
208 fn = None
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
209 else:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
210 fn = None
23611
21b1d742e4a7 Replaced boolean test with 'is None'
Raymond Hettinger <python@rcn.com>
parents: 21924
diff changeset
211 if fn is None:
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
212 f = sys.stdin
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
213 else:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
214 f = open(fn)
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
215 source = f.read()
23611
21b1d742e4a7 Replaced boolean test with 'is None'
Raymond Hettinger <python@rcn.com>
parents: 21924
diff changeset
216 if fn is not None:
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
217 f.close()
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
218 else:
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
219 fn = "<stdin>"
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
220 code = compile(source, fn, "exec")
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
221 dis(code)
11709
5d625a365e76 Added a simple test program to disassemble a file, invoked as __main__.
Guido van Rossum <guido@python.org>
parents: 10822
diff changeset
222
5d625a365e76 Added a simple test program to disassemble a file, invoked as __main__.
Guido van Rossum <guido@python.org>
parents: 10822
diff changeset
223 if __name__ == "__main__":
15867
f5a3d98a35bc Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14171
diff changeset
224 _test()