3
\F @ 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 ZdJddddZdKddddZdddddddd d d d
Zd d! Zd"d# Zd$d Zd%d& Zddd'd Zejd(d)Zd*ej_ d+ej_ d,ej _ d-ej _ d.ej _ d/ej _ d0ej!_ d1ej"_ G d2d
d
eZ#dd3d4d Z$d5d6 Z%d7d8 Z&dLd9d:Z'dNdddd?Z)ddd@dAZ*e(Z+dBdC Z,dDd
Z-dEd Z.G dFd dZ/dGdH Z0e1dIkre0 dS )Q N)*)__all__ code_infodis disassembledistbdiscofindlinestarts
findlabels show_codeget_instructions InstructionBytecode FORMAT_VALUEc C s6 yt | |d}W n tk
r0 t | |d}Y nX |S )Nevalexec)compile SyntaxError)sourcenamec r /usr/lib64/python3.6/dis.py _try_compile s
r )filec C s> | d krt |d d S t| dr&| j} t| dr6| j} t| drF| j} t| drt| jj }x|D ]n\}}t|t rdt
d| |d yt ||d W n0 t k
r } zt
d||d W Y d d }~X nX t
|d qdW nbt| drt
| |d nJt| ttfrt| |d n,t| tr(t| |d nt d t| j d S )
N)r __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 sZ | d krBy
t j} W n tk
r. tdY nX x| jr@| j} q2W 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 E s
Z OPTIMIZEDZ NEWLOCALSZVARARGSZ VARKEYWORDSZNESTEDZ GENERATORZNOFREEZ COROUTINEZITERABLE_COROUTINEZASYNC_GENERATOR)
@ i c C sb g }xRt dD ]8}d|> }| |@ r|jtj|t| | |N } | sP qW |jt| dj|S )Nr> r9 z, )rangeappendCOMPILER_FLAG_NAMESgethexjoin)flagsnamesiflagr r r pretty_flags_ s rL c C sh t | dr| j} t | dr | j} t | dr0| j} t| trDt| d} t | drR| S tdt| j d S )Nr r r z
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_objectm s
rM c C s t t| S )N)_format_code_inforM )r. r r r r | s c C s g }|j d| j |j d| j |j d| j |j d| j |j d| j |j d| j |j dt| j | j r|j d x t
| j D ]}|j d | qW | j r|j d
x t
| j D ]}|j d | qW | j r|j d x t
| j D ]}|j d | qW | j
rF|j d
x"t
| j
D ]}|j d | q.W | jr||j d x"t
| jD ]}|j d | qdW dj|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:
)rC co_name co_filename co_argcountco_kwonlyargcount
co_nlocals co_stacksizerL co_flags co_consts enumerateco_names co_varnames co_freevars co_cellvarsrG )colinesZi_cZi_nr r r rN s:
rN c C s t t| |d d S )N)r )r% r )r] 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 Falsec @ s e Zd ZdddZdS )r
Fc C s g }|r:| j d k r,d| }|j|| j n|jd| |rJ|jd n
|jd | jrf|jd n
|jd |jt| jjd |j| jjd | jd k r|jt| jjd | j r|jd
| j d dj
|j S ) Nz%%%dd z-->z z>>z r; ()) starts_linerC is_jump_targetreproffsetrjustopnameljustargargreprrG rstrip)self lineno_widthZmark_as_currentZfieldsZ
lineno_fmtr r r _disassemble s&
zInstruction._disassembleN)r` F)r-
__module__ __qualname__rr 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 ) rM r\ r[ dictr co_firstlineno_get_instructions_bytesr rZ rY rW )r. ru r]
cell_names
linestarts line_offsetr r r r s c C s | }|d k r|| }|t |fS )N)rh )Z const_indexZ
const_listargvalr r r _get_const_info s r} c C s* | }|d k r|| }|}nt |}||fS )N)rh )Z
name_indexZ name_listr| rn r r r _get_name_info s r~ c
c sx t | }d }xdt| D ]V\} }
} |d k rH|j| 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|d
7 }|d 7 }tt|
|
| |
|| || V qW d S )
N r: zto r` r; r* rh asciir9 z, z with format)r r* rh r )r
_unpack_opargsrE Zhasconstr} Zhasnamer~ hasjrelrh ZhaslocalZ
hascompareZcmp_opZhasfreer r* r boolr
rk )codevarnamesrI constantscellsrz r{ labelsrf ri oprm rg r| rn r r r rx sH
rx r9 c
C s< | j | j }tt| }t| j|| j| j| j|||d d S )N)r ) r\ r[ rv r r) r rZ rY rW )r] lastir ry rz r r r r J s )r r{ c C s| |d k } | rdnd}
xbt | ||||||dD ]H} | oF| jd k oF| jdk} | rVt|d | j|k}
t| j|
|
|d q,W d S )Nr` r )r{ )r )rx rf ri r% rr )r r r rI r r rz r r{ Z show_linenorq ZinstrZnew_source_lineZis_current_instrr r r r) Q s
r) c C s t t| d|d d S )Nz)r )r r )r r r r r r+ c s r+ c c sh d}x^t dt| dD ]J}| | }|tkrP| |d |B }|tkrJ|d> nd}nd }|||fV qW d S )Nr r: r9 r< )rB lenZ
HAVE_ARGUMENTZ EXTENDED_ARG)r Z extended_argrJ r rm r r r r i s r c C s` g }xVt | D ]J\}}}|d k r|tkr6|d | }n|tkr|}nq||kr|j| qW |S )Nr: )r r ZhasjabsrC )r r ri r rm Zlabelr r r r
t 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: r9 r@ rA ) co_lnotabrw 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)ru current_offsetc C sd t | | _}|d kr&|j| _d| _n|| _||j | _|j|j | _tt || _
|| _ || _ d S )Nr )
rM codeobjrw ru _line_offsetr\ r[ _cell_namesrv r _linestarts_original_objectr )rp r. ru r r] r r r __init__ s zBytecode.__init__c C s* | j }t|j|j|j|j| j| j| jdS )N)r{ ) r rx r rZ rY rW r r r )rp r] r r r __iter__ s
zBytecode.__iter__c C s dj | jj| jS )Nz{}({!r}))format __class__r- r )rp r r r __repr__ s
zBytecode.__repr__c C s$ x|j r|j }qW | |jj|jdS )N)r )r4 r5 r6 r7 )clsr8 r r r from_traceback s
zBytecode.from_tracebackc C s
t | jS )N)rN r )rp r r r info s z
Bytecode.infoc C s` | j }| jd k r| j}nd}tj 2}t|j|j|j|j| j | j
| j ||d |j S Q R X d S )Nr9 )r rI r r rz r{ r r )
r r ioStringIOr) r rZ rY rW r r r getvalue)rp r] ri outputr r r r s
z Bytecode.dis)
r- rs rt r r r classmethodr r r r r r r r s
c C sd dd l } | j }|jd| j ddd |j }|j}|j }W d Q R X t||jjd}t | d S )Nr infile?-)r, 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 r )r NNNNN)2r0 types collectionsr Zopcoder Z _opcodes_all
MethodType FunctionTypeCodeTyper staticmethodr, r$ Zopmapr r r r rD rL rM r rN r
namedtupler_ rk __doc__rm r| rn ri rf rg r
r r} r~ rx r r) r+ r r r
r r r r- r r r r sv
# 3
2 =