File: //usr/lib/python3.9/site-packages/fail2ban/server/__pycache__/asyncserver.cpython-39.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 Jaquier�GPL� )�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 �EMPTY�joinZCLOSE�close_when_doner � Exception�logSysr �getEffectiveLevelr �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 r3 r �str� traceback�
format_exc�
splitlinesr r r6 r r r2 r1 )r �e1�e2r8 r9 r r r � handle_error� s
zRequestHandler.handle_errorN) r( r) r* r r! r" r& r2 r+ r: rA 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 rC �callableZpoll2�hasattr�selectr3 �debug�floatr2 �args�errnoZENOTCONN�EBADF�infor; r � exception�EMFILE�sum�values�critical)Zactive�timeout�use_poll� err_countrC r9 r r r �loop� sD
�� rY 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 )�acceptrB ) 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 ) Nr[ r rE zAccept socket error: %sr, z.Too many acceptor errors - stop logging errorsrF r rD rG )r[ r2 ra r3 �warningr r7 r rM rN rR rS rT rU �stoprZ � _AsyncServer__markCloseOnExecr r] )r r �addrr9 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>)rV rW rX F) � threading�current_thread�_AsyncServer__workerr^ �os�path�existsr3 r rd � _remove_sock�AsyncServerExceptionZ
create_socketr �AF_UNIX� SOCK_STREAMZset_reuse_addr�bindr2 rZ rf rB r_ ri r` rb rY ra re )r �sock�forcerV rW 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 �r` r r# r r rj % rk z#AsyncServer.close.<locals>.<lambda>r TzRemoved socket file zSocket shutdown)r` ri r r r r r r\ r rl rm rn r
�wait_forr_ ro rp rq r^ rr r3 rK )r Zstopflgr r# r r s$
zAsyncServer.closec C s | j rt�d� d | _ d S )Nz Stop communication, shutdown)r] r3 rK r# r r r �stop_communication2 s
z AsyncServer.stop_communicationc C s | � � | �� d S r )r| r r# r r r re : s zAsyncServer.stopc C s | j S r rz 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 )ro �remover^ �OSErrorrN �ENOENT)r r9 r r r rr E s
zAsyncServer._remove_sockc C s0 | � � }t�|tj�}t�|tj|tjB � d S r )�fileno�fcntlZF_GETFDZF_SETFDZ
FD_CLOEXEC)rw �fd�flagsr r r Z__markCloseOnExecR s z AsyncServer.__markCloseOnExec)NF)r( r) r* r rc rh ry r r| re r} rr � staticmethodrf r r r r rZ � s
rZ c @ s e Zd ZdS )rs Nr' r r r r rs \ s rs )NFN)#�
__author__�
__copyright__� __license__�pickler r r r � ImportError�compatr rN r� ro r �sysrl r< �utilsr
�protocolr Zhelpersr r
r r( r3 r r rY r\ rZ r2 rs r r r r �<module> s6 X
6