Merge tag 'tty-3.10-rc4' of git://git.kernel.org/pub/scm/linux/kernel/git/gregkh/tty
[GitHub/mt8127/android_kernel_alcatel_ttab.git] / net / sched / sch_sfq.c
CommitLineData
1da177e4
LT
1/*
2 * net/sched/sch_sfq.c Stochastic Fairness Queueing discipline.
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: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
10 */
11
1da177e4 12#include <linux/module.h>
1da177e4
LT
13#include <linux/types.h>
14#include <linux/kernel.h>
15#include <linux/jiffies.h>
16#include <linux/string.h>
1da177e4
LT
17#include <linux/in.h>
18#include <linux/errno.h>
1da177e4 19#include <linux/init.h>
1da177e4 20#include <linux/skbuff.h>
32740ddc 21#include <linux/jhash.h>
5a0e3ad6 22#include <linux/slab.h>
817fb15d 23#include <linux/vmalloc.h>
0ba48053 24#include <net/netlink.h>
1da177e4 25#include <net/pkt_sched.h>
11fca931 26#include <net/flow_keys.h>
ddecf0f4 27#include <net/red.h>
1da177e4
LT
28
29
30/* Stochastic Fairness Queuing algorithm.
31 =======================================
32
33 Source:
34 Paul E. McKenney "Stochastic Fairness Queuing",
35 IEEE INFOCOMM'90 Proceedings, San Francisco, 1990.
36
37 Paul E. McKenney "Stochastic Fairness Queuing",
38 "Interworking: Research and Experience", v.2, 1991, p.113-131.
39
40
41 See also:
42 M. Shreedhar and George Varghese "Efficient Fair
43 Queuing using Deficit Round Robin", Proc. SIGCOMM 95.
44
45
10297b99 46 This is not the thing that is usually called (W)FQ nowadays.
1da177e4
LT
47 It does not use any timestamp mechanism, but instead
48 processes queues in round-robin order.
49
50 ADVANTAGE:
51
52 - It is very cheap. Both CPU and memory requirements are minimal.
53
54 DRAWBACKS:
55
10297b99 56 - "Stochastic" -> It is not 100% fair.
1da177e4
LT
57 When hash collisions occur, several flows are considered as one.
58
59 - "Round-robin" -> It introduces larger delays than virtual clock
60 based schemes, and should not be used for isolating interactive
61 traffic from non-interactive. It means, that this scheduler
62 should be used as leaf of CBQ or P3, which put interactive traffic
63 to higher priority band.
64
65 We still need true WFQ for top level CSZ, but using WFQ
66 for the best effort traffic is absolutely pointless:
67 SFQ is superior for this purpose.
68
69 IMPLEMENTATION:
18cb8098
ED
70 This implementation limits :
71 - maximal queue length per flow to 127 packets.
72 - max mtu to 2^18-1;
73 - max 65408 flows,
74 - number of hash buckets to 65536.
1da177e4
LT
75
76 It is easy to increase these values, but not in flight. */
77
18cb8098
ED
78#define SFQ_MAX_DEPTH 127 /* max number of packets per flow */
79#define SFQ_DEFAULT_FLOWS 128
80#define SFQ_MAX_FLOWS (0x10000 - SFQ_MAX_DEPTH - 1) /* max number of flows */
81#define SFQ_EMPTY_SLOT 0xffff
817fb15d
ED
82#define SFQ_DEFAULT_HASH_DIVISOR 1024
83
eeaeb068
ED
84/* We use 16 bits to store allot, and want to handle packets up to 64K
85 * Scale allot by 8 (1<<3) so that no overflow occurs.
86 */
87#define SFQ_ALLOT_SHIFT 3
88#define SFQ_ALLOT_SIZE(X) DIV_ROUND_UP(X, 1 << SFQ_ALLOT_SHIFT)
1da177e4 89
18cb8098
ED
90/* This type should contain at least SFQ_MAX_DEPTH + 1 + SFQ_MAX_FLOWS values */
91typedef u16 sfq_index;
1da177e4 92
eda83e3b
ED
93/*
94 * We dont use pointers to save space.
18cb8098
ED
95 * Small indexes [0 ... SFQ_MAX_FLOWS - 1] are 'pointers' to slots[] array
96 * while following values [SFQ_MAX_FLOWS ... SFQ_MAX_FLOWS + SFQ_MAX_DEPTH]
eda83e3b
ED
97 * are 'pointers' to dep[] array
98 */
cc7ec456 99struct sfq_head {
1da177e4
LT
100 sfq_index next;
101 sfq_index prev;
102};
103
eda83e3b
ED
104struct sfq_slot {
105 struct sk_buff *skblist_next;
106 struct sk_buff *skblist_prev;
107 sfq_index qlen; /* number of skbs in skblist */
18cb8098 108 sfq_index next; /* next slot in sfq RR chain */
eda83e3b
ED
109 struct sfq_head dep; /* anchor in dep[] chains */
110 unsigned short hash; /* hash value (index in ht[]) */
111 short allot; /* credit for this slot */
ddecf0f4
ED
112
113 unsigned int backlog;
114 struct red_vars vars;
eda83e3b
ED
115};
116
cc7ec456 117struct sfq_sched_data {
18cb8098
ED
118/* frequently used fields */
119 int limit; /* limit of total number of packets in this qdisc */
817fb15d 120 unsigned int divisor; /* number of slots in hash table */
ddecf0f4
ED
121 u8 headdrop;
122 u8 maxdepth; /* limit of packets per flow */
18cb8098 123
32740ddc 124 u32 perturbation;
ddecf0f4
ED
125 u8 cur_depth; /* depth of longest slot */
126 u8 flags;
eeaeb068 127 unsigned short scaled_quantum; /* SFQ_ALLOT_SIZE(quantum) */
ddecf0f4 128 struct tcf_proto *filter_list;
18cb8098
ED
129 sfq_index *ht; /* Hash table ('divisor' slots) */
130 struct sfq_slot *slots; /* Flows table ('maxflows' entries) */
131
ddecf0f4
ED
132 struct red_parms *red_parms;
133 struct tc_sfqred_stats stats;
134 struct sfq_slot *tail; /* current slot in round */
135
18cb8098
ED
136 struct sfq_head dep[SFQ_MAX_DEPTH + 1];
137 /* Linked lists of slots, indexed by depth
138 * dep[0] : list of unused flows
139 * dep[1] : list of flows with 1 packet
140 * dep[X] : list of flows with X packets
141 */
142
ddecf0f4 143 unsigned int maxflows; /* number of flows in flows array */
18cb8098
ED
144 int perturb_period;
145 unsigned int quantum; /* Allotment per round: MUST BE >= MTU */
146 struct timer_list perturb_timer;
1da177e4
LT
147};
148
eda83e3b
ED
149/*
150 * sfq_head are either in a sfq_slot or in dep[] array
151 */
152static inline struct sfq_head *sfq_dep_head(struct sfq_sched_data *q, sfq_index val)
153{
18cb8098 154 if (val < SFQ_MAX_FLOWS)
eda83e3b 155 return &q->slots[val].dep;
18cb8098 156 return &q->dep[val - SFQ_MAX_FLOWS];
eda83e3b
ED
157}
158
225d9b89
ED
159/*
160 * In order to be able to quickly rehash our queue when timer changes
161 * q->perturbation, we store flow_keys in skb->cb[]
162 */
163struct sfq_skb_cb {
164 struct flow_keys keys;
165};
166
167static inline struct sfq_skb_cb *sfq_skb_cb(const struct sk_buff *skb)
168{
16bda13d
DM
169 qdisc_cb_private_validate(skb, sizeof(struct sfq_skb_cb));
170 return (struct sfq_skb_cb *)qdisc_skb_cb(skb)->data;
225d9b89
ED
171}
172
11fca931
ED
173static unsigned int sfq_hash(const struct sfq_sched_data *q,
174 const struct sk_buff *skb)
1da177e4 175{
225d9b89 176 const struct flow_keys *keys = &sfq_skb_cb(skb)->keys;
11fca931
ED
177 unsigned int hash;
178
225d9b89
ED
179 hash = jhash_3words((__force u32)keys->dst,
180 (__force u32)keys->src ^ keys->ip_proto,
181 (__force u32)keys->ports, q->perturbation);
11fca931 182 return hash & (q->divisor - 1);
1da177e4
LT
183}
184
7d2681a6
PM
185static unsigned int sfq_classify(struct sk_buff *skb, struct Qdisc *sch,
186 int *qerr)
187{
188 struct sfq_sched_data *q = qdisc_priv(sch);
189 struct tcf_result res;
190 int result;
191
192 if (TC_H_MAJ(skb->priority) == sch->handle &&
193 TC_H_MIN(skb->priority) > 0 &&
817fb15d 194 TC_H_MIN(skb->priority) <= q->divisor)
7d2681a6
PM
195 return TC_H_MIN(skb->priority);
196
225d9b89
ED
197 if (!q->filter_list) {
198 skb_flow_dissect(skb, &sfq_skb_cb(skb)->keys);
7d2681a6 199 return sfq_hash(q, skb) + 1;
225d9b89 200 }
7d2681a6 201
c27f339a 202 *qerr = NET_XMIT_SUCCESS | __NET_XMIT_BYPASS;
7d2681a6
PM
203 result = tc_classify(skb, q->filter_list, &res);
204 if (result >= 0) {
205#ifdef CONFIG_NET_CLS_ACT
206 switch (result) {
207 case TC_ACT_STOLEN:
208 case TC_ACT_QUEUED:
378a2f09 209 *qerr = NET_XMIT_SUCCESS | __NET_XMIT_STOLEN;
7d2681a6
PM
210 case TC_ACT_SHOT:
211 return 0;
212 }
213#endif
817fb15d 214 if (TC_H_MIN(res.classid) <= q->divisor)
7d2681a6
PM
215 return TC_H_MIN(res.classid);
216 }
217 return 0;
218}
219
eda83e3b 220/*
18cb8098 221 * x : slot number [0 .. SFQ_MAX_FLOWS - 1]
eda83e3b 222 */
1da177e4
LT
223static inline void sfq_link(struct sfq_sched_data *q, sfq_index x)
224{
225 sfq_index p, n;
18cb8098
ED
226 struct sfq_slot *slot = &q->slots[x];
227 int qlen = slot->qlen;
eda83e3b 228
18cb8098 229 p = qlen + SFQ_MAX_FLOWS;
eda83e3b 230 n = q->dep[qlen].next;
1da177e4 231
18cb8098
ED
232 slot->dep.next = n;
233 slot->dep.prev = p;
eda83e3b
ED
234
235 q->dep[qlen].next = x; /* sfq_dep_head(q, p)->next = x */
236 sfq_dep_head(q, n)->prev = x;
1da177e4
LT
237}
238
eda83e3b
ED
239#define sfq_unlink(q, x, n, p) \
240 n = q->slots[x].dep.next; \
241 p = q->slots[x].dep.prev; \
242 sfq_dep_head(q, p)->next = n; \
243 sfq_dep_head(q, n)->prev = p
244
245
1da177e4
LT
246static inline void sfq_dec(struct sfq_sched_data *q, sfq_index x)
247{
248 sfq_index p, n;
eda83e3b 249 int d;
1da177e4 250
eda83e3b 251 sfq_unlink(q, x, n, p);
1da177e4 252
eda83e3b
ED
253 d = q->slots[x].qlen--;
254 if (n == p && q->cur_depth == d)
255 q->cur_depth--;
1da177e4
LT
256 sfq_link(q, x);
257}
258
259static inline void sfq_inc(struct sfq_sched_data *q, sfq_index x)
260{
261 sfq_index p, n;
262 int d;
263
eda83e3b 264 sfq_unlink(q, x, n, p);
1da177e4 265
eda83e3b
ED
266 d = ++q->slots[x].qlen;
267 if (q->cur_depth < d)
268 q->cur_depth = d;
1da177e4
LT
269 sfq_link(q, x);
270}
271
eda83e3b
ED
272/* helper functions : might be changed when/if skb use a standard list_head */
273
274/* remove one skb from tail of slot queue */
275static inline struct sk_buff *slot_dequeue_tail(struct sfq_slot *slot)
276{
277 struct sk_buff *skb = slot->skblist_prev;
278
279 slot->skblist_prev = skb->prev;
ee09b3c1 280 skb->prev->next = (struct sk_buff *)slot;
eda83e3b
ED
281 skb->next = skb->prev = NULL;
282 return skb;
283}
284
285/* remove one skb from head of slot queue */
286static inline struct sk_buff *slot_dequeue_head(struct sfq_slot *slot)
287{
288 struct sk_buff *skb = slot->skblist_next;
289
290 slot->skblist_next = skb->next;
18c8d82a 291 skb->next->prev = (struct sk_buff *)slot;
eda83e3b
ED
292 skb->next = skb->prev = NULL;
293 return skb;
294}
295
296static inline void slot_queue_init(struct sfq_slot *slot)
297{
18cb8098 298 memset(slot, 0, sizeof(*slot));
eda83e3b
ED
299 slot->skblist_prev = slot->skblist_next = (struct sk_buff *)slot;
300}
301
302/* add skb to slot queue (tail add) */
303static inline void slot_queue_add(struct sfq_slot *slot, struct sk_buff *skb)
304{
305 skb->prev = slot->skblist_prev;
306 skb->next = (struct sk_buff *)slot;
307 slot->skblist_prev->next = skb;
308 slot->skblist_prev = skb;
309}
310
311#define slot_queue_walk(slot, skb) \
312 for (skb = slot->skblist_next; \
313 skb != (struct sk_buff *)slot; \
314 skb = skb->next)
315
1da177e4
LT
316static unsigned int sfq_drop(struct Qdisc *sch)
317{
318 struct sfq_sched_data *q = qdisc_priv(sch);
eda83e3b 319 sfq_index x, d = q->cur_depth;
1da177e4
LT
320 struct sk_buff *skb;
321 unsigned int len;
eda83e3b 322 struct sfq_slot *slot;
1da177e4 323
eda83e3b 324 /* Queue is full! Find the longest slot and drop tail packet from it */
1da177e4 325 if (d > 1) {
eda83e3b
ED
326 x = q->dep[d].next;
327 slot = &q->slots[x];
328drop:
18cb8098 329 skb = q->headdrop ? slot_dequeue_head(slot) : slot_dequeue_tail(slot);
0abf77e5 330 len = qdisc_pkt_len(skb);
ddecf0f4 331 slot->backlog -= len;
1da177e4 332 sfq_dec(q, x);
eda83e3b 333 kfree_skb(skb);
1da177e4
LT
334 sch->q.qlen--;
335 sch->qstats.drops++;
f5539eb8 336 sch->qstats.backlog -= len;
1da177e4
LT
337 return len;
338 }
339
340 if (d == 1) {
341 /* It is difficult to believe, but ALL THE SLOTS HAVE LENGTH 1. */
eda83e3b
ED
342 x = q->tail->next;
343 slot = &q->slots[x];
344 q->tail->next = slot->next;
345 q->ht[slot->hash] = SFQ_EMPTY_SLOT;
346 goto drop;
1da177e4
LT
347 }
348
349 return 0;
350}
351
ddecf0f4
ED
352/* Is ECN parameter configured */
353static int sfq_prob_mark(const struct sfq_sched_data *q)
354{
355 return q->flags & TC_RED_ECN;
356}
357
358/* Should packets over max threshold just be marked */
359static int sfq_hard_mark(const struct sfq_sched_data *q)
360{
361 return (q->flags & (TC_RED_ECN | TC_RED_HARDDROP)) == TC_RED_ECN;
362}
363
364static int sfq_headdrop(const struct sfq_sched_data *q)
365{
366 return q->headdrop;
367}
368
1da177e4 369static int
6f9e98f7 370sfq_enqueue(struct sk_buff *skb, struct Qdisc *sch)
1da177e4
LT
371{
372 struct sfq_sched_data *q = qdisc_priv(sch);
7d2681a6 373 unsigned int hash;
8efa8854 374 sfq_index x, qlen;
eda83e3b 375 struct sfq_slot *slot;
7f3ff4f6 376 int uninitialized_var(ret);
ddecf0f4
ED
377 struct sk_buff *head;
378 int delta;
7d2681a6
PM
379
380 hash = sfq_classify(skb, sch, &ret);
381 if (hash == 0) {
c27f339a 382 if (ret & __NET_XMIT_BYPASS)
7d2681a6
PM
383 sch->qstats.drops++;
384 kfree_skb(skb);
385 return ret;
386 }
387 hash--;
1da177e4
LT
388
389 x = q->ht[hash];
eda83e3b
ED
390 slot = &q->slots[x];
391 if (x == SFQ_EMPTY_SLOT) {
392 x = q->dep[0].next; /* get a free slot */
18cb8098
ED
393 if (x >= SFQ_MAX_FLOWS)
394 return qdisc_drop(skb, sch);
eda83e3b
ED
395 q->ht[hash] = x;
396 slot = &q->slots[x];
397 slot->hash = hash;
ddecf0f4
ED
398 slot->backlog = 0; /* should already be 0 anyway... */
399 red_set_vars(&slot->vars);
400 goto enqueue;
401 }
402 if (q->red_parms) {
403 slot->vars.qavg = red_calc_qavg_no_idle_time(q->red_parms,
404 &slot->vars,
405 slot->backlog);
406 switch (red_action(q->red_parms,
407 &slot->vars,
408 slot->vars.qavg)) {
409 case RED_DONT_MARK:
410 break;
411
412 case RED_PROB_MARK:
413 sch->qstats.overlimits++;
414 if (sfq_prob_mark(q)) {
415 /* We know we have at least one packet in queue */
416 if (sfq_headdrop(q) &&
417 INET_ECN_set_ce(slot->skblist_next)) {
418 q->stats.prob_mark_head++;
419 break;
420 }
421 if (INET_ECN_set_ce(skb)) {
422 q->stats.prob_mark++;
423 break;
424 }
425 }
426 q->stats.prob_drop++;
427 goto congestion_drop;
428
429 case RED_HARD_MARK:
430 sch->qstats.overlimits++;
431 if (sfq_hard_mark(q)) {
432 /* We know we have at least one packet in queue */
433 if (sfq_headdrop(q) &&
434 INET_ECN_set_ce(slot->skblist_next)) {
435 q->stats.forced_mark_head++;
436 break;
437 }
438 if (INET_ECN_set_ce(skb)) {
439 q->stats.forced_mark++;
440 break;
441 }
442 }
443 q->stats.forced_drop++;
444 goto congestion_drop;
445 }
1da177e4 446 }
6f9e98f7 447
18cb8098 448 if (slot->qlen >= q->maxdepth) {
ddecf0f4
ED
449congestion_drop:
450 if (!sfq_headdrop(q))
18cb8098
ED
451 return qdisc_drop(skb, sch);
452
ddecf0f4 453 /* We know we have at least one packet in queue */
18cb8098 454 head = slot_dequeue_head(slot);
ddecf0f4
ED
455 delta = qdisc_pkt_len(head) - qdisc_pkt_len(skb);
456 sch->qstats.backlog -= delta;
457 slot->backlog -= delta;
18cb8098
ED
458 qdisc_drop(head, sch);
459
18cb8098
ED
460 slot_queue_add(slot, skb);
461 return NET_XMIT_CN;
462 }
32740ddc 463
ddecf0f4 464enqueue:
0abf77e5 465 sch->qstats.backlog += qdisc_pkt_len(skb);
ddecf0f4 466 slot->backlog += qdisc_pkt_len(skb);
eda83e3b 467 slot_queue_add(slot, skb);
1da177e4 468 sfq_inc(q, x);
eda83e3b
ED
469 if (slot->qlen == 1) { /* The flow is new */
470 if (q->tail == NULL) { /* It is the first flow */
471 slot->next = x;
1da177e4 472 } else {
eda83e3b
ED
473 slot->next = q->tail->next;
474 q->tail->next = x;
1da177e4 475 }
cc34eb67
ED
476 /* We put this flow at the end of our flow list.
477 * This might sound unfair for a new flow to wait after old ones,
478 * but we could endup servicing new flows only, and freeze old ones.
479 */
480 q->tail = slot;
ddecf0f4 481 /* We could use a bigger initial quantum for new flows */
eeaeb068 482 slot->allot = q->scaled_quantum;
1da177e4 483 }
9190b3b3 484 if (++sch->q.qlen <= q->limit)
9871e50e 485 return NET_XMIT_SUCCESS;
1da177e4 486
8efa8854 487 qlen = slot->qlen;
1da177e4 488 sfq_drop(sch);
8efa8854
ED
489 /* Return Congestion Notification only if we dropped a packet
490 * from this flow.
491 */
e1738bd9
ED
492 if (qlen != slot->qlen)
493 return NET_XMIT_CN;
494
495 /* As we dropped a packet, better let upper stack know this */
496 qdisc_tree_decrease_qlen(sch, 1);
497 return NET_XMIT_SUCCESS;
1da177e4
LT
498}
499
1da177e4 500static struct sk_buff *
6f9e98f7 501sfq_dequeue(struct Qdisc *sch)
1da177e4
LT
502{
503 struct sfq_sched_data *q = qdisc_priv(sch);
504 struct sk_buff *skb;
aa3e2199 505 sfq_index a, next_a;
eda83e3b 506 struct sfq_slot *slot;
1da177e4
LT
507
508 /* No active slots */
eda83e3b 509 if (q->tail == NULL)
1da177e4
LT
510 return NULL;
511
eeaeb068 512next_slot:
eda83e3b
ED
513 a = q->tail->next;
514 slot = &q->slots[a];
eeaeb068
ED
515 if (slot->allot <= 0) {
516 q->tail = slot;
517 slot->allot += q->scaled_quantum;
518 goto next_slot;
519 }
eda83e3b 520 skb = slot_dequeue_head(slot);
1da177e4 521 sfq_dec(q, a);
9190b3b3 522 qdisc_bstats_update(sch, skb);
1da177e4 523 sch->q.qlen--;
0abf77e5 524 sch->qstats.backlog -= qdisc_pkt_len(skb);
ddecf0f4 525 slot->backlog -= qdisc_pkt_len(skb);
1da177e4 526 /* Is the slot empty? */
eda83e3b
ED
527 if (slot->qlen == 0) {
528 q->ht[slot->hash] = SFQ_EMPTY_SLOT;
529 next_a = slot->next;
aa3e2199 530 if (a == next_a) {
eda83e3b 531 q->tail = NULL; /* no more active slots */
1da177e4
LT
532 return skb;
533 }
eda83e3b 534 q->tail->next = next_a;
eeaeb068
ED
535 } else {
536 slot->allot -= SFQ_ALLOT_SIZE(qdisc_pkt_len(skb));
1da177e4
LT
537 }
538 return skb;
539}
540
541static void
6f9e98f7 542sfq_reset(struct Qdisc *sch)
1da177e4
LT
543{
544 struct sk_buff *skb;
545
546 while ((skb = sfq_dequeue(sch)) != NULL)
547 kfree_skb(skb);
548}
549
225d9b89
ED
550/*
551 * When q->perturbation is changed, we rehash all queued skbs
552 * to avoid OOO (Out Of Order) effects.
553 * We dont use sfq_dequeue()/sfq_enqueue() because we dont want to change
554 * counters.
555 */
18cb8098 556static void sfq_rehash(struct Qdisc *sch)
225d9b89 557{
18cb8098 558 struct sfq_sched_data *q = qdisc_priv(sch);
225d9b89
ED
559 struct sk_buff *skb;
560 int i;
561 struct sfq_slot *slot;
562 struct sk_buff_head list;
18cb8098 563 int dropped = 0;
225d9b89
ED
564
565 __skb_queue_head_init(&list);
566
18cb8098 567 for (i = 0; i < q->maxflows; i++) {
225d9b89
ED
568 slot = &q->slots[i];
569 if (!slot->qlen)
570 continue;
571 while (slot->qlen) {
572 skb = slot_dequeue_head(slot);
573 sfq_dec(q, i);
574 __skb_queue_tail(&list, skb);
575 }
ddecf0f4
ED
576 slot->backlog = 0;
577 red_set_vars(&slot->vars);
225d9b89
ED
578 q->ht[slot->hash] = SFQ_EMPTY_SLOT;
579 }
580 q->tail = NULL;
581
582 while ((skb = __skb_dequeue(&list)) != NULL) {
583 unsigned int hash = sfq_hash(q, skb);
584 sfq_index x = q->ht[hash];
585
586 slot = &q->slots[x];
587 if (x == SFQ_EMPTY_SLOT) {
588 x = q->dep[0].next; /* get a free slot */
18cb8098
ED
589 if (x >= SFQ_MAX_FLOWS) {
590drop: sch->qstats.backlog -= qdisc_pkt_len(skb);
591 kfree_skb(skb);
592 dropped++;
593 continue;
594 }
225d9b89
ED
595 q->ht[hash] = x;
596 slot = &q->slots[x];
597 slot->hash = hash;
598 }
18cb8098
ED
599 if (slot->qlen >= q->maxdepth)
600 goto drop;
225d9b89 601 slot_queue_add(slot, skb);
ddecf0f4
ED
602 if (q->red_parms)
603 slot->vars.qavg = red_calc_qavg(q->red_parms,
604 &slot->vars,
605 slot->backlog);
606 slot->backlog += qdisc_pkt_len(skb);
225d9b89
ED
607 sfq_inc(q, x);
608 if (slot->qlen == 1) { /* The flow is new */
609 if (q->tail == NULL) { /* It is the first flow */
610 slot->next = x;
611 } else {
612 slot->next = q->tail->next;
613 q->tail->next = x;
614 }
615 q->tail = slot;
616 slot->allot = q->scaled_quantum;
617 }
618 }
18cb8098
ED
619 sch->q.qlen -= dropped;
620 qdisc_tree_decrease_qlen(sch, dropped);
225d9b89
ED
621}
622
1da177e4
LT
623static void sfq_perturbation(unsigned long arg)
624{
6f9e98f7 625 struct Qdisc *sch = (struct Qdisc *)arg;
1da177e4 626 struct sfq_sched_data *q = qdisc_priv(sch);
225d9b89 627 spinlock_t *root_lock = qdisc_lock(qdisc_root_sleeping(sch));
1da177e4 628
225d9b89 629 spin_lock(root_lock);
d46f8dd8 630 q->perturbation = net_random();
225d9b89 631 if (!q->filter_list && q->tail)
18cb8098 632 sfq_rehash(sch);
225d9b89 633 spin_unlock(root_lock);
1da177e4 634
32740ddc
AK
635 if (q->perturb_period)
636 mod_timer(&q->perturb_timer, jiffies + q->perturb_period);
1da177e4
LT
637}
638
1e90474c 639static int sfq_change(struct Qdisc *sch, struct nlattr *opt)
1da177e4
LT
640{
641 struct sfq_sched_data *q = qdisc_priv(sch);
1e90474c 642 struct tc_sfq_qopt *ctl = nla_data(opt);
18cb8098 643 struct tc_sfq_qopt_v1 *ctl_v1 = NULL;
5e50da01 644 unsigned int qlen;
ddecf0f4 645 struct red_parms *p = NULL;
1da177e4 646
1e90474c 647 if (opt->nla_len < nla_attr_size(sizeof(*ctl)))
1da177e4 648 return -EINVAL;
18cb8098
ED
649 if (opt->nla_len >= nla_attr_size(sizeof(*ctl_v1)))
650 ctl_v1 = nla_data(opt);
119b3d38 651 if (ctl->divisor &&
652 (!is_power_of_2(ctl->divisor) || ctl->divisor > 65536))
653 return -EINVAL;
ddecf0f4
ED
654 if (ctl_v1 && ctl_v1->qth_min) {
655 p = kmalloc(sizeof(*p), GFP_KERNEL);
656 if (!p)
657 return -ENOMEM;
658 }
1da177e4 659 sch_tree_lock(sch);
18cb8098
ED
660 if (ctl->quantum) {
661 q->quantum = ctl->quantum;
662 q->scaled_quantum = SFQ_ALLOT_SIZE(q->quantum);
663 }
6f9e98f7 664 q->perturb_period = ctl->perturb_period * HZ;
18cb8098
ED
665 if (ctl->flows)
666 q->maxflows = min_t(u32, ctl->flows, SFQ_MAX_FLOWS);
667 if (ctl->divisor) {
817fb15d 668 q->divisor = ctl->divisor;
18cb8098
ED
669 q->maxflows = min_t(u32, q->maxflows, q->divisor);
670 }
671 if (ctl_v1) {
672 if (ctl_v1->depth)
673 q->maxdepth = min_t(u32, ctl_v1->depth, SFQ_MAX_DEPTH);
ddecf0f4
ED
674 if (p) {
675 swap(q->red_parms, p);
676 red_set_parms(q->red_parms,
677 ctl_v1->qth_min, ctl_v1->qth_max,
678 ctl_v1->Wlog,
679 ctl_v1->Plog, ctl_v1->Scell_log,
680 NULL,
681 ctl_v1->max_P);
682 }
683 q->flags = ctl_v1->flags;
18cb8098
ED
684 q->headdrop = ctl_v1->headdrop;
685 }
686 if (ctl->limit) {
687 q->limit = min_t(u32, ctl->limit, q->maxdepth * q->maxflows);
688 q->maxflows = min_t(u32, q->maxflows, q->limit);
689 }
690
5e50da01 691 qlen = sch->q.qlen;
5588b40d 692 while (sch->q.qlen > q->limit)
1da177e4 693 sfq_drop(sch);
5e50da01 694 qdisc_tree_decrease_qlen(sch, qlen - sch->q.qlen);
1da177e4
LT
695
696 del_timer(&q->perturb_timer);
697 if (q->perturb_period) {
32740ddc 698 mod_timer(&q->perturb_timer, jiffies + q->perturb_period);
d46f8dd8 699 q->perturbation = net_random();
1da177e4
LT
700 }
701 sch_tree_unlock(sch);
ddecf0f4 702 kfree(p);
1da177e4
LT
703 return 0;
704}
705
bd16a6cc
ED
706static void *sfq_alloc(size_t sz)
707{
708 void *ptr = kmalloc(sz, GFP_KERNEL | __GFP_NOWARN);
709
710 if (!ptr)
711 ptr = vmalloc(sz);
712 return ptr;
713}
714
715static void sfq_free(void *addr)
716{
717 if (addr) {
718 if (is_vmalloc_addr(addr))
719 vfree(addr);
720 else
721 kfree(addr);
722 }
723}
724
725static void sfq_destroy(struct Qdisc *sch)
726{
727 struct sfq_sched_data *q = qdisc_priv(sch);
728
729 tcf_destroy_chain(&q->filter_list);
730 q->perturb_period = 0;
731 del_timer_sync(&q->perturb_timer);
732 sfq_free(q->ht);
18cb8098 733 sfq_free(q->slots);
ddecf0f4 734 kfree(q->red_parms);
bd16a6cc
ED
735}
736
1e90474c 737static int sfq_init(struct Qdisc *sch, struct nlattr *opt)
1da177e4
LT
738{
739 struct sfq_sched_data *q = qdisc_priv(sch);
740 int i;
741
d3e99483 742 q->perturb_timer.function = sfq_perturbation;
c19a28e1 743 q->perturb_timer.data = (unsigned long)sch;
d3e99483 744 init_timer_deferrable(&q->perturb_timer);
1da177e4 745
18cb8098
ED
746 for (i = 0; i < SFQ_MAX_DEPTH + 1; i++) {
747 q->dep[i].next = i + SFQ_MAX_FLOWS;
748 q->dep[i].prev = i + SFQ_MAX_FLOWS;
1da177e4 749 }
6f9e98f7 750
18cb8098
ED
751 q->limit = SFQ_MAX_DEPTH;
752 q->maxdepth = SFQ_MAX_DEPTH;
eda83e3b
ED
753 q->cur_depth = 0;
754 q->tail = NULL;
817fb15d 755 q->divisor = SFQ_DEFAULT_HASH_DIVISOR;
18cb8098 756 q->maxflows = SFQ_DEFAULT_FLOWS;
02a9098e
ED
757 q->quantum = psched_mtu(qdisc_dev(sch));
758 q->scaled_quantum = SFQ_ALLOT_SIZE(q->quantum);
759 q->perturb_period = 0;
760 q->perturbation = net_random();
761
762 if (opt) {
1da177e4
LT
763 int err = sfq_change(sch, opt);
764 if (err)
765 return err;
766 }
6f9e98f7 767
bd16a6cc 768 q->ht = sfq_alloc(sizeof(q->ht[0]) * q->divisor);
18cb8098
ED
769 q->slots = sfq_alloc(sizeof(q->slots[0]) * q->maxflows);
770 if (!q->ht || !q->slots) {
bd16a6cc 771 sfq_destroy(sch);
817fb15d 772 return -ENOMEM;
bd16a6cc 773 }
817fb15d
ED
774 for (i = 0; i < q->divisor; i++)
775 q->ht[i] = SFQ_EMPTY_SLOT;
776
18cb8098 777 for (i = 0; i < q->maxflows; i++) {
18c8d82a 778 slot_queue_init(&q->slots[i]);
1da177e4 779 sfq_link(q, i);
18c8d82a 780 }
23624935
ED
781 if (q->limit >= 1)
782 sch->flags |= TCQ_F_CAN_BYPASS;
783 else
784 sch->flags &= ~TCQ_F_CAN_BYPASS;
1da177e4
LT
785 return 0;
786}
787
1da177e4
LT
788static int sfq_dump(struct Qdisc *sch, struct sk_buff *skb)
789{
790 struct sfq_sched_data *q = qdisc_priv(sch);
27a884dc 791 unsigned char *b = skb_tail_pointer(skb);
18cb8098 792 struct tc_sfq_qopt_v1 opt;
ddecf0f4 793 struct red_parms *p = q->red_parms;
18cb8098
ED
794
795 memset(&opt, 0, sizeof(opt));
796 opt.v0.quantum = q->quantum;
797 opt.v0.perturb_period = q->perturb_period / HZ;
798 opt.v0.limit = q->limit;
799 opt.v0.divisor = q->divisor;
800 opt.v0.flows = q->maxflows;
801 opt.depth = q->maxdepth;
802 opt.headdrop = q->headdrop;
1da177e4 803
ddecf0f4
ED
804 if (p) {
805 opt.qth_min = p->qth_min >> p->Wlog;
806 opt.qth_max = p->qth_max >> p->Wlog;
807 opt.Wlog = p->Wlog;
808 opt.Plog = p->Plog;
809 opt.Scell_log = p->Scell_log;
810 opt.max_P = p->max_P;
811 }
812 memcpy(&opt.stats, &q->stats, sizeof(opt.stats));
813 opt.flags = q->flags;
814
1b34ec43
DM
815 if (nla_put(skb, TCA_OPTIONS, sizeof(opt), &opt))
816 goto nla_put_failure;
1da177e4
LT
817
818 return skb->len;
819
1e90474c 820nla_put_failure:
dc5fc579 821 nlmsg_trim(skb, b);
1da177e4
LT
822 return -1;
823}
824
41065fba
JP
825static struct Qdisc *sfq_leaf(struct Qdisc *sch, unsigned long arg)
826{
827 return NULL;
828}
829
7d2681a6
PM
830static unsigned long sfq_get(struct Qdisc *sch, u32 classid)
831{
832 return 0;
833}
834
eb4a5527
JP
835static unsigned long sfq_bind(struct Qdisc *sch, unsigned long parent,
836 u32 classid)
837{
23624935
ED
838 /* we cannot bypass queue discipline anymore */
839 sch->flags &= ~TCQ_F_CAN_BYPASS;
eb4a5527
JP
840 return 0;
841}
842
da7115d9
JP
843static void sfq_put(struct Qdisc *q, unsigned long cl)
844{
845}
846
7d2681a6
PM
847static struct tcf_proto **sfq_find_tcf(struct Qdisc *sch, unsigned long cl)
848{
849 struct sfq_sched_data *q = qdisc_priv(sch);
850
851 if (cl)
852 return NULL;
853 return &q->filter_list;
854}
855
94de78d1
PM
856static int sfq_dump_class(struct Qdisc *sch, unsigned long cl,
857 struct sk_buff *skb, struct tcmsg *tcm)
858{
859 tcm->tcm_handle |= TC_H_MIN(cl);
860 return 0;
861}
862
863static int sfq_dump_class_stats(struct Qdisc *sch, unsigned long cl,
864 struct gnet_dump *d)
865{
866 struct sfq_sched_data *q = qdisc_priv(sch);
ee09b3c1
ED
867 sfq_index idx = q->ht[cl - 1];
868 struct gnet_stats_queue qs = { 0 };
869 struct tc_sfq_xstats xstats = { 0 };
c4266263 870
ee09b3c1
ED
871 if (idx != SFQ_EMPTY_SLOT) {
872 const struct sfq_slot *slot = &q->slots[idx];
94de78d1 873
eeaeb068 874 xstats.allot = slot->allot << SFQ_ALLOT_SHIFT;
ee09b3c1 875 qs.qlen = slot->qlen;
ddecf0f4 876 qs.backlog = slot->backlog;
ee09b3c1 877 }
94de78d1
PM
878 if (gnet_stats_copy_queue(d, &qs) < 0)
879 return -1;
880 return gnet_stats_copy_app(d, &xstats, sizeof(xstats));
881}
882
7d2681a6
PM
883static void sfq_walk(struct Qdisc *sch, struct qdisc_walker *arg)
884{
94de78d1
PM
885 struct sfq_sched_data *q = qdisc_priv(sch);
886 unsigned int i;
887
888 if (arg->stop)
889 return;
890
817fb15d 891 for (i = 0; i < q->divisor; i++) {
eda83e3b 892 if (q->ht[i] == SFQ_EMPTY_SLOT ||
94de78d1
PM
893 arg->count < arg->skip) {
894 arg->count++;
895 continue;
896 }
897 if (arg->fn(sch, i + 1, arg) < 0) {
898 arg->stop = 1;
899 break;
900 }
901 arg->count++;
902 }
7d2681a6
PM
903}
904
905static const struct Qdisc_class_ops sfq_class_ops = {
41065fba 906 .leaf = sfq_leaf,
7d2681a6 907 .get = sfq_get,
da7115d9 908 .put = sfq_put,
7d2681a6 909 .tcf_chain = sfq_find_tcf,
eb4a5527 910 .bind_tcf = sfq_bind,
da7115d9 911 .unbind_tcf = sfq_put,
94de78d1
PM
912 .dump = sfq_dump_class,
913 .dump_stats = sfq_dump_class_stats,
7d2681a6
PM
914 .walk = sfq_walk,
915};
916
20fea08b 917static struct Qdisc_ops sfq_qdisc_ops __read_mostly = {
7d2681a6 918 .cl_ops = &sfq_class_ops,
1da177e4
LT
919 .id = "sfq",
920 .priv_size = sizeof(struct sfq_sched_data),
921 .enqueue = sfq_enqueue,
922 .dequeue = sfq_dequeue,
07bd8df5 923 .peek = qdisc_peek_dequeued,
1da177e4
LT
924 .drop = sfq_drop,
925 .init = sfq_init,
926 .reset = sfq_reset,
927 .destroy = sfq_destroy,
928 .change = NULL,
929 .dump = sfq_dump,
930 .owner = THIS_MODULE,
931};
932
933static int __init sfq_module_init(void)
934{
935 return register_qdisc(&sfq_qdisc_ops);
936}
10297b99 937static void __exit sfq_module_exit(void)
1da177e4
LT
938{
939 unregister_qdisc(&sfq_qdisc_ops);
940}
941module_init(sfq_module_init)
942module_exit(sfq_module_exit)
943MODULE_LICENSE("GPL");