support IP addresses and CIDRs

nettica-com · Jul 1, 2024 · 1e11b6f · 1e11b6f
1 parent 7fc8f42
commit 1e11b6f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/util/util.go b/util/util.go
@@ -135,6 +135,15 @@ func IsInCidr(cidr string, subnet string) bool {
 	if err != nil {
 		return false
 	}
+	parts := strings.Split(cidr, "/")
+	if len(parts) == 1 {
+		ip := net.ParseIP(cidr)
+		if ip == nil {
+			return false
+		}
+		return ipnet.Contains(ip)
+	}
+
 	ip, _, err := net.ParseCIDR(cidr)
 	if err != nil {
 		return false