]> asedeno.scripts.mit.edu Git - linux.git/blob - net/sched/sch_api.c
Merge branch 'xtensa-sim-params' into xtensa-fixes
[linux.git] / net / sched / sch_api.c
1 /*
2  * net/sched/sch_api.c  Packet scheduler API.
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:     Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
10  *
11  * Fixes:
12  *
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
16  */
17
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>
32 #include <linux/hashtable.h>
33
34 #include <net/net_namespace.h>
35 #include <net/sock.h>
36 #include <net/netlink.h>
37 #include <net/pkt_sched.h>
38
39 static int qdisc_notify(struct net *net, struct sk_buff *oskb,
40                         struct nlmsghdr *n, u32 clid,
41                         struct Qdisc *old, struct Qdisc *new);
42 static int tclass_notify(struct net *net, struct sk_buff *oskb,
43                          struct nlmsghdr *n, struct Qdisc *q,
44                          unsigned long cl, int event);
45
46 /*
47
48    Short review.
49    -------------
50
51    This file consists of two interrelated parts:
52
53    1. queueing disciplines manager frontend.
54    2. traffic classes manager frontend.
55
56    Generally, queueing discipline ("qdisc") is a black box,
57    which is able to enqueue packets and to dequeue them (when
58    device is ready to send something) in order and at times
59    determined by algorithm hidden in it.
60
61    qdisc's are divided to two categories:
62    - "queues", which have no internal structure visible from outside.
63    - "schedulers", which split all the packets to "traffic classes",
64      using "packet classifiers" (look at cls_api.c)
65
66    In turn, classes may have child qdiscs (as rule, queues)
67    attached to them etc. etc. etc.
68
69    The goal of the routines in this file is to translate
70    information supplied by user in the form of handles
71    to more intelligible for kernel form, to make some sanity
72    checks and part of work, which is common to all qdiscs
73    and to provide rtnetlink notifications.
74
75    All real intelligent work is done inside qdisc modules.
76
77
78
79    Every discipline has two major routines: enqueue and dequeue.
80
81    ---dequeue
82
83    dequeue usually returns a skb to send. It is allowed to return NULL,
84    but it does not mean that queue is empty, it just means that
85    discipline does not want to send anything this time.
86    Queue is really empty if q->q.qlen == 0.
87    For complicated disciplines with multiple queues q->q is not
88    real packet queue, but however q->q.qlen must be valid.
89
90    ---enqueue
91
92    enqueue returns 0, if packet was enqueued successfully.
93    If packet (this one or another one) was dropped, it returns
94    not zero error code.
95    NET_XMIT_DROP        - this packet dropped
96      Expected action: do not backoff, but wait until queue will clear.
97    NET_XMIT_CN          - probably this packet enqueued, but another one dropped.
98      Expected action: backoff or ignore
99
100    Auxiliary routines:
101
102    ---peek
103
104    like dequeue but without removing a packet from the queue
105
106    ---reset
107
108    returns qdisc to initial state: purge all buffers, clear all
109    timers, counters (except for statistics) etc.
110
111    ---init
112
113    initializes newly created qdisc.
114
115    ---destroy
116
117    destroys resources allocated by init and during lifetime of qdisc.
118
119    ---change
120
121    changes qdisc parameters.
122  */
123
124 /* Protects list of registered TC modules. It is pure SMP lock. */
125 static DEFINE_RWLOCK(qdisc_mod_lock);
126
127
128 /************************************************
129  *      Queueing disciplines manipulation.      *
130  ************************************************/
131
132
133 /* The list of all installed queueing disciplines. */
134
135 static struct Qdisc_ops *qdisc_base;
136
137 /* Register/unregister queueing discipline */
138
139 int register_qdisc(struct Qdisc_ops *qops)
140 {
141         struct Qdisc_ops *q, **qp;
142         int rc = -EEXIST;
143
144         write_lock(&qdisc_mod_lock);
145         for (qp = &qdisc_base; (q = *qp) != NULL; qp = &q->next)
146                 if (!strcmp(qops->id, q->id))
147                         goto out;
148
149         if (qops->enqueue == NULL)
150                 qops->enqueue = noop_qdisc_ops.enqueue;
151         if (qops->peek == NULL) {
152                 if (qops->dequeue == NULL)
153                         qops->peek = noop_qdisc_ops.peek;
154                 else
155                         goto out_einval;
156         }
157         if (qops->dequeue == NULL)
158                 qops->dequeue = noop_qdisc_ops.dequeue;
159
160         if (qops->cl_ops) {
161                 const struct Qdisc_class_ops *cops = qops->cl_ops;
162
163                 if (!(cops->get && cops->put && cops->walk && cops->leaf))
164                         goto out_einval;
165
166                 if (cops->tcf_chain && !(cops->bind_tcf && cops->unbind_tcf))
167                         goto out_einval;
168         }
169
170         qops->next = NULL;
171         *qp = qops;
172         rc = 0;
173 out:
174         write_unlock(&qdisc_mod_lock);
175         return rc;
176
177 out_einval:
178         rc = -EINVAL;
179         goto out;
180 }
181 EXPORT_SYMBOL(register_qdisc);
182
183 int unregister_qdisc(struct Qdisc_ops *qops)
184 {
185         struct Qdisc_ops *q, **qp;
186         int err = -ENOENT;
187
188         write_lock(&qdisc_mod_lock);
189         for (qp = &qdisc_base; (q = *qp) != NULL; qp = &q->next)
190                 if (q == qops)
191                         break;
192         if (q) {
193                 *qp = q->next;
194                 q->next = NULL;
195                 err = 0;
196         }
197         write_unlock(&qdisc_mod_lock);
198         return err;
199 }
200 EXPORT_SYMBOL(unregister_qdisc);
201
202 /* Get default qdisc if not otherwise specified */
203 void qdisc_get_default(char *name, size_t len)
204 {
205         read_lock(&qdisc_mod_lock);
206         strlcpy(name, default_qdisc_ops->id, len);
207         read_unlock(&qdisc_mod_lock);
208 }
209
210 static struct Qdisc_ops *qdisc_lookup_default(const char *name)
211 {
212         struct Qdisc_ops *q = NULL;
213
214         for (q = qdisc_base; q; q = q->next) {
215                 if (!strcmp(name, q->id)) {
216                         if (!try_module_get(q->owner))
217                                 q = NULL;
218                         break;
219                 }
220         }
221
222         return q;
223 }
224
225 /* Set new default qdisc to use */
226 int qdisc_set_default(const char *name)
227 {
228         const struct Qdisc_ops *ops;
229
230         if (!capable(CAP_NET_ADMIN))
231                 return -EPERM;
232
233         write_lock(&qdisc_mod_lock);
234         ops = qdisc_lookup_default(name);
235         if (!ops) {
236                 /* Not found, drop lock and try to load module */
237                 write_unlock(&qdisc_mod_lock);
238                 request_module("sch_%s", name);
239                 write_lock(&qdisc_mod_lock);
240
241                 ops = qdisc_lookup_default(name);
242         }
243
244         if (ops) {
245                 /* Set new default */
246                 module_put(default_qdisc_ops->owner);
247                 default_qdisc_ops = ops;
248         }
249         write_unlock(&qdisc_mod_lock);
250
251         return ops ? 0 : -ENOENT;
252 }
253
254 /* We know handle. Find qdisc among all qdisc's attached to device
255  * (root qdisc, all its children, children of children etc.)
256  * Note: caller either uses rtnl or rcu_read_lock()
257  */
258
259 static struct Qdisc *qdisc_match_from_root(struct Qdisc *root, u32 handle)
260 {
261         struct Qdisc *q;
262
263         if (!qdisc_dev(root))
264                 return (root->handle == handle ? root : NULL);
265
266         if (!(root->flags & TCQ_F_BUILTIN) &&
267             root->handle == handle)
268                 return root;
269
270         hash_for_each_possible_rcu(qdisc_dev(root)->qdisc_hash, q, hash, handle) {
271                 if (q->handle == handle)
272                         return q;
273         }
274         return NULL;
275 }
276
277 void qdisc_hash_add(struct Qdisc *q)
278 {
279         if ((q->parent != TC_H_ROOT) && !(q->flags & TCQ_F_INGRESS)) {
280                 struct Qdisc *root = qdisc_dev(q)->qdisc;
281
282                 WARN_ON_ONCE(root == &noop_qdisc);
283                 ASSERT_RTNL();
284                 hash_add_rcu(qdisc_dev(q)->qdisc_hash, &q->hash, q->handle);
285         }
286 }
287 EXPORT_SYMBOL(qdisc_hash_add);
288
289 void qdisc_hash_del(struct Qdisc *q)
290 {
291         if ((q->parent != TC_H_ROOT) && !(q->flags & TCQ_F_INGRESS)) {
292                 ASSERT_RTNL();
293                 hash_del_rcu(&q->hash);
294         }
295 }
296 EXPORT_SYMBOL(qdisc_hash_del);
297
298 struct Qdisc *qdisc_lookup(struct net_device *dev, u32 handle)
299 {
300         struct Qdisc *q;
301
302         q = qdisc_match_from_root(dev->qdisc, handle);
303         if (q)
304                 goto out;
305
306         if (dev_ingress_queue(dev))
307                 q = qdisc_match_from_root(
308                         dev_ingress_queue(dev)->qdisc_sleeping,
309                         handle);
310 out:
311         return q;
312 }
313
314 static struct Qdisc *qdisc_leaf(struct Qdisc *p, u32 classid)
315 {
316         unsigned long cl;
317         struct Qdisc *leaf;
318         const struct Qdisc_class_ops *cops = p->ops->cl_ops;
319
320         if (cops == NULL)
321                 return NULL;
322         cl = cops->get(p, classid);
323
324         if (cl == 0)
325                 return NULL;
326         leaf = cops->leaf(p, cl);
327         cops->put(p, cl);
328         return leaf;
329 }
330
331 /* Find queueing discipline by name */
332
333 static struct Qdisc_ops *qdisc_lookup_ops(struct nlattr *kind)
334 {
335         struct Qdisc_ops *q = NULL;
336
337         if (kind) {
338                 read_lock(&qdisc_mod_lock);
339                 for (q = qdisc_base; q; q = q->next) {
340                         if (nla_strcmp(kind, q->id) == 0) {
341                                 if (!try_module_get(q->owner))
342                                         q = NULL;
343                                 break;
344                         }
345                 }
346                 read_unlock(&qdisc_mod_lock);
347         }
348         return q;
349 }
350
351 /* The linklayer setting were not transferred from iproute2, in older
352  * versions, and the rate tables lookup systems have been dropped in
353  * the kernel. To keep backward compatible with older iproute2 tc
354  * utils, we detect the linklayer setting by detecting if the rate
355  * table were modified.
356  *
357  * For linklayer ATM table entries, the rate table will be aligned to
358  * 48 bytes, thus some table entries will contain the same value.  The
359  * mpu (min packet unit) is also encoded into the old rate table, thus
360  * starting from the mpu, we find low and high table entries for
361  * mapping this cell.  If these entries contain the same value, when
362  * the rate tables have been modified for linklayer ATM.
363  *
364  * This is done by rounding mpu to the nearest 48 bytes cell/entry,
365  * and then roundup to the next cell, calc the table entry one below,
366  * and compare.
367  */
368 static __u8 __detect_linklayer(struct tc_ratespec *r, __u32 *rtab)
369 {
370         int low       = roundup(r->mpu, 48);
371         int high      = roundup(low+1, 48);
372         int cell_low  = low >> r->cell_log;
373         int cell_high = (high >> r->cell_log) - 1;
374
375         /* rtab is too inaccurate at rates > 100Mbit/s */
376         if ((r->rate > (100000000/8)) || (rtab[0] == 0)) {
377                 pr_debug("TC linklayer: Giving up ATM detection\n");
378                 return TC_LINKLAYER_ETHERNET;
379         }
380
381         if ((cell_high > cell_low) && (cell_high < 256)
382             && (rtab[cell_low] == rtab[cell_high])) {
383                 pr_debug("TC linklayer: Detected ATM, low(%d)=high(%d)=%u\n",
384                          cell_low, cell_high, rtab[cell_high]);
385                 return TC_LINKLAYER_ATM;
386         }
387         return TC_LINKLAYER_ETHERNET;
388 }
389
390 static struct qdisc_rate_table *qdisc_rtab_list;
391
392 struct qdisc_rate_table *qdisc_get_rtab(struct tc_ratespec *r,
393                                         struct nlattr *tab)
394 {
395         struct qdisc_rate_table *rtab;
396
397         if (tab == NULL || r->rate == 0 || r->cell_log == 0 ||
398             nla_len(tab) != TC_RTAB_SIZE)
399                 return NULL;
400
401         for (rtab = qdisc_rtab_list; rtab; rtab = rtab->next) {
402                 if (!memcmp(&rtab->rate, r, sizeof(struct tc_ratespec)) &&
403                     !memcmp(&rtab->data, nla_data(tab), 1024)) {
404                         rtab->refcnt++;
405                         return rtab;
406                 }
407         }
408
409         rtab = kmalloc(sizeof(*rtab), GFP_KERNEL);
410         if (rtab) {
411                 rtab->rate = *r;
412                 rtab->refcnt = 1;
413                 memcpy(rtab->data, nla_data(tab), 1024);
414                 if (r->linklayer == TC_LINKLAYER_UNAWARE)
415                         r->linklayer = __detect_linklayer(r, rtab->data);
416                 rtab->next = qdisc_rtab_list;
417                 qdisc_rtab_list = rtab;
418         }
419         return rtab;
420 }
421 EXPORT_SYMBOL(qdisc_get_rtab);
422
423 void qdisc_put_rtab(struct qdisc_rate_table *tab)
424 {
425         struct qdisc_rate_table *rtab, **rtabp;
426
427         if (!tab || --tab->refcnt)
428                 return;
429
430         for (rtabp = &qdisc_rtab_list;
431              (rtab = *rtabp) != NULL;
432              rtabp = &rtab->next) {
433                 if (rtab == tab) {
434                         *rtabp = rtab->next;
435                         kfree(rtab);
436                         return;
437                 }
438         }
439 }
440 EXPORT_SYMBOL(qdisc_put_rtab);
441
442 static LIST_HEAD(qdisc_stab_list);
443
444 static const struct nla_policy stab_policy[TCA_STAB_MAX + 1] = {
445         [TCA_STAB_BASE] = { .len = sizeof(struct tc_sizespec) },
446         [TCA_STAB_DATA] = { .type = NLA_BINARY },
447 };
448
449 static struct qdisc_size_table *qdisc_get_stab(struct nlattr *opt)
450 {
451         struct nlattr *tb[TCA_STAB_MAX + 1];
452         struct qdisc_size_table *stab;
453         struct tc_sizespec *s;
454         unsigned int tsize = 0;
455         u16 *tab = NULL;
456         int err;
457
458         err = nla_parse_nested(tb, TCA_STAB_MAX, opt, stab_policy);
459         if (err < 0)
460                 return ERR_PTR(err);
461         if (!tb[TCA_STAB_BASE])
462                 return ERR_PTR(-EINVAL);
463
464         s = nla_data(tb[TCA_STAB_BASE]);
465
466         if (s->tsize > 0) {
467                 if (!tb[TCA_STAB_DATA])
468                         return ERR_PTR(-EINVAL);
469                 tab = nla_data(tb[TCA_STAB_DATA]);
470                 tsize = nla_len(tb[TCA_STAB_DATA]) / sizeof(u16);
471         }
472
473         if (tsize != s->tsize || (!tab && tsize > 0))
474                 return ERR_PTR(-EINVAL);
475
476         list_for_each_entry(stab, &qdisc_stab_list, list) {
477                 if (memcmp(&stab->szopts, s, sizeof(*s)))
478                         continue;
479                 if (tsize > 0 && memcmp(stab->data, tab, tsize * sizeof(u16)))
480                         continue;
481                 stab->refcnt++;
482                 return stab;
483         }
484
485         stab = kmalloc(sizeof(*stab) + tsize * sizeof(u16), GFP_KERNEL);
486         if (!stab)
487                 return ERR_PTR(-ENOMEM);
488
489         stab->refcnt = 1;
490         stab->szopts = *s;
491         if (tsize > 0)
492                 memcpy(stab->data, tab, tsize * sizeof(u16));
493
494         list_add_tail(&stab->list, &qdisc_stab_list);
495
496         return stab;
497 }
498
499 static void stab_kfree_rcu(struct rcu_head *head)
500 {
501         kfree(container_of(head, struct qdisc_size_table, rcu));
502 }
503
504 void qdisc_put_stab(struct qdisc_size_table *tab)
505 {
506         if (!tab)
507                 return;
508
509         if (--tab->refcnt == 0) {
510                 list_del(&tab->list);
511                 call_rcu_bh(&tab->rcu, stab_kfree_rcu);
512         }
513 }
514 EXPORT_SYMBOL(qdisc_put_stab);
515
516 static int qdisc_dump_stab(struct sk_buff *skb, struct qdisc_size_table *stab)
517 {
518         struct nlattr *nest;
519
520         nest = nla_nest_start(skb, TCA_STAB);
521         if (nest == NULL)
522                 goto nla_put_failure;
523         if (nla_put(skb, TCA_STAB_BASE, sizeof(stab->szopts), &stab->szopts))
524                 goto nla_put_failure;
525         nla_nest_end(skb, nest);
526
527         return skb->len;
528
529 nla_put_failure:
530         return -1;
531 }
532
533 void __qdisc_calculate_pkt_len(struct sk_buff *skb,
534                                const struct qdisc_size_table *stab)
535 {
536         int pkt_len, slot;
537
538         pkt_len = skb->len + stab->szopts.overhead;
539         if (unlikely(!stab->szopts.tsize))
540                 goto out;
541
542         slot = pkt_len + stab->szopts.cell_align;
543         if (unlikely(slot < 0))
544                 slot = 0;
545
546         slot >>= stab->szopts.cell_log;
547         if (likely(slot < stab->szopts.tsize))
548                 pkt_len = stab->data[slot];
549         else
550                 pkt_len = stab->data[stab->szopts.tsize - 1] *
551                                 (slot / stab->szopts.tsize) +
552                                 stab->data[slot % stab->szopts.tsize];
553
554         pkt_len <<= stab->szopts.size_log;
555 out:
556         if (unlikely(pkt_len < 1))
557                 pkt_len = 1;
558         qdisc_skb_cb(skb)->pkt_len = pkt_len;
559 }
560 EXPORT_SYMBOL(__qdisc_calculate_pkt_len);
561
562 void qdisc_warn_nonwc(const char *txt, struct Qdisc *qdisc)
563 {
564         if (!(qdisc->flags & TCQ_F_WARN_NONWC)) {
565                 pr_warn("%s: %s qdisc %X: is non-work-conserving?\n",
566                         txt, qdisc->ops->id, qdisc->handle >> 16);
567                 qdisc->flags |= TCQ_F_WARN_NONWC;
568         }
569 }
570 EXPORT_SYMBOL(qdisc_warn_nonwc);
571
572 static enum hrtimer_restart qdisc_watchdog(struct hrtimer *timer)
573 {
574         struct qdisc_watchdog *wd = container_of(timer, struct qdisc_watchdog,
575                                                  timer);
576
577         rcu_read_lock();
578         __netif_schedule(qdisc_root(wd->qdisc));
579         rcu_read_unlock();
580
581         return HRTIMER_NORESTART;
582 }
583
584 void qdisc_watchdog_init(struct qdisc_watchdog *wd, struct Qdisc *qdisc)
585 {
586         hrtimer_init(&wd->timer, CLOCK_MONOTONIC, HRTIMER_MODE_ABS_PINNED);
587         wd->timer.function = qdisc_watchdog;
588         wd->qdisc = qdisc;
589 }
590 EXPORT_SYMBOL(qdisc_watchdog_init);
591
592 void qdisc_watchdog_schedule_ns(struct qdisc_watchdog *wd, u64 expires)
593 {
594         if (test_bit(__QDISC_STATE_DEACTIVATED,
595                      &qdisc_root_sleeping(wd->qdisc)->state))
596                 return;
597
598         if (wd->last_expires == expires)
599                 return;
600
601         wd->last_expires = expires;
602         hrtimer_start(&wd->timer,
603                       ns_to_ktime(expires),
604                       HRTIMER_MODE_ABS_PINNED);
605 }
606 EXPORT_SYMBOL(qdisc_watchdog_schedule_ns);
607
608 void qdisc_watchdog_cancel(struct qdisc_watchdog *wd)
609 {
610         hrtimer_cancel(&wd->timer);
611 }
612 EXPORT_SYMBOL(qdisc_watchdog_cancel);
613
614 static struct hlist_head *qdisc_class_hash_alloc(unsigned int n)
615 {
616         unsigned int size = n * sizeof(struct hlist_head), i;
617         struct hlist_head *h;
618
619         if (size <= PAGE_SIZE)
620                 h = kmalloc(size, GFP_KERNEL);
621         else
622                 h = (struct hlist_head *)
623                         __get_free_pages(GFP_KERNEL, get_order(size));
624
625         if (h != NULL) {
626                 for (i = 0; i < n; i++)
627                         INIT_HLIST_HEAD(&h[i]);
628         }
629         return h;
630 }
631
632 static void qdisc_class_hash_free(struct hlist_head *h, unsigned int n)
633 {
634         unsigned int size = n * sizeof(struct hlist_head);
635
636         if (size <= PAGE_SIZE)
637                 kfree(h);
638         else
639                 free_pages((unsigned long)h, get_order(size));
640 }
641
642 void qdisc_class_hash_grow(struct Qdisc *sch, struct Qdisc_class_hash *clhash)
643 {
644         struct Qdisc_class_common *cl;
645         struct hlist_node *next;
646         struct hlist_head *nhash, *ohash;
647         unsigned int nsize, nmask, osize;
648         unsigned int i, h;
649
650         /* Rehash when load factor exceeds 0.75 */
651         if (clhash->hashelems * 4 <= clhash->hashsize * 3)
652                 return;
653         nsize = clhash->hashsize * 2;
654         nmask = nsize - 1;
655         nhash = qdisc_class_hash_alloc(nsize);
656         if (nhash == NULL)
657                 return;
658
659         ohash = clhash->hash;
660         osize = clhash->hashsize;
661
662         sch_tree_lock(sch);
663         for (i = 0; i < osize; i++) {
664                 hlist_for_each_entry_safe(cl, next, &ohash[i], hnode) {
665                         h = qdisc_class_hash(cl->classid, nmask);
666                         hlist_add_head(&cl->hnode, &nhash[h]);
667                 }
668         }
669         clhash->hash     = nhash;
670         clhash->hashsize = nsize;
671         clhash->hashmask = nmask;
672         sch_tree_unlock(sch);
673
674         qdisc_class_hash_free(ohash, osize);
675 }
676 EXPORT_SYMBOL(qdisc_class_hash_grow);
677
678 int qdisc_class_hash_init(struct Qdisc_class_hash *clhash)
679 {
680         unsigned int size = 4;
681
682         clhash->hash = qdisc_class_hash_alloc(size);
683         if (clhash->hash == NULL)
684                 return -ENOMEM;
685         clhash->hashsize  = size;
686         clhash->hashmask  = size - 1;
687         clhash->hashelems = 0;
688         return 0;
689 }
690 EXPORT_SYMBOL(qdisc_class_hash_init);
691
692 void qdisc_class_hash_destroy(struct Qdisc_class_hash *clhash)
693 {
694         qdisc_class_hash_free(clhash->hash, clhash->hashsize);
695 }
696 EXPORT_SYMBOL(qdisc_class_hash_destroy);
697
698 void qdisc_class_hash_insert(struct Qdisc_class_hash *clhash,
699                              struct Qdisc_class_common *cl)
700 {
701         unsigned int h;
702
703         INIT_HLIST_NODE(&cl->hnode);
704         h = qdisc_class_hash(cl->classid, clhash->hashmask);
705         hlist_add_head(&cl->hnode, &clhash->hash[h]);
706         clhash->hashelems++;
707 }
708 EXPORT_SYMBOL(qdisc_class_hash_insert);
709
710 void qdisc_class_hash_remove(struct Qdisc_class_hash *clhash,
711                              struct Qdisc_class_common *cl)
712 {
713         hlist_del(&cl->hnode);
714         clhash->hashelems--;
715 }
716 EXPORT_SYMBOL(qdisc_class_hash_remove);
717
718 /* Allocate an unique handle from space managed by kernel
719  * Possible range is [8000-FFFF]:0000 (0x8000 values)
720  */
721 static u32 qdisc_alloc_handle(struct net_device *dev)
722 {
723         int i = 0x8000;
724         static u32 autohandle = TC_H_MAKE(0x80000000U, 0);
725
726         do {
727                 autohandle += TC_H_MAKE(0x10000U, 0);
728                 if (autohandle == TC_H_MAKE(TC_H_ROOT, 0))
729                         autohandle = TC_H_MAKE(0x80000000U, 0);
730                 if (!qdisc_lookup(dev, autohandle))
731                         return autohandle;
732                 cond_resched();
733         } while (--i > 0);
734
735         return 0;
736 }
737
738 void qdisc_tree_reduce_backlog(struct Qdisc *sch, unsigned int n,
739                                unsigned int len)
740 {
741         const struct Qdisc_class_ops *cops;
742         unsigned long cl;
743         u32 parentid;
744         int drops;
745
746         if (n == 0 && len == 0)
747                 return;
748         drops = max_t(int, n, 0);
749         rcu_read_lock();
750         while ((parentid = sch->parent)) {
751                 if (TC_H_MAJ(parentid) == TC_H_MAJ(TC_H_INGRESS))
752                         break;
753
754                 if (sch->flags & TCQ_F_NOPARENT)
755                         break;
756                 /* TODO: perform the search on a per txq basis */
757                 sch = qdisc_lookup(qdisc_dev(sch), TC_H_MAJ(parentid));
758                 if (sch == NULL) {
759                         WARN_ON_ONCE(parentid != TC_H_ROOT);
760                         break;
761                 }
762                 cops = sch->ops->cl_ops;
763                 if (cops->qlen_notify) {
764                         cl = cops->get(sch, parentid);
765                         cops->qlen_notify(sch, cl);
766                         cops->put(sch, cl);
767                 }
768                 sch->q.qlen -= n;
769                 sch->qstats.backlog -= len;
770                 __qdisc_qstats_drop(sch, drops);
771         }
772         rcu_read_unlock();
773 }
774 EXPORT_SYMBOL(qdisc_tree_reduce_backlog);
775
776 static void notify_and_destroy(struct net *net, struct sk_buff *skb,
777                                struct nlmsghdr *n, u32 clid,
778                                struct Qdisc *old, struct Qdisc *new)
779 {
780         if (new || old)
781                 qdisc_notify(net, skb, n, clid, old, new);
782
783         if (old)
784                 qdisc_destroy(old);
785 }
786
787 /* Graft qdisc "new" to class "classid" of qdisc "parent" or
788  * to device "dev".
789  *
790  * When appropriate send a netlink notification using 'skb'
791  * and "n".
792  *
793  * On success, destroy old qdisc.
794  */
795
796 static int qdisc_graft(struct net_device *dev, struct Qdisc *parent,
797                        struct sk_buff *skb, struct nlmsghdr *n, u32 classid,
798                        struct Qdisc *new, struct Qdisc *old)
799 {
800         struct Qdisc *q = old;
801         struct net *net = dev_net(dev);
802         int err = 0;
803
804         if (parent == NULL) {
805                 unsigned int i, num_q, ingress;
806
807                 ingress = 0;
808                 num_q = dev->num_tx_queues;
809                 if ((q && q->flags & TCQ_F_INGRESS) ||
810                     (new && new->flags & TCQ_F_INGRESS)) {
811                         num_q = 1;
812                         ingress = 1;
813                         if (!dev_ingress_queue(dev))
814                                 return -ENOENT;
815                 }
816
817                 if (dev->flags & IFF_UP)
818                         dev_deactivate(dev);
819
820                 if (new && new->ops->attach)
821                         goto skip;
822
823                 for (i = 0; i < num_q; i++) {
824                         struct netdev_queue *dev_queue = dev_ingress_queue(dev);
825
826                         if (!ingress)
827                                 dev_queue = netdev_get_tx_queue(dev, i);
828
829                         old = dev_graft_qdisc(dev_queue, new);
830                         if (new && i > 0)
831                                 atomic_inc(&new->refcnt);
832
833                         if (!ingress)
834                                 qdisc_destroy(old);
835                 }
836
837 skip:
838                 if (!ingress) {
839                         notify_and_destroy(net, skb, n, classid,
840                                            dev->qdisc, new);
841                         if (new && !new->ops->attach)
842                                 atomic_inc(&new->refcnt);
843                         dev->qdisc = new ? : &noop_qdisc;
844
845                         if (new && new->ops->attach)
846                                 new->ops->attach(new);
847                 } else {
848                         notify_and_destroy(net, skb, n, classid, old, new);
849                 }
850
851                 if (dev->flags & IFF_UP)
852                         dev_activate(dev);
853         } else {
854                 const struct Qdisc_class_ops *cops = parent->ops->cl_ops;
855
856                 err = -EOPNOTSUPP;
857                 if (cops && cops->graft) {
858                         unsigned long cl = cops->get(parent, classid);
859                         if (cl) {
860                                 err = cops->graft(parent, cl, new, &old);
861                                 cops->put(parent, cl);
862                         } else
863                                 err = -ENOENT;
864                 }
865                 if (!err)
866                         notify_and_destroy(net, skb, n, classid, old, new);
867         }
868         return err;
869 }
870
871 /* lockdep annotation is needed for ingress; egress gets it only for name */
872 static struct lock_class_key qdisc_tx_lock;
873 static struct lock_class_key qdisc_rx_lock;
874
875 /*
876    Allocate and initialize new qdisc.
877
878    Parameters are passed via opt.
879  */
880
881 static struct Qdisc *qdisc_create(struct net_device *dev,
882                                   struct netdev_queue *dev_queue,
883                                   struct Qdisc *p, u32 parent, u32 handle,
884                                   struct nlattr **tca, int *errp)
885 {
886         int err;
887         struct nlattr *kind = tca[TCA_KIND];
888         struct Qdisc *sch;
889         struct Qdisc_ops *ops;
890         struct qdisc_size_table *stab;
891
892         ops = qdisc_lookup_ops(kind);
893 #ifdef CONFIG_MODULES
894         if (ops == NULL && kind != NULL) {
895                 char name[IFNAMSIZ];
896                 if (nla_strlcpy(name, kind, IFNAMSIZ) < IFNAMSIZ) {
897                         /* We dropped the RTNL semaphore in order to
898                          * perform the module load.  So, even if we
899                          * succeeded in loading the module we have to
900                          * tell the caller to replay the request.  We
901                          * indicate this using -EAGAIN.
902                          * We replay the request because the device may
903                          * go away in the mean time.
904                          */
905                         rtnl_unlock();
906                         request_module("sch_%s", name);
907                         rtnl_lock();
908                         ops = qdisc_lookup_ops(kind);
909                         if (ops != NULL) {
910                                 /* We will try again qdisc_lookup_ops,
911                                  * so don't keep a reference.
912                                  */
913                                 module_put(ops->owner);
914                                 err = -EAGAIN;
915                                 goto err_out;
916                         }
917                 }
918         }
919 #endif
920
921         err = -ENOENT;
922         if (ops == NULL)
923                 goto err_out;
924
925         sch = qdisc_alloc(dev_queue, ops);
926         if (IS_ERR(sch)) {
927                 err = PTR_ERR(sch);
928                 goto err_out2;
929         }
930
931         sch->parent = parent;
932
933         if (handle == TC_H_INGRESS) {
934                 sch->flags |= TCQ_F_INGRESS;
935                 handle = TC_H_MAKE(TC_H_INGRESS, 0);
936                 lockdep_set_class(qdisc_lock(sch), &qdisc_rx_lock);
937         } else {
938                 if (handle == 0) {
939                         handle = qdisc_alloc_handle(dev);
940                         err = -ENOMEM;
941                         if (handle == 0)
942                                 goto err_out3;
943                 }
944                 lockdep_set_class(qdisc_lock(sch), &qdisc_tx_lock);
945                 if (!netif_is_multiqueue(dev))
946                         sch->flags |= TCQ_F_ONETXQUEUE;
947         }
948
949         sch->handle = handle;
950
951         /* This exist to keep backward compatible with a userspace
952          * loophole, what allowed userspace to get IFF_NO_QUEUE
953          * facility on older kernels by setting tx_queue_len=0 (prior
954          * to qdisc init), and then forgot to reinit tx_queue_len
955          * before again attaching a qdisc.
956          */
957         if ((dev->priv_flags & IFF_NO_QUEUE) && (dev->tx_queue_len == 0)) {
958                 dev->tx_queue_len = DEFAULT_TX_QUEUE_LEN;
959                 netdev_info(dev, "Caught tx_queue_len zero misconfig\n");
960         }
961
962         if (!ops->init || (err = ops->init(sch, tca[TCA_OPTIONS])) == 0) {
963                 if (qdisc_is_percpu_stats(sch)) {
964                         sch->cpu_bstats =
965                                 netdev_alloc_pcpu_stats(struct gnet_stats_basic_cpu);
966                         if (!sch->cpu_bstats)
967                                 goto err_out4;
968
969                         sch->cpu_qstats = alloc_percpu(struct gnet_stats_queue);
970                         if (!sch->cpu_qstats)
971                                 goto err_out4;
972                 }
973
974                 if (tca[TCA_STAB]) {
975                         stab = qdisc_get_stab(tca[TCA_STAB]);
976                         if (IS_ERR(stab)) {
977                                 err = PTR_ERR(stab);
978                                 goto err_out4;
979                         }
980                         rcu_assign_pointer(sch->stab, stab);
981                 }
982                 if (tca[TCA_RATE]) {
983                         seqcount_t *running;
984
985                         err = -EOPNOTSUPP;
986                         if (sch->flags & TCQ_F_MQROOT)
987                                 goto err_out4;
988
989                         if ((sch->parent != TC_H_ROOT) &&
990                             !(sch->flags & TCQ_F_INGRESS) &&
991                             (!p || !(p->flags & TCQ_F_MQROOT)))
992                                 running = qdisc_root_sleeping_running(sch);
993                         else
994                                 running = &sch->running;
995
996                         err = gen_new_estimator(&sch->bstats,
997                                                 sch->cpu_bstats,
998                                                 &sch->rate_est,
999                                                 NULL,
1000                                                 running,
1001                                                 tca[TCA_RATE]);
1002                         if (err)
1003                                 goto err_out4;
1004                 }
1005
1006                 qdisc_hash_add(sch);
1007
1008                 return sch;
1009         }
1010         /* ops->init() failed, we call ->destroy() like qdisc_create_dflt() */
1011         ops->destroy(sch);
1012 err_out3:
1013         dev_put(dev);
1014         kfree((char *) sch - sch->padded);
1015 err_out2:
1016         module_put(ops->owner);
1017 err_out:
1018         *errp = err;
1019         return NULL;
1020
1021 err_out4:
1022         free_percpu(sch->cpu_bstats);
1023         free_percpu(sch->cpu_qstats);
1024         /*
1025          * Any broken qdiscs that would require a ops->reset() here?
1026          * The qdisc was never in action so it shouldn't be necessary.
1027          */
1028         qdisc_put_stab(rtnl_dereference(sch->stab));
1029         if (ops->destroy)
1030                 ops->destroy(sch);
1031         goto err_out3;
1032 }
1033
1034 static int qdisc_change(struct Qdisc *sch, struct nlattr **tca)
1035 {
1036         struct qdisc_size_table *ostab, *stab = NULL;
1037         int err = 0;
1038
1039         if (tca[TCA_OPTIONS]) {
1040                 if (sch->ops->change == NULL)
1041                         return -EINVAL;
1042                 err = sch->ops->change(sch, tca[TCA_OPTIONS]);
1043                 if (err)
1044                         return err;
1045         }
1046
1047         if (tca[TCA_STAB]) {
1048                 stab = qdisc_get_stab(tca[TCA_STAB]);
1049                 if (IS_ERR(stab))
1050                         return PTR_ERR(stab);
1051         }
1052
1053         ostab = rtnl_dereference(sch->stab);
1054         rcu_assign_pointer(sch->stab, stab);
1055         qdisc_put_stab(ostab);
1056
1057         if (tca[TCA_RATE]) {
1058                 /* NB: ignores errors from replace_estimator
1059                    because change can't be undone. */
1060                 if (sch->flags & TCQ_F_MQROOT)
1061                         goto out;
1062                 gen_replace_estimator(&sch->bstats,
1063                                       sch->cpu_bstats,
1064                                       &sch->rate_est,
1065                                       NULL,
1066                                       qdisc_root_sleeping_running(sch),
1067                                       tca[TCA_RATE]);
1068         }
1069 out:
1070         return 0;
1071 }
1072
1073 struct check_loop_arg {
1074         struct qdisc_walker     w;
1075         struct Qdisc            *p;
1076         int                     depth;
1077 };
1078
1079 static int check_loop_fn(struct Qdisc *q, unsigned long cl,
1080                          struct qdisc_walker *w);
1081
1082 static int check_loop(struct Qdisc *q, struct Qdisc *p, int depth)
1083 {
1084         struct check_loop_arg   arg;
1085
1086         if (q->ops->cl_ops == NULL)
1087                 return 0;
1088
1089         arg.w.stop = arg.w.skip = arg.w.count = 0;
1090         arg.w.fn = check_loop_fn;
1091         arg.depth = depth;
1092         arg.p = p;
1093         q->ops->cl_ops->walk(q, &arg.w);
1094         return arg.w.stop ? -ELOOP : 0;
1095 }
1096
1097 static int
1098 check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w)
1099 {
1100         struct Qdisc *leaf;
1101         const struct Qdisc_class_ops *cops = q->ops->cl_ops;
1102         struct check_loop_arg *arg = (struct check_loop_arg *)w;
1103
1104         leaf = cops->leaf(q, cl);
1105         if (leaf) {
1106                 if (leaf == arg->p || arg->depth > 7)
1107                         return -ELOOP;
1108                 return check_loop(leaf, arg->p, arg->depth + 1);
1109         }
1110         return 0;
1111 }
1112
1113 /*
1114  * Delete/get qdisc.
1115  */
1116
1117 static int tc_get_qdisc(struct sk_buff *skb, struct nlmsghdr *n)
1118 {
1119         struct net *net = sock_net(skb->sk);
1120         struct tcmsg *tcm = nlmsg_data(n);
1121         struct nlattr *tca[TCA_MAX + 1];
1122         struct net_device *dev;
1123         u32 clid;
1124         struct Qdisc *q = NULL;
1125         struct Qdisc *p = NULL;
1126         int err;
1127
1128         if ((n->nlmsg_type != RTM_GETQDISC) &&
1129             !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
1130                 return -EPERM;
1131
1132         err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1133         if (err < 0)
1134                 return err;
1135
1136         dev = __dev_get_by_index(net, tcm->tcm_ifindex);
1137         if (!dev)
1138                 return -ENODEV;
1139
1140         clid = tcm->tcm_parent;
1141         if (clid) {
1142                 if (clid != TC_H_ROOT) {
1143                         if (TC_H_MAJ(clid) != TC_H_MAJ(TC_H_INGRESS)) {
1144                                 p = qdisc_lookup(dev, TC_H_MAJ(clid));
1145                                 if (!p)
1146                                         return -ENOENT;
1147                                 q = qdisc_leaf(p, clid);
1148                         } else if (dev_ingress_queue(dev)) {
1149                                 q = dev_ingress_queue(dev)->qdisc_sleeping;
1150                         }
1151                 } else {
1152                         q = dev->qdisc;
1153                 }
1154                 if (!q)
1155                         return -ENOENT;
1156
1157                 if (tcm->tcm_handle && q->handle != tcm->tcm_handle)
1158                         return -EINVAL;
1159         } else {
1160                 q = qdisc_lookup(dev, tcm->tcm_handle);
1161                 if (!q)
1162                         return -ENOENT;
1163         }
1164
1165         if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
1166                 return -EINVAL;
1167
1168         if (n->nlmsg_type == RTM_DELQDISC) {
1169                 if (!clid)
1170                         return -EINVAL;
1171                 if (q->handle == 0)
1172                         return -ENOENT;
1173                 err = qdisc_graft(dev, p, skb, n, clid, NULL, q);
1174                 if (err != 0)
1175                         return err;
1176         } else {
1177                 qdisc_notify(net, skb, n, clid, NULL, q);
1178         }
1179         return 0;
1180 }
1181
1182 /*
1183  * Create/change qdisc.
1184  */
1185
1186 static int tc_modify_qdisc(struct sk_buff *skb, struct nlmsghdr *n)
1187 {
1188         struct net *net = sock_net(skb->sk);
1189         struct tcmsg *tcm;
1190         struct nlattr *tca[TCA_MAX + 1];
1191         struct net_device *dev;
1192         u32 clid;
1193         struct Qdisc *q, *p;
1194         int err;
1195
1196         if (!netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
1197                 return -EPERM;
1198
1199 replay:
1200         /* Reinit, just in case something touches this. */
1201         err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1202         if (err < 0)
1203                 return err;
1204
1205         tcm = nlmsg_data(n);
1206         clid = tcm->tcm_parent;
1207         q = p = NULL;
1208
1209         dev = __dev_get_by_index(net, tcm->tcm_ifindex);
1210         if (!dev)
1211                 return -ENODEV;
1212
1213
1214         if (clid) {
1215                 if (clid != TC_H_ROOT) {
1216                         if (clid != TC_H_INGRESS) {
1217                                 p = qdisc_lookup(dev, TC_H_MAJ(clid));
1218                                 if (!p)
1219                                         return -ENOENT;
1220                                 q = qdisc_leaf(p, clid);
1221                         } else if (dev_ingress_queue_create(dev)) {
1222                                 q = dev_ingress_queue(dev)->qdisc_sleeping;
1223                         }
1224                 } else {
1225                         q = dev->qdisc;
1226                 }
1227
1228                 /* It may be default qdisc, ignore it */
1229                 if (q && q->handle == 0)
1230                         q = NULL;
1231
1232                 if (!q || !tcm->tcm_handle || q->handle != tcm->tcm_handle) {
1233                         if (tcm->tcm_handle) {
1234                                 if (q && !(n->nlmsg_flags & NLM_F_REPLACE))
1235                                         return -EEXIST;
1236                                 if (TC_H_MIN(tcm->tcm_handle))
1237                                         return -EINVAL;
1238                                 q = qdisc_lookup(dev, tcm->tcm_handle);
1239                                 if (!q)
1240                                         goto create_n_graft;
1241                                 if (n->nlmsg_flags & NLM_F_EXCL)
1242                                         return -EEXIST;
1243                                 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
1244                                         return -EINVAL;
1245                                 if (q == p ||
1246                                     (p && check_loop(q, p, 0)))
1247                                         return -ELOOP;
1248                                 atomic_inc(&q->refcnt);
1249                                 goto graft;
1250                         } else {
1251                                 if (!q)
1252                                         goto create_n_graft;
1253
1254                                 /* This magic test requires explanation.
1255                                  *
1256                                  *   We know, that some child q is already
1257                                  *   attached to this parent and have choice:
1258                                  *   either to change it or to create/graft new one.
1259                                  *
1260                                  *   1. We are allowed to create/graft only
1261                                  *   if CREATE and REPLACE flags are set.
1262                                  *
1263                                  *   2. If EXCL is set, requestor wanted to say,
1264                                  *   that qdisc tcm_handle is not expected
1265                                  *   to exist, so that we choose create/graft too.
1266                                  *
1267                                  *   3. The last case is when no flags are set.
1268                                  *   Alas, it is sort of hole in API, we
1269                                  *   cannot decide what to do unambiguously.
1270                                  *   For now we select create/graft, if
1271                                  *   user gave KIND, which does not match existing.
1272                                  */
1273                                 if ((n->nlmsg_flags & NLM_F_CREATE) &&
1274                                     (n->nlmsg_flags & NLM_F_REPLACE) &&
1275                                     ((n->nlmsg_flags & NLM_F_EXCL) ||
1276                                      (tca[TCA_KIND] &&
1277                                       nla_strcmp(tca[TCA_KIND], q->ops->id))))
1278                                         goto create_n_graft;
1279                         }
1280                 }
1281         } else {
1282                 if (!tcm->tcm_handle)
1283                         return -EINVAL;
1284                 q = qdisc_lookup(dev, tcm->tcm_handle);
1285         }
1286
1287         /* Change qdisc parameters */
1288         if (q == NULL)
1289                 return -ENOENT;
1290         if (n->nlmsg_flags & NLM_F_EXCL)
1291                 return -EEXIST;
1292         if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
1293                 return -EINVAL;
1294         err = qdisc_change(q, tca);
1295         if (err == 0)
1296                 qdisc_notify(net, skb, n, clid, NULL, q);
1297         return err;
1298
1299 create_n_graft:
1300         if (!(n->nlmsg_flags & NLM_F_CREATE))
1301                 return -ENOENT;
1302         if (clid == TC_H_INGRESS) {
1303                 if (dev_ingress_queue(dev))
1304                         q = qdisc_create(dev, dev_ingress_queue(dev), p,
1305                                          tcm->tcm_parent, tcm->tcm_parent,
1306                                          tca, &err);
1307                 else
1308                         err = -ENOENT;
1309         } else {
1310                 struct netdev_queue *dev_queue;
1311
1312                 if (p && p->ops->cl_ops && p->ops->cl_ops->select_queue)
1313                         dev_queue = p->ops->cl_ops->select_queue(p, tcm);
1314                 else if (p)
1315                         dev_queue = p->dev_queue;
1316                 else
1317                         dev_queue = netdev_get_tx_queue(dev, 0);
1318
1319                 q = qdisc_create(dev, dev_queue, p,
1320                                  tcm->tcm_parent, tcm->tcm_handle,
1321                                  tca, &err);
1322         }
1323         if (q == NULL) {
1324                 if (err == -EAGAIN)
1325                         goto replay;
1326                 return err;
1327         }
1328
1329 graft:
1330         err = qdisc_graft(dev, p, skb, n, clid, q, NULL);
1331         if (err) {
1332                 if (q)
1333                         qdisc_destroy(q);
1334                 return err;
1335         }
1336
1337         return 0;
1338 }
1339
1340 static int tc_fill_qdisc(struct sk_buff *skb, struct Qdisc *q, u32 clid,
1341                          u32 portid, u32 seq, u16 flags, int event)
1342 {
1343         struct gnet_stats_basic_cpu __percpu *cpu_bstats = NULL;
1344         struct gnet_stats_queue __percpu *cpu_qstats = NULL;
1345         struct tcmsg *tcm;
1346         struct nlmsghdr  *nlh;
1347         unsigned char *b = skb_tail_pointer(skb);
1348         struct gnet_dump d;
1349         struct qdisc_size_table *stab;
1350         __u32 qlen;
1351
1352         cond_resched();
1353         nlh = nlmsg_put(skb, portid, seq, event, sizeof(*tcm), flags);
1354         if (!nlh)
1355                 goto out_nlmsg_trim;
1356         tcm = nlmsg_data(nlh);
1357         tcm->tcm_family = AF_UNSPEC;
1358         tcm->tcm__pad1 = 0;
1359         tcm->tcm__pad2 = 0;
1360         tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
1361         tcm->tcm_parent = clid;
1362         tcm->tcm_handle = q->handle;
1363         tcm->tcm_info = atomic_read(&q->refcnt);
1364         if (nla_put_string(skb, TCA_KIND, q->ops->id))
1365                 goto nla_put_failure;
1366         if (q->ops->dump && q->ops->dump(q, skb) < 0)
1367                 goto nla_put_failure;
1368         qlen = q->q.qlen;
1369
1370         stab = rtnl_dereference(q->stab);
1371         if (stab && qdisc_dump_stab(skb, stab) < 0)
1372                 goto nla_put_failure;
1373
1374         if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
1375                                          NULL, &d, TCA_PAD) < 0)
1376                 goto nla_put_failure;
1377
1378         if (q->ops->dump_stats && q->ops->dump_stats(q, &d) < 0)
1379                 goto nla_put_failure;
1380
1381         if (qdisc_is_percpu_stats(q)) {
1382                 cpu_bstats = q->cpu_bstats;
1383                 cpu_qstats = q->cpu_qstats;
1384         }
1385
1386         if (gnet_stats_copy_basic(qdisc_root_sleeping_running(q),
1387                                   &d, cpu_bstats, &q->bstats) < 0 ||
1388             gnet_stats_copy_rate_est(&d, &q->rate_est) < 0 ||
1389             gnet_stats_copy_queue(&d, cpu_qstats, &q->qstats, qlen) < 0)
1390                 goto nla_put_failure;
1391
1392         if (gnet_stats_finish_copy(&d) < 0)
1393                 goto nla_put_failure;
1394
1395         nlh->nlmsg_len = skb_tail_pointer(skb) - b;
1396         return skb->len;
1397
1398 out_nlmsg_trim:
1399 nla_put_failure:
1400         nlmsg_trim(skb, b);
1401         return -1;
1402 }
1403
1404 static bool tc_qdisc_dump_ignore(struct Qdisc *q)
1405 {
1406         return (q->flags & TCQ_F_BUILTIN) ? true : false;
1407 }
1408
1409 static int qdisc_notify(struct net *net, struct sk_buff *oskb,
1410                         struct nlmsghdr *n, u32 clid,
1411                         struct Qdisc *old, struct Qdisc *new)
1412 {
1413         struct sk_buff *skb;
1414         u32 portid = oskb ? NETLINK_CB(oskb).portid : 0;
1415
1416         skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1417         if (!skb)
1418                 return -ENOBUFS;
1419
1420         if (old && !tc_qdisc_dump_ignore(old)) {
1421                 if (tc_fill_qdisc(skb, old, clid, portid, n->nlmsg_seq,
1422                                   0, RTM_DELQDISC) < 0)
1423                         goto err_out;
1424         }
1425         if (new && !tc_qdisc_dump_ignore(new)) {
1426                 if (tc_fill_qdisc(skb, new, clid, portid, n->nlmsg_seq,
1427                                   old ? NLM_F_REPLACE : 0, RTM_NEWQDISC) < 0)
1428                         goto err_out;
1429         }
1430
1431         if (skb->len)
1432                 return rtnetlink_send(skb, net, portid, RTNLGRP_TC,
1433                                       n->nlmsg_flags & NLM_F_ECHO);
1434
1435 err_out:
1436         kfree_skb(skb);
1437         return -EINVAL;
1438 }
1439
1440 static int tc_dump_qdisc_root(struct Qdisc *root, struct sk_buff *skb,
1441                               struct netlink_callback *cb,
1442                               int *q_idx_p, int s_q_idx, bool recur)
1443 {
1444         int ret = 0, q_idx = *q_idx_p;
1445         struct Qdisc *q;
1446         int b;
1447
1448         if (!root)
1449                 return 0;
1450
1451         q = root;
1452         if (q_idx < s_q_idx) {
1453                 q_idx++;
1454         } else {
1455                 if (!tc_qdisc_dump_ignore(q) &&
1456                     tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).portid,
1457                                   cb->nlh->nlmsg_seq, NLM_F_MULTI,
1458                                   RTM_NEWQDISC) <= 0)
1459                         goto done;
1460                 q_idx++;
1461         }
1462
1463         /* If dumping singletons, there is no qdisc_dev(root) and the singleton
1464          * itself has already been dumped.
1465          *
1466          * If we've already dumped the top-level (ingress) qdisc above and the global
1467          * qdisc hashtable, we don't want to hit it again
1468          */
1469         if (!qdisc_dev(root) || !recur)
1470                 goto out;
1471
1472         hash_for_each(qdisc_dev(root)->qdisc_hash, b, q, hash) {
1473                 if (q_idx < s_q_idx) {
1474                         q_idx++;
1475                         continue;
1476                 }
1477                 if (!tc_qdisc_dump_ignore(q) &&
1478                     tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).portid,
1479                                   cb->nlh->nlmsg_seq, NLM_F_MULTI,
1480                                   RTM_NEWQDISC) <= 0)
1481                         goto done;
1482                 q_idx++;
1483         }
1484
1485 out:
1486         *q_idx_p = q_idx;
1487         return ret;
1488 done:
1489         ret = -1;
1490         goto out;
1491 }
1492
1493 static int tc_dump_qdisc(struct sk_buff *skb, struct netlink_callback *cb)
1494 {
1495         struct net *net = sock_net(skb->sk);
1496         int idx, q_idx;
1497         int s_idx, s_q_idx;
1498         struct net_device *dev;
1499
1500         s_idx = cb->args[0];
1501         s_q_idx = q_idx = cb->args[1];
1502
1503         idx = 0;
1504         ASSERT_RTNL();
1505         for_each_netdev(net, dev) {
1506                 struct netdev_queue *dev_queue;
1507
1508                 if (idx < s_idx)
1509                         goto cont;
1510                 if (idx > s_idx)
1511                         s_q_idx = 0;
1512                 q_idx = 0;
1513
1514                 if (tc_dump_qdisc_root(dev->qdisc, skb, cb, &q_idx, s_q_idx,
1515                                        true) < 0)
1516                         goto done;
1517
1518                 dev_queue = dev_ingress_queue(dev);
1519                 if (dev_queue &&
1520                     tc_dump_qdisc_root(dev_queue->qdisc_sleeping, skb, cb,
1521                                        &q_idx, s_q_idx, false) < 0)
1522                         goto done;
1523
1524 cont:
1525                 idx++;
1526         }
1527
1528 done:
1529         cb->args[0] = idx;
1530         cb->args[1] = q_idx;
1531
1532         return skb->len;
1533 }
1534
1535
1536
1537 /************************************************
1538  *      Traffic classes manipulation.           *
1539  ************************************************/
1540
1541
1542
1543 static int tc_ctl_tclass(struct sk_buff *skb, struct nlmsghdr *n)
1544 {
1545         struct net *net = sock_net(skb->sk);
1546         struct tcmsg *tcm = nlmsg_data(n);
1547         struct nlattr *tca[TCA_MAX + 1];
1548         struct net_device *dev;
1549         struct Qdisc *q = NULL;
1550         const struct Qdisc_class_ops *cops;
1551         unsigned long cl = 0;
1552         unsigned long new_cl;
1553         u32 portid;
1554         u32 clid;
1555         u32 qid;
1556         int err;
1557
1558         if ((n->nlmsg_type != RTM_GETTCLASS) &&
1559             !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
1560                 return -EPERM;
1561
1562         err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1563         if (err < 0)
1564                 return err;
1565
1566         dev = __dev_get_by_index(net, tcm->tcm_ifindex);
1567         if (!dev)
1568                 return -ENODEV;
1569
1570         /*
1571            parent == TC_H_UNSPEC - unspecified parent.
1572            parent == TC_H_ROOT   - class is root, which has no parent.
1573            parent == X:0         - parent is root class.
1574            parent == X:Y         - parent is a node in hierarchy.
1575            parent == 0:Y         - parent is X:Y, where X:0 is qdisc.
1576
1577            handle == 0:0         - generate handle from kernel pool.
1578            handle == 0:Y         - class is X:Y, where X:0 is qdisc.
1579            handle == X:Y         - clear.
1580            handle == X:0         - root class.
1581          */
1582
1583         /* Step 1. Determine qdisc handle X:0 */
1584
1585         portid = tcm->tcm_parent;
1586         clid = tcm->tcm_handle;
1587         qid = TC_H_MAJ(clid);
1588
1589         if (portid != TC_H_ROOT) {
1590                 u32 qid1 = TC_H_MAJ(portid);
1591
1592                 if (qid && qid1) {
1593                         /* If both majors are known, they must be identical. */
1594                         if (qid != qid1)
1595                                 return -EINVAL;
1596                 } else if (qid1) {
1597                         qid = qid1;
1598                 } else if (qid == 0)
1599                         qid = dev->qdisc->handle;
1600
1601                 /* Now qid is genuine qdisc handle consistent
1602                  * both with parent and child.
1603                  *
1604                  * TC_H_MAJ(portid) still may be unspecified, complete it now.
1605                  */
1606                 if (portid)
1607                         portid = TC_H_MAKE(qid, portid);
1608         } else {
1609                 if (qid == 0)
1610                         qid = dev->qdisc->handle;
1611         }
1612
1613         /* OK. Locate qdisc */
1614         q = qdisc_lookup(dev, qid);
1615         if (!q)
1616                 return -ENOENT;
1617
1618         /* An check that it supports classes */
1619         cops = q->ops->cl_ops;
1620         if (cops == NULL)
1621                 return -EINVAL;
1622
1623         /* Now try to get class */
1624         if (clid == 0) {
1625                 if (portid == TC_H_ROOT)
1626                         clid = qid;
1627         } else
1628                 clid = TC_H_MAKE(qid, clid);
1629
1630         if (clid)
1631                 cl = cops->get(q, clid);
1632
1633         if (cl == 0) {
1634                 err = -ENOENT;
1635                 if (n->nlmsg_type != RTM_NEWTCLASS ||
1636                     !(n->nlmsg_flags & NLM_F_CREATE))
1637                         goto out;
1638         } else {
1639                 switch (n->nlmsg_type) {
1640                 case RTM_NEWTCLASS:
1641                         err = -EEXIST;
1642                         if (n->nlmsg_flags & NLM_F_EXCL)
1643                                 goto out;
1644                         break;
1645                 case RTM_DELTCLASS:
1646                         err = -EOPNOTSUPP;
1647                         if (cops->delete)
1648                                 err = cops->delete(q, cl);
1649                         if (err == 0)
1650                                 tclass_notify(net, skb, n, q, cl,
1651                                               RTM_DELTCLASS);
1652                         goto out;
1653                 case RTM_GETTCLASS:
1654                         err = tclass_notify(net, skb, n, q, cl, RTM_NEWTCLASS);
1655                         goto out;
1656                 default:
1657                         err = -EINVAL;
1658                         goto out;
1659                 }
1660         }
1661
1662         new_cl = cl;
1663         err = -EOPNOTSUPP;
1664         if (cops->change)
1665                 err = cops->change(q, clid, portid, tca, &new_cl);
1666         if (err == 0)
1667                 tclass_notify(net, skb, n, q, new_cl, RTM_NEWTCLASS);
1668
1669 out:
1670         if (cl)
1671                 cops->put(q, cl);
1672
1673         return err;
1674 }
1675
1676
1677 static int tc_fill_tclass(struct sk_buff *skb, struct Qdisc *q,
1678                           unsigned long cl,
1679                           u32 portid, u32 seq, u16 flags, int event)
1680 {
1681         struct tcmsg *tcm;
1682         struct nlmsghdr  *nlh;
1683         unsigned char *b = skb_tail_pointer(skb);
1684         struct gnet_dump d;
1685         const struct Qdisc_class_ops *cl_ops = q->ops->cl_ops;
1686
1687         cond_resched();
1688         nlh = nlmsg_put(skb, portid, seq, event, sizeof(*tcm), flags);
1689         if (!nlh)
1690                 goto out_nlmsg_trim;
1691         tcm = nlmsg_data(nlh);
1692         tcm->tcm_family = AF_UNSPEC;
1693         tcm->tcm__pad1 = 0;
1694         tcm->tcm__pad2 = 0;
1695         tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
1696         tcm->tcm_parent = q->handle;
1697         tcm->tcm_handle = q->handle;
1698         tcm->tcm_info = 0;
1699         if (nla_put_string(skb, TCA_KIND, q->ops->id))
1700                 goto nla_put_failure;
1701         if (cl_ops->dump && cl_ops->dump(q, cl, skb, tcm) < 0)
1702                 goto nla_put_failure;
1703
1704         if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
1705                                          NULL, &d, TCA_PAD) < 0)
1706                 goto nla_put_failure;
1707
1708         if (cl_ops->dump_stats && cl_ops->dump_stats(q, cl, &d) < 0)
1709                 goto nla_put_failure;
1710
1711         if (gnet_stats_finish_copy(&d) < 0)
1712                 goto nla_put_failure;
1713
1714         nlh->nlmsg_len = skb_tail_pointer(skb) - b;
1715         return skb->len;
1716
1717 out_nlmsg_trim:
1718 nla_put_failure:
1719         nlmsg_trim(skb, b);
1720         return -1;
1721 }
1722
1723 static int tclass_notify(struct net *net, struct sk_buff *oskb,
1724                          struct nlmsghdr *n, struct Qdisc *q,
1725                          unsigned long cl, int event)
1726 {
1727         struct sk_buff *skb;
1728         u32 portid = oskb ? NETLINK_CB(oskb).portid : 0;
1729
1730         skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1731         if (!skb)
1732                 return -ENOBUFS;
1733
1734         if (tc_fill_tclass(skb, q, cl, portid, n->nlmsg_seq, 0, event) < 0) {
1735                 kfree_skb(skb);
1736                 return -EINVAL;
1737         }
1738
1739         return rtnetlink_send(skb, net, portid, RTNLGRP_TC,
1740                               n->nlmsg_flags & NLM_F_ECHO);
1741 }
1742
1743 struct qdisc_dump_args {
1744         struct qdisc_walker     w;
1745         struct sk_buff          *skb;
1746         struct netlink_callback *cb;
1747 };
1748
1749 static int qdisc_class_dump(struct Qdisc *q, unsigned long cl,
1750                             struct qdisc_walker *arg)
1751 {
1752         struct qdisc_dump_args *a = (struct qdisc_dump_args *)arg;
1753
1754         return tc_fill_tclass(a->skb, q, cl, NETLINK_CB(a->cb->skb).portid,
1755                               a->cb->nlh->nlmsg_seq, NLM_F_MULTI,
1756                               RTM_NEWTCLASS);
1757 }
1758
1759 static int tc_dump_tclass_qdisc(struct Qdisc *q, struct sk_buff *skb,
1760                                 struct tcmsg *tcm, struct netlink_callback *cb,
1761                                 int *t_p, int s_t)
1762 {
1763         struct qdisc_dump_args arg;
1764
1765         if (tc_qdisc_dump_ignore(q) ||
1766             *t_p < s_t || !q->ops->cl_ops ||
1767             (tcm->tcm_parent &&
1768              TC_H_MAJ(tcm->tcm_parent) != q->handle)) {
1769                 (*t_p)++;
1770                 return 0;
1771         }
1772         if (*t_p > s_t)
1773                 memset(&cb->args[1], 0, sizeof(cb->args)-sizeof(cb->args[0]));
1774         arg.w.fn = qdisc_class_dump;
1775         arg.skb = skb;
1776         arg.cb = cb;
1777         arg.w.stop  = 0;
1778         arg.w.skip = cb->args[1];
1779         arg.w.count = 0;
1780         q->ops->cl_ops->walk(q, &arg.w);
1781         cb->args[1] = arg.w.count;
1782         if (arg.w.stop)
1783                 return -1;
1784         (*t_p)++;
1785         return 0;
1786 }
1787
1788 static int tc_dump_tclass_root(struct Qdisc *root, struct sk_buff *skb,
1789                                struct tcmsg *tcm, struct netlink_callback *cb,
1790                                int *t_p, int s_t)
1791 {
1792         struct Qdisc *q;
1793         int b;
1794
1795         if (!root)
1796                 return 0;
1797
1798         if (tc_dump_tclass_qdisc(root, skb, tcm, cb, t_p, s_t) < 0)
1799                 return -1;
1800
1801         if (!qdisc_dev(root))
1802                 return 0;
1803
1804         hash_for_each(qdisc_dev(root)->qdisc_hash, b, q, hash) {
1805                 if (tc_dump_tclass_qdisc(q, skb, tcm, cb, t_p, s_t) < 0)
1806                         return -1;
1807         }
1808
1809         return 0;
1810 }
1811
1812 static int tc_dump_tclass(struct sk_buff *skb, struct netlink_callback *cb)
1813 {
1814         struct tcmsg *tcm = nlmsg_data(cb->nlh);
1815         struct net *net = sock_net(skb->sk);
1816         struct netdev_queue *dev_queue;
1817         struct net_device *dev;
1818         int t, s_t;
1819
1820         if (nlmsg_len(cb->nlh) < sizeof(*tcm))
1821                 return 0;
1822         dev = dev_get_by_index(net, tcm->tcm_ifindex);
1823         if (!dev)
1824                 return 0;
1825
1826         s_t = cb->args[0];
1827         t = 0;
1828
1829         if (tc_dump_tclass_root(dev->qdisc, skb, tcm, cb, &t, s_t) < 0)
1830                 goto done;
1831
1832         dev_queue = dev_ingress_queue(dev);
1833         if (dev_queue &&
1834             tc_dump_tclass_root(dev_queue->qdisc_sleeping, skb, tcm, cb,
1835                                 &t, s_t) < 0)
1836                 goto done;
1837
1838 done:
1839         cb->args[0] = t;
1840
1841         dev_put(dev);
1842         return skb->len;
1843 }
1844
1845 /* Main classifier routine: scans classifier chain attached
1846  * to this qdisc, (optionally) tests for protocol and asks
1847  * specific classifiers.
1848  */
1849 int tc_classify(struct sk_buff *skb, const struct tcf_proto *tp,
1850                 struct tcf_result *res, bool compat_mode)
1851 {
1852         __be16 protocol = tc_skb_protocol(skb);
1853 #ifdef CONFIG_NET_CLS_ACT
1854         const int max_reclassify_loop = 4;
1855         const struct tcf_proto *old_tp = tp;
1856         int limit = 0;
1857
1858 reclassify:
1859 #endif
1860         for (; tp; tp = rcu_dereference_bh(tp->next)) {
1861                 int err;
1862
1863                 if (tp->protocol != protocol &&
1864                     tp->protocol != htons(ETH_P_ALL))
1865                         continue;
1866
1867                 err = tp->classify(skb, tp, res);
1868 #ifdef CONFIG_NET_CLS_ACT
1869                 if (unlikely(err == TC_ACT_RECLASSIFY && !compat_mode))
1870                         goto reset;
1871 #endif
1872                 if (err >= 0)
1873                         return err;
1874         }
1875
1876         return TC_ACT_UNSPEC; /* signal: continue lookup */
1877 #ifdef CONFIG_NET_CLS_ACT
1878 reset:
1879         if (unlikely(limit++ >= max_reclassify_loop)) {
1880                 net_notice_ratelimited("%s: reclassify loop, rule prio %u, protocol %02x\n",
1881                                        tp->q->ops->id, tp->prio & 0xffff,
1882                                        ntohs(tp->protocol));
1883                 return TC_ACT_SHOT;
1884         }
1885
1886         tp = old_tp;
1887         protocol = tc_skb_protocol(skb);
1888         goto reclassify;
1889 #endif
1890 }
1891 EXPORT_SYMBOL(tc_classify);
1892
1893 #ifdef CONFIG_PROC_FS
1894 static int psched_show(struct seq_file *seq, void *v)
1895 {
1896         seq_printf(seq, "%08x %08x %08x %08x\n",
1897                    (u32)NSEC_PER_USEC, (u32)PSCHED_TICKS2NS(1),
1898                    1000000,
1899                    (u32)NSEC_PER_SEC / hrtimer_resolution);
1900
1901         return 0;
1902 }
1903
1904 static int psched_open(struct inode *inode, struct file *file)
1905 {
1906         return single_open(file, psched_show, NULL);
1907 }
1908
1909 static const struct file_operations psched_fops = {
1910         .owner = THIS_MODULE,
1911         .open = psched_open,
1912         .read  = seq_read,
1913         .llseek = seq_lseek,
1914         .release = single_release,
1915 };
1916
1917 static int __net_init psched_net_init(struct net *net)
1918 {
1919         struct proc_dir_entry *e;
1920
1921         e = proc_create("psched", 0, net->proc_net, &psched_fops);
1922         if (e == NULL)
1923                 return -ENOMEM;
1924
1925         return 0;
1926 }
1927
1928 static void __net_exit psched_net_exit(struct net *net)
1929 {
1930         remove_proc_entry("psched", net->proc_net);
1931 }
1932 #else
1933 static int __net_init psched_net_init(struct net *net)
1934 {
1935         return 0;
1936 }
1937
1938 static void __net_exit psched_net_exit(struct net *net)
1939 {
1940 }
1941 #endif
1942
1943 static struct pernet_operations psched_net_ops = {
1944         .init = psched_net_init,
1945         .exit = psched_net_exit,
1946 };
1947
1948 static int __init pktsched_init(void)
1949 {
1950         int err;
1951
1952         err = register_pernet_subsys(&psched_net_ops);
1953         if (err) {
1954                 pr_err("pktsched_init: "
1955                        "cannot initialize per netns operations\n");
1956                 return err;
1957         }
1958
1959         register_qdisc(&pfifo_fast_ops);
1960         register_qdisc(&pfifo_qdisc_ops);
1961         register_qdisc(&bfifo_qdisc_ops);
1962         register_qdisc(&pfifo_head_drop_qdisc_ops);
1963         register_qdisc(&mq_qdisc_ops);
1964         register_qdisc(&noqueue_qdisc_ops);
1965
1966         rtnl_register(PF_UNSPEC, RTM_NEWQDISC, tc_modify_qdisc, NULL, NULL);
1967         rtnl_register(PF_UNSPEC, RTM_DELQDISC, tc_get_qdisc, NULL, NULL);
1968         rtnl_register(PF_UNSPEC, RTM_GETQDISC, tc_get_qdisc, tc_dump_qdisc,
1969                       NULL);
1970         rtnl_register(PF_UNSPEC, RTM_NEWTCLASS, tc_ctl_tclass, NULL, NULL);
1971         rtnl_register(PF_UNSPEC, RTM_DELTCLASS, tc_ctl_tclass, NULL, NULL);
1972         rtnl_register(PF_UNSPEC, RTM_GETTCLASS, tc_ctl_tclass, tc_dump_tclass,
1973                       NULL);
1974
1975         return 0;
1976 }
1977
1978 subsys_initcall(pktsched_init);