1 /* 2 * net/sched/sch_gred.c Generic Random Early Detection queue. 3 * 4 * 5 * This program is free software; you can redistribute it and/or 6 * modify it under the terms of the GNU General Public License 7 * as published by the Free Software Foundation; either version 8 * 2 of the License, or (at your option) any later version. 9 * 10 * Authors: J Hadi Salim (hadi@cyberus.ca) 1998-2002 11 * 12 * 991129: - Bug fix with grio mode 13 * - a better sing. AvgQ mode with Grio(WRED) 14 * - A finer grained VQ dequeue based on sugestion 15 * from Ren Liu 16 * - More error checks 17 * 18 * For all the glorious comments look at include/net/red.h 19 */ 20 21 #include <linux/slab.h> 22 #include <linux/module.h> 23 #include <linux/types.h> 24 #include <linux/kernel.h> 25 #include <linux/skbuff.h> 26 #include <net/pkt_cls.h> 27 #include <net/pkt_sched.h> 28 #include <net/red.h> 29 30 #define GRED_DEF_PRIO (MAX_DPs / 2) 31 #define GRED_VQ_MASK (MAX_DPs - 1) 32 33 #define GRED_VQ_RED_FLAGS (TC_RED_ECN | TC_RED_HARDDROP) 34 35 struct gred_sched_data; 36 struct gred_sched; 37 38 struct gred_sched_data { 39 u32 limit; /* HARD maximal queue length */ 40 u32 DP; /* the drop parameters */ 41 u32 red_flags; /* virtualQ version of red_flags */ 42 u64 bytesin; /* bytes seen on virtualQ so far*/ 43 u32 packetsin; /* packets seen on virtualQ so far*/ 44 u32 backlog; /* bytes on the virtualQ */ 45 u8 prio; /* the prio of this vq */ 46 47 struct red_parms parms; 48 struct red_vars vars; 49 struct red_stats stats; 50 }; 51 52 enum { 53 GRED_WRED_MODE = 1, 54 GRED_RIO_MODE, 55 }; 56 57 struct gred_sched { 58 struct gred_sched_data *tab[MAX_DPs]; 59 unsigned long flags; 60 u32 red_flags; 61 u32 DPs; 62 u32 def; 63 struct red_vars wred_set; 64 }; 65 66 static inline int gred_wred_mode(struct gred_sched *table) 67 { 68 return test_bit(GRED_WRED_MODE, &table->flags); 69 } 70 71 static inline void gred_enable_wred_mode(struct gred_sched *table) 72 { 73 __set_bit(GRED_WRED_MODE, &table->flags); 74 } 75 76 static inline void gred_disable_wred_mode(struct gred_sched *table) 77 { 78 __clear_bit(GRED_WRED_MODE, &table->flags); 79 } 80 81 static inline int gred_rio_mode(struct gred_sched *table) 82 { 83 return test_bit(GRED_RIO_MODE, &table->flags); 84 } 85 86 static inline void gred_enable_rio_mode(struct gred_sched *table) 87 { 88 __set_bit(GRED_RIO_MODE, &table->flags); 89 } 90 91 static inline void gred_disable_rio_mode(struct gred_sched *table) 92 { 93 __clear_bit(GRED_RIO_MODE, &table->flags); 94 } 95 96 static inline int gred_wred_mode_check(struct Qdisc *sch) 97 { 98 struct gred_sched *table = qdisc_priv(sch); 99 int i; 100 101 /* Really ugly O(n^2) but shouldn't be necessary too frequent. */ 102 for (i = 0; i < table->DPs; i++) { 103 struct gred_sched_data *q = table->tab[i]; 104 int n; 105 106 if (q == NULL) 107 continue; 108 109 for (n = i + 1; n < table->DPs; n++) 110 if (table->tab[n] && table->tab[n]->prio == q->prio) 111 return 1; 112 } 113 114 return 0; 115 } 116 117 static inline unsigned int gred_backlog(struct gred_sched *table, 118 struct gred_sched_data *q, 119 struct Qdisc *sch) 120 { 121 if (gred_wred_mode(table)) 122 return sch->qstats.backlog; 123 else 124 return q->backlog; 125 } 126 127 static inline u16 tc_index_to_dp(struct sk_buff *skb) 128 { 129 return skb->tc_index & GRED_VQ_MASK; 130 } 131 132 static inline void gred_load_wred_set(const struct gred_sched *table, 133 struct gred_sched_data *q) 134 { 135 q->vars.qavg = table->wred_set.qavg; 136 q->vars.qidlestart = table->wred_set.qidlestart; 137 } 138 139 static inline void gred_store_wred_set(struct gred_sched *table, 140 struct gred_sched_data *q) 141 { 142 table->wred_set.qavg = q->vars.qavg; 143 table->wred_set.qidlestart = q->vars.qidlestart; 144 } 145 146 static int gred_use_ecn(struct gred_sched_data *q) 147 { 148 return q->red_flags & TC_RED_ECN; 149 } 150 151 static int gred_use_harddrop(struct gred_sched_data *q) 152 { 153 return q->red_flags & TC_RED_HARDDROP; 154 } 155 156 static bool gred_per_vq_red_flags_used(struct gred_sched *table) 157 { 158 unsigned int i; 159 160 /* Local per-vq flags couldn't have been set unless global are 0 */ 161 if (table->red_flags) 162 return false; 163 for (i = 0; i < MAX_DPs; i++) 164 if (table->tab[i] && table->tab[i]->red_flags) 165 return true; 166 return false; 167 } 168 169 static int gred_enqueue(struct sk_buff *skb, struct Qdisc *sch, 170 struct sk_buff **to_free) 171 { 172 struct gred_sched_data *q = NULL; 173 struct gred_sched *t = qdisc_priv(sch); 174 unsigned long qavg = 0; 175 u16 dp = tc_index_to_dp(skb); 176 177 if (dp >= t->DPs || (q = t->tab[dp]) == NULL) { 178 dp = t->def; 179 180 q = t->tab[dp]; 181 if (!q) { 182 /* Pass through packets not assigned to a DP 183 * if no default DP has been configured. This 184 * allows for DP flows to be left untouched. 185 */ 186 if (likely(sch->qstats.backlog + qdisc_pkt_len(skb) <= 187 sch->limit)) 188 return qdisc_enqueue_tail(skb, sch); 189 else 190 goto drop; 191 } 192 193 /* fix tc_index? --could be controversial but needed for 194 requeueing */ 195 skb->tc_index = (skb->tc_index & ~GRED_VQ_MASK) | dp; 196 } 197 198 /* sum up all the qaves of prios < ours to get the new qave */ 199 if (!gred_wred_mode(t) && gred_rio_mode(t)) { 200 int i; 201 202 for (i = 0; i < t->DPs; i++) { 203 if (t->tab[i] && t->tab[i]->prio < q->prio && 204 !red_is_idling(&t->tab[i]->vars)) 205 qavg += t->tab[i]->vars.qavg; 206 } 207 208 } 209 210 q->packetsin++; 211 q->bytesin += qdisc_pkt_len(skb); 212 213 if (gred_wred_mode(t)) 214 gred_load_wred_set(t, q); 215 216 q->vars.qavg = red_calc_qavg(&q->parms, 217 &q->vars, 218 gred_backlog(t, q, sch)); 219 220 if (red_is_idling(&q->vars)) 221 red_end_of_idle_period(&q->vars); 222 223 if (gred_wred_mode(t)) 224 gred_store_wred_set(t, q); 225 226 switch (red_action(&q->parms, &q->vars, q->vars.qavg + qavg)) { 227 case RED_DONT_MARK: 228 break; 229 230 case RED_PROB_MARK: 231 qdisc_qstats_overlimit(sch); 232 if (!gred_use_ecn(q) || !INET_ECN_set_ce(skb)) { 233 q->stats.prob_drop++; 234 goto congestion_drop; 235 } 236 237 q->stats.prob_mark++; 238 break; 239 240 case RED_HARD_MARK: 241 qdisc_qstats_overlimit(sch); 242 if (gred_use_harddrop(q) || !gred_use_ecn(q) || 243 !INET_ECN_set_ce(skb)) { 244 q->stats.forced_drop++; 245 goto congestion_drop; 246 } 247 q->stats.forced_mark++; 248 break; 249 } 250 251 if (gred_backlog(t, q, sch) + qdisc_pkt_len(skb) <= q->limit) { 252 q->backlog += qdisc_pkt_len(skb); 253 return qdisc_enqueue_tail(skb, sch); 254 } 255 256 q->stats.pdrop++; 257 drop: 258 return qdisc_drop(skb, sch, to_free); 259 260 congestion_drop: 261 qdisc_drop(skb, sch, to_free); 262 return NET_XMIT_CN; 263 } 264 265 static struct sk_buff *gred_dequeue(struct Qdisc *sch) 266 { 267 struct sk_buff *skb; 268 struct gred_sched *t = qdisc_priv(sch); 269 270 skb = qdisc_dequeue_head(sch); 271 272 if (skb) { 273 struct gred_sched_data *q; 274 u16 dp = tc_index_to_dp(skb); 275 276 if (dp >= t->DPs || (q = t->tab[dp]) == NULL) { 277 net_warn_ratelimited("GRED: Unable to relocate VQ 0x%x after dequeue, screwing up backlog\n", 278 tc_index_to_dp(skb)); 279 } else { 280 q->backlog -= qdisc_pkt_len(skb); 281 282 if (gred_wred_mode(t)) { 283 if (!sch->qstats.backlog) 284 red_start_of_idle_period(&t->wred_set); 285 } else { 286 if (!q->backlog) 287 red_start_of_idle_period(&q->vars); 288 } 289 } 290 291 return skb; 292 } 293 294 return NULL; 295 } 296 297 static void gred_reset(struct Qdisc *sch) 298 { 299 int i; 300 struct gred_sched *t = qdisc_priv(sch); 301 302 qdisc_reset_queue(sch); 303 304 for (i = 0; i < t->DPs; i++) { 305 struct gred_sched_data *q = t->tab[i]; 306 307 if (!q) 308 continue; 309 310 red_restart(&q->vars); 311 q->backlog = 0; 312 } 313 } 314 315 static void gred_offload(struct Qdisc *sch, enum tc_gred_command command) 316 { 317 struct gred_sched *table = qdisc_priv(sch); 318 struct net_device *dev = qdisc_dev(sch); 319 struct tc_gred_qopt_offload opt = { 320 .command = command, 321 .handle = sch->handle, 322 .parent = sch->parent, 323 }; 324 325 if (!tc_can_offload(dev) || !dev->netdev_ops->ndo_setup_tc) 326 return; 327 328 if (command == TC_GRED_REPLACE) { 329 unsigned int i; 330 331 opt.set.grio_on = gred_rio_mode(table); 332 opt.set.wred_on = gred_wred_mode(table); 333 opt.set.dp_cnt = table->DPs; 334 opt.set.dp_def = table->def; 335 336 for (i = 0; i < table->DPs; i++) { 337 struct gred_sched_data *q = table->tab[i]; 338 339 if (!q) 340 continue; 341 opt.set.tab[i].present = true; 342 opt.set.tab[i].limit = q->limit; 343 opt.set.tab[i].prio = q->prio; 344 opt.set.tab[i].min = q->parms.qth_min >> q->parms.Wlog; 345 opt.set.tab[i].max = q->parms.qth_max >> q->parms.Wlog; 346 opt.set.tab[i].is_ecn = gred_use_ecn(q); 347 opt.set.tab[i].is_harddrop = gred_use_harddrop(q); 348 opt.set.tab[i].probability = q->parms.max_P; 349 opt.set.tab[i].backlog = &q->backlog; 350 } 351 opt.set.qstats = &sch->qstats; 352 } 353 354 dev->netdev_ops->ndo_setup_tc(dev, TC_SETUP_QDISC_GRED, &opt); 355 } 356 357 static int gred_offload_dump_stats(struct Qdisc *sch) 358 { 359 struct gred_sched *table = qdisc_priv(sch); 360 struct tc_gred_qopt_offload *hw_stats; 361 unsigned int i; 362 int ret; 363 364 hw_stats = kzalloc(sizeof(*hw_stats), GFP_KERNEL); 365 if (!hw_stats) 366 return -ENOMEM; 367 368 hw_stats->command = TC_GRED_STATS; 369 hw_stats->handle = sch->handle; 370 hw_stats->parent = sch->parent; 371 372 for (i = 0; i < MAX_DPs; i++) 373 if (table->tab[i]) 374 hw_stats->stats.xstats[i] = &table->tab[i]->stats; 375 376 ret = qdisc_offload_dump_helper(sch, TC_SETUP_QDISC_GRED, hw_stats); 377 /* Even if driver returns failure adjust the stats - in case offload 378 * ended but driver still wants to adjust the values. 379 */ 380 for (i = 0; i < MAX_DPs; i++) { 381 if (!table->tab[i]) 382 continue; 383 table->tab[i]->packetsin += hw_stats->stats.bstats[i].packets; 384 table->tab[i]->bytesin += hw_stats->stats.bstats[i].bytes; 385 table->tab[i]->backlog += hw_stats->stats.qstats[i].backlog; 386 387 _bstats_update(&sch->bstats, 388 hw_stats->stats.bstats[i].bytes, 389 hw_stats->stats.bstats[i].packets); 390 sch->qstats.qlen += hw_stats->stats.qstats[i].qlen; 391 sch->qstats.backlog += hw_stats->stats.qstats[i].backlog; 392 sch->qstats.drops += hw_stats->stats.qstats[i].drops; 393 sch->qstats.requeues += hw_stats->stats.qstats[i].requeues; 394 sch->qstats.overlimits += hw_stats->stats.qstats[i].overlimits; 395 } 396 397 kfree(hw_stats); 398 return ret; 399 } 400 401 static inline void gred_destroy_vq(struct gred_sched_data *q) 402 { 403 kfree(q); 404 } 405 406 static int gred_change_table_def(struct Qdisc *sch, struct nlattr *dps, 407 struct netlink_ext_ack *extack) 408 { 409 struct gred_sched *table = qdisc_priv(sch); 410 struct tc_gred_sopt *sopt; 411 bool red_flags_changed; 412 int i; 413 414 if (!dps) 415 return -EINVAL; 416 417 sopt = nla_data(dps); 418 419 if (sopt->DPs > MAX_DPs) { 420 NL_SET_ERR_MSG_MOD(extack, "number of virtual queues too high"); 421 return -EINVAL; 422 } 423 if (sopt->DPs == 0) { 424 NL_SET_ERR_MSG_MOD(extack, 425 "number of virtual queues can't be 0"); 426 return -EINVAL; 427 } 428 if (sopt->def_DP >= sopt->DPs) { 429 NL_SET_ERR_MSG_MOD(extack, "default virtual queue above virtual queue count"); 430 return -EINVAL; 431 } 432 if (sopt->flags && gred_per_vq_red_flags_used(table)) { 433 NL_SET_ERR_MSG_MOD(extack, "can't set per-Qdisc RED flags when per-virtual queue flags are used"); 434 return -EINVAL; 435 } 436 437 sch_tree_lock(sch); 438 table->DPs = sopt->DPs; 439 table->def = sopt->def_DP; 440 red_flags_changed = table->red_flags != sopt->flags; 441 table->red_flags = sopt->flags; 442 443 /* 444 * Every entry point to GRED is synchronized with the above code 445 * and the DP is checked against DPs, i.e. shadowed VQs can no 446 * longer be found so we can unlock right here. 447 */ 448 sch_tree_unlock(sch); 449 450 if (sopt->grio) { 451 gred_enable_rio_mode(table); 452 gred_disable_wred_mode(table); 453 if (gred_wred_mode_check(sch)) 454 gred_enable_wred_mode(table); 455 } else { 456 gred_disable_rio_mode(table); 457 gred_disable_wred_mode(table); 458 } 459 460 if (red_flags_changed) 461 for (i = 0; i < table->DPs; i++) 462 if (table->tab[i]) 463 table->tab[i]->red_flags = 464 table->red_flags & GRED_VQ_RED_FLAGS; 465 466 for (i = table->DPs; i < MAX_DPs; i++) { 467 if (table->tab[i]) { 468 pr_warn("GRED: Warning: Destroying shadowed VQ 0x%x\n", 469 i); 470 gred_destroy_vq(table->tab[i]); 471 table->tab[i] = NULL; 472 } 473 } 474 475 gred_offload(sch, TC_GRED_REPLACE); 476 return 0; 477 } 478 479 static inline int gred_change_vq(struct Qdisc *sch, int dp, 480 struct tc_gred_qopt *ctl, int prio, 481 u8 *stab, u32 max_P, 482 struct gred_sched_data **prealloc, 483 struct netlink_ext_ack *extack) 484 { 485 struct gred_sched *table = qdisc_priv(sch); 486 struct gred_sched_data *q = table->tab[dp]; 487 488 if (!red_check_params(ctl->qth_min, ctl->qth_max, ctl->Wlog)) { 489 NL_SET_ERR_MSG_MOD(extack, "invalid RED parameters"); 490 return -EINVAL; 491 } 492 493 if (!q) { 494 table->tab[dp] = q = *prealloc; 495 *prealloc = NULL; 496 if (!q) 497 return -ENOMEM; 498 q->red_flags = table->red_flags & GRED_VQ_RED_FLAGS; 499 } 500 501 q->DP = dp; 502 q->prio = prio; 503 if (ctl->limit > sch->limit) 504 q->limit = sch->limit; 505 else 506 q->limit = ctl->limit; 507 508 if (q->backlog == 0) 509 red_end_of_idle_period(&q->vars); 510 511 red_set_parms(&q->parms, 512 ctl->qth_min, ctl->qth_max, ctl->Wlog, ctl->Plog, 513 ctl->Scell_log, stab, max_P); 514 red_set_vars(&q->vars); 515 return 0; 516 } 517 518 static const struct nla_policy gred_vq_policy[TCA_GRED_VQ_MAX + 1] = { 519 [TCA_GRED_VQ_DP] = { .type = NLA_U32 }, 520 [TCA_GRED_VQ_FLAGS] = { .type = NLA_U32 }, 521 }; 522 523 static const struct nla_policy gred_vqe_policy[TCA_GRED_VQ_ENTRY_MAX + 1] = { 524 [TCA_GRED_VQ_ENTRY] = { .type = NLA_NESTED }, 525 }; 526 527 static const struct nla_policy gred_policy[TCA_GRED_MAX + 1] = { 528 [TCA_GRED_PARMS] = { .len = sizeof(struct tc_gred_qopt) }, 529 [TCA_GRED_STAB] = { .len = 256 }, 530 [TCA_GRED_DPS] = { .len = sizeof(struct tc_gred_sopt) }, 531 [TCA_GRED_MAX_P] = { .type = NLA_U32 }, 532 [TCA_GRED_LIMIT] = { .type = NLA_U32 }, 533 [TCA_GRED_VQ_LIST] = { .type = NLA_NESTED }, 534 }; 535 536 static void gred_vq_apply(struct gred_sched *table, const struct nlattr *entry) 537 { 538 struct nlattr *tb[TCA_GRED_VQ_MAX + 1]; 539 u32 dp; 540 541 nla_parse_nested_deprecated(tb, TCA_GRED_VQ_MAX, entry, 542 gred_vq_policy, NULL); 543 544 dp = nla_get_u32(tb[TCA_GRED_VQ_DP]); 545 546 if (tb[TCA_GRED_VQ_FLAGS]) 547 table->tab[dp]->red_flags = nla_get_u32(tb[TCA_GRED_VQ_FLAGS]); 548 } 549 550 static void gred_vqs_apply(struct gred_sched *table, struct nlattr *vqs) 551 { 552 const struct nlattr *attr; 553 int rem; 554 555 nla_for_each_nested(attr, vqs, rem) { 556 switch (nla_type(attr)) { 557 case TCA_GRED_VQ_ENTRY: 558 gred_vq_apply(table, attr); 559 break; 560 } 561 } 562 } 563 564 static int gred_vq_validate(struct gred_sched *table, u32 cdp, 565 const struct nlattr *entry, 566 struct netlink_ext_ack *extack) 567 { 568 struct nlattr *tb[TCA_GRED_VQ_MAX + 1]; 569 int err; 570 u32 dp; 571 572 err = nla_parse_nested_deprecated(tb, TCA_GRED_VQ_MAX, entry, 573 gred_vq_policy, extack); 574 if (err < 0) 575 return err; 576 577 if (!tb[TCA_GRED_VQ_DP]) { 578 NL_SET_ERR_MSG_MOD(extack, "Virtual queue with no index specified"); 579 return -EINVAL; 580 } 581 dp = nla_get_u32(tb[TCA_GRED_VQ_DP]); 582 if (dp >= table->DPs) { 583 NL_SET_ERR_MSG_MOD(extack, "Virtual queue with index out of bounds"); 584 return -EINVAL; 585 } 586 if (dp != cdp && !table->tab[dp]) { 587 NL_SET_ERR_MSG_MOD(extack, "Virtual queue not yet instantiated"); 588 return -EINVAL; 589 } 590 591 if (tb[TCA_GRED_VQ_FLAGS]) { 592 u32 red_flags = nla_get_u32(tb[TCA_GRED_VQ_FLAGS]); 593 594 if (table->red_flags && table->red_flags != red_flags) { 595 NL_SET_ERR_MSG_MOD(extack, "can't change per-virtual queue RED flags when per-Qdisc flags are used"); 596 return -EINVAL; 597 } 598 if (red_flags & ~GRED_VQ_RED_FLAGS) { 599 NL_SET_ERR_MSG_MOD(extack, 600 "invalid RED flags specified"); 601 return -EINVAL; 602 } 603 } 604 605 return 0; 606 } 607 608 static int gred_vqs_validate(struct gred_sched *table, u32 cdp, 609 struct nlattr *vqs, struct netlink_ext_ack *extack) 610 { 611 const struct nlattr *attr; 612 int rem, err; 613 614 err = nla_validate_nested_deprecated(vqs, TCA_GRED_VQ_ENTRY_MAX, 615 gred_vqe_policy, extack); 616 if (err < 0) 617 return err; 618 619 nla_for_each_nested(attr, vqs, rem) { 620 switch (nla_type(attr)) { 621 case TCA_GRED_VQ_ENTRY: 622 err = gred_vq_validate(table, cdp, attr, extack); 623 if (err) 624 return err; 625 break; 626 default: 627 NL_SET_ERR_MSG_MOD(extack, "GRED_VQ_LIST can contain only entry attributes"); 628 return -EINVAL; 629 } 630 } 631 632 if (rem > 0) { 633 NL_SET_ERR_MSG_MOD(extack, "Trailing data after parsing virtual queue list"); 634 return -EINVAL; 635 } 636 637 return 0; 638 } 639 640 static int gred_change(struct Qdisc *sch, struct nlattr *opt, 641 struct netlink_ext_ack *extack) 642 { 643 struct gred_sched *table = qdisc_priv(sch); 644 struct tc_gred_qopt *ctl; 645 struct nlattr *tb[TCA_GRED_MAX + 1]; 646 int err, prio = GRED_DEF_PRIO; 647 u8 *stab; 648 u32 max_P; 649 struct gred_sched_data *prealloc; 650 651 if (opt == NULL) 652 return -EINVAL; 653 654 err = nla_parse_nested_deprecated(tb, TCA_GRED_MAX, opt, gred_policy, 655 extack); 656 if (err < 0) 657 return err; 658 659 if (tb[TCA_GRED_PARMS] == NULL && tb[TCA_GRED_STAB] == NULL) { 660 if (tb[TCA_GRED_LIMIT] != NULL) 661 sch->limit = nla_get_u32(tb[TCA_GRED_LIMIT]); 662 return gred_change_table_def(sch, tb[TCA_GRED_DPS], extack); 663 } 664 665 if (tb[TCA_GRED_PARMS] == NULL || 666 tb[TCA_GRED_STAB] == NULL || 667 tb[TCA_GRED_LIMIT] != NULL) { 668 NL_SET_ERR_MSG_MOD(extack, "can't configure Qdisc and virtual queue at the same time"); 669 return -EINVAL; 670 } 671 672 max_P = tb[TCA_GRED_MAX_P] ? nla_get_u32(tb[TCA_GRED_MAX_P]) : 0; 673 674 ctl = nla_data(tb[TCA_GRED_PARMS]); 675 stab = nla_data(tb[TCA_GRED_STAB]); 676 677 if (ctl->DP >= table->DPs) { 678 NL_SET_ERR_MSG_MOD(extack, "virtual queue index above virtual queue count"); 679 return -EINVAL; 680 } 681 682 if (tb[TCA_GRED_VQ_LIST]) { 683 err = gred_vqs_validate(table, ctl->DP, tb[TCA_GRED_VQ_LIST], 684 extack); 685 if (err) 686 return err; 687 } 688 689 if (gred_rio_mode(table)) { 690 if (ctl->prio == 0) { 691 int def_prio = GRED_DEF_PRIO; 692 693 if (table->tab[table->def]) 694 def_prio = table->tab[table->def]->prio; 695 696 printk(KERN_DEBUG "GRED: DP %u does not have a prio " 697 "setting default to %d\n", ctl->DP, def_prio); 698 699 prio = def_prio; 700 } else 701 prio = ctl->prio; 702 } 703 704 prealloc = kzalloc(sizeof(*prealloc), GFP_KERNEL); 705 sch_tree_lock(sch); 706 707 err = gred_change_vq(sch, ctl->DP, ctl, prio, stab, max_P, &prealloc, 708 extack); 709 if (err < 0) 710 goto err_unlock_free; 711 712 if (tb[TCA_GRED_VQ_LIST]) 713 gred_vqs_apply(table, tb[TCA_GRED_VQ_LIST]); 714 715 if (gred_rio_mode(table)) { 716 gred_disable_wred_mode(table); 717 if (gred_wred_mode_check(sch)) 718 gred_enable_wred_mode(table); 719 } 720 721 sch_tree_unlock(sch); 722 kfree(prealloc); 723 724 gred_offload(sch, TC_GRED_REPLACE); 725 return 0; 726 727 err_unlock_free: 728 sch_tree_unlock(sch); 729 kfree(prealloc); 730 return err; 731 } 732 733 static int gred_init(struct Qdisc *sch, struct nlattr *opt, 734 struct netlink_ext_ack *extack) 735 { 736 struct nlattr *tb[TCA_GRED_MAX + 1]; 737 int err; 738 739 if (!opt) 740 return -EINVAL; 741 742 err = nla_parse_nested_deprecated(tb, TCA_GRED_MAX, opt, gred_policy, 743 extack); 744 if (err < 0) 745 return err; 746 747 if (tb[TCA_GRED_PARMS] || tb[TCA_GRED_STAB]) { 748 NL_SET_ERR_MSG_MOD(extack, 749 "virtual queue configuration can't be specified at initialization time"); 750 return -EINVAL; 751 } 752 753 if (tb[TCA_GRED_LIMIT]) 754 sch->limit = nla_get_u32(tb[TCA_GRED_LIMIT]); 755 else 756 sch->limit = qdisc_dev(sch)->tx_queue_len 757 * psched_mtu(qdisc_dev(sch)); 758 759 return gred_change_table_def(sch, tb[TCA_GRED_DPS], extack); 760 } 761 762 static int gred_dump(struct Qdisc *sch, struct sk_buff *skb) 763 { 764 struct gred_sched *table = qdisc_priv(sch); 765 struct nlattr *parms, *vqs, *opts = NULL; 766 int i; 767 u32 max_p[MAX_DPs]; 768 struct tc_gred_sopt sopt = { 769 .DPs = table->DPs, 770 .def_DP = table->def, 771 .grio = gred_rio_mode(table), 772 .flags = table->red_flags, 773 }; 774 775 if (gred_offload_dump_stats(sch)) 776 goto nla_put_failure; 777 778 opts = nla_nest_start_noflag(skb, TCA_OPTIONS); 779 if (opts == NULL) 780 goto nla_put_failure; 781 if (nla_put(skb, TCA_GRED_DPS, sizeof(sopt), &sopt)) 782 goto nla_put_failure; 783 784 for (i = 0; i < MAX_DPs; i++) { 785 struct gred_sched_data *q = table->tab[i]; 786 787 max_p[i] = q ? q->parms.max_P : 0; 788 } 789 if (nla_put(skb, TCA_GRED_MAX_P, sizeof(max_p), max_p)) 790 goto nla_put_failure; 791 792 if (nla_put_u32(skb, TCA_GRED_LIMIT, sch->limit)) 793 goto nla_put_failure; 794 795 /* Old style all-in-one dump of VQs */ 796 parms = nla_nest_start_noflag(skb, TCA_GRED_PARMS); 797 if (parms == NULL) 798 goto nla_put_failure; 799 800 for (i = 0; i < MAX_DPs; i++) { 801 struct gred_sched_data *q = table->tab[i]; 802 struct tc_gred_qopt opt; 803 unsigned long qavg; 804 805 memset(&opt, 0, sizeof(opt)); 806 807 if (!q) { 808 /* hack -- fix at some point with proper message 809 This is how we indicate to tc that there is no VQ 810 at this DP */ 811 812 opt.DP = MAX_DPs + i; 813 goto append_opt; 814 } 815 816 opt.limit = q->limit; 817 opt.DP = q->DP; 818 opt.backlog = gred_backlog(table, q, sch); 819 opt.prio = q->prio; 820 opt.qth_min = q->parms.qth_min >> q->parms.Wlog; 821 opt.qth_max = q->parms.qth_max >> q->parms.Wlog; 822 opt.Wlog = q->parms.Wlog; 823 opt.Plog = q->parms.Plog; 824 opt.Scell_log = q->parms.Scell_log; 825 opt.other = q->stats.other; 826 opt.early = q->stats.prob_drop; 827 opt.forced = q->stats.forced_drop; 828 opt.pdrop = q->stats.pdrop; 829 opt.packets = q->packetsin; 830 opt.bytesin = q->bytesin; 831 832 if (gred_wred_mode(table)) 833 gred_load_wred_set(table, q); 834 835 qavg = red_calc_qavg(&q->parms, &q->vars, 836 q->vars.qavg >> q->parms.Wlog); 837 opt.qave = qavg >> q->parms.Wlog; 838 839 append_opt: 840 if (nla_append(skb, sizeof(opt), &opt) < 0) 841 goto nla_put_failure; 842 } 843 844 nla_nest_end(skb, parms); 845 846 /* Dump the VQs again, in more structured way */ 847 vqs = nla_nest_start_noflag(skb, TCA_GRED_VQ_LIST); 848 if (!vqs) 849 goto nla_put_failure; 850 851 for (i = 0; i < MAX_DPs; i++) { 852 struct gred_sched_data *q = table->tab[i]; 853 struct nlattr *vq; 854 855 if (!q) 856 continue; 857 858 vq = nla_nest_start_noflag(skb, TCA_GRED_VQ_ENTRY); 859 if (!vq) 860 goto nla_put_failure; 861 862 if (nla_put_u32(skb, TCA_GRED_VQ_DP, q->DP)) 863 goto nla_put_failure; 864 865 if (nla_put_u32(skb, TCA_GRED_VQ_FLAGS, q->red_flags)) 866 goto nla_put_failure; 867 868 /* Stats */ 869 if (nla_put_u64_64bit(skb, TCA_GRED_VQ_STAT_BYTES, q->bytesin, 870 TCA_GRED_VQ_PAD)) 871 goto nla_put_failure; 872 if (nla_put_u32(skb, TCA_GRED_VQ_STAT_PACKETS, q->packetsin)) 873 goto nla_put_failure; 874 if (nla_put_u32(skb, TCA_GRED_VQ_STAT_BACKLOG, 875 gred_backlog(table, q, sch))) 876 goto nla_put_failure; 877 if (nla_put_u32(skb, TCA_GRED_VQ_STAT_PROB_DROP, 878 q->stats.prob_drop)) 879 goto nla_put_failure; 880 if (nla_put_u32(skb, TCA_GRED_VQ_STAT_PROB_MARK, 881 q->stats.prob_mark)) 882 goto nla_put_failure; 883 if (nla_put_u32(skb, TCA_GRED_VQ_STAT_FORCED_DROP, 884 q->stats.forced_drop)) 885 goto nla_put_failure; 886 if (nla_put_u32(skb, TCA_GRED_VQ_STAT_FORCED_MARK, 887 q->stats.forced_mark)) 888 goto nla_put_failure; 889 if (nla_put_u32(skb, TCA_GRED_VQ_STAT_PDROP, q->stats.pdrop)) 890 goto nla_put_failure; 891 if (nla_put_u32(skb, TCA_GRED_VQ_STAT_OTHER, q->stats.other)) 892 goto nla_put_failure; 893 894 nla_nest_end(skb, vq); 895 } 896 nla_nest_end(skb, vqs); 897 898 return nla_nest_end(skb, opts); 899 900 nla_put_failure: 901 nla_nest_cancel(skb, opts); 902 return -EMSGSIZE; 903 } 904 905 static void gred_destroy(struct Qdisc *sch) 906 { 907 struct gred_sched *table = qdisc_priv(sch); 908 int i; 909 910 for (i = 0; i < table->DPs; i++) { 911 if (table->tab[i]) 912 gred_destroy_vq(table->tab[i]); 913 } 914 gred_offload(sch, TC_GRED_DESTROY); 915 } 916 917 static struct Qdisc_ops gred_qdisc_ops __read_mostly = { 918 .id = "gred", 919 .priv_size = sizeof(struct gred_sched), 920 .enqueue = gred_enqueue, 921 .dequeue = gred_dequeue, 922 .peek = qdisc_peek_head, 923 .init = gred_init, 924 .reset = gred_reset, 925 .destroy = gred_destroy, 926 .change = gred_change, 927 .dump = gred_dump, 928 .owner = THIS_MODULE, 929 }; 930 931 static int __init gred_module_init(void) 932 { 933 return register_qdisc(&gred_qdisc_ops); 934 } 935 936 static void __exit gred_module_exit(void) 937 { 938 unregister_qdisc(&gred_qdisc_ops); 939 } 940 941 module_init(gred_module_init) 942 module_exit(gred_module_exit) 943 944 MODULE_LICENSE("GPL"); 945