CasperSecurity
o
;s*b�' � @ 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 Jaquier�GPL� )�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__buffer�set_terminatorr �END)�self�conn� transmitter� r �=/usr/lib/python3/dist-packages/fail2ban/server/asyncserver.pyr 5 s
zRequestHandler.__init__c C sH | j r"| j }d | _ z
|�tj� |�� W d S tjy! Y d S w d S r )r �shutdown�socket� SHUT_RDWR�close�error)r r r r r �__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 @ � e Zd ZdS )zRequestHandler.LoadErrorN��__name__�
__module__� __qualname__r r r r � LoadErrorP � r- c
C s zZ| j }g | _ tj�|�}|tjkr| �� W d S zt|�}W n ty= } ztj d|t�
� t j k d� t
�|��d }~ww | jrH| j�|�}ndg}t|t�}| �|tj � W d S ty� } z)t|t
j�sutj d|t�
� t j kd� td| t�}| �|tj � W Y d }~d S d }~ww )Nz$PROTO-error: load message failed: %s��exc_info�SHUTDOWNz Caught unhandled exception: %r� ERROR: %s)r r �EMPTY�join�CLOSE�close_when_doner � Exception�logSysr �getEffectiveLevelr
�DEBUGr
r- r �proceedr r �pushr �
isinstance)r �message�er r r �found_terminatorX s<
�
��
� ��z RequestHandler.found_terminatorc
C s� z(t � \}}t�dt|� � t�t�� �� � td| t�}| � |t
j � W n t y: } zW Y d }~nd }~ww | �
� d S )Nz"Unexpected communication error: %sr2 )r r8 r �str� traceback�
format_exc�
splitlinesr r r< r r r7 r6 )r �e1�e2r> r? r r r � handle_error~ s
�� zRequestHandler.handle_errorN) r* r+ r, r r" r# r' r7 r- r@ rG r r r r r
3 s
&r
Fc
C s� |si }d|d< |du rt j}tj}t|�r|}n|r2tjr2ttjd�r2t� d� t
|�d }tj}| � r�z||� |d rF|d d8 < W n{ t y� } zo| � sYW Y d}~dS |d d7 < |d dk r�|j d t
jt
jfv r{t�d t|�� n=t�d t|�� n4|d dkr�t�|� t�d
� n#|d d kr�|j d t
jks�t|�� �dkr�t�d |� W Y d}~dS W Y d}~nd}~ww | � s5dS dS )
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 �DEFAULT_SLEEP_TIME�asyncorerI �callable�poll2�hasattr�selectr8 �debug�floatr7 �args�errno�ENOTCONN�EBADF�inforA r � exception�EMFILE�sum�values�critical)�active�timeout�use_poll� err_countrI r? r r r �loop� sH
�
���rd 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 )�acceptrH ) rO �
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 ng tyo } z[| jd d7 < | jd dk r.tjd|| jd dkd� n6| jd dkr;t�d� n)| jd dkrdt|tj�rP|jd t j
ksYt | j� � �d krdt�
d
| j� | �� W Y d }~d S d }~ww | jd r~| jd d8 < t�|� t|| j� d S ) Nrf r rK zAccept socket error: %sr/ z.Too many acceptor errors - stop logging errorsrL r rJ rM )rf r7 rl r8 �warningr r= r rV rW r\ r] r^ r_ �stopre � _AsyncServer__markCloseOnExecr
rh )r r �addrr? r r r �
handle_accept� s0 �
���
zAsyncServer.handle_acceptNFc s� t �� � _|� _tj�|�r#t�d� |r t� d� � �
� nt d��� � t
jt
j� � �� z� �|� W n tyD t d� j ��w 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> s z#AsyncServer.start.<locals>.<lambda>)ra rb rc F) � threading�current_thread�_AsyncServer__workerri �os�path�existsr8 r rp � _remove_sock�AsyncServerException�
create_socketr �AF_UNIX� SOCK_STREAM�set_reuse_addr�bindr7 re rr rH rj ru rk rm rd rl rq )r �sock�forcera rb r r$ r �start� s.
�
zAsyncServer.startc s� d}� j r8d� _� jr z � j�tj� W n
tjy Y nw tj�� � t �
� � j kr6t �
� fdd�d� d}� jrNtj�� j�rN� �� t�d� j � |rUt�d� d� _ d S )NFc s � j S r �rk r r$ r r rv s z#AsyncServer.close.<locals>.<lambda>r TzRemoved socket file zSocket shutdown)rk ru r r r r rO rg r rw rx ry r �wait_forrj rz r{ r| ri r} r8 rT )r �stopflgr r$ r r s& �
zAsyncServer.closec C s | j r
t�d� d | _ d S d S )Nz Stop communication, shutdown)rh r8 rT r$ r r r �stop_communication, s
�z AsyncServer.stop_communicationc C s | � � | �� d S r )r� r r$ r r r rq 4 s zAsyncServer.stopc C s | j S r r� r$ r r r �isActive9 s zAsyncServer.isActivec
C sH z t �| j� W d S ty# } z|jtjkr� W Y d }~d S d }~ww r )rz �removeri �OSErrorrW �ENOENT)r r? r r r r} ? s ���zAsyncServer._remove_sockc C s0 | � � }t�|tj�}t�|tj|tjB � d S r )�fileno�fcntl�F_GETFD�F_SETFD�
FD_CLOEXEC)r� �fd�flagsr r r �__markCloseOnExecL s z AsyncServer.__markCloseOnExecrn )r* r+ r, r ro rt r� r r� rq r� r} � staticmethodrr r r r r re � s
re c @ r( )r~ Nr) r r r r r~ V r. r~ )NFN)!�
__author__�
__copyright__� __license__�pickler r r r rO rW r� rz r �sysrw rB �utilsr �protocolr �helpersr
r r r* r8 r r
rd rg re r7 r~ r r r r �<module> s,
X6