[mirror_ubuntu-jammy-kernel.git] / net / sched / sch_multiq.c

// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (c) 2008, Intel Corporation.
 *
 * Author: Alexander Duyck <alexander.h.duyck@intel.com>
 */

#include <linux/module.h>
#include <linux/slab.h>
#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/string.h>
#include <linux/errno.h>
#include <linux/skbuff.h>
#include <net/netlink.h>
#include <net/pkt_sched.h>
#include <net/pkt_cls.h>

struct multiq_sched_data {
	u16 bands;
	u16 max_bands;
	u16 curband;
	struct tcf_proto __rcu *filter_list;
	struct tcf_block *block;
	struct Qdisc **queues;
};


static struct Qdisc *
multiq_classify(struct sk_buff *skb, struct Qdisc *sch, int *qerr)
{
	struct multiq_sched_data *q = qdisc_priv(sch);
	u32 band;
	struct tcf_result res;
	struct tcf_proto *fl = rcu_dereference_bh(q->filter_list);
	int err;

	*qerr = NET_XMIT_SUCCESS | __NET_XMIT_BYPASS;
	err = tcf_classify(skb, fl, &res, false);
#ifdef CONFIG_NET_CLS_ACT
	switch (err) {
	case TC_ACT_STOLEN:
	case TC_ACT_QUEUED:
	case TC_ACT_TRAP:
		*qerr = NET_XMIT_SUCCESS | __NET_XMIT_STOLEN;
		fallthrough;
	case TC_ACT_SHOT:
		return NULL;
	}
#endif
	band = skb_get_queue_mapping(skb);

	if (band >= q->bands)
		return q->queues[0];

	return q->queues[band];
}

static int
multiq_enqueue(struct sk_buff *skb, struct Qdisc *sch,
	       struct sk_buff **to_free)
{
	struct Qdisc *qdisc;
	int ret;

	qdisc = multiq_classify(skb, sch, &ret);
#ifdef CONFIG_NET_CLS_ACT
	if (qdisc == NULL) {

		if (ret & __NET_XMIT_BYPASS)
			qdisc_qstats_drop(sch);
		__qdisc_drop(skb, to_free);
		return ret;
	}
#endif

	ret = qdisc_enqueue(skb, qdisc, to_free);
	if (ret == NET_XMIT_SUCCESS) {
		sch->q.qlen++;
		return NET_XMIT_SUCCESS;
	}
	if (net_xmit_drop_count(ret))
		qdisc_qstats_drop(sch);
	return ret;
}

static struct sk_buff *multiq_dequeue(struct Qdisc *sch)
{
	struct multiq_sched_data *q = qdisc_priv(sch);
	struct Qdisc *qdisc;
	struct sk_buff *skb;
	int band;

	for (band = 0; band < q->bands; band++) {
		/* cycle through bands to ensure fairness */
		q->curband++;
		if (q->curband >= q->bands)
			q->curband = 0;

		/* Check that target subqueue is available before
		 * pulling an skb to avoid head-of-line blocking.
		 */
		if (!netif_xmit_stopped(
		    netdev_get_tx_queue(qdisc_dev(sch), q->curband))) {
			qdisc = q->queues[q->curband];
			skb = qdisc->dequeue(qdisc);
			if (skb) {
				qdisc_bstats_update(sch, skb);
				sch->q.qlen--;
				return skb;
			}
		}
	}
	return NULL;

}

static struct sk_buff *multiq_peek(struct Qdisc *sch)
{
	struct multiq_sched_data *q = qdisc_priv(sch);
	unsigned int curband = q->curband;
	struct Qdisc *qdisc;
	struct sk_buff *skb;
	int band;

	for (band = 0; band < q->bands; band++) {
		/* cycle through bands to ensure fairness */
		curband++;
		if (curband >= q->bands)
			curband = 0;

		/* Check that target subqueue is available before
		 * pulling an skb to avoid head-of-line blocking.
		 */
		if (!netif_xmit_stopped(
		    netdev_get_tx_queue(qdisc_dev(sch), curband))) {
			qdisc = q->queues[curband];
			skb = qdisc->ops->peek(qdisc);
			if (skb)
				return skb;
		}
	}
	return NULL;

}

static void
multiq_reset(struct Qdisc *sch)
{
	u16 band;
	struct multiq_sched_data *q = qdisc_priv(sch);

	for (band = 0; band < q->bands; band++)
		qdisc_reset(q->queues[band]);
	sch->q.qlen = 0;
	q->curband = 0;
}

static void
multiq_destroy(struct Qdisc *sch)
{
	int band;
	struct multiq_sched_data *q = qdisc_priv(sch);

	tcf_block_put(q->block);
	for (band = 0; band < q->bands; band++)
		qdisc_put(q->queues[band]);

	kfree(q->queues);
}

static int multiq_tune(struct Qdisc *sch, struct nlattr *opt,
		       struct netlink_ext_ack *extack)
{
	struct multiq_sched_data *q = qdisc_priv(sch);
	struct tc_multiq_qopt *qopt;
	struct Qdisc **removed;
	int i, n_removed = 0;

	if (!netif_is_multiqueue(qdisc_dev(sch)))
		return -EOPNOTSUPP;
	if (nla_len(opt) < sizeof(*qopt))
		return -EINVAL;

	qopt = nla_data(opt);

	qopt->bands = qdisc_dev(sch)->real_num_tx_queues;

	removed = kmalloc(sizeof(*removed) * (q->max_bands - q->bands),
			  GFP_KERNEL);
	if (!removed)
		return -ENOMEM;

	sch_tree_lock(sch);
	q->bands = qopt->bands;
	for (i = q->bands; i < q->max_bands; i++) {
		if (q->queues[i] != &noop_qdisc) {
			struct Qdisc *child = q->queues[i];

			q->queues[i] = &noop_qdisc;
			qdisc_purge_queue(child);
			removed[n_removed++] = child;
		}
	}

	sch_tree_unlock(sch);

	for (i = 0; i < n_removed; i++)
		qdisc_put(removed[i]);
	kfree(removed);

	for (i = 0; i < q->bands; i++) {
		if (q->queues[i] == &noop_qdisc) {
			struct Qdisc *child, *old;
			child = qdisc_create_dflt(sch->dev_queue,
						  &pfifo_qdisc_ops,
						  TC_H_MAKE(sch->handle,
							    i + 1), extack);
			if (child) {
				sch_tree_lock(sch);
				old = q->queues[i];
				q->queues[i] = child;
				if (child != &noop_qdisc)
					qdisc_hash_add(child, true);

				if (old != &noop_qdisc)
					qdisc_purge_queue(old);
				sch_tree_unlock(sch);
				qdisc_put(old);
			}
		}
	}
	return 0;
}

static int multiq_init(struct Qdisc *sch, struct nlattr *opt,
		       struct netlink_ext_ack *extack)
{
	struct multiq_sched_data *q = qdisc_priv(sch);
	int i, err;

	q->queues = NULL;

	if (!opt)
		return -EINVAL;

	err = tcf_block_get(&q->block, &q->filter_list, sch, extack);
	if (err)
		return err;

	q->max_bands = qdisc_dev(sch)->num_tx_queues;

	q->queues = kcalloc(q->max_bands, sizeof(struct Qdisc *), GFP_KERNEL);
	if (!q->queues)
		return -ENOBUFS;
	for (i = 0; i < q->max_bands; i++)
		q->queues[i] = &noop_qdisc;

	return multiq_tune(sch, opt, extack);
}

static int multiq_dump(struct Qdisc *sch, struct sk_buff *skb)
{
	struct multiq_sched_data *q = qdisc_priv(sch);
	unsigned char *b = skb_tail_pointer(skb);
	struct tc_multiq_qopt opt;

	opt.bands = q->bands;
	opt.max_bands = q->max_bands;

	if (nla_put(skb, TCA_OPTIONS, sizeof(opt), &opt))
		goto nla_put_failure;

	return skb->len;

nla_put_failure:
	nlmsg_trim(skb, b);
	return -1;
}

static int multiq_graft(struct Qdisc *sch, unsigned long arg, struct Qdisc *new,
			struct Qdisc **old, struct netlink_ext_ack *extack)
{
	struct multiq_sched_data *q = qdisc_priv(sch);
	unsigned long band = arg - 1;

	if (new == NULL)
		new = &noop_qdisc;

	*old = qdisc_replace(sch, new, &q->queues[band]);
	return 0;
}

static struct Qdisc *
multiq_leaf(struct Qdisc *sch, unsigned long arg)
{
	struct multiq_sched_data *q = qdisc_priv(sch);
	unsigned long band = arg - 1;

	return q->queues[band];
}

static unsigned long multiq_find(struct Qdisc *sch, u32 classid)
{
	struct multiq_sched_data *q = qdisc_priv(sch);
	unsigned long band = TC_H_MIN(classid);

	if (band - 1 >= q->bands)
		return 0;
	return band;
}

static unsigned long multiq_bind(struct Qdisc *sch, unsigned long parent,
				 u32 classid)
{
	return multiq_find(sch, classid);
}


static void multiq_unbind(struct Qdisc *q, unsigned long cl)
{
}

static int multiq_dump_class(struct Qdisc *sch, unsigned long cl,
			     struct sk_buff *skb, struct tcmsg *tcm)
{
	struct multiq_sched_data *q = qdisc_priv(sch);

	tcm->tcm_handle |= TC_H_MIN(cl);
	tcm->tcm_info = q->queues[cl - 1]->handle;
	return 0;
}

static int multiq_dump_class_stats(struct Qdisc *sch, unsigned long cl,
				 struct gnet_dump *d)
{
	struct multiq_sched_data *q = qdisc_priv(sch);
	struct Qdisc *cl_q;

	cl_q = q->queues[cl - 1];
	if (gnet_stats_copy_basic(qdisc_root_sleeping_running(sch),
				  d, cl_q->cpu_bstats, &cl_q->bstats) < 0 ||
	    qdisc_qstats_copy(d, cl_q) < 0)
		return -1;

	return 0;
}

static void multiq_walk(struct Qdisc *sch, struct qdisc_walker *arg)
{
	struct multiq_sched_data *q = qdisc_priv(sch);
	int band;

	if (arg->stop)
		return;

	for (band = 0; band < q->bands; band++) {
		if (arg->count < arg->skip) {
			arg->count++;
			continue;
		}
		if (arg->fn(sch, band + 1, arg) < 0) {
			arg->stop = 1;
			break;
		}
		arg->count++;
	}
}

static struct tcf_block *multiq_tcf_block(struct Qdisc *sch, unsigned long cl,
					  struct netlink_ext_ack *extack)
{
	struct multiq_sched_data *q = qdisc_priv(sch);

	if (cl)
		return NULL;
	return q->block;
}

static const struct Qdisc_class_ops multiq_class_ops = {
	.graft		=	multiq_graft,
	.leaf		=	multiq_leaf,
	.find		=	multiq_find,
	.walk		=	multiq_walk,
	.tcf_block	=	multiq_tcf_block,
	.bind_tcf	=	multiq_bind,
	.unbind_tcf	=	multiq_unbind,
	.dump		=	multiq_dump_class,
	.dump_stats	=	multiq_dump_class_stats,
};

static struct Qdisc_ops multiq_qdisc_ops __read_mostly = {
	.next		=	NULL,
	.cl_ops		=	&multiq_class_ops,
	.id		=	"multiq",
	.priv_size	=	sizeof(struct multiq_sched_data),
	.enqueue	=	multiq_enqueue,
	.dequeue	=	multiq_dequeue,
	.peek		=	multiq_peek,
	.init		=	multiq_init,
	.reset		=	multiq_reset,
	.destroy	=	multiq_destroy,
	.change		=	multiq_tune,
	.dump		=	multiq_dump,
	.owner		=	THIS_MODULE,
};

static int __init multiq_module_init(void)
{
	return register_qdisc(&multiq_qdisc_ops);
}

static void __exit multiq_module_exit(void)
{
	unregister_qdisc(&multiq_qdisc_ops);
}

module_init(multiq_module_init)
module_exit(multiq_module_exit)

MODULE_LICENSE("GPL");
Commit	Line	Data
9952f691	1	// SPDX-License-Identifier: GPL-2.0-only
92651940 AD	2	/*
	3	* Copyright (c) 2008, Intel Corporation.
	4	*
92651940 AD	5	* Author: Alexander Duyck <alexander.h.duyck@intel.com>
	6	*/
	7
	8	#include <linux/module.h>
5a0e3ad6	9	#include <linux/slab.h>
92651940 AD	10	#include <linux/types.h>
	11	#include <linux/kernel.h>
	12	#include <linux/string.h>
	13	#include <linux/errno.h>
	14	#include <linux/skbuff.h>
	15	#include <net/netlink.h>
	16	#include <net/pkt_sched.h>
cf1facda	17	#include <net/pkt_cls.h>
92651940 AD	18
	19	struct multiq_sched_data {
	20	u16 bands;
	21	u16 max_bands;
	22	u16 curband;
25d8c0d5	23	struct tcf_proto __rcu *filter_list;
6529eaba	24	struct tcf_block *block;
92651940 AD	25	struct Qdisc **queues;
	26	};
	27
	28
	29	static struct Qdisc *
	30	multiq_classify(struct sk_buff skb, struct Qdisc sch, int *qerr)
	31	{
	32	struct multiq_sched_data *q = qdisc_priv(sch);
	33	u32 band;
	34	struct tcf_result res;
25d8c0d5	35	struct tcf_proto *fl = rcu_dereference_bh(q->filter_list);
92651940 AD	36	int err;
	37
	38	*qerr = NET_XMIT_SUCCESS \| __NET_XMIT_BYPASS;
87d83093	39	err = tcf_classify(skb, fl, &res, false);
92651940 AD	40	#ifdef CONFIG_NET_CLS_ACT
	41	switch (err) {
	42	case TC_ACT_STOLEN:
	43	case TC_ACT_QUEUED:
e25ea21f	44	case TC_ACT_TRAP:
92651940	45	*qerr = NET_XMIT_SUCCESS \| __NET_XMIT_STOLEN;
964201de	46	fallthrough;
92651940 AD	47	case TC_ACT_SHOT:
	48	return NULL;
	49	}
	50	#endif
	51	band = skb_get_queue_mapping(skb);
	52
	53	if (band >= q->bands)
	54	return q->queues[0];
	55
	56	return q->queues[band];
	57	}
	58
	59	static int
ac5c66f2	60	multiq_enqueue(struct sk_buff skb, struct Qdisc sch,
520ac30f	61	struct sk_buff **to_free)
92651940 AD	62	{
	63	struct Qdisc *qdisc;
	64	int ret;
	65
	66	qdisc = multiq_classify(skb, sch, &ret);
	67	#ifdef CONFIG_NET_CLS_ACT
	68	if (qdisc == NULL) {
	69
	70	if (ret & __NET_XMIT_BYPASS)
25331d6c	71	qdisc_qstats_drop(sch);
520ac30f	72	__qdisc_drop(skb, to_free);
92651940 AD	73	return ret;
	74	}
	75	#endif
	76
ac5c66f2	77	ret = qdisc_enqueue(skb, qdisc, to_free);
92651940	78	if (ret == NET_XMIT_SUCCESS) {
92651940 AD	79	sch->q.qlen++;
	80	return NET_XMIT_SUCCESS;
	81	}
	82	if (net_xmit_drop_count(ret))
25331d6c	83	qdisc_qstats_drop(sch);
92651940 AD	84	return ret;
	85	}
	86
92651940 AD	87	static struct sk_buff multiq_dequeue(struct Qdisc sch)
	88	{
	89	struct multiq_sched_data *q = qdisc_priv(sch);
	90	struct Qdisc *qdisc;
	91	struct sk_buff *skb;
	92	int band;
	93
	94	for (band = 0; band < q->bands; band++) {
	95	/* cycle through bands to ensure fairness */
	96	q->curband++;
	97	if (q->curband >= q->bands)
	98	q->curband = 0;
	99
	100	/* Check that target subqueue is available before
f30ab418	101	* pulling an skb to avoid head-of-line blocking.
92651940	102	*/
73466498 TH	103	if (!netif_xmit_stopped(
73466498 TH	104	netdev_get_tx_queue(qdisc_dev(sch), q->curband))) {
92651940 AD	105	qdisc = q->queues[q->curband];
	106	skb = qdisc->dequeue(qdisc);
	107	if (skb) {
9190b3b3	108	qdisc_bstats_update(sch, skb);
92651940 AD	109	sch->q.qlen--;
	110	return skb;
	111	}
	112	}
	113	}
	114	return NULL;
	115
	116	}
	117
8e3af978 JP	118	static struct sk_buff multiq_peek(struct Qdisc sch)
	119	{
	120	struct multiq_sched_data *q = qdisc_priv(sch);
	121	unsigned int curband = q->curband;
	122	struct Qdisc *qdisc;
	123	struct sk_buff *skb;
	124	int band;
	125
	126	for (band = 0; band < q->bands; band++) {
	127	/* cycle through bands to ensure fairness */
	128	curband++;
	129	if (curband >= q->bands)
	130	curband = 0;
	131
	132	/* Check that target subqueue is available before
f30ab418	133	* pulling an skb to avoid head-of-line blocking.
8e3af978	134	*/
73466498 TH	135	if (!netif_xmit_stopped(
73466498 TH	136	netdev_get_tx_queue(qdisc_dev(sch), curband))) {
8e3af978 JP	137	qdisc = q->queues[curband];
	138	skb = qdisc->ops->peek(qdisc);
	139	if (skb)
	140	return skb;
	141	}
	142	}
	143	return NULL;
	144
	145	}
	146
92651940 AD	147	static void
	148	multiq_reset(struct Qdisc *sch)
	149	{
	150	u16 band;
	151	struct multiq_sched_data *q = qdisc_priv(sch);
	152
	153	for (band = 0; band < q->bands; band++)
	154	qdisc_reset(q->queues[band]);
	155	sch->q.qlen = 0;
	156	q->curband = 0;
	157	}
	158
	159	static void
	160	multiq_destroy(struct Qdisc *sch)
	161	{
	162	int band;
	163	struct multiq_sched_data *q = qdisc_priv(sch);
	164
6529eaba	165	tcf_block_put(q->block);
92651940	166	for (band = 0; band < q->bands; band++)
86bd446b	167	qdisc_put(q->queues[band]);
92651940 AD	168
	169	kfree(q->queues);
	170	}
	171
2030721c AA	172	static int multiq_tune(struct Qdisc sch, struct nlattr opt,
2030721c AA	173	struct netlink_ext_ack *extack)
92651940 AD	174	{
	175	struct multiq_sched_data *q = qdisc_priv(sch);
	176	struct tc_multiq_qopt *qopt;
c2999f7f VB	177	struct Qdisc **removed;
c2999f7f VB	178	int i, n_removed = 0;
92651940 AD	179
92651940 AD	180	if (!netif_is_multiqueue(qdisc_dev(sch)))
149490f1	181	return -EOPNOTSUPP;
92651940 AD	182	if (nla_len(opt) < sizeof(*qopt))
	183	return -EINVAL;
	184
	185	qopt = nla_data(opt);
	186
	187	qopt->bands = qdisc_dev(sch)->real_num_tx_queues;
	188
c2999f7f VB	189	removed = kmalloc(sizeof(removed) (q->max_bands - q->bands),
	190	GFP_KERNEL);
	191	if (!removed)
	192	return -ENOMEM;
	193
92651940 AD	194	sch_tree_lock(sch);
	195	q->bands = qopt->bands;
	196	for (i = q->bands; i < q->max_bands; i++) {
f07d1501	197	if (q->queues[i] != &noop_qdisc) {
b94c8afc	198	struct Qdisc *child = q->queues[i];
e5f0e8f8	199
b94c8afc	200	q->queues[i] = &noop_qdisc;
c2999f7f VB	201	qdisc_purge_queue(child);
c2999f7f VB	202	removed[n_removed++] = child;
92651940 AD	203	}
	204	}
	205
	206	sch_tree_unlock(sch);
	207
c2999f7f VB	208	for (i = 0; i < n_removed; i++)
	209	qdisc_put(removed[i]);
	210	kfree(removed);
	211
92651940 AD	212	for (i = 0; i < q->bands; i++) {
92651940 AD	213	if (q->queues[i] == &noop_qdisc) {
b94c8afc	214	struct Qdisc child, old;
3511c913	215	child = qdisc_create_dflt(sch->dev_queue,
92651940 AD	216	&pfifo_qdisc_ops,
92651940 AD	217	TC_H_MAKE(sch->handle,
a38a9882	218	i + 1), extack);
92651940 AD	219	if (child) {
92651940 AD	220	sch_tree_lock(sch);
b94c8afc PM	221	old = q->queues[i];
b94c8afc PM	222	q->queues[i] = child;
49b49971 JK	223	if (child != &noop_qdisc)
49b49971 JK	224	qdisc_hash_add(child, true);
92651940	225
c2999f7f VB	226	if (old != &noop_qdisc)
c2999f7f VB	227	qdisc_purge_queue(old);
92651940	228	sch_tree_unlock(sch);
c2999f7f	229	qdisc_put(old);
92651940 AD	230	}
	231	}
	232	}
	233	return 0;
	234	}
	235
e63d7dfd AA	236	static int multiq_init(struct Qdisc sch, struct nlattr opt,
e63d7dfd AA	237	struct netlink_ext_ack *extack)
92651940 AD	238	{
92651940 AD	239	struct multiq_sched_data *q = qdisc_priv(sch);
f07d1501	240	int i, err;
92651940 AD	241
	242	q->queues = NULL;
	243
ac8ef4ab	244	if (!opt)
92651940 AD	245	return -EINVAL;
92651940 AD	246
8d1a77f9	247	err = tcf_block_get(&q->block, &q->filter_list, sch, extack);
6529eaba JP	248	if (err)
	249	return err;
	250
92651940 AD	251	q->max_bands = qdisc_dev(sch)->num_tx_queues;
	252
	253	q->queues = kcalloc(q->max_bands, sizeof(struct Qdisc *), GFP_KERNEL);
	254	if (!q->queues)
	255	return -ENOBUFS;
	256	for (i = 0; i < q->max_bands; i++)
	257	q->queues[i] = &noop_qdisc;
	258
2030721c	259	return multiq_tune(sch, opt, extack);
92651940 AD	260	}
	261
	262	static int multiq_dump(struct Qdisc sch, struct sk_buff skb)
	263	{
	264	struct multiq_sched_data *q = qdisc_priv(sch);
	265	unsigned char *b = skb_tail_pointer(skb);
	266	struct tc_multiq_qopt opt;
	267
	268	opt.bands = q->bands;
	269	opt.max_bands = q->max_bands;
	270
1b34ec43 DM	271	if (nla_put(skb, TCA_OPTIONS, sizeof(opt), &opt))
1b34ec43 DM	272	goto nla_put_failure;
92651940 AD	273
	274	return skb->len;
	275
	276	nla_put_failure:
	277	nlmsg_trim(skb, b);
	278	return -1;
	279	}
	280
	281	static int multiq_graft(struct Qdisc sch, unsigned long arg, struct Qdisc new,
653d6fd6	282	struct Qdisc *old, struct netlink_ext_ack extack)
92651940 AD	283	{
	284	struct multiq_sched_data *q = qdisc_priv(sch);
	285	unsigned long band = arg - 1;
	286
92651940 AD	287	if (new == NULL)
	288	new = &noop_qdisc;
	289
86a7996c	290	*old = qdisc_replace(sch, new, &q->queues[band]);
92651940 AD	291	return 0;
	292	}
	293
	294	static struct Qdisc *
	295	multiq_leaf(struct Qdisc *sch, unsigned long arg)
	296	{
	297	struct multiq_sched_data *q = qdisc_priv(sch);
	298	unsigned long band = arg - 1;
	299
92651940 AD	300	return q->queues[band];
	301	}
	302
143976ce	303	static unsigned long multiq_find(struct Qdisc *sch, u32 classid)
92651940 AD	304	{
	305	struct multiq_sched_data *q = qdisc_priv(sch);
	306	unsigned long band = TC_H_MIN(classid);
	307
	308	if (band - 1 >= q->bands)
	309	return 0;
	310	return band;
	311	}
	312
	313	static unsigned long multiq_bind(struct Qdisc *sch, unsigned long parent,
	314	u32 classid)
	315	{
143976ce	316	return multiq_find(sch, classid);
92651940 AD	317	}
	318
	319
143976ce	320	static void multiq_unbind(struct Qdisc *q, unsigned long cl)
92651940	321	{
92651940 AD	322	}
92651940 AD	323
92651940 AD	324	static int multiq_dump_class(struct Qdisc *sch, unsigned long cl,
	325	struct sk_buff skb, struct tcmsg tcm)
	326	{
	327	struct multiq_sched_data *q = qdisc_priv(sch);
	328
92651940	329	tcm->tcm_handle \|= TC_H_MIN(cl);
cc7ec456	330	tcm->tcm_info = q->queues[cl - 1]->handle;
92651940 AD	331	return 0;
	332	}
	333
	334	static int multiq_dump_class_stats(struct Qdisc *sch, unsigned long cl,
	335	struct gnet_dump *d)
	336	{
	337	struct multiq_sched_data *q = qdisc_priv(sch);
	338	struct Qdisc *cl_q;
	339
	340	cl_q = q->queues[cl - 1];
edb09eb1	341	if (gnet_stats_copy_basic(qdisc_root_sleeping_running(sch),
14e54ab9	342	d, cl_q->cpu_bstats, &cl_q->bstats) < 0 \|\|
5dd431b6	343	qdisc_qstats_copy(d, cl_q) < 0)
92651940 AD	344	return -1;
	345
	346	return 0;
	347	}
	348
	349	static void multiq_walk(struct Qdisc sch, struct qdisc_walker arg)
	350	{
	351	struct multiq_sched_data *q = qdisc_priv(sch);
	352	int band;
	353
	354	if (arg->stop)
	355	return;
	356
	357	for (band = 0; band < q->bands; band++) {
	358	if (arg->count < arg->skip) {
	359	arg->count++;
	360	continue;
	361	}
cc7ec456	362	if (arg->fn(sch, band + 1, arg) < 0) {
92651940 AD	363	arg->stop = 1;
	364	break;
	365	}
	366	arg->count++;
	367	}
	368	}
	369
cbaacc4e AA	370	static struct tcf_block multiq_tcf_block(struct Qdisc sch, unsigned long cl,
cbaacc4e AA	371	struct netlink_ext_ack *extack)
92651940 AD	372	{
	373	struct multiq_sched_data *q = qdisc_priv(sch);
	374
	375	if (cl)
	376	return NULL;
6529eaba	377	return q->block;
92651940 AD	378	}
	379
	380	static const struct Qdisc_class_ops multiq_class_ops = {
	381	.graft = multiq_graft,
	382	.leaf = multiq_leaf,
143976ce	383	.find = multiq_find,
92651940	384	.walk = multiq_walk,
6529eaba	385	.tcf_block = multiq_tcf_block,
92651940	386	.bind_tcf = multiq_bind,
143976ce	387	.unbind_tcf = multiq_unbind,
92651940 AD	388	.dump = multiq_dump_class,
	389	.dump_stats = multiq_dump_class_stats,
	390	};
	391
	392	static struct Qdisc_ops multiq_qdisc_ops __read_mostly = {
	393	.next = NULL,
	394	.cl_ops = &multiq_class_ops,
	395	.id = "multiq",
	396	.priv_size = sizeof(struct multiq_sched_data),
	397	.enqueue = multiq_enqueue,
	398	.dequeue = multiq_dequeue,
8e3af978	399	.peek = multiq_peek,
92651940 AD	400	.init = multiq_init,
	401	.reset = multiq_reset,
	402	.destroy = multiq_destroy,
	403	.change = multiq_tune,
	404	.dump = multiq_dump,
	405	.owner = THIS_MODULE,
	406	};
	407
	408	static int __init multiq_module_init(void)
	409	{
	410	return register_qdisc(&multiq_qdisc_ops);
	411	}
	412
	413	static void __exit multiq_module_exit(void)
	414	{
	415	unregister_qdisc(&multiq_qdisc_ops);
	416	}
	417
	418	module_init(multiq_module_init)
	419	module_exit(multiq_module_exit)
	420
	421	MODULE_LICENSE("GPL");