o hw1@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 ;/var/www/html/venv/lib/python3.10/site-packages/dns/node.py_matches_type_or_its_signature,sr c@sXeZdZdZdZdZdZedej j dej j ddfdd Z ed ej j ddfd d Zd S)NodeKindzRdatasets in nodesrrr returncCs*tt||r tjStt||rtjStjSr)r _cname_typesr CNAME_neutral_typesNEUTRALREGULAR)clsrr r r r classify8s  zNodeKind.classifyrdatasetcCs||j|jSr)rrr )rrr r r classify_rdatasetCszNodeKind.classify_rdatasetN)__name__ __module__ __qualname____doc__rrr classmethodrr RdataTyperrRdatasetrr r r r r 0s  r c @sDeZdZdZdgZddZdejjde e e fde fdd Z d d Z d d ZddZddZddZddZejjdfdejjdejjdejjdedejjf ddZejjdfdejjdejjdejjdedeejjf ddZejjfdejjdejjdejjddfd d!Zd"ejjddfd#d$Zde fd%d&Z!defd'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__Zs z Node.__init__namekwrcKsRt}|jD]}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] }||jvr dSq|jD] }||jvrdSqdS)NFTr#)r%otherrdr r r __eq__us    z Node.__eq__cCs || Sr)r9)r%r7r r r __ne__s z Node.__ne__cC t|jSr)r-r"r$r r r __len__ z Node.__len__cCr;r)iterr"r$r r r __iter__r=z Node.__iter__cCsbt|jdkr)t|}|tjkrdd|jD|_n|tjkr)dd|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. rcS g|] }t|tjkr|qSr )r rr.0r2r r r z)Node._append_rdataset..cSr@r )r rrrAr r r rCrDN)r-r"r rrrappendr%rkindr r r _append_rdatasets   zNode._append_rdatasetFrdclassrr createcCsH|jD] }||||r|Sq|sttj|||}|||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 rHr%rIrr rJr2r r r find_rdatasets  zNode.find_rdatasetcCs0z |||||}W|Styd}Y|Sw)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)rNrLrMr r r get_rdatasets zNode.get_rdatasetNcCs*||||}|dur|j|dSdS)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)rOr"remove)r%rIrr r2r r r delete_rdatasetszNode.delete_rdataset replacementcCsNt|tjjs tdt|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_rdatasetrQrIrr rHr%rRr r r replace_rdataset s zNode.replace_rdatasetcCs4|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 rrFr r r r$s  z Node.classifycCdS)NFr r$r r r is_immutable9zNode.is_immutable)#rrrr __slots__r&rr'Namerr4rr/r6r9r:r<r?rHrNONE rdataclass RdataClassrboolrr rNrrOrQrVr rrXr r r r r!Hsd"   -  ) r!c seZdZfddZejjdfdejjdejj dejj de dej j f fd d Z ejjdfdejjdejj dejj de deej j f fd d Zejjfdejjdejj dejj dd fddZdej j dd fddZde fddZZS) ImmutableNodecs$ttdd|jD|_dS)NcSsg|]}tj|qSr )rrZImmutableRdatasetrAr r r rCBsz*ImmutableNode.__init__..)superr&tupler")r%node __class__r r r&?s  zImmutableNode.__init__FrIrr rJrc|rtdt|||dSN immutableF) TypeErrorrarNr%rIrr rJrdr r rNEzImmutableNode.find_rdatasetcrfrg)rirarOrjrdr r rOPrkzImmutableNode.get_rdatasetNcCtdNrhri)r%rIrr r r r rQ[szImmutableNode.delete_rdatasetrRcCrlrmrnrUr r r rVcszImmutableNode.replace_rdatasetcCrW)NTr r$r r r rXfrYzImmutableNode.is_immutable)rrrr&rrr\r]r^rr_rr rNrrOrQrVrX __classcell__r r rdr r`=sP    r`)renumr+typingrrrZ dns.immutablerZdns.nameZdns.rdataclassZ dns.rdatasetZ dns.rdatatypeZ dns.rendererZ dns.rrsetrrrZNSECZNSEC3ZKEYrr uniqueEnumr r!rhr`r r r r s0v