a sd_: @spdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ z(ddlZddlmZddlmZdZWneydZYn0z ddlZWn eyGdddZYn0ejZGd d d e jjZGd d d e jj Z!Gd dde jjZ"Gddde jjZ#ddZ$ddZ%ddZ&ddZ'ddZ(e)edre%a*ne&a*ddZ+ddZ,d d!Z-d"d#Z.dId$d%Z/dJd&d'Z0dKd*d+Z1dLd,d-Z2dMd/d0Z3dNd2d3Z4dOd4d5Z5d6d7Z6d8d9Z7dPd:d;Z8dQdd?Z:dRd@dAZ;dBdCZj?e j@jAdd1dddddddde jBjCf dGdHZDdS)TzTalk to a DNS server.N)SourceAddressAdapter)HostHeaderSSLAdapterTFc@sBeZdZGdddeZGdddeZGdddZddZd S) sslc@s eZdZdS)zssl.WantReadExceptionN__name__ __module__ __qualname__r r +/usr/lib/python3/dist-packages/dns/query.pyWantReadException4sr c@s eZdZdS)zssl.WantWriteExceptionNrr r r r WantWriteException7sr c@s eZdZdS)z ssl.SSLSocketNrr r r r SSLSocket:sr cOs tddS)Nzno ssl support) Exception)selfargskwargsr r r create_default_context=szssl.create_default_contextN)rrrrr r r rr r r r r2src@seZdZdZdS)UnexpectedSourcez=A DNS query response came from an unexpected address or port.Nrrr__doc__r r r r rDsrc@seZdZdZdS) BadResponsezt|tjrV|rV|dkrVWdSt|||||sntjjWn8t y}z |j dt j kr|WYd}~n d}~00d}qdS)NFgrT) r"r exceptionZTimeout isinstancerr pending_polling_backendOSErrorrerrnoZEINTR)r)r*r+r, expirationdoner#er r r _wait_fors   r8cCs|adSr!)r2)fnr r r _set_polling_backendsr:r&cCst|ddd|dS)NTFr8sr5r r r _wait_for_readablesr>cCst|ddd|dS)NFTr;r<r r r _wait_for_writablesr?cCsbz(tj||d}tj||d}Wntjjy@YdS0||ko`|dd|ddkS)NrF)rinetZ inet_ptonr/ SyntaxError)afZa1Za2Zn1Zn2r r r _addresses_equals rDcCs`|sdSt|||s|j|d|dWS|WSWntyb|Yn0dS)NF)Zdo_handshake_on_connectserver_hostname)socket_factoryZ setblockingZbindZ wrap_socketrclose)rCtyperP ssl_contextrSr=r r r _make_sockets     rX /dns-queryc Cstst|} t||||d\}}}d}ddi}zBtj|}|tjkrZd ||| }n|tj krrd ||| }WnJt y| durt j |}|j|d<||j| }t}n|}Yn0|durt|}t}|s|tj}|r|||| r2|dtt| d|j||| || d }n(t|  d } |j!|||| d | id }Wdn1sp0Y|j"d ks|j"dkrt d ||j"|j#tj$j%|j#|j&|j'||d}|j(|_)|*|st+|S)aReturn the response obtained after sending a query via DNS-over-HTTPS. *q*, a ``dns.message.Message``, the query to send. *where*, a ``str``, the nameserver IP address or the full URL. If an IP address is given, the URL will be constructed using the following schema: https://:/. *timeout*, a ``float`` or ``None``, the number of seconds to wait before the query times out. If ``None``, the default, wait forever. *port*, a ``int``, the port to send the query to. The default is 443. *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. *session*, a ``requests.session.Session``. If provided, the session to use to send the queries. *path*, a ``str``. If *where* is an IP address, then *path* will be used to construct the URL to send the DNS query to. *post*, a ``bool``. If ``True``, the default, POST method will be used. *bootstrap_address*, a ``str``, the IP address to use to bypass the system's DNS resolver. *verify*, a ``str``, containing a path to a certificate file or directory. Returns a ``dns.message.Message``. FNZacceptzapplication/dns-messagezhttps://{}:{}{}zhttps://[{}]:{}{}ZHost)z content-typezcontent-length)headersdatar#verify=r)r[r#r]paramsi+z2{} responded with status code {} Response body: {}keyring request_macone_rr_per_rrsetignore_trailing),have_dohr to_wirerRrrArIrKrLformatrMrJurllibparseZurlsplitZhostnamereplacerr contextlib ExitStack enter_contextrequestsZsessionsZSessionZmountupdatestrlenpostbase64Zurlsafe_b64encoderstripgetZ status_codeZcontentr from_wirerbrcelapsedr" is_responser)qrNr#rOrPrQrdreZsessionpathrsZbootstrap_addressr]wirerCrFZtransport_adapterr[Zwhere_afZurlZ split_urlstackresponserr r r httpssv-            &  rcCs<t|tjjr|}t||t}|||}||fS)aSend a DNS message to the specified UDP socket. *sock*, a ``socket``. *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. )r0rrMessagergr?r"Zsendto)sockwhatrFr5 sent_timenr r r send_udps   rc Csjd} t|||d\} } t|j| ||rq0qt} tjj| |||||d} |r\| | fS| | | fSdS)a~Read a DNS message from a UDP socket. *sock*, a ``socket``. *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. If *destination* is not ``None``, returns a ``(dns.message.Message, float)`` tuple of the received message and the received time. If *destination* is ``None``, returns a ``(dns.message.Message, float, tuple)`` tuple of the received message, the received time, and the address where the message arrived from. r)rbrcrdreraise_on_truncationN)r>recvfromrHZfamilyr"rrrw) rrFr5rGrdrbrcrerr|rE received_timerr r r receive_udps -   r5c  Cs|} t||||\} } }t|\}}tx}| r<| }n|t| tj|}t || | |t || ||||j |j || \}}|||_ ||st|WdS1s0YdS)aLReturn 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 ``socket.socket``, 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 nonblocking datagram socket, and the *source* and *source_port* are ignored. Returns a ``dns.message.Message``. N)rgrRr$rlrmrnrXrK SOCK_DGRAMrrrbmacr"ryr)rzrNr#rOrPrQrGrdrerrr|rCrF begin_timer5r}r=rrr r r udps&)      rc Csbz&t|||||||||d| } | dfWStjjy\t||||||||| } | 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 ``socket.socket``, 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, it must be a nonblocking datagram socket, and the *source* and *source_port* are ignored for the UDP query. *tcp_sock*, a ``socket.socket``, 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, it must be a nonblocking connected stream socket, and *where*, *source* and *source_port* are ignored for the TCP query. Returns a (``dns.message.Message``, tcp) tuple where tcp is ``True`` if and only if TCP was used. TFN)rrrZ Truncatedtcp) rzrNr#rOrPrQrGrdreZudp_sockZtcp_sockr~r r r udp_with_fallbacks. rcCsd}|dkrt||z||}Wn8tjy<YqYn"tjy\t||YqYn0|dkrjt|t|}||}q|S)zRead the specified number of bytes from sock. Keep trying until we either get the desired amount, or we hit EOF. A Timeout exception will be raised if the operation is not completed by the expiration time. rr)r>ZrecvrSSLWantReadErrorSSLWantWriteErrorr?EOFErrorrr)rcountr5r=rr r r _net_readQs     rcCsxd}t|}||krtt||z||||d7}Wq tjyZt||Yq Yq tjypYq Yq 0q dS)zWrite the specified data to the socket. A Timeout exception will be raised if the operation is not completed by the expiration time. rN)rrr?sendrrr>r)rr\r5Zcurrentlr r r _net_writehs  rcCsXt|tjjr|}t|}td||}t||t }t |||t||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. !H) r0rrrrgrrstructpackr?r"r)rrr5rtcpmsgrr r r send_tcpzs  rc CsLt|d|}td|\}t|||}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. rra)rrunpackr"rrrw) rr5rdrbrcreldatarr|rrr r r receive_tcps   rcCs`||}|dkrdS|tjtjtjfvrDt|||tjtj }|dkr\t |t |dS)Nr) Z connect_exr4Z EINPROGRESSZ EWOULDBLOCKZEALREADYr?Z getsockoptrKZ SOL_SOCKETZSO_ERRORr3osstrerror)r=Zaddressr5errr r r _connects  rc Cs|} t|\} } t} |r0||} n4t||||\}}}| t|tj |} t | || t | | | t | | ||j |j|\}}|| |_||st|WdS1s0YdS)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 ``socket.socket``, 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 nonblocking connected stream socket, and *where*, *port*, *source* and *source_port* are ignored. Returns a ``dns.message.Message``. N)rgr$rlrmZ getpeernamerRrnrXrK SOCK_STREAMrrrrbrr"ryr)rzrNr#rOrPrQrdrerr|rr5r}r=rCrFrrr r r rs,"         rcCsPz|WdStjy,t||YqtjyHt||Yq0qdSr!)Z do_handshakerrr>rr?r<r r r _tls_handshakesrUc  Cs|rt||||||||| S|} t|\} } t||||\}}}| durf|sft} | durfd| _t|tj || | dh}t ||| t || t || | t || ||j|j|\}}|| |_||st|WdS1s0YdS)a:Return 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 ``ssl.SSLSocket``, 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 nonblocking connected SSL stream socket, and *where*, *port*, *source*, *source_port*, and *ssl_context* are ignored. *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``. NF)rWrS)rrgr$rRrrZcheck_hostnamerXrKrrrrrrbrr"ryr)rzrNr#rOrPrQrdrerrWrSr|rr5rCrFr=rrr r r tls s6-          rc( cst|trtj|}tjj|}tj |||}|tjj krdtj |dddd| }|j ||dur||j|||d|}t||| | \}}} | r|tjj krtd| rtjntj}t||| }t| \}}t|||t|}| r t||||ntd||}t|||d }d }d }d}|rL|}tjj}nd}|}d} |st|\}}!|!dus|dur|!|kr|}!| rt||| d \}}"n&t!|d |!}#t"d|#\}t!|||!}|tjj k}$tjj#||j$|j%d || d |$d }%|%&}&|&tj&j'krt(|&|%j)} d}'|dur|%j*rJ|%j*dj|krVtj+,d|%j*d}|j-tjj.kr|tj+,dd}'|/}|tjj krtj01|dj0| krd }nd }|%j*|'dD]}|rtj+,d|j-tjj.krf|j|krf|r |dj0| krtj+,dd }n|tjj kr4| }||krx|tjj2ks`|tjj krx|rxd }n|rtjj2}d }q|r|j$r|%j3stj+,d|%VqXWdn1s0YdS)aReturn a generator for the responses to a zone transfer. *where*, a ``str`` containing an IPv4 or IPv6 address, where to send the message. *zone*, a ``dns.name.Name`` or ``str``, the name of the zone to transfer. *rdtype*, an ``int`` or ``str``, 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 ``str``, 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 ``str``, 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. *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 ``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. *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 ``str``, 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 AXFRrFTrr)rbrcxfrorigintsig_ctxZmultirdz No answer or RRset not for qnamezfirst RRset is not an SOAr@zanswers after final SOAzIXFR base serial mismatchz missing TSIG)4r0rqrnameZ from_text rdatatypeZ RdataTypeZmakerZ make_queryZIXFRrrsetZ authorityappendZuse_tsigrgrRrJrKrrrXr$rrrr?rrrremptyr>rrrrwrbrrZNOERRORrrZanswerr/ FormErrorrdtypercopyserialZSerialAXFRZhad_tsig)(rNZzonerZrdclassr#rOrbZkeynameZ relativizeZlifetimerPrQrZuse_udpZ keyalgorithmrzrr|rCrFZ sock_typer=_r5rrr6Z delete_modeZ expecting_SOAZ soa_rrsetrZonamerZ mexpirationrErZis_ixfrrrZ answer_indexr r r rTs8                          r)T)NN) NrYNrFFNrZTNT)N)NNFFNrFF) NrNrFFFFN) NrNrFFFNN)N)NFNrF)NrNrFFN) NrNrFFNNN)Errlr4rr%rKrr"rtZ urllib.parseriZ dns.exceptionrZdns.inetZdns.nameZ dns.messageZ dns.rcodeZdns.rdataclassZ dns.rdatatypeZ dns.serialroZ!requests_toolbelt.adapters.sourcerZ*requests_toolbelt.adapters.host_header_sslrrf ImportErrorrrTr/Z DNSExceptionrrrrr r$r-r.r8r:hasattrr2r>r?rDrHrRrXrrrrrrrrrrrrrrrZ rdataclassrZtsigZdefault_algorithmrr r r r s           +  q  > < 8  $  <  J