← Back
Editing: rpc.cpython-36.pyc
3 � f�R � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZdd� Zdd� Zd+dd�ZG d d � d ej�Zd,Zd ZG dd� de j�Zi Zejd�Zejd�Z G dd� de �Z G dd� de �Z dd� Z G dd� de �Z!G dd� de j"e �Z#G dd� de �Z$G d d � d e �Z%d d � Z&d d!� Z'G d"d#� d#e �Z(d$d%� Z)e*d&k�r�dd'l+m,Z, e,d(d)d*� dS )-a RPC 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. � Nc C s t j| �}t|tj�st�|S )z*Return code object from marshal string ms.)�marshal�loads� isinstance�types�CodeType�AssertionError)�ms�co� r �0/opt/alt/python36/lib64/python3.6/idlelib/rpc.py� unpickle_code- s r c C s$ t | tj�st�tj| �}t|ffS )zBReturn unpickle function and tuple with marshalled co code object.)r r r r r �dumpsr )r r r r r � pickle_code3 s r c C s$ t j� }t||�}|j| � |j� S )z.Return pickled (or marshalled) string for obj.)�io�BytesIO� CodePickler�dump�getvalue)�objZprotocol�f�pr r r r 9 s r c @ s" e Zd ZejeiZejej� dS )r N) �__name__� __module__� __qualname__r r r �dispatch_table�update�copyregr 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 kr t }tjj| ||� d S )N)� RPCHandler� socketserver� TCPServer�__init__)�self�addrZ handlerclassr r r r" L s zRPCServer.__init__c C s dS )z@Override TCPServer method, no bind() phase for connecting entityNr )r# r r r � server_bindQ s zRPCServer.server_bindc C s | j j| j� dS )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# r r r �server_activateU s zRPCServer.server_activatec C s | j | jfS )z:Override TCPServer method, return already connected socket)r& r' )r# r r r � get_request^ s zRPCServer.get_requestc C s� y� W n� t k r � Y n� tj}tddd |d� td|d� tdtj� j |d� td||d� tdt|�|d� tj |d� td |d� tdd |d� t j d � Y nX d S ) 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_addressZerfr r r � handle_errorb s zRPCServer.handle_error)N)r r r r" r% r( r) r; r r r r r J 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 j� | _|d k r|| _|| _|d kr*t}|| _i | _i | _d S )N) r2 r3 � sockthread� debugging�sock� objecttable�objtable� responses�cvars)r# r? rA r> r r r r" � s zSocketIO.__init__c C s | j }d | _ |d k r |j� d S )N)r? �close)r# r? r r r rD � s zSocketIO.closec C s t jd� dS )z!override for specific exit actionr N)r8 r9 )r# r r r �exithook� s zSocketIO.exithookc G sR | j s d S | jd ttj� j� }x |D ]}|d t|� }q(W t|tjd� d S )N� )r- ) r> �location�strr2 r3 r4 r1 r/ r0 )r# �args�s�ar r r �debug� s zSocketIO.debugc C s || j |< d S )N)rA )r# �oid�objectr r r �register� s zSocketIO.registerc C s& y | j |= W n tk r Y nX d S )N)rA �KeyError)r# rM r r r � unregister� s zSocketIO.unregisterc C s� | j d|� y|\}\}}}}W n tk r4 dS X || jkrNdd|f fS | j| }|dkrvi } t|| � d| fS |dkr�i } t|| � d| fS t||�s�dd|f fS t||�} y^|d kr�| ||�} t| t�r�t | �} d| fS |d k�rt j || ||ff� dS dd | fS W n� t k �r, � Y n� t k �rB � Y nn tk �rX � Y nX tk �r| } zd | fS d } ~ X n4 d}t||| |f tjd� tjtjd� dS d S )Nz localcall:�ERROR�Bad request formatzUnknown object id: %r� __methods__�OK�__attributes__zUnsupported method name: %r�CALL�QUEUE�QUEUEDz Unsupported message type: %s�CALLEXCzU*** Internal Error: rpc.py:SocketIO.localcall() Object: %s Method: %s Args: %s )r- � EXCEPTION)rR rS )rY N)r[ N)rL � TypeErrorrA � _getmethods�_getattributes�hasattr�getattrr � RemoteObject� remoteref� request_queueZputr. �KeyboardInterrupt�OSError� Exceptionr1 r/ r0 r6 r7 )r# �seqr: �howrM � methodnamerI �kwargsr �methods� attributes�methodZretZex�msgr r r � localcall� sT zSocketIO.localcallc C s( | j d||� | j||||�}| j|�S )Nzremotecall:asynccall: )rL � asynccall� asyncreturn)r# rM ri rI rj rg r r r � remotecall� s zSocketIO.remotecallc C s( | j d||� | j||||�}| j|�S )Nzremotequeue:asyncqueue: )rL � asyncqueuerq )r# rM ri rI rj rg r r r � remotequeue� s zSocketIO.remotequeuec C s` d||||ff}| j � }tj� | jkr8tj� }|| j|< | jd| ||||� | j||f� |S )NrW z asynccall:%d:)�newseqr2 r3 r= � ConditionrC rL � putmessage)r# rM ri rI rj r: rg �cvarr r r rp � s zSocketIO.asynccallc C s` d||||ff}| j � }tj� | jkr8tj� }|| j|< | jd| ||||� | j||f� |S )NrX zasyncqueue:%d:)ru r2 r3 r= rv rC rL rw )r# rM ri rI rj r: rg rx r r r rs � s zSocketIO.asyncqueuec C s6 | j d| � | j|dd�}| j d| |� | j|�S )Nz#asyncreturn:%d:call getresponse(): g�������?)�waitzasyncreturn:%d:response: )rL � getresponse�decoderesponse)r# rg �responser r r rq � s zSocketIO.asyncreturnc C s� |\}}|dkr|S |dkr d S |dkr6| j d� d S |dkrT| j d� | j� d S |dkrp| j d|� t|��|d kr�| j d |� |�t||��d S ) NrU rY r[ zdecoderesponse: EXCEPTION�EOFzdecoderesponse: EOFrR z decoderesponse: Internal ERROR:rZ z decoderesponse: Call Exception:)rL �decode_interrupthook� RuntimeError� SystemError)r# r| rh �whatr r r r{ � s& zSocketIO.decoderesponsec C s t �dS )� N)�EOFError)r# r r r r~ s z SocketIO.decode_interrupthookc C s6 y| j ddd� W n tk r0 | jd� dS X dS )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. Ng�������?)�myseqry zmainloop:return)rz r� rL )r# r r r �mainloop s zSocketIO.mainloopc C s6 | j ||�}|d k r2|\}}|dkr2|| j|�f}|S )NrU )� _getresponse�_proxify)r# r� ry r| rh r� r r r rz # s zSocketIO.getresponsec C s4 t |t�rt| |j�S t |t�r0tt| j|��S |S )N)r � RemoteProxy�RPCProxyrM �list�mapr� )r# r r r r r� + s zSocketIO._proxifyc C s� | j d|� tj� | jkr:x�| j||�}|d k r |S q W nb| j| }|j� x|| jkrb|j� qNW | j| }| j d||f � | j|= | j|= |j � |S d S )Nz_getresponse:myseq:z-_getresponse:%s: thread woke up: response: %s) rL r2 r3 r= � pollresponserC �acquirerB ry �release)r# r� ry r| rx r r r r� 3 s" zSocketIO._getresponsec C s | j d | _ }|S )N� )�nextseq)r# rg r r r ru I s zSocketIO.newseqc C s� | j d|d � y t|�}W n, tjk rJ tdt|�tjd� � Y nX tj dt |��| }xnt |�dkr�y0t j g | j gg �\}}}| j j |d t� �}W n ttfk r� td��Y nX ||d � }qbW d S )Nzputmessage:%d:r zCannot pickle:)r- z<izsocket no longer exists)rL r �pickleZ PicklingErrorr1 r5 r/ r0 �structZpack�len�selectr? �send�BUFSIZE�AttributeErrorr\ re )r# �messagerJ �r�w�x�nr r r rw M s zSocketIO.putmessage� � c C s� | j � t| j�| jk r�tj| jj� gg g |�\}}}t|�dkrFd S y| jjt�}W n t k rn t �Y nX t|�dkr�t �| j|7 _| j � | j � S )Nr ) �_stage0r� �buff�bufneedr� r? �filenoZrecvr� re r� �_stage1)r# ry r� r� r� rJ r r r � pollpacketa s zSocketIO.pollpacketc C sR | j dkrNt| j�dkrN| jd d� }| jdd � | _tjd|�d | _d| _ d S )Nr r� z<i� )�bufstater� r� r� Zunpackr� )r# rJ r r r r� q s zSocketIO._stage0c C sP | j dkrLt| j�| jkrL| jd | j� }| j| jd � | _d| _d| _ |S d S )Nr� r� r )r� r� r� r� )r# �packetr r r r� x s zSocketIO._stage1c C s� | j |�}|d krd S ytj|�}W nV tjk rz tdtjd� tdt|�tjd� tj tjd� tdtjd� � Y nX |S )Nz-----------------------)r- zcannot unpickle packet:) r� r� r ZUnpicklingErrorr1 r/ r0 r5 r6 � print_stack)r# ry r� r� r r r � pollmessage� s zSocketIO.pollmessagec C s` �xXyt jd�}W n tjk r( Y n X |\}}|d|ff}| j|� y| j|�}|dkr`dS W n. tk r~ | j� dS tk r� dS X |\}}|d }| j d||f � |d k�r| j d| � | j ||�}| j d||f � |dk�r| j||f� q|dkrqq||k�r |S | j j|d�} | dk r| j � || j |< | j� | j� qqW dS ) aR Handle 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 rU Nzpollresponse:%d:myseq:%srW rX z pollresponse:%d:localcall:call:z%pollresponse:%d:localcall:response:%s)rW rX )�response_queue�get�queueZEmptyrw r� r� � handle_EOFr� rL ro rC r� rB �notifyr� ) r# r� ry Zqmsgrg r| r� Zresqrh �cvr r r r� � sL zSocketIO.pollresponsec C sZ | j � | jd� x:| jD ]0}| j| }|j� d| j|< |j� |j� qW | j� dS )z+action taken upon link being closed by peerr� r} N)r} N)�EOFhookrL rC r� rB r� r� rE )r# �keyr� r r r r� � s zSocketIO.handle_EOFc C s dS )zBClasses using rpc client/server can override to augment EOF actionNr )r# r r r r� � s zSocketIO.EOFhook)NN)"r r r r� r" rD rE rL rO rQ ro rr rt rp rs rq r{ r~ r� rz r� r� ru rw r� r� r� r� r� r� r� r� r� r� r r r r r< � s> . I r< c @ s e Zd ZdS )ra N)r r r r r r r ra � s ra c C s t | �}| t|< t|�S )N)�idr@ r� )r rM r r r rb � s rb c @ s e Zd Zdd� ZdS )r� c C s || _ d S )N)rM )r# rM 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( | |_ tj| |� tjj| |||� d S )N)Zcurrent_handlerr< r" r �BaseRequestHandler)r# r? r$ Zsvrr r r r" s zRPCHandler.__init__c C s | j � dS )z(handle() method required by socketserverN)r� )r# r r r �handle s zRPCHandler.handlec C s t | |�S )N)r� )r# rM r r r �get_remote_proxy s 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 ||�| _| jj|� | jjd� d S )Nr� )r& �listening_sockZbindZlisten)r# �addressZfamily�typer r r r" s zRPCClient.__init__c C sV | j j� \}}| jr$td|tjd� |d tkr>tj| |� ntd|tjd� t �d S )Nz ****** Connection request from )r- r z** Invalid host: ) r� �acceptr> r1 r/ r0 � LOCALHOSTr< r" re )r# Z working_sockr� r r r r� s zRPCClient.acceptc C s t | |�S )N)r� )r# rM r r r r� # s 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 )N)�sockiorM )r# r� rM r r r r" , s zRPCProxy.__init__c C sp | j d kr| j� | j j|�r.t| j| j|�S | jd kr@| j� || jkrd| jj| jd|fi �}|S t |��d S )N�__getattribute__) �_RPCProxy__methods�_RPCProxy__getmethodsr� � MethodProxyr� rM �_RPCProxy__attributes�_RPCProxy__getattributesrr r� )r# r4 �valuer r r � __getattr__0 s zRPCProxy.__getattr__c C s | j j| jdf i �| _d S )NrV )r� rr rM r� )r# r r r Z__getattributes> s zRPCProxy.__getattributesc C s | j j| jdf i �| _d S )NrT )r� rr rM 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| t�rPx| jD ]}t||� q>W d S )Nr� )�dirr` �callabler r� � __bases__r] )r rk r4 �attr�superr r r r] F s r] c C s0 x*t | �D ] }t| |�}t|�s d||< q W d S )Nr� )r� r` r� )r rl r4 r� r r r r^ Q s r^ c @ s e Zd Zdd� Zdd� ZdS )r� c C s || _ || _|| _d S )N)r� rM r4 )r# r� rM r4 r r r r" Z s zMethodProxy.__init__c O s | j j| j| j||�}|S )N)r� rr rM r4 )r# rI rj 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~ | dkr dS dt _t| �}ytjj|� W n< tk rf d}|j|d�}|j|d�}tjj|� Y nX tjjd� | t _dS )z9Override standard display hook to use non-locale encodingN�ascii�backslashreplace�strictr* ) �builtins�_r5 r/ �stdout�write�UnicodeEncodeError�encode�decode)r� �text�encoding�bytesr r r � displayhookg s r� �__main__)�mainzidlelib.idle_test.test_rpcr� )� verbosity)Ni )-�__doc__r� r r r r8 r� r� r� r& r r� r/ r2 r6 r r r r ZPicklerr r� r� r! r r@ ZQueuerc r� rN r< ra rb r� r� r r� r� r] r^ r� r� r Zunittestr� r r r r �<module> sR 2 k
Save File
Cancel