Update AWS IMDSv1 calls to IMDSv2 in fbcode/torchsnapshot/github/.github/workflows

CodemodService Bot · facebook-github-bot · commit 66e2c445f93c · 2024-11-04T13:22:06.000-08:00
Differential Revision: D65188986
diff --git a/.github/workflows/run_tests.yaml b/.github/workflows/run_tests.yaml
@@ -63,7 +63,7 @@ jobs:
             # Pulled from instance metadata endpoint for EC2
             # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html
             category=$1
-            curl -fsSL "http://169.254.169.254/latest/meta-data/${category}"
+            curl -H "X-aws-ec2-metadata-token: $(curl -s -X PUT "http://169.254.169.254/latest/api/token" -H "X-aws-ec2-metadata-token-ttl-seconds: 30")" -fsSL "http://169.254.169.254/latest/meta-data/${category}"
           }
           echo "ami-id: $(get_ec2_metadata ami-id)"
           echo "instance-id: $(get_ec2_metadata instance-id)"

Original file line number	Diff line number	Diff line change
`@@ -63,7 +63,7 @@ jobs:`
`63`	`63`	`# Pulled from instance metadata endpoint for EC2`
`64`	`64`	`# see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html`
`65`	`65`	`category=$1`
`66`		`- curl -fsSL "http://169.254.169.254/latest/meta-data/${category}"`
	`66`	`+ curl -H "X-aws-ec2-metadata-token: $(curl -s -X PUT "http://169.254.169.254/latest/api/token" -H "X-aws-ec2-metadata-token-ttl-seconds: 30")" -fsSL "http://169.254.169.254/latest/meta-data/${category}"`
`67`	`67`	`}`
`68`	`68`	`echo "ami-id: $(get_ec2_metadata ami-id)"`
`69`	`69`	`echo "instance-id: $(get_ec2_metadata instance-id)"`