[linux-2.6-block.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;
		/* 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
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;
f3ae608e	46	/* fall through */
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
520ac30f ED	60	multiq_enqueue(struct sk_buff skb, struct Qdisc sch,
520ac30f ED	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
520ac30f	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;
	177	int i;
	178
	179	if (!netif_is_multiqueue(qdisc_dev(sch)))
149490f1	180	return -EOPNOTSUPP;
92651940 AD	181	if (nla_len(opt) < sizeof(*qopt))
	182	return -EINVAL;
	183
	184	qopt = nla_data(opt);
	185
	186	qopt->bands = qdisc_dev(sch)->real_num_tx_queues;
	187
	188	sch_tree_lock(sch);
	189	q->bands = qopt->bands;
	190	for (i = q->bands; i < q->max_bands; i++) {
f07d1501	191	if (q->queues[i] != &noop_qdisc) {
b94c8afc	192	struct Qdisc *child = q->queues[i];
e5f0e8f8	193
b94c8afc	194	q->queues[i] = &noop_qdisc;
e5f0e8f8	195	qdisc_tree_flush_backlog(child);
86bd446b	196	qdisc_put(child);
92651940 AD	197	}
	198	}
	199
	200	sch_tree_unlock(sch);
	201
	202	for (i = 0; i < q->bands; i++) {
	203	if (q->queues[i] == &noop_qdisc) {
b94c8afc	204	struct Qdisc child, old;
3511c913	205	child = qdisc_create_dflt(sch->dev_queue,
92651940 AD	206	&pfifo_qdisc_ops,
92651940 AD	207	TC_H_MAKE(sch->handle,
a38a9882	208	i + 1), extack);
92651940 AD	209	if (child) {
92651940 AD	210	sch_tree_lock(sch);
b94c8afc PM	211	old = q->queues[i];
b94c8afc PM	212	q->queues[i] = child;
49b49971 JK	213	if (child != &noop_qdisc)
49b49971 JK	214	qdisc_hash_add(child, true);
92651940	215
b94c8afc	216	if (old != &noop_qdisc) {
e5f0e8f8	217	qdisc_tree_flush_backlog(old);
86bd446b	218	qdisc_put(old);
92651940 AD	219	}
	220	sch_tree_unlock(sch);
	221	}
	222	}
	223	}
	224	return 0;
	225	}
	226
e63d7dfd AA	227	static int multiq_init(struct Qdisc sch, struct nlattr opt,
e63d7dfd AA	228	struct netlink_ext_ack *extack)
92651940 AD	229	{
92651940 AD	230	struct multiq_sched_data *q = qdisc_priv(sch);
f07d1501	231	int i, err;
92651940 AD	232
	233	q->queues = NULL;
	234
ac8ef4ab	235	if (!opt)
92651940 AD	236	return -EINVAL;
92651940 AD	237
8d1a77f9	238	err = tcf_block_get(&q->block, &q->filter_list, sch, extack);
6529eaba JP	239	if (err)
	240	return err;
	241
92651940 AD	242	q->max_bands = qdisc_dev(sch)->num_tx_queues;
	243
	244	q->queues = kcalloc(q->max_bands, sizeof(struct Qdisc *), GFP_KERNEL);
	245	if (!q->queues)
	246	return -ENOBUFS;
	247	for (i = 0; i < q->max_bands; i++)
	248	q->queues[i] = &noop_qdisc;
	249
2030721c	250	return multiq_tune(sch, opt, extack);
92651940 AD	251	}
	252
	253	static int multiq_dump(struct Qdisc sch, struct sk_buff skb)
	254	{
	255	struct multiq_sched_data *q = qdisc_priv(sch);
	256	unsigned char *b = skb_tail_pointer(skb);
	257	struct tc_multiq_qopt opt;
	258
	259	opt.bands = q->bands;
	260	opt.max_bands = q->max_bands;
	261
1b34ec43 DM	262	if (nla_put(skb, TCA_OPTIONS, sizeof(opt), &opt))
1b34ec43 DM	263	goto nla_put_failure;
92651940 AD	264
	265	return skb->len;
	266
	267	nla_put_failure:
	268	nlmsg_trim(skb, b);
	269	return -1;
	270	}
	271
	272	static int multiq_graft(struct Qdisc sch, unsigned long arg, struct Qdisc new,
653d6fd6	273	struct Qdisc *old, struct netlink_ext_ack extack)
92651940 AD	274	{
	275	struct multiq_sched_data *q = qdisc_priv(sch);
	276	unsigned long band = arg - 1;
	277
92651940 AD	278	if (new == NULL)
	279	new = &noop_qdisc;
	280
86a7996c	281	*old = qdisc_replace(sch, new, &q->queues[band]);
92651940 AD	282	return 0;
	283	}
	284
	285	static struct Qdisc *
	286	multiq_leaf(struct Qdisc *sch, unsigned long arg)
	287	{
	288	struct multiq_sched_data *q = qdisc_priv(sch);
	289	unsigned long band = arg - 1;
	290
92651940 AD	291	return q->queues[band];
	292	}
	293
143976ce	294	static unsigned long multiq_find(struct Qdisc *sch, u32 classid)
92651940 AD	295	{
	296	struct multiq_sched_data *q = qdisc_priv(sch);
	297	unsigned long band = TC_H_MIN(classid);
	298
	299	if (band - 1 >= q->bands)
	300	return 0;
	301	return band;
	302	}
	303
	304	static unsigned long multiq_bind(struct Qdisc *sch, unsigned long parent,
	305	u32 classid)
	306	{
143976ce	307	return multiq_find(sch, classid);
92651940 AD	308	}
	309
	310
143976ce	311	static void multiq_unbind(struct Qdisc *q, unsigned long cl)
92651940	312	{
92651940 AD	313	}
92651940 AD	314
92651940 AD	315	static int multiq_dump_class(struct Qdisc *sch, unsigned long cl,
	316	struct sk_buff skb, struct tcmsg tcm)
	317	{
	318	struct multiq_sched_data *q = qdisc_priv(sch);
	319
92651940	320	tcm->tcm_handle \|= TC_H_MIN(cl);
cc7ec456	321	tcm->tcm_info = q->queues[cl - 1]->handle;
92651940 AD	322	return 0;
	323	}
	324
	325	static int multiq_dump_class_stats(struct Qdisc *sch, unsigned long cl,
	326	struct gnet_dump *d)
	327	{
	328	struct multiq_sched_data *q = qdisc_priv(sch);
	329	struct Qdisc *cl_q;
	330
	331	cl_q = q->queues[cl - 1];
edb09eb1 ED	332	if (gnet_stats_copy_basic(qdisc_root_sleeping_running(sch),
edb09eb1 ED	333	d, NULL, &cl_q->bstats) < 0 \|\|
5dd431b6	334	qdisc_qstats_copy(d, cl_q) < 0)
92651940 AD	335	return -1;
	336
	337	return 0;
	338	}
	339
	340	static void multiq_walk(struct Qdisc sch, struct qdisc_walker arg)
	341	{
	342	struct multiq_sched_data *q = qdisc_priv(sch);
	343	int band;
	344
	345	if (arg->stop)
	346	return;
	347
	348	for (band = 0; band < q->bands; band++) {
	349	if (arg->count < arg->skip) {
	350	arg->count++;
	351	continue;
	352	}
cc7ec456	353	if (arg->fn(sch, band + 1, arg) < 0) {
92651940 AD	354	arg->stop = 1;
	355	break;
	356	}
	357	arg->count++;
	358	}
	359	}
	360
cbaacc4e AA	361	static struct tcf_block multiq_tcf_block(struct Qdisc sch, unsigned long cl,
cbaacc4e AA	362	struct netlink_ext_ack *extack)
92651940 AD	363	{
	364	struct multiq_sched_data *q = qdisc_priv(sch);
	365
	366	if (cl)
	367	return NULL;
6529eaba	368	return q->block;
92651940 AD	369	}
	370
	371	static const struct Qdisc_class_ops multiq_class_ops = {
	372	.graft = multiq_graft,
	373	.leaf = multiq_leaf,
143976ce	374	.find = multiq_find,
92651940	375	.walk = multiq_walk,
6529eaba	376	.tcf_block = multiq_tcf_block,
92651940	377	.bind_tcf = multiq_bind,
143976ce	378	.unbind_tcf = multiq_unbind,
92651940 AD	379	.dump = multiq_dump_class,
	380	.dump_stats = multiq_dump_class_stats,
	381	};
	382
	383	static struct Qdisc_ops multiq_qdisc_ops __read_mostly = {
	384	.next = NULL,
	385	.cl_ops = &multiq_class_ops,
	386	.id = "multiq",
	387	.priv_size = sizeof(struct multiq_sched_data),
	388	.enqueue = multiq_enqueue,
	389	.dequeue = multiq_dequeue,
8e3af978	390	.peek = multiq_peek,
92651940 AD	391	.init = multiq_init,
	392	.reset = multiq_reset,
	393	.destroy = multiq_destroy,
	394	.change = multiq_tune,
	395	.dump = multiq_dump,
	396	.owner = THIS_MODULE,
	397	};
	398
	399	static int __init multiq_module_init(void)
	400	{
	401	return register_qdisc(&multiq_qdisc_ops);
	402	}
	403
	404	static void __exit multiq_module_exit(void)
	405	{
	406	unregister_qdisc(&multiq_qdisc_ops);
	407	}
	408
	409	module_init(multiq_module_init)
	410	module_exit(multiq_module_exit)
	411
	412	MODULE_LICENSE("GPL");