B X<˳H@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddZddZd*ddZGd d d ejZd Zd ZGd dde jZiZedZedZGdddeZGdddeZddZ GdddeZ!Gddde j"eZ#GdddeZ$GdddeZ%ddZ&dd Z'Gd!d"d"eZ(d#d$Z)e*d%krdd&l+m,Z,e,d'd(d)dS)+aRPC 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. NcCst|}t|tjst|S)z*Return code object from marshal string ms.)marshalloads isinstancetypesCodeTypeAssertionError)mscor 0/opt/alt/python37/lib64/python3.7/idlelib/rpc.py unpickle_code-s r cCs$t|tjstt|}t|ffS)zBReturn unpickle function and tuple with marshalled co code object.)rrrrrdumpsr )r rr r r pickle_code3s rcCs$t}t||}|||S)z.Return pickled (or marshalled) string for obj.)ioBytesIO CodePicklerdumpgetvalue)objZprotocolfpr r r r 9s  r c@s"eZdZejeiZeejdS)rN) __name__ __module__ __qualname__rrrdispatch_tableupdatecopyregr r r r rBs ri z 127.0.0.1c@s6eZdZd ddZddZddZdd Zd d ZdS) RPCServerNcCs |dkr t}tj|||dS)N) RPCHandler socketserver TCPServer__init__)selfaddrZ handlerclassr r r r!LszRPCServer.__init__cCsdS)z@Override TCPServer method, no bind() phase for connecting entityNr )r"r r r server_bindQszRPCServer.server_bindcCs|j|jdS)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_activateUszRPCServer.server_activatecCs |j|jfS)z:Override TCPServer method, return already connected socket)r%r&)r"r r r get_request^szRPCServer.get_requestcCsyWntk rYntj}td|dtd|dtdtj|dtd||dtdt||dtj |dtd|dtd|dt d YnXd 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. z) ----------------------------------------)filezUnhandled server exception!z Thread: %szClient Address: z Request: z# *** Unrecoverable, server exiting!z(----------------------------------------rN) SystemExitsys __stderr__print threadingcurrent_threadnamerepr traceback print_excos_exit)r"requestZclient_addressZerfr r r handle_errorbs     zRPCServer.handle_error)N)rrrr!r$r'r(r7r r r r rJs   rc@seZdZdZd;ddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+Zd,ZdZd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Z d9d:Z!dS)<SocketIOrNcCs@t|_|dk r||_||_|dkr*t}||_i|_i|_dS)N) r.r/ sockthread debuggingsock objecttableobjtable responsescvars)r"r;r=r:r r r r!s zSocketIO.__init__cCs |j}d|_|dk r|dS)N)r;close)r"r;r r r r@szSocketIO.closecCstddS)z!override for specific exit actionrN)r4r5)r"r r r exithookszSocketIO.exithookcGsR|js dS|jdttj}x|D]}|dt|}q(Wt|tjddS)N )r)) r:locationstrr.r/r0r-r+r,)r"argssar r r debugs  zSocketIO.debugcCs||j|<dS)N)r=)r"oidobjectr r r registerszSocketIO.registercCs&y |j|=Wntk r YnXdS)N)r=KeyError)r"rIr r r unregisters zSocketIO.unregisterc Cs|d|y|\}\}}}}Wntk r4dSX||jkrNdd|ffS|j|}|dkrvi} t|| d| fS|dkri} t|| d| fSt||sdd|ffSt||} y^|d kr| ||} t| trt | } d| fS|d krt || ||ffd Sdd |fSWnt k r,Ynt k rBYnptk rXYnZtk r~} zd | fSd} ~ XYn4d}t||| |ftjdtjtjddSdS)Nz localcall:)ERRORzBad request formatrNzUnknown object id: %r __methods__OK__attributes__zUnsupported method name: %rCALLQUEUE)QUEUEDNzUnsupported message type: %sCALLEXCzU*** Internal Error: rpc.py:SocketIO.localcall() Object: %s Method: %s Args: %s )r)) EXCEPTIONN)rH TypeErrorr= _getmethods_getattributeshasattrgetattrr RemoteObject remoteref request_queueZputr*KeyboardInterruptOSError Exceptionr-r+r,r2r3)r"seqr6howrI methodnamerEkwargsrmethods attributesmethodZretZexmsgr r r localcallsT          zSocketIO.localcallcCs(|d|||||||}||S)Nzremotecall:asynccall: )rH asynccall asyncreturn)r"rIrdrErerbr r r remotecallszSocketIO.remotecallcCs(|d|||||||}||S)Nzremotequeue:asyncqueue: )rH asyncqueuerl)r"rIrdrErerbr r r remotequeueszSocketIO.remotequeuecCs`d||||ff}|}t|jkr8t}||j|<|d||||||||f|S)NrRz asynccall:%d:)newseqr.r/r9 Conditionr?rH putmessage)r"rIrdrErer6rbcvarr r r rks zSocketIO.asynccallcCs`d||||ff}|}t|jkr8t}||j|<|d||||||||f|S)NrSzasyncqueue:%d:)rpr.r/r9rqr?rHrr)r"rIrdrErer6rbrsr r r rns zSocketIO.asyncqueuecCs6|d||j|dd}|d||||S)Nz#asyncreturn:%d:call getresponse(): g?)waitzasyncreturn:%d:response: )rH getresponsedecoderesponse)r"rbresponser r r rlszSocketIO.asyncreturncCs|\}}|dkr|S|dkr dS|dkr6|ddS|dkrT|d|dS|dkrp|d|t||d kr|d ||t||dS) NrPrTrVzdecoderesponse: EXCEPTIONEOFzdecoderesponse: EOFrNzdecoderesponse: Internal ERROR:rUzdecoderesponse: Call Exception:)rHdecode_interrupthook RuntimeError SystemError)r"rwrcwhatr r r rvs&    zSocketIO.decoderesponsecCstdS)N)EOFError)r"r r r ryszSocketIO.decode_interrupthookcCs6y|jdddWntk r0|ddSXdS)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?)myseqrtzmainloop:return)rur~rH)r"r r r mainloops  zSocketIO.mainloopcCs6|||}|dk r2|\}}|dkr2|||f}|S)NrP) _getresponse_proxify)r"rrtrwrcr|r r r ru#s  zSocketIO.getresponsecCs4t|trt||jSt|tr0tt|j|S|S)N)r RemoteProxyRPCProxyrIlistmapr)r"rr r r r+s    zSocketIO._proxifycCs|d|t|jkr:x|||}|dk r|SqWnb|j|}|x||jkrb|qNW|j|}|d||f|j|=|j|=| |SdS)Nz_getresponse:myseq:z-_getresponse:%s: thread woke up: response: %s) rHr.r/r9 pollresponser?acquirer>rtrelease)r"rrtrwrsr r r r3s"        zSocketIO._getresponsecCs|jd|_}|S)N)nextseq)r"rbr r r rpIszSocketIO.newseqc Cs|d|dy t|}Wn,tjk rJtdt|tjdYnXt dt ||}xnt |dkry0t g|j gg\}}}|j |dt}Wn ttfk rtdYnX||d}qbWdS)Nzputmessage:%d:rzCannot pickle:)r)znotifyr) r"rrtZqmsgrbrwrZresqrccvr r r rsL         zSocketIO.pollresponsecCsZ||dx:|jD]0}|j|}|d|j|<||qW|dS)z+action taken upon link being closed by peerr)rxNN)EOFhookrHr?rr>rrrA)r"keyrr r r rs     zSocketIO.handle_EOFcCsdS)zBClasses using rpc client/server can override to augment EOF actionNr )r"r r r rszSocketIO.EOFhook)NN)"rrrrr!r@rArHrKrMrjrmrorkrnrlrvryrrurrrprrrrrrrrrrrrr r r r r8s> .   I r8c@s eZdZdS)r\N)rrrr r r r r\sr\cCst|}|t|<t|S)N)idr<r)rrIr r r r]sr]c@seZdZddZdS)rcCs ||_dS)N)rI)r"rIr r r r!szRemoteProxy.__init__N)rrrr!r r r r rsrc@s,eZdZdZdZddZddZddZd S) rFz#ScCs(||_t||tj||||dS)N)Zcurrent_handlerr8r!rBaseRequestHandler)r"r;r#Zsvrr r r r!s zRPCHandler.__init__cCs |dS)z(handle() method required by socketserverN)r)r"r r r handleszRPCHandler.handlecCs t||S)N)r)r"rIr r r get_remote_proxy szRPCHandler.get_remote_proxyN)rrrr:rCr!rrr r r r rs rc@s:eZdZdZdZdZejejfddZ ddZ dd Z d S) RPCClientFz#CrcCs*t|||_|j||jddS)Nr)r%listening_sockZbindZlisten)r"addressZfamilytyper r r r!s zRPCClient.__init__cCsV|j\}}|jr$td|tjd|dtkr>t||ntd|tjdt dS)Nz****** Connection request from )r)rz** Invalid host: ) racceptr:r-r+r, LOCALHOSTr8r!r`)r"Z working_sockrr r r rs zRPCClient.acceptcCs t||S)N)r)r"rIr r r r#szRPCClient.get_remote_proxyN) rrrr:rCrr%ZAF_INETZ SOCK_STREAMr!rrr r r r r s  rc@s4eZdZdZdZddZddZddZdd ZdS) rNcCs||_||_dS)N)sockiorI)r"rrIr r r r!,szRPCProxy.__init__cCsp|jdkr||j|r.t|j|j|S|jdkr@|||jkrd|j|jd|fi}|St |dS)N__getattribute__) _RPCProxy__methods_RPCProxy__getmethodsr MethodProxyrrI_RPCProxy__attributes_RPCProxy__getattributesrmr)r"r0valuer r r __getattr__0s      zRPCProxy.__getattr__cCs|j|jddi|_dS)NrQr )rrmrIr)r"r r r Z__getattributes>s zRPCProxy.__getattributescCs|j|jddi|_dS)NrOr )rrmrIr)r"r r r Z __getmethodsBs zRPCProxy.__getmethods) rrrrrr!rrrr r r r r's rcCsTx*t|D]}t||}t|r d||<q Wt|trPx|jD]}t||q>WdS)Nr)dirr[callablerr __bases__rX)rrfr0attrsuperr r r rXFs    rXcCs0x*t|D]}t||}t|s d||<q WdS)Nr)rr[r)rrgr0rr r r rYQs rYc@seZdZddZddZdS)rcCs||_||_||_dS)N)rrIr0)r"rrIr0r r r r!ZszMethodProxy.__init__cOs|j|j|j||}|S)N)rrmrIr0)r"rErerr r r __call___szMethodProxy.__call__N)rrrr!rr r r r rXsrcCs~|dkr dSdt_t|}ytj|Wn<tk rfd}||d}||d}tj|YnXtjd|t_dS)z9Override standard display hook to use non-locale encodingNasciibackslashreplacestrict ) builtins_r1r+stdoutwriteUnicodeEncodeErrorencodedecode)rtextencodingbytesr r r displayhookgs   r__main__)mainzidlelib.idle_test.test_rpcr) verbosity)N)-__doc__rrrrr4rrrr%rrr+r.r2rr rr ZPicklerrrrr rr<ZQueuer^rrJr8r\r]rrrrrrXrYrrrZunittestrr r r r sR 2  k