U mfw1@sdZddlZddlZddlmZmZmZddlZddl Zddl Zddl Zddl Zddl ZddlZejjhZejjejjejjhZddZejGdddejZGdd d ZejjGd d d eZdS) z)DNS nodes. A node is a set of rdatasets.N)AnyDictOptionalcCs||kp|tjjko||kSN)dns rdatatypeZRRSIG)Zrdtypesrdtypecoversr 6/opt/hc_python/lib/python3.8/site-packages/dns/node.py_matches_type_or_its_signature,sr c@sReZdZdZdZdZdZeej j ej j ddddZ eej j ddd d Zd S) NodeKindzRdatasets in nodesr)rr returncCs.tt||rtjStt||r$tjStjSdSr)r _cname_typesr CNAME_neutral_typesNEUTRALREGULAR)clsrr r r r classify8s   zNodeKind.classify)rdatasetrcCs||j|jSr)rrr )rrr r r classify_rdatasetCszNodeKind.classify_rdatasetN)__name__ __module__ __qualname____doc__rrr classmethodrr RdataTyperrRdatasetrr r r r r 0s r c@s(eZdZdZdgZddZejje e e fe dddZ dd Z d d Zd d ZddZddZddZejjdfejjejjejjeejjdddZejjdfejjejjejjeeejjdddZejjfejjejjejjddddZejjdddd Ze d!d"d#Z!ed!d$d%Z"dS)&NodeaA Node is a set of rdatasets. A node is either a CNAME node or an "other data" node. A CNAME node contains only CNAME, KEY, NSEC, and NSEC3 rdatasets along with their covering RRSIG rdatasets. An "other data" node contains any rdataset other than a CNAME or RRSIG(CNAME) rdataset. When changes are made to a node, the CNAME or "other data" state is always consistent with the update, i.e. the most recent change wins. For example, if you have a node which contains a CNAME rdataset, and then add an MX rdataset to it, then the CNAME rdataset will be deleted. Likewise if you have a node containing an MX rdataset and add a CNAME rdataset, the MX rdataset will be deleted. rdatasetscCs g|_dSrr"selfr r r __init__Zsz Node.__init__)namekwrcKsNt}|jD].}t|dkr||j|f||dq|ddS)a"Convert a node to text format. Each rdataset at the node is printed. Any keyword arguments to this method are passed on to the rdataset's to_text() method. *name*, a ``dns.name.Name``, the owner name of the rdatasets. Returns a ``str``. r N)ioStringIOr"lenwriteto_textgetvalue)r%r'r(srdsr r r r/^s    z Node.to_textcCsdtt|dS)Nz )stridr$r r r __repr__rsz Node.__repr__cCs<|jD]}||jkrdSq|jD]}||jkr"dSq"dS)NFTr#)r%otherrdr r r __eq__us    z Node.__eq__cCs || Sr)r9)r%r7r r r __ne__sz Node.__ne__cCs t|jSr)r-r"r$r r r __len__sz Node.__len__cCs t|jSr)iterr"r$r r r __iter__sz Node.__iter__cCsbt|jdkrRt|}|tjkr6dd|jD|_n|tjkrRdd|jD|_|j|dS)aAppend rdataset to the node with special handling for CNAME and other data conditions. Specifically, if the rdataset being appended has ``NodeKind.CNAME``, then all rdatasets other than KEY, NSEC, NSEC3, and their covering RRSIGs are deleted. If the rdataset being appended has ``NodeKind.REGULAR`` then CNAME and RRSIG(CNAME) are deleted. rcSs g|]}t|tjkr|qSr )r rr.0r2r r r sz)Node._append_rdataset..cSs g|]}t|tjkr|qSr )r rrr>r r r r@sN)r-r"r rrrappendr%rkindr r r _append_rdatasets     zNode._append_rdatasetFrdclassrr creatercCsH|jD]}||||r|Sq|s*ttj|||}|||S)a*Find an rdataset matching the specified properties in the current node. *rdclass*, a ``dns.rdataclass.RdataClass``, the class of the rdataset. *rdtype*, a ``dns.rdatatype.RdataType``, the type of the rdataset. *covers*, a ``dns.rdatatype.RdataType``, the covered type. Usually this value is ``dns.rdatatype.NONE``, but if the rdtype is ``dns.rdatatype.SIG`` or ``dns.rdatatype.RRSIG``, then the covers value will be the rdata type the SIG/RRSIG covers. The library treats the SIG and RRSIG types as if they were a family of types, e.g. RRSIG(A), RRSIG(NS), RRSIG(SOA). This makes RRSIGs much easier to work with than if RRSIGs covering different rdata types were aggregated into a single RRSIG rdataset. *create*, a ``bool``. If True, create the rdataset if it is not found. Raises ``KeyError`` if an rdataset of the desired type and class does not exist and *create* is not ``True``. Returns a ``dns.rdataset.Rdataset``. )r"matchKeyErrorrrr rDr%rFrr rGr2r r r find_rdatasets   zNode.find_rdatasetcCs2z|||||}Wntk r,d}YnX|S)aGet an rdataset matching the specified properties in the current node. None is returned if an rdataset of the specified type and class does not exist and *create* is not ``True``. *rdclass*, an ``int``, the class of the rdataset. *rdtype*, an ``int``, the type of the rdataset. *covers*, an ``int``, the covered type. Usually this value is dns.rdatatype.NONE, but if the rdtype is dns.rdatatype.SIG or dns.rdatatype.RRSIG, then the covers value will be the rdata type the SIG/RRSIG covers. The library treats the SIG and RRSIG types as if they were a family of types, e.g. RRSIG(A), RRSIG(NS), RRSIG(SOA). This makes RRSIGs much easier to work with than if RRSIGs covering different rdata types were aggregated into a single RRSIG rdataset. *create*, a ``bool``. If True, create the rdataset if it is not found. Returns a ``dns.rdataset.Rdataset`` or ``None``. N)rKrIrJr r r get_rdatasets  zNode.get_rdatasetNrFrr rcCs&||||}|dk r"|j|dS)aDDelete the rdataset matching the specified properties in the current node. If a matching rdataset does not exist, it is not an error. *rdclass*, an ``int``, the class of the rdataset. *rdtype*, an ``int``, the type of the rdataset. *covers*, an ``int``, the covered type. N)rLr"remove)r%rFrr r2r r r delete_rdatasetszNode.delete_rdataset replacementrcCsNt|tjjstdt|tjjr,|}||j |j |j | |dS)aReplace an rdataset. It is not an error if there is no rdataset matching *replacement*. Ownership of the *replacement* object is transferred to the node; in other words, this method does not store a copy of *replacement* at the node, it stores *replacement* itself. *replacement*, a ``dns.rdataset.Rdataset``. Raises ``ValueError`` if *replacement* is not a ``dns.rdataset.Rdataset``. zreplacement is not an rdatasetN) isinstancerrr ValueErrorZrrsetZRRsetZ to_rdatasetrOrFrr rDr%rQr r r replace_rdataset szNode.replace_rdatasetrcCs4|jD]&}t|j|j}|tjkr|SqtjS)aTClassify a node. A node which contains a CNAME or RRSIG(CNAME) is a ``NodeKind.CNAME`` node. A node which contains only "neutral" types, i.e. types allowed to co-exist with a CNAME, is a ``NodeKind.NEUTRAL`` node. The neutral types are NSEC, NSEC3, KEY, and their associated RRSIGS. An empty node is also considered neutral. A node which contains some rdataset which is not a CNAME, RRSIG(CNAME), or a neutral type is a a ``NodeKind.REGULAR`` node. Regular nodes are also commonly referred to as "other data". )r"r rrr rrBr r r r$s    z Node.classifycCsdS)NFr r$r r r is_immutable9szNode.is_immutable)#rrrr __slots__r&rr'Namerr4rr/r6r9r:r;r=rDrNONE rdataclass RdataClassrboolrr rKrrLrOrUr rrWr r r r r!HsL   - ) r!cseZdZfddZejjdfejjejj ejj e ej j dfdd Z ejjdfejjejj ejj e eej j dfdd Zejjfejjejj ejj d d d d Zej j d d ddZe dddZZS) ImmutableNodecs$ttdd|jD|_dS)NcSsg|]}tj|qSr )rrZImmutableRdatasetr>r r r r@Bsz*ImmutableNode.__init__..)superr&tupler")r%node __class__r r r&?s zImmutableNode.__init__FrEcs|r tdt|||dSN immutableF) TypeErrorr_rKr%rFrr rGrbr r rKEszImmutableNode.find_rdatasetcs|r tdt|||dSrd)rfr_rLrgrbr r rLPszImmutableNode.get_rdatasetNrMcCs tddSNrerf)r%rFrr r r r rO[szImmutableNode.delete_rdatasetrPcCs tddSrhrirTr r r rUcszImmutableNode.replace_rdatasetrVcCsdS)NTr r$r r r rWfszImmutableNode.is_immutable)rrrr&rrrZr[r\rr]rr rKrrLrOrUrW __classcell__r r rbr r^=s8   r^)renumr+typingrrrZ dns.immutablerZdns.nameZdns.rdataclassZ dns.rdatasetZ dns.rdatatypeZ dns.rendererZ dns.rrsetrrrZNSECZNSEC3ZKEYrr uniqueEnumr r!rer^r r r r s.v