CasperSecurity
o
;s*b� � @ s� d Z dZdZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z dd l
mZmZmZmZmZ dd
lmZ dd lmZ ddl m
Z
d d
� ZG dd� de�ZG dd� de
�ZG dd� de
�ZdS )zSteven Hiscocksz"Copyright (c) 2013 Steven Hiscocks�GPL� N� )�LogCaptureTestCase� )�protocol)�asyncore�RequestHandler�loop� AsyncServer�AsyncServerException)�Utils)�CSocketc G s t d��)N�test unpickle error�� Exception��args� r �?/usr/lib/python3/dist-packages/fail2ban/tests/sockettestcase.py� TestMsgError, s r c @ s$ e Zd Zedffdd�Zdd� ZdS )�TestMsgr c C s
|| _ d S �N��unpickle)�selfr r r r �__init__/ s
zTestMsg.__init__c C s | j S r r �r r r r �
__reduce__1 s zTestMsg.__reduce__N)�__name__�
__module__� __qualname__r r r r r r r r . s r c sz e Zd Z� fdd�Zdd� Zedd� �Zddd �Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Z� ZS )�Socketc sV t �| � tt| ��� t| �| _t�dd�\}}t� |� t�
|� || _ d| _ dS )z Call before every test case.z
fail2ban.sockz
f2b-socketN)
r �setUp�superr! r
�server�tempfile�mkstemp�os�close�remove� sock_name� serverThread)r �sock_fdr* �� __class__r r r" 7 s
z Socket.setUpc C s& | j r | j�� | �� t�| � dS )zCall after every test case.N)r+ r$ �stop�_stopServerThreadr �tearDownr r r r r1 B s
zSocket.tearDownc C s | S )z<Test transmitter proceed method which just returns first argr )�messager r r �proceedI s zSocket.proceedFc C sN t j| jj| j|fd� | _}d|_|�� | �t� | jj
t j �
d��� |S )N)�targetr T�
)� threading�Threadr$ �startr* r+ �daemon�
assertTruer �wait_for�isActive�unittest�F2B� maxWaitTime)r �forcer+ r r r �_createServerThreadN s � zSocket._createServerThreadc s, | j � t�� fdd�tj�d�� d | _ d S )Nc s � � � p
� �tj�S r )�is_alive�joinr �DEFAULT_SLEEP_TIMEr �r+ r r �<lambda>Z s
z*Socket._stopServerThread.<locals>.<lambda>r5 )r+ r r; r= r> r? r r rE r r0 W s
�
zSocket._stopServerThreadc C sZ | � � }| j�� | �� | �|�� � | j�� | �| j�� � | �tj �
| j �� d S r ) rA r$ r( r0 � assertFalserB r/ r< r' �path�existsr* �r r+ r r r �testStopPerCloseUnexpected^ s
z!Socket.testStopPerCloseUnexpectedc
C s4 zt | j�W S ty } zW Y d }~d S d }~ww r )r
r* r )r �er r r �
_serverSocketk s ��zSocket._serverSocketc C s� | � � }t�| jd�}g d�}| �|�|�|� | �|�t� gg�d� | jdddd� | �|�|�|� |�� |�� | j �
� t�| jd�}| �|�|�dg� | j � � | � � | �
|�� � | �
| j �� � | �
tj�| j�� d S ) Nr ��A�testr2 �ERROR: test unpickle errorz!PROTO-error: load message failed:r T��all�SHUTDOWN)rA r r; rM � assertEqual�sendr � assertLoggedr( r$ �stop_communicationr/ r0 rG rB r< r' rH rI r* )r r+ �client� testMessager r r �
testSocketq s"
zSocket.testSocketc sf | � � }t�| jd�� dddtjjgg�tj}zdd� t_| � t
d� �fdd�� W |t_d S |t_w ) Nr rO rP r2 c S s | � � S r )r( r r r r rF � s z0Socket.testSocketConnectBroken.<locals>.<lambda>zreset by peer|Broken pipec s � j �tj�d�d�S )Nr5 )�timeout)rV r= r> r? r �rY rZ r r rF � s ) rA r r; rM r �CSPROTO�ENDr �found_terminator�assertRaisesRegexr )r r+ � org_handlerr r] r �testSocketConnectBroken� s
�z Socket.testSocketConnectBrokenc C s� | � � }t�| jd�}g d�}| �|�|�|� tj}zdd� t_| �|�|�d� W |t_n|t_w | jdddd � | j �
� | � � | � |�
� � d S )
Nr rN c S s t � S r )r r r r r rF � s z.Socket.testStopByCommunicate.<locals>.<lambda>rQ z Unexpected communication errorr TrR )rA r r; rM rU rV r r` rW r$ r/ r0 rG rB )r r+ rY rZ rb r r r �testStopByCommunicate� s
z Socket.testStopByCommunicatec st t j�ddi� � �fdd�}z |t _| �� }| �t�� fdd�tj�d��� W �t _n�t _w | j dd d
d � d S ) N�cntrr c s, � d d7 < � d dk rt d���| � S )Nre r �2 ztest errors in poll)� RuntimeErrorr ��err�org_pollr r �_produce_error� s z-Socket.testLoopErrors.<locals>._produce_errorc s � d dkS )Nre rf r r )ri r r rF � s z'Socket.testLoopErrors.<locals>.<lambda>r5 z1Server connection was closed: test errors in poll�0Too many errors - stop logging connection errorsTrR )
r �pollrA r: r r; r= r> r? rW )r rk r+ r rh r �testLoopErrors� s $
�zSocket.testLoopErrorsc C sx t | jd��� | �t| jj| jd� | �d�}| j�� | � � | �
|� � � | �
| j� � � | �
t
j�| j�� d S )N�wFT)�openr* r( � assertRaisesr r$ r8 rA r/ r0 rG rB r< r' rH rI rJ r r r �testSocketForce� s �
zSocket.testSocketForce)F)r r r r" r1 � staticmethodr3 rA r0 rK rM r[ rc rd rn rr �
__classcell__r r r- r r! 5 s
!r! c @ s e Zd Zdd� Zdd� ZdS )�
ClientMiscc s^ ddi� � fdd�}� fdd�}t ||d� | jdd d
d d � | �d
� | jdddd d � d S )Nre r c s � d dk S )Nre �( r r ��phaser r �_active� s z,ClientMisc.testErrorsInLoop.<locals>._activec s � d d7 < t d� d ��)Nre r z test *%d*r r rw r r �_poll� s z*ClientMisc.testErrorsInLoop.<locals>._poll)�use_pollztest *1*z test *10*z test *20*TrR rl z test *21*z test *22*z test *23*)r rW �assertNotLogged)r ry rz r rw r �testErrorsInLoop� s
zClientMisc.testErrorsInLoopc C s: t j}ttjd�t _zt�� t�� W |t _d S |t _w )Nro )�sys�stdoutrp r' �devnullr �printFormatted� printWiki)r � saved_stdoutr r r �testPrintFormattedAndWiki� s
z$ClientMisc.testPrintFormattedAndWikiN)r r r r} r� r r r r ru � s
ru ) �
__author__�
__copyright__� __license__r' r~ r% r6 �timer= �utilsr � r �server.asyncserverr r r r
r � server.utilsr �client.csocketr
r �objectr r! ru r r r r �<module> s( ,