o ht$@sdZddlmZmZddlZddlmZddlmZm Z m Z Gddde Z Gd d d e Z Gd d d e ZGd dde ZGddde ZddZdS)z5Implementing support for MySQL Authentication Plugins)sha1sha256N)errors)PY2isstr UNICODE_TYPESc@s6eZdZdZdZdZ  d ddZddZd d ZdS) BaseAuthPluginaBase class for authentication plugins Classes inheriting from BaseAuthPlugin should implement the method prepare_password(). When instantiating, auth_data argument is required. The username, password and database are optional. The ssl_enabled argument can be used to tell the plugin whether SSL is active or not. The method auth_response() method is used to retrieve the password which was prepared by prepare_password(). FNcCs"||_||_||_||_||_dS)InitializationN) _auth_data _username _password _database _ssl_enabled)self auth_datausernamepassworddatabase ssl_enabledrP/var/www/html/env/lib/python3.10/site-packages/mysql/connector/authentication.py__init__7s  zBaseAuthPlugin.__init__cCst)zPrepares and returns password to be send to MySQL This method needs to be implemented by classes inheriting from this class. It is used by the auth_response() method. Raises NotImplementedError. )NotImplementedErrorrrrrprepare_password@szBaseAuthPlugin.prepare_passwordcCs(|jr|jstdj|jd|S)zReturns the prepared password to send to MySQL Raises InterfaceError on errors. For example, when SSL is required by not enabled. Returns str {name} requires SSLname) requires_sslrrInterfaceErrorformat plugin_namerrrrr auth_responseJs zBaseAuthPlugin.auth_response)NNNF) __name__ __module__ __qualname____doc__r r#rrr$rrrrr &s   r c@ eZdZdZdZdZddZdS)MySQLNativePasswordAuthPluginzBClass implementing the MySQL Native Password authentication pluginFmysql_native_passwordc Cs|jstd|js dS|j}t|jr|jd}n|j}tr9t|}zt|j}Wnty8tdw|}|j}d}z7t | }t | }t || }trbddt ||D}n ddt ||D}t j d g|R}W|Sty}ztd |d}~ww) z;Prepares and returns password as native MySQL 4.1+ password"Missing authentication data (seed)utf-8Authentication data incorrectNcS g|] \}}t|t|AqSrord.0h1h3rrr | zBMySQLNativePasswordAuthPlugin.prepare_password..cSg|]\}}||AqSrrr3rrrr7~20BzFailed scrambling password; {0})r rr!rrencoderbuffer TypeErrorrdigestzipstructpack Exceptionr") rrrhash4hash1hash2hash3xoredexcrrrr^sB      z.MySQLNativePasswordAuthPlugin.prepare_passwordNr%r&r'r(r r#rrrrrr*X  r*c@r))MySQLClearPasswordAuthPluginzAClass implementing the MySQL Clear Password authentication pluginTmysql_clear_passwordcCL|jsdS|j}trt|tr|d}|dSt|tr"|d}|dS!Returns password as as clear textutf8rr isinstanceunicoder<strrrrrrr   z-MySQLClearPasswordAuthPlugin.prepare_passwordNrJrrrrrLrKrLc@r))MySQLSHA256PasswordAuthPluginzClass implementing the MySQL SHA256 authentication plugin Note that encrypting using RSA is not supported since the Python Standard Library does not provide this OpenSSL functionality. Tsha256_passwordcCrNrOrSrWrrrrrXz.MySQLSHA256PasswordAuthPlugin.prepare_passwordNrJrrrrrYs  rYc@s8eZdZdZdZdZdZdZddZdd Z d d Z d S) "MySQLCachingSHA2PasswordAuthPluginzClass implementing the MySQL caching_sha2_password authentication plugin Note that encrypting using RSA is not supported since the Python Standard Library does not provide this OpenSSL functionality. Fcaching_sha2_passwordcCs|jstd|js dSt|jtr|jdn|j}tr6t|}zt|j}Wnt y5tdw|}|j}t | }t }| t | | || }trcddt ||D}n ddt ||D}tjdg|R}|S) z Returns a scramble of the password using a Nonce sent by the server. The scramble is of the form: XOR(SHA2(password), SHA2(SHA2(SHA2(password)), Nonce)) r,r-r.r/cSr0rr1r4r5h2rrrr7r8z@MySQLCachingSHA2PasswordAuthPlugin._scramble..cSr9rrr_rrrr7r:32B)r rr!rrTrr<rr=r>rr?updater@rArB)rrrrErFrHrGrrr _scrambles6      z,MySQLCachingSHA2PasswordAuthPlugin._scramblecCs2t|jdkr |S|jd|jkr|SdS)Nrr)lenr rcperform_full_authentication_full_authenticationrrrrrs z3MySQLCachingSHA2PasswordAuthPlugin.prepare_passwordcCsf|js tdj|jd|jsdS|j}tr%t|tr!| d}|dSt|t r/| d}|dS)rPrrrQrR) rrr!r"r#rrrTrUr<rVrWrrrrfs   z7MySQLCachingSHA2PasswordAuthPlugin._full_authenticationN) r%r&r'r(r r#refast_auth_successrcrrfrrrrr[s' r[cCs0tD] }|j|kr|Sqtd|)a.Return authentication class based on plugin name This function returns the class for the authentication plugin plugin_name. The returned class is a subclass of BaseAuthPlugin. Raises errors.NotSupportedError when plugin_name is not supported. Returns subclass of BaseAuthPlugin. z,Authentication plugin '{0}' is not supported)r __subclasses__r#rNotSupportedErrorr")r# authclassrrrget_auth_plugins rk)r(hashlibrrrAr rcatch23rrrobjectr r*rLrYr[rkrrrrs 2/ L