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� ZdUddd�dd�ZdVdd�dd�Zd d d d d d!d"d#d$d%d&�
Zd'd(� Zd)d*� Zd+d� Z d,d-� Z dd�d.d �Z e� d/d0�Z 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.dWdBdC�Z/dXdd�dEd�Z0ddd�dFdG�Z1dYdd dH�dIdJ�Z2dKdL� Z3e0Z4dMdN� Z5dOd
� Z6dPd � Z7G dQd� d�Z8dRdS� Z9e:dTk�re9� dS )Z� N)�*)�__all__� code_info�dis� disassemble�distb�disco�findlinestarts�
findlabels� show_code�get_instructions� Instruction�Bytecode� FORMAT_VALUE)N� �str�repr�ascii�
MAKE_FUNCTION)�defaultsZ
kwdefaultsZ annotationsZclosurec C s6 zt | |d�}W n tk
r0 t | |d�}Y nX |S )N�eval�exec)�compile� SyntaxError)�source�name�c� r �/usr/lib64/python3.8/dis.py� _try_compile s
r ��file�depthc C sh | d krt |d� d S t| d�r&| j} t| d�r8| j} n4t| d�rJ| j} n"t| d�r\| j} nt| d�rl| j} t| d�r�t| j� � �}|D ]p\}}t
|t �r�t d| |d� zt
|||d � W n0 tk
r� } zt d
||d� W 5 d }~X Y nX t |d� q�nht| d ��rt| ||d � nLt
| ttf��r6t| |d� n.t
| t��rRt| ||d � ntd t| �j ��d S )
N�r! �__func__�__code__�gi_code�ag_code�cr_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_code�printr � TypeError�_disassemble_recursive�bytes� bytearray�_disassemble_bytesr �_disassemble_str�type�__name__)�xr! r" r. r Zx1�msgr r r r + s@
�r# c C sX | d kr@z
t j} W n tk
r0 td�d �Y nX | jr@| j} q2t| jj| j|d� d S )Nz no last traceback to disassembler# ) �sys�last_traceback�AttributeError� RuntimeError�tb_nextr �tb_frame�f_code�tb_lasti)�tbr! r r r r X s
Z OPTIMIZEDZ NEWLOCALSZVARARGSZ VARKEYWORDSZNESTEDZ GENERATORZNOFREEZ COROUTINEZITERABLE_COROUTINEZASYNC_GENERATOR)
� � � � � � �@ � � i c C s` g }t d�D ]:}d|> }| |@ r |�t�|t|��� | |N } | s qVq |�t| �� d�|�S )NrJ rE �, )�range�append�COMPILER_FLAG_NAMES�get�hex�join)�flags�names�i�flagr r r � pretty_flagsr s rY c C s� t | d�r| j} t | d�r"| j} n4t | d�r4| j} n"t | d�rF| j} nt | d�rV| j} t| t�rjt| d�} t | d�rx| 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 | j�rH|� d� t | j�D ]}|� d | � �q2| j�rz|� d� t | j�D ]}|� 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_argcount�co_posonlyargcount�co_kwonlyargcount�
co_nlocals� co_stacksizerY �co_flags� co_consts� enumerate�co_names� co_varnames� co_freevars� co_cellvarsrT )�co�linesZi_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 Zddd�ZdS )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��|�� |�| j�t�� | 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 �offset�rjust�opname�ljust�
_OPNAME_WIDTH�arg� _OPARG_WIDTH�argreprrT �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_list�argvalr 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�|
tkr�t� |�\} }
n�|
tkr�| d � } dt| � }
n�|
t kr�t� |�\} }
n�|
t
kr�t � } | }
n�|
t kr�t� |�\} }
nr|
t
k�rFt� d@ \} }
| t� d@ �f} | d �rl|
�r<|
d7 }
|
d7 }
n&|
tk�rld�� fd d
�tt�D ��}
tt|
|
� | |
| || �V qd S ) Nr rF zto rs rG rE rN z with formatc 3 s" | ]\}}� d |> @ r|V qdS )rE Nr )�.0rW �s�r~ r r � g s �z*_get_instructions_bytes..)r
�_unpack_opargsrR Zhasconstr� Zhasnamer� �hasjrelr ZhaslocalZ
hascompareZcmp_opZhasfreer �FORMAT_VALUE_CONVERTERS�boolr rT rh �MAKE_FUNCTION_FLAGSr
r{ )�code�varnamesrV � constants�cellsr� 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|d�r2t|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| �d�D ]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
lastlineno�linenoZaddrZ byte_incrZ line_incrr r r r � s&
c @ sH e Zd Zddd�dd�Zdd� 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| jd�S )Nr� ) r� r� r* rj ri rg r� r� r� )r� rm r r r �__iter__� s �zBytecode.__iter__c C s d� | jj| j�S )Nz{}({!r}))�format� __class__r9 r� �r� r r r �__repr__� s
�zBytecode.__repr__c C s |j r|j }q | |jj|jd�S )N)r� )r@ rA rB rC )�clsrD r r r �from_traceback� s zBytecode.from_tracebackc C s
t | j�S 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� �io�StringIOr6 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| �d�ddd� |�� }|j�}|�� }W 5 Q R X t||jjd�}t |� d S )Nr �infile�rb�?�-)r8 �nargs�defaultr )
�argparse�ArgumentParser� add_argumentZFileType�
parse_argsr� �readr r r )r� �parser�argsr� r r� r r r �_test s r� �__main__)N)N)NNNNNr )r� )r� NNNNN);r< �types� collectionsr� Zopcoder Z _opcodes_all�
MethodType� FunctionType�CodeTyper� � 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 � � =