3
UB(dG? � @ sR d Z dZdZddlZddlZddlmZmZ ddlm Z ee
�Z G dd � d �Z dS )
z
Cyril Jaquierz Copyright (c) 2004 Cyril JaquierZGPL� N� )� getLogger�logging)�versionc @ s> e Zd Zdd� Zdd� Zdd� Zddd �Zd
d � Zd d
� ZdS )� Transmitterc C s || _ d| _d S )Nr )�_Transmitter__server�_Transmitter__quiet)�selfZserver� r
�!/usr/lib/python3.6/transmitter.py�__init__, s zTransmitter.__init__c C sp t jdd|� y| j|�}d|f}W nF tk
rj } z*t jd||t j� tjkd� d|f}W Y d d }~X nX |S )N� z Command: %rr z"Command %r has failed. Received %r)�exc_info� )�logSys�log� _Transmitter__commandHandler� Exception�errorZgetEffectiveLevelr �DEBUG)r �command�retZack�er
r
r �proceed6 s
zTransmitter.proceedc
C s |d }|dkrdS |dkrr|d }|dkr:t d|f ��y |d }W n tk
r^ d }Y nX | jj||� |S |d
kr�| j|dd � d �S |d kr�| j|dd � �S |d
kr�|d }| jj|� d S |dk�r t|�dkr�| jj� n.|d dk�r| jj� n|d }| jj |� d S |dk�r�|dd� }d| _
z4| jj |dd i� x |d D ]}| j |� �qVW W d d| _
| jj |ddi� X dS |dk�r�t|�dk�r�|dd � }t|�dk�r�|d dk�r�| jj
� S | jj
d |�S |dk�r| jjd |dd � �S |dk�r|dd � S |dk�r2tjd� dS |dk�rrd| _
z"x |d D ]}| j |� �qNW W d d| _
X d S |dk�r�|d }tjt|�� d S |d k�r�| jj� S |d k�r�| j|dd � �S |d k�r�| j|dd � �S |d k�r�tjS |d k�rtj|d � d S t d!��d S )"Nr ZpingZpong�addr z--allzReserved name %rr �autoz multi-setT�set�start�stop�reload� ZbeginFZOKZunban�bannedZechoz
server-statusz
Status: readyz Server readyz
server-stream�sleepZ flushlogs�get�statusr z config-errorzInvalid command)r �
IndexErrorr ZaddJail�_Transmitter__commandSetZ startJail�len�quitZ stopAllJailZstopJailr Z reloadJailsr �
setUnbanIPr! r �debug�timer" �floatZ flushLogs�_Transmitter__commandGetr$ r r )r r �nameZbackendZopts�cmd�valuer
r
r Z__commandHandlerH s�
z Transmitter.__commandHandlerFc C sF |d }|dkr8|d }| j j|� | jr.d S | j j� S |dkrt|d }| j j|�rh| jr^d S | j j� S td���
n�|dkr�|d }| j j|�r�| jr�d S | j j� S td���
n�|dkr�|d }| j j |� | jr�d S |S |d k� r�|d }| j j
|�S |d
k�r@| j j |d � | j j � }|d k�r*d S | j�r6d S |j
S � n�|d k�r�| j j � }|d k�rptjtjd � d S t|d �|_| j�r�d S |jS � n�|d
k�r�| j j � }|d k�r�tjtjd� d S |d |_| j�r�d S |jS � nL|d dk�rN|d dk�r| j j|d� n&|d dk�r.| j j|d� ntd��| j�rBd S | j j|�S |d dk�r�|d }| j j||� | j�r~d S | j j|�S |d dk�r�x$|dd � D ]}| j j||� �q�W | j�r�d S | j j|�S |d dk�r|d }| j j||� | j�rd S | j j|�S |d dk�rN|d }| j j ||� | j�rBd S | j j |�S |d dk�r�|d }| j j ||� | j�r~d S | j j |�S |d dk�r$|d }d}t |�d k�r�|d j!� d k�r�d}n|d j!� d k�r�t"d ��nt |�d k�r�t"d!��| j j#|||� | j�rd S | j j$|�S |d d"k�r`|d }| j j%||� | j�rTd S | j j$|�S |d d#k�r�|d }| j j&||� | j�r�d S | j j'|�S |d d$k�r�|dd � }| j j(||� | j�r�d S | j j)|�S |d d%k�r |dd � }| j j*||� | j�rd S | j j)|�S |d d&k�rj|d }| j j+||� | j�rLd S | j j,|�}|�rf|j-� S d'S |d d(k�r�|d }| j j.|||d)� |�r�dS | j�r�d S | j j/|�S |d d*k�r�t|d �}| j j0||� | j�r�d S | j j/|�S |d d+k�r>|d }| j j1|||d)� |�r&dS | j�r2d S | j j2|�S |d d,k�r~t|d �}| j j3||� | j�rrd S | j j2|�S |d d-k�r�|d }| j j4||� | j�r�d S | j j5|�S |d d.k�r�|d }| j j6||� | j�r�d S | j j7|�S |d d/k�r2|d }| j j8||� | j�r&d S | j j9|�S |d d0k�rn|d }| j j:||� | j�rbd S | j j;|�S |d d1k�r�|d }| j j<|t|�� | j�r�d S | j j=|�S |d d2k�r�|d }| j j>|t|�� | j�r�d S | j j?|�S |d d3k�r.|d }| j j@|t|�� | j�r"d S | j jA|�S |d d4k�rj|d }| j jB||� | j�r^d S | j jC|�S |d d5k�r�|dd � }| j�r�d S | j jD|f|�� S |d jEd6��r�|d }|d t d6�d � }| j jF|||� | j�r�d S | j jG||�S |d d7k� r |dd � }| j jH||�S |d d8k� rpd} |d d9k� rN|dd � }nd} |d d � }| j jI||| d:�S |d d;k� r�|d g}
t |�d k� r�|
jJ|d tKjL|d �g� | j jM|f|
�� | j� r�d S |
d S |d d<k�
r |d }| j jN||� d S |d d=k� r0|d } | j jO|| �} |�
r�x�|d D ]t}
tjd>d?|
� |
d }tPtQ| |d ���
r�t |
�dk�
r|tKjL|
d �ni }tQ| |�f |� n|
d }tR| ||� �
q4W dS |d }tPtQ| |d ��� rt |�d k�
r�tKjL|d �ni }| j�
r�d S tQ| |�f |�S |d }tR| ||� | j� r&d S tQ| |�S td@|d f ��d S )ANr �loglevelr � logtargetzFailed to change log target� syslogsocketz Failed to change syslog socketZ allowipv6�thread�dbfile� dbmaxmatchesz6dbmaxmatches setting was not in effect since no db yet�
dbpurgeagez4dbpurgeage setting was not in effect since no db yetZidler ZonTZoffFz*Invalid idle option, must be 'on' or 'off'�
ignoreselfZ addignoreipZ delignoreip�
ignorecommand� ignorecacheZ
addlogpath� r �tail�headz$File option must be 'head' or 'tail'z$Only one file can be added at a timeZ
dellogpath� logencodingZaddjournalmatchZdeljournalmatch� prefregex� Z addfailregex)�multipleZ delfailregexZaddignoreregexZdelignoreregex�usedns�findtime� datepattern� logtimezone�
maxmatches�maxretry�maxlines�bantimeZattemptzbantime.�banipZunbanipz--report-absent)�ifexistsZ addactionZ delaction�actionr
z %rz9Invalid command %r (no set action or not yet implemented))Sr Z setLogLevelr � getLogLevelZ setLogTarget� getLogTargetr ZsetSyslogSocket�getSyslogSocketZsetIPv6IsAllowedZsetThreadOptionsZ setDatabase� getDatabase�filenamer r r ZMSG�int�
maxMatches�purgeageZ setIdleJailZ getIdleJailZ
setIgnoreSelf�
getIgnoreSelfZ addIgnoreIP� getIgnoreIPZ delIgnoreIPZsetIgnoreCommand�getIgnoreCommandZsetIgnoreCache�getIgnoreCacher' �lower�
ValueErrorZ
addLogPath�
getLogPathZ
delLogPathZsetLogEncoding�getLogEncodingZaddJournalMatch�getJournalMatchZdelJournalMatchZ setPrefRegex� getPrefRegex�getRegexZ addFailRegex� getFailRegexZ delFailRegexZaddIgnoreRegex�getIgnoreRegexZdelIgnoreRegexZ setUseDns� getUseDnsZ setFindTime� getFindTimeZsetDatePattern�getDatePatternZsetLogTimeZone�getLogTimeZoneZ
setMaxMatches�
getMaxMatchesZ setMaxRetry� getMaxRetryZ setMaxLines� getMaxLinesZ
setBanTime�
getBanTimeZ addAttemptIP�
startswithZsetBanTimeExtra�getBanTimeExtraZsetBanIPr) �extend�json�loadsZ addActionZ delAction� getAction�callable�getattr�setattr)r r rA r. r0 �dbr< �v�optrK �args�
actionnamerL r/ Z actionkey� actionvaluer
r
r Z __commandSet� s&
zTransmitter.__commandSetc s |d }|dkr| j j� S |dkr,| j j� S |dkr>| j j� S |dkrP| j j� S |dkrx| j j� }|d krnd S |jS �n�|dkr�| j j� }|d kr�d S |jS �nd|dkr�| j j� }|d kr�d S |jS �n<|d d
k� r�| j j ||d d � �S |d d k�r| j j
|�S |d d
k�r | j j |�S |d dk�r:| j j |�S |d dk�rT| j j
|�S |d dk�rn| j j|�S |d dk�r�| j j|�S |d dk�r�| j j|�S |d dk�r�| j j|�}|�r�|j� S dS |d dk�r�| j j|�S |d dk�r| j j|�S |d dk�r | j j|�S |d dk�r6| j j|�S |d dk�rP| j j|�S |d dk�rj| j j|�S |d dk�r�| j j|�S |d d k�r�| j j|�S |d d k�r�| j j|�S |d d k�r�| j j |�S |d d k�r| j j |t |�d k�o |d d kd!�S |d j d"��r8|d t d"�d � }| j j ||�S |d d#k�rZt!| j j"|�j#� �S |d d$k�r�|d }|d% }| j j$||�� t%� |�S |d d&k�r�|d }| j j$||�� � fd'd(�t&� �D �S |d d)k�r|d }| j j$||�� � fd*d(�t&� �D �S t'd+��d S ),Nr r1 r2 r3 r4 r5 r6 r7 r r! r Zlogpathr> Z journalmatchr8 Zignoreipr9 r: r? r@ Z failregexZ ignoreregexrB rC rD rE rF rG rH rI rJ z --with-time)ZwithTimezbantime.ZactionsrL r Zactionpropertiesc s, g | ]$}|j d � rtt� |�� r|�qS )�_)rj rp rq )�.0�key)rL r
r �
<listcomp>� s z,Transmitter.__commandGet.<locals>.<listcomp>Z
actionmethodsc s* g | ]"}|j d � rtt� |��r|�qS )ry )rj rp rq )rz r{ )rL r
r r| � s z6Invalid command (no get action or not yet implemented))(r rM rN rO ZgetThreadOptionsrP rQ rS rT r! r[ r\ r] rU rV rW rX r^ r_ r` ra rb rc rd re rf rg rh ri Z
getBanListr' rj rk �listZ
getActions�keysro rq �dirr )r r r. rs rt ru rw rx r
)rL r Z __commandGet� s�
zTransmitter.__commandGetc C sn t |�dkr| jj� S t |�dkr6|d }| jj|�S t |�dkrb|d }|d }| jj||d�S td��d S )Nr r r )�flavorzInvalid command (no status))r' r r$ Z
statusJailr )r r r. r� r
r
r r$ s
zTransmitter.statusN)F) �__name__�
__module__� __qualname__r r r r& r- r$ r
r
r
r r % s
R
er )
�
__author__Z
__copyright__Z __license__r+ rm Zhelpersr r r@ r r� r r r
r
r
r �<module> s |