o g(1@sdZdgZddlZddlmZmZmZmZmZddl m Z m Z ddl m Z mZmZddlmZdd lmZmZmZdd lmZmZmZmZdd lmZmZd d lmZd dl m!Z!m"Z"d dl#m$Z$GdddeZdS)z,Implements the MySQL Client/Server protocol. MySQLProtocolN)AnyDictListOptionalTuple) ClientFlag ServerCmd)InterfaceErrorProgrammingError get_exception)logger)DEFAULT_CHARSET_IDDEFAULT_MAX_ALLOWED_PACKETr)BinaryProtocolTypeDescriptionType EofPacketType HandShakeType)lc_intread_lc_string_list) MySQLSocket)MySQLAuthPluginget_auth_plugin)"MySQLCachingSHA2PasswordAuthPluginc @s|eZdZdZe   d$dededededeed ed ee ee fd e ee ffd d Z edededdddddf dedededeededededeedeedee eefded ed ee ee fd e ee ffddZ   d%dedeedededeed e ee edfeeff dd Z  d&ded!e edfdedeed e ee eedfeeff d"d#ZdS)'rzSImplements MySQL client/server protocol. Create and parses MySQL packets. NF auth_datausernamepassword auth_pluginauth_plugin_class ssl_enabled plugin_configreturnc Cs|s|dkrdt|||dfS|duri}zt|||||d}|j|fi|}Wnttfy@} ztd| | d} ~ ww|durMtd|jtt||}||fS)aPrepare 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 )rr auth_response TypeErrorr namerlen) rrrrr r!r" auth_strategyr&errr,V/var/www/html/api-tag/env/lib/python3.10/site-packages/mysql/connector/aio/protocol.pyauth_plugin_first_response:s. "z(MySQLProtocol.auth_plugin_first_responser handshakedatabasecharset client_flagsmax_allowed_packet conn_attrsis_change_user_requestc  Cs|} g}| r tdtd|durtdd|ddur'tddz|p-|d}WnttfyF}z td|d dd}~wwtd || ra|t d t | d t j | nd }|t d|t | d |||| t j|d||||| | d\}}|||t ||| r|t d||tj@r||d|tj@r| dur|t | d||fS)aMake 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` requestzStarting authorization phaseNzGot a NULL handshakerz$Handshake misses authentication inforz-Handshake misses authentication plugin info ()z#The provided initial strategy is %szd}||| dd|}|durL|durL||n |durX|durXt| | d7} q )zxRead MySQL binary protocol result. Reads all or given number of binary resultset rows from the socket. NrTr)read parse_eof_parse_binary_valuesr>r ) selfrLrMrNr1rOrowseofvaluesipacketr,r,r-read_binary_results,      z MySQLProtocol.read_binary_resultversionc s<|}g}d}d}d} |s| |kr ||fS||IdH} | dr_| ddg} ||IdH} | drN| | dd||IdH} | ds8| | ddtd| }n | ddkrs| ddkrs|| }d}n d}tt| dd}|dur|dur||n |dur|durt| | d 7} q ) zRead 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. NrTsrPr9rQr)rS startswithr>rrFrTbytesr ) rVrLr]rNrO_rWrXrowdatarZr[datasr,r,r-read_text_results>      zMySQLProtocol.read_text_result)NFN)rrKN)rN)__name__ __module__ __qualname____doc__ staticmethodr`strrboolrrrrr.rrrintrJrrrrrr\rdr,r,r,r-r4s  <       ~ ) )%rh__all__r?typingrrrrr constantsr r errorsr r r rprotocolrrr_MySQLProtocoltypesrrrrutilsrrnetworkrpluginsrrplugins.caching_sha2_passwordrr,r,r,r-s