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@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 28static void usage(void); 29 30static void usage(void) 31{ 32 fprintf(stderr, "Usage: tc class [ add | del | change | get ] 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 43int 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 147int filter_ifindex; 148__u32 filter_qdisc; 149 150static int print_class(const struct sockaddr_nl *who, 151 struct nlmsghdr *n, void *arg) 152{ 153 FILE *fp = (FILE*)arg; 154 struct tcmsg *t = NLMSG_DATA(n); 155 int len = n->nlmsg_len; 156 struct rtattr * tb[TCA_MAX+1]; 157 struct qdisc_util *q; 158 char abuf[256]; 159 160 if (n->nlmsg_type != RTM_NEWTCLASS && n->nlmsg_type != RTM_DELTCLASS) { 161 fprintf(stderr, "Not a class\n"); 162 return 0; 163 } 164 len -= NLMSG_LENGTH(sizeof(*t)); 165 if (len < 0) { 166 fprintf(stderr, "Wrong len %d\n", len); 167 return -1; 168 } 169 if (filter_qdisc && TC_H_MAJ(t->tcm_handle^filter_qdisc)) 170 return 0; 171 172 memset(tb, 0, sizeof(tb)); 173 parse_rtattr(tb, TCA_MAX, TCA_RTA(t), len); 174 175 if (tb[TCA_KIND] == NULL) { 176 fprintf(stderr, "print_class: NULL kind\n"); 177 return -1; 178 } 179 180 if (n->nlmsg_type == RTM_DELTCLASS) 181 fprintf(fp, "deleted "); 182 183 abuf[0] = 0; 184 if (t->tcm_handle) { 185 if (filter_qdisc) 186 print_tc_classid(abuf, sizeof(abuf), TC_H_MIN(t->tcm_handle)); 187 else 188 print_tc_classid(abuf, sizeof(abuf), t->tcm_handle); 189 } 190 fprintf(fp, "class %s %s ", (char*)RTA_DATA(tb[TCA_KIND]), abuf); 191 192 if (filter_ifindex == 0) 193 fprintf(fp, "dev %s ", ll_index_to_name(t->tcm_ifindex)); 194 195 if (t->tcm_parent == TC_H_ROOT) 196 fprintf(fp, "root "); 197 else { 198 if (filter_qdisc) 199 print_tc_classid(abuf, sizeof(abuf), TC_H_MIN(t->tcm_parent)); 200 else 201 print_tc_classid(abuf, sizeof(abuf), t->tcm_parent); 202 fprintf(fp, "parent %s ", abuf); 203 } 204 if (t->tcm_info) 205 fprintf(fp, "leaf %x: ", t->tcm_info>>16); 206 q = get_qdisc_kind(RTA_DATA(tb[TCA_KIND])); 207 if (tb[TCA_OPTIONS]) { 208 if (q && q->print_copt) 209 q->print_copt(q, fp, tb[TCA_OPTIONS]); 210 else 211 fprintf(fp, "[cannot parse class parameters]"); 212 } 213 fprintf(fp, "\n"); 214 if (show_stats) { 215 struct rtattr *xstats = NULL; 216 217 if (tb[TCA_STATS] || tb[TCA_STATS2]) { 218 print_tcstats_attr(fp, tb, " ", &xstats); 219 fprintf(fp, "\n"); 220 } 221 if (q && (xstats || tb[TCA_XSTATS]) && q->print_xstats) { 222 q->print_xstats(q, fp, xstats ? : tb[TCA_XSTATS]); 223 fprintf(fp, "\n"); 224 } 225 } 226 fflush(fp); 227 return 0; 228} 229 230 231int tc_class_list(int argc, char **argv) 232{ 233 struct tcmsg t; 234 char d[16]; 235 236 memset(&t, 0, sizeof(t)); 237 t.tcm_family = AF_UNSPEC; 238 memset(d, 0, sizeof(d)); 239 240 while (argc > 0) { 241 if (strcmp(*argv, "dev") == 0) { 242 NEXT_ARG(); 243 if (d[0]) 244 duparg("dev", *argv); 245 strncpy(d, *argv, sizeof(d)-1); 246 } else if (strcmp(*argv, "qdisc") == 0) { 247 NEXT_ARG(); 248 if (filter_qdisc) 249 duparg("qdisc", *argv); 250 if (get_qdisc_handle(&filter_qdisc, *argv)) 251 invarg(*argv, "invalid qdisc ID"); 252 } else if (strcmp(*argv, "root") == 0) { 253 if (t.tcm_parent) { 254 fprintf(stderr, "Error: \"root\" is duplicate parent ID\n"); 255 return -1; 256 } 257 t.tcm_parent = TC_H_ROOT; 258 } else if (strcmp(*argv, "parent") == 0) { 259 __u32 handle; 260 if (t.tcm_parent) 261 duparg("parent", *argv); 262 NEXT_ARG(); 263 if (get_tc_classid(&handle, *argv)) 264 invarg(*argv, "invalid parent ID"); 265 t.tcm_parent = handle; 266 } else if (matches(*argv, "help") == 0) { 267 usage(); 268 } else { 269 fprintf(stderr, "What is \"%s\"? Try \"tc class help\".\n", *argv); 270 return -1; 271 } 272 273 argc--; argv++; 274 } 275 276 ll_init_map(&rth); 277 278 if (d[0]) { 279 if ((t.tcm_ifindex = ll_name_to_index(d)) == 0) { 280 fprintf(stderr, "Cannot find device \"%s\"\n", d); 281 return 1; 282 } 283 filter_ifindex = t.tcm_ifindex; 284 } 285 286 if (rtnl_dump_request(&rth, RTM_GETTCLASS, &t, sizeof(t)) < 0) { 287 perror("Cannot send dump request"); 288 return 1; 289 } 290 291 if (rtnl_dump_filter(&rth, print_class, stdout, NULL, NULL) < 0) { 292 fprintf(stderr, "Dump terminated\n"); 293 return 1; 294 } 295 296 return 0; 297} 298 299int do_class(int argc, char **argv) 300{ 301 if (argc < 1) 302 return tc_class_list(0, NULL); 303 if (matches(*argv, "add") == 0) 304 return tc_class_modify(RTM_NEWTCLASS, NLM_F_EXCL|NLM_F_CREATE, argc-1, argv+1); 305 if (matches(*argv, "change") == 0) 306 return tc_class_modify(RTM_NEWTCLASS, 0, argc-1, argv+1); 307 if (matches(*argv, "replace") == 0) 308 return tc_class_modify(RTM_NEWTCLASS, NLM_F_CREATE, argc-1, argv+1); 309 if (matches(*argv, "delete") == 0) 310 return tc_class_modify(RTM_DELTCLASS, 0, argc-1, argv+1); 311#if 0 312 if (matches(*argv, "get") == 0) 313 return tc_class_get(RTM_GETTCLASS, 0, argc-1, argv+1); 314#endif 315 if (matches(*argv, "list") == 0 || matches(*argv, "show") == 0 316 || matches(*argv, "lst") == 0) 317 return tc_class_list(argc-1, argv+1); 318 if (matches(*argv, "help") == 0) 319 usage(); 320 fprintf(stderr, "Command \"%s\" is unknown, try \"tc class help\".\n", *argv); 321 return -1; 322} 323