a sd_&@sdZddlZddlZddlZddlZddlZGdddejjZ e e j GdddZ Gddde ZGd d d e Ze jeiZd d Zd dZddZdS)z EDNS OptionsNc@s@eZdZdZdZdZdZdZdZdZ dZ d Z d Z e d d Zd S) OptionType cCsdS)Ni)clsr r */usr/lib/python3/dist-packages/dns/edns.py_maximum1szOptionType._maximumN)__name__ __module__ __qualname__ZNSIDZDAUZDHUZN3UECSZEXPIREZCOOKIEZ KEEPALIVEZPADDINGZCHAIN classmethodrr r r rrsrc@sneZdZdZddZdddZeddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZdS)Optionz%Base class for all EDNS option types.cCs ||_dS)zPInitialize an option. *otype*, an ``int``, is the option type. Notype)selfrr r r__init__;szOption.__init__NcCstdS)zUConvert an option to wire format. Returns a ``bytes`` or ``None``. NNotImplementedErrorrfiler r rto_wireBszOption.to_wirecCstdS)zBuild an EDNS option object from wire format. *otype*, an ``int``, is the option type. *parser*, a ``dns.wire.Parser``, the parser, which should be restructed to the option length. Returns a ``dns.edns.Option``. Nrrrparserr r rfrom_wire_parserJs zOption.from_wire_parsercCs,|}|}||krdS||kr(dSdS)zCompare an EDNS option with another option of the same type. Returns < 0 if < *other*, 0 if == *other*, and > 0 if > *other*. r)r)rotherwireZowirer r r_cmpWsz Option._cmpcCs,t|tsdS|j|jkrdS||dkS)NFr isinstancerrr'rr%r r r__eq__ds   z Option.__eq__cCs,t|tsdS|j|jkrdS||dkS)NTrr(r*r r r__ne__ks   z Option.__ne__cCs(t|tr|j|jkrtS||dkSNrr)rrNotImplementedr'r*r r r__lt__rs   z Option.__lt__cCs(t|tr|j|jkrtS||dkSr-r.r*r r r__le__xs   z Option.__le__cCs(t|tr|j|jkrtS||dkSr-r.r*r r r__ge__~s   z Option.__ge__cCs(t|tr|j|jkrtS||dkSr-r.r*r r r__gt__s   z Option.__gt__cCs|SN)to_textrr r r__str__szOption.__str__)N)rrr__doc__rrrr"r'r+r,r0r1r2r3r7r r r rr7s   rcs>eZdZdZfddZd ddZddZed d ZZ S) GenericOptionzwGeneric Option Class This class is used for EDNS option types for which we have no better implementation. cst|||_dSr4)superrdata)rrr; __class__r rrs zGenericOption.__init__NcCs|r||jn|jSdSr4)writer;rr r rrszGenericOption.to_wirecCs d|jS)Nz Generic %drr6r r rr5szGenericOption.to_textcCs|||Sr4)Z get_remainingr r r rr"szGenericOption.from_wire_parser)N) rrrr8rrr5rr" __classcell__r r r<rr9s   r9csLeZdZdZdfdd ZddZedd Zdd d Ze d d Z Z S) ECSOptionz!EDNS Client Subnet (ECS, RFC7871)Nrc sttjtj|}|tjkr8d|_ |dur^d}n&|tj krVd|_ |dur^d}nt d||_ ||_ ||_tj||}tt|d}|d||_|d}|d krtd t|jd dd d|>@}|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. N8r#z Bad ip family @rrBr$)r:rrrdnsZinetZaf_for_addresssocketZAF_INET6familyZAF_INET ValueErroraddresssrclenscopelenZ inet_ptonintmathceiladdrdatastructpackord) rrKrLrMZafrQnbytesZnbitsZlastr<r rrs.    zECSOption.__init__cCsd|j|j|jS)NzECS {}/{} scope/{})formatrKrLrMr6r r rr5s zECSOption.to_textcCs$d}|}d}t|dkr&|d}n>t|dkrV|d|krLtd||d}ntd||d}|dkr|d\}}d}n(|dkr|d\}}}ntd|z t|}Wn$tytdd |Yn0z t|}Wn&tytd d |Yn0t|||S) aConvert 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') rNr#rrAzcould not parse ECS from "{}"/zinvalid scope z"{}": scope must be an integerzinvalid srclen z"{}": srclen must be an integer)splitlenrJrVcountrNr@)textZoptional_prefixtokensZecs_textZ n_slashesrKrLscoper r r from_texts<           zECSOption.from_textcCs4td|j|j|j|j}|r,||n|SdS)N!HBB)rRrSrIrLrMrQr>)rrvaluer r rrs  zECSOption.to_wirec Cs|d\}}}tt|d}||}|dkrRd|}tj|d|} n.|dkrxd|}tj|d|} nt d|| ||S) Nr_rDr#rAzunsupported family) Z get_structrNrOrPZ get_bytesrGZipv4Z inet_ntoaZipv6rJ) rrr!rIsrcr]ZaddrlenprefixZpadZaddrr r rr"s zECSOption.from_wire_parser)Nr)N) rrrr8rr5 staticmethodr^rrr"r?r r r<rr@s) 6 r@cCst|}|durt}|S)zReturn the class for the specified option type. The GenericOption class is used if a more specific class is not known. N)_type_to_classgetr9)rrr r rget_option_class,s ricCst|}t|}|||S)aBuild 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``. )rirZmaker")rr!rr r roption_from_wire_parser9s  rjcCsFtj||}||t||WdS1s80YdS)arBuild 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)rGr&ZParserZ restrict_torj)rr&ZcurrentZolenr!r r roption_from_wireHs rk)r8rOrHrRZdns.enumrGZdns.inetenumIntEnumrglobalsupdate __members__rr9r@rrgrirjrkr r r rs W