fix: Add IPv6 support for signing HTTP request

Add in a new function isIPv6() for IPv6 host check.

Author: Yue Yin

go.mod

@@ -14,4 +14,4 @@ require (
 	github.com/vmware/vmw-guestinfo v0.0.0-20170707015358-25eff159a728
 )
 
-require github.com/kr/text v0.1.0 // indirect
+require	github.com/kr/text v0.1.0 // indirect

sts/signer.go

@@ -30,6 +30,7 @@ import (
 	"io"
 	"io/ioutil"
 	mrand "math/rand"
+	"net"
 	"net/http"
 	"net/url"
 	"strings"
@@ -274,11 +275,18 @@ func (s *Signer) SignRequest(req *http.Request) error {
 		}
 
 		var buf bytes.Buffer
+		host := req.URL.Hostname()
+
+		// Check if the host IP is in IPv6 format. If yes, add the opening and closing square brackets.
+		if isIPv6(host) {
+			host = fmt.Sprintf("%s%s%s", "[", host, "]")
+		}
+
 		msg := []string{
 			nonce,
 			req.Method,
 			req.URL.Path,
-			strings.ToLower(req.URL.Hostname()),
+			strings.ToLower(host),
 			port,
 		}
 		for i := range msg {
@@ -329,3 +337,11 @@ func (s *Signer) NewRequest() TokenRequest {
 		KeyID:       s.keyID,
 	}
 }
+
+func isIPv6(s string) bool {
+	ip := net.ParseIP(s)
+	if ip == nil {
+		return false
+	}
+	return len(ip) == net.IPv6len
+}

sts/signer_test.go

@@ -133,4 +133,10 @@ func TestSigner(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+
+	req, _ = http.NewRequest(http.MethodPost, "https://[0:0:0:0:0:0:0:1]", nil)
+	err = s.SignRequest(req)
+	if err != nil {
+		t.Fatal(err)
+	}
 }