\ddZddlZddlZddlZddlZddlmZmZmZddl m Z Gddej Z Gddej Z Gd d ej ZGd d ejZGd dejZdS)z) Implementation of the SOCKSv4 protocol. N)deferprotocolreactor)logc,eZdZdZdZdZdZdZdS)SOCKSv4Outgoingc||_dSN)socksselfr s 7lib/python3.11/site-packages/twisted/protocols/socks.py__init__zSOCKSv4Outgoing.__init__s  c|j}|jdd|j|j||j_dS)NZr)portip) transportgetPeerr makeReplyrhost otherConn)r peers rconnectionMadezSOCKSv4Outgoing.connectionMadesH~%%'' RtyAAA# rcB|jjdSr r rloseConnectionr reasons rconnectionLostzSOCKSv4Outgoing.connectionLost ++-----rc:|j|dSr r writer datas r dataReceivedzSOCKSv4Outgoing.dataReceived! rcp|j|||j|dSr r rrr%r&s rr%zSOCKSv4Outgoing.write$4 tT""" T"""""rN)__name__ __module__ __qualname__rrr!r(r%rrrrs_$$$ ...#####rrc&eZdZdZdZdZdZdS)SOCKSv4Incomingc,||_||j_dSr )r rr s rrzSOCKSv4Incoming.__init__*s # rcB|jjdSr rrs rr!zSOCKSv4Incoming.connectionLost.r"rc:|j|dSr r$r&s rr(zSOCKSv4Incoming.dataReceived1r)rcp|j|||j|dSr r+r&s rr%zSOCKSv4Incoming.write4r,rN)r-r.r/rr!r(r%r0rrr2r2)sP$$$...#####rr2c\eZdZdZdefdZdZdZdZdZ dZ d Z d Z dd Z dZdZdS)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. Nc"||_||_dSr )loggingr)r r:rs rrzSOCKSv4.__init__Ms  rc"d|_d|_dS)Nr)bufr)r s rrzSOCKSv4.connectionMadeQsrc |jr|j|dS|j|z|_|j}d|jddvrB|jdd|jddc}|_tjd|dd\}}}|jdd\}|_|dddkr|dddkrd|jvr ||_dS|jdd\}|_|j|} | |j ||||| |fd dStj |dd}| |||||dSdS) z Called whenever data is received. @type data: L{bytes} @param data: Part or all of a SOCKSv4 packet. N!BBHsc,|dSN[rresultr s rz&SOCKSv4.dataReceived..pst~~b7I7Ir) rr%r<structunpacksplitrresolve addCallback_dataReceived2 addErrbacksocket inet_ntoa) r r'completeBufferheadversioncoderuserserverds rr(zSOCKSv4.dataReceivedUs >  N  & & & F8d? dhqrrl " "!Xbqb\48ABB>'1#=#= L((00 d14$MMM IIIJJJ)$qs)44   gtT B B B B B) # "rc|dks Jd|z|||||s|ddS|dkr6|||t|}||fdnN|dkr6|dt ||}||fd ntd ||j d ksJd t|j zdS) 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. rAzBad version code: %srFNrBc,|dSrErGrHs rrJz(SOCKSv4._dataReceived2..s4>>"3E3ErrcJ|dd|d|dS)NrrrBrG)xr s rrJz(SOCKSv4._dataReceived2..s t~~b!QqT1Q4/P/PrzBad Connect Code: rz hmm, still stuff in buffer... %s) authorizer connectClassrrQ listenClassSOCKSv4IncomingFactoryrO RuntimeErrorr<repr)r rYrXrVrWrrZs rrPzSOCKSv4._dataReceived2ws .!|||3g=|||~~dFD$77  NN2    F 199!!&$FFA LLTEEE F F F F QYY  $:D&IIA MMPPP Q Q Q Q:D::;; ;x3 BT$(^^ SrcT|jr |jjdSdSr )rrrrs rr!zSOCKSv4.connectionLosts3 > 6 N $ 3 3 5 5 5 5 5 6 6rc Htjd|d|d|d|d dS)Nzcode z connection to :z (user z ) authorizedrB)rmsg)r rWrYrrXs rr`zSOCKSv4.authorizes= ttVVVTTT444 )   qrcZtjt|g|R||Sr )r ClientCreatorr connectTCP)r rrklassargss rrazSOCKSv4.connectClasss-%guc&dt|zS)Nz%02X)ord)r_s rrJzSOCKSv4.log..sfs1vvor . )r:rrropenr%timectimerrstringjoinmaplenreclose)r protor'r their_peerfpcs rrz SOCKSv4.logs|  F~%%''^-5577 s # #  "   4-'C/C    3B3ibcctA GGFK$=$=q A A3GG#M N N N GGR#a&&[A%+ , , ,  tAww<rs]   6555555555#####h'###( # # # # #h' # # # \\\\\h\\\~ . . . . .X% . . .X-r