RPf d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
d dl Z
d dl m
Z
d dlmZmZmZmZmZ G d de Z G d de Z G d d e Zy)
N) NullContext)QUIC_MAX_DATAGRAMAsyncQuicConnectionAsyncQuicManagerBaseQuicStream
UnexpectedEOFc L e Zd Z fdZd Zd dZd
dZd Zd Zd Z d Z
xZ S ) TrioQuicStreamc X t | || t j | _ y N)super__init__trio Condition_wake_up)self
connection stream_id __class__s H/var/www/fastapitest/venv/lib/python3.12/site-packages/dns/quic/_trio.pyr zTrioQuicStream.__init__ s
Y/ (
c & K | j j | ry || _ | j 4 d { | j j d { d d d d { d| _ r7 >7 7 # 1 d { 7 sw Y xY ww)Nr )_bufferhave
_expectingr wait)r amounts r wait_forzTrioQuicStream.wait_for s || ( $DO}}
+
+mm((***
+
+ DO
+*
+
+
+
+sV 5BA6B A<A8A< B)A:*
B8A<:B<BB B
Bc K | t d }nt j | }|5 | j d d { t j
d| j j d \ }| j | d { | j j | cd d d S 7 r7 *# 1 sw Y t j j xY ww)N z!H) r r
move_on_afterr structunpackr getdns exceptionTimeout)r timeoutcontextsizes r receivezTrioQuicStream.receive' s ?!$'G((1G
*--" " " mmD$,,*:*:1*=>GT--% % %<<##D) * * " % *
mm###s? &CB4B0A B4B2 B4&
C0B42B44C9 Cc K | j | }| j j | j || d { y 7 wr ) _encapsulate _connectionwrite
_stream_id)r datagramis_enddatas r sendzTrioQuicStream.send3 s8 * $$T__dFCCCs >tyy{ K #, 6KHa ,,++H5556/ jj4 DJ $ $ ( ( *?
3 JY Y , 6 DJ $ $ ( ( *s IA H4 H!,H4
H A H4 ,(H$H!=H$3H4 H0A H4 $H2%H4 9"IH4 H4 !H$$H-)H4 2H4 4#IIc K d} | j j }|y t |t j j
j rX| j j |j }|r8|j |j |j d { nt |t j j
j r| j j nt |t j j
j r"d| _ | j$ j' not |t j j
j( rA| j j |j }|r|j dd d { |dz
}|dkD r d}t+ j, d d { 7 :7 27 w)Nr Tr
)r.
next_event
isinstanceaioquicquiceventsStreamDataReceived_streamsr$ r r8 r3
end_streamHandshakeCompletedrV rq ConnectionTerminatedrg rT r< StreamResetr sleep)r counteventstreams r ro z!TrioQuicConnection._handle_eventsx sd $$//1E}% !4!4!G!GH **5??; ++EJJ8H8HIIIE7<<#6#6#I#IJ((,,.E7<<#6#6#K#KL !
""$E7<<#6#6#B#BC **5??; ++C666 QJErzjjm##+ J 7
$s7 B G#"G #D G#,G -(G#G!G# G#!G#c K | j j ||| d| _ | j | j j y y w)NT)r. send_stream_datarY rX cancel)r r r3 r2 s r r/ zTrioQuicConnection.write sI ))&$? ! ) % % ' *s A
Ac K | j ry t j 4 d { }|j | j d d d d { | j
j
y 7 J7 !# 1 d { 7 sw Y 1xY wwr )_closedr open_nursery
start_soonrv rW rq )r nurserys r runzTrioQuicConnection.run so << $$& - -' t|| , - - - - - - -sD &B A3B A7 B A5 B 5B 7B =B >B B c K | t d }nt j | }|5 | j j d { | j
rt | j j d }t | | }|| j |< |cd d d S 7 W# 1 sw Y t j j xY wwrC )r r r! rV r rg r r. get_next_available_stream_idr
r r% r& r' )r r( r) r r s r make_streamz TrioQuicConnection.make_stream s ?!$'G((1G
**//1 1 1zz## ((EEeLI#D)4F'-DMM) $ 1 mm###s/ &C B B A
B
C B C% Cc v K | j s| j j | j d | j d d| _ | j j d| _ | j | j j | j j d { y y 7 w)Nr rx T) r _managerclosedrf r. r< rY rX r rW r r; s r r< zTrioQuicConnection.close s || MM A
1
> DL " " $!%D !!-""))+..%%' ' '
(s B.B90B71B9r rH ) rI rJ rK r rv ro r/ r r r< rL rM s @r rO rO K s' #"+H$2(
$ (r rO c P e Zd Zdej df fd Z ddZd Zd Z xZ S )TrioQuicManagerNc @ t | ||t | || _ y r )r
r rO _nursery)r r conf verify_mode server_namer s r r zTrioQuicManager.__init__ s {,> L
r c | j ||||| \ }}|r%| j j |j |S r )_connectr r r )r rZ r[ r\ r] want_session_ticketr starts r re zTrioQuicManager.connect sD #mm T6;0C
U MM $ $Z^^ 4r c K | S wr r> r; s r r? zTrioQuicManager.__aenter__ r@ rA c K t | j j }|D ] }|j d { y7 wrC )list _connectionsvaluesr< )r rD rE rF connectionsr s r rG zTrioQuicManager.__aexit__ sJ 4,,3356 % %J ""$ $ $ %
%s r r r s\
! ) /^ /dj(, j(Z& r