U ec?@sjdZddlZddlZddlmZddZGdddZGd d d eZGd d d eZGd ddeZ dS)z.Implementation of MySQL Authentication Plugin.N)hexlifycCs*ddt||D}tj|df|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).0h1h2rr9/tmp/pip-unpacked-wheel-ix8dg7li/mysqlx/authentication.py 0szxor_string..B)zipstructpack)hash1hash2Z hash_sizexoredrrr xor_string%s rc@s*eZdZdZd ddZddZddZdS) BaseAuthPluginz7Base class for implementing the authentication plugins.NcCs||_||_dS)N _username _password)selfusernamepasswordrrr__init__7szBaseAuthPlugin.__init__cCstdS)UReturns the plugin name. Returns: str: The plugin name. NNotImplementedErrorrrrrname;szBaseAuthPlugin.namecCstdS)eReturns the authentication name. Returns: str: The authentication name. Nrrrrr auth_nameCszBaseAuthPlugin.auth_name)NN)__name__ __module__ __qualname____doc__rrr rrrrr4s rc@s(eZdZdZddZddZddZdS) MySQL41AuthPluginzCClass implementing the MySQL Native Password authentication plugin.cCsdS)rzMySQL 4.1 Authentication PluginrrrrrrOszMySQL41AuthPlugin.namecCsdS)rZMYSQL41rrrrrr WszMySQL41AuthPlugin.auth_namecCs|jrrt|jtr|jdn|j}t|}t|}t|t||d}d|jdt |dSd|jdS)zHashing for MySQL 4.1 authentication. Args: data (str): The authentication data. Returns: str: The authentication response. utf-8z*) r isinstancestrencodehashlibsha1digestrrrrdatarrrrrrr auth_data_s  zMySQL41AuthPlugin.auth_dataNr!r"r#r$rr r1rrrrr%Lsr%c@s(eZdZdZddZddZddZdS) PlainAuthPluginz9Class implementing the MySQL Plain authentication plugin.cCsdS)rzPlain Authentication PluginrrrrrrxszPlainAuthPlugin.namecCsdS)rZPLAINrrrrrr szPlainAuthPlugin.auth_namecCsd|jd|jS)zeReturns the authentication data. Returns: str: The authentication data. r(rrrrrr1szPlainAuthPlugin.auth_dataNr2rrrrr3usr3c@s(eZdZdZddZddZddZdS) Sha256MemoryAuthPluginz;Class implementing the SHA256_MEMORY authentication plugin.cCsdS)rz#SHA256_MEMORY Authentication PluginrrrrrrszSha256MemoryAuthPlugin.namecCsdS)rZ SHA256_MEMORYrrrrrr sz Sha256MemoryAuthPlugin.auth_namecCsjt|jtr|jdn|j}t|}tt||}t||d}d|jdt |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. r& r() r)rr*r+r,sha256r.rrrr/rrrr1s  z Sha256MemoryAuthPlugin.auth_dataNr2rrrrr4sr4) r$r,r Zhelpersrrrr%r3r4rrrrs )