• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * net/sched/sch_api.c	Packet scheduler API.
4  *
5  * Authors:	Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
6  *
7  * Fixes:
8  *
9  * Rani Assaf <rani@magic.metawire.com> :980802: JIFFIES and CPU clock sources are repaired.
10  * Eduardo J. Blanco <ejbs@netlabs.com.uy> :990222: kmod support
11  * Jamal Hadi Salim <hadi@nortelnetworks.com>: 990601: ingress support
12  */
13 
14 #include <linux/module.h>
15 #include <linux/types.h>
16 #include <linux/kernel.h>
17 #include <linux/string.h>
18 #include <linux/errno.h>
19 #include <linux/skbuff.h>
20 #include <linux/init.h>
21 #include <linux/proc_fs.h>
22 #include <linux/seq_file.h>
23 #include <linux/kmod.h>
24 #include <linux/list.h>
25 #include <linux/hrtimer.h>
26 #include <linux/slab.h>
27 #include <linux/hashtable.h>
28 
29 #include <net/net_namespace.h>
30 #include <net/sock.h>
31 #include <net/netlink.h>
32 #include <net/pkt_sched.h>
33 #include <net/pkt_cls.h>
34 
35 #include <trace/events/qdisc.h>
36 
37 /*
38 
39    Short review.
40    -------------
41 
42    This file consists of two interrelated parts:
43 
44    1. queueing disciplines manager frontend.
45    2. traffic classes manager frontend.
46 
47    Generally, queueing discipline ("qdisc") is a black box,
48    which is able to enqueue packets and to dequeue them (when
49    device is ready to send something) in order and at times
50    determined by algorithm hidden in it.
51 
52    qdisc's are divided to two categories:
53    - "queues", which have no internal structure visible from outside.
54    - "schedulers", which split all the packets to "traffic classes",
55      using "packet classifiers" (look at cls_api.c)
56 
57    In turn, classes may have child qdiscs (as rule, queues)
58    attached to them etc. etc. etc.
59 
60    The goal of the routines in this file is to translate
61    information supplied by user in the form of handles
62    to more intelligible for kernel form, to make some sanity
63    checks and part of work, which is common to all qdiscs
64    and to provide rtnetlink notifications.
65 
66    All real intelligent work is done inside qdisc modules.
67 
68 
69 
70    Every discipline has two major routines: enqueue and dequeue.
71 
72    ---dequeue
73 
74    dequeue usually returns a skb to send. It is allowed to return NULL,
75    but it does not mean that queue is empty, it just means that
76    discipline does not want to send anything this time.
77    Queue is really empty if q->q.qlen == 0.
78    For complicated disciplines with multiple queues q->q is not
79    real packet queue, but however q->q.qlen must be valid.
80 
81    ---enqueue
82 
83    enqueue returns 0, if packet was enqueued successfully.
84    If packet (this one or another one) was dropped, it returns
85    not zero error code.
86    NET_XMIT_DROP 	- this packet dropped
87      Expected action: do not backoff, but wait until queue will clear.
88    NET_XMIT_CN	 	- probably this packet enqueued, but another one dropped.
89      Expected action: backoff or ignore
90 
91    Auxiliary routines:
92 
93    ---peek
94 
95    like dequeue but without removing a packet from the queue
96 
97    ---reset
98 
99    returns qdisc to initial state: purge all buffers, clear all
100    timers, counters (except for statistics) etc.
101 
102    ---init
103 
104    initializes newly created qdisc.
105 
106    ---destroy
107 
108    destroys resources allocated by init and during lifetime of qdisc.
109 
110    ---change
111 
112    changes qdisc parameters.
113  */
114 
115 /* Protects list of registered TC modules. It is pure SMP lock. */
116 static DEFINE_RWLOCK(qdisc_mod_lock);
117 
118 
119 /************************************************
120  *	Queueing disciplines manipulation.	*
121  ************************************************/
122 
123 
124 /* The list of all installed queueing disciplines. */
125 
126 static struct Qdisc_ops *qdisc_base;
127 
128 /* Register/unregister queueing discipline */
129 
register_qdisc(struct Qdisc_ops * qops)130 int register_qdisc(struct Qdisc_ops *qops)
131 {
132 	struct Qdisc_ops *q, **qp;
133 	int rc = -EEXIST;
134 
135 	write_lock(&qdisc_mod_lock);
136 	for (qp = &qdisc_base; (q = *qp) != NULL; qp = &q->next)
137 		if (!strcmp(qops->id, q->id))
138 			goto out;
139 
140 	if (qops->enqueue == NULL)
141 		qops->enqueue = noop_qdisc_ops.enqueue;
142 	if (qops->peek == NULL) {
143 		if (qops->dequeue == NULL)
144 			qops->peek = noop_qdisc_ops.peek;
145 		else
146 			goto out_einval;
147 	}
148 	if (qops->dequeue == NULL)
149 		qops->dequeue = noop_qdisc_ops.dequeue;
150 
151 	if (qops->cl_ops) {
152 		const struct Qdisc_class_ops *cops = qops->cl_ops;
153 
154 		if (!(cops->find && cops->walk && cops->leaf))
155 			goto out_einval;
156 
157 		if (cops->tcf_block && !(cops->bind_tcf && cops->unbind_tcf))
158 			goto out_einval;
159 	}
160 
161 	qops->next = NULL;
162 	*qp = qops;
163 	rc = 0;
164 out:
165 	write_unlock(&qdisc_mod_lock);
166 	return rc;
167 
168 out_einval:
169 	rc = -EINVAL;
170 	goto out;
171 }
172 EXPORT_SYMBOL(register_qdisc);
173 
unregister_qdisc(struct Qdisc_ops * qops)174 int unregister_qdisc(struct Qdisc_ops *qops)
175 {
176 	struct Qdisc_ops *q, **qp;
177 	int err = -ENOENT;
178 
179 	write_lock(&qdisc_mod_lock);
180 	for (qp = &qdisc_base; (q = *qp) != NULL; qp = &q->next)
181 		if (q == qops)
182 			break;
183 	if (q) {
184 		*qp = q->next;
185 		q->next = NULL;
186 		err = 0;
187 	}
188 	write_unlock(&qdisc_mod_lock);
189 	return err;
190 }
191 EXPORT_SYMBOL(unregister_qdisc);
192 
193 /* Get default qdisc if not otherwise specified */
qdisc_get_default(char * name,size_t len)194 void qdisc_get_default(char *name, size_t len)
195 {
196 	read_lock(&qdisc_mod_lock);
197 	strlcpy(name, default_qdisc_ops->id, len);
198 	read_unlock(&qdisc_mod_lock);
199 }
200 
qdisc_lookup_default(const char * name)201 static struct Qdisc_ops *qdisc_lookup_default(const char *name)
202 {
203 	struct Qdisc_ops *q = NULL;
204 
205 	for (q = qdisc_base; q; q = q->next) {
206 		if (!strcmp(name, q->id)) {
207 			if (!try_module_get(q->owner))
208 				q = NULL;
209 			break;
210 		}
211 	}
212 
213 	return q;
214 }
215 
216 /* Set new default qdisc to use */
qdisc_set_default(const char * name)217 int qdisc_set_default(const char *name)
218 {
219 	const struct Qdisc_ops *ops;
220 
221 	if (!capable(CAP_NET_ADMIN))
222 		return -EPERM;
223 
224 	write_lock(&qdisc_mod_lock);
225 	ops = qdisc_lookup_default(name);
226 	if (!ops) {
227 		/* Not found, drop lock and try to load module */
228 		write_unlock(&qdisc_mod_lock);
229 		request_module("sch_%s", name);
230 		write_lock(&qdisc_mod_lock);
231 
232 		ops = qdisc_lookup_default(name);
233 	}
234 
235 	if (ops) {
236 		/* Set new default */
237 		module_put(default_qdisc_ops->owner);
238 		default_qdisc_ops = ops;
239 	}
240 	write_unlock(&qdisc_mod_lock);
241 
242 	return ops ? 0 : -ENOENT;
243 }
244 
245 #ifdef CONFIG_NET_SCH_DEFAULT
246 /* Set default value from kernel config */
sch_default_qdisc(void)247 static int __init sch_default_qdisc(void)
248 {
249 	return qdisc_set_default(CONFIG_DEFAULT_NET_SCH);
250 }
251 late_initcall(sch_default_qdisc);
252 #endif
253 
254 /* We know handle. Find qdisc among all qdisc's attached to device
255  * (root qdisc, all its children, children of children etc.)
256  * Note: caller either uses rtnl or rcu_read_lock()
257  */
258 
qdisc_match_from_root(struct Qdisc * root,u32 handle)259 static struct Qdisc *qdisc_match_from_root(struct Qdisc *root, u32 handle)
260 {
261 	struct Qdisc *q;
262 
263 	if (!qdisc_dev(root))
264 		return (root->handle == handle ? root : NULL);
265 
266 	if (!(root->flags & TCQ_F_BUILTIN) &&
267 	    root->handle == handle)
268 		return root;
269 
270 	hash_for_each_possible_rcu(qdisc_dev(root)->qdisc_hash, q, hash, handle,
271 				   lockdep_rtnl_is_held()) {
272 		if (q->handle == handle)
273 			return q;
274 	}
275 	return NULL;
276 }
277 
qdisc_hash_add(struct Qdisc * q,bool invisible)278 void qdisc_hash_add(struct Qdisc *q, bool invisible)
279 {
280 	if ((q->parent != TC_H_ROOT) && !(q->flags & TCQ_F_INGRESS)) {
281 		ASSERT_RTNL();
282 		hash_add_rcu(qdisc_dev(q)->qdisc_hash, &q->hash, q->handle);
283 		if (invisible)
284 			q->flags |= TCQ_F_INVISIBLE;
285 	}
286 }
287 EXPORT_SYMBOL(qdisc_hash_add);
288 
qdisc_hash_del(struct Qdisc * q)289 void qdisc_hash_del(struct Qdisc *q)
290 {
291 	if ((q->parent != TC_H_ROOT) && !(q->flags & TCQ_F_INGRESS)) {
292 		ASSERT_RTNL();
293 		hash_del_rcu(&q->hash);
294 	}
295 }
296 EXPORT_SYMBOL(qdisc_hash_del);
297 
qdisc_lookup(struct net_device * dev,u32 handle)298 struct Qdisc *qdisc_lookup(struct net_device *dev, u32 handle)
299 {
300 	struct Qdisc *q;
301 
302 	if (!handle)
303 		return NULL;
304 	q = qdisc_match_from_root(rtnl_dereference(dev->qdisc), handle);
305 	if (q)
306 		goto out;
307 
308 	if (dev_ingress_queue(dev))
309 		q = qdisc_match_from_root(
310 			dev_ingress_queue(dev)->qdisc_sleeping,
311 			handle);
312 out:
313 	return q;
314 }
315 
qdisc_lookup_rcu(struct net_device * dev,u32 handle)316 struct Qdisc *qdisc_lookup_rcu(struct net_device *dev, u32 handle)
317 {
318 	struct netdev_queue *nq;
319 	struct Qdisc *q;
320 
321 	if (!handle)
322 		return NULL;
323 	q = qdisc_match_from_root(rcu_dereference(dev->qdisc), handle);
324 	if (q)
325 		goto out;
326 
327 	nq = dev_ingress_queue_rcu(dev);
328 	if (nq)
329 		q = qdisc_match_from_root(nq->qdisc_sleeping, handle);
330 out:
331 	return q;
332 }
333 
qdisc_leaf(struct Qdisc * p,u32 classid)334 static struct Qdisc *qdisc_leaf(struct Qdisc *p, u32 classid)
335 {
336 	unsigned long cl;
337 	const struct Qdisc_class_ops *cops = p->ops->cl_ops;
338 
339 	if (cops == NULL)
340 		return NULL;
341 	cl = cops->find(p, classid);
342 
343 	if (cl == 0)
344 		return NULL;
345 	return cops->leaf(p, cl);
346 }
347 
348 /* Find queueing discipline by name */
349 
qdisc_lookup_ops(struct nlattr * kind)350 static struct Qdisc_ops *qdisc_lookup_ops(struct nlattr *kind)
351 {
352 	struct Qdisc_ops *q = NULL;
353 
354 	if (kind) {
355 		read_lock(&qdisc_mod_lock);
356 		for (q = qdisc_base; q; q = q->next) {
357 			if (nla_strcmp(kind, q->id) == 0) {
358 				if (!try_module_get(q->owner))
359 					q = NULL;
360 				break;
361 			}
362 		}
363 		read_unlock(&qdisc_mod_lock);
364 	}
365 	return q;
366 }
367 
368 /* The linklayer setting were not transferred from iproute2, in older
369  * versions, and the rate tables lookup systems have been dropped in
370  * the kernel. To keep backward compatible with older iproute2 tc
371  * utils, we detect the linklayer setting by detecting if the rate
372  * table were modified.
373  *
374  * For linklayer ATM table entries, the rate table will be aligned to
375  * 48 bytes, thus some table entries will contain the same value.  The
376  * mpu (min packet unit) is also encoded into the old rate table, thus
377  * starting from the mpu, we find low and high table entries for
378  * mapping this cell.  If these entries contain the same value, when
379  * the rate tables have been modified for linklayer ATM.
380  *
381  * This is done by rounding mpu to the nearest 48 bytes cell/entry,
382  * and then roundup to the next cell, calc the table entry one below,
383  * and compare.
384  */
__detect_linklayer(struct tc_ratespec * r,__u32 * rtab)385 static __u8 __detect_linklayer(struct tc_ratespec *r, __u32 *rtab)
386 {
387 	int low       = roundup(r->mpu, 48);
388 	int high      = roundup(low+1, 48);
389 	int cell_low  = low >> r->cell_log;
390 	int cell_high = (high >> r->cell_log) - 1;
391 
392 	/* rtab is too inaccurate at rates > 100Mbit/s */
393 	if ((r->rate > (100000000/8)) || (rtab[0] == 0)) {
394 		pr_debug("TC linklayer: Giving up ATM detection\n");
395 		return TC_LINKLAYER_ETHERNET;
396 	}
397 
398 	if ((cell_high > cell_low) && (cell_high < 256)
399 	    && (rtab[cell_low] == rtab[cell_high])) {
400 		pr_debug("TC linklayer: Detected ATM, low(%d)=high(%d)=%u\n",
401 			 cell_low, cell_high, rtab[cell_high]);
402 		return TC_LINKLAYER_ATM;
403 	}
404 	return TC_LINKLAYER_ETHERNET;
405 }
406 
407 static struct qdisc_rate_table *qdisc_rtab_list;
408 
qdisc_get_rtab(struct tc_ratespec * r,struct nlattr * tab,struct netlink_ext_ack * extack)409 struct qdisc_rate_table *qdisc_get_rtab(struct tc_ratespec *r,
410 					struct nlattr *tab,
411 					struct netlink_ext_ack *extack)
412 {
413 	struct qdisc_rate_table *rtab;
414 
415 	if (tab == NULL || r->rate == 0 ||
416 	    r->cell_log == 0 || r->cell_log >= 32 ||
417 	    nla_len(tab) != TC_RTAB_SIZE) {
418 		NL_SET_ERR_MSG(extack, "Invalid rate table parameters for searching");
419 		return NULL;
420 	}
421 
422 	for (rtab = qdisc_rtab_list; rtab; rtab = rtab->next) {
423 		if (!memcmp(&rtab->rate, r, sizeof(struct tc_ratespec)) &&
424 		    !memcmp(&rtab->data, nla_data(tab), 1024)) {
425 			rtab->refcnt++;
426 			return rtab;
427 		}
428 	}
429 
430 	rtab = kmalloc(sizeof(*rtab), GFP_KERNEL);
431 	if (rtab) {
432 		rtab->rate = *r;
433 		rtab->refcnt = 1;
434 		memcpy(rtab->data, nla_data(tab), 1024);
435 		if (r->linklayer == TC_LINKLAYER_UNAWARE)
436 			r->linklayer = __detect_linklayer(r, rtab->data);
437 		rtab->next = qdisc_rtab_list;
438 		qdisc_rtab_list = rtab;
439 	} else {
440 		NL_SET_ERR_MSG(extack, "Failed to allocate new qdisc rate table");
441 	}
442 	return rtab;
443 }
444 EXPORT_SYMBOL(qdisc_get_rtab);
445 
qdisc_put_rtab(struct qdisc_rate_table * tab)446 void qdisc_put_rtab(struct qdisc_rate_table *tab)
447 {
448 	struct qdisc_rate_table *rtab, **rtabp;
449 
450 	if (!tab || --tab->refcnt)
451 		return;
452 
453 	for (rtabp = &qdisc_rtab_list;
454 	     (rtab = *rtabp) != NULL;
455 	     rtabp = &rtab->next) {
456 		if (rtab == tab) {
457 			*rtabp = rtab->next;
458 			kfree(rtab);
459 			return;
460 		}
461 	}
462 }
463 EXPORT_SYMBOL(qdisc_put_rtab);
464 
465 static LIST_HEAD(qdisc_stab_list);
466 
467 static const struct nla_policy stab_policy[TCA_STAB_MAX + 1] = {
468 	[TCA_STAB_BASE]	= { .len = sizeof(struct tc_sizespec) },
469 	[TCA_STAB_DATA] = { .type = NLA_BINARY },
470 };
471 
qdisc_get_stab(struct nlattr * opt,struct netlink_ext_ack * extack)472 static struct qdisc_size_table *qdisc_get_stab(struct nlattr *opt,
473 					       struct netlink_ext_ack *extack)
474 {
475 	struct nlattr *tb[TCA_STAB_MAX + 1];
476 	struct qdisc_size_table *stab;
477 	struct tc_sizespec *s;
478 	unsigned int tsize = 0;
479 	u16 *tab = NULL;
480 	int err;
481 
482 	err = nla_parse_nested_deprecated(tb, TCA_STAB_MAX, opt, stab_policy,
483 					  extack);
484 	if (err < 0)
485 		return ERR_PTR(err);
486 	if (!tb[TCA_STAB_BASE]) {
487 		NL_SET_ERR_MSG(extack, "Size table base attribute is missing");
488 		return ERR_PTR(-EINVAL);
489 	}
490 
491 	s = nla_data(tb[TCA_STAB_BASE]);
492 
493 	if (s->tsize > 0) {
494 		if (!tb[TCA_STAB_DATA]) {
495 			NL_SET_ERR_MSG(extack, "Size table data attribute is missing");
496 			return ERR_PTR(-EINVAL);
497 		}
498 		tab = nla_data(tb[TCA_STAB_DATA]);
499 		tsize = nla_len(tb[TCA_STAB_DATA]) / sizeof(u16);
500 	}
501 
502 	if (tsize != s->tsize || (!tab && tsize > 0)) {
503 		NL_SET_ERR_MSG(extack, "Invalid size of size table");
504 		return ERR_PTR(-EINVAL);
505 	}
506 
507 	list_for_each_entry(stab, &qdisc_stab_list, list) {
508 		if (memcmp(&stab->szopts, s, sizeof(*s)))
509 			continue;
510 		if (tsize > 0 && memcmp(stab->data, tab, tsize * sizeof(u16)))
511 			continue;
512 		stab->refcnt++;
513 		return stab;
514 	}
515 
516 	if (s->size_log > STAB_SIZE_LOG_MAX ||
517 	    s->cell_log > STAB_SIZE_LOG_MAX) {
518 		NL_SET_ERR_MSG(extack, "Invalid logarithmic size of size table");
519 		return ERR_PTR(-EINVAL);
520 	}
521 
522 	stab = kmalloc(sizeof(*stab) + tsize * sizeof(u16), GFP_KERNEL);
523 	if (!stab)
524 		return ERR_PTR(-ENOMEM);
525 
526 	stab->refcnt = 1;
527 	stab->szopts = *s;
528 	if (tsize > 0)
529 		memcpy(stab->data, tab, tsize * sizeof(u16));
530 
531 	list_add_tail(&stab->list, &qdisc_stab_list);
532 
533 	return stab;
534 }
535 
qdisc_put_stab(struct qdisc_size_table * tab)536 void qdisc_put_stab(struct qdisc_size_table *tab)
537 {
538 	if (!tab)
539 		return;
540 
541 	if (--tab->refcnt == 0) {
542 		list_del(&tab->list);
543 		kfree_rcu(tab, rcu);
544 	}
545 }
546 EXPORT_SYMBOL(qdisc_put_stab);
547 
qdisc_dump_stab(struct sk_buff * skb,struct qdisc_size_table * stab)548 static int qdisc_dump_stab(struct sk_buff *skb, struct qdisc_size_table *stab)
549 {
550 	struct nlattr *nest;
551 
552 	nest = nla_nest_start_noflag(skb, TCA_STAB);
553 	if (nest == NULL)
554 		goto nla_put_failure;
555 	if (nla_put(skb, TCA_STAB_BASE, sizeof(stab->szopts), &stab->szopts))
556 		goto nla_put_failure;
557 	nla_nest_end(skb, nest);
558 
559 	return skb->len;
560 
561 nla_put_failure:
562 	return -1;
563 }
564 
__qdisc_calculate_pkt_len(struct sk_buff * skb,const struct qdisc_size_table * stab)565 void __qdisc_calculate_pkt_len(struct sk_buff *skb,
566 			       const struct qdisc_size_table *stab)
567 {
568 	int pkt_len, slot;
569 
570 	pkt_len = skb->len + stab->szopts.overhead;
571 	if (unlikely(!stab->szopts.tsize))
572 		goto out;
573 
574 	slot = pkt_len + stab->szopts.cell_align;
575 	if (unlikely(slot < 0))
576 		slot = 0;
577 
578 	slot >>= stab->szopts.cell_log;
579 	if (likely(slot < stab->szopts.tsize))
580 		pkt_len = stab->data[slot];
581 	else
582 		pkt_len = stab->data[stab->szopts.tsize - 1] *
583 				(slot / stab->szopts.tsize) +
584 				stab->data[slot % stab->szopts.tsize];
585 
586 	pkt_len <<= stab->szopts.size_log;
587 out:
588 	if (unlikely(pkt_len < 1))
589 		pkt_len = 1;
590 	qdisc_skb_cb(skb)->pkt_len = pkt_len;
591 }
592 EXPORT_SYMBOL(__qdisc_calculate_pkt_len);
593 
qdisc_warn_nonwc(const char * txt,struct Qdisc * qdisc)594 void qdisc_warn_nonwc(const char *txt, struct Qdisc *qdisc)
595 {
596 	if (!(qdisc->flags & TCQ_F_WARN_NONWC)) {
597 		pr_warn("%s: %s qdisc %X: is non-work-conserving?\n",
598 			txt, qdisc->ops->id, qdisc->handle >> 16);
599 		qdisc->flags |= TCQ_F_WARN_NONWC;
600 	}
601 }
602 EXPORT_SYMBOL(qdisc_warn_nonwc);
603 
qdisc_watchdog(struct hrtimer * timer)604 static enum hrtimer_restart qdisc_watchdog(struct hrtimer *timer)
605 {
606 	struct qdisc_watchdog *wd = container_of(timer, struct qdisc_watchdog,
607 						 timer);
608 
609 	rcu_read_lock();
610 	__netif_schedule(qdisc_root(wd->qdisc));
611 	rcu_read_unlock();
612 
613 	return HRTIMER_NORESTART;
614 }
615 
qdisc_watchdog_init_clockid(struct qdisc_watchdog * wd,struct Qdisc * qdisc,clockid_t clockid)616 void qdisc_watchdog_init_clockid(struct qdisc_watchdog *wd, struct Qdisc *qdisc,
617 				 clockid_t clockid)
618 {
619 	hrtimer_init(&wd->timer, clockid, HRTIMER_MODE_ABS_PINNED);
620 	wd->timer.function = qdisc_watchdog;
621 	wd->qdisc = qdisc;
622 }
623 EXPORT_SYMBOL(qdisc_watchdog_init_clockid);
624 
qdisc_watchdog_init(struct qdisc_watchdog * wd,struct Qdisc * qdisc)625 void qdisc_watchdog_init(struct qdisc_watchdog *wd, struct Qdisc *qdisc)
626 {
627 	qdisc_watchdog_init_clockid(wd, qdisc, CLOCK_MONOTONIC);
628 }
629 EXPORT_SYMBOL(qdisc_watchdog_init);
630 
qdisc_watchdog_schedule_range_ns(struct qdisc_watchdog * wd,u64 expires,u64 delta_ns)631 void qdisc_watchdog_schedule_range_ns(struct qdisc_watchdog *wd, u64 expires,
632 				      u64 delta_ns)
633 {
634 	if (test_bit(__QDISC_STATE_DEACTIVATED,
635 		     &qdisc_root_sleeping(wd->qdisc)->state))
636 		return;
637 
638 	if (hrtimer_is_queued(&wd->timer)) {
639 		/* If timer is already set in [expires, expires + delta_ns],
640 		 * do not reprogram it.
641 		 */
642 		if (wd->last_expires - expires <= delta_ns)
643 			return;
644 	}
645 
646 	wd->last_expires = expires;
647 	hrtimer_start_range_ns(&wd->timer,
648 			       ns_to_ktime(expires),
649 			       delta_ns,
650 			       HRTIMER_MODE_ABS_PINNED);
651 }
652 EXPORT_SYMBOL(qdisc_watchdog_schedule_range_ns);
653 
qdisc_watchdog_cancel(struct qdisc_watchdog * wd)654 void qdisc_watchdog_cancel(struct qdisc_watchdog *wd)
655 {
656 	hrtimer_cancel(&wd->timer);
657 }
658 EXPORT_SYMBOL(qdisc_watchdog_cancel);
659 
qdisc_class_hash_alloc(unsigned int n)660 static struct hlist_head *qdisc_class_hash_alloc(unsigned int n)
661 {
662 	struct hlist_head *h;
663 	unsigned int i;
664 
665 	h = kvmalloc_array(n, sizeof(struct hlist_head), GFP_KERNEL);
666 
667 	if (h != NULL) {
668 		for (i = 0; i < n; i++)
669 			INIT_HLIST_HEAD(&h[i]);
670 	}
671 	return h;
672 }
673 
qdisc_class_hash_grow(struct Qdisc * sch,struct Qdisc_class_hash * clhash)674 void qdisc_class_hash_grow(struct Qdisc *sch, struct Qdisc_class_hash *clhash)
675 {
676 	struct Qdisc_class_common *cl;
677 	struct hlist_node *next;
678 	struct hlist_head *nhash, *ohash;
679 	unsigned int nsize, nmask, osize;
680 	unsigned int i, h;
681 
682 	/* Rehash when load factor exceeds 0.75 */
683 	if (clhash->hashelems * 4 <= clhash->hashsize * 3)
684 		return;
685 	nsize = clhash->hashsize * 2;
686 	nmask = nsize - 1;
687 	nhash = qdisc_class_hash_alloc(nsize);
688 	if (nhash == NULL)
689 		return;
690 
691 	ohash = clhash->hash;
692 	osize = clhash->hashsize;
693 
694 	sch_tree_lock(sch);
695 	for (i = 0; i < osize; i++) {
696 		hlist_for_each_entry_safe(cl, next, &ohash[i], hnode) {
697 			h = qdisc_class_hash(cl->classid, nmask);
698 			hlist_add_head(&cl->hnode, &nhash[h]);
699 		}
700 	}
701 	clhash->hash     = nhash;
702 	clhash->hashsize = nsize;
703 	clhash->hashmask = nmask;
704 	sch_tree_unlock(sch);
705 
706 	kvfree(ohash);
707 }
708 EXPORT_SYMBOL(qdisc_class_hash_grow);
709 
qdisc_class_hash_init(struct Qdisc_class_hash * clhash)710 int qdisc_class_hash_init(struct Qdisc_class_hash *clhash)
711 {
712 	unsigned int size = 4;
713 
714 	clhash->hash = qdisc_class_hash_alloc(size);
715 	if (!clhash->hash)
716 		return -ENOMEM;
717 	clhash->hashsize  = size;
718 	clhash->hashmask  = size - 1;
719 	clhash->hashelems = 0;
720 	return 0;
721 }
722 EXPORT_SYMBOL(qdisc_class_hash_init);
723 
qdisc_class_hash_destroy(struct Qdisc_class_hash * clhash)724 void qdisc_class_hash_destroy(struct Qdisc_class_hash *clhash)
725 {
726 	kvfree(clhash->hash);
727 }
728 EXPORT_SYMBOL(qdisc_class_hash_destroy);
729 
qdisc_class_hash_insert(struct Qdisc_class_hash * clhash,struct Qdisc_class_common * cl)730 void qdisc_class_hash_insert(struct Qdisc_class_hash *clhash,
731 			     struct Qdisc_class_common *cl)
732 {
733 	unsigned int h;
734 
735 	INIT_HLIST_NODE(&cl->hnode);
736 	h = qdisc_class_hash(cl->classid, clhash->hashmask);
737 	hlist_add_head(&cl->hnode, &clhash->hash[h]);
738 	clhash->hashelems++;
739 }
740 EXPORT_SYMBOL(qdisc_class_hash_insert);
741 
qdisc_class_hash_remove(struct Qdisc_class_hash * clhash,struct Qdisc_class_common * cl)742 void qdisc_class_hash_remove(struct Qdisc_class_hash *clhash,
743 			     struct Qdisc_class_common *cl)
744 {
745 	hlist_del(&cl->hnode);
746 	clhash->hashelems--;
747 }
748 EXPORT_SYMBOL(qdisc_class_hash_remove);
749 
750 /* Allocate an unique handle from space managed by kernel
751  * Possible range is [8000-FFFF]:0000 (0x8000 values)
752  */
qdisc_alloc_handle(struct net_device * dev)753 static u32 qdisc_alloc_handle(struct net_device *dev)
754 {
755 	int i = 0x8000;
756 	static u32 autohandle = TC_H_MAKE(0x80000000U, 0);
757 
758 	do {
759 		autohandle += TC_H_MAKE(0x10000U, 0);
760 		if (autohandle == TC_H_MAKE(TC_H_ROOT, 0))
761 			autohandle = TC_H_MAKE(0x80000000U, 0);
762 		if (!qdisc_lookup(dev, autohandle))
763 			return autohandle;
764 		cond_resched();
765 	} while	(--i > 0);
766 
767 	return 0;
768 }
769 
qdisc_tree_reduce_backlog(struct Qdisc * sch,int n,int len)770 void qdisc_tree_reduce_backlog(struct Qdisc *sch, int n, int len)
771 {
772 	bool qdisc_is_offloaded = sch->flags & TCQ_F_OFFLOADED;
773 	const struct Qdisc_class_ops *cops;
774 	unsigned long cl;
775 	u32 parentid;
776 	bool notify;
777 	int drops;
778 
779 	if (n == 0 && len == 0)
780 		return;
781 	drops = max_t(int, n, 0);
782 	rcu_read_lock();
783 	while ((parentid = sch->parent)) {
784 		if (TC_H_MAJ(parentid) == TC_H_MAJ(TC_H_INGRESS))
785 			break;
786 
787 		if (sch->flags & TCQ_F_NOPARENT)
788 			break;
789 		/* Notify parent qdisc only if child qdisc becomes empty.
790 		 *
791 		 * If child was empty even before update then backlog
792 		 * counter is screwed and we skip notification because
793 		 * parent class is already passive.
794 		 *
795 		 * If the original child was offloaded then it is allowed
796 		 * to be seem as empty, so the parent is notified anyway.
797 		 */
798 		notify = !sch->q.qlen && !WARN_ON_ONCE(!n &&
799 						       !qdisc_is_offloaded);
800 		/* TODO: perform the search on a per txq basis */
801 		sch = qdisc_lookup(qdisc_dev(sch), TC_H_MAJ(parentid));
802 		if (sch == NULL) {
803 			WARN_ON_ONCE(parentid != TC_H_ROOT);
804 			break;
805 		}
806 		cops = sch->ops->cl_ops;
807 		if (notify && cops->qlen_notify) {
808 			cl = cops->find(sch, parentid);
809 			cops->qlen_notify(sch, cl);
810 		}
811 		sch->q.qlen -= n;
812 		sch->qstats.backlog -= len;
813 		__qdisc_qstats_drop(sch, drops);
814 	}
815 	rcu_read_unlock();
816 }
817 EXPORT_SYMBOL(qdisc_tree_reduce_backlog);
818 
qdisc_offload_dump_helper(struct Qdisc * sch,enum tc_setup_type type,void * type_data)819 int qdisc_offload_dump_helper(struct Qdisc *sch, enum tc_setup_type type,
820 			      void *type_data)
821 {
822 	struct net_device *dev = qdisc_dev(sch);
823 	int err;
824 
825 	sch->flags &= ~TCQ_F_OFFLOADED;
826 	if (!tc_can_offload(dev) || !dev->netdev_ops->ndo_setup_tc)
827 		return 0;
828 
829 	err = dev->netdev_ops->ndo_setup_tc(dev, type, type_data);
830 	if (err == -EOPNOTSUPP)
831 		return 0;
832 
833 	if (!err)
834 		sch->flags |= TCQ_F_OFFLOADED;
835 
836 	return err;
837 }
838 EXPORT_SYMBOL(qdisc_offload_dump_helper);
839 
qdisc_offload_graft_helper(struct net_device * dev,struct Qdisc * sch,struct Qdisc * new,struct Qdisc * old,enum tc_setup_type type,void * type_data,struct netlink_ext_ack * extack)840 void qdisc_offload_graft_helper(struct net_device *dev, struct Qdisc *sch,
841 				struct Qdisc *new, struct Qdisc *old,
842 				enum tc_setup_type type, void *type_data,
843 				struct netlink_ext_ack *extack)
844 {
845 	bool any_qdisc_is_offloaded;
846 	int err;
847 
848 	if (!tc_can_offload(dev) || !dev->netdev_ops->ndo_setup_tc)
849 		return;
850 
851 	err = dev->netdev_ops->ndo_setup_tc(dev, type, type_data);
852 
853 	/* Don't report error if the graft is part of destroy operation. */
854 	if (!err || !new || new == &noop_qdisc)
855 		return;
856 
857 	/* Don't report error if the parent, the old child and the new
858 	 * one are not offloaded.
859 	 */
860 	any_qdisc_is_offloaded = new->flags & TCQ_F_OFFLOADED;
861 	any_qdisc_is_offloaded |= sch && sch->flags & TCQ_F_OFFLOADED;
862 	any_qdisc_is_offloaded |= old && old->flags & TCQ_F_OFFLOADED;
863 
864 	if (any_qdisc_is_offloaded)
865 		NL_SET_ERR_MSG(extack, "Offloading graft operation failed.");
866 }
867 EXPORT_SYMBOL(qdisc_offload_graft_helper);
868 
qdisc_offload_graft_root(struct net_device * dev,struct Qdisc * new,struct Qdisc * old,struct netlink_ext_ack * extack)869 static void qdisc_offload_graft_root(struct net_device *dev,
870 				     struct Qdisc *new, struct Qdisc *old,
871 				     struct netlink_ext_ack *extack)
872 {
873 	struct tc_root_qopt_offload graft_offload = {
874 		.command	= TC_ROOT_GRAFT,
875 		.handle		= new ? new->handle : 0,
876 		.ingress	= (new && new->flags & TCQ_F_INGRESS) ||
877 				  (old && old->flags & TCQ_F_INGRESS),
878 	};
879 
880 	qdisc_offload_graft_helper(dev, NULL, new, old,
881 				   TC_SETUP_ROOT_QDISC, &graft_offload, extack);
882 }
883 
tc_fill_qdisc(struct sk_buff * skb,struct Qdisc * q,u32 clid,u32 portid,u32 seq,u16 flags,int event)884 static int tc_fill_qdisc(struct sk_buff *skb, struct Qdisc *q, u32 clid,
885 			 u32 portid, u32 seq, u16 flags, int event)
886 {
887 	struct gnet_stats_basic_cpu __percpu *cpu_bstats = NULL;
888 	struct gnet_stats_queue __percpu *cpu_qstats = NULL;
889 	struct tcmsg *tcm;
890 	struct nlmsghdr  *nlh;
891 	unsigned char *b = skb_tail_pointer(skb);
892 	struct gnet_dump d;
893 	struct qdisc_size_table *stab;
894 	u32 block_index;
895 	__u32 qlen;
896 
897 	cond_resched();
898 	nlh = nlmsg_put(skb, portid, seq, event, sizeof(*tcm), flags);
899 	if (!nlh)
900 		goto out_nlmsg_trim;
901 	tcm = nlmsg_data(nlh);
902 	tcm->tcm_family = AF_UNSPEC;
903 	tcm->tcm__pad1 = 0;
904 	tcm->tcm__pad2 = 0;
905 	tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
906 	tcm->tcm_parent = clid;
907 	tcm->tcm_handle = q->handle;
908 	tcm->tcm_info = refcount_read(&q->refcnt);
909 	if (nla_put_string(skb, TCA_KIND, q->ops->id))
910 		goto nla_put_failure;
911 	if (q->ops->ingress_block_get) {
912 		block_index = q->ops->ingress_block_get(q);
913 		if (block_index &&
914 		    nla_put_u32(skb, TCA_INGRESS_BLOCK, block_index))
915 			goto nla_put_failure;
916 	}
917 	if (q->ops->egress_block_get) {
918 		block_index = q->ops->egress_block_get(q);
919 		if (block_index &&
920 		    nla_put_u32(skb, TCA_EGRESS_BLOCK, block_index))
921 			goto nla_put_failure;
922 	}
923 	if (q->ops->dump && q->ops->dump(q, skb) < 0)
924 		goto nla_put_failure;
925 	if (nla_put_u8(skb, TCA_HW_OFFLOAD, !!(q->flags & TCQ_F_OFFLOADED)))
926 		goto nla_put_failure;
927 	qlen = qdisc_qlen_sum(q);
928 
929 	stab = rtnl_dereference(q->stab);
930 	if (stab && qdisc_dump_stab(skb, stab) < 0)
931 		goto nla_put_failure;
932 
933 	if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
934 					 NULL, &d, TCA_PAD) < 0)
935 		goto nla_put_failure;
936 
937 	if (q->ops->dump_stats && q->ops->dump_stats(q, &d) < 0)
938 		goto nla_put_failure;
939 
940 	if (qdisc_is_percpu_stats(q)) {
941 		cpu_bstats = q->cpu_bstats;
942 		cpu_qstats = q->cpu_qstats;
943 	}
944 
945 	if (gnet_stats_copy_basic(qdisc_root_sleeping_running(q),
946 				  &d, cpu_bstats, &q->bstats) < 0 ||
947 	    gnet_stats_copy_rate_est(&d, &q->rate_est) < 0 ||
948 	    gnet_stats_copy_queue(&d, cpu_qstats, &q->qstats, qlen) < 0)
949 		goto nla_put_failure;
950 
951 	if (gnet_stats_finish_copy(&d) < 0)
952 		goto nla_put_failure;
953 
954 	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
955 	return skb->len;
956 
957 out_nlmsg_trim:
958 nla_put_failure:
959 	nlmsg_trim(skb, b);
960 	return -1;
961 }
962 
tc_qdisc_dump_ignore(struct Qdisc * q,bool dump_invisible)963 static bool tc_qdisc_dump_ignore(struct Qdisc *q, bool dump_invisible)
964 {
965 	if (q->flags & TCQ_F_BUILTIN)
966 		return true;
967 	if ((q->flags & TCQ_F_INVISIBLE) && !dump_invisible)
968 		return true;
969 
970 	return false;
971 }
972 
qdisc_notify(struct net * net,struct sk_buff * oskb,struct nlmsghdr * n,u32 clid,struct Qdisc * old,struct Qdisc * new)973 static int qdisc_notify(struct net *net, struct sk_buff *oskb,
974 			struct nlmsghdr *n, u32 clid,
975 			struct Qdisc *old, struct Qdisc *new)
976 {
977 	struct sk_buff *skb;
978 	u32 portid = oskb ? NETLINK_CB(oskb).portid : 0;
979 
980 	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
981 	if (!skb)
982 		return -ENOBUFS;
983 
984 	if (old && !tc_qdisc_dump_ignore(old, false)) {
985 		if (tc_fill_qdisc(skb, old, clid, portid, n->nlmsg_seq,
986 				  0, RTM_DELQDISC) < 0)
987 			goto err_out;
988 	}
989 	if (new && !tc_qdisc_dump_ignore(new, false)) {
990 		if (tc_fill_qdisc(skb, new, clid, portid, n->nlmsg_seq,
991 				  old ? NLM_F_REPLACE : 0, RTM_NEWQDISC) < 0)
992 			goto err_out;
993 	}
994 
995 	if (skb->len)
996 		return rtnetlink_send(skb, net, portid, RTNLGRP_TC,
997 				      n->nlmsg_flags & NLM_F_ECHO);
998 
999 err_out:
1000 	kfree_skb(skb);
1001 	return -EINVAL;
1002 }
1003 
notify_and_destroy(struct net * net,struct sk_buff * skb,struct nlmsghdr * n,u32 clid,struct Qdisc * old,struct Qdisc * new)1004 static void notify_and_destroy(struct net *net, struct sk_buff *skb,
1005 			       struct nlmsghdr *n, u32 clid,
1006 			       struct Qdisc *old, struct Qdisc *new)
1007 {
1008 	if (new || old)
1009 		qdisc_notify(net, skb, n, clid, old, new);
1010 
1011 	if (old)
1012 		qdisc_put(old);
1013 }
1014 
qdisc_clear_nolock(struct Qdisc * sch)1015 static void qdisc_clear_nolock(struct Qdisc *sch)
1016 {
1017 	sch->flags &= ~TCQ_F_NOLOCK;
1018 	if (!(sch->flags & TCQ_F_CPUSTATS))
1019 		return;
1020 
1021 	free_percpu(sch->cpu_bstats);
1022 	free_percpu(sch->cpu_qstats);
1023 	sch->cpu_bstats = NULL;
1024 	sch->cpu_qstats = NULL;
1025 	sch->flags &= ~TCQ_F_CPUSTATS;
1026 }
1027 
1028 /* Graft qdisc "new" to class "classid" of qdisc "parent" or
1029  * to device "dev".
1030  *
1031  * When appropriate send a netlink notification using 'skb'
1032  * and "n".
1033  *
1034  * On success, destroy old qdisc.
1035  */
1036 
qdisc_graft(struct net_device * dev,struct Qdisc * parent,struct sk_buff * skb,struct nlmsghdr * n,u32 classid,struct Qdisc * new,struct Qdisc * old,struct netlink_ext_ack * extack)1037 static int qdisc_graft(struct net_device *dev, struct Qdisc *parent,
1038 		       struct sk_buff *skb, struct nlmsghdr *n, u32 classid,
1039 		       struct Qdisc *new, struct Qdisc *old,
1040 		       struct netlink_ext_ack *extack)
1041 {
1042 	struct Qdisc *q = old;
1043 	struct net *net = dev_net(dev);
1044 
1045 	if (parent == NULL) {
1046 		unsigned int i, num_q, ingress;
1047 
1048 		ingress = 0;
1049 		num_q = dev->num_tx_queues;
1050 		if ((q && q->flags & TCQ_F_INGRESS) ||
1051 		    (new && new->flags & TCQ_F_INGRESS)) {
1052 			num_q = 1;
1053 			ingress = 1;
1054 			if (!dev_ingress_queue(dev)) {
1055 				NL_SET_ERR_MSG(extack, "Device does not have an ingress queue");
1056 				return -ENOENT;
1057 			}
1058 		}
1059 
1060 		if (dev->flags & IFF_UP)
1061 			dev_deactivate(dev);
1062 
1063 		qdisc_offload_graft_root(dev, new, old, extack);
1064 
1065 		if (new && new->ops->attach)
1066 			goto skip;
1067 
1068 		for (i = 0; i < num_q; i++) {
1069 			struct netdev_queue *dev_queue = dev_ingress_queue(dev);
1070 
1071 			if (!ingress)
1072 				dev_queue = netdev_get_tx_queue(dev, i);
1073 
1074 			old = dev_graft_qdisc(dev_queue, new);
1075 			if (new && i > 0)
1076 				qdisc_refcount_inc(new);
1077 
1078 			if (!ingress)
1079 				qdisc_put(old);
1080 		}
1081 
1082 skip:
1083 		if (!ingress) {
1084 			old = rtnl_dereference(dev->qdisc);
1085 			if (new && !new->ops->attach)
1086 				qdisc_refcount_inc(new);
1087 			rcu_assign_pointer(dev->qdisc, new ? : &noop_qdisc);
1088 
1089 			notify_and_destroy(net, skb, n, classid, old, new);
1090 
1091 			if (new && new->ops->attach)
1092 				new->ops->attach(new);
1093 		} else {
1094 			notify_and_destroy(net, skb, n, classid, old, new);
1095 		}
1096 
1097 		if (dev->flags & IFF_UP)
1098 			dev_activate(dev);
1099 	} else {
1100 		const struct Qdisc_class_ops *cops = parent->ops->cl_ops;
1101 		unsigned long cl;
1102 		int err;
1103 
1104 		/* Only support running class lockless if parent is lockless */
1105 		if (new && (new->flags & TCQ_F_NOLOCK) && !(parent->flags & TCQ_F_NOLOCK))
1106 			qdisc_clear_nolock(new);
1107 
1108 		if (!cops || !cops->graft)
1109 			return -EOPNOTSUPP;
1110 
1111 		cl = cops->find(parent, classid);
1112 		if (!cl) {
1113 			NL_SET_ERR_MSG(extack, "Specified class not found");
1114 			return -ENOENT;
1115 		}
1116 
1117 		if (new && new->ops == &noqueue_qdisc_ops) {
1118 			NL_SET_ERR_MSG(extack, "Cannot assign noqueue to a class");
1119 			return -EINVAL;
1120 		}
1121 
1122 		err = cops->graft(parent, cl, new, &old, extack);
1123 		if (err)
1124 			return err;
1125 		notify_and_destroy(net, skb, n, classid, old, new);
1126 	}
1127 	return 0;
1128 }
1129 
qdisc_block_indexes_set(struct Qdisc * sch,struct nlattr ** tca,struct netlink_ext_ack * extack)1130 static int qdisc_block_indexes_set(struct Qdisc *sch, struct nlattr **tca,
1131 				   struct netlink_ext_ack *extack)
1132 {
1133 	u32 block_index;
1134 
1135 	if (tca[TCA_INGRESS_BLOCK]) {
1136 		block_index = nla_get_u32(tca[TCA_INGRESS_BLOCK]);
1137 
1138 		if (!block_index) {
1139 			NL_SET_ERR_MSG(extack, "Ingress block index cannot be 0");
1140 			return -EINVAL;
1141 		}
1142 		if (!sch->ops->ingress_block_set) {
1143 			NL_SET_ERR_MSG(extack, "Ingress block sharing is not supported");
1144 			return -EOPNOTSUPP;
1145 		}
1146 		sch->ops->ingress_block_set(sch, block_index);
1147 	}
1148 	if (tca[TCA_EGRESS_BLOCK]) {
1149 		block_index = nla_get_u32(tca[TCA_EGRESS_BLOCK]);
1150 
1151 		if (!block_index) {
1152 			NL_SET_ERR_MSG(extack, "Egress block index cannot be 0");
1153 			return -EINVAL;
1154 		}
1155 		if (!sch->ops->egress_block_set) {
1156 			NL_SET_ERR_MSG(extack, "Egress block sharing is not supported");
1157 			return -EOPNOTSUPP;
1158 		}
1159 		sch->ops->egress_block_set(sch, block_index);
1160 	}
1161 	return 0;
1162 }
1163 
1164 /*
1165    Allocate and initialize new qdisc.
1166 
1167    Parameters are passed via opt.
1168  */
1169 
qdisc_create(struct net_device * dev,struct netdev_queue * dev_queue,struct Qdisc * p,u32 parent,u32 handle,struct nlattr ** tca,int * errp,struct netlink_ext_ack * extack)1170 static struct Qdisc *qdisc_create(struct net_device *dev,
1171 				  struct netdev_queue *dev_queue,
1172 				  struct Qdisc *p, u32 parent, u32 handle,
1173 				  struct nlattr **tca, int *errp,
1174 				  struct netlink_ext_ack *extack)
1175 {
1176 	int err;
1177 	struct nlattr *kind = tca[TCA_KIND];
1178 	struct Qdisc *sch;
1179 	struct Qdisc_ops *ops;
1180 	struct qdisc_size_table *stab;
1181 
1182 	ops = qdisc_lookup_ops(kind);
1183 #ifdef CONFIG_MODULES
1184 	if (ops == NULL && kind != NULL) {
1185 		char name[IFNAMSIZ];
1186 		if (nla_strlcpy(name, kind, IFNAMSIZ) < IFNAMSIZ) {
1187 			/* We dropped the RTNL semaphore in order to
1188 			 * perform the module load.  So, even if we
1189 			 * succeeded in loading the module we have to
1190 			 * tell the caller to replay the request.  We
1191 			 * indicate this using -EAGAIN.
1192 			 * We replay the request because the device may
1193 			 * go away in the mean time.
1194 			 */
1195 			rtnl_unlock();
1196 			request_module("sch_%s", name);
1197 			rtnl_lock();
1198 			ops = qdisc_lookup_ops(kind);
1199 			if (ops != NULL) {
1200 				/* We will try again qdisc_lookup_ops,
1201 				 * so don't keep a reference.
1202 				 */
1203 				module_put(ops->owner);
1204 				err = -EAGAIN;
1205 				goto err_out;
1206 			}
1207 		}
1208 	}
1209 #endif
1210 
1211 	err = -ENOENT;
1212 	if (!ops) {
1213 		NL_SET_ERR_MSG(extack, "Specified qdisc kind is unknown");
1214 		goto err_out;
1215 	}
1216 
1217 	sch = qdisc_alloc(dev_queue, ops, extack);
1218 	if (IS_ERR(sch)) {
1219 		err = PTR_ERR(sch);
1220 		goto err_out2;
1221 	}
1222 
1223 	sch->parent = parent;
1224 
1225 	if (handle == TC_H_INGRESS) {
1226 		if (!(sch->flags & TCQ_F_INGRESS)) {
1227 			NL_SET_ERR_MSG(extack,
1228 				       "Specified parent ID is reserved for ingress and clsact Qdiscs");
1229 			err = -EINVAL;
1230 			goto err_out3;
1231 		}
1232 		handle = TC_H_MAKE(TC_H_INGRESS, 0);
1233 	} else {
1234 		if (handle == 0) {
1235 			handle = qdisc_alloc_handle(dev);
1236 			if (handle == 0) {
1237 				NL_SET_ERR_MSG(extack, "Maximum number of qdisc handles was exceeded");
1238 				err = -ENOSPC;
1239 				goto err_out3;
1240 			}
1241 		}
1242 		if (!netif_is_multiqueue(dev))
1243 			sch->flags |= TCQ_F_ONETXQUEUE;
1244 	}
1245 
1246 	sch->handle = handle;
1247 
1248 	/* This exist to keep backward compatible with a userspace
1249 	 * loophole, what allowed userspace to get IFF_NO_QUEUE
1250 	 * facility on older kernels by setting tx_queue_len=0 (prior
1251 	 * to qdisc init), and then forgot to reinit tx_queue_len
1252 	 * before again attaching a qdisc.
1253 	 */
1254 	if ((dev->priv_flags & IFF_NO_QUEUE) && (dev->tx_queue_len == 0)) {
1255 		dev->tx_queue_len = DEFAULT_TX_QUEUE_LEN;
1256 		netdev_info(dev, "Caught tx_queue_len zero misconfig\n");
1257 	}
1258 
1259 	err = qdisc_block_indexes_set(sch, tca, extack);
1260 	if (err)
1261 		goto err_out3;
1262 
1263 	if (ops->init) {
1264 		err = ops->init(sch, tca[TCA_OPTIONS], extack);
1265 		if (err != 0)
1266 			goto err_out5;
1267 	}
1268 
1269 	if (tca[TCA_STAB]) {
1270 		stab = qdisc_get_stab(tca[TCA_STAB], extack);
1271 		if (IS_ERR(stab)) {
1272 			err = PTR_ERR(stab);
1273 			goto err_out4;
1274 		}
1275 		rcu_assign_pointer(sch->stab, stab);
1276 	}
1277 	if (tca[TCA_RATE]) {
1278 		seqcount_t *running;
1279 
1280 		err = -EOPNOTSUPP;
1281 		if (sch->flags & TCQ_F_MQROOT) {
1282 			NL_SET_ERR_MSG(extack, "Cannot attach rate estimator to a multi-queue root qdisc");
1283 			goto err_out4;
1284 		}
1285 
1286 		if (sch->parent != TC_H_ROOT &&
1287 		    !(sch->flags & TCQ_F_INGRESS) &&
1288 		    (!p || !(p->flags & TCQ_F_MQROOT)))
1289 			running = qdisc_root_sleeping_running(sch);
1290 		else
1291 			running = &sch->running;
1292 
1293 		err = gen_new_estimator(&sch->bstats,
1294 					sch->cpu_bstats,
1295 					&sch->rate_est,
1296 					NULL,
1297 					running,
1298 					tca[TCA_RATE]);
1299 		if (err) {
1300 			NL_SET_ERR_MSG(extack, "Failed to generate new estimator");
1301 			goto err_out4;
1302 		}
1303 	}
1304 
1305 	qdisc_hash_add(sch, false);
1306 	trace_qdisc_create(ops, dev, parent);
1307 
1308 	return sch;
1309 
1310 err_out5:
1311 	/* ops->init() failed, we call ->destroy() like qdisc_create_dflt() */
1312 	if (ops->destroy)
1313 		ops->destroy(sch);
1314 err_out3:
1315 	dev_put(dev);
1316 	qdisc_free(sch);
1317 err_out2:
1318 	module_put(ops->owner);
1319 err_out:
1320 	*errp = err;
1321 	return NULL;
1322 
1323 err_out4:
1324 	/*
1325 	 * Any broken qdiscs that would require a ops->reset() here?
1326 	 * The qdisc was never in action so it shouldn't be necessary.
1327 	 */
1328 	qdisc_put_stab(rtnl_dereference(sch->stab));
1329 	if (ops->destroy)
1330 		ops->destroy(sch);
1331 	goto err_out3;
1332 }
1333 
qdisc_change(struct Qdisc * sch,struct nlattr ** tca,struct netlink_ext_ack * extack)1334 static int qdisc_change(struct Qdisc *sch, struct nlattr **tca,
1335 			struct netlink_ext_ack *extack)
1336 {
1337 	struct qdisc_size_table *ostab, *stab = NULL;
1338 	int err = 0;
1339 
1340 	if (tca[TCA_OPTIONS]) {
1341 		if (!sch->ops->change) {
1342 			NL_SET_ERR_MSG(extack, "Change operation not supported by specified qdisc");
1343 			return -EINVAL;
1344 		}
1345 		if (tca[TCA_INGRESS_BLOCK] || tca[TCA_EGRESS_BLOCK]) {
1346 			NL_SET_ERR_MSG(extack, "Change of blocks is not supported");
1347 			return -EOPNOTSUPP;
1348 		}
1349 		err = sch->ops->change(sch, tca[TCA_OPTIONS], extack);
1350 		if (err)
1351 			return err;
1352 	}
1353 
1354 	if (tca[TCA_STAB]) {
1355 		stab = qdisc_get_stab(tca[TCA_STAB], extack);
1356 		if (IS_ERR(stab))
1357 			return PTR_ERR(stab);
1358 	}
1359 
1360 	ostab = rtnl_dereference(sch->stab);
1361 	rcu_assign_pointer(sch->stab, stab);
1362 	qdisc_put_stab(ostab);
1363 
1364 	if (tca[TCA_RATE]) {
1365 		/* NB: ignores errors from replace_estimator
1366 		   because change can't be undone. */
1367 		if (sch->flags & TCQ_F_MQROOT)
1368 			goto out;
1369 		gen_replace_estimator(&sch->bstats,
1370 				      sch->cpu_bstats,
1371 				      &sch->rate_est,
1372 				      NULL,
1373 				      qdisc_root_sleeping_running(sch),
1374 				      tca[TCA_RATE]);
1375 	}
1376 out:
1377 	return 0;
1378 }
1379 
1380 struct check_loop_arg {
1381 	struct qdisc_walker	w;
1382 	struct Qdisc		*p;
1383 	int			depth;
1384 };
1385 
1386 static int check_loop_fn(struct Qdisc *q, unsigned long cl,
1387 			 struct qdisc_walker *w);
1388 
check_loop(struct Qdisc * q,struct Qdisc * p,int depth)1389 static int check_loop(struct Qdisc *q, struct Qdisc *p, int depth)
1390 {
1391 	struct check_loop_arg	arg;
1392 
1393 	if (q->ops->cl_ops == NULL)
1394 		return 0;
1395 
1396 	arg.w.stop = arg.w.skip = arg.w.count = 0;
1397 	arg.w.fn = check_loop_fn;
1398 	arg.depth = depth;
1399 	arg.p = p;
1400 	q->ops->cl_ops->walk(q, &arg.w);
1401 	return arg.w.stop ? -ELOOP : 0;
1402 }
1403 
1404 static int
check_loop_fn(struct Qdisc * q,unsigned long cl,struct qdisc_walker * w)1405 check_loop_fn(struct Qdisc *q, unsigned long cl, struct qdisc_walker *w)
1406 {
1407 	struct Qdisc *leaf;
1408 	const struct Qdisc_class_ops *cops = q->ops->cl_ops;
1409 	struct check_loop_arg *arg = (struct check_loop_arg *)w;
1410 
1411 	leaf = cops->leaf(q, cl);
1412 	if (leaf) {
1413 		if (leaf == arg->p || arg->depth > 7)
1414 			return -ELOOP;
1415 		return check_loop(leaf, arg->p, arg->depth + 1);
1416 	}
1417 	return 0;
1418 }
1419 
1420 const struct nla_policy rtm_tca_policy[TCA_MAX + 1] = {
1421 	[TCA_KIND]		= { .type = NLA_STRING },
1422 	[TCA_RATE]		= { .type = NLA_BINARY,
1423 				    .len = sizeof(struct tc_estimator) },
1424 	[TCA_STAB]		= { .type = NLA_NESTED },
1425 	[TCA_DUMP_INVISIBLE]	= { .type = NLA_FLAG },
1426 	[TCA_CHAIN]		= { .type = NLA_U32 },
1427 	[TCA_INGRESS_BLOCK]	= { .type = NLA_U32 },
1428 	[TCA_EGRESS_BLOCK]	= { .type = NLA_U32 },
1429 };
1430 
1431 /*
1432  * Delete/get qdisc.
1433  */
1434 
tc_get_qdisc(struct sk_buff * skb,struct nlmsghdr * n,struct netlink_ext_ack * extack)1435 static int tc_get_qdisc(struct sk_buff *skb, struct nlmsghdr *n,
1436 			struct netlink_ext_ack *extack)
1437 {
1438 	struct net *net = sock_net(skb->sk);
1439 	struct tcmsg *tcm = nlmsg_data(n);
1440 	struct nlattr *tca[TCA_MAX + 1];
1441 	struct net_device *dev;
1442 	u32 clid;
1443 	struct Qdisc *q = NULL;
1444 	struct Qdisc *p = NULL;
1445 	int err;
1446 
1447 	if ((n->nlmsg_type != RTM_GETQDISC) &&
1448 	    !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
1449 		return -EPERM;
1450 
1451 	err = nlmsg_parse_deprecated(n, sizeof(*tcm), tca, TCA_MAX,
1452 				     rtm_tca_policy, extack);
1453 	if (err < 0)
1454 		return err;
1455 
1456 	dev = __dev_get_by_index(net, tcm->tcm_ifindex);
1457 	if (!dev)
1458 		return -ENODEV;
1459 
1460 	clid = tcm->tcm_parent;
1461 	if (clid) {
1462 		if (clid != TC_H_ROOT) {
1463 			if (TC_H_MAJ(clid) != TC_H_MAJ(TC_H_INGRESS)) {
1464 				p = qdisc_lookup(dev, TC_H_MAJ(clid));
1465 				if (!p) {
1466 					NL_SET_ERR_MSG(extack, "Failed to find qdisc with specified classid");
1467 					return -ENOENT;
1468 				}
1469 				q = qdisc_leaf(p, clid);
1470 			} else if (dev_ingress_queue(dev)) {
1471 				q = dev_ingress_queue(dev)->qdisc_sleeping;
1472 			}
1473 		} else {
1474 			q = rtnl_dereference(dev->qdisc);
1475 		}
1476 		if (!q) {
1477 			NL_SET_ERR_MSG(extack, "Cannot find specified qdisc on specified device");
1478 			return -ENOENT;
1479 		}
1480 
1481 		if (tcm->tcm_handle && q->handle != tcm->tcm_handle) {
1482 			NL_SET_ERR_MSG(extack, "Invalid handle");
1483 			return -EINVAL;
1484 		}
1485 	} else {
1486 		q = qdisc_lookup(dev, tcm->tcm_handle);
1487 		if (!q) {
1488 			NL_SET_ERR_MSG(extack, "Failed to find qdisc with specified handle");
1489 			return -ENOENT;
1490 		}
1491 	}
1492 
1493 	if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id)) {
1494 		NL_SET_ERR_MSG(extack, "Invalid qdisc name");
1495 		return -EINVAL;
1496 	}
1497 
1498 	if (n->nlmsg_type == RTM_DELQDISC) {
1499 		if (!clid) {
1500 			NL_SET_ERR_MSG(extack, "Classid cannot be zero");
1501 			return -EINVAL;
1502 		}
1503 		if (q->handle == 0) {
1504 			NL_SET_ERR_MSG(extack, "Cannot delete qdisc with handle of zero");
1505 			return -ENOENT;
1506 		}
1507 		err = qdisc_graft(dev, p, skb, n, clid, NULL, q, extack);
1508 		if (err != 0)
1509 			return err;
1510 	} else {
1511 		qdisc_notify(net, skb, n, clid, NULL, q);
1512 	}
1513 	return 0;
1514 }
1515 
1516 /*
1517  * Create/change qdisc.
1518  */
1519 
tc_modify_qdisc(struct sk_buff * skb,struct nlmsghdr * n,struct netlink_ext_ack * extack)1520 static int tc_modify_qdisc(struct sk_buff *skb, struct nlmsghdr *n,
1521 			   struct netlink_ext_ack *extack)
1522 {
1523 	struct net *net = sock_net(skb->sk);
1524 	struct tcmsg *tcm;
1525 	struct nlattr *tca[TCA_MAX + 1];
1526 	struct net_device *dev;
1527 	u32 clid;
1528 	struct Qdisc *q, *p;
1529 	int err;
1530 
1531 	if (!netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
1532 		return -EPERM;
1533 
1534 replay:
1535 	/* Reinit, just in case something touches this. */
1536 	err = nlmsg_parse_deprecated(n, sizeof(*tcm), tca, TCA_MAX,
1537 				     rtm_tca_policy, extack);
1538 	if (err < 0)
1539 		return err;
1540 
1541 	tcm = nlmsg_data(n);
1542 	clid = tcm->tcm_parent;
1543 	q = p = NULL;
1544 
1545 	dev = __dev_get_by_index(net, tcm->tcm_ifindex);
1546 	if (!dev)
1547 		return -ENODEV;
1548 
1549 
1550 	if (clid) {
1551 		if (clid != TC_H_ROOT) {
1552 			if (clid != TC_H_INGRESS) {
1553 				p = qdisc_lookup(dev, TC_H_MAJ(clid));
1554 				if (!p) {
1555 					NL_SET_ERR_MSG(extack, "Failed to find specified qdisc");
1556 					return -ENOENT;
1557 				}
1558 				q = qdisc_leaf(p, clid);
1559 			} else if (dev_ingress_queue_create(dev)) {
1560 				q = dev_ingress_queue(dev)->qdisc_sleeping;
1561 			}
1562 		} else {
1563 			q = rtnl_dereference(dev->qdisc);
1564 		}
1565 
1566 		/* It may be default qdisc, ignore it */
1567 		if (q && q->handle == 0)
1568 			q = NULL;
1569 
1570 		if (!q || !tcm->tcm_handle || q->handle != tcm->tcm_handle) {
1571 			if (tcm->tcm_handle) {
1572 				if (q && !(n->nlmsg_flags & NLM_F_REPLACE)) {
1573 					NL_SET_ERR_MSG(extack, "NLM_F_REPLACE needed to override");
1574 					return -EEXIST;
1575 				}
1576 				if (TC_H_MIN(tcm->tcm_handle)) {
1577 					NL_SET_ERR_MSG(extack, "Invalid minor handle");
1578 					return -EINVAL;
1579 				}
1580 				q = qdisc_lookup(dev, tcm->tcm_handle);
1581 				if (!q)
1582 					goto create_n_graft;
1583 				if (n->nlmsg_flags & NLM_F_EXCL) {
1584 					NL_SET_ERR_MSG(extack, "Exclusivity flag on, cannot override");
1585 					return -EEXIST;
1586 				}
1587 				if (tca[TCA_KIND] &&
1588 				    nla_strcmp(tca[TCA_KIND], q->ops->id)) {
1589 					NL_SET_ERR_MSG(extack, "Invalid qdisc name");
1590 					return -EINVAL;
1591 				}
1592 				if (q->flags & TCQ_F_INGRESS) {
1593 					NL_SET_ERR_MSG(extack,
1594 						       "Cannot regraft ingress or clsact Qdiscs");
1595 					return -EINVAL;
1596 				}
1597 				if (q == p ||
1598 				    (p && check_loop(q, p, 0))) {
1599 					NL_SET_ERR_MSG(extack, "Qdisc parent/child loop detected");
1600 					return -ELOOP;
1601 				}
1602 				if (clid == TC_H_INGRESS) {
1603 					NL_SET_ERR_MSG(extack, "Ingress cannot graft directly");
1604 					return -EINVAL;
1605 				}
1606 				qdisc_refcount_inc(q);
1607 				goto graft;
1608 			} else {
1609 				if (!q)
1610 					goto create_n_graft;
1611 
1612 				/* This magic test requires explanation.
1613 				 *
1614 				 *   We know, that some child q is already
1615 				 *   attached to this parent and have choice:
1616 				 *   either to change it or to create/graft new one.
1617 				 *
1618 				 *   1. We are allowed to create/graft only
1619 				 *   if CREATE and REPLACE flags are set.
1620 				 *
1621 				 *   2. If EXCL is set, requestor wanted to say,
1622 				 *   that qdisc tcm_handle is not expected
1623 				 *   to exist, so that we choose create/graft too.
1624 				 *
1625 				 *   3. The last case is when no flags are set.
1626 				 *   Alas, it is sort of hole in API, we
1627 				 *   cannot decide what to do unambiguously.
1628 				 *   For now we select create/graft, if
1629 				 *   user gave KIND, which does not match existing.
1630 				 */
1631 				if ((n->nlmsg_flags & NLM_F_CREATE) &&
1632 				    (n->nlmsg_flags & NLM_F_REPLACE) &&
1633 				    ((n->nlmsg_flags & NLM_F_EXCL) ||
1634 				     (tca[TCA_KIND] &&
1635 				      nla_strcmp(tca[TCA_KIND], q->ops->id))))
1636 					goto create_n_graft;
1637 			}
1638 		}
1639 	} else {
1640 		if (!tcm->tcm_handle) {
1641 			NL_SET_ERR_MSG(extack, "Handle cannot be zero");
1642 			return -EINVAL;
1643 		}
1644 		q = qdisc_lookup(dev, tcm->tcm_handle);
1645 	}
1646 
1647 	/* Change qdisc parameters */
1648 	if (!q) {
1649 		NL_SET_ERR_MSG(extack, "Specified qdisc not found");
1650 		return -ENOENT;
1651 	}
1652 	if (n->nlmsg_flags & NLM_F_EXCL) {
1653 		NL_SET_ERR_MSG(extack, "Exclusivity flag on, cannot modify");
1654 		return -EEXIST;
1655 	}
1656 	if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], q->ops->id)) {
1657 		NL_SET_ERR_MSG(extack, "Invalid qdisc name");
1658 		return -EINVAL;
1659 	}
1660 	err = qdisc_change(q, tca, extack);
1661 	if (err == 0)
1662 		qdisc_notify(net, skb, n, clid, NULL, q);
1663 	return err;
1664 
1665 create_n_graft:
1666 	if (!(n->nlmsg_flags & NLM_F_CREATE)) {
1667 		NL_SET_ERR_MSG(extack, "Qdisc not found. To create specify NLM_F_CREATE flag");
1668 		return -ENOENT;
1669 	}
1670 	if (clid == TC_H_INGRESS) {
1671 		if (dev_ingress_queue(dev)) {
1672 			q = qdisc_create(dev, dev_ingress_queue(dev), p,
1673 					 tcm->tcm_parent, tcm->tcm_parent,
1674 					 tca, &err, extack);
1675 		} else {
1676 			NL_SET_ERR_MSG(extack, "Cannot find ingress queue for specified device");
1677 			err = -ENOENT;
1678 		}
1679 	} else {
1680 		struct netdev_queue *dev_queue;
1681 
1682 		if (p && p->ops->cl_ops && p->ops->cl_ops->select_queue)
1683 			dev_queue = p->ops->cl_ops->select_queue(p, tcm);
1684 		else if (p)
1685 			dev_queue = p->dev_queue;
1686 		else
1687 			dev_queue = netdev_get_tx_queue(dev, 0);
1688 
1689 		q = qdisc_create(dev, dev_queue, p,
1690 				 tcm->tcm_parent, tcm->tcm_handle,
1691 				 tca, &err, extack);
1692 	}
1693 	if (q == NULL) {
1694 		if (err == -EAGAIN)
1695 			goto replay;
1696 		return err;
1697 	}
1698 
1699 graft:
1700 	err = qdisc_graft(dev, p, skb, n, clid, q, NULL, extack);
1701 	if (err) {
1702 		if (q)
1703 			qdisc_put(q);
1704 		return err;
1705 	}
1706 
1707 	return 0;
1708 }
1709 
tc_dump_qdisc_root(struct Qdisc * root,struct sk_buff * skb,struct netlink_callback * cb,int * q_idx_p,int s_q_idx,bool recur,bool dump_invisible)1710 static int tc_dump_qdisc_root(struct Qdisc *root, struct sk_buff *skb,
1711 			      struct netlink_callback *cb,
1712 			      int *q_idx_p, int s_q_idx, bool recur,
1713 			      bool dump_invisible)
1714 {
1715 	int ret = 0, q_idx = *q_idx_p;
1716 	struct Qdisc *q;
1717 	int b;
1718 
1719 	if (!root)
1720 		return 0;
1721 
1722 	q = root;
1723 	if (q_idx < s_q_idx) {
1724 		q_idx++;
1725 	} else {
1726 		if (!tc_qdisc_dump_ignore(q, dump_invisible) &&
1727 		    tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).portid,
1728 				  cb->nlh->nlmsg_seq, NLM_F_MULTI,
1729 				  RTM_NEWQDISC) <= 0)
1730 			goto done;
1731 		q_idx++;
1732 	}
1733 
1734 	/* If dumping singletons, there is no qdisc_dev(root) and the singleton
1735 	 * itself has already been dumped.
1736 	 *
1737 	 * If we've already dumped the top-level (ingress) qdisc above and the global
1738 	 * qdisc hashtable, we don't want to hit it again
1739 	 */
1740 	if (!qdisc_dev(root) || !recur)
1741 		goto out;
1742 
1743 	hash_for_each(qdisc_dev(root)->qdisc_hash, b, q, hash) {
1744 		if (q_idx < s_q_idx) {
1745 			q_idx++;
1746 			continue;
1747 		}
1748 		if (!tc_qdisc_dump_ignore(q, dump_invisible) &&
1749 		    tc_fill_qdisc(skb, q, q->parent, NETLINK_CB(cb->skb).portid,
1750 				  cb->nlh->nlmsg_seq, NLM_F_MULTI,
1751 				  RTM_NEWQDISC) <= 0)
1752 			goto done;
1753 		q_idx++;
1754 	}
1755 
1756 out:
1757 	*q_idx_p = q_idx;
1758 	return ret;
1759 done:
1760 	ret = -1;
1761 	goto out;
1762 }
1763 
tc_dump_qdisc(struct sk_buff * skb,struct netlink_callback * cb)1764 static int tc_dump_qdisc(struct sk_buff *skb, struct netlink_callback *cb)
1765 {
1766 	struct net *net = sock_net(skb->sk);
1767 	int idx, q_idx;
1768 	int s_idx, s_q_idx;
1769 	struct net_device *dev;
1770 	const struct nlmsghdr *nlh = cb->nlh;
1771 	struct nlattr *tca[TCA_MAX + 1];
1772 	int err;
1773 
1774 	s_idx = cb->args[0];
1775 	s_q_idx = q_idx = cb->args[1];
1776 
1777 	idx = 0;
1778 	ASSERT_RTNL();
1779 
1780 	err = nlmsg_parse_deprecated(nlh, sizeof(struct tcmsg), tca, TCA_MAX,
1781 				     rtm_tca_policy, cb->extack);
1782 	if (err < 0)
1783 		return err;
1784 
1785 	for_each_netdev(net, dev) {
1786 		struct netdev_queue *dev_queue;
1787 
1788 		if (idx < s_idx)
1789 			goto cont;
1790 		if (idx > s_idx)
1791 			s_q_idx = 0;
1792 		q_idx = 0;
1793 
1794 		if (tc_dump_qdisc_root(rtnl_dereference(dev->qdisc),
1795 				       skb, cb, &q_idx, s_q_idx,
1796 				       true, tca[TCA_DUMP_INVISIBLE]) < 0)
1797 			goto done;
1798 
1799 		dev_queue = dev_ingress_queue(dev);
1800 		if (dev_queue &&
1801 		    tc_dump_qdisc_root(dev_queue->qdisc_sleeping, skb, cb,
1802 				       &q_idx, s_q_idx, false,
1803 				       tca[TCA_DUMP_INVISIBLE]) < 0)
1804 			goto done;
1805 
1806 cont:
1807 		idx++;
1808 	}
1809 
1810 done:
1811 	cb->args[0] = idx;
1812 	cb->args[1] = q_idx;
1813 
1814 	return skb->len;
1815 }
1816 
1817 
1818 
1819 /************************************************
1820  *	Traffic classes manipulation.		*
1821  ************************************************/
1822 
tc_fill_tclass(struct sk_buff * skb,struct Qdisc * q,unsigned long cl,u32 portid,u32 seq,u16 flags,int event)1823 static int tc_fill_tclass(struct sk_buff *skb, struct Qdisc *q,
1824 			  unsigned long cl,
1825 			  u32 portid, u32 seq, u16 flags, int event)
1826 {
1827 	struct tcmsg *tcm;
1828 	struct nlmsghdr  *nlh;
1829 	unsigned char *b = skb_tail_pointer(skb);
1830 	struct gnet_dump d;
1831 	const struct Qdisc_class_ops *cl_ops = q->ops->cl_ops;
1832 
1833 	cond_resched();
1834 	nlh = nlmsg_put(skb, portid, seq, event, sizeof(*tcm), flags);
1835 	if (!nlh)
1836 		goto out_nlmsg_trim;
1837 	tcm = nlmsg_data(nlh);
1838 	tcm->tcm_family = AF_UNSPEC;
1839 	tcm->tcm__pad1 = 0;
1840 	tcm->tcm__pad2 = 0;
1841 	tcm->tcm_ifindex = qdisc_dev(q)->ifindex;
1842 	tcm->tcm_parent = q->handle;
1843 	tcm->tcm_handle = q->handle;
1844 	tcm->tcm_info = 0;
1845 	if (nla_put_string(skb, TCA_KIND, q->ops->id))
1846 		goto nla_put_failure;
1847 	if (cl_ops->dump && cl_ops->dump(q, cl, skb, tcm) < 0)
1848 		goto nla_put_failure;
1849 
1850 	if (gnet_stats_start_copy_compat(skb, TCA_STATS2, TCA_STATS, TCA_XSTATS,
1851 					 NULL, &d, TCA_PAD) < 0)
1852 		goto nla_put_failure;
1853 
1854 	if (cl_ops->dump_stats && cl_ops->dump_stats(q, cl, &d) < 0)
1855 		goto nla_put_failure;
1856 
1857 	if (gnet_stats_finish_copy(&d) < 0)
1858 		goto nla_put_failure;
1859 
1860 	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
1861 	return skb->len;
1862 
1863 out_nlmsg_trim:
1864 nla_put_failure:
1865 	nlmsg_trim(skb, b);
1866 	return -1;
1867 }
1868 
tclass_notify(struct net * net,struct sk_buff * oskb,struct nlmsghdr * n,struct Qdisc * q,unsigned long cl,int event)1869 static int tclass_notify(struct net *net, struct sk_buff *oskb,
1870 			 struct nlmsghdr *n, struct Qdisc *q,
1871 			 unsigned long cl, int event)
1872 {
1873 	struct sk_buff *skb;
1874 	u32 portid = oskb ? NETLINK_CB(oskb).portid : 0;
1875 	int err = 0;
1876 
1877 	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1878 	if (!skb)
1879 		return -ENOBUFS;
1880 
1881 	if (tc_fill_tclass(skb, q, cl, portid, n->nlmsg_seq, 0, event) < 0) {
1882 		kfree_skb(skb);
1883 		return -EINVAL;
1884 	}
1885 
1886 	err = rtnetlink_send(skb, net, portid, RTNLGRP_TC,
1887 			     n->nlmsg_flags & NLM_F_ECHO);
1888 	if (err > 0)
1889 		err = 0;
1890 	return err;
1891 }
1892 
tclass_del_notify(struct net * net,const struct Qdisc_class_ops * cops,struct sk_buff * oskb,struct nlmsghdr * n,struct Qdisc * q,unsigned long cl)1893 static int tclass_del_notify(struct net *net,
1894 			     const struct Qdisc_class_ops *cops,
1895 			     struct sk_buff *oskb, struct nlmsghdr *n,
1896 			     struct Qdisc *q, unsigned long cl)
1897 {
1898 	u32 portid = oskb ? NETLINK_CB(oskb).portid : 0;
1899 	struct sk_buff *skb;
1900 	int err = 0;
1901 
1902 	if (!cops->delete)
1903 		return -EOPNOTSUPP;
1904 
1905 	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1906 	if (!skb)
1907 		return -ENOBUFS;
1908 
1909 	if (tc_fill_tclass(skb, q, cl, portid, n->nlmsg_seq, 0,
1910 			   RTM_DELTCLASS) < 0) {
1911 		kfree_skb(skb);
1912 		return -EINVAL;
1913 	}
1914 
1915 	err = cops->delete(q, cl);
1916 	if (err) {
1917 		kfree_skb(skb);
1918 		return err;
1919 	}
1920 
1921 	err = rtnetlink_send(skb, net, portid, RTNLGRP_TC,
1922 			     n->nlmsg_flags & NLM_F_ECHO);
1923 	if (err > 0)
1924 		err = 0;
1925 	return err;
1926 }
1927 
1928 #ifdef CONFIG_NET_CLS
1929 
1930 struct tcf_bind_args {
1931 	struct tcf_walker w;
1932 	unsigned long base;
1933 	unsigned long cl;
1934 	u32 classid;
1935 };
1936 
tcf_node_bind(struct tcf_proto * tp,void * n,struct tcf_walker * arg)1937 static int tcf_node_bind(struct tcf_proto *tp, void *n, struct tcf_walker *arg)
1938 {
1939 	struct tcf_bind_args *a = (void *)arg;
1940 
1941 	if (tp->ops->bind_class) {
1942 		struct Qdisc *q = tcf_block_q(tp->chain->block);
1943 
1944 		sch_tree_lock(q);
1945 		tp->ops->bind_class(n, a->classid, a->cl, q, a->base);
1946 		sch_tree_unlock(q);
1947 	}
1948 	return 0;
1949 }
1950 
1951 struct tc_bind_class_args {
1952 	struct qdisc_walker w;
1953 	unsigned long new_cl;
1954 	u32 portid;
1955 	u32 clid;
1956 };
1957 
tc_bind_class_walker(struct Qdisc * q,unsigned long cl,struct qdisc_walker * w)1958 static int tc_bind_class_walker(struct Qdisc *q, unsigned long cl,
1959 				struct qdisc_walker *w)
1960 {
1961 	struct tc_bind_class_args *a = (struct tc_bind_class_args *)w;
1962 	const struct Qdisc_class_ops *cops = q->ops->cl_ops;
1963 	struct tcf_block *block;
1964 	struct tcf_chain *chain;
1965 
1966 	block = cops->tcf_block(q, cl, NULL);
1967 	if (!block)
1968 		return 0;
1969 	for (chain = tcf_get_next_chain(block, NULL);
1970 	     chain;
1971 	     chain = tcf_get_next_chain(block, chain)) {
1972 		struct tcf_proto *tp;
1973 
1974 		for (tp = tcf_get_next_proto(chain, NULL, true);
1975 		     tp; tp = tcf_get_next_proto(chain, tp, true)) {
1976 			struct tcf_bind_args arg = {};
1977 
1978 			arg.w.fn = tcf_node_bind;
1979 			arg.classid = a->clid;
1980 			arg.base = cl;
1981 			arg.cl = a->new_cl;
1982 			tp->ops->walk(tp, &arg.w, true);
1983 		}
1984 	}
1985 
1986 	return 0;
1987 }
1988 
tc_bind_tclass(struct Qdisc * q,u32 portid,u32 clid,unsigned long new_cl)1989 static void tc_bind_tclass(struct Qdisc *q, u32 portid, u32 clid,
1990 			   unsigned long new_cl)
1991 {
1992 	const struct Qdisc_class_ops *cops = q->ops->cl_ops;
1993 	struct tc_bind_class_args args = {};
1994 
1995 	if (!cops->tcf_block)
1996 		return;
1997 	args.portid = portid;
1998 	args.clid = clid;
1999 	args.new_cl = new_cl;
2000 	args.w.fn = tc_bind_class_walker;
2001 	q->ops->cl_ops->walk(q, &args.w);
2002 }
2003 
2004 #else
2005 
tc_bind_tclass(struct Qdisc * q,u32 portid,u32 clid,unsigned long new_cl)2006 static void tc_bind_tclass(struct Qdisc *q, u32 portid, u32 clid,
2007 			   unsigned long new_cl)
2008 {
2009 }
2010 
2011 #endif
2012 
tc_ctl_tclass(struct sk_buff * skb,struct nlmsghdr * n,struct netlink_ext_ack * extack)2013 static int tc_ctl_tclass(struct sk_buff *skb, struct nlmsghdr *n,
2014 			 struct netlink_ext_ack *extack)
2015 {
2016 	struct net *net = sock_net(skb->sk);
2017 	struct tcmsg *tcm = nlmsg_data(n);
2018 	struct nlattr *tca[TCA_MAX + 1];
2019 	struct net_device *dev;
2020 	struct Qdisc *q = NULL;
2021 	const struct Qdisc_class_ops *cops;
2022 	unsigned long cl = 0;
2023 	unsigned long new_cl;
2024 	u32 portid;
2025 	u32 clid;
2026 	u32 qid;
2027 	int err;
2028 
2029 	if ((n->nlmsg_type != RTM_GETTCLASS) &&
2030 	    !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
2031 		return -EPERM;
2032 
2033 	err = nlmsg_parse_deprecated(n, sizeof(*tcm), tca, TCA_MAX,
2034 				     rtm_tca_policy, extack);
2035 	if (err < 0)
2036 		return err;
2037 
2038 	dev = __dev_get_by_index(net, tcm->tcm_ifindex);
2039 	if (!dev)
2040 		return -ENODEV;
2041 
2042 	/*
2043 	   parent == TC_H_UNSPEC - unspecified parent.
2044 	   parent == TC_H_ROOT   - class is root, which has no parent.
2045 	   parent == X:0	 - parent is root class.
2046 	   parent == X:Y	 - parent is a node in hierarchy.
2047 	   parent == 0:Y	 - parent is X:Y, where X:0 is qdisc.
2048 
2049 	   handle == 0:0	 - generate handle from kernel pool.
2050 	   handle == 0:Y	 - class is X:Y, where X:0 is qdisc.
2051 	   handle == X:Y	 - clear.
2052 	   handle == X:0	 - root class.
2053 	 */
2054 
2055 	/* Step 1. Determine qdisc handle X:0 */
2056 
2057 	portid = tcm->tcm_parent;
2058 	clid = tcm->tcm_handle;
2059 	qid = TC_H_MAJ(clid);
2060 
2061 	if (portid != TC_H_ROOT) {
2062 		u32 qid1 = TC_H_MAJ(portid);
2063 
2064 		if (qid && qid1) {
2065 			/* If both majors are known, they must be identical. */
2066 			if (qid != qid1)
2067 				return -EINVAL;
2068 		} else if (qid1) {
2069 			qid = qid1;
2070 		} else if (qid == 0)
2071 			qid = rtnl_dereference(dev->qdisc)->handle;
2072 
2073 		/* Now qid is genuine qdisc handle consistent
2074 		 * both with parent and child.
2075 		 *
2076 		 * TC_H_MAJ(portid) still may be unspecified, complete it now.
2077 		 */
2078 		if (portid)
2079 			portid = TC_H_MAKE(qid, portid);
2080 	} else {
2081 		if (qid == 0)
2082 			qid = rtnl_dereference(dev->qdisc)->handle;
2083 	}
2084 
2085 	/* OK. Locate qdisc */
2086 	q = qdisc_lookup(dev, qid);
2087 	if (!q)
2088 		return -ENOENT;
2089 
2090 	/* An check that it supports classes */
2091 	cops = q->ops->cl_ops;
2092 	if (cops == NULL)
2093 		return -EINVAL;
2094 
2095 	/* Now try to get class */
2096 	if (clid == 0) {
2097 		if (portid == TC_H_ROOT)
2098 			clid = qid;
2099 	} else
2100 		clid = TC_H_MAKE(qid, clid);
2101 
2102 	if (clid)
2103 		cl = cops->find(q, clid);
2104 
2105 	if (cl == 0) {
2106 		err = -ENOENT;
2107 		if (n->nlmsg_type != RTM_NEWTCLASS ||
2108 		    !(n->nlmsg_flags & NLM_F_CREATE))
2109 			goto out;
2110 	} else {
2111 		switch (n->nlmsg_type) {
2112 		case RTM_NEWTCLASS:
2113 			err = -EEXIST;
2114 			if (n->nlmsg_flags & NLM_F_EXCL)
2115 				goto out;
2116 			break;
2117 		case RTM_DELTCLASS:
2118 			err = tclass_del_notify(net, cops, skb, n, q, cl);
2119 			/* Unbind the class with flilters with 0 */
2120 			tc_bind_tclass(q, portid, clid, 0);
2121 			goto out;
2122 		case RTM_GETTCLASS:
2123 			err = tclass_notify(net, skb, n, q, cl, RTM_NEWTCLASS);
2124 			goto out;
2125 		default:
2126 			err = -EINVAL;
2127 			goto out;
2128 		}
2129 	}
2130 
2131 	if (tca[TCA_INGRESS_BLOCK] || tca[TCA_EGRESS_BLOCK]) {
2132 		NL_SET_ERR_MSG(extack, "Shared blocks are not supported for classes");
2133 		return -EOPNOTSUPP;
2134 	}
2135 
2136 	new_cl = cl;
2137 	err = -EOPNOTSUPP;
2138 	if (cops->change)
2139 		err = cops->change(q, clid, portid, tca, &new_cl, extack);
2140 	if (err == 0) {
2141 		tclass_notify(net, skb, n, q, new_cl, RTM_NEWTCLASS);
2142 		/* We just create a new class, need to do reverse binding. */
2143 		if (cl != new_cl)
2144 			tc_bind_tclass(q, portid, clid, new_cl);
2145 	}
2146 out:
2147 	return err;
2148 }
2149 
2150 struct qdisc_dump_args {
2151 	struct qdisc_walker	w;
2152 	struct sk_buff		*skb;
2153 	struct netlink_callback	*cb;
2154 };
2155 
qdisc_class_dump(struct Qdisc * q,unsigned long cl,struct qdisc_walker * arg)2156 static int qdisc_class_dump(struct Qdisc *q, unsigned long cl,
2157 			    struct qdisc_walker *arg)
2158 {
2159 	struct qdisc_dump_args *a = (struct qdisc_dump_args *)arg;
2160 
2161 	return tc_fill_tclass(a->skb, q, cl, NETLINK_CB(a->cb->skb).portid,
2162 			      a->cb->nlh->nlmsg_seq, NLM_F_MULTI,
2163 			      RTM_NEWTCLASS);
2164 }
2165 
tc_dump_tclass_qdisc(struct Qdisc * q,struct sk_buff * skb,struct tcmsg * tcm,struct netlink_callback * cb,int * t_p,int s_t)2166 static int tc_dump_tclass_qdisc(struct Qdisc *q, struct sk_buff *skb,
2167 				struct tcmsg *tcm, struct netlink_callback *cb,
2168 				int *t_p, int s_t)
2169 {
2170 	struct qdisc_dump_args arg;
2171 
2172 	if (tc_qdisc_dump_ignore(q, false) ||
2173 	    *t_p < s_t || !q->ops->cl_ops ||
2174 	    (tcm->tcm_parent &&
2175 	     TC_H_MAJ(tcm->tcm_parent) != q->handle)) {
2176 		(*t_p)++;
2177 		return 0;
2178 	}
2179 	if (*t_p > s_t)
2180 		memset(&cb->args[1], 0, sizeof(cb->args)-sizeof(cb->args[0]));
2181 	arg.w.fn = qdisc_class_dump;
2182 	arg.skb = skb;
2183 	arg.cb = cb;
2184 	arg.w.stop  = 0;
2185 	arg.w.skip = cb->args[1];
2186 	arg.w.count = 0;
2187 	q->ops->cl_ops->walk(q, &arg.w);
2188 	cb->args[1] = arg.w.count;
2189 	if (arg.w.stop)
2190 		return -1;
2191 	(*t_p)++;
2192 	return 0;
2193 }
2194 
tc_dump_tclass_root(struct Qdisc * root,struct sk_buff * skb,struct tcmsg * tcm,struct netlink_callback * cb,int * t_p,int s_t,bool recur)2195 static int tc_dump_tclass_root(struct Qdisc *root, struct sk_buff *skb,
2196 			       struct tcmsg *tcm, struct netlink_callback *cb,
2197 			       int *t_p, int s_t, bool recur)
2198 {
2199 	struct Qdisc *q;
2200 	int b;
2201 
2202 	if (!root)
2203 		return 0;
2204 
2205 	if (tc_dump_tclass_qdisc(root, skb, tcm, cb, t_p, s_t) < 0)
2206 		return -1;
2207 
2208 	if (!qdisc_dev(root) || !recur)
2209 		return 0;
2210 
2211 	if (tcm->tcm_parent) {
2212 		q = qdisc_match_from_root(root, TC_H_MAJ(tcm->tcm_parent));
2213 		if (q && q != root &&
2214 		    tc_dump_tclass_qdisc(q, skb, tcm, cb, t_p, s_t) < 0)
2215 			return -1;
2216 		return 0;
2217 	}
2218 	hash_for_each(qdisc_dev(root)->qdisc_hash, b, q, hash) {
2219 		if (tc_dump_tclass_qdisc(q, skb, tcm, cb, t_p, s_t) < 0)
2220 			return -1;
2221 	}
2222 
2223 	return 0;
2224 }
2225 
tc_dump_tclass(struct sk_buff * skb,struct netlink_callback * cb)2226 static int tc_dump_tclass(struct sk_buff *skb, struct netlink_callback *cb)
2227 {
2228 	struct tcmsg *tcm = nlmsg_data(cb->nlh);
2229 	struct net *net = sock_net(skb->sk);
2230 	struct netdev_queue *dev_queue;
2231 	struct net_device *dev;
2232 	int t, s_t;
2233 
2234 	if (nlmsg_len(cb->nlh) < sizeof(*tcm))
2235 		return 0;
2236 	dev = dev_get_by_index(net, tcm->tcm_ifindex);
2237 	if (!dev)
2238 		return 0;
2239 
2240 	s_t = cb->args[0];
2241 	t = 0;
2242 
2243 	if (tc_dump_tclass_root(rtnl_dereference(dev->qdisc),
2244 				skb, tcm, cb, &t, s_t, true) < 0)
2245 		goto done;
2246 
2247 	dev_queue = dev_ingress_queue(dev);
2248 	if (dev_queue &&
2249 	    tc_dump_tclass_root(dev_queue->qdisc_sleeping, skb, tcm, cb,
2250 				&t, s_t, false) < 0)
2251 		goto done;
2252 
2253 done:
2254 	cb->args[0] = t;
2255 
2256 	dev_put(dev);
2257 	return skb->len;
2258 }
2259 
2260 #ifdef CONFIG_PROC_FS
psched_show(struct seq_file * seq,void * v)2261 static int psched_show(struct seq_file *seq, void *v)
2262 {
2263 	seq_printf(seq, "%08x %08x %08x %08x\n",
2264 		   (u32)NSEC_PER_USEC, (u32)PSCHED_TICKS2NS(1),
2265 		   1000000,
2266 		   (u32)NSEC_PER_SEC / hrtimer_resolution);
2267 
2268 	return 0;
2269 }
2270 
psched_net_init(struct net * net)2271 static int __net_init psched_net_init(struct net *net)
2272 {
2273 	struct proc_dir_entry *e;
2274 
2275 	e = proc_create_single("psched", 0, net->proc_net, psched_show);
2276 	if (e == NULL)
2277 		return -ENOMEM;
2278 
2279 	return 0;
2280 }
2281 
psched_net_exit(struct net * net)2282 static void __net_exit psched_net_exit(struct net *net)
2283 {
2284 	remove_proc_entry("psched", net->proc_net);
2285 }
2286 #else
psched_net_init(struct net * net)2287 static int __net_init psched_net_init(struct net *net)
2288 {
2289 	return 0;
2290 }
2291 
psched_net_exit(struct net * net)2292 static void __net_exit psched_net_exit(struct net *net)
2293 {
2294 }
2295 #endif
2296 
2297 static struct pernet_operations psched_net_ops = {
2298 	.init = psched_net_init,
2299 	.exit = psched_net_exit,
2300 };
2301 
pktsched_init(void)2302 static int __init pktsched_init(void)
2303 {
2304 	int err;
2305 
2306 	err = register_pernet_subsys(&psched_net_ops);
2307 	if (err) {
2308 		pr_err("pktsched_init: "
2309 		       "cannot initialize per netns operations\n");
2310 		return err;
2311 	}
2312 
2313 	register_qdisc(&pfifo_fast_ops);
2314 	register_qdisc(&pfifo_qdisc_ops);
2315 	register_qdisc(&bfifo_qdisc_ops);
2316 	register_qdisc(&pfifo_head_drop_qdisc_ops);
2317 	register_qdisc(&mq_qdisc_ops);
2318 	register_qdisc(&noqueue_qdisc_ops);
2319 
2320 	rtnl_register(PF_UNSPEC, RTM_NEWQDISC, tc_modify_qdisc, NULL, 0);
2321 	rtnl_register(PF_UNSPEC, RTM_DELQDISC, tc_get_qdisc, NULL, 0);
2322 	rtnl_register(PF_UNSPEC, RTM_GETQDISC, tc_get_qdisc, tc_dump_qdisc,
2323 		      0);
2324 	rtnl_register(PF_UNSPEC, RTM_NEWTCLASS, tc_ctl_tclass, NULL, 0);
2325 	rtnl_register(PF_UNSPEC, RTM_DELTCLASS, tc_ctl_tclass, NULL, 0);
2326 	rtnl_register(PF_UNSPEC, RTM_GETTCLASS, tc_ctl_tclass, tc_dump_tclass,
2327 		      0);
2328 
2329 	return 0;
2330 }
2331 
2332 subsys_initcall(pktsched_init);
2333