3 [_]s8 @s<dZddlmZddlZddlZddlZddlZddlZddl Zddl Zddl Zddl m Z mZmZy ddlZWnek rddlZYnXdZefddZdZefd d Zed Zd d ZddZGdddeZGdddeZiZdZ ej!Z"ddZ#d!ddZ$d"ddZ%Gdddej&j'Z(dej)j*fdd Z+dS)#z DNS rdata.)BytesION)xrange string_types text_type cs4tj|djfddtdtDjS)z{Convert a binary string into its hex encoding, broken up into chunks of chunksize characters separated by a space.  csg|]}||qSr ).0i) chunksizeliner ./tmp/pip-build-yz_zf6az/dnspython/dns/rdata.py .sz_hexify..r)binasciihexlifyjoinrangelendecode)datar r )r r r_hexify(s rcs4tj|djfddtdtDjS)z~Convert a binary string into its base64 encoding, broken up into chunks of chunksize characters separated by a space. rcsg|]}||qSr r )r r )r r r rr;sz_base64ify..r)base64 b64encoderrrr)rr r )r r r _base64ify5s rs"\cCs~t|tr|j}t|ts$t|}d}xP|D]H}|tkrL|dt|7}q.|dkrj|dkrj|t|7}q.|d|7}q.W|S)z7Escape the characters in a quoted string which need it.\rz\%03d) isinstancerencode bytearray __escapedchr)Zqstringtextcr r r _escapifyAs   r%cCsFx8tt|dddD] }||dkr|d|dSqW|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. rrr&)rr)whatr r r r_truncate_bitmapTs r(c@seZdZdZddgZddZddZdd Zd.d d Zd/ddZ d0ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zed1d(d)Zed2d*d+Zd3d,d-Zd S)4Rdataz#Base class for all DNS rdata types.rdclassrdtypecCs||_||_dS)zInitialize an rdata. *rdclass*, an ``int`` is the rdataclass of the Rdata. *rdtype*, an ``int`` is the rdatatype of the Rdata. N)r*r+)selfr*r+r r r__init__dszRdata.__init__cCstjjS)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 an ``int``. )dns rdatatypeNONE)r,r r rcoversns z Rdata.coverscCs|jd>|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``. )r1r+)r,r r rextended_rdatatype|szRdata.extended_rdatatypeNTcKstdS)zFConvert an rdata to text format. Returns a ``text``. N)NotImplementedError)r,origin relativizekwr r rto_textsz Rdata.to_textcCstdS)zHConvert an rdata to wire format. Returns a ``binary``. N)r4)r,filecompressr5r r rto_wiresz Rdata.to_wirecCst}|j|d||jS)zConvert rdata to a format suitable for digesting in hashes. This is also the DNSSEC canonical form. Returns a ``binary``. N)rr;getvalue)r,r5fr r r to_digestableszRdata.to_digestablecCstjj|j|j|jdS)a9Check that the current contents of the rdata's fields are valid. If you change an rdata by assigning to its fields, it is a good idea to call validate() when you are done making changes. Raises various exceptions if there are problems. Returns ``None``. N)r.rdata from_textr*r+r8)r,r r rvalidates zRdata.validatecCsf|j}|tjjkrd}ndtjj|d}dtjj|jdtjj|j|dt|dS)Nr()z) r1r.r/r0r8 rdataclassr*r+str)r,r1Zctextr r r__repr__s  zRdata.__repr__cCs|jS)N)r8)r,r r r__str__sz Rdata.__str__cCs<|jtjj}|jtjj}||kr(dS||kr4dSdSdS)zCompare an rdata with another rdata of the same rdtype and rdclass. Return < 0 if self < other in the DNSSEC ordering, 0 if self == other, and > 0 if self > other. rrNr&)r>r.nameroot)r,otherZourZtheirr r r_cmpsz Rdata._cmpcCs8t|tsdS|j|jks&|j|jkr*dS|j|dkS)NFr)rr)r*r+rM)r,rLr r r__eq__s  z Rdata.__eq__cCs8t|tsdS|j|jks&|j|jkr*dS|j|dkS)NTr)rr)r*r+rM)r,rLr r r__ne__s  z Rdata.__ne__cCs6t|t s$|j|jks$|j|jkr(tS|j|dkS)Nr)rr)r*r+NotImplementedrM)r,rLr r r__lt__s z Rdata.__lt__cCs6t|t s$|j|jks$|j|jkr(tS|j|dkS)Nr)rr)r*r+rPrM)r,rLr r r__le__s z Rdata.__le__cCs6t|t s$|j|jks$|j|jkr(tS|j|dkS)Nr)rr)r*r+rPrM)r,rLr r r__ge__s z Rdata.__ge__cCs6t|t s$|j|jks$|j|jkr(tS|j|dkS)Nr)rr)r*r+rPrM)r,rLr r r__gt__s z Rdata.__gt__cCst|jtjjS)N)hashr>r.rJrK)r,r r r__hash__szRdata.__hash__cCstdS)N)r4)clsr*r+tokr5r6r r rr@szRdata.from_textcCstdS)N)r4)rWr*r+wirecurrentrdlenr5r r r from_wireszRdata.from_wirecCsdS)zWConvert any domain names in the rdata to the specified relativization. Nr )r,r5r6r r rchoose_relativityszRdata.choose_relativity)NT)NN)N)NT)N)NT)__name__ __module__ __qualname____doc__ __slots__r-r1r3r8r;r>rArHrIrMrNrOrQrRrSrTrV classmethodr@r\r]r r r rr)_s0       r)csVeZdZdZdgZfddZdddZedd d Zdd d Z edd dZ Z S) GenericRdatazGeneric Rdata Class This class is used for rdata types for which we have no better implementation. It implements the DNS "unknown RRs" scheme. rcstt|j||||_dS)N)superrdr-r)r,r*r+r) __class__r rr-szGenericRdata.__init__NTcKsdt|jt|jS)Nz\# %d )rrr)r,r5r6r7r r rr8szGenericRdata.to_textc Cs|j}|j s|jdkr(tjjd|j}g}x&|j}|jrHP|j|jj q6Wdj |} t j | } t | |krtjjd|||| S)Nz\#z$generic rdata does not start with \#z'generic rdata hex data has wrong length)get is_identifiervaluer. exception SyntaxErrorZget_intZ is_eol_or_eofappendrrr unhexlifyr) rWr*r+rXr5r6tokenlengthchunkshexrr r rr@s"   zGenericRdata.from_textcCs|j|jdS)N)writer)r,r9r:r5r r rr;*szGenericRdata.to_wirecCs|||||||S)Nr )rWr*r+rYrZr[r5r r rr\-szGenericRdata.from_wire)NT)NT)NN)N) r^r_r`rarbr-r8rcr@r;r\ __classcell__r r )rfrrds    rdz dns.rdtypescCsdd}tj||f}tjj|}tjj|}|jdd}|stjtjj|f}|sy$|djt ||g}|t||f<WnVt k ry(|djt d|g}|ttjj|f<Wnt k rd}YnXYnX|rt ||}nt }|S)Nc SsFt8t|}|jd}x|ddD]}t||}q&W|SQRXdS)N.r) _import_lock __import__splitgetattr)rJmod componentscompr r r import_module7s  z&get_rdata_class..import_module-_ruANY) _rdata_modulesrhr.rFr8r/replacerr_module_prefix ImportErrorryrd)r*r+r}rzZ rdclass_text rdtype_textrWr r rget_rdata_class5s.      rTcCst|trtjj|}t||}|tkrx|j}|j||j rx|j dkrxtj |||||}t |||j dt|j |S|j |||||S)aNBuild 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 ``text``, then a tokenizer is created and the string is used as its input. *rdclass*, an ``int``, the rdataclass. *rdtype*, an ``int``, the rdatatype. *tok*, a ``dns.tokenizer.Tokenizer`` or a ``text``. *origin*, a ``dns.name.Name`` (or ``None``), the origin to use for relative names. *relativize*, a ``bool``. If true, name will be relativized to the specified origin. Returns an instance of the chosen Rdata subclass. z\#r)rrr. tokenizer Tokenizerrrdrhungetrirjr@r\rr)r*r+rXr5r6rWror?r r rr@Xs      r@cCs*tjj|}t||}|j||||||S)a2Build 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 ``binary``, 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. )r.ZwiredataZ maybe_wraprr\)r*r+rYrZr[r5rWr r rr\s  r\c@seZdZdZddhZdZdS)RdatatypeExistszDNS rdatatype already exists.r*r+z8The rdata type with class {rdclass} and rdtype {rdtype} zalready exists.NzGThe rdata type with class {rdclass} and rdtype {rdtype} already exists.)r^r_r`raZ supp_kwargsfmtr r r rrsrFcCs>t||}|tkrt||d|t||f<tjj|||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 ``text``, 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. )r*r+N)rrdrrr.r/ register_type)implementationr+rZ is_singletonr*Z existing_clsr r rrs    r)NT)N),raiorrrZ dns.exceptionr.Zdns.nameZdns.rdataclassZ dns.rdatatypeZ dns.tokenizerZ dns.wiredata_compatrrr threading _threadingrdummy_threadingZ_hex_chunksizerZ_base64_chunksizerr r!r%r(objectr)rdrrLockrvrr@r\rkZ DNSExceptionrrFINrr r r rsB     ',# 2 !