tc/m_ife.c - platform/external/iproute2 - Gitiles

 /*
  * m_ife.c	IFE 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 (jhs@mojatatu.com)
  *
  */

 #include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
 #include <syslog.h>
 #include <fcntl.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>
 #include <string.h>
 #include <linux/netdevice.h>

 #include "rt_names.h"
 #include "utils.h"
 #include "tc_util.h"
 #include <linux/tc_act/tc_ife.h>

 static void ife_explain(void)
 {
 	fprintf(stderr,
 		"Usage:... ife {decode|encode} [{ALLOW|USE} ATTR] [dst DMAC] [src SMAC] [type TYPE] [CONTROL] [index INDEX]\n");
 	fprintf(stderr,
 		"\tALLOW := Encode direction. Allows encoding specified metadata\n"
 		"\t\t e.g \"allow mark\"\n"
 		"\tUSE := Encode direction. Enforce Static encoding of specified metadata\n"
 		"\t\t e.g \"use mark 0x12\"\n"
 		"\tATTR := mark (32-bit), prio (32-bit), tcindex (16-bit)\n"
 		"\tDMAC := 6 byte Destination MAC address to encode\n"
 		"\tSMAC := optional 6 byte Source MAC address to encode\n"
 		"\tTYPE := optional 16 bit ethertype to encode\n"
 		"\tCONTROL := reclassify|pipe|drop|continue|ok\n"
 		"\tINDEX := optional IFE table index value used\n");
 	fprintf(stderr, "encode is used for sending IFE packets\n");
 	fprintf(stderr, "decode is used for receiving IFE packets\n");
 }

 static void ife_usage(void)
 {
 	ife_explain();
 	exit(-1);
 }

 static int parse_ife(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;
 	struct tc_ife p = { .action = TC_ACT_PIPE };	/* good default */
 	struct rtattr *tail;
 	struct rtattr *tail2;
 	char dbuf[ETH_ALEN];
 	char sbuf[ETH_ALEN];
 	__u16 ife_type = 0;
 	__u32 ife_prio = 0;
 	__u32 ife_prio_v = 0;
 	__u32 ife_mark = 0;
 	__u32 ife_mark_v = 0;
 	__u16 ife_tcindex = 0;
 	__u16 ife_tcindex_v = 0;
 	char *daddr = NULL;
 	char *saddr = NULL;

 	if (argc <= 0)
 		return -1;

 	while (argc > 0) {
 		if (matches(*argv, "ife") == 0) {
 			NEXT_ARG();
 			continue;
 		} else if (matches(*argv, "decode") == 0) {
 			p.flags = IFE_DECODE; /* readability aid */
 			ok++;
 		} else if (matches(*argv, "encode") == 0) {
 			p.flags = IFE_ENCODE;
 			ok++;
 		} else if (matches(*argv, "allow") == 0) {
 			NEXT_ARG();
 			if (matches(*argv, "mark") == 0) {
 				ife_mark = IFE_META_SKBMARK;
 			} else if (matches(*argv, "prio") == 0) {
 				ife_prio = IFE_META_PRIO;
 			} else if (matches(*argv, "tcindex") == 0) {
 				ife_prio = IFE_META_TCINDEX;
 			} else {
 				fprintf(stderr, "Illegal meta define <%s>\n",
 					*argv);
 				return -1;
 			}
 		} else if (matches(*argv, "use") == 0) {
 			NEXT_ARG();
 			if (matches(*argv, "mark") == 0) {
 				NEXT_ARG();
 				if (get_u32(&ife_mark_v, *argv, 0))
 					invarg("ife mark val is invalid",
 					       *argv);
 			} else if (matches(*argv, "prio") == 0) {
 				NEXT_ARG();
 				if (get_u32(&ife_prio_v, *argv, 0))
 					invarg("ife prio val is invalid",
 					       *argv);
 			} else if (matches(*argv, "tcindex") == 0) {
 				NEXT_ARG();
 				if (get_u16(&ife_tcindex_v, *argv, 0))
 					invarg("ife tcindex val is invalid",
 					       *argv);
 			} else {
 				fprintf(stderr, "Illegal meta use type <%s>\n",
 					*argv);
 				return -1;
 			}
 		} else if (matches(*argv, "type") == 0) {
 			NEXT_ARG();
 			if (get_u16(&ife_type, *argv, 0))
 				invarg("ife type is invalid", *argv);
 			fprintf(stderr, "IFE type 0x%x\n", ife_type);
 		} else if (matches(*argv, "dst") == 0) {
 			NEXT_ARG();
 			daddr = *argv;
 			if (sscanf(daddr, "%hhx:%hhx:%hhx:%hhx:%hhx:%hhx",
 				   dbuf, dbuf + 1, dbuf + 2,
 				   dbuf + 3, dbuf + 4, dbuf + 5) != 6) {
 				fprintf(stderr, "Invalid mac address %s\n",
 					daddr);
 			}
 			fprintf(stderr, "dst MAC address <%s>\n", daddr);

 		} else if (matches(*argv, "src") == 0) {
 			NEXT_ARG();
 			saddr = *argv;
 			if (sscanf(saddr, "%hhx:%hhx:%hhx:%hhx:%hhx:%hhx",
 				   sbuf, sbuf + 1, sbuf + 2,
 				   sbuf + 3, sbuf + 4, sbuf + 5) != 6) {
 				fprintf(stderr, "Invalid mac address %s\n",
 					saddr);
 			}
 			fprintf(stderr, "src MAC address <%s>\n", saddr);
 		} else if (matches(*argv, "help") == 0) {
 			ife_usage();
 		} else {
 			break;
 		}

 		argc--;
 		argv++;
 	}

 	if (argc && !action_a2n(*argv, &p.action, false))
 		NEXT_ARG_FWD();

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

 	if (!ok) {
 		fprintf(stderr, "IFE requires decode/encode specified\n");
 		ife_usage();
 	}

 	tail = NLMSG_TAIL(n);
 	addattr_l(n, MAX_MSG, tca_id, NULL, 0);
 	addattr_l(n, MAX_MSG, TCA_IFE_PARMS, &p, sizeof(p));

 	if (!(p.flags & IFE_ENCODE))
 		goto skip_encode;

 	if (daddr)
 		addattr_l(n, MAX_MSG, TCA_IFE_DMAC, dbuf, ETH_ALEN);
 	if (ife_type)
 		addattr_l(n, MAX_MSG, TCA_IFE_TYPE, &ife_type, 2);
 	if (saddr)
 		addattr_l(n, MAX_MSG, TCA_IFE_SMAC, sbuf, ETH_ALEN);

 	tail2 = NLMSG_TAIL(n);
 	addattr_l(n, MAX_MSG, TCA_IFE_METALST, NULL, 0);
 	if (ife_mark || ife_mark_v) {
 		if (ife_mark_v)
 			addattr_l(n, MAX_MSG, IFE_META_SKBMARK, &ife_mark_v, 4);
 		else
 			addattr_l(n, MAX_MSG, IFE_META_SKBMARK, NULL, 0);
 	}
 	if (ife_prio || ife_prio_v) {
 		if (ife_prio_v)
 			addattr_l(n, MAX_MSG, IFE_META_PRIO, &ife_prio_v, 4);
 		else
 			addattr_l(n, MAX_MSG, IFE_META_PRIO, NULL, 0);
 	}
 	if (ife_tcindex || ife_tcindex_v) {
 		if (ife_tcindex_v)
 			addattr_l(n, MAX_MSG, IFE_META_TCINDEX, &ife_tcindex_v,
 				  2);
 		else
 			addattr_l(n, MAX_MSG, IFE_META_TCINDEX, NULL, 0);
 	}

 	tail2->rta_len = (void *)NLMSG_TAIL(n) - (void *)tail2;

 skip_encode:
 	tail->rta_len = (void *)NLMSG_TAIL(n) - (void *)tail;

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

 static int print_ife(struct action_util *au, FILE *f, struct rtattr *arg)
 {
 	struct tc_ife *p = NULL;
 	struct rtattr *tb[TCA_IFE_MAX + 1];
 	__u16 ife_type = 0;
 	__u32 mmark = 0;
 	__u16 mtcindex = 0;
 	__u32 mprio = 0;
 	int has_optional = 0;
 	SPRINT_BUF(b2);

 	if (arg == NULL)
 		return -1;

 	parse_rtattr_nested(tb, TCA_IFE_MAX, arg);

 	if (tb[TCA_IFE_PARMS] == NULL) {
 		fprintf(f, "[NULL ife parameters]");
 		return -1;
 	}
 	p = RTA_DATA(tb[TCA_IFE_PARMS]);

 	fprintf(f, "ife %s action %s ",
 		(p->flags & IFE_ENCODE) ? "encode" : "decode",
 		action_n2a(p->action));

 	if (tb[TCA_IFE_TYPE]) {
 		ife_type = rta_getattr_u16(tb[TCA_IFE_TYPE]);
 		has_optional = 1;
 		fprintf(f, "type 0x%X ", ife_type);
 	}

 	if (has_optional)
 		fprintf(f, "\n\t ");

 	if (tb[TCA_IFE_METALST]) {
 		struct rtattr *metalist[IFE_META_MAX + 1];
 		int len = 0;

 		parse_rtattr_nested(metalist, IFE_META_MAX,
 				    tb[TCA_IFE_METALST]);

 		if (metalist[IFE_META_SKBMARK]) {
 			len = RTA_PAYLOAD(metalist[IFE_META_SKBMARK]);
 			if (len) {
 				mmark = rta_getattr_u32(metalist[IFE_META_SKBMARK]);
 				fprintf(f, "use mark %u ", mmark);
 			} else
 				fprintf(f, "allow mark ");
 		}

 		if (metalist[IFE_META_TCINDEX]) {
 			len = RTA_PAYLOAD(metalist[IFE_META_TCINDEX]);
 			if (len) {
 				mtcindex =
 					rta_getattr_u16(metalist[IFE_META_TCINDEX]);
 				fprintf(f, "use tcindex %d ", mtcindex);
 			} else
 				fprintf(f, "allow tcindex ");
 		}

 		if (metalist[IFE_META_PRIO]) {
 			len = RTA_PAYLOAD(metalist[IFE_META_PRIO]);
 			if (len) {
 				mprio = rta_getattr_u32(metalist[IFE_META_PRIO]);
 				fprintf(f, "use prio %u ", mprio);
 			} else
 				fprintf(f, "allow prio ");
 		}

 	}

 	if (tb[TCA_IFE_DMAC]) {
 		has_optional = 1;
 		fprintf(f, "dst %s ",
 			ll_addr_n2a(RTA_DATA(tb[TCA_IFE_DMAC]),
 				    RTA_PAYLOAD(tb[TCA_IFE_DMAC]), 0, b2,
 				    sizeof(b2)));

 	}

 	if (tb[TCA_IFE_SMAC]) {
 		has_optional = 1;
 		fprintf(f, "src %s ",
 			ll_addr_n2a(RTA_DATA(tb[TCA_IFE_SMAC]),
 				    RTA_PAYLOAD(tb[TCA_IFE_SMAC]), 0, b2,
 				    sizeof(b2)));
 	}

 	fprintf(f, "\n\t index %d ref %d bind %d", p->index, p->refcnt,
 		p->bindcnt);
 	if (show_stats) {
 		if (tb[TCA_IFE_TM]) {
 			struct tcf_t *tm = RTA_DATA(tb[TCA_IFE_TM]);

 			print_tm(f, tm);
 		}
 	}

 	fprintf(f, "\n");

 	return 0;
 }

 struct action_util ife_action_util = {
 	.id = "ife",
 	.parse_aopt = parse_ife,
 	.print_aopt = print_ife,
 };
	/*
	* m_ife.c IFE 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 (jhs@mojatatu.com)
	*
	*/

	#include <stdio.h>
	#include <stdlib.h>
	#include <unistd.h>
	#include <syslog.h>
	#include <fcntl.h>
	#include <sys/socket.h>
	#include <netinet/in.h>
	#include <arpa/inet.h>
	#include <string.h>
	#include <linux/netdevice.h>

	#include "rt_names.h"
	#include "utils.h"
	#include "tc_util.h"
	#include <linux/tc_act/tc_ife.h>

	static void ife_explain(void)
	{
	fprintf(stderr,
	"Usage:... ife {decode\|encode} [{ALLOW\|USE} ATTR] [dst DMAC] [src SMAC] [type TYPE] [CONTROL] [index INDEX]\n");
	fprintf(stderr,
	"\tALLOW := Encode direction. Allows encoding specified metadata\n"
	"\t\t e.g \"allow mark\"\n"
	"\tUSE := Encode direction. Enforce Static encoding of specified metadata\n"
	"\t\t e.g \"use mark 0x12\"\n"
	"\tATTR := mark (32-bit), prio (32-bit), tcindex (16-bit)\n"
	"\tDMAC := 6 byte Destination MAC address to encode\n"
	"\tSMAC := optional 6 byte Source MAC address to encode\n"
	"\tTYPE := optional 16 bit ethertype to encode\n"
	"\tCONTROL := reclassify\|pipe\|drop\|continue\|ok\n"
	"\tINDEX := optional IFE table index value used\n");
	fprintf(stderr, "encode is used for sending IFE packets\n");
	fprintf(stderr, "decode is used for receiving IFE packets\n");
	}

	static void ife_usage(void)
	{
	ife_explain();
	exit(-1);
	}

	static int parse_ife(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;
	struct tc_ife p = { .action = TC_ACT_PIPE }; /* good default */
	struct rtattr *tail;
	struct rtattr *tail2;
	char dbuf[ETH_ALEN];
	char sbuf[ETH_ALEN];
	__u16 ife_type = 0;
	__u32 ife_prio = 0;
	__u32 ife_prio_v = 0;
	__u32 ife_mark = 0;
	__u32 ife_mark_v = 0;
	__u16 ife_tcindex = 0;
	__u16 ife_tcindex_v = 0;
	char *daddr = NULL;
	char *saddr = NULL;

	if (argc <= 0)
	return -1;

	while (argc > 0) {
	if (matches(*argv, "ife") == 0) {
	NEXT_ARG();
	continue;
	} else if (matches(*argv, "decode") == 0) {
	p.flags = IFE_DECODE; /* readability aid */
	ok++;
	} else if (matches(*argv, "encode") == 0) {
	p.flags = IFE_ENCODE;
	ok++;
	} else if (matches(*argv, "allow") == 0) {
	NEXT_ARG();
	if (matches(*argv, "mark") == 0) {
	ife_mark = IFE_META_SKBMARK;
	} else if (matches(*argv, "prio") == 0) {
	ife_prio = IFE_META_PRIO;
	} else if (matches(*argv, "tcindex") == 0) {
	ife_prio = IFE_META_TCINDEX;
	} else {
	fprintf(stderr, "Illegal meta define <%s>\n",
	*argv);
	return -1;
	}
	} else if (matches(*argv, "use") == 0) {
	NEXT_ARG();
	if (matches(*argv, "mark") == 0) {
	NEXT_ARG();
	if (get_u32(&ife_mark_v, *argv, 0))
	invarg("ife mark val is invalid",
	*argv);
	} else if (matches(*argv, "prio") == 0) {
	NEXT_ARG();
	if (get_u32(&ife_prio_v, *argv, 0))
	invarg("ife prio val is invalid",
	*argv);
	} else if (matches(*argv, "tcindex") == 0) {
	NEXT_ARG();
	if (get_u16(&ife_tcindex_v, *argv, 0))
	invarg("ife tcindex val is invalid",
	*argv);
	} else {
	fprintf(stderr, "Illegal meta use type <%s>\n",
	*argv);
	return -1;
	}
	} else if (matches(*argv, "type") == 0) {
	NEXT_ARG();
	if (get_u16(&ife_type, *argv, 0))
	invarg("ife type is invalid", *argv);
	fprintf(stderr, "IFE type 0x%x\n", ife_type);
	} else if (matches(*argv, "dst") == 0) {
	NEXT_ARG();
	daddr = *argv;
	if (sscanf(daddr, "%hhx:%hhx:%hhx:%hhx:%hhx:%hhx",
	dbuf, dbuf + 1, dbuf + 2,
	dbuf + 3, dbuf + 4, dbuf + 5) != 6) {
	fprintf(stderr, "Invalid mac address %s\n",
	daddr);
	}
	fprintf(stderr, "dst MAC address <%s>\n", daddr);

	} else if (matches(*argv, "src") == 0) {
	NEXT_ARG();
	saddr = *argv;
	if (sscanf(saddr, "%hhx:%hhx:%hhx:%hhx:%hhx:%hhx",
	sbuf, sbuf + 1, sbuf + 2,
	sbuf + 3, sbuf + 4, sbuf + 5) != 6) {
	fprintf(stderr, "Invalid mac address %s\n",
	saddr);
	}
	fprintf(stderr, "src MAC address <%s>\n", saddr);
	} else if (matches(*argv, "help") == 0) {
	ife_usage();
	} else {
	break;
	}

	argc--;
	argv++;
	}

	if (argc && !action_a2n(*argv, &p.action, false))
	NEXT_ARG_FWD();

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

	if (!ok) {
	fprintf(stderr, "IFE requires decode/encode specified\n");
	ife_usage();
	}

	tail = NLMSG_TAIL(n);
	addattr_l(n, MAX_MSG, tca_id, NULL, 0);
	addattr_l(n, MAX_MSG, TCA_IFE_PARMS, &p, sizeof(p));

	if (!(p.flags & IFE_ENCODE))
	goto skip_encode;

	if (daddr)
	addattr_l(n, MAX_MSG, TCA_IFE_DMAC, dbuf, ETH_ALEN);
	if (ife_type)
	addattr_l(n, MAX_MSG, TCA_IFE_TYPE, &ife_type, 2);
	if (saddr)
	addattr_l(n, MAX_MSG, TCA_IFE_SMAC, sbuf, ETH_ALEN);

	tail2 = NLMSG_TAIL(n);
	addattr_l(n, MAX_MSG, TCA_IFE_METALST, NULL, 0);
	if (ife_mark \|\| ife_mark_v) {
	if (ife_mark_v)
	addattr_l(n, MAX_MSG, IFE_META_SKBMARK, &ife_mark_v, 4);
	else
	addattr_l(n, MAX_MSG, IFE_META_SKBMARK, NULL, 0);
	}
	if (ife_prio \|\| ife_prio_v) {
	if (ife_prio_v)
	addattr_l(n, MAX_MSG, IFE_META_PRIO, &ife_prio_v, 4);
	else
	addattr_l(n, MAX_MSG, IFE_META_PRIO, NULL, 0);
	}
	if (ife_tcindex \|\| ife_tcindex_v) {
	if (ife_tcindex_v)
	addattr_l(n, MAX_MSG, IFE_META_TCINDEX, &ife_tcindex_v,
	2);
	else
	addattr_l(n, MAX_MSG, IFE_META_TCINDEX, NULL, 0);
	}

	tail2->rta_len = (void )NLMSG_TAIL(n) - (void )tail2;

	skip_encode:
	tail->rta_len = (void )NLMSG_TAIL(n) - (void )tail;

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

	static int print_ife(struct action_util au, FILE f, struct rtattr *arg)
	{
	struct tc_ife *p = NULL;
	struct rtattr *tb[TCA_IFE_MAX + 1];
	__u16 ife_type = 0;
	__u32 mmark = 0;
	__u16 mtcindex = 0;
	__u32 mprio = 0;
	int has_optional = 0;
	SPRINT_BUF(b2);

	if (arg == NULL)
	return -1;

	parse_rtattr_nested(tb, TCA_IFE_MAX, arg);

	if (tb[TCA_IFE_PARMS] == NULL) {
	fprintf(f, "[NULL ife parameters]");
	return -1;
	}
	p = RTA_DATA(tb[TCA_IFE_PARMS]);

	fprintf(f, "ife %s action %s ",
	(p->flags & IFE_ENCODE) ? "encode" : "decode",
	action_n2a(p->action));

	if (tb[TCA_IFE_TYPE]) {
	ife_type = rta_getattr_u16(tb[TCA_IFE_TYPE]);
	has_optional = 1;
	fprintf(f, "type 0x%X ", ife_type);
	}

	if (has_optional)
	fprintf(f, "\n\t ");

	if (tb[TCA_IFE_METALST]) {
	struct rtattr *metalist[IFE_META_MAX + 1];
	int len = 0;

	parse_rtattr_nested(metalist, IFE_META_MAX,
	tb[TCA_IFE_METALST]);

	if (metalist[IFE_META_SKBMARK]) {
	len = RTA_PAYLOAD(metalist[IFE_META_SKBMARK]);
	if (len) {
	mmark = rta_getattr_u32(metalist[IFE_META_SKBMARK]);
	fprintf(f, "use mark %u ", mmark);
	} else
	fprintf(f, "allow mark ");
	}

	if (metalist[IFE_META_TCINDEX]) {
	len = RTA_PAYLOAD(metalist[IFE_META_TCINDEX]);
	if (len) {
	mtcindex =
	rta_getattr_u16(metalist[IFE_META_TCINDEX]);
	fprintf(f, "use tcindex %d ", mtcindex);
	} else
	fprintf(f, "allow tcindex ");
	}

	if (metalist[IFE_META_PRIO]) {
	len = RTA_PAYLOAD(metalist[IFE_META_PRIO]);
	if (len) {
	mprio = rta_getattr_u32(metalist[IFE_META_PRIO]);
	fprintf(f, "use prio %u ", mprio);
	} else
	fprintf(f, "allow prio ");
	}

	}

	if (tb[TCA_IFE_DMAC]) {
	has_optional = 1;
	fprintf(f, "dst %s ",
	ll_addr_n2a(RTA_DATA(tb[TCA_IFE_DMAC]),
	RTA_PAYLOAD(tb[TCA_IFE_DMAC]), 0, b2,
	sizeof(b2)));

	}

	if (tb[TCA_IFE_SMAC]) {
	has_optional = 1;
	fprintf(f, "src %s ",
	ll_addr_n2a(RTA_DATA(tb[TCA_IFE_SMAC]),
	RTA_PAYLOAD(tb[TCA_IFE_SMAC]), 0, b2,
	sizeof(b2)));
	}

	fprintf(f, "\n\t index %d ref %d bind %d", p->index, p->refcnt,
	p->bindcnt);
	if (show_stats) {
	if (tb[TCA_IFE_TM]) {
	struct tcf_t *tm = RTA_DATA(tb[TCA_IFE_TM]);

	print_tm(f, tm);
	}
	}

	fprintf(f, "\n");

	return 0;
	}

	struct action_util ife_action_util = {
	.id = "ife",
	.parse_aopt = parse_ife,
	.print_aopt = print_ife,
	};