From: Denis Vlasenko Date: Sun, 7 Jan 2007 19:35:11 +0000 (-0000) Subject: I *always* forgotting svn add X-Git-Tag: 1_4_0~89 X-Git-Url: https://git.librecmc.org/?a=commitdiff_plain;h=88e2b1cb626761b1924305b761a5dfc723613c4e;p=oweals%2Fbusybox.git I *always* forgotting svn add --- diff --git a/networking/arp.c b/networking/arp.c new file mode 100644 index 000000000..10934c254 --- /dev/null +++ b/networking/arp.c @@ -0,0 +1,500 @@ +/* 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 "busybox.h" +#include "inet_common.h" + +#include +#include +#include +#include +#include + +#define DEBUG 0 + +#define DFLT_AF "inet" +#define DFLT_HW "ether" + +#define _PATH_PROCNET_ARP "/proc/net/arp" + +#define ARP_OPT_A (0x1) +#define ARP_OPT_p (0x2) +#define ARP_OPT_H (0x4) +#define ARP_OPT_t (0x8) +#define ARP_OPT_i (0x10) +#define ARP_OPT_a (0x20) +#define ARP_OPT_d (0x40) +#define ARP_OPT_n (0x80) /* do not resolve addresses */ +#define ARP_OPT_D (0x100) /* HW-address is devicename */ +#define ARP_OPT_s (0x200) +#define ARP_OPT_v (0x400 * DEBUG) /* debugging output flag */ + + +static const struct aftype *ap; /* current address family */ +static const struct hwtype *hw; /* current hardware type */ +static int sockfd; /* active socket descriptor */ +static smallint hw_set; /* flag if hw-type was set (-H) */ +static char *device = ""; /* current device */ + +static const char *const options[] = { + "pub", + "priv", + "temp", + "trail", + "dontpub", + "auto", + "dev", + "netmask", + NULL +}; + +/* Delete an entry from the ARP cache. */ +/* Called only from main, once */ +static int arp_del(char **args) +{ + char host[128]; + struct arpreq req; + struct sockaddr sa; + int flags = 0; + int err; + + memset(&req, 0, sizeof(req)); + + /* Resolve the host name. */ + safe_strncpy(host, *args, 128); + if (ap->input(0, 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) { + if (option_mask32 & ARP_OPT_v) + bb_error_msg("args=%s", *args); + + switch (index_in_str_array(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) { + safe_strncpy(host, *args, 128); + if (ap->input(0, host, &sa) < 0) { + bb_herror_msg("%s", host); + return -1; + } + 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); + if (ioctl(sockfd, SIOCGIFHWADDR, &ifr) < 0) { + bb_perror_msg_and_die("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((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[128]; + struct arpreq req; + struct sockaddr sa; + int flags; + + memset(&req, 0, sizeof(req)); + + safe_strncpy(host, *args++, 128); + if (ap->input(0, 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_str_array(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) { + safe_strncpy(host, *args++, 128); + if (ap->input(0, 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()"); + if (ioctl(sockfd, SIOCSARP, &req) < 0) { + bb_perror_msg_and_die("SIOCSARP"); + } + return 0; +} + + +/* Print the contents of an ARP request block. */ +static void +arp_disp(char *name, char *ip, int type, int arp_flags, + char *hwa, char *mask, char *dev) +{ + 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); + + if (arp_flags & ATF_PERM) + printf("PERM "); + if (arp_flags & ATF_PUBL) + printf("PUP "); +#ifdef HAVE_ATF_MAGIC + if (arp_flags & ATF_MAGIC) + printf("AUTO "); +#endif +#ifdef HAVE_ATF_DONTPUB + if (arp_flags & ATF_DONTPUB) + printf("DONTPUB "); +#endif + if (arp_flags & ATF_USETRAILERS) + printf("TRAIL "); + + 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) +{ + char host[100]; + struct sockaddr sa; + char ip[100]; + char hwa[100]; + char mask[100]; + char line[200]; + char dev[100]; + int type, flags; + FILE *fp; + char *hostname; + int num, entries = 0, shown = 0; + + host[0] = '\0'; + + if (name != NULL) { + /* Resolve the host name. */ + safe_strncpy(host, name, (sizeof host)); + if (ap->input(0, host, &sa) < 0) { + bb_herror_msg_and_die("%s", host); + } + safe_strncpy(host, ap->sprint(&sa, 1), sizeof(host)); + } + /* Open the PROCps kernel table. */ + fp = xfopen(_PATH_PROCNET_ARP, "r"); + /* Bypass header -- read until newline */ + if (fgets(line, sizeof(line), fp) != (char *) NULL) { + mask[0] = '-'; mask[1] = '\0'; + dev[0] = '-'; dev[1] = '\0'; + /* Read the ARP cache entries. */ + for (; fgets(line, sizeof(line), fp);) { + num = sscanf(line, "%s 0x%x 0x%x %100s %100s %100s\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[0] && strcmp(ip, host) != 0) + continue; + + /* if the user specified device differs, skip it */ + if (device && strcmp(dev, device) != 0) + continue; + + shown++; + /* This IS ugly but it works -be */ + if (option_mask32 & ARP_OPT_n) + hostname = "?"; + else { + if (ap->input(0, 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[0] || device) + printf("No match found in %d entries\n", entries); + } + + fclose(fp); + + return 0; +} + +int arp_main(int argc, char **argv) +{ + char *hw_type; + char *protocol; + + /* Initialize variables... */ + ap = get_aftype(DFLT_AF); + if (!ap) + bb_error_msg_and_die("%s: %s not supported", DFLT_AF, "address family"); + + getopt32(argc, argv, "A:p:H:t:i:adnDsv", &protocol, &protocol, + &hw_type, &hw_type, &device); + argv += optind; + if (option_mask32 & ARP_OPT_A || option_mask32 & ARP_OPT_p) { + ap = get_aftype(protocol); + if (ap == NULL) + bb_error_msg_and_die("%s: unknown %s", protocol, "address family"); + } + if (option_mask32 & ARP_OPT_A || option_mask32 & 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 (option_mask32 & 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"); + } + sockfd = xsocket(AF_INET, SOCK_DGRAM, 0); + + /* Now see what we have to do here... */ + if (option_mask32 & (ARP_OPT_d|ARP_OPT_s)) { + if (argv[0] == NULL) + bb_error_msg_and_die("need host name"); + if (option_mask32 & ARP_OPT_s) + return arp_set(argv); + return arp_del(argv); + } + //if (option_mask32 & ARP_OPT_a) - default + return arp_show(argv[0]); +}