File: //usr/lib/python3.9/site-packages/fail2ban/server/__pycache__/ticket.cpython-39.opt-2.pyc
a
C
;%�� @ sl d Z dZdZddlmZ ddlmZ ddlmZ ee �Z
G dd � d e �Z G d
d � d e �Z
G d d
� d
e
�ZdS )z
Cyril Jaquierz Copyright (c) 2004 Cyril JaquierZGPL� )� getLogger� )�IPAddr)�MyTimec @ s: e Zd ZdZdZdZdZdddi dfdd�Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd>d d �Zd?d d �Zd@d!d"�Zd#d$� ZdAd%d&�ZdBd'd(�Zd)d*� Zd+d,� Zd-d.� Zd/d0� Ze d1d2� �Z e j d3d2� �Z e d4d5� �Z e j d6d5� �Z d7d8� Z dCd9d:�Z!e d;d<� �Z"e"j d=d<� �Z"dS )D�Ticket)�_id�_flags� _banCount�_banTime�_time�_data�_retryZ
_lastResetl ��� r � Nc C s� | � |� d| _d| _d| _|dur(|nt�� | _|p8g dd�| _|durn|�� D ]\}}|durR|| j|< qR|r|| � |� dS )z{Ticket constructor
@param ip the IP address
@param time the ban time
@param matches (log) lines caused the ticket
� N)�matches�failures)
�setIDr r r
r �timer r �items�update)�self�ipr r �data�ticket�k�v� r �:/usr/lib/python3.9/site-packages/fail2ban/server/ticket.py�__init__+ s
zTicket.__init__c C s< d| j j�d�d | j| j| j| j| jd | j�dg �f S )Nz@%s: ip=%s time=%s bantime=%s bancount=%s #attempts=%d matches=%r�.���r r ) � __class__�__name__�splitr r r
r r �get�r r r r �__str__B s ��zTicket.__str__c C s t | �S �N)�strr% r r r �__repr__H s zTicket.__repr__c C sL z2| j |j ko0t| jd�t|jd�ko0| j|jkW S tyF Y dS 0 d S )Nr F)r �roundr r �AttributeError)r �otherr r r �__eq__K s �
� z
Ticket.__eq__c C s0 |j D ]$}t||d �}|d urt| ||� qd S r' )� __slots__�getattr�setattr)r r �nr r r r r S s
z
Ticket.updatec C s t |t�rt|�}|| _d S r' )�
isinstancer( r r �r �valuer r r r Y s
z Ticket.setIDc C s | j S r' )r r% r r r �getID_ s z Ticket.getIDc C s | j �d| j�S )Nr )r r$ r r% r r r �getIPb s z Ticket.getIPc C s
|| _ d S r' �r r3 r r r �setTimee s zTicket.setTimec C s | j S r' r7 r% r r r �getTimeh s zTicket.getTimec C s
|| _ d S r' �r
r3 r r r �
setBanTimek s zTicket.setBanTimec C s | j d ur| j S |S r' r: )r � defaultBTr r r �
getBanTimen s zTicket.getBanTimeFc C s |s|| j kr|| _ d S r' �r )r r4 �alwaysr r r � setBanCountq s zTicket.setBanCountc C s | j |7 _ d S r' r> r3 r r r � incrBanCountu s zTicket.incrBanCountc C s | j S r' r> r% r r r � getBanCountx s zTicket.getBanCountc C s, | j d ur| j n|}|dkr"tjS | j| S )Nr )r
r �MAX_TIMEr )r r< �bantimer r r �getEndOfBanTime{ s zTicket.getEndOfBanTimec C s. | j d ur| j n|}|dkr dS || j| kS )Nr F)r
r )r r r< rD r r r �
isTimedOut� s zTicket.isTimedOutc C s || j d< d S �Nr �r r3 r r r �
setAttempt� s zTicket.setAttemptc C s
| j d S rG rH r% r r r �
getAttempt� s zTicket.getAttemptc C s4 |r|| j d< n z | j d= W n ty. Y n0 d S )Nr )r �KeyError)r r r r r �
setMatches� s zTicket.setMatchesc C s dd� | j �dd�D �S )Nc S s( g | ] }t |ttf�s|nd �|��qS )� )r2 �list�tuple�join)�.0�liner r r �
<listcomp>� s �z%Ticket.getMatches.<locals>.<listcomp>r r )r r$ r% r r r �
getMatches� s �zTicket.getMatchesc C s | j tj@ S r' �r r �RESTOREDr% r r r �restored� s zTicket.restoredc C s, |r| j tjO _ n| j tj M _ d S r' rU r3 r r r rW � s c C s | j tj@ S r' �r r �BANNEDr% r r r �banned� s z
Ticket.bannedc C s, |r| j tjO _ n| j tj M _ d S r' rX r3 r r r rZ � s c O s� t |�dkr*tdd� |d �� D ��| _nLt |�dkrF| j�|f� n0t |�dkrv| j�dd� tt|�gd � D �� t |�r�| j�|� tdd� | j�� D ��| _d S ) Nr c S s g | ]\}}|d ur||f�qS r' r �rQ r r r r r rS � � z"Ticket.setData.<locals>.<listcomp>r r c s s | ]\}}||fV qd S r' r r[ r r r � <genexpr>� r\ z!Ticket.setData.<locals>.<genexpr>c S s g | ]\}}|d ur||f�qS r' r r[ r r r rS � r\ )�len�dictr r r �zip�iter)r �args�argvr r r �setData� s $ zTicket.setDatac s� � d u r| j S | j s|S t� ttd �ttttf�s|t� �rVt � fdd�| j �
� D ��S t � d�r|t � fdd�| j �
� D ��S | j � � |�S )Nc s g | ]\}}� |�r||f�qS r r r[ ��keyr r rS � r\ z"Ticket.getData.<locals>.<listcomp>�__iter__c s g | ]\}}|� v r||f�qS r r r[ re r r rS � r\ )
r r2 r( �type�int�float�bool�complex�callabler_ r �hasattrr$ )r rf �defaultr re r �getData� s
zTicket.getDatac C s t | dd�S )N� _banEpochr )r/ r% r r r �banEpoch� s zTicket.banEpochc C s
|| _ d S r' )rq r3 r r r rr � s )N)F)r )N)N)NN)#r" �
__module__� __qualname__r. rC rV rY r r& r) r- r r r5 r6 r8 r9 r; r= r@ rA rB rE rF rI rJ rL rT �propertyrW �setterrZ rd rp rr r r r r r # sN
r c @ sN e Zd Zdddi dfdd�Zdd� Zdd� Zdd � Zdd d �Zed
d� �Z dS )�
FailTicketNc C sT d | _ d| _t�| |||||� t|t�sP|d ur6|n| �� | _ | j�dd�| _d S )Nr r ) �
_firstTimer
r r r2 rw r9 r r$ )r r r r r r r r r r � s
zFailTicket.__init__c C s6 || _ | jd sd| jd< |s2d| jd< g | jd< dS )z� Set artificial retry count, normally equal failures / attempt,
used in incremental features (BanTimeIncr) to increase retry count for bad IPs
r r r r N�r
r r3 r r r �setRetry� s
zFailTicket.setRetryc C s | j S )zV Returns failures / attempt count or
artificial retry count increased for bad IPs
)r
r% r r r �getRetry� s zFailTicket.getRetryc C sN || j krJ| j|| k rDtt| jt|| j � | ��| _|| | _|| _ dS )z� Adjust time of ticket and current attempts count considering given maxTime
as estimation from rate by previous known interval (if it exceeds the findTime)
N)r rx ri r* r
rj )r r ZmaxTimer r r �
adjustTime� s
"
zFailTicket.adjustTimer c C sR | j |7 _ | jd |7 < |rN| jd rD| jd | | jd< n
|| jd< d S )Nr r ry )r r Zattempt�countr r r �inc s
zFailTicket.incc C s
t | _| S r' )rw r! ��or r r �wrap s zFailTicket.wrap)Nr r )
r" rs rt r rz r{ r| r~ � staticmethodr� r r r r rw � s
rw c @ s e Zd Zedd� �ZdS )� BanTicketc C s
t | _| S r' )r� r! r r r r r� " s zBanTicket.wrapN)r" rs rt r� r� r r r r r� s r� N)�
__author__Z
__copyright__Z __license__Zhelpersr Zipdnsr Zmytimer r" ZlogSys�objectr rw r� r r r r �<module> s ??