File: //lib/python3/dist-packages/fail2ban/tests/__pycache__/samplestestcase.cpython-38.pyc
U
]^W/ � @ s� d Z dZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl m Z ddl
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>�q dS )z"Copyright (c) 2013 Steven HiscocksZGPL� N� )�Regex)�Filter)� 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__� �@/usr/lib/python3/dist-packages/fail2ban/tests/samplestestcase.pyr 4 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)�.0Ztestr r r �
<listcomp>C 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 �testReWrongGreedyCatchAllH s �����z,FilterSamplesRegex.testReWrongGreedyCatchAllNc C s� | j �|�}|r|S td �}d|_d|_d|_d|_|d krBt� }|�� }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
getOptionsZconvert�failZ prefRegexZ addFailRegexZaddIgnoreRegexZ setMaxLinesZsetDatePattern� getFailRegexr# r$ �AssertionError�str�replacer% r+ ) r �fltName�namer( �opts�fltZ
filterConfZoptZoptval� regexList�frr r r � _readFilterV sZ �
�
z FilterSamplesRegex._readFilterc C s t dd� t| �� �D ��S )Nc s s$ | ] \}}|� d �s||fV qdS )ztest.Nr )r �k�vr r r � <genexpr>� s
z4FilterSamplesRegex._filterOptions.<locals>.<genexpr>)r �list�items�r8 r r r �_filterOptions� s z!FilterSamplesRegex._filterOptions)N) �__name__�
__module__� __qualname__r r r" r' r< � staticmethodrC �
__classcell__r r r r r
2 s
5r
c s � �fdd�}|S )Nc ! s� | � tj�tj�td���d� � �g}t� }i }i }d}|t|�k �r�|| }|d7 }t� tj�td|��}d}|D �]} t
� d| �}
|
�r�z&t �
|
�d��}|
�d�dkr�|}W qx|
�d�d k�r�g | _d}t|t�r�|n|gD ]�}|�r|�� } | �|� n|} | � t| t�� | �d
��r6t| �d
�� }|s�| �| �} | �d �}
|
�sf| �rbt| �nd }
�|
}
| j|
�� | d
�}| j�|
|| f� q�W qx|
�d�dk�r�|�|� W qxW n> tk
�r� } z td||� � |� � f ��W 5 d }~X Y nX t |�} n |sx| � d�sx| � � �s qxni }|�r*qx| j�sV�}
| j|
�� d d
�}|
|i fg| _| jD �] \}
}} |\}}|�!� }d}�zxi }d}| �d�dk�r�|�"| �}n6d}| �d��r�| �d�| } |�"d t#| �$d�ft%�}|�s |�d��r�|�d�k�rW �q\| �&|�dd�d� W �q\|d \}}}}|�dd��sn|d k�sP|�d��rn|�'|� |�'|| � W �q\| � |�dd�d� | �(t|�dd d d � |D � � t|�)� �D ]�\}}|d k�r�|�|d �}|d k�r�|d k�r�|}|d!k�r�t|�d"i ��}t|tttf��r | �*||� �q�| �(||� �q�|�d#d �}|�sL|d k �r�zt+j+�,|d$�} W n$ tk
�r� t+j+�,|d%�} Y nX t-�.| �/� �} | | j0d& 7 } | �(|| d'|t-�1d$t-�2|��| t-�1d$t-�2| ��|| f � |�'|� |�'|| � W n� t3k
�rz } zddd l4} t3d(|
||� � |� � | ||dk�r<|| nd d)�| �5|��6� �d)�| �5|��6� �f ��W 5 d }~X Y nX �q\qxq<t| j7�)� �D ]R\}
}|\}}|�!� }t8|�D ].\} } | � | |k�p�| |kd*|
| | f � �q��q�d S )+NZlogsz,No sample log file available for '%s' filterr r Fz4^#+ ?(failJSON|(?:file|filter)Options|addFILE):(.+)$r Z fileOptionsZ
filterOptionsztest.conditionztest.filter-name� rB ZaddFILEz %s: %s:%i�#���TZlogtypeZjournalztest.prefix-linez
�filter�matchz!Line not matched when should haveZnofailz Line matched when shouldn't havezMultiple regexs matched %rc S s g | ] }|d �qS )r r )r �xr r r r � s z?testSampleRegexsFactory.<locals>.testFilter.<locals>.<listcomp>)�timerM ZdescrL ZhostZattemptsZmatchesrO 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)z@%s: %s on: %s:%i, line:
%sregex (%s):
%s
faildata: %s
fail: %s�
z %s: Regex has no samples: %i: %r)9r �os�path�isfile�join�TEST_FILES_DIRr+ r � fileinputZ FileInput�rerM �json�loads�groupr �
isinstancer@ r- �updater r, �evalrC r4 r< �append�
ValueError�filenameZ
filelineno�nextr �stripr2 Z processLine� TEST_NOW_STR�rstripr r& �addr/ rA ZassertSortedEqual�datetime�strptimerO �mktimeZ timetupleZ microsecond�strftime�gmtimer3 �pprintZpformat�
splitlinesr � enumerate)!r � filenamesZ regexsUsedReZ
commonOptsZfaildata�ir` ZlogFileZ ignoreBlock�lineZ jsonREMatchr8 ZfltOptsr6 r9 �eZ
regexsUsedIdxr: Z failregexr1 Z checktimeZretZfidZ fail2banTimer= r> Zfv�tZ
jsonTimeLocalZjsonTimerk ZfailRegexIndexZ failRegex�r( r7 r r �
testFilter� s& � �
�
�$
� �
���
�� ��z+testSampleRegexsFactory.<locals>.testFilterr )r7 r( rt r rs r �testSampleRegexsFactory� s %ru c C s | � d� o| � d�S )Nz common.conf�.conf)�endswith�rN r r r �<lambda>7 � ry c C s | � d�o| �d�S )Nzzzz-rv )r rw rx r r r ry 8 rz zfilter.d�.ztestSampleRegexs%s)0Z
__copyright__Z __license__rf rV r! rX rQ rW �sysrO r. Zserver.failregexr Z
server.filterr Zclient.filterreaderr Zutilsr r r r
ZutcfromtimestampZ isoformatrc rR rT �dirname�__file__ZTEST_CONFIG_DIRrU Z_resolveHostTagr% �compile�escaper# ZTestCaser
ru Zbasedir_Zfilter_rL �listdir�
rpartitionZ
filterNamer �setattr�upperr r r r �<module> sF
] )
� �
�