fume-manage-python.git

U  
£ý°d3ã@sdZddlZddlZddlmZddlmZddlmZm    Z    m
Z
ddlmZddl m Z z8dd    lmZdd
lmZmZddlmZddlmZWn ek
r°e d ¡dYnXzddlmZmZWn ek
ræe d¡dYnXddlmZdZdZdZdZ dZ!GdddeZ"dS)zOCI Authentication Plugin.éN)Ú    b64encode)ÚPath)ÚAnyÚDictÚOptionalé)Úerrors)Úlogger)ÚUnsupportedAlgorithm)ÚhashesÚ serialization)Úpadding)ÚPRIVATE_KEY_TYPESz'Package 'cryptography' is not installed)ÚconfigÚ
exceptionszGPackage 'oci' (Oracle Cloud Infrastructure Python SDK) is not installedé)ÚBaseAuthPluginÚMySQLOCIAuthPlugini(z0Ephemeral security token is too large (10KB max)zGEphemeral security token file ('security_token_file') could not be readzKOCI configuration file does not contain a 'fingerprint' or 'key_file' entryc@s®eZdZUdZdZeed<dZeed<dZ    e
ed<dZeed    <ej Zeed
<eeeee
feddd ZeeedddZeee
fdddZdeeedddZdS)rz2Implement the MySQL OCI IAM authentication plugin.Zauthentication_oci_clientÚplugin_nameFÚrequires_sslNÚcontextÚDEFAULTÚoci_config_profileÚoci_config_file)Ú    signatureÚ
oci_configÚreturnc
Cst|}|d| ¡d}| d¡rz8t|d}| ¡jtkrJt t    ¡|j
dd|d<Wn2ttfk
r}zt t ¡|W5d}~XYnXtj|dd    S)
a=Prepare client's authentication response
 
        Prepares client's authentication response in JSON format
        Args:
            signature (bytes):  server's nonce to be signed by client.
            oci_config (dict): OCI configuration object.
 
        Returns:
            str: JSON string with the following format:
                 {"fingerprint": str, "signature": str, "token": base64.base64.base64}
 
        Raises:
            ProgrammingError: If the ephemeral security token file can't be open or the
                              token is too large.
        Úfingerprint)rrÚsecurity_token_filezutf-8)ÚencodingÚtokenN)ú,ú:)Ú
separators)rÚdecodeÚgetrÚstatÚst_sizeÚOCI_SECURITY_TOKEN_MAX_SIZErÚProgrammingErrorÚOCI_SECURITY_TOKEN_TOO_LARGEÚ    read_textÚOSErrorÚUnicodeErrorÚ%OCI_SECURITY_TOKEN_FILE_NOT_AVAILABLEÚjsonÚdumps)rrZsignature_64Ú auth_responserÚerr©r3úhd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\mysql/connector/plugins/authentication_oci_client.pyÚ_prepare_auth_responsePs"þ    
 
ÿþz)MySQLOCIAuthPlugin._prepare_auth_response)Úkey_pathrcCszz4ttj |¡d}tj| ¡dd}W5QRXWn@ttt    t
fk
rt}zt d|d|¡W5d}~XYnX|S)z+Get the private_key form the given locationÚrbN)Úpasswordz2An error occurred while reading the API_KEY from "z": ) ÚopenÚosÚpathÚ
expanduserrZload_pem_private_keyÚreadÚ    TypeErrorr,Ú
ValueErrorr
rr))r6Úkey_fileÚprivate_keyr2r3r3r4Ú_get_private_keyxsþÿz#MySQLOCIAuthPlugin._get_private_key)rcCsg}ddddd}i}zt |jp*tj|jp2d¡}| ¡D]X\}}z*||rn|||sn| d|d¡Wq>tk
r| d|¡Yq>Xq>WnDtj    tj
tjtjtj fk
rÞ}z| t|¡W5d    }~XYnX|rüt d
|jd|¡|S)z=Get a valid OCI config from the given configuration file pathcSst|dkS)Né )Úlen©Úxr3r3r4Ú<lambda>óz:MySQLOCIAuthPlugin._get_valid_oci_config.<locals>.<lambda>cSstj tj |¡¡S)N)r:r;Úexistsr<rEr3r3r4rGrH)rr@rzParameter "z" is invalidzDoes not contain parameter NzInvalid oci-config-file: z. Errors found: )rÚ    from_filerÚDEFAULT_LOCATIONrÚitemsÚappendÚKeyErrorrZConfigFileNotFoundZ InvalidConfigZInvalidKeyFilePathZInvalidPrivateKeyZProfileNotFoundÚstrrr))ÚselfZ
error_listZreq_keysrZreq_keyZ    req_valuer2r3r3r4Ú_get_valid_oci_configs:þ
þû ÿz(MySQLOCIAuthPlugin._get_valid_oci_config)Ú    auth_datarcCsdt d|jt|j¡| ¡}| |d¡}| |jt ¡t     
¡¡}| ||¡}t d|¡| ¡S)z-Prepare authentication string for the server.zserver nonce: %s, len %dr@zauthentication response: %s) r    ÚdebugZ
_auth_datarDrQrBÚsignr ZPKCS1v15rÚSHA256r5Úencode)rPrRrrArr1r3r3r4r1±sÿz MySQLOCIAuthPlugin.auth_response)N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__rrOÚ__annotations__rÚboolrrrrrKrÚstaticmethodÚbytesrr5rrBrQrr1r3r3r3r4rGs
'()#rZr/r:Úbase64rÚpathlibrÚtypingrrrÚrr    Zcryptography.exceptionsr
Zcryptography.hazmat.primitivesrrZ)cryptography.hazmat.primitives.asymmetricr Z/cryptography.hazmat.primitives.asymmetric.typesrÚImportErrorr)ZocirrrZAUTHENTICATION_PLUGIN_CLASSr(r*r.ZOCI_PROFILE_MISSING_PROPERTIESrr3r3r3r4Ú<module>s>ÿþÿÿ