blob: 0fc4a18fd96f7d83ec21e95bc6d0740025f6a219 [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))
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700210 list_add_tail(&q->list, &qdisc_root_sleeping(q)->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{
David S. Miller30723672008-07-18 22:50:15 -0700222 unsigned int i;
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700223 struct Qdisc *q;
224
David S. Miller30723672008-07-18 22:50:15 -0700225 for (i = 0; i < dev->num_tx_queues; i++) {
226 struct netdev_queue *txq = netdev_get_tx_queue(dev, i);
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700227 struct Qdisc *txq_root = txq->qdisc_sleeping;
David S. Miller30723672008-07-18 22:50:15 -0700228
David S. Miller8123b422008-08-08 23:23:39 -0700229 q = qdisc_match_from_root(txq_root, handle);
230 if (q)
Jarek Poplawskif6486d42008-11-25 13:56:06 -0800231 goto out;
Patrick McHardy43effa12006-11-29 17:35:48 -0800232 }
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700233
234 q = qdisc_match_from_root(dev->rx_queue.qdisc_sleeping, handle);
Jarek Poplawskif6486d42008-11-25 13:56:06 -0800235out:
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700236 return q;
Patrick McHardy43effa12006-11-29 17:35:48 -0800237}
238
Linus Torvalds1da177e2005-04-16 15:20:36 -0700239static struct Qdisc *qdisc_leaf(struct Qdisc *p, u32 classid)
240{
241 unsigned long cl;
242 struct Qdisc *leaf;
Eric Dumazet20fea082007-11-14 01:44:41 -0800243 const struct Qdisc_class_ops *cops = p->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700244
245 if (cops == NULL)
246 return NULL;
247 cl = cops->get(p, classid);
248
249 if (cl == 0)
250 return NULL;
251 leaf = cops->leaf(p, cl);
252 cops->put(p, cl);
253 return leaf;
254}
255
256/* Find queueing discipline by name */
257
Patrick McHardy1e904742008-01-22 22:11:17 -0800258static struct Qdisc_ops *qdisc_lookup_ops(struct nlattr *kind)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700259{
260 struct Qdisc_ops *q = NULL;
261
262 if (kind) {
263 read_lock(&qdisc_mod_lock);
264 for (q = qdisc_base; q; q = q->next) {
Patrick McHardy1e904742008-01-22 22:11:17 -0800265 if (nla_strcmp(kind, q->id) == 0) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700266 if (!try_module_get(q->owner))
267 q = NULL;
268 break;
269 }
270 }
271 read_unlock(&qdisc_mod_lock);
272 }
273 return q;
274}
275
276static struct qdisc_rate_table *qdisc_rtab_list;
277
Patrick McHardy1e904742008-01-22 22:11:17 -0800278struct qdisc_rate_table *qdisc_get_rtab(struct tc_ratespec *r, struct nlattr *tab)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700279{
280 struct qdisc_rate_table *rtab;
281
282 for (rtab = qdisc_rtab_list; rtab; rtab = rtab->next) {
283 if (memcmp(&rtab->rate, r, sizeof(struct tc_ratespec)) == 0) {
284 rtab->refcnt++;
285 return rtab;
286 }
287 }
288
Patrick McHardy5feb5e12008-01-23 20:35:19 -0800289 if (tab == NULL || r->rate == 0 || r->cell_log == 0 ||
290 nla_len(tab) != TC_RTAB_SIZE)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700291 return NULL;
292
293 rtab = kmalloc(sizeof(*rtab), GFP_KERNEL);
294 if (rtab) {
295 rtab->rate = *r;
296 rtab->refcnt = 1;
Patrick McHardy1e904742008-01-22 22:11:17 -0800297 memcpy(rtab->data, nla_data(tab), 1024);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700298 rtab->next = qdisc_rtab_list;
299 qdisc_rtab_list = rtab;
300 }
301 return rtab;
302}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800303EXPORT_SYMBOL(qdisc_get_rtab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700304
305void qdisc_put_rtab(struct qdisc_rate_table *tab)
306{
307 struct qdisc_rate_table *rtab, **rtabp;
308
309 if (!tab || --tab->refcnt)
310 return;
311
312 for (rtabp = &qdisc_rtab_list; (rtab=*rtabp) != NULL; rtabp = &rtab->next) {
313 if (rtab == tab) {
314 *rtabp = rtab->next;
315 kfree(rtab);
316 return;
317 }
318 }
319}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800320EXPORT_SYMBOL(qdisc_put_rtab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700321
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700322static LIST_HEAD(qdisc_stab_list);
323static DEFINE_SPINLOCK(qdisc_stab_lock);
324
325static const struct nla_policy stab_policy[TCA_STAB_MAX + 1] = {
326 [TCA_STAB_BASE] = { .len = sizeof(struct tc_sizespec) },
327 [TCA_STAB_DATA] = { .type = NLA_BINARY },
328};
329
330static struct qdisc_size_table *qdisc_get_stab(struct nlattr *opt)
331{
332 struct nlattr *tb[TCA_STAB_MAX + 1];
333 struct qdisc_size_table *stab;
334 struct tc_sizespec *s;
335 unsigned int tsize = 0;
336 u16 *tab = NULL;
337 int err;
338
339 err = nla_parse_nested(tb, TCA_STAB_MAX, opt, stab_policy);
340 if (err < 0)
341 return ERR_PTR(err);
342 if (!tb[TCA_STAB_BASE])
343 return ERR_PTR(-EINVAL);
344
345 s = nla_data(tb[TCA_STAB_BASE]);
346
347 if (s->tsize > 0) {
348 if (!tb[TCA_STAB_DATA])
349 return ERR_PTR(-EINVAL);
350 tab = nla_data(tb[TCA_STAB_DATA]);
351 tsize = nla_len(tb[TCA_STAB_DATA]) / sizeof(u16);
352 }
353
354 if (!s || tsize != s->tsize || (!tab && tsize > 0))
355 return ERR_PTR(-EINVAL);
356
David S. Millerf3b96052008-08-18 22:33:05 -0700357 spin_lock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700358
359 list_for_each_entry(stab, &qdisc_stab_list, list) {
360 if (memcmp(&stab->szopts, s, sizeof(*s)))
361 continue;
362 if (tsize > 0 && memcmp(stab->data, tab, tsize * sizeof(u16)))
363 continue;
364 stab->refcnt++;
David S. Millerf3b96052008-08-18 22:33:05 -0700365 spin_unlock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700366 return stab;
367 }
368
David S. Millerf3b96052008-08-18 22:33:05 -0700369 spin_unlock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700370
371 stab = kmalloc(sizeof(*stab) + tsize * sizeof(u16), GFP_KERNEL);
372 if (!stab)
373 return ERR_PTR(-ENOMEM);
374
375 stab->refcnt = 1;
376 stab->szopts = *s;
377 if (tsize > 0)
378 memcpy(stab->data, tab, tsize * sizeof(u16));
379
David S. Millerf3b96052008-08-18 22:33:05 -0700380 spin_lock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700381 list_add_tail(&stab->list, &qdisc_stab_list);
David S. Millerf3b96052008-08-18 22:33:05 -0700382 spin_unlock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700383
384 return stab;
385}
386
387void qdisc_put_stab(struct qdisc_size_table *tab)
388{
389 if (!tab)
390 return;
391
David S. Millerf3b96052008-08-18 22:33:05 -0700392 spin_lock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700393
394 if (--tab->refcnt == 0) {
395 list_del(&tab->list);
396 kfree(tab);
397 }
398
David S. Millerf3b96052008-08-18 22:33:05 -0700399 spin_unlock(&qdisc_stab_lock);
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700400}
401EXPORT_SYMBOL(qdisc_put_stab);
402
403static int qdisc_dump_stab(struct sk_buff *skb, struct qdisc_size_table *stab)
404{
405 struct nlattr *nest;
406
407 nest = nla_nest_start(skb, TCA_STAB);
Patrick McHardy3aa46142008-11-20 04:07:14 -0800408 if (nest == NULL)
409 goto nla_put_failure;
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700410 NLA_PUT(skb, TCA_STAB_BASE, sizeof(stab->szopts), &stab->szopts);
411 nla_nest_end(skb, nest);
412
413 return skb->len;
414
415nla_put_failure:
416 return -1;
417}
418
419void qdisc_calculate_pkt_len(struct sk_buff *skb, struct qdisc_size_table *stab)
420{
421 int pkt_len, slot;
422
423 pkt_len = skb->len + stab->szopts.overhead;
424 if (unlikely(!stab->szopts.tsize))
425 goto out;
426
427 slot = pkt_len + stab->szopts.cell_align;
428 if (unlikely(slot < 0))
429 slot = 0;
430
431 slot >>= stab->szopts.cell_log;
432 if (likely(slot < stab->szopts.tsize))
433 pkt_len = stab->data[slot];
434 else
435 pkt_len = stab->data[stab->szopts.tsize - 1] *
436 (slot / stab->szopts.tsize) +
437 stab->data[slot % stab->szopts.tsize];
438
439 pkt_len <<= stab->szopts.size_log;
440out:
441 if (unlikely(pkt_len < 1))
442 pkt_len = 1;
443 qdisc_skb_cb(skb)->pkt_len = pkt_len;
444}
445EXPORT_SYMBOL(qdisc_calculate_pkt_len);
446
Patrick McHardy41794772007-03-16 01:19:15 -0700447static enum hrtimer_restart qdisc_watchdog(struct hrtimer *timer)
448{
449 struct qdisc_watchdog *wd = container_of(timer, struct qdisc_watchdog,
450 timer);
451
452 wd->qdisc->flags &= ~TCQ_F_THROTTLED;
David S. Miller8608db02008-08-18 20:51:18 -0700453 __netif_schedule(qdisc_root(wd->qdisc));
Stephen Hemminger19365022007-03-22 12:18:35 -0700454
Patrick McHardy41794772007-03-16 01:19:15 -0700455 return HRTIMER_NORESTART;
456}
457
458void qdisc_watchdog_init(struct qdisc_watchdog *wd, struct Qdisc *qdisc)
459{
460 hrtimer_init(&wd->timer, CLOCK_MONOTONIC, HRTIMER_MODE_ABS);
461 wd->timer.function = qdisc_watchdog;
462 wd->qdisc = qdisc;
463}
464EXPORT_SYMBOL(qdisc_watchdog_init);
465
466void qdisc_watchdog_schedule(struct qdisc_watchdog *wd, psched_time_t expires)
467{
468 ktime_t time;
469
Jarek Poplawski2540e052008-08-21 05:11:14 -0700470 if (test_bit(__QDISC_STATE_DEACTIVATED,
471 &qdisc_root_sleeping(wd->qdisc)->state))
472 return;
473
Patrick McHardy41794772007-03-16 01:19:15 -0700474 wd->qdisc->flags |= TCQ_F_THROTTLED;
475 time = ktime_set(0, 0);
476 time = ktime_add_ns(time, PSCHED_US2NS(expires));
477 hrtimer_start(&wd->timer, time, HRTIMER_MODE_ABS);
478}
479EXPORT_SYMBOL(qdisc_watchdog_schedule);
480
481void qdisc_watchdog_cancel(struct qdisc_watchdog *wd)
482{
483 hrtimer_cancel(&wd->timer);
484 wd->qdisc->flags &= ~TCQ_F_THROTTLED;
485}
486EXPORT_SYMBOL(qdisc_watchdog_cancel);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700487
Adrian Bunka94f7792008-07-22 14:20:11 -0700488static struct hlist_head *qdisc_class_hash_alloc(unsigned int n)
Patrick McHardy6fe1c7a2008-07-05 23:21:31 -0700489{
490 unsigned int size = n * sizeof(struct hlist_head), i;
491 struct hlist_head *h;
492
493 if (size <= PAGE_SIZE)
494 h = kmalloc(size, GFP_KERNEL);
495 else
496 h = (struct hlist_head *)
497 __get_free_pages(GFP_KERNEL, get_order(size));
498
499 if (h != NULL) {
500 for (i = 0; i < n; i++)
501 INIT_HLIST_HEAD(&h[i]);
502 }
503 return h;
504}
505
506static void qdisc_class_hash_free(struct hlist_head *h, unsigned int n)
507{
508 unsigned int size = n * sizeof(struct hlist_head);
509
510 if (size <= PAGE_SIZE)
511 kfree(h);
512 else
513 free_pages((unsigned long)h, get_order(size));
514}
515
516void qdisc_class_hash_grow(struct Qdisc *sch, struct Qdisc_class_hash *clhash)
517{
518 struct Qdisc_class_common *cl;
519 struct hlist_node *n, *next;
520 struct hlist_head *nhash, *ohash;
521 unsigned int nsize, nmask, osize;
522 unsigned int i, h;
523
524 /* Rehash when load factor exceeds 0.75 */
525 if (clhash->hashelems * 4 <= clhash->hashsize * 3)
526 return;
527 nsize = clhash->hashsize * 2;
528 nmask = nsize - 1;
529 nhash = qdisc_class_hash_alloc(nsize);
530 if (nhash == NULL)
531 return;
532
533 ohash = clhash->hash;
534 osize = clhash->hashsize;
535
536 sch_tree_lock(sch);
537 for (i = 0; i < osize; i++) {
538 hlist_for_each_entry_safe(cl, n, next, &ohash[i], hnode) {
539 h = qdisc_class_hash(cl->classid, nmask);
540 hlist_add_head(&cl->hnode, &nhash[h]);
541 }
542 }
543 clhash->hash = nhash;
544 clhash->hashsize = nsize;
545 clhash->hashmask = nmask;
546 sch_tree_unlock(sch);
547
548 qdisc_class_hash_free(ohash, osize);
549}
550EXPORT_SYMBOL(qdisc_class_hash_grow);
551
552int qdisc_class_hash_init(struct Qdisc_class_hash *clhash)
553{
554 unsigned int size = 4;
555
556 clhash->hash = qdisc_class_hash_alloc(size);
557 if (clhash->hash == NULL)
558 return -ENOMEM;
559 clhash->hashsize = size;
560 clhash->hashmask = size - 1;
561 clhash->hashelems = 0;
562 return 0;
563}
564EXPORT_SYMBOL(qdisc_class_hash_init);
565
566void qdisc_class_hash_destroy(struct Qdisc_class_hash *clhash)
567{
568 qdisc_class_hash_free(clhash->hash, clhash->hashsize);
569}
570EXPORT_SYMBOL(qdisc_class_hash_destroy);
571
572void qdisc_class_hash_insert(struct Qdisc_class_hash *clhash,
573 struct Qdisc_class_common *cl)
574{
575 unsigned int h;
576
577 INIT_HLIST_NODE(&cl->hnode);
578 h = qdisc_class_hash(cl->classid, clhash->hashmask);
579 hlist_add_head(&cl->hnode, &clhash->hash[h]);
580 clhash->hashelems++;
581}
582EXPORT_SYMBOL(qdisc_class_hash_insert);
583
584void qdisc_class_hash_remove(struct Qdisc_class_hash *clhash,
585 struct Qdisc_class_common *cl)
586{
587 hlist_del(&cl->hnode);
588 clhash->hashelems--;
589}
590EXPORT_SYMBOL(qdisc_class_hash_remove);
591
Linus Torvalds1da177e2005-04-16 15:20:36 -0700592/* Allocate an unique handle from space managed by kernel */
593
594static u32 qdisc_alloc_handle(struct net_device *dev)
595{
596 int i = 0x10000;
597 static u32 autohandle = TC_H_MAKE(0x80000000U, 0);
598
599 do {
600 autohandle += TC_H_MAKE(0x10000U, 0);
601 if (autohandle == TC_H_MAKE(TC_H_ROOT, 0))
602 autohandle = TC_H_MAKE(0x80000000U, 0);
603 } while (qdisc_lookup(dev, autohandle) && --i > 0);
604
605 return i>0 ? autohandle : 0;
606}
607
David S. Miller99194cf2008-07-17 04:54:10 -0700608/* Attach toplevel qdisc to device queue. */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700609
David S. Miller99194cf2008-07-17 04:54:10 -0700610static struct Qdisc *dev_graft_qdisc(struct netdev_queue *dev_queue,
611 struct Qdisc *qdisc)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700612{
David S. Miller8d50b532008-07-30 02:37:46 -0700613 struct Qdisc *oqdisc = dev_queue->qdisc_sleeping;
David S. Miller53049972008-07-16 03:00:19 -0700614 spinlock_t *root_lock;
David S. Miller53049972008-07-16 03:00:19 -0700615
Jarek Poplawski666d9bb2008-08-27 02:12:52 -0700616 root_lock = qdisc_lock(oqdisc);
David S. Miller53049972008-07-16 03:00:19 -0700617 spin_lock_bh(root_lock);
618
David S. Miller8d50b532008-07-30 02:37:46 -0700619 /* Prune old scheduler */
620 if (oqdisc && atomic_read(&oqdisc->refcnt) <= 1)
621 qdisc_reset(oqdisc);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700622
David S. Miller8d50b532008-07-30 02:37:46 -0700623 /* ... and graft new one */
624 if (qdisc == NULL)
625 qdisc = &noop_qdisc;
626 dev_queue->qdisc_sleeping = qdisc;
Jarek Poplawskif7a54c12008-08-27 02:22:07 -0700627 rcu_assign_pointer(dev_queue->qdisc, &noop_qdisc);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700628
David S. Miller53049972008-07-16 03:00:19 -0700629 spin_unlock_bh(root_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700630
Linus Torvalds1da177e2005-04-16 15:20:36 -0700631 return oqdisc;
632}
633
Patrick McHardy43effa12006-11-29 17:35:48 -0800634void qdisc_tree_decrease_qlen(struct Qdisc *sch, unsigned int n)
635{
Eric Dumazet20fea082007-11-14 01:44:41 -0800636 const struct Qdisc_class_ops *cops;
Patrick McHardy43effa12006-11-29 17:35:48 -0800637 unsigned long cl;
638 u32 parentid;
639
640 if (n == 0)
641 return;
642 while ((parentid = sch->parent)) {
Jarek Poplawski066a3b52008-04-14 15:10:42 -0700643 if (TC_H_MAJ(parentid) == TC_H_MAJ(TC_H_INGRESS))
644 return;
645
David S. Miller5ce2d482008-07-08 17:06:30 -0700646 sch = qdisc_lookup(qdisc_dev(sch), TC_H_MAJ(parentid));
Patrick McHardyffc8fef2007-07-30 17:11:50 -0700647 if (sch == NULL) {
648 WARN_ON(parentid != TC_H_ROOT);
649 return;
650 }
Patrick McHardy43effa12006-11-29 17:35:48 -0800651 cops = sch->ops->cl_ops;
652 if (cops->qlen_notify) {
653 cl = cops->get(sch, parentid);
654 cops->qlen_notify(sch, cl);
655 cops->put(sch, cl);
656 }
657 sch->q.qlen -= n;
658 }
659}
660EXPORT_SYMBOL(qdisc_tree_decrease_qlen);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700661
David S. Miller99194cf2008-07-17 04:54:10 -0700662static void notify_and_destroy(struct sk_buff *skb, struct nlmsghdr *n, u32 clid,
663 struct Qdisc *old, struct Qdisc *new)
664{
665 if (new || old)
666 qdisc_notify(skb, n, clid, old, new);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700667
David S. Miller4d8863a2008-08-18 21:03:15 -0700668 if (old)
David S. Miller99194cf2008-07-17 04:54:10 -0700669 qdisc_destroy(old);
David S. Miller99194cf2008-07-17 04:54:10 -0700670}
671
672/* Graft qdisc "new" to class "classid" of qdisc "parent" or
673 * to device "dev".
674 *
675 * When appropriate send a netlink notification using 'skb'
676 * and "n".
677 *
678 * On success, destroy old qdisc.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700679 */
680
681static int qdisc_graft(struct net_device *dev, struct Qdisc *parent,
David S. Miller99194cf2008-07-17 04:54:10 -0700682 struct sk_buff *skb, struct nlmsghdr *n, u32 classid,
683 struct Qdisc *new, struct Qdisc *old)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700684{
David S. Miller99194cf2008-07-17 04:54:10 -0700685 struct Qdisc *q = old;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700686 int err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700687
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900688 if (parent == NULL) {
David S. Miller99194cf2008-07-17 04:54:10 -0700689 unsigned int i, num_q, ingress;
690
691 ingress = 0;
692 num_q = dev->num_tx_queues;
David S. Miller8d50b532008-07-30 02:37:46 -0700693 if ((q && q->flags & TCQ_F_INGRESS) ||
694 (new && new->flags & TCQ_F_INGRESS)) {
David S. Miller99194cf2008-07-17 04:54:10 -0700695 num_q = 1;
696 ingress = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700697 }
David S. Miller99194cf2008-07-17 04:54:10 -0700698
699 if (dev->flags & IFF_UP)
700 dev_deactivate(dev);
701
702 for (i = 0; i < num_q; i++) {
703 struct netdev_queue *dev_queue = &dev->rx_queue;
704
705 if (!ingress)
706 dev_queue = netdev_get_tx_queue(dev, i);
707
David S. Miller8d50b532008-07-30 02:37:46 -0700708 old = dev_graft_qdisc(dev_queue, new);
709 if (new && i > 0)
710 atomic_inc(&new->refcnt);
711
David S. Miller99194cf2008-07-17 04:54:10 -0700712 notify_and_destroy(skb, n, classid, old, new);
713 }
714
715 if (dev->flags & IFF_UP)
716 dev_activate(dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700717 } else {
Eric Dumazet20fea082007-11-14 01:44:41 -0800718 const struct Qdisc_class_ops *cops = parent->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700719
720 err = -EINVAL;
721
722 if (cops) {
723 unsigned long cl = cops->get(parent, classid);
724 if (cl) {
David S. Miller99194cf2008-07-17 04:54:10 -0700725 err = cops->graft(parent, cl, new, &old);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700726 cops->put(parent, cl);
727 }
728 }
David S. Miller99194cf2008-07-17 04:54:10 -0700729 if (!err)
730 notify_and_destroy(skb, n, classid, old, new);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700731 }
732 return err;
733}
734
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700735/* lockdep annotation is needed for ingress; egress gets it only for name */
736static struct lock_class_key qdisc_tx_lock;
737static struct lock_class_key qdisc_rx_lock;
738
Linus Torvalds1da177e2005-04-16 15:20:36 -0700739/*
740 Allocate and initialize new qdisc.
741
742 Parameters are passed via opt.
743 */
744
745static struct Qdisc *
David S. Millerbb949fb2008-07-08 16:55:56 -0700746qdisc_create(struct net_device *dev, struct netdev_queue *dev_queue,
747 u32 parent, u32 handle, struct nlattr **tca, int *errp)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700748{
749 int err;
Patrick McHardy1e904742008-01-22 22:11:17 -0800750 struct nlattr *kind = tca[TCA_KIND];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700751 struct Qdisc *sch;
752 struct Qdisc_ops *ops;
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700753 struct qdisc_size_table *stab;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700754
755 ops = qdisc_lookup_ops(kind);
Johannes Berg95a5afc2008-10-16 15:24:51 -0700756#ifdef CONFIG_MODULES
Linus Torvalds1da177e2005-04-16 15:20:36 -0700757 if (ops == NULL && kind != NULL) {
758 char name[IFNAMSIZ];
Patrick McHardy1e904742008-01-22 22:11:17 -0800759 if (nla_strlcpy(name, kind, IFNAMSIZ) < IFNAMSIZ) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700760 /* We dropped the RTNL semaphore in order to
761 * perform the module load. So, even if we
762 * succeeded in loading the module we have to
763 * tell the caller to replay the request. We
764 * indicate this using -EAGAIN.
765 * We replay the request because the device may
766 * go away in the mean time.
767 */
768 rtnl_unlock();
769 request_module("sch_%s", name);
770 rtnl_lock();
771 ops = qdisc_lookup_ops(kind);
772 if (ops != NULL) {
773 /* We will try again qdisc_lookup_ops,
774 * so don't keep a reference.
775 */
776 module_put(ops->owner);
777 err = -EAGAIN;
778 goto err_out;
779 }
780 }
781 }
782#endif
783
Jamal Hadi Salimb9e2cc02006-08-03 16:36:51 -0700784 err = -ENOENT;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700785 if (ops == NULL)
786 goto err_out;
787
David S. Miller5ce2d482008-07-08 17:06:30 -0700788 sch = qdisc_alloc(dev_queue, ops);
Thomas Graf3d54b822005-07-05 14:15:09 -0700789 if (IS_ERR(sch)) {
790 err = PTR_ERR(sch);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700791 goto err_out2;
Thomas Graf3d54b822005-07-05 14:15:09 -0700792 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700793
Patrick McHardyffc8fef2007-07-30 17:11:50 -0700794 sch->parent = parent;
795
Thomas Graf3d54b822005-07-05 14:15:09 -0700796 if (handle == TC_H_INGRESS) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700797 sch->flags |= TCQ_F_INGRESS;
Thomas Graf3d54b822005-07-05 14:15:09 -0700798 handle = TC_H_MAKE(TC_H_INGRESS, 0);
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700799 lockdep_set_class(qdisc_lock(sch), &qdisc_rx_lock);
Patrick McHardyfd44de72007-04-16 17:07:08 -0700800 } else {
Patrick McHardyfd44de72007-04-16 17:07:08 -0700801 if (handle == 0) {
802 handle = qdisc_alloc_handle(dev);
803 err = -ENOMEM;
804 if (handle == 0)
805 goto err_out3;
806 }
Jarek Poplawski25bfcd52008-08-18 20:53:34 -0700807 lockdep_set_class(qdisc_lock(sch), &qdisc_tx_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700808 }
809
Thomas Graf3d54b822005-07-05 14:15:09 -0700810 sch->handle = handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700811
Patrick McHardy1e904742008-01-22 22:11:17 -0800812 if (!ops->init || (err = ops->init(sch, tca[TCA_OPTIONS])) == 0) {
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700813 if (tca[TCA_STAB]) {
814 stab = qdisc_get_stab(tca[TCA_STAB]);
815 if (IS_ERR(stab)) {
816 err = PTR_ERR(stab);
817 goto err_out3;
818 }
819 sch->stab = stab;
820 }
Patrick McHardy1e904742008-01-22 22:11:17 -0800821 if (tca[TCA_RATE]) {
Jarek Poplawskif6f9b932008-08-27 02:25:17 -0700822 spinlock_t *root_lock;
823
824 if ((sch->parent != TC_H_ROOT) &&
825 !(sch->flags & TCQ_F_INGRESS))
826 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]);
Thomas Graf023e09a2005-07-05 14:15:53 -0700832 if (err) {
833 /*
834 * Any broken qdiscs that would require
835 * a ops->reset() here? The qdisc was never
836 * in action so it shouldn't be necessary.
837 */
838 if (ops->destroy)
839 ops->destroy(sch);
840 goto err_out3;
841 }
842 }
Jarek Poplawskif6e0b232008-08-22 03:24:05 -0700843
844 qdisc_list_add(sch);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700845
Linus Torvalds1da177e2005-04-16 15:20:36 -0700846 return sch;
847 }
848err_out3:
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700849 qdisc_put_stab(sch->stab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700850 dev_put(dev);
Thomas Graf3d54b822005-07-05 14:15:09 -0700851 kfree((char *) sch - sch->padded);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700852err_out2:
853 module_put(ops->owner);
854err_out:
855 *errp = err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700856 return NULL;
857}
858
Patrick McHardy1e904742008-01-22 22:11:17 -0800859static int qdisc_change(struct Qdisc *sch, struct nlattr **tca)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700860{
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700861 struct qdisc_size_table *stab = NULL;
862 int err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700863
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700864 if (tca[TCA_OPTIONS]) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700865 if (sch->ops->change == NULL)
866 return -EINVAL;
Patrick McHardy1e904742008-01-22 22:11:17 -0800867 err = sch->ops->change(sch, tca[TCA_OPTIONS]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700868 if (err)
869 return err;
870 }
Jussi Kivilinna175f9c12008-07-20 00:08:47 -0700871
872 if (tca[TCA_STAB]) {
873 stab = qdisc_get_stab(tca[TCA_STAB]);
874 if (IS_ERR(stab))
875 return PTR_ERR(stab);
876 }
877
878 qdisc_put_stab(sch->stab);
879 sch->stab = stab;
880
Patrick McHardy1e904742008-01-22 22:11:17 -0800881 if (tca[TCA_RATE])
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800882 /* NB: ignores errors from replace_estimator
883 because change can't be undone. */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700884 gen_replace_estimator(&sch->bstats, &sch->rate_est,
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800885 qdisc_root_sleeping_lock(sch),
886 tca[TCA_RATE]);
887
Linus Torvalds1da177e2005-04-16 15:20:36 -0700888 return 0;
889}
890
891struct check_loop_arg
892{
893 struct qdisc_walker w;
894 struct Qdisc *p;
895 int depth;
896};
897
898static int check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w);
899
900static int check_loop(struct Qdisc *q, struct Qdisc *p, int depth)
901{
902 struct check_loop_arg arg;
903
904 if (q->ops->cl_ops == NULL)
905 return 0;
906
907 arg.w.stop = arg.w.skip = arg.w.count = 0;
908 arg.w.fn = check_loop_fn;
909 arg.depth = depth;
910 arg.p = p;
911 q->ops->cl_ops->walk(q, &arg.w);
912 return arg.w.stop ? -ELOOP : 0;
913}
914
915static int
916check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w)
917{
918 struct Qdisc *leaf;
Eric Dumazet20fea082007-11-14 01:44:41 -0800919 const struct Qdisc_class_ops *cops = q->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700920 struct check_loop_arg *arg = (struct check_loop_arg *)w;
921
922 leaf = cops->leaf(q, cl);
923 if (leaf) {
924 if (leaf == arg->p || arg->depth > 7)
925 return -ELOOP;
926 return check_loop(leaf, arg->p, arg->depth + 1);
927 }
928 return 0;
929}
930
931/*
932 * Delete/get qdisc.
933 */
934
935static int tc_get_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
936{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900937 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700938 struct tcmsg *tcm = NLMSG_DATA(n);
Patrick McHardy1e904742008-01-22 22:11:17 -0800939 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700940 struct net_device *dev;
941 u32 clid = tcm->tcm_parent;
942 struct Qdisc *q = NULL;
943 struct Qdisc *p = NULL;
944 int err;
945
Denis V. Lunevb8542722007-12-01 00:21:31 +1100946 if (net != &init_net)
947 return -EINVAL;
948
Eric W. Biederman881d9662007-09-17 11:56:21 -0700949 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700950 return -ENODEV;
951
Patrick McHardy1e904742008-01-22 22:11:17 -0800952 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
953 if (err < 0)
954 return err;
955
Linus Torvalds1da177e2005-04-16 15:20:36 -0700956 if (clid) {
957 if (clid != TC_H_ROOT) {
958 if (TC_H_MAJ(clid) != TC_H_MAJ(TC_H_INGRESS)) {
959 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
960 return -ENOENT;
961 q = qdisc_leaf(p, clid);
962 } else { /* ingress */
David S. Miller8123b422008-08-08 23:23:39 -0700963 q = dev->rx_queue.qdisc_sleeping;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900964 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700965 } else {
David S. Millere8a04642008-07-17 00:34:19 -0700966 struct netdev_queue *dev_queue;
967 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Millerb0e1e642008-07-08 17:42:10 -0700968 q = dev_queue->qdisc_sleeping;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700969 }
970 if (!q)
971 return -ENOENT;
972
973 if (tcm->tcm_handle && q->handle != tcm->tcm_handle)
974 return -EINVAL;
975 } else {
976 if ((q = qdisc_lookup(dev, tcm->tcm_handle)) == NULL)
977 return -ENOENT;
978 }
979
Patrick McHardy1e904742008-01-22 22:11:17 -0800980 if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700981 return -EINVAL;
982
983 if (n->nlmsg_type == RTM_DELQDISC) {
984 if (!clid)
985 return -EINVAL;
986 if (q->handle == 0)
987 return -ENOENT;
David S. Miller99194cf2008-07-17 04:54:10 -0700988 if ((err = qdisc_graft(dev, p, skb, n, clid, NULL, q)) != 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700989 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700990 } else {
991 qdisc_notify(skb, n, clid, NULL, q);
992 }
993 return 0;
994}
995
996/*
997 Create/change qdisc.
998 */
999
1000static int tc_modify_qdisc(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
1001{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001002 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001003 struct tcmsg *tcm;
Patrick McHardy1e904742008-01-22 22:11:17 -08001004 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -07001005 struct net_device *dev;
1006 u32 clid;
1007 struct Qdisc *q, *p;
1008 int err;
1009
Denis V. Lunevb8542722007-12-01 00:21:31 +11001010 if (net != &init_net)
1011 return -EINVAL;
1012
Linus Torvalds1da177e2005-04-16 15:20:36 -07001013replay:
1014 /* Reinit, just in case something touches this. */
1015 tcm = NLMSG_DATA(n);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001016 clid = tcm->tcm_parent;
1017 q = p = NULL;
1018
Eric W. Biederman881d9662007-09-17 11:56:21 -07001019 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001020 return -ENODEV;
1021
Patrick McHardy1e904742008-01-22 22:11:17 -08001022 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1023 if (err < 0)
1024 return err;
1025
Linus Torvalds1da177e2005-04-16 15:20:36 -07001026 if (clid) {
1027 if (clid != TC_H_ROOT) {
1028 if (clid != TC_H_INGRESS) {
1029 if ((p = qdisc_lookup(dev, TC_H_MAJ(clid))) == NULL)
1030 return -ENOENT;
1031 q = qdisc_leaf(p, clid);
1032 } else { /*ingress */
David S. Miller8123b422008-08-08 23:23:39 -07001033 q = dev->rx_queue.qdisc_sleeping;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001034 }
1035 } else {
David S. Millere8a04642008-07-17 00:34:19 -07001036 struct netdev_queue *dev_queue;
1037 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Millerb0e1e642008-07-08 17:42:10 -07001038 q = dev_queue->qdisc_sleeping;
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)
David S. Millerbb949fb2008-07-08 16:55:56 -07001115 q = qdisc_create(dev, &dev->rx_queue,
1116 tcm->tcm_parent, tcm->tcm_parent,
Patrick McHardyffc8fef2007-07-30 17:11:50 -07001117 tca, &err);
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001118 else
David S. Millere8a04642008-07-17 00:34:19 -07001119 q = qdisc_create(dev, netdev_get_tx_queue(dev, 0),
David S. Millerbb949fb2008-07-08 16:55:56 -07001120 tcm->tcm_parent, tcm->tcm_handle,
Patrick McHardyffc8fef2007-07-30 17:11:50 -07001121 tca, &err);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001122 if (q == NULL) {
1123 if (err == -EAGAIN)
1124 goto replay;
1125 return err;
1126 }
1127
1128graft:
Ilpo Järvinene5befbd2008-08-18 22:30:01 -07001129 err = qdisc_graft(dev, p, skb, n, clid, q, NULL);
1130 if (err) {
1131 if (q)
1132 qdisc_destroy(q);
1133 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001134 }
Ilpo Järvinene5befbd2008-08-18 22:30:01 -07001135
Linus Torvalds1da177e2005-04-16 15:20:36 -07001136 return 0;
1137}
1138
1139static int tc_fill_qdisc(struct sk_buff *skb, struct Qdisc *q, u32 clid,
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001140 u32 pid, u32 seq, u16 flags, int event)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001141{
1142 struct tcmsg *tcm;
1143 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001144 unsigned char *b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001145 struct gnet_dump d;
1146
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001147 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001148 tcm = NLMSG_DATA(nlh);
1149 tcm->tcm_family = AF_UNSPEC;
Patrick McHardy9ef1d4c2005-06-28 12:55:30 -07001150 tcm->tcm__pad1 = 0;
1151 tcm->tcm__pad2 = 0;
David S. Miller5ce2d482008-07-08 17:06:30 -07001152 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001153 tcm->tcm_parent = clid;
1154 tcm->tcm_handle = q->handle;
1155 tcm->tcm_info = atomic_read(&q->refcnt);
Patrick McHardy57e1c482008-01-23 20:34:28 -08001156 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001157 if (q->ops->dump && q->ops->dump(q, skb) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001158 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001159 q->qstats.qlen = q->q.qlen;
1160
Jussi Kivilinna175f9c12008-07-20 00:08:47 -07001161 if (q->stab && qdisc_dump_stab(skb, q->stab) < 0)
1162 goto nla_put_failure;
1163
Jarek Poplawski102396a2008-08-29 14:21:52 -07001164 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
1165 qdisc_root_sleeping_lock(q), &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001166 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001167
1168 if (q->ops->dump_stats && q->ops->dump_stats(q, &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001169 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001170
1171 if (gnet_stats_copy_basic(&d, &q->bstats) < 0 ||
Linus Torvalds1da177e2005-04-16 15:20:36 -07001172 gnet_stats_copy_rate_est(&d, &q->rate_est) < 0 ||
Linus Torvalds1da177e2005-04-16 15:20:36 -07001173 gnet_stats_copy_queue(&d, &q->qstats) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001174 goto nla_put_failure;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001175
Linus Torvalds1da177e2005-04-16 15:20:36 -07001176 if (gnet_stats_finish_copy(&d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001177 goto nla_put_failure;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001178
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001179 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001180 return skb->len;
1181
1182nlmsg_failure:
Patrick McHardy1e904742008-01-22 22:11:17 -08001183nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -07001184 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001185 return -1;
1186}
1187
1188static int qdisc_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1189 u32 clid, struct Qdisc *old, struct Qdisc *new)
1190{
1191 struct sk_buff *skb;
1192 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1193
1194 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1195 if (!skb)
1196 return -ENOBUFS;
1197
1198 if (old && old->handle) {
1199 if (tc_fill_qdisc(skb, old, clid, pid, n->nlmsg_seq, 0, RTM_DELQDISC) < 0)
1200 goto err_out;
1201 }
1202 if (new) {
1203 if (tc_fill_qdisc(skb, new, clid, pid, n->nlmsg_seq, old ? NLM_F_REPLACE : 0, RTM_NEWQDISC) < 0)
1204 goto err_out;
1205 }
1206
1207 if (skb->len)
Denis V. Lunev97c53ca2007-11-19 22:26:51 -08001208 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001209
1210err_out:
1211 kfree_skb(skb);
1212 return -EINVAL;
1213}
1214
David S. Miller30723672008-07-18 22:50:15 -07001215static bool tc_qdisc_dump_ignore(struct Qdisc *q)
1216{
1217 return (q->flags & TCQ_F_BUILTIN) ? true : false;
1218}
1219
1220static int tc_dump_qdisc_root(struct Qdisc *root, struct sk_buff *skb,
1221 struct netlink_callback *cb,
1222 int *q_idx_p, int s_q_idx)
1223{
1224 int ret = 0, q_idx = *q_idx_p;
1225 struct Qdisc *q;
1226
1227 if (!root)
1228 return 0;
1229
1230 q = root;
1231 if (q_idx < s_q_idx) {
1232 q_idx++;
1233 } else {
1234 if (!tc_qdisc_dump_ignore(q) &&
1235 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1236 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1237 goto done;
1238 q_idx++;
1239 }
1240 list_for_each_entry(q, &root->list, list) {
1241 if (q_idx < s_q_idx) {
1242 q_idx++;
1243 continue;
1244 }
1245 if (!tc_qdisc_dump_ignore(q) &&
1246 tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).pid,
1247 cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWQDISC) <= 0)
1248 goto done;
1249 q_idx++;
1250 }
1251
1252out:
1253 *q_idx_p = q_idx;
1254 return ret;
1255done:
1256 ret = -1;
1257 goto out;
1258}
1259
Linus Torvalds1da177e2005-04-16 15:20:36 -07001260static int tc_dump_qdisc(struct sk_buff *skb, struct netlink_callback *cb)
1261{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001262 struct net *net = sock_net(skb->sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001263 int idx, q_idx;
1264 int s_idx, s_q_idx;
1265 struct net_device *dev;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001266
Denis V. Lunevb8542722007-12-01 00:21:31 +11001267 if (net != &init_net)
1268 return 0;
1269
Linus Torvalds1da177e2005-04-16 15:20:36 -07001270 s_idx = cb->args[0];
1271 s_q_idx = q_idx = cb->args[1];
1272 read_lock(&dev_base_lock);
Pavel Emelianov7562f872007-05-03 15:13:45 -07001273 idx = 0;
Eric W. Biederman881d9662007-09-17 11:56:21 -07001274 for_each_netdev(&init_net, dev) {
David S. Miller30723672008-07-18 22:50:15 -07001275 struct netdev_queue *dev_queue;
1276
Linus Torvalds1da177e2005-04-16 15:20:36 -07001277 if (idx < s_idx)
Pavel Emelianov7562f872007-05-03 15:13:45 -07001278 goto cont;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001279 if (idx > s_idx)
1280 s_q_idx = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001281 q_idx = 0;
David S. Miller30723672008-07-18 22:50:15 -07001282
1283 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Miller827ebd62008-08-07 20:26:40 -07001284 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 -07001285 goto done;
1286
1287 dev_queue = &dev->rx_queue;
David S. Miller827ebd62008-08-07 20:26:40 -07001288 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 -07001289 goto done;
1290
Pavel Emelianov7562f872007-05-03 15:13:45 -07001291cont:
1292 idx++;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001293 }
1294
1295done:
1296 read_unlock(&dev_base_lock);
1297
1298 cb->args[0] = idx;
1299 cb->args[1] = q_idx;
1300
1301 return skb->len;
1302}
1303
1304
1305
1306/************************************************
1307 * Traffic classes manipulation. *
1308 ************************************************/
1309
1310
1311
1312static int tc_ctl_tclass(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
1313{
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001314 struct net *net = sock_net(skb->sk);
David S. Millerb0e1e642008-07-08 17:42:10 -07001315 struct netdev_queue *dev_queue;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001316 struct tcmsg *tcm = NLMSG_DATA(n);
Patrick McHardy1e904742008-01-22 22:11:17 -08001317 struct nlattr *tca[TCA_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -07001318 struct net_device *dev;
1319 struct Qdisc *q = NULL;
Eric Dumazet20fea082007-11-14 01:44:41 -08001320 const struct Qdisc_class_ops *cops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001321 unsigned long cl = 0;
1322 unsigned long new_cl;
1323 u32 pid = tcm->tcm_parent;
1324 u32 clid = tcm->tcm_handle;
1325 u32 qid = TC_H_MAJ(clid);
1326 int err;
1327
Denis V. Lunevb8542722007-12-01 00:21:31 +11001328 if (net != &init_net)
1329 return -EINVAL;
1330
Eric W. Biederman881d9662007-09-17 11:56:21 -07001331 if ((dev = __dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001332 return -ENODEV;
1333
Patrick McHardy1e904742008-01-22 22:11:17 -08001334 err = nlmsg_parse(n, sizeof(*tcm), tca, TCA_MAX, NULL);
1335 if (err < 0)
1336 return err;
1337
Linus Torvalds1da177e2005-04-16 15:20:36 -07001338 /*
1339 parent == TC_H_UNSPEC - unspecified parent.
1340 parent == TC_H_ROOT - class is root, which has no parent.
1341 parent == X:0 - parent is root class.
1342 parent == X:Y - parent is a node in hierarchy.
1343 parent == 0:Y - parent is X:Y, where X:0 is qdisc.
1344
1345 handle == 0:0 - generate handle from kernel pool.
1346 handle == 0:Y - class is X:Y, where X:0 is qdisc.
1347 handle == X:Y - clear.
1348 handle == X:0 - root class.
1349 */
1350
1351 /* Step 1. Determine qdisc handle X:0 */
1352
David S. Millere8a04642008-07-17 00:34:19 -07001353 dev_queue = netdev_get_tx_queue(dev, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001354 if (pid != TC_H_ROOT) {
1355 u32 qid1 = TC_H_MAJ(pid);
1356
1357 if (qid && qid1) {
1358 /* If both majors are known, they must be identical. */
1359 if (qid != qid1)
1360 return -EINVAL;
1361 } else if (qid1) {
1362 qid = qid1;
1363 } else if (qid == 0)
David S. Millerb0e1e642008-07-08 17:42:10 -07001364 qid = dev_queue->qdisc_sleeping->handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001365
1366 /* Now qid is genuine qdisc handle consistent
1367 both with parent and child.
1368
1369 TC_H_MAJ(pid) still may be unspecified, complete it now.
1370 */
1371 if (pid)
1372 pid = TC_H_MAKE(qid, pid);
1373 } else {
1374 if (qid == 0)
David S. Millerb0e1e642008-07-08 17:42:10 -07001375 qid = dev_queue->qdisc_sleeping->handle;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001376 }
1377
1378 /* OK. Locate qdisc */
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001379 if ((q = qdisc_lookup(dev, qid)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001380 return -ENOENT;
1381
1382 /* An check that it supports classes */
1383 cops = q->ops->cl_ops;
1384 if (cops == NULL)
1385 return -EINVAL;
1386
1387 /* Now try to get class */
1388 if (clid == 0) {
1389 if (pid == TC_H_ROOT)
1390 clid = qid;
1391 } else
1392 clid = TC_H_MAKE(qid, clid);
1393
1394 if (clid)
1395 cl = cops->get(q, clid);
1396
1397 if (cl == 0) {
1398 err = -ENOENT;
1399 if (n->nlmsg_type != RTM_NEWTCLASS || !(n->nlmsg_flags&NLM_F_CREATE))
1400 goto out;
1401 } else {
1402 switch (n->nlmsg_type) {
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001403 case RTM_NEWTCLASS:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001404 err = -EEXIST;
1405 if (n->nlmsg_flags&NLM_F_EXCL)
1406 goto out;
1407 break;
1408 case RTM_DELTCLASS:
1409 err = cops->delete(q, cl);
1410 if (err == 0)
1411 tclass_notify(skb, n, q, cl, RTM_DELTCLASS);
1412 goto out;
1413 case RTM_GETTCLASS:
1414 err = tclass_notify(skb, n, q, cl, RTM_NEWTCLASS);
1415 goto out;
1416 default:
1417 err = -EINVAL;
1418 goto out;
1419 }
1420 }
1421
1422 new_cl = cl;
1423 err = cops->change(q, clid, pid, tca, &new_cl);
1424 if (err == 0)
1425 tclass_notify(skb, n, q, new_cl, RTM_NEWTCLASS);
1426
1427out:
1428 if (cl)
1429 cops->put(q, cl);
1430
1431 return err;
1432}
1433
1434
1435static int tc_fill_tclass(struct sk_buff *skb, struct Qdisc *q,
1436 unsigned long cl,
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001437 u32 pid, u32 seq, u16 flags, int event)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001438{
1439 struct tcmsg *tcm;
1440 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001441 unsigned char *b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001442 struct gnet_dump d;
Eric Dumazet20fea082007-11-14 01:44:41 -08001443 const struct Qdisc_class_ops *cl_ops = q->ops->cl_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001444
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -07001445 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*tcm), flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001446 tcm = NLMSG_DATA(nlh);
1447 tcm->tcm_family = AF_UNSPEC;
David S. Miller5ce2d482008-07-08 17:06:30 -07001448 tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001449 tcm->tcm_parent = q->handle;
1450 tcm->tcm_handle = q->handle;
1451 tcm->tcm_info = 0;
Patrick McHardy57e1c482008-01-23 20:34:28 -08001452 NLA_PUT_STRING(skb, TCA_KIND, q->ops->id);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001453 if (cl_ops->dump && cl_ops->dump(q, cl, skb, tcm) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001454 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001455
Jarek Poplawski102396a2008-08-29 14:21:52 -07001456 if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
1457 qdisc_root_sleeping_lock(q), &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001458 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001459
1460 if (cl_ops->dump_stats && cl_ops->dump_stats(q, cl, &d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001461 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001462
1463 if (gnet_stats_finish_copy(&d) < 0)
Patrick McHardy1e904742008-01-22 22:11:17 -08001464 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001465
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001466 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001467 return skb->len;
1468
1469nlmsg_failure:
Patrick McHardy1e904742008-01-22 22:11:17 -08001470nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -07001471 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001472 return -1;
1473}
1474
1475static int tclass_notify(struct sk_buff *oskb, struct nlmsghdr *n,
1476 struct Qdisc *q, unsigned long cl, int event)
1477{
1478 struct sk_buff *skb;
1479 u32 pid = oskb ? NETLINK_CB(oskb).pid : 0;
1480
1481 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1482 if (!skb)
1483 return -ENOBUFS;
1484
1485 if (tc_fill_tclass(skb, q, cl, pid, n->nlmsg_seq, 0, event) < 0) {
1486 kfree_skb(skb);
1487 return -EINVAL;
1488 }
1489
Denis V. Lunev97c53ca2007-11-19 22:26:51 -08001490 return rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001491}
1492
1493struct qdisc_dump_args
1494{
1495 struct qdisc_walker w;
1496 struct sk_buff *skb;
1497 struct netlink_callback *cb;
1498};
1499
1500static int qdisc_class_dump(struct Qdisc *q, unsigned long cl, struct qdisc_walker *arg)
1501{
1502 struct qdisc_dump_args *a = (struct qdisc_dump_args *)arg;
1503
1504 return tc_fill_tclass(a->skb, q, cl, NETLINK_CB(a->cb->skb).pid,
1505 a->cb->nlh->nlmsg_seq, NLM_F_MULTI, RTM_NEWTCLASS);
1506}
1507
David S. Miller30723672008-07-18 22:50:15 -07001508static int tc_dump_tclass_qdisc(struct Qdisc *q, struct sk_buff *skb,
1509 struct tcmsg *tcm, struct netlink_callback *cb,
1510 int *t_p, int s_t)
1511{
1512 struct qdisc_dump_args arg;
1513
1514 if (tc_qdisc_dump_ignore(q) ||
1515 *t_p < s_t || !q->ops->cl_ops ||
1516 (tcm->tcm_parent &&
1517 TC_H_MAJ(tcm->tcm_parent) != q->handle)) {
1518 (*t_p)++;
1519 return 0;
1520 }
1521 if (*t_p > s_t)
1522 memset(&cb->args[1], 0, sizeof(cb->args)-sizeof(cb->args[0]));
1523 arg.w.fn = qdisc_class_dump;
1524 arg.skb = skb;
1525 arg.cb = cb;
1526 arg.w.stop = 0;
1527 arg.w.skip = cb->args[1];
1528 arg.w.count = 0;
1529 q->ops->cl_ops->walk(q, &arg.w);
1530 cb->args[1] = arg.w.count;
1531 if (arg.w.stop)
1532 return -1;
1533 (*t_p)++;
1534 return 0;
1535}
1536
1537static int tc_dump_tclass_root(struct Qdisc *root, struct sk_buff *skb,
1538 struct tcmsg *tcm, struct netlink_callback *cb,
1539 int *t_p, int s_t)
1540{
1541 struct Qdisc *q;
1542
1543 if (!root)
1544 return 0;
1545
1546 if (tc_dump_tclass_qdisc(root, skb, tcm, cb, t_p, s_t) < 0)
1547 return -1;
1548
1549 list_for_each_entry(q, &root->list, list) {
1550 if (tc_dump_tclass_qdisc(q, skb, tcm, cb, t_p, s_t) < 0)
1551 return -1;
1552 }
1553
1554 return 0;
1555}
1556
Linus Torvalds1da177e2005-04-16 15:20:36 -07001557static int tc_dump_tclass(struct sk_buff *skb, struct netlink_callback *cb)
1558{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001559 struct tcmsg *tcm = (struct tcmsg*)NLMSG_DATA(cb->nlh);
David S. Miller30723672008-07-18 22:50:15 -07001560 struct net *net = sock_net(skb->sk);
1561 struct netdev_queue *dev_queue;
1562 struct net_device *dev;
1563 int t, s_t;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001564
Denis V. Lunevb8542722007-12-01 00:21:31 +11001565 if (net != &init_net)
1566 return 0;
1567
Linus Torvalds1da177e2005-04-16 15:20:36 -07001568 if (cb->nlh->nlmsg_len < NLMSG_LENGTH(sizeof(*tcm)))
1569 return 0;
Eric W. Biederman881d9662007-09-17 11:56:21 -07001570 if ((dev = dev_get_by_index(&init_net, tcm->tcm_ifindex)) == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001571 return 0;
1572
1573 s_t = cb->args[0];
1574 t = 0;
1575
David S. Miller30723672008-07-18 22:50:15 -07001576 dev_queue = netdev_get_tx_queue(dev, 0);
David S. Miller8123b422008-08-08 23:23:39 -07001577 if (tc_dump_tclass_root(dev_queue->qdisc_sleeping, skb, tcm, cb, &t, s_t) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001578 goto done;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001579
David S. Miller30723672008-07-18 22:50:15 -07001580 dev_queue = &dev->rx_queue;
David S. Miller8123b422008-08-08 23:23:39 -07001581 if (tc_dump_tclass_root(dev_queue->qdisc_sleeping, skb, tcm, cb, &t, s_t) < 0)
David S. Miller30723672008-07-18 22:50:15 -07001582 goto done;
1583
1584done:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001585 cb->args[0] = t;
1586
1587 dev_put(dev);
1588 return skb->len;
1589}
1590
1591/* Main classifier routine: scans classifier chain attached
1592 to this qdisc, (optionally) tests for protocol and asks
1593 specific classifiers.
1594 */
Patrick McHardy73ca4912007-07-15 00:02:31 -07001595int tc_classify_compat(struct sk_buff *skb, struct tcf_proto *tp,
1596 struct tcf_result *res)
1597{
1598 __be16 protocol = skb->protocol;
1599 int err = 0;
1600
1601 for (; tp; tp = tp->next) {
1602 if ((tp->protocol == protocol ||
1603 tp->protocol == htons(ETH_P_ALL)) &&
1604 (err = tp->classify(skb, tp, res)) >= 0) {
1605#ifdef CONFIG_NET_CLS_ACT
1606 if (err != TC_ACT_RECLASSIFY && skb->tc_verd)
1607 skb->tc_verd = SET_TC_VERD(skb->tc_verd, 0);
1608#endif
1609 return err;
1610 }
1611 }
1612 return -1;
1613}
1614EXPORT_SYMBOL(tc_classify_compat);
1615
Linus Torvalds1da177e2005-04-16 15:20:36 -07001616int tc_classify(struct sk_buff *skb, struct tcf_proto *tp,
Patrick McHardy73ca4912007-07-15 00:02:31 -07001617 struct tcf_result *res)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001618{
1619 int err = 0;
Patrick McHardy73ca4912007-07-15 00:02:31 -07001620 __be16 protocol;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001621#ifdef CONFIG_NET_CLS_ACT
1622 struct tcf_proto *otp = tp;
1623reclassify:
1624#endif
1625 protocol = skb->protocol;
1626
Patrick McHardy73ca4912007-07-15 00:02:31 -07001627 err = tc_classify_compat(skb, tp, res);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001628#ifdef CONFIG_NET_CLS_ACT
Patrick McHardy73ca4912007-07-15 00:02:31 -07001629 if (err == TC_ACT_RECLASSIFY) {
1630 u32 verd = G_TC_VERD(skb->tc_verd);
1631 tp = otp;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001632
Patrick McHardy73ca4912007-07-15 00:02:31 -07001633 if (verd++ >= MAX_REC_LOOP) {
1634 printk("rule prio %u protocol %02x reclassify loop, "
1635 "packet dropped\n",
1636 tp->prio&0xffff, ntohs(tp->protocol));
1637 return TC_ACT_SHOT;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001638 }
Patrick McHardy73ca4912007-07-15 00:02:31 -07001639 skb->tc_verd = SET_TC_VERD(skb->tc_verd, verd);
1640 goto reclassify;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001641 }
Patrick McHardy73ca4912007-07-15 00:02:31 -07001642#endif
1643 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001644}
Patrick McHardy73ca4912007-07-15 00:02:31 -07001645EXPORT_SYMBOL(tc_classify);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001646
Patrick McHardya48b5a62007-03-23 11:29:43 -07001647void tcf_destroy(struct tcf_proto *tp)
1648{
1649 tp->ops->destroy(tp);
1650 module_put(tp->ops->owner);
1651 kfree(tp);
1652}
1653
Patrick McHardyff31ab52008-07-01 19:52:38 -07001654void tcf_destroy_chain(struct tcf_proto **fl)
Patrick McHardya48b5a62007-03-23 11:29:43 -07001655{
1656 struct tcf_proto *tp;
1657
Patrick McHardyff31ab52008-07-01 19:52:38 -07001658 while ((tp = *fl) != NULL) {
1659 *fl = tp->next;
Patrick McHardya48b5a62007-03-23 11:29:43 -07001660 tcf_destroy(tp);
1661 }
1662}
1663EXPORT_SYMBOL(tcf_destroy_chain);
1664
Linus Torvalds1da177e2005-04-16 15:20:36 -07001665#ifdef CONFIG_PROC_FS
1666static int psched_show(struct seq_file *seq, void *v)
1667{
Patrick McHardy3c0cfc12007-10-10 16:32:41 -07001668 struct timespec ts;
1669
1670 hrtimer_get_res(CLOCK_MONOTONIC, &ts);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001671 seq_printf(seq, "%08x %08x %08x %08x\n",
Patrick McHardy641b9e02007-03-16 01:18:42 -07001672 (u32)NSEC_PER_USEC, (u32)PSCHED_US2NS(1),
Patrick McHardy514bca32007-03-16 12:34:52 -07001673 1000000,
Patrick McHardy3c0cfc12007-10-10 16:32:41 -07001674 (u32)NSEC_PER_SEC/(u32)ktime_to_ns(timespec_to_ktime(ts)));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001675
1676 return 0;
1677}
1678
1679static int psched_open(struct inode *inode, struct file *file)
1680{
1681 return single_open(file, psched_show, PDE(inode)->data);
1682}
1683
Arjan van de Venda7071d2007-02-12 00:55:36 -08001684static const struct file_operations psched_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001685 .owner = THIS_MODULE,
1686 .open = psched_open,
1687 .read = seq_read,
1688 .llseek = seq_lseek,
1689 .release = single_release,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001690};
Linus Torvalds1da177e2005-04-16 15:20:36 -07001691#endif
1692
Linus Torvalds1da177e2005-04-16 15:20:36 -07001693static int __init pktsched_init(void)
1694{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001695 register_qdisc(&pfifo_qdisc_ops);
1696 register_qdisc(&bfifo_qdisc_ops);
Eric W. Biederman457c4cb2007-09-12 12:01:34 +02001697 proc_net_fops_create(&init_net, "psched", 0, &psched_fops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001698
Thomas Grafbe577dd2007-03-22 11:55:50 -07001699 rtnl_register(PF_UNSPEC, RTM_NEWQDISC, tc_modify_qdisc, NULL);
1700 rtnl_register(PF_UNSPEC, RTM_DELQDISC, tc_get_qdisc, NULL);
1701 rtnl_register(PF_UNSPEC, RTM_GETQDISC, tc_get_qdisc, tc_dump_qdisc);
1702 rtnl_register(PF_UNSPEC, RTM_NEWTCLASS, tc_ctl_tclass, NULL);
1703 rtnl_register(PF_UNSPEC, RTM_DELTCLASS, tc_ctl_tclass, NULL);
1704 rtnl_register(PF_UNSPEC, RTM_GETTCLASS, tc_ctl_tclass, tc_dump_tclass);
1705
Linus Torvalds1da177e2005-04-16 15:20:36 -07001706 return 0;
1707}
1708
1709subsys_initcall(pktsched_init);