a  zeTR@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddZddZd*ddZGd d d ejZd Zd ZGd dde jZiZedZedZGdddZGdddZddZGdddZ Gddde j!eZ"GdddeZ#GdddZ$ddZ%dd Z&Gd!d"d"Z'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|}|S)z*Return code object from marshal string ms.)marshalloads)mscor?/opt/bitninja-python-dojo/embedded/lib/python3.9/idlelib/rpc.py unpickle_code-s rcCst|}t|ffS)zBReturn unpickle function and tuple with marshalled co code object.)rdumpsr)rrrrr pickle_code3s r cCs$t}t||}|||S)z.Return pickled (or marshalled) string for obj.)ioBytesIO CodePicklerdumpgetvalue)objprotocolfprrrr 9s  r c@seZdZejeiejZdS)r N)__name__ __module__ __qualname__typesCodeTyper copyregdispatch_tablerrrrr Bsr i z 127.0.0.1c@s6eZdZd ddZddZddZdd Zd d ZdS) RPCServerNcCs |dur t}tj|||dSN) RPCHandler socketserver TCPServer__init__)selfaddrZ handlerclassrrrr KszRPCServer.__init__cCsdS)z@Override TCPServer method, no bind() phase for connecting entityNrr!rrr server_bindPszRPCServer.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)socketconnectserver_addressr#rrrserver_activateTszRPCServer.server_activatecCs |j|jfS)z:Override TCPServer method, return already connected socket)r%r'r#rrr get_request]szRPCServer.get_requestcCszWntyYntj}td|dtd|dtdtj|dtd||dtdt||dtj |dtd|dtd|dt d Yn0d 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_addresserfrrr handle_erroras      zRPCServer.handle_error)N)rrrr r$r(r)r:rrrrrIs   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|_|dur||_||_|dur*t}||_i|_i|_dSr) r0r1 sockthread debuggingsock objecttableobjtable responsescvars)r!r>r@r=rrrr s zSocketIO.__init__cCs |j}d|_|dur|dSr)r>close)r!r>rrrrCszSocketIO.closecCstddS)z!override for specific exit actionrN)r6r7r#rrrexithookszSocketIO.exithookcGsN|js dS|jdttj}|D]}|dt|}q&t|tjddS)N r*) r=locationstrr0r1r2r/r-r.)r!argssarrrdebugs zSocketIO.debugcCs||j|<dSr)r@)r!oidobjectrrrregisterszSocketIO.registercCs$z |j|=WntyYn0dSr)r@KeyErrorr!rLrrr unregisters  zSocketIO.unregisterc Cs|d|z|\}\}}}}Wnty4YdS0||jvrNdd|ffS|j|}|dkrvi} t|| d| fS|dkri} t|| d| fSt||sdd|ffSt||} zh|d kr| |i|} t| trt | } d| fWS|d krt || ||ffWd Sdd |fWSWnt y4Ynt yHYnxty\Yndty} zd | fWYd} ~ Sd} ~ 0d}t||| |ftjdtjtjdYdS0dS)Nz localcall:)ERRORzBad request formatrRzUnknown 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)rK TypeErrorr@ _getmethods_getattributeshasattrgetattr isinstance RemoteObject remoteref request_queueputr,KeyboardInterruptOSError Exceptionr/r-r.r4r5)r!seqr8howrL methodnamerHkwargsrmethods attributesmethodretexmsgrrr localcallsT           zSocketIO.localcallcCs(|d|||||||}||S)Nzremotecall:asynccall: )rK asynccall asyncreturnr!rLrjrHrkrhrrr remotecallszSocketIO.remotecallcCs(|d|||||||}||S)Nzremotequeue:asyncqueue: )rK asyncqueuertrurrr remotequeueszSocketIO.remotequeuecCs`d||||ff}|}t|jkr8t}||j|<|d||||||||f|S)NrVz asynccall:%d:newseqr0r1r< ConditionrBrK putmessager!rLrjrHrkr8rhcvarrrrrss zSocketIO.asynccallcCs`d||||ff}|}t|jkr8t}||j|<|d||||||||f|S)NrWzasyncqueue:%d:ryr}rrrrws zSocketIO.asyncqueuecCs6|d||j|dd}|d||||S)Nz#asyncreturn:%d:call getresponse(): 皙?)waitzasyncreturn:%d:response: )rK getresponsedecoderesponse)r!rhresponserrrrtszSocketIO.asyncreturncCs|\}}|dkr|S|dkr dS|dkr6|ddS|dkrT|d|dS|dkrp|d|t||d kr|d ||t||dS) NrTrXrZzdecoderesponse: EXCEPTIONEOFzdecoderesponse: EOFrRzdecoderesponse: Internal ERROR:rYzdecoderesponse: Call Exception:)rKdecode_interrupthook RuntimeError SystemError)r!rriwhatrrrrs&    zSocketIO.decoderesponsecCstdS)N)EOFErrorr#rrrrszSocketIO.decode_interrupthookcCs6z|jdddWnty0|dYdS0dS)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. Nr)myseqrzmainloop:return)rrrKr#rrrmainloops   zSocketIO.mainloopcCs6|||}|dur2|\}}|dkr2|||f}|S)NrT) _getresponse_proxify)r!rrrrirrrrr"s  zSocketIO.getresponsecCs4t|trt||jSt|tr0tt|j|S|Sr)r` RemoteProxyRPCProxyrLlistmapr)r!rrrrr*s    zSocketIO._proxifycCs|d|t|jur6|||}|dur|Sqn^|j|}|||jvr\|qH|j|}|d||f|j|=|j|=| |SdS)Nz_getresponse:myseq:z-_getresponse:%s: thread woke up: response: %s) rKr0r1r< pollresponserBacquirerArrelease)r!rrrr~rrrr2s"      zSocketIO._getresponsecCs|jd|_}|S)N)nextseq)r!rhrrrrzHszSocketIO.newseqc Cs|d|dz t|}Wn*tjyHtdt|tjdYn0t dt ||}t |dkrz0t g|j gg\}}}|j |dt}WnttfytdYn0||d}q^dS)Nzputmessage:%d:rzCannot pickle:r*sendBUFSIZEAttributeErrorr[rf)r!messagerIrwxnrrrr|Ls  zSocketIO.putmessagecCs|t|j|jkrt|jggg|\}}}t|dkrFdSz|jt}Wnt ylt Yn0t|dkr~t |j|7_|| S)Nr) _stage0rbuffbufneedrr>filenorecvrrfr_stage1)r!rrrrrIrrr pollpacket`s    zSocketIO.pollpacketcCsR|jdkrNt|jdkrN|jdd}|jdd|_td|d|_d|_dS)Nrrr)bufstaterrrunpackr)r!rIrrrrps zSocketIO._stage0cCsP|jdkrLt|j|jkrL|jd|j}|j|jd|_d|_d|_|SdS)Nrrr)rrrr)r!packetrrrrws zSocketIO._stage1cCs~||}|durdSzt|}WnTtjyxtdtjdtdt|tjdtj tjdtdtjdYn0|S)Nz-----------------------r*zcannot unpickle packet:) rrrUnpicklingErrorr/r-r.r3r4 print_stack)r!rrrrrr pollmessages zSocketIO.pollmessagec CsZztd}Wntjy"Yn 0|\}}|d|ff}||z||}|dur\WdSWn.tyz|YdStyYdS0|\}}|d}| d||f|dvr | d|| ||}| d||f|dkr|||fq|d krqq||kr|S|j |d} | dur| ||j |<| | qqdS) aRHandle messages received on the socket. Some messages received may be asynchronous 'call' or 'queue' requests, and some may be responses for other threads. 'call' requests are passed to self.localcall() with the expectation of immediate execution, during which time the socket is not serviced. 'queue' requests are used for tasks (which may block or hang) to be processed in a different thread. These requests are fed into request_queue by self.localcall(). Responses to queued requests are taken from response_queue and sent across the link with the associated sequence numbers. Messages in the queues are (sequence_number, request/response) tuples and code using this module removing messages from the request_queue is responsible for returning the correct sequence number in the response_queue. pollresponse() will loop until a response message with the myseq sequence number is received, and will save other responses in self.responses and notify the owning thread. rrTNzpollresponse:%d:myseq:%s)rVrWzpollresponse:%d:localcall:call:z%pollresponse:%d:localcall:response:%srVrW)response_queuegetqueueEmptyr|rr handle_EOFrrKrrrBrrAnotifyr) r!rrZqmsgrhrrZresqricvrrrrsL           zSocketIO.pollresponsecCsV||d|jD]0}|j|}|d|j|<||q|dS)z+action taken upon link being closed by peerr)rNN)EOFhookrKrBrrArrrD)r!keyrrrrrs     zSocketIO.handle_EOFcCsdS)zBClasses using rpc client/server can override to augment EOF actionNrr#rrrrszSocketIO.EOFhook)NN)"rrrrr rCrDrKrNrQrrrvrxrsrwrtrrrrrrrzr|rrrrrrrrrrrrrrr;s> .   I r;c@s eZdZdS)raN)rrrrrrrrasracCst|}|t|<t|Sr)idr?r)rrLrrrrbsrbc@seZdZddZdS)rcCs ||_dSr)rLrPrrrr szRemoteProxy.__init__N)rrrr rrrrrsrc@s,eZdZdZdZddZddZddZd S) rFz#ScCs(||_t||tj||||dSr)Zcurrent_handlerr;r rBaseRequestHandler)r!r>r"Zsvrrrrr s zRPCHandler.__init__cCs |dS)z(handle() method required by socketserverN)rr#rrrhandleszRPCHandler.handlecCs t||SrrrPrrrget_remote_proxyszRPCHandler.get_remote_proxyN)rrrr=rFr rrrrrrrs rc@s:eZdZdZdZdZejejfddZ ddZ dd Z d S) RPCClientFz#CrcCs*t|||_|j||jddSNr)r%listening_sockbindlisten)r!addressfamilytyperrrr 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. LOCALHOSTr;r rf)r!Z working_sockrrrrrs zRPCClient.acceptcCs t||SrrrPrrrr"szRPCClient.get_remote_proxyN) rrrr=rFrr%AF_INET SOCK_STREAMr rrrrrrr s  rc@s4eZdZdZdZddZddZddZdd ZdS) rNcCs||_||_dSr)sockiorL)r!rrLrrrr +szRPCProxy.__init__cCsp|jdur||j|r.t|j|j|S|jdur@|||jvrd|j|jd|fi}|St |dS)N__getattribute__) _RPCProxy__methods_RPCProxy__getmethodsr MethodProxyrrL_RPCProxy__attributes_RPCProxy__getattributesrvr)r!r2valuerrr __getattr__/s     zRPCProxy.__getattr__cCs|j|jddi|_dS)NrUr)rrvrLrr#rrrZ__getattributes=s zRPCProxy.__getattributescCs|j|jddi|_dS)NrSr)rrvrLrr#rrrZ __getmethodsAs zRPCProxy.__getmethods) rrrrrr rrrrrrrr&s rcCsLt|D]}t||}t|rd||<qt|trH|jD]}t||q8dSr)dirr_callabler`r __bases__r\)rrlr2attrsuperrrrr\Es     r\cCs,t|D]}t||}t|sd||<qdSr)rr_r)rrmr2rrrrr]Ps  r]c@seZdZddZddZdS)rcCs||_||_||_dSr)rrLr2)r!rrLr2rrrr YszMethodProxy.__init__cOs|j|j|j||}|Sr)rrvrLr2)r!rHrkrrrr__call__^szMethodProxy.__call__N)rrrr rrrrrrWsrcCs||dur dSdt_t|}ztj|Wn:tydd}||d}||d}tj|Yn0tjd|t_dS)z9Override standard display hook to use non-locale encodingNasciibackslashreplacestrict ) builtins_r3r-stdoutwriteUnicodeEncodeErrorencodedecode)rtextencodingbytesrrr displayhookfs    r__main__)mainzidlelib.idle_test.test_rpcr) verbosity)N),__doc__rrr rr6rrrr%rrr-r0r4rrr r Picklerr rrrrr?Queuercrr;rarbrrrrrr\r]rrrZunittestrrrrrsR 2  k