a sd_BJ @sdZddlZddlZddlZddlZddlZddlZddlZddl Zddl Zddl Zddl Zddl mZmZmZmZejjZddZdddZddd Zd d d Zd!ddZd"ddZd#ddZddZd$ddZd%ddZd&ddZdS)'zTalk to a DNS server.N)_compute_times_matches_destination BadResponsesslcCsN|s|rF|dur>|tjkr d}n|tjkr0d}ntd|||fSdSdS)Nz0.0.0.0z::zunknown address family )socketZAF_INETZAF_INET6NotImplementedError)afZaddressportr 0/usr/lib/python3/dist-packages/dns/asyncquery.py _source_tuple(s  r cCs&|r|st}t||dSdSdS)Nr)timemax) expirationZnowr r r _timeout8s rcs@t|tjjr|}t}|||t||IdH}||fS)aSend a DNS message to the specified UDP socket. *sock*, a ``dns.asyncbackend.DatagramSocket``. *what*, a ``bytes`` or ``dns.message.Message``, the message to send. *destination*, a destination tuple appropriate for the address family of the socket, specifying where to send the query. *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. N) isinstancednsmessageMessageto_wirer Zsendtor)sockwhat destinationr sent_timenr r r send_udpAs rFc s\d} |dt|IdH\} } t|j| ||rq2qt} tjj| |||||d} | | | fS)aRead a DNS message from a UDP socket. *sock*, a ``dns.asyncbackend.DatagramSocket``. *destination*, a destination tuple appropriate for the address family of the socket, specifying where the message is expected to arrive from. When receiving a response, this would be where the associated query was sent. *expiration*, a ``float`` or ``None``, the absolute time at which a timeout exception should be raised. If ``None``, no timeout will occur. *ignore_unexpected*, a ``bool``. If ``True``, ignore responses from unexpected sources. *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 ``bytes``, the MAC of the request (for TSIG). *ignore_trailing*, a ``bool``. If ``True``, ignore trailing junk at end of the received message. *raise_on_truncation*, a ``bool``. If ``True``, raise an exception if the TC bit is set. Raises if the message is malformed, if network errors occur, of if there is a timeout. Returns a ``(dns.message.Message, float, tuple)`` tuple of the received message, the received time, and the address where the message arrived from. riN)keyring request_macone_rr_per_rrsetignore_trailingraise_on_truncation)ZrecvfromrrZfamilyr rr from_wire) rrrignore_unexpectedrrrr r!wireZ from_address received_timerr r r receive_udpYs(  r'5c  s|} t|\} }d}ztj|}t||f|}| r>| }n2| sLtj} t|||}| |t j d|IdH}t || ||IdHt ||||||j|j|| IdH\}}}|| |_||st|W| s|r|IdHSn| s|r|IdH0dS)aReturn the response obtained after sending a query via UDP. *q*, a ``dns.message.Message``, the query to send *where*, a ``str`` 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. *source*, a ``str`` 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. *ignore_unexpected*, a ``bool``. If ``True``, ignore responses from unexpected sources. *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. *raise_on_truncation*, a ``bool``. If ``True``, raise an exception if the TC bit is set. *sock*, a ``dns.asyncbackend.DatagramSocket``, or ``None``, the socket to use for the query. If ``None``, the default, a socket is created. Note that if a socket is provided, the *source*, *source_port*, and *backend* are ignored. *backend*, a ``dns.asyncbackend.Backend``, or ``None``. If ``None``, the default, then dnspython will use the default backend. Returns a ``dns.message.Message``. Nr)rrrinetaf_for_address_lltuple asyncbackendget_default_backendr make_socketrZ SOCK_DGRAMrr'rmacr is_responserclose)qwheretimeoutr source source_portr#rr r!rbackendr$ begin_timersrrstupler&r%_r r r udps>,       r<c srz.t|||||||||d| | IdH} | dfWStjjylt||||||||| | IdH} | dfYS0dS)aReturn the response to the query, trying UDP first and falling back to TCP if UDP results in a truncated response. *q*, a ``dns.message.Message``, the query to send *where*, a ``str`` 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. *source*, a ``str`` 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. *ignore_unexpected*, a ``bool``. If ``True``, ignore responses from unexpected sources. *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. *udp_sock*, a ``dns.asyncbackend.DatagramSocket``, or ``None``, the socket to use for the UDP query. If ``None``, the default, a socket is created. Note that if a socket is provided the *source*, *source_port*, and *backend* are ignored for the UDP query. *tcp_sock*, a ``dns.asyncbackend.StreamSocket``, or ``None``, the socket to use for the TCP query. If ``None``, the default, a socket is created. Note that if a socket is provided *where*, *source*, *source_port*, and *backend* are ignored for the TCP query. *backend*, a ``dns.asyncbackend.Backend``, or ``None``. If ``None``, the default, then dnspython will use the default backend. Returns a (``dns.message.Message``, tcp) tuple where tcp is ``True`` if and only if TCP was used. TNF)r<rrZ Truncatedtcp) r2r3r4r r5r6r#rr Zudp_sockZtcp_sockr7responser r r udp_with_fallbacks0   r?csTt|tjjr|}t|}td||}t}| ||IdHt||fS)a{Send a DNS message to the specified TCP socket. *sock*, a ``socket``. *what*, a ``bytes`` 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. !HN) rrrrrlenstructZpackr Zsendall)rrrlZtcpmsgrr r r send_tcpsrDcsHd}|dkrD||t|IdH}|dkr.t|t|}||}q|S)z|Read the specified number of bytes from stream. Keep trying until we either get the desired amount, or we hit EOF. rrN)ZrecvrEOFErrorrA)rcountrr9rr r r _read_exactly-s  rGc sXt|d|IdH}td|\}t|||IdH}t} 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 ``bytes``, 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, float)`` tuple of the received message and the received time. Nr@)rrrr )rGrBZunpackr rrr") rrrrrr ZldatarCr$r%r&r r r receive_tcp;s rIc s|} t|\} } d} z|r2|IdH|} nJtj|}t|||}||f}| s`tj} | |t j d|||IdH} t | | | IdHt | | ||j|j|IdH\}}|| |_||st|W|s| r| IdHSn|s| r| IdH0dS)aReturn the response obtained after sending a query via TCP. *q*, a ``dns.message.Message``, the query to send *where*, a ``str`` 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. *source*, a ``str`` 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. *sock*, a ``dns.asyncbacket.StreamSocket``, or ``None``, the socket to use for the query. If ``None``, the default, a socket is created. Note that if a socket is provided *where*, *port*, *source*, *source_port*, and *backend* are ignored. *backend*, a ``dns.asyncbackend.Backend``, or ``None``. If ``None``, the default, then dnspython will use the default backend. Returns a ``dns.message.Message``. Nr)rrZ getpeernamerr)r*r r,r-r.r SOCK_STREAMrDrIrr/r r0rr1)r2r3r4r r5r6rr rr7r$r8rr9rr:dtupler&r%r r r r=`s>&        r=Uc  st|\} } |s| dur0t} | dur8d| _nd} d} tj|}t|||}||f}| sftj } | |t j d|||| | IdH}n|}zXt | }t|||||||||| IdH}t}|| |_|W|s|r|IdHSn|s|r|IdH0dS)aReturn the response obtained after sending a query via TLS. *q*, a ``dns.message.Message``, the query to send *where*, a ``str`` 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 853. *source*, a ``str`` 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. *sock*, an ``asyncbackend.StreamSocket``, or ``None``, the socket to use for the query. If ``None``, the default, a socket is created. Note that if a socket is provided, it must be a connected SSL stream socket, and *where*, *port*, *source*, *source_port*, *backend*, *ssl_context*, and *server_hostname* are ignored. *backend*, a ``dns.asyncbackend.Backend``, or ``None``. If ``None``, the default, then dnspython will use the default backend. *ssl_context*, an ``ssl.SSLContext``, the context to use when establishing a TLS connection. If ``None``, the default, creates one with the default configuration. *server_hostname*, a ``str`` containing the server's hostname. The default is ``None``, which means that no hostname is known, and if an SSL context is created, hostname checking will be disabled. Returns a ``dns.message.Message``. NFr)rrZcreate_default_contextZcheck_hostnamerr)r*r r,r-r.rrJrr=r r1)r2r3r4r r5r6rr rr7Z ssl_contextZserver_hostnamer8rrr:rKr9r>Zend_timer r r tlssB0        rM)N)N)NNFFNrFF) Nr(NrFFFFNN) Nr(NrFFFNNN)N)NFNrF)Nr(NrFFNN) NrLNrFFNNNN) __doc__rrBr Zdns.asyncbackendrZ dns.exceptionZdns.inetZdns.nameZ dns.messageZ dns.rcodeZdns.rdataclassZ dns.rdatatypeZ dns.queryrrrrr)Zlow_level_address_tupler+r rrr'r<r?rDrGrIr=rMr r r r sX  5 I <  % H