File: //usr/lib/python3.6/site-packages/fail2ban/server/__pycache__/utils.cpython-36.pyc
3
UB(d�. � @ s� d Z dZdZddlZddlZddlZddlZddlZddlZddl m
Z
ddl Z ddl m
Z
mZmZ ddlmZ ejdkr�ddlZnddlZe
e�Zd
d iZed d
� ejj� D ��Z G dd� d�Z dS )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$ | ] \}}|j d �r||fV qdS )ZSIGN)�
startswith)�.0�nameZnum� r
�/usr/lib/python3.6/utils.py� <genexpr>7 s r c @ s� e Zd ZdZdZdZdZed ZG dd� de�Z e
dd � �Z e
d
d � �Z e
d dd��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 ||� t� | _t� | _d S )N)�
setOptionsr �_cacher � _Cache__lock)�self�args�kwargsr
r
r �__init__H s zUtils.Cache.__init__�� �< c C s || _ || _d S )N)�maxCount�maxTime)r r r r
r
r r M s zUtils.Cache.setOptionsc C s
t | j�S )N)�lenr )r r
r
r �__len__Q s zUtils.Cache.__len__Nc C s6 | j j|�}|r2|d tj� kr(|d S | j|� |S )N� r )r �get�time�unset)r �kZdefv�vr
r
r r T s
zUtils.Cache.getc
C sz t j � }| j}| j�\ t|�| jkrZx4|rX|jdd�\}}|d |kr&t|�| jk r&P q&W ||| j f||< W d Q R X d S )NF)Zlastr )r! r r r r �popitemr )r r# r$ �t�cacheZckZcvr
r
r �set\ s zUtils.Cache.setc
C s$ | j � | jj|d � W d Q R X d S )N)r r �pop)r r# r
r
r r" l s zUtils.Cache.unsetc C s | j � | jj� W d Q R X d S )N)r r �clear)r r
r
r r* p s zUtils.Cache.clear)r r )N) �__name__�
__module__� __qualname__�__doc__r r r r r( r" r* r
r
r
r �CacheD s
r/ c C s> t j | t j�}|s |tjO }n |tj M }t j | t j|� |S )N)�fcntlZF_GETFL�os�
O_NONBLOCKZF_SETFL)Zfhandle�value�flagsr
r
r �
setFBlockModeu s zUtils.setFBlockModec C sp d}t | t�s| g} t| �d }x6|j� D ]*\}}|d||f 7 }| j|� |d7 }q*W |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 TFr Nc 1 s� d }}d} d � }
|r6|r*t j�|��n ttj|�}
t�����fdd�} �ytj�tjtj||
tj d�� � j
� } | dkr�� fdd�} t j | |t j �} | r�| d } | dk�rZ| r�| t
j� d} tjd�|f � tj� j�}
tj|
tj� tjt j� � j
� } | dk�s|�r<tj|
tj� tjt j� | dk�r<� j
� } | dk�rZt j|
� �rZtj} W nj tk
�r� } zL| �r�| t
j� d} d �|f }tj|� � �s�|�s�d
S d
||| fS W Y dd}~X nX | |k�r�t
j nt
j}|tj � k�r| �r| |t
j k�r|d nt
j� d} |�s(|tj � k�r�� j �r�y0| dk�sF| d k �rTt j � j d
� � j j � }W n0 t!k
�r� } ztjd |� W Y dd}~X nX |dk �r�|d
k�r�|tj � k�r�x&|j"� D ]}tj#|d�t$|�� �q�W � j%�r�y0| dk�s�| d k �rt j � j%d
� � j%j � }W n0 t!k
�rD } ztjd|� W Y dd}~X nX |dk �r�|d
k�r�|tj � k�r�x&|j"� D ]}tj#|d�t$|�� �qrW � j �r�� j j&� � j%�r�� j%j&� d
}| |k�r�tj'd�| � d}n�| dk�r�tjd�� j� n~| d k �s| dk�r<| d k �r| n| d }tjd�t(j)|d| �| � n4t*j)| d�}tjd�| � |�rptj+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 j| d�� �S )Nz%x -- exec: %s)�logSys�log)�level)r< � realCmdIdr
r �<lambda>� s z"Utils.executeCmd.<locals>.<lambda>)�stdout�stderr�shell�envZ
preexec_fnc s � j � } | d k rd| 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 %sr6 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 r1 �environ�id�
subprocess�Popen�PIPE�setsidrI �wait_for�DEFAULT_SHORTEST_INTERVAL�loggingZERRORr@ �error�getpgid�pid�killpg�signal�SIGTERMr! �sleep�DEFAULT_SLEEP_INTERVAL�SIGKILL�
pid_exists�OSError�DEBUGZgetEffectiveLevelrE r5 �read�IOError�
splitlinesrA r rF �close�debug�signamer �_RETCODE_HINTS�info�localsr )r< �timeoutrG �outputZtout_kill_treeZ
success_codesr= rE rF rJ rH ZlogCmdrL Zpgid�eZ std_level�l�successZsigcode�msgr
)rK r< rC r �
executeCmd� s�
( " "
zUtils.executeCmdc sv d}xl| � }|r|S |rFd }}t |�sBtj� | � � fdd�}n|}|� rNP t||pZtj tj�}tj|� qW |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 j � � kS )N)r! r
)�time0r
r rD < s z Utils.wait_for.<locals>.<lambda>)�callabler! �minr r^ �DEFAULT_SLEEP_TIMEr] )Zcondrl ZintervalZini�retZstmZ timeout_exprr
)rs r rT s zUtils.wait_for�posixc C sV ddl }| dk rdS ytj| d� W n( tk
rL } z |j |jkS d}~X nX dS dS )z6Check whether pid exists in the current process table.r NFT)�errnor1 �killra ZEPERM)rY ry rn r
r
r r` H s zUtils.pid_existsc C s@ dd l }|jj}d}|j|d| �}|dkr8|j|� dS dS d S )Nr i TF)�ctypesZwindll�kernel32Z OpenProcessZ CloseHandle)rY r{ r| Z SYNCHRONIZEZprocessr
r
r r` U s
c C sF t jjt jj| ��d }tjdkr6tjj|| �j � }n t
j || �}|S )Nr r )r r ) r1 �path�splitext�basename�sys� version_info� importlib� machinery�SourceFileLoader� load_module�impZ load_source)Z pythonModuleZpythonModuleName�modr
r
r �load_python_moduleb s
zUtils.load_python_module�r )r TFTr� N)N)r+ r, r- r. rv r^ ZDEFAULT_SHORT_INTERVALrU �objectr/ � staticmethodr5 r? rr rT r1 r r` r� r
r
r
r r : s$ 1
)
r )r r ) �
__author__Z
__copyright__Z __license__r0 rV r1 r[ rP r� Z threadingr r! Zhelpersr r r � collectionsr r� Zimportlib.machineryr� r� r+ r@ ri �dict�__dict__r: rh r r
r
r
r �<module> s(