inNdZddgZddlZddlZddlZ ddlZn #e$rdZYnwxYwddlmZm Z m Z m Z ere dGddej Z Gd dejjZ dd d d ed edede de ejeff dZde de de dejfdZdS)z Utilities. to_threadopen_connectionN) TYPE_CHECKINGAnyCallableTuple StreamWriterc,eZdZdZdejddfdZdS)StreamReaderProtocolzExtends asyncio.streams.StreamReaderProtocol for adding start_tls(). The ``start_tls()`` is based on ``asyncio.streams.StreamWriter`` introduced in Python 3.11. It provides the same functionality for older Python versions. writerreturnNch|j}||_||_|ddu|_dS)zQReplace stream writer. Args: writer: Stream Writer. sslcontextN) transport_stream_writer _transportget_extra_info _over_ssl)selfr rs S/var/www/cyber-strat/venv/lib/python3.11/site-packages/mysql/connector/aio/utils.py_replace_writerz$StreamReaderProtocol._replace_writer:s9 $ $#"11,??tK)__name__ __module__ __qualname____doc__asyncior rrrr r 3sK Lg&: Lt L L L L L Lrr c <eZdZdZddddejdededdfdZdS) r zExtends asyncio.streams.StreamWriter for adding start_tls(). The ``start_tls()`` is based on ``asyncio.streams.StreamWriter`` introduced in Python 3.11. It provides the same functionality for older Python versions. N)server_hostnamessl_handshake_timeout ssl_contextr r!r cK|jjdu}|j}|d{V|j|j|||||d{V}||_||dS)zUpgrade an existing stream-based connection to TLS. Args: ssl_context: Configured SSL context. server_hostname: Server host name. ssl_handshake_timeout: SSL handshake timeout. N) server_sider r!) _protocol_client_connected_cbdrain_loop start_tlsrr)rr"r r!r$protocol new_transports rr)zStreamWriter.start_tlsMsn9E >jjll"j22 O  #+"73             &&&&&r) rrrrssl SSLContextstrintr)rrrr r Fsl $%) '''^' ' # '  ''''''ri)limithostportr0kwdsr c Ktj}tj||}t || |j fd||fi|d{V\}}t | ||}||fS)aA wrapper for create_connection() returning a (reader, writer) pair. This function is based on ``asyncio.streams.open_connection`` and adds a custom stream reader. MySQL expects TLS negotiation to happen in the middle of a TCP connection, not at the start. This function in conjunction with ``_StreamReaderProtocol`` and ``_StreamWriter`` allows the TLS negotiation on an existing connection. Args: host: Server host name. port: Server port. limit: The buffer size limit used by the returned ``StreamReader`` instance. By default the limit is set to 64 KiB. Returns: tuple: Returns a pair of reader and writer objects that are instances of ``StreamReader`` and ``StreamWriter`` classes. )r0loop)r5cS)Nr)r*srz!open_connection..srN)rget_running_loopstreams StreamReaderr create_connectionr ) r1r2r0r3r5readerr_r r*s @rrrms.  # % %D _ ) )D ) A AF#F666H//0@0@0@0@$UUPTUUUUUUUULIq )Xvt < rfuncargskwargscKtj}tj}t j|j|g|Ri|}|d|d{VS)aQAsynchronously run function ``func`` in a separate thread. This function is based on ``asyncio.to_thread()`` introduced in Python 3.9, which provides the same functionality for older Python versions. Returns: coroutine: A coroutine that can be awaited to get the eventual result of ``func``. N)rr8 contextvars copy_context functoolspartialrunrun_in_executor)r>r?r@r5ctx func_calls rrrsp  # % %D  " $ $C!#'4A$AAA&AAI%%dI66 6 6 6 6 6 66r)NN)r__all__rrBrDr, ImportErrortypingrrrrappendr r9r r.r/r:rFuturerrrrrOs@ ) *JJJJ CCC766666666666# NN>"""LLLLL77LLL&$'$'$'$'$'7?/$'$'$'P#'8= 25GJ 7  -.> 7( 73 7# 7'. 7 7 7 7 7 7s ##