U  W[4 @sLdZddlmZmZeZddlmZddlm Z ddl m Z ddl m Z ddlmZmZmZddlmZmZmZdd lmZmZdd lmZmZmZdd lmZdd lm 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/m0Z0Wne1k r"dZ0YnXddl2m3Z3m4Z4ddl2m5Z5GdddZ6Gddde7Z8eeGddde7Z9Gddde,e8Z:Gddde7Z;Gd d!d!e6e e-ZGd&d'd'e6e8e&e;e(e)e Z?e@Ae>Be@Aes  z'ContextGeneratingMixin.getServerContextcCstSNr!r.r)r)r*getClientContextHsz'ContextGeneratingMixin.getClientContext) r$r%r&Ztwistedrr__file__ZsiblingZchildr,r/r2r)r)r)r*r+8s r+c@s eZdZdZddZddZdS)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. cCs||_||_dSr0)wrappedcontextFactory)r.r5r6r)r)r*__init__YszStartTLSClientEndpoint.__init__cs$Gfdddt}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. cseZdZfddZdS)z6StartTLSClientEndpoint.connect..WrapperFactorycs(|jffdd }|_S)Ncsjj|dSr0) transportstartTLSr6)Zorig)protocolr.r)r*connectionMadeisz\StartTLSClientEndpoint.connect..WrapperFactory.buildProtocol..connectionMade) buildProtocolr;)Z wrapperSelfZaddrr;factoryr.)r:r*r<gs zDStartTLSClientEndpoint.connect..WrapperFactory.buildProtocolN)r$r%r&r<r)r=r)r*WrapperFactoryfsr?)rr5connect)r.r>r?r)r=r*r@^s zStartTLSClientEndpoint.connectN)r$r%r&__doc__r7r@r)r)r)r*r4Ms r4c@s eZdZdZddZddZdS)StartTLSClientCreatorz{ Create L{ITLSTransport.startTLS} endpoint for the client, and normal SSL for server just because it's easier. cCst|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. rrr/r.reactorr)r)r*serverxszStartTLSClientCreator.servercCstt|d|jtS)zS Construct a TCP client endpoint wrapped to immediately start TLS. 127.0.0.1)r4rportr"r.rEZ serverAddressr)r)r*clientszStartTLSClientCreator.clientNr$r%r&rArFrJr)r)r)r*rBss rBc@seZdZdZddZdS)BadContextTestsMixinz Mixin for L{ReactorBuilder} subclasses which defines a helper for testing the handling of broken context factories. cCs0|}|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 ValueErrorrZ assertEqualmessagestr)r.useItrEexcr)r)r*_testBadContexts z$BadContextTestsMixin._testBadContextN)r$r%r&rArSr)r)r)r*rLsrLc@seZdZdZeZdS)StartTLSClientTestsMixinz Tests for TLS connections established using L{ITLSTransport.startTLS} (as opposed to L{IReactorSSL.connectSSL} or L{IReactorSSL.listenSSL}). N)r$r%r&rArB endpointsr)r)r)r*rTsrTc@s eZdZdZddZddZdS) SSLCreatorz Create SSL endpoints. cCst|d|S)zQ Create an SSL server endpoint on a TCP/IP-stack allocated port. rrCrDr)r)r*rFszSSLCreator.servercCst|d|jtS)z Create an SSL client endpoint which will connect localhost on the port given by C{serverAddress}. @type serverAddress: L{IPv4Address} rG)rrHr"rIr)r)r*rJs zSSLCreator.clientNrKr)r)r)r*rVsrVc@s&eZdZdZeZddZddZdS)SSLClientTestsMixinz< Mixin defining tests relating to L{ITLSTransport}. cCsdd}||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}. cSs|ddt|S)NrGi) connectSSLr rEr6r)r)r*rQs z2SSLClientTestsMixin.test_badContext..useItNrSr.rQr)r)r*test_badContextsz#SSLClientTestsMixin.test_badContextcs Gdddt}|t}t|_||_||_t}t|_||_| |_|jj |j_ gt |j|jgdd}fdd}| |j d|dd }||j|j|j|}||j| fd d |dtd td 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$eZdZddZddZddZdS)zQSSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS..ShortProtocolcSsNt|js.|jj}d|j_|tddS|j|jj|j ddS)NzNo ITLSTransport supportx) r Z providedByr8r>finishedZerrbackrr9contextwrite)r.r^r)r)r*r;s z`SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS..ShortProtocol.connectionMadecSs|jd|jdS)Ny)r8r`ZloseConnection)r.datar)r)r* dataReceiveds z^SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS..ShortProtocol.dataReceivedcSs&|jj}|dk r"d|j_||dSr0)r>r^callback)r.reasonr^r)r)r*connectionLostsz`SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS..ShortProtocol.connectionLostN)r$r%r&r;rcrfr)r)r)r* ShortProtocols rgT)Z consumeErrorscs"|dd|ddgdS)Nr)extend)Zresults)lostConnectionResultsr)r* cbFinishedszNSSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS..cbFinishedrrG)Z interfacecsSr0)stop)ZignrEr)r*zLSSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS..rhN)r rMrrr^r:r/r_r r2methodrZ addCallbackZ listenTCPZ addCleanupZ stopListeningZ connectTCPgetHostZhostrHZ disconnectZ runReactorZtrapr)r.rgZ serverFactoryZ clientFactoryr^rkrHZ connectorr))rjrEr*&test_disconnectAfterWriteAfterStartTLSs<!         z:SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLSN)r$r%r&rArVrUr\rrr)r)r)r*rWs rWc@s8eZdZdZddZddZddZdd Zd d Zd S) TLSPortTestsBuilderz, Tests for L{IReactorSSL.listenSSL} cCs|d||S)z0 Get a TLS port from a reactor. r) listenSSLr/)r.rEr>r)r)r*getListeningPort*sz$TLSPortTestsBuilder.getListeningPortcCsd||jfS)zY Get the message expected to be logged when a TLS port starts listening. z%s (TLS) starting on %drqrH)r.rHr>r)r)r*#getExpectedStartListeningLogMessage1sz7TLSPortTestsBuilder.getExpectedStartListeningLogMessagecCsd|jfS)zJ Get the expected connection lost message for a TLS port. z(TLS Port %s Closed)rv)r.rHr)r)r*getExpectedConnectionLostLogMsg8sz3TLSPortTestsBuilder.getExpectedConnectionLostLogMsgcCsdd}||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}. cSs|dt|S)Nr)rtrrYr)r)r*rQEsz2TLSPortTestsBuilder.test_badContext..useItNrZr[r)r)r*r\?sz#TLSPortTestsBuilder.test_badContextcCs||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 )rXZ LISTENER_HOSTrHr2)r.rEZaddressr>r)r)r*connectToListenerJs z%TLSPortTestsBuilder.connectToListenerN) r$r%r&rArurwrxr\ryr)r)r)r*rs"s  rsc@s,eZdZdZefZeZddZddZ dS)AbortSSLConnectionTestsz- C{abortConnection} tests using SSL. cs<t|ddlm}tfddd}||d|jS)Nr)_producer_helperscs d|S)Ngh㈵>)Z callLater)xrmr)r*rnyroz6AbortSSLConnectionTests.buildReactor..)Z scheduler cooperate)rrMZtwisted.internetr{rZpatchr})r.r{Z cooperatorr)rmr*rMrs   z$AbortSSLConnectionTests.buildReactorcCstdkrtddS)NzOpenSSL not available.)rrr1r)r)r*setUp~szAbortSSLConnectionTests.setUpN) r$r%r&rAr r'rVrUrMr~r)r)r)r*rzks  rz)DrAZ __future__rrtypeZ __metaclass__Zzope.interfacerZtwisted.python.compatrZtwisted.python.filepathrZ#twisted.internet.test.reactormixinsrZtwisted.internet.protocolrr r Ztwisted.internet.interfacesr r r Ztwisted.internet.deferrrZtwisted.internet.endpointsrrrZtwisted.internet.errorrZtwisted.internet.taskrZtwisted.trial.unittestrZtwisted.python.runtimerZtwisted.internet.test.test_corerZtwisted.internet.test.test_tcprrrZ&twisted.internet.test.connectionmixinsrrrZOpenSSL.cryptor ImportErrorZtwisted.internet.sslrr r"r#objectr+r4rBrLrTrVrWrsglobalsupdateZmakeTestCaseClassesrzr)r)r)r*s^            % _C