\dXDdZddlZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZGd d e jZGd d ZGd de jZGddejZGddejZdS)z_ Tests for L{twisted.protocol.socks}, an implementation of the SOCKSv4 and SOCKSv4a protocols. N)addressdefer)DNSLookupError)socks) iterbytes) proto_helpers)unittestc(eZdZdZdZdZdZdZdS)StringTCPTransportFNc|jS)N)peerselfs 7lib/python3.11/site-packages/twisted/test/test_socks.pygetPeerzStringTCPTransport.getPeers yc.tjdddS)NTCPz2.3.4.5*)r IPv4Addressrs rgetHostzStringTCPTransport.getHosts"5)R888rcd|_dS)NT)stringTCPTransport_closingrs rloseConnectionz!StringTCPTransport.loseConnections*.'''r)__name__ __module__ __qualname__rr rrrrrr r sL!& D999/////rr ceZdZdZdZdZdS)FakeResolverReactorzP Bare-bones reactor with deterministic behavior for the resolve method. c||_dS)z @type names: L{dict} containing L{str} keys and L{str} values. @param names: A hostname to IP address mapping. The IP addresses are stringified dotted quads. N)names)rr"s r__init__zFakeResolverReactor.__init__'s  rc  tj|j|S#t$r:tjt d|dzcYSwxYw)zQ Resolve a hostname by looking it up in the C{names} dictionary. z"FakeResolverReactor couldn't find zutf-8)rsucceedr"KeyErrorfailrdecode)rhostnames rresolvezFakeResolverReactor.resolve/sx =H!566 6   :88??7;S;SS    s!AA%$A%N)rrr__doc__r#r*rrrr r "s<     rr c"eZdZdZdZdZdZdS) SOCKSv4DriverNc||}t|_tjd|||j_|||_tj|S)Nr) r transportrrr connectionMadedriver_outgoingrr%)rhostportklassargsprotos r connectClasszSOCKSv4Driver.connectClassDs]t ,..&25$EE $}U###rcV||}||_|dkrd}tjd|fS)Nr6.7.8.9) driver_listenrr%)rr3r4r5factorys r listenClasszSOCKSv4Driver.listenClassMs8%,$ 199D}i.///r)rrrr1r;r7r=rrrr-r-=s>OM$$$00000rr-cBeZdZdZdZdZdZdZdZdZ dZ d Z d S) ConnectTestszV Tests for SOCKS and SOCKSv4a connect requests using the L{SOCKSv4} protocol. ct|_t|j_|jt ddi|j_dSNs localhost 127.0.0.1r-sockr r/r0r reactorrs rsetUpzConnectTests.setUp[N!OO 022    """/{0KLL rcf|jj}|"||jjddSdS)Nz-Outgoing SOCKS connections need to be closed.)rDr1 assertTruer/r)routgoings rtearDownzConnectTests.tearDownasF9,   OO"=?      rc |jtjddddt jdzdzdz|jj}|jj| |tjddd dt jdz| |jjj | |jj |jd | |jj jd |jj d | |jjd |jd dS) N!BBH"1.2.3.4fooBARrZ hello, worldhi there fake reason)rD dataReceivedstructpacksocket inet_atonr/valueclear assertEqual assertFalserassertIsNotNoner1connectionLost)rsents r test_simplezConnectTests.test_simpleis  K1b ) )y)) *     y"((** !!###  &+faR0063CI3N3NN    ,GHHH TY6777 /// 2<BBDDoVVV !..{;;; ,2244kBBB   /////rc tjddddtjdzdzdz}t |D]}|j||jj}|jj | |tjddd dtjd z| |jjj | |jj|jd | |jjjd |jjd | |jjd |jd dS)u If the destination IP address has zeros for the first three octets and non-zero for the fourth octet, the client is attempting a v4a connection. A hostname is specified after the user ID string and the server connects to the address that hostname resolves to. @see: U{http://en.wikipedia.org/wiki/SOCKS#SOCKS_4a_protocol} rMrNrOrP0.0.0.1fooBAZ localhostrrTrBrUrVrWN)rYrZr[r\rrDrXr/r]r^r_r`rrar1rbr clientRequestbytercs r test_socks4aSuccessfulResolutionz-ConnectTests.test_socks4aSuccessfulResolutions K1b ) )y)) *   m,, ) )D I " "4 ( ( ( (y"((** !!###  &+faR0063CK3P3PP    ,GHHH TY6777 /// 2<BBDDoVVV !..{;;; ,2244kBBB   /////rc tjddddtjdzdzdz}t |D]}|j||jj}| |tjddd dtjd z| |jjj | |jj d S) Failed hostname resolution on a SOCKSv4a packet results in a 91 error response and the connection getting closed. rMrNrOrPrgrh failinghostr[0.0.0.0NrYrZr[r\rrDrXr/r]r_rIr assertIsNoner1rjs rtest_socks4aFailedResolutionz)ConnectTests.test_socks4aFailedResolution K1b ) )y)) *   m,, ) )D I " "4 ( ( ( (y"((**  &+faQ//&2B92M2MM    +FGGG $)344444rc d|j_|jtjddddt jdzdzdz||jj tjdd d d t jd z| |jjj | |jj dS) NcdSNrrcodeserverr3users rz0ConnectTests.test_accessDenied..qrrMrNrO10.2.3.4rRrSrrqrr)rD authorizerXrYrZr[r\r_r/r]rIrrtr1rs rtest_accessDeniedzConnectTests.test_accessDenieds@@   K1d + +z** +      I  % % ' ' K2q ) )F,.rrrMrNrrrrRrSrrqrr)rDrrXrYrZr[r\r_r/r]rIrrtr;rs rrzBindTests.test_accessDenieds@@   K1d + +z** +      I  % % ' ' K2q ) )F,HHHHHrr)r+r[rYtwisted.internetrrtwisted.internet.errorrtwisted.protocolsrtwisted.python.compatr twisted.testr twisted.trialr StringTransportr r SOCKSv4r-TestCaser?rrrrrs  ++++++++111111######++++++&&&&&&"""""" / / / / /6 / / /600000EM0002e0e0e0e0e08$e0e0e0PtHtHtHtHtH!tHtHtHtHtHr