o pkj @sdZddlZddlZddlZddlZddlmZddlmZddl m Z ddl Z e eZ ddededeed ejfd d Zd d ZddZ dddZdddZdS)z< Helper functions for mTLS in async for discovery of certs. N)Optional) exceptions)secure_cert_key_paths cert_bytes key_bytes passphrasereturnc Csz3t|||d!\}}}ttjj}|r|}|j|||d|WdWS1s,wYWdStjttt t t fyL}zt d|d}~ww)aCreates an SSLContext with the given client certificate and key. This function writes the certificate and key to temporary files so that ssl.create_default_context can load them, as the ssl module requires file paths for client certificates. These temporary files are deleted immediately after the SSL context is created. Args: cert_bytes (bytes): The client certificate content in PEM format. key_bytes (bytes): The client private key content in PEM format. passphrase (Optional[bytes]): The passphrase for the private key, if any. Returns: ssl.SSLContext: The configured SSL context with client certificate. Raises: google.auth.exceptions.TransportError: If there is an error loading the certificate. )r)certfilekeyfilepasswordNz3Failed to load client certificate and key for mTLS.)rsslcreate_default_contextPurpose SERVER_AUTHload_cert_chainSSLErrorOSErrorIOError ValueError RuntimeError TypeErrorrTransportError) rrr cert_pathkey_pathpassphrase_valcontextr excr\/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/google/auth/aio/transport/mtls.pymake_client_cert_ssl_context s.(rcsRz tj|g|RIdHWSty(t}|jd|g|RIdHYSw)zRun a blocking function in an executor to avoid blocking the event loop. This implements the non-blocking execution strategy for disk I/O operations. N)asyncio to_threadAttributeErrorget_running_looprun_in_executor)funcargslooprrr_run_in_executorGs r(cCs(tjjjjddstddd}|S)aGet a callback which returns the default client SSL credentials. Returns: Awaitable[Callable[[], Tuple[bytes, bytes]]]: A callback which returns the default client certificate bytes and private key bytes, both in PEM format. Raises: google.auth.exceptions.DefaultClientCertSourceError: If the default client SSL credentials don't exist or are malformed. F)include_context_awarez(Default client cert source doesn't existc sNztIdH\}}}W||fStttfy&}zt|}||d}~wwN)get_client_cert_and_keyrrrrMutualTLSChannelError)_rr caught_excnew_excrrrcallbackgs z,default_client_cert_source..callback)googleauth transportmtlshas_default_client_cert_sourcerr,)r0rrrdefault_client_cert_sourceUs  r6cs8ttjjjj|dIdH\}}|r|rd||dfSdS)aReturns the client side certificate, private key and passphrase. We look for certificates and keys with the following order of priority: 1. Certificate and key specified by certificate_config.json. Currently, only X.509 workload certificates are supported. Args: certificate_config_path (str): The certificate_config.json file path. Returns: Tuple[bool, bytes, bytes, bytes]: A boolean indicating if cert, key and passphrase are obtained, the cert bytes and key bytes both in PEM format, and passphrase bytes. Raises: google.auth.exceptions.ClientCertError: if problems occurs when getting the cert, key and passphrase. FNT)FNNN)r(r1r2r3 _mtls_helper_get_workload_cert_and_key)certificate_config_pathcertkeyrrrget_client_ssl_credentialsss  r<csV|r|}t|r|IdH\}}n|\}}d||fStIdH\}}}}|||fS)aReturns the client side certificate and private key. The function first tries to get certificate and key from client_cert_callback; if the callback is None or doesn't provide certificate and key, the function tries application default SSL credentials. Args: client_cert_callback (Optional[Callable[[], (bytes, bytes)]]): An optional callback which returns client certificate bytes and private key bytes both in PEM format. Returns: Tuple[bool, bytes, bytes]: A boolean indicating if cert and key are obtained, the cert bytes and key bytes both in PEM format. Raises: google.auth.exceptions.ClientCertError: if problems occurs when getting the cert and key. NT)inspect isawaitabler<)client_cert_callbackresultr:r;has_certr-rrrr+s   r+r*)__doc__r r=loggingr typingr google.authr"google.auth.transport._mtls_helperrgoogle.auth.transport.mtlsr1 getLogger__name___LOGGERbytes SSLContextrr(r6r<r+rrrrs2     ' #