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

More IMDSv2 tokens.

parent 9e506f39
No related branches found
No related tags found
No related merge requests found
#!/usr/bin/python
import boto3
import urllib2
import json
import requests
import sys
#import re
#import tempfile
......@@ -41,8 +40,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.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment