/*
 * ipmroute.c		"ip mroute".
 *
 *		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.
 *
 * Authors:	Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <inttypes.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>

#include <linux/netdevice.h>
#include <linux/if.h>
#include <linux/if_arp.h>
#include <linux/sockios.h>

#include <rt_names.h>
#include "utils.h"
#include "ip_common.h"
#include "json_print.h"

static void usage(void) __attribute__((noreturn));

static void usage(void)
{
	fprintf(stderr,
		"Usage: ip mroute show [ [ to ] PREFIX ] [ from PREFIX ] [ iif DEVICE ]\n"
		"                      [ table TABLE_ID ]\n"
		"TABLE_ID := [ local | main | default | all | NUMBER ]\n"
#if 0
	"Usage: ip mroute [ add | del ] DESTINATION from SOURCE [ iif DEVICE ] [ oif DEVICE ]\n"
#endif
	);
	exit(-1);
}

static struct rtfilter {
	int tb;
	int af;
	int iif;
	inet_prefix mdst;
	inet_prefix msrc;
} filter;

int print_mroute(struct nlmsghdr *n, void *arg)
{
	struct rtmsg *r = NLMSG_DATA(n);
	int len = n->nlmsg_len;
	struct rtattr *tb[RTA_MAX+1];
	FILE *fp = arg;
	const char *src, *dst;
	SPRINT_BUF(b1);
	SPRINT_BUF(b2);
	__u32 table;
	int iif = 0;
	int family;

	if ((n->nlmsg_type != RTM_NEWROUTE &&
	     n->nlmsg_type != RTM_DELROUTE)) {
		fprintf(stderr, "Not a multicast route: %08x %08x %08x\n",
			n->nlmsg_len, n->nlmsg_type, n->nlmsg_flags);
		return 0;
	}
	len -= NLMSG_LENGTH(sizeof(*r));
	if (len < 0) {
		fprintf(stderr, "BUG: wrong nlmsg len %d\n", len);
		return -1;
	}

	if (r->rtm_type != RTN_MULTICAST) {
		fprintf(stderr,
			"Non multicast route received, kernel does support IP multicast?\n");
		return -1;
	}

	parse_rtattr(tb, RTA_MAX, RTM_RTA(r), len);
	table = rtm_get_table(r, tb);

	if (filter.tb > 0 && filter.tb != table)
		return 0;

	if (tb[RTA_IIF])
		iif = rta_getattr_u32(tb[RTA_IIF]);
	if (filter.iif && filter.iif != iif)
		return 0;

	if (filter.af && filter.af != r->rtm_family)
		return 0;

	if (inet_addr_match_rta(&filter.mdst, tb[RTA_DST]))
		return 0;

	if (inet_addr_match_rta(&filter.msrc, tb[RTA_SRC]))
		return 0;

	family = get_real_family(r->rtm_type, r->rtm_family);

	open_json_object(NULL);
	if (n->nlmsg_type == RTM_DELROUTE)
		print_bool(PRINT_ANY, "deleted", "Deleted ", true);

	if (tb[RTA_SRC])
		src = rt_addr_n2a_r(family, RTA_PAYLOAD(tb[RTA_SRC]),
				    RTA_DATA(tb[RTA_SRC]), b1, sizeof(b1));
	else
		src = "unknown";

	if (tb[RTA_DST])
		dst = rt_addr_n2a_r(family, RTA_PAYLOAD(tb[RTA_DST]),
				    RTA_DATA(tb[RTA_DST]), b2, sizeof(b2));
	else
		dst = "unknown";

	if (is_json_context()) {
		print_string(PRINT_JSON, "src", NULL, src);
		print_string(PRINT_JSON, "dst", NULL, dst);
	} else {
		char obuf[256];

		snprintf(obuf, sizeof(obuf), "(%s,%s)", src, dst);
		print_string(PRINT_FP, NULL,
			     "%-32s Iif: ", obuf);
	}

	if (iif)
		print_color_string(PRINT_ANY, COLOR_IFNAME,
				   "iif", "%-10s ", ll_index_to_name(iif));
	else
		print_string(PRINT_ANY,"iif", "%s ", "unresolved");

	if (tb[RTA_MULTIPATH]) {
		struct rtnexthop *nh = RTA_DATA(tb[RTA_MULTIPATH]);
		int first = 1;

		open_json_array(PRINT_JSON, "multipath");
		len = RTA_PAYLOAD(tb[RTA_MULTIPATH]);

		for (;;) {
			if (len < sizeof(*nh))
				break;
			if (nh->rtnh_len > len)
				break;

			open_json_object(NULL);
			if (first) {
				print_string(PRINT_FP, NULL, "Oifs: ", NULL);
				first = 0;
			}

			print_color_string(PRINT_ANY, COLOR_IFNAME,
					   "oif", "%s", ll_index_to_name(nh->rtnh_ifindex));

			if (nh->rtnh_hops > 1)
				print_uint(PRINT_ANY,
					   "ttl", "(ttl %u) ", nh->rtnh_hops);
			else
				print_string(PRINT_FP, NULL, " ", NULL);

			close_json_object();
			len -= NLMSG_ALIGN(nh->rtnh_len);
			nh = RTNH_NEXT(nh);
		}
		close_json_array(PRINT_JSON, NULL);
	}

	print_string(PRINT_ANY, "state", " State: %s",
		     (r->rtm_flags & RTNH_F_UNRESOLVED) ? "unresolved" : "resolved");

	if (r->rtm_flags & RTNH_F_OFFLOAD)
		print_null(PRINT_ANY, "offload", " offload", NULL);

	if (show_stats && tb[RTA_MFC_STATS]) {
		struct rta_mfc_stats *mfcs = RTA_DATA(tb[RTA_MFC_STATS]);

		print_nl();
		print_u64(PRINT_ANY, "packets", "  %"PRIu64" packets,",
			   mfcs->mfcs_packets);
		print_u64(PRINT_ANY, "bytes", " %"PRIu64" bytes", mfcs->mfcs_bytes);

		if (mfcs->mfcs_wrong_if)
			print_u64(PRINT_ANY, "wrong_if",
				   ", %"PRIu64" arrived on wrong iif.",
				   mfcs->mfcs_wrong_if);
	}

	if (show_stats && tb[RTA_EXPIRES]) {
		struct timeval tv;
		double age;

		__jiffies_to_tv(&tv, rta_getattr_u64(tb[RTA_EXPIRES]));
		age = tv.tv_sec;
		age += tv.tv_usec / 1000000.;
		print_float(PRINT_ANY, "expires",
			    ", Age %.2f", age);
	}

	if (table && (table != RT_TABLE_MAIN || show_details > 0) && !filter.tb)
		print_string(PRINT_ANY, "table", " Table: %s",
			     rtnl_rttable_n2a(table, b1, sizeof(b1)));

	print_string(PRINT_FP, NULL, "\n", NULL);
	close_json_object();
	fflush(fp);
	return 0;
}

void ipmroute_reset_filter(int ifindex)
{
	memset(&filter, 0, sizeof(filter));
	filter.mdst.bitlen = -1;
	filter.msrc.bitlen = -1;
	filter.iif = ifindex;
}

static int iproute_dump_filter(struct nlmsghdr *nlh, int reqlen)
{
	int err;

	if (filter.tb) {
		err = addattr32(nlh, reqlen, RTA_TABLE, filter.tb);
		if (err)
			return err;
	}

	return 0;
}

static int mroute_list(int argc, char **argv)
{
	char *id = NULL;
	int family = preferred_family;

	ipmroute_reset_filter(0);
	if (family == AF_INET || family == AF_UNSPEC) {
		family = RTNL_FAMILY_IPMR;
		filter.af = RTNL_FAMILY_IPMR;
		filter.tb = RT_TABLE_DEFAULT;  /* for backward compatibility */
	} else if (family == AF_INET6) {
		family = RTNL_FAMILY_IP6MR;
		filter.af = RTNL_FAMILY_IP6MR;
	} else {
		/* family does not have multicast routing */
		return 0;
	}

	filter.msrc.family = filter.mdst.family = family;

	while (argc > 0) {
		if (matches(*argv, "table") == 0) {
			__u32 tid;

			NEXT_ARG();
			if (rtnl_rttable_a2n(&tid, *argv)) {
				if (strcmp(*argv, "all") == 0) {
					filter.tb = 0;
				} else if (strcmp(*argv, "help") == 0) {
					usage();
				} else {
					invarg("table id value is invalid\n", *argv);
				}
			} else
				filter.tb = tid;
		} else if (strcmp(*argv, "iif") == 0) {
			NEXT_ARG();
			id = *argv;
		} else if (matches(*argv, "from") == 0) {
			NEXT_ARG();
			if (get_prefix(&filter.msrc, *argv, family))
				invarg("from value is invalid\n", *argv);
		} else {
			if (strcmp(*argv, "to") == 0) {
				NEXT_ARG();
			}
			if (matches(*argv, "help") == 0)
				usage();
			if (get_prefix(&filter.mdst, *argv, family))
				invarg("to value is invalid\n", *argv);
		}
		argc--; argv++;
	}

	ll_init_map(&rth);

	if (id)  {
		int idx;

		idx = ll_name_to_index(id);
		if (!idx)
			return nodev(id);
		filter.iif = idx;
	}

	if (rtnl_routedump_req(&rth, filter.af, iproute_dump_filter) < 0) {
		perror("Cannot send dump request");
		return 1;
	}

	new_json_obj(json);
	if (rtnl_dump_filter(&rth, print_mroute, stdout) < 0) {
		delete_json_obj();
		fprintf(stderr, "Dump terminated\n");
		exit(1);
	}
	delete_json_obj();

	return 0;
}

int do_multiroute(int argc, char **argv)
{
	if (argc < 1)
		return mroute_list(0, NULL);
#if 0
	if (matches(*argv, "add") == 0)
		return mroute_modify(RTM_NEWADDR, argc-1, argv+1);
	if (matches(*argv, "delete") == 0)
		return mroute_modify(RTM_DELADDR, argc-1, argv+1);
	if (matches(*argv, "get") == 0)
		return mroute_get(argc-1, argv+1);
#endif
	if (matches(*argv, "list") == 0 || matches(*argv, "show") == 0
	    || matches(*argv, "lst") == 0)
		return mroute_list(argc-1, argv+1);
	if (matches(*argv, "help") == 0)
		usage();
	fprintf(stderr, "Command \"%s\" is unknown, try \"ip mroute help\".\n", *argv);
	exit(-1);
}
