blob: 2a78d54101540f10562dbba2332271038f379694 [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
Jarek Poplawski99c0db22008-10-31 00:45:27 -0700100 ---peek
101
102 like dequeue but without removing a packet from the queue
103
Linus Torvalds1da177e2005-04-16 15:20:36 -0700104 ---reset
105
106 returns qdisc to initial state: purge all buffers, clear all
107 timers, counters (except for statistics) etc.
108
109 ---init
110
111 initializes newly created qdisc.
112
113 ---destroy
114
115 destroys resources allocated by init and during lifetime of qdisc.
116
117 ---change
118
119 changes qdisc parameters.
120 */
121
122/* Protects list of registered TC modules. It is pure SMP lock. */
123static DEFINE_RWLOCK(qdisc_mod_lock);
124
125
126/************************************************
127 * Queueing disciplines manipulation. *
128 ************************************************/
129
130
131/* The list of all installed queueing disciplines. */
132
133static struct Qdisc_ops *qdisc_base;
134
135/* Register/uregister queueing discipline */
136
137int register_qdisc(struct Qdisc_ops *qops)
138{
139 struct Qdisc_ops *q, **qp;
140 int rc = -EEXIST;
141
142 write_lock(&qdisc_mod_lock);
143 for (qp = &qdisc_base; (q = *qp) != NULL; qp = &q->next)
144 if (!strcmp(qops->id, q->id))
145 goto out;
146
147 if (qops->enqueue == NULL)
148 qops->enqueue = noop_qdisc_ops.enqueue;
Jarek Poplawski99c0db22008-10-31 00:45:27 -0700149 if (qops->peek == NULL) {
150 if (qops->dequeue == NULL) {
151 qops->peek = noop_qdisc_ops.peek;
152 } else {
153 rc = -EINVAL;
154 goto out;
155 }
156 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700157 if (qops->dequeue == NULL)
158 qops->dequeue = noop_qdisc_ops.dequeue;
159
160 qops->next = NULL;
161 *qp = qops;
162 rc = 0;
163out:
164 write_unlock(&qdisc_mod_lock);
165 return rc;
166}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800167EXPORT_SYMBOL(register_qdisc);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700168
169int unregister_qdisc(struct Qdisc_ops *qops)
170{
171 struct Qdisc_ops *q, **qp;
172 int err = -ENOENT;
173
174 write_lock(&qdisc_mod_lock);
175 for (qp = &qdisc_base; (q=*qp)!=NULL; qp = &q->next)
176 if (q == qops)
177 break;
178 if (q) {
179 *qp = q->next;
180 q->next = NULL;
181 err = 0;
182 }
183 write_unlock(&qdisc_mod_lock);
184 return err;
185}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800186EXPORT_SYMBOL(unregister_qdisc);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700187
188/* We know handle. Find qdisc among all qdisc's attached to device
189 (root qdisc, all its children, children of children etc.)
190 */
191
Hannes Eder6113b742008-11-28 03:06:46 -0800192static struct Qdisc *qdisc_match_from_root(struct Qdisc *root, u32 handle)
David S. Miller8123b422008-08-08 23:23:39 -0700193{
194 struct Qdisc *q;
195
196 if (!(root->flags & TCQ_F_BUILTIN) &&
197 root->handle == handle)
198 return root;
199
200 list_for_each_entry(q, &root->list, list) {
201 if (q->handle == handle)
202 return q;
203 }
204 return NULL;
205}
206
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700207static void qdisc_list_add(struct Qdisc *q)
208{
Jarek Poplawskif6486d42008-11-25 13:56:06 -0800209 if ((q->parent != TC_H_ROOT) && !(q->flags & TCQ_F_INGRESS))
Patrick McHardyaf356af2009-09-04 06:41:18 +0000210 list_add_tail(&q->list, &qdisc_dev(q)->qdisc->list);
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700211}
212
213void qdisc_list_del(struct Qdisc *q)
214{
Jarek Poplawskif6486d42008-11-25 13:56:06 -0800215 if ((q->parent != TC_H_ROOT) && !(q->flags & TCQ_F_INGRESS))
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700216 list_del(&q->list);
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700217}
218EXPORT_SYMBOL(qdisc_list_del);
219
David S. Milleread81cc2008-07-17 00:50:32 -0700220struct Qdisc *qdisc_lookup(struct net_device *dev, u32 handle)
Patrick McHardy43effa12006-11-29 17:35:48 -0800221{
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700222 struct Qdisc *q;
223
Patrick McHardyaf356af2009-09-04 06:41:18 +0000224 q = qdisc_match_from_root(dev->qdisc, handle);
225 if (q)
226 goto out;
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700227
228 q = qdisc_match_from_root(dev->rx_queue.qdisc_sleeping, handle);
Jarek Poplawskif6486d42008-11-25 13:56:06 -0800229out:
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700230 return q;
Patrick McHardy43effa12006-11-29 17:35:48 -0800231}
232
Linus Torvalds1da177e2005-04-16 15:20:36 -0700233static struct Qdisc *qdisc_leaf(struct Qdisc *p, u32 classid)
234{
235 unsigned long cl;
236 struct Qdisc *leaf;
Eric Dumazet20fea082007-11-14 01:44:41 -0800237 const struct Qdisc_class_ops *cops = p->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700238
239 if (cops == NULL)
240 return NULL;
241 cl = cops->get(p, classid);
242
243 if (cl == 0)
244 return NULL;
245 leaf = cops->leaf(p, cl);
246 cops->put(p, cl);
247 return leaf;
248}
249
250/* Find queueing discipline by name */
251
Patrick McHardy1e904742008-01-22 22:11:17 -0800252static struct Qdisc_ops *qdisc_lookup_ops(struct nlattr *kind)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700253{
254 struct Qdisc_ops *q = NULL;
255
256 if (kind) {
257 read_lock(&qdisc_mod_lock);
258 for (q = qdisc_base; q; q = q->next) {
Patrick McHardy1e904742008-01-22 22:11:17 -0800259 if (nla_strcmp(kind, q->id) == 0) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700260 if (!try_module_get(q->owner))
261 q = NULL;
262 break;
263 }
264 }
265 read_unlock(&qdisc_mod_lock);
266 }
267 return q;
268}
269
270static struct qdisc_rate_table *qdisc_rtab_list;
271
Patrick McHardy1e904742008-01-22 22:11:17 -0800272struct qdisc_rate_table *qdisc_get_rtab(struct tc_ratespec *r, struct nlattr *tab)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700273{
274 struct qdisc_rate_table *rtab;
275
276 for (rtab = qdisc_rtab_list; rtab; rtab = rtab->next) {
277 if (memcmp(&rtab->rate, r, sizeof(struct tc_ratespec)) == 0) {
278 rtab->refcnt++;
279 return rtab;
280 }
281 }
282
Patrick McHardy5feb5e12008-01-23 20:35:19 -0800283 if (tab == NULL || r->rate == 0 || r->cell_log == 0 ||
284 nla_len(tab) != TC_RTAB_SIZE)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700285 return NULL;
286
287 rtab = kmalloc(sizeof(*rtab), GFP_KERNEL);
288 if (rtab) {
289 rtab->rate = *r;
290 rtab->refcnt = 1;
Patrick McHardy1e904742008-01-22 22:11:17 -0800291 memcpy(rtab->data, nla_data(tab), 1024);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700292 rtab->next = qdisc_rtab_list;
293 qdisc_rtab_list = rtab;
294 }
295 return rtab;
296}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800297EXPORT_SYMBOL(qdisc_get_rtab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700298
299void qdisc_put_rtab(struct qdisc_rate_table *tab)
300{
301 struct qdisc_rate_table *rtab, **rtabp;
302
303 if (!tab || --tab->refcnt)
304 return;
305
306 for (rtabp = &qdisc_rtab_list; (rtab=*rtabp) != NULL; rtabp = &rtab->next) {
307 if (rtab == tab) {
308 *rtabp = rtab->next;
309 kfree(rtab);
310 return;
311 }
312 }
313}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800314EXPORT_SYMBOL(qdisc_put_rtab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700315
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700316static LIST_HEAD(qdisc_stab_list);
317static DEFINE_SPINLOCK(qdisc_stab_lock);
318
319static const struct nla_policy stab_policy[TCA_STAB_MAX + 1] = {
320 [TCA_STAB_BASE] = { .len = sizeof(struct tc_sizespec) },
321 [TCA_STAB_DATA] = { .type = NLA_BINARY },
322};
323
324static struct qdisc_size_table *qdisc_get_stab(struct nlattr *opt)
325{
326 struct nlattr *tb[TCA_STAB_MAX + 1];
327 struct qdisc_size_table *stab;
328 struct tc_sizespec *s;
329 unsigned int tsize = 0;
330 u16 *tab = NULL;
331 int err;
332
333 err = nla_parse_nested(tb, TCA_STAB_MAX, opt, stab_policy);
334 if (err < 0)
335 return ERR_PTR(err);
336 if (!tb[TCA_STAB_BASE])
337 return ERR_PTR(-EINVAL);
338
339 s = nla_data(tb[TCA_STAB_BASE]);
340
341 if (s->tsize > 0) {
342 if (!tb[TCA_STAB_DATA])
343 return ERR_PTR(-EINVAL);
344 tab = nla_data(tb[TCA_STAB_DATA]);
345 tsize = nla_len(tb[TCA_STAB_DATA]) / sizeof(u16);
346 }
347
348 if (!s || tsize != s->tsize || (!tab && tsize > 0))
349 return ERR_PTR(-EINVAL);
350
David S. Millerf3b96052008-08-18 22:33:05 -0700351 spin_lock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700352
353 list_for_each_entry(stab, &qdisc_stab_list, list) {
354 if (memcmp(&stab->szopts, s, sizeof(*s)))
355 continue;
356 if (tsize > 0 && memcmp(stab->data, tab, tsize * sizeof(u16)))
357 continue;
358 stab->refcnt++;
David S. Millerf3b96052008-08-18 22:33:05 -0700359 spin_unlock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700360 return stab;
361 }
362
David S. Millerf3b96052008-08-18 22:33:05 -0700363 spin_unlock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700364
365 stab = kmalloc(sizeof(*stab) + tsize * sizeof(u16), GFP_KERNEL);
366 if (!stab)
367 return ERR_PTR(-ENOMEM);
368
369 stab->refcnt = 1;
370 stab->szopts = *s;
371 if (tsize > 0)
372 memcpy(stab->data, tab, tsize * sizeof(u16));
373
David S. Millerf3b96052008-08-18 22:33:05 -0700374 spin_lock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700375 list_add_tail(&stab->list, &qdisc_stab_list);
David S. Millerf3b96052008-08-18 22:33:05 -0700376 spin_unlock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700377
378 return stab;
379}
380
381void qdisc_put_stab(struct qdisc_size_table *tab)
382{
383 if (!tab)
384 return;
385
David S. Millerf3b96052008-08-18 22:33:05 -0700386 spin_lock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700387
388 if (--tab->refcnt == 0) {
389 list_del(&tab->list);
390 kfree(tab);
391 }
392
David S. Millerf3b96052008-08-18 22:33:05 -0700393 spin_unlock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700394}
395EXPORT_SYMBOL(qdisc_put_stab);
396
397static int qdisc_dump_stab(struct sk_buff *skb, struct qdisc_size_table *stab)
398{
399 struct nlattr *nest;
400
401 nest = nla_nest_start(skb, TCA_STAB);
Patrick McHardy3aa46142008-11-20 04:07:14 -0800402 if (nest == NULL)
403 goto nla_put_failure;
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700404 NLA_PUT(skb, TCA_STAB_BASE, sizeof(stab->szopts), &stab->szopts);
405 nla_nest_end(skb, nest);
406
407 return skb->len;
408
409nla_put_failure:
410 return -1;
411}
412
413void qdisc_calculate_pkt_len(struct sk_buff *skb, struct qdisc_size_table *stab)
414{
415 int pkt_len, slot;
416
417 pkt_len = skb->len + stab->szopts.overhead;
418 if (unlikely(!stab->szopts.tsize))
419 goto out;
420
421 slot = pkt_len + stab->szopts.cell_align;
422 if (unlikely(slot < 0))
423 slot = 0;
424
425 slot >>= stab->szopts.cell_log;
426 if (likely(slot < stab->szopts.tsize))
427 pkt_len = stab->data[slot];
428 else
429 pkt_len = stab->data[stab->szopts.tsize - 1] *
430 (slot / stab->szopts.tsize) +
431 stab->data[slot % stab->szopts.tsize];
432
433 pkt_len <<= stab->szopts.size_log;
434out:
435 if (unlikely(pkt_len < 1))
436 pkt_len = 1;
437 qdisc_skb_cb(skb)->pkt_len = pkt_len;
438}
439EXPORT_SYMBOL(qdisc_calculate_pkt_len);
440
Jarek Poplawskib00355d2009-02-01 01:12:42 -0800441void qdisc_warn_nonwc(char *txt, struct Qdisc *qdisc)
442{
443 if (!(qdisc->flags & TCQ_F_WARN_NONWC)) {
444 printk(KERN_WARNING
445 "%s: %s qdisc %X: is non-work-conserving?\n",
446 txt, qdisc->ops->id, qdisc->handle >> 16);
447 qdisc->flags |= TCQ_F_WARN_NONWC;
448 }
449}
450EXPORT_SYMBOL(qdisc_warn_nonwc);
451
Patrick McHardy41794772007-03-16 01:19:15 -0700452static enum hrtimer_restart qdisc_watchdog(struct hrtimer *timer)
453{
454 struct qdisc_watchdog *wd = container_of(timer, struct qdisc_watchdog,
455 timer);
456
457 wd->qdisc->flags &= ~TCQ_F_THROTTLED;
David S. Miller8608db02008-08-18 20:51:18 -0700458 __netif_schedule(qdisc_root(wd->qdisc));
Stephen Hemminger19365022007-03-22 12:18:35 -0700459
Patrick McHardy41794772007-03-16 01:19:15 -0700460 return HRTIMER_NORESTART;
461}
462
463void qdisc_watchdog_init(struct qdisc_watchdog *wd, struct Qdisc *qdisc)
464{
465 hrtimer_init(&wd->timer, CLOCK_MONOTONIC, HRTIMER_MODE_ABS);
466 wd->timer.function = qdisc_watchdog;
467 wd->qdisc = qdisc;
468}
469EXPORT_SYMBOL(qdisc_watchdog_init);
470
471void qdisc_watchdog_schedule(struct qdisc_watchdog *wd, psched_time_t expires)
472{
473 ktime_t time;
474
Jarek Poplawski2540e052008-08-21 05:11:14 -0700475 if (test_bit(__QDISC_STATE_DEACTIVATED,
476 &qdisc_root_sleeping(wd->qdisc)->state))
477 return;
478
Patrick McHardy41794772007-03-16 01:19:15 -0700479 wd->qdisc->flags |= TCQ_F_THROTTLED;
480 time = ktime_set(0, 0);
Jarek Poplawskica44d6e2009-06-15 02:31:47 -0700481 time = ktime_add_ns(time, PSCHED_TICKS2NS(expires));
Patrick McHardy41794772007-03-16 01:19:15 -0700482 hrtimer_start(&wd->timer, time, HRTIMER_MODE_ABS);
483}
484EXPORT_SYMBOL(qdisc_watchdog_schedule);
485
486void qdisc_watchdog_cancel(struct qdisc_watchdog *wd)
487{
488 hrtimer_cancel(&wd->timer);
489 wd->qdisc->flags &= ~TCQ_F_THROTTLED;
490}
491EXPORT_SYMBOL(qdisc_watchdog_cancel);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700492
Adrian Bunka94f7792008-07-22 14:20:11 -0700493static struct hlist_head *qdisc_class_hash_alloc(unsigned int n)
Patrick McHardy6fe1c7a2008-07-05 23:21:31 -0700494{
495 unsigned int size = n * sizeof(struct hlist_head), i;
496 struct hlist_head *h;
497
498 if (size <= PAGE_SIZE)
499 h = kmalloc(size, GFP_KERNEL);
500 else
501 h = (struct hlist_head *)
502 __get_free_pages(GFP_KERNEL, get_order(size));
503
504 if (h != NULL) {
505 for (i = 0; i < n; i++)
506 INIT_HLIST_HEAD(&h[i]);
507 }
508 return h;
509}
510
511static void qdisc_class_hash_free(struct hlist_head *h, unsigned int n)
512{
513 unsigned int size = n * sizeof(struct hlist_head);
514
515 if (size <= PAGE_SIZE)
516 kfree(h);
517 else
518 free_pages((unsigned long)h, get_order(size));
519}
520
521void qdisc_class_hash_grow(struct Qdisc *sch, struct Qdisc_class_hash *clhash)
522{
523 struct Qdisc_class_common *cl;
524 struct hlist_node *n, *next;
525 struct hlist_head *nhash, *ohash;
526 unsigned int nsize, nmask, osize;
527 unsigned int i, h;
528
529 /* Rehash when load factor exceeds 0.75 */
530 if (clhash->hashelems * 4 <= clhash->hashsize * 3)
531 return;
532 nsize = clhash->hashsize * 2;
533 nmask = nsize - 1;
534 nhash = qdisc_class_hash_alloc(nsize);
535 if (nhash == NULL)
536 return;
537
538 ohash = clhash->hash;
539 osize = clhash->hashsize;
540
541 sch_tree_lock(sch);
542 for (i = 0; i < osize; i++) {
543 hlist_for_each_entry_safe(cl, n, next, &ohash[i], hnode) {
544 h = qdisc_class_hash(cl->classid, nmask);
545 hlist_add_head(&cl->hnode, &nhash[h]);
546 }
547 }
548 clhash->hash = nhash;
549 clhash->hashsize = nsize;
550 clhash->hashmask = nmask;
551 sch_tree_unlock(sch);
552
553 qdisc_class_hash_free(ohash, osize);
554}
555EXPORT_SYMBOL(qdisc_class_hash_grow);
556
557int qdisc_class_hash_init(struct Qdisc_class_hash *clhash)
558{
559 unsigned int size = 4;
560
561 clhash->hash = qdisc_class_hash_alloc(size);
562 if (clhash->hash == NULL)
563 return -ENOMEM;
564 clhash->hashsize = size;
565 clhash->hashmask = size - 1;
566 clhash->hashelems = 0;
567 return 0;
568}
569EXPORT_SYMBOL(qdisc_class_hash_init);
570
571void qdisc_class_hash_destroy(struct Qdisc_class_hash *clhash)
572{
573 qdisc_class_hash_free(clhash->hash, clhash->hashsize);
574}
575EXPORT_SYMBOL(qdisc_class_hash_destroy);
576
577void qdisc_class_hash_insert(struct Qdisc_class_hash *clhash,
578 struct Qdisc_class_common *cl)
579{
580 unsigned int h;
581
582 INIT_HLIST_NODE(&cl->hnode);
583 h = qdisc_class_hash(cl->classid, clhash->hashmask);
584 hlist_add_head(&cl->hnode, &clhash->hash[h]);
585 clhash->hashelems++;
586}
587EXPORT_SYMBOL(qdisc_class_hash_insert);
588
589void qdisc_class_hash_remove(struct Qdisc_class_hash *clhash,
590 struct Qdisc_class_common *cl)
591{
592 hlist_del(&cl->hnode);
593 clhash->hashelems--;
594}
595EXPORT_SYMBOL(qdisc_class_hash_remove);
596
Linus Torvalds1da177e2005-04-16 15:20:36 -0700597/* Allocate an unique handle from space managed by kernel */
598
599static u32 qdisc_alloc_handle(struct net_device *dev)
600{
601 int i = 0x10000;
602 static u32 autohandle = TC_H_MAKE(0x80000000U, 0);
603
604 do {
605 autohandle += TC_H_MAKE(0x10000U, 0);
606 if (autohandle == TC_H_MAKE(TC_H_ROOT, 0))
607 autohandle = TC_H_MAKE(0x80000000U, 0);
608 } while (qdisc_lookup(dev, autohandle) && --i > 0);
609
610 return i>0 ? autohandle : 0;
611}
612
Patrick McHardy43effa12006-11-29 17:35:48 -0800613void qdisc_tree_decrease_qlen(struct Qdisc *sch, unsigned int n)
614{
Eric Dumazet20fea082007-11-14 01:44:41 -0800615 const struct Qdisc_class_ops *cops;
Patrick McHardy43effa12006-11-29 17:35:48 -0800616 unsigned long cl;
617 u32 parentid;
618
619 if (n == 0)
620 return;
621 while ((parentid = sch->parent)) {
Jarek Poplawski066a3b52008-04-14 15:10:42 -0700622 if (TC_H_MAJ(parentid) == TC_H_MAJ(TC_H_INGRESS))
623 return;
624
David S. Miller5ce2d482008-07-08 17:06:30 -0700625 sch = qdisc_lookup(qdisc_dev(sch), TC_H_MAJ(parentid));
Patrick McHardyffc8fef2007-07-30 17:11:50 -0700626 if (sch == NULL) {
627 WARN_ON(parentid != TC_H_ROOT);
628 return;
629 }
Patrick McHardy43effa12006-11-29 17:35:48 -0800630 cops = sch->ops->cl_ops;
631 if (cops->qlen_notify) {
632 cl = cops->get(sch, parentid);
633 cops->qlen_notify(sch, cl);
634 cops->put(sch, cl);
635 }
636 sch->q.qlen -= n;
637 }
638}
639EXPORT_SYMBOL(qdisc_tree_decrease_qlen);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700640
David S. Miller99194cf2008-07-17 04:54:10 -0700641static void notify_and_destroy(struct sk_buff *skb, struct nlmsghdr *n, u32 clid,
642 struct Qdisc *old, struct Qdisc *new)
643{
644 if (new || old)
645 qdisc_notify(skb, n, clid, old, new);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700646
David S. Miller4d8863a2008-08-18 21:03:15 -0700647 if (old)
David S. Miller99194cf2008-07-17 04:54:10 -0700648 qdisc_destroy(old);
David S. Miller99194cf2008-07-17 04:54:10 -0700649}
650
651/* Graft qdisc "new" to class "classid" of qdisc "parent" or
652 * to device "dev".
653 *
654 * When appropriate send a netlink notification using 'skb'
655 * and "n".
656 *
657 * On success, destroy old qdisc.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700658 */
659
660static int qdisc_graft(struct net_device *dev, struct Qdisc *parent,
David S. Miller99194cf2008-07-17 04:54:10 -0700661 struct sk_buff *skb, struct nlmsghdr *n, u32 classid,
662 struct Qdisc *new, struct Qdisc *old)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700663{
David S. Miller99194cf2008-07-17 04:54:10 -0700664 struct Qdisc *q = old;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700665 int err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700666
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900667 if (parent == NULL) {
David S. Miller99194cf2008-07-17 04:54:10 -0700668 unsigned int i, num_q, ingress;
669
670 ingress = 0;
671 num_q = dev->num_tx_queues;
David S. Miller8d50b532008-07-30 02:37:46 -0700672 if ((q && q->flags & TCQ_F_INGRESS) ||
673 (new && new->flags & TCQ_F_INGRESS)) {
David S. Miller99194cf2008-07-17 04:54:10 -0700674 num_q = 1;
675 ingress = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700676 }
David S. Miller99194cf2008-07-17 04:54:10 -0700677
678 if (dev->flags & IFF_UP)
679 dev_deactivate(dev);
680
David S. Miller6ec1c692009-09-06 01:58:51 -0700681 if (new && new->ops->attach) {
682 new->ops->attach(new);
683 num_q = 0;
684 }
685
David S. Miller99194cf2008-07-17 04:54:10 -0700686 for (i = 0; i < num_q; i++) {
687 struct netdev_queue *dev_queue = &dev->rx_queue;
688
689 if (!ingress)
690 dev_queue = netdev_get_tx_queue(dev, i);
691
David S. Miller8d50b532008-07-30 02:37:46 -0700692 old = dev_graft_qdisc(dev_queue, new);
693 if (new && i > 0)
694 atomic_inc(&new->refcnt);
695
Patrick McHardyaf356af2009-09-04 06:41:18 +0000696 qdisc_destroy(old);
David S. Miller99194cf2008-07-17 04:54:10 -0700697 }
698
Patrick McHardyaf356af2009-09-04 06:41:18 +0000699 notify_and_destroy(skb, n, classid, dev->qdisc, new);
David S. Miller6ec1c692009-09-06 01:58:51 -0700700 if (new && !new->ops->attach)
Patrick McHardyaf356af2009-09-04 06:41:18 +0000701 atomic_inc(&new->refcnt);
702 dev->qdisc = new ? : &noop_qdisc;
703
David S. Miller99194cf2008-07-17 04:54:10 -0700704 if (dev->flags & IFF_UP)
705 dev_activate(dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700706 } else {
Eric Dumazet20fea082007-11-14 01:44:41 -0800707 const struct Qdisc_class_ops *cops = parent->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700708
Patrick McHardyc9f1d032009-09-04 06:41:13 +0000709 err = -EOPNOTSUPP;
710 if (cops && cops->graft) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700711 unsigned long cl = cops->get(parent, classid);
712 if (cl) {
David S. Miller99194cf2008-07-17 04:54:10 -0700713 err = cops->graft(parent, cl, new, &old);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700714 cops->put(parent, cl);
Patrick McHardyc9f1d032009-09-04 06:41:13 +0000715 } else
716 err = -ENOENT;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700717 }
David S. Miller99194cf2008-07-17 04:54:10 -0700718 if (!err)
719 notify_and_destroy(skb, n, classid, old, new);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700720 }
721 return err;
722}
723
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700724/* lockdep annotation is needed for ingress; egress gets it only for name */
725static struct lock_class_key qdisc_tx_lock;
726static struct lock_class_key qdisc_rx_lock;
727
Linus Torvalds1da177e2005-04-16 15:20:36 -0700728/*
729 Allocate and initialize new qdisc.
730
731 Parameters are passed via opt.
732 */
733
734static struct Qdisc *
David S. Millerbb949fb2008-07-08 16:55:56 -0700735qdisc_create(struct net_device *dev, struct netdev_queue *dev_queue,
736 u32 parent, u32 handle, struct nlattr **tca, int *errp)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700737{
738 int err;
Patrick McHardy1e904742008-01-22 22:11:17 -0800739 struct nlattr *kind = tca[TCA_KIND];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700740 struct Qdisc *sch;
741 struct Qdisc_ops *ops;
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700742 struct qdisc_size_table *stab;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700743
744 ops = qdisc_lookup_ops(kind);
Johannes Berg95a5afc2008-10-16 15:24:51 -0700745#ifdef CONFIG_MODULES
Linus Torvalds1da177e2005-04-16 15:20:36 -0700746 if (ops == NULL && kind != NULL) {
747 char name[IFNAMSIZ];
Patrick McHardy1e904742008-01-22 22:11:17 -0800748 if (nla_strlcpy(name, kind, IFNAMSIZ) < IFNAMSIZ) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700749 /* We dropped the RTNL semaphore in order to
750 * perform the module load. So, even if we
751 * succeeded in loading the module we have to
752 * tell the caller to replay the request. We
753 * indicate this using -EAGAIN.
754 * We replay the request because the device may
755 * go away in the mean time.
756 */
757 rtnl_unlock();
758 request_module("sch_%s", name);
759 rtnl_lock();
760 ops = qdisc_lookup_ops(kind);
761 if (ops != NULL) {
762 /* We will try again qdisc_lookup_ops,
763 * so don't keep a reference.
764 */
765 module_put(ops->owner);
766 err = -EAGAIN;
767 goto err_out;
768 }
769 }
770 }
771#endif
772
Jamal Hadi Salimb9e2cc02006-08-03 16:36:51 -0700773 err = -ENOENT;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700774 if (ops == NULL)
775 goto err_out;
776
David S. Miller5ce2d482008-07-08 17:06:30 -0700777 sch = qdisc_alloc(dev_queue, ops);
Thomas Graf3d54b822005-07-05 14:15:09 -0700778 if (IS_ERR(sch)) {
779 err = PTR_ERR(sch);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700780 goto err_out2;
Thomas Graf3d54b822005-07-05 14:15:09 -0700781 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700782
Patrick McHardyffc8fef2007-07-30 17:11:50 -0700783 sch->parent = parent;
784
Thomas Graf3d54b822005-07-05 14:15:09 -0700785 if (handle == TC_H_INGRESS) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700786 sch->flags |= TCQ_F_INGRESS;
Thomas Graf3d54b822005-07-05 14:15:09 -0700787 handle = TC_H_MAKE(TC_H_INGRESS, 0);
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700788 lockdep_set_class(qdisc_lock(sch), &qdisc_rx_lock);
Patrick McHardyfd44de72007-04-16 17:07:08 -0700789 } else {
Patrick McHardyfd44de72007-04-16 17:07:08 -0700790 if (handle == 0) {
791 handle = qdisc_alloc_handle(dev);
792 err = -ENOMEM;
793 if (handle == 0)
794 goto err_out3;
795 }
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700796 lockdep_set_class(qdisc_lock(sch), &qdisc_tx_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700797 }
798
Thomas Graf3d54b822005-07-05 14:15:09 -0700799 sch->handle = handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700800
Patrick McHardy1e904742008-01-22 22:11:17 -0800801 if (!ops->init || (err = ops->init(sch, tca[TCA_OPTIONS])) == 0) {
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700802 if (tca[TCA_STAB]) {
803 stab = qdisc_get_stab(tca[TCA_STAB]);
804 if (IS_ERR(stab)) {
805 err = PTR_ERR(stab);
806 goto err_out3;
807 }
808 sch->stab = stab;
809 }
Patrick McHardy1e904742008-01-22 22:11:17 -0800810 if (tca[TCA_RATE]) {
Jarek Poplawskif6f9b932008-08-27 02:25:17 -0700811 spinlock_t *root_lock;
812
813 if ((sch->parent != TC_H_ROOT) &&
814 !(sch->flags & TCQ_F_INGRESS))
815 root_lock = qdisc_root_sleeping_lock(sch);
816 else
817 root_lock = qdisc_lock(sch);
818
Thomas Graf023e09a2005-07-05 14:15:53 -0700819 err = gen_new_estimator(&sch->bstats, &sch->rate_est,
Jarek Poplawskif6f9b932008-08-27 02:25:17 -0700820 root_lock, tca[TCA_RATE]);
Thomas Graf023e09a2005-07-05 14:15:53 -0700821 if (err) {
822 /*
823 * Any broken qdiscs that would require
824 * a ops->reset() here? The qdisc was never
825 * in action so it shouldn't be necessary.
826 */
827 if (ops->destroy)
828 ops->destroy(sch);
829 goto err_out3;
830 }
831 }
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700832
833 qdisc_list_add(sch);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700834
Linus Torvalds1da177e2005-04-16 15:20:36 -0700835 return sch;
836 }
837err_out3:
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700838 qdisc_put_stab(sch->stab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700839 dev_put(dev);
Thomas Graf3d54b822005-07-05 14:15:09 -0700840 kfree((char *) sch - sch->padded);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700841err_out2:
842 module_put(ops->owner);
843err_out:
844 *errp = err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700845 return NULL;
846}
847
Patrick McHardy1e904742008-01-22 22:11:17 -0800848static int qdisc_change(struct Qdisc *sch, struct nlattr **tca)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700849{
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700850 struct qdisc_size_table *stab = NULL;
851 int err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700852
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700853 if (tca[TCA_OPTIONS]) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700854 if (sch->ops->change == NULL)
855 return -EINVAL;
Patrick McHardy1e904742008-01-22 22:11:17 -0800856 err = sch->ops->change(sch, tca[TCA_OPTIONS]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700857 if (err)
858 return err;
859 }
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700860
861 if (tca[TCA_STAB]) {
862 stab = qdisc_get_stab(tca[TCA_STAB]);
863 if (IS_ERR(stab))
864 return PTR_ERR(stab);
865 }
866
867 qdisc_put_stab(sch->stab);
868 sch->stab = stab;
869
Patrick McHardy1e904742008-01-22 22:11:17 -0800870 if (tca[TCA_RATE])
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800871 /* NB: ignores errors from replace_estimator
872 because change can't be undone. */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700873 gen_replace_estimator(&sch->bstats, &sch->rate_est,
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800874 qdisc_root_sleeping_lock(sch),
875 tca[TCA_RATE]);
876
Linus Torvalds1da177e2005-04-16 15:20:36 -0700877 return 0;
878}
879
880struct check_loop_arg
881{
882 struct qdisc_walker w;
883 struct Qdisc *p;
884 int depth;
885};
886
887static int check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w);
888
889static int check_loop(struct Qdisc *q, struct Qdisc *p, int depth)
890{
891 struct check_loop_arg arg;
892
893 if (q->ops->cl_ops == NULL)
894 return 0;
895
896 arg.w.stop = arg.w.skip = arg.w.count = 0;
897 arg.w.fn = check_loop_fn;
898 arg.depth = depth;
899 arg.p = p;
900 q->ops->cl_ops->walk(q, &arg.w);
901 return arg.w.stop ? -ELOOP : 0;
902}
903
904static int
905check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w)
906{
907 struct Qdisc *leaf;
Eric Dumazet20fea082007-11-14 01:44:41 -0800908 const struct Qdisc_class_ops *cops = q->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700909 struct check_loop_arg *arg = (struct check_loop_arg *)w;
910
911 leaf = cops->leaf(q, cl);
912 if (leaf) {
913 if (leaf == arg->p || arg->depth > 7)
914 return -ELOOP;
915 return check_loop(leaf, arg->p, arg->depth + 1);
916 }
917 return 0;
918}
919
920/*
921 * Delete/get qdisc.
922 */
923
924static int tc_get_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
925{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900926 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700927 struct tcmsg *tcm = NLMSG_DATA(n);
Patrick McHardy1e904742008-01-22 22:11:17 -0800928 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700929 struct net_device *dev;
930 u32 clid = tcm->tcm_parent;
931 struct Qdisc *q = NULL;
932 struct Qdisc *p = NULL;
933 int err;
934
Denis V. Lunevb8542722007-12-01 00:21:31 +1100935 if (net != &init_net)
936 return -EINVAL;
937
Eric W. Biederman881d9662007-09-17 11:56:21 -0700938 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700939 return -ENODEV;
940
Patrick McHardy1e904742008-01-22 22:11:17 -0800941 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
942 if (err < 0)
943 return err;
944
Linus Torvalds1da177e2005-04-16 15:20:36 -0700945 if (clid) {
946 if (clid != TC_H_ROOT) {
947 if (TC_H_MAJ(clid) != TC_H_MAJ(TC_H_INGRESS)) {
948 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
949 return -ENOENT;
950 q = qdisc_leaf(p, clid);
951 } else { /* ingress */
David S. Miller8123b422008-08-08 23:23:39 -0700952 q = dev->rx_queue.qdisc_sleeping;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900953 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700954 } else {
Patrick McHardyaf356af2009-09-04 06:41:18 +0000955 q = dev->qdisc;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700956 }
957 if (!q)
958 return -ENOENT;
959
960 if (tcm->tcm_handle && q->handle != tcm->tcm_handle)
961 return -EINVAL;
962 } else {
963 if ((q = qdisc_lookup(dev, tcm->tcm_handle)) == NULL)
964 return -ENOENT;
965 }
966
Patrick McHardy1e904742008-01-22 22:11:17 -0800967 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700968 return -EINVAL;
969
970 if (n->nlmsg_type == RTM_DELQDISC) {
971 if (!clid)
972 return -EINVAL;
973 if (q->handle == 0)
974 return -ENOENT;
David S. Miller99194cf2008-07-17 04:54:10 -0700975 if ((err = qdisc_graft(dev, p, skb, n, clid, NULL, q)) != 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700976 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700977 } else {
978 qdisc_notify(skb, n, clid, NULL, q);
979 }
980 return 0;
981}
982
983/*
984 Create/change qdisc.
985 */
986
987static int tc_modify_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
988{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900989 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700990 struct tcmsg *tcm;
Patrick McHardy1e904742008-01-22 22:11:17 -0800991 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700992 struct net_device *dev;
993 u32 clid;
994 struct Qdisc *q, *p;
995 int err;
996
Denis V. Lunevb8542722007-12-01 00:21:31 +1100997 if (net != &init_net)
998 return -EINVAL;
999
Linus Torvalds1da177e2005-04-16 15:20:36 -07001000replay:
1001 /* Reinit, just in case something touches this. */
1002 tcm = NLMSG_DATA(n);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001003 clid = tcm->tcm_parent;
1004 q = p = NULL;
1005
Eric W. Biederman881d9662007-09-17 11:56:21 -07001006 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001007 return -ENODEV;
1008
Patrick McHardy1e904742008-01-22 22:11:17 -08001009 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1010 if (err < 0)
1011 return err;
1012
Linus Torvalds1da177e2005-04-16 15:20:36 -07001013 if (clid) {
1014 if (clid != TC_H_ROOT) {
1015 if (clid != TC_H_INGRESS) {
1016 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
1017 return -ENOENT;
1018 q = qdisc_leaf(p, clid);
1019 } else { /*ingress */
David S. Miller8123b422008-08-08 23:23:39 -07001020 q = dev->rx_queue.qdisc_sleeping;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001021 }
1022 } else {
Patrick McHardyaf356af2009-09-04 06:41:18 +00001023 q = dev->qdisc;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001024 }
1025
1026 /* It may be default qdisc, ignore it */
1027 if (q && q->handle == 0)
1028 q = NULL;
1029
1030 if (!q || !tcm->tcm_handle || q->handle != tcm->tcm_handle) {
1031 if (tcm->tcm_handle) {
1032 if (q && !(n->nlmsg_flags&NLM_F_REPLACE))
1033 return -EEXIST;
1034 if (TC_H_MIN(tcm->tcm_handle))
1035 return -EINVAL;
1036 if ((q = qdisc_lookup(dev, tcm->tcm_handle)) == NULL)
1037 goto create_n_graft;
1038 if (n->nlmsg_flags&NLM_F_EXCL)
1039 return -EEXIST;
Patrick McHardy1e904742008-01-22 22:11:17 -08001040 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001041 return -EINVAL;
1042 if (q == p ||
1043 (p && check_loop(q, p, 0)))
1044 return -ELOOP;
1045 atomic_inc(&q->refcnt);
1046 goto graft;
1047 } else {
1048 if (q == NULL)
1049 goto create_n_graft;
1050
1051 /* This magic test requires explanation.
1052 *
1053 * We know, that some child q is already
1054 * attached to this parent and have choice:
1055 * either to change it or to create/graft new one.
1056 *
1057 * 1. We are allowed to create/graft only
1058 * if CREATE and REPLACE flags are set.
1059 *
1060 * 2. If EXCL is set, requestor wanted to say,
1061 * that qdisc tcm_handle is not expected
1062 * to exist, so that we choose create/graft too.
1063 *
1064 * 3. The last case is when no flags are set.
1065 * Alas, it is sort of hole in API, we
1066 * cannot decide what to do unambiguously.
1067 * For now we select create/graft, if
1068 * user gave KIND, which does not match existing.
1069 */
1070 if ((n->nlmsg_flags&NLM_F_CREATE) &&
1071 (n->nlmsg_flags&NLM_F_REPLACE) &&
1072 ((n->nlmsg_flags&NLM_F_EXCL) ||
Patrick McHardy1e904742008-01-22 22:11:17 -08001073 (tca[TCA_KIND] &&
1074 nla_strcmp(tca[TCA_KIND], q->ops->id))))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001075 goto create_n_graft;
1076 }
1077 }
1078 } else {
1079 if (!tcm->tcm_handle)
1080 return -EINVAL;
1081 q = qdisc_lookup(dev, tcm->tcm_handle);
1082 }
1083
1084 /* Change qdisc parameters */
1085 if (q == NULL)
1086 return -ENOENT;
1087 if (n->nlmsg_flags&NLM_F_EXCL)
1088 return -EEXIST;
Patrick McHardy1e904742008-01-22 22:11:17 -08001089 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001090 return -EINVAL;
1091 err = qdisc_change(q, tca);
1092 if (err == 0)
1093 qdisc_notify(skb, n, clid, NULL, q);
1094 return err;
1095
1096create_n_graft:
1097 if (!(n->nlmsg_flags&NLM_F_CREATE))
1098 return -ENOENT;
1099 if (clid == TC_H_INGRESS)
David S. Millerbb949fb2008-07-08 16:55:56 -07001100 q = qdisc_create(dev, &dev->rx_queue,
1101 tcm->tcm_parent, tcm->tcm_parent,
Patrick McHardyffc8fef2007-07-30 17:11:50 -07001102 tca, &err);
David S. Miller6ec1c692009-09-06 01:58:51 -07001103 else {
1104 unsigned int ntx = 0;
1105
1106 if (p && p->ops->cl_ops && p->ops->cl_ops->select_queue)
1107 ntx = p->ops->cl_ops->select_queue(p, tcm);
1108
1109 q = qdisc_create(dev, netdev_get_tx_queue(dev, ntx),
David S. Millerbb949fb2008-07-08 16:55:56 -07001110 tcm->tcm_parent, tcm->tcm_handle,
Patrick McHardyffc8fef2007-07-30 17:11:50 -07001111 tca, &err);
David S. Miller6ec1c692009-09-06 01:58:51 -07001112 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001113 if (q == NULL) {
1114 if (err == -EAGAIN)
1115 goto replay;
1116 return err;
1117 }
1118
1119graft:
Ilpo Järvinene5befbd2008-08-18 22:30:01 -07001120 err = qdisc_graft(dev, p, skb, n, clid, q, NULL);
1121 if (err) {
1122 if (q)
1123 qdisc_destroy(q);
1124 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001125 }
Ilpo Järvinene5befbd2008-08-18 22:30:01 -07001126
Linus Torvalds1da177e2005-04-16 15:20:36 -07001127 return 0;
1128}
1129
1130static int tc_fill_qdisc(struct sk_buff *skb, struct Qdisc *q, u32 clid,
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001131 u32 pid, u32 seq, u16 flags, int event)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001132{
1133 struct tcmsg *tcm;
1134 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001135 unsigned char *b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001136 struct gnet_dump d;
1137
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001138 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001139 tcm = NLMSG_DATA(nlh);
1140 tcm->tcm_family = AF_UNSPEC;
Patrick McHardy9ef1d4c2005-06-28 12:55:30 -07001141 tcm->tcm__pad1 = 0;
1142 tcm->tcm__pad2 = 0;
David S. Miller5ce2d482008-07-08 17:06:30 -07001143 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001144 tcm->tcm_parent = clid;
1145 tcm->tcm_handle = q->handle;
1146 tcm->tcm_info = atomic_read(&q->refcnt);
Patrick McHardy57e1c482008-01-23 20:34:28 -08001147 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001148 if (q->ops->dump && q->ops->dump(q, skb) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001149 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001150 q->qstats.qlen = q->q.qlen;
1151
Jussi Kivilinna175f9c12008-07-20 00:08:47 -07001152 if (q->stab && qdisc_dump_stab(skb, q->stab) < 0)
1153 goto nla_put_failure;
1154
Jarek Poplawski102396a2008-08-29 14:21:52 -07001155 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
1156 qdisc_root_sleeping_lock(q), &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001157 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001158
1159 if (q->ops->dump_stats && q->ops->dump_stats(q, &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001160 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001161
1162 if (gnet_stats_copy_basic(&d, &q->bstats) < 0 ||
Linus Torvalds1da177e2005-04-16 15:20:36 -07001163 gnet_stats_copy_rate_est(&d, &q->rate_est) < 0 ||
Linus Torvalds1da177e2005-04-16 15:20:36 -07001164 gnet_stats_copy_queue(&d, &q->qstats) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001165 goto nla_put_failure;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001166
Linus Torvalds1da177e2005-04-16 15:20:36 -07001167 if (gnet_stats_finish_copy(&d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001168 goto nla_put_failure;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001169
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001170 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001171 return skb->len;
1172
1173nlmsg_failure:
Patrick McHardy1e904742008-01-22 22:11:17 -08001174nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -07001175 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001176 return -1;
1177}
1178
1179static int qdisc_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1180 u32 clid, struct Qdisc *old, struct Qdisc *new)
1181{
1182 struct sk_buff *skb;
1183 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1184
1185 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1186 if (!skb)
1187 return -ENOBUFS;
1188
1189 if (old && old->handle) {
1190 if (tc_fill_qdisc(skb, old, clid, pid, n->nlmsg_seq, 0, RTM_DELQDISC) < 0)
1191 goto err_out;
1192 }
1193 if (new) {
1194 if (tc_fill_qdisc(skb, new, clid, pid, n->nlmsg_seq, old ? NLM_F_REPLACE : 0, RTM_NEWQDISC) < 0)
1195 goto err_out;
1196 }
1197
1198 if (skb->len)
Denis V. Lunev97c53ca2007-11-19 22:26:51 -08001199 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001200
1201err_out:
1202 kfree_skb(skb);
1203 return -EINVAL;
1204}
1205
David S. Miller30723672008-07-18 22:50:15 -07001206static bool tc_qdisc_dump_ignore(struct Qdisc *q)
1207{
1208 return (q->flags & TCQ_F_BUILTIN) ? true : false;
1209}
1210
1211static int tc_dump_qdisc_root(struct Qdisc *root, struct sk_buff *skb,
1212 struct netlink_callback *cb,
1213 int *q_idx_p, int s_q_idx)
1214{
1215 int ret = 0, q_idx = *q_idx_p;
1216 struct Qdisc *q;
1217
1218 if (!root)
1219 return 0;
1220
1221 q = root;
1222 if (q_idx < s_q_idx) {
1223 q_idx++;
1224 } else {
1225 if (!tc_qdisc_dump_ignore(q) &&
1226 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1227 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1228 goto done;
1229 q_idx++;
1230 }
1231 list_for_each_entry(q, &root->list, list) {
1232 if (q_idx < s_q_idx) {
1233 q_idx++;
1234 continue;
1235 }
1236 if (!tc_qdisc_dump_ignore(q) &&
1237 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1238 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1239 goto done;
1240 q_idx++;
1241 }
1242
1243out:
1244 *q_idx_p = q_idx;
1245 return ret;
1246done:
1247 ret = -1;
1248 goto out;
1249}
1250
Linus Torvalds1da177e2005-04-16 15:20:36 -07001251static int tc_dump_qdisc(struct sk_buff *skb, struct netlink_callback *cb)
1252{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001253 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001254 int idx, q_idx;
1255 int s_idx, s_q_idx;
1256 struct net_device *dev;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001257
Denis V. Lunevb8542722007-12-01 00:21:31 +11001258 if (net != &init_net)
1259 return 0;
1260
Linus Torvalds1da177e2005-04-16 15:20:36 -07001261 s_idx = cb->args[0];
1262 s_q_idx = q_idx = cb->args[1];
1263 read_lock(&dev_base_lock);
Pavel Emelianov7562f872007-05-03 15:13:45 -07001264 idx = 0;
Eric W. Biederman881d9662007-09-17 11:56:21 -07001265 for_each_netdev(&init_net, dev) {
David S. Miller30723672008-07-18 22:50:15 -07001266 struct netdev_queue *dev_queue;
1267
Linus Torvalds1da177e2005-04-16 15:20:36 -07001268 if (idx < s_idx)
Pavel Emelianov7562f872007-05-03 15:13:45 -07001269 goto cont;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001270 if (idx > s_idx)
1271 s_q_idx = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001272 q_idx = 0;
David S. Miller30723672008-07-18 22:50:15 -07001273
Patrick McHardyaf356af2009-09-04 06:41:18 +00001274 if (tc_dump_qdisc_root(dev->qdisc, skb, cb, &q_idx, s_q_idx) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001275 goto done;
1276
1277 dev_queue = &dev->rx_queue;
David S. Miller827ebd62008-08-07 20:26:40 -07001278 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 -07001279 goto done;
1280
Pavel Emelianov7562f872007-05-03 15:13:45 -07001281cont:
1282 idx++;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001283 }
1284
1285done:
1286 read_unlock(&dev_base_lock);
1287
1288 cb->args[0] = idx;
1289 cb->args[1] = q_idx;
1290
1291 return skb->len;
1292}
1293
1294
1295
1296/************************************************
1297 * Traffic classes manipulation. *
1298 ************************************************/
1299
1300
1301
1302static int tc_ctl_tclass(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
1303{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001304 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001305 struct tcmsg *tcm = NLMSG_DATA(n);
Patrick McHardy1e904742008-01-22 22:11:17 -08001306 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -07001307 struct net_device *dev;
1308 struct Qdisc *q = NULL;
Eric Dumazet20fea082007-11-14 01:44:41 -08001309 const struct Qdisc_class_ops *cops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001310 unsigned long cl = 0;
1311 unsigned long new_cl;
1312 u32 pid = tcm->tcm_parent;
1313 u32 clid = tcm->tcm_handle;
1314 u32 qid = TC_H_MAJ(clid);
1315 int err;
1316
Denis V. Lunevb8542722007-12-01 00:21:31 +11001317 if (net != &init_net)
1318 return -EINVAL;
1319
Eric W. Biederman881d9662007-09-17 11:56:21 -07001320 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001321 return -ENODEV;
1322
Patrick McHardy1e904742008-01-22 22:11:17 -08001323 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1324 if (err < 0)
1325 return err;
1326
Linus Torvalds1da177e2005-04-16 15:20:36 -07001327 /*
1328 parent == TC_H_UNSPEC - unspecified parent.
1329 parent == TC_H_ROOT - class is root, which has no parent.
1330 parent == X:0 - parent is root class.
1331 parent == X:Y - parent is a node in hierarchy.
1332 parent == 0:Y - parent is X:Y, where X:0 is qdisc.
1333
1334 handle == 0:0 - generate handle from kernel pool.
1335 handle == 0:Y - class is X:Y, where X:0 is qdisc.
1336 handle == X:Y - clear.
1337 handle == X:0 - root class.
1338 */
1339
1340 /* Step 1. Determine qdisc handle X:0 */
1341
1342 if (pid != TC_H_ROOT) {
1343 u32 qid1 = TC_H_MAJ(pid);
1344
1345 if (qid && qid1) {
1346 /* If both majors are known, they must be identical. */
1347 if (qid != qid1)
1348 return -EINVAL;
1349 } else if (qid1) {
1350 qid = qid1;
1351 } else if (qid == 0)
Patrick McHardyaf356af2009-09-04 06:41:18 +00001352 qid = dev->qdisc->handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001353
1354 /* Now qid is genuine qdisc handle consistent
1355 both with parent and child.
1356
1357 TC_H_MAJ(pid) still may be unspecified, complete it now.
1358 */
1359 if (pid)
1360 pid = TC_H_MAKE(qid, pid);
1361 } else {
1362 if (qid == 0)
Patrick McHardyaf356af2009-09-04 06:41:18 +00001363 qid = dev->qdisc->handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001364 }
1365
1366 /* OK. Locate qdisc */
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001367 if ((q = qdisc_lookup(dev, qid)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001368 return -ENOENT;
1369
1370 /* An check that it supports classes */
1371 cops = q->ops->cl_ops;
1372 if (cops == NULL)
1373 return -EINVAL;
1374
1375 /* Now try to get class */
1376 if (clid == 0) {
1377 if (pid == TC_H_ROOT)
1378 clid = qid;
1379 } else
1380 clid = TC_H_MAKE(qid, clid);
1381
1382 if (clid)
1383 cl = cops->get(q, clid);
1384
1385 if (cl == 0) {
1386 err = -ENOENT;
1387 if (n->nlmsg_type != RTM_NEWTCLASS || !(n->nlmsg_flags&NLM_F_CREATE))
1388 goto out;
1389 } else {
1390 switch (n->nlmsg_type) {
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001391 case RTM_NEWTCLASS:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001392 err = -EEXIST;
1393 if (n->nlmsg_flags&NLM_F_EXCL)
1394 goto out;
1395 break;
1396 case RTM_DELTCLASS:
Patrick McHardyde6d5cd2009-09-04 06:41:16 +00001397 err = -EOPNOTSUPP;
1398 if (cops->delete)
1399 err = cops->delete(q, cl);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001400 if (err == 0)
1401 tclass_notify(skb, n, q, cl, RTM_DELTCLASS);
1402 goto out;
1403 case RTM_GETTCLASS:
1404 err = tclass_notify(skb, n, q, cl, RTM_NEWTCLASS);
1405 goto out;
1406 default:
1407 err = -EINVAL;
1408 goto out;
1409 }
1410 }
1411
1412 new_cl = cl;
Patrick McHardyde6d5cd2009-09-04 06:41:16 +00001413 err = -EOPNOTSUPP;
1414 if (cops->change)
1415 err = cops->change(q, clid, pid, tca, &new_cl);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001416 if (err == 0)
1417 tclass_notify(skb, n, q, new_cl, RTM_NEWTCLASS);
1418
1419out:
1420 if (cl)
1421 cops->put(q, cl);
1422
1423 return err;
1424}
1425
1426
1427static int tc_fill_tclass(struct sk_buff *skb, struct Qdisc *q,
1428 unsigned long cl,
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001429 u32 pid, u32 seq, u16 flags, int event)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001430{
1431 struct tcmsg *tcm;
1432 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001433 unsigned char *b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001434 struct gnet_dump d;
Eric Dumazet20fea082007-11-14 01:44:41 -08001435 const struct Qdisc_class_ops *cl_ops = q->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001436
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001437 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001438 tcm = NLMSG_DATA(nlh);
1439 tcm->tcm_family = AF_UNSPEC;
David S. Miller5ce2d482008-07-08 17:06:30 -07001440 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001441 tcm->tcm_parent = q->handle;
1442 tcm->tcm_handle = q->handle;
1443 tcm->tcm_info = 0;
Patrick McHardy57e1c482008-01-23 20:34:28 -08001444 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001445 if (cl_ops->dump && cl_ops->dump(q, cl, skb, tcm) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001446 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001447
Jarek Poplawski102396a2008-08-29 14:21:52 -07001448 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
1449 qdisc_root_sleeping_lock(q), &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001450 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001451
1452 if (cl_ops->dump_stats && cl_ops->dump_stats(q, cl, &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001453 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001454
1455 if (gnet_stats_finish_copy(&d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001456 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001457
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001458 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001459 return skb->len;
1460
1461nlmsg_failure:
Patrick McHardy1e904742008-01-22 22:11:17 -08001462nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -07001463 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001464 return -1;
1465}
1466
1467static int tclass_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1468 struct Qdisc *q, unsigned long cl, int event)
1469{
1470 struct sk_buff *skb;
1471 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1472
1473 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1474 if (!skb)
1475 return -ENOBUFS;
1476
1477 if (tc_fill_tclass(skb, q, cl, pid, n->nlmsg_seq, 0, event) < 0) {
1478 kfree_skb(skb);
1479 return -EINVAL;
1480 }
1481
Denis V. Lunev97c53ca2007-11-19 22:26:51 -08001482 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001483}
1484
1485struct qdisc_dump_args
1486{
1487 struct qdisc_walker w;
1488 struct sk_buff *skb;
1489 struct netlink_callback *cb;
1490};
1491
1492static int qdisc_class_dump(struct Qdisc *q, unsigned long cl, struct qdisc_walker *arg)
1493{
1494 struct qdisc_dump_args *a = (struct qdisc_dump_args *)arg;
1495
1496 return tc_fill_tclass(a->skb, q, cl, NETLINK_CB(a->cb->skb).pid,
1497 a->cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWTCLASS);
1498}
1499
David S. Miller30723672008-07-18 22:50:15 -07001500static int tc_dump_tclass_qdisc(struct Qdisc *q, struct sk_buff *skb,
1501 struct tcmsg *tcm, struct netlink_callback *cb,
1502 int *t_p, int s_t)
1503{
1504 struct qdisc_dump_args arg;
1505
1506 if (tc_qdisc_dump_ignore(q) ||
1507 *t_p < s_t || !q->ops->cl_ops ||
1508 (tcm->tcm_parent &&
1509 TC_H_MAJ(tcm->tcm_parent) != q->handle)) {
1510 (*t_p)++;
1511 return 0;
1512 }
1513 if (*t_p > s_t)
1514 memset(&cb->args[1], 0, sizeof(cb->args)-sizeof(cb->args[0]));
1515 arg.w.fn = qdisc_class_dump;
1516 arg.skb = skb;
1517 arg.cb = cb;
1518 arg.w.stop = 0;
1519 arg.w.skip = cb->args[1];
1520 arg.w.count = 0;
1521 q->ops->cl_ops->walk(q, &arg.w);
1522 cb->args[1] = arg.w.count;
1523 if (arg.w.stop)
1524 return -1;
1525 (*t_p)++;
1526 return 0;
1527}
1528
1529static int tc_dump_tclass_root(struct Qdisc *root, struct sk_buff *skb,
1530 struct tcmsg *tcm, struct netlink_callback *cb,
1531 int *t_p, int s_t)
1532{
1533 struct Qdisc *q;
1534
1535 if (!root)
1536 return 0;
1537
1538 if (tc_dump_tclass_qdisc(root, skb, tcm, cb, t_p, s_t) < 0)
1539 return -1;
1540
1541 list_for_each_entry(q, &root->list, list) {
1542 if (tc_dump_tclass_qdisc(q, skb, tcm, cb, t_p, s_t) < 0)
1543 return -1;
1544 }
1545
1546 return 0;
1547}
1548
Linus Torvalds1da177e2005-04-16 15:20:36 -07001549static int tc_dump_tclass(struct sk_buff *skb, struct netlink_callback *cb)
1550{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001551 struct tcmsg *tcm = (struct tcmsg*)NLMSG_DATA(cb->nlh);
David S. Miller30723672008-07-18 22:50:15 -07001552 struct net *net = sock_net(skb->sk);
1553 struct netdev_queue *dev_queue;
1554 struct net_device *dev;
1555 int t, s_t;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001556
Denis V. Lunevb8542722007-12-01 00:21:31 +11001557 if (net != &init_net)
1558 return 0;
1559
Linus Torvalds1da177e2005-04-16 15:20:36 -07001560 if (cb->nlh->nlmsg_len < NLMSG_LENGTH(sizeof(*tcm)))
1561 return 0;
Eric W. Biederman881d9662007-09-17 11:56:21 -07001562 if ((dev = dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001563 return 0;
1564
1565 s_t = cb->args[0];
1566 t = 0;
1567
Patrick McHardyaf356af2009-09-04 06:41:18 +00001568 if (tc_dump_tclass_root(dev->qdisc, skb, tcm, cb, &t, s_t) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001569 goto done;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001570
David S. Miller30723672008-07-18 22:50:15 -07001571 dev_queue = &dev->rx_queue;
David S. Miller8123b422008-08-08 23:23:39 -07001572 if (tc_dump_tclass_root(dev_queue->qdisc_sleeping, skb, tcm, cb, &t, s_t) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001573 goto done;
1574
1575done:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001576 cb->args[0] = t;
1577
1578 dev_put(dev);
1579 return skb->len;
1580}
1581
1582/* Main classifier routine: scans classifier chain attached
1583 to this qdisc, (optionally) tests for protocol and asks
1584 specific classifiers.
1585 */
Patrick McHardy73ca4912007-07-15 00:02:31 -07001586int tc_classify_compat(struct sk_buff *skb, struct tcf_proto *tp,
1587 struct tcf_result *res)
1588{
1589 __be16 protocol = skb->protocol;
1590 int err = 0;
1591
1592 for (; tp; tp = tp->next) {
1593 if ((tp->protocol == protocol ||
1594 tp->protocol == htons(ETH_P_ALL)) &&
1595 (err = tp->classify(skb, tp, res)) >= 0) {
1596#ifdef CONFIG_NET_CLS_ACT
1597 if (err != TC_ACT_RECLASSIFY && skb->tc_verd)
1598 skb->tc_verd = SET_TC_VERD(skb->tc_verd, 0);
1599#endif
1600 return err;
1601 }
1602 }
1603 return -1;
1604}
1605EXPORT_SYMBOL(tc_classify_compat);
1606
Linus Torvalds1da177e2005-04-16 15:20:36 -07001607int tc_classify(struct sk_buff *skb, struct tcf_proto *tp,
Patrick McHardy73ca4912007-07-15 00:02:31 -07001608 struct tcf_result *res)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001609{
1610 int err = 0;
Patrick McHardy73ca4912007-07-15 00:02:31 -07001611 __be16 protocol;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001612#ifdef CONFIG_NET_CLS_ACT
1613 struct tcf_proto *otp = tp;
1614reclassify:
1615#endif
1616 protocol = skb->protocol;
1617
Patrick McHardy73ca4912007-07-15 00:02:31 -07001618 err = tc_classify_compat(skb, tp, res);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001619#ifdef CONFIG_NET_CLS_ACT
Patrick McHardy73ca4912007-07-15 00:02:31 -07001620 if (err == TC_ACT_RECLASSIFY) {
1621 u32 verd = G_TC_VERD(skb->tc_verd);
1622 tp = otp;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001623
Patrick McHardy73ca4912007-07-15 00:02:31 -07001624 if (verd++ >= MAX_REC_LOOP) {
1625 printk("rule prio %u protocol %02x reclassify loop, "
1626 "packet dropped\n",
1627 tp->prio&0xffff, ntohs(tp->protocol));
1628 return TC_ACT_SHOT;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001629 }
Patrick McHardy73ca4912007-07-15 00:02:31 -07001630 skb->tc_verd = SET_TC_VERD(skb->tc_verd, verd);
1631 goto reclassify;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001632 }
Patrick McHardy73ca4912007-07-15 00:02:31 -07001633#endif
1634 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001635}
Patrick McHardy73ca4912007-07-15 00:02:31 -07001636EXPORT_SYMBOL(tc_classify);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001637
Patrick McHardya48b5a62007-03-23 11:29:43 -07001638void tcf_destroy(struct tcf_proto *tp)
1639{
1640 tp->ops->destroy(tp);
1641 module_put(tp->ops->owner);
1642 kfree(tp);
1643}
1644
Patrick McHardyff31ab52008-07-01 19:52:38 -07001645void tcf_destroy_chain(struct tcf_proto **fl)
Patrick McHardya48b5a62007-03-23 11:29:43 -07001646{
1647 struct tcf_proto *tp;
1648
Patrick McHardyff31ab52008-07-01 19:52:38 -07001649 while ((tp = *fl) != NULL) {
1650 *fl = tp->next;
Patrick McHardya48b5a62007-03-23 11:29:43 -07001651 tcf_destroy(tp);
1652 }
1653}
1654EXPORT_SYMBOL(tcf_destroy_chain);
1655
Linus Torvalds1da177e2005-04-16 15:20:36 -07001656#ifdef CONFIG_PROC_FS
1657static int psched_show(struct seq_file *seq, void *v)
1658{
Patrick McHardy3c0cfc12007-10-10 16:32:41 -07001659 struct timespec ts;
1660
1661 hrtimer_get_res(CLOCK_MONOTONIC, &ts);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001662 seq_printf(seq, "%08x %08x %08x %08x\n",
Jarek Poplawskica44d6e2009-06-15 02:31:47 -07001663 (u32)NSEC_PER_USEC, (u32)PSCHED_TICKS2NS(1),
Patrick McHardy514bca32007-03-16 12:34:52 -07001664 1000000,
Patrick McHardy3c0cfc12007-10-10 16:32:41 -07001665 (u32)NSEC_PER_SEC/(u32)ktime_to_ns(timespec_to_ktime(ts)));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001666
1667 return 0;
1668}
1669
1670static int psched_open(struct inode *inode, struct file *file)
1671{
1672 return single_open(file, psched_show, PDE(inode)->data);
1673}
1674
Arjan van de Venda7071d2007-02-12 00:55:36 -08001675static const struct file_operations psched_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001676 .owner = THIS_MODULE,
1677 .open = psched_open,
1678 .read = seq_read,
1679 .llseek = seq_lseek,
1680 .release = single_release,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001681};
Linus Torvalds1da177e2005-04-16 15:20:36 -07001682#endif
1683
Linus Torvalds1da177e2005-04-16 15:20:36 -07001684static int __init pktsched_init(void)
1685{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001686 register_qdisc(&pfifo_qdisc_ops);
1687 register_qdisc(&bfifo_qdisc_ops);
David S. Miller6ec1c692009-09-06 01:58:51 -07001688 register_qdisc(&mq_qdisc_ops);
Eric W. Biederman457c4cb2007-09-12 12:01:34 +02001689 proc_net_fops_create(&init_net, "psched", 0, &psched_fops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001690
Thomas Grafbe577dd2007-03-22 11:55:50 -07001691 rtnl_register(PF_UNSPEC, RTM_NEWQDISC, tc_modify_qdisc, NULL);
1692 rtnl_register(PF_UNSPEC, RTM_DELQDISC, tc_get_qdisc, NULL);
1693 rtnl_register(PF_UNSPEC, RTM_GETQDISC, tc_get_qdisc, tc_dump_qdisc);
1694 rtnl_register(PF_UNSPEC, RTM_NEWTCLASS, tc_ctl_tclass, NULL);
1695 rtnl_register(PF_UNSPEC, RTM_DELTCLASS, tc_ctl_tclass, NULL);
1696 rtnl_register(PF_UNSPEC, RTM_GETTCLASS, tc_ctl_tclass, tc_dump_tclass);
1697
Linus Torvalds1da177e2005-04-16 15:20:36 -07001698 return 0;
1699}
1700
1701subsys_initcall(pktsched_init);