File: //usr/lib/python3.6/site-packages/fail2ban/server/__pycache__/asyncserver.cpython-36.pyc
3
TB(d�' � @ 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 jj| |� || _|| _g | _| jtj� d S )N) �asynchat�
async_chat�__init__�_RequestHandler__conn� _RequestHandler__transmitter�_RequestHandler__bufferZset_terminatorr �END)�self�conn� transmitter� r �!/usr/lib/python3.6/asyncserver.pyr 5 s
zRequestHandler.__init__c C sF | j rB| j }d | _ y|jtj� |j� W n tjk
r@ Y nX d S )N)r �shutdown�socket� SHUT_RDWR�close�error)r r r r r Z__close= s zRequestHandler.__closec C s | j � tjj| � d S )N)�_RequestHandler__closer
r � handle_close)r r r r r G s zRequestHandler.handle_closec C s | j j|� d S )N)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( y�| j }g | _ tjj|�}|tjkr0| j� d S y t|�}W nF tk
r� } z*tj d|tj
� t j k d� t
j|��W Y d d }~X nX | jr�| jj|�}ndg}t|t�}| j|tj � W nh tk
�r" } zJt|t
j�s�tj d|tj
� t j kd� td| t�}| j|tj � W Y d d }~X nX d S )Nz$PROTO-error: load message failed: %s)�exc_infoZSHUTDOWNz Caught unhandled exception: %rz 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 s0
z RequestHandler.found_terminatorc C s� yPt � \}}tjdt|� � tjtj� j� � td| t�}| j |t
j � W n" t k
rr } zW Y d d }~X nX | j
� d S )Nz"Unexpected communication error: %sz ERROR: %s)r r+ r �str� traceback�
format_exc�
splitlinesr r r- r r r* r) )r Ze1Ze2r/ r0 r r r � handle_error~ s
zRequestHandler.handle_errorN) r# r$ r% r r r r" r* r& r1 r6 r r r r r 3 s
&r Fc C s� |si }d|d< |dkr t j}tj}t|�r2|}n2|rdtjrdttjd�rdtj d� t
|�d }tj}�x| � �r�y$||� |d r�|d d8 < W qh t k
�r| } z�| � s�P |d d7 < |d dk �r|j d t
jt
jfkr�tjd t|�� ntjd t|�� nd|d dk�r,tj|� tjd
� n@|d d k�rl|j d t
jk�s^t|j� �dk�rltjd |� P W Y dd}~X qhX qhW 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 polli� r � z Server connection was closed: %sz0Too many errors - stop logging connection errors�d z+Too many errors - critical count reached %r)r ZDEFAULT_SLEEP_TIME�asyncorer8 �callableZpoll2�hasattrZselectr+ �debug�floatr* �args�errnoZENOTCONNZEBADF�infor2 r Z exception�EMFILE�sum�values�critical)Zactive�timeout�use_poll� err_countr8 r0 r r r �loop� sB
rJ 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 jj| � || _d| _d| _d| _ddd�| _d | _d S )Nz /var/run/fail2ban/fail2ban.sockFr )�acceptr7 ) 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 y| j � \}}W n� tk
r� } z�| jd d7 < | jd dk r^tjd|| jd dkd� nl| jd dkrxtjd� nR| jd dkr�t|tj�r�|jd t j
ks�t | jj � �d kr�tj
d
| j� | j� d S d }~X nX | jd r�| jd d8 < tj|� t|| j� d S ) NrL r r9 zAccept socket error: %s)r' z.Too many acceptor errors - stop logging errorsr: r i� z+Too many errors - critical count reached %r)rL r* rR r+ �warningr r. r r@ rA rC rD rE rF �stoprK � _AsyncServer__markCloseOnExecr rN )r r Zaddrr0 r r r �
handle_accept� s&
zAsyncServer.handle_acceptNFc s� t j� � _|� _tjj|�rFtjd� |r>tj d� � j
� nt d��� j t
jt
j� � j� y� j|� W n" tk
r� t d� j ��Y nX tj� j
� � jd� d � _ � _� _� jrƈ j� t� fdd�||� jd � d
� _� j � d S ) Nz$Fail2ban seems to be already runningz Forcing execution of the serverzServer already runningzUnable to bind socket %sr Tc s � j S )N)�_AsyncServer__loopr )r r r �<lambda> s z#AsyncServer.start.<locals>.<lambda>)rG rH rI F) � threading�current_thread�_AsyncServer__workerrO �os�path�existsr+ r rU � _remove_sock�AsyncServerExceptionZ
create_socketr ZAF_UNIXZ SOCK_STREAMZset_reuse_addrZbindr* rK rW r7 rP rY rQ rS rJ rR rV )r �sock�forcerG rH r )r r �start� s,
zAsyncServer.startc s� d}� j rrd� _� jr@y� jjtj� W n tjk
r> Y nX tjj� � t j
� � j krnt j
� fdd�d� d}� jr�tjj� j�r�� j� tjd� j � |r�tjd� d� _ d S )NFc s � j S )N)rQ r )r r r rZ s z#AsyncServer.close.<locals>.<lambda>r TzRemoved socket file zSocket shutdown)rQ rY r r r r r; rM r r[ r\ r] r Zwait_forrP r^ r_ r` rO ra r+ r> )r Zstopflgr )r r r s$
zAsyncServer.closec C s | j rtjd� d | _ d S )Nz Stop communication, shutdown)rN r+ r> )r r r r �stop_communication, s
z AsyncServer.stop_communicationc C s | j � | j� d S )N)rf r )r r r r rV 4 s zAsyncServer.stopc C s | j S )N)rQ )r r r r �isActive9 s zAsyncServer.isActivec C sF yt j| j� W n0 tk
r@ } z|jtjkr0� W Y d d }~X nX d S )N)r^ �removerO �OSErrorrA �ENOENT)r r0 r r r ra ? s
zAsyncServer._remove_sockc C s0 | j � }tj|tj�}tj|tj|tjB � d S )N)�fileno�fcntlZF_GETFDZF_SETFDZ
FD_CLOEXEC)rc �fd�flagsr r r Z__markCloseOnExecL s z AsyncServer.__markCloseOnExec)NF)r# r$ r% r rT rX re r rf rV rg ra � staticmethodrW r r r r rK � s
rK c @ s e Zd ZdS )rb N)r# r$ r% r r r r rb V s rb )NFN)!�
__author__Z
__copyright__Z __license__�pickler r r r
r; rA rl r^ r �sysr[ r3 Zutilsr Zprotocolr Zhelpersr r
r r# r+ r r rJ rM rK r* rb r r r r �<module> s* X
6