File: //lib/python3.9/site-packages/fail2ban/server/__pycache__/action.cpython-39.pyc
a
�5��;��� @ st d Z dZdZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl m Z zddl
mZ W n ey� ddlmZ Y n0 ddlmZ dd lmZ dd
lmZ dd lmZ d d
lmZmZm Z m Z m Z m Z m Z ee!�Z"e �#� Z$ej%Z&e�'d�Z(dZ)e�'d�Z*dd� dd� d�Z+ddd�Z,e,�-e+� G dd� dee.�Z/G dd� de.e d �Z0d d d �Z1G d d!� d!e0�Z2dS )"z'Cyril Jaquier and Fail2Ban Contributorsz>Copyright (c) 2004 Cyril Jaquier, 2011-2012 Yaroslav Halchenko�GPL� N)�ABCMeta)�MutableMapping� )�
mapTag2Opt)�DNSUtils)�MyTime)�Utils� )� getLogger�_merge_copy_dicts�
splitwords�substituteRecursiveTags�
uni_string�TAG_CRE�MAX_TAG_REPLACE_COUNTz<F-([A-Z0-9_\-]+)>)�inet4�inet6z^(\w+)\?(family)=(.*)$c C s t tjdd��S )NT�Zfqdn��strr Z getHostname� r r �:/usr/lib/python3.9/site-packages/fail2ban/server/action.py�<lambda>? � r c C s t tjdd��S )NFr r r r r r r @ r )z fq-hostnamez sh-hostname�
� )�br�spc @ s� e Zd ZdZdZdZdd� Zd dd�Zd d
d �ZeZ d!d
d�Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd S )"�
CallingMapa" A Mapping type which returns the result of callable values.
`CallingMap` behaves similar to a standard python dictionary,
with the exception that any values which are callable, are called
and the result is returned as the value.
No error handling is in place, such that any errors raised in the
callable will raised as usual.
Actual dictionary is stored in property `data`, and can be accessed
to obtain original callable values.
Attributes
----------
data : dict
The dictionary data which can be accessed to obtain items uncalled
r )�data�storage� immutableZ
__org_datac O s" t � | _d| _t |i |��| _d S )NT)�dictr! r" r )�self�args�kwargsr r r �__init___ s zCallingMap.__init__Tc C s2 t � | _z | j| _W n ty& Y n0 || _d S �N)r# r! �_CallingMap__org_datar �AttributeErrorr" )r$ r" r r r �resetd s zCallingMap.resetFc C s d| j j| �|t�f S )Nz%s(%r))� __class__�__name__�_asdictr )r$ �
calculatedr r r �_asreprl s zCallingMap._asreprNc s� t � jfi � j��}|s2t � fdd�|�� D ��S t|�� �D ]@\}}t|�r>z"� �|�}|rf||� |||< W q> Y q>0 q>|S )Nc 3 s, | ]$\}}t |�r |� jv r||fV qd S r( )�callable�
CM_REPR_ITEMS)�.0�n�v�r$ r r � <genexpr>u s
�z%CallingMap._asdict.<locals>.<genexpr>)r# r r! �items�listr1 � __getitem__)r$ r/ Zchecker�dr4 r5 r r6 r r. r s
zCallingMap._asdictc C s0 z| j | }W n ty* | j| }Y n0 |S r( )r! �KeyErrorr �r$ �key�valuer r r �
getRawItem� s
zCallingMap.getRawItemc C sb z| j | }W n ty* | j| }Y n0 t|�r^t|d�rN|jjrN|| �n|� }|| j |< |S )N�__code__)r! r< r r1 �hasattrrA � co_argcountr= r r r r: � s
zCallingMap.__getitem__c C s: | j r,| j�� | _| j| _| j�� | _d| _ || j|< d S �NF)r" r! �copyr r) r= r r r � __setitem__� s zCallingMap.__setitem__c C s t d| ��d S )NzKey %r was deleted)r< �r$ r> r r r Z
__unavailable� s zCallingMap.__unavailablec C sX | j r,| j�� | _| j| _| j�� | _d| _ z | j|= W n tyJ Y n0 | j|= d S rD )r" r! rE r r) r< rG r r r � __delitem__� s zCallingMap.__delitem__c C s
t | j�S r( )�iterr r6 r r r �__iter__� s zCallingMap.__iter__c C s
t | j�S r( )�lenr r6 r r r �__len__� s zCallingMap.__len__c C s | � t| j| j��S r( )r, r r r! r6 r r r rE � s zCallingMap.copy)T)F)FN)r- �
__module__� __qualname__�__doc__r2 � __slots__r' r+ r0 �__repr__r. r@ r: rF Z_CallingMap__unavailablerH rJ rL rE r r r r r J s
r c @ sX e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
e dd� �Z dd� Z
dS )�
ActionBasea� An abstract base class for actions in Fail2Ban.
Action Base is a base definition of what methods need to be in
place to create a Python based action for Fail2Ban. This class can
be inherited from to ease implementation.
Required methods:
- __init__(jail, name)
- start()
- stop()
- ban(aInfo)
- unban(aInfo)
Called when action is created, but before the jail/actions is
started. This should carry out necessary methods to initialise
the action but not "start" the action.
Parameters
----------
jail : Jail
The jail in which the action belongs to.
name : str
Name assigned to the action.
Notes
-----
Any additional arguments specified in `jail.conf` or passed
via `fail2ban-client` will be passed as keyword arguments.
c C s( d}|D ]}t t||d ��s dS qdS )N)�start�stop�ban�reban�unbanFT)r1 �getattr)�cls�C�required�methodr r r �__subclasshook__� s
zActionBase.__subclasshook__c C s" || _ || _td| jj �| _d S )Nz fail2ban.%s)�_jail�_namer r, r- �_logSys�r$ Zjail�namer r r r' � s zActionBase.__init__c C s dS )z,Executed when the jail/action is started.
Nr r6 r r r rS � s zActionBase.startc C s dS )z,Executed when the jail/action is stopped.
Nr r6 r r r rT � s zActionBase.stopc C s dS )��Executed when a ban occurs.
Parameters
----------
aInfo : dict
Dictionary which includes information in relation to
the ban.
Nr �r$ �aInfor r r rU � s zActionBase.banc C s
| � |�S )rc )rU rd r r r rV s zActionBase.rebanc C s dS rD r r6 r r r � _prolongable s zActionBase._prolongablec C s dS )z�Executed when a ban expires.
Parameters
----------
aInfo : dict
Dictionary which includes information in relation to
the ban.
Nr rd r r r rW s zActionBase.unbanN)r- rM rN rO � classmethodr] r' rS rT rU rV �propertyrf rW r r r r rR � s
rR )� metaclass� str2seconds�ignore)�timeoutZbantimec s� e Zd ZdZed�Zdd� Z� fdd�Zedd� �Z d d
� Z
e
Z d d � Z e
d
d� �Ze
dd� �Zdd� Zdd� Zg dfdd�Ze
dd� �Ze
dd� �Ze
d d � �Zd d � ZdNd!d"�ZdOd$d%�Ze
d&d'� �Zd(d)� Zd*d+� Zd,d-� Z d.d/� Z d0d1� Z dPd2d3�Z d4d5� Z dQd6d7�Z!e"�#d8�Z$ed9d:� �Z%edRd<d=��Z&e"�#d>�Z$e"�#d?�Z'edSd@dA��Z(e
dBdC� �Z)dDdE� Z*dTdGdH�Z+dUdIdJ�Z,e-dVdLdM��Z.� Z/S )W�
CommandActiona� A action which executes OS shell commands.
This is the default type of action which Fail2Ban uses.
Default sets all commands for actions as empty string, such
no command is executed.
Parameters
----------
jail : Jail
The jail in which the action belongs to.
name : str
Name assigned to the action.
Attributes
----------
actionban
actioncheck
actionreban
actionreload
actionrepair
actionstart
actionstop
actionunban
timeout
)�matchesZ ipmatchesZ
ipjailmatchesc C sZ d| _ zFd| _d| _d| _d| _d| _d| _d| _d| _d| _ d| _
W d| _ nd| _ 0 dS )z8 Clear all lists/dicts parameters (used by reloading)
r �< � r N) �_CommandAction__initrl Z actionstartZ actionban� actionrebanZ actionunban� actioncheckZ actionrepairZ actionflushZ
actionstopZ actionreloadr6 r r r �clearAllParams> s z CommandAction.clearAllParamsc sH t t| ��||� d| _d | _i | _i | _| �� | j� d| j
� d S )Nr z
Created %s) �superrm r' rq �_CommandAction__properties�_CommandAction__started�_CommandAction__substCachert r` �debugr, ra �r, r r r' X s zCommandAction.__init__c C s t S r( )�NotImplemented)rY rZ r r r r] a s z CommandAction.__subclasshook__c C sn |� d�s`| js`t|�s`t�|�}|dkr.d S |dkr@t�|�}d | _| j� � | j
� d||� || j |< d S )N�_rk rj z
Set %s = %r)
�
startswithrq r1 �WRAP_CMD_PARAMS�getr rj rv rx �clearr` ry �__dict__)r$ rb r? Zwrpr r r � __setattr__e s
zCommandAction.__setattr__c C s4 |� d�s(d | _| j�� | j�d|� | j|= d S )Nr| z
Unset %s)r} rv rx r� r` ry r� )r$ rb r r r � __delattr__w s
zCommandAction.__delattr__c s2 � j dur� j S t� fdd�t� �D ��� _ � j S )z`A dictionary of the actions properties.
This is used to substitute "tags" in the commands.
Nc 3 s4 | ],}|� d �stt� |��s|t� |�fV qdS )r| N)r} r1 rX )r3 r> r6 r r r7 � s �z,CommandAction._properties.<locals>.<genexpr>)rv r# �dirr6 r r6 r � _properties� s
�
zCommandAction._propertiesc C s | j S r( )rx r6 r r r � _substCache� s zCommandAction._substCachec C sF | j || j|rd| nd| jd�}|r.d|vr2|S | �|d|i�}|S )N�family=rp �� conditional�cache�<�family)�
replaceTagr� rx �replaceDynamicTags)r$ �tagr� �cmdr r r �
_getOperation� s
�zCommandAction._getOperationc s� d|f}t |�sLt� �s,| j�|i ��� �S � fdd�| j�|i ��� D �S |d }|r�z| j| }W n ty� i }| j|< Y n0 ||� < nPz<| j| }|�� �}t|�� �D ]\� }||kr�|� = q�W n ty� Y n0 dS )z? Get, set or delete command of operation considering family.
Z__eOpCmdc s g | ]\}}� |�r|�qS r r �r3 �fr5 �r� r r �
<listcomp>� r z4CommandAction._operationExecuted.<locals>.<listcomp>r N)rK r1 rx r r8 r< �popr9 )r$ r� r� r% r> r� Zfamdr5 r r� r �_operationExecuted� s(
z CommandAction._operationExecutedNc sB d}d}|s dd� | j �� D �}|D ]�� z�| �|� �}d}|r�|| �|� fdd��vr�|} | jr�t| jjdd�}
|
s�| jj�d�}
t| jjd|
� t �
� |
d < � |
d
< | � ||
�} | � | | j
�}||M }|r�|� |� | �|� |r�|nd� W q$ t�y } zd }| }W Y d} ~ q$d} ~ 0 0 q$|�s>td || j| j|f ��|S )
z�Executes the operation commands (like "actionstart", "actionstop", etc).
Replace the tags in the action command with actions properties
and executes the resulting command.
Tz Script errorc S s g | ]\}}|r|�qS r r )r3 �famoperr5 r r r r� � r z3CommandAction._executeOperation.<locals>.<listcomp>c s | � kS r( r )r� �r� r r r � r z1CommandAction._executeOperation.<locals>.<lambda>Z
actionInfoN�timer� FzError %s action %s/%s: %r)rw r8 r� r� r^ rX �actionsZ_getActionInfo�setattrr r� r� �
executeCmdrl �
ValueError� RuntimeErrorr_ ) r$ r� Z operationr� � afterExec�res�errr� �ret�realCmdre �er r� r �_executeOperation� s8 z CommandAction._executeOperationc C sH | j �d�}|d ur|S d}| j D ]}t�|�r"d} q:q"|| j d< |S )NZ__hasCondSectionFT)r� r �CONDITIONAL_FAM_RE�match)r$ r5 r4 r r r �_hasCondSection� s
z CommandAction._hasCondSectionc C sl | j �d�}|r|S | j �d�}|r<t|ttf�s<t|�}n"| jrXt� rPddgndg}ndg}|| j d< |S )NZ
__familiesZfamiliesr r rp )r� r �
isinstancer9 �setr
r� � allowed_ipv6�r$ r5 r r r � _families� s
zCommandAction._familiesc C s, | j �d�}|dur|S | j}|| j d< |S )z1Checks the action depends on family (conditional)Zactionstart_on_demandN)r� r r� r� r r r �_startOnDemand� s
z CommandAction._startOnDemandc C s | � � S )��Executes the "actionstart" command.
Replace the tags in the action command with actions properties
and executes the resulting command.
)�_startr6 r r r rS s zCommandAction.startFc sZ � j r|s$dS n|s$� j�|�r$dS |dur2|gn� j}� fdd�}� jdd||d�}|S )r� TNc s |r � � d| d � d� j| < d S )N� <actionstop>r )r� rw �r� r� r6 r r �_started s z&CommandAction._start.<locals>._started�
<actionstart>Zstarting�r� r� )r� rw r r� r� )r$ r� �
forceStartr� r� r r6 r r� s zCommandAction._start� <actionban>c C s` |� dd�}| jr,| j� |�s,| j|dd� | �||�sDtd| ��| j� |d�dB | j|< dS ) a) Executes the given command ("actionban" or "actionreban").
Replaces the tags in the action command with actions properties
and ban information, and executes the resulting command.
Parameters
----------
aInfo : dict
Dictionary which includes information in relation to
the ban.
r� rp T�r� zError banning %(ip)sr � N)r r� rw r� � _processCmdr� )r$ re r� r� r r r rU % s
zCommandAction.banc C s t | d�o | jo t| j��� S )N�
actionprolong)rB r� r �isspacer6 r r r rf ; s �zCommandAction._prolongablec C s | � d|�std| ��dS )a Executes the "actionprolong" command.
Replaces the tags in the action command with actions properties
and ban information, and executes the resulting command.
Parameters
----------
aInfo : dict
Dictionary which includes information in relation to
the ban.
z<actionprolong>zError prolonging %(ip)sN)r� r� rd r r r �prolong@ s zCommandAction.prolongc C s: |� dd�}| j� |d�d@ r6| �d|�s6td| ��dS )a Executes the "actionunban" command.
Replaces the tags in the action command with actions properties
and ban information, and executes the resulting command.
Parameters
----------
aInfo : dict
Dictionary which includes information in relation to
the ban.
r� rp r r
�
<actionunban>zError unbanning %(ip)sN)r rw r� r� )r$ re r� r r r rW O s zCommandAction.unbanc C s | � || jrdnd�S )aD Executes the "actionreban" command if available, otherwise simply repeat "actionban".
Replaces the tags in the action command with actions properties
and ban information, and executes the resulting command.
Parameters
----------
aInfo : dict
Dictionary which includes information in relation to
the ban.
z
<actionreban>r� )rU rr rd r r r rV ` s
zCommandAction.rebanc s: dd� � j �� D �}|s dS � fdd�}� jdd||d�S ) a Executes the "actionflush" command.
Command executed in order to flush all bans at once (e. g. by stop/shutdown
the system), instead of unbanning of each single ticket.
Replaces the tags in the action command with actions properties
and executes the resulting command.
c S s g | ]\}}|d @ d kr|�qS )r� r r� r r r r� y r z'CommandAction.flush.<locals>.<listcomp>Tc s& |r"� j �| �r"� j | dM < d S )N���)rw r r� r6 r r � _afterFlush} s z(CommandAction.flush.<locals>._afterFlushz
<actionflush>Zflushingr� )rw r8 r� )r$ r� r� r r6 r �flusho s
zCommandAction.flushc C s | � � S )��Executes the "actionstop" command.
Replaces the tags in the action command with actions properties
and executes the resulting command.
)�_stopr6 r r r rT � s zCommandAction.stopc s| |du r,dd� � j �� D �}|s$dS i � _ n2z � j | dM < |g}W n ty\ Y dS 0 � fdd�}� jdd ||d
�S ) r� Nc S s g | ]\}}|r|�qS r r r� r r r r� � r z'CommandAction._stop.<locals>.<listcomp>Tr c s |r� � d| d � d S )Nr� )r� r� r6 r r �_stopped� s z%CommandAction._stop.<locals>._stoppedr� Zstoppingr� )rw r8 r< r� )r$ r� r� r r6 r r� � s
zCommandAction._stopc K s | � dd�S )z�Executes the "actionreload" command.
Parameters
----------
kwargs : dict
Currently unused, because CommandAction do not support initOpts
Replaces the tags in the action command with actions properties
and executes the resulting command.
z<actionreload>Z reloading)r� )r$ r& r r r �reload� s zCommandAction.reloadc C sV d}| j rRt| j�� �D ]8\}}|r| �||�sd| j|< | �d|d� |dM }q|S )zFExecutes the invariant check with repair if expected (conditional).
Tr r� NF)rs r9 rw r8 �_invariantCheckr� )r$ � beforeRepairr� r� �startedr r r �consistencyCheck� s
z CommandAction.consistencyCheckz [\\#&;`|*?~<>^()\[\]{}$'"\n\r]c s( ddd�� � fdd�}| j �||�}|S )a5 Escape characters which may be used for command injection.
Parameters
----------
value : str
A string of which characters will be escaped.
Returns
-------
str
`value` with certain characters escaped.
Notes
-----
The following characters are escaped::
\#&;`|*?~<>^()[]{}$'"
r4 �r)r �
c s | � � }d� �||� S )N�\)�groupr )�m�c�Z_map2cr r � substChar� s z*CommandAction.escapeTag.<locals>.substChar)�
ESCAPE_CRE�sub)rY r? r� r r� r � escapeTag� s
zCommandAction.escapeTagrp c s< d|vr |S |dur:|�f}z
|| W S t y8 Y n0 t|t�}|�|s�d �}|dur�dt|��f}z || �W n t y� Y n0 �du r�t|��j|d��|dur��||< tt�� � ���fdd�} td }
t� | |�} |r�q&| |k�s&d| v�r�q&| }|
d8 }
|
dkr�t
d |f ��q�|du�r8| ||< | S )
z�Replaces tags in `query` with property values.
Parameters
----------
query : str
String with tags.
aInfo : dict
Tags(keys) and associated values for substitution in query.
Returns
-------
str
`query` string with tags replaced.
r� Nz
subst-tags)rk �addreplc sn | � d�}d }�r$��|d � �}|d u rN��|�}|d u rN� �|| � � �S t|�}|�jv rj��|�}|S )Nr �?)r� r r � _escapedTagsr� �r� r� r? ��ADD_REPL_TAGS_CMrY r� ZsubInfor r �substVal s
z*CommandAction.replaceTag.<locals>.substValr r zaunexpected too long replacement interpolation, possible self referencing definitions in query: %s) r< r� r �idr r� �
ADD_REPL_TAGSr r r� r� ) rY �queryre r� r� r� ZckeyZ noRecReplZcsubkeyr� �countr? r r� r r� � sP
� ��
zCommandAction.replaceTagz![\\#&;`|*?~<>\^\(\)\[\]{}$'"\n\r]z\Wc s� t � ��s��fdd��tt�� � ��fdd�}t�||�}d|v rp��d���sVi ���fdd�}t�||�}�r�t�|��}|S ) a% Replaces dynamical tags in `query` with property values.
**Important**
-------------
Because this tags are dynamic resp. foreign (user) input:
- values should be escaped (using "escape" as shell variable)
- no recursive substitution (no interpolation for <a<b>>)
- don't use cache
Parameters
----------
query : str
String with tags.
aInfo : dict
Tags(keys) and associated values for substitution in query.
Returns
-------
str
shell script as string or array with tags replaced (direct or as variables).
c s2 � j �|�r.d� j�d| � } |�| < d| }|S )Nzf2bV_%sr| �$)r� �search�
ESCAPE_VN_CREr� )r� r? )rY �varsDictr r � escapeValS s
z3CommandAction.replaceDynamicTags.<locals>.escapeValc sL | � d�}z �| }W n" ty8 � �|| � � � Y S 0 t|�}�||�S )Nr )r� r< r r r� )r� re r� r r r� a s
z2CommandAction.replaceDynamicTags.<locals>.substValr� zF-*c sB t | �d��}zt�| �}W n ty2 Y dS 0 � d| |�S )Nr rp ZF_)r r� r r< r� )r� �tickDatar r �substTags s z2CommandAction.replaceDynamicTags.<locals>.substTag) r# r r� r r� r � FCUSTAG_CREr Z
buildShellCmd)rY r� re r� r� r� r )r� re rY r� r� r� r r� 8 s
z CommandAction.replaceDynamicTagsc C s t | dd�S )N� _banEpochr )rX r6 r r r �banEpoch� s zCommandAction.banEpochc C s6 | j dur&| j jjd | _| j j_n | jd | _dS )z�Increments ban epoch of jail and this action, so already banned tickets would cause
a re-ban for all tickets with previous epoch.Nr )r^ r� r� r� r6 r r r �invalidateBanEpoch� s
z CommandAction.invalidateBanEpochTc C s |s|dur|| j vrdS | �d|�}|r8| �|| j�r<dS |rJ|� sJdS | j�d� | �� | �d|�}|r�| �|| j�s�d| j |< | j�d� dS d| j |< n8z| �|� W n t y� Y n0 | j
||p�| j d � | j � |��r| �|| j��s| j�d� dS dS )
z0Executes a substituted `actioncheck` command.
Nr z
<actioncheck>���z<Invariant check failed. Trying to restore a sane environmentz<actionrepair>r z Unable to restore environmentr� )
rw r� r� rl r` �errorr� �criticalr� r� r� r� r )r$ r� r� r� ZcheckCmdZ repairCmdr r r r� � s6
�
z CommandAction._invariantCheckc s� � dkr�j �d� dS z |d }W n ttfy> d}Y n0 d}|r|�jr|� �fdd�}�j||� dkd �}|d
kr|d S �j� �j|r�d | nd�jd
�}|dur��� ||�}n� }��
|�j �}|d
7 }|s�|d
krD|S qDdS )a� Executes a command with preliminary checks and substitutions.
Before executing any commands, executes the "check" command first
in order to check if pre-requirements are met. If this check fails,
it tries to restore a sane environment before executing the real
command.
Parameters
----------
cmd : str
The command to execute.
aInfo : dictionary
Dynamic properties.
Returns
-------
bool
True if the command succeeded.
rp �
Nothing to doTr� r c s( � dkr$�j �d�s$�j�d� dS dS )Nr� Zactionrepair_on_unbanz,Invariant check failed. Unban is impossible.FT)r� r r` r� r �r� r$ r r �
_beforeRepair� s z0CommandAction._processCmd.<locals>._beforeRepairr� r� r Fr� r� N) r` ry r< � TypeErrorrs r� r� r� rx r� r� rl )r$ r� re r� Zrepcntr� r� r� r r� r r� � s.
� zCommandAction._processCmdro c K sn t �� tjk rt �d| � | s,t �d� dS t�* tj| |fddd�|��W d � S 1 s`0 Y dS )aI Executes a command.
Parameters
----------
realCmd : str
The command to execute.
timeout : int
The time out in seconds for the command.
Returns
-------
bool
True if the command succeeded.
Raises
------
OSError
If command fails to be executed.
RuntimeError
If command execution times out.
� r� TF)�shell�outputN) �logSys�getEffectiveLevel�logging�DEBUG�logry � _cmd_lockr r� )r� rl r& r r r r� � s
zCommandAction.executeCmd)NF)r� )N)N)rp NN)N)NNT)N)ro )0r- rM rN rO r� r� rt r' rg r] r� rF r� rh r� r� r� r� r� r� r� r� rS r� rU rf r� rW rV r� rT r� r� r� �re�compiler� r� r� r� r� r� r� r� r� � staticmethodr� �
__classcell__r r rz r rm sf
'
Y
I
(
?rm )3�
__author__�
__copyright__� __license__r� �osr� �signal�
subprocess�tempfile� threadingr� �abcr �collections.abcr � ImportError� collectionsZ failregexr Zipdnsr Zmytimer �utilsr Zhelpersr r r
r r r r r- r� �Lockr� Z
IPv6IsAllowedr� r� r� Z
COND_FAMILIESr� Z
DYN_REPL_TAGSr� �update�objectr rR r~ rm r r r r �<module> sN $
��
oc�