관리-도구
편집 파일: io_object.cpython-36.pyc
3 @)�f5 � @ 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