File: //usr/lib/python3.9/site-packages/fail2ban/server/__pycache__/banmanager.cpython-39.opt-2.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 JaquierZGPL� )�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 ZorderedZwithTimeZlst�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�errorr8 r9 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+ ZgetPTRZisIPv4Zquery�
ValueError�setZto_text�splitr6 �add�join�sortedZNXDOMAIN� exceptionZ DNSExceptionZ
NoNameserversZTimeout�levelr �DEBUG� Exception)r rA Z return_dictr@ �eZbanIPsZipZquestionZanswersZasnsZ countriesZrirsZrdatar8 Znetr9 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& r8 r r r r( � r! z3BanManager.geBanListExtendedASN.<locals>.<listcomp>r8 zFailed to lookup ASN�rN rC r< rK �r Z
cymru_inforO 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& r9 r r r r( � r! z7BanManager.geBanListExtendedCountry.<locals>.<listcomp>r9 zFailed to lookup CountryrR rS 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 RIRrR rS 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 ZexpiredFr/ Zprolong���r 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 | j�S 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._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[ r/ �� unBanListr r � <genexpr>d s
�z'BanManager.unBanList.<locals>.<genexpr>)
r
r r) r r r �itemsr, r
r5 r* r r+ )r rX ZmaxCountZ
nextUnbanTimer[ r/ r0 r r` r ra 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)r7 )r_ )�__name__�
__module__� __qualname__r r r r r r1 r3 � staticmethodr6 rQ rT rU rV r\ r] r^ ra rd rg r r r r r + s&
O %
*
r N)�
__author__Z
__copyright__Z __license__� threadingr r/ r Zmytimer Zhelpersr r rh rC r r r r r �<module> s