o bIr@sdZddlZddlmZddlmZddlmZddlm Z edZ ddl m Z e r3dd 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 D/usr/lib/python3/dist-packages/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:s zTestChannel.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=s zTestChannel.channelOpencCs ||_dS)zD Opening the channel failed. Store the reason why. N)openFailureReason)rreasonr r r openFailedJ zTestChannel.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.extReceivedcC d|_dS)z1 EOF was received. Remember it. TN)rrr r r eofReceiveder"zTestChannel.eofReceivedcCr,)z3 Close was received. Remember it. TN)rrr r r closeReceivedkr"zTestChannel.closeReceivedcCr,)z7 The channel is closed. Rembember it. TN)rrr r rclosedqr"zTestChannel.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|tjkr t||||dS|dkrt|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)rr1r ConchError_ARGS_ERROR_CODE)r channelType windowSize maxPacketr&r r r lookupChannels zTestAvatar.lookupChannelcCs |dkrdS|dkrd|fSdS)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. TestGlobalTTestDataFr )r requestTyper&r r rgotGlobalRequests zTestAvatar.gotGlobalRequestN)r r r r0r:r>rBr r r rr2xs  r2c@s@eZdZdZes dZddZddZddZd 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 cryptographycCdS)NrCr rr r rrszTestConnection.logPrefixcCrE)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. )r4r5r&)rrrr<r=r&r r rchannel_TestChannelsz"TestConnection.channel_TestChannelcCstd)zU The other side is requesting the ErrorChannel. Raise an exception. z no such thing)AssertionErrorrHr r rchannel_ErrorChannelsz#TestConnection.channel_ErrorChannelN) r r r r0rskiprrFrGrIrKr r r rrCs rCc@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)EConnectionTestsrDcCs6td|_t|j_t|_|j|j_|jdSN)r FakeTransport transportr2r6rCconnserviceStartedrr r rsetUps   zConnectionTests.setUpcCs@|j||jjdd|j_|jtd|jdddS)z? Open the channel with the default connection. Nz>2Ls)rQ openChannelrPpacketsssh_CHANNEL_OPEN_CONFIRMATIONstructpackrrrr r r _openChannels zConnectionTests._openChannelcCs|jdSrN)rQserviceStoppedrr r rtearDownzConnectionTests.tearDowncCs||jjj|jdS)z_ Test that the connection links itself to the avatar in the transport. N)assertIsrPr6rQrr 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) rrQrVrX assertTruer assertFalserr]twisted.internet.errorrbassertIsInstancer)rchannel1channel2rbr 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) rQssh_GLOBAL_REQUESTrNS assertEqualrPrWr MSG_REQUEST_SUCCESSMSG_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$)ro)r&rr rcheck"sz3ConnectionTests.test_REQUEST_SUCCESS..check)rQsendGlobalRequestssh_REQUEST_SUCCESS addCallback addErrbackfailrdrur 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. rsr$Tcs|jjddSrt)rovaluer&)frr rru1z3ConnectionTests.test_REQUEST_FAILURE..check)rQrvssh_REQUEST_FAILURErxrzryr{r 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 channelrks ErrorChannelssunknown failureN)rPr6rQssh_CHANNEL_OPENrrnrcrrror&rremoteWindowLeftr5rWr MSG_CHANNEL_OPEN_CONFIRMATIONflushLoggedErrorsMSG_CHANNEL_OPEN_FAILURErr r rtest_CHANNEL_OPEN8sV  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. r7rr#zExpected one error, got: r)r3r8s{serror args in wrong orderrkN)rPr6r:rQrrrnrrr9rolenr~argsrWr r)rr*errorsr r r_lookupChannelErrorTestls$   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. r3N)rrr 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) rrQrVrXrorr5rchannelsToRemoteChannellocalToRemoteChannelr[r 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) rrQrVssh_CHANNEL_OPEN_FAILURErrnrorr assertIsNonechannelsgetr[r 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)rr\rrQssh_CHANNEL_WINDOW_ADJUSTro)rr 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#rN)rr\rQssh_CHANNEL_DATArrnrorrPrWr MSG_CHANNEL_WINDOW_ADJUSTlocalWindowLeftMSG_CHANNEL_CLOSErrrlongDatabigDatar 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)rr\rQssh_CHANNEL_EXTENDED_DATArrnrorrPrWr 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)rr\rQssh_CHANNEL_EOFrcrr[r 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) rr\rcrrdrrrQ sendClosessh_CHANNEL_CLOSEr[r r rtest_CHANNEL_CLOSEs       z"ConnectionTests.test_CHANNEL_CLOSEcspt}|jdtdd|jdjdtddd}fdd}|||S) zS Test that channel requests that succeed send MSG_CHANNEL_SUCCESS. rtestrlr#rjr$cjjtjdfgdSNr)rorPrWr MSG_CHANNEL_SUCCESSresultrr rru2 z;ConnectionTests.test_CHANNEL_REQUEST_success..check) rr\rQssh_CHANNEL_REQUESTrrnrorrxrrr|rur 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. rrrjcrr)rorPrWr MSG_CHANNEL_FAILURErrr rruErz;ConnectionTests.test_CHANNEL_REQUEST_failure..check) rr\rQrrrnrxrzryrr 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|dSrN)rcrrr rruYr_z;ConnectionTests.test_CHANNEL_REQUEST_SUCCESS..check)rr\rQ sendRequestssh_CHANNEL_SUCCESSrr rrtest_CHANNEL_REQUEST_SUCCESSOs    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. rrkTrcs|jjddS)Nzchannel request failed)ror~rrr rruhrz;ConnectionTests.test_CHANNEL_REQUEST_FAILURE..check)rr\rQrssh_CHANNEL_FAILURErxrzryrr 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$TcSdSrNr failurer r ruz8ConnectionTests.test_sendGlobalRequest..snoReplyrkFsdatarlglobalN) rQrvryrorPrWr MSG_GLOBAL_REQUESTrrn deferredsrr|r r rtest_sendGlobalRequestosz&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) rrQrVrorPrWr MSG_CHANNEL_OPENrrnrlocalChannelIDr[r 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. rTcSrrNr rr r rrrz2ConnectionTests.test_sendRequest..test2rkFstest3rstestrlrN)rr\rQrry localClosedrorPrWr MSG_CHANNEL_REQUESTrrnrrrr|r 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) rr\rrQ adjustWindowrorrPrWr rr[r 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) rr\rQsendDatarrorPrWr MSG_CHANNEL_DATArrnr[r 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) rr\rQsendExtendedDatarrorPrWr MSG_CHANNEL_EXTENDED_DATArrnr[r r rtest_sendExtendedDatas  z%ConnectionTests.test_sendExtendedDatacCsbt}|||j|||jjtjdfgd|_ |j|||jjtjdfgdS)zN Test that channel EOF messages are sent in the right format. rTN) rr\rQsendEOFrorPrWr MSG_CHANNEL_EOFrr[r 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)rr\rQrrcrrorPrWr rrrdr remoteClosed)rrrhr 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$rN) rQ getChannelror&rr5 assertRaisesrr9r[r 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)rPr6rcrQrBrordrr 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# wantReply)rr\rQr assertFailurerr9 channelClosedrr r r9test_channelClosedCausesLeftoverChannelDeferredsToErrback)s   zIConnectionTests.test_channelClosedCausesLeftoverChannelDeferredsToErrbackN)&r r r rrLrSr\r^rarirrr}rrrrrrrrrrrrrrrrrrrrrrrrrrr r r rrMsH 4   %+  rMc@s(eZdZdZes dZddZddZdS)CleanConnectionShutdownTestszL Check whether correct cleanup is performed on connection shutdown. rDcCs,td|_t|j_t|_|j|j_dSrN)rrOrPr2r6rCrQrr r rrS?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)rQrRrvrrr9r]rr r r9test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackEs  zVCleanConnectionShutdownTests.test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackN)r r r r0rrLrSrr r r rr7s  r)r0rYtwisted.conch.sshrtwisted.conch.testrtwisted.python.reflectr twisted.trialrr twisted.conchrrr SSHChannelrr2rrCTestCaserMrr r r rs(     \),n