| Current File : //lib/python3.9/site-packages/firewall/core/io/__pycache__/helper.cpython-39.pyc |
a
���g2 � @ s� g d �Z ddlmZ ddlZddlZddlZddlmZ ddlm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZ ddlmZ G dd � d e �ZG d
d � d e
�Zd d
� Zddd�ZdS ))�Helper�
helper_reader�
helper_writer� N)�config)� IO_Object�IO_Object_ContentHandler�IO_Object_XMLGenerator�
check_port� check_tcpudp)�log)�errors)�
FirewallErrorc sv e Zd ZdddddddgffZdZd d
gZd d d gd
�Zg d�ddgd�Z� fdd�Zdd� Z dd� Z
dd� Z � Z S )r )�version� )�shortr )� descriptionr )�familyr )�moduler �ports)r r z (sssssa(ss))�-�.Nr )r r �helper)�namer r �port�protocol)r r c s6 t t| ��� d| _d| _d| _d| _d| _g | _d S �Nr ) �superr �__init__r r r r r r ��self�� __class__� �;/usr/lib/python3.9/site-packages/firewall/core/io/helper.pyr : s zHelper.__init__c C s. d| _ d| _d| _d| _d| _| jd d �= d S r )r r r r r r r r" r" r# �cleanupC s zHelper.cleanupc C s( ddg}||vr$t tjd||f ��d S )NZipv4Zipv6z'%s' not in '%s')r
r Z INVALID_IPV)r ZipvZipvsr" r" r# � check_ipvK s
�zHelper.check_ipvc C sv |dkr,|D ] }t |d � t|d � q nF|dkrr|�d�sNttjd| ��t|�dd��dk rrttjd| ��d S ) Nr r � r �
nf_conntrack_�('%s' does not start with 'nf_conntrack_'r �Module name '%s' too short)r r
�
startswithr
r �INVALID_MODULE�len�replace)r r �itemZ
all_configZall_io_objectsr r" r" r# �
_check_configQ s
��zHelper._check_config)
�__name__�
__module__� __qualname__ZIMPORT_EXPORT_STRUCTUREZDBUS_SIGNATUREZADDITIONAL_ALNUM_CHARSZ PARSER_REQUIRED_ELEMENT_ATTRSZ PARSER_OPTIONAL_ELEMENT_ATTRSr r$ r% r/ �
__classcell__r" r" r r# r % s( ��� r c @ s e Zd Zdd� ZdS )�helper_ContentHandlerc C s> t �| ||� | j�||� |dkr�d|v r8|d | j_d|v r\| j�|d � |d | j_d|v r�|d �d�s�tt j
d|d ��t |d � dd��dk r�tt j
d |d ��|d | j_
nz|d
kr�np|d kr�nf|d k�r:t|d � t|d
� |d |d
f}|| jjv�r$| jj�|� nt�d|d |d
� d S )Nr r r r r' r( r r& r) r r r r z#Port '%s/%s' already set, ignoring.)r � startElementr. Zparser_check_element_attrsr r% r r* r
r r+ r, r- r r r
r �appendr Zwarning)r r �attrs�entryr" r" r# r5 b sF ��
�
�z"helper_ContentHandler.startElementN)r0 r1 r2 r5 r" r" r" r# r4 a s r4 c C s t � }| �d�s ttjd| ��| d d� |_|�|j� | |_||_|� t
j �rVdnd|_ |j |_
t|�}t�� }|�|� d|| f }t|d��n}t�d �}|�|� z|�|� W n: tjy� } z ttjd|�� ��W Y d }~n
d }~0 0 W d � n1 �s
0 Y ~~|S ) Nz.xmlz'%s' is missing .xml suffix���FT�%s/%s�rbznot a valid helper file: %s)r �endswithr
r Z INVALID_NAMEr Z
check_name�filename�pathr* r �
ETC_FIREWALLDZbuiltin�defaultr4 �saxZ make_parserZsetContentHandler�openZ InputSourceZ
setByteStream�parseZSAXParseExceptionZINVALID_HELPERZ getException) r= r> r �handler�parserr �f�source�msgr" r" r# r � s:
�
��:r c
C sL |r|n| j }| jr$d|| jf }nd|| jf }tj �|�r�zt�|d| � W n2 ty� } zt� d||� W Y d }~n
d }~0 0 tj �
|�}|� t j
�r�tj �|�s�tj �t j
�s�t�t j
d� t�|d� tj|ddd�}t|�}|�� i }| j|d < | j�r | jd
k�r | j|d < | j�r<| jd
k�r<| j|d < |�d
|� |�d� | j�r�| jd
k�r�|�d� |�di � |�| j� |�d� |�d� | j�r�| jd
k�r�|�d� |�di � |�| j� |�d� |�d� | j D ]4} |�d� |� d| d | d d�� |�d� �q�|�d
� |�d� |� � |� � ~d S )Nr: z %s/%s.xmlz%s.oldz Backup of file '%s' failed: %si� ZwtzUTF-8)�mode�encodingr r r r r �
z r r r r r& )r r ) r> r= r �os�exists�shutil�copy2� Exceptionr �error�dirnamer* r r? �mkdir�iorB r Z
startDocumentr r r r5 ZignorableWhitespacer Z
charactersZ
endElementr r Z
simpleElementZ endDocument�close)
r r> �_pathr rH �dirpathrF rD r7 r r" r" r# r � s\ $
r )N)�__all__Zxml.saxrA rL rT rN Zfirewallr Zfirewall.core.io.io_objectr r r r r
Zfirewall.core.loggerr r Zfirewall.errorsr
r r4 r r r" r" r" r# �<module> s <#