+ DfjHrPRt^RIt^RIHt^RIHt^RIHt^RIH t ]!R4t ^RI H t ] 'd ^RIH t HtM !R R 4t!R R ]P4t!R R4t!RR]P$4t!RR] P(4t!RR] P(4tR#)z1 This module tests twisted.conch.ssh.connection. Nchannel) test_userauth) requireModule)unittest cryptography)error)common connectionc*]tRt^t!RR4tRtR#)r c]tRt^tRtR#)connection.SSHConnectionN)__name__ __module__ __qualname____firstlineno____static_attributes__rD/usr/lib/python3/dist-packages/twisted/conch/test/test_connection.py SSHConnectionr s rrrN)rrrrrrrrrr r s   rr cfa]tRt^toRtRtRtRtRtRt Rt Rt Rt R t R tR tR tR tVtR#) TestChannelaz 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} TestChannelFc(RVP,#)zTestChannel %i)idselfs&r logPrefixTestChannel.logPrefix:s$''))rc tRVnWn.Vn.Vn^VnRVnRVnRVnR#)z6 The channel is open. Set up the instance variables. TFN)gotOpen specificDatainBuffer extBuffernumberRequestsgotEOF gotOneClose gotClosed)rr"s&&r channelOpenTestChannel.channelOpen=s= (   rc WnR#)z4 Opening the channel failed. Store the reason why. N)openFailureReason)rreasons&&r openFailedTestChannel.openFailedJs "(rc >V;P^, unVR8H#)zG A test request. Return True if data is 'data'. @type data: L{bytes} data)r%rdatas&&r request_testTestChannel.request_testPs q wrc <VPPV4R#)z- Data was received. Store it in the buffer. N)r#appendr2s&&r dataReceivedTestChannel.dataReceivedYs T"rc >VPPW34R#)z6 Extended data was received. Store it in the buffer. N)r$r7)rcoder3s&&&r extReceivedTestChannel.extReceived_s tl+rc RVnR#)z! EOF was received. Remember it. TN)r&rs&r eofReceivedTestChannel.eofReceivedes  rc RVnR#)z# Close was received. Remember it. TN)r'rs&r closeReceivedTestChannel.closeReceivedks  rc RVnR#)z' The channel is closed. Rembember it. TN)r(rs&rclosedTestChannel.closedqs r) r$r(r&r'r!r#r%r,r"N)rrrr__doc__namer!r(rr)r.r4r8r<r?rBrEr__classdictcell__ __classdict__s@rrrsM2 DGI* ( # ,   rrc4a]tRt^xtoRt^{tRtRtRtVt R#) TestAvatarz7 A mocked-up version of twisted.conch.avatar.ConchUser c V\P8Xd\VVVVR7#VR8Xd"\P!VPR4hR#)zz The server wants us to return a channel. If the requested channel is our TestChannel, return it, otherwise return None. ) remoteWindowremoteMaxPacketr3avatarconch-error-argserror args in wrong orderN)rrHr ConchError_ARGS_ERROR_CODE)r channelType windowSize maxPacketr3s&&&&&r lookupChannelTestAvatar.lookupChannelsW +** *' )   / / ""4#8#8:UV V 0rc .VR8XdR#VR8XdRV3#R#)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 requestTyper3s&&&rgotGlobalRequestTestAvatar.gotGlobalRequests$ - ' K ': rrN) rrrrrGrUrYr_rrIrJs@rrMrMxs"W&  rrMcVa]tRt^toRt]'gRtRtRtRt Rt Rt Rt Vt R #) TestConnectionzm A subclass of SSHConnection for testing. @ivar channel: the current channel. @type channel. C{TestChannel} Cannot run without cryptographycR#)rbrrs&rrTestConnection.logPrefixsrc R#)zD The other side made the 'TestGlobal' global request. Return True. Trr2s&&rglobal_TestGlobal TestConnection.global_TestGlobalsrc RV3#)z\ The other side made the 'Test-Data' global request. Return True and the data we received. Trr2s&&rglobal_Test_DataTestConnection.global_Test_Datas Tzrc >\WVR7VnVP#)zk The other side is requesting the TestChannel. Create a C{TestChannel} instance, store it, and return it. )rOrPr3)rrrrWrXr3s&&&&rchannel_TestChannel"TestConnection.channel_TestChannels! ##T  ||rc \R4h)zE The other side is requesting the ErrorChannel. Raise an exception. z no such thing)AssertionErrorrms&&&&rchannel_ErrorChannel#TestConnection.channel_ErrorChannels_--rrN)rrrrrGrskiprrgrjrnrrrrIrJs@rrbrbs2 0  ..rrbca]tRt^to]'gRtRtRtRtRt Rt Rt Rt R t R tR tR tR tRtRtRtRtRtRtRtRtRtRtRtRtRtRtRt Rt!Rt"Rt#R t$R!t%R"t&R#t'Vt(R$#)%ConnectionTestsrcc\P!R4Vn\4VPn\ 4VnVPVP nVP P4R#N)r FakeTransport transportrMrQrbconnserviceStartedrs&rsetUpConnectionTests.setUpsL&44T: * "$ "nn    "rc VPPV4VPPRRVPnVPP \ P !RVP^4R,4R#)z/ Open the channel with the default connection. Nz>2Ls)r{ openChannelrzpacketsssh_CHANNEL_OPEN_CONFIRMATIONstructpackrrrs&&r _openChannelConnectionTests._openChannels_ g&!%!7!7!< // KKwzz3 /2U U rc:VPP4R#rx)r{serviceStoppedrs&rtearDownConnectionTests.tearDowns   "rc zVPVPPPVP4R#)zG Test that the connection links itself to the avatar in the transport. N)assertIsrzrQr{rs&rtest_linkAvatarConnectionTests.test_linkAvatars& dnn++00$)).check!s   T7 +r)r{sendGlobalRequestssh_REQUEST_SUCCESS addCallback addErrbackfailrdrsf rtest_REQUEST_SUCCESS$ConnectionTests.test_REQUEST_SUCCESSsW II ' ' GT B %%g. , e TYYrc aSPPRRR4pSPPR4V3RlpVPSP4VP V4V#)zO Test that global request failure packets cause the Deferred to be erred back. rr1TcT<SPVPPR4R#r)rvaluer3)frs&rr3ConnectionTests.test_REQUEST_FAILURE..check0s   QWW\\7 3r)r{rssh_REQUEST_FAILURErrrrsf rtest_REQUEST_FAILURE$ConnectionTests.test_REQUEST_FAILURE(sW II ' ' GT B %%g. 4 dii  Urc ZVP=VPP\P !R4R ,4VP VPPP4VPVPPPVP4VPVPPPR4VPVPPPR4VPVPPP^4VPVPPP^4VPVPP\P R3.4.VPnVPP\P !R4R ,4VP#4VPVPP\P$R\P !R4,\P !R4,3.4.VPnVPP\P !R4R ,4VP#4VPVPP\P$R \P !R 4,\P !R4,3.4R #)zn Test that open channel packets cause a channel to be created and opened or a failure message to be returned. rs BadChannelssunknown channelrs ErrorChannelssunknown failureNs)rzrQr{ssh_CHANNEL_OPENr rrrr!rr3r"remoteWindowLeftrPrr MSG_CHANNEL_OPEN_CONFIRMATIONflushLoggedErrorsMSG_CHANNEL_OPEN_FAILURErs&rtest_CHANNEL_OPEN!ConnectionTests.test_CHANNEL_OPEN7sF NN ! ""699^#  NN " "<<(  "$ ""699]#;>U#UV    NN " "777ii 234iin%  "$ ""699_#=@W#WX    NN " "777ii 234iin%  rc XWPPnVPP \ P !R4R,4VP\P4pVP\V4^RV: 24VPV^,PPR4VPVPP\P R\ P !R4,\ P !R4,3.4R#) 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. rRzExpected one error, got: s{serror args in wrong orderrNr){rS)rzrQrUr{rr rrrrTrlenrargsrr r)rr;errorss&& r_lookupChannelErrorTest'ConnectionTests._lookupChannelErrorTestks 26. "" II) *-D D ''(8(89 Va+DVJ)OP --/QR  NN " "778ii <=>iin%   rc (VP^{4R#)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. N)rrs&rtest_lookupChannelError'ConnectionTests.test_lookupChannelErrors $$S)rc \4pVPPV4VPPR4VP VP ^4VP VP ^4VP VPR4VP VPPV,^4VP VPP^,^4R#)z^ Test that channel open confirmation packets cause the channel to be notified that it's open. Ns) rr{rrrrrPr"channelsToRemoteChannellocalToRemoteChannelrs& rtest_CHANNEL_OPEN_CONFIRMATION.ConnectionTests.test_CHANNEL_OPEN_CONFIRMATIONs - g& //0GH 1115 00!4 --/BC ::7CQG 77:A>rc p\4pVPPV4VPPR\P !R4,4VP VPPR4VPVPPPV44R#)zb Test that channel open failure packets cause the channel to be notified that its opening failed. failure!N)r) rr{rssh_CHANNEL_OPEN_FAILUREr rrr,r assertIsNonechannelsgetrs& rtest_CHANNEL_OPEN_FAILURE)ConnectionTests.test_CHANNEL_OPEN_FAILUREs - g& ** 3fii 6L L  22779IJ $)),,009:rc \4pVPV4VPpVPP R4VP VPV^,4R#)zK Test that channel window adjust messages add bytes to the channel window. rN)rrrr{ssh_CHANNEL_WINDOW_ADJUSTr)rr oldWindowSizes& rtest_CHANNEL_WINDOW_ADJUST*ConnectionTests.test_CHANNEL_WINDOW_ADJUSTsR - '"00  ++,ST 11=13DErc ^\^^R7pVPV4VPPR\P !R4,4VP VPR.4VP VPP\PR3.4.VPn RVP^,,pVPPR\P !V4,4VP VPR.4VP VPP\PR3.4\4pVPV4RVP^,,p.VPn VPPR\P !V4,4VP VP.4VP VPP\PR3.4R#) 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.  localWindowlocalMaxPacketrr1arN)rrr{ssh_CHANNEL_DATAr rrr#rzrr MSG_CHANNEL_WINDOW_ADJUSTlocalWindowLeftMSG_CHANNEL_CLOSErrrlongDatabigDatas& rtest_CHANNEL_DATA!ConnectionTests.test_CHANNEL_DATAs !A> '" ""#679K#KL ))G95  NN " "88;  "$722Q67 ""#689L#LM ))G95  NN " "**,? @ A - '"'00145!# ""#679K#KL ))2.  NN " "**,? @ A rc ^\^^R7pVPV4VPPR\P !R4,4VP VPR .4VP VPP\PR3.4.VPn RVP^,,pVPPR\P !V4,4VP VPR .4VP VPP\PR3.4\4pVPV4RVP^,,p.VPn VPPR\P !V4,4VP VP.4VP VPP\PR3.4R#) 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. rsr1rrrsN)r1)rrr{ssh_CHANNEL_EXTENDED_DATAr rrr$rzrr rrrrrs& rtest_CHANNEL_EXTENDED_DATA*ConnectionTests.test_CHANNEL_EXTENDED_DATAs !A> '" ++ 3fii6H H  **\N;  NN " "88;  "$722Q67 ++ 3fii6I I  **\N;  NN " "**,? @ A - '"'00145!# ++ 3fii6H H  **B/  NN " "**,? @ A rc \4pVPV4VPPR4VP VP 4R#)z> Test that channel eof messages are passed up to the channel. rN)rrr{ssh_CHANNEL_EOFrr&rs& rtest_CHANNEL_EOF ConnectionTests.test_CHANNEL_EOF s<- '" !!"56 'rc \4pVPV4VPVP4VP VP 4VP VP 4VPPV4VPPR4VPVP 4VPVP 4R#)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) rrrr!rr'r(r{ sendClosessh_CHANNEL_CLOSErs& rtest_CHANNEL_CLOSE"ConnectionTests.test_CHANNEL_CLOSEs - '" ( ,,- **+ G$ ##$78 ++, ))*rc a\4pSPV4SPPR\P !R4,R,4SP VP^4SPPR\P !R4,R,R,4pV3RlpVPV4V#)zC Test that channel requests that succeed send MSG_CHANNEL_SUCCESS. rtestrrr1cv<SPSPP\PR3.4R#rN)rrzrr MSG_CHANNEL_SUCCESSresultrs&rr;ConnectionTests.test_CHANNEL_REQUEST_success..check10   &&002EFG r) rrr{ssh_CHANNEL_REQUESTr rrr%rrrrrsf rtest_CHANNEL_REQUEST_success,ConnectionTests.test_CHANNEL_REQUEST_success#s- '" %% &))G"4 4w >  //3 II ) ) &))G"4 4w > H   erc a\4pSPV4SPPR\P !R4,R,4pV3RlpVP SP4VPV4V#)z@ Test that channel requests that fail send MSG_CHANNEL_FAILURE. rrrcv<SPSPP\PR3.4R#r)rrzrr MSG_CHANNEL_FAILUREr s&rr;ConnectionTests.test_CHANNEL_REQUEST_failure..checkDr r) rrr{r r rrrrrsf rtest_CHANNEL_REQUEST_failure,ConnectionTests.test_CHANNEL_REQUEST_failure:sl- '" II ) ) &))G"4 4w >   dii  Urc a\4pSPV4SPPVRRR4pSPP R4V3RlpV#)zR Test that channel request success messages cause the Deferred to be called back. rr1Trc*<SPV4R#rx)rr s&rr;ConnectionTests.test_CHANNEL_REQUEST_SUCCESS..checkXs OOF #r)rrr{ sendRequestssh_CHANNEL_SUCCESSrsf rtest_CHANNEL_REQUEST_SUCCESS,ConnectionTests.test_CHANNEL_REQUEST_SUCCESSNsS - '" II ! !'7GT B %%&9: $rc a\4pSPV4SPPVRRR4pSPP R4V3RlpVP SP 4VPV4V#)zQ Test that channel request failure messages cause the Deferred to be erred back. rrTrcT<SPVPPR4R#)zchannel request failedN)rrr s&rr;ConnectionTests.test_CHANNEL_REQUEST_FAILURE..checkgs   V\\//1I Jr)rrr{rssh_CHANNEL_FAILURErrrrsf rtest_CHANNEL_REQUEST_FAILURE,ConnectionTests.test_CHANNEL_REQUEST_FAILURE]sp - '" II ! !'7C > %%&9: K dii  Urc VPPRRR4pVPR4VPPRRR4VPVPP \ P\P!R4R,3\ P\P!R4R ,3.4VPVPPR V./4R #) zA Test that global request messages are sent in the right format. s wantReplyr1TcR#rxrfailures&r8ConnectionTests.test_sendGlobalRequest..tTrsnoReplyrFsdatarglobalN) r{rrrrzrr MSG_GLOBAL_REQUESTr r deferredsrrs& rtest_sendGlobalRequest&ConnectionTests.test_sendGlobalRequestns II ' ' gt D )* ##JU;  NN " ".. ,0G+0UV.. *0E0OP   ,,x!o>rc z\4pVPPVR4VPVPP \ P\P!R4R,3.4VPVP^4VPVPP^4R#)z? Test that open channel messages are sent in the right format. saaaarsaaaaN) rr{rrrzrr MSG_CHANNEL_OPENr rrlocalChannelIDrs& rtest_openChannel ConnectionTests.test_openChannels- gw/  NN " "//IIn-MN   Q' 1115rc  \4pVPV4VPPVRRR4pVP R4VPPVRRR4RVnVPPVRRR4VP VPP\PR\P!R4,R ,3\PR\P!R4,R ,3.4VP VPP^,V.4R #) zB Test that channel request messages are sent in the right format. rTcR#rxrr&s&rr(2ConnectionTests.test_sendRequest..r*rtest2rFstest3rstestrN)rrr{rr localClosedrrzrr MSG_CHANNEL_REQUESTr rr-rrrs& rtest_sendRequest ConnectionTests.test_sendRequests- '" II ! !'7GT B )* gxe<" gxd;  NN " "22'&))G*<<{J 22'&))H*==G   ,,Q/!5rc \^R7pVPV4^VnVPP V^4VP VP^4RVnVPP V^4VP VP^4VP VPP\PR3.4R#)zQ Test that channel window adjust messages cause bytes to be added to the window. )rTN) rrrr{ adjustWindowrr:rzrr rrs& rtest_adjustWindow!ConnectionTests.test_adjustWindows !, '""# w* 00!4" w* 00!4  NN " "88;  rc ^\4pVPV4VPPVR4RVnVPPVR4VP VP P\PR\P!R4,3.4R#)z? Test that channel data messages are sent in the right format. rTbrN) rrr{sendDatar:rrzrr MSG_CHANNEL_DATAr rrs& r test_sendDataConnectionTests.test_sendDatas- '" 7D)" 7D)  NN " "))+>4+P Q R rc b\4pVPV4VPPV^R4RVnVPPV^R4VP VP P\PR\P!R4,3.4R#)zH Test that channel extended data messages are sent in the right format. rTr9Nr@) rrr{sendExtendedDatar:rrzrr MSG_CHANNEL_EXTENDED_DATAr rrs& rtest_sendExtendedData%ConnectionTests.test_sendExtendedDatas- '" ""7Aw7" ""7Ax8  NN " "88= '@RR  rc \4pVPV4VPPV4VP VP P \PR3.4RVn VPPV4VP VP P \PR3.4R#)z> Test that channel EOF messages are sent in the right format. rTN) rrr{sendEOFrrzrr MSG_CHANNEL_EOFr:rs& r test_sendEOFConnectionTests.test_sendEOFs- '" '"  NN " "j&@&@BU%V$W # '"  NN " "j&@&@BU%V$W rc \4pVPV4VPPV4VP VP 4VP VPP\PR3.4VPPV4VP VPP\PR3.4\4pVPV4VP VP4VPVP4RVnVPPV4VP VP4R#)z@ Test that channel close messages are sent in the right format. rTN)rrr{rrr:rrzrr rr!rr( remoteClosed)rrrs& rtest_sendCloseConnectionTests.test_sendCloses- '" G$ ++,  NN " "**,? @ A  G$  NN " "**,? @ A = (# (() ++, $ H% **+rc ZVPPR^2^R4pVPVPR4VPVP^24VPVP ^4VP \PVPPR^2^R4R#)z Test that getChannel dispatches to the avatar when an avatar is present. Correct functioning without the avatar is verified in test_CHANNEL_OPEN. rr1rN) r{ getChannelrr3rrP assertRaisesrrTrs& rtest_getChannelWithAvatar)ConnectionTests.test_getChannelWithAvatars ))&&~r2wG w/ 1126 00"5    dii22M2r7 rc "VP=VPVPP RR44VP VPP RR4R4VP VPP RR44R#)zG Test that gotGlobalRequests dispatches to global_* without an avatar. r\r1s Test-Datas BadGlobalN)Tr1)rzrQrr{r_rrrs&r"test_gotGlobalRequestWithoutAvatar2ConnectionTests.test_gotGlobalRequestWithoutAvatarsp NN !  22='JK  II & &|W =  33L'JKrc \4pVPV4VPPVRR^R7pVP V\ P 4pVPPV4V#)z Whenever an SSH channel gets closed any Deferred that was returned by a sendRequest() on its parent connection must be errbacked. dummyrequest dummydata wantReply)rrr{r assertFailurerrT channelClosedr<s& r9test_channelClosedCausesLeftoverChannelDeferredsToErrbackIConnectionTests.test_channelClosedCausesLeftoverChannelDeferredsToErrback(sf - '" II ! !'?LTU ! V   q%"2"2 3 (rr{rzN))rrrrrrtr}rrrrrrrrrrrrrrrrrrrrr"r/r4r=rBrHrMrRrVr[r^rgrrIrJs@rrvrvs 0# #=J*50  2 h 8 * ? ; F# J) V(+ .( "?"6&66 .   & ,4   L  rrvcDa]tRtRtoRt]'gRtRtRtRt Vt R#)CleanConnectionShutdownTestsi6zD Check whether correct cleanup is performed on connection shutdown. rcc\P!R4Vn\4VPn\ 4VnVPVP nR#rx)rryrzrMrQrbr{rs&rr}"CleanConnectionShutdownTests.setUp>s<&44T: * "$ "nn rc VPP4VPPRR^R7pVPV\P 4pVPP 4V#)zt Once the service is stopped any leftover global deferred returned by a sendGlobalRequest() call must be errbacked. rarbrc)r{r|rrerrTrr.s& r9test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackVCleanConnectionShutdownTests.test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackDs]   " II ' 'QR ' S   q%"2"2 3   "rriN) rrrrrGrrtr}rorrIrJs@rrkrk6s# 0-  rrk)rGrtwisted.conch.sshrtwisted.conch.testrtwisted.python.reflectr twisted.trialrr twisted.conchrr r SSHChannelrrMrrbTestCaservrkrrrrxs%,0"^, 44 Y'$$Yx&&R).Z--).Xf h''f R8#4#4r