a Mew1@sdZddlZddlZddlmZmZmZddlZddl Zddl Zddl Zddl Zddl ZddlZejjhZejjejjejjhZddZejGdddejZGdd d ZejjGd d d eZdS) z)DNS nodes. A node is a set of rdatasets.N)AnyDictOptionalcCs||vp|tjjko||vSN)dns rdatatypeZRRSIG)Zrdtypesrdtypecoversr ,/usr/lib/python3.9/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__)namekwrcKsRt}|jD]2}t|dkr||j|fi||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]}||jvrdSq|jD]}||jvr"dSq"dS)NFTr#)r%otherZrdr r r __eq__us    z Node.__eq__cCs || Sr)r8)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 rCr%rErr rFr2r r r find_rdatasets   zNode.find_rdatasetcCs0z|||||}Wnty*d}Yn0|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)rJrHrIr r r get_rdatasets   zNode.get_rdatasetNrErr rcCs&||||}|dur"|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)rKr"remove)r%rErr 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_rdatasetrNrErr rCr%rPr r r replace_rdataset s zNode.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 rrAr r r r$s    z Node.classifycCsdS)NFr r$r r r is_immutable9szNode.is_immutable)#rrrr __slots__r&rr'Namerr4rr/r6r8r9r:r<rCrNONE rdataclass RdataClassrboolrr rJrrKrNrTr rrVr 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?Bz*ImmutableNode.__init__..)superr&tupler")r%Znode __class__r r r&?s zImmutableNode.__init__FrDcs|r tdt|||dSN immutableF) TypeErrorr_rJr%rErr rFrar r rJEszImmutableNode.find_rdatasetcs|r tdt|||dSrc)rer_rKrfrar r rKPszImmutableNode.get_rdatasetNrLcCs tddSNrdre)r%rErr r r r rN[szImmutableNode.delete_rdatasetrOcCs tddSrgrhrSr r r rTcszImmutableNode.replace_rdatasetrUcCsdS)NTr r$r r r rVfszImmutableNode.is_immutable)rrrr&rrrYrZr[rr\rr rJrrKrNrTrV __classcell__r r rar r]=s8   r])renumr+typingrrrZ dns.immutablerZdns.nameZdns.rdataclassZ dns.rdatasetZ dns.rdatatypeZ dns.rendererZ dns.rrsetrrrZNSECZNSEC3ZKEYrr uniqueEnumr r!rdr]r r r r s.v