File: //var/opt/nydus/ops/mysql/connector/aio/__pycache__/protocol.cpython-39.pyc
a
�,�h51 � @ s� d Z dgZddlZddlmZmZmZmZmZ ddl m
Z
m Z ddl m
Z
mZmZ ddlmZ dd lmZmZmZ dd
lmZmZmZmZ dd lm Z m Z d d
l m Z d dl m!Z!m"Z" d dl#m$Z$ G dd� de�ZdS )z,Implements the MySQL Client/Server protocol.�
MySQLProtocol� N)�Any�Dict�List�Optional�Tuple� )�
ClientFlag� ServerCmd)�InterfaceError�ProgrammingError�
get_exception)�logger)�DEFAULT_CHARSET_ID�DEFAULT_MAX_ALLOWED_PACKETr )�BinaryProtocolType�DescriptionType�
EofPacketType�
HandShakeType)� int1store�read_lc_string_list� )� MySQLSocket)�MySQLAuthPlugin�get_auth_plugin)�"MySQLCachingSHA2PasswordAuthPluginc @ s( e Zd ZdZedeeeeeee e ee
ee f e ee
f d� dd��Zedededdddddf
eeeee eeeee ee ee
eef e e ee
ee f e ee
f d�d d
��Zdeee eee ee ed
f ee f d�dd�Zdee ed
f ee ee ee d
f ee f d�dd�ZdS )r zSImplements MySQL client/server protocol.
Create and parses MySQL packets.
NF) � auth_data�username�password� client_flags� auth_plugin�auth_plugin_class� ssl_enabled�
plugin_config�returnc
C s� |sdt |||d�fS |du r"i }z*t||�|||d�}|j| fi |��} W n8 ttfy� }
z td|
� ��|
�W Y d}
~
n
d}
~
0 0 | du r�td|j� ���|tj@ r�tt | ��| n| d } | |fS )a Prepare the first authentication response.
Args:
auth_data: Authorization data from initial handshake.
username: Account's username.
password: Account's password.
client_flags: Integer representing client capabilities flags.
auth_plugin: Authorization plugin name.
auth_plugin_class: Authorization plugin class (has higher precedence
than the authorization plugin name).
ssl_enabled: Whether SSL is enabled or not.
plugin_config: Custom configuration to be passed to the auth plugin
when invoked. The parameters defined here will override
the ones defined in the auth plugin itself.
Returns:
auth_response: Authorization plugin response.
auth_strategy: Authorization plugin instance created based
on the provided `auth_plugin` and `auth_plugin_class`
parameters.
Raises:
InterfaceError: If authentication fails or when got a NULL auth response.
� )r" NzFailed authentication: z8Got NULL auth response while authenticating with plugin )
r r �
auth_response� TypeErrorr �namer ZSECURE_CONNECTIONr �len) r r r r r r! r" r# �
auth_strategyr& �err� r, �O/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/connector/aio/protocol.py�auth_plugin_first_response: s0 #��&����z(MySQLProtocol.auth_plugin_first_responser )� handshaker r �database�charsetr �max_allowed_packetr r! �
conn_attrs�is_change_user_requestr" r# r$ c
C s� |� � }
g }|
rt�d� t�d� | du r6td�d�| �d�du rNtd�d�z|pZ| d }W n: ttfy� } z td|� d ��d�W Y d}~n
d}~0 0 t�d
|� |
r�|�t� d t
|
�� d �t j |
�� n,d
}|�t� d|� t
|
�� d �||||
�� t
j| d |||||| | d�\}}|�|� |�t
�||�� |
�rP|�t� d|�� |tj@ �rn|�|� � d � |tj@ �r�| du�r�|�t
�| �� d�|�|fS )a� Make a MySQL Authentication packet.
Args:
handshake: Initial handshake.
username: Account's username.
password: Account's password.
database: Initial database name for the connection
charset: Client charset (see [2]), only the lower 8-bits.
client_flags: Integer representing client capabilities flags.
max_allowed_packet: Maximum packet size.
auth_plugin: Authorization plugin name.
auth_plugin_class: Authorization plugin class (has higher precedence
than the authorization plugin name).
conn_attrs: Connection attributes.
is_change_user_request: Whether is a `change user request` operation or not.
ssl_enabled: Whether SSL is enabled or not.
plugin_config: Custom configuration to be passed to the auth plugin
when invoked. The parameters defined here will override
the one defined in the auth plugin itself.
Returns:
handshake_response: Handshake response as per [1].
auth_strategy: Authorization plugin instance created based
on the provided `auth_plugin` and `auth_plugin_class`.
Raises:
ProgrammingError: Handshake misses authentication info.
References:
[1]: https://dev.mysql.com/doc/dev/mysql-server/latest/ page_protocol_connection_phase_packets_protocol_handshake_response.html
[2]: https://dev.mysql.com/doc/dev/mysql-server/latest/ page_protocol_basic_character_set.html#a_protocol_character_set
zGot a `change user` requestz Starting authorization phaseNzGot a NULL handshaker z$Handshake misses authentication infor z-Handshake misses authentication plugin info (�)z#The provided initial strategy is %sz<BZsxZxxxxxxxxxxxxxxxxxxxxxxz<IIH)r r r r r r! r" r# z<Hr% � )�encoder �debugr �getr' �KeyError�append�struct�packr) r
Z CHANGE_USERr r. Zconnect_with_dbr Z PLUGIN_AUTHZ CONNECT_ARGSZmake_conn_attrs�join)r/ r r r0 r1 r r2 r r! r3 r4 r" r# Z
b_usernameZresponse_payloadr+ �fillerr& r* r, r, r- � make_auth| sn 3
�� ���� �
zMySQLProtocol.make_authr �utf-8.)�sock�columns�countr1 r$ c
� s� g }d}d}d}|s�||kr q�|� � I dH } | d dkrH| �| �}d}n&| d dkrnd}| �|| dd� |�}|du r�|dur�|�|� n|du r�|du r�t| ��|d7 }q||fS )zxRead MySQL binary protocol result.
Reads all or given number of binary resultset rows from the socket.
Nr � � � r )�read� parse_eofZ_parse_binary_valuesr; r
)
�selfrB rC rD r1 �rows�eof�values�i�packetr, r, r- �read_binary_result� s&
z MySQLProtocol.read_binary_result)rB �versionrD r$ c � s4 |}g }d}d}d}|s ||kr$�q,|� � I dH } | �d�r�| dd� g}
|� � I dH } | �d�r�|
�| dd� � |� � I dH } qX|
�| dd� � td�|
��}n@| d dkr�| d dk r�| �| �}d}nd}tt| dd� ��}|du �r|du�r|�|� n |du �r"|du �r"t| ��|d7 }q||fS ) z�Read MySQL text result.
Reads all or given number of rows from the socket.
Returns a tuple with 2 elements: a list with all rows and
the EOF packet.
Nr s ���rE r6 rF � r )rH �
startswithr; r r> rI �bytesr
) rJ rB rQ rD �_rK rL ZrowdatarN rO Zdatasr, r, r- �read_text_result s6
z MySQLProtocol.read_text_result)NFN)r rA )r )�__name__�
__module__� __qualname__�__doc__� staticmethodrT �str�intr �boolr r r r r. r r r r@ r r r r r rP rV r, r, r, r- r 4 s� �
�A�
� ���� % ����)%rZ �__all__r< �typingr r r r r � constantsr r
�errorsr r r
r �protocolr r r Z_MySQLProtocol�typesr r r r �utilsr r �networkr Zpluginsr r Z plugins.caching_sha2_passwordr r, r, r, r- �<module> s