/usr/lib/python3/dist-packages/twisted/protocols/__pycache__
Edit: /usr/lib/python3/dist-packages/twisted/protocols/__pycache__/loopback.cpython-310.pyc (12680B)
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 ZeejejG d
d dZdd Zdd ZefddZdd Zdd ZeejejG dd dZG 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
interfacesmainprotocol)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)_queueself 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)queuetargetr* 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)serverclient
pumpPolicyserverToClientclientToServerr r r
loopbackAsync s
rS c
s fdd} d } }}|| ||}|||| }|s3|s3t } | |_| |_| t| ||| | S |jr?d}|| || n |jrJd}|||| |rc| tt j
|tt j
t dS 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 sentrP r r pump s
z _loopbackAsyncBody.
.pump FTN) r Deferredr addCallback_loopbackAsyncContinuer# connectionLostr
Failurer CONNECTION_DONEsucceed)
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 ZdZdZdZdZdddZdd 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| jdt| 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 | jdt| j | j}d| _| j| | j dkr@d| _ | j t
t j
d S d S )Nzloopback 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 Nloopbackr 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 jjdS )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| _ | jd d S )NrY )rv rw r )r connectorreasonr 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| | | jd 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} tt }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 h z loopbackTCP..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 fclientFr r r r r loopbackTCPa s
r c s t }ddlm} tt }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..c r r r r r r r r r c r r r r r r r r r )tempfilemktempre 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 s0 0 &G 6