o jI~#@sdZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z mZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZddlmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'zddl(Z(Wne)yddl*m+Z(Ynwe&rddl,Z,ej-j.Z/ddZ0dYdd Z1 dYd ej2j3d ej4j5e6Bd e d e7dBde e8e7ff ddZ9          dZd ej2j3d e dBd e7dBde:de:de ej;jfdBde6dBde:de:de:dej4j5dBde fddZ?           d[dej4j5de@d e7dBd!e8d"e@dBd#e8de:de:de:de:d ej2j3dBd$ej2jAdBde:dej4j5fd%d&ZB           d\dej4j5de@d e7dBd!e8d"e@dBd#e8de:de:de:d'ej2j3dBd(ej2jCdBd$ej2jAdBde:de ej4j5e:ffd)d*ZD dYd ej2jCd ej4j5e6Bd e7dBde e8e7ffd+d,ZEd-d.ZF     d]d ej2jCd e7dBde:de ej;jfdBde6dBde:de ej4j5e7ffd/d0ZG        d^dej4j5de@d e7dBd!e8d"e@dBd#e8de:de:d ej2jCdBd$ej2jAdBdej4j5fd1d2ZH  3         4d_dej4j5de@d e7dBd!e8d"e@dBd#e8de:de:d ej2jCdBd$ej2jAdBd5e(jIdBd6e@dBd7e:e@Bdej4j5fd8d9ZJd:e d;dd;fddddddd?d4d4ddejLejMfdej4j5de@d e7dBd!e8d"e@dBd#e8de:de:d@e dAdBe@dCe:d7e:e@Be(jIBdDe@dBd:e d;dEe8dFedej4j5f"dGdHZN  >     4  4 d`dej4j5de@dIe@d e7dBd!e8d"e@dBd#e8de:de:d7e:e@Be(jIBd$ej2jAdBdCe:dJejOjPdBdej4j5fdKdLZQ  3      4   dadej4j5de@d e7dBd!e8d"e@dBd#e8de:de:dJejOjPdBd7e:e@Bd$ej2jAdBdMe@dBd6e@dBdej4j5fdNdOZOdPejRjSdQej2jTdej4j5dRe8dBd e7dBd e7de fdSdTZUdddddde!jVdfde@dPejRjSdej4j5dBd!e8d e7dBdUe7dBd"e@dBd#e8dVe!d$ej2jAdBddfdWdXZWdS)bzTalk to a DNS server.N)AnyDictOptionalTuplecast) NullContext) BadResponse HTTPVersionNoDOHNoDOQUDPMode _check_status_compute_times_matches_destination _remaininghave_dohmake_ssl_contextcCsV|s|r)|dur%|tjkrd}||fS|tjkrd}||fStd|||fSdS)Nz0.0.0.0z::zunknown address family )socketAF_INETAF_INET6NotImplementedError)afaddressportr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/dns/asyncquery.py _source_tupleBs  rcCs&|dur|s t}t||dSdSNr)timemax) expirationnowrrr_timeoutRs r"sockwhat destinationr returncsBt|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. The expiration value is meaningless for the asyncio backend, as asyncio's transport sendto() never blocks. Returns an ``(int, float)`` tuple of bytes sent and the sent time. N) isinstancednsmessageMessageto_wirersendtor")r#r$r%r sent_timenrrrsend_udp[s r/Fignore_unexpectedone_rr_per_rrsetkeyring request_macignore_trailingraise_on_truncation ignore_errorsqueryc  sd} |dt|IdH\} } t|j| ||sqt} ztjj| |||||d}Wn-tjjyN}z| rI| durI| |sIWYd}~qd}~wt yY| rXYqw| rf| durf| |sfq|| | fS)ajRead a DNS message from a UDP socket. *sock*, a ``dns.asyncbackend.DatagramSocket``. See :py:func:`dns.query.receive_udp()` for the documentation of the other parameters, and exceptions. Returns a ``(dns.message.Message, float, tuple)`` tuple of the received message, the received time, and the address where the message arrived from. r0TN)r3r4r2r5r6) recvfromr"rfamilyrr(r) from_wire Truncated is_response Exception)r#r%r r1r2r3r4r5r6r7r8wire from_address received_timererrr receive_udpysF      rE5qwheretimeoutrsource source_portbackendc  s$|} t|\}}tj|}t||f|}| rt| }n%| s&tj} t |||}| r5||f}nd}| |t j d||IdH}|4IdH;}t|| ||IdHt||||||j|j|| | | IdH\}}}|||_| sy||syt|WdIdHS1IdHswYdS)aFReturn the response obtained after sending a query via UDP. *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. See :py:func:`dns.query.udp()` for the documentation of the other parameters, exceptions, and return type of this method. Nr)r+rr(inetaf_for_address_lltupler asyncbackendget_default_backendrdatagram_connection_required make_socketr SOCK_DGRAMr/rEr3macrr>r)rGrHrIrrJrKr1r2r5r6r#rLr7r@ begin_timer rr%cmstupledtuplesrCrB_rrrudpsD       0r\udp_socktcp_sockc srzt|||||||||d| | | IdH} | dfWStjjy8t||||||||| | IdH} | dfYSw)aReturn the response to the query, trying UDP first and falling back to TCP if UDP results in a truncated response. *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. See :py:func:`dns.query.udp_with_fallback()` for the documentation of the other parameters, exceptions, and return type of this method. TNF)r\r(r)r=tcp)rGrHrIrrJrKr1r2r5r]r^rLr7responserrrudp_with_fallbacksB#  racs^t|tjjr|jdd}n t|dd|}t}||t ||IdHt||fS)zSend a DNS message to the specified TCP socket. *sock*, a ``dns.asyncbackend.StreamSocket``. See :py:func:`dns.query.send_tcp()` for the documentation of the other parameters, exceptions, and return type of this method. T)prepend_lengthbigN) r'r(r)r*r+lento_bytesrsendallr")r#r$r tcpmsgr-rrrsend_tcp<s  ricsTd}|dkr(||t|IdH}|dkrtd|t|}||}|dks|S)z|Read the specified number of bytes from stream. Keep trying until we either get the desired amount, or we hit EOF. r0rNEOF)recvr"EOFErrorre)r#countr rZr.rrr _read_exactlyUs rnc sZt|d|IdH}td|\}t|||IdH}t} tjj|||||d} | | fS)zRead a DNS message from a TCP socket. *sock*, a ``dns.asyncbackend.StreamSocket``. See :py:func:`dns.query.receive_tcp()` for the documentation of the other parameters, exceptions, and return type of this method. rcN!Hr3r4r2r5)rnstructunpackrr(r)r<) r#r r2r3r4r5ldatalr@rBrCrrr receive_tcpcsruc  s|} t|\} } |r|IdHt|} n%tj|}t|||}||f}| s0tj } | |t j d|||IdH} | 4IdH2}t || | IdHt|| ||j|j|IdH\}}|| |_||sjt|WdIdHS1IdHs|wYdS)aOReturn the response obtained after sending a query via TCP. *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. See :py:func:`dns.query.tcp()` for the documentation of the other parameters, exceptions, and return type of this method. Nr)r+r getpeernamerr(rMrNrrPrQrSr SOCK_STREAMrirur3rUrr>r)rGrHrIrrJrKr2r5r#rLr@rVr rWrrXrYrZrCrBrrrr_s:        0r_UT ssl_contextserver_hostnameverifyc  st|\} }|rt|}n4| durt| | dudg} tj|}t|||}||f}| s2tj} | |t j d|||| | IdH}|4IdH*}t |}t |||||||||| IdH}t}|| |_|WdIdHS1IdHsxwYdS)aReturn the response obtained after sending a query via TLS. *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. See :py:func:`dns.query.tls()` for the documentation of the other parameters, exceptions, and return type of this method. Ndotr)rrrr(rMrNrrPrQrSrrwr"r_r)rGrHrIrrJrKr2r5r#rLryrzr{rVr rWrrXrYrZr`end_timerrrtlssN         0r~resolverzdns.asyncresolver.ResolvercCs|dur ddl}|j}|Sr)dns.asyncresolver asyncresolverResolver)rr(rrr_maybe_get_resolvers rz /dns-queryclientz.httpx.AsyncClient|dns.quic.AsyncQuicConnectionpathpostbootstrap_addressr; http_versionc"sfztj|}Wn tyd}Ynw|}|dur@tj|r@|tjkr1d|d|| }n|tjkr@d|d|| }i}| durltj |}|j durUtdtj|j rd|j } |j |d<|j durl|j }|t jksx|t jkrts| durt| } |j dusJ| |j |IdH}tt|} |rt|tjjstd|dust|tjjsJt|| |||||||| | |d IdHStst|rt|tjstd |}d d i}|t jt jfv}|t j t jfv}tj!"}|durd}d }n|}|}|r t#|}n|$|||| || | |d}tj||| |d}|4IdHI}| rG|%d t&t'|d|(|j)||||d|IdH}nt*+|,d}|-} |(|j.||d| i|d|IdH}WdIdHn 1IdHsvwY|j/dks|j/dkrt|d|j/d|j0tj1j2|j0|j3|j4||d}!|j56|!_7|8|!st9|!S)aReturn the response obtained after sending a query via DNS-over-HTTPS. *client*, a ``httpx.AsyncClient``. If provided, the client to use for the query. Unlike the other dnspython async functions, a backend cannot be provided in this function because httpx always auto-detects the async backend. See :py:func:`dns.query.https()` for the documentation of the other parameters, exceptions, and return type of this method. Nzhttps://:z https://[z]:zno hostname in URL sni_hostnamez8client parameter must be a dns.quic.AsyncQuicConnection.)r{r connectionz-client parameter must be an httpx.AsyncClientacceptzapplication/dns-messager) local_addresshttp1http2r{ local_portrrr;)rrr{ transport)z content-typezcontent-length)headerscontent extensions=r()rparamsri+z responded with status code z Response body: rp):r(rMrN ValueError is_addressrrrurllibparseurlparsehostnamerr H3DEFAULTrr resolve_namerandomchoicelist addressesr'quicAsyncQuicConnection_http3r httpx AsyncClientr+H1H2rPrQrget_transport_classupdatestrrewait_forrbase64urlsafe_b64encoderstripdecodeget status_coderr)r<r3r4elapsed total_secondsrr>r)"rGrHrIrrJrKr2r5rrrr{rrr;rrurlrparsedanswersr@rh1h2rLrrrWr the_clientr`twirerCrrrhttpss                 *"  rrrc  stjjs tdtj|} | j}|dusJ| jdur | j}d|_ | }| r2tjj }tjj }ntj | \}}|4IdH}||| |dd4IdH_}| rT| }n| ||||}t|\}}||IdH}|4IdH$|||| |t|IdH}t|||WdIdHn 1IdHswYt}WdIdHn 1IdHswYtjj||j|j||d}WdIdHn 1IdHswYt||d|_||st|S)Nz DNS-over-HTTP3 is not available.rT) verify_mode server_nameh3rp)r(r have_quicr rrrrridr+ null_factoryfactories_for_backendconnectr make_streamsend_h3receiverr rrr)r<r3r4rr>r)rGrHrrIrrJrKr2r5r{rLrr url_partsrr@cfactorymfactorycontext the_managerthe_connectionstartr streamfinishrCrrrrsZ     ( (( rrc  stjjs td| dur| dur| } d|_|} |r'tjj}tjj}|}ntj| \}}|4IdH}||| | d4IdHV}|sM|||||}t |\}}| |IdH}|4IdH| | dIdH| t |IdH} WdIdHn 1IdHswYt}WdIdHn 1IdHswYtjj| |j|j||d}WdIdHn 1IdHswYt||d|_||st|S)aiReturn the response obtained after sending an asynchronous query via DNS-over-QUIC. *backend*, a ``dns.asyncbackend.Backend``, or ``None``. If ``None``, the default, then dnspython will use the default backend. See :py:func:`dns.query.quic()` for the documentation of the other parameters, exceptions, and return type of this method. zDNS-over-QUIC is not available.Nr)rrTrpr)r(rrr rr+rrrrrsendrrrr)r<r3r4rr>r)rGrHrIrrJrKr2r5rr{rLrrzr@rrrrrrr rrrCrrrrsV ( (( r txn_managerrZserialc Cs|jdj}|tjjk}|}|} |jtj k} | r1t tj j |} | | dt|IdHnt tj j|} tdt| | } | | |IdHtj|||| }d}d}d}|st|\}}|duso|durq||krq|}| rt|}| d|IdH\}}nt| d|IdH}td|\}t| ||IdH}tjj||j|jd||| |d}||}|V|j}|r]|jr|dur|j stj!"d WddSWddSWddS1swYdS) z'Given a socket, does the zone transfer.rNroFr9rcT)r3r4xfrorigintsig_ctxmultir2z missing TSIG)#questionrdtyper( rdatatypeIXFRfrom_wire_originr+typerrTrrPDatagramSocketr,r" StreamSocketrqpackrergrInboundrr:rnrrr)r<r3rUprocess_messagerhad_tsig exception FormError)rrZr8rrIr ris_ixfrrr@is_udpr]r^rhinbounddonerrCr[ mexpirationrwirersrtrrr _inbound_xfrHsb     "rlifetimeudp_modec  s|durtj|\}} ntj|} tj|} t| ||} ||f} | s+tj} t |\}}|j dj tj j kr|tjkr| | tjd| | t|IdH}|4IdH:zt|||| ||2z3dHW}qa6WWdIdHdStjjy|tjkrYnwWdIdHn 1IdHswY| | tjd| | t|IdH}|4IdHt|||| ||2z3dHW}q6WdIdHdS1IdHswYdS)amConduct an inbound transfer and apply it via a transaction from the txn_manager. *backend*, a ``dns.asyncbackend.Backend``, or ``None``. If ``None``, the default, then dnspython will use the default backend. See :py:func:`dns.query.inbound_xfr()` for the documentation of the other parameters, exceptions, and return type of this method. Nr)r(r make_queryextract_serial_from_queryrMrNrrPrQrrrrrr NEVERrSrrTr"rUseTCPONLYrw)rHrr8rrIrrJrKrrLrrrXrYr[r rZrrr inbound_xfrsZ        (  .r)N) NNFFNr0FFFN) NrFNrFFFFNNF) NrFNrFFFNNNF)NFNr0F)NrFNrFFNN) NrxNrFFNNNNT) NrNrFFTNTN) NrxNrFFNTNNN)X__doc__r contextlibrrrqr urllib.parsertypingrrrrrdns.asyncbackendr( dns.exceptiondns.inet dns.messagedns.namedns.quic dns.rdatatypedns.transactiondns.tsigdns.xfrdns._asyncbackendr dns.queryrr r r r r rrrrrssl ImportError dns._no_ssl_no_sslrrMlow_level_address_tuplerOrr"rPrr)r*bytesfloatintr/boolnameNametsigKeyrErBackendr\rrarirnrur_ SSLContextr~r AF_UNSPECrrrrr transactionTransactionManagerSocketrrrrrrrsp 4            @      E      G    !   @      F       5      C      F :