Added check AF_INET/AF_INET6 for dual-stack environments

nik-netlox · Oct 15, 2024 · f354fa8 · f354fa8
1 parent f919f5a
commit f354fa8
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/api/loxinlp/nlp.go b/api/loxinlp/nlp.go
@@ -1188,10 +1188,19 @@ func DelNeigh(neigh nlp.Neigh, link nlp.Link) int {
 func AddRoute(route nlp.Route) int {
 	var ipNet net.IPNet
 	if route.Dst == nil {
-		r := net.IPv4(0, 0, 0, 0)
-		m := net.CIDRMask(0, 32)
-		r = r.Mask(m)
-		ipNet = net.IPNet{IP: r, Mask: m}
+		if route.Family == 2 {
+			r := net.IPv4(0, 0, 0, 0)
+			m := net.CIDRMask(0, 32)
+			r = r.Mask(m)
+			ipNet = net.IPNet{IP: r, Mask: m}
+		} else if route.Family == 10 {
+			r := net.ParseIP("::")
+			m := net.CIDRMask(0, 128)
+			r = r.Mask(m)
+			ipNet = net.IPNet{IP: r, Mask: m}
+		} else {
+			return -1
+		}
 	} else {
 		ipNet = *route.Dst
 	}