libbb: add forgotten file from previous commit :(
[oweals/busybox.git] / networking / ipcalc.c
index 2c23d17f6df700ed9ef88799ffffbb45711b9758..d8fa5f3f2fa224998ecd76fd2d3609c821002c09 100644 (file)
-/* vi: set sw=4 ts=4 ai: */
+/* vi: set sw=4 ts=4: */
 /*
  * Mini ipcalc implementation for busybox
  *
  * 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
+ *
+ * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  */
 
-#include <stdio.h>
-#include <stdlib.h>
-#include <ctype.h>
-#include <getopt.h>
 #include <sys/socket.h>
-#include <netinet/in.h>
 #include <arpa/inet.h>
 
-#include "busybox.h"
+#include "libbb.h"
 
-#define IPCALC_MSG(CMD,ALTCMD) if (mode & SILENT) {ALTCMD;} else {CMD;}
+#define CLASS_A_NETMASK        ntohl(0xFF000000)
+#define CLASS_B_NETMASK        ntohl(0xFFFF0000)
+#define CLASS_C_NETMASK        ntohl(0xFFFFFF00)
 
 static unsigned long get_netmask(unsigned long ipaddr)
 {
-       if (ipaddr & 0xC0) {
-               return 0x00FFFFFF;      /* Class C */
-       }
-       if (ipaddr & 0x10) {
-               return 0x0000FFFF;      /* Class B */
+       ipaddr = htonl(ipaddr);
+
+       if ((ipaddr & 0xC0000000) == 0xC0000000)
+               return CLASS_C_NETMASK;
+       else if ((ipaddr & 0x80000000) == 0x80000000)
+               return CLASS_B_NETMASK;
+       else if ((ipaddr & 0x80000000) == 0)
+               return CLASS_A_NETMASK;
+       else
+               return 0;
+}
+
+#if ENABLE_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 0x000000FF;      /* Class A */
+       return ret;
 }
+#else
+int get_prefix(unsigned long netmask);
+#endif
+
 
 #define NETMASK   0x01
 #define BROADCAST 0x02
 #define NETWORK   0x04
-#define HOSTNAME  0x08
-#define SILENT    0x80
+#define NETPREFIX 0x08
+#define HOSTNAME  0x10
+#define SILENT    0x20
+
+#if ENABLE_FEATURE_IPCALC_LONG_OPTIONS
+       static const char ipcalc_longopts[] ALIGN1 =
+               "netmask\0"   No_argument "m"
+               "broadcast\0" No_argument "b"
+               "network\0"   No_argument "n"
+# if ENABLE_FEATURE_IPCALC_FANCY
+               "prefix\0"    No_argument "p"
+               "hostname\0"  No_argument "h"
+               "silent\0"    No_argument "s"
+# endif
+               ;
+#endif
 
+int ipcalc_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 int ipcalc_main(int argc, char **argv)
 {
-       unsigned char mode = 0;
-
-       unsigned long netmask = 0;
-       unsigned long broadcast = 0;
-       unsigned long network = 0;
-       unsigned long ipaddr = 0;
-
-       int opt = 0;
-
-       struct option long_options[] = {
-               {"netmask", no_argument, NULL, 'n'},
-               {"broadcast", no_argument, NULL, 'b'},
-               {"network", no_argument, NULL, 'w'},
-#ifdef CONFIG_FEATURE_IPCALC_FANCY
-               {"hostname", no_argument, NULL, 'h'},
-               {"silent", no_argument, NULL, 's'},
+       unsigned opt;
+       int have_netmask = 0;
+       in_addr_t netmask, broadcast, network, ipaddr;
+       struct in_addr a;
+       char *ipstr;
+
+#if ENABLE_FEATURE_IPCALC_LONG_OPTIONS
+       applet_long_options = ipcalc_longopts;
 #endif
-               {NULL, 0, NULL, 0}
-       };
-
-
-       while ((opt = getopt_long(argc, argv,
-#ifdef CONFIG_FEATURE_IPCALC_FANCY
-                                                         "nbwhs",
-#else
-                                                         "nbw",
-#endif
-                                                         long_options, NULL)) != EOF) {
-               if (opt == 'n')
-                       mode |= NETMASK;
-               else if (opt == 'b')
-                       mode |= BROADCAST;
-               else if (opt == 'w')
-                       mode |= NETWORK;
-#ifdef CONFIG_FEATURE_IPCALC_FANCY
-               else if (opt == 'h')
-                       mode |= HOSTNAME;
-               else if (opt == 's')
-                       mode |= SILENT;
-#endif
-               else {
-                       bb_show_usage();
-               }
-       }
-
-       if (mode & (BROADCAST | NETWORK)) {
-               if (argc - optind > 2) {
+       opt = getopt32(argv, "mbn" USE_FEATURE_IPCALC_FANCY("phs"));
+       argc -= optind;
+       argv += optind;
+       if (opt & (BROADCAST | NETWORK | NETPREFIX)) {
+               if (argc > 2 || argc <= 0)
                        bb_show_usage();
-               }
        } else {
-               if (argc - optind != 1) {
+               if (argc != 1)
                        bb_show_usage();
+       }
+       if (opt & SILENT)
+               logmode = LOGMODE_NONE; /* Suppress error_msg() output */
+
+       ipstr = argv[0];
+       if (ENABLE_FEATURE_IPCALC_FANCY) {
+               unsigned long netprefix = 0;
+               char *prefixstr;
+
+               prefixstr = ipstr;
+
+               while (*prefixstr) {
+                       if (*prefixstr == '/') {
+                               *prefixstr = (char)0;
+                               prefixstr++;
+                               if (*prefixstr) {
+                                       unsigned msk;
+                                       netprefix = xatoul_range(prefixstr, 0, 32);
+                                       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);
 
-       ipaddr = inet_addr(argv[optind]);
-
-       if (ipaddr == INADDR_NONE) {
-               IPCALC_MSG(bb_error_msg_and_die("bad IP address: %s\n", argv[optind]),
-                                  exit(EXIT_FAILURE));
+       if (ipaddr == 0) {
+               bb_error_msg_and_die("bad IP address: %s", argv[0]);
        }
+       ipaddr = a.s_addr;
 
+       if (argc == 2) {
+               if (ENABLE_FEATURE_IPCALC_FANCY && have_netmask) {
+                       bb_error_msg_and_die("use prefix or netmask, not both");
+               }
 
-       if (argc - optind == 2) {
-               netmask = inet_addr(argv[optind + 1]);
-       }
-
-       if (ipaddr == INADDR_NONE) {
-               IPCALC_MSG(bb_error_msg_and_die("bad netmask: %s\n", argv[optind + 1]),
-                                  exit(EXIT_FAILURE));
-       }
+               netmask = inet_aton(argv[1], &a);
+               if (netmask == 0) {
+                       bb_error_msg_and_die("bad netmask: %s", argv[1]);
+               }
+               netmask = a.s_addr;
+       } else {
 
-       /* JHC - If the netmask wasn't provided then calculate it */
-       if (!netmask) {
-               netmask = get_netmask(ipaddr);
+               /* JHC - If the netmask wasn't provided then calculate it */
+               if (!ENABLE_FEATURE_IPCALC_FANCY || !have_netmask)
+                       netmask = get_netmask(ipaddr);
        }
 
-       if (mode & NETMASK) {
+       if (opt & NETMASK) {
                printf("NETMASK=%s\n", inet_ntoa((*(struct in_addr *) &netmask)));
        }
 
-       if (mode & BROADCAST) {
+       if (opt & BROADCAST) {
                broadcast = (ipaddr & netmask) | ~netmask;
                printf("BROADCAST=%s\n", inet_ntoa((*(struct in_addr *) &broadcast)));
        }
 
-       if (mode & NETWORK) {
+       if (opt & NETWORK) {
                network = ipaddr & netmask;
                printf("NETWORK=%s\n", inet_ntoa((*(struct in_addr *) &network)));
        }
-#ifdef CONFIG_FEATURE_IPCALC_FANCY
-       if (mode & HOSTNAME) {
-               struct hostent *hostinfo;
-               int x;
-
-               hostinfo = gethostbyaddr((char *) &ipaddr, sizeof(ipaddr), AF_INET);
-               if (!hostinfo) {
-                       IPCALC_MSG(bb_error_msg("cannot find hostname for %s", argv[optind]);
-                                          herror(NULL);
-                                          putc('\n', stderr);,);
-                       exit(EXIT_FAILURE);
-               }
-               for (x = 0; hostinfo->h_name[x]; x++) {
-                       hostinfo->h_name[x] = tolower(hostinfo->h_name[x]);
+
+       if (ENABLE_FEATURE_IPCALC_FANCY) {
+               if (opt & NETPREFIX) {
+                       printf("PREFIX=%i\n", get_prefix(netmask));
                }
 
-               printf("HOSTNAME=%s\n", hostinfo->h_name);
+               if (opt & HOSTNAME) {
+                       struct hostent *hostinfo;
+
+                       hostinfo = gethostbyaddr((char *) &ipaddr, sizeof(ipaddr), AF_INET);
+                       if (!hostinfo) {
+                               bb_herror_msg_and_die("cannot find hostname for %s", argv[0]);
+                       }
+                       str_tolower(hostinfo->h_name);
+
+                       printf("HOSTNAME=%s\n", hostinfo->h_name);
+               }
        }
-#endif
 
        return EXIT_SUCCESS;
 }
-
-/* END CODE */
-/*
-Local Variables:
-c-file-style: "linux"
-c-basic-offset: 4
-tab-width: 4
-End:
-*/