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