a
�M�ew1 � @ s� d Z ddlZddlZddlmZmZmZ ddlZddl Zddl
Zddl Zddl Zddl
ZddlZejjhZejjejjejjhZdd� ZejG dd� dej��ZG dd � d �ZejjG d
d � d e��Z dS ) z)DNS nodes. A node is a set of rdatasets.� N)�Any�Dict�Optionalc C s || v p|t jjko|| v S �N)�dns� rdatatypeZRRSIG)Zrdtypes�rdtype�covers� r
�,/usr/lib/python3.9/site-packages/dns/node.py� _matches_type_or_its_signature, s r c @ sR e Zd ZdZdZdZdZeej j
ej j
d d�dd��Z eej j
d d�d d
��Zd S ) �NodeKindzRdatasets in nodesr � � )r r �returnc C s. t t||�rtjS t t||�r$tjS tjS d S r )r � _cname_typesr
�CNAME�_neutral_types�NEUTRAL�REGULAR)�clsr r r
r
r �classify8 s
zNodeKind.classify)�rdatasetr c C s | � |j|j�S r )r r r )r r r
r
r �classify_rdatasetC s zNodeKind.classify_rdatasetN)�__name__�
__module__� __qualname__�__doc__r r r � classmethodr r � RdataTyper r �Rdatasetr r
r
r
r r
0 s �
r
c @ s( e Zd ZdZdgZdd� Zejje e
e f e
d�dd�Z dd � Z
d
d � Zd d
� Zdd� Zdd� Zdd� Zejjdfejjejjejjeejjd�dd�Zejjdfejjejjejjee ejj d�dd�Z ejjfejjejjejjdd�d d �Z ejjdd �d d �Z e d!�d"d#�Z!ed!�d$d%�Z"dS )&�Nodea� A 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.
� rdatasetsc C s
g | _ d S r �r" ��selfr
r
r �__init__Z s z
Node.__init__)�name�kwr c K sR t �� }| jD ]2}t|�dkr|�|j|fi |��� |�d� q|�� dd� S )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���)�io�StringIOr" �len�write�to_text�getvalue)r% r' r( �s�rdsr
r
r r/ ^ s
z Node.to_textc C s dt t| �� d S )Nz
<DNS node �>)�str�idr$ r
r
r �__repr__r s z
Node.__repr__c C s< | j D ]}||j vr dS q|j D ]}|| j vr" dS q"dS )NFTr# )r% �otherZrdr
r
r �__eq__u s
z Node.__eq__c C s | � |� S r )r8 )r% r7 r
r
r �__ne__� s z Node.__ne__c C s
t | j�S r )r- r" r$ r
r
r �__len__� s z Node.__len__c C s
t | j�S r )�iterr" r$ r
r
r �__iter__� s z
Node.__iter__c C sb t | j�dkrRt�|�}|tjkr6dd� | jD �| _n |tjkrRdd� | jD �| _| j�|� dS )a� Append 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.
r c S s g | ]}t �|�t jkr|�qS r
)r
r r ��.0r2 r
r
r �
<listcomp>� s �z)Node._append_rdataset.<locals>.<listcomp>c S s g | ]}t �|�t jkr|�qS r
)r
r r r= r
r
r r? � s �N)r- r" r
r r r �append�r% r �kindr
r
r �_append_rdataset� s
�
�zNode._append_rdatasetF��rdclassr r �creater c C sH | j D ]}|�|||�r| S q|s*t�tj�|||�}| �|� |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" �match�KeyErrorr r r rC �r% rE r r rF r2 r
r
r �
find_rdataset� s
zNode.find_rdatasetc C s0 z| � ||||�}W n ty* d}Y n0 |S )a� Get 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)rJ rH rI r
r
r � get_rdataset� s
zNode.get_rdatasetN�rE r r r c C s& | � |||�}|dur"| j�|� dS )aD Delete 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)rK r" �remove)r% rE r r r2 r
r
r �delete_rdataset� s zNode.delete_rdataset�� replacementr c C sN t |tjj�std��t |tjj�r,|�� }| �|j |j
|j � | � |� dS )a� Replace 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``.
z replacement is not an rdatasetN)
�
isinstancer r r �
ValueErrorZrrsetZRRsetZ to_rdatasetrN rE r r rC �r% rP r
r
r �replace_rdataset
s �zNode.replace_rdataset�r c C s4 | j D ]&}t�|j|j�}|tjkr| S qtjS )aT Classify 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
r r r r rA r
r
r r $ s
z
Node.classifyc C s dS )NFr
r$ r
r
r � is_immutable9 s zNode.is_immutable)#r r r r � __slots__r& r r' �Namer r4 r r/ r6 r8 r9 r: r< rC r �NONE�
rdataclass�
RdataClassr �boolr r rJ r rK rN rT r
r rV r
r
r
r r! H sL �� -�
� )�� r! c s� e Zd Z� fdd�Zejjdfejjejj ejj e
ej j d�� fdd�
Z
ejjdfejjejj ejj e
eej j d�� fdd�
Zejjfejjejj ejj d d
�d d �Zej j d d
�dd�Ze
d�dd�Z� ZS )�
ImmutableNodec s$ t � �� tdd� |jD ��| _d S )Nc S s g | ]}t j�|��qS r
)r r ZImmutableRdatasetr= r
r
r r? B � z*ImmutableNode.__init__.<locals>.<listcomp>)�superr& �tupler" )r% Znode�� __class__r
r r& ? s
�zImmutableNode.__init__FrD c s |r t d��t� �|||d�S �N� immutableF)� TypeErrorr_ rJ �r% rE r r rF ra r
r rJ E s zImmutableNode.find_rdatasetc s |r t d��t� �|||d�S rc )re r_ rK rf ra r
r rK P s zImmutableNode.get_rdatasetNrL c C s t d��d S �Nrd �re )r% rE r r r
r
r rN [ s z ImmutableNode.delete_rdatasetrO c C s t d��d S rg rh rS r
r
r rT c s z ImmutableNode.replace_rdatasetrU c C s dS )NTr
r$ r
r
r rV f s zImmutableNode.is_immutable)r r r r& r r rY rZ r[ r r\ r r rJ r rK rN rT rV �
__classcell__r
r
ra r r] = s8
���
��� r] ) r �enumr+ �typingr r r Z
dns.immutabler Zdns.nameZdns.rdataclassZ dns.rdatasetZ
dns.rdatatypeZ dns.rendererZ dns.rrsetr r r ZNSECZNSEC3ZKEYr r �unique�Enumr
r! rd r] r
r
r
r �<module> s. �� v