JezK
Edit File: edns.cpython-39.pyc
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 s^ 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 ��|dus�J �|| _|| _|| _tj�||�}tt�|d ��}|d|� | _|d }|dk�rZt�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, � s8 &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 = � � �