Blame misc/arpd.c

Packit Service 3880ab
/*
Packit Service 3880ab
 * arpd.c	ARP helper daemon.
Packit Service 3880ab
 *
Packit Service 3880ab
 *		This program is free software; you can redistribute it and/or
Packit Service 3880ab
 *		modify it under the terms of the GNU General Public License
Packit Service 3880ab
 *		as published by the Free Software Foundation; either version
Packit Service 3880ab
 *		2 of the License, or (at your option) any later version.
Packit Service 3880ab
 *
Packit Service 3880ab
 * Authors:	Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
Packit Service 3880ab
 */
Packit Service 3880ab
Packit Service 3880ab
#include <stdio.h>
Packit Service 3880ab
#include <syslog.h>
Packit Service 3880ab
#include <malloc.h>
Packit Service 3880ab
#include <string.h>
Packit Service 3880ab
#include <unistd.h>
Packit Service 3880ab
#include <stdlib.h>
Packit Service 3880ab
#include <netdb.h>
Packit Service 3880ab
#include <db_185.h>
Packit Service 3880ab
#include <sys/ioctl.h>
Packit Service 3880ab
#include <sys/poll.h>
Packit Service 3880ab
#include <errno.h>
Packit Service 3880ab
#include <fcntl.h>
Packit Service 3880ab
#include <sys/uio.h>
Packit Service 3880ab
#include <sys/socket.h>
Packit Service 3880ab
#include <sys/time.h>
Packit Service 3880ab
#include <time.h>
Packit Service 3880ab
#include <signal.h>
Packit Service 3880ab
#include <linux/if.h>
Packit Service 3880ab
#include <linux/if_ether.h>
Packit Service 3880ab
#include <linux/if_arp.h>
Packit Service 3880ab
#include <netinet/in.h>
Packit Service 3880ab
#include <arpa/inet.h>
Packit Service 3880ab
#include <linux/if_packet.h>
Packit Service 3880ab
#include <linux/filter.h>
Packit Service 3880ab
Packit Service 3880ab
#include "libnetlink.h"
Packit Service 3880ab
#include "utils.h"
Packit Service 3880ab
#include "rt_names.h"
Packit Service 3880ab
Packit Service 3880ab
DB	*dbase;
Packit Service 3880ab
char	*dbname = "/var/lib/arpd/arpd.db";
Packit Service 3880ab
Packit Service 3880ab
int	ifnum;
Packit Service 3880ab
int	*ifvec;
Packit Service 3880ab
char	**ifnames;
Packit Service 3880ab
Packit Service 3880ab
struct dbkey {
Packit Service 3880ab
	__u32	iface;
Packit Service 3880ab
	__u32	addr;
Packit Service 3880ab
};
Packit Service 3880ab
Packit Service 3880ab
#define IS_NEG(x)	(((__u8 *)(x))[0] == 0xFF)
Packit Service 3880ab
#define NEG_TIME(x)	(((x)[2]<<24)|((x)[3]<<16)|((x)[4]<<8)|(x)[5])
Packit Service 3880ab
#define NEG_AGE(x)	((__u32)time(NULL) - NEG_TIME((__u8 *)x))
Packit Service 3880ab
#define NEG_VALID(x)	(NEG_AGE(x) < negative_timeout)
Packit Service 3880ab
#define NEG_CNT(x)	(((__u8 *)(x))[1])
Packit Service 3880ab
Packit Service 3880ab
struct rtnl_handle rth;
Packit Service 3880ab
Packit Service 3880ab
struct pollfd pset[2];
Packit Service 3880ab
int udp_sock = -1;
Packit Service 3880ab
Packit Service 3880ab
volatile int do_exit;
Packit Service 3880ab
volatile int do_sync;
Packit Service 3880ab
volatile int do_stats;
Packit Service 3880ab
Packit Service 3880ab
struct {
Packit Service 3880ab
	unsigned long arp_new;
Packit Service 3880ab
	unsigned long arp_change;
Packit Service 3880ab
Packit Service 3880ab
	unsigned long app_recv;
Packit Service 3880ab
	unsigned long app_success;
Packit Service 3880ab
	unsigned long app_bad;
Packit Service 3880ab
	unsigned long app_neg;
Packit Service 3880ab
	unsigned long app_suppressed;
Packit Service 3880ab
Packit Service 3880ab
	unsigned long kern_neg;
Packit Service 3880ab
	unsigned long kern_new;
Packit Service 3880ab
	unsigned long kern_change;
Packit Service 3880ab
Packit Service 3880ab
	unsigned long probes_sent;
Packit Service 3880ab
	unsigned long probes_suppressed;
Packit Service 3880ab
} stats;
Packit Service 3880ab
Packit Service 3880ab
int active_probing;
Packit Service 3880ab
int negative_timeout = 60;
Packit Service 3880ab
int no_kernel_broadcasts;
Packit Service 3880ab
int broadcast_rate = 1000;
Packit Service 3880ab
int broadcast_burst = 3000;
Packit Service 3880ab
int poll_timeout = 30000;
Packit Service 3880ab
Packit Service 3880ab
static void usage(void)
Packit Service 3880ab
{
Packit Service 3880ab
	fprintf(stderr,
Packit Service 3880ab
		"Usage: arpd [ -lkh? ] [ -a N ] [ -b dbase ] [ -B number ] [ -f file ] [ -n time ] [-p interval ] [ -R rate ] [ interfaces ]\n");
Packit Service 3880ab
	exit(1);
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int handle_if(int ifindex)
Packit Service 3880ab
{
Packit Service 3880ab
	int i;
Packit Service 3880ab
Packit Service 3880ab
	if (ifnum == 0)
Packit Service 3880ab
		return 1;
Packit Service 3880ab
Packit Service 3880ab
	for (i = 0; i < ifnum; i++)
Packit Service 3880ab
		if (ifvec[i] == ifindex)
Packit Service 3880ab
			return 1;
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
int sysctl_adjusted;
Packit Service 3880ab
Packit Service 3880ab
static void do_sysctl_adjustments(void)
Packit Service 3880ab
{
Packit Service 3880ab
	int i;
Packit Service 3880ab
Packit Service 3880ab
	if (!ifnum)
Packit Service 3880ab
		return;
Packit Service 3880ab
Packit Service 3880ab
	for (i = 0; i < ifnum; i++) {
Packit Service 3880ab
		char buf[128];
Packit Service 3880ab
		FILE *fp;
Packit Service 3880ab
Packit Service 3880ab
		if (active_probing) {
Packit Service 3880ab
			sprintf(buf, "/proc/sys/net/ipv4/neigh/%s/mcast_solicit", ifnames[i]);
Packit Service 3880ab
			if ((fp = fopen(buf, "w")) != NULL) {
Packit Service 3880ab
				if (no_kernel_broadcasts)
Packit Service 3880ab
					strcpy(buf, "0\n");
Packit Service 3880ab
				else
Packit Service 3880ab
					sprintf(buf, "%d\n", active_probing >= 2 ? 1 : 3-active_probing);
Packit Service 3880ab
				fputs(buf, fp);
Packit Service 3880ab
				fclose(fp);
Packit Service 3880ab
			}
Packit Service 3880ab
		}
Packit Service 3880ab
Packit Service 3880ab
		sprintf(buf, "/proc/sys/net/ipv4/neigh/%s/app_solicit", ifnames[i]);
Packit Service 3880ab
		if ((fp = fopen(buf, "w")) != NULL) {
Packit Service 3880ab
			sprintf(buf, "%d\n", active_probing <= 1 ? 1 : active_probing);
Packit Service 3880ab
			fputs(buf, fp);
Packit Service 3880ab
			fclose(fp);
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
	sysctl_adjusted = 1;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static void undo_sysctl_adjustments(void)
Packit Service 3880ab
{
Packit Service 3880ab
	int i;
Packit Service 3880ab
Packit Service 3880ab
	if (!sysctl_adjusted)
Packit Service 3880ab
		return;
Packit Service 3880ab
Packit Service 3880ab
	for (i = 0; i < ifnum; i++) {
Packit Service 3880ab
		char buf[128];
Packit Service 3880ab
		FILE *fp;
Packit Service 3880ab
Packit Service 3880ab
		if (active_probing) {
Packit Service 3880ab
			sprintf(buf, "/proc/sys/net/ipv4/neigh/%s/mcast_solicit", ifnames[i]);
Packit Service 3880ab
			if ((fp = fopen(buf, "w")) != NULL) {
Packit Service 3880ab
				strcpy(buf, "3\n");
Packit Service 3880ab
				fputs(buf, fp);
Packit Service 3880ab
				fclose(fp);
Packit Service 3880ab
			}
Packit Service 3880ab
		}
Packit Service 3880ab
		sprintf(buf, "/proc/sys/net/ipv4/neigh/%s/app_solicit", ifnames[i]);
Packit Service 3880ab
		if ((fp = fopen(buf, "w")) != NULL) {
Packit Service 3880ab
			strcpy(buf, "0\n");
Packit Service 3880ab
			fputs(buf, fp);
Packit Service 3880ab
			fclose(fp);
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
	sysctl_adjusted = 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
Packit Service 3880ab
static int send_probe(int ifindex, __u32 addr)
Packit Service 3880ab
{
Packit Service 3880ab
	struct ifreq ifr = { .ifr_ifindex = ifindex };
Packit Service 3880ab
	struct sockaddr_in dst = {
Packit Service 3880ab
		.sin_family = AF_INET,
Packit Service 3880ab
		.sin_port = htons(1025),
Packit Service 3880ab
		.sin_addr.s_addr = addr,
Packit Service 3880ab
	};
Packit Service 3880ab
	socklen_t len;
Packit Service 3880ab
	unsigned char buf[256];
Packit Service 3880ab
	struct arphdr *ah = (struct arphdr *)buf;
Packit Service 3880ab
	unsigned char *p = (unsigned char *)(ah+1);
Packit Service 3880ab
	struct sockaddr_ll sll = {
Packit Service 3880ab
		.sll_family = AF_PACKET,
Packit Service 3880ab
		.sll_ifindex = ifindex,
Packit Service 3880ab
		.sll_protocol = htons(ETH_P_ARP),
Packit Service 3880ab
	};
Packit Service 3880ab
Packit Service 3880ab
	if (ioctl(udp_sock, SIOCGIFNAME, &ifr))
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	if (ioctl(udp_sock, SIOCGIFHWADDR, &ifr))
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	if (ifr.ifr_hwaddr.sa_family != ARPHRD_ETHER)
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	if (setsockopt(udp_sock, SOL_SOCKET, SO_BINDTODEVICE, ifr.ifr_name, strlen(ifr.ifr_name)+1) < 0)
Packit Service 3880ab
		return -1;
Packit Service 3880ab
Packit Service 3880ab
	if (connect(udp_sock, (struct sockaddr *)&dst, sizeof(dst)) < 0)
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	len = sizeof(dst);
Packit Service 3880ab
	if (getsockname(udp_sock, (struct sockaddr *)&dst, &len) < 0)
Packit Service 3880ab
		return -1;
Packit Service 3880ab
Packit Service 3880ab
	ah->ar_hrd = htons(ifr.ifr_hwaddr.sa_family);
Packit Service 3880ab
	ah->ar_pro = htons(ETH_P_IP);
Packit Service 3880ab
	ah->ar_hln = 6;
Packit Service 3880ab
	ah->ar_pln = 4;
Packit Service 3880ab
	ah->ar_op  = htons(ARPOP_REQUEST);
Packit Service 3880ab
Packit Service 3880ab
	memcpy(p, ifr.ifr_hwaddr.sa_data, ah->ar_hln);
Packit Service 3880ab
	p += ah->ar_hln;
Packit Service 3880ab
Packit Service 3880ab
	memcpy(p, &dst.sin_addr, 4);
Packit Service 3880ab
	p += 4;
Packit Service 3880ab
Packit Service 3880ab
	memset(sll.sll_addr, 0xFF, sizeof(sll.sll_addr));
Packit Service 3880ab
	memcpy(p, &sll.sll_addr, ah->ar_hln);
Packit Service 3880ab
	p += ah->ar_hln;
Packit Service 3880ab
Packit Service 3880ab
	memcpy(p, &addr, 4);
Packit Service 3880ab
	p += 4;
Packit Service 3880ab
Packit Service 3880ab
	if (sendto(pset[0].fd, buf, p-buf, 0, (struct sockaddr *)&sll, sizeof(sll)) < 0)
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	stats.probes_sent++;
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
/* Be very tough on sending probes: 1 per second with burst of 3. */
Packit Service 3880ab
Packit Service 3880ab
static int queue_active_probe(int ifindex, __u32 addr)
Packit Service 3880ab
{
Packit Service 3880ab
	static struct timeval prev;
Packit Service 3880ab
	static int buckets;
Packit Service 3880ab
	struct timeval now;
Packit Service 3880ab
Packit Service 3880ab
	gettimeofday(&now, NULL);
Packit Service 3880ab
	if (prev.tv_sec) {
Packit Service 3880ab
		int diff = (now.tv_sec-prev.tv_sec)*1000+(now.tv_usec-prev.tv_usec)/1000;
Packit Service 3880ab
Packit Service 3880ab
		buckets += diff;
Packit Service 3880ab
	} else {
Packit Service 3880ab
		buckets = broadcast_burst;
Packit Service 3880ab
	}
Packit Service 3880ab
	if (buckets > broadcast_burst)
Packit Service 3880ab
		buckets = broadcast_burst;
Packit Service 3880ab
	if (buckets >= broadcast_rate && !send_probe(ifindex, addr)) {
Packit Service 3880ab
		buckets -= broadcast_rate;
Packit Service 3880ab
		prev = now;
Packit Service 3880ab
		return 0;
Packit Service 3880ab
	}
Packit Service 3880ab
	stats.probes_suppressed++;
Packit Service 3880ab
	return -1;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int respond_to_kernel(int ifindex, __u32 addr, char *lla, int llalen)
Packit Service 3880ab
{
Packit Service 3880ab
	struct {
Packit Service 3880ab
		struct nlmsghdr	n;
Packit Service 3880ab
		struct ndmsg		ndm;
Packit Service 3880ab
		char			buf[256];
Packit Service 3880ab
	} req = {
Packit Service 3880ab
		.n.nlmsg_len = NLMSG_LENGTH(sizeof(struct ndmsg)),
Packit Service 3880ab
		.n.nlmsg_flags = NLM_F_REQUEST,
Packit Service 3880ab
		.n.nlmsg_type = RTM_NEWNEIGH,
Packit Service 3880ab
		.ndm.ndm_family = AF_INET,
Packit Service 3880ab
		.ndm.ndm_state = NUD_STALE,
Packit Service 3880ab
		.ndm.ndm_ifindex = ifindex,
Packit Service 3880ab
		.ndm.ndm_type = RTN_UNICAST,
Packit Service 3880ab
	};
Packit Service 3880ab
Packit Service 3880ab
	addattr_l(&req.n, sizeof(req), NDA_DST, &addr, 4);
Packit Service 3880ab
	addattr_l(&req.n, sizeof(req), NDA_LLADDR, lla, llalen);
Packit Service 3880ab
	return rtnl_send(&rth, &req, req.n.nlmsg_len) <= 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static void prepare_neg_entry(__u8 *ndata, __u32 stamp)
Packit Service 3880ab
{
Packit Service 3880ab
	ndata[0] = 0xFF;
Packit Service 3880ab
	ndata[1] = 0;
Packit Service 3880ab
	ndata[2] = stamp>>24;
Packit Service 3880ab
	ndata[3] = stamp>>16;
Packit Service 3880ab
	ndata[4] = stamp>>8;
Packit Service 3880ab
	ndata[5] = stamp;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
Packit Service 3880ab
static int do_one_request(struct nlmsghdr *n)
Packit Service 3880ab
{
Packit Service 3880ab
	struct ndmsg *ndm = NLMSG_DATA(n);
Packit Service 3880ab
	int len = n->nlmsg_len;
Packit Service 3880ab
	struct rtattr *tb[NDA_MAX+1];
Packit Service 3880ab
	struct dbkey key;
Packit Service 3880ab
	DBT dbkey, dbdat;
Packit Service 3880ab
	int do_acct = 0;
Packit Service 3880ab
Packit Service 3880ab
	if (n->nlmsg_type == NLMSG_DONE) {
Packit Service 3880ab
		dbase->sync(dbase, 0);
Packit Service 3880ab
Packit Service 3880ab
		/* Now we have at least mirror of kernel db, so that
Packit Service 3880ab
		 * may start real resolution.
Packit Service 3880ab
		 */
Packit Service 3880ab
		do_sysctl_adjustments();
Packit Service 3880ab
		return 0;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (n->nlmsg_type != RTM_GETNEIGH && n->nlmsg_type != RTM_NEWNEIGH)
Packit Service 3880ab
		return 0;
Packit Service 3880ab
Packit Service 3880ab
	len -= NLMSG_LENGTH(sizeof(*ndm));
Packit Service 3880ab
	if (len < 0)
Packit Service 3880ab
		return -1;
Packit Service 3880ab
Packit Service 3880ab
	if (ndm->ndm_family != AF_INET ||
Packit Service 3880ab
	    (ifnum && !handle_if(ndm->ndm_ifindex)) ||
Packit Service 3880ab
	    ndm->ndm_flags ||
Packit Service 3880ab
	    ndm->ndm_type != RTN_UNICAST ||
Packit Service 3880ab
	    !(ndm->ndm_state&~NUD_NOARP))
Packit Service 3880ab
		return 0;
Packit Service 3880ab
Packit Service 3880ab
	parse_rtattr(tb, NDA_MAX, NDA_RTA(ndm), len);
Packit Service 3880ab
Packit Service 3880ab
	if (!tb[NDA_DST])
Packit Service 3880ab
		return 0;
Packit Service 3880ab
Packit Service 3880ab
	key.iface = ndm->ndm_ifindex;
Packit Service 3880ab
	memcpy(&key.addr, RTA_DATA(tb[NDA_DST]), 4);
Packit Service 3880ab
	dbkey.data = &ke;;
Packit Service 3880ab
	dbkey.size = sizeof(key);
Packit Service 3880ab
Packit Service 3880ab
	if (dbase->get(dbase, &dbkey, &dbdat, 0) != 0) {
Packit Service 3880ab
		dbdat.data = 0;
Packit Service 3880ab
		dbdat.size = 0;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (n->nlmsg_type == RTM_GETNEIGH) {
Packit Service 3880ab
		if (!(n->nlmsg_flags&NLM_F_REQUEST))
Packit Service 3880ab
			return 0;
Packit Service 3880ab
Packit Service 3880ab
		if (!(ndm->ndm_state&(NUD_PROBE|NUD_INCOMPLETE))) {
Packit Service 3880ab
			stats.app_bad++;
Packit Service 3880ab
			return 0;
Packit Service 3880ab
		}
Packit Service 3880ab
Packit Service 3880ab
		if (ndm->ndm_state&NUD_PROBE) {
Packit Service 3880ab
			/* If we get this, kernel still has some valid
Packit Service 3880ab
			 * address, but unicast probing failed and host
Packit Service 3880ab
			 * is either dead or changed its mac address.
Packit Service 3880ab
			 * Kernel is going to initiate broadcast resolution.
Packit Service 3880ab
			 * OK, we invalidate our information as well.
Packit Service 3880ab
			 */
Packit Service 3880ab
			if (dbdat.data && !IS_NEG(dbdat.data))
Packit Service 3880ab
				stats.app_neg++;
Packit Service 3880ab
Packit Service 3880ab
			dbase->del(dbase, &dbkey, 0);
Packit Service 3880ab
		} else {
Packit Service 3880ab
			/* If we get this kernel does not have any information.
Packit Service 3880ab
			 * If we have something tell this to kernel. */
Packit Service 3880ab
			stats.app_recv++;
Packit Service 3880ab
			if (dbdat.data && !IS_NEG(dbdat.data)) {
Packit Service 3880ab
				stats.app_success++;
Packit Service 3880ab
				respond_to_kernel(key.iface, key.addr, dbdat.data, dbdat.size);
Packit Service 3880ab
				return 0;
Packit Service 3880ab
			}
Packit Service 3880ab
Packit Service 3880ab
			/* Sheeit! We have nothing to tell. */
Packit Service 3880ab
			/* If we have recent negative entry, be silent. */
Packit Service 3880ab
			if (dbdat.data && NEG_VALID(dbdat.data)) {
Packit Service 3880ab
				if (NEG_CNT(dbdat.data) >= active_probing) {
Packit Service 3880ab
					stats.app_suppressed++;
Packit Service 3880ab
					return 0;
Packit Service 3880ab
				}
Packit Service 3880ab
				do_acct = 1;
Packit Service 3880ab
			}
Packit Service 3880ab
		}
Packit Service 3880ab
Packit Service 3880ab
		if (active_probing &&
Packit Service 3880ab
		    queue_active_probe(ndm->ndm_ifindex, key.addr) == 0 &&
Packit Service 3880ab
		    do_acct) {
Packit Service 3880ab
			NEG_CNT(dbdat.data)++;
Packit Service 3880ab
			dbase->put(dbase, &dbkey, &dbdat, 0);
Packit Service 3880ab
		}
Packit Service 3880ab
	} else if (n->nlmsg_type == RTM_NEWNEIGH) {
Packit Service 3880ab
		if (n->nlmsg_flags&NLM_F_REQUEST)
Packit Service 3880ab
			return 0;
Packit Service 3880ab
Packit Service 3880ab
		if (ndm->ndm_state&NUD_FAILED) {
Packit Service 3880ab
			/* Kernel was not able to resolve. Host is dead.
Packit Service 3880ab
			 * Create negative entry if it is not present
Packit Service 3880ab
			 * or renew it if it is too old. */
Packit Service 3880ab
			if (!dbdat.data ||
Packit Service 3880ab
			    !IS_NEG(dbdat.data) ||
Packit Service 3880ab
			    !NEG_VALID(dbdat.data)) {
Packit Service 3880ab
				__u8 ndata[6];
Packit Service 3880ab
Packit Service 3880ab
				stats.kern_neg++;
Packit Service 3880ab
				prepare_neg_entry(ndata, time(NULL));
Packit Service 3880ab
				dbdat.data = ndata;
Packit Service 3880ab
				dbdat.size = sizeof(ndata);
Packit Service 3880ab
				dbase->put(dbase, &dbkey, &dbdat, 0);
Packit Service 3880ab
			}
Packit Service 3880ab
		} else if (tb[NDA_LLADDR]) {
Packit Service 3880ab
			if (dbdat.data && !IS_NEG(dbdat.data)) {
Packit Service 3880ab
				if (memcmp(RTA_DATA(tb[NDA_LLADDR]), dbdat.data, dbdat.size) == 0)
Packit Service 3880ab
					return 0;
Packit Service 3880ab
				stats.kern_change++;
Packit Service 3880ab
			} else {
Packit Service 3880ab
				stats.kern_new++;
Packit Service 3880ab
			}
Packit Service 3880ab
			dbdat.data = RTA_DATA(tb[NDA_LLADDR]);
Packit Service 3880ab
			dbdat.size = RTA_PAYLOAD(tb[NDA_LLADDR]);
Packit Service 3880ab
			dbase->put(dbase, &dbkey, &dbdat, 0);
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static void load_initial_table(void)
Packit Service 3880ab
{
Packit Service 3880ab
	if (rtnl_neighdump_req(&rth, AF_INET, NULL) < 0) {
Packit Service 3880ab
		perror("dump request failed");
Packit Service 3880ab
		exit(1);
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static void get_kern_msg(void)
Packit Service 3880ab
{
Packit Service 3880ab
	int status;
Packit Service 3880ab
	struct nlmsghdr *h;
Packit Service 3880ab
	struct sockaddr_nl nladdr = {};
Packit Service 3880ab
	struct iovec iov;
Packit Service 3880ab
	char   buf[8192];
Packit Service 3880ab
	struct msghdr msg = {
Packit Service 3880ab
		(void *)&nladdr, sizeof(nladdr),
Packit Service 3880ab
		&iov,	1,
Packit Service 3880ab
		NULL,	0,
Packit Service 3880ab
		0
Packit Service 3880ab
	};
Packit Service 3880ab
Packit Service 3880ab
	iov.iov_base = buf;
Packit Service 3880ab
	iov.iov_len = sizeof(buf);
Packit Service 3880ab
Packit Service 3880ab
	status = recvmsg(rth.fd, &msg, MSG_DONTWAIT);
Packit Service 3880ab
Packit Service 3880ab
	if (status <= 0)
Packit Service 3880ab
		return;
Packit Service 3880ab
Packit Service 3880ab
	if (msg.msg_namelen != sizeof(nladdr))
Packit Service 3880ab
		return;
Packit Service 3880ab
Packit Service 3880ab
	if (nladdr.nl_pid)
Packit Service 3880ab
		return;
Packit Service 3880ab
Packit Service 3880ab
	for (h = (struct nlmsghdr *)buf; status >= sizeof(*h); ) {
Packit Service 3880ab
		int len = h->nlmsg_len;
Packit Service 3880ab
		int l = len - sizeof(*h);
Packit Service 3880ab
Packit Service 3880ab
		if (l < 0 || len > status)
Packit Service 3880ab
			return;
Packit Service 3880ab
Packit Service 3880ab
		if (do_one_request(h) < 0)
Packit Service 3880ab
			return;
Packit Service 3880ab
Packit Service 3880ab
		status -= NLMSG_ALIGN(len);
Packit Service 3880ab
		h = (struct nlmsghdr *)((char *)h + NLMSG_ALIGN(len));
Packit Service 3880ab
	}
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
/* Receive gratuitous ARP messages and store them, that's all. */
Packit Service 3880ab
static void get_arp_pkt(void)
Packit Service 3880ab
{
Packit Service 3880ab
	unsigned char buf[1024];
Packit Service 3880ab
	struct sockaddr_ll sll;
Packit Service 3880ab
	socklen_t sll_len = sizeof(sll);
Packit Service 3880ab
	struct arphdr *a = (struct arphdr *)buf;
Packit Service 3880ab
	struct dbkey key;
Packit Service 3880ab
	DBT dbkey, dbdat;
Packit Service 3880ab
	int n;
Packit Service 3880ab
Packit Service 3880ab
	n = recvfrom(pset[0].fd, buf, sizeof(buf), MSG_DONTWAIT,
Packit Service 3880ab
		     (struct sockaddr *)&sll, &sll_len);
Packit Service 3880ab
	if (n < 0) {
Packit Service 3880ab
		if (errno != EINTR && errno != EAGAIN)
Packit Service 3880ab
			syslog(LOG_ERR, "recvfrom: %m");
Packit Service 3880ab
		return;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (ifnum && !handle_if(sll.sll_ifindex))
Packit Service 3880ab
		return;
Packit Service 3880ab
Packit Service 3880ab
	/* Sanity checks */
Packit Service 3880ab
Packit Service 3880ab
	if (n < sizeof(*a) ||
Packit Service 3880ab
	    (a->ar_op != htons(ARPOP_REQUEST) &&
Packit Service 3880ab
	     a->ar_op != htons(ARPOP_REPLY)) ||
Packit Service 3880ab
	    a->ar_pln != 4 ||
Packit Service 3880ab
	    a->ar_pro != htons(ETH_P_IP) ||
Packit Service 3880ab
	    a->ar_hln != sll.sll_halen ||
Packit Service 3880ab
	    sizeof(*a) + 2*4 + 2*a->ar_hln > n)
Packit Service 3880ab
		return;
Packit Service 3880ab
Packit Service 3880ab
	key.iface = sll.sll_ifindex;
Packit Service 3880ab
	memcpy(&key.addr, (char *)(a+1) + a->ar_hln, 4);
Packit Service 3880ab
Packit Service 3880ab
	/* DAD message, ignore. */
Packit Service 3880ab
	if (key.addr == 0)
Packit Service 3880ab
		return;
Packit Service 3880ab
Packit Service 3880ab
	dbkey.data = &ke;;
Packit Service 3880ab
	dbkey.size = sizeof(key);
Packit Service 3880ab
Packit Service 3880ab
	if (dbase->get(dbase, &dbkey, &dbdat, 0) == 0 && !IS_NEG(dbdat.data)) {
Packit Service 3880ab
		if (memcmp(dbdat.data, a+1, dbdat.size) == 0)
Packit Service 3880ab
			return;
Packit Service 3880ab
		stats.arp_change++;
Packit Service 3880ab
	} else {
Packit Service 3880ab
		stats.arp_new++;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	dbdat.data = a+1;
Packit Service 3880ab
	dbdat.size = a->ar_hln;
Packit Service 3880ab
	dbase->put(dbase, &dbkey, &dbdat, 0);
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static void catch_signal(int sig, void (*handler)(int))
Packit Service 3880ab
{
Packit Service 3880ab
	struct sigaction sa = { .sa_handler = handler };
Packit Service 3880ab
Packit Service 3880ab
#ifdef SA_INTERRUPT
Packit Service 3880ab
	sa.sa_flags = SA_INTERRUPT;
Packit Service 3880ab
#endif
Packit Service 3880ab
	sigaction(sig, &sa, NULL);
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
#include <setjmp.h>
Packit Service 3880ab
sigjmp_buf env;
Packit Service 3880ab
volatile int in_poll;
Packit Service 3880ab
Packit Service 3880ab
static void sig_exit(int signo)
Packit Service 3880ab
{
Packit Service 3880ab
	do_exit = 1;
Packit Service 3880ab
	if (in_poll)
Packit Service 3880ab
		siglongjmp(env, 1);
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static void sig_sync(int signo)
Packit Service 3880ab
{
Packit Service 3880ab
	do_sync = 1;
Packit Service 3880ab
	if (in_poll)
Packit Service 3880ab
		siglongjmp(env, 1);
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static void sig_stats(int signo)
Packit Service 3880ab
{
Packit Service 3880ab
	do_sync = 1;
Packit Service 3880ab
	do_stats = 1;
Packit Service 3880ab
	if (in_poll)
Packit Service 3880ab
		siglongjmp(env, 1);
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static void send_stats(void)
Packit Service 3880ab
{
Packit Service 3880ab
	syslog(LOG_INFO, "arp_rcv: n%lu c%lu app_rcv: tot %lu hits %lu bad %lu neg %lu sup %lu",
Packit Service 3880ab
	       stats.arp_new, stats.arp_change,
Packit Service 3880ab
Packit Service 3880ab
	       stats.app_recv, stats.app_success,
Packit Service 3880ab
	       stats.app_bad, stats.app_neg, stats.app_suppressed
Packit Service 3880ab
	       );
Packit Service 3880ab
	syslog(LOG_INFO, "kern: n%lu c%lu neg %lu arp_send: %lu rlim %lu",
Packit Service 3880ab
	       stats.kern_new, stats.kern_change, stats.kern_neg,
Packit Service 3880ab
Packit Service 3880ab
	       stats.probes_sent, stats.probes_suppressed
Packit Service 3880ab
	       );
Packit Service 3880ab
	do_stats = 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
Packit Service 3880ab
int main(int argc, char **argv)
Packit Service 3880ab
{
Packit Service 3880ab
	int opt;
Packit Service 3880ab
	int do_list = 0;
Packit Service 3880ab
	char *do_load = NULL;
Packit Service 3880ab
Packit Service 3880ab
	while ((opt = getopt(argc, argv, "h?b:lf:a:n:p:kR:B:")) != EOF) {
Packit Service 3880ab
		switch (opt) {
Packit Service 3880ab
		case 'b':
Packit Service 3880ab
			dbname = optarg;
Packit Service 3880ab
			break;
Packit Service 3880ab
		case 'f':
Packit Service 3880ab
			if (do_load) {
Packit Service 3880ab
				fprintf(stderr, "Duplicate option -f\n");
Packit Service 3880ab
				usage();
Packit Service 3880ab
			}
Packit Service 3880ab
			do_load = optarg;
Packit Service 3880ab
			break;
Packit Service 3880ab
		case 'l':
Packit Service 3880ab
			do_list = 1;
Packit Service 3880ab
			break;
Packit Service 3880ab
		case 'a':
Packit Service 3880ab
			active_probing = atoi(optarg);
Packit Service 3880ab
			break;
Packit Service 3880ab
		case 'n':
Packit Service 3880ab
			negative_timeout = atoi(optarg);
Packit Service 3880ab
			break;
Packit Service 3880ab
		case 'k':
Packit Service 3880ab
			no_kernel_broadcasts = 1;
Packit Service 3880ab
			break;
Packit Service 3880ab
		case 'p':
Packit Service 3880ab
			if ((poll_timeout = 1000 * strtod(optarg, NULL)) < 100) {
Packit Service 3880ab
				fprintf(stderr, "Invalid poll timeout\n");
Packit Service 3880ab
				exit(-1);
Packit Service 3880ab
			}
Packit Service 3880ab
			break;
Packit Service 3880ab
		case 'R':
Packit Service 3880ab
			if ((broadcast_rate = atoi(optarg)) <= 0 ||
Packit Service 3880ab
			    (broadcast_rate = 1000/broadcast_rate) <= 0) {
Packit Service 3880ab
				fprintf(stderr, "Invalid ARP rate\n");
Packit Service 3880ab
				exit(-1);
Packit Service 3880ab
			}
Packit Service 3880ab
			break;
Packit Service 3880ab
		case 'B':
Packit Service 3880ab
			if ((broadcast_burst = atoi(optarg)) <= 0 ||
Packit Service 3880ab
			    (broadcast_burst = 1000*broadcast_burst) <= 0) {
Packit Service 3880ab
				fprintf(stderr, "Invalid ARP burst\n");
Packit Service 3880ab
				exit(-1);
Packit Service 3880ab
			}
Packit Service 3880ab
			break;
Packit Service 3880ab
		case 'h':
Packit Service 3880ab
		case '?':
Packit Service 3880ab
		default:
Packit Service 3880ab
			usage();
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
	argc -= optind;
Packit Service 3880ab
	argv += optind;
Packit Service 3880ab
Packit Service 3880ab
	if (argc > 0) {
Packit Service 3880ab
		ifnum = argc;
Packit Service 3880ab
		ifnames = argv;
Packit Service 3880ab
		ifvec = malloc(argc*sizeof(int));
Packit Service 3880ab
		if (!ifvec) {
Packit Service 3880ab
			perror("malloc");
Packit Service 3880ab
			exit(-1);
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if ((udp_sock = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
Packit Service 3880ab
		perror("socket");
Packit Service 3880ab
		exit(-1);
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (ifnum) {
Packit Service 3880ab
		int i;
Packit Service 3880ab
		struct ifreq ifr = {};
Packit Service 3880ab
Packit Service 3880ab
		for (i = 0; i < ifnum; i++) {
Packit Service 3880ab
			if (get_ifname(ifr.ifr_name, ifnames[i]))
Packit Service 3880ab
				invarg("not a valid ifname", ifnames[i]);
Packit Service 3880ab
			if (ioctl(udp_sock, SIOCGIFINDEX, &ifr)) {
Packit Service 3880ab
				perror("ioctl(SIOCGIFINDEX)");
Packit Service 3880ab
				exit(-1);
Packit Service 3880ab
			}
Packit Service 3880ab
			ifvec[i] = ifr.ifr_ifindex;
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	dbase = dbopen(dbname, O_CREAT|O_RDWR, 0644, DB_HASH, NULL);
Packit Service 3880ab
	if (dbase == NULL) {
Packit Service 3880ab
		perror("db_open");
Packit Service 3880ab
		exit(-1);
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (do_load) {
Packit Service 3880ab
		char buf[128];
Packit Service 3880ab
		FILE *fp;
Packit Service 3880ab
		struct dbkey k;
Packit Service 3880ab
		DBT dbkey, dbdat;
Packit Service 3880ab
Packit Service 3880ab
		dbkey.data = &k;
Packit Service 3880ab
		dbkey.size = sizeof(k);
Packit Service 3880ab
Packit Service 3880ab
		if (strcmp(do_load, "-") == 0 || strcmp(do_load, "--") == 0) {
Packit Service 3880ab
			fp = stdin;
Packit Service 3880ab
		} else if ((fp = fopen(do_load, "r")) == NULL) {
Packit Service 3880ab
			perror("fopen");
Packit Service 3880ab
			goto do_abort;
Packit Service 3880ab
		}
Packit Service 3880ab
Packit Service 3880ab
		buf[sizeof(buf)-1] = 0;
Packit Service 3880ab
		while (fgets(buf, sizeof(buf), fp)) {
Packit Service 3880ab
			__u8 b1[6];
Packit Service 3880ab
			char ipbuf[128];
Packit Service 3880ab
			char macbuf[128];
Packit Service 3880ab
Packit Service 3880ab
			if (buf[0] == '#')
Packit Service 3880ab
				continue;
Packit Service 3880ab
Packit Service 3880ab
			if (sscanf(buf, "%u%s%s", &k.iface, ipbuf, macbuf) != 3) {
Packit Service 3880ab
				fprintf(stderr, "Wrong format of input file \"%s\"\n", do_load);
Packit Service 3880ab
				goto do_abort;
Packit Service 3880ab
			}
Packit Service 3880ab
			if (strncmp(macbuf, "FAILED:", 7) == 0)
Packit Service 3880ab
				continue;
Packit Service 3880ab
			if (!inet_aton(ipbuf, (struct in_addr *)&k.addr)) {
Packit Service 3880ab
				fprintf(stderr, "Invalid IP address: \"%s\"\n", ipbuf);
Packit Service 3880ab
				goto do_abort;
Packit Service 3880ab
			}
Packit Service 3880ab
Packit Service 3880ab
			if (ll_addr_a2n((char *) b1, 6, macbuf) != 6)
Packit Service 3880ab
				goto do_abort;
Packit Service 3880ab
			dbdat.size = 6;
Packit Service 3880ab
Packit Service 3880ab
			if (dbase->put(dbase, &dbkey, &dbdat, 0)) {
Packit Service 3880ab
				perror("hash->put");
Packit Service 3880ab
				goto do_abort;
Packit Service 3880ab
			}
Packit Service 3880ab
		}
Packit Service 3880ab
		dbase->sync(dbase, 0);
Packit Service 3880ab
		if (fp != stdin)
Packit Service 3880ab
			fclose(fp);
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (do_list) {
Packit Service 3880ab
		DBT dbkey, dbdat;
Packit Service 3880ab
Packit Service 3880ab
		printf("%-8s %-15s %s\n", "#Ifindex", "IP", "MAC");
Packit Service 3880ab
		while (dbase->seq(dbase, &dbkey, &dbdat, R_NEXT) == 0) {
Packit Service 3880ab
			struct dbkey *key = dbkey.data;
Packit Service 3880ab
Packit Service 3880ab
			if (handle_if(key->iface)) {
Packit Service 3880ab
				if (!IS_NEG(dbdat.data)) {
Packit Service 3880ab
					char b1[18];
Packit Service 3880ab
Packit Service 3880ab
					printf("%-8d %-15s %s\n",
Packit Service 3880ab
					       key->iface,
Packit Service 3880ab
					       inet_ntoa(*(struct in_addr *)&key->addr),
Packit Service 3880ab
					       ll_addr_n2a(dbdat.data, 6, ARPHRD_ETHER, b1, 18));
Packit Service 3880ab
				} else {
Packit Service 3880ab
					printf("%-8d %-15s FAILED: %dsec ago\n",
Packit Service 3880ab
					       key->iface,
Packit Service 3880ab
					       inet_ntoa(*(struct in_addr *)&key->addr),
Packit Service 3880ab
					       NEG_AGE(dbdat.data));
Packit Service 3880ab
				}
Packit Service 3880ab
			}
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (do_load || do_list)
Packit Service 3880ab
		goto out;
Packit Service 3880ab
Packit Service 3880ab
	pset[0].fd = socket(PF_PACKET, SOCK_DGRAM, 0);
Packit Service 3880ab
	if (pset[0].fd < 0) {
Packit Service 3880ab
		perror("socket");
Packit Service 3880ab
		exit(-1);
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (1) {
Packit Service 3880ab
		struct sockaddr_ll sll = {
Packit Service 3880ab
			.sll_family = AF_PACKET,
Packit Service 3880ab
			.sll_protocol = htons(ETH_P_ARP),
Packit Service 3880ab
			.sll_ifindex = (ifnum == 1 ? ifvec[0] : 0),
Packit Service 3880ab
		};
Packit Service 3880ab
Packit Service 3880ab
		if (bind(pset[0].fd, (struct sockaddr *)&sll, sizeof(sll)) < 0) {
Packit Service 3880ab
			perror("bind");
Packit Service 3880ab
			goto do_abort;
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (rtnl_open(&rth, RTMGRP_NEIGH) < 0) {
Packit Service 3880ab
		perror("rtnl_open");
Packit Service 3880ab
		goto do_abort;
Packit Service 3880ab
	}
Packit Service 3880ab
	pset[1].fd = rth.fd;
Packit Service 3880ab
Packit Service 3880ab
	load_initial_table();
Packit Service 3880ab
Packit Service 3880ab
	if (daemon(0, 0)) {
Packit Service 3880ab
		perror("arpd: daemon");
Packit Service 3880ab
		goto do_abort;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	openlog("arpd", LOG_PID | LOG_CONS, LOG_DAEMON);
Packit Service 3880ab
	catch_signal(SIGINT, sig_exit);
Packit Service 3880ab
	catch_signal(SIGTERM, sig_exit);
Packit Service 3880ab
	catch_signal(SIGHUP, sig_sync);
Packit Service 3880ab
	catch_signal(SIGUSR1, sig_stats);
Packit Service 3880ab
Packit Service 3880ab
#define EVENTS (POLLIN|POLLPRI|POLLERR|POLLHUP)
Packit Service 3880ab
	pset[0].events = EVENTS;
Packit Service 3880ab
	pset[0].revents = 0;
Packit Service 3880ab
	pset[1].events = EVENTS;
Packit Service 3880ab
	pset[1].revents = 0;
Packit Service 3880ab
Packit Service 3880ab
	sigsetjmp(env, 1);
Packit Service 3880ab
Packit Service 3880ab
	for (;;) {
Packit Service 3880ab
		in_poll = 1;
Packit Service 3880ab
Packit Service 3880ab
		if (do_exit)
Packit Service 3880ab
			break;
Packit Service 3880ab
		if (do_sync) {
Packit Service 3880ab
			in_poll = 0;
Packit Service 3880ab
			dbase->sync(dbase, 0);
Packit Service 3880ab
			do_sync = 0;
Packit Service 3880ab
			in_poll = 1;
Packit Service 3880ab
		}
Packit Service 3880ab
		if (do_stats)
Packit Service 3880ab
			send_stats();
Packit Service 3880ab
		if (poll(pset, 2, poll_timeout) > 0) {
Packit Service 3880ab
			in_poll = 0;
Packit Service 3880ab
			if (pset[0].revents&EVENTS)
Packit Service 3880ab
				get_arp_pkt();
Packit Service 3880ab
			if (pset[1].revents&EVENTS)
Packit Service 3880ab
				get_kern_msg();
Packit Service 3880ab
		} else {
Packit Service 3880ab
			do_sync = 1;
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	undo_sysctl_adjustments();
Packit Service 3880ab
out:
Packit Service 3880ab
	dbase->close(dbase);
Packit Service 3880ab
	exit(0);
Packit Service 3880ab
Packit Service 3880ab
do_abort:
Packit Service 3880ab
	dbase->close(dbase);
Packit Service 3880ab
	exit(-1);
Packit Service 3880ab
}