a
�M�e�; � @ s� U d Z ddlZddlZddlZddlZddlmZmZmZm Z ddl
Z ddl Z ddl
Z ddlZ 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�ZG dd� de�ZejeejeejeiZeeef e d< eed�dd�Z e ee f ded�dd�Z e ee f e e!e!ed�dd �Z"eedd �d d �Z#ejZej$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 s dS �Ni�� � ��clsr r �,/usr/lib/python3.9/site-packages/dns/edns.py�_maximum8 s 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 ZdZeeef d�dd�Zd ee ee
d�dd�Z ed �d
d �Z e
ed d d
�dd��Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � 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 s t �dS )zUConvert an option to wire format.
Returns a ``bytes`` or ``None``.
N��NotImplementedError�r+ r. r r r �to_wireG s zOption.to_wire�r/ c C s t �d S �Nr0 �r+ r r r �to_textO s zOption.to_text�dns.wire.Parser�r) �parserr/ c C s t �dS )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``.
Nr0 �r r) r: r r r �from_wire_parserR s zOption.from_wire_parserc C s, | � � }|� � }||kr dS ||kr(dS 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 � ���)r3 )r+ �other�wireZowirer r r �_cmp_ s z Option._cmpc C s, t |t�sdS | j|jkr dS | �|�dkS )NFr ��
isinstancer' r) rA �r+ r? r r r �__eq__l s
z
Option.__eq__c C s, t |t�sdS | j|jkr dS | �|�dkS )NTr rB rD r r r �__ne__s s
z
Option.__ne__c C s( t |t�r| j|jkrtS | �|�dk S �Nr �rC r' r) �NotImplementedrA rD r r r �__lt__z s z
Option.__lt__c C s( t |t�r| j|jkrtS | �|�dkS rG rH rD r r r �__le__ s z
Option.__le__c C s( t |t�r| j|jkrtS | �|�dkS rG rH rD r r r �__ge__� s z
Option.__ge__c C s( t |t�r| j|jkrtS | �|�dkS rG rH rD r r r �__gt__� s z
Option.__gt__c C s | � � S r5 )r7 r6 r r r �__str__� s zOption.__str__)N)r r r �__doc__r r �strr, r r �bytesr3 r7 r& r< rA rE rF rJ rK rL rM rN r r r r r' = s
r' c s~ e Zd ZdZeeef eeef d�� fdd� Zde e
e e d�dd�Z ed �d
d �Z e
eeef d ed
�dd��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�RdataZ _as_bytesrS )r+ r) rS �� __class__r r r, � s zGenericOption.__init__Nr- c C s |r|� | j� d S | jS d S r5 )�writerS r2 r r r r3 � s zGenericOption.to_wirer4 c C s
d| j S )Nz
Generic %dr( r6 r r r r7 � s zGenericOption.to_textr8 r9 c C s | ||� � �S r5 ��
get_remainingr; r r r r< � s z GenericOption.from_wire_parser)N)r r r rO r r rP rQ r, r r r3 r7 r& r' r< �
__classcell__r r rX r rR � s $ �rR c s� e Zd ZdZdeee ed�� fdd�
Zed�dd �Ze ee
d
�d d ��Z dee ee
d
�dd�Zeeeef de
d�dd��Z� ZS )� ECSOptionz!EDNS Client Subnet (ECS, RFC7871)Nr )�address�srclen�scopelenc sR t � �tj� tj�|�}|tjkrjd| _ |du r6d}tj
j � |�}tj
j �
|dd�}tj
j �
|dd�}nX|tjkr�d| _ |du r�d}tj
j �|�}tj
j �
|dd�}tj
j �
|dd�}ntd ��|| _|| _|| _tj�||�}tt�|d
��}|d|� | _|d }|dk�rNt�d t| jd
d� �dd | > @ �}| jdd
� | | _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 � r= � � zBad address family� @r �Br> � ) rT r, r r rU ZinetZaf_for_address�socketZAF_INET6�familyrV rW Z_as_ipv6_address�_as_intZAF_INETZ_as_ipv4_address�
ValueErrorr_ r` ra Z inet_pton�int�math�ceil�addrdata�struct�pack�ord) r+ r_ r` ra �afrq �nbytesZnbitsZlastrX r r r, � s6
&zECSOption.__init__r4 c C s d� | j| j| j�S )NzECS {}/{} scope/{})�formatr_ r` ra r6 r r r r7 � s zECSOption.to_text)�textr/ c
C s$ d}| � � }d}t|�dkr&|d }n>t|�dkrV|d |krLtd�| ���|d }ntd�| ���|�d�}|dkr�|� d�\}}d}n(|dkr�|� d�\}}}ntd�| ���z t|�}W n$ ty� td d
�|� ��Y n0 z t|�} W n& t�y td d �|� ��Y n0 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 Nr= r rb z could not parse ECS from "{}"�/�0zinvalid scope z "{}": scope must be an integerzinvalid srclen z "{}": srclen must be an integer)�split�lenrm rw �countrn r^ )
rx Zoptional_prefix�tokensZecs_textZ n_slashesr_ ZtsrclenZtscope�scoper` r r r � from_text� s<
�
�
zECSOption.from_textr- c C s6 t �d| j| j| j�| j }|r.|�|� d S |S d S )N�!HBB)rr rs rk r` ra rq rZ �r+ r. �valuer r r r3 s �
zECSOption.to_wirer8 r9 c
C s� |� d�\}}}tt�|d ��}|�|�}|dkrRd| }tj�|d| �} n.|dkrxd| }tj�|d| �} nt d��| | ||�S ) Nr� rg r= � � rb � zunsupported family)
Z
get_structrn ro rp Z get_bytesrU Zipv4Z inet_ntoaZipv6rm )
r r) r: rk �srcr Zaddrlen�prefixZpadZaddrr r r r< & s
zECSOption.from_wire_parser)Nr )N)r r r rO rP r rn r, r7 � staticmethodr' r� r rQ r3 r& r r r< r] r r rX 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 r= rb r r� r r r
r r r
r r r � r r� � � � � � � � re c C s dS r r r r r r r T s zEDECode._maximumN) r r r ZOTHERZ UNSUPPORTED_DNSKEY_ALGORITHMZUNSUPPORTED_DS_DIGEST_TYPEZ STALE_ANSWERZ
FORGED_ANSWERZDNSSEC_INDETERMINATEZ DNSSEC_BOGUSZSIGNATURE_EXPIREDZSIGNATURE_NOT_YET_VALIDZDNSKEY_MISSINGZRRSIGS_MISSINGZNO_ZONE_KEY_BIT_SETZ NSEC_MISSINGZ CACHED_ERRORZ NOT_READYZBLOCKEDZCENSOREDZFILTEREDZ
PROHIBITEDZSTALE_NXDOMAIN_ANSWERZNOT_AUTHORITATIVEZ
NOT_SUPPORTEDZNO_REACHABLE_AUTHORITYZ
NETWORK_ERRORZ INVALID_DATAr& r r r r r r� 9 s6 r� c s� e Zd ZdZh d�Zdeeef ee d�� fdd�
Z ed�dd �Z
dee ee d
�d d �Z
eeeef d
ed�dd��Z� ZS )� EDEOptionz!Extended DNS Error (EDE, RFC8914)> ZDSZDNSKEYZNSECZDNSSECZNXDOMAINZRRSIGsN)�coderx c s> t � �tj� t�|�| _|dur4t|t�s4t 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) rT r, r r% r� r* r� rC rP rm rx )r+ r� rx rX r r r, ^ s
zEDEOption.__init__r4 c sn d� j � �}� j tv rPt�� j �}d�� fdd�|�d�D ��}|d|� d�7 }� jd urj|d� j� �7 }|S ) NzEDE � c 3 s$ | ] }|� j v r|n|�� V qd S r5 )�_preserve_case�title)�.0Zwordr6 r r � <genexpr>q s �z$EDEOption.to_text.<locals>.<genexpr>�_z (�)z: )r� r� r7 �joinr{ rx )r+ �outputZdescr r6 r r7 m s
�
zEDEOption.to_textr- c C sB t �d| j�}| jd ur(|| j�d�7 }|r:|�|� d S |S d S )Nz!H�utf8)rr rs r� rx �encoderZ r� r r r r3 z s
zEDEOption.to_wirer8 r9 c C sL t �|�� �}|�� }|r>|d dkr2|d d� }|�d�}nd }| ||�S )Nr> r r� )r� r* Z
get_uint16r\ �decode)r r) r: r� rx Zbtextr r r r< � s zEDEOption.from_wire_parser)N)N)r r r rO r� r r� rP r r, r7 r rQ r3 r& r r' r< r] r r rX r r� Y s "
�r� c sh e Zd Zed�� fdd� Zdeee d�dd�Zed�d d
�Z e
e e ef e
jjed �d d
��Z� ZS )�
NSIDOption)�nsidc s t � �tj� || _d S r5 )rT r, r r r� )r+ r� rX r r r, � s zNSIDOption.__init__Nr- c C s |r|� | j� d S | jS d S r5 )rZ r� r2 r r r r3 � s zNSIDOption.to_wirer4 c C s: t dd� | jD ��r | j�� }nt�| j��� }d|� �S )Nc s s | ]}|d ko|dkV qdS )rf �~ Nr )r� �cr r r r� � � z%NSIDOption.to_text.<locals>.<genexpr>zNSID )�allr� r� �binasciiZhexlify)r+ r� r r r r7 � s zNSIDOption.to_textr9 c C s | |� � �S r5 r[ r; r r r r< � s zNSIDOption.from_wire_parser)N)r r r rQ r, r r r3 rP r7 r& r r rU r@ �Parserr' r< r] r r rX r r� � s �r� �_type_to_class)r) r/ c C s t �| �}|du rt}|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� �getrR )r) r r r r �get_option_class� s
r� r8 r9 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� r< )r) r: r r r r �option_from_wire_parser� s
r� )r) r@ �current�olenr/ c C sF t j�||�}|�|�� t| |�W d � S 1 s80 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)rU r@ r� Z restrict_tor� )r) r@ r� r� r: r r r �option_from_wire� s r� )�implementationr) r/ c 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� ),rO r� ro rj rr �typingr r r r Zdns.enumrU Zdns.inetZ dns.rdataZdns.wire�enum�IntEnumr r' rR r^ r� r� r� r r% r r� �__annotations__r� rP r� rQ rn r� r� r r r r r! r" r# r$ r r r r �<module> sP U = � � �