3 [_]@sdZddlmZddlZddlZddlZdZdZdZ dZ dZ d Z d Z d Zd Zd ZGdddeZGdddeZGdddeZe eiZddZddZdS)z EDNS Options)absolute_importN c@sdeZdZdZddZddZeddZdd Zd d Z d d Z ddZ ddZ ddZ ddZdS)Optionz%Base class for all EDNS option types.cCs ||_dS)zPInitialize an option. *otype*, an ``int``, is the option type. N)otype)selfrr-/tmp/pip-build-yz_zf6az/dnspython/dns/edns.py__init__4szOption.__init__cCstdS)z*Convert an option to wire format. N)NotImplementedError)rfilerrrto_wire;szOption.to_wirecCstdS)atBuild an EDNS option object from wire format. *otype*, an ``int``, is the option type. *wire*, a ``binary``, 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 a ``dns.edns.Option``. N)r)clsrwirecurrentolenrrr from_wire@szOption.from_wirecCstdS)zCompare an EDNS option with another option of the same type. Returns < 0 if < *other*, 0 if == *other*, and > 0 if > *other*. N)r)rotherrrr_cmpRsz Option._cmpcCs,t|tsdS|j|jkrdS|j|dkS)NFr) isinstancer rr)rrrrr__eq__Ys   z Option.__eq__cCs,t|tsdS|j|jkrdS|j|dkS)NFr)rr rr)rrrrr__ne__`s   z Option.__ne__cCs*t|t s|j|jkrtS|j|dkS)Nr)rr rNotImplementedr)rrrrr__lt__gs  z Option.__lt__cCs*t|t s|j|jkrtS|j|dkS)Nr)rr rr r)rrrrr__le__ms  z Option.__le__cCs*t|t s|j|jkrtS|j|dkS)Nr)rr rr r)rrrrr__ge__ss  z Option.__ge__cCs*t|t s|j|jkrtS|j|dkS)Nr)rr rr r)rrrrr__gt__ys  z Option.__gt__N)__name__ __module__ __qualname____doc__rr classmethodrrrrr!r"r#r$rrrrr 0s r csDeZdZdZfddZddZddZedd Zd d Z Z S) GenericOptionzwGeneric Option Class This class is used for EDNS option types for which we have no better implementation. cstt|j|||_dS)N)superr*rdata)rrr,) __class__rrrszGenericOption.__init__cCs|j|jdS)N)writer,)rrrrrrszGenericOption.to_wirecCs d|jS)Nz Generic %d)r)rrrrto_textszGenericOption.to_textcCs||||||S)Nr)rrrrrrrrrszGenericOption.from_wirecCs$|j|jkrdS|j|jkr dSdS)Nr)r,)rrrrrrs   zGenericOption._cmp) r%r&r'r(rrr/r)rr __classcell__rr)r-rr*s   r*csFeZdZdZdfdd ZddZdd Zed d Zd d Z Z S) ECSOptionz!EDNS Client Subnet (ECS, RFC7871)Nrc stt|jttjj|}|tjjkr@}|jdd ||_dS)a*address*, a ``text``, 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. N8r0z Bad ip familyg @rrBr1r1)r+r3rECSdnsinetZaf_for_addressAF_INET6familyAF_INET ValueErroraddresssrclenscopelen inet_ptonintmathceiladdrdatastructpackord) rr@rArBafrGnbytesnbitslast)r-rrrs*    "zECSOption.__init__cCsdj|j|j|jS)NzECS {}/{} scope/{})formatr@rArB)rrrrr/s zECSOption.to_textcCs<|jtjd|j|jtjd|j|j|j|jdS)Nz!Hz!BB)r.rHrIr=rArBrG)rrrrrrszECSOption.to_wirec Cstjd|||d\}}}|d7}ttj|d}|dkrRtjj} d|} n"|dkrltjj} d|} nt dtjj | ||||d| } || ||S) Nz!HBBg @r0r4zunsupported family) rHunpackrDrErFr:r;r>r<r? inet_ntop) rrrcurrr=srcZscopeZaddrlenrKpadaddrrrrrs  "zECSOption.from_wirecCs$|j|jkrdS|j|jkr dSdS)Nrr0r1)rG)rrrrrrs   zECSOption._cmp)Nr) r%r&r'r(rr/rr)rrr2rr)r-rr3s ( r3cCstj|}|dkrt}|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_classgetr*)rrrrrget_option_classs r[cCst|}|j||||S)asBuild an EDNS option object from wire format. *otype*, an ``int``, is the option type. *wire*, a ``binary``, 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``. )r[r)rrrrrrrroption_from_wiresr\)r( __future__rrErHZdns.inetr:ZNSIDZDAUZDHUZN3Ur9ZEXPIREZCOOKIEZ KEEPALIVEZPADDINGZCHAINobjectr r*r3rYr[r\rrrrs( PO