hush/networking/arp.c
Denis Vlasenko fb79a2e2cf xioctl and friends by Tito <farmatito@tiscali.it>
function                                             old     new   delta
do_iptunnel                                          203     977    +774
process_dev                                         5328    5494    +166
ioctl_or_perror                                        -      54     +54
ioctl_or_perror_and_die                                -      51     +51
ioctl_alt_func                                         -      49     +49
bb_ioctl_or_warn                                       -      47     +47
do_add_ioctl                                         102     145     +43
bb_xioctl                                              -      39     +39
print_value_on_off                                     -      31     +31
get_lcm                                              105     123     +18
arp_main                                            2155    2167     +12
..................
zcip_main                                           1576    1566     -10
setlogcons_main                                       92      82     -10
dumpkmap_main                                        263     253     -10
do_get_ioctl                                          85      75     -10
setkeycodes_main                                     165     154     -11
write_table                                          244     232     -12
vconfig_main                                         318     306     -12
do_del_ioctl                                          93      81     -12
set_address                                           75      62     -13
maybe_set_utc                                         30      16     -14
loadfont_main                                        495     479     -16
slattach_main                                        712     695     -17
do_loadfont                                          191     174     -17
do_iplink                                           1155    1136     -19
getty_main                                          2583    2562     -21
fbset_main                                          2058    2035     -23
do_time                                              588     565     -23
xioctl                                                25       -     -25
read_rtc                                             186     160     -26
parse_conf                                          1299    1270     -29
udhcp_read_interface                                 269     239     -30
bb_ioctl                                              45       -     -45
bb_ioctl_alt                                          70       -     -70
bb_ioctl_on_off                                       78       -     -78
.rodata                                           129370  129018    -352
do_show                                              799       -    -799
------------------------------------------------------------------------------
(add/remove: 6/5 grow/shrink: 13/49 up/down: 1316/-1864)     Total: -548 bytes
   text	   data	    bss	    dec	    hex	filename
 675352	   2740	  13968	 692060	  a8f5c	busybox_old
 674804	   2740	  13968	 691512	  a8d38	busybox_unstripped
2007-07-14 22:07:14 +00:00

494 lines
11 KiB
C

/* 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, <waltje at uwalt.nl.mugnet.org>
* Busybox port: Paul van Gool <pvangool at mimotech.com>
*
* modified for getopt32 by Arne Bernin <arne [at] alamut.de>
*/
#include "libbb.h"
#include "inet_common.h"
#include <arpa/inet.h>
#include <net/if.h>
#include <net/if_arp.h>
#include <netinet/ether.h>
#include <netpacket/packet.h>
#define DEBUG 0
#define DFLT_AF "inet"
#define DFLT_HW "ether"
#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 const 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;
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_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) {
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((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_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) {
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)
{
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("<incomplete> ");
} 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)
{
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("/proc/net/arp", "r");
/* 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);
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]);
}