a
àMÒeö ã @ s¬ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
d dl Z
d dl m
Z
d dlmZmZmZmZmZ G dd„ deƒZG dd„ deƒZG dd „ d eƒZdS )
é N)Ú NullContext)ÚQUIC_MAX_DATAGRAMÚAsyncQuicConnectionÚAsyncQuicManagerÚBaseQuicStreamÚ
UnexpectedEOFc sX e Zd Z‡ fdd„Zdd„ Zddd„Zdd d
„Zd d „ Zd
d„ Zdd„ Z dd„ Z
‡ Z S )ÚTrioQuicStreamc s t ƒ ||¡ t ¡ | _d S ©N)ÚsuperÚ__init__ÚtrioÚ ConditionÚ_wake_up)ÚselfÚ
connectionÚ stream_id©Ú __class__© ú2/usr/lib/python3.9/site-packages/dns/quic/_trio.pyr s zTrioQuicStream.__init__c à sl | j |¡rd S || _| j4 I d H š& | j ¡ I d H W d ƒI d H q`1 I d H sV0 Y d| _q d S )Nr )Ú_bufferZhaveZ
_expectingr Úwait)r Úamountr r r Úwait_for s :zTrioQuicStream.wait_forNc à sŽ |d u rt d ƒ}n
t |¡}|R | d¡I d H t d| j d¡¡\}| |¡I d H | j |¡W d ƒ S 1 sx0 Y tj j
‚d S )Né z!H) r r Ú
move_on_afterr ÚstructÚunpackr ÚgetÚdnsÚ exceptionÚTimeout)r ÚtimeoutÚcontextÚsizer r r Úreceive' s
*zTrioQuicStream.receiveFc à s& | |¡}| j | j||¡I d H d S r )Z _encapsulateÚ _connectionÚwriteZ
_stream_id)r ÚdatagramÚis_endÚdatar r r Úsend3 s
zTrioQuicStream.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_input7 s zTrioQuicStream._add_inputc à s | ¡ d S r )Z_close©r r r r Úclose< s zTrioQuicStream.closec à s | S r r r. r r r Ú
__aenter__A s zTrioQuicStream.__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__D s 4zTrioQuicStream.__aexit__)N)F) Ú__name__Ú
__module__Ú __qualname__r r r% r+ r- r/ r0 r5 Ú
__classcell__r r r r r s
r c sR e Zd Zd‡ fdd„ Zdd„ Zdd„ Zdd d
„Zd d „ Zdd
d„Zdd„ Z ‡ Z
S )ÚTrioQuicConnectionNc sP t ƒ ||||||¡ tj | jtjd¡| _t ¡ | _t ¡ | _ d | _
d| _ d S )Nr F) r
r r ÚsocketÚ_afZ
SOCK_DGRAMÚ_socketÚEventÚ_handshake_completeÚ _run_doneÚ
_worker_scopeÚ
_send_pending)r r ÚaddressÚportÚsourceÚ source_portZmanagerr r r r L s
zTrioQuicConnection.__init__c à s< z"| j r*| j tj | j | j¡¡I d H | j | j¡I d H | j s|
d¡\}}| j r^d}t j
t ¡ | d:| _| j t¡I d H }| j || jt ¡ ¡ W d ƒ n1 s²0 Y d | _| |¡ | ¡ I d H d| _ | j t ¡ ¡}|D ]\}}| j |¡I d H qôq>W d| _ | j ¡ nd| _ | j ¡ 0 d S )NFg )ZdeadlineT)Ú_sourcer= Zbindr ZinetZlow_level_address_tupler< ÚconnectÚ_peerÚ_doneZ_get_timer_valuesrB r Z CancelScopeZ current_timerA Zrecvr r& Zreceive_datagramÚtimeZ
_handle_timerÚ_handle_eventsZdatagrams_to_sendr+ r? Úset)r Z
expirationÚintervalr( Z datagramsÚ_r r r Ú_workerT s6 ÿ
ÿ4
ÿzTrioQuicConnection._workerc à 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ó é é
)r& Z
next_eventÚ
isinstanceÚaioquicZquicZeventsZStreamDataReceivedÚ_streamsr r r- r* Z
end_streamZHandshakeCompletedr? rM ZConnectionTerminatedrJ r= r/ Z StreamResetr Úsleep)r ÚcountZeventÚstreamr r r rL x s*
z!TrioQuicConnection._handle_eventsFc à s. | j |||¡ d| _| jd ur*| j ¡ d S )NT)r& Zsend_stream_datarB rA Úcancel)r rY r* r) r r r r' ‘ s
zTrioQuicConnection.writec à s` | j r
d S t ¡ 4 I d H š"}| | j¡ W d ƒI d H qR1 I d H sH0 Y | j ¡ d S r )Ú_closedr Z open_nurseryÚ
start_soonrP r@ rM )r Únurseryr r r Úrun— s
6zTrioQuicConnection.runc à sŠ |d u rt d ƒ}n
t |¡}|N | j ¡ I d H | jr s 2m