File: //usr/lib/python3/dist-packages/fail2ban/server/__pycache__/banmanager.cpython-310.pyc
o
;s*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/dist-packages/fail2ban/server/banmanager.py�__init__2 s
zBanManager.__init__c C s t |�| _d S �N)�intr �r �valuer r r �
setBanTimeD � zBanManager.setBanTimec C � | j S r )r r r r r �
getBanTimeM � zBanManager.getBanTimec C s
|| _ d S r �r r r r r � setBanTotalU �
zBanManager.setBanTotalc C r r r r r r r � getBanTotal] r zBanManager.getBanTotalFc s� |s t � j�� �S � j� g }� j�� D ]}|�� j�}|�||f� qW d � n1 s.w Y |jdd� d� t j
�|rJ� �fdd�|D �S dd� |D �S )Nc S s | d S )Nr r )�tr r r �<lambda>m s 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 �t2sr r �
<listcomp>p s �
&
�z)BanManager.getBanList.<locals>.<listcomp>c S s g | ]}|d � � �qS )r )r' r) r r r r- t s ) �listr
�keysr �values�getEndOfBanTimer �append�sortr �time2str)r �ordered�withTime�lst�ticket�eobr r+ r �
getBanListe s � � �zBanManager.getBanListc C s t t| j�� ��S r )�iterr. r
r0 r r r r �__iter__{ s zBanManager.__iter__c C s | d u s
t | �dkr dS | S )Nr �unknown)�len)r r r r �handleBlankResult� s z BanManager.handleBlankResult�
c C sv g g g d�}t | d�sZzdd ladd latj�� }||_|d |_|| _W n4 t yY } z(t
� d� t |�|d< |d �
d� |d �
d� |d �
d� |W Y d }~S d }~ww | j� d
d � t| j�� �D �}W d � n1 stw Y �z|D �]
}|�|jr�d nd
�}z{| j}|�|d�}|s�td��t� } t� }
t� } |D ]:} dd � | �� �d�D �\}
}}}}| �|
�}
| �|�}| �|�}| �| �|
�� |
�| �|�� | �| �|�� q�|d �
d�t | ��� |d �
d�t |
��� |d �
d�t | ��� W q} tjj �y& |d �
d� |d �
d� |d �
d� Y q} tj j tjj tj j!f�y] } z"t
� d||� t
j"t#j$k�rJt
� |� t |�|d< W Y d }~ W |S d }~w t%�y� } z"t
� d||� t
j"t#j$k�rxt
� |� t |�|d< W Y d }~ W |S d }~ww W |S t%�y� } z t
� d|� t
j"t#j$k�r�t
� |� t |�|d< W Y d }~|S d }~ww )N)�asn�country�rir� dnsResolverr r z7dnspython package is required but could not be imported�errorrA rB rC c S s g | ]}|� � �qS r )�getIP)r* �banDatar r r r- � s z:BanManager.getBanListExtendedCymruInfo.<locals>.<listcomp>zorigin.asn.cymru.comzorigin6.asn.cymru.com�TXTzNo data retrievedc S s g | ]}|� d ��qS )z'" )�strip)r* �answerr r r r- � s �|z, �nxdomainz)DNSException %r querying Cymru for %s TXTz0Unhandled Exception %r querying Cymru for %s TXTz*Failure looking up extended Cymru info: %s)&�hasattr�
dns.exception�dns� dns.resolver�resolver�Resolver�lifetime�timeoutrD � ImportError�logSysrE �reprr2 r r. r
r0 �getPTR�isIPv4�query�
ValueError�set�to_text�splitr? �add�join�sorted�NXDOMAIN� exception� DNSException�
NoNameservers�Timeout�levelr �DEBUG� Exception)r rT � return_dictrQ �e�banIPs�ip�question�answers�asns� countries�rirs�rdatarA �netrB rC �changed�dnser r r �getBanListExtendedCymruInfo� s�
�� �
� �
"
��
���/�
��z&BanManager.getBanListExtendedCymruInfoc
C �R z
dd� |d D �W S t y( } zt�d� t�|� g W Y d }~S d }~ww )Nc S � g | ]}|�qS r r )r* rA r r r r- � � z3BanManager.geBanListExtendedASN.<locals>.<listcomp>rA zFailed to lookup ASN�ri rV rE rc �r �
cymru_infork r r r �geBanListExtendedASN� �
��z BanManager.geBanListExtendedASNc
C rx )Nc S ry r r )r* rB r r r r- � rz z7BanManager.geBanListExtendedCountry.<locals>.<listcomp>rB zFailed to lookup Countryr{ r| r r r �geBanListExtendedCountry� r z#BanManager.geBanListExtendedCountryc
C rx )Nc S ry r r )r* rC r r r r- � rz z3BanManager.geBanListExtendedRIR.<locals>.<listcomp>rC zFailed to lookup RIRr{ r| r r r �geBanListExtendedRIR� r z BanManager.geBanListExtendedRIRc C s
|� | j�}|t�� k rd|d< dS | j�f |�� }| j�|�}|rZ||d< ||� | j�krQd|d< |�| j�}|dkrL|� � |� � }|dkrL||7 }|�
|� W d � dS || j|< | j d7 _ |� � | j
|krr|| _
W d � dS 1 s~w Y d S ) Nr �expiredFr8 �prolong���r T)r1 r r �timer r' r
�getr r( r r � incrBanCountr )r r8 �reasonr9 �fid� oldticket�btm�diftmr r r � addBanTicket s4
�
$�zBanManager.addBanTicketc C s
t | j�S r )r> r
r r r r �size1 r" zBanManager.sizec C s |� � | jv S r )r' r
)r r8 r r r �
_inBanList< r zBanManager._inBanList��c s | j �} | j}||krt� W d � S i � tj}| j�� D ]$\}}|�| j�}||kr=|� |< t � �|kr<| j} nq ||krC|}q || _t � �rst � �d t | j�d krd� �
� D ]}| j|= q\nt � fdd�| j�� D ��| _t� � � �W d � S 1 s�w Y d S )Ng @g @c 3 s$ � | ]
\}}|� vr||fV qd S r r )r* r� r8 �� unBanListr r � <genexpr>d s �
��z'BanManager.unBanList.<locals>.<genexpr>)
r r r. r r r
�itemsr1 r r> r/ r r0 )r r� �maxCount�
nextUnbanTimer� r8 r9 r r� r r� F s6 � ��
�
$�zBanManager.unBanListc C sD | j � t| j�� �}t� | _|W d � S 1 sw Y d S r )r r. r
r0 r )r �uBListr r r � flushBanListp s
$�zBanManager.flushBanListc C s\ | j �! z| j�|�W W d � S ty Y nw W d � d S 1 s'w Y d S r )r r
�pop�KeyError)r r� r r r �
getTicketByIDz s � �
��zBanManager.getTicketByIDN)FF)r@ )r� )�__name__�
__module__� __qualname__r r r r! r# r: r< � staticmethodr? rw r~ r� r� r� r� r� r� r� r� r r r r r
+ s(
O %
*
r
N)�
__author__�
__copyright__� __license__� threadingr r8 r �mytimer �helpersr r r� rV r
r r r r �<module> s