a w=icwr@sdZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddl mZddl mZdd l mZd Zd Zd Zd ZGdddeZddZddZddZdifddZGdddejZdS)aAWS Credentials and AWS Signature V4 Request Signer. This module provides credentials to access Google Cloud resources from Amazon Web Services (AWS) workloads. These credentials are recommended over the use of service account credentials in AWS as they do not involve the management of long-live service account private keys. AWS Credentials are initialized using external_account arguments which are typically loaded from the external credentials JSON file. Unlike other Credentials that can be initialized with a list of explicit arguments, secrets or credentials, external account clients use the environment and hints/guidelines provided by the external_account JSON file to retrieve credentials and exchange them for Google access tokens. This module also provides a basic implementation of the `AWS Signature Version 4`_ request signing algorithm. AWS Credentials use serialized signed requests to the `AWS STS GetCallerIdentity`_ API that can be exchanged for Google access tokens via the GCP STS endpoint. .. _AWS Signature Version 4: https://docs.aws.amazon.com/general/latest/gr/signature-version-4.html .. _AWS STS GetCallerIdentity: https://docs.aws.amazon.com/STS/latest/APIReference/API_GetCallerIdentity.html N) http_client)urllib)urljoin)_helpers)environment_vars) exceptions)external_accountzAWS4-HMAC-SHA256 aws4_requestzx-amz-security-tokenz x-amz-datec@s&eZdZdZddZdifddZdS) RequestSignerzImplements an AWS request signer based on the AWS Signature Version 4 signing process. https://docs.aws.amazon.com/general/latest/gr/signature-version-4.html cCs ||_dS)zInstantiates an AWS request signer used to compute authenticated signed requests to AWS APIs based on the AWS Signature Version 4 signing process. Args: region_name (str): The AWS region to use. N) _region_name)self region_namer`/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/google/auth/aws.py__init__IszRequestSigner.__init__c Cs|d}|d}|d}|p$i}tj|} tjt|t| j} | jr\| j dkrdt dt | j| jprdt | j ||j|||||d } | d| jd } d | vr| d | t<|D]} || | | <q|d ur|| t<||| d }|r||d <|S)a^Generates the signed request for the provided HTTP request for calling an AWS API. This follows the steps described at: https://docs.aws.amazon.com/general/latest/gr/sigv4_signing.html Args: aws_security_credentials (Mapping[str, str]): A dictionary containing the AWS security credentials. url (str): The AWS service URL containing the canonical URI and query string. method (str): The HTTP method used to call this API. request_payload (Optional[str]): The optional request payload if available. additional_headers (Optional[Mapping[str, str]]): The optional additional headers needed for the requested AWS API. Returns: Mapping[str, str]: The AWS signed request dictionary object. access_key_idsecret_access_keysecurity_tokenhttpszInvalid AWS service URL/) host canonical_uricanonical_querystringmethodregion access_key secret_keyrrequest_payloadadditional_headersauthorization_header) Authorizationramz_dateNurlrheadersdata)getrparseurlparser posixpathnormpathpathhostnamescheme ValueError#_generate_authentication_header_map_get_canonical_querystringqueryr _AWS_DATE_HEADER_AWS_SECURITY_TOKEN_HEADER)r aws_security_credentialsr$rrrrrruriZnormalized_uriZ header_mapr%keyZsigned_requestrrrget_request_optionsSsF      z!RequestSigner.get_request_optionsN)__name__ __module__ __qualname____doc__rr8rrrrr Cs r cCstj|}i}|D]P}tjj|dd}g||<||D]}||tjj|ddq8||qt|}|g}|D]$}||D]}|d||qqd |S)aGenerates the canonical query string given a raw query string. Logic is based on https://docs.aws.amazon.com/general/latest/gr/sigv4-create-canonical-request.html Args: query (str): The raw query string. Returns: str: The canonical query string. z-_.~)safez{}={}&) rr(parse_qsquoteappendsortlistkeysformatjoin)r2Z querystringZquerystring_encoded_mapr7Z quote_keyitemZ sorted_keysZquerystring_encoded_pairsrrrr1s"    r1cCst||dtjS)zCreates the HMAC-SHA256 hash of the provided message using the provided key. Args: key (str): The HMAC-SHA256 key to use. msg (str): The message to hash. Returns: str: The computed hash bytes. utf-8)hmacnewencodehashlibsha256digest)r7msgrrr_signs rPcCs6td|d|}t||}t||}t|d}|S)aCalculates the signing key used to calculate the signature for AWS Signature Version 4 based on: https://docs.aws.amazon.com/general/latest/gr/sigv4-calculate-signature.html Args: key (str): The AWS secret access key. date_stamp (str): The '%Y%m%d' date format. region_name (str): The AWS region. service_name (str): The AWS service name, eg. sts. Returns: str: The signing key bytes. ZAWS4rHr )rPrK)r7 date_stampr service_nameZk_dateZk_regionZ k_serviceZ k_signingrrr_get_signing_keys    rSrc  Csb|dd} t} | d} | d} i}| D]}| |||<q2|durX||t<||d<d|vrp| |t<d}t|}| |D]}d ||||}qd |}t |pdd }d ||||||}d | || t}d t| |t |d }t|| || }t||d t j }d t||||}d|i}d|vr^| |d<|S)a Generates the authentication header map needed for generating the AWS Signature Version 4 signed request. Args: host (str): The AWS service URL hostname. canonical_uri (str): The AWS service URL path name. canonical_querystring (str): The AWS service URL query string. method (str): The HTTP method used to call this API. region (str): The AWS region. access_key (str): The AWS access key ID. secret_key (str): The AWS secret access key. security_token (Optional[str]): The AWS security session token. This is available for temporary sessions. request_payload (Optional[str]): The optional request payload if available. additional_headers (Optional[Mapping[str, str]]): The optional additional headers needed for the requested AWS API. Returns: Mapping[str, str]: The AWS authentication header dictionary object. This contains the x-amz-date and authorization header information. .rz%Y%m%dT%H%M%SZz%Y%m%dNrdaterz{}{}:{} ;rHz{} {} {} {} {} {}z {}/{}/{}/{}z {} {} {} {}z3{} Credential={}/{}, SignedHeaders={}, Signature={}r r")splitrutcnowstrftimelowerr4r3rCrDrBrErFrLrMrK hexdigest_AWS_REQUEST_TYPE_AWS_ALGORITHMrSrIrJ)rrrrrrrrrrrR current_timer"rQZ full_headersr7Zcanonical_headers header_keysZsigned_headersZ payload_hashZcanonical_requestZcredential_scopeZstring_to_signZ signing_key signaturer Zauthentication_headerrrrr0sb$        r0csjeZdZdZdfdd ZddZddZd d Zd d Zd dZ e fddZ e fddZ Z S) CredentialszAWS external account credentials. This is used to exchange serialized AWS signature v4 signed requests to AWS STS GetCallerIdentity service for Google access tokens. Nc stt|j|||||d||p&i}|dp4d|_|d|_|d|_|d|_|d|_d|_ d|_ ||_ t d |j}|r|\}} nd \}} |d ks|jdurtd nt| pdd krtd| dS)a]Instantiates an AWS workload external account credentials object. Args: audience (str): The STS audience field. subject_token_type (str): The subject token type. token_url (str): The STS endpoint URL. credential_source (Mapping): The credential source dictionary used to provide instructions on how to retrieve external credential to be exchanged for Google access tokens. args (List): Optional positional arguments passed into the underlying :meth:`~external_account.Credentials.__init__` method. kwargs (Mapping): Optional keyword arguments passed into the underlying :meth:`~external_account.Credentials.__init__` method. Raises: google.auth.exceptions.RefreshError: If an error is encountered during access token retrieval logic. ValueError: For invalid parameters. .. note:: Typically one of the helper constructors :meth:`from_file` or :meth:`from_info` are used instead of calling the constructor directly. )audiencesubject_token_type token_urlcredential_sourceZenvironment_idrZ region_urlr$Zregional_cred_verification_urlZimdsv2_session_token_urlNz^(aws)([\d]+)$)NNZawsz)No valid AWS 'credential_source' providedz7aws version '{}' is not supported in the current build.)superrarr'Z_environment_id _region_url_security_credentials_url_cred_verification_url_imdsv2_session_token_url_region_request_signer_target_resourcerematchgroupsr/intrE) r rbrcrdreargskwargsmatchesZenv_idZ env_version __class__rrr\sD    zCredentials.__init__c Cs|durJ|jdurJddi}||jd|d}|jdkrBtd|j|j}nd}|jdurv|||j||_t |j|_| ||}|j ||j d|jd }|d }|j|d <i}|d |d <|d |d <g|d <t|D]} |d | || dqtjtj|dddS)aiRetrieves the subject token using the credential_source object. The subject token is a serialized `AWS GetCallerIdentity signed request`_. The logic is summarized as: Retrieve the AWS region from the AWS_REGION or AWS_DEFAULT_REGION environment variable or from the AWS metadata server availability-zone if not found in the environment variable. Check AWS credentials in environment variables. If not found, retrieve from the AWS metadata server security-credentials endpoint. When retrieving AWS credentials from the metadata server security-credentials endpoint, the AWS role needs to be determined by calling the security-credentials endpoint without any argument. Then the credentials can be retrieved via: security-credentials/role_name Generate the signed request to AWS STS GetCallerIdentity action. Inject x-goog-cloud-target-resource into header and serialize the signed request. This will be the subject-token to pass to GCP STS. .. _AWS GetCallerIdentity signed request: https://cloud.google.com/iam/docs/access-resources-aws#exchange-token Args: request (google.auth.transport.Request): A callable used to make HTTP requests. Returns: str: The retrieved subject token. Nz$X-aws-ec2-metadata-token-ttl-secondsZ300PUTr#z$Unable to retrieve AWS Session Tokenz{region}POSTr%zx-goog-cloud-target-resourcer$r)r7value),:T) separators sort_keys)rkstatusr RefreshErrorr&rm _get_regionrhrlr _get_security_credentialsr8rjreplacer'rnsortedrDrArr(r@jsondumps) r requestr%Zimdsv2_session_token_responseimdsv2_session_tokenr5Zrequest_optionsZrequest_headersZaws_signed_reqr7rrrretrieve_subject_tokensL!      z"Credentials.retrieve_subject_tokencCstjtj}|dur|Stjtj}|dur4|S|jsDtdd}|durXd|i}||jd|d}t |j dr|j dn|j }|j dkrtd ||dd S) aRetrieves the current AWS region from either the AWS_REGION or AWS_DEFAULT_REGION environment variable or from the AWS metadata server. Args: request (google.auth.transport.Request): A callable used to make HTTP requests. url (str): The AWS metadata server region URL. imdsv2_session_token (str): The AWS IMDSv2 session token to be added as a header in the requests to AWS metadata endpoint. Returns: str: The current AWS region. Raises: google.auth.exceptions.RefreshError: If an error occurs while retrieving the AWS region. NzUnable to determine AWS regionX-aws-ec2-metadata-tokenGETr#decoderHryzUnable to retrieve AWS region) osenvironr'rZ AWS_REGIONZAWS_DEFAULT_REGIONrhrrhasattrr&rr)r rr$rZenv_aws_regionr%response response_bodyrrrr s*   zCredentials._get_regioncCsvtjtj}tjtj}tjtj}|r>|r>|||dS|||}||||}|d|d|ddS)aRetrieves the AWS security credentials required for signing AWS requests from either the AWS security credentials environment variables or from the AWS metadata server. Args: request (google.auth.transport.Request): A callable used to make HTTP requests. imdsv2_session_token (str): The AWS IMDSv2 session token to be added as a header in the requests to AWS metadata endpoint. Returns: Mapping[str, str]: The AWS security credentials dictionary object. Raises: google.auth.exceptions.RefreshError: If an error occurs while retrieving the AWS security credentials. )rrrZ AccessKeyIdZSecretAccessKeyToken) rrr'rZAWS_ACCESS_KEY_IDZAWS_SECRET_ACCESS_KEYZAWS_SESSION_TOKEN_get_metadata_role_name"_get_metadata_security_credentials)r rrZenv_aws_access_key_idZenv_aws_secret_access_keyZenv_aws_session_token role_name credentialsrrrrAs$ z%Credentials._get_security_credentialscCstddi}|dur||d<|d|j|d|d}t|jdrH|jd n|j}|jtjkrft d |t |}|S) aJRetrieves the AWS security credentials required for signing AWS requests from the AWS metadata server. Args: request (google.auth.transport.Request): A callable used to make HTTP requests. role_name (str): The AWS role name required by the AWS metadata server security_credentials endpoint in order to return the credentials. imdsv2_session_token (str): The AWS IMDSv2 session token to be added as a header in the requests to AWS metadata endpoint. Returns: Mapping[str, str]: The AWS metadata server security credentials response. Raises: google.auth.exceptions.RefreshError: If an error occurs while retrieving the AWS security credentials. z Content-Typezapplication/jsonNrz{}/{}rr#rrHz+Unable to retrieve AWS security credentials) rErirr&rrrOKrrrloads)r rrrr%rrZcredentials_responserrrrqs$    z.Credentials._get_metadata_security_credentialscCsr|jdurtdd}|dur(d|i}||jd|d}t|jdrP|jdn|j}|jtjkrntd||S) aRetrieves the AWS role currently attached to the current AWS workload by querying the AWS metadata server. This is needed for the AWS metadata server security credentials endpoint in order to retrieve the AWS security credentials needed to sign requests to AWS APIs. Args: request (google.auth.transport.Request): A callable used to make HTTP requests. imdsv2_session_token (str): The AWS IMDSv2 session token to be added as a header in the requests to AWS metadata endpoint. Returns: str: The AWS role name. Raises: google.auth.exceptions.RefreshError: If an error occurs while retrieving the AWS role name. NzIUnable to determine the AWS metadata server security credentials endpointrrr#rrHz Unable to retrieve AWS role name) rirrrr&rrrr)r rrr%rrrrrrs&   z#Credentials._get_metadata_role_namec stt|j|fi|S)aCreates an AWS Credentials instance from parsed external account info. Args: info (Mapping[str, str]): The AWS external account info in Google format. kwargs: Additional arguments to pass to the constructor. Returns: google.auth.aws.Credentials: The constructed credentials. Raises: ValueError: For invalid parameters. )rgra from_info)clsinfortrvrrrszCredentials.from_infoc stt|j|fi|S)aHCreates an AWS Credentials instance from an external account json file. Args: filename (str): The path to the AWS external account json file. kwargs: Additional arguments to pass to the constructor. Returns: google.auth.aws.Credentials: The constructed credentials. )rgra from_file)rfilenamertrvrrrs zCredentials.from_file)N)r9r:r;r<rrrrrr classmethodrr __classcell__rrrvrraVs Dk601.ra)r<rLrIrrr*roZ six.movesrrZsix.moves.urllib.parserZ google.authrrrrr]r\r4r3objectr r1rPrSr0rarrrrs2       ]$ o