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