𝗖𝗨𝗥𝗥𝗘𝗡𝗧 𝗙𝗜𝗟𝗘 : /usr/lib/python3.9/site-packages/dns/__pycache__//zone.cpython-39.opt-1.pyc
a
�M�ev� � @ s~ d Z ddlZddlZddlZddlZddlmZmZmZm Z m
Z
m Z m Z m
Z
mZmZ ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Zddl Zddl Zddl Zddl Zddl!m"Z"m#Z#m$Z$ G dd� dej%j&�Z'G dd� de'�Z(G d d
� d
e'�Z)G d d � d e'�Z*G d
d� dej%j&�Z+G dd� dej%j&�Z,G dd� dej%j&�Z-G dd� dej%j&�Z.ej/j0e ej/j0 e1ej/j0d�dd�Z2G dd� dej3j4�Z5G dd� dej6j7�Z8ej9j9G d d � d e8��Z:G d d � d �Z;G d d!� d!e;�Z<ej9j9G d"d#� d#e;��Z=G d$d%� d%ej3j>�Z>dej?j@d&e5dd'd&dd&f ee eej/j0eAf ej?jBe1ee eA e1e1e ej/jC ee1eeA f e5d(� d)d*�ZDdej?j@d&e5dd'd&dd&f eAe eej/j0eAf ej?jBe1ee eA e1e1e ej/jC ee1eeA f e5d(� d+d,�ZEdej?j@d&e5dd&d&dd&f ee eej/j0eAf ej?jBe1ee eA e1e1e ej/jC ee1eeA f e5d-� d.d/�ZFe5d&d&feee1e1e5d0�d1d2�ZGdS )3z
DNS Zones.� N)
�Any�Callable�Iterable�Iterator�List�MutableMapping�Optional�Set�Tuple�Union)�DigestHashAlgorithm� DigestScheme�_digest_hashersc @ s e Zd ZdZdS )�BadZonezThe DNS zone is malformed.N��__name__�
__module__� __qualname__�__doc__� r r �,/usr/lib/python3.9/site-packages/dns/zone.pyr 8 s r c @ s e Zd ZdZdS )�NoSOAz)The DNS zone has no SOA RR at its origin.Nr r r r r r < s r c @ s e Zd ZdZdS )�NoNSz+The DNS zone has no NS RRset at its origin.Nr r r r r r @ s r c @ s e Zd ZdZdS )�
UnknownOriginz!The DNS zone's origin is unknown.Nr r r r r r D s r c @ s e Zd ZdZdS )�UnsupportedDigestSchemez(The zone digest's scheme is unsupported.Nr r r r r r H s r c @ s e Zd ZdZdS )� UnsupportedDigestHashAlgorithmz(The zone digest's origin is unsupported.Nr r r r r r L s r c @ s e Zd ZdZdS )�NoDigestz/The DNS zone has no ZONEMD RRset at its origin.Nr r r r r r P s r c @ s e Zd ZdZdS )�DigestVerificationFailurez#The ZONEMD digest failed to verify.Nr r r r r r T s r )�name�origin�
relativize�returnc C st |d u rt d��| �� r:| �|�s*t d��|rp| �|�} n6z| �|�}W n tjjyf t d��Y n0 |sp|} | S )Nzno zone origin is definedz5name parameter must be a subdomain of the zone originz relative name too long for zone)�KeyError� is_absoluteZ is_subdomainr Z derelativize�dnsr Z NameTooLong)r r r Zabs_namer r r �_validate_nameX s
r% c @ s� e Zd ZU dZejjZeg ejjf e d< e
Z eg e ej
jejjf f e d< dZeeg df e d< dZeeg df e d< g d �Zejjd
feeej
jef ejjed �d d
�Zdd� Zdd� Zeej
jef ej
jd�dd�Z dd� Z dd� Z dd� Z dd � Z d d � Z!d d � Z"d!d"� Z#d#d$� Z$d%d&� Z%dieej
jef eejjd(�d)d*�Z&djeej
jef eeejj d(�d+d,�Z'eej
jef dd�d-d.�Z(ej)j*d'feej
jef eej)j+ef eej)j+ef eej,j-d/�d0d1�Z.ej)j*d'feej
jef eej)j+ef eej)j+ef eeej,j- d/�d2d3�Z/ej)j*feej
jef eej)j+ef eej)j+ef dd4�d5d6�Z0eej
jef ej,j-dd7�d8d9�Z1ej)j*feej
jef eej)j+ef eej)j+ef ej2j3d4�d:d;�Z4ej)j*feej
jef eej)j+ef eej)j+ef eej2j3 d4�d<d=�Z5ej)j6ej)j*feej)j+ef eej)j+ef e7e8ej
jej,j-f d>�d?d@�Z9ej)j6ej)j*feej)j+ef eej)j+ef e7e8ej
je:ej;j<f d>�dAdB�Z=dke>eeee eeddC�dDdE�Z?dleeee eeedF�dGdH�Z@ddI�dJdK�ZAdmeejBjC ejDj6jEjEdL�dMdN�ZFeGjHfeIeGeJdO�dPdQ�ZKeGjHfeIeGejDj6jLjLdO�dRdS�ZMdneejDj6jLjL ddT�dUdV�ZNdWdI�dXdY�ZOdoedWdZ�d[d\�ZPe8eej
j eeej
j f dI�d]d^�ZQd_d`� ZRdadb� ZSdcdd� ZTdedf� ZUdgdh� ZVdS )p�Zoneaz A DNS zone.
A ``Zone`` is a mapping from names to nodes. The zone object may be
treated like a Python dictionary, e.g. ``zone[name]`` will retrieve
the node associated with that name. The *name* may be a
``dns.name.Name object``, or it may be a string. In either case,
if the name is relative it is treated as relative to the origin of
the zone.
� node_factory� map_factoryN�WritableVersion�writable_version_factory�ImmutableVersion�immutable_version_factory)�rdclassr �nodesr T)r r- r c C sf |durFt |t�r tj�|�}nt |tjj�s6td��|�� sFtd��|| _|| _ | �
� | _ || _ dS )a� Initialize a zone object.
*origin* is the origin of the zone. It may be a ``dns.name.Name``,
a ``str``, or ``None``. If ``None``, then the zone's origin will
be set by the first ``$ORIGIN`` line in a zone file.
*rdclass*, an ``int``, the zone's rdata class; the default is class IN.
*relativize*, a ``bool``, determine's whether domain names are
relativized to the zone's origin. The default is ``True``.
Nz2origin parameter must be convertible to a DNS namez)origin parameter must be an absolute name)
�
isinstance�strr$ r � from_text�Name�
ValueErrorr# r r- r( r. r )�selfr r- r r r r �__init__� s
z
Zone.__init__c C s: t |t�sdS | j|jks2| j|jks2| j|jkr6dS dS )zqTwo zones are equal if they have the same origin, class, and
nodes.
Returns a ``bool``.
FT)r/ r& r- r r. �r4 �otherr r r �__eq__� s
�
�
�z Zone.__eq__c C s | � |� S )z>Are two zones not equal?
Returns a ``bool``.
)r8 r6 r r r �__ne__� s z Zone.__ne__�r r! c C s@ t |t�rtj�|d �}nt |tjj�s0td��t|| j| j �S )Nz0name parameter must be convertible to a DNS name)
r/ r0 r$ r r1 r2 r" r% r r �r4 r r r r r% � s
zZone._validate_namec C s | � |�}| j| S �N�r% r. �r4 �keyr r r � __getitem__� s
zZone.__getitem__c C s | � |�}|| j|< d S r< r= )r4 r? �valuer r r � __setitem__� s
zZone.__setitem__c C s | � |�}| j|= d S r< r= r> r r r � __delitem__� s
zZone.__delitem__c C s
| j �� S r< )r. �__iter__�r4 r r r rD � s z
Zone.__iter__c C s
| j �� S r< �r. �keysrE r r r rG � s z Zone.keysc C s
| j �� S r< )r. �valuesrE r r r rH � s z Zone.valuesc C s
| j �� S r< �r. �itemsrE r r r rJ � s z
Zone.itemsc C s | � |�}| j�|�S r< �r% r. �getr> r r r rL � s
zZone.getc C s | � |�}|| jv S r< r= r> r r r � __contains__� s
zZone.__contains__F)r �creater! c C s< | � |�}| j�|�}|du r8|s&t�| �� }|| j|< |S )aN Find a node in the zone, possibly creating it.
*name*: the name of the node to find.
The value may be a ``dns.name.Name`` or a ``str``. If absolute, the
name must be a subdomain of the zone's origin. If ``zone.relativize``
is ``True``, then the name will be relativized.
*create*, a ``bool``. If true, the node will be created if it does
not exist.
Raises ``KeyError`` if the name is not known and create was
not specified, or if the name was not a subdomain of the origin.
Returns a ``dns.node.Node``.
N)r% r. rL r" r' �r4 r rN �noder r r � find_node� s
zZone.find_nodec C s, z| � ||�}W n ty& d}Y n0 |S )a� Get a node in the zone, possibly creating it.
This method is like ``find_node()``, except it returns None instead
of raising an exception if the node does not exist and creation
has not been requested.
*name*: the name of the node to find.
The value may be a ``dns.name.Name`` or a ``str``. If absolute, the
name must be a subdomain of the zone's origin. If ``zone.relativize``
is ``True``, then the name will be relativized.
*create*, a ``bool``. If true, the node will be created if it does
not exist.
Returns a ``dns.node.Node`` or ``None``.
N)rQ r" rO r r r �get_node s
z
Zone.get_nodec C s | � |�}|| jv r | j|= dS )ak Delete the specified node if it exists.
*name*: the name of the node to find.
The value may be a ``dns.name.Name`` or a ``str``. If absolute, the
name must be a subdomain of the zone's origin. If ``zone.relativize``
is ``True``, then the name will be relativized.
It is not an error if the node does not exist.
Nr= r; r r r � delete_node s
zZone.delete_node)r �rdtype�coversrN r! c C sD | � |�}tjj�|�}tjj�|�}| �||�}|�| j|||�S )a� Look for an rdataset with the specified name and type in the zone,
and return an rdataset encapsulating it.
The rdataset returned is not a copy; changes to it will change
the zone.
KeyError is raised if the name or type are not found.
*name*: the name of the node to find.
The value may be a ``dns.name.Name`` or a ``str``. If absolute, the
name must be a subdomain of the zone's origin. If ``zone.relativize``
is ``True``, then the name will be relativized.
*rdtype*, a ``dns.rdatatype.RdataType`` or ``str``, the rdata type desired.
*covers*, a ``dns.rdatatype.RdataType`` or ``str`` 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, the node will be created if it does
not exist.
Raises ``KeyError`` if the name is not known and create was
not specified, or if the name was not a subdomain of the origin.
Returns a ``dns.rdataset.Rdataset``.
)r% r$ � rdatatype� RdataType�makerQ �
find_rdatasetr- )r4 r rT rU rN rP r r r rY / s
)
zZone.find_rdatasetc C s0 z| � ||||�}W n ty* d}Y n0 |S )a% Look for an rdataset with the specified name and type in the zone.
This method is like ``find_rdataset()``, except it returns None instead
of raising an exception if the rdataset does not exist and creation
has not been requested.
The rdataset returned is not a copy; changes to it will change
the zone.
*name*: the name of the node to find.
The value may be a ``dns.name.Name`` or a ``str``. If absolute, the
name must be a subdomain of the zone's origin. If ``zone.relativize``
is ``True``, then the name will be relativized.
*rdtype*, a ``dns.rdatatype.RdataType`` or ``str``, the rdata type desired.
*covers*, a ``dns.rdatatype.RdataType`` or ``str``, 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, the node will be created if it does
not exist.
Raises ``KeyError`` if the name is not known and create was
not specified, or if the name was not a subdomain of the origin.
Returns a ``dns.rdataset.Rdataset`` or ``None``.
N)rY r" )r4 r rT rU rN �rdatasetr r r � get_rdataset^ s
*
zZone.get_rdataset�r rT rU r! c C sb | � |�}tjj�|�}tjj�|�}| �|�}|dur^|�| j||� t|�dkr^| � |� dS )a� Delete the rdataset matching *rdtype* and *covers*, if it
exists at the node specified by *name*.
It is not an error if the node does not exist, or if there is no matching
rdataset at the node.
If the node has no rdatasets after the deletion, it will itself be deleted.
*name*: the name of the node to find. The value may be a ``dns.name.Name`` or a
``str``. If absolute, the name must be a subdomain of the zone's origin. If
``zone.relativize`` is ``True``, then the name will be relativized.
*rdtype*, a ``dns.rdatatype.RdataType`` or ``str``, the rdata type desired.
*covers*, a ``dns.rdatatype.RdataType`` or ``str`` or ``None``, 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.
Nr )
r% r$ rV rW rX rR �delete_rdatasetr- �lenrS �r4 r rT rU rP r r r r] � s
zZone.delete_rdataset)r � replacementr! c C s. |j | j krtd��| �|d�}|�|� dS )a� Replace an rdataset at name.
It is not an error if there is no rdataset matching I{replacement}.
Ownership of the *replacement* object is transferred to the zone;
in other words, this method does not store a copy of *replacement*
at the node, it stores *replacement* itself.
If the node does not exist, it is created.
*name*: the name of the node to find.
The value may be a ``dns.name.Name`` or a ``str``. If absolute, the
name must be a subdomain of the zone's origin. If ``zone.relativize``
is ``True``, then the name will be relativized.
*replacement*, a ``dns.rdataset.Rdataset``, the replacement rdataset.
z#replacement.rdclass != zone.rdclassTN)r- r3 rQ �replace_rdataset)r4 r r` rP r r r ra � s zZone.replace_rdatasetc C s^ | � |�}tjj�|�}tjj�|�}| j| �| j||�}tj� || j||�}|�
|� |S )a� Look for an rdataset with the specified name and type in the zone,
and return an RRset encapsulating it.
This method is less efficient than the similar
``find_rdataset()`` because it creates an RRset instead of
returning the matching rdataset. It may be more convenient
for some uses since it returns an object which binds the owner
name to the rdataset.
This method may not be used to create new nodes or rdatasets;
use ``find_rdataset`` instead.
*name*: the name of the node to find.
The value may be a ``dns.name.Name`` or a ``str``. If absolute, the
name must be a subdomain of the zone's origin. If ``zone.relativize``
is ``True``, then the name will be relativized.
*rdtype*, a ``dns.rdatatype.RdataType`` or ``str``, the rdata type desired.
*covers*, a ``dns.rdatatype.RdataType`` or ``str``, 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, the node will be created if it does
not exist.
Raises ``KeyError`` if the name is not known and create was
not specified, or if the name was not a subdomain of the origin.
Returns a ``dns.rrset.RRset`` or ``None``.
) r% r$ rV rW rX r. rY r- �rrset�RRset�update)r4 r rT rU ZvnamerZ rb r r r �
find_rrset� s ,
zZone.find_rrsetc C s. z| � |||�}W n ty( d}Y n0 |S )a. Look for an rdataset with the specified name and type in the zone,
and return an RRset encapsulating it.
This method is less efficient than the similar ``get_rdataset()``
because it creates an RRset instead of returning the matching
rdataset. It may be more convenient for some uses since it
returns an object which binds the owner name to the rdataset.
This method may not be used to create new nodes or rdatasets;
use ``get_rdataset()`` instead.
*name*: the name of the node to find.
The value may be a ``dns.name.Name`` or a ``str``. If absolute, the
name must be a subdomain of the zone's origin. If ``zone.relativize``
is ``True``, then the name will be relativized.
*rdtype*, a ``dns.rdataset.Rdataset`` or ``str``, the rdata type desired.
*covers*, a ``dns.rdataset.Rdataset`` or ``str``, 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, the node will be created if it does
not exist.
Returns a ``dns.rrset.RRset`` or ``None``.
N)re r" )r4 r rT rU rb r r r � get_rrset s
(
zZone.get_rrset)rT rU r! c c sf t jj�|�}t jj�|�}| �� D ]<\}}|D ].}|t jjksT|j|kr0|j|kr0||fV q0q$dS )a� Return a generator which yields (name, rdataset) tuples for
all rdatasets in the zone which have the specified *rdtype*
and *covers*. If *rdtype* is ``dns.rdatatype.ANY``, the default,
then all rdatasets will be matched.
*rdtype*, a ``dns.rdataset.Rdataset`` or ``str``, the rdata type desired.
*covers*, a ``dns.rdataset.Rdataset`` or ``str``, 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.
N)r$ rV rW rX rJ �ANYrT rU )r4 rT rU r rP �rdsr r r �iterate_rdatasets0 s ��zZone.iterate_rdatasetsc c st t jj�|�}t jj�|�}| �� D ]J\}}|D ]<}|t jjksT|j|kr0|j|kr0|D ]}||j|fV qXq0q$dS )a� Return a generator which yields (name, ttl, rdata) tuples for
all rdatas in the zone which have the specified *rdtype*
and *covers*. If *rdtype* is ``dns.rdatatype.ANY``, the default,
then all rdatas will be matched.
*rdtype*, a ``dns.rdataset.Rdataset`` or ``str``, the rdata type desired.
*covers*, a ``dns.rdataset.Rdataset`` or ``str``, 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.
N) r$ rV rW rX rJ rg rT rU �ttl)r4 rT rU r rP rh �rdatar r r �iterate_rdatasP s ��zZone.iterate_rdatas)�f�sortedr �nl�
want_comments� want_originr! c
C s� t |t�rt|d�}n
t�|�}|��P}t|dd�}|du r@d}|du rZtj�|�} d}n"t |t�rp|�|�} n |} |� � }|r�d| j
� � }
|
�|�} z|� | � |� | � W n& t
y� |� |
� |� |� Y n0 |r�t| �� �} | �� n| �� } | D ]h}
| |
j |
| j
||d�}
|
�|�} z|� | � |� | � W q� t
�yd |� |
� |� |� Y q�0 q�W d � n1 �s~0 Y dS )a= Write a zone to a file.
*f*, a file or `str`. If *f* is a string, it is treated
as the name of a file to open.
*sorted*, a ``bool``. If True, the default, then the file
will be written with the names sorted in DNSSEC order from
least to greatest. Otherwise the names will be written in
whatever order they happen to have in the zone's dictionary.
*relativize*, a ``bool``. If True, the default, then domain
names in the output will be relativized to the zone's origin
if possible.
*nl*, a ``str`` or None. The end of line string. If not
``None``, the output will use the platform's native
end-of-line marker (i.e. LF on POSIX, CRLF on Windows).
*want_comments*, a ``bool``. If ``True``, emit end-of-line comments
as part of writing the file. If ``False``, the default, do not
emit them.
*want_origin*, a ``bool``. If ``True``, emit a $ORIGIN line at
the start of the file. If ``False``, the default, do not emit
one.
�wb�encodingNzutf-8�
z$ORIGIN )r r rp )r/ r0 �open�
contextlib� nullcontext�getattr�os�linesep�encode�decoder �to_text�write� TypeError�listrG �sort)r4 rm rn r ro rp rq �cmZfile_encZnl_b�lZl_b�names�nr r r �to_fileq sR $
�
z Zone.to_file)rn r ro rp rq r! c C s0 t �� }| �||||||� |�� }|�� |S )a Return a zone's text as though it were written to a file.
*sorted*, a ``bool``. If True, the default, then the file
will be written with the names sorted in DNSSEC order from
least to greatest. Otherwise the names will be written in
whatever order they happen to have in the zone's dictionary.
*relativize*, a ``bool``. If True, the default, then domain
names in the output will be relativized to the zone's origin
if possible.
*nl*, a ``str`` or None. The end of line string. If not
``None``, the output will use the platform's native
end-of-line marker (i.e. LF on POSIX, CRLF on Windows).
*want_comments*, a ``bool``. If ``True``, emit end-of-line comments
as part of writing the file. If ``False``, the default, do not
emit them.
*want_origin*, a ``bool``. If ``True``, emit a $ORIGIN line at
the start of the output. If ``False``, the default, do not emit
one.
Returns a ``str``.
)�io�StringIOr� �getvalue�close)r4 rn r ro rp rq Z temp_bufferZ return_valuer r r r} � s
!z Zone.to_text�r! c C sJ | j rtjj}n| j}| �|tjj�du r.t�| �|tjj �du rFt
�dS )z�Do some simple checking of the zone's origin.
Raises ``dns.zone.NoSOA`` if there is no SOA RRset.
Raises ``dns.zone.NoNS`` if there is no NS RRset.
Raises ``KeyError`` if there is no origin node.
N) r r$ r �emptyr r[ rV �SOAr ZNSr r; r r r � check_origin� s
zZone.check_origin)�txnr! c C s^ | j rtjj}n| jdu r t�| j}|r:|�|tjj�}n| � |tjj�}|du rVt�|d S )z�Get the zone SOA rdata.
Raises ``dns.zone.NoSOA`` if there is no SOA RRset.
Returns a ``dns.rdtypes.ANY.SOA.SOA`` Rdata.
Nr )
r r$ r r� r r rL rV r� r[ )r4 r� � origin_nameZsoar r r �get_soa s
z Zone.get_soa)�hash_algorithm�schemer! c s� t �|�}|st�|tjkr t�� jr0tjj }n� j
}|� }t � � � �D ]�\}}|�
� j
�}t |dd� d�D ]~} ||kr�tjj| j| jfv r�qlt�d| j| j| j�}
� fdd�| D �} t | �D ]*} t�dt| ��}
|�||
|
| � q�qlqH|�� S )Nc S s | j | jfS r< )rT rU )rh r r r �<lambda>1 � z&Zone._compute_digest.<locals>.<lambda>)r? z!HHIc s g | ]}|� � j��qS r )�
to_digestabler )�.0rk rE r r �
<listcomp>: r� z(Zone._compute_digest.<locals>.<listcomp>z!H)r rL r r
�SIMPLEr r r$ r r� r rn rJ r� rV �ZONEMDrT rU �struct�packr- rj r^ rd �digest)r4 r� r� Zhashinfor� Zhasherr rP Z rrnamebufrZ ZrrfixedZrdatasrk Zrrlenr rE r �_compute_digest s2
�� zZone._compute_digestc C s6 | � � j}| �||�}tjjj�| jtjj||||�S r< ) r� �serialr� r$ �rdtypesrg r� r- rV )r4 r� r� r� r� r r r �compute_digest@ s
�zZone.compute_digest)�zonemdr! c C sz |r |g}n"| � | jtjj�}|d u r*t�|}|D ]>}z&| �|j|j�}||j krZW d S W q2 t
yn Y q20 q2t �d S r< ) r[ r r$ rV r� r r� r� r� r� � Exceptionr )r4 r� Zdigestsrh r� Zcomputedr r r �
verify_digestK s
zZone.verify_digest� Transactionc C s t | dt| d| j| j��S )NF� )r� �Versionr. r rE r r r �readerb s z Zone.reader�r` r! c C s t | |�}|�� |S r< )r� �_setup_version)r4 r` r� r r r �writere s
z Zone.writerc C s$ | j rtjj}n| j}| j| j |fS r< )r r$ r r� r )r4 � effectiver r r �origin_informationj s
zZone.origin_informationc C s | j S r< )r- rE r r r � get_classt s zZone.get_classc C s d S r< r �r4 r� r r r � _end_ready s zZone._end_readc C s d S r< r r� r r r �
_end_write| s zZone._end_writec C s |j | _ | jd u r|| _d S r< )r. r )r4 �_�versionr r r r �_commit_version s
zZone._commit_versionc C s dS )Nr� r rE r r r �_get_next_version_id� s zZone._get_next_version_id)F)F)TTNFF)TTNFF)N)N)F)Wr r r r r$ rP �Noder' r �__annotations__�dictr( r r r2 r* r r, � __slots__�
rdataclass�INr r0 �
RdataClass�boolr5 r8 r9 r% r@ rB rC rD rG rH rJ rL rM rQ rR rS rV �NONErW rZ �RdatasetrY r[ r] ra rb rc re rf rg r r
ri �intrk ZRdatarl r r� r} r� � transactionr� r� r� r� r
r� r �bytesr� r� r� r� r� r� r� r� r� r� r� r� r r r r r& x s.
$�� �� �
� �� 3�
� 4�� '� �� 8�
� 0�� "�� $ �� [ �� ' �
� �� %�
� �� �
r& c s" e Zd ZdgZ� fdd�Z� ZS )�
VersionedNode�idc s t � �� d| _d S �Nr )�superr5 r� rE �� __class__r r r5 � s
zVersionedNode.__init__)r r r r� r5 �
__classcell__r r r� r r� � s 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 )�ImmutableVersionedNodec s, t � �� |j| _tdd� |jD ��| _d S )Nc S s g | ]}t j�|��qS r )r$ rZ ZImmutableRdataset)r� rh r r r r� � r� z3ImmutableVersionedNode.__init__.<locals>.<listcomp>)r� r5 r� �tuple� rdatasets)r4 rP r� r r r5 � s
�z ImmutableVersionedNode.__init__F)r- rT rU rN r! c s |r t d��t� �|||d�S �N� immutableF)r r� rY �r4 r- rT rU rN r� r r rY � s z$ImmutableVersionedNode.find_rdatasetc s |r t d��t� �|||d�S r� )r r� r[ r� r� r r r[ � s z#ImmutableVersionedNode.get_rdatasetN)r- rT rU r! c C s t d��d S �Nr� �r )r4 r- rT rU r r r r] � s z&ImmutableVersionedNode.delete_rdatasetr� c C s t d��d S r� r� )r4 r` r r r ra � s z'ImmutableVersionedNode.replace_rdatasetr� c C s dS �NTr rE r r r � is_immutable� s z#ImmutableVersionedNode.is_immutable)r r r r5 r$ rV r� r� r� rW r� rZ r� rY r r[ r] ra r� r� r r r� r r� � s8 ���
��� r� c @ s� e Zd Zdeeeeejj ej
j f eejj d�dd�Z ejj ejj d�dd�Z
ejj eej
j d�dd �Zejj ejjejjeejj d
�d d �Zd
d� Zdd� ZdS )r� N)�zoner� r. r c C s0 || _ || _|d ur || _n
|�� | _|| _d S r< )r� r� r. r( r )r4 r� r� r. r r r r r5 � s
zVersion.__init__r: c C s t || j| jj�S r< )r% r r� r r; r r r r% � s zVersion._validate_namec C s | � |�}| j�|�S r< rK r; r r r rR � s
zVersion.get_noder\ c C s( | � |�}|d u rd S |�| jj||�S r< )rR r[ r� r- r_ r r r r[ � s
zVersion.get_rdatasetc C s
| j �� S r< rF rE r r r rG � s z Version.keysc C s
| j �� S r< rI rE r r r rJ � s z
Version.items)NN)r r r r& r� r r r$ r r2 rP r� r5 r% rR rV rW rZ r� r[ rG rJ r r r r r� � s" �
�
� r� c s� e Zd Zdeed�� fdd�
Zejjej j
d�dd�Z ejjdd�d d
�Z ejjej
jdd �d d
�Zejjejjejjdd�dd�Z� ZS )r) F)r� r` c s< |� � }t� �||� |s(| j�|j� |j| _t� | _d S r< )r� r� r5 r. rd r �set�changed)r4 r� r` r� r� r r r5 � s zWritableVersion.__init__r: c C s| | � |�}| j�|�}|d u s(|| jvrt| j�� }t|d�rD| j|_|d urZ|j� |j� || j|< | j�
|� |S |S d S )Nr� ) r% r. rL r� r� r' �hasattrr� r� �extend�add)r4 r rP Znew_noder r r �
_maybe_cow s
zWritableVersion._maybe_cowNc C s, | � |�}|| jv r(| j|= | j�|� d S r< )r% r. r� r� r; r r r rS s
zWritableVersion.delete_node)r rZ r! c C s | � |�}|�|� d S r< )r� ra )r4 r rZ rP r r r � put_rdataset s
z WritableVersion.put_rdatasetr\ c C s4 | � |�}|�| jj||� t|�dkr0| j|= d S r� )r� r] r� r- r^ r. r_ r r r r] $ s
z WritableVersion.delete_rdataset)F)r r r r& r� r5 r$ r r2 rP r� r� rS rZ r� r� rV rW r] r� r r r� r r) � s � �r) c s" e Zd Zed�� fdd� Z� ZS )r+ )r� c sf t � �|jd� |j| _|j| _|jD ]"}|j�|�}|r&t|�|j|< q&t j
� |jd| jj �| _d S r� )
r� r5 r� r� r r� r. rL r� r$ r� �Dictr( )r4 r� r rP r� r r r5 2 s
�zImmutableVersion.__init__)r r r r) r5 r� r r r� r r+ 0 s r+ c s� e Zd Zd!� fdd� Zedd� �Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Z� ZS )"r� NFc s( |d u}t � �|||� || _|| _d S r< )r� r5 r� �make_immutable)r4 r� r` r� r� � read_onlyr� r r r5 I s zTransaction.__init__c C s | j S r< )�managerrE r r r r� O s zTransaction.zonec C s( | j j}|d u rt}|| j| j�| _d S r< )r� r* r) r� r` r� )r4 �factoryr r r r� S s zTransaction._setup_versionc C s | j �|||�S r< )r� r[ �r4 r rT rU r r r �
_get_rdatasetZ s zTransaction._get_rdatasetc C s | j �||� d S r< )r� r� )r4 r rZ r r r �
_put_rdataset] s zTransaction._put_rdatasetc C s | j �|� d S r< )r� rS r; r r r � _delete_namea s zTransaction._delete_namec C s | j �|||� d S r< )r� r] r� r r r �_delete_rdatasete s z Transaction._delete_rdatasetc C s | j �|�d uS r< �r� rR r; r r r � _name_existsi s zTransaction._name_existsc C s | j r
dS t| jj�dkS d S )NFr )r� r^ r� r� rE r r r �_changedl s zTransaction._changedc C sz | j r| j�| � nb|rjt| jj�dkrj| jrN| jj}|d u rBt }|| j�}n| j}| j�
| || jj � n | j� | � d S r� )
r� r� r� r^ r� r� r� r� r, r+ r� r r� )r4 Zcommitr� r� r r r �_end_transactionr s z Transaction._end_transactionc C s | j jd u r|| j _d S r< )r� r )r4 r r r r � _set_origin� s zTransaction._set_originc c s, | j �� D ] \}}|D ]}||fV qq
d S r< )r� rJ )r4 r rP rZ r r r �_iterate_rdatasets� s z Transaction._iterate_rdatasetsc C s
| j �� S r< )r� rG rE r r r �_iterate_names� s zTransaction._iterate_namesc C s | j �|�S r< r� r; r r r � _get_node� s zTransaction._get_nodec C sH | j �� \}}}|d u r>| jjd ur>| jj}|r:tjj}n|}|||fS r< )r� r� r� r r$ r r� )r4 �absoluter r� r r r �_origin_information� s
z Transaction._origin_information)NF)r r r r5 �propertyr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r� r r� H s
r� TF) �textr r- r � zone_factory�filename�
allow_includer� �
idna_codec�allow_directivesr! c
C s� |d u r d}||||d�}
|
� d��d} tjj| ||d�} tjj| || || d�}
z |
�� W n tjjyx tjj�Y n0 W d � n1 s�0 Y |r�|
� � |
S )Nz<string>�r T)r� )r� r� )
r� r$ Z tokenizer� TokenizerZzonefile�Reader�readr r� r� )r� r r- r r� r� r� r� r� r� r� r� �tokr� r r r �
_from_text� s& � ,r� c
C s t | ||||||||| �
S )a Build a zone object from a zone file format string.
*text*, a ``str``, the zone file format input.
*origin*, a ``dns.name.Name``, a ``str``, or ``None``. The origin
of the zone; if not specified, the first ``$ORIGIN`` statement in the
zone file will determine the origin of the zone.
*rdclass*, a ``dns.rdataclass.RdataClass``, the zone's rdata class; the default is
class IN.
*relativize*, a ``bool``, determine's whether domain names are
relativized to the zone's origin. The default is ``True``.
*zone_factory*, the zone factory to use or ``None``. If ``None``, then
``dns.zone.Zone`` will be used. The value may be any class or callable
that returns a subclass of ``dns.zone.Zone``.
*filename*, a ``str`` or ``None``, the filename to emit when
describing where an error occurred; the default is ``'<string>'``.
*allow_include*, a ``bool``. If ``True``, the default, then ``$INCLUDE``
directives are permitted. If ``False``, then encoutering a ``$INCLUDE``
will raise a ``SyntaxError`` exception.
*check_origin*, a ``bool``. If ``True``, the default, then sanity
checks of the origin node will be made by calling the zone's
``check_origin()`` method.
*idna_codec*, a ``dns.name.IDNACodec``, specifies the IDNA
encoder/decoder. If ``None``, the default IDNA 2003 encoder/decoder
is used.
*allow_directives*, a ``bool`` or an iterable of `str`. If ``True``, the default,
then directives are permitted, and the *allow_include* parameter controls whether
``$INCLUDE`` is permitted. If ``False`` or an empty iterable, then no directive
processing is done and any directive-like text will be treated as a regular owner
name. If a non-empty iterable, then only the listed directives (including the
``$``) are allowed.
Raises ``dns.zone.NoSOA`` if there is no SOA RRset.
Raises ``dns.zone.NoNS`` if there is no NS RRset.
Raises ``KeyError`` if there is no origin node.
Returns a subclass of ``dns.zone.Zone``.
)r� )
r� r r- r r� r� r� r� r� r� r r r r1 � s <�r1 ) rm r r- r r� r� r� r� r� r� r! c
C sl t | t�r |du r| }t| �}
n
t�| �}
|
�*} t| ||||||||| �
W d � S 1 s^0 Y dS )a Read a zone file and build a zone object.
*f*, a file or ``str``. If *f* is a string, it is treated
as the name of a file to open.
*origin*, a ``dns.name.Name``, a ``str``, or ``None``. The origin
of the zone; if not specified, the first ``$ORIGIN`` statement in the
zone file will determine the origin of the zone.
*rdclass*, an ``int``, the zone's rdata class; the default is class IN.
*relativize*, a ``bool``, determine's whether domain names are
relativized to the zone's origin. The default is ``True``.
*zone_factory*, the zone factory to use or ``None``. If ``None``, then
``dns.zone.Zone`` will be used. The value may be any class or callable
that returns a subclass of ``dns.zone.Zone``.
*filename*, a ``str`` or ``None``, the filename to emit when
describing where an error occurred; the default is ``'<string>'``.
*allow_include*, a ``bool``. If ``True``, the default, then ``$INCLUDE``
directives are permitted. If ``False``, then encoutering a ``$INCLUDE``
will raise a ``SyntaxError`` exception.
*check_origin*, a ``bool``. If ``True``, the default, then sanity
checks of the origin node will be made by calling the zone's
``check_origin()`` method.
*idna_codec*, a ``dns.name.IDNACodec``, specifies the IDNA
encoder/decoder. If ``None``, the default IDNA 2003 encoder/decoder
is used.
*allow_directives*, a ``bool`` or an iterable of `str`. If ``True``, the default,
then directives are permitted, and the *allow_include* parameter controls whether
``$INCLUDE`` is permitted. If ``False`` or an empty iterable, then no directive
processing is done and any directive-like text will be treated as a regular owner
name. If a non-empty iterable, then only the listed directives (including the
``$``) are allowed.
Raises ``dns.zone.NoSOA`` if there is no SOA RRset.
Raises ``dns.zone.NoNS`` if there is no NS RRset.
Raises ``KeyError`` if there is no origin node.
Returns a subclass of ``dns.zone.Zone``.
N)r/ r0 ru rv rw r� ) rm r r- r r� r� r� r� r� r� r� r r r � from_file s& =
�" r� )�xfrr� r r� r! c C s� d}| D ]�}|du rF|r |j }n |jd j}|jd j}||||d�}|jD ]`}|j�|j�} | sv|�� } | |j|j< | �|j|j|j d�}
|
�
|j � |D ]} |
� | � q�qLq|du r�t
d��|r�|�� |S )aN Convert the output of a zone transfer generator into a zone object.
*xfr*, a generator of ``dns.message.Message`` objects, typically
``dns.query.xfr()``.
*relativize*, a ``bool``, determine's whether domain names are
relativized to the zone's origin. The default is ``True``.
It is essential that the relativize setting matches the one specified
to the generator.
*check_origin*, a ``bool``. If ``True``, the default, then sanity
checks of the origin node will be made by calling the zone's
``check_origin()`` method.
Raises ``dns.zone.NoSOA`` if there is no SOA RRset.
Raises ``dns.zone.NoNS`` if there is no NS RRset.
Raises ``KeyError`` if there is no origin node.
Raises ``ValueError`` if no messages are yielded by the generator.
Returns a subclass of ``dns.zone.Zone``.
Nr r� Tzempty transfer)r Zanswerr r- r. rL r' rY rT rU Z
update_ttlrj r� r3 r� ) r� r� r r� �z�rr r- rb ZznodeZzrdsZrdr r r �from_xfre s,
r )Hr rv r� ry r� �typingr r r r r r r r r
r Z
dns.exceptionr$ Z
dns.grangeZ
dns.immutableZdns.nameZdns.nodeZ dns.rdataZdns.rdataclassZ dns.rdatasetZ
dns.rdatatypeZdns.rdtypes.ANY.SOAZdns.rdtypes.ANY.ZONEMDZ dns.rrsetZ
dns.tokenizerZdns.transactionZdns.ttlZ dns.zonefileZ
dns.zonetypesr r
r � exceptionZ DNSExceptionr r r r r r r r r r2 r� r% r� ZTransactionManagerr& rP r� r� r� r� r� r) r+ r� r� r� r0 r� Z IDNACodecr� r1 r� r r r r r �<module> s� 0
� -)?X�
� ,�
� L�
� U��