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.
9 * Authors: Martin Devera, devik@cdi.cz
17 #include <sys/socket.h>
18 #include <netinet/in.h>
19 #include <arpa/inet.h>
25 #define HTB_TC_VER 0x30003
26 #if HTB_TC_VER >> 16 != TC_HTB_PROTOVER
27 #error "Different kernel and TC HTB versions"
30 static void explain(void)
32 fprintf(stderr
, "Usage: ... qdisc add ... htb [default N] [r2q N]\n"
34 " default minor id of class to which unclassified packets are sent {0}\n"
35 " r2q DRR quantums are computed as rate in Bps/r2q {10}\n"
36 " debug string of 16 numbers each 0-3 {0}\n\n"
37 " direct_qlen Limit of the direct queue {in packets}\n"
38 "... class add ... htb rate R1 [burst B1] [mpu B] [overhead O]\n"
39 " [prio P] [slot S] [pslot PS]\n"
40 " [ceil R2] [cburst B2] [mtu MTU] [quantum Q]\n"
41 " rate rate allocated to this class (class can still borrow)\n"
42 " burst max bytes burst which can be accumulated during idle period {computed}\n"
43 " mpu minimum packet size used in rate computations\n"
44 " overhead per-packet size overhead used in rate computations\n"
45 " linklay adapting to a linklayer e.g. atm\n"
46 " ceil definite upper class rate (no borrows) {rate}\n"
47 " cburst burst but for ceil {computed}\n"
48 " mtu max packet size we create rate map for {1600}\n"
49 " prio priority of leaf; lower are served first {0}\n"
50 " quantum how much bytes to serve from leaf at once {use r2q}\n"
51 "\nTC HTB version %d.%d\n", HTB_TC_VER
>>16, HTB_TC_VER
&0xffff
55 static void explain1(char *arg
)
57 fprintf(stderr
, "Illegal \"%s\"\n", arg
);
61 static int htb_parse_opt(struct qdisc_util
*qu
, int argc
,
62 char **argv
, struct nlmsghdr
*n
, const char *dev
)
64 unsigned int direct_qlen
= ~0U;
65 struct tc_htb_glob opt
= {
70 unsigned int i
; char *p
;
73 if (matches(*argv
, "r2q") == 0) {
75 if (get_u32(&opt
.rate2quantum
, *argv
, 10)) {
76 explain1("r2q"); return -1;
78 } else if (matches(*argv
, "default") == 0) {
80 if (get_u32(&opt
.defcls
, *argv
, 16)) {
81 explain1("default"); return -1;
83 } else if (matches(*argv
, "debug") == 0) {
84 NEXT_ARG(); p
= *argv
;
85 for (i
= 0; i
< 16; i
++, p
++) {
86 if (*p
< '0' || *p
> '3') break;
87 opt
.debug
|= (*p
-'0')<<(2*i
);
89 } else if (matches(*argv
, "direct_qlen") == 0) {
91 if (get_u32(&direct_qlen
, *argv
, 10)) {
92 explain1("direct_qlen"); return -1;
95 fprintf(stderr
, "What is \"%s\"?\n", *argv
);
101 tail
= NLMSG_TAIL(n
);
102 addattr_l(n
, 1024, TCA_OPTIONS
, NULL
, 0);
103 addattr_l(n
, 2024, TCA_HTB_INIT
, &opt
, NLMSG_ALIGN(sizeof(opt
)));
104 if (direct_qlen
!= ~0U)
105 addattr_l(n
, 2024, TCA_HTB_DIRECT_QLEN
,
106 &direct_qlen
, sizeof(direct_qlen
));
107 tail
->rta_len
= (void *) NLMSG_TAIL(n
) - (void *) tail
;
111 static int htb_parse_class_opt(struct qdisc_util
*qu
, int argc
, char **argv
, struct nlmsghdr
*n
, const char *dev
)
114 struct tc_htb_opt opt
= {};
115 __u32 rtab
[256], ctab
[256];
116 unsigned buffer
= 0, cbuffer
= 0;
117 int cell_log
= -1, ccell_log
= -1;
118 unsigned int mtu
= 1600; /* eth packet len */
119 unsigned short mpu
= 0;
120 unsigned short overhead
= 0;
121 unsigned int linklayer
= LINKLAYER_ETHERNET
; /* Assume ethernet */
123 __u64 ceil64
= 0, rate64
= 0;
126 if (matches(*argv
, "prio") == 0) {
128 if (get_u32(&opt
.prio
, *argv
, 10)) {
129 explain1("prio"); return -1;
132 } else if (matches(*argv
, "mtu") == 0) {
134 if (get_u32(&mtu
, *argv
, 10)) {
135 explain1("mtu"); return -1;
137 } else if (matches(*argv
, "mpu") == 0) {
139 if (get_u16(&mpu
, *argv
, 10)) {
140 explain1("mpu"); return -1;
142 } else if (matches(*argv
, "overhead") == 0) {
144 if (get_u16(&overhead
, *argv
, 10)) {
145 explain1("overhead"); return -1;
147 } else if (matches(*argv
, "linklayer") == 0) {
149 if (get_linklayer(&linklayer
, *argv
)) {
150 explain1("linklayer"); return -1;
152 } else if (matches(*argv
, "quantum") == 0) {
154 if (get_u32(&opt
.quantum
, *argv
, 10)) {
155 explain1("quantum"); return -1;
157 } else if (matches(*argv
, "burst") == 0 ||
158 strcmp(*argv
, "buffer") == 0 ||
159 strcmp(*argv
, "maxburst") == 0) {
161 if (get_size_and_cell(&buffer
, &cell_log
, *argv
) < 0) {
166 } else if (matches(*argv
, "cburst") == 0 ||
167 strcmp(*argv
, "cbuffer") == 0 ||
168 strcmp(*argv
, "cmaxburst") == 0) {
170 if (get_size_and_cell(&cbuffer
, &ccell_log
, *argv
) < 0) {
175 } else if (strcmp(*argv
, "ceil") == 0) {
178 fprintf(stderr
, "Double \"ceil\" spec\n");
181 if (strchr(*argv
, '%')) {
182 if (get_percent_rate64(&ceil64
, *argv
, dev
)) {
186 } else if (get_rate64(&ceil64
, *argv
)) {
191 } else if (strcmp(*argv
, "rate") == 0) {
194 fprintf(stderr
, "Double \"rate\" spec\n");
197 if (strchr(*argv
, '%')) {
198 if (get_percent_rate64(&rate64
, *argv
, dev
)) {
202 } else if (get_rate64(&rate64
, *argv
)) {
207 } else if (strcmp(*argv
, "help") == 0) {
211 fprintf(stderr
, "What is \"%s\"?\n", *argv
);
222 fprintf(stderr
, "\"rate\" is required.\n");
225 /* if ceil params are missing, use the same as rate */
229 opt
.rate
.rate
= (rate64
>= (1ULL << 32)) ? ~0U : rate64
;
230 opt
.ceil
.rate
= (ceil64
>= (1ULL << 32)) ? ~0U : ceil64
;
232 /* compute minimal allowed burst from rate; mtu is added here to make
233 sute that buffer is larger than mtu and to have some safeguard space */
235 buffer
= rate64
/ get_hz() + mtu
;
237 cbuffer
= ceil64
/ get_hz() + mtu
;
239 opt
.ceil
.overhead
= overhead
;
240 opt
.rate
.overhead
= overhead
;
245 if (tc_calc_rtable(&opt
.rate
, rtab
, cell_log
, mtu
, linklayer
) < 0) {
246 fprintf(stderr
, "htb: failed to calculate rate table.\n");
249 opt
.buffer
= tc_calc_xmittime(rate64
, buffer
);
251 if (tc_calc_rtable(&opt
.ceil
, ctab
, ccell_log
, mtu
, linklayer
) < 0) {
252 fprintf(stderr
, "htb: failed to calculate ceil rate table.\n");
255 opt
.cbuffer
= tc_calc_xmittime(ceil64
, cbuffer
);
257 tail
= NLMSG_TAIL(n
);
258 addattr_l(n
, 1024, TCA_OPTIONS
, NULL
, 0);
260 if (rate64
>= (1ULL << 32))
261 addattr_l(n
, 1124, TCA_HTB_RATE64
, &rate64
, sizeof(rate64
));
263 if (ceil64
>= (1ULL << 32))
264 addattr_l(n
, 1224, TCA_HTB_CEIL64
, &ceil64
, sizeof(ceil64
));
266 addattr_l(n
, 2024, TCA_HTB_PARMS
, &opt
, sizeof(opt
));
267 addattr_l(n
, 3024, TCA_HTB_RTAB
, rtab
, 1024);
268 addattr_l(n
, 4024, TCA_HTB_CTAB
, ctab
, 1024);
269 tail
->rta_len
= (void *) NLMSG_TAIL(n
) - (void *) tail
;
273 static int htb_print_opt(struct qdisc_util
*qu
, FILE *f
, struct rtattr
*opt
)
275 struct rtattr
*tb
[TCA_HTB_MAX
+ 1];
276 struct tc_htb_opt
*hopt
;
277 struct tc_htb_glob
*gopt
;
278 double buffer
, cbuffer
;
279 unsigned int linklayer
;
280 __u64 rate64
, ceil64
;
289 parse_rtattr_nested(tb
, TCA_HTB_MAX
, opt
);
291 if (tb
[TCA_HTB_PARMS
]) {
292 hopt
= RTA_DATA(tb
[TCA_HTB_PARMS
]);
293 if (RTA_PAYLOAD(tb
[TCA_HTB_PARMS
]) < sizeof(*hopt
)) return -1;
296 print_int(PRINT_ANY
, "prio", "prio ", (int)hopt
->prio
);
298 print_int(PRINT_ANY
, "quantum", "quantum ",
302 rate64
= hopt
->rate
.rate
;
303 if (tb
[TCA_HTB_RATE64
] &&
304 RTA_PAYLOAD(tb
[TCA_HTB_RATE64
]) >= sizeof(rate64
)) {
305 rate64
= rta_getattr_u64(tb
[TCA_HTB_RATE64
]);
308 ceil64
= hopt
->ceil
.rate
;
309 if (tb
[TCA_HTB_CEIL64
] &&
310 RTA_PAYLOAD(tb
[TCA_HTB_CEIL64
]) >= sizeof(ceil64
))
311 ceil64
= rta_getattr_u64(tb
[TCA_HTB_CEIL64
]);
313 fprintf(f
, "rate %s ", sprint_rate(rate64
, b1
));
314 if (hopt
->rate
.overhead
)
315 fprintf(f
, "overhead %u ", hopt
->rate
.overhead
);
316 buffer
= tc_calc_xmitsize(rate64
, hopt
->buffer
);
318 fprintf(f
, "ceil %s ", sprint_rate(ceil64
, b1
));
319 cbuffer
= tc_calc_xmitsize(ceil64
, hopt
->cbuffer
);
320 linklayer
= (hopt
->rate
.linklayer
& TC_LINKLAYER_MASK
);
321 if (linklayer
> TC_LINKLAYER_ETHERNET
|| show_details
)
322 fprintf(f
, "linklayer %s ", sprint_linklayer(linklayer
, b3
));
324 fprintf(f
, "burst %s/%u mpu %s ",
325 sprint_size(buffer
, b1
),
326 1<<hopt
->rate
.cell_log
,
327 sprint_size(hopt
->rate
.mpu
, b2
));
328 fprintf(f
, "cburst %s/%u mpu %s ",
329 sprint_size(cbuffer
, b1
),
330 1<<hopt
->ceil
.cell_log
,
331 sprint_size(hopt
->ceil
.mpu
, b2
));
332 fprintf(f
, "level %d ", (int)hopt
->level
);
334 fprintf(f
, "burst %s ", sprint_size(buffer
, b1
));
335 fprintf(f
, "cburst %s ", sprint_size(cbuffer
, b1
));
338 fprintf(f
, "buffer [%08x] cbuffer [%08x] ",
339 hopt
->buffer
, hopt
->cbuffer
);
341 if (tb
[TCA_HTB_INIT
]) {
342 gopt
= RTA_DATA(tb
[TCA_HTB_INIT
]);
343 if (RTA_PAYLOAD(tb
[TCA_HTB_INIT
]) < sizeof(*gopt
)) return -1;
345 print_int(PRINT_ANY
, "r2q", "r2q %d", gopt
->rate2quantum
);
346 print_uint(PRINT_ANY
, "default", " default %u", gopt
->defcls
);
347 print_uint(PRINT_ANY
, "direct_packets_stat",
348 " direct_packets_stat %u", gopt
->direct_pkts
);
350 sprintf(b1
, "%d.%d", gopt
->version
>> 16, gopt
->version
& 0xffff);
351 print_string(PRINT_ANY
, "ver", " ver %s", b1
);
354 if (tb
[TCA_HTB_DIRECT_QLEN
] &&
355 RTA_PAYLOAD(tb
[TCA_HTB_DIRECT_QLEN
]) >= sizeof(__u32
)) {
356 __u32 direct_qlen
= rta_getattr_u32(tb
[TCA_HTB_DIRECT_QLEN
]);
358 print_uint(PRINT_ANY
, "direct_qlen", " direct_qlen %u",
364 static int htb_print_xstats(struct qdisc_util
*qu
, FILE *f
, struct rtattr
*xstats
)
366 struct tc_htb_xstats
*st
;
371 if (RTA_PAYLOAD(xstats
) < sizeof(*st
))
374 st
= RTA_DATA(xstats
);
375 fprintf(f
, " lended: %u borrowed: %u giants: %u\n",
376 st
->lends
, st
->borrows
, st
->giants
);
377 fprintf(f
, " tokens: %d ctokens: %d\n", st
->tokens
, st
->ctokens
);
381 struct qdisc_util htb_qdisc_util
= {
383 .parse_qopt
= htb_parse_opt
,
384 .print_qopt
= htb_print_opt
,
385 .print_xstats
= htb_print_xstats
,
386 .parse_copt
= htb_parse_class_opt
,
387 .print_copt
= htb_print_opt
,