File: //lib/python3/dist-packages/firewall/core/io/__pycache__/helper.cpython-310.pyc
o
bhAb> � @ 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__� �9/usr/lib/python3/dist-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}||vrt tjd||f ��d S )N�ipv4�ipv6z'%s' not in '%s')r
r � INVALID_IPV)r �ipv�ipvsr" r" r# � check_ipvK s
��zHelper.check_ipvc C s| |dkr|D ]}t |d � t|d � qd S |dkr:|�d�s(ttjd| ��t|�dd��dk r<ttjd| ��d S 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 �item�
all_config�all_io_objectsr r" r" r# �
_check_configQ s" �
���zHelper._check_config)
�__name__�
__module__� __qualname__�IMPORT_EXPORT_STRUCTURE�DBUS_SIGNATURE�ADDITIONAL_ALNUM_CHARS� PARSER_REQUIRED_ELEMENT_ATTRS� PARSER_OPTIONAL_ELEMENT_ATTRSr r$ r* r6 �
__classcell__r" r" r r# r % s* ��� r c @ s e Zd Zdd� ZdS )�helper_ContentHandlerc C sJ t �| ||� | j�||� |dkrcd|v r |d | j_d|v r.| j�|d � |d | j_d|v ra|d �d�sCtt j
d|d ��t |d � dd��dk rYtt j
d |d ��|d | j_
d S d S |d
krid S |d krod S |d kr�t|d � t|d
� |d |d
f}|| jjvr�| jj�|� d S t�d|d |d
� d S d S )Nr r r r r, r- r r+ r. r r r r z#Port '%s/%s' already set, ignoring.)r � startElementr3 �parser_check_element_attrsr r* r r/ r
r r0 r1 r2 r r r
r �appendr �warning)r r �attrs�entryr" r" r# rA b sJ ��
�� ��z"helper_ContentHandler.startElementN)r7 r8 r9 rA r" r" r" r# r@ a s r@ c C s t � }| �d�sttjd| ��| d d� |_|�|j� | |_||_|� t
j �r+dnd|_ |j |_
t|�}t�� }|�|� d|| f }t|d��4}t�d �}|�|� z|�|� W n tjys } z
ttjd|�� ��d }~ww W d � ~~|S 1 s�w Y ~~|S ) Nz.xmlz'%s' is missing .xml suffix���FT�%s/%s�rbznot a valid helper file: %s)r �endswithr
r � INVALID_NAMEr �
check_name�filename�pathr/ r �
ETC_FIREWALLD�builtin�defaultr@ �sax� make_parser�setContentHandler�open� InputSource�
setByteStream�parse�SAXParseException�INVALID_HELPER� getException) rM rN r �handler�parserr �f�source�msgr" r" r# r � sJ
�
�����
� � r c
C s: |r|n| j }| jrd|| jf }nd|| jf }tj �|�rCz
t�|d| � W n tyB } z
t� d||� W Y d }~nd }~ww tj �
|�}|� t j
�ritj �|�sitj �t j
�sct�t j
d� t�|d� tj|ddd�}t|�}|�� i }| j|d < | jr�| jd
kr�| j|d < | jr�| jd
kr�| j|d < |�d
|� |�d� | jr�| jd
kr�|�d� |�di � |�| j� |�d� |�d� | jr�| jd
kr�|�d� |�di � |�| j� |�d� |�d� | j D ]} |�d� |� d| d | d d�� |�d� q�|�d
� |�d� |� � |� � ~d S )NrH z %s/%s.xmlz%s.oldz Backup of file '%s' failed: %si� �wtzUTF-8)�mode�encodingr r r r r �
z r r r r r+ )r r ) rN rM r �os�exists�shutil�copy2� Exceptionr �error�dirnamer/ r rO �mkdir�iorU r �
startDocumentr r r rA �ignorableWhitespacer �
characters�
endElementr r �
simpleElement� endDocument�close)
r rN �_pathr r` �dirpathr^ r\ rE r r" r" r# r � s` ��
r )N)�__all__�xml.saxrR re rm rg �firewallr �firewall.core.io.io_objectr r r r r
�firewall.core.loggerr r �firewall.errorsr
r r@ r r r" r" r" r# �<module> s <#