← Back
Editing: loopback.cpython-38.pyc
U W[�. � @ s: d Z ddlmZmZ ddlZddlmZ ddlmZ ddl m Z m Z m Z m Z ddlmZ ddlmZ dd lmZ G d d � d e�Zee�G d d � d e��Zee je j�G dd� de��Zdd� Zdd� Zefdd�Z dd� Z dd� Z ee je j�G dd� d��Z G d d � d e j �Z!G d d � d ej"�Z#d%d!d"�Z$d&d#d$�Z%dS )'zF Testing support for protocols -- loopback between client and server. � )�division�absolute_importN)� implementer)�policies)� interfaces�protocol�main�defer)� deferLater)�failure)�IAddressc @ s< e Zd ZdZdZdZdd� Zdd� Zdd � ZeZ d d � Z dS ) �_LoopbackQueuez� Trivial wrapper around a list to give it an interface like a queue, which the addition of also sending notifications by way of a Deferred whenever the list has something added to it. NFc C s g | _ d S �N)�_queue��self� r �</usr/lib/python3/dist-packages/twisted/protocols/loopback.py�__init__" s z_LoopbackQueue.__init__c C s2 | j �|� | jd k r.| jd }| _|�d � d S r )r �append�_notificationDeferred�callback)r �v�dr r r �put& s z_LoopbackQueue.putc C s t | j�S r )�boolr r r r r � __nonzero__- s z_LoopbackQueue.__nonzero__c C s | j �d�S �Nr )r �popr r r r �get2 s z_LoopbackQueue.get) �__name__� __module__� __qualname__�__doc__r � disconnectr r r �__bool__r r r r r r s r c @ s e Zd ZdS )�_LoopbackAddressN)r r! r" r r r r r& 7 s r& c @ sd e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� ZdS )�_LoopbackTransportFNc C s || _ d S r )�q)r r( r r r r C s z_LoopbackTransport.__init__c C s" t |t�std��| j�|� d S )Nz"Can only write bytes to ITransport)� isinstance�bytes� TypeErrorr( r �r �datar r r �writeF s z_LoopbackTransport.writec C s | j �d�|�� d S �N� )r( r �join�r Ziovecr r r � writeSequenceK s z _LoopbackTransport.writeSequencec C s d| j _| j �d � d S )NT)r( r$ r r r r r �loseConnectionN s z!_LoopbackTransport.loseConnectionc C s | � � dS )zB Abort the connection. Same as L{loseConnection}. N)r4 r r r r �abortConnectionS s z"_LoopbackTransport.abortConnectionc C s t � S r �r& r r r r �getPeerZ s z_LoopbackTransport.getPeerc C s t � S r r6 r r r r �getHost] s z_LoopbackTransport.getHostc C s& | j d kst�|| _ || _| �� d S r )�producer�AssertionError�streamingProducer� _pollProducer�r r9 Z streamingr r r �registerProducera s z#_LoopbackTransport.registerProducerc C s | j d k st�d | _ d S r )r9 r: r r r r �unregisterProducerg s z%_LoopbackTransport.unregisterProducerc C s | j d k r| js| j �� d S r )r9 r; �resumeProducingr r r r r< k s z _LoopbackTransport._pollProducer)r r! r" � disconnectingr9 r r. r3 r4 r5 r7 r8 r> r? r< r r r r r' = s r' c C s& | r"| � � }|dkrq"|�|� q dS )z� L{identityPumpPolicy} is a policy which delivers each chunk of data written to the given queue as-is to the target. This isn't a particularly realistic policy. @see: L{loopbackAsync} N)r � dataReceived)�queue�targetr* r r r �identityPumpPolicyq s rE c C s> g }| r&| � � }|dkrq&|�|� q|r:|�d�|�� dS )z� L{collapsingPumpPolicy} is a policy which collapses all outstanding chunks into a single string and delivers it to the target. @see: L{loopbackAsync} Nr0 )r r rB r1 )rC rD r* �chunkr r r �collapsingPumpPolicy� s rG c C s8 t � }t � }| �t|�� |�t|�� t| ||||�S )a Establish a connection between C{server} and C{client} then transfer data between them until the connection is closed. This is often useful for testing a protocol. @param server: The protocol instance representing the server-side of this connection. @param client: The protocol instance representing the client-side of this connection. @param pumpPolicy: When either C{server} or C{client} writes to its transport, the string passed in is added to a queue of data for the other protocol. Eventually, C{pumpPolicy} will be called with one such queue and the corresponding protocol object. The pump policy callable is responsible for emptying the queue and passing the strings it contains to the given protocol's C{dataReceived} method. The signature of C{pumpPolicy} is C{(queue, protocol)}. C{queue} is an object with a C{get} method which will return the next string written to the transport, or L{None} if the transport has been disconnected, and which evaluates to C{True} if and only if there are more items to be retrieved via C{get}. @return: A L{Deferred} which fires when the connection has been closed and both sides have received notification of this. )r ZmakeConnectionr' �_loopbackAsyncBody)�server�client� pumpPolicy�serverToClient�clientToServerr r r � loopbackAsync� s �rN c s� � fdd�}d } }}|| ||�}|||| �}|sd|sdt �� } | |_| |_| �t| |||� � | S |jr|d}|| ||� n|jr�d}|||| � |r | �t�t j �� |�t�t j �� t � d�S q dS )ak Transfer bytes from the output queue of each protocol to the input of the other. @param server: The protocol instance representing the server-side of this connection. @param serverToClient: The L{_LoopbackQueue} holding the server's output. @param client: The protocol instance representing the client-side of this connection. @param clientToServer: The L{_LoopbackQueue} holding the client's output. @param pumpPolicy: See L{loopbackAsync}. @return: A L{Deferred} which fires when the connection has been closed and both sides have received notification of this. c s, d}|r� ||� d}|r(|s(| j �� |S )NFT)Z transportr< )�sourcer( rD Zsent�rK r r �pump� s z _loopbackAsyncBody.<locals>.pumpFTN) r �Deferredr � addCallback�_loopbackAsyncContinuer$ �connectionLostr �Failurer �CONNECTION_DONEZsucceed) rI rL rJ rM rK rQ r$ Z clientSentZ serverSentr r rP r rH � s6 � rH c C s. d |_ d |_ ddlm} t|dt|||||�S )Nr ��reactor)r �twisted.internetrY r rH )ZignoredrI rL rJ rM rK rY r r r rT � s �rT c @ sn e Zd ZdZdZdZdZddd�Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� Zdd� Zdd� ZdS )� LoopbackRelayr0 r Nc C s || _ || _d S r )rD �logFile)r rD r\ r r r r s zLoopbackRelay.__init__c C s* | j | | _ | jr&| j�dt|� � d S )Nzloopback writing %s )�bufferr\ r. �reprr, r r r r. s zLoopbackRelay.writec C s | � d�|�� d S r/ )r. r1 r2 r r r r3 s zLoopbackRelay.writeSequencec C s� | j dkrd S | jr | j�� | jrX| jr@| j�dt| j� � | j}d| _| j�|� | j dkr|d| _ | j� t � t j �� d S )N���zloopback receiving %s r0 � )� shouldLoser9 r@ r] r\ r. r^ rD rB rU r rV r rW )r r] r r r � clearBuffer# s zLoopbackRelay.clearBufferc C s | j dkrd| _ d S )Nr_ r` )ra r r r r r4 3 s z LoopbackRelay.loseConnectionc C s dS �NZloopbackr r r r r r8 7 s zLoopbackRelay.getHostc C s dS rc r r r r r r7 : s zLoopbackRelay.getPeerc C s || _ d S r �r9 r= r r r r> = s z LoopbackRelay.registerProducerc C s d | _ d S r rd r r r r r? @ s z LoopbackRelay.unregisterProducerc C s d| j jjf S )Nz Loopback(%r))rD � __class__r r r r r � logPrefixC s zLoopbackRelay.logPrefix)N)r r! r" r] ra rA r9 r r. r3 rb r4 r8 r7 r> r? rf r r r r r[ s r[ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�LoopbackClientFactoryc C s d| _ t�� | _|| _d S r )� disconnectedr rR �deferredr )r r r r r r J s z LoopbackClientFactory.__init__c C s | j S r )r )r �addrr r r � buildProtocolO s z#LoopbackClientFactory.buildProtocolc C s d| _ | j�d � d S )Nr` )rh ri r )r Z connector�reasonr r r �clientConnectionLostR s z*LoopbackClientFactory.clientConnectionLostN)r r! r" r rk rm r r r r rg H s rg c @ s e Zd Zdd� Zdd� ZdS )� _FireOnClosec C s t j�| ||� t�� | _d S r )r �ProtocolWrapperr r rR ri )r r �factoryr r r r X s z_FireOnClose.__init__c C s t j�| |� | j�d � d S r )r ro rU ri r )r rl r r r rU \ s z_FireOnClose.connectionLostN)r r! r" r rU r r r r rn W s rn Tc s� ddl m} t�t�� �}t|| ��||_�fdd�|_|j ||dd�� t |�}||_|� d� � � j |� |j}|��fdd�� |�� fdd�� |S ) zBRun session between server and client protocol instances over TCP.r rX c s � S r r �rj �� serverWrapperr r �<lambda>g r0 z loopbackTCP.<locals>.<lambda>z 127.0.0.1)Z interfacec s � j S r �ri ��xrr r r rt m r0 c s � � � S r �Z stopListeningrv �� serverPortr r rt n r0 )rZ rY r �WrappingFactoryr �Factoryrn �noisyrk Z listenTCPrg Z connectTCPr8 �portri rS )rI rJ r~ r} rY �f�clientFr r �rz rs r � loopbackTCPa s r� c s� t �� }ddlm} t�t�� �}t|| ��||_ �fdd�|_ |� ||�� t |�}||_ |� ||� |j}|��fdd�� |�� fdd�� |S )zJRun session between server and client protocol instances over UNIX socket.r rX c s � S r r rq rr r r rt y r0 z loopbackUNIX.<locals>.<lambda>c s � j S r ru rv rr r r rt r0 c s � � � S r rx rv ry r r rt � r0 )�tempfileZmktemprZ rY r r{ r r| rn r} rk Z listenUNIXrg Z connectUNIXri rS )rI rJ r} �pathrY r r� r r r� r � loopbackUNIXr s r� )r T)T)&r# Z __future__r r r� Zzope.interfacer Ztwisted.protocolsr rZ r r r r Ztwisted.internet.taskr Ztwisted.pythonr Ztwisted.internet.interfacesr �objectr r& Z ITransportZ IConsumerr' rE rG rN rH rT r[ Z ClientFactoryrg ro rn r� r� r r r r �<module> s0 3 &C 7
Save File
Cancel