관리-도구
편집 파일: zonefile.cpython-38.pyc
U /?�fm � @ 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 �:/opt/hc_python/lib/python3.8/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 kr$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 k r:|} | d krPd| _d| _ n | | _d| _ | j| _ || _ || _ g | _ d | _| |dkr�dddh| _|r�| j�d� n&|dkr�t� | _ntdd � |D ��| _|| _|| _|| _| | _| j �t� d S ) Nr FT� $GENERATE�$ORIGIN�$TTL�$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_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 rA � is_identifierr � exception� SyntaxErrorrC r r r �_get_identifier� s zReader._get_identifierc C sV d}| j dk r| j }n�| jdkr$t�| jjdd�}|�� sN| j�|| j�| _n"| j�� }|�� rddS | j� |� | j}|� | j �s�| � � dS | j r�|� | j �}| jdk r�| j}|| _d| _nV| �� }d}z"tj�|j�}|| _d| _d}W n& tjjk �r | j� |� Y nX | jdk �r(| j}nv| �� }ztj�|j�}W nB tjjk �r^ � Y n( tk �r� | j }| j� |� Y nX || j k�r�tj�d��|dk�r | �� }d}z"tj�|j�}|| _d| _d}W nD tjjk �r | j �r�| j }n| j�r | j}| j� |� Y nX | j dk �r0| j }nD| �� }ztj �|j�}W n( tk �rr tj�d|j ��Y nX z"tj!�||| j| j| j | j �}W n` tjjk �r� � Y nF tk �r� t"�#� dd� \}}tj�d�$t%|�t%|����Y nX | j �s,|tj j&k�r,|j'| _ d| _ |dk�r,|j'}|dk�rBtj�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" r3 r r rA Z is_whitespaceZas_namer7 rB �unget� is_subdomainr1 rE r2 r# r4 r5 rI r �ttl� from_text�value�BadTTLr$ � rdataclassrG rH � Exceptionr8 r6 r& r% � rdatatype�rdata�sys�exc_info�format�strZSOA�minimumr r< ) r> rD r rP 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| �} |dkr�tj�d | ��| d k�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 rG rH ) r> rb Z is_generate1Z is_generate2Z is_generate3�g1�mod�sign�offset�width�base�g2Zg3r r r � _parse_modify sD zReader._parse_modifyc s� | j dkrt�| j�� }z2tj�|j�\}}}| j�� }|�� sHtj j �W n t k rf tj j �Y nX z$|j}| j�� }|�� s�tj j �W n t k r� tj j �Y nX z8tj �|j�}|| _ d| _| j�� }|�� s�tj j �W nT tj jk �r6 | j�s| j�stj � d��| j�r$| j}n| j�r2| j }Y nX z.tj�|j�}| j�� }|�� �sdtj j �W n< tj j k �r� tj j �Y n t k �r� | j}Y nX || jk�r�tj � d��z.tj�|j�}| j�� }|�� �s�tj j �W n( t k �r tj � d|j ��Y nX |j} ttttd�dd�} ttttd �� fd d � � | �|�\} } } }}| �| �\}}}}}t||d |�D �]6}| || | �}| |||�}� |||�}� |||�}|�d | |�}| �d | |�}tj�|| j | jj�| _ | j }|� | j ��s| � � dS | j �r"|� | j �}z tj!�|||| j | j | j �} W n` tj j k �r^ � Y nF t k �r� t"�#� dd� \} } tj � dt| �t| �f ��Y nX | j$�%||| � �q~dS )zPProcess one line containing the GENERATE statement from a DNS zone file.NTrM rJ rK )�counter� offset_signrt rc c S s |dkr|d9 }| | S )z0Calculate the index from the counter and offset.rf ���r )ry rz rt r r r �_calculate_index� s z/Reader._generate_line.<locals>._calculate_index)�indexrv ru rc c sT |dkrt | |��|�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 �.Nr{ rk )rZ �zfill�joinr )r} rv ru ZhexaZnibbles�� _format_indexr r r� � s z,Reader._generate_line.<locals>._format_index� z$%srL zcaught exception %s: %s)&r3 r r rA r ZgrangerQ rR rF rG rH rU rP r4 r5 rS r6 r& rT r8 rV rp r[ rx �range�replacer � idna_codecr7 rO r1 rE r2 rW rX rY r r< ) r> rD �start�stop�step�lhsrP r r] �rhsr| ZlmodZlsignZloffsetZlwidthZlbaseZrmodZrsignZroffsetZrwidthZrbase�iZlindex�rindexZlzfindexZrzfindexr rW r^ r_ r` r r� r �_generate_line@ s� � �� zReader._generate_line)rc c C s� �zr| j �dd�}|�� rr| jdk r.| j�� t| j�dkrj| j�d�\| _ | _| _ | _| _ | _ | _ | _ q�qr�n�|�� r�q�n�|�� r�| j �� q�n�|jd dk�r\t| j�dk�r\|j�� }|| jkr�tj�d|� d���|dk�r,| j �� }|�� �s tj�d ��tj�|j�| _ d| _ | j �� q|d k�rp| j �� | _| j �� | jdk�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 nr tjjk �r� } zN| j �(� \}}|dk�r�d}tj�d|||f �}t)�*� d }|�+|�d�W 5 d}~X Y nX 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 r{ 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: %srL ),r rA Zis_eofr: �close�lenr9 �popr3 r7 r4 r5 r& r6 Zis_eolZ is_commentZget_eolrR r; r r rG rH rF rP rQ �get_namer1 r � _set_originr r� rB �append�open� tokenizer� Tokenizerr� rN ra �whererX rY �with_traceback) r> rD �c�filenameZ new_origin�detail� line_number�ex�tbr r r �read� s� � � � �� � � z Reader.read)FTNNNNN) r r r r r r� r� rT � RdataClass� transaction� Transaction�boolr r r[ r r �Namerp rV � RdataTyper@ rE rI ra r rx 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" |rt �t� �|||� i | _d S r+ )�AssertionError�superr@ � rdatasets)r> r/ � replacementZ read_only�� __class__r r r@ + s z RRsetsReaderTransaction.__init__c C s | j �|||f�S r+ )r� rA �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 �Node)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 tk r& Y nX 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� r/ � set_rrsets)r> �commit�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 r@ 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� r2 r r� )r> r� r2 r r r r r@ p s z RRSetsReaderManager.__init__c C s t �d S r+ r� r� r r r �readerx s zRRSetsReaderManager.readerc C s |dks t �t| dd�S r� )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+ )r2 r r �emptyr� )r> Z effectiver r r r0 � 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 �rootrT �INr@ r� r� r� r0 r� r r r r r� o s � r� F) �textr rP r �default_rdclassr] r& r� r� r2 rc c C s t |t�r tj�|tjj|�}t |t�r8tj�|||�}t |t�rNtj�|�}t |t�rdtj�|�}|dk r|tjj� |�}nd}tjj� |�}|dk r�tj j � |�}nd}t || |�} | � d��:} tjj| d|d�} t| || d|||||d� } | �� W 5 Q R X | 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 rQ r� rP rT r� �makerV r� r� r� r� r� r r� r� )r� r rP r r� r] r& r� r� r2 r/ r r r� r r r � read_rrsets� s> = � r� )6r rl rX �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.ttlrG Z DNSExceptionr r r r� r� r r� rp r� ZSavedStateTyper r r� r� r� ZTransactionManagerr� rT r� r� r[ r� rV r� Z IDNACodecr� r� r� r r r r �<module> st $ �� UE � �