B 0²ô`ÿã@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 úF/home/dcms/DCMS/lib/python3.7/site-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 ¡dS)N)rrÚloseConnection)rÚreasonr r r ÚconnectionLostszSOCKSv4Outgoing.connectionLostcCs|j |¡dS)N)rÚwrite)rÚdatar r r Ú dataReceived szSOCKSv4Outgoing.dataReceivedcCs|j ||¡|j |¡dS)N)rrrr)rrr r r r#szSOCKSv4Outgoing.writeN)Ú__name__Ú __module__Ú __qualname__r rrrrr r r r rs rc@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚSOCKSv4IncomingcCs||_||j_dS)N)rr)rrr r r r )szSOCKSv4Incoming.__init__cCs|jj ¡dS)N)rrr)rrr r r r-szSOCKSv4Incoming.connectionLostcCs|j |¡dS)N)rr)rrr r r r0szSOCKSv4Incoming.dataReceivedcCs|j ||¡|j |¡dS)N)rrrr)rrr r r r3szSOCKSv4Incoming.writeN)rrrr rrrr r r r r(src@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||_||_dS)N)Úloggingr)rr!rr r r r LszSOCKSv4.__init__cCsd|_d|_dS)Nó)Úbufr)rr r r rPszSOCKSv4.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óéz!BBHéééscSs | d¡S)Né[)r)Úresultrr r r Úor"z&SOCKSv4.dataReceived..) rrr#ÚstructÚunpackÚsplitrÚresolveÚ addCallbackÚ_dataReceived2Ú addErrbackÚsocketÚ inet_ntoa) rrZcompleteBufferÚheadÚversionÚcoder ÚuserÚserverÚdr r r rTs(   $  zSOCKSv4.dataReceivedcCs¸|dkstd|ƒ‚| ||||¡s2| d¡dS|dkr^| ||t|¡}| |fdd„¡n:|dkrŠ| d t||¡}| |fd d„¡nt d   |¡ƒ‚|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: %sr)Nr'cSs | d¡S)Nr))r)r*rr r r r+“r"z(SOCKSv4._dataReceived2..ércSs| dd|d|d¡S)Nr rr')r)Úxrr r r r+–r"zBad Connect Code: {}r"z hmm, still stuff in buffer... %s) ÚAssertionErrorÚ authorizerÚ connectClassrr2Ú listenClassÚSOCKSv4IncomingFactoryr0Ú RuntimeErrorÚformatr#Úrepr)rr9r8r6r7r r:r r r r1vs zSOCKSv4._dataReceived2cCs|jr|jj ¡dS)N)rrr)rrr r r r›szSOCKSv4.connectionLostcCst d||||f¡dS)Nz0code %s connection to %s:%s (user %s) authorizedr')rÚmsg)rr7r9r r8r r r r>ŸszSOCKSv4.authorizecGstjt|f|žŽ ||¡S)N)rZ ClientCreatorrZ connectTCP)rrr ÚklassÚargsr r r r?¦szSOCKSv4.connectClasscGs&t |||Ž¡}t | ¡dd…¡S)Nr')rZ listenTCPrZsucceedZgetHost)rr rFrGZservr r r r@©szSOCKSv4.listenClassrú0.0.0.0cCs8|j t d|||¡t |¡¡|dkr4|j ¡dS)Nz!BBHr )rrr,Úpackr3Ú inet_atonr)rZreplyr6r rr r r r­szSOCKSv4.makeReplycCs| ||¡|j |¡dS)N)rrr)rrr r r r´s z SOCKSv4.writec Cs|js dS|j ¡}|jj ¡}t|jdƒ}| dt ¡|j|j ||krLdpNd|j|j f¡xš|rø|dd…|dd…}}| t   t dd„|ƒd¡d¡| dt |ƒd d¡x2|D]*}t t|ƒƒd krÞ| d ¡q¾| |¡q¾W| d ¡q`W| d ¡| ¡dS) NÚaz%s %s:%d %s %s:%d ú<ú>écSs dt|ƒS)Nz%02X)Úord)r<r r r r+Ër"zSOCKSv4.log..ú éÚ.Ú )r!rrrÚopenrÚtimeÚctimerr ÚstringÚjoinÚmapÚlenrDÚclose)rÚprotorrZ their_peerÚfÚpÚcr r r r¸s0        z SOCKSv4.log)rrrH)rrrÚ__doc__rr rrr1rr>r?r@rrrr r r r r 8s"% r c@s eZdZdZdd„Zdd„ZdS)ÚSOCKSv4Factoryz` A factory for a SOCKSv4 proxy. Constructor accepts one argument, a log file name. cCs ||_dS)N)r!)rrr r r r ÞszSOCKSv4Factory.__init__cCs t|jtƒS)N)r r!r)rÚaddrr r r Ú buildProtocolászSOCKSv4Factory.buildProtocolN)rrrr`r rcr r r r ra×srac@s eZdZdZdd„Zdd„ZdS)rAzJ A utility class for building protocols for incoming connections. cCs||_||_dS)N)rr)rrrr r r r êszSOCKSv4IncomingFactory.__init__cCsV|d|jkr,d|_|j dd¡t|jƒS|jdkr:dS|j dd¡d|_dSdS)NrÚr r))rrrr)rrbr r r rcîs  z$SOCKSv4IncomingFactory.buildProtocolN)rrrr`r rcr r r r rAåsrA)r`r,rWr3rUZtwisted.internetrrrZtwisted.pythonrZProtocolrrr ZFactoryrarAr r r r Ús