File: //lib/python3.6/site-packages/fail2ban/client/__pycache__/fail2banclient.cpython-36.pyc
3
TB(d:= � @ 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 j� jjS )N)� threadingZcurrent_thread� __class__�__name__� r r �$/usr/lib/python3.6/fail2banclient.py� _thread_name, s r c C s t t�S )N)�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 j| � tj| � 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� tjd| � 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�pingr F)�timeout���)�_Fail2banClient__processCmd)r r$ r r r Z__pingJ s zFail2banClient.__pingc C s | j r | j S t� | _ | j S )N)r r )r r r r �
beautifierN s zFail2banClient.beautifierTr c
C sB d }�z�| j }d}�x�|D �]�}|j|� y�|sDt| jd |d�}n|dkrV|j|� | jd dkrrtjdd|� |j|�}|d d kr�tjdd
|d � |s�|d dkr�t|j |d �� n,tj
d
|d j � |r�t|j |d �� d}W q t
j
k
�r` } zT|�s| jd dk�rR|�s,|d dk�rB| j| |d dk� ntjdd|| � dS d } ~ X q tk
�r� } z@|�s�| jd dk�r�| jd dk�r�tj| � n
tj
| � dS d } ~ X qX qW W d |�r y |j� W nD tk
�r } z&|�s| jd dk�r tj| � W Y d d } ~ X nX |�s2|d dk�r<tjj� X |S )NT�socket)r$ r �verboser � zCMD: %rr zOK : %r�echo�
server-statuszNOK: %rFr# z -- %s failed -- %rr% )r+ r, )r+ r, )r' Z setInputCmdr �_confZ
settimeoutr �log�sendr
Zbeautify�error�argsZ
beautifyErrorr( � _Fail2banClient__logSocketError� Exception� exception�close�debug�sys�stdout�flush)
r �cmdZshowRetr$ �clientr' Z streamRet�c�ret�er r r Z __processCmdU sV
zFail2banClient.__processCmdr Fc C s� y|t j| jd t j�rht j| jd t j�rT|r:tj|� qftjd|rLd| nd� qztjd| jd � ntjd| jd � W n> tk
r� } z"tjd| jd � tj|� W Y d d }~X nX 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 r0 r3 )r Z prevErrorZ errorOnlyr> r r r Z__logSocketError� s z Fail2banClient.__logSocketErrorc C sd | j � rtjd� d S | j� \}}|s*d S | jd rVtjj| jd �rVtjd� d S d|gdggS )NzServer already running�forcer( zLFail2ban seems to be in unexpected state (not running but the socket exists)z
server-streamz
server-status)�_Fail2banClient__pingr r0 �
readConfigr- r? �path�exists)r r= �streamr r r Z__prepareStartServer� s
z#Fail2banClient.__prepareStartServerc C s
|| _ d S )N)r )r �sr r r � _set_server� s zFail2banClient._set_serverc C s ddl m} | j� }d| _|s"dS y~|rF|j| j� | j|d�s�dS nXt� }| j||d� d| _ |j
| jd| j �| _ |j
dd�s�| j r�| j j� d | _ td� W nz tk
r� � Y nf tk
�r } zHtd� tjd |r�d
nd � | jd dk�rtj|� n
tj|� dS d }~X nX dS )
Nr )�Fail2banServerTF)�phaserH �doner r z Exception while starting server �
backgroundZ
foregroundr) )Zfail2banserverrK �#_Fail2banClient__prepareStartServerr ZstartServerAsyncr- �,_Fail2banClient__processStartStreamAfterWait�dict�configureServer�daemonZstartServerDirectrJ r �get�quitr r
r3 r
r r0 r4 )r rN rK rH rL r> r r r Z
__startServer� s:
z Fail2banClient.__startServerNc s` |r�� d k r"� fdd�}|| j d< ttj| d� |fd�}d|_|j� |d kr�� d k r�tj� fdd�| j d d
� tj d d � � � j
d
d�s�t d��dS � d k r�d� d
< tj d d� � |d kr�| j � }� d k r�|r�dnd � d< � d
< tj d d� � |� s�dS � d k �r>tj� fdd�dd
� |�r(dnd� d< tj d d� � | j
|d�}� d k �r\|� d< |S )Nc s d� d< t jdd� � d S )NTz start-readyr* z server phase %s)r r. r )rL r r �
_server_ready� s z5Fail2banClient.configureServer.<locals>._server_readyZonstartF)�targetr1 Tc s � j dd �d k S )N�ready)rT r )rL r r �<lambda>� s z0Fail2banClient.configureServer.<locals>.<lambda>r$ g����MbP?r* z server phase %s�startz$Async configuration of server failedz client phase %srX c s � j dd �d k S )Nz start-ready)rT r )rL r r rY s g �?Z configurerM )r- r r rR rS rZ r Zwait_forr r. rT r rO rP )r ZnonsyncrL rH rV Zthr= r )rL r rR � s>
z Fail2banClient.configureServerc C s~ t |t�st|�}t|�dkrF|d dkrF| j| jd �}|sBdS |S t|�dko\|d dk�r t|�dkr�ddg|dd�< | j|�S | jjd d�r�td
� | jd g� | jd�s�t j
d � dS | jjd d�� r�td
� | j � | j | j
�}|d k � r�|S | jjd d��rtd� | jdg�S t|�dk�r:|d dk�r:g }x`t|�dk�r�|d dk�rr|j|d � |d= n(t|�dk�r�t j
d|dd � � dS P �q@W | jdd��r*t|�dk�s�|d dk�r�d}| j� \}}n|d }| j|�\}}|�s�dS | jjd d��rtd� | jd|||ggd�S t j
d� dS n@t|�dk�rn|d dk�rn| j|gt|d �d�S | j|g�S d S )Nr r rZ rN FZrestart�reload� --restart� interactivez ## stop ... �stopzCould not stop serverz ## load configuration ... z ## start ... r �--unban� --if-existsz%Unexpected argument(s) for reload: %r)r$ z--allz ## reload ... TzCould not find serverr# )r\ r_ r` r% )�
isinstance�list�len� _Fail2banClient__startServerr- � _Fail2banClient__processCommandrT r
� _Fail2banClient__waitOnServerr r0 Z resetConf� initCmdLine�_argv�appendrD rE r& �float)r r: r= ZoptsZjailrH r r r Z__processCommand sh
z Fail2banClient.__processCommandc G s� d}y$| j � s tjd� dS | j|� }W nR tk
rz } z6| jd dkrRtj|� tjd| jd d � W Y d d }~X nX | r�| jr�| jj� d | _|S )NFz%Could not find server, waiting failedr) 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) rf r r0 r& r r- r4 r rU )r r1 r= r> r r r Z __processStartStreamAfterWaitZ s
&
z,Fail2banClient.__processStartStreamAfterWaitc s� |d kr� j d }tj� }tjdd||f� d�� �fdd�}t� j d ���}xz� jr�|� }||krhd S tj� | }tjdd
|� |d kr�|j� ||kr�td ��t�d |d
kr�dnd��tj �� qPW W d Q R X dS )Nr$ r* z__waitOnServer: %rg�������?r c s t jj� jd �o � j�d�S )Nr( )r$ )r? rF rG r- rD r )r �sltimer r rY x s z/Fail2banClient.__waitOnServer.<locals>.<lambda>r) Tz wait-time: %sr zFailed to start serverg�������?g �?g�������?Fg������y?)
r- �timer r. �
VisualWaitr � heartbeatr �minZsleep)r �aliveZmaxtimeZ starttimeZtestZvisZrunfZwaittimer )r rk r Z__waitOnServerq s(
z Fail2banClient.__waitOnServerc
* C sT i }t � dkr@x0tjtjfD ] }tj|�||< tj|| j� q W �z�y�| jd krt| j|�}|d k rt|rldS td��| j }| j
j dd��r�y dd l }W n t
k
r� td��Y nX y�d}t|�dkr�| j|�}|�r�|jd� | j� x�t� }|d kp�|d
k�rdS |d k�r| j� q�|d ks�y| jtj|�� W q� tk
�r� } z,| j
d
dk�rftj|� n
tj|� W Y d d }~X q�X q�W W n$ ttfk
�r� t d � � Y nX n$t|�dk �r�| j� dS | j|�S W nH tk
�r } z*| j
d
dk�rtj|� n
tj|� dS d }~X nX W d d| _ x"|j � D ]\}} tj|| � �q4W X d S )NZ _MainThreadTzInit of command line failedr] Fr zReadline not availablez
tab: completer rU �helpr r) r ) r �signal�SIGTERM�SIGINT� getsignal� _Fail2banClient__sigTERMhandlerrh rg r Z_argsr- rT �readline� ImportErrorrc re �parse_and_bindr r Z dispUsage�shlex�splitr3 r r4 r0 �EOFError�KeyboardInterruptr
r �items)
r �argvZ
_prev_signalsrI r= r1 rw r: r> Zshr r r rZ � sj
$
zFail2banClient.start)r" r% )Tr% )r F)T)TNN)TN)r �
__module__� __qualname__r r rv rD �propertyr' r& r2 rO rJ rd rR re rP rf rZ 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 )N)�maxpos)r r� r r r r � s z_VisualWait.__init__c C s | S )Nr )r r r r � __enter__� s z_VisualWait.__enter__c G s2 | j r.tjjddd| j d � tjj� d S )N�
� �# )�posr7 r8 �writer� r9 )r r1 r r r �__exit__� s z_VisualWait.__exit__c C s� | j s tjjdd| j d � | j | j7 _ | jdkrN| j dkrHdnd}nd}tjj|� tjj� | 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 Nr% )r� r7 r8 r� r� �deltar9 )r rI r r r rn � s
z_VisualWait.heartbeatN)r� )
r r� r� �__doc__r� r� r r� r� rn 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 )Nr )r r r r r� � s z_NotVisualWait.__enter__c G s d S )Nr )r r1 r r r r� � s z_NotVisualWait.__exit__c C s d S )Nr )r r r r rn � s z_NotVisualWait.heartbeatN)r r� r� r� r� r� rn r r r r r� � s r� c O s | dkrt ||�S t� S )z3Wonderful visual progress indication (if verbose)
r )r� r� )r) r1 �kwargsr r r rm � s rm c C s& t � }|j| �rtd� ntd� d S )Nr r )r rZ r )r r; r r r �exec_command_line� s
r� )!�
__author__Z
__copyright__Z __license__r? rz rr r( r7 rl r r r Zcsocketr r' r Zfail2bancmdliner r r
r r r
Z server.utilsr r r r r r� r� rm r� r r r r �<module> s4