File: //home/arjun/projects/env/lib/python3.10/site-packages/dns/__pycache__/wire.cpython-310.pyc
o
!we � @ sF d dl Z d dlZd dlmZmZmZ d dlZd dlZG dd� d�Z dS )� N)�Iterator�Optional�Tuplec @ s e Zd Zd)dedefdd�Zdefdd�Zd edefd
d �Zd*d
edefdd�Zdefdd�Z defdd�Z
defdd�Z defdd�Z defdd�Z
dedefdd �Zd+d ed dd fd d!�Zd"edd fd#d$�Zejd edefd%d&��Zejdefd'd(��Zd S ),�Parserr �wire�currentc C s0 || _ d| _t| j �| _|r| �|� || _d S �Nr )r r �len�end�seek�furthest)�selfr r � r �A/home/arjun/projects/env/lib/python3.10/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�sizec C sX |dksJ �|| � � krtjj�| 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_sizec 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 � t �d| �d��d S )Nz!Br r ��struct�unpackr r r r r � get_uint8' � zParser.get_uint8c C r% )Nz!H� r r& r r r r �
get_uint16* r* zParser.get_uint16c C r% )Nz!I� r r& r r r r �
get_uint32- r* zParser.get_uint32c C s t �| �d�d�S )N� r r r r r r �
get_uint480 s zParser.get_uint48�formatc C s t �|| �t �|���S r )r' r( r �calcsize)r
r1 r r r �
get_struct3 s zParser.get_structN�originz
dns.name.Namec C s t j�| �}|r
|�|�}|S r )r �name�from_wire_parser�
relativize)r
r4 r5 r r r �get_name6 s
zParser.get_name�wherec C s$ |dk s || j kr
tjj�|| _d S r )r
r r r r )r
r9 r r r r < s
z Parser.seekc c sd � |dksJ �|| � � krtjj�| j}z| j| | _d V | j| jkr(tjj�W || _d S || _w r )r r r r r
r )r
r � saved_endr r r � restrict_toC s � �zParser.restrict_toc c s" � z
d V W | j | _d S | j | _w r )r r r r r r �restore_furthestT s �zParser.restore_furthest)r )r r )�__name__�
__module__� __qualname__�bytesr r r r r# r$ r) r, r. r0 �strr r3 r r8 r �
contextlib�contextmanagerr r; r< r r r r r s" r )
rB r' �typingr r r �
dns.exceptionr �dns.namer r r r r �<module> s