a
=*f @ sp d Z dZdZddlmZ ddlZddlmZmZ dd l m
Z
m Z e
e Z
ejZG d
d d ZG d d
d
eZdS )z
Cyril Jaquierz Copyright (c) 2004 Cyril JaquierZGPL )LockN )
FailTicket BanTicket ) getLogger BgServicec @ sx e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
d ddZ dd Z dd Z
dd Zd d d ZdS ) FailManagerc C s4 t | _t | _d| _d| _d| _d| _t | _ d S )N iX r )
r _FailManager__lockdict_FailManager__failList_FailManager__maxRetry_FailManager__maxTime_FailManager__failTotal
maxMatchesr _FailManager__bgSvcself r ?/usr/lib/python3.9/site-packages/fail2ban/server/failmanager.py__init__( s zFailManager.__init__c C s
|| _ d S Nr r valuer r r setFailTotal1 s zFailManager.setFailTotalc C s | j S r r r r r r getFailTotal4 s zFailManager.getFailTotalc C sP | j 6 t| jtdd t| j D fW d S 1 sB0 Y d S )Nc S s g | ] }| qS r getRetry).0fr r r
: z,FailManager.getFailCount..)r lenr sumlistvaluesr r r r getFailCount7 s zFailManager.getFailCountc C s
|| _ d S r r r r r r setMaxRetry< s zFailManager.setMaxRetryc C s | j S r r* r r r r getMaxRetry? s zFailManager.getMaxRetryc C s
|| _ d S r r r r r r
setMaxTimeB s zFailManager.setMaxTimec C s | j S r r- r r r r
getMaxTimeE s zFailManager.getMaxTimer Fc C s d}| j | }z| j| }||u r4d }d}n*| jrB| nd }| }|dkr^|d7 }| } || | j | ||| | jr| }t
|| jkr| || j d n
| d W nz t y< |st
|tr| Y W d S t
|tr |}n
t|}|| kr.|| || j|< Y n0 | }| jd7 _t tkrddd | j D }
ttd| jt
| j|
f W d n1 s0 Y | j |S )Nr r z, c S s g | ]\}}d || f qS )z%s:%dr )r! kvr r r r# v s z*FailManager.addFailure..zMTotal # of detected failures: %d. Current failures from %d IPs (IP:count): %s) r ZgetIDr r Z
getMatchesZ
getAttemptgetTimeZ
adjustTimer Zincr% Z
setMatchesKeyError
isinstancer r r wrapZsetRetryr logSysZgetEffectiveLevellogLeveljoinitemslogr service) r ticketcountZobservedZattemptsfidZfDatamatchesZattemptZunixTimeZfailures_summaryr r r
addFailureH sP
&
zFailManager.addFailurec C s
t | jS r )r% r r r r r size~ s zFailManager.sizec s | j 8 | j fdd| j D }t|t| jkrFt | _nt|s^W d d S t|d t| jd kr|D ] }| j|= q|n t fdd| j D | _W d n1 s0 Y | j d S )Nc s g | ]\}}| kr|qS r r2 r! r> itemtimer r r# s z'FailManager.cleanup..g @g @c 3 s& | ] \}}| kr||fV qd S r rB rC rE r r s
z&FailManager.cleanup..)r r r r9 r% r
r r; )r rF Ztodeleter> r rE r cleanup s
<zFailManager.cleanupc C sJ | j 0 z | j|= W n ty& Y n0 W d n1 s<0 Y d S r )r r r3 )r r> r r r
delFailure s
zFailManager.delFailureNc C s | j j |d ur || jv r |gn| jD ]:}| j| }| | jkr&| j|= | W d S q&W d n1 sv0 Y | j td S r )r r r r r r; FailManagerEmpty)r r> datar r r toBan s "
6
zFailManager.toBan)r F)N)__name__
__module__ __qualname__r r r r) r+ r, r. r/ r@ rA rH rI rL r r r r r & s
6r c @ s e Zd ZdS )rJ N)rM rN rO r r r r rJ s rJ )
__author__Z
__copyright__Z __license__ threadingr Zloggingr< r r Zhelpersr r rM r6 DEBUGr7 r ExceptionrJ r r r r s