B
YI~d�R � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZdd� Zdd� Zd)dd�ZG dd � d ej�Zd
Zd ZG d d
� d
e j�Zi Ze�d �Ze�d �ZG 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� de �Z$d d � Z%d d � Z&G d d!� d!e �Z'd"d#� Z(e)d$k�r�d d%l*m+Z+ e+d&d'd(� dS )*� Nc C s t �| �}|S )N)�marshal�loads)�ms�co� r �0/opt/alt/python37/lib64/python3.7/idlelib/rpc.py�
unpickle_code- s
r c C s t �| �}t|ffS )N)r �dumpsr )r r r r r � pickle_code3 s
r
c C s$ t �� }t||�}|�| � |�� S )N)�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__�types�CodeTyper
�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 d S )Nr )r r r r � server_bindQ s zRPCServer.server_bindc C s | j �| j� d S )N)�socketZconnect�server_address)r r r r �server_activateU s zRPCServer.server_activatec C s | j | jfS )N)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 )
Nz)
----------------------------------------)�filezUnhandled server exception!z
Thread: %szClient Address: z Request: z#
*** Unrecoverable, server exiting!z(----------------------------------------r ) �
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& r5 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 r9 r; r8 r r r r � s
zSocketIO.__init__c C s | j }d | _ |d k r |�� d S )N)r9 �close)r r9 r r r r> � s zSocketIO.closec C s t �d� d S )Nr )r2 r3 )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' ) r8 �location�strr, r- r. 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 rG 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 formatrL 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)rF � TypeErrorr; � _getmethods�_getattributes�hasattr�getattr�
isinstance� RemoteObject� remoteref�
request_queueZputr( �KeyboardInterrupt�OSError� Exceptionr+ r) r* r0 r1 )r �seqr4 �howrG �
methodnamerC �kwargsr �methods�
attributes�method�retZex�msgr r r � localcall� sT
zSocketIO.localcallc C s( | � d||� | �||||�}| �|�S )Nzremotecall:asynccall: )rF � asynccall� asyncreturn)r rG rc rC rd ra r r r �
remotecall� s zSocketIO.remotecallc C s( | � d||� | �||||�}| �|�S )Nzremotequeue:asyncqueue: )rF �
asyncqueuerl )r rG rc rC rd ra r r r � remotequeue� s zSocketIO.remotequeuec C s` d||||ff}| � � }t�� | jkr8t�� }|| j|< | �d| ||||� | �||f� |S )NrP z
asynccall:%d:)�newseqr, r- r7 � Conditionr= rF �
putmessage)r rG rc rC rd r4 ra �cvarr r r rk � s
zSocketIO.asynccallc C s` d||||ff}| � � }t�� | jkr8t�� }|| j|< | �d| ||||� | �||f� |S )NrQ zasyncqueue:%d:)rp r, r- r7 rq r= rF rr )r rG rc rC rd r4 ra rs r r r rn � s
zSocketIO.asyncqueuec C s6 | � d| � | j|dd�}| � d| |� | �|�S )Nz#asyncreturn:%d:call getresponse(): g�������?)�waitzasyncreturn:%d:response: )rF � getresponse�decoderesponse)r ra �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 ) NrN rR rT zdecoderesponse: EXCEPTION�EOFzdecoderesponse: EOFrL z decoderesponse: Internal ERROR:rS z decoderesponse: Call Exception:)rF �decode_interrupthook� RuntimeError� SystemError)r rw rb �whatr r r rv � s&
zSocketIO.decoderesponsec C s t �d S )N)�EOFError)r r r r ry s z SocketIO.decode_interrupthookc C s6 y| j d dd� W n tk
r0 | �d� d S X d S )Ng�������?)�myseqrt zmainloop:return)ru r} rF )r r r r �mainloop s
zSocketIO.mainloopc C s6 | � ||�}|d k r2|\}}|dkr2|| �|�f}|S )NrN )� _getresponse�_proxify)r r~ rt rw rb r| r r r ru # s zSocketIO.getresponsec C s4 t |t�rt| |j�S t |t�r0tt| j|��S |S )N)rZ � RemoteProxy�RPCProxyrG �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)
rF r, r- r7 � 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 ra 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 r? rF rI rK 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 r6 � s>
.
I
r6 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 rG r r r r\ � s r\ c @ s e Zd Zdd� ZdS )r� c C s
|| _ d S )N)rG )r rG 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_handlerr6 r r �BaseRequestHandler)r r9 r! Zsvrr r r r s zRPCHandler.__init__c C s | � � d S )N)r )r r r r �handle s zRPCHandler.handlec C s
t | |�S )N)r� )r rG r r r �get_remote_proxy s zRPCHandler.get_remote_proxyN)r r r r8 rA 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� �acceptr8 r+ r) r* � LOCALHOSTr6 r r_ )r Z working_sockr� r r r r� s zRPCClient.acceptc C s
t | |�S )N)r� )r rG r r r r� # s zRPCClient.get_remote_proxyN) r r r r8 rA 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)�sockiorG )r r� rG 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� rG �_RPCProxy__attributes�_RPCProxy__getattributesrm r� )r r. �valuer r r � __getattr__0 s
zRPCProxy.__getattr__c C s | j �| jddi �| _d S )NrO r )r� rm rG r� )r r r r Z__getattributes> s
zRPCProxy.__getattributesc C s | j �| jddi �| _d S )NrM r )r� rm rG 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� )�dirrY �callablerZ r� � __bases__rV )r re r. �attr�superr r r rV F s
rV c C s0 x*t | �D ] }t| |�}t|�s
d||< q
W d S )Nr� )r� rY r� )r rf r. r� r r r rW Q s
rW c @ s e Zd Zdd� Zdd� ZdS )r� c C s || _ || _|| _d S )N)r� rG r. )r r� rG r. r r r r Z s zMethodProxy.__init__c O s | j �| j| j||�}|S )N)r� rm rG r. )r rC rd 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~ | d kr d S d t _t| �}ytj�|� W n< tk
rf d}|�|d�}|�|d�}tj�|� Y nX tj�d� | t _d S )N�ascii�backslashreplace�strict�
) �builtins�_r/ r) �stdout�write�UnicodeEncodeError�encode�decode)r� �text�encoding�bytesr r r � displayhookg s r� �__main__)�mainzidlelib.idle_test.test_rpcr� )� verbosity)N),r� r r r r2 r� r� r� r# r r� r) r, r0 r r r
r ZPicklerr
r� r� r r r: ZQueuer] r� rH r6 r[ r\ r� r� r r� r� rV rW r� r� r Zunittestr� r r r r � sP
2
k