$42 GRAYBYTE WORDPRESS FILE MANAGER $20

�
e f.Q�@sdZd
dl
Z
d
dlZd
dlZd
dlZd
dlZd
dlZd
dlZd
dlZd
dl	Z	d
dl
Z
d
dlZd
dlZd
dl
Z
d
dlZd
dlZdd�Zdd�Zddd�
ZGd	d
�d
ej�ZddZd
ZGdd�dej�ZiZe
jd
�
Ze
jd
�
ZGdd�de�ZGdd�de�Zdd�Z Gdd�de�Z!Gdd�dej"e�Z#Gdd�de�Z$Gdd�de�Z%dd�Z&d d!�Z'Gd"d#�d#e�Z(d$d%�Z)dS)&aRPC Implemention, 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
Cstj
|�
}
|
S)
N)�marshal�loads)�ms�co�r�0/opt/alt/python34/lib64/python3.4/idlelib/rpc.py�
unpickle_code/s
rc
Cstj
|�
}
t|
f
fS)
N)r�dumpsr)rrrrr�pickle_code4s
r
cCs2tj
�}t||
�}|j|�

|j�S)
N)�io�BytesIO�CodePickler�dump�getvalue)�objZprotocol�
f�
prrrr	9s




r	c@s0eZ
dZi
eej6Zejej�

d
S)r
N)	�__name__�
__module__�__qualname__r
�types�CodeType�dispatch_table�update�copyregrrrrr
?s

r
�iz	127.0.0.1c@sOeZ
dZd
dd�
Zdd�Zdd�Zdd	�Zd
d�Zd
S)�	RPCServerNcCs/|dkrt}nt
jj||
|�
dS)
N)�
RPCHandler�socketserver�	TCPServer�__init__)�self�addrZhandlerclassrrrr Hs

	
zRPCServer.__init__c


Csd
S)z@Override TCPServer method, no bind() phase for connecting entityNr)
r!rrr�server_bindMszRPCServer.server_bindc

Cs|jj
|j�

d
S)z�Override 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)�socketZconnect�server_address)
r!rrr�server_activateQszRPCServer.server_activatec

Cs|j|j
fS)
z:Override TCPServer method, return already connected socket)r$r%)
r!rrr�get_requestZszRPCServer.get_requestcCs�y�Wn�tk
r


�Yn�


t
j}td
ddd|�


tdd|�


tdtj�jd|�


td|d|�

tdt|
�
d|�

tj	d|�

td	d|�


tddd|�


t
jd
�

Yn
XdS)z�Override 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)�
SystemExit�sys�
__stderr__�print�	threading�current_thread�name�repr�	traceback�	print_exc�os�_exit)r!�requestZclient_addressZerfrrr�handle_error^s





	








zRPCServer.handle_error)rrrr r#r&r'r9rrrrrFs
	rc@sr
eZ
dZd
Zdddd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+Zd,Zd
Zd-d.�Zd/d0�Zd1d2�Zd3d4�Zd5d6�Zd7d8�Z d9d:�Z!dS);�SocketIOr
NcCsdtj
�|_|dk	r'||_n|
|_|dkrEt}n||_i|_i|_dS)
N)	r0r1�
sockthread�	debugging�sock�objecttable�objtable�	responses�cvars)r!r=r?r<rrrr �s



	

	
	
	
zSocketIO.__init__c
Cs/|j}
d|_|
dk	r+|
j
�
ndS)
N)r=�close)r!r=rrrrB�s
	
	

zSocketIO.closec

Cstj
d
�

dS)z!override for specific exit actionr
N)r6r7)
r!rrr�exithook�szSocketIO.exithookc
Gsi|js
dS|j
d
ttj�j�
}x"|
D]}|d
t|�
}q4Wt|dtj�


dS)N�
 r+)	r<�location�strr0r1r2r/r-r.)r!�args�
s�
arrr�debug�s
	

 



zSocketIO.debugcCs||j|
<dS)
N)
r?)r!�oid�objectrrr�register�s
zSocketIO.registercCs'y|j|
=Wnt
k
r"


Yn
XdS)
N)r?�KeyError)r!rKrrr�
unregister�s




zSocketIO.unregistercCs*|jd
|�
y|\}\}}}}Wnt
k
rD


dSYn
X||jkredd|f
fS|j|}|dkr�i}	t||	�
d|	fS|dkr�i}
t||
�
d|
fSt||�s�dd|f
fSt||�}y�|d	kr9
|||�}t|t�r/
t	|�
}nd|fS|d
kre
t
j|
|||ff�

dSdd|fSWn�tk
r�



�Yn�t
k
r�



�Yn�tk
r�



�Ynstk
r�

}

zd
|
fSWYdd}
~
XnE


d}t||||fdtj�


tjdtj�

dSYn
XdS)Nz
localcall:�ERROR�Bad request formatzUnknown object id: %r�__methods__�OK�__attributes__zUnsupported method name: %r�CALL�QUEUE�QUEUEDzUnsupported message type: %s�CALLEXCzU*** Internal Error: rpc.py:SocketIO.localcall()

 Object: %s 
 Method: %s 
 Args: %s
r+�	EXCEPTION)rPrQ)rWN)rYN)rJ�	TypeErrorr?�_getmethods�_getattributes�hasattr�getattr�
isinstance�RemoteObject�	remoteref�
request_queueZputr,�KeyboardInterrupt�OSError�	Exceptionr/r-r.r4r5)r!�seqr8�howrK�
methodnamerG�kwargsr�methods�
attributes�methodZretZex�msgrrr�	localcall�sT





	









































 

zSocketIO.localcallcCs8|jd
|
|�
|j
|
|||�}|j|�
S)Nzremotecall:asynccall: )rJ�	asynccall�asyncreturn)r!rKrhrGrirfrrr�
remotecall�s


zSocketIO.remotecallcCs8|jd
|
|�
|j
|
|||�}|j|�
S)Nzremotequeue:asyncqueue: )rJ�
asyncqueuerp)r!rKrhrGrirfrrr�remotequeue�s


zSocketIO.remotequeuecCs�d
|
|||ff}|j�}t
j�|jkrUt
j�}||j|<n|jd||
|||�
|j||f�

|S)NrUz
asynccall:%d:)�newseqr0r1r;�	ConditionrArJ�
putmessage)r!rKrhrGrir8rf�cvarrrrro�s







zSocketIO.asynccallcCs�d
|
|||ff}|j�}t
j�|jkrUt
j�}||j|<n|jd||
|||�
|j||f�

|S)NrVzasyncqueue:%d:)rtr0r1r;rurArJrv)r!rKrhrGrir8rfrwrrrrr�s







zSocketIO.asyncqueuecCsG|jd
|
�

|j
|
dd�

}|jd|
|�
|j|�
S)Nz#asyncreturn:%d:call getresponse(): �waitg�������?zasyncreturn:%d:response: )rJ�getresponse�decoderesponse)r!rf�responserrrrp�s



zSocketIO.asyncreturncCs�|
\}}|d
kr|S|dkr,dS|dkrI|jd�

dS|dkrp|jd�

|j
�
dS|dkr�|jd|�
t|�
�
n|d	kr�|jd
|�
|�
nt||��
dS)NrSrWrYzdecoderesponse: EXCEPTION�EOFzdecoderesponse: EOFrPzdecoderesponse: Internal ERROR:rXzdecoderesponse: Call Exception:)rJ�decode_interrupthook�RuntimeError�SystemError)r!r{rg�whatrrrrz�s&




















	
zSocketIO.decoderesponsec


Cs
t�
d
S)�N)
�EOFError)
r!rrrr}
szSocketIO.decode_interrupthookc

CsDy|jd
ddd�
Wn#t
k
r?


|jd�

dSYn
XdS)z�Listen on socket until I/O not ready or EOF

        pollresponse() will loop looking for seq number None, which
        never comes, and exit on EOFError.

        �myseqNrxg�������?zmainloop:return)ryr�rJ)
r!rrr�mainloop
s






zSocketIO.mainloopcCsU|j|
|�}|dk	rQ|\}}|d
krQ||j
|�
f}qQn|S)NrS)�_getresponse�_proxify)r!r�rxr{rgr�rrrry
s





zSocketIO.getresponsecCsHt|
t
�rt||
j�St|
t�rDtt|j|
��
S|
S)
N)r_�RemoteProxy�RPCProxyrK�list�mapr�)r!rrrrr�'
s




zSocketIO._proxifycCs�|jd
|
�
t
j�|jkrQx�|j|
|�}|dk	r(|Sq(Wn}|j|
}|j�
x|
|jkr�|j�
qkW|j|
}|jd|
|f�

|j|
=|j|
=|j	�
|SdS)Nz_getresponse:myseq:z-_getresponse:%s: thread woke up: response: %s)
rJr0r1r;�pollresponserA�acquirer@rx�release)r!r�rxr{rwrrrr�/
s"












	







zSocketIO._getresponsec
Cs|jd
|_}
|
S)N�)
�nextseq)r!rfrrrrtE
s

zSocketIO.newseqcCs
|jd
|
d�

yt
|
�
}Wn4tjk
r[


tdt|
�
dtj�

�Yn
Xtj	dt
|�
�|}x�t
|�
dkr
yDtjg|jg
g�\}}}|jj
|dt��
}Wn$ttfk
r�


td�
�
Yn
X||d�}q{WdS)Nzputmessage:%d:r
zCannot pickle:r+z<izsocket no longer exists)rJr	�pickleZ
PicklingErrorr/r3r-r.�structZpack�len�selectr=�send�BUFSIZE�AttributeErrorrZrd)r!�messagerH�
r�
w�
x�
nrrrrvI
s









$
 


zSocketIO.putmessage��cCs�|j�
t
|j�
|jkr�tj|jj�g
gg|
�\}}}t
|�
d
kredSy|jjt�
}Wnt	k
r�


t
�
Yn
Xt
|�
d
kr�t
�
n|j|7_|j�
n|j�S)Nr
)�_stage0r��buff�bufneedr�r=�filenoZrecvr�rdr��_stage1)r!rxr�r�r�rHrrr�
pollpacket]
s



-








	



zSocketIO.pollpacketc
Csv|jd
krrt
|j�
dkrr|jdd�}
|jdd�|_tjd|
�d
|_d|_ndS)Nr
r�z<i�
)�bufstater�r�r�Zunpackr�)r!rHrrrr�m
s

$



zSocketIO._stage0c
Csp|jd
krlt
|j�
|jkrl|jd|j�}
|j|jd�|_d|_d|_|
SdS)Nr�r�r
)r�r�r�r�)r!�packetrrrr�t
s
'


	
	
zSocketIO._stage1cCs�|j|
�
}|dkrdSyt
j|�
}Wnmt
jk
r�


td
dtj�


tdt|�
dtj�

tj	dtj�

td
dtj�


�Yn
X|S)Nz-----------------------r+zcannot unpickle packet:)
r�r�rZUnpicklingErrorr/r-r.r3r4�print_stack)r!rxr�r�rrr�pollmessage|
s











zSocketIO.pollmessagec
Cs�
x�
ytj
d
�
}Wntjk
r-


Yn,X|\}}|d|ff}|j|�

y#|j|�
}|dkr{dSWn5tk
r�


|j�
dSYntk
r�


dSYn
X|\}}|d
}|j	d||
f�

|d	kr^
|j	d|�

|j
||�}|j	d||f�

|dkrI
|j||f�

q|dkrqqq||
krn
|S|jj
|d�}	|	dk	r|	j�
||j
|<|	j�
|	j�
qqqWdS)
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.

        r
rSNzpollresponse:%d:myseq:%srUrVzpollresponse:%d:localcall:call:z%pollresponse:%d:localcall:response:%s)zCALLzQUEUE)�response_queue�get�queueZEmptyrvr�r��
handle_EOFr�rJrnrAr�r@�notifyr�)
r!r�rxZqmsgrfr{r�Zresqrg�cvrrrr��
sN
















	






	














zSocketIO.pollresponsec
Csq|j�
|j
d
�

xI|jD]>}
|j|
}|j�
d|j|
<|j�
|j�
q!W|j�
dS)z+action taken upon link being closed by peerr�r|N)zEOFN)�EOFhookrJrAr�r@r�r�rC)r!�keyr�rrrr��
s












zSocketIO.handle_EOFc


Csd
S)zBClasses using rpc client/server can override to augment EOF actionNr)
r!rrrr��
szSocketIO.EOFhook)"rrrr�r rBrCrJrMrOrnrqrsrorrrprzr}r�ryr�r�rtrvr�r�r�r�r�r�r�r�r�r�rrrrr:}s>.




I
r:c
@seZ
dZd
S)r`N)rrrrrrrr`�
sr`c
Cs t|�
}
|t
|
<t|
�
S)
N)�idr>r�)rrKrrrra�
s



rac@seZ
dZd
d�ZdS)r�cCs
|
|_dS)
N)
rK)r!rKrrrr �
s
zRemoteProxy.__init__N)rrrr rrrrr��
sr�c@s@eZ
dZd
ZdZdd�Zdd�Zdd�Zd	S)
rFz#ScCs6||_t
j||
�
tjj||
||�
dS)
N)Zcurrent_handlerr:r r�BaseRequestHandler)r!r=r"Zsvrrrrr �
s
	

zRPCHandler.__init__c


Cs|j�
d
S)z(handle() method required by socketserverN)
r�)
r!rrr�handle�
szRPCHandler.handlecCs
t||
�S)
N)
r�)r!rKrrr�get_remote_proxys
zRPCHandler.get_remote_proxyN)rrrr<rEr r�r�rrrrr�
s

rc@sReZ
dZd
ZdZdZejejdd�Z	dd�Z
dd	�Zd
S)�	RPCClientFz#Cr�cCs9tj||�|_
|j
j|
�

|j
jd
�

dS)Nr�)r$�listening_sockZbindZlisten)r!�addressZfamily�typerrrr s


zRPCClient.__init__c
Csz|jj
�\}
}|jr7td
|dtj�

n|dtkrZtj||
�
ntd|dtj�

t	�
dS)Nz****** Connection request from r+r
z** Invalid host: )
r��acceptr<r/r-r.�	LOCALHOSTr:r rd)r!Zworking_sockr�rrrr�s

	



zRPCClient.acceptcCs
t||
�S)
N)
r�)r!rKrrrr�s
zRPCClient.get_remote_proxyN)rrrr<rEr�r$ZAF_INETZSOCK_STREAMr r�r�rrrrr�s

r�c@sLeZ
dZd
Zd
Zdd�Zdd�Zdd�Zdd	�Zd
S)
r�NcCs|
|_||_
dS)
N)�sockiorK)r!r�rKrrrr #s
	
zRPCProxy.__init__cCs�|jdkr|j
�
n|jj|
�
rDt|j|j|
�S|jdkr`|j�
n|
|jkr�|jj|jd
|
f
i�}|St	|
�
�
dS)N�__getattribute__)
�_RPCProxy__methods�_RPCProxy__getmethodsr��MethodProxyr�rK�_RPCProxy__attributes�_RPCProxy__getattributesrqr�)r!r2�valuerrr�__getattr__'s











zRPCProxy.__getattr__c

Cs%|jj
|jd
fi�|_dS)NrT)r�rqrKr�)
r!rrrZ__getattributes5s

zRPCProxy.__getattributesc

Cs%|jj
|jd
fi�|_dS)NrR)r�rqrKr�)
r!rrrZ__getmethods9s

zRPCProxy.__getmethods)	rrrr�r�r r�r�r�rrrrr�s
r�cCsvx<t|�
D].}t
||�}t|�
r
d
|
|<q
q
Wt|t�rrx!|jD]}t||
�
qXWndS)Nr�)�dirr^�callabler_r��	__bases__r[)rrjr2�attr�superrrrr[=s





r[cCsCx<t|�
D].}t
||�}t|�
s
d
|
|<q
q
WdS)Nr�)r�r^r�)rrkr2r�rrrr\Hs



r\c@s(eZ
dZd
d�Zdd�ZdS)r�cCs|
|_||_
||_dS)
N)r�rKr2)r!r�rKr2rrrr Ps
	
	
zMethodProxy.__init__c
Os%|jj
|j|j|
|�}|S)
N)r�rqrKr2)r!rGrir�rrr�__call__Us
!
zMethodProxy.__call__N)rrrr r�rrrrr�Nsr�c
Cs�|d
krd
Sd
t_
t|�
}
ytjj|
�

WnLtk
r�


d}|
j|d�}|j|d�}
tjj|
�

Yn
Xtjjd�

|t_
d
S)z9Override standard display hook to use non-locale encodingN�ascii�backslashreplace�strictr()	�builtins�
_r3r-�stdout�write�UnicodeEncodeError�encode�decode)r��text�encoding�bytesrrr�displayhook]s
	









r�)*�__doc__r-r6rr$r�rr�r�r0r�r4rrrr�rr
r	ZPicklerr
r�r�rrr>ZQueuerbr�rLr:r`rar�r�rr�r�r[r\r�r�rrrr�<module>sJ















2

�j