File: //home/arjun/projects/env/lib/python3.10/site-packages/dns/__pycache__/zonefile.cpython-310.pyc
o
!wem � @ 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 dedee
e j j e.f dee! dee
e j+j/e.f de
e j+j/e.f dee
e j0j1e.f dee
e!e.f dee j j2 dee
e j j e.f d e"d ee j3j4 fd d �Z5dS ) z
DNS Zones.� N)�Any�Iterable�List�Optional�Set�Tuple�Unionc @ � e Zd ZdZdS )�
UnknownOriginzUnknown originN��__name__�
__module__� __qualname__�__doc__� r r �E/home/arjun/projects/env/lib/python3.10/site-packages/dns/zonefile.pyr
& � r
c @ r )�CNAMEAndOtherDataz!A node has a CNAME and other dataNr r r r r r * r r c C s| t jj�|�}| �|�}|d u rd S |�� }|t jjjkr(|t jjjkr(td��|t jjjkr:|t jjjkr<td��d S d S )Nz1rdataset type is not compatible with a CNAME nodez9CNAME rdataset is not compatible with a regular data node) �dns�node�NodeKind�classify_rdataset�get_node�classify�CNAME�REGULARr )�txn�name�rdataset�
rdataset_kindr � node_kindr r r �_check_cname_and_other_data. s
��r! c C s | � � } | �d�s
d| } | S )N�$)�upper�
startswith)�sr r r �_upper_dollarizeQ s
r& c @ s� e Zd ZdZ d dejjdejjdej j
de d e e e
e f d
eejj d ee d eejj d
eejj dee fdd�Zdd� Zdd� Zdd� Zdedeeeeeef fdd�Zdd � Zd d d �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 r(d| _d| _ n|
| _d| _ | j| _
|| _ || _ g | _
d | _| |du rRh d�| _|rQ| j�d� n|du r[t� | _n
tdd� |D ��| _|| _|| _|| _| | _| j �t� d S )Nr FT> �$TTL�$ORIGIN� $GENERATE�$INCLUDEc s s � | ]}t |�V qd S �N)r& )�.0�dr r r � <genexpr>� s � z"Reader.__init__.<locals>.<genexpr>)r( �manager�origin_information� zone_origin�
relativize�current_origin�last_ttl�last_ttl_knownr0 �default_ttl_known� last_name� zone_rdclassr � saved_state� current_file�allowed_directives�add�setr, r- r. r/ �check_put_rdatasetr! ) �selfr( r) r r* r+ r, r- r. r/ r0 �_r r r �__init__\ s>
�
zReader.__init__c C s | j �� }|�� r d S qr5 )r( �get�
is_eol_or_eof�rI �tokenr r r � _eat_line� s
�zReader._eat_linec C s | j �� }|�� s
tjj�|S r5 )r( rL �
is_identifierr � exception� SyntaxErrorrN r r r �_get_identifier� s
zReader._get_identifierc
C s& d}| j dur | j }nE| jdu rt�| jjdd�}|�� s'| j�|| j�| _n| j�� }|�� r2dS | j� |� | j}|�
| j �sG| � � dS | j
rP|�
| j �}| jdur_| j}|| _d| _n)| �� }d}ztj�|j�}|| _d| _d}W n tjjy� | j� |� Y nw | jdur�| j}n4| �� }z tj�|j�}W n tjjy� � ty� | j }| j� |� Y nw || j kr�tj�d��|du r�| �� }d}ztj�|j�}|| _d| _d}W n tjjy� | j r�| j }n| jr�| j}| j� |� Y nw | j du�r | j }n | �� }z tj �|j�}W n t�y' tj�d|j ��w ztj!�||| j| j| j
| j �}W n* tjj�yD � t�yc t"�#� dd� \}}tj�d�$t%|�t%|����w | j �s~|tj j&k�r~|j'| _ d| _ |du �r~|j'}|du �r�tj�d��| j(�)|||� dS ) z&Process one line from a DNS zone file.NT)� want_leading� RR class is not zone's class�unknown rdatatype '%s'� zcaught exception {}: {}�Missing default TTL value)*r, r= r
r( rL �
is_whitespace�as_namerA rM �unget� is_subdomainr; rP r< r- r> r? rT r �ttl� from_text�value�BadTTLr. �
rdataclassrR rS � ExceptionrB r@ r0 r/ � rdatatype�rdata�sys�exc_info�format�str�SOA�minimumr rF ) rI rO r r^ r) �rdtype�rd�ty�var r r �_rr_line� s�
�
�
� ����
zReader._rr_line�side�returnc
C s t �d�}t �d�}t �d�}|�|�}|r%|�� \}}}} }
|dkr%d}|�|�} | r=| �� \}}}|dkr9d}d} d}
|�|�} | rT| �� \}}}} |dkrRd}d}
|sd| sd| sdd}d}d}d} d}
t|�}t| �} |dvrxtj�d | ��|
d
vr�tj�d |
��|||| |
fS ) Nz"^.*\$({(\+|-?)(\d+),(\d+),(.)}).*$z^.*\$({(\+|-?)(\d+)}).*$z ^.*\$({(\+|-?)(\d+),(\d+)}).*$� �+r r7 )rt �-zinvalid offset sign %s)r7 �o�x�X�n�Nzinvalid type %s)�re�compile�match�groups�intr rR rS )
rI rq � is_generate1� is_generate2� is_generate3�g1�mod�sign�offset�width�base�g2�g3r r r �
_parse_modify sD
zReader._parse_modifyc
s� | j du rt�| j�� }ztj�|j�\}}}| j�� }|�� s$tj j
�W n t y0 tj j
�w z|j}| j�� }|�� sBtj j
�W n t yN tj j
�w z tj �|j�}|| _
d| _| j�� }|�� sjtj j
�W n$ tj jy� | js�| js�tj �
d��| jr�| j}n| jr�| j
}Y nw ztj�|j�}| j�� }|�� s�tj j
�W n tj j
y� tj j
� t y� | j}Y nw || jkr�tj �
d��ztj�|j�}| j�� }|�� s�tj j
�W n t y� tj �
d|j ��w |j} dtdtdtd tfd
d �}
d td
tdtd tf� fdd� � | �|�\} } }
}}| �| �\}}}}}t||d |�D ]�}|
|| |
�}|
|||�}� |||�}� |||�}|�d| |�}| �d| |�}tj�|| j | jj�| _ | j }|� | j ��ss| � � dS | j �r}|� | j �}ztj!�|||| j | j | j �} W n* tj j
�y� � t �y� t"�#� dd� \} } tj �
dt| �t| �f ��w | j$�%||| � �q,dS )zPProcess one line containing the GENERATE statement from a DNS
zone file.NTrY rV rW �counter� offset_signr� rr c S s |dkr|d9 }| | S )z0Calculate the index from the counter and offset.ru ���r )r� r� r� r r r �_calculate_index� s z/Reader._generate_line.<locals>._calculate_index�indexr� r� c sT |dv r t | |��|�S � | d|�}d�|ddd� �d|� }|dkr(|�� }|S )zVFormat the index with the given base, and zero-fill it
to the given width.)r7 rv rw rx rw �.Nr� rz )rh �zfill�joinr# )r� r� r� �hexa�nibbles��
_format_indexr r r� � s z,Reader._generate_line.<locals>._format_index� z$%srX zcaught exception %s: %s)&r= r
r( rL r �granger_ r` rQ rR rS rc r^ r> r? ra r@ r0 rb rB rd r ri r� �range�replacer �
idna_codecrA r] r; rP r< re rf rg r rF ) rI rO �start�stop�step�lhsr^ r) rl �rhsr� �lmod�lsign�loffset�lwidth�lbase�rmod�rsign�roffset�rwidth�rbase�i�lindex�rindex�lzfindex�rzfindexr re rm rn ro r r� r �_generate_lineA s�
� �
� �
� ��
�
�
� �
� ��� �zReader._generate_linec
C s� �z, | j �dd�}|�� r9| jdur| j�� t| j�dkr6| j�d�\| _ | _| _ | _| _
| _ | _ | _
qW dS |�� r>q|�� rH| j �� q|jd dk�r#t| j�dk�r#|j�� }|| jvrltj�d|� d���|d kr�| j �� }|�� stj�d
��tj�|j�| _ d| _
| j �� n�|d kr�| j �� | _| j �� | jdu r�| j| _| j � | j� nr|d k�r| j �� }|j}| j �� }|�� r�tj �|j| j| j j �}| j �� n
|� � s�tj�d
��| j}| j�!| j | j| j | j| j
| j | j | j
f� t"|d�| _tj#�$| j|�| _ || _n|dk�r| �%� n
tj�d|� d���q| j �&|� | �'� q tjj�ya } z%| j �(� \}}|du �rFd}tj�d|||f �}t)�*� d }|�+|�d�d}~ww )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
r� TNr r� r" zzone file directive 'z' is not allowedr1 zbad $TTLr2 r4 zbad origin in $INCLUDE�rr3 z Unknown zone file directive '�'z syntax errorz %s:%d: %srX ),r( rL �is_eofrD �close�lenrC �popr= rA r> r? r0 r@ �is_eol�
is_comment�get_eolr` rE r# r rR rS rQ r^ r_ �get_namer; r � _set_originr r� rM �append�open� tokenizer� Tokenizerr� r\ rp �whererf rg �with_traceback) rI rO �c�filename�
new_origin�detail� line_number�ex�tbr r r �read� s�
�
�
� ��
� �R
� ��z Reader.read)FTNNNNN)rr N) r r
r r r r� r� rb �
RdataClass� transaction� Transaction�boolr r ri r r �Namer rd � RdataTyperK rP rT rp r r� r� r� r r r r r' X sJ ������
��
�
�
�2 z- 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" |rJ �t � �|||� i | _d S r5 )�superrK � rdatasets)rI r9 � replacement� read_only�� __class__r r rK , s
z RRsetsReaderTransaction.__init__c C s | j �|||f�S r5 )r� rL �rI r rl �coversr r r �
_get_rdataset1 s z%RRsetsReaderTransaction._get_rdatasetc C sT g }| j �� D ]\\}}}}||kr|�|� qt|�dkr d S tj�� }||_ |S �Nr )r� �itemsr� r� r r �Node)rI r r� �
rdataset_namerJ r r r r r � _get_node4 s
�
z!RRsetsReaderTransaction._get_nodec C s || j ||j|jf< d S r5 )r� rl r� )rI r r r r r �
_put_rdataset? s z%RRsetsReaderTransaction._put_rdatasetc C sL g }| j D ]
}|d |kr|�|� qt|�dkr"|D ]}| j |= qd S d S r� )r� r� r� )rI r �remove�keyr r r � _delete_nameB s
�
�z$RRsetsReaderTransaction._delete_namec C s* z
| j |||f= W d S ty Y d S w r5 )r� �KeyErrorr� r r r �_delete_rdatasetL s
�z(RRsetsReaderTransaction._delete_rdatasetc C s$ | j D ] \}}}||kr dS qdS �NTF)r� )rI r ry rJ r r r � _name_existsR s
�z$RRsetsReaderTransaction._name_existsc C s t | j�dkS r� )r� r� �rI r r r �_changedX s z RRsetsReaderTransaction._changedc C sn |r3| � � r5g }| j�� D ] \\}}}}tj�||j|j|j�}|� |� |�
|� q
| j � |� d S d S d S r5 )
r� r� r� r �rrset�RRsetr) rl r� �updater� r9 �
set_rrsets)rI �commit�rrsetsr rJ r r� r r r �_end_transaction[ s �
�z(RRsetsReaderTransaction._end_transactionc C s d S r5 r )rI �originr r r r� f � z#RRsetsReaderTransaction._set_originc C � t �r5 ��NotImplementedErrorr� r r r �_iterate_rdatasetsi r� z*RRsetsReaderTransaction._iterate_rdatasetsc C r� r5 r� r� r r r �_iterate_namesl r� z&RRsetsReaderTransaction._iterate_names)r r
r rK 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 r5 )r� r< r) r� )rI r� r< r) r r r rK q s
z RRSetsReaderManager.__init__c C r� r5 r� r� r r r �readery r� zRRSetsReaderManager.readerc C s |du sJ �t | dd�S r� )r� )rI r� r r r �writer| s zRRSetsReaderManager.writerc C s | j S r5 )r) r� r r r � get_class� s z RRSetsReaderManager.get_classc C s$ | j rtjj}n| j}| j| j |fS r5 )r< r r �emptyr� )rI � effectiver r r r: � s
z&RRSetsReaderManager.origin_informationc C s
|| _ d S r5 )r� )rI r� r r r r� � s
z RRSetsReaderManager.set_rrsetsN)F)r r
r r r �rootrb �INrK r� r� r� r: r� r r r r r� p s
�
r� F�textr r^ r) �default_rdclassrl r0 r� r� r< rr c
C s t |t�rtj�|tjj|�}t |t�r tj�|||�}t |t�r'tj�|�}t |t�r2tj�|�}|dur>tjj� |�}nd}tjj� |�}|durStj
j � |�}nd}t || |�}
|
�
d��$} tjj| d|d�} t| || d|||||d� }
|
�� W d � |
jS 1 s�w 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/ r0 )�
isinstanceri r r r_ r� r^ rb r� �makerd r� r� r� r� r� r' r� r� )r r r^ r) r rl r0 r� r� r< r9 r r( r� r r r � read_rrsets� sD
=
�
��r )6r r{ rf �typingr r r r r r r �
dns.exceptionr �
dns.grange�dns.name�dns.node� dns.rdata�dns.rdataclass�
dns.rdatatype�dns.rdtypes.ANY.SOA� dns.rrset�
dns.tokenizer�dns.transaction�dns.ttlrR � DNSExceptionr
r r! r� r� r r� r r� �SavedStateTyper& r' r� r� r� �TransactionManagerr� rb r r� ri r� rd r� � IDNACodecr� r� r r r r r �<module> s� $
�� VE ��������
� �
�
�