o .j*#@sddlmZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddlmZmZdd lmZmZmZmZdd lmZdd lmZmZd d lmZmZdgZGdddejZ edddddd%d$dZ!dS)&) annotationsN)partial)Callablecast)pull_quic_header)QuicConfiguration)NetworkAddressQuicConnection) QuicHeaderQuicPacketTypeencode_quic_retryencode_quic_version_negotiation)QuicRetryTokenHandler)SessionTicketFetcherSessionTicketHandler)QuicConnectionProtocolQuicStreamHandlerservec@s^eZdZedddddd*ddZddZd+ddZd,ddZd-d$d%Zd.d&d'Z d/d(d)Z dS)0 QuicServerNFcreate_protocolsession_ticket_fetchersession_ticket_handlerretrystream_handler configurationrrrrSessionTicketFetcher | NonerSessionTicketHandler | NonerboolrQuicStreamHandler | NonereturnNonecCsN||_||_t|_i|_||_||_d|_||_ |r"t |_ dSd|_ dSN) _configuration_create_protocolasyncioget_running_loop_loop _protocols_session_ticket_fetcher_session_ticket_handler _transport_stream_handlerr_retry)selfrrrrrrr1G/home/thesage/.local/lib/python3.10/site-packages/qh3/asyncio/server.py__init__s    zQuicServer.__init__cCs4t|jD]}|q|j|jdSr$)setr*valuescloseclearr-)r0protocolr1r1r2r62s  zQuicServer.close transportasyncio.BaseTransportcCsttj||_dSr$)rr'DatagramTransportr-)r0r9r1r1r2connection_made8szQuicServer.connection_madedata bytes | straddrr c Cstt|}zt|d|jj\ }}}}}}} } } } Wn ty#YdSwt|t|||||| t| d} | j durR| j |jj vrR|j t | j| j|jj d|dS|j| jd}d}d}|durt|dkr| jtjkr|jdur| jstd}|j t| j || j| j|j|| j|d|dSz |j|| j\}}Wn tyYdSw| j}t|j|||j|jd}|j||j d}|!|j t"|j#|d |_$t"|j%|d |_&t"|j'|d |_(||j| j<||j|j)<|dur|*||dSdS) Nr)version packet_type packet_lengthdestination_cid source_cidtoken integrity_tagsupported_versions)rDrCrGi)r@rDrCoriginal_destination_cid retry_token)r"original_destination_connection_idretry_source_connection_idrr)r)r8)+rbytes_pull_quic_header_rawr%connection_id_length ValueErrorr r listr@rGr-sendtorrCrDr*getlenrAINITIALr/rEosurandomr create_tokenvalidate_tokenr r+r,r&r.r<r_connection_id_issued_connection_id_issued_handler_connection_id_retired_connection_id_retired_handler_connection_terminated_connection_terminated_handlerhost_ciddatagram_received)r0r=r?_version _packet_type_packet_length_destination_cid _source_cid_token_integrity_tag_supported_versions_encrypted_offset _end_offsetheaderr8rKrLrD connectionr1r1r2ra;s              zQuicServer.datagram_receivedcidrMr8rcCs||j|<dSr$r*r0rnr8r1r1r2rZsz QuicServer._connection_id_issuedcCs|j||ks J|j|=dSr$rorpr1r1r2r\s z!QuicServer._connection_id_retiredcCs,t|jD] \}}||kr|j|=qdSr$)rQr*items)r0r8rnprotor1r1r2r^s z!QuicServer._connection_terminated)rrrrrrrrrr rr!r"r#)r9r:r"r#)r=r>r?r r"r#)rnrMr8r)rnrMr8rr"r#)r8r) __name__ __module__ __qualname__rr3r6r<rarZr\r^r1r1r1r2rs   q rFrhoststrportintrrrrrrrrrr rrr"c s<t}|jfdd||fdIdH\} } | 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`. cstdS)N)rrrrrr)rr1rrrrrrr1r2szserve..) local_addrN)r'r(create_datagram_endpoint) rvrxrrrrrrloop_r8r1rzr2rs$ )rvrwrxryrrrrrrrrrr rrr"r)" __future__rr'rV functoolsrtypingrr_hazmatrrNquic.configurationrquic.connectionr r quic.packetr r r r quic.retryrtlsrrr8rr__all__DatagramProtocolrrr1r1r1r2s*     )