|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/fail2ban/tests/__pycache__/ |
Upload File : |
o
� �_:& � @ sd d Z dZdZddlZddlmZmZ ddlmZ dd l m
Z
G d
d � d ej �Z G d d
� d
ej �Z
dS )z
Cyril Jaquierz Copyright (c) 2004 Cyril Jaquier�GPL� N� )� setUpMyTime�tearDownMyTime� )�
BanManager)� BanTicketc sp e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
� ZS )�
AddFailurec s, t t| ��� t� tdd�| _t� | _dS )� Call before every test case.�
193.168.0.128� �;f�AN)�superr �setUpr r �_AddFailure__ticketr �_AddFailure__banManager��self�� __class__� �C/usr/lib/python3/dist-packages/fail2ban/tests/banmanagertestcase.pyr # s zAddFailure.setUpc � t t| ��� t� dS �zCall after every test case.N)r
r �tearDownr r r r r r * �
zAddFailure.tearDownc C sZ | � | j�| j�� | �| j�� d� | �| j�� d� | j�d� | �| j�� d� d S )Nr r )�
assertTruer � addBanTicketr � assertEqual�size� getBanTotal� setBanTotalr r r r �testAdd/ s
zAddFailure.testAddc C s> | � | j�| j�� | �| j�| j�� | �| j�� d� d S )Nr )r r r r � assertFalser r r r r r �testAddDuplicate6 s zAddFailure.testAddDuplicatec C s� | j �� }d}dD ]j\}}tdd�}td|�}|d ur |�|� | �| j �|�� | �| j �|�� | �| j �� d� | j � |�
� �}| �|� |�|� |�� | �|� |�|k� |� |�}| �|� � d� |dkrs| �|�|�d� q d S )Nr ))r N)g �Tf�AN)r i�Q )r ���r r r r$ )
r �
getBanTimer �
setBanTimer r r" r r �
getTicketByID�getID�getEndOfBanTime�getTime)r �
defBanTime�prevEndOfBanTime�tnew�btnew�ticket1�ticket2� banticketr r r �testAddDuplicateWithTime; s&
��z#AddFailure.testAddDuplicateWithTimec C s4 | � | j�| j�� tdd�}| � | j�|�� d S )Nr r )r r r r r �
_inBanList�r �ticketr r r � testInListOK[ �
zAddFailure.testInListOKc C s4 | � | j�| j�� tdd�}| �| j�|�� d S )Nz
111.111.1.111r )r r r r r r" r3 r4 r r r �
testInListNOK` r7 zAddFailure.testInListNOKc C s t | j�� | j�� �}d}dD ]3}| j�| j� |d7 }|�|� | �| j�|�� | �t | j�
|�� ��d|�� |�� ||f � q| j�| j� |d7 }|�d� | �| j�|�� |�d� | �| j�|�� | �t | j�
|�� ��d|�� |�� d|f � d S )Nr )�� i� r$ r zFBanTicket: ip=%s time=%s bantime=%s bancount=%s #attempts=0 matches=[]r$ r9 ) r r �getIPr* r r r& r" r �strr' )r r5 �c�ir r r �testBanTimeIncre s"
�
�zAddFailure.testBanTimeIncrc C s� | j �� }| j�� }| �| j �| j�� | �| j �| j�� | �| j �|�g � | �| j �|| d �| jg� | �| j � � d� | �| j �| j�� t
| j� � |d �}| � | j �|�� | �t
| j �|| d ��d� | �t
| j �|| d d ��d� td�D ] }t
d| |�}|�|�|�|d � | �| j �|�� q�| �t
| j �|| d d ��d� | �t
| j �|| d d ��d � | �| j � � d� d S )
Nr r iX � z 193.168.0.%s�
r �2 � )r r% r r* r r r3 r � unBanListr r r( r" �len�ranger& )r �btime�stimer5 r= r r r � testUnbanx s&
$ $$zAddFailure.testUnbanc C s� | j �� }| j �d� z7| �| j �| j�� | �| j �| j�� | �| j �| j� � | d �g � | �| j �
� d� W | j �|� d S | j �|� w )Nr$ r ) r r% r& r r r r3 r rC r* r )r rF r r r �testUnbanPermanent� s
" z AddFailure.testUnbanPermanentc C sX t dd�t dd�g}|d �d� |D ]}| j�|� q| �| jjddd�dd g� d S )
Nz 192.0.2.1r z 192.0.2.2r r$ T)�ordered�withTimez:192.0.2.1 2006-12-31 23:59:59 + 600 = 2007-01-01 00:09:59z9192.0.2.2 2006-12-31 23:59:59 + -1 = 9999-12-31 23:59:59)r r& r r �assertSortedEqual�
getBanList)r �tickets�tr r r � testBanList� s ���zAddFailure.testBanList)�__name__�
__module__� __qualname__r r r! r# r2 r6 r8 r> rH rI rP �
__classcell__r r r r r " s r c s\ e Zd Z� fdd�Z� fdd�ZdZdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z � Z S )�StatusExtendedCymruInfoc s` t t| ��� tj�� t� d| _d| _d| _ d| _
t | jd�}t � | _
| �| j
�|�� dS )r
z
93.184.216.34�15133�EU�ripenccr N)r
rU r �unittest�F2B�SkipIfNoNetworkr � _StatusExtendedCymruInfo__ban_ip� _StatusExtendedCymruInfo__asn�!_StatusExtendedCymruInfo__country� _StatusExtendedCymruInfo__rirr r �$_StatusExtendedCymruInfo__banManagerr r r4 r r r r � s
z StatusExtendedCymruInfo.setUpc r r )r
rU r r r r r r r � r z StatusExtendedCymruInfo.tearDown)TNc C s^ t }|jd r| jjtjjrdndd�}n|jd }|�d�r-d|f|_t�d|d ��|S ) Nr r � )�timeoutr �errorFz.Skip test because service is not available: %s) rU � availabler` �getBanListExtendedCymruInforY rZ �fast�get�SkipTest)r �tc�
cymru_infor r r � _getBanListExtendedCymruInfo� s
�
z4StatusExtendedCymruInfo._getBanListExtendedCymruInfoc C s, | � � }| �|| jg| jg| jgd�� d S )N��asn�country�rir)rk �assertDictEqualr] r^ r_ )r rj r r r �
testCymruInfo� s ��z%StatusExtendedCymruInfo.testCymruInfoc C � | � | j�| �� �| jg� d S �N)r r` �geBanListExtendedASNrk r] r r r r �testCymruInfoASN� � �z(StatusExtendedCymruInfo.testCymruInfoASNc C rr rs )r r` �geBanListExtendedCountryrk r^ r r r r �testCymruInfoCountry� rv z,StatusExtendedCymruInfo.testCymruInfoCountryc C rr rs )r r` �geBanListExtendedRIRrk r_ r r r r �testCymruInfoRIR� rv z(StatusExtendedCymruInfo.testCymruInfoRIRc C s� t � | _tdd�}| �| j�|�� | �� }| �|dgdgdgd�� tdd�}| �| j�|�� | �� }| j|ddgddgdd gd�d
td � d S ) Nz0.0.0.0r �nxdomainrl z8.0.0.0g <f�A�3356�US�arinr$ )�level�key) r r` r r r rk rp rL r; )r r5 rj r r r �testCymruInfoNxdomain� s(
��
�
�z-StatusExtendedCymruInfo.testCymruInfoNxdomain)
rQ rR rS r r rd rk rq ru rx rz r� rT r r r r rU � s
rU )�
__author__�
__copyright__� __license__rY �utilsr r �server.banmanagerr �
server.ticketr �TestCaser rU r r r r �<module> s