Your IP : 216.73.216.213
B
;��]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 �J/opt/alt/python37/lib64/python3.7/site-packages/twisted/test/test_socks.py�getPeer s zStringTCPTransport.getPeerc C s t �ddd�S )N�TCPz2.3.4.5�* )r � IPv4Address)r r r r
�getHost s zStringTCPTransport.getHostc C s
d| _ d S )NT)�stringTCPTransport_closing)r 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> yt �| j| �S tk
r8 t �td|�d� ��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 �hostnamer 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 �host�port�klass�args�protor r r
� connectClassE s zSOCKSv4Driver.connectClassc G s( ||� }|| _ |dkrd}t�d|f�S )Nr i� z6.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 localhostz 127.0.0.1)r! �sockr r" r# r �reactor)r 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.)r/ 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 )
Nz!BBH� � �" z1.2.3.4s fooBAR� r �Z s hello, worlds hi therez fake reason)r/ � dataReceived�struct�pack�socket� inet_atonr" �value�clear� assertEqual� assertFalser �assertIsNotNoner$ �connectionLost)r �sentr r r
� test_simplek s $ zConnectTests.test_simplec C s� t �dddd�t�d� d d }x t|�D ]}| j�|� q,W | jj�� }| jj� � | �
|t �ddd d�t�d
� � | � | jjj � | �
| jj� | j�d � | �
| jjj�� d � | jj�d � | �
| jj�� d � | j�d
� dS )au
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}
z!BBHr4 r5 r6 z0.0.0.1s fooBAZ s
localhost r r8 z 127.0.0.1s hello, worlds hi therez fake reasonN)r: r; r< r= r r/ r9 r" r> r? r@ rA r rB r$ rC )r �
clientRequest�byterD r r r
� test_socks4aSuccessfulResolution� s " z-ConnectTests.test_socks4aSuccessfulResolutionc C s� t �dddd�t�d� d d }x t|�D ]}| j�|� q,W | jj�� }| � |t �ddd d�t�d
� � | �
| jjj � | � | jj
� d S ) z�
Failed hostname resolution on a SOCKSv4a packet results in a 91 error
response and the connection getting closed.
z!BBHr4 r5 r6 z0.0.0.1s fooBAZ s failinghost r �[ z0.0.0.0N)r: r; r< r= r r/ r9 r" r> r@ r2 r � assertIsNoner$ )r rF rG rD 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 )�code�serverr& �userr r r
�<lambda>� � z0ConnectTests.test_accessDenied.<locals>.<lambda>z!BBHr4 r5 i� z10.2.3.4s fooBARr7 r rI z0.0.0.0)r/ � authorizer9 r: r; r< r= r@ r" r> r2 r rJ 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 )
Nz!BBHr4 r5 r6 z1.2.3.4s fooBARr7 s hello, worldz fake reason)
r/ r9 r: r; r< r= r" r? r@ r$ r> r rC )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 )
Nz!BBHr4 r5 r6 z1.2.3.4s fooBARr7 s hello, worldz fake reason) r/ r9 r: r; r< r= r" r? r@ r$ r> rC )r r r r
�
test_eofLocal� s $ zConnectTests.test_eofLocalN) r r r r r1 r3 rE rH rK rR rS rT 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 )Ns localhostz 127.0.0.1)r! r/ r r" r# r r0 )r r r r
r1 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 )Nz!BBHr4 � r6 z1.2.3.4s fooBARr7 r r8 i� z6.7.8.9)z1.2.3.4i� z0.0.0.0s hello, worlds hi therez fake reason)r/ r9 r: r; r< r= r" r> r? r@ rA r rB r+ �
buildProtocolr r# rC )r rD �incomingr r r
rE s2 $
zBindTests.test_simplec C sX t �dddd�t�d� d d }x t|�D ]}| j�|� q,W | 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 )au
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}
z!BBHr4 rV r6 z0.0.0.1s fooBAZ s
localhost r r8 i� z6.7.8.9)z 127.0.0.1i� z0.0.0.0Ns hi therez fake reason)r: r; r< r= r r/ r9 r" r> r? r@ rA r rB r+ rW r r# Z assertIsNotrC )r rF rG rD rX r r r
� test_socks4a< s4 "
zBindTests.test_socks4ac C s� t �dddd�t�d� d d }x t|�D ]}| j�|� q,W | jj�� }| � |t �ddd d�t�d
� � | �
| jjj � | � | jj
� d S ) z�
Failed hostname resolution on a SOCKSv4a packet results in a 91 error
response and the connection getting closed.
z!BBHr4 rV r6 z0.0.0.1s fooBAZ s failinghost r rI z0.0.0.0N)r: r; r< r= r r/ r9 r" r> r@ r2 r rJ r$ )r rF rG rD r r r
rK 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 )Nr r )rL rM r& rN r r r
rO � rP z-BindTests.test_accessDenied.<locals>.<lambda>z!BBHr4 rV i� z10.2.3.4s fooBARr7 r rI z0.0.0.0)r/ rQ r9 r: r; r< r= r@ r" r> r2 r rJ r+ )r r r r
rR � 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 )Nz!BBHr4 rV r6 z1.2.3.4s fooBARr7 )z1.2.3.4i� r r8 z0.0.0.0s hello, worldz fake reason)r/ r9 r: r; r< r= r" r> r? r+ rW rB r r# r@ rA r r rC )r rD rX r r r
rS � 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 )Nz!BBHr4 rV r6 z1.2.3.4s fooBARr7 )z1.2.3.4i� r r8 z0.0.0.0s hello, worldz fake reason)r/ r9 r: r; r< r= r" r> r? r+ rW rB r r# r@ rA r rC )r rD rX r r r
rT � s$ $
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 ) Nz!BBHr4 rV r6 z1.2.3.4s fooBARr7 )z1.6.6.6i� r rI z0.0.0.0)r/ r9 r: r; r< r= r" r> r? r+ rW rJ r@ r2 r )r rD rX r r r
�test_badSource� s $
zBindTests.test_badSourceN) r r r r r1 rE rY rK rR rS rT rZ r r r r
rU s
(< ! rU )r r: r< Ztwisted.internetr r Ztwisted.internet.errorr Ztwisted.python.compatr Ztwisted.protocolsr Z twisted.testr Z
twisted.trialr ZStringTransportr r ZSOCKSv4r! ZTestCaser. rU r r r r
�<module> s +