|
Server : LiteSpeed System : Linux host 5.15.0-161-generic #171-Ubuntu SMP Sat Oct 11 08:17:01 UTC 2025 x86_64 User : idnco5810 ( 1093) PHP Version : 8.2.29 Disable Function : NONE Directory : /lib/python3/dist-packages/twisted/test/__pycache__/ |
Upload File : |
o
�bXD � @ s� d Z ddlZddlZddlmZmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ G d d
� d
e
j�ZG d d � d �ZG d
d� de j�ZG dd� dej�ZG dd� dej�ZdS )z_
Tests for L{twisted.protocol.socks}, an implementation of the SOCKSv4 and
SOCKSv4a protocols.
� N)�address�defer)�DNSLookupError)�socks)� iterbytes)�
proto_helpers)�unittestc @ s, e Zd ZdZdZdd� Zdd� Zdd� ZdS ) �StringTCPTransportFNc C s | j S )N)�peer��self� r
�9/usr/lib/python3/dist-packages/twisted/test/test_socks.py�getPeer s zStringTCPTransport.getPeerc C s t �ddd�S )N�TCPz2.3.4.5�* )r � IPv4Addressr r
r
r �getHost s zStringTCPTransport.getHostc C s
d| _ d S )NT)�stringTCPTransport_closingr r
r
r �loseConnection s
z!StringTCPTransport.loseConnection)�__name__�
__module__� __qualname__r r
r r r r
r
r
r r s r c @ s e Zd ZdZdd� Zdd� ZdS )�FakeResolverReactorzP
Bare-bones reactor with deterministic behavior for the resolve method.
c C s
|| _ dS )z�
@type names: L{dict} containing L{str} keys and L{str} values.
@param names: A hostname to IP address mapping. The IP addresses are
stringified dotted quads.
N)�names)r r r
r
r �__init__' s
z FakeResolverReactor.__init__c
C s> z t �| j| �W S ty t �td|�d� �� Y S w )zQ
Resolve a hostname by looking it up in the C{names} dictionary.
z"FakeResolverReactor couldn't find zutf-8)r �succeedr �KeyError�failr �decode)r �hostnamer
r
r �resolve/ s ���zFakeResolverReactor.resolveN)r r r �__doc__r r! r
r
r
r r " s r c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�
SOCKSv4DriverNc G s: ||� }t � |_t�d||�|j_|�� || _t�|�S )Nr ) r � transportr r r
�connectionMade�driver_outgoingr r )r �host�port�klass�args�protor
r
r � connectClassD s
zSOCKSv4Driver.connectClassc G s( ||� }|| _ |dkr
d}t�d|f�S )Nr �� �6.7.8.9)�
driver_listenr r )r r( r) r* �factoryr
r
r � listenClassM s
zSOCKSv4Driver.listenClass)r r r r& r/ r, r1 r
r
r
r r# = s
r# c @ �P e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )� ConnectTestszV
Tests for SOCKS and SOCKSv4a connect requests using the L{SOCKSv4} protocol.
c C �0 t � | _t� | j_| j�� tddi�| j_d S �Ns localhost� 127.0.0.1�r# �sockr r$ r% r �reactorr r
r
r �setUp[ �
zConnectTests.setUpc C s( | j j}|d ur| �|jjd� d S d S )Nz-Outgoing SOCKS connections need to be closed.)r8 r& �
assertTruer$ r )r �outgoingr
r
r �tearDowna s ��zConnectTests.tearDownc C s� | j �t�dddd�t�d� d d � | j j�� }| j j�� | � |t�ddd d�t�d� � | �
| j jj � | � | j j
� | j �d
� | � | j j
j�� d
� | j j
�d � | � | j j�� d � | j �d � d S )
N�!BBH� � �" �1.2.3.4� fooBAR� r �Z � hello, world� hi there� fake reason)r8 � dataReceived�struct�pack�socket� inet_atonr$ �value�clear� assertEqual� assertFalser �assertIsNotNoner& �connectionLost)r �sentr
r
r � test_simplei s* ���� � zConnectTests.test_simplec C s� t �dddd�t�d� d d }t|�D ]}| j�|� q| jj�� }| jj� � | �
|t �ddd d�t�d
� � | � | jjj � | �
| jj� | j�d � | �
| jjj�� d � | jj�d � | �
| jj�� d � | j�d
� dS )�u
If the destination IP address has zeros for the first three octets and
non-zero for the fourth octet, the client is attempting a v4a
connection. A hostname is specified after the user ID string and the
server connects to the address that hostname resolves to.
@see: U{http://en.wikipedia.org/wiki/SOCKS#SOCKS_4a_protocol}
r? r@ rA rB �0.0.0.1� fooBAZ �
localhost r rF r6 rG rH rI N)rK rL rM rN r r8 rJ r$ rO rP rQ rR r rS r& rT �r �
clientRequest�byterU r
r
r � test_socks4aSuccessfulResolution� s, ����
� z-ConnectTests.test_socks4aSuccessfulResolutionc C � t �dddd�t�d� d d }t|�D ]}| j�|� q| jj�� }| � |t �ddd d�t�d
� � | �
| jjj � | � | jj
� d S ) ��
Failed hostname resolution on a SOCKSv4a packet results in a 91 error
response and the connection getting closed.
r? r@ rA rB rX rY � failinghost r �[ �0.0.0.0N�rK rL rM rN r r8 rJ r$ rO rQ r< r � assertIsNoner& r[ r
r
r � test_socks4aFailedResolution� � ����
�z)ConnectTests.test_socks4aFailedResolutionc C � dd� | j _| j �t�dddd�t�d� d d � | �| j j� � t�dd
d d
�t�d � � | �
| j jj � | � | j j
� d S )
Nc S � dS �Nr r
��code�serverr( �userr
r
r �<lambda>� � z0ConnectTests.test_accessDenied.<locals>.<lambda>r? r@ rA � �10.2.3.4rD rE r rb rc )r8 � authorizerJ rK rL rM rN rQ r$ rO r< r re r& r r
r
r �test_accessDenied� � ����
�z ConnectTests.test_accessDeniedc C sx | j �t�dddd�t�d� d d � | j j�� | j �d� | �| j j j�
� d� | j j j� � | j j � d � d S �
Nr? r@ rA rB rC rD rE rG rI )
r8 rJ rK rL rM rN r$ rP rQ r& rO r rT r r
r
r �test_eofRemote� s ���� zConnectTests.test_eofRemotec C sh | j �t�dddd�t�d� d d � | j j�� | j �d� | �| j j j�
� d� | j � d � d S rv ) r8 rJ rK rL rM rN r$ rP rQ r& rO rT r r
r
r �
test_eofLocal� s ���� zConnectTests.test_eofLocalN) r r r r" r: r> rV r^ rf rt rw rx r
r
r
r r3 V s . r3 c @ r2 )� BindTestszS
Tests for SOCKS and SOCKSv4a bind requests using the L{SOCKSv4} protocol.
c C r4 r5 r7 r r
r
r r: r; zBindTests.setUpc C s@ | j �t�dddd�t�d� d d � | j j�� }| j j�� | � |t�ddd d
�t�d � � | �
| j jj � | � | j j
� | j j
�d �}| � |� t� |_|�� | j j�� }| j j�� | � |t�ddd d�t�d
� � | �
| j jj � | j �d� | � |j�� d� |�d� | � | j j�� d� | j �d� d S )Nr? r@ � rB rC rD rE r rF r- r. �rC �� rc rG rH rI )r8 rJ rK rL rM rN r$ rO rP rQ rR r rS r/ �
buildProtocolr r% rT �r rU �incomingr
r
r rV s> ���� �
�
zBindTests.test_simplec C sT t �dddd�t�d� d d }t|�D ]}| j�|� q| jj�� }| jj� � | �
|t �ddd d
�t�d � � | � | jjj � | �
| jj� | jj�d �}| �
|� t� |_|�� | jj�� }| jj� � | �
|t �ddd d�t�d
� � | �| jjj d� | j�d� | �
|j�� d� |�d� | �
| jj�� d� | j�d� dS )rW r? r@ rz rB rX rY rZ r rF r- r. )r6 r| rc NrH rI )rK rL rM rN r r8 rJ r$ rO rP rQ rR r rS r/ r} r r% � assertIsNotrT )r r\ r] rU r r
r
r � test_socks4a7 s@ ����
�
�
zBindTests.test_socks4ac C r_ ) r` r? r@ rz rB rX rY ra r rb rc Nrd r[ r
r
r rf r rg z&BindTests.test_socks4aFailedResolutionc C rh )
Nc S ri rj r
rk r
r
r ro � rp z-BindTests.test_accessDenied.<locals>.<lambda>r? r@ rz rq rr rD rE r rb rc )r8 rs rJ rK rL rM rN rQ r$ rO r< r re r/ r r
r
r rt � ru zBindTests.test_accessDeniedc C s� | j �t�dddd�t�d� d d � | j j�� }| j j�� | j j �
d�}| � |� t � |_|�
� | j j�� }| j j�� | �|t�dd d
d �t�d � � | �| j jj� | j �d � | �|j�� d � |j�� |�d
� d S �Nr? r@ rz rB rC rD rE r{ r rF rc rG rI )r8 rJ rK rL rM rN r$ rO rP r/ r} rS r r% rQ rR r r rT r~ r
r
r rw � s2 ����
�
zBindTests.test_eofRemotec C s� | j �t�dddd�t�d� d d � | j j�� }| j j�� | j j �
d�}| � |� t � |_|�
� | j j�� }| j j�� | �|t�dd d
d �t�d � � | �| j jj� | j �d � | �|j�� d � | j �d
� d S r� )r8 rJ rK rL rM rN r$ rO rP r/ r} rS r r% rQ rR r rT r~ r
r
r rx � s0 ����
� zBindTests.test_eofLocalc C s� | j �t�dddd�t�d� d d � | j j�� }| j j�� | j j �
d�}| � |� | j j�� }| j j�� | � |t�dd d
d �t�d � � | �
| j jj� d S ) Nr? r@ rz rB rC rD rE )z1.6.6.6i� r rb rc )r8 rJ rK rL rM rN r$ rO rP r/ r} re rQ r< r r~ r
r
r �test_badSource� s& ����
�zBindTests.test_badSourceN) r r r r" r: rV r� rf rt rw rx r� r
r
r
r ry � s
'; ry )r" rM rK �twisted.internetr r �twisted.internet.errorr �twisted.protocolsr �twisted.python.compatr � twisted.testr �
twisted.trialr �StringTransportr r �SOCKSv4r# �TestCaser3 ry r
r
r
r �<module> s )