Add utils to get all IP addresses (aws#4642)

* Add utils to get all IP addresses

* Update comments

Author: xxx0624

agent/vendor/github.com/aws/amazon-ecs-agent/ecs-agent/tmds/utils/netconfig/netconfig_linux.go

@@ -83,6 +83,7 @@ github.com/aws/amazon-ecs-agent/ecs-agent/utils/execwrapper/mocks
 github.com/aws/amazon-ecs-agent/ecs-agent/utils/httpproxy
 github.com/aws/amazon-ecs-agent/ecs-agent/utils/netlinkwrapper
 github.com/aws/amazon-ecs-agent/ecs-agent/utils/netlinkwrapper/mocks
+github.com/aws/amazon-ecs-agent/ecs-agent/utils/netwrapper
 github.com/aws/amazon-ecs-agent/ecs-agent/utils/retry
 github.com/aws/amazon-ecs-agent/ecs-agent/utils/retry/mock
 github.com/aws/amazon-ecs-agent/ecs-agent/utils/ttime

agent/vendor/github.com/aws/amazon-ecs-agent/ecs-agent/tmds/utils/netconfig/netconfig_unsupported.go

@@ -17,9 +17,12 @@
 package netconfig
 
 import (
+	"net"
+
 	"github.com/aws/amazon-ecs-agent/ecs-agent/logger"
 	"github.com/aws/amazon-ecs-agent/ecs-agent/logger/field"
 	"github.com/aws/amazon-ecs-agent/ecs-agent/utils/netlinkwrapper"
+	"github.com/aws/amazon-ecs-agent/ecs-agent/utils/netwrapper"
 
 	"github.com/vishvananda/netlink"
 )
@@ -69,3 +72,32 @@ func DefaultNetInterfaceName(netlinkClient netlinkwrapper.NetLink) (string, erro
 	}
 	return "", nil
 }
+
+// GetInterfaceGlobalIPAddresses returns all global unicast IP addresses (both IPv4 and IPv6)
+// assigned to the given network interface. It excludes link-local, loopback, multicast,
+// and unspecified addresses. Returns an empty list if no global unicast addresses are found,
+// or an error if the interface cannot be accessed.
+func GetInterfaceGlobalIPAddresses(nw netwrapper.Net, ifaceName string) ([]string, error) {
+	iface, err := nw.InterfaceByName(ifaceName)
+	if err != nil {
+		return nil, err
+	}
+
+	allAddrs, err := nw.Addrs(iface)
+	if err != nil {
+		return nil, err
+	}
+
+	ipAddrs := make([]string, 0)
+	for _, addr := range allAddrs {
+		ipNet, ok := addr.(*net.IPNet)
+		if !ok {
+			continue
+		}
+
+		if ipNet.IP.IsGlobalUnicast() {
+			ipAddrs = append(ipAddrs, ipNet.IP.String())
+		}
+	}
+	return ipAddrs, nil
+}

agent/vendor/github.com/aws/amazon-ecs-agent/ecs-agent/tmds/utils/netconfig/netconfig_windows.go

@@ -17,10 +17,13 @@
 package netconfig
 
 import (
+	"errors"
 	"net"
+	"reflect"
 	"testing"
 
 	mock_netlinkwrapper "github.com/aws/amazon-ecs-agent/ecs-agent/utils/netlinkwrapper/mocks"
+	mock_nw "github.com/aws/amazon-ecs-agent/ecs-agent/utils/netwrapper/mocks"
 
 	"github.com/golang/mock/gomock"
 	"github.com/stretchr/testify/assert"
@@ -167,3 +170,83 @@ func TestDefaultNetInterfaceName(t *testing.T) {
 		})
 	}
 }
+
+func TestGetInterfaceGlobalIPAddresses(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	tests := []struct {
+		name            string
+		addrs           []net.Addr
+		ifaceErrMsg     string
+		ifaceAddrErrMsg string
+		expectedAddrs   []string
+		expectedErrMsg  string
+	}{
+		{
+			name: "success with mixed IPs",
+			addrs: []net.Addr{
+				&net.IPNet{IP: net.ParseIP("192.168.1.100"), Mask: net.CIDRMask(24, 32)},
+				&net.IPNet{IP: net.ParseIP("fe80::1"), Mask: net.CIDRMask(64, 128)},
+				&net.IPNet{IP: net.ParseIP("2001:db8::1"), Mask: net.CIDRMask(64, 128)},
+				&net.IPNet{IP: net.ParseIP("127.0.0.1"), Mask: net.CIDRMask(8, 32)},
+				&net.IPNet{IP: net.ParseIP("::1"), Mask: net.CIDRMask(128, 128)},
+			},
+			expectedAddrs:  []string{"192.168.1.100", "2001:db8::1"},
+			expectedErrMsg: "",
+		},
+		{
+			name:           "empty address list",
+			addrs:          nil,
+			expectedAddrs:  []string{},
+			expectedErrMsg: "",
+		},
+		{
+			name:           "interface error",
+			ifaceErrMsg:    "fail to get the interface",
+			expectedErrMsg: "fail to get the interface",
+		},
+		{
+			name:            "addrs error",
+			ifaceAddrErrMsg: "fail to get the IP address",
+			expectedErrMsg:  "fail to get the IP address",
+		},
+		{
+			name: "invalid addr",
+			addrs: []net.Addr{
+				&net.TCPAddr{IP: net.ParseIP("192.168.1.100"), Port: 80},
+			},
+			expectedAddrs:  []string{},
+			expectedErrMsg: "",
+		},
+	}
+
+	for _, tc := range tests {
+		tc := tc
+		t.Run(tc.name, func(t *testing.T) {
+			t.Parallel()
+
+			nw := mock_nw.NewMockNet(ctrl)
+			if tc.ifaceErrMsg != "" {
+				nw.EXPECT().InterfaceByName("test0").Return(nil, errors.New(tc.ifaceErrMsg))
+			} else {
+				nw.EXPECT().InterfaceByName("test0").Return(nil, nil)
+				if tc.ifaceAddrErrMsg != "" {
+					nw.EXPECT().Addrs(nil).Return(nil, errors.New(tc.ifaceAddrErrMsg))
+				} else {
+					nw.EXPECT().Addrs(nil).Return(tc.addrs, nil)
+				}
+			}
+
+			got, err := GetInterfaceGlobalIPAddresses(nw, "test0")
+			if err != nil {
+				assert.EqualValues(t, tc.expectedErrMsg, err.Error())
+				return
+			}
+
+			if !reflect.DeepEqual(got, tc.expectedAddrs) {
+				t.Errorf("GetInterfaceGlobalIPAddresses() = %v, want %v", got, tc.expectedAddrs)
+			}
+		})
+	}
+}

agent/vendor/github.com/aws/amazon-ecs-agent/ecs-agent/utils/netwrapper/generate_mocks.go

@@ -30,3 +30,9 @@ func NewNetworkConfigClient() *NetworkConfigClient {
 func DefaultNetInterfaceName(unknown interface{}) (string, error) {
 	return "", errors.New("not supported on unknown platform")
 }
+
+// GetInterfaceGlobalIPAddresses returns all global unicast IP addresses (both IPv4 and IPv6)
+// assigned to the given network interface. This is only supported on linux as of now.
+func GetInterfaceGlobalIPAddresses(unknown interface{}, ifaceName string) ([]string, error) {
+	return nil, errors.New("not supported on windows")
+}

agent/vendor/github.com/aws/amazon-ecs-agent/ecs-agent/utils/netwrapper/net.go

@@ -33,3 +33,9 @@ func NewNetworkConfigClient() *NetworkConfigClient {
 func DefaultNetInterfaceName(unknown interface{}) (string, error) {
 	return "", errors.New("not supported on windows")
 }
+
+// GetInterfaceGlobalIPAddresses returns all global unicast IP addresses (both IPv4 and IPv6)
+// assigned to the given network interface. This is only supported on linux as of now.
+func GetInterfaceGlobalIPAddresses(unknown interface{}, ifaceName string) ([]string, error) {
+	return nil, errors.New("not supported on windows")
+}