B
� f�R � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZdd� Zdd� Zd*dd�ZG d d
� d
ej�Zd Zd ZG d
d� de
j�Zi Ze�d�Ze�d�Z G dd� de �Z G dd� de �Z dd� Z G dd� de �Z!G dd� de
j"e �Z#G dd� de �Z$G dd � d e �Z%d d � Z&d d � Z'G d!d"� d"e �Z(d#d$� Z)e*d%k�r�dd&l+m,Z, e,d'd(d)� dS )+a RPC Implementation, originally written for the Python Idle IDE
For security reasons, GvR requested that Idle's Python execution server process
connect to the Idle process, which listens for the connection. Since Idle has
only one client per server, this was not a limitation.
+---------------------------------+ +-------------+
| socketserver.BaseRequestHandler | | SocketIO |
+---------------------------------+ +-------------+
^ | register() |
| | unregister()|
| +-------------+
| ^ ^
| | |
| + -------------------+ |
| | |
+-------------------------+ +-----------------+
| RPCHandler | | RPCClient |
| [attribute of RPCServer]| | |
+-------------------------+ +-----------------+
The RPCServer handler class is expected to provide register/unregister methods.
RPCHandler inherits the mix-in class SocketIO, which provides these methods.
See the Idle run.main() docstring for further information on how this was
accomplished in Idle.
� Nc C s t �| �}t|tj�st�|S )z*Return code object from marshal string ms.)�marshal�loads�
isinstance�types�CodeType�AssertionError)�ms�co� r
�0/opt/alt/python37/lib64/python3.7/idlelib/rpc.py�
unpickle_code- s
r c C s$ t | tj�st�t�| �}t|ffS )zBReturn unpickle function and tuple with marshalled co code object.)r r r r r �dumpsr )r r r
r
r � pickle_code3 s
r c C s$ t �� }t||�}|�| � |�� S )z.Return pickled (or marshalled) string for obj.)�io�BytesIO� CodePickler�dump�getvalue)�objZprotocol�f�pr
r
r r
9 s
r
c @ s" e Zd ZejeiZe�ej� dS )r N) �__name__�
__module__� __qualname__r r r �dispatch_table�update�copyregr
r
r
r r B s
r i z 127.0.0.1c @ s6 e Zd Zd dd�Zdd� Zdd� Zdd � Zd
d � ZdS )
� RPCServerNc C s |d kr t }tj�| ||� d S )N)�
RPCHandler� socketserver� TCPServer�__init__)�self�addrZ handlerclassr
r
r r! L s zRPCServer.__init__c C s dS )z@Override TCPServer method, no bind() phase for connecting entityNr
)r" r
r
r � server_bindQ s zRPCServer.server_bindc C s | j �| j� dS )z�Override TCPServer method, connect() instead of listen()
Due to the reversed connection, self.server_address is actually the
address of the Idle Client to which we are connecting.
N)�socketZconnect�server_address)r" r
r
r �server_activateU s zRPCServer.server_activatec C s | j | jfS )z:Override TCPServer method, return already connected socket)r% r& )r" r
r
r � get_request^ s zRPCServer.get_requestc C s� y� W n� t k
r � Y n� tj}td|d� td|d� tdt�� j |d� td||d� tdt|�|d� tj |d� td|d� td|d� t
� d � Y nX d
S ) z�Override TCPServer method
Error message goes to __stderr__. No error message if exiting
normally or socket raised EOF. Other exceptions not handled in
server code will cause os._exit.
z)
----------------------------------------)�filezUnhandled server exception!z
Thread: %szClient Address: z Request: z#
*** Unrecoverable, server exiting!z(----------------------------------------r N) �
SystemExit�sys�
__stderr__�print� threading�current_thread�name�repr� traceback� print_exc�os�_exit)r" �requestZclient_addressZerfr
r
r � handle_errorb s zRPCServer.handle_error)N)r r r r! r$ r' r( r7 r
r
r
r r J s
r c @ s� e Zd ZdZd;dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+Zd,ZdZd-d.� Zd/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z!dS )<�SocketIOr Nc C s@ t �� | _|d k r|| _|| _|d kr*t}|| _i | _i | _d S )N) r. r/ �
sockthread� debugging�sock� objecttable�objtable� responses�cvars)r" r; r= r: r
r
r r! � s
zSocketIO.__init__c C s | j }d | _ |d k r |�� d S )N)r; �close)r" r; r
r
r r@ � s zSocketIO.closec C s t �d� dS )z!override for specific exit actionr N)r4 r5 )r" r
r
r �exithook� s zSocketIO.exithookc G sR | j s
d S | jd tt�� j� }x |D ]}|d t|� }q(W t|tjd� d S )N� )r) ) r: �location�strr. r/ r0 r- r+ r, )r" �args�s�ar
r
r �debug� s
zSocketIO.debugc C s || j |< d S )N)r= )r" �oid�objectr
r
r �register� s zSocketIO.registerc C s& y | j |= W n tk
r Y nX d S )N)r= �KeyError)r" rI r
r
r �
unregister� s zSocketIO.unregisterc
C s� | � d|� y|\}\}}}}W n tk
r4 dS X || jkrNdd|f fS | j| }|dkrvi } t|| � d| fS |dkr�i }
t||
� d|
fS t||�s�dd|f fS t||�} y^|d kr�| ||�} t| t�r�t | �} d| fS |d
k�rt
� || ||ff� d S dd | fS W n� t k
�r, � Y n� t
k
�rB � Y np tk
�rX � Y nZ tk
�r~ }
zd
|
fS d }
~
X Y n4 d}t||| |f tjd� tjtjd� dS d S )Nz
localcall:)�ERRORzBad request formatrN zUnknown object id: %r� __methods__�OK�__attributes__zUnsupported method name: %r�CALL�QUEUE)�QUEUEDNz Unsupported message type: %s�CALLEXCzU*** Internal Error: rpc.py:SocketIO.localcall()
Object: %s
Method: %s
Args: %s
)r) )� EXCEPTIONN)rH � TypeErrorr= � _getmethods�_getattributes�hasattr�getattrr � RemoteObject� remoteref�
request_queueZputr* �KeyboardInterrupt�OSError� Exceptionr- r+ r, r2 r3 )r" �seqr6 �howrI �
methodnamerE �kwargsr �methods�
attributes�methodZretZex�msgr
r
r � localcall� sT
zSocketIO.localcallc C s( | � d||� | �||||�}| �|�S )Nzremotecall:asynccall: )rH � asynccall� asyncreturn)r" rI rd rE re rb r
r
r �
remotecall� s zSocketIO.remotecallc C s( | � d||� | �||||�}| �|�S )Nzremotequeue:asyncqueue: )rH �
asyncqueuerl )r" rI rd rE re rb r
r
r � remotequeue� s zSocketIO.remotequeuec C s` d||||ff}| � � }t�� | jkr8t�� }|| j|< | �d| ||||� | �||f� |S )NrR z
asynccall:%d:)�newseqr. r/ r9 � Conditionr? rH �
putmessage)r" rI rd rE re r6 rb �cvarr
r
r rk � s
zSocketIO.asynccallc C s` d||||ff}| � � }t�� | jkr8t�� }|| j|< | �d| ||||� | �||f� |S )NrS zasyncqueue:%d:)rp r. r/ r9 rq r? rH rr )r" rI rd rE re r6 rb rs r
r
r rn � s
zSocketIO.asyncqueuec C s6 | � d| � | j|dd�}| � d| |� | �|�S )Nz#asyncreturn:%d:call getresponse(): g�������?)�waitzasyncreturn:%d:response: )rH � getresponse�decoderesponse)r" rb �responser
r
r rl � s zSocketIO.asyncreturnc C s� |\}}|dkr|S |dkr d S |dkr6| � d� d S |dkrT| � d� | �� d S |dkrp| � d|� t|��|d kr�| � d
|� |�t||��d S ) NrP rT rV zdecoderesponse: EXCEPTION�EOFzdecoderesponse: EOFrN z decoderesponse: Internal ERROR:rU z decoderesponse: Call Exception:)rH �decode_interrupthook� RuntimeError� SystemError)r" rw rc �whatr
r
r rv � s&
zSocketIO.decoderesponsec C s t �dS )� N)�EOFError)r" r
r
r ry s z SocketIO.decode_interrupthookc C s6 y| j ddd� W n tk
r0 | �d� dS X dS )z�Listen on socket until I/O not ready or EOF
pollresponse() will loop looking for seq number None, which
never comes, and exit on EOFError.
Ng�������?)�myseqrt zmainloop:return)ru r~ rH )r" r
r
r �mainloop s
zSocketIO.mainloopc C s6 | � ||�}|d k r2|\}}|dkr2|| �|�f}|S )NrP )� _getresponse�_proxify)r" r rt rw rc r| r
r
r ru # s zSocketIO.getresponsec C s4 t |t�rt| |j�S t |t�r0tt| j|��S |S )N)r � RemoteProxy�RPCProxyrI �list�mapr� )r" r r
r
r r� + s
zSocketIO._proxifyc C s� | � d|� t�� | jkr:x�| �||�}|d k r |S q W nb| j| }|�� x|| jkrb|�� qNW | j| }| � d||f � | j|= | j|= |� � |S d S )Nz_getresponse:myseq:z-_getresponse:%s: thread woke up: response: %s)
rH r. r/ r9 � pollresponser? �acquirer> rt �release)r" r rt rw rs r
r
r r� 3 s"
zSocketIO._getresponsec C s | j d | _ }|S )N� )�nextseq)r" rb r
r
r rp I s zSocketIO.newseqc C s� | � d|d � y t|�}W n, tjk
rJ tdt|�tjd� � Y nX t� dt
|��| }xnt
|�dkr�y0t � g | j gg �\}}}| j �
|d t� �}W n ttfk
r� td��Y nX ||d � }qbW d S )Nzputmessage:%d:r zCannot pickle:)r) z �notifyr� )
r" r rt Zqmsgrb rw r� Zresqrc �cvr
r
r r� � sL
zSocketIO.pollresponsec C sZ | � � | �d� x:| jD ]0}| j| }|�� d| j|< |�� |�� qW | �� dS )z+action taken upon link being closed by peerr� )rx NN)�EOFhookrH r? r� r> r� r� rA )r" �keyr� r
r
r r� � s
zSocketIO.handle_EOFc C s dS )zBClasses using rpc client/server can override to augment EOF actionNr
)r" r
r
r r� � s zSocketIO.EOFhook)NN)"r r r r� r! r@ rA rH rK rM rj rm ro rk rn rl rv ry r� ru r� r� rp rr r� r� r� r� r� r� r� r� r� r� r
r
r
r r8 � s>
.
I
r8 c @ s e Zd ZdS )r\ N)r r r r
r
r
r r\ � s r\ c C s t | �}| t|< t|�S )N)�idr< r� )r rI r
r
r r] � s r] c @ s e Zd Zdd� ZdS )r� c C s
|| _ d S )N)rI )r" rI r
r
r r! � s zRemoteProxy.__init__N)r r r r! r
r
r
r r� � s r� c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S )
r Fz#Sc C s( | |_ t�| |� tj�| |||� d S )N)Zcurrent_handlerr8 r! r �BaseRequestHandler)r" r; r# Zsvrr
r
r r! s zRPCHandler.__init__c C s | � � dS )z(handle() method required by socketserverN)r� )r" r
r
r �handle s zRPCHandler.handlec C s
t | |�S )N)r� )r" rI r
r
r �get_remote_proxy s zRPCHandler.get_remote_proxyN)r r r r: rC r! r� r� r
r
r
r r � s
r c @ s: e Zd ZdZdZdZejejfdd�Z dd� Z
dd � Z d
S ) � RPCClientFz#Cr� c C s* t � ||�| _| j�|� | j�d� d S )Nr� )r% �listening_sockZbindZlisten)r" �addressZfamily�typer
r
r r! s zRPCClient.__init__c C sV | j �� \}}| jr$td|tjd� |d tkr>t�| |� ntd|tjd� t �d S )Nz ****** Connection request from )r) r z** Invalid host: )
r� �acceptr: r- r+ r, � LOCALHOSTr8 r! r` )r" Z working_sockr� r
r
r r� s zRPCClient.acceptc C s
t | |�S )N)r� )r" rI r
r
r r� # s zRPCClient.get_remote_proxyN) r r r r: rC r� r% ZAF_INETZ SOCK_STREAMr! r� r� r
r
r
r r�
s
r� c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zdd � ZdS )
r� Nc C s || _ || _d S )N)�sockiorI )r" r� rI r
r
r r! , s zRPCProxy.__init__c C sp | j d kr| �� | j �|�r.t| j| j|�S | jd kr@| �� || jkrd| j�| jd|fi �}|S t |��d S )N�__getattribute__)
�_RPCProxy__methods�_RPCProxy__getmethodsr� � MethodProxyr� rI �_RPCProxy__attributes�_RPCProxy__getattributesrm r� )r" r0 �valuer
r
r � __getattr__0 s
zRPCProxy.__getattr__c C s | j �| jddi �| _d S )NrQ r
)r� rm rI r� )r" r
r
r Z__getattributes> s
zRPCProxy.__getattributesc C s | j �| jddi �| _d S )NrO r
)r� rm rI r� )r" r
r
r Z __getmethodsB s
zRPCProxy.__getmethods) r r r r� r� r! r� r� r� r
r
r
r r� ' s r� c C sT x*t | �D ] }t| |�}t|�r
d||< q
W t| t�rPx| jD ]}t||� q>W d S )Nr� )�dirr[ �callabler r� � __bases__rX )r rf r0 �attr�superr
r
r rX F s
rX c C s0 x*t | �D ] }t| |�}t|�s
d||< q
W d S )Nr� )r� r[ r� )r rg r0 r� r
r
r rY Q s
rY c @ s e Zd Zdd� Zdd� ZdS )r� c C s || _ || _|| _d S )N)r� rI r0 )r" r� rI r0 r
r
r r! Z s zMethodProxy.__init__c O s | j �| j| j||�}|S )N)r� rm rI r0 )r" rE re r� r
r
r �__call___ s zMethodProxy.__call__N)r r r r! r� r
r
r
r r� X s r� c C s~ | dkr dS dt _t| �}ytj�|� W n< tk
rf d}|�|d�}|�|d�}tj�|� Y nX tj�d� | t _dS )z9Override standard display hook to use non-locale encodingN�ascii�backslashreplace�strict�
) �builtins�_r1 r+ �stdout�write�UnicodeEncodeError�encode�decode)r� �text�encoding�bytesr
r
r � displayhookg s r� �__main__)�mainzidlelib.idle_test.test_rpcr� )� verbosity)N)-�__doc__r� r r r r4 r� r� r� r% r r� r+ r. r2 r r r r
ZPicklerr r� r� r r r< ZQueuer^ r� rJ r8 r\ r] r� r� r r� r� rX rY r� r� r Zunittestr� r
r
r
r � sR
2
k