/usr/lib/python3/dist-packages/fail2ban/tests/__pycache__
Edit: /usr/lib/python3/dist-packages/fail2ban/tests/__pycache__/databasetestcase.cpython-310.pyc (16452B)
o
;s*bâY ã @ sè d Z dZddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl m
Z
ddlmZmZ d d
lmZ z dd lmZ ejZW n eyR dZY nw d d lmZmZ ej ej e ¡d
¡Z dd„ Z G dd„ deƒZ!dS )z"Copyright (c) 2013 Steven HiscocksÚGPLé Né )Ú
FileContainer)ÚMyTime)Ú
FailTicket)ÚActionsÚUtilsé )Ú DummyJail)Údatabase)ÚLogCaptureTestCaseÚlogSysÚfilesc C s t jjrtdƒS t| ƒS )Nú:memory:)ÚunittestÚF2BÚ memory_dbÚ
Fail2BanDb)Úfilename© r úA/usr/lib/python3/dist-packages/fail2ban/tests/databasetestcase.pyÚ
getFail2BanDb0 s r c sò e Zd Z‡ fdd„Zedd„ ƒZejdd„ ƒZ‡ fdd„Zdd „ Zd
d „ Z d d
„ Z
dd„ Z dd„ Z dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zd d „ Zd d „ Zd d!„ Zd"d#„ Zd$d%„ Zd&d'„ Zd(d)„ Zd*d+„ Zd,d-„ Zd.d/„ Zd0d1„ Z d2d3„ Z ‡ Z S )4Ú DatabaseTestc sJ t t| ƒ ¡ tdu rt d¡‚d| _tjjs t
dd¡\}| _d| _ dS )z Call before every test case.NzEUnable to import fail2ban database module as sqlite is not available.ú.dbÚ fail2ban_ú:auto-create-in-memory:) Úsuperr ÚsetUpr r ÚSkipTestÚ
dbFilenamer r ÚtempfileÚmkstempÚ_db)ÚselfÚ_©Ú __class__r r r 8 s ÿ
zDatabaseTest.setUpc C s( t | jtƒr| jdkrt| jƒ| _| jS )Nr )Ú
isinstancer" Ústrr r ©r# r r r ÚdbD s zDatabaseTest.dbc C s t | jtƒr | j ¡ || _d S ©N)r' r" r Úclose)r# Úvaluer r r r* I s
c s8 t t| ƒ ¡ tdu r
dS | jdurt | j¡ dS dS )zCall after every test case.N)r r ÚtearDownr r ÚosÚremover) r% r r r. O s
ÿzDatabaseTest.tearDownc C s, | j jdkr t d¡‚| | j| j j¡ d S )Nr úin :memory: database)r* r r r Ú assertEqualr r) r r r ÚtestGetFilenameX s
z DatabaseTest.testGetFilenamec C sD | | jjd¡ d| j_| | jjd¡ d| j_| | jjd¡ d S )Ni€Q Ú1y6mon15d5h30mi¨_æz2y 12mon 30d 10h 60miP¿Ì)r2 r* Úpurgeager) r r r Ú testPurgeAge] s
zDatabaseTest.testPurgeAgec C s | tjtd¡ d S )Nz/this/path/should/not/exist)Ú assertRaisesÚsqlite3ÚOperationalErrorr r) r r r ÚtestCreateInvalidPathd s
ýz"DatabaseTest.testCreateInvalidPathc C sH | j jdkr t d¡‚| ¡ t| jƒ| _ | | jj | j
¡ v d¡ d S )Nr r1 z3Jail not retained in Db after disconnect reconnect.) r* r r r Ú testAddJailr r Ú
assertTrueÚjailÚnameÚ getJailNamesr) r r r ÚtestCreateAndReconnectj s
þz#DatabaseTest.testCreateAndReconnectc
C sv t d¡s
t d¡‚d | _| jd u rt dd¡\}| _dD ]›}| d| ¡ t
t j
td¡| j¡ t | jt j¡}t ||¡ t |¡ z_t| jƒ| _|dkrq| jd d
d d d
| | j ¡ tdgƒ¡ | t| j ¡ ƒd¡ n | jdd
dd d
| t| j ¡ ƒd¡ | t| j ¡ ƒd¡ W | jr¤| jjdkr¤t | jj ¡ d | _q | jr¸| jjdkr¸t | jj ¡ d | _w d S )Nzsqlite3 --versionzno sqlite3 commandr r )é°6 i z)[test-repair], next phase - file-size: %dúdatabase_v1.dbrA z Repair seems to be successfulzCheck integrityzDatabase updatedT)Úallú/tmp/Fail2BanDb_pUlZJh.logr zRepair seems to be failedzNew database created.r r ) r Ú
executeCmdr r r* r r r! ÚpruneLogÚshutilÚcopyfiler/ ÚpathÚjoinÚTEST_FILES_DIRÚopenÚO_RDWRÚ ftruncater, r Ú assertLoggedr2 Ú getLogPathsÚsetÚlenr? Ú _dbFilenamer0 Ú_dbBackupFilename)r# r$ Ú truncSizeÚfr r r Ú testRepairDbu sF
ÿ
ÿÿ€€þèzDatabaseTest.testRepairDbc C sT d | _ z“| jd u rt dd¡\}| _t tj t d¡| j¡ t
| jƒ| _ | | j ¡ t
dgƒ¡ | | j ¡ t
dgƒ¡ tdddgƒ}| | j ¡ d |¡ | | j t
j¡t
j¡ | t| j jt
jd
¡ | j jd d d
}| t|ƒd
¡ | |d ¡ d ¡ W | j r“| j jdkr•t | j j¡ d S d S d S | j r¨| j jdkr©t | j j¡ w w w )Nr r rB z"DummyJail #29162448 with 0 ticketsrD ú 127.0.0.1g×£ÆÕ®ÔAúabc
r r iW»Ri@â )ÚfromtimeÚcorrectBanTimer )r* r r r! rG rH r/ rI rJ rK r r2 r? rQ rP r ÚgetBansÚupdateDbÚ __version__r7 ÚNotImplementedErrorÚgetCurrentBansrR Ú
getBanTimerS r0 rT )r# r$ ÚticketÚticketsr r r Ú testUpdateDb˜ s,
ÿ ÿÿzDatabaseTest.testUpdateDbc C sR d | _ | jd u rt dd¡\}| _t tj t d¡| j¡ t
| jƒ| _ | | j ¡ t
dgƒ¡ | | j ¡ t
dgƒ¡ | j ¡ }| t|ƒd¡ tddg d ¢ƒ}| d
¡ | |d |¡ | |d ¡ d
¡ | | j t
j¡t
j¡ tdd| _| j j| jdd}| t|ƒd¡ | |d ¡ d¡ | t| j jt
jd ¡ t | j j ¡ d S )Nr r zdatabase_v2.dbz pam-genericz/var/log/auth.logr z1.2.3.7ifÊ~T)zŠDec 3 09:31:08 f2btest test:auth[27658]: pam_unix(test:auth): authentication failure; logname= uid=0 euid=0 tty=test ruser= rhost=1.2.3.7zŠDec 3 09:31:32 f2btest test:auth[27671]: pam_unix(test:auth): authentication failure; logname= uid=0 euid=0 tty=test ruser= rhost=1.2.3.7zŠDec 3 09:31:34 f2btest test:auth[27673]: pam_unix(test:auth): authentication failure; logname= uid=0 euid=0 tty=test ruser= rhost=1.2.3.7é r r z1.2.3.8©r> )r= rZ iX ) r* r r r! rG rH r/ rI rJ rK r r2 r? rQ rP r\ rR r Ú
setAttemptÚgetIPr] r^ r
r= r` ra r7 r_ r0 rT )r# r$ Úbansrb rc r r r Ú
testUpdateDb2¯ s,
ÿ
zDatabaseTest.testUpdateDb2c C s6 t ƒ | _| j | j¡ | | jj| j d¡v d¡ d S )NTzJail not added to database)r
r= r* ÚaddJailr< r> r? r) r r r r; Ð s þzDatabaseTest.testAddJailc C sZ | ¡ t dd¡\}}t|dƒ| _| j | j| j¡ | || j | j¡¡ t
|¡ d S )Nz.logÚ Fail2BanDb_úutf-8) r; r r! r Ú
fileContainerr* ÚaddLogr= ÚassertInrP r/ r0 )r# r$ r r r r Ú
testAddLog× s zDatabaseTest.testAddLogc C s | ¡ | j ¡ }t|dƒ}| d¡ | ¡ | j ¡ | j ¡ | j ¡ | j ¡ }| |dk¡ | j
| j | j¡ t |dƒ| _|
| j ¡ d¡ |
| j | j | j¡|¡ t|dƒ}| d¡ | ¡ t |dƒ| _|
| j ¡ d¡ |
| j | j | j¡d ¡ t |¡ d S )NÚwz,Some text to write which will change md5sum
r rm z%Some different text to change md5sum
)rq rn Ú getFileNamerL Úwriter, ÚreadlineÚgetPosr< r* Ú updateLogr= r r2 ro r/ r0 )r# r Úfile_ÚlastPosr r r Ú
testUpdateLogâ s2
ÿ
ÿzDatabaseTest.testUpdateLogc C sb | ¡ | | j | jd¡d ¡ dD ] }| j | jd|dt|ƒ ¡ | | j | jd¡|¡ qd S )Nzsystemd-journal)i /hYi/hYi/hYÚTEST)r; r2 r* Ú
getJournalPosr= Ú
updateJournalr( )r# Útr r r ÚtestUpdateJournal s þz DatabaseTest.testUpdateJournalc C s^ | ¡ tdddgƒ}| j | j|¡ | jj| jd}| t|ƒd¡ | t |d tƒ¡ d S )NrX r rY ©r= r )
r; r r* ÚaddBanr= r\ r2 rR r< r' )r# rb rc r r r Ú
testAddBan s ÿzDatabaseTest.testAddBanc C sô | ¡ tddg d¢ƒtddg d¢ƒtddg d¢ƒtddg d¢ƒtd dd
d gƒtd dd
d gƒtd
dd
dgƒg}|D ]
}| j | j|¡ q9| d¡ | jj| jd}| d¡ | t|ƒd¡ t |ƒD ]4\}}t
d||| ¡ ¡ t
d|| ¡ ¡ | ||
¡ |
¡ ¡ | t|| ¡ ƒt| ¡ ƒ¡ qb| d¡ tj}z/dt_|D ]
}| j | j|¡ q¥| d¡ | jj| jd}| d¡ | t|ƒd¡ W |t_n|t_w | d¡ | j | jtdƒ¡ | jj| jd}| t|ƒd¡ | dd¡ d S )NrX r )ú user "test"õ user "Ñâåòà "u user "äöüß"ú 127.0.0.2z 127.0.0.3)rƒ s user "Ñâåòà"s user "äöüß"z 127.0.0.4)rƒ r„ u user "äöüß"z 127.0.0.5rƒ u unterminated Ãz 127.0.0.6z 127.0.0.7s unterminated Ïzjson dumps failedr€ zjson loads failedé zreadtickets[%d]: %rz == tickets[%d]: %rz [test-phase 2] simulate errorsz f2b-test::non-existing-encodingé z [test-phase 3] still operable?z 127.0.0.8é )r; r r* r r= ÚassertNotLoggedr\ r2 rR Ú enumerateÚ DefLogSysÚdebugÚgetDatarh Ú
getMatchesrF r Ú
PREFER_ENCrO )r# rc rb Ú readticketsÚiÚpriorEncr r r ÚtestAddBanInvalidEncoded sH ù
"
z%DatabaseTest.testAddBanInvalidEncodedc C sX | ¡ dD ]}td| ddgƒ}| j | j|¡ q| jj| jd}| t|ƒd¡ |S )N)r r re z
192.0.2.%dr ztest
r€ re )r; r r* r r= r\ r2 rR )r# r‘ rb rc r r r Ú
_testAdd3BansX s zDatabaseTest._testAdd3Bansc C s~ | ¡ }| j | j|d ¡ ¡ | t| jj| jdƒd¡ | j | j|d ¡ |d ¡ ¡ | t| jj| jdƒd¡ d S )Nr r€ r r )r” r* ÚdelBanr= rh r2 rR r\ )r# rc r r r Ú
testDelBana s
" zDatabaseTest.testDelBanc C s6 | ¡ | j | j¡ | t| jj| jdƒd¡ d S )Nr€ r )r” r* r• r= r2 rR r\ r) r r r Ú
testFlushBansj s zDatabaseTest.testFlushBansc C sª | ¡ | j | jtdt ¡ d dgƒ¡ | j | jtdt ¡ d dgƒ¡ | t| jj | jddƒd¡ | t| jj | jddƒd ¡ | t| jj | jd
dƒd ¡ d S ) NrX é< rY é( é2 )r= Úbantimer é r éÿÿÿÿr )
r; r* r r= r r Útimer2 rR r\ r) r r r ÚtestGetBansWithTimep s ÿÿ "z DatabaseTest.testGetBansWithTimec C s | ¡ d}dgtdgƒdœdgtdgƒdœdgtddgƒdœdgtddgƒdœg}d d
„ |D ƒ}d }|D ] }|d 8 }td
t ¡ | |d}| d¡ | j | j|¡ q5|| j_ | j
d
¡}| | ¡ d
¡ | |
¡ t|ƒ¡ | t| ¡ ƒ|¡ | | ¡ || d … ¡ td
t ¡ d |dtddgƒid}| t|ƒ¡ | j | j|¡ | j
d
¡}| |
¡ dt|ƒ ¡ | t| ¡ ƒ|¡ | | ¡ || d … ¡ | jj| jd
t ¡ d d}| |d u¡ | |
¡ t|ƒ¡ | t| ¡ ƒ|¡ | | ¡ || d … ¡ | jj| jd
t ¡ d dd}| t| ¡ ƒd¡ | | ¡ |d d ¡ | jj| jd
t ¡ d dd}| t| ¡ ƒd¡ | g d¢¡ d| j_ | j | j|¡ | jj| jd
t ¡ d d}| |d u¡ | |
¡ t|ƒ¡ | t| ¡ ƒd¡ d S )Nr rY Útest)ÚmatchesÚuserú123
úABC
Úrootz1234
c S s g | ]}|d d ‘qS )r¡ r r )Ú.0rV r r r Ú
… s z=DatabaseTest.testGetBansMerged_MaxMatches..éP é
rX )Údatar r¢ éd )rZ )rZ Ú
maxmatchesre r¡ r )Ú1Ú2Ú3)r; rQ r r rž rg r* r r= Ú
maxMatchesÚ
getBansMergedr2 rh Ú
getAttemptrR rŽ r` r< Ú
setMatches)r# r° ÚfailuresÚ matches2findr‘ rV rb r r r Ú testGetBansMerged_MaxMatches| sd ü
ÿ ÿÿ z)DatabaseTest.testGetBansMerged_MaxMatchesc C s„ | ¡ tdd}| j |¡ tdt ¡ d dgƒ}| d¡ | j | j |¡ tdt ¡ d dgƒ}| d ¡ | j | j |¡ td
t ¡ d d gƒ}| d¡ | j | j |¡ tdt ¡ d d gƒ}| d¡ | j ||¡ | j
d¡}| | ¡ d¡ | |
¡ d ¡ | | ¡ g d
¢¡ | jj
d| j d}| | ¡ d¡ | |
¡ d¡ | | ¡ ddg¡ | t|ƒt| jj
d| j dƒ¡ td
t ¡ d d gƒ}| d¡ | j | j |¡ | t|ƒt| jj
d| j dƒ¡ tdt ¡ d d gƒ}| d¡ | j | j |¡ | t|ƒt| jj
d| j dƒ¡ | j
¡ }| t|ƒd¡ | ttdd„ |D ƒƒƒdd„ |D ƒ¡ | jj
|d}| t|ƒd¡ | jj
dd}| t|ƒd¡ | jj
dd}| t|ƒd¡ | jj
dd}| t|ƒd¡ | jj
dd}| t|ƒd¡ | jj| j d}| t|ƒd¡ | jjd dd}| | ¡ d¡ | jj| j dt ¡ d }| t|ƒd¡ | jj| j dt ¡ t d ¡ d }| t|ƒd¡ | jj| j dt ¡ t d ¡ d }| t|ƒd¡ | d¡ | j | j |¡ | jj| j dt ¡ t d ¡ d }| t|ƒd¡ | d | j ¡ ¡ | j j d¡ | jj| j dt ¡ t d ¡ d }| t|ƒd¡ | |d ¡ d¡ d S ) Nz DummyJail-2rf rX r™ rY r© é r£ rœ r… r¤ éF )rY r£ r¤ r€ r c s s | ]}| ¡ V qd S r+ ©rh ©r¦ rb r r r Ú ì s € z1DatabaseTest.testGetBansMerged..c S s g | ]}| ¡ ‘qS r r¹ rº r r r r§ í s z2DatabaseTest.testGetBansMerged..r é )r› rˆ é r r )r= Úip)r= Ú
forbantimerZ Ú1yearz(ignore ticket (with new max ban-time %r)) r; r
r* rk r r rž rg r r= r± r2 rh r² rŽ ÚidÚassertNotEqualrR ÚassertSortedEqualÚlistrQ r` Ú str2secondsÚ
setBanTimerO Ú
getMaxBanTimeÚactionsra )r# Újail2rb Ú newTicketrc r r r ÚtestGetBansMergedµ s¬
þ
þ
þ
þ ÿ ÿ ÿ
ÿ ÿz DatabaseTest.testGetBansMergedc C sˆ | ¡ | j| j_| j | j¡ t| jƒ}| dtj t
d¡i ¡ t dƒ}| d¡ |
ddg¡ | j |¡ | ¡ | dd ¡ d S )NÚaction_checkainfoz action.d/action_checkainfo.pyz1.2.3.4r½ r zban ainfo %s, %s, %s, %s)TTTT)r; r* r= r rk r Úaddr/ rI rJ rK r rg r³ Ú
putFailTicketÚ_Actions__checkBanrO )r# rÈ rb r r r ÚtestActionWithDB s
ý
z DatabaseTest.testActionWithDBc C s | ¡ | j | j¡ | j ¡ }| t|ƒdko| jj|¡ | jjdd}| t|ƒdko0| jj|¡ | jjdd}| t|ƒdk¡ | j | j¡ | j ¡ }| t|ƒdkoZ| jj|¡ | jjdd}| t|ƒdkoo| jj|¡ | jjdd}| t|ƒdk¡ d S )Nr F)ÚenabledTr )
r; r* ÚdelJailr= r? rp rR r> r< rk )r# Újailsr r r ÚtestDelAndAddJail/ s
z DatabaseTest.testDelAndAddJailc C s | ¡ | j ¡ | t| j ¡ ƒd¡ | j | j¡ | j ¡ | t| j ¡ ƒd¡ | ¡ | j | j¡ | j ¡ | t| j ¡ ƒd¡ | t| jj | jdƒd¡ | ¡ | j
| jt dt
¡ dgƒ¡ | j | j¡ | j ¡ | t| j ¡ ƒd¡ | t| jj | jdƒd¡ d S )Nr r r€ rX rY )r; r* Úpurger2 rR r? rÒ r= r‚ r\ r r r rž r) r r r Ú testPurgeB s&
ÿ
zDatabaseTest.testPurge) Ú__name__Ú
__module__Ú __qualname__r Úpropertyr* Úsetterr. r3 r6 r: r@ rW rd rj r; rq rz r r‚ r“ r” r– r— rŸ r¶ rË rÐ rÔ rÖ Ú
__classcell__r r r% r r 6 s:
#! '
< 9ir )"Ú
__copyright__Ú __license__r/ Úsysr r r8 rG Ú
server.filterr Ú
server.mytimer Ú
server.ticketr Úserver.actionsr r Ú dummyjailr
Úserverr r Ú ImportErrorÚutilsr r
r‹ rI rJ ÚdirnameÚ__file__rK r r r r r r Ú s.
ÿ