U f7@sldZddlZddlZddlmZddZGdddeZGd d d eZGd d d eZ Gd ddeZ dS)z.Implementation of MySQL Authentication Plugin.N)hexlifycCs*ddt||D}tjd|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).0h1h2rr9/tmp/pip-unpacked-wheel-mzf9r1ws/mysqlx/authentication.py 0szxor_string..z{0}B)zipstructpackformat)hash1hash2Z hash_sizexoredrrr xor_string%s rc@s*eZdZdZd ddZddZddZdS) BaseAuthPluginz7Base class for implementing the authentication plugins.NcCs||_||_dS)N) _username _password)selfusernamepasswordrrr__init__6szBaseAuthPlugin.__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_nameBszBaseAuthPlugin.auth_name)NN)__name__ __module__ __qualname____doc__rrrrrrrr4s rc@s(eZdZdZddZddZddZdS) MySQL41AuthPluginzCClass implementing the MySQL Native Password authentication plugin.cCsdS)rzMySQL 4.1 Authentication PluginrrrrrrMszMySQL41AuthPlugin.namecCsdS)rZMYSQL41rrrrrrUszMySQL41AuthPlugin.auth_namecCs||jrnt|jtr|jdn|j}t|}t|}t|t||d}dd|j t |Sdd|j S)zHashing for MySQL 4.1 authentication. Args: data (str): The authentication data. Returns: str: The authentication response. utf-8z {0}{1}*{2}z{0}{1}) r isinstancestrencodehashlibsha1digestrr rrrdatarrrrrrr auth_data]s  zMySQL41AuthPlugin.auth_dataNr r!r"r#rrr0rrrrr$Ksr$c@s(eZdZdZddZddZddZdS) PlainAuthPluginz9Class implementing the MySQL Plain authentication plugin.cCsdS)rzPlain Authentication PluginrrrrrrrszPlainAuthPlugin.namecCsdS)rZPLAINrrrrrrzszPlainAuthPlugin.auth_namecCsd|j|jS)zeReturns the authentication data. Returns: str: The authentication data. {0}{1})r rrrrrrr0szPlainAuthPlugin.auth_dataNr1rrrrr2psr2c@s(eZdZdZddZddZddZdS) Sha256MemoryAuthPluginz;Class implementing the SHA256_MEMORY authentication plugin.cCsdS)rz#SHA256_MEMORY Authentication PluginrrrrrrszSha256MemoryAuthPlugin.namecCsdS)rZ SHA256_MEMORYrrrrrrsz Sha256MemoryAuthPlugin.auth_namecCsft|jtr|jdn|j}t|}tt||}t||d}d|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. r% r3) r(rr)r*r+sha256r-rr rrr.rrrr0s  z Sha256MemoryAuthPlugin.auth_dataNr1rrrrr4sr4) r#r+r Zhelpersrrobjectrr$r2r4rrrrs %