From 4ec71dada9218b52bc8f377c2000af6aa76d4e7a Mon Sep 17 00:00:00 2001
From: Tim Steiner <tim.steiner@nebraska.edu>
Date: Tue, 16 Jan 2024 11:50:49 -0600
Subject: [PATCH] More IMDSv2 tokens for EFS.
---
update-efs-mount.py | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/update-efs-mount.py b/update-efs-mount.py
index 696a4a7..4a91f92 100755
--- a/update-efs-mount.py
+++ b/update-efs-mount.py
@@ -1,8 +1,7 @@
#!/usr/bin/python
import boto3
-import urllib2
-import json
+import requests
import sys
import re
import tempfile
@@ -42,8 +41,8 @@ def main():
# Returns the region the instance is currently running in (eg: 'us-east')
def get_region():
- instance_info = urllib2.urlopen('http://169.254.169.254/2018-09-24/dynamic/instance-identity/document/').read()
- instance_info = json.loads(instance_info)
+ session_token = requests.put('http://169.254.169.254/latest/api/token', headers={'X-aws-ec2-metadata-token-ttl-seconds': 60}).text
+ instance_info = requests.get('http://169.254.169.254/2018-09-24/dynamic/instance-identity/document/', headers={'X-aws-ec2-metadata-token': session_token}).json()
return instance_info['region']
# Returns the instance_id of this VM.
--
GitLab