3 [_]\@sdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddlZ ddlZ ddlZ ddlmZmZmZereZnejZejZGddde jjZGdd d e jjZGd d d e jjZd d ZddZ ddZ!ddZ"ddZ#e$edre a%ne!a%ddZ&ddZ'ddZ(ddZ)d7dd Z*d8d#d$Z+d9d&d'Z,d(d)Z-d*d+Z.d:d,d-Z/d;d.d/Z0d0d1Z1dd}t|}x,||kr8t||||j||d7}qWdS)zWrite the specified data to the socket. A Timeout exception will be raised if the operation is not completed by the expiration time. rN)rmr7send)rJdatar/currentlr r r _net_writecs   rtcCsXt|tjjr|j}t|}tjd||}t||t j }t |||t||fS)a|Send a DNS message to the specified TCP socket. *sock*, a ``socket``. *what*, a ``binary`` or ``dns.message.Message``, the message to send. *expiration*, a ``float`` or ``None``, the absolute time at which a timeout exception should be raised. If ``None``, no timeout will occur. Returns an ``(int, float)`` tuple of bytes sent and the sent time. z!H) rFrrrGrHrmstructpackr7rrt)rJrKr/rstcpmsgrLr r r send_tcpos  rxc CsLt|d|}tjd|\}t|||}tj} tjj|||||d} | | fS)aRead a DNS message from a TCP socket. *sock*, a ``socket``. *expiration*, a ``float`` or ``None``, the absolute time at which a timeout exception should be raised. If ``None``, no timeout will occur. *one_rr_per_rrset*, a ``bool``. If ``True``, put each RR into its own RRset. *keyring*, a ``dict``, the keyring to use for TSIG. *request_mac*, a ``binary``, the MAC of the request (for TSIG). *ignore_trailing*, a ``bool``. If ``True``, ignore trailing junk at end of the received message. Raises if the message is malformed, if network errors occur, of if there is a timeout. Returns a ``dns.message.Message`` object. z!H)rPrQrRrS)roruunpackrrrrW) rJr/rRrPrQrSldatarsrYr[r\r r r receive_tcps   r|c Cspy|j|Wn\tjk rjtjdd\}}t|drF|j}n|d}|tjtjtj gkrf|YnXdS)Nryr.r) connectr`r%sysexc_infohasattrr.Z EINPROGRESS EWOULDBLOCKZEALREADY)r5addresstyvZv_errr r r _connects rc  Cs|j} t|||||\}} }t|tjd} d} d} z`t|}| jdtj} |dk rb| j|t | | t | | |t | |||j |j |\}} Wd| dks| dkrd}n| | }| jX||_|j|st|S)aReturn the response obtained after sending a query via TCP. *q*, a ``dns.message.Message``, the query to send *where*, a ``text`` containing an IPv4 or IPv6 address, where to send the message. *timeout*, a ``float`` or ``None``, the number of seconds to wait before the query times out. If ``None``, the default, wait forever. *port*, an ``int``, the port send the message to. The default is 53. *af*, an ``int``, the address family to use. The default is ``None``, which causes the address family to use to be inferred from the form of *where*. If the inference attempt fails, AF_INET is used. This parameter is historical; you need never set it. *source*, a ``text`` containing an IPv4 or IPv6 address, specifying the source address. The default is the wildcard address. *source_port*, an ``int``, the port from which to send the message. The default is 0. *one_rr_per_rrset*, a ``bool``. If ``True``, put each RR into its own RRset. *ignore_trailing*, a ``bool``. If ``True``, ignore trailing junk at end of the received message. Returns a ``dns.message.Message``. rN)rHrEr_r` SOCK_STREAMrrbrrcrrxr|rPrdrerfr)rgr@rrAr;rBrCrRrSrYrDr5Z begin_timer[r/r\rir r r tcps0"      rTc' cst|trtjj|}t|tr,tjj|}tjj|||}|tjjkrltj j|dddd| }|j j ||dk r|j |||d|j }t| ||| | \} }} |r|tjjkrtdt| tjd}nt| tjd}|jd| dk r|j| t| }t||t|}|r*t|||j|ntjd||}t|||d }d }d }d}|rj|}tjj}nd}|}d}d }xJ|st|} | dks| |kr|} |rt|||j d \}}!n&t!|d | }"tj"d|"\}t!||| }|tjjk}#tjj#||j$|j%d ||d ||#d }$|$j&}%|%tj&j'kr6t(|%|$j)}d }d}&|dkr|$j* sj|$j*dj|krvtj+j,d|$j*d}|j-tjj.krtj+j,dd}&|j/}|tjjkr|dj0| krd }nd }x|$j*|&dD]}|rtj+j,d|j-tjj.kr|j|kr|r:|dj0| kr4tj+j,dd }n|tjjkrN| }||kr|tjj1ksz|tjjkr|rd }n|rtjj1}d }qW|r|j$r|$j2 rtj+j,d|$Vq~W|j3dS)aReturn a generator for the responses to a zone transfer. *where*. If the inference attempt fails, AF_INET is used. This parameter is historical; you need never set it. *zone*, a ``dns.name.Name`` or ``text``, the name of the zone to transfer. *rdtype*, an ``int`` or ``text``, the type of zone transfer. The default is ``dns.rdatatype.AXFR``. ``dns.rdatatype.IXFR`` can be used to do an incremental transfer instead. *rdclass*, an ``int`` or ``text``, the class of the zone transfer. The default is ``dns.rdataclass.IN``. *timeout*, a ``float``, the number of seconds to wait for each response message. If None, the default, wait forever. *port*, an ``int``, the port send the message to. The default is 53. *keyring*, a ``dict``, the keyring to use for TSIG. *keyname*, a ``dns.name.Name`` or ``text``, the name of the TSIG key to use. *relativize*, a ``bool``. If ``True``, all names in the zone will be relativized to the zone origin. It is essential that the relativize setting matches the one specified to ``dns.zone.from_xfr()`` if using this generator to make a zone. *af*, an ``int``, the address family to use. The default is ``None``, which causes the address family to use to be inferred from the form of *where*. If the inference attempt fails, AF_INET is used. This parameter is historical; you need never set it. *lifetime*, a ``float``, the total number of seconds to spend doing the transfer. If ``None``, the default, then there is no limit on the time the transfer may take. *source*, a ``text`` containing an IPv4 or IPv6 address, specifying the source address. The default is the wildcard address. *source_port*, an ``int``, the port from which to send the message. The default is 0. *serial*, an ``int``, the SOA serial number to use as the base for an IXFR diff sequence (only meaningful if *rdtype* is ``dns.rdatatype.IXFR``). *use_udp*, a ``bool``. If ``True``, use UDP (only meaningful for IXFR). *keyalgorithm*, a ``dns.name.Name`` or ``text``, the TSIG algorithm to use. Raises on errors, and so does the generator. Returns a generator of ``dns.message.Message`` objects. rINSOAz. . %u 0 0 0 0N) algorithmzcannot do a UDP AXFRz!HFTiry)rPrQxfrorigintsig_ctxZmultifirstrRz No answer or RRset not for qnamezfirst RRset is not an SOArzanswers after final SOAzIXFR base serial mismatchz missing TSIG)4rFrrnameZ from_text rdatatyperZ make_queryZIXFRrrset authorityappendZuse_tsigrHrE ValueErrorr_r`rarrbrcrrrmr7rprurvrtemptyr6rTrorzrWrPrdrZNOERRORrrZanswerr) FormErrorrdtypercopyserialAXFRZhad_tsigre)'r@zonerZrdclassrrArPZkeynameZ relativizer;ZlifetimerBrCrZuse_udpZ keyalgorithmrgrrYrDr5r/rsrwr0Z delete_modeZ expecting_SOAZ soa_rrsetrZonamerrZ mexpirationrZr{Zis_ixfrr\rZ answer_indexr r r rs=                            r)N)NFFNrOF)Nr^NNrFFF)N)NFNrOF)Nr^NNrFF):r __future__rr.rr`rur~rZ dns.exceptionrZdns.inetZdns.nameZ dns.messageZ dns.rcodeZdns.rdataclassZ dns.rdatatype_compatrrrOSErrorr,r%r_r)Z DNSExceptionrrrrrr'r(r2r4rr+r6r7r<rErNr]rjrortrxr|rrrrZ rdataclassrZtsigZdefault_algorithmrr r r r sh       1 ?  " =