org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 1 | /* |
| 2 | * q_htb.c HTB. |
| 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: Martin Devera, devik@cdi.cz |
| 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 | |
| 23 | #include "utils.h" |
| 24 | #include "tc_util.h" |
| 25 | |
| 26 | #define HTB_TC_VER 0x30003 |
| 27 | #if HTB_TC_VER >> 16 != TC_HTB_PROTOVER |
| 28 | #error "Different kernel and TC HTB versions" |
| 29 | #endif |
| 30 | |
| 31 | static void explain(void) |
| 32 | { |
| 33 | 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" |
osdl.net!shemminger | a166d24 | 2004-07-30 20:26:15 +0000 | [diff] [blame] | 37 | "... class add ... htb rate R1 [burst B1] [mpu B] [overhead O]\n" |
| 38 | " [prio P] [slot S] [pslot PS]\n" |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 39 | " [ceil R2] [cburst B2] [mtu MTU] [quantum Q]\n" |
| 40 | " rate rate allocated to this class (class can still borrow)\n" |
| 41 | " burst max bytes burst which can be accumulated during idle period {computed}\n" |
osdl.net!shemminger | a166d24 | 2004-07-30 20:26:15 +0000 | [diff] [blame] | 42 | " mpu minimum packet size used in rate computations\n" |
| 43 | " overhead per-packet size overhead used in rate computations\n" |
Jesper Dangaard Brouer | 292f29b | 2008-04-09 23:01:01 +0200 | [diff] [blame] | 44 | " linklay adapting to a linklayer e.g. atm\n" |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 45 | " ceil definite upper class rate (no borrows) {rate}\n" |
| 46 | " cburst burst but for ceil {computed}\n" |
| 47 | " mtu max packet size we create rate map for {1600}\n" |
| 48 | " prio priority of leaf; lower are served first {0}\n" |
| 49 | " quantum how much bytes to serve from leaf at once {use r2q}\n" |
| 50 | "\nTC HTB version %d.%d\n",HTB_TC_VER>>16,HTB_TC_VER&0xffff |
| 51 | ); |
| 52 | } |
| 53 | |
| 54 | static void explain1(char *arg) |
| 55 | { |
| 56 | fprintf(stderr, "Illegal \"%s\"\n", arg); |
| 57 | explain(); |
| 58 | } |
| 59 | |
| 60 | |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 61 | static int htb_parse_opt(struct qdisc_util *qu, int argc, char **argv, struct nlmsghdr *n) |
| 62 | { |
| 63 | struct tc_htb_glob opt; |
| 64 | struct rtattr *tail; |
| 65 | unsigned i; char *p; |
| 66 | memset(&opt,0,sizeof(opt)); |
| 67 | opt.rate2quantum = 10; |
| 68 | opt.version = 3; |
| 69 | |
| 70 | while (argc > 0) { |
| 71 | if (matches(*argv, "r2q") == 0) { |
| 72 | NEXT_ARG(); |
| 73 | if (get_u32(&opt.rate2quantum, *argv, 10)) { |
| 74 | explain1("r2q"); return -1; |
| 75 | } |
| 76 | } else if (matches(*argv, "default") == 0) { |
| 77 | NEXT_ARG(); |
| 78 | if (get_u32(&opt.defcls, *argv, 16)) { |
| 79 | explain1("default"); return -1; |
| 80 | } |
| 81 | } else if (matches(*argv, "debug") == 0) { |
| 82 | NEXT_ARG(); p = *argv; |
| 83 | for (i=0; i<16; i++,p++) { |
| 84 | if (*p<'0' || *p>'3') break; |
| 85 | opt.debug |= (*p-'0')<<(2*i); |
| 86 | } |
| 87 | } else { |
| 88 | fprintf(stderr, "What is \"%s\"?\n", *argv); |
| 89 | explain(); |
| 90 | return -1; |
| 91 | } |
| 92 | argc--; argv++; |
| 93 | } |
6!tgraf | 1b52a76 | 2005-01-18 01:24:18 +0000 | [diff] [blame] | 94 | tail = NLMSG_TAIL(n); |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 95 | addattr_l(n, 1024, TCA_OPTIONS, NULL, 0); |
| 96 | addattr_l(n, 2024, TCA_HTB_INIT, &opt, NLMSG_ALIGN(sizeof(opt))); |
6!tgraf | 1b52a76 | 2005-01-18 01:24:18 +0000 | [diff] [blame] | 97 | tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail; |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 98 | return 0; |
| 99 | } |
| 100 | |
| 101 | static int htb_parse_class_opt(struct qdisc_util *qu, int argc, char **argv, struct nlmsghdr *n) |
| 102 | { |
| 103 | int ok=0; |
| 104 | struct tc_htb_opt opt; |
| 105 | __u32 rtab[256],ctab[256]; |
| 106 | unsigned buffer=0,cbuffer=0; |
osdl.net!shemminger | a166d24 | 2004-07-30 20:26:15 +0000 | [diff] [blame] | 107 | int cell_log=-1,ccell_log = -1; |
Jesper Dangaard Brouer | bccd014 | 2007-09-11 16:59:58 +0200 | [diff] [blame] | 108 | unsigned mtu; |
| 109 | unsigned short mpu = 0; |
| 110 | unsigned short overhead = 0; |
Jesper Dangaard Brouer | 292f29b | 2008-04-09 23:01:01 +0200 | [diff] [blame] | 111 | unsigned int linklayer = LINKLAYER_ETHERNET; /* Assume ethernet */ |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 112 | struct rtattr *tail; |
| 113 | |
| 114 | memset(&opt, 0, sizeof(opt)); mtu = 1600; /* eth packet len */ |
| 115 | |
| 116 | while (argc > 0) { |
| 117 | if (matches(*argv, "prio") == 0) { |
| 118 | NEXT_ARG(); |
| 119 | if (get_u32(&opt.prio, *argv, 10)) { |
| 120 | explain1("prio"); return -1; |
| 121 | } |
| 122 | ok++; |
| 123 | } else if (matches(*argv, "mtu") == 0) { |
| 124 | NEXT_ARG(); |
| 125 | if (get_u32(&mtu, *argv, 10)) { |
| 126 | explain1("mtu"); return -1; |
| 127 | } |
osdl.net!shemminger | a166d24 | 2004-07-30 20:26:15 +0000 | [diff] [blame] | 128 | } else if (matches(*argv, "mpu") == 0) { |
| 129 | NEXT_ARG(); |
Jesper Dangaard Brouer | bccd014 | 2007-09-11 16:59:58 +0200 | [diff] [blame] | 130 | if (get_u16(&mpu, *argv, 10)) { |
osdl.net!shemminger | a166d24 | 2004-07-30 20:26:15 +0000 | [diff] [blame] | 131 | explain1("mpu"); return -1; |
| 132 | } |
| 133 | } else if (matches(*argv, "overhead") == 0) { |
| 134 | NEXT_ARG(); |
Jesper Dangaard Brouer | bccd014 | 2007-09-11 16:59:58 +0200 | [diff] [blame] | 135 | if (get_u16(&overhead, *argv, 10)) { |
osdl.net!shemminger | a166d24 | 2004-07-30 20:26:15 +0000 | [diff] [blame] | 136 | explain1("overhead"); return -1; |
| 137 | } |
Jesper Dangaard Brouer | 292f29b | 2008-04-09 23:01:01 +0200 | [diff] [blame] | 138 | } else if (matches(*argv, "linklayer") == 0) { |
| 139 | NEXT_ARG(); |
| 140 | if (get_linklayer(&linklayer, *argv)) { |
| 141 | explain1("linklayer"); return -1; |
| 142 | } |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 143 | } else if (matches(*argv, "quantum") == 0) { |
| 144 | NEXT_ARG(); |
| 145 | if (get_u32(&opt.quantum, *argv, 10)) { |
| 146 | explain1("quantum"); return -1; |
| 147 | } |
| 148 | } else if (matches(*argv, "burst") == 0 || |
| 149 | strcmp(*argv, "buffer") == 0 || |
| 150 | strcmp(*argv, "maxburst") == 0) { |
| 151 | NEXT_ARG(); |
| 152 | if (get_size_and_cell(&buffer, &cell_log, *argv) < 0) { |
| 153 | explain1("buffer"); |
| 154 | return -1; |
| 155 | } |
| 156 | ok++; |
| 157 | } else if (matches(*argv, "cburst") == 0 || |
| 158 | strcmp(*argv, "cbuffer") == 0 || |
| 159 | strcmp(*argv, "cmaxburst") == 0) { |
| 160 | NEXT_ARG(); |
| 161 | if (get_size_and_cell(&cbuffer, &ccell_log, *argv) < 0) { |
| 162 | explain1("cbuffer"); |
| 163 | return -1; |
| 164 | } |
| 165 | ok++; |
| 166 | } else if (strcmp(*argv, "ceil") == 0) { |
| 167 | NEXT_ARG(); |
| 168 | if (opt.ceil.rate) { |
| 169 | fprintf(stderr, "Double \"ceil\" spec\n"); |
| 170 | return -1; |
| 171 | } |
| 172 | if (get_rate(&opt.ceil.rate, *argv)) { |
| 173 | explain1("ceil"); |
| 174 | return -1; |
| 175 | } |
| 176 | ok++; |
| 177 | } else if (strcmp(*argv, "rate") == 0) { |
| 178 | NEXT_ARG(); |
| 179 | if (opt.rate.rate) { |
| 180 | fprintf(stderr, "Double \"rate\" spec\n"); |
| 181 | return -1; |
| 182 | } |
| 183 | if (get_rate(&opt.rate.rate, *argv)) { |
| 184 | explain1("rate"); |
| 185 | return -1; |
| 186 | } |
| 187 | ok++; |
| 188 | } else if (strcmp(*argv, "help") == 0) { |
| 189 | explain(); |
| 190 | return -1; |
| 191 | } else { |
| 192 | fprintf(stderr, "What is \"%s\"?\n", *argv); |
| 193 | explain(); |
| 194 | return -1; |
| 195 | } |
| 196 | argc--; argv++; |
| 197 | } |
| 198 | |
| 199 | /* if (!ok) |
| 200 | return 0;*/ |
| 201 | |
| 202 | if (opt.rate.rate == 0) { |
| 203 | fprintf(stderr, "\"rate\" is required.\n"); |
| 204 | return -1; |
| 205 | } |
| 206 | /* if ceil params are missing, use the same as rate */ |
| 207 | if (!opt.ceil.rate) opt.ceil = opt.rate; |
| 208 | |
| 209 | /* compute minimal allowed burst from rate; mtu is added here to make |
| 210 | sute that buffer is larger than mtu and to have some safeguard space */ |
osdl.org!shemminger | d0d0e26 | 2004-06-09 18:10:55 +0000 | [diff] [blame] | 211 | if (!buffer) buffer = opt.rate.rate / get_hz() + mtu; |
| 212 | if (!cbuffer) cbuffer = opt.ceil.rate / get_hz() + mtu; |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 213 | |
Jesper Dangaard Brouer | bccd014 | 2007-09-11 16:59:58 +0200 | [diff] [blame] | 214 | opt.ceil.overhead = overhead; |
| 215 | opt.rate.overhead = overhead; |
| 216 | |
| 217 | opt.ceil.mpu = mpu; |
| 218 | opt.rate.mpu = mpu; |
osdl.net!shemminger | a166d24 | 2004-07-30 20:26:15 +0000 | [diff] [blame] | 219 | |
Jesper Dangaard Brouer | 292f29b | 2008-04-09 23:01:01 +0200 | [diff] [blame] | 220 | if (tc_calc_rtable(&opt.rate, rtab, cell_log, mtu, linklayer) < 0) { |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 221 | fprintf(stderr, "htb: failed to calculate rate table.\n"); |
| 222 | return -1; |
| 223 | } |
| 224 | opt.buffer = tc_calc_xmittime(opt.rate.rate, buffer); |
Stephen Hemminger | ae665a5 | 2006-12-05 10:10:22 -0800 | [diff] [blame] | 225 | |
Jesper Dangaard Brouer | 292f29b | 2008-04-09 23:01:01 +0200 | [diff] [blame] | 226 | if (tc_calc_rtable(&opt.ceil, ctab, ccell_log, mtu, linklayer) < 0) { |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 227 | fprintf(stderr, "htb: failed to calculate ceil rate table.\n"); |
| 228 | return -1; |
| 229 | } |
| 230 | opt.cbuffer = tc_calc_xmittime(opt.ceil.rate, cbuffer); |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 231 | |
6!tgraf | 1b52a76 | 2005-01-18 01:24:18 +0000 | [diff] [blame] | 232 | tail = NLMSG_TAIL(n); |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 233 | addattr_l(n, 1024, TCA_OPTIONS, NULL, 0); |
| 234 | addattr_l(n, 2024, TCA_HTB_PARMS, &opt, sizeof(opt)); |
| 235 | addattr_l(n, 3024, TCA_HTB_RTAB, rtab, 1024); |
| 236 | addattr_l(n, 4024, TCA_HTB_CTAB, ctab, 1024); |
6!tgraf | 1b52a76 | 2005-01-18 01:24:18 +0000 | [diff] [blame] | 237 | tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail; |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 238 | return 0; |
| 239 | } |
| 240 | |
| 241 | static int htb_print_opt(struct qdisc_util *qu, FILE *f, struct rtattr *opt) |
| 242 | { |
| 243 | struct rtattr *tb[TCA_HTB_RTAB+1]; |
| 244 | struct tc_htb_opt *hopt; |
| 245 | struct tc_htb_glob *gopt; |
| 246 | double buffer,cbuffer; |
| 247 | SPRINT_BUF(b1); |
| 248 | SPRINT_BUF(b2); |
osdl.net!shemminger | a166d24 | 2004-07-30 20:26:15 +0000 | [diff] [blame] | 249 | SPRINT_BUF(b3); |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 250 | |
| 251 | if (opt == NULL) |
| 252 | return 0; |
| 253 | |
6!tgraf | 14ee9e6 | 2005-01-18 22:11:58 +0000 | [diff] [blame] | 254 | parse_rtattr_nested(tb, TCA_HTB_RTAB, opt); |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 255 | |
| 256 | if (tb[TCA_HTB_PARMS]) { |
| 257 | |
| 258 | hopt = RTA_DATA(tb[TCA_HTB_PARMS]); |
| 259 | if (RTA_PAYLOAD(tb[TCA_HTB_PARMS]) < sizeof(*hopt)) return -1; |
| 260 | |
| 261 | if (!hopt->level) { |
| 262 | fprintf(f, "prio %d ", (int)hopt->prio); |
| 263 | if (show_details) |
| 264 | fprintf(f, "quantum %d ", (int)hopt->quantum); |
| 265 | } |
| 266 | fprintf(f, "rate %s ", sprint_rate(hopt->rate.rate, b1)); |
Patrick McHardy | 76dc0aa | 2007-03-04 20:14:57 +0100 | [diff] [blame] | 267 | buffer = tc_calc_xmitsize(hopt->rate.rate, hopt->buffer); |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 268 | fprintf(f, "ceil %s ", sprint_rate(hopt->ceil.rate, b1)); |
Patrick McHardy | 76dc0aa | 2007-03-04 20:14:57 +0100 | [diff] [blame] | 269 | cbuffer = tc_calc_xmitsize(hopt->ceil.rate, hopt->cbuffer); |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 270 | if (show_details) { |
osdl.net!shemminger | a166d24 | 2004-07-30 20:26:15 +0000 | [diff] [blame] | 271 | fprintf(f, "burst %s/%u mpu %s overhead %s ", |
| 272 | sprint_size(buffer, b1), |
| 273 | 1<<hopt->rate.cell_log, |
| 274 | sprint_size(hopt->rate.mpu&0xFF, b2), |
| 275 | sprint_size((hopt->rate.mpu>>8)&0xFF, b3)); |
| 276 | fprintf(f, "cburst %s/%u mpu %s overhead %s ", |
| 277 | sprint_size(cbuffer, b1), |
| 278 | 1<<hopt->ceil.cell_log, |
| 279 | sprint_size(hopt->ceil.mpu&0xFF, b2), |
| 280 | sprint_size((hopt->ceil.mpu>>8)&0xFF, b3)); |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 281 | fprintf(f, "level %d ", (int)hopt->level); |
| 282 | } else { |
| 283 | fprintf(f, "burst %s ", sprint_size(buffer, b1)); |
| 284 | fprintf(f, "cburst %s ", sprint_size(cbuffer, b1)); |
| 285 | } |
| 286 | if (show_raw) |
Stephen Hemminger | ae665a5 | 2006-12-05 10:10:22 -0800 | [diff] [blame] | 287 | fprintf(f, "buffer [%08x] cbuffer [%08x] ", |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 288 | hopt->buffer,hopt->cbuffer); |
| 289 | } |
| 290 | if (tb[TCA_HTB_INIT]) { |
| 291 | gopt = RTA_DATA(tb[TCA_HTB_INIT]); |
| 292 | if (RTA_PAYLOAD(tb[TCA_HTB_INIT]) < sizeof(*gopt)) return -1; |
| 293 | |
Stephen Hemminger | ae665a5 | 2006-12-05 10:10:22 -0800 | [diff] [blame] | 294 | fprintf(f, "r2q %d default %x direct_packets_stat %u", |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 295 | gopt->rate2quantum,gopt->defcls,gopt->direct_pkts); |
| 296 | if (show_details) |
| 297 | fprintf(f," ver %d.%d",gopt->version >> 16,gopt->version & 0xffff); |
| 298 | } |
| 299 | return 0; |
| 300 | } |
| 301 | |
| 302 | static int htb_print_xstats(struct qdisc_util *qu, FILE *f, struct rtattr *xstats) |
| 303 | { |
| 304 | struct tc_htb_xstats *st; |
| 305 | if (xstats == NULL) |
| 306 | return 0; |
| 307 | |
| 308 | if (RTA_PAYLOAD(xstats) < sizeof(*st)) |
| 309 | return -1; |
| 310 | |
| 311 | st = RTA_DATA(xstats); |
Stephen Hemminger | ae665a5 | 2006-12-05 10:10:22 -0800 | [diff] [blame] | 312 | fprintf(f, " lended: %u borrowed: %u giants: %u\n", |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 313 | st->lends,st->borrows,st->giants); |
| 314 | fprintf(f, " tokens: %d ctokens: %d\n", st->tokens,st->ctokens); |
| 315 | return 0; |
| 316 | } |
| 317 | |
net[shemminger]!kaber | 95812b5 | 2004-09-28 18:35:49 +0000 | [diff] [blame] | 318 | struct qdisc_util htb_qdisc_util = { |
osdl.net!shemminger | f2f99e2 | 2004-08-31 17:45:21 +0000 | [diff] [blame] | 319 | .id = "htb", |
| 320 | .parse_qopt = htb_parse_opt, |
| 321 | .print_qopt = htb_print_opt, |
| 322 | .print_xstats = htb_print_xstats, |
| 323 | .parse_copt = htb_parse_class_opt, |
| 324 | .print_copt = htb_print_opt, |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 325 | }; |
| 326 | |
| 327 | /* for testing of old one */ |
net[shemminger]!kaber | 95812b5 | 2004-09-28 18:35:49 +0000 | [diff] [blame] | 328 | struct qdisc_util htb2_qdisc_util = { |
osdl.net!shemminger | f2f99e2 | 2004-08-31 17:45:21 +0000 | [diff] [blame] | 329 | .id = "htb2", |
| 330 | .parse_qopt = htb_parse_opt, |
| 331 | .print_qopt = htb_print_opt, |
| 332 | .print_xstats = htb_print_xstats, |
| 333 | .parse_copt = htb_parse_class_opt, |
| 334 | .print_copt = htb_print_opt, |
org[shemminger]!shemminger | 9e9d615 | 2004-06-07 22:04:51 +0000 | [diff] [blame] | 335 | }; |