CasperSecurity
o
;s*bO4 � @ s� d Z dZdZddlZddlZddlmZmZ ddlm Z m
Z
m Z dd l m
Z
mZmZmZ e
e�Zeed
�Zd d � ZG d
d� d�ZG dd� de
�ZG dd� de�ZdS )z?Cyril Jaquier, Yaroslav Halchenko, Serg G. Brester (aka sebres)z\Copyright (c) 2004 Cyril Jaquier, 2007 Yaroslav Halchenko, 2015 Serg G. Brester (aka sebres)�GPL� N)�
NoOptionError�NoSectionError� )�sys� SafeConfigParserWithIncludes�logLevel� )� getLogger�_as_bool� _merge_dicts�substituteRecursiveTags)�bool�intc c sv � t | ttf�r(| D ]}t|�dkr|\}}}n|d\}}}|||fV q
dS | D ]}| | \}}|||fV q*dS )z�Iterator over the options template with default options.
Each options entry is composed of an array or tuple with:
[[type, name, ?default?], ...]
Or it is a dict:
{name: [type, default], ...}
r N)�
isinstance�list�tuple�len)�options�optname�opttype�optvalue� r �>/usr/lib/python3/dist-packages/fail2ban/client/configreader.py�_OptionsTemplateGen) s � � �r c @ s� e Zd ZdZd"dd�Zdd� Zdd� Zed d
� �Zd#d d
�Z d$dd�Z
dd� Z dd� Z d#dd�Z
dd� Zdd� Zd%d d �Zdi fd d �Zd d!� ZdS )&� ConfigReaderzkGeneric config reader class.
A caching adapter which automatically reuses already shared configuration.
Nc K sZ d | _ d | _|d ur
|| _|d ur || _ || _d | _d S | jd u r+tdi |��| _d S d S )Nr )�
_cfg_share�_cfg�_cfg_share_kwargs�_cfg_share_basedir�ConfigReaderUnshared)�self�
use_config� share_config�kwargsr r r �__init__D s
�zConfigReader.__init__c C s | j r | j �|� d S || _d S �N)r �
setBaseDirr �r! �basedirr r r r' R s
zConfigReader.setBaseDirc C s | j r| j �� S | jS r& )r �
getBaseDirr �r! r r r r* X s
zConfigReader.getBaseDirc C � | j S r& )r r+ r r r r# ^ s zConfigReader.share_configTc C sT | j s| �|� |r| j jdur| j jS t�d|| j �� � | j �|�}|| j _|S )z� Overloads a default (not shared) read of config reader.
To prevent mutiple reads of config files with it includes, reads into
the config reader, if it was not yet cached/shared by 'name'.
Nz Loading configs for %s under %s )r �_create_unshared�read_cfg_files�logSys�infor* �read)r! �name�once�retr r r r1 b s
zConfigReader.read� c C s~ | j s4| jdur4| j�|�| _ | j s2tdd| ji| j��| _ | jdur*| j �| j� | j | j|< dS dS tdi | j��| _ dS )z� Allocates and share a config file by it name.
Automatically allocates unshared or reuses shared handle by given 'name' and
init arguments inside a given shared storage.
Nr# r )r r �getr r r r' )r! r2 r r r r- w s
�z ConfigReader._create_unsharedc C s. z dd� | j �� D �W S ty g Y S w )Nc s s � | ]
}|� d �s|V qdS )zKNOWN/N)�
startswith)�.0�nr r r � <genexpr>� s � z(ConfigReader.sections.<locals>.<genexpr>)r �sections�AttributeErrorr+ r r r r; � �
�zConfigReader.sectionsc C s$ z| j �|�W S ty Y dS w )NF)r � has_sectionr< )r! �secr r r r> � s
�zConfigReader.has_sectionc C s& |r | j �||�S || j j�|i �v S r& )r �
has_option� _sectionsr6 )r! r? �opt� withDefaultr r r r@ � s �zConfigReader.has_optionc C s | j �� �|� d S r& )r � get_defaults�update)r! �dr r r �merge_defaults� s zConfigReader.merge_defaultsc O �4 z| j j|g|�R i |��W S ty t|��w r& )r �
merge_sectionr< r �r! �section�argsr$ r r r rI � �
�zConfigReader.merge_sectionFc C s( z| j �||�W S ty t|��w )z�Return a list of option names for the given section name.
Parameter `withDefault` controls the include of names from section `[DEFAULT]`
)r r r< r )r! rK rC r r r r � s
�zConfigReader.optionsc C s. z | j j||||d�W S ty t|��w )N)�raw�vars)r r6 r< r )r! r? rB rN rO r r r r6 � r= zConfigReader.getc O rH r& )r �
getOptionsr< r rJ r r r rP � rM zConfigReader.getOptions)NN)T)r5 �F)�__name__�
__module__� __qualname__�__doc__r% r' r* �propertyr# r1 r- r; r> r@ rG rI r r6 rP r r r r r > s"
r c @ s@ e Zd ZdZdZddd�Zdd� Zdd � Zd
d � Zddd�Z dS )r z�Unshared config reader (previously ConfigReader).
Do not use this class (internal not shared/cached represenation).
Use ConfigReader instead.
z
/etc/fail2banNc O s, t j| g|�R i |�� d | _| �|� d S r& )r r% r. r' )r! r) rL r$ r r r r% � s z ConfigReaderUnshared.__init__c C s |d u rt j}|�d�| _d S )N�/)r �DEFAULT_BASEDIR�rstrip�_basedirr( r r r r' � s z ConfigReaderUnshared.setBaseDirc C r, r& )rZ r+ r r r r* � � z ConfigReaderUnshared.getBaseDirc sT t j�| j�std| j ��|�d�rt j�|�}t j�| j|�}t� d|| j� |d g}|d }|t
t � d| ��7 }|� |d � |t
t � d| ��7 }t
tt jj|��}t|�r�t� d d
�|�� t�| |�� � fd d �|D �}|rt�d
d
�|�� � r�dS t�d|| �� � dS t�d| d| �� dd
�|� gtt|�� � dS )Nz/Base configuration directory %s does not exist �./z Reading configs for %s under %s z.confz.dz %s/*.confz.localz
%s/*.localzReading config files: %sz, c s g | ]}|� vr|�qS r r )r8 �cf��config_files_readr r �
<listcomp>� s z-ConfigReaderUnshared.read.<locals>.<listcomp>z Could not read config files: %sTz0Found no accessible config files for %r under %sFz(Found no accessible config files for %r zunder %szamong existing ones: )�os�path�existsrZ �
ValueErrorr7 �abspath�joinr/ �debug�sorted�glob�appendr �filterr r r1 �errorr* r )r! �filename�basename� config_files�
config_dir�missedr r^ r r1 � sD �
�
�
��zConfigReaderUnshared.readFTc
C s( t � }|d u r i }t|�D ]�\}}} ||v rq
z#| j|||d�}
|
||< |r9t�|�} | r9|
d u r3W q
| |
�||< W q
tyX } z|rE� t�| � | ||< W Y d } ~ q
d } ~ w tyr | d urpt�d||| f � | ||< Y q
t y� t�d| d | d t
| � d � | ||< Y q
w |S )N�rO z/'%s' not defined in '%s'. Using default one: %rzWrong value for 'z' in 'z'. Using default one: '�') �dictr r6 � CONVERTERr r/ rl r �warningrd �repr)
r! r? r �pOptions� shouldExist�convert�valuesr r r �v�conv�er r r rP � sN
�
� �� ��� �z ConfigReaderUnshared.getOptionsr& )NFT)
rR rS rT rU rX r% r' r* r1 rP r r r r r � s
2r c @ sx e Zd ZdZg Zdd� Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z d dd�Z dd� Z
dd� Zd dd�Zdd� Zd S ) �DefinitionInitConfigReaderz�Config reader for files with options grouped in [Definition] and
[Init] sections.
Is a base class for readers of filters and actions, where definitions
in jails might provide custom values for options defined in [Init]
section.
c K sV t j| fi |�� |�d�rtj�|�}| �|� | �|� || _t � | _
t � | _ d S )Nr\ ) r r% r7 ra rb re �setFile� setJailName� _initOptsrt �_pOpts� _defCache)r! �file_�jailName�initOptsr$ r r r r% / s
z#DefinitionInitConfigReader.__init__c C s || _ i | _d S r& )�_filer� )r! �fileNamer r r r� 9 s
z"DefinitionInitConfigReader.setFilec C r, r& )r� r+ r r r �getFile= r[ z"DefinitionInitConfigReader.getFilec C s
|| _ d S r& �� _jailName)r! r� r r r r� @ s
z&DefinitionInitConfigReader.setJailNamec C r, r& r� r+ r r r � getJailNameC r[ z&DefinitionInitConfigReader.getJailNamec C s t �| | j�S r& )r r1 r� r+ r r r r1 F s z DefinitionInitConfigReader.readc C s" | j s | �| j� t�| j | j�S r& )r r- r� r r1 r+ r r r � readexplicitJ s z'DefinitionInitConfigReader.readexplicitFc s |st � }� jrt|� j�}tj� d� j|dd�� _|� _� �d�r_� fdd�}� j ddd�D ]/}|dkr6q/d }|�
d �sL|d u rE||�}|� jd | < |� jvr^|d u rY||�}|� j|< q/|r�� �d�r�� � d�D ]}|dksv|� jv rwqk� � d|�� j|< qkd S d S d S )
N�
DefinitionF)rz �Initc s � � d| �S )Nr� )r6 )rB r+ r r �<lambda>\ s z7DefinitionInitConfigReader.getOptions.<locals>.<lambda>)rC rR zknown/) rt r� r r rP � _configOpts�_optsr� r> r r7 r6 )r! �pOpts�all�getoptrB r| r r+ r rP O s4 �
��z%DefinitionInitConfigReader.getOptionsc C sx t |�D ]5\}}}t�|�}|r9|�|�}|du rqz||�||< W q ty8 t�d||||� |||< Y qw qdS )z:Convert interpolated combined options to expected type.
Nz/Wrong %s value %r for %r. Using default one: %r)r ru r6 rd r/ rv )r! �opts�
configOptsr r r r} r| r r r �convertOptionsm s
� ���z)DefinitionInitConfigReader.convertOptionsc
C sd z| j | W S ty1 z | jjd|| jd�}W n tttfy' d}Y nw || j |< | Y S w )z�Get combined definition option (as string) using pre-set and init
options as preselection (values with higher precedence as specified in section).
Can be used only after calling of getOptions.
r� rr N)r� �KeyErrorr �get_exr� r r rd )r! r r| r r r �
getCombOption| s �
�z(DefinitionInitConfigReader.getCombOptionr c C s� | j }| jr t|| j�}t|�si S t|��� }|D ]}tj�|�}|r/|� � \}}|�
|� qt ||| j d�}|s>t
d��| �|| j� |S )N)�ignore�addreplz/recursive tag definitions unable to be resolved)r� r� r r �set�copyr �CONDITIONAL_RE�match�groups�addr
r� rd r� r� )r! r� � combinedoptsr9 �condr� r r r � getCombined� s&
��z&DefinitionInitConfigReader.getCombinedc C s t �r& )�NotImplementedErrorr+ r r r rz � s z"DefinitionInitConfigReader.convertNrQ )r )rR rS rT rU r� r% r� r� r� r� r1 r� rP r� r� r� rz r r r r r $ s
r )�
__author__�
__copyright__� __license__ri ra � configparserr r �configparserincr r r �helpersr
r r r
rR r/ r ru r r r r r r r r �<module> s �ym