a
�M�e � @ sF d dl Z d dlZd dlmZmZmZ d dlZd dlZG dd� d�Z dS )� N)�Iterator�Optional�Tuplec @ s� e Zd Zd(eed�dd�Zed�dd�Zeed�d d
�Zd)eed �d
d�Zed�dd�Z ed�dd�Z
ed�dd�Z ed�dd�Z ed�dd�Z
eed�dd�Zd*ed d d �d d �Zed d!�d"d#�Zejeed�d$d%��Zejed�d&d'��Zd S )+�Parserr )�wire�currentc C s0 || _ d| _t| j �| _|r&| �|� || _d S �Nr )r r �len�end�seek�furthest)�selfr r � r �,/usr/lib/python3.9/site-packages/dns/wire.py�__init__ s
zParser.__init__)�returnc C s | j | j S �N)r
r �r
r r r � remaining s zParser.remaining)�sizer c C sX |dks J �|| � � kr tjj�| j| j| j| � }| j|7 _t| j| j�| _|S r )r �dns� exception� FormErrorr r �maxr )r
r �outputr r r � get_bytes s zParser.get_bytes� )� length_sizer c C s t �| �|�d�}| �|�S )N�big��int�
from_bytesr )r
r �lengthr r r �get_counted_bytes s zParser.get_counted_bytesc C s | � | �� �S r )r r r r r r �
get_remaining$ s zParser.get_remainingc C s t �d| �d��d S )Nz!Br r ��struct�unpackr r r r r � get_uint8' s zParser.get_uint8c C s t �d| �d��d S )Nz!H� r r% r r r r �
get_uint16* s zParser.get_uint16c C s t �d| �d��d S )Nz!I� r r% r r r r �
get_uint32- s zParser.get_uint32c C s t �| �d�d�S )N� r r r r r r �
get_uint480 s zParser.get_uint48)�formatr c C s t �|| �t �|���S r )r& r' r �calcsize)r
r/ r r r �
get_struct3 s zParser.get_structNz
dns.name.Name)�originr c C s t j�| �}|r|�|�}|S r )r �nameZfrom_wire_parserZ
relativize)r
r2 r3 r r r �get_name6 s
zParser.get_name)�wherer c C s$ |dk s|| j krtjj�|| _d S r )r
r r r r )r
r5 r r r r < s z Parser.seekc c sd |dks J �|| � � kr tjj�| j}z0| j| | _d V | j| jkrNtjj�W || _n|| _0 d S r )r r r r r
r )r
r Z saved_endr r r � restrict_toC s
zParser.restrict_toc c s" zd V W | j | _n
| j | _0 d S r )r r r r r r �restore_furthestT s zParser.restore_furthest)r )r )N)�__name__�
__module__� __qualname__�bytesr r r r r# r$ r( r* r, r. �strr r1 r r4 r �
contextlib�contextmanagerr r6 r7 r r r r r s r )
r= r& �typingr r r Z
dns.exceptionr Zdns.namer r r r r �<module> s