Blame ip/rtmon.c

Packit d3f73b
/*
Packit d3f73b
 * rtmon.c		RTnetlink listener.
Packit d3f73b
 *
Packit d3f73b
 *		This program is free software; you can redistribute it and/or
Packit d3f73b
 *		modify it under the terms of the GNU General Public License
Packit d3f73b
 *		as published by the Free Software Foundation; either version
Packit d3f73b
 *		2 of the License, or (at your option) any later version.
Packit d3f73b
 *
Packit d3f73b
 * Authors:	Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
Packit d3f73b
 *
Packit d3f73b
 */
Packit d3f73b
Packit d3f73b
#include <stdio.h>
Packit d3f73b
#include <stdlib.h>
Packit d3f73b
#include <unistd.h>
Packit d3f73b
#include <fcntl.h>
Packit d3f73b
#include <sys/socket.h>
Packit d3f73b
#include <sys/time.h>
Packit d3f73b
#include <netinet/in.h>
Packit d3f73b
#include <string.h>
Packit d3f73b
Packit d3f73b
#include "SNAPSHOT.h"
Packit d3f73b
Packit d3f73b
#include "utils.h"
Packit d3f73b
#include "libnetlink.h"
Packit d3f73b
Packit d3f73b
static int init_phase = 1;
Packit d3f73b
Packit d3f73b
static void write_stamp(FILE *fp)
Packit d3f73b
{
Packit d3f73b
	char buf[128];
Packit d3f73b
	struct nlmsghdr *n1 = (void *)buf;
Packit d3f73b
	struct timeval tv;
Packit d3f73b
Packit d3f73b
	n1->nlmsg_type = NLMSG_TSTAMP;
Packit d3f73b
	n1->nlmsg_flags = 0;
Packit d3f73b
	n1->nlmsg_seq = 0;
Packit d3f73b
	n1->nlmsg_pid = 0;
Packit d3f73b
	n1->nlmsg_len = NLMSG_LENGTH(4*2);
Packit d3f73b
	gettimeofday(&tv, NULL);
Packit d3f73b
	((__u32 *)NLMSG_DATA(n1))[0] = tv.tv_sec;
Packit d3f73b
	((__u32 *)NLMSG_DATA(n1))[1] = tv.tv_usec;
Packit d3f73b
	fwrite((void *)n1, 1, NLMSG_ALIGN(n1->nlmsg_len), fp);
Packit d3f73b
}
Packit d3f73b
Packit d3f73b
static int dump_msg(struct rtnl_ctrl_data *ctrl,
Packit d3f73b
		    struct nlmsghdr *n, void *arg)
Packit d3f73b
{
Packit d3f73b
	FILE *fp = (FILE *)arg;
Packit d3f73b
Packit d3f73b
	if (!init_phase)
Packit d3f73b
		write_stamp(fp);
Packit d3f73b
	fwrite((void *)n, 1, NLMSG_ALIGN(n->nlmsg_len), fp);
Packit d3f73b
	fflush(fp);
Packit d3f73b
	return 0;
Packit d3f73b
}
Packit d3f73b
Packit d3f73b
static int dump_msg2(struct nlmsghdr *n, void *arg)
Packit d3f73b
{
Packit d3f73b
	return dump_msg(NULL, n, arg);
Packit d3f73b
}
Packit d3f73b
Packit d3f73b
static void usage(void)
Packit d3f73b
{
Packit d3f73b
	fprintf(stderr,
Packit d3f73b
		"Usage: rtmon [ OPTIONS ] file FILE [ all | LISTofOBJECTS ]\n"
Packit d3f73b
		"OPTIONS := { -f[amily] { inet | inet6 | link | help } |\n"
Packit d3f73b
		"	     -4 | -6 | -0 | -V[ersion] }\n"
Packit d3f73b
		"LISTofOBJECTS := [ link ] [ address ] [ route ]\n");
Packit d3f73b
	exit(-1);
Packit d3f73b
}
Packit d3f73b
Packit d3f73b
int
Packit d3f73b
main(int argc, char **argv)
Packit d3f73b
{
Packit d3f73b
	FILE *fp;
Packit d3f73b
	struct rtnl_handle rth;
Packit d3f73b
	int family = AF_UNSPEC;
Packit d3f73b
	unsigned int groups = ~0U;
Packit d3f73b
	int llink = 0;
Packit d3f73b
	int laddr = 0;
Packit d3f73b
	int lroute = 0;
Packit d3f73b
	char *file = NULL;
Packit d3f73b
Packit d3f73b
	while (argc > 1) {
Packit d3f73b
		if (matches(argv[1], "-family") == 0) {
Packit d3f73b
			argc--;
Packit d3f73b
			argv++;
Packit d3f73b
			if (argc <= 1)
Packit d3f73b
				usage();
Packit d3f73b
			if (strcmp(argv[1], "inet") == 0)
Packit d3f73b
				family = AF_INET;
Packit d3f73b
			else if (strcmp(argv[1], "inet6") == 0)
Packit d3f73b
				family = AF_INET6;
Packit d3f73b
			else if (strcmp(argv[1], "link") == 0)
Packit d3f73b
				family = AF_INET6;
Packit d3f73b
			else if (strcmp(argv[1], "help") == 0)
Packit d3f73b
				usage();
Packit d3f73b
			else {
Packit d3f73b
				fprintf(stderr, "Protocol ID \"%s\" is unknown, try \"rtmon help\".\n", argv[1]);
Packit d3f73b
				exit(-1);
Packit d3f73b
			}
Packit d3f73b
		} else if (strcmp(argv[1], "-4") == 0) {
Packit d3f73b
			family = AF_INET;
Packit d3f73b
		} else if (strcmp(argv[1], "-6") == 0) {
Packit d3f73b
			family = AF_INET6;
Packit d3f73b
		} else if (strcmp(argv[1], "-0") == 0) {
Packit d3f73b
			family = AF_PACKET;
Packit d3f73b
		} else if (matches(argv[1], "-Version") == 0) {
Packit d3f73b
			printf("rtmon utility, iproute2-ss%s\n", SNAPSHOT);
Packit d3f73b
			exit(0);
Packit d3f73b
		} else if (matches(argv[1], "file") == 0) {
Packit d3f73b
			argc--;
Packit d3f73b
			argv++;
Packit d3f73b
			if (argc <= 1)
Packit d3f73b
				usage();
Packit d3f73b
			file = argv[1];
Packit d3f73b
		} else if (matches(argv[1], "link") == 0) {
Packit d3f73b
			llink = 1;
Packit d3f73b
			groups = 0;
Packit d3f73b
		} else if (matches(argv[1], "address") == 0) {
Packit d3f73b
			laddr = 1;
Packit d3f73b
			groups = 0;
Packit d3f73b
		} else if (matches(argv[1], "route") == 0) {
Packit d3f73b
			lroute = 1;
Packit d3f73b
			groups = 0;
Packit d3f73b
		} else if (strcmp(argv[1], "all") == 0) {
Packit d3f73b
			groups = ~0U;
Packit d3f73b
		} else if (matches(argv[1], "help") == 0) {
Packit d3f73b
			usage();
Packit d3f73b
		} else {
Packit d3f73b
			fprintf(stderr, "Argument \"%s\" is unknown, try \"rtmon help\".\n", argv[1]);
Packit d3f73b
			exit(-1);
Packit d3f73b
		}
Packit d3f73b
		argc--;	argv++;
Packit d3f73b
	}
Packit d3f73b
Packit d3f73b
	if (file == NULL) {
Packit d3f73b
		fprintf(stderr, "Not enough information: argument \"file\" is required\n");
Packit d3f73b
		exit(-1);
Packit d3f73b
	}
Packit d3f73b
	if (llink)
Packit d3f73b
		groups |= nl_mgrp(RTNLGRP_LINK);
Packit d3f73b
	if (laddr) {
Packit d3f73b
		if (!family || family == AF_INET)
Packit d3f73b
			groups |= nl_mgrp(RTNLGRP_IPV4_IFADDR);
Packit d3f73b
		if (!family || family == AF_INET6)
Packit d3f73b
			groups |= nl_mgrp(RTNLGRP_IPV6_IFADDR);
Packit d3f73b
	}
Packit d3f73b
	if (lroute) {
Packit d3f73b
		if (!family || family == AF_INET)
Packit d3f73b
			groups |= nl_mgrp(RTNLGRP_IPV4_ROUTE);
Packit d3f73b
		if (!family || family == AF_INET6)
Packit d3f73b
			groups |= nl_mgrp(RTNLGRP_IPV6_ROUTE);
Packit d3f73b
	}
Packit d3f73b
Packit d3f73b
	fp = fopen(file, "w");
Packit d3f73b
	if (fp == NULL) {
Packit d3f73b
		perror("Cannot fopen");
Packit d3f73b
		exit(-1);
Packit d3f73b
	}
Packit d3f73b
Packit d3f73b
	if (rtnl_open(&rth, groups) < 0)
Packit d3f73b
		exit(1);
Packit d3f73b
Packit d3f73b
	if (rtnl_linkdump_req(&rth, AF_UNSPEC) < 0) {
Packit d3f73b
		perror("Cannot send dump request");
Packit d3f73b
		exit(1);
Packit d3f73b
	}
Packit d3f73b
Packit d3f73b
	write_stamp(fp);
Packit d3f73b
Packit d3f73b
	if (rtnl_dump_filter(&rth, dump_msg2, fp) < 0) {
Packit d3f73b
		fprintf(stderr, "Dump terminated\n");
Packit d3f73b
		return 1;
Packit d3f73b
	}
Packit d3f73b
Packit d3f73b
	init_phase = 0;
Packit d3f73b
Packit d3f73b
	if (rtnl_listen(&rth, dump_msg, (void *)fp) < 0)
Packit d3f73b
		exit(2);
Packit d3f73b
Packit d3f73b
	exit(0);
Packit d3f73b
}