B
cGDF+d@M @ 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 Zdd ZdOdddddZdPddddZddddddd d d d d
Zd!d" Zd#d$ Zd%d Zd&d' Zddd(d Zed)d*Zd+ej_ d,ej_ d-ej _ d.ej _ d/ej _ d0ej _ d1ej!_ d2ej"_ d3Z#d4Z$G d5d
d
eZ%dd6d7d Z&d8d9 Z'd:d; Z(dQd
)r c C s\ | d krDy
t j} W n tk
r0 tdd Y nX x| jrB| j} q4W t| jj| j|d d S )Nz no last traceback to disassemble)r ) syslast_tracebackAttributeError RuntimeErrortb_nextr tb_framef_codetb_lasti)tbr r r r r O s
Z OPTIMIZEDZ NEWLOCALSZVARARGSZ VARKEYWORDSZNESTEDZ GENERATORZNOFREEZ COROUTINEZITERABLE_COROUTINEZASYNC_GENERATOR)
@ i c C sb g }xRt dD ]8}d|> }| |@ r|t|t| | |N } | sP qW |t| d|S )NrB r= z, )rangeappendCOMPILER_FLAG_NAMESgethexjoin)flagsnamesiflagr r r pretty_flagsi s rP 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" z(don't know how to disassemble %s objects) r# r r r r r r& r. r r) r0 r1 )r2 r r r _get_code_objectw s
rQ c C s t t| S )N)_format_code_inforQ )r2 r r r r s c C s g }| d| j | d| j | d| j | d| j | d| j | d| j | dt| j | j r| d x t
| j D ]}| d | qW | j r| d
x t
| j D ]}| d | qW | j r| d x t
| j D ]}| d | qW | j
rF| d
x"t
| j
D ]}| d | q.W | jr|| d x"t
| jD ]}| d | qdW d|S )NzName: %szFilename: %szArgument count: %szKw-only arguments: %szNumber of locals: %szStack size: %szFlags: %sz
Constants:z%4d: %rzNames:z%4d: %szVariable names:zFree variables:zCell variables:
)rG co_name co_filename co_argcountco_kwonlyargcount
co_nlocals co_stacksizerP co_flags co_consts enumerateco_names co_varnames co_freevars co_cellvarsrK )colinesZi_cZi_nr r r rR s:
rR c C s t t| |d d S )N)r )r( r )ra 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
Fr? 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_linerG is_jump_targetreproffsetrjustopnameljust
_OPNAME_WIDTHarg _OPARG_WIDTHargreprrK rstrip)self lineno_widthZmark_as_current offset_widthZfieldsZ
lineno_fmtr r r _disassemble s&
zInstruction._disassembleN)rf Fr? )r1
__module__ __qualname__ry 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 ) rQ r` r_ dictr co_firstlineno_get_instructions_bytesr" r^ r] r[ )r2 r| ra
cell_names
linestarts line_offsetr r r r s c C s | }|d k r|| }|t |fS )N)rl )Z const_indexZ
const_listargvalr r r _get_const_info s r c C s* | }|d k r|| }|}nt |}||fS )N)rl )Z
name_indexZ name_listr rt r r r _get_name_info s r c
c sx t | }d }xdt| D ]V\} }
} |d k rH|| d }|d k rH||7 }| |k} d }
d}| d k rT| }
|
tkr~t| |\}
}n|
tkrt| |\}
}n|
tkr| d | }
dt|
}n|
t krt| |\}
}n|
t
krt | }
|
}nn|
t krt| |\}
}nV|
t
krTd tttf| d@ t| d@ f}
d| d@ }|
d rT|rL|d7 }|d 7 }tt|
|
| |
|| || V qW d S )
N r> zto rf r? )r r. rl asciir= z, z with format)r
_unpack_opargsrI Zhasconstr Zhasnamer hasjrelrl ZhaslocalZ
hascompareZcmp_opZhasfreer r. r boolr
ro )codevarnamesrM constantscellsr r labelsrj rm oprr rk r rt r r r r - sH
r c
C s< | j | j }tt| }t| j|| j| j| j|||d d S )N)r ) r` r_ r} r r- r" r^ r] r[ )ra lastir r r r r r r ` s c C st t | |d |d ks |dkrp|d k r,|d }xB| jD ]8}t|dr4t|d td|f |d t|||d q4W d S )N)r r r= r" zDisassembly of %r:)r r )r r[ r# r( r* )ra r r r2 r r r r* g s
r* )r r c C s |d k } | r8t | | }
|
dkr2tt|
} q i' r? )r )r ) maxvalueslenr. r rj rm r( ry )r r r rM r r r r r Z show_linenoZ maxlinenorw Z maxoffsetrx ZinstrZnew_source_lineZis_current_instrr r r r- r s,
r- c K s t t| df| d S )Nz)r* r )r kwargsr r r r/ s r/ c c sh d}x^t dt| dD ]J}| | }|tkrP| |d |B }|tkrJ|d> nd}nd }|||fV qW d S )Nr r> r= r@ )rF r Z
HAVE_ARGUMENTZ EXTENDED_ARG)r Z extended_argrN r rr r r r r s r c C s` g }xVt | D ]J\}}}|d k r|tkr6|d | }n|tkr|}nq||kr|| qW |S )Nr> )r r ZhasjabsrG )r r rm r rr Zlabelr r r r
s c c s | j dd d }| j dd d }d }| j}d}xPt||D ]B\}}|rd||kr\||fV |}||7 }|dkrt|d8 }||7 }q:W ||kr||fV d S )Nr r> r= rD rE ) co_lnotabr~ zip)r Zbyte_incrementsZline_incrementsZ
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 )Nr )
rQ codeobjr~ r| _line_offsetr` r_ _cell_namesr} r _linestarts_original_objectr )rv r2 r| r ra r r r __init__ s zBytecode.__init__c C s* | j }t|j|j|j|j| j| j| jdS )N)r ) r r r" r^ r] r[ r r r )rv ra r r r __iter__ s
zBytecode.__iter__c C s d | jj| jS )Nz{}({!r}))format __class__r1 r )rv r r r __repr__ s
zBytecode.__repr__c C s$ x|j r|j }qW | |jj|jdS )N)r )r8 r9 r: r; )clsr< r r r from_traceback s
zBytecode.from_tracebackc C s
t | jS )N)rR r )rv r r r info s z
Bytecode.infoc C s` | j }| jd k r| j}nd}t 2}t|j|j|j|j| j | j
| j ||d | S Q R X d S )Nr )r rM r r r r r r )
r r ioStringIOr- r" r^ r] r[ r r r getvalue)rv ra rm outputr r r r s
z Bytecode.dis)
r1 rz r{ r r r classmethodr r r r r r r r s
c C sd dd l } | }|jd| ddd | }|j}| }W d Q R X t||jjd}t | d S )Nr infile?-)r0 nargsdefaultr )
argparseArgumentParser add_argumentFileType
parse_argsr readr r r )r parserargsr r r r r r _test
s r __main__)N)N)NNNNNr )r )r NNNNN)5r4 types collectionsr Zopcoder Z _opcodes_all
MethodType FunctionTypeCodeTyper staticmethodr0 r' Zopmapr r r r rH rP rQ r rR r
namedtuplerc ro __doc__rr r rt rm rj rk rq rs r
r r r r r r* r- r/ r r r
r r r r1 r r r r s|
- 4
2 =