[thirdparty/kernel/stable.git] / net / sched / sch_multiq.c

/*
 * Copyright (c) 2008, Intel Corporation.
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, see <http://www.gnu.org/licenses/>.
 *
 * 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;
		/* fall through */
	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;
	int i;

	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;

	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_tree_flush_backlog(child);
			qdisc_put(child);
		}
	}

	sch_tree_unlock(sch);

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