tf2dZddlmZddlmZddlmZ ddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZd Zn#e$rd ZeZ YnwxYwddlmZGdde Z dddZeeeGddeZdS)z Test ssh/channel.py. ) annotations)skipIf) verifyObject)channel)SSHTransportAddress) SSHService)SSHServerTransport) interfaces) IPv4Address)StringTransportzConch SSH not supported.)TestCasec:eZdZdZddZddZdd ZddZddZdS)MockConnectiona A mock for twisted.conch.ssh.connection.SSHConnection. Record the data that channels send, and when they try to close the connection. @ivar data: a L{dict} mapping channel id #s to lists of data sent by that channel. @ivar extData: a L{dict} mapping channel id #s to lists of 2-tuples (extended data type, data) sent by that channel. @ivar closes: a L{dict} mapping channel id #s to True if that channel sent a close message. returnNonec0i|_i|_i|_dS)N)dataextDataclosesselfs ^/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/conch/test/test_channel.py__init__zMockConnection.__init__*s;= JL 68 strcdS)z, Return our logging prefix. rrs r logPrefixzMockConnection.logPrefix/s  rrchannel.SSHChannelrbytescb|j|g|dS)z' Record the sent data. N)r setdefaultappend)rrrs rsendDatazMockConnection.sendData5s0 Wb))0066666rtypeintcf|j|g||fdS)z0 Record the sent extended data. N)rr#r$)rrr&rs rsendExtendedDatazMockConnection.sendExtendedData;s4 ,,33T4LAAAAArcd|j|<dS)z? Record that the channel sent a close message. TN)r)rrs r sendClosezMockConnection.sendCloseCs $ GrNrr)rr)rr rr!rr)rr r&r'rr!rr)rr rr) __name__ __module__ __qualname____doc__rrr%r)r+rrrrrs  9999     7777 BBBB$$$$$$rrNservicer hostAddressinterfaces.IAddress | None peerAddressrrct}|t||||dS)a Connect a SSHTransport which is already connected to a remote peer to the channel under test. @param service: Service used over the connected transport. @type service: L{SSHService} @param hostAddress: Local address of the connected transport. @type hostAddress: L{interfaces.IAddress} @param peerAddress: Remote address of the connected transport. @type peerAddress: L{interfaces.IAddress} )r2r4N)r makeConnectionr setService)r1r2r4 transports rconnectSSHTransportr9JsS$#$$I K[III!!!!!rceZdZdZddZddZddZddZddZdd Z dd Z dd Z dd Z dd Z ddZddZddZddZddZdS) ChannelTestsz" Tests for L{SSHChannel}. rrct|_tj|jd|_d|j_dS)z Initialize the channel. remoteMaxPacket is 10 so that data is able to be sent (the default of 0 means no data is sent because no packets are made). )connremoteMaxPacketschannelN)rr>r SSHChannelnamers rsetUpzChannelTests.setUpis: #$$ )ty"MMM & rcj|ttj|jdS)zK L{SSHChannel} instances provide L{interfaces.ITransport}. N) assertTruerr ITransportrrs rtest_interfacezChannelTests.test_interfacess+  Z%:DLIIJJJJJrc tj|j}||jd||jd||jd||jd||jd||j|j| |j | |j tjddddd d d }||jd||jd||jd||jd||jd||jd ||j d ||j d d S) a Test that SSHChannel initializes correctly. localWindowSize defaults to 131072 (2**17) and localMaxPacket to 32768 (2**15) as reasonable defaults (what OpenSSH uses for those variables). The values in the second set of assertions are meaningless; they serve only to verify that the instance variables are assigned in the correct order. )r>iirN) rr@r> assertEquallocalWindowSizelocalWindowLeftlocalMaxPacketremoteWindowLeftr? assertIsNoneravatar)rcc2s r test_initzChannelTests.test_initys  DI . . . *F333 *F333 )5111 +Q/// *A... +++ !&!!! !(###  1aAq! 4 4 +Q/// +Q/// *A... ,a000 +Q/// !$$$ !$$$ A&&&&&rc|t|jd|ttjdddS)z{ Test that str(SSHChannel) works gives the channel name and local and remote windows at a glance.. z%rH localWindowzN)rOrrr@rs rtest_strzChannelTests.test_strsc T\**,STTT  "q111 2 24S     rc||jd|tjdddS)z Test that bytes(SSHChannel) works, gives the channel name and local and remote windows at a glance.. s%rHrZsN)rOr __bytes__r@rs r test_byteszChannelTests.test_bytesst  L " " $ $&N      1 - - - 7 7 9 9 ,     rc`||jddS)z Test that SSHChannel.logPrefix gives the name of the channel, the local channel ID and the underlying connection. z.SSHChannel channel (unknown) on MockConnectionN)rOrrrs rtest_logPrefixzChannelTests.test_logPrefixs:  L " " $ $&V     rcdgdfd }||j_|jd|jdd|jd||jjd||jj|d ||jj d ||j j |jdg||jj g||j j |jd gdd <|jd |d |jd |j|jd |d dS)zq Test that addWindowBytes adds bytes to the window and resumes writing if it was paused. Frrcdd<dSNTrrcbsrstubStartWritingz:ChannelTests.test_addWindowBytes..stubStartWritingBqEEErtestrH2*rr)rHrisPaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaNr,)r startWritingwrite writeExtendedaddWindowBytesrOrSrD areWritingbufr>rextBufr assertFalseloseConnection)rrgrfs @rtest_addWindowBytesz ChannelTests.test_addWindowBytess W      %5 ! 7### ""1g... ##B''' 6 CCC  /000 1 )3///  5yAAA ,b111 *4<8<.III1 ##B''' A 9%%% ##%%% ##B''' Arc4d|j_||jdd||jdd||jdddS)zn Test that requestReceived handles requests by dispatching them to request_* methods. c|dkS)Nrr)rs rz3ChannelTests.test_requestReceived..s  rs test-methodras bad-methodN)rrequest_test_methodrDrequestReceivedrtrs rtest_requestReceivedz!ChannelTests.test_requestReceiveds ,D+C (  44^SIIJJJ 55ndKKLLL 55mSIIJJJJJrc||jj|j||jjdS)zM Test that the default closeReceieved closes the connection. N)rtrclosing closeReceivedrDrs rtest_closeReceievedz ChannelTests.test_closeReceievedsN -... ""$$$  ,-----rcdgdfd }||j_|jd|jd||jj|d|jd|jd |jj|j}| |d d g| |jj d |jd | |gd | |jj ddd<|jd||jj|d| |gd| |jj d| |jj ddS)z Test that write handles data correctly. Send data up to the size of the remote window, splitting the data into packets of length remoteMaxPacket. Frrcdd<dSrdrresrstubStopWritingz0ChannelTests.test_write..stubStopWritingrhrdrzrrltada 12345678901)rr 12345678901rL123456)rrrr123456Nr,) r stopWritingrnrtrqrDrpr>rrOrSrrrrrrfs @r test_writezChannelTests.test_writes W      $3   4    4    0111 1 ##B''' 5!!!y~dl+ u~... 6;;; >*** BBBCCC 6:::1 9%%% 0111 1 LLLMMM )4000 6:::::rc`dgdfd }||j_|jdd|jdd|jdd ||jj|d |jd |jdd|jj|j}| |gd | |jj d |jdd| |gd| |jj ddd <|jdd||jj|d | |gd| |jj ddgg| |jj d dS)z Test that writeExtended handles data correctly. Send extended data up to the size of the window, splitting the extended data into packets of length remoteMaxPacket. Frrcdd<dSrdrresrrz8ChannelTests.test_writeExtended..stubStopWritingrhrrHrrzrItrrl)rHrrIrrIrzrrJr)rrrrJrrJrrLrKr)rrrrr)rKrrNr,) rrrortrqrDrpr>rrOrSrsrs @rtest_writeExtendedzChannelTests.test_writeExtended s; W      $3   ""1d+++ ""1d+++ ""1d+++ 0111 1 ##B''' ""1d+++y . AAABBB 6;;; ""1n555  SSS    6:::1 ""1i000 0111 1       ,4yk::: 6:::::rc|jd|jdtdD||jj|jdgdS)zS Test that writeSequence is equivalent to write(''.join(sequece)). rlc3"K|] }d|fzV dS)s%dNr).0is r z2ChannelTests.test_writeSequence..Cs("C"CA5A4<"C"C"C"C"C"Crr=s 0123456789N)rrp writeSequencerangerOr>rrs rtest_writeSequencezChannelTests.test_writeSequence>sn ##B''' """C"Cr"C"C"CCCC  5 GGGGGrcT|jd|jdd|j||jj|j|jd||jj|j|jd| |jj|jdS)zk Tesyt that loseConnection() doesn't close the channel until all the data is sent. sdatarHsdatadatarKN) rrnrorurTr>rgetrprDrs rtest_loseConnectionz ChannelTests.test_loseConnectionFs 7### ""1k222 ##%%% $)*..t|<<=== ##A&&& $)*..t|<<=== ##A&&&  (,,T\::;;;;;rctddd}t|jj||t ||jdS)z L{SSHChannel.getPeer} returns the same object as the underlying transport's C{getPeer} method returns. TCPz 192.168.0.1i1)r1r4N)r r9rr>rOrgetPeer)rpeers r test_getPeerzChannelTests.test_getPeerTs` 5-77DL$54HHHH ,T22DL4H4H4J4JKKKKKrctddd}t|jj||t ||jdS)z L{SSHChannel.getHost} returns the same object as the underlying transport's C{getHost} method returns. rz 127.0.0.1i90)r1r2N)r r9rr>rOrgetHost)rhosts r test_getHostzChannelTests.test_getHost^s` 5+u55DL$54HHHH ,T22DL4H4H4J4JKKKKKrNr,)r-r.r/r0rBrFrXr\r_rarvr}rrrrrrrrrrr;r;csb''''KKKK ''''<                 >KKKK....";";";";H/;/;/;/;bHHHH < < < <LLLLLLLLLLrr;)NN)r1rr2r3r4r3rr)r0 __future__runittestrzope.interface.verifyrtwisted.conch.sshrtwisted.conch.ssh.addressrtwisted.conch.ssh.servicertwisted.conch.ssh.transportr twisted.internetr twisted.internet.addressr twisted.internet.testingr skipTest ImportErrorobjecttwisted.trial.unittestrrr9r;rrrrs#"""""...... ))))))======444444>>>>>>++++++444444888888HH)HJJJ,+++++*$*$*$*$*$Z*$*$*$^/3.2"""""2(BLBLBLBLBL8BLBLBLBLBLs,A AA