HEX

File: //usr/lib/python3/dist-packages/twisted/protocols/test/__pycache__/test_tls.cpython-310.pyc
o

�bk�@sfdZddlZddlmZmZmZddlmZddlm	Z	z&ddl
mZddlm
Z
mZmZmZmZmZmZddlmZmZmZmZWneySd	ZdZZZYnwdd
lmZmZddlmZm Z m!Z!ddl"m#Z#dd
l$m%Z%m&Z&ddl'm(Z(m)Z)ddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5m6Z6ddl7m8Z8ddl9m:Z:m;Z;ddl<m=Z=ddl>m?Z?ddl@mAZAddlBmCZCddlDmEZEmFZFddlGmHZHddlImJZJmKZKGdd�d�ZLGdd�de5�ZM	d1dd �ZNGd!d"�d"eK�ZO	d2d#d$�ZPGd%d&�d&eJ�ZQGd'd(�d(eK�ZRGd)d*�d*eK�ZSGd+d,�d,eK�ZTee.�Gd-d.�d.e3��ZUee.�Gd/d0�d0e6��ZVdS)3z%
Tests for L{twisted.protocols.tls}.
�N)�	Interface�directlyProvides�implementer)�verifyObject)�	iterbytes)�crypto)�
Connection�Context�Error�TLSv1_1_METHOD�TLSv1_2_METHOD�TLSv1_METHOD�
WantReadError)�TLSMemoryBIOFactory�TLSMemoryBIOProtocol�_ProducerMembrane�_PullToPushz;pyOpenSSL 16.0.0 or newer required for twisted.protocol.tls)�PrivateCertificate�optionsForClientTLS)�ClientTLSContext�ServerTLSContext�certPath)�!certificatesForAuthorityAndServer)�Deferred�
gatherResults)�ConnectionDone�ConnectionLost)�IHandshakeListener�IOpenSSLClientConnectionCreator�IOpenSSLServerConnectionCreator�IProtocolNegotiationFactory�
IPushProducer�
ISSLTransport�
ISystemHandle)�
ClientFactory�Factory�Protocol�
ServerFactory)�TaskStopped)�collapsingPumpPolicy�
loopbackAsync)�log)�Failure)�FilePath)�connectedServerAndClient)�NonStreamingProducer�StringTransport��ConnectionLostNotifyingProtocol)�SynchronousTestCase�TestCasec@s<eZdZdZdZefdd�Zedd��Zdd�Z	d	d
�Z
dS)�HandshakeCallbackContextFactorya
    L{HandshakeCallbackContextFactory} is a factory for SSL contexts which
    allows applications to get notification when the SSL handshake completes.

    @ivar _finished: A L{Deferred} which will be called back when the handshake
        is done.
    � cCst�|_||_dS�N)r�	_finished�_method)�self�method�r<�A/usr/lib/python3/dist-packages/twisted/protocols/test/test_tls.py�__init__Os
z(HandshakeCallbackContextFactory.__init__cCs|�}||jfS)z�
        Create a new L{HandshakeCallbackContextFactory} and return a two-tuple
        of it and a L{Deferred} which will fire when a connection created with
        it completes a TLS handshake.
        )r8)�cls�contextFactoryr<r<r=�factoryAndDeferredSs
z2HandshakeCallbackContextFactory.factoryAndDeferredcCs||j@r
|j�d�dSdS)a
        This is the "info callback" on the context.  It will be called
        periodically by pyOpenSSL with information about the state of a
        connection.  When it indicates the handshake is complete, it will fire
        C{self._finished}.
        N)�SSL_CB_HANDSHAKE_DONEr8�callback)r:�
connection�where�retr<r<r=�_info]s
�z%HandshakeCallbackContextFactory._infocCst|j�}|�|j�|S)zp
        Create and return an SSL context configured to use L{self._info} as the
        info callback.
        )r	r9�set_info_callbackrG)r:�contextr<r<r=�
getContextgs
z*HandshakeCallbackContextFactory.getContextN)�__name__�
__module__�__qualname__�__doc__rBr
r>�classmethodrArGrJr<r<r<r=r5Bs

	
r5c@s0eZdZdZdd�Zdd�Zdd�Zdd	�Zd
S)�AccumulatingProtocola.
    A protocol which collects the bytes it receives and closes its connection
    after receiving a certain minimum of data.

    @ivar howMany: The number of bytes of data to wait for before closing the
        connection.

    @ivar received: A L{list} of L{bytes} of the bytes received so far.
    cC�
||_dSr7)�howMany)r:rRr<r<r=r>|�
zAccumulatingProtocol.__init__cC�
g|_dSr7)�received�r:r<r<r=�connectionMaderSz#AccumulatingProtocol.connectionMadecCs4|j�|�ttt|j��|jkr|j��dSdSr7)rU�append�sum�map�lenrR�	transport�loseConnection�r:�datar<r<r=�dataReceived�s�z!AccumulatingProtocol.dataReceivedcCs|�t�st�|�dSdSr7)�checkrr+�err�r:�reasonr<r<r=�connectionLost�s
�z#AccumulatingProtocol.connectionLostN)rKrLrMrNr>rWr`rer<r<r<r=rPqs
rPFc	s�td��t�}�fdd�|_�r"ttt�G�fdd�d��}|�}n|r*t|d�}nt�}t|||�}|�	d�}|dur@t
�}|�|��|fS)a

    Create a protocol hooked up to a TLS transport hooked up to a
    StringTransport.

    @param serverMethod: The TLS method accepted by the server-side and used by the created protocol. Set to to C{None} to use the default method used by your OpenSSL library.
    l�J)�c��Sr7r<r<��clientProtocolr<r=�<lambda>��z"buildTLSProtocol.<locals>.<lambda>cseZdZ�fdd�ZeZdS)z-buildTLSProtocol.<locals>.HardCodedConnectioncrfr7r<)r:�tlsProtocol��fakeConnectionr<r=�clientConnectionForTLS��zDbuildTLSProtocol.<locals>.HardCodedConnection.clientConnectionForTLSN)rKrLrMrn�serverConnectionForTLSr<rlr<r=�HardCodedConnection�srq�r;N)rPr$�protocolrrrrrr�
buildProtocolr0�makeConnection)	�serverr\rm�serverMethod�
clientFactoryrqr@�wrapperFactory�sslProtocolr<)rhrmr=�buildTLSProtocol�s 


r{c@s(eZdZdZdd�Zdd�Zdd�ZdS)	�TLSMemoryBIOFactoryTestsz<
    Ensure TLSMemoryBIOFactory logging acts correctly.
    cCsnt�}g}|j}t�|�|�tj|�t�}dd�|_dd�|_t	|d|�}|��|��|�
|g�dS)zs
        L{TLSMemoryBIOFactory.doStart} and L{TLSMemoryBIOFactory.doStop} do
        not log any messages.
        cS�dSr7r<r<r<r<r=ri�rjz5TLSMemoryBIOFactoryTests.test_quiet.<locals>.<lambda>cSr}r7r<r<r<r<r=ri�rjFN)rrXr+�addObserver�
addCleanup�removeObserverr'�doStart�doStopr�assertEqual)r:r@�logs�logger�wrappedFactory�factoryr<r<r=�
test_quiet�s


z#TLSMemoryBIOFactoryTests.test_quietcCs(t�}t|dt��}|�d|���dS)z�
        L{TLSMemoryBIOFactory.logPrefix} amends the wrapped factory's log prefix
        with a short string (C{"TLS"}) indicating the wrapping, rather than its
        full class name.
        FzServerFactory (TLS)N)rrr'r��	logPrefix)r:r@r�r<r<r=�test_logPrefix�sz'TLSMemoryBIOFactoryTests.test_logPrefixcCs6Gdd�d�}t�}t|d|��}|�d|���dS)z�
        If the wrapped factory does not provide L{ILoggingContext},
        L{TLSMemoryBIOFactory.logPrefix} uses the wrapped factory's class name.
        c@�eZdZdS)zBTLSMemoryBIOFactoryTests.test_logPrefixFallback.<locals>.NoFactoryN�rKrLrMr<r<r<r=�	NoFactory��r�FzNoFactory (TLS)N)rrr�r�)r:r�r@r�r<r<r=�test_logPrefixFallback�sz/TLSMemoryBIOFactoryTests.test_logPrefixFallbackN)rKrLrMrNr�r�r�r<r<r<r=r|�s

r|cs�t�\}}tt�G��fdd�dt���tt�Gdd�dt���ttd|d�dt��fdd	��d
��t|��dt	��fdd	��d
��t
�fd
d	��fdd	�dd�\}}}|||fS)a
    Construct a client and server L{TLSMemoryBIOProtocol} connected by an IO
    pump.

    @param greetingData: The data which should be written in L{connectionMade}.
    @type greetingData: L{bytes}

    @return: 3-tuple of client, server, L{twisted.test.iosim.IOPump}
    cs:eZdZdZdZ��fdd�Z�fdd�Zdd�Z�ZS)	z*handshakingClientAndServer.<locals>.ClientFNcs&t����dur|j���dSdSr7)�superrWr\�writerV)�	__class__�clientGreetingDatar<r=rW�s
�z9handshakingClientAndServer.<locals>.Client.connectionMadecs(d|_|j��|_�r|j��dSdS�NT)�	handshookr\�getPeerCertificate�peerAfterHandshake�abortConnectionrV)�clientAbortAfterHandshaker<r=�handshakeCompleted�s
�z=handshakingClientAndServer.<locals>.Client.handshakeCompletedcSr}r7r<rcr<r<r=reroz9handshakingClientAndServer.<locals>.Client.connectionLost)	rKrLrMr�r�rWr�re�
__classcell__r<)r�r�)r�r=�Client�sr�c@� eZdZdZdd�Zdd�ZdS)z*handshakingClientAndServer.<locals>.ServerFcS�
d|_dSr�)�
handshakedrVr<r<r=r�rSz=handshakingClientAndServer.<locals>.Server.handshakeCompletedcSr}r7r<rcr<r<r=reroz9handshakingClientAndServer.<locals>.Server.connectionLostN)rKrLrMr�r�rer<r<r<r=�Serversr��example.com��	trustRootTc��d�S�Ni?Br<r<)r�r<r=ri�z,handshakingClientAndServer.<locals>.<lambda>)�isClientr�Fcr�r�r<r<)r�r<r=rir�c�
��d�Sr7�rtr<)�serverFr<r=ri�
cr�r7r�r<)�clientFr<r=rir�)�greet)rrrrPrrr$�forProtocol�optionsr'r.)r�r��authCert�
serverCert�clientrv�pumpr<)r�r�r�r�r�r�r=�handshakingClientAndServer�s*
	
��

�
r�c@r�)�DeterministicTLSMemoryBIOTestsz�
    Test for the implementation of L{ISSLTransport} which runs over another
    transport.

    @note: Prefer to add test cases to this suite, in this style, using
        L{connectedServerAndClient}, rather than returning L{Deferred}s.
    cCsht�\}}}|�|jjd�|�|jjd�|��|�|jjd�|�|jjd�|�|jjd�dS)a
        The completion of the TLS handshake calls C{handshakeCompleted} on
        L{Protocol} objects that provide L{IHandshakeListener}.  At the time
        C{handshakeCompleted} is invoked, the transport's peer certificate will
        have been initialized.
        FTN)r�r��wrappedProtocolr�r��flush�assertIsNotr�)r:r�rvr�r<r<r=�test_handshakeNotification)sz9DeterministicTLSMemoryBIOTests.test_handshakeNotificationcCs0tdd�\}}}|j}|��|�|jg�dS)a
        If some data is written to the transport in C{connectionMade}, but
        C{handshakeDone} doesn't like something it sees about the handshake, it
        can use C{abortConnection} to ensure that the application never
        receives that data.
        s
untrustworthyTN)r�r�r�r�rU)r:r�rvr��wrappedServerProtocolr<r<r=�test_handshakeStopWriting8sz8DeterministicTLSMemoryBIOTests.test_handshakeStopWritingN)rKrLrMrNr�r�r<r<r<r=r� sr�c@s�eZdZdZdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd�Zefd d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd0d1�Zd2d3�Zd4d5�Zd6d7�Zd8S)9�TLSMemoryBIOTestsze
    Tests for the implementation of L{ISSLTransport} which runs over another
    L{ITransport}.
    cCs.tdd�}|�t�|��|�t�|��dS)zj
        L{TLSMemoryBIOProtocol} instances provide L{ISSLTransport} and
        L{ISystemHandle}.
        N)r�
assertTruer"�
providedByr#)r:�protor<r<r=�test_interfacesKs
z!TLSMemoryBIOTests.test_interfacescCspGdd�dt�}Gdd�d�}t�}t�}t|d|�}|�}t||�t|t��}|�|�|�|�	|��dS)zw
        L{TLSMemoryBIOProtocol} instances provide the interfaces provided by
        the transport they wrap.
        c@r�)zDTLSMemoryBIOTests.test_wrappedProtocolInterfaces.<locals>.ITransportNr�r<r<r<r=�
ITransportZr�r�c@�eZdZdd�ZdS)zETLSMemoryBIOTests.test_wrappedProtocolInterfaces.<locals>.MyTransportcSr}r7r<r^r<r<r=r�^rozKTLSMemoryBIOTests.test_wrappedProtocolInterfaces.<locals>.MyTransport.writeN)rKrLrMr�r<r<r<r=�MyTransport]�r�TN)
rr$rrrrr&rur�r�)r:r�r�rxr@ryr\rkr<r<r=�test_wrappedProtocolInterfacesTs

z0TLSMemoryBIOTests.test_wrappedProtocolInterfacescCsHt�}t�}t|d|�}t|t��}t�}|�|�|�|��t	�dS)a/
        L{TLSMemoryBIOProtocol.getHandle} returns the L{OpenSSL.SSL.Connection}
        instance it uses to actually implement TLS.

        This may seem odd.  In fact, it is.  The L{OpenSSL.SSL.Connection} is
        not actually the "system handle" here, nor even an object the reactor
        knows about directly.  However, L{twisted.internet.ssl.Certificate}'s
        C{peerFromTransport} and C{hostFromTransport} methods depend on being
        able to get an L{OpenSSL.SSL.Connection} object in order to work
        properly.  Implementing L{ISystemHandle.getHandle} like this is the
        easiest way for those APIs to be made to work.  If they are changed,
        then it may make sense to get rid of this implementation of
        L{ISystemHandle} and return the underlying socket instead.
        TN)
r$rrrr&r0ru�assertIsInstance�	getHandler)r:r�r@ryr�r\r<r<r=�test_getHandleks
z TLSMemoryBIOTests.test_getHandlecsrt��t�}�fdd�|_t�}t|d|�}|�d�}t�}|�|�|��j	�|�
�j	|�|��j	|�dS)z�
        When L{TLSMemoryBIOProtocol} is connected to a transport, it connects
        the protocol it wraps to a transport.
        crfr7r<r<rgr<r=ri�rjz7TLSMemoryBIOTests.test_makeConnection.<locals>.<lambda>TN)r&r$rsrrrtr0ru�assertIsNotNoner\r��assertIs)r:rxr@ryrzr\r<rgr=�test_makeConnection�s

z%TLSMemoryBIOTests.test_makeConnectionc
Cslt�}t|_t��\}}t|d|�}|�d�}t�}t|_t�}t|d|�}|�d�}t	||�}	||||	fS)z@
        Start handshake between TLS client and server.
        TNF)
r$r&rsr5rArrtr'rr*)
r:rx�clientContextFactory�handshakeDeferredry�sslClientProtocol�
serverFactory�serverContextFactory�sslServerProtocol�connectionDeferredr<r<r=�handshakeProtocols�s&�


�z$TLSMemoryBIOTests.handshakeProtocolscCs|��\}}}}|S)zr
        The TLS handshake is performed when L{TLSMemoryBIOProtocol} is
        connected to a transport.
        )r�)r:�	tlsClient�	tlsServerr��_r<r<r=�test_handshake�sz TLSMemoryBIOTests.test_handshakecs�t��t�}�fdd�|_t�}t|d|�}|�d�}t��t�}�fdd�|_tt��	�}t
�|�}|�|�}t|d|�}|�d�}	t
|	|�}
dd�}��|���|�t��|
g�S)	z�
        L{TLSMemoryBIOProtocol} reports errors in the handshake process to the
        application-level protocol object using its C{connectionLost} method
        and disconnects the underlying transport.
        c�t��Sr7r1r<��clientConnectionLostr<r=ri���z9TLSMemoryBIOTests.test_handshakeFailure.<locals>.<lambda>TNcr�r7r1r<)�serverConnectionLostr<r=ri�r�FcSs|j�t�dSr7)�lostConnectionReason�trapr
)rsr<r<r=�cbConnectionLost�szATLSMemoryBIOTests.test_handshakeFailure.<locals>.cbConnectionLost)rr$rsr5rrtr'r-r�
getContentr�loadPEMr�r*�addCallbackr)r:rxr�ryr�r��certificateData�certificater�r�r�r�r<)r�r�r=�test_handshakeFailure�s*






�z'TLSMemoryBIOTests.test_handshakeFailurec	s|t�}t|_t��\}}t|d|�}|�d��t�}t|_t�}t|d|�}|�d�}t	|����fdd�}|�
|�|S)z�
        L{TLSMemoryBIOProtocol.getPeerCertificate} returns the
        L{OpenSSL.crypto.X509} instance representing the peer's
        certificate.
        TNFcs,���}��|tj���|�d�d�dS)N�sha1s;23:4B:72:99:2E:5D:5E:2B:02:C3:BC:1B:7C:50:67:05:4F:60:FF:C9)r�r�r�X509r��digest)�ignored�cert�r:r�r<r=�cbHandshook	s�z>TLSMemoryBIOTests.test_getPeerCertificate.<locals>.cbHandshook)r$r&rsr5rArrtr'rr*r�)	r:rxr�r�ryr�r�r�r�r<r�r=�test_getPeerCertificate�s"�




z)TLSMemoryBIOTests.test_getPeerCertificatecs�d�t��t�}�fdd�|_t��\}}t|d|�}|�d�}tt����t	�}�fdd�|_t
�}t|d|�}|�d�}t||�����fdd	�}	|�|	����fd
d�}
|�|
�|S)��
        Bytes written to L{TLSMemoryBIOProtocol} before the handshake is
        complete are received by the protocol on the other side of the
        connection once the handshake succeeds.
        �
some bytescrfr7r<r<rgr<r=ri rjz<TLSMemoryBIOTests.test_writeAfterHandshake.<locals>.<lambda>TNcrfr7r<r<��serverProtocolr<r=ri+rjFcs�j����Sr7�r\r��r�)rhr�r_r<r=r�4sz?TLSMemoryBIOTests.test_writeAfterHandshake.<locals>.cbHandshookc���d��j���dS�N��r��joinrUr��r_r:r�r<r=�cbDisconnected>�zBTLSMemoryBIOTests.test_writeAfterHandshake.<locals>.cbDisconnected)
r&r$rsr5rArrtrPr[r'rr*r�)r:rxr�r�ryr�r�r�r�r�r�r<)rhr�r_r:r�r=�test_writeAfterHandshakes,�




z*TLSMemoryBIOTests.test_writeAfterHandshakec
s�t�}||_t��\}}t|d|�}|�d�}tt����t�}�fdd�|_t	�}	t|	d|�}|�d�}
t
|
|�}���fdd�}|�|�|S)z{
        Run test where client sends data before handshake, given the sending
        protocol and expected bytes.
        TNcrfr7r<r<r�r<r=riVrjz<TLSMemoryBIOTests.writeBeforeHandshakeTest.<locals>.<lambda>Fcr�r�r�r�r�r<r=�cbConnectionDone`r�zDTLSMemoryBIOTests.writeBeforeHandshakeTest.<locals>.cbConnectionDone)r$rsr5rArrtrPr[r'rr*r�)
r:�sendingProtocolr_rxr�r�ryr�r�r�r�r�r�r<r�r=�writeBeforeHandshakeTestEs$�



z*TLSMemoryBIOTests.writeBeforeHandshakeTestc�$d�G�fdd�dt�}|�|��S)r�r�c�eZdZ�fdd�ZdS)zJTLSMemoryBIOTests.test_writeBeforeHandshake.<locals>.SimpleSendingProtocolcs|j���dSr7r�rV�r_r<r=rWo�zYTLSMemoryBIOTests.test_writeBeforeHandshake.<locals>.SimpleSendingProtocol.connectionMadeN�rKrLrMrWr<rr<r=�SimpleSendingProtocoln�r�r&r��r:rr<rr=�test_writeBeforeHandshakefsz+TLSMemoryBIOTests.test_writeBeforeHandshakecr)z�
        Bytes written to L{TLSMemoryBIOProtocol} with C{writeSequence} are
        received by the protocol on the other side of the connection.
        r�cr)zCTLSMemoryBIOTests.test_writeSequence.<locals>.SimpleSendingProtocolcs|j�tt����dSr7)r\�
writeSequence�listrrVrr<r=rW|r�zRTLSMemoryBIOTests.test_writeSequence.<locals>.SimpleSendingProtocol.connectionMadeNrr<rr<r=r{rrrrr<rr=�test_writeSequencetsz$TLSMemoryBIOTests.test_writeSequencecr)z�
        Bytes written to L{TLSMemoryBIOProtocol} after C{loseConnection} is
        called are not transmitted (unless there is a registered producer,
        which will be tested elsewhere).
        r�cr)zNTLSMemoryBIOTests.test_writeAfterLoseConnection.<locals>.SimpleSendingProtocolcs4|j���|j��|j�d�|j�dg�dS)N�hello�world)r\r�r]r
rVrr<r=rW�s
z]TLSMemoryBIOTests.test_writeAfterLoseConnection.<locals>.SimpleSendingProtocol.connectionMadeNrr<rr<r=r�rrrrr<rr=�test_writeAfterLoseConnection�sz/TLSMemoryBIOTests.test_writeAfterLoseConnectioncs>d�g�G��fdd�dt�}��|d�}|���fdd��S)zV
        Writing C{unicode} to L{TLSMemoryBIOProtocol} throws a C{TypeError}.
        �helloc�eZdZ��fdd�ZdS)zQTLSMemoryBIOTests.test_writeUnicodeRaisesTypeError.<locals>.SimpleSendingProtocolcsHz|j���Wnty��d�Ynw|j�d�|j��dS)NT�bytes)r\r��	TypeErrorrXr]rV��notBytes�resultr<r=rW�s�z`TLSMemoryBIOTests.test_writeUnicodeRaisesTypeError.<locals>.SimpleSendingProtocol.connectionMadeNrr<rr<r=r��rrcs���dg�Sr�)r���ign)rr:r<r=ri�szDTLSMemoryBIOTests.test_writeUnicodeRaisesTypeError.<locals>.<lambda>)r&r�r�)r:r�dr<)rrr:r=� test_writeUnicodeRaisesTypeError�s
	z2TLSMemoryBIOTests.test_writeUnicodeRaisesTypeErrorcs�gd��G�fdd�dt�}t�}||_t�}t|d|�}|�d�}tttt	�����t
�}�fdd�|_t�}t|d|�}|�d�}t||t
�}	���fd	d
�}
|	�|
�|	S)z�
        If multiple separate TLS messages are received in a single chunk from
        the underlying transport, all of the application bytes from each
        message are delivered to the application-level protocol.
        )	�a�b�c�d�e�f�g�h�icr)zDTLSMemoryBIOTests.test_multipleWrites.<locals>.SimpleSendingProtocolcs�D]}|j�|�qdSr7r�)r:�brr<r=rW�s�zSTLSMemoryBIOTests.test_multipleWrites.<locals>.SimpleSendingProtocol.connectionMadeNrr<rr<r=r�rrTNcrfr7r<r<r�r<r=ri�rjz7TLSMemoryBIOTests.test_multipleWrites.<locals>.<lambda>Fcs��d��j�d����dSr�r�r�r�r<r=r���z?TLSMemoryBIOTests.test_multipleWrites.<locals>.cbConnectionDone)r&r$rsr5rrtrPrYrZr[r'rr*r)r�)r:rrxr�ryr�r�r�r�r�r�r<r�r=�test_multipleWrites�s&

�
z%TLSMemoryBIOTests.test_multipleWritescs�d�d�G��fdd�dt�}t�}||_t|d�}t|d|�}|�d�}tt�����t�}�fdd	�|_t	|d�}t|d
|�}|�d�}	t
|	|�}
����fdd�}|
�|�|
S)
z�
        If a very long string is passed to L{TLSMemoryBIOProtocol.write}, any
        trailing part of it which cannot be send immediately is buffered and
        sent later.
        r�icr)z:TLSMemoryBIOTests.hugeWrite.<locals>.SimpleSendingProtocolcs|j����dSr7r�rV�r_�factorr<r=rW�szITLSMemoryBIOTests.hugeWrite.<locals>.SimpleSendingProtocol.connectionMadeNrr<r(r<r=r�rrrrTNcrfr7r<r<r�r<r=ri�rjz-TLSMemoryBIOTests.hugeWrite.<locals>.<lambda>Fcs��d��j����dSr�r�r��r_r)r:r�r<r=r��sz5TLSMemoryBIOTests.hugeWrite.<locals>.cbConnectionDone)r&r$rsr5rrtrPr[r'rr*r�)r:r;rrxr�ryr�r�r�r�r�r�r<r*r=�	hugeWrite�s$





zTLSMemoryBIOTests.hugeWritecCs|��Sr7)r+rVr<r<r=�test_hugeWrite_TLSv1��z&TLSMemoryBIOTests.test_hugeWrite_TLSv1cC�|jtd�S�Nrr)r+rrVr<r<r=�test_hugeWrite_TLSv1_1��z(TLSMemoryBIOTests.test_hugeWrite_TLSv1_1cCr.r/)r+rrVr<r<r=�test_hugeWrite_TLSv1_2�r1z(TLSMemoryBIOTests.test_hugeWrite_TLSv1_2csft��t�}�fdd�|_t�}t|d|�}|�d�}t�}t||�|j�	�dd�}��
|��S)z~
        If a L{TLSMemoryBIOProtocol} loses its connection unexpectedly, this is
        reported to the application.
        cr�r7r1r<r�r<r=rir�z;TLSMemoryBIOTests.test_disorderlyShutdown.<locals>.<lambda>TNcSs|j�tt�dSr7)r�r�r
rrgr<r<r=r�szATLSMemoryBIOTests.test_disorderlyShutdown.<locals>.cbDisconnected)rr$rsr5rrtr&r*r\r]r�)r:rxr�ryr�r�r�r<r�r=�test_disorderlyShutdown�s



z)TLSMemoryBIOTests.test_disorderlyShutdowncs�Gdd�dt�}t��t�}|����fdd�|_t��\}}t|d|�}|�d�}t��|���t�}�fdd�|_t	�}t|d|�}|�d�}	t
|	|�d	d
������fdd�}
|�|
���fd
d�}|�|�|S)z�
        L{TLSMemoryBIOProtocol.loseConnection} sends a TLS close alert and
        shuts down the underlying connection cleanly on both sides, after
        transmitting all buffered data.
        c@seZdZdd�Zdd�ZdS)zNTLSMemoryBIOTests.test_loseConnectionAfterHandshake.<locals>.NotifyingProtocolcSst�||�g|_dSr7)r2r>r_)r:�onConnectionLostr<r<r=r>"s
zWTLSMemoryBIOTests.test_loseConnectionAfterHandshake.<locals>.NotifyingProtocol.__init__cSs|j�|�dSr7)r_rXr^r<r<r=r`&rz[TLSMemoryBIOTests.test_loseConnectionAfterHandshake.<locals>.NotifyingProtocol.dataReceivedN)rKrLrMr>r`r<r<r<r=�NotifyingProtocol!sr5crfr7r<r<rgr<r=ri,rjzETLSMemoryBIOTests.test_loseConnectionAfterHandshake.<locals>.<lambda>TNcrfr7r<r<r�r<r=ri8rjFs	123456890i��cs.�j����j�d��j��t��g�S)N�x)r\r�r]rr�)�chunkOfBytesr�rhr�r�r<r=�cbHandshakeBs
zHTLSMemoryBIOTests.test_loseConnectionAfterHandshake.<locals>.cbHandshakecsX|\}}|j�t�|j�t���d�|j�����|jjj	���|jjj	�dSr�)
r�r�rr�r�r_r�r\�q�
disconnect)rrhr�)r7r:r<r=r�UszMTLSMemoryBIOTests.test_loseConnectionAfterHandshake.<locals>.cbConnectionDone)r2rr$rsr5rArrtr'rr*r�)r:r5rxr�r�ryr�r�r�r�r8r�r<)r7r�rhr:r�r�r=�!test_loseConnectionAfterHandshakes2�




z3TLSMemoryBIOTests.test_loseConnectionAfterHandshakecCs�Gdd�dt�}tt�dt��}|�}t||�}t�}|�|�|�d�|�|j	�|�
|j�|�t
td���|�|j�t��|�|jjjd�dS)zz
        The user protocol's connectionLost is only called when transport
        underlying TLS is disconnected.
        c@seZdZdZdd�ZdS)zTTLSMemoryBIOTests.test_connectionLostOnlyAfterUnderlyingCloses.<locals>.LostProtocolNcSrQr7)�disconnectedrcr<r<r=rerrSzcTLSMemoryBIOTests.test_connectionLostOnlyAfterUnderlyingCloses.<locals>.LostProtocol.connectionLost)rKrLrMr<rer<r<r<r=�LostProtocolo�r=TN�ono)r?)r&rrr$rr0ru�_tlsShutdownFinishedr��
disconnecting�assertIsNoner<rer,rrar��value�args)r:r=ryrsrkr\r<r<r=�,test_connectionLostOnlyAfterUnderlyingClosesis


z>TLSMemoryBIOTests.test_connectionLostOnlyAfterUnderlyingClosescs||��\}}}}|�|�g�|jf�fdd�	}||_|�d�|��|�|j�|��dg�|��|��dg�|S)z�
        If TLSMemoryBIOProtocol.loseConnection is called multiple times, all
        but the first call have no effect.
        c���d�|�S�N��rX��shutdown��callsr<r=�_shutdownTLS��
z@TLSMemoryBIOTests.test_loseConnectionTwice.<locals>._shutdownTLSr6rH)r��successResultOfrNr�r]r�rAr�)r:r�r�r��disconnectDeferredrNr<rLr=�test_loseConnectionTwice�s"
�

z*TLSMemoryBIOTests.test_loseConnectionTwicecs\���\}�}}g�|jf�fdd�	}|�_��d�|�����fdd�}|�|�|S)zq
        If TLSMemoryBIOProtocol.loseConnection is called after connectionLost,
        it does nothing.
        crFrGrIrJrLr<r=rN�rOzNTLSMemoryBIOTests.test_loseConnectionAfterConnectionLost.<locals>._shutdownTLSr6cs(���dg�������dg�dSrG)r�r]�r��rMr:r�r<r=r<�szNTLSMemoryBIOTests.test_loseConnectionAfterConnectionLost.<locals>.disconnected)r�rNr�r]r�)r:r�r�rQrNr<r<rTr=�&test_loseConnectionAfterConnectionLost�s
�

z8TLSMemoryBIOTests.test_loseConnectionAfterConnectionLostcs`���\�}}}|j}g}g�|j|_�j|_�fdd�}|�|���fdd�}|�|�|S)zP
        Unexpected disconnects get converted to ConnectionLost errors.
        cs��d��j��dS�Nr
)r�r\r]r�r�r<r=�
handshakeDone�s
z;TLSMemoryBIOTests.test_unexpectedEOF.<locals>.handshakeDonec����d�t��d�dS�Nr)r�rarr�rdr:r<r=r<�r&z:TLSMemoryBIOTests.test_unexpectedEOF.<locals>.disconnected)r�r�rXr`rer�)r:r�r�rQr�r_rXr<r<�rdr:r�r=�test_unexpectedEOF�s 	�

z$TLSMemoryBIOTests.test_unexpectedEOFcsj���\�}}}g��j�j_Gdd�d�}|�j��_�fdd�}|�|���fdd�}|�|�|S)zN
        Errors while writing cause the protocols to be disconnected.
        c@s$eZdZdd�Zdd�Zdd�ZdS)z4TLSMemoryBIOTests.test_errorWriting.<locals>.WrappercSrQr7)�_wrapped)r:�wrappedr<r<r=r>�rSz=TLSMemoryBIOTests.test_errorWriting.<locals>.Wrapper.__init__cSst|j|�Sr7)�getattrr^)r:�attrr<r<r=�__getattr__�r1z@TLSMemoryBIOTests.test_errorWriting.<locals>.Wrapper.__getattr__cWstd��)NzONO!)r
)r:rDr<r<r=�sendr-z9TLSMemoryBIOTests.test_errorWriting.<locals>.Wrapper.sendN)rKrLrMr>rbrcr<r<r<r=�Wrapper�srdc���d�dSrV)r�rrWr<r=rX�z:TLSMemoryBIOTests.test_errorWriting.<locals>.handshakeDonecrYrZ)r�rar
rr[r<r=r<
r&z9TLSMemoryBIOTests.test_errorWriting.<locals>.disconnected)r�rXr�re�_tlsConnectionr�)r:r�r�rQrdrXr<r<r\r=�test_errorWriting�s	�



z#TLSMemoryBIOTests.test_errorWritingcCs�dd�}|�tj�t��Gdd�dt�}Gdd�dt�}|t�}||�}t�\}}t|��dt	�
|��}ttd|d	�d
t	�
|��}	tt||��t|	|���|t�}
||�}|�
|
|�|�
||�dS)z�
        TLSMemoryBIOProtocol doesn't leave circular references that keep
        it in memory after connection is closed.
        cst�fdd�t��D��S)z�
            Return the number of instances of a given type in memory.

            @param type: Type whose instances to find.

            @return: The number of instances found.
            c3s�|]
}t|��rdVqdS)rHN)�
isinstance)�.0�x��typer<r=�	<genexpr>!s�zVTLSMemoryBIOTests.test_noCircularReferences.<locals>.nObjectsOfType.<locals>.<genexpr>)rY�gc�get_objectsrlr<rlr=�nObjectsOfTypeszCTLSMemoryBIOTests.test_noCircularReferences.<locals>.nObjectsOfTypec@r�)zCTLSMemoryBIOTests.test_noCircularReferences.<locals>.CloserProtocolcSs|j��dSr7)r\r]r^r<r<r=r`'rfzPTLSMemoryBIOTests.test_noCircularReferences.<locals>.CloserProtocol.dataReceivedN)rKrLrMr`r<r<r<r=�CloserProtocol&r�rrc@r�)zDTLSMemoryBIOTests.test_noCircularReferences.<locals>.GreeterProtocolcSs|j�d�dSrVr�rVr<r<r=rW+rzSTLSMemoryBIOTests.test_noCircularReferences.<locals>.GreeterProtocol.connectionMadeNrr<r<r<r=�GreeterProtocol*r�rsFr�r�TN)rro�enable�disabler&rrrr�r%r�rr*r�)r:rqrrrs�
origTLSProtos�origServerProtosr�r�r�rx�newTLSProtos�newServerProtosr<r<r=�test_noCircularReferencess0

�
�

�z+TLSMemoryBIOTests.test_noCircularReferencesN) rKrLrMrNr�r�r�r�r�r�r�r�r�r�r	rrrr'r
r+r,r0r2r3r;rErRrUr]rhrzr<r<r<r=r�Es:	0(/!
(&O!# 'r�c@s�eZdZdZd.dd�Z	d/dd�Zdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�ZdS)0�TLSProducerTestszA
    The TLS transport must support the IConsumer interface.
    FcCs(|��}|��|�t|p|�d�|S)a�
        Drain the bytes currently pending write from a L{StringTransport}, then
        clear it, since those bytes have been consumed.

        @param transport: The L{StringTransport} to get the bytes from.
        @type transport: L{StringTransport}

        @param allowEmpty: Allow the test to pass even if the transport has no
            outgoing bytes in it.
        @type allowEmpty: L{bool}

        @return: the outgoing bytes from the given transport
        @rtype: L{bytes}
        T)rC�clearr��bool)r:r\�
allowEmptyrCr<r<r=�drainJszTLSProducerTests.drainNc	CsPGdd�dt�}t||||d�\}}|�}|j�|d�|�|jj�|||fS)a�
        Create a new client-side protocol that is connected to a remote TLS server.

        @param serverMethod: The TLS method accepted by the server-side. Set to to C{None} to use the default method used by your OpenSSL library.

        @return: A tuple with high level client protocol, the low-level client-side TLS protocol, and a producer that is used to send data to the client.
        c@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
zGTLSProducerTests.setupStreamingProducer.<locals>.HistoryStringTransportcSst�|�g|_dSr7)r0r>�producerHistoryrVr<r<r=r>js

zPTLSProducerTests.setupStreamingProducer.<locals>.HistoryStringTransport.__init__cS�|j�d�t�|�dS)N�pause)r�rXr0�pauseProducingrVr<r<r=r�n�zVTLSProducerTests.setupStreamingProducer.<locals>.HistoryStringTransport.pauseProducingcSr�)N�resume)r�rXr0�resumeProducingrVr<r<r=r�rr�zWTLSProducerTests.setupStreamingProducer.<locals>.HistoryStringTransport.resumeProducingcSr�)N�stop)r�rXr0�
stopProducingrVr<r<r=r�vr�zUTLSProducerTests.setupStreamingProducer.<locals>.HistoryStringTransport.stopProducingN)rKrLrMr>r�r�r�r<r<r<r=�HistoryStringTransportis
r�)r\rmrvrwT)r0r{r\�registerProducerr��	streaming)	r:r\rmrvrwr��applicationProtocolrk�producerr<r<r=�setupStreamingProducer^s
�
z'TLSProducerTests.setupStreamingProducercCsztd�D]$}|�|jd�}|r|�|�|�|jd�}|r"|�|�|s(|s(nq|�|j��d�|�|j��d�dS)zJ
        Transfer bytes back and forth between two TLS protocols.
        �Tr�N)�rangerr\r`r�rC)r:rk�serverTLSProtocol�i�
clientData�
serverDatar<r<r=�flushTwoTLSProtocols�s

�z%TLSProducerTests.flushTwoTLSProtocolscCs�|��\}}}|jdtd�\}}}|�||�|j�d�|j�d�|�||�|�dg|j�|�dg|j�|j}|�d|���|�	�|�
t|j�|�
�|�|�|j��d}|j�|�|j��|�|�|j��|�|j�}	|	dd�|	dd	�|	d	d
�fD]}
|�|
�q�|�|jjd�|j��|�||�|�|jjd�|�ddg|j�|�|jdd
g�d
S)z�
        If we write some data to a TLS connection that is blocked waiting for a
        renegotiation with its peer, it will pause and resume its registered
        producer exactly once.

        Renegotiation only works on TLSv1.2 or less.
        T)rvrwsdata from clientsdata from serverzTLSv1.2spayloadrrH�NFr�r�)r�rr�r\r�r�rUrg�get_cipher_version�renegotiate�assertRaisesr�do_handshake�
_flushSendBIOr`rr]rA�unregisterProducerr�)r:rh�clientTransportr�r��serverTransport�serverProducer�tlsc�payload�messageThatUnblocksTheServer�fragmentr<r<r=� test_producerDuringRenegotiation�s@�



�
z1TLSProducerTests.test_producerDuringRenegotiationcCsJ|��\}}}|jj��|�|jd�|�|jdg�|�|jj	�dS)z�
        When the TLS transport is not blocked on reads, it correctly calls
        pauseProducing on the registered producer.
        �pausedr�N)
r�r\r�r�r��
producerStater�r��	_producer�_producerPaused�r:r�rkr�r<r<r=�(test_streamingProducerPausedInNormalMode�s
z9TLSProducerTests.test_streamingProducerPausedInNormalModecCsh|��\}}}|jj��|�|jdg�|jj��|�|jd�|�|jddg�|�|j	j
�dS)z�
        When the TLS transport is not blocked on reads, it correctly calls
        resumeProducing on the registered producer.
        r��	producingr�N)r�r\r�r�r�r�r�r��assertFalser�r�r�r<r<r=�)test_streamingProducerResumedInNormalMode�sz:TLSProducerTests.test_streamingProducerResumedInNormalModecCsJ|��\}}}|j�d�|�|jd�|�|jdg�|�|jj�dS)z�
        When the TLS transport is blocked on reads, it correctly calls
        pauseProducing on the registered producer.
        r
r�r�N)	r�r\r�r�r�r�r�r�r��r:rhrkr�r<r<r=�4test_streamingProducerPausedInWriteBlockedOnReadMode�s
zETLSProducerTests.test_streamingProducerPausedInWriteBlockedOnReadModecCs�|��\}}}|j�dd�|�|jdg�tdd�\}}|�||�|�|jddg�|�|jj	�|�|jj
�|�|jd�dS)	z�
        When the TLS transport is blocked on reads, it correctly calls
        resumeProducing on the registered producer.
        shello worldi�r�T�rvr�r�N)r�r\r�r�r�r{r�r�r�r�rAr�)r:rhrkr�r�r�r<r<r=�5test_streamingProducerResumedInWriteBlockedOnReadMode�szFTLSProducerTests.test_streamingProducerResumedInWriteBlockedOnReadModecCs@|��\}}}|j}t�}|�t|jj|d�|�|j|�dS)zM
        Registering a streaming producer twice throws an exception.
        TN)r�r��objectr��RuntimeErrorr\r�r�)r:rhrkr��originalProducer�	producer2r<r<r=�test_streamingProducerTwices�z,TLSProducerTests.test_streamingProducerTwicecCs6|��\}}}|j��|�|j�|�|jj�dS)z\
        Unregistering a streaming producer removes it, reverting to initial state.
        N�r�r\r�rBr�r�r�r<r<r=� test_streamingProducerUnregister"s
z1TLSProducerTests.test_streamingProducerUnregistercCs@|��\}}}|j��|j��|�|j�|�|jj�dS)zd
        Unregistering a streaming producer when no producer is registered is
        safe.
        Nr�r�r<r<r=�%test_streamingProducerUnregisterTwice+s


z6TLSProducerTests.test_streamingProducerUnregisterTwicecCs|��\}}}tdd�\}}|s|�||�n	|j�d�|j��|�||�|�|jj�|�d|jv�|j�d�|j�	ddg�|j�
�|�|j��d�|�|jj�|j�d	�|j�	d
g�|�||�|�
|jj�|�d�|j�d�dS)
zl
        Common code for tests involving writes by producer after
        loseConnection is called.
        Tr�sx r�r
� rr�swon'tswon't!s
x hello worldN)r�r{r�r\r�r]r�rAr�r
r��assertNotEqualrCr�r�r�rU)r:�writeBlockedOnReadrhrkr�r�r�r<r<r=�loseConnectionWithProducer6s(

z+TLSProducerTests.loseConnectionWithProducercC�
|�d�S)z�
        loseConnection() waits for the producer to unregister itself, then
        does a clean TLS close alert, then closes the underlying connection.
        F�r�rVr<r<r=�0test_streamingProducerLoseConnectionWithProducerfs
zATLSProducerTests.test_streamingProducerLoseConnectionWithProducercCr�)z�
        Even when writes are blocked on reading, loseConnection() waits for
        the producer to unregister itself, then does a clean TLS close alert,
        then closes the underlying connection.
        Tr�rVr<r<r=�4test_streamingProducerLoseConnectionWithProducerWBORms
zETLSProducerTests.test_streamingProducerLoseConnectionWithProducerWBORcCs�Gdd�dt�}Gdd�d�}|�}|j||�d�\}}}|�|jd�|j�d�|�|jd�|�|jd	g�|jj��|�|jd�|�|jd	d
g�|�	d�|�|jd�|�|jd	d
g�dS)z�
        pauseProducing() events can come from both the TLS transport layer and
        the underlying transport. In this case, both decide to pause,
        underlying first.
        c@�eZdZdZdd�ZdS)zbTLSProducerTests.test_streamingProducerBothTransportsDecideToPause.<locals>.PausingStringTransportFcSs0|js|jdurd|_|j��t�||�dSr�)�	_didPauser�r�r0r�r^r<r<r=r�s
zhTLSProducerTests.test_streamingProducerBothTransportsDecideToPause.<locals>.PausingStringTransport.writeN)rKrLrMr�r�r<r<r<r=�PausingStringTransport|r>r�c@sDeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dS)zYTLSProducerTests.test_streamingProducerBothTransportsDecideToPause.<locals>.TLSConnectioncSrTr7)�lrVr<r<r=r>�rSzbTLSProducerTests.test_streamingProducerBothTransportsDecideToPause.<locals>.TLSConnection.__init__cSsF|js	|dd�}t|j�dkr|j�d�t��|j�|�t|�S)N���rHr�)r�r[rXrr^r<r<r=rc�sz^TLSProducerTests.test_streamingProducerBothTransportsDecideToPause.<locals>.TLSConnection.sendcSr}r7r<rVr<r<r=�set_connect_state�rozkTLSProducerTests.test_streamingProducerBothTransportsDecideToPause.<locals>.TLSConnection.set_connect_statecSr}r7r<rVr<r<r=r��rozfTLSProducerTests.test_streamingProducerBothTransportsDecideToPause.<locals>.TLSConnection.do_handshakecSr}r7r<r^r<r<r=�	bio_write�rozcTLSProducerTests.test_streamingProducerBothTransportsDecideToPause.<locals>.TLSConnection.bio_writecSsdS)N�Xr<�r:�sizer<r<r=�bio_read�rozbTLSProducerTests.test_streamingProducerBothTransportsDecideToPause.<locals>.TLSConnection.bio_readcS�t��r7)rr�r<r<r=�recv��z^TLSProducerTests.test_streamingProducerBothTransportsDecideToPause.<locals>.TLSConnection.recvN)
rKrLrMr>rcr�r�r�r�r�r<r<r<r=�
TLSConnection�sr�rlr�r
r�r�r�N)
r0r�r�r�r\r�r�r�r�r`)r:r�r�r\rhrkr�r<r<r=�1test_streamingProducerBothTransportsDecideToPauseus 	�
zBTLSProducerTests.test_streamingProducerBothTransportsDecideToPausecCs,|��\}}}|jj��|�|jd�dS)z�
        If the underlying transport tells its producer to stopProducing(),
        this is passed on to the high-level producer.
        �stoppedN)r�r\r�r�r�r�r�r<r<r=�#test_streamingProducerStopProducing�sz4TLSProducerTests.test_streamingProducerStopProducingcs�t�\}�t|j��|j��d��jjj����t����j�����j	|j���
�jj�����fdd�}�j�
|�tdd�\}}���|��jS)zM
        Non-streaming producers get wrapped as streaming producers.
        Fcs*���j����jj����j�dSr7)rB�consumerr\r�r�r8��ignore�r�r:�streamingProducerrkr<r=�done�sz8TLSProducerTests.test_nonStreamingProducer.<locals>.doneTr�)r{r/r\r�r�r�r�rr��	_consumerr�r�rr�r�)r:rhr�r�r�r<r�r=�test_nonStreamingProducer�s


z*TLSProducerTests.test_nonStreamingProducercCs"t�}t|�}|�tt|��dS)zC
        L{_ProducerMembrane} implements L{IPushProducer}.
        N)r0rr�rr!)r:r��membraner<r<r=�test_interface�szTLSProducerTests.test_interfacecCsbt�\}}dd�|_Gdd�d�}|�tt���|�}|�|d�|�|jj�|�|j	�dS)z�
        If a producer is registered after the transport has disconnected, the
        producer is not used, and its stopProducing method is called.
        cSs|�tt�Sr7)r�r
r�rdr<r<r=ri�s�zFTLSProducerTests.registerProducerAfterConnectionLost.<locals>.<lambda>c@r�)zFTLSProducerTests.registerProducerAfterConnectionLost.<locals>.ProducerFcSsddS)NrHrr<rVr<r<r=r��r-zVTLSProducerTests.registerProducerAfterConnectionLost.<locals>.Producer.resumeProducingcSr�r�)r�rVr<r<r=r��rSzTTLSProducerTests.registerProducerAfterConnectionLost.<locals>.Producer.stopProducingN)rKrLrMr�r�r�r<r<r<r=�Producer�sr�FN)
r{rer,rr�rBr\r�r�r�)r:r�rhrkr�r�r<r<r=�#registerProducerAfterConnectionLost�s


z4TLSProducerTests.registerProducerAfterConnectionLostcC�|�d�dS)z�
        If a streaming producer is registered after the transport has
        disconnected, the producer is not used, and its stopProducing method
        is called.
        TN�r�rVr<r<r=�)test_streamingProducerAfterConnectionLost
�z:TLSProducerTests.test_streamingProducerAfterConnectionLostcCr�)z�
        If a non-streaming producer is registered after the transport has
        disconnected, the producer is not used, and its stopProducing method
        is called.
        FNr�rVr<r<r=�,test_nonStreamingProducerAfterConnectionLostr�z=TLSProducerTests.test_nonStreamingProducerAfterConnectionLost)F)NNFN)rKrLrMrNrr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r<r<r<r=r{Es0

�'8	0G	 r{c@s`eZdZdZdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dd�Z
dd�Zdd�Zdd�Z
dS)�NonStreamingProducerTestszP
    Non-streaming producers can be adapted into being streaming producers.
    csbt��}t|������d��jf�fdd�	}|�_|j}���fdd�}|�|����|S)zd
        Verify the consumer writes out all its data, but is not called after
        that.
        Tc�|����dSr7��
stopStreaming��orig�r�r<r=�
unregister-�z<NonStreamingProducerTests.streamUntilEnd.<locals>.unregistercs,�����d����j����j�dS)Ns
0123456789)r�rCrBr�r�r8rS�r�r:r�r<r=�
doneStreaming5�z?NonStreamingProducerTests.streamUntilEnd.<locals>.doneStreaming)r/rr�r�rr��startStreaming)r:r��
nsProducerr�r�r�r<r�r=�streamUntilEnd s

z(NonStreamingProducerTests.streamUntilEndcCst�}|�|�S)z�
        When converted to a streaming producer, the non-streaming producer
        writes out all its data, but is not called after that.
        �r0r�)r:r�r<r<r=�test_writeUntilDoneBs
z-NonStreamingProducerTests.test_writeUntilDonecsvGdd�dt�}|�}t|�}t||��|��d��fdd�}|j�|�|j}��fdd�}|�|����|S)z}
        When the streaming producer is paused, the underlying producer stops
        getting resumeProducing calls.
        c@r�)zDNonStreamingProducerTests.test_pause.<locals>.PausingStringTransportrcSst�|�t�|_dSr7)r0r>rr�rVr<r<r=r>Ss
zMNonStreamingProducerTests.test_pause.<locals>.PausingStringTransport.__init__cSsJ|jd7_t�||�|jdkr#|j��|j}|`|�d�dSdS�NrHr�)�writesr0r�r�r�r�rC)r:r_rr<r<r=r�Ws

�zJNonStreamingProducerTests.test_pause.<locals>.PausingStringTransport.writeN)rKrLrMr�r>r�r<r<r<r=r�Psr�Tcre)Nz$BUG: The producer should not finish!)�failr�rVr<r=�shouldNotBeCalledfrfz?NonStreamingProducerTests.test_pause.<locals>.shouldNotBeCalledcs���jjd�dSrG)r��	_coopTask�_pauseCountr��r:r�r<r=r�msz4NonStreamingProducerTests.test_pause.<locals>.paused)r0r/rr�rr�r�r�)r:r�r�r�r�r�r�r<r�r=�
test_pauseJs

z$NonStreamingProducerTests.test_pausecCs Gdd�dt�}|�}|�|�S)a
        When the streaming producer is paused and then resumed, the underlying
        producer starts getting resumeProducing calls again after the resume.

        The test will never finish (or rather, time out) if the resume
        producing call is not working.
        c@r�)zENonStreamingProducerTests.test_resume.<locals>.PausingStringTransportrcSs@|jd7_t�||�|jdkr|j��|j��dSdSr�)r�r0r�r�r�r�r^r<r<r=r��s

�zKNonStreamingProducerTests.test_resume.<locals>.PausingStringTransport.writeN�rKrLrMr�r�r<r<r<r=r��r>r�r�)r:r�r�r<r<r=�test_resumews	

z%NonStreamingProducerTests.test_resumecsbGdd�dt�}|��t���t�������d��j}����fdd�}|�|����|S)z�
        When the streaming producer is stopped by the consumer, the underlying
        producer is stopped, and streaming is stopped.
        c@r�)zMNonStreamingProducerTests.test_stopProducing.<locals>.StoppingStringTransportrcSs6|jd7_t�||�|jdkr|j��dSdSr�)r�r0r�r�r�r^r<r<r=r��s

�zSNonStreamingProducerTests.test_stopProducing.<locals>.StoppingStringTransport.writeNr�r<r<r<r=�StoppingStringTransport�r>rTcs,�����d����j����j�dS)Ns012)r�rCr�r�r8rS�r�r�r:r�r<r=r��r�zCNonStreamingProducerTests.test_stopProducing.<locals>.doneStreaming)r0r/rr�rr�r�)r:rr�r�r<rr=�test_stopProducing�s	

z,NonStreamingProducerTests.test_stopProducingcs�Gdd�dt�}|��}t|������d�g�t��j���tj�j��jf�fdd�	}|�_��	��j
��}|�dd�������fdd	�}|�
|�|S)
z�
        Common implementation for tests where the underlying producer throws
        an exception when its resumeProducing is called.
        c@r�)zINonStreamingProducerTests.resumeProducingRaises.<locals>.ThrowingProducercSs |jdkr	ddSt�|�dS)Nr�rHr)�counterr/r�rVr<r<r=r��s
zYNonStreamingProducerTests.resumeProducingRaises.<locals>.ThrowingProducer.resumeProducingN)rKrLrMr�r<r<r<r=�ThrowingProducer�r�rTcr�r7r�r�r�r<r=r��r�zCNonStreamingProducerTests.resumeProducingRaises.<locals>.unregistercSs
|�t�Sr7)r�r(r�r<r<r=ri�r�zANonStreamingProducerTests.resumeProducingRaises.<locals>.<lambda>csx�����d����}��t|�t���t|���D]\}\}}}��|�|����||d�q���j�dS)Ns01�why)	r�rC�flushLoggedErrorsr[�zipr�ra�assertInr8)r�errors�f�expected�msg�logMsg�r��expectedExceptions�
loggedMsgsr:r�r<r=r��s�z@NonStreamingProducerTests.resumeProducingRaises.<locals>.stopped)r/rr�r+r~rXrr�r�r�r��whenDone�
addErrbackr�)r:r�rrr�r�r�r�r<rr=�resumeProducingRaises�s



z/NonStreamingProducerTests.resumeProducingRaisescs4t�����tdfg�}��fdd�}|�|�|S)z�
        If the underlying producer raises an exception when resumeProducing is
        called, the streaming wrapper should log the error, unregister from
        the consumer and stop streaming.
        �!failed, producing will be stoppedcs���j�dSr7)rBr�r��r�r:r<r=�
cleanShutdown�szKNonStreamingProducerTests.test_resumeProducingRaises.<locals>.cleanShutdown)r0r�ZeroDivisionErrorr�)r:r�rr<rr=�test_resumeProducingRaises�s
�
z4NonStreamingProducerTests.test_resumeProducingRaisescCs,t�}dd�}||_|�|tdftdfg�S)a
        If the underlying producer raises an exception when resumeProducing is
        called, the streaming wrapper should log the error, unregister from
        the consumer and stop streaming even if the unregisterProducer call
        also raise.
        cSr�r7)r�r<r<r<r=�raiser�r�z^NonStreamingProducerTests.test_resumeProducingRaiseAndUnregisterProducerRaises.<locals>.raiserrzfailed to unregister producer)r0r�rrr�)r:r�rr<r<r=�4test_resumeProducingRaiseAndUnregisterProducerRaises�s��zNNonStreamingProducerTests.test_resumeProducingRaiseAndUnregisterProducerRaisescCs@t�}t|�}t||�}|��|��|��|�|j�dS)z�
        stopStreaming() can be called more than once without blowing
        up. This is useful for error-handling paths.
        N)r0r/rr�r�r�r8�r:r�r�r�r<r<r=�test_stopStreamingTwice
s
z1NonStreamingProducerTests.test_stopStreamingTwicecCs,t�}t|�}t||�}|�tt|��dS)z=
        L{_PullToPush} implements L{IPushProducer}.
        N)r0r/rr�rr!rr<r<r=r�s
z(NonStreamingProducerTests.test_interfaceN)rKrLrMrNr�r�r�rrrrrrr�r<r<r<r=r�s"-#3
r�c@r�)�ClientNegotiationFactoryza
    A L{ClientFactory} that has a set of acceptable protocols for NPN/ALPN
    negotiation.
    cC�
||_dS)z�
        Create a L{ClientNegotiationFactory}.

        @param acceptableProtocols: The protocols the client will accept
            speaking after the TLS handshake is complete.
        @type acceptableProtocols: L{list} of L{bytes}
        N��_acceptableProtocols�r:�acceptableProtocolsr<r<r=r>(�
z!ClientNegotiationFactory.__init__cC�|jS�a
        Returns a list of protocols that can be spoken by the connection
        factory in the form of ALPN tokens, as laid out in the IANA registry
        for ALPN tokens.

        @return: a list of ALPN tokens in order of preference.
        @rtype: L{list} of L{bytes}
        r rVr<r<r=r#2�	z,ClientNegotiationFactory.acceptableProtocolsN�rKrLrMrNr>r#r<r<r<r=r!�
rc@r�)�ServerNegotiationFactoryza
    A L{ServerFactory} that has a set of acceptable protocols for NPN/ALPN
    negotiation.
    cCr)z�
        Create a L{ServerNegotiationFactory}.

        @param acceptableProtocols: The protocols the server will accept
            speaking after the TLS handshake is complete.
        @type acceptableProtocols: L{list} of L{bytes}
        Nr r"r<r<r=r>Er$z!ServerNegotiationFactory.__init__cCr%r&r rVr<r<r=r#Or'z,ServerNegotiationFactory.acceptableProtocolsNr(r<r<r<r=r*>r)r*)FNNN)NF)WrNro�zope.interfacerrr�zope.interface.verifyr�twisted.python.compatr�OpenSSLr�OpenSSL.SSLrr	r
rrr
r�twisted.protocols.tlsrrrr�ImportError�skip�twisted.internet.sslrr�twisted.test.ssl_helpersrrr�twisted.test.test_sslverifyr�twisted.internet.deferrr�twisted.internet.errorrr�twisted.internet.interfacesrrrr r!r"r#�twisted.internet.protocolr$r%r&r'�twisted.internet.taskr(�twisted.protocols.loopbackr)r*�twisted.pythonr+�twisted.python.failurer,�twisted.python.filepathr-�twisted.test.iosimr.�twisted.test.proto_helpersr/r0�twisted.test.test_tcpr2�twisted.trial.unittestr3r4r5rPr{r|r�r�r�r{r�rr*r<r<r<r=�<module>sn$
�$	/
�&2
�=%Y