File: //lib/python3.9/site-packages/fail2ban/client/__pycache__/fail2banclient.cpython-39.pyc
a
�I�W4�G}� @ s� d Z dZdZddlZddlZddlZddlZddlZddlZddl Z ddl m
Z
ddl m Z dd l m
Z
dd
lmZ dd lmZmZmZmZmZmZ dd lmZ d
Zdd� Zdd� ZG dd� dee
�Z G dd� d�Z G dd� d�Z dd� Z dd� Z dS ) zFail2Ban Developersz^Copyright (c) 2004-2008 Cyril Jaquier, 2012-2014 Yaroslav Halchenko, 2014-2016 Serg G. Brester�GPL� N)�Thread� )�version� )�CSocket)�
Beautifier)�Fail2banCmdLine�ServerExecutionException�
ExitException�logSys�exit�output)�Utilsz
fail2ban> c C s t �� jjS �N)� threading�current_thread� __class__�__name__� r r �B/usr/lib/python3.9/site-packages/fail2ban/client/fail2banclient.py� _thread_name, s r c C s t t�S r )�input�PROMPTr r r r �
input_command/ s r c @ s� e Zd Zdd� Zdd� Zdd� Zd%dd �Zed
d � �Zd&dd�Z d'dd�Z
dd� Z dd� Z d(dd�Z
d)dd �Zd d � Zd d � Zd*d!d"�Zd#d$� ZdS )+�Fail2banClientc C s* t �| � t�| � d| _d | _d | _d S )NT)r �__init__r �_alive�_server� _beautifier��selfr r r r 8 s
zFail2banClient.__init__c C s$ t dt d � t d� t d� d S )Nz
Fail2Ban vz5 reads log file that contains password failure reportz=and bans the corresponding IP addresses using firewall rules.� )r r r r r r �dispInteractive? s z Fail2banClient.dispInteractivec C s" t d� t�d| � td� d S )Nr" zCaught signal %d. Exiting� )r r �warningr
)r! �signum�framer r r Z__sigTERMhandlerD s z Fail2banClient.__sigTERMhandler皙�����?c C s& | j dg|dkr|gng gd|d�S )N�ping���F��timeout)�_Fail2banClient__processCmd)r! r, r r r Z__pingJ s �zFail2banClient.__pingc C s | j r | j S t� | _ | j S r )r r r r r r �
beautifierN s zFail2banClient.beautifierTr* c
C s� d }�z4| j }d}|D �]�}|�|� z�|s@t| jd |d�}n|dkrR|�|� | jd dkrnt�dd|� |�|�}|d d kr�t�dd
|d � |s�|d d v r�t|� |d �� n,t�
d
|d j � |r�t|� |d �� d}W q t
j
�y� } z�|�s| jd d k�rL|�s&|d dk�r<| �| |d dk� nt�dd|| � W Y d } ~ W |�r�z |�� W nF t�y� } z,|�s�| jd d k�r�t�| � W Y d } ~ n
d } ~ 0 0 |�s�|d d v �r�tj�� dS d } ~ 0 t�y� } z�|�s| jd d k�r*| jd d k�r t�| � n
t�
| � W Y d } ~ W |�r�z |�� W nF t�y� } z,|�sr| jd d k�r|t�| � W Y d } ~ n
d } ~ 0 0 |�s�|d d v �r�tj�� dS d } ~ 0 0 qW |�r z |�� W nF t�y } z,|�s�| jd d k�rt�| � W Y d } ~ n
d } ~ 0 0 |�s0|d d v �r�tj�� nz|�r�z |�� W nF t�y� } z,|�sv| jd d k�r�t�| � W Y d } ~ n
d } ~ 0 0 |�s�|d d v �r�tj�� 0 |S )NT�socketr+ r* �verboser � zCMD: %rr zOK : %rr )Zecho�
server-statuszNOK: %rFr) z -- %s failed -- %r)r. Z setInputCmdr �_conf�
settimeoutr �log�sendr Zbeautify�error�argsZ
beautifyErrorr/ � _Fail2banClient__logSocketError�close� Exception�debug�sys�stdout�flush� exception)
r! �cmdZshowRetr, �clientr. Z streamRet�c�ret�er r r Z __processCmdU s�
�
� � zFail2banClient.__processCmdr" Fc
C s� z|t �| jd t j�rht �| jd t j�rT|r:t�|� qft�d|rLd| nd� qzt�d| jd � nt�d| jd � W n@ ty� } z(t�d| jd � t�|� W Y d }~n
d }~0 0 d S )Nr/ z*%sUnable to contact server. Is it running?z[%s] r" z3Permission denied to socket: %s, (you must be root)z6Failed to access socket path: %s. Is fail2ban running?z*Exception while checking socket access: %s)�os�accessr3 �F_OK�W_OKr r7 r; )r! Z prevErrorZ errorOnlyrE r r r Z__logSocketError� s& ����z Fail2banClient.__logSocketErrorc C sb | � � rt�d� d S | �� \}}|s*d S | jd sTtj�| jd �rTt�d� d S d|gdggS )NzServer already running�forcer/ zLFail2ban seems to be in unexpected state (not running but the socket exists)z
server-streamr2 )�_Fail2banClient__pingr r7 �
readConfigr3 rF �path�exists)r! rD �streamr r r Z__prepareStartServer� s
z#Fail2banClient.__prepareStartServerc C s
|| _ d S r )r �r! �sr r r � _set_server� s zFail2banClient._set_serverc
C s$ ddl m} | �� }d| _|s"dS z�|rH|�| j� | �|d�s�W dS nXt� }| j||d� d| _ |�
| jd| j �| _ |�
dd�s�| j r�| j �� d | _ td� W n| ty� � Y nj t�y } zPtd� t�d |r�d
nd � | jd dkr�t�|� n
t�|� W Y d }~dS d }~0 0 dS )
Nr )�Fail2banServerTF)�phaserO �doner$ r" z Exception while starting server �
backgroundZ
foregroundr0 )Zfail2banserverrS �#_Fail2banClient__prepareStartServerr ZstartServerAsyncr3 �,_Fail2banClient__processStartStreamAfterWait�dict�configureServer�daemonZstartServerDirectrR r �get�quitr
r r; r r r7 r@ )r! rV rS rO rT rE r r r Z
__startServer� s:
z Fail2banClient.__startServerNc s^ |r�� d ur"� fdd�}|| j d< ttj| d� |fd�}d|_|�� |d u r�� d ur�t�� fdd�| j d d
� t� d d � � � �
d
d�s�t d��dS � d ur�d� d
< t� d d� � |d u r�| � � }� d ur�|r�dnd � d< � d
< t� d d� � |s�dS � d u�r<t�� fdd�dd
� |�r&dnd� d< t� d d� � | �
|d�}� d u�rZ|� d< |S )Nc s d� d< t �dd� � d S )NT� start-readyr1 � server phase %s)r r5 r �rT r r �
_server_ready� s z5Fail2banClient.configureServer.<locals>._server_readyZonstartF)�targetr8 Tc s � � dd �d uS )N�ready�r\ r r` r r �<lambda>� � z0Fail2banClient.configureServer.<locals>.<lambda>r, g����MbP?r1 r_ �startz$Async configuration of server failedz client phase %src c s � � dd �d uS )Nr^ rd r r` r r re rf � �?Z configurerU )r3 r r rZ r[ rg r �wait_forr r5 r\ r
rW rX )r! ZnonsyncrT rO ra �thrD r r` r rZ � s>
z Fail2banClient.configureServerc C sz t |t�st|�}t|�dkrF|d dkrF| �| jd �}|sBdS |S t|�dk�r |d dk�r t|�dkr�ddg|dd�< | �|�S | j�d d�r�td
� | �d g� | �d�s�t �
d � dS | j�d d�r�td
� | � � | � | j
�}|d ur�|S | j�d d��rtd� | �dg�S t|�dk�r6|d dk�r6g }t|�dk�r�|d dv �rn|�|d � |d= n*t|�dk�r�t �
d|dd � � dS �q��q<| jdd��r&t|�dk�s�|d dk�r�d}| �� \}}n|d }| �|�\}}|�s�dS | j�d d��rtd� | �d|||ggd�S t �
d� dS n@t|�dk�rj|d dk�rj| j|gt|d �d�S | �|g�S d S )Nr r rg rV FZrestart�reload� --restart� interactivez ## stop ... �stopzCould not stop serverz ## load configuration ... z ## start ... r )rl z--unbanz --if-existsz%Unexpected argument(s) for reload: %rr* r+ z--allz ## reload ... TzCould not find serverr) )�
isinstance�list�len� _Fail2banClient__startServerr3 � _Fail2banClient__processCommandr\ r � _Fail2banClient__waitOnServerr r7 Z resetConf� initCmdLine�_argv�appendrK rL r- �float)r! rA rD �optsZjailrO r r r Z__processCommand sh
z Fail2banClient.__processCommandc
G s� d}z&| � � s t�d� W dS | j|� }W nT ty~ } z<| jd dkrRt�|� t�d| jd d � W Y d }~n
d }~0 0 |s�| jr�| j�� d | _|S )NFz%Could not find server, waiting failedr0 r zQCould not start server. Maybe an old socket file is still present. Try to remove r/ zR. If you used fail2ban-client to start the server, adding the -x option will do it) rt r r7 r- r
r3 r@ r r] )r! r8 rD rE r r r Z __processStartStreamAfterWaitZ s$
��
z,Fail2banClient.__processStartStreamAfterWaitc s� |d u r� j d }t�� }t�dd||f� d�� �fdd�}t� j d ���}� jr�|� }||krrW d � dS t�� | }t�dd |� |d
kr�|�� ||kr�td ��t�d |d
kr�dnd��t� �� qNW d � n1 s�0 Y dS )Nr, r1 z__waitOnServer: %rg������y?c s t j�� jd �o � j�d�S )Nr/ r+ )rF rM rN r3 rK r �r! Zsltimer r re x rf z/Fail2banClient.__waitOnServer.<locals>.<lambda>r0 Tz wait-time: %sr zFailed to start serverr g�������?rh r( F)
r3 �timer r5 �
VisualWaitr � heartbeatr
�min�sleep)r! �aliveZmaxtimeZ starttime�testZvisZrunf�waittimer rz r Z__waitOnServerq s(
*z Fail2banClient.__waitOnServerc
C sJ i }t � dkr<tjtjfD ] }t�|�||< t�|| j� q�zސz:| jd u r�| �|�}|d ur�|r�W W d| _|� � D ]\}}t�||� qvdS t
d��| j }| j �
dd��rz dd l}W n ty� t
d��Y n0 �zd}t|�dkr�| �|�}|�r�|�d� | �� t� }|d k�s&|d
k�rVW W W d| _|� � D ]\}}t�||� �q:dS |d k�rj| �� nl|d k�s z| �t�|�� W nL t�y� } z2| j d
dk�r�t�| � n
t�| � W Y d } ~ n
d } ~ 0 0 �q W n" t t f�y� t d � � Y n0 nxt|�dk �rF| �� W W d| _|� � D ]\}}t�||� �q*dS | �|�W W d| _|� � D ]\}}t�||� �q`S W nv t�y� } z\| j d
dk�r�t�| � n
t�| � W Y d } ~ W d| _|� � D ]\}}t�||� �q�dS d } ~ 0 0 W d| _|� � D ]\}}t�||� �qn(d| _|� � D ]\}}t�||� �q,0 d S )N� _MainThreadFTzInit of command line failedrm r zReadline not availablez
tab: completer
r] �helpr" r0 r ) r �signal�SIGTERM�SIGINT� getsignal� _Fail2banClient__sigTERMhandlerrv ru r �itemsr
�_argsr3 r\ �readline� ImportErrorrq rs �parse_and_bindr# r Z dispUsage�shlex�splitr; r r@ r7 �EOFError�KeyboardInterruptr )
r! �argvZ
_prev_signalsrQ rD �shr8 r� rA rE r r r rg � s�
2�
�
(
� �
��zFail2banClient.start)r( )Tr* )r" F)T)TNN)TN)r �
__module__� __qualname__r r# r� rK �propertyr. r- r9 rW rR rr rZ rs rX rt rg r r r r r 6 s
2
*
1I
r c @ s: e Zd ZdZdZdZddd�Zdd� Zd d
� Zd d � Z d
S )� _VisualWaitzJSmall progress indication (as "wonderful visual") during waiting process
r r �
c C s
|| _ d S r )�maxpos)r! r� r r r r � s z_VisualWait.__init__c C s | S r r r r r r � __enter__� s z_VisualWait.__enter__c G s2 | j r.tj�ddd| j d � tj�� d S )N�
� �# )�posr= r> �writer� r? �r! r8 r r r �__exit__� s z_VisualWait.__exit__c C s� | j s tj�dd| j d � | j | j7 _ | jdkrN| j dkrHdnd}nd}tj�|� tj�� | j | jkr|d | _n| j d
k r�d| _d S ) z&Show or step for progress indicator
z
INFO [#r� z ] Waiting on the server...
[8Cr r z #[1Dz# [2Dz
[1D# [2Dr* r N)r� r= r> r� r� �deltar? rP r r r r} � s
z_VisualWait.heartbeatN)r� )
r r� r� �__doc__r� r� r r� r� r} r r r r r� � s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �_NotVisualWaitz8Mockup for invisible progress indication (not verbose)
c C s | S r r r r r r r� � s z_NotVisualWait.__enter__c G s d S r r r� r r r r� � s z_NotVisualWait.__exit__c C s d S r r r r r r r} � s z_NotVisualWait.heartbeatN)r r� r� r� r� r� r} r r r r r� � s r� c O s | dkrt |i |��S t� S )z3Wonderful visual progress indication (if verbose)
r )r� r� )r0 r8 �kwargsr r r r| � s r| c C s& t � }|�| �rtd� ntd� d S )Nr r$ )r rg r
)r� rB r r r �exec_command_line� s
r� )!�
__author__�
__copyright__� __license__rF r� r� r/ r= r{ r r r Zcsocketr r. r Zfail2bancmdliner r
r r r
r Z server.utilsr r r r r r� r� r| r� r r r r �<module> s4