U j51@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@s(eZdZdZedeeeeeeee ee ee fe ee fd ddZedededdddddf eeeeeeeeeeeeee eefe e ee ee fe ee fdd d Zdeeeeee ee ed feefdddZdee ed fee ee eed feefdddZdS)rzSImplements MySQL client/server protocol. Create and parses MySQL packets. NF) auth_datausernamepassword client_flags auth_pluginauth_plugin_class ssl_enabled plugin_configreturnc Cs|sdt|||dfS|dkr"i}z&t|||||d}|j|f|} Wn6ttfk r~} ztd| | W5d} ~ XYnX| dkrtd|j|tj@rtt | | 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 ZSECURE_CONNECTIONrlen) rrrrr r!r"r# auth_strategyr&errr,@/tmp/pip-unpacked-wheel-d6bt0v6z/mysql/connector/aio/protocol.pyauth_plugin_first_response:s4#" z(MySQLProtocol.auth_plugin_first_responser) handshakerrdatabasecharsetrmax_allowed_packetr r! conn_attrsis_change_user_requestr"r#r$c  Cs|} g}| rtdtd|dkr6tdd|ddkrNtddz|pZ|d}Wn8ttfk r}ztd|d dW5d}~XYnXtd || r|t d t | d t j | n,d }|t d|t | d |||| t j|d|||||| | d\}}|||t ||| rN|t d||tj@rl||d|tj@r| dk r|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 %szrHbytesr ) rIrArPrC_rJrKZrowdatarMrNZdatasr,r,r-read_text_results6      zMySQLProtocol.read_text_result)NFN)rr@)r)__name__ __module__ __qualname____doc__ staticmethodrSstrintrboolrrrrr.rrrr?rrrrrrOrUr,r,r,r-r4s A  % )%rY__all__r<typingrrrrr constantsr r errorsr r r rprotocolrrrZ_MySQLProtocoltypesrrrrutilsrrnetworkrZpluginsrrZplugins.caching_sha2_passwordrr,r,r,r-s