File: //usr/lib/python3/dist-packages/fail2ban/server/__pycache__/asyncserver.cpython-39.pyc
a
���`�' � @ s� d Z dZdZddlmZmZmZ ddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlmZ dd lmZ dd
lmZmZmZ ee�ZG d d � d ej�Zddd�Z G dd� dej �Z G dd� de �Z dS )z
Cyril Jaquierz Copyright (c) 2004 Cyril JaquierZGPL� )�dumps�loads�HIGHEST_PROTOCOLN� )�Utils� )�CSPROTO)�logging� getLogger�formatExceptionInfoc @ sL e Zd Zdd� Zdd� Zdd� Zdd� ZG d d
� d
e�Zd d � Z d
d� Z
dS )�RequestHandlerc C s0 t j�| |� || _|| _g | _| �tj� d S �N) �asynchat�
async_chat�__init__�_RequestHandler__conn� _RequestHandler__transmitter�_RequestHandler__bufferZset_terminatorr �END)�self�conn� transmitter� r �=/usr/lib/python3/dist-packages/fail2ban/server/asyncserver.pyr 5 s
zRequestHandler.__init__c C sD | j r@| j }d | _ z|�tj� |�� W n tjy> Y n0 d S r
)r �shutdown�socket� SHUT_RDWR�close�error)r r r r r Z__close= s zRequestHandler.__closec C s | � � tj�| � d S r
)�_RequestHandler__closer r � handle_close�r r r r r G s zRequestHandler.handle_closec C s | j �|� d S r
)r �append)r �datar r r �collect_incoming_dataK s z$RequestHandler.collect_incoming_datac @ s e Zd ZdS )zRequestHandler.LoadErrorN��__name__�
__module__� __qualname__r r r r � LoadErrorP s r) c
C s. z�| j }g | _ tj�|�}|tjkr2| �� W d S z t|�}W nH ty� } z0tj d|t�
� t j k d� t
�|��W Y d }~n
d }~0 0 | jr�| j�|�}ndg}t|t�}| �|tj � W nj t�y( } zPt|t
j�s�tj d|t�
� t j kd� td| t�}| �|tj � W Y d }~n
d }~0 0 d S )Nz$PROTO-error: load message failed: %s��exc_infoZSHUTDOWNz Caught unhandled exception: %r� ERROR: %s)r r ZEMPTY�joinZCLOSE�close_when_doner � Exception�logSysr ZgetEffectiveLevelr �DEBUGr r) r Zproceedr r �pushr �
isinstance)r �message�er r r �found_terminatorX s4
�
�z RequestHandler.found_terminatorc
C s� zPt � \}}t�dt|� � t�t�� �� � td| t�}| � |t
j � W n$ t yt } z W Y d }~n
d }~0 0 | �
� d S )Nz"Unexpected communication error: %sr, )r r0 r �str� traceback�
format_exc�
splitlinesr r r2 r r r/ r. )r Ze1Ze2r4 r5 r r r � handle_error~ s
zRequestHandler.handle_errorN) r& r' r( r r r r$ r/ r) r6 r; r r r r r 3 s
&r Fc
C s� |si }d|d< |du r t j}tj}t|�r2|}n2|rdtjrdttjd�rdt� d� t
|�d }tj}| � �r�z$||� |d r�|d d8 < W qd t �y� } z�| � s�W Y d}~�q�|d d7 < |d dk �r|j d t
jt
jfv r�t�d t|�� nt�d t|�� np|d dk�r2t�|� t�d
� nL|d d k�r~|j d t
jk�sdt|�� �dk�r~t�d |� W Y d}~�q�W Y d}~qdd}~0 0 qddS )
z�Custom event loop implementation
Uses poll instead of loop to respect `active` flag,
to avoid loop timeout mistake: different in poll and poll2 (sec vs ms),
and to prevent sporadic errors like EBADF 'Bad file descriptor' etc. (see gh-161)
r �listenN�pollz"Server listener (select) uses poll�� r � z Server connection was closed: %sz0Too many errors - stop logging connection errors�d �+Too many errors - critical count reached %r)r ZDEFAULT_SLEEP_TIME�asyncorer= �callableZpoll2�hasattrZselectr0 �debug�floatr/ �args�errnoZENOTCONNZEBADF�infor7 r Z exception�EMFILE�sum�values�critical)Zactive�timeout�use_poll� err_countr= r5 r r r �loop� sD
�� rQ c @ sb e Zd Zdd� Zdd� Zdd� Zdd d
�Zd d � Zd
d� Zdd� Z dd� Z
dd� Z e dd� �Z
dS )� AsyncServerc C s: t j�| � || _d| _d| _d| _ddd�| _d | _d S )Nz /var/run/fail2ban/fail2ban.sockFr )�acceptr< ) rB �
dispatcherr �_AsyncServer__transmitter�_AsyncServer__sock�_AsyncServer__init�_AsyncServer__active�_AsyncServer__errCount�onstart)r r r r r r � s zAsyncServer.__init__c C s dS )NFr r! r r r �writable� s zAsyncServer.writablec
C s z| � � \}}W n� ty� } z�| jd d7 < | jd dk r\tjd|| jd dkd� nl| jd dkrvt�d� nR| jd dkr�t|tj�r�|jd t j
ks�t | j� � �d kr�t�
d
| j� | �� W Y d }~d S d }~0 0 | jd � r�| jd d8 < t�|� t|| j� d S ) NrS r r? zAccept socket error: %sr* z.Too many acceptor errors - stop logging errorsr@ r r> rA )rS r/ rY r0 �warningr r3 r rG rH rJ rK rL rM �stoprR � _AsyncServer__markCloseOnExecr rU )r r Zaddrr5 r r r �
handle_accept� s0 �
���
zAsyncServer.handle_acceptNFc s� t �� � _|� _tj�|�rFt�d� |r>t� d� � �
� nt d��� � t
jt
j� � �� z� �|� W n ty� t d� j ��Y n0 t�� j
� � �d� d � _ � _� _� jrĈ �� t� fdd�||� jd � d
� _� � � d S ) Nz$Fail2ban seems to be already runningz Forcing execution of the serverzServer already runningzUnable to bind socket %sr Tc s � j S r
)�_AsyncServer__loopr r! r r �<lambda> � z#AsyncServer.start.<locals>.<lambda>)rN rO rP F) � threading�current_thread�_AsyncServer__workerrV �os�path�existsr0 r r\ � _remove_sock�AsyncServerExceptionZ
create_socketr ZAF_UNIXZ SOCK_STREAMZset_reuse_addrZbindr/ rR r^ r< rW r` rX rZ rQ rY r] )r �sockZforcerN rO r r! r �start� s,
zAsyncServer.startc s� d}� j rpd� _� jr>z� j�tj� W n tjy< Y n0 tj�� � t �
� � j krlt �
� fdd�d� d}� jr�tj�� j�r�� �� t�d� j � |r�t�d� d� _ d S )NFc s � j S r
�rX r r! r r ra rb z#AsyncServer.close.<locals>.<lambda>r TzRemoved socket file zSocket shutdown)rX r` r r r r rB rT r rc rd re r Zwait_forrW rf rg rh rV ri r0 rE )r Zstopflgr r! r r s$
zAsyncServer.closec C s | j rt�d� d | _ d S )Nz Stop communication, shutdown)rU r0 rE r! r r r �stop_communication, s
z AsyncServer.stop_communicationc C s | � � | �� d S r
)rn r r! r r r r] 4 s zAsyncServer.stopc C s | j S r
rm r! r r r �isActive9 s zAsyncServer.isActivec
C sH zt �| j� W n2 tyB } z|jtjkr.� W Y d }~n
d }~0 0 d S r
)rf �removerV �OSErrorrH ZENOENT)r r5 r r r ri ? s
zAsyncServer._remove_sockc C s0 | � � }t�|tj�}t�|tj|tjB � d S r
)�fileno�fcntlZF_GETFDZF_SETFDZ
FD_CLOEXEC)rk �fd�flagsr r r Z__markCloseOnExecL s z AsyncServer.__markCloseOnExec)NF)r& r' r( r r[ r_ rl r rn r] ro ri � staticmethodr^ r r r r rR � s
rR c @ s e Zd ZdS )rj Nr% r r r r rj V s rj )NFN)!�
__author__Z
__copyright__Z __license__�pickler r r r rB rH rs rf r �sysrc r8 Zutilsr Zprotocolr Zhelpersr r
r r& r0 r r rQ rT rR r/ rj r r r r �<module> s* X
6