mirror of
https://github.com/fog/fog-aws.git
synced 2022-11-09 13:50:52 -05:00
87 lines
3.4 KiB
Ruby
87 lines
3.4 KiB
Ruby
module Fog
|
|
module AWS
|
|
module CredentialFetcher
|
|
|
|
INSTANCE_METADATA_HOST = "http://169.254.169.254"
|
|
INSTANCE_METADATA_PATH = "/latest/meta-data/iam/security-credentials/"
|
|
INSTANCE_METADATA_AZ = "/latest/meta-data/placement/availability-zone/"
|
|
|
|
CONTAINER_CREDENTIALS_HOST = "http://169.254.170.2"
|
|
|
|
module ServiceMethods
|
|
def fetch_credentials(options)
|
|
if options[:use_iam_profile] && Fog.mocking?
|
|
return Fog::AWS::Compute::Mock.data[:iam_role_based_creds]
|
|
end
|
|
if options[:use_iam_profile]
|
|
begin
|
|
role_data = nil
|
|
region = options[:region]
|
|
|
|
if ENV["AWS_CONTAINER_CREDENTIALS_RELATIVE_URI"]
|
|
connection = options[:connection] || Excon.new(CONTAINER_CREDENTIALS_HOST)
|
|
credential_path = options[:credential_path] || ENV["AWS_CONTAINER_CREDENTIALS_RELATIVE_URI"]
|
|
role_data = connection.get(:path => credential_path, :idempotent => true, :expects => 200).body
|
|
|
|
if region.nil?
|
|
connection = options[:metadata_connection] || Excon.new(INSTANCE_METADATA_HOST)
|
|
region = connection.get(:path => INSTANCE_METADATA_AZ, :idempotent => true, :expects => 200).body[0..-2]
|
|
end
|
|
else
|
|
connection = options[:connection] || Excon.new(INSTANCE_METADATA_HOST)
|
|
role_name = connection.get(:path => INSTANCE_METADATA_PATH, :idempotent => true, :expects => 200).body
|
|
role_data = connection.get(:path => INSTANCE_METADATA_PATH+role_name, :idempotent => true, :expects => 200).body
|
|
region ||= connection.get(:path => INSTANCE_METADATA_AZ, :idempotent => true, :expects => 200).body[0..-2]
|
|
end
|
|
|
|
session = Fog::JSON.decode(role_data)
|
|
credentials = {}
|
|
credentials[:aws_access_key_id] = session['AccessKeyId']
|
|
credentials[:aws_secret_access_key] = session['SecretAccessKey']
|
|
credentials[:aws_session_token] = session['Token']
|
|
credentials[:aws_credentials_expire_at] = Time.xmlschema session['Expiration']
|
|
|
|
# set region by default to the one the instance is in.
|
|
credentials[:region] = region
|
|
#these indicate the metadata service is unavailable or has no profile setup
|
|
credentials
|
|
rescue Excon::Error => e
|
|
Fog::Logger.warning("Unable to fetch credentials: #{e.message}")
|
|
super
|
|
end
|
|
else
|
|
super
|
|
end
|
|
end
|
|
end
|
|
|
|
module ConnectionMethods
|
|
def refresh_credentials_if_expired
|
|
refresh_credentials if credentials_expired?
|
|
end
|
|
|
|
private
|
|
|
|
def credentials_expired?
|
|
@use_iam_profile &&
|
|
(!@aws_credentials_expire_at ||
|
|
(@aws_credentials_expire_at && Fog::Time.now > @aws_credentials_expire_at - 15)) #new credentials become available from around 5 minutes before expiration time
|
|
end
|
|
|
|
def refresh_credentials
|
|
if @use_iam_profile
|
|
new_credentials = service.fetch_credentials :use_iam_profile => @use_iam_profile, :region => @region
|
|
if new_credentials.any?
|
|
setup_credentials new_credentials
|
|
return true
|
|
else
|
|
false
|
|
end
|
|
else
|
|
false
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|