o .jX@s UdZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z dZded <d Zded <ed Zded <edZded<dZded<dZded<dZded<dZded<dZded<dZded<dZded<ejdkZded<ejd kZded!<ejZded"<e ed#dZ!ded$<e ed%dZ"ded&<e#ed're$ej%ndZ&ded(<e'd)d*e ed+de ed,dfDZ(ded-<d.Z)ded/<dZd4d5Z*d[d8d9Z+d[d:d;Z,d\d?d@Z-d]dFdGZ.d^dIdJZ/d_dMdNZ0GdOdPdPej1Z2 Qd`dadXdYZ3dS)bz: 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 _ANCBUFSIZEccs|] }|dur|VqdSN).0er"r"K/home/thesage/.local/lib/python3.10/site-packages/qh3/asyncio/_transport.py 7sr&EMSGSIZE WSAEMSGSIZE_MSG_TOO_BIG_ERRNOSi8'_WSAEMSGSIZE_WINERRORexcOSErrorreturnboolcCs|jtvrdSt|ddtkS)z@Return True when *exc* means the datagram exceeded the path MTU.TwinerrorN)errnor)getattrr*)r+r"r"r%_is_msg_too_bigEs r2sock socket.socketcCs>tsdSz|ttd|ttdkWStyYdSw)z?Enable UDP GRO on *sock*. Returns True on success (Linux only).F)r setsockoptrr getsockoptr,r3r"r"r% enable_groL r9cCs>tsdSz|ttdt|ttWStyYdSw)zFReturn True if the kernel supports UDP_SEGMENT on *sock* (Linux only).F)rr6rr r.r7r,r8r"r"r%has_gsoWr:r<ancdatalist[tuple[int, int, bytes]] int | NonecCsv|D]6\}}}|tkr8|tkr8t|}|tjkr$tt|ddS|tjkr5tt|ddSdSqdSNr)rrlenr sizeint unpack_fromr )r= cmsg_level cmsg_type cmsg_datanr"r"r%_parse_gro_segment_sizebs  rIbufbytes segment_sizerC list[bytes]cs:t}dks |krgSfddtd|DS)Nrcsg|] }||qSr"r")r#irJrLr"r% rsz%_split_gro_buffer..)rArange)rJrLrHr"rOr%_split_gro_buffernsrRrBcCs0|dkrtSt|}|dkrdS|tkr|StSNrr5)rr)rBcapr"r"r%_max_segments_forus rU datagramslist[tuple[int, list[bytes]]]cCs|sgSg}|d}t|}|g}t|}|ddD]C}t|}||kr1t||kr1||q||krLt||krL|||||fg}d}q|rU|||f|}|g}t|}q|rh|||f|SrS)rArUappend)rVgroupsfirst current_size current_grouprTdgramrBr"r"r%_group_for_gso~s0   r^cs2eZdZdZdZdXfdd ZdYdZddZd[ddZd\ddZd]d d!Z d^d"d#Z d_d$d%Z d_d&d'Z d`d*d+Z d_d,d-Zd_d.d/Zd_d0d1Zd_d2d3Zd_d4d5Zd_d6d7Zdad;d<ZdYdad=d>ZdYdbdAdBZdbdCdDZdadEdFZd_dGdHZd_dIdJZdYdcdMdNZd_dOdPZdddRdSZd_dTdUZd_dVdWZZ S)eOptimizedDatagramTransportzDatagramTransport 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.AbstractEventLoopr3r4protocolasyncio.DatagramProtocoladdresstuple[str, int] | None gro_enabledr. gso_enabledgro_segment_sizerCr-Nonec st||_||_||_||_||_||_||_ ||_ d|_ d|_ d|_ d|_d|_t|_d|_d|_t|d|_|rBtnt|d|_d|_zt|j|_Wn ty\Ynwz|}Wn tynd}Ynw||||j|jd|_ dS)NFrdatagrams_receivedi)peernamesocketsocknamefamilytype)!super__init__r`rafilenorbrcrdrerfrgrirjrkrprqrrmrnrohasattrrrr maxrhrs_UdpSocketState Exception getsocknamer,rrrl) selfrtr3rvrxrzr{r|r __class__r"r%rsJ       z#OptimizedDatagramTransport.__init__Nnamestrdefault typing.AnycCs|j||Sr!)rlget)rrrr"r"r%get_extra_infosz)OptimizedDatagramTransport.get_extra_infocC|jSr!)rirr"r"r% is_closingz%OptimizedDatagramTransport.is_closingasyncio.BaseProtocolcCrr!)rcrr"r"r% get_protocolrz'OptimizedDatagramTransport.get_protocolcCs||_t|d|_dS)Nr~)rcrrr)rrvr"r"r% set_protocolsz'OptimizedDatagramTransport.set_protocolcCrr!)rnrr"r"r%get_write_buffer_sizerz0OptimizedDatagramTransport.get_write_buffer_sizecCs6|jrdSd|_||js|j|jddSdSNT)ri_unregister_readerrmr` call_soon_call_connection_lostrr"r"r%closesz OptimizedDatagramTransport.closecCs4|jrdSd|_|jd|_|j|jddS)NTr)rirmclearrnr`rrrr"r"r%abort s  z OptimizedDatagramTransport.abortr+Exception | NonecCs|jrdSd|_||z2z|j|Wnttfy#ty+YnwWd|_z|j WdSt yBYdSwd|_z|j Wwt yVYwwr) rjr_unregister_writerrcconnection_lostKeyboardInterrupt SystemExit BaseExceptionrsrarr,)rr+r"r"r%rs2   z0OptimizedDatagramTransport._call_connection_lostc CH|js|jrdSz|j|j|jd|_WdSttfy#YdSwr)rqrjr` add_readerrb _on_readabler, ValueErrorrr"r"r%_register_reader&  z+OptimizedDatagramTransport._register_readerc C>|jsdSd|_z |j|jWdSttfyYdSwNF)rqr` remove_readerrbr,rrr"r"r%r/z-OptimizedDatagramTransport._unregister_readerc Crr)rprjr` add_writerrb_on_write_readyr,rrr"r"r%_register_writer8rz+OptimizedDatagramTransport._register_writerc Crr)rpr` remove_writerrbr,rrr"r"r%rArz-OptimizedDatagramTransport._unregister_writercCs|js d|_|dSdSr)rkrrr"r"r% pause_readingJ z(OptimizedDatagramTransport.pause_readingcCs|jr d|_|dSdSr)rkrrr"r"r%resume_readingOrz)OptimizedDatagramTransport.resume_readingdatarKaddrc Csz'|dur|j||WdS|jdur|j||jWdS|j|WdSty?}z t|s4WYd}~dSd}~wwr!)rasendtordsendr,r2)rrrr$r"r"r% _raw_sendTs z$OptimizedDatagramTransport._raw_sendc Cs|jrdS|jr|||dSz |||WdSty-||||YdStyE}z |j|WYd}~dSd}~wwr!) rirm _queue_writerBlockingIOErrorrr,rcerror_received)rrrr+r"r"r%r`s  z!OptimizedDatagramTransport.sendtorVrMcCs|js|sdS|jr|D]}|||q dS|j}|dur_|dur$|n|j}|dur_z||t|dt|dWdStyV| |D]}|||qJYdSt y^Ynw|j rj| ||dS|D]}| |||jsz|jr}dSqldS)z2Send multiple datagrams, using GSO when available.Nrr5)rirmrrsrdrrrCrrr,rf_send_gso_pythonrrj)rrVrr]statetargetr"r"r% sendto_manyps:     z&OptimizedDatagramTransport.sendto_manycCsNt|}|j}|dur |n|j}t|D]\}\}}z2t|dkr+||d|n!|dur?||ttt |fgd|n ||ttt |fgWqt yr| ||dD]\} } | D]} | | |qdq^YdSty$} zt|dkr|D]} z|| |Wqt y| | | ||| d} || dD]}| ||q||ddD]\} } | D]}| ||qqYWYd} ~ dSty}z |j||js|jrWYd}~WYd} ~ dSWYd}~qd}~wwn|j| |js|jrWYd} ~ dSWYd} ~ qd} ~ wwdS)z/Python fallback: sendmsg with UDP_SEGMENT cmsg.Nr5r)r^rard enumeraterArsendmsgrr r packrrrr,indexrcrrirj)rrVrrYr3rrNrLgroup_szgr]r+idxtaildinnerr"r"r%rsp        z+OptimizedDatagramTransport._send_gso_pythonc Csx|j||f|jt|7_|jtkr8|js:d|_z|jWdStt fy.t y7YdSwdSdSr) rmrXrnrArrorc pause_writingrrr)rrrr"r"r%rs z'OptimizedDatagramTransport._queue_writec Cs@|j}|j}|r|d\}}z|||Wn[tyYdSty'Yqtyp}z>||jt|8_|z|j |Wnt t fyQt yYYnw|js`|jrfWYd}~nWYd}~qd}~ww||jt|8_||s|s||jr|js|j|jddSdSdSdSr@)rmrrInterruptedErrorr,popleftrnrA_maybe_resume_protocolrcrrrrrirjrr`rr)rqueueraw_sendrrr+r"r"r%rsH        z*OptimizedDatagramTransport._on_write_readyc CsV|jr'|jtkr)d|_z|jWdSttfyty&YdSwdSdSr)rornrrcresume_writingrrrrr"r"r%rs z1OptimizedDatagramTransport._maybe_resume_protocolwaiterasyncio.Future[None] | NonecCs@|j|jj||j|j|dur|j|jddSdSr!)r`rrcconnection_mader set_result)rrr"r"r%_starts z!OptimizedDatagramTransport._startcCsD|jrdS|j}|dur||dS|jr|dS|dSr!)rirs _recv_rustre_recv_gro_python _recv_plain)rrr"r"r%rs  z'OptimizedDatagramTransport._on_readablerc Cs|j}|jr |jnd}|j}d}|durFg}d}ttD]}z|\} } Wn ty/Ynw| s4n || | }qd}|rE|||n.|j} ttD]$}z|\} }Wn tyaYdSw| sgdS| D]} | | |qiqMd}|r|j s|j r|j |j dSdSdSdS)z!Batch-receive via quinn-udp Rust.NFT)rcrrr~recvrQrr,extenddatagram_receivedrirqr`rr) rrrvbatch_cb_recv hit_limit all_segmentsr_segmentsarsegr"r"r%rsP        z%OptimizedDatagramTransport._recv_rustc Cs|j}|j}ttD]B}z |d\}}Wn+ty YdSty(Yq ty@}z ||WYd}~dSd}~ww|sFdS| ||q dS)Nr) rarcrQrrecvfromrrr,rr)rr3rvrrrr+r"r"r%rCs$    z&OptimizedDatagramTransport._recv_plainc Cs|jj}|j}|j}|jr|jnd}|j}t}tt D]}z ||j |\}} } } Wn@t y4YdSt y<Yqt yi} z"z|| WnttfyTty\YnwWYd} ~ dSd} ~ ww|sodS| t@r|j } | tkrt| dt|_ q| t@r||| qt| }|dur||| q|dkr|n|}t||kr||| qt||}|dur||| q|D]}||| qq|js|jr|j|jdSdSdS)z/Python fallback: recvmsg with GRO cmsg parsing.Nrr)rarecvmsgrcrrrr~rgr rQrrhrrr,rrrrrr minrrIrArRrirqr`rr)r sock_recvmsgrvrrdefault_segment_size ancbufsizerrr=flagsrr+bufsizeparsedrLrrr"r"r%rTsp             z+OptimizedDatagramTransport._recv_gro_python)rtrur3r4rvrwrxryrzr.r{r.r|rCr-r}r!)rrrrr-r)r-r.)r-r)rvrr-r})r-rC)r-r})r+rr-r})rrKrrr-r})rVrMrrr-r})rrr-r})rrr-r})!__name__ __module__ __qualname____doc__ __slots__rrrrrrrrrrrrrrrrrrrrrrrrrrr __classcell__r"r"rr%r_s< 7            $ 5 #  /r_r;rtruprotocol_factory-typing.Callable[[], asyncio.DatagramProtocol]r|:tuple[asyncio.DatagramTransport, asyncio.DatagramProtocol]c stt|}t|}ts|j||dIdHS|d|}t|||d|||d}|}|||IdH||fS)z?Create a DatagramTransport with optimized UDP I/O if available.r8NF)rtr3rvrxrzr{r|)r9r<rcreate_datagram_endpoint setblockingr_ create_futurer) rtrr3r|rzr{rv transportrr"r"r%#create_optimized_datagram_transports(   r)r+r,r-r.)r3r4r-r.)r=r>r-r?)rJrKrLrCr-rM)rBrCr-rC)rVrMr-rW)r;) rtrurrr3r4r|rCr-r)4r __future__rasyncior0rstructsystyping collectionsr_hazmatrrr__annotations__r Structr r r r rrrrrplatformrrSOL_UDPrr1rrrrrBr frozensetr)r*r2r9r<rIrRrUr^DatagramTransportr_rr"r"r"r%s^                    ~