a
=Æ*f°D ã @ sÀ d Z dZdZddlZddlmZ ddlmZ ddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZdd lmZ dd
lmZ dd lmZ eeƒZG d d
„ d
eƒZG dd„ dƒZeƒ ZdS )zSerg G. Brester (sebres)z"Copyright (c) 2014 Serg G. BresterZGPLé Né )Ú
JailThread)ÚFailManagerEmptyé )Ú getLogger)ÚMyTime)ÚUtilsc sN e Zd ZdZejd Z‡ fdd„Zdd„ Zdd„ Zd d
„ Z d d „ Z
d
d„ Z dd„ Z dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd d „ Zd d „ Zd!d"„ ZdGd$d%„Z‡ fd&d'„ZdHd*d+„Zed,d-„ ƒZdId.d/„ZdJd0d1„Z ed2d3„ ƒZ e j d4d3„ ƒZ ed5d6„ ƒZ d7d8„ Z d9d:„ Z!d;d<„ Z"G d=d>„ d>ƒZ#d?d@„ Z$dAdB„ Z%dCdD„ Z&dEdF„ Z'‡ Z(S )KÚObserverThreada8 Handles observing a database, managing bad ips and ban increment.
Parameters
----------
Attributes
----------
daemon
ident
name
status
active : bool
Control the state of the thread.
idle : bool
Control the idle state of the thread.
sleeptime : int
The time the thread sleeps for in the loop.
é
c sZ t t| ƒjdd d| _t ¡ | _g | _t ¡ | _ d| _
i | _ d| _ d | _
d| _d| _d S )Nz f2b/observer)ÚnameTé< Fi )Úsuperr Ú__init__ÚidleÚ threadingÚRLockÚ _queue_lockÚ_queueÚEventÚ_notifyÚ sleeptimeÚ_timersÚ_pausedÚ_ObserverThread__dbÚ"_ObserverThread__db_purge_intervalÚdaemon©Úself©Ú __class__© ú | j | j |¡ W d ƒ n1 s(0 Y | ¡ dS )z5Add a event to queue and notify thread to wake up.
N)r r Úappendr@ ©r r7 r r r! r4 › s *zObserverThread.addc G s6 | j | j |¡ W d ƒ n1 s(0 Y dS )zÆ ó z$ObserverThread.run..c S s dS rI r r r r r! rJ Ç rK ) ÚcallÚdb_setrH Úis_aliveZ is_activer5 ÚstopÚnopÚshutdownrN FNr r ú%sT©Úexc_infoz&Observer stopped, %s events remaining.z Observer stopped after error: %s)!ÚlogSysÚinfor9 r rF rM rH ÚisAliveÚisActiver5 rO r4 Úactiver r r r+ r ÚpopÚcallabler1 ÚgetattrÚ ExceptionÚerrorr Úwaitr Úclearr< Úsleepr r: Úis_full)r Z_ObserverThread__methZevÚmethÚer? r r r! Úrun s^
ô
* (
&&zObserverThread.runc C s dS )NTr r r r r! rW ý s zObserverThread.isAliveNc C s | j S r( )rY )r ZfromStrr r r! rX s zObserverThread.isActivec s> | j $ | js tt| ƒ ¡ W d ƒ n1 s00 Y d S r( )r rY r
r r5 r r r r! r5 s zObserverThread.starté Tc C sª | j r¦| jr¦t d|¡ | j0 | d¡ | j}| j ¡ d | _W d ƒ n1 sT0 Y | |¡sl|rˆ| ¡ d| _ d| _ d | _
n|| _| t |dƒ¡o¤| j
S dS )Nz-Observer stop ... try to end queue %s secondsrQ FTg à?)rY r rU rV r rC r> Ú
wait_emptyr` r r Ú wait_idleÚminrb )r ZwtimeZ forceQuitr? r r r! rO s
$zObserverThread.stopc C s< | j " t| jƒrdndW d ƒ S 1 s.0 Y d S )NTF)r r+ r r r r r! rb # s zObserverThread.is_fullc C s„ t tj¡ |dur t ¡ | }| jdurH| d¡ | jrH| jrH| ¡ | jrr|durdt ¡ |krdqrt tj¡ qH|
d¡ | j S )zWWait observer is running and returns if observer has no more events (queue is empty)
NrP gü©ñÒMbP?) r< ra r r: r r rC rb r r@ rh ©r r rd r r r! rg ( s
zObserverThread.wait_emptyc C sZ t tj¡ | jrdS |dur*t ¡ | }| jsT|durFt ¡ |krFqTt tj¡ q*| jS )zJWait observer is running and returns if observer idle (observer sleeps)
TN)r< ra r r: r r rj r r r! rh < s zObserverThread.wait_idlec C s | j S r( )r r r r r! ÚpausedJ s zObserverThread.pausedc C s | j |krd S || _ | ¡ d S r( )r r@ )r Úpauser r r! rk N s
c C s dS )z/Status of observer to be implemented. [TODO]
)Ú rm r r r r r! ÚstatusW s zObserverThread.statusc C s
|| _ d S r( )r )r Údbr r r! rM a s zObserverThread.db_setc C s2 t d¡ | jd ur | j ¡ | d| jd¡ d S )Nz Purge database event occurredrG rH )rU Údebugr Úpurger9 r r r r r! rH d s
zObserverThread.db_purgec
C s’ | ¡ r| d¡sdS | ¡ }| ¡ }t d|j|¡ d}d}d}z|jj ¡ }|j
} | durÚ| ||¡D ]8\}}}
t ||
¡ ƒ}d|dk r’|nd> d d } q¦qlt||ƒ}|durÚ||krÚt d|j|||¡ W dS |dkrèW dS t d |j|t |¡||||krd
nd ¡ |jj ||d d ¡}| |¡ ||krJ|j |¡ W n@ tyŒ } z&tjd
| t ¡ tjkd W Y d} ~ n
d} ~ 0 0 dS )z} Notify observer a failure for ip was found
Observer will check ip was known (bad) and possibly increase an retry count
Ú incrementNz [%s] Observer: failure found %sr r é r z8[%s] Ignore failure %s before last ban %s < %s, restoredz%[%s] Found %s, bad - %s, %s # -> %s%sz, Banrm TrR rS )rW ÚgetBanTimeExtraÚgetIDÚgetTimerU rp r ÚfilterZ failManagerZ getMaxRetryÚdatabaseÚgetBanÚmaxÚ getBanCountri rV r Útime2strZ
addFailureÚ setBanCountZ
performBanr] r^ ÚgetEffectiveLevelÚloggingÚDEBUG) r ÚjailÚticketÚipZunixTimeÚbanCountZ
retryCountÚ timeOfBanZmaxRetryro Ú lastBanTimerd r r r! Ú failureFoundo sD
ÿ þ
zObserverThread.failureFoundc @ s e Zd Zdd„ ZdS )zObserverThread.BanTimeIncrc C s || _ || _d S r( )ZTimeZCount)r ÚbanTimer„ r r r! r ¢ s z#ObserverThread.BanTimeIncr.__init__N©Ú__name__Ú
__module__Ú __qualname__r r r r r! Ú BanTimeIncr¡ s r c C s | ¡ }|d | ||¡ƒS )NÚ evformula)rt r )r r rˆ r„ Úber r r! Ú calcBanTime¦ s zObserverThread.calcBanTimec C s@ | ¡ r|js|S | ¡ }| ¡ }|}zÔ|dkrø| dd¡rø|jj||| dd¡dD ] \}}} || ¡ krz| |d ¡ t d|||¡ |dkr¦|d |
||¡ƒ}| |¡ | ¡ |krìt
d |j||t |¡t |¡t |¡f ¡ nd
|_ qøqVW n@ ty: }
z&tjd |
t ¡ tjkd W Y d
}
~
n
d
}
~
0 0 |S )ztCheck for IP address to increment ban time (if was already banned).
Returns
-------
float
new ban time.
r rr FÚ overalljails)r‘ r z"IP %s was already banned: %s #, %srŽ z/[%s] IP %s is bad: %s # last %s - incr %s to %sTrR rS N)rW rx rt ru r1 ry r{ r} rU rp r Ú
setBanTimerv rV r r r| Ú seconds2strÚrestoredr] r^ r~ r r€ ) r r rˆ r‚ r rƒ Z
orgBanTimer„ r… r† rd r r r! Ú incrBanTimeª s2 ÿ
þ
0zObserverThread.incrBanTimec
C sˆ |j r
dS z6|}| ¡ }t d|j||¡ |dkrh| ¡ du rh| |||¡}|dks^||krh| |¡ |dkr´| ¡ | }t
|¡t |¡f}|t ¡ k r¸t d|d ¡ W dS nd}||kr tj
d|j|| ¡ g|¢R Ž t d d
|j|||f¡ | td td || d ƒƒ| j||¡ |jdur@|j s@|j ||¡ W n@ ty‚ } z&tjd
|t ¡ tjkd W Y d}~n
d}~0 0 dS )úÁ Notify observer a ban occurred for ip
Observer will check ip was known (bad) and possibly increase/prolong a ban time
Secondary we will actualize the bans and bips (bad ip) in database
Nz [%s] Observer: ban found %s, %séÿÿÿÿzIgnore old bantime %sr F)Z permanentZinfinitez$[%s] Increase Ban %s (%d # %s -> %s)rf z [%s] Observer: prolong %s in %sr
r rR rS )r” ru rU rp r Ú
getBanTimer• r’ rv r r“ r| r< Znoticer{ Úlogr= ri rz Ú
prolongBanrx ZaddBanr] r^ r~ r r€ ) r r‚ r ÚbtimeZoldbtimerƒ ZbendtimeZlogtimerd r r r! ÚbanFoundÓ s:
ÿÿ&zObserverThread.banFoundc
C sv z2| ¡ }| ¡ }t d|j||¡ |j |¡ W n> typ } z&tjd|t ¡ t
j kd W Y d}~n
d}~0 0 dS )r– z [%s] Observer: prolong %s, %srR rS N) r˜ ru rU rp r ÚactionsZ _prolongBanr] r^ r~ r r€ )r r‚ r r› rƒ rd r r r! rš ý s zObserverThread.prolongBan)N)rf T)N)N))rŠ r‹ rŒ Ú__doc__r r: r r& r' r* r, r. r0 r9 r= r; r@ r4 rC rF re rW rX r5 rO Úpropertyrb rg rh rk Úsetterrn rM rH r‡ r r r• rœ rš Ú
__classcell__r r r r! r ( sN
P
2)*r c @ s e Zd Zdd„ ZdS )Ú
_Observersc C s
d | _ d S r( )ZMainr r r r! r s z_Observers.__init__Nr‰ r r r r! r¢
s r¢ )Ú
__author__Z
__copyright__Z __license__r Z
jailthreadr Z failmanagerr Úosr r< ZdatetimeZmathZjsonZrandomÚsysZhelpersr Zmytimer Zutilsr rŠ rU r r¢ Z Observersr r r r! Ú s" 8 h