Skip to content
Snippets Groups Projects
Commit 4ec71dad authored by Tim Steiner's avatar Tim Steiner
Browse files

More IMDSv2 tokens for EFS.

parent 758f8661
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
Showing
with 3 additions and 4 deletions
update-efs-mount.py
+ 3
− 4
View file @ 4ec71dad
#!/usr/bin/python #!/usr/bin/python
import boto3 import boto3
import urllib2 import requests
import json
import sys import sys
import re import re
import tempfile import tempfile
...@@ -42,8 +41,8 @@ def main(): ...@@ -42,8 +41,8 @@ def main():
# Returns the region the instance is currently running in (eg: 'us-east') # Returns the region the instance is currently running in (eg: 'us-east')
def get_region(): def get_region():
instance_info = urllib2.urlopen('http://169.254.169.254/2018-09-24/dynamic/instance-identity/document/').read() session_token = requests.put('http://169.254.169.254/latest/api/token', headers={'X-aws-ec2-metadata-token-ttl-seconds': 60}).text
instance_info = json.loads(instance_info) 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'] return instance_info['region']
# Returns the instance_id of this VM. # Returns the instance_id of this VM.
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment