RPf d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
d dl Z
d dl Z
d dl
mZmZmZmZmZ G d de Z G d de Z G d de Zy) N)QUIC_MAX_DATAGRAMAsyncQuicConnectionAsyncQuicManagerBaseQuicStream
UnexpectedEOFc R e Zd Z fdZd Zd Zd
dZd dZd Zd Z d Z
d Z xZ S ) AsyncioQuicStreamc X t | || t j | _ y N)super__init__asyncio Condition_wake_up)self
connection stream_id __class__s K/var/www/fastapitest/venv/lib/python3.12/site-packages/dns/quic/_asyncio.pyr
zAsyncioQuicStream.__init__ s"
Y/ ))+
c K | j 4 d { | j j d { d d d d { y 7 77 7 # 1 d { 7 sw Y y xY wwr )r waitr s r _wait_for_wake_upz#AsyncioQuicStream._wait_for_wake_up sV == ' '--$$& & & ' ' ' & ' ' ' 'S A&A A& AA
A A&AA&
AA&A#A A# A&c 0 K | j | }| j j | ry || _ t j
| j
| d { d| _ i7 # t $ r t j j w xY ww)Nr ) _timeout_from_expiration_bufferhave
_expectingr wait_forr TimeoutErrordns exceptionTimeout)r amount
expirationtimeouts r r! zAsyncioQuicStream.wait_for" s 33J?G|| ( $DO
, &&t'='='?III DO J
,mm+++
,s. 6B'A/ A-!A/ %B-A/ /$BBc 4 K | j | }| j d| d { t j d| j j d \ }| j || d { | j j | S 7 j7 !w)N z!H)_expiration_from_timeoutr! structunpackr get)r r( r' sizes r receivezAsyncioQuicStream.receive. s} 227;
mmAz***--dll&6&6q&9:mmD*---|| %% +-s" 'BBA
B4B5 BBc K | j | }| j j | j || d { y 7 wr ) _encapsulate _connectionwrite
_stream_id)r datagramis_enddatas r sendzAsyncioQuicStream.send5 s8 * $$T__dFCCCs s
s c K | S wr r s r
__aenter__z AsyncioQuicStream.__aenter__C c K | j d { | j 4 d { | j j d d d d { y7 D7 17 # 1 d { 7 sw Y yxY wwNF)r@ r r< )r exc_typeexc_valexc_tbs r __aexit__zAsyncioQuicStream.__aexit__F sg jjl== # # MM " # # # # # # #sT A6AA6A A6A!
A6A A6 A6 A6!A3'A* (A3/A6r F)
__name__
__module__ __qualname__r
r r! r0 r9 r= r@ rC rK
__classcell__r s @r r r s1 ,'
&D'
r r c T e Zd Zd
fd Zd Zd Zd Zd Zd dZd Z d
dZ
d Z xZ S ) AsyncioQuicConnectionc t | |||||| d | _ t j | _ t j | _ t j | _ d | _ d | _
y r ) r r
_socketr Event_handshake_complete_socket_createdr _wake_timer_receiver_task _sender_task)r r addressportsource source_portmanagerr s r r
z AsyncioQuicConnection.__init__N s^
WdFKQ #*==? &}} ",,. " r c K t j j | j }t j j d }|j
|t j d| j | j d | j d f d { | _ | j j | j 4 d { | j s| j j t d d { \ }}|d | j d k7 s|d | j d k7 rb| j" j% ||t' j& | j( 4 d { | j( j+ d d d d { | j sd d d d { d| _ | j( 4 d { | j( j+ d d d d { | j. j y 7 7 V7 %7 7 # 1 d { 7 sw Y xY w7 # 1 d { 7 sw Y xY w# t, $ r Y w xY w7 7 i# 1 d { 7 sw Y yxY w# d| _ | j( 4 d { 7 | j( j+ d d d d { 7 n# 1 d { 7 sw Y nxY w| j. j w xY ww)Nr r T)r# inetaf_for_address_address asyncbackend get_backend make_socketsocket
SOCK_DGRAM_source_peerrU rX set_donerecvfromr r3 receive_datagramtimerY
notify_all ExceptionrW )r afbackendr6 r\ s r _receiverz AsyncioQuicConnection._receiverW sq +(( 7B&&229=G ")!4!4F%%q$,,A
ST
8V" DL
$ $ &||
6
6 **04 0E0E)41 +'Xw qzTZZ]2gajDJJqM6Q $$55hU $ / / 6 6 ((3356 6 **
6
6 DJ''
.
. ++-
.
. $ $ ( ( *-
6+6 6 6 6 6
6
6
6
6
.
.
.
.
. DJ''
.
. ++-
.
.
.
.
. $ $ ( ( *sn K=BI H 3I H"
I 0I H%A1I2H(3I6H, I H* I. I 9I:I >K=I'K=I+4 K=?I) K= I "I %I(I*I,H> 2H53H> :II I I
II I$!J #I$$J 'K=)K=+I=1I4 2I=9K= K:J
K: K
8 K:KK:
K KK "K::K=c K | j 4 d { | j j d { d d d d { y 7 77 7 # 1 d { 7 sw Y y xY wwr )rY r r s r _wait_for_wake_timerz*AsyncioQuicConnection._wait_for_wake_timeru sZ ## * *""'') ) ) * * * ) * * * *r c r K | j j d { | j s| j j t j
}|D ]E \ }}|| j k( sJ | j j || j d d { G | j \ }} t j | j | d { | j | | j d { | j sy y 7 7 7 @# t $ r Y Iw xY w7 (wr )rX r rn r3 datagrams_to_sendrq rl rU sendto_get_timer_valuesr r! rx rs
_handle_timer_handle_events)r datagramsr6 r\ r' intervals r _senderz AsyncioQuicConnection._sendery s ""'')))** ((::499;GI%.
F!' $**,,,ll))(DJJEEE
F &*%;%;%= "Z
&&t'@'@'BHMMM
z *%%' ' ' ** *
F N
(sj D7D B D7!D""D7<'D& #D$$D& ($D7 D5
D7 D7"D7$D& & D2/D71D22D7c 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 rb
)r3
next_event
isinstanceaioquicquiceventsStreamDataReceived_streamsr. r r= r8
end_streamHandshakeCompletedrW rm ConnectionTerminatedrn rZ cancel StreamResetr sleep)r counteventstreams r r~ z$AsyncioQuicConnection._handle_events sg $$//1E}% !4!4!G!GH **5??; ++EJJ8H8HIIIE7<<#6#6#I#IJ((,,.E7<<#6#6#K#KL !
##**,E7<<#6#6#B#BC **5??; ++C666 QJErz mmA&&&- J 7 's7 B G#"G #D G#,G -(G#G!G# G#!G#c K | j j ||| | j 4 d { | j j d d d d { y 7 /7 # 1 d { 7 sw Y y xY wwr )r3 send_stream_datarY rr )r r r8 r7 s r r4 zAsyncioQuicConnection.write sh ))&$?## * * ' ' ) * * * * * * *sD .A9A A9A$ A9A"A9"A9$A6*A- +A62A9c | j ry t j | j | _ t j | j | _ y r )_closedr Taskrv rZ r r[ r s r runzAsyncioQuicConnection.run s; << %ll4>>+; < #LL 8r c ^ K t j | j j | d { | j rt | j j d }t | | }|| j |< |S 7 N# t $ r t
j j w xY wwrG )r r! rW r r" r# r$ r% rn r r3 get_next_available_stream_idr r )r r( r r s r make_streamz!AsyncioQuicConnection.make_stream s (""4#;#;#@#@#BGL L L :: $$AA%H "43#)
i
M (--'' ' (s, B-1B BB A
B-B $B**B-c K | j s| j j | j d | j d d| _ | j j | j j | j 4 d { | j j d d d d { | j d { | j d { | j j d { y y 7 |7 T# 1 d { 7 sw Y dxY w7 V# t j $ r Y iw xY w7 \# t j $ r Y ow xY w7 Uw)Nr rb T)r _managerclosedrl r3 r@ rX rm rY rr rZ r CancelledErrorr[ rU r s r r@ zAsyncioQuicConnection.close s) || MM A
1
> DL " " $ $ $ &''
.
. ++-
.
.
))))
'''' ,,$$& & &!
.
.
.
.
. *))
())
&s BE D
E D/ E :D;E D( D&D( E $E%E ) E E E E D#D D# E &D( (D>;E =D>>E E EE EE r rL )
rM rN rO r
rv rx r r~ r4 r r r@ rP rQ s @r rS rS M s0 ! +<*
( '4*
9
'r rS c P e Zd Zdej df fd Z ddZd Zd Z xZ S )AsyncioQuicManagerNc 2 t | ||t | y r )r r
rS )r conf verify_mode server_namer s r r
zAsyncioQuicManager.__init__ s
{,A;Or c Z | j ||||| \ }}|r|j |S r )_connectr )r r\ r] r^ r_ want_session_ticketr starts r connectzAsyncioQuicManager.connect s8 #mm T6;0C
U NN r c K | S wr rB r s r rC z AsyncioQuicManager.__aenter__ rD rE c K t | j j }|D ] }|j d { y7 wrG )list _connectionsvaluesr@ )r rH rI rJ connectionsr s r rK z AsyncioQuicManager.__aexit__ sJ 4,,3356 % %J ""$ $ $ %
%s r s\
! 1 1h}' / }'@) r