store the token for temporary credentials

ex-aws · Jun 29, 2015 · 956e778 · 956e778
1 parent cc0ac7f
commit 956e778
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/lib/ex_aws/auth.ex b/lib/ex_aws/auth.ex
@@ -2,6 +2,8 @@ defmodule ExAws.Auth do
   import ExAws.Auth.Utils
   alias Timex.DateFormat
 
+  @moduledoc false
+
   def headers(http_method, url, service, config, headers, body) do
     now = %{Timex.Date.now | ms: 0}
     headers = [
@@ -22,7 +24,13 @@ defmodule ExAws.Auth do
       now)
 
     [{"Authorization", auth_header} | headers ]
+    |> handle_temp_credentials(config)
+  end
+
+  def handle_temp_credentials(headers, %{token: token}) do
+    [{"X-Amz-Security-Token", token} | headers]
   end
+  def handle_temp_credentials(headers, _), do: headers
 
   def auth_header(access_key, secret_key, http_method, url, region, service, headers, body, now) do
     date  = DateFormat.format!(now, "%Y%m%d", :strftime)

diff --git a/lib/ex_aws/instance_meta.ex b/lib/ex_aws/instance_meta.ex
@@ -18,6 +18,7 @@ defmodule ExAws.InstanceMeta do
     %{
       access_key_id: result["AccessKeyId"],
       secret_access_key: result["SecretAccessKey"],
+      token: result["Token"],
       expiration: result["Expiration"]
     }
   end