File: //home/arjun/projects/env/lib/python3.10/site-packages/dns/__pycache__/edns.cpython-310.pyc
o
!we�6 �
@ s� U d Z ddlZddlZddlZddlmZmZmZmZ ddl Z
ddl Z
ddl Z
ddl
Z
G dd� de
jj�ZG dd� d�ZG dd � d e�ZG d
d � d e�ZG d d
� d
e
jj�ZG dd� de�ZejeejeiZeeef ed< dedefdd�Zdeeef dddefdd�Z deeef de de de def
d d �Z d ededdfd d �Z ej!Z!ej"Z"ej#Z#ej$Z$ejZej%Z%ej&Z&ej'Z'ej(Z(ej)Z)ejZdS )!z EDNS Options� N)�Any�Dict�Optional�Unionc @ sD e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
ed d
� �ZdS )�
OptionType� � � � � � �
� � �
� c C � dS �Ni�� � ��clsr r �A/home/arjun/projects/env/lib/python3.10/site-packages/dns/edns.py�_maximum7 � zOptionType._maximumN)�__name__�
__module__� __qualname__�NSID�DAU�DHU�N3U�ECS�EXPIRE�COOKIE� KEEPALIVE�PADDING�CHAIN�EDE� classmethodr r r r r r s r c @ s� e Zd ZdZdeeef fdd�Zd dee dee
fdd �Z e ded
d dd fd d
��Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � ZdS ) �Optionz%Base class for all EDNS option types.�otypec C s t �|�| _dS )z_Initialize an option.
*otype*, a ``dns.edns.OptionType``, is the option type.
N)r �maker* )�selfr* r r r �__init__@ s zOption.__init__N�file�returnc C � t �)zUConvert an option to wire format.
Returns a ``bytes`` or ``None``.
��NotImplementedError�r, r. r r r �to_wireG s zOption.to_wire�parser�dns.wire.Parserc C r0 )a Build an EDNS option object from wire format.
*otype*, a ``dns.edns.OptionType``, is the option type.
*parser*, a ``dns.wire.Parser``, the parser, which should be
restructed to the option length.
Returns a ``dns.edns.Option``.
r1 �r r* r5 r r r �from_wire_parserO s zOption.from_wire_parserc C s, | � � }|� � }||krdS ||krdS dS )z�Compare an EDNS option with another option of the same type.
Returns < 0 if < *other*, 0 if == *other*, and > 0 if > *other*.
r � ���)r4 )r, �other�wire�owirer r r �_cmp\ s z Option._cmpc C s, t |t�sdS | j|jkrdS | �|�dkS )NFr ��
isinstancer) r* r> �r, r; r r r �__eq__i �
z
Option.__eq__c C s, t |t�sdS | j|jkrdS | �|�dkS )NTr r? rA r r r �__ne__p rC z
Option.__ne__c C s( t |t�r | j|jkr
tS | �|�dk S �Nr �r@ r) r* �NotImplementedr> rA r r r �__lt__w � z
Option.__lt__c C s( t |t�r | j|jkr
tS | �|�dkS rE rF rA r r r �__le__| rI z
Option.__le__c C s( t |t�r | j|jkr
tS | �|�dkS rE rF rA r r r �__ge__� rI z
Option.__ge__c C s( t |t�r | j|jkr
tS | �|�dkS rE rF rA r r r �__gt__� rI z
Option.__gt__c C s | � � S �N)�to_text�r, r r r �__str__� s zOption.__str__rM )r r r �__doc__r r �strr- r r �bytesr4 r( r8 r> rB rD rH rJ rK rL rP r r r r r) < s
r) c s� e Zd ZdZdeeef deeef f� fdd� Zdde e
de e fd d
�Z defd d �Z e
deeef d
ddefdd��Z� ZS )�
GenericOptionzwGeneric Option Class
This class is used for EDNS option types for which we have no better
implementation.
r* �datac s" t � �|� tjj�|d�| _d S )NT)�superr- �dns�rdata�Rdata� _as_bytesrU )r, r* rU �� __class__r r r- � s zGenericOption.__init__Nr. r/ c C s |r
|� | j� d S | jS rM )�writerU r3 r r r r4 � s zGenericOption.to_wirec C s
d| j S )Nz
Generic %d)r* rO r r r rN � s
zGenericOption.to_textr5 r6 c C s | ||� � �S rM )�
get_remainingr7 r r r r8 � s z GenericOption.from_wire_parserrM )r r r rQ r r rR rS r- r r r4 rN r( r) r8 �
__classcell__r r r[ r rT � s &
���rT c s� e Zd ZdZddedee def� fdd�
Zd efd
d �Ze d ed e
fd
d��Z ddee d ee
fdd�Zedeeef ddd e
fdd��Z� ZS )� ECSOptionz!EDNS Client Subnet (ECS, RFC7871)Nr �address�srclen�scopelenc s` t � �tj� tj�|�}|tjkr5d| _ |du rd}tj
j � |�}tj
j �
|dd�}tj
j �
|dd�}n,|tjkr]d| _ |du rCd}tj
j �|�}tj
j �
|dd�}tj
j �
|dd�}ntd ��|dusgJ �|| _|| _|| _tj�||�}tt�|d
��}|d|� | _|d }|dkr�t�d t| jd
d� �dd | > @ �}| jdd
� | | _dS dS )a� *address*, a ``str``, is the client address information.
*srclen*, an ``int``, the source prefix length, which is the
leftmost number of bits of the address to be used for the
lookup. The default is 24 for IPv4 and 56 for IPv6.
*scopelen*, an ``int``, the scope prefix length. This value
must be 0 in queries, and should be set in responses.
� N�8 r � r9 � � zBad address family� @r �Br: � ) rV r- r r! rW �inet�af_for_address�socket�AF_INET6�familyrX rY �_as_ipv6_address�_as_int�AF_INET�_as_ipv4_address�
ValueErrorra rb rc � inet_pton�int�math�ceil�addrdata�struct�pack�ord) r, ra rb rc �afrz �nbytes�nbits�lastr[ r r r- � s:
&�zECSOption.__init__r/ c C s d� | j| j| j�S )NzECS {}/{} scope/{})�formatra rb rc rO r r r rN � s zECSOption.to_text�textc
C s d}| � � }d}t|�dkr|d }n t|�dkr+|d |kr&td�| ���|d }ntd�| ���|�d�}|dkrE|� d�\}}d}n|dkrR|� d�\}}}ntd�| ���zt|�}W n tyo td d
�|� ��w zt|�} W n ty� td d �|� ��w t|| |�S )
a� Convert a string into a `dns.edns.ECSOption`
*text*, a `str`, the text form of the option.
Returns a `dns.edns.ECSOption`.
Examples:
>>> import dns.edns
>>>
>>> # basic example
>>> dns.edns.ECSOption.from_text('1.2.3.4/24')
>>>
>>> # also understands scope
>>> dns.edns.ECSOption.from_text('1.2.3.4/24/32')
>>>
>>> # IPv6
>>> dns.edns.ECSOption.from_text('2001:4b98::1/64/64')
>>>
>>> # it understands results from `dns.edns.ECSOption.to_text()`
>>> dns.edns.ECSOption.from_text('ECS 1.2.3.4/24/32')
r! Nr9 r rd z could not parse ECS from "{}"�/�0zinvalid scope z "{}": scope must be an integerzinvalid srclen z "{}": srclen must be an integer)�split�lenru r� �countrw r` )
r� �optional_prefix�tokens�ecs_text� n_slashesra �tsrclen�tscope�scoperb r r r � from_text� s@
�� �� zECSOption.from_textr. c C s2 t �d| j| j| j�| j }|r|�|� d S |S )N�!HBB)r{ r| rp rb rc rz r] �r, r. �valuer r r r4 s �
zECSOption.to_wirer* r5 r6 c
C s� |� d�\}}}tt�|d ��}|�|�}|dkr)d| }tj�|d| �} n|dkr<d| }tj�|d| �} nt d��| | ||�S ) Nr� ri r9 � � rd � zunsupported family)
�
get_structrw rx ry � get_bytesrW �ipv4� inet_ntoa�ipv6ru )
r r* r5 rp �srcr� �addrlen�prefix�pad�addrr r r r8 $ s
zECSOption.from_wire_parserrE rM )r r r rQ rR r rw r- rN � staticmethodr) r� r rS r4 r( r r r8 r_ r r r[ r r` � s / 8
���r` c @ s| e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZe dd� �Z d S ) �EDECoder r9 rd r r� r r r
r r r
r r r � r r� � � � � � � � rg c C r r r r r r r r R r zEDECode._maximumN) r r r �OTHER� UNSUPPORTED_DNSKEY_ALGORITHM�UNSUPPORTED_DS_DIGEST_TYPE� STALE_ANSWER�
FORGED_ANSWER�DNSSEC_INDETERMINATE� DNSSEC_BOGUS�SIGNATURE_EXPIRED�SIGNATURE_NOT_YET_VALID�DNSKEY_MISSING�RRSIGS_MISSING�NO_ZONE_KEY_BIT_SET� NSEC_MISSING� CACHED_ERROR� NOT_READY�BLOCKED�CENSORED�FILTERED�
PROHIBITED�STALE_NXDOMAIN_ANSWER�NOT_AUTHORITATIVE�
NOT_SUPPORTED�NO_REACHABLE_AUTHORITY�
NETWORK_ERROR� INVALID_DATAr( r r r r r r� 7 s8 r� c s� e Zd ZdZddeeef dee f� fdd�
Zdefdd �Z dd
ee
dee fd d �Z e
d
eeef dddefdd��Z� ZS )� EDEOptionz!Extended DNS Error (EDE, RFC8914)N�coder� c s> t � �tj� t�|�| _|durt|t�st d��|| _
dS )z�*code*, a ``dns.edns.EDECode`` or ``str``, the info code of the
extended error.
*text*, a ``str`` or ``None``, specifying additional information about
the error.
Nztext must be string or None) rV r- r r' r� r+ r� r@ rR ru r� )r, r� r� r[ r r r- Z s
zEDEOption.__init__r/ c C s* d| j � �}| jd ur|d| j� �7 }|S )NzEDE z: )r� r� )r, �outputr r r rN i s
zEDEOption.to_textr. c C s> t �d| j�}| jd ur|| j�d�7 }|r |�|� d S |S )Nz!H�utf8)r{ r| r� r� �encoder] r� r r r r4 o s
zEDEOption.to_wirer* r5 r6 c C sL t �|�� �}|�� }|r |d dkr|d d� }|�d�}nd }| ||�S )Nr: r r� )r� r+ �
get_uint16r^ �decode)r r* r5 r� r� �btextr r r r8 z s
zEDEOption.from_wire_parserrM )r r r rQ r r� rR r r- rN r rS r4 r( r r) r8 r_ r r r[ r r� W s $
���r� �_type_to_classr* r/ c C s t �| �}|du r t}|S )z�Return the class for the specified option type.
The GenericOption class is used if a more specific class is not
known.
N)r� �getrT )r* r r r r �get_option_class� s
r� r5 r6 c C s t �| �} t| �}|�| |�S )a Build an EDNS option object from wire format.
*otype*, an ``int``, is the option type.
*parser*, a ``dns.wire.Parser``, the parser, which should be
restricted to the option length.
Returns an instance of a subclass of ``dns.edns.Option``.
)r r+ r� r8 )r* r5 r r r r �option_from_wire_parser� s
r� r< �current�olenc C sF t j�||�}|�|��
t| |�W d � S 1 s w Y dS )ar Build an EDNS option object from wire format.
*otype*, an ``int``, is the option type.
*wire*, a ``bytes``, is the wire-format message.
*current*, an ``int``, is the offset in *wire* of the beginning
of the rdata.
*olen*, an ``int``, is the length of the wire-format option data
Returns an instance of a subclass of ``dns.edns.Option``.
N)rW r< �Parser� restrict_tor� )r* r< r� r� r5 r r r �option_from_wire� s $�r� �implementationc C s | t |< dS )z�Register the implementation of an option type.
*implementation*, a ``class``, is a subclass of ``dns.edns.Option``.
*otype*, an ``int``, is the option type.
N)r� )r� r* r r r �
register_type� s r� )*rQ rx rn r{ �typingr r r r �dns.enumrW �dns.inet� dns.rdata�dns.wire�enum�IntEnumr r) rT r` r� r� r! r' r� �__annotations__r� rR r� rS rw r� r� r r r r r" r# r$ r% r& r r r r �<module> sd S 5�
��
�
����
�