U “jHr@sdZddlZddlmZddlmZddlmZddlm Z edZ ddl m Z e rfdd lm Z mZnGd d d ZGd d d ejZGdddZGdddejZGddde jZGddde jZdS)z1 This module tests twisted.conch.ssh.connection. N)channel) test_userauth) requireModule)unittest cryptography)error)common connectionc@seZdZGdddZdS)r c@s eZdZdS)zconnection.SSHConnectionN)__name__ __module__ __qualname__r r F/tmp/pip-unpacked-wheel-59kc99xd/twisted/conch/test/test_connection.py SSHConnectionsrN)r r r rr r r rr sr c@sdeZdZdZdZdZdZddZddZdd Z d d Z d d Z ddZ ddZ ddZddZdS) TestChannela A mocked-up version of twisted.conch.ssh.channel.SSHChannel. @ivar gotOpen: True if channelOpen has been called. @type gotOpen: L{bool} @ivar specificData: the specific channel open data passed to channelOpen. @type specificData: L{bytes} @ivar openFailureReason: the reason passed to openFailed. @type openFailed: C{error.ConchError} @ivar inBuffer: a C{list} of strings received by the channel. @type inBuffer: C{list} @ivar extBuffer: a C{list} of 2-tuples (type, extended data) of received by the channel. @type extBuffer: C{list} @ivar numberRequests: the number of requests that have been made to this channel. @type numberRequests: L{int} @ivar gotEOF: True if the other side sent EOF. @type gotEOF: L{bool} @ivar gotOneClose: True if the other side closed the connection. @type gotOneClose: L{bool} @ivar gotClosed: True if the channel is closed. @type gotClosed: L{bool} TestChannelFcCs d|jS)NzTestChannel %i)idselfr r r logPrefix:szTestChannel.logPrefixcCs4d|_||_g|_g|_d|_d|_d|_d|_dS)zF The channel is open. Set up the instance variables. TrFN)gotOpen specificDatainBuffer extBuffernumberRequestsgotEOF gotOneClose gotClosed)rrr r r channelOpen=szTestChannel.channelOpencCs ||_dS)zD Opening the channel failed. Store the reason why. N)openFailureReason)rreasonr r r openFailedJszTestChannel.openFailedcCs|jd7_|dkS)z_ A test request. Return True if data is 'data'. @type data: L{bytes} data)rrdatar r r request_testPszTestChannel.request_testcCs|j|dS)z= Data was received. Store it in the buffer. N)rappendr$r r r dataReceivedYszTestChannel.dataReceivedcCs|j||fdS)zF Extended data was received. Store it in the buffer. N)rr')rcoder%r r r extReceived_szTestChannel.extReceivedcCs d|_dS)z1 EOF was received. Remember it. TN)rrr r r eofReceivedeszTestChannel.eofReceivedcCs d|_dS)z3 Close was received. Remember it. TN)rrr r r closeReceivedkszTestChannel.closeReceivedcCs d|_dS)z7 The channel is closed. Rembember it. TN)rrr r rclosedqszTestChannel.closedN)r r r __doc__namerrrrr!r&r(r*r+r,r-r r r rrs  rc@s$eZdZdZdZddZddZdS) TestAvatarz? A mocked-up version of twisted.conch.avatar.ConchUser {cCs4|tjkrt||||dS|dkr0t|jddS)z The server wants us to return a channel. If the requested channel is our TestChannel, return it, otherwise return None. ) remoteWindowremoteMaxPacketr%avatarconch-error-argserror args in wrong orderN)rr/r ConchError_ARGS_ERROR_CODE)rZ channelType windowSize maxPacketr%r r r lookupChannels zTestAvatar.lookupChannelcCs$|dkr dS|dkrd|fSdSdS)z The client has made a global request. If the global request is 'TestGlobal', return True. If the global request is 'TestData', return True and the request-specific data we received. Otherwise, return False. TestGlobalTTestDataFNr )rZ requestTyper%r r rgotGlobalRequests zTestAvatar.gotGlobalRequestN)r r r r.r8r;r>r r r rr0xsr0c@s@eZdZdZesdZddZddZddZd d Z d d Z d S)TestConnectionz} A subclass of SSHConnection for testing. @ivar channel: the current channel. @type channel. C{TestChannel} Cannot run without cryptographycCsdS)Nr?r rr r rrszTestConnection.logPrefixcCsdS)zT The other side made the 'TestGlobal' global request. Return True. Tr r$r r rglobal_TestGlobalsz TestConnection.global_TestGlobalcCsd|fS)zt The other side made the 'Test-Data' global request. Return True and the data we received. Tr r$r r rglobal_Test_DataszTestConnection.global_Test_DatacCst|||d|_|jS)z The other side is requesting the TestChannel. Create a C{TestChannel} instance, store it, and return it. )r2r3r%)rrrr9r:r%r r rchannel_TestChannels z"TestConnection.channel_TestChannelcCs tddS)zU The other side is requesting the ErrorChannel. Raise an exception. z no such thingN)AssertionErrorrCr r rchannel_ErrorChannelsz#TestConnection.channel_ErrorChannelN) r r r r.rskiprrArBrDrFr r r rr?s r?c@seZdZesdZddZddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Z d:d;Z!dd?Z#d@dAZ$dBdCZ%dDS)EConnectionTestsr@cCs6td|_t|j_t|_|j|j_|jdSN)r FakeTransport transportr0r4r?connserviceStartedrr r rsetUps    zConnectionTests.setUpcCs@|j||jjdd|j_|jtd|jdddS)z? Open the channel with the default connection. Nz>2Ls)rL openChannelrKpacketsssh_CHANNEL_OPEN_CONFIRMATIONstructpackrrrr r r _openChannels  zConnectionTests._openChannelcCs|jdSrI)rLserviceStoppedrr r rtearDownszConnectionTests.tearDowncCs||jjj|jdS)z_ Test that the connection links itself to the avatar in the transport. N)ZassertIsrKr4rLrr r rtest_linkAvatarszConnectionTests.test_linkAvatarcCst}t}|j||j||jd||j||j||j||j|j||j||j||jddl m }| |j |dS)zF Test that serviceStopped() closes any open channels. sr)ConnectionLostN) rrLrQrS assertTruer assertFalserrXZtwisted.internet.errorr[ZassertIsInstancer)rZchannel1channel2r[r r rtest_serviceStoppeds            z#ConnectionTests.test_serviceStoppedcCs|jtdd||jjtjdfgg|j_|jtddd||jjtjdfgg|j_|jtdd||jjtj dfgg|j_|jtdd||jjgdS) z Test that global request packets are dispatched to the global_* methods and the return values are translated into success or failure messages. r<r=s test datasTestBadN) rLZssh_GLOBAL_REQUESTrNS assertEqualrKrRr ZMSG_REQUEST_SUCCESSZMSG_REQUEST_FAILURErr r rtest_GLOBAL_REQUESTs(   z#ConnectionTests.test_GLOBAL_REQUESTcsBjddd}jdfdd}|||j|S)zh Test that global request success packets cause the Deferred to be called back. requestr#Tcs|ddSNr#)rd)r%rr rcheck!sz3ConnectionTests.test_REQUEST_SUCCESS..check)rLsendGlobalRequestZssh_REQUEST_SUCCESS addCallback addErrbackfailrdrhr rrtest_REQUEST_SUCCESSs     z$ConnectionTests.test_REQUEST_SUCCESScsBjddd}jdfdd}|j|||S)zg Test that global request failure packets cause the Deferred to be erred back. rfr#Tcs|jjddSrg)rdvaluer%)frr rrh0sz3ConnectionTests.test_REQUEST_FAILURE..check)rLriZssh_REQUEST_FAILURErjrlrkrmr rrtest_REQUEST_FAILURE(s     z$ConnectionTests.test_REQUEST_FAILUREcCsH|j`|jtdd||jjj| |jjj|j| |jjj d| |jjj d| |jjj d| |jjj d| |jjtjdfgg|j_|jtdd|| |jjtjdtd td fgg|j_|jtd d|| |jjtjd td td fgdS)z Test that open channel packets cause a channel to be created and opened or a failure message to be returned. rr"s BadChannelsssunknown channelras ErrorChannelssunknown failureN)rKr4rLssh_CHANNEL_OPENrrcr\rrrdr%rremoteWindowLeftr3rRr ZMSG_CHANNEL_OPEN_CONFIRMATIONflushLoggedErrorsMSG_CHANNEL_OPEN_FAILURErr r rtest_CHANNEL_OPEN7sV  z!ConnectionTests.test_CHANNEL_OPENcCs||jj_|jtdd|tj }| t |dd|| |dj j d| |jjtjdtdtd fgd S) z Deliver a request for a channel open which will result in an exception being raised during channel lookup. Assert that an error response is delivered as a result. r5rsr"zExpected one error, got: r)r1r6s{serror args in wrong orderraN)rKr4r8rLrvrrcrxrr7rdlenrpargsrRr ry)rr)errorsr r r_lookupChannelErrorTestks$   z'ConnectionTests._lookupChannelErrorTestcCs|ddS)a If a C{lookupChannel} implementation raises L{error.ConchError} with the arguments in the wrong order, a C{MSG_CHANNEL_OPEN} failure is still sent in response to the message. This is a temporary work-around until L{error.ConchError} is given better attributes and all of the Conch code starts constructing instances of it properly. Eventually this functionality should be deprecated and then removed. r1N)r~rr r rtest_lookupChannelErrors z'ConnectionTests.test_lookupChannelErrorcCstt}|j||jd||jd||jd||jd||jj|d||jj dddS)zv Test that channel open confirmation packets cause the channel to be notified that it's open. srN) rrLrQrSrdrwr3rZchannelsToRemoteChannelZlocalToRemoteChannelrVr r rtest_CHANNEL_OPEN_CONFIRMATIONs  z.ConnectionTests.test_CHANNEL_OPEN_CONFIRMATIONcCsPt}|j||jdtd||jjd| |jj |dS)zz Test that channel open failure packets cause the channel to be notified that its opening failed. failure!)rr"N) rrLrQZssh_CHANNEL_OPEN_FAILURErrcrdrr|Z assertIsNoneZchannelsgetrVr r rtest_CHANNEL_OPEN_FAILUREs  z)ConnectionTests.test_CHANNEL_OPEN_FAILUREcCs8t}|||j}|jd||j|ddS)zc Test that channel window adjust messages add bytes to the channel window. rr"N)rrWrwrLZssh_CHANNEL_WINDOW_ADJUSTrd)rrZ oldWindowSizer r rtest_CHANNEL_WINDOW_ADJUSTs   z*ConnectionTests.test_CHANNEL_WINDOW_ADJUSTcCstddd}|||jdtd||jdg||jj t j dfgg|j_ d|j d}|jdt|||jdg||jj t j d fgt}||d|jd}g|j_ |jd t|||jg||jj t j d fgd S) z Test that channel data messages are passed up to the channel, or cause the channel to be closed if the data is too large.  localWindowlocalMaxPacketrr#ar"rtN)rrWrLZssh_CHANNEL_DATArrcrdrrKrRr MSG_CHANNEL_WINDOW_ADJUSTlocalWindowLeftMSG_CHANNEL_CLOSErrrZlongDataZbigDatar r rtest_CHANNEL_DATAs:      z!ConnectionTests.test_CHANNEL_DATAcCstddd}|||jdtd||jdg||jj t j dfgg|j_ d|j d }|jdt|||jdg||jj t j d fgt}||d|jd }g|j_ |jd t|||jg||jj t j d fgd S) z Test that channel extended data messages are passed up to the channel, or cause the channel to be closed if they're too big. rrrsr#)rr#rrr"rsN)rrWrLZssh_CHANNEL_EXTENDED_DATArrcrdrrKrRr rrrrrr r rtest_CHANNEL_EXTENDED_DATAsF         z*ConnectionTests.test_CHANNEL_EXTENDED_DATAcCs,t}|||jd||jdS)zN Test that channel eof messages are passed up to the channel. rN)rrWrLZssh_CHANNEL_EOFr\rrVr r rtest_CHANNEL_EOF s  z ConnectionTests.test_CHANNEL_EOFcCsht}||||j||j||j|j||j d||j||jdS)z Test that channel close messages are passed up to the channel. Also, test that channel.close() is called if both sides are closed when this message is received. rN) rrWr\rr]rrrL sendCloseZssh_CHANNEL_CLOSErVr r rtest_CHANNEL_CLOSEs       z"ConnectionTests.test_CHANNEL_CLOSEcspt}|jdtdd|jdjdtddd}fdd}|||S) zS Test that channel requests that succeed send MSG_CHANNEL_SUCCESS. rtestrbr"r`r#csjjtjdfgdSNr)rdrKrRr ZMSG_CHANNEL_SUCCESSresultrr rrh1s z;ConnectionTests.test_CHANNEL_REQUEST_success..check) rrWrLssh_CHANNEL_REQUESTrrcrdrrjrrrnrhr rrtest_CHANNEL_REQUEST_success#s   z,ConnectionTests.test_CHANNEL_REQUEST_successcsPt}|jdtdd}fdd}|j|||S)zP Test that channel requests that fail send MSG_CHANNEL_FAILURE. rrr`csjjtjdfgdSr)rdrKrRr ZMSG_CHANNEL_FAILURErrr rrhDs z;ConnectionTests.test_CHANNEL_REQUEST_failure..check) rrWrLrrrcrjrlrkrr rrtest_CHANNEL_REQUEST_failure:s    z,ConnectionTests.test_CHANNEL_REQUEST_failurecs>t}|j|ddd}jdfdd}|S)zj Test that channel request success messages cause the Deferred to be called back. rr#Trcs|dSrI)r\rrr rrhXsz;ConnectionTests.test_CHANNEL_REQUEST_SUCCESS..check)rrWrL sendRequestZssh_CHANNEL_SUCCESSrr rrtest_CHANNEL_REQUEST_SUCCESSNs    z,ConnectionTests.test_CHANNEL_REQUEST_SUCCESScsTt}|j|ddd}jdfdd}|j|||S)zi Test that channel request failure messages cause the Deferred to be erred back. rraTrcs|jjddS)Nzchannel request failed)rdrprrr rrhgsz;ConnectionTests.test_CHANNEL_REQUEST_FAILURE..check)rrWrLrZssh_CHANNEL_FAILURErjrlrkrr rrtest_CHANNEL_REQUEST_FAILURE]s     z,ConnectionTests.test_CHANNEL_REQUEST_FAILUREcCs||jddd}|dd|jddd||jjtjt dd ftjt dd fg||jj d |gid S) zQ Test that global request messages are sent in the right format. s wantReplyr#TcSsdSrIr Zfailurer r rtraz8ConnectionTests.test_sendGlobalRequest..snoReplyraFsdatarbglobalN) rLrirkrdrKrRr ZMSG_GLOBAL_REQUESTrrc deferredsrrnr r rtest_sendGlobalRequestnsz&ConnectionTests.test_sendGlobalRequestcCsXt}|j|d||jjtjt ddfg||j d||jj ddS)zO Test that open channel messages are sent in the right format. saaaarsaaaarr"N) rrLrQrdrKrRr ZMSG_CHANNEL_OPENrrcrZlocalChannelIDrVr r rtest_openChannels z ConnectionTests.test_openChannelc Cst}|||j|ddd}|dd|j|dddd|_|j|ddd||jjt j d t dd ft j d t dd fg||jj d |gd S)zR Test that channel request messages are sent in the right format. rTcSsdSrIr rr r rrraz2ConnectionTests.test_sendRequest..test2raFstest3rstestrbrN)rrWrLrrk localClosedrdrKrRr ZMSG_CHANNEL_REQUESTrrcrrrrnr r rtest_sendRequests$  z ConnectionTests.test_sendRequestcCsttdd}||d|_|j|d||jdd|_|j|d||jd||jjt j dfgdS) zi Test that channel window adjust messages cause bytes to be added to the window. r)rrr"TN) rrWrrLZ adjustWindowrdrrKrRr rrVr r rtest_adjustWindows  z!ConnectionTests.test_adjustWindowcCsXt}|||j|dd|_|j|d||jjtj dt dfgdS)zO Test that channel data messages are sent in the right format. rTbrN) rrWrLZsendDatarrdrKrRr ZMSG_CHANNEL_DATArrcrVr r r test_sendDatas zConnectionTests.test_sendDatacCs\t}|||j|ddd|_|j|dd||jjtj dt dfgdS)zX Test that channel extended data messages are sent in the right format. r"rTrrrN) rrWrLZsendExtendedDatarrdrKrRr ZMSG_CHANNEL_EXTENDED_DATArrcrVr r rtest_sendExtendedDatas  z%ConnectionTests.test_sendExtendedDatacCsbt}|||j|||jjtjdfgd|_ |j|||jjtjdfgdS)zN Test that channel EOF messages are sent in the right format. rTN) rrWrLZsendEOFrdrKrRr ZMSG_CHANNEL_EOFrrVr r r test_sendEOFs     zConnectionTests.test_sendEOFcCst}|||j|||j||jjt j dfg|j|||jjt j dfgt}||||j | |j d|_|j|||j dS)zP Test that channel close messages are sent in the right format. rTN)rrWrLrr\rrdrKrRr rrr]rZ remoteClosed)rrr^r r rtest_sendCloses(          zConnectionTests.test_sendClosecCsZ|jdddd}||jd||jd||jd|tj|jjdddddS)z Test that getChannel dispatches to the avatar when an avatar is present. Correct functioning without the avatar is verified in test_CHANNEL_OPEN. r2r#ruN) rLZ getChannelrdr%rwr3 assertRaisesrr7rVr r rtest_getChannelWithAvatarsz)ConnectionTests.test_getChannelWithAvatarcCsH|j`||jdd||jddd||jdddS)zW Test that gotGlobalRequests dispatches to global_* without an avatar. r<r#s Test-Data)Tr#s BadGlobalN)rKr4r\rLr>rdr]rr r r"test_gotGlobalRequestWithoutAvatars z2ConnectionTests.test_gotGlobalRequestWithoutAvatarcCsBt}|||jj|dddd}||tj}|j||S)z Whenever an SSH channel gets closed any Deferred that was returned by a sendRequest() on its parent connection must be errbacked. dummyrequest dummydatar"Z wantReply)rrWrLr assertFailurerr7Z channelClosedrr r r9test_channelClosedCausesLeftoverChannelDeferredsToErrback(s   zIConnectionTests.test_channelClosedCausesLeftoverChannelDeferredsToErrbackN)&r r r rrGrNrWrYrZr_rerorrrzr~rrrrrrrrrrrrrrrrrrrrrrrr r r rrHsF 4   %+  rHc@s(eZdZdZesdZddZddZdS)CleanConnectionShutdownTestszL Check whether correct cleanup is performed on connection shutdown. r@cCs,td|_t|j_t|_|j|j_dSrI)rrJrKr0r4r?rLrr r rrN>s  z"CleanConnectionShutdownTests.setUpcCs8|j|jjdddd}||tj}|j|S)z Once the service is stopped any leftover global deferred returned by a sendGlobalRequest() call must be errbacked. rrr"r)rLrMrirrr7rXrr r r9test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackDs   zVCleanConnectionShutdownTests.test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackN)r r r r.rrGrNrr r r rr6s r)r.rTZtwisted.conch.sshrZtwisted.conch.testrZtwisted.python.reflectrZ twisted.trialrrZ twisted.conchrrr Z SSHChannelrr0rr?ZTestCaserHrr r r rs&     \),m