sch_cbq.c 41.3 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
/*
 * net/sched/sch_cbq.c	Class-Based Queueing discipline.
 *
 *		This program is free software; you can redistribute it and/or
 *		modify it under the terms of the GNU General Public License
 *		as published by the Free Software Foundation; either version
 *		2 of the License, or (at your option) any later version.
 *
 * Authors:	Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
 *
 */

#include <linux/module.h>
14
#include <linux/slab.h>
L
Linus Torvalds 已提交
15 16 17 18 19
#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/string.h>
#include <linux/errno.h>
#include <linux/skbuff.h>
20
#include <net/netlink.h>
L
Linus Torvalds 已提交
21 22 23 24 25 26 27
#include <net/pkt_sched.h>


/*	Class-Based Queueing (CBQ) algorithm.
	=======================================

	Sources: [1] Sally Floyd and Van Jacobson, "Link-sharing and Resource
28
		 Management Models for Packet Networks",
L
Linus Torvalds 已提交
29 30
		 IEEE/ACM Transactions on Networking, Vol.3, No.4, 1995

31
		 [2] Sally Floyd, "Notes on CBQ and Guaranteed Service", 1995
L
Linus Torvalds 已提交
32

33
		 [3] Sally Floyd, "Notes on Class-Based Queueing: Setting
L
Linus Torvalds 已提交
34 35 36 37 38 39 40 41 42 43 44 45 46
		 Parameters", 1996

		 [4] Sally Floyd and Michael Speer, "Experimental Results
		 for Class-Based Queueing", 1998, not published.

	-----------------------------------------------------------------------

	Algorithm skeleton was taken from NS simulator cbq.cc.
	If someone wants to check this code against the LBL version,
	he should take into account that ONLY the skeleton was borrowed,
	the implementation is different. Particularly:

	--- The WRR algorithm is different. Our version looks more
47 48 49 50 51 52
	reasonable (I hope) and works when quanta are allowed to be
	less than MTU, which is always the case when real time classes
	have small rates. Note, that the statement of [3] is
	incomplete, delay may actually be estimated even if class
	per-round allotment is less than MTU. Namely, if per-round
	allotment is W*r_i, and r_1+...+r_k = r < 1
L
Linus Torvalds 已提交
53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74

	delay_i <= ([MTU/(W*r_i)]*W*r + W*r + k*MTU)/B

	In the worst case we have IntServ estimate with D = W*r+k*MTU
	and C = MTU*r. The proof (if correct at all) is trivial.


	--- It seems that cbq-2.0 is not very accurate. At least, I cannot
	interpret some places, which look like wrong translations
	from NS. Anyone is advised to find these differences
	and explain to me, why I am wrong 8).

	--- Linux has no EOI event, so that we cannot estimate true class
	idle time. Workaround is to consider the next dequeue event
	as sign that previous packet is finished. This is wrong because of
	internal device queueing, but on a permanently loaded link it is true.
	Moreover, combined with clock integrator, this scheme looks
	very close to an ideal solution.  */

struct cbq_sched_data;


E
Eric Dumazet 已提交
75
struct cbq_class {
76
	struct Qdisc_class_common common;
L
Linus Torvalds 已提交
77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121
	struct cbq_class	*next_alive;	/* next class with backlog in this priority band */

/* Parameters */
	unsigned char		priority;	/* class priority */
	unsigned char		priority2;	/* priority to be used after overlimit */
	unsigned char		ewma_log;	/* time constant for idle time calculation */

	u32			defmap;

	/* Link-sharing scheduler parameters */
	long			maxidle;	/* Class parameters: see below. */
	long			offtime;
	long			minidle;
	u32			avpkt;
	struct qdisc_rate_table	*R_tab;

	/* General scheduler (WRR) parameters */
	long			allot;
	long			quantum;	/* Allotment per WRR round */
	long			weight;		/* Relative allotment: see below */

	struct Qdisc		*qdisc;		/* Ptr to CBQ discipline */
	struct cbq_class	*split;		/* Ptr to split node */
	struct cbq_class	*share;		/* Ptr to LS parent in the class tree */
	struct cbq_class	*tparent;	/* Ptr to tree parent in the class tree */
	struct cbq_class	*borrow;	/* NULL if class is bandwidth limited;
						   parent otherwise */
	struct cbq_class	*sibling;	/* Sibling chain */
	struct cbq_class	*children;	/* Pointer to children chain */

	struct Qdisc		*q;		/* Elementary queueing discipline */


/* Variables */
	unsigned char		cpriority;	/* Effective priority */
	unsigned char		delayed;
	unsigned char		level;		/* level of the class in hierarchy:
						   0 for leaf classes, and maximal
						   level of children + 1 for nodes.
						 */

	psched_time_t		last;		/* Last end of service */
	psched_time_t		undertime;
	long			avgidle;
	long			deficit;	/* Saved deficit for WRR */
122
	psched_time_t		penalized;
123
	struct gnet_stats_basic_packed bstats;
L
Linus Torvalds 已提交
124
	struct gnet_stats_queue qstats;
125
	struct gnet_stats_rate_est64 rate_est;
L
Linus Torvalds 已提交
126 127
	struct tc_cbq_xstats	xstats;

J
John Fastabend 已提交
128
	struct tcf_proto __rcu	*filter_list;
L
Linus Torvalds 已提交
129 130 131 132

	int			refcnt;
	int			filters;

E
Eric Dumazet 已提交
133
	struct cbq_class	*defaults[TC_PRIO_MAX + 1];
L
Linus Torvalds 已提交
134 135
};

E
Eric Dumazet 已提交
136
struct cbq_sched_data {
137
	struct Qdisc_class_hash	clhash;			/* Hash table of all classes */
E
Eric Dumazet 已提交
138 139
	int			nclasses[TC_CBQ_MAXPRIO + 1];
	unsigned int		quanta[TC_CBQ_MAXPRIO + 1];
L
Linus Torvalds 已提交
140 141 142

	struct cbq_class	link;

E
Eric Dumazet 已提交
143 144
	unsigned int		activemask;
	struct cbq_class	*active[TC_CBQ_MAXPRIO + 1];	/* List of all classes
L
Linus Torvalds 已提交
145 146
								   with backlog */

147
#ifdef CONFIG_NET_CLS_ACT
L
Linus Torvalds 已提交
148 149 150 151 152 153
	struct cbq_class	*rx_class;
#endif
	struct cbq_class	*tx_class;
	struct cbq_class	*tx_borrowed;
	int			tx_len;
	psched_time_t		now;		/* Cached timestamp */
E
Eric Dumazet 已提交
154
	unsigned int		pmask;
L
Linus Torvalds 已提交
155

156
	struct hrtimer		delay_timer;
157
	struct qdisc_watchdog	watchdog;	/* Watchdog timer,
L
Linus Torvalds 已提交
158 159 160
						   started when CBQ has
						   backlog, but cannot
						   transmit just now */
161
	psched_tdiff_t		wd_expires;
L
Linus Torvalds 已提交
162 163 164 165 166
	int			toplevel;
	u32			hgenerator;
};


E
Eric Dumazet 已提交
167
#define L2T(cl, len)	qdisc_l2t((cl)->R_tab, len)
L
Linus Torvalds 已提交
168

E
Eric Dumazet 已提交
169
static inline struct cbq_class *
L
Linus Torvalds 已提交
170 171
cbq_class_lookup(struct cbq_sched_data *q, u32 classid)
{
172
	struct Qdisc_class_common *clc;
L
Linus Torvalds 已提交
173

174 175 176 177
	clc = qdisc_class_find(&q->clhash, classid);
	if (clc == NULL)
		return NULL;
	return container_of(clc, struct cbq_class, common);
L
Linus Torvalds 已提交
178 179
}

180
#ifdef CONFIG_NET_CLS_ACT
L
Linus Torvalds 已提交
181 182 183 184

static struct cbq_class *
cbq_reclassify(struct sk_buff *skb, struct cbq_class *this)
{
E
Eric Dumazet 已提交
185
	struct cbq_class *cl;
L
Linus Torvalds 已提交
186

E
Eric Dumazet 已提交
187 188
	for (cl = this->tparent; cl; cl = cl->tparent) {
		struct cbq_class *new = cl->defaults[TC_PRIO_BESTEFFORT];
L
Linus Torvalds 已提交
189

E
Eric Dumazet 已提交
190 191 192
		if (new != NULL && new != this)
			return new;
	}
L
Linus Torvalds 已提交
193 194 195 196 197 198
	return NULL;
}

#endif

/* Classify packet. The procedure is pretty complicated, but
E
Eric Dumazet 已提交
199 200 201 202 203 204 205
 * it allows us to combine link sharing and priority scheduling
 * transparently.
 *
 * Namely, you can put link sharing rules (f.e. route based) at root of CBQ,
 * so that it resolves to split nodes. Then packets are classified
 * by logical priority, or a more specific classifier may be attached
 * to the split node.
L
Linus Torvalds 已提交
206 207 208 209 210 211 212 213 214 215
 */

static struct cbq_class *
cbq_classify(struct sk_buff *skb, struct Qdisc *sch, int *qerr)
{
	struct cbq_sched_data *q = qdisc_priv(sch);
	struct cbq_class *head = &q->link;
	struct cbq_class **defmap;
	struct cbq_class *cl = NULL;
	u32 prio = skb->priority;
J
John Fastabend 已提交
216
	struct tcf_proto *fl;
L
Linus Torvalds 已提交
217 218 219 220 221
	struct tcf_result res;

	/*
	 *  Step 1. If skb->priority points to one of our classes, use it.
	 */
E
Eric Dumazet 已提交
222
	if (TC_H_MAJ(prio ^ sch->handle) == 0 &&
L
Linus Torvalds 已提交
223 224 225
	    (cl = cbq_class_lookup(q, prio)) != NULL)
		return cl;

226
	*qerr = NET_XMIT_SUCCESS | __NET_XMIT_BYPASS;
L
Linus Torvalds 已提交
227 228 229 230
	for (;;) {
		int result = 0;
		defmap = head->defaults;

J
John Fastabend 已提交
231
		fl = rcu_dereference_bh(head->filter_list);
L
Linus Torvalds 已提交
232 233 234
		/*
		 * Step 2+n. Apply classifier.
		 */
235
		result = tc_classify(skb, fl, &res, true);
J
John Fastabend 已提交
236
		if (!fl || result < 0)
L
Linus Torvalds 已提交
237 238
			goto fallback;

E
Eric Dumazet 已提交
239 240
		cl = (void *)res.class;
		if (!cl) {
L
Linus Torvalds 已提交
241 242
			if (TC_H_MAJ(res.classid))
				cl = cbq_class_lookup(q, res.classid);
E
Eric Dumazet 已提交
243
			else if ((cl = defmap[res.classid & TC_PRIO_MAX]) == NULL)
L
Linus Torvalds 已提交
244 245
				cl = defmap[TC_PRIO_BESTEFFORT];

246
			if (cl == NULL)
L
Linus Torvalds 已提交
247 248
				goto fallback;
		}
249 250
		if (cl->level >= head->level)
			goto fallback;
L
Linus Torvalds 已提交
251 252 253
#ifdef CONFIG_NET_CLS_ACT
		switch (result) {
		case TC_ACT_QUEUED:
254
		case TC_ACT_STOLEN:
255
			*qerr = NET_XMIT_SUCCESS | __NET_XMIT_STOLEN;
L
Linus Torvalds 已提交
256 257
		case TC_ACT_SHOT:
			return NULL;
258 259
		case TC_ACT_RECLASSIFY:
			return cbq_reclassify(skb, cl);
L
Linus Torvalds 已提交
260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279
		}
#endif
		if (cl->level == 0)
			return cl;

		/*
		 * Step 3+n. If classifier selected a link sharing class,
		 *	   apply agency specific classifier.
		 *	   Repeat this procdure until we hit a leaf node.
		 */
		head = cl;
	}

fallback:
	cl = head;

	/*
	 * Step 4. No success...
	 */
	if (TC_H_MAJ(prio) == 0 &&
E
Eric Dumazet 已提交
280
	    !(cl = head->defaults[prio & TC_PRIO_MAX]) &&
L
Linus Torvalds 已提交
281 282 283 284 285 286 287
	    !(cl = head->defaults[TC_PRIO_BESTEFFORT]))
		return head;

	return cl;
}

/*
E
Eric Dumazet 已提交
288 289 290
 * A packet has just been enqueued on the empty class.
 * cbq_activate_class adds it to the tail of active class list
 * of its priority band.
L
Linus Torvalds 已提交
291 292
 */

E
Eric Dumazet 已提交
293
static inline void cbq_activate_class(struct cbq_class *cl)
L
Linus Torvalds 已提交
294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311
{
	struct cbq_sched_data *q = qdisc_priv(cl->qdisc);
	int prio = cl->cpriority;
	struct cbq_class *cl_tail;

	cl_tail = q->active[prio];
	q->active[prio] = cl;

	if (cl_tail != NULL) {
		cl->next_alive = cl_tail->next_alive;
		cl_tail->next_alive = cl;
	} else {
		cl->next_alive = cl;
		q->activemask |= (1<<prio);
	}
}

/*
E
Eric Dumazet 已提交
312 313 314
 * Unlink class from active chain.
 * Note that this same procedure is done directly in cbq_dequeue*
 * during round-robin procedure.
L
Linus Torvalds 已提交
315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347
 */

static void cbq_deactivate_class(struct cbq_class *this)
{
	struct cbq_sched_data *q = qdisc_priv(this->qdisc);
	int prio = this->cpriority;
	struct cbq_class *cl;
	struct cbq_class *cl_prev = q->active[prio];

	do {
		cl = cl_prev->next_alive;
		if (cl == this) {
			cl_prev->next_alive = cl->next_alive;
			cl->next_alive = NULL;

			if (cl == q->active[prio]) {
				q->active[prio] = cl_prev;
				if (cl == q->active[prio]) {
					q->active[prio] = NULL;
					q->activemask &= ~(1<<prio);
					return;
				}
			}
			return;
		}
	} while ((cl_prev = cl) != q->active[prio]);
}

static void
cbq_mark_toplevel(struct cbq_sched_data *q, struct cbq_class *cl)
{
	int toplevel = q->toplevel;

348
	if (toplevel > cl->level && !(qdisc_is_throttled(cl->q))) {
V
Vasily Averin 已提交
349
		psched_time_t now = psched_get_time();
L
Linus Torvalds 已提交
350 351

		do {
P
Patrick McHardy 已提交
352
			if (cl->undertime < now) {
L
Linus Torvalds 已提交
353 354 355
				q->toplevel = cl->level;
				return;
			}
E
Eric Dumazet 已提交
356
		} while ((cl = cl->borrow) != NULL && toplevel > cl->level);
L
Linus Torvalds 已提交
357 358 359 360 361 362 363
	}
}

static int
cbq_enqueue(struct sk_buff *skb, struct Qdisc *sch)
{
	struct cbq_sched_data *q = qdisc_priv(sch);
364
	int uninitialized_var(ret);
L
Linus Torvalds 已提交
365 366
	struct cbq_class *cl = cbq_classify(skb, sch, &ret);

367
#ifdef CONFIG_NET_CLS_ACT
L
Linus Torvalds 已提交
368 369 370
	q->rx_class = cl;
#endif
	if (cl == NULL) {
371
		if (ret & __NET_XMIT_BYPASS)
372
			qdisc_qstats_drop(sch);
L
Linus Torvalds 已提交
373 374 375 376
		kfree_skb(skb);
		return ret;
	}

377 378
	ret = qdisc_enqueue(skb, cl->q);
	if (ret == NET_XMIT_SUCCESS) {
L
Linus Torvalds 已提交
379 380 381 382 383 384 385
		sch->q.qlen++;
		cbq_mark_toplevel(q, cl);
		if (!cl->next_alive)
			cbq_activate_class(cl);
		return ret;
	}

386
	if (net_xmit_drop_count(ret)) {
387
		qdisc_qstats_drop(sch);
388 389 390
		cbq_mark_toplevel(q, cl);
		cl->qstats.drops++;
	}
L
Linus Torvalds 已提交
391 392 393
	return ret;
}

394 395
/* Overlimit action: penalize leaf class by adding offtime */
static void cbq_overlimit(struct cbq_class *cl)
L
Linus Torvalds 已提交
396 397
{
	struct cbq_sched_data *q = qdisc_priv(cl->qdisc);
P
Patrick McHardy 已提交
398
	psched_tdiff_t delay = cl->undertime - q->now;
L
Linus Torvalds 已提交
399 400 401 402

	if (!cl->delayed) {
		delay += cl->offtime;

403
		/*
E
Eric Dumazet 已提交
404 405 406 407 408
		 * Class goes to sleep, so that it will have no
		 * chance to work avgidle. Let's forgive it 8)
		 *
		 * BTW cbq-2.0 has a crap in this
		 * place, apparently they forgot to shift it by cl->ewma_log.
L
Linus Torvalds 已提交
409 410 411 412 413 414 415
		 */
		if (cl->avgidle < 0)
			delay -= (-cl->avgidle) - ((-cl->avgidle) >> cl->ewma_log);
		if (cl->avgidle < cl->minidle)
			cl->avgidle = cl->minidle;
		if (delay <= 0)
			delay = 1;
416
		cl->undertime = q->now + delay;
L
Linus Torvalds 已提交
417 418 419 420 421 422 423 424

		cl->xstats.overactions++;
		cl->delayed = 1;
	}
	if (q->wd_expires == 0 || q->wd_expires > delay)
		q->wd_expires = delay;

	/* Dirty work! We must schedule wakeups based on
E
Eric Dumazet 已提交
425 426
	 * real available rate, rather than leaf rate,
	 * which may be tiny (even zero).
L
Linus Torvalds 已提交
427 428 429 430 431 432
	 */
	if (q->toplevel == TC_CBQ_MAXLEVEL) {
		struct cbq_class *b;
		psched_tdiff_t base_delay = q->wd_expires;

		for (b = cl->borrow; b; b = b->borrow) {
P
Patrick McHardy 已提交
433
			delay = b->undertime - q->now;
L
Linus Torvalds 已提交
434 435 436 437 438 439 440 441 442 443 444
			if (delay < base_delay) {
				if (delay <= 0)
					delay = 1;
				base_delay = delay;
			}
		}

		q->wd_expires = base_delay;
	}
}

445 446
static psched_tdiff_t cbq_undelay_prio(struct cbq_sched_data *q, int prio,
				       psched_time_t now)
L
Linus Torvalds 已提交
447 448 449
{
	struct cbq_class *cl;
	struct cbq_class *cl_prev = q->active[prio];
450
	psched_time_t sched = now;
L
Linus Torvalds 已提交
451 452

	if (cl_prev == NULL)
453
		return 0;
L
Linus Torvalds 已提交
454 455 456

	do {
		cl = cl_prev->next_alive;
457
		if (now - cl->penalized > 0) {
L
Linus Torvalds 已提交
458 459 460 461 462 463 464 465 466 467 468 469 470 471 472
			cl_prev->next_alive = cl->next_alive;
			cl->next_alive = NULL;
			cl->cpriority = cl->priority;
			cl->delayed = 0;
			cbq_activate_class(cl);

			if (cl == q->active[prio]) {
				q->active[prio] = cl_prev;
				if (cl == q->active[prio]) {
					q->active[prio] = NULL;
					return 0;
				}
			}

			cl = cl_prev->next_alive;
473
		} else if (sched - cl->penalized > 0)
L
Linus Torvalds 已提交
474 475 476
			sched = cl->penalized;
	} while ((cl_prev = cl) != q->active[prio]);

477
	return sched - now;
L
Linus Torvalds 已提交
478 479
}

480
static enum hrtimer_restart cbq_undelay(struct hrtimer *timer)
L
Linus Torvalds 已提交
481
{
482
	struct cbq_sched_data *q = container_of(timer, struct cbq_sched_data,
483
						delay_timer);
484 485 486
	struct Qdisc *sch = q->watchdog.qdisc;
	psched_time_t now;
	psched_tdiff_t delay = 0;
E
Eric Dumazet 已提交
487
	unsigned int pmask;
L
Linus Torvalds 已提交
488

489
	now = psched_get_time();
490

L
Linus Torvalds 已提交
491 492 493 494 495
	pmask = q->pmask;
	q->pmask = 0;

	while (pmask) {
		int prio = ffz(~pmask);
496
		psched_tdiff_t tmp;
L
Linus Torvalds 已提交
497 498 499

		pmask &= ~(1<<prio);

500
		tmp = cbq_undelay_prio(q, prio, now);
L
Linus Torvalds 已提交
501 502 503 504 505 506 507 508
		if (tmp > 0) {
			q->pmask |= 1<<prio;
			if (tmp < delay || delay == 0)
				delay = tmp;
		}
	}

	if (delay) {
509 510 511
		ktime_t time;

		time = ktime_set(0, 0);
512
		time = ktime_add_ns(time, PSCHED_TICKS2NS(now + delay));
E
Eric Dumazet 已提交
513
		hrtimer_start(&q->delay_timer, time, HRTIMER_MODE_ABS_PINNED);
L
Linus Torvalds 已提交
514 515
	}

516
	qdisc_unthrottled(sch);
517
	__netif_schedule(qdisc_root(sch));
518
	return HRTIMER_NORESTART;
L
Linus Torvalds 已提交
519 520
}

521
/*
E
Eric Dumazet 已提交
522 523 524 525 526 527 528
 * It is mission critical procedure.
 *
 * We "regenerate" toplevel cutoff, if transmitting class
 * has backlog and it is not regulated. It is not part of
 * original CBQ description, but looks more reasonable.
 * Probably, it is wrong. This question needs further investigation.
 */
L
Linus Torvalds 已提交
529

E
Eric Dumazet 已提交
530
static inline void
L
Linus Torvalds 已提交
531 532 533 534 535 536
cbq_update_toplevel(struct cbq_sched_data *q, struct cbq_class *cl,
		    struct cbq_class *borrowed)
{
	if (cl && q->toplevel >= borrowed->level) {
		if (cl->q->q.qlen > 1) {
			do {
537
				if (borrowed->undertime == PSCHED_PASTPERFECT) {
L
Linus Torvalds 已提交
538 539 540
					q->toplevel = borrowed->level;
					return;
				}
E
Eric Dumazet 已提交
541
			} while ((borrowed = borrowed->borrow) != NULL);
L
Linus Torvalds 已提交
542
		}
543
#if 0
L
Linus Torvalds 已提交
544 545 546 547 548 549 550 551 552 553 554 555 556 557
	/* It is not necessary now. Uncommenting it
	   will save CPU cycles, but decrease fairness.
	 */
		q->toplevel = TC_CBQ_MAXLEVEL;
#endif
	}
}

static void
cbq_update(struct cbq_sched_data *q)
{
	struct cbq_class *this = q->tx_class;
	struct cbq_class *cl = this;
	int len = q->tx_len;
558
	psched_time_t now;
L
Linus Torvalds 已提交
559 560

	q->tx_class = NULL;
561 562 563 564
	/* Time integrator. We calculate EOS time
	 * by adding expected packet transmission time.
	 */
	now = q->now + L2T(&q->link, len);
L
Linus Torvalds 已提交
565 566 567 568 569 570 571 572 573

	for ( ; cl; cl = cl->share) {
		long avgidle = cl->avgidle;
		long idle;

		cl->bstats.packets++;
		cl->bstats.bytes += len;

		/*
E
Eric Dumazet 已提交
574 575 576 577
		 * (now - last) is total time between packet right edges.
		 * (last_pktlen/rate) is "virtual" busy time, so that
		 *
		 *	idle = (now - last) - last_pktlen/rate
L
Linus Torvalds 已提交
578 579
		 */

580
		idle = now - cl->last;
L
Linus Torvalds 已提交
581 582 583 584 585 586
		if ((unsigned long)idle > 128*1024*1024) {
			avgidle = cl->maxidle;
		} else {
			idle -= L2T(cl, len);

		/* true_avgidle := (1-W)*true_avgidle + W*idle,
E
Eric Dumazet 已提交
587 588 589
		 * where W=2^{-ewma_log}. But cl->avgidle is scaled:
		 * cl->avgidle == true_avgidle/W,
		 * hence:
L
Linus Torvalds 已提交
590 591 592 593 594 595 596 597 598 599 600 601 602
		 */
			avgidle += idle - (avgidle>>cl->ewma_log);
		}

		if (avgidle <= 0) {
			/* Overlimit or at-limit */

			if (avgidle < cl->minidle)
				avgidle = cl->minidle;

			cl->avgidle = avgidle;

			/* Calculate expected time, when this class
E
Eric Dumazet 已提交
603 604 605 606 607 608
			 * will be allowed to send.
			 * It will occur, when:
			 * (1-W)*true_avgidle + W*delay = 0, i.e.
			 * idle = (1/W - 1)*(-true_avgidle)
			 * or
			 * idle = (1 - W)*(-cl->avgidle);
L
Linus Torvalds 已提交
609 610 611 612
			 */
			idle = (-avgidle) - ((-avgidle) >> cl->ewma_log);

			/*
E
Eric Dumazet 已提交
613 614 615 616 617 618
			 * That is not all.
			 * To maintain the rate allocated to the class,
			 * we add to undertime virtual clock,
			 * necessary to complete transmitted packet.
			 * (len/phys_bandwidth has been already passed
			 * to the moment of cbq_update)
L
Linus Torvalds 已提交
619 620 621 622 623
			 */

			idle -= L2T(&q->link, len);
			idle += L2T(cl, len);

624
			cl->undertime = now + idle;
L
Linus Torvalds 已提交
625 626 627
		} else {
			/* Underlimit */

628
			cl->undertime = PSCHED_PASTPERFECT;
L
Linus Torvalds 已提交
629 630 631 632 633
			if (avgidle > cl->maxidle)
				cl->avgidle = cl->maxidle;
			else
				cl->avgidle = avgidle;
		}
634 635
		if ((s64)(now - cl->last) > 0)
			cl->last = now;
L
Linus Torvalds 已提交
636 637 638 639 640
	}

	cbq_update_toplevel(q, this, q->tx_borrowed);
}

E
Eric Dumazet 已提交
641
static inline struct cbq_class *
L
Linus Torvalds 已提交
642 643 644 645 646 647 648 649
cbq_under_limit(struct cbq_class *cl)
{
	struct cbq_sched_data *q = qdisc_priv(cl->qdisc);
	struct cbq_class *this_cl = cl;

	if (cl->tparent == NULL)
		return cl;

650
	if (cl->undertime == PSCHED_PASTPERFECT || q->now >= cl->undertime) {
L
Linus Torvalds 已提交
651 652 653 654 655 656
		cl->delayed = 0;
		return cl;
	}

	do {
		/* It is very suspicious place. Now overlimit
E
Eric Dumazet 已提交
657 658 659 660 661 662 663 664
		 * action is generated for not bounded classes
		 * only if link is completely congested.
		 * Though it is in agree with ancestor-only paradigm,
		 * it looks very stupid. Particularly,
		 * it means that this chunk of code will either
		 * never be called or result in strong amplification
		 * of burstiness. Dangerous, silly, and, however,
		 * no another solution exists.
L
Linus Torvalds 已提交
665
		 */
E
Eric Dumazet 已提交
666 667
		cl = cl->borrow;
		if (!cl) {
L
Linus Torvalds 已提交
668
			this_cl->qstats.overlimits++;
669
			cbq_overlimit(this_cl);
L
Linus Torvalds 已提交
670 671 672 673
			return NULL;
		}
		if (cl->level > q->toplevel)
			return NULL;
674
	} while (cl->undertime != PSCHED_PASTPERFECT && q->now < cl->undertime);
L
Linus Torvalds 已提交
675 676 677 678 679

	cl->delayed = 0;
	return cl;
}

E
Eric Dumazet 已提交
680
static inline struct sk_buff *
L
Linus Torvalds 已提交
681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703
cbq_dequeue_prio(struct Qdisc *sch, int prio)
{
	struct cbq_sched_data *q = qdisc_priv(sch);
	struct cbq_class *cl_tail, *cl_prev, *cl;
	struct sk_buff *skb;
	int deficit;

	cl_tail = cl_prev = q->active[prio];
	cl = cl_prev->next_alive;

	do {
		deficit = 0;

		/* Start round */
		do {
			struct cbq_class *borrow = cl;

			if (cl->q->q.qlen &&
			    (borrow = cbq_under_limit(cl)) == NULL)
				goto skip_class;

			if (cl->deficit <= 0) {
				/* Class exhausted its allotment per
E
Eric Dumazet 已提交
704
				 * this round. Switch to the next one.
L
Linus Torvalds 已提交
705 706 707 708 709 710 711 712 713
				 */
				deficit = 1;
				cl->deficit += cl->quantum;
				goto next_class;
			}

			skb = cl->q->dequeue(cl->q);

			/* Class did not give us any skb :-(
E
Eric Dumazet 已提交
714 715
			 * It could occur even if cl->q->q.qlen != 0
			 * f.e. if cl->q == "tbf"
L
Linus Torvalds 已提交
716 717 718 719
			 */
			if (skb == NULL)
				goto skip_class;

720
			cl->deficit -= qdisc_pkt_len(skb);
L
Linus Torvalds 已提交
721 722 723 724 725 726 727
			q->tx_class = cl;
			q->tx_borrowed = borrow;
			if (borrow != cl) {
#ifndef CBQ_XSTATS_BORROWS_BYTES
				borrow->xstats.borrows++;
				cl->xstats.borrows++;
#else
728 729
				borrow->xstats.borrows += qdisc_pkt_len(skb);
				cl->xstats.borrows += qdisc_pkt_len(skb);
L
Linus Torvalds 已提交
730 731
#endif
			}
732
			q->tx_len = qdisc_pkt_len(skb);
L
Linus Torvalds 已提交
733 734 735 736 737 738 739 740 741 742 743

			if (cl->deficit <= 0) {
				q->active[prio] = cl;
				cl = cl->next_alive;
				cl->deficit += cl->quantum;
			}
			return skb;

skip_class:
			if (cl->q->q.qlen == 0 || prio != cl->cpriority) {
				/* Class is empty or penalized.
E
Eric Dumazet 已提交
744
				 * Unlink it from active chain.
L
Linus Torvalds 已提交
745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782
				 */
				cl_prev->next_alive = cl->next_alive;
				cl->next_alive = NULL;

				/* Did cl_tail point to it? */
				if (cl == cl_tail) {
					/* Repair it! */
					cl_tail = cl_prev;

					/* Was it the last class in this band? */
					if (cl == cl_tail) {
						/* Kill the band! */
						q->active[prio] = NULL;
						q->activemask &= ~(1<<prio);
						if (cl->q->q.qlen)
							cbq_activate_class(cl);
						return NULL;
					}

					q->active[prio] = cl_tail;
				}
				if (cl->q->q.qlen)
					cbq_activate_class(cl);

				cl = cl_prev;
			}

next_class:
			cl_prev = cl;
			cl = cl->next_alive;
		} while (cl_prev != cl_tail);
	} while (deficit);

	q->active[prio] = cl_prev;

	return NULL;
}

E
Eric Dumazet 已提交
783
static inline struct sk_buff *
L
Linus Torvalds 已提交
784 785 786 787
cbq_dequeue_1(struct Qdisc *sch)
{
	struct cbq_sched_data *q = qdisc_priv(sch);
	struct sk_buff *skb;
E
Eric Dumazet 已提交
788
	unsigned int activemask;
L
Linus Torvalds 已提交
789

E
Eric Dumazet 已提交
790
	activemask = q->activemask & 0xFF;
L
Linus Torvalds 已提交
791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807
	while (activemask) {
		int prio = ffz(~activemask);
		activemask &= ~(1<<prio);
		skb = cbq_dequeue_prio(sch, prio);
		if (skb)
			return skb;
	}
	return NULL;
}

static struct sk_buff *
cbq_dequeue(struct Qdisc *sch)
{
	struct sk_buff *skb;
	struct cbq_sched_data *q = qdisc_priv(sch);
	psched_time_t now;

808
	now = psched_get_time();
809 810

	if (q->tx_class)
L
Linus Torvalds 已提交
811
		cbq_update(q);
812 813

	q->now = now;
L
Linus Torvalds 已提交
814 815 816 817 818 819

	for (;;) {
		q->wd_expires = 0;

		skb = cbq_dequeue_1(sch);
		if (skb) {
820
			qdisc_bstats_update(sch, skb);
L
Linus Torvalds 已提交
821
			sch->q.qlen--;
822
			qdisc_unthrottled(sch);
L
Linus Torvalds 已提交
823 824 825 826
			return skb;
		}

		/* All the classes are overlimit.
E
Eric Dumazet 已提交
827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842
		 *
		 * It is possible, if:
		 *
		 * 1. Scheduler is empty.
		 * 2. Toplevel cutoff inhibited borrowing.
		 * 3. Root class is overlimit.
		 *
		 * Reset 2d and 3d conditions and retry.
		 *
		 * Note, that NS and cbq-2.0 are buggy, peeking
		 * an arbitrary class is appropriate for ancestor-only
		 * sharing, but not for toplevel algorithm.
		 *
		 * Our version is better, but slower, because it requires
		 * two passes, but it is unavoidable with top-level sharing.
		 */
L
Linus Torvalds 已提交
843 844

		if (q->toplevel == TC_CBQ_MAXLEVEL &&
845
		    q->link.undertime == PSCHED_PASTPERFECT)
L
Linus Torvalds 已提交
846 847 848
			break;

		q->toplevel = TC_CBQ_MAXLEVEL;
849
		q->link.undertime = PSCHED_PASTPERFECT;
L
Linus Torvalds 已提交
850 851 852
	}

	/* No packets in scheduler or nobody wants to give them to us :-(
E
Eric Dumazet 已提交
853 854
	 * Sigh... start watchdog timer in the last case.
	 */
L
Linus Torvalds 已提交
855 856

	if (sch->q.qlen) {
857
		qdisc_qstats_overlimit(sch);
858 859
		if (q->wd_expires)
			qdisc_watchdog_schedule(&q->watchdog,
860
						now + q->wd_expires);
L
Linus Torvalds 已提交
861 862 863 864 865 866 867 868 869 870 871 872 873 874 875
	}
	return NULL;
}

/* CBQ class maintanance routines */

static void cbq_adjust_levels(struct cbq_class *this)
{
	if (this == NULL)
		return;

	do {
		int level = 0;
		struct cbq_class *cl;

E
Eric Dumazet 已提交
876 877
		cl = this->children;
		if (cl) {
L
Linus Torvalds 已提交
878 879 880 881 882
			do {
				if (cl->level > level)
					level = cl->level;
			} while ((cl = cl->sibling) != this->children);
		}
E
Eric Dumazet 已提交
883
		this->level = level + 1;
L
Linus Torvalds 已提交
884 885 886 887 888 889
	} while ((this = this->tparent) != NULL);
}

static void cbq_normalize_quanta(struct cbq_sched_data *q, int prio)
{
	struct cbq_class *cl;
890
	unsigned int h;
L
Linus Torvalds 已提交
891 892 893 894

	if (q->quanta[prio] == 0)
		return;

895
	for (h = 0; h < q->clhash.hashsize; h++) {
896
		hlist_for_each_entry(cl, &q->clhash.hash[h], common.hnode) {
L
Linus Torvalds 已提交
897
			/* BUGGGG... Beware! This expression suffer of
E
Eric Dumazet 已提交
898
			 * arithmetic overflows!
L
Linus Torvalds 已提交
899 900 901 902 903
			 */
			if (cl->priority == prio) {
				cl->quantum = (cl->weight*cl->allot*q->nclasses[prio])/
					q->quanta[prio];
			}
904 905
			if (cl->quantum <= 0 ||
			    cl->quantum > 32*qdisc_dev(cl->qdisc)->mtu) {
906 907
				pr_warn("CBQ: class %08x has bad quantum==%ld, repaired.\n",
					cl->common.classid, cl->quantum);
908
				cl->quantum = qdisc_dev(cl->qdisc)->mtu/2 + 1;
L
Linus Torvalds 已提交
909 910 911 912 913 914 915 916 917
			}
		}
	}
}

static void cbq_sync_defmap(struct cbq_class *cl)
{
	struct cbq_sched_data *q = qdisc_priv(cl->qdisc);
	struct cbq_class *split = cl->split;
E
Eric Dumazet 已提交
918
	unsigned int h;
L
Linus Torvalds 已提交
919 920 921 922 923
	int i;

	if (split == NULL)
		return;

E
Eric Dumazet 已提交
924 925
	for (i = 0; i <= TC_PRIO_MAX; i++) {
		if (split->defaults[i] == cl && !(cl->defmap & (1<<i)))
L
Linus Torvalds 已提交
926 927 928
			split->defaults[i] = NULL;
	}

E
Eric Dumazet 已提交
929
	for (i = 0; i <= TC_PRIO_MAX; i++) {
L
Linus Torvalds 已提交
930 931 932 933 934
		int level = split->level;

		if (split->defaults[i])
			continue;

935
		for (h = 0; h < q->clhash.hashsize; h++) {
L
Linus Torvalds 已提交
936 937
			struct cbq_class *c;

938
			hlist_for_each_entry(c, &q->clhash.hash[h],
939
					     common.hnode) {
L
Linus Torvalds 已提交
940
				if (c->split == split && c->level < level &&
E
Eric Dumazet 已提交
941
				    c->defmap & (1<<i)) {
L
Linus Torvalds 已提交
942 943 944 945 946 947 948 949 950 951 952 953 954
					split->defaults[i] = c;
					level = c->level;
				}
			}
		}
	}
}

static void cbq_change_defmap(struct cbq_class *cl, u32 splitid, u32 def, u32 mask)
{
	struct cbq_class *split = NULL;

	if (splitid == 0) {
E
Eric Dumazet 已提交
955 956
		split = cl->split;
		if (!split)
L
Linus Torvalds 已提交
957
			return;
958
		splitid = split->common.classid;
L
Linus Torvalds 已提交
959 960
	}

961
	if (split == NULL || split->common.classid != splitid) {
L
Linus Torvalds 已提交
962
		for (split = cl->tparent; split; split = split->tparent)
963
			if (split->common.classid == splitid)
L
Linus Torvalds 已提交
964 965 966 967 968 969 970 971 972 973
				break;
	}

	if (split == NULL)
		return;

	if (cl->split != split) {
		cl->defmap = 0;
		cbq_sync_defmap(cl);
		cl->split = split;
E
Eric Dumazet 已提交
974
		cl->defmap = def & mask;
L
Linus Torvalds 已提交
975
	} else
E
Eric Dumazet 已提交
976
		cl->defmap = (cl->defmap & ~mask) | (def & mask);
L
Linus Torvalds 已提交
977 978 979 980 981 982 983 984 985

	cbq_sync_defmap(cl);
}

static void cbq_unlink_class(struct cbq_class *this)
{
	struct cbq_class *cl, **clp;
	struct cbq_sched_data *q = qdisc_priv(this->qdisc);

986
	qdisc_class_hash_remove(&q->clhash, &this->common);
L
Linus Torvalds 已提交
987 988

	if (this->tparent) {
E
Eric Dumazet 已提交
989
		clp = &this->sibling;
L
Linus Torvalds 已提交
990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004
		cl = *clp;
		do {
			if (cl == this) {
				*clp = cl->sibling;
				break;
			}
			clp = &cl->sibling;
		} while ((cl = *clp) != this->sibling);

		if (this->tparent->children == this) {
			this->tparent->children = this->sibling;
			if (this->sibling == this)
				this->tparent->children = NULL;
		}
	} else {
1005
		WARN_ON(this->sibling != this);
L
Linus Torvalds 已提交
1006 1007 1008 1009 1010 1011 1012 1013 1014
	}
}

static void cbq_link_class(struct cbq_class *this)
{
	struct cbq_sched_data *q = qdisc_priv(this->qdisc);
	struct cbq_class *parent = this->tparent;

	this->sibling = this;
1015
	qdisc_class_hash_insert(&q->clhash, &this->common);
L
Linus Torvalds 已提交
1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028

	if (parent == NULL)
		return;

	if (parent->children == NULL) {
		parent->children = this;
	} else {
		this->sibling = parent->children->sibling;
		parent->children->sibling = this;
	}
}

static void
E
Eric Dumazet 已提交
1029
cbq_reset(struct Qdisc *sch)
L
Linus Torvalds 已提交
1030 1031 1032 1033
{
	struct cbq_sched_data *q = qdisc_priv(sch);
	struct cbq_class *cl;
	int prio;
E
Eric Dumazet 已提交
1034
	unsigned int h;
L
Linus Torvalds 已提交
1035 1036 1037 1038 1039

	q->activemask = 0;
	q->pmask = 0;
	q->tx_class = NULL;
	q->tx_borrowed = NULL;
1040
	qdisc_watchdog_cancel(&q->watchdog);
1041
	hrtimer_cancel(&q->delay_timer);
L
Linus Torvalds 已提交
1042
	q->toplevel = TC_CBQ_MAXLEVEL;
1043
	q->now = psched_get_time();
L
Linus Torvalds 已提交
1044 1045 1046 1047

	for (prio = 0; prio <= TC_CBQ_MAXPRIO; prio++)
		q->active[prio] = NULL;

1048
	for (h = 0; h < q->clhash.hashsize; h++) {
1049
		hlist_for_each_entry(cl, &q->clhash.hash[h], common.hnode) {
L
Linus Torvalds 已提交
1050 1051 1052
			qdisc_reset(cl->q);

			cl->next_alive = NULL;
1053
			cl->undertime = PSCHED_PASTPERFECT;
L
Linus Torvalds 已提交
1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064
			cl->avgidle = cl->maxidle;
			cl->deficit = cl->quantum;
			cl->cpriority = cl->priority;
		}
	}
	sch->q.qlen = 0;
}


static int cbq_set_lss(struct cbq_class *cl, struct tc_cbq_lssopt *lss)
{
E
Eric Dumazet 已提交
1065 1066 1067
	if (lss->change & TCF_CBQ_LSS_FLAGS) {
		cl->share = (lss->flags & TCF_CBQ_LSS_ISOLATED) ? NULL : cl->tparent;
		cl->borrow = (lss->flags & TCF_CBQ_LSS_BOUNDED) ? NULL : cl->tparent;
L
Linus Torvalds 已提交
1068
	}
E
Eric Dumazet 已提交
1069
	if (lss->change & TCF_CBQ_LSS_EWMA)
L
Linus Torvalds 已提交
1070
		cl->ewma_log = lss->ewma_log;
E
Eric Dumazet 已提交
1071
	if (lss->change & TCF_CBQ_LSS_AVPKT)
L
Linus Torvalds 已提交
1072
		cl->avpkt = lss->avpkt;
E
Eric Dumazet 已提交
1073
	if (lss->change & TCF_CBQ_LSS_MINIDLE)
L
Linus Torvalds 已提交
1074
		cl->minidle = -(long)lss->minidle;
E
Eric Dumazet 已提交
1075
	if (lss->change & TCF_CBQ_LSS_MAXIDLE) {
L
Linus Torvalds 已提交
1076 1077 1078
		cl->maxidle = lss->maxidle;
		cl->avgidle = lss->maxidle;
	}
E
Eric Dumazet 已提交
1079
	if (lss->change & TCF_CBQ_LSS_OFFTIME)
L
Linus Torvalds 已提交
1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106
		cl->offtime = lss->offtime;
	return 0;
}

static void cbq_rmprio(struct cbq_sched_data *q, struct cbq_class *cl)
{
	q->nclasses[cl->priority]--;
	q->quanta[cl->priority] -= cl->weight;
	cbq_normalize_quanta(q, cl->priority);
}

static void cbq_addprio(struct cbq_sched_data *q, struct cbq_class *cl)
{
	q->nclasses[cl->priority]++;
	q->quanta[cl->priority] += cl->weight;
	cbq_normalize_quanta(q, cl->priority);
}

static int cbq_set_wrr(struct cbq_class *cl, struct tc_cbq_wrropt *wrr)
{
	struct cbq_sched_data *q = qdisc_priv(cl->qdisc);

	if (wrr->allot)
		cl->allot = wrr->allot;
	if (wrr->weight)
		cl->weight = wrr->weight;
	if (wrr->priority) {
E
Eric Dumazet 已提交
1107
		cl->priority = wrr->priority - 1;
L
Linus Torvalds 已提交
1108 1109
		cl->cpriority = cl->priority;
		if (cl->priority >= cl->priority2)
E
Eric Dumazet 已提交
1110
			cl->priority2 = TC_CBQ_MAXPRIO - 1;
L
Linus Torvalds 已提交
1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122
	}

	cbq_addprio(q, cl);
	return 0;
}

static int cbq_set_fopt(struct cbq_class *cl, struct tc_cbq_fopt *fopt)
{
	cbq_change_defmap(cl, fopt->split, fopt->defmap, fopt->defchange);
	return 0;
}

1123 1124 1125 1126 1127 1128 1129 1130 1131 1132
static const struct nla_policy cbq_policy[TCA_CBQ_MAX + 1] = {
	[TCA_CBQ_LSSOPT]	= { .len = sizeof(struct tc_cbq_lssopt) },
	[TCA_CBQ_WRROPT]	= { .len = sizeof(struct tc_cbq_wrropt) },
	[TCA_CBQ_FOPT]		= { .len = sizeof(struct tc_cbq_fopt) },
	[TCA_CBQ_OVL_STRATEGY]	= { .len = sizeof(struct tc_cbq_ovl) },
	[TCA_CBQ_RATE]		= { .len = sizeof(struct tc_ratespec) },
	[TCA_CBQ_RTAB]		= { .type = NLA_BINARY, .len = TC_RTAB_SIZE },
	[TCA_CBQ_POLICE]	= { .len = sizeof(struct tc_cbq_police) },
};

1133
static int cbq_init(struct Qdisc *sch, struct nlattr *opt)
L
Linus Torvalds 已提交
1134 1135
{
	struct cbq_sched_data *q = qdisc_priv(sch);
1136
	struct nlattr *tb[TCA_CBQ_MAX + 1];
L
Linus Torvalds 已提交
1137
	struct tc_ratespec *r;
1138 1139
	int err;

1140
	err = nla_parse_nested(tb, TCA_CBQ_MAX, opt, cbq_policy);
1141 1142
	if (err < 0)
		return err;
L
Linus Torvalds 已提交
1143

1144
	if (tb[TCA_CBQ_RTAB] == NULL || tb[TCA_CBQ_RATE] == NULL)
L
Linus Torvalds 已提交
1145 1146
		return -EINVAL;

1147
	r = nla_data(tb[TCA_CBQ_RATE]);
L
Linus Torvalds 已提交
1148

1149
	if ((q->link.R_tab = qdisc_get_rtab(r, tb[TCA_CBQ_RTAB])) == NULL)
L
Linus Torvalds 已提交
1150 1151
		return -EINVAL;

1152 1153 1154 1155
	err = qdisc_class_hash_init(&q->clhash);
	if (err < 0)
		goto put_rtab;

L
Linus Torvalds 已提交
1156 1157
	q->link.refcnt = 1;
	q->link.sibling = &q->link;
1158
	q->link.common.classid = sch->handle;
L
Linus Torvalds 已提交
1159
	q->link.qdisc = sch;
1160 1161 1162
	q->link.q = qdisc_create_dflt(sch->dev_queue, &pfifo_qdisc_ops,
				      sch->handle);
	if (!q->link.q)
L
Linus Torvalds 已提交
1163 1164
		q->link.q = &noop_qdisc;

E
Eric Dumazet 已提交
1165 1166 1167
	q->link.priority = TC_CBQ_MAXPRIO - 1;
	q->link.priority2 = TC_CBQ_MAXPRIO - 1;
	q->link.cpriority = TC_CBQ_MAXPRIO - 1;
1168
	q->link.allot = psched_mtu(qdisc_dev(sch));
L
Linus Torvalds 已提交
1169 1170 1171 1172 1173 1174 1175
	q->link.quantum = q->link.allot;
	q->link.weight = q->link.R_tab->rate.rate;

	q->link.ewma_log = TC_CBQ_DEF_EWMA;
	q->link.avpkt = q->link.allot/2;
	q->link.minidle = -0x7FFFFFFF;

1176
	qdisc_watchdog_init(&q->watchdog, sch);
E
Eric Dumazet 已提交
1177
	hrtimer_init(&q->delay_timer, CLOCK_MONOTONIC, HRTIMER_MODE_ABS_PINNED);
L
Linus Torvalds 已提交
1178 1179
	q->delay_timer.function = cbq_undelay;
	q->toplevel = TC_CBQ_MAXLEVEL;
1180
	q->now = psched_get_time();
L
Linus Torvalds 已提交
1181 1182 1183

	cbq_link_class(&q->link);

1184 1185
	if (tb[TCA_CBQ_LSSOPT])
		cbq_set_lss(&q->link, nla_data(tb[TCA_CBQ_LSSOPT]));
L
Linus Torvalds 已提交
1186 1187 1188

	cbq_addprio(q, &q->link);
	return 0;
1189 1190 1191 1192

put_rtab:
	qdisc_put_rtab(q->link.R_tab);
	return err;
L
Linus Torvalds 已提交
1193 1194
}

E
Eric Dumazet 已提交
1195
static int cbq_dump_rate(struct sk_buff *skb, struct cbq_class *cl)
L
Linus Torvalds 已提交
1196
{
1197
	unsigned char *b = skb_tail_pointer(skb);
L
Linus Torvalds 已提交
1198

1199 1200
	if (nla_put(skb, TCA_CBQ_RATE, sizeof(cl->R_tab->rate), &cl->R_tab->rate))
		goto nla_put_failure;
L
Linus Torvalds 已提交
1201 1202
	return skb->len;

1203
nla_put_failure:
1204
	nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
1205 1206 1207
	return -1;
}

E
Eric Dumazet 已提交
1208
static int cbq_dump_lss(struct sk_buff *skb, struct cbq_class *cl)
L
Linus Torvalds 已提交
1209
{
1210
	unsigned char *b = skb_tail_pointer(skb);
L
Linus Torvalds 已提交
1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224
	struct tc_cbq_lssopt opt;

	opt.flags = 0;
	if (cl->borrow == NULL)
		opt.flags |= TCF_CBQ_LSS_BOUNDED;
	if (cl->share == NULL)
		opt.flags |= TCF_CBQ_LSS_ISOLATED;
	opt.ewma_log = cl->ewma_log;
	opt.level = cl->level;
	opt.avpkt = cl->avpkt;
	opt.maxidle = cl->maxidle;
	opt.minidle = (u32)(-cl->minidle);
	opt.offtime = cl->offtime;
	opt.change = ~0;
1225 1226
	if (nla_put(skb, TCA_CBQ_LSSOPT, sizeof(opt), &opt))
		goto nla_put_failure;
L
Linus Torvalds 已提交
1227 1228
	return skb->len;

1229
nla_put_failure:
1230
	nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
1231 1232 1233
	return -1;
}

E
Eric Dumazet 已提交
1234
static int cbq_dump_wrr(struct sk_buff *skb, struct cbq_class *cl)
L
Linus Torvalds 已提交
1235
{
1236
	unsigned char *b = skb_tail_pointer(skb);
L
Linus Torvalds 已提交
1237 1238
	struct tc_cbq_wrropt opt;

1239
	memset(&opt, 0, sizeof(opt));
L
Linus Torvalds 已提交
1240 1241
	opt.flags = 0;
	opt.allot = cl->allot;
E
Eric Dumazet 已提交
1242 1243
	opt.priority = cl->priority + 1;
	opt.cpriority = cl->cpriority + 1;
L
Linus Torvalds 已提交
1244
	opt.weight = cl->weight;
1245 1246
	if (nla_put(skb, TCA_CBQ_WRROPT, sizeof(opt), &opt))
		goto nla_put_failure;
L
Linus Torvalds 已提交
1247 1248
	return skb->len;

1249
nla_put_failure:
1250
	nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
1251 1252 1253
	return -1;
}

E
Eric Dumazet 已提交
1254
static int cbq_dump_fopt(struct sk_buff *skb, struct cbq_class *cl)
L
Linus Torvalds 已提交
1255
{
1256
	unsigned char *b = skb_tail_pointer(skb);
L
Linus Torvalds 已提交
1257 1258 1259
	struct tc_cbq_fopt opt;

	if (cl->split || cl->defmap) {
1260
		opt.split = cl->split ? cl->split->common.classid : 0;
L
Linus Torvalds 已提交
1261 1262
		opt.defmap = cl->defmap;
		opt.defchange = ~0;
1263 1264
		if (nla_put(skb, TCA_CBQ_FOPT, sizeof(opt), &opt))
			goto nla_put_failure;
L
Linus Torvalds 已提交
1265 1266 1267
	}
	return skb->len;

1268
nla_put_failure:
1269
	nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285
	return -1;
}

static int cbq_dump_attr(struct sk_buff *skb, struct cbq_class *cl)
{
	if (cbq_dump_lss(skb, cl) < 0 ||
	    cbq_dump_rate(skb, cl) < 0 ||
	    cbq_dump_wrr(skb, cl) < 0 ||
	    cbq_dump_fopt(skb, cl) < 0)
		return -1;
	return 0;
}

static int cbq_dump(struct Qdisc *sch, struct sk_buff *skb)
{
	struct cbq_sched_data *q = qdisc_priv(sch);
1286
	struct nlattr *nest;
L
Linus Torvalds 已提交
1287

1288 1289 1290
	nest = nla_nest_start(skb, TCA_OPTIONS);
	if (nest == NULL)
		goto nla_put_failure;
L
Linus Torvalds 已提交
1291
	if (cbq_dump_attr(skb, &q->link) < 0)
1292
		goto nla_put_failure;
1293
	return nla_nest_end(skb, nest);
L
Linus Torvalds 已提交
1294

1295
nla_put_failure:
1296
	nla_nest_cancel(skb, nest);
L
Linus Torvalds 已提交
1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312
	return -1;
}

static int
cbq_dump_stats(struct Qdisc *sch, struct gnet_dump *d)
{
	struct cbq_sched_data *q = qdisc_priv(sch);

	q->link.xstats.avgidle = q->link.avgidle;
	return gnet_stats_copy_app(d, &q->link.xstats, sizeof(q->link.xstats));
}

static int
cbq_dump_class(struct Qdisc *sch, unsigned long arg,
	       struct sk_buff *skb, struct tcmsg *tcm)
{
E
Eric Dumazet 已提交
1313
	struct cbq_class *cl = (struct cbq_class *)arg;
1314
	struct nlattr *nest;
L
Linus Torvalds 已提交
1315 1316

	if (cl->tparent)
1317
		tcm->tcm_parent = cl->tparent->common.classid;
L
Linus Torvalds 已提交
1318 1319
	else
		tcm->tcm_parent = TC_H_ROOT;
1320
	tcm->tcm_handle = cl->common.classid;
L
Linus Torvalds 已提交
1321 1322
	tcm->tcm_info = cl->q->handle;

1323 1324 1325
	nest = nla_nest_start(skb, TCA_OPTIONS);
	if (nest == NULL)
		goto nla_put_failure;
L
Linus Torvalds 已提交
1326
	if (cbq_dump_attr(skb, cl) < 0)
1327
		goto nla_put_failure;
1328
	return nla_nest_end(skb, nest);
L
Linus Torvalds 已提交
1329

1330
nla_put_failure:
1331
	nla_nest_cancel(skb, nest);
L
Linus Torvalds 已提交
1332 1333 1334 1335 1336 1337 1338 1339
	return -1;
}

static int
cbq_dump_class_stats(struct Qdisc *sch, unsigned long arg,
	struct gnet_dump *d)
{
	struct cbq_sched_data *q = qdisc_priv(sch);
E
Eric Dumazet 已提交
1340
	struct cbq_class *cl = (struct cbq_class *)arg;
L
Linus Torvalds 已提交
1341 1342 1343 1344

	cl->xstats.avgidle = cl->avgidle;
	cl->xstats.undertime = 0;

1345
	if (cl->undertime != PSCHED_PASTPERFECT)
P
Patrick McHardy 已提交
1346
		cl->xstats.undertime = cl->undertime - q->now;
L
Linus Torvalds 已提交
1347

1348 1349
	if (gnet_stats_copy_basic(qdisc_root_sleeping_running(sch),
				  d, NULL, &cl->bstats) < 0 ||
1350
	    gnet_stats_copy_rate_est(d, &cl->bstats, &cl->rate_est) < 0 ||
1351
	    gnet_stats_copy_queue(d, NULL, &cl->qstats, cl->q->q.qlen) < 0)
L
Linus Torvalds 已提交
1352 1353 1354 1355 1356 1357 1358 1359
		return -1;

	return gnet_stats_copy_app(d, &cl->xstats, sizeof(cl->xstats));
}

static int cbq_graft(struct Qdisc *sch, unsigned long arg, struct Qdisc *new,
		     struct Qdisc **old)
{
E
Eric Dumazet 已提交
1360
	struct cbq_class *cl = (struct cbq_class *)arg;
L
Linus Torvalds 已提交
1361

1362
	if (new == NULL) {
1363
		new = qdisc_create_dflt(sch->dev_queue,
1364 1365 1366
					&pfifo_qdisc_ops, cl->common.classid);
		if (new == NULL)
			return -ENOBUFS;
L
Linus Torvalds 已提交
1367
	}
1368

1369
	*old = qdisc_replace(sch, new, &cl->q);
1370
	return 0;
L
Linus Torvalds 已提交
1371 1372
}

E
Eric Dumazet 已提交
1373
static struct Qdisc *cbq_leaf(struct Qdisc *sch, unsigned long arg)
L
Linus Torvalds 已提交
1374
{
E
Eric Dumazet 已提交
1375
	struct cbq_class *cl = (struct cbq_class *)arg;
L
Linus Torvalds 已提交
1376

1377
	return cl->q;
L
Linus Torvalds 已提交
1378 1379
}

1380 1381 1382 1383 1384 1385 1386 1387
static void cbq_qlen_notify(struct Qdisc *sch, unsigned long arg)
{
	struct cbq_class *cl = (struct cbq_class *)arg;

	if (cl->q->q.qlen == 0)
		cbq_deactivate_class(cl);
}

L
Linus Torvalds 已提交
1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403
static unsigned long cbq_get(struct Qdisc *sch, u32 classid)
{
	struct cbq_sched_data *q = qdisc_priv(sch);
	struct cbq_class *cl = cbq_class_lookup(q, classid);

	if (cl) {
		cl->refcnt++;
		return (unsigned long)cl;
	}
	return 0;
}

static void cbq_destroy_class(struct Qdisc *sch, struct cbq_class *cl)
{
	struct cbq_sched_data *q = qdisc_priv(sch);

1404
	WARN_ON(cl->filters);
L
Linus Torvalds 已提交
1405

1406
	tcf_destroy_chain(&cl->filter_list);
L
Linus Torvalds 已提交
1407 1408 1409 1410 1411 1412 1413
	qdisc_destroy(cl->q);
	qdisc_put_rtab(cl->R_tab);
	gen_kill_estimator(&cl->bstats, &cl->rate_est);
	if (cl != &q->link)
		kfree(cl);
}

E
Eric Dumazet 已提交
1414
static void cbq_destroy(struct Qdisc *sch)
L
Linus Torvalds 已提交
1415 1416
{
	struct cbq_sched_data *q = qdisc_priv(sch);
1417
	struct hlist_node *next;
L
Linus Torvalds 已提交
1418
	struct cbq_class *cl;
E
Eric Dumazet 已提交
1419
	unsigned int h;
L
Linus Torvalds 已提交
1420

1421
#ifdef CONFIG_NET_CLS_ACT
L
Linus Torvalds 已提交
1422 1423 1424 1425 1426 1427 1428
	q->rx_class = NULL;
#endif
	/*
	 * Filters must be destroyed first because we don't destroy the
	 * classes from root to leafs which means that filters can still
	 * be bound to classes which have been destroyed already. --TGR '04
	 */
1429
	for (h = 0; h < q->clhash.hashsize; h++) {
1430
		hlist_for_each_entry(cl, &q->clhash.hash[h], common.hnode)
1431
			tcf_destroy_chain(&cl->filter_list);
1432
	}
1433
	for (h = 0; h < q->clhash.hashsize; h++) {
1434
		hlist_for_each_entry_safe(cl, next, &q->clhash.hash[h],
1435
					  common.hnode)
L
Linus Torvalds 已提交
1436 1437
			cbq_destroy_class(sch, cl);
	}
1438
	qdisc_class_hash_destroy(&q->clhash);
L
Linus Torvalds 已提交
1439 1440 1441 1442
}

static void cbq_put(struct Qdisc *sch, unsigned long arg)
{
E
Eric Dumazet 已提交
1443
	struct cbq_class *cl = (struct cbq_class *)arg;
L
Linus Torvalds 已提交
1444 1445

	if (--cl->refcnt == 0) {
1446
#ifdef CONFIG_NET_CLS_ACT
1447
		spinlock_t *root_lock = qdisc_root_sleeping_lock(sch);
L
Linus Torvalds 已提交
1448 1449
		struct cbq_sched_data *q = qdisc_priv(sch);

1450
		spin_lock_bh(root_lock);
L
Linus Torvalds 已提交
1451 1452
		if (q->rx_class == cl)
			q->rx_class = NULL;
1453
		spin_unlock_bh(root_lock);
L
Linus Torvalds 已提交
1454 1455 1456 1457 1458 1459 1460
#endif

		cbq_destroy_class(sch, cl);
	}
}

static int
1461
cbq_change_class(struct Qdisc *sch, u32 classid, u32 parentid, struct nlattr **tca,
L
Linus Torvalds 已提交
1462 1463 1464 1465
		 unsigned long *arg)
{
	int err;
	struct cbq_sched_data *q = qdisc_priv(sch);
E
Eric Dumazet 已提交
1466
	struct cbq_class *cl = (struct cbq_class *)*arg;
1467 1468
	struct nlattr *opt = tca[TCA_OPTIONS];
	struct nlattr *tb[TCA_CBQ_MAX + 1];
L
Linus Torvalds 已提交
1469 1470 1471
	struct cbq_class *parent;
	struct qdisc_rate_table *rtab = NULL;

1472
	if (opt == NULL)
L
Linus Torvalds 已提交
1473 1474
		return -EINVAL;

1475
	err = nla_parse_nested(tb, TCA_CBQ_MAX, opt, cbq_policy);
1476 1477 1478
	if (err < 0)
		return err;

1479
	if (tb[TCA_CBQ_OVL_STRATEGY] || tb[TCA_CBQ_POLICE])
1480 1481
		return -EOPNOTSUPP;

L
Linus Torvalds 已提交
1482 1483 1484
	if (cl) {
		/* Check parent */
		if (parentid) {
1485 1486
			if (cl->tparent &&
			    cl->tparent->common.classid != parentid)
L
Linus Torvalds 已提交
1487 1488 1489 1490 1491
				return -EINVAL;
			if (!cl->tparent && parentid != TC_H_ROOT)
				return -EINVAL;
		}

1492
		if (tb[TCA_CBQ_RATE]) {
1493 1494
			rtab = qdisc_get_rtab(nla_data(tb[TCA_CBQ_RATE]),
					      tb[TCA_CBQ_RTAB]);
L
Linus Torvalds 已提交
1495 1496 1497 1498
			if (rtab == NULL)
				return -EINVAL;
		}

1499
		if (tca[TCA_RATE]) {
1500 1501
			err = gen_replace_estimator(&cl->bstats, NULL,
						    &cl->rate_est,
1502 1503
						    NULL,
						    qdisc_root_sleeping_running(sch),
1504 1505
						    tca[TCA_RATE]);
			if (err) {
1506
				qdisc_put_rtab(rtab);
1507 1508 1509 1510
				return err;
			}
		}

L
Linus Torvalds 已提交
1511 1512 1513 1514 1515 1516 1517
		/* Change class parameters */
		sch_tree_lock(sch);

		if (cl->next_alive != NULL)
			cbq_deactivate_class(cl);

		if (rtab) {
1518 1519
			qdisc_put_rtab(cl->R_tab);
			cl->R_tab = rtab;
L
Linus Torvalds 已提交
1520 1521
		}

1522 1523
		if (tb[TCA_CBQ_LSSOPT])
			cbq_set_lss(cl, nla_data(tb[TCA_CBQ_LSSOPT]));
L
Linus Torvalds 已提交
1524

1525
		if (tb[TCA_CBQ_WRROPT]) {
L
Linus Torvalds 已提交
1526
			cbq_rmprio(q, cl);
1527
			cbq_set_wrr(cl, nla_data(tb[TCA_CBQ_WRROPT]));
L
Linus Torvalds 已提交
1528 1529
		}

1530 1531
		if (tb[TCA_CBQ_FOPT])
			cbq_set_fopt(cl, nla_data(tb[TCA_CBQ_FOPT]));
L
Linus Torvalds 已提交
1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543

		if (cl->q->q.qlen)
			cbq_activate_class(cl);

		sch_tree_unlock(sch);

		return 0;
	}

	if (parentid == TC_H_ROOT)
		return -EINVAL;

1544 1545
	if (tb[TCA_CBQ_WRROPT] == NULL || tb[TCA_CBQ_RATE] == NULL ||
	    tb[TCA_CBQ_LSSOPT] == NULL)
L
Linus Torvalds 已提交
1546 1547
		return -EINVAL;

1548
	rtab = qdisc_get_rtab(nla_data(tb[TCA_CBQ_RATE]), tb[TCA_CBQ_RTAB]);
L
Linus Torvalds 已提交
1549 1550 1551 1552 1553
	if (rtab == NULL)
		return -EINVAL;

	if (classid) {
		err = -EINVAL;
E
Eric Dumazet 已提交
1554 1555
		if (TC_H_MAJ(classid ^ sch->handle) ||
		    cbq_class_lookup(q, classid))
L
Linus Torvalds 已提交
1556 1557 1558
			goto failure;
	} else {
		int i;
E
Eric Dumazet 已提交
1559
		classid = TC_H_MAKE(sch->handle, 0x8000);
L
Linus Torvalds 已提交
1560

E
Eric Dumazet 已提交
1561
		for (i = 0; i < 0x8000; i++) {
L
Linus Torvalds 已提交
1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581
			if (++q->hgenerator >= 0x8000)
				q->hgenerator = 1;
			if (cbq_class_lookup(q, classid|q->hgenerator) == NULL)
				break;
		}
		err = -ENOSR;
		if (i >= 0x8000)
			goto failure;
		classid = classid|q->hgenerator;
	}

	parent = &q->link;
	if (parentid) {
		parent = cbq_class_lookup(q, parentid);
		err = -EINVAL;
		if (parent == NULL)
			goto failure;
	}

	err = -ENOBUFS;
1582
	cl = kzalloc(sizeof(*cl), GFP_KERNEL);
L
Linus Torvalds 已提交
1583 1584
	if (cl == NULL)
		goto failure;
1585 1586

	if (tca[TCA_RATE]) {
1587
		err = gen_new_estimator(&cl->bstats, NULL, &cl->rate_est,
1588 1589
					NULL,
					qdisc_root_sleeping_running(sch),
1590 1591 1592 1593 1594 1595 1596
					tca[TCA_RATE]);
		if (err) {
			kfree(cl);
			goto failure;
		}
	}

L
Linus Torvalds 已提交
1597 1598 1599
	cl->R_tab = rtab;
	rtab = NULL;
	cl->refcnt = 1;
1600 1601
	cl->q = qdisc_create_dflt(sch->dev_queue, &pfifo_qdisc_ops, classid);
	if (!cl->q)
L
Linus Torvalds 已提交
1602
		cl->q = &noop_qdisc;
1603
	cl->common.classid = classid;
L
Linus Torvalds 已提交
1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616
	cl->tparent = parent;
	cl->qdisc = sch;
	cl->allot = parent->allot;
	cl->quantum = cl->allot;
	cl->weight = cl->R_tab->rate.rate;

	sch_tree_lock(sch);
	cbq_link_class(cl);
	cl->borrow = cl->tparent;
	if (cl->tparent != &q->link)
		cl->share = cl->tparent;
	cbq_adjust_levels(parent);
	cl->minidle = -0x7FFFFFFF;
1617 1618
	cbq_set_lss(cl, nla_data(tb[TCA_CBQ_LSSOPT]));
	cbq_set_wrr(cl, nla_data(tb[TCA_CBQ_WRROPT]));
E
Eric Dumazet 已提交
1619
	if (cl->ewma_log == 0)
L
Linus Torvalds 已提交
1620
		cl->ewma_log = q->link.ewma_log;
E
Eric Dumazet 已提交
1621
	if (cl->maxidle == 0)
L
Linus Torvalds 已提交
1622
		cl->maxidle = q->link.maxidle;
E
Eric Dumazet 已提交
1623
	if (cl->avpkt == 0)
L
Linus Torvalds 已提交
1624
		cl->avpkt = q->link.avpkt;
1625 1626
	if (tb[TCA_CBQ_FOPT])
		cbq_set_fopt(cl, nla_data(tb[TCA_CBQ_FOPT]));
L
Linus Torvalds 已提交
1627 1628
	sch_tree_unlock(sch);

1629 1630
	qdisc_class_hash_grow(sch, &q->clhash);

L
Linus Torvalds 已提交
1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641
	*arg = (unsigned long)cl;
	return 0;

failure:
	qdisc_put_rtab(rtab);
	return err;
}

static int cbq_delete(struct Qdisc *sch, unsigned long arg)
{
	struct cbq_sched_data *q = qdisc_priv(sch);
E
Eric Dumazet 已提交
1642
	struct cbq_class *cl = (struct cbq_class *)arg;
1643
	unsigned int qlen, backlog;
L
Linus Torvalds 已提交
1644 1645 1646 1647 1648 1649

	if (cl->filters || cl->children || cl == &q->link)
		return -EBUSY;

	sch_tree_lock(sch);

1650
	qlen = cl->q->q.qlen;
1651
	backlog = cl->q->qstats.backlog;
1652
	qdisc_reset(cl->q);
1653
	qdisc_tree_reduce_backlog(cl->q, qlen, backlog);
1654

L
Linus Torvalds 已提交
1655 1656 1657 1658 1659 1660 1661 1662 1663
	if (cl->next_alive)
		cbq_deactivate_class(cl);

	if (q->tx_borrowed == cl)
		q->tx_borrowed = q->tx_class;
	if (q->tx_class == cl) {
		q->tx_class = NULL;
		q->tx_borrowed = NULL;
	}
1664
#ifdef CONFIG_NET_CLS_ACT
L
Linus Torvalds 已提交
1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676
	if (q->rx_class == cl)
		q->rx_class = NULL;
#endif

	cbq_unlink_class(cl);
	cbq_adjust_levels(cl->tparent);
	cl->defmap = 0;
	cbq_sync_defmap(cl);

	cbq_rmprio(q, cl);
	sch_tree_unlock(sch);

1677 1678 1679 1680 1681
	BUG_ON(--cl->refcnt == 0);
	/*
	 * This shouldn't happen: we "hold" one cops->get() when called
	 * from tc_ctl_tclass; the destroy method is done from cops->put().
	 */
L
Linus Torvalds 已提交
1682 1683 1684 1685

	return 0;
}

J
John Fastabend 已提交
1686 1687
static struct tcf_proto __rcu **cbq_find_tcf(struct Qdisc *sch,
					     unsigned long arg)
L
Linus Torvalds 已提交
1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701
{
	struct cbq_sched_data *q = qdisc_priv(sch);
	struct cbq_class *cl = (struct cbq_class *)arg;

	if (cl == NULL)
		cl = &q->link;

	return &cl->filter_list;
}

static unsigned long cbq_bind_filter(struct Qdisc *sch, unsigned long parent,
				     u32 classid)
{
	struct cbq_sched_data *q = qdisc_priv(sch);
E
Eric Dumazet 已提交
1702
	struct cbq_class *p = (struct cbq_class *)parent;
L
Linus Torvalds 已提交
1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715
	struct cbq_class *cl = cbq_class_lookup(q, classid);

	if (cl) {
		if (p && p->level <= cl->level)
			return 0;
		cl->filters++;
		return (unsigned long)cl;
	}
	return 0;
}

static void cbq_unbind_filter(struct Qdisc *sch, unsigned long arg)
{
E
Eric Dumazet 已提交
1716
	struct cbq_class *cl = (struct cbq_class *)arg;
L
Linus Torvalds 已提交
1717 1718 1719 1720 1721 1722 1723

	cl->filters--;
}

static void cbq_walk(struct Qdisc *sch, struct qdisc_walker *arg)
{
	struct cbq_sched_data *q = qdisc_priv(sch);
1724
	struct cbq_class *cl;
E
Eric Dumazet 已提交
1725
	unsigned int h;
L
Linus Torvalds 已提交
1726 1727 1728 1729

	if (arg->stop)
		return;

1730
	for (h = 0; h < q->clhash.hashsize; h++) {
1731
		hlist_for_each_entry(cl, &q->clhash.hash[h], common.hnode) {
L
Linus Torvalds 已提交
1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744
			if (arg->count < arg->skip) {
				arg->count++;
				continue;
			}
			if (arg->fn(sch, (unsigned long)cl, arg) < 0) {
				arg->stop = 1;
				return;
			}
			arg->count++;
		}
	}
}

1745
static const struct Qdisc_class_ops cbq_class_ops = {
L
Linus Torvalds 已提交
1746 1747
	.graft		=	cbq_graft,
	.leaf		=	cbq_leaf,
1748
	.qlen_notify	=	cbq_qlen_notify,
L
Linus Torvalds 已提交
1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760
	.get		=	cbq_get,
	.put		=	cbq_put,
	.change		=	cbq_change_class,
	.delete		=	cbq_delete,
	.walk		=	cbq_walk,
	.tcf_chain	=	cbq_find_tcf,
	.bind_tcf	=	cbq_bind_filter,
	.unbind_tcf	=	cbq_unbind_filter,
	.dump		=	cbq_dump_class,
	.dump_stats	=	cbq_dump_class_stats,
};

1761
static struct Qdisc_ops cbq_qdisc_ops __read_mostly = {
L
Linus Torvalds 已提交
1762 1763 1764 1765 1766 1767
	.next		=	NULL,
	.cl_ops		=	&cbq_class_ops,
	.id		=	"cbq",
	.priv_size	=	sizeof(struct cbq_sched_data),
	.enqueue	=	cbq_enqueue,
	.dequeue	=	cbq_dequeue,
1768
	.peek		=	qdisc_peek_dequeued,
L
Linus Torvalds 已提交
1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781
	.init		=	cbq_init,
	.reset		=	cbq_reset,
	.destroy	=	cbq_destroy,
	.change		=	NULL,
	.dump		=	cbq_dump,
	.dump_stats	=	cbq_dump_stats,
	.owner		=	THIS_MODULE,
};

static int __init cbq_module_init(void)
{
	return register_qdisc(&cbq_qdisc_ops);
}
1782
static void __exit cbq_module_exit(void)
L
Linus Torvalds 已提交
1783 1784 1785 1786 1787 1788
{
	unregister_qdisc(&cbq_qdisc_ops);
}
module_init(cbq_module_init)
module_exit(cbq_module_exit)
MODULE_LICENSE("GPL");