U
e5dZP @ s d dl Z d dlZd dlZd dlZd dlT d dlmZ dddddd d
d d d
dg e Z[ejejej e
e e fZ
ed ZdedfedfedffZed ZdZdd ZdUdddddZdVddddZd d d d d d!d"d#d$d%d&
Zd'd( Zd)d* Zd+d Z d,d- Z ddd.d Z e d/d0Z d1e j!_"d2e j_"d3e j#_"d4e j$_"d5e j%_"d6e j&_"d7e j'_"d8e j(_"d9Z)d:Z*G d;d
d
e Z+dd<d=d Z,d>d? Z-d@dA Z.dWdBdCZ/dXdddEdZ0ddddFdGZ1dYdd dHdIdJZ2dKdL Z3e0Z4dMdN Z5dOd
Z6dPd Z7G dQd dZ8dRdS Z9e:dTkre9 dS )Z 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 )Nevalexec)compile SyntaxError)sourcenamec r /usr/lib64/python3.8/dis.py _try_compile s
r filedepthc C sh | d krt |d d S 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 d S )
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 | d kr@z
t j} W n tk
r0 tdd Y nX | jr@| j} q2t| jj| j|d d S )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 )NrJ 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 ) Nr$ r% r& r' r( z
r* r+ ) 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 N)_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 d S Nr# )r1 r )rm 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 ZdddZdS )r
FrG c C s g }|r:| j d k r,d| }||| j n|d| |rJ|d n
|d | jrf|d n
|d |t| j| || jt | j d k r|t| j t
| j r|d| j d d |
S ) 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)rs FrG )r9
__module__ __qualname__r r r r r r
s )
first_linec C sT t | }|j|j }tt|}|d k r4||j }nd}t|j|j|j |j
|||S Nr ) r[ rl rk dictr co_firstlineno_get_instructions_bytesr* rj ri rg )r: r rm
cell_names
linestarts line_offsetr r r r s c C s | }|d k r|| }|t |fS r\ r )Z const_indexZ
const_listargvalr r r _get_const_info s r c C s* | }|d k r|| }|}nt |}||fS r\ r )Z
name_indexZ name_listr r r r r _get_name_info' s r c
# s t | }d }t| D ]r\} }
|d k rD|| d }|d k rD||7 }| |k} d } d}
d k 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 }
|
d7 }
n&|
tkrld fd d
ttD }
tt|
|
| |
| || V qd S ) Nr rF zto rs 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 rh MAKE_FUNCTION_FLAGSr
r{ )codevarnamesrV constantscellsr r labelsrw ry oprx r r r r r r 7 sV
r c
C s< | j | j }tt| }t| j|| j| j| j|||d d S ro ) rl rk r r r6 r* rj ri rg )rm 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 rg r, r1 r3 )rm r! r" r: r r r r3 t s
r3 )r! r c C s |d k } | r8t | | }
|
dkr2tt|
} q)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\}}}|d k r |tkr4|d | }n|tkr |}nq ||kr || q |S )NrF )r r ZhasjabsrP )r r ry r r~ Zlabelr r r r
s c c s | j dd d }| j dd d }t| j}d }| j}d}t||D ]P\}}|rz||krd||fV |}||7 }||krz d S |dkr|d8 }||7 }qB||kr||fV d S )Nr rF rE rL rM ) co_lnotabr r* r zip) r Zbyte_incrementsZline_incrementsZ bytecode_lenZ
lastlinenolinenoZaddrZ byte_incrZ line_incrr r r r s&
c @ sH e Zd ZdddddZdd Zdd Zed d
Zd d Zd
d Z dS )r N)r current_offsetc C sd t | | _}|d kr&|j| _d| _n|| _||j | _|j|j | _tt || _
|| _ || _ d S r )
r[ codeobjr r _line_offsetrl rk _cell_namesr r _linestarts_original_objectr )r r: r r rm r r r __init__ s zBytecode.__init__c C s* | j }t|j|j|j|j| j| j| jdS )Nr ) r r r* rj ri rg r r r )r rm 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 )N)r )r@ rA rB rC )clsrD r r r from_traceback s zBytecode.from_tracebackc C s
t | jS r\ )r] r r r r r info s z
Bytecode.infoc C sl | j }| jd k r| j}nd}t >}t|j|j|j|j| j | j
| j ||d | W 5 Q R S Q R X d S )Nr )r rV r r r r r! r )
r r ioStringIOr6 r* rj ri rg r r r getvalue)r rm ry outputr r r r
s
z Bytecode.dis)
r9 r r r r r classmethodr r r r r r r r s
c C sf dd l } | }|jd| dddd | }|j}| }W 5 Q R X t||jjd}t | d S )Nr infilerb?-)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< 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
namedtuplerp r{ __doc__r~ r r ry rw rx 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 =