Blame tc/m_skbedit.c

Packit Service 3880ab
/*
Packit Service 3880ab
 * m_skbedit.c		SKB Editing module
Packit Service 3880ab
 *
Packit Service 3880ab
 * Copyright (c) 2008, Intel Corporation.
Packit Service 3880ab
 *
Packit Service 3880ab
 * This program is free software; you can redistribute it and/or modify it
Packit Service 3880ab
 * under the terms and conditions of the GNU General Public License,
Packit Service 3880ab
 * version 2, as published by the Free Software Foundation.
Packit Service 3880ab
 *
Packit Service 3880ab
 * This program is distributed in the hope it will be useful, but WITHOUT
Packit Service 3880ab
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
Packit Service 3880ab
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
Packit Service 3880ab
 * more details.
Packit Service 3880ab
 *
Packit Service 3880ab
 * You should have received a copy of the GNU General Public License
Packit Service 3880ab
 * along with this program; if not, see <http://www.gnu.org/licenses>.
Packit Service 3880ab
 *
Packit Service 3880ab
 * Authors:	Alexander Duyck <alexander.h.duyck@intel.com>
Packit Service 3880ab
 *
Packit Service 3880ab
 */
Packit Service 3880ab
Packit Service 3880ab
#include <stdio.h>
Packit Service 3880ab
#include <stdlib.h>
Packit Service 3880ab
#include <unistd.h>
Packit Service 3880ab
#include <string.h>
Packit Service 3880ab
#include "utils.h"
Packit Service 3880ab
#include "tc_util.h"
Packit Service 3880ab
#include <linux/tc_act/tc_skbedit.h>
Packit Service 3880ab
#include <linux/if_packet.h>
Packit Service 3880ab
Packit Service 3880ab
static void explain(void)
Packit Service 3880ab
{
Packit Service 3880ab
	fprintf(stderr, "Usage: ... skbedit <[QM] [PM] [MM] [PT] [IF]>\n"
Packit Service 3880ab
		"QM = queue_mapping QUEUE_MAPPING\n"
Packit Service 3880ab
		"PM = priority PRIORITY\n"
Packit Service 3880ab
		"MM = mark MARK[/MASK]\n"
Packit Service 3880ab
		"PT = ptype PACKETYPE\n"
Packit Service 3880ab
		"IF = inheritdsfield\n"
Packit Service 3880ab
		"PACKETYPE = is one of:\n"
Packit Service 3880ab
		"  host, otherhost, broadcast, multicast\n"
Packit Service 3880ab
		"QUEUE_MAPPING = device transmit queue to use\n"
Packit Service 3880ab
		"PRIORITY = classID to assign to priority field\n"
Packit Service 3880ab
		"MARK = firewall mark to set\n"
Packit Service 3880ab
		"MASK = mask applied to firewall mark (0xffffffff by default)\n"
Packit Service 3880ab
		"note: inheritdsfield maps DS field to skb->priority\n");
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static void
Packit Service 3880ab
usage(void)
Packit Service 3880ab
{
Packit Service 3880ab
	explain();
Packit Service 3880ab
	exit(-1);
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int
Packit Service 3880ab
parse_skbedit(struct action_util *a, int *argc_p, char ***argv_p, int tca_id,
Packit Service 3880ab
	      struct nlmsghdr *n)
Packit Service 3880ab
{
Packit Service 3880ab
	int argc = *argc_p;
Packit Service 3880ab
	char **argv = *argv_p;
Packit Service 3880ab
	int ok = 0;
Packit Service 3880ab
	struct rtattr *tail;
Packit Service 3880ab
	unsigned int tmp;
Packit Service 3880ab
	__u16 queue_mapping, ptype;
Packit Service 3880ab
	__u32 flags = 0, priority, mark, mask;
Packit Service 3880ab
	__u64 pure_flags = 0;
Packit Service 3880ab
	struct tc_skbedit sel = { 0 };
Packit Service 3880ab
Packit Service 3880ab
	if (matches(*argv, "skbedit") != 0)
Packit Service 3880ab
		return -1;
Packit Service 3880ab
Packit Service 3880ab
	NEXT_ARG();
Packit Service 3880ab
Packit Service 3880ab
	while (argc > 0) {
Packit Service 3880ab
		if (matches(*argv, "queue_mapping") == 0) {
Packit Service 3880ab
			flags |= SKBEDIT_F_QUEUE_MAPPING;
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			if (get_unsigned(&tmp, *argv, 10) || tmp > 65535) {
Packit Service 3880ab
				fprintf(stderr, "Illegal queue_mapping\n");
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
			queue_mapping = tmp;
Packit Service 3880ab
			ok++;
Packit Service 3880ab
		} else if (matches(*argv, "priority") == 0) {
Packit Service 3880ab
			flags |= SKBEDIT_F_PRIORITY;
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			if (get_tc_classid(&priority, *argv)) {
Packit Service 3880ab
				fprintf(stderr, "Illegal priority\n");
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
			ok++;
Packit Service 3880ab
		} else if (matches(*argv, "mark") == 0) {
Packit Service 3880ab
			char *slash;
Packit Service 3880ab
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			slash = strchr(*argv, '/');
Packit Service 3880ab
			if (slash)
Packit Service 3880ab
				*slash = '\0';
Packit Service 3880ab
Packit Service 3880ab
			flags |= SKBEDIT_F_MARK;
Packit Service 3880ab
			if (get_u32(&mark, *argv, 0)) {
Packit Service 3880ab
				fprintf(stderr, "Illegal mark\n");
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
Packit Service 3880ab
			if (slash) {
Packit Service 3880ab
				if (get_u32(&mask, slash + 1, 0)) {
Packit Service 3880ab
					fprintf(stderr, "Illegal mask\n");
Packit Service 3880ab
					return -1;
Packit Service 3880ab
				}
Packit Service 3880ab
				flags |= SKBEDIT_F_MASK;
Packit Service 3880ab
			}
Packit Service 3880ab
			ok++;
Packit Service 3880ab
		} else if (matches(*argv, "ptype") == 0) {
Packit Service 3880ab
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			if (matches(*argv, "host") == 0) {
Packit Service 3880ab
				ptype = PACKET_HOST;
Packit Service 3880ab
			} else if (matches(*argv, "broadcast") == 0) {
Packit Service 3880ab
				ptype = PACKET_BROADCAST;
Packit Service 3880ab
			} else if (matches(*argv, "multicast") == 0) {
Packit Service 3880ab
				ptype = PACKET_MULTICAST;
Packit Service 3880ab
			} else if (matches(*argv, "otherhost") == 0) {
Packit Service 3880ab
				ptype = PACKET_OTHERHOST;
Packit Service 3880ab
			} else {
Packit Service 3880ab
				fprintf(stderr, "Illegal ptype (%s)\n",
Packit Service 3880ab
					*argv);
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
			flags |= SKBEDIT_F_PTYPE;
Packit Service 3880ab
			ok++;
Packit Service 3880ab
		} else if (matches(*argv, "inheritdsfield") == 0) {
Packit Service 3880ab
			pure_flags |= SKBEDIT_F_INHERITDSFIELD;
Packit Service 3880ab
			ok++;
Packit Service 3880ab
		} else if (matches(*argv, "help") == 0) {
Packit Service 3880ab
			usage();
Packit Service 3880ab
		} else {
Packit Service 3880ab
			break;
Packit Service 3880ab
		}
Packit Service 3880ab
		argc--;
Packit Service 3880ab
		argv++;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	parse_action_control_dflt(&argc, &argv, &sel.action,
Packit Service 3880ab
				  false, TC_ACT_PIPE);
Packit Service 3880ab
Packit Service 3880ab
	if (argc) {
Packit Service 3880ab
		if (matches(*argv, "index") == 0) {
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			if (get_u32(&sel.index, *argv, 10)) {
Packit Service 3880ab
				fprintf(stderr, "skbedit: Illegal \"index\"\n");
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
			argc--;
Packit Service 3880ab
			argv++;
Packit Service 3880ab
			ok++;
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (!ok) {
Packit Service 3880ab
		explain();
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
Packit Service 3880ab
	tail = addattr_nest(n, MAX_MSG, tca_id);
Packit Service 3880ab
	addattr_l(n, MAX_MSG, TCA_SKBEDIT_PARMS, &sel, sizeof(sel));
Packit Service 3880ab
	if (flags & SKBEDIT_F_QUEUE_MAPPING)
Packit Service 3880ab
		addattr_l(n, MAX_MSG, TCA_SKBEDIT_QUEUE_MAPPING,
Packit Service 3880ab
			  &queue_mapping, sizeof(queue_mapping));
Packit Service 3880ab
	if (flags & SKBEDIT_F_PRIORITY)
Packit Service 3880ab
		addattr_l(n, MAX_MSG, TCA_SKBEDIT_PRIORITY,
Packit Service 3880ab
			  &priority, sizeof(priority));
Packit Service 3880ab
	if (flags & SKBEDIT_F_MARK)
Packit Service 3880ab
		addattr_l(n, MAX_MSG, TCA_SKBEDIT_MARK,
Packit Service 3880ab
			  &mark, sizeof(mark));
Packit Service 3880ab
	if (flags & SKBEDIT_F_MASK)
Packit Service 3880ab
		addattr_l(n, MAX_MSG, TCA_SKBEDIT_MASK,
Packit Service 3880ab
			  &mask, sizeof(mask));
Packit Service 3880ab
	if (flags & SKBEDIT_F_PTYPE)
Packit Service 3880ab
		addattr_l(n, MAX_MSG, TCA_SKBEDIT_PTYPE,
Packit Service 3880ab
			  &ptype, sizeof(ptype));
Packit Service 3880ab
	if (pure_flags != 0)
Packit Service 3880ab
		addattr64(n, MAX_MSG, TCA_SKBEDIT_FLAGS, pure_flags);
Packit Service 3880ab
	addattr_nest_end(n, tail);
Packit Service 3880ab
Packit Service 3880ab
	*argc_p = argc;
Packit Service 3880ab
	*argv_p = argv;
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int print_skbedit(struct action_util *au, FILE *f, struct rtattr *arg)
Packit Service 3880ab
{
Packit Service 3880ab
	struct rtattr *tb[TCA_SKBEDIT_MAX + 1];
Packit Service 3880ab
Packit Service 3880ab
	SPRINT_BUF(b1);
Packit Service 3880ab
	__u32 priority;
Packit Service 3880ab
	__u16 ptype;
Packit Service 3880ab
	struct tc_skbedit *p;
Packit Service 3880ab
Packit Service 3880ab
	if (arg == NULL)
Packit Service 3880ab
		return -1;
Packit Service 3880ab
Packit Service 3880ab
	parse_rtattr_nested(tb, TCA_SKBEDIT_MAX, arg);
Packit Service 3880ab
Packit Service 3880ab
	if (tb[TCA_SKBEDIT_PARMS] == NULL) {
Packit Service 3880ab
		fprintf(stderr, "Missing skbedit parameters\n");
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
	p = RTA_DATA(tb[TCA_SKBEDIT_PARMS]);
Packit Service 3880ab
Packit Service 3880ab
	print_string(PRINT_ANY, "kind", "%s ", "skbedit");
Packit Service 3880ab
Packit Service 3880ab
	if (tb[TCA_SKBEDIT_QUEUE_MAPPING] != NULL) {
Packit Service 3880ab
		print_uint(PRINT_ANY, "queue_mapping", "queue_mapping %u",
Packit Service 3880ab
			   rta_getattr_u16(tb[TCA_SKBEDIT_QUEUE_MAPPING]));
Packit Service 3880ab
	}
Packit Service 3880ab
	if (tb[TCA_SKBEDIT_PRIORITY] != NULL) {
Packit Service 3880ab
		priority = rta_getattr_u32(tb[TCA_SKBEDIT_PRIORITY]);
Packit Service 3880ab
		print_string(PRINT_ANY, "priority", " priority %s",
Packit Service 3880ab
			     sprint_tc_classid(priority, b1));
Packit Service 3880ab
	}
Packit Service 3880ab
	if (tb[TCA_SKBEDIT_MARK] != NULL) {
Packit Service 3880ab
		print_uint(PRINT_ANY, "mark", " mark %u",
Packit Service 3880ab
			   rta_getattr_u32(tb[TCA_SKBEDIT_MARK]));
Packit Service 3880ab
	}
Packit Service 3880ab
	if (tb[TCA_SKBEDIT_MASK]) {
Packit Service 3880ab
		print_hex(PRINT_ANY, "mask", "/%#x",
Packit Service 3880ab
			  rta_getattr_u32(tb[TCA_SKBEDIT_MASK]));
Packit Service 3880ab
	}
Packit Service 3880ab
	if (tb[TCA_SKBEDIT_PTYPE] != NULL) {
Packit Service 3880ab
		ptype = rta_getattr_u16(tb[TCA_SKBEDIT_PTYPE]);
Packit Service 3880ab
		if (ptype == PACKET_HOST)
Packit Service 3880ab
			print_string(PRINT_ANY, "ptype", " ptype %s", "host");
Packit Service 3880ab
		else if (ptype == PACKET_BROADCAST)
Packit Service 3880ab
			print_string(PRINT_ANY, "ptype", " ptype %s",
Packit Service 3880ab
				     "broadcast");
Packit Service 3880ab
		else if (ptype == PACKET_MULTICAST)
Packit Service 3880ab
			print_string(PRINT_ANY, "ptype", " ptype %s",
Packit Service 3880ab
				     "multicast");
Packit Service 3880ab
		else if (ptype == PACKET_OTHERHOST)
Packit Service 3880ab
			print_string(PRINT_ANY, "ptype", " ptype %s",
Packit Service 3880ab
				     "otherhost");
Packit Service 3880ab
		else
Packit Service 3880ab
			print_uint(PRINT_ANY, "ptype", " ptype %u", ptype);
Packit Service 3880ab
	}
Packit Service 3880ab
	if (tb[TCA_SKBEDIT_FLAGS] != NULL) {
Packit Service 3880ab
		__u64 flags = rta_getattr_u64(tb[TCA_SKBEDIT_FLAGS]);
Packit Service 3880ab
Packit Service 3880ab
		if (flags & SKBEDIT_F_INHERITDSFIELD)
Packit Service 3880ab
			print_null(PRINT_ANY, "inheritdsfield", " %s",
Packit Service 3880ab
				     "inheritdsfield");
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	print_action_control(f, " ", p->action, "");
Packit Service 3880ab
Packit Service 3880ab
	print_nl();
Packit Service 3880ab
	print_uint(PRINT_ANY, "index", "\t index %u", p->index);
Packit Service 3880ab
	print_int(PRINT_ANY, "ref", " ref %d", p->refcnt);
Packit Service 3880ab
	print_int(PRINT_ANY, "bind", " bind %d", p->bindcnt);
Packit Service 3880ab
Packit Service 3880ab
	if (show_stats) {
Packit Service 3880ab
		if (tb[TCA_SKBEDIT_TM]) {
Packit Service 3880ab
			struct tcf_t *tm = RTA_DATA(tb[TCA_SKBEDIT_TM]);
Packit Service 3880ab
Packit Service 3880ab
			print_tm(f, tm);
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	print_nl();
Packit Service 3880ab
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
struct action_util skbedit_action_util = {
Packit Service 3880ab
	.id = "skbedit",
Packit Service 3880ab
	.parse_aopt = parse_skbedit,
Packit Service 3880ab
	.print_aopt = print_skbedit,
Packit Service 3880ab
};