F2jX UdZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z dZded <d Zded <ejd Zded <ejdZded<dZded<dZded<dZded<dZded<dZded<dZded<dZded<ejdkZded<ejd kZded!<ejZded"<e ed#dZ!ded$<e ed%dZ"ded&<e#ed'rej$ej%ndZ&ded(<e'd)e ed*de ed+dfDZ(ded,<d-Z)ded.<dQd3Z*dRd6Z+dRd7Z,dSd;Z-dTdAZ.dUdCZ/dVdFZ0GdGdHej1Z2 dWdXdPZ3dS)Yz: Custom asyncio DatagramTransport with optimized UDP I/O. ) annotationsN)deque)UdpSocketStatehz typing.FinalUDP_GROg UDP_SEGMENTz=H_UINT16z@i _GRO_CMSGi_DEFAULT_GRO_BUFi _MAX_GRO_BUF_HIGH_WATERMARKi@_LOW_WATERMARK@_GSO_MAX_SEGMENTSi_GSO_MAX_PAYLOAD _RECV_BURST_LIMITlinux _IS_LINUXwin32_IS_UNIX_SOL_UDP MSG_TRUNC _MSG_TRUNC MSG_CTRUNC _MSG_CTRUNC CMSG_SPACE _ANCBUFSIZEc#K|]}||V dSN).0es ^/home/thesage/.hermes/hermes-agent/venv/lib/python3.11/site-packages/qh3/asyncio/_transport.py r(7s4..  }  }}} ..EMSGSIZE WSAEMSGSIZE_MSG_TOO_BIG_ERRNOSi8'_WSAEMSGSIZE_WINERRORexcOSErrorreturnboolcV|jtvrdSt|ddtkS)z@Return True when *exc* means the datagram exceeded the path MTU.TwinerrorN)errnor,getattrr-)r.s r'_is_msg_too_bigr6Es- y'''t 3 D ) )-B BBr)sock socket.socketctsdS |ttd|ttdkS#t $rYdSwxYw)z?Enable UDP GRO on *sock*. Returns True on success (Linux only).F)r setsockoptrr getsockoptr/r7s r' enable_gror>Lse u '1---x11Q66 uusAA AActsdS |ttdt |ttS#t $rYdSwxYw)zFReturn True if the kernel supports UDP_SEGMENT on *sock* (Linux only).F)rr;rr r1r<r/r=s r'has_gsorAWsh u +t444DOOHk::;;; uusA A A'&A'ancdatalist[tuple[int, int, bytes]] int | Nonech|D]\}}}|tkr|tkrt|}|tjkr0t t|ddcS|tjkr0t t|ddcSdSdSNr)rrlenr sizeint unpack_fromr )rB cmsg_level cmsg_type cmsg_datans r'_parse_gro_segment_sizerObs,3( Iy  ! !i7&:&:IAIN""900A>>qABBBBBGL  7..y!<z%_split_gro_buffer..rs( I I I!CA $$ % I I Ir))rGrange)rPrRrNs`` r'_split_gro_bufferrYnsT CAqA--u I I I I IuQd?dZd@dZdAdZdBdZ dCdZ dDdZ dDdZ dEd"Z dDd#ZdDd$ZdDd%ZdDd&ZdDd'ZdDd(ZdFd,Zd>dFd-Zd>dGd0ZdGd1ZdFd2ZdDd3ZdDd4Zd>dHd7ZdDd8ZdId:ZdDd;ZdDd<ZxZ S)JOptimizedDatagramTransportzDatagramTransport that uses recvmmsg/sendmsg for GRO/GSO on Linux. When quinn-udp Rust is available, uses it for batched recvmmsg and GSO sendmsg. Falls back to Python recvmsg/sendmsg otherwise. )_loop_sock_sock_fd _protocol_address _gro_enabled _gso_enabled_gro_segment_size_recv_buf_size_closing_closed_paused_extra _send_queue _buffer_size_protocol_paused_writer_registered_reader_registered_protocol_supports_batch _udp_stateloopasyncio.AbstractEventLoopr7r8protocolasyncio.DatagramProtocoladdresstuple[str, int] | None gro_enabledr1 gso_enabledgro_segment_sizerIr0Nonect||_||_||_||_||_||_||_ ||_ d|_ d|_ d|_ d|_d|_t!|_d|_d|_t)|d|_|rt,nt/|d|_d|_ t5|j|_n#t6$rYnwxYw |}n#t:$rd}YnwxYw||||j|jd|_ dS)NFrdatagrams_receivedi)peernamesocketsocknamefamilytype)!super__init__rjrkfilenorlrmrnrorprqrsrtrurzr{rrwrxryhasattrr|r maxrrr}_UdpSocketState Exception getsocknamer/rrrv) selfr~r7rrrrrr __class__s r'rz#OptimizedDatagramTransport.__init__sy     ! ''!1   "'"'||_t|d|_dS)Nr)rmrr|)rrs r' set_protocolz'OptimizedDatagramTransport.set_protocols"!(/:N(O(O%%%r)c|jSr#)rxrs r'get_write_buffer_sizez0OptimizedDatagramTransport.get_write_buffer_sizes   r)c|jrdSd|_||js"|j|jddSdSNT)rs_unregister_readerrwrj call_soon_call_connection_lostrs r'closez OptimizedDatagramTransport.closesc =  F  !!! C J !;T B B B B B C Cr)c|jrdSd|_|jd|_|j|jddS)NTr)rsrwclearrxrjrrrs r'abortz OptimizedDatagramTransport.abort sV =  F      T7>>>>>r)r.Exception | Nonec|jrdSd|_|| |j|n!#t t f$rt$rYnwxYwd|_ |j dS#t$rYdSwxYw#d|_ |j w#t$rYwwxYwxYwr) rtr_unregister_writerrmconnection_lostKeyboardInterrupt SystemExit BaseExceptionr}rkrr/)rr.s r'rz0OptimizedDatagramTransport._call_connection_losts' <  F  !!! !!!  N * *3 / / / /!:.        D #DO    """""    #DO    """"    s_AB*A30B*2A33B*>B B'&B'*C3C  C CCCCc|js|jrdS |j|j|jd|_dS#t tf$rYdSwxYwr)r{rtrj add_readerrl _on_readabler/ ValueErrorrs r'_register_readerz+OptimizedDatagramTransport._register_reader&sr  " dl  F  J ! !$-1B C C C&*D # # #$    DD ,AAAc|jsdSd|_ |j|jdS#tt f$rYdSwxYwNF)r{rj remove_readerrlr/rrs r'rz-OptimizedDatagramTransport._unregister_reader/d&  F"'  J $ $T] 3 3 3 3 3$    DD 3AAc|js|jrdS |j|j|jd|_dS#t tf$rYdSwxYwr)rzrtrj add_writerrl_on_write_readyr/rrs r'_register_writerz+OptimizedDatagramTransport._register_writer8sr  " dl  F  J ! !$-1E F F F&*D # # #$    DD rc|jsdSd|_ |j|jdS#tt f$rYdSwxYwr)rzrj remove_writerrlr/rrs r'rz-OptimizedDatagramTransport._unregister_writerArrcN|jsd|_|dSdSr)rurrs r' pause_readingz(OptimizedDatagramTransport.pause_readingJs5| &DL  # # % % % % % & &r)cN|jrd|_|dSdSr)rurrs r'resume_readingz)OptimizedDatagramTransport.resume_readingOs5 < $ DL  ! ! # # # # # $ $r)datarQaddrc ||j||dS|j"|j||jdS|j|dS#t$r}t |sYd}~dSd}~wwxYwr#)rksendtornsendr/r6)rrrr&s r' _raw_sendz$OptimizedDatagramTransport._raw_sendTs  !!$-----* !!$ 66666 %%%%%   "1%%        s"A&'A& A&& B 0BB cT|jrdS|jr|||dS |||dS#t$r.||||YdSt $r%}|j|Yd}~dSd}~wwxYwr#) rsrw _queue_writerBlockingIOErrorrr/rmerror_received)rrrr.s r'rz!OptimizedDatagramTransport.sendto`s =  F      dD ) ) ) F / NN4 & & & & & * * *  ! ! # # #   dD ) ) ) ) ) ) / / / N ) )# . . . . . . . . . /sA4B'9 B'B""B'r^rSc<|js|sdS|jr|D]}|||dS|j}|||n|j}| ||t |dt|ddS#t$r3| |D]}|||YdSt$rYnwxYw|j r| ||dS|D])}| |||js|jrdS*dS)z2Send multiple datagrams, using GSO when available.Nrr:)rsrwrr}rnrrrIrrr/rp_send_gso_pythonrrt)rr^rrfstatetargets r' sendto_manyz&OptimizedDatagramTransport.sendto_manyps =    F   " / /!!%.... F  !-TT4=F! JJy#fQi..#fQi..IIIF&))+++!*77))%6666FFD     ! !)T 2 2 2 2 2"   E4(((=DLFF  s=B9C C Cc Dt|}|j}||n|j}t|D]m\}\}} t |dkr||d|n}|?||ttt |fgd|n<||ttt |fg#t$rD| ||dD] \} } | D]} | | |!YdSt$r]} t |dkr|D] } || |#t$r| | | ||| dz} || dD]}| ||||dzdD] \} } | D]}| ||!YYd} ~ dSt$r?}|j||js|jr Yd}~Yd} ~ dSYd}~d}~wwxYwn/|j| |js|jrYd} ~ dSYd} ~ gd} ~ wwxYwdS)z/Python fallback: sendmsg with UDP_SEGMENT cmsg.Nr:r)rgrkrn enumeraterGrsendmsgrr r packrrrr/indexrmrrsrt)rr^rrbr7rrVrRgroup_szgrfr.idxtaildinners r'rz+OptimizedDatagramTransport._send_gso_pythonse **z)t}(1&(9(9- -  $A$ e, u::??NN58T2222) !& W\\,5O5OPQ"  !& W\\,5O5OPQ#   %%'''$QRRj77FC!"77))%66667   u::>>!&''' NN5$7777. # # # 11333 --eT:::"'++e"4"4q"8C(-cdd >> $ 1 1$ = = = =*0Q/??Q)*!?!?A$($5$5a$>$>$>$>!?"FFFFFFFF&''' N99%@@@#}' ' &'''''''$N11#666} - /- - s\B-C))A J6 JJE10J1BI J I (I JJI ,JJc4|j||f|xjt|z c_|jtkrL|jsGd|_ |jdS#ttf$rt$rYdSwxYwdSdSr) rwrarxrGrryrm pause_writingrrr)rrrs r'rz'OptimizedDatagramTransport._queue_writes t --- SYY&   / /8M /$(D ! ,,.....%z2         0 / / /sA44BBc|j}|j}|r$|d\}} |||n#t$rYdSt$rY4t$r}||xjt|zc_| |j |n!#ttf$rt$rYnwxYw|js|jrYd}~nUYd}~d}~wwxYw||xjt|zc_||$|sD||jr+|js&|j|jddSdSdSdSrF)rwrrInterruptedErrorr/popleftrxrG_maybe_resume_protocolrmrrrrrsrtrrjrr)rqueueraw_sendrrr.s r'rz*OptimizedDatagramTransport._on_write_readys > *qJD$ t$$$$"   #       !!SYY.!!++---N11#6666):6$D=DLEEEE  MMOOO   T *    ' ' ) ) )1 *4 G  # # % % %} GT\ G $$T%?FFFFF G G G G G GsE + C+ C+ C+ AC&B-,C&-C C& C  C&&C+c|jrU|jtkrGd|_ |jdS#t t f$rt$rYdSwxYwdSdSr)ryrxrrmresume_writingrrrrs r'rz1OptimizedDatagramTransport._maybe_resume_protocols   T%6.%H%H$)D ! --/////%z2           %H%Hs;AAwaiterasyncio.Future[None] | Nonec|j|jj||j|j|"|j|jddSdSr#)rjrrmconnection_mader set_result)rrs r'_startz!OptimizedDatagramTransport._startsg T^;TBBB T2333   J !2D 9 9 9 9 9  r)c|jrdS|j}|||dS|jr|dS|dSr#)rsr} _recv_rustro_recv_gro_python _recv_plain)rrs r'rz'OptimizedDatagramTransport._on_readablesq =  F   OOE " " " " "     ! ! # # # # #        r)rc0|j}|jr|jnd}|j}d}|gg}d}t t D]=} |\} } n#t $rYn"wxYw| sn|| | }>d}|r |||nW|j} t t D]9} |\} }n#t $rYdSwxYw| sdS| D]} | | |:d}|r/|j s*|j r%|j |j dSdSdSdS)z!Batch-receive via quinn-udp Rust.NFT)rmr|rrecvrXrr/extenddatagram_receivedrsr{rjrr) rrrbatch_cb_recv hit_limit all_segmentsr_segmentsarsegs r'rz%OptimizedDatagramTransport._recv_rusts>, H ' '     (*L#D,-- ! !"'%''KHaaEEE##H---  -t,,, ( : ,-- ! !%*UWWNHddFFFFF#11C%%c400001!   4T] 4t/F 4 J !2 3 3 3 3 3 4 4 4 4 4 4s# A  AA( B66 CCcF|j}|j}ttD]}} |d\}}nF#t $rYdSt $rY5t$r!}||Yd}~dSd}~wwxYw|sdS| ||~dS)Nr) rkrmrXrrecvfromrrr/rr)rr7rrrrr.s r'rz&OptimizedDatagramTransport._recv_plainCsz>()) 3 3A !]]511 dd"   #      '',,,    & &tT 2 2 2 2 3 3s!? B B B!A==Bc|jj}|j}|j}|jr|jnd}|j}t}ttD]T} ||j |\}} } } ni#t$rYdSt$rY4t$rD} || n!#tt f$rt"$rYnwxYwYd} ~ dSd} ~ wwxYw|sdS| t$zr0|j } | t&krt)| dzt&|_ | t*zr ||| t-| }| ||| |dkr|n|}t/||kr||| #t1||}|||| C|D]}||| V|js(|jr#|j|jdSdSdS)z/Python fallback: recvmsg with GRO cmsg parsing.Nrr)rkrecvmsgrmrr|rrqr!rXrrrrrr/rrrrrrminrrOrGrYrsr{rjrr)r sock_recvmsgrrrdefault_segment_size ancbufsizerrrBflagsrr.bufsizeparsedrRrr s r'rz+OptimizedDatagramTransport._recv_gro_pythonTsz) >$6, H ' '  $5 ()). 1. 1A -9\'..*gudd#   #      ++C0000):6$D  z! -\))*-gk<*H*HD'{" !!$---,W55F~!!$---%+aZZ665IL4yyL((!!$---(|<B??C  C)r~rr7r8rrrrrr1rr1rrIr0rr#)rrrrr0r)r0r1)r0r)rrr0r)r0rI)r0r)r.rr0r)rrQrrr0r)r^rSrrr0r)rrr0r)rrr0r)!__name__ __module__ __qualname____doc__ __slots__rrrrrrrrrrrrrrrrrrrrrrrrrrr __classcell__)rs@r'riris{ I.5 5 5 5 5 5 n.....PPPP!!!!CCCC????&&&&& $$$$    ///// """""H3333j    !G!G!G!GF:::::    -4-4-4-4^3333">4>4>4>4>4>4>4>4r)rir@r~rprotocol_factory-typing.Callable[[], asyncio.DatagramProtocol]r:tuple[asyncio.DatagramTransport, asyncio.DatagramProtocol]c ^Kt|}t|}ts|||d{VS|d|}t |||d|||}|}|||d{V||fS)z?Create a DatagramTransport with optimized UDP I/O if available.r=NF)r~r7rrrrr)r>rArcreate_datagram_endpoint setblockingri create_futurer) r~rr7rrrr transportrs r'#create_optimized_datagram_transportr#sT""K$--K P223C$2OOOOOOOOOU!!H*  )I   ! !F V LLLLLLL h r))r.r/r0r1)r7r8r0r1)rBrCr0rD)rPrQrRrIr0rS)rHrIr0rI)r^rSr0r_)r@) r~rrrr7r8rrIr0r)4r __future__rasyncior4rstructsystyping collectionsr_hazmatrrr__annotations__r Structr r r rrrrrrplatformrrSOL_UDPrr5rrrr rHr! frozensetr,r-r6r>rArOrYr]rgDatagramTransportrir#r$r)r'r1s#""""" 877777 & d++++++'&--- ----"'&&&&# ####!*))))(((((#%$$$$!&&&&&#%$$$$,'1 111100000''''"76;:: ::::#GFL!<< <<<<)0)F)FMFin%%%A %.I.. z4((}d++...%%',++++CCCC    JJJJAAAA@t4t4t4t4t4!:t4t4t4v! r)