File: //usr/lib/python3/dist-packages/fail2ban/server/__pycache__/ticket.cpython-310.pyc
o
;s*b ! � @ 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$� Zd>d%d&�Zd>d'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 dAd9d:�Z!e d;d<� �Z"e"j d=d<� �Z"dS )B�Ticket)�_ip�_flags� _banCount�_banTime�_time�_data�_retry�
_lastResetl ��� r � Nc C s� | � |� d| _d| _d| _|dur|nt�� | _|p g dd�| _|dur7|�� D ]
\}}|dur6|| j|< q)|r@| � |� dS dS )z{Ticket constructor
@param ip the IP address
@param time the ban time
@param matches (log) lines caused the ticket
� N)�matches�failures)
�setIPr r
r r �timer r
�items�update)�self�ipr r �data�ticket�k�v� r �8/usr/lib/python3/dist-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 sH z| j |j kot| jd�t|jd�ko| j|jkW S ty# Y dS w )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�r t|�}|| _d S r) )�
isinstancer* r r �r �valuer r r r Z s
z Ticket.setIPc C s | j �d| j�S )N�fid)r
r& r r' r r r �getID` s z Ticket.getIDc C � | j S r) )r r' r r r �getIPc � z Ticket.getIPc C �
|| _ d S r) �r r5 r r r �setTimef �
zTicket.setTimec C r9 r) r= r' r r r �getTimei r; zTicket.getTimec C r< r) �r r5 r r r �
setBanTimel r? zTicket.setBanTimec C s | j d ur| j S |S r) rA )r � defaultBTr r r �
getBanTimeo s zTicket.getBanTimeFc C s |s|| j kr || _ d S d S r) �r
)r r6 �alwaysr r r � setBanCountr s
�zTicket.setBanCountc C s | j |7 _ d S r) rE r5 r r r � incrBanCountv s zTicket.incrBanCountc C r9 r) rE r' r r r � getBanCounty r; zTicket.getBanCountc C s, | j d ur| j n|}|dkrtjS | j| S )Nr" )r r �MAX_TIMEr )r rC �bantimer r r �getEndOfBanTime| s
zTicket.getEndOfBanTimec C s. | j d ur| j n|}|dkrdS || j| kS )Nr" F)r r )r r rC rK r r r �
isTimedOut� s zTicket.isTimedOutc C s || j d< d S �Nr �r
r5 r r r �
setAttempt� s zTicket.setAttemptc C s
| j d S rN rO r' r r r �
getAttempt� r? zTicket.getAttemptc C s6 |r || j d< d S z| j d= W d S ty Y d S w )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 )� )r4 �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 � | j tj@ S r) �r r �RESTOREDr' r r r �restored� � zTicket.restoredc C �. |r | j tjO _ d S | j tj M _ d S r) r] r5 r r r r_ � � c C r\ r) �r r �BANNEDr' r r r �banned� r` z
Ticket.bannedc C ra r) rc r5 r r r re � rb c O s� t |�dkrtdd� |d �� D ��| _n&t |�dkr#| j�|f� nt |�dkr;| j�dd� tt|�gd � D �� t |�rE| j�|� tdd� | j�� D ��| _d S ) Nr c S � g | ] \}}|d ur||f�qS r) r �rX r r r r r rZ � � z"Ticket.setData.<locals>.<listcomp>r r c s s � | ] \}}||fV qd S r) r rg r r r � <genexpr>� s � z!Ticket.setData.<locals>.<genexpr>c S rf r) r rg r r r rZ � rh )�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� �r+t � fdd�| j �
� D ��S t � d�r>t � fdd�| j �
� D ��S | j � � |�S )Nc s g | ] \}}� |�r||f�qS r r rg ��keyr r rZ � rh z"Ticket.getData.<locals>.<listcomp>�__iter__c s g | ] \}}|� v r||f�qS r r rg rq r r rZ � rh )
r
r4 r* �type�int�float�bool�complex�callablerk r �hasattrr& )r rr �defaultr rq r �getData� s
zTicket.getDatac C s t | dd�S )N� _banEpochr )r1 r' r r r �banEpoch� r` zTicket.banEpochc C r< r) )r} r5 r r r r~ � s
r) )F)r )NN)#r$ �
__module__� __qualname__r0 rJ r^ rd r r( r+ r/ r r r8 r: r> r@ rB rD rG rH rI rL rM rP rQ rS r[ �propertyr_ �setterre rp r| r~ r r r r r # sP
r c @ sN e Zd Zdddi dfdd�Zdd� Zdd� Zdd � Zdd d �Zed
d� �Z dS )�
FailTicketNc C sX d | _ d| _t�| |||||� t|t�s*|d ur|n| �� | _ | j�dd�| _d S d S )Nr r ) �
_firstTimer r r r4 r� r@ r
r& )r r r r r r r r r r � s
�zFailTicket.__init__c C s: || _ | jd s
d| jd< |sd| jd< g | jd< dS 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
r5 r r r �setRetry� s
�zFailTicket.setRetryc C r9 )zV Returns failures / attempt count or
artificial retry count increased for bad IPs
)r r' r r r �getRetry� s zFailTicket.getRetryc C sR || j kr'| j|| k r"tt| jt|| j � | ��| _|| | _|| _ dS 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 r� ru r, r rv )r r �maxTimer r r �
adjustTime s
"
�zFailTicket.adjustTimer c C sX | j |7 _ | jd |7 < |r*| jd r#| jd | | jd< d S || jd< d S d S )Nr r r� )r r �attempt�countr r r �inc
s
�zFailTicket.incc C �
t | _| S r) )r� r# ��or r r �wrap � zFailTicket.wrap)Nr r )
r$ r r� r r� r� r� r� � staticmethodr� r r r r r� � s
r� c @ s e Zd Zedd� �ZdS )� BanTicketc C r� r) )r� r# r� r r r r� # r� zBanTicket.wrapN)r$ r r� r� r� r r r r r� ! s r� N)�
__author__�
__copyright__� __license__�helpersr �ipdnsr �mytimer r$ �logSys�objectr r� r� r r r r �<module> s @?