Blame tc/q_ets.c

Packit Service 3880ab
// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
Packit Service 3880ab
Packit Service 3880ab
/*
Packit Service 3880ab
 * Enhanced Transmission Selection - 802.1Qaz-based Qdisc
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 <fcntl.h>
Packit Service 3880ab
#include <sys/socket.h>
Packit Service 3880ab
#include <netinet/in.h>
Packit Service 3880ab
#include <arpa/inet.h>
Packit Service 3880ab
#include <string.h>
Packit Service 3880ab
Packit Service 3880ab
#include "utils.h"
Packit Service 3880ab
#include "tc_util.h"
Packit Service 3880ab
Packit Service 3880ab
static void explain(void)
Packit Service 3880ab
{
Packit Service 3880ab
	fprintf(stderr, "Usage: ... ets [bands NUMBER] [strict NUMBER] [quanta Q1 Q2...] [priomap P1 P2...]\n");
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static void cexplain(void)
Packit Service 3880ab
{
Packit Service 3880ab
	fprintf(stderr, "Usage: ... ets [quantum Q1]\n");
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static unsigned int parse_quantum(const char *arg)
Packit Service 3880ab
{
Packit Service 3880ab
	unsigned int quantum;
Packit Service 3880ab
Packit Service 3880ab
	if (get_unsigned(&quantum, arg, 10)) {
Packit Service 3880ab
		fprintf(stderr, "Illegal \"quanta\" element\n");
Packit Service 3880ab
		return 0;
Packit Service 3880ab
	}
Packit Service 3880ab
	if (!quantum)
Packit Service 3880ab
		fprintf(stderr, "\"quanta\" must be > 0\n");
Packit Service 3880ab
	return quantum;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int parse_nbands(const char *arg, __u8 *pnbands, const char *what)
Packit Service 3880ab
{
Packit Service 3880ab
	unsigned int tmp;
Packit Service 3880ab
Packit Service 3880ab
	if (get_unsigned(&tmp, arg, 10)) {
Packit Service 3880ab
		fprintf(stderr, "Illegal \"%s\"\n", what);
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
	if (tmp > TCQ_ETS_MAX_BANDS) {
Packit Service 3880ab
		fprintf(stderr, "The number of \"%s\" must be <= %d\n",
Packit Service 3880ab
			what, TCQ_ETS_MAX_BANDS);
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	*pnbands = tmp;
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int ets_parse_opt(struct qdisc_util *qu, int argc, char **argv,
Packit Service 3880ab
			 struct nlmsghdr *n, const char *dev)
Packit Service 3880ab
{
Packit Service 3880ab
	__u8 nbands = 0;
Packit Service 3880ab
	__u8 nstrict = 0;
Packit Service 3880ab
	bool quanta_mode = false;
Packit Service 3880ab
	unsigned int nquanta = 0;
Packit Service 3880ab
	__u32 quanta[TCQ_ETS_MAX_BANDS];
Packit Service 3880ab
	bool priomap_mode = false;
Packit Service 3880ab
	unsigned int nprio = 0;
Packit Service 3880ab
	__u8 priomap[TC_PRIO_MAX + 1];
Packit Service 3880ab
	unsigned int tmp;
Packit Service 3880ab
	struct rtattr *tail, *nest;
Packit Service 3880ab
Packit Service 3880ab
	while (argc > 0) {
Packit Service 3880ab
		if (strcmp(*argv, "bands") == 0) {
Packit Service 3880ab
			if (nbands) {
Packit Service 3880ab
				fprintf(stderr, "Duplicate \"bands\"\n");
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			if (parse_nbands(*argv, &nbands, "bands"))
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			priomap_mode = quanta_mode = false;
Packit Service 3880ab
		} else if (strcmp(*argv, "strict") == 0) {
Packit Service 3880ab
			if (nstrict) {
Packit Service 3880ab
				fprintf(stderr, "Duplicate \"strict\"\n");
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			if (parse_nbands(*argv, &nstrict, "strict"))
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			priomap_mode = quanta_mode = false;
Packit Service 3880ab
		} else if (strcmp(*argv, "quanta") == 0) {
Packit Service 3880ab
			if (nquanta) {
Packit Service 3880ab
				fprintf(stderr, "Duplicate \"quanta\"\n");
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			priomap_mode = false;
Packit Service 3880ab
			quanta_mode = true;
Packit Service 3880ab
			goto parse_quantum;
Packit Service 3880ab
		} else if (strcmp(*argv, "priomap") == 0) {
Packit Service 3880ab
			if (nprio) {
Packit Service 3880ab
				fprintf(stderr, "Duplicate \"priomap\"\n");
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			priomap_mode = true;
Packit Service 3880ab
			quanta_mode = false;
Packit Service 3880ab
			goto parse_priomap;
Packit Service 3880ab
		} else if (strcmp(*argv, "help") == 0) {
Packit Service 3880ab
			explain();
Packit Service 3880ab
			return -1;
Packit Service 3880ab
		} else if (quanta_mode) {
Packit Service 3880ab
			unsigned int quantum;
Packit Service 3880ab
Packit Service 3880ab
parse_quantum:
Packit Service 3880ab
			quantum = parse_quantum(*argv);
Packit Service 3880ab
			if (!quantum)
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			quanta[nquanta++] = quantum;
Packit Service 3880ab
		} else if (priomap_mode) {
Packit Service 3880ab
			unsigned int band;
Packit Service 3880ab
Packit Service 3880ab
parse_priomap:
Packit Service 3880ab
			if (get_unsigned(&band, *argv, 10)) {
Packit Service 3880ab
				fprintf(stderr, "Illegal \"priomap\" element\n");
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
			if (nprio > TC_PRIO_MAX) {
Packit Service 3880ab
				fprintf(stderr, "\"priomap\" index cannot be higher than %u\n", TC_PRIO_MAX);
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
			priomap[nprio++] = band;
Packit Service 3880ab
		} else {
Packit Service 3880ab
			fprintf(stderr, "What is \"%s\"?\n", *argv);
Packit Service 3880ab
			explain();
Packit Service 3880ab
			return -1;
Packit Service 3880ab
		}
Packit Service 3880ab
		argc--; argv++;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (!nbands)
Packit Service 3880ab
		nbands = nquanta + nstrict;
Packit Service 3880ab
	if (!nbands) {
Packit Service 3880ab
		fprintf(stderr, "One of \"bands\", \"quanta\" or \"strict\" needs to be specified\n");
Packit Service 3880ab
		explain();
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
	if (nbands < 1) {
Packit Service 3880ab
		fprintf(stderr, "The number of \"bands\" must be >= 1\n");
Packit Service 3880ab
		explain();
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
	if (nstrict + nquanta > nbands) {
Packit Service 3880ab
		fprintf(stderr, "Not enough total bands to cover all the strict bands and quanta\n");
Packit Service 3880ab
		explain();
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
	for (tmp = 0; tmp < nprio; tmp++) {
Packit Service 3880ab
		if (priomap[tmp] >= nbands) {
Packit Service 3880ab
			fprintf(stderr, "\"priomap\" element is out of bounds\n");
Packit Service 3880ab
			return -1;
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	tail = addattr_nest(n, 1024, TCA_OPTIONS | NLA_F_NESTED);
Packit Service 3880ab
	addattr_l(n, 1024, TCA_ETS_NBANDS, &nbands, sizeof(nbands));
Packit Service 3880ab
	if (nstrict)
Packit Service 3880ab
		addattr_l(n, 1024, TCA_ETS_NSTRICT, &nstrict, sizeof(nstrict));
Packit Service 3880ab
	if (nquanta) {
Packit Service 3880ab
		nest = addattr_nest(n, 1024, TCA_ETS_QUANTA | NLA_F_NESTED);
Packit Service 3880ab
		for (tmp = 0; tmp < nquanta; tmp++)
Packit Service 3880ab
			addattr_l(n, 1024, TCA_ETS_QUANTA_BAND,
Packit Service 3880ab
				  &quanta[tmp], sizeof(quanta[0]));
Packit Service 3880ab
		addattr_nest_end(n, nest);
Packit Service 3880ab
	}
Packit Service 3880ab
	if (nprio) {
Packit Service 3880ab
		nest = addattr_nest(n, 1024, TCA_ETS_PRIOMAP | NLA_F_NESTED);
Packit Service 3880ab
		for (tmp = 0; tmp < nprio; tmp++)
Packit Service 3880ab
			addattr_l(n, 1024, TCA_ETS_PRIOMAP_BAND,
Packit Service 3880ab
				  &priomap[tmp], sizeof(priomap[0]));
Packit Service 3880ab
		addattr_nest_end(n, nest);
Packit Service 3880ab
	}
Packit Service 3880ab
	addattr_nest_end(n, tail);
Packit Service 3880ab
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int ets_parse_copt(struct qdisc_util *qu, int argc, char **argv,
Packit Service 3880ab
			  struct nlmsghdr *n, const char *dev)
Packit Service 3880ab
{
Packit Service 3880ab
	unsigned int quantum = 0;
Packit Service 3880ab
	struct rtattr *tail;
Packit Service 3880ab
Packit Service 3880ab
	while (argc > 0) {
Packit Service 3880ab
		if (strcmp(*argv, "quantum") == 0) {
Packit Service 3880ab
			if (quantum) {
Packit Service 3880ab
				fprintf(stderr, "Duplicate \"quantum\"\n");
Packit Service 3880ab
				return -1;
Packit Service 3880ab
			}
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			quantum = parse_quantum(*argv);
Packit Service 3880ab
			if (!quantum)
Packit Service 3880ab
				return -1;
Packit Service 3880ab
		} else if (strcmp(*argv, "help") == 0) {
Packit Service 3880ab
			cexplain();
Packit Service 3880ab
			return -1;
Packit Service 3880ab
		} else {
Packit Service 3880ab
			fprintf(stderr, "What is \"%s\"?\n", *argv);
Packit Service 3880ab
			cexplain();
Packit Service 3880ab
			return -1;
Packit Service 3880ab
		}
Packit Service 3880ab
		argc--; argv++;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	tail = addattr_nest(n, 1024, TCA_OPTIONS | NLA_F_NESTED);
Packit Service 3880ab
	if (quantum)
Packit Service 3880ab
		addattr_l(n, 1024, TCA_ETS_QUANTA_BAND, &quantum,
Packit Service 3880ab
			  sizeof(quantum));
Packit Service 3880ab
	addattr_nest_end(n, tail);
Packit Service 3880ab
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int ets_print_opt_quanta(struct rtattr *opt)
Packit Service 3880ab
{
Packit Service 3880ab
	int len = RTA_PAYLOAD(opt);
Packit Service 3880ab
	unsigned int offset;
Packit Service 3880ab
Packit Service 3880ab
	open_json_array(PRINT_ANY, "quanta");
Packit Service 3880ab
	for (offset = 0; offset < len; ) {
Packit Service 3880ab
		struct rtattr *tb[TCA_ETS_MAX + 1] = {NULL};
Packit Service 3880ab
		struct rtattr *attr;
Packit Service 3880ab
		__u32 quantum;
Packit Service 3880ab
Packit Service 3880ab
		attr = RTA_DATA(opt) + offset;
Packit Service 3880ab
		parse_rtattr(tb, TCA_ETS_MAX, attr, len - offset);
Packit Service 3880ab
		offset += RTA_LENGTH(RTA_PAYLOAD(attr));
Packit Service 3880ab
Packit Service 3880ab
		if (!tb[TCA_ETS_QUANTA_BAND]) {
Packit Service 3880ab
			fprintf(stderr, "No ETS band quantum\n");
Packit Service 3880ab
			return -1;
Packit Service 3880ab
		}
Packit Service 3880ab
Packit Service 3880ab
		quantum = rta_getattr_u32(tb[TCA_ETS_QUANTA_BAND]);
Packit Service 3880ab
		print_uint(PRINT_ANY, NULL, " %u", quantum);
Packit Service 3880ab
Packit Service 3880ab
	}
Packit Service 3880ab
	close_json_array(PRINT_ANY, " ");
Packit Service 3880ab
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int ets_print_opt_priomap(struct rtattr *opt)
Packit Service 3880ab
{
Packit Service 3880ab
	int len = RTA_PAYLOAD(opt);
Packit Service 3880ab
	unsigned int offset;
Packit Service 3880ab
Packit Service 3880ab
	open_json_array(PRINT_ANY, "priomap");
Packit Service 3880ab
	for (offset = 0; offset < len; ) {
Packit Service 3880ab
		struct rtattr *tb[TCA_ETS_MAX + 1] = {NULL};
Packit Service 3880ab
		struct rtattr *attr;
Packit Service 3880ab
		__u8 band;
Packit Service 3880ab
Packit Service 3880ab
		attr = RTA_DATA(opt) + offset;
Packit Service 3880ab
		parse_rtattr(tb, TCA_ETS_MAX, attr, len - offset);
Packit Service 3880ab
		offset += RTA_LENGTH(RTA_PAYLOAD(attr)) + 3 /* padding */;
Packit Service 3880ab
Packit Service 3880ab
		if (!tb[TCA_ETS_PRIOMAP_BAND]) {
Packit Service 3880ab
			fprintf(stderr, "No ETS priomap band\n");
Packit Service 3880ab
			return -1;
Packit Service 3880ab
		}
Packit Service 3880ab
Packit Service 3880ab
		band = rta_getattr_u8(tb[TCA_ETS_PRIOMAP_BAND]);
Packit Service 3880ab
		print_uint(PRINT_ANY, NULL, " %u", band);
Packit Service 3880ab
Packit Service 3880ab
	}
Packit Service 3880ab
	close_json_array(PRINT_ANY, " ");
Packit Service 3880ab
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int ets_print_opt(struct qdisc_util *qu, FILE *f, struct rtattr *opt)
Packit Service 3880ab
{
Packit Service 3880ab
	struct rtattr *tb[TCA_ETS_MAX + 1];
Packit Service 3880ab
	__u8 nbands;
Packit Service 3880ab
	__u8 nstrict;
Packit Service 3880ab
	int err;
Packit Service 3880ab
Packit Service 3880ab
	if (opt == NULL)
Packit Service 3880ab
		return 0;
Packit Service 3880ab
Packit Service 3880ab
	parse_rtattr_nested(tb, TCA_ETS_MAX, opt);
Packit Service 3880ab
Packit Service 3880ab
	if (!tb[TCA_ETS_NBANDS] || !tb[TCA_ETS_PRIOMAP]) {
Packit Service 3880ab
		fprintf(stderr, "Incomplete ETS options\n");
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	nbands = rta_getattr_u8(tb[TCA_ETS_NBANDS]);
Packit Service 3880ab
	print_uint(PRINT_ANY, "bands", "bands %u ", nbands);
Packit Service 3880ab
Packit Service 3880ab
	if (tb[TCA_ETS_NSTRICT]) {
Packit Service 3880ab
		nstrict = rta_getattr_u8(tb[TCA_ETS_NSTRICT]);
Packit Service 3880ab
		print_uint(PRINT_ANY, "strict", "strict %u ", nstrict);
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (tb[TCA_ETS_QUANTA]) {
Packit Service 3880ab
		err = ets_print_opt_quanta(tb[TCA_ETS_QUANTA]);
Packit Service 3880ab
		if (err)
Packit Service 3880ab
			return err;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	return ets_print_opt_priomap(tb[TCA_ETS_PRIOMAP]);
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int ets_print_copt(struct qdisc_util *qu, FILE *f, struct rtattr *opt)
Packit Service 3880ab
{
Packit Service 3880ab
	struct rtattr *tb[TCA_ETS_MAX + 1];
Packit Service 3880ab
	__u32 quantum;
Packit Service 3880ab
Packit Service 3880ab
	if (opt == NULL)
Packit Service 3880ab
		return 0;
Packit Service 3880ab
Packit Service 3880ab
	parse_rtattr_nested(tb, TCA_ETS_MAX, opt);
Packit Service 3880ab
Packit Service 3880ab
	if (tb[TCA_ETS_QUANTA_BAND]) {
Packit Service 3880ab
		quantum = rta_getattr_u32(tb[TCA_ETS_QUANTA_BAND]);
Packit Service 3880ab
		print_uint(PRINT_ANY, "quantum", "quantum %u ", quantum);
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
struct qdisc_util ets_qdisc_util = {
Packit Service 3880ab
	.id		= "ets",
Packit Service 3880ab
	.parse_qopt	= ets_parse_opt,
Packit Service 3880ab
	.parse_copt	= ets_parse_copt,
Packit Service 3880ab
	.print_qopt	= ets_print_opt,
Packit Service 3880ab
	.print_copt	= ets_print_copt,
Packit Service 3880ab
};