File: //lib/python3.9/site-packages/fail2ban/__pycache__/helpers.cpython-39.opt-2.pyc
a
��}a[/&� @ s* d Z dZddlZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z ddl
Z
zddlZe�d�ZW n dZY n0 e�� Ze�d�r�ejr�ejjdur�ejj�d�s�ejjZned d
� d D ��r�d Zed
fdd�Zdd� Zdd� Zdd� Zdd� Zdd� Z G dd� de �Z G d d � d ej �Z d e_!dGd d �Z"ej#j$Z%d!d"� Z&e&ej#_$ej'j(Z)d#d$� Z*e*ej'_(d%d&� Z+d'd(� Z,dHd+d,�Z-d-d.� Z.d/d0� Z/d1d2� Z0d3d4� Z1e�2d5ej3�Z4e�2d6ej3�Z5e�2d7ej3�Z6d8d9� Z7d:d;� Z8d<Z9e�2d=�Z:dId@dA�Z;e�rdBdC� Z<ndDdC� Z<G dEdF� dFe �Z=dS )Jz<Cyril Jaquier, Arturo 'Buanzo' Busleiman, Yaroslav HalchenkoZGPL� N)�Lock� )�MyTimez libcap.so.2ZANSI_c c s | ]}t �|�d v V qdS ))N� N)�os�getenv��.0�v� r �4/usr/lib/python3.9/site-packages/fail2ban/helpers.py� <genexpr>0 � r
)ZLANGUAGE�LC_ALL�LC_CTYPEZLANG�UTF-8�strictc C sP z t | t�r| �||�W S | W S ttfyJ |dkr:� | �|d� Y S 0 d S )Nr �replace)�
isinstance�bytes�decode�UnicodeDecodeError�UnicodeEncodeError)�x�enc�errorsr r r �
uni_decode@ s
r c C s t | t�st| �S | �td�S )Nr )r r �strr �
PREFER_ENC�r r r r �
uni_stringI s
r c C s
t | d�S )Nr )r r r r r � uni_bytesM s r! c C s t | t�st| �S | �� dv S )N)�1�on�true�yes)r r �bool�lower)�valr r r �_as_boolP s
�r) c C s" t �� dd� \} }| jt|�fS )z+ Consistently format exception information N� )�sys�exc_info�__name__r )Zcla�excr r r �formatExceptionInfoU s r/ c C sR t j�| �}|�d�r"|dd� }|tddg�v rNt j�t j�| ��d | }|S )zaCustom function to include directory name if filename is too common
Also strip .py at the end
z.pyN����base�__init__�.)r �path�basename�endswith�set�dirname)�sr1 r r r � mbasenamea s
r: c @ s" e Zd ZdZddd�Zdd� ZdS ) � TraceBackz7Customized traceback to be included in debug messages
Fc C s d| _ || _dS )z�Initialize TrackBack metric
Parameters
----------
compress : bool
if True then prefix common with previous invocation gets
replaced with ...
r N)�_TraceBack__prev�_TraceBack__compress)�self�compressr r r r2 r s zTraceBack.__init__c C s� t jdd�d d� }dd� |D �}dd� |D �}|d g}|dd � D ]@}|d |d d kr||d d d
|d 7 < qF|�|� qFd �d d� |D ��}| jr�|}tj�| j|f�}t �
d
d|�}|dkr�d|t |�d � }|| _|S )N�d )�limit���c S s2 g | ]*}t |d �tj�|d �t|d �g�qS )r r )r: r r4 r8 r �r r r r r �
<listcomp>� s z&TraceBack.__call__.<locals>.<listcomp>c S s6 g | ].}|d dv s|d � d�s|d |d g�qS )r )Zunittestzlogging.__init__r z /unittestr* )r6 )r �er r r rD � s �r r ���z,%s�>c S s$ g | ] }d t |d �|d f �qS )z%s:%sr r )r: rC r r r rD � s � �z>[^>]*$r z...) � traceback�
extract_stack�append�joinr= r r4 � commonprefixr< �re�sub�len) r> Zftb�entriesZ entries_out�entryZsftbZ prev_nextZ
common_prefixZcommon_prefix2r r r �__call__~ s* �
�zTraceBack.__call__N)F)r- �
__module__� __qualname__�__doc__r2 rR r r r r r; n s
r; c @ s e Zd ZdZdd� Zdd� ZdS )�FormatterWithTraceBackz}Custom formatter which expands %(tb) and %(tbc) with tracebacks
TODO: might need locking in case of compressed tracebacks
c O s6 t jj| g|�R d|i|�� d|v }t|d�| _d S )N�fmtz%(tbc)s)r? )�logging� Formatterr2 r; �_tb)r> rW �args�kwargsr? r r r r2 � s z FormatterWithTraceBack.__init__c C s | � � |_|_tj�| |�S �N)rZ Ztbc�tbrX rY �format)r> �recordr r r r_ � s z FormatterWithTraceBack.formatN)r- rS rT rU r2 r_ r r r r rV � s rV Fc C sh | r t | j�r | �| jd � |r,dd� |_dd� tj_tjrdztj �� W n Y n0 t�
d� d S )Nr c S s d S r] r r r r r �<lambda>� r z!__stopOnIOError.<locals>.<lambda>c S s d S r] r �r> r r r ra � r ) rO �handlersZ
removeHandler�closerX �
StreamHandler�flush�
exitOnIOErrorr+ �stderr�exit)ZlogSysZlogHndlrr r r �__stopOnIOError� s
rj c
K s� zt | |||fi |�� W n� ttfyV } z |jdkr@t| � � W Y d}~n�d}~0 ty� } zlzRd|t|�ffddd� |D �fffD ](}zt | |g|�R � W q� Y q�0 q�W n Y n0 W Y d}~n
d}~0 0 dS )a� Safe log inject to avoid possible errors by unsafe log-handlers,
concat, str. conversion, representation fails, etc.
Used to intrude exception-safe _log-method instead of _log-method
of Logger class to be always safe by logging and to get more-info about.
See testSafeLogging test-case for more information. At least the errors
covered in phase 3 seems to affected in all known pypy/python versions
until now.
� Nzlogging failed: %r on %sz
args: %rc S s g | ] }t |��qS r )r )r �ar r r rD � r z __safeLog.<locals>.<listcomp>)� __origLog�BrokenPipeError�IOError�errnorj � Exceptionr )r> �level�msgr[ r\ rE r r r � __safeLog� s"
� rt c
C sP z t | � W n> ttfyJ } z"|jdkr4td| � � W Y d}~n
d}~0 0 dS )zQSafe flush inject stopping endless logging on closed streams (redirected pipe).
rk N)�__origLogFlushrn ro rp rj )r> rE r r r �__safeLogFlush� s
rv c C s$ d| v rd| � d�d } t�| �S )zBGet logging.Logger instance with Fail2Ban logger name convention
r3 z fail2ban.%srF )�
rpartitionrX � getLogger��namer r r rx � s rx c C sR z.t | t�s| �� r t| �}ntt| �� �}W n tyL td| ��Y n0 |S )NzInvalid log level %r)r �int�isdigit�getattrrX �upper�AttributeError�
ValueError)�valueZllr r r � str2LogLevel� s
r� � %(message)sTc C sn | dkr@| dkrd| }| dkr*d| }qTd| }|rTd| }nd| }|rTd | }|sjt �d
d d � |�}|S )
z(Custom log format for the verbose runs
r � z6 | %(module)15.15s-%(levelno)-2d: %(funcName)-20.20s |r* zB +%(relativeCreated)5d %(thread)X %(name)-25.25s %(levelname)-5.5sz %(thread)X %(levelname)-5.5sz %(asctime)-15sz)%(name)-24s[%(process)d]: %(levelname)-7sz %(asctime)s z(?<=\))-?\d+(?:\.\d+)?sc S s dS )Nr9 r )�mr r r ra
r z$getVerbosityFormat.<locals>.<lambda>)rM rN )� verbosityrW ZaddtimeZpaddingr r r �getVerbosityFormat� s
r� c C s t d�jddd� t�| ||�S )z>Except hook used to log unhandled exceptions to Fail2Ban log
Zfail2banz Unhandled exception in Fail2Ban:T)r, )rx Zcriticalr+ �__excepthook__)�exctyper� rH r r r �
excepthook s �r� c C s( | sg S t ttdd� t�d| �D ���S )z�Helper to split words on any comma, space, or a new line
Returns empty list if input is empty (or None) and filters
out empty entries
c S s g | ] }|� � �qS r )�stripr r r r rD r z splitwords.<locals>.<listcomp>z[\s,]+)�list�filterr& rM �split)r9 r r r �
splitwords s r� c C s |ri | �|�S | S )zHelper to merge dicts.
r �r �yr r r � _merge_dicts s r� c C s i | �|�S )zBHelper to merge dicts to guarantee a copy result (r is never x).
r r� r r r �_merge_copy_dicts& s r� z^([^\[]+)(?:\[(.*)\])?\s*$zf\s*([\w\-_\.]+)=(?:"([^"]*)"|\'([^\']*)\'|([^,\]]*))(?:,|\]\s*\[|$|(?P<wrngA>.+))|,?\s*$|(?P<wrngB>.+)zs(?:[^\[\s]+(?:\s*\[\s*(?:[\w\-_\.]+=(?:"[^"]*"|\'[^\']*\'|[^,\]]*)\s*(?:,|\]\s*\[)?\s*)*\])?\s*|\S+)(?=\n\s*|\s+|$)c C s� t �| �}|std��|�� \}}t� }|r�t�|�D ]�}|�d�rptd|�d�|�d�|�d�dd� f ��|�d�r�td|�d�|�d�dd� f ��|�d�}|s�q6d d
� |�d d d
�D �d }|� � ||� � < q6||fS )Nzunexpected option syntaxZwrngAz+unexpected syntax at %d after option %r: %sr r � ZwrngBz'expected option, wrong syntax at %d: %sc S s g | ]}|d ur|�qS r] r )r r( r r r rD K s z"extractOptions.<locals>.<listcomp>r* r� � )
�
OPTION_CRE�matchr� �groups�dict�OPTION_EXTRACT_CRE�finditer�group�startr� )�optionr� Z option_nameZoptstrZ option_optsZoptmatch�optr� r r r �extractOptions; s0
�
�
��r� c C s
t �| �S r] )�OPTION_SPLIT_CRE�findall)r� r r r �splitWithOptionsP s r� r� z <([^ <>]+)>r r c C s� t j}| }t|�}t� }t|d�}i }d} |�� D �]�}
|
|v s2|
|v rJq2|r^t|�|
��r^q2t||
� } } || �}
|�|
i �}|
�r�|
� d�}||v r�|| |
�
� �}
q�||
ks�|�|d�t kr�t d|
||| f ��d}|r�|�|d | �}|du �r |�|�}|du �r |du�r ||�}|du �r:|| |
�
� �}
q�t
|t��sNt|�}| �d| |�} |�|d�d ||< || |
�� �}
q�| | k�r�|| ��r�|||
< d } t|�t| �k�r�| �� }| ||
< d
| vr2|�|
� q2| s&�q�q&|S ) a| Sort out tag definitions within other tags.
Since v.0.9.2 supports embedded interpolation (see test cases for examples).
so: becomes:
a = 3 a = 3
b = <a>_3 b = 3_3
Parameters
----------
inptags : dict
Dictionary of tags(keys) and their values.
Returns
-------
dict
Dictionary of tags(keys) and their values, with tags
within the values recursively replaced.
�
getRawItemFr zpproperties contain self referencing definitions and cannot be resolved, fail tag: %s, found: %s in %s, value: %sN�?z<%s>r T�<)�TAG_CRE�searchr7 �hasattr�keys�callabler� r �getr� �end�MAX_TAG_REPLACE_COUNTr� r r r r� �id�copy�add)ZinptagsZ conditional�ignoreZaddreplZ
tre_searchZtagsZdoneZ noRecReplZ repCountsZrepFlag�tagr� Zorgvalr� ZrplcZrtag�replr r r �substituteRecursiveTags^ sb
��
r� c C s* z| � � } t�d| � W n Y n0 dS )z�Helper to set real thread name (used for identification and diagnostic purposes).
Side effect: name can be silently truncated to 15 bytes (16 bytes with NTS zero)
� N)�encode�_libcapZprctlry r r r �prctl_set_th_name� s
r� c C s d S r] r ry r r r r� � s c s< e Zd ZdZe� ZdZ� fdd�Zdd� Zd
dd �Z � Z
S ) � BgServicez{Background servicing
Prevents memory leak on some platforms/python versions,
using forced GC in periodical intervals.
Nc s | j stt| ��| �| _ | j S r] )� _instance�superr� �__new__)�cls�� __class__r r r� � s �zBgService.__new__c C s2 d| _ d| _d| _| j| _ttd�r.t�d� d S )Ni �� r@ �
set_thresholdr )�_BgService__serviceTime�_BgService__periodTime�_BgService__threshold�_BgService__countr� �gcr� rb r r r r2 � s
zBgService.__init__Fc C s� | j d8 _ |s.| j dks*t�� | jk r.dS tj�|�s>dS zNt�� | jk r^W tj�� dS t� � t�� | j
| _| j | _ W tj�� dS tj�� 0 dS )Nr r FT) r� r �timer� r� �_mutex�acquire�releaser� Zcollectr� r� )r> �force�waitr r r �service� s$
�
� zBgService.service)FF) r- rS rT rU r r� r� r� r2 r� �
__classcell__r r r� r r� � s r� )NN)r� TT)r r N)>�
__author__Z __license__r� ZlocalerX r rM r+ rH � threadingr Z
server.mytimer � importlibZctypesZCDLLr� �getpreferredencodingr �
startswith�stdout�encoding�allr r r! r) r/ r: �objectr; rY rV rg rj ZLogger�_logrm rt re rf ru rv rx r� r� r� r� r� r� �compile�DOTALLr� r� r� r� r� r� r� r� r� r� r r r r �<module> s~
.
��
�
c