o >h)2 @s$dZddlmZmZmZddlmZmZddlm Z m Z ddl m Z m Z mZddlmZddlmZmZmZddlmZmZmZdd lmZmZmZdd lmZdd lm Z m!Z!m"Z"dd l#m$Z$dd l%m&Z&ddl'm(Z(ddl)m*Z*zddl+m,Z,Wn e-ydZ,Yn wddl.m/Z/m0Z0m1Z1GdddZ2GdddZ3eeGdddZ4Gdddee3Z5GdddZ6Gddde2eeZ7Gdd d ee3Z8Gd!d"d"e2ee3ee6Z9Gd#d$d$e2e3e6e!e"eZ:e;e;=dS)'z0 Tests for implementations of L{ITLSTransport}. )OptionalSequenceType) Interface implementer)Deferred DeferredList)SSL4ClientEndpointSSL4ServerEndpointTCP4ClientEndpoint)ConnectionClosed) IReactorSSLIStreamClientEndpoint ITLSTransport) ClientFactoryProtocol ServerFactory)BrokenContextFactoryConnectionTestsMixinEndpointCreator)ReactorBuilder)AbortConnectionMixinConnectToTCPListenerMixinStreamTransportTestsMixin) networkString)FilePath)platform)SkipTest) FILETYPE_PEMN)ClientContextFactoryKeyPairPrivateCertificatec@sBeZdZUegZeeeee d<e rdZ e e dZ dSdS)TLSMixinrequiredInterfaceszcFor 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 r#rrrr__annotations__r isWindowsmsgskippedReactorsr+r+~/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/internet/test/test_tls.pyr"2s  r"c@s@eZdZddlZeeejddZ [ddZ ddZ dS) ContextGeneratingMixinrNstests server.pemcCs(|j}t|t|tt}|S)zM Return a new SSL context suitable for use in a test server. )_pem getContentr!loadr roptions)selfpemcertr+r+r,getServerContextHs z'ContextGeneratingMixin.getServerContextcCstSN)rr2r+r+r,getClientContextRsz'ContextGeneratingMixin.getClientContext) r$r%r&twistedrr__file__siblingchildr.r5r8r+r+r+r,r-@s r-c@ 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||_||_dSr6)wrappedcontextFactory)r2r?r@r+r+r,__init__bs zStartTLSClientEndpoint.__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|dSr6) transportstartTLSr@)orig)protocolr2r+r,connectionMadess z\StartTLSClientEndpoint.connect..WrapperFactory.buildProtocol..connectionMade) buildProtocolrF) wrapperSelfaddrrFfactoryr2)rEr,rGps zDStartTLSClientEndpoint.connect..WrapperFactory.buildProtocolN)r$r%r&rGr+rJr+r,WrapperFactoryosrL)rr?connect)r2rKrLr+rJr,rMfs  zStartTLSClientEndpoint.connectN)r$r%r&__doc__rArMr+r+r+r,r>Vs r>c@r=)StartTLSClientCreatorz{ Create L{ITLSTransport.startTLS} endpoint for the client, and normal SSL for server just because it's easier. cCt|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. rr r5r2reactorr+r+r,serverszStartTLSClientCreator.servercCstt|d|jtS)zS Construct a TCP client endpoint wrapped to immediately start TLS. 127.0.0.1)r>r portrr2rS serverAddressr+r+r,clients zStartTLSClientCreator.clientNr$r%r&rNrTrYr+r+r+r,rO}s rOc@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) buildReactor assertRaises ValueErrorr assertEqualmessagestr)r2useItrSexcr+r+r,_testBadContexts z$BadContextTestsMixin._testBadContextN)r$r%r&rNrdr+r+r+r,r[s r[c@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&rNrO endpointsr+r+r+r,res rec@r=) SSLCreatorz Create SSL endpoints. cCrP)zQ Create an SSL server endpoint on a TCP/IP-stack allocated port. rrQrRr+r+r,rTszSSLCreator.servercCst|d|jtS)z Create an SSL client endpoint which will connect localhost on the port given by C{serverAddress}. @type serverAddress: L{IPv4Address} rU)r rVrrWr+r+r,rYs zSSLCreator.clientNrZr+r+r+r,rgs rgc@s&eZdZdZeZddZddZdS)SSLClientTestsMixinz< Mixin defining tests relating to L{ITLSTransport}. cCdd}||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)NrUi) connectSSLrrSr@r+r+r,rbs z2SSLClientTestsMixin.test_badContext..useItNrdr2rbr+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 providedByrBrKfinishederrbackrrCcontextwrite)r2rqr+r+r,rFs z`SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS..ShortProtocol.connectionMadecSs|jd|jdS)Ny)rBrtloseConnection)r2datar+r+r, dataReceiveds z^SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS..ShortProtocol.dataReceivedcSs*|jj}|durd|j_||dSdSr6)rKrqcallback)r2reasonrqr+r+r,connectionLosts z`SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS..ShortProtocol.connectionLostN)r$r%r&rFrxr{r+r+r+r, ShortProtocols r|T) consumeErrorscs"|dd|ddgdS)Nr)extend)results)lostConnectionResultsr+r, cbFinished"s"zNSSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS..cbFinishedrrU) interfacecsSr6)stop)ign)rSr+r,/szLSSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLS..r~N)rr\rrrqrEr5rsrr8methodr addCallback listenTCP addCleanup stopListening connectTCPgetHosthostrV disconnect runReactortrapr )r2r| serverFactory clientFactoryrqrrV connectorr+)rrSr,&test_disconnectAfterWriteAfterStartTLSs6 !         z:SSLClientTestsMixin.test_disconnectAfterWriteAfterStartTLSN)r$r%r&rNrgrfrnrr+r+r+r,rhs  rhc@s8eZdZdZddZddZddZdd Zd d Zd S) TLSPortTestsBuilderz, Tests for L{IReactorSSL.listenSSL} cCs|d||S)z0 Get a TLS port from a reactor. r) listenSSLr5)r2rSrKr+r+r,getListeningPortAz$TLSPortTestsBuilder.getListeningPortcCsd||jfS)zY Get the message expected to be logged when a TLS port starts listening. z%s (TLS) starting on %drrV)r2rVrKr+r+r,#getExpectedStartListeningLogMessageGrz7TLSPortTestsBuilder.getExpectedStartListeningLogMessagecCsd|jdS)zJ Get the expected connection lost message for a TLS port. z (TLS Port z Closed)r)r2rVr+r+r,getExpectedConnectionLostLogMsgMrz3TLSPortTestsBuilder.getExpectedConnectionLostLogMsgcCri)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)rrrkr+r+r,rbZsz2TLSPortTestsBuilder.test_badContext..useItNrlrmr+r+r,rnSsz#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 )rj LISTENER_HOSTrVr8)r2rSaddressrKr+r+r,connectToListener_s z%TLSPortTestsBuilder.connectToListenerN) r$r%r&rNrrrrnrr+r+r+r,r5s rc@s$eZdZdZefZeZddZdS)AbortSSLConnectionTestsz- C{abortConnection} tests using SSL. cCstdurtddS)NzOpenSSL not available.)rrr7r+r+r,setUpszAbortSSLConnectionTests.setUpN) r$r%r&rNr r#rgrfrr+r+r+r,r~s  r)?rNtypingrrrzope.interfacerrtwisted.internet.deferrrtwisted.internet.endpointsr r r twisted.internet.errorr twisted.internet.interfacesr rrtwisted.internet.protocolrrr&twisted.internet.test.connectionmixinsrrr#twisted.internet.test.reactormixinsrtwisted.internet.test.test_tcprrrtwisted.python.compatrtwisted.python.filepathrtwisted.python.runtimertwisted.trial.unittestrOpenSSL.cryptor ImportErrortwisted.internet.sslrr r!r"r-r>rOr[rergrhrglobalsupdatemakeTestCaseClassesrr+r+r+r,sf       &  jD