ruạṛ
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
cải xoăn