File: //usr/lib/python3.9/site-packages/fail2ban/server/__pycache__/ticket.cpython-39.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 Jaquier�GPL� )� 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 r4 r r r �setTimee s zTicket.setTimec C s | j S r( r8 r& r r r �getTimeh s zTicket.getTimec C s
|| _ d S r( �r r4 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 r5 �alwaysr r r � setBanCountq s zTicket.setBanCountc C s | j |7 _ d S r( r? r4 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= rE r r r �
isTimedOut� s zTicket.isTimedOutc C s || j d< d S �Nr �r
r4 r r r �
setAttempt� s zTicket.setAttemptc C s
| j d S rH rI 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 )� )r3 �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( rV r4 r r r rX � 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( rY r4 r r r r[ � 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 �rR r r r r r rT � � 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 rT � 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 rT � r] z"Ticket.getData.<locals>.<listcomp>�__iter__c s g | ]\}}|� v r||f�qS r r r\ rf r r rT � r] )
r
r3 r) �type�int�float�bool�complex�callabler` r �hasattrr% )r rg �defaultr rf r �getData� s
zTicket.getDatac C s t | dd�S )N� _banEpochr )r0 r& r r r �banEpoch� s zTicket.banEpochc C s
|| _ d S r( )rr r4 r r r rs � s )N)F)r )N)N)NN)#r# �
__module__� __qualname__r/ rD rW rZ r r' r* r. r r r6 r7 r9 r: r< r> rA rB rC rF rG rJ rK rM rU �propertyrX �setterr[ re rq rs 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 r3 rx r: 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
r4 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 ry rj r+ r rk )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 rz )r r �attempt�countr r r �inc s
zFailTicket.incc C s
t | _| S r( )rx r" ��or r r �wrap s zFailTicket.wrap)Nr r )
r# rt ru r r{ r| r} r� � staticmethodr� r r r r rx � s
rx c @ s e Zd Zedd� �ZdS )� BanTicketc C s
t | _| S r( )r� r" r� r r r r� " s zBanTicket.wrapN)r# rt ru r� r� r r r r r� s r� N)�
__author__�
__copyright__� __license__Zhelpersr Zipdnsr Zmytimer r# �logSys�objectr rx r� r r r r �<module> s ??