File: //lib/python3.6/site-packages/fail2ban/server/__pycache__/utils.cpython-36.pyc
3
=�Z]�- � @ s� d Z dZdZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z m
Z
ejdkrlddlZnddlZe e�Zdd iZed
d � ejj� 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� )� getLogger� _merge_dicts�
uni_decode� � 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>5 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 @ sD e Zd ZdZdd� Zddd�Zdd � Zdd d �Zd
d� Zdd� Z d
S )z Utils.Cachez.A simple cache with a TTL and limit on size
c O s | j ||� i | _d S )N)�
setOptions�_cache)�self�args�kwargsr r r �__init__F s zUtils.Cache.__init__�� �< c C s || _ || _d S )N)�maxCount�maxTime)r r r r r r r J s zUtils.Cache.setOptionsc C s
t | j�S )N)�lenr )r r r r �__len__N s zUtils.Cache.__len__Nc C s4 | j j|�}|r0|d tj� kr(|d S | j |= |S )N� r )r �get�time)r �kZdefv�vr r r r Q s zUtils.Cache.getc C st t j � }| j}t|�| jkr^x*t|j� �D ]\}}|d |k r*||= q*W t|�| jkr^|j� ||| j f||< d S )Nr )r r r r �list�items�popitemr )r r r �t�cacheZckZcvr r r �setY s
zUtils.Cache.setc C s& y | j |= W n tk
r Y nX d S )N)r �KeyError)r r r r r �unsetf s zUtils.Cache.unset)r r )N)
�__name__�
__module__� __qualname__�__doc__r r r r r& r( r r r r �CacheB 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 �
setFBlockModem 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 )�
isinstancer! r r" �append)�realCmd�varsDictZvarsStat�ir r r r r �
buildShellCmdw s
zUtils.buildShellCmdr TFr Nc 1 sz 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| |d � d} |�s|tj � k�r�� j �r�y0| dk�s4| d k �rBt 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 j%� � j&�r�y0| dk�s�| d k �r t 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$|�� �qjW � 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�| � |�rNtj+d| |t,� � |�r`|||| fS t-|�dk�rr|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)r8 � 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 %sr4 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 r/ �environ�id�
subprocess�Popen�PIPE�setsidrE �wait_for�DEFAULT_SHORTEST_INTERVAL�loggingZERRORr< �error�getpgid�pid�killpg�signal�SIGTERMr �sleep�DEFAULT_SLEEP_INTERVAL�SIGKILL�
pid_exists�OSError�DEBUGZgetEffectiveLevelrA r3 �read�IOError�
splitlinesr= r �closerB �debug�signamer �_RETCODE_HINTS�info�localsr )r8 �timeoutrC �outputZtout_kill_treeZ
success_codesr9 rA rB rF rD ZlogCmdrH Zpgid�eZ std_level�l�successZsigcode�msgr )rG r8 r? r �
executeCmd� s�
( "
"
zUtils.executeCmdc sz d}xp| � }|r|S |rPd }}t |�sBtj� | � � fdd�}n|}|sPtj}|� rXP t|| 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 r@ 3 s z Utils.wait_for.<locals>.<lambda>)�callabler r rZ �min�DEFAULT_SLEEP_TIMErY )Zcondrh ZintervalZini�retZstmZ timeout_exprr )ro r rP 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)�errnor/ �killr] ZEPERM)rU ru rj r r r r\ A 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)rU rw rx Z SYNCHRONIZEZprocessr r r r\ N s
c C sF t jjt jj| ��d }tjdkr6tjj|| �j � }n t
j || �}|S )Nr r )r r ) r/ �path�splitext�basename�sys� version_info� importlib� machinery�SourceFileLoader� load_module�impZ load_source)Z pythonModuleZpythonModuleName�modr r r �load_python_module[ s
zUtils.load_python_module�r )r TFTr� N)N)r) r* r+ r, rr rZ ZDEFAULT_SHORT_INTERVALrQ �objectr- � staticmethodr3 r; rn rP r/ r
r\ r� r r r r r 8 s$ +
+
r )r r )�
__author__Z
__copyright__Z __license__r. rR r/ rW rL r| r Zhelpersr r r r} Zimportlib.machineryr~ r� r) r< re �dict�__dict__r" rd r r r r r �<module> s$