[mirror_iproute2.git] / tc / q_dsmark.c

/* SPDX-License-Identifier: GPL-2.0 */
/*
 * q_dsmark.c		Differentiated Services field marking.
 *
 * Hacked 1998,1999 by Werner Almesberger, EPFL ICA
 *
 */

#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"


static void explain(void)
{
	fprintf(stderr,"Usage: dsmark indices INDICES [ default_index DEFAULT_INDEX ] [ set_tc_index ]\n");
}


static int dsmark_parse_opt(struct qdisc_util *qu, int argc, char **argv,
	struct nlmsghdr *n, const char *dev)
{
	struct rtattr *tail;
	__u16 ind;
	char *end;
	int dflt, set_tc_index;

	ind = set_tc_index = 0;
	dflt = -1;
	while (argc > 0) {
		if (!strcmp(*argv, "indices")) {
			NEXT_ARG();
			ind = strtoul(*argv, &end, 0);
			if (*end) {
				explain();
				return -1;
			}
		} else if (!strcmp(*argv,"default_index") || !strcmp(*argv,
		    "default")) {
			NEXT_ARG();
			dflt = strtoul(*argv, &end, 0);
			if (*end) {
				explain();
				return -1;
			}
		} else if (!strcmp(*argv,"set_tc_index")) {
			set_tc_index = 1;
		} else {
			explain();
			return -1;
		}
		argc--;
		argv++;
	}
	if (!ind) {
		explain();
		return -1;
	}
	tail = addattr_nest(n, 1024, TCA_OPTIONS);
	addattr_l(n, 1024, TCA_DSMARK_INDICES, &ind, sizeof(ind));
	if (dflt != -1) {
	    __u16 tmp = dflt;

	    addattr_l(n, 1024, TCA_DSMARK_DEFAULT_INDEX, &tmp, sizeof(tmp));
	}
	if (set_tc_index)
		addattr_l(n, 1024, TCA_DSMARK_SET_TC_INDEX, NULL, 0);
	addattr_nest_end(n, tail);
	return 0;
}


static void explain_class(void)
{
	fprintf(stderr, "Usage: ... dsmark [ mask MASK ] [ value VALUE ]\n");
}


static int dsmark_parse_class_opt(struct qdisc_util *qu, int argc, char **argv,
	struct nlmsghdr *n, const char *dev)
{
	struct rtattr *tail;
	__u8 tmp;
	char *end;

	tail = addattr_nest(n, 1024, TCA_OPTIONS);
	while (argc > 0) {
		if (!strcmp(*argv, "mask")) {
			NEXT_ARG();
			tmp = strtoul(*argv, &end, 0);
			if (*end) {
				explain_class();
				return -1;
			}
			addattr_l(n, 1024, TCA_DSMARK_MASK, &tmp, 1);
		} else if (!strcmp(*argv,"value")) {
			NEXT_ARG();
			tmp = strtoul(*argv, &end, 0);
			if (*end) {
				explain_class();
				return -1;
			}
			addattr_l(n, 1024, TCA_DSMARK_VALUE, &tmp, 1);
		} else {
			explain_class();
			return -1;
		}
		argc--;
		argv++;
	}
	addattr_nest_end(n, tail);
	return 0;
}


static int dsmark_print_opt(struct qdisc_util *qu, FILE *f, struct rtattr *opt)
{
	struct rtattr *tb[TCA_DSMARK_MAX+1];

	if (!opt) return 0;
	parse_rtattr(tb, TCA_DSMARK_MAX, RTA_DATA(opt), RTA_PAYLOAD(opt));
	if (tb[TCA_DSMARK_MASK]) {
		if (!RTA_PAYLOAD(tb[TCA_DSMARK_MASK]))
			fprintf(stderr, "dsmark: empty mask\n");
		else fprintf(f, "mask 0x%02x ",
			    rta_getattr_u8(tb[TCA_DSMARK_MASK]));
	}
	if (tb[TCA_DSMARK_VALUE]) {
		if (!RTA_PAYLOAD(tb[TCA_DSMARK_VALUE]))
			fprintf(stderr, "dsmark: empty value\n");
		else fprintf(f, "value 0x%02x ",
			    rta_getattr_u8(tb[TCA_DSMARK_VALUE]));
	}
	if (tb[TCA_DSMARK_INDICES]) {
		if (RTA_PAYLOAD(tb[TCA_DSMARK_INDICES]) < sizeof(__u16))
			fprintf(stderr, "dsmark: indices too short\n");
		else fprintf(f, "indices 0x%04x ",
			    rta_getattr_u16(tb[TCA_DSMARK_INDICES]));
	}
	if (tb[TCA_DSMARK_DEFAULT_INDEX]) {
		if (RTA_PAYLOAD(tb[TCA_DSMARK_DEFAULT_INDEX]) < sizeof(__u16))
			fprintf(stderr, "dsmark: default_index too short\n");
		else fprintf(f, "default_index 0x%04x ",
			    rta_getattr_u16(tb[TCA_DSMARK_DEFAULT_INDEX]));
	}
	if (tb[TCA_DSMARK_SET_TC_INDEX]) fprintf(f, "set_tc_index ");
	return 0;
}


struct qdisc_util dsmark_qdisc_util = {
	.id		= "dsmark",
	.parse_qopt	= dsmark_parse_opt,
	.print_qopt	= dsmark_print_opt,
	.parse_copt	= dsmark_parse_class_opt,
	.print_copt	= dsmark_print_opt,
};
Commit	Line	Data
6054c1eb	1	/* SPDX-License-Identifier: GPL-2.0 */
aba5acdf SH	2	/*
	3	* q_dsmark.c Differentiated Services field marking.
	4	*
	5	* Hacked 1998,1999 by Werner Almesberger, EPFL ICA
	6	*
	7	*/
	8
	9	#include <stdio.h>
	10	#include <stdlib.h>
	11	#include <unistd.h>
aba5acdf SH	12	#include <fcntl.h>
	13	#include <sys/socket.h>
	14	#include <netinet/in.h>
	15	#include <arpa/inet.h>
	16	#include <string.h>
	17
	18	#include "utils.h"
	19	#include "tc_util.h"
	20
	21
aba5acdf SH	22	static void explain(void)
aba5acdf SH	23	{
32a121cb	24	fprintf(stderr,"Usage: dsmark indices INDICES [ default_index DEFAULT_INDEX ] [ set_tc_index ]\n");
aba5acdf SH	25	}
	26
	27
	28	static int dsmark_parse_opt(struct qdisc_util qu, int argc, char *argv,
927e3cfb	29	struct nlmsghdr n, const char dev)
aba5acdf SH	30	{
	31	struct rtattr *tail;
	32	__u16 ind;
	33	char *end;
32a121cb	34	int dflt, set_tc_index;
aba5acdf SH	35
	36	ind = set_tc_index = 0;
	37	dflt = -1;
	38	while (argc > 0) {
32a121cb	39	if (!strcmp(*argv, "indices")) {
aba5acdf	40	NEXT_ARG();
32a121cb	41	ind = strtoul(*argv, &end, 0);
aba5acdf SH	42	if (*end) {
	43	explain();
	44	return -1;
	45	}
32a121cb	46	} else if (!strcmp(argv,"default_index") \|\| !strcmp(argv,
aba5acdf SH	47	"default")) {
aba5acdf SH	48	NEXT_ARG();
32a121cb	49	dflt = strtoul(*argv, &end, 0);
aba5acdf SH	50	if (*end) {
	51	explain();
	52	return -1;
	53	}
32a121cb	54	} else if (!strcmp(*argv,"set_tc_index")) {
aba5acdf	55	set_tc_index = 1;
32a121cb	56	} else {
aba5acdf SH	57	explain();
	58	return -1;
	59	}
	60	argc--;
	61	argv++;
	62	}
	63	if (!ind) {
	64	explain();
	65	return -1;
	66	}
c14f9d92	67	tail = addattr_nest(n, 1024, TCA_OPTIONS);
32a121cb	68	addattr_l(n, 1024, TCA_DSMARK_INDICES, &ind, sizeof(ind));
aba5acdf SH	69	if (dflt != -1) {
	70	__u16 tmp = dflt;
	71
32a121cb	72	addattr_l(n, 1024, TCA_DSMARK_DEFAULT_INDEX, &tmp, sizeof(tmp));
aba5acdf	73	}
c14f9d92 SP	74	if (set_tc_index)
	75	addattr_l(n, 1024, TCA_DSMARK_SET_TC_INDEX, NULL, 0);
	76	addattr_nest_end(n, tail);
aba5acdf SH	77	return 0;
	78	}
	79
	80
	81	static void explain_class(void)
	82	{
	83	fprintf(stderr, "Usage: ... dsmark [ mask MASK ] [ value VALUE ]\n");
	84	}
	85
	86
	87	static int dsmark_parse_class_opt(struct qdisc_util qu, int argc, char *argv,
927e3cfb	88	struct nlmsghdr n, const char dev)
aba5acdf SH	89	{
	90	struct rtattr *tail;
	91	__u8 tmp;
	92	char *end;
	93
c14f9d92	94	tail = addattr_nest(n, 1024, TCA_OPTIONS);
aba5acdf	95	while (argc > 0) {
32a121cb	96	if (!strcmp(*argv, "mask")) {
aba5acdf	97	NEXT_ARG();
32a121cb	98	tmp = strtoul(*argv, &end, 0);
aba5acdf SH	99	if (*end) {
	100	explain_class();
	101	return -1;
	102	}
32a121cb SH	103	addattr_l(n, 1024, TCA_DSMARK_MASK, &tmp, 1);
32a121cb SH	104	} else if (!strcmp(*argv,"value")) {
aba5acdf	105	NEXT_ARG();
32a121cb	106	tmp = strtoul(*argv, &end, 0);
aba5acdf SH	107	if (*end) {
	108	explain_class();
	109	return -1;
	110	}
32a121cb SH	111	addattr_l(n, 1024, TCA_DSMARK_VALUE, &tmp, 1);
32a121cb SH	112	} else {
aba5acdf SH	113	explain_class();
	114	return -1;
	115	}
	116	argc--;
	117	argv++;
	118	}
c14f9d92	119	addattr_nest_end(n, tail);
aba5acdf SH	120	return 0;
	121	}
	122
	123
	124
	125	static int dsmark_print_opt(struct qdisc_util qu, FILE f, struct rtattr *opt)
	126	{
	127	struct rtattr *tb[TCA_DSMARK_MAX+1];
	128
	129	if (!opt) return 0;
aba5acdf SH	130	parse_rtattr(tb, TCA_DSMARK_MAX, RTA_DATA(opt), RTA_PAYLOAD(opt));
	131	if (tb[TCA_DSMARK_MASK]) {
	132	if (!RTA_PAYLOAD(tb[TCA_DSMARK_MASK]))
32a121cb SH	133	fprintf(stderr, "dsmark: empty mask\n");
32a121cb SH	134	else fprintf(f, "mask 0x%02x ",
ff24746c	135	rta_getattr_u8(tb[TCA_DSMARK_MASK]));
aba5acdf SH	136	}
	137	if (tb[TCA_DSMARK_VALUE]) {
	138	if (!RTA_PAYLOAD(tb[TCA_DSMARK_VALUE]))
32a121cb SH	139	fprintf(stderr, "dsmark: empty value\n");
32a121cb SH	140	else fprintf(f, "value 0x%02x ",
ff24746c	141	rta_getattr_u8(tb[TCA_DSMARK_VALUE]));
aba5acdf SH	142	}
	143	if (tb[TCA_DSMARK_INDICES]) {
	144	if (RTA_PAYLOAD(tb[TCA_DSMARK_INDICES]) < sizeof(__u16))
32a121cb SH	145	fprintf(stderr, "dsmark: indices too short\n");
32a121cb SH	146	else fprintf(f, "indices 0x%04x ",
ff24746c	147	rta_getattr_u16(tb[TCA_DSMARK_INDICES]));
aba5acdf SH	148	}
	149	if (tb[TCA_DSMARK_DEFAULT_INDEX]) {
	150	if (RTA_PAYLOAD(tb[TCA_DSMARK_DEFAULT_INDEX]) < sizeof(__u16))
32a121cb SH	151	fprintf(stderr, "dsmark: default_index too short\n");
32a121cb SH	152	else fprintf(f, "default_index 0x%04x ",
ff24746c	153	rta_getattr_u16(tb[TCA_DSMARK_DEFAULT_INDEX]));
aba5acdf	154	}
32a121cb	155	if (tb[TCA_DSMARK_SET_TC_INDEX]) fprintf(f, "set_tc_index ");
aba5acdf SH	156	return 0;
	157	}
	158
	159
95812b56	160	struct qdisc_util dsmark_qdisc_util = {
f2f99e2e SH	161	.id = "dsmark",
	162	.parse_qopt = dsmark_parse_opt,
	163	.print_qopt = dsmark_print_opt,
	164	.parse_copt = dsmark_parse_class_opt,
	165	.print_copt = dsmark_print_opt,
aba5acdf	166	};