Your IP : 216.73.216.61
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)�reactor�protocol�defer)�logc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �SOCKSv4Outgoingc C s
|| _ d S )N)�socks)�selfr � r �J/opt/alt/python37/lib64/python3.7/site-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 )N)r r �loseConnection)r �reasonr r r
�connectionLost s z SOCKSv4Outgoing.connectionLostc C s | j �|� d S )N)r �write)r �datar r r
� dataReceived# s z SOCKSv4Outgoing.dataReceivedc C s | j �| |� | j�|� d S )N)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 )N)r r )r r r r r
r . s zSOCKSv4Incoming.__init__c C s | j j�� d S )N)r r r )r r r r r
r 3 s z SOCKSv4Incoming.connectionLostc C s | j �|� d S )N)r r )r r r r r
r 7 s z SOCKSv4Incoming.dataReceivedc C s | j �| |� | j�|� d S )N)r r r 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 )N)�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� � z!BBH� � � s c S s
|� d�S )N�[ )r )�resultr r r r
�<lambda>z r" z&SOCKSv4.dataReceived.<locals>.<lambda>)
r r r# �struct�unpack�splitr �resolve� addCallback�_dataReceived2�
addErrback�socket� inet_ntoa)
r r ZcompleteBuffer�head�version�coder
�user�server�dr r r
r ^ s* $
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: %sr) Nr' c S s
|� d�S )Nr) )r )r* r r r r
r+ � r" z(SOCKSv4._dataReceived2.<locals>.<lambda>� r c S s |� dd| d | d �S )Nr r r' )r )�xr r r r
r+ � s zBad Connect Code: %sr" z hmm, still stuff in buffer... %s) �AssertionError� authorizer � connectClassr r2 � listenClass�SOCKSv4IncomingFactoryr0 � RuntimeErrorr# �repr)r r9 r8 r6 r7 r
r: r r r
r1 � s
zSOCKSv4._dataReceived2c C s | j r| j j�� d S )N)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 r7 r9 r
r8 r r r
r> � s zSOCKSv4.authorizec G s t jt|f|�� �||�S )N)r Z
ClientCreatorr Z
connectTCP)r r r
�klass�argsr r r
r? � s zSOCKSv4.connectClassc G s& t �|||� �}t�|�� dd � �S )Nr' )r Z listenTCPr ZsucceedZgetHost)r r
rE rF Zservr r r
r@ � s zSOCKSv4.listenClassr �0.0.0.0c C s8 | j �t�d|||�t�|� � |dkr4| j �� d S )Nz!BBHr )r r r, �packr3 � inet_atonr )r Zreplyr6 r
r r r r
r � s " zSOCKSv4.makeReplyc C s | � | |� | j�|� d S )N)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 � x�|r�|d d� |dd � }}|�t
� t dd� |�d�d � |�dt
|� d d � x2|D ]*}t
t|��d kr�|�d
� q�|�|� q�W |�d � q`W |�d � |�� d S ) N�az%s %s:%d %s %s:%d
�<�>� c S s dt | � S )Nz%02X)�ord)r< r r r
r+ � r" z SOCKSv4.log.<locals>.<lambda>� � �.�
)r! r r r �openr �time�ctimer r
�string�join�map�lenrC �close)r �protor r Z
their_peer�f�p�cr r r
r � s(
z SOCKSv4.log)r r rG )r r r �__doc__r r r r r1 r r> r? r@ 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 )N)r! )r r r r r
r � s zSOCKSv4Factory.__init__c C s t | jt�S )N)r r! r )r �addrr r r
�
buildProtocol� s z SOCKSv4Factory.buildProtocolN)r r r r_ r rb r r r r
r` � s r` c @ s e Zd ZdZdd� Zdd� ZdS )rA zJ
A utility class for building protocols for incoming connections.
c C s || _ || _d S )N)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 r) )r r r r )r ra r r r
rb � s
z$SOCKSv4IncomingFactory.buildProtocolN)r r r r_ r rb r r r r
rA � s rA )r_ r, rV r3 rT Ztwisted.internetr r r Ztwisted.pythonr ZProtocolr r r ZFactoryr` rA r r r r
�<module> s