Linux Audio

Check our new training course

Linux debugging, profiling, tracing and performance analysis training

Mar 24-27, 2025, special US time zones
Register
Loading...
Note: File does not exist in v4.6.
  1/*
  2 * net/sched/sch_cbs.c	Credit Based Shaper
  3 *
  4 *		This program is free software; you can redistribute it and/or
  5 *		modify it under the terms of the GNU General Public License
  6 *		as published by the Free Software Foundation; either version
  7 *		2 of the License, or (at your option) any later version.
  8 *
  9 * Authors:	Vinicius Costa Gomes <vinicius.gomes@intel.com>
 10 *
 11 */
 12
 13/* Credit Based Shaper (CBS)
 14 * =========================
 15 *
 16 * This is a simple rate-limiting shaper aimed at TSN applications on
 17 * systems with known traffic workloads.
 18 *
 19 * Its algorithm is defined by the IEEE 802.1Q-2014 Specification,
 20 * Section 8.6.8.2, and explained in more detail in the Annex L of the
 21 * same specification.
 22 *
 23 * There are four tunables to be considered:
 24 *
 25 *	'idleslope': Idleslope is the rate of credits that is
 26 *	accumulated (in kilobits per second) when there is at least
 27 *	one packet waiting for transmission. Packets are transmitted
 28 *	when the current value of credits is equal or greater than
 29 *	zero. When there is no packet to be transmitted the amount of
 30 *	credits is set to zero. This is the main tunable of the CBS
 31 *	algorithm.
 32 *
 33 *	'sendslope':
 34 *	Sendslope is the rate of credits that is depleted (it should be a
 35 *	negative number of kilobits per second) when a transmission is
 36 *	ocurring. It can be calculated as follows, (IEEE 802.1Q-2014 Section
 37 *	8.6.8.2 item g):
 38 *
 39 *	sendslope = idleslope - port_transmit_rate
 40 *
 41 *	'hicredit': Hicredit defines the maximum amount of credits (in
 42 *	bytes) that can be accumulated. Hicredit depends on the
 43 *	characteristics of interfering traffic,
 44 *	'max_interference_size' is the maximum size of any burst of
 45 *	traffic that can delay the transmission of a frame that is
 46 *	available for transmission for this traffic class, (IEEE
 47 *	802.1Q-2014 Annex L, Equation L-3):
 48 *
 49 *	hicredit = max_interference_size * (idleslope / port_transmit_rate)
 50 *
 51 *	'locredit': Locredit is the minimum amount of credits that can
 52 *	be reached. It is a function of the traffic flowing through
 53 *	this qdisc (IEEE 802.1Q-2014 Annex L, Equation L-2):
 54 *
 55 *	locredit = max_frame_size * (sendslope / port_transmit_rate)
 56 */
 57
 58#include <linux/module.h>
 59#include <linux/types.h>
 60#include <linux/kernel.h>
 61#include <linux/string.h>
 62#include <linux/errno.h>
 63#include <linux/skbuff.h>
 64#include <net/netlink.h>
 65#include <net/sch_generic.h>
 66#include <net/pkt_sched.h>
 67
 68#define BYTES_PER_KBIT (1000LL / 8)
 69
 70struct cbs_sched_data {
 71	bool offload;
 72	int queue;
 73	s64 port_rate; /* in bytes/s */
 74	s64 last; /* timestamp in ns */
 75	s64 credits; /* in bytes */
 76	s32 locredit; /* in bytes */
 77	s32 hicredit; /* in bytes */
 78	s64 sendslope; /* in bytes/s */
 79	s64 idleslope; /* in bytes/s */
 80	struct qdisc_watchdog watchdog;
 81	int (*enqueue)(struct sk_buff *skb, struct Qdisc *sch);
 82	struct sk_buff *(*dequeue)(struct Qdisc *sch);
 83};
 84
 85static int cbs_enqueue_offload(struct sk_buff *skb, struct Qdisc *sch)
 86{
 87	return qdisc_enqueue_tail(skb, sch);
 88}
 89
 90static int cbs_enqueue_soft(struct sk_buff *skb, struct Qdisc *sch)
 91{
 92	struct cbs_sched_data *q = qdisc_priv(sch);
 93
 94	if (sch->q.qlen == 0 && q->credits > 0) {
 95		/* We need to stop accumulating credits when there's
 96		 * no enqueued packets and q->credits is positive.
 97		 */
 98		q->credits = 0;
 99		q->last = ktime_get_ns();
100	}
101
102	return qdisc_enqueue_tail(skb, sch);
103}
104
105static int cbs_enqueue(struct sk_buff *skb, struct Qdisc *sch,
106		       struct sk_buff **to_free)
107{
108	struct cbs_sched_data *q = qdisc_priv(sch);
109
110	return q->enqueue(skb, sch);
111}
112
113/* timediff is in ns, slope is in bytes/s */
114static s64 timediff_to_credits(s64 timediff, s64 slope)
115{
116	return div64_s64(timediff * slope, NSEC_PER_SEC);
117}
118
119static s64 delay_from_credits(s64 credits, s64 slope)
120{
121	if (unlikely(slope == 0))
122		return S64_MAX;
123
124	return div64_s64(-credits * NSEC_PER_SEC, slope);
125}
126
127static s64 credits_from_len(unsigned int len, s64 slope, s64 port_rate)
128{
129	if (unlikely(port_rate == 0))
130		return S64_MAX;
131
132	return div64_s64(len * slope, port_rate);
133}
134
135static struct sk_buff *cbs_dequeue_soft(struct Qdisc *sch)
136{
137	struct cbs_sched_data *q = qdisc_priv(sch);
138	s64 now = ktime_get_ns();
139	struct sk_buff *skb;
140	s64 credits;
141	int len;
142
143	if (q->credits < 0) {
144		credits = timediff_to_credits(now - q->last, q->idleslope);
145
146		credits = q->credits + credits;
147		q->credits = min_t(s64, credits, q->hicredit);
148
149		if (q->credits < 0) {
150			s64 delay;
151
152			delay = delay_from_credits(q->credits, q->idleslope);
153			qdisc_watchdog_schedule_ns(&q->watchdog, now + delay);
154
155			q->last = now;
156
157			return NULL;
158		}
159	}
160
161	skb = qdisc_dequeue_head(sch);
162	if (!skb)
163		return NULL;
164
165	len = qdisc_pkt_len(skb);
166
167	/* As sendslope is a negative number, this will decrease the
168	 * amount of q->credits.
169	 */
170	credits = credits_from_len(len, q->sendslope, q->port_rate);
171	credits += q->credits;
172
173	q->credits = max_t(s64, credits, q->locredit);
174	q->last = now;
175
176	return skb;
177}
178
179static struct sk_buff *cbs_dequeue_offload(struct Qdisc *sch)
180{
181	return qdisc_dequeue_head(sch);
182}
183
184static struct sk_buff *cbs_dequeue(struct Qdisc *sch)
185{
186	struct cbs_sched_data *q = qdisc_priv(sch);
187
188	return q->dequeue(sch);
189}
190
191static const struct nla_policy cbs_policy[TCA_CBS_MAX + 1] = {
192	[TCA_CBS_PARMS]	= { .len = sizeof(struct tc_cbs_qopt) },
193};
194
195static void cbs_disable_offload(struct net_device *dev,
196				struct cbs_sched_data *q)
197{
198	struct tc_cbs_qopt_offload cbs = { };
199	const struct net_device_ops *ops;
200	int err;
201
202	if (!q->offload)
203		return;
204
205	q->enqueue = cbs_enqueue_soft;
206	q->dequeue = cbs_dequeue_soft;
207
208	ops = dev->netdev_ops;
209	if (!ops->ndo_setup_tc)
210		return;
211
212	cbs.queue = q->queue;
213	cbs.enable = 0;
214
215	err = ops->ndo_setup_tc(dev, TC_SETUP_QDISC_CBS, &cbs);
216	if (err < 0)
217		pr_warn("Couldn't disable CBS offload for queue %d\n",
218			cbs.queue);
219}
220
221static int cbs_enable_offload(struct net_device *dev, struct cbs_sched_data *q,
222			      const struct tc_cbs_qopt *opt,
223			      struct netlink_ext_ack *extack)
224{
225	const struct net_device_ops *ops = dev->netdev_ops;
226	struct tc_cbs_qopt_offload cbs = { };
227	int err;
228
229	if (!ops->ndo_setup_tc) {
230		NL_SET_ERR_MSG(extack, "Specified device does not support cbs offload");
231		return -EOPNOTSUPP;
232	}
233
234	cbs.queue = q->queue;
235
236	cbs.enable = 1;
237	cbs.hicredit = opt->hicredit;
238	cbs.locredit = opt->locredit;
239	cbs.idleslope = opt->idleslope;
240	cbs.sendslope = opt->sendslope;
241
242	err = ops->ndo_setup_tc(dev, TC_SETUP_QDISC_CBS, &cbs);
243	if (err < 0) {
244		NL_SET_ERR_MSG(extack, "Specified device failed to setup cbs hardware offload");
245		return err;
246	}
247
248	q->enqueue = cbs_enqueue_offload;
249	q->dequeue = cbs_dequeue_offload;
250
251	return 0;
252}
253
254static int cbs_change(struct Qdisc *sch, struct nlattr *opt,
255		      struct netlink_ext_ack *extack)
256{
257	struct cbs_sched_data *q = qdisc_priv(sch);
258	struct net_device *dev = qdisc_dev(sch);
259	struct nlattr *tb[TCA_CBS_MAX + 1];
260	struct tc_cbs_qopt *qopt;
261	int err;
262
263	err = nla_parse_nested(tb, TCA_CBS_MAX, opt, cbs_policy, extack);
264	if (err < 0)
265		return err;
266
267	if (!tb[TCA_CBS_PARMS]) {
268		NL_SET_ERR_MSG(extack, "Missing CBS parameter which are mandatory");
269		return -EINVAL;
270	}
271
272	qopt = nla_data(tb[TCA_CBS_PARMS]);
273
274	if (!qopt->offload) {
275		struct ethtool_link_ksettings ecmd;
276		s64 link_speed;
277
278		if (!__ethtool_get_link_ksettings(dev, &ecmd))
279			link_speed = ecmd.base.speed;
280		else
281			link_speed = SPEED_1000;
282
283		q->port_rate = link_speed * 1000 * BYTES_PER_KBIT;
284
285		cbs_disable_offload(dev, q);
286	} else {
287		err = cbs_enable_offload(dev, q, qopt, extack);
288		if (err < 0)
289			return err;
290	}
291
292	/* Everything went OK, save the parameters used. */
293	q->hicredit = qopt->hicredit;
294	q->locredit = qopt->locredit;
295	q->idleslope = qopt->idleslope * BYTES_PER_KBIT;
296	q->sendslope = qopt->sendslope * BYTES_PER_KBIT;
297	q->offload = qopt->offload;
298
299	return 0;
300}
301
302static int cbs_init(struct Qdisc *sch, struct nlattr *opt,
303		    struct netlink_ext_ack *extack)
304{
305	struct cbs_sched_data *q = qdisc_priv(sch);
306	struct net_device *dev = qdisc_dev(sch);
307
308	if (!opt) {
309		NL_SET_ERR_MSG(extack, "Missing CBS qdisc options  which are mandatory");
310		return -EINVAL;
311	}
312
313	q->queue = sch->dev_queue - netdev_get_tx_queue(dev, 0);
314
315	q->enqueue = cbs_enqueue_soft;
316	q->dequeue = cbs_dequeue_soft;
317
318	qdisc_watchdog_init(&q->watchdog, sch);
319
320	return cbs_change(sch, opt, extack);
321}
322
323static void cbs_destroy(struct Qdisc *sch)
324{
325	struct cbs_sched_data *q = qdisc_priv(sch);
326	struct net_device *dev = qdisc_dev(sch);
327
328	qdisc_watchdog_cancel(&q->watchdog);
329
330	cbs_disable_offload(dev, q);
331}
332
333static int cbs_dump(struct Qdisc *sch, struct sk_buff *skb)
334{
335	struct cbs_sched_data *q = qdisc_priv(sch);
336	struct tc_cbs_qopt opt = { };
337	struct nlattr *nest;
338
339	nest = nla_nest_start(skb, TCA_OPTIONS);
340	if (!nest)
341		goto nla_put_failure;
342
343	opt.hicredit = q->hicredit;
344	opt.locredit = q->locredit;
345	opt.sendslope = div64_s64(q->sendslope, BYTES_PER_KBIT);
346	opt.idleslope = div64_s64(q->idleslope, BYTES_PER_KBIT);
347	opt.offload = q->offload;
348
349	if (nla_put(skb, TCA_CBS_PARMS, sizeof(opt), &opt))
350		goto nla_put_failure;
351
352	return nla_nest_end(skb, nest);
353
354nla_put_failure:
355	nla_nest_cancel(skb, nest);
356	return -1;
357}
358
359static struct Qdisc_ops cbs_qdisc_ops __read_mostly = {
360	.id		=	"cbs",
361	.priv_size	=	sizeof(struct cbs_sched_data),
362	.enqueue	=	cbs_enqueue,
363	.dequeue	=	cbs_dequeue,
364	.peek		=	qdisc_peek_dequeued,
365	.init		=	cbs_init,
366	.reset		=	qdisc_reset_queue,
367	.destroy	=	cbs_destroy,
368	.change		=	cbs_change,
369	.dump		=	cbs_dump,
370	.owner		=	THIS_MODULE,
371};
372
373static int __init cbs_module_init(void)
374{
375	return register_qdisc(&cbs_qdisc_ops);
376}
377
378static void __exit cbs_module_exit(void)
379{
380	unregister_qdisc(&cbs_qdisc_ops);
381}
382module_init(cbs_module_init)
383module_exit(cbs_module_exit)
384MODULE_LICENSE("GPL");