- Added Multicast and Broadcast checks for route and next hop for static route command.

1 files changed, 27 insertions, 58 deletions

diff --git a/src/check_prefix_boundary.c b/src/check_prefix_boundary.c
index 1c9ee30a..b432c788 100644
--- a/src/check_prefix_boundary.c
+++ b/src/check_prefix_boundary.c
@@ -1,33 +1,4 @@
-/*
- * Check format of network prefix
- */
-#include <stdio.h>
-#include <stdarg.h>
-#include <stdint.h>
-#include <stdlib.h>
-#include <string.h>
-#include <sys/types.h>
-#include <sys/socket.h>
-#include <arpa/inet.h>
-
-typedef struct
-{
-	uint8_t family;
-	uint8_t bytelen;
-	unsigned int plen;
-	uint32_t data[4];
-} inet_prefix;
-
-static void err(const char *fmt, ...)
-{
-	va_list ap;
-	
-	va_start(ap, fmt);
-	vfprintf(stderr, fmt, ap);
-	va_end(ap);
-
-	exit(1);
-}
+#include "check_ucast_static.h"
 
 static void usage(void)
 {
@@ -35,32 +6,6 @@ static void usage(void)
 	exit(1);
 }
 
-static void get_addr_1(inet_prefix *addr, const char *name, int family)
-{
-	memset(addr, 0, sizeof(*addr));
-
-	if (strchr(name, ':')) {
-		addr->family = AF_INET6;
-		addr->bytelen = 16;
-		if (family != AF_UNSPEC && family != AF_INET6)
-			err("IPV6 address not allowed\n");
-
-		if (inet_pton(AF_INET6, name, addr->data) <= 0)
-			err("Invalid IPV6 address: %s\n", name);
-
-		return;
-	}
-
-	addr->family = AF_INET;
-	addr->bytelen = 4;
-	if (family != AF_UNSPEC && family != AF_INET)
-		err("IPV4 address not allowed\n");
-
-	if (inet_pton(AF_INET, name, addr->data) <= 0)
-		err("Invalid IPV4 address: %s\n", name);
-	return;
-}
-
 static void get_prefix_1(inet_prefix *dst, char *arg, int family)
 {
 	char *slash, *endp;
@@ -103,12 +48,12 @@ static void get_netmask(inet_prefix *msk, const inet_prefix *dst)
 int main(int argc, char **argv)
 {
 	int family = AF_UNSPEC;
-
+    
 	while (--argc) {
 		char *arg = *++argv;
 		inet_prefix dst, msk;
 
-		if (arg[0] == '-') 
+		if (arg[0] == '-') { 
 			switch(arg[1]) {
 			case '4':
 				family = AF_INET;
@@ -119,6 +64,8 @@ int main(int argc, char **argv)
 			default:
 				usage();
 			}
+            continue;
+        }
 
 		get_prefix_1(&dst, arg, family);
 		get_netmask(&msk, &dst);
@@ -129,6 +76,28 @@ int main(int argc, char **argv)
 			    "Did you mean %s?\n", 
 			    inet_ntop(msk.family, msk.data, buf, sizeof buf));
 		}
+
+        /*
+         * Macros to check for Mcast are based on:
+         *
+         *    Addr          dst.data
+         * 224.1.2.2    ==> 0x030201e0
+         * ff01:0203::  ==> 0x030201ff
+         *
+         */
+        if (family == AF_INET) {
+            if (IS_MULTICAST(dst.data[0])) {
+                err("Invalid Prefix...Route cannot be Multicast\n");
+            } 
+            if (IS_BROADCAST(dst.data[0])) {
+                err("Invalid Prefix...Route cannot be Broadcast\n");
+            } 
+        } else if (family == AF_INET6) {
+            if (IS_IPV6_MULTICAST(dst.data[0])) {
+                err("Invalid Prefix...Route cannot be IPv6 Multicast\n");
+            } 
+        }
+
 	}
 
 	return 0;