File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_socks.cpython-38.pyc
U
W[JE � @ 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)�defer�address)�DNSLookupError)� iterbytes)�socks)�
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 sD zt �| j| �W S tk
r> t �td|�d� �� Y S X dS )zQ
Resolve a hostname by looking it up in the C{names} dictionary.
z"FakeResolverReactor couldn't find zutf-8N)r �succeedr �KeyErrorZfailr �decode)r Zhostnamer
r
r �resolve1 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 Zhost�port�klass�args�protor
r
r � connectClassE s zSOCKSv4Driver.connectClassc G s( ||� }|| _ |dkrd}t�d|f�S )Nr �� �6.7.8.9)�
driver_listenr r )r r% r& r' �factoryr
r
r � listenClassO s
zSOCKSv4Driver.listenClass)r r r r$ r, r) r. r
r
r
r r! > s
r! c @ sP 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 s0 t � | _t� | j_| j�� tddi�| j_d S �Ns localhost� 127.0.0.1�r! �sockr r" r# r Zreactorr r
r
r �setUp] s
zConnectTests.setUpc C s$ | j j}|d k r | �|jjd� d S )Nz-Outgoing SOCKS connections need to be closed.)r3 r$ �
assertTruer" r )r Zoutgoingr
r
r �tearDownd 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)r3 � dataReceived�struct�pack�socket� inet_atonr" �value�clear� assertEqual� assertFalser �assertIsNotNoner$ �connectionLost)r �sentr
r
r � test_simplek s2 ���� �� �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}
r7 r8 r9 r: �0.0.0.1� fooBAZ �
localhost r r> r1 r? r@ rA N)rC rD rE rF r r3 rB r" rG rH rI rJ r rK r$ rL �r �
clientRequest�byterM r
r
r � test_socks4aSuccessfulResolution� s4 ���� � �z-ConnectTests.test_socks4aSuccessfulResolutionc C s� 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.
r7 r8 r9 r: rP rQ � failinghost r �[ �0.0.0.0N�rC rD rE rF r r3 rB r" rG rI r5 r � assertIsNoner$ rS r
r
r � test_socks4aFailedResolution� s" ���� �z)ConnectTests.test_socks4aFailedResolutionc C s� 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 s dS �Nr r
��codeZserverr% �userr
r
r �<lambda>� � z0ConnectTests.test_accessDenied.<locals>.<lambda>r7 r8 r9 � �10.2.3.4r< r= r rY rZ )r3 � authorizerB rC rD rE rF rI r" rG r5 r r\ r$ r r
r
r �test_accessDenied� s" ������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 �
Nr7 r8 r9 r: r; r< r= r? rA )
r3 rB rC rD rE rF r" rH rI r$ rG r rL 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 rh ) r3 rB rC rD rE rF r" rH rI r$ rG rL r r
r
r �
test_eofLocal� s ���� �zConnectTests.test_eofLocalN) r r r r r4 r6 rN rV r] rg ri rj r
r
r
r r/ Y s / r/ c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )� BindTestszS
Tests for SOCKS and SOCKSv4a bind requests using the L{SOCKSv4} protocol.
c C s0 t � | _t� | j_| j�� tddi�| j_d S r0 r2 r r
r
r r4 s
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 )Nr7 r8 � r: r; r< r= r r> r* r+ �r; �� rZ r? r@ rA )r3 rB rC rD rE rF r" rG rH rI rJ r rK r, �
buildProtocolr r# rL �r rM �incomingr
r
r rN sJ ���� ��
�� �
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 )rO r7 r8 rl r: rP rQ rR r r> r* r+ )r1 rn rZ Nr@ rA )rC rD rE rF r r3 rB r" rG rH rI rJ r rK r, ro r r# Z assertIsNotrL )r rT rU rM rq r
r
r � test_socks4a< sL ���� �
�� �
zBindTests.test_socks4ac C s� t �dddd�t�d� d d }t|�D ]}| j�|� q*| jj�� }| � |t �ddd d�t�d
� � | �
| jjj � | � | jj
� d S ) rW r7 r8 rl r: rP rQ rX r rY rZ Nr[ rS r
r
r r] x s" ���� �z&BindTests.test_socks4aFailedResolutionc C s� 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 s dS r^ r
r_ r
r
r rb � rc z-BindTests.test_accessDenied.<locals>.<lambda>r7 r8 rl rd re r< r= r rY rZ )r3 rf rB rC rD rE rF rI r" rG r5 r r\ r, r r
r
r rg � s" ������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 �Nr7 r8 rl r: r; r< r= rm r r> rZ r? rA )r3 rB rC rD rE rF r" rG rH r, ro rK r r# rI rJ r r rL rp r
r
r ri � s: ����
�� �
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 rs )r3 rB rC rD rE rF r" rG rH r, ro rK r r# rI rJ r rL rp r
r
r rj � s8 ����
�� �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 ) Nr7 r8 rl r: r; r< r= )z1.6.6.6i� r rY rZ )r3 rB rC rD rE rF r" rG rH r, ro r\ rI r5 r rp r
r
r �test_badSource� s* ����
��zBindTests.test_badSourceN) r r r r r4 rN rr r] rg ri rj rt r
r
r
r rk s
(< ! rk )r rC rE Ztwisted.internetr r Ztwisted.internet.errorr Ztwisted.python.compatr Ztwisted.protocolsr Z twisted.testr Z
twisted.trialr ZStringTransportr r ZSOCKSv4r! ZTestCaser/ rk r
r
r
r �<module> s +