a
=*fW( @ 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 eeZG d
d dej Z dddZ G dd de
j Z G dd de!Z"dS )z
Cyril Jaquierz Copyright (c) 2004 Cyril JaquierZGPL )dumpsloadsHIGHEST_PROTOCOLN )asynchat)asyncore )Utils)CSPROTO)logging getLoggerformatExceptionInfoc @ sL e Zd Zdd Zdd Zdd Zdd ZG d d
d
eZd 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)selfconn 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 shutdownsocket SHUT_RDWRcloseerror)r r r r r Z__closeC s zRequestHandler.__closec C s | tj| d S r )_RequestHandler__closer r handle_closer 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 ty( } zPt|t
jstj d|t
t j kd td| t}| |tj W Y d }~n
d }~0 0 d S )Nz$PROTO-error: load message failed: %sexc_infoZSHUTDOWNz Caught unhandled exception: %r ERROR: %s)r r
ZEMPTYjoinZCLOSEclose_when_doner ExceptionlogSysr ZgetEffectiveLevelr DEBUGr r* r Zproceedr r pushr
isinstance)r messageer r r found_terminator^ s4
z RequestHandler.found_terminatorc
C s zPt \}}tdt| tt 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 e1e2r5 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|rdtjrdttjdrdt d t
|d }tj}| rz$|| |d r|d d8 < W qd t y } z| sW Y d}~q|d d7 < |d dk r|j d t
jt
jfv rtd t| ntd t| np|d dkr2t| td
nL|d d kr~|j d t
jksdt| dkr~td | W Y d}~qW Y d}~qdd}~0 0 qddS )
zCustom 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 listenNpollz"Server listener (select) uses poll r z Server connection was closed: %sz0Too many errors - stop logging connection errorsd +Too many errors - critical count reached %r)r ZDEFAULT_SLEEP_TIMEr r@ callableZpoll2hasattrZselectr1 debugfloatr0 argserrnoZENOTCONNEBADFinfor8 r exceptionEMFILEsumvaluescritical)Zactivetimeoutuse_poll err_countr@ r6 r r r loop sD
rU 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__errCountonstart)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 dkrvtd nR| jd dkrt|tjr|jd t j
kst | j d krt
d
| j | W Y d }~d S d }~0 0 | jd r| jd d8 < t| t|| j d S ) NrW r rB zAccept socket error: %sr+ z.Too many acceptor errors - stop logging errorsrC r rA rD )rW r0 r] r1 warningr r4 r rI rJ rN rO rP rQ stoprV _AsyncServer__markCloseOnExecr rY )r r addrr6 r r r
handle_accept s0
zAsyncServer.handle_acceptNFc s t _| _tj|rFtd |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 z#AsyncServer.start..)rR rS rT F) threadingcurrent_thread_AsyncServer__workerrZ ospathexistsr1 r r` _remove_sockAsyncServerExceptionZ
create_socketr ZAF_UNIXZ SOCK_STREAMZset_reuse_addrZbindr0 rV rb r? r[ re r\ r^ rU r] ra )r sockforcerR rS r r" r start s,
zAsyncServer.startc s d} j rpd _ jr>z jtj W n tjy< Y n0 tj t
j krlt
fddd d} jrtj jr td j |rtd d _ d S )NFc s j S r r\ r r" r r rf % rg z#AsyncServer.close..r TzRemoved socket file zSocket shutdown)r\ re r r r r r rX r rh ri rj r wait_forr[ rk rl rm rZ rn r1 rG )r Zstopflgr r" r r s$
zAsyncServer.closec C s | j rtd d | _ d S )Nz Stop communication, shutdown)rY r1 rG r" r r r stop_communication2 s
z AsyncServer.stop_communicationc C s | | d S r )ru r r" r r r ra : s zAsyncServer.stopc C s | j S r rs 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 )rk removerZ OSErrorrJ ENOENT)r r6 r r r rn E s
zAsyncServer._remove_sockc C s0 | }t|tj}t|tj|tjB d S r )filenofcntlZF_GETFDZF_SETFDZ
FD_CLOEXEC)rp fdflagsr r r Z__markCloseOnExecR s z AsyncServer.__markCloseOnExec)NF)r' r( r) r r_ rd rr r ru ra rv rn staticmethodrb r r r r rV s
rV c @ s e Zd ZdS )ro Nr& r r r r ro \ s ro )NFN)#
__author__Z
__copyright__Z __license__pickler r r r ImportErrorcompatr rJ r{ rk r sysrh r9 Zutilsr Zprotocolr
Zhelpersr r r
r' r1 r r rU rX rV r0 ro r r r r s6 X
6