
    EqgB              
          U d Z ddlZddlZddlZddlZddlmZmZmZm	Z	 ddl
ZddlZddlZddlZ G d dej        j                  Z G d d          Z G d d	e          Z G d
 de          Z G d dej        j                  Z G d de          Z G d de          Z G d de          Z G d de          Zej        eej        eej        eej        eej        eiZeeef         e d<   dedefdZ!de	ee"f         dddefdZ#de	ee"f         de$de%de%def
d Z&d!ededdfd"Z'ej        Zej(        Z(ej)        Z)ej*        Z*ej        Zej+        Z+ej        Zej,        Z,ej-        Z-ej.        Z.ej        Zej        ZdS )#zEDNS Options    N)AnyDictOptionalUnionc                   T    e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZed             ZdS )
OptionType               	   
                  c                     dS Ni   clss    :/var/www/html/env/lib/python3.11/site-packages/dns/edns.py_maximumzOptionType._maximum:       u    N)__name__
__module____qualname__NSIDDAUDHUN3UECSEXPIRECOOKIE	KEEPALIVEPADDINGCHAINEDEREPORTCHANNELclassmethodr   r   r   r   r   r       si        D
C
C
C
CFFIGE
CM  [  r   r   c                       e Zd ZdZdeeef         fdZddee	         dee
         fdZdefdZeded	d
dd fd            Zd Zd Zd Zd Zd Zd Zd Zd ZdS )Optionz%Base class for all EDNS option types.otypec                 D    t                               |          | _        dS )z_Initialize an option.

        *otype*, a ``dns.edns.OptionType``, is the option type.
        N)r   maker0   )selfr0   s     r   __init__zOption.__init__B   s    
  __U++


r   Nfilereturnc                     t           )zUConvert an option to wire format.

        Returns a ``bytes`` or ``None``.

        NotImplementedErrorr3   r5   s     r   to_wirezOption.to_wireI   s
     "!r   c                     t           Nr8   r3   s    r   to_textzOption.to_textQ   s    !!r   parserdns.wire.Parserc                     t           )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``.
        r8   r   r0   r@   s      r   from_wire_parserzOption.from_wire_parserT   s
     "!r   c                 v    |                                  }|                                 }||k    rdS ||k    rdS dS )zCompare an EDNS option with another option of the same type.

        Returns < 0 if < *other*, 0 if == *other*, and > 0 if > *other*.
        r      )r;   )r3   otherwireowires       r   _cmpzOption._cmpa   s?    
 ||~~5==1%<<1rr   c                     t          |t                    sdS | j        |j        k    rdS |                     |          dk    S )NFr   
isinstancer/   r0   rK   r3   rH   s     r   __eq__zOption.__eq__n   sD    %(( 	5:$$5yy1$$r   c                     t          |t                    sdS | j        |j        k    rdS |                     |          dk    S )NTr   rM   rO   s     r   __ne__zOption.__ne__u   sD    %(( 	4:$$4yy1$$r   c                     t          |t                    r| j        |j        k    rt          S |                     |          dk     S Nr   rN   r/   r0   NotImplementedrK   rO   s     r   __lt__zOption.__lt__|   ?    %(( 	"DJ%+,E,E!!yy!##r   c                     t          |t                    r| j        |j        k    rt          S |                     |          dk    S rT   rU   rO   s     r   __le__zOption.__le__   ?    %(( 	"DJ%+,E,E!!yy1$$r   c                     t          |t                    r| j        |j        k    rt          S |                     |          dk    S rT   rU   rO   s     r   __ge__zOption.__ge__   r[   r   c                     t          |t                    r| j        |j        k    rt          S |                     |          dk    S rT   rU   rO   s     r   __gt__zOption.__gt__   rX   r   c                 *    |                                  S r=   )r?   r>   s    r   __str__zOption.__str__   s    ||~~r   r=   )r   r   r    __doc__r   r   strr4   r   r   bytesr;   r?   r-   rD   rK   rP   rR   rW   rZ   r]   r_   ra   r   r   r   r/   r/   ?   s6       //,eJO4 , , , ," "HSM "Xe_ " " " "" " " " " 
"Z 
"9J 
"x 
" 
" 
" [
"  % % %% % %$ $ $
% % %
% % %
$ $ $
    r   r/   c                        e Zd ZdZdeeef         deeef         f fdZdde	e
         de	e         fdZdefd	Zedeeef         d
ddefd            Z xZS )GenericOptionzwGeneric Option Class

    This class is used for EDNS option types for which we have no better
    implementation.
    r0   datac                     t                                          |           t          j        j                            |d          | _        d S )NT)superr4   dnsrdataRdata	_as_bytesrg   )r3   r0   rg   	__class__s      r   r4   zGenericOption.__init__   s:    IO--dD99			r   Nr5   r6   c                 L    |r|                     | j                   d S | j        S r=   )writerg   r:   s     r   r;   zGenericOption.to_wire   +     	JJty!!!49r   c                     d| j         z  S )Nz
Generic %d)r0   r>   s    r   r?   zGenericOption.to_text   s    dj((r   r@   rA   c                 >     | ||                                           S r=   get_remainingrC   s      r   rD   zGenericOption.from_wire_parser   s"     s5&..00111r   r=   )r   r   r    rb   r   r   rc   rd   r4   r   r   r;   r?   r-   r/   rD   __classcell__rn   s   @r   rf   rf      s         :eJO4 :E%*<M : : : : : : HSM Xe_    ) ) ) ) ) 2*c/*24E2	2 2 2 [2 2 2 2 2r   rf   c                        e Zd ZdZddedee         def fdZdefd	Ze	d
ede
fd            Zddee         dee         fdZedeeef         ddde
fd            Z xZS )	ECSOptionz!EDNS Client Subnet (ECS, RFC7871)Nr   addresssrclenscopelenc                 z   t                                          t          j                   t          j                            |          }|t          j        k    r|d| _	        |d}t          j
        j                            |          }t          j
        j                            |dd          }t          j
        j                            |dd          }n|t          j        k    r|d| _	        |d}t          j
        j                            |          }t          j
        j                            |dd          }t          j
        j                            |dd          }nt!          d	          |J || _        || _        || _        t          j                            ||          }t+          t-          j        |d
z                      }|d|         | _        |dz  }|dk    rQt3          j        dt7          | j        dd                   dd|z
  z  z            }| j        dd         |z   | _        dS dS )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      rF          zBad address family       @r   BrG      )ri   r4   r   r%   rj   inetaf_for_addresssocketAF_INET6familyrk   rl   _as_ipv6_address_as_intAF_INET_as_ipv4_address
ValueErrorrz   r{   r|   	inet_ptonintmathceiladdrdatastructpackord)
r3   rz   r{   r|   afr   nbytesnbitslastrn   s
            r   r4   zECSOption.__init__   s    	(((X$$W--  DK~io66w??GY_,,VQ<<Fy..xC@@HH6>!!DK~io66w??GY_,,VQ;;Fy..xB??HH1222!!! 8%%b'22TYv|,,-- !&)
A::;sCbcc(:$;$;tE	?R$STTD M#2#.5DMMM :r   r6   c                 6    d| j          d| j         d| j         S )NzECS /z scope/)rz   r{   r|   r>   s    r   r?   zECSOption.to_text   s&    HdlHHT[HHHHHr   textc                    d}|                                  }d}t          |          dk    r	|d         }nNt          |          dk    r(|d         |k    rt          d|  d          |d         }nt          d|  d          |                    d          }|dk    r|                     d          \  }}d	}n3|dk    r|                     d          \  }}}nt          d|  d          	 t	          |          }n$# t          $ r t          d
d| dz             w xY w	 t	          |          }	n$# t          $ r t          dd| dz             w xY w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%   NrF   r   r~   zcould not parse ECS from ""r   0zinvalid scope z": scope must be an integerzinvalid srclen z": srclen must be an integer)splitlenr   countr   ry   )
r   optional_prefixtokensecs_text	n_slashesrz   tsrclentscopescoper{   s
             r   	from_textzECSOption.from_text   s   0  v;;!ayHH[[AayO++ !Ed!E!E!EFFFayHHA$AAABBBNN3''	>>'~~c22GWFF!^^'/~~c':':$GWffA$AAABBB	YKKEE 	Y 	Y 	Y-0WF0W0W0WWXXX	Y	\\FF 	 	 	!$M$M$M$MM  	 &%000s   -C= =!D"D2 2!Er5   c                     t          j        d| j        | j        | j                  | j        z   }|r|                    |           d S |S )N!HBB)r   r   r   r{   r|   r   rp   r3   r5   values      r   r;   zECSOption.to_wire  sM    KT[$-HH4=X 	  	JJu4Lr   r0   r@   rA   c                    |                     d          \  }}}t          t          j        |dz                      }|                    |          }|dk    r+d|z
  }t
          j                            |d|z  z             }	n@|dk    r+d|z
  }t
          j                            |d|z  z             }	nt          d           | |	||          S )	Nr   r   rF          r~      zunsupported family)

get_structr   r   r   	get_bytesrj   ipv4	inet_ntoaipv6r   )
r   r0   r@   r   srcr   addrlenprefixpadaddrs
             r   rD   zECSOption.from_wire_parser&  s     $..v66Udic	**++!!'**Q;;g+C8%%fw}&<==DDq[[w,C8%%fw}&<==DD1222s4e$$$r   rT   r=   )r   r   r    rb   rc   r   r   r4   r?   staticmethodr/   r   r   rd   r;   r-   r   r   rD   rv   rw   s   @r   ry   ry      s!       ++-6 -6 -6Xc] -6S -6 -6 -6 -6 -6 -6^I I I I I 41 41 41 41 41 \41l HSM Xe_     %*c/*%4E%	% % % [% % % % %r   ry   c                       e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZed             ZdS )EDECoder   rF   r~   r	   r   r
   r   r   r   r   r   r   r   r      r   r      r                  r   c                     dS r   r   r   s    r   r   zEDECode._maximumT  r   r   N)r   r   r    OTHERUNSUPPORTED_DNSKEY_ALGORITHMUNSUPPORTED_DS_DIGEST_TYPESTALE_ANSWERFORGED_ANSWERDNSSEC_INDETERMINATEDNSSEC_BOGUSSIGNATURE_EXPIREDSIGNATURE_NOT_YET_VALIDDNSKEY_MISSINGRRSIGS_MISSINGNO_ZONE_KEY_BIT_SETNSEC_MISSINGCACHED_ERROR	NOT_READYBLOCKEDCENSOREDFILTERED
PROHIBITEDSTALE_NXDOMAIN_ANSWERNOT_AUTHORITATIVENOT_SUPPORTEDNO_REACHABLE_AUTHORITYNETWORK_ERRORINVALID_DATAr-   r   r   r   r   r   r   9  s        E#$ !"LMLNNLLIGHHJMML  [  r   r   c                        e Zd ZdZh dZddeeef         dee         f fdZ	defdZ
dd	ee         dee         fd
Zedeeef         dddefd            Z xZS )	EDEOptionz!Extended DNS Error (EDE, RFC8914)>   DSNSECDNSKEYDNSSECRRSIGsNXDOMAINNcoder   c                     t                                          t          j                   t                              |          | _        |$t          |t                    st          d          || _
        dS )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.
        Nztext must be string or None)ri   r4   r   r+   r   r2   r   rN   rc   r   r   )r3   r   r   rn   s      r   r4   zEDEOption.__init__^  sa     	(((LL&&	JtS$9$9:;;;			r   r6   c                     d j          } j         t          v r\t                               j                   }d                     fd|                    d          D                       }|d| dz  } j        |d j         z  }|S )NzEDE  c              3   V   K   | ]#}|j         v r|n|                                V  $d S r=   )_preserve_casetitle).0wordr3   s     r   	<genexpr>z$EDEOption.to_text.<locals>.<genexpr>q  sP          333     r   _z ()z: )r   r   r?   joinr   r   )r3   outputdescs   `  r   r?   zEDEOption.to_textm  s    #	##9??49--D88     JJsOO    D l4lll"F9 &49&&&Fr   r5   c                     t          j        d| j                  }| j        || j                            d          z  }|r|                    |           d S |S )Nz!Hutf8)r   r   r   r   encoderp   r   s      r   r;   zEDEOption.to_wirez  s[    D$),,9 TY%%f---E 	JJu4Lr   r0   r@   rA   c                     t                               |                                          }|                                }|r,|d         dk    r
|d d         }|                    d          }nd } | ||          S )NrG   r   r   )r   r2   
get_uint16ru   decode)r   r0   r@   r   r   btexts         r   rD   zEDEOption.from_wire_parser  s~     ||F--//00##%% 	Bx1}}CRCyKK''EEEs4r   r=   )r   r   r    rb   r   r   r   rc   r   r4   r?   r   rd   r;   r-   r   r/   rD   rv   rw   s   @r   r   r   Y  s        ++MMMN U7C<0           	 	HSM 	Xe_ 	 	 	 	  *c/* 4E 	      [         r   r   c                        e Zd Zdef fdZddedee         fdZdefdZ	e
deeef         d	ej        j        defd
            Z xZS )
NSIDOptionnsidc                 l    t                                          t          j                   || _        d S r=   )ri   r4   r   r!   r  )r3   r  rn   s     r   r4   zNSIDOption.__init__  s(    )))			r   Nr5   r6   c                 L    |r|                     | j                   d S | j        S r=   )rp   r  r:   s     r   r;   zNSIDOption.to_wire  rq   r   c                     t          d | j        D                       r| j                                        }n+t          j        | j                                                  }d| S )Nc              3   .   K   | ]}|d k    o|dk    V  dS )r   ~   Nr   )r   cs     r   r   z%NSIDOption.to_text.<locals>.<genexpr>  s/      ::1qDy&Q$Y::::::r   zNSID )allr  r   binasciihexlify)r3   r   s     r   r?   zNSIDOption.to_text  sb    ::	::::: 	9I$$&&EE$TY//6688Eur   r0   r@   c                 <     | |                                           S r=   rt   rC   s      r   rD   zNSIDOption.from_wire_parser  s      s6''))***r   r=   r   r   r    rd   r4   r   r   r;   rc   r?   r-   r   r   rj   rI   Parserr/   rD   rv   rw   s   @r   r  r    s        U       C 8E?         +*c/*+47HO+	+ + + [+ + + + +r   r  c                        e Zd Zdedef fdZddedee         fdZdefdZ	e
d	eeef         d
ej        j        defd            Z xZS )CookieOptionclientserverc                 f   t                                          t          j        j        j                   || _        || _        t          |          dk    rt          d          t          |          dk    r5t          |          dk     st          |          dk    rt          d          d S d S )Nr   zclient cookie must be 8 bytesr   r   z5server cookie must be empty or between 8 and 32 bytes)
ri   r4   rj   ednsr   r'   r  r  r   r   )r3   r  r  rn   s      r   r4   zCookieOption.__init__  s    ,3444v;;!<===v;;!VqCKK"4D4DTUUU 4D4Dr   Nr5   r6   c                     |rN|                     | j                   t          | j                  dk    r|                     | j                   d S | j        | j        z   S rT   )rp   r  r   r  r:   s     r   r;   zCookieOption.to_wire  s[     	-JJt{###4;!##

4;'''4;,,r   c                     t          j        | j                                                  }t	          | j                  dk    r,t          j        | j                                                  }nd}d| | S )Nr    zCOOKIE )r  r  r  r   r   r  )r3   r  r  s      r   r?   zCookieOption.to_text  sm    !$+..5577t{a%dk2299;;FFF)))))r   r0   r@   c                 d     | |                     d          |                                          S )Nr   )r   ru   rC   s      r   rD   zCookieOption.from_wire_parser  s0     s6##A&&(<(<(>(>???r   r=   r  rw   s   @r   r  r    s        Vu Ve V V V V V V- -C -8E? - - - -* * * * * @*c/*@47HO@	@ @ @ [@ @ @ @ @r   r  c                        e Zd Zdej        j        f fdZddedee	         fdZ
defdZedeeef         d	ej        j        defd
            Z xZS )ReportChannelOptionagent_domainc                 l    t                                          t          j                   || _        d S r=   )ri   r4   r   r,   r  )r3   r  rn   s     r   r4   zReportChannelOption.__init__  s,    1222(r   Nr5   r6   c                 6    | j                             |          S r=   )r  r;   r:   s     r   r;   zReportChannelOption.to_wire  s     ((...r   c                 :    d| j                                         z   S )NzREPORTCHANNEL )r  r?   r>   s    r   r?   zReportChannelOption.to_text  s    $"3";";"="===r   r0   r@   c                 <     | |                                           S r=   )get_namerC   s      r   rD   z$ReportChannelOption.from_wire_parser  s     s6??$$%%%r   r=   )r   r   r    rj   nameNamer4   r   r   rd   r;   rc   r?   r-   r   r   rI   r  r/   rD   rv   rw   s   @r   r  r    s        )SX] ) ) ) ) ) )/ /C /8E? / / / /> > > > > &*c/*&47HO&	& & & [& & & & &r   r  _type_to_classr0   r6   c                 L    t                               |           }|t          }|S )zReturn the class for the specified option type.

    The GenericOption class is used if a more specific class is not
    known.
    )r$  getrf   )r0   r   s     r   get_option_classr'    s&     

U
#
#C
{Jr   r@   rA   c                     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   r2   r'  rD   )r0   r@   r   s      r   option_from_wire_parserr)    s8     OOE""E
5
!
!Cv...r   rI   currentolenc                     t           j                            ||          }|                    |          5  t	          | |          cddd           S # 1 swxY w Y   dS )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)rj   rI   r  restrict_tor)  )r0   rI   r*  r+  r@   s        r   option_from_wirer.    s      X__T7++F			D	!	! 6 6&uf556 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s   AAAimplementationc                     | t           |<   dS )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/  r0   s     r   register_typer1  "  s     +N5r   )/rb   r  r   r   r   typingr   r   r   r   dns.enumrj   dns.inet	dns.rdatadns.wireenumIntEnumr   r/   rf   ry   r   r   r  r  r  r%   r+   r!   r'   r,   r$  __annotations__r'  rc   r)  rd   r   r.  r1  r"   r#   r$   r&   r(   r)   r*   r   r   r   <module>r:     s  $        - - - - - - - - - - - -           !   >R R R R R R R Rj2 2 2 2 2F 2 2 28F% F% F% F% F% F% F% F%R    ch   @:  :  :  :  :  :  :  : z+ + + + + + + +6@ @ @ @ @6 @ @ @D& & & & && & & &( NINIOZ|1)Z_%   
J 
3 
 
 
 
/S!/+<// / / /"6S!6).69<6DG66 6 6 6*+# +j +T + + + + nnnn				 	

n(r   