blob: 7d7070b1eebd3dd5290776fe3cdf230ee8232b70 [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
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
Linus Torvalds1da177e2005-04-16 15:20:36 -070018#include <linux/module.h>
19#include <linux/types.h>
20#include <linux/kernel.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070021#include <linux/string.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070022#include <linux/errno.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070023#include <linux/skbuff.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070024#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>
Patrick McHardy41794772007-03-16 01:19:15 -070029#include <linux/hrtimer.h>
Jarek Poplawski25bfcd52008-08-18 20:53:34 -070030#include <linux/lockdep.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070031
Eric W. Biederman457c4cb2007-09-12 12:01:34 +020032#include <net/net_namespace.h>
Denis V. Lunevb8542722007-12-01 00:21:31 +110033#include <net/sock.h>
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -070034#include <net/netlink.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070035#include <net/pkt_sched.h>
36
Linus Torvalds1da177e2005-04-16 15:20:36 -070037static int qdisc_notify(struct sk_buff *oskb, struct nlmsghdr *n, u32 clid,
38 struct Qdisc *old, struct Qdisc *new);
39static int tclass_notify(struct sk_buff *oskb, struct nlmsghdr *n,
40 struct Qdisc *q, unsigned long cl, int event);
41
42/*
43
44 Short review.
45 -------------
46
47 This file consists of two interrelated parts:
48
49 1. queueing disciplines manager frontend.
50 2. traffic classes manager frontend.
51
52 Generally, queueing discipline ("qdisc") is a black box,
53 which is able to enqueue packets and to dequeue them (when
54 device is ready to send something) in order and at times
55 determined by algorithm hidden in it.
56
57 qdisc's are divided to two categories:
58 - "queues", which have no internal structure visible from outside.
59 - "schedulers", which split all the packets to "traffic classes",
60 using "packet classifiers" (look at cls_api.c)
61
62 In turn, classes may have child qdiscs (as rule, queues)
63 attached to them etc. etc. etc.
64
65 The goal of the routines in this file is to translate
66 information supplied by user in the form of handles
67 to more intelligible for kernel form, to make some sanity
68 checks and part of work, which is common to all qdiscs
69 and to provide rtnetlink notifications.
70
71 All real intelligent work is done inside qdisc modules.
72
73
74
75 Every discipline has two major routines: enqueue and dequeue.
76
77 ---dequeue
78
79 dequeue usually returns a skb to send. It is allowed to return NULL,
80 but it does not mean that queue is empty, it just means that
81 discipline does not want to send anything this time.
82 Queue is really empty if q->q.qlen == 0.
83 For complicated disciplines with multiple queues q->q is not
84 real packet queue, but however q->q.qlen must be valid.
85
86 ---enqueue
87
88 enqueue returns 0, if packet was enqueued successfully.
89 If packet (this one or another one) was dropped, it returns
90 not zero error code.
91 NET_XMIT_DROP - this packet dropped
92 Expected action: do not backoff, but wait until queue will clear.
93 NET_XMIT_CN - probably this packet enqueued, but another one dropped.
94 Expected action: backoff or ignore
95 NET_XMIT_POLICED - dropped by police.
96 Expected action: backoff or error to real-time apps.
97
98 Auxiliary routines:
99
100 ---requeue
101
102 requeues once dequeued packet. It is used for non-standard or
David S. Millere65d22e2008-07-08 16:46:01 -0700103 just buggy devices, which can defer output even if netif_queue_stopped()=0.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700104
105 ---reset
106
107 returns qdisc to initial state: purge all buffers, clear all
108 timers, counters (except for statistics) etc.
109
110 ---init
111
112 initializes newly created qdisc.
113
114 ---destroy
115
116 destroys resources allocated by init and during lifetime of qdisc.
117
118 ---change
119
120 changes qdisc parameters.
121 */
122
123/* Protects list of registered TC modules. It is pure SMP lock. */
124static DEFINE_RWLOCK(qdisc_mod_lock);
125
126
127/************************************************
128 * Queueing disciplines manipulation. *
129 ************************************************/
130
131
132/* The list of all installed queueing disciplines. */
133
134static struct Qdisc_ops *qdisc_base;
135
136/* Register/uregister queueing discipline */
137
138int register_qdisc(struct Qdisc_ops *qops)
139{
140 struct Qdisc_ops *q, **qp;
141 int rc = -EEXIST;
142
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;
147
148 if (qops->enqueue == NULL)
149 qops->enqueue = noop_qdisc_ops.enqueue;
150 if (qops->requeue == NULL)
151 qops->requeue = noop_qdisc_ops.requeue;
152 if (qops->dequeue == NULL)
153 qops->dequeue = noop_qdisc_ops.dequeue;
154
155 qops->next = NULL;
156 *qp = qops;
157 rc = 0;
158out:
159 write_unlock(&qdisc_mod_lock);
160 return rc;
161}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800162EXPORT_SYMBOL(register_qdisc);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700163
164int unregister_qdisc(struct Qdisc_ops *qops)
165{
166 struct Qdisc_ops *q, **qp;
167 int err = -ENOENT;
168
169 write_lock(&qdisc_mod_lock);
170 for (qp = &qdisc_base; (q=*qp)!=NULL; qp = &q->next)
171 if (q == qops)
172 break;
173 if (q) {
174 *qp = q->next;
175 q->next = NULL;
176 err = 0;
177 }
178 write_unlock(&qdisc_mod_lock);
179 return err;
180}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800181EXPORT_SYMBOL(unregister_qdisc);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700182
183/* We know handle. Find qdisc among all qdisc's attached to device
184 (root qdisc, all its children, children of children etc.)
185 */
186
David S. Miller8123b422008-08-08 23:23:39 -0700187struct Qdisc *qdisc_match_from_root(struct Qdisc *root, u32 handle)
188{
189 struct Qdisc *q;
190
191 if (!(root->flags & TCQ_F_BUILTIN) &&
192 root->handle == handle)
193 return root;
194
195 list_for_each_entry(q, &root->list, list) {
196 if (q->handle == handle)
197 return q;
198 }
199 return NULL;
200}
201
David S. Milleread81cc2008-07-17 00:50:32 -0700202struct Qdisc *qdisc_lookup(struct net_device *dev, u32 handle)
Patrick McHardy43effa12006-11-29 17:35:48 -0800203{
David S. Miller30723672008-07-18 22:50:15 -0700204 unsigned int i;
Patrick McHardy43effa12006-11-29 17:35:48 -0800205
David S. Miller30723672008-07-18 22:50:15 -0700206 for (i = 0; i < dev->num_tx_queues; i++) {
207 struct netdev_queue *txq = netdev_get_tx_queue(dev, i);
David S. Miller827ebd62008-08-07 20:26:40 -0700208 struct Qdisc *q, *txq_root = txq->qdisc_sleeping;
David S. Miller30723672008-07-18 22:50:15 -0700209
David S. Miller8123b422008-08-08 23:23:39 -0700210 q = qdisc_match_from_root(txq_root, handle);
211 if (q)
212 return q;
Patrick McHardy43effa12006-11-29 17:35:48 -0800213 }
David S. Miller8123b422008-08-08 23:23:39 -0700214 return qdisc_match_from_root(dev->rx_queue.qdisc_sleeping, handle);
Patrick McHardy43effa12006-11-29 17:35:48 -0800215}
216
Linus Torvalds1da177e2005-04-16 15:20:36 -0700217static struct Qdisc *qdisc_leaf(struct Qdisc *p, u32 classid)
218{
219 unsigned long cl;
220 struct Qdisc *leaf;
Eric Dumazet20fea082007-11-14 01:44:41 -0800221 const struct Qdisc_class_ops *cops = p->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700222
223 if (cops == NULL)
224 return NULL;
225 cl = cops->get(p, classid);
226
227 if (cl == 0)
228 return NULL;
229 leaf = cops->leaf(p, cl);
230 cops->put(p, cl);
231 return leaf;
232}
233
234/* Find queueing discipline by name */
235
Patrick McHardy1e904742008-01-22 22:11:17 -0800236static struct Qdisc_ops *qdisc_lookup_ops(struct nlattr *kind)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700237{
238 struct Qdisc_ops *q = NULL;
239
240 if (kind) {
241 read_lock(&qdisc_mod_lock);
242 for (q = qdisc_base; q; q = q->next) {
Patrick McHardy1e904742008-01-22 22:11:17 -0800243 if (nla_strcmp(kind, q->id) == 0) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700244 if (!try_module_get(q->owner))
245 q = NULL;
246 break;
247 }
248 }
249 read_unlock(&qdisc_mod_lock);
250 }
251 return q;
252}
253
254static struct qdisc_rate_table *qdisc_rtab_list;
255
Patrick McHardy1e904742008-01-22 22:11:17 -0800256struct qdisc_rate_table *qdisc_get_rtab(struct tc_ratespec *r, struct nlattr *tab)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700257{
258 struct qdisc_rate_table *rtab;
259
260 for (rtab = qdisc_rtab_list; rtab; rtab = rtab->next) {
261 if (memcmp(&rtab->rate, r, sizeof(struct tc_ratespec)) == 0) {
262 rtab->refcnt++;
263 return rtab;
264 }
265 }
266
Patrick McHardy5feb5e12008-01-23 20:35:19 -0800267 if (tab == NULL || r->rate == 0 || r->cell_log == 0 ||
268 nla_len(tab) != TC_RTAB_SIZE)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700269 return NULL;
270
271 rtab = kmalloc(sizeof(*rtab), GFP_KERNEL);
272 if (rtab) {
273 rtab->rate = *r;
274 rtab->refcnt = 1;
Patrick McHardy1e904742008-01-22 22:11:17 -0800275 memcpy(rtab->data, nla_data(tab), 1024);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700276 rtab->next = qdisc_rtab_list;
277 qdisc_rtab_list = rtab;
278 }
279 return rtab;
280}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800281EXPORT_SYMBOL(qdisc_get_rtab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700282
283void qdisc_put_rtab(struct qdisc_rate_table *tab)
284{
285 struct qdisc_rate_table *rtab, **rtabp;
286
287 if (!tab || --tab->refcnt)
288 return;
289
290 for (rtabp = &qdisc_rtab_list; (rtab=*rtabp) != NULL; rtabp = &rtab->next) {
291 if (rtab == tab) {
292 *rtabp = rtab->next;
293 kfree(rtab);
294 return;
295 }
296 }
297}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800298EXPORT_SYMBOL(qdisc_put_rtab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700299
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700300static LIST_HEAD(qdisc_stab_list);
301static DEFINE_SPINLOCK(qdisc_stab_lock);
302
303static const struct nla_policy stab_policy[TCA_STAB_MAX + 1] = {
304 [TCA_STAB_BASE] = { .len = sizeof(struct tc_sizespec) },
305 [TCA_STAB_DATA] = { .type = NLA_BINARY },
306};
307
308static struct qdisc_size_table *qdisc_get_stab(struct nlattr *opt)
309{
310 struct nlattr *tb[TCA_STAB_MAX + 1];
311 struct qdisc_size_table *stab;
312 struct tc_sizespec *s;
313 unsigned int tsize = 0;
314 u16 *tab = NULL;
315 int err;
316
317 err = nla_parse_nested(tb, TCA_STAB_MAX, opt, stab_policy);
318 if (err < 0)
319 return ERR_PTR(err);
320 if (!tb[TCA_STAB_BASE])
321 return ERR_PTR(-EINVAL);
322
323 s = nla_data(tb[TCA_STAB_BASE]);
324
325 if (s->tsize > 0) {
326 if (!tb[TCA_STAB_DATA])
327 return ERR_PTR(-EINVAL);
328 tab = nla_data(tb[TCA_STAB_DATA]);
329 tsize = nla_len(tb[TCA_STAB_DATA]) / sizeof(u16);
330 }
331
332 if (!s || tsize != s->tsize || (!tab && tsize > 0))
333 return ERR_PTR(-EINVAL);
334
Jarek Poplawski1cfa2662008-08-11 18:11:06 -0700335 spin_lock_bh(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700336
337 list_for_each_entry(stab, &qdisc_stab_list, list) {
338 if (memcmp(&stab->szopts, s, sizeof(*s)))
339 continue;
340 if (tsize > 0 && memcmp(stab->data, tab, tsize * sizeof(u16)))
341 continue;
342 stab->refcnt++;
Jarek Poplawski1cfa2662008-08-11 18:11:06 -0700343 spin_unlock_bh(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700344 return stab;
345 }
346
Jarek Poplawski1cfa2662008-08-11 18:11:06 -0700347 spin_unlock_bh(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700348
349 stab = kmalloc(sizeof(*stab) + tsize * sizeof(u16), GFP_KERNEL);
350 if (!stab)
351 return ERR_PTR(-ENOMEM);
352
353 stab->refcnt = 1;
354 stab->szopts = *s;
355 if (tsize > 0)
356 memcpy(stab->data, tab, tsize * sizeof(u16));
357
Jarek Poplawski1cfa2662008-08-11 18:11:06 -0700358 spin_lock_bh(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700359 list_add_tail(&stab->list, &qdisc_stab_list);
Jarek Poplawski1cfa2662008-08-11 18:11:06 -0700360 spin_unlock_bh(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700361
362 return stab;
363}
364
365void qdisc_put_stab(struct qdisc_size_table *tab)
366{
367 if (!tab)
368 return;
369
Jarek Poplawski1cfa2662008-08-11 18:11:06 -0700370 spin_lock_bh(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700371
372 if (--tab->refcnt == 0) {
373 list_del(&tab->list);
374 kfree(tab);
375 }
376
Jarek Poplawski1cfa2662008-08-11 18:11:06 -0700377 spin_unlock_bh(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700378}
379EXPORT_SYMBOL(qdisc_put_stab);
380
381static int qdisc_dump_stab(struct sk_buff *skb, struct qdisc_size_table *stab)
382{
383 struct nlattr *nest;
384
385 nest = nla_nest_start(skb, TCA_STAB);
386 NLA_PUT(skb, TCA_STAB_BASE, sizeof(stab->szopts), &stab->szopts);
387 nla_nest_end(skb, nest);
388
389 return skb->len;
390
391nla_put_failure:
392 return -1;
393}
394
395void qdisc_calculate_pkt_len(struct sk_buff *skb, struct qdisc_size_table *stab)
396{
397 int pkt_len, slot;
398
399 pkt_len = skb->len + stab->szopts.overhead;
400 if (unlikely(!stab->szopts.tsize))
401 goto out;
402
403 slot = pkt_len + stab->szopts.cell_align;
404 if (unlikely(slot < 0))
405 slot = 0;
406
407 slot >>= stab->szopts.cell_log;
408 if (likely(slot < stab->szopts.tsize))
409 pkt_len = stab->data[slot];
410 else
411 pkt_len = stab->data[stab->szopts.tsize - 1] *
412 (slot / stab->szopts.tsize) +
413 stab->data[slot % stab->szopts.tsize];
414
415 pkt_len <<= stab->szopts.size_log;
416out:
417 if (unlikely(pkt_len < 1))
418 pkt_len = 1;
419 qdisc_skb_cb(skb)->pkt_len = pkt_len;
420}
421EXPORT_SYMBOL(qdisc_calculate_pkt_len);
422
Patrick McHardy41794772007-03-16 01:19:15 -0700423static enum hrtimer_restart qdisc_watchdog(struct hrtimer *timer)
424{
425 struct qdisc_watchdog *wd = container_of(timer, struct qdisc_watchdog,
426 timer);
427
428 wd->qdisc->flags &= ~TCQ_F_THROTTLED;
Stephen Hemminger11274e52007-03-22 12:17:42 -0700429 smp_wmb();
David S. Miller8608db02008-08-18 20:51:18 -0700430 __netif_schedule(qdisc_root(wd->qdisc));
Stephen Hemminger19365022007-03-22 12:18:35 -0700431
Patrick McHardy41794772007-03-16 01:19:15 -0700432 return HRTIMER_NORESTART;
433}
434
435void qdisc_watchdog_init(struct qdisc_watchdog *wd, struct Qdisc *qdisc)
436{
437 hrtimer_init(&wd->timer, CLOCK_MONOTONIC, HRTIMER_MODE_ABS);
438 wd->timer.function = qdisc_watchdog;
439 wd->qdisc = qdisc;
440}
441EXPORT_SYMBOL(qdisc_watchdog_init);
442
443void qdisc_watchdog_schedule(struct qdisc_watchdog *wd, psched_time_t expires)
444{
445 ktime_t time;
446
447 wd->qdisc->flags |= TCQ_F_THROTTLED;
448 time = ktime_set(0, 0);
449 time = ktime_add_ns(time, PSCHED_US2NS(expires));
450 hrtimer_start(&wd->timer, time, HRTIMER_MODE_ABS);
451}
452EXPORT_SYMBOL(qdisc_watchdog_schedule);
453
454void qdisc_watchdog_cancel(struct qdisc_watchdog *wd)
455{
456 hrtimer_cancel(&wd->timer);
457 wd->qdisc->flags &= ~TCQ_F_THROTTLED;
458}
459EXPORT_SYMBOL(qdisc_watchdog_cancel);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700460
Adrian Bunka94f7792008-07-22 14:20:11 -0700461static struct hlist_head *qdisc_class_hash_alloc(unsigned int n)
Patrick McHardy6fe1c7a2008-07-05 23:21:31 -0700462{
463 unsigned int size = n * sizeof(struct hlist_head), i;
464 struct hlist_head *h;
465
466 if (size <= PAGE_SIZE)
467 h = kmalloc(size, GFP_KERNEL);
468 else
469 h = (struct hlist_head *)
470 __get_free_pages(GFP_KERNEL, get_order(size));
471
472 if (h != NULL) {
473 for (i = 0; i < n; i++)
474 INIT_HLIST_HEAD(&h[i]);
475 }
476 return h;
477}
478
479static void qdisc_class_hash_free(struct hlist_head *h, unsigned int n)
480{
481 unsigned int size = n * sizeof(struct hlist_head);
482
483 if (size <= PAGE_SIZE)
484 kfree(h);
485 else
486 free_pages((unsigned long)h, get_order(size));
487}
488
489void qdisc_class_hash_grow(struct Qdisc *sch, struct Qdisc_class_hash *clhash)
490{
491 struct Qdisc_class_common *cl;
492 struct hlist_node *n, *next;
493 struct hlist_head *nhash, *ohash;
494 unsigned int nsize, nmask, osize;
495 unsigned int i, h;
496
497 /* Rehash when load factor exceeds 0.75 */
498 if (clhash->hashelems * 4 <= clhash->hashsize * 3)
499 return;
500 nsize = clhash->hashsize * 2;
501 nmask = nsize - 1;
502 nhash = qdisc_class_hash_alloc(nsize);
503 if (nhash == NULL)
504 return;
505
506 ohash = clhash->hash;
507 osize = clhash->hashsize;
508
509 sch_tree_lock(sch);
510 for (i = 0; i < osize; i++) {
511 hlist_for_each_entry_safe(cl, n, next, &ohash[i], hnode) {
512 h = qdisc_class_hash(cl->classid, nmask);
513 hlist_add_head(&cl->hnode, &nhash[h]);
514 }
515 }
516 clhash->hash = nhash;
517 clhash->hashsize = nsize;
518 clhash->hashmask = nmask;
519 sch_tree_unlock(sch);
520
521 qdisc_class_hash_free(ohash, osize);
522}
523EXPORT_SYMBOL(qdisc_class_hash_grow);
524
525int qdisc_class_hash_init(struct Qdisc_class_hash *clhash)
526{
527 unsigned int size = 4;
528
529 clhash->hash = qdisc_class_hash_alloc(size);
530 if (clhash->hash == NULL)
531 return -ENOMEM;
532 clhash->hashsize = size;
533 clhash->hashmask = size - 1;
534 clhash->hashelems = 0;
535 return 0;
536}
537EXPORT_SYMBOL(qdisc_class_hash_init);
538
539void qdisc_class_hash_destroy(struct Qdisc_class_hash *clhash)
540{
541 qdisc_class_hash_free(clhash->hash, clhash->hashsize);
542}
543EXPORT_SYMBOL(qdisc_class_hash_destroy);
544
545void qdisc_class_hash_insert(struct Qdisc_class_hash *clhash,
546 struct Qdisc_class_common *cl)
547{
548 unsigned int h;
549
550 INIT_HLIST_NODE(&cl->hnode);
551 h = qdisc_class_hash(cl->classid, clhash->hashmask);
552 hlist_add_head(&cl->hnode, &clhash->hash[h]);
553 clhash->hashelems++;
554}
555EXPORT_SYMBOL(qdisc_class_hash_insert);
556
557void qdisc_class_hash_remove(struct Qdisc_class_hash *clhash,
558 struct Qdisc_class_common *cl)
559{
560 hlist_del(&cl->hnode);
561 clhash->hashelems--;
562}
563EXPORT_SYMBOL(qdisc_class_hash_remove);
564
Linus Torvalds1da177e2005-04-16 15:20:36 -0700565/* Allocate an unique handle from space managed by kernel */
566
567static u32 qdisc_alloc_handle(struct net_device *dev)
568{
569 int i = 0x10000;
570 static u32 autohandle = TC_H_MAKE(0x80000000U, 0);
571
572 do {
573 autohandle += TC_H_MAKE(0x10000U, 0);
574 if (autohandle == TC_H_MAKE(TC_H_ROOT, 0))
575 autohandle = TC_H_MAKE(0x80000000U, 0);
576 } while (qdisc_lookup(dev, autohandle) && --i > 0);
577
578 return i>0 ? autohandle : 0;
579}
580
David S. Miller99194cf2008-07-17 04:54:10 -0700581/* Attach toplevel qdisc to device queue. */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700582
David S. Miller99194cf2008-07-17 04:54:10 -0700583static struct Qdisc *dev_graft_qdisc(struct netdev_queue *dev_queue,
584 struct Qdisc *qdisc)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700585{
David S. Miller8d50b532008-07-30 02:37:46 -0700586 struct Qdisc *oqdisc = dev_queue->qdisc_sleeping;
David S. Miller53049972008-07-16 03:00:19 -0700587 spinlock_t *root_lock;
David S. Miller53049972008-07-16 03:00:19 -0700588
589 root_lock = qdisc_root_lock(oqdisc);
590 spin_lock_bh(root_lock);
591
David S. Miller8d50b532008-07-30 02:37:46 -0700592 /* Prune old scheduler */
593 if (oqdisc && atomic_read(&oqdisc->refcnt) <= 1)
594 qdisc_reset(oqdisc);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700595
David S. Miller8d50b532008-07-30 02:37:46 -0700596 /* ... and graft new one */
597 if (qdisc == NULL)
598 qdisc = &noop_qdisc;
599 dev_queue->qdisc_sleeping = qdisc;
600 dev_queue->qdisc = &noop_qdisc;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700601
David S. Miller53049972008-07-16 03:00:19 -0700602 spin_unlock_bh(root_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700603
Linus Torvalds1da177e2005-04-16 15:20:36 -0700604 return oqdisc;
605}
606
Patrick McHardy43effa12006-11-29 17:35:48 -0800607void qdisc_tree_decrease_qlen(struct Qdisc *sch, unsigned int n)
608{
Eric Dumazet20fea082007-11-14 01:44:41 -0800609 const struct Qdisc_class_ops *cops;
Patrick McHardy43effa12006-11-29 17:35:48 -0800610 unsigned long cl;
611 u32 parentid;
612
613 if (n == 0)
614 return;
615 while ((parentid = sch->parent)) {
Jarek Poplawski066a3b52008-04-14 15:10:42 -0700616 if (TC_H_MAJ(parentid) == TC_H_MAJ(TC_H_INGRESS))
617 return;
618
David S. Miller5ce2d482008-07-08 17:06:30 -0700619 sch = qdisc_lookup(qdisc_dev(sch), TC_H_MAJ(parentid));
Patrick McHardyffc8fef2007-07-30 17:11:50 -0700620 if (sch == NULL) {
621 WARN_ON(parentid != TC_H_ROOT);
622 return;
623 }
Patrick McHardy43effa12006-11-29 17:35:48 -0800624 cops = sch->ops->cl_ops;
625 if (cops->qlen_notify) {
626 cl = cops->get(sch, parentid);
627 cops->qlen_notify(sch, cl);
628 cops->put(sch, cl);
629 }
630 sch->q.qlen -= n;
631 }
632}
633EXPORT_SYMBOL(qdisc_tree_decrease_qlen);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700634
David S. Miller99194cf2008-07-17 04:54:10 -0700635static void notify_and_destroy(struct sk_buff *skb, struct nlmsghdr *n, u32 clid,
636 struct Qdisc *old, struct Qdisc *new)
637{
638 if (new || old)
639 qdisc_notify(skb, n, clid, old, new);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700640
David S. Miller99194cf2008-07-17 04:54:10 -0700641 if (old) {
Jarek Poplawski3a76e372008-08-17 22:02:11 -0700642 sch_tree_lock(old);
David S. Miller99194cf2008-07-17 04:54:10 -0700643 qdisc_destroy(old);
Jarek Poplawski3a76e372008-08-17 22:02:11 -0700644 sch_tree_unlock(old);
David S. Miller99194cf2008-07-17 04:54:10 -0700645 }
646}
647
648/* Graft qdisc "new" to class "classid" of qdisc "parent" or
649 * to device "dev".
650 *
651 * When appropriate send a netlink notification using 'skb'
652 * and "n".
653 *
654 * On success, destroy old qdisc.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700655 */
656
657static int qdisc_graft(struct net_device *dev, struct Qdisc *parent,
David S. Miller99194cf2008-07-17 04:54:10 -0700658 struct sk_buff *skb, struct nlmsghdr *n, u32 classid,
659 struct Qdisc *new, struct Qdisc *old)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700660{
David S. Miller99194cf2008-07-17 04:54:10 -0700661 struct Qdisc *q = old;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700662 int err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700663
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900664 if (parent == NULL) {
David S. Miller99194cf2008-07-17 04:54:10 -0700665 unsigned int i, num_q, ingress;
666
667 ingress = 0;
668 num_q = dev->num_tx_queues;
David S. Miller8d50b532008-07-30 02:37:46 -0700669 if ((q && q->flags & TCQ_F_INGRESS) ||
670 (new && new->flags & TCQ_F_INGRESS)) {
David S. Miller99194cf2008-07-17 04:54:10 -0700671 num_q = 1;
672 ingress = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700673 }
David S. Miller99194cf2008-07-17 04:54:10 -0700674
675 if (dev->flags & IFF_UP)
676 dev_deactivate(dev);
677
678 for (i = 0; i < num_q; i++) {
679 struct netdev_queue *dev_queue = &dev->rx_queue;
680
681 if (!ingress)
682 dev_queue = netdev_get_tx_queue(dev, i);
683
David S. Miller8d50b532008-07-30 02:37:46 -0700684 old = dev_graft_qdisc(dev_queue, new);
685 if (new && i > 0)
686 atomic_inc(&new->refcnt);
687
David S. Miller99194cf2008-07-17 04:54:10 -0700688 notify_and_destroy(skb, n, classid, old, new);
689 }
690
691 if (dev->flags & IFF_UP)
692 dev_activate(dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700693 } else {
Eric Dumazet20fea082007-11-14 01:44:41 -0800694 const struct Qdisc_class_ops *cops = parent->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700695
696 err = -EINVAL;
697
698 if (cops) {
699 unsigned long cl = cops->get(parent, classid);
700 if (cl) {
David S. Miller99194cf2008-07-17 04:54:10 -0700701 err = cops->graft(parent, cl, new, &old);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700702 cops->put(parent, cl);
703 }
704 }
David S. Miller99194cf2008-07-17 04:54:10 -0700705 if (!err)
706 notify_and_destroy(skb, n, classid, old, new);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700707 }
708 return err;
709}
710
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700711/* lockdep annotation is needed for ingress; egress gets it only for name */
712static struct lock_class_key qdisc_tx_lock;
713static struct lock_class_key qdisc_rx_lock;
714
Linus Torvalds1da177e2005-04-16 15:20:36 -0700715/*
716 Allocate and initialize new qdisc.
717
718 Parameters are passed via opt.
719 */
720
721static struct Qdisc *
David S. Millerbb949fb2008-07-08 16:55:56 -0700722qdisc_create(struct net_device *dev, struct netdev_queue *dev_queue,
723 u32 parent, u32 handle, struct nlattr **tca, int *errp)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700724{
725 int err;
Patrick McHardy1e904742008-01-22 22:11:17 -0800726 struct nlattr *kind = tca[TCA_KIND];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700727 struct Qdisc *sch;
728 struct Qdisc_ops *ops;
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700729 struct qdisc_size_table *stab;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700730
731 ops = qdisc_lookup_ops(kind);
732#ifdef CONFIG_KMOD
733 if (ops == NULL && kind != NULL) {
734 char name[IFNAMSIZ];
Patrick McHardy1e904742008-01-22 22:11:17 -0800735 if (nla_strlcpy(name, kind, IFNAMSIZ) < IFNAMSIZ) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700736 /* We dropped the RTNL semaphore in order to
737 * perform the module load. So, even if we
738 * succeeded in loading the module we have to
739 * tell the caller to replay the request. We
740 * indicate this using -EAGAIN.
741 * We replay the request because the device may
742 * go away in the mean time.
743 */
744 rtnl_unlock();
745 request_module("sch_%s", name);
746 rtnl_lock();
747 ops = qdisc_lookup_ops(kind);
748 if (ops != NULL) {
749 /* We will try again qdisc_lookup_ops,
750 * so don't keep a reference.
751 */
752 module_put(ops->owner);
753 err = -EAGAIN;
754 goto err_out;
755 }
756 }
757 }
758#endif
759
Jamal Hadi Salimb9e2cc02006-08-03 16:36:51 -0700760 err = -ENOENT;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700761 if (ops == NULL)
762 goto err_out;
763
David S. Miller5ce2d482008-07-08 17:06:30 -0700764 sch = qdisc_alloc(dev_queue, ops);
Thomas Graf3d54b822005-07-05 14:15:09 -0700765 if (IS_ERR(sch)) {
766 err = PTR_ERR(sch);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700767 goto err_out2;
Thomas Graf3d54b822005-07-05 14:15:09 -0700768 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700769
Patrick McHardyffc8fef2007-07-30 17:11:50 -0700770 sch->parent = parent;
771
Thomas Graf3d54b822005-07-05 14:15:09 -0700772 if (handle == TC_H_INGRESS) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700773 sch->flags |= TCQ_F_INGRESS;
Thomas Graf3d54b822005-07-05 14:15:09 -0700774 handle = TC_H_MAKE(TC_H_INGRESS, 0);
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700775 lockdep_set_class(qdisc_lock(sch), &qdisc_rx_lock);
Patrick McHardyfd44de72007-04-16 17:07:08 -0700776 } else {
Patrick McHardyfd44de72007-04-16 17:07:08 -0700777 if (handle == 0) {
778 handle = qdisc_alloc_handle(dev);
779 err = -ENOMEM;
780 if (handle == 0)
781 goto err_out3;
782 }
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700783 lockdep_set_class(qdisc_lock(sch), &qdisc_tx_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700784 }
785
Thomas Graf3d54b822005-07-05 14:15:09 -0700786 sch->handle = handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700787
Patrick McHardy1e904742008-01-22 22:11:17 -0800788 if (!ops->init || (err = ops->init(sch, tca[TCA_OPTIONS])) == 0) {
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700789 if (tca[TCA_STAB]) {
790 stab = qdisc_get_stab(tca[TCA_STAB]);
791 if (IS_ERR(stab)) {
792 err = PTR_ERR(stab);
793 goto err_out3;
794 }
795 sch->stab = stab;
796 }
Patrick McHardy1e904742008-01-22 22:11:17 -0800797 if (tca[TCA_RATE]) {
Thomas Graf023e09a2005-07-05 14:15:53 -0700798 err = gen_new_estimator(&sch->bstats, &sch->rate_est,
David S. Miller7698b4f2008-07-16 01:42:40 -0700799 qdisc_root_lock(sch),
Patrick McHardy1e904742008-01-22 22:11:17 -0800800 tca[TCA_RATE]);
Thomas Graf023e09a2005-07-05 14:15:53 -0700801 if (err) {
802 /*
803 * Any broken qdiscs that would require
804 * a ops->reset() here? The qdisc was never
805 * in action so it shouldn't be necessary.
806 */
807 if (ops->destroy)
808 ops->destroy(sch);
809 goto err_out3;
810 }
811 }
David S. Milleree7af822008-08-06 23:35:59 -0700812 if ((parent != TC_H_ROOT) && !(sch->flags & TCQ_F_INGRESS))
David S. Miller827ebd62008-08-07 20:26:40 -0700813 list_add_tail(&sch->list, &dev_queue->qdisc_sleeping->list);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700814
Linus Torvalds1da177e2005-04-16 15:20:36 -0700815 return sch;
816 }
817err_out3:
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700818 qdisc_put_stab(sch->stab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700819 dev_put(dev);
Thomas Graf3d54b822005-07-05 14:15:09 -0700820 kfree((char *) sch - sch->padded);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700821err_out2:
822 module_put(ops->owner);
823err_out:
824 *errp = err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700825 return NULL;
826}
827
Patrick McHardy1e904742008-01-22 22:11:17 -0800828static int qdisc_change(struct Qdisc *sch, struct nlattr **tca)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700829{
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700830 struct qdisc_size_table *stab = NULL;
831 int err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700832
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700833 if (tca[TCA_OPTIONS]) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700834 if (sch->ops->change == NULL)
835 return -EINVAL;
Patrick McHardy1e904742008-01-22 22:11:17 -0800836 err = sch->ops->change(sch, tca[TCA_OPTIONS]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700837 if (err)
838 return err;
839 }
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700840
841 if (tca[TCA_STAB]) {
842 stab = qdisc_get_stab(tca[TCA_STAB]);
843 if (IS_ERR(stab))
844 return PTR_ERR(stab);
845 }
846
847 qdisc_put_stab(sch->stab);
848 sch->stab = stab;
849
Patrick McHardy1e904742008-01-22 22:11:17 -0800850 if (tca[TCA_RATE])
Linus Torvalds1da177e2005-04-16 15:20:36 -0700851 gen_replace_estimator(&sch->bstats, &sch->rate_est,
David S. Miller7698b4f2008-07-16 01:42:40 -0700852 qdisc_root_lock(sch), tca[TCA_RATE]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700853 return 0;
854}
855
856struct check_loop_arg
857{
858 struct qdisc_walker w;
859 struct Qdisc *p;
860 int depth;
861};
862
863static int check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w);
864
865static int check_loop(struct Qdisc *q, struct Qdisc *p, int depth)
866{
867 struct check_loop_arg arg;
868
869 if (q->ops->cl_ops == NULL)
870 return 0;
871
872 arg.w.stop = arg.w.skip = arg.w.count = 0;
873 arg.w.fn = check_loop_fn;
874 arg.depth = depth;
875 arg.p = p;
876 q->ops->cl_ops->walk(q, &arg.w);
877 return arg.w.stop ? -ELOOP : 0;
878}
879
880static int
881check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w)
882{
883 struct Qdisc *leaf;
Eric Dumazet20fea082007-11-14 01:44:41 -0800884 const struct Qdisc_class_ops *cops = q->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700885 struct check_loop_arg *arg = (struct check_loop_arg *)w;
886
887 leaf = cops->leaf(q, cl);
888 if (leaf) {
889 if (leaf == arg->p || arg->depth > 7)
890 return -ELOOP;
891 return check_loop(leaf, arg->p, arg->depth + 1);
892 }
893 return 0;
894}
895
896/*
897 * Delete/get qdisc.
898 */
899
900static int tc_get_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
901{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900902 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700903 struct tcmsg *tcm = NLMSG_DATA(n);
Patrick McHardy1e904742008-01-22 22:11:17 -0800904 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700905 struct net_device *dev;
906 u32 clid = tcm->tcm_parent;
907 struct Qdisc *q = NULL;
908 struct Qdisc *p = NULL;
909 int err;
910
Denis V. Lunevb8542722007-12-01 00:21:31 +1100911 if (net != &init_net)
912 return -EINVAL;
913
Eric W. Biederman881d9662007-09-17 11:56:21 -0700914 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700915 return -ENODEV;
916
Patrick McHardy1e904742008-01-22 22:11:17 -0800917 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
918 if (err < 0)
919 return err;
920
Linus Torvalds1da177e2005-04-16 15:20:36 -0700921 if (clid) {
922 if (clid != TC_H_ROOT) {
923 if (TC_H_MAJ(clid) != TC_H_MAJ(TC_H_INGRESS)) {
924 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
925 return -ENOENT;
926 q = qdisc_leaf(p, clid);
927 } else { /* ingress */
David S. Miller8123b422008-08-08 23:23:39 -0700928 q = dev->rx_queue.qdisc_sleeping;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900929 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700930 } else {
David S. Millere8a04642008-07-17 00:34:19 -0700931 struct netdev_queue *dev_queue;
932 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Millerb0e1e642008-07-08 17:42:10 -0700933 q = dev_queue->qdisc_sleeping;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700934 }
935 if (!q)
936 return -ENOENT;
937
938 if (tcm->tcm_handle && q->handle != tcm->tcm_handle)
939 return -EINVAL;
940 } else {
941 if ((q = qdisc_lookup(dev, tcm->tcm_handle)) == NULL)
942 return -ENOENT;
943 }
944
Patrick McHardy1e904742008-01-22 22:11:17 -0800945 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700946 return -EINVAL;
947
948 if (n->nlmsg_type == RTM_DELQDISC) {
949 if (!clid)
950 return -EINVAL;
951 if (q->handle == 0)
952 return -ENOENT;
David S. Miller99194cf2008-07-17 04:54:10 -0700953 if ((err = qdisc_graft(dev, p, skb, n, clid, NULL, q)) != 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700954 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700955 } else {
956 qdisc_notify(skb, n, clid, NULL, q);
957 }
958 return 0;
959}
960
961/*
962 Create/change qdisc.
963 */
964
965static int tc_modify_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
966{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900967 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700968 struct tcmsg *tcm;
Patrick McHardy1e904742008-01-22 22:11:17 -0800969 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700970 struct net_device *dev;
971 u32 clid;
972 struct Qdisc *q, *p;
973 int err;
974
Denis V. Lunevb8542722007-12-01 00:21:31 +1100975 if (net != &init_net)
976 return -EINVAL;
977
Linus Torvalds1da177e2005-04-16 15:20:36 -0700978replay:
979 /* Reinit, just in case something touches this. */
980 tcm = NLMSG_DATA(n);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700981 clid = tcm->tcm_parent;
982 q = p = NULL;
983
Eric W. Biederman881d9662007-09-17 11:56:21 -0700984 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700985 return -ENODEV;
986
Patrick McHardy1e904742008-01-22 22:11:17 -0800987 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
988 if (err < 0)
989 return err;
990
Linus Torvalds1da177e2005-04-16 15:20:36 -0700991 if (clid) {
992 if (clid != TC_H_ROOT) {
993 if (clid != TC_H_INGRESS) {
994 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
995 return -ENOENT;
996 q = qdisc_leaf(p, clid);
997 } else { /*ingress */
David S. Miller8123b422008-08-08 23:23:39 -0700998 q = dev->rx_queue.qdisc_sleeping;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700999 }
1000 } else {
David S. Millere8a04642008-07-17 00:34:19 -07001001 struct netdev_queue *dev_queue;
1002 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Millerb0e1e642008-07-08 17:42:10 -07001003 q = dev_queue->qdisc_sleeping;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001004 }
1005
1006 /* It may be default qdisc, ignore it */
1007 if (q && q->handle == 0)
1008 q = NULL;
1009
1010 if (!q || !tcm->tcm_handle || q->handle != tcm->tcm_handle) {
1011 if (tcm->tcm_handle) {
1012 if (q && !(n->nlmsg_flags&NLM_F_REPLACE))
1013 return -EEXIST;
1014 if (TC_H_MIN(tcm->tcm_handle))
1015 return -EINVAL;
1016 if ((q = qdisc_lookup(dev, tcm->tcm_handle)) == NULL)
1017 goto create_n_graft;
1018 if (n->nlmsg_flags&NLM_F_EXCL)
1019 return -EEXIST;
Patrick McHardy1e904742008-01-22 22:11:17 -08001020 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001021 return -EINVAL;
1022 if (q == p ||
1023 (p && check_loop(q, p, 0)))
1024 return -ELOOP;
1025 atomic_inc(&q->refcnt);
1026 goto graft;
1027 } else {
1028 if (q == NULL)
1029 goto create_n_graft;
1030
1031 /* This magic test requires explanation.
1032 *
1033 * We know, that some child q is already
1034 * attached to this parent and have choice:
1035 * either to change it or to create/graft new one.
1036 *
1037 * 1. We are allowed to create/graft only
1038 * if CREATE and REPLACE flags are set.
1039 *
1040 * 2. If EXCL is set, requestor wanted to say,
1041 * that qdisc tcm_handle is not expected
1042 * to exist, so that we choose create/graft too.
1043 *
1044 * 3. The last case is when no flags are set.
1045 * Alas, it is sort of hole in API, we
1046 * cannot decide what to do unambiguously.
1047 * For now we select create/graft, if
1048 * user gave KIND, which does not match existing.
1049 */
1050 if ((n->nlmsg_flags&NLM_F_CREATE) &&
1051 (n->nlmsg_flags&NLM_F_REPLACE) &&
1052 ((n->nlmsg_flags&NLM_F_EXCL) ||
Patrick McHardy1e904742008-01-22 22:11:17 -08001053 (tca[TCA_KIND] &&
1054 nla_strcmp(tca[TCA_KIND], q->ops->id))))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001055 goto create_n_graft;
1056 }
1057 }
1058 } else {
1059 if (!tcm->tcm_handle)
1060 return -EINVAL;
1061 q = qdisc_lookup(dev, tcm->tcm_handle);
1062 }
1063
1064 /* Change qdisc parameters */
1065 if (q == NULL)
1066 return -ENOENT;
1067 if (n->nlmsg_flags&NLM_F_EXCL)
1068 return -EEXIST;
Patrick McHardy1e904742008-01-22 22:11:17 -08001069 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001070 return -EINVAL;
1071 err = qdisc_change(q, tca);
1072 if (err == 0)
1073 qdisc_notify(skb, n, clid, NULL, q);
1074 return err;
1075
1076create_n_graft:
1077 if (!(n->nlmsg_flags&NLM_F_CREATE))
1078 return -ENOENT;
1079 if (clid == TC_H_INGRESS)
David S. Millerbb949fb2008-07-08 16:55:56 -07001080 q = qdisc_create(dev, &dev->rx_queue,
1081 tcm->tcm_parent, tcm->tcm_parent,
Patrick McHardyffc8fef2007-07-30 17:11:50 -07001082 tca, &err);
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001083 else
David S. Millere8a04642008-07-17 00:34:19 -07001084 q = qdisc_create(dev, netdev_get_tx_queue(dev, 0),
David S. Millerbb949fb2008-07-08 16:55:56 -07001085 tcm->tcm_parent, tcm->tcm_handle,
Patrick McHardyffc8fef2007-07-30 17:11:50 -07001086 tca, &err);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001087 if (q == NULL) {
1088 if (err == -EAGAIN)
1089 goto replay;
1090 return err;
1091 }
1092
1093graft:
1094 if (1) {
David S. Miller53049972008-07-16 03:00:19 -07001095 spinlock_t *root_lock;
1096
David S. Miller99194cf2008-07-17 04:54:10 -07001097 err = qdisc_graft(dev, p, skb, n, clid, q, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001098 if (err) {
1099 if (q) {
David S. Miller53049972008-07-16 03:00:19 -07001100 root_lock = qdisc_root_lock(q);
1101 spin_lock_bh(root_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001102 qdisc_destroy(q);
David S. Miller53049972008-07-16 03:00:19 -07001103 spin_unlock_bh(root_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001104 }
1105 return err;
1106 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001107 }
1108 return 0;
1109}
1110
1111static int tc_fill_qdisc(struct sk_buff *skb, struct Qdisc *q, u32 clid,
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001112 u32 pid, u32 seq, u16 flags, int event)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001113{
1114 struct tcmsg *tcm;
1115 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001116 unsigned char *b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001117 struct gnet_dump d;
1118
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001119 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001120 tcm = NLMSG_DATA(nlh);
1121 tcm->tcm_family = AF_UNSPEC;
Patrick McHardy9ef1d4c2005-06-28 12:55:30 -07001122 tcm->tcm__pad1 = 0;
1123 tcm->tcm__pad2 = 0;
David S. Miller5ce2d482008-07-08 17:06:30 -07001124 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001125 tcm->tcm_parent = clid;
1126 tcm->tcm_handle = q->handle;
1127 tcm->tcm_info = atomic_read(&q->refcnt);
Patrick McHardy57e1c482008-01-23 20:34:28 -08001128 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001129 if (q->ops->dump && q->ops->dump(q, skb) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001130 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001131 q->qstats.qlen = q->q.qlen;
1132
Jussi Kivilinna175f9c12008-07-20 00:08:47 -07001133 if (q->stab && qdisc_dump_stab(skb, q->stab) < 0)
1134 goto nla_put_failure;
1135
Linus Torvalds1da177e2005-04-16 15:20:36 -07001136 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS,
David S. Miller7698b4f2008-07-16 01:42:40 -07001137 TCA_XSTATS, qdisc_root_lock(q), &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001138 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001139
1140 if (q->ops->dump_stats && q->ops->dump_stats(q, &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001141 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001142
1143 if (gnet_stats_copy_basic(&d, &q->bstats) < 0 ||
Linus Torvalds1da177e2005-04-16 15:20:36 -07001144 gnet_stats_copy_rate_est(&d, &q->rate_est) < 0 ||
Linus Torvalds1da177e2005-04-16 15:20:36 -07001145 gnet_stats_copy_queue(&d, &q->qstats) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001146 goto nla_put_failure;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001147
Linus Torvalds1da177e2005-04-16 15:20:36 -07001148 if (gnet_stats_finish_copy(&d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001149 goto nla_put_failure;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001150
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001151 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001152 return skb->len;
1153
1154nlmsg_failure:
Patrick McHardy1e904742008-01-22 22:11:17 -08001155nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -07001156 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001157 return -1;
1158}
1159
1160static int qdisc_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1161 u32 clid, struct Qdisc *old, struct Qdisc *new)
1162{
1163 struct sk_buff *skb;
1164 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1165
1166 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1167 if (!skb)
1168 return -ENOBUFS;
1169
1170 if (old && old->handle) {
1171 if (tc_fill_qdisc(skb, old, clid, pid, n->nlmsg_seq, 0, RTM_DELQDISC) < 0)
1172 goto err_out;
1173 }
1174 if (new) {
1175 if (tc_fill_qdisc(skb, new, clid, pid, n->nlmsg_seq, old ? NLM_F_REPLACE : 0, RTM_NEWQDISC) < 0)
1176 goto err_out;
1177 }
1178
1179 if (skb->len)
Denis V. Lunev97c53ca2007-11-19 22:26:51 -08001180 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001181
1182err_out:
1183 kfree_skb(skb);
1184 return -EINVAL;
1185}
1186
David S. Miller30723672008-07-18 22:50:15 -07001187static bool tc_qdisc_dump_ignore(struct Qdisc *q)
1188{
1189 return (q->flags & TCQ_F_BUILTIN) ? true : false;
1190}
1191
1192static int tc_dump_qdisc_root(struct Qdisc *root, struct sk_buff *skb,
1193 struct netlink_callback *cb,
1194 int *q_idx_p, int s_q_idx)
1195{
1196 int ret = 0, q_idx = *q_idx_p;
1197 struct Qdisc *q;
1198
1199 if (!root)
1200 return 0;
1201
1202 q = root;
1203 if (q_idx < s_q_idx) {
1204 q_idx++;
1205 } else {
1206 if (!tc_qdisc_dump_ignore(q) &&
1207 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1208 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1209 goto done;
1210 q_idx++;
1211 }
1212 list_for_each_entry(q, &root->list, list) {
1213 if (q_idx < s_q_idx) {
1214 q_idx++;
1215 continue;
1216 }
1217 if (!tc_qdisc_dump_ignore(q) &&
1218 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1219 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1220 goto done;
1221 q_idx++;
1222 }
1223
1224out:
1225 *q_idx_p = q_idx;
1226 return ret;
1227done:
1228 ret = -1;
1229 goto out;
1230}
1231
Linus Torvalds1da177e2005-04-16 15:20:36 -07001232static int tc_dump_qdisc(struct sk_buff *skb, struct netlink_callback *cb)
1233{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001234 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001235 int idx, q_idx;
1236 int s_idx, s_q_idx;
1237 struct net_device *dev;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001238
Denis V. Lunevb8542722007-12-01 00:21:31 +11001239 if (net != &init_net)
1240 return 0;
1241
Linus Torvalds1da177e2005-04-16 15:20:36 -07001242 s_idx = cb->args[0];
1243 s_q_idx = q_idx = cb->args[1];
1244 read_lock(&dev_base_lock);
Pavel Emelianov7562f872007-05-03 15:13:45 -07001245 idx = 0;
Eric W. Biederman881d9662007-09-17 11:56:21 -07001246 for_each_netdev(&init_net, dev) {
David S. Miller30723672008-07-18 22:50:15 -07001247 struct netdev_queue *dev_queue;
1248
Linus Torvalds1da177e2005-04-16 15:20:36 -07001249 if (idx < s_idx)
Pavel Emelianov7562f872007-05-03 15:13:45 -07001250 goto cont;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001251 if (idx > s_idx)
1252 s_q_idx = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001253 q_idx = 0;
David S. Miller30723672008-07-18 22:50:15 -07001254
1255 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Miller827ebd62008-08-07 20:26:40 -07001256 if (tc_dump_qdisc_root(dev_queue->qdisc_sleeping, skb, cb, &q_idx, s_q_idx) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001257 goto done;
1258
1259 dev_queue = &dev->rx_queue;
David S. Miller827ebd62008-08-07 20:26:40 -07001260 if (tc_dump_qdisc_root(dev_queue->qdisc_sleeping, skb, cb, &q_idx, s_q_idx) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001261 goto done;
1262
Pavel Emelianov7562f872007-05-03 15:13:45 -07001263cont:
1264 idx++;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001265 }
1266
1267done:
1268 read_unlock(&dev_base_lock);
1269
1270 cb->args[0] = idx;
1271 cb->args[1] = q_idx;
1272
1273 return skb->len;
1274}
1275
1276
1277
1278/************************************************
1279 * Traffic classes manipulation. *
1280 ************************************************/
1281
1282
1283
1284static int tc_ctl_tclass(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
1285{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001286 struct net *net = sock_net(skb->sk);
David S. Millerb0e1e642008-07-08 17:42:10 -07001287 struct netdev_queue *dev_queue;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001288 struct tcmsg *tcm = NLMSG_DATA(n);
Patrick McHardy1e904742008-01-22 22:11:17 -08001289 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -07001290 struct net_device *dev;
1291 struct Qdisc *q = NULL;
Eric Dumazet20fea082007-11-14 01:44:41 -08001292 const struct Qdisc_class_ops *cops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001293 unsigned long cl = 0;
1294 unsigned long new_cl;
1295 u32 pid = tcm->tcm_parent;
1296 u32 clid = tcm->tcm_handle;
1297 u32 qid = TC_H_MAJ(clid);
1298 int err;
1299
Denis V. Lunevb8542722007-12-01 00:21:31 +11001300 if (net != &init_net)
1301 return -EINVAL;
1302
Eric W. Biederman881d9662007-09-17 11:56:21 -07001303 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001304 return -ENODEV;
1305
Patrick McHardy1e904742008-01-22 22:11:17 -08001306 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1307 if (err < 0)
1308 return err;
1309
Linus Torvalds1da177e2005-04-16 15:20:36 -07001310 /*
1311 parent == TC_H_UNSPEC - unspecified parent.
1312 parent == TC_H_ROOT - class is root, which has no parent.
1313 parent == X:0 - parent is root class.
1314 parent == X:Y - parent is a node in hierarchy.
1315 parent == 0:Y - parent is X:Y, where X:0 is qdisc.
1316
1317 handle == 0:0 - generate handle from kernel pool.
1318 handle == 0:Y - class is X:Y, where X:0 is qdisc.
1319 handle == X:Y - clear.
1320 handle == X:0 - root class.
1321 */
1322
1323 /* Step 1. Determine qdisc handle X:0 */
1324
David S. Millere8a04642008-07-17 00:34:19 -07001325 dev_queue = netdev_get_tx_queue(dev, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001326 if (pid != TC_H_ROOT) {
1327 u32 qid1 = TC_H_MAJ(pid);
1328
1329 if (qid && qid1) {
1330 /* If both majors are known, they must be identical. */
1331 if (qid != qid1)
1332 return -EINVAL;
1333 } else if (qid1) {
1334 qid = qid1;
1335 } else if (qid == 0)
David S. Millerb0e1e642008-07-08 17:42:10 -07001336 qid = dev_queue->qdisc_sleeping->handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001337
1338 /* Now qid is genuine qdisc handle consistent
1339 both with parent and child.
1340
1341 TC_H_MAJ(pid) still may be unspecified, complete it now.
1342 */
1343 if (pid)
1344 pid = TC_H_MAKE(qid, pid);
1345 } else {
1346 if (qid == 0)
David S. Millerb0e1e642008-07-08 17:42:10 -07001347 qid = dev_queue->qdisc_sleeping->handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001348 }
1349
1350 /* OK. Locate qdisc */
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001351 if ((q = qdisc_lookup(dev, qid)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001352 return -ENOENT;
1353
1354 /* An check that it supports classes */
1355 cops = q->ops->cl_ops;
1356 if (cops == NULL)
1357 return -EINVAL;
1358
1359 /* Now try to get class */
1360 if (clid == 0) {
1361 if (pid == TC_H_ROOT)
1362 clid = qid;
1363 } else
1364 clid = TC_H_MAKE(qid, clid);
1365
1366 if (clid)
1367 cl = cops->get(q, clid);
1368
1369 if (cl == 0) {
1370 err = -ENOENT;
1371 if (n->nlmsg_type != RTM_NEWTCLASS || !(n->nlmsg_flags&NLM_F_CREATE))
1372 goto out;
1373 } else {
1374 switch (n->nlmsg_type) {
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001375 case RTM_NEWTCLASS:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001376 err = -EEXIST;
1377 if (n->nlmsg_flags&NLM_F_EXCL)
1378 goto out;
1379 break;
1380 case RTM_DELTCLASS:
1381 err = cops->delete(q, cl);
1382 if (err == 0)
1383 tclass_notify(skb, n, q, cl, RTM_DELTCLASS);
1384 goto out;
1385 case RTM_GETTCLASS:
1386 err = tclass_notify(skb, n, q, cl, RTM_NEWTCLASS);
1387 goto out;
1388 default:
1389 err = -EINVAL;
1390 goto out;
1391 }
1392 }
1393
1394 new_cl = cl;
1395 err = cops->change(q, clid, pid, tca, &new_cl);
1396 if (err == 0)
1397 tclass_notify(skb, n, q, new_cl, RTM_NEWTCLASS);
1398
1399out:
1400 if (cl)
1401 cops->put(q, cl);
1402
1403 return err;
1404}
1405
1406
1407static int tc_fill_tclass(struct sk_buff *skb, struct Qdisc *q,
1408 unsigned long cl,
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001409 u32 pid, u32 seq, u16 flags, int event)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001410{
1411 struct tcmsg *tcm;
1412 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001413 unsigned char *b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001414 struct gnet_dump d;
Eric Dumazet20fea082007-11-14 01:44:41 -08001415 const struct Qdisc_class_ops *cl_ops = q->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001416
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001417 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001418 tcm = NLMSG_DATA(nlh);
1419 tcm->tcm_family = AF_UNSPEC;
David S. Miller5ce2d482008-07-08 17:06:30 -07001420 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001421 tcm->tcm_parent = q->handle;
1422 tcm->tcm_handle = q->handle;
1423 tcm->tcm_info = 0;
Patrick McHardy57e1c482008-01-23 20:34:28 -08001424 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001425 if (cl_ops->dump && cl_ops->dump(q, cl, skb, tcm) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001426 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001427
1428 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS,
David S. Miller7698b4f2008-07-16 01:42:40 -07001429 TCA_XSTATS, qdisc_root_lock(q), &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001430 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001431
1432 if (cl_ops->dump_stats && cl_ops->dump_stats(q, cl, &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001433 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001434
1435 if (gnet_stats_finish_copy(&d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001436 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001437
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001438 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001439 return skb->len;
1440
1441nlmsg_failure:
Patrick McHardy1e904742008-01-22 22:11:17 -08001442nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -07001443 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001444 return -1;
1445}
1446
1447static int tclass_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1448 struct Qdisc *q, unsigned long cl, int event)
1449{
1450 struct sk_buff *skb;
1451 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1452
1453 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1454 if (!skb)
1455 return -ENOBUFS;
1456
1457 if (tc_fill_tclass(skb, q, cl, pid, n->nlmsg_seq, 0, event) < 0) {
1458 kfree_skb(skb);
1459 return -EINVAL;
1460 }
1461
Denis V. Lunev97c53ca2007-11-19 22:26:51 -08001462 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001463}
1464
1465struct qdisc_dump_args
1466{
1467 struct qdisc_walker w;
1468 struct sk_buff *skb;
1469 struct netlink_callback *cb;
1470};
1471
1472static int qdisc_class_dump(struct Qdisc *q, unsigned long cl, struct qdisc_walker *arg)
1473{
1474 struct qdisc_dump_args *a = (struct qdisc_dump_args *)arg;
1475
1476 return tc_fill_tclass(a->skb, q, cl, NETLINK_CB(a->cb->skb).pid,
1477 a->cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWTCLASS);
1478}
1479
David S. Miller30723672008-07-18 22:50:15 -07001480static int tc_dump_tclass_qdisc(struct Qdisc *q, struct sk_buff *skb,
1481 struct tcmsg *tcm, struct netlink_callback *cb,
1482 int *t_p, int s_t)
1483{
1484 struct qdisc_dump_args arg;
1485
1486 if (tc_qdisc_dump_ignore(q) ||
1487 *t_p < s_t || !q->ops->cl_ops ||
1488 (tcm->tcm_parent &&
1489 TC_H_MAJ(tcm->tcm_parent) != q->handle)) {
1490 (*t_p)++;
1491 return 0;
1492 }
1493 if (*t_p > s_t)
1494 memset(&cb->args[1], 0, sizeof(cb->args)-sizeof(cb->args[0]));
1495 arg.w.fn = qdisc_class_dump;
1496 arg.skb = skb;
1497 arg.cb = cb;
1498 arg.w.stop = 0;
1499 arg.w.skip = cb->args[1];
1500 arg.w.count = 0;
1501 q->ops->cl_ops->walk(q, &arg.w);
1502 cb->args[1] = arg.w.count;
1503 if (arg.w.stop)
1504 return -1;
1505 (*t_p)++;
1506 return 0;
1507}
1508
1509static int tc_dump_tclass_root(struct Qdisc *root, struct sk_buff *skb,
1510 struct tcmsg *tcm, struct netlink_callback *cb,
1511 int *t_p, int s_t)
1512{
1513 struct Qdisc *q;
1514
1515 if (!root)
1516 return 0;
1517
1518 if (tc_dump_tclass_qdisc(root, skb, tcm, cb, t_p, s_t) < 0)
1519 return -1;
1520
1521 list_for_each_entry(q, &root->list, list) {
1522 if (tc_dump_tclass_qdisc(q, skb, tcm, cb, t_p, s_t) < 0)
1523 return -1;
1524 }
1525
1526 return 0;
1527}
1528
Linus Torvalds1da177e2005-04-16 15:20:36 -07001529static int tc_dump_tclass(struct sk_buff *skb, struct netlink_callback *cb)
1530{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001531 struct tcmsg *tcm = (struct tcmsg*)NLMSG_DATA(cb->nlh);
David S. Miller30723672008-07-18 22:50:15 -07001532 struct net *net = sock_net(skb->sk);
1533 struct netdev_queue *dev_queue;
1534 struct net_device *dev;
1535 int t, s_t;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001536
Denis V. Lunevb8542722007-12-01 00:21:31 +11001537 if (net != &init_net)
1538 return 0;
1539
Linus Torvalds1da177e2005-04-16 15:20:36 -07001540 if (cb->nlh->nlmsg_len < NLMSG_LENGTH(sizeof(*tcm)))
1541 return 0;
Eric W. Biederman881d9662007-09-17 11:56:21 -07001542 if ((dev = dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001543 return 0;
1544
1545 s_t = cb->args[0];
1546 t = 0;
1547
David S. Miller30723672008-07-18 22:50:15 -07001548 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Miller8123b422008-08-08 23:23:39 -07001549 if (tc_dump_tclass_root(dev_queue->qdisc_sleeping, skb, tcm, cb, &t, s_t) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001550 goto done;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001551
David S. Miller30723672008-07-18 22:50:15 -07001552 dev_queue = &dev->rx_queue;
David S. Miller8123b422008-08-08 23:23:39 -07001553 if (tc_dump_tclass_root(dev_queue->qdisc_sleeping, skb, tcm, cb, &t, s_t) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001554 goto done;
1555
1556done:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001557 cb->args[0] = t;
1558
1559 dev_put(dev);
1560 return skb->len;
1561}
1562
1563/* Main classifier routine: scans classifier chain attached
1564 to this qdisc, (optionally) tests for protocol and asks
1565 specific classifiers.
1566 */
Patrick McHardy73ca4912007-07-15 00:02:31 -07001567int tc_classify_compat(struct sk_buff *skb, struct tcf_proto *tp,
1568 struct tcf_result *res)
1569{
1570 __be16 protocol = skb->protocol;
1571 int err = 0;
1572
1573 for (; tp; tp = tp->next) {
1574 if ((tp->protocol == protocol ||
1575 tp->protocol == htons(ETH_P_ALL)) &&
1576 (err = tp->classify(skb, tp, res)) >= 0) {
1577#ifdef CONFIG_NET_CLS_ACT
1578 if (err != TC_ACT_RECLASSIFY && skb->tc_verd)
1579 skb->tc_verd = SET_TC_VERD(skb->tc_verd, 0);
1580#endif
1581 return err;
1582 }
1583 }
1584 return -1;
1585}
1586EXPORT_SYMBOL(tc_classify_compat);
1587
Linus Torvalds1da177e2005-04-16 15:20:36 -07001588int tc_classify(struct sk_buff *skb, struct tcf_proto *tp,
Patrick McHardy73ca4912007-07-15 00:02:31 -07001589 struct tcf_result *res)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001590{
1591 int err = 0;
Patrick McHardy73ca4912007-07-15 00:02:31 -07001592 __be16 protocol;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001593#ifdef CONFIG_NET_CLS_ACT
1594 struct tcf_proto *otp = tp;
1595reclassify:
1596#endif
1597 protocol = skb->protocol;
1598
Patrick McHardy73ca4912007-07-15 00:02:31 -07001599 err = tc_classify_compat(skb, tp, res);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001600#ifdef CONFIG_NET_CLS_ACT
Patrick McHardy73ca4912007-07-15 00:02:31 -07001601 if (err == TC_ACT_RECLASSIFY) {
1602 u32 verd = G_TC_VERD(skb->tc_verd);
1603 tp = otp;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001604
Patrick McHardy73ca4912007-07-15 00:02:31 -07001605 if (verd++ >= MAX_REC_LOOP) {
1606 printk("rule prio %u protocol %02x reclassify loop, "
1607 "packet dropped\n",
1608 tp->prio&0xffff, ntohs(tp->protocol));
1609 return TC_ACT_SHOT;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001610 }
Patrick McHardy73ca4912007-07-15 00:02:31 -07001611 skb->tc_verd = SET_TC_VERD(skb->tc_verd, verd);
1612 goto reclassify;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001613 }
Patrick McHardy73ca4912007-07-15 00:02:31 -07001614#endif
1615 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001616}
Patrick McHardy73ca4912007-07-15 00:02:31 -07001617EXPORT_SYMBOL(tc_classify);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001618
Patrick McHardya48b5a62007-03-23 11:29:43 -07001619void tcf_destroy(struct tcf_proto *tp)
1620{
1621 tp->ops->destroy(tp);
1622 module_put(tp->ops->owner);
1623 kfree(tp);
1624}
1625
Patrick McHardyff31ab52008-07-01 19:52:38 -07001626void tcf_destroy_chain(struct tcf_proto **fl)
Patrick McHardya48b5a62007-03-23 11:29:43 -07001627{
1628 struct tcf_proto *tp;
1629
Patrick McHardyff31ab52008-07-01 19:52:38 -07001630 while ((tp = *fl) != NULL) {
1631 *fl = tp->next;
Patrick McHardya48b5a62007-03-23 11:29:43 -07001632 tcf_destroy(tp);
1633 }
1634}
1635EXPORT_SYMBOL(tcf_destroy_chain);
1636
Linus Torvalds1da177e2005-04-16 15:20:36 -07001637#ifdef CONFIG_PROC_FS
1638static int psched_show(struct seq_file *seq, void *v)
1639{
Patrick McHardy3c0cfc12007-10-10 16:32:41 -07001640 struct timespec ts;
1641
1642 hrtimer_get_res(CLOCK_MONOTONIC, &ts);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001643 seq_printf(seq, "%08x %08x %08x %08x\n",
Patrick McHardy641b9e02007-03-16 01:18:42 -07001644 (u32)NSEC_PER_USEC, (u32)PSCHED_US2NS(1),
Patrick McHardy514bca32007-03-16 12:34:52 -07001645 1000000,
Patrick McHardy3c0cfc12007-10-10 16:32:41 -07001646 (u32)NSEC_PER_SEC/(u32)ktime_to_ns(timespec_to_ktime(ts)));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001647
1648 return 0;
1649}
1650
1651static int psched_open(struct inode *inode, struct file *file)
1652{
1653 return single_open(file, psched_show, PDE(inode)->data);
1654}
1655
Arjan van de Venda7071d2007-02-12 00:55:36 -08001656static const struct file_operations psched_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001657 .owner = THIS_MODULE,
1658 .open = psched_open,
1659 .read = seq_read,
1660 .llseek = seq_lseek,
1661 .release = single_release,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001662};
Linus Torvalds1da177e2005-04-16 15:20:36 -07001663#endif
1664
Linus Torvalds1da177e2005-04-16 15:20:36 -07001665static int __init pktsched_init(void)
1666{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001667 register_qdisc(&pfifo_qdisc_ops);
1668 register_qdisc(&bfifo_qdisc_ops);
Eric W. Biederman457c4cb2007-09-12 12:01:34 +02001669 proc_net_fops_create(&init_net, "psched", 0, &psched_fops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001670
Thomas Grafbe577dd2007-03-22 11:55:50 -07001671 rtnl_register(PF_UNSPEC, RTM_NEWQDISC, tc_modify_qdisc, NULL);
1672 rtnl_register(PF_UNSPEC, RTM_DELQDISC, tc_get_qdisc, NULL);
1673 rtnl_register(PF_UNSPEC, RTM_GETQDISC, tc_get_qdisc, tc_dump_qdisc);
1674 rtnl_register(PF_UNSPEC, RTM_NEWTCLASS, tc_ctl_tclass, NULL);
1675 rtnl_register(PF_UNSPEC, RTM_DELTCLASS, tc_ctl_tclass, NULL);
1676 rtnl_register(PF_UNSPEC, RTM_GETTCLASS, tc_ctl_tclass, tc_dump_tclass);
1677
Linus Torvalds1da177e2005-04-16 15:20:36 -07001678 return 0;
1679}
1680
1681subsys_initcall(pktsched_init);