Merge master.kernel.org:/home/rmk/linux-2.6-arm
[linux-2.6/linux-acpi-2.6/ibm-acpi-2.6.git] / net / sched / sch_api.c
blob145268ca57cfeffdfea5751e19c738964b2637db
1 /*
2 * net/sched/sch_api.c Packet scheduler API.
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: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
11 * Fixes:
13 * Rani Assaf <rani@magic.metawire.com> :980802: JIFFIES and CPU clock sources are repaired.
14 * Eduardo J. Blanco <ejbs@netlabs.com.uy> :990222: kmod support
15 * Jamal Hadi Salim <hadi@nortelnetworks.com>: 990601: ingress support
18 #include <linux/module.h>
19 #include <linux/types.h>
20 #include <linux/kernel.h>
21 #include <linux/string.h>
22 #include <linux/errno.h>
23 #include <linux/skbuff.h>
24 #include <linux/init.h>
25 #include <linux/proc_fs.h>
26 #include <linux/seq_file.h>
27 #include <linux/kmod.h>
28 #include <linux/list.h>
29 #include <linux/hrtimer.h>
30 #include <linux/lockdep.h>
31 #include <linux/slab.h>
33 #include <net/net_namespace.h>
34 #include <net/sock.h>
35 #include <net/netlink.h>
36 #include <net/pkt_sched.h>
38 static int qdisc_notify(struct sk_buff *oskb, struct nlmsghdr *n, u32 clid,
39 struct Qdisc *old, struct Qdisc *new);
40 static int tclass_notify(struct sk_buff *oskb, struct nlmsghdr *n,
41 struct Qdisc *q, unsigned long cl, int event);
45 Short review.
46 -------------
48 This file consists of two interrelated parts:
50 1. queueing disciplines manager frontend.
51 2. traffic classes manager frontend.
53 Generally, queueing discipline ("qdisc") is a black box,
54 which is able to enqueue packets and to dequeue them (when
55 device is ready to send something) in order and at times
56 determined by algorithm hidden in it.
58 qdisc's are divided to two categories:
59 - "queues", which have no internal structure visible from outside.
60 - "schedulers", which split all the packets to "traffic classes",
61 using "packet classifiers" (look at cls_api.c)
63 In turn, classes may have child qdiscs (as rule, queues)
64 attached to them etc. etc. etc.
66 The goal of the routines in this file is to translate
67 information supplied by user in the form of handles
68 to more intelligible for kernel form, to make some sanity
69 checks and part of work, which is common to all qdiscs
70 and to provide rtnetlink notifications.
72 All real intelligent work is done inside qdisc modules.
76 Every discipline has two major routines: enqueue and dequeue.
78 ---dequeue
80 dequeue usually returns a skb to send. It is allowed to return NULL,
81 but it does not mean that queue is empty, it just means that
82 discipline does not want to send anything this time.
83 Queue is really empty if q->q.qlen == 0.
84 For complicated disciplines with multiple queues q->q is not
85 real packet queue, but however q->q.qlen must be valid.
87 ---enqueue
89 enqueue returns 0, if packet was enqueued successfully.
90 If packet (this one or another one) was dropped, it returns
91 not zero error code.
92 NET_XMIT_DROP - this packet dropped
93 Expected action: do not backoff, but wait until queue will clear.
94 NET_XMIT_CN - probably this packet enqueued, but another one dropped.
95 Expected action: backoff or ignore
96 NET_XMIT_POLICED - dropped by police.
97 Expected action: backoff or error to real-time apps.
99 Auxiliary routines:
101 ---peek
103 like dequeue but without removing a packet from the queue
105 ---reset
107 returns qdisc to initial state: purge all buffers, clear all
108 timers, counters (except for statistics) etc.
110 ---init
112 initializes newly created qdisc.
114 ---destroy
116 destroys resources allocated by init and during lifetime of qdisc.
118 ---change
120 changes qdisc parameters.
123 /* Protects list of registered TC modules. It is pure SMP lock. */
124 static DEFINE_RWLOCK(qdisc_mod_lock);
127 /************************************************
128 * Queueing disciplines manipulation. *
129 ************************************************/
132 /* The list of all installed queueing disciplines. */
134 static struct Qdisc_ops *qdisc_base;
136 /* Register/uregister queueing discipline */
138 int register_qdisc(struct Qdisc_ops *qops)
140 struct Qdisc_ops *q, **qp;
141 int rc = -EEXIST;
143 write_lock(&qdisc_mod_lock);
144 for (qp = &qdisc_base; (q = *qp) != NULL; qp = &q->next)
145 if (!strcmp(qops->id, q->id))
146 goto out;
148 if (qops->enqueue == NULL)
149 qops->enqueue = noop_qdisc_ops.enqueue;
150 if (qops->peek == NULL) {
151 if (qops->dequeue == NULL) {
152 qops->peek = noop_qdisc_ops.peek;
153 } else {
154 rc = -EINVAL;
155 goto out;
158 if (qops->dequeue == NULL)
159 qops->dequeue = noop_qdisc_ops.dequeue;
161 qops->next = NULL;
162 *qp = qops;
163 rc = 0;
164 out:
165 write_unlock(&qdisc_mod_lock);
166 return rc;
168 EXPORT_SYMBOL(register_qdisc);
170 int unregister_qdisc(struct Qdisc_ops *qops)
172 struct Qdisc_ops *q, **qp;
173 int err = -ENOENT;
175 write_lock(&qdisc_mod_lock);
176 for (qp = &qdisc_base; (q=*qp)!=NULL; qp = &q->next)
177 if (q == qops)
178 break;
179 if (q) {
180 *qp = q->next;
181 q->next = NULL;
182 err = 0;
184 write_unlock(&qdisc_mod_lock);
185 return err;
187 EXPORT_SYMBOL(unregister_qdisc);
189 /* We know handle. Find qdisc among all qdisc's attached to device
190 (root qdisc, all its children, children of children etc.)
193 static struct Qdisc *qdisc_match_from_root(struct Qdisc *root, u32 handle)
195 struct Qdisc *q;
197 if (!(root->flags & TCQ_F_BUILTIN) &&
198 root->handle == handle)
199 return root;
201 list_for_each_entry(q, &root->list, list) {
202 if (q->handle == handle)
203 return q;
205 return NULL;
208 static void qdisc_list_add(struct Qdisc *q)
210 if ((q->parent != TC_H_ROOT) && !(q->flags & TCQ_F_INGRESS))
211 list_add_tail(&q->list, &qdisc_dev(q)->qdisc->list);
214 void qdisc_list_del(struct Qdisc *q)
216 if ((q->parent != TC_H_ROOT) && !(q->flags & TCQ_F_INGRESS))
217 list_del(&q->list);
219 EXPORT_SYMBOL(qdisc_list_del);
221 struct Qdisc *qdisc_lookup(struct net_device *dev, u32 handle)
223 struct Qdisc *q;
225 q = qdisc_match_from_root(dev->qdisc, handle);
226 if (q)
227 goto out;
229 q = qdisc_match_from_root(dev->rx_queue.qdisc_sleeping, handle);
230 out:
231 return q;
234 static struct Qdisc *qdisc_leaf(struct Qdisc *p, u32 classid)
236 unsigned long cl;
237 struct Qdisc *leaf;
238 const struct Qdisc_class_ops *cops = p->ops->cl_ops;
240 if (cops == NULL)
241 return NULL;
242 cl = cops->get(p, classid);
244 if (cl == 0)
245 return NULL;
246 leaf = cops->leaf(p, cl);
247 cops->put(p, cl);
248 return leaf;
251 /* Find queueing discipline by name */
253 static struct Qdisc_ops *qdisc_lookup_ops(struct nlattr *kind)
255 struct Qdisc_ops *q = NULL;
257 if (kind) {
258 read_lock(&qdisc_mod_lock);
259 for (q = qdisc_base; q; q = q->next) {
260 if (nla_strcmp(kind, q->id) == 0) {
261 if (!try_module_get(q->owner))
262 q = NULL;
263 break;
266 read_unlock(&qdisc_mod_lock);
268 return q;
271 static struct qdisc_rate_table *qdisc_rtab_list;
273 struct qdisc_rate_table *qdisc_get_rtab(struct tc_ratespec *r, struct nlattr *tab)
275 struct qdisc_rate_table *rtab;
277 for (rtab = qdisc_rtab_list; rtab; rtab = rtab->next) {
278 if (memcmp(&rtab->rate, r, sizeof(struct tc_ratespec)) == 0) {
279 rtab->refcnt++;
280 return rtab;
284 if (tab == NULL || r->rate == 0 || r->cell_log == 0 ||
285 nla_len(tab) != TC_RTAB_SIZE)
286 return NULL;
288 rtab = kmalloc(sizeof(*rtab), GFP_KERNEL);
289 if (rtab) {
290 rtab->rate = *r;
291 rtab->refcnt = 1;
292 memcpy(rtab->data, nla_data(tab), 1024);
293 rtab->next = qdisc_rtab_list;
294 qdisc_rtab_list = rtab;
296 return rtab;
298 EXPORT_SYMBOL(qdisc_get_rtab);
300 void qdisc_put_rtab(struct qdisc_rate_table *tab)
302 struct qdisc_rate_table *rtab, **rtabp;
304 if (!tab || --tab->refcnt)
305 return;
307 for (rtabp = &qdisc_rtab_list; (rtab=*rtabp) != NULL; rtabp = &rtab->next) {
308 if (rtab == tab) {
309 *rtabp = rtab->next;
310 kfree(rtab);
311 return;
315 EXPORT_SYMBOL(qdisc_put_rtab);
317 static LIST_HEAD(qdisc_stab_list);
318 static DEFINE_SPINLOCK(qdisc_stab_lock);
320 static const struct nla_policy stab_policy[TCA_STAB_MAX + 1] = {
321 [TCA_STAB_BASE] = { .len = sizeof(struct tc_sizespec) },
322 [TCA_STAB_DATA] = { .type = NLA_BINARY },
325 static struct qdisc_size_table *qdisc_get_stab(struct nlattr *opt)
327 struct nlattr *tb[TCA_STAB_MAX + 1];
328 struct qdisc_size_table *stab;
329 struct tc_sizespec *s;
330 unsigned int tsize = 0;
331 u16 *tab = NULL;
332 int err;
334 err = nla_parse_nested(tb, TCA_STAB_MAX, opt, stab_policy);
335 if (err < 0)
336 return ERR_PTR(err);
337 if (!tb[TCA_STAB_BASE])
338 return ERR_PTR(-EINVAL);
340 s = nla_data(tb[TCA_STAB_BASE]);
342 if (s->tsize > 0) {
343 if (!tb[TCA_STAB_DATA])
344 return ERR_PTR(-EINVAL);
345 tab = nla_data(tb[TCA_STAB_DATA]);
346 tsize = nla_len(tb[TCA_STAB_DATA]) / sizeof(u16);
349 if (!s || tsize != s->tsize || (!tab && tsize > 0))
350 return ERR_PTR(-EINVAL);
352 spin_lock(&qdisc_stab_lock);
354 list_for_each_entry(stab, &qdisc_stab_list, list) {
355 if (memcmp(&stab->szopts, s, sizeof(*s)))
356 continue;
357 if (tsize > 0 && memcmp(stab->data, tab, tsize * sizeof(u16)))
358 continue;
359 stab->refcnt++;
360 spin_unlock(&qdisc_stab_lock);
361 return stab;
364 spin_unlock(&qdisc_stab_lock);
366 stab = kmalloc(sizeof(*stab) + tsize * sizeof(u16), GFP_KERNEL);
367 if (!stab)
368 return ERR_PTR(-ENOMEM);
370 stab->refcnt = 1;
371 stab->szopts = *s;
372 if (tsize > 0)
373 memcpy(stab->data, tab, tsize * sizeof(u16));
375 spin_lock(&qdisc_stab_lock);
376 list_add_tail(&stab->list, &qdisc_stab_list);
377 spin_unlock(&qdisc_stab_lock);
379 return stab;
382 void qdisc_put_stab(struct qdisc_size_table *tab)
384 if (!tab)
385 return;
387 spin_lock(&qdisc_stab_lock);
389 if (--tab->refcnt == 0) {
390 list_del(&tab->list);
391 kfree(tab);
394 spin_unlock(&qdisc_stab_lock);
396 EXPORT_SYMBOL(qdisc_put_stab);
398 static int qdisc_dump_stab(struct sk_buff *skb, struct qdisc_size_table *stab)
400 struct nlattr *nest;
402 nest = nla_nest_start(skb, TCA_STAB);
403 if (nest == NULL)
404 goto nla_put_failure;
405 NLA_PUT(skb, TCA_STAB_BASE, sizeof(stab->szopts), &stab->szopts);
406 nla_nest_end(skb, nest);
408 return skb->len;
410 nla_put_failure:
411 return -1;
414 void qdisc_calculate_pkt_len(struct sk_buff *skb, struct qdisc_size_table *stab)
416 int pkt_len, slot;
418 pkt_len = skb->len + stab->szopts.overhead;
419 if (unlikely(!stab->szopts.tsize))
420 goto out;
422 slot = pkt_len + stab->szopts.cell_align;
423 if (unlikely(slot < 0))
424 slot = 0;
426 slot >>= stab->szopts.cell_log;
427 if (likely(slot < stab->szopts.tsize))
428 pkt_len = stab->data[slot];
429 else
430 pkt_len = stab->data[stab->szopts.tsize - 1] *
431 (slot / stab->szopts.tsize) +
432 stab->data[slot % stab->szopts.tsize];
434 pkt_len <<= stab->szopts.size_log;
435 out:
436 if (unlikely(pkt_len < 1))
437 pkt_len = 1;
438 qdisc_skb_cb(skb)->pkt_len = pkt_len;
440 EXPORT_SYMBOL(qdisc_calculate_pkt_len);
442 void qdisc_warn_nonwc(char *txt, struct Qdisc *qdisc)
444 if (!(qdisc->flags & TCQ_F_WARN_NONWC)) {
445 printk(KERN_WARNING
446 "%s: %s qdisc %X: is non-work-conserving?\n",
447 txt, qdisc->ops->id, qdisc->handle >> 16);
448 qdisc->flags |= TCQ_F_WARN_NONWC;
451 EXPORT_SYMBOL(qdisc_warn_nonwc);
453 static enum hrtimer_restart qdisc_watchdog(struct hrtimer *timer)
455 struct qdisc_watchdog *wd = container_of(timer, struct qdisc_watchdog,
456 timer);
458 wd->qdisc->flags &= ~TCQ_F_THROTTLED;
459 __netif_schedule(qdisc_root(wd->qdisc));
461 return HRTIMER_NORESTART;
464 void qdisc_watchdog_init(struct qdisc_watchdog *wd, struct Qdisc *qdisc)
466 hrtimer_init(&wd->timer, CLOCK_MONOTONIC, HRTIMER_MODE_ABS);
467 wd->timer.function = qdisc_watchdog;
468 wd->qdisc = qdisc;
470 EXPORT_SYMBOL(qdisc_watchdog_init);
472 void qdisc_watchdog_schedule(struct qdisc_watchdog *wd, psched_time_t expires)
474 ktime_t time;
476 if (test_bit(__QDISC_STATE_DEACTIVATED,
477 &qdisc_root_sleeping(wd->qdisc)->state))
478 return;
480 wd->qdisc->flags |= TCQ_F_THROTTLED;
481 time = ktime_set(0, 0);
482 time = ktime_add_ns(time, PSCHED_TICKS2NS(expires));
483 hrtimer_start(&wd->timer, time, HRTIMER_MODE_ABS);
485 EXPORT_SYMBOL(qdisc_watchdog_schedule);
487 void qdisc_watchdog_cancel(struct qdisc_watchdog *wd)
489 hrtimer_cancel(&wd->timer);
490 wd->qdisc->flags &= ~TCQ_F_THROTTLED;
492 EXPORT_SYMBOL(qdisc_watchdog_cancel);
494 static struct hlist_head *qdisc_class_hash_alloc(unsigned int n)
496 unsigned int size = n * sizeof(struct hlist_head), i;
497 struct hlist_head *h;
499 if (size <= PAGE_SIZE)
500 h = kmalloc(size, GFP_KERNEL);
501 else
502 h = (struct hlist_head *)
503 __get_free_pages(GFP_KERNEL, get_order(size));
505 if (h != NULL) {
506 for (i = 0; i < n; i++)
507 INIT_HLIST_HEAD(&h[i]);
509 return h;
512 static void qdisc_class_hash_free(struct hlist_head *h, unsigned int n)
514 unsigned int size = n * sizeof(struct hlist_head);
516 if (size <= PAGE_SIZE)
517 kfree(h);
518 else
519 free_pages((unsigned long)h, get_order(size));
522 void qdisc_class_hash_grow(struct Qdisc *sch, struct Qdisc_class_hash *clhash)
524 struct Qdisc_class_common *cl;
525 struct hlist_node *n, *next;
526 struct hlist_head *nhash, *ohash;
527 unsigned int nsize, nmask, osize;
528 unsigned int i, h;
530 /* Rehash when load factor exceeds 0.75 */
531 if (clhash->hashelems * 4 <= clhash->hashsize * 3)
532 return;
533 nsize = clhash->hashsize * 2;
534 nmask = nsize - 1;
535 nhash = qdisc_class_hash_alloc(nsize);
536 if (nhash == NULL)
537 return;
539 ohash = clhash->hash;
540 osize = clhash->hashsize;
542 sch_tree_lock(sch);
543 for (i = 0; i < osize; i++) {
544 hlist_for_each_entry_safe(cl, n, next, &ohash[i], hnode) {
545 h = qdisc_class_hash(cl->classid, nmask);
546 hlist_add_head(&cl->hnode, &nhash[h]);
549 clhash->hash = nhash;
550 clhash->hashsize = nsize;
551 clhash->hashmask = nmask;
552 sch_tree_unlock(sch);
554 qdisc_class_hash_free(ohash, osize);
556 EXPORT_SYMBOL(qdisc_class_hash_grow);
558 int qdisc_class_hash_init(struct Qdisc_class_hash *clhash)
560 unsigned int size = 4;
562 clhash->hash = qdisc_class_hash_alloc(size);
563 if (clhash->hash == NULL)
564 return -ENOMEM;
565 clhash->hashsize = size;
566 clhash->hashmask = size - 1;
567 clhash->hashelems = 0;
568 return 0;
570 EXPORT_SYMBOL(qdisc_class_hash_init);
572 void qdisc_class_hash_destroy(struct Qdisc_class_hash *clhash)
574 qdisc_class_hash_free(clhash->hash, clhash->hashsize);
576 EXPORT_SYMBOL(qdisc_class_hash_destroy);
578 void qdisc_class_hash_insert(struct Qdisc_class_hash *clhash,
579 struct Qdisc_class_common *cl)
581 unsigned int h;
583 INIT_HLIST_NODE(&cl->hnode);
584 h = qdisc_class_hash(cl->classid, clhash->hashmask);
585 hlist_add_head(&cl->hnode, &clhash->hash[h]);
586 clhash->hashelems++;
588 EXPORT_SYMBOL(qdisc_class_hash_insert);
590 void qdisc_class_hash_remove(struct Qdisc_class_hash *clhash,
591 struct Qdisc_class_common *cl)
593 hlist_del(&cl->hnode);
594 clhash->hashelems--;
596 EXPORT_SYMBOL(qdisc_class_hash_remove);
598 /* Allocate an unique handle from space managed by kernel */
600 static u32 qdisc_alloc_handle(struct net_device *dev)
602 int i = 0x10000;
603 static u32 autohandle = TC_H_MAKE(0x80000000U, 0);
605 do {
606 autohandle += TC_H_MAKE(0x10000U, 0);
607 if (autohandle == TC_H_MAKE(TC_H_ROOT, 0))
608 autohandle = TC_H_MAKE(0x80000000U, 0);
609 } while (qdisc_lookup(dev, autohandle) && --i > 0);
611 return i>0 ? autohandle : 0;
614 void qdisc_tree_decrease_qlen(struct Qdisc *sch, unsigned int n)
616 const struct Qdisc_class_ops *cops;
617 unsigned long cl;
618 u32 parentid;
620 if (n == 0)
621 return;
622 while ((parentid = sch->parent)) {
623 if (TC_H_MAJ(parentid) == TC_H_MAJ(TC_H_INGRESS))
624 return;
626 sch = qdisc_lookup(qdisc_dev(sch), TC_H_MAJ(parentid));
627 if (sch == NULL) {
628 WARN_ON(parentid != TC_H_ROOT);
629 return;
631 cops = sch->ops->cl_ops;
632 if (cops->qlen_notify) {
633 cl = cops->get(sch, parentid);
634 cops->qlen_notify(sch, cl);
635 cops->put(sch, cl);
637 sch->q.qlen -= n;
640 EXPORT_SYMBOL(qdisc_tree_decrease_qlen);
642 static void notify_and_destroy(struct sk_buff *skb, struct nlmsghdr *n, u32 clid,
643 struct Qdisc *old, struct Qdisc *new)
645 if (new || old)
646 qdisc_notify(skb, n, clid, old, new);
648 if (old)
649 qdisc_destroy(old);
652 /* Graft qdisc "new" to class "classid" of qdisc "parent" or
653 * to device "dev".
655 * When appropriate send a netlink notification using 'skb'
656 * and "n".
658 * On success, destroy old qdisc.
661 static int qdisc_graft(struct net_device *dev, struct Qdisc *parent,
662 struct sk_buff *skb, struct nlmsghdr *n, u32 classid,
663 struct Qdisc *new, struct Qdisc *old)
665 struct Qdisc *q = old;
666 int err = 0;
668 if (parent == NULL) {
669 unsigned int i, num_q, ingress;
671 ingress = 0;
672 num_q = dev->num_tx_queues;
673 if ((q && q->flags & TCQ_F_INGRESS) ||
674 (new && new->flags & TCQ_F_INGRESS)) {
675 num_q = 1;
676 ingress = 1;
679 if (dev->flags & IFF_UP)
680 dev_deactivate(dev);
682 if (new && new->ops->attach) {
683 new->ops->attach(new);
684 num_q = 0;
687 for (i = 0; i < num_q; i++) {
688 struct netdev_queue *dev_queue = &dev->rx_queue;
690 if (!ingress)
691 dev_queue = netdev_get_tx_queue(dev, i);
693 old = dev_graft_qdisc(dev_queue, new);
694 if (new && i > 0)
695 atomic_inc(&new->refcnt);
697 if (!ingress)
698 qdisc_destroy(old);
701 if (!ingress) {
702 notify_and_destroy(skb, n, classid, dev->qdisc, new);
703 if (new && !new->ops->attach)
704 atomic_inc(&new->refcnt);
705 dev->qdisc = new ? : &noop_qdisc;
706 } else {
707 notify_and_destroy(skb, n, classid, old, new);
710 if (dev->flags & IFF_UP)
711 dev_activate(dev);
712 } else {
713 const struct Qdisc_class_ops *cops = parent->ops->cl_ops;
715 err = -EOPNOTSUPP;
716 if (cops && cops->graft) {
717 unsigned long cl = cops->get(parent, classid);
718 if (cl) {
719 err = cops->graft(parent, cl, new, &old);
720 cops->put(parent, cl);
721 } else
722 err = -ENOENT;
724 if (!err)
725 notify_and_destroy(skb, n, classid, old, new);
727 return err;
730 /* lockdep annotation is needed for ingress; egress gets it only for name */
731 static struct lock_class_key qdisc_tx_lock;
732 static struct lock_class_key qdisc_rx_lock;
735 Allocate and initialize new qdisc.
737 Parameters are passed via opt.
740 static struct Qdisc *
741 qdisc_create(struct net_device *dev, struct netdev_queue *dev_queue,
742 struct Qdisc *p, u32 parent, u32 handle,
743 struct nlattr **tca, int *errp)
745 int err;
746 struct nlattr *kind = tca[TCA_KIND];
747 struct Qdisc *sch;
748 struct Qdisc_ops *ops;
749 struct qdisc_size_table *stab;
751 ops = qdisc_lookup_ops(kind);
752 #ifdef CONFIG_MODULES
753 if (ops == NULL && kind != NULL) {
754 char name[IFNAMSIZ];
755 if (nla_strlcpy(name, kind, IFNAMSIZ) < IFNAMSIZ) {
756 /* We dropped the RTNL semaphore in order to
757 * perform the module load. So, even if we
758 * succeeded in loading the module we have to
759 * tell the caller to replay the request. We
760 * indicate this using -EAGAIN.
761 * We replay the request because the device may
762 * go away in the mean time.
764 rtnl_unlock();
765 request_module("sch_%s", name);
766 rtnl_lock();
767 ops = qdisc_lookup_ops(kind);
768 if (ops != NULL) {
769 /* We will try again qdisc_lookup_ops,
770 * so don't keep a reference.
772 module_put(ops->owner);
773 err = -EAGAIN;
774 goto err_out;
778 #endif
780 err = -ENOENT;
781 if (ops == NULL)
782 goto err_out;
784 sch = qdisc_alloc(dev_queue, ops);
785 if (IS_ERR(sch)) {
786 err = PTR_ERR(sch);
787 goto err_out2;
790 sch->parent = parent;
792 if (handle == TC_H_INGRESS) {
793 sch->flags |= TCQ_F_INGRESS;
794 handle = TC_H_MAKE(TC_H_INGRESS, 0);
795 lockdep_set_class(qdisc_lock(sch), &qdisc_rx_lock);
796 } else {
797 if (handle == 0) {
798 handle = qdisc_alloc_handle(dev);
799 err = -ENOMEM;
800 if (handle == 0)
801 goto err_out3;
803 lockdep_set_class(qdisc_lock(sch), &qdisc_tx_lock);
806 sch->handle = handle;
808 if (!ops->init || (err = ops->init(sch, tca[TCA_OPTIONS])) == 0) {
809 if (tca[TCA_STAB]) {
810 stab = qdisc_get_stab(tca[TCA_STAB]);
811 if (IS_ERR(stab)) {
812 err = PTR_ERR(stab);
813 goto err_out4;
815 sch->stab = stab;
817 if (tca[TCA_RATE]) {
818 spinlock_t *root_lock;
820 err = -EOPNOTSUPP;
821 if (sch->flags & TCQ_F_MQROOT)
822 goto err_out4;
824 if ((sch->parent != TC_H_ROOT) &&
825 !(sch->flags & TCQ_F_INGRESS) &&
826 (!p || !(p->flags & TCQ_F_MQROOT)))
827 root_lock = qdisc_root_sleeping_lock(sch);
828 else
829 root_lock = qdisc_lock(sch);
831 err = gen_new_estimator(&sch->bstats, &sch->rate_est,
832 root_lock, tca[TCA_RATE]);
833 if (err)
834 goto err_out4;
837 qdisc_list_add(sch);
839 return sch;
841 err_out3:
842 dev_put(dev);
843 kfree((char *) sch - sch->padded);
844 err_out2:
845 module_put(ops->owner);
846 err_out:
847 *errp = err;
848 return NULL;
850 err_out4:
852 * Any broken qdiscs that would require a ops->reset() here?
853 * The qdisc was never in action so it shouldn't be necessary.
855 qdisc_put_stab(sch->stab);
856 if (ops->destroy)
857 ops->destroy(sch);
858 goto err_out3;
861 static int qdisc_change(struct Qdisc *sch, struct nlattr **tca)
863 struct qdisc_size_table *stab = NULL;
864 int err = 0;
866 if (tca[TCA_OPTIONS]) {
867 if (sch->ops->change == NULL)
868 return -EINVAL;
869 err = sch->ops->change(sch, tca[TCA_OPTIONS]);
870 if (err)
871 return err;
874 if (tca[TCA_STAB]) {
875 stab = qdisc_get_stab(tca[TCA_STAB]);
876 if (IS_ERR(stab))
877 return PTR_ERR(stab);
880 qdisc_put_stab(sch->stab);
881 sch->stab = stab;
883 if (tca[TCA_RATE]) {
884 /* NB: ignores errors from replace_estimator
885 because change can't be undone. */
886 if (sch->flags & TCQ_F_MQROOT)
887 goto out;
888 gen_replace_estimator(&sch->bstats, &sch->rate_est,
889 qdisc_root_sleeping_lock(sch),
890 tca[TCA_RATE]);
892 out:
893 return 0;
896 struct check_loop_arg
898 struct qdisc_walker w;
899 struct Qdisc *p;
900 int depth;
903 static int check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w);
905 static int check_loop(struct Qdisc *q, struct Qdisc *p, int depth)
907 struct check_loop_arg arg;
909 if (q->ops->cl_ops == NULL)
910 return 0;
912 arg.w.stop = arg.w.skip = arg.w.count = 0;
913 arg.w.fn = check_loop_fn;
914 arg.depth = depth;
915 arg.p = p;
916 q->ops->cl_ops->walk(q, &arg.w);
917 return arg.w.stop ? -ELOOP : 0;
920 static int
921 check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w)
923 struct Qdisc *leaf;
924 const struct Qdisc_class_ops *cops = q->ops->cl_ops;
925 struct check_loop_arg *arg = (struct check_loop_arg *)w;
927 leaf = cops->leaf(q, cl);
928 if (leaf) {
929 if (leaf == arg->p || arg->depth > 7)
930 return -ELOOP;
931 return check_loop(leaf, arg->p, arg->depth + 1);
933 return 0;
937 * Delete/get qdisc.
940 static int tc_get_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
942 struct net *net = sock_net(skb->sk);
943 struct tcmsg *tcm = NLMSG_DATA(n);
944 struct nlattr *tca[TCA_MAX + 1];
945 struct net_device *dev;
946 u32 clid = tcm->tcm_parent;
947 struct Qdisc *q = NULL;
948 struct Qdisc *p = NULL;
949 int err;
951 if (!net_eq(net, &init_net))
952 return -EINVAL;
954 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
955 return -ENODEV;
957 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
958 if (err < 0)
959 return err;
961 if (clid) {
962 if (clid != TC_H_ROOT) {
963 if (TC_H_MAJ(clid) != TC_H_MAJ(TC_H_INGRESS)) {
964 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
965 return -ENOENT;
966 q = qdisc_leaf(p, clid);
967 } else { /* ingress */
968 q = dev->rx_queue.qdisc_sleeping;
970 } else {
971 q = dev->qdisc;
973 if (!q)
974 return -ENOENT;
976 if (tcm->tcm_handle && q->handle != tcm->tcm_handle)
977 return -EINVAL;
978 } else {
979 if ((q = qdisc_lookup(dev, tcm->tcm_handle)) == NULL)
980 return -ENOENT;
983 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
984 return -EINVAL;
986 if (n->nlmsg_type == RTM_DELQDISC) {
987 if (!clid)
988 return -EINVAL;
989 if (q->handle == 0)
990 return -ENOENT;
991 if ((err = qdisc_graft(dev, p, skb, n, clid, NULL, q)) != 0)
992 return err;
993 } else {
994 qdisc_notify(skb, n, clid, NULL, q);
996 return 0;
1000 Create/change qdisc.
1003 static int tc_modify_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
1005 struct net *net = sock_net(skb->sk);
1006 struct tcmsg *tcm;
1007 struct nlattr *tca[TCA_MAX + 1];
1008 struct net_device *dev;
1009 u32 clid;
1010 struct Qdisc *q, *p;
1011 int err;
1013 if (!net_eq(net, &init_net))
1014 return -EINVAL;
1016 replay:
1017 /* Reinit, just in case something touches this. */
1018 tcm = NLMSG_DATA(n);
1019 clid = tcm->tcm_parent;
1020 q = p = NULL;
1022 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
1023 return -ENODEV;
1025 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1026 if (err < 0)
1027 return err;
1029 if (clid) {
1030 if (clid != TC_H_ROOT) {
1031 if (clid != TC_H_INGRESS) {
1032 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
1033 return -ENOENT;
1034 q = qdisc_leaf(p, clid);
1035 } else { /*ingress */
1036 q = dev->rx_queue.qdisc_sleeping;
1038 } else {
1039 q = dev->qdisc;
1042 /* It may be default qdisc, ignore it */
1043 if (q && q->handle == 0)
1044 q = NULL;
1046 if (!q || !tcm->tcm_handle || q->handle != tcm->tcm_handle) {
1047 if (tcm->tcm_handle) {
1048 if (q && !(n->nlmsg_flags&NLM_F_REPLACE))
1049 return -EEXIST;
1050 if (TC_H_MIN(tcm->tcm_handle))
1051 return -EINVAL;
1052 if ((q = qdisc_lookup(dev, tcm->tcm_handle)) == NULL)
1053 goto create_n_graft;
1054 if (n->nlmsg_flags&NLM_F_EXCL)
1055 return -EEXIST;
1056 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
1057 return -EINVAL;
1058 if (q == p ||
1059 (p && check_loop(q, p, 0)))
1060 return -ELOOP;
1061 atomic_inc(&q->refcnt);
1062 goto graft;
1063 } else {
1064 if (q == NULL)
1065 goto create_n_graft;
1067 /* This magic test requires explanation.
1069 * We know, that some child q is already
1070 * attached to this parent and have choice:
1071 * either to change it or to create/graft new one.
1073 * 1. We are allowed to create/graft only
1074 * if CREATE and REPLACE flags are set.
1076 * 2. If EXCL is set, requestor wanted to say,
1077 * that qdisc tcm_handle is not expected
1078 * to exist, so that we choose create/graft too.
1080 * 3. The last case is when no flags are set.
1081 * Alas, it is sort of hole in API, we
1082 * cannot decide what to do unambiguously.
1083 * For now we select create/graft, if
1084 * user gave KIND, which does not match existing.
1086 if ((n->nlmsg_flags&NLM_F_CREATE) &&
1087 (n->nlmsg_flags&NLM_F_REPLACE) &&
1088 ((n->nlmsg_flags&NLM_F_EXCL) ||
1089 (tca[TCA_KIND] &&
1090 nla_strcmp(tca[TCA_KIND], q->ops->id))))
1091 goto create_n_graft;
1094 } else {
1095 if (!tcm->tcm_handle)
1096 return -EINVAL;
1097 q = qdisc_lookup(dev, tcm->tcm_handle);
1100 /* Change qdisc parameters */
1101 if (q == NULL)
1102 return -ENOENT;
1103 if (n->nlmsg_flags&NLM_F_EXCL)
1104 return -EEXIST;
1105 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
1106 return -EINVAL;
1107 err = qdisc_change(q, tca);
1108 if (err == 0)
1109 qdisc_notify(skb, n, clid, NULL, q);
1110 return err;
1112 create_n_graft:
1113 if (!(n->nlmsg_flags&NLM_F_CREATE))
1114 return -ENOENT;
1115 if (clid == TC_H_INGRESS)
1116 q = qdisc_create(dev, &dev->rx_queue, p,
1117 tcm->tcm_parent, tcm->tcm_parent,
1118 tca, &err);
1119 else {
1120 struct netdev_queue *dev_queue;
1122 if (p && p->ops->cl_ops && p->ops->cl_ops->select_queue)
1123 dev_queue = p->ops->cl_ops->select_queue(p, tcm);
1124 else if (p)
1125 dev_queue = p->dev_queue;
1126 else
1127 dev_queue = netdev_get_tx_queue(dev, 0);
1129 q = qdisc_create(dev, dev_queue, p,
1130 tcm->tcm_parent, tcm->tcm_handle,
1131 tca, &err);
1133 if (q == NULL) {
1134 if (err == -EAGAIN)
1135 goto replay;
1136 return err;
1139 graft:
1140 err = qdisc_graft(dev, p, skb, n, clid, q, NULL);
1141 if (err) {
1142 if (q)
1143 qdisc_destroy(q);
1144 return err;
1147 return 0;
1150 static int tc_fill_qdisc(struct sk_buff *skb, struct Qdisc *q, u32 clid,
1151 u32 pid, u32 seq, u16 flags, int event)
1153 struct tcmsg *tcm;
1154 struct nlmsghdr *nlh;
1155 unsigned char *b = skb_tail_pointer(skb);
1156 struct gnet_dump d;
1158 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
1159 tcm = NLMSG_DATA(nlh);
1160 tcm->tcm_family = AF_UNSPEC;
1161 tcm->tcm__pad1 = 0;
1162 tcm->tcm__pad2 = 0;
1163 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
1164 tcm->tcm_parent = clid;
1165 tcm->tcm_handle = q->handle;
1166 tcm->tcm_info = atomic_read(&q->refcnt);
1167 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
1168 if (q->ops->dump && q->ops->dump(q, skb) < 0)
1169 goto nla_put_failure;
1170 q->qstats.qlen = q->q.qlen;
1172 if (q->stab && qdisc_dump_stab(skb, q->stab) < 0)
1173 goto nla_put_failure;
1175 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
1176 qdisc_root_sleeping_lock(q), &d) < 0)
1177 goto nla_put_failure;
1179 if (q->ops->dump_stats && q->ops->dump_stats(q, &d) < 0)
1180 goto nla_put_failure;
1182 if (gnet_stats_copy_basic(&d, &q->bstats) < 0 ||
1183 gnet_stats_copy_rate_est(&d, &q->bstats, &q->rate_est) < 0 ||
1184 gnet_stats_copy_queue(&d, &q->qstats) < 0)
1185 goto nla_put_failure;
1187 if (gnet_stats_finish_copy(&d) < 0)
1188 goto nla_put_failure;
1190 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
1191 return skb->len;
1193 nlmsg_failure:
1194 nla_put_failure:
1195 nlmsg_trim(skb, b);
1196 return -1;
1199 static int qdisc_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1200 u32 clid, struct Qdisc *old, struct Qdisc *new)
1202 struct sk_buff *skb;
1203 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1205 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1206 if (!skb)
1207 return -ENOBUFS;
1209 if (old && old->handle) {
1210 if (tc_fill_qdisc(skb, old, clid, pid, n->nlmsg_seq, 0, RTM_DELQDISC) < 0)
1211 goto err_out;
1213 if (new) {
1214 if (tc_fill_qdisc(skb, new, clid, pid, n->nlmsg_seq, old ? NLM_F_REPLACE : 0, RTM_NEWQDISC) < 0)
1215 goto err_out;
1218 if (skb->len)
1219 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
1221 err_out:
1222 kfree_skb(skb);
1223 return -EINVAL;
1226 static bool tc_qdisc_dump_ignore(struct Qdisc *q)
1228 return (q->flags & TCQ_F_BUILTIN) ? true : false;
1231 static int tc_dump_qdisc_root(struct Qdisc *root, struct sk_buff *skb,
1232 struct netlink_callback *cb,
1233 int *q_idx_p, int s_q_idx)
1235 int ret = 0, q_idx = *q_idx_p;
1236 struct Qdisc *q;
1238 if (!root)
1239 return 0;
1241 q = root;
1242 if (q_idx < s_q_idx) {
1243 q_idx++;
1244 } else {
1245 if (!tc_qdisc_dump_ignore(q) &&
1246 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1247 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1248 goto done;
1249 q_idx++;
1251 list_for_each_entry(q, &root->list, list) {
1252 if (q_idx < s_q_idx) {
1253 q_idx++;
1254 continue;
1256 if (!tc_qdisc_dump_ignore(q) &&
1257 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1258 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1259 goto done;
1260 q_idx++;
1263 out:
1264 *q_idx_p = q_idx;
1265 return ret;
1266 done:
1267 ret = -1;
1268 goto out;
1271 static int tc_dump_qdisc(struct sk_buff *skb, struct netlink_callback *cb)
1273 struct net *net = sock_net(skb->sk);
1274 int idx, q_idx;
1275 int s_idx, s_q_idx;
1276 struct net_device *dev;
1278 if (!net_eq(net, &init_net))
1279 return 0;
1281 s_idx = cb->args[0];
1282 s_q_idx = q_idx = cb->args[1];
1284 rcu_read_lock();
1285 idx = 0;
1286 for_each_netdev_rcu(&init_net, dev) {
1287 struct netdev_queue *dev_queue;
1289 if (idx < s_idx)
1290 goto cont;
1291 if (idx > s_idx)
1292 s_q_idx = 0;
1293 q_idx = 0;
1295 if (tc_dump_qdisc_root(dev->qdisc, skb, cb, &q_idx, s_q_idx) < 0)
1296 goto done;
1298 dev_queue = &dev->rx_queue;
1299 if (tc_dump_qdisc_root(dev_queue->qdisc_sleeping, skb, cb, &q_idx, s_q_idx) < 0)
1300 goto done;
1302 cont:
1303 idx++;
1306 done:
1307 rcu_read_unlock();
1309 cb->args[0] = idx;
1310 cb->args[1] = q_idx;
1312 return skb->len;
1317 /************************************************
1318 * Traffic classes manipulation. *
1319 ************************************************/
1323 static int tc_ctl_tclass(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
1325 struct net *net = sock_net(skb->sk);
1326 struct tcmsg *tcm = NLMSG_DATA(n);
1327 struct nlattr *tca[TCA_MAX + 1];
1328 struct net_device *dev;
1329 struct Qdisc *q = NULL;
1330 const struct Qdisc_class_ops *cops;
1331 unsigned long cl = 0;
1332 unsigned long new_cl;
1333 u32 pid = tcm->tcm_parent;
1334 u32 clid = tcm->tcm_handle;
1335 u32 qid = TC_H_MAJ(clid);
1336 int err;
1338 if (!net_eq(net, &init_net))
1339 return -EINVAL;
1341 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
1342 return -ENODEV;
1344 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1345 if (err < 0)
1346 return err;
1349 parent == TC_H_UNSPEC - unspecified parent.
1350 parent == TC_H_ROOT - class is root, which has no parent.
1351 parent == X:0 - parent is root class.
1352 parent == X:Y - parent is a node in hierarchy.
1353 parent == 0:Y - parent is X:Y, where X:0 is qdisc.
1355 handle == 0:0 - generate handle from kernel pool.
1356 handle == 0:Y - class is X:Y, where X:0 is qdisc.
1357 handle == X:Y - clear.
1358 handle == X:0 - root class.
1361 /* Step 1. Determine qdisc handle X:0 */
1363 if (pid != TC_H_ROOT) {
1364 u32 qid1 = TC_H_MAJ(pid);
1366 if (qid && qid1) {
1367 /* If both majors are known, they must be identical. */
1368 if (qid != qid1)
1369 return -EINVAL;
1370 } else if (qid1) {
1371 qid = qid1;
1372 } else if (qid == 0)
1373 qid = dev->qdisc->handle;
1375 /* Now qid is genuine qdisc handle consistent
1376 both with parent and child.
1378 TC_H_MAJ(pid) still may be unspecified, complete it now.
1380 if (pid)
1381 pid = TC_H_MAKE(qid, pid);
1382 } else {
1383 if (qid == 0)
1384 qid = dev->qdisc->handle;
1387 /* OK. Locate qdisc */
1388 if ((q = qdisc_lookup(dev, qid)) == NULL)
1389 return -ENOENT;
1391 /* An check that it supports classes */
1392 cops = q->ops->cl_ops;
1393 if (cops == NULL)
1394 return -EINVAL;
1396 /* Now try to get class */
1397 if (clid == 0) {
1398 if (pid == TC_H_ROOT)
1399 clid = qid;
1400 } else
1401 clid = TC_H_MAKE(qid, clid);
1403 if (clid)
1404 cl = cops->get(q, clid);
1406 if (cl == 0) {
1407 err = -ENOENT;
1408 if (n->nlmsg_type != RTM_NEWTCLASS || !(n->nlmsg_flags&NLM_F_CREATE))
1409 goto out;
1410 } else {
1411 switch (n->nlmsg_type) {
1412 case RTM_NEWTCLASS:
1413 err = -EEXIST;
1414 if (n->nlmsg_flags&NLM_F_EXCL)
1415 goto out;
1416 break;
1417 case RTM_DELTCLASS:
1418 err = -EOPNOTSUPP;
1419 if (cops->delete)
1420 err = cops->delete(q, cl);
1421 if (err == 0)
1422 tclass_notify(skb, n, q, cl, RTM_DELTCLASS);
1423 goto out;
1424 case RTM_GETTCLASS:
1425 err = tclass_notify(skb, n, q, cl, RTM_NEWTCLASS);
1426 goto out;
1427 default:
1428 err = -EINVAL;
1429 goto out;
1433 new_cl = cl;
1434 err = -EOPNOTSUPP;
1435 if (cops->change)
1436 err = cops->change(q, clid, pid, tca, &new_cl);
1437 if (err == 0)
1438 tclass_notify(skb, n, q, new_cl, RTM_NEWTCLASS);
1440 out:
1441 if (cl)
1442 cops->put(q, cl);
1444 return err;
1448 static int tc_fill_tclass(struct sk_buff *skb, struct Qdisc *q,
1449 unsigned long cl,
1450 u32 pid, u32 seq, u16 flags, int event)
1452 struct tcmsg *tcm;
1453 struct nlmsghdr *nlh;
1454 unsigned char *b = skb_tail_pointer(skb);
1455 struct gnet_dump d;
1456 const struct Qdisc_class_ops *cl_ops = q->ops->cl_ops;
1458 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
1459 tcm = NLMSG_DATA(nlh);
1460 tcm->tcm_family = AF_UNSPEC;
1461 tcm->tcm__pad1 = 0;
1462 tcm->tcm__pad2 = 0;
1463 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
1464 tcm->tcm_parent = q->handle;
1465 tcm->tcm_handle = q->handle;
1466 tcm->tcm_info = 0;
1467 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
1468 if (cl_ops->dump && cl_ops->dump(q, cl, skb, tcm) < 0)
1469 goto nla_put_failure;
1471 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
1472 qdisc_root_sleeping_lock(q), &d) < 0)
1473 goto nla_put_failure;
1475 if (cl_ops->dump_stats && cl_ops->dump_stats(q, cl, &d) < 0)
1476 goto nla_put_failure;
1478 if (gnet_stats_finish_copy(&d) < 0)
1479 goto nla_put_failure;
1481 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
1482 return skb->len;
1484 nlmsg_failure:
1485 nla_put_failure:
1486 nlmsg_trim(skb, b);
1487 return -1;
1490 static int tclass_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1491 struct Qdisc *q, unsigned long cl, int event)
1493 struct sk_buff *skb;
1494 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1496 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1497 if (!skb)
1498 return -ENOBUFS;
1500 if (tc_fill_tclass(skb, q, cl, pid, n->nlmsg_seq, 0, event) < 0) {
1501 kfree_skb(skb);
1502 return -EINVAL;
1505 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
1508 struct qdisc_dump_args
1510 struct qdisc_walker w;
1511 struct sk_buff *skb;
1512 struct netlink_callback *cb;
1515 static int qdisc_class_dump(struct Qdisc *q, unsigned long cl, struct qdisc_walker *arg)
1517 struct qdisc_dump_args *a = (struct qdisc_dump_args *)arg;
1519 return tc_fill_tclass(a->skb, q, cl, NETLINK_CB(a->cb->skb).pid,
1520 a->cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWTCLASS);
1523 static int tc_dump_tclass_qdisc(struct Qdisc *q, struct sk_buff *skb,
1524 struct tcmsg *tcm, struct netlink_callback *cb,
1525 int *t_p, int s_t)
1527 struct qdisc_dump_args arg;
1529 if (tc_qdisc_dump_ignore(q) ||
1530 *t_p < s_t || !q->ops->cl_ops ||
1531 (tcm->tcm_parent &&
1532 TC_H_MAJ(tcm->tcm_parent) != q->handle)) {
1533 (*t_p)++;
1534 return 0;
1536 if (*t_p > s_t)
1537 memset(&cb->args[1], 0, sizeof(cb->args)-sizeof(cb->args[0]));
1538 arg.w.fn = qdisc_class_dump;
1539 arg.skb = skb;
1540 arg.cb = cb;
1541 arg.w.stop = 0;
1542 arg.w.skip = cb->args[1];
1543 arg.w.count = 0;
1544 q->ops->cl_ops->walk(q, &arg.w);
1545 cb->args[1] = arg.w.count;
1546 if (arg.w.stop)
1547 return -1;
1548 (*t_p)++;
1549 return 0;
1552 static int tc_dump_tclass_root(struct Qdisc *root, struct sk_buff *skb,
1553 struct tcmsg *tcm, struct netlink_callback *cb,
1554 int *t_p, int s_t)
1556 struct Qdisc *q;
1558 if (!root)
1559 return 0;
1561 if (tc_dump_tclass_qdisc(root, skb, tcm, cb, t_p, s_t) < 0)
1562 return -1;
1564 list_for_each_entry(q, &root->list, list) {
1565 if (tc_dump_tclass_qdisc(q, skb, tcm, cb, t_p, s_t) < 0)
1566 return -1;
1569 return 0;
1572 static int tc_dump_tclass(struct sk_buff *skb, struct netlink_callback *cb)
1574 struct tcmsg *tcm = (struct tcmsg*)NLMSG_DATA(cb->nlh);
1575 struct net *net = sock_net(skb->sk);
1576 struct netdev_queue *dev_queue;
1577 struct net_device *dev;
1578 int t, s_t;
1580 if (!net_eq(net, &init_net))
1581 return 0;
1583 if (cb->nlh->nlmsg_len < NLMSG_LENGTH(sizeof(*tcm)))
1584 return 0;
1585 if ((dev = dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
1586 return 0;
1588 s_t = cb->args[0];
1589 t = 0;
1591 if (tc_dump_tclass_root(dev->qdisc, skb, tcm, cb, &t, s_t) < 0)
1592 goto done;
1594 dev_queue = &dev->rx_queue;
1595 if (tc_dump_tclass_root(dev_queue->qdisc_sleeping, skb, tcm, cb, &t, s_t) < 0)
1596 goto done;
1598 done:
1599 cb->args[0] = t;
1601 dev_put(dev);
1602 return skb->len;
1605 /* Main classifier routine: scans classifier chain attached
1606 to this qdisc, (optionally) tests for protocol and asks
1607 specific classifiers.
1609 int tc_classify_compat(struct sk_buff *skb, struct tcf_proto *tp,
1610 struct tcf_result *res)
1612 __be16 protocol = skb->protocol;
1613 int err = 0;
1615 for (; tp; tp = tp->next) {
1616 if ((tp->protocol == protocol ||
1617 tp->protocol == htons(ETH_P_ALL)) &&
1618 (err = tp->classify(skb, tp, res)) >= 0) {
1619 #ifdef CONFIG_NET_CLS_ACT
1620 if (err != TC_ACT_RECLASSIFY && skb->tc_verd)
1621 skb->tc_verd = SET_TC_VERD(skb->tc_verd, 0);
1622 #endif
1623 return err;
1626 return -1;
1628 EXPORT_SYMBOL(tc_classify_compat);
1630 int tc_classify(struct sk_buff *skb, struct tcf_proto *tp,
1631 struct tcf_result *res)
1633 int err = 0;
1634 __be16 protocol;
1635 #ifdef CONFIG_NET_CLS_ACT
1636 struct tcf_proto *otp = tp;
1637 reclassify:
1638 #endif
1639 protocol = skb->protocol;
1641 err = tc_classify_compat(skb, tp, res);
1642 #ifdef CONFIG_NET_CLS_ACT
1643 if (err == TC_ACT_RECLASSIFY) {
1644 u32 verd = G_TC_VERD(skb->tc_verd);
1645 tp = otp;
1647 if (verd++ >= MAX_REC_LOOP) {
1648 printk("rule prio %u protocol %02x reclassify loop, "
1649 "packet dropped\n",
1650 tp->prio&0xffff, ntohs(tp->protocol));
1651 return TC_ACT_SHOT;
1653 skb->tc_verd = SET_TC_VERD(skb->tc_verd, verd);
1654 goto reclassify;
1656 #endif
1657 return err;
1659 EXPORT_SYMBOL(tc_classify);
1661 void tcf_destroy(struct tcf_proto *tp)
1663 tp->ops->destroy(tp);
1664 module_put(tp->ops->owner);
1665 kfree(tp);
1668 void tcf_destroy_chain(struct tcf_proto **fl)
1670 struct tcf_proto *tp;
1672 while ((tp = *fl) != NULL) {
1673 *fl = tp->next;
1674 tcf_destroy(tp);
1677 EXPORT_SYMBOL(tcf_destroy_chain);
1679 #ifdef CONFIG_PROC_FS
1680 static int psched_show(struct seq_file *seq, void *v)
1682 struct timespec ts;
1684 hrtimer_get_res(CLOCK_MONOTONIC, &ts);
1685 seq_printf(seq, "%08x %08x %08x %08x\n",
1686 (u32)NSEC_PER_USEC, (u32)PSCHED_TICKS2NS(1),
1687 1000000,
1688 (u32)NSEC_PER_SEC/(u32)ktime_to_ns(timespec_to_ktime(ts)));
1690 return 0;
1693 static int psched_open(struct inode *inode, struct file *file)
1695 return single_open(file, psched_show, PDE(inode)->data);
1698 static const struct file_operations psched_fops = {
1699 .owner = THIS_MODULE,
1700 .open = psched_open,
1701 .read = seq_read,
1702 .llseek = seq_lseek,
1703 .release = single_release,
1705 #endif
1707 static int __init pktsched_init(void)
1709 register_qdisc(&pfifo_qdisc_ops);
1710 register_qdisc(&bfifo_qdisc_ops);
1711 register_qdisc(&pfifo_head_drop_qdisc_ops);
1712 register_qdisc(&mq_qdisc_ops);
1713 proc_net_fops_create(&init_net, "psched", 0, &psched_fops);
1715 rtnl_register(PF_UNSPEC, RTM_NEWQDISC, tc_modify_qdisc, NULL);
1716 rtnl_register(PF_UNSPEC, RTM_DELQDISC, tc_get_qdisc, NULL);
1717 rtnl_register(PF_UNSPEC, RTM_GETQDISC, tc_get_qdisc, tc_dump_qdisc);
1718 rtnl_register(PF_UNSPEC, RTM_NEWTCLASS, tc_ctl_tclass, NULL);
1719 rtnl_register(PF_UNSPEC, RTM_DELTCLASS, tc_ctl_tclass, NULL);
1720 rtnl_register(PF_UNSPEC, RTM_GETTCLASS, tc_ctl_tclass, tc_dump_tclass);
1722 return 0;
1725 subsys_initcall(pktsched_init);