3
fR @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZdd Zdd Zd+ddZG d d
d
ejZd,Zd
ZG dd de
jZi ZejdZejdZ 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 d d d e Z%d d Z&d d! Z'G d"d# d#e Z(d$d% Z)e*d&krdd'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 j| }t|tjst|S )z*Return code object from marshal string ms.)marshalloads
isinstancetypesCodeTypeAssertionError)msco r
0/opt/alt/python36/lib64/python3.6/idlelib/rpc.py
unpickle_code- s
r c C s$ t | tjsttj| }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 j }t||}|j| |j S )z.Return pickled (or marshalled) string for obj.)ioBytesIO CodePicklerdumpgetvalue)objZprotocolfpr
r
r r
9 s
r
c @ s" e Zd ZejeiZejej dS )r N) __name__
__module__ __qualname__r r r 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 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| j dS )zOverride 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)socketZconnectserver_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}tddd |d td|d tdtj j |d td||d tdt||d tj |d td |d tdd |d t
j d
Y nX d S ) zOverride 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.
-( )filezUnhandled server exception!z
Thread: %szClient Address: z Request: z#
*** Unrecoverable, server exiting!r N)
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) r; 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) r2 r3
sockthread debuggingsock objecttableobjtable responsescvars)r# r? rA 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 rD s zSocketIO.closec C s t jd dS )z!override for specific exit actionr N)r8 r9 )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> locationstrr2 r3 r4 r1 r/ r0 )r# argssar
r
r debug s
zSocketIO.debugc C s || j |< d S )N)rA )r# oidobjectr
r
r register s zSocketIO.registerc C s& y | j |= W n tk
r Y nX d S )N)rA KeyError)r# rM 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)rR rS )rY N)r[ N)rL TypeErrorrA _getmethods_getattributeshasattrgetattrr RemoteObject remoteref
request_queueZputr. KeyboardInterruptOSError Exceptionr1 r/ r0 r6 r7 )r# seqr: howrM
methodnamerI kwargsr methods
attributesmethodZretZexmsgr
r
r localcall sT
zSocketIO.localcallc C s( | j d|| | j||||}| j|S )Nzremotecall:asynccall: )rL asynccall asyncreturn)r# rM ri rI rj rg r
r
r
remotecall s zSocketIO.remotecallc C s( | j d|| | j||||}| j|S )Nzremotequeue:asyncqueue: )rL
asyncqueuerq )r# rM ri rI rj rg r
r
r remotequeue s zSocketIO.remotequeuec C s` d||||ff}| j }tj | jkr8tj }|| j|< | jd| |||| | j||f |S )NrW z
asynccall:%d:)newseqr2 r3 r= ConditionrC rL
putmessage)r# rM ri rI rj r: rg cvarr
r
r rp s
zSocketIO.asynccallc C s` d||||ff}| j }tj | jkr8tj }|| j|< | jd| |||| | j||f |S )NrX zasyncqueue:%d:)ru r2 r3 r= rv rC rL rw )r# rM ri rI rj r: rg 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: )rL getresponsedecoderesponse)r# rg 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 ) NrU rY r[ zdecoderesponse: EXCEPTIONEOFzdecoderesponse: EOFrR z decoderesponse: Internal ERROR:rZ z decoderesponse: Call Exception:)rL decode_interrupthook RuntimeError SystemError)r# r| rh whatr
r
r r{ s&
zSocketIO.decoderesponsec C s t dS ) N)EOFError)r# r
r
r r~ s z SocketIO.decode_interrupthookc C s6 y| j ddd W n tk
r0 | jd dS X dS )zListen 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?)myseqry zmainloop:return)rz r rL )r# r
r
r mainloop s
zSocketIO.mainloopc C s6 | j ||}|d k r2|\}}|dkr2|| j|f}|S )NrU ) _getresponse_proxify)r# r ry r| rh r r
r
r rz # s zSocketIO.getresponsec C s4 t |trt| |jS t |tr0tt| j|S |S )N)r RemoteProxyRPCProxyrM 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)
rL r2 r3 r= pollresponserC acquirerB ry release)r# r ry r| rx r
r
r r 3 s"
zSocketIO._getresponsec C s | j d | _ }|S )N )nextseq)r# rg 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 )ra N)r r r r
r
r
r ra s ra c C s t | }| t|< t|S )N)idr@ r )r rM r
r
r rb s rb c @ s e Zd Zdd ZdS )r c C s
|| _ d S )N)rM )r# rM 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 dS )z(handle() method required by socketserverN)r )r# r
r
r handle s zRPCHandler.handlec C s
t | |S )N)r )r# rM r
r
r get_remote_proxy s zRPCHandler.get_remote_proxyN)r r r r> rG 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> r1 r/ r0 LOCALHOSTr< r" re )r# Z working_sockr r
r
r r s zRPCClient.acceptc C s
t | |S )N)r )r# rM r
r
r r # s zRPCClient.get_remote_proxyN) r r r r> rG 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)sockiorM )r# r rM 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 rM _RPCProxy__attributes_RPCProxy__getattributesrr r )r# r4 valuer
r
r __getattr__0 s
zRPCProxy.__getattr__c C s | j j| jdf i | _d S )NrV )r rr rM r )r# r
r
r Z__getattributes> s
zRPCProxy.__getattributesc C s | j j| jdf i | _d S )NrT )r rr rM 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 rk r4 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 rl r4 r r
r
r r^ Q s
r^ c @ s e Zd Zdd Zdd ZdS )r c C s || _ || _|| _d S )N)r rM r4 )r# r rM r4 r
r
r r" Z s zMethodProxy.__init__c O s | j j| j| j||}|S )N)r rr rM r4 )r# rI rj 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| }ytjj| W n< tk
rf d}|j|d}|j|d}tjj| Y nX tjjd | t _dS )z9Override standard display hook to use non-locale encodingNasciibackslashreplacestrictr* ) builtins_r5 r/ stdoutwriteUnicodeEncodeErrorencodedecode)r textencodingbytesr
r
r displayhookg s r __main__)mainzidlelib.idle_test.test_rpcr ) verbosity)Ni )-__doc__r r r r r8 r r r r& r r r/ r2 r6 r r r r
ZPicklerr r r r! r r@ ZQueuerc r rN r< ra rb r r r r r r] r^ r r r Zunittestr r
r
r
r sR
2
k