]C @ sn d d l Z d d l Z d d l Z d d l Z d d l Td d l m Z d d d d d d d
d d d
d g e Z [ e j e j e j e
e e f Z
d d Z d d d d d Z d d d d d Z d d d d d d d d d d d d d d! d" d# d$ d% i Z d& d' Z d( d) Z d* d Z d+ d, Z d d d- d Z e j d. d/ Z Gd0 d
d
e Z d1 d d2 d Z d3 d4 Z d5 d6 Z d d d d d d d7 d8 Z dG d d d9 d Z dH d d d d d d d d: d d; d< Z d d d= d> Z e Z! d? d@ Z" dA d
Z# dB d Z$ GdC d d Z% dD dE Z& e' dF k rje& d S)I N)*)__all__ code_infodis disassembledistbdiscofindlinestarts
findlabels show_codeget_instructions InstructionBytecodec C sA y t | | d } Wn$ t k
r< t | | d } Yn X| S)Nevalexec)compile SyntaxError)sourcenamec r (/opt/alt/python35/lib64/python3.5/dis.py _try_compile s
r filec C s | d k r t d | d St | d r5 | j } t | d rM | j } t | d re | j } t | d r"t | j j } x| D] \ } } t | t r t
d | d | y t | d | Wn8 t k
r
} z t
d | d | WYd d } ~ Xn Xt
d | q Wn t | d rDt
| d | nc t | t t f rlt | d | n; t | t rt | d | n t d t | j d S)
Nr __func____code__gi_code__dict__zDisassembly of %s:zSorry:co_codez(don't know how to disassemble %s objects)r hasattrr r r sortedr items
isinstance
_have_codeprintr TypeErrorr bytes bytearray_disassemble_bytesstr_disassemble_strtype__name__)xr r! r Zx1msgr r r r s6
&c
C ss | d k rS y
t j } Wn t k
r9 t d Yn Xx | j rR | j } q= Wt | j j | j d | d S)Nz no last traceback to disassembler ) syslast_tracebackAttributeError RuntimeErrortb_nextr tb_framef_codetb_lasti)tbr r r r r C s
Z OPTIMIZED Z NEWLOCALS ZVARARGS Z VARKEYWORDS ZNESTED Z GENERATOR@ ZNOFREE Z COROUTINE ZITERABLE_COROUTINEc C s g } xk t d D]J } d | >} | | @r | j t j | t | | | N} | s Pq W| j t | d j | S)Nr= r8 z, )rangeappendCOMPILER_FLAG_NAMESgethexjoin)flagsnamesiflagr r r pretty_flags\ s
rK c C s t | d r | j } t | d r0 | j } t | d rH | j } t | t rf t | d } t | d ry | St d t | j d S)Nr r r z
<disassembly>r z(don't know how to disassemble %s objects)
r r r r r" r) r r% r+ r, )r- r r r _get_code_objectj s rL c C s t t | S)N)_format_code_inforL )r- r r r r y s c C s g } | j d | j | j d | j | j d | j | j d | j | j d | j | j d | j | j d t | j | j r | j d x( t
| j D] } | j d | q W| j r| j d
x( t
| j D] } | j d | q W| j r[| j d x( t
| j D] } | j d | q@W| j
r| j d
x( t
| j
D] } | j d | qW| j r| j d x( t
| j D] } | j d | qWd j | 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:
)rB co_name co_filename co_argcountco_kwonlyargcount
co_nlocals co_stacksizerK co_flags co_consts enumerateco_names co_varnames co_freevars co_cellvarsrF )colinesZi_cZi_nr r r rM } s:
rM c C s t t | d | d S)Nr )r$ r )r\ r r r r r s _InstructionzBopname opcode arg argval argrepr offset starts_line is_jump_targetc @ s" e Z d Z d d d d Z d S)r
Fc C s0 g } | rM | j d k r< d | } | j | | j n | j d | | rc | j d n
| j d | j r | j d n
| j d | j t | j j d | j | j j d | j d k r | j t | j j d | j r | j d
| j d d j
| j S) Nz%%%dd z-->z z>>z r: ()) starts_linerB is_jump_targetreproffsetrjustopnameljustargargreprrF rstrip)self lineno_widthZmark_as_currentZfieldsZ
lineno_fmtr r r _disassemble s&
zInstruction._disassembleN)r,
__module__ __qualname__rq r r r r r
s
first_linec C sx t | } | j | j } t t | } | d k rJ | | j } n d } t | j | j | j | j
| | | S)Nr ) rL r[ rZ dictr co_firstlineno_get_instructions_bytesr rY rX rV )r- rt r\
cell_names
linestarts line_offsetr r r r s c C s, | } | d k r | | } | t | f S)N)rg )Z const_indexZ
const_listargvalr r r _get_const_info s
r| c C s; | } | d k r% | | } | } n t | } | | f S)N)rg )Z
name_indexZ name_listr{ rm r r r _get_name_info s
r} c
c s t | } d } d } xt | D]\ }
} } | d k rh | j |
d } | d k rh | | 7} |
| k }
d } d } | d k r| } | t k r t | | \ } } n | t k r t | | \ } } n | t k r|
d | } d t | } n | t k r+t | | \ } } ng | t
k rJt | } | } nH | t k rnt | | \ } } n$ | t
k rd | d | d f } t t | | | | | |
| |
Vq% Wd S)N r_ zto z %d positional, %d keyword pairr@ )r
_unpack_opargsrD Zhasconstr| Zhasnamer} hasjrelrg ZhaslocalZ
hascompareZcmp_opZhasfreeZhasnargsr
rj )codevarnamesrH constantscellsry rz labelsre freerh oprl rf r{ rm r r r rw s@
rw c
C sT | j | j } t t | } t | j | | j | j | j | | d | d S)Nr ) r[ rZ ru r r( r rY rX rV )r\ lastir rx ry r r r r : s rz c C s | d k } | r d n d }
x t | | | | | | d | D]h } | og | j d k og | j d k } | r} t d | | j | k }
t | j |
|
d | q@ Wd S)Nr_ r rz r )rw re rh r$ rq )r r r rH r r ry r rz Z show_linenorp ZinstrZnew_source_lineZis_current_instrr r r r( A s
r( c C s t t | d d | d S)Nz<dis>r )r r )r r r r r r* S s r* c c s d } t | } d } x | | k r | | } | } | d } d } | t k r | | | | d d | } d } | d } | t k r | d } | | | f Vq Wd S)Nr r8 r@ r9 i )lenZ
HAVE_ARGUMENTZ EXTENDED_ARG)r Z extended_argnrI r rh rl r r r r Y s
r c C s g } x t | D]u \ } } } | d k r d } | t k rQ | d | } n | t k rc | } | d k r | | k r | j | q W| S)Nr8 r_ r )r r ZhasjabsrB )r r rh r rl Zlabelr r r r
l s c c s t | j d d d } t | j d d d } d } | j } d } xT t | | D]C \ } } | r | | k r | | f V| } | | 7} | | 7} q] W| | k r | | f Vd S)Nr r9 r8 )list co_lnotabrv zip)r Zbyte_incrementsZline_incrementsZ
lastlinenolinenoaddrZ byte_incrZ line_incrr r r r s
c @ sj e Z d Z d d d d d d Z d d Z d d Z e d
d Z d d
Z d d Z d S)r rt Ncurrent_offsetc C s t | | _ } | d k r7 | j | _ d | _ n | | _ | | j | _ | j | j | _ t t | | _
| | _ | | _ d S)Nr )
rL codeobjrv rt _line_offsetr[ rZ _cell_namesru r _linestarts_original_objectr )ro r- rt r r\ r r r __init__ s zBytecode.__init__c C s= | j } t | j | j | j | j | j | j d | j S)Nrz ) r rw r rY rX rV r r r )ro r\ r r r __iter__ s
zBytecode.__iter__c C s d j | j j | j S)Nz{}({!r}))format __class__r, r )ro r r r __repr__ s zBytecode.__repr__c C s2 x | j r | j } q W| | j j d | j S)Nr )r3 r4 r5 r6 )clsr7 r r r from_traceback s
zBytecode.from_tracebackc C s
t | j S)N)rM r )ro r r r info s z
Bytecode.infoc C s | j } | j d k r$ | j } n d
} t j ` } t | j d | j d | j d | j d | j d | j
d | j d | d | | j SWd QRXd S) Nr8 r rH r r ry rz r r r )
r r ioStringIOr( r rY rX rV r r r getvalue)ro r\ rh outputr r r r s z Bytecode.dis)
r, rr rs r r r classmethodr r r r r r r r s
c C s d d l } | j } | j d d | j d d d d | j } | j } | j } Wd QRXt | | j j d } t | d S) Nr infiler+ nargs?default-r )
argparseArgumentParser add_argumentFileType
parse_argsr readr r r )r parserargsr r r r r r _test s % r __main__r r )(r/ types collectionsr Zopcoder Z _opcodes_all
MethodType FunctionTypeCodeTyper staticmethodr+ r# r r r rC rK rL r rM r
namedtupler^ r
r r| r} rw r r( r* r r r
r r r r, r r r r <module> sb
# 3 . =