a
=*f=. @ s d Z dZdZddlZddlZddlZddlZddlZddlZddl m
Z
ddl Z ddl m
Z
mZmZ ddlmZ ddlZe
eZd d
iZed d ej D ZG d
d dZdS )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).0nameZnum r 9/usr/lib/python3.9/site-packages/fail2ban/server/utils.py 4 s r c @ s e Zd ZdZdZdZdZed ZG dd deZ e
dd Z e
d
d Z e
dddZ
e
d ddZejdkr|e
dd Zn e
dd Ze
dd ZdS ) UtilszPUtilities provide diverse static methods like executes OS shell commands, etc.
r g?gMbP?d c @ sL e Zd ZdZdd ZdddZdd 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)selfargskwargsr r r
__init__E s zUtils.Cache.__init__ < c C s || _ || _d S r )maxCountmaxTime)r r r r r r
r J s zUtils.Cache.setOptionsc C s
t | jS 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 gettimeunset)r kZdefvvr r r
r! Q s
zUtils.Cache.getc C s t }| j}| jb t|| jkrV|rV|jdd\}}|d |kr$t|| jk r$qVq$||| j f||< W d n1 s|0 Y d S )NF)lastr )r" r r r r popitemr )r r$ r% tcacheZckZcvr 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
r1 c C s> t | t j}|s |tjO }n |tj M }t | t j| |S r )fcntlZF_GETFLos
O_NONBLOCKZF_SETFL)Zfhandlevalueflagsr r r
setFBlockModer s zUtils.setFBlockModec C sl d}t | ts| 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 )
isinstancelistr itemsappend)realCmdvarsDictZvarsStatir$ r% r r r
buildShellCmd| s
zUtils.buildShellCmdr TFr Nc
s d }}d} d }
|r6|r*t |n ttj|}
tfdd} z tjtjtj||
tj d
} | du r fdd} t | |t j } | r| d } | du rX| r| t
j d} td|f t j}
t|
tj tt j
} | du s|r realCmdIdr r
z"Utils.executeCmd..)stdoutstderrshellenvZ
preexec_fnc s } | d urd| fS d S )NT)poll)retcode)popenr r
_popen_wait_end s z)Utils.executeCmd.._popen_wait_endr z!%x -- timed out after %s seconds.z%s -- failed with %sFr z ... -- failed to read stdout %sr8 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 rA r r3 environid
subprocessPopenPIPEsetsidrM wait_forDEFAULT_SHORTEST_INTERVALloggingZERRORrC errorgetpgidpidkillpgsignalSIGTERMr" sleepDEFAULT_SLEEP_INTERVALSIGKILL
pid_existsOSErrorDEBUGZgetEffectiveLevelrI r7 readIOError
splitlinesrD r rJ closedebugsignamer! _RETCODE_HINTSinfolocalsr )r> timeoutrK outputZtout_kill_treeZ
success_codesr? rI rJ rN rL ZlogCmdrP ZpgideZ std_levellsuccessZsigcodemsgr )rO r> rF 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
rG 9 rH z Utils.wait_for..)callabler" minr rb DEFAULT_SLEEP_TIMEra )Zcondrp intervalZiniretZstmZ timeout_exprr rw r
rX s zUtils.wait_forposixc
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)errnor3 killre EPERM)r] r~ rr r r r
rd E s "zUtils.pid_existsc C s@ dd l }|jj}d}||d| }|dkr8|| dS dS d S )Nr i TF)ctypesZwindllkernel32Z OpenProcessZ CloseHandle)r] r r Z SYNCHRONIZEZprocessr r r
rd R s
c C s. t jt j| d }tj|| }|S )Nr )r3 pathsplitextbasename importlib machinerySourceFileLoader load_module)Z pythonModuleZpythonModuleNamemodr r r
load_python_module_ s
zUtils.load_python_module)r TFTrB N)N)r- r. r/ r0 rz rb ZDEFAULT_SHORT_INTERVALrY objectr1 staticmethodr7 rA rv rX r3 r rd r r r r r
r 7 s0 1
)
r )
__author__Z
__copyright__Z __license__r2 rZ r3 r_ rT sys threadingr r" Zhelpersr r r collectionsr importlib.machineryr r- rC rm dict__dict__r< rl r r r r r
s(