/usr/lib/python3/dist-packages/fail2ban/tests/__pycache__
Edit: /usr/lib/python3/dist-packages/fail2ban/tests/__pycache__/observertestcase.cpython-310.pyc (15625B)
o
;s*bY @ s d Z dZdZddlZddlZddlZddlZddlZddlm Z ddl
m Z m Z ddl
mZ dd lmZmZ dd
lmZ d d lmZ d d
lmZ d dlmZmZ G dd deZG dd deZ G dd deZ dS )zSerg G. Brester (sebres)z"Copyright (c) 2014 Serg G. BresterGPL N )MyTime)
FailTicket BanTicket) FailManager) ObserversObserverThread)Utils )LogCaptureTestCase) DummyJail)
getFail2BanDb
Fail2BanDbc sJ e Zd Z fddZ fddZdd Zddd Zd
d Zd d
Z Z S ) BanTimeIncrc s, t t| t | _| j| j_t | _dS Call before every test case.N)superr setUpr
_BanTimeIncr__jail calcBanTimer Observerself __class__ A/usr/lib/python3/dist-packages/fail2ban/tests/observertestcase.pyr . s
zBanTimeIncr.setUpc s t t| d S N)r r tearDownr r r r r 5 zBanTimeIncr.tearDownc C s | j | j||S r )r r r )r banTimebanCountr r r r 8 r zBanTimeIncr.calcBanTimeNc sB | j dd | dd dd | dd dd dd d | | fd
d td d
D g d dd g d}|d urqt|d}|d
k rq|d|d ||d gd
| }| fdd td d
D | dd dd | fdd td d
D g d dd | fdd td d
D g d dd dd | fd d td d
D g d dd dd | d fd!d td"D v dd | d fd#d td"D v d d dd dd dd d S )$N incrementtrueTmaxtime1dQ rndtimefactor multipliersc g | ]} d |qS X r .0iar r
G z+BanTimeIncr.testDefault..r
` % K , r' r' r' 30d)
r8 r9 r: r; r< r= r> i X i i ` r r c r+ r, r. r/ r2 r r r4 S r5 2c r+ r, r. r/ r2 r r r4 Z r5 )
r9 r: r; r< r= r> r' r' r' r' z1.33c g | ]
}t d |qS r, intr r/ r2 r r r4 ` )
i< ix i i1 ic i r' r' r' r' 12hc r+ r, r. r/ r2 r r r4 g r5
r8 r9 r: r; r< r= rH rH rH 24h5mFc & g | ]}d fddt dD v qS )r8 c g | ]} d dqS r- r r. r/ r2 r r r4 n r5 6BanTimeIncr.testDefault...
ranger0 cr2 r r r4 n & rO c rK )r8 c rL rM r. r/ r2 r r r4 r r5 rN rO rP rR r2 r r r4 r rT ) r setBanTimeExtra assertEqualgetBanTimeExtrarQ lensplit
assertTrue assertFalse)r r* arrmultcntr r2 r testDefault; sh & zBanTimeIncr.testDefaultc C s, | d | ddd tdD d S )Nz1 2 4 8 16 32 64 128 256r@ c S s g | ]}t d |> qS )r )strr/ r r r r4 ~ r5 z/BanTimeIncr.testMultipliers.. )r^ joinrQ r r r r testMultipliersz s
"zBanTimeIncr.testMultipliersc s | j dd dd dd dd dd | fd d
td d D g d
dd | fdd
td d D g d dd dd | fdd
td d D g d dd dd | fdd
td d D g d dd dd | d fdd
tdD v dd | d fdd
tdD v dd dd dd dd dd d S ) Nr% rI r( formulazGban.Time * math.exp(float(ban.Count+1)*banFactor)/math.exp(1*banFactor)r) z2.0 / 2.885385r* c rB r, rC r/ r2 r r r4 rE z+BanTimeIncr.testFormula..r r6 r7 r? c rB r, rC r/ r2 r r r4 rE )
r8 r9 r: r; r< r= r> iX i i` 1c rB r, rC r/ r2 r r r4 rE )
i^ iQ i/ i r' r' r' r' r' r' rF c rB r, rC r/ r2 r r r4 rE rG rJ Fc rK )r8 c g | ]
}t d dqS rM rC r/ r2 r r r4 rE 6BanTimeIncr.testFormula...rO rP rR r2 r r r4 rT rO c rK )r8 c re rM rC r/ r2 r r r4 rE rf rO rP rR r2 r r r4 rT )r rU rV rQ rZ r[ r r r2 r testFormula sR zBanTimeIncr.testFormular )
__name__
__module__ __qualname__r r r r^ rb rg
__classcell__r r r r r , s
?r c sB e Zd Z fddZ fddZd ddZdd Zd
d Z ZS )
BanTimeIncrDBc sz t t| tdu rtjdkrtdtdu rdS t dd\}| _
t | j
| _ t
| _| j | j_t | _| jt_dS )r N)r zEUnable to import fail2ban database module as sqlite is not available.z.db fail2ban_)r rl r r sys version_infounittestSkipTesttempfilemkstemp
dbFilenamer dbr
jaildatabaser r r Main)r _r r r r s
zBanTimeIncrDB.setUpc s: t du rdS | j dt_t| j tt |
dS zCall after every test case.N) r r stopr ry osremoveru r rl r r r r r r s
zBanTimeIncrDB.tearDownNc C s< | j }|d u r||j }|d | j|||}|S r )rw
getBanTimeactions
setBanTimer incrBanTime)r ticketr! rw incrTimer r r r s
zBanTimeIncrDB.incrBanTimec
s t d u rd S j} j| |jd |dd |dd d}tt }t
||g fddt d D g d
j| d d j||d d D d
|dfg |d dd j| dd j||d d D d|d dfg dd j|dd dD d|d dfg dd jj|||dD d|d dfg jjd|d dg jj|d d} t|d||d f jj|d d} t|d ||d f jj||d d } t|d ||d f d}t dD ]#}|| d d} ||d j| |}q|| d d} ||d || j| |}t
|d |d g }|d! |
j|| t
|d" |d g }|d# |
j|| jj|d d} t|d t|d$ d%||| d |f t|d
d&|d" |d d#f jj|d' d d} t|d t|d d&|d |d d!f |d
| |d
| |d | |d |d' t
|d( |d# g |d) |d* d* |d) |d* d) |d* d*
j| jj|d d} t|d t|d d&|d( |d# d*f j jj|d d} t|d d)
j| j jj|d d} t|d |d$ | d+ j_ j jj|d d} t|d
|d$ |d" d, j_ j jj|d d} |g t d-d.} d | j _ | dd j| _ j| t d/d-d0}
d |
j _ j|
_ j|
t
||g } | d1 |
j| | t
||d1 g }|d2 | d
|
j|
| jj| |d d } t|d
t|d$ d&||d1f jj|
|d d } t|d
t|d$ d3||d1 d2f j|| D ] } | d
|d1f j||
D ]
} | d|d1 d2f jj|dd4D ] } | d |d5f | j!d d6 | " j# td&||d1f |
j!d d6 |
" d | d7d8 | ! | " j# td&||d)f |
! |
" d d S )9NrO r# r$ r* z&1 2 4 8 16 32 64 128 256 512 1024 2048 192.0.2.1c s g | ]} d qS )rO )r r/ r r r r r4 r5 z1BanTimeIncrDB.testBanTimeIncr.. )rO rO rO c S g | ]
\}}}|||fqS r r r0 r" timeOfBan lastBanTimer r r r4 rE Fr c S r r r r r r r r4 rE r c S r r r r r r r r4 rE Tc S r r r r r r r r4 rE )
forbantimefromtimei)r r correctBanTime)ipr zFFailTicket: ip=%s time=%s bantime=20 bancount=2 #attempts=0 matches=[])r r zH[FailTicket: ip=%s time=%s bantime=20 bancount=2 #attempts=0 matches=[]]rw r r rA r' rH rd i@ r zGFailTicket: ip=%s time=%s bantime=%s bancount=13 #attempts=0 matches=[]zFFailTicket: ip=%s time=%s bantime=%s bancount=1 #attempts=0 matches=[]i 3r- i ] polling)backendz DummyJail-2)namer ip i. zFFailTicket: ip=%s time=%s bantime=%s bancount=2 #attempts=0 matches=[]) overalljailsiPF )r r% 10m)$r rw rv addJailr r rU rD r timer rV rQ incrBanCountaddBangetBansetTimer getCurrentBansr_ assertNotEqualrX r[
isTimedOutrZ purgegetIP _purgeAger
filter
ignoreSelfrx setBanCountrestoreCurrentBans
getFailTicketrestored)
r rw r stimerestored_ticketsr r1 r! ticket2jail1jail2ticket1rowr r r testBanTimeIncr sv
z BanTimeIncrDB.testBanTimeIncrc C s t d u rd S | j}| j| |jd |dd tj}| |
| j | d | d d| j_
|ddd | jd
d d |d
| d | d tt }| j }| |g d}t||d g }t }|d tdD ]}|| | d||| qz| d | | d | | | d | j | j| | jj ||d dd} | t!| d t||g }t }|d || | d|| j| | d t"#|j d}
| |
| |
$ |% |
} t&'| }
| |
| | d|
|d | d | |
( d | |
d | jj ||dd} | t!| d | | d ( d | | d d t||d dg}|)| | *|j+ t|t dg}|)| | |j+ | d | jj ||dd} | t!| d | | d ( d | | d d t|d t d g}|d |)| | |j+ | d t|d t d g}|d! |)| | *|j+ | d | jj ||dd} | t!| d" | | d ( d | | d d |, d S )#NrO r# r$ nopr r DB_PURGEgMbP?db_purgez Purge database event occurredT)waitg?r x r failureFoundr Fr r banFound < ztest-expired-ban-timez test-actionsi@ rd ztest-permanentr r- r )-r rw rv r r r rU r ry startdb_setadd
wait_emptyr add_named_timer assertLogged wait_idlerD r r getBansrV r r setMaxRetryrQ
addFailure getBanCountrZ r r r r rX r
wait_forgetRetry getMaxRetryr wrapr
putFailTicketr[ checkBanr| ) r rw obsr ticketsr r failManagerr1 r r failticket2r r r testObserver s
zBanTimeIncrDB.testObserverr ) rh ri rj r r r r r rk r r r r rl s
jrl c sH e Zd Z fddZ fddZdd ZG dd deZd d
Z Z S ) ObserverTestc t t| dS r )r r r r r r r r 6 zObserverTest.setUpc r r{ )r r r r r r r r : r zObserverTest.tearDownc C s t }| |d tdg}|d|j |d|jd |d | |j | |tdg d|_
|d|j |d|jd |dt j |
|j | |tdg d|_
|d | |tdg |
| |
| | d }d S ) Nr testcalltest2Ttest3rO F)r r r setr clearr r[ is_fullrV pausedr
DEFAULT_SLEEP_TIMErZ isActiveisAliver| )r r or r r testObserverBanTimeIncr> s,
z$ObserverTest.testObserverBanTimeIncrc @ s e Zd Zdd ZdS )zObserverTest._BadObserverc C s t d)Nrun bad thread exception) RuntimeErrorr r r r run^ s z ObserverTest._BadObserver.runN)rh ri rj r r r r r _BadObserver] s r c s t }dd |_tj}g fddt_z | | | t
fddd W |t_n|t_w d t
d d d t td d d d S )
Nc S s dS )Nr r )vr r r d s z1ObserverTest.testObserverBadRun..c s
| S r )append)args)xr r r h s
c s t o dS )NUnhandled exception)rX
_is_loggedr r r r r r m s r r r r r )r r r ro __excepthook__r r| ra rZ r
r r rV rX r r_ )r r prev_exchookr r r testObserverBadRuna s
z ObserverTest.testObserverBadRun)
rh ri rj r r r r r r rk r r r r r 4 s r )
__author__
__copyright__ __license__r} ro rq rs r
server.mytimer
server.ticketr r server.failmanagerr server.observerr r server.utilsr
utilsr dummyjailr
databasetestcaser r r rl r r r r r s.