File: //usr/lib/python3.9/site-packages/fail2ban/server/__pycache__/asyncserver.cpython-39.opt-2.pyc
a
� 8D )� @ s& d Z dZdZddlmZmZmZ z ddlZW n eyJ ddl mZ Y n0 z ddl
Z
W n eyv ddl m
Z
Y n0 ddl Z ddl Z ddl
Z
ddlZddlZddlZddlZd d
lmZ dd lmZ dd lmZmZmZ ee�ZG d
d� dej �Z ddd�Z G dd� de
j �Z G dd� de!�Z"dS )z
Cyril Jaquierz Copyright (c) 2004 Cyril JaquierZGPL� )�dumps�loads�HIGHEST_PROTOCOLN� )�asynchat)�asyncore� )�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) r �
async_chat�__init__�_RequestHandler__conn� _RequestHandler__transmitter�_RequestHandler__bufferZset_terminatorr
�END)�self�conn� transmitter� r �?/usr/lib/python3.9/site-packages/fail2ban/server/asyncserver.pyr ; 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__closeC s zRequestHandler.__closec C s | � � tj�| � d S r )�_RequestHandler__closer r � handle_close�r r r r r! M s zRequestHandler.handle_closec C s | j �|� d S r )r �append)r �datar r r �collect_incoming_dataQ s z$RequestHandler.collect_incoming_datac @ s e Zd ZdS )zRequestHandler.LoadErrorN��__name__�
__module__� __qualname__r r r r � LoadErrorV 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_terminator^ 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
r1 r �str� traceback�
format_exc�
splitlinesr r r3 r
r r0 r/ )r Ze1Ze2r5 r6 r r r � handle_error� s
zRequestHandler.handle_errorN) r' r( r) r r r! r% r0 r* r7 r< r r r r r 9 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_TIMEr r> �callableZpoll2�hasattr�selectr1 �debug�floatr0 �args�errnoZENOTCONNZEBADF�infor8 r Z exception�EMFILE�sum�values�critical)Zactive�timeout�use_poll� err_countr> r6 r r r �loop� sD
�� rR 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= ) r �
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 ) NrT r r@ zAccept socket error: %sr+ z.Too many acceptor errors - stop logging errorsrA r r? rB )rT r0 rZ r1 �warningr r4 r rH rI rK rL rM rN �stoprS � _AsyncServer__markCloseOnExecr rV )r r Zaddrr6 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>)rO rP rQ F) � threading�current_thread�_AsyncServer__workerrW �os�path�existsr1 r r] � _remove_sock�AsyncServerExceptionZ
create_socketr ZAF_UNIXZ SOCK_STREAMZset_reuse_addrZbindr0 rS r_ r= rX ra rY r[ rR rZ r^ )r �sock�forcerO rP 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 �rY r r" r r rb % rc z#AsyncServer.close.<locals>.<lambda>r TzRemoved socket file zSocket shutdown)rY ra r r r r r rU r rd re rf r �wait_forrX rg rh ri rW rj r1 rF )r Zstopflgr r" r r s$
zAsyncServer.closec C s | j rt�d� d | _ d S )Nz Stop communication, shutdown)rV r1 rF r" r r r �stop_communication2 s
z AsyncServer.stop_communicationc C s | � � | �� d S r )rq r r" r r r r^ : s zAsyncServer.stopc C s | j S r ro r" r r r �isActive? s zAsyncServer.isActivec
C sH zt �| j� W n2 tyB } z|jtjkr.� W Y d }~n
d }~0 0 d S r )rg �removerW �OSErrorrI ZENOENT)r r6 r r r rj E s
zAsyncServer._remove_sockc C s0 | � � }t�|tj�}t�|tj|tjB � d S r )�fileno�fcntlZF_GETFDZF_SETFDZ
FD_CLOEXEC)rl �fd�flagsr r r Z__markCloseOnExecR s z AsyncServer.__markCloseOnExec)NF)r' r( r) r r\ r` rn r rq r^ rr rj � staticmethodr_ r r r r rS � s
rS c @ s e Zd ZdS )rk Nr& r r r r rk \ s rk )NFN)#�
__author__Z
__copyright__Z __license__�pickler r r r � ImportError�compatr rI rv rg r �sysrd r9 Zutilsr Zprotocolr
Zhelpersr r r
r' r1 r r rR rU rS r0 rk r r r r �<module> s6 X
6