File: //usr/lib/python3/dist-packages/fail2ban/__pycache__/helpers.cpython-39.pyc
a
���`�E � @ s� d Z dZddlZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z ddl
Z
zddlZe�d�ZW n dZY n0 e�� at�d�r�ejr�ejjdur�ejj�d�s�ejjaned d
� d D ��r�d aejd
k r�dd� Zet� [ejd
k�rtdfdd�Zdd� Zn.tdfdd�Ze�� � � d k�rDdd� Zne Zdd� Z dd� Z dd � Z G d d � d e!�Z"G d d � d ej#�Z$d!e_%dMd"d#�Z&ze'Z'W n e(�y� e)Z'Y n0 ej*j+Z,d$d%� Z-e-ej*_+ej.j/Z0d&d'� Z1e1ej._/d(d)� Z2d*d+� Z3dNd.d/�Z4d0d1� Z5d2d3� Z6ejd4k�r,e7e8d5e9d6�� nd7d8� Z:d9d:� Z;e�8d;ej<�Z=e�8d<ej<�Z>e�8d=ej<�Z?d>d?� Z@d@dA� ZAdBZBe�8dC�ZCdOdFdG�ZDe�r�dHdI� ZEndJdI� ZEG dKdL� dLe!�ZFdS )Pz<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 �2/usr/lib/python3/dist-packages/fail2ban/helpers.py� <genexpr>0 � r
)ZLANGUAGE�LC_ALL�LC_CTYPEZLANGzUTF-8�� c C s� t �� �� }|dkr~|t�� kr~t }t|d�sjzddlm} |dd�}W n tyh t� t � t }Y n0 t|d�r~|�
| � d S )N�ASCII�setdefaultencodingr )� load_dynamic�_sys�sys) r �getdefaultencoding�upper�
PREFER_ENC�hasattr�impr � ImportError� importlib�reloadr )�encodingZoder Z__ldmr r r �__resetDefaultEncoding8 s
r! �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_decodeY s
r. c C s t | t�st| �S | �td�S �Nr$ )r% r& �strr' r �r+ r r r �
uni_stringb s
r2 c C sP z t | t�r| �||�W S | W S ttfyJ |dkr:� | �|d� Y S 0 d S r# )r% r0 �encoder( r) r* r r r r. g s
c C s t | t�st| �S | �td�S r/ )r% r0 r3 r r1 r r r r2 q s
c C s t | t�st| �S | �� dv S )N)�1Zon�trueZyes)r% r0 �bool�lower)�valr r r �_as_booly s
�r9 c C s" t �� dd� \} }| jt|�fS )z+ Consistently format exception information N� )r �exc_info�__name__r2 )Zcla�excr r r �formatExceptionInfo~ 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)�sr@ r r r � mbasename� s
rI 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 rA � 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 )rI r rC rG r0 �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: )rE )r �er r r rS � s �r r ���z,%s�>c S s$ g | ] }d t |d �|d f �qS )z%s:%sr r )rI rR r r r rS � s � �z>[^>]*$r z...) � traceback�
extract_stack�append�joinrL r rC � commonprefixrK �re�sub�len) rM Zftb�entriesZ entries_out�entryZsftbZ prev_nextZ
common_prefixZcommon_prefix2r r r �__call__� s* �
�zTraceBack.__call__N)F)r< �
__module__� __qualname__�__doc__rA ra r r r r rJ � s
rJ 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)rN )�logging� FormatterrA rJ �_tb)rM rf �args�kwargsrN r r r rA � s z FormatterWithTraceBack.__init__c C s | � � |_|_tj�| |�S �N)ri Ztbc�tbrg rh �format)rM �recordr r r rn � s z FormatterWithTraceBack.formatN)r< rb rc rd rA rn r r r r re � s re 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 rl r r r r r �<lambda>� r z!__stopOnIOError.<locals>.<lambda>c S s d S rl r �rM r r r rp � r ) r^ �handlersZ
removeHandler�closerg �
StreamHandler�flush�
exitOnIOErrorr �stderr�exit)ZlogSysZlogHndlrr r r �__stopOnIOError� s
ry 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 )r2 )r �ar r r rS � r z __safeLog.<locals>.<listcomp>)� __origLog�BrokenPipeError�IOError�errnory � Exceptionr2 )rM �level�msgrj rk rT r r r � __safeLog� s"
� r� 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).
rz N)�__origLogFlushr} r~ r ry )rM rT 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
rB z fail2ban.%srU )�
rpartitionrg � 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�getattrrg r �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 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 )NrH r )�mr r r rp 8 r z$getVerbosityFormat.<locals>.<lambda>)r\ r] )� verbosityrf 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; )r� Zcriticalr �__excepthook__)�exctyper� rW 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 rS K r z splitwords.<locals>.<listcomp>z[ ,
]+)�list�filterr6 r\ �split)rH r r r �
splitwordsC s r� )r � z�if 1:
def _merge_dicts(x, y):
"""Helper to merge dicts.
"""
if y:
return {**x, **y}
return x
def _merge_copy_dicts(x, y):
"""Helper to merge dicts to guarantee a copy result (r is never x).
"""
return {**x, **y}
�execc C s | }|r| � � }|�|� |S )zHelper to merge dicts.
��copy�update�r+ �y�rr r r � _merge_dicts\ s
r� c C s | � � }|r|�|� |S )zCHelper to merge dicts to guarantee a copy result (r is never x).
r� r� r r r �_merge_copy_dictsd s
r� z^([^\[]+)(?:\[(.*)\])?\s*$z@([\w\-_\.]+)=(?:"([^"]*)"|\'([^\']*)\'|([^,\]]*))(?:,|\]\s*\[|$)zs(?:[^\[\s]+(?:\s*\[\s*(?:[\w\-_\.]+=(?:"[^"]*"|\'[^\']*\'|[^,\]]*)\s*(?:,|\]\s*\[)?\s*)*\])?\s*|\S+)(?=\n\s*|\s+|$)c C sv t �| �}|sdS |�� \}}t� }|rnt�|�D ]:}|�d�}dd� |�ddd�D �d }|�� ||�� < q2||fS ) N)NNr c S s g | ]}|d ur|�qS rl r )r r8 r r r rS � s z"extractOptions.<locals>.<listcomp>r: r � r )�
OPTION_CRE�match�groups�dict�OPTION_EXTRACT_CRE�finditer�groupr� )�optionr� Z option_nameZoptstrZ option_optsZoptmatchZoptr� r r r �extractOptions| s
��r� c C s
t �| �S rl )�OPTION_SPLIT_CRE�findall)r� r r r �splitWithOptions� s 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�searchrF r �keys�callabler� r2 �getr� �end�MAX_TAG_REPLACE_COUNTr� r% r0 r$ �start�idr� �add)ZinptagsZ conditional�ignoreZaddreplZ
tre_searchZtagsZdoneZ noRecReplZ repCountsZrepFlag�tagr� Zorgvalr� ZrplcZrtag�replr r r �substituteRecursiveTags� sb
��
r� c C s> z,t jdkr| �� } nt| �} 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)
r � N)r � version_infor3 r& �_libcapZprctlr� r r r �prctl_set_th_name� s
r� c C s d S rl 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 rl )� _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 �� rO �
set_thresholdr )�_BgService__serviceTime�_BgService__periodTime�_BgService__threshold�_BgService__countr �gcr� rq r r r rA 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� )rM Zforce�waitr r r �service) s$
�
� zBgService.service)FF) r< rb rc rd r r� r� r� rA r� �
__classcell__r r r� r r� s r� )NN)r� TT)r r N)G�
__author__Z __license__r� Zlocalerg r r\ r rW Z threadingr Z
server.mytimer r ZctypesZCDLLr� Zgetpreferredencodingr �
startswith�stdoutr �allr� r! r. r2 r r r0 r9 r> rI �objectrJ rh re rv ry r} � NameErrorr~ ZLoggerZ_logr| r� rt ru r� r� r� r� r� r� r� �eval�compile�__file__r� r� �DOTALLr� r� r� r� r� r� r� r� r� r� r r r r �<module> s�
.
�
��
�
c