Bump version to 1.32.0
[oweals/busybox.git] / networking / ipcalc.c
index d75c883b8e62bd446da1ee2bd06b2ce8211c8f83..09b14687244f1ddc19646ef2d6d7c35ee229698c 100644 (file)
@@ -1,30 +1,61 @@
-/* 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 source tree.
  */
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <ctype.h>
-#include <getopt.h>
-#include <sys/socket.h>
-#include <netinet/in.h>
+//config:config IPCALC
+//config:      bool "ipcalc (4.4 kb)"
+//config:      default y
+//config:      help
+//config:      ipcalc takes an IP address and netmask and calculates the
+//config:      resulting broadcast, network, and host range.
+//config:
+//config:config FEATURE_IPCALC_LONG_OPTIONS
+//config:      bool "Enable long options"
+//config:      default y
+//config:      depends on IPCALC && LONG_OPTS
+//config:
+//config:config FEATURE_IPCALC_FANCY
+//config:      bool "Fancy IPCALC, more options, adds 1 kbyte"
+//config:      default y
+//config:      depends on IPCALC
+//config:      help
+//config:      Adds the options hostname, prefix and silent to the output of
+//config:      "ipcalc".
+
+//applet:IF_IPCALC(APPLET_NOEXEC(ipcalc, ipcalc, BB_DIR_BIN, BB_SUID_DROP, ipcalc))
+
+//kbuild:lib-$(CONFIG_IPCALC) += ipcalc.o
+
+//usage:#define ipcalc_trivial_usage
+//usage:       "[OPTIONS] ADDRESS"
+//usage:       IF_FEATURE_IPCALC_FANCY("[/PREFIX]") " [NETMASK]"
+//usage:#define ipcalc_full_usage "\n\n"
+//usage:       "Calculate and display network settings from IP address\n"
+//usage:     "\n       -b      Broadcast address"
+//usage:     "\n       -n      Network address"
+//usage:     "\n       -m      Default netmask for IP"
+//usage:       IF_FEATURE_IPCALC_FANCY(
+//usage:     "\n       -p      Prefix for IP/NETMASK"
+//usage:     "\n       -h      Resolved host name"
+//usage:     "\n       -s      No error messages"
+//usage:       )
+
+#include "libbb.h"
+/* After libbb.h, because on some systems it needs other includes */
 #include <arpa/inet.h>
 
-#include "busybox.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)
+#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)
 {
@@ -40,22 +71,25 @@ static unsigned long get_netmask(unsigned long ipaddr)
                return 0;
 }
 
-#ifdef CONFIG_FEATURE_IPCALC_FANCY
+#if ENABLE_FEATURE_IPCALC_FANCY
 static int get_prefix(unsigned long netmask)
 {
        unsigned long msk = 0x80000000;
-        int ret = 0;
+       int ret = 0;
 
        netmask = htonl(netmask);
-       while(msk) {
+       while (msk) {
                if (netmask & msk)
                        ret++;
                msk >>= 1;
-        }
-        return ret;
+       }
+       return ret;
 }
+#else
+int get_prefix(unsigned long netmask);
 #endif
 
+
 #define NETMASK   0x01
 #define BROADCAST 0x02
 #define NETWORK   0x04
@@ -63,161 +97,134 @@ static int get_prefix(unsigned long netmask)
 #define HOSTNAME  0x10
 #define SILENT    0x20
 
-
-int ipcalc_main(int argc, char **argv)
-{
-       unsigned long mode;
-
-       unsigned long netmask;
-       unsigned long broadcast;
-       unsigned long network;
-       unsigned long 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, 'm'},
-               {"broadcast", no_argument, NULL, 'b'},
-               {"network", no_argument, NULL, 'n'},
-#ifdef CONFIG_FEATURE_IPCALC_FANCY
-                {"prefix", no_argument, NULL, 'p'},
-               {"hostname", no_argument, NULL, 'h'},
-               {"silent", no_argument, NULL, 's'},
-#endif
-               {NULL, 0, NULL, 0}
-       };
-
-       bb_applet_long_options = long_options;
-       mode = bb_getopt_ulflags(argc, argv,
-#ifdef CONFIG_FEATURE_IPCALC_FANCY
-                                                         "mbnphs");
+#if ENABLE_FEATURE_IPCALC_LONG_OPTIONS
+       static const char ipcalc_longopts[] ALIGN1 =
+               "netmask\0"   No_argument "m" // netmask from IP (assuming complete class A, B, or C network)
+               "broadcast\0" No_argument "b" // broadcast from IP [netmask]
+               "network\0"   No_argument "n" // network from IP [netmask]
+# if ENABLE_FEATURE_IPCALC_FANCY
+               "prefix\0"    No_argument "p" // prefix from IP[/prefix] [netmask]
+               "hostname\0"  No_argument "h" // hostname from IP
+               "silent\0"    No_argument "s" // don’t ever display error messages
+# endif
+               ;
+# define GETOPT32 getopt32long
+# define LONGOPTS ,ipcalc_longopts
 #else
-                                                         "mbn");
+# define GETOPT32 getopt32
+# define LONGOPTS
 #endif
-       if (mode & (BROADCAST | NETWORK | NETPREFIX)) {
-               if (argc - optind > 2) {
-                       bb_show_usage();
-               }
-       } else {
-               if (argc - optind != 1) {
+
+int ipcalc_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int ipcalc_main(int argc UNUSED_PARAM, char **argv)
+{
+       unsigned opt;
+       bool have_netmask = 0;
+       struct in_addr s_netmask, s_broadcast, s_network, s_ipaddr;
+       /* struct in_addr { in_addr_t s_addr; }  and  in_addr_t
+        * (which in turn is just a typedef to uint32_t)
+        * are essentially the same type. A few macros for less verbosity: */
+#define netmask   (s_netmask.s_addr)
+#define broadcast (s_broadcast.s_addr)
+#define network   (s_network.s_addr)
+#define ipaddr    (s_ipaddr.s_addr)
+       char *ipstr;
+
+       opt = GETOPT32(argv, "^"
+                       "mbn" IF_FEATURE_IPCALC_FANCY("phs")
+                       "\0" "-1:?2"/*min 1, max 2 args*/
+                       LONGOPTS
+       );
+       argv += optind;
+       if (opt & SILENT)
+               logmode = LOGMODE_NONE; /* suppress error_msg() output */
+       opt &= ~SILENT;
+       if (!(opt & (BROADCAST | NETWORK | NETPREFIX))) {
+               /* if no options at all or
+                * (no broadcast,network,prefix) and (two args)... */
+               if (!opt || argv[1])
                        bb_show_usage();
-               }
        }
 
-#ifdef CONFIG_FEATURE_IPCALC_FANCY
-        prefixstr = ipstr = argv[optind];
-
-        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[optind], &a);
-#endif
-
-       if (ipaddr == 0) {
-               IPCALC_MSG(bb_error_msg_and_die("bad IP address: %s", argv[optind]),
-                                  exit(EXIT_FAILURE));
+       ipstr = argv[0];
+       if (ENABLE_FEATURE_IPCALC_FANCY) {
+               unsigned long netprefix = 0;
+               char *prefixstr;
+
+               prefixstr = ipstr;
+
+               while (*prefixstr) {
+                       if (*prefixstr == '/') {
+                               *prefixstr++ = '\0';
+                               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 = a.s_addr;
-
-       if (argc - optind == 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));
-                }
 
-#endif
-               netmask = inet_aton(argv[optind + 1], &a);
-               if (netmask == 0) {
-               IPCALC_MSG(bb_error_msg_and_die("bad netmask: %s", argv[optind + 1]),
-                                  exit(EXIT_FAILURE));
+       if (inet_aton(ipstr, &s_ipaddr) == 0) {
+               bb_error_msg_and_die("bad IP address: %s", argv[0]);
        }
-               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 (argv[1]) {
+               if (ENABLE_FEATURE_IPCALC_FANCY && have_netmask) {
+                       bb_simple_error_msg_and_die("use prefix or netmask, not both");
+               }
+               if (inet_aton(argv[1], &s_netmask) == 0) {
+                       bb_error_msg_and_die("bad netmask: %s", argv[1]);
+               }
+       } else {
+               /* JHC - If the netmask wasn't provided then calculate it */
+               if (!ENABLE_FEATURE_IPCALC_FANCY || !have_netmask)
+                       netmask = get_netmask(ipaddr);
        }
 
-       if (mode & NETMASK) {
-               printf("NETMASK=%s\n", inet_ntoa((*(struct in_addr *) &netmask)));
+       if (opt & NETMASK) {
+               printf("NETMASK=%s\n", inet_ntoa(s_netmask));
        }
 
-       if (mode & BROADCAST) {
+       if (opt & BROADCAST) {
                broadcast = (ipaddr & netmask) | ~netmask;
-               printf("BROADCAST=%s\n", inet_ntoa((*(struct in_addr *) &broadcast)));
+               printf("BROADCAST=%s\n", inet_ntoa(s_broadcast));
        }
 
-       if (mode & NETWORK) {
+       if (opt & NETWORK) {
                network = ipaddr & netmask;
-               printf("NETWORK=%s\n", inet_ntoa((*(struct in_addr *) &network)));
+               printf("NETWORK=%s\n", inet_ntoa(s_network));
        }
 
-#ifdef CONFIG_FEATURE_IPCALC_FANCY
-        if (mode & NETPREFIX) {
-                printf("PREFIX=%i\n", get_prefix(netmask));
-        }
+       if (ENABLE_FEATURE_IPCALC_FANCY) {
+               if (opt & NETPREFIX) {
+                       printf("PREFIX=%i\n", get_prefix(netmask));
+               }
 
-       if (mode & HOSTNAME) {
-               struct hostent *hostinfo;
-               int x;
+               if (opt & HOSTNAME) {
+                       struct hostent *hostinfo;
 
-               hostinfo = gethostbyaddr((char *) &ipaddr, sizeof(ipaddr), AF_INET);
-               if (!hostinfo) {
-                       IPCALC_MSG(bb_herror_msg_and_die(
-                               "cannot find hostname for %s", argv[optind]),);
-                       exit(EXIT_FAILURE);
-               }
-               for (x = 0; hostinfo->h_name[x]; x++) {
-                       hostinfo->h_name[x] = tolower(hostinfo->h_name[x]);
-               }
+                       hostinfo = gethostbyaddr((char *) &ipaddr, sizeof(ipaddr), AF_INET);
+                       if (!hostinfo) {
+                               bb_herror_msg_and_die("can't find hostname for %s", argv[0]);
+                       }
+                       str_tolower(hostinfo->h_name);
 
-               printf("HOSTNAME=%s\n", 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:
-*/