a
�M�em �
@ s� d Z ddlZddlZddlmZmZmZmZmZm Z m
Z
ddl Z ddl
Z ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ G dd� de jj�ZG dd� de jj�Zdd � Z e e j j ee j j ee j j ee e!e"e!e"f Z#d
d � Z$G d d
� d
�Z%G dd� de j&j'�Z(G dd� de j&j)�Z*dde j+j,e j+j,ddde j j-df eee
e j j e.f ee! ee
e j+j/e.f e
e j+j/e.f ee
e j0j1e.f ee
e!e.f ee j j2 ee
e j j e.f e"ee j3j4 d� dd�Z5dS )z
DNS Zones.� N)�Any�Iterable�List�Optional�Set�Tuple�Unionc @ s e Zd ZdZdS )�
UnknownOriginzUnknown originN��__name__�
__module__� __qualname__�__doc__� r r �0/usr/lib/python3.9/site-packages/dns/zonefile.pyr & s r c @ s e Zd ZdZdS )�CNAMEAndOtherDataz!A node has a CNAME and other dataNr
r r r r r * s r c C sz t jj�|�}| �|�}|d u r$d S |�� }|t jjjkrR|t jjjkrRtd��n$|t jjjkrv|t jjjkrvtd��d S )Nz1rdataset type is not compatible with a CNAME nodez9CNAME rdataset is not compatible with a regular data node) �dns�nodeZNodeKindZclassify_rdatasetZget_nodeZclassifyZCNAMEZREGULARr )�txn�name�rdatasetZ
rdataset_kindr Z node_kindr r r �_check_cname_and_other_data. s"
� �
� ��r c C s | � � } | �d�sd| } | S )N�$)�upper�
startswith)�sr r r �_upper_dollarizeQ s
r c @ s� e Zd ZdZdejjejjej j
e e e e
e f eejj ee eejj eejj ee d�
dd�Zdd � Zd
d � Zd d
� Zeeeeeeef d�dd�Zdd� Zdd�dd�Z dS )�Readerz(Read a DNS zone file into a transaction.FTN)
�tok�rdclassr �
allow_include�allow_directives�
force_name� force_ttl�
force_rdclass� force_rdtype� default_ttlc C s� || _ |j�� \| _| _} | j| _d| _d| _|d ur:|}
|
d u rPd| _d| _ n |
| _d| _ | j| _
|| _ || _ g | _
d | _| |du r�h d�| _|r�| j�d� n&|du r�t� | _ntdd� |D ��| _|| _|| _|| _| | _| j �t� d S )Nr FT> � $GENERATE�$TTL�$ORIGIN�$INCLUDEc s s | ]}t |�V qd S �N)r )�.0�dr r r � <genexpr>� � z"Reader.__init__.<locals>.<genexpr>)r �manager�origin_information� zone_origin�
relativize�current_origin�last_ttl�last_ttl_knownr& �default_ttl_known� last_name� zone_rdclassr � saved_state� current_file�allowed_directives�add�setr" r# r$ r% Zcheck_put_rdatasetr ) �selfr r r r r! r" r# r$ r% r& �_r r r �__init__[ s<
zReader.__init__c C s | j �� }|�� r qq d S r+ )r �get�
is_eol_or_eof�r? �tokenr r r � _eat_line� s
zReader._eat_linec C s | j �� }|�� stjj�|S r+ )r rB �
is_identifierr � exception� SyntaxErrorrD r r r �_get_identifier� s
zReader._get_identifierc
C sH d}| j dur| j }n�| jdu r$t�| jjdd�}|�� sN| j�|| j�| _n"| j�� }|�� rddS | j� |� | j}|�
| j �s�| � � dS | j
r�|�
| j �}| jdur�| j}|| _d| _nT| �� }d}z"tj�|j�}|| _d| _d}W n$ tjj�y | j� |� Y n0 | jdu�r&| j}nr| �� }ztj�|j�}W n> tjj�yZ � Y n& t�y~ | j }| j� |� Y n0 || j k�r�tj�d��|du �r| �� }d}z"tj�|j�}|| _d| _d}W nB tjj�y | j �r�| j }n| j�r| j}| j� |� Y n0 | j du�r(| j }nB| �� }ztj �|j�}W n& t�yh tj�d|j ��Y n0 z"tj!�||| j| j| j
| j �}W n\ tjj�y� � Y nD t�y� t"�#� dd� \}}tj�d�$t%|�t%|����Y n0 | j �s |tj j&k�r |j'| _ d| _ |du �r |j'}|du �r4tj�d��| j(�)|||� dS ) z&Process one line from a DNS zone file.NT)Z want_leading� RR class is not zone's class�unknown rdatatype '%s'� zcaught exception {}: {}�Missing default TTL value)*r" r4 r r rB Z
is_whitespaceZas_namer8 rC �unget� is_subdomainr2 rF r3 r# r5 r6 rJ r �ttl� from_text�value�BadTTLr$ �
rdataclassrH rI � Exceptionr9 r7 r& r% � rdatatype�rdata�sys�exc_info�format�strZSOAZminimumr r= ) r? rE r rQ r �rdtype�rd�ty�var r r �_rr_line� s�
��
zReader._rr_line)�side�returnc
C s t �d�}t �d�}t �d�}|�|�}|rJ|�� \}}}} }
|dkrJd}|�|�} | rz| �� \}}}|dkrrd}d} d}
|�|�} | r�| �� \}}}} |dkr�d}d}
|s�| s�| s�d}d}d}d} d}
t|�}t| �} |dvr�tj�d | ��|
d
v�r
tj�d |
��|||| |
fS ) Nz"^.*\$({(\+|-?)(\d+),(\d+),(.)}).*$z^.*\$({(\+|-?)(\d+)}).*$z ^.*\$({(\+|-?)(\d+),(\d+)}).*$� �+r r- )re �-zinvalid offset sign %s)r- �o�x�X�n�Nzinvalid type %s)�re�compile�match�groups�intr rH rI )
r? rb Z is_generate1Z is_generate2Z is_generate3Zg1�mod�sign�offset�width�baseZg2Zg3r r r �
_parse_modify sD
zReader._parse_modifyc
s� | j du rt�| j�� }z2tj�|j�\}}}| j�� }|�� sHtj j
�W n t yd tj j
�Y n0 z$|j}| j�� }|�� s�tj j
�W n t y� tj j
�Y n0 z8tj �|j�}|| _
d| _| j�� }|�� s�tj j
�W nR tj j�y0 | j�s| j�stj �
d��| j�r | j}n| j�r,| j
}Y n0 z.tj�|j�}| j�� }|�� �s^tj j
�W n8 tj j
�y� tj j
�Y n t �y� | j}Y n0 || jk�r�tj �
d��z.tj�|j�}| j�� }|�� �s�tj j
�W n& t �y tj �
d|j ��Y n0 |j} ttttd�dd�}
ttttd �� fd
d � � | �|�\} } }
}}| �| �\}}}}}t||d |�D �]2}|
|| |
�}|
|||�}� |||�}� |||�}|�d
| |�}| �d
| |�}tj�|| j | jj�| _ | j }|� | j ��s| � � dS | j �r|� | j �}z tj!�|||| j | j | j �} W n\ tj j
�yP � Y nD t �y� t"�#� dd� \} } tj �
dt| �t| �f ��Y n0 | j$�%||| � �qrdS )zPProcess one line containing the GENERATE statement from a DNS
zone file.NTrN rK rL )�counter� offset_signrs rc c S s |dkr|d9 }| | S )z0Calculate the index from the counter and offset.rf ���r )rw rx rs r r r �_calculate_index� s z/Reader._generate_line.<locals>._calculate_index)�indexru rt rc c sT |dv rt | |��|�S � | d|�}d�|ddd� �d|� }|dkrP|�� }|S )zVFormat the index with the given base, and zero-fill it
to the given width.)r- rg rh ri rh �.Nry rk )r[ �zfill�joinr )r{ ru rt ZhexaZnibbles��
_format_indexr r r� � s z,Reader._generate_line.<locals>._format_index� z$%srM zcaught exception %s: %s)&r4 r r rB r ZgrangerR rS rG rH rI rV rQ r5 r6 rT r7 r& rU r9 rW rp r\ rv �range�replacer �
idna_codecr8 rP r2 rF r3 rX rY rZ r r= ) r? rE �start�stop�stepZlhsrQ r r] Zrhsrz ZlmodZlsignZloffsetZlwidthZlbaseZrmodZrsignZroffsetZrwidthZrbase�iZlindex�rindexZlzfindexZrzfindexr rX r^ r_ r` r r r �_generate_line@ s�
� ��
zReader._generate_line)rc c
C s� �zr| j �dd�}|�� rr| jdur.| j�� t| j�dkrj| j�d�\| _ | _| _ | _| _
| _ | _ | _
q�qr�n�|�� r�q�n�|�� r�| j �� q�n�|jd dk�r\t| j�dk�r\|j�� }|| jvr�tj�d|� d���|dk�r,| j �� }|�� �s
tj�d ��tj�|j�| _ d| _
| j �� q|d
k�rp| j �� | _| j �� | jdu �r`| j| _| j � | j� q|d k�r2| j �� }|j}| j �� }|�� �r�tj �|j| j| j j �}| j �� n |� � �s�tj�d ��n| j}| j�!| j | j| j | j| j
| j | j | j
f� t"|d
�| _tj#�$| j|�| _ || _q|dk�rF| �%� qtj�d|� d���q| j �&|� | �'� qW nt tjj�y� } zV| j �(� \}}|du �r�d}tj�d|||f �}t)�*� d }|�+|�d�W Y d}~n
d}~0 0 dS )z�Read a DNS zone file and build a zone object.
@raises dns.zone.NoSOA: No SOA RR was found at the zone origin
@raises dns.zone.NoNS: No NS RRset was found at the zone origin
TNr ry r zzone file directive 'z' is not allowedr( zbad $TTLr) r* zbad origin in $INCLUDE�rr' z Unknown zone file directive '�'z syntax errorz %s:%d: %srM ),r rB Zis_eofr; �close�lenr: �popr4 r8 r5 r6 r& r7 Zis_eolZ
is_commentZget_eolrS r< r r rH rI rG rQ rR Zget_namer2 r � _set_originr r� rC �append�open� tokenizer� Tokenizerr� rO ra �whererY rZ �with_traceback) r? rE �c�filenameZ
new_originZdetailZ line_number�ex�tbr r r �read� s�
�
�
�
��
�
� z Reader.read)FTNNNNN) r r r
r r r� r� rU �
RdataClass� transaction� Transaction�boolr r r\ r r �Namerp rW � RdataTyperA rF rJ ra r rv r� r� r r r r r X s6 �
� 2z - r c st e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Z� ZS )�RRsetsReaderTransactionc s t � �|||� i | _d S r+ )�superrA � rdatasets)r? r0 � replacementZ read_only�� __class__r r rA + s z RRsetsReaderTransaction.__init__c C s | j �|||f�S r+ )r� rB �r? r r] �coversr r r �
_get_rdataset0 s z%RRsetsReaderTransaction._get_rdatasetc C sT g }| j �� D ] \\}}}}||kr|�|� qt|�dkr@d S tj�� }||_ |S �Nr )r� �itemsr� r� r r ZNode)r? r r� Z
rdataset_namer@ r r r r r � _get_node3 s
z!RRsetsReaderTransaction._get_nodec C s || j ||j|jf< d S r+ )r� r] r� )r? r r r r r �
_put_rdataset> s z%RRsetsReaderTransaction._put_rdatasetc C sH g }| j D ]}|d |kr
|�|� q
t|�dkrD|D ] }| j |= q6d S r� )r� r� r� )r? r �remove�keyr r r � _delete_nameA s
z$RRsetsReaderTransaction._delete_namec C s* z| j |||f= W n ty$ Y n0 d S r+ )r� �KeyErrorr� r r r �_delete_rdatasetK s z(RRsetsReaderTransaction._delete_rdatasetc C s$ | j D ]\}}}||kr dS qdS �NTF)r� )r? r rj r@ r r r � _name_existsQ s z$RRsetsReaderTransaction._name_existsc C s t | j�dkS r� )r� r� �r? r r r �_changedW s z RRsetsReaderTransaction._changedc C sf |rb| � � rbg }| j�� D ]:\\}}}}tj�||j|j|j�}|� |� |�
|� q| j � |� d S r+ )
r� r� r� r �rrset�RRsetr r] r� �updater� r0 �
set_rrsets)r? Zcommit�rrsetsr r@ r r� r r r �_end_transactionZ s �
z(RRsetsReaderTransaction._end_transactionc C s d S r+ r )r? �originr r r r� e s z#RRsetsReaderTransaction._set_originc C s t �d S r+ ��NotImplementedErrorr� r r r �_iterate_rdatasetsh s z*RRsetsReaderTransaction._iterate_rdatasetsc C s t �d S r+ r� r� r r r �_iterate_namesk s z&RRsetsReaderTransaction._iterate_names)r r r
rA r� r� r� r� r� r� r� r� r� r� r� �
__classcell__r r r� r r� * s
r� c @ sN e Zd Zejjdejjfdd�Zdd� Z ddd�Z
dd � Z d
d � Z d d
� Z
dS )�RRSetsReaderManagerFc C s || _ || _|| _g | _d S r+ )r� r3 r r� )r? r� r3 r r r r rA p s z RRSetsReaderManager.__init__c C s t �d S r+ r� r� r r r �readerx s zRRSetsReaderManager.readerc C s t | dd�S r� )r� )r? r� r r r �writer{ s zRRSetsReaderManager.writerc C s | j S r+ )r r� r r r � get_class s z RRSetsReaderManager.get_classc C s$ | j rtjj}n| j}| j| j |fS r+ )r3 r r �emptyr� )r? Z effectiver r r r1 � s
z&RRSetsReaderManager.origin_informationc C s
|| _ d S r+ )r� )r? r� r r r r� � s z RRSetsReaderManager.set_rrsetsN)F)r r r
r r �rootrU �INrA r� r� r� r1 r� r r r r r� o s �
r� F) �textr rQ r �default_rdclassr] r& r� r� r3 rc c
C s t |t�r tj�|tjj|�}t |t�r8tj�|||�}t |t�rNtj�|�}t |t�rdtj�|�}|dur|tjj� |�}nd}tjj� |�}|dur�tj
j � |�}nd}t || |�}
|
�
d��D} tjj| d|d�} t| || d|||||d� }
|
�� W d � n1 �s 0 Y |
jS )a� Read one or more rrsets from the specified text, possibly subject
to restrictions.
*text*, a file object or a string, is the input to process.
*name*, a string, ``dns.name.Name``, or ``None``, is the owner name of
the rrset. If not ``None``, then the owner name is "forced", and the
input must not specify an owner name. If ``None``, then any owner names
are allowed and must be present in the input.
*ttl*, an ``int``, string, or None. If not ``None``, the the TTL is
forced to be the specified value and the input must not specify a TTL.
If ``None``, then a TTL may be specified in the input. If it is not
specified, then the *default_ttl* will be used.
*rdclass*, a ``dns.rdataclass.RdataClass``, string, or ``None``. If
not ``None``, then the class is forced to the specified value, and the
input must not specify a class. If ``None``, then the input may specify
a class that matches *default_rdclass*. Note that it is not possible to
return rrsets with differing classes; specifying ``None`` for the class
simply allows the user to optionally type a class as that may be convenient
when cutting and pasting.
*default_rdclass*, a ``dns.rdataclass.RdataClass`` or string. The class
of the returned rrsets.
*rdtype*, a ``dns.rdatatype.RdataType``, string, or ``None``. If not
``None``, then the type is forced to the specified value, and the
input must not specify a type. If ``None``, then a type must be present
for each RR.
*default_ttl*, an ``int``, string, or ``None``. If not ``None``, then if
the TTL is not forced and is not specified, then this value will be used.
if ``None``, then if the TTL is not forced an error will occur if the TTL
is not specified.
*idna_codec*, a ``dns.name.IDNACodec``, specifies the IDNA
encoder/decoder. If ``None``, the default IDNA 2003 encoder/decoder
is used. Note that codecs only apply to the owner name; dnspython does
not do IDNA for names in rdata, as there is no IDNA zonefile format.
*origin*, a string, ``dns.name.Name``, or ``None``, is the origin for any
relative names in the input, and also the origin to relativize to if
*relativize* is ``True``.
*relativize*, a bool. If ``True``, names are relativized to the *origin*;
if ``False`` then any relative names in the input are made absolute by
appending the *origin*.
NTz<input>)r� F)r! r" r# r$ r% r& )�
isinstancer\ r r rR r� rQ rU r� ZmakerW r� r� r� r� r� r r� r� )r� r rQ r r� r] r& r� r� r3 r0 r r r� r r r � read_rrsets� s> =
� (r� )6r rl rY �typingr r r r r r r Z
dns.exceptionr Z
dns.grangeZdns.nameZdns.nodeZ dns.rdataZdns.rdataclassZ
dns.rdatatypeZdns.rdtypes.ANY.SOAZ dns.rrsetZ
dns.tokenizerZdns.transactionZdns.ttlrH Z DNSExceptionr r r r� r� r r� rp r� ZSavedStateTyper r r� r� r� ZTransactionManagerr� rU r� r� r\ r� rW r� Z IDNACodecr� r� r� r r r r �<module> st $
�� UE �
�