[mirror_iproute2.git] / tc / m_mirred.c

/*
 * m_egress.c		ingress/egress packet mirror/redir actions module
 *
 *		This program is free software; you can distribute 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:  J Hadi Salim (hadi@cyberus.ca)
 *
 * TODO: Add Ingress support
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include "utils.h"
#include "tc_util.h"
#include "tc_common.h"
#include <linux/tc_act/tc_mirred.h>

static void
explain(void)
{
	fprintf(stderr,
		"Usage: mirred <DIRECTION> <ACTION> [index INDEX] <dev DEVICENAME>\n"
		"where:\n"
		"\tDIRECTION := <ingress | egress>\n"
		"\tACTION := <mirror | redirect>\n"
		"\tINDEX  is the specific policy instance id\n"
		"\tDEVICENAME is the devicename\n");
}

static void
usage(void)
{
	explain();
	exit(-1);
}

static const char *mirred_n2a(int action)
{
	switch (action) {
	case TCA_EGRESS_REDIR:
		return "Egress Redirect";
	case TCA_INGRESS_REDIR:
		return "Ingress Redirect";
	case TCA_EGRESS_MIRROR:
		return "Egress Mirror";
	case TCA_INGRESS_MIRROR:
		return "Ingress Mirror";
	default:
		return "unknown";
	}
}

static const char *mirred_direction(int action)
{
	switch (action) {
	case TCA_EGRESS_REDIR:
	case TCA_EGRESS_MIRROR:
		return "egress";
	case TCA_INGRESS_REDIR:
	case TCA_INGRESS_MIRROR:
		return "ingress";
	default:
		return "unknown";
	}
}

static const char *mirred_action(int action)
{
	switch (action) {
	case TCA_EGRESS_REDIR:
	case TCA_INGRESS_REDIR:
		return "redirect";
	case TCA_EGRESS_MIRROR:
	case TCA_INGRESS_MIRROR:
		return "mirror";
	default:
		return "unknown";
	}
}

static int
parse_direction(struct action_util *a, int *argc_p, char ***argv_p,
		int tca_id, struct nlmsghdr *n)
{

	int argc = *argc_p;
	char **argv = *argv_p;
	int ok = 0, iok = 0, mirror = 0, redir = 0, ingress = 0, egress = 0;
	struct tc_mirred p = {};
	struct rtattr *tail;
	char d[IFNAMSIZ] = {};

	while (argc > 0) {

		if (matches(*argv, "action") == 0) {
			NEXT_ARG();
			break;
		} else if (!egress && matches(*argv, "egress") == 0) {
			egress = 1;
			if (ingress) {
				fprintf(stderr,
					"Can't have both egress and ingress\n");
				return -1;
			}
			NEXT_ARG();
			ok++;
			continue;
		} else if (!ingress && matches(*argv, "ingress") == 0) {
			ingress = 1;
			if (egress) {
				fprintf(stderr,
					"Can't have both ingress and egress\n");
				return -1;
			}
			NEXT_ARG();
			ok++;
			continue;
		} else {

			if (matches(*argv, "index") == 0) {
				NEXT_ARG();
				if (get_u32(&p.index, *argv, 10)) {
					fprintf(stderr, "Illegal \"index\"\n");
					return -1;
				}
				iok++;
				if (!ok) {
					argc--;
					argv++;
					break;
				}
			} else if (!ok) {
				fprintf(stderr,
					"was expecting egress or ingress (%s)\n",
					*argv);
				break;

			} else if (!mirror && matches(*argv, "mirror") == 0) {
				mirror = 1;
				if (redir) {
					fprintf(stderr,
						"Can't have both mirror and redir\n");
					return -1;
				}
				p.eaction = egress ? TCA_EGRESS_MIRROR :
					TCA_INGRESS_MIRROR;
				p.action = TC_ACT_PIPE;
				ok++;
			} else if (!redir && matches(*argv, "redirect") == 0) {
				redir = 1;
				if (mirror) {
					fprintf(stderr,
						"Can't have both mirror and redir\n");
					return -1;
				}
				p.eaction = egress ? TCA_EGRESS_REDIR :
					TCA_INGRESS_REDIR;
				p.action = TC_ACT_STOLEN;
				ok++;
			} else if ((redir || mirror) &&
				   matches(*argv, "dev") == 0) {
				NEXT_ARG();
				if (strlen(d))
					duparg("dev", *argv);

				strncpy(d, *argv, sizeof(d)-1);
				argc--;
				argv++;

				break;

			}
		}

		NEXT_ARG();
	}

	if (!ok && !iok)
		return -1;

	if (d[0])  {
		int idx;

		ll_init_map(&rth);

		idx = ll_name_to_index(d);
		if (!idx)
			return nodev(d);

		p.ifindex = idx;
	}


	if (p.eaction == TCA_EGRESS_MIRROR || p.eaction == TCA_INGRESS_MIRROR)
		parse_action_control(&argc, &argv, &p.action, false);

	if (argc) {
		if (iok && matches(*argv, "index") == 0) {
			fprintf(stderr, "mirred: Illegal double index\n");
			return -1;
		}

		if (matches(*argv, "index") == 0) {
			NEXT_ARG();
			if (get_u32(&p.index, *argv, 10)) {
				fprintf(stderr,
					"mirred: Illegal \"index\"\n");
				return -1;
			}
			argc--;
			argv++;
		}
	}

	tail = addattr_nest(n, MAX_MSG, tca_id);
	addattr_l(n, MAX_MSG, TCA_MIRRED_PARMS, &p, sizeof(p));
	addattr_nest_end(n, tail);

	*argc_p = argc;
	*argv_p = argv;
	return 0;
}


static int
parse_mirred(struct action_util *a, int *argc_p, char ***argv_p,
	     int tca_id, struct nlmsghdr *n)
{

	int argc = *argc_p;
	char **argv = *argv_p;

	if (argc < 0) {
		fprintf(stderr, "mirred bad argument count %d\n", argc);
		return -1;
	}

	if (matches(*argv, "mirred") == 0) {
		NEXT_ARG();
	} else {
		fprintf(stderr, "mirred bad argument %s\n", *argv);
		return -1;
	}


	if (matches(*argv, "egress") == 0 || matches(*argv, "ingress") == 0 ||
	    matches(*argv, "index") == 0) {
		int ret = parse_direction(a, &argc, &argv, tca_id, n);

		if (ret == 0) {
			*argc_p = argc;
			*argv_p = argv;
			return 0;
		}

	} else if (matches(*argv, "help") == 0) {
		usage();
	} else {
		fprintf(stderr, "mirred option not supported %s\n", *argv);
	}

	return -1;

}

static int
print_mirred(struct action_util *au, FILE *f, struct rtattr *arg)
{
	struct tc_mirred *p;
	struct rtattr *tb[TCA_MIRRED_MAX + 1];
	const char *dev;

	if (arg == NULL)
		return -1;

	parse_rtattr_nested(tb, TCA_MIRRED_MAX, arg);

	if (tb[TCA_MIRRED_PARMS] == NULL) {
		print_string(PRINT_FP, NULL, "%s", "[NULL mirred parameters]");
		return -1;
	}
	p = RTA_DATA(tb[TCA_MIRRED_PARMS]);

	dev = ll_index_to_name(p->ifindex);
	if (dev == 0) {
		fprintf(stderr, "Cannot find device %d\n", p->ifindex);
		return -1;
	}

	print_string(PRINT_ANY, "kind", "%s ", "mirred");
	print_string(PRINT_FP, NULL, "(%s", mirred_n2a(p->eaction));
	print_string(PRINT_JSON, "mirred_action", NULL,
		     mirred_action(p->eaction));
	print_string(PRINT_JSON, "direction", NULL,
		     mirred_direction(p->eaction));
	print_string(PRINT_ANY, "to_dev", " to device %s)", dev);
	print_action_control(f, " ", p->action, "");

	print_uint(PRINT_ANY, "index", "\n \tindex %u", p->index);
	print_int(PRINT_ANY, "ref", " ref %d", p->refcnt);
	print_int(PRINT_ANY, "bind", " bind %d", p->bindcnt);

	if (show_stats) {
		if (tb[TCA_MIRRED_TM]) {
			struct tcf_t *tm = RTA_DATA(tb[TCA_MIRRED_TM]);

			print_tm(f, tm);
		}
	}
	print_string(PRINT_FP, NULL, "%s", "\n ");
	return 0;
}

struct action_util mirred_action_util = {
	.id = "mirred",
	.parse_aopt = parse_mirred,
	.print_aopt = print_mirred,
};
Commit	Line	Data
00fa8480	1	/*
ae665a52	2	* m_egress.c ingress/egress packet mirror/redir actions module
00fa8480	3	*
	4	* This program is free software; you can distribute it and/or
	5	* modify it under the terms of the GNU General Public License
	6	* as published by the Free Software Foundation; either version
	7	* 2 of the License, or (at your option) any later version.
	8	*
ae665a52 SH	9	* Authors: J Hadi Salim (hadi@cyberus.ca)
ae665a52 SH	10	*
00fa8480	11	* TODO: Add Ingress support
	12	*
	13	*/
	14
	15	#include <stdio.h>
	16	#include <stdlib.h>
	17	#include <unistd.h>
00fa8480	18	#include <fcntl.h>
	19	#include <sys/socket.h>
	20	#include <netinet/in.h>
	21	#include <arpa/inet.h>
	22	#include <string.h>
	23	#include "utils.h"
	24	#include "tc_util.h"
c1027a75	25	#include "tc_common.h"
00fa8480	26	#include <linux/tc_act/tc_mirred.h>
00fa8480	27
00fa8480	28	static void
	29	explain(void)
	30	{
c6a656f4 SH	31	fprintf(stderr,
	32	"Usage: mirred <DIRECTION> <ACTION> [index INDEX] <dev DEVICENAME>\n"
	33	"where:\n"
	34	"\tDIRECTION := <ingress \| egress>\n"
	35	"\tACTION := <mirror \| redirect>\n"
	36	"\tINDEX is the specific policy instance id\n"
	37	"\tDEVICENAME is the devicename\n");
00fa8480	38	}
00fa8480	39
ebf32083 JHS	40	static void
	41	usage(void)
	42	{
	43	explain();
	44	exit(-1);
	45	}
00fa8480	46
d1f28cf1	47	static const char *mirred_n2a(int action)
00fa8480	48	{
	49	switch (action) {
	50	case TCA_EGRESS_REDIR:
	51	return "Egress Redirect";
	52	case TCA_INGRESS_REDIR:
	53	return "Ingress Redirect";
	54	case TCA_EGRESS_MIRROR:
	55	return "Egress Mirror";
	56	case TCA_INGRESS_MIRROR:
	57	return "Ingress Mirror";
	58	default:
	59	return "unknown";
	60	}
	61	}
	62
502c4adf JP	63	static const char *mirred_direction(int action)
	64	{
	65	switch (action) {
	66	case TCA_EGRESS_REDIR:
	67	case TCA_EGRESS_MIRROR:
	68	return "egress";
	69	case TCA_INGRESS_REDIR:
	70	case TCA_INGRESS_MIRROR:
	71	return "ingress";
	72	default:
	73	return "unknown";
	74	}
	75	}
	76
	77	static const char *mirred_action(int action)
	78	{
	79	switch (action) {
	80	case TCA_EGRESS_REDIR:
	81	case TCA_INGRESS_REDIR:
	82	return "redirect";
	83	case TCA_EGRESS_MIRROR:
	84	case TCA_INGRESS_MIRROR:
	85	return "mirror";
	86	default:
	87	return "unknown";
	88	}
	89	}
	90
d1f28cf1	91	static int
5eca0a37 SL	92	parse_direction(struct action_util a, int argc_p, char ***argv_p,
5eca0a37 SL	93	int tca_id, struct nlmsghdr *n)
00fa8480	94	{
	95
	96	int argc = *argc_p;
	97	char *argv = argv_p;
5eca0a37	98	int ok = 0, iok = 0, mirror = 0, redir = 0, ingress = 0, egress = 0;
d17b136f	99	struct tc_mirred p = {};
00fa8480	100	struct rtattr *tail;
b317557f	101	char d[IFNAMSIZ] = {};
00fa8480	102
	103	while (argc > 0) {
	104
	105	if (matches(*argv, "action") == 0) {
75ef7b18	106	NEXT_ARG();
00fa8480	107	break;
5eca0a37 SL	108	} else if (!egress && matches(*argv, "egress") == 0) {
	109	egress = 1;
	110	if (ingress) {
c6a656f4 SH	111	fprintf(stderr,
c6a656f4 SH	112	"Can't have both egress and ingress\n");
5eca0a37 SL	113	return -1;
	114	}
	115	NEXT_ARG();
	116	ok++;
	117	continue;
	118	} else if (!ingress && matches(*argv, "ingress") == 0) {
	119	ingress = 1;
	120	if (egress) {
c6a656f4 SH	121	fprintf(stderr,
c6a656f4 SH	122	"Can't have both ingress and egress\n");
5eca0a37 SL	123	return -1;
5eca0a37 SL	124	}
00fa8480	125	NEXT_ARG();
	126	ok++;
	127	continue;
	128	} else {
	129
	130	if (matches(*argv, "index") == 0) {
	131	NEXT_ARG();
	132	if (get_u32(&p.index, *argv, 10)) {
	133	fprintf(stderr, "Illegal \"index\"\n");
	134	return -1;
	135	}
	136	iok++;
	137	if (!ok) {
	138	argc--;
	139	argv++;
	140	break;
	141	}
32a121cb	142	} else if (!ok) {
c6a656f4 SH	143	fprintf(stderr,
	144	"was expecting egress or ingress (%s)\n",
	145	*argv);
00fa8480	146	break;
	147
	148	} else if (!mirror && matches(*argv, "mirror") == 0) {
32a121cb	149	mirror = 1;
00fa8480	150	if (redir) {
c6a656f4 SH	151	fprintf(stderr,
c6a656f4 SH	152	"Can't have both mirror and redir\n");
00fa8480	153	return -1;
00fa8480	154	}
5eca0a37	155	p.eaction = egress ? TCA_EGRESS_MIRROR :
c6a656f4	156	TCA_INGRESS_MIRROR;
00fa8480	157	p.action = TC_ACT_PIPE;
	158	ok++;
	159	} else if (!redir && matches(*argv, "redirect") == 0) {
32a121cb	160	redir = 1;
00fa8480	161	if (mirror) {
c6a656f4 SH	162	fprintf(stderr,
c6a656f4 SH	163	"Can't have both mirror and redir\n");
00fa8480	164	return -1;
00fa8480	165	}
5eca0a37	166	p.eaction = egress ? TCA_EGRESS_REDIR :
c6a656f4	167	TCA_INGRESS_REDIR;
00fa8480	168	p.action = TC_ACT_STOLEN;
00fa8480	169	ok++;
c6a656f4 SH	170	} else if ((redir \|\| mirror) &&
c6a656f4 SH	171	matches(*argv, "dev") == 0) {
00fa8480	172	NEXT_ARG();
	173	if (strlen(d))
	174	duparg("dev", *argv);
	175
	176	strncpy(d, *argv, sizeof(d)-1);
	177	argc--;
	178	argv++;
	179
	180	break;
	181
	182	}
	183	}
	184
	185	NEXT_ARG();
	186	}
	187
c6a656f4	188	if (!ok && !iok)
00fa8480	189	return -1;
00fa8480	190
	191	if (d[0]) {
	192	int idx;
32a121cb	193
00fa8480	194	ll_init_map(&rth);
00fa8480	195
c6a656f4	196	idx = ll_name_to_index(d);
fe99adbc SP	197	if (!idx)
fe99adbc SP	198	return nodev(d);
00fa8480	199
00fa8480	200	p.ifindex = idx;
00fa8480	201	}
	202
	203
75ef7b18	204	if (p.eaction == TCA_EGRESS_MIRROR \|\| p.eaction == TCA_INGRESS_MIRROR)
e67aba55	205	parse_action_control(&argc, &argv, &p.action, false);
00fa8480	206
	207	if (argc) {
	208	if (iok && matches(*argv, "index") == 0) {
	209	fprintf(stderr, "mirred: Illegal double index\n");
	210	return -1;
c6a656f4 SH	211	}
	212
	213	if (matches(*argv, "index") == 0) {
	214	NEXT_ARG();
	215	if (get_u32(&p.index, *argv, 10)) {
	216	fprintf(stderr,
	217	"mirred: Illegal \"index\"\n");
	218	return -1;
00fa8480	219	}
c6a656f4 SH	220	argc--;
c6a656f4 SH	221	argv++;
00fa8480	222	}
	223	}
	224
c14f9d92	225	tail = addattr_nest(n, MAX_MSG, tca_id);
32a121cb	226	addattr_l(n, MAX_MSG, TCA_MIRRED_PARMS, &p, sizeof(p));
c14f9d92	227	addattr_nest_end(n, tail);
00fa8480	228
	229	*argc_p = argc;
	230	*argv_p = argv;
	231	return 0;
	232	}
	233
	234
d1f28cf1 SH	235	static int
	236	parse_mirred(struct action_util a, int argc_p, char ***argv_p,
	237	int tca_id, struct nlmsghdr *n)
00fa8480	238	{
	239
	240	int argc = *argc_p;
	241	char *argv = argv_p;
	242
	243	if (argc < 0) {
32a121cb	244	fprintf(stderr, "mirred bad argument count %d\n", argc);
00fa8480	245	return -1;
	246	}
	247
	248	if (matches(*argv, "mirred") == 0) {
	249	NEXT_ARG();
	250	} else {
32a121cb	251	fprintf(stderr, "mirred bad argument %s\n", *argv);
00fa8480	252	return -1;
	253	}
	254
	255
5eca0a37 SL	256	if (matches(argv, "egress") == 0 \|\| matches(argv, "ingress") == 0 \|\|
	257	matches(*argv, "index") == 0) {
	258	int ret = parse_direction(a, &argc, &argv, tca_id, n);
32a121cb	259
00fa8480	260	if (ret == 0) {
	261	*argc_p = argc;
	262	*argv_p = argv;
	263	return 0;
	264	}
	265
ebf32083 JHS	266	} else if (matches(*argv, "help") == 0) {
ebf32083 JHS	267	usage();
00fa8480	268	} else {
32a121cb	269	fprintf(stderr, "mirred option not supported %s\n", *argv);
00fa8480	270	}
	271
	272	return -1;
ae665a52	273
00fa8480	274	}
00fa8480	275
d1f28cf1	276	static int
c6a656f4	277	print_mirred(struct action_util au, FILE f, struct rtattr *arg)
00fa8480	278	{
	279	struct tc_mirred *p;
	280	struct rtattr *tb[TCA_MIRRED_MAX + 1];
00fa8480	281	const char *dev;
32a121cb	282
00fa8480	283	if (arg == NULL)
	284	return -1;
	285
5cb5ee34	286	parse_rtattr_nested(tb, TCA_MIRRED_MAX, arg);
00fa8480	287
00fa8480	288	if (tb[TCA_MIRRED_PARMS] == NULL) {
502c4adf	289	print_string(PRINT_FP, NULL, "%s", "[NULL mirred parameters]");
00fa8480	290	return -1;
	291	}
	292	p = RTA_DATA(tb[TCA_MIRRED_PARMS]);
	293
c6a656f4 SH	294	dev = ll_index_to_name(p->ifindex);
c6a656f4 SH	295	if (dev == 0) {
00fa8480	296	fprintf(stderr, "Cannot find device %d\n", p->ifindex);
00fa8480	297	return -1;
	298	}
	299
502c4adf JP	300	print_string(PRINT_ANY, "kind", "%s ", "mirred");
	301	print_string(PRINT_FP, NULL, "(%s", mirred_n2a(p->eaction));
	302	print_string(PRINT_JSON, "mirred_action", NULL,
	303	mirred_action(p->eaction));
	304	print_string(PRINT_JSON, "direction", NULL,
	305	mirred_direction(p->eaction));
	306	print_string(PRINT_ANY, "to_dev", " to device %s)", dev);
e67aba55	307	print_action_control(f, " ", p->action, "");
00fa8480	308
502c4adf JP	309	print_uint(PRINT_ANY, "index", "\n \tindex %u", p->index);
	310	print_int(PRINT_ANY, "ref", " ref %d", p->refcnt);
	311	print_int(PRINT_ANY, "bind", " bind %d", p->bindcnt);
00fa8480	312
	313	if (show_stats) {
	314	if (tb[TCA_MIRRED_TM]) {
	315	struct tcf_t *tm = RTA_DATA(tb[TCA_MIRRED_TM]);
32a121cb SH	316
32a121cb SH	317	print_tm(f, tm);
00fa8480	318	}
00fa8480	319	}
502c4adf	320	print_string(PRINT_FP, NULL, "%s", "\n ");
00fa8480	321	return 0;
	322	}
	323
6ce88ca6	324	struct action_util mirred_action_util = {
00fa8480	325	.id = "mirred",
	326	.parse_aopt = parse_mirred,
	327	.print_aopt = print_mirred,
	328	};