File: //usr/lib/python3/dist-packages/twisted/internet/test/__pycache__/test_tls.cpython-38.pyc
U
W[�4 �
@ sL d Z ddlmZmZ eZddlmZ ddlm Z ddl
m Z ddl m
Z
ddlmZmZmZ ddlmZmZmZ dd lmZmZ dd
lmZmZm Z dd l m Z dd l m Z dd
l!m"Z" ddl#m$Z$ ddl%m&Z& ddl'm(Z(m)Z)m*Z* ddl+m,Z,m-Z-m.Z. zddl/m0Z0 W n e1k
�r" dZ0Y n X ddl2m3Z3m4Z4 ddl2m5Z5 G dd� d�Z6G dd� de7�Z8ee�G dd� de7��Z9G d d � d e,e8�Z:G d d � d e7�Z;G d d!� d!e6e
e-�Z<G d"d#� d#e,e8�Z=G d$d%� d%e6e
e8e-e;�Z>G d&d'� d'e6e8e&e;e(e)e
� Z?e@� �Ae>�B� � e@� �Ae<�B� � e@� �Ae?� �B� � G d(d)� d)e
e*e8�ZCe@� �AeC�B� � dS )*z0
Tests for implementations of L{ITLSTransport}.
� )�division�absolute_import)� implementer)�
networkString)�FilePath)�ReactorBuilder)�
ServerFactory�
ClientFactory�Protocol)� IReactorSSL�
ITLSTransport�IStreamClientEndpoint)�Deferred� DeferredList)�SSL4ServerEndpoint�SSL4ClientEndpoint�TCP4ClientEndpoint)�ConnectionClosed)�
Cooperator)�SkipTest)�platform)�ObjectModelIntegrationMixin)�ConnectToTCPListenerMixin�StreamTransportTestsMixin�AbortConnectionMixin)�EndpointCreator�ConnectionTestsMixin�BrokenContextFactory)� FILETYPE_PEMN)�PrivateCertificate�KeyPair��ClientContextFactoryc @ s( e Zd ZegZe�� r$dZeed�ZdS )�TLSMixinzcFor some reason, these reactors don't deal with SSL disconnection correctly on Windows. See #3371.)z*twisted.internet.glib2reactor.Glib2Reactorz(twisted.internet.gtk2reactor.Gtk2ReactorN) �__name__�
__module__� __qualname__r �requiredInterfacesr Z isWindows�msgZskippedReactors� r) r) �@/usr/lib/python3/dist-packages/twisted/internet/test/test_tls.pyr# , s ��r# c @ s@ e Zd ZddlZeeej���d��d�Z [dd� Z
dd� Z dS ) �ContextGeneratingMixinr Ns tests
server.pemc C s( | j �� }t�|t�|t�t�}|�� S )zM
Return a new SSL context suitable for use in a test server.
)�_pemZ
getContentr �loadr r Zoptions)�selfZpemZcertr) r) r* �getServerContext> s
�z'ContextGeneratingMixin.getServerContextc C s t � S �Nr! �r. r) r) r* �getClientContextH s z'ContextGeneratingMixin.getClientContext) r$ r% r& Ztwistedr r �__file__ZsiblingZchildr, r/ r2 r) r) r) r* r+ 8 s ���
r+ c @ s e Zd ZdZdd� Zdd� ZdS )�StartTLSClientEndpointa!
An endpoint which wraps another one and adds a TLS layer immediately when
connections are set up.
@ivar wrapped: A L{IStreamClientEndpoint} provider which will be used to
really set up connections.
@ivar contextFactory: A L{ContextFactory} to use to do TLS.
c C s || _ || _d S r0 )�wrapped�contextFactory)r. r5 r6 r) r) r* �__init__Y s z StartTLSClientEndpoint.__init__c s$ G � �fdd�dt �}�j�|� �S )z�
Establish a connection using a protocol build by C{factory} and
immediately start TLS on it. Return a L{Deferred} which fires with the
protocol instance.
c s e Zd Z� �fdd�ZdS )z6StartTLSClientEndpoint.connect.<locals>.WrapperFactoryc s( �� |�� � jf� �fdd� }|� _� S )Nc s � j ��j� | � d S r0 )� transport�startTLSr6 )Zorig)�protocolr. r) r* �connectionMadei s z\StartTLSClientEndpoint.connect.<locals>.WrapperFactory.buildProtocol.<locals>.connectionMade)�
buildProtocolr; )Z wrapperSelfZaddrr; ��factoryr. )r: r* r< g s
zDStartTLSClientEndpoint.connect.<locals>.WrapperFactory.buildProtocolN)r$ r% r& r< r) r= r) r* �WrapperFactoryf s r? )r r5 �connect)r. r> r? r) r= r* r@ ^ s z StartTLSClientEndpoint.connectN)r$ r% r& �__doc__r7 r@ r) r) r) r* r4 M s
r4 c @ s e Zd ZdZdd� Zdd� ZdS )�StartTLSClientCreatorz{
Create L{ITLSTransport.startTLS} endpoint for the client, and normal SSL
for server just because it's easier.
c C s t |d| �� �S )z�
Construct an SSL server endpoint. This should be constructing a TCP
server endpoint which immediately calls C{startTLS} instead, but that
is hard.
r �r r/ �r. �reactorr) r) r* �serverx s z StartTLSClientCreator.serverc C s t t|d|j�t� �S )zS
Construct a TCP client endpoint wrapped to immediately start TLS.
� 127.0.0.1)r4 r �portr" �r. rE Z
serverAddressr) r) r* �client� s ��z StartTLSClientCreator.clientN�r$ r% r& rA rF rJ r) r) r) r* rB s s rB c @ s e Zd ZdZdd� ZdS )�BadContextTestsMixinz�
Mixin for L{ReactorBuilder} subclasses which defines a helper for testing
the handling of broken context factories.
c C s0 | � � }| �t||t� �}| �tjt|�� dS )a�
Assert that the exception raised by a broken context factory's
C{getContext} method is raised by some reactor method. If it is not, an
exception will be raised to fail the test.
@param useIt: A two-argument callable which will be called with a
reactor and a broken context factory and which is expected to raise
the same exception as the broken context factory's C{getContext}
method.
N)� buildReactorZ assertRaises�
ValueErrorr Z assertEqual�message�str)r. �useItrE �excr) r) r* �_testBadContext� s �z$BadContextTestsMixin._testBadContextN)r$ r% r& rA rS r) r) r) r* rL � s rL c @ s e Zd ZdZe� ZdS )�StartTLSClientTestsMixinz�
Tests for TLS connections established using L{ITLSTransport.startTLS} (as
opposed to L{IReactorSSL.connectSSL} or L{IReactorSSL.listenSSL}).
N)r$ r% r& rA rB � endpointsr) r) r) r* rT � s rT c @ s e Zd ZdZdd� Zdd� ZdS )�
SSLCreatorz
Create SSL endpoints.
c C s t |d| �� �S )zQ
Create an SSL server endpoint on a TCP/IP-stack allocated port.
r rC rD r) r) r* rF � s zSSLCreator.serverc C s t |d|jt� �S )z�
Create an SSL client endpoint which will connect localhost on
the port given by C{serverAddress}.
@type serverAddress: L{IPv4Address}
rG )r rH r" rI r) r) r* rJ � s �zSSLCreator.clientNrK r) r) r) r* rV � s rV c @ s& e Zd ZdZe� Zdd� Zdd� ZdS )�SSLClientTestsMixinz<
Mixin defining tests relating to L{ITLSTransport}.
c C s dd� }| � |� dS )z�
If the context factory passed to L{IReactorSSL.connectSSL} raises an
exception from its C{getContext} method, that exception is raised by
L{IReactorSSL.connectSSL}.
c S s | � ddt� |�S )NrG i� )�
connectSSLr �rE r6 r) r) r* rQ � s �z2SSLClientTestsMixin.test_badContext.<locals>.useItN�rS �r. rQ r) r) r* �test_badContext� s z#SSLClientTestsMixin.test_badContextc s G dd� dt �}| �� �t� }t� |_||_| �� |_t� }t� |_||_| � � |_|jj
|j_
g � t |j|jgdd�}� fdd�}|� |� �j
d|dd �}| �|j� ��|�� j|�� j|�}| �|j� |� �fd
d �� | ��� � d �t� � d �t� d
S )aO
L{ITCPTransport.loseConnection} ends a connection which was set up with
L{ITLSTransport.startTLS} and which has recently been written to. This
is intended to verify that a socket send error masked by the TLS
implementation doesn't prevent the connection from being reported as
closed.
c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zQSSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS.<locals>.ShortProtocolc S sN t �| j�s.| jj}d | j_|�td�� d S | j�| jj� | j� d� d S )NzNo ITLSTransport support� x)
r Z
providedByr8 r> �finishedZerrbackr r9 �context�write)r. r^ r) r) r* r; � s z`SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS.<locals>.ShortProtocol.connectionMadec S s | j �d� | j �� d S )N� y)r8 r` ZloseConnection)r. �datar) r) r* � dataReceived� s z^SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS.<locals>.ShortProtocol.dataReceivedc S s& | j j}|d k r"d | j _|�|� d S r0 )r> r^ �callback)r. �reasonr^ r) r) r* �connectionLost� s z`SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS.<locals>.ShortProtocol.connectionLostN)r$ r% r& r; rc rf r) r) r) r* �
ShortProtocol� s
rg T)Z
consumeErrorsc s" � � | d d | d d g� d S )Nr � )�extend)Zresults)�lostConnectionResultsr) r* �
cbFinished s zNSSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS.<locals>.cbFinishedr rG )Z interfacec s � � � S r0 )�stop)Zign�rE r) r* �<lambda> � zLSSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS.<locals>.<lambda>rh N)r
rM r r r^ r: r/ r_ r r2 �methodr Z addCallbackZ listenTCPZ
addCleanupZ
stopListeningZ
connectTCP�getHostZhostrH Z
disconnectZ
runReactorZtrapr )r. rg Z
serverFactoryZ
clientFactoryr^ rk rH Z connectorr) )rj rE r* �&test_disconnectAfterWriteAfterStartTLS� s< !
�
�
z:SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLSN)r$ r% r& rA rV rU r\ rr r) r) r) r* rW � s rW c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�TLSPortTestsBuilderz,
Tests for L{IReactorSSL.listenSSL}
c C s |� d|| �� �S )z0
Get a TLS port from a reactor.
r )� listenSSLr/ )r. rE r> r) r) r* �getListeningPort* s z$TLSPortTestsBuilder.getListeningPortc C s d||� � jf S )zY
Get the message expected to be logged when a TLS port starts listening.
z%s (TLS) starting on %d�rq rH )r. rH r> r) r) r* �#getExpectedStartListeningLogMessage1 s z7TLSPortTestsBuilder.getExpectedStartListeningLogMessagec C s d|� � jf S )zJ
Get the expected connection lost message for a TLS port.
z(TLS Port %s Closed)rv )r. rH r) r) r* � getExpectedConnectionLostLogMsg8 s z3TLSPortTestsBuilder.getExpectedConnectionLostLogMsgc C s dd� }| � |� dS )z�
If the context factory passed to L{IReactorSSL.listenSSL} raises an
exception from its C{getContext} method, that exception is raised by
L{IReactorSSL.listenSSL}.
c S s | � dt� |�S )Nr )rt r rY r) r) r* rQ E s z2TLSPortTestsBuilder.test_badContext.<locals>.useItNrZ r[ r) r) r* r\ ? s z#TLSPortTestsBuilder.test_badContextc C s |� | j|j|| �� �S )a
Connect to the given listening TLS port, assuming the
underlying transport is TCP.
@param reactor: The reactor under test.
@type reactor: L{IReactorSSL}
@param address: The listening's address. Only the C{port}
component is used; see
L{ConnectToTCPListenerMixin.LISTENER_HOST}.
@type address: L{IPv4Address} or L{IPv6Address}
@param factory: The client factory.
@type factory: L{ClientFactory}
@return: The connector
)rX Z
LISTENER_HOSTrH r2 )r. rE Zaddressr> r) r) r* �connectToListenerJ s �z%TLSPortTestsBuilder.connectToListenerN) r$ r% r& rA ru rw rx r\ ry r) r) r) r* rs " s rs c @ s, e Zd ZdZefZe� Zdd� Zdd� Z dS )�AbortSSLConnectionTestsz-
C{abortConnection} tests using SSL.
c s< t �| �� ddlm} t� fdd�d�}| �|d|j� � S )Nr )�_producer_helpersc s � � d| �S )Ng�h㈵��>)Z callLater)�xrm r) r* rn y ro z6AbortSSLConnectionTests.buildReactor.<locals>.<lambda>)Z scheduler� cooperate)r rM Ztwisted.internetr{ r Zpatchr} )r. r{ Z
cooperatorr) rm r* rM r s
�z$AbortSSLConnectionTests.buildReactorc C s t d krtd��d S )NzOpenSSL not available.)r r r1 r) r) r* �setUp~ s z AbortSSLConnectionTests.setUpN)
r$ r% r& rA r r' rV rU rM r~ r) r) r) r* rz k s
rz )DrA Z
__future__r r �typeZ
__metaclass__Zzope.interfacer Ztwisted.python.compatr Ztwisted.python.filepathr Z#twisted.internet.test.reactormixinsr Ztwisted.internet.protocolr r r
Ztwisted.internet.interfacesr r r
Ztwisted.internet.deferr r Ztwisted.internet.endpointsr r r Ztwisted.internet.errorr Ztwisted.internet.taskr Ztwisted.trial.unittestr Ztwisted.python.runtimer Z twisted.internet.test.test_corer Z twisted.internet.test.test_tcpr r r Z&twisted.internet.test.connectionmixinsr r r ZOpenSSL.cryptor � ImportErrorZtwisted.internet.sslr r r" r# �objectr+ r4 rB rL rT rV rW rs �globals�updateZmakeTestCaseClassesrz r) r) r) r* �<module> s^
% �_ �C