o
    'h\                     @   s   d Z ddlZddlZddlmZmZmZmZ ddlZ	ddl
Z	ddlZ	h dZdhZdZdZdZdZd	Zd
ZdZG dd de	jjZG dd dZG dd dZdS )zTokenize DNS zone file format    N)AnyListOptionalTuple>   
	 "();r	                     c                   @   s   e Zd ZdZdS )UngetBufferFullzDAn attempt was made to unget a token when the unget buffer was full.N)__name__
__module____qualname____doc__ r   r   I/var/www/html/olx_land/venv/lib/python3.10/site-packages/dns/tokenizer.pyr   (   s    r   c                
   @   s   e Zd ZdZ			d&dedededee fd	d
Z	defddZ
defddZdefddZdefddZdefddZdefddZdefddZdefddZdd Zdd Zd d! Zd'd"d#Zd'd$d%ZdS )(TokenzA DNS zone file format token.

    ttype: The token type
    value: The token value
    has_escape: Does the token value contain escapes?
     FNttypevalue
has_escapecommentc                 C   s   || _ || _|| _|| _dS )zInitialize a token instance.N)r   r   r   r   )selfr   r   r   r   r   r   r   __init__4   s   	
zToken.__init__returnc                 C   
   | j tkS N)r   EOFr    r   r   r   is_eofB      
zToken.is_eofc                 C   r#   r$   )r   EOLr&   r   r   r   is_eolE   r(   zToken.is_eolc                 C   r#   r$   )r   
WHITESPACEr&   r   r   r   is_whitespaceH   r(   zToken.is_whitespacec                 C   r#   r$   )r   
IDENTIFIERr&   r   r   r   is_identifierK   r(   zToken.is_identifierc                 C   r#   r$   )r   QUOTED_STRINGr&   r   r   r   is_quoted_stringN   r(   zToken.is_quoted_stringc                 C   r#   r$   )r   COMMENTr&   r   r   r   
is_commentQ   r(   zToken.is_commentc                 C   r#   r$   )r   	DELIMITERr&   r   r   r   is_delimiterT   r(   zToken.is_delimiterc                 C   s   | j tkp	| j tkS r$   )r   r)   r%   r&   r   r   r   is_eol_or_eofW   s   zToken.is_eol_or_eofc                 C   s&   t |tsdS | j|jko| j|jkS )NF
isinstancer   r   r   r    otherr   r   r   __eq__Z      
zToken.__eq__c                 C   s&   t |tsdS | j|jkp| j|jkS )NTr6   r8   r   r   r   __ne___   r;   zToken.__ne__c                 C   s   d| j | jf S )Nz%d "%s")r   r   r&   r   r   r   __str__d   s   zToken.__str__c                 C   s  | j s| S d}t| j}d}||k r| j| }|d7 }|dkr~||kr'tjj| j| }|d7 }| r~||kr<tjj| j| }|d7 }||krMtjj| j| }|d7 }| r^| sbtjjt|d t|d  t| }|dkrztjjt	|}||7 }||k st
| j|S )Nr   r   r   \d   
      )r   lenr   dns	exceptionUnexpectedEndisdigitSyntaxErrorintchrr   r   r    	unescapedlicc2c3	codepointr   r   r   unescapeg   s>   




 zToken.unescapec                 C   s*  d}t | j}d}||k r| j| }|d7 }|dkr||kr"tjj| j| }|d7 }| r|||kr7tjj| j| }|d7 }||krHtjj| j| }|d7 }| rY| s]tjjt|d t|d  t| }|dkrutjj|d| 7 }n|| 7 }n|| 7 }||k st	| j
t|S )	N    r   r   r>   r?   r@   rA   s   %c)rB   r   rC   rD   rE   rF   rG   rH   encoder   r   bytesrJ   r   r   r   unescape_to_bytes   s<   




  zToken.unescape_to_bytes)r   FN)r"   r   )r   r   r   r   rH   r   boolr   strr!   r'   r*   r,   r.   r0   r2   r4   r5   r:   r<   r=   rR   rV   r   r   r   r   r   ,   s6    


 r   c                   @   s  e Zd ZdZejddfdedee dee	j
j fddZdefd	d
Zdeeef fddZdeddfddZdefddZdBdededefddZdeddfddZdd ZeZdd ZdCdedefd d!Zdefd"d#ZdCdedefd$d%ZdCdedefd&d'ZdCdedefd(d)ZdDd*ee defd+d,Zdefd-d.Z dDd/ee de!e fd0d1Z"dEd2edefd3d4Z#			dFded5ee	j
j$ d6ed7ee	j
j$ de	j
j$f
d8d9Z%			dFd5ee	j
j$ d6ed7ee	j
j$ de	j
j$fd:d;Z&defd<d=Z'defd>d?Z(defd@dAZ)dS )G	Tokenizera  A DNS zone file format tokenizer.

    A token object is basically a (type, value) tuple.  The valid
    types are EOF, EOL, WHITESPACE, IDENTIFIER, QUOTED_STRING,
    COMMENT, and DELIMITER.

    file: The file to tokenize

    ungotten_char: The most recently ungotten character, or None.

    ungotten_token: The most recently ungotten token, or None.

    multiline: The current multiline level.  This value is increased
    by one every time a '(' delimiter is read, and decreased by one every time
    a ')' delimiter is read.

    quoting: This variable is true if the tokenizer is currently
    reading a quoted string.

    eof: This variable is true if the tokenizer has encountered EOF.

    delimiters: The current delimiter dictionary.

    line_number: The current line number

    filename: A filename that will be returned by the where() method.

    idna_codec: A dns.name.IDNACodec, specifies the IDNA
    encoder/decoder.  If None, the default IDNA 2003
    encoder/decoder is used.
    Nffilename
idna_codecc                 C   s   t |trt|}|du rd}n!t |tr$t| }|du r#d}n|du r2|tju r0d}nd}|| _d| _	d| _
d| _d| _d| _t| _d| _|dusPJ || _|du r^tjj| _dS || _dS )a  Initialize a tokenizer instance.

        f: The file to tokenize.  The default is sys.stdin.
        This parameter may also be a string, in which case the tokenizer
        will take its input from the contents of the string.

        filename: the name of the filename that the where() method
        will return.

        idna_codec: A dns.name.IDNACodec, specifies the IDNA
        encoder/decoder.  If None, the default IDNA 2003
        encoder/decoder is used.
        Nz<string>z<stdin>z<file>r   Fr   )r7   rX   ioStringIOrU   decodesysstdinfileungotten_charungotten_token	multilinequotingeof_DELIMITERS
delimitersline_numberr[   rC   name	IDNA_2003r\   )r    rZ   r[   r\   r   r   r   r!      s6   




zTokenizer.__init__r"   c                 C   s`   | j du r(| jrd}|S | jd}|dkrd| _|S |dkr&|  jd7  _|S | j }d| _ |S )zRead a character from input.Nr   r   Tr   )rc   rg   rb   readrj   r    rN   r   r   r   	_get_char  s   

zTokenizer._get_charc                 C   s   | j | jfS )zReturn the current location in the input.

        Returns a (string, int) tuple.  The first item is the filename of
        the input, the second is the current line number.
        )r[   rj   r&   r   r   r   where)  s   zTokenizer.whererN   c                 C      | j durt|| _ dS )a%  Unget a character.

        The unget buffer for characters is only one character large; it is
        an error to try to unget a character when the unget buffer is not
        empty.

        c: the character to unget
        raises UngetBufferFull: there is already an ungotten char
        N)rc   r   rn   r   r   r   _unget_char2  s   

zTokenizer._unget_charc                 C   sD   d}	 |   }|dkr|dkr|dks| js| | |S |d7 }q)aF  Consume input until a non-whitespace character is encountered.

        The non-whitespace character is then ungotten, and the number of
        whitespace characters consumed is returned.

        If the tokenizer is in multiline mode, then newlines are whitespace.

        Returns the number of characters skipped.
        r   Tr   r   r   r   )ro   re   rr   )r    skippedrN   r   r   r   skip_whitespaceB  s   
zTokenizer.skip_whitespaceFwant_leadingwant_commentc           	      C   s  | j dur| j }d| _ | r|r|S n| r|r|S n|S |  }|r.|dkr.ttdS d}t}d}	 |  }|dksB|| jv r|dkrM| j	rMt
jj|dkr|tkr|dkre|  jd7  _|   q4|d	kr~| jdkrrt
jj|  jd8  _|   q4|d
kr| j	sd| _	t| _t}q4d| _	t| _|   q4|dkrttdS |dkr	 |  }|dks|dkrn||7 }q|r| | tt|S |dkr| jrt
jdtt|dS | jr|   d}q4ttd|dS |}t}n| | n5| j	r|dkrt
jd|dkr$||7 }d}|  }|dks |dkr$| j	s$t
jj||7 }q5|dkr?|tkr?| jr=t
jdt}t|||S )a  Get the next token.

        want_leading: If True, return a WHITESPACE token if the
        first character read is whitespace.  The default is False.

        want_comment: If True, return a COMMENT token if the
        first token read is a comment.  The default is False.

        Raises dns.exception.UnexpectedEnd: input ended prematurely

        Raises dns.exception.SyntaxError: input was badly formed

        Returns a Token.
        Nr   r   r   FTr
   r   r   r	   r   r   zunbalanced parentheses)r   znewline in quoted stringr>   )rd   r,   r2   rt   r   r+   r-   ro   ri   rf   rC   rD   rE   r/   re   rG   _QUOTING_DELIMITERSrh   r)   rr   r1   r%   r3   )	r    ru   rv   utokenrs   tokenr   r   rN   r   r   r   getV  s   







HzTokenizer.getry   c                 C   rq   )a  Unget a token.

        The unget buffer for tokens is only one token large; it is
        an error to try to unget a token when the unget buffer is not
        empty.

        token: the token to unget

        Raises UngetBufferFull: there is already an ungotten token
        N)rd   r   r    ry   r   r   r   unget  s   

zTokenizer.ungetc                 C   s   |   }| r
t|S )zHReturn the next item in an iteration.

        Returns a Token.
        )rz   r'   StopIterationr{   r   r   r   next  s   zTokenizer.nextc                 C   s   | S r$   r   r&   r   r   r   __iter__  s   zTokenizer.__iter__r@   basec                 C   sB   |    }| stjd|j stjdt|j|S )zRead the next token and interpret it as an unsigned integer.

        Raises dns.exception.SyntaxError if not an unsigned integer.

        Returns an int.
        expecting an identifierzexpecting an integer)	rz   rR   r.   rC   rD   rG   r   rF   rH   )r    r   ry   r   r   r   get_int  s   
zTokenizer.get_intc                 C   s,   |   }|dk s|dkrtjd| |S )zRead the next token and interpret it as an 8-bit unsigned
        integer.

        Raises dns.exception.SyntaxError if not an 8-bit unsigned integer.

        Returns an int.
        r   rA   z#%d is not an unsigned 8-bit integerr   rC   rD   rG   )r    r   r   r   r   	get_uint8  s   	zTokenizer.get_uint8c                 C   sL   | j |d}|dk s|dkr$|dkrtj|ddtjd| |S )zRead the next token and interpret it as a 16-bit unsigned
        integer.

        Raises dns.exception.SyntaxError if not a 16-bit unsigned integer.

        Returns an int.
        r   r   i     oz( is not an octal unsigned 16-bit integerz$%d is not an unsigned 16-bit integerr   r    r   r   r   r   r   
get_uint16  s   	
zTokenizer.get_uint16c                 C   0   | j |d}|dk s|dkrtjd| |S )zRead the next token and interpret it as a 32-bit unsigned
        integer.

        Raises dns.exception.SyntaxError if not a 32-bit unsigned integer.

        Returns an int.
        r   r   l    z$%d is not an unsigned 32-bit integerr   r   r   r   r   
get_uint32     	zTokenizer.get_uint32c                 C   r   )zRead the next token and interpret it as a 48-bit unsigned
        integer.

        Raises dns.exception.SyntaxError if not a 48-bit unsigned integer.

        Returns an int.
        r   r   l    z$%d is not an unsigned 48-bit integerr   r   r   r   r   
get_uint48+  r   zTokenizer.get_uint48
max_lengthc                 C   sL   |    }| s| stjd|r#t|j|kr#tjd|jS )zRead the next token and interpret it as a string.

        Raises dns.exception.SyntaxError if not a string.
        Raises dns.exception.SyntaxError if token value length
        exceeds max_length (if specified).

        Returns a string.
        zexpecting a stringzstring too long)	rz   rR   r.   r0   rC   rD   rG   rB   r   )r    r   ry   r   r   r   
get_string;  s   
zTokenizer.get_stringc                 C   s&   |    }| stjd|jS )zRead the next token, which should be an identifier.

        Raises dns.exception.SyntaxError if not an identifier.

        Returns a string.
        r   )rz   rR   r.   rC   rD   rG   r   r{   r   r   r   get_identifierL  s   zTokenizer.get_identifier
max_tokensc                 C   sD   g }	 |   }| r| | 	 |S || t||kr!	 |S q)zReturn the remaining tokens on the line, until an EOL or EOF is seen.

        max_tokens: If not None, stop after this number of tokens.

        Returns a list of tokens.
        )rz   r5   r|   appendrB   )r    r   tokensry   r   r   r   get_remainingY  s   

zTokenizer.get_remainingallow_emptyc                 C   sZ   d}	 |    }| r| | n| stjj||j7 }q|s+|s+tjd|S )a  Read the remaining tokens on the line, which should be identifiers.

        Raises dns.exception.SyntaxError if there are no remaining tokens,
        unless `allow_empty=True` is given.

        Raises dns.exception.SyntaxError if a token is seen that is not an
        identifier.

        Returns a string containing a concatenation of the remaining
        identifiers.
        r   Tzexpecting another identifier)	rz   rR   r5   r|   r.   rC   rD   rG   r   )r    r   sry   r   r   r   !concatenate_remaining_identifiersl  s   

z+Tokenizer.concatenate_remaining_identifiersorigin
relativizerelativize_toc                 C   s8   |  s
tjdtj|j|| j}||p||S )zTry to interpret the token as a DNS name.

        Raises dns.exception.SyntaxError if not a name.

        Returns a dns.name.Name.
        r   )	r.   rC   rD   rG   rk   	from_textr   r\   choose_relativity)r    ry   r   r   r   rk   r   r   r   as_name  s   zTokenizer.as_namec                 C   s   |   }| ||||S )zRead the next token and interpret it as a DNS name.

        Raises dns.exception.SyntaxError if not a name.

        Returns a dns.name.Name.
        )rz   r   )r    r   r   r   ry   r   r   r   get_name  s   zTokenizer.get_namec                 C   s,   |   }| stjd|j|jf |S )znRead the next token and raise an exception if it isn't EOL or
        EOF.

        Returns a string.
        z expected EOL or EOF, got %d "%s")rz   r5   rC   rD   rG   r   r   r{   r   r   r   get_eol_as_token  s   zTokenizer.get_eol_as_tokenc                 C   s
   |   jS r$   )r   r   r&   r   r   r   get_eol  r(   zTokenizer.get_eolc                 C   s.   |    }| stjdtj|jS )zRead the next token and interpret it as a DNS TTL.

        Raises dns.exception.SyntaxError or dns.ttl.BadTTL if not an
        identifier or badly formed.

        Returns an int.
        r   )	rz   rR   r.   rC   rD   rG   ttlr   r   r{   r   r   r   get_ttl  s   	zTokenizer.get_ttl)FF)r@   r$   )F)NFN)*r   r   r   r   r`   ra   r   r   rX   rC   rk   	IDNACodecr!   ro   r   rH   rp   rr   rt   rW   r   rz   r|   r~   __next__r   r   r   r   r   r   r   r   r   r   r   Namer   r   r   r   r   r   r   r   r   rY      sv    "

1	o





rY   )r   r]   r`   typingr   r   r   r   dns.exceptionrC   dns.namedns.ttlrh   rw   r%   r)   r+   r-   r/   r1   r3   rD   DNSExceptionr   r   rY   r   r   r   r   <module>   s(    