3 Y�j5�@sdZddddddddgZd d ljZd d ljjZd d lZd d lZd dlm Z d dl mZd d lm Z d dl mZd dlmZejdkZGdd�de�ZGdd�de�ZGdd�de�ZGdd�de�ZGdd�dejj�ZGdd�dej�Zdd�Zdd�Zdd�Z dd�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|eZdZdZfZdZgZiZiZdd�Z dd�Z dd�Zd d �Zdd�Z d d�Zdd�Zdd�Zdd�Zdd�Zdd�ZdS)rz; Abstract IO_Object as base for icmptype, service and zone z()cCs"d|_d|_d|_d|_d|_dS)N�F)�filename�path�name�defaultZbuiltin)�self�r�/usr/lib/python3.6/io_object.py�__init__2s zIO_Object.__init__cCs6g}x(|jD]}|jtjt||d��qWt|�S)Nr )�IMPORT_EXPORT_STRUCTURE�append�copy�deepcopy�getattr�tuple)r�ret�xrrr� export_config9szIO_Object.export_configcCsXi}tdd�|jD��}x:|D]2}t||�sAsz0IO_Object.export_config_dict..)�dictrr� isinstance�boolrr)r�conf�type_formats�keyrrr�export_config_dict?s zIO_Object.export_config_dictcCs�|j|�x�t|j�D]~\}\}}t||t�r~g}t�}x,||D] }||krD|j|�|j|�qDW~t||t j |��qt||t j ||��qWdS)N)�check_config� enumeraterr&�list�setr�add�setattrrr)rr(�i�elementZdummyZ_confZ_setr rrr� import_configGs zIO_Object.import_configc Cs~|j|�xn|D]f}t||�s0ttjdj|��t||t�r`t||tt j tj||��qt||tj||��qWdS)Nz-Internal error. '{}' is not a valid attribute) �check_config_dict�hasattrrr Z UNKNOWN_ERROR�formatr&r.r1r �fromkeysrr)rr(r*rrr�import_config_dictWs "zIO_Object.import_config_dictcCszt|t�s(ttjd|td�t|�f��t|�dkr@ttjd��x4|D],}|j�rF||j krFttjd||f��qFWdS)Nz'%s' not of type %s, but %srr"zname can't be emptyz'%s' is not allowed in '%s') r&�strrr �INVALID_TYPE�type�lenZINVALID_NAME�isalnum�ADDITIONAL_ALNUM_CHARS)rr�charrrr� check_namecs zIO_Object.check_namecCsjt|�t|j�kr0ttjdt|�t|j�f��i}x&t|j�D]\}\}}||||<q@W|j|�dS)Nz structure size mismatch %d != %d)r=rrr r;r-r5)rr(Z conf_dictr2r �yrrrr,pszIO_Object.check_configcCsrtdd�|jD��}xX|D]P}|dd�|jD�krDttjdj|��|j||||�|j||||�qWdS)NcSsg|]}|d|df�qS)r r"r)r#r rrrr$|sz/IO_Object.check_config_dict..cSsg|]\}}|�qSrr)r#r rBrrrr$~szoption '{}' is not valid)r%rrr ZINVALID_OPTIONr7�_check_config_structure� _check_config)rr(r)r*rrrr5{s zIO_Object.check_config_dictcCsdS)Nr)rZdummy1Zdummy2Zdummy3rrrrD�szIO_Object._check_configc Cs`t|t|��s,ttjd|t|�t|�f��t|t�rrt|�dkrRttjd|��x|D]}|j||d�qXWn�t|t�r�t|�t|�kr�ttjd|t|�f��x�t |�D]\}}|j|||�q�Wn�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�WdS)Nz'%s' not of type %s, but %sr"zlen('%s') != 1r zlen('%s') != %d)r&r<rr r;r.r=rCrr-r%�items) rr(Z structurer r2�valueZskeyZsvaluer*rrrrC�s8 z!IO_Object._check_config_structurecCs�|j�}d}||jkrdd}|j|dk rdx:|j|D],}||krL|j|�q4ttjd||f��q4W||jkr�d}x$|j|D]}||kr~|j|�q~W|s�ttjd|��x |D]}ttjd||f��q�WdS)NFTzMissing attribute %s for %szUnexpected element %sz%s: Unexpected attribute %s)ZgetNames�PARSER_REQUIRED_ELEMENT_ATTRS�removerr ZPARSE_ERROR�PARSER_OPTIONAL_ELEMENT_ATTRS)rr�attrsZ_attrs�foundr rrr�parser_check_element_attrs�s, z$IO_Object.parser_check_element_attrsN)�__name__� __module__�__qualname__�__doc__rZDBUS_SIGNATUREr?rGrIrr!r+r4r9rAr,r5rDrCrLrrrrr)s" !cs$eZdZ�fdd�Zdd�Z�ZS)�UnexpectedElementErrorcstt|�j�||_dS)N)�superrQrr)rr)� __class__rrr�szUnexpectedElementError.__init__cCs d|jS)NzUnexpected element '%s')r)rrrr�__str__�szUnexpectedElementError.__str__)rMrNrOrrT� __classcell__rr)rSrrQ�srQcs$eZdZ�fdd�Zdd�Z�ZS)�MissingAttributeErrorcstt|�j�||_||_dS)N)rRrVrr� attribute)rrrW)rSrrr�szMissingAttributeError.__init__cCsd|j|jfS)Nz$Element '%s': missing '%s' attribute)rrW)rrrrrT�szMissingAttributeError.__str__)rMrNrOrrTrUrr)rSrrV�srVcs$eZdZ�fdd�Zdd�Z�ZS)�UnexpectedAttributeErrorcstt|�j�||_||_dS)N)rRrXrrrW)rrrW)rSrrr�sz!UnexpectedAttributeError.__init__cCsd|j|jfS)Nz'Element '%s': unexpected attribute '%s')rrW)rrrrrT�sz UnexpectedAttributeError.__str__)rMrNrOrrTrUrr)rSrrX�srXc@s4eZdZdd�Zdd�Zdd�Zdd�Zd d �ZdS)rcCs||_d|_dS)Nr)�item�_element)rrYrrrr�sz!IO_Object_ContentHandler.__init__cCs d|_dS)Nr)rZ)rrrr� startDocument�sz&IO_Object_ContentHandler.startDocumentcCs d|_dS)Nr)rZ)rrrJrrr�startElement�sz%IO_Object_ContentHandler.startElementcCs*|dkr|j|j_n|dkr&|j|j_dS)N�short�description)rZrYr]r^)rrrrr� endElement�sz#IO_Object_ContentHandler.endElementcCs|j|jdd�7_dS)N� � )rZ�replace)r�contentrrr� characters�sz#IO_Object_ContentHandler.charactersN)rMrNrOrr[r\r_rdrrrrr�s c@s<eZdZdd�Zdd�Zdd�Zdd�Zd d �Zdd�Zd S)rcCsNtjjj|�|j|_|j|_ig|_|jd|_ g|_ d|_d|_d|_ dS)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�outrrrr�szIO_Object_XMLGenerator.__init__cCs*trdd�|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. cSsi|]\}}t|�t|��qSr)r)r#rrFrrr� sz7IO_Object_XMLGenerator.startElement..N)rrE�saxutils�XMLGeneratorr\)rrrJrrrr\sz#IO_Object_XMLGenerator.startElementcCs�trX|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() �N)rrjrrErnZ quoteattr)rrrJrFrrr� simpleElementsz$IO_Object_XMLGenerator.simpleElementcCstjj|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)rnror_r)rrrrrr_sz!IO_Object_XMLGenerator.endElementcCstjj|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)rnrordr)rrcrrrrd%sz!IO_Object_XMLGenerator.characterscCstjj|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)rnro�ignorableWhitespacer)rrcrrrrr-sz*IO_Object_XMLGenerator.ignorableWhitespaceN) rMrNrOrr\rqr_rdrrrrrrr�s cCs�tj|�}|dkr$ttjd|��n`|dkr>ttjd|��nF|dkrXttjd|��n,t|�dkr�|d|dkr�ttjd|��dS) N�zport number in '%s' is too bigr"z'%s' is invalid port rangezport range '%s' is ambiguousr ��re)rZgetPortRangerr ZINVALID_PORTr=)ZportZ port_rangerrrr5s cCs|dkrttjd|��dS)N�tcp�udp�sctp�dccpz)'%s' not from {'tcp'|'udp'|'sctp'|'dccp'})rurvrwrx)rr �INVALID_PROTOCOL)�protocolrrrrDscCstj|�sttj|��dS)N)rZ checkProtocolrr ry)rzrrrrJs cCs$tj||�s ttjd||f��dS)Nz'%s' is not valid %s address)rrrr ZINVALID_ADDR)ZipvZaddrrrrrNs)"rP�__all__Zxml.saxrfZxml.sax.saxutilsrnr�sys�collectionsr ZfirewallrZfirewall.functionsrr Zfirewall.errorsr�versionr�objectr� ExceptionrQrVrXrgrhrrorrrrrrrrr�s0 C