File: //usr/lib/python3.9/site-packages/fail2ban/tests/__pycache__/samplestestcase.cpython-39.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 Hiscocks�GPL� N� )�Regex)�Filter�
FileContainer)� FilterReader� )� setUpMyTime�tearDownMyTime�TEST_NOW�
CONFIG_DIR�config�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�inspect�
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* r0 ) r �fltName�namer- �opts�fltZ
filterConf�opt�optval� 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, rD � staticmethodrK �
__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� rJ ZaddFILEz %s: %s:%i�#z
�
constraint����logtypeZjournalztest.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! rH z?testSampleRegexsFactory.<locals>.testFilter.<locals>.<listcomp>)�timerV �descrS �hostZattempts�matchesrX 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_DIRr0 r$ r ZwaitForLineEnd�rerV �json�loads�groupr �
isinstance�listr2 �updater r1 �evalrK r: rD �append�
ValueErrorZ getFileName�nextr �strip�rstripr8 Z processLine� TEST_NOW_STRr �addr+ r4 rI ZassertSortedEqual�datetime�strptimerX �mktime� timetuple� microsecond�strftime�gmtimer9 �pprint�pformat�
splitlinesr � enumerate)"r � filenamesZ regexsUsedReZ
commonOptsZfaildata�i�filenameZlogFileZ ignoreBlockZlnnum�lineZ jsonREMatchr> ZfltOptsr< r? �eZ
regexsUsedIdxrB Z failregexr7 �ret�found�fidZ fail2banTimerE rF Zfv�tZ
jsonTimeLocalZjsonTimerx ZfailRegexIndexZ failRegex�r- r= r r �
testFilter� s* � ��
�
�� �
���
��$��z+testSampleRegexsFactory.<locals>.testFilterr )r= r- r� r r� r �testSampleRegexsFactory� s /r� c C s | � d� o| � d�S )Nz common.conf�.conf)�endswith�rW r r r �<lambda>@ rH r� c C s | � d�o| �d�S )Nzzzz-r� )r r� r� r r r r� A rH zfilter.d�.ztestSampleRegexs%s)0�
__copyright__� __license__rq r% rc r] rb �sysrX r3 Zserver.failregexr Z
server.filterr r Zclient.filterreaderr �utilsr r
r r �utcfromtimestamp� isoformatro r^ r` �dirname�__file__ZTEST_CONFIG_DIRra Z_resolveHostTagr* �compile�escaper( ZTestCaser r� Zbasedir_Zfilter_�filter�listdir�
rpartitionZ
filterNamer �setattr�upperr r r r �<module> sD
] 3
� �
�