B
YI~dR @ 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
d
e jZi Zed Zed 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 de 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 | }|S )N)marshalloads)msco r 0/opt/alt/python37/lib64/python3.7/idlelib/rpc.py
unpickle_code- s
r c C s t | }t|ffS )N)r dumpsr )r r r r r pickle_code3 s
r
c C s$ t }t||}|| | S )N)ioBytesIO CodePicklerdumpgetvalue)objZprotocolfpr r r r 9 s
r c @ s" e Zd ZejeiZeej 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 }tj| || 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 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}td|d td|d tdt j |d td||d tdt||d tj |d td|d td|d t
d Y nX d S )
Nz)
----------------------------------------)filezUnhandled server exception!z
Thread: %szClient Address: z Request: z#
*** Unrecoverable, server exiting!z(----------------------------------------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& r5 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 | _|d k r|| _|| _|d kr*t}|| _i | _i | _d S )N) r, r-
sockthread debuggingsock objecttableobjtable responsescvars)r r9 r; r8 r r r r s
zSocketIO.__init__c C s | j }d | _ |d k r | d S )N)r9 close)r r9 r r r r> s zSocketIO.closec C s t d d S )Nr )r2 r3 )r r r r exithook s zSocketIO.exithookc G sR | j s
d S | jd tt j }x |D ]}|d t| }q(W t|tjd d S )N )r' ) r8 locationstrr, r- r. 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 rG r r r
unregister s zSocketIO.unregisterc
C s | 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
|| ||ff d S dd | fS W n t k
r, Y n t
k
rB Y np tk
rX Y nZ tk
r~ }
zd
|
fS d }
~
X Y n4 d}t||| |f tjd tjtjd dS d S )Nz
localcall:)ERRORzBad request formatrL zUnknown object id: %r __methods__OK__attributes__zUnsupported method name: %rCALLQUEUE)QUEUEDNz Unsupported message type: %sCALLEXCzU*** Internal Error: rpc.py:SocketIO.localcall()
Object: %s
Method: %s
Args: %s
)r' ) EXCEPTIONN)rF TypeErrorr; _getmethods_getattributeshasattrgetattr
isinstance RemoteObject remoteref
request_queueZputr( KeyboardInterruptOSError Exceptionr+ r) r* r0 r1 )r seqr4 howrG
methodnamerC kwargsr methods
attributesmethodretZexmsgr r r localcall sT
zSocketIO.localcallc C s( | d|| | ||||}| |S )Nzremotecall:asynccall: )rF asynccall asyncreturn)r rG rc rC rd ra r r r
remotecall s zSocketIO.remotecallc C s( | d|| | ||||}| |S )Nzremotequeue:asyncqueue: )rF
asyncqueuerl )r rG rc rC rd ra r r r remotequeue s zSocketIO.remotequeuec C s` d||||ff}| }t | jkr8t }|| j|< | d| |||| | ||f |S )NrP z
asynccall:%d:)newseqr, r- r7 Conditionr= rF
putmessage)r rG rc rC rd r4 ra cvarr r r rk s
zSocketIO.asynccallc C s` d||||ff}| }t | jkr8t }|| j|< | d| |||| | ||f |S )NrQ zasyncqueue:%d:)rp r, r- r7 rq r= rF rr )r rG rc rC rd r4 ra rs r r r rn s
zSocketIO.asyncqueuec C s6 | d| | j|dd}| d| | | |S )Nz#asyncreturn:%d:call getresponse(): g?)waitzasyncreturn:%d:response: )rF getresponsedecoderesponse)r ra responser r r rl s zSocketIO.asyncreturnc C s |\}}|dkr|S |dkr d S |dkr6| d d S |dkrT| d | d S |dkrp| d| t||d kr| d
| |t||d S ) NrN rR rT zdecoderesponse: EXCEPTIONEOFzdecoderesponse: EOFrL z decoderesponse: Internal ERROR:rS z decoderesponse: Call Exception:)rF decode_interrupthook RuntimeError SystemError)r rw rb whatr r r rv s&
zSocketIO.decoderesponsec C s t d S )N)EOFError)r r r r ry s z SocketIO.decode_interrupthookc C s6 y| j d dd W n tk
r0 | d d S X d S )Ng?)myseqrt zmainloop:return)ru r} rF )r r r r mainloop s
zSocketIO.mainloopc C s6 | ||}|d k r2|\}}|dkr2|| |f}|S )NrN ) _getresponse_proxify)r r~ rt rw rb r| r r r ru # s zSocketIO.getresponsec C s4 t |trt| |jS t |tr0tt| j|S |S )N)rZ RemoteProxyRPCProxyrG listmapr )r r r r r r + s
zSocketIO._proxifyc C s | d| t | jkr:x| ||}|d k r |S q W nb| j| }| x|| jkrb| qNW | j| }| d||f | j|= | j|= | |S d S )Nz_getresponse:myseq:z-_getresponse:%s: thread woke up: response: %s)
rF r, r- r7 pollresponser= acquirer< rt release)r r~ rt rw rs r r r r 3 s"
zSocketIO._getresponsec C s | j d | _ }|S )N )nextseq)r ra r r r rp I s zSocketIO.newseqc C s | d|d y t|}W n, tjk
rJ tdt|tjd Y nX t dt
|| }xnt
|dkry0t g | j gg \}}}| j
|d t }W n ttfk
r tdY nX ||d }qbW d S )Nzputmessage:%d:r zCannot pickle:)r' z r? rF rI rK rj rm ro rk rn rl rv ry r ru r r rp rr r r r r r r r r r r r r r r r6 s>
.
I
r6 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 rG r r r r\ s r\ c @ s e Zd Zdd ZdS )r c C s
|| _ d S )N)rG )r rG 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( | |_ t| | tj| ||| d S )N)Zcurrent_handlerr6 r r BaseRequestHandler)r r9 r! Zsvrr r r r s zRPCHandler.__init__c C s | d S )N)r )r r r r handle s zRPCHandler.handlec C s
t | |S )N)r )r rG r r r get_remote_proxy s zRPCHandler.get_remote_proxyN)r r r r8 rA 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| | jd d S )Nr )r# listening_sockZbindZlisten)r addressZfamilytyper r r r s zRPCClient.__init__c C sV | j \}}| jr$td|tjd |d tkr>t| | ntd|tjd t d S )Nz ****** Connection request from )r' r z** Invalid host: )
r acceptr8 r+ r) r* LOCALHOSTr6 r r_ )r Z working_sockr r r r r s zRPCClient.acceptc C s
t | |S )N)r )r rG r r r r # s zRPCClient.get_remote_proxyN) r r r r8 rA 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)sockiorG )r r rG r r r r , s zRPCProxy.__init__c C sp | j d kr| | j |r.t| j| j|S | jd kr@| || jkrd| j| jd|fi }|S t |d S )N__getattribute__)
_RPCProxy__methods_RPCProxy__getmethodsr MethodProxyr rG _RPCProxy__attributes_RPCProxy__getattributesrm r )r r. valuer r r __getattr__0 s
zRPCProxy.__getattr__c C s | j | jddi | _d S )NrO r )r rm rG r )r r r r Z__getattributes> s
zRPCProxy.__getattributesc C s | j | jddi | _d S )NrM r )r rm rG 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 )dirrY callablerZ r __bases__rV )r re r. attrsuperr r r rV F s
rV c C s0 x*t | D ] }t| |}t|s
d||< q
W d S )Nr )r rY r )r rf r. r r r r rW Q s
rW c @ s e Zd Zdd Zdd ZdS )r c C s || _ || _|| _d S )N)r rG r. )r r rG r. r r r r Z s zMethodProxy.__init__c O s | j | j| j||}|S )N)r rm rG r. )r rC rd 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| }ytj| W n< tk
rf d}||d}||d}tj| Y nX tjd | t _d S )Nasciibackslashreplacestrict
) builtins_r/ r) stdoutwriteUnicodeEncodeErrorencodedecode)r textencodingbytesr r r displayhookg s r __main__)mainzidlelib.idle_test.test_rpcr ) verbosity)N),r r r r r2 r r r r# r r r) r, r0 r r r
r ZPicklerr
r r r r r: ZQueuer] r rH r6 r[ r\ r r r r r rV rW r r r Zunittestr r r r r sP
2
k