@OOf &ddlZddlZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl mZmZddlmZmZddlmZmZmZmZdd lmZdd lmZmZd d lmZmZd gZ GddejBZ"edddddde#de$dededeedeede%dede"fdZ&y)N)partial)CallableDictOptionalTextUnioncast)Buffer)SMALLEST_MAX_DATAGRAM_SIZEQuicConfiguration)NetworkAddressQuicConnection)PACKET_TYPE_INITIALencode_quic_retryencode_quic_version_negotiationpull_quic_header)QuicRetryTokenHandler)SessionTicketFetcherSessionTicketHandler)QuicConnectionProtocolQuicStreamHandlerserveceZdZeddddddededeedeede d ee d dfd Z d Z d e jd dfdZdeeefded dfdZdedefdZdeded dfdZdefdZy) QuicServerNFcreate_protocolsession_ticket_fetchersession_ticket_handlerretrystream_handler configurationrrr r!r"returnc||_||_tj|_i|_||_||_d|_||_ |rt|_ yd|_ yN) _configuration_create_protocolasyncioget_event_loop_loop _protocols_session_ticket_fetcher_session_ticket_handler _transport_stream_handlerr_retry)selfr#rrr r!r"s W/var/www/premiumrankchecker/venv/lib/python3.12/site-packages/aioquic/asyncio/server.py__init__zQuicServer.__init__s^, /++- ?A'=$'=$?C- /1DKDKct|jjD]}|j|jj |j jyr&)setr,valuescloseclearr/)r2protocols r3r9zQuicServer.close0sMDOO2245 H NN    r5 transportcBttj||_yr&)r r)DatagramTransportr/)r2r<s r3connection_madezQuicServer.connection_made6sw88)Dr5dataaddrc tt|}t|} t||jj }|js|j|jjvrQ|jjt|j|j|jj|y|jj|jd}d}d}|t!|t"k\r|j$t&k(r|j(|j*st-j.d}|jjt1|j||j|j|j(j3||j||y |j(j5||j*\}}n |j}t7|j|||j8|j:} |j=| |j>}|jA|jtC|jD||_#tC|jH||_%tC|jL||_'||j|j<||j| jP<||jS||yy#t $rYywxYw#t $rYywxYw) N)r@)host_cid_length) source_ciddestination_cidsupported_versions)versionrDrEoriginal_destination_cid retry_token)r#"original_destination_connection_idretry_source_connection_idrr )r")r;)*r bytesr rr'connection_id_length ValueErrorrHrFr/sendtorrErDr,getlenr packet_typerr1tokenosurandomr create_tokenvalidate_tokenrr-r.r(r0r?r_connection_id_issued_connection_id_issued_handler_connection_id_retired_connection_id_retired_handler_connection_terminated_connection_terminated_handlerhost_ciddatagram_received) r2r@rAbufheaderr;rKrLrD connections r3r`zQuicServer.datagram_received9sE4 $ %T%8%8%M%MF NN &d&9&9&L&LL OO " "/%55$*$5$5'+':':'M'M    ??&&v'='=tD>B*6:"  D 77""&99{{&||!#AJOO**)$*NN'1,2,=,=5;5K5K(, (@(@ $f&<&6 6<5K5K2("113U+E'+'C'C'+'C'C J,,4+?+?-H  $ $T__ 56=**X6H 27>++h7H 37>++h7H 37?DOOF22 33;DOOJ// 0    & &tT 2 g   f&s#!K&.)K5& K21K25 LLcidr;c"||j|<yr&r,r2rdr;s r3rYz QuicServer._connection_id_issueds'r5cF|j||k(sJ|j|=yr&rfrgs r3r[z!QuicServer._connection_id_retireds's#x/// OOC r5ct|jjD]\}}||k(s |j|=yr&)listr,items)r2r;rdprotos r3r]z!QuicServer._connection_terminateds;t4467 )JC OOC( )r5)__name__ __module__ __qualname__rr rrrrboolrr4r9r) BaseTransportr?rrMrrr`rYr[r]r5r3rrs %;AEAE6:)"  !))= >  !))= > !!23 2 E)>)>E4E\3eE4K&8\3\3SW\3|((:P(!!$:! ! )/E)r5rFrhostportr#rrr r!r"r$cKtj}|jfd||fd{\} } | S7 w)a] Start a QUIC server at the given `host` and `port`. :func:`serve` requires a :class:`~aioquic.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:`~aioquic.asyncio.QuicConnectionProtocol` and returning an instance of :class:`~aioquic.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)r#rrr r!r")r)r#rr!rr r"sr3zserve..s '+#9#9)  r5) local_addrN)r)r*create_datagram_endpoint) rsrtr#rrr r!r"loop_r;s `````` r3rrsUH  ! ! #D55  $<6  KAx O s5A A A )'r)rU functoolsrtypingrrrrrr bufferr quic.configurationr r quic.connectionrr quic.packetrrrr quic.retryrtlsrrr;rr__all__DatagramProtocolrstrintrprrrr5r3rs >>N< /<? )M)))M)j!7=A=A(,1 1 1% 1  1 %%9: 1%%9:1 1&11r5