o pkjt; @s(dZddlmZddlZddlZddlmZddlmZddlm Z ddl m Z zddl Z Wne y?Zze dedZ[wwee d dZeeervgZed D]Zz eeeWqReygYnwerveed krved eeeZGd dde jZ  dddZ GdddZ!dS)zAuthorization support for gRPC.)absolute_importN) exceptions) _mtls_helper)mtls)service_accountzWgRPC is not installed from please install the grpcio package to use the gRPC transport. __version__.)Sra`grpcio < 1.83.0 does not support Post-Quantum Cryptography (PQC). Support for non-PQC environments is deprecated. In October 2026, google-auth will raise its minimum requirements to enforce grpcio >= 1.83.0. For more details on Google Cloud's post-quantum security migration, visit: https://cloud.google.com/security/resources/post-quantum-cryptographycs:eZdZdZ d ddfdd ZddZd d ZZS) AuthMetadataPluginaA `gRPC AuthMetadataPlugin`_ that inserts the credentials into each request. .. _gRPC AuthMetadataPlugin: http://www.grpc.io/grpc/python/grpc.html#grpc.AuthMetadataPlugin Args: credentials (google.auth.credentials.Credentials): The credentials to add to requests. request (google.auth.transport.Request): A HTTP transport request object used to refresh credentials as needed. default_host (Optional[str]): A host like "pubsub.googleapis.com". This is used when a self-signed JWT is created from service account credentials. suppress_metrics_header (bool): When enabled, ``x-goog-api-client`` will be stripped from authorization headers. NF)suppress_metrics_headercs*tt|||_||_||_||_dSN)superr __init__ _credentials_request _default_host_suppress_metrics_header)self credentialsrequest default_hostr  __class__X/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/google/auth/transport/grpc.pyrLs  zAuthMetadataPlugin.__init__cCshi}t|jtjr|j|jrd|jnd|j|j|j |j ||j r.d|vr.|d=t | S)zGets the authorization headers for a request. Returns: Sequence[Tuple[str, str]]: A list of request headers (key, value) to add to the request. z https://{}/Nzx-goog-api-client) isinstancerr Credentials_create_self_signed_jwtrformatbefore_requestr method_name service_urlrlistitems)rcontextheadersrrr_get_authorization_headersXs z-AuthMetadataPlugin._get_authorization_headerscCs|||ddS)a Passes authorization metadata into the given callback. Args: context (grpc.AuthMetadataContext): The RPC context. callback (grpc.AuthMetadataPluginCallback): The callback that will be invoked to pass in the authorization metadata. N)r')rr%callbackrrr__call__sszAuthMetadataPlugin.__call__r )__name__ __module__ __qualname____doc__rr'r) __classcell__rrrrr 9s r c Kst||}t|}|r|rtd|s7t}|r*|r*|\} } tj| | d}n |r3t} | j }nt}t ||} tj || fi|S)auCreates a secure authorized gRPC channel. This creates a channel with SSL and :class:`AuthMetadataPlugin`. This channel can be used to create a stub that can make authorized requests. Users can configure client certificate or rely on device certificates to establish a mutual TLS channel, if the `GOOGLE_API_USE_CLIENT_CERTIFICATE` variable is explicitly set to `true`. Example:: import google.auth import google.auth.transport.grpc import google.auth.transport.requests from google.cloud.speech.v1 import cloud_speech_pb2 # Get credentials. credentials, _ = google.auth.default() # Get an HTTP request function to refresh credentials. request = google.auth.transport.requests.Request() # Create a channel. channel = google.auth.transport.grpc.secure_authorized_channel( credentials, regular_endpoint, request, ssl_credentials=grpc.ssl_channel_credentials()) # Use the channel to create a stub. cloud_speech.create_Speech_stub(channel) Usage: There are actually a couple of options to create a channel, depending on if you want to create a regular or mutual TLS channel. First let's list the endpoints (regular vs mutual TLS) to choose from:: regular_endpoint = 'speech.googleapis.com:443' mtls_endpoint = 'speech.mtls.googleapis.com:443' Option 1: create a regular (non-mutual) TLS channel by explicitly setting the ssl_credentials:: regular_ssl_credentials = grpc.ssl_channel_credentials() channel = google.auth.transport.grpc.secure_authorized_channel( credentials, request, regular_endpoint, ssl_credentials=regular_ssl_credentials) Option 2: create a mutual TLS channel by calling a callback which returns the client side certificate and the key (Note that `GOOGLE_API_USE_CLIENT_CERTIFICATE` environment variable must be explicitly set to `true`):: def my_client_cert_callback(): code_to_load_client_cert_and_key() if loaded: return (pem_cert_bytes, pem_key_bytes) raise MyClientCertFailureException() try: channel = google.auth.transport.grpc.secure_authorized_channel( credentials, request, mtls_endpoint, client_cert_callback=my_client_cert_callback) except MyClientCertFailureException: # handle the exception Option 3: use application default SSL credentials. It searches and uses the command in a context aware metadata file, which is available on devices with endpoint verification support (Note that `GOOGLE_API_USE_CLIENT_CERTIFICATE` environment variable must be explicitly set to `true`). See https://cloud.google.com/endpoint-verification/docs/overview:: try: default_ssl_credentials = SslCredentials() except: # Exception can be raised if the context aware metadata is malformed. # See :class:`SslCredentials` for the possible exceptions. # Choose the endpoint based on the SSL credentials type. if default_ssl_credentials.is_mtls: endpoint_to_use = mtls_endpoint else: endpoint_to_use = regular_endpoint channel = google.auth.transport.grpc.secure_authorized_channel( credentials, request, endpoint_to_use, ssl_credentials=default_ssl_credentials) Option 4: not setting ssl_credentials and client_cert_callback. For devices without endpoint verification support or `GOOGLE_API_USE_CLIENT_CERTIFICATE` environment variable is not `true`, a regular TLS channel is created; otherwise, a mutual TLS channel is created, however, the call should be wrapped in a try/except block in case of malformed context aware metadata. The following code uses regular_endpoint, it works the same no matter the created channle is regular or mutual TLS. Regular endpoint ignores client certificate and key:: channel = google.auth.transport.grpc.secure_authorized_channel( credentials, request, regular_endpoint) The following code uses mtls_endpoint, if the created channle is regular, and API mtls_endpoint is confgured to require client SSL credentials, API calls using this channel will be rejected:: channel = google.auth.transport.grpc.secure_authorized_channel( credentials, request, mtls_endpoint) Args: credentials (google.auth.credentials.Credentials): The credentials to add to requests. request (google.auth.transport.Request): A HTTP transport request object used to refresh credentials as needed. Even though gRPC is a separate transport, there's no way to refresh the credentials without using a standard http transport. target (str): The host and port of the service. ssl_credentials (grpc.ChannelCredentials): Optional SSL channel credentials. This can be used to specify different certificates. This argument is mutually exclusive with client_cert_callback; providing both will raise an exception. If ssl_credentials and client_cert_callback are None, application default SSL credentials are used if `GOOGLE_API_USE_CLIENT_CERTIFICATE` environment variable is explicitly set to `true`, otherwise one way TLS SSL credentials are used. client_cert_callback (Callable[[], (bytes, bytes)]): Optional callback function to obtain client certicate and key for mutual TLS connection. This argument is mutually exclusive with ssl_credentials; providing both will raise an exception. This argument does nothing unless `GOOGLE_API_USE_CLIENT_CERTIFICATE` environment variable is explicitly set to `true`. kwargs: Additional arguments to pass to :func:`grpc.secure_channel`. Returns: grpc.Channel: The created gRPC channel. Raises: google.auth.exceptions.MutualTLSChannelError: If mutual TLS channel creation failed for any reason. zUReceived both ssl_credentials and client_cert_callback; these are mutually exclusive.certificate_chain private_key) r grpcmetadata_call_credentialsrMalformedErrorrcheck_use_client_certssl_channel_credentialsSslCredentialsssl_credentialscomposite_channel_credentialssecure_channel) rrtargetr8client_cert_callbackkwargsmetadata_plugingoogle_auth_credentialsuse_client_certcertkeyadc_ssl_credentilscomposite_credentialsrrrsecure_authorized_channel~s,   rEc@s0eZdZdZddZeddZeddZdS) r7a*Class for application default SSL credentials. Mutual TLS (mTLS) is enabled if either: 1. The `GOOGLE_API_USE_CLIENT_CERTIFICATE` environment variable is explicitly set to `"true"`. 2. The `GOOGLE_API_USE_CLIENT_CERTIFICATE` environment variable is unset or empty, but a valid workload certificate configuration is found (e.g., via the `GOOGLE_API_CERTIFICATE_CONFIG` environment variable or the default gcloud config path). See https://google.aip.dev/auth/4114 for client certificate discovery details. If client certificate usage is enabled, then for devices with endpoint verification support, a device certificate will be automatically loaded and mutual TLS will be established. See https://cloud.google.com/endpoint-verification/docs/overview. cCs$t}|s d|_dSt|_dS)NF)rr5_is_mtlsrhas_default_client_cert_source)rr@rrrrHs zSslCredentials.__init__c Cs|jr>z#t\}}}}|rtj||d|_n t|_d|_W|jSW|jStjtfy=}zt |}||d}~wwt|_|jS)aGet the created SSL channel credentials. For devices with endpoint verification support, if the device certificate loading has any problems, corresponding exceptions will be raised. For a device without endpoint verification support, no exceptions will be raised. Returns: grpc.ChannelCredentials: The created grpc channel credentials. Raises: google.auth.exceptions.MutualTLSChannelError: If mutual TLS channel creation failed for any reason. r/FN) rFrget_client_ssl_credentialsr2r6_ssl_credentialsrClientCertErrorOSErrorMutualTLSChannelError)rhas_certrArB_ caught_excnew_excrrrr8Os&    zSslCredentials.ssl_credentialscCs|jS)z?Indicates if the created SSL channel credentials is mutual TLS.)rF)rrrris_mtlsqszSslCredentials.is_mtlsN)r*r+r,r-rpropertyr8rQrrrrr75s !r7)NN)"r- __future__rloggingwarnings google.authrgoogle.auth.transportrr google.oauth2rr2 ImportErrorrOgetattr _grpc_ver_strrstr_partssplit_partappendint ValueErrortuplewarn FutureWarning getLoggerr*_LOGGERr rEr7rrrrsN          I 8