a
=*f> @ s* d Z dZddlZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z ddl
Z
zddlZedZW n dZY n0 e ZedrejrejjdurejjdsejjZned d
d D rd Zed
fddZdd Zdd Zdd Zdd Zdd Z G dd de Z G d d d ej Z d e_!dGd d Z"ej#j$Z%d!d" Z&e&ej#_$ej'j(Z)d#d$ Z*e*ej'_(d%d& Z+d'd( Z,dHd+d,Z-d-d. Z.d/d0 Z/d1d2 Z0d3d4 Z1e2d5ej3Z4e2d6ej3Z5e2d7ej3Z6d8d9 Z7d:d; Z8d0 r
)LANGUAGELC_ALLLC_CTYPELANGUTF-8strictc C sP z t | tr| ||W S | W S ttfyJ |dkr: | |d Y S 0 d S )Nr replace)
isinstancebytesdecodeUnicodeDecodeErrorUnicodeEncodeError)xencerrorsr r r
uni_decode@ s
r c C s t | tst| S | tdS )Nr )r r strr
PREFER_ENCr r r r
uni_stringI s
r" c C s
t | dS )Nr )r r! r r r uni_bytesM s r# c C s t | tst| S | dv S )N)1ZontrueZyes)r r boollower)valr r r _as_boolP s
r) c C s" t dd \} }| jt|fS )z+ Consistently format exception information N )sysexc_info__name__r" )Zclaexcr r r formatExceptionInfoU s r/ c C sR t j| }|dr"|dd }|tddgv rNt jt j| d | }|S )zaCustom function to include directory name if filename is too common
Also strip .py at the end
z.pyNbase__init__.)r pathbasenameendswithsetdirname)sr1 r r r mbasenamea s
r: c @ s" e Zd ZdZdddZdd ZdS ) TraceBackz7Customized traceback to be included in debug messages
Fc C s d| _ || _dS )zInitialize TrackBack metric
Parameters
----------
compress : bool
if True then prefix common with previous invocation gets
replaced with ...
r N)_TraceBack__prev_TraceBack__compress)selfcompressr r r r2 r s zTraceBack.__init__c C s t jddd d }dd |D }dd |D }|d g}|dd D ]@}|d |d d kr||d d d
|d 7 < qF|| qFd d d |D }| jr|}tj| j|f}t
d
d|}|dkrd|t |d }|| _|S )Nd )limitc S s2 g | ]*}t |d tj|d t|d gqS )r r )r: r r4 r8 r r r r r r
s z&TraceBack.__call__..c S s6 g | ].}|d dv s|d ds|d |d gqS )r )Zunittestzlogging.__init__r z /unittestr* )r6 )r er r r rD s r r z,%s>c S s$ g | ] }d t |d |d f qS )z%s:%sr r )r: rC r r r rD s z>[^>]*$r z...) traceback
extract_stackappendjoinr= r r4 commonprefixr< resublen) r> ZftbentriesZ entries_outentryZsftbZ prev_nextZ
common_prefixZcommon_prefix2r r r __call__~ s*
zTraceBack.__call__N)F)r-
__module__ __qualname____doc__r2 rR r r r r r; n s
r; c @ s e Zd ZdZdd Zdd ZdS )FormatterWithTraceBackz}Custom formatter which expands %(tb) and %(tbc) with tracebacks
TODO: might need locking in case of compressed tracebacks
c O s6 t jj| g|R d|i| d|v }t|d| _d S )Nfmtz%(tbc)s)r? )logging Formatterr2 r; _tb)r> rW argskwargsr? r r r r2 s z FormatterWithTraceBack.__init__c C s | |_|_tj| |S N)rZ ZtbctbrX rY format)r> recordr r r r_ s z FormatterWithTraceBack.formatN)r- rS rT rU r2 r_ r r r r rV s rV Fc C sh | r t | jr | | jd |r,dd |_dd tj_tjrdztj W n Y n0 t
d d S )Nr c S s d S r] r r r r r r z!__stopOnIOError..c S s d S r] r r> r r r ra r ) rO handlersZ
removeHandlercloserX
StreamHandlerflush
exitOnIOErrorr+ stderrexit)ZlogSysZlogHndlrr r r __stopOnIOError s
rj c
K s zt | |||fi | W n ttfyV } z |jdkr@t| W Y d}~nd}~0 ty } zlzRd|t|ffddd |D fffD ](}zt | |g|R W q Y q0 qW n Y n0 W Y d}~n
d}~0 0 dS )a Safe log inject to avoid possible errors by unsafe log-handlers,
concat, str. conversion, representation fails, etc.
Used to intrude exception-safe _log-method instead of _log-method
of Logger class to be always safe by logging and to get more-info about.
See testSafeLogging test-case for more information. At least the errors
covered in phase 3 seems to affected in all known pypy/python versions
until now.
Nzlogging failed: %r on %sz
args: %rc S s g | ] }t |qS r )r" )r ar r r rD r z __safeLog..) __origLogBrokenPipeErrorIOErrorerrnorj Exceptionr" )r> levelmsgr[ r\ rE r r r __safeLog s"
rt c
C sP z t | W n> ttfyJ } z"|jdkr4td| W Y d}~n
d}~0 0 dS )zQSafe flush inject stopping endless logging on closed streams (redirected pipe).
rk N)__origLogFlushrn ro rp rj )r> rE r r r __safeLogFlush s
rv c C s$ d| v rd| dd } t| S )zBGet logging.Logger instance with Fail2Ban logger name convention
r3 z fail2ban.%srF )
rpartitionrX getLoggernamer r r rx s rx c C sR z.t | ts| r t| }ntt| }W n tyL td| Y n0 |S )NzInvalid log level %r)r intisdigitgetattrrX upperAttributeError
ValueError)valueZllr r r str2LogLevel s
r %(message)sTc C sn | dkr@| dkrd| }| dkr*d| }qTd| }|rTd| }nd| }|rTd | }|sjt d
d d |}|S )
z(Custom log format for the verbose runs
r z6 | %(module)15.15s-%(levelno)-2d: %(funcName)-20.20s |r* zB +%(relativeCreated)5d %(thread)X %(name)-25.25s %(levelname)-5.5sz %(thread)X %(levelname)-5.5sz %(asctime)-15sz)%(name)-24s[%(process)d]: %(levelname)-7sz %(asctime)s z(?<=\))-?\d+(?:\.\d+)?sc S s dS )Nr9 r )mr r r ra
r z$getVerbosityFormat..)rM rN ) verbosityrW ZaddtimeZpaddingr r r getVerbosityFormat s
r c C s t djddd t| ||S )z>Except hook used to log unhandled exceptions to Fail2Ban log
Zfail2banz Unhandled exception in Fail2Ban:T)r, )rx Zcriticalr+ __excepthook__)exctyper rH r r r
excepthook s r c C s( | sg S t ttdd td| D S )zHelper to split words on any comma, space, or a new line
Returns empty list if input is empty (or None) and filters
out empty entries
c S s g | ] }| qS r )stripr r r r rD r z splitwords..z[\s,]+)listfilterr& rM split)r9 r r r
splitwords s r c C s |ri | |S | S )zHelper to merge dicts.
r r yr r r _merge_dicts s r c C s i | |S )zBHelper to merge dicts to guarantee a copy result (r is never x).
r r r r r _merge_copy_dicts& s r z^([^\[]+)(?:\[(.*)\])?\s*$zf\s*([\w\-_\.]+)=(?:"([^"]*)"|\'([^\']*)\'|([^,\]]*))(?:,|\]\s*\[|$|(?P.+))|,?\s*$|(?P.+)zs(?:[^\[\s]+(?:\s*\[\s*(?:[\w\-_\.]+=(?:"[^"]*"|\'[^\']*\'|[^,\]]*)\s*(?:,|\]\s*\[)?\s*)*\])?\s*|\S+)(?=\n\s*|\s+|$)c C s t | }|std| \}}t }|rt|D ]}|drptd|d|d|ddd f |drtd|d|ddd f |d}|sq6d d
|d d d
D d }| || < q6||fS )Nzunexpected option syntaxZwrngAz+unexpected syntax at %d after option %r: %sr r ZwrngBz'expected option, wrong syntax at %d: %sc S s g | ]}|d ur|qS r] r )r r( r r r rD K s z"extractOptions..r* r )
OPTION_CREmatchr groupsdictOPTION_EXTRACT_CREfinditergroupstartr )optionr Z option_nameZoptstrZ option_optsZoptmatchoptr r r r extractOptions; s0
r c C s
t | S r] )OPTION_SPLIT_CREfindall)r r r r splitWithOptionsP s r r z <([^ <>]+)>r r c C s t j}| }t|}t }t|d}i }d} | D ]}
|
|v s2|
|v rJq2|r^t||
r^q2t||
} } || }
||
i }|
r|
d}||v r|| |
}
q||
ks||dt krt d|
||| f d}|r||d | }|du r ||}|du r |dur ||}|du r:|| |
}
qt
|tsNt|}| d| |} ||dd ||< || |
}
q| | kr|| r|||
< d } t|t| kr| }| ||
< d
| vr2||
q2| s&qq&|S ) a| Sort out tag definitions within other tags.
Since v.0.9.2 supports embedded interpolation (see test cases for examples).
so: becomes:
a = 3 a = 3
b = _3 b = 3_3
Parameters
----------
inptags : dict
Dictionary of tags(keys) and their values.
Returns
-------
dict
Dictionary of tags(keys) and their values, with tags
within the values recursively replaced.
getRawItemFr zpproperties contain self referencing definitions and cannot be resolved, fail tag: %s, found: %s in %s, value: %sN?z<%s>r T<)TAG_CREsearchr7 hasattrkeyscallabler r" getr endMAX_TAG_REPLACE_COUNTr r r r r idcopyadd)ZinptagsZ conditionalignoreZaddreplZ
tre_searchtagsZdoneZ noRecReplZ repCountsZrepFlagtagr Zorgvalr ZrplcZrtagreplr r r substituteRecursiveTags^ sb
r c C s* z| } td| W n Y n0 dS )zHelper to set real thread name (used for identification and diagnostic purposes).
Side effect: name can be silently truncated to 15 bytes (16 bytes with NTS zero)
N)encode_libcapZprctlry r r r prctl_set_th_name s
r c C s d S r] r ry r r r r s c s< e Zd ZdZe ZdZ fddZdd Zd
dd Z Z
S ) BgServicez{Background servicing
Prevents memory leak on some platforms/python versions,
using forced GC in periodical intervals.
Nc s | j stt| | | _ | j S r] ) _instancesuperr __new__)cls __class__r r r s zBgService.__new__c C s2 d| _ d| _d| _| j| _ttdr.td d S )Ni r@
set_thresholdr )_BgService__serviceTime_BgService__periodTime_BgService__threshold_BgService__countr gcr rb r r r r2 s
zBgService.__init__Fc C s | j d8 _ |s.| j dks*t | jk r.dS tj|s>dS zNt | jk r^W tj dS t t | j
| _| j | _ W tj dS tj 0 dS )Nr r FT) r r timer r _mutexacquirereleaser Zcollectr r )r> forcewaitr r r service s$
zBgService.service)FF) r- rS rT rU r r r r r2 r
__classcell__r r r r r s r )NN)r TT)r r N)>
__author__Z __license__r localerX r rM r+ rH threadingr Z
server.mytimer importlibZctypesZCDLLr getpreferredencodingr
startswithstdoutencodingallr r" r# r) r/ r: objectr; rY rV rg rj ZLoggerZ_logrm rt re rf ru rv rx r r r r r r compileDOTALLr r r r r r r r r r r r r r s~
.
c