File: //usr/lib/python3/dist-packages/twisted/protocols/__pycache__/loopback.cpython-310.pyc
o
�b�. � @ s$ d Z ddlZddlmZ ddlmZmZmZmZ ddl m
Z
ddl m Z ddl
mZ ddlmZ G d d
� d
�Zee
�G d d � d ��Zeejej�G d
d� d��Zdd� Zdd� Zefdd�Zdd� Zdd� Zeejej�G dd� d��ZG dd � d ej �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.
� N)� implementer)�defer�
interfaces�main�protocol)�IAddress)�
deferLater)�policies)�failurec @ 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__" �
z_LoopbackQueue.__init__c C s6 | j �|� | jd ur| jd }| _|�d � d S 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__+ r z_LoopbackQueue.__nonzero__c C s | j �d�S �Nr )r
�popr r r r �get0 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% 4 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 �
|| _ d S r )�q)r r( r r r r ? r z_LoopbackTransport.__init__c C s" t |t�s td��| j�|� d S )Nz"Can only write bytes to ITransport)�
isinstance�bytes� TypeErrorr( r �r �datar r r �writeB s
z_LoopbackTransport.writec C s | j �d�|�� d S �N� )r( r �join�r �iovecr r r �
writeSequenceG s z _LoopbackTransport.writeSequencec C s d| j _| j �d � d S )NT)r( r# r r r r r �loseConnectionJ s z!_LoopbackTransport.loseConnectionc C s | � � dS )zB
Abort the connection. Same as L{loseConnection}.
N)r5 r r r r �abortConnectionN s z"_LoopbackTransport.abortConnectionc C � t � S r �r% r r r r �getPeerT � z_LoopbackTransport.getPeerc C r7 r r8 r r r r �getHostW r: z_LoopbackTransport.getHostc C s& | j d u sJ �|| _ || _| �� d S r )�producer�streamingProducer�
_pollProducer�r r< � streamingr r r �registerProducer[ s z#_LoopbackTransport.registerProducerc C s | j d usJ �d | _ d S r �r< r r r r �unregisterProducera s
z%_LoopbackTransport.unregisterProducerc C s& | j d ur| js| j �� d S d S d S r )r< r= �resumeProducingr r r r r> e s �z _LoopbackTransport._pollProducer)r r r! �
disconnectingr< r r. r4 r5 r6 r9 r; rA rC r> r r r r r& 9 s r& c C s. | r| � � }|du r dS |�|� | sdS 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 �identityPumpPolicyj s
�rI c C sD g }| r| � � }|du r
n|�|� | s|r |�d�|�� dS 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 rF r1 )rG rH r* �chunkr r r �collapsingPumpPolicyz s
��rK 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 �makeConnectionr& �_loopbackAsyncBody)�server�client�
pumpPolicy�serverToClient�clientToServerr r r �
loopbackAsync� s
�rS c
s� � fdd�} d } }}|| ||�}|||| �}|s3|s3t �� } | |_| |_| �t| |||� � | S |jr?d}|| ||� n |jrJd}|||| � |rc| �t�t j
�� |�t�t j
�� t � d�S q)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)� transportr> )�sourcer( rH �sent�rP r r �pump� s
z _loopbackAsyncBody.<locals>.pump� FTN) r �Deferredr � addCallback�_loopbackAsyncContinuer# �connectionLostr
�Failurer �CONNECTION_DONE�succeed)
rN rQ rO rR rP rX r# �
clientSent�
serverSentr r rW r rM � s: �
�rM c C s. d |_ d |_ ddlm} t|dt|||||�S )Nr ��reactor)r �twisted.internetrd r rM )�ignoredrN rQ rO rR rP rd r r r r\ � s �r\ 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 )rH �logFile)r rH rh r r r r s
zLoopbackRelay.__init__c C s. | j | | _ | jr| j�dt|� � d S d S )Nzloopback writing %s
)�bufferrh r. �reprr, r r r r. s �zLoopbackRelay.writec C s | � d�|�� d S r/ )r. r1 r2 r r r r4 " s zLoopbackRelay.writeSequencec C s� | j dkrd S | jr| j�� | jr,| jr | j�dt| j� � | j}d| _| j�|� | j dkr@d| _ | j� t
� t j
�� d S d S )N���zloopback receiving %s
r0 rY )�
shouldLoser< rD ri rh r. rj rH rF r] r
r^ r r_ )r ri r r r � clearBuffer% s
�zLoopbackRelay.clearBufferc C s | j dkr
d| _ d S d S )Nrk rY )rl r r r r r5 5 s
�z LoopbackRelay.loseConnectionc C � dS �N�loopbackr r r r r r; 9 � zLoopbackRelay.getHostc C rn ro r r r r r r9 < rq zLoopbackRelay.getPeerc C r' r rB r? r r r rA ? r z LoopbackRelay.registerProducerc C s
d | _ d S r rB r r r r rC B r z LoopbackRelay.unregisterProducerc C s d| j jj�d�S )Nz Loopback(�))rH � __class__r r r r r � logPrefixE s zLoopbackRelay.logPrefixr )r r r! ri rl rE r< r r. r4 rm r5 r; r9 rA rC rt r r r r rg s
rg c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�LoopbackClientFactoryc C s d| _ t�� | _|| _d S r )� disconnectedr rZ �deferredr )r r r r r r J s
z LoopbackClientFactory.__init__c C s | j S r )r )r �addrr r r �
buildProtocolO r: z#LoopbackClientFactory.buildProtocolc C s d| _ | j�d � d S )NrY )rv rw r )r � connector�reasonr r r �clientConnectionLostR s z*LoopbackClientFactory.clientConnectionLostN)r r r! r ry r| r r r r ru I s ru c @ s e Zd Zdd� Zdd� ZdS )� _FireOnClosec C s t j�| ||� t�� | _d S r )r �ProtocolWrapperr r rZ rw )r r �factoryr r r r X s z_FireOnClose.__init__c C s t j�| |� | j�d � d S r )r r~ r] rw r )r r{ r r r r] \ s z_FireOnClose.connectionLostN)r r r! r r] r r r r r} W s r} 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 rc c � � S r r �rx ��
serverWrapperr r �<lambda>h � z loopbackTCP.<locals>.<lambda>z 127.0.0.1)� interfacec � � j S r �rw ��xr� r r r� n � c � � � � S r ��
stopListeningr� ��
serverPortr r r� o � )re rd r �WrappingFactoryr �Factoryr} �noisyry � listenTCPru �
connectTCPr; �portrw r[ )rN rO r� r� rd �f�clientFr r �r� r� 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 rc c r� r r r� r� r r r� { r� z loopbackUNIX.<locals>.<lambda>c r� r r� r� r� r r r� � r� c r� r r� r� r� r r r� � r� )�tempfile�mktempre rd r r� r r� r} r� ry �
listenUNIXru � connectUNIXrw r[ )rN rO r� �pathrd r� r� r r r� r � loopbackUNIXs s
r� )r T)T)"r" r� �zope.interfacer re r r r r �twisted.internet.interfacesr �twisted.internet.taskr �twisted.protocolsr �twisted.pythonr
r r% �
ITransport� IConsumerr& rI rK rS rM r\ rg �
ClientFactoryru r~ r} r� r� r r r r �<module> s0 0 &G 6