File: //lib/python3.9/site-packages/fail2ban/server/__pycache__/utils.cpython-39.opt-2.pyc
a
늠���� @ s� d Z dZdZddlZddlZddlZddlZddlZddlZddl m
Z
ddl Z ddl m
Z
mZmZ ddlmZ ddlZe
e�Zd d
iZed d � ej�� D ��ZG d
d� d�ZdS )z2Serg G. Brester (sebres) and Fail2Ban ContributorszYCopyright (c) 2004 Cyril Jaquier, 2011-2012 Yaroslav Halchenko, 2012-2015 Serg G. BresterZGPL� N)�Lock� )� getLogger� _merge_dicts�
uni_decode)� OrderedDict� am "Command not found". Make sure that all commands in %(realCmd)r are in the PATH of fail2ban-server process (grep -a PATH= /proc/`pidof -x fail2ban-server`/environ). You may want to start "fail2ban-server -f" separately, initiate it with "fail2ban-client reload" in another shell session and observe if additional informative error messages appear in the terminals.c c s$ | ] \}}|� d �r||fV qdS )ZSIGN)�
startswith)�.0�nameZnum� r �9/usr/lib/python3.9/site-packages/fail2ban/server/utils.py� <genexpr>4 s �r c @ s� e Zd ZdZdZdZdZed ZG dd� de�Z e
dd � �Z e
d
d � �Z e
ddd��Z
e
d dd��Zejdkr|e
dd� �Zn e
dd� �Ze
dd� �ZdS ) �UtilszPUtilities provide diverse static methods like executes OS shell commands, etc.
r g�������?g����MbP?�d c @ sL e Zd ZdZdd� Zddd�Zdd � Zdd d �Zd
d� Zdd� Z dd� Z
d
S )z Utils.Cachez.A simple cache with a TTL and limit on size
c O s$ | j |i |�� t� | _t� | _d S �N)�
setOptionsr �_cacher � _Cache__lock)�self�args�kwargsr r r
�__init__E s zUtils.Cache.__init__�� �< c C s || _ || _d S r )�maxCount�maxTime)r r r r r r
r J s zUtils.Cache.setOptionsc C s
t | j�S r )�lenr �r r r r
�__len__N s zUtils.Cache.__len__Nc C s6 | j �|�}|r2|d t�� kr(|d S | �|� |S )N� r )r �get�time�unset)r �kZdefv�vr r r
r! Q s
zUtils.Cache.getc C s� t � � }| j}| j�b t|�| jkrV|rV|jdd�\}}|d |kr$t|�| jk r$qVq$||| j f||< W d � n1 s|0 Y d S )NF)Zlastr )r" r r r r �popitemr )r r$ r% �t�cacheZckZcvr r r
�setY s zUtils.Cache.setc C s8 | j � | j�|d � W d � n1 s*0 Y d S r )r r �pop)r r$ r r r
r# i s zUtils.Cache.unsetc C s4 | j � | j�� W d � n1 s&0 Y d S r )r r �clearr r r r
r+ m s zUtils.Cache.clear)r r )N) �__name__�
__module__� __qualname__�__doc__r r r r! r) r# r+ r r r r
�CacheA s
r0 c C s> t � | t j�}|s |tjO }n |tj M }t � | t j|� |S r )�fcntlZF_GETFL�os�
O_NONBLOCKZF_SETFL)Zfhandle�value�flagsr r r
�
setFBlockModer s zUtils.setFBlockModec C sl d}t | t�s| g} t| �d }|�� D ]*\}}|d||f 7 }| �|� |d7 }q(|d | d | d< | S )a� Generates new shell command as array, contains map as variables to
arguments statement (varsStat), the command (realCmd) used this variables and
the list of the arguments, mapped from varsDict
Example:
buildShellCmd('echo "V2: $v2, V1: $v1"', {"v1": "val 1", "v2": "val 2", "vUnused": "unused var"})
returns:
['v1=$0 v2=$1 vUnused=$2
echo "V2: $v2, V1: $v1"', 'val 1', 'val 2', 'unused var']
� r z%s=$%s �
r )�
isinstance�listr �items�append)�realCmd�varsDictZvarsStat�ir$ r% r r r
�
buildShellCmd| s
zUtils.buildShellCmdr TF�r Nc
s� d }}d} d � }
|r6|r*t ��|��n ttj|�}
t�����fdd�} �z tj�tjtj||
tj d�� � �
� } | du r�� fdd�} t � | |t j �} | r�| d } | du �rX| r�| t
j� d} t�d�|f � t�� j�}
t�|
tj� t�t j� � �
� } | du �s|�r<t�|
tj� t�t j� | du �r<� �
� } | du �rXt �|
��sXtj} W nx t�y� } z^| �r�| t
j� d} d �|f }t�|� � �s�|�s�d
n
d
||| fW Y d}~S W Y d}~n
d}~0 0 | |v �r�t
j nt
j}|t� � k�r | �r | |t
j k�r|d nt
j� d} |�s4|t� � k�r�� j �r�z0| du �sR| d k �r`t � � j d
� � j � � }W n2 t!�y� } zt�d |� W Y d}~n
d}~0 0 |du�r�|d
k�r�|t� � k�r�|�"� D ]}t�#|d�t$|�� �qʈ j%�r�z0| du �s| d k �rt � � j%d
� � j%� � }W n2 t!�yP } zt�d|� W Y d}~n
d}~0 0 |du�r�|d
k�r�|t� � k�r�|�"� D ]}t�#|d�t$|�� �q|� j �r�� j �&� � j%�r�� j%�&� d
}| |v �r�t�'d�| � d}n�| du �r�t�d�� j� n~| d k �s| dk�rD| d k �r | n| d }t�d�t(�)|d| �| � n4t*�)| d�}t�d�| � |�rxt�+d| |t,� � |�r�|||| fS t-|�dk�r�|S || fS )a Executes a command.
Parameters
----------
realCmd : str
The command to execute.
timeout : int
The time out in seconds for the command.
shell : bool
If shell is True (default), the specified command (may be a string) will be
executed through the shell.
output : bool
If output is True, the function returns tuple (success, stdoutdata, stderrdata, returncode).
If False, just indication of success is returned
varsDict: dict
variables supplied to the command (or to the shell script)
Returns
-------
bool or (bool, str, str, int)
True if the command succeeded and with stdout, stderr, returncode if output was set to True
Raises
------
OSError
If command fails to be executed.
RuntimeError
If command execution times out.
Nc s t �| d�� �S )Nz%x -- exec: %s)�logSys�log)�level)r= � realCmdIdr r
�<lambda>� � z"Utils.executeCmd.<locals>.<lambda>)�stdout�stderr�shell�env�
preexec_fnc s � � � } | d urd| fS d S )NT)�poll)�retcode)�popenr r
�_popen_wait_end� s z)Utils.executeCmd.<locals>._popen_wait_endr z!%x -- timed out after %s seconds.z%s -- failed with %sFr z ... -- failed to read stdout %sr7 z%x -- stdout: %rz ... -- failed to read stderr %sz%x -- stderr: %rz %x -- returned successfully %iTz%x -- unable to kill PID %i� z&%x -- killed with %s (return code: %s)z signal %iz%x -- returned %izHINT on %i: %s).r r@ r r2 �environ�id�
subprocess�Popen�PIPE�setsidrM �wait_for�DEFAULT_SHORTEST_INTERVAL�logging�ERRORrB �error�getpgid�pid�killpg�signal�SIGTERMr" �sleep�DEFAULT_SLEEP_INTERVAL�SIGKILL�
pid_exists�OSError�DEBUGZgetEffectiveLevelrH r6 �read�IOError�
splitlinesrC r rI �close�debug�signamer! �_RETCODE_HINTS�info�localsr )r= �timeoutrJ �outputZtout_kill_treeZ
success_codesr> rH rI rN rK ZlogCmdrP Zpgid�eZ std_level�lZsuccessZsigcode�msgr )rO r= rE r
�
executeCmd� s� �
�
8("" ""
� zUtils.executeCmdc sr d}| � }|r|S |rDd }}t |�s@t�� | � � fdd�}n|}|� rLqnt||pXtj tj�}t�|� q|S )a5 Wait until condition expression `cond` is True, up to `timeout` sec
Parameters
----------
cond : callable
The expression to check condition
(should return equivalent to bool True if wait successful).
timeout : float or callable
The time out for end of wait
(in seconds or callable that returns True if timeout occurred).
interval : float (optional)
Polling start interval for wait cycle in seconds.
Returns
-------
variable
The return value of the last call of `cond`,
logical False (or None, 0, etc) if timeout occurred.
r r c s t � � � kS r )r" r �Ztime0r r
rF 9 rG z Utils.wait_for.<locals>.<lambda>)�callabler" �minr rc �DEFAULT_SLEEP_TIMErb )Zcondrq �intervalZiniZretZstmZ timeout_exprr rw r
rX s zUtils.wait_for�posixc
C s^ ddl }| dk rdS zt�| d� W n0 tyT } z|j |jkW Y d}~S d}~0 0 dS dS )z6Check whether pid exists in the current process table.r NFT)�errnor2 �killrf ZEPERM)r^ r} rs r r r
re E s "zUtils.pid_existsc C s@ dd l }|jj}d}|�|d| �}|dkr8|�|� dS dS d S )Nr i TF)�ctypesZwindll�kernel32Z OpenProcessZ CloseHandle)r^ r r� Z SYNCHRONIZE�processr r r
re R s
c C s. t j�t j�| ��d }tj�|| ��� }|S )Nr )r2 �path�splitext�basename� importlib� machinery�SourceFileLoader� load_module)Z pythonModuleZpythonModuleName�modr r r
�load_python_module_ s
���zUtils.load_python_module)r TFTrA N)N)r, r- r. r/ rz rc ZDEFAULT_SHORT_INTERVALrY �objectr0 � staticmethodr6 r@ rv rX r2 r re r� r r r r
r 7 s0 1
� )
r ) �
__author__Z
__copyright__Z __license__r1 rZ r2 r` rT �sys� threadingr r" Zhelpersr r r � collectionsr Zimportlib.machineryr� r, rB rn �dict�__dict__r; rm r r r r r
�<module> s( � �