U
e5dZP @ s d Z ddlZddlZddlZddlZddlT ddlmZ ddddd d
d d d
ddg e Z[ejej ej
e e e
fZed ZdedfedfedffZed ZdZdd ZdVdddddZdWddd dZd d d d d!d"d#d$d%d&d'
Zd(d) Zd*d+ Z d,d Z d-d. Z ddd/d Z e d0d1Z!d2e!j"_ d3e!j_ d4e!j#_ d5e!j$_ d6e!j%_ d7e!j&_ d8e!j'_ d9e!j(_ d:Z)d;Z*G dd
Z,d?d@ Z-dAdB Z.dXdCdDZ/dYdddFdZ0ddddGdHZ1dZdddIdJdKZ2dLdM Z3e0Z4dNdO Z5dPd Z6dQd
Z7G dRd dZ8dSdT Z9e:dUkre9 dS )[z0Disassembler of Python byte code into mnemonics. N)*)__all__ code_infodis disassembledistbdiscofindlinestarts
findlabels show_codeget_instructions InstructionBytecode FORMAT_VALUE)N strreprascii
MAKE_FUNCTION)defaultsZ
kwdefaultsZ annotationsZclosurec C s6 zt | |d}W n tk
r0 t | |d}Y nX |S )zAttempts to compile the given source, first as an expression and
then as a statement if the first approach fails.
Utility function to accept strings in functions that otherwise
expect code objects
evalexec)compile SyntaxError)sourcenamec r /usr/lib64/python3.8/dis.py _try_compile s
r filedepthc C sh | dkrt |d dS t| dr&| j} t| dr8| j} n4t| drJ| j} n"t| dr\| j} nt| drl| j} t| drt| j }|D ]p\}}t
|t rt d | |d zt
|||d
W n0 tk
r } zt d ||d W 5 d}~X Y nX t |d qnht| d rt| ||d
nLt
| ttfr6t| |d n.t
| trRt| ||d
ntd
t| j dS )a0 Disassemble classes, methods, functions, and other compiled objects.
With no argument, disassemble the last traceback.
Compiled objects currently include generator objects, async generator
objects, and coroutine objects, all of which store their code object
in a special attribute.
Nr! __func____code__gi_codeag_codecr_code__dict__zDisassembly of %s:r zSorry:co_code(don't know how to disassemble %s objects)r hasattrr$ r% r& r' r( sortedr) items
isinstance
_have_codeprintr TypeError_disassemble_recursivebytes bytearray_disassemble_bytesr _disassemble_strtype__name__)xr! r" r. r Zx1msgr r r r + s@
r# c C sX | dkr@z
t j} W n tk
r0 tddY nX | jr@| j} q2t| jj| j|d dS )z2Disassemble a traceback (default: last traceback).Nz no last traceback to disassembler# ) syslast_tracebackAttributeError RuntimeErrortb_nextr tb_framef_codetb_lasti)tbr! r r r r X s
Z OPTIMIZEDZ NEWLOCALSZVARARGSZ VARKEYWORDSZNESTEDZ GENERATORZNOFREEZ COROUTINEZITERABLE_COROUTINEZASYNC_GENERATOR)
@ i c C s` g }t dD ]:}d|> }| |@ r |t|t| | |N } | s qVq |t| d|S )z+Return pretty representation of code flags.rJ rE , )rangeappendCOMPILER_FLAG_NAMESgethexjoin)flagsnamesiflagr r r pretty_flagsr s rY c C s t | dr| j} t | dr"| j} n4t | dr4| j} n"t | drF| j} nt | drV| j} t| trjt| d} t | drx| S t dt
| j d S )
zDHelper to handle methods, compiled or raw code objects, and strings.r$ r% r& r' r( z
r* r+ N) r, r$ r% r& r' r( r/ r r r2 r8 r9 r: r r r _get_code_object s"
r[ c C s t t| S )z1Formatted details of methods, functions, or code.)_format_code_infor[ rZ r r r r s c C s g }| d| j | d| j | d| j | d| j | d| j | d| j | d| j | dt| j | j
r| d t | j
D ]}| d
| q| j r| d t | j D ]}| d | q| j
r| d
t | j
D ]}| d | q | jrH| d t | jD ]}| d | q2| jrz| d t | jD ]}| d | qdd|S )NzName: %szFilename: %szArgument count: %sz Positional-only arguments: %szKw-only arguments: %szNumber of locals: %szStack size: %szFlags: %sz
Constants:z%4d: %rzNames:z%4d: %szVariable names:zFree variables:zCell variables:
)rP co_name co_filename co_argcountco_posonlyargcountco_kwonlyargcount
co_nlocals co_stacksizerY co_flags co_consts enumerateco_names co_varnames co_freevars co_cellvarsrT )colinesZi_cZi_nr r r r\ s<
r\ c C s t t| |d dS )z}Print details of methods, functions, or code to *file*.
If *file* is not provided, the output is printed on stdout.
r# N)r1 r )rl r! r r r r s _InstructionzBopname opcode arg argval argrepr offset starts_line is_jump_targetz!Human readable name for operationzNumeric code for operationz6Numeric argument to operation (if any), otherwise Nonez4Resolved arg value (if known), otherwise same as argz0Human readable description of operation argumentz1Start index of operation within bytecode sequencez4Line started by this opcode (if any), otherwise Nonez1True if other code jumps to here, otherwise False c @ s e Zd ZdZdddZdS ) r
aK Details for a bytecode operation
Defined fields:
opname - human readable name for operation
opcode - numeric code for operation
arg - numeric argument to operation (if any), otherwise None
argval - resolved arg value (if known), otherwise same as arg
argrepr - human readable description of operation argument
offset - start index of operation within bytecode sequence
starts_line - line started by this opcode (if any), otherwise None
is_jump_target - True if other code jumps to here, otherwise False
FrG c C s g }|r:| j dk r,d| }||| j n|d| |rJ|d n
|d | jrf|d n
|d |t| j| || jt | j dk r|t| j t
| j r|d| j d d |
S )
a% Format instruction details for inclusion in disassembly output
*lineno_width* sets the width of the line number field (0 omits it)
*mark_as_current* inserts a '-->' marker arrow as part of the line
*offset_width* sets the width of the instruction offset field
Nz%%%dd z-->z z>>z ()) starts_linerP is_jump_targetr offsetrjustopnameljust
_OPNAME_WIDTHarg _OPARG_WIDTHargreprrT rstrip)self lineno_widthZmark_as_current offset_widthZfieldsZ
lineno_fmtr r r _disassemble s&
zInstruction._disassembleN)rq FrG )r9
__module__ __qualname____doc__r r r r r r
s
)
first_linec C sT t | }|j|j }tt|}|dk r4||j }nd}t|j|j|j |j
|||S )a Iterator for the opcodes in methods, functions or code
Generates a series of Instruction named tuples giving the details of
each operations in the supplied code.
If *first_line* is not None, it indicates the line number that should
be reported for the first source line in the disassembled code.
Otherwise, the source line information (if any) is taken directly from
the disassembled code object.
Nr ) r[ rk rj dictr co_firstlineno_get_instructions_bytesr* ri rh rf )r: r rl
cell_names
linestarts line_offsetr r r r s c C s | }|dk r|| }|t |fS )zHelper to get optional details about const references
Returns the dereferenced constant and its repr if the constant
list is defined.
Otherwise returns the constant index and its repr().
Nr )Z const_indexZ
const_listargvalr r r _get_const_info s r c C s* | }|dk r|| }|}nt |}||fS )zHelper to get optional details about named references
Returns the dereferenced name as both value and repr if the name
list is defined.
Otherwise returns the name index and its repr().
Nr )Z
name_indexZ name_listr r~ r r r _get_name_info' s r c
# s t | }d}t| D ]r\} }
|dk rD|| d}|dk rD||7 }| |k} d} d}
dk rl } |
tkrzt |\} }
n|
tkrt |\} }
n|
tkr| d } dt| }
n|
t krt |\} }
n|
t
krt } | }
n|
t krt |\} }
nr|
t
krFt d@ \} }
| t d@ f} | d rl|
r<|
d7 }
|
d 7 }
n&|
tkrld fd
d ttD }
tt|
|
| |
| || V qdS ) a& Iterate over the instructions in a bytecode string.
Generates a sequence of Instruction namedtuples giving the details of each
opcode. Additional information about the code's runtime environment
(e.g. variable names, constants) can be specified using optional
arguments.
Nr rF zto rq rG rE rN z with formatc 3 s" | ]\}} d |> @ r|V qdS )rE Nr ).0rW sr| r r g s z*_get_instructions_bytes..)r
_unpack_opargsrR Zhasconstr Zhasnamer hasjrelr ZhaslocalZ
hascompareZcmp_opZhasfreer FORMAT_VALUE_CONVERTERSboolr rT rg MAKE_FUNCTION_FLAGSr
ry )codevarnamesrV constantscellsr r labelsru rw oprv r r~ r r r r 7 sV
r c
C s< | j | j }tt| }t| j|| j| j| j|||d dS )zDisassemble a code object.r# N) rk rj r r r6 r* ri rh rf )rl lastir! r r r r r r m s c C sp t | |d |d ks |dkrl|d k r,|d }| jD ]8}t|dr2t|d td|f |d t|||d q2d S )Nr# r rE r* zDisassembly of %r:r )r rf r, r1 r3 )rl r! r" r: r r r r3 t s
r3 )r! r c C s |d k } | r8t | | }
|
dkr2tt|
} qN)r3 r )r kwargsr r r r7 s r7 c c sd d}t dt| dD ]J}| | }|tkrN| |d |B }|tkrH|d> nd}nd }|||fV qd S )Nr rF rE rH )rO r Z
HAVE_ARGUMENTZ EXTENDED_ARG)r Z extended_argrW r r| r r r r s r c C s\ g }t | D ]J\}}}|dk r |tkr4|d | }n|tkr |}nq ||kr || q |S )z`Detect all offsets in a byte code which are jump targets.
Return the list of offsets.
NrF )r r ZhasjabsrP )r r rw r r| Zlabelr r r r
s c c s | j ddd }| j ddd }t| j}d}| j}d}t||D ]P\}}|rz||krd||fV |}||7 }||krz dS |dkr|d8 }||7 }qB||kr||fV dS )zFind the offsets in a byte code which are start of lines in the source.
Generate pairs (offset, lineno) as described in Python/compile.c.
r NrF rE rL rM ) co_lnotabr r* r zip) r Zbyte_incrementsZline_incrementsZ bytecode_lenZ
lastlinenolinenoZaddrZ byte_incrZ line_incrr r r r s&
c @ sL e Zd ZdZdddddZdd Zdd Zed
d Zd d
Z dd Z
dS )r a The bytecode operations of a piece of code
Instantiate this with a function, method, other compiled object, string of
code, or a code object (as returned by compile()).
Iterating over this yields the bytecode operations as Instruction instances.
N)r current_offsetc C sd t | | _}|d kr&|j| _d| _n|| _||j | _|j|j | _tt || _
|| _ || _ d S )Nr )
r[ codeobjr r _line_offsetrk rj _cell_namesr r _linestarts_original_objectr )r r: r r rl r r r __init__ s zBytecode.__init__c C s* | j }t|j|j|j|j| j| j| jdS )Nr ) r r r* ri rh rf r r r )r rl r r r __iter__ s zBytecode.__iter__c C s d | jj| jS )Nz{}({!r}))format __class__r9 r r r r r __repr__ s
zBytecode.__repr__c C s |j r|j }q | |jj|jdS )z/ Construct a Bytecode from the given traceback )r )r@ rA rB rC )clsrD r r r from_traceback s zBytecode.from_tracebackc C s
t | jS )z3Return formatted information about the code object.)r\ r r r r r info s z
Bytecode.infoc C sl | j }| jdk r| j}nd}t >}t|j|j|j|j| j | j
| j ||d | W 5 Q R S Q R X dS )z3Return a formatted view of the bytecode operations.Nr )r rV r r r r r! r )
r r ioStringIOr6 r* ri rh rf r r r getvalue)r rl rw outputr r r r
s
z Bytecode.dis) r9 r r r r r r classmethodr r r r r r r r s
c C sf ddl } | }|jd| dddd | }|j}| }W 5 Q R X t||jjd}t | dS ) z*Simple test program to disassemble a file.r Ninfilerb?-)r8 nargsdefaultr )
argparseArgumentParser add_argumentZFileType
parse_argsr readr r r )r parserargsr r r r r r _test s r __main__)N)N)NNNNNr )r )r NNNNN);r r< types collectionsr Zopcoder Z _opcodes_all
MethodType FunctionTypeCodeTyper staticmethodr8 r0 Zopmapr r r r r r r r r r rQ rY r[ r r\ r
namedtuplern ry r| r r~ rw ru rv r{ r} r
r r r r r r3 r6 r7 r r r
r r r r9 r r r r s
-
4
6 =