3 S^k@stdZddlZddlZddlmZmZmZddZGdddeZ Gd d d e Z Gd d d e Z Gd dde Z dS)z.Implementation of MySQL Authentication Plugin.N)PY3 UNICODE_TYPEShexlifycCsDtrddt||D}nddt||D}tjdj|f|S)zEncrypt/Decrypt function used for password encryption in authentication, using a simple XOR. Args: hash1 (str): The first hash. hash2 (str): The second hash. Returns: str: A string with the xor applied. cSsg|]\}}||AqSr).0h1h2rrG/tmp/pip-build-8app2_gc/mysql-connector-python/mysqlx/authentication.py 1szxor_string..cSs g|]\}}t|t|AqSr)ord)rrr rrr r 3sz{0}B)rzipstructpackformat)hash1hash2Z hash_sizexoredrrr xor_string%s rc@s*eZdZdZd ddZddZddZdS) BaseAuthPluginz7Base class for implementing the authentication plugins.NcCs||_||_dS)N) _username _password)selfusernamepasswordrrr __init__9szBaseAuthPlugin.__init__cCstdS)zUReturns the plugin name. Returns: str: The plugin name. N)NotImplementedError)rrrr name=szBaseAuthPlugin.namecCstdS)zeReturns the authentication name. Returns: str: The authentication name. N)r)rrrr auth_nameEszBaseAuthPlugin.auth_name)NN)__name__ __module__ __qualname____doc__rrrrrrr r7s rc@s(eZdZdZddZddZddZdS) MySQL41AuthPluginzCClass implementing the MySQL Native Password authentication plugin.cCsdS)zUReturns the plugin name. Returns: str: The plugin name. zMySQL 4.1 Authentication Pluginr)rrrr rPszMySQL41AuthPlugin.namecCsdS)zeReturns the authentication name. Returns: str: The authentication name. ZMYSQL41r)rrrr rXszMySQL41AuthPlugin.auth_namecCs||jrnt|jtr|jjdn|j}tj|j}tj|j}t|tj||jd}djd|j t |Sdjd|j S)zHashing for MySQL 4.1 authentication. Args: data (str): The authentication data. Returns: str: The authentication response. zutf-8z {0}{1}*{2}z{0}{1}) r isinstancerencodehashlibsha1digestrrrr)rdatarrrrrrr auth_data`s zMySQL41AuthPlugin.auth_dataN)rr r!r"rrr,rrrr r#Nsr#c@s(eZdZdZddZddZddZdS) PlainAuthPluginz9Class implementing the MySQL Plain authentication plugin.cCsdS)zUReturns the plugin name. Returns: str: The plugin name. zPlain Authentication Pluginr)rrrr ruszPlainAuthPlugin.namecCsdS)zeReturns the authentication name. Returns: str: The authentication name. ZPLAINr)rrrr r}szPlainAuthPlugin.auth_namecCs2t|jtrt r|jjdn|j}dj|j|S)zeReturns the authentication data. Returns: str: The authentication data. zutf-8z{0}{1})r&rrrr'rr)rrrrr r,szPlainAuthPlugin.auth_dataN)rr r!r"rrr,rrrr r-ssr-c@s(eZdZdZddZddZddZdS) Sha256MemoryAuthPluginz;Class implementing the SHA256_MEMORY authentication plugin.cCsdS)zUReturns the plugin name. Returns: str: The plugin name. z#SHA256_MEMORY Authentication Pluginr)rrrr rszSha256MemoryAuthPlugin.namecCsdS)zeReturns the authentication name. Returns: str: The authentication name. Z SHA256_MEMORYr)rrrr rsz Sha256MemoryAuthPlugin.auth_namecCsft|jtr|jjdn|j}tj|j}tjtj|j|j}t||d}dj|j t |S)a Hashing for SHA256_MEMORY authentication. The scramble is of the form: SHA256(SHA256(SHA256(PASSWORD)),NONCE) XOR SHA256(PASSWORD) Args: data (str): The authentication data. Returns: str: The authentication response. zutf-8 z{0}{1}) r&rrr'r(sha256r*rrrr)rr+rrrrrrr r,s  z Sha256MemoryAuthPlugin.auth_dataN)rr r!r"rrr,rrrr r.sr.) r"r(rcompatrrrrobjectrr#r-r.rrrr s%