F2j7ddlmZddlZddlZddlmZmZmZddlm Z m Z m Z ddl m Z ddlmZmZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZ d!dZ!GddZ"GddZ#Gdd Z$dS)") annotationsN)AnyCallableIterable)QuicPacketPacerQuicRttMonitorRangeSet)QuicLoggerTrace)QuicDeliveryStateQuicSentPacketMbP?g?gư>g? g?gffffff?g@gMbp?gMb?xfloatreturnc&|dkr| dz S|dzS)NrgUUUUUU?rs Y/home/thesage/.hermes/hermes-agent/venv/lib/python3.11/site-packages/qh3/quic/recovery.py _cubic_rootr$s&1uu")$%%  ceZdZddZdS)QuicPacketSpacerNonecd|_t|_d|_d|_d|_d|_d|_d|_d|_ d|_ i|_ d|_ dS)NFr) ack_atr ack_queue discardedexpected_packet_numberlargest_received_packetlargest_received_time packet_numberack_eliciting_in_flightlargest_acked_packet loss_time sent_packets!time_of_last_ack_eliciting_packetselfs r__init__zQuicPacketSpace.__init__+sl$( !&'#')$37"()$$%!'+799<...rNrr )__name__ __module__ __qualname__r2rrrrr*s(<<<<<>#q47{q&88>I?T%<<===rc|jtz|_d|_d|_d|_d|_|j|_d|_d|_ d|_ | dS)NTFr#r) r<rBrCrDrErFrGrHrIrJrK_hystart_resetr0s r_resetzQuicCongestionControl._resethsi!%!8;K!K !%.3+,    rcd|_d|_tj|_tj|_tj|_d|_d|_dS)z Reset HyStart++ slow-start exit detector. Called when entering a fresh slow-start phase: at construction, after idle restart, and after persistent congestion collapse. FrN) rMrNrOrPrQrRrSrTrUr0s rr^z$QuicCongestionControl._hystart_resettsH %"#-1X*+/8(.2h+)*&#'   rnowc||_|j|_|j|_|j|jz }|j|jz }t ||z tz |_dSN) rKrCrJrIrHr<rrYrG)r1ra W_max_segcwnd_segs r _start_epochz"QuicCongestionControl._start_epochsZ 1& K$"99 #d&==y83y@AArpacketrc|xj|jzc_|j|_|jr|j|j}|j|jkrm|j|_ tj |_d|_ |j dz|_|jr3|xjdz c_|jt kr|j|_d|_|j|j|jkr=|jr|xj|jt$zz c_dS|xj|jz c_dS|jr4|js-d|_|j|_||j|jr(d|_d|_||jt/|j|j|j|jz zz|_|j|jz }|||jz}||jkr|j}n5|t/d|jzkrt/d|jz}n|}|||jkr|j|_dSt/|j||jz |j|jz zz|_dS)Nrr Fg?)rA sent_bytes sent_timer@rLrDrUr*rSrRrOrPrTrVrMrNK_HYSTART_CSS_ROUNDSrCK_HYSTART_CSS_GROWTH_DIVISORrErFrHrfr:rIr<rKr\r?)r1rgrWW_cubictargets ron_packet_ackedz%QuicCongestionControl.on_packet_ackeds  11)   1 %(4$(@@@/3/RD ,26(D /-.D *'+'Dq'HD $# 1''1,''*.BBB$($:DM+0D( = D$:T]$J$J# <&&%)EE&&&&&&&*;;&&&&% 4d.Q 4).&"4 !!&"23332 46;3).&!!&"2333 )V->AW-WXYDK  4=0AmmA M22G////3sT%;;<<<<S4#99:: }}Q$+--)-&&&*-* 66.1GGII**&&&rc$|xj|jz c_|j|jkr |j|_|jr|j|j |j|_|jdkr0|j|jz }|tkr| dSdSdS)Nr#) rArir*rVrLrDrUr@rjK_CUBIC_MAX_IDLE_TIMEr_)r1rg elapsed_idles ron_packet_sentz$QuicCongestionControl.on_packet_sents  11  $"? ? ?,2,@D )   < %(0'-';D $ >C  !+dnrCrHr:K_CUBIC_LOSS_REDUCTION_FACTORmaxr<K_MINIMUM_WINDOWrDrFrM)r1rtralost_largest_timergs ron_packets_lostz%QuicCongestionControl.on_packets_losts 1 1F  F$5 5 & 0   tC C C36D 0% 33!*a2O.OPSTT #4 %(D*-JJKK'*::&&D "!2DM26D / $)D - D Cr latest_rttc||_|jdS|jr ||jkr||_|xjdz c_|jt krt j|jrt j|jrtttt|jtz }|jr3|j|jkr!d|_d|_t j|_dSdS|j|j|zkr"d|_|j|_d|_dSdSdSdSdS|j||r|j|_dSdS)Nr FrT)r?rDrLrSrTK_HYSTART_N_RTT_SAMPLErOisfiniterRr~K_HYSTART_MIN_RTT_THRESHminK_HYSTART_MAX_RTT_THRESHK_HYSTART_MIN_RTT_DIVISORrMrQrNrPr=is_rtt_increasingrC)r1rra rtt_threshs ron_rtt_measurementz(QuicCongestionControl.on_rtt_measurements = $ F  , 7D???6@3  * *a / * *.2HHHM$"BCCIM$"EFFI!,08;TT '4 ;<==05,23/=AX::: ==;;jHII04, ?:34///CIHHHHH2II 22:sCC 7 $ 6  7 7rc||_|jtz|_d|_d|_d|_d|_|j|_d|_ d|_ d|_ | dS)a| RFC 9002 7.6: on persistent congestion the sender's cwnd is collapsed to the minimum window and slow-start state is reset so the controller re-probes capacity. Set ``_congestion_recovery_start_time`` to ``now`` (the time of the collapse), not 0.0. Otherwise, the very next packet declared lost (e.g. a probe sent just after collapse) passes the ``lost_largest_time > _congestion_recovery_start_time`` guard in ``on_packets_lost`` and pins ``ssthresh = MINIMUM_WINDOW``, terminating the new slow-start phase that this collapse was meant to start. NTFr#r) r>r<rrCrDrErFrGrHrIrJrKr^)r1ras ron_persistent_congestionz.QuicCongestionControl.on_persistent_congestionJss03,!%!8;K!K !%.3+,    rN)r9r:rr )rWrrr:r3rarrr )rgrrr )rtrurr )rtrurarrr )rrrarrr )r4r5r6__doc__r2r\r_r^rfrorsryr{rrrrrrr8r8>s0000@>>>>     ( ( ( (BBBBIIIIV$6666 6 6 6 6))))@17171717frr8ceZdZdZ d0d1dZed2dZed2dZd3dZd4dZ d4dZ d5dZ d6d7d!Z d8d"Z d9d%Zd8d&Zd:d'Zd;d)ZdQuicPacketRecoveryz0 Packet loss and congestion controller. N initial_rttr!peer_completed_address_validationbool send_probeCallable[[], None]r9r:loggerlogging.LoggerAdapter | None quic_loggerQuicLoggerTrace | Nonerr cXd|_||_g|_||_||_||_d|_d|_d|_||_ d|_ d|_ d|_ tj|_d|_d|_d|_t'||_t+||_dS)Ng?rFr#) max_ack_delayrspaces_logger _quic_logger _send_probe _pto_count _pto_total _loss_total _rtt_initial_rtt_initialized _rtt_latest_rtt_latest_rawrOrP_rtt_min _rtt_smoothed _rtt_variance'_time_of_last_sent_ack_eliciting_packetr8_ccr_pacer)r1rrrr9rrs rr2zQuicPacketRecovery.__init__js#1R.-/  '%' % #   7:4)):;;%&788 rc|jjSrc)rrAr0s rrAz"QuicPacketRecovery.bytes_in_flights x''rc|jjSrc)rrCr0s rrCz$QuicPacketRecovery.congestion_windows x))rspacerc@||jvsJ|jtd|j|jd|_d|_d|_ d|_ |j | dSdS)Nc|jSrc) in_flightrs rz2QuicPacketRecovery.discard_space..sQ[rr) rrryfilterr.valuesclearr$r+r-rr_log_metrics_updated)r1rs r discard_spacez QuicPacketRecovery.discard_spaces #### ## ((%*<*C*C*E*E F F      """ ()%   (  % % ' ' ' ' ' ) (rc<|}||jS|js)|d|jzz}|j|zSd}|jD]}|jdkr|j}|||kr|}|dS|d|jzz}||zS)Nrr) _get_loss_spacer-rget_probe_timeoutrrrr+r/)r1 loss_spacetimeoutearliestrrWs rget_loss_detection_timez*QuicPacketRecovery.get_loss_detection_times))++  !' ' 5 J,,..!T_2DEG?'I I!%[ ! !E,q00;#q8|| H  4((**a.@A'!!rc~|js d|jzS|jtd|jzt z|jzS)Nrr)rrrr~r K_GRANULARITYrr0s rrz$QuicPacketRecovery.get_probe_timeoutsK$ )t(( (  !d((-88 9  ! rcnd|_d|_d|_tj|_d|_d|_dS)a RFC 9000 9.4 / RFC 9002 5.1: when a path is changed, RTT samples from the prior path are no longer representative. ``min_rtt`` in particular MUST be reset because the new path may have higher latency, and a stale ``min_rtt`` would cause spurious loss declarations (since loss_delay scales with max(latest_rtt, smoothed_rtt) but min_rtt feeds the ack-delay floor in ``on_ack_received``). We additionally reset the smoothed RTT so a fresh sample is taken on the new path. Fr#N)rrrrOrPrrrr0s rreset_for_new_pathz%QuicPacketRecovery.reset_for_new_paths;!&"   rT ack_rangesetr ack_delayrareset_pto_countcd}|ddz }d}d} ||jkr||_t|jD]} | |krn| |vr|j| } | jrd}| js|xjdzc_| j r|j | | }| j } | j } | | D]\} }| tjg|R|dS||kr0|r-|| z }d}t!||j}t%|d|_|j|_|j|jkr |j|_|j|j|zkr|xj|zc_|jsd|_|dz |_||_nId|jzdt3|j|jz zz|_d |jzd |jzz|_|j || |j|j j|j nd}||| |rd |_|j |!|dSdS)a Update metrics as the result of an ACK being received. ``reset_pto_count`` MUST be False when the caller is a client processing an ACK in the Initial packet space and the server has not yet been confirmed to have validated the client's address (RFC 9002 6.2.1). Resetting in that case would prematurely clear the PTO backoff and let a stuck handshake under-probe. Fr NTrrg?g?g?g?rarC smoothed_rttr)log_rtt)"boundsr,sortedr.keyspopis_ack_eliciting is_pmtu_prober+rrrorjdelivery_handlersr ACKEDrrr~rrrrrrabsrr update_raterC _detect_lossrrr)r1rrrrarr largest_ackedlargest_newly_ackedlargest_sent_timer*rgdhhandlerargsrrs ron_ack_receivedz"QuicPacketRecovery.on_ack_receiveds"!$++--a014 "  55 5 5)6E &#E$6$;$;$=$=>> @ @M},, ,,+// >>*;'+$!/;55:55#5H,,V444&3#$*$4!->)+@@  1 7?$?????  & F / / /4D /00JGIt'9::I #:u55D $(#3D $-// $ 0 4=9#<<<  I-  ( (,%%/!^"%/""%*T-?%?%#&)99KKC&"D..9I1II" H ' '  ' < < < K # #"&("<!/ $     G %S)))  DO   (  % %g % 6 6 6 6 6 ) (rc|}||||dS|xjdz c_|xjdz c_||dS)Nrr )rrrrreschedule_data)r1rars ron_loss_detection_timeoutz,QuicPacketRecovery.on_loss_detection_timeoutBsu))++  !   jc  2 2 2 2 2 OOq OO OOq OO  S ) ) ) ) )rrgrc,||j|j<|jr|js|xjdz c_|jr]|jr|js|j|_|j|_|j ||j | dSdSdS)Nr ) r.r*rrr+rrjrr/rrsrr)r1rgrs rrsz!QuicPacketRecovery.on_packet_sentKs396/0  " /6+? /  ) )Q . ) )   ,& Kv/C K?E?O<:@:J7 H # #F + + + ,))+++++ , ,-,rcd}|jD]S}ttd|j}|r||||d}T|r!|j|jdd}|s|jD]}|js g}t|j D]G}|j|}|j r1|j r*| |t|dkrnH|r||||d}|s|s|dSdSdS)a Schedule some data for retransmission upon PTO expiry. Per RFC 9002 6.2.4, on PTO the sender MUST send one or two ack-eliciting packets. A PTO event MUST NOT mark prior packets as lost or trigger a congestion-control reduction. Here we requeue outstanding CRYPTO (or up to two oldest application data packets) so their content is included with the probe; the bytes are reclaimed via on_packets_rescheduled rather than on_packets_lost to avoid an unwarranted cwnd reduction. If nothing was rescheduled the caller emits a PING frame as the ack-eliciting probe. Fc|jSrc)is_crypto_packet)is rrz4QuicPacketRecovery.reschedule_data..ps !3rrraTNz(Scheduled CRYPTO data for retransmissionr)rtuplerr.r_on_packets_rescheduledrdebugrrrrappendlenr) r1racrypto_scheduledrrtapp_rescheduled to_reschedulepnpkts rrz"QuicPacketRecovery.reschedule_data_s![ ( (E33U5G5N5N5P5PQQG (,,WEs,KKK#'  K 8 L  I J J J  + + +) "  !3!8!8!:!:;;""B,R0C+" "%,,S111}--22!E +00eQT0UUU&*O               rctt|jrt|j|jn|jzt }|jtz }||z }g}d|_ |j D]\\}}||jkrnK||ks |j |kr| |9|j |z} |j |j | kr| |_ ]||||dS)zD Check whether any packets should be declared lost. Nr)r~K_TIME_THRESHOLDrrrrrr,K_PACKET_THRESHOLDr-r.itemsrjr_on_packets_lost) r1rra loss_delaypacket_thresholdtime_threshold lost_packetsr*rgpacket_loss_times rrzQuicPacketRecovery._detect_losss! ('D($*<===&     !58JJz) %*%7%=%=%?%? 7 7 !M6u999 000F4D4V4V##F++++#)#3j#@ ?*eo@P.P.P&6EO l%SAAAAArQuicPacketSpace | NonecTd}|jD]}|j||j|jkr|}|Src)rr-)r1rrs rrz"QuicPacketRecovery._get_loss_spacesA [ # #E*"eo 8L&L&L" rFc|jj|jjd}|jj|jj|d<|r||j|j|j|j|j|j |j|j d|j dd|dS)N)rAcwndrD)rmin_rttr rtt_variancerecoverymetrics_updatedcategoryeventdata) rrArCrDupdater encode_timerrrr log_event)r1rr s rrz'QuicPacketRecovery._log_metrics_updateds#x7H.   8  (#x0D    KK"&"3"?"?@P"Q"Q#0<z7QuicPacketRecovery._on_packets_lost..s NNN11;MNQNNNrrrrc3$K|] }|jV dSrcrjrs r z6QuicPacketRecovery._on_packets_lost.. s$>>qq{>>>>>>rc3$K|] }|jV dSrcrrs rrz6QuicPacketRecovery._on_packets_lost.. s=EE() EEEEEErzDPersistent congestion detected (span=%.3fs > %.3fs); collapsing cwnd)!rr.r*rrrrArirrr+rr  packet_typerrr LOSTrrrrCrrrr~rrrrrrr) r1rtrralost_packets_ccrgrrr eliciting pc_durationspans rrz#QuicPacketRecovery._on_packets_losts ; ;F    !  "6#78 3'3 H,,0AA,,,#**6222& 3v/C 3--2-- ,!++'' $ 1 = =f>P Q Q)/)=,))+++)B~%';;MGTG-2:T::::: % , H $ $_# $ > > > K # #"&("<!/ $   $ NNNNN y>>Q&&*a$"44mDDE,- #K >>I>>>>>EE-6EEEBBDk))<3 L..!2 $ +  99#>>> //.2h.H)-);0 ,))+++++K% ,% ,H-,rc`g}|D]m}|j|j=|jr|||jr|js|xjdzc_|j}||D]\}}|tj g|Rn|r7|j ||j | dSdSdS)au Requeue the contents of in-flight packets without declaring loss. Used by PTO probes (RFC 9002 6.2.4): the packet is removed from sent_packets and its delivery_handlers fire LOST so the stream sender re-enqueues the data, but congestion control is NOT informed and bytes_in_flight is reclaimed without applying a reduction. r N)r.r*rrrrr+rr rrr{rr) r1rtrrarescheduled_ccrgrrrs rrz*QuicPacketRecovery._on_packets_rescheduleds02 ; ;F"6#78 .%%f---& 3v/C 3--2--)B~%';;MGTG-2:T:::::  , H + +N ; ; ; ,))+++++ , ,,,r)rNN)rrrrrrr9r:rrrrrr )rr:)rrrr )rrr3)T) rrrr rrrarrrrr r)rgrrrrr )rrrarrr )rr)F)rtrurrrarrr )r4r5r6rr2propertyrArCrrrrrrrsrrrrrrrrrrres"&/3.2%9%9%9%9%9N(((X(***X*(((($""""2    !!!!0!% a7a7a7a7a7F****,,,,(----^!B!B!B!BF     ,L,L,L,L,\,,,,,,rr)rrrr)% __future__rloggingrOtypingrrr_hazmatrr r rr packet_builderr rrrrK_MICRO_SECONDK_SECONDrBrrYr}rqrrrrrlrkrrr8rrrrr)s"""""" **********??????????######========    #!   <<<<<<<<(ddddddddN V,V,V,V,V,V,V,V,V,V,r