a
i- @ s d Z g dZddlmZ ddlm mZ ddlZddlmZ ddl m
Z
ddl m Z ddl m
Z
G dd d eZG d
d d eZG d d
d
eZG dd deZG dd dejjZG dd dejZdd Zdd Zdd Zdd Z dS ) z5Generic io_object handler, io specific check methods.) IO_ObjectIO_Object_ContentHandlerIO_Object_XMLGenerator
check_port check_tcpudpcheck_protocol
check_address N) OrderedDict) functions)errors)
FirewallErrorc @ s e Zd ZdZdZdZg Zi Zi Zdd Z dd Z
dd Z d
d Z d d
Z
dd Zi fddZdd 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)filenamepathnamedefaultZbuiltinselfr
r
>/usr/lib/python3.9/site-packages/firewall/core/io/io_object.py__init__- s
zIO_Object.__init__c C s2 g }| j D ] }|tt| |d q
t|S )Nr )IMPORT_EXPORT_STRUCTUREappendcopydeepcopygetattrtuple)r retxr
r
r
export_config4 s
zIO_Object.export_configc C sT i }t dd | jD }|D ]2}t| |s:tt| |tr tt| |||< q |S )Nc S s g | ]}|d |d fqS r r
.0r r
r
r
< z0IO_Object.export_config_dict..)dictr r
isinstanceboolr r )r conf type_formatskeyr
r
r export_config_dict: s z IO_Object.export_config_dictc C s | || t| jD ]z\}\}}t|| trzg }t }|| D ] }||vrB|| || qB~t| |t
| qt| |t
|| qd S N) check_config enumerater r' listsetr addsetattrr r ) r r) all_io_objectsielementZdummyZ_confZ_setr r
r
r
import_configB s
zIO_Object.import_configc C s| | || |D ]f}t| |s0ttjd|t|| tr`t| |tt
t || qt| |t || qd S )Nz-Internal error. '{}' is not a valid attribute)
check_config_dicthasattrr r Z
UNKNOWN_ERRORformatr' r0 r3 r fromkeysr r )r r) r4 r+ r
r
r import_config_dictR s
"z IO_Object.import_config_dictc C st t |ts(ttjd|tdt|f t|dk r@ttjd|D ]*}| sD|| j vrDttjd||f qDd S )N'%s' not of type %s, but %sr r! zname can't be emptyz'%s' is not allowed in '%s')
r' strr r INVALID_TYPEtypelenZ INVALID_NAMEisalnumADDITIONAL_ALNUM_CHARS)r r charr
r
r
check_name^ s
zIO_Object.check_namec C sh t |t | jkr0ttjdt |t | jf i }t| jD ]\}\}}|| ||< q>| || d S )Nz structure size mismatch %d != %d)rA r r r r? r/ r8 )r r) r4 Z conf_dictr5 r yr
r
r r. k s zIO_Object.check_configc C sp t dd | jD }|D ]R}|dd | jD vrBttjd|| || || | || ||| qd S )Nc S s g | ]}|d |d fqS r r
r" r
r
r r$ w r% z/IO_Object.check_config_dict..c S s g | ] \}}|qS r
r
)r# r rF r
r
r r$ y r% zoption '{}' is not valid)r& r r r ZINVALID_OPTIONr: _check_config_structure
_check_config)r r) r4 r* r+ r
r
r r8 v s zIO_Object.check_config_dictc C s d S r- r
)r Zdummy1Zdummy2Zdummy3Zdummy4r
r
r rH ~ s zIO_Object._check_configc C sT t |t|s,ttjd|t|t|f t |trnt|dkrRttjd| |D ]}| ||d qVnt |trt|t|krttjd|t|f t |D ]\}}| || | qnt |t
rPt| d \}}| D ]b\}}t |t|s"ttjd|t|t|f t |t|sttjd|t|t|f qd S )Nr= r! zlen('%s') != 1r zlen('%s') != %d) r' r@ r r r? r0 rA rG r r/ r& items) r r) Z structurer r5 valueZskeyZsvaluer+ r
r
r rG sJ
z!IO_Object._check_config_structurec C s | }d}|| jv r`d}| j| d ur`| j| D ],}||v rJ|| q2ttjd||f q2|| jv rd}| j| D ]}||v rx|| qx|sttjd| |D ]}ttjd||f qd S )NFTzMissing attribute %s for %szUnexpected element %sz%s: Unexpected attribute %s)ZgetNames PARSER_REQUIRED_ELEMENT_ATTRSremover r Z PARSE_ERROR PARSER_OPTIONAL_ELEMENT_ATTRS)r r attrsZ_attrsfoundr r
r
r parser_check_element_attrs s2
z$IO_Object.parser_check_element_attrsN)__name__
__module__ __qualname____doc__r ZDBUS_SIGNATURErC rK rM r r r, r7 r< rE r. r8 rH rG rP r
r
r
r r $ s"
!r c s$ e Zd Z fddZdd Z ZS )UnexpectedElementErrorc s t t| || _d S r- )superrU 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__rQ rR rS r rZ
__classcell__r
r
rX r rU s rU c s$ e Zd Z fddZdd Z ZS )MissingAttributeErrorc s t t| || _|| _d S r- )rV r] r r attributer r r^ rX r
r r s z MissingAttributeError.__init__c C s d| j | jf S )Nz$Element '%s': missing '%s' attributer r^ r r
r
r rZ s
z MissingAttributeError.__str__r[ r
r
rX r r] s r] c s$ e Zd Z fddZdd Z ZS )UnexpectedAttributeErrorc s t t| || _|| _d S r- )rV ra r r r^ r_ rX r
r r s z!UnexpectedAttributeError.__init__c C s d| j | jf S )Nz'Element '%s': unexpected attribute '%s'r` r r
r
r rZ s
z UnexpectedAttributeError.__str__r[ r
r
rX r ra s ra c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S ) r c C s || _ d| _d S Nr )item_element)r rc r
r
r r s z!IO_Object_ContentHandler.__init__c C s
d| _ d S rb rd r r
r
r
startDocument s z&IO_Object_ContentHandler.startDocumentc C s
d| _ d S rb re )r r rN r
r
r startElement s z%IO_Object_ContentHandler.startElementc C s* |dkr| j | j_n|dkr&| j | j_d S )Nshort description)rd rc rh ri rW r
r
r
endElement s z#IO_Object_ContentHandler.endElementc C s | j |dd7 _ d S )N
)rd replace)r Zcontentr
r
r
characters s z#IO_Object_ContentHandler.charactersN)rQ rR rS r rf rg rj rn r
r
r
r r s
r c @ s e Zd Zdd Zdd ZdS )r c C sN t jj| |j| _|j| _i g| _| jd | _ g | _
d| _ d| _ d| _
d S )Nzutf-8F)saxhandlerContentHandlerr write_writeflushZ_flushZ _ns_contextsZ_current_contextZ_undeclared_ns_maps _encodingZ_pending_start_elementZ_short_empty_elements)r outr
r
r r s z IO_Object_XMLGenerator.__init__c C sF | d| | D ] \}}| d|t|f q| d dS )z* slightly modified startElement()
N)rt rI saxutilsZ quoteattr)r r rN rJ r
r
r
simpleElement s z$IO_Object_XMLGenerator.simpleElementN)rQ rR rS r rz r
r
r
r r s r c C s t | }|dkr$ttjd| n`|dkr>ttjd| nF|d u rXttjd| n,t|dkr|d |d krttjd| d S ) Nz port number in '%s' is too bigro z'%s' is invalid port rangez port range '%s' is ambiguous r r! )r
Z getPortRanger r Z INVALID_PORTrA )portZ
port_ranger
r
r r s"
r c C s | dvrt tjd| d S )N)ZtcpZudpZsctpZdccpz)'%s' not from {'tcp'|'udp'|'sctp'|'dccp'})r r INVALID_PROTOCOLZprotocolr
r
r r s r c C s t | sttj| d S r- )r
Z
checkProtocolr r r~ r r
r
r r s
r c C s$ t | |s ttjd|| f d S )Nz '%s' is not valid %s address)r
r r r Z INVALID_ADDR)Zipvaddrr
r
r r s
r ) rT __all__Zxml.saxrp Zxml.sax.saxutilsry r collectionsr Zfirewallr
r Zfirewall.errorsr objectr ExceptionrU r] ra rq rr r Z XMLGeneratorr r r r r r
r
r
r s&