Home | History | Annotate | Download | only in tc
      1 /*
      2  * tc_class.c		"tc class".
      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 #include <math.h>
     23 
     24 #include "utils.h"
     25 #include "tc_util.h"
     26 #include "tc_common.h"
     27 
     28 static void usage(void);
     29 
     30 static void usage(void)
     31 {
     32 	fprintf(stderr, "Usage: tc class [ add | del | change | replace | show ] dev STRING\n");
     33 	fprintf(stderr, "       [ classid CLASSID ] [ root | parent CLASSID ]\n");
     34 	fprintf(stderr, "       [ [ QDISC_KIND ] [ help | OPTIONS ] ]\n");
     35 	fprintf(stderr, "\n");
     36 	fprintf(stderr, "       tc class show [ dev STRING ] [ root | parent CLASSID ]\n");
     37 	fprintf(stderr, "Where:\n");
     38 	fprintf(stderr, "QDISC_KIND := { prio | cbq | etc. }\n");
     39 	fprintf(stderr, "OPTIONS := ... try tc class add <desired QDISC_KIND> help\n");
     40 	return;
     41 }
     42 
     43 int tc_class_modify(int cmd, unsigned flags, int argc, char **argv)
     44 {
     45 	struct {
     46 		struct nlmsghdr 	n;
     47 		struct tcmsg 		t;
     48 		char   			buf[4096];
     49 	} req;
     50 	struct qdisc_util *q = NULL;
     51 	struct tc_estimator est;
     52 	char  d[16];
     53 	char  k[16];
     54 
     55 	memset(&req, 0, sizeof(req));
     56 	memset(&est, 0, sizeof(est));
     57 	memset(d, 0, sizeof(d));
     58 	memset(k, 0, sizeof(k));
     59 
     60 	req.n.nlmsg_len = NLMSG_LENGTH(sizeof(struct tcmsg));
     61 	req.n.nlmsg_flags = NLM_F_REQUEST|flags;
     62 	req.n.nlmsg_type = cmd;
     63 	req.t.tcm_family = AF_UNSPEC;
     64 
     65 	while (argc > 0) {
     66 		if (strcmp(*argv, "dev") == 0) {
     67 			NEXT_ARG();
     68 			if (d[0])
     69 				duparg("dev", *argv);
     70 			strncpy(d, *argv, sizeof(d)-1);
     71 		} else if (strcmp(*argv, "classid") == 0) {
     72 			__u32 handle;
     73 			NEXT_ARG();
     74 			if (req.t.tcm_handle)
     75 				duparg("classid", *argv);
     76 			if (get_tc_classid(&handle, *argv))
     77 				invarg(*argv, "invalid class ID");
     78 			req.t.tcm_handle = handle;
     79 		} else if (strcmp(*argv, "handle") == 0) {
     80 			fprintf(stderr, "Error: try \"classid\" instead of \"handle\"\n");
     81 			return -1;
     82  		} else if (strcmp(*argv, "root") == 0) {
     83 			if (req.t.tcm_parent) {
     84 				fprintf(stderr, "Error: \"root\" is duplicate parent ID.\n");
     85 				return -1;
     86 			}
     87 			req.t.tcm_parent = TC_H_ROOT;
     88 		} else if (strcmp(*argv, "parent") == 0) {
     89 			__u32 handle;
     90 			NEXT_ARG();
     91 			if (req.t.tcm_parent)
     92 				duparg("parent", *argv);
     93 			if (get_tc_classid(&handle, *argv))
     94 				invarg(*argv, "invalid parent ID");
     95 			req.t.tcm_parent = handle;
     96 		} else if (matches(*argv, "estimator") == 0) {
     97 			if (parse_estimator(&argc, &argv, &est))
     98 				return -1;
     99 		} else if (matches(*argv, "help") == 0) {
    100 			usage();
    101 		} else {
    102 			strncpy(k, *argv, sizeof(k)-1);
    103 
    104 			q = get_qdisc_kind(k);
    105 			argc--; argv++;
    106 			break;
    107 		}
    108 		argc--; argv++;
    109 	}
    110 
    111 	if (k[0])
    112 		addattr_l(&req.n, sizeof(req), TCA_KIND, k, strlen(k)+1);
    113 	if (est.ewma_log)
    114 		addattr_l(&req.n, sizeof(req), TCA_RATE, &est, sizeof(est));
    115 
    116 	if (q) {
    117 		if (q->parse_copt == NULL) {
    118 			fprintf(stderr, "Error: Qdisc \"%s\" is classless.\n", k);
    119 			return 1;
    120 		}
    121 		if (q->parse_copt(q, argc, argv, &req.n))
    122 			return 1;
    123 	} else {
    124 		if (argc) {
    125 			if (matches(*argv, "help") == 0)
    126 				usage();
    127 			fprintf(stderr, "Garbage instead of arguments \"%s ...\". Try \"tc class help\".", *argv);
    128 			return -1;
    129 		}
    130 	}
    131 
    132 	if (d[0])  {
    133 		ll_init_map(&rth);
    134 
    135 		if ((req.t.tcm_ifindex = ll_name_to_index(d)) == 0) {
    136 			fprintf(stderr, "Cannot find device \"%s\"\n", d);
    137 			return 1;
    138 		}
    139 	}
    140 
    141 	if (rtnl_talk(&rth, &req.n, 0, 0, NULL, NULL, NULL) < 0)
    142 		return 2;
    143 
    144 	return 0;
    145 }
    146 
    147 int filter_ifindex;
    148 __u32 filter_qdisc;
    149 __u32 filter_classid;
    150 
    151 int print_class(const struct sockaddr_nl *who,
    152 		       struct nlmsghdr *n, void *arg)
    153 {
    154 	FILE *fp = (FILE*)arg;
    155 	struct tcmsg *t = NLMSG_DATA(n);
    156 	int len = n->nlmsg_len;
    157 	struct rtattr * tb[TCA_MAX+1];
    158 	struct qdisc_util *q;
    159 	char abuf[256];
    160 
    161 	if (n->nlmsg_type != RTM_NEWTCLASS && n->nlmsg_type != RTM_DELTCLASS) {
    162 		fprintf(stderr, "Not a class\n");
    163 		return 0;
    164 	}
    165 	len -= NLMSG_LENGTH(sizeof(*t));
    166 	if (len < 0) {
    167 		fprintf(stderr, "Wrong len %d\n", len);
    168 		return -1;
    169 	}
    170 	if (filter_qdisc && TC_H_MAJ(t->tcm_handle^filter_qdisc))
    171 		return 0;
    172 
    173 	if (filter_classid && t->tcm_handle != filter_classid)
    174 		return 0;
    175 
    176 	memset(tb, 0, sizeof(tb));
    177 	parse_rtattr(tb, TCA_MAX, TCA_RTA(t), len);
    178 
    179 	if (tb[TCA_KIND] == NULL) {
    180 		fprintf(stderr, "print_class: NULL kind\n");
    181 		return -1;
    182 	}
    183 
    184 	if (n->nlmsg_type == RTM_DELTCLASS)
    185 		fprintf(fp, "deleted ");
    186 
    187 	abuf[0] = 0;
    188 	if (t->tcm_handle) {
    189 		if (filter_qdisc)
    190 			print_tc_classid(abuf, sizeof(abuf), TC_H_MIN(t->tcm_handle));
    191 		else
    192 			print_tc_classid(abuf, sizeof(abuf), t->tcm_handle);
    193 	}
    194 	fprintf(fp, "class %s %s ", (char*)RTA_DATA(tb[TCA_KIND]), abuf);
    195 
    196 	if (filter_ifindex == 0)
    197 		fprintf(fp, "dev %s ", ll_index_to_name(t->tcm_ifindex));
    198 
    199 	if (t->tcm_parent == TC_H_ROOT)
    200 		fprintf(fp, "root ");
    201 	else {
    202 		if (filter_qdisc)
    203 			print_tc_classid(abuf, sizeof(abuf), TC_H_MIN(t->tcm_parent));
    204 		else
    205 			print_tc_classid(abuf, sizeof(abuf), t->tcm_parent);
    206 		fprintf(fp, "parent %s ", abuf);
    207 	}
    208 	if (t->tcm_info)
    209 		fprintf(fp, "leaf %x: ", t->tcm_info>>16);
    210 	q = get_qdisc_kind(RTA_DATA(tb[TCA_KIND]));
    211 	if (tb[TCA_OPTIONS]) {
    212 		if (q && q->print_copt)
    213 			q->print_copt(q, fp, tb[TCA_OPTIONS]);
    214 		else
    215 			fprintf(fp, "[cannot parse class parameters]");
    216 	}
    217 	fprintf(fp, "\n");
    218 	if (show_stats) {
    219 		struct rtattr *xstats = NULL;
    220 
    221 		if (tb[TCA_STATS] || tb[TCA_STATS2]) {
    222 			print_tcstats_attr(fp, tb, " ", &xstats);
    223 			fprintf(fp, "\n");
    224 		}
    225 		if (q && (xstats || tb[TCA_XSTATS]) && q->print_xstats) {
    226 			q->print_xstats(q, fp, xstats ? : tb[TCA_XSTATS]);
    227 			fprintf(fp, "\n");
    228 		}
    229 	}
    230 	fflush(fp);
    231 	return 0;
    232 }
    233 
    234 
    235 int tc_class_list(int argc, char **argv)
    236 {
    237 	struct tcmsg t;
    238 	char d[16];
    239 
    240 	memset(&t, 0, sizeof(t));
    241 	t.tcm_family = AF_UNSPEC;
    242 	memset(d, 0, sizeof(d));
    243 
    244 	while (argc > 0) {
    245 		if (strcmp(*argv, "dev") == 0) {
    246 			NEXT_ARG();
    247 			if (d[0])
    248 				duparg("dev", *argv);
    249 			strncpy(d, *argv, sizeof(d)-1);
    250 		} else if (strcmp(*argv, "qdisc") == 0) {
    251 			NEXT_ARG();
    252 			if (filter_qdisc)
    253 				duparg("qdisc", *argv);
    254 			if (get_qdisc_handle(&filter_qdisc, *argv))
    255 				invarg(*argv, "invalid qdisc ID");
    256 		} else if (strcmp(*argv, "classid") == 0) {
    257 			NEXT_ARG();
    258 			if (filter_classid)
    259 				duparg("classid", *argv);
    260 			if (get_tc_classid(&filter_classid, *argv))
    261 				invarg(*argv, "invalid class ID");
    262 		} else if (strcmp(*argv, "root") == 0) {
    263 			if (t.tcm_parent) {
    264 				fprintf(stderr, "Error: \"root\" is duplicate parent ID\n");
    265 				return -1;
    266 			}
    267 			t.tcm_parent = TC_H_ROOT;
    268 		} else if (strcmp(*argv, "parent") == 0) {
    269 			__u32 handle;
    270 			if (t.tcm_parent)
    271 				duparg("parent", *argv);
    272 			NEXT_ARG();
    273 			if (get_tc_classid(&handle, *argv))
    274 				invarg(*argv, "invalid parent ID");
    275 			t.tcm_parent = handle;
    276 		} else if (matches(*argv, "help") == 0) {
    277 			usage();
    278 		} else {
    279 			fprintf(stderr, "What is \"%s\"? Try \"tc class help\".\n", *argv);
    280 			return -1;
    281 		}
    282 
    283 		argc--; argv++;
    284 	}
    285 
    286  	ll_init_map(&rth);
    287 
    288 	if (d[0]) {
    289 		if ((t.tcm_ifindex = ll_name_to_index(d)) == 0) {
    290 			fprintf(stderr, "Cannot find device \"%s\"\n", d);
    291 			return 1;
    292 		}
    293 		filter_ifindex = t.tcm_ifindex;
    294 	}
    295 
    296  	if (rtnl_dump_request(&rth, RTM_GETTCLASS, &t, sizeof(t)) < 0) {
    297 		perror("Cannot send dump request");
    298 		return 1;
    299 	}
    300 
    301  	if (rtnl_dump_filter(&rth, print_class, stdout, NULL, NULL) < 0) {
    302 		fprintf(stderr, "Dump terminated\n");
    303 		return 1;
    304 	}
    305 
    306 	return 0;
    307 }
    308 
    309 int do_class(int argc, char **argv)
    310 {
    311 	if (argc < 1)
    312 		return tc_class_list(0, NULL);
    313 	if (matches(*argv, "add") == 0)
    314 		return tc_class_modify(RTM_NEWTCLASS, NLM_F_EXCL|NLM_F_CREATE, argc-1, argv+1);
    315 	if (matches(*argv, "change") == 0)
    316 		return tc_class_modify(RTM_NEWTCLASS, 0, argc-1, argv+1);
    317 	if (matches(*argv, "replace") == 0)
    318 		return tc_class_modify(RTM_NEWTCLASS, NLM_F_CREATE, argc-1, argv+1);
    319 	if (matches(*argv, "delete") == 0)
    320 		return tc_class_modify(RTM_DELTCLASS, 0,  argc-1, argv+1);
    321 #if 0
    322 	if (matches(*argv, "get") == 0)
    323 		return tc_class_get(RTM_GETTCLASS, 0,  argc-1, argv+1);
    324 #endif
    325 	if (matches(*argv, "list") == 0 || matches(*argv, "show") == 0
    326 	    || matches(*argv, "lst") == 0)
    327 		return tc_class_list(argc-1, argv+1);
    328 	if (matches(*argv, "help") == 0) {
    329 		usage();
    330 		return 0;
    331 	}
    332 	fprintf(stderr, "Command \"%s\" is unknown, try \"tc class help\".\n", *argv);
    333 	return -1;
    334 }
    335