Home | History | Annotate | Download | only in tc 
      1 /*
      2  * q_prio.c		PRIO.
      3  *
      4  *		This program is free software; you can redistribute it and/or
      5  *		modify it under the terms of the GNU General Public License
      6  *		as published by the Free Software Foundation; either version
      7  *		2 of the License, or (at your option) any later version.
      8  *
      9  * Authors:	Alexey Kuznetsov, <kuznet (at) ms2.inr.ac.ru>
     10  *
     11  */
     12 
     13 #include <stdio.h>
     14 #include <stdlib.h>
     15 #include <unistd.h>
     16 #include <syslog.h>
     17 #include <fcntl.h>
     18 #include <sys/socket.h>
     19 #include <netinet/in.h>
     20 #include <arpa/inet.h>
     21 #include <string.h>
     22 
     23 #include "utils.h"
     24 #include "tc_util.h"
     25 
     26 static void explain(void)
     27 {
     28 	fprintf(stderr, "Usage: ... prio bands NUMBER priomap P1 P2...[multiqueue]\n");
     29 }
     30 
     31 static int prio_parse_opt(struct qdisc_util *qu, int argc, char **argv, struct nlmsghdr *n)
     32 {
     33 	int pmap_mode = 0;
     34 	int idx = 0;
     35 	struct tc_prio_qopt opt = {3, { 1, 2, 2, 2, 1, 2, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1 } };
     36 	struct rtattr *nest;
     37 	unsigned char mq = 0;
     38 
     39 	while (argc > 0) {
     40 		if (strcmp(*argv, "bands") == 0) {
     41 			if (pmap_mode)
     42 				explain();
     43 			NEXT_ARG();
     44 			if (get_integer(&opt.bands, *argv, 10)) {
     45 				fprintf(stderr, "Illegal \"bands\"\n");
     46 				return -1;
     47 			}
     48 		} else if (strcmp(*argv, "priomap") == 0) {
     49 			if (pmap_mode) {
     50 				fprintf(stderr, "Error: duplicate priomap\n");
     51 				return -1;
     52 			}
     53 			pmap_mode = 1;
     54 		} else if (strcmp(*argv, "multiqueue") == 0) {
     55 			mq = 1;
     56 		} else if (strcmp(*argv, "help") == 0) {
     57 			explain();
     58 			return -1;
     59 		} else {
     60 			unsigned int band;
     61 
     62 			if (!pmap_mode) {
     63 				fprintf(stderr, "What is \"%s\"?\n", *argv);
     64 				explain();
     65 				return -1;
     66 			}
     67 			if (get_unsigned(&band, *argv, 10)) {
     68 				fprintf(stderr, "Illegal \"priomap\" element\n");
     69 				return -1;
     70 			}
     71 			if (band >= opt.bands) {
     72 				fprintf(stderr, "\"priomap\" element is out of bands\n");
     73 				return -1;
     74 			}
     75 			if (idx > TC_PRIO_MAX) {
     76 				fprintf(stderr, "\"priomap\" index > TC_PRIO_MAX=%u\n", TC_PRIO_MAX);
     77 				return -1;
     78 			}
     79 			opt.priomap[idx++] = band;
     80 		}
     81 		argc--; argv++;
     82 	}
     83 
     84 /*
     85 	if (pmap_mode) {
     86 		for (; idx < TC_PRIO_MAX; idx++)
     87 			opt.priomap[idx] = opt.priomap[TC_PRIO_BESTEFFORT];
     88 	}
     89 */
     90 	nest = addattr_nest_compat(n, 1024, TCA_OPTIONS, &opt, sizeof(opt));
     91 	if (mq)
     92 		addattr_l(n, 1024, TCA_PRIO_MQ, NULL, 0);
     93 	addattr_nest_compat_end(n, nest);
     94 	return 0;
     95 }
     96 
     97 int prio_print_opt(struct qdisc_util *qu, FILE *f, struct rtattr *opt)
     98 {
     99 	int i;
    100 	struct tc_prio_qopt *qopt;
    101 	struct rtattr *tb[TCA_PRIO_MAX+1];
    102 
    103 	if (opt == NULL)
    104 		return 0;
    105 
    106 	if (parse_rtattr_nested_compat(tb, TCA_PRIO_MAX, opt, qopt,
    107 					sizeof(*qopt)))
    108 		return -1;
    109 
    110 	fprintf(f, "bands %u priomap ", qopt->bands);
    111 	for (i = 0; i <= TC_PRIO_MAX; i++)
    112 		fprintf(f, " %d", qopt->priomap[i]);
    113 
    114 	if (tb[TCA_PRIO_MQ])
    115 		fprintf(f, " multiqueue: %s ",
    116 			rta_getattr_u8(tb[TCA_PRIO_MQ]) ? "on" : "off");
    117 
    118 	return 0;
    119 }
    120 
    121 struct qdisc_util prio_qdisc_util = {
    122 	.id		= "prio",
    123 	.parse_qopt	= prio_parse_opt,
    124 	.print_qopt	= prio_print_opt,
    125 };
    126