a
=*f* @ sZ d Z dZdZddlmZ ddlmZ ddlmZ dd l m
Z
m Z e
e Z
G d
d d Zd S )
z
Cyril Jaquierz Copyright (c) 2004 Cyril JaquierZGPL )Lock ) BanTicket)MyTime ) getLoggerloggingc @ s e Zd Zdd Zdd Zdd Zdd Zd d
Zd)d d
Zdd Z e
dd Z d*ddZ dd Z
dd Zdd Zi fdd Zd d Zd d Zd+d"d#Zd$d% Zd&d' Zd(S ),
BanManagerc C s( t | _t | _d| _d| _tj| _d S )NiX r ) r _BanManager__lockdict_BanManager__banList_BanManager__banTime_BanManager__banTotalr MAX_TIME_nextUnbanTimeself r >/usr/lib/python3.9/site-packages/fail2ban/server/banmanager.py__init__2 s
zBanManager.__init__c C s t || _d S N)intr
r valuer r r
setBanTimeD s zBanManager.setBanTimec C s | j S r )r
r r r r
getBanTimeM s zBanManager.getBanTimec C s
|| _ d S r r r r r r setBanTotalU s zBanManager.setBanTotalc C s | j S r r r r r r getBanTotal] s zBanManager.getBanTotalFc s |st j S j> g } j D ] }| j}|||f q(W d n1 s\0 Y |jdd d t j
|r fdd|D S dd |D S )Nc S s | d S )Nr r )tr r r m z'BanManager.getBanList..)keyc sD g | ]<}d |d |d |d j|d f qS )z%s %s + %d = %sr r )getIDgetTimer r
.0r r Zt2sr r
p s
&z)BanManager.getBanList..c S s g | ]}|d qS )r )r# r% r r r r( t r! ) listr keysr
valuesgetEndOfBanTimer
appendsortr Ztime2str)r ZorderedZwithTimeZlstticketeobr r' r
getBanListe s . zBanManager.getBanListc C s t t| j S r )iterr) r r+ r r r r __iter__{ s zBanManager.__iter__c C s | d u st | dkrdS | S d S )Nr unknown)len)r r r r handleBlankResult s z BanManager.handleBlankResult
c C s g g g d}t | dsz4dd ladd latj }||_|d |_|| _W nh t y } zPt
d t ||d< |d
d |d
d |d
d |W Y d }~S d }~0 0 | j( d
d t| j D }W d n1 s0 Y zF|D ]:}||jrd nd
}z| j}||d}|s8tdt } t }
t } |D ]v} dd | dD \}
}}}}| |
}
| |}| |}| | |
|
| | | | | qN|d
dt | |d
dt |
|d
dt | W q tjj yT |d
d |d
d |d
d Y n tj j tjj tj j!fy } zNt
d|| t
j"t#j$krt
| t ||d< W Y d }~ q8W Y d }~npd }~0 t%y2 } zNt
d|| t
j"t#j$krt
| t ||d< W Y d }~ q8W Y d }~n
d }~0 0 qW nV t%y } z.zorigin.asn.cymru.comzorigin6.asn.cymru.comZTXTzNo data retrievedc S s g | ]}| d qS )z'" )strip)r& Zanswerr r r r( r! |z, Znxdomainz)DNSException %r querying Cymru for %s TXTz0Unhandled Exception %r querying Cymru for %s TXTz*Failure looking up extended Cymru info: %s)&hasattrZ
dns.exceptionZdnsZ dns.resolverresolverZResolverZlifetimetimeoutr; ImportErrorlogSysr< reprr- r
r) r r+ ZgetPTRZisIPv4query
ValueErrorsetZto_textsplitr6 addjoinsortedZNXDOMAIN exceptionZ DNSExceptionZ
NoNameserversZTimeoutlevelr DEBUG Exception)r rA Z return_dictr@ eZbanIPsipZquestionZanswersZasnsZ countriesZrirsZrdatar8 netr9 r: ZchangedZdnser r r getBanListExtendedCymruInfo s
6
"
$
.
"z&BanManager.getBanListExtendedCymruInfoc
C sV zdd |d D W S t yP } z$td t| g W Y d }~S d }~0 0 d S )Nc S s g | ]}|qS r r )r& r8 r r r r( r! z3BanManager.geBanListExtendedASN..r8 zFailed to lookup ASNrO rC r< rL r Z
cymru_inforP r r r geBanListExtendedASN s
z BanManager.geBanListExtendedASNc
C sV zdd |d D W S t yP } z$td t| g W Y d }~S d }~0 0 d S )Nc S s g | ]}|qS r r )r& r9 r r r r( r! z7BanManager.geBanListExtendedCountry..r9 zFailed to lookup CountryrT rU r r r geBanListExtendedCountry s
z#BanManager.geBanListExtendedCountryc
C sV zdd |d D W S t yP } z$td t| g W Y d }~S d }~0 0 d S )Nc S s g | ]}|qS r r )r& r: r r r r( r! z3BanManager.geBanListExtendedRIR..r: zFailed to lookup RIRrT rU r r r geBanListExtendedRIR s
z BanManager.geBanListExtendedRIRc C s | | j}|t k r$d|d< dS | j | }| j|}|r||d< || | jkrd|d< || j}|dkr| | }|dkr||7 }|
| W d dS || j|< | j d7 _ | | j
|kr|| _
W d dS 1 s0 Y d S ) Nr ZexpiredFr/ Zprolongr T)r, r
r timer
r# r getr r$ r r Z incrBanCountr )r r/ reasonr0 fidZ oldticketZbtmZdiftmr r r addBanTicket s0
zBanManager.addBanTicketc C s
t | jS r )r5 r r r r r size1 s zBanManager.sizec C s | | jv S r )r# r )r r/ r r r
_inBanList< s zBanManager._inBanListc s | j | j}||kr,t W d S i tj}| j D ]H\}}|| j}||kr|| |< t |kr| j} qq@||kr@|}q@|| _t rt d t | jd krʈ
D ] }| j|= qn t fdd| j D | _t W d S 1 s
0 Y d S )Ng @g @c 3 s" | ]\}}| vr||fV qd S r r )r& r] r/ unBanListr r d s
z'BanManager.unBanList..)
r
r r) r r r itemsr, r
r5 r* r r+ )r rZ ZmaxCountZ
nextUnbanTimer] r/ r0 r rb r rc F s,
zBanManager.unBanListc C sD | j * t| j }t | _|W d S 1 s60 Y d S r )r
r) r r+ r )r ZuBListr r r flushBanListp s zBanManager.flushBanListc C sZ | j @ z | j|W W d S ty6 Y n0 W d n1 sL0 Y d S r )r
r popKeyError)r r] r r r
getTicketByIDz s $zBanManager.getTicketByIDN)FF)r7 )ra )__name__
__module__ __qualname__r r r r r r1 r3 staticmethodr6 rS rV rW rX r^ r_ r` rc rf ri r r r r r + s&
O %
*
r N)
__author__Z
__copyright__Z __license__ threadingr r/ r Zmytimer Zhelpersr r rj rC r r r r r s