관리-도구
편집 파일: asyncquery.cpython-38.pyc
U /?�f�h � @ sf 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"r�ddl$Z$e j%j&Z'dd� Z(d6dd �Z)d7e j*j+e e j,j-e.f ee e/ e e0e/f d �d d �Z1d8e j*j+e e e e/ e2e2e ee j3j4e j5j6f e e. e2e2e2e e j,j- ed� dd�Z7d9e j,j-e8e e/ e0e e8 e0e2e2e2e2e e j*j+ e e j*j9 e2e j,j-d�dd�Z:d:e j,j-e8e e/ e0e e8 e0e2e2e2e e j*j+ e e j*j; e e j*j9 e2e e j,j-e2f d�dd�Z<d;e j*j;e e j,j-e.f e e/ e e0e/f d�dd�Z=d d � Z>d<e j*j;e e/ e2e ee j3j4e j5j6f e e. e2e e j,j-e/f d �d d �Z?d=e j,j-e8e e/ e0e e8 e0e2e2e e j*j; e e j*j9 e j,j-d!� d"d#�Z@d>e j,j-e8e e/ e0e e8 e0e2e2e e j*j; e e j*j9 e e#jA e e8 e e2e8f e j,j-d&�d'd(�ZBdd)ddd d dd*d%d%ddejCf e j,j-e8e e/ e0e e8 e0e2e2e d+ e8e2e e2e8f e e8 e d, e e0 e j,j-d-�d.d/�ZDdddddde jEdfe8e jFjGe e j,j- e0e e/ e e/ e e8 e0e e e j*j9 dd0� d1d2�ZHd?e j,j-e8e e/ e0e e8 e0e2e2e e jIjJ e e2e8f e e j*j9 e e8 e j,j-d3� d4d5�ZIdS )@zTalk to a DNS server.� N)�Any�Dict�Optional�Tuple�Union)� NullContext) � BadResponse�NoDOH�NoDOQ�UDPMode�_compute_times�_make_dot_ssl_context�_matches_destination� _remaining�have_doh�sslc C sN |s|rF|d kr>| t jkr d}n | t jkr0d}ntd| � ���||fS d S d S )Nz0.0.0.0z::zunknown address family )�socket�AF_INET�AF_INET6�NotImplementedError)�af�address�port� r �</opt/hc_python/lib/python3.8/site-packages/dns/asyncquery.py� _source_tuple: s r c C s* | d k r"|st � � }t| | d�S d S d S )Nr )�time�max)� expiration�nowr r r �_timeoutJ s r )�sock�what� destinationr �returnc � s@ 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� ) r! r# r �ignore_unexpected�one_rr_per_rrset�keyring� request_mac�ignore_trailing�raise_on_truncation� ignore_errors�queryr$ c � s� d} | � dt|��I dH \} } t| j| ||�s0qt�� } z tjj| |||||d�}W nj tjjk r� } z*| r�| dk r�| � |�� �s�W Y � qn� W 5 d}~X Y n t k r� | r�Y qn� Y nX | r�| dk r�| � |�s�q|| | 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� Truncated� is_response� Exception)r! r# r r/ r0 r1 r2 r3 r4 r5 r6 �wire� from_address� received_time�r�er r r � receive_udpq sF �� �� � rC �5 )�q�where�timeoutr �source� source_portr/ r0 r3 r4 r! �backendr5 r$ c � s | � � } t|�\}}tj�|�}t||f|�}| r<t| �}nJ| sJtj�� } t |||�}| � � rh||f}nd}| � |t j d||�I dH }|4 I dH �t}t|| ||�I dH t|||||| j| j|| | | � I dH \}}}|| |_| s�| �|�s�t�|W 5 Q I dH R � S Q I dH R X 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 Z datagram_connection_required� make_socketr � SOCK_DGRAMr- rC r1 �macr r< r )rE rF rG r rH rI r/ r0 r3 r4 r! rJ r5 r> � begin_timer r r# �cm�stuple�dtuple�srA r@ �_r r r �udp� s@ � rY )rE rF rG r rH rI r/ r0 r3 �udp_sock�tcp_sockrJ r5 r$ c � sv z0t | ||||||||d| | | � I dH } | dfW S tjjk rp t| |||||||| | � I dH } | df Y S X dS )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)rY r& r' r; �tcp)rE rF rG r rH rI r/ r0 r3 rZ r[ rJ r5 �responser r r �udp_with_fallback� s> #� � r^ )r! r" r r$ c � s\ t |tjj�r |jdd�}nt|��dd�| }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. T)Zprepend_length� �bigN) r% r& r' r( r) �len�to_bytesr �sendallr )r! r" r �tcpmsgr+ r r r �send_tcp4 s re c � sH d}|dkrD| � |t|��I dH }|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. r. r N)�recvr �EOFErrorra )r! �countr rW r, r r r � _read_exactlyM s ri )r! r r0 r1 r2 r3 r$ c � sX 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. r_ N�!H�r1 r2 r0 r3 )ri �struct�unpackr r& r' r: ) r! r r0 r1 r2 r3 �ldata�lr> r@ rA r r r � receive_tcp[ s �rp ) rE rF rG r rH rI r0 r3 r! rJ r$ c � s� | � � } t|�\} } |r0|�� I dH t|�} nJtj�|�}t|||�}||f}| s^tj� � } | � |t j d|||�I dH } | 4 I dH �b}t || | �I dH t|| || j| j|�I dH \}}|| |_| �|�s�t�|W 5 Q I dH R � S Q I dH R X 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& rK rL r rN rO rP r � SOCK_STREAMre rp r1 rR r r< r )rE rF rG r rH rI r0 r3 r! rJ r> rS r rT r rU rV rW rA r@ r r r r\ y s@ � � r\ �U T)rE rF rG r rH rI r0 r3 r! rJ � ssl_context�server_hostname�verifyr$ c � s� t |�\} }|rt|�}n`| dkr,t| | �} tj�|�}t|||�}||f}| sZtj�� } | � |t j d|||| | �I dH }|4 I dH �R}t |�}t | ||||||||| � I dH }t�� }|| |_|W 5 Q I dH R � S Q I dH R X 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. Nr )r r r r& rK rL r rN rO rP r rr r r\ r )rE rF rG r rH rI r0 r3 r! rJ rt ru rv rS r rT r rU rV rW r] Zend_timer r r �tls� sJ � � rw i� z /dns-queryzhttpx.AsyncClientzdns.asyncresolver.Resolver)rE rF rG r rH rI r0 r3 �client�path�postrv �bootstrap_address�resolverr9 r$ c � s t st�|r t|tj�s td��| �� }ztj� |�}W n tk rP d}Y nX d}ddi}|dk r�tj� |�r�|t j kr�d� ||| �}q�|t jkr�d� ||| �}n|}tj�� }|dkr�d}d}n|}|}|�� |dd| || | |d �}|r�t|�}ntjdd| |d �}|4 I dH �~}| �rV|�dtt|��d �� |�|j|||d �|�I dH }n:t�|��d �}|�� }|�|j ||d|id�|�I dH }W 5 Q I dH R X |j dk �s�|j dk�r�td� ||j |j ���tj j |j | j!| j"||d�}|j#�$� |_%| �&|��st'�|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�http2rv � local_portr{ r| r9 )r r� rv � transport)z content-typezcontent-length)�headers�content� =r& )r� �params�� i+ z4{} responded with status code {} Response body: {!r}rk )(r r r% �httpxZ AsyncClient� ValueErrorr) r& rK rL Z is_addressr r �formatr rN rO Zget_transport_classr �update�strra �wait_forrz �base64�urlsafe_b64encode�rstrip�decode�get� status_coder� r' r: r1 r2 �elapsed� total_secondsr r< r ) rE rF rG r rH rI r0 r3 rx ry rz rv r{ r| r9 r> r r� r� �urlrJ r~ r� rT Z the_clientr] ZtwirerA r r r �https� s� � � �� � � ��� r� ) rF � txn_managerr6 r rG �lifetimerH rI �udp_moderJ r$ c # � s� |dkrt j�|�\}} n t j�|�} |jd j} | t jjk} |�� } |� � }t j � | �}t |||�}| |f}t |�\}}d}|�r�d}| r�|tjkr�tj}d}n tj}d}| s�t j�� } | �||d||t|��I dH }|4 I dH ��� |�r|�||t|��I dH n&t�dt|��| }|�||�I dH t j� || | |���p}d}d}|�s�t |�\}} | dk�s||dk �r�| |k�r�|} |�r�t | |f|�} t| �}|� d|�I dH \} } t || | d��r��q��q�n2t |d| �I dH } t�!d| �\}!t ||!| �I dH } | t jjk} t j"j#| |j$|j%d| || | d�}"z|�&|"�}W nF t jj'k �r� |�sZt(�|tj)k�rh� d}d}tj}Y �qLY nX |"j*}�qL|�s�|j$�r�|"j+�s�t j,�-d ��W 5 Q R X W 5 Q I dH R X q~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 TFrj r7 r_ )r1 r2 �xfr�origin�tsig_ctx�multir0 z missing TSIG).r& r� Z make_queryZextract_serial_from_query�question�rdtypeZ rdatatypeZIXFRZfrom_wire_originr) rK rL r r r �NEVERr rQ rr rN rO rP r r* rl �packra rc ZInboundrM r8 r ri rm r' r: r1 rR Zprocess_messageZUseTCP�AssertionErrorZONLYr� Zhad_tsig� exceptionZ FormError)#rF r� r6 r rG r� rH rI r� rJ �serialr� Zis_ixfrr� r> r rU rV rX r �retryZ sock_typeZis_udprW rd Zinbound�doner� Z mexpirationr# Zrwirer? rn ro rA r r r � inbound_xfrl s� � �� � � r� ) rE rF rG r rH rI r0 r3 � connectionrv rJ ru r$ c � sJ t jjstd��d| _| �� } |r8t jj} t jj}|}nt j�| �\} }| � 4 I dH ��}||| | d�4 I dH ��}|s�|�||||�}t |�\}}|� |�I dH }|4 I dH �, |� | d�I dH |� t |��I dH } W 5 Q I dH R X t�� }W 5 Q I dH R X t jj| | j| j||d�}W 5 Q I dH R X t|| d�|_| �|��sFt�|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_modeZ server_nameTrk g )r& �quicZ have_quicr �idr) Z null_factoryZfactories_for_backend�connectr Z make_stream�sendZreceiver r r' r: r1 r2 r r< r )rE rF rG r rH rI r0 r3 r� rv rJ ru r> ZcfactoryZmfactoryZthe_connection�contextZ the_manager�startr �stream�finishrA r r r r� � sF � $� r� )N)N) NNFFNr. FFFN) NrD Nr FFFFNNF) NrD Nr FFFNNNF)N)NFNr. F)NrD Nr FFNN) Nrs Nr FFNNNNT) Nrs Nr FFNTNN)K�__doc__r� � contextlibr rl r �typingr r r r r Zdns.asyncbackendr& Z dns.exceptionZdns.inetZ dns.messageZdns.nameZdns.quicZ dns.rcodeZdns.rdataclassZ dns.rdatatypeZdns.transactionZdns._asyncbackendr Z dns.queryr r r r r r r r r r r� rK Zlow_level_address_tuplerM r r rN ZDatagramSocketr' r( �bytes�float�intr- �bool�name�NameZtsig�KeyrC r� �BackendrY Z StreamSocketr^ re ri rp r\ � SSLContextrw � AF_UNSPECr� r� Z transactionZTransactionManagerr� r� ZAsyncQuicConnectionr r r r �<module> s 0 � � � � @ � � E � � G � � �� ! � � ; � � I� � x� � h � �