3
2aR @ 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*ddZG dd d ejZd+Zd ZG d
d de jZi 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%krd d&l*m+Z+ e+d'd(d) dS ), Nc C s t j| }|S )N)marshalloads)msco 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)ioBytesIO CodePicklerdumpgetvalue)objZprotocolfpr r r r 9 s
r c @ s" e Zd ZejeiZejej dS )r
N) __name__
__module__ __qualname__typesCodeTyper
dispatch_tableupdatecopyregr r r r r
B s
r
i z 127.0.0.1c @ s6 e Zd Zd ddZdd Zdd Zdd Zd
d ZdS )
RPCServerNc C s |d kr t }tjj| || d S )N)
RPCHandler socketserver TCPServer__init__)selfaddrZ 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)socketZconnectserver_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 )
SystemExitsys
__stderr__print threadingcurrent_threadnamerepr traceback print_excos_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;ddZdd 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 debuggingsock objecttableobjtable responsescvars)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< locationstrr0 r1 r2 r/ r- r. )r! argssar r r debug s
zSocketIO.debugc C s || j |< d S )N)r? )r! oidobjectr 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 |dkri }
t||
d|
fS t||sdd|f fS t||} y^|d kr| ||} t| trt | } d| fS |d
krt
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:ERRORBad request formatzUnknown object id: %r __methods__OK__attributes__zUnsupported method name: %rCALLQUEUEQUEUEDz Unsupported message type: %sCALLEXCzU*** Internal Error: rpc.py:SocketIO.localcall()
Object: %s
Method: %s
Args: %s
)r+ EXCEPTION)rP rQ )rW N)rY N)rJ TypeErrorr? _getmethods_getattributeshasattrgetattr
isinstance RemoteObject remoteref
request_queueZputr, KeyboardInterruptOSError Exceptionr/ r- r. r4 r5 )r! seqr8 howrK
methodnamerG kwargsr methods
attributesmethodretZexmsgr 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 getresponsedecoderesponse)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: EXCEPTIONEOFzdecoderesponse: 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 |trt| |jS t |tr0tt| j|S |S )N)r_ RemoteProxyRPCProxyrK listmapr )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
|dkry0t j g | j gg \}}}| j j
|d t }W n ttfk
r tdY nX ||d }qbW d S )Nzputmessage:%d:r zCannot pickle:)r+ z
.
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 ZdZdZdZejejfddZ dd Z
dd Z d
S ) RPCClientFz#Cr c C s* t j ||| _| jj| | jjd d S )Nr )r$ listening_sockZbindZlisten)r! addressZfamilytyper 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| trPx| jD ]}t|| q>W d S )Nr )dirr^ callabler_ r __bases__r[ )r rj r2 attrsuperr 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 )Nasciibackslashreplacestrictr( ) builtins_r3 r- stdoutwriteUnicodeEncodeErrorencodedecode)r textencodingbytesr 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 sP
2
k