a
Me # @ s" d Z ddlZddlZddlZddlZddlZddlZddlZej dkZ
dd Z G dd dZ dd Z
G d
d d ejjZG d d
d
ejjZejdrddlZddlZddlZddlZejZejjjZddl m Z m Z m Z G dd deZ G dd dej!Z"nejj#Z"G dd dejj$Z$dS )z asyncio library query support Nwin32c C s* z
t W S ty$ t Y S 0 d S N)asyncioZget_running_loopAttributeErrorZget_event_loop r r 8/usr/lib/python3.9/site-packages/dns/_asyncio_backend.py_get_running_loop s
r c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )_DatagramProtocolc C s d | _ d | _d S r ) transportrecvfromselfr r r __init__ s z_DatagramProtocol.__init__c C s
|| _ d S r )r
)r
r
r r r connection_made s z!_DatagramProtocol.connection_madec C s$ | j r | j s | j ||f d S r )r doneZ
set_result)r
dataZaddrr r r datagram_received s z#_DatagramProtocol.datagram_receivedc C s | j r | j s | j | d S r )r r
set_exception)r
excr r r error_received$ s z _DatagramProtocol.error_receivedc
C sd | j r`| j s`|d u rTztW q` tyP } z| j | W Y d }~q`d }~0 0 n | j | d S r )r r EOFErrorr )r
r er r r connection_lost( s $z!_DatagramProtocol.connection_lostc C s | j d S r )r
closer r r r r 3 s z_DatagramProtocol.closeN) __name__
__module__ __qualname__r r r r r r r r r r r s r c sP |d urBzt | |I d H W S t jy> tjj|dY qL0 n
| I d H S d S )N)timeout)r wait_for TimeoutErrordns exceptionZTimeout) awaitabler r r r _maybe_wait_for7 s r# c sL e Zd Z fddZdd Zdd Zdd Zd d
Zd d Zd
d Z Z
S )DatagramSocketc s t | || _|| _d S r )superr r
protocol)r
familyr
r& __class__r r r B s zDatagramSocket.__init__c s | j || t|S r )r
sendtolen)r
what destinationr r r r r* G s zDatagramSocket.sendtoc sD t }z*|| j_t||I d H | W d | j_S d | j_0 d S r )r Z
create_futurer& r r# result)r
sizer r r r r r L s
zDatagramSocket.recvfromc s | j d S r )r& r r r r r r W s zDatagramSocket.closec s | j dS NZpeernamer
get_extra_infor r r r getpeernameZ s zDatagramSocket.getpeernamec s | j dS NZsocknamer1 r r r r getsockname] s zDatagramSocket.getsocknamec s t d S r NotImplementedErrorr
r r r r getpeercert` s zDatagramSocket.getpeercert) r r r r r* r r r3 r5 r9
__classcell__r r r( r r$ A s r$ c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dS ) StreamSocketc C s || _ || _|| _d S r )r' readerwriter)r
afr< r= r r r r e s zStreamSocket.__init__c s" | j | t| j |I d H S r )r= writer# Zdrain)r
r, r r r r sendallj s zStreamSocket.sendallc s t | j||I d H S r )r# r< read)r
r/ r r r r recvn s zStreamSocket.recvc s | j d S r )r= r r r r r r q s zStreamSocket.closec s | j dS r0 r= r2 r r r r r3 t s zStreamSocket.getpeernamec s | j dS r4 rC r r r r r5 w s zStreamSocket.getsocknamec s | j dS )NZpeercertrC r8 r r r r9 z s zStreamSocket.getpeercertN)
r r r r r@ rB r r3 r5 r9 r r r r r; d s r; Zdoh)_compute_times _expiration_for_this_attempt
_remainingc s8 e Zd Z fddZd
ddZd ddZdd Z ZS ) _NetworkBackendc s6 t || _|| _|| _|| _|dkr2tdd S )Nr z9the asyncio transport for HTTPX cannot set the local port)r% r Z _local_port _resolver_bootstrap_address_familyr7 )r
resolver
local_portbootstrap_addressr' r( r r r s
z_NetworkBackend.__init__Nc s g }t |\}}tj|r(|| nV| jd ur@|| j n>t|}| j} |r^tj|} | j j
|| |dI d H }
|
}|D ]x} z`t d|} t| }t
|& t
j| ||dI d H }
W d n1 s0 Y t|
W S ty Y q0 qtjd S )N)r' Zlifetimeg @)Z remote_hostZ remote_portZ
local_host)rD r inetZ
is_addressappendrI rF rJ Zaf_for_addressrH resolve_name addressesrE anyioZ
fail_after connect_tcp_CoreAnyIOStream ExceptionhttpcoreZ ConnectError)r
hostportr Z
local_addresssocket_optionsrQ _Z
expirationr' ZanswersZaddressZattempt_expirationstreamr r r rS s8
* z_NetworkBackend.connect_tcpc s t d S r r6 )r
pathr rY r r r connect_unix_socket s z#_NetworkBackend.connect_unix_socketc s t |I d H d S r )rR sleep)r
Zsecondsr r r r^ s z_NetworkBackend.sleep)N)N)r r r r rS r] r^ r: r r r( r rG s
"
rG c s* e Zd Zdddejd fdd
Z ZS )_HTTPTransportr N)rL rM rK r' c sB |d u rdd l }|j }t j|i | t||||| j_d S )Nr )Zdns.asyncresolverZ
asyncresolverZResolverr% r rG Z_poolZ_network_backend)r
rL rM rK r' argskwargsr r( r r r s
z_HTTPTransport.__init__)r r r socketZ AF_UNSPECr r: r r r( r r_ s
r_ c @ s> e Zd Zdd ZdddZdd Zd d
Zd d Zd
d ZdS )Backendc C s dS )Nr r r r r r name s z Backend.namer Nc
s t } |tjkrVtr,|d u r,tj|df}| jt||||dI d H \}
} t ||
| S |tj
kr|d u rpt dt t
j|d |d |||||d|I d H \} }
t|| |
S tdd| d S )Nr )r' protoZ remote_addrz'destination required for stream sockets )Zsslr' re Z
local_addrserver_hostnamezunsupported socket ztype )r rb Z
SOCK_DGRAM _is_win32r rN Z
any_for_afZcreate_datagram_endpointr r$ Z SOCK_STREAM
ValueErrorr# r Zopen_connectionr; r7 )r
r> Zsocktypere sourcer- r Z ssl_contextrg Zloopr
r& rwr r r make_socket s>
zBackend.make_socketc s t |I d H d S r )r r^ )r
intervalr r r r^ s z
Backend.sleepc C s dS )NFr r r r r datagram_connection_required s z$Backend.datagram_connection_requiredc C s t S r )r_ r r r r get_transport_class s zBackend.get_transport_classc s t ||I d H S r )r# )r
r" r r r r r s zBackend.wait_for)r NNNNN) r r r rd rm r^ ro rp r r r r r rc s
/rc )%__doc__r rb sysZdns._asyncbackendr Z
dns._featuresZ
dns.exceptionZdns.inetplatformrh r r r# Z
_asyncbackendr$ r; Z _featuresZhaverR rV Zhttpcore._backends.anyioZhttpxZAsyncNetworkBackendZ_CoreAsyncNetworkBackendZ _backendsZ AnyIOStreamrT Z dns.queryrD rE rF rG ZAsyncHTTPTransportr_ Z
NullTransportrc r r r r s2
#
5