o Òó·iHrã@sÎdZddlZddlmZddlmZddlmZddlm Z edƒZ ddl m Z e r3dd 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 úhC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\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:s zTestChannel.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=s zTestChannel.channelOpencCs ||_dS)zD Opening the channel failed. Store the reason why. N)ÚopenFailureReason)rÚreasonr r rÚ openFailedJó zTestChannel.openFailedcCs|jd7_|dkS)z_ A test request. Return True if data is 'data'. @type data: L{bytes} éódataN)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.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 rÚclosedqr"zTestChannel.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|tjkr t||||dS|dkrt |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)rr1rÚ ConchErrorÚ_ARGS_ERROR_CODE)rZ 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. ó TestGlobalTóTestDataFNr )rZ requestTyper&r r rÚgotGlobalRequest’s zTestAvatar.gotGlobalRequestN)r r r r0r:r=r@r r r rr2xs  r2c@s@eZdZdZes dZdd„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 cryptographycCódS)NrAr rr r rr¬szTestConnection.logPrefixcCrC)zT The other side made the 'TestGlobal' global request. Return True. TNr 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. TNr 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. )r4r5r&N)rr©rr;r<r&r r rÚchannel_TestChannel¼sÿz"TestConnection.channel_TestChannelcCstdƒ‚)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 r0rÚskiprrDrErGrIr r r rrA¡s rAc@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ÚConnectionTestsrBcCs6t d¡|_tƒ|j_tƒ|_|j|j_|j ¡dS©N)rÚ FakeTransportÚ transportr2r6rAÚ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€)rOÚ openChannelrNÚpacketsÚssh_CHANNEL_OPEN_CONFIRMATIONÚstructÚpackr©rrr r rÚ _openChannelØs ÿzConnectionTests._openChannelcCs|j ¡dSrL)rOÚserviceStoppedrr r rÚtearDownâózConnectionTests.tearDowncCs| |jjj|j¡dS)z_ Test that the connection links itself to the avatar in the transport. N)ZassertIsrNr6rOrr 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) rrOrTrVÚ assertTruerÚ assertFalserr[Ztwisted.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) rOZssh_GLOBAL_REQUESTrÚNSÚ assertEqualrNrUr 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$)rh)r&rr rÚcheck!sz3ConnectionTests.test_REQUEST_SUCCESS..checkN)rOÚsendGlobalRequestZssh_REQUEST_SUCCESSÚ addCallbackÚ addErrbackÚfail©rÚdrlr 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. rjr$Tcsˆ |jjd¡dSrk)rhÚvaluer&)Úfrr rrl0óz3ConnectionTests.test_REQUEST_FAILURE..checkN)rOrmZssh_REQUEST_FAILURErnrprorqr 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 channelres ErrorChannelssunknown failureN)rNr6rOÚssh_CHANNEL_OPENrrgr`rrrhr&rÚremoteWindowLeftr5rUr 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. r7rxr#zExpected one error, got: r)r3r8s{serror args in wrong orderreN)rNr6r:rOr{rrgr}rr9rhÚlenrtÚargsrUr r~)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. r3N)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) rrOrTrVrhr|r5rZchannelsToRemoteChannelZlocalToRemoteChannelrYr 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) rrOrTZssh_CHANNEL_OPEN_FAILURErrgrhrrZ 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)rrZr|rOZssh_CHANNEL_WINDOW_ADJUSTrh)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#óÿryN)rrZrOZssh_CHANNEL_DATArrgrhrrNrUr Ú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ŒrrŽsr$)rr$r‘r’r#r“sN)rrZrOZssh_CHANNEL_EXTENDED_DATArrgrhrrNrUr r”r•r–rr—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)rrZrOZssh_CHANNEL_EOFr`rrYr 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) rrZr`rrarrrOÚ sendCloseZssh_CHANNEL_CLOSErYr 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…ótestrfr#rdr$cóˆ ˆjjtjdfg¡dS©Nr“)rhrNrUr ZMSG_CHANNEL_SUCCESS©Úresultrr rrl1ó þz;ConnectionTests.test_CHANNEL_REQUEST_success..checkN) rrZrOÚssh_CHANNEL_REQUESTrrgrhrrn©rrrrrlr 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…rrdcržrŸ)rhrNrUr ZMSG_CHANNEL_FAILUREr rr rrlDr¢z;ConnectionTests.test_CHANNEL_REQUEST_failure..checkN) rrZrOr£rrgrnrpror¤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. rr$Tr…csˆ |¡dSrL)r`r rr rrlXr]z;ConnectionTests.test_CHANNEL_REQUEST_SUCCESS..checkN)rrZrOÚ 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. rreTr…csˆ |jjd¡dS)Nzchannel request failed)rhrtr rr rrlgrvz;ConnectionTests.test_CHANNEL_REQUEST_FAILURE..checkN)rrZrOr§Zssh_CHANNEL_FAILURErnrpror¤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$TcSódSrLr ©Zfailurer r rÚtóz8ConnectionTests.test_sendGlobalRequest..snoReplyreFsÿdatarfÚglobalN) rOrmrorhrNrUr ZMSG_GLOBAL_REQUESTrrgÚ deferreds©rrrr 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) rrOrTrhrNrUr ZMSG_CHANNEL_OPENrrgrZlocalChannelIDrYr 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. rTcSrªrLr r«r r rr¬šr­z2ConnectionTests.test_sendRequest..ótest2reFstest3r“stestrfrN)rrZrOr§roÚ localClosedrhrNrUr ZMSG_CHANNEL_REQUESTrrgr¯©rrrrr 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)rrr#TéóÿN) rrZr•rOZ adjustWindowrhr´rNrUr r”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) rrZrOZsendDatar´rhrNrUr ZMSG_CHANNEL_DATArrgrYr 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#rTr·r³r¸N) rrZrOZsendExtendedDatar´rhrNrUr ZMSG_CHANNEL_EXTENDED_DATArrgrYr 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) rrZrOZsendEOFrhrNrUr ZMSG_CHANNEL_EOFr´rYr 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)rrZrOr›r`r´rhrNrUr r–rrarZ remoteClosed)rrrbr 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$rzN) rOZ getChannelrhr&r|r5Ú assertRaisesrr9rYr 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)rNr6r`rOr@rhrarr 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 wantReplyN)rrZrOr§Ú assertFailurerr9Z channelClosedrµr r rÚ9test_channelClosedCausesLeftoverChannelDeferredsToErrback(s   zIConnectionTests.test_channelClosedCausesLeftoverChannelDeferredsToErrbackN)&r r r rrJrQrZr\r^rcrirsrwrrƒ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 rrKÍsH 4   %+  rKc@s(eZdZdZes dZdd„Zdd„ZdS)ÚCleanConnectionShutdownTestszL Check whether correct cleanup is performed on connection shutdown. rBcCs,t d¡|_tƒ|j_tƒ|_|j|j_dSrL)rrMrNr2r6rArOrr r rrQ>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ÆN)rOrPrmrÇrr9r[r°r r rÚ9test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackDs  zVCleanConnectionShutdownTests.test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackN)r r r r0rrJrQrÊr r r rrÉ6s  rÉ)r0rWZtwisted.conch.sshrZtwisted.conch.testrZtwisted.python.reflectrZ twisted.trialrrZ twisted.conchrrr Z SSHChannelrr2rrAZTestCaserKrÉr r r rÚs(     \),m