add ipv6 check (#212)

vultr · Sep 27, 2023 · d767651 · d767651
1 parent ee60acd
commit d767651
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/vultr/loadbalancers.go b/vultr/loadbalancers.go
@@ -795,11 +795,16 @@ func buildFirewallRules(service *v1.Service) ([]govultr.LBFirewallRule, error) {
 		}
 
 		source := rules[0]
+		ipType := "v4"
 		if source != "cloudflare" {
-			_, _, err := net.ParseCIDR(source)
+			ip, _, err := net.ParseCIDR(source)
 			if err != nil {
 				return nil, fmt.Errorf("loadbalancer fw rules : source %s is invalid", source)
 			}
+
+			if ip.To4() == nil {
+				ipType = "v6"
+			}
 		}
 
 		port, err := strconv.Atoi(rules[1])
@@ -808,7 +813,7 @@ func buildFirewallRules(service *v1.Service) ([]govultr.LBFirewallRule, error) {
 		}
 
 		fwRule.Source = source
-		fwRule.IPType = "v4"
+		fwRule.IPType = ipType
 		fwRule.Port = port
 		lbFWRules = append(lbFWRules, fwRule)
 	}