֛hHrLSrSSKrSSKJr SSKJr SSKJr SSKJ r \"S5r SSK J r \ (a SS KJ r Jr O "S S 5r"S S \R5r"SS5r"SS\R$5r"SS\ R(5r"SS\ R(5rg)z1 This module tests twisted.conch.ssh.connection. Nchannel) test_userauth) requireModule)unittest cryptography)error)common connectionc(\rSrSr"SS5rSrg)r c\rSrSrSrg)connection.SSHConnectionN)__name__ __module__ __qualname____firstlineno____static_attributes__rfC:\Users\psant\Downloads\Noiseback\school\venv\Lib\site-packages\twisted/conch/test/test_connection.py SSHConnectionrs rrrN)rrrrrrrrrr r s   rr cZ\rSrSrSrSrSrSrSrSr Sr Sr S r S r S rS rS rSrg) 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 SUR-$)NzTestChannel %i)idselfs r logPrefixTestChannel.logPrefix:s$''))rcrSUlXl/Ul/UlSUlSUlSUlSUlg)z6 The channel is open. Set up the instance variables. TrFN)gotOpen specificDatainBuffer extBuffernumberRequestsgotEOF gotOneClose gotClosed)r!r&s r channelOpenTestChannel.channelOpen=s= (   rcXlg)z4 Opening the channel failed. Store the reason why. N)openFailureReason)r!reasons r openFailedTestChannel.openFailedJs "(rc6U=RS- slUS:H$)zG A test request. Return True if data is 'data'. @type data: L{bytes} data)r)r!datas r request_testTestChannel.request_testPs q wrc:URRU5 g)z- Data was received. Store it in the buffer. N)r'appendr7s r dataReceivedTestChannel.dataReceivedYs T"rc<URRX45 g)z6 Extended data was received. Store it in the buffer. N)r(r<)r!coder8s r extReceivedTestChannel.extReceived_s tl+rcSUlg)z! EOF was received. Remember it. TN)r*r s r eofReceivedTestChannel.eofReceivedes  rcSUlg)z# Close was received. Remember it. TN)r+r s r closeReceivedTestChannel.closeReceivedks  rcSUlg)z' The channel is closed. Rembember it. TN)r,r s rclosedTestChannel.closedqs r) r(r,r*r+r%r'r)r0r&N)rrrr__doc__namer%r,r"r-r2r9r=rArDrGrJrrrrrrsE2 DGI* ( # ,   rrc(\rSrSrSrSrSrSrSrg) TestAvatarxz7 A mocked-up version of twisted.conch.avatar.ConchUser {cU[R:Xa [UUUUS9$US:Xa![R"URS5eg)zz The server wants us to return a channel. If the requested channel is our TestChannel, return it, otherwise return None. ) remoteWindowremoteMaxPacketr8avatarconch-error-argserror args in wrong orderN)rrMr ConchError_ARGS_ERROR_CODE)r! channelType windowSize maxPacketr8s r lookupChannelTestAvatar.lookupChannelsW +** *' )   / / ""4#8#8:UV V 0rc&US:XagUS:XaSU4$g)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! requestTyper8s rgotGlobalRequestTestAvatar.gotGlobalRequests$ - ' K ': rrN) rrrrrLrYr]rcrrrrrOrOxsW& rrOcH\rSrSrSr\(dSrSrSrSr Sr Sr S r g ) TestConnectionzm A subclass of SSHConnection for testing. @ivar channel: the current channel. @type channel. C{TestChannel} Cannot run without cryptographycg)Nrfrr s rr"TestConnection.logPrefixsrcg)zD The other side made the 'TestGlobal' global request. Return True. Trr7s rglobal_TestGlobal TestConnection.global_TestGlobalsrc SU4$)z\ The other side made the 'Test-Data' global request. Return True and the data we received. Trr7s rglobal_Test_DataTestConnection.global_Test_Datas Tzrc8[XUS9UlUR$)zk The other side is requesting the TestChannel. Create a C{TestChannel} instance, store it, and return it. )rSrTr8)rrr!r[r\r8s rchannel_TestChannel"TestConnection.channel_TestChannels! ##T  ||rc[S5e)zE The other side is requesting the ErrorChannel. Raise an exception. z no such thing)AssertionErrorrrs rchannel_ErrorChannel#TestConnection.channel_ErrorChannels_--rrN) rrrrrLrskipr"rlrorsrwrrrrrfrfs* 0  .rrfc\rSrSr\(dSrSrSrSrSr Sr Sr S r S r S rS rS rSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSr Sr!Sr"S r#S!r$S"r%S#r&S$r'g%)&ConnectionTestsrhc[R"S5Ul[5URl[ 5UlURUR lUR R5 gN)r FakeTransport transportrOrUrfconnserviceStartedr s rsetUpConnectionTests.setUpsL&44T: * "$ "nn    "rcURRU5 URRSSURlURR [ R "SURS5S-5 g)z/ Open the channel with the default connection. Nz>2Ls)r openChannelrpacketsssh_CHANNEL_OPEN_CONFIRMATIONstructpackrr!rs r _openChannelConnectionTests._openChannels_ g&!%!7!7!< // KKwzz3 /2U U rc8URR5 gr~)rserviceStoppedr s rtearDownConnectionTests.tearDowns   "rcxURURRRUR5 g)zG Test that the connection links itself to the avatar in the transport. N)assertIsrrUrr s rtest_linkAvatarConnectionTests.test_linkAvatars& dnn++00$))TRUS5 gNr6)r)r8r!s rcheck3ConnectionTests.test_REQUEST_SUCCESS..check!s   T7 +r)rsendGlobalRequestssh_REQUEST_SUCCESS addCallback addErrbackfailr!drs` rtest_REQUEST_SUCCESS$ConnectionTests.test_REQUEST_SUCCESSsW II ' ' GT B %%g. , e TYYrc^TRRSSS5nTRRS5 U4SjnURTR5 UR U5 U$)zO Test that global request failure packets cause the Deferred to be erred back. rr6TcR>TRURRS5 gr)rvaluer8)fr!s rr3ConnectionTests.test_REQUEST_FAILURE..check0s   QWW\\7 3r)rrssh_REQUEST_FAILURErrrrs` rtest_REQUEST_FAILURE$ConnectionTests.test_REQUEST_FAILURE(sW II ' ' GT B %%g. 4 dii  Urc UR?URR[R "S5S-5 UR URRR5 URURRRUR5 URURRRS5 URURRRS5 URURRRS5 URURRRS5 URURR[R S4/5 /URlURR[R "S5S-5 UR#5 URURR[R$S[R "S 5-[R "S 5-4/5 /URlURR[R "S 5S-5 UR#5 URURR[R$S [R "S 5-[R "S 5-4/5 g)zn Test that open channel packets cause a channel to be created and opened or a failure message to be returned. rr5s BadChannelsssunknown channelrs ErrorChannelssunknown failureN)rrUrssh_CHANNEL_OPENr rrrr%rr8r&remoteWindowLeftrTrr MSG_CHANNEL_OPEN_CONFIRMATIONflushLoggedErrorsMSG_CHANNEL_OPEN_FAILUREr s rtest_CHANNEL_OPEN!ConnectionTests.test_CHANNEL_OPEN7sF NN ! ""699^#  NN " "<<(  "$ ""699]#;>U#UV    NN " "777ii 234iin%  "$ ""699_#=@W#WX    NN " "777ii 234iin%  rc6XRRlURR [ R "S5S-5 UR[R5nUR[U5SSU<35 URUSRRS5 URURR[R S[ R "S5-[ R "S 5-4/5 g ) 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. rVrr5zExpected one error, got: r)rQrWs{serror args in wrong orderrN)rrUrYrrr rrr rXrlenrargsrr r)r!r@errorss r_lookupChannelErrorTest'ConnectionTests._lookupChannelErrorTestks 26. "" II) *-D D ''(8(89 Va+DVJ)OP --/QR  NN " "778ii <=>iin%   rc&URS5 g)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. rQN)rr s rtest_lookupChannelError'ConnectionTests.test_lookupChannelErrors $$S)rc[5nURRU5 URRS5 UR UR S5 UR UR S5 UR URS5 UR URRUS5 UR URRSS5 g)z^ Test that channel open confirmation packets cause the channel to be notified that it's open. srN) rrrrrrrTr&channelsToRemoteChannellocalToRemoteChannelrs rtest_CHANNEL_OPEN_CONFIRMATION.ConnectionTests.test_CHANNEL_OPEN_CONFIRMATIONs - g& //0GH 1115 00!4 --/BC ::7CQG 77:A>rcf[5nURRU5 URRS[R "S5-5 UR URRS5 URURRRU55 g)zb Test that channel open failure packets cause the channel to be notified that its opening failed. failure!)rr5N) rrrssh_CHANNEL_OPEN_FAILUREr rrr0r assertIsNonechannelsgetrs rtest_CHANNEL_OPEN_FAILURE)ConnectionTests.test_CHANNEL_OPEN_FAILUREs - g& ** 3fii 6L L  22779IJ $)),,009:rc[5nURU5 URnURR S5 UR URUS-5 g)zK Test that channel window adjust messages add bytes to the channel window. rr5N)rrrrssh_CHANNEL_WINDOW_ADJUSTr)r!r oldWindowSizes rtest_CHANNEL_WINDOW_ADJUST*ConnectionTests.test_CHANNEL_WINDOW_ADJUSTsR - '"00  ++,ST 11=13DErc[SSS9nURU5 URRS[R "S5-5 UR URS/5 UR URR[RS4/5 /URl SURS--nURRS[R "U5-5 UR URS/5 UR URR[RS 4/5 [5nURU5 SURS--n/URl URRS [R "U5-5 UR UR/5 UR URR[RS 4/5 g ) 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.  localWindowlocalMaxPacketrr6ar5rN)rrrssh_CHANNEL_DATAr rrr'rrr MSG_CHANNEL_WINDOW_ADJUSTlocalWindowLeftMSG_CHANNEL_CLOSErr!rlongDatabigDatas 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[SSS9nURU5 URRS[R "S5-5 UR URS/5 UR URR[RS4/5 /URl SURS --nURRS[R "U5-5 UR URS/5 UR URR[RS 4/5 [5nURU5 SURS --n/URl URRS [R "U5-5 UR UR/5 UR URR[RS 4/5 g ) 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. rrrsr6)rr6rrr5rsN)rrrssh_CHANNEL_EXTENDED_DATAr rrr(rrr 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[5nURU5 URRS5 UR UR 5 g)z> Test that channel eof messages are passed up to the channel. rN)rrrssh_CHANNEL_EOFrr*rs rtest_CHANNEL_EOF ConnectionTests.test_CHANNEL_EOF s<- '" !!"56 'rc[5nURU5 URUR5 UR UR 5 UR UR 5 URRU5 URRS5 URUR 5 URUR 5 g)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~^[5nTRU5 TRRS[R "S5-S-5 TR URS5 TRRS[R "S5-S-S-5nU4SjnURU5 U$)zC Test that channel requests that succeed send MSG_CHANNEL_SUCCESS. rtestrr5rr6ct>TRTRR[RS4/5 gNr)rrrr MSG_CHANNEL_SUCCESSresultr!s rr;ConnectionTests.test_CHANNEL_REQUEST_success..check10   &&002EFG r) rrrssh_CHANNEL_REQUESTr rrr)rr!rrrs` rtest_CHANNEL_REQUEST_success,ConnectionTests.test_CHANNEL_REQUEST_success#s- '" %% &))G"4 4w >  //3 II ) ) &))G"4 4w > H   erc ^[5nTRU5 TRRS[R "S5-S-5nU4SjnUR TR5 URU5 U$)z@ Test that channel requests that fail send MSG_CHANNEL_FAILURE. rr rct>TRTRR[RS4/5 gr )rrrr MSG_CHANNEL_FAILURErs rr;ConnectionTests.test_CHANNEL_REQUEST_failure..checkDrr) rrrrr rrrrrs` rtest_CHANNEL_REQUEST_failure,ConnectionTests.test_CHANNEL_REQUEST_failure:sl- '" II ) ) &))G"4 4w >   dii  Urc^[5nTRU5 TRRUSSS5nTRR S5 U4SjnU$)zR Test that channel request success messages cause the Deferred to be called back. r r6Trc(>TRU5 gr~)rrs rr;ConnectionTests.test_CHANNEL_REQUEST_SUCCESS..checkXs OOF #r)rrr sendRequestssh_CHANNEL_SUCCESSrs` rtest_CHANNEL_REQUEST_SUCCESS,ConnectionTests.test_CHANNEL_REQUEST_SUCCESSNsS - '" II ! !'7GT B %%&9: $rc^[5nTRU5 TRRUSSS5nTRR S5 U4SjnUR TR 5 URU5 U$)zQ Test that channel request failure messages cause the Deferred to be erred back. r rTrcR>TRURRS5 g)Nzchannel request failed)rrrs rr;ConnectionTests.test_CHANNEL_REQUEST_FAILURE..checkgs   V\\//1I Jr)rrrr ssh_CHANNEL_FAILURErrrrs` rtest_CHANNEL_REQUEST_FAILURE,ConnectionTests.test_CHANNEL_REQUEST_FAILURE]sp - '" II ! !'7C > %%&9: K dii  UrcURRSSS5nURS5 URRSSS5 URURR [ R[R"S5S-4[ R[R"S5S -4/5 URURRS U/05 g ) zA Test that global request messages are sent in the right format. s wantReplyr6Tcgr~rfailures r8ConnectionTests.test_sendGlobalRequest..tTrsnoReplyrFsdatarglobalN) rrrrrrr MSG_GLOBAL_REQUESTr r deferredsr!rs rtest_sendGlobalRequest&ConnectionTests.test_sendGlobalRequestns II ' ' gt D )* ##JU;  NN " ".. ,0G+0UV.. *0E0OP   ,,x!o>rcp[5nURRUS5 URURR [ R[R"S5S-4/5 URURS5 URURRS5 g)z? Test that open channel messages are sent in the right format. saaaarsaaaarr5N) rrrrrrr MSG_CHANNEL_OPENr rrlocalChannelIDrs rtest_openChannel ConnectionTests.test_openChannels- gw/  NN " "//IIn-MN   Q' 1115rc l[5nURU5 URRUSSS5nUR S5 URRUSSS5 SUlURRUSSS5 UR URR[RS[R"S5-S -4[RS[R"S5-S -4/5 UR URRS U/5 g ) zB Test that channel request messages are sent in the right format. r Tcgr~rr,s rr.2ConnectionTests.test_sendRequest..r0rtest2rFstest3rstestrrN)rrrr r localClosedrrrr MSG_CHANNEL_REQUESTr rr3r!rrs rtest_sendRequest ConnectionTests.test_sendRequests- '" II ! !'7GT B )* gxe<" gxd;  NN " "22'&))G*<<{J 22'&))H*==G   ,,Q/!5rc[SS9nURU5 SUlURR US5 UR URS5 SUlURR US5 UR URS5 UR URR[RS4/5 g) zQ Test that channel window adjust messages cause bytes to be added to the window. r)rrr5TN) rrrr adjustWindowrr@rrr rrs rtest_adjustWindow!ConnectionTests.test_adjustWindows !, '""# w* 00!4" w* 00!4  NN " "88;  rcT[5nURU5 URRUS5 SUlURRUS5 UR UR R[RS[R"S5-4/5 g)z? Test that channel data messages are sent in the right format. rTbrN) rrrsendDatar@rrrr MSG_CHANNEL_DATAr rrs r test_sendDataConnectionTests.test_sendDatas- '" 7D)" 7D)  NN " "))+>4+P Q R rcX[5nURU5 URRUSS5 SUlURRUSS5 UR UR R[RS[R"S5-4/5 g)zH Test that channel extended data messages are sent in the right format. r5r TrFr?rGN) rrrsendExtendedDatar@rrrr MSG_CHANNEL_EXTENDED_DATAr rrs rtest_sendExtendedData%ConnectionTests.test_sendExtendedDatas- '" ""7Aw7" ""7Ax8  NN " "88= '@RR  rc[5nURU5 URRU5 UR UR R [RS4/5 SUl URRU5 UR UR R [RS4/5 g)z> Test that channel EOF messages are sent in the right format. rTN) rrrsendEOFrrrr MSG_CHANNEL_EOFr@rs r test_sendEOFConnectionTests.test_sendEOFs- '" '"  NN " "j&@&@BU%V$W # '"  NN " "j&@&@BU%V$W rc[5nURU5 URRU5 UR UR 5 UR URR[RS4/5 URRU5 UR URR[RS4/5 [5nURU5 UR UR5 URUR5 SUlURRU5 UR UR5 g)z@ Test that channel close messages are sent in the right format. rTN)rrrrrr@rrrr rr%rr, remoteClosed)r!rrs rtest_sendCloseConnectionTests.test_sendCloses- '" G$ ++,  NN " "**,? @ A  G$  NN " "**,? @ A = (# (() ++, $ H% **+rcXURRSSSS5nURURS5 URURS5 URUR S5 UR [RURRSSSS5 g)z Test that getChannel dispatches to the avatar when an avatar is present. Correct functioning without the avatar is verified in test_CHANNEL_OPEN. r2r6rN) r getChannelrr8rrT assertRaisesr rXrs rtest_getChannelWithAvatar)ConnectionTests.test_getChannelWithAvatars ))&&~r2wG w/ 1126 00"5    dii22M2r7 rc UR?URURR SS55 UR URR SS5S5 UR URR SS55 g)zG Test that gotGlobalRequests dispatches to global_* without an avatar. r`r6s Test-Data)Tr6s BadGlobalN)rrUrrrcrrr s r"test_gotGlobalRequestWithoutAvatar2ConnectionTests.test_gotGlobalRequestWithoutAvatarsp NN !  22='JK  II & &|W =  33L'JKrc[5nURU5 URRUSSSS9nUR U[ R 5nURRU5 U$)z Whenever an SSH channel gets closed any Deferred that was returned by a sendRequest() on its parent connection must be errbacked. dummyrequest dummydatar5 wantReply)rrrr  assertFailurer rX channelClosedrBs r9test_channelClosedCausesLeftoverChannelDeferredsToErrbackIConnectionTests.test_channelClosedCausesLeftoverChannelDeferredsToErrback(sf - '" II ! !'?LTU ! V   q%"2"2 3 (rrrN)(rrrrrryrrrrrrrrrrrrrrrrrrrrr"r(r5r:rCrIrOrTrYr]rdrgrprrrrr{r{s 0# #=J*50  2 h 8 * ? ; F# J) V(+ .( "?"6&66 .   & ,4   L rr{c6\rSrSrSr\(dSrSrSrSr g)CleanConnectionShutdownTestsi6zD Check whether correct cleanup is performed on connection shutdown. rhc[R"S5Ul[5URl[ 5UlURUR lgr~)rrrrOrUrfrr s rr"CleanConnectionShutdownTests.setUp>s<&44T: * "$ "nn rcURR5 URRSSSS9nURU[R 5nURR 5 U$)zt Once the service is stopped any leftover global deferred returned by a sendGlobalRequest() call must be errbacked. rjrkr5rl)rrrrnr rXrr4s r9test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackVCleanConnectionShutdownTests.test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackDs]   " II ' 'QR ' S   q%"2"2 3   "rrrN) rrrrrLrryrrxrrrrrtrt6s 0- rrt)rLrtwisted.conch.sshrtwisted.conch.testrtwisted.python.reflectr twisted.trialrr twisted.conchr r r SSHChannelrrOrrfTestCaser{rtrrrrs%,0"^, 44 Y'$$Yx&&R).Z--).Xf h''f R8#4#4r