File: //var/opt/nydus/ops/mysql/connector/aio/__pycache__/network.cpython-39.pyc
a
�,�h`a � @ sJ d Z ddgZddlZddlZddlZz"ddlZejejejej d�Z
W n e y\ dZY n0 ddl m
Z
mZ ddlmZ ddlmZmZmZmZmZ d d
lmZmZmZmZ d d l m Z m Z m Z m Z d d
l!m"Z"m#Z# e$e%d�dd�Z&G dd� de
�Z'G dd� de'�Z(G dd� de(�Z)G dd� de
�Z*G dd� de*�Z+G dd� de*�Z,dS )zFModule implementing low-level socket communication with MySQL servers.�MySQLTcpSocket�MySQLUnixSocket� N)�TLSv1�TLSv1.1�TLSv1.2�TLSv1.3)�ABC�abstractmethod)�deque)�Any�Deque�List�Optional�Tuple� )�InterfaceError�NotSupportedError�OperationalError�ProgrammingError)� COMPRESSED_PACKET_HEADER_LENGTH�MAX_PAYLOAD_LENGTH�MIN_COMPRESS_LENGTH�PACKET_HEADER_LENGTH� )� StreamWriter�open_connection)�err�returnc C s | j st| �S | j � d| j� �S )z`Reformat the IOError error message.
This function reformats the IOError error message.
� )�errno�str�strerror)r � r" �N/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/connector/aio/network.py� _strioerrorG s r$ c @ sN e Zd ZdZed eeeee ee dd�dd��Z
ee j ee
d�dd��ZdS )
�
NetworkBrokeraP Broker class interface.
The network object is a broker used as a delegate by a socket object. Whenever the
socket wants to deliver or get packets to or from the MySQL server it needs to rely
on its network broker (netbroker).
The netbroker sends `payloads` and receives `packets`.
A packet is a bytes sequence, it has a header and body (referred to as payload).
The first `PACKET_HEADER_LENGTH` or `COMPRESSED_PACKET_HEADER_LENGTH`
(as appropriate) bytes correspond to the `header`, the remaining ones represent the
`payload`.
The maximum payload length allowed to be sent per packet to the server is
`MAX_PAYLOAD_LENGTH`. When `send` is called with a payload whose length is greater
than `MAX_PAYLOAD_LENGTH` the netbroker breaks it down into packets, so the caller
of `send` can provide payloads of arbitrary length.
Finally, data received by the netbroker comes directly from the server, expect to
get a packet for each call to `recv`. The received packet contains a header and
payload, the latter respecting `MAX_PAYLOAD_LENGTH`.
N��writer�address�payload�
packet_number�compressed_packet_numberr c � s dS )a� Send `payload` to the MySQL server.
If provided a payload whose length is greater than `MAX_PAYLOAD_LENGTH`, it is
broken down into packets.
Args:
sock: Object holding the socket connection.
address: Socket's location.
payload: Packet's body to send.
packet_number: Sequence id (packet ID) to attach to the header when sending
plain packets.
compressed_packet_number: Same as `packet_number` but used when sending
compressed packets.
Raises:
:class:`OperationalError`: If something goes wrong while sending packets to
the MySQL server.
Nr" )�selfr' r( r) r* r+ r" r" r# �writeg s zNetworkBroker.write��readerr( r c � s dS )a) Get the next available packet from the MySQL server.
Args:
sock: Object holding the socket connection.
address: Socket's location.
Returns:
packet: A packet from the MySQL server.
Raises:
:class:`OperationalError`: If something goes wrong while receiving packets
from the MySQL server.
:class:`InterfaceError`: If something goes wrong while receiving packets
from the MySQL server.
Nr" )r, r/ r( r" r" r# �read� s zNetworkBroker.read)NN)�__name__�
__module__� __qualname__�__doc__r r r �bytesr �intr- �asyncio� StreamReader� bytearrayr0 r" r" r" r# r% O s ��r% c @ s� e Zd ZdZdd�dd�Zeeeeef d�dd��Z de
e dd �d
d �Z e e
edd �d
d�Zdejeed�dd�Zde e
ee
e e
e dd�dd�Zeje
ed�dd�ZdS ) �NetworkBrokerPlain�,Broker class for MySQL socket communication.N�r c C s
d| _ d S �N�����_pktnr�r, r" r" r# �__init__� s zNetworkBrokerPlain.__init__��pktr c C sB t | �tk rtd��t�d| dd� d �d | d }}||fS )�-Recover the header information from a packet.�3Can't recover header info from an incomplete packet�<Ir � � )�lenr �
ValueError�struct�unpack)rD �pll�seqidr" r" r# �
get_header� s �z NetworkBrokerPlain.get_header��next_idr c C s0 |du r| j d7 _ n|| _ | j d; _ dS �z:Set the given packet id, if any, else increment packet id.Nr � r? �r, rR r" r" r# �_set_next_pktnr� s z"NetworkBrokerPlain._set_next_pktnr�r' r( rD r c
� s� z |� |� |�� I dH W nh tyV } z"td|t|�fd�|�W Y d}~n8d}~0 ty� } ztdd�|�W Y d}~n
d}~0 0 dS )z!Write packet to the comm channel.N� �r �valuesi� �r )r- �drain�IOErrorr r$ �AttributeError)r, r' r( rD r r" r" r# �
_write_pkt� s
��z NetworkBrokerPlain._write_pktr )r/ �sizer c � sH t d�}t|�|k rD|�|t|� �I dH }|s:tdd��||7 }q|S )z(Read `size` bytes from the comm channel.� Ni� r[ )r9 rJ r0 r )r, r/ r` rD �chunkr" r" r# � _read_chunk� s
z NetworkBrokerPlain._read_chunkr& c
� s� | � |� d}tt|�t �D ]F}| �||dt�d| j� |||t � �I dH | � � |t7 }q | �||t�dt|�| �dd� t�d| j� ||d� �I dH dS )z�Send payload to the MySQL server.
If provided a payload whose length is greater than `MAX_PAYLOAD_LENGTH`, it is
broken down into packets.
r � ����<BNrG rH )rV �rangerJ r r_ rL �packr@ )r, r' r( r) r* r+ �offset�_r" r" r# r- � s.
���
�
��zNetworkBrokerPlain.writer. c
� s~ z>| j |td�I dH }| �|�\}| _|| j ||d�I dH W S tyx } z"td|t|�fd�|�W Y d}~n
d}~0 0 dS )z+Receive `one` packet from the MySQL server.�r` NrX rY )rc r rP r@ r] r r$ )r, r/ r( �headerZ payload_lenr r" r" r# r0 � s ��zNetworkBrokerPlain.read)N)r )NN)r1 r2 r3 r4 rB � staticmethodr5 r r6 rP r rV r r r_ r7 r8 r9 rc r- r0 r" r" r" r# r: � s,
�� �� &r: c s� e Zd ZdZdd�� fdd� Zeeeee d�dd��Z eee
eeef d �d
d ��Z de e dd �d
d�Z
eeedd�� fdd� Zd eeee e e e dd�� fdd�
Zejedd�� fdd� Zejeed�� fdd� Z� ZS ) �NetworkBrokerCompressedr; Nr< c s t � �� d| _t� | _d S r= )�superrB �_compressed_pktnrr
� _queue_readrA �� __class__r" r# rB s
z NetworkBrokerCompressed.__init__)r) �pktnrr c C s� d}g }t t| �t �D ]>}|�dt�d|� | ||t � � |d d }|t7 }q|�t�dt| �| �dd� t�d|� | |d� � |S ) z2Prepare a payload for sending to the MySQL server.r rd re r rT rG rH N)rf rJ r �appendrL rg )r) rs rh Zpktsri r" r" r# �_prepare_packets s(
���
�
��z(NetworkBrokerCompressed._prepare_packetsrC c C sb t | �tk rtd��t�d| dd� d �d | d t�d| dd� d �d }}}|||fS )rE rF rG r rH rI � � )rJ r rK rL rM )rD �compressed_pllrO �uncompressed_pllr" r" r# rP ! s �
z"NetworkBrokerCompressed.get_headerrQ c C s0 |du r| j d7 _ n|| _ | j d; _ dS rS )ro rU r" r" r# �_set_next_compressed_pktnr/ s z2NetworkBrokerCompressed._set_next_compressed_pktnrrW c � sb t �|�}t�dt|��dd� t�d| j� t�dt|��dd� | }t� �|||�I dH S )z1Compress packet and write it to the comm channel.rG r rH re N)�zlib�compressrL rg rJ ro rn r_ )r, r' r( rD �compressed_pktrq r" r# r_ 7 s
����z"NetworkBrokerCompressed._write_pktr& c � s | � |� | �|� td��| �|| j��}t|�tt kr�d}t t|�t �D ]4}| �
|||||t � �I dH | �� |t7 }qP| �
||||d� �I dH nrt|�t kr�| �
|||�I dH nPt � �
||t
�dt|��dd� t
�d| j� t
�dd�dd� | �I dH dS )z�Send `payload` as compressed packets to the MySQL server.
If provided a payload whose length is greater than `MAX_PAYLOAD_LENGTH`, it is
broken down into packets.
ra r NrG rH re )rV rz r9 �joinru r@ rJ r r rf r_ r rn rL rg ro ) r, r' r( r) r* r+ Z payload_preprh ri rq r" r# r- B s2
�
����z NetworkBrokerCompressed.write)r/ rx r c � s� t t�t� j||d�I dH ��}d}|t|�k r�t�d|||t d � d �d }t| t|�| kr�t� j|t d�I dH }| �
|�\}| _ }t� j||d�I dH }||dkr�|nt�|�7 }| j �
|||t | � � |t| 7 }q$dS )z&Handle reading of a compressed packet.rj Nr rG r rI )r9 r{ �
decompressrn rc rJ rL rM r r rP ro rp rt ) r, r/ rx rD rh rN rk ry r} rq r" r# �_read_compressed_pkty s6 � ��� ��� z,NetworkBrokerCompressed._read_compressed_pktr. c
� s� | j s�zft� j|td�I dH }| �|�\}| _}|dkrX| j �t� j||d�I dH � n| �||�I dH W n: ty� } z"t d|t
|�fd�|�W Y d}~n
d}~0 0 | j s�dS | j � � }|d | _ |S )z{Receive `one` or `several` packets from the MySQL server, enqueue them, and
return the packet at the head.
rj Nr rX rY rH )
rp rn rc r rP ro rt r� r] r r$ �popleftr@ )r, r/ r( rk rx ry r rD rq r" r# r0 � s4 � �� ��
z NetworkBrokerCompressed.read)N)NN)r1 r2 r3 r4 rB rl r5 r6 r
ru r rP r rz r r r_ r- r7 r8 r� r9 r0 �
__classcell__r" r" rq r# rm s,
��8�(rm c
@ s e Zd ZdZdd�dd�Zeed�dd��Zee dd�d d
��Z
dd�d d �Z e d�d
d�Z
edd�dd�Zdd�dd�Zejdd�dd�Zd eee ee dd�dd�Zed�dd�Zdddd d g g fee ee ee ee ee eee eee ejd �d d �ZdS )!� MySQLSocketz�MySQL socket communication interface.
Examples:
Subclasses: network.MySQLTCPSocket and network.MySQLUnixSocket.
Nr< c C s* d| _ d| _d| _d| _t� | _d| _dS )zsNetwork layer where transactions are made with plain (uncompressed) packets
is enabled by default.
NF)�_reader�_writer�_connection_timeout�_addressr: �
_netbroker�
_is_connectedrA r" r" r# rB � s zMySQLSocket.__init__c C s | j S )zSocket location.)r� rA r" r" r# r( � s zMySQLSocket.address��kwargsr c � s dS )zOpen the socket.Nr" �r, r� r" r" r# r � s zMySQLSocket.open_connectionc � sB | j r8| j �� | j jdur(| j j�� | j �� I dH d| _dS )zClose the connection.NF)r� �close� transport�abort� wait_closedr� rA r" r" r# �close_connection� s
z MySQLSocket.close_connectionc C s | j S )z�Check if the socket is connected.
Return:
bool: Returns `True` if the socket is connected to MySQL server.
)r� rA r" r" r# � is_connected� s zMySQLSocket.is_connected)�timeoutr c C s
|| _ dS )zSet the connection timeout.N)r� )r, r� r" r" r# �set_connection_timeout� s z"MySQLSocket.set_connection_timeoutc C s t � | _dS )zIEnable network layer where transactions are made with compressed packets.N)rm r� rA r" r" r# �switch_to_compressed_mode� s z%MySQLSocket.switch_to_compressed_mode)� ssl_contextr c � sD | j dusJ �| j j�d�}|jdkr.td��| j �|�I dH dS )a? Upgrade an existing stream-based connection to TLS.
The `start_tls()` method from `asyncio.streams.StreamWriter` is only available
in Python 3.11. This method is used as a workaround.
The MySQL TLS negotiation happens in the middle of the TCP connection.
Therefore, passing a socket to open connection will cause it to negotiate
TLS on an existing connection.
Args:
ssl_context: The SSL Context to be used.
Raises:
RuntimeError: If the transport does not expose the socket instance.
N�socketr z,SSL is not supported when using Unix sockets)r� r� �get_extra_info�familyr � start_tls)r, r� r� r" r" r# �
switch_to_ssl� s
zMySQLSocket.switch_to_ssl)r) r* r+ r c � s$ | j j| j| j|||d�I dH dS )z!Send packets to the MySQL server.)r* r+ N)r� r- r� r( )r, r) r* r+ r" r" r# r- s �zMySQLSocket.writec � s | j �| j| j�I dH S )z#Read packets from the MySQL server.N)r� r0 r� r( rA r" r" r# r0 & s zMySQLSocket.readF)�ssl_ca�ssl_cert�ssl_key�ssl_verify_cert�ssl_verify_identity� tls_versions�tls_cipher_suitesr c C s | j stdd��tdu r td���zx|r�|jdd� |d }t| } t�| �}
|dkr�d |vrn|
jtjO _d
|vr�|
jtj O _d |vr�|
jtj
O _nt� � }
||
_ |r�tj
|
_n|r�tj|
_ntj|
_|
�� |�r,z|
�|� W n< ttjf�y* } z td | � ��| �W Y d} ~ n
d} ~ 0 0 |�r�z|
�||� W n< ttjf�y~ } z td
| � ��| �W Y d} ~ n
d} ~ 0 0 |�r�|
�d�|�� |
W S t�y� } ztd�| �W Y d} ~ nHd} ~ 0 tttjtjf�y } ztt | ��| �W Y d} ~ n
d} ~ 0 0 dS )zBuild a SSLContext.i r[ Nz&Python installation has no SSL supportT)�reverser r r r r zInvalid CA Certificate: zInvalid Certificate/Key: �:) r� r �ssl� RuntimeError�sort� TLS_VERSIONS�
SSLContext�options�
OP_NO_TLSv1_2�
OP_NO_TLSv1_1� OP_NO_TLSv1�create_default_context�check_hostname�
CERT_REQUIRED� verify_mode�
CERT_OPTIONAL� CERT_NONE�load_default_certs�load_verify_locationsr] �SSLError�load_cert_chain� set_ciphersr~ � NameErrorr �NotImplementedError�CertificateErrorr ) r, r� r� r� r� r� r� r� Z tls_version� ssl_protocol�contextr r" r" r# �build_ssl_context* s^
&& �z MySQLSocket.build_ssl_context)NN)r1 r2 r3 r4 rB �propertyr r( r r r r� �boolr� r6 r� r� r� r� r� r5 r r- r9 r0 r
r� r" r" r" r# r� � sJ
�� �
�r� c s< e Zd ZdZd eeed�� fdd�
Zedd �d
d �Z � Z
S )
r z�MySQL socket class using TCP/IP.
Args:
host: MySQL host name.
port: MySQL port.
force_ipv6: Force IPv6 usage.
� 127.0.0.1�� F)�host�port�
force_ipv6c s0 t � �� || _|| _|| _|� d|� �| _d S )Nr� )rn rB �_host�_portZ _force_ipv6r� )r, r� r� r� rq r" r# rB z s
zMySQLTcpSocket.__init__Nr� c � s0 t f | j| jd�|��I dH \| _| _d| _dS )zOpen TCP/IP connection.)r� r� NT)r r� r� r� r� r� r� r" r" r# r � s ��z MySQLTcpSocket.open_connection)r� r� F) r1 r2 r3 r4 r r6 r� rB r r r� r" r" rq r# r q s �� c s8 e Zd ZdZd
ed�� fdd�
Zedd�dd �Z� ZS ) r zbMySQL socket class using UNIX sockets.
Args:
unix_socket: UNIX socket file path.
�/tmp/mysql.sock)� unix_socketc s t � �� || _d S )N)rn rB r� )r, r� rq r" r# rB � s
zMySQLUnixSocket.__init__Nr� c � s. t jf d| ji|��I dH \| _| _d| _dS )z Open UNIX socket connection.�pathNT)r7 �open_unix_connectionr� r� r� r� r� r" r" r# r � s ��
�z MySQLUnixSocket.open_connection)r� ) r1 r2 r3 r4 r rB r r r� r" r" rq r# r � s )-r4 �__all__r7 rL r{ r� �PROTOCOL_TLSv1�PROTOCOL_TLSv1_1�PROTOCOL_TLSv1_2� PROTOCOL_TLSr� � ImportError�abcr r � collectionsr
�typingr r r
r r �errorsr r r r �networkr r r r �utilsr r r] r r$ r% r: rm r� r r r" r" r" r# �<module> s8 �
Gj J )