Current File : //usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyo
�
��&dc @ s� d Z d Z d 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 m Z m Z e j j e j j e � d � Z d e
f d
� � YZ d S( s Daniel Blacks Copyright (c) 2013 Daniel Blackt GPLi����Ni ( t
FailTicket( t Utilsi ( t DummyJail( t LogCaptureTestCaset
with_alt_timet with_tmpdirt MyTimet filest ExecuteActionsc B s� e Z d � Z d � Z i d � Z d � Z d � Z d � Z d � Z d � Z d � Z
d � Z e d
� � Z
d � Z d � Z e e d
� � � Z RS( c C s2 t t | � j � t � | _ | j j | _ d S( N( t superR t setUpR t _ExecuteActions__jailt actionst _ExecuteActions__actions( t self( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyR ) s c C s t t | � j � d S( N( R
R t tearDown( R ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyR / s c C s� | j j d � | j d } d | j d d � | _ d | j d d � | _ d | j d d � | _ d | j d
d � | _ d | j d d � | _ d
| j d d � | _ | S( Nt ips
echo ip startt startt s echo ip ban <ip>t bans echo ip unban <ip>t unbans
echo ip checkt checks
echo ip flusht flushs echo ip stopt stop( R t addt gett actionstartt actionbant actionunbant actioncheckt actionflusht
actionstop( R t ot act( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt
defaultAction2 s
c C s- | j j d � | j t | j j d � d S( Nt test( R R t assertRaisest
ValueError( R ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt testActionsAddDuplicateName= s c C s� | j j d � | j | j d � | j d | j � | j d | j � | j j d � | j d =| j d =| j d | j � | j t | j � d � | j j d � | j | j j � d � | j t
| j j d � d S( NR$ s nonexistant actiont test1i i s 127.0.0.1( R R t
assertTruet assertInt assertNotInt assertEqualt lent
setBanTimet
getBanTimeR% R& t removeBannedIP( R ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt testActionsManipulationA s
c C s� | j | j j d � d � | j d � | j � | j | j j d d d g � d � | j d � | j d � | j d � | j d � d S(
Ns 192.0.2.1i s
Ban 192.0.2.1s 192.0.2.2s 192.0.2.3i s 192.0.2.1 already banneds
Ban 192.0.2.2s
Ban 192.0.2.3( R, R t addBannedIPt assertLoggedt pruneLogt assertNotLogged( R ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt testAddBannedIPP s
%
c C s� | j � | j j � | j d d t �| j j � | j j � | j d d
� | j | j j � d d d g f g � | j | j j d
� d d g � d S( Ns
stdout: %rs ip startt waits ip flushs ip stops Currently bannedi s Total banneds Banned IP listt shorts stdout: 'ip start's stdout: 'ip flush's stdout: 'ip stop'( s Currently bannedi ( s Total bannedi ( s Currently bannedi ( s Total bannedi ( R# R R R3 t TrueR t joinR, t status( R ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt testActionsOutputZ s
c sV � j j d t j j t d � i d d 6� � j d � � j j � � j t j
� f d � d � � � j j � � j j � � j d � � j t
� j j d d
i � � j j d t j j t d � i d d 6d d
6� � j t � j j d t j j t d � i d d 6d d
6d d 6� � j t � j j d t j j t d � i � d S( Nt Actions action.d/action.pyt valuet opt1s TestAction initialisedc s
� j d � S( Ns TestAction action start( t
_is_logged( ( R ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt <lambda>n s i s TestAction action stopt Action3s /does/not/exist.pyt Action4t value2t opt2t Action5t value3t opt3( R R t ost pathR: t TEST_FILES_DIRR3 R R) R t wait_forR R% t IOErrort TypeError( R ( ( R sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt testAddActionPythonf s*
"
c s� � j t � j j d t j j t d � i � � j t � j j d t j j t d � i � � j j d t j j t d � i � � j j � � j t
j � f d � d � � � j j � � j j � � j
d � d S( NR= s action.d/action_noAction.pys action.d/action_nomethod.pys action.d/action_errors.pyc s
� j d � S( Ns Failed to start( R@ ( ( R ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyRA � s i s Failed to stop( R% t RuntimeErrorR R RI RJ R: RK R R) R RL R R3 ( R ( ( R sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt testAddPythonActionNOK� s"
"
c C s� | j j d t j j t d � i � | j j d t j j t d � i � | j j t d � � | j j � | j
d � | j d � | j d � | j j � | j
d � | j d � | j d
� d S( Nt action1s action.d/action_modifyainfo.pyt action2s 1.2.3.4s Failed to execute bans action1 ban deleted aInfo IPs action2 ban deleted aInfo IPs Failed to execute unbans action1 unban deleted aInfo IPs action2 unban deleted aInfo IP(
R R RI RJ R: RK R t
putFailTicketR t _Actions__checkBanR5 R3 t _Actions__flushBan( R ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt testBanActionsAInfo� s"
c C s3 d | j _ d | j _ | j j d � | j j � t j d � d } x= | d k r� d | } | j j t | d � � | d 7} qK W| j
d d t �t j d
� x= | d k r� d | } | j j t | d
� � | d 7} q� W| j
d d t �| j j � | j j
� | j
d
� | j d � d S( Ni i id i i s
192.0.2.%di s / 20,R7 i� i2 s / 50,s Unbanned 30, 0 ticket(s)s Unbanned 50, 0 ticket(s)( R t
banPrecedencet
unbanMaxCountR. R R t setTimeR RT R R3 R9 R R: R5 ( R t iR ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt testUnbanOnBusyBanBombing� s*
c C s� | j i d d 6d d 6� } | j d | d <t | _ | j j � | j d$ � | j | j j d � d � | j | j j d
� d � | j d d d% d d
d d d t d t �| j
d � | j d | d <| j j d
� | j d d t �| j
d � | j j
� | j d d d d d d d d& d d t d t �| j
d � | j | d <| j | j j d
� d � | j d d' d d d t d t �| j d d d t �| j
d � | j j
� | j d d d d d d d t d t �| j d d d( d) d d t �| j
d � | j | d <| j j � | j j � | j d d d* d! d t d t �| j d" d d d# d t �d S(+ Ns <family>R R s ; exit 1s actionflush?family=inet6s
stdout: %rs ip starts 192.0.2.1i s 2001:db8::1s
Ban 192.0.2.1s Ban 2001:db8::1s ip ban 192.0.2.1s ip ban 2001:db8::1t allR7 s> [test-phase 1a] simulate inconsistent irreparable env by unbans actioncheck?family=inet6s, Invariant check failed. Unban is impossible.s> [test-phase 1b] simulate inconsistent irreparable env by flushs ip flush inet4s ip flush inet6s Failed to flush banss' No flush occurred, do consistency checks< Invariant check failed. Trying to restore a sane environments ip stops [test-phase 2] consistent envs ip check inet4s- [test-phase 3] failed flush in consistent envs ip check inet6s Unable to restore environments! [test-phase end] flush successfuls action ip terminatedt ERRORs Unban tickets each individualys stdout: 'ip start's stdout: 'ip start's stdout: 'ip stop's stdout: 'ip start's stdout: 'ip stop's stdout: 'ip start's stdout: 'ip stop'( R# R R9 t actionstart_on_demandR R R5 R, R2 R3 R4 R R0 RV R R: ( R R" ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt testActionsConsistencyCheck� s~
c C s� | j i d d 6d d 6d d 6d d 6� } | j d | d <t | _ t | _ | j j � | j d, � | j | j j d
� d � | j | j j d � d � | j
d
d d d d d d d d d d t d t �| j d | d <| j d � | j j
d � | j
d d d d t d t �| j d d d d d d d t �| j d � | j | j j d � d � | j
d d d d d d d d d d t d t �| j d d d d d t �| j | d <| j | j j d � d � | j d | d <| j d � | j j � | j
d d d d! d" d# d d d d$ d t d t �| j d d d t �| j d% � | j | d <| j | j j d � d � | j
d d d d d d t d t �| j d d& d d d t �| j d' � | j | d <| j j � | j
d" d# d d! d d d t d t �| j d d d d d d d d d d d t �| j d( � | j | d <| j j � | j j � | j
d d! d d d d d) d t d t �| j d* d d d+ d t �d S(- Ns <family>R R R R s ; exit 1s actionflush?family=inet6s
stdout: %rs ip starts 192.0.2.1i s 2001:db8::1s
Ban 192.0.2.1s Ban 2001:db8::1s ip start inet4s ip ban 192.0.2.1s ip start inet6s ip ban 2001:db8::1R] R7 s actioncheck?family=inet6s> [test-phase 1a] simulate inconsistent irreparable env by unbans< Invariant check failed. Trying to restore a sane environments
ip stop inet6s
ip stop inet4s< [test-phase 1b] simulate inconsistent irreparable env by bans ip check inet6s Unable to restore environments Failed to execute bans 2001:db8::2s> [test-phase 1c] simulate inconsistent irreparable env by flushs ip flush inet4s ip flush inet6s Failed to flush banss' No flush occurred, do consistency checks Failed to flush bans in jails [test-phase 2] consistent envs ip check inet4s- [test-phase 3] failed flush in consistent envs! [test-phase end] flush successfuls action ip terminatedR^ s Unban tickets each individualys stdout: 'ip start'( R# R R9 R_ t actionrepair_on_unbanR R R5 R, R2 R3 R R4 R0 RV R R: ( R R" ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt" testActionsConsistencyCheckDiffFam" s� +
c C s� | j i d d 6d d 6d d 6d d 6d d
6� } | d | d <t | _ d
| _ d | _ | j d � | j | j j d d g � d � | j d d d d d d d d d d d t �| j d � | j
| d � | j d � | j d � | j � | j
| d � | j d � | j d � t j
t j � d � | j d! � | j | j j d d d" g � d# � | j d$ d% d& d d' d d( d t �| j
| d � | j
| d � | j d) d* d d d d+ d d d d, d- d. d t �t j
t j � d � t j | d � t j | d � | j d/ � | j | j j d d g � d � | j d0 d d1 d d2 d3 d4 d d+ d d, d t �| j d5 � | j | j j d" � d# � | j d6 d d7 d t �| j d0 d d1 d d2 d3 d4 d d+ d d, d t �| j d8 � | j
| d � | j d9 � | j d: � | j � | j
| d � | j d: d; d t �| j d9 � d< | _ d= | _ | j | j j t d d> � � d> � | j d? d@ d t �d S(A Ns <family>; touch "<FN>"R s <family>; test -f "<FN>"R s <family>; echo -n "" > "<FN>"R s <family>; rm -f "<FN>"R s) <family>; echo "<ip> <family>" >> "<FN>"R s /<family>t FNs% echo ip repair <family>; touch "<FN>"sG echo ip reban <ip> <family>; echo "<ip> <family> -- rebanned" >> "<FN>"s [test-phase 0] initial bans 192.0.2.1s 2001:db8::1i s
Ban 192.0.2.1s Ban 2001:db8::1s
stdout: %rs ip start inet4s ip ban 192.0.2.1 inet4s ip start inet6s ip ban 2001:db8::1 inet6R] s [test-phase 1] check bans /inet4s 192.0.2.1 inet4s 2001:db8::1 inet6s /inet6i s# [test-phase 2] check already banneds 2001:db8::2i s 192.0.2.1 already banneds 2001:db8::1 already banneds Ban 2001:db8::2s ip check inet4s ip check inet6s Reban 192.0.2.1s Reban 2001:db8::1s ip reban 192.0.2.1 inet4s ip reban 2001:db8::1 inet6s 192.0.2.1 inet4 -- repaireds 2001:db8::1 inet6 -- repaireds3 [test-phase 3a] check reban after sane env repaireds< Invariant check failed. Trying to restore a sane environments ip repair inet4s ip repair inet6s Reban 192.0.2.1, action 'ip's Reban 2001:db8::1, action 'ip's> [test-phase 3a] check reban by epoch mismatch (without repair)s Reban 2001:db8::2, action 'ip's ip reban 2001:db8::2 inet6s [test-phase 4] check rebans 192.0.2.1 inet4 -- rebanneds 2001:db8::1 inet6 -- rebanneds 2001:db8::2 inet6 -- rebannedR s exit 1i s Failed to execute rebans Error banning 192.0.2.1( R# R9 R_ t actionrepairt actionrebanR4 R, R R2 R3 t dumpFileR5 R RZ t timeRI t removeR t _Actions__reBanR ( R t tmpR" ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt! testActionsRebanBrokenAfterRepair� s�
"
%
"
%( t __name__t
__module__R R R# R' R1 R6 R< RO RQ RW R R\ R` Rb R Rk ( ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyR ' s
& K o( t
__author__t
__copyright__t __license__Rg RI t tempfilet
server.ticketR t server.utilsR t dummyjailR t utilsR R R R RJ R: t dirnamet __file__RK R ( ( ( sB /usr/lib/python2.7/site-packages/fail2ban/tests/actionstestcase.pyt <module> s "!
Mini Shell