Current File : //usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyc
�
��&dc @ s& d Z 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 m Z d d l
m Z m Z d d l
m Z d d l m Z m Z d d
l m Z d d l m Z d d
l m Z d d l m Z m Z d e f d � � YZ d e f d � � YZ d e f d � � YZ d S( s Serg G. Brester (sebres)s" Copyright (c) 2014 Serg G. Brestert GPLi����Ni ( t MyTime( t
FailTickett BanTicket( t FailManager( t Observerst ObserverThread( t Utilsi ( t LogCaptureTestCase( t DummyJail( t
getFail2BanDbt
Fail2BanDbt BanTimeIncrc B sA e Z d � Z d � Z d � Z d d � Z d � Z d � Z RS( c C s> t t | � j � t � | _ | j | j _ t � | _ d S( s Call before every test case.N( t superR t setUpR t _BanTimeIncr__jailt calcBanTimeR t Observer( t self( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR . s c C s t t | � j � d S( N( R
R t tearDown( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR 5 s c C s | j j | j | | � S( N( R R R ( R t banTimet banCount( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR 8 s c C s" | j } | j d d � | j | j d � t � | j d d � | j | j d � d* � | j d d � | j d d � | j d | � | j g t d
d � D] } | j d | � ^ q� d
d d d d d d d d d g
� | j d d � d
d d d d d d d d d g
} | d k rpt | j d � � } | d k rp| d | d
!| | d g d | } qpn | j g t d
d � D] } | j d | � ^ q�| � | j d d � | j d d � | j g t d
d � D] } | j d | � ^ q�d d d d d d d d d d g
� | j d d � | j g t d
d � D] } t
| j d | � � ^ qDd d d d! d" d# d d d d g
� | j d d � | j d d$ � | j g t d
d � D] } | j d | � ^ q�d
d d d d d d% d% d% d% g
� | j d d& � | j d d' � | j t g t d( � D]7 } d
g t d( � D] } | j d d
� ^ qLk ^ q6k � | j d d � | j
t g t d( � D]7 } d
g t d( � D] } | j d d
� ^ q�k ^ q�k � | j d d � | j d d � | j d d& � | j d d � d S(+ Nt incrementt truet maxtimet 1di i< t rndtimet factort multipliersi i iX i� i` i� i�% i K i � i , i�Q t 30di X i � i ` t i i t 2s 1.33i< ix i� i�1 i�c i�� t 12hi�� t 24ht 5mi
i� i�Q ( R t setBanTimeExtrat assertEqualt getBanTimeExtrat Truet Nonet xrangeR t lent splitt intt
assertTruet Falset assertFalse( R R t at it arrt multcntt c( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyt testDefault; sZ +%$ .++%1%+%QQc C sJ | j d � | j d j g t d � D] } t d | >� ^ q& � � d S( Ns 1 2 4 8 16 32 64 128 256R i i ( R4 t joinR( t str( R R0 ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyt testMultipliersz s
c C sW | j } | j d d � | j d d � | j d d � | j d d � | j d d � | j g t d d
� D] } t | j d | � � ^ qo d d
d d d d d d d d g
� | j d d � | j g t d d
� D] } t | j d | � � ^ q� d d
d d d d d d d d g
� | j d d � | j d d � | j g t d d
� D] } t | j d | � � ^ qWd d d d d d d d d d g
� | j d d � | j d d � | j g t d d
� D] } t | j d | � � ^ q�d d
d d d d d d d d g
� | j d d � | j d d � | j t g t d � D]= } d g t d � D] } t | j d d � � ^ qek ^ qOk � | j d d � | j t g t d � D]= } d g t d � D] } t | j d d � � ^ q�k ^ q�k � | j d d � | j d d � | j d d � | j d d � | j d d � d S(! NR R! R t formulasG ban.Time * math.exp(float(ban.Count+1)*banFactor)/math.exp(1*banFactor)R s 2.0 / 2.885385R i i iX i� i` i� i�% i K i � i , i�Q R iX i� i` t 1i^ iQ i/ i� R i�� R" i
(
R R# R' R$ R( R+ R R, R- R. ( R R/ R0 R3 ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyt testFormula� sF 1%1%1%1%WWN( t __name__t
__module__R R R R' R4 R7 R: ( ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR , s ? t
BanTimeIncrDBc B s8 e Z d � Z d � Z d d � Z d � Z d � Z RS( c C s� t t | � j � t d k r@ t j d k r@ t j d � � n t d k rP d St j
d d � \ } | _ t | j � | _
t � | _ | j
| j _ t � | _ | j t _ d S( s Call before every test case.i i sE Unable to import fail2ban database module as sqlite is not available.Ns .dbt fail2ban_( i i ( R
R= R R R' t syst version_infot unittestt SkipTestt tempfilet mkstempt
dbFilenameR
t dbR t jailt databaseR R R t Main( R t _( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR � s c C sM t d k r d S| j j � d t _ t j | j � t t
| � j � d S( s Call after every test case.N( R R' R t stopR RI t ost removeRE R
R= R ( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR � s
c C sY | j } | d k r0 | j | j j � � } n | j d � | j j | | | � } | S( N( RG R' t
getBanTimet actionst
setBanTimeR t incrBanTime( R t ticketR RG t incrTime( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyRQ � s
c C sJ t d k r d S| j } | j j | � | j j d � | j d d � | j d d � d } t t j
� � } t | | g � } | j g t
d � D] } | j | d � ^ q� d d d g � | j � | j j | | � | j g | j j | | d t � D] \ } } } | | | f ^ q d | d f g � | j | d � | j | j | d � d
� | j j | | � | j g | j j | | d t � D] \ } } } | | | f ^ q�d | d d
f g � | j g | j j | d d t � D] \ } } } | | | f ^ q�d | d d
f g � | j g | j j | d
| d | �D] \ } } } | | | f ^ qMd | d d
f g � | j | j j d
d1 d | d t � g � | j j d | d t � } | j t | � d | | d f � | j j d | d t � } | j t | � d | | d f � | j j d | d | d t � } | j t | � d | | d f � d
} xh t
d � D]Z } | j | | d � | j | d � }
| j |
| d � | j j | | � |
} q�W| j | | d � | j | d � }
| j |
| d � | j |
| � | j j | | � |
} t | d | d3 g � } | j d5 � | j � | j j | | � t | d | d7 g � } | j d9 � | j � | j j | | � | j j d | d t � } | j t | � d � | j t | d � d | | | d | f � | j t | d � d | d | d; d= f � | j j d | d? d t � } | j t | � d � | j t | d � d | d | dA dC f � | j | d j | � � | j | d j | � � | j | d j | � � | j | d j | dE � � t | d | dG g � } | j | j | d! � � | j | j | d" � � | j d" � | j | j | d! � � | j | j | d" � � | j d! � | j | j | d" � � | j d" � | j � | j j | | � | j j d | d t � } | j t | � d � | j t | d � d | d | dI d" f � | j j � | j j d | d t � } | j t | � d � | j d! � | j � | j j | | � | j j � | j j d | d t � } | j t | � d � | j | d j � | � dK | j _ | j j � | j j d | d t � } | j t | � d � | j | d j � | d � dM | j _ | j j � | j j d | d t � } | j | g � t d% d& � } t | j _! | j d d � | j | _" | j j | � t d' d( d% d& � }
t |
j _! | j |
_" | j j |
� t | | g � } | j d) � | j � | j j | | � t | | d) g � } | j d* � | j# d � | j � | j j |
| � | j j d | d | d t � } | j t | � d � | j t | d � d | | d) f � | j j d |
d | d t � } | j t | � d � | j t | d � d+ | | d) d* f � x7 | j j | | � D] } | j | d | d) f � Pq�
Wx; | j j | |
� D]$ } | j | d | d) d* f � Pq�
Wx: | j j | d, t �D] } | j | d | d- f � Pq4 W| j$ d t � | j% � } | j | j& � | j t | � d | | d) f � |
j$ d t � | j |
j% � t � | j d. d/ � | j$ � | j% � } | j | j& � | j t | � d | | d! f � |
j$ � | j |
j% � t � d S(N Ni
R R R s& 1 2 4 8 16 32 64 128 256 512 1024 2048s 192.0.2.1i i i i i t t
forbantimet fromtimei��i< t correctBanTimet ipsF FailTicket: ip=%s time=%s bantime=20 bancount=2 #attempts=0 matches=[]sH [FailTicket: ip=%s time=%s bantime=20 bancount=2 #attempts=0 matches=[]]RG i R i i R9 i$ i sG FailTicket: ip=%s time=%s bantime=%s bancount=13 #attempts=0 matches=[]sF FailTicket: ip=%s time=%s bantime=%s bancount=1 #attempts=0 matches=[]i t 3iX i����i���i���t backendt pollingt names DummyJail-2ip i�. sF FailTicket: ip=%s time=%s bantime=%s bancount=2 #attempts=0 matches=[]t overalljailsiPF R t 10mi`���i����i� i�Q i� i�� i� i�Q ip i@� i� i�Q ip i@� i8 i � i� i�Q i� i�� i8 i � ip i@� ip i@� i��i ]��i���i ��(' R R' RG RF t addJailRO RP R# R+ R t timeR R$ R( RQ t incrBanCountt addBant getBanR- t setTimeR& t getCurrentBansR6 t assertNotEqualR) R. t
isTimedOutR, t purget getIPt _purgeAgeR t filtert
ignoreSelfRH t setBanCountt restoreCurrentBanst
getFailTickett restored( R RG RX t stimeRR R0 R t timeOfBant lastBanTimet restored_ticketsR t ticket2t jail1t jail2t ticket1t row( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyt testBanTimeIncr� sT (
= ==@ !
!
!
c C s� t d k r d S| j } | j j | � | j j d � | j d d � t j } | j
� | j | j � | j d � | j
d � d$ | j _ | j d d d
� | j d d t �| j d
� | j d � | j
d � t t j � � } | j j � } | j | g � d } t | | d g � } t � } | j d � x7 t d � D]) } | j | � | j d | | | � qOW| j
d � | j | j � d � | j | j � � | j d � | j j! | j | � | j j" d | d | d d t# � } | j t$ | � d � t | | g � } t � } | j d � | j | � | j d | | j | � | j
d � t% j& | j d � }
| j |
� | j |
j' � | j( � � |
} t) j* | � }
| j |
| � | j d |
| d � | j
d � | j |
j+ � d � | j |
j � d � | j j" d | d | d t# � } | j t$ | � d � | j | d j+ � d � | j | d j � d � t | | d d g � } | j, | � | j- | j j. � � t | t j � d g � } | j, | � | j | j j. � � | j
d � | j j" d | d | d t# � } | j t$ | � d � | j | d j+ � d � | j | d j � d � t | d t j � d g � } | j d � | j, | � | j | j j. � � | j
d � t | d t j � d g � } | j d! � | j, | � | j- | j j. � � | j
d � | j j" d | d | d t# � } | j t$ | � d" � | j | d j+ � d � | j | d j � d � | j/ � d S(% Ni
R R t nopi i���i< t DB_PURGEg����MbP?t db_purges Purge database event occurredt waitg�������?s 192.0.2.1ix i t failureFoundi i RG RV RW i t banFoundi� s test-expired-ban-times test-actionsi@ i R9 s test-permanenti����iX i i���i ��(0 R R' RG RF R_ RO RP R# R RI t startt db_sett addt
wait_emptyRj t add_named_timert assertLoggedR& t wait_idleR+ R R` t getBansR$ R R t setMaxRetryR( t
addFailuret getBanCountR, Ro Rm Rb Re R- R) R t wait_fort getRetryt getMaxRetryR t wrapRN t
putFailTicketR. t checkBanRK ( R RG t obsRq t ticketsRX RR t failManagerR0 Rt Ru t failticket2( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyt testObserver� s�
%
!
!
!N( R; R< R R R' RQ Rz R� ( ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR= � s
�t ObserverTestc B sB e Z d � Z d � Z d � Z d e f d � � YZ d � Z RS( c C s t t | � j � d S( s Call before every test case.N( R
R� R ( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR 6 s c C s t t | � j � d S( s Call after every test case.N( R
R� R ( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR : s c C sf t � } | j � | j d � t d g � } | j d | j � | j d | j d � | j d � | j | j � | j | t d g � � t
| _ | j d | j � | j d | j d � | j d t j
� | j | j � | j | t d g � � t | _ | j d � | j | t d g � � | j | j � � | j | j � � | j � d } d S( Ni t testt callt test2t test3i
( R R� R� t setR� t clearR� R. t is_fullR$ R& t pausedR t DEFAULT_SLEEP_TIMER, R- t isActivet isAliveRK R' ( R R� t o( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyt testObserverBanTimeIncr> s,
t _BadObserverc B s e Z d � Z RS( c C s t d � � d S( Ns run bad thread exception( t RuntimeError( R ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyt run^ s ( R; R< R� ( ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR� ] s c s� t j � } d � | _ t j } g � � f d � t _ zG | j � | j � | j � � j t j
� � f d � d � � Wd | t _ X� j d � � j t
� � d � � j � d d t � � j t � d d � d � d S( Nc S s d S( N( ( ( t v( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyt <lambda>d s c s
� j | � S( N( t append( t args( t x( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR� h s c s t � � o � j d � S( Ns Unhandled exception( R) t
_is_logged( ( R R� ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR� m s i s Unhandled exceptioni i s run bad thread exception( R� R� R� R? t __excepthook__R� RK R5 R, R R� R� R$ R) R� R6 ( R R� t prev_exchook( ( R R� sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyt testObserverBadRuna s
)
( R; R< R R R� R R� R� ( ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyR� 4 s
( t
__author__t
__copyright__t __license__RL R? RA RC R` t
server.mytimeR t
server.ticketR R t server.failmanagerR t server.observerR R t server.utilsR t utilsR t dummyjailR t databasetestcaseR
R R R= R� ( ( ( sC /usr/lib/python2.7/site-packages/fail2ban/tests/observertestcase.pyt <module> s& �� �
Mini Shell