File: //home/arjun/projects/env/lib/python3.10/site-packages/dns/__pycache__/asyncquery.cpython-310.pyc
o
!we�e � ! @ s� d Z ddlZddlZddlZddlZddlZddlmZmZm Z m
Z
m Z ddl Z
ddlZ
ddlZ
ddlZ
ddlZ
ddlZ
ddlZ
ddlZ
ddlZ
ddlZ
ddlmZ ddlmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z# e"rpddl$Z$e
j%j&Z'dd� Z(dMdd �Z) dMd
e
j*j+d e e
j,j-e.f d ed
e e/ de
e0e/f f
dd�Z1 dNd
e
j*j+d e e d
e e/ de2de2de ee
j3j4e
j5j6f de e. de2de2defdd�Z7 dOd e
j,j-d e8d e e/ d e0d e e8 d!e0de2de2de2de2d
e e
j*j+ d"e e
j*j9 de
j,j-fd#d$�Z: dPd e
j,j-d e8d e e/ d e0d e e8 d!e0de2de2de2d%e e
j*j+ d&e e
j*j; d"e e
j*j9 de
e
j,j-e2f fd'd(�Z< dMd
e
j*j;d e e
j,j-e.f d
e e/ de
e0e/f fd)d*�Z=d+d,� Z> dQd
e
j*j;d
e e/ de2de ee
j3j4e
j5j6f de e. de2de
e
j,j-e/f fd-d.�Z? dRd e
j,j-d e8d e e/ d e0d e e8 d!e0de2de2d
e e
j*j; d"e e
j*j9 de
j,j-fd/d0�Z@ 1 dSd e
j,j-d e8d e e/ d e0d e e8 d!e0de2de2d
e e
j*j; d"e e
j*j9 d2e e#jA d3e e8 de
j,j-fd4d5�ZBdd6dddddd7d8d8ddejCf
d e
j,j-d e8d e e/ d e0d e e8 d!e0de2de2d9e d: d;e8d<e2d=e e2e8f d>e e8 d?e d@ dAe e0 de
j,j-f dBdC�ZDdddddde jEdfd e8dDe
jFjGdEe e
j,j- d e0d e e/ dFe e/ d e e8 d!e0dGe d"e e
j*j9 ddfdHdI�ZH 1 8 dTd e
j,j-d e8d e e/ d e0d e e8 d!e0de2de2dJe e
jIjJ d=e e2e8f d"e e
j*j9 d3e e8 de
j,j-fdKdL�ZIdS )UzTalk to a DNS server.� N)�Any�Dict�Optional�Tuple�Union)� NullContext)
� BadResponse�NoDOH�NoDOQ�UDPMode�_compute_times� _have_http2�_matches_destination�
_remaining�have_doh�sslc C sV |s|r)|d u r%| t jkrd}||fS | t jkr d}||fS td| � ���||fS d S )Nz0.0.0.0z::zunknown address family )�socket�AF_INET�AF_INET6�NotImplementedError)�af�address�port� r �G/home/arjun/projects/env/lib/python3.10/site-packages/dns/asyncquery.py�
_source_tuple: s
��r c C s& | d ur|s
t � � }t| | d�S d S )Nr )�time�max)�
expiration�nowr r r �_timeoutJ s
r �sock�what� destinationr �returnc � sB �t |tjj�r |�� }t�� }| �||t||��I dH }||fS )a� Send 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)�
isinstance�dns�message�Message�to_wirer �sendtor )r! r" r# r � sent_time�nr r r �send_udpS s �r- F� �ignore_unexpected�one_rr_per_rrset�keyring� request_mac�ignore_trailing�raise_on_truncationc
� s` �d} | � dt|��I dH \} }
t| j|
||�rnqt�� } tjj| |||||d�} | | |
fS )aj Read 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.
r. � � N)r1 r2 r0 r3 r4 )�recvfromr r �familyr r& r' � from_wire)
r! r# r r/ r0 r1 r2 r3 r4 �wire� from_address�
received_time�rr r r � receive_udpq s& �
���
r> �5 �q�where�timeoutr �source� source_port�backendc � s �| � � } t|�\}
}tj�|�}t||f|�}|
r t|
�}n%| s&tj�� } t |||�}| �
� r5||f}nd}| � |t j
d||�I dH }|4 I dH �7}t|| ||�I dH t|||||| j| j|| � I dH \}}}||
|_| �|�sut�|W d �I dH S 1 I dH s�w Y dS )aF Return 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) r r& �inet�af_for_address�_lltupler � asyncbackend�get_default_backendr � datagram_connection_required� make_socketr �
SOCK_DGRAMr- r> r1 �macr � is_responser )r@ rA rB r rC rD r/ r0 r3 r4 r! rE r: �
begin_timer r r# �cm�stuple�dtuple�sr= r< �_r r r �udp� s@ �
�
0�rV �udp_sock�tcp_sockc
� sp �zt | ||||||||d| | � I dH } | dfW S tjjy7 t| ||||||||
| �
I dH } | df Y S w )a� Return 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)rV r& r' � Truncated�tcp)
r@ rA rB r rC rD r/ r0 r3 rW rX rE �responser r r �udp_with_fallback� s@ �"
�
� �r\ c � sb �t |tjj�r
|�� }n|}t|�}t�d|�| }t�� }| � |t
||��I dH t|�|fS )z�Send 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.
�!HN) r% r& r' r( r) �len�struct�packr �sendallr )r! r" r r: �l�tcpmsgr+ r r r �send_tcp s �
rd c � sP �d}|dkr&| � |t|��I dH }|dkrt�|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.
r. r N)�recvr �EOFErrorr^ )r! �countr rT r, r r r �
_read_exactly4 s � �rh c � sZ �t | d|�I dH }t�d|�\}t | ||�I dH }t�� } tjj|||||d�}
|
| fS )z�Read 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.
� Nr] �r1 r2 r0 r3 )rh r_ �unpackr r& r' r9 ) r! r r0 r1 r2 r3 �ldatarb r: r< r= r r r � receive_tcpB s ��rm c
� s �| � � }
t|�\} } |r|�� I dH t|�}
n%tj�|�}t|||�}||f}| s0tj� � } | �
|t j d|||�I dH }
|
4 I dH �2}t
||
| �I dH t|| || j| j|�I dH \}}|| |_| �|�sjt�|W d �I dH S 1 I dH s|w Y dS )aO Return 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 � getpeernamer r& rF rG r rI rJ rL r � SOCK_STREAMrd rm r1 rN r rO r )r@ rA rB r rC rD r0 r3 r! rE r: rP r rQ r rR rS rT r= r< r r r rZ ` s0 �
��
0�rZ �U � ssl_context�server_hostnamec � s �t |�\} }
|rt|�}n;|
du r"t�� }
tjj|
_| du r"d|
_tj �
|�}t |||�}||f}| s9tj �
� } | �|tjd||||
| �I dH }|4 I dH �*}t|
�}t| ||||||||| �
I dH }t�� }|| |_|W d �I dH S 1 I dH sw Y dS )a� Return 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.
NFr )r r r �create_default_context�
TLSVersion�TLSv1_2�minimum_version�check_hostnamer& rF rG r rI rJ rL r ro r rZ r )r@ rA rB r rC rD r0 r3 r! rE rq rr rP r rQ r rR rS rT r[ �end_timer r r �tls� sT �
�
�
0�ry i� z
/dns-queryT�clientzhttpx.AsyncClient�path�post�verify�bootstrap_address�resolverzdns.asyncresolver.Resolverr8 c
� s �t st�|rt|tj�std��| �� }ztj� |�}W n ty( d}Y nw d}ddi}|durStj�
|�rS|t j krFd�
||| �}n|t jkrRd�
||| �}n|}tj�� }|du rcd}d}n|}|}|�� |dt| || |
|d �}|r|t|�}n tjdt| |d
�}|4 I dH �F}|
r�|�dtt|��d �� |�|j|||d �|�I dH }n t�|��d
�}|� � }|�|j ||d|id�|�I dH }W d �I dH n 1 I dH s�w Y |j dk s�|j dkr�td�
||j |j ���tj j!|j | j"| j#||d�}|j$�%� |_&| �'|��s
t(�|S )a� Return 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.
z.session parameter must be an httpx.AsyncClientN�acceptzapplication/dns-messagezhttps://{}:{}{}zhttps://[{}]:{}{}r T)�
local_address�http1�http2r} �
local_portr~ r r8 )r� r� r} � transport)z content-typezcontent-length)�headers�content� =r& )r� �params�� i+ z4{} responded with status code {}
Response body: {!r}rj ))r r r% �httpx� AsyncClient�
ValueErrorr) r& rF rG �
is_addressr r �formatr rI rJ �get_transport_classr
r �update�strr^ �wait_forr| �base64�urlsafe_b64encode�rstrip�decode�get� status_coder� r' r9 r1 r2 �elapsed�
total_secondsr rO r ) r@ rA rB r rC rD r0 r3 rz r{ r| r} r~ r r8 r: r r� r� �urlrE r� r� rQ �
the_clientr[ �twirer= r r r �https� s� � �
�
�
�
�� � �(�
��� r� � txn_manager�query�lifetime�udp_modec
# � s� �|du rt j�|�\}}
nt j�|�}
|jd j} | t jjk} |�� }
|� � }t j
� | �}t |||�}| |f}t
|�\}}d}|�rud}| rR|tjkrRtj}d}ntj}d}| s^t j�� } | �||d||t|��I dH }|4 I dH �� |r�|�||t|��I dH nt�dt|��| }|�||�I dH t j� || |
|���}d}d}|�s:t
|�\}} | du s�|dur�| |kr�|} |r�t | |f|�} t| �}|� d|�I dH \} } t || | d�r�nq�nt |d| �I dH } t�!d| �\}!t ||!| �I dH } | t jjk} t j"j#| |j$|j%d|
|| | d�}"z|�&|"�}W n t jj'�y4 |�s$J �|tj(k�r+� d}d}tj}Y q�w |"j)}|r�|�sK|j$�rK|"j*�sKt j+�,d ��W d � n 1 �sVw Y W d �I dH n 1 I dH �slw Y |sCdS dS )
am Conduct 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 TFr] r6 ri )r1 r2 �xfr�origin�tsig_ctx�multir0 z missing TSIG)-r& r� �
make_query�extract_serial_from_query�question�rdtype� rdatatype�IXFR�from_wire_originr) rF rG r r r �NEVERr rM ro rI rJ rL r r* r_ r` r^ ra �InboundrH r7 r rh rk r' r9 r1 rN �process_message�UseTCP�ONLYr� �had_tsig� exception� FormError)#rA r� r� r rB r� rC rD r� rE �serialr� �is_ixfrr� r: r rR rS rU r �retry� sock_type�is_udprT rc �inbound�doner� � mexpirationr# �rwirer; rl rb r= r r r � inbound_xfrV s� �
� �� �
��) � �*� �r� �
connectionc � s� �t jjs td��d| _| �� } |r t jj}
t jj}|}nt j�|
�\}
}|
� 4 I dH ��}||| | d�4 I dH �V}|sC|�||||�}t |�\}}|�
|�I dH }|4 I dH � |� | d�I dH |� t
|��I dH } W d �I dH n 1 I dH s{w Y t�� }W d �I dH n 1 I dH s�w Y t jj| | j| j||d�}W d �I dH n 1 I dH s�w Y t|| d�|_| �|�s�t�|S )ai Return 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.
z DNS-over-QUIC is not available.r N)� verify_mode� server_nameTrj g )r& �quic� have_quicr
�idr) � null_factory�factories_for_backend�connectr � make_stream�send�receiver r r' r9 r1 r2 r rO r )r@ rA rB r rC rD r0 r3 r� r} rE rr r: �cfactory�mfactory�the_connection�context� the_manager�startr �stream�finishr= r r r r� � sJ �� (�
(� �(�
r� )N)NNFFNr. FF)
Nr? Nr FFFFNN)
Nr? Nr FFFNNN)NFNr. F)Nr? Nr FFNN)
Nrp Nr FFNNNN)
Nrp Nr FFNTNN)K�__doc__r� �
contextlibr r_ r �typingr r r r r �dns.asyncbackendr& �
dns.exception�dns.inet� dns.message�dns.name�dns.quic� dns.rcode�dns.rdataclass�
dns.rdatatype�dns.transaction�dns._asyncbackendr � dns.queryr r r
r r r
r r r r r� rF �low_level_address_tuplerH r r rI �DatagramSocketr' r( �bytes�float�intr- �bool�name�Name�tsig�Keyr> r� �BackendrV � StreamSocketr\ rd rh rm rZ �
SSLContextry � AF_UNSPECr� r� � transaction�TransactionManagerr� r� �AsyncQuicConnectionr r r r �<module> s� 0
�����
� ��������� �
�,��������� �
�
�
�
�@��������� �
�
�
�
�E����
��������
�!���������
�
�
�;���������
�
� � �
�L��������� �
� �
�
���
�x���
������ �
�
�h���������
�
�
� �
�