File: //usr/lib/python3/dist-packages/fail2ban/client/__pycache__/fail2banclient.cpython-39.pyc
a
���` 9 � @ 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. BresterZGPL� N)�Thread� )�version� )�CSocket)�
Beautifier)�Fail2banCmdLine�ServerExecutionException�
ExitException�logSys�exit�output)�Utilsz
fail2ban> c C s t �� jjS �N)� threadingZcurrent_thread� __class__�__name__� r r �@/usr/lib/python3/dist-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 d&dd�Z d'dd�Z
dd � 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 Zwarningr )r Zsignum�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 �_confZ
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�accessr/ �F_OK�W_OKr r2 r6 )r Z prevErrorZ errorOnlyr@ 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 runningZforcer+ zLFail2ban seems to be in unexpected state (not running but the socket exists)z
server-streamr. )�_Fail2banClient__pingr r2 �
readConfigr/ rA �path�exists)r r? �streamr r r Z__prepareStartServer� s
z#Fail2banClient.__prepareStartServerc
C s ddl m} | �� }d| _|s"dS zd|rH|�| j� | �|d�s�W dS n<ttj| |dfd�}d|_ |�
� | � d� |� | jd�| _
W nz ty� Y nj t�y } zPtd� t�d|r�dnd � | jd
dkr�t�|� n
t�|� W Y d }~dS d }~0 0 dS ) Nr )�Fail2banServerTF��targetr3 r z Exception while starting server �
backgroundZ
foregroundr, )Zfail2banserverrJ �#_Fail2banClient__prepareStartServerr ZstartServerAsyncr/ �,_Fail2banClient__processStartStreamAfterWaitr r �daemon�startZ setDaemonZstartServerDirectr r
r6 r
r r2 r; )r rM rJ rI �thr@ r r r Z
__startServer� s2
z Fail2banClient.__startServerNc s� |r&t tj| d� fd�}d|_|�� S � d urDd� d< t�dd� � | �� }� d urz|r\dnd � d< � d< t�dd� � |s�dS � d ur�t� � fdd �d
d � |r�dnd� d < t�dd� � | �
|d�}� d ur�|� d
< |S )NFrK TrQ r- z client phase %sZreadyc s � � dd �d uS )Nz start-ready)�getr ��phaser r �<lambda>� � z0Fail2banClient.configureServer.<locals>.<lambda>� �?g����MbP?Z configureZdone) r r �configureServerrP rQ r r0 rN r Zwait_forrO )r ZnonsyncrU rR rI r? r rT r rY � 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 rQ rM FZrestart�reload� --restart� interactivez ## stop ... �stopzCould not stop serverz ## load configuration ... z ## start ... r )r[ z--unbanz --if-existsz%Unexpected argument(s) for reload: %rr& r' z--allz ## reload ... TzCould not find serverr% )�
isinstance�list�len� _Fail2banClient__startServerr/ � _Fail2banClient__processCommandrS r
� _Fail2banClient__waitOnServerr r2 Z resetConf� initCmdLine�_argv�appendrE rF r) �float)r r< r? ZoptsZjailrI r r r Z__processCommand� sh
z Fail2banClient.__processCommandc
G s� z&| � � st�d� W dS | j|� W nf ty� } zN| jd dkrNt�|� t�d| jd d � | jrv| j�� W Y d }~dS d }~0 0 dS ) Nz%Could not find server, waiting failedFr, 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 itT) rc r r2 r) r r/ r; r �quit)r r3 r@ r r r Z __processStartStreamAfterWaitC 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( r- z__waitOnServer: %rg������y?c s t j�� jd �o � j�d�S )Nr+ r' )rA rG rH r/ rE r �r Zsltimer r rV _ rW z/Fail2banClient.__waitOnServer.<locals>.<lambda>r, Tz wait-time: %sr zFailed to start serverr g�������?rX r$ F)
r/ �timer r0 �
VisualWaitr � heartbeatr �min�sleep)r �aliveZmaxtimeZ starttime�testZvisZrunfZwaittimer ri r Z__waitOnServerX 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 )NZ _MainThreadFTzInit of command line failedr\ r zReadline not availablez
tab: completer rh �helpr r, r ) r �signal�SIGTERM�SIGINT� getsignal� _Fail2banClient__sigTERMhandlerre rd r �itemsr Z_argsr/ rS �readline� ImportErrorr` rb �parse_and_bindr! r Z dispUsage�shlex�splitr6 r r; r2 �EOFError�KeyboardInterruptr
)
r �argvZ
_prev_signals�sr? Zshr3 rx r< r@ r r r rQ r s�
2�
�
(
� �
��zFail2banClient.start)r$ )Tr& )r F)T)TN)TN)r �
__module__� __qualname__r r! rv rE �propertyr* r) r4 rN ra rY rb rO rc rQ r r r r r 6 s
2
&
!I
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�
� �# )�posr8 r9 �writer� r: �r r3 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� r8 r9 r� r� �deltar: )r r� r r r rl � s
z_VisualWait.heartbeatN)r� )
r r� r� �__doc__r� r� r r� r� rl 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 rl � s z_NotVisualWait.heartbeatN)r r� r� r� r� r� rl r r r r r� � s r� c O s | dkrt |i |��S t� S )z3Wonderful visual progress indication (if verbose)
r )r� r� )r, r3 �kwargsr r r rk � s rk c C s& t � }|�| �rtd� ntd� d S )Nr r" )r rQ r )r r= r r r �exec_command_line� s
r� )!�
__author__Z
__copyright__Z __license__rA r{ rr r+ r8 rj r r r Zcsocketr r* r Zfail2bancmdliner r r
r r r
Z server.utilsr r r r r r� r� rk r� r r r r �<module> s4