busybox/networking/ipcalc.c

238 lines
6.7 KiB
C
Raw Normal View History

/* 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 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.
*/
//config:config IPCALC
//config: bool "ipcalc"
//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(ipcalc, BB_DIR_BIN, BB_SUID_DROP))
//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 IP network settings from a IP address\n"
//usage: IF_FEATURE_IPCALC_LONG_OPTIONS(
//usage: "\n -b,--broadcast Display calculated broadcast address"
//usage: "\n -n,--network Display calculated network address"
//usage: "\n -m,--netmask Display default netmask for IP"
//usage: IF_FEATURE_IPCALC_FANCY(
//usage: "\n -p,--prefix Display the prefix for IP/NETMASK"
//usage: "\n -h,--hostname Display first resolved host name"
//usage: "\n -s,--silent Don't ever display error messages"
//usage: )
//usage: )
//usage: IF_NOT_FEATURE_IPCALC_LONG_OPTIONS(
//usage: "\n -b Display calculated broadcast address"
//usage: "\n -n Display calculated network address"
//usage: "\n -m Display default netmask for IP"
//usage: IF_FEATURE_IPCALC_FANCY(
//usage: "\n -p Display the prefix for IP/NETMASK"
//usage: "\n -h Display first resolved host name"
//usage: "\n -s Don't ever display error messages"
//usage: )
//usage: )
#include "libbb.h"
/* After libbb.h, because on some systems it needs other includes */
#include <arpa/inet.h>
#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)
{
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;
}
2007-08-13 16:06:25 +05:30
#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 ret;
}
2006-06-03 00:33:01 +05:30
#else
int get_prefix(unsigned long netmask);
#endif
2006-10-08 18:19:22 +05:30
#define NETMASK 0x01
#define BROADCAST 0x02
#define NETWORK 0x04
#define NETPREFIX 0x08
#define HOSTNAME 0x10
2003-09-05 08:07:15 +05:30
#define SILENT 0x20
#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" // dont ever display error messages
# endif
;
#endif
int ipcalc_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int ipcalc_main(int argc UNUSED_PARAM, char **argv)
2006-06-03 00:33:01 +05:30
{
2006-10-08 18:19:22 +05:30
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)
2006-06-03 00:33:01 +05:30
char *ipstr;
#if ENABLE_FEATURE_IPCALC_LONG_OPTIONS
2007-07-23 22:44:14 +05:30
applet_long_options = ipcalc_longopts;
#endif
opt_complementary = "-1:?2"; /* minimum 1 arg, maximum 2 args */
opt = getopt32(argv, "mbn" IF_FEATURE_IPCALC_FANCY("phs"));
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])
2003-03-19 14:43:01 +05:30
bb_show_usage();
}
2006-06-03 00:33:01 +05:30
ipstr = argv[0];
if (ENABLE_FEATURE_IPCALC_FANCY) {
unsigned long netprefix = 0;
char *prefixstr;
prefixstr = ipstr;
2006-10-08 18:19:22 +05:30
while (*prefixstr) {
2006-06-03 00:33:01 +05:30
if (*prefixstr == '/') {
*prefixstr++ = '\0';
2006-06-03 00:33:01 +05:30
if (*prefixstr) {
2006-10-08 18:19:22 +05:30
unsigned msk;
netprefix = xatoul_range(prefixstr, 0, 32);
2006-06-03 00:33:01 +05:30
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;
}
2006-06-03 00:33:01 +05:30
break;
}
2006-06-03 00:33:01 +05:30
prefixstr++;
}
}
if (inet_aton(ipstr, &s_ipaddr) == 0) {
2006-10-08 18:19:22 +05:30
bb_error_msg_and_die("bad IP address: %s", argv[0]);
}
if (argv[1]) {
2006-06-03 00:33:01 +05:30
if (ENABLE_FEATURE_IPCALC_FANCY && have_netmask) {
2006-10-08 18:19:22 +05:30
bb_error_msg_and_die("use prefix or netmask, not both");
}
if (inet_aton(argv[1], &s_netmask) == 0) {
2006-10-08 18:19:22 +05:30
bb_error_msg_and_die("bad netmask: %s", argv[1]);
}
} else {
2006-06-03 00:33:01 +05:30
/* JHC - If the netmask wasn't provided then calculate it */
if (!ENABLE_FEATURE_IPCALC_FANCY || !have_netmask)
netmask = get_netmask(ipaddr);
}
2006-10-08 18:19:22 +05:30
if (opt & NETMASK) {
printf("NETMASK=%s\n", inet_ntoa(s_netmask));
}
2006-10-08 18:19:22 +05:30
if (opt & BROADCAST) {
broadcast = (ipaddr & netmask) | ~netmask;
printf("BROADCAST=%s\n", inet_ntoa(s_broadcast));
}
2006-10-08 18:19:22 +05:30
if (opt & NETWORK) {
network = ipaddr & netmask;
printf("NETWORK=%s\n", inet_ntoa(s_network));
}
2006-06-03 00:33:01 +05:30
if (ENABLE_FEATURE_IPCALC_FANCY) {
2006-10-08 18:19:22 +05:30
if (opt & NETPREFIX) {
2006-06-03 00:33:01 +05:30
printf("PREFIX=%i\n", get_prefix(netmask));
}
2006-10-08 18:19:22 +05:30
if (opt & HOSTNAME) {
2006-06-03 00:33:01 +05:30
struct hostent *hostinfo;
2006-06-03 00:33:01 +05:30
hostinfo = gethostbyaddr((char *) &ipaddr, sizeof(ipaddr), AF_INET);
if (!hostinfo) {
bb_herror_msg_and_die("can't find hostname for %s", argv[0]);
2006-06-03 00:33:01 +05:30
}
str_tolower(hostinfo->h_name);
2006-06-03 00:33:01 +05:30
printf("HOSTNAME=%s\n", hostinfo->h_name);
}
}
return EXIT_SUCCESS;
}