F2j*#ddlmZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddlmZmZdd lmZmZmZmZdd lmZdd lmZmZd d lmZmZdgZGddejZ edddddd%d$Z!dS)&) annotationsN)partial)Callablecast)pull_quic_header)QuicConfiguration)NetworkAddressQuicConnection) QuicHeaderQuicPacketTypeencode_quic_retryencode_quic_version_negotiation)QuicRetryTokenHandler)SessionTicketFetcherSessionTicketHandler)QuicConnectionProtocolQuicStreamHandlerservecReZdZedddddd#dZdZd$dZd%dZd&d Zd'd!Z d(d"Z dS)) QuicServerNFcreate_protocolsession_ticket_fetchersession_ticket_handlerretrystream_handler configurationr rrrSessionTicketFetcher | NonerSessionTicketHandler | NonerboolrQuicStreamHandler | NonereturnNonec||_||_tj|_i|_||_||_d|_||_ |rt|_ dSd|_ dSN) _configuration_create_protocolasyncioget_running_loop_loop _protocols_session_ticket_fetcher_session_ticket_handler _transport_stream_handlerr_retry)selfrrrrrrs Z/home/thesage/.hermes/hermes-agent/venv/lib/python3.11/site-packages/qh3/asyncio/server.py__init__zQuicServer.__init__sn, /-// ?A'=$'=$<@-  /11DKKKDKKKct|jD]}||j|jdSr')setr-valuescloseclearr0)r3protocols r4r:zQuicServer.close2shDO224455  H NN      r6 transportasyncio.BaseTransportcDttj||_dSr')rr*DatagramTransportr0)r3r=s r4connection_madezQuicServer.connection_made8sw8)DDr6data bytes | straddrr c tt|} t|d|jj\ }}}}}}} } } } n#t $rYdSwxYwt |t|||||| t| } | j T| j |jj vrA|j t| j| j|jj |dS|j| jd}d}d}|t%|dkr| jtjkr|j| jsqt/jd}|j t3| j || j| j|j|| j||dS |j|| j\}}n#t $rYdSwxYw| j}t9|j|||j|j}|||j }|!|j tE|j#| |_$tE|j%| |_&tE|j'| |_(||j| j<||j|j)<||*||dSdS) Nr)version packet_type packet_lengthdestination_cid source_cidtoken integrity_tagsupported_versions)rJrIrMi)rFrJrIoriginal_destination_cid retry_token)r"original_destination_connection_idretry_source_connection_idrr)r)r<)+rbytes_pull_quic_header_rawr(connection_id_length ValueErrorr r listrFrMr0sendtorrIrJr-getlenrGINITIALr2rKosurandomr create_tokenvalidate_tokenr r.r/r)r1rAr_connection_id_issued_connection_id_issued_handler_connection_id_retired_connection_id_retired_handler_connection_terminated_connection_terminated_handlerhost_ciddatagram_received)r3rBrD_version _packet_type_packet_length_destination_cid _source_cid_token_integrity_tag_supported_versions_encrypted_offset _end_offsetheaderr<rQrRrJ connections r4rgzQuicServer.datagram_received;sgE4   &dAt/B/WXX  #!     FF &|44(,"(#$788     N &d&9&LLL O " "/%5$*$5'+':'M      F?&&v'=tDD;?*37"  D T!!"n&<<<{&|!#AJO**)$*N'1,2,=5;5K(, (@(@ $f&66%6<5K2("13U+E'+'C'+'C J,,4+?-H  $ $T_ 5 5 56=*X666H 27>+h777H 37>+h777H 37?DOF2 33;DOJ/ 0    & &tT 2 2 2 2 2 s!&> A  A ,#G GGcidrSr<rc||j|<dSr'r-r3rtr<s r4r`z QuicServer._connection_id_issueds'r6c<|j||ksJ|j|=dSr'rvrws r4rbz!QuicServer._connection_id_retireds,s#x//// OC r6czt|jD]\}}||kr|j|=dSr')rWr-items)r3r<rtprotos r4rdz!QuicServer._connection_terminatedsKt446677 ) )JC  OC( ) )r6)rr rrrr rr!rr"rr#r$r%)r=r>r$r%)rBrCrDr r$r%)rtrSr<r)rtrSr<rr$r%)r<r) __name__ __module__ __qualname__rr5r:rArgr`rbrdr6r4rrs %;>B>B372   EEEEo3o3o3o3b((((!!!! ))))))r6rFrhoststrportintrr rrrr rr!rr"rrr$cKtj}|fd||fd{V\} } | S)aQ Start a QUIC server at the given `host` and `port`. :func:`serve` requires a :class:`~qh3.quic.configuration.QuicConfiguration` containing TLS certificate and private key as the ``configuration`` argument. :func:`serve` also accepts the following optional arguments: * ``create_protocol`` allows customizing the :class:`~asyncio.Protocol` that manages the connection. It should be a callable or class accepting the same arguments as :class:`~qh3.asyncio.QuicConnectionProtocol` and returning an instance of :class:`~qh3.asyncio.QuicConnectionProtocol` or a subclass. * ``session_ticket_fetcher`` is a callback which is invoked by the TLS engine when a session ticket is presented by the peer. It should return the session ticket with the specified ID or `None` if it is not found. * ``session_ticket_handler`` is a callback which is invoked by the TLS engine when a new session ticket is issued. It should store the session ticket for future lookup. * ``retry`` specifies whether client addresses should be validated prior to the cryptographic handshake using a retry packet. * ``stream_handler`` is a callback which is invoked whenever a stream is created. It must accept two arguments: a :class:`asyncio.StreamReader` and a :class:`asyncio.StreamWriter`. c.tS)N)rrrrrr)r)rrrrrrsr4zserve..s( '+#9#9)    r6) local_addrN)r*r+create_datagram_endpoint) rrrrrrrrloop_r<s `````` r4rrsH  # % %D55         $<6        KAx Or6)rrrrrr rrrr rr!rr"rrr$r)" __future__rr*r\ functoolsrtypingrr_hazmatrrTquic.configurationr quic.connectionr r quic.packetr r rr quic.retryrtlsrrr<rr__all__DatagramProtocolrrrr6r4rs"""""" !!!!!!!!??????222222<<<<<<<< /.....<<<<<<<<???????? )`)`)`)`)`))`)`)`)P!7:>:>(,11111111r6