Your IP : 3.144.98.43
3
��2a�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� de j�Zi Zejd �Zejd �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 � d e �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 j| �}|S )N)�marshal�loads)�ms�co� r �(/opt/alt/python36/lib64/python3.6/rpc.py�
unpickle_code- s
r c C s t j| �}t|ffS )N)r �dumpsr )r r r r r � pickle_code3 s
r
c C s$ t j� }t||�}|j| � |j� S )N)�io�BytesIO� CodePickler�dump�getvalue)�objZprotocol�f�pr r r r 9 s
r c @ s" e Zd ZejeiZejej� 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 }tjj| ||� 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| 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}tddd |d� td|d� tdtj� j |d� td||d� tdt|�|d� tj |d� td |d� tdd |d� t
j d
� Y nX d S ) N�
�-�( )�filezUnhandled server exception!z
Thread: %szClient Address: z Request: z#
*** Unrecoverable, server exiting!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' r9 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 j� | _|d k r|| _|| _|d kr*t}|| _i | _i | _d S )N) r0 r1 �
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 |j� d S )N)r= �close)r! r= r r r rB � s zSocketIO.closec C s t jd� d S )Nr )r6 r7 )r! r r r �exithook� s zSocketIO.exithookc G sR | j s
d S | jd ttj� j� }x |D ]}|d t|� }q(W t|tjd� d S )N� )r+ ) r< �location�strr0 r1 r2 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! rK r r r �
unregister� s zSocketIO.unregisterc C s� | j 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
j || ||ff� dS dd | fS W n� t k
�r, � Y n� t
k
�rB � Y nn tk
�rX � Y nX tk
�r| }
zd
|
fS d }
~
X n4 d}t||| |f tjd� tjtjd� dS d S )Nz
localcall:�ERROR�Bad request formatzUnknown object id: %r� __methods__�OK�__attributes__zUnsupported method name: %r�CALL�QUEUE�QUEUEDz Unsupported message type: %s�CALLEXCzU*** Internal Error: rpc.py:SocketIO.localcall()
Object: %s
Method: %s
Args: %s
)r+ � EXCEPTION)rP rQ )rW N)rY N)rJ � TypeErrorr? � _getmethods�_getattributes�hasattr�getattr�
isinstance� RemoteObject� remoteref�
request_queueZputr, �KeyboardInterrupt�OSError� Exceptionr/ r- r. r4 r5 )r! �seqr8 �howrK �
methodnamerG �kwargsr �methods�
attributes�method�retZex�msgr r r � localcall� sT
zSocketIO.localcallc C s( | j d||� | j||||�}| j|�S )Nzremotecall:asynccall: )rJ � asynccall� asyncreturn)r! rK rh rG ri rf r r r �
remotecall� s zSocketIO.remotecallc C s( | j d||� | j||||�}| j|�S )Nzremotequeue:asyncqueue: )rJ �
asyncqueuerq )r! rK rh rG ri rf r r r � remotequeue� s zSocketIO.remotequeuec C s` d||||ff}| j � }tj� | jkr8tj� }|| j|< | jd| ||||� | j||f� |S )NrU z
asynccall:%d:)�newseqr0 r1 r; � ConditionrA rJ �
putmessage)r! rK rh rG ri r8 rf �cvarr r r rp � s
zSocketIO.asynccallc C s` d||||ff}| j � }tj� | jkr8tj� }|| j|< | jd| ||||� | j||f� |S )NrV zasyncqueue:%d:)ru r0 r1 r; rv rA rJ rw )r! rK rh rG ri r8 rf rx r r r rs � s
zSocketIO.asyncqueuec C s6 | j d| � | j|dd�}| j d| |� | j|�S )Nz#asyncreturn:%d:call getresponse(): g�������?)�waitzasyncreturn:%d:response: )rJ � getresponse�decoderesponse)r! rf �responser r r rq � s zSocketIO.asyncreturnc C s� |\}}|dkr|S |dkr d S |dkr6| j d� d S |dkrT| j d� | j� d S |dkrp| j d|� t|��|d kr�| j d
|� |�t||��d S ) NrS rW rY zdecoderesponse: EXCEPTION�EOFzdecoderesponse: EOFrP z decoderesponse: Internal ERROR:rX z decoderesponse: Call Exception:)rJ �decode_interrupthook� RuntimeError� SystemError)r! r| rg �whatr r r r{ � s&
zSocketIO.decoderesponsec C s t �d S )N)�EOFError)r! r r r r~ s z SocketIO.decode_interrupthookc C s6 y| j d dd� W n tk
r0 | jd� d S X d S )Ng�������?)�myseqry zmainloop:return)rz r� rJ )r! r r r �mainloop s
zSocketIO.mainloopc C s6 | j ||�}|d k r2|\}}|dkr2|| j|�f}|S )NrS )� _getresponse�_proxify)r! r� ry r| rg r� r r r rz # s zSocketIO.getresponsec C s4 t |t�rt| |j�S t |t�r0tt| j|��S |S )N)r_ � RemoteProxy�RPCProxyrK �list�mapr� )r! r r r r r� + s
zSocketIO._proxifyc C s� | j d|� tj� | jkr:x�| j||�}|d k r |S q W nb| j| }|j� x|| jkrb|j� qNW | j| }| j d||f � | j|= | j|= |j � |S d S )Nz_getresponse:myseq:z-_getresponse:%s: thread woke up: response: %s)
rJ r0 r1 r; � pollresponserA �acquirer@ ry �release)r! r� ry r| rx r r r r� 3 s"
zSocketIO._getresponsec C s | j d | _ }|S )N� )�nextseq)r! rf r r r ru I s zSocketIO.newseqc C s� | j d|d � y t|�}W n, tjk
rJ tdt|�tjd� � Y nX tj dt
|��| }xnt
|�dkr�y0t j g | j gg �\}}}| j j
|d t� �}W n ttfk
r� td��Y nX ||d � }qbW d S )Nzputmessage:%d:r zCannot pickle:)r+ z<izsocket no longer exists)rJ r �pickleZ
PicklingErrorr/ r3 r- r. �struct�pack�len�selectr= �send�BUFSIZE�AttributeErrorrZ rd )r! �messagerH �r�w�x�nr r r rw M s zSocketIO.putmessage� � c C s� | j � t| j�| jk r�tj| jj� gg g |�\}}}t|�dkrFd S y| jjt�}W n t k
rn t
�Y nX t|�dkr�t
�| j|7 _| j � | j � S )Nr ) �_stage0r� �buff�bufneedr� r= �filenoZrecvr� rd r� �_stage1)r! ry r� r� r� rH r r r �
pollpacketa s
zSocketIO.pollpacketc C sR | j dkrNt| j�dkrN| jd d� }| jdd � | _tjd|�d | _d| _ d S )Nr r� z<i� )�bufstater� r� r� �unpackr� )r! rH r r r r� q s
zSocketIO._stage0c C sP | j dkrLt| j�| jkrL| jd | j� }| j| jd � | _d| _d| _ |S d S )Nr� r� r )r� r� r� r� )r! �packetr r r r� x s zSocketIO._stage1c C s� | j |�}|d krd S ytj|�}W nV tjk
rz tdtjd� tdt|�tjd� tj tjd� tdtjd� � Y nX |S )Nz-----------------------)r+ zcannot unpickle packet:)
r� r� r ZUnpicklingErrorr/ r- r. r3 r4 � print_stack)r! ry r� r� r r r � pollmessage� s
zSocketIO.pollmessagec
C s` �xXyt jd�}W n tjk
r( Y n X |\}}|d|ff}| j|� y| j|�}|d kr`d S W n. tk
r~ | j� d S tk
r� d S X |\}}|d }| j d||f � |dk�r| j d| � | j
||�}| j d||f � |dk�r| j||f� q|dkrqq||k�r |S | j j|d �} | d k r| j � || j
|< | j� | j� qqW d S ) Nr rS zpollresponse:%d:myseq:%srU rV z pollresponse:%d:localcall:call:z%pollresponse:%d:localcall:response:%s)rU rV )�response_queue�get�queueZEmptyrw r� r� �
handle_EOFr� rJ ro rA r� r@ �notifyr� )
r! r� ry Zqmsgrf r| r� Zresqrg �cvr r r r� � sL
zSocketIO.pollresponsec C sZ | j � | jd� x:| jD ]0}| j| }|j� d| j|< |j� |j� qW | j� d S )Nr� r} )r} N)�EOFhookrJ rA r� r@ r� r� rC )r! �keyr� r r r r� � s
zSocketIO.handle_EOFc C s d S )Nr )r! r r r r� � s zSocketIO.EOFhook)NN)"r r r r� r rB rC rJ rM rO ro rr rt rp rs rq r{ r~ r� rz r� r� ru rw r� r� r� r� r� r� r� r� r� r� r r r r r: � s>
.
I
r: 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 rK r r r ra � s ra c @ s e Zd Zdd� ZdS )r� c C s
|| _ d S )N)rK )r! rK 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( | |_ tj| |� tjj| |||� d S )N)Zcurrent_handlerr: r r �BaseRequestHandler)r! r= r" Zsvrr r r r s zRPCHandler.__init__c C s | j � d S )N)r� )r! r r r �handle s zRPCHandler.handlec C s
t | |�S )N)r� )r! rK r r r �get_remote_proxy s zRPCHandler.get_remote_proxyN)r r r r< rE 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 ||�| _| jj|� | jjd� d S )Nr� )r$ �listening_sockZbindZlisten)r! �addressZfamily�typer r r r s zRPCClient.__init__c C sV | j j� \}}| jr$td|tjd� |d tkr>tj| |� ntd|tjd� t �d S )Nz ****** Connection request from )r+ r z** Invalid host: )
r� �acceptr< r/ r- r. � LOCALHOSTr: r rd )r! Z working_sockr� r r r r� s zRPCClient.acceptc C s
t | |�S )N)r� )r! rK r r r r� # s zRPCClient.get_remote_proxyN) r r r r< rE 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)�sockiorK )r! r� rK r r r r , s zRPCProxy.__init__c C sp | j d kr| j� | j j|�r.t| j| j|�S | jd kr@| j� || jkrd| jj| jd|fi �}|S t |��d S )N�__getattribute__)
�_RPCProxy__methods�_RPCProxy__getmethodsr� � MethodProxyr� rK �_RPCProxy__attributes�_RPCProxy__getattributesrr r� )r! r2 �valuer r r � __getattr__0 s
zRPCProxy.__getattr__c C s | j j| jdf i �| _d S )NrT )r� rr rK r� )r! r r r Z__getattributes> s
zRPCProxy.__getattributesc C s | j j| jdf i �| _d S )NrR )r� rr rK 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__r[ )r rj r2 �attr�superr r r r[ F s
r[ c C s0 x*t | �D ] }t| |�}t|�s
d||< q
W d S )Nr� )r� r^ r� )r rk r2 r� r r r r\ Q s
r\ c @ s e Zd Zdd� Zdd� ZdS )r� c C s || _ || _|| _d S )N)r� rK r2 )r! r� rK r2 r r r r Z s zMethodProxy.__init__c O s | j j| j| j||�}|S )N)r� rr rK r2 )r! rG ri 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| �}ytjj|� W n< tk
rf d}|j|d�}|j|d�}tjj|� Y nX tjjd� | t _d S )N�ascii�backslashreplace�strictr( ) �builtins�_r3 r- �stdout�write�UnicodeEncodeError�encode�decode)r� �text�encoding�bytesr r r � displayhookg s r� �__main__)�mainzidlelib.idle_test.test_rpcr� )� verbosity)Ni ),r� r r r r6 r� r� r� r$ r r� r- r0 r4 r r r
r ZPicklerr
r� r� r r r> ZQueuerb r� rL r: r` ra r� r� r r� r� r[ r\ r� r� r Zunittestr� r r r r �<module> sP
2
k