Your IP : 216.73.216.85
o
0�iZR � @ 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)dd�ZG dd � d ej�Zd
Zd ZG d d
� d
e j�Zi Ze�d �Ze�d �ZG dd� d�Z G dd� d�Z dd� Z G dd� d�Z G dd� de j e �Z!G dd� de �Z"G dd� d�Z#d d � Z$d d � Z%G d d!� d!�Z&d"d#� Z'e(d$kr�d d%l)m*Z* e*d&d'd(� dS dS )*� Nc C s t �| �}|S �N)�marshal�loads)�ms�co� r �2/opt/alt/python310/lib64/python3.10/idlelib/rpc.py�
unpickle_code- s
r c C s t �| �}t|ffS r )r �dumpsr )r r r r r � pickle_code3 s
r c C s& t �� }t||�}|�| � |�� S r )�io�BytesIO� CodePickler�dump�getvalue)�objZprotocol�f�pr r r r
9 s
r
c @ s e Zd Zejeiej�ZdS )r N)�__name__�
__module__� __qualname__�types�CodeTyper �copyreg�dispatch_tabler r r r r B s r i z 127.0.0.1c @ s6 e Zd Zd dd�Zdd� Zdd� Zdd � Zd
d � ZdS )
� RPCServerNc C s |d u rt }tj�| ||� d S r )�
RPCHandler� socketserver� TCPServer�__init__)�self�addrZ handlerclassr r r r K s zRPCServer.__init__c C � d S r r �r r r r � server_bindP � zRPCServer.server_bindc C s | j �| j� d S r )�socketZconnect�server_addressr# r r r �server_activateT s zRPCServer.server_activatec C s | j | jfS r )r& r' r# r r r � get_request] � zRPCServer.get_requestc C s� z� t y
� tj}td|d� td|d� tdt�� j |d� td||d� tdt|�|d� tj |d� td|d� td|d� t
� d � Y d S )
Nz)
----------------------------------------��filezUnhandled server exception!z
Thread: %szClient Address: z Request: z#
*** Unrecoverable, server exiting!z(----------------------------------------r ) �
SystemExit�sys�
__stderr__�print� threading�current_thread�name�repr� traceback� print_exc�os�_exit)r �requestZclient_addressZerfr r r � handle_errora s zRPCServer.handle_errorr )r r r r r$ r( r) r: r r r r r I s
r c @ s� e Zd ZdZd;dd�Zdd� 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 ur || _|| _|d u rt}|| _i | _i | _d S r ) r1 r2 �
sockthread� debugging�sock� objecttable�objtable� responses�cvars)r r> r@ r= r r r r � s
zSocketIO.__init__c C s$ | j }d | _ |d ur|�� d S d S r )r> �close)r r> r r r rC � s
�zSocketIO.closec C s t �d� d S �Nr )r7 r8 r# r r r �exithook� s zSocketIO.exithookc G sN | j sd S | jd tt�� j� }|D ]
}|d t|� }qt|tjd� d S )N� r+ ) r= �location�strr1 r2 r3 r0 r. r/ )r �args�s�ar r r �debug� s zSocketIO.debugc C s || j |< d S r )r@ )r �oid�objectr r r �register� s zSocketIO.registerc C s$ z| j |= W d S ty Y d S w r )r@ �KeyError�r rM r r r �
unregister� s
�zSocketIO.unregisterc
C s� | � d|� z
|\}\}}}}W n
ty Y dS w || jvr'dd|f fS | j| }|dkr;i } t|| � d| fS |dkrJi }
t||
� d|
fS t||�sVdd|f fS t||�} z1|d kru| |i |��} t| t�rpt | �} d| fW S |d
kr�t
� || ||ff� W d S dd | fW S t y� � t
y� � ty� � ty� }
z
d
|
fW Y d }
~
S d }
~
w d}t||| |f tjd� tjtjd� Y dS )Nz
localcall:)�ERRORzBad request formatrS zUnknown object id: %r� __methods__�OK�__attributes__zUnsupported method name: %r�CALL�QUEUE)�QUEUEDNz Unsupported message type: %s�CALLEXCzU*** Internal Error: rpc.py:SocketIO.localcall()
Object: %s
Method: %s
Args: %s
r+ )� EXCEPTIONN)rL � TypeErrorr@ � _getmethods�_getattributes�hasattr�getattr�
isinstance� RemoteObject� remoteref�
request_queueZputr- �KeyboardInterrupt�OSError� Exceptionr0 r. r/ r5 r6 )r �seqr9 �howrM �
methodnamerI �kwargsr �methods�
attributes�methodZretZex�msgr r r � localcall� sX �
�zSocketIO.localcallc C �( | � d||� | �||||�}| �|�S )Nzremotecall:asynccall: )rL � asynccall� asyncreturn�r rM rj rI rk rh r r r �
remotecall� �
zSocketIO.remotecallc C rq )Nzremotequeue:asyncqueue: )rL �
asyncqueuers rt r r r � remotequeue� rv zSocketIO.remotequeuec C �` d||||ff}| � � }t�� | jkr t�� }|| j|< | �d| ||||� | �||f� |S )NrW z
asynccall:%d:��newseqr1 r2 r< Z ConditionrB rL �
putmessage�r rM rj rI rk r9 rh �cvarr r r rr � �
zSocketIO.asynccallc C ry )NrX zasyncqueue:%d:rz r} r r r rw � r zSocketIO.asyncqueuec C s6 | � d| � | j|dd�}| � d| |� | �|�S )Nz#asyncreturn:%d:call getresponse(): 皙�����?)�waitzasyncreturn:%d:response: )rL � getresponse�decoderesponse)r rh �responser r r rs � s
zSocketIO.asyncreturnc C s� |\}}|dkr
|S |dkrd S |dkr| � d� d S |dkr*| � d� | �� d S |dkr8| � d|� t|��|d krD| � d
|� |�t||��) NrU rY r[ zdecoderesponse: EXCEPTION�EOFzdecoderesponse: EOFrS z decoderesponse: Internal ERROR:rZ z decoderesponse: Call Exception:)rL �decode_interrupthook� RuntimeError� SystemError)r r� ri �whatr r r r� � s&
zSocketIO.decoderesponsec C s t �r )�EOFErrorr# r r r r� r% z SocketIO.decode_interrupthookc C s6 z
| j d dd� W d S ty | �d� Y d S w )Nr� )�myseqr� zmainloop:return)r� r� rL r# r r r �mainloop s
�zSocketIO.mainloopc C s6 | � ||�}|d ur|\}}|dkr|| �|�f}|S )NrU )� _getresponse�_proxify)r r� r� r� ri r� r r r r� " s zSocketIO.getresponsec C s4 t |t�r t| |j�S t |t�rtt| j|��S |S r )ra � RemoteProxy�RPCProxyrM �list�mapr� )r r r r r r� * s
zSocketIO._proxifyc C s� | � d|� t�� | ju r | �||�}|d ur|S q| j| }|�� || jvr2|�� || jvs)| j| }| � d||f � | j|= | j|= |� � |S )Nz_getresponse:myseq:Tz-_getresponse:%s: thread woke up: response: %s)
rL r1 r2 r< � pollresponserB �acquirerA r� �release)r r� r� r� r~ r r r r� 2 s( �
�
�zSocketIO._getresponsec C s | j d | _ }|S )N� )�nextseq)r rh r r r r{ H s zSocketIO.newseqc C s� | � d|d � zt|�}W n tjy" tdt|�tjd� � w t� dt
|��| }t
|�dkrgzt � g | j gg �\}}}| j �
|d t� �}W n
ttfyX td��w ||d � }t
|�dks3d S d S )Nzputmessage:%d:r zCannot pickle:r+ �<izsocket no longer exists)rL r
�pickleZ
PicklingErrorr0 r4 r. r/ �structZpack�len�selectr> �send�BUFSIZE�AttributeErrorr\ rf )r �messagerJ �r�w�x�nr r r r| L s" � � �zSocketIO.putmessage� � c C s� | � � t| j�| jk rHt�| j�� gg g |�\}}}t|�dkr#d S z| j�t�}W n t y4 t
�w t|�dkr=t
�| j|7 _| � � | � � S rD ) �_stage0r� �buff�bufneedr� r> �filenoZrecvr� rf r� �_stage1)r r� r� r� r� rJ r r r �
pollpacket` s � zSocketIO.pollpacketc C sZ | j dkr)t| j�dkr+| jd d� }| jdd � | _t�d|�d | _d| _ d S d S d S )Nr r� r� � )�bufstater� r� r� Zunpackr� )r rJ r r r r� p s
�zSocketIO._stage0c C sT | j dkr&t| j�| jkr(| jd | j� }| j| jd � | _d| _d| _ |S d S d S )Nr� r� r )r� r� r� r� )r �packetr r r r� w s �zSocketIO._stage1c C sx | � |�}|d u r d S zt�|�}W |S tjy; tdtjd� tdt|�tjd� tj tjd� tdtjd� � w )Nz-----------------------r+ zcannot unpickle packet:)
r� r� r ZUnpicklingErrorr0 r. r/ r4 r5 � print_stack)r r� r� r� r r r � pollmessage s
��zSocketIO.pollmessagec
C sR zt �d�}W n
tjy Y nw |\}}|d|ff}| �|� z| �|�}|d u r0W d S W n ty? | �� Y d S tyH Y d S w |\}}|d }| � d||f � |dv r�| � d| � | �
||�}| � d||f � |dkr�| �||f� n|d kr� q||kr�|S | j �|d �} | d ur�| � � || j
|< | �� | �� q)
NTr rU zpollresponse:%d:myseq:%s)rW rX z pollresponse:%d:localcall:call:z%pollresponse:%d:localcall:response:%srW rX )�response_queue�get�queueZEmptyr| r� r� �
handle_EOFr� rL rp rB r� rA �notifyr� )
r r� r� Zqmsgrh r� r� Zresqri �cvr r r r� � sX �
� � �
zSocketIO.pollresponsec C sX | � � | �d� | jD ]}| j| }|�� d| j|< |�� |�� q
| �� d S )Nr� )r� N)�EOFhookrL rB r� rA r� r� rE )r �keyr� r r r r� � s
zSocketIO.handle_EOFc C r" r r r# r r r r� � r% zSocketIO.EOFhook)NN)"r r r r� r rC rE rL rO rR rp ru rx rr rw rs r� r� r� r� r� r� r{ r| r� r� r� r� r� r� r� r� r� r� r r r r r; � s@
.
I
r; c @ s e Zd ZdS )rb N)r r r r r r r rb � s rb c C s t | �}| t|< t|�S r )�idr? r� )r rM r r r rc � s rc c @ s e Zd Zdd� ZdS )r� c C s
|| _ d S r )rM rQ r r r r � �
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 r )Zcurrent_handlerr; r r �BaseRequestHandler)r r> r! Zsvrr r r r � s zRPCHandler.__init__c C s | � � d S r )r� r# r r r �handle r* zRPCHandler.handlec C �
t | |�S r �r� rQ r r r �get_remote_proxy r� zRPCHandler.get_remote_proxyN)r r r r= rG r r� r� r r r r r � s r c @ s: e Zd ZdZdZdZejejfdd�Z dd� Z
dd � Z d
S ) � RPCClientFz#Cr� c C s* t � ||�| _| j�|� | j�d� d S �Nr� )r& �listening_sockZbindZlisten)r �addressZfamily�typer r r r s zRPCClient.__init__c C sT | j �� \}}| jrtd|tjd� |d tkr t�| |� d S td|tjd� t �)Nz ****** Connection request from r+ r z** Invalid host: )
r� �acceptr= r0 r. r/ � LOCALHOSTr; r rf )r Z working_sockr� r r r r� s zRPCClient.acceptc C r� r r� rQ r r r r� " r� 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 r )�sockiorM )r r� rM r r r r + s
zRPCProxy.__init__c C sl | j d u r | �� | j �|�rt| j| j|�S | jd u r | �� || jv r2| j�| jd|fi �}|S t |��)N�__getattribute__)
�_RPCProxy__methods�_RPCProxy__getmethodsr� � MethodProxyr� rM �_RPCProxy__attributes�_RPCProxy__getattributesru r� )r r3 �valuer r r � __getattr__/ s
�zRPCProxy.__getattr__c C � | j �| jddi �| _d S )NrV r )r� ru rM r� r# r r r Z__getattributes= �
�zRPCProxy.__getattributesc C r� )NrT r )r� ru rM r� r# r r r Z __getmethodsA r� zRPCProxy.__getmethods) r r r r� r� r r� r� r� r r r r r� & s r� c C sP t | �D ]}t| |�}t|�rd||< qt| t�r$| jD ] }t||� q d S d S r� )�dirr` �callablera r� � __bases__r] )r rl r3 �attr�superr r r r] E s
�
�r] c C s, t | �D ]}t| |�}t|�sd||< qd S r� )r� r` r� )r rm r3 r� r r r r^ P s
��r^ c @ s e Zd Zdd� Zdd� ZdS )r� c C s || _ || _|| _d S r )r� rM r3 )r r� rM r3 r r r r Y s
zMethodProxy.__init__c O s | j �| j| j||�}|S r )r� ru rM r3 )r rI rk r� r r r �__call__^ s zMethodProxy.__call__N)r r r r r� r r r r r� W s r� c C s~ | d u rd S d t _t| �}ztj�|� W n ty3 d}|�|d�}|�|d�}tj�|� Y nw tj�d� | t _d S )N�ascii�backslashreplace�strict�
) �builtins�_r4 r. �stdout�write�UnicodeEncodeError�encode�decode)r� �text�encoding�bytesr r r � displayhookf s �
r� �__main__)�mainzidlelib.idle_test.test_rpcr� )� verbosityr )+r� r r r r7 r� r� r� r& r r� r. r1 r5 r r r r
ZPicklerr r� r� r r r? ZQueuerd r� r; rb rc r� r� r r� r� r] r^ r� r� r Zunittestr� r r r r �<module> sV
2
k �