a
àMÒeß ã @ s¨ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
d dl Z
d dl Z
d dl
mZmZmZmZmZ G dd„ deƒZG dd„ deƒZG dd„ deƒZdS ) é N)ÚQUIC_MAX_DATAGRAMÚAsyncQuicConnectionÚAsyncQuicManagerÚBaseQuicStreamÚ
UnexpectedEOFc s` e Zd Z‡ fdd„Zdd„ Zdd„ Zddd „Zdd d „Zd
d„ Zdd„ Z dd„ Z
dd„ Z ‡ Z S )ÚAsyncioQuicStreamc s t ƒ ||¡ t ¡ | _d S ©N)ÚsuperÚ__init__ÚasyncioÚ ConditionÚ_wake_up)ÚselfÚ
connectionÚ stream_id©Ú __class__© ú5/usr/lib/python3.9/site-packages/dns/quic/_asyncio.pyr
s zAsyncioQuicStream.__init__c à sN | j 4 I d H š& | j ¡ I d H W d ƒI d H qJ1 I d H s@0 Y d S r )r
Úwait©r r r r Ú_wait_for_wake_up s z#AsyncioQuicStream._wait_for_wake_upc à sb | |¡}| j |¡rd S || _zt | ¡ |¡I d H W n tyT tj j
‚Y n0 d| _q d S )Nr ) Z_timeout_from_expirationÚ_bufferZhaveZ
_expectingr Úwait_forr Ú TimeoutErrorÚdnsÚ exceptionÚTimeout)r ÚamountÚ
expirationÚtimeoutr r r r " s
zAsyncioQuicStream.wait_forNc à sP | |¡}| d|¡I d H t d| j d¡¡\}| ||¡I d H | j |¡S )Né z!H)Z_expiration_from_timeoutr ÚstructÚunpackr Úget)r r r Úsizer r r Úreceive. s
zAsyncioQuicStream.receiveFc à s& | |¡}| j | j||¡I d H d S r )Z _encapsulateÚ _connectionÚwriteZ
_stream_id)r ÚdatagramÚis_endÚdatar r r Úsend5 s
zAsyncioQuicStream.sendc à sT | ||¡rP| j4 I d H š | j ¡ W d ƒI d H qP1 I d H sF0 Y d S r )Z_common_add_inputr
Únotify)r r+ r* r r r Ú
_add_input9 s z AsyncioQuicStream._add_inputc à s | ¡ d S r )Z_closer r r r Úclose> s zAsyncioQuicStream.closec à s | S r r r r r r Ú
__aenter__C s z AsyncioQuicStream.__aenter__c à sV | ¡ I d H | j4 I d H š | j ¡ W d ƒI d H qR1 I d H sH0 Y dS ©NF)r/ r
r- )r Úexc_typeÚexc_valÚexc_tbr r r Ú __aexit__F s 4zAsyncioQuicStream.__aexit__)N)F)
Ú__name__Ú
__module__Ú __qualname__r
r r r& r, r. r/ r0 r5 Ú
__classcell__r r r r r s
r c sb e Zd Zd‡ fdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zdd
d„Zdd„ Z ddd„Z
dd„ Z ‡ Z S )ÚAsyncioQuicConnectionNc sJ t ƒ ||||||¡ d | _t ¡ | _t ¡ | _t ¡ | _d | _ d | _
d S r ) r r
Ú_socketr ÚEventÚ_handshake_completeÚ_socket_createdr Ú _wake_timerÚ_receiver_taskÚ _sender_task)r r ÚaddressÚportÚsourceÚ source_portZmanagerr r r r
N s
z AsyncioQuicConnection.__init__c à s zªz6t j | j¡}t j d¡}| |tjd| j | j
d | j
d f¡I d H | _ | j
¡ | j 4 I d H šº | js | j td ¡I d H \}}|d | j
d ksj|d | j
d kr°qj| j ||t ¡ ¡ | j4 I d H š | j ¡ W d ƒI d H qj1 I d H s 0 Y qjW d ƒI d H q:1 I d H s00 Y W n tyP Y n0 W d| _| j4 I d H š" | j ¡ W d ƒI d H q¢1 I d H s˜0 Y | j
¡ nZd| _| j4 I d H š" | j ¡ W d ƒI d H qü1 I d H sò0 Y | j
¡ 0 d S )Nr r é T)r ZinetZaf_for_addressZ_addressZ asyncbackendZ get_backendZ make_socketÚsocketZ
SOCK_DGRAMÚ_sourceÚ_peerr; r> ÚsetÚ_doneZrecvfromr r' Zreceive_datagramÚtimer? Ú
notify_allÚ Exceptionr= )r ÚafZbackendr) rB r r r Ú _receiverW s6 ÿ
ÿ$j8 ý8z AsyncioQuicConnection._receiverc à sN | j 4 I d H š& | j ¡ I d H W d ƒI d H qJ1 I d H s@0 Y d S r )r? r r r r r Ú_wait_for_wake_timeru s z*AsyncioQuicConnection._wait_for_wake_timerc à s¤ | j ¡ I d H | js | j t ¡ ¡}|D ] \}}| j || jd ¡I d H q*| ¡ \}}zt
| ¡ |¡I d H W n t
y„ Y n0 | |¡ | ¡ I d H qd S r )r> r rK r' Zdatagrams_to_sendrL r; ZsendtorI Z_get_timer_valuesr r rQ rN Z
_handle_timerÚ_handle_events)r Z datagramsr) rB r Úintervalr r r Ú_sendery s
z AsyncioQuicConnection._senderc à sð d}| j ¡ }|d u rd S t|tjjjƒrT| j |j ¡}|rÆ|
|j |j ¡I d H nrt|tjjj
ƒrp| j ¡ nVt|tjjjƒr’d| _| j ¡ n4t|tjjjƒrÆ| j |j ¡}|rÆ|
dd¡I d H |d7 }|dkrd}t d¡I d H qd S )Nr Tó rF é
)r' Z
next_eventÚ
isinstanceÚaioquicZquicZeventsZStreamDataReceivedÚ_streamsr$ r r. r+ Z
end_streamZHandshakeCompletedr= rJ ZConnectionTerminatedrK r@ ÚcancelZ StreamResetr Úsleep)r ÚcountZeventÚstreamr r r rR ˆ s*
z$AsyncioQuicConnection._handle_eventsFc à sX | j |||¡ | j4 I d H š | j ¡ W d ƒI d H qT1 I d H sJ0 Y d S r )r' Zsend_stream_datar? rM )r r] r+ r* r r r r( ¢ s zAsyncioQuicConnection.writec C s. | j r
d S t | ¡ ¡| _t | ¡ ¡| _d S r )Ú_closedr ZTaskrP r@ rT rA r r r r Úrun§ s zAsyncioQuicConnection.runc à sf z t | j ¡ |¡I d H W n ty6 tjj‚Y n0 | jrBt ‚| j
d¡}t | |ƒ}|| j
|< |S r1 )r r r= r r r r r rK r r' Z get_next_available_stream_idr rY )r r r r] r r r Ú make_stream s
z!AsyncioQuicConnection.make_streamc à sÞ | j sÚ| j | jd | jd ¡ d| _ | j ¡ | j ¡ | j4 I d H š | j ¡ W d ƒI d H q~1 I d H st0 Y z| j
I d H W n t j y¢ Y n0 z| j
I d H W n t j yÈ Y n0 | j ¡ I d H d S )Nr rF T)r^ Z_managerÚclosedrI r' r/ r> rJ r? rM r@ r ZCancelledErrorrA r; r r r r r/ ¹ s
4zAsyncioQuicConnection.close)N)F)N)
r6 r7 r8 r
rP rQ rT rR r( r_ r` r/ r9 r r r r r: M s
r: c s@ e Zd Zdejdf‡ fdd„ Zd
dd„Zd d
„ Zd d „ Z‡ Z S )ÚAsyncioQuicManagerNc s t ƒ ||t|¡ d S r )r r
r: )r ZconfZ verify_modeZ server_namer r r r
Î s zAsyncioQuicManager.__init__éU r Tc C s& | |||||¡\}}|r"| ¡ |S r )Z_connectr_ )r rB rC rD rE Zwant_session_ticketr Ústartr r r ÚconnectÑ s
ÿzAsyncioQuicManager.connectc à s | S r r r r r r r0 Û s z AsyncioQuicManager.__aenter__c à s* t | j ¡ ƒ}|D ]}| ¡ I d H qdS r1 )ÚlistZ _connectionsÚvaluesr/ )r r2 r3 r4 Z connectionsr r r r r5 Þ s z AsyncioQuicManager.__aexit__)rc Nr T)
r6 r7 r8 ÚsslZ
CERT_REQUIREDr
re r0 r5 r9 r r r r rb Í s
ÿ
rb )r rG rh r" rL Zaioquic.quic.configurationrX Zaioquic.quic.connectionZaioquic.quic.eventsZdns.asyncbackendr Z
dns.exceptionZdns.inetZdns.quic._commonr r r r r r r: rb r r r r Ú s 4