← Back
Editing: 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
Save File
Cancel