3
��g5 � @ s d Z ddddddddgZd d
ljZd d
ljjZd d
lZd d
lZd d lm Z d d l
m Z d d
l m
Z
d dl
mZ d dlmZ ejdk ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� dejj�ZG dd� dej �Z dd� Z dd� Z d d� Z d d� Z!d
S ) z5Generic io_object handler, io specific check methods.�PY2� IO_Object�IO_Object_ContentHandler�IO_Object_XMLGenerator�
check_port� check_tcpudp�check_protocol�
check_address� N)� OrderedDict)� functions)�b2u)�errors)�
FirewallError�3c @ s| e Zd ZdZf ZdZg Zi Zi Zdd� Z dd� Z
dd� Z d d
� Z d d � Z
d
d� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )r z; Abstract IO_Object as base for icmptype, service and zone z()c C s" d| _ d| _d| _d| _d| _d S )N� F)�filename�path�name�defaultZbuiltin)�self� r � /usr/lib/python3.6/io_object.py�__init__2 s
zIO_Object.__init__c C s6 g }x(| j D ] }|jtjt| |d ��� q W t|�S )Nr )�IMPORT_EXPORT_STRUCTURE�append�copy�deepcopy�getattr�tuple)r �ret�xr r r �
export_config9 s zIO_Object.export_configc C sX i }t dd� | jD ��}x:|D ]2}t| |�s<tt| |�t�r tjt| |��||< q W |S )Nc S s g | ]}|d |d f�qS )r � r )�.0r r r r �
<listcomp>A s z0IO_Object.export_config_dict.<locals>.<listcomp>)�dictr r �
isinstance�boolr r )r �conf� type_formats�keyr r r �export_config_dict? s
z IO_Object.export_config_dictc C s� | j |� x�t| j�D ]~\}\}}t|| t�r~g }t� }x,|| D ] }||krD|j|� |j|� qDW ~t| |t j
|�� qt| |t j
|| �� qW d S )N) � check_config� enumerater r& �list�setr �add�setattrr r )r r( �i�elementZdummyZ_confZ_setr r r r �
import_configG s
zIO_Object.import_configc C s~ | j |� xn|D ]f}t| |�s0ttjdj|���t|| t�r`t| |tt j
t j || ���� qt| |t j || �� qW d S )Nz-Internal error. '{}' is not a valid attribute)
�check_config_dict�hasattrr r
Z
UNKNOWN_ERROR�formatr& r. r1 r
�fromkeysr r )r r( r* r r r �import_config_dictW s
"z IO_Object.import_config_dictc C sz t |t�s(ttjd|td�t|�f ��t|�dk r@ttjd��x4|D ],}|j� rF|| j krFttjd||f ��qFW d S )Nz'%s' not of type %s, but %sr r" zname can't be emptyz'%s' is not allowed in '%s')
r& �strr r
� INVALID_TYPE�type�lenZ INVALID_NAME�isalnum�ADDITIONAL_ALNUM_CHARS)r r �charr r r �
check_namec s
zIO_Object.check_namec C sj t |�t | j�kr0ttjdt |�t | j�f ��i }x&t| j�D ]\}\}}|| ||< q@W | j|� d S )Nz structure size mismatch %d != %d)r= r r r
r; r- r5 )r r( Z conf_dictr2 r �yr r r r, p s zIO_Object.check_configc C sr t dd� | jD ��}xX|D ]P}|dd� | jD �krDttjdj|���| j|| || � | j|| ||� qW d S )Nc S s g | ]}|d |d f�qS )r r" r )r# r r r r r$ | s z/IO_Object.check_config_dict.<locals>.<listcomp>c S s g | ] \}}|�qS r r )r# r rB r r r r$ ~ s zoption '{}' is not valid)r% r r r
ZINVALID_OPTIONr7 �_check_config_structure�
_check_config)r r( r) r* r r r r5 { s
zIO_Object.check_config_dictc C s d S )Nr )r Zdummy1Zdummy2Zdummy3r r r rD � s zIO_Object._check_configc C s` t |t|��s,ttjd|t|�t|�f ��t |t�rrt|�dkrRttjd| ��x |D ]}| j||d � qXW n�t |t�r�t|�t|�kr�ttjd|t|�f ��x�t |�D ]\}}| j|| |� q�W n�t |t
��r\t|j � �d \}}xn|j � D ]b\}}t |t|���s,ttjd|t|�t|�f ��t |t|��s�ttjd|t|�t|�f ��q�W d S )Nz'%s' not of type %s, but %sr" zlen('%s') != 1r zlen('%s') != %d) r& r< r r
r; r. r= rC r r- r% �items) r r( Z structurer r2 �valueZskeyZsvaluer* r r r rC � s8
z!IO_Object._check_config_structurec C s� |j � }d}|| jkrdd}| j| d k rdx:| j| D ],}||krL|j|� q4ttjd||f ��q4W || jkr�d}x$| j| D ]}||kr~|j|� q~W |s�ttjd| ��x |D ]}ttjd||f ��q�W d S )NFTzMissing attribute %s for %szUnexpected element %sz%s: Unexpected attribute %s)ZgetNames� PARSER_REQUIRED_ELEMENT_ATTRS�remover r
Z PARSE_ERROR� PARSER_OPTIONAL_ELEMENT_ATTRS)r r �attrsZ_attrs�foundr r r r �parser_check_element_attrs� s,
z$IO_Object.parser_check_element_attrsN)�__name__�
__module__� __qualname__�__doc__r ZDBUS_SIGNATUREr? rG rI r r! r+ r4 r9 rA r, r5 rD rC rL r r r r r ) s"
!c s$ e Zd Z� fdd�Zdd� Z� ZS )�UnexpectedElementErrorc s t t| �j� || _d S )N)�superrQ r r )r r )� __class__r r r � s z UnexpectedElementError.__init__c C s
d| j S )NzUnexpected element '%s')r )r r r r �__str__� s z UnexpectedElementError.__str__)rM rN rO r rT �
__classcell__r r )rS r rQ � s rQ c s$ e Zd Z� fdd�Zdd� Z� ZS )�MissingAttributeErrorc s t t| �j� || _|| _d S )N)rR rV r r � attribute)r r rW )rS r r r � s z MissingAttributeError.__init__c C s d| j | jf S )Nz$Element '%s': missing '%s' attribute)r rW )r r r r rT � s z MissingAttributeError.__str__)rM rN rO r rT rU r r )rS r rV � s rV c s$ e Zd Z� fdd�Zdd� Z� ZS )�UnexpectedAttributeErrorc s t t| �j� || _|| _d S )N)rR rX r r rW )r r rW )rS r r r � s z!UnexpectedAttributeError.__init__c C s d| j | jf S )Nz'Element '%s': unexpected attribute '%s')r rW )r r r r rT � s z UnexpectedAttributeError.__str__)rM rN rO r rT rU r r )rS r rX � s rX c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) r c C s || _ d| _d S )Nr )�item�_element)r rY r r r r � s z!IO_Object_ContentHandler.__init__c C s
d| _ d S )Nr )rZ )r r r r �
startDocument� s z&IO_Object_ContentHandler.startDocumentc C s
d| _ d S )Nr )rZ )r r rJ r r r � startElement� s z%IO_Object_ContentHandler.startElementc C s* |dkr| j | j_n|dkr&| j | j_d S )N�short� description)rZ rY r] r^ )r r r r r �
endElement� s z#IO_Object_ContentHandler.endElementc C s | j |jdd�7 _ d S )N�
� )rZ �replace)r �contentr r r �
characters� s z#IO_Object_ContentHandler.charactersN)rM rN rO r r[ r\ r_ rd r r r r r � s
c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )r c C sN t jjj| � |j| _|j| _i g| _| jd | _ g | _
d| _ d| _ d| _
d S )Nr" zutf-8F���)�sax�handler�ContentHandlerr �write�_write�flushZ_flushZ _ns_contextsZ_current_contextZ_undeclared_ns_mapsZ _encodingZ_pending_start_elementZ_short_empty_elements)r �outr r r r � s z IO_Object_XMLGenerator.__init__c C s* t rdd� |j� D �}tjj| ||� dS )a saxutils.XMLGenerator.startElement() expects name and attrs to be
unicode and bad things happen if any of them is (utf-8) encoded.
We override the method here to sanitize this case.
Can be removed once we drop Python2 support.
c S s i | ]\}}t |�t |��qS r )r )r# r rF r r r �
<dictcomp> s z7IO_Object_XMLGenerator.startElement.<locals>.<dictcomp>N)r rE �saxutils� XMLGeneratorr\ )r r rJ r r r r\ s z#IO_Object_XMLGenerator.startElementc C s� t rX| jdt|� � x4|j� D ](\}}| jdt|�tjt|��f � q W | jd� nF| jd| � x,|j� D ] \}}| jd|tj|�f � qpW | jd� dS )z* slightly modified startElement()
�<z %s=%sz/>N)r rj r rE rn Z quoteattr)r r rJ rF r r r �
simpleElement s z$IO_Object_XMLGenerator.simpleElementc C s t jj| t|�� dS )z� saxutils.XMLGenerator.endElement() expects name to be
unicode and bad things happen if it's (utf-8) encoded.
We override the method here to sanitize this case.
Can be removed once we drop Python2 support.
N)rn ro r_ r )r r r r r r_ s z!IO_Object_XMLGenerator.endElementc C s t jj| t|�� dS )z� saxutils.XMLGenerator.characters() expects content to be
unicode and bad things happen if it's (utf-8) encoded.
We override the method here to sanitize this case.
Can be removed once we drop Python2 support.
N)rn ro rd r )r rc r r r rd % s z!IO_Object_XMLGenerator.charactersc C s t jj| t|�� dS )a saxutils.XMLGenerator.ignorableWhitespace() expects content to be
unicode and bad things happen if it's (utf-8) encoded.
We override the method here to sanitize this case.
Can be removed once we drop Python2 support.
N)rn ro �ignorableWhitespacer )r rc r r r rr - s z*IO_Object_XMLGenerator.ignorableWhitespaceN) rM rN rO r r\ rq r_ rd rr r r r r r � s
c C s� t j| �}|dkr$ttjd| ��n`|dkr>ttjd| ��nF|d krXttjd| ��n,t|�dkr�|d |d kr�ttjd| ��d S ) N� z port number in '%s' is too bigr" z'%s' is invalid port rangez port range '%s' is ambiguousr ���re )r Z getPortRanger r
Z INVALID_PORTr= )ZportZ
port_ranger r r r 5 s
c C s | dkrt tjd| ��d S )N�tcp�udp�sctp�dccpz)'%s' not from {'tcp'|'udp'|'sctp'|'dccp'})ru rv rw rx )r r
�INVALID_PROTOCOL)�protocolr r r r D s c C s t j| �sttj| ��d S )N)r Z
checkProtocolr r
ry )rz r r r r J s
c C s$ t j| |�s ttjd|| f ��d S )Nz '%s' is not valid %s address)r r r r
Z INVALID_ADDR)ZipvZaddrr r r r N s )"rP �__all__Zxml.saxrf Zxml.sax.saxutilsrn r �sys� collectionsr
Zfirewallr Zfirewall.functionsr r
Zfirewall.errorsr �versionr �objectr � ExceptionrQ rV rX rg rh r ro r r r r r r r r r �<module> s0
C