3
OfL @ s d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z
ddl Z
ddl Z
ddl
Z
ddlZ
ddlZ
ddlmZmZ ejd/kreZnejZejZG dd de
jjZG d d
d
e
jjZ d d Z d
d Z dd Z dd Z dd Z!e"edre a#ne a#dd Z$dd Z%dd Z&d d Z'd d Z(d0d"d#Z)d$d% Z*d&d' Z+d(d) Z,d1d*d+Z-e
j.j/e
j0j1dd ddd,dddddd!e
j2j3fd-d.Z4dS )2zTalk to a DNS server. )
generatorsN )long string_types c @ s e Zd ZdZdS )UnexpectedSourcez=A DNS query response came from an unexpected address or port.N)__name__
__module__ __qualname____doc__ r r /usr/lib/python3.6/query.pyr , s r c @ s e Zd ZdZdS ) BadResponsez d}t |}x,||k r8t| | || j||d 7 }qW dS )zWrite the specified data to the socket.
A Timeout exception will be raised if the operation is not completed
by the expiration time.
r N)rQ r* send)rR datar" Zcurrentlr r r
_net_write/ s
rY c C sp y| j | W n\ tjk
rj tj d d \}}t|drF|j}n|d }|tjtjtj gkrf|Y nX d S )N r! r )
Zconnectr> r sysexc_infohasattrr! Z EINPROGRESSZ EWOULDBLOCKZEALREADY)r( ZaddressZtyvZv_errr r r
_connect; s
r_ c C s | j }t|||||\}} }t|tjd}
d} zt|} |
jd tj } |dk r^|
j| t |
| t
|}
t j d|
| }t
|
|| t|
d| }t jd|\}
t|
|
| }W d| dkrd}n tj | }|
j X tjj|| j| j|d}||_| j|st|S )a Return the response obtained after sending a query via TCP.
@param q: the query
@type q: dns.message.Message object
@param where: where to send the message
@type where: string containing an IPv4 or IPv6 address
@param timeout: The number of seconds to wait before the query times out.
If None, the default, wait forever.
@type timeout: float
@param port: The port to which to send the message. The default is 53.
@type port: int
@param af: 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.
@type af: int
@rtype: dns.message.Message object
@param source: source address. The default is the wildcard address.
@type source: string
@param source_port: The port from which to send the message.
The default is 0.
@type source_port: int
@param one_rr_per_rrset: Put each RR into its own RRset
@type one_rr_per_rrset: bool
r Nz!HrZ )r9 r: r; )r< r7 r= r> SOCK_STREAMr r@ r rA r_ rQ structpackrY rU unpackrG r rC rD r9 rE rF r )rH r3 r r4 r, r5 r6 r; rJ r/ r( rK r" rX tcpmsgldatarL rM r r r
tcpI s8
rf Tc & c s t |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| n tj d|| }t ||| d }d
}d }d}|rj|} tjj } nd} |} d} d
} x6|st|} | dks|dk r| |kr|} |rt || |j d \}}!n&t!|d | }"tj"d|"\}t!||| }|tjjk}#tjj#||j$|j%d
| | d
| |#d
}$|$j&} d } d}%|dkr|$j' sV|$j'd j| krbtj(j)d|$j'd }|j*tjj+krtj(j)dd}%|j, }|tjjkr|d j-|
krd
}nd
}x|$j'|%d D ]}|rtj(j)d|j*tjj+krl|j| krl|r&|d j-|
kr tj(j)dd }n|tjjkr:| }||kr~|tjj.ksf|tjjkr~|r~d
}n|rtjj.}d }qW |r|j$r|$j/ rtj(j)d|$V q~W |j0 dS )a Return a generator for the responses to a zone transfer.
@param where: where to send the message
@type where: string containing an IPv4 or IPv6 address
@param zone: The name of the zone to transfer
@type zone: dns.name.Name object or string
@param rdtype: The type of zone transfer. The default is
dns.rdatatype.AXFR.
@type rdtype: int or string
@param rdclass: The class of the zone transfer. The default is
dns.rdataclass.IN.
@type rdclass: int or string
@param timeout: The number of seconds to wait for each response message.
If None, the default, wait forever.
@type timeout: float
@param port: The port to which to send the message. The default is 53.
@type port: int
@param keyring: The TSIG keyring to use
@type keyring: dict
@param keyname: The name of the TSIG key to use
@type keyname: dns.name.Name object or string
@param relativize: 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().
@type relativize: bool
@param af: 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.
@type af: int
@param lifetime: 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.
@type lifetime: float
@rtype: generator of dns.message.Message objects.
@param source: source address. The default is the wildcard address.
@type source: string
@param source_port: The port from which to send the message.
The default is 0.
@type source_port: int
@param serial: The SOA serial number to use as the base for an IXFR diff
sequence (only meaningful if rdtype == dns.rdatatype.IXFR).
@type serial: int
@param use_udp: Use UDP (only meaningful for IXFR)
@type use_udp: bool
@param keyalgorithm: The TSIG algorithm to use; defaults to
dns.tsig.default_algorithm
@type keyalgorithm: string
r INSOAz. . %u 0 0 0 0N) algorithmzcannot do a UDP AXFRz!HFTi rZ )r9 r: xfrorigintsig_ctxZmultifirstr; z No answer or RRset not for qnamezfirst RRset is not an SOAr zanswers after final SOAzIXFR base serial mismatchz missing TSIG)1
isinstancer r nameZ from_text rdatatyperC Z
make_queryZIXFRrrsetZ authorityappendZuse_tsigr< r7
ValueErrorr= r> r? r` r@ rA r r_ rQ r* rV ra rb rY emptyr) rB rU rc rD r9 rE rl Zanswerr FormErrorrdtyperh copyserialAXFRZhad_tsigrG )&r3 Zzonerv Zrdclassr r4 r9 ZkeynameZ
relativizer, Zlifetimer5 r6 rx Zuse_udpZ keyalgorithmrH rq rJ r/ r( r" rX rd r# Z delete_modeZ
expecting_SOAZ soa_rrsetrk Zonamerl rm Z mexpirationr. re Zis_ixfrrM Z answer_indexr r r
rj s 5
rj )r )Nr8 NNr FFF)Nr8 NNr F)5r Z
__future__r r! r r> ra r[ r Z
dns.exceptionr Zdns.inetZdns.nameZ dns.messageZdns.rdataclassZ
dns.rdatatypeZ_compatr r version_infoOSErrorr r r= r Z DNSExceptionr ru r r r r r% r' r] r r) r* r- r1 r7 rN rU rY r_ rf rp ry Z
rdataclassrg ZtsigZdefault_algorithmrj r r r r
sX
I
?