blob: d91a2338877cbc108a650cf27df5f147a165de23 [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. Miller4d8863a2008-08-18 21:03:15 -0700641 if (old)
David S. Miller99194cf2008-07-17 04:54:10 -0700642 qdisc_destroy(old);
David S. Miller99194cf2008-07-17 04:54:10 -0700643}
644
645/* Graft qdisc "new" to class "classid" of qdisc "parent" or
646 * to device "dev".
647 *
648 * When appropriate send a netlink notification using 'skb'
649 * and "n".
650 *
651 * On success, destroy old qdisc.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700652 */
653
654static int qdisc_graft(struct net_device *dev, struct Qdisc *parent,
David S. Miller99194cf2008-07-17 04:54:10 -0700655 struct sk_buff *skb, struct nlmsghdr *n, u32 classid,
656 struct Qdisc *new, struct Qdisc *old)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700657{
David S. Miller99194cf2008-07-17 04:54:10 -0700658 struct Qdisc *q = old;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700659 int err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700660
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900661 if (parent == NULL) {
David S. Miller99194cf2008-07-17 04:54:10 -0700662 unsigned int i, num_q, ingress;
663
664 ingress = 0;
665 num_q = dev->num_tx_queues;
David S. Miller8d50b532008-07-30 02:37:46 -0700666 if ((q && q->flags & TCQ_F_INGRESS) ||
667 (new && new->flags & TCQ_F_INGRESS)) {
David S. Miller99194cf2008-07-17 04:54:10 -0700668 num_q = 1;
669 ingress = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700670 }
David S. Miller99194cf2008-07-17 04:54:10 -0700671
672 if (dev->flags & IFF_UP)
673 dev_deactivate(dev);
674
675 for (i = 0; i < num_q; i++) {
676 struct netdev_queue *dev_queue = &dev->rx_queue;
677
678 if (!ingress)
679 dev_queue = netdev_get_tx_queue(dev, i);
680
David S. Miller8d50b532008-07-30 02:37:46 -0700681 old = dev_graft_qdisc(dev_queue, new);
682 if (new && i > 0)
683 atomic_inc(&new->refcnt);
684
David S. Miller99194cf2008-07-17 04:54:10 -0700685 notify_and_destroy(skb, n, classid, old, new);
686 }
687
688 if (dev->flags & IFF_UP)
689 dev_activate(dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700690 } else {
Eric Dumazet20fea082007-11-14 01:44:41 -0800691 const struct Qdisc_class_ops *cops = parent->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700692
693 err = -EINVAL;
694
695 if (cops) {
696 unsigned long cl = cops->get(parent, classid);
697 if (cl) {
David S. Miller99194cf2008-07-17 04:54:10 -0700698 err = cops->graft(parent, cl, new, &old);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700699 cops->put(parent, cl);
700 }
701 }
David S. Miller99194cf2008-07-17 04:54:10 -0700702 if (!err)
703 notify_and_destroy(skb, n, classid, old, new);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700704 }
705 return err;
706}
707
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700708/* lockdep annotation is needed for ingress; egress gets it only for name */
709static struct lock_class_key qdisc_tx_lock;
710static struct lock_class_key qdisc_rx_lock;
711
Linus Torvalds1da177e2005-04-16 15:20:36 -0700712/*
713 Allocate and initialize new qdisc.
714
715 Parameters are passed via opt.
716 */
717
718static struct Qdisc *
David S. Millerbb949fb2008-07-08 16:55:56 -0700719qdisc_create(struct net_device *dev, struct netdev_queue *dev_queue,
720 u32 parent, u32 handle, struct nlattr **tca, int *errp)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700721{
722 int err;
Patrick McHardy1e904742008-01-22 22:11:17 -0800723 struct nlattr *kind = tca[TCA_KIND];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700724 struct Qdisc *sch;
725 struct Qdisc_ops *ops;
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700726 struct qdisc_size_table *stab;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700727
728 ops = qdisc_lookup_ops(kind);
729#ifdef CONFIG_KMOD
730 if (ops == NULL && kind != NULL) {
731 char name[IFNAMSIZ];
Patrick McHardy1e904742008-01-22 22:11:17 -0800732 if (nla_strlcpy(name, kind, IFNAMSIZ) < IFNAMSIZ) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700733 /* We dropped the RTNL semaphore in order to
734 * perform the module load. So, even if we
735 * succeeded in loading the module we have to
736 * tell the caller to replay the request. We
737 * indicate this using -EAGAIN.
738 * We replay the request because the device may
739 * go away in the mean time.
740 */
741 rtnl_unlock();
742 request_module("sch_%s", name);
743 rtnl_lock();
744 ops = qdisc_lookup_ops(kind);
745 if (ops != NULL) {
746 /* We will try again qdisc_lookup_ops,
747 * so don't keep a reference.
748 */
749 module_put(ops->owner);
750 err = -EAGAIN;
751 goto err_out;
752 }
753 }
754 }
755#endif
756
Jamal Hadi Salimb9e2cc02006-08-03 16:36:51 -0700757 err = -ENOENT;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700758 if (ops == NULL)
759 goto err_out;
760
David S. Miller5ce2d482008-07-08 17:06:30 -0700761 sch = qdisc_alloc(dev_queue, ops);
Thomas Graf3d54b822005-07-05 14:15:09 -0700762 if (IS_ERR(sch)) {
763 err = PTR_ERR(sch);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700764 goto err_out2;
Thomas Graf3d54b822005-07-05 14:15:09 -0700765 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700766
Patrick McHardyffc8fef2007-07-30 17:11:50 -0700767 sch->parent = parent;
768
Thomas Graf3d54b822005-07-05 14:15:09 -0700769 if (handle == TC_H_INGRESS) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700770 sch->flags |= TCQ_F_INGRESS;
Thomas Graf3d54b822005-07-05 14:15:09 -0700771 handle = TC_H_MAKE(TC_H_INGRESS, 0);
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700772 lockdep_set_class(qdisc_lock(sch), &qdisc_rx_lock);
Patrick McHardyfd44de72007-04-16 17:07:08 -0700773 } else {
Patrick McHardyfd44de72007-04-16 17:07:08 -0700774 if (handle == 0) {
775 handle = qdisc_alloc_handle(dev);
776 err = -ENOMEM;
777 if (handle == 0)
778 goto err_out3;
779 }
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700780 lockdep_set_class(qdisc_lock(sch), &qdisc_tx_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700781 }
782
Thomas Graf3d54b822005-07-05 14:15:09 -0700783 sch->handle = handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700784
Patrick McHardy1e904742008-01-22 22:11:17 -0800785 if (!ops->init || (err = ops->init(sch, tca[TCA_OPTIONS])) == 0) {
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700786 if (tca[TCA_STAB]) {
787 stab = qdisc_get_stab(tca[TCA_STAB]);
788 if (IS_ERR(stab)) {
789 err = PTR_ERR(stab);
790 goto err_out3;
791 }
792 sch->stab = stab;
793 }
Patrick McHardy1e904742008-01-22 22:11:17 -0800794 if (tca[TCA_RATE]) {
Thomas Graf023e09a2005-07-05 14:15:53 -0700795 err = gen_new_estimator(&sch->bstats, &sch->rate_est,
David S. Miller7698b4f2008-07-16 01:42:40 -0700796 qdisc_root_lock(sch),
Patrick McHardy1e904742008-01-22 22:11:17 -0800797 tca[TCA_RATE]);
Thomas Graf023e09a2005-07-05 14:15:53 -0700798 if (err) {
799 /*
800 * Any broken qdiscs that would require
801 * a ops->reset() here? The qdisc was never
802 * in action so it shouldn't be necessary.
803 */
804 if (ops->destroy)
805 ops->destroy(sch);
806 goto err_out3;
807 }
808 }
David S. Milleree7af822008-08-06 23:35:59 -0700809 if ((parent != TC_H_ROOT) && !(sch->flags & TCQ_F_INGRESS))
David S. Miller827ebd62008-08-07 20:26:40 -0700810 list_add_tail(&sch->list, &dev_queue->qdisc_sleeping->list);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700811
Linus Torvalds1da177e2005-04-16 15:20:36 -0700812 return sch;
813 }
814err_out3:
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700815 qdisc_put_stab(sch->stab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700816 dev_put(dev);
Thomas Graf3d54b822005-07-05 14:15:09 -0700817 kfree((char *) sch - sch->padded);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700818err_out2:
819 module_put(ops->owner);
820err_out:
821 *errp = err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700822 return NULL;
823}
824
Patrick McHardy1e904742008-01-22 22:11:17 -0800825static int qdisc_change(struct Qdisc *sch, struct nlattr **tca)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700826{
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700827 struct qdisc_size_table *stab = NULL;
828 int err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700829
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700830 if (tca[TCA_OPTIONS]) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700831 if (sch->ops->change == NULL)
832 return -EINVAL;
Patrick McHardy1e904742008-01-22 22:11:17 -0800833 err = sch->ops->change(sch, tca[TCA_OPTIONS]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700834 if (err)
835 return err;
836 }
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700837
838 if (tca[TCA_STAB]) {
839 stab = qdisc_get_stab(tca[TCA_STAB]);
840 if (IS_ERR(stab))
841 return PTR_ERR(stab);
842 }
843
844 qdisc_put_stab(sch->stab);
845 sch->stab = stab;
846
Patrick McHardy1e904742008-01-22 22:11:17 -0800847 if (tca[TCA_RATE])
Linus Torvalds1da177e2005-04-16 15:20:36 -0700848 gen_replace_estimator(&sch->bstats, &sch->rate_est,
David S. Miller7698b4f2008-07-16 01:42:40 -0700849 qdisc_root_lock(sch), tca[TCA_RATE]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700850 return 0;
851}
852
853struct check_loop_arg
854{
855 struct qdisc_walker w;
856 struct Qdisc *p;
857 int depth;
858};
859
860static int check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w);
861
862static int check_loop(struct Qdisc *q, struct Qdisc *p, int depth)
863{
864 struct check_loop_arg arg;
865
866 if (q->ops->cl_ops == NULL)
867 return 0;
868
869 arg.w.stop = arg.w.skip = arg.w.count = 0;
870 arg.w.fn = check_loop_fn;
871 arg.depth = depth;
872 arg.p = p;
873 q->ops->cl_ops->walk(q, &arg.w);
874 return arg.w.stop ? -ELOOP : 0;
875}
876
877static int
878check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w)
879{
880 struct Qdisc *leaf;
Eric Dumazet20fea082007-11-14 01:44:41 -0800881 const struct Qdisc_class_ops *cops = q->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700882 struct check_loop_arg *arg = (struct check_loop_arg *)w;
883
884 leaf = cops->leaf(q, cl);
885 if (leaf) {
886 if (leaf == arg->p || arg->depth > 7)
887 return -ELOOP;
888 return check_loop(leaf, arg->p, arg->depth + 1);
889 }
890 return 0;
891}
892
893/*
894 * Delete/get qdisc.
895 */
896
897static int tc_get_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
898{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900899 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700900 struct tcmsg *tcm = NLMSG_DATA(n);
Patrick McHardy1e904742008-01-22 22:11:17 -0800901 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700902 struct net_device *dev;
903 u32 clid = tcm->tcm_parent;
904 struct Qdisc *q = NULL;
905 struct Qdisc *p = NULL;
906 int err;
907
Denis V. Lunevb8542722007-12-01 00:21:31 +1100908 if (net != &init_net)
909 return -EINVAL;
910
Eric W. Biederman881d9662007-09-17 11:56:21 -0700911 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700912 return -ENODEV;
913
Patrick McHardy1e904742008-01-22 22:11:17 -0800914 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
915 if (err < 0)
916 return err;
917
Linus Torvalds1da177e2005-04-16 15:20:36 -0700918 if (clid) {
919 if (clid != TC_H_ROOT) {
920 if (TC_H_MAJ(clid) != TC_H_MAJ(TC_H_INGRESS)) {
921 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
922 return -ENOENT;
923 q = qdisc_leaf(p, clid);
924 } else { /* ingress */
David S. Miller8123b422008-08-08 23:23:39 -0700925 q = dev->rx_queue.qdisc_sleeping;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900926 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700927 } else {
David S. Millere8a04642008-07-17 00:34:19 -0700928 struct netdev_queue *dev_queue;
929 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Millerb0e1e642008-07-08 17:42:10 -0700930 q = dev_queue->qdisc_sleeping;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700931 }
932 if (!q)
933 return -ENOENT;
934
935 if (tcm->tcm_handle && q->handle != tcm->tcm_handle)
936 return -EINVAL;
937 } else {
938 if ((q = qdisc_lookup(dev, tcm->tcm_handle)) == NULL)
939 return -ENOENT;
940 }
941
Patrick McHardy1e904742008-01-22 22:11:17 -0800942 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700943 return -EINVAL;
944
945 if (n->nlmsg_type == RTM_DELQDISC) {
946 if (!clid)
947 return -EINVAL;
948 if (q->handle == 0)
949 return -ENOENT;
David S. Miller99194cf2008-07-17 04:54:10 -0700950 if ((err = qdisc_graft(dev, p, skb, n, clid, NULL, q)) != 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700951 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700952 } else {
953 qdisc_notify(skb, n, clid, NULL, q);
954 }
955 return 0;
956}
957
958/*
959 Create/change qdisc.
960 */
961
962static int tc_modify_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
963{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900964 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700965 struct tcmsg *tcm;
Patrick McHardy1e904742008-01-22 22:11:17 -0800966 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700967 struct net_device *dev;
968 u32 clid;
969 struct Qdisc *q, *p;
970 int err;
971
Denis V. Lunevb8542722007-12-01 00:21:31 +1100972 if (net != &init_net)
973 return -EINVAL;
974
Linus Torvalds1da177e2005-04-16 15:20:36 -0700975replay:
976 /* Reinit, just in case something touches this. */
977 tcm = NLMSG_DATA(n);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700978 clid = tcm->tcm_parent;
979 q = p = NULL;
980
Eric W. Biederman881d9662007-09-17 11:56:21 -0700981 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700982 return -ENODEV;
983
Patrick McHardy1e904742008-01-22 22:11:17 -0800984 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
985 if (err < 0)
986 return err;
987
Linus Torvalds1da177e2005-04-16 15:20:36 -0700988 if (clid) {
989 if (clid != TC_H_ROOT) {
990 if (clid != TC_H_INGRESS) {
991 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
992 return -ENOENT;
993 q = qdisc_leaf(p, clid);
994 } else { /*ingress */
David S. Miller8123b422008-08-08 23:23:39 -0700995 q = dev->rx_queue.qdisc_sleeping;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700996 }
997 } else {
David S. Millere8a04642008-07-17 00:34:19 -0700998 struct netdev_queue *dev_queue;
999 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Millerb0e1e642008-07-08 17:42:10 -07001000 q = dev_queue->qdisc_sleeping;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001001 }
1002
1003 /* It may be default qdisc, ignore it */
1004 if (q && q->handle == 0)
1005 q = NULL;
1006
1007 if (!q || !tcm->tcm_handle || q->handle != tcm->tcm_handle) {
1008 if (tcm->tcm_handle) {
1009 if (q && !(n->nlmsg_flags&NLM_F_REPLACE))
1010 return -EEXIST;
1011 if (TC_H_MIN(tcm->tcm_handle))
1012 return -EINVAL;
1013 if ((q = qdisc_lookup(dev, tcm->tcm_handle)) == NULL)
1014 goto create_n_graft;
1015 if (n->nlmsg_flags&NLM_F_EXCL)
1016 return -EEXIST;
Patrick McHardy1e904742008-01-22 22:11:17 -08001017 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001018 return -EINVAL;
1019 if (q == p ||
1020 (p && check_loop(q, p, 0)))
1021 return -ELOOP;
1022 atomic_inc(&q->refcnt);
1023 goto graft;
1024 } else {
1025 if (q == NULL)
1026 goto create_n_graft;
1027
1028 /* This magic test requires explanation.
1029 *
1030 * We know, that some child q is already
1031 * attached to this parent and have choice:
1032 * either to change it or to create/graft new one.
1033 *
1034 * 1. We are allowed to create/graft only
1035 * if CREATE and REPLACE flags are set.
1036 *
1037 * 2. If EXCL is set, requestor wanted to say,
1038 * that qdisc tcm_handle is not expected
1039 * to exist, so that we choose create/graft too.
1040 *
1041 * 3. The last case is when no flags are set.
1042 * Alas, it is sort of hole in API, we
1043 * cannot decide what to do unambiguously.
1044 * For now we select create/graft, if
1045 * user gave KIND, which does not match existing.
1046 */
1047 if ((n->nlmsg_flags&NLM_F_CREATE) &&
1048 (n->nlmsg_flags&NLM_F_REPLACE) &&
1049 ((n->nlmsg_flags&NLM_F_EXCL) ||
Patrick McHardy1e904742008-01-22 22:11:17 -08001050 (tca[TCA_KIND] &&
1051 nla_strcmp(tca[TCA_KIND], q->ops->id))))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001052 goto create_n_graft;
1053 }
1054 }
1055 } else {
1056 if (!tcm->tcm_handle)
1057 return -EINVAL;
1058 q = qdisc_lookup(dev, tcm->tcm_handle);
1059 }
1060
1061 /* Change qdisc parameters */
1062 if (q == NULL)
1063 return -ENOENT;
1064 if (n->nlmsg_flags&NLM_F_EXCL)
1065 return -EEXIST;
Patrick McHardy1e904742008-01-22 22:11:17 -08001066 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001067 return -EINVAL;
1068 err = qdisc_change(q, tca);
1069 if (err == 0)
1070 qdisc_notify(skb, n, clid, NULL, q);
1071 return err;
1072
1073create_n_graft:
1074 if (!(n->nlmsg_flags&NLM_F_CREATE))
1075 return -ENOENT;
1076 if (clid == TC_H_INGRESS)
David S. Millerbb949fb2008-07-08 16:55:56 -07001077 q = qdisc_create(dev, &dev->rx_queue,
1078 tcm->tcm_parent, tcm->tcm_parent,
Patrick McHardyffc8fef2007-07-30 17:11:50 -07001079 tca, &err);
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001080 else
David S. Millere8a04642008-07-17 00:34:19 -07001081 q = qdisc_create(dev, netdev_get_tx_queue(dev, 0),
David S. Millerbb949fb2008-07-08 16:55:56 -07001082 tcm->tcm_parent, tcm->tcm_handle,
Patrick McHardyffc8fef2007-07-30 17:11:50 -07001083 tca, &err);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001084 if (q == NULL) {
1085 if (err == -EAGAIN)
1086 goto replay;
1087 return err;
1088 }
1089
1090graft:
1091 if (1) {
David S. Miller99194cf2008-07-17 04:54:10 -07001092 err = qdisc_graft(dev, p, skb, n, clid, q, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001093 if (err) {
David S. Miller4d8863a2008-08-18 21:03:15 -07001094 if (q)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001095 qdisc_destroy(q);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001096 return err;
1097 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001098 }
1099 return 0;
1100}
1101
1102static int tc_fill_qdisc(struct sk_buff *skb, struct Qdisc *q, u32 clid,
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001103 u32 pid, u32 seq, u16 flags, int event)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001104{
1105 struct tcmsg *tcm;
1106 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001107 unsigned char *b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001108 struct gnet_dump d;
1109
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001110 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001111 tcm = NLMSG_DATA(nlh);
1112 tcm->tcm_family = AF_UNSPEC;
Patrick McHardy9ef1d4c2005-06-28 12:55:30 -07001113 tcm->tcm__pad1 = 0;
1114 tcm->tcm__pad2 = 0;
David S. Miller5ce2d482008-07-08 17:06:30 -07001115 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001116 tcm->tcm_parent = clid;
1117 tcm->tcm_handle = q->handle;
1118 tcm->tcm_info = atomic_read(&q->refcnt);
Patrick McHardy57e1c482008-01-23 20:34:28 -08001119 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001120 if (q->ops->dump && q->ops->dump(q, skb) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001121 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001122 q->qstats.qlen = q->q.qlen;
1123
Jussi Kivilinna175f9c12008-07-20 00:08:47 -07001124 if (q->stab && qdisc_dump_stab(skb, q->stab) < 0)
1125 goto nla_put_failure;
1126
Linus Torvalds1da177e2005-04-16 15:20:36 -07001127 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS,
David S. Miller7698b4f2008-07-16 01:42:40 -07001128 TCA_XSTATS, qdisc_root_lock(q), &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001129 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001130
1131 if (q->ops->dump_stats && q->ops->dump_stats(q, &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001132 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001133
1134 if (gnet_stats_copy_basic(&d, &q->bstats) < 0 ||
Linus Torvalds1da177e2005-04-16 15:20:36 -07001135 gnet_stats_copy_rate_est(&d, &q->rate_est) < 0 ||
Linus Torvalds1da177e2005-04-16 15:20:36 -07001136 gnet_stats_copy_queue(&d, &q->qstats) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001137 goto nla_put_failure;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001138
Linus Torvalds1da177e2005-04-16 15:20:36 -07001139 if (gnet_stats_finish_copy(&d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001140 goto nla_put_failure;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001141
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001142 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001143 return skb->len;
1144
1145nlmsg_failure:
Patrick McHardy1e904742008-01-22 22:11:17 -08001146nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -07001147 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001148 return -1;
1149}
1150
1151static int qdisc_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1152 u32 clid, struct Qdisc *old, struct Qdisc *new)
1153{
1154 struct sk_buff *skb;
1155 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1156
1157 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1158 if (!skb)
1159 return -ENOBUFS;
1160
1161 if (old && old->handle) {
1162 if (tc_fill_qdisc(skb, old, clid, pid, n->nlmsg_seq, 0, RTM_DELQDISC) < 0)
1163 goto err_out;
1164 }
1165 if (new) {
1166 if (tc_fill_qdisc(skb, new, clid, pid, n->nlmsg_seq, old ? NLM_F_REPLACE : 0, RTM_NEWQDISC) < 0)
1167 goto err_out;
1168 }
1169
1170 if (skb->len)
Denis V. Lunev97c53ca2007-11-19 22:26:51 -08001171 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001172
1173err_out:
1174 kfree_skb(skb);
1175 return -EINVAL;
1176}
1177
David S. Miller30723672008-07-18 22:50:15 -07001178static bool tc_qdisc_dump_ignore(struct Qdisc *q)
1179{
1180 return (q->flags & TCQ_F_BUILTIN) ? true : false;
1181}
1182
1183static int tc_dump_qdisc_root(struct Qdisc *root, struct sk_buff *skb,
1184 struct netlink_callback *cb,
1185 int *q_idx_p, int s_q_idx)
1186{
1187 int ret = 0, q_idx = *q_idx_p;
1188 struct Qdisc *q;
1189
1190 if (!root)
1191 return 0;
1192
1193 q = root;
1194 if (q_idx < s_q_idx) {
1195 q_idx++;
1196 } else {
1197 if (!tc_qdisc_dump_ignore(q) &&
1198 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1199 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1200 goto done;
1201 q_idx++;
1202 }
1203 list_for_each_entry(q, &root->list, list) {
1204 if (q_idx < s_q_idx) {
1205 q_idx++;
1206 continue;
1207 }
1208 if (!tc_qdisc_dump_ignore(q) &&
1209 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1210 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1211 goto done;
1212 q_idx++;
1213 }
1214
1215out:
1216 *q_idx_p = q_idx;
1217 return ret;
1218done:
1219 ret = -1;
1220 goto out;
1221}
1222
Linus Torvalds1da177e2005-04-16 15:20:36 -07001223static int tc_dump_qdisc(struct sk_buff *skb, struct netlink_callback *cb)
1224{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001225 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001226 int idx, q_idx;
1227 int s_idx, s_q_idx;
1228 struct net_device *dev;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001229
Denis V. Lunevb8542722007-12-01 00:21:31 +11001230 if (net != &init_net)
1231 return 0;
1232
Linus Torvalds1da177e2005-04-16 15:20:36 -07001233 s_idx = cb->args[0];
1234 s_q_idx = q_idx = cb->args[1];
1235 read_lock(&dev_base_lock);
Pavel Emelianov7562f872007-05-03 15:13:45 -07001236 idx = 0;
Eric W. Biederman881d9662007-09-17 11:56:21 -07001237 for_each_netdev(&init_net, dev) {
David S. Miller30723672008-07-18 22:50:15 -07001238 struct netdev_queue *dev_queue;
1239
Linus Torvalds1da177e2005-04-16 15:20:36 -07001240 if (idx < s_idx)
Pavel Emelianov7562f872007-05-03 15:13:45 -07001241 goto cont;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001242 if (idx > s_idx)
1243 s_q_idx = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001244 q_idx = 0;
David S. Miller30723672008-07-18 22:50:15 -07001245
1246 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Miller827ebd62008-08-07 20:26:40 -07001247 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 -07001248 goto done;
1249
1250 dev_queue = &dev->rx_queue;
David S. Miller827ebd62008-08-07 20:26:40 -07001251 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 -07001252 goto done;
1253
Pavel Emelianov7562f872007-05-03 15:13:45 -07001254cont:
1255 idx++;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001256 }
1257
1258done:
1259 read_unlock(&dev_base_lock);
1260
1261 cb->args[0] = idx;
1262 cb->args[1] = q_idx;
1263
1264 return skb->len;
1265}
1266
1267
1268
1269/************************************************
1270 * Traffic classes manipulation. *
1271 ************************************************/
1272
1273
1274
1275static int tc_ctl_tclass(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
1276{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001277 struct net *net = sock_net(skb->sk);
David S. Millerb0e1e642008-07-08 17:42:10 -07001278 struct netdev_queue *dev_queue;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001279 struct tcmsg *tcm = NLMSG_DATA(n);
Patrick McHardy1e904742008-01-22 22:11:17 -08001280 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -07001281 struct net_device *dev;
1282 struct Qdisc *q = NULL;
Eric Dumazet20fea082007-11-14 01:44:41 -08001283 const struct Qdisc_class_ops *cops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001284 unsigned long cl = 0;
1285 unsigned long new_cl;
1286 u32 pid = tcm->tcm_parent;
1287 u32 clid = tcm->tcm_handle;
1288 u32 qid = TC_H_MAJ(clid);
1289 int err;
1290
Denis V. Lunevb8542722007-12-01 00:21:31 +11001291 if (net != &init_net)
1292 return -EINVAL;
1293
Eric W. Biederman881d9662007-09-17 11:56:21 -07001294 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001295 return -ENODEV;
1296
Patrick McHardy1e904742008-01-22 22:11:17 -08001297 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1298 if (err < 0)
1299 return err;
1300
Linus Torvalds1da177e2005-04-16 15:20:36 -07001301 /*
1302 parent == TC_H_UNSPEC - unspecified parent.
1303 parent == TC_H_ROOT - class is root, which has no parent.
1304 parent == X:0 - parent is root class.
1305 parent == X:Y - parent is a node in hierarchy.
1306 parent == 0:Y - parent is X:Y, where X:0 is qdisc.
1307
1308 handle == 0:0 - generate handle from kernel pool.
1309 handle == 0:Y - class is X:Y, where X:0 is qdisc.
1310 handle == X:Y - clear.
1311 handle == X:0 - root class.
1312 */
1313
1314 /* Step 1. Determine qdisc handle X:0 */
1315
David S. Millere8a04642008-07-17 00:34:19 -07001316 dev_queue = netdev_get_tx_queue(dev, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001317 if (pid != TC_H_ROOT) {
1318 u32 qid1 = TC_H_MAJ(pid);
1319
1320 if (qid && qid1) {
1321 /* If both majors are known, they must be identical. */
1322 if (qid != qid1)
1323 return -EINVAL;
1324 } else if (qid1) {
1325 qid = qid1;
1326 } else if (qid == 0)
David S. Millerb0e1e642008-07-08 17:42:10 -07001327 qid = dev_queue->qdisc_sleeping->handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001328
1329 /* Now qid is genuine qdisc handle consistent
1330 both with parent and child.
1331
1332 TC_H_MAJ(pid) still may be unspecified, complete it now.
1333 */
1334 if (pid)
1335 pid = TC_H_MAKE(qid, pid);
1336 } else {
1337 if (qid == 0)
David S. Millerb0e1e642008-07-08 17:42:10 -07001338 qid = dev_queue->qdisc_sleeping->handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001339 }
1340
1341 /* OK. Locate qdisc */
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001342 if ((q = qdisc_lookup(dev, qid)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001343 return -ENOENT;
1344
1345 /* An check that it supports classes */
1346 cops = q->ops->cl_ops;
1347 if (cops == NULL)
1348 return -EINVAL;
1349
1350 /* Now try to get class */
1351 if (clid == 0) {
1352 if (pid == TC_H_ROOT)
1353 clid = qid;
1354 } else
1355 clid = TC_H_MAKE(qid, clid);
1356
1357 if (clid)
1358 cl = cops->get(q, clid);
1359
1360 if (cl == 0) {
1361 err = -ENOENT;
1362 if (n->nlmsg_type != RTM_NEWTCLASS || !(n->nlmsg_flags&NLM_F_CREATE))
1363 goto out;
1364 } else {
1365 switch (n->nlmsg_type) {
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001366 case RTM_NEWTCLASS:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001367 err = -EEXIST;
1368 if (n->nlmsg_flags&NLM_F_EXCL)
1369 goto out;
1370 break;
1371 case RTM_DELTCLASS:
1372 err = cops->delete(q, cl);
1373 if (err == 0)
1374 tclass_notify(skb, n, q, cl, RTM_DELTCLASS);
1375 goto out;
1376 case RTM_GETTCLASS:
1377 err = tclass_notify(skb, n, q, cl, RTM_NEWTCLASS);
1378 goto out;
1379 default:
1380 err = -EINVAL;
1381 goto out;
1382 }
1383 }
1384
1385 new_cl = cl;
1386 err = cops->change(q, clid, pid, tca, &new_cl);
1387 if (err == 0)
1388 tclass_notify(skb, n, q, new_cl, RTM_NEWTCLASS);
1389
1390out:
1391 if (cl)
1392 cops->put(q, cl);
1393
1394 return err;
1395}
1396
1397
1398static int tc_fill_tclass(struct sk_buff *skb, struct Qdisc *q,
1399 unsigned long cl,
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001400 u32 pid, u32 seq, u16 flags, int event)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001401{
1402 struct tcmsg *tcm;
1403 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001404 unsigned char *b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001405 struct gnet_dump d;
Eric Dumazet20fea082007-11-14 01:44:41 -08001406 const struct Qdisc_class_ops *cl_ops = q->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001407
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001408 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001409 tcm = NLMSG_DATA(nlh);
1410 tcm->tcm_family = AF_UNSPEC;
David S. Miller5ce2d482008-07-08 17:06:30 -07001411 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001412 tcm->tcm_parent = q->handle;
1413 tcm->tcm_handle = q->handle;
1414 tcm->tcm_info = 0;
Patrick McHardy57e1c482008-01-23 20:34:28 -08001415 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001416 if (cl_ops->dump && cl_ops->dump(q, cl, skb, tcm) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001417 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001418
1419 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS,
David S. Miller7698b4f2008-07-16 01:42:40 -07001420 TCA_XSTATS, qdisc_root_lock(q), &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001421 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001422
1423 if (cl_ops->dump_stats && cl_ops->dump_stats(q, cl, &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001424 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001425
1426 if (gnet_stats_finish_copy(&d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001427 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001428
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001429 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001430 return skb->len;
1431
1432nlmsg_failure:
Patrick McHardy1e904742008-01-22 22:11:17 -08001433nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -07001434 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001435 return -1;
1436}
1437
1438static int tclass_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1439 struct Qdisc *q, unsigned long cl, int event)
1440{
1441 struct sk_buff *skb;
1442 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1443
1444 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1445 if (!skb)
1446 return -ENOBUFS;
1447
1448 if (tc_fill_tclass(skb, q, cl, pid, n->nlmsg_seq, 0, event) < 0) {
1449 kfree_skb(skb);
1450 return -EINVAL;
1451 }
1452
Denis V. Lunev97c53ca2007-11-19 22:26:51 -08001453 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001454}
1455
1456struct qdisc_dump_args
1457{
1458 struct qdisc_walker w;
1459 struct sk_buff *skb;
1460 struct netlink_callback *cb;
1461};
1462
1463static int qdisc_class_dump(struct Qdisc *q, unsigned long cl, struct qdisc_walker *arg)
1464{
1465 struct qdisc_dump_args *a = (struct qdisc_dump_args *)arg;
1466
1467 return tc_fill_tclass(a->skb, q, cl, NETLINK_CB(a->cb->skb).pid,
1468 a->cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWTCLASS);
1469}
1470
David S. Miller30723672008-07-18 22:50:15 -07001471static int tc_dump_tclass_qdisc(struct Qdisc *q, struct sk_buff *skb,
1472 struct tcmsg *tcm, struct netlink_callback *cb,
1473 int *t_p, int s_t)
1474{
1475 struct qdisc_dump_args arg;
1476
1477 if (tc_qdisc_dump_ignore(q) ||
1478 *t_p < s_t || !q->ops->cl_ops ||
1479 (tcm->tcm_parent &&
1480 TC_H_MAJ(tcm->tcm_parent) != q->handle)) {
1481 (*t_p)++;
1482 return 0;
1483 }
1484 if (*t_p > s_t)
1485 memset(&cb->args[1], 0, sizeof(cb->args)-sizeof(cb->args[0]));
1486 arg.w.fn = qdisc_class_dump;
1487 arg.skb = skb;
1488 arg.cb = cb;
1489 arg.w.stop = 0;
1490 arg.w.skip = cb->args[1];
1491 arg.w.count = 0;
1492 q->ops->cl_ops->walk(q, &arg.w);
1493 cb->args[1] = arg.w.count;
1494 if (arg.w.stop)
1495 return -1;
1496 (*t_p)++;
1497 return 0;
1498}
1499
1500static int tc_dump_tclass_root(struct Qdisc *root, struct sk_buff *skb,
1501 struct tcmsg *tcm, struct netlink_callback *cb,
1502 int *t_p, int s_t)
1503{
1504 struct Qdisc *q;
1505
1506 if (!root)
1507 return 0;
1508
1509 if (tc_dump_tclass_qdisc(root, skb, tcm, cb, t_p, s_t) < 0)
1510 return -1;
1511
1512 list_for_each_entry(q, &root->list, list) {
1513 if (tc_dump_tclass_qdisc(q, skb, tcm, cb, t_p, s_t) < 0)
1514 return -1;
1515 }
1516
1517 return 0;
1518}
1519
Linus Torvalds1da177e2005-04-16 15:20:36 -07001520static int tc_dump_tclass(struct sk_buff *skb, struct netlink_callback *cb)
1521{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001522 struct tcmsg *tcm = (struct tcmsg*)NLMSG_DATA(cb->nlh);
David S. Miller30723672008-07-18 22:50:15 -07001523 struct net *net = sock_net(skb->sk);
1524 struct netdev_queue *dev_queue;
1525 struct net_device *dev;
1526 int t, s_t;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001527
Denis V. Lunevb8542722007-12-01 00:21:31 +11001528 if (net != &init_net)
1529 return 0;
1530
Linus Torvalds1da177e2005-04-16 15:20:36 -07001531 if (cb->nlh->nlmsg_len < NLMSG_LENGTH(sizeof(*tcm)))
1532 return 0;
Eric W. Biederman881d9662007-09-17 11:56:21 -07001533 if ((dev = dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001534 return 0;
1535
1536 s_t = cb->args[0];
1537 t = 0;
1538
David S. Miller30723672008-07-18 22:50:15 -07001539 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Miller8123b422008-08-08 23:23:39 -07001540 if (tc_dump_tclass_root(dev_queue->qdisc_sleeping, skb, tcm, cb, &t, s_t) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001541 goto done;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001542
David S. Miller30723672008-07-18 22:50:15 -07001543 dev_queue = &dev->rx_queue;
David S. Miller8123b422008-08-08 23:23:39 -07001544 if (tc_dump_tclass_root(dev_queue->qdisc_sleeping, skb, tcm, cb, &t, s_t) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001545 goto done;
1546
1547done:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001548 cb->args[0] = t;
1549
1550 dev_put(dev);
1551 return skb->len;
1552}
1553
1554/* Main classifier routine: scans classifier chain attached
1555 to this qdisc, (optionally) tests for protocol and asks
1556 specific classifiers.
1557 */
Patrick McHardy73ca4912007-07-15 00:02:31 -07001558int tc_classify_compat(struct sk_buff *skb, struct tcf_proto *tp,
1559 struct tcf_result *res)
1560{
1561 __be16 protocol = skb->protocol;
1562 int err = 0;
1563
1564 for (; tp; tp = tp->next) {
1565 if ((tp->protocol == protocol ||
1566 tp->protocol == htons(ETH_P_ALL)) &&
1567 (err = tp->classify(skb, tp, res)) >= 0) {
1568#ifdef CONFIG_NET_CLS_ACT
1569 if (err != TC_ACT_RECLASSIFY && skb->tc_verd)
1570 skb->tc_verd = SET_TC_VERD(skb->tc_verd, 0);
1571#endif
1572 return err;
1573 }
1574 }
1575 return -1;
1576}
1577EXPORT_SYMBOL(tc_classify_compat);
1578
Linus Torvalds1da177e2005-04-16 15:20:36 -07001579int tc_classify(struct sk_buff *skb, struct tcf_proto *tp,
Patrick McHardy73ca4912007-07-15 00:02:31 -07001580 struct tcf_result *res)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001581{
1582 int err = 0;
Patrick McHardy73ca4912007-07-15 00:02:31 -07001583 __be16 protocol;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001584#ifdef CONFIG_NET_CLS_ACT
1585 struct tcf_proto *otp = tp;
1586reclassify:
1587#endif
1588 protocol = skb->protocol;
1589
Patrick McHardy73ca4912007-07-15 00:02:31 -07001590 err = tc_classify_compat(skb, tp, res);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001591#ifdef CONFIG_NET_CLS_ACT
Patrick McHardy73ca4912007-07-15 00:02:31 -07001592 if (err == TC_ACT_RECLASSIFY) {
1593 u32 verd = G_TC_VERD(skb->tc_verd);
1594 tp = otp;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001595
Patrick McHardy73ca4912007-07-15 00:02:31 -07001596 if (verd++ >= MAX_REC_LOOP) {
1597 printk("rule prio %u protocol %02x reclassify loop, "
1598 "packet dropped\n",
1599 tp->prio&0xffff, ntohs(tp->protocol));
1600 return TC_ACT_SHOT;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001601 }
Patrick McHardy73ca4912007-07-15 00:02:31 -07001602 skb->tc_verd = SET_TC_VERD(skb->tc_verd, verd);
1603 goto reclassify;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001604 }
Patrick McHardy73ca4912007-07-15 00:02:31 -07001605#endif
1606 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001607}
Patrick McHardy73ca4912007-07-15 00:02:31 -07001608EXPORT_SYMBOL(tc_classify);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001609
Patrick McHardya48b5a62007-03-23 11:29:43 -07001610void tcf_destroy(struct tcf_proto *tp)
1611{
1612 tp->ops->destroy(tp);
1613 module_put(tp->ops->owner);
1614 kfree(tp);
1615}
1616
Patrick McHardyff31ab52008-07-01 19:52:38 -07001617void tcf_destroy_chain(struct tcf_proto **fl)
Patrick McHardya48b5a62007-03-23 11:29:43 -07001618{
1619 struct tcf_proto *tp;
1620
Patrick McHardyff31ab52008-07-01 19:52:38 -07001621 while ((tp = *fl) != NULL) {
1622 *fl = tp->next;
Patrick McHardya48b5a62007-03-23 11:29:43 -07001623 tcf_destroy(tp);
1624 }
1625}
1626EXPORT_SYMBOL(tcf_destroy_chain);
1627
Linus Torvalds1da177e2005-04-16 15:20:36 -07001628#ifdef CONFIG_PROC_FS
1629static int psched_show(struct seq_file *seq, void *v)
1630{
Patrick McHardy3c0cfc12007-10-10 16:32:41 -07001631 struct timespec ts;
1632
1633 hrtimer_get_res(CLOCK_MONOTONIC, &ts);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001634 seq_printf(seq, "%08x %08x %08x %08x\n",
Patrick McHardy641b9e02007-03-16 01:18:42 -07001635 (u32)NSEC_PER_USEC, (u32)PSCHED_US2NS(1),
Patrick McHardy514bca32007-03-16 12:34:52 -07001636 1000000,
Patrick McHardy3c0cfc12007-10-10 16:32:41 -07001637 (u32)NSEC_PER_SEC/(u32)ktime_to_ns(timespec_to_ktime(ts)));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001638
1639 return 0;
1640}
1641
1642static int psched_open(struct inode *inode, struct file *file)
1643{
1644 return single_open(file, psched_show, PDE(inode)->data);
1645}
1646
Arjan van de Venda7071d2007-02-12 00:55:36 -08001647static const struct file_operations psched_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001648 .owner = THIS_MODULE,
1649 .open = psched_open,
1650 .read = seq_read,
1651 .llseek = seq_lseek,
1652 .release = single_release,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001653};
Linus Torvalds1da177e2005-04-16 15:20:36 -07001654#endif
1655
Linus Torvalds1da177e2005-04-16 15:20:36 -07001656static int __init pktsched_init(void)
1657{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001658 register_qdisc(&pfifo_qdisc_ops);
1659 register_qdisc(&bfifo_qdisc_ops);
Eric W. Biederman457c4cb2007-09-12 12:01:34 +02001660 proc_net_fops_create(&init_net, "psched", 0, &psched_fops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001661
Thomas Grafbe577dd2007-03-22 11:55:50 -07001662 rtnl_register(PF_UNSPEC, RTM_NEWQDISC, tc_modify_qdisc, NULL);
1663 rtnl_register(PF_UNSPEC, RTM_DELQDISC, tc_get_qdisc, NULL);
1664 rtnl_register(PF_UNSPEC, RTM_GETQDISC, tc_get_qdisc, tc_dump_qdisc);
1665 rtnl_register(PF_UNSPEC, RTM_NEWTCLASS, tc_ctl_tclass, NULL);
1666 rtnl_register(PF_UNSPEC, RTM_DELTCLASS, tc_ctl_tclass, NULL);
1667 rtnl_register(PF_UNSPEC, RTM_GETTCLASS, tc_ctl_tclass, tc_dump_tclass);
1668
Linus Torvalds1da177e2005-04-16 15:20:36 -07001669 return 0;
1670}
1671
1672subsys_initcall(pktsched_init);