File: //opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/__pycache__/resolver.cpython-310.pyc
o
;j�& � @ s� U d dl Z d dlZd dlZd dlmZmZmZ ddlmZm Z dZ
z d dl Z e e j
d�ZW n
ey: dZ dZY nw ejejB ZejejB ZejZe ed�rTeejM ZG d d
� d
e�ZG d d � d e�ZG d
d� d�Ze eeB Z erueneZ e e d< dS )� N)�Any�Final�Optional� )�AbstractResolver�
ResolveResult)�ThreadedResolver�
AsyncResolver�DefaultResolver� getaddrinfoF�AI_MASKc
@ s\ e Zd ZdZddejdB ddfdd�Zdejfde d e
d
ej de e
fd d �Zdd
d�ZdS )r z�Threaded resolver.
Uses an Executor for synchronous getaddrinfo() calls.
concurrent.futures.ThreadPoolExecutor is used by default.
N�loop�returnc C s |pt �� | _d S �N)�asyncio�get_running_loop�_loop)�selfr
� r �D/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/resolver.py�__init__" s zThreadedResolver.__init__r �host�port�familyc � s� �| j j||tj|td�I d H }g }|D ]I\}}}}}|tjkrEt|�dk r'q|d r<| j �|t�I d H \} }
t |
�}n|d d� \} }n |tj
ksLJ �|\} }|� t || |||t
d�� q|S )N)�typer �flags� � ��hostnamer r r �protor )r r �socket� SOCK_STREAM�_AI_ADDRCONFIG�AF_INET6�len� getnameinfo�_NAME_SOCKET_FLAGS�int�AF_INET�appendr �_NUMERIC_SOCKET_FLAGS) r r r r �infos�hosts�_r �address�
resolved_hostZ_portr r r �resolve% s@ � �
�
�� zThreadedResolver.resolvec � s �d S r r �r r r r �closeQ s �zThreadedResolver.closer �r N)�__name__�
__module__� __qualname__�__doc__r �AbstractEventLoopr r! r) �strr( �
AddressFamily�listr r1 r3 r r r r r s ����
�,r c @ s� e Zd ZdZ ddejdB dededdfdd�Zd ej fd
e
d e d ej de
e fd
d�Zd ej fd
e
d e d e de
ee
ef fdd�Zddd�ZdS )r z9Use the `aiodns` package to make asynchronous DNS lookupsNr
�args�kwargsr c O sx t d u rtd��|p
t�� | _d | _|s|r!t j|i |��| _d S t� | _| j� | | j�| _t
| jd�s:| j | _ d S d S )Nz Resolver requires aiodns library�
gethostbyname)
�aiodns� RuntimeErrorr r r �_manager� DNSResolver� _resolver�_DNSResolverManager� get_resolver�hasattr�_resolve_with_queryr1 )r r
r= r> r r r r X s �zAsyncResolver.__init__r r r r c � sB �z| j j||tj|td�I d H }W n" tjjy4 } zt|j �dkr(|j d nd}t
d |�|�d }~ww g }|j D ]]}|j } |j
tjkrvt| �dkrj| d rj| j �| d �d�g| dd � �R t�I d H }
|
j} n | d �d�} | d }n|j
tjks~J �| d �d�} | d }|�t|| ||j
dtd�� q:|s�t
d d��|S )N)r r r r r �DNS lookup failedr r �asciir )rD r r! r" r# r@ �error�DNSErrorr% r= �OSErrorZnodes�addrr r$ r& �decoder' �noder) r* r r+ ) r r r r �resp�exc�msgr- rP r/ �resultr0 r r r r1 q sT �� ��
�
��
zAsyncResolver.resolvec
� s� �|t jkrdnd}z | j�||�I d H }W n" tjjy8 } zt|j�dkr,|jd nd}t d |�|�d }~ww g }|D ]} |�
|| j ||dt j d�� q=|sUt d d��|S )NZAAAA�Ar rI r r )
r! r$ rD �queryr@ rK rL r% r= rM r* r �AI_NUMERICHOST)
r r r r ZqtyperQ rR rS r- Zrrr r r rH � s. � ����
z!AsyncResolver._resolve_with_queryc � sF �| j r| j �| | j� d | _ d | _d S | jd ur | j�� d | _d S r )rB �release_resolverr rD �cancelr2 r r r r3 � s �
zAsyncResolver.closer r4 )r5 r6 r7 r8 r r9 r r r! r) r: r( r; r< r r1 �dictrH r3 r r r r r U sB ����
�����
�3����
� r c sn e Zd ZU dZdZed ed< d� fdd� Zddd�Zd d
d e j
dd fd
d�Z d d
d e j
ddfdd�Z � Z
S )rE z�Manager for aiodns.DNSResolver objects.
This class manages shared aiodns.DNSResolver instances
with no custom arguments across different event loops.
N� _instancer c s( | j d u rt� �| �| _ | j �� | j S r )r[ �super�__new__�_init)�cls�� __class__r r r] � s
z_DNSResolverManager.__new__c C s t �� | _d S r )�weakref�WeakKeyDictionary�
_loop_datar2 r r r r^ � s �z_DNSResolverManager._init�clientr r
zaiodns.DNSResolverc C sJ || j vrtj|d�}t�� }||f| j |< n| j | \}}|�|� |S )a& Get or create the shared aiodns.DNSResolver instance for a specific event loop.
Args:
client: The AsyncResolver instance requesting the resolver.
This is required to track resolver usage.
loop: The event loop to use for the resolver.
)r
)rd r@ rC rb �WeakSet�add)r re r
�resolver�
client_setr r r rF � s
z _DNSResolverManager.get_resolverc C sN | j �|�}|du r dS |\}}|�|� |s%|dur |�� | j |= dS dS )z�Release the resolver for an AsyncResolver client when it's closed.
Args:
client: The AsyncResolver instance to release.
loop: The event loop the resolver was using.
N)rd �get�discardrY )r re r
Zcurrent_loop_datarh ri r r r rX � s
�z$_DNSResolverManager.release_resolver)r rE r4 )r5 r6 r7 r8 r[ r �__annotations__r] r^ r r9 rF rX �
__classcell__r r r` r rE � s&
��
����rE r
) r r! rb �typingr r r �abcr r �__all__r@ rG rC Zaiodns_default� ImportErrorrW �AI_NUMERICSERVr+ �NI_NUMERICHOST�NI_NUMERICSERVr' �
AI_ADDRCONFIGr# r r r rE r Z _DefaultTyper
rl r r r r �<module> s.
�
:x C