| Server IP : 172.67.201.108 / Your IP : 216.73.216.55 Web Server : Apache/2.4.68 (Amazon Linux) OpenSSL/3.5.7 System : Linux ip-172-31-69-123.ec2.internal 6.1.177-224.371.amzn2023.x86_64 #1 SMP PREEMPT_DYNAMIC Mon Jul 27 20:28:29 UTC 2026 x86_64 User : ec2-user ( 1000) PHP Version : 8.4.24 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /lib/python3.9/site-packages/awscli/botocore/__pycache__/ |
Upload File : |
a
wA�ig� � @ sZ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
d dl m Z d dl
mZ d dlZd dlZd dlZd dlmZ d dlmZ d dlmZmZ d dlmZ d d lm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z* d d
l+m,Z, d d l-m.Z.m/Z/ d d l0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z< d d
lm=Z= d dl>m?Z?m@Z@ e�AeB�ZCe
dg d�dd�ZDdZEdZFdXdd�ZGG dd� d�ZHdd� ZIdd � ZJd d � ZKdYd d!�ZLd"d#� ZMd$d%� ZNd&d'� ZOG d(d)� d)�ZPG d*d+� d+eP�ZQG d,d-� d-eQ�ZRG d.d/� d/�ZSG d0d1� d1eS�ZTG d2d3� d3eT�ZUG d4d5� d5eT�ZVG d6d7� d7�ZWG d8d9� d9eW�ZXG d:d;� d;eW�ZYG d<d=� d=eW�ZZG d>d?� d?eW�Z[G d@dA� dAeW�Z\G dBdC� dCeW�Z]G dDdE� dEeW�Z^G dFdG� dGeW�Z_G dHdI� dIeW�Z`G dJdK� dK�ZaG dLdM� dMeW�ZbG dNdO� dO�ZcG dPdQ� dQeS�ZdG dRdS� dSeW�ZeG dTdU� dU�ZfG dVdW� dWeW�ZgdS )Z� N)�
namedtuple)�deepcopy)�sha1)�EC)�UNSIGNED)�compat_shell_split�
total_seconds)�Config)� ClientError�ConfigNotFound�CredentialRetrievalError�InfiniteLoopConfigError�InvalidConfigError�
LoginError� LoginInsufficientPermissions�LoginRefreshPasswordChanged�LoginRefreshTokenExpired�LoginTokenLoadError�MetadataRetrievalError�PartialCredentialsError� RefreshWithMFAUnsupportedError�UnauthorizedSSOTokenError�UnknownCredentialError)�SSOTokenProvider)�register_feature_id�register_feature_ids) � ArnParser�ContainerMetadataFetcher�FileWebIdentityTokenLoader�InstanceMetadataFetcher�
JSONFileCache�LoginCredentialsLoader�SSOTokenLoader� build_add_dpop_header_handler� get_login_token_cache_directory�original_ld_library_path�parse_key_val_file�resolve_imds_endpoint_mode��parse)�tzlocal�tzutc�ReadOnlyCredentials)�
access_key�
secret_key�token�
account_id�N)�defaultsiX � c s � � d�p d}� � d�}� � d�}� �� �d�du}� � d�t� �t� � d�d�}|du r^i }t� }t� } tt||� j d d
�|d �d �}
t
� ||d
�} t � fdd�t � |�||t
|| |
g�| d�} || g}
| j||d�}t� t� | |
g}|
| | }|�r|�|� t�d� t|d�}|S )z�Create a default credential resolver.
This creates a pre-configured credential resolver
that includes the default lookup chain for
credentials.
�profile�defaultZmetadata_service_timeoutZ metadata_service_num_attemptsN� ec2_metadata_service_endpoint�ec2_metadata_v1_disabled)r6 Z"ec2_metadata_service_endpoint_modeZ ec2_credential_refresh_windowr7 T)�truncate)Ztimeout� num_attempts�
user_agent�config)�iam_role_fetcher)�cache� region_namec s � j S r1 )� full_config� ��sessionr@ �?/usr/lib/python3.9/site-packages/awscli/botocore/credentials.py�<lambda>} � z,create_credential_resolver.<locals>.<lambda>)� load_config�client_creatorr= � profile_name�credential_sourcer�profile_provider_builder�rH �disable_env_varszWSkipping environment variable credential check because profile name was explicitly set.�� providers)�get_config_variableZinstance_variables�getr' �!_DEFAULT_ADVISORY_REFRESH_TIMEOUT� EnvProvider�ContainerProvider�InstanceMetadataProviderr r: �ProfileProviderBuilder�AssumeRoleProvider�_get_client_creator� CanonicalNameCredentialSourcerrN �OriginalEC2Provider� BotoProvider�remove�logger�debug�CredentialResolver)rB r= r> rH Zmetadata_timeoutr9 rL Z imds_configZ env_providerZcontainer_providerZinstance_metadata_providerrJ �assume_role_providerZ pre_profile�profile_providersZ post_profilerN �resolverr@ rA rC �create_credential_resolverQ sx
����
�� �
�� ���
�
rb c @ sT e Zd ZdZddd�Zddd�Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dS )rU a� This class handles the creation of profile based providers.
NOTE: This class is only intended for internal use.
This class handles the creation and ordering of the various credential
providers that primarly source their configuration from the shared config.
This is needed to enable sharing between the default credential chain and
the source profile chain created by the assume role provider.
Nc C s" || _ || _|| _|| _|| _d S r1 )�_session�_cache� _region_name�_sso_token_cache�_signin_token_cache)�selfrB r= r> Zsso_token_cacheZsignin_token_cacher@ r@ rC �__init__� s
z ProfileProviderBuilder.__init__Fc C s6 | � ||�| �|�| �|�| �|�| �|�| �|�gS r1 )� _create_web_identity_provider�_create_sso_provider�"_create_shared_credential_provider�_create_login_provider�_create_process_provider�_create_config_provider�rh rH rL r@ r@ rC rN � s ��z ProfileProviderBuilder.providersc s t |� fdd�d�S )Nc s � j jS r1 �rc r? r@ �rh r@ rC rD � rE zAProfileProviderBuilder._create_process_provider.<locals>.<lambda>)rH rF )�ProcessProvider�rh rH r@ rr rC rn � s
�z/ProfileProviderBuilder._create_process_providerc C s | j �d�}t||d�S )NZcredentials_file)rH �creds_filename)rc rO �SharedCredentialProvider)rh rH Zcredential_filer@ r@ rC rl � s
�z9ProfileProviderBuilder._create_shared_credential_providerc C s | j �d�}t||d�S )N� config_file)rH �config_filename)rc rO �ConfigProvider)rh rH rw r@ r@ rC ro � s
�z.ProfileProviderBuilder._create_config_providerc s& t � fdd�t� j� j�� j||d�S )Nc s � j jS r1 rq r@ rr r@ rC rD � rE zFProfileProviderBuilder._create_web_identity_provider.<locals>.<lambda>)rF rG r= rH rL )�!AssumeRoleWithWebIdentityProviderrW rc re rd rp r@ rr rC rj � s
��z4ProfileProviderBuilder._create_web_identity_providerc s2 t � fdd�� jj|� j� jt� j� j|d�d�S )Nc s � j jS r1 rq r@ rr r@ rC rD � rE z=ProfileProviderBuilder._create_sso_provider.<locals>.<lambda>)r= rH )rF rG rH r= � token_cache�token_provider)� SSOProviderrc �
create_clientrd rf r rt r@ rr rC rk � s
��z+ProfileProviderBuilder._create_sso_providerc s t � j� fdd�|� jjd�S )Nc s � j jS r1 rq r@ rr r@ rC rD rE z?ProfileProviderBuilder._create_login_provider.<locals>.<lambda>)r{ rF rH rG )�
LoginProviderrg rc r~ rt r@ rr rC rm s
�z-ProfileProviderBuilder._create_login_provider)NNNN)F) �__name__�
__module__� __qualname__�__doc__ri rN rn rl ro rj rk rm r@ r@ r@ rC rU � s
�
rU c C s t | �}|�� S r1 )rb �load_credentials)rB ra r@ r@ rC �get_credentials
s r� c C s t j �t� �S r1 )�datetimeZnowr* r@ r@ r@ rC �
_local_now s r� c C s t | tj�r| S t| �S r1 )�
isinstancer� r) )�valuer@ r@ rC �_parse_if_needed s r� Fc C s& t | tj�r"|r| �� S | �d�S | S )Nz%Y-%m-%dT%H:%M:%S%Z)r� r� Z isoformat�strftime)r� �isor@ r@ rC �_serialize_if_needed s
r� c s � �fdd�}|S )Nc s* d� i}|j f i |�� �j| fi |��S )Nr> )�updater~ )Z service_name�kwargsZcreate_client_kwargs�r> rB r@ rC rG % s z+_get_client_creator.<locals>.client_creatorr@ )rB r> rG r@ r� rC rW $ s rW c s � �fdd�}|S )Nc s: � j f i ���} | d }|d |d |d t|d �d�S )N� Credentials� AccessKeyId�SecretAccessKey� SessionToken�
Expiration�r- r. r/ � expiry_time)� assume_roler� )�response� credentials��client�paramsr@ rC �refresh. s
�z-create_assume_role_refresher.<locals>.refreshr@ )r� r� r� r@ r� rC � create_assume_role_refresher- s r� c C s G dd� d�}|| �S )Nc @ s e Zd Zdd� Zdd� ZdS )z/create_mfa_serial_refresher.<locals>._Refresherc S s || _ d| _d S )NF)�_refresh�_has_been_called)rh r� r@ r@ rC ri ? s z8create_mfa_serial_refresher.<locals>._Refresher.__init__c S s | j r t� �d| _ | �� S �NT)r� r r� rr r@ r@ rC �__call__C s z8create_mfa_serial_refresher.<locals>._Refresher.__call__N)r� r� r� ri r� r@ r@ r@ rC �
_Refresher> s r� r@ )Zactual_refreshr� r@ r@ rC �create_mfa_serial_refresher= s r� c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� ZdS ) r� a�
Holds the credentials needed to authenticate requests.
:param str access_key: The access key part of the credentials.
:param str secret_key: The secret key part of the credentials.
:param str token: The security token, valid only for session credentials.
:param str method: A string which identifies where the credentials
were found.
:param str account_id: (optional) An account ID associated with the credentials.
Nc C s6 || _ || _|| _|d u r d}|| _|| _| �� d S )NZexplicit)r- r. r/ �methodr0 �
_normalize)rh r- r. r/ r� r0 r@ r@ rC ri [ s zCredentials.__init__c C s$ t j�| j�| _t j�| j�| _d S r1 )�botocore�compat�ensure_unicoder- r. rr r@ r@ rC r� i s zCredentials._normalizec C s t | j| j| j| j�S r1 )r, r- r. r/ r0 rr r@ r@ rC �get_frozen_credentialss s �z"Credentials.get_frozen_credentialsc s � �fdd�}|S )Nc s t �� d �S r1 )�getattrr@ ��
property_namerh r@ rC � get_propertyy s z7Credentials.get_deferred_property.<locals>.get_propertyr@ )rh r� r� r@ r� rC �get_deferred_propertyx s z!Credentials.get_deferred_property)NNN)r� r� r� r� ri r� r� r� r@ r@ r@ rC r� O s �
r� c @ s� e Zd ZdZeZeZedddfdd�Z dd� Z
e d%dd��Z e
d d
� �Zejd d
� �Ze
d d
� �Zejdd
� �Ze
dd� �Zejdd� �Ze
dd� �Zejdd� �Zdd� Zd&dd�Zdd� Zdd � Zd d � Zed d � �Zd!d"� Zd#d$� ZdS )'�RefreshableCredentialsa@
Holds the credentials needed to authenticate requests. In addition, it
knows how to refresh itself.
:param str access_key: The access key part of the credentials.
:param str secret_key: The secret key part of the credentials.
:param str token: The security token, valid only for session credentials.
:param function refresh_using: Callback function to refresh the credentials.
:param str method: A string which identifies where the credentials
were found.
:param function time_fetcher: Callback function to retrieve current time.
Nc C sr || _ || _|| _|| _|
| _|| _|| _t�� | _ || _
t ||||
�| _ | �
� |d ur`|| _| d urn| | _d S r1 )�_refresh_using� _access_key� _secret_key�_token� _account_id� _expiry_time�
_time_fetcher� threading�Lock�
_refresh_lockr� r, �_frozen_credentialsr� �_advisory_refresh_timeout�_mandatory_refresh_timeout) rh r- r. r/ r� �
refresh_usingr� � time_fetcher�advisory_timeout�mandatory_timeoutr0 r@ r@ rC ri � s"
�z RefreshableCredentials.__init__c C s$ t j�| j�| _t j�| j�| _d S r1 )r� r� r� r� r� rr r@ r@ rC r� � s z!RefreshableCredentials._normalizec C sb i }|d ur||d< |d ur$||d< | f |d |d |d | � |d �|||�d�d�|��}|S ) Nr� r� r- r. r/ r� r0 )r- r. r/ r� r� r� r0 )�_expiry_datetimerP )�cls�metadatar� r� r� r� r� �instancer@ r@ rC �create_from_metadata� s" ��
z+RefreshableCredentials.create_from_metadatac C s | � � | jS �z�Warning: Using this property can lead to race conditions if you
access another property subsequently along the refresh boundary.
Please use get_frozen_credentials instead.
)r� r� rr r@ r@ rC r- � s z!RefreshableCredentials.access_keyc C s
|| _ d S r1 )r� �rh r� r@ r@ rC r- � s c C s | � � | jS r� )r� r� rr r@ r@ rC r. � s z!RefreshableCredentials.secret_keyc C s
|| _ d S r1 )r� r� r@ r@ rC r. � s c C s | � � | jS r� )r� r� rr r@ r@ rC r/ � s z RefreshableCredentials.tokenc C s
|| _ d S r1 )r� r� r@ r@ rC r/ � s c C s | � � | jS r� )r� r� rr r@ r@ rC r0 � s z!RefreshableCredentials.account_idc C s
|| _ d S r1 )r� r� r@ r@ rC r0 s c C s | j | �� }t|�S r1 )r� r� r )rh Zdeltar@ r@ rC �_seconds_remaining s z)RefreshableCredentials._seconds_remainingc C s: | j du rdS |du r | j}| �� |kr,dS t�d� dS )a� Check if a refresh is needed.
A refresh is needed if the expiry time associated
with the temporary credentials is less than the
provided ``refresh_in``. If ``time_delta`` is not
provided, ``self.advisory_refresh_needed`` will be used.
For example, if your temporary credentials expire
in 10 minutes and the provided ``refresh_in`` is
``15 * 60``, then this function will return ``True``.
:type refresh_in: int
:param refresh_in: The number of seconds before the
credentials expire in which refresh attempts should
be made.
:return: True if refresh needed, False otherwise.
NFz!Credentials need to be refreshed.T)r� r� r� r\ r] �rh �
refresh_inr@ r@ rC �refresh_needed
s
z%RefreshableCredentials.refresh_neededc C s | j dd�S )Nr )r� )r� rr r@ r@ rC � _is_expired, s z"RefreshableCredentials._is_expiredc C s� | � | j�sd S | j�d�rpzD| � | j�s:W | j�� d S | � | j�}| j|d� W | j�� d S | j�� 0 nZ| � | j�r�| j�8 | � | j�s�W d � d S | jdd� W d � n1 s�0 Y d S )NF)� is_mandatoryT)r� r� r� �acquire�releaser� �_protected_refresh)rh Zis_mandatory_refreshr@ r@ rC r� 0 s*
��
� z RefreshableCredentials._refreshc C s� z | � � }W n6 tyB |r"dnd}tjd|dd� |r<� Y d S 0 | �|� t| j| j| j| j �| _
| � � r�d}t�|� t |��d S )NZ mandatoryZadvisoryzARefreshing temporary credentials failed during %s refresh period.T��exc_infozLCredentials were refreshed, but the refreshed credentials are still expired.)
r� � Exceptionr\ Zwarning�_set_from_datar, r� r� r� r� r� r� � RuntimeError)rh r� r� Z period_name�msgr@ r@ rC r� N s* �
��
z)RefreshableCredentials._protected_refreshc C s t | �S r1 r( )Ztime_strr@ r@ rC r� u s z'RefreshableCredentials._expiry_datetimec s� g d�}� s|}n� fdd�|D �}|rDd}t | j|d�|� d��� d | _� d | _� d | _t� d
�| _� �d �| _ t
� d | j� | � � d S )
Nr� c s g | ]}|� vr|�qS r@ r@ )�.0�k��datar@ rC �
<listcomp>~ rE z9RefreshableCredentials._set_from_data.<locals>.<listcomp>z7Credential refresh failed, response did not contain: %s�, ��provider� error_msgr- r. r/ r� r0 z(Retrieved credentials will expire at: %s)
r r� �joinr- r. r/ r) r� rP r0 r\ r] r� )rh r� Z
expected_keysZ missing_keys�messager@ r� rC r� y s& �
�z%RefreshableCredentials._set_from_datac C s | � � | jS )a� Return immutable credentials.
The ``access_key``, ``secret_key``, and ``token`` properties
on this class will always check and refresh credentials if
needed before returning the particular credentials.
This has an edge case where you can get inconsistent
credentials. Imagine this:
# Current creds are "t1"
tmp.access_key ---> expired? no, so return t1.access_key
# ---- time is now expired, creds need refreshing to "t2" ----
tmp.secret_key ---> expired? yes, refresh and return t2.secret_key
This means we're using the access key from t1 with the secret key
from t2. To fix this issue, you can request a frozen credential object
which is guaranteed not to change.
The frozen credentials returned from this method should be used
immediately and then discarded. The typical usage pattern would
be::
creds = RefreshableCredentials(...)
some_code = SomeSignerObject()
# I'm about to sign the request.
# The frozen credentials are only used for the
# duration of generate_presigned_url and will be
# immediately thrown away.
request = some_code.sign_some_request(
with_credentials=creds.get_frozen_credentials())
print("Signed request:", request)
)r� r� rr r@ r@ rC r� � s "z-RefreshableCredentials.get_frozen_credentials)NN)N) r� r� r� r� rQ r� �"_DEFAULT_MANDATORY_REFRESH_TIMEOUTr� r� ri r� � classmethodr� �propertyr- �setterr. r/ r0 r� r� r� r� r� � staticmethodr� r� r� r@ r@ r@ rC r� sL
�
�
" '
r� c s. e Zd ZdZefdd�Zd� fdd� Z� ZS )� DeferredRefreshableCredentialszyRefreshable credentials that don't require initial credentials.
refresh_using will be called upon first access.
c C sD || _ d | _d | _d | _d | _d | _|| _t�� | _ || _
d | _ d S r1 ) r� r� r� r� r� r� r� r� r� r� r� r� )rh r� r� r� r@ r@ rC ri � s
z'DeferredRefreshableCredentials.__init__Nc s | j d u rdS t� �|�S r� )r� �superr� r� �� __class__r@ rC r� � s
z-DeferredRefreshableCredentials.refresh_needed)N)r� r� r� r� r� ri r� �
__classcell__r@ r@ r� rC r� � s r� c @ sZ e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�CachedCredentialFetcherr3 Nc C s< |d u r i }|| _ | �� | _|d u r*| j}|| _t� | _d S r1 )rd �_create_cache_key�
_cache_key� DEFAULT_EXPIRY_WINDOW_SECONDS�_expiry_window_seconds�set� feature_ids)rh r= �expiry_window_secondsr@ r@ rC ri � s
z CachedCredentialFetcher.__init__c C s t d��d S )Nz_create_cache_key()��NotImplementedErrorrr r@ r@ rC r� � s z)CachedCredentialFetcher._create_cache_keyc C s$ |� dd�� tjjd�}|� dd�S )N�:�_�/)�replace�os�path�sep)rh �filenamer@ r@ rC �_make_file_safe� s z'CachedCredentialFetcher._make_file_safec C s t d��d S )Nz_get_credentials()r� rr r@ r@ rC �_get_credentials� s z(CachedCredentialFetcher._get_credentialsc C s | � � S r1 )�_get_cached_credentialsrr r@ r@ rC �fetch_credentials� s z)CachedCredentialFetcher.fetch_credentialsc C sl | � � }|du r$| �� }| �|� n
t�d� |d }t|d dd�}|d |d |d ||�d
�d �}|S ) z�Get up-to-date credentials.
This will check the cache for up-to-date credentials, calling assume
role if none are available.
Nz*Credentials for role retrieved from cache.r� r� T)r� r� r� r� � AccountId�r- r. r/ r� r0 )�_load_from_cacher �_write_to_cacher\ r] r� rP )rh r� �creds�
expirationr� r@ r@ rC r � s
�z/CachedCredentialFetcher._get_cached_credentialsc C s8 | j | jv r4t| j| j �}| �|�s*|S t�d� d S )Nz6Credentials were found in cache, but they are expired.)r� rd r r� r\ r] )rh r r@ r@ rC r s
�z(CachedCredentialFetcher._load_from_cachec C s t |�| j| j< d S r1 )r rd r� )rh r� r@ r@ rC r s z'CachedCredentialFetcher._write_to_cachec C s( t |d d �}t|t� �}|| jk S )z!Check if credentials are expired.r� r� )r� r r� r� )rh r� Zend_time�secondsr@ r@ rC r� s z#CachedCredentialFetcher._is_expired)NN)
r� r� r� r� ri r� r r r r r r r� r@ r@ r@ rC r� � s
r� c s6 e Zd Zd
� fdd� Zdd� Zdd� Zdd � Z� ZS ) � BaseAssumeRoleCredentialFetcherNc sf || _ || _|d u r i | _n
t|�| _| j| jd< | j�d�| _d| _| jsT| �� t� � ||� d S )NZRoleArn�RoleSessionNameF)
�_client_creatorZ _role_arn�_assume_kwargsr rP �_role_session_name�_using_default_session_name�_generate_assume_role_namer� ri )rh rG �role_arn�
extra_argsr= r� r� r@ rC ri s
z(BaseAssumeRoleCredentialFetcher.__init__c C s* dt t�� �� �| _| j| jd< d| _d S )Nzbotocore-session-r T)�int�timer r r rr r@ r@ rC r 1 s z:BaseAssumeRoleCredentialFetcher._generate_assume_role_namec C sZ t | j�}| jr|d= d|v r0t�|d �|d< tj|dd�}t|�d���� }| � |�S )��Create a predictable cache key for the current configuration.
The cache key is intended to be compatible with file names.
r ZPolicyT)� sort_keys�utf-8)
r r r �json�loads�dumpsr �encode� hexdigestr �rh �argsZ
argument_hashr@ r@ rC r� 6 s
z1BaseAssumeRoleCredentialFetcher._create_cache_keyc C sR |� di �� d�}t�|�r>t� }|�|�d }||d d< nt�d|� �� d S )NZAssumedRoleUserZArnZaccountr� r z'Unable to extract account ID from Arn: )rP r Zis_arnZ parse_arnr\ r] )rh r� r Z
arn_parserr0 r@ r@ rC �_add_account_id_to_responseL s
z;BaseAssumeRoleCredentialFetcher._add_account_id_to_response)NNN)r� r� r� ri r r� r r� r@ r@ r� rC r s �r c s6 e Zd Zd
� fdd� Zdd� Zdd� Zdd � Z� ZS ) �AssumeRoleCredentialFetcherNc s8 || _ || _| jdu r tj| _t� j|||||d� dS )a�
:type client_creator: callable
:param client_creator: A callable that creates a client taking
arguments like ``Session.create_client``.
:type source_credentials: Credentials
:param source_credentials: The credentials to use to create the
client for the call to AssumeRole.
:type role_arn: str
:param role_arn: The ARN of the role to be assumed.
:type extra_args: dict
:param extra_args: Any additional arguments to add to the assume
role request using the format of the botocore operation.
Possible keys include, but may not be limited to,
DurationSeconds, Policy, SerialNumber, ExternalId and
RoleSessionName.
:type mfa_prompter: callable
:param mfa_prompter: A callable that returns input provided by the
user (i.e raw_input, getpass.getpass, etc.).
:type cache: dict
:param cache: An object that supports ``__getitem__``,
``__setitem__``, and ``__contains__``. An example of this is
the ``JSONFileCache`` class in aws-cli.
:type expiry_window_seconds: int
:param expiry_window_seconds: The amount of time, in seconds,
N�r r= r� )�_source_credentials�
_mfa_prompter�getpassr� ri )rh rG �source_credentialsr r � mfa_prompterr= r� r� r@ rC ri W s )
�z$AssumeRoleCredentialFetcher.__init__c C s8 t | j� | �� }| �� }|jf i |��}| �|� |S )�'Get credentials by calling assume role.)r r� �_assume_role_kwargs�_create_clientr� r )rh r� r� r� r@ r@ rC r � s
z,AssumeRoleCredentialFetcher._get_credentialsc C sX t | j�}|�d�}|dur:d|� d�}| �|�}||d< |�d�}|durT||d< |S )�AGet the arguments for assume role based on current configuration.� SerialNumberNzEnter MFA code for z: Z TokenCode�DurationSeconds)r r rP r$ )rh �assume_role_kwargs�
mfa_serial�promptZ
token_code�duration_secondsr@ r@ rC r) � s
z/AssumeRoleCredentialFetcher._assume_role_kwargsc C s" | j �� }| jd|j|j|jd�S )z2Create an STS client using the source credentials.�sts)�aws_access_key_id�aws_secret_access_key�aws_session_token)r# r� r
r- r. r/ )rh Zfrozen_credentialsr@ r@ rC r* � s
�z*AssumeRoleCredentialFetcher._create_client)NNNN)r� r� r� ri r r) r* r� r@ r@ r� rC r! V s �6 r! c s. e Zd Zd� fdd� Zdd� Zdd� Z� ZS ) �*AssumeRoleWithWebIdentityCredentialFetcherNc s || _ t� j|||||d� dS )aG
:type client_creator: callable
:param client_creator: A callable that creates a client taking
arguments like ``Session.create_client``.
:type web_identity_token_loader: callable
:param web_identity_token_loader: A callable that takes no arguments
and returns a web identity token str.
:type role_arn: str
:param role_arn: The ARN of the role to be assumed.
:type extra_args: dict
:param extra_args: Any additional arguments to add to the assume
role request using the format of the botocore operation.
Possible keys include, but may not be limited to,
DurationSeconds, Policy, SerialNumber, ExternalId and
RoleSessionName.
:type cache: dict
:param cache: An object that supports ``__getitem__``,
``__setitem__``, and ``__contains__``. An example of this is
the ``JSONFileCache`` class in aws-cli.
:type expiry_window_seconds: int
:param expiry_window_seconds: The amount of time, in seconds,
r" N)�_web_identity_token_loaderr� ri )rh rG �web_identity_token_loaderr r r= r� r� r@ rC ri � s $�z3AssumeRoleWithWebIdentityCredentialFetcher.__init__c C sH t | j� | �� }ttd�}| jd|d�}|jf i |��}| �|� |S )r( ��signature_versionr2 �r; )r r� r) r r r
Z assume_role_with_web_identityr )rh r� r; r� r� r@ r@ rC r � s
z;AssumeRoleWithWebIdentityCredentialFetcher._get_credentialsc C s t | j�}| �� }||d< |S )r+ ZWebIdentityToken)r r r7 )rh r. Zidentity_tokenr@ r@ rC r) � s
z>AssumeRoleWithWebIdentityCredentialFetcher._assume_role_kwargs)NNN)r� r� r� ri r r) r� r@ r@ r� rC r6 � s �. r6 c @ s. e Zd ZdZdZddd�Zdd� Zdd� ZdS ) �CredentialProviderNc C s
|| _ d S r1 rA )rh rB r@ r@ rC ri s zCredentialProvider.__init__c C s dS )a}
Loads the credentials from their source & sets them on the object.
Subclasses should implement this method (by reading from disk, the
environment, the network or wherever), returning ``True`` if they were
found & loaded.
If not found, this method should return ``False``, indicating that the
``CredentialResolver`` should fall back to the next available method.
The default implementation does nothing, assuming the user has set the
``access_key/secret_key/token`` themselves.
:returns: Whether credentials were found & set
:rtype: Credentials
Tr@ rr r@ r@ rC �load s zCredentialProvider.loadc G sF g }|D ]8}z|� || � W q ty> t| j|d��Y q0 q|S )N�r� Zcred_var)�append�KeyErrorr �METHOD)rh �mappingZ key_names�foundZkey_namer@ r@ rC �_extract_creds_from_mapping s �z.CredentialProvider._extract_creds_from_mapping)N)r� r� r� rA �CANONICAL_NAMEri r= rD r@ r@ r@ rC r< � s
r< c @ sN e Zd ZdZejfdd�Zdd� Zdd� Ze dd � �Z
e d
d � �Z d d
� Z dS )rs zcustom-processc C s || _ || _d | _|| _d S r1 )�
_profile_name� _load_config�_loaded_config�_popen)rh rH rF �popenr@ r@ rC ri * s zProcessProvider.__init__c s| �j � � d u rd S td� ��� �}td� |�d�d urTt�|� �fdd��j�S t|d |d |�d��j|�d �d
�S ) NZCREDENTIALS_PROFILE_PROCESSZCREDENTIALS_PROCESSr� c s
�� � �S r1 )�_retrieve_credentials_usingr@ ��credential_processrh r@ rC rD ; rE z&ProcessProvider.load.<locals>.<lambda>r- r. r/ r0 )r- r. r/ r� r0 )�_credential_processr rK rP r� r� rA r� )rh Z
creds_dictr@ rL rC r= 0 s&
��zProcessProvider.loadc
C s t |�}t� �0 | j|tjtjd�}|�� \}}W d � n1 sD0 Y |jdkrlt| j|� d�d��t
j j �
|� d��}|�dd�}|dkr�t| jd|� d �d��z,|d
|d |�d �|�d
�| �|�d�W S t�y } z t| jd|� �d��W Y d }~n
d }~0 0 d S )N)�stdout�stderrr r r� ZVersionz<Version key not provided>� zUnsupported version 'z8' for credential process provider, supported versions: 1r� r� r� r� r z"Missing required key in response: )r r% rI �
subprocess�PIPEZ communicate�
returncoder rA �decoder� r� r r rP �_get_account_idr@ ) rh rM Z process_list�prO rP �parsed�version�er@ r@ rC rK G s:
�*
�
���z+ProcessProvider._retrieve_credentials_usingc C s | j �d�S )NrM )�profile_configrP rr r@ r@ rC rN l s z#ProcessProvider._credential_processc C s0 | j d u r| �� | _ | j �di ��| ji �}|S �N�profiles)rH rG rP rF )rh r[ r@ r@ rC r[ p s
�z ProcessProvider.profile_configc C s |� d�}|p| j� d�S )Nr �aws_account_id)rP r[ )rh rX r0 r@ r@ rC rV y s
z ProcessProvider._get_account_idN)
r� r� r� rA rR �Popenri r= rK r� rN r[ rV r@ r@ r@ rC rs ' s %
rs c @ s$ e Zd ZdZdZdd� Zdd� ZdS )rT ziam-role�Ec2InstanceMetadatac C s
|| _ d S r1 )�
_role_fetcher)rh r< r@ r@ rC ri � s z!InstanceMetadataProvider.__init__c C sF | j }|�� }|sd S td� t�d|d � tj|| j|jd�}|S )N�CREDENTIALS_IMDSz#Found credentials from IAM Role: %s� role_name�r� r� )ra Z retrieve_iam_role_credentialsr r\ r] r� r� rA )rh �fetcherr� r r@ r@ rC r= � s ��z InstanceMetadataProvider.loadN)r� r� r� rA rE ri r= r@ r@ r@ rC rT ~ s rT c @ sN e Zd ZdZdZdZdZddgZdZdZ dd
d �Z
d d
� Z dd� Z dd� Z
d S )rR �env� EnvironmentZAWS_ACCESS_KEY_IDZAWS_SECRET_ACCESS_KEYZAWS_SECURITY_TOKENZAWS_SESSION_TOKENZAWS_CREDENTIAL_EXPIRATIONZAWS_ACCOUNT_IDNc C s$ |du rt j}|| _| �|�| _dS )a�
:param environ: The environment variables (defaults to
``os.environ`` if no value is provided).
:param mapping: An optional mapping of variable names to
environment variable names. Use this if you want to
change the mapping of access_key->AWS_ACCESS_KEY_ID, etc.
The dict can have up to 5 keys:
* ``access_key``
* ``secret_key``
* ``token``
* ``expiry_time``
* ``account_id``
N)r� �environ�_build_mapping�_mapping)rh rh rB r@ r@ rC ri � s zEnvProvider.__init__c C s� i }|d u r@| j |d< | j|d< | j|d< | j|d< | j|d< nv|�d| j �|d< |�d| j�|d< |�d| j�|d< t|d t�s�|d g|d< |�d| j�|d< |�d| j�|d< |S )Nr- r. r/ r� r0 )�
ACCESS_KEY�
SECRET_KEY�TOKENS� EXPIRY_TIME�
ACCOUNT_IDrP r� �list)rh rB Z var_mappingr@ r@ rC ri � s.
����zEnvProvider._build_mappingc C s� | j �| jd d�}|r�t�d� | �� }|dd�}td� |d }|dur|t|�}t|d |d |d
||| j |d d �S t
|d |d |d
| j |d d
�S dS dS )zK
Search for credentials in explicit environment variables.
r- � z+Found credentials in environment variables.F)�require_expiry�CREDENTIALS_ENV_VARSr� Nr. r/ r0 )r� r� r0 �r� r0 ) rh rP rj r\ �info�_create_credentials_fetcherr r) r� rA r� )rh r- re r� r� r@ r@ rC r= � s4
�
�zEnvProvider.loadc s( | j �| j�| j� d� ��fdd� }|S )NTc s� i }� � �d d�}|s(t��d d��||d< � � �d d�}|sTt��d d��||d< d |d< �d D ] }� � |d�}|rl||d< q�qld |d< � � �d d�}|r�||d< | r�|s�t��d d��d |d< � � �d d�}|r�||d< |S )Nr- rq r> r. r/ r� r0 )rP r )rr r� r- r. Z
token_env_varr/ r� r0 �rh rB r� r@ rC r s@ �� �zBEnvProvider._create_credentials_fetcher.<locals>.fetch_credentials)T)rj rA rh )rh r r@ rw rC rv � s
(z'EnvProvider._create_credentials_fetcher)NN)r� r� r� rA rE rk rl rm rn ro ri ri r= rv r@ r@ r@ rC rR � s
$rR c @ s2 e Zd ZdZdZdZdZdZd dd�Zd d
� Z dS ) rY zec2-credentials-fileZ Ec2Config�AWS_CREDENTIAL_FILEZAWSAccessKeyIdZ AWSSecretKeyNc C s* |d u rt j}|d u rt}|| _|| _d S r1 )r� rh r& �_environ�_parser)rh rh �parserr@ r@ rC ri 3 s z OriginalEC2Provider.__init__c C sh d| j v r`tj�| j d �}| �|�}| j|v rdt�d� || j }|| j }t ||| j
d�S ndS dS )zN
Search for a credential file used by original EC2 CLI tools.
rx z)Found credentials in AWS_CREDENTIAL_FILE.�r� N) ry r� r� �
expanduserrz rk r\ ru rl r� rA )rh � full_pathr r- r. r@ r@ rC r= ; s
�
zOriginalEC2Provider.load)NN)
r� r� r� rA rE Z
CRED_FILE_ENVrk rl ri r= r@ r@ r@ rC rY + s
rY c @ sJ e Zd ZdZdZdZdZddgZdZdd d
�Z d d � Z
d
d� Z dd� Z dS )rv zshared-credentials-fileZSharedCredentialsr3 r4 �aws_security_tokenr5 r^ Nc C s2 || _ |d u rd}|| _|d u r(tjj}|| _d S )Nr5 )�_creds_filenamerF r� � configloader�raw_config_parse� _ini_parser)rh ru rH �
ini_parserr@ r@ rC ri Z s z!SharedCredentialProvider.__init__c C s� z| � | j�}W n ty$ Y d S 0 | j|v r�|| j }| j|v r�t�d| j� | �|| j| j�\}}| � |�}| �
|�}t d� t |||| j
|d�S d S )Nz0Found credentials in shared credentials file: %s�CREDENTIALS_PROFILErt )r� r� r rF rk r\ ru rD rl �_get_session_tokenrV r r� rA )rh Zavailable_credsr; r- r. r/ r0 r@ r@ rC r= c s0
�
�
�z SharedCredentialProvider.loadc C s$ | j D ]}||v r|| S qd S r1 �rm )rh r; Z token_envvarr@ r@ rC r� } s
z+SharedCredentialProvider._get_session_tokenc C s |� | j�S r1 �rP ro �rh r; r@ r@ rC rV � s z(SharedCredentialProvider._get_account_id)NN)
r� r� r� rA rE rk rl rm ro ri r= r� rV r@ r@ r@ rC rv N s
rv c @ sN e Zd ZdZdZdZdZdZddgZdZ dd
d �Z
d d
� Z dd� Z dd� Z
d S )ry z0INI based config provider with profile sections.z config-fileZ SharedConfigr3 r4 r r5 r^ Nc C s& || _ || _|du r tjj}|| _dS )a
:param config_filename: The session configuration scoped to the current
profile. This is available via ``session.config``.
:param profile_name: The name of the current profile.
:param config_parser: A config parser callable.
N)�_config_filenamerF r� r� rF �_config_parser)rh rx rH Z
config_parserr@ r@ rC ri � s
zConfigProvider.__init__c C s� z| � | j�}W n ty$ Y dS 0 | j|d v r�|d | j }| j|v r�t�d| j� | �|| j| j�\}}| � |�}| �
|�}t d� t |||| j
|d�S ndS dS )zr
If there is are credentials in the configuration associated with
the session, use those.
Nr] z$Credentials found in config file: %sr� rt )r� r� r rF rk r\ ru rD rl r� rV r r� rA )rh r? r[ r- r. r/ r0 r@ r@ rC r= � s2
�
�
�zConfigProvider.loadc C s$ | j D ]}||v r|| S qd S r1 r� )rh r[ Z
token_namer@ r@ rC r� � s
z!ConfigProvider._get_session_tokenc C s |� | j�S r1 r� r� r@ r@ rC rV � s z ConfigProvider._get_account_id)N)r� r� r� r� rA rE rk rl rm ro ri r= r� rV r@ r@ r@ rC ry � s
ry c @ s: e Zd ZdZdZdZddgZdZdZd
d d
�Z d d � Z
dS )rZ z boto-configZ Boto2ConfigZ BOTO_CONFIGz
/etc/boto.cfgz~/.botor3 r4 Nc C s. |d u rt j}|d u r tjj}|| _|| _d S r1 )r� rh r� r� r� ry r� )rh rh r� r@ r@ rC ri � s zBotoProvider.__init__c C s� | j | jv r | j| j g}n| j}|D ]�}z| �|�}W n tyN Y q&Y n0 d|v r&|d }| j|v r&t�d|� | �|| j| j �\}}t
d� t ||| j d� S q&dS )z;
Look for credentials in boto config file.
r� z)Found credentials in boto config file: %sZ CREDENTIALS_BOTO2_CONFIG_FILEr| N)
�BOTO_CONFIG_ENVry �DEFAULT_CONFIG_FILENAMESr� r rk r\ ru rD rl r r� rA )rh Zpotential_locationsr� r; r� r- r. r@ r@ rC r= � s*
�
��zBotoProvider.load)NN) r� r� r� rA rE r� r� rk rl ri r= r@ r@ r@ rC rZ � s
rZ c @ s� e Zd ZdZdZdZdZdZdddd �Ze j ddfd
d �Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� ZdS )$rV � assume-roleNr �web_identity_token_filer3 rb rs �CREDENTIALS_HTTP)r` rg � EcsContainerc C sF || _ || _|| _|| _|| _i | _|| _|| _| jg| _t � | _
dS )a�
:type load_config: callable
:param load_config: A function that accepts no arguments, and
when called, will return the full configuration dictionary
for the session (``session.full_config``).
:type client_creator: callable
:param client_creator: A factory function that will create
a client when called. Has the same interface as
``botocore.session.Session.create_client``.
:type cache: dict
:param cache: An object that supports ``__getitem__``,
``__setitem__``, and ``__contains__``. An example
of this is the ``JSONFileCache`` class in the CLI.
:type profile_name: str
:param profile_name: The name of the profile.
:type prompter: callable
:param prompter: A callable that returns input provided
by the user (i.e raw_input, getpass.getpass, etc.).
:type credential_sourcer: CanonicalNameCredentialSourcer
:param credential_sourcer: A credential provider that takes a
configuration, which is used to provide the source credentials
for the STS call.
N) r= rG r
rF � _prompterrH �_credential_sourcer�_profile_provider_builder�_visited_profilesr� � _feature_ids)rh rF rG r= rH ZprompterrI rJ r@ r@ rC ri s +
zAssumeRoleProvider.__init__c C s@ | � � | _| j�di �}|�| ji �}| �|�r<| �| j�S d S r\ )rG rH rP rF � _has_assume_role_config_vars�_load_creds_via_assume_role)rh r] r4 r@ r@ rC r= M s
zAssumeRoleProvider.loadc C s | j |v o| j|vS r1 )�ROLE_CONFIG_VAR�WEB_IDENTITY_TOKE_FILE_VAR�rh r4 r@ r@ rC r� T s
�z/AssumeRoleProvider._has_assume_role_config_varsc C s� | � |�}| �||�}i }|�d�}|d ur4||d< |�d�}|d urN||d< |�d�}|d urh||d< |�d�}|d ur�||d< t| j||d || j| jd
�} | j�� | _ | j
}
|d ur�t |
�}
| j� d � t
| j� t| j|
td �S )
N�role_session_namer � external_idZ
ExternalIdr/ r, r1 r- r )rG r& r r r' r= ZCREDENTIALS_STS_ASSUME_ROLE)r� r� r� )�_get_role_config�_resolve_source_credentialsrP r! r
r� r= r� �copyr� r r� �addr r� rA r� ) rh rH � role_configr& r r� r� r/ r1 re Z refresherr@ r@ rC r� _ sH
�
�
�z.AssumeRoleProvider._load_creds_via_assume_rolec C s | j �di �}|| }|�d�}|d }|�d�}|�d�}|�d�}|�d�} |�d�}
|||| ||d �} |
d
ur�zt|
�| d< W n ty� Y n0 |d
ur�|d
ur�td |� d �d
��nB|d
u r�|d
u r�t| jdd��n"|d
ur�| �||� n | �||� | S )z?Retrieves and validates the role configuration for the profile.r] �source_profiler �credential_sourcer/ r� r� r1 )r r� r/ r� r� r� N�
The profile "z5" contains both source_profile and credential_source.�r� z#source_profile or credential_sourcer> ) rH rP r �
ValueErrorr r rA �_validate_credential_source�_validate_source_profile) rh rH r] r4 r� r r� r/ r� r� r1 r� r@ r@ rC r� � sD
�
�� z#AssumeRoleProvider._get_role_configc C sJ | j d u r"td|� d|� d�d��| j �|�sFtd|� d|� d�d��d S )NzThe credential_source "z" is specified in profile "z)", but no source provider was configured.r� zThe credential source "z" referenced in profile "z" is not valid.)r� r � is_supported)rh Zparent_profiler� r@ r@ rC r� � s
� �z.AssumeRoleProvider._validate_credential_sourcec C s t | �|�| �|�g�S r1 )�any�_has_static_credentialsr� r� r@ r@ rC � _source_profile_has_credentials� s
��z2AssumeRoleProvider._source_profile_has_credentialsc C sv | j �di �}||vr.td|� d|� d�d��|| }|| jvrDd S ||krZt|| jd��| �|�srt|| jd��d S )Nr] zThe source_profile "z " referenced in the profile "z" does not exist.r� )r� Zvisited_profiles)rH rP r r� r
r� )rh Zparent_profile_nameZsource_profile_namer] r� r@ r@ rC r� � s( ��
�
�z+AssumeRoleProvider._validate_source_profilec s ddg}t � fdd�|D ��S )Nr4 r3 c 3 s | ]}|� v V qd S r1 r@ )r� Z
static_key�r4 r@ rC � <genexpr>� rE z=AssumeRoleProvider._has_static_credentials.<locals>.<genexpr>)r� )rh r4 Z static_keysr@ r� rC r� � s z*AssumeRoleProvider._has_static_credentialsc C sT |� d�}|d ur*| j�d� | �||�S |d }| j�|� | j�d� | �|�S )Nr� Z"CREDENTIALS_PROFILE_NAMED_PROVIDERr� Z"CREDENTIALS_PROFILE_SOURCE_PROFILE)rP r� r� � _resolve_credentials_from_sourcer� r? �!_resolve_credentials_from_profile)rh r� rH r� r� r@ r@ rC r� s
� z.AssumeRoleProvider._resolve_source_credentialsc C s� | j �di �}|| }| j�d� | �|�r<| js<| �|�S | �|�sP| �|�s�| jj|dd�}t |�}|�
� }|d u r�d}t || d��|S | � |�S )Nr] r� TrK z.The source profile "%s" must have credentials.r� )
rH rP r� r� r� r� �(_resolve_static_credentials_from_profiler� rN r^ r� r r� )rh rH r] r4 r` Z
profile_chainr� Z
error_messager@ r@ rC r� s6 ��
�����z4AssumeRoleProvider._resolve_credentials_from_profilec
C sZ z t |d |d |�d�d�W S tyT } z t| jt|�d��W Y d }~n
d }~0 0 d S )Nr3 r4 r5 )r- r. r/ r> )r� rP r@ r rA �str)rh r4 rZ r@ r@ rC r� / s �
�z;AssumeRoleProvider._resolve_static_credentials_from_profilec C sF | j �|�}|d u r&t|d|� �d��| j�|�}|rB| j�|� |S )Nz@No credentials found in credential_source referenced in profile r� )r� r& r �NAMED_PROVIDER_FEATURE_MAPrP r� r� )rh r� rH r� Znamed_provider_feature_idr@ r@ rC r� ; s ���� z3AssumeRoleProvider._resolve_credentials_from_source)r� r� r� rA rE r� r� ZEXPIRY_WINDOW_SECONDSr� r% ri r= r� r� r� r� r� r� r� r� r� r� r� r@ r@ r@ rC rV � s2 � �
? /1&
! rV c @ sR e Zd ZdZdZdddd�Zddd �Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dS )rz z assume-role-with-web-identityNZAWS_WEB_IDENTITY_TOKEN_FILEZAWS_ROLE_SESSION_NAMEZ AWS_ROLE_ARN)r� r� r Fc C sB || _ || _|| _|| _d | _|| _|d u r0t}|| _t� | _ d S r1 )
r= rG r
rF �_profile_config�_disable_env_varsr �_token_loader_clsr� r� )rh rF rG rH r= rL Ztoken_loader_clsr@ r@ rC ri Z s z*AssumeRoleWithWebIdentityProvider.__init__c C s | � � S r1 )� _assume_role_with_web_identityrr r@ r@ rC r= n s z&AssumeRoleWithWebIdentityProvider.loadc C s: | j d u r.| �� }|�di �}|�| ji �| _ | j �|�S r\ )r� rG rP rF )rh �key�
loaded_configr] r@ r@ rC �_get_profile_configq s
z5AssumeRoleWithWebIdentityProvider._get_profile_configc C s2 | j r
d S | j�|�}|r.|tjv r.tj| S d S r1 )r� �_CONFIG_TO_ENV_VARrP r� rh )rh r� Zenv_keyr@ r@ rC �_get_env_configx s
z1AssumeRoleWithWebIdentityProvider._get_env_configc C sH | � |�}|d ur"| j�d� |S | �|�}|d urD| j�d� |S d S )NZ%CREDENTIALS_ENV_VARS_STS_WEB_ID_TOKENZ$CREDENTIALS_PROFILE_STS_WEB_ID_TOKEN)r� r� r� r� )rh r� Z env_valueZ config_valuer@ r@ rC � _get_config� s
z-AssumeRoleWithWebIdentityProvider._get_configc C s� | � d�}|sd S | �|�}| � d�}|s8d}t|d��i }| � d�}|d urV||d< t| j|||| jd�}| j�� |_| j� d� t
| j� t | j |j
d �S )
Nr� r z�The provided profile or the current environment is configured to assume role with web identity but has no role ARN configured. Ensure that the profile has the role_arnconfiguration set or the AWS_ROLE_ARN env var is set.r� r� r )rG r8 r r r= Z"CREDENTIALS_STS_ASSUME_ROLE_WEB_IDrd )r� r� r r6 r
r= r� r� r� r� r r� rA r )rh Z
token_path� token_loaderr r� r r� re r@ r@ rC r� � s6
�
�
�z@AssumeRoleWithWebIdentityProvider._assume_role_with_web_identity)NFN) r� r� r� rA rE r� ri r= r� r� r� r� r@ r@ r@ rC rz Q s � �
rz c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )rX c C s
|| _ d S r1 ��
_providers�rh rN r@ r@ rC ri � s z'CanonicalNameCredentialSourcer.__init__c C s |dd� | j D �v S )aL Validates a given source name.
:type source_name: str
:param source_name: The value of credential_source in the config
file. This is the canonical name of the credential provider.
:rtype: bool
:returns: True if the credential provider is supported,
False otherwise.
c S s g | ]
}|j �qS r@ )rE �r� rW r@ r@ rC r� � rE z?CanonicalNameCredentialSourcer.is_supported.<locals>.<listcomp>r� )rh � source_namer@ r@ rC r� � s z+CanonicalNameCredentialSourcer.is_supportedc C s$ | � |�}t|t�r |�� S |�� S )a Loads source credentials based on the provided configuration.
:type source_name: str
:param source_name: The value of credential_source in the config
file. This is the canonical name of the credential provider.
:rtype: Credentials
)�
_get_providerr� r^ r� r= )rh r� �sourcer@ r@ rC r& � s
z1CanonicalNameCredentialSourcer.source_credentialsc C sV | � |�}|�� dv r@| �d�}|dur@|du r4|S t||g�S |du rRt|d��|S )a# Return a credential provider by its canonical name.
:type canonical_name: str
:param canonical_name: The canonical name of the provider.
:raises UnknownCredentialError: Raised if no
credential provider by the provided name
is found.
)Z sharedconfigZsharedcredentialsr� N��name)� _get_provider_by_canonical_name�lower�_get_provider_by_methodr^ r )rh �canonical_namer� r_ r@ r@ rC r� � s
z,CanonicalNameCredentialSourcer._get_providerc C s2 | j D ]&}|j}|r|�� |�� kr| S qdS )z�Return a credential provider by its canonical name.
This function is strict, it does not attempt to address
compatibility issues.
N)r� rE r� )rh r� r� r� r@ r@ rC r� � s
z>CanonicalNameCredentialSourcer._get_provider_by_canonical_namec C s" | j D ]}|j|kr| S qdS )z0Return a credential provider by its METHOD name.N)r� rA )rh r� r� r@ r@ rC r� s
z6CanonicalNameCredentialSourcer._get_provider_by_methodN) r� r� r� ri r� r& r� r� r� r@ r@ r@ rC rX � s
& rX c @ s^ e Zd ZdZdZdZdZdZdZddd �Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� ZdS )rS zcontainer-roler� Z&AWS_CONTAINER_CREDENTIALS_RELATIVE_URIZ"AWS_CONTAINER_CREDENTIALS_FULL_URIZ!AWS_CONTAINER_AUTHORIZATION_TOKENZ&AWS_CONTAINER_AUTHORIZATION_TOKEN_FILENc C s, |d u rt j}|d u r t� }|| _|| _d S r1 )r� rh r ry �_fetcher)rh rh re r@ r@ rC ri s zContainerProvider.__init__c C s$ | j | jv s| j| jv r | �� S d S r1 )�ENV_VARry � ENV_VAR_FULL�_retrieve_or_failrr r@ r@ rC r= s zContainerProvider.loadc
C sv | � � r | j�| j| j �}n | j| j }| �� }| �||�}|� }t|d |d |d | j t
|d �||� d�d�S )Nr- r. r/ r� r0 )r- r. r/ r� r� r� r0 ) �_provided_relative_urir� Zfull_urlry r� r� �_build_headers�_create_fetcherr� rA r� rP )rh �full_uri�headersre r r@ r@ rC r� $ s
�z#ContainerProvider._retrieve_or_failc C s� d }| j | jv rN| j| j }t|��}|�� }W d � qf1 sB0 Y n| j| jv rf| j| j }|d ur�| �|� d|iS d S )NZ
Authorization)�ENV_VAR_AUTH_TOKEN_FILEry �open�read�ENV_VAR_AUTH_TOKEN�_validate_auth_token)rh �
auth_tokenZauth_token_file_pathZ
token_filer@ r@ rC r� 6 s
(
z ContainerProvider._build_headersc C s d|v sd|v rt d��d S )N�
�
z,Auth token value is not a legal header value)r� )rh r� r@ r@ rC r� B s z&ContainerProvider._validate_auth_tokenc s � ��fdd�}|S )Nc
s� z �j j� �d�} td� W nF tyb } z.tjd|dd� t�jt|�d��W Y d }~n
d }~0 0 | d | d | d | d
| � d �d �S )
N)r� r� z'Error retrieving container metadata: %sTr� r� r� r� �Tokenr� r r )
r� Zretrieve_full_urir r r\ r] r rA r� rP )r� rZ �r� r� rh r@ rC � fetch_credsG s$ � �
� �z6ContainerProvider._create_fetcher.<locals>.fetch_credsr@ )rh r� r� r� r@ r� rC r� F s z!ContainerProvider._create_fetcherc C s | j | jv S r1 )r� ry rr r@ r@ rC r� ^ s z(ContainerProvider._provided_relative_uri)NN)r� r� r� rA rE r� r� r� r� ri r= r� r� r� r� r� r@ r@ r@ rC rS s
rS c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )r^ c C s
|| _ dS )zQ
:param providers: A list of ``CredentialProvider`` instances.
NrM r� r@ r@ rC ri c s zCredentialResolver.__init__c C sJ zdd� | j D ��|�}W n ty6 t|d��Y n0 | j �||� dS )a=
Inserts a new instance of ``CredentialProvider`` into the chain that
will be tried before an existing one.
:param name: The short name of the credentials you'd like to insert the
new credentials before. (ex. ``env`` or ``config``). Existing names
& ordering can be discovered via ``self.available_methods``.
:type name: string
:param cred_instance: An instance of the new ``Credentials`` object
you'd like to add to the chain.
:type cred_instance: A subclass of ``Credentials``
c S s g | ]
}|j �qS r@ �rA r� r@ r@ rC r� z rE z4CredentialResolver.insert_before.<locals>.<listcomp>r� N)rN �indexr� r �insert�rh r� Zcredential_provider�offsetr@ r@ rC �
insert_beforek s
z CredentialResolver.insert_beforec C s | � |�}| j�|d |� dS )a9
Inserts a new type of ``Credentials`` instance into the chain that will
be tried after an existing one.
:param name: The short name of the credentials you'd like to insert the
new credentials after. (ex. ``env`` or ``config``). Existing names
& ordering can be discovered via ``self.available_methods``.
:type name: string
:param cred_instance: An instance of the new ``Credentials`` object
you'd like to add to the chain.
:type cred_instance: A subclass of ``Credentials``
rQ N)�_get_provider_offsetrN r� r� r@ r@ rC � insert_after s
z CredentialResolver.insert_afterc C s6 dd� | j D �}||vr dS |�|�}| j �|� dS )z�
Removes a given ``Credentials`` instance from the chain.
:param name: The short name of the credentials instance to remove.
:type name: string
c S s g | ]
}|j �qS r@ r� r� r@ r@ rC r� � rE z-CredentialResolver.remove.<locals>.<listcomp>N)rN r� �pop)rh r� Zavailable_methodsr� r@ r@ rC r[ � s
zCredentialResolver.removec C s | j | �|� S )z�Return a credential provider by name.
:type name: str
:param name: The name of the provider.
:raises UnknownCredentialError: Raised if no
credential provider by the provided name
is found.
)rN r� �rh r� r@ r@ rC � get_provider� s
z CredentialResolver.get_providerc C s: zdd� | j D ��|�W S ty4 t|d��Y n0 d S )Nc S s g | ]
}|j �qS r@ r� r� r@ r@ rC r� � rE z;CredentialResolver._get_provider_offset.<locals>.<listcomp>r� )rN r� r� r r� r@ r@ rC r� � s z'CredentialResolver._get_provider_offsetc C s6 | j D ]*}t�d|j� |�� }|dur| S qdS )zw
Goes through the credentials chain, returning the first ``Credentials``
that could be loaded.
z Looking for credentials via: %sN)rN r\ r] rA r= )rh r� r r@ r@ rC r� � s
z#CredentialResolver.load_credentialsN)
r� r� r� ri r� r� r[ r� r� r� r@ r@ r@ rC r^ b s r^ c sF e Zd ZdZdddddef� fdd� Zdd� Zdd� Zd d
� Z� Z S ) �SSOCredentialFetcher�%Y-%m-%dT%H:%M:%SZNc sH || _ || _|| _|| _|| _|| _| | _|
| _| | _t � �
||� d S r1 ) r
� _sso_region�
_role_namer� �
_start_url�
_token_loader�_token_provider�_sso_session_namer� r� ri ) rh � start_url�
sso_regionrc r0 rG r� r= r� r| �sso_session_namer� r� r@ rC ri � s z SSOCredentialFetcher.__init__c C sV | j | jd�}| jr | j|d< n
| j|d< tj|ddd�}t|�d���� }| � |�S )r )�roleName� accountIdZ sessionNameZstartUrlT)�,r� )r Z
separatorsr )
r� r� r� r� r r r r r r r r@ r@ rC r� � s �
z&SSOCredentialFetcher._create_cache_keyc C s$ |d }t j �|t� �}|�| j�S )Ng @�@)r� Z
fromtimestampr+ r� �_UTC_DATE_FORMAT)rh Z timestamp_msZtimestamp_secondsZ timestampr@ r@ rC �_parse_timestamp� s z%SSOCredentialFetcher._parse_timestampc C s t t| jd�}| jd|d�}| jr8| j�� }|�� j}nB| �| j �}|d }t
j � |d �}t
|| �� �}|dkrzt� �| j| j|d�}z t| j� |jf i |��} W n |jjy� t� �Y n0 | d }
d|
d |
d
|
d | �|
d �| jd
�d�}
|
S )z4Get credentials by calling SSO get role credentials.)r: r> �ssor; � accessToken� expiresAtr )r� r� r� ZroleCredentials� accessKeyId�secretAccessKey� sessionTokenr )r� r� r� r� r )Z ProviderTyper� )r r r� r
r� �
load_tokenZget_frozen_tokenr/ r� r� �dateutilr{ r) r r� r r� r� r r� Zget_role_credentials�
exceptionsZUnauthorizedExceptionr� ) rh r; r� Zinitial_token_datar/ Z
token_dictr Z remainingr� r� r� r@ r@ rC r � sB �
�
��
z%SSOCredentialFetcher._get_credentials)
r� r� r� r� r� ri r� r� r r� r@ r@ r� rC r� � s �r� c @ s\ e Zd ZdZej�ej�dddd��ZdZ dZ
e e
Z ddd �Z d
d � Z
d d
� Zdd� ZdS )r} r� �~z.awsr= )�
sso_role_name�sso_account_id)�
sso_start_urlr� Nc C sN |d u rt | j�}|| _|| _|d u r*i }|| _|| _|| _|| _t� | _ d S r1 )
r �_SSO_TOKEN_CACHE_DIR� _token_cacher� r= rG r
rF r� r� )rh rF rG rH r= r{ r| r@ r@ rC ri ? s
zSSOProvider.__init__c s� | � � }|�di �}| j}|�| ji �� |�di �}t� fdd�| jD ��rPd S | �� |�\}}i }g }| j| } | D ]$}
|
|v r�||
||
< qv|�|
� qv|r�d�|�} t d|� d| � �d��|S ) Nr] � sso_sessionsc 3 s | ]}|� vV qd S r1 r@ )r� �c�r[ r@ rC r� \ s z/SSOProvider._load_sso_config.<locals>.<genexpr>r� r� zB" is configured to use SSO but is missing required configuration: r� )
rG rP rF �all� _PROFILE_REQUIRED_CONFIG_VARS� _resolve_sso_session_reference�_ALL_REQUIRED_CONFIG_VARSr? r� r ) rh r� r] rH r Zresolved_configZ
extra_reqsr; Zmissing_config_varsZall_required_configs�
config_varZmissingr@ r
rC �_load_sso_configT s6 ��
��z SSOProvider._load_sso_configc C s� |� d�}|d u r|dfS ||vr8d|� d�}t|d��|�� }|| }|�� D ]F\}}|� ||�|kr�d|� d|| � d|� d �}t|d��|||< qP|d
fS ) N� sso_sessionr@ z+The specified sso-session does not exist: "�"r� zThe value for z" is inconsistent between profile (z) and sso-session (z).)r )rP r r� �items) rh r[ r r� r� r; rB r �valr@ r@ rC r x s&
���
z*SSOProvider._resolve_sso_session_referencec C s� | � � }|sd S |d |d |d |d | jt| jd�| jd�}d|v }|rp|d |d< | j|d < | j�d
� n | j�d � tf i |��}| j� � |_
|r�| j�d � n | j�d
� t | j� t | j
|jd�S )Nr r� r r )r= )r� r� rc r0 rG r� r= r r� r| ZCREDENTIALS_PROFILE_SSOZ CREDENTIALS_PROFILE_SSO_LEGACYZCREDENTIALS_SSOZCREDENTIALS_SSO_LEGACYrd )r r
r" r
r= r� r� r� r� r� r� r r� rA r )rh Z
sso_configZfetcher_kwargsZsso_session_in_configZ sso_fetcherr@ r@ rC r= � s6
�
�zSSOProvider.load)NNN)r� r� r� rA r� r� r} r� r r Z_SSO_REQUIRED_CONFIG_VARSr ri r r r= r@ r@ r@ rC r} - s �� �
$r} c @ sJ e Zd ZdZdZedfdd�Zdd� Zdd � Ze d
d � �Z
e d d
� �Z dS )�LoginCredentialFetcherzm
Converts login access tokens from the cached token to
credentials, and supports refreshing them
i, r@ c C s" || _ || _|| _|| _|| _d S r1 )�
_session_namer� r
r� r� )rh � session_namer� rG r� r� r@ r@ rC ri � s
z LoginCredentialFetcher.__init__c s\ | j �| j�� � du r*td| j� d�d��d� vsHt� fdd�dD ��sRtd d��| �� �S )
z6Loads cached credentials without checking their expiryNz4Unable to load a existing login session for session z), Please reauthenticate with 'aws login'.r� r� c 3 s | ]}|� v V qd S r1 r@ )r� r� �r/ r@ rC r� � s �zALoginCredentialFetcher.load_cached_credentials.<locals>.<genexpr>)r� � refreshToken�dpopKey�clientIdzRFailed to load access token from token cache, missing one or more required fields.)r� r r r r �_token_to_credentialsrr r@ r rC �load_cached_credentials� s ����z.LoginCredentialFetcher.load_cached_credentialsc
C s� | j rt| j � | j�| j�}| �|�}|rtd|v rtd|d v rtt|d d �}t|| �� �}|| j krt| �
|�S t j j
t jd�}| jd|d�}|jj�dt|�� z |j|d |d d d
�d �}W np |jj�y6 } zR|j�d d
�} | dkr�t� �n"| dk�r t� �n| dk�r t� �t |�W Y d}~n
d}~0 0 |du �sLd|v�rVt dd��|�d�}
| �� � t � �t j!|
d d� } |�"|
d d |
d d |
d d |d d | �#d�d �|
d d �� | j�$| j|� | �
|�S ) z?Refreshes login credentials, including saving them to the cacher� r� r9 Zsigninr; z$before-call.signin.CreateOAuth2Tokenr r Z
refresh_token)r r Z grantType)Z
tokenInput�errorrq Z
TOKEN_EXPIREDZUSER_CREDENTIALS_CHANGEDZINSUFFICIENT_PERMISSIONSNZ tokenOutputz"Failed to refresh an access token.r� Z expiresIn)r
r� r r r� r� )r� r r r� r� )r� r )%r� r r� r r �_load_private_keyr� r r� �_REFRESH_THRESHOLDr r� r; r r r
�metaZevents�registerr# Zcreate_o_auth2_tokenr ZAccessDeniedExceptionr� rP r r r r r Z
astimezoner+ r� Z timedeltar� r� Z
save_token) rh r/ Z private_keyr� Zremaining_timer; r� r� rZ Z
error_type�outputZexpires_timestampr@ r@ rC �refresh_credentials� s�
��
�
�����
�
��
�
�� ��z*LoginCredentialFetcher.refresh_credentialsc C s8 | d d | d d | d d | d d | d d d�S )Nr� r� r r r� r� r r@ r r@ r@ rC r 2
s
�z,LoginCredentialFetcher._token_to_credentialsc
C s� d| vrt dd��| d �� }dd� |D �}d�|��� }zt�t�|��W S ty� } z t d|� �d��W Y d }~n
d }~0 0 d S )Nr z&Private key not found in cached token.r� c S s$ g | ] }|� d �s|� d�s|�qS )z
-----BEGINz-----END)�
startswith)r� �liner@ r@ rC r� E
s
�z<LoginCredentialFetcher._load_private_key.<locals>.<listcomp>rq z.Unable to load private key from cached token: ) r �
splitlinesr� �stripr Znew_key_from_der_data�base64Z b64decoder� )r/ �linesZ
content_lines�contentsrZ r@ r@ rC r" <
s � ��z(LoginCredentialFetcher._load_private_keyN) r� r� r� r� r# r� ri r r' r� r r" r@ r@ r@ rC r � s �
R
r c s( e Zd ZdZ� fdd�Zdd� Z� ZS )r Zloginc s> t � �� |d u r tt� �}|| _|| _|| _|| _d| _d S )N)ZCREDENTIALS_PROFILE_LOGINZCREDENTIALS_LOGIN) r� ri r r$ r
rF rG r
r� )rh r{ rF rH rG r� r@ rC ri Z
s
zLoginProvider.__init__c
C s� | � � }|�di �}|�| ji �}d|vr.d S t|d t| j�| jt| jd�}t | j� |�
� }t |d |d |d t |d �|d | j
|jtd �S )
Nr] Z
login_session)r r� rG r� r� r- r. r/ r� r0 )r- r. r/ r� r0 r� r� r� )rG rP rF r r! r
r
r� r� r r r� r� rA r' )rh r� r] r[ re Zcached_credentialsr@ r@ rC r= k
s0 �
�zLoginProvider.load)r� r� r� rA ri r= r� r@ r@ r� rC r W
s r )NN)F)hr, r� r% r Zloggingr� rR r� r � collectionsr r� r Zhashlibr Zbotocore.compatr� Zbotocore.configloaderZdateutil.parserr Z
awscrt.cryptor r r r Zbotocore.configr Zbotocore.exceptionsr
r r r
r r r r r r r r r r r Zbotocore.tokensr Zbotocore.useragentr r Zbotocore.utilsr r r r r r! r" r# r$ r% r& r' r) Z dateutil.tzr* r+ Z getLoggerr� r\ r, r� rQ rb rU r� r� r� r� rW r� r� r� r� r� r� r r! r6 r< rs rT rR rY rv ry rZ rV rz rX rS r^ r� r} r r r@ r@ r@ rC �<module> s� D 8
�
`\
0 :I>]
�F.W #8F. YeXTdg #