dZddlmZddlZddlZddlZdZdZdZ dZ dZ d Z d Z d Zd Zd ZGddeZGddeZGddeZe eiZdZdZy)z EDNS Options)absolute_importN cVeZdZdZdZdZedZdZdZ dZ dZ d Z d Z d Zy ) Optionz%Base class for all EDNS option types.c||_y)zPInitialize an option. *otype*, an ``int``, is the option type. Notype)selfrs lib/third_party/dns/edns.py__init__zOption.__init__4s  ct)z*Convert an option to wire format. NotImplementedErrorrfiles rto_wirezOption.to_wire;s "!rct)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``. rclsrwirecurrentolens r from_wirezOption.from_wire@s "!rct)zCompare an EDNS option with another option of the same type. Returns < 0 if < *other*, 0 if == *other*, and > 0 if > *other*. rrothers r_cmpz Option._cmpRs "!rct|tsy|j|jk7ry|j|dk(SNFr isinstancerrr'r%s r__eq__z Option.__eq__Y6%( :: $yy1$$rct|tsy|j|jk7ry|j|dk7Sr)r*r%s r__ne__z Option.__ne__`r-rct|tr|j|jk7rtS|j |dkSNrr+rrNotImplementedr'r%s r__lt__z Option.__lt__g6%( ekk)! !yy!##rct|tr|j|jk7rtS|j |dkSr1r2r%s r__le__z Option.__le__m6%( ekk)! !yy1$$rct|tr|j|jk7rtS|j |dk\Sr1r2r%s r__ge__z Option.__ge__sr8rct|tr|j|jk7rtS|j |dkDSr1r2r%s r__gt__z Option.__gt__yr5rN)__name__ __module__ __qualname____doc__rr classmethodr#r'r,r/r4r7r:r<rrrr0sF/" """"%%$ % % $rrcDeZdZdZfdZdZdZedZdZ xZ S) GenericOptionzwGeneric Option Class This class is used for EDNS option types for which we have no better implementation. c:tt| |||_yN)superrDrdata)rrrH __class__s rrzGenericOption.__init__s mT+E2 rc:|j|jyrF)writerHrs rrzGenericOption.to_wires 499rc d|jzS)Nz Generic %drrs rto_textzGenericOption.to_textsdjj((rc ||||||zSrFrBrs rr#zGenericOption.from_wires5$w$788rcl|j|jk(ry|j|jkDryyNr)rHr%s rr'zGenericOption._cmps+ 99 " 99uzz !r) r=r>r?r@rrrNrAr#r' __classcell__rIs@rrDrDs0 )99rrDcFeZdZdZdfd ZdZdZedZdZ xZ S) ECSOptionz!EDNS Client Subnet (ECS, RFC7871)ctt| ttj j |}|tj jk(r d|_|7d}n4|tj jk(r d|_|d}n td||_ ||_ ||_ tj j||}tt!j"|dz }|d||_|dz}|d k7rJt'j(d t+|j$d dd |zz}|j$dd |z|_yy) 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. N8rRz Bad ip family @rrBrS)rGrWrECSdnsinetaf_for_addressAF_INET6familyAF_INET ValueErroraddresssrclenscopelen inet_ptonintmathceiladdrdatastructpackord) rrgrhriafrnnbytesnbitslastrIs rrzECSOption.__init__s  i', XX $ $W - "" "DK~ 388## #DK~_- -    88%%b'2TYYvcz*+!&)   A:;;sC bc(:$;tu}$MND MM#2.5DM rcddj|j|j|jS)NzECS {}/{} scope/{})formatrgrhrirMs rrNzECSOption.to_texts(#**4<<+/==: :rc |jtjd|j|jtjd|j|j |j|j y)Nz!Hz!BB)rKrorprdrhrirnrs rrzECSOption.to_wiresN 6;;tT[[12 6;;udkk4==AB 4==!rctjd|||dz\}}}|dz }ttj|dz }|dk(r t j j} d|z } n0|dk(r t j j} d|z } n tdt j j| ||||zd| zz} || ||S) Nz!HBBr\rRrYzunsupported family) rounpackrkrlrmr`rarercrf inet_ntop) rrr curr"rdsrcscopeaddrlenrrpadaddrs rr#zECSOption.from_wires#]]64CE?CU qdiiC() Q;!!Bg+C q[""Bw,C12 2xx!!"d3s7{&;gm&KL4e$$rcl|j|jk(ry|j|jkDryyrQ)rnr%s rr'zECSOption._cmps+ ==ENN * ==5>> )rr1) r=r>r?r@rrNrrAr#r'rTrUs@rrWrWs/+&6P:" %%$rrWc@tj|}|t}|S)zReturn the class for the specified option type. The GenericOption class is used if a more specific class is not known. )_type_to_classgetrD)rrs rget_option_classrs#   U #C { Jrc@t|}|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``. )rr#)rr r!r"rs roption_from_wirers# 5 !C ==gt 44r)r@ __future__rrlrodns.inetr`NSIDDAUDHUN3Ur_EXPIRECOOKIE KEEPALIVEPADDINGCHAINobjectrrDrWrrrrBrrrs$&       M$VM$`F