File: //lib/python3.9/site-packages/fail2ban/tests/__pycache__/samplestestcase.cpython-39.opt-2.pyc
a
TPUG��8� @ s~ d Z dZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl m
Z
mZ ddlmZ dd lmZmZmZmZ ej�e��� Zej�ej�e �d
�Z ej�ej�e �d �Z e � d �Z e�!d
e�"e � d �Z#G dd� de j$�Z%dd� Z&edd� fe dd� ffD ]`\Z'Z(e)e(e�*ej�e'd���D ]:Z(e(�+d�d Z,e,�-d��s:e.e%de,�/� e&e,e'�� �q:�qdS )z"Copyright (c) 2013 Steven HiscocksZGPL� N� )�Regex)�Filter�
FileContainer)� FilterReader� )� setUpMyTime�tearDownMyTime�TEST_NOW�
CONFIG_DIRZconfig�files�<HOST>z\.[+\*](?!\?)[^\$\^]*z.*(?:\.[+\*].*|[^\$])$c sN e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zdd
d �Zed d
� �Z � Z
S )�FilterSamplesRegexc s& t t| ��� t� | _d| _t� dS )z Call before every test case.N)�superr �setUp�dict�_filters� _filterTestsr ��self�� __class__� �B/usr/lib/python3.9/site-packages/fail2ban/tests/samplestestcase.pyr 3 s zFilterSamplesRegex.setUpc s t t| ��� t� dS )zCall after every test case.N)r r �tearDownr r r r r r : s zFilterSamplesRegex.tearDownc C s( | � tdd� t�| �D ��dkd� dS )z Check to ensure some tests existc S s g | ]}|d � d�r|�qS )r ZtestSampleRegexs��
startswith)�.0�testr r r �
<listcomp>B s �z9FilterSamplesRegex.testFiltersPresent.<locals>.<listcomp>�
z&Expected more FilterSampleRegexs testsN)�
assertTrue�len�inspectZ
getmembersr r r r �testFiltersPresent? s ��z%FilterSamplesRegex.testFiltersPresentc C s| | � t�dt d �� | � t�dt d �� | �t�dt d �� | �t�dt d �� | �t�dt d �� d S )
z9Tests regexp RE_WRONG_GREED is intact (positive/negative)zgreedy .* testz test not hard-anchoredzgreedy .+ testz test vary .* anchored$z" test no catch-all, hard-anchored$znon-greedy .*? testznon-greedy .+? testz! test vary catch-all .* anchored$N)r! �RE_WRONG_GREED�search�RE_HOST� assertFalser r r r �testReWrongGreedyCatchAllG s �����z,FilterSamplesRegex.testReWrongGreedyCatchAllNc C s� | j �|�}|r|S td �}d|_d|_d|_d|_|d u rBt� }|�� }t |d||t
j j d�}| �
|�� |� | �
|�� d� |�� |�i � |�� D ]�}|d dkr�|d }n(|d dkr�|d g}n| �d |f � |D ]|}|d
d kr�||_q�|d
d k�r|�|� q�|d
d
k�r,|�|� q�|d
dk�rF|�|� q�|d
dkr�|�|� q�q�|�� } | D ].}
t�|
��rlt d|t |
�� t d�f ���ql|t � g}|| j |< |S )NTFZjail)�basedir� share_configr z multi-set� �setz$Unexpected config-token %r in streamr Z prefregexZ addfailregexZaddignoreregexZmaxlinesZ datepatternz�Following regexp of "%s" contains greedy catch-all before <HOST>, that is not hard-anchored at end or has not precise sub expression after <HOST>:
%sr
)!r �getr Z
returnRawHostZ
checkAllRegexZ
checkFindTimeZactiver �copyr �unittestZF2Br+ � assertEqualZgetFileZ getJailName�readZ
getOptions�convert�failZ prefRegexZ addFailRegexZaddIgnoreRegexZ setMaxLinesZsetDatePattern� getFailRegexr% r& �AssertionError�str�replacer' r- ) r �fltName�namer* �opts�fltZ
filterConf�optZoptval� regexList�frr r r � _readFilterU sV �
�
z FilterSamplesRegex._readFilterc C s t dd� | �� D ��S )Nc s s$ | ] \}}|� d �s||fV qdS )ztest.Nr )r �k�vr r r � <genexpr>� � z4FilterSamplesRegex._filterOptions.<locals>.<genexpr>)r �items�r; r r r �_filterOptions� s z!FilterSamplesRegex._filterOptions)N) �__name__�
__module__� __qualname__r r r$ r) r@ � staticmethodrG �
__classcell__r r r r r 1 s
5r c s � �fdd�}|S )Nc " s | � tj�tj�td���d� � �g}t� }i }i }d}|t|�k �r�|| }|d7 }ttj�td|�ddd�}d|_ d}d} |D �] }
| d7 } t
� d |
�} | �r�z&t �
| �d
��}| �d�d kr�|}W q�| �d�d k�r�g | _d}t|t�r�|n|gD ]�}|�r |�� } | �|� n|} | � t| t�� | �d
��rLt| �d
�� }|s�| �| �}
| �d�}|�s||
�rxt|
�nd}�| }| j|�� |
d�}| j�||| f� q�W q�| �d�dk�r�|�|� W q�W n< t�y
} z"td||� � | f ��W Y d }~n
d }~0 0 t |�}
n |s�|
� d�s�|
� � �s2q�ni }|�r>q�| j�sj�}| j|�� d d�}||i fg| _|
� d�}
| jD �]*\}}} |�d��r�t|d ��s��qz|\}}|�!� }d}�zfi }| �d�dk�r�|�"|
�}n,| �d��r�| �d�|
}
|�"dt#|
ft$�}|�rjg }|D ]N}|\}}}}|d u �s<|�d��rX|�%|� |�%|| � �q|�|� �q|}|�s�| �&|�dd�d � W �qz| � |�dd�d � | �'t|�dd d d � |D � � |D �]X}|\}}}}|�(� D ]�\}}|d!v�r�|�|d �}|d u �r,|d"k�r|}|d#k�r,t|�d$i ��}t|tttf��rN| �)||� �q�| �'||� �q�|�d%d �} | d u�rzt*j*�+| d&�} W n" t�y� t*j*�+| d'�} Y n0 t,�-| �.� �} | | j/d( 7 } | �'|| d)|t,�0d&t,�1|��| t,�0d&t,�1| ��|| f � |�%|� |�%|| � �q�W n� t2�y� } zhdd l3} t2d*|||� � | |
||dk�r`|| nd d+�| �4|��5� �d+�| �4|��5� �f ��W Y d }~n
d }~0 0 �qzq�q<| j6�(� D ]R\}}|\}}|�!� }t7|�D ].\} }!| � | |v �p�|!|v d,|| |!f � �qq�d S )-NZlogsz,No sample log file available for '%s' filterr r zUTF-8T)ZdoOpenFz4^#+ ?(failJSON|(?:file|filter)Options|addFILE):(.+)$r Z fileOptionsZ
filterOptionsztest.conditionztest.filter-name� rF ZaddFILEz %s: %s:%i�#z
�
constraint���ZlogtypeZjournalztest.prefix-lineZnofail�matchz!Line not matched when should havez Line matched when shouldn't havezMultiple regexs matched %rc S s g | ] }|d �qS )r r )r �xr r r r rD z?testSampleRegexsFactory.<locals>.testFilter.<locals>.<listcomp>)�timerQ ZdescrO �hostZattemptsZmatchesrS z%Y-%m-%dT%H:%M:%Sz%Y-%m-%dT%H:%M:%S.%fg ��.Az:UTC Time mismatch %s (%s) != %s (%s) (diff %.3f seconds)zC%s: %s on: %s:%i, line:
%s
regex (%s):
%s
faildata: %s
fail: %s�
z %s: Regex has no samples: %i: %r)8r! �os�path�isfile�join�TEST_FILES_DIRr- r" r ZwaitForLineEnd�rerQ �json�loads�groupr �
isinstance�listr/ �updater r. �evalrG r7 r@ �append�
ValueErrorZ getFileName�nextr �strip�rstripr5 Z processLine� TEST_NOW_STRr
�addr( r1 rE ZassertSortedEqual�datetime�strptimerS �mktimeZ timetupleZ microsecond�strftime�gmtimer6 �pprintZpformat�
splitlinesr � enumerate)"r � filenamesZ regexsUsedReZ
commonOptsZfaildata�i�filenameZlogFileZ ignoreBlockZlnnum�lineZ jsonREMatchr; ZfltOptsr9 r< �eZ
regexsUsedIdxr> Z failregexr4 Zret�foundZfidZ fail2banTimerA rB Zfv�tZ
jsonTimeLocalZjsonTimero ZfailRegexIndexZ failRegex�r* r: r r �
testFilter� s* � ��
�
�� �
���
��$��z+testSampleRegexsFactory.<locals>.testFilterr )r: r* rz r ry r �testSampleRegexsFactory� s /r{ c C s | � d� o| � d�S )Nz common.conf�.conf)�endswith�rR r r r �<lambda>@ rD r c C s | � d�o| �d�S )Nzzzz-r| )r r} r~ r r r r A rD zfilter.d�.ztestSampleRegexs%s)0Z
__copyright__Z __license__rj r# r\ rV r[ �sysrS r0 Zserver.failregexr Z
server.filterr r Zclient.filterreaderr Zutilsr r r
r ZutcfromtimestampZ isoformatrh rW rY �dirname�__file__ZTEST_CONFIG_DIRrZ Z_resolveHostTagr' �compile�escaper% ZTestCaser r{ Zbasedir_Zfilter_�filter�listdir�
rpartitionZ
filterNamer �setattr�upperr r r r �<module> sD
] 3
� �
�