← Back
Editing: 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
Save File
Cancel