tfh<dZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZeejGd d ZGd d ejZGd dejZGddejZGddejZdS)z$ Tests for L{twisted.conch.telnet}. ) implementer) verifyObject)telnet)defer) iterbytes) proto_helpers)unittestc^eZdZdZdZdZdZdZdZdZ dZ dZ d Z d Z d Zd Zd ZdS) TestProtocolchd|_g|_g|_g|_g|_g|_g|_dS)N)datasubcmdcalls enabledLocal enabledRemote disabledLocaldisabledRemoteselfs ]/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/conch/test/test_telnet.py__init__zTestProtocol.__init__s=    rcix}|_j|d<|jx}|_dD]}|ffd |t t |< dS)N) EORNOPDMBRKIPAOAYTECELGAc8j|SN)rappend)argcmdrs rz-TestProtocol.makeConnection..)s4:;L;LS;Q;Qr)negotiationMapneg_TEST_COMMAND commandMapcopygetattrr)r transportdr*s` rmakeConnectionzTestProtocol.makeConnection#s{'))I $*' #,#7#<#<#>#>>I S R RC69&Q&Q&Q&Q&QAgfc"" # # R Rrc&|xj|z c_dSr')rrrs r dataReceivedzTestProtocol.dataReceived+s T rcdSr'r )rreasons rconnectionLostzTestProtocol.connectionLost.s rc||_dSr')r)rpayloads rr-zTestProtocol.neg_TEST_COMMAND1s  rcP||jvr|j|dSdSNTF)localEnableablerr(roptions r enableLocalzTestProtocol.enableLocal4s1 T) ) )   $ $V , , ,4urc:|j|dSr')rr(r?s r disableLocalzTestProtocol.disableLocal:s !!&)))))rcP||jvr|j|dSdSr=)remoteEnableablerr(r?s r enableRemotezTestProtocol.enableRemote=s1 T* * *   % %f - - -4urc:|j|dSr')rr(r?s r disableRemotezTestProtocol.disableRemoteCs ""6*****rcdSr'r rs rconnectionMadezTestProtocol.connectionMadeF rcdSr'r )rcommandarguments runhandledCommandzTestProtocol.unhandledCommandJrKrcdSr'r rrMrs runhandledSubnegotiationz$TestProtocol.unhandledSubnegotiationNrKrN)__name__ __module__ __qualname__r>rErr3r6r9r-rArCrFrHrJrOrRr rrr r sO!!!RRR    *** +++           rr ceZdZdZdS)InterfacesTestsc`tj}ttj|dS)zO L{telnet.TelnetProtocol} implements L{telnet.ITelnetProtocol} N)rTelnetProtocolrITelnetProtocol)rps rtest_interfacezInterfacesTests.test_interfaceTs+  ! # #V+Q/////rN)rSrTrUr\r rrrWrWSs#00000rrWceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZggggfdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'S)(TelnetTransportTestsz. Tests for L{telnet.TelnetTransport}. ctjt|_t j|_|j|jdSr')rTelnetTransportr r[rStringTransporttr3rs rsetUpzTelnetTransportTests.setUpasB' 55.00 df%%%%%rc|jj}gd}|D]}|j|||jd|dS)N)shere are some bytes la la lassome more arrive hereslots of bytes to play withsla la lasta de dasdumr)r[protocolr6 assertEqualrjoinrhLbs rtestRegularBytesz%TelnetTransportTests.testRegularBytesfsp FO    # #A F   " " " " !-----rc4|jj}gd}|D]}|j|||j|ddddz|dddzdz|dddzdz|dddzdzdS) N)shere is the first line shere is the second line shere is the third line shere is the last line r  )r[rer6rfrrhs rtestNewlineHandlingz(TelnetTransportTests.testNewlineHandlingys FO    # #A F   " " " "  F aD"I d3B3i  d3B3i      d3B3i    rc|jj}gd}|D]}|j|||jd|dddS)N)s*here are some bytes with an embedded IACs&and here is a test of a border escapes did you get that IAC?rs)r[rer6rfrrgreplacerhs r testIACEscapez"TelnetTransportTests.testIACEscapes FO     # #A F   " " " " !!4!4['!J!JKKKKKrcp|jj}tjt t|z}dd|zdzg}|D]}|j|||j|g||jd | |ddS)NsHere's some bytes, tra la lasBut ono!s an interruptr) r[rerIACr0r6rfrrrgrw)rcmdNamerir*rjrks r_simpleCommandTestz'TelnetTransportTests._simpleCommandTests FOj767333 ,kC.?BR.R S # #A F   " " " " 7),,, !!4!4S#!>!>?????rc0|ddS)Nr r|rs r testInterruptz"TelnetTransportTests.testInterrupt %%%%%rc0|ddS)Nrr~rs rtestEndOfRecordz$TelnetTransportTests.testEndOfRecord &&&&&rc0|ddS)Nrr~rs rtestNoOperationz$TelnetTransportTests.testNoOperationrrc0|ddS)Nrr~rs r testDataMarkz!TelnetTransportTests.testDataMarkrrc0|ddS)Nrr~rs r testBreakzTelnetTransportTests.testBreakrrc0|ddS)Nr!r~rs rtestAbortOutputz$TelnetTransportTests.testAbortOutputrrc0|ddS)Nr"r~rs rtestAreYouTherez$TelnetTransportTests.testAreYouThererrc0|ddS)Nr#r~rs rtestEraseCharacterz'TelnetTransportTests.testEraseCharacterrrc0|ddS)Nr$r~rs r testEraseLinez"TelnetTransportTests.testEraseLinerrc0|ddS)Nr%r~rs r testGoAheadz TelnetTransportTests.testGoAheadrrc|jj}tjtjzdztjztjz}d|zdg}|D]}|j|||jd | |d||j ttddS)Ns hello worldsThese are some bytes but soonsthere will be some morers hello world)r[rerrzSBSEr6rfrrgrwrlistrrrir*rjrks rtestSubnegotiationz'TelnetTransportTests.testSubnegotiations FOj69$'99FJFR - 35O P # #A F   " " " " !!4!4S#!>!>??? 4 .(A(A#B#BCCCCCrc|jj}tjtjzdztjztjztjz}d|zdzdg}|D]}|j|||jd | |d||j tjgdS)NrsSome bytes are heresand herer) r[rerrzrrr6rfrrgrwrrs r testSubnegotiationWithEmbeddedSEz5TelnetTransportTests.testSubnegotiationWithEmbeddedSEs FOj69$w.:VZG&)S #c )K 7 E # #A F   " " " " !!4!4S#!>!>??? FI;/////rc tjtjzdztjzdztjztjz}t t |D]}t x}|j_| |j|d|||d}}d|z|dzg}|D]}|j || |j d ||d| |jtjgt!t#dzdS)Nrshellos first parts last partr)rrzrrrangelenr r[rer3r6rfrrgrwrrr)rr*iriarkrjrs rtestBoundarySubnegotiationz/TelnetTransportTests.testBoundarySubnegotiations\ Ji  i    j   i   s3xx P PA"... 0A  TV $ $ $rr7CGqA"A $45A * *##D))))   QVSXXa[[%8%8c%B%B C C C   QX {T)H:M:M5N5N'N O O O O P Prc||j|||j|||j|||j|dSr')rfrrrr)roeLeRdLdRs r_enabledHelperz#TelnetTransportTests._enabledHelper si ,,, "--- "--- )2.....rctjtjzdz}d|zdz}|j|||jjj||d||j tjtj zdz| |jjdSNrssurrounding bytessto spice things upr) rrzWILLr[r6rfrerrwrbvalueDONTrrr*rs rtestRefuseWillz#TelnetTransportTests.testRefuseWillsj6;&0#c),AA D!!! -t||C/E/EFFF fk)AG)KLLL DFO,,,,,rctjtjzdz}d|zdz}|j|||jjj||d||j tjtj zdz| |jjdSr) rrzDOr[r6rfrerrwrbrWONTrrs r testRefuseDoz!TelnetTransportTests.testRefuseDosj69$w.#c),AA D!!! -t||C/E/EFFF fk)AG)KLLL DFO,,,,,rcftjtjzdz}d|zdz}|jj}d|_|j|||j tjtj zdz| |dgdS)Npaddingstrailer)r)r) rrzrr[rer>r6rfrbrrrrr*rris r testAcceptDoz!TelnetTransportTests.testAcceptDo&sj69$w.C*, FO& D!!! fk)AG)KLLL A7),,,,,rcftjtjzdz}d|zdz}|jj}d|_|j|||j tjtj zdz| |dgdS)Nsheaderr)rr) rrzrr[rerEr6rfrbrrrrs rtestAcceptWillz#TelnetTransportTests.testAcceptWill3sj6;&03+ FO' D!!! fi)?')IJJJ A7),,,,,rcLtjtjzdz}|jd}d|j_d|z}|j|||jj j | |d||j tjtjzdz||jjd||jj dgdS)N)yess fiddle deernor)rrzrr[getOptionStatehimstater6rfrerrwrbrrrrr*srs rtestAcceptWontz#TelnetTransportTests.testAcceptWont?sj6;&0 F ! !' * * s" D!!! -t||C/E/EFFF fk)AG)KLLL d+++ DFO :::::rcLtjtjzdz}|jd}d|j_d|z}|j|||jj j | |d||j tjtjzdz||jjd||jj dgdS)Nrrs fiddle dum rr)r)rrzrr[rusrr6rfrerrwrbrrrrs rtestAcceptDontz#TelnetTransportTests.testAcceptDontSsj6;&0 F ! !' * * # D!!! -t||C/E/EFFF fk)AG)KLLL T*** DFO :::::rctjtjzdz}d|zdz}|j|||jjj||d||j d| |jjdSNG dum de dum tra la lar) rrzrr[r6rfrerrwrbrrrs rtestIgnoreWontz#TelnetTransportTests.testIgnoreWontgsj6;&0s"\1 D!!! -t||C/E/EFFF --- DFO,,,,,rctjtjzdz}d|zdz}|j|||jjj||d||j d| |jjdSr) rrzrr[r6rfrerrwrbrrrs rtestIgnoreDontz#TelnetTransportTests.testIgnoreDontssj6;&0s"\1 D!!! -t||C/E/EFFF --- DFO,,,,,rctjtjzdz}|jd}d|j_d|zdz}|j|||jj j | |d||j d||jj dSNVrrrr)rrzrr[rrrr6rfrerrwrbrrrs rtestIgnoreWillz#TelnetTransportTests.testIgnoreWillsj6;&0 F ! !' * * c!M1 D!!! -t||C/E/EFFF --- DFO,,,,,rctjtjzdz}|jd}d|j_d|zdz}|j|||jj j | |d||j d||jj dSr)rrzrr[rrrr6rfrerrwrbrrrs r testIgnoreDoz!TelnetTransportTests.testIgnoreDosj69$w. F ! !' * * c!M1 D!!! -t||C/E/EFFF --- DFO,,,,,rcjd}jjd_jtjtj zdzj tjtj zdz| jd| fd|S)NBrTc4dgS)Nrr)r)_rirs rr+z@TelnetTransportTests.testAcceptedEnableRequest..s 3 3A7) 3 D Dr) r[dorerErfrbrrrzrr6r addCallback)rr2ris` @rtestAcceptedEnableRequestz.TelnetTransportTests.testAcceptedEnableRequests FIIg   FO' fi)?')IJJJ FJ4w>??? d&--- DDDDDEEErc(djj_jd}jtjtj zdzj dj j dj j dj jj jjtjtjzdz|tj}|fd|fd|S)z If the peer refuses to enable an option we request it to enable, the L{Deferred} returned by L{TelnetProtocol.do} fires with an L{OptionRefused} L{Failure}. rrrcBjjSr'rr[reignoredrs rr+z@TelnetTransportTests.test_refusedEnableRequest..d&9&9$&/&J&JrcBjjSr') assertFalser negotiatingrrrs rr+z@TelnetTransportTests.test_refusedEnableRequest..sd&6&6qu7H&I&Ir)r[rerErrfrbrrrzrrrrr assertTruerrr6r assertFailure OptionRefusedrrr2rs` @rtest_refusedEnableRequestz.TelnetTransportTests.test_refusedEnableRequestsH,6( FIIg   fi)?')IJJJ F ! !' * * d+++ T*** )*** )*** FJ4w>???   q&"6 7 7 JJJJKKK IIIIIJJJrc(djj_jd}jtjtj zdzj dj j dj j dj jj jjtjtjzdz|tj}|fd|fd|S)z If the peer refuses to allow us to enable an option, the L{Deferred} returned by L{TelnetProtocol.will} fires with an L{OptionRefused} L{Failure}. rrrcBjjSr'rrs rr+z>TelnetTransportTests.test_refusedEnableOffer..rrcBjjSr')rrrrs rr+z>TelnetTransportTests.test_refusedEnableOffer..sd&6&6qt7G&H&Hr)r[rer>willrfrbrrrzrrrrrrrrr6rrrrrs` @rtest_refusedEnableOfferz,TelnetTransportTests.test_refusedEnableOffersH+5' FKK  fk)AG)KLLL F ! !' * * d+++ T*** *+++ ())) FJ4w>???   q&"6 7 7 JJJJKKK HHHHHIIIrcjd}d|j_jd}jtj tj zdzj tj tj zdz| jd| fd|S)NrrTcHjjdgS)Nrrrrrs rr+zATelnetTransportTests.testAcceptedDisableRequest..s  3 3DFO 3 R Rr)r[rrrdontrfrbrrrzrr6rrrrr2s` rtestAcceptedDisableRequestz/TelnetTransportTests.testAcceptedDisableRequests F ! !' * * FKK  fk)AG)KLLL FJ4w>??? d&--- RRRRSSSrcjd}d|j_jdfd}fd}fd}|d}|||||S)N$rcxjd}|tjSNr)r[rrrAlreadyNegotiatingxr2rs r_dozITelnetTransportTests.testNegotiationBlocksFurtherNegotiation.._do s/ '""A%%a)BCC Crcxjd}|tjSr)r[rrrrrs r_dontzKTelnetTransportTests.testNegotiationBlocksFurtherNegotiation.._donts/ G$$A%%a)BCC Crcjtjtjzdzjjdgdjj_jd}jtjtj zdz| j d| fd|S)Nrr)rTcLjjdgdgS)Nr)rrrrs rr+z^TelnetTransportTests.testNegotiationBlocksFurtherNegotiation.._final..s+$--FO wi.r) r[r6rrzrrrerErrrrfrs r_finalzLTelnetTransportTests.testNegotiationBlocksFurtherNegotiation.._finals F   V[ 87 B C C C   WI  > > >/9DFO , '""A F   V[ 87 B C C C MM$*D 1 1 1 MM    Hr)r[rrrrr)rrrrrr2s` r'testNegotiationBlocksFurtherNegotiationz.TestExceptionN)rSrTrUr rr TestExceptionr8s DrrzTotal failure!)r[r Exceptionr9rr gatherResults)rd1d2d3rs r testLostConnectionFailsDeferredsz5TelnetTransportTests.testLostConnectionFailsDeferreds3s VYYw   VYYw   VYYw       I    mm,<==>>>   M 2 2   M 2 2   M 2 2"BB<000rN))rSrTrU__doc__rcrlrtrxr|rrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrr rrr^r^\sf&&& ...&   6LLL" @ @ @&&&''''''&&&'''&&&'''&&&&&&&&& D D D 0 0 0PPP6$&"//// - - - - - - - - - - - -;;;(;;;( - - - - - ----&---&$88$###J=== <<< 1 1 1 1 1rr^c*eZdZdZdZdZdZdZdS) TestTelnetzP A trivial extension of the telnet protocol class useful to unit tests. cRtj|g|_dSr')rTelnetreventsrs rrzTestTelnet.__init__Hs# t$$$ rc>|jd|fdS)z: Record the given data in C{self.events}. bytesNr!r(r5s rapplicationDataReceivedz"TestTelnet.applicationDataReceivedLs$ GT?+++++rc@|jd||fdS)z= Record the given command in C{self.events}. rMNr$rQs rrOzTestTelnet.unhandledCommandRs' Iw566666rc@|jd||fdS)zL Record the given subnegotiation command in C{self.events}. negotiateNr$rQs rrRz"TestTelnet.unhandledSubnegotiationXs' K$788888rN)rSrTrUrrr%rOrRr rrrrCsZ,,, 777 99999rrcreZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdS) TelnetTestsz Tests for L{telnet.Telnet}. L{telnet.Telnet} implements the TELNET protocol (RFC 854), including option and suboption negotiation, and option state tracking. c,t|_dS)zM Create an unconnected L{telnet.Telnet} to be used by tests. N)rrers rrczTelnetTests.setUpgs#  rc`||jddS)z L{telnet.Telnet.enableLocal} should reject all options, since L{telnet.Telnet} does not know how to implement any options. N)rrerArs rtest_enableLocalzTelnetTests.test_enableLocalms. 22599:::::rc`||jddS)z L{telnet.Telnet.enableRemote} should reject all options, since L{telnet.Telnet} does not know how to implement any options. r-N)rrerFrs rtest_enableRemotezTelnetTests.test_enableRemotets. 33E::;;;;;rcR|t|jjddS)a  It is an error for L{telnet.Telnet.disableLocal} to be called, since L{telnet.Telnet.enableLocal} will never allow any options to be enabled locally. If a subclass overrides enableLocal, it must also override disableLocal. r-N) assertRaisesNotImplementedErrorrerCrs rtest_disableLocalzTelnetTests.test_disableLocal{s( -t}/I5QQQQQrcR|t|jjddS)a It is an error for L{telnet.Telnet.disableRemote} to be called, since L{telnet.Telnet.enableRemote} will never allow any options to be enabled remotely. If a subclass overrides enableRemote, it must also override disableRemote. r-N)r2r3rerHrs rtest_disableRemotezTelnetTests.test_disableRemotes( -t}/JERRRRRrctj}|j||jdd||ddS)z L{telnet.Telnet.requestNegotiation} formats the feature byte and the payload bytes into the subnegotiation format and sends them. See RFC 855. ssNrrarer3requestNegotiationrfrrr1s rtest_requestNegotiationz#TelnetTests.test_requestNegotiationsr"133  $$Y/// ((+>>>  OO   +     rctj}|j||jdd||ddS)z If the payload for a subnegotiation includes I{IAC}, it is escaped by L{telnet.Telnet.requestNegotiation} with another I{IAC}. See RFC 855. r8rvsNr9r;s r!test_requestNegotiationEscapesIACz-TelnetTests.test_requestNegotiationEscapesIACsg"133  $$Y/// (('::: **,KLLLLLrcgx}|j_|j|||t |dS)z Pass the given bytes to the protocol's C{dataReceived} method and assert that the given events occur. N)rer!r6rfr)rrexpectedreceiveds r_deliverzTelnetTests._deliversK +-,4=' ""4((( 4>>22222rc2|dddS)zc One application-data byte in the default state gets delivered right away. a)r#rDNrBrs rtest_oneApplicationDataBytez'TelnetTests.test_oneApplicationDataBytes dO,,,,,rc2|dddS)za Two application-data bytes in the default state get delivered together. bc)r#rHNrErs rtest_twoApplicationDataBytesz(TelnetTests.test_twoApplicationDataBytess e-.....rcL|dtjzddS)z~ Three application-data bytes followed by a control byte get delivered, but the control byte doesn't. def)r#rKNrBrrzrs rtest_threeApplicationDataBytesz*TelnetTests.test_threeApplicationDataBytess& fvz)+<=====rc|tj|tjdzdtjdzfdS)zy IAC in the escaped state gets delivered and so does another application-data byte following it. gr#NrLrs rtest_escapedControlzTelnetTests.test_escapedControlsE fj!!! fj4''6:3D)EFFFFFrc:|d|dd|dd|d|dd|dd|d|dd |d d |d|tjtjzd zd dtjzd zfd S)a A carriage return only puts the protocol into the newline state. A linefeed in the newline state causes just the newline to be delivered. A nul in the newline state causes a carriage return to be delivered. An IAC in the newline state causes a carriage return to be delivered and puts the protocol into the escaped state. Anything else causes a carriage return and that thing to be delivered. rqro)r#ros r-)r#rqs rD)r# arRxr#NrLrs rtest_carriageReturnzTelnetTests.test_carriageReturns e e-... g/000 e e-... g/000 e d-... f/000 e J #d *Wefj6H46O,P     rc|dtjztjzddtjdfdS)zt Application bytes received before a command are delivered before the command is processed. rS)r#rSrMN)rBrrzrrs r'test_applicationDataBeforeSimpleCommandz3TelnetTests.test_applicationDataBeforeSimpleCommandsD 6:  *   D )     rci|j_|dtjztjzdzddtjdfdS)z~ Application bytes received before a WILL/WONT/DO/DONT are delivered before the command is processed. yr-)r#rXrMN)rer.rBrrzrrs r!test_applicationDataBeforeCommandz-TelnetTests.test_applicationDataBeforeCommandsS $&   6:  +g 5   W -     rc|dtjztjzdztjztjzddddgfdS)z Application bytes received before a subnegotiation command are delivered before the negotiation is processed. zsQx)r#r[r(QrSN)rBrrzrrrs r(test_applicationDataBeforeSubnegotiationz4TelnetTests.test_applicationDataBeforeSubnegotiationsW 6:  )E 1FJ > J  $ '     rN)rSrTrUrrcr.r0r4r6r<r>rBrFrIrMrPrTrVrYr]r rrr*r*_s%%% ;;;<<<RRRSSS   M M M333---///>>>GGG   6              rr*N)rzope.interfacerzope.interface.verifyr twisted.conchrtwisted.internetrtwisted.python.compatr twisted.testr twisted.trialr rZr TestCaserWr^r rr*r rrrfs  '&&&&&...... """"""++++++&&&&&&"""""" V #$$; ; ; ; ; ; ; %$; |00000h'000d1d1d1d1d18,d1d1d1N999999998k k k k k (#k k k k k r