Current File : //lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyc
�
��&dc @ s; d Z d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l
m Z d d l m
Z
d d l m Z m Z d d
l m Z y d d l m Z e j Z Wn e k
r� d Z n Xd d l m Z m Z e j j e j j e � d
� Z d � Z! d e f d � � YZ" d S( s" Copyright (c) 2013 Steven Hiscockst GPLi����Ni ( t
FileContainer( t MyTime( t
FailTicket( t Actionst Utilsi ( t DummyJail( t database( t LogCaptureTestCaset logSyst filesc C s t j j r t d � St | � S( Ns :memory:( t unittestt F2Bt memory_dbt
Fail2BanDb( t filename( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt
getFail2BanDb0 s
t DatabaseTestc B s e Z d � Z e d � � Z e j d � � Z d � Z d � Z d � Z d � Z d � Z
d � Z d � Z d
� Z
d � Z d � Z d
� Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C sq t t | � j � t d k r1 t j d � � n d | _ t j j sd t
j d d � \ } | _ n d | _ d S( s Call before every test case.sE Unable to import fail2ban database module as sqlite is not available.s .dbt fail2ban_s :auto-create-in-memory:N(
t superR t setUpR t NoneR t SkipTestt
dbFilenameR R
t tempfilet mkstempt _db( t selft _( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyR 8 s c C s= t | j t � r6 | j d k r6 t | j � | _ n | j S( Ns :auto-create-in-memory:( t
isinstanceR t
basestringR R ( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt dbD s !c C s/ t | j t � r" | j j � n | | _ d S( N( R R R t close( R t value( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyR I s c C sI t t | � j � t d k r# d S| j d k rE t j | j � n d S( s Call after every test case.N( R R t tearDownR R R t ost remove( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyR" O s
c C sA | j j d k r$ t j d � � n | j | j | j j � d S( Ns :memory:s in :memory: database( R R R R t assertEqualR ( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testGetFilenameX s c C s^ | j | j j d � d | j _ | j | j j d � d | j _ | j | j j d � d S( Ni�Q t 1y6mon15d5h30mi�_�s 2y 12mon 30d 10h 60mi iP��( R% R t purgeage( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testPurgeAge] s
c C s | j t j t d � d S( Ns /this/path/should/not/exist( t assertRaisest sqlite3t OperationalErrorR ( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testCreateInvalidPathd s c C si | j j d k r$ t j d � � n | j � t | j � | _ | j | j j | j j
� k d � d S( Ns :memory:s in :memory: databases3 Jail not retained in Db after disconnect reconnect.( R R R R t testAddJailR R t
assertTruet jailt namet getJailNames( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testCreateAndReconnectj s
c C s� t j d � s! t j d � � n d | _ | j d k rW t j d d � \ } | _ n x�d D]�} | j d | � t
j t j
j t d � | j � t j | j t j � } t j | | � t j | � z� t | j � | _ | d k rJ| j d d
d d t �| j | j j � t d
g � � | j t | j j � � d � nW | j d d
d d t �| j t | j j � � d � | j t | j j � � d � Wd | j r�| j j d k r�t j | j j � d | _ n Xq^ Wd S( Ns sqlite3 --versions no sqlite3 commands .dbR i�6 i� s) [test-repair], next phase - file-size: %ds database_v1.dbs Repair seems to be successfuls Check integritys Database updatedt alls /tmp/Fail2BanDb_pUlZJh.logi s Repair seems to be faileds New database created.i s :memory:( i�6 i� ( R t
executeCmdR R R R R R R t pruneLogt shutilt copyfileR# t patht joint TEST_FILES_DIRt opent O_RDWRt ftruncateR R t assertLoggedt TrueR% t getLogPathst sett lenR2 t _dbFilenameR$ t _dbBackupFilename( R R t truncSizet f( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testRepairDbu s4
"" #c C s� d | _ zh| j d k r9 t j d d � \ } | _ n t j t j j t
d � | j � t | j � | _ | j | j j
� t d g � � | j | j j � t d g � � t d d d g � } | j | j j � d | � | j | j j t j � t j � | j t | j j t j d
� | j j d d d
d � } | j t | � d
� | j | d j � d � Wd | j r�| j j d k r�t j | j j � n Xd S( Ns .dbR s database_v1.dbs" DummyJail #29162448 with 0 ticketss /tmp/Fail2BanDb_pUlZJh.logs 127.0.0.1gף��ծ�Au abc
i i t fromtimeiW�Rt correctBanTimei@� s :memory:( R R R R R R7 R8 R# R9 R: R; R R% R2 RB RA R t getBanst updateDbt __version__R* t NotImplementedErrort getCurrentBansRC t
getBanTimeRD R$ RE ( R R t tickett tickets( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testUpdateDb� s$ "" " c C s� d | _ | j d k r6 t j d d � \ } | _ n t j t j j t
d � | j � t | j � | _ | j | j j
� t d g � � | j | j j � t d g � � | j j � } | j t | � d � t d d d d
d g � } | j d � | j | d
| � | j | d j � d � | j | j j t j � t j � t d d � | _ | j j d | j d d � } | j t | � d � | j | d
j � d � | j t | j j t j d � t j | j j � d S( Ns .dbR s database_v2.dbs pam-generics /var/log/auth.logi s 1.2.3.7if�~Tu� 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.7u� 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.7u� 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.7i i i s 1.2.3.8R1 R0 RI iX ( R R R R R R7 R8 R# R9 R: R; R R% R2 RB RA RK RC R t
setAttemptt getIPRL RM R R0 RO RP R* RN R$ RE ( R R t bansRQ RR ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt
testUpdateDb2� s0 ""
" c C sK t � | _ | j j | j � | j | j j | j j t � k d � d S( Ns Jail not added to database( R R0 R t addJailR/ R1 R2 R@ ( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyR. � s
c C s} | j � t j d d � \ } } t | d � | _ | j j | j | j � | j | | j j | j � � t
j | � d S( Ns .logt Fail2BanDb_s utf-8( R. R R R t
fileContainerR t addLogR0 t assertInRA R# R$ ( R R R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt
testAddLog� s
c C sx | j � | j j � } t | d � } | j d � | j � | j j � | j j � | j j � | j j � } | j | d k � | j j
| j | j � t | d � | _ | j
| j j � d � | j
| j j | j | j � | � t | d � } | j d � | j � t | d � | _ | j
| j j � d � | j
| j j | j | j � d � t j | � d S( Nt ws, Some text to write which will change md5sum
i s utf-8s% Some different text to change md5sum
( R] RZ t getFileNameR< t writeR t readlinet getPosR/ R t updateLogR0 R R% R[ R R# R$ ( R R t file_t lastPos( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt
testUpdateLog� s.
c C s� | j � | j | j j | j d � d � xV d D]N } | j j | j d | d t | � � | j | j j | j d � | � q3 Wd S( Ns systemd-journali /hYi/hYi/hYt TEST( i /hYi/hYi/hY( R. R% R t
getJournalPosR0 R t
updateJournalt str( R t t( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testUpdateJournal s
"
&c C s� | j � t d d d g � } | j j | j | � | j j d | j � } | j t | � d � | j t | d t � � d S( Ns 127.0.0.1i s abc
R0 i (
R. R R t addBanR0 RK R% RC R/ R ( R RQ RR ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt
testAddBan s
c C s� | j � t d d d d d g � t d d d d d g � t d d d d d g � t d
d d d d g � t d d d d
g � t d d d d g � t d d d d
g � g } x$ | D] } | j j | j | � q� W| j d � | j j d | j � } | j d � | j t | � d � x� t | � D]� \ } } t
j d | | | j � � t
j d | | j � � | j | | j
� | j
� � | j t | | j � � t | j � � � q+W| j d � t j } z| d t _ x$ | D] } | j j | j | � q�W| j d � | j j d | j � } | j d � | j t | � d � Wd | t _ X| j d � | j j | j t d � � | j j d | j � } | j t | � d � | j d d � d S( Ns 127.0.0.1i s user "test"s user "�����"s user "äöüß"s 127.0.0.2u user "Ñâåòà"u user "äöüÃ"s 127.0.0.3s 127.0.0.4u user "äöüß"s 127.0.0.5s unterminated �s 127.0.0.6u unterminated Ïs 127.0.0.7s json dumps failedR0 s json loads failedi s readtickets[%d]: %rs == tickets[%d]: %rs [test-phase 2] simulate errorss f2b-test::non-existing-encodingi s [test-phase 3] still operable?s 127.0.0.8i ( R. R R Rm R0 t assertNotLoggedRK R% RC t enumeratet DefLogSyst debugt getDataRU t
getMatchesR6 R t
PREFER_ENCR? ( R RR RQ t readticketst it priorEnc( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testAddBanInvalidEncoded sF
0
c C s| | j � x= d D]5 } t d | d d g � } | j j | j | � q W| j j d | j � } | j t | � d � | S( Ni i i s
192.0.2.%di s test
R0 ( i i i ( R. R R Rm R0 RK R% RC ( R Rw RQ RR ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt
_testAdd3BansX s
c C s� | j � } | j j | j | d j � � | j t | j j d | j � � d � | j j | j | d j � | d j � � | j t | j j d | j � � d � d S( Ni R0 i i ( Rz R t delBanR0 RU R% RC RK ( R RR ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt
testDelBana s
(-c C sI | j � | j j | j � | j t | j j d | j � � d � d S( NR0 i ( Rz R R{ R0 R% RC RK ( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt
testFlushBansj s
c C s� | j � | j j | j t d t j � d d g � � | j j | j t d t j � d d g � � | j t | j j d | j d d � � d � | j t | j j d | j d d � � d
� | j t | j j d | j d d � � d � d S(
Ns 127.0.0.1i< s abc
i( R0 t bantimei2 i i i i����i (
R. R Rm R0 R R t timeR% RC RK ( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testGetBansWithTimep s
& &..c C sQ | j � d } i d g d 6t d g � d 6i d g d 6t d g � d 6i d g d 6t d d g � d 6i d g d 6t d d g � d 6g } g | D] } | d d
^ q� } d } xZ | D]R } | d 8} t d
t j � | d | �} | j d � | j j | j | � q� W| | j _ | j j
d
� } | j | j � d
� | j | j
� t | � � | j t | j � � | � | j | j � | | � t d
t j � d | d i t d d g � d 6�} | j t | � � | j j | j | � | j j
d
� } | j | j
� d t | � � | j t | j � � | � | j | j � | | � | j j | j d
d t j � d �} | j | d k � | j | j
� t | � � | j t | j � � | � | j | j � | | � | j j | j d
d t j � d d d �} | j t | j � � d � | j | j � | d d � | j j | j d
d t j � d d d
�} | j t | j � � d
� | j d d d g � d
| j _ | j j | j | � | j j | j d
d t j � d �} | j | d k � | j | j
� t | � � | j t | j � � d
� d S( Ni s abc
t matchest testt users 123
s ABC
t roots 1234
i iP i
s 127.0.0.1t datai RI id t
maxmatchesi t 1t 2t 3( R. RB R R R RT R Rm R0 t
maxMatchest
getBansMergedR% RU t
getAttemptRC Rt RO R/ R t
setMatches( R R� t failuresRG t matches2findRw RQ ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testGetBansMerged_MaxMatches| s\
&!
( % % ( c C s� | j � t d d � } | j j | � t d t j � d d g � } | j d � | j j | j | � t d t j � d d g � } | j d � | j j | j | � t d
t j � d d g � } | j d � | j j | j | � t d t j � d d g � } | j d � | j j | | � | j j
d � } | j | j � d � | j | j
� d � | j | j � d d d g � | j j
d d
| j �} | j | j � d � | j | j
� d � | j | j � d d g � | j t | � t | j j
d d
| j �� � t d
t j � d d g � } | j d � | j j | j | � | j t | � t | j j
d d
| j �� � t d t j � d d g � } | j d � | j j | j | � | j t | � t | j j
d d
| j �� � | j j
� } | j t | � d � | j t t d � | D� � � g | D] } | j � ^ qP� | j j
d
| � } | j t | � d � | j j
d d � } | j t | � d � | j j
d d � } | j t | � d � | j j
d d � } | j t | � d � | j j
d d � } | j t | � d � | j j d
| j � } | j t | � d � | j j d
d d d � } | j | j � d � | j j d
| j d d d t j � � } | j t | � d � | j j d
| j d d d t j � t j d � � } | j t | � d � | j j d
| j d d d t j � t j d � � } | j t | � d � | j d � | j j | j | � | j j d
| j d d d t j � t j d � � } | j t | � d � | j d | j j � � | j j j d � | j j d
| j d d d t j � t j d � � } | j t | � d � | j | d j � d � d S( NR1 s DummyJail-2s 127.0.0.1i( s abc
i
i s 123
i s 127.0.0.2s ABC
iF R0 i c s s | ] } | j � Vq d S( N( RU ( t .0RQ ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pys <genexpr>� s i R~ i i i i i����t ipt
forbantimeRI t 1years( ignore ticket (with new max ban-time %r)( R. R R RX R R R RT Rm R0 R� R% RU R� Rt t idt assertNotEqualRC t assertSortedEqualt listRB RO R t str2secondst
setBanTimeR? t
getMaxBanTimet actionsRP ( R t jail2RQ t newTicketRR ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testGetBansMerged� s�
"
"
"
c C s� | j � | j | j _ | j j | j � t | j � } | j d t j j t
d � i � t d � } | j d � | j
d d g � | j j | � | j � | j d t t t t f � d S( Nt action_checkainfos action.d/action_checkainfo.pys 1.2.3.4i R� s ban ainfo %s, %s, %s, %s( R. R R0 R RX R t addR# R9 R: R; R RT R� t
putFailTickett _Actions__checkBanR? R@ ( R R� RQ ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testActionWithDB s
c C sx | j � | j j | j � | j j � } | j t | � d k oM | j j | � | j j d t � } | j t | � d k o� | j j | � | j j d t � } | j
t | � d k � | j j | j � | j j � } | j t | � d k o| j j | � | j j d t � } | j t | � d k o?| j j | � | j j d t � } | j
t | � d k � d S( Ni t enabledi ( R. R t delJailR0 R2 R\ RC R1 t FalseR@ R/ RX ( R t jails( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testDelAndAddJail/ s
((((c C s� | j � | j j � | j t | j j � � d � | j j | j � | j j � | j t | j j � � d � | j � | j j | j � | j j � | j t | j j � � d � | j t | j j d | j � � d � | j � | j j
| j t d t j
� d g � � | j j | j � | j j � | j t | j j � � d � | j t | j j d | j � � d � d S( Ni i R0 s 127.0.0.1s abc
( R. R t purgeR% RC R2 R� R0 Rn RK Rm R R R ( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt testPurgeB s$
(
"
( t __name__t
__module__R t propertyR t setterR" R& R) R- R3 RH RS RW R. R] Rf Rl Rn Ry Rz R| R} R� R� R� R� R� R� ( ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyR 6 s4 # ! '
< 9 i (# t
__copyright__t __license__R# t sysR R R+ R7 t
server.filterR t
server.mytimeR t
server.ticketR t server.actionsR R t dummyjailR t serverR R t ImportErrorR t utilsR R Rq R9 R: t dirnamet __file__R; R R ( ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/databasetestcase.pyt <module> s*
!
Mini Shell