Edit File: zone.cpython-36.pyc
3 Y eYM � @ s d ddgZ ddljZddlZddlZddlZddlmZ ddlm Z m Z m Z m Z m Z mZmZ ddlmZmZ ddlmZmZmZmZ dd lmZmZmZm Z dd l m Z dd l m Z dd lm!Z! dd l"m#Z# G dd � d e�Z$G dd� de�Z%ddd�Z&ddd�Z'dS )�Zone� zone_reader� zone_writer� N)�config)� checkIPnMask� checkIP6nMask�checkInterface�uniqify�max_zone_name_len� u2b_if_py2� check_mac)�DEFAULT_ZONE_TARGET� ZONE_TARGETS)�PY2� IO_Object�IO_Object_ContentHandler�IO_Object_XMLGenerator)�common_startElement�common_endElement�common_check_config� common_writer)�rich)�log)�errors)� FirewallErrorc s` e Zd ZdZd?d@dAdBdCd dgfd dDgfd dgfdEd dFgfddgfddgfddgfddgfddGgfdHdIfZdddgZddddgddgdgdgdddgdgddd gd gddgdddddd gd gdd �Zdd ddgd!gd"d#gd$d%gd d&d'd$d(gd'gd)d*gd+gd,�Zed-d.� �Z � fd/d0�Z d1d2� Z d3d4� Z � fd5d6�Z � fd7d8�Zd9d:� Z� fd;d<�Zd=d>� Z� ZS )Jr z Zone class �version� �short� description�UNUSEDF�target�services�ports� icmp_blocks� masquerade� forward_ports� interfaces�sources� rules_str� protocols� source_ports�icmp_block_inversion�forward�_�-�/N�name�port�protocol�address�value�set)r r �zone�servicer1 z icmp-blockz icmp-typer, z forward-port� interface�rule�source� destinationr2 z source-portr ZauditZaccept�rejectZdropZmark�limitzicmp-block-inversion� immutableZenabledzto-portzto-addr�familyZpriority�mac�invert�ipset�prefix�level�type)r6 r$ z forward-portr9 r: r; r r< c C s8 x&t tj�D ]\}\}}|| kr |S q W ttjd��d S )Nz index_of())� enumerater �IMPORT_EXPORT_STRUCTUREr r Z UNKNOWN_ERROR)�element�iZelZdummy� rJ �/usr/lib/python3.6/zone.py�index_ofc s z Zone.index_ofc s� t t| �j� d| _d| _d| _d| _t| _g | _ g | _ g | _ g | _ d| _ d| _g | _g | _g | _g | _d | _g | _g | _d| _d| _d| _d S )Nr F)�superr �__init__r r r r r r r! r" r) r# r, r$ r% r* r& r' � fw_config�rulesr( r+ �combined�applied)�self)� __class__rJ rK rN j s, z Zone.__init__c C s� d| _ d| _d| _d| _t| _| jd d �= | jd d �= | jd d �= | j d d �= d| _ d| _ | j d d �= | j d d �= | jd d �= | jd d �= d | _| jd d �= | jd d �= d| _d| _d| _d S )Nr F)r r r r r r r! r" r) r# r, r$ r% r* r& r' rO rP r( r+ rQ rR )rS rJ rJ rK �cleanup� s* z Zone.cleanupc C s� t | j�| _t | j�| _t | j�| _t | j�| _dd� | jD �| _dd� | jD �| _dd� | jD �| _dd� | jD �| _dd� | j D �| _ dd� | j D �| _ dd� | j D �| _ d d� | j D �| _ d d� | j D �| _ d d� | jD �| _d S ) z� HACK. I haven't been able to make sax parser return strings encoded (because of python 2) instead of in unicode. Get rid of it once we throw out python 2 support.c S s g | ] }t |��qS rJ )r )�.0�srJ rJ rK � <listcomp>� s z'Zone.encode_strings.<locals>.<listcomp>c S s g | ]\}}t |�t |�f�qS rJ )r )rV �po�prrJ rJ rK rX � s c S s g | ] }t |��qS rJ )r )rV rZ rJ rJ rK rX � s c S s g | ] }t |��qS rJ )r )rV rI rJ rJ rK rX � s c S s0 g | ](\}}}}t |�t |�t |�t |�f�qS rJ )r )rV Zp1Zp2Zp3Zp4rJ rJ rK rX � s c S s g | ]\}}t |�t |�f�qS rJ )r )rV rY rZ rJ rJ rK rX � s c S s g | ] }t |��qS rJ )r )rV rI rJ rJ rK rX � s c S s g | ] }t |��qS rJ )r )rV rW rJ rJ rK rX � s c S s g | ] }t |��qS rJ )r )rV rW rJ rJ rK rX � s c S s g | ] }t |��qS rJ )r )rV rW rJ rJ rK rX � s N)r r r r r r! r" r) r# r% r* r& r' rP r( )rS rJ rJ rK �encode_strings� s zZone.encode_stringsc sN |dkr8dd� |D �| _ tt| �j|dd� | j D �� ntt| �j||� d S )Nr( c S s g | ]}t j|d ��qS ))Zrule_str)r Z Rich_Rule)rV rW rJ rJ rK rX � s z$Zone.__setattr__.<locals>.<listcomp>c S s g | ] }t |��qS rJ )�str)rV rW rJ rJ rK rX � s )rP rM r � __setattr__)rS r0 r4 )rT rJ rK r] � s zZone.__setattr__c s t t| �j� }|d= |S )Nr )rM r �export_config_dict)rS Zconf)rT rJ rK r^ � s zZone.export_config_dictc C sL t | |||� |dkr.|tkr*ttj|���n|dkr�xl|D ]d}t|�sTttj|��| jr<xD| jj� D ]6}|| j krvqf|| jj |�j krfttjdj ||���qfW q<W n�|dk�rHx�|D ]�}t |� r�t|� r�t|� r�|jd� r�ttj|��| jr�xL| jj� D ]>}|| j k�r�q|| jj |�jk�rttjdj ||����qW q�W d S )Nr r& z)interface '{}' already bound to zone '{}'r' zipset:z&source '{}' already bound to zone '{}')r r r r �INVALID_TARGETr ZINVALID_INTERFACErO Z get_zonesr0 Zget_zoner& �formatr r r � startswith� INVALID_ADDRr' )rS r �itemZ all_configr8 r6 r: rJ rJ rK � _check_config� s6 zZone._check_configc s� t t| �j|� |jd�r,ttjd| ��n�|jd�rHttjd| ��n�|jd�dkrhttjd| ��nnd|kr�|d |j d�� }n|}t |�t � kr�ttjd|t |�t � | j f ��| j r�|| j j� kr�ttjd��d S )Nr/ z'%s' can't start with '/'z'%s' can't end with '/'� zmore than one '/' in '%s'z'Zone of '%s' has %d chars, max is %d %sz+Zones can't have the same name as a policy.)rM r � check_namera r r � INVALID_NAME�endswith�count�find�lenr rQ rO Zget_policy_objectsZ NAME_CONFLICT)rS r0 Z checked_name)rT rJ rK rf � s, zZone.check_namec C s� d| _ d | _d| _d| _d| _x$|jD ]}|| jkr&| jj|� q&W x$|jD ]}|| jkrL| jj|� qLW x$|jD ]}|| jkrr| jj|� qrW x$|j D ]}|| j kr�| j j|� q�W x$|j D ]}|| j kr�| j j|� q�W x$|j D ]}|| j kr�| j j|� q�W |j �rd| _ |j �r d| _ x(|jD ] }|| jk�r&| jj|� �q&W x(|jD ] }|| jk�rP| jj|� �qPW x,|jD ]"} | jj| � | jjt| �� �qzW |j�r�d| _d S )NTr )rQ �filenamer r r r& �appendr' r! r" r) r# r, r$ r% r* rP r( r\ r+ ) rS r6 r8 r: r7 r1 �protoZicmpr, r9 rJ rJ rK �combine� sL z Zone.combine)r r )r r )r r )r F)r r )r r )r$ F)r r r r )r r )r+ F)r, F)�__name__� __module__� __qualname__�__doc__rG ZADDITIONAL_ALNUM_CHARSZ PARSER_REQUIRED_ELEMENT_ATTRSZ PARSER_OPTIONAL_ELEMENT_ATTRS� staticmethodrL rN rU r[ r] r^ rd rf ro � __classcell__rJ rJ )rT rK r ( sv c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�zone_ContentHandlerc C s" t j| |� d | _d| _d | _d S )NF)r rN �_rule� _rule_errorZ _limit_ok)rS rc rJ rJ rK rN s z zone_ContentHandler.__init__c C s� t j| ||� | jrd S | jj||� t| ||�r6d S |dkr�d|krVtjd|d � d|krj|d | j_d|kr�tjd|d � d|kr�|d }|t kr�t t j |��|dkr�|t kr�|| j_�n�|d kr�| jjr�tjd � nd | j_�n�|d k�rh| j�rtjd � d | _d S d|k�r.tjd� d | _d S |d | jjk�rT| jjj|d � ntjd|d � �n8|dk�rf| j�r | jj�r�tjdt| j�� d | _d S d}d|k�r�|d j� d$k�r�d }d } }}d|k�r�|d }d|k�r�|d }d|k�r|d }tj||||d�| j_d S d|k�rBd|k�rBtjd� d S d|k�rdd|k�rdtjd� d S d |k�r~tjd |d � d|k�r�tjd � d S d|k�r�t|d � �r�t|d � �r�t|d � �r�t t j|d ��d|k�r$d |d }|| jj k�r| jj j|� ntjd |d � d|k�r�|d }|| jj k�rT| jj j|� ntjd |d � n:|d!k�r�| jj �r�tjd"� nd | j_ ntjd#|� d S d S )%Nr6 r0 z'Ignoring deprecated attribute name='%s'r r> z,Ignoring deprecated attribute immutable='%s'r r r, z Forward already set, ignoring.Tr8 z$Invalid rule: interface use in rule.z Invalid interface: Name missing.z%Interface '%s' already set, ignoring.r: z:Invalid rule: More than one source in rule '%s', ignoring.FrA �yes�truer3 r@ rB )rA z$Invalid source: No address no ipset.z"Invalid source: Address and ipset.r? z)Ignoring deprecated attribute family='%s'z+Invalid source: Invertion not allowed here.zipset:%sz"Source '%s' already set, ignoring.zicmp-block-inversionz+Icmp-Block-Inversion already set, ignoring.zUnknown XML element '%s')ry rz ) r � startElementrx rc Zparser_check_element_attrsr r Zwarningr r r r r_ r r r, rw r& rm r: r\ �lowerr Z Rich_Sourcer r r rb r' r+ ) rS r0 �attrsr rA Zaddrr@ rB �entryrJ rJ rK r{ % s� z zone_ContentHandler.startElementc C s t j| |� t| |� d S )N)r � endElementr )rS r0 rJ rJ rK r � s z zone_ContentHandler.endElementN)rp rq rr rN r{ r rJ rJ rJ rK rv s prv Fc C s t � }| jd�s ttjd| ��| d d � |_|s>|j|j� | |_||_|j t j �rZdnd|_ |j |_ t|�}tj� }|j|� d|| f }t|d��b}tjd �}|j|� y|j|� W n8 tjk r� } zttjd| j� ��W Y d d } ~ X nX W d Q R X ~~t�r|j� |S ) Nz.xmlz'%s' is missing .xml suffix� FTz%s/%s�rbznot a valid zone file: %s���)r rh r r rg r0 rf rl �pathra r � ETC_FIREWALLDZbuiltin�defaultrv �saxZ make_parserZsetContentHandler�openZ InputSourceZ setByteStream�parseZSAXParseExceptionZ INVALID_ZONEZ getExceptionr r[ ) rl r� Z no_check_namer6 �handler�parserr0 �fr: �msgrJ rJ rK r � s: (c C s\ |r|n| j }| jr$d|| jf }nd|| jf }tj j|�r�ytj|d| � W n0 tk r� } ztj d||� W Y d d }~X nX tj j |�}|j t j �r�tj j|� r�tj jt j �s�tjt j d� tj|d� tj|ddd�}t|�}|j� i }| j�r| jd k�r| j|d < | jtk�r*| j|d < |jd |� |jd � t| |� x8t| j�D ]*} |jd� |jdd| i� |jd � �qVW x\t| j �D ]N} |jd� d| k�r�|jdd| dd � i� n|jdd| i� |jd � �q�W | j �r |jd� |jdi � |jd � | j �r2|jd� |jdi � |jd � |j d � |jd � |j � |j!� ~d S )Nz%s/%sz %s/%s.xmlz%s.oldz Backup of file '%s' failed: %si� ZwtzUTF-8)�mode�encodingr r r r6 � z r8 r0 zipset:r: rB � r3 zicmp-block-inversionr, )"r� rl r0 �os�exists�shutilZcopy2� Exceptionr �error�dirnamera r r� �mkdir�ior� r Z startDocumentr r r r{ ZignorableWhitespacer r r& Z simpleElementr' r+ r, r Z endDocument�close) r6 r� �_pathr0 r� �dirpathr� r� r} r8 r: rJ rJ rK r � s` )F)N)(�__all__Zxml.saxr� r� r� r� Zfirewallr Zfirewall.functionsr r r r r r r Zfirewall.core.baser r Zfirewall.core.io.io_objectr r r r Zfirewall.core.io.policyr r r r Z firewall.corer Zfirewall.core.loggerr r Zfirewall.errorsr r rv r r rJ rJ rJ rK �<module> s$ $ w|