From 4aca87515a5083ae0e31ce3177189fd43b6d05ac Mon Sep 17 00:00:00 2001 From: Andreas Baumann Date: Sat, 3 Jan 2015 13:58:15 +0100 Subject: patch to Vanilla Tomato 1.28 --- release/src/router/busybox/networking/arp.c | 515 ++++++++++++++++++++++++++++ 1 file changed, 515 insertions(+) create mode 100644 release/src/router/busybox/networking/arp.c (limited to 'release/src/router/busybox/networking/arp.c') diff --git a/release/src/router/busybox/networking/arp.c b/release/src/router/busybox/networking/arp.c new file mode 100644 index 00000000..278f2dc9 --- /dev/null +++ b/release/src/router/busybox/networking/arp.c @@ -0,0 +1,515 @@ +/* vi: set sw=4 ts=4: */ +/* + * arp.c - Manipulate the system ARP cache + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version + * 2 of the License, or (at your option) any later version. + * + * Author: Fred N. van Kempen, + * Busybox port: Paul van Gool + * + * modified for getopt32 by Arne Bernin + */ + +#include "libbb.h" +#include "inet_common.h" + +#include +#include +#include +#include +#include + +#define DEBUG 0 + +#define DFLT_AF "inet" +#define DFLT_HW "ether" + +enum { + ARP_OPT_A = (1 << 0), + ARP_OPT_p = (1 << 1), + ARP_OPT_H = (1 << 2), + ARP_OPT_t = (1 << 3), + ARP_OPT_i = (1 << 4), + ARP_OPT_a = (1 << 5), + ARP_OPT_d = (1 << 6), + ARP_OPT_n = (1 << 7), /* do not resolve addresses */ + ARP_OPT_D = (1 << 8), /* HW-address is devicename */ + ARP_OPT_s = (1 << 9), + ARP_OPT_v = (1 << 10) * DEBUG, /* debugging output flag */ +}; + +enum { + sockfd = 3, /* active socket descriptor */ +}; + +struct globals { + const struct aftype *ap; /* current address family */ + const struct hwtype *hw; /* current hardware type */ + const char *device; /* current device */ + smallint hw_set; /* flag if hw-type was set (-H) */ + +}; +#define G (*(struct globals*)&bb_common_bufsiz1) +#define ap (G.ap ) +#define hw (G.hw ) +#define device (G.device ) +#define hw_set (G.hw_set ) +#define INIT_G() do { \ + device = ""; \ +} while (0) + + +static const char options[] ALIGN1 = + "pub\0" + "priv\0" + "temp\0" + "trail\0" + "dontpub\0" + "auto\0" + "dev\0" + "netmask\0"; + +/* Delete an entry from the ARP cache. */ +/* Called only from main, once */ +static int arp_del(char **args) +{ + char *host; + struct arpreq req; + struct sockaddr sa; + int flags = 0; + int err; + + memset(&req, 0, sizeof(req)); + + /* Resolve the host name. */ + host = *args; + if (ap->input(host, &sa) < 0) { + bb_herror_msg_and_die("%s", host); + } + + /* If a host has more than one address, use the correct one! */ + memcpy(&req.arp_pa, &sa, sizeof(struct sockaddr)); + + if (hw_set) + req.arp_ha.sa_family = hw->type; + + req.arp_flags = ATF_PERM; + args++; + while (*args != NULL) { + switch (index_in_strings(options, *args)) { + case 0: /* "pub" */ + flags |= 1; + args++; + break; + case 1: /* "priv" */ + flags |= 2; + args++; + break; + case 2: /* "temp" */ + req.arp_flags &= ~ATF_PERM; + args++; + break; + case 3: /* "trail" */ + req.arp_flags |= ATF_USETRAILERS; + args++; + break; + case 4: /* "dontpub" */ +#ifdef HAVE_ATF_DONTPUB + req.arp_flags |= ATF_DONTPUB; +#else + bb_error_msg("feature ATF_DONTPUB is not supported"); +#endif + args++; + break; + case 5: /* "auto" */ +#ifdef HAVE_ATF_MAGIC + req.arp_flags |= ATF_MAGIC; +#else + bb_error_msg("feature ATF_MAGIC is not supported"); +#endif + args++; + break; + case 6: /* "dev" */ + if (*++args == NULL) + bb_show_usage(); + device = *args; + args++; + break; + case 7: /* "netmask" */ + if (*++args == NULL) + bb_show_usage(); + if (strcmp(*args, "255.255.255.255") != 0) { + host = *args; + if (ap->input(host, &sa) < 0) { + bb_herror_msg_and_die("%s", host); + } + memcpy(&req.arp_netmask, &sa, sizeof(struct sockaddr)); + req.arp_flags |= ATF_NETMASK; + } + args++; + break; + default: + bb_show_usage(); + break; + } + } + if (flags == 0) + flags = 3; + + strncpy(req.arp_dev, device, sizeof(req.arp_dev)); + + err = -1; + + /* Call the kernel. */ + if (flags & 2) { + if (option_mask32 & ARP_OPT_v) + bb_error_msg("SIOCDARP(nopub)"); + err = ioctl(sockfd, SIOCDARP, &req); + if (err < 0) { + if (errno == ENXIO) { + if (flags & 1) + goto nopub; + printf("No ARP entry for %s\n", host); + return -1; + } + bb_perror_msg_and_die("SIOCDARP(priv)"); + } + } + if ((flags & 1) && err) { + nopub: + req.arp_flags |= ATF_PUBL; + if (option_mask32 & ARP_OPT_v) + bb_error_msg("SIOCDARP(pub)"); + if (ioctl(sockfd, SIOCDARP, &req) < 0) { + if (errno == ENXIO) { + printf("No ARP entry for %s\n", host); + return -1; + } + bb_perror_msg_and_die("SIOCDARP(pub)"); + } + } + return 0; +} + +/* Get the hardware address to a specified interface name */ +static void arp_getdevhw(char *ifname, struct sockaddr *sa, + const struct hwtype *hwt) +{ + struct ifreq ifr; + const struct hwtype *xhw; + + strcpy(ifr.ifr_name, ifname); + ioctl_or_perror_and_die(sockfd, SIOCGIFHWADDR, &ifr, + "cant get HW-Address for '%s'", ifname); + if (hwt && (ifr.ifr_hwaddr.sa_family != hw->type)) { + bb_error_msg_and_die("protocol type mismatch"); + } + memcpy(sa, &(ifr.ifr_hwaddr), sizeof(struct sockaddr)); + + if (option_mask32 & ARP_OPT_v) { + xhw = get_hwntype(ifr.ifr_hwaddr.sa_family); + if (!xhw || !xhw->print) { + xhw = get_hwntype(-1); + } + bb_error_msg("device '%s' has HW address %s '%s'", + ifname, xhw->name, + xhw->print((unsigned char *) &ifr.ifr_hwaddr.sa_data)); + } +} + +/* Set an entry in the ARP cache. */ +/* Called only from main, once */ +static int arp_set(char **args) +{ + char *host; + struct arpreq req; + struct sockaddr sa; + int flags; + + memset(&req, 0, sizeof(req)); + + host = *args++; + if (ap->input(host, &sa) < 0) { + bb_herror_msg_and_die("%s", host); + } + /* If a host has more than one address, use the correct one! */ + memcpy(&req.arp_pa, &sa, sizeof(struct sockaddr)); + + /* Fetch the hardware address. */ + if (*args == NULL) { + bb_error_msg_and_die("need hardware address"); + } + if (option_mask32 & ARP_OPT_D) { + arp_getdevhw(*args++, &req.arp_ha, hw_set ? hw : NULL); + } else { + if (hw->input(*args++, &req.arp_ha) < 0) { + bb_error_msg_and_die("invalid hardware address"); + } + } + + /* Check out any modifiers. */ + flags = ATF_PERM | ATF_COM; + while (*args != NULL) { + switch (index_in_strings(options, *args)) { + case 0: /* "pub" */ + flags |= ATF_PUBL; + args++; + break; + case 1: /* "priv" */ + flags &= ~ATF_PUBL; + args++; + break; + case 2: /* "temp" */ + flags &= ~ATF_PERM; + args++; + break; + case 3: /* "trail" */ + flags |= ATF_USETRAILERS; + args++; + break; + case 4: /* "dontpub" */ +#ifdef HAVE_ATF_DONTPUB + flags |= ATF_DONTPUB; +#else + bb_error_msg("feature ATF_DONTPUB is not supported"); +#endif + args++; + break; + case 5: /* "auto" */ +#ifdef HAVE_ATF_MAGIC + flags |= ATF_MAGIC; +#else + bb_error_msg("feature ATF_MAGIC is not supported"); +#endif + args++; + break; + case 6: /* "dev" */ + if (*++args == NULL) + bb_show_usage(); + device = *args; + args++; + break; + case 7: /* "netmask" */ + if (*++args == NULL) + bb_show_usage(); + if (strcmp(*args, "255.255.255.255") != 0) { + host = *args; + if (ap->input(host, &sa) < 0) { + bb_herror_msg_and_die("%s", host); + } + memcpy(&req.arp_netmask, &sa, sizeof(struct sockaddr)); + flags |= ATF_NETMASK; + } + args++; + break; + default: + bb_show_usage(); + break; + } + } + + /* Fill in the remainder of the request. */ + req.arp_flags = flags; + + strncpy(req.arp_dev, device, sizeof(req.arp_dev)); + + /* Call the kernel. */ + if (option_mask32 & ARP_OPT_v) + bb_error_msg("SIOCSARP()"); + xioctl(sockfd, SIOCSARP, &req); + return 0; +} + + +/* Print the contents of an ARP request block. */ +static void +arp_disp(const char *name, char *ip, int type, int arp_flags, + char *hwa, char *mask, char *dev) +{ + static const int arp_masks[] = { + ATF_PERM, ATF_PUBL, +#ifdef HAVE_ATF_MAGIC + ATF_MAGIC, +#endif +#ifdef HAVE_ATF_DONTPUB + ATF_DONTPUB, +#endif + ATF_USETRAILERS, + }; + static const char arp_labels[] ALIGN1 = "PERM\0""PUP\0" +#ifdef HAVE_ATF_MAGIC + "AUTO\0" +#endif +#ifdef HAVE_ATF_DONTPUB + "DONTPUB\0" +#endif + "TRAIL\0" + ; + + const struct hwtype *xhw; + + xhw = get_hwntype(type); + if (xhw == NULL) + xhw = get_hwtype(DFLT_HW); + + printf("%s (%s) at ", name, ip); + + if (!(arp_flags & ATF_COM)) { + if (arp_flags & ATF_PUBL) + printf("* "); + else + printf(" "); + } else { + printf("%s [%s] ", hwa, xhw->name); + } + + if (arp_flags & ATF_NETMASK) + printf("netmask %s ", mask); + + print_flags_separated(arp_masks, arp_labels, arp_flags, " "); + printf(" on %s\n", dev); +} + +/* Display the contents of the ARP cache in the kernel. */ +/* Called only from main, once */ +static int arp_show(char *name) +{ + const char *host; + const char *hostname; + FILE *fp; + struct sockaddr sa; + int type, flags; + int num; + unsigned entries = 0, shown = 0; + char ip[128]; + char hwa[128]; + char mask[128]; + char line[128]; + char dev[128]; + + host = NULL; + if (name != NULL) { + /* Resolve the host name. */ + if (ap->input(name, &sa) < 0) { + bb_herror_msg_and_die("%s", name); + } + host = xstrdup(ap->sprint(&sa, 1)); + } + fp = xfopen_for_read("/proc/net/arp"); + /* Bypass header -- read one line */ + fgets(line, sizeof(line), fp); + + /* Read the ARP cache entries. */ + while (fgets(line, sizeof(line), fp)) { + + mask[0] = '-'; mask[1] = '\0'; + dev[0] = '-'; dev[1] = '\0'; + /* All these strings can't overflow + * because fgets above reads limited amount of data */ + num = sscanf(line, "%s 0x%x 0x%x %s %s %s\n", + ip, &type, &flags, hwa, mask, dev); + if (num < 4) + break; + + entries++; + /* if the user specified hw-type differs, skip it */ + if (hw_set && (type != hw->type)) + continue; + + /* if the user specified address differs, skip it */ + if (host && strcmp(ip, host) != 0) + continue; + + /* if the user specified device differs, skip it */ + if (device[0] && strcmp(dev, device) != 0) + continue; + + shown++; + /* This IS ugly but it works -be */ + hostname = "?"; + if (!(option_mask32 & ARP_OPT_n)) { + if (ap->input(ip, &sa) < 0) + hostname = ip; + else + hostname = ap->sprint(&sa, (option_mask32 & ARP_OPT_n) | 0x8000); + if (strcmp(hostname, ip) == 0) + hostname = "?"; + } + + arp_disp(hostname, ip, type, flags, hwa, mask, dev); + } + if (option_mask32 & ARP_OPT_v) + printf("Entries: %d\tSkipped: %d\tFound: %d\n", + entries, entries - shown, shown); + + if (!shown) { + if (hw_set || host || device[0]) + printf("No match found in %d entries\n", entries); + } + if (ENABLE_FEATURE_CLEAN_UP) { + free((char*)host); + fclose(fp); + } + return 0; +} + +int arp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; +int arp_main(int argc UNUSED_PARAM, char **argv) +{ + const char *hw_type = "ether"; + const char *protocol; + unsigned opts; + + INIT_G(); + + xmove_fd(xsocket(AF_INET, SOCK_DGRAM, 0), sockfd); + ap = get_aftype(DFLT_AF); + if (!ap) + bb_error_msg_and_die("%s: %s not supported", DFLT_AF, "address family"); + + opts = getopt32(argv, "A:p:H:t:i:adnDsv", &protocol, &protocol, + &hw_type, &hw_type, &device); + argv += optind; + if (opts & (ARP_OPT_A | ARP_OPT_p)) { + ap = get_aftype(protocol); + if (ap == NULL) + bb_error_msg_and_die("%s: unknown %s", protocol, "address family"); + } + if (opts & (ARP_OPT_A | ARP_OPT_p)) { + hw = get_hwtype(hw_type); + if (hw == NULL) + bb_error_msg_and_die("%s: unknown %s", hw_type, "hardware type"); + hw_set = 1; + } + //if (opts & ARP_OPT_i)... -i + + if (ap->af != AF_INET) { + bb_error_msg_and_die("%s: kernel only supports 'inet'", ap->name); + } + + /* If no hw type specified get default */ + if (!hw) { + hw = get_hwtype(DFLT_HW); + if (!hw) + bb_error_msg_and_die("%s: %s not supported", DFLT_HW, "hardware type"); + } + + if (hw->alen <= 0) { + bb_error_msg_and_die("%s: %s without ARP support", + hw->name, "hardware type"); + } + + /* Now see what we have to do here... */ + if (opts & (ARP_OPT_d | ARP_OPT_s)) { + if (argv[0] == NULL) + bb_error_msg_and_die("need host name"); + if (opts & ARP_OPT_s) + return arp_set(argv); + return arp_del(argv); + } + //if (opts & ARP_OPT_a) - default + return arp_show(argv[0]); +} -- cgit v1.2.3-54-g00ecf