File: //lib/python3/dist-packages/twisted/protocols/__pycache__/socks.cpython-310.pyc
o
�b� � @ s� d Z ddlZddlZddlZddlZddlmZmZmZ ddl m
Z
G dd� dej �Z G dd� dej �Z
G d d
� d
ej �ZG d d � d ej�ZG d
d� dej�ZdS )z)
Implementation of the SOCKSv4 protocol.
� N)�defer�protocol�reactor)�logc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �SOCKSv4Outgoingc C �
|| _ d S �N)�socks��selfr � r �9/usr/lib/python3/dist-packages/twisted/protocols/socks.py�__init__ �
zSOCKSv4Outgoing.__init__c C s. | j �� }| jjdd|j|jd� | | j_d S )N�Z r )�port�ip)� transport�getPeerr � makeReplyr �host� otherConn)r �peerr r r
�connectionMade s
z SOCKSv4Outgoing.connectionMadec C � | j j�� d S r �r r �loseConnection�r �reasonr r r
�connectionLost � z SOCKSv4Outgoing.connectionLostc C � | j �|� d S r �r �write�r �datar r r
� dataReceived! r z SOCKSv4Outgoing.dataReceivedc C � | j �| |� | j�|� d S r �r r r r# r$ r r r
r# $ � zSOCKSv4Outgoing.writeN)�__name__�
__module__� __qualname__r r r r& r# r r r r
r s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�SOCKSv4Incomingc C s || _ | | j _d S r )r r r
r r r
r * s zSOCKSv4Incoming.__init__c C r r r r r r r
r . r z SOCKSv4Incoming.connectionLostc C r! r r" r$ r r r
r& 1 r z SOCKSv4Incoming.dataReceivedc C r' r r( r$ r r r
r# 4 r) zSOCKSv4Incoming.writeN)r* r+ r, r r r& r# r r r r
r- ) s
r- c @ sp e Zd ZdZdefdd�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.
Nc C � || _ || _d S r )�loggingr )r r0 r r r r
r M �
zSOCKSv4.__init__c C s d| _ d | _d S )N� )�bufr )r r r r
r Q r1 zSOCKSv4.connectionMadec
C s2 | j r | j �|� dS | j| | _| j}d| jdd� v r�| jdd� | jdd� }| _t�d|dd� �\}}}| j�dd�\}| _|dd� dkr�|dd� dkr�d| jvr^|| _dS | j�dd�\}| _| j�|�} | �| j ||||� | �
| fd d
�� dS t � |dd� �}| � |||||� dS dS ) z�
Called whenever data is received.
@type data: L{bytes}
@param data: Part or all of a SOCKSv4 packet.
N� � �!BBH� � � s c S �
|� d�S �N�[ �r ��resultr r r r
�<lambda>p �
z&SOCKSv4.dataReceived.<locals>.<lambda>)
r r# r3 �struct�unpack�splitr �resolve� addCallback�_dataReceived2�
addErrback�socket� inet_ntoa)
r r% �completeBuffer�head�version�coder �user�server�dr r r
r& U s*
�zSOCKSv4.dataReceivedc C s� |dks
J d| ��| � ||||�s| �d� dS |dkr/| �||t| �}|�| fdd�� n |dkrE| �d t| |�}|�| fd
d�� ntd |� ���| j d ksZJ d
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.
r7 zBad version code: %sr< Nr8 c S r: r; r= r> r r r
r@ � rA z(SOCKSv4._dataReceived2.<locals>.<lambda>� r c S s |� dd| d | d �S )Nr r r8 r= )�xr r r r
r@ � s zBad Connect Code: r2 z hmm, still stuff in buffer... %s) � authorizer � connectClassr rH � listenClass�SOCKSv4IncomingFactoryrF � RuntimeErrorr3 �repr)r rP rO rM rN r rQ r r r
rG w s
zSOCKSv4._dataReceived2c C s | j r | j j�� d S d S r )r r r r r r r
r � s �zSOCKSv4.connectionLostc C s t �d||||f � dS )Nz0code %s connection to %s:%s (user %s) authorizedr8 )r �msg)r rN rP r rO r r r
rT � s
��zSOCKSv4.authorizec G s t jt|g|�R � �||�S r )r �
ClientCreatorr �
connectTCP)r r r �klass�argsr r r
rU � s zSOCKSv4.connectClassc G s& t �|||� �}t�|�� dd � �S )Nr8 )r � listenTCPr �succeed�getHost)r r r] r^ �servr r r
rV � s zSOCKSv4.listenClassr �0.0.0.0c C s< | j �t�d|||�t�|� � |dkr | j �� d S d S )Nr6 r )r r# rB �packrI � inet_atonr )r �replyrM r r r r r
r � s ��zSOCKSv4.makeReplyc C s | � | |� | j�|� d S r )r r r# r$ r r r
r# � s z
SOCKSv4.writec C s
| j sd S | j�� }| jj�� }t| j d�}|�dt�� |j|j || kr&dp'd|j|j f � |rz|d d� |dd � }}|�t
� t dd� |�d�d � |�dt
|� d d � |D ]}t
t|��d krm|�d
� q]|�|� q]|�d � |s1|�d � |�� d S ) N�az%s %s:%d %s %s:%d
�<�>� c S s dt | � S )Nz%02X)�ord)rS r r r
r@ � s z SOCKSv4.log.<locals>.<lambda>� � �.�
)r0 r r r �openr# �time�ctimer r �string�join�map�lenrY �close)r �protor% r �
their_peer�f�p�cr r r
r � s8
���
�
z SOCKSv4.log)r r rc )r* r+ r, �__doc__r r r r& rG r rT rU rV r r# r r r r r
r. 9 s "%
r. c @ � e Zd ZdZdd� Zdd� ZdS )�SOCKSv4Factoryz`
A factory for a SOCKSv4 proxy.
Constructor accepts one argument, a log file name.
c C r r )r0 )r r r r r
r � r zSOCKSv4Factory.__init__c C s t | jt�S r )r. r0 r �r �addrr r r
�
buildProtocol� s z SOCKSv4Factory.buildProtocolN�r* r+ r, r} r r� r r r r
r � s r c @ r~ )rW zJ
A utility class for building protocols for incoming connections.
c C r/ r )r r )r r r r r r
r � r1 z SOCKSv4IncomingFactory.__init__c C sR |d | j krd| _ | j�dd� t| j�S | j dkr d S | j�dd� d| _ d S )Nr � r r< )r r r r- r� r r r
r� � s
z$SOCKSv4IncomingFactory.buildProtocolNr� r r r r
rW � s rW )r} rI rs rB rq �twisted.internetr r r �twisted.pythonr �Protocolr r- r. �Factoryr rW r r r r
�<module> s