blob: 63ce99ffef9510af1ef45e79f352912f8fce9cbd [file] [log] [blame]
"Robert P. J. Day"63fc1a92006-07-02 19:47:05 +00001/* vi: set sw=4 ts=4: */
Glenn L McGrathc11986d2002-11-10 23:42:27 +00002/*
3 * Mini ipcalc implementation for busybox
4 *
5 * By Jordan Crouse <jordan@cosmicpenguin.net>
6 * Stephan Linz <linz@li-pro.net>
7 *
Eric Andersenaff114c2004-04-14 17:51:38 +00008 * This is a complete reimplementation of the ipcalc program
9 * from Red Hat. I didn't look at their source code, but there
Glenn L McGrathc11986d2002-11-10 23:42:27 +000010 * is no denying that this is a loving reimplementation
Bernhard Reutner-Fischerb1629b12006-05-19 19:29:19 +000011 *
12 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
Glenn L McGrathc11986d2002-11-10 23:42:27 +000013 */
14
Rob Landleyff97ee92006-06-02 19:03:01 +000015#include "busybox.h"
Glenn L McGrathc11986d2002-11-10 23:42:27 +000016#include <ctype.h>
17#include <getopt.h>
18#include <sys/socket.h>
Glenn L McGrathc11986d2002-11-10 23:42:27 +000019#include <arpa/inet.h>
20
Eric Andersen7207b882003-07-05 07:59:30 +000021#define CLASS_A_NETMASK ntohl(0xFF000000)
22#define CLASS_B_NETMASK ntohl(0xFFFF0000)
23#define CLASS_C_NETMASK ntohl(0xFFFFFF00)
24
Glenn L McGrathc11986d2002-11-10 23:42:27 +000025static unsigned long get_netmask(unsigned long ipaddr)
26{
Eric Andersen7207b882003-07-05 07:59:30 +000027 ipaddr = htonl(ipaddr);
28
29 if ((ipaddr & 0xC0000000) == 0xC0000000)
30 return CLASS_C_NETMASK;
31 else if ((ipaddr & 0x80000000) == 0x80000000)
32 return CLASS_B_NETMASK;
33 else if ((ipaddr & 0x80000000) == 0)
34 return CLASS_A_NETMASK;
35 else
36 return 0;
Glenn L McGrathc11986d2002-11-10 23:42:27 +000037}
38
Glenn L McGrath530ea422003-09-02 06:59:57 +000039#ifdef CONFIG_FEATURE_IPCALC_FANCY
40static int get_prefix(unsigned long netmask)
41{
Glenn L McGrath14092a12003-09-12 00:44:50 +000042 unsigned long msk = 0x80000000;
Eric Andersena4375042004-04-13 19:25:57 +000043 int ret = 0;
Glenn L McGrath530ea422003-09-02 06:59:57 +000044
Glenn L McGrath14092a12003-09-12 00:44:50 +000045 netmask = htonl(netmask);
Denis Vlasenkobf0a2012006-12-26 10:42:51 +000046 while (msk) {
Glenn L McGrath14092a12003-09-12 00:44:50 +000047 if (netmask & msk)
48 ret++;
49 msk >>= 1;
Eric Andersena4375042004-04-13 19:25:57 +000050 }
51 return ret;
Glenn L McGrath530ea422003-09-02 06:59:57 +000052}
Rob Landleyff97ee92006-06-02 19:03:01 +000053#else
54int get_prefix(unsigned long netmask);
Glenn L McGrath530ea422003-09-02 06:59:57 +000055#endif
56
Denis Vlasenko13858992006-10-08 12:49:22 +000057
Glenn L McGrathc11986d2002-11-10 23:42:27 +000058#define NETMASK 0x01
59#define BROADCAST 0x02
60#define NETWORK 0x04
Glenn L McGrath530ea422003-09-02 06:59:57 +000061#define NETPREFIX 0x08
62#define HOSTNAME 0x10
Glenn L McGrathd6bdd5d2003-09-05 02:37:15 +000063#define SILENT 0x20
Glenn L McGrathc11986d2002-11-10 23:42:27 +000064
Bernhard Reutner-Fischer01d23ad2006-05-26 20:19:22 +000065#if ENABLE_FEATURE_IPCALC_LONG_OPTIONS
Glenn L McGrath530ea422003-09-02 06:59:57 +000066 static const struct option long_options[] = {
Denis Vlasenkoc61852a2006-11-29 11:09:43 +000067 { "netmask", no_argument, NULL, 'm' },
68 { "broadcast", no_argument, NULL, 'b' },
69 { "network", no_argument, NULL, 'n' },
70# if ENABLE_FEATURE_IPCALC_FANCY
71 { "prefix", no_argument, NULL, 'p' },
72 { "hostname", no_argument, NULL, 'h' },
73 { "silent", no_argument, NULL, 's' },
74# endif
75 { NULL, 0, NULL, 0 }
Glenn L McGrathc11986d2002-11-10 23:42:27 +000076 };
Glenn L McGrathc11986d2002-11-10 23:42:27 +000077#endif
Denis Vlasenkoc61852a2006-11-29 11:09:43 +000078
Denis Vlasenko06af2162007-02-03 17:28:39 +000079int ipcalc_main(int argc, char **argv);
Rob Landleyff97ee92006-06-02 19:03:01 +000080int ipcalc_main(int argc, char **argv)
81{
Denis Vlasenko13858992006-10-08 12:49:22 +000082 unsigned opt;
Rob Landleyff97ee92006-06-02 19:03:01 +000083 int have_netmask = 0;
84 in_addr_t netmask, broadcast, network, ipaddr;
85 struct in_addr a;
86 char *ipstr;
87
Denis Vlasenkoc61852a2006-11-29 11:09:43 +000088#if ENABLE_FEATURE_IPCALC_LONG_OPTIONS
89 applet_long_options = long_options;
90#endif
Denis Vlasenko13858992006-10-08 12:49:22 +000091 opt = getopt32(argc, argv, "mbn" USE_FEATURE_IPCALC_FANCY("phs"));
Mike Frysingerb7f88e02005-09-24 23:48:18 +000092 argc -= optind;
93 argv += optind;
Denis Vlasenko13858992006-10-08 12:49:22 +000094 if (opt & (BROADCAST | NETWORK | NETPREFIX)) {
Mike Frysingerb7f88e02005-09-24 23:48:18 +000095 if (argc > 2 || argc <= 0)
Manuel Novoa III cad53642003-03-19 09:13:01 +000096 bb_show_usage();
Glenn L McGrathc11986d2002-11-10 23:42:27 +000097 } else {
Mike Frysingerb7f88e02005-09-24 23:48:18 +000098 if (argc != 1)
Manuel Novoa III cad53642003-03-19 09:13:01 +000099 bb_show_usage();
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000100 }
Denis Vlasenko13858992006-10-08 12:49:22 +0000101 if (opt & SILENT)
102 logmode = LOGMODE_NONE; /* Suppress error_msg() output */
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000103
Rob Landleyff97ee92006-06-02 19:03:01 +0000104 ipstr = argv[0];
105 if (ENABLE_FEATURE_IPCALC_FANCY) {
106 unsigned long netprefix = 0;
107 char *prefixstr;
Glenn L McGrath530ea422003-09-02 06:59:57 +0000108
Rob Landleyff97ee92006-06-02 19:03:01 +0000109 prefixstr = ipstr;
Glenn L McGrath14092a12003-09-12 00:44:50 +0000110
Denis Vlasenko13858992006-10-08 12:49:22 +0000111 while (*prefixstr) {
Rob Landleyff97ee92006-06-02 19:03:01 +0000112 if (*prefixstr == '/') {
113 *prefixstr = (char)0;
114 prefixstr++;
115 if (*prefixstr) {
Denis Vlasenko13858992006-10-08 12:49:22 +0000116 unsigned msk;
117 netprefix = xatoul_range(prefixstr, 0, 32);
Rob Landleyff97ee92006-06-02 19:03:01 +0000118 netmask = 0;
119 msk = 0x80000000;
120 while (netprefix > 0) {
121 netmask |= msk;
122 msk >>= 1;
123 netprefix--;
124 }
125 netmask = htonl(netmask);
126 /* Even if it was 0, we will signify that we have a netmask. This allows */
127 /* for specification of default routes, etc which have a 0 netmask/prefix */
128 have_netmask = 1;
Eric Andersena4375042004-04-13 19:25:57 +0000129 }
Rob Landleyff97ee92006-06-02 19:03:01 +0000130 break;
Eric Andersena4375042004-04-13 19:25:57 +0000131 }
Rob Landleyff97ee92006-06-02 19:03:01 +0000132 prefixstr++;
Eric Andersena4375042004-04-13 19:25:57 +0000133 }
Eric Andersena4375042004-04-13 19:25:57 +0000134 }
Glenn L McGrath14092a12003-09-12 00:44:50 +0000135 ipaddr = inet_aton(ipstr, &a);
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000136
Glenn L McGrath14092a12003-09-12 00:44:50 +0000137 if (ipaddr == 0) {
Denis Vlasenko13858992006-10-08 12:49:22 +0000138 bb_error_msg_and_die("bad IP address: %s", argv[0]);
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000139 }
Glenn L McGrath14092a12003-09-12 00:44:50 +0000140 ipaddr = a.s_addr;
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000141
Mike Frysingerb7f88e02005-09-24 23:48:18 +0000142 if (argc == 2) {
Rob Landleyff97ee92006-06-02 19:03:01 +0000143 if (ENABLE_FEATURE_IPCALC_FANCY && have_netmask) {
Denis Vlasenko13858992006-10-08 12:49:22 +0000144 bb_error_msg_and_die("use prefix or netmask, not both");
Eric Andersena4375042004-04-13 19:25:57 +0000145 }
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000146
Mike Frysingerb7f88e02005-09-24 23:48:18 +0000147 netmask = inet_aton(argv[1], &a);
Glenn L McGrath14092a12003-09-12 00:44:50 +0000148 if (netmask == 0) {
Denis Vlasenko13858992006-10-08 12:49:22 +0000149 bb_error_msg_and_die("bad netmask: %s", argv[1]);
Eric Andersena4375042004-04-13 19:25:57 +0000150 }
Glenn L McGrath14092a12003-09-12 00:44:50 +0000151 netmask = a.s_addr;
152 } else {
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000153
Rob Landleyff97ee92006-06-02 19:03:01 +0000154 /* JHC - If the netmask wasn't provided then calculate it */
155 if (!ENABLE_FEATURE_IPCALC_FANCY || !have_netmask)
Eric Andersena4375042004-04-13 19:25:57 +0000156 netmask = get_netmask(ipaddr);
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000157 }
158
Denis Vlasenko13858992006-10-08 12:49:22 +0000159 if (opt & NETMASK) {
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000160 printf("NETMASK=%s\n", inet_ntoa((*(struct in_addr *) &netmask)));
161 }
162
Denis Vlasenko13858992006-10-08 12:49:22 +0000163 if (opt & BROADCAST) {
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000164 broadcast = (ipaddr & netmask) | ~netmask;
165 printf("BROADCAST=%s\n", inet_ntoa((*(struct in_addr *) &broadcast)));
166 }
167
Denis Vlasenko13858992006-10-08 12:49:22 +0000168 if (opt & NETWORK) {
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000169 network = ipaddr & netmask;
170 printf("NETWORK=%s\n", inet_ntoa((*(struct in_addr *) &network)));
171 }
Glenn L McGrath530ea422003-09-02 06:59:57 +0000172
Rob Landleyff97ee92006-06-02 19:03:01 +0000173 if (ENABLE_FEATURE_IPCALC_FANCY) {
Denis Vlasenko13858992006-10-08 12:49:22 +0000174 if (opt & NETPREFIX) {
Rob Landleyff97ee92006-06-02 19:03:01 +0000175 printf("PREFIX=%i\n", get_prefix(netmask));
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000176 }
177
Denis Vlasenko13858992006-10-08 12:49:22 +0000178 if (opt & HOSTNAME) {
Rob Landleyff97ee92006-06-02 19:03:01 +0000179 struct hostent *hostinfo;
180 int x;
181
182 hostinfo = gethostbyaddr((char *) &ipaddr, sizeof(ipaddr), AF_INET);
183 if (!hostinfo) {
Denis Vlasenko13858992006-10-08 12:49:22 +0000184 bb_herror_msg_and_die("cannot find hostname for %s", argv[0]);
Rob Landleyff97ee92006-06-02 19:03:01 +0000185 }
186 for (x = 0; hostinfo->h_name[x]; x++) {
187 hostinfo->h_name[x] = tolower(hostinfo->h_name[x]);
188 }
189
190 printf("HOSTNAME=%s\n", hostinfo->h_name);
191 }
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000192 }
Glenn L McGrathc11986d2002-11-10 23:42:27 +0000193
194 return EXIT_SUCCESS;
195}