U 4Je@sdZddlZddlZddlmZddlmZeeeedddZ Gd d d Z Gd d d e Z Gd dde Z Gddde Z dS)z.Implementation of MySQL Authentication Plugin.N)Optional)hexlify)hash1hash2 hash_sizereturncCs*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 ).0h1h2r r 9/tmp/pip-unpacked-wheel-7_167w8m/mysqlx/authentication.py 2szxor_string..B)zipstructpack)rrrxoredr r r xor_string's rc@sFeZdZdZd eeeedddZedddZedd d ZdS) BaseAuthPluginz7Base class for implementing the authentication plugins.N)usernamepasswordcCs||_||_dS)N _username _password)selfrrr r r __init__9szBaseAuthPlugin.__init__rcCstdS)UReturns the plugin name. Returns: str: The plugin name. NNotImplementedErrorrr r r name=szBaseAuthPlugin.namecCstdS)eReturns the authentication name. Returns: str: The authentication name. Nrr!r r r auth_nameEszBaseAuthPlugin.auth_name)NN) __name__ __module__ __qualname____doc__rstrrr"r$r r r r r6src@s<eZdZdZedddZedddZeeddd Zd S) MySQL41AuthPluginzCClass implementing the MySQL Native Password authentication plugin.rcCsdS)rzMySQL 4.1 Authentication Pluginr r!r r r r"QszMySQL41AuthPlugin.namecCsdS)r#ZMYSQL41r r!r r r r$YszMySQL41AuthPlugin.auth_namedatarcCs|jrrt|jtr|jdn|j}t|}t|}t|t||d}d|jdt |dSd|jdS)zHashing for MySQL 4.1 authentication. Args: data (bytes): The authentication data. Returns: str: The authentication response. utf-8z*) r isinstancer)encodehashlibsha1digestrrrrr,rrrrr r r auth_dataas  zMySQL41AuthPlugin.auth_dataN r%r&r'r(r)r"r$bytesr6r r r r r*Nsr*c@s:eZdZdZedddZedddZedddZd S) PlainAuthPluginz9Class implementing the MySQL Plain authentication plugin.rcCsdS)rzPlain Authentication Pluginr r!r r r r"zszPlainAuthPlugin.namecCsdS)r#ZPLAINr r!r r r r$szPlainAuthPlugin.auth_namecCsd|jd|jS)zeReturns the authentication data. Returns: str: The authentication data. r/rr!r r r r6szPlainAuthPlugin.auth_dataN)r%r&r'r(r)r"r$r6r r r r r9wsr9c@s<eZdZdZedddZedddZeeddd Zd S) Sha256MemoryAuthPluginz;Class implementing the SHA256_MEMORY authentication plugin.rcCsdS)rz#SHA256_MEMORY Authentication Pluginr r!r r r r"szSha256MemoryAuthPlugin.namecCsdS)r#Z SHA256_MEMORYr r!r r r r$sz Sha256MemoryAuthPlugin.auth_namer+cCsjt|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 (bytes): The authentication data. Returns: str: The authentication response. r- r/) r0rr)r1r2sha256r4rrrr5r r r r6s  z Sha256MemoryAuthPlugin.auth_dataNr7r r r r r:sr:)r(r2rtypingrZhelpersrr8intrrr*r9r:r r r r s  )