annotate Lib/dis.py @ 58925:eecf1bef052d 2.6

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