Dual stack EC2 client (#4658)

Author: amogh09

agent/app/agent.go

@@ -210,7 +210,11 @@ func newAgent(blackholeEC2Metadata bool, acceptInsecureCert *bool) (agent, error
 		cfg.NoIID = true
 	}
 
-	ec2Client, err := ec2.NewClientImpl(cfg.AWSRegion)
+	ec2ClientDualStackEndpointState := aws.DualStackEndpointStateDisabled
+	if cfg.InstanceIPCompatibility.IsIPv6Only() {
+		ec2ClientDualStackEndpointState = aws.DualStackEndpointStateEnabled
+	}
+	ec2Client, err := ec2.NewClientImpl(cfg.AWSRegion, ec2ClientDualStackEndpointState)
 	if err != nil {
 		logger.Critical("Error creating EC2 client", logger.Fields{
 			field.Error: err,

agent/vendor/github.com/aws/amazon-ecs-agent/ecs-agent/ec2/ec2_client.go

@@ -54,7 +54,10 @@ type ClientImpl struct {
 	client ClientSDK
 }
 
-func NewClientImpl(awsRegion string) (Client, error) {
+func NewClientImpl(
+	awsRegion string,
+	dualStackEndpointState aws.DualStackEndpointState,
+) (Client, error) {
 	credentialsProvider := providers.NewInstanceCredentialsCache(
 		false,
 		providers.NewRotatingSharedCredentialsProviderV2(),
@@ -65,6 +68,7 @@ func NewClientImpl(awsRegion string) (Client, error) {
 		config.WithRegion(awsRegion),
 		config.WithCredentialsProvider(credentialsProvider),
 		config.WithRetryMaxAttempts(clientRetriesNum),
+		config.WithUseDualStackEndpoint(dualStackEndpointState),
 	)
 	if err != nil {
 		return nil, err

ecs-agent/ec2/ec2_client.go

@@ -34,7 +34,7 @@ func TestCreateTags(t *testing.T) {
 	defer ctrl.Finish()
 
 	mockClientSDK := mock_ec2.NewMockClientSDK(ctrl)
-	testClient, err := ec2.NewClientImpl("us-west-2")
+	testClient, err := ec2.NewClientImpl("us-west-2", aws.DualStackEndpointStateDisabled)
 	assert.NoError(t, err)
 	testClient.(*ec2.ClientImpl).SetClientSDK(mockClientSDK)
 
@@ -54,7 +54,7 @@ func TestDescribeECSTagsForInstance(t *testing.T) {
 
 	instanceID := "iid"
 	mockClientSDK := mock_ec2.NewMockClientSDK(ctrl)
-	testClient, err := ec2.NewClientImpl("us-west-2")
+	testClient, err := ec2.NewClientImpl("us-west-2", aws.DualStackEndpointStateDisabled)
 	assert.NoError(t, err)
 	testClient.(*ec2.ClientImpl).SetClientSDK(mockClientSDK)