blob: 109b2322778f2353d82ebfbc8e40578e80ae7099 [file] [log] [blame]
Eric Dumazetafe4fd02013-08-29 15:49:55 -07001/*
2 * net/sched/sch_fq.c Fair Queue Packet Scheduler (per flow pacing)
3 *
Eric Dumazet86b3bfe2015-01-28 06:06:36 -08004 * Copyright (C) 2013-2015 Eric Dumazet <edumazet@google.com>
Eric Dumazetafe4fd02013-08-29 15:49:55 -07005 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version
9 * 2 of the License, or (at your option) any later version.
10 *
Simon Horman05e8bb82015-04-02 11:20:23 +090011 * Meant to be mostly used for locally generated traffic :
Eric Dumazetafe4fd02013-08-29 15:49:55 -070012 * Fast classification depends on skb->sk being set before reaching us.
13 * If not, (router workload), we use rxhash as fallback, with 32 bits wide hash.
14 * All packets belonging to a socket are considered as a 'flow'.
15 *
16 * Flows are dynamically allocated and stored in a hash table of RB trees
17 * They are also part of one Round Robin 'queues' (new or old flows)
18 *
19 * Burst avoidance (aka pacing) capability :
20 *
21 * Transport (eg TCP) can set in sk->sk_pacing_rate a rate, enqueue a
22 * bunch of packets, and this packet scheduler adds delay between
23 * packets to respect rate limitation.
24 *
25 * enqueue() :
26 * - lookup one RB tree (out of 1024 or more) to find the flow.
27 * If non existent flow, create it, add it to the tree.
28 * Add skb to the per flow list of skb (fifo).
29 * - Use a special fifo for high prio packets
30 *
31 * dequeue() : serves flows in Round Robin
32 * Note : When a flow becomes empty, we do not immediately remove it from
33 * rb trees, for performance reasons (its expected to send additional packets,
34 * or SLAB cache will reuse socket for another flow)
35 */
36
37#include <linux/module.h>
38#include <linux/types.h>
39#include <linux/kernel.h>
40#include <linux/jiffies.h>
41#include <linux/string.h>
42#include <linux/in.h>
43#include <linux/errno.h>
44#include <linux/init.h>
45#include <linux/skbuff.h>
46#include <linux/slab.h>
47#include <linux/rbtree.h>
48#include <linux/hash.h>
Eric Dumazet08f89b92013-08-30 09:46:43 -070049#include <linux/prefetch.h>
Eric Dumazetc3bd8542013-12-15 13:15:25 -080050#include <linux/vmalloc.h>
Eric Dumazetafe4fd02013-08-29 15:49:55 -070051#include <net/netlink.h>
52#include <net/pkt_sched.h>
53#include <net/sock.h>
54#include <net/tcp_states.h>
Eric Dumazet98781962015-02-03 18:31:53 -080055#include <net/tcp.h>
Eric Dumazetafe4fd02013-08-29 15:49:55 -070056
57/*
58 * Per flow structure, dynamically allocated
59 */
60struct fq_flow {
61 struct sk_buff *head; /* list of skbs for this flow : first skb */
62 union {
63 struct sk_buff *tail; /* last skb in the list */
64 unsigned long age; /* jiffies when flow was emptied, for gc */
65 };
Simon Horman05e8bb82015-04-02 11:20:23 +090066 struct rb_node fq_node; /* anchor in fq_root[] trees */
Eric Dumazetafe4fd02013-08-29 15:49:55 -070067 struct sock *sk;
68 int qlen; /* number of packets in flow queue */
69 int credit;
70 u32 socket_hash; /* sk_hash */
71 struct fq_flow *next; /* next pointer in RR lists, or &detached */
72
73 struct rb_node rate_node; /* anchor in q->delayed tree */
74 u64 time_next_packet;
75};
76
77struct fq_flow_head {
78 struct fq_flow *first;
79 struct fq_flow *last;
80};
81
82struct fq_sched_data {
83 struct fq_flow_head new_flows;
84
85 struct fq_flow_head old_flows;
86
87 struct rb_root delayed; /* for rate limited flows */
88 u64 time_next_delayed_flow;
89
90 struct fq_flow internal; /* for non classified or high prio packets */
91 u32 quantum;
92 u32 initial_quantum;
Eric Dumazetf52ed892013-11-15 08:58:14 -080093 u32 flow_refill_delay;
Eric Dumazetafe4fd02013-08-29 15:49:55 -070094 u32 flow_max_rate; /* optional max rate per flow */
95 u32 flow_plimit; /* max packets per flow */
Eric Dumazet06eb3952015-02-04 21:30:40 -080096 u32 orphan_mask; /* mask for orphaned skb */
Eric Dumazetafe4fd02013-08-29 15:49:55 -070097 struct rb_root *fq_root;
98 u8 rate_enable;
99 u8 fq_trees_log;
100
101 u32 flows;
102 u32 inactive_flows;
103 u32 throttled_flows;
104
105 u64 stat_gc_flows;
106 u64 stat_internal_packets;
107 u64 stat_tcp_retrans;
108 u64 stat_throttled;
109 u64 stat_flows_plimit;
110 u64 stat_pkts_too_long;
111 u64 stat_allocation_errors;
112 struct qdisc_watchdog watchdog;
113};
114
115/* special value to mark a detached flow (not on old/new list) */
116static struct fq_flow detached, throttled;
117
118static void fq_flow_set_detached(struct fq_flow *f)
119{
120 f->next = &detached;
Eric Dumazetf52ed892013-11-15 08:58:14 -0800121 f->age = jiffies;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700122}
123
124static bool fq_flow_is_detached(const struct fq_flow *f)
125{
126 return f->next == &detached;
127}
128
129static void fq_flow_set_throttled(struct fq_sched_data *q, struct fq_flow *f)
130{
131 struct rb_node **p = &q->delayed.rb_node, *parent = NULL;
132
133 while (*p) {
134 struct fq_flow *aux;
135
136 parent = *p;
137 aux = container_of(parent, struct fq_flow, rate_node);
138 if (f->time_next_packet >= aux->time_next_packet)
139 p = &parent->rb_right;
140 else
141 p = &parent->rb_left;
142 }
143 rb_link_node(&f->rate_node, parent, p);
144 rb_insert_color(&f->rate_node, &q->delayed);
145 q->throttled_flows++;
146 q->stat_throttled++;
147
148 f->next = &throttled;
149 if (q->time_next_delayed_flow > f->time_next_packet)
150 q->time_next_delayed_flow = f->time_next_packet;
151}
152
153
154static struct kmem_cache *fq_flow_cachep __read_mostly;
155
156static void fq_flow_add_tail(struct fq_flow_head *head, struct fq_flow *flow)
157{
158 if (head->first)
159 head->last->next = flow;
160 else
161 head->first = flow;
162 head->last = flow;
163 flow->next = NULL;
164}
165
166/* limit number of collected flows per round */
167#define FQ_GC_MAX 8
168#define FQ_GC_AGE (3*HZ)
169
170static bool fq_gc_candidate(const struct fq_flow *f)
171{
172 return fq_flow_is_detached(f) &&
173 time_after(jiffies, f->age + FQ_GC_AGE);
174}
175
176static void fq_gc(struct fq_sched_data *q,
177 struct rb_root *root,
178 struct sock *sk)
179{
180 struct fq_flow *f, *tofree[FQ_GC_MAX];
181 struct rb_node **p, *parent;
182 int fcnt = 0;
183
184 p = &root->rb_node;
185 parent = NULL;
186 while (*p) {
187 parent = *p;
188
189 f = container_of(parent, struct fq_flow, fq_node);
190 if (f->sk == sk)
191 break;
192
193 if (fq_gc_candidate(f)) {
194 tofree[fcnt++] = f;
195 if (fcnt == FQ_GC_MAX)
196 break;
197 }
198
199 if (f->sk > sk)
200 p = &parent->rb_right;
201 else
202 p = &parent->rb_left;
203 }
204
205 q->flows -= fcnt;
206 q->inactive_flows -= fcnt;
207 q->stat_gc_flows += fcnt;
208 while (fcnt) {
209 struct fq_flow *f = tofree[--fcnt];
210
211 rb_erase(&f->fq_node, root);
212 kmem_cache_free(fq_flow_cachep, f);
213 }
214}
215
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700216static struct fq_flow *fq_classify(struct sk_buff *skb, struct fq_sched_data *q)
217{
218 struct rb_node **p, *parent;
219 struct sock *sk = skb->sk;
220 struct rb_root *root;
221 struct fq_flow *f;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700222
223 /* warning: no starvation prevention... */
Maciej Żenczykowski2abc2f02013-11-14 08:50:43 -0800224 if (unlikely((skb->priority & TC_PRIO_MAX) == TC_PRIO_CONTROL))
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700225 return &q->internal;
226
Eric Dumazetca6fb062015-10-02 11:43:35 -0700227 /* SYNACK messages are attached to a TCP_NEW_SYN_RECV request socket
Eric Dumazete446f9d2015-10-08 05:01:55 -0700228 * or a listener (SYNCOOKIE mode)
Eric Dumazetca6fb062015-10-02 11:43:35 -0700229 * 1) request sockets are not full blown,
230 * they do not contain sk_pacing_rate
231 * 2) They are not part of a 'flow' yet
232 * 3) We do not want to rate limit them (eg SYNFLOOD attack),
Eric Dumazet06eb3952015-02-04 21:30:40 -0800233 * especially if the listener set SO_MAX_PACING_RATE
Eric Dumazetca6fb062015-10-02 11:43:35 -0700234 * 4) We pretend they are orphaned
Eric Dumazet06eb3952015-02-04 21:30:40 -0800235 */
Eric Dumazete446f9d2015-10-08 05:01:55 -0700236 if (!sk || sk_listener(sk)) {
Eric Dumazet06eb3952015-02-04 21:30:40 -0800237 unsigned long hash = skb_get_hash(skb) & q->orphan_mask;
238
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700239 /* By forcing low order bit to 1, we make sure to not
240 * collide with a local flow (socket pointers are word aligned)
241 */
Eric Dumazet06eb3952015-02-04 21:30:40 -0800242 sk = (struct sock *)((hash << 1) | 1UL);
243 skb_orphan(skb);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700244 }
245
246 root = &q->fq_root[hash_32((u32)(long)sk, q->fq_trees_log)];
247
248 if (q->flows >= (2U << q->fq_trees_log) &&
249 q->inactive_flows > q->flows/2)
250 fq_gc(q, root, sk);
251
252 p = &root->rb_node;
253 parent = NULL;
254 while (*p) {
255 parent = *p;
256
257 f = container_of(parent, struct fq_flow, fq_node);
258 if (f->sk == sk) {
259 /* socket might have been reallocated, so check
260 * if its sk_hash is the same.
261 * It not, we need to refill credit with
262 * initial quantum
263 */
264 if (unlikely(skb->sk &&
265 f->socket_hash != sk->sk_hash)) {
266 f->credit = q->initial_quantum;
267 f->socket_hash = sk->sk_hash;
Eric Dumazetfc59d5b2013-10-27 16:26:43 -0700268 f->time_next_packet = 0ULL;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700269 }
270 return f;
271 }
272 if (f->sk > sk)
273 p = &parent->rb_right;
274 else
275 p = &parent->rb_left;
276 }
277
278 f = kmem_cache_zalloc(fq_flow_cachep, GFP_ATOMIC | __GFP_NOWARN);
279 if (unlikely(!f)) {
280 q->stat_allocation_errors++;
281 return &q->internal;
282 }
283 fq_flow_set_detached(f);
284 f->sk = sk;
285 if (skb->sk)
286 f->socket_hash = sk->sk_hash;
287 f->credit = q->initial_quantum;
288
289 rb_link_node(&f->fq_node, parent, p);
290 rb_insert_color(&f->fq_node, root);
291
292 q->flows++;
293 q->inactive_flows++;
294 return f;
295}
296
297
298/* remove one skb from head of flow queue */
Eric Dumazet8d34ce12013-09-27 14:20:01 -0700299static struct sk_buff *fq_dequeue_head(struct Qdisc *sch, struct fq_flow *flow)
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700300{
301 struct sk_buff *skb = flow->head;
302
303 if (skb) {
304 flow->head = skb->next;
305 skb->next = NULL;
306 flow->qlen--;
John Fastabend25331d62014-09-28 11:53:29 -0700307 qdisc_qstats_backlog_dec(sch, skb);
Eric Dumazet8d34ce12013-09-27 14:20:01 -0700308 sch->q.qlen--;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700309 }
310 return skb;
311}
312
313/* We might add in the future detection of retransmits
314 * For the time being, just return false
315 */
316static bool skb_is_retransmit(struct sk_buff *skb)
317{
318 return false;
319}
320
321/* add skb to flow queue
322 * flow queue is a linked list, kind of FIFO, except for TCP retransmits
323 * We special case tcp retransmits to be transmitted before other packets.
324 * We rely on fact that TCP retransmits are unlikely, so we do not waste
325 * a separate queue or a pointer.
326 * head-> [retrans pkt 1]
327 * [retrans pkt 2]
328 * [ normal pkt 1]
329 * [ normal pkt 2]
330 * [ normal pkt 3]
331 * tail-> [ normal pkt 4]
332 */
333static void flow_queue_add(struct fq_flow *flow, struct sk_buff *skb)
334{
335 struct sk_buff *prev, *head = flow->head;
336
337 skb->next = NULL;
338 if (!head) {
339 flow->head = skb;
340 flow->tail = skb;
341 return;
342 }
343 if (likely(!skb_is_retransmit(skb))) {
344 flow->tail->next = skb;
345 flow->tail = skb;
346 return;
347 }
348
349 /* This skb is a tcp retransmit,
350 * find the last retrans packet in the queue
351 */
352 prev = NULL;
353 while (skb_is_retransmit(head)) {
354 prev = head;
355 head = head->next;
356 if (!head)
357 break;
358 }
359 if (!prev) { /* no rtx packet in queue, become the new head */
360 skb->next = flow->head;
361 flow->head = skb;
362 } else {
363 if (prev == flow->tail)
364 flow->tail = skb;
365 else
366 skb->next = prev->next;
367 prev->next = skb;
368 }
369}
370
371static int fq_enqueue(struct sk_buff *skb, struct Qdisc *sch)
372{
373 struct fq_sched_data *q = qdisc_priv(sch);
374 struct fq_flow *f;
375
376 if (unlikely(sch->q.qlen >= sch->limit))
377 return qdisc_drop(skb, sch);
378
379 f = fq_classify(skb, q);
380 if (unlikely(f->qlen >= q->flow_plimit && f != &q->internal)) {
381 q->stat_flows_plimit++;
382 return qdisc_drop(skb, sch);
383 }
384
385 f->qlen++;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700386 if (skb_is_retransmit(skb))
387 q->stat_tcp_retrans++;
John Fastabend25331d62014-09-28 11:53:29 -0700388 qdisc_qstats_backlog_inc(sch, skb);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700389 if (fq_flow_is_detached(f)) {
390 fq_flow_add_tail(&q->new_flows, f);
Eric Dumazetf52ed892013-11-15 08:58:14 -0800391 if (time_after(jiffies, f->age + q->flow_refill_delay))
392 f->credit = max_t(u32, f->credit, q->quantum);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700393 q->inactive_flows--;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700394 }
Eric Dumazetf52ed892013-11-15 08:58:14 -0800395
396 /* Note: this overwrites f->age */
397 flow_queue_add(f, skb);
398
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700399 if (unlikely(f == &q->internal)) {
400 q->stat_internal_packets++;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700401 }
402 sch->q.qlen++;
403
404 return NET_XMIT_SUCCESS;
405}
406
407static void fq_check_throttled(struct fq_sched_data *q, u64 now)
408{
409 struct rb_node *p;
410
411 if (q->time_next_delayed_flow > now)
412 return;
413
414 q->time_next_delayed_flow = ~0ULL;
415 while ((p = rb_first(&q->delayed)) != NULL) {
416 struct fq_flow *f = container_of(p, struct fq_flow, rate_node);
417
418 if (f->time_next_packet > now) {
419 q->time_next_delayed_flow = f->time_next_packet;
420 break;
421 }
422 rb_erase(p, &q->delayed);
423 q->throttled_flows--;
424 fq_flow_add_tail(&q->old_flows, f);
425 }
426}
427
428static struct sk_buff *fq_dequeue(struct Qdisc *sch)
429{
430 struct fq_sched_data *q = qdisc_priv(sch);
Eric Dumazetd2de8752014-08-22 18:32:09 -0700431 u64 now = ktime_get_ns();
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700432 struct fq_flow_head *head;
433 struct sk_buff *skb;
434 struct fq_flow *f;
Eric Dumazet0eab5eb2013-10-01 09:10:16 -0700435 u32 rate;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700436
Eric Dumazet8d34ce12013-09-27 14:20:01 -0700437 skb = fq_dequeue_head(sch, &q->internal);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700438 if (skb)
439 goto out;
440 fq_check_throttled(q, now);
441begin:
442 head = &q->new_flows;
443 if (!head->first) {
444 head = &q->old_flows;
445 if (!head->first) {
446 if (q->time_next_delayed_flow != ~0ULL)
447 qdisc_watchdog_schedule_ns(&q->watchdog,
Eric Dumazetf2600cf2014-10-04 10:11:31 -0700448 q->time_next_delayed_flow,
449 false);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700450 return NULL;
451 }
452 }
453 f = head->first;
454
455 if (f->credit <= 0) {
456 f->credit += q->quantum;
457 head->first = f->next;
458 fq_flow_add_tail(&q->old_flows, f);
459 goto begin;
460 }
461
Eric Dumazet98781962015-02-03 18:31:53 -0800462 skb = f->head;
463 if (unlikely(skb && now < f->time_next_packet &&
464 !skb_is_tcp_pure_ack(skb))) {
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700465 head->first = f->next;
466 fq_flow_set_throttled(q, f);
467 goto begin;
468 }
469
Eric Dumazet8d34ce12013-09-27 14:20:01 -0700470 skb = fq_dequeue_head(sch, f);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700471 if (!skb) {
472 head->first = f->next;
473 /* force a pass through old_flows to prevent starvation */
474 if ((head == &q->new_flows) && q->old_flows.first) {
475 fq_flow_add_tail(&q->old_flows, f);
476 } else {
477 fq_flow_set_detached(f);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700478 q->inactive_flows++;
479 }
480 goto begin;
481 }
Eric Dumazet08f89b92013-08-30 09:46:43 -0700482 prefetch(&skb->end);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700483 f->credit -= qdisc_pkt_len(skb);
484
Eric Dumazet0eab5eb2013-10-01 09:10:16 -0700485 if (f->credit > 0 || !q->rate_enable)
486 goto out;
487
Eric Dumazet98781962015-02-03 18:31:53 -0800488 /* Do not pace locally generated ack packets */
489 if (skb_is_tcp_pure_ack(skb))
490 goto out;
491
Eric Dumazet7eec4172013-10-08 15:16:00 -0700492 rate = q->flow_max_rate;
Eric Dumazet86b3bfe2015-01-28 06:06:36 -0800493 if (skb->sk)
Eric Dumazet7eec4172013-10-08 15:16:00 -0700494 rate = min(skb->sk->sk_pacing_rate, rate);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700495
Eric Dumazet7eec4172013-10-08 15:16:00 -0700496 if (rate != ~0U) {
Eric Dumazet0eab5eb2013-10-01 09:10:16 -0700497 u32 plen = max(qdisc_pkt_len(skb), q->quantum);
498 u64 len = (u64)plen * NSEC_PER_SEC;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700499
Eric Dumazet7eec4172013-10-08 15:16:00 -0700500 if (likely(rate))
501 do_div(len, rate);
Eric Dumazet0eab5eb2013-10-01 09:10:16 -0700502 /* Since socket rate can change later,
Eric Dumazetced7a042014-11-25 08:57:29 -0800503 * clamp the delay to 1 second.
504 * Really, providers of too big packets should be fixed !
Eric Dumazet0eab5eb2013-10-01 09:10:16 -0700505 */
Eric Dumazetced7a042014-11-25 08:57:29 -0800506 if (unlikely(len > NSEC_PER_SEC)) {
507 len = NSEC_PER_SEC;
Eric Dumazet0eab5eb2013-10-01 09:10:16 -0700508 q->stat_pkts_too_long++;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700509 }
Eric Dumazet0eab5eb2013-10-01 09:10:16 -0700510
511 f->time_next_packet = now + len;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700512 }
513out:
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700514 qdisc_bstats_update(sch, skb);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700515 return skb;
516}
517
518static void fq_reset(struct Qdisc *sch)
519{
Eric Dumazet8d34ce12013-09-27 14:20:01 -0700520 struct fq_sched_data *q = qdisc_priv(sch);
521 struct rb_root *root;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700522 struct sk_buff *skb;
Eric Dumazet8d34ce12013-09-27 14:20:01 -0700523 struct rb_node *p;
524 struct fq_flow *f;
525 unsigned int idx;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700526
Eric Dumazet8d34ce12013-09-27 14:20:01 -0700527 while ((skb = fq_dequeue_head(sch, &q->internal)) != NULL)
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700528 kfree_skb(skb);
Eric Dumazet8d34ce12013-09-27 14:20:01 -0700529
530 if (!q->fq_root)
531 return;
532
533 for (idx = 0; idx < (1U << q->fq_trees_log); idx++) {
534 root = &q->fq_root[idx];
535 while ((p = rb_first(root)) != NULL) {
536 f = container_of(p, struct fq_flow, fq_node);
537 rb_erase(p, root);
538
539 while ((skb = fq_dequeue_head(sch, f)) != NULL)
540 kfree_skb(skb);
541
542 kmem_cache_free(fq_flow_cachep, f);
543 }
544 }
545 q->new_flows.first = NULL;
546 q->old_flows.first = NULL;
547 q->delayed = RB_ROOT;
548 q->flows = 0;
549 q->inactive_flows = 0;
550 q->throttled_flows = 0;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700551}
552
553static void fq_rehash(struct fq_sched_data *q,
554 struct rb_root *old_array, u32 old_log,
555 struct rb_root *new_array, u32 new_log)
556{
557 struct rb_node *op, **np, *parent;
558 struct rb_root *oroot, *nroot;
559 struct fq_flow *of, *nf;
560 int fcnt = 0;
561 u32 idx;
562
563 for (idx = 0; idx < (1U << old_log); idx++) {
564 oroot = &old_array[idx];
565 while ((op = rb_first(oroot)) != NULL) {
566 rb_erase(op, oroot);
567 of = container_of(op, struct fq_flow, fq_node);
568 if (fq_gc_candidate(of)) {
569 fcnt++;
570 kmem_cache_free(fq_flow_cachep, of);
571 continue;
572 }
573 nroot = &new_array[hash_32((u32)(long)of->sk, new_log)];
574
575 np = &nroot->rb_node;
576 parent = NULL;
577 while (*np) {
578 parent = *np;
579
580 nf = container_of(parent, struct fq_flow, fq_node);
581 BUG_ON(nf->sk == of->sk);
582
583 if (nf->sk > of->sk)
584 np = &parent->rb_right;
585 else
586 np = &parent->rb_left;
587 }
588
589 rb_link_node(&of->fq_node, parent, np);
590 rb_insert_color(&of->fq_node, nroot);
591 }
592 }
593 q->flows -= fcnt;
594 q->inactive_flows -= fcnt;
595 q->stat_gc_flows += fcnt;
596}
597
Eric Dumazetc3bd8542013-12-15 13:15:25 -0800598static void *fq_alloc_node(size_t sz, int node)
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700599{
Eric Dumazetc3bd8542013-12-15 13:15:25 -0800600 void *ptr;
601
602 ptr = kmalloc_node(sz, GFP_KERNEL | __GFP_REPEAT | __GFP_NOWARN, node);
603 if (!ptr)
604 ptr = vmalloc_node(sz, node);
605 return ptr;
606}
607
608static void fq_free(void *addr)
609{
WANG Cong4cb28972014-06-02 15:55:22 -0700610 kvfree(addr);
Eric Dumazetc3bd8542013-12-15 13:15:25 -0800611}
612
613static int fq_resize(struct Qdisc *sch, u32 log)
614{
615 struct fq_sched_data *q = qdisc_priv(sch);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700616 struct rb_root *array;
Eric Dumazet2d8d40a2014-03-06 22:57:52 -0800617 void *old_fq_root;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700618 u32 idx;
619
620 if (q->fq_root && log == q->fq_trees_log)
621 return 0;
622
Eric Dumazetc3bd8542013-12-15 13:15:25 -0800623 /* If XPS was setup, we can allocate memory on right NUMA node */
624 array = fq_alloc_node(sizeof(struct rb_root) << log,
625 netdev_queue_numa_node_read(sch->dev_queue));
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700626 if (!array)
627 return -ENOMEM;
628
629 for (idx = 0; idx < (1U << log); idx++)
630 array[idx] = RB_ROOT;
631
Eric Dumazet2d8d40a2014-03-06 22:57:52 -0800632 sch_tree_lock(sch);
633
634 old_fq_root = q->fq_root;
635 if (old_fq_root)
636 fq_rehash(q, old_fq_root, q->fq_trees_log, array, log);
637
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700638 q->fq_root = array;
639 q->fq_trees_log = log;
640
Eric Dumazet2d8d40a2014-03-06 22:57:52 -0800641 sch_tree_unlock(sch);
642
643 fq_free(old_fq_root);
644
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700645 return 0;
646}
647
648static const struct nla_policy fq_policy[TCA_FQ_MAX + 1] = {
649 [TCA_FQ_PLIMIT] = { .type = NLA_U32 },
650 [TCA_FQ_FLOW_PLIMIT] = { .type = NLA_U32 },
651 [TCA_FQ_QUANTUM] = { .type = NLA_U32 },
652 [TCA_FQ_INITIAL_QUANTUM] = { .type = NLA_U32 },
653 [TCA_FQ_RATE_ENABLE] = { .type = NLA_U32 },
654 [TCA_FQ_FLOW_DEFAULT_RATE] = { .type = NLA_U32 },
655 [TCA_FQ_FLOW_MAX_RATE] = { .type = NLA_U32 },
656 [TCA_FQ_BUCKETS_LOG] = { .type = NLA_U32 },
Eric Dumazetf52ed892013-11-15 08:58:14 -0800657 [TCA_FQ_FLOW_REFILL_DELAY] = { .type = NLA_U32 },
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700658};
659
660static int fq_change(struct Qdisc *sch, struct nlattr *opt)
661{
662 struct fq_sched_data *q = qdisc_priv(sch);
663 struct nlattr *tb[TCA_FQ_MAX + 1];
664 int err, drop_count = 0;
665 u32 fq_log;
666
667 if (!opt)
668 return -EINVAL;
669
670 err = nla_parse_nested(tb, TCA_FQ_MAX, opt, fq_policy);
671 if (err < 0)
672 return err;
673
674 sch_tree_lock(sch);
675
676 fq_log = q->fq_trees_log;
677
678 if (tb[TCA_FQ_BUCKETS_LOG]) {
679 u32 nval = nla_get_u32(tb[TCA_FQ_BUCKETS_LOG]);
680
681 if (nval >= 1 && nval <= ilog2(256*1024))
682 fq_log = nval;
683 else
684 err = -EINVAL;
685 }
686 if (tb[TCA_FQ_PLIMIT])
687 sch->limit = nla_get_u32(tb[TCA_FQ_PLIMIT]);
688
689 if (tb[TCA_FQ_FLOW_PLIMIT])
690 q->flow_plimit = nla_get_u32(tb[TCA_FQ_FLOW_PLIMIT]);
691
Kenneth Klette Jonassen3725a262015-02-03 17:49:18 +0100692 if (tb[TCA_FQ_QUANTUM]) {
693 u32 quantum = nla_get_u32(tb[TCA_FQ_QUANTUM]);
694
695 if (quantum > 0)
696 q->quantum = quantum;
697 else
698 err = -EINVAL;
699 }
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700700
701 if (tb[TCA_FQ_INITIAL_QUANTUM])
Eric Dumazetede869c2013-10-07 12:50:18 -0700702 q->initial_quantum = nla_get_u32(tb[TCA_FQ_INITIAL_QUANTUM]);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700703
704 if (tb[TCA_FQ_FLOW_DEFAULT_RATE])
Eric Dumazet65c5189a2013-11-15 08:57:26 -0800705 pr_warn_ratelimited("sch_fq: defrate %u ignored.\n",
706 nla_get_u32(tb[TCA_FQ_FLOW_DEFAULT_RATE]));
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700707
708 if (tb[TCA_FQ_FLOW_MAX_RATE])
709 q->flow_max_rate = nla_get_u32(tb[TCA_FQ_FLOW_MAX_RATE]);
710
711 if (tb[TCA_FQ_RATE_ENABLE]) {
712 u32 enable = nla_get_u32(tb[TCA_FQ_RATE_ENABLE]);
713
714 if (enable <= 1)
715 q->rate_enable = enable;
716 else
717 err = -EINVAL;
718 }
719
Eric Dumazetf52ed892013-11-15 08:58:14 -0800720 if (tb[TCA_FQ_FLOW_REFILL_DELAY]) {
721 u32 usecs_delay = nla_get_u32(tb[TCA_FQ_FLOW_REFILL_DELAY]) ;
722
723 q->flow_refill_delay = usecs_to_jiffies(usecs_delay);
724 }
725
Eric Dumazet06eb3952015-02-04 21:30:40 -0800726 if (tb[TCA_FQ_ORPHAN_MASK])
727 q->orphan_mask = nla_get_u32(tb[TCA_FQ_ORPHAN_MASK]);
728
Eric Dumazet2d8d40a2014-03-06 22:57:52 -0800729 if (!err) {
730 sch_tree_unlock(sch);
Eric Dumazetc3bd8542013-12-15 13:15:25 -0800731 err = fq_resize(sch, fq_log);
Eric Dumazet2d8d40a2014-03-06 22:57:52 -0800732 sch_tree_lock(sch);
733 }
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700734 while (sch->q.qlen > sch->limit) {
735 struct sk_buff *skb = fq_dequeue(sch);
736
Eric Dumazet8d34ce12013-09-27 14:20:01 -0700737 if (!skb)
738 break;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700739 kfree_skb(skb);
740 drop_count++;
741 }
742 qdisc_tree_decrease_qlen(sch, drop_count);
743
744 sch_tree_unlock(sch);
745 return err;
746}
747
748static void fq_destroy(struct Qdisc *sch)
749{
750 struct fq_sched_data *q = qdisc_priv(sch);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700751
Eric Dumazet8d34ce12013-09-27 14:20:01 -0700752 fq_reset(sch);
Eric Dumazetc3bd8542013-12-15 13:15:25 -0800753 fq_free(q->fq_root);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700754 qdisc_watchdog_cancel(&q->watchdog);
755}
756
757static int fq_init(struct Qdisc *sch, struct nlattr *opt)
758{
759 struct fq_sched_data *q = qdisc_priv(sch);
760 int err;
761
762 sch->limit = 10000;
763 q->flow_plimit = 100;
764 q->quantum = 2 * psched_mtu(qdisc_dev(sch));
765 q->initial_quantum = 10 * psched_mtu(qdisc_dev(sch));
Eric Dumazetf52ed892013-11-15 08:58:14 -0800766 q->flow_refill_delay = msecs_to_jiffies(40);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700767 q->flow_max_rate = ~0U;
768 q->rate_enable = 1;
769 q->new_flows.first = NULL;
770 q->old_flows.first = NULL;
771 q->delayed = RB_ROOT;
772 q->fq_root = NULL;
773 q->fq_trees_log = ilog2(1024);
Eric Dumazet06eb3952015-02-04 21:30:40 -0800774 q->orphan_mask = 1024 - 1;
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700775 qdisc_watchdog_init(&q->watchdog, sch);
776
777 if (opt)
778 err = fq_change(sch, opt);
779 else
Eric Dumazetc3bd8542013-12-15 13:15:25 -0800780 err = fq_resize(sch, q->fq_trees_log);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700781
782 return err;
783}
784
785static int fq_dump(struct Qdisc *sch, struct sk_buff *skb)
786{
787 struct fq_sched_data *q = qdisc_priv(sch);
788 struct nlattr *opts;
789
790 opts = nla_nest_start(skb, TCA_OPTIONS);
791 if (opts == NULL)
792 goto nla_put_failure;
793
Eric Dumazet65c5189a2013-11-15 08:57:26 -0800794 /* TCA_FQ_FLOW_DEFAULT_RATE is not used anymore */
795
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700796 if (nla_put_u32(skb, TCA_FQ_PLIMIT, sch->limit) ||
797 nla_put_u32(skb, TCA_FQ_FLOW_PLIMIT, q->flow_plimit) ||
798 nla_put_u32(skb, TCA_FQ_QUANTUM, q->quantum) ||
799 nla_put_u32(skb, TCA_FQ_INITIAL_QUANTUM, q->initial_quantum) ||
800 nla_put_u32(skb, TCA_FQ_RATE_ENABLE, q->rate_enable) ||
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700801 nla_put_u32(skb, TCA_FQ_FLOW_MAX_RATE, q->flow_max_rate) ||
Eric Dumazetf52ed892013-11-15 08:58:14 -0800802 nla_put_u32(skb, TCA_FQ_FLOW_REFILL_DELAY,
803 jiffies_to_usecs(q->flow_refill_delay)) ||
Eric Dumazet06eb3952015-02-04 21:30:40 -0800804 nla_put_u32(skb, TCA_FQ_ORPHAN_MASK, q->orphan_mask) ||
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700805 nla_put_u32(skb, TCA_FQ_BUCKETS_LOG, q->fq_trees_log))
806 goto nla_put_failure;
807
Yang Yingliangd59b7d82014-03-12 10:20:32 +0800808 return nla_nest_end(skb, opts);
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700809
810nla_put_failure:
811 return -1;
812}
813
814static int fq_dump_stats(struct Qdisc *sch, struct gnet_dump *d)
815{
816 struct fq_sched_data *q = qdisc_priv(sch);
Eric Dumazetd2de8752014-08-22 18:32:09 -0700817 u64 now = ktime_get_ns();
Eric Dumazetafe4fd02013-08-29 15:49:55 -0700818 struct tc_fq_qd_stats st = {
819 .gc_flows = q->stat_gc_flows,
820 .highprio_packets = q->stat_internal_packets,
821 .tcp_retrans = q->stat_tcp_retrans,
822 .throttled = q->stat_throttled,
823 .flows_plimit = q->stat_flows_plimit,
824 .pkts_too_long = q->stat_pkts_too_long,
825 .allocation_errors = q->stat_allocation_errors,
826 .flows = q->flows,
827 .inactive_flows = q->inactive_flows,
828 .throttled_flows = q->throttled_flows,
829 .time_next_delayed_flow = q->time_next_delayed_flow - now,
830 };
831
832 return gnet_stats_copy_app(d, &st, sizeof(st));
833}
834
835static struct Qdisc_ops fq_qdisc_ops __read_mostly = {
836 .id = "fq",
837 .priv_size = sizeof(struct fq_sched_data),
838
839 .enqueue = fq_enqueue,
840 .dequeue = fq_dequeue,
841 .peek = qdisc_peek_dequeued,
842 .init = fq_init,
843 .reset = fq_reset,
844 .destroy = fq_destroy,
845 .change = fq_change,
846 .dump = fq_dump,
847 .dump_stats = fq_dump_stats,
848 .owner = THIS_MODULE,
849};
850
851static int __init fq_module_init(void)
852{
853 int ret;
854
855 fq_flow_cachep = kmem_cache_create("fq_flow_cache",
856 sizeof(struct fq_flow),
857 0, 0, NULL);
858 if (!fq_flow_cachep)
859 return -ENOMEM;
860
861 ret = register_qdisc(&fq_qdisc_ops);
862 if (ret)
863 kmem_cache_destroy(fq_flow_cachep);
864 return ret;
865}
866
867static void __exit fq_module_exit(void)
868{
869 unregister_qdisc(&fq_qdisc_ops);
870 kmem_cache_destroy(fq_flow_cachep);
871}
872
873module_init(fq_module_init)
874module_exit(fq_module_exit)
875MODULE_AUTHOR("Eric Dumazet");
876MODULE_LICENSE("GPL");