U  W[ˆã@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)ÚreactorÚprotocolÚdefer)Úlogc@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd S) ÚSOCKSv4OutgoingcCs ||_dS©N)Úsocks©Úselfr©r ú9/usr/lib/python3/dist-packages/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 r3szSOCKSv4Incoming.connectionLostcCs|j |¡dSrrrr r r r!7szSOCKSv4Incoming.dataReceivedcCs|j ||¡|j |¡dSrr"rr r r r;szSOCKSv4Incoming.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 TszSOCKSv4.__init__cCsd|_d|_dS)Nó)Úbufr)r r r r rYszSOCKSv4.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 Úzr)z&SOCKSv4.dataReceived..) rrr*ÚstructZunpackÚsplitrZresolveÚ addCallbackÚ_dataReceived2Ú addErrbackÚsocketZ inet_ntoa) r r ZcompleteBufferÚheadÚversionÚcoderÚuserÚserverÚdr r r r!^s0   $   ÿzSOCKSv4.dataReceivedcCs¸|dkstd|ƒ‚| ||||¡s2| d¡dS|dkr^| ||t|¡}| |fdd„¡n:|dkrŠ| d t||¡}| |fd d„¡nt d |fƒ‚|j d ks´td t |j ƒƒ‚dS)a3 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 comand 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¢szBad Connect Code: %sr)z hmm, still stuff in buffer... %s) ÚAssertionErrorÚ authorizerÚ connectClassrr;Ú listenClassÚSOCKSv4IncomingFactoryr9Ú RuntimeErrorr*Úrepr)r rAr@r>r?rrBr r r r:‚s  ÿÿ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 r?rArr@r r r rF¯szSOCKSv4.authorizecGstjt|f|žŽ ||¡Sr)rZ ClientCreatorrZ connectTCP)r rrÚklassÚargsr r r rG´szSOCKSv4.connectClasscGs&t |||Ž¡}t | ¡dd…¡S)Nr/)rZ listenTCPrZsucceedZgetHost)r rrMrNZservr r r rH¸szSOCKSv4.listenClassrú0.0.0.0cCs8|j t d|||¡t |¡¡|dkr4|j ¡dS)Nr-r)rrr7Zpackr<Z inet_atonr)r Zreplyr>rrr 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)rDr r r r6Òr)zSOCKSv4.log..ú éÚ.Ú )r(rrrÚopenrÚtimeÚctimerrÚstringÚjoinÚmapÚlenrKÚclose)r Úprotor rZ their_peerÚfÚpÚcr r r rÇs.    ý     z SOCKSv4.log)rrrO)r#r$r%Ú__doc__rr rr!r:rrFrGrHrrrr r r r r'As$( 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 Zaddrr r r Ú buildProtocolçszSOCKSv4Factory.buildProtocolN©r#r$r%rer rhr r r r rfÝsrfc@s eZdZdZdd„Zdd„ZdS)rIzJ 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&rgr r r rhõs  z$SOCKSv4IncomingFactory.buildProtocolNrir r r r rIìsrI)rer7r\r<rZZtwisted.internetrrrZtwisted.pythonrZProtocolrr&r'ZFactoryrfrIr r r r Ús