]>
Commit | Line | Data |
---|---|---|
aba5acdf SH |
1 | /* |
2 | * q_gred.c GRED. | |
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 | * | |
ae665a52 SH |
9 | * Authors: J Hadi Salim(hadi@nortelnetworks.com) |
10 | * code ruthlessly ripped from | |
11 | * Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru> | |
aba5acdf SH |
12 | * |
13 | */ | |
14 | ||
15 | #include <stdio.h> | |
16 | #include <stdlib.h> | |
17 | #include <unistd.h> | |
aba5acdf SH |
18 | #include <fcntl.h> |
19 | #include <sys/socket.h> | |
20 | #include <netinet/in.h> | |
21 | #include <arpa/inet.h> | |
22 | #include <string.h> | |
1b6f0bb5 | 23 | #include <math.h> |
aba5acdf SH |
24 | |
25 | #include "utils.h" | |
26 | #include "tc_util.h" | |
27 | ||
28 | #include "tc_red.h" | |
29 | ||
30 | ||
31 | #if 0 | |
32a121cb | 32 | #define DPRINTF(format, args...) fprintf(stderr, format, ##args) |
aba5acdf | 33 | #else |
32a121cb | 34 | #define DPRINTF(format, args...) |
aba5acdf SH |
35 | #endif |
36 | ||
37 | static void explain(void) | |
38 | { | |
357c45ad | 39 | fprintf(stderr, "Usage: tc qdisc { add | replace | change } ... gred setup vqs NUMBER\n"); |
aacee269 | 40 | fprintf(stderr, " default DEFAULT_VQ [ grio ] [ limit BYTES ]\n"); |
357c45ad DW |
41 | fprintf(stderr, " tc qdisc change ... gred vq VQ [ prio VALUE ] limit BYTES\n"); |
42 | fprintf(stderr, " min BYTES max BYTES avpkt BYTES [ burst PACKETS ]\n"); | |
43 | fprintf(stderr, " [ probability PROBABILITY ] [ bandwidth KBPS ]\n"); | |
aba5acdf SH |
44 | } |
45 | ||
3d0b7439 | 46 | static int init_gred(struct qdisc_util *qu, int argc, char **argv, |
ebde8780 | 47 | struct nlmsghdr *n) |
aba5acdf SH |
48 | { |
49 | ||
50 | struct rtattr *tail; | |
cb4bd0ec | 51 | struct tc_gred_sopt opt = { 0 }; |
aacee269 | 52 | __u32 limit = 0; |
eb6d7d6a DW |
53 | |
54 | opt.def_DP = MAX_DPs; | |
aba5acdf SH |
55 | |
56 | while (argc > 0) { | |
32a121cb | 57 | DPRINTF(stderr, "init_gred: invoked with %s\n", *argv); |
357c45ad DW |
58 | if (strcmp(*argv, "vqs") == 0 || |
59 | strcmp(*argv, "DPs") == 0) { | |
aba5acdf | 60 | NEXT_ARG(); |
eb6d7d6a | 61 | if (get_unsigned(&opt.DPs, *argv, 10)) { |
357c45ad | 62 | fprintf(stderr, "Illegal \"vqs\"\n"); |
eb6d7d6a DW |
63 | return -1; |
64 | } else if (opt.DPs > MAX_DPs) { | |
32a121cb SH |
65 | fprintf(stderr, "GRED: only %u VQs are currently supported\n", |
66 | MAX_DPs); | |
aba5acdf SH |
67 | return -1; |
68 | } | |
69 | } else if (strcmp(*argv, "default") == 0) { | |
eb6d7d6a | 70 | if (opt.DPs == 0) { |
32a121cb | 71 | fprintf(stderr, "\"default\" must be defined after \"vqs\"\n"); |
aba5acdf SH |
72 | return -1; |
73 | } | |
eb6d7d6a DW |
74 | NEXT_ARG(); |
75 | if (get_unsigned(&opt.def_DP, *argv, 10)) { | |
76 | fprintf(stderr, "Illegal \"default\"\n"); | |
77 | return -1; | |
78 | } else if (opt.def_DP >= opt.DPs) { | |
32a121cb | 79 | fprintf(stderr, "\"default\" must be less than \"vqs\"\n"); |
aba5acdf SH |
80 | return -1; |
81 | } | |
82 | } else if (strcmp(*argv, "grio") == 0) { | |
cb4bd0ec | 83 | opt.grio = 1; |
aacee269 DW |
84 | } else if (strcmp(*argv, "limit") == 0) { |
85 | NEXT_ARG(); | |
86 | if (get_size(&limit, *argv)) { | |
87 | fprintf(stderr, "Illegal \"limit\"\n"); | |
88 | return -1; | |
89 | } | |
aba5acdf SH |
90 | } else if (strcmp(*argv, "help") == 0) { |
91 | explain(); | |
92 | return -1; | |
93 | } else { | |
94 | fprintf(stderr, "What is \"%s\"?\n", *argv); | |
95 | explain(); | |
96 | return -1; | |
97 | } | |
98 | argc--; argv++; | |
ebde8780 | 99 | } |
aba5acdf | 100 | |
eb6d7d6a | 101 | if (!opt.DPs || opt.def_DP == MAX_DPs) { |
32a121cb | 102 | fprintf(stderr, "Illegal gred setup parameters\n"); |
ebde8780 SH |
103 | return -1; |
104 | } | |
105 | ||
32a121cb SH |
106 | DPRINTF("TC_GRED: sending DPs=%u def_DP=%u\n", opt.DPs, opt.def_DP); |
107 | n->nlmsg_flags |= NLM_F_CREATE; | |
c14f9d92 | 108 | tail = addattr_nest(n, 1024, TCA_OPTIONS); |
aba5acdf | 109 | addattr_l(n, 1024, TCA_GRED_DPS, &opt, sizeof(struct tc_gred_sopt)); |
aacee269 DW |
110 | if (limit) |
111 | addattr32(n, 1024, TCA_GRED_LIMIT, limit); | |
c14f9d92 | 112 | addattr_nest_end(n, tail); |
ebde8780 | 113 | return 0; |
aba5acdf SH |
114 | } |
115 | /* | |
116 | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
117 | */ | |
927e3cfb | 118 | static int gred_parse_opt(struct qdisc_util *qu, int argc, char **argv, struct nlmsghdr *n, const char *dev) |
aba5acdf | 119 | { |
32a121cb | 120 | int ok = 0; |
eb6d7d6a | 121 | struct tc_gred_qopt opt = { 0 }; |
32a121cb SH |
122 | unsigned int burst = 0; |
123 | unsigned int avpkt = 0; | |
aba5acdf | 124 | double probability = 0.02; |
32a121cb | 125 | unsigned int rate = 0; |
9d9a67c7 | 126 | int parm; |
aba5acdf SH |
127 | __u8 sbuf[256]; |
128 | struct rtattr *tail; | |
1b6f0bb5 | 129 | __u32 max_P; |
aba5acdf | 130 | |
eb6d7d6a | 131 | opt.DP = MAX_DPs; |
aba5acdf SH |
132 | |
133 | while (argc > 0) { | |
134 | if (strcmp(*argv, "limit") == 0) { | |
135 | NEXT_ARG(); | |
136 | if (get_size(&opt.limit, *argv)) { | |
137 | fprintf(stderr, "Illegal \"limit\"\n"); | |
138 | return -1; | |
139 | } | |
140 | ok++; | |
141 | } else if (strcmp(*argv, "setup") == 0) { | |
142 | if (ok) { | |
143 | fprintf(stderr, "Illegal \"setup\"\n"); | |
144 | return -1; | |
145 | } | |
d73e0408 | 146 | return init_gred(qu, argc-1, argv+1, n); |
aba5acdf SH |
147 | } else if (strcmp(*argv, "min") == 0) { |
148 | NEXT_ARG(); | |
149 | if (get_size(&opt.qth_min, *argv)) { | |
150 | fprintf(stderr, "Illegal \"min\"\n"); | |
151 | return -1; | |
152 | } | |
153 | ok++; | |
154 | } else if (strcmp(*argv, "max") == 0) { | |
155 | NEXT_ARG(); | |
156 | if (get_size(&opt.qth_max, *argv)) { | |
157 | fprintf(stderr, "Illegal \"max\"\n"); | |
158 | return -1; | |
159 | } | |
160 | ok++; | |
357c45ad DW |
161 | } else if (strcmp(*argv, "vq") == 0 || |
162 | strcmp(*argv, "DP") == 0) { | |
aba5acdf | 163 | NEXT_ARG(); |
eb6d7d6a | 164 | if (get_unsigned(&opt.DP, *argv, 10)) { |
357c45ad | 165 | fprintf(stderr, "Illegal \"vq\"\n"); |
aba5acdf | 166 | return -1; |
eb6d7d6a | 167 | } else if (opt.DP >= MAX_DPs) { |
32a121cb SH |
168 | fprintf(stderr, "GRED: only %u VQs are currently supported\n", |
169 | MAX_DPs); | |
eb6d7d6a DW |
170 | return -1; |
171 | } /* need a better error check */ | |
aba5acdf SH |
172 | ok++; |
173 | } else if (strcmp(*argv, "burst") == 0) { | |
174 | NEXT_ARG(); | |
d73e0408 | 175 | if (get_unsigned(&burst, *argv, 0)) { |
aba5acdf SH |
176 | fprintf(stderr, "Illegal \"burst\"\n"); |
177 | return -1; | |
178 | } | |
179 | ok++; | |
180 | } else if (strcmp(*argv, "avpkt") == 0) { | |
181 | NEXT_ARG(); | |
182 | if (get_size(&avpkt, *argv)) { | |
183 | fprintf(stderr, "Illegal \"avpkt\"\n"); | |
184 | return -1; | |
185 | } | |
186 | ok++; | |
187 | } else if (strcmp(*argv, "probability") == 0) { | |
188 | NEXT_ARG(); | |
189 | if (sscanf(*argv, "%lg", &probability) != 1) { | |
190 | fprintf(stderr, "Illegal \"probability\"\n"); | |
191 | return -1; | |
192 | } | |
193 | ok++; | |
194 | } else if (strcmp(*argv, "prio") == 0) { | |
195 | NEXT_ARG(); | |
32a121cb | 196 | opt.prio = strtol(*argv, (char **)NULL, 10); |
aba5acdf SH |
197 | /* some error check here */ |
198 | ok++; | |
199 | } else if (strcmp(*argv, "bandwidth") == 0) { | |
200 | NEXT_ARG(); | |
927e3cfb ND |
201 | if (strchr(*argv, '%')) { |
202 | if (get_percent_rate(&rate, *argv, dev)) { | |
203 | fprintf(stderr, "Illegal \"bandwidth\"\n"); | |
204 | return -1; | |
205 | } | |
206 | } else if (get_rate(&rate, *argv)) { | |
aba5acdf SH |
207 | fprintf(stderr, "Illegal \"bandwidth\"\n"); |
208 | return -1; | |
209 | } | |
210 | ok++; | |
211 | } else if (strcmp(*argv, "help") == 0) { | |
212 | explain(); | |
213 | return -1; | |
214 | } else { | |
215 | fprintf(stderr, "What is \"%s\"?\n", *argv); | |
216 | explain(); | |
217 | return -1; | |
218 | } | |
219 | argc--; argv++; | |
220 | } | |
221 | ||
a77905ef DW |
222 | if (!ok) { |
223 | explain(); | |
224 | return -1; | |
225 | } | |
eb6d7d6a DW |
226 | if (opt.DP == MAX_DPs || !opt.limit || !opt.qth_min || !opt.qth_max || |
227 | !avpkt) { | |
32a121cb | 228 | fprintf(stderr, "Required parameter (vq, limit, min, max, avpkt) is missing\n"); |
aba5acdf SH |
229 | return -1; |
230 | } | |
ab15aeac ED |
231 | if (!burst) { |
232 | burst = (2 * opt.qth_min + opt.qth_max) / (3 * avpkt); | |
233 | fprintf(stderr, "GRED: set burst to %u\n", burst); | |
234 | } | |
d93c909a DW |
235 | if (!rate) { |
236 | get_rate(&rate, "10Mbit"); | |
237 | fprintf(stderr, "GRED: set bandwidth to 10Mbit\n"); | |
238 | } | |
9d9a67c7 | 239 | if ((parm = tc_red_eval_ewma(opt.qth_min, burst, avpkt)) < 0) { |
aba5acdf SH |
240 | fprintf(stderr, "GRED: failed to calculate EWMA constant.\n"); |
241 | return -1; | |
242 | } | |
9d9a67c7 | 243 | if (parm >= 10) |
32a121cb SH |
244 | fprintf(stderr, "GRED: WARNING. Burst %u seems to be too large.\n", |
245 | burst); | |
9d9a67c7 DW |
246 | opt.Wlog = parm; |
247 | if ((parm = tc_red_eval_P(opt.qth_min, opt.qth_max, probability)) < 0) { | |
aba5acdf SH |
248 | fprintf(stderr, "GRED: failed to calculate probability.\n"); |
249 | return -1; | |
250 | } | |
9d9a67c7 DW |
251 | opt.Plog = parm; |
252 | if ((parm = tc_red_eval_idle_damping(opt.Wlog, avpkt, rate, sbuf)) < 0) | |
aba5acdf | 253 | { |
32a121cb | 254 | fprintf(stderr, "GRED: failed to calculate idle damping table.\n"); |
aba5acdf SH |
255 | return -1; |
256 | } | |
9d9a67c7 | 257 | opt.Scell_log = parm; |
aba5acdf | 258 | |
c14f9d92 | 259 | tail = addattr_nest(n, 1024, TCA_OPTIONS); |
aba5acdf SH |
260 | addattr_l(n, 1024, TCA_GRED_PARMS, &opt, sizeof(opt)); |
261 | addattr_l(n, 1024, TCA_GRED_STAB, sbuf, 256); | |
1b6f0bb5 ED |
262 | max_P = probability * pow(2, 32); |
263 | addattr32(n, 1024, TCA_GRED_MAX_P, max_P); | |
c14f9d92 | 264 | addattr_nest_end(n, tail); |
aba5acdf SH |
265 | return 0; |
266 | } | |
267 | ||
c3e1cd28 JK |
268 | static void gred_print_stats(struct tc_gred_qopt *qopt) |
269 | { | |
270 | SPRINT_BUF(b1); | |
271 | ||
272 | if (!is_json_context()) | |
273 | printf("\n Queue size: "); | |
274 | ||
275 | print_uint(PRINT_JSON, "qave", NULL, qopt->qave); | |
276 | print_string(PRINT_FP, NULL, "average %s ", | |
277 | sprint_size(qopt->qave, b1)); | |
278 | ||
279 | print_uint(PRINT_JSON, "backlog", NULL, qopt->backlog); | |
280 | print_string(PRINT_FP, NULL, "current %s ", | |
281 | sprint_size(qopt->backlog, b1)); | |
282 | ||
283 | if (!is_json_context()) | |
284 | printf("\n Dropped packets: "); | |
285 | ||
286 | print_uint(PRINT_ANY, "forced_drop", "forced %u ", qopt->forced); | |
287 | print_uint(PRINT_ANY, "prob_drop", "early %u ", qopt->early); | |
288 | print_uint(PRINT_ANY, "pdrop", "pdrop %u ", qopt->pdrop); | |
289 | print_uint(PRINT_ANY, "other", "other %u ", qopt->other); | |
290 | ||
291 | if (!is_json_context()) | |
292 | printf("\n Total packets: "); | |
293 | ||
294 | print_uint(PRINT_ANY, "packets", "%u ", qopt->packets); | |
295 | ||
296 | print_uint(PRINT_JSON, "bytes", NULL, qopt->bytesin); | |
297 | print_string(PRINT_FP, NULL, "(%s) ", sprint_size(qopt->bytesin, b1)); | |
298 | } | |
299 | ||
aba5acdf SH |
300 | static int gred_print_opt(struct qdisc_util *qu, FILE *f, struct rtattr *opt) |
301 | { | |
1b6f0bb5 | 302 | struct rtattr *tb[TCA_GRED_MAX + 1]; |
1693a4d3 | 303 | struct tc_gred_sopt *sopt; |
aba5acdf | 304 | struct tc_gred_qopt *qopt; |
1b6f0bb5 | 305 | __u32 *max_p = NULL; |
aacee269 | 306 | __u32 *limit = NULL; |
32a121cb SH |
307 | unsigned int i; |
308 | ||
aba5acdf | 309 | SPRINT_BUF(b1); |
aba5acdf SH |
310 | |
311 | if (opt == NULL) | |
312 | return 0; | |
313 | ||
1b6f0bb5 | 314 | parse_rtattr_nested(tb, TCA_GRED_MAX, opt); |
aba5acdf SH |
315 | |
316 | if (tb[TCA_GRED_PARMS] == NULL) | |
317 | return -1; | |
a5a6f1e8 | 318 | |
1b6f0bb5 ED |
319 | if (tb[TCA_GRED_MAX_P] && |
320 | RTA_PAYLOAD(tb[TCA_GRED_MAX_P]) >= sizeof(__u32) * MAX_DPs) | |
321 | max_p = RTA_DATA(tb[TCA_GRED_MAX_P]); | |
322 | ||
aacee269 DW |
323 | if (tb[TCA_GRED_LIMIT] && |
324 | RTA_PAYLOAD(tb[TCA_GRED_LIMIT]) == sizeof(__u32)) | |
325 | limit = RTA_DATA(tb[TCA_GRED_LIMIT]); | |
326 | ||
1693a4d3 | 327 | sopt = RTA_DATA(tb[TCA_GRED_DPS]); |
aba5acdf | 328 | qopt = RTA_DATA(tb[TCA_GRED_PARMS]); |
1693a4d3 DW |
329 | if (RTA_PAYLOAD(tb[TCA_GRED_DPS]) < sizeof(*sopt) || |
330 | RTA_PAYLOAD(tb[TCA_GRED_PARMS]) < sizeof(*qopt)*MAX_DPs) { | |
32a121cb | 331 | fprintf(f, "\n GRED received message smaller than expected\n"); |
aba5acdf | 332 | return -1; |
1693a4d3 | 333 | } |
ae665a52 | 334 | |
6475e6a5 JK |
335 | print_uint(PRINT_ANY, "dp_cnt", "vqs %u ", sopt->DPs); |
336 | print_uint(PRINT_ANY, "dp_default", "default %u ", sopt->def_DP); | |
1693a4d3 | 337 | |
6475e6a5 JK |
338 | if (sopt->grio) |
339 | print_bool(PRINT_ANY, "grio", "grio ", true); | |
340 | else | |
341 | print_bool(PRINT_ANY, "grio", NULL, false); | |
342 | ||
343 | if (limit) { | |
344 | print_uint(PRINT_JSON, "limit", NULL, *limit); | |
345 | print_string(PRINT_FP, NULL, "limit %s ", | |
346 | sprint_size(*limit, b1)); | |
347 | } | |
aacee269 | 348 | |
6475e6a5 | 349 | open_json_array(PRINT_JSON, "vqs"); |
32a121cb | 350 | for (i = 0; i < MAX_DPs; i++, qopt++) { |
6475e6a5 JK |
351 | if (qopt->DP >= MAX_DPs) |
352 | continue; | |
353 | ||
354 | open_json_object(NULL); | |
355 | ||
356 | print_uint(PRINT_ANY, "vq", "\n vq %u ", qopt->DP); | |
357 | print_hhu(PRINT_ANY, "prio", "prio %hhu ", qopt->prio); | |
358 | ||
359 | print_uint(PRINT_JSON, "limit", NULL, qopt->limit); | |
360 | print_string(PRINT_FP, NULL, "limit %s ", | |
361 | sprint_size(qopt->limit, b1)); | |
362 | ||
363 | print_uint(PRINT_JSON, "min", NULL, qopt->qth_min); | |
364 | print_string(PRINT_FP, NULL, "min %s ", | |
365 | sprint_size(qopt->qth_min, b1)); | |
366 | ||
367 | print_uint(PRINT_JSON, "max", NULL, qopt->qth_max); | |
368 | print_string(PRINT_FP, NULL, "max %s ", | |
369 | sprint_size(qopt->qth_max, b1)); | |
370 | ||
1693a4d3 | 371 | if (show_details) { |
6475e6a5 | 372 | print_uint(PRINT_ANY, "ewma", "ewma %u ", qopt->Wlog); |
1693a4d3 | 373 | if (max_p) |
6475e6a5 JK |
374 | print_float(PRINT_ANY, "probability", |
375 | "probability %lg ", | |
376 | max_p[i] / pow(2, 32)); | |
1693a4d3 | 377 | else |
6475e6a5 JK |
378 | print_uint(PRINT_ANY, "Plog", "Plog %u ", |
379 | qopt->Plog); | |
380 | print_uint(PRINT_ANY, "Scell_log", "Scell_log %u ", | |
381 | qopt->Scell_log); | |
1693a4d3 | 382 | } |
c3e1cd28 JK |
383 | if (show_stats) |
384 | gred_print_stats(qopt); | |
6475e6a5 | 385 | close_json_object(); |
aba5acdf | 386 | } |
6475e6a5 | 387 | close_json_array(PRINT_JSON, "vqs"); |
aba5acdf SH |
388 | return 0; |
389 | } | |
390 | ||
95812b56 | 391 | struct qdisc_util gred_qdisc_util = { |
f2f99e2e SH |
392 | .id = "gred", |
393 | .parse_qopt = gred_parse_opt, | |
394 | .print_qopt = gred_print_opt, | |
aba5acdf | 395 | }; |