tfHrFdZddlZddlmZddlmZddlmZddlm Z edZ ddl m Z e r dd lm Z mZn Gd d ZGd d ejZGddZGddejZGdde jZGdde jZdS)z1 This module tests twisted.conch.ssh.connection. N)channel) test_userauth) requireModule)unittest cryptography)error)common connectionc(eZdZGddZdS)r ceZdZdS)connection.SSHConnectionN)__name__ __module__ __qualname__a/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/conch/test/test_connection.py SSHConnectionr s DrrN)rrrrrrrr r s<          rr cTeZdZdZdZdZdZdZdZdZ dZ dZ d Z d Z d Zd Zd S) 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} TestChannelFcd|jzS)NzTestChannel %i)idselfs r logPrefixzTestChannel.logPrefix:s$'))rcvd|_||_g|_g|_d|_d|_d|_d|_dS)zF The channel is open. Set up the instance variables. TrFN)gotOpen specificDatainBuffer extBuffernumberRequestsgotEOF gotOneClose gotClosed)rrs r channelOpenzTestChannel.channelOpen=sC (   rc||_dS)zD Opening the channel failed. Store the reason why. N)openFailureReason)rreasons r openFailedzTestChannel.openFailedJs"(rc.|xjdz c_|dkS)z_ A test request. Return True if data is 'data'. @type data: L{bytes} data)r"rdatas r request_testzTestChannel.request_testPs" q wrc:|j|dS)z= Data was received. Store it in the buffer. N)r appendr.s r dataReceivedzTestChannel.dataReceivedYs  T"""""rc>|j||fdS)zF Extended data was received. Store it in the buffer. N)r!r2)rcoder/s r extReceivedzTestChannel.extReceived_s$ tTl+++++rcd|_dS)z1 EOF was received. Remember it. TN)r#rs r eofReceivedzTestChannel.eofReceivedes rcd|_dS)z3 Close was received. Remember it. TN)r$rs r closeReceivedzTestChannel.closeReceivedks rcd|_dS)z7 The channel is closed. Rembember it. TN)r%rs rclosedzTestChannel.closedqsrN)rrr__doc__namerr%rr&r*r0r3r6r8r:r<rrrrrs2 DGI***   ((( ### ,,,     rrc"eZdZdZdZdZdZdS) TestAvatarz? A mocked-up version of twisted.conch.avatar.ConchUser {c|tjkrt||||S|dkrtj|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)r channelType windowSize maxPacketr/s r lookupChannelzTestAvatar.lookupChannelsd +* * *' )   / / / "4#8:UVV V 0 /rc*|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/s rgotGlobalRequestzTestAvatar.gotGlobalRequests. - ' '4 K ' ': 5rN)rrrr=rIrMrRrrrr@r@xsGWWW&     rr@c8eZdZdZesdZdZdZdZdZ dZ dS) TestConnectionz} A subclass of SSHConnection for testing. @ivar channel: the current channel. @type channel. C{TestChannel} Cannot run without cryptographycdS)NrTrrs rrzTestConnection.logPrefixsrcdS)zT The other side made the 'TestGlobal' global request. Return True. Trr.s rglobal_TestGlobalz TestConnection.global_TestGlobals trc d|fS)zt The other side made the 'Test-Data' global request. Return True and the data we received. Trr.s rglobal_Test_DatazTestConnection.global_Test_Datas Tzrc>t||||_|jS)z The other side is requesting the TestChannel. Create a C{TestChannel} instance, store it, and return it. )rCrDr/)rrrrKrLr/s rchannel_TestChannelz"TestConnection.channel_TestChannels+ ##YT    |rc td)zU The other side is requesting the ErrorChannel. Raise an exception. z no such thing)AssertionErrorr\s rchannel_ErrorChannelz#TestConnection.channel_ErrorChannels_---rN) rrrr=rskiprrXrZr]r`rrrrTrTsx 10    .....rrTceZdZesdZdZdZdZdZdZ dZ dZ d Z d Z d Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"d Z#d!Z$d"Z%d#S)$ConnectionTestsrUctjd|_t|j_t |_|j|j_|jdSN)r FakeTransport transportr@rErTconnserviceStartedrs rsetUpzConnectionTests.setUpsU&4T:: * "$$ "n    """""rc|j||jjdd|j_|jt jd|jddzdS)z? Open the channel with the default connection. Nz>2Ls)rh openChannelrgpacketsssh_CHANNEL_OPEN_CONFIRMATIONstructpackrrrs r _openChannelzConnectionTests._openChannelso g&&&!%!7!< // Kwz3 / /2U U     rc8|jdSre)rhserviceStoppedrs rtearDownzConnectionTests.tearDowns   """""rcZ||jjj|jdS)z_ Test that the connection links itself to the avatar in the transport. N)assertIsrgrErhrs rtest_linkAvatarzConnectionTests.test_linkAvatars( dn+0$)<<<<.check!s   T7 + + + + +r)rhsendGlobalRequestssh_REQUEST_SUCCESS addCallback addErrbackfailrdrs` rtest_REQUEST_SUCCESSz$ConnectionTests.test_REQUEST_SUCCESSsz I ' ' GT B B %%g... , , , , , e TYrcjddd}jdfd}|j|||S)zg Test that global request failure packets cause the Deferred to be erred back. rr-TcH|jjddSr)rvaluer/)frs rrz3ConnectionTests.test_REQUEST_FAILURE..check0s#   QW\7 3 3 3 3 3r)rhrssh_REQUEST_FAILURErrrrs` rtest_REQUEST_FAILUREz$ConnectionTests.test_REQUEST_FAILURE(sz I ' ' GT B B %%g... 4 4 4 4 4 di    Urc(|j`|jt jddz||jjj| |jjj|j| |jjj d| |jjj d| |jjj d| |jjj d| |jjtjdfgg|j_|jt jddz|| |jjtjdt jd zt jd zfgg|j_|jt jd dz|| |jjtjd t jd zt jd zfgdS)z Test that open channel packets cause a channel to be created and opened or a failure message to be returned. rr,s BadChannelsssunknown channelrs ErrorChannelssunknown failureN)rgrErhssh_CHANNEL_OPENr rr}rrrr/rremoteWindowLeftrDror MSG_CHANNEL_OPEN_CONFIRMATIONflushLoggedErrorsMSG_CHANNEL_OPEN_FAILURErs rtest_CHANNEL_OPENz!ConnectionTests.test_CHANNEL_OPEN7s_ N ! ""69^#<#>>  N "<(  "$ ""69]#;#;>U#UVVV      N "77i 2334inn%  "$ ""69_#=#=@W#WXXX      N "77i 2334inn%  rc ||jj_|jt jddz|tj }| t|dd|| |dj j d| |jjtjdt jdzt jd zfgd 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. rFrr,zExpected one error, got: r)rArGs{serror args in wrong orderrN)rgrErIrhrr rrrrHrlenrargsror r)rr5errorss r_lookupChannelErrorTestz'ConnectionTests._lookupChannelErrorTestks 26. "" I) * *-D D   ''(899 Va)OV)O)OPPP -/QRRR  N "78i <==>inn%   rc0|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. rAN)rrs rtest_lookupChannelErrorz'ConnectionTests.test_lookupChannelErrors $$S)))))rct}|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) rrhrnrprrrDrchannelsToRemoteChannellocalToRemoteChannelrss rtest_CHANNEL_OPEN_CONFIRMATIONz.ConnectionTests.test_CHANNEL_OPEN_CONFIRMATIONs -- g&&& //0GHHH 11555 0!444 -/BCCC :7CQGGG 7:A>>>>>rcXt}|j||jdt jdz||jjd| |jj |dS)zz Test that channel open failure packets cause the channel to be notified that its opening failed. failure!)rr,N) rrhrnssh_CHANNEL_OPEN_FAILUREr rrr(r assertIsNonechannelsgetrss rtest_CHANNEL_OPEN_FAILUREz)ConnectionTests.test_CHANNEL_OPEN_FAILUREs -- g&&& ** 3fi 6L6L L    279IJJJ $),0099:::::rct}|||j}|jd||j|dzdS)zc Test that channel window adjust messages add bytes to the channel window. rr,N)rrtrrhssh_CHANNEL_WINDOW_ADJUSTr)rr oldWindowSizes rtest_CHANNEL_WINDOW_ADJUSTz*ConnectionTests.test_CHANNEL_WINDOW_ADJUSTsd -- '"""0  ++,STTT 1=13DEEEEErctdd}|||jdt jdz||jdg||jj tj dfgg|j_ d|j dzz}|jdt j|z||jdg||jj tj d fgt}||d|jdzz}g|j_ |jd t j|z||jg||jj tj 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)rrtrhssh_CHANNEL_DATAr rrr rgror MSG_CHANNEL_WINDOW_ADJUSTlocalWindowLeftMSG_CHANNEL_CLOSErrrlongDatabigDatas rtest_CHANNEL_DATAz!ConnectionTests.test_CHANNEL_DATAs !A>>> '""" ""#679K9K#KLLL )G9555  N "8;    "$72Q67 ""#689L9L#LMMM )G9555  N "*,? @ A   -- '"""'0145!# ""#679K9K#KLLL )2...  N "*,? @ A     rctdd}|||jdt jdz||jdg||jj tj dfgg|j_ d|j d zz}|jdt j|z||jdg||jj tj d fgt}||d|jd zz}g|j_ |jd t j|z||jg||jj tj 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)rrtrhssh_CHANNEL_EXTENDED_DATAr rrr!rgror rrrrrs rtest_CHANNEL_EXTENDED_DATAz*ConnectionTests.test_CHANNEL_EXTENDED_DATAs !A>>> '""" ++ 3fi6H6H H    *\N;;;  N "8;    "$72Q67 ++ 3fi6I6I I    *\N;;;  N "*,? @ A   -- '"""'0145!# ++ 3fi6H6H H    *B///  N "*,? @ A     rct}|||jd||jdS)zN Test that channel eof messages are passed up to the channel. rN)rrtrhssh_CHANNEL_EOFr}r#rss rtest_CHANNEL_EOFz ConnectionTests.test_CHANNEL_EOF sR-- '""" !!"5666 '''''rct}||||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) rrtr}rr~r$r%rh sendClosessh_CHANNEL_CLOSErss rtest_CHANNEL_CLOSEz"ConnectionTests.test_CHANNEL_CLOSEs -- '""" ((( ,--- *+++ G$$$ ##$7888 +,,, )*****rct}|jdt jdzdz|jdjdt jdzdzdz}fd}|||S)zS Test that channel requests that succeed send MSG_CHANNEL_SUCCESS. rtestrr,rr-cbjjtjdfgdSNr)rrgror MSG_CHANNEL_SUCCESSresultrs rrz;ConnectionTests.test_CHANNEL_REQUEST_success..check1<   &02EFG     r) rrtrhssh_CHANNEL_REQUESTr rrr"rrrrrs` rtest_CHANNEL_REQUEST_successz,ConnectionTests.test_CHANNEL_REQUEST_success#s-- '""" %% &)G"4"4 4w >    /333 I ) ) &)G"4"4 4w > H        erct}|jdt jdzdz}fd}|j|||S)zP Test that channel requests that fail send MSG_CHANNEL_FAILURE. rrrcbjjtjdfgdSr)rrgror MSG_CHANNEL_FAILURErs rrz;ConnectionTests.test_CHANNEL_REQUEST_failure..checkDrr) rrtrhrr rrrrrs` rtest_CHANNEL_REQUEST_failurez,ConnectionTests.test_CHANNEL_REQUEST_failure:s-- '""" I ) ) &)G"4"4 4w >        di    Urct}|j|ddd}jdfd}|S)zj Test that channel request success messages cause the Deferred to be called back. rr-Trc2|dSre)r}rs rrz;ConnectionTests.test_CHANNEL_REQUEST_SUCCESS..checkXs OOF # # # # #r)rrtrh sendRequestssh_CHANNEL_SUCCESSrs` rtest_CHANNEL_REQUEST_SUCCESSz,ConnectionTests.test_CHANNEL_REQUEST_SUCCESSNst -- '""" I ! !'7GT B B %%&9::: $ $ $ $ $rc$t}|j|ddd}jdfd}|j|||S)zi Test that channel request failure messages cause the Deferred to be erred back. rrTrcH|jjddS)Nzchannel request failed)rrrs rrz;ConnectionTests.test_CHANNEL_REQUEST_FAILURE..checkgs%   V\/1I J J J J Jr)rrtrhrssh_CHANNEL_FAILURErrrrs` rtest_CHANNEL_REQUEST_FAILUREz,ConnectionTests.test_CHANNEL_REQUEST_FAILURE]s -- '""" I ! !'7C > > %%&9::: K K K K K di    Urc|jddd}|d|jddd||jjt jtj ddzft jtj dd zfg||jj d |gid S) zQ Test that global request messages are sent in the right format. s wantReplyr-TcdSrerfailures rz8ConnectionTests.test_sendGlobalRequest..tTrsnoReplyrFsdatarglobalN) rhrrrrgror MSG_GLOBAL_REQUESTr r deferredsrrs rtest_sendGlobalRequestz&ConnectionTests.test_sendGlobalRequestns I ' ' gt D D ))*** ##JU;;;  N ". ,0G0G+0UV. *0E0E0OP     ,x!o>>>>>rcRt}|j|d||jjt jtj ddzfg||j d||jj ddS)zO Test that open channel messages are sent in the right format. saaaarsaaaarr,N) rrhrnrrgror MSG_CHANNEL_OPENr rrlocalChannelIDrss rtest_openChannelz ConnectionTests.test_openChannels-- gw///  N "/In--MN   Q''' 1155555rc Vt}|||j|ddd}|d|j|dddd|_|j|ddd||jjtj dtj dzd zftj dtj dzd zfg||jj d |gd S) zR Test that channel request messages are sent in the right format. rTcdSrerrs rrz2ConnectionTests.test_sendRequest..rrtest2rFstest3rstestrrN)rrtrhrr localClosedrrgror MSG_CHANNEL_REQUESTr rrrrrs rtest_sendRequestz ConnectionTests.test_sendRequests)-- '""" I ! !'7GT B B ))*** gxe<<<" gxd;;;  N "2'&)G*<*<<{J 2'&)H*=*==G   ,Q/!55555rctd}||d|_|j|d||jdd|_|j|d||jd||jjtj dfgdS) zi Test that channel window adjust messages cause bytes to be added to the window. r)rrr,TN) rrtrrh adjustWindowrrrgror rrss rtest_adjustWindowz!ConnectionTests.test_adjustWindows !,,, '""""# w*** 0!444" w*** 0!444  N "8;      rcJt}|||j|dd|_|j|d||jjtj dtj dzfgdS)zO Test that channel data messages are sent in the right format. rTbrN) rrtrhsendDatarrrgror MSG_CHANNEL_DATAr rrss r test_sendDatazConnectionTests.test_sendDatas-- '""" 7D)))" 7D)))  N ")+>4+P Q R     rcNt}|||j|ddd|_|j|dd||jjtj dtj dzfgdS)zX Test that channel extended data messages are sent in the right format. r,rTr rr N) rrtrhsendExtendedDatarrrgror MSG_CHANNEL_EXTENDED_DATAr rrss rtest_sendExtendedDataz%ConnectionTests.test_sendExtendedDatas-- '""" ""7Aw777" ""7Ax888  N "8= '@R@RR      rcvt}|||j|||jjtjdfgd|_ |j|||jjtjdfgdS)zN Test that channel EOF messages are sent in the right format. rTN) rrtrhsendEOFrrgror MSG_CHANNEL_EOFrrss r test_sendEOFzConnectionTests.test_sendEOFs-- '""" '"""  N "j&@BU%V$W   # '"""  N "j&@BU%V$W     rct}|||j|||j||jjtj dfg|j|||jjtj dfgt}||||j | |j d|_|j|||j dS)zP Test that channel close messages are sent in the right format. rTN)rrtrhrr}rrrgror rrr~r% remoteClosed)rrrs rtest_sendClosezConnectionTests.test_sendClosesF-- '""" G$$$ +,,,  N "*,? @ A    G$$$  N "*,? @ A   == (### ())) +,,, $ H%%% *+++++rc>|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) rh getChannelrr/rrD assertRaisesrrHrss rtest_getChannelWithAvatarz)ConnectionTests.test_getChannelWithAvatars )&&~r2wGG w/// 12666 0"555   di2M2r7     rc*|j`||jdd||jddd||jdddS)zW Test that gotGlobalRequests dispatches to global_* without an avatar. rOr-s Test-Data)Tr-s BadGlobalN)rgrEr}rhrRrr~rs r"test_gotGlobalRequestWithoutAvatarz2ConnectionTests.test_gotGlobalRequestWithoutAvatars N !  22='JJKKK  I & &|W = =    33L'JJKKKKKrct}|||j|ddd}||t j}|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)rrtrhr assertFailurerrH channelClosedr s r9test_channelClosedCausesLeftoverChannelDeferredsToErrbackzIConnectionTests.test_channelClosedCausesLeftoverChannelDeferredsToErrback(ss -- '""" I ! !'?LTU ! V V   q%"2 3 3 (((rN)&rrrrrarjrtrwrzrrrrrrrrrrrrrrrrrrrrr rrrrrr%r'r/rrrrcrcs 10###   ###===JJJ*5550      2 2 2 h   8 * * * ? ? ? ; ; ; F F F# # # J) ) ) V(((+++ .(   "???"666&6666   .       &   ,,,4     L L L     rrcc&eZdZdZesdZdZdZdS)CleanConnectionShutdownTestszL Check whether correct cleanup is performed on connection shutdown. rUctjd|_t|j_t |_|j|j_dSre)rrfrgr@rErTrhrs rrjz"CleanConnectionShutdownTests.setUp>sA&4T:: * "$$ "n rc|j|jddd}||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+)rhrirr-rrHrvrs r9test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackzVCleanConnectionShutdownTests.test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackDsf   """ I ' 'QR ' S S   q%"2 3 3   """rN)rrrr=rrarjr4rrrr1r16sK 10---     rr1)r=rqtwisted.conch.sshrtwisted.conch.testrtwisted.python.reflectr twisted.trialrr twisted.conchrr r SSHChannelrr@rrTTestCasercr1rrrr<s %%%%%%,,,,,,000000""""""}^,, 444444444 YYYYY'$YYYx&&&&&&&&R).).).).).Z-).).).Xf f f f f h'f f f R8#4r