Add support for AWS IMDSv2 (#158)

Co-authored-by: Alexander Tarasov <[email protected]>
widdix · Jul 31, 2020 · 00a62f9 · 00a62f9
1 parent 1196ff4
commit 00a62f9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/import_users.sh b/import_users.sh
@@ -61,8 +61,9 @@ fi
 : ${USERDEL_ARGS:="--force --remove"}
 
 # Initizalize INSTANCE variable
-INSTANCE_ID=$(curl -s http://169.254.169.254/latest/meta-data/instance-id)
-REGION=$(curl -s http://169.254.169.254/latest/dynamic/instance-identity/document | grep region | awk -F\" '{print $4}')
+METADATA_TOKEN=$(curl -s -X PUT -H 'x-aws-ec2-metadata-token-ttl-seconds: 60' http://169.254.169.254/latest/api/token)
+INSTANCE_ID=$(curl -s -H "x-aws-ec2-metadata-token: ${METADATA_TOKEN}" http://169.254.169.254/latest/meta-data/instance-id)
+REGION=$(curl -s -H "x-aws-ec2-metadata-token: ${METADATA_TOKEN}" http://169.254.169.254/latest/dynamic/instance-identity/document | grep region | awk -F\" '{print $4}')
 
 function setup_aws_credentials() {
     local stscredentials