blob: c6e4063f698c5ea768a4f0122bf3474ccf5dd50b [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,
David S. Miller2fbd3da2009-09-01 17:59:25 -0700455 timer);
Patrick McHardy41794772007-03-16 01:19:15 -0700456
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{
David S. Miller2fbd3da2009-09-01 17:59:25 -0700465 hrtimer_init(&wd->timer, CLOCK_MONOTONIC, HRTIMER_MODE_ABS);
466 wd->timer.function = qdisc_watchdog;
Patrick McHardy41794772007-03-16 01:19:15 -0700467 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));
David S. Miller2fbd3da2009-09-01 17:59:25 -0700482 hrtimer_start(&wd->timer, time, HRTIMER_MODE_ABS);
Patrick McHardy41794772007-03-16 01:19:15 -0700483}
484EXPORT_SYMBOL(qdisc_watchdog_schedule);
485
486void qdisc_watchdog_cancel(struct qdisc_watchdog *wd)
487{
David S. Miller2fbd3da2009-09-01 17:59:25 -0700488 hrtimer_cancel(&wd->timer);
Patrick McHardy41794772007-03-16 01:19:15 -0700489 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
Jarek Poplawski036d6a62009-09-13 22:35:44 +0000696 if (!ingress)
697 qdisc_destroy(old);
David S. Miller99194cf2008-07-17 04:54:10 -0700698 }
699
Jarek Poplawski036d6a62009-09-13 22:35:44 +0000700 if (!ingress) {
701 notify_and_destroy(skb, n, classid, dev->qdisc, new);
702 if (new && !new->ops->attach)
703 atomic_inc(&new->refcnt);
704 dev->qdisc = new ? : &noop_qdisc;
705 } else {
706 notify_and_destroy(skb, n, classid, old, new);
707 }
Patrick McHardyaf356af2009-09-04 06:41:18 +0000708
David S. Miller99194cf2008-07-17 04:54:10 -0700709 if (dev->flags & IFF_UP)
710 dev_activate(dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700711 } else {
Eric Dumazet20fea082007-11-14 01:44:41 -0800712 const struct Qdisc_class_ops *cops = parent->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700713
Patrick McHardyc9f1d032009-09-04 06:41:13 +0000714 err = -EOPNOTSUPP;
715 if (cops && cops->graft) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700716 unsigned long cl = cops->get(parent, classid);
717 if (cl) {
David S. Miller99194cf2008-07-17 04:54:10 -0700718 err = cops->graft(parent, cl, new, &old);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700719 cops->put(parent, cl);
Patrick McHardyc9f1d032009-09-04 06:41:13 +0000720 } else
721 err = -ENOENT;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700722 }
David S. Miller99194cf2008-07-17 04:54:10 -0700723 if (!err)
724 notify_and_destroy(skb, n, classid, old, new);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700725 }
726 return err;
727}
728
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700729/* lockdep annotation is needed for ingress; egress gets it only for name */
730static struct lock_class_key qdisc_tx_lock;
731static struct lock_class_key qdisc_rx_lock;
732
Linus Torvalds1da177e2005-04-16 15:20:36 -0700733/*
734 Allocate and initialize new qdisc.
735
736 Parameters are passed via opt.
737 */
738
739static struct Qdisc *
David S. Millerbb949fb2008-07-08 16:55:56 -0700740qdisc_create(struct net_device *dev, struct netdev_queue *dev_queue,
Patrick McHardy23bcf632009-09-09 18:11:23 -0700741 struct Qdisc *p, u32 parent, u32 handle,
742 struct nlattr **tca, int *errp)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700743{
744 int err;
Patrick McHardy1e904742008-01-22 22:11:17 -0800745 struct nlattr *kind = tca[TCA_KIND];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700746 struct Qdisc *sch;
747 struct Qdisc_ops *ops;
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700748 struct qdisc_size_table *stab;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700749
750 ops = qdisc_lookup_ops(kind);
Johannes Berg95a5afc2008-10-16 15:24:51 -0700751#ifdef CONFIG_MODULES
Linus Torvalds1da177e2005-04-16 15:20:36 -0700752 if (ops == NULL && kind != NULL) {
753 char name[IFNAMSIZ];
Patrick McHardy1e904742008-01-22 22:11:17 -0800754 if (nla_strlcpy(name, kind, IFNAMSIZ) < IFNAMSIZ) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700755 /* We dropped the RTNL semaphore in order to
756 * perform the module load. So, even if we
757 * succeeded in loading the module we have to
758 * tell the caller to replay the request. We
759 * indicate this using -EAGAIN.
760 * We replay the request because the device may
761 * go away in the mean time.
762 */
763 rtnl_unlock();
764 request_module("sch_%s", name);
765 rtnl_lock();
766 ops = qdisc_lookup_ops(kind);
767 if (ops != NULL) {
768 /* We will try again qdisc_lookup_ops,
769 * so don't keep a reference.
770 */
771 module_put(ops->owner);
772 err = -EAGAIN;
773 goto err_out;
774 }
775 }
776 }
777#endif
778
Jamal Hadi Salimb9e2cc02006-08-03 16:36:51 -0700779 err = -ENOENT;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700780 if (ops == NULL)
781 goto err_out;
782
David S. Miller5ce2d482008-07-08 17:06:30 -0700783 sch = qdisc_alloc(dev_queue, ops);
Thomas Graf3d54b822005-07-05 14:15:09 -0700784 if (IS_ERR(sch)) {
785 err = PTR_ERR(sch);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700786 goto err_out2;
Thomas Graf3d54b822005-07-05 14:15:09 -0700787 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700788
Patrick McHardyffc8fef2007-07-30 17:11:50 -0700789 sch->parent = parent;
790
Thomas Graf3d54b822005-07-05 14:15:09 -0700791 if (handle == TC_H_INGRESS) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700792 sch->flags |= TCQ_F_INGRESS;
Thomas Graf3d54b822005-07-05 14:15:09 -0700793 handle = TC_H_MAKE(TC_H_INGRESS, 0);
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700794 lockdep_set_class(qdisc_lock(sch), &qdisc_rx_lock);
Patrick McHardyfd44de72007-04-16 17:07:08 -0700795 } else {
Patrick McHardyfd44de72007-04-16 17:07:08 -0700796 if (handle == 0) {
797 handle = qdisc_alloc_handle(dev);
798 err = -ENOMEM;
799 if (handle == 0)
800 goto err_out3;
801 }
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700802 lockdep_set_class(qdisc_lock(sch), &qdisc_tx_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700803 }
804
Thomas Graf3d54b822005-07-05 14:15:09 -0700805 sch->handle = handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700806
Patrick McHardy1e904742008-01-22 22:11:17 -0800807 if (!ops->init || (err = ops->init(sch, tca[TCA_OPTIONS])) == 0) {
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700808 if (tca[TCA_STAB]) {
809 stab = qdisc_get_stab(tca[TCA_STAB]);
810 if (IS_ERR(stab)) {
811 err = PTR_ERR(stab);
812 goto err_out3;
813 }
814 sch->stab = stab;
815 }
Patrick McHardy1e904742008-01-22 22:11:17 -0800816 if (tca[TCA_RATE]) {
Jarek Poplawskif6f9b932008-08-27 02:25:17 -0700817 spinlock_t *root_lock;
818
Patrick McHardy23bcf632009-09-09 18:11:23 -0700819 err = -EOPNOTSUPP;
820 if (sch->flags & TCQ_F_MQROOT)
821 goto err_out4;
822
Jarek Poplawskif6f9b932008-08-27 02:25:17 -0700823 if ((sch->parent != TC_H_ROOT) &&
Patrick McHardy23bcf632009-09-09 18:11:23 -0700824 !(sch->flags & TCQ_F_INGRESS) &&
825 (!p || !(p->flags & TCQ_F_MQROOT)))
Jarek Poplawskif6f9b932008-08-27 02:25:17 -0700826 root_lock = qdisc_root_sleeping_lock(sch);
827 else
828 root_lock = qdisc_lock(sch);
829
Thomas Graf023e09a2005-07-05 14:15:53 -0700830 err = gen_new_estimator(&sch->bstats, &sch->rate_est,
Jarek Poplawskif6f9b932008-08-27 02:25:17 -0700831 root_lock, tca[TCA_RATE]);
Patrick McHardy23bcf632009-09-09 18:11:23 -0700832 if (err)
833 goto err_out4;
Thomas Graf023e09a2005-07-05 14:15:53 -0700834 }
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700835
836 qdisc_list_add(sch);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700837
Linus Torvalds1da177e2005-04-16 15:20:36 -0700838 return sch;
839 }
840err_out3:
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700841 qdisc_put_stab(sch->stab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700842 dev_put(dev);
Thomas Graf3d54b822005-07-05 14:15:09 -0700843 kfree((char *) sch - sch->padded);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700844err_out2:
845 module_put(ops->owner);
846err_out:
847 *errp = err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700848 return NULL;
Patrick McHardy23bcf632009-09-09 18:11:23 -0700849
850err_out4:
851 /*
852 * Any broken qdiscs that would require a ops->reset() here?
853 * The qdisc was never in action so it shouldn't be necessary.
854 */
855 if (ops->destroy)
856 ops->destroy(sch);
857 goto err_out3;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700858}
859
Patrick McHardy1e904742008-01-22 22:11:17 -0800860static int qdisc_change(struct Qdisc *sch, struct nlattr **tca)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700861{
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700862 struct qdisc_size_table *stab = NULL;
863 int err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700864
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700865 if (tca[TCA_OPTIONS]) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700866 if (sch->ops->change == NULL)
867 return -EINVAL;
Patrick McHardy1e904742008-01-22 22:11:17 -0800868 err = sch->ops->change(sch, tca[TCA_OPTIONS]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700869 if (err)
870 return err;
871 }
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700872
873 if (tca[TCA_STAB]) {
874 stab = qdisc_get_stab(tca[TCA_STAB]);
875 if (IS_ERR(stab))
876 return PTR_ERR(stab);
877 }
878
879 qdisc_put_stab(sch->stab);
880 sch->stab = stab;
881
Patrick McHardy23bcf632009-09-09 18:11:23 -0700882 if (tca[TCA_RATE]) {
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800883 /* NB: ignores errors from replace_estimator
884 because change can't be undone. */
Patrick McHardy23bcf632009-09-09 18:11:23 -0700885 if (sch->flags & TCQ_F_MQROOT)
886 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700887 gen_replace_estimator(&sch->bstats, &sch->rate_est,
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800888 qdisc_root_sleeping_lock(sch),
889 tca[TCA_RATE]);
Patrick McHardy23bcf632009-09-09 18:11:23 -0700890 }
891out:
Linus Torvalds1da177e2005-04-16 15:20:36 -0700892 return 0;
893}
894
895struct check_loop_arg
896{
897 struct qdisc_walker w;
898 struct Qdisc *p;
899 int depth;
900};
901
902static int check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w);
903
904static int check_loop(struct Qdisc *q, struct Qdisc *p, int depth)
905{
906 struct check_loop_arg arg;
907
908 if (q->ops->cl_ops == NULL)
909 return 0;
910
911 arg.w.stop = arg.w.skip = arg.w.count = 0;
912 arg.w.fn = check_loop_fn;
913 arg.depth = depth;
914 arg.p = p;
915 q->ops->cl_ops->walk(q, &arg.w);
916 return arg.w.stop ? -ELOOP : 0;
917}
918
919static int
920check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w)
921{
922 struct Qdisc *leaf;
Eric Dumazet20fea082007-11-14 01:44:41 -0800923 const struct Qdisc_class_ops *cops = q->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700924 struct check_loop_arg *arg = (struct check_loop_arg *)w;
925
926 leaf = cops->leaf(q, cl);
927 if (leaf) {
928 if (leaf == arg->p || arg->depth > 7)
929 return -ELOOP;
930 return check_loop(leaf, arg->p, arg->depth + 1);
931 }
932 return 0;
933}
934
935/*
936 * Delete/get qdisc.
937 */
938
939static int tc_get_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
940{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900941 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700942 struct tcmsg *tcm = NLMSG_DATA(n);
Patrick McHardy1e904742008-01-22 22:11:17 -0800943 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700944 struct net_device *dev;
945 u32 clid = tcm->tcm_parent;
946 struct Qdisc *q = NULL;
947 struct Qdisc *p = NULL;
948 int err;
949
Denis V. Lunevb8542722007-12-01 00:21:31 +1100950 if (net != &init_net)
951 return -EINVAL;
952
Eric W. Biederman881d9662007-09-17 11:56:21 -0700953 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700954 return -ENODEV;
955
Patrick McHardy1e904742008-01-22 22:11:17 -0800956 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
957 if (err < 0)
958 return err;
959
Linus Torvalds1da177e2005-04-16 15:20:36 -0700960 if (clid) {
961 if (clid != TC_H_ROOT) {
962 if (TC_H_MAJ(clid) != TC_H_MAJ(TC_H_INGRESS)) {
963 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
964 return -ENOENT;
965 q = qdisc_leaf(p, clid);
966 } else { /* ingress */
David S. Miller8123b422008-08-08 23:23:39 -0700967 q = dev->rx_queue.qdisc_sleeping;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900968 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700969 } else {
Patrick McHardyaf356af2009-09-04 06:41:18 +0000970 q = dev->qdisc;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700971 }
972 if (!q)
973 return -ENOENT;
974
975 if (tcm->tcm_handle && q->handle != tcm->tcm_handle)
976 return -EINVAL;
977 } else {
978 if ((q = qdisc_lookup(dev, tcm->tcm_handle)) == NULL)
979 return -ENOENT;
980 }
981
Patrick McHardy1e904742008-01-22 22:11:17 -0800982 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700983 return -EINVAL;
984
985 if (n->nlmsg_type == RTM_DELQDISC) {
986 if (!clid)
987 return -EINVAL;
988 if (q->handle == 0)
989 return -ENOENT;
David S. Miller99194cf2008-07-17 04:54:10 -0700990 if ((err = qdisc_graft(dev, p, skb, n, clid, NULL, q)) != 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700991 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700992 } else {
993 qdisc_notify(skb, n, clid, NULL, q);
994 }
995 return 0;
996}
997
998/*
999 Create/change qdisc.
1000 */
1001
1002static int tc_modify_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
1003{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001004 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001005 struct tcmsg *tcm;
Patrick McHardy1e904742008-01-22 22:11:17 -08001006 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -07001007 struct net_device *dev;
1008 u32 clid;
1009 struct Qdisc *q, *p;
1010 int err;
1011
Denis V. Lunevb8542722007-12-01 00:21:31 +11001012 if (net != &init_net)
1013 return -EINVAL;
1014
Linus Torvalds1da177e2005-04-16 15:20:36 -07001015replay:
1016 /* Reinit, just in case something touches this. */
1017 tcm = NLMSG_DATA(n);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001018 clid = tcm->tcm_parent;
1019 q = p = NULL;
1020
Eric W. Biederman881d9662007-09-17 11:56:21 -07001021 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001022 return -ENODEV;
1023
Patrick McHardy1e904742008-01-22 22:11:17 -08001024 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1025 if (err < 0)
1026 return err;
1027
Linus Torvalds1da177e2005-04-16 15:20:36 -07001028 if (clid) {
1029 if (clid != TC_H_ROOT) {
1030 if (clid != TC_H_INGRESS) {
1031 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
1032 return -ENOENT;
1033 q = qdisc_leaf(p, clid);
1034 } else { /*ingress */
David S. Miller8123b422008-08-08 23:23:39 -07001035 q = dev->rx_queue.qdisc_sleeping;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001036 }
1037 } else {
Patrick McHardyaf356af2009-09-04 06:41:18 +00001038 q = dev->qdisc;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001039 }
1040
1041 /* It may be default qdisc, ignore it */
1042 if (q && q->handle == 0)
1043 q = NULL;
1044
1045 if (!q || !tcm->tcm_handle || q->handle != tcm->tcm_handle) {
1046 if (tcm->tcm_handle) {
1047 if (q && !(n->nlmsg_flags&NLM_F_REPLACE))
1048 return -EEXIST;
1049 if (TC_H_MIN(tcm->tcm_handle))
1050 return -EINVAL;
1051 if ((q = qdisc_lookup(dev, tcm->tcm_handle)) == NULL)
1052 goto create_n_graft;
1053 if (n->nlmsg_flags&NLM_F_EXCL)
1054 return -EEXIST;
Patrick McHardy1e904742008-01-22 22:11:17 -08001055 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001056 return -EINVAL;
1057 if (q == p ||
1058 (p && check_loop(q, p, 0)))
1059 return -ELOOP;
1060 atomic_inc(&q->refcnt);
1061 goto graft;
1062 } else {
1063 if (q == NULL)
1064 goto create_n_graft;
1065
1066 /* This magic test requires explanation.
1067 *
1068 * We know, that some child q is already
1069 * attached to this parent and have choice:
1070 * either to change it or to create/graft new one.
1071 *
1072 * 1. We are allowed to create/graft only
1073 * if CREATE and REPLACE flags are set.
1074 *
1075 * 2. If EXCL is set, requestor wanted to say,
1076 * that qdisc tcm_handle is not expected
1077 * to exist, so that we choose create/graft too.
1078 *
1079 * 3. The last case is when no flags are set.
1080 * Alas, it is sort of hole in API, we
1081 * cannot decide what to do unambiguously.
1082 * For now we select create/graft, if
1083 * user gave KIND, which does not match existing.
1084 */
1085 if ((n->nlmsg_flags&NLM_F_CREATE) &&
1086 (n->nlmsg_flags&NLM_F_REPLACE) &&
1087 ((n->nlmsg_flags&NLM_F_EXCL) ||
Patrick McHardy1e904742008-01-22 22:11:17 -08001088 (tca[TCA_KIND] &&
1089 nla_strcmp(tca[TCA_KIND], q->ops->id))))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001090 goto create_n_graft;
1091 }
1092 }
1093 } else {
1094 if (!tcm->tcm_handle)
1095 return -EINVAL;
1096 q = qdisc_lookup(dev, tcm->tcm_handle);
1097 }
1098
1099 /* Change qdisc parameters */
1100 if (q == NULL)
1101 return -ENOENT;
1102 if (n->nlmsg_flags&NLM_F_EXCL)
1103 return -EEXIST;
Patrick McHardy1e904742008-01-22 22:11:17 -08001104 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001105 return -EINVAL;
1106 err = qdisc_change(q, tca);
1107 if (err == 0)
1108 qdisc_notify(skb, n, clid, NULL, q);
1109 return err;
1110
1111create_n_graft:
1112 if (!(n->nlmsg_flags&NLM_F_CREATE))
1113 return -ENOENT;
1114 if (clid == TC_H_INGRESS)
Patrick McHardy23bcf632009-09-09 18:11:23 -07001115 q = qdisc_create(dev, &dev->rx_queue, p,
David S. Millerbb949fb2008-07-08 16:55:56 -07001116 tcm->tcm_parent, tcm->tcm_parent,
Patrick McHardyffc8fef2007-07-30 17:11:50 -07001117 tca, &err);
David S. Miller6ec1c692009-09-06 01:58:51 -07001118 else {
1119 unsigned int ntx = 0;
1120
1121 if (p && p->ops->cl_ops && p->ops->cl_ops->select_queue)
1122 ntx = p->ops->cl_ops->select_queue(p, tcm);
1123
Patrick McHardy23bcf632009-09-09 18:11:23 -07001124 q = qdisc_create(dev, netdev_get_tx_queue(dev, ntx), p,
David S. Millerbb949fb2008-07-08 16:55:56 -07001125 tcm->tcm_parent, tcm->tcm_handle,
Patrick McHardyffc8fef2007-07-30 17:11:50 -07001126 tca, &err);
David S. Miller6ec1c692009-09-06 01:58:51 -07001127 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001128 if (q == NULL) {
1129 if (err == -EAGAIN)
1130 goto replay;
1131 return err;
1132 }
1133
1134graft:
Ilpo Järvinene5befbd2008-08-18 22:30:01 -07001135 err = qdisc_graft(dev, p, skb, n, clid, q, NULL);
1136 if (err) {
1137 if (q)
1138 qdisc_destroy(q);
1139 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001140 }
Ilpo Järvinene5befbd2008-08-18 22:30:01 -07001141
Linus Torvalds1da177e2005-04-16 15:20:36 -07001142 return 0;
1143}
1144
1145static int tc_fill_qdisc(struct sk_buff *skb, struct Qdisc *q, u32 clid,
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001146 u32 pid, u32 seq, u16 flags, int event)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001147{
1148 struct tcmsg *tcm;
1149 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001150 unsigned char *b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001151 struct gnet_dump d;
1152
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001153 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001154 tcm = NLMSG_DATA(nlh);
1155 tcm->tcm_family = AF_UNSPEC;
Patrick McHardy9ef1d4c2005-06-28 12:55:30 -07001156 tcm->tcm__pad1 = 0;
1157 tcm->tcm__pad2 = 0;
David S. Miller5ce2d482008-07-08 17:06:30 -07001158 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001159 tcm->tcm_parent = clid;
1160 tcm->tcm_handle = q->handle;
1161 tcm->tcm_info = atomic_read(&q->refcnt);
Patrick McHardy57e1c482008-01-23 20:34:28 -08001162 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001163 if (q->ops->dump && q->ops->dump(q, skb) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001164 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001165 q->qstats.qlen = q->q.qlen;
1166
Jussi Kivilinna175f9c12008-07-20 00:08:47 -07001167 if (q->stab && qdisc_dump_stab(skb, q->stab) < 0)
1168 goto nla_put_failure;
1169
Jarek Poplawski102396a2008-08-29 14:21:52 -07001170 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
1171 qdisc_root_sleeping_lock(q), &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001172 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001173
1174 if (q->ops->dump_stats && q->ops->dump_stats(q, &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001175 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001176
1177 if (gnet_stats_copy_basic(&d, &q->bstats) < 0 ||
Linus Torvalds1da177e2005-04-16 15:20:36 -07001178 gnet_stats_copy_rate_est(&d, &q->rate_est) < 0 ||
Linus Torvalds1da177e2005-04-16 15:20:36 -07001179 gnet_stats_copy_queue(&d, &q->qstats) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001180 goto nla_put_failure;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001181
Linus Torvalds1da177e2005-04-16 15:20:36 -07001182 if (gnet_stats_finish_copy(&d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001183 goto nla_put_failure;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001184
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001185 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001186 return skb->len;
1187
1188nlmsg_failure:
Patrick McHardy1e904742008-01-22 22:11:17 -08001189nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -07001190 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001191 return -1;
1192}
1193
1194static int qdisc_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1195 u32 clid, struct Qdisc *old, struct Qdisc *new)
1196{
1197 struct sk_buff *skb;
1198 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1199
1200 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1201 if (!skb)
1202 return -ENOBUFS;
1203
1204 if (old && old->handle) {
1205 if (tc_fill_qdisc(skb, old, clid, pid, n->nlmsg_seq, 0, RTM_DELQDISC) < 0)
1206 goto err_out;
1207 }
1208 if (new) {
1209 if (tc_fill_qdisc(skb, new, clid, pid, n->nlmsg_seq, old ? NLM_F_REPLACE : 0, RTM_NEWQDISC) < 0)
1210 goto err_out;
1211 }
1212
1213 if (skb->len)
Denis V. Lunev97c53ca2007-11-19 22:26:51 -08001214 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001215
1216err_out:
1217 kfree_skb(skb);
1218 return -EINVAL;
1219}
1220
David S. Miller30723672008-07-18 22:50:15 -07001221static bool tc_qdisc_dump_ignore(struct Qdisc *q)
1222{
1223 return (q->flags & TCQ_F_BUILTIN) ? true : false;
1224}
1225
1226static int tc_dump_qdisc_root(struct Qdisc *root, struct sk_buff *skb,
1227 struct netlink_callback *cb,
1228 int *q_idx_p, int s_q_idx)
1229{
1230 int ret = 0, q_idx = *q_idx_p;
1231 struct Qdisc *q;
1232
1233 if (!root)
1234 return 0;
1235
1236 q = root;
1237 if (q_idx < s_q_idx) {
1238 q_idx++;
1239 } else {
1240 if (!tc_qdisc_dump_ignore(q) &&
1241 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1242 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1243 goto done;
1244 q_idx++;
1245 }
1246 list_for_each_entry(q, &root->list, list) {
1247 if (q_idx < s_q_idx) {
1248 q_idx++;
1249 continue;
1250 }
1251 if (!tc_qdisc_dump_ignore(q) &&
1252 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1253 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1254 goto done;
1255 q_idx++;
1256 }
1257
1258out:
1259 *q_idx_p = q_idx;
1260 return ret;
1261done:
1262 ret = -1;
1263 goto out;
1264}
1265
Linus Torvalds1da177e2005-04-16 15:20:36 -07001266static int tc_dump_qdisc(struct sk_buff *skb, struct netlink_callback *cb)
1267{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001268 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001269 int idx, q_idx;
1270 int s_idx, s_q_idx;
1271 struct net_device *dev;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001272
Denis V. Lunevb8542722007-12-01 00:21:31 +11001273 if (net != &init_net)
1274 return 0;
1275
Linus Torvalds1da177e2005-04-16 15:20:36 -07001276 s_idx = cb->args[0];
1277 s_q_idx = q_idx = cb->args[1];
1278 read_lock(&dev_base_lock);
Pavel Emelianov7562f872007-05-03 15:13:45 -07001279 idx = 0;
Eric W. Biederman881d9662007-09-17 11:56:21 -07001280 for_each_netdev(&init_net, dev) {
David S. Miller30723672008-07-18 22:50:15 -07001281 struct netdev_queue *dev_queue;
1282
Linus Torvalds1da177e2005-04-16 15:20:36 -07001283 if (idx < s_idx)
Pavel Emelianov7562f872007-05-03 15:13:45 -07001284 goto cont;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001285 if (idx > s_idx)
1286 s_q_idx = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001287 q_idx = 0;
David S. Miller30723672008-07-18 22:50:15 -07001288
Patrick McHardyaf356af2009-09-04 06:41:18 +00001289 if (tc_dump_qdisc_root(dev->qdisc, skb, cb, &q_idx, s_q_idx) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001290 goto done;
1291
1292 dev_queue = &dev->rx_queue;
David S. Miller827ebd62008-08-07 20:26:40 -07001293 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 -07001294 goto done;
1295
Pavel Emelianov7562f872007-05-03 15:13:45 -07001296cont:
1297 idx++;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001298 }
1299
1300done:
1301 read_unlock(&dev_base_lock);
1302
1303 cb->args[0] = idx;
1304 cb->args[1] = q_idx;
1305
1306 return skb->len;
1307}
1308
1309
1310
1311/************************************************
1312 * Traffic classes manipulation. *
1313 ************************************************/
1314
1315
1316
1317static int tc_ctl_tclass(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
1318{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001319 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001320 struct tcmsg *tcm = NLMSG_DATA(n);
Patrick McHardy1e904742008-01-22 22:11:17 -08001321 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -07001322 struct net_device *dev;
1323 struct Qdisc *q = NULL;
Eric Dumazet20fea082007-11-14 01:44:41 -08001324 const struct Qdisc_class_ops *cops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001325 unsigned long cl = 0;
1326 unsigned long new_cl;
1327 u32 pid = tcm->tcm_parent;
1328 u32 clid = tcm->tcm_handle;
1329 u32 qid = TC_H_MAJ(clid);
1330 int err;
1331
Denis V. Lunevb8542722007-12-01 00:21:31 +11001332 if (net != &init_net)
1333 return -EINVAL;
1334
Eric W. Biederman881d9662007-09-17 11:56:21 -07001335 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001336 return -ENODEV;
1337
Patrick McHardy1e904742008-01-22 22:11:17 -08001338 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1339 if (err < 0)
1340 return err;
1341
Linus Torvalds1da177e2005-04-16 15:20:36 -07001342 /*
1343 parent == TC_H_UNSPEC - unspecified parent.
1344 parent == TC_H_ROOT - class is root, which has no parent.
1345 parent == X:0 - parent is root class.
1346 parent == X:Y - parent is a node in hierarchy.
1347 parent == 0:Y - parent is X:Y, where X:0 is qdisc.
1348
1349 handle == 0:0 - generate handle from kernel pool.
1350 handle == 0:Y - class is X:Y, where X:0 is qdisc.
1351 handle == X:Y - clear.
1352 handle == X:0 - root class.
1353 */
1354
1355 /* Step 1. Determine qdisc handle X:0 */
1356
1357 if (pid != TC_H_ROOT) {
1358 u32 qid1 = TC_H_MAJ(pid);
1359
1360 if (qid && qid1) {
1361 /* If both majors are known, they must be identical. */
1362 if (qid != qid1)
1363 return -EINVAL;
1364 } else if (qid1) {
1365 qid = qid1;
1366 } else if (qid == 0)
Patrick McHardyaf356af2009-09-04 06:41:18 +00001367 qid = dev->qdisc->handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001368
1369 /* Now qid is genuine qdisc handle consistent
1370 both with parent and child.
1371
1372 TC_H_MAJ(pid) still may be unspecified, complete it now.
1373 */
1374 if (pid)
1375 pid = TC_H_MAKE(qid, pid);
1376 } else {
1377 if (qid == 0)
Patrick McHardyaf356af2009-09-04 06:41:18 +00001378 qid = dev->qdisc->handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001379 }
1380
1381 /* OK. Locate qdisc */
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001382 if ((q = qdisc_lookup(dev, qid)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001383 return -ENOENT;
1384
1385 /* An check that it supports classes */
1386 cops = q->ops->cl_ops;
1387 if (cops == NULL)
1388 return -EINVAL;
1389
1390 /* Now try to get class */
1391 if (clid == 0) {
1392 if (pid == TC_H_ROOT)
1393 clid = qid;
1394 } else
1395 clid = TC_H_MAKE(qid, clid);
1396
1397 if (clid)
1398 cl = cops->get(q, clid);
1399
1400 if (cl == 0) {
1401 err = -ENOENT;
1402 if (n->nlmsg_type != RTM_NEWTCLASS || !(n->nlmsg_flags&NLM_F_CREATE))
1403 goto out;
1404 } else {
1405 switch (n->nlmsg_type) {
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001406 case RTM_NEWTCLASS:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001407 err = -EEXIST;
1408 if (n->nlmsg_flags&NLM_F_EXCL)
1409 goto out;
1410 break;
1411 case RTM_DELTCLASS:
Patrick McHardyde6d5cd2009-09-04 06:41:16 +00001412 err = -EOPNOTSUPP;
1413 if (cops->delete)
1414 err = cops->delete(q, cl);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001415 if (err == 0)
1416 tclass_notify(skb, n, q, cl, RTM_DELTCLASS);
1417 goto out;
1418 case RTM_GETTCLASS:
1419 err = tclass_notify(skb, n, q, cl, RTM_NEWTCLASS);
1420 goto out;
1421 default:
1422 err = -EINVAL;
1423 goto out;
1424 }
1425 }
1426
1427 new_cl = cl;
Patrick McHardyde6d5cd2009-09-04 06:41:16 +00001428 err = -EOPNOTSUPP;
1429 if (cops->change)
1430 err = cops->change(q, clid, pid, tca, &new_cl);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001431 if (err == 0)
1432 tclass_notify(skb, n, q, new_cl, RTM_NEWTCLASS);
1433
1434out:
1435 if (cl)
1436 cops->put(q, cl);
1437
1438 return err;
1439}
1440
1441
1442static int tc_fill_tclass(struct sk_buff *skb, struct Qdisc *q,
1443 unsigned long cl,
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001444 u32 pid, u32 seq, u16 flags, int event)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001445{
1446 struct tcmsg *tcm;
1447 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001448 unsigned char *b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001449 struct gnet_dump d;
Eric Dumazet20fea082007-11-14 01:44:41 -08001450 const struct Qdisc_class_ops *cl_ops = q->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001451
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001452 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001453 tcm = NLMSG_DATA(nlh);
1454 tcm->tcm_family = AF_UNSPEC;
Eric Dumazet16ebb5e2009-09-02 02:40:09 +00001455 tcm->tcm__pad1 = 0;
1456 tcm->tcm__pad2 = 0;
David S. Miller5ce2d482008-07-08 17:06:30 -07001457 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001458 tcm->tcm_parent = q->handle;
1459 tcm->tcm_handle = q->handle;
1460 tcm->tcm_info = 0;
Patrick McHardy57e1c482008-01-23 20:34:28 -08001461 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001462 if (cl_ops->dump && cl_ops->dump(q, cl, skb, tcm) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001463 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001464
Jarek Poplawski102396a2008-08-29 14:21:52 -07001465 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
1466 qdisc_root_sleeping_lock(q), &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001467 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001468
1469 if (cl_ops->dump_stats && cl_ops->dump_stats(q, cl, &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001470 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001471
1472 if (gnet_stats_finish_copy(&d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001473 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001474
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001475 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001476 return skb->len;
1477
1478nlmsg_failure:
Patrick McHardy1e904742008-01-22 22:11:17 -08001479nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -07001480 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001481 return -1;
1482}
1483
1484static int tclass_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1485 struct Qdisc *q, unsigned long cl, int event)
1486{
1487 struct sk_buff *skb;
1488 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1489
1490 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1491 if (!skb)
1492 return -ENOBUFS;
1493
1494 if (tc_fill_tclass(skb, q, cl, pid, n->nlmsg_seq, 0, event) < 0) {
1495 kfree_skb(skb);
1496 return -EINVAL;
1497 }
1498
Denis V. Lunev97c53ca2007-11-19 22:26:51 -08001499 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001500}
1501
1502struct qdisc_dump_args
1503{
1504 struct qdisc_walker w;
1505 struct sk_buff *skb;
1506 struct netlink_callback *cb;
1507};
1508
1509static int qdisc_class_dump(struct Qdisc *q, unsigned long cl, struct qdisc_walker *arg)
1510{
1511 struct qdisc_dump_args *a = (struct qdisc_dump_args *)arg;
1512
1513 return tc_fill_tclass(a->skb, q, cl, NETLINK_CB(a->cb->skb).pid,
1514 a->cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWTCLASS);
1515}
1516
David S. Miller30723672008-07-18 22:50:15 -07001517static int tc_dump_tclass_qdisc(struct Qdisc *q, struct sk_buff *skb,
1518 struct tcmsg *tcm, struct netlink_callback *cb,
1519 int *t_p, int s_t)
1520{
1521 struct qdisc_dump_args arg;
1522
1523 if (tc_qdisc_dump_ignore(q) ||
1524 *t_p < s_t || !q->ops->cl_ops ||
1525 (tcm->tcm_parent &&
1526 TC_H_MAJ(tcm->tcm_parent) != q->handle)) {
1527 (*t_p)++;
1528 return 0;
1529 }
1530 if (*t_p > s_t)
1531 memset(&cb->args[1], 0, sizeof(cb->args)-sizeof(cb->args[0]));
1532 arg.w.fn = qdisc_class_dump;
1533 arg.skb = skb;
1534 arg.cb = cb;
1535 arg.w.stop = 0;
1536 arg.w.skip = cb->args[1];
1537 arg.w.count = 0;
1538 q->ops->cl_ops->walk(q, &arg.w);
1539 cb->args[1] = arg.w.count;
1540 if (arg.w.stop)
1541 return -1;
1542 (*t_p)++;
1543 return 0;
1544}
1545
1546static int tc_dump_tclass_root(struct Qdisc *root, struct sk_buff *skb,
1547 struct tcmsg *tcm, struct netlink_callback *cb,
1548 int *t_p, int s_t)
1549{
1550 struct Qdisc *q;
1551
1552 if (!root)
1553 return 0;
1554
1555 if (tc_dump_tclass_qdisc(root, skb, tcm, cb, t_p, s_t) < 0)
1556 return -1;
1557
1558 list_for_each_entry(q, &root->list, list) {
1559 if (tc_dump_tclass_qdisc(q, skb, tcm, cb, t_p, s_t) < 0)
1560 return -1;
1561 }
1562
1563 return 0;
1564}
1565
Linus Torvalds1da177e2005-04-16 15:20:36 -07001566static int tc_dump_tclass(struct sk_buff *skb, struct netlink_callback *cb)
1567{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001568 struct tcmsg *tcm = (struct tcmsg*)NLMSG_DATA(cb->nlh);
David S. Miller30723672008-07-18 22:50:15 -07001569 struct net *net = sock_net(skb->sk);
1570 struct netdev_queue *dev_queue;
1571 struct net_device *dev;
1572 int t, s_t;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001573
Denis V. Lunevb8542722007-12-01 00:21:31 +11001574 if (net != &init_net)
1575 return 0;
1576
Linus Torvalds1da177e2005-04-16 15:20:36 -07001577 if (cb->nlh->nlmsg_len < NLMSG_LENGTH(sizeof(*tcm)))
1578 return 0;
Eric W. Biederman881d9662007-09-17 11:56:21 -07001579 if ((dev = dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001580 return 0;
1581
1582 s_t = cb->args[0];
1583 t = 0;
1584
Patrick McHardyaf356af2009-09-04 06:41:18 +00001585 if (tc_dump_tclass_root(dev->qdisc, skb, tcm, cb, &t, s_t) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001586 goto done;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001587
David S. Miller30723672008-07-18 22:50:15 -07001588 dev_queue = &dev->rx_queue;
David S. Miller8123b422008-08-08 23:23:39 -07001589 if (tc_dump_tclass_root(dev_queue->qdisc_sleeping, skb, tcm, cb, &t, s_t) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001590 goto done;
1591
1592done:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001593 cb->args[0] = t;
1594
1595 dev_put(dev);
1596 return skb->len;
1597}
1598
1599/* Main classifier routine: scans classifier chain attached
1600 to this qdisc, (optionally) tests for protocol and asks
1601 specific classifiers.
1602 */
Patrick McHardy73ca4912007-07-15 00:02:31 -07001603int tc_classify_compat(struct sk_buff *skb, struct tcf_proto *tp,
1604 struct tcf_result *res)
1605{
1606 __be16 protocol = skb->protocol;
1607 int err = 0;
1608
1609 for (; tp; tp = tp->next) {
1610 if ((tp->protocol == protocol ||
1611 tp->protocol == htons(ETH_P_ALL)) &&
1612 (err = tp->classify(skb, tp, res)) >= 0) {
1613#ifdef CONFIG_NET_CLS_ACT
1614 if (err != TC_ACT_RECLASSIFY && skb->tc_verd)
1615 skb->tc_verd = SET_TC_VERD(skb->tc_verd, 0);
1616#endif
1617 return err;
1618 }
1619 }
1620 return -1;
1621}
1622EXPORT_SYMBOL(tc_classify_compat);
1623
Linus Torvalds1da177e2005-04-16 15:20:36 -07001624int tc_classify(struct sk_buff *skb, struct tcf_proto *tp,
Patrick McHardy73ca4912007-07-15 00:02:31 -07001625 struct tcf_result *res)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001626{
1627 int err = 0;
Patrick McHardy73ca4912007-07-15 00:02:31 -07001628 __be16 protocol;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001629#ifdef CONFIG_NET_CLS_ACT
1630 struct tcf_proto *otp = tp;
1631reclassify:
1632#endif
1633 protocol = skb->protocol;
1634
Patrick McHardy73ca4912007-07-15 00:02:31 -07001635 err = tc_classify_compat(skb, tp, res);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001636#ifdef CONFIG_NET_CLS_ACT
Patrick McHardy73ca4912007-07-15 00:02:31 -07001637 if (err == TC_ACT_RECLASSIFY) {
1638 u32 verd = G_TC_VERD(skb->tc_verd);
1639 tp = otp;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001640
Patrick McHardy73ca4912007-07-15 00:02:31 -07001641 if (verd++ >= MAX_REC_LOOP) {
1642 printk("rule prio %u protocol %02x reclassify loop, "
1643 "packet dropped\n",
1644 tp->prio&0xffff, ntohs(tp->protocol));
1645 return TC_ACT_SHOT;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001646 }
Patrick McHardy73ca4912007-07-15 00:02:31 -07001647 skb->tc_verd = SET_TC_VERD(skb->tc_verd, verd);
1648 goto reclassify;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001649 }
Patrick McHardy73ca4912007-07-15 00:02:31 -07001650#endif
1651 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001652}
Patrick McHardy73ca4912007-07-15 00:02:31 -07001653EXPORT_SYMBOL(tc_classify);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001654
Patrick McHardya48b5a62007-03-23 11:29:43 -07001655void tcf_destroy(struct tcf_proto *tp)
1656{
1657 tp->ops->destroy(tp);
1658 module_put(tp->ops->owner);
1659 kfree(tp);
1660}
1661
Patrick McHardyff31ab52008-07-01 19:52:38 -07001662void tcf_destroy_chain(struct tcf_proto **fl)
Patrick McHardya48b5a62007-03-23 11:29:43 -07001663{
1664 struct tcf_proto *tp;
1665
Patrick McHardyff31ab52008-07-01 19:52:38 -07001666 while ((tp = *fl) != NULL) {
1667 *fl = tp->next;
Patrick McHardya48b5a62007-03-23 11:29:43 -07001668 tcf_destroy(tp);
1669 }
1670}
1671EXPORT_SYMBOL(tcf_destroy_chain);
1672
Linus Torvalds1da177e2005-04-16 15:20:36 -07001673#ifdef CONFIG_PROC_FS
1674static int psched_show(struct seq_file *seq, void *v)
1675{
Patrick McHardy3c0cfc12007-10-10 16:32:41 -07001676 struct timespec ts;
1677
1678 hrtimer_get_res(CLOCK_MONOTONIC, &ts);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001679 seq_printf(seq, "%08x %08x %08x %08x\n",
Jarek Poplawskica44d6e2009-06-15 02:31:47 -07001680 (u32)NSEC_PER_USEC, (u32)PSCHED_TICKS2NS(1),
Patrick McHardy514bca32007-03-16 12:34:52 -07001681 1000000,
Patrick McHardy3c0cfc12007-10-10 16:32:41 -07001682 (u32)NSEC_PER_SEC/(u32)ktime_to_ns(timespec_to_ktime(ts)));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001683
1684 return 0;
1685}
1686
1687static int psched_open(struct inode *inode, struct file *file)
1688{
1689 return single_open(file, psched_show, PDE(inode)->data);
1690}
1691
Arjan van de Venda7071d2007-02-12 00:55:36 -08001692static const struct file_operations psched_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001693 .owner = THIS_MODULE,
1694 .open = psched_open,
1695 .read = seq_read,
1696 .llseek = seq_lseek,
1697 .release = single_release,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001698};
Linus Torvalds1da177e2005-04-16 15:20:36 -07001699#endif
1700
Linus Torvalds1da177e2005-04-16 15:20:36 -07001701static int __init pktsched_init(void)
1702{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001703 register_qdisc(&pfifo_qdisc_ops);
1704 register_qdisc(&bfifo_qdisc_ops);
David S. Miller6ec1c692009-09-06 01:58:51 -07001705 register_qdisc(&mq_qdisc_ops);
Eric W. Biederman457c4cb2007-09-12 12:01:34 +02001706 proc_net_fops_create(&init_net, "psched", 0, &psched_fops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001707
Thomas Grafbe577dd2007-03-22 11:55:50 -07001708 rtnl_register(PF_UNSPEC, RTM_NEWQDISC, tc_modify_qdisc, NULL);
1709 rtnl_register(PF_UNSPEC, RTM_DELQDISC, tc_get_qdisc, NULL);
1710 rtnl_register(PF_UNSPEC, RTM_GETQDISC, tc_get_qdisc, tc_dump_qdisc);
1711 rtnl_register(PF_UNSPEC, RTM_NEWTCLASS, tc_ctl_tclass, NULL);
1712 rtnl_register(PF_UNSPEC, RTM_DELTCLASS, tc_ctl_tclass, NULL);
1713 rtnl_register(PF_UNSPEC, RTM_GETTCLASS, tc_ctl_tclass, tc_dump_tclass);
1714
Linus Torvalds1da177e2005-04-16 15:20:36 -07001715 return 0;
1716}
1717
1718subsys_initcall(pktsched_init);