mbi.ddlmZddlmZddlmZddlmZmZm Z m Z ddl m Z Gdd e ZGd d e ZGd d ZGddZGddZdS)) annotations)RangeSet)events) QuicErrorCodeQuicResetStreamFrameQuicStopSendingFrameQuicStreamFrame)QuicDeliveryStateceZdZdS)FinalSizeErrorN__name__ __module__ __qualname___/var/www/html/mdtn/previsions/meteo_cartes/venv/lib/python3.11/site-packages/qh3/quic/stream.pyrrDrrceZdZdS)StreamFinishedErrorNrrrrrrrrrcteZdZdZdZdd Zdd Zd d Zd!dZe j dd"dZ d#dZ e j fd$dZ d%dZdS)&QuicStreamReceiverz The receive part of a QUIC stream. It finishes: - immediately for a send-only stream - upon reception of a STREAM_RESET frame - upon reception of a data frame with the FIN bit set ) highest_offset is_finished stop_pending_buffer _buffer_start _final_size_ranges _stream_id_stop_error_code stream_id int | NonereadableboolreturnNonecd|_d|_d|_t|_d|_d|_t|_||_ d|_ dS)NrF) rrr bytearrayrrr rr!r"r#)selfr$r&s r__init__zQuicStreamReceiver.__init__,sU ! {{ '+zz #,0rr cFd|_t|j|jS)NF error_coder$)rr r#r"r,s rget_stop_framez!QuicStreamReceiver.get_stop_frame8s,!#,o    rintc|jSN)rr1s rstarting_offsetz"QuicStreamReceiver.starting_offset?s !!rframer events.StreamDataReceived | Nonec|j|jz }t|j}|j|z}|j;||jkrt d|jr||jkrt d|jr||_||jkr||_|dkrM|rK|jsD|xj|z c_|jrd|_ tj |j|j|j S|dkr;|j| d|_|xj|zc_d}t|j}||jkr |j |j||t|jz }|dkr|xjt|z c_|j|j|||z<|}|j|jk}|rd|_ |s|rtj |||j SdS)z2 Handle a frame of received data. NzData received beyond final sizeCannot change final sizerT)data end_streamr$)offsetrlenr;r rfinrrrrStreamDataReceivedr"r!addr+ _pull_data)r,r7poscount frame_endgapr;r<s r handle_framezQuicStreamReceiver.handle_frameBs%lT//EJL5(    '4+++$%FGGG AyD,<<<$%?@@@ 9 )(D  t* * *"+D  !888dl8   % '  y (#' ,ZEI  77SDEE*EJ LLC LLC OOE u| # # L  U\9 5 5 5C %%% 77 LLIcNN *LL*/* S3;&'  '4+;;  $#D   : ,jDO 4r)r0 final_sizer0events.StreamReset | Nonec|j||jkrtd||_d|_tj||jS)zX Handle an abrupt termination of the receiving part of the QUIC stream. Nr:Tr/)r rrr StreamResetr")r,rHr0s r handle_resetzQuicStreamReceiver.handle_reset}sS   'J$:J,J,J !;<< <&!Z4?SSSSrdeliveryr c8|tjkr d|_dSdS)z8 Callback when a STOP_SENDING is ACK'd. TN)r ACKEDrr,rMs ron_stop_sending_deliveryz+QuicStreamReceiver.on_stop_sending_deliverys) (. . . $D    / .rc"||_d|_dS)zH Request the peer stop sending data on the QUIC stream. TN)r#rr,r0s rstopzQuicStreamReceiver.stops!+ rbytesc( |jdd|jk}n#t$rd}YnwxYw|sdS|j}|d|dz }t |jd|}|jd|=|d|_|S)z; Remove data from the front of the buffer. rFrrN)r!r IndexErrorshiftrUr)r,has_data_to_readrrCr;s rrBzQuicStreamReceiver._pull_datas %#|Aq1T5GG   % % %$    % 3 L   dQqTkT\$3$'(( L# qT s  ..N)r$r%r&r'r(r))r(r r(r3)r7r r(r8)rHr3r0r3r(rIrMr r(r)r0r3r(r))r(rU)rrr__doc__ __slots__r-r2r6rGrNO_ERRORrLrQrTrBrrrrrs I 1 1 1 1    """"9999x5B4J T T T T T T%%%%&3%;!!!!!rrcleZdZdZdZd d Zed!d Z d"d#dZd$dZ d%dZ d&dZ d'dZ d(d)dZ d S)*QuicStreamSenderz The send part of a QUIC stream. It finishes: - immediately for a receive-only stream - upon acknowledgement of a STREAM_RESET frame - upon acknowledgement of a data frame with the FIN bit set )buffer_is_emptyrr reset_pending_ackedr _buffer_finr _buffer_stop_pending _pending_eof_reset_error_coder"send_buffer_emptyr$r%writabler'r(r)cd|_d|_| |_d|_t |_t |_d|_d|_ d|_ t |_ d|_ d|_ ||_dS)NTrF)rcrrrdrrer+rrfrrgrhrirjr")r,r$rls rr-zQuicStreamSender.__init__sx#'<"jj  {{ '+  !-1#rr3cX |jddS#t$r |jcYSwxYw)z The offset for the next frame to send. This is used to determine the space needed for the frame's `offset` field. r)rhrWrgr1s r next_offsetzQuicStreamSender.next_offsetsC %=#A& & % % %$ $ $ $ %s ))Nmax_size max_offsetQuicStreamFrame | Nonec |jd}n>#t$r1|jrd|_td|jcYSd|_YdSwxYw|d}t |d||z}|||kr|}||krdStt|j||j z ||j z |}|j ||||j kr||_ |j|krd|_ d|_|S)z. Get a frame of data to send. rFT)r?r=Nr)r;r=) rhrWrir rfrcminrUrrsubtractrr?)r,rprqrZstartrTr7s r get_framezQuicStreamSender.get_framesR  a AA     J$)!&48HIIIIII#'D 44 !1Q4)**  !dZ&7&7D 5==4  UT%77$AS:SST     ud+++ $% % %"&D   t # #EI %D  s .A A  A r cRd|_t|j|j|jS)NF)r0rHr$)rdr rjrr"r1s rget_reset_framez QuicStreamSender.get_reset_frames2"#-*o    rrMr rvrTcd|_|tjkr||kr}|j|||jd}|d|jkrD|d|dz }|j|xj|z c_|jd|=|j|jkr d|_ dSdS||kr|j ||||jkrd|_ d|_ dSdS)z3 Callback when sent data is ACK'd. FrrNT) rcr rOrerArrXrrfrrhrkri)r,rMrvrT first_rangesizes ron_data_deliveryz!QuicStreamSender.on_data_deliverys % (. . .e|| t,,,"k!n q>T%777&q>KN:DK%%'''&&$.&& UdU+!T%555#'   65e|| !!%...t''').&$(!!!('rcF|tjkr d|_dSd|_dS)z1 Callback when a reset is ACK'd. TN)r rOrrdrPs ron_reset_deliveryz"QuicStreamSender.on_reset_delivery4s/ (. . .#D   !%D   rr0cB|j||_d|_d|_dSdS)z Abruptly terminate the sending part of the QUIC stream. Once this method has been called, any further calls to it will have no effect. NT)rjrdrcrSs rresetzQuicStreamSender.reset>s5  ! )%/D "!%D $(D * )rFr;rUr<cF|j Jd|j Jdt|}|rOd|_|j|j|j|z|xj|z c_|xj|z c_|rd|_|j|_d|_dSdS)z; Write some data bytes to the QUIC stream. Nzcannot call write() after FINz!cannot call write() after reset()FT) rfrjr>rcrhrArgrri)r,r;r<r|s rwritezQuicStreamSender.writeLs'')H'''%--/R---4yy  &#(D M  d/1BT1I J J J LLD LL    %    %#(D #0D  $D    % %r)r$r%rlr'r(r)r[r5)rpr3rqr%r(rr)r(r )rMr rvr3rTr3r(r)r\r])F)r;rUr<r'r(r))rrrr^r_r-propertyrorwryr}rrrrrrrbrbsI"$$$$  % % %X %7;,,,,,\    ))))6&&&& ( ( ( (%%%%%%%rrbc>eZdZdZ dddZeddZdS) QuicStream) is_blockedmax_stream_data_localmax_stream_data_local_sentmax_stream_data_remotereceiversenderr$NrTr$r%rr3rr&r'rlr(r)cd|_||_||_||_t |||_t |||_||_dS)NF)r$r&)r$rl) rrrrrrrbrr$)r,r$rrr&rls rr-zQuicStream.__init__jsV %:"*?'&<#*YRRR &XNNN "rc2|jjo |jjSr5)rrrr1s rrzQuicStream.is_finishedzs}(DT[-DDr)NrrTT) r$r%rr3rr3r&r'rlr'r(r))r(r')rrrr_r-rrrrrrr_skI!%%&&' ##### EEEXEEErrN) __future__r_hazmatrrpacketrr r r packet_builderr Exceptionrrrrbrrrrrs}"""""" .-----     Y        )   SSSSSSSSlp%p%p%p%p%p%p%p%fEEEEEEEEEEr