Add length checks

bgp · Jul 27, 2021 · d1a7ac3 · d1a7ac3
1 parent f82e338
commit d1a7ac3
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/cmd/stayrtr/stayrtr.go b/cmd/stayrtr/stayrtr.go
@@ -10,6 +10,7 @@ import (
 	"flag"
 	"fmt"
 	"io/ioutil"
+	"net"
 	"net/http"
 	"os"
 	"os/signal"
@@ -200,12 +201,22 @@ func processData(vrplistjson []prefixfile.VRPJson) ([]rtr.VRP, int, int, int) {
 			continue
 		}
 
-		count++
 		if prefix.IP.To4() != nil {
+			l, _ := net.IPMask.Size(prefix.Mask)
+			if (uint8(l) > v.Length || v.Length > 32) {
+				log.Errorf("%s Maxlength wrong: %d - %d", prefix, l, v.Length)
+				continue
+			}
 			countv4++
 		} else if prefix.IP.To16() != nil {
+			l, _ := net.IPMask.Size(prefix.Mask)
+			if (uint8(l) > v.Length || v.Length > 128) {
+				log.Errorf("%s Maxlength wrong: %d - %d", prefix, l, v.Length)
+				continue
+			}
 			countv6++
 		}
+		count++
 
 		key := fmt.Sprintf("%s,%d,%d", prefix, asn, v.Length)
 		_, exists := filterDuplicates[key]