File: //lib/python3/dist-packages/twisted/protocols/__pycache__/socks.cpython-38.pyc
U
W[� � @ 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)�reactor�protocol�defer)�logc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �SOCKSv4Outgoingc C s
|| _ d S �N)�socks��selfr � r �9/usr/lib/python3/dist-packages/twisted/protocols/socks.py�__init__ s 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 s | j j�� d S r �r r �loseConnection�r
�reasonr r r �connectionLost s z SOCKSv4Outgoing.connectionLostc C s | j �|� d S r �r �write�r
�datar r r � dataReceived# s z SOCKSv4Outgoing.dataReceivedc C s | j �| |� | j�|� d S r �r r r r r r r r r ' s 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 s | j j�� d S r r r r r r r 3 s z SOCKSv4Incoming.connectionLostc C s | j �|� d S r r r r r r r! 7 s z SOCKSv4Incoming.dataReceivedc C s | j �| |� | j�|� d S r r" r r r r r ; s 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 s || _ || _d S r )�loggingr )r
r( r r r r r
T s zSOCKSv4.__init__c C s d| _ d | _d S )N� )�bufr )r
r r r r Y s zSOCKSv4.connectionMadec
C s4 | j r| j �|� dS | j| | _| j}d| jdd� k�r0| jdd� | jdd� }| _t�d|dd� �\}}}| j�dd�\}| _|dd� dk�r |dd� dk�r d| jkr�|| _dS | j�dd�\}| _| j�|�} | �| j ||||� | �
| fd d
�� dS t � |dd� �}| � |||||� 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 s
|� d�S �N�[ �r ��resultr
r r r �<lambda>z r) z&SOCKSv4.dataReceived.<locals>.<lambda>)
r r r* �structZunpack�splitr Zresolve� addCallback�_dataReceived2�
addErrback�socketZ inet_ntoa)
r
r ZcompleteBuffer�head�version�coder �user�server�dr r r r! ^ s0 $
�zSOCKSv4.dataReceivedc C s� |dkst d| ��| �||||�s2| �d� dS |dkr^| �||t| �}|�| fdd�� n:|dkr�| �d t| |�}|�| fd
d�� nt d |f ��| j
d ks�t d
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: %sr2 Nr/ c S s
|� d�S r1 r3 r4 r r r r6 � r) z(SOCKSv4._dataReceived2.<locals>.<lambda>� r c S s |� dd| d | d �S )Nr r r/ r3 )�xr
r r r r6 � s zBad Connect Code: %sr) z hmm, still stuff in buffer... %s) �AssertionError� authorizer � connectClassr r; � listenClass�SOCKSv4IncomingFactoryr9 � RuntimeErrorr* �repr)r
rA r@ r> r? r rB r r r r: � s
��zSOCKSv4._dataReceived2c C s | j r| j j�� 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) authorizedr/ )r �msg)r
r? rA r r@ r r r rF � s zSOCKSv4.authorizec G s t jt|f|�� �||�S r )r Z
ClientCreatorr Z
connectTCP)r
r r �klass�argsr r r rG � s zSOCKSv4.connectClassc G s& t �|||� �}t�|�� dd � �S )Nr/ )r Z listenTCPr ZsucceedZgetHost)r
r rM rN Zservr r r rH � s zSOCKSv4.listenClassr �0.0.0.0c C s8 | j �t�d|||�t�|� � |dkr4| j �� d S )Nr- r )r r r7 Zpackr< Z inet_atonr )r
Zreplyr> 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 s
d S | j�� }| jj�� }t| j d�}|�dt�� |j|j || krLdpNd|j|j f � |r�|d d� |dd � }}|�t
� t dd� |�d�d � |�dt
|� d d � |D ]*}t
t|��d kr�|�d
� q�|�|� q�|�d � q^|�d � |�� d S ) N�az%s %s:%d %s %s:%d
�<�>� c S s dt | � S )Nz%02X)�ord)rD r r r r6 � r) z SOCKSv4.log.<locals>.<lambda>� � �.�
)r( r r r �openr �time�ctimer r �string�join�map�lenrK �close)r
�protor r Z
their_peer�f�p�cr r r r � s.
�
z SOCKSv4.log)r r rO )r# r$ r% �__doc__r r
r r! r: r rF rG rH r r r r r r r r' A s $(
r' c @ s e Zd ZdZdd� Zdd� ZdS )�SOCKSv4Factoryz`
A factory for a SOCKSv4 proxy.
Constructor accepts one argument, a log file name.
c C s
|| _ d S r )r( )r
r r r r r
� s zSOCKSv4Factory.__init__c C s t | jt�S r )r' r( r �r
Zaddrr r r �
buildProtocol� s z SOCKSv4Factory.buildProtocolN�r# r$ r% re r
rh r r r r rf � s rf c @ s e Zd ZdZdd� Zdd� ZdS )rI zJ
A utility class for building protocols for incoming connections.
c C s || _ || _d S r )r r )r
r r r r r r
� s z SOCKSv4IncomingFactory.__init__c C sV |d | j kr,d| _ | j�dd� t| j�S | j dkr:d S | j�dd� d| _ d S d S )Nr � r r2 )r r r r& rg r r r rh � s
z$SOCKSv4IncomingFactory.buildProtocolNri r r r r rI � s rI )re r7 r\ r< rZ Ztwisted.internetr r r Ztwisted.pythonr ZProtocolr r& r' ZFactoryrf rI r r r r �<module> s