cephfs: update fetchIP to support ipv6 addresses

Signed-off-by: Riya Singhal <[email protected]>
ceph · Nov 8, 2023 · 0631c15 · 0631c15
1 parent 304462c
commit 0631c15
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/internal/csi-addons/networkfence/fencing.go b/internal/csi-addons/networkfence/fencing.go
@@ -212,9 +212,12 @@ func (ac *activeClient) fetchIP() (string, error) {
 	clientInfo := ac.Inst
 	parts := strings.Fields(clientInfo)
 	if len(parts) >= 2 {
-		ip := strings.Split(parts[1], ":")[0]
-
-		return ip, nil
+		lastColonIndex := strings.LastIndex(parts[1], ":")
+		firstPart := parts[1][:lastColonIndex]
+		ip := net.ParseIP(firstPart)
+		if ip != nil {
+			return ip.String(), nil
+		}
 	}
 
 	return "", fmt.Errorf("failed to extract IP address, incorrect format: %s", clientInfo)