U  W[õmć@sźdZddlmZmZddlZddlmZedƒZddlm Z erRddl m Z m Z nGdd „d ƒZ dd l m Z dd lmZdd lmZdd lmZGdd„de jƒZGdd„dƒZGdd„de jƒZGdd„dejƒZGdd„dejƒZdS)z1 This module tests twisted.conch.ssh.connection. é)ŚdivisionŚabsolute_importN)Ś requireModuleŚ cryptography)Śerror)ŚcommonŚ connectionc@seZdZGdd„dƒZdS)rc@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 r rsr)Śchannel)Ślong)Śunittest)Ś test_userauthc@s`eZdZdZdZdZdd„Zdd„Zdd „Zd d „Z d d „Z dd„Z dd„Z dd„Z dd„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 Ś logPrefix8szTestChannel.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 Ś openFailedHszTestChannel.openFailedcCs|jd7_|dkS)z_ A test request. Return True if data is 'data'. @type data: L{bytes} éódata)r©rŚdatar r r Ś request_testNszTestChannel.request_testcCs|j |”dS)z= Data was received. Store it in the buffer. N)rŚappendr'r r r Ś dataReceivedWszTestChannel.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 Ś eofReceivedcszTestChannel.eofReceivedcCs d|_dS)z3 Close was received. Remember it. TN)rrr r r Ś closeReceivediszTestChannel.closeReceivedcCs d|_dS)z7 The channel is closed. Rembember it. TN)r rr r r ŚclosedoszTestChannel.closedN)r r r Ś__doc__Śnamerrr!r$r)r+r-r.r/r0r r r r rs  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)rr2rŚ ConchErrorŚ_ARGS_ERROR_CODE)rZ channelTypeŚ windowSizeŚ maxPacketr(r r r Ś lookupChannel{s ž’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 r1r;r>rAr r r r r3usr3c@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)NrBr rr r r rØ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. )r5r6r()rr©rr<r=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)ŚAssertionErrorrFr r r Śchannel_ErrorChannelĮsz#TestConnection.channel_ErrorChannelN) r r r r1rŚskiprrDrErGrIr r r r rBs rBc@s2eZdZesdZejdkrdZdd„Zdd„Zdd „Z d d „Z d d „Z dd„Z dd„Z dd„Zdd„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„Z d8d9„Z!d:d;„Z"dd?„Z$d@dA„Z%dBdC„Z&dDdE„Z'dFdG„Z(dS)HŚConnectionTestsrCNś/Cannot run without both cryptography and pyasn1cCs6t d”|_tƒ|j_tƒ|_|j|j_|j ”dS©N)rŚ FakeTransportŚ transportr3r7rBŚ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€)rPŚ openChannelrOŚpacketsŚssh_CHANNEL_OPEN_CONFIRMATIONŚstructZpackr©rrr r r Ś _openChannel×s  ’’zConnectionTests._openChannelcCs|j ”dSrM)rPŚserviceStoppedrr r r ŚtearDownąszConnectionTests.tearDowncCs| |jjj|j”dS)z_ Test that the connection links itself to the avatar in the transport. N)ZassertIsrOr7rPrr r r Śtest_linkAvatarćszConnectionTests.test_linkAvatarcCsbtƒ}tƒ}|j |”|j |”|j d”| |j”| |j”|j ”| |j”dS)zF Test that serviceStopped() closes any open channels. sN) rrPrUrWŚ assertTruerŚ assertFalser[r )rZchannel1Śchannel2r 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) rPZssh_GLOBAL_REQUESTrŚNSŚ assertEqualrOrVrZMSG_REQUEST_SUCCESSZMSG_REQUEST_FAILURErr r r Śtest_GLOBAL_REQUESTųs&  ’’  ’  ’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&)rf)r(rr r Śchecksz3ConnectionTests.test_REQUEST_SUCCESS..check)rPŚsendGlobalRequestZssh_REQUEST_SUCCESSŚ addCallbackŚ addErrbackŚfail©rŚdrjr 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. rhr&Tcsˆ |jjd”dSri)rfŚvaluer()Śfrr r rj"sz3ConnectionTests.test_REQUEST_FAILURE..check)rPrkZssh_REQUEST_FAILURErlrnrmror rr Śtest_REQUEST_FAILUREs     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 channelrcs ErrorChannelssunknown failureN)rOr7rPŚssh_CHANNEL_OPENrrer^rrrfr(rŚremoteWindowLeftr6rVrZMSG_CHANNEL_OPEN_CONFIRMATIONŚflushLoggedErrorsŚMSG_CHANNEL_OPEN_FAILURErr r r Śtest_CHANNEL_OPEN(sV’ ’’’ ’’’’’ ’’’’z!ConnectionTests.test_CHANNEL_OPENcCs||jj_|j t d”d”| tj ”}|  t |ƒdd|f”|  |dj j tdƒdf”|  |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. r8rur%zExpected one error, got: %rrr4r9s{serror args in wrong orderrcN)rOr7r;rPrxrrerzrr:rfŚlenrrŚargsrrVrr{)rr,Śerrorsr r r Ś_lookupChannelErrorTestLs*  ’ ’’’üž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. r4N)r€rr r r Śtest_lookupChannelErrorcs z'ConnectionTests.test_lookupChannelErrorcCs| tdƒ”dS)z“ Like L{test_lookupChannelError}, but for the case where the failure code is represented as a L{long} instead of a L{int}. r4N)r€rrr r r Śtest_lookupChannelErrorLongCodeqsz/ConnectionTests.test_lookupChannelErrorLongCodecCsttƒ}|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) rrPrUrWrfryr6rZchannelsToRemoteChannelZlocalToRemoteChannelrYr r r Śtest_CHANNEL_OPEN_CONFIRMATIONys  ’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) rrPrUZssh_CHANNEL_OPEN_FAILURErrerfr"r~Z assertIsNoneZchannelsŚgetrYr 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)rrZryrPZssh_CHANNEL_WINDOW_ADJUSTrf)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%ó’rvN)rrZrPZssh_CHANNEL_DATArrerfrrOrVrŚ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)rrZrPZssh_CHANNEL_EXTENDED_DATArrerfrrOrVrr’r“r”rŽr•r r r Śtest_CHANNEL_EXTENDED_DATA¼s:  ’  ’’  ’ ’  ’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)rrZrPZssh_CHANNEL_EOFr^rrYr r r Śtest_CHANNEL_EOFŚs  z ConnectionTests.test_CHANNEL_EOFcCsDtƒ}| |”|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)rrZrPŚ sendCloseZssh_CHANNEL_CLOSEr^rr rYr r r Śtest_CHANNEL_CLOSEćs     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ƒótestrdr%rbr&csˆ ˆjjtjdfg”dS©Nr‘)rfrOrVrZMSG_CHANNEL_SUCCESS©Śresultrr r rjūs  ’z;ConnectionTests.test_CHANNEL_REQUEST_success..check) rrZrPŚssh_CHANNEL_REQUESTrrerfrrl©rrrprjr 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›rbcsˆ ˆjjtjdfg”dSrœ)rfrOrVrZMSG_CHANNEL_FAILURErrr r rj s  ’z;ConnectionTests.test_CHANNEL_REQUEST_failure..check) rrZrPrŸrrerlrnrmr r rr Śtest_CHANNEL_REQUEST_failures  ’’   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ˆ |”dSrM)r^rrr r rjsz;ConnectionTests.test_CHANNEL_REQUEST_SUCCESS..check)rrZrPŚ sendRequestZssh_CHANNEL_SUCCESSr r rr Śtest_CHANNEL_REQUEST_SUCCESSs    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›rcTrƒcsˆ |jjd”dS)Nzchannel request failed)rfrrrrr r rj'sz;ConnectionTests.test_CHANNEL_REQUEST_FAILURE..check)rrZrPr£Zssh_CHANNEL_FAILURErlrnrmr r rr Śtest_CHANNEL_REQUEST_FAILUREs     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&TcSsdSrMr ©Zfailurer r r Ś3rcz8ConnectionTests.test_sendGlobalRequest..snoReplyrcFs’datardŚglobalN) rPrkrmrfrOrVrZMSG_GLOBAL_REQUESTrreŚ deferreds©rrpr r r Śtest_sendGlobalRequest-s  ’ ’ž’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) rrPrUrfrOrVrZMSG_CHANNEL_OPENrrerZlocalChannelIDrYr r r Śtest_openChannel<s  ’’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›TcSsdSrMr r¦r r r r§Prcz2ConnectionTests.test_sendRequest..ótest2rcFstest3r‘stestrdrN)rrZrPr£rmŚ localClosedrfrOrVrZMSG_CHANNEL_REQUESTrrer©©rrrpr r r Śtest_sendRequestHs*  ’’’’ž’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) rrZr“rPZ adjustWindowrfr®rOrVrr’rYr 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) rrZrPZsendDatar®rfrOrVrZMSG_CHANNEL_DATArrerYr r r Ś test_sendDatals  ’’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) rrZrPZsendExtendedDatar®rfrOrVrZMSG_CHANNEL_EXTENDED_DATArrerYr r r Śtest_sendExtendedDatays  ’’z%ConnectionTests.test_sendExtendedDatacCsbtƒ}| |”|j |”| |jjtjdfg”d|_ |j |”| |jjtjdfg”dS)zN Test that channel EOF messages are sent in the right format. r‘TN) rrZrPZsendEOFrfrOrVrZMSG_CHANNEL_EOFr®rYr r r Ś test_sendEOF†s    ’   ’zConnectionTests.test_sendEOFcCs–tƒ}| |”|j |”| |j”| |jjt j dfg”|j |”| |jjt j dfg”tƒ}| |”d|_ |j |”| |j ”dS)zP Test that channel close messages are sent in the right format. r‘TN) rrZrPr™r^r®rfrOrVrr”Z remoteClosedr )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&rwN) rPZ getChannelrfr(ryr6Z assertRaisesrr:rYr r r Śtest_getChannelWithAvatarØs’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)rOr7r^rPrArfr_rr r r Ś"test_gotGlobalRequestWithoutAvatarµs ’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)rrZrPr£Ś assertFailurerr:Z channelClosedrÆr r r Ś9test_channelClosedCausesLeftoverChannelDeferredsToErrbackĄs ’ zIConnectionTests.test_channelClosedCausesLeftoverChannelDeferredsToErrback))r r r rrJrrOrRrZr\r]rargrqrtr|r€rr‚r„rˆr‰r–r—r˜ršr”r¢r¤r„r«r¬r°r³rµr¶r·rør»r¼rĮr r r r rKÉsL    $          rKc@s6eZdZdZesdZejdkr"dZdd„Zdd„Z dS) ŚCleanConnectionShutdownTestszL Check whether correct cleanup is performed on connection shutdown. rCNrLcCs,t d”|_tƒ|j_tƒ|_|j|j_dSrM)rrNrOr3r7rBrPrr r r rRŚ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æ)rPrQrkrĄrr:r[rŖr r r Ś9test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackįs ’ zVCleanConnectionShutdownTests.test_serviceStoppedCausesLeftoverGlobalDeferredsToErrback) r r r r1rrJrrOrRrĆr r r r rĀŠs rĀ)r1Z __future__rrrXZtwisted.python.reflectrrZ twisted.conchrZtwisted.conch.sshrrrZtwisted.python.compatrZ twisted.trialrZtwisted.conch.testrZ SSHChannelrr3rrBZTestCaserKrĀr r r r Śs*      Y(,