o h@sdZddlZddlZddlmZddlmZdededed efd d Z Gd d d Z Gddde Z Gddde Z Gddde Z dS)z.Implementation of MySQL Authentication Plugin.N)Optional)hexlifyhash1hash2 hash_sizereturncCs,ddt||D}tj|dg|RS)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 P/var/www/html/scripts/venv/lib/python3.10/site-packages/mysqlx/authentication.py 2szxor_string..B)zipstructpack)rrrxoredr r r xor_string's rc@sHeZdZdZd deedeefddZdefdd Zdefd d ZdS) BaseAuthPluginz7Base class for implementing the authentication plugins.NusernamepasswordcCs||_||_dS)N _username _password)selfrrr r r __init__9s zBaseAuthPlugin.__init__rcCt)UReturns the plugin name. Returns: str: The plugin name. NotImplementedErrorrr r r name=zBaseAuthPlugin.namecCr)eReturns the authentication name. Returns: str: The authentication name. rr!r r r auth_nameEr#zBaseAuthPlugin.auth_name)NN) __name__ __module__ __qualname____doc__rstrrr"r%r r r r r6s rc@>eZdZdZdefddZdefddZdedefdd Zd S) MySQL41AuthPluginzCClass implementing the MySQL Native Password authentication plugin.rcCdS)rzMySQL 4.1 Authentication Pluginr r!r r r r"Qr#zMySQL41AuthPlugin.namecCr-)r$MYSQL41r r!r r r r%Yr#zMySQL41AuthPlugin.auth_namedatacCs|jr9t|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%bytesr9r r r r r,N r,c@s:eZdZdZdefddZdefddZdefddZd S) PlainAuthPluginz9Class implementing the MySQL Plain authentication plugin.rcCr-)rzPlain Authentication Pluginr r!r r r r"zr#zPlainAuthPlugin.namecCr-)r$PLAINr r!r r r r%r#zPlainAuthPlugin.auth_namecCsd|jd|jS)zeReturns the authentication data. Returns: str: The authentication data. r2rr!r r r r9szPlainAuthPlugin.auth_dataN)r&r'r(r)r*r"r%r9r r r r r=ws r=c@r+) Sha256MemoryAuthPluginz;Class implementing the SHA256_MEMORY authentication plugin.rcCr-)rz#SHA256_MEMORY Authentication Pluginr r!r r r r"r#zSha256MemoryAuthPlugin.namecCr-)r$ SHA256_MEMORYr r!r r r r%r#z 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. r0 r2) r3rr*r4r5sha256r7rrrr8r r r r9s  z Sha256MemoryAuthPlugin.auth_dataNr:r r r r r?r<r?)r)r5rtypingrhelpersrr;intrrr,r=r?r r r r s  )