U T‚gùã@s¢dZddlZddlZddlZddlZddlmZmZmZddl m Z Gdd„dej ƒZ Gdd„dej ƒZ Gd d „d ej ƒZGd d „d ejƒZGd d„dejƒZdS)z) Implementation of the SOCKSv4 protocol. éN)ÚdeferÚprotocolÚreactor)Úlogc@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd S) ÚSOCKSv4OutgoingcCs ||_dS©N)Úsocks©Úselfr©r ú;/tmp/pip-unpacked-wheel-1_7di8ic/twisted/protocols/socks.pyÚ__init__szSOCKSv4Outgoing.__init__cCs.|j ¡}|jjdd|j|jd||j_dS)NéZr)ÚportÚip)Ú transportÚgetPeerrÚ makeReplyrÚhostÚ otherConn)r Úpeerr r r ÚconnectionMades zSOCKSv4Outgoing.connectionMadecCs|jj ¡dSr©rrÚloseConnection©r Úreasonr r r ÚconnectionLostszSOCKSv4Outgoing.connectionLostcCs|j |¡dSr©rÚwrite©r Údatar r r Ú dataReceived!szSOCKSv4Outgoing.dataReceivedcCs|j ||¡|j |¡dSr©rrrrrr r r r$szSOCKSv4Outgoing.writeN)Ú__name__Ú __module__Ú __qualname__r rrr!rr r r r rs rc@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚSOCKSv4IncomingcCs||_||j_dSr)rrr r r r r *szSOCKSv4Incoming.__init__cCs|jj ¡dSrrrr r r r.szSOCKSv4Incoming.connectionLostcCs|j |¡dSrrrr r r r!1szSOCKSv4Incoming.dataReceivedcCs|j ||¡|j |¡dSrr"rr r r r4szSOCKSv4Incoming.writeN)r#r$r%r rr!rr r r r r&)sr&c@speZdZdZdefdd„Zdd„Zdd„Zd d „Zd d „Z d d„Z dd„Z dd„Z ddd„Z dd„Zdd„ZdS)ÚSOCKSv4aË An implementation of the SOCKSv4 protocol. @type logging: L{str} or L{None} @ivar logging: If not L{None}, the name of the logfile to which connection information will be written. @type reactor: object providing L{twisted.internet.interfaces.IReactorTCP} @ivar reactor: The reactor used to create connections. @type buf: L{str} @ivar buf: Part of a SOCKSv4 connection request. @type otherConn: C{SOCKSv4Incoming}, C{SOCKSv4Outgoing} or L{None} @ivar otherConn: Until the connection has been established, C{otherConn} is L{None}. After that, it is the proxy-to-destination protocol instance along which the client's connection is being forwarded. NcCs||_||_dSr)Úloggingr)r r(rr r r r MszSOCKSv4.__init__cCsd|_d|_dS)Nó)Úbufr)r r r r rQszSOCKSv4.connectionMadec Cs4|jr|j |¡dS|j||_|j}d|jdd…kr0|jdd…|jdd…}|_t d|dd…¡\}}}|j dd¡\}|_|dd…dkr |dd…dkr d|jkrÂ||_dS|j dd¡\}|_|j |¡} |  |j ||||¡|   |fd d „¡dSt   |dd…¡}|  |||||¡dS) z‡ Called whenever data is received. @type data: L{bytes} @param data: Part or all of a SOCKSv4 packet. Nóéú!BBHéééscSs | d¡S©Né[©r©Úresultr r r r Úpr)z&SOCKSv4.dataReceived..) rrr*ÚstructÚunpackÚsplitrÚresolveÚ addCallbackÚ_dataReceived2Ú addErrbackÚsocketÚ inet_ntoa) r r ZcompleteBufferÚheadÚversionÚcoderÚuserÚserverÚdr r r r!Us(   $  zSOCKSv4.dataReceivedcCs¸|dkstd|ƒ‚| ||||¡s2| d¡dS|dkr^| ||t|¡}| |fdd„¡n:|dkrŠ| d t||¡}| |fd d„¡nt d |›ƒ‚|j d ks´td t |j ƒƒ‚dS)a4 The second half of the SOCKS connection setup. For a SOCKSv4 packet this is after the server address has been extracted from the header. For a SOCKSv4a packet this is after the host name has been resolved. @type server: L{str} @param server: The IP address of the destination, represented as a dotted quad. @type user: L{str} @param user: The username associated with the connection. @type version: L{int} @param version: The SOCKS protocol version number. @type code: L{int} @param code: The command code. 1 means establish a TCP/IP stream connection, and 2 means establish a TCP/IP port binding. @type port: L{int} @param port: The port number associated with the connection. r.zBad version code: %sr2Nr/cSs | d¡Sr1r3r4r r r r6”r)z(SOCKSv4._dataReceived2..ércSs| dd|d|d¡S)Nrrr/r3)Úxr r r r r6—r)zBad Connect Code: r)z hmm, still stuff in buffer... %s) ÚAssertionErrorÚ authorizerÚ connectClassrr=Ú listenClassÚSOCKSv4IncomingFactoryr;Ú RuntimeErrorr*Úrepr)r rDrCrArBrrEr r r r<ws zSOCKSv4._dataReceived2cCs|jr|jj ¡dSr)rrrrr r r rœszSOCKSv4.connectionLostcCst d||||f¡dS)Nz0code %s connection to %s:%s (user %s) authorizedr/)rÚmsg)r rBrDrrCr r r rI s  ÿÿzSOCKSv4.authorizecGstjt|f|žŽ ||¡Sr)rZ ClientCreatorrZ connectTCP)r rrÚklassÚargsr r r rJ§szSOCKSv4.connectClasscGs&t |||Ž¡}t | ¡dd…¡S)Nr/)rZ listenTCPrZsucceedZgetHost)r rrPrQZservr r r rKªszSOCKSv4.listenClassrú0.0.0.0cCs8|j t d|||¡t |¡¡|dkr4|j ¡dS)Nr-r)rrr7Úpackr>Ú inet_atonr)r ZreplyrArrr r r r®s ÿzSOCKSv4.makeReplycCs| ||¡|j |¡dSr)rrrrr r r rµs z SOCKSv4.writec Cs|js dS|j ¡}|jj ¡}t|jdƒ}| dt ¡|j|j ||krLdpNd|j|j f¡|rò|dd…|dd…}}| t   t dd„|ƒd¡d¡| dt |ƒd d¡|D]*}t t|ƒƒd krÚ| d ¡qº| |¡qº| d ¡q^| d ¡| ¡dS) NÚaz%s %s:%d %s %s:%d ú<ú>écSs dt|ƒS)Nz%02X)Úord)rGr r r r6Ìr)zSOCKSv4.log..ú éÚ.Ú )r(rrrÚopenrÚtimeÚctimerrÚstringÚjoinÚmapÚlenrNÚclose)r Úprotor rZ their_peerÚfÚpÚcr r r r¹s6   úÿÿ      z SOCKSv4.log)rrrR)r#r$r%Ú__doc__rr rr!r<rrIrJrKrrrr r r r r'9s"% r'c@s eZdZdZdd„Zdd„ZdS)ÚSOCKSv4Factoryz` A factory for a SOCKSv4 proxy. Constructor accepts one argument, a log file name. cCs ||_dSr)r()r rr r r r ßszSOCKSv4Factory.__init__cCs t|jtƒSr)r'r(r©r Úaddrr r r Ú buildProtocolâszSOCKSv4Factory.buildProtocolN©r#r$r%rjr rnr r r r rkØsrkc@s eZdZdZdd„Zdd„ZdS)rLzJ A utility class for building protocols for incoming connections. cCs||_||_dSr)rr)r rrr r r r ëszSOCKSv4IncomingFactory.__init__cCsV|d|jkr,d|_|j dd¡t|jƒS|jdkr:dS|j dd¡d|_dSdS)NrÚrr2)rrrr&rlr r r rnïs  z$SOCKSv4IncomingFactory.buildProtocolNror r r r rLæsrL)rjr>rar7r_Ztwisted.internetrrrZtwisted.pythonrÚProtocolrr&r'ZFactoryrkrLr r r r Ús