o #f0@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) int1storeread_lc_string_list) MySQLSocket)MySQLAuthPluginget_auth_plugin)"MySQLCachingSHA2PasswordAuthPluginc @sleZdZdZe   d#dededededed eed e d ee ee fd e ee ffd dZedededdddddf dedededeededededeed eedee eefde d e d ee ee fd e ee ffddZ  d$dedeededed e ee edfeeff ddZ d%ded e edfded e ee eedfeeffd!d"ZdS)&rzSImplements MySQL client/server protocol. Create and parses MySQL packets. NF auth_datausernamepassword client_flags auth_pluginauth_plugin_class ssl_enabled plugin_configreturnc Cs|s dt|||dfS|duri}zt|||||d}|j|fi|} Wnttfy<} ztd| | d} ~ ww| durItd|j|tj@rVtt | | n| d} | |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 namer SECURE_CONNECTIONrlen) rrrrr r!r"r# auth_strategyr&errr-R/home/ubuntu/dev/venv/lib/python3.10/site-packages/mysql/connector/aio/protocol.pyauth_plugin_first_response:s4#z(MySQLProtocol.auth_plugin_first_responser handshakedatabasecharsetmax_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 infor z-Handshake misses authentication plugin info ()z#The provided initial strategy is %szr ) selfrLrMrNr2rowseofvaluesipacketr-r-r.read_binary_results,       z MySQLProtocol.read_binary_resultversionc s6|}g}d}d}d} |s||kr ||fS|IdH} | dr^| ddg} |IdH} | drK| | dd|IdH} | ds6| | ddttd| }n| ddkrr| ddkrr|| }d}n d}t| 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. NrTsrOr9rPr)rR startswithr>r bytearrayrFrSr ) rUrLr\rN_rVrWrowdatarYrZdatasr-r-r.read_text_results>       zMySQLProtocol.read_text_result)NFN)rrK)r)__name__ __module__ __qualname____doc__ staticmethodbytesstrintrboolrrrrr/rrrrJrrrrrr[rcr-r-r-r.r4s A        " )%rg__all__r?typingrrrrr constantsr r errorsr r r rprotocolrrr_MySQLProtocoltypesrrrrutilsrrnetworkrpluginsrrplugins.caching_sha2_passwordrr-r-r-r.s