File: //usr/lib/python3.9/site-packages/fail2ban/server/__pycache__/banmanager.cpython-39.pyc
a
g� B1`b�� @ 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 Jaquier�GPL� )�Lock� )� BanTicket)�MyTime� )� getLogger�loggingc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd)d d
�Zdd� Z e
dd� �Z d*dd�Z 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__lock�dict�_BanManager__banList�_BanManager__banTime�_BanManager__banTotalr �MAX_TIME�_nextUnbanTime��self� 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 �<lambda>m � z'BanManager.getBanList.<locals>.<lambda>)�keyc sD g | ]<}d |d � � �|d �� �|d �� j��|d �f �qS )z%s %s + %d = %sr r )�getID�getTimer r ��.0r �r Zt2sr r �
<listcomp>p s
�
&�z)BanManager.getBanList.<locals>.<listcomp>c S s g | ]}|d � � �qS )r )r$ r& r r r r) t r" ) �listr
�keysr �values�getEndOfBanTimer �append�sortr Ztime2str)r ZorderedZwithTime�lst�ticket�eobr 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 | d�s�z4dd 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 s�0 Y �zF|D �]:}|�|j�rd nd
�}z�| j}|�|d�}|�s8td��t� } t� }
t� } |D ]v} dd � | �� �d�D �\}
}}}}| �|
�}
| �|�}| �|�}| �| �|
�� |
�| �|�� | �| �|�� �qN|d �
d�t | ��� |d �
d�t |
��� |d �
d�t | ��� W � q� tjj �yT |d �
d� |d �
d� |d �
d� Y n� tj j tjj tj j!f�y� } zNt
� d||� t
j"t#j$k�r�t
� |� t |�|d< W Y d }~ �q8W Y d }~npd }~0 t%�y2 } zNt
� d||� t
j"t#j$k�rt
� |� t |�|d< W Y d }~ �q8W Y d }~n
d }~0 0 � q�W nV t%�y� } z<t
� d|� t
j"t#j$k�rpt
� |� t |�|d< W Y d }~n
d }~0 0 |S )N)�asn�country�rir� dnsResolverr r z7dnspython package is required but could not be imported�errorr: r; r< c S s g | ] }|� � �qS r )ZgetIP)r' ZbanDatar r r r) � r" z:BanManager.getBanListExtendedCymruInfo.<locals>.<listcomp>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.resolver�resolverZResolverZlifetime�timeoutr= � ImportError�logSysr> �reprr. r r* r
r, ZgetPTRZisIPv4�query�
ValueError�setZto_text�splitr8 �add�join�sortedZNXDOMAIN� exceptionZ DNSExceptionZ
NoNameserversZTimeout�levelr �DEBUG� Exception)r rC Z return_dictrB �eZbanIPs�ipZquestionZanswersZasnsZ countriesZrirsZrdatar: �netr; r< �changedZdnser r r �getBanListExtendedCymruInfo� s�
6
� �
"
$
.
"z&BanManager.getBanListExtendedCymruInfoc
C sV zdd� |d D �W S t yP } z$t�d� 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.geBanListExtendedASN.<locals>.<listcomp>r: zFailed to lookup ASN�rQ rE r> rN �r Z
cymru_inforR r r r �geBanListExtendedASN� s
z BanManager.geBanListExtendedASNc
C sV zdd� |d D �W S t yP } z$t�d� 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" z7BanManager.geBanListExtendedCountry.<locals>.<listcomp>r; zFailed to lookup CountryrW rX r r r �geBanListExtendedCountry� s
z#BanManager.geBanListExtendedCountryc
C sV zdd� |d D �W S t yP } z$t�d� 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.<locals>.<listcomp>r< zFailed to lookup RIRrW rX r r r �geBanListExtendedRIR� s
z BanManager.geBanListExtendedRIRc C s |� | j�}|t�� k r$d|d< dS | j�� |�� }| j�|�}|r�||d< ||� | j�kr�d|d< |�| j�}|dkr�|� � |� � }|dkr�||7 }|�
|� W d � dS || j|< | j d7 _ |� � | j
|kr�|| _
W d � dS 1 � s�0 Y d S ) Nr ZexpiredFr1 Zprolong���r T)r- r r �timer r$ r
�getr r% r r Z incrBanCountr )r r1 �reasonr2 �fidZ oldticketZbtmZdiftmr r r � addBanTicket s0
zBanManager.addBanTicketc C s
t | j�S r )r7 r
r r r r �size1 s zBanManager.sizec C s |� � | jv S r )r$ r
)r r1 r r r �
_inBanList< s zBanManager._inBanList��c s | j � �� | j}||kr,t� W d � S i � tj}| j�� D ]H\}}|�| j�}||kr||� |< t � �|kr�| j} q�q@||kr@|}q@|| _t � �r�t � �d t | j�d krʈ �
� D ] }| j|= q�n 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` r1 �� unBanListr r � <genexpr>d s
�z'BanManager.unBanList.<locals>.<genexpr>)
r r r* r r r
�itemsr- r r7 r+ r r, )r r] ZmaxCountZ
nextUnbanTimer` r1 r2 r re r rf 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
�pop�KeyError)r r` r r r �
getTicketByIDz s $zBanManager.getTicketByIDN)FF)r9 )rd )�__name__�
__module__� __qualname__r r r r r r3 r5 � staticmethodr8 rV rY rZ r[ ra rb rc rf ri rl r r r r r
+ s&
O %
*
r
N)�
__author__�
__copyright__� __license__� threadingr r1 r Zmytimer Zhelpersr r rm rE r
r r r r �<module> s