Rob Sullivan sent in some cleanups, which I beat on slightly.
[oweals/busybox.git] / networking / ipcalc.c
index 94b747ef69beec5cc1b5e83ea39dceca90e5d1dd..c5f2ba121ec4520b5f90d380c30dfc5cc2d4c9cc 100644 (file)
@@ -5,8 +5,8 @@
  * By Jordan Crouse <jordan@cosmicpenguin.net>
  *    Stephan Linz  <linz@li-pro.net>
  *
- * This is a complete reimplentation of the ipcalc program
- * from Redhat.  I didn't look at their source code, but there
+ * This is a complete reimplementation of the ipcalc program
+ * from Red Hat.  I didn't look at their source code, but there
  * is no denying that this is a loving reimplementation
  */
 
@@ -40,28 +40,54 @@ static unsigned long get_netmask(unsigned long ipaddr)
                return 0;
 }
 
+#ifdef CONFIG_FEATURE_IPCALC_FANCY
+static int get_prefix(unsigned long netmask)
+{
+       unsigned long msk = 0x80000000;
+       int ret = 0;
+
+       netmask = htonl(netmask);
+       while(msk) {
+               if (netmask & msk)
+                       ret++;
+               msk >>= 1;
+       }
+       return ret;
+}
+#endif
+
 #define NETMASK   0x01
 #define BROADCAST 0x02
 #define NETWORK   0x04
-#define HOSTNAME  0x08
-#define SILENT    0x10
+#define NETPREFIX 0x08
+#define HOSTNAME  0x10
+#define SILENT    0x20
+
 
 int ipcalc_main(int argc, char **argv)
 {
        unsigned long mode;
 
-       unsigned long netmask = 0;
-       unsigned long broadcast;
-       unsigned long network;
-       unsigned long ipaddr;
+       in_addr_t netmask;
+       in_addr_t broadcast;
+       in_addr_t network;
+       in_addr_t ipaddr;
+       struct in_addr a;
+
+#ifdef CONFIG_FEATURE_IPCALC_FANCY
+       unsigned long netprefix = 0;
+       int have_netmask = 0;
+       char *ipstr, *prefixstr;
+#endif
 
        static const struct option long_options[] = {
-               {"netmask", no_argument, NULL, 'n'},
-               {"broadcast", no_argument, NULL, 'b'},
-               {"network", no_argument, NULL, 'w'},
+               {"netmask",             no_argument, NULL, 'm'},
+               {"broadcast",   no_argument, NULL, 'b'},
+               {"network",             no_argument, NULL, 'n'},
 #ifdef CONFIG_FEATURE_IPCALC_FANCY
-               {"hostname", no_argument, NULL, 'h'},
-               {"silent", no_argument, NULL, 's'},
+               {"prefix",              no_argument, NULL, 'p'},
+               {"hostname",    no_argument, NULL, 'h'},
+               {"silent",              no_argument, NULL, 's'},
 #endif
                {NULL, 0, NULL, 0}
        };
@@ -69,40 +95,84 @@ int ipcalc_main(int argc, char **argv)
        bb_applet_long_options = long_options;
        mode = bb_getopt_ulflags(argc, argv,
 #ifdef CONFIG_FEATURE_IPCALC_FANCY
-                                                         "nbwhs");
+                       "mbnphs"
 #else
-                                                         "nbw");
+                       "mbn"
 #endif
-       if (mode & (BROADCAST | NETWORK)) {
-               if (argc - optind > 2) {
+                       );
+
+       argc -= optind;
+       argv += optind;
+       if (mode & (BROADCAST | NETWORK | NETPREFIX)) {
+               if (argc > 2 || argc <= 0)
                        bb_show_usage();
-               }
        } else {
-               if (argc - optind != 1) {
+               if (argc != 1)
                        bb_show_usage();
-               }
        }
 
-       ipaddr = inet_addr(argv[optind]);
-
-       if (ipaddr == INADDR_NONE) {
-               IPCALC_MSG(bb_error_msg_and_die("bad IP address: %s", argv[optind]),
-                                  exit(EXIT_FAILURE));
+#ifdef CONFIG_FEATURE_IPCALC_FANCY
+       prefixstr = ipstr = argv[0];
+
+       while(*prefixstr) {
+               if (*prefixstr == '/') {
+                       *prefixstr = (char)0;
+                       prefixstr++;
+                       if (*prefixstr) {
+                               unsigned int msk;
+
+                               if (safe_strtoul(prefixstr, &netprefix) || netprefix > 32) {
+                                       IPCALC_MSG(bb_error_msg_and_die("bad IP prefix: %s\n", prefixstr),
+                                                       exit(EXIT_FAILURE));
+                               }
+                               netmask = 0;
+                               msk = 0x80000000;
+                               while (netprefix > 0) {
+                                       netmask |= msk;
+                                       msk >>= 1;
+                                       netprefix--;
+                               }
+                               netmask = htonl(netmask);
+                               /* Even if it was 0, we will signify that we have a netmask. This allows */
+                               /* for specification of default routes, etc which have a 0 netmask/prefix */
+                               have_netmask = 1;
+                       }
+                       break;
+               }
+               prefixstr++;
        }
+       ipaddr = inet_aton(ipstr, &a);
+#else
+       ipaddr = inet_aton(argv[0], &a);
+#endif
 
-
-       if (argc - optind == 2) {
-               netmask = inet_addr(argv[optind + 1]);
+       if (ipaddr == 0) {
+               IPCALC_MSG(bb_error_msg_and_die("bad IP address: %s", argv[0]),
+                               exit(EXIT_FAILURE));
        }
+       ipaddr = a.s_addr;
 
-       if (ipaddr == INADDR_NONE) {
-               IPCALC_MSG(bb_error_msg_and_die("bad netmask: %s", argv[optind + 1]),
-                                  exit(EXIT_FAILURE));
-       }
+       if (argc == 2) {
+#ifdef CONFIG_FEATURE_IPCALC_FANCY
+               if (have_netmask) {
+                       IPCALC_MSG(bb_error_msg_and_die("Both prefix and netmask were specified, use one or the other.\n"),
+                                       exit(EXIT_FAILURE));
+               }
 
-       /* JHC - If the netmask wasn't provided then calculate it */
-       if (!netmask) {
-               netmask = get_netmask(ipaddr);
+#endif
+               netmask = inet_aton(argv[1], &a);
+               if (netmask == 0) {
+                       IPCALC_MSG(bb_error_msg_and_die("bad netmask: %s", argv[1]),
+                                       exit(EXIT_FAILURE));
+               }
+               netmask = a.s_addr;
+       } else {
+#ifdef CONFIG_FEATURE_IPCALC_FANCY
+
+               if (!have_netmask)
+#endif
+                       /* JHC - If the netmask wasn't provided then calculate it */
+                       netmask = get_netmask(ipaddr);
        }
 
        if (mode & NETMASK) {
@@ -118,7 +188,12 @@ int ipcalc_main(int argc, char **argv)
                network = ipaddr & netmask;
                printf("NETWORK=%s\n", inet_ntoa((*(struct in_addr *) &network)));
        }
+
 #ifdef CONFIG_FEATURE_IPCALC_FANCY
+       if (mode & NETPREFIX) {
+               printf("PREFIX=%i\n", get_prefix(netmask));
+       }
+
        if (mode & HOSTNAME) {
                struct hostent *hostinfo;
                int x;
@@ -126,7 +201,7 @@ int ipcalc_main(int argc, char **argv)
                hostinfo = gethostbyaddr((char *) &ipaddr, sizeof(ipaddr), AF_INET);
                if (!hostinfo) {
                        IPCALC_MSG(bb_herror_msg_and_die(
-                               "cannot find hostname for %s", argv[optind]),);
+                                               "cannot find hostname for %s", argv[0]),);
                        exit(EXIT_FAILURE);
                }
                for (x = 0; hostinfo->h_name[x]; x++) {