o 'h.y@sUdZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZdZdZGdddejjZed fd d Zed fd d Z ed fddZ!dZ"ddZ#ddZ$ej%j&Z'ej%j%GdddZ(ej%j%Gddde(Z)iZ*e e ej+j,ej-j.fe fe/d<dZ0da1d9ddZ2d9ddZ3    d:deej+j,e4fd eej-j.e4fd!eej5j6e4fd"e ej7j8d#e9d$e ej7j8d%e ej7j:d&e(fd'd(Z; d;deej+j,e4fd eej-j.e4fd)ej d;deej+j,e4fd eej-j.e4fd,e?d-e@d.e@d"e ej7j8d&e(fd/d0ZAGd1d2d2ejjZBd3ej+jCfd4e d e@d5e4d6e9dej+j,d&df d7d8ZDdS)Csz_wordbreak..r)decodejoinrangelen)rr separatorrrr _wordbreak;srcKtt|||S)zConvert a binary string into its hex encoding, broken up into chunks of chunksize characters separated by a separator. )rbinasciihexlifyrrrkwrrr_hexifyJr"cKr)zConvert a binary string into its base64 encoding, broken up into chunks of chunksize characters separated by a separator. )rbase64 b64encoder rrr _base64ifyRr#r&s"\cCszt|tr |}t|tst|}d}|D]$}|tvr%|dt|7}q|dkr4|dkr4|t|7}q|d|7}q|S)z7Escape the characters in a quoted string which need it.\rz\%03d) isinstancestrencode bytearray __escapedchr)qstringtextcrrr _escapify_s  r3cCsFtt|dddD]}||dkr|d|dSq |ddS)zDetermine the index of greatest byte that isn't all zeros, and return the bitmap that contains all the bytes less than that index. r)rr)whatrrrr_truncate_bitmaprs   r7c@seZdZdZgdZddZddZddZd d Zd e j j fd d Z d e fddZ  dfdee jjdedeeefd efddZ   dgdeedee jjdee jjded df ddZ    dhdeedee jjdee jjded eef ddZ didee jjd dfd d!Zdidee jjd efd"d#Zd$d%Zd&d'Zd(d)Zd*d+Z d,d-Z!d.d/Z"d0d1Z#d2d3Z$d4d5Z%d6d7Z&e'   djd8e j(j)d9e j j d:e j*j+dee jjded;ee jjd dfde j-j.dee jjd df d?d@Z/dAed dfdBdCZ0e'dDdEZ1e'dFdGZ2e'   dkdHedIedJee dKed ef dLdMZ3e'dNdOZ4e'dPdQZ5e'dRdSZ6e'dTdUZ7e'dVdWZ8e'dldXdYZ9e'dZd[Z:e'd\d]Z;e'd^d_Ze'dddeZ?dS)mRdataz#Base class for all DNS rdata types.)rdclassrdtype rdcommentcCs"|||_|||_d|_dS)zInitialize an rdata. *rdclass*, an ``int`` is the rdataclass of the Rdata. *rdtype*, an ``int`` is the rdatatype of the Rdata. N)_as_rdataclassr9 _as_rdatatyper:r;)selfr9r:rrr__init__s   zRdata.__init__cCstjdd|jjDS)Ncss|] }t|dgVqdS) __slots__N)getattr)rclsrrr s  z'Rdata._get_all_slots..) itertoolschain from_iterable __class____mro__r>rrr_get_all_slotss zRdata._get_all_slotscCs$i}|D] }t||||<q|SN)rJrA)r>stateslotrrr __getstate__s zRdata.__getstate__cCs@|D] \}}t|||qt|dst|dddSdS)Nr;)itemsobject __setattr__hasattr)r>rLrMvalrrr __setstate__s  zRdata.__setstate__returncCstjjS)aReturn the type a Rdata covers. DNS SIG/RRSIG rdatas apply to a specific type; this type is returned by the covers() function. If the rdata type is not SIG or RRSIG, dns.rdatatype.NONE is returned. This is useful when creating rdatasets, allowing the rdataset to contain only RRSIGs of a particular type, e.g. RRSIG(NS). Returns a ``dns.rdatatype.RdataType``. )dns rdatatypeNONErIrrrcoverss z Rdata.coverscCs|d>|jBS)zReturn a 32-bit type value, the least significant 16 bits of which are the ordinary DNS type, and the upper 16 bits of which are the "covered" type, if any. Returns an ``int``. )rYr:rIrrrextended_rdatatypeszRdata.extended_rdatatypeNTorigin relativizer!cKt)zEConvert an rdata to text format. Returns a ``str``. NotImplementedErrorr>r\r]r!rrrto_texts z Rdata.to_textFfilecompress canonicalizecCr^rKr_r>rcrdr\rerrr_to_wireszRdata._to_wirecCs8|r |||||dSt}||||||S)z~Convert an rdata to wire format. Returns a ``bytes`` if no output file was specified, or ``None`` otherwise. N)rgioBytesIOgetvalue)r>rcrdr\refrrrto_wires  z Rdata.to_wirezdns.rdata.GenericRdatacCstj|j|j|j|dS)zrCreates a dns.rdata.GenericRdata equivalent of this rdata. Returns a ``dns.rdata.GenericRdata``. )r\)rVrdata GenericRdatar9r:rl)r>r\rrr to_genericszRdata.to_genericcCs|j|dd}|dus J|S)zConvert rdata to a format suitable for digesting in hashes. This is also the DNSSEC canonical form. Returns a ``bytes``. T)r\reN)rl)r>r\wirerrr to_digestables zRdata.to_digestablecCsf|}|tjjkr d}n dtj|d}dtj|jdtj|j|dt|dS)Nr'()z) rYrVrWrXrb rdataclassr9r:r+)r>rYctextrrr__repr__ s(   zRdata.__repr__cCs|SrK)rbrIrrr__str__sz Rdata.__str__cCsz|}d}Wntjjytr|tjj}d}Ynwz|}d}Wntjjy=tr9|tjj}d}YnwtrK||krJ|rHdSdSn|sO|rQt||krWdS||kr]dSdS)aCompare an rdata with another rdata of the same rdtype and rdclass. For rdata with only absolute names: Return < 0 if self < other in the DNSSEC ordering, 0 if self == other, and > 0 if self > other. For rdata with at least one relative names: The rdata sorts before any rdata with only absolute names. When compared with another relative rdata, all names are made absolute as if they were relative to the root, as the proper origin is not available. While this creates a stable ordering, it is NOT guaranteed to be the DNSSEC ordering. In the future, all ordering comparisons for rdata with relative names will be disallowed. FTr5r4r)rqrVnameNeedAbsoluteNameOrOrigin_allow_relative_comparisonsrootr )r>otherour our_relativetheirtheir_relativerrr_cmps:z Rdata._cmpcCst|tsdS|j|jks|j|jkrdSd}d}z|}Wntjjy3|tjj}d}Ynwz|}WntjjyN|tjj}d}Ynw||krUdS||kS)NFT) r*r8r9r:rqrVrzr{r})r>r~rrrrrrr__eq__Ls*   z Rdata.__eq__cCs6t|tsdS|j|jks|j|jkrdS|| SNT)r*r8r9r:rr>r~rrr__ne__as  z Rdata.__ne__cCs4t|tr|j|jks|j|jkrtS||dkSNrr*r8r9r:NotImplementedrrrrr__lt__h   z Rdata.__lt__cCs4t|tr|j|jks|j|jkrtS||dkSrrrrrr__le__qrz Rdata.__le__cCs4t|tr|j|jks|j|jkrtS||dkSrrrrrr__ge__zrz Rdata.__ge__cCs4t|tr|j|jks|j|jkrtS||dkSrrrrrr__gt__rz Rdata.__gt__cCst|tjjSrK)hashrqrVrzr}rIrrr__hash__szRdata.__hash__r9r:tok relativize_tocCr^rKr_)rBr9r:rr\r]rrrr from_texts zRdata.from_textparsercCr^rKr_rBr9r:rr\rrrfrom_wire_parserszRdata.from_wire_parserkwargsc stjj}D])}|dkrq ||vr!tdjjd|d|dvr2tdjjd|dq fdd|D}j|}dj}|d urTt |d||S) a Create a new Rdata instance based on the instance replace was invoked on. It is possible to pass different parameters to override the corresponding properties of the base Rdata. Any field specific to the Rdata type can be replaced, but the *rdtype* and *rdclass* fields cannot. Returns an instance of the same Rdata subclass as *self*. r;'z' object has no attribute 'r9r:zCannot overwrite 'z ' attribute 'c3s"|] }|t|VqdSrK)getrA)rkeyrr>rrrCs z Rdata.replace..N) inspect signaturer? parametersAttributeErrorrGr rr;rPrQ)r>rrrargsrdr;rrrreplaces&  z Rdata.replacecCtjj|SrK)rVrv RdataClassmakerBvaluerrrr<zRdata._as_rdataclasscCrrK)rVrW RdataTyperrrrrr=rzRdata._as_rdatatyperr, max_lengthempty_okcCs||r t|tr |}nt|trt|}n t|tr|}ntd|dur0t||kr0td|ss   z Rdata._as_ttlcs6z|fWStytfdd|DYSw)Nc3s|]}|VqdSrKr)rvas_valuerrrCPsz"Rdata._as_tuple..) Exceptiontuple)rBrrrrr _as_tupleGs   zRdata._as_tuplecCst|}t||SrK)listrandomshuffle)rBiterablerOrrr_processing_orderTs zRdata._processing_orderrNNF)NNNFrKNTN)FNT)NN)@r r r r r@r?rJrNrTrVrWrrYrr[rrzrrrr+rrb CompressTypergrrlrorqrxryrrrrrrrr classmethodrvr tokenizer TokenizerrrpParserrrr<r=rrrrrrrrrrrrrrrrrr8s                /           -               r8c s~eZdZdZdgZfddZ  ddeejj de d e e e fd e fd d Ze dd dZdddZedddZZS)rnzGeneric Rdata Class This class is used for rdata types for which we have no better implementation. It implements the DNS "unknown RRs" scheme. rcst||||_dSrK)superr?r)r>r9r:rrGrrr?es zGenericRdata.__init__NTr\r]r!rUcKs dt|jt|jfi|S)Nz\# %d )rrr"rarrrrbis zGenericRdata.to_textc Csj|}|r |jdkrtjd|}|d} t | } t | |kr/tjd|||| S)N\#z$generic rdata does not start with \#Tz'generic rdata hex data has wrong length) r is_identifierrrV exception SyntaxErrorget_int!concatenate_remaining_identifiersr,r unhexlifyr) rBr9r:rr\r]rtokenlengthhexrrrrrqs     zGenericRdata.from_textFcCs||jdSrK)writerrfrrrrgszGenericRdata._to_wirecCs||||SrK) get_remainingrrrrrszGenericRdata.from_wire_parserrrrrK)r r r r r@r?rrVrzrrrr+rrbrrrgr __classcell__rrrrrn[s*     rn_rdata_classesz dns.rdtypesc Cst||f}|srttjj|f}|srtrrtj|}tj|}|dd}zt d t ||g}t ||}|t||f<Wn2t yqzt d t d|g}t ||}|ttjj|f<|t||f<Wn t ynYnwYnw|s~|r~t}|t||f<|S)N-_.ANY)rrrVrWr_dynamic_load_allowedrvrbrrr_module_prefixrA ImportErrorrn)r9r: use_genericrB rdclass_text rdtype_textmodrrrget_rdata_classs8        rcCsBtjjD] }ttjj|dqttjjtjjd|rdadSdS)aLoad all rdata types for which dnspython has a non-generic implementation. Normally dnspython loads DNS rdatatype implementations on demand, but in some specialized cases loading all types at an application-controlled time is preferred. If *disable_dynamic_load*, a ``bool``, is ``True`` then dnspython will not attempt to use its dynamic loading mechanism if an unknown type is subsequently encountered, and will simply use the ``GenericRdata`` class. FN) rVrWrrrvINCHAr)disable_dynamic_loadr:rrrload_all_typess rr9r:rr\r]r idna_codecrUc Cs2t|tr tjj||d}tjj|}tjj |}t ||}tj tj j gd}|tkrg|} || | rg| jdkrgt||||||} t||| jdt| j|}|} | | jkrgtj d|duru|||||||}|} | jdurt|d| j|WdS1swYdS)a3Build an rdata object from text format. This function attempts to dynamically load a class which implements the specified rdata class and type. If there is no class-and-type-specific implementation, the GenericRdata class is used. Once a class is chosen, its from_text() class method is called with the parameters to this function. If *tok* is a ``str``, then a tokenizer is created and the string is used as its input. *rdclass*, a ``dns.rdataclass.RdataClass`` or ``str``, the rdataclass. *rdtype*, a ``dns.rdatatype.RdataType`` or ``str``, the rdatatype. *tok*, a ``dns.tokenizer.Tokenizer`` or a ``str``. *origin*, a ``dns.name.Name`` (or ``None``), the origin to use for relative names. *relativize*, a ``bool``. If true, name will be relativized. *relativize_to*, a ``dns.name.Name`` (or ``None``), the origin to use when relativizing names. If not set, the *origin* value will be used. *idna_codec*, a ``dns.name.IDNACodec``, specifies the IDNA encoder/decoder to use if a tokenizer needs to be created. If ``None``, the default IDNA 2003 encoder/decoder is used. If a tokenizer is not created, then the codec associated with the tokenizer is the one that is used. Returns an instance of the chosen Rdata subclass. )rNrrz9compressed data in generic syntax form of known rdatatyper;)r*r+rVrrrvrrrWrrrExceptionWrapperrrnrungetrrr from_wirerrrlget_eol_as_tokencommentrPrQ) r9r:rr\r]rrrBrmrgrdatarwirerrrrs> -      $rrcCsjtjj|}tjj|}t||}tjtjj | ||||WdS1s.wYdS)aBuild an rdata object from wire format This function attempts to dynamically load a class which implements the specified rdata class and type. If there is no class-and-type-specific implementation, the GenericRdata class is used. Once a class is chosen, its from_wire() class method is called with the parameters to this function. *rdclass*, a ``dns.rdataclass.RdataClass`` or ``str``, the rdataclass. *rdtype*, a ``dns.rdatatype.RdataType`` or ``str``, the rdatatype. *parser*, a ``dns.wire.Parser``, the parser, which should be restricted to the rdata length. *origin*, a ``dns.name.Name`` (or ``None``). If not ``None``, then names will be relativized to this origin. Returns an instance of the chosen Rdata subclass. N) rVrvrrrWrrrr FormErrorr)r9r:rr\rBrrrrs  $rrpcurrentrdlencCsJtj||}||t||||WdS1swYdS)a1Build an rdata object from wire format This function attempts to dynamically load a class which implements the specified rdata class and type. If there is no class-and-type-specific implementation, the GenericRdata class is used. Once a class is chosen, its from_wire() class method is called with the parameters to this function. *rdclass*, an ``int``, the rdataclass. *rdtype*, an ``int``, the rdatatype. *wire*, a ``bytes``, the wire-format message. *current*, an ``int``, the offset in wire of the beginning of the rdata. *rdlen*, an ``int``, the length of the wire-format rdata *origin*, a ``dns.name.Name`` (or ``None``). If not ``None``, then names will be relativized to this origin. Returns an instance of the chosen Rdata subclass. N)rVrpr restrict_tor)r9r:rpr r r\rrrrr@s"  $rc@seZdZdZddhZdZdS)RdatatypeExistszDNS rdatatype already exists.r9r:zKThe rdata type with class {rdclass:d} and rdtype {rdtype:d} already exists.N)r r r r supp_kwargsfmtrrrrr gs r Fimplementationr is_singletoncCsftjj|}t||}|tkstj|rt||dt|| ddt ||f<tj |||dS)aDynamically register a module to handle an rdatatype. *implementation*, a module implementing the type in the usual dnspython way. *rdtype*, an ``int``, the rdatatype to register. *rdtype_text*, a ``str``, the textual form of the rdatatype. *is_singleton*, a ``bool``, indicating if the type is a singleton (i.e. RRsets of the type can have only one member.) *rdclass*, the rdataclass of the type, or ``dns.rdataclass.ANY`` if it applies to all classes. rrrN) rVrWrrrrn is_metatyper rArr register_type)rr:rrr9 existing_clsrrrrqs    r)T)NTNNrK)Er r$rrrhrDr importlibrtypingrrrrr dns.exceptionrV dns.immutabledns.ipv4dns.ipv6dns.namedns.rdataclass dns.rdatatype dns.tokenizerdns.ttldns.wire _chunksizer|r DNSExceptionr rr"r&r.r3r7 immutableconstify _constifyr8rnrrvrrWr__annotations__rrrrr+rrrzrr IDNACodecrrprrrrrr rrrrrrs   \,"      ^  *  '