a
Me; @ s U d Z ddlZddlZddlZddlZddlmZmZmZm Z ddl
Z ddl Z ddl
Z ddlZ G dd de jjZG dd dZG dd d eZG d
d d eZG d d
d
e jjZG dd deZG dd deZejeejeejeiZeeef e d< eedddZ e ee f dedddZ e ee f e e!e!eddd 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)AnyDictOptionalUnionc @ 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__NSIDDAUDHUN3UECSEXPIRECOOKIE KEEPALIVEPADDINGCHAINEDE classmethodr r r r r r s r c @ s e Zd ZdZeeef dddZd ee ee
dddZ ed d
d Z e
ed d d
ddZdd 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__Nfilereturnc C s t dS )zUConvert an option to wire format.
Returns a ``bytes`` or ``None``.
NNotImplementedErrorr+ r. r r r to_wireG s zOption.to_wirer/ c C s t d S Nr0 r+ r r r to_textO s zOption.to_textdns.wire.Parserr) 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 )zCompare an EDNS option with another option of the same type.
Returns < 0 if < *other*, 0 if == *other*, and > 0 if > *other*.
r )r3 )r+ otherwireZowirer r r _cmp_ s z Option._cmpc C s, t |tsdS | j|jkr dS | |dkS )NFr
isinstancer' r) rA r+ r? r r r __eq__l s
z
Option.__eq__c C s, t |tsdS | j|jkr dS | |dkS )NTr rB rD r r r __ne__s s
z
Option.__ne__c C s( t |tr| j|jkrtS | |dk S Nr rC r' r) NotImplementedrA rD r r r __lt__z s z
Option.__lt__c C s( t |tr| j|jkrtS | |dkS rG rH rD r r r __le__ s z
Option.__le__c C s( t |tr| j|jkrtS | |dkS rG rH rD r r r __ge__ s z
Option.__ge__c C s( t |tr| 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 dddZ ed d
d Z e
eeef d ed
ddZ 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, dnsrdataRdataZ _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
Zeddd Ze ee
d
d d Z dee ee
d
ddZeeeef de
dddZ ZS ) ECSOptionz!EDNS Client Subnet (ECS, RFC7871)Nr )addresssrclenscopelenc sR t tj tj|}|tjkrjd| _ |du r6d}tj
j |}tj
j
|dd}tj
j
|dd}nX|tjkrd| _ |du rd}tj
j |}tj
j
|dd}tj
j
|dd}ntd || _|| _|| _tj||}tt|d
}|d| | _|d }|dkrNtd 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.
N8 r r= zBad address family @r Br> ) rT r, r r rU ZinetZaf_for_addresssocketZAF_INET6familyrV rW Z_as_ipv6_address_as_intZAF_INETZ_as_ipv4_address
ValueErrorr_ r` ra Z inet_ptonintmathceiladdrdatastructpackord) r+ r_ r` ra afrq nbytesZnbitsZlastrX r r r, s6
&zECSOption.__init__r4 c C s d | j| j| jS )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& ty 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)splitlenrm rw countrn r^ )
rx Zoptional_prefixtokensZecs_textZ n_slashesr_ ZtsrclenZtscopescoper` 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 ZaddrlenprefixZpadZaddrr 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 dZdeeef ee d fdd
Z eddd Z
dee ee d
d d Z
eeeef d
edddZ ZS ) EDEOptionz!Extended DNS Error (EDE, RFC8914)> ZDSZDNSKEYZNSECZDNSSECZNXDOMAINZRRSIGsN)coderx c s> t tj t|| _|dur4t|ts4t 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|dD }|d| d7 } jd urj|d j 7 }|S ) NzEDE c 3 s$ | ] }| j v r|n| V qd S r5 )_preserve_casetitle).0Zwordr6 r r q s z$EDEOption.to_text.._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(|| jd7 }|r:|| d S |S d S )Nz!Hutf8)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 dddZedd 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..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 )zReturn 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@ currentolenr/ 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 )zRegister 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.wireenumIntEnumr 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 sP U =