U n5{iHrć@sĪdZddlZddlmZddlmZddlmZddlm Z edƒZ ddl m Z e rfdd lm Z mZnGd d „d ƒZGd d „d ejƒZGdd„dƒZGdd„dejƒZGdd„de jƒZGdd„de jƒZdS)z1 This module tests twisted.conch.ssh.connection. éN)Śchannel)Ś test_userauth)Ś requireModule)ŚunittestŚ cryptography)Śerror)ŚcommonŚ connectionc@seZdZGdd„dƒZdS)r c@s eZdZdS)zconnection.SSHConnectionN)Ś__name__Ś __module__Ś __qualname__©r r śF/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/conch/test/test_connection.pyŚ SSHConnectionsrN)r r r rr r r rr sr c@sdeZdZdZdZdZdZdd„Zdd„Zdd „Z d d „Z d d „Z dd„Z dd„Z dd„Zdd„ZdS)Ś 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)Śid©Śselfr r rŚ logPrefix:szTestChannel.logPrefixcCs4d|_||_g|_g|_d|_d|_d|_d|_dS)zF The channel is open. Set up the instance variables. TrFN)ŚgotOpenŚ specificDataŚinBufferŚ extBufferŚnumberRequestsŚgotEOFŚ gotOneCloseŚ gotClosed)rrr r rŚ channelOpen=szTestChannel.channelOpencCs ||_dS)zD Opening the channel failed. Store the reason why. N)ŚopenFailureReason)rŚreasonr r rŚ openFailedJszTestChannel.openFailedcCs|jd7_|dkS)z_ A test request. Return True if data is 'data'. @type data: L{bytes} éódata)r©rŚdatar r rŚ request_testPszTestChannel.request_testcCs|j |”dS)z= Data was received. Store it in the buffer. N)rŚappendr$r r rŚ dataReceivedYszTestChannel.dataReceivedcCs|j ||f”dS)zF Extended data was received. Store it in the buffer. N)rr')rŚcoder%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 rŚclosedqszTestChannel.closedN)r r r Ś__doc__Śnamerrrrr!r&r(r*r+r,r-r r r rrs  rc@s$eZdZdZdZdd„Zdd„ZdS)Ś TestAvatarz? A mocked-up version of twisted.conch.avatar.ConchUser é{cCs4|tjkrt||||dS|dkr0t |jd”‚dS)z’ The server wants us to return a channel. If the requested channel is our TestChannel, return it, otherwise return None. )Ś remoteWindowŚremoteMaxPacketr%Śavataróconch-error-argsśerror 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. ó TestGlobalTóTestDataFNr )rZ requestTyper%r r rŚgotGlobalRequest’s zTestAvatar.gotGlobalRequestN)r r r r.r8r;r>r r r rr0xsr0c@s@eZdZdZesdZdd„Zdd„Zdd„Zd 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 rr¬szTestConnection.logPrefixcCsdS)zT The other side made the 'TestGlobal' global request. Return True. Tr r$r r rŚglobal_TestGlobalÆsz 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 rŚglobal_Test_DataµszTestConnection.global_Test_DatacCst|||d|_|jS)zƒ The other side is requesting the TestChannel. Create a C{TestChannel} instance, store it, and return it. )r2r3r%)rr©rr9r:r%r r rŚchannel_TestChannel¼s ’z"TestConnection.channel_TestChannelcCs tdƒ‚dS)zU The other side is requesting the ErrorChannel. Raise an exception. z no such thingN)ŚAssertionErrorrCr r rŚchannel_ErrorChannelĘsz#TestConnection.channel_ErrorChannelN) r r r r.rŚskiprrArBrDrFr r r rr?”s r?c@seZdZesdZdd„Zdd„Zdd„Zdd „Zd d „Z d d „Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0d1„Zd2d3„Zd4d5„Zd6d7„Zd8d9„Z d:d;„Z!dd?„Z#d@dA„Z$dBdC„Z%dDS)EŚConnectionTestsr@cCs6t d”|_tƒ|j_tƒ|_|j|j_|j ”dS©N)rŚ FakeTransportŚ transportr0r4r?ŚconnŚserviceStartedrr r rŚsetUpŃs    zConnectionTests.setUpcCs@|j |”|jjdd…|j_|j t d|jd”d”dS)z? Open the channel with the default connection. Né’’’’z>2Lé’s€)rLŚ openChannelrKŚpacketsŚssh_CHANNEL_OPEN_CONFIRMATIONŚstructŚpackr©rrr r rŚ _openChannelŲs  ’zConnectionTests._openChannelcCs|j ”dSrI)rLŚserviceStoppedrr r rŚtearDownāszConnectionTests.tearDowncCs| |jjj|j”dS)z_ Test that the connection links itself to the avatar in the transport. N)ZassertIsrKr4rLrr r rŚtest_linkAvataråszConnectionTests.test_linkAvatarcCs¬tƒ}tƒ}|j |”|j |”|j d”| |j”| |j”| |j”| |j”|j ”| |j”| |j”| |j”ddl m }|  |j |”dS)zF Test that serviceStopped() closes any open channels. sr)ŚConnectionLostN) rrLrQrSŚ assertTruerŚ assertFalserrXZtwisted.internet.errorr[ZassertIsInstancer)rZchannel1Śchannel2r[r r rŚtest_serviceStoppedģs            z#ConnectionTests.test_serviceStoppedcCsŠ|j t d”d”| |jjtjdfg”g|j_|j t d”dd”| |jjtjdfg”g|j_|j t d”d”| |jjtj dfg”g|j_|j t d”d”| |jjg”dS) 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 datasTestBadóN) rLZssh_GLOBAL_REQUESTrŚNSŚ assertEqualrKrRr ZMSG_REQUEST_SUCCESSZMSG_REQUEST_FAILURErr r rŚtest_GLOBAL_REQUESTs( ’ ’ ’z#ConnectionTests.test_GLOBAL_REQUESTcsBˆj ddd”}ˆj d”‡fdd„}| |”| ˆj”|S)zh Test that global request success packets cause the Deferred to be called back. órequestr#Tcsˆ |d”dS©Nr#)rd)r%rr rŚcheck!sz3ConnectionTests.test_REQUEST_SUCCESS..check)rLŚsendGlobalRequestZssh_REQUEST_SUCCESSŚ addCallbackŚ addErrbackŚfail©rŚdrhr rrŚtest_REQUEST_SUCCESSs     z$ConnectionTests.test_REQUEST_SUCCESScsBˆj ddd”}ˆj d”‡fdd„}| ˆj”| |”|S)zg Test that global request failure packets cause the Deferred to be erred back. rfr#Tcsˆ |jjd”dSrg)rdŚvaluer%)Śfrr rrh0sz3ConnectionTests.test_REQUEST_FAILURE..check)rLriZssh_REQUEST_FAILURErjrlrkrmr rrŚtest_REQUEST_FAILURE(s     z$ConnectionTests.test_REQUEST_FAILUREcCsH|j`|j t d”d”| |jjj”|  |jjj|j”|  |jjj d”|  |jjj d”|  |jjj d”|  |jjj d”|  |jjtjdfg”g|j_|j t d”d”| ”|  |jjtjdt d ”t d ”fg”g|j_|j t d ”d”| ”|  |jjtjd t d ”t d ”fg”dS)z† Test that open channel packets cause a channel to be created and opened or a failure message to be returned. róór"s€ó BadChannelsssunknown channelras ErrorChannelssunknown failureN)rKr4rLŚssh_CHANNEL_OPENrrcr\rrrdr%rŚremoteWindowLeftr3rRr ZMSG_CHANNEL_OPEN_CONFIRMATIONŚflushLoggedErrorsŚMSG_CHANNEL_OPEN_FAILURErr r rŚtest_CHANNEL_OPEN7sVž’ž ’žž’ž ’žž’žz!ConnectionTests.test_CHANNEL_OPENcCsˆ||jj_|j t d”d”| tj ”}|  t |ƒdd|›”|  |dj j d”|  |jjtjdt d”t d ”fg”d 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)rKr4r8rLrvrrcrxrr7rdŚlenrpŚargsrRr ry)rr)Śerrorsr r rŚ_lookupChannelErrorTestks$  ’ ’žū’žz'ConnectionTests._lookupChannelErrorTestcCs| d”dS)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 rŚtest_lookupChannelError‡s z'ConnectionTests.test_lookupChannelErrorcCsttƒ}|j |”|j d”| |jd”| |jd”| |jd”| |jj|d”| |jj dd”dS)zv Test that channel open confirmation packets cause the channel to be notified that it's open. sróN) rrLrQrSrdrwr3rZchannelsToRemoteChannelZlocalToRemoteChannelrVr r rŚtest_CHANNEL_OPEN_CONFIRMATION”s  z.ConnectionTests.test_CHANNEL_OPEN_CONFIRMATIONcCsPtƒ}|j |”|j dt d””| |jjd”|  |jj   |””dS)zz Test that channel open failure packets cause the channel to be notified that its opening failed. óófailure!)rƒr"N) rrLrQZssh_CHANNEL_OPEN_FAILURErrcrdrr|Z assertIsNoneZchannelsŚgetrVr r rŚtest_CHANNEL_OPEN_FAILURE¢s  ’z)ConnectionTests.test_CHANNEL_OPEN_FAILUREcCs8tƒ}| |”|j}|j d”| |j|d”dS)zc Test that channel window adjust messages add bytes to the channel window. r‚r"N)rrWrwrLZssh_CHANNEL_WINDOW_ADJUSTrd)rrZ oldWindowSizer r rŚtest_CHANNEL_WINDOW_ADJUSTÆs   z*ConnectionTests.test_CHANNEL_WINDOW_ADJUSTcCstddd}| |”|j dt d””| |jdg”| |jj t j dfg”g|j_ d|j d}|j dt |””| |jdg”| |jj t j d fg”tƒ}| |”d|jd}g|j_ |j d t |””| |jg”| |jj t j d fg”d 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. éé©Ś localWindowŚlocalMaxPacketr€r#ó’óar"ó’rtN)rrWrLZssh_CHANNEL_DATArrcrdrrKrRr ŚMSG_CHANNEL_WINDOW_ADJUSTŚlocalWindowLeftŚMSG_CHANNEL_CLOSEr‹©rrZlongDataZbigDatar r rŚtest_CHANNEL_DATAŗs:  ž’ž  ž  žz!ConnectionTests.test_CHANNEL_DATAcCstddd}| |”|j dt d””| |jdg”| |jj t j dfg”g|j_ d|j d }|j dt |””| |jdg”| |jj t j d fg”tƒ}| |”d|jd }g|j_ |j d t |””| |jg”| |jj t j d fg”d 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. r‡rˆr‰sr#)rr#rŒrr"rŽsN)rrWrLZssh_CHANNEL_EXTENDED_DATArrcrdrrKrRr rrr‘r‹r’r r rŚtest_CHANNEL_EXTENDED_DATAßsF   ’ž’ž  ’ ž  ’ žz*ConnectionTests.test_CHANNEL_EXTENDED_DATAcCs,tƒ}| |”|j d”| |j”dS)zN Test that channel eof messages are passed up to the channel. r€N)rrWrLZssh_CHANNEL_EOFr\rrVr r rŚtest_CHANNEL_EOF s  z ConnectionTests.test_CHANNEL_EOFcCshtƒ}| |”| |j”| |j”| |j”|j |”|j  d”| |j”| |j”dS)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. r€N) rrWr\rr]rrrLŚ sendCloseZssh_CHANNEL_CLOSErVr r rŚtest_CHANNEL_CLOSEs       z"ConnectionTests.test_CHANNEL_CLOSEcsptƒ}ˆ |”ˆj dt d”d”ˆ |jd”ˆj dt d”dd”}‡fdd„}| |”|S) zS Test that channel requests that succeed send MSG_CHANNEL_SUCCESS. r€ótestrbr"r`r#csˆ ˆjjtjdfg”dS©NrŽ)rdrKrRr ZMSG_CHANNEL_SUCCESS©Śresultrr rrh1s žz;ConnectionTests.test_CHANNEL_REQUEST_success..check) rrWrLŚssh_CHANNEL_REQUESTrrcrdrrj©rrrnrhr rrŚtest_CHANNEL_REQUEST_success#s ’’  z,ConnectionTests.test_CHANNEL_REQUEST_successcsPtƒ}ˆ |”ˆj dt d”d”}‡fdd„}| ˆj”| |”|S)zP Test that channel requests that fail send MSG_CHANNEL_FAILURE. r€r˜r`csˆ ˆjjtjdfg”dSr™)rdrKrRr ZMSG_CHANNEL_FAILUREršrr rrhDs žz;ConnectionTests.test_CHANNEL_REQUEST_failure..check) rrWrLrœrrcrjrlrkrr rrŚtest_CHANNEL_REQUEST_failure:s ’   z,ConnectionTests.test_CHANNEL_REQUEST_failurecs>tƒ}ˆ |”ˆj |ddd”}ˆj d”‡fdd„}|S)zj Test that channel request success messages cause the Deferred to be called back. r˜r#Tr€csˆ |”dSrI)r\ršrr rrhXsz;ConnectionTests.test_CHANNEL_REQUEST_SUCCESS..check)rrWrLŚ sendRequestZssh_CHANNEL_SUCCESSrr rrŚtest_CHANNEL_REQUEST_SUCCESSNs    z,ConnectionTests.test_CHANNEL_REQUEST_SUCCESScsTtƒ}ˆ |”ˆj |ddd”}ˆj d”‡fdd„}| ˆj”| |”|S)zi Test that channel request failure messages cause the Deferred to be erred back. r˜raTr€csˆ |jjd”dS)Nzchannel request failed)rdrpršrr rrhgsz;ConnectionTests.test_CHANNEL_REQUEST_FAILURE..check)rrWrLr Zssh_CHANNEL_FAILURErjrlrkrr rrŚtest_CHANNEL_REQUEST_FAILURE]s     z,ConnectionTests.test_CHANNEL_REQUEST_FAILUREcCs||j ddd”}| dd„”|j ddd”| |jjtjt  d”d ftjt  d”d fg”| |jj d |gi”d S) zQ Test that global request messages are sent in the right format. s wantReplyr#TcSsdSrIr ©Zfailurer r rŚtraz8ConnectionTests.test_sendGlobalRequest..snoReplyraFs’datarbŚglobalN) rLrirkrdrKrRr ZMSG_GLOBAL_REQUESTrrcŚ deferreds©rrnr r rŚtest_sendGlobalRequestnsžžz&ConnectionTests.test_sendGlobalRequestcCsXtƒ}|j |d”| |jjtjt  d”dfg”| |j d”| |jj d”dS)zO Test that open channel messages are sent in the right format. saaaars€aaaarr"N) rrLrQrdrKrRr ZMSG_CHANNEL_OPENrrcrZlocalChannelIDrVr r rŚtest_openChannels’ž’ž z ConnectionTests.test_openChannelc Cs°tƒ}| |”|j |ddd”}| dd„”|j |ddd”d|_|j |ddd”| |jjt j d t   d”d ft j d t   d”d fg”| |jj d |g”d S)zR Test that channel request messages are sent in the right format. r˜TcSsdSrIr r£r r rr¤šraz2ConnectionTests.test_sendRequest..ótest2raFstest3rŽstestrbrN)rrWrLr rkŚ localClosedrdrKrRr ZMSG_CHANNEL_REQUESTrrcr¦©rrrnr r rŚtest_sendRequest’s$ žžūž z ConnectionTests.test_sendRequestcCsttdd}| |”d|_|j |d”| |jd”d|_|j |d”| |jd”| |jjt j dfg”dS) zi Test that channel window adjust messages cause bytes to be added to the window. rˆ)rŠrr"Téó’N) rrWrrLZ adjustWindowrdr«rKrRr rrVr r rŚtest_adjustWindow­s  ž’žz!ConnectionTests.test_adjustWindowcCsXtƒ}| |”|j |d”d|_|j |d”| |jjtj dt   d”fg”dS)zO Test that channel data messages are sent in the right format. rTóbrŽN) rrWrLZsendDatar«rdrKrRr ZMSG_CHANNEL_DATArrcrVr r rŚ test_sendDataÄs žzConnectionTests.test_sendDatacCs\tƒ}| |”|j |dd”d|_|j |dd”| |jjtj dt   d”fg”dS)zX Test that channel extended data messages are sent in the right format. r"r˜Tr®rŖrÆN) rrWrLZsendExtendedDatar«rdrKrRr ZMSG_CHANNEL_EXTENDED_DATArrcrVr r rŚtest_sendExtendedDataŅs  ž’žz%ConnectionTests.test_sendExtendedDatacCsbtƒ}| |”|j |”| |jjtjdfg”d|_ |j |”| |jjtjdfg”dS)zN Test that channel EOF messages are sent in the right format. rŽTN) rrWrLZsendEOFrdrKrRr ZMSG_CHANNEL_EOFr«rVr r rŚ test_sendEOFås   ’  ’zConnectionTests.test_sendEOFcCs®tƒ}| |”|j |”| |j”| |jjt j dfg”|j |”| |jjt j dfg”tƒ}| |”| |j ”|  |j ”d|_|j |”| |j ”dS)zP Test that channel close messages are sent in the right format. rŽTN)rrWrLr–r\r«rdrKrRr r‘rr]rZ remoteClosed)rrr^r r rŚtest_sendCloseõs(    ž  ž    zConnectionTests.test_sendClosecCsZ|j dddd”}| |jd”| |jd”| |jd”| tj|jjdddd”dS)z³ Test that getChannel dispatches to the avatar when an avatar is present. Correct functioning without the avatar is verified in test_CHANNEL_OPEN. ré2ér#ruN) rLZ getChannelrdr%rwr3Ś assertRaisesrr7rVr r rŚtest_getChannelWithAvatars’z)ConnectionTests.test_getChannelWithAvatarcCsH|j`| |j dd””| |j dd”d”| |j dd””dS)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 channelClosedr¬r r rŚ9test_channelClosedCausesLeftoverChannelDeferredsToErrback(s   zIConnectionTests.test_channelClosedCausesLeftoverChannelDeferredsToErrbackN)&r r r rrGrNrWrYrZr_rerorrrzr~rrr…r†r“r”r•r—ržrŸr”r¢rØr©r­r°r²r³r“rµr¹rŗrær r r rrHĶsF 4   %+  rHc@s(eZdZdZesdZdd„Zdd„ZdS)ŚCleanConnectionShutdownTestszL Check whether correct cleanup is performed on connection shutdown. r@cCs,t d”|_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. r»r¼r"r½)rLrMrir¾rr7rXr§r r rŚ9test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackDs   zVCleanConnectionShutdownTests.test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackN)r r r r.rrGrNrĮr r r rrĄ6s rĄ)r.rTZtwisted.conch.sshrZtwisted.conch.testrZtwisted.python.reflectrZ twisted.trialrrZ twisted.conchrrr Z SSHChannelrr0rr?ZTestCaserHrĄr r r rŚs&     \),m