o .j@sddlmZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z ddl mZddlmZd d lmZd d lmZmZd gZeed sMde_ededddddd#d"d ZdS)$) annotationsN)asynccontextmanager)AsyncGeneratorCallablecast)QuicConfiguration)QuicConnection)SessionTicketHandler)#create_optimized_datagram_transport)QuicConnectionProtocolQuicStreamHandlerconnect IPPROTO_IPV6)T) configurationcreate_protocolsession_ticket_handlerstream_handlerwait_connected local_porthoststrportintrQuicConfiguration | NonerCallable | NonerSessionTicketHandler | NonerQuicStreamHandler | Nonerboolrreturn&AsyncGenerator[QuicConnectionProtocol]cst}d} z t|d} Wn ty|} Ynw|j||tjdIdH} | dd} t| dkr@d| d| dddf} |durIt d d }|j durQ| |_ t ||d ttj tj} d }z| tjtjd| | |ddfd }W|s}| n|s| wwt|fd d| dIdH\}}tt|}z#|| |r|IdH|VW||IdH|dS||IdH|w)aa Connect to a QUIC server at the given `host` and `port`. :meth:`connect()` returns an awaitable. Awaiting it yields a :class:`~qh3.asyncio.QuicConnectionProtocol` which can be used to create streams. :func:`connect` also accepts the following optional arguments: * ``configuration`` is a :class:`~qh3.quic.configuration.QuicConfiguration` configuration object. * ``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_handler`` is a callback which is invoked by the TLS engine when a new session ticket is received. * ``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`. * ``local_port`` is the UDP port number that this client wants to bind. z::N)typerrz::ffff:r T) is_client)rrFcs dS)N)rr& connectionrrr&G/home/thesage/.local/lib/python3.10/site-packages/qh3/asyncio/client.py_s zconnect..)sock)asyncioget_running_loop ipaddress ip_address ValueError getaddrinfosocket SOCK_DGRAMlenr server_namer AF_INET6 setsockoptr IPV6_V6ONLYbindcloser rr rr wait_closed)rrrrrrrrloop local_hostr5infosaddrr+ completed transportprotocolr&r'r)rsb"           )rrrrrrrrrrrrrr rrr!r") __future__rr,r.r2 contextlibrtypingrrrquic.configurationrquic.connectionr tlsr _transportr rBr r__all__hasattrrrr&r&r&r)s,