[mirror_iproute2.git] / tc / m_skbedit.c

/*
 * m_skbedit.c		SKB Editing module
 *
 * Copyright (c) 2008, Intel Corporation.
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see <http://www.gnu.org/licenses>.
 *
 * Authors:	Alexander Duyck <alexander.h.duyck@intel.com>
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include "utils.h"
#include "tc_util.h"
#include <linux/tc_act/tc_skbedit.h>
#include <linux/if_packet.h>

static void explain(void)
{
	fprintf(stderr, "Usage: ... skbedit <[QM] [PM] [MM] [PT] [IF]>\n"
		"QM = queue_mapping QUEUE_MAPPING\n"
		"PM = priority PRIORITY\n"
		"MM = mark MARK[/MASK]\n"
		"PT = ptype PACKETYPE\n"
		"IF = inheritdsfield\n"
		"PACKETYPE = is one of:\n"
		"  host, otherhost, broadcast, multicast\n"
		"QUEUE_MAPPING = device transmit queue to use\n"
		"PRIORITY = classID to assign to priority field\n"
		"MARK = firewall mark to set\n"
		"MASK = mask applied to firewall mark (0xffffffff by default)\n"
		"note: inheritdsfield maps DS field to skb->priority\n");
}

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

static int
parse_skbedit(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 rtattr *tail;
	unsigned int tmp;
	__u16 queue_mapping, ptype;
	__u32 flags = 0, priority, mark, mask;
	__u64 pure_flags = 0;
	struct tc_skbedit sel = { 0 };

	if (matches(*argv, "skbedit") != 0)
		return -1;

	NEXT_ARG();

	while (argc > 0) {
		if (matches(*argv, "queue_mapping") == 0) {
			flags |= SKBEDIT_F_QUEUE_MAPPING;
			NEXT_ARG();
			if (get_unsigned(&tmp, *argv, 10) || tmp > 65535) {
				fprintf(stderr, "Illegal queue_mapping\n");
				return -1;
			}
			queue_mapping = tmp;
			ok++;
		} else if (matches(*argv, "priority") == 0) {
			flags |= SKBEDIT_F_PRIORITY;
			NEXT_ARG();
			if (get_tc_classid(&priority, *argv)) {
				fprintf(stderr, "Illegal priority\n");
				return -1;
			}
			ok++;
		} else if (matches(*argv, "mark") == 0) {
			char *slash;

			NEXT_ARG();
			slash = strchr(*argv, '/');
			if (slash)
				*slash = '\0';

			flags |= SKBEDIT_F_MARK;
			if (get_u32(&mark, *argv, 0)) {
				fprintf(stderr, "Illegal mark\n");
				return -1;
			}

			if (slash) {
				if (get_u32(&mask, slash + 1, 0)) {
					fprintf(stderr, "Illegal mask\n");
					return -1;
				}
				flags |= SKBEDIT_F_MASK;
			}
			ok++;
		} else if (matches(*argv, "ptype") == 0) {

			NEXT_ARG();
			if (matches(*argv, "host") == 0) {
				ptype = PACKET_HOST;
			} else if (matches(*argv, "broadcast") == 0) {
				ptype = PACKET_BROADCAST;
			} else if (matches(*argv, "multicast") == 0) {
				ptype = PACKET_MULTICAST;
			} else if (matches(*argv, "otherhost") == 0) {
				ptype = PACKET_OTHERHOST;
			} else {
				fprintf(stderr, "Illegal ptype (%s)\n",
					*argv);
				return -1;
			}
			flags |= SKBEDIT_F_PTYPE;
			ok++;
		} else if (matches(*argv, "inheritdsfield") == 0) {
			pure_flags |= SKBEDIT_F_INHERITDSFIELD;
			ok++;
		} else if (matches(*argv, "help") == 0) {
			usage();
		} else {
			break;
		}
		argc--;
		argv++;
	}

	parse_action_control_dflt(&argc, &argv, &sel.action,
				  false, TC_ACT_PIPE);

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

	if (!ok) {
		explain();
		return -1;
	}


	tail = addattr_nest(n, MAX_MSG, tca_id);
	addattr_l(n, MAX_MSG, TCA_SKBEDIT_PARMS, &sel, sizeof(sel));
	if (flags & SKBEDIT_F_QUEUE_MAPPING)
		addattr_l(n, MAX_MSG, TCA_SKBEDIT_QUEUE_MAPPING,
			  &queue_mapping, sizeof(queue_mapping));
	if (flags & SKBEDIT_F_PRIORITY)
		addattr_l(n, MAX_MSG, TCA_SKBEDIT_PRIORITY,
			  &priority, sizeof(priority));
	if (flags & SKBEDIT_F_MARK)
		addattr_l(n, MAX_MSG, TCA_SKBEDIT_MARK,
			  &mark, sizeof(mark));
	if (flags & SKBEDIT_F_MASK)
		addattr_l(n, MAX_MSG, TCA_SKBEDIT_MASK,
			  &mask, sizeof(mask));
	if (flags & SKBEDIT_F_PTYPE)
		addattr_l(n, MAX_MSG, TCA_SKBEDIT_PTYPE,
			  &ptype, sizeof(ptype));
	if (pure_flags != 0)
		addattr64(n, MAX_MSG, TCA_SKBEDIT_FLAGS, pure_flags);
	addattr_nest_end(n, tail);

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

static int print_skbedit(struct action_util *au, FILE *f, struct rtattr *arg)
{
	struct rtattr *tb[TCA_SKBEDIT_MAX + 1];

	SPRINT_BUF(b1);
	__u32 priority;
	__u16 ptype;
	struct tc_skbedit *p;

	if (arg == NULL)
		return -1;

	parse_rtattr_nested(tb, TCA_SKBEDIT_MAX, arg);

	if (tb[TCA_SKBEDIT_PARMS] == NULL) {
		fprintf(stderr, "Missing skbedit parameters\n");
		return -1;
	}
	p = RTA_DATA(tb[TCA_SKBEDIT_PARMS]);

	print_string(PRINT_ANY, "kind", "%s ", "skbedit");

	if (tb[TCA_SKBEDIT_QUEUE_MAPPING] != NULL) {
		print_uint(PRINT_ANY, "queue_mapping", "queue_mapping %u",
			   rta_getattr_u16(tb[TCA_SKBEDIT_QUEUE_MAPPING]));
	}
	if (tb[TCA_SKBEDIT_PRIORITY] != NULL) {
		priority = rta_getattr_u32(tb[TCA_SKBEDIT_PRIORITY]);
		print_string(PRINT_ANY, "priority", " priority %s",
			     sprint_tc_classid(priority, b1));
	}
	if (tb[TCA_SKBEDIT_MARK] != NULL) {
		print_uint(PRINT_ANY, "mark", " mark %u",
			   rta_getattr_u32(tb[TCA_SKBEDIT_MARK]));
	}
	if (tb[TCA_SKBEDIT_MASK]) {
		print_hex(PRINT_ANY, "mask", "/%#x",
			  rta_getattr_u32(tb[TCA_SKBEDIT_MASK]));
	}
	if (tb[TCA_SKBEDIT_PTYPE] != NULL) {
		ptype = rta_getattr_u16(tb[TCA_SKBEDIT_PTYPE]);
		if (ptype == PACKET_HOST)
			print_string(PRINT_ANY, "ptype", " ptype %s", "host");
		else if (ptype == PACKET_BROADCAST)
			print_string(PRINT_ANY, "ptype", " ptype %s",
				     "broadcast");
		else if (ptype == PACKET_MULTICAST)
			print_string(PRINT_ANY, "ptype", " ptype %s",
				     "multicast");
		else if (ptype == PACKET_OTHERHOST)
			print_string(PRINT_ANY, "ptype", " ptype %s",
				     "otherhost");
		else
			print_uint(PRINT_ANY, "ptype", " ptype %u", ptype);
	}
	if (tb[TCA_SKBEDIT_FLAGS] != NULL) {
		__u64 flags = rta_getattr_u64(tb[TCA_SKBEDIT_FLAGS]);

		if (flags & SKBEDIT_F_INHERITDSFIELD)
			print_null(PRINT_ANY, "inheritdsfield", " %s",
				     "inheritdsfield");
	}

	print_action_control(f, " ", p->action, "");

	print_string(PRINT_FP, NULL, "%s", _SL_);
	print_uint(PRINT_ANY, "index", "\t index %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_SKBEDIT_TM]) {
			struct tcf_t *tm = RTA_DATA(tb[TCA_SKBEDIT_TM]);

			print_tm(f, tm);
		}
	}

	print_string(PRINT_FP, NULL, "%s", _SL_);

	return 0;
}

struct action_util skbedit_action_util = {
	.id = "skbedit",
	.parse_aopt = parse_skbedit,
	.print_aopt = print_skbedit,
};
Commit	Line	Data
f72a7aab AD	1	/*
	2	* m_skbedit.c SKB Editing module
	3	*
	4	* Copyright (c) 2008, Intel Corporation.
	5	*
	6	* This program is free software; you can redistribute it and/or modify it
	7	* under the terms and conditions of the GNU General Public License,
	8	* version 2, as published by the Free Software Foundation.
	9	*
	10	* This program is distributed in the hope it will be useful, but WITHOUT
	11	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	12	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
	13	* more details.
	14	*
4d98ab00 SH	15	* You should have received a copy of the GNU General Public License
4d98ab00 SH	16	* along with this program; if not, see <http://www.gnu.org/licenses>.
f72a7aab AD	17	*
	18	* Authors: Alexander Duyck <alexander.h.duyck@intel.com>
	19	*
	20	*/
	21
	22	#include <stdio.h>
	23	#include <stdlib.h>
	24	#include <unistd.h>
	25	#include <string.h>
	26	#include "utils.h"
	27	#include "tc_util.h"
	28	#include <linux/tc_act/tc_skbedit.h>
1d1e0fd2	29	#include <linux/if_packet.h>
f72a7aab	30
1d1e0fd2	31	static void explain(void)
f72a7aab	32	{
697dce7b	33	fprintf(stderr, "Usage: ... skbedit <[QM] [PM] [MM] [PT] [IF]>\n"
e04dd30a	34	"QM = queue_mapping QUEUE_MAPPING\n"
32a121cb	35	"PM = priority PRIORITY\n"
82f3df20	36	"MM = mark MARK[/MASK]\n"
1d1e0fd2	37	"PT = ptype PACKETYPE\n"
697dce7b	38	"IF = inheritdsfield\n"
1d1e0fd2 JHS	39	"PACKETYPE = is one of:\n"
1d1e0fd2 JHS	40	" host, otherhost, broadcast, multicast\n"
e04dd30a JHS	41	"QUEUE_MAPPING = device transmit queue to use\n"
e04dd30a JHS	42	"PRIORITY = classID to assign to priority field\n"
697dce7b	43	"MARK = firewall mark to set\n"
82f3df20	44	"MASK = mask applied to firewall mark (0xffffffff by default)\n"
697dce7b	45	"note: inheritdsfield maps DS field to skb->priority\n");
f72a7aab AD	46	}
	47
	48	static void
	49	usage(void)
	50	{
	51	explain();
	52	exit(-1);
	53	}
	54
	55	static int
	56	parse_skbedit(struct action_util a, int argc_p, char ***argv_p, int tca_id,
	57	struct nlmsghdr *n)
	58	{
f72a7aab AD	59	int argc = *argc_p;
	60	char *argv = argv_p;
	61	int ok = 0;
	62	struct rtattr *tail;
	63	unsigned int tmp;
1d1e0fd2	64	__u16 queue_mapping, ptype;
82f3df20	65	__u32 flags = 0, priority, mark, mask;
697dce7b	66	__u64 pure_flags = 0;
46a65732	67	struct tc_skbedit sel = { 0 };
f72a7aab AD	68
	69	if (matches(*argv, "skbedit") != 0)
	70	return -1;
	71
	72	NEXT_ARG();
	73
	74	while (argc > 0) {
	75	if (matches(*argv, "queue_mapping") == 0) {
	76	flags \|= SKBEDIT_F_QUEUE_MAPPING;
	77	NEXT_ARG();
	78	if (get_unsigned(&tmp, *argv, 10) \|\| tmp > 65535) {
	79	fprintf(stderr, "Illegal queue_mapping\n");
	80	return -1;
	81	}
	82	queue_mapping = tmp;
	83	ok++;
	84	} else if (matches(*argv, "priority") == 0) {
	85	flags \|= SKBEDIT_F_PRIORITY;
	86	NEXT_ARG();
	87	if (get_tc_classid(&priority, *argv)) {
	88	fprintf(stderr, "Illegal priority\n");
	89	return -1;
	90	}
	91	ok++;
e04dd30a	92	} else if (matches(*argv, "mark") == 0) {
82f3df20 RM	93	char *slash;
82f3df20 RM	94
e04dd30a	95	NEXT_ARG();
82f3df20 RM	96	slash = strchr(*argv, '/');
	97	if (slash)
	98	*slash = '\0';
	99
	100	flags \|= SKBEDIT_F_MARK;
e906975a	101	if (get_u32(&mark, *argv, 0)) {
e04dd30a JHS	102	fprintf(stderr, "Illegal mark\n");
	103	return -1;
	104	}
82f3df20 RM	105
	106	if (slash) {
	107	if (get_u32(&mask, slash + 1, 0)) {
	108	fprintf(stderr, "Illegal mask\n");
	109	return -1;
	110	}
	111	flags \|= SKBEDIT_F_MASK;
	112	}
e04dd30a	113	ok++;
1d1e0fd2 JHS	114	} else if (matches(*argv, "ptype") == 0) {
	115
	116	NEXT_ARG();
	117	if (matches(*argv, "host") == 0) {
	118	ptype = PACKET_HOST;
	119	} else if (matches(*argv, "broadcast") == 0) {
	120	ptype = PACKET_BROADCAST;
	121	} else if (matches(*argv, "multicast") == 0) {
	122	ptype = PACKET_MULTICAST;
	123	} else if (matches(*argv, "otherhost") == 0) {
	124	ptype = PACKET_OTHERHOST;
	125	} else {
	126	fprintf(stderr, "Illegal ptype (%s)\n",
	127	*argv);
	128	return -1;
	129	}
	130	flags \|= SKBEDIT_F_PTYPE;
	131	ok++;
697dce7b QF	132	} else if (matches(*argv, "inheritdsfield") == 0) {
	133	pure_flags \|= SKBEDIT_F_INHERITDSFIELD;
	134	ok++;
f72a7aab AD	135	} else if (matches(*argv, "help") == 0) {
	136	usage();
	137	} else {
	138	break;
	139	}
	140	argc--;
	141	argv++;
	142	}
	143
e67aba55 JP	144	parse_action_control_dflt(&argc, &argv, &sel.action,
e67aba55 JP	145	false, TC_ACT_PIPE);
f72a7aab AD	146
	147	if (argc) {
	148	if (matches(*argv, "index") == 0) {
	149	NEXT_ARG();
	150	if (get_u32(&sel.index, *argv, 10)) {
82f3df20	151	fprintf(stderr, "skbedit: Illegal \"index\"\n");
f72a7aab AD	152	return -1;
	153	}
	154	argc--;
	155	argv++;
	156	ok++;
	157	}
	158	}
	159
	160	if (!ok) {
	161	explain();
	162	return -1;
	163	}
	164
	165
c14f9d92	166	tail = addattr_nest(n, MAX_MSG, tca_id);
f72a7aab AD	167	addattr_l(n, MAX_MSG, TCA_SKBEDIT_PARMS, &sel, sizeof(sel));
	168	if (flags & SKBEDIT_F_QUEUE_MAPPING)
	169	addattr_l(n, MAX_MSG, TCA_SKBEDIT_QUEUE_MAPPING,
	170	&queue_mapping, sizeof(queue_mapping));
	171	if (flags & SKBEDIT_F_PRIORITY)
	172	addattr_l(n, MAX_MSG, TCA_SKBEDIT_PRIORITY,
	173	&priority, sizeof(priority));
e04dd30a JHS	174	if (flags & SKBEDIT_F_MARK)
	175	addattr_l(n, MAX_MSG, TCA_SKBEDIT_MARK,
	176	&mark, sizeof(mark));
82f3df20 RM	177	if (flags & SKBEDIT_F_MASK)
	178	addattr_l(n, MAX_MSG, TCA_SKBEDIT_MASK,
	179	&mask, sizeof(mask));
1d1e0fd2 JHS	180	if (flags & SKBEDIT_F_PTYPE)
	181	addattr_l(n, MAX_MSG, TCA_SKBEDIT_PTYPE,
	182	&ptype, sizeof(ptype));
697dce7b QF	183	if (pure_flags != 0)
697dce7b QF	184	addattr64(n, MAX_MSG, TCA_SKBEDIT_FLAGS, pure_flags);
c14f9d92	185	addattr_nest_end(n, tail);
f72a7aab AD	186
	187	*argc_p = argc;
	188	*argv_p = argv;
	189	return 0;
	190	}
	191
	192	static int print_skbedit(struct action_util au, FILE f, struct rtattr *arg)
	193	{
f72a7aab	194	struct rtattr *tb[TCA_SKBEDIT_MAX + 1];
32a121cb	195
f72a7aab	196	SPRINT_BUF(b1);
7b177017 RM	197	__u32 priority;
7b177017 RM	198	__u16 ptype;
d5ddb441	199	struct tc_skbedit *p;
f72a7aab AD	200
	201	if (arg == NULL)
	202	return -1;
	203
	204	parse_rtattr_nested(tb, TCA_SKBEDIT_MAX, arg);
	205
	206	if (tb[TCA_SKBEDIT_PARMS] == NULL) {
d5ddb441	207	fprintf(stderr, "Missing skbedit parameters\n");
f72a7aab AD	208	return -1;
f72a7aab AD	209	}
02b1d345	210	p = RTA_DATA(tb[TCA_SKBEDIT_PARMS]);
f72a7aab	211
7b177017	212	print_string(PRINT_ANY, "kind", "%s ", "skbedit");
f72a7aab AD	213
f72a7aab AD	214	if (tb[TCA_SKBEDIT_QUEUE_MAPPING] != NULL) {
7b177017 RM	215	print_uint(PRINT_ANY, "queue_mapping", "queue_mapping %u",
7b177017 RM	216	rta_getattr_u16(tb[TCA_SKBEDIT_QUEUE_MAPPING]));
f72a7aab AD	217	}
f72a7aab AD	218	if (tb[TCA_SKBEDIT_PRIORITY] != NULL) {
7b177017 RM	219	priority = rta_getattr_u32(tb[TCA_SKBEDIT_PRIORITY]);
	220	print_string(PRINT_ANY, "priority", " priority %s",
	221	sprint_tc_classid(priority, b1));
f72a7aab	222	}
e04dd30a	223	if (tb[TCA_SKBEDIT_MARK] != NULL) {
7b177017 RM	224	print_uint(PRINT_ANY, "mark", " mark %u",
7b177017 RM	225	rta_getattr_u32(tb[TCA_SKBEDIT_MARK]));
e04dd30a	226	}
82f3df20 RM	227	if (tb[TCA_SKBEDIT_MASK]) {
	228	print_hex(PRINT_ANY, "mask", "/%#x",
	229	rta_getattr_u32(tb[TCA_SKBEDIT_MASK]));
	230	}
1d1e0fd2	231	if (tb[TCA_SKBEDIT_PTYPE] != NULL) {
7b177017 RM	232	ptype = rta_getattr_u16(tb[TCA_SKBEDIT_PTYPE]);
	233	if (ptype == PACKET_HOST)
	234	print_string(PRINT_ANY, "ptype", " ptype %s", "host");
	235	else if (ptype == PACKET_BROADCAST)
	236	print_string(PRINT_ANY, "ptype", " ptype %s",
	237	"broadcast");
	238	else if (ptype == PACKET_MULTICAST)
	239	print_string(PRINT_ANY, "ptype", " ptype %s",
	240	"multicast");
	241	else if (ptype == PACKET_OTHERHOST)
	242	print_string(PRINT_ANY, "ptype", " ptype %s",
	243	"otherhost");
1d1e0fd2	244	else
7b177017	245	print_uint(PRINT_ANY, "ptype", " ptype %u", ptype);
1d1e0fd2	246	}
697dce7b QF	247	if (tb[TCA_SKBEDIT_FLAGS] != NULL) {
	248	__u64 flags = rta_getattr_u64(tb[TCA_SKBEDIT_FLAGS]);
	249
	250	if (flags & SKBEDIT_F_INHERITDSFIELD)
	251	print_null(PRINT_ANY, "inheritdsfield", " %s",
	252	"inheritdsfield");
	253	}
f72a7aab	254
e67aba55	255	print_action_control(f, " ", p->action, "");
878babff	256
7b177017 RM	257	print_string(PRINT_FP, NULL, "%s", _SL_);
	258	print_uint(PRINT_ANY, "index", "\t index %u", p->index);
	259	print_int(PRINT_ANY, "ref", " ref %d", p->refcnt);
	260	print_int(PRINT_ANY, "bind", " bind %d", p->bindcnt);
02b1d345	261
f72a7aab AD	262	if (show_stats) {
	263	if (tb[TCA_SKBEDIT_TM]) {
	264	struct tcf_t *tm = RTA_DATA(tb[TCA_SKBEDIT_TM]);
32a121cb	265
f72a7aab AD	266	print_tm(f, tm);
	267	}
	268	}
	269
7b177017	270	print_string(PRINT_FP, NULL, "%s", _SL_);
02b1d345	271
f72a7aab AD	272	return 0;
	273	}
	274
	275	struct action_util skbedit_action_util = {
	276	.id = "skbedit",
	277	.parse_aopt = parse_skbedit,
	278	.print_aopt = print_skbedit,
	279	};