dfC 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 J/home/api-vastappli/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 wwNr ) _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)rA 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" r1 r: r> rA rD rL
__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% || j d 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 c7 27 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_peerrV rY set_donerecvfromr r4 receive_datagramtimerZ
notify_all ExceptionrX )r afbackendr7 r] s r _receiverz AsyncioQuicConnection._receiverW s| +(( 7B&&229=G ")!4!4F%%q$,,A
ST
8V" DL
$ $ &||
6
6 **04 0E0E)41 +'Xw qzTZZ]2gajDJJqM6Q $$55 $**Q-
$ / / 6 6 ((3356 6 **
6
6" DJ''
.
. ++-
.
. $ $ ( ( *1
6+6 6 6 6 6
6
6
6
6
.
.
.
.
. DJ''
.
. ++-
.
.
.
.
. $ $ ( ( *sn L
BI% H,3I% H/
I% 0I H2A>I?H5 IH9 I)H7*I; I% II% L
"I4#L
&I8 L
I6
L
,I% /I% 2I5I7I9I ?I I II% I"I I" I% % I1.J
0I11J
4L
6L
8J
>J ?J
L
L%J(
&L*K LKLK) K K) %"LL
c K | j 4 d { | j j d { d d d d { y 7 77 7 # 1 d { 7 sw Y y xY wwr )rZ r r s r _wait_for_wake_timerz*AsyncioQuicConnection._wait_for_wake_timerw sZ ## * *""'') ) ) * * * ) * * * *r c z K | j j d { | j s| j j t j
}|D ]H \ }}|| j d k( sJ | j j || j d d { J | j \ }} t j | j | d { | j | | j d { | j sy y 7 7 7 A# t $ r Y Jw xY w7 )wr )rY r ro r4 datagrams_to_sendrr rm rV sendto_get_timer_valuesr r" ry rt
_handle_timer_handle_events)r datagramsr7 r] r( intervals r _senderz AsyncioQuicConnection._sender{ s ""'')))** ((::499;GI%.
F!' $**Q-//ll))(DJJEEE
F &*%;%;%= "Z
&&t'@'@'BHMMM
z *%%' ' ' ** *
F N
(sj D;D#BD;$D&%D;?'D* &D('D* +$D;D9D;!D;&D;(D* * D63D;5D66D;c < K d} | j j }|y t |t j j
j rV| j j |j }|r|j |j |j d { nt |t j j
j r| j j n}t |t j j
j s.t |t j j
j" r!d| _ | j& j) |dz
}|dkD r d}t+ j, d d { 7 7 w)Nr Trc
)r4
next_event
isinstanceaioquicquiceventsStreamDataReceived_streamsr/ r r> r9
end_streamHandshakeCompletedrX rn ConnectionTerminated StreamResetro r[ cancelr sleep)r counteventstreams r r z$AsyncioQuicConnection._handle_events s3 $$//1E}% !4!4!G!GH **5??; ++EJJ8H8HIIIE7<<#6#6#I#IJ((,,.w||**??E7<<#6#6#B#BC !
##**, QJErz mmA&&&' J 's% B F !F"C/F FF F c K | j j ||| | j 4 d { | j j d d d d { y 7 /7 # 1 d { 7 sw Y y xY wwr )r4 send_stream_datarZ rs )r r r9 r8 s r r5 zAsyncioQuicConnection.write sh ))&$?## * * ' ' ) * * * * * * *sD .A9A A9A$ A9A"A9"A9$A6*A- +A62A9c | j ry t j | j | _ t j | j | _ y r )_closedr Taskrw r[ 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 wwrH )r r" rX r r# r$ r% r& ro r r4 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 j d { | j 4 d { | j j d d d d { | j d { | j d { y y 7 o7 \7 4# 1 d { 7 sw Y DxY w7 6# t j $ r Y Iw xY w7 <# t j $ r Y y w xY ww)Nr rc T)r _managerclosedrm r4 rA rY rn rV rZ rs r[ r CancelledErrorr\ r s r rA zAsyncioQuicConnection.close s) || MM A
1
> DL " " $ $ $ &,,$$& & &''
.
. ++-
.
.
))))
''''
'
.
.
.
.
. *))
())
s BE D
E 2D3E 6D E D E "D* 1D(2D* 7E EE E E E D%D D%!E (D* *E =E ?E E E EE EE r rM )
rN rO rP r
rw ry r r r5 r r rA rQ rR s @r rT rT M s1 ! +@*
( '.*
9
r rT c N e Zd Zdej df fd ZddZd Zd Z xZ S )AsyncioQuicManagerNc 2 t | ||t | y r )r r
rT )r conf verify_mode server_namer s r r
zAsyncioQuicManager.__init__ s
{,A;Or c X | j |||| \ }}|r|j |S r )_connectr )r r] r^ r_ r` r starts r connectzAsyncioQuicManager.connect s. "mmGT6; OU NN r c K | S wr rC r s r rD z AsyncioQuicManager.__aenter__ rE rF c K t | j j }|D ] }|j d { y7 wrH )list _connectionsvaluesrA )r rI rJ rK connectionsr s r rL z AsyncioQuicManager.__aexit__ sJ 4,,3356 % %J ""$ $ $ %
%s r s\
! 1 1h| / |~) r