File: //usr/lib/python3.9/site-packages/fail2ban/__pycache__/helpers.cpython-39.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 Halchenko�GPL� 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 )�LANGUAGE�LC_ALL�LC_CTYPE�LANG�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 r2 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)�sr4 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
r5 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 r7 r; 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- )r9 )r
�er r r
rG � s �r r ���z,%s�>c S s$ g | ] }d t |d �|d f �qS )z%s:%sr r )r= rF r r r
rG � s � �z>[^>]*$r z...) � traceback�
extract_stack�append�joinr@ r r7 � commonprefixr? �re�sub�len) rA Zftb�entriesZ entries_out�entryZsftbZ prev_nextZ
common_prefixZcommon_prefix2r r r
�__call__~ s* �
�zTraceBack.__call__N)F)r0 �
__module__� __qualname__�__doc__r5 rU 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)rB )�logging� Formatterr5 r> �_tb)rA rZ �args�kwargsrB r r r
r5 � s z FormatterWithTraceBack.__init__c C s | � � |_|_tj�| |�S �N)r] Ztbc�tbr[ r\ �format)rA �recordr r r
rb � s z FormatterWithTraceBack.formatN)r0 rV rW rX r5 rb r r r r
rY � s rY 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 �rA r r r
rd � r ) rR �handlers�
removeHandler�closer[ �
StreamHandler�flush�
exitOnIOErrorr. �stderr�exit)�logSysZlogHndlrr r r
�__stopOnIOError� s
ro 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
rG � r z __safeLog.<locals>.<listcomp>)� __origLog�BrokenPipeError�IOError�errnoro � Exceptionr# )rA �level�msgr^ r_ rH r r r
� __safeLog� s"
� ry 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).
rp N)�__origLogFlushrs rt ru ro )rA rH r r r
�__safeLogFlush� s
r{ c C s$ d| v rd| � d�d } t�| �S )zBGet logging.Logger instance with Fail2Ban logger name convention
r6 z fail2ban.%srI )�
rpartitionr[ � getLogger��namer r r
r} � s r} c C sR z.t | t�s| �� r t| �}ntt| �� �}W n tyL td| ��Y n0 |S )NzInvalid log level %r)r �int�isdigit�getattrr[ �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 )Nr< r )�mr r r
rd
r z$getVerbosityFormat.<locals>.<lambda>)rP rQ )� verbosityrZ Zaddtime�paddingr r r
�getVerbosityFormat� s
r� c C s t d�jddd� t�| ||�S )z>Except hook used to log unhandled exceptions to Fail2Ban log
�fail2banz Unhandled exception in Fail2Ban:T)r/ )r} �criticalr. �__excepthook__)�exctyper� rK 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
rG r z splitwords.<locals>.<listcomp>z[\s,]+)�list�filterr) rP �split)r< 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
rG K s z"extractOptions.<locals>.<listcomp>r- r� � )
�
OPTION_CRE�matchr� �groups�dict�OPTION_EXTRACT_CRE�finditer�group�startr� )�optionr� � 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�searchr: �hasattr�keys�callabler� r# �getr� �end�MAX_TAG_REPLACE_COUNTr� r r r r� �id�copy�add)ZinptagsZ conditional�ignoreZaddreplZ
tre_searchZtags�doneZ 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�_libcapZprctlr~ r r r
�prctl_set_th_name� s
r� c C s d S r` r r~ 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 �� rC �
set_thresholdr )�_BgService__serviceTime�_BgService__periodTime�_BgService__threshold�_BgService__countr� �gcr� re r r r
r5 � 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� )rA �force�waitr r r
�service� s$
�
� zBgService.service)FF) r0 rV rW rX r r� r� r� r5 r� �
__classcell__r r r� r
r� � s r� )NN)r� TT)r r N)>�
__author__� __license__r� �localer[ r rP r. rK � threadingr Z
server.mytimer � importlib�ctypes�CDLLr� �getpreferredencodingr! �
startswith�stdout�encoding�allr r# r$ r, r2 r= �objectr> r\ rY rk ro �Logger�_logrr ry ri rj rz r{ r} r� r� r� r� r� r� �compile�DOTALLr� r� r� r� r� r� r� r� r� r� r r r r
�<module> s~
.
��
�
c