sch_fq.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815
  1. /*
  2. * net/sched/sch_fq.c Fair Queue Packet Scheduler (per flow pacing)
  3. *
  4. * Copyright (C) 2013 Eric Dumazet <edumazet@google.com>
  5. *
  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. *
  11. * Meant to be mostly used for localy generated traffic :
  12. * 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. #include <linux/module.h>
  37. #include <linux/types.h>
  38. #include <linux/kernel.h>
  39. #include <linux/jiffies.h>
  40. #include <linux/string.h>
  41. #include <linux/in.h>
  42. #include <linux/errno.h>
  43. #include <linux/init.h>
  44. #include <linux/skbuff.h>
  45. #include <linux/slab.h>
  46. #include <linux/rbtree.h>
  47. #include <linux/hash.h>
  48. #include <linux/prefetch.h>
  49. #include <net/netlink.h>
  50. #include <net/pkt_sched.h>
  51. #include <net/sock.h>
  52. #include <net/tcp_states.h>
  53. /*
  54. * Per flow structure, dynamically allocated
  55. */
  56. struct fq_flow {
  57. struct sk_buff *head; /* list of skbs for this flow : first skb */
  58. union {
  59. struct sk_buff *tail; /* last skb in the list */
  60. unsigned long age; /* jiffies when flow was emptied, for gc */
  61. };
  62. struct rb_node fq_node; /* anchor in fq_root[] trees */
  63. struct sock *sk;
  64. int qlen; /* number of packets in flow queue */
  65. int credit;
  66. u32 socket_hash; /* sk_hash */
  67. struct fq_flow *next; /* next pointer in RR lists, or &detached */
  68. struct rb_node rate_node; /* anchor in q->delayed tree */
  69. u64 time_next_packet;
  70. };
  71. struct fq_flow_head {
  72. struct fq_flow *first;
  73. struct fq_flow *last;
  74. };
  75. struct fq_sched_data {
  76. struct fq_flow_head new_flows;
  77. struct fq_flow_head old_flows;
  78. struct rb_root delayed; /* for rate limited flows */
  79. u64 time_next_delayed_flow;
  80. struct fq_flow internal; /* for non classified or high prio packets */
  81. u32 quantum;
  82. u32 initial_quantum;
  83. u32 flow_default_rate;/* rate per flow : bytes per second */
  84. u32 flow_max_rate; /* optional max rate per flow */
  85. u32 flow_plimit; /* max packets per flow */
  86. struct rb_root *fq_root;
  87. u8 rate_enable;
  88. u8 fq_trees_log;
  89. u32 flows;
  90. u32 inactive_flows;
  91. u32 throttled_flows;
  92. u64 stat_gc_flows;
  93. u64 stat_internal_packets;
  94. u64 stat_tcp_retrans;
  95. u64 stat_throttled;
  96. u64 stat_flows_plimit;
  97. u64 stat_pkts_too_long;
  98. u64 stat_allocation_errors;
  99. struct qdisc_watchdog watchdog;
  100. };
  101. /* special value to mark a detached flow (not on old/new list) */
  102. static struct fq_flow detached, throttled;
  103. static void fq_flow_set_detached(struct fq_flow *f)
  104. {
  105. f->next = &detached;
  106. }
  107. static bool fq_flow_is_detached(const struct fq_flow *f)
  108. {
  109. return f->next == &detached;
  110. }
  111. static void fq_flow_set_throttled(struct fq_sched_data *q, struct fq_flow *f)
  112. {
  113. struct rb_node **p = &q->delayed.rb_node, *parent = NULL;
  114. while (*p) {
  115. struct fq_flow *aux;
  116. parent = *p;
  117. aux = container_of(parent, struct fq_flow, rate_node);
  118. if (f->time_next_packet >= aux->time_next_packet)
  119. p = &parent->rb_right;
  120. else
  121. p = &parent->rb_left;
  122. }
  123. rb_link_node(&f->rate_node, parent, p);
  124. rb_insert_color(&f->rate_node, &q->delayed);
  125. q->throttled_flows++;
  126. q->stat_throttled++;
  127. f->next = &throttled;
  128. if (q->time_next_delayed_flow > f->time_next_packet)
  129. q->time_next_delayed_flow = f->time_next_packet;
  130. }
  131. static struct kmem_cache *fq_flow_cachep __read_mostly;
  132. static void fq_flow_add_tail(struct fq_flow_head *head, struct fq_flow *flow)
  133. {
  134. if (head->first)
  135. head->last->next = flow;
  136. else
  137. head->first = flow;
  138. head->last = flow;
  139. flow->next = NULL;
  140. }
  141. /* limit number of collected flows per round */
  142. #define FQ_GC_MAX 8
  143. #define FQ_GC_AGE (3*HZ)
  144. static bool fq_gc_candidate(const struct fq_flow *f)
  145. {
  146. return fq_flow_is_detached(f) &&
  147. time_after(jiffies, f->age + FQ_GC_AGE);
  148. }
  149. static void fq_gc(struct fq_sched_data *q,
  150. struct rb_root *root,
  151. struct sock *sk)
  152. {
  153. struct fq_flow *f, *tofree[FQ_GC_MAX];
  154. struct rb_node **p, *parent;
  155. int fcnt = 0;
  156. p = &root->rb_node;
  157. parent = NULL;
  158. while (*p) {
  159. parent = *p;
  160. f = container_of(parent, struct fq_flow, fq_node);
  161. if (f->sk == sk)
  162. break;
  163. if (fq_gc_candidate(f)) {
  164. tofree[fcnt++] = f;
  165. if (fcnt == FQ_GC_MAX)
  166. break;
  167. }
  168. if (f->sk > sk)
  169. p = &parent->rb_right;
  170. else
  171. p = &parent->rb_left;
  172. }
  173. q->flows -= fcnt;
  174. q->inactive_flows -= fcnt;
  175. q->stat_gc_flows += fcnt;
  176. while (fcnt) {
  177. struct fq_flow *f = tofree[--fcnt];
  178. rb_erase(&f->fq_node, root);
  179. kmem_cache_free(fq_flow_cachep, f);
  180. }
  181. }
  182. static const u8 prio2band[TC_PRIO_MAX + 1] = {
  183. 1, 2, 2, 2, 1, 2, 0, 0 , 1, 1, 1, 1, 1, 1, 1, 1
  184. };
  185. static struct fq_flow *fq_classify(struct sk_buff *skb, struct fq_sched_data *q)
  186. {
  187. struct rb_node **p, *parent;
  188. struct sock *sk = skb->sk;
  189. struct rb_root *root;
  190. struct fq_flow *f;
  191. int band;
  192. /* warning: no starvation prevention... */
  193. band = prio2band[skb->priority & TC_PRIO_MAX];
  194. if (unlikely(band == 0))
  195. return &q->internal;
  196. if (unlikely(!sk)) {
  197. /* By forcing low order bit to 1, we make sure to not
  198. * collide with a local flow (socket pointers are word aligned)
  199. */
  200. sk = (struct sock *)(skb_get_rxhash(skb) | 1L);
  201. }
  202. root = &q->fq_root[hash_32((u32)(long)sk, q->fq_trees_log)];
  203. if (q->flows >= (2U << q->fq_trees_log) &&
  204. q->inactive_flows > q->flows/2)
  205. fq_gc(q, root, sk);
  206. p = &root->rb_node;
  207. parent = NULL;
  208. while (*p) {
  209. parent = *p;
  210. f = container_of(parent, struct fq_flow, fq_node);
  211. if (f->sk == sk) {
  212. /* socket might have been reallocated, so check
  213. * if its sk_hash is the same.
  214. * It not, we need to refill credit with
  215. * initial quantum
  216. */
  217. if (unlikely(skb->sk &&
  218. f->socket_hash != sk->sk_hash)) {
  219. f->credit = q->initial_quantum;
  220. f->socket_hash = sk->sk_hash;
  221. }
  222. return f;
  223. }
  224. if (f->sk > sk)
  225. p = &parent->rb_right;
  226. else
  227. p = &parent->rb_left;
  228. }
  229. f = kmem_cache_zalloc(fq_flow_cachep, GFP_ATOMIC | __GFP_NOWARN);
  230. if (unlikely(!f)) {
  231. q->stat_allocation_errors++;
  232. return &q->internal;
  233. }
  234. fq_flow_set_detached(f);
  235. f->sk = sk;
  236. if (skb->sk)
  237. f->socket_hash = sk->sk_hash;
  238. f->credit = q->initial_quantum;
  239. rb_link_node(&f->fq_node, parent, p);
  240. rb_insert_color(&f->fq_node, root);
  241. q->flows++;
  242. q->inactive_flows++;
  243. return f;
  244. }
  245. /* remove one skb from head of flow queue */
  246. static struct sk_buff *fq_dequeue_head(struct Qdisc *sch, struct fq_flow *flow)
  247. {
  248. struct sk_buff *skb = flow->head;
  249. if (skb) {
  250. flow->head = skb->next;
  251. skb->next = NULL;
  252. flow->qlen--;
  253. sch->qstats.backlog -= qdisc_pkt_len(skb);
  254. sch->q.qlen--;
  255. }
  256. return skb;
  257. }
  258. /* We might add in the future detection of retransmits
  259. * For the time being, just return false
  260. */
  261. static bool skb_is_retransmit(struct sk_buff *skb)
  262. {
  263. return false;
  264. }
  265. /* add skb to flow queue
  266. * flow queue is a linked list, kind of FIFO, except for TCP retransmits
  267. * We special case tcp retransmits to be transmitted before other packets.
  268. * We rely on fact that TCP retransmits are unlikely, so we do not waste
  269. * a separate queue or a pointer.
  270. * head-> [retrans pkt 1]
  271. * [retrans pkt 2]
  272. * [ normal pkt 1]
  273. * [ normal pkt 2]
  274. * [ normal pkt 3]
  275. * tail-> [ normal pkt 4]
  276. */
  277. static void flow_queue_add(struct fq_flow *flow, struct sk_buff *skb)
  278. {
  279. struct sk_buff *prev, *head = flow->head;
  280. skb->next = NULL;
  281. if (!head) {
  282. flow->head = skb;
  283. flow->tail = skb;
  284. return;
  285. }
  286. if (likely(!skb_is_retransmit(skb))) {
  287. flow->tail->next = skb;
  288. flow->tail = skb;
  289. return;
  290. }
  291. /* This skb is a tcp retransmit,
  292. * find the last retrans packet in the queue
  293. */
  294. prev = NULL;
  295. while (skb_is_retransmit(head)) {
  296. prev = head;
  297. head = head->next;
  298. if (!head)
  299. break;
  300. }
  301. if (!prev) { /* no rtx packet in queue, become the new head */
  302. skb->next = flow->head;
  303. flow->head = skb;
  304. } else {
  305. if (prev == flow->tail)
  306. flow->tail = skb;
  307. else
  308. skb->next = prev->next;
  309. prev->next = skb;
  310. }
  311. }
  312. static int fq_enqueue(struct sk_buff *skb, struct Qdisc *sch)
  313. {
  314. struct fq_sched_data *q = qdisc_priv(sch);
  315. struct fq_flow *f;
  316. if (unlikely(sch->q.qlen >= sch->limit))
  317. return qdisc_drop(skb, sch);
  318. f = fq_classify(skb, q);
  319. if (unlikely(f->qlen >= q->flow_plimit && f != &q->internal)) {
  320. q->stat_flows_plimit++;
  321. return qdisc_drop(skb, sch);
  322. }
  323. f->qlen++;
  324. flow_queue_add(f, skb);
  325. if (skb_is_retransmit(skb))
  326. q->stat_tcp_retrans++;
  327. sch->qstats.backlog += qdisc_pkt_len(skb);
  328. if (fq_flow_is_detached(f)) {
  329. fq_flow_add_tail(&q->new_flows, f);
  330. if (q->quantum > f->credit)
  331. f->credit = q->quantum;
  332. q->inactive_flows--;
  333. qdisc_unthrottled(sch);
  334. }
  335. if (unlikely(f == &q->internal)) {
  336. q->stat_internal_packets++;
  337. qdisc_unthrottled(sch);
  338. }
  339. sch->q.qlen++;
  340. return NET_XMIT_SUCCESS;
  341. }
  342. static void fq_check_throttled(struct fq_sched_data *q, u64 now)
  343. {
  344. struct rb_node *p;
  345. if (q->time_next_delayed_flow > now)
  346. return;
  347. q->time_next_delayed_flow = ~0ULL;
  348. while ((p = rb_first(&q->delayed)) != NULL) {
  349. struct fq_flow *f = container_of(p, struct fq_flow, rate_node);
  350. if (f->time_next_packet > now) {
  351. q->time_next_delayed_flow = f->time_next_packet;
  352. break;
  353. }
  354. rb_erase(p, &q->delayed);
  355. q->throttled_flows--;
  356. fq_flow_add_tail(&q->old_flows, f);
  357. }
  358. }
  359. static struct sk_buff *fq_dequeue(struct Qdisc *sch)
  360. {
  361. struct fq_sched_data *q = qdisc_priv(sch);
  362. u64 now = ktime_to_ns(ktime_get());
  363. struct fq_flow_head *head;
  364. struct sk_buff *skb;
  365. struct fq_flow *f;
  366. u32 rate;
  367. skb = fq_dequeue_head(sch, &q->internal);
  368. if (skb)
  369. goto out;
  370. fq_check_throttled(q, now);
  371. begin:
  372. head = &q->new_flows;
  373. if (!head->first) {
  374. head = &q->old_flows;
  375. if (!head->first) {
  376. if (q->time_next_delayed_flow != ~0ULL)
  377. qdisc_watchdog_schedule_ns(&q->watchdog,
  378. q->time_next_delayed_flow);
  379. return NULL;
  380. }
  381. }
  382. f = head->first;
  383. if (f->credit <= 0) {
  384. f->credit += q->quantum;
  385. head->first = f->next;
  386. fq_flow_add_tail(&q->old_flows, f);
  387. goto begin;
  388. }
  389. if (unlikely(f->head && now < f->time_next_packet)) {
  390. head->first = f->next;
  391. fq_flow_set_throttled(q, f);
  392. goto begin;
  393. }
  394. skb = fq_dequeue_head(sch, f);
  395. if (!skb) {
  396. head->first = f->next;
  397. /* force a pass through old_flows to prevent starvation */
  398. if ((head == &q->new_flows) && q->old_flows.first) {
  399. fq_flow_add_tail(&q->old_flows, f);
  400. } else {
  401. fq_flow_set_detached(f);
  402. f->age = jiffies;
  403. q->inactive_flows++;
  404. }
  405. goto begin;
  406. }
  407. prefetch(&skb->end);
  408. f->time_next_packet = now;
  409. f->credit -= qdisc_pkt_len(skb);
  410. if (f->credit > 0 || !q->rate_enable)
  411. goto out;
  412. rate = q->flow_max_rate;
  413. if (skb->sk && skb->sk->sk_state != TCP_TIME_WAIT)
  414. rate = min(skb->sk->sk_pacing_rate, rate);
  415. if (rate != ~0U) {
  416. u32 plen = max(qdisc_pkt_len(skb), q->quantum);
  417. u64 len = (u64)plen * NSEC_PER_SEC;
  418. if (likely(rate))
  419. do_div(len, rate);
  420. /* Since socket rate can change later,
  421. * clamp the delay to 125 ms.
  422. * TODO: maybe segment the too big skb, as in commit
  423. * e43ac79a4bc ("sch_tbf: segment too big GSO packets")
  424. */
  425. if (unlikely(len > 125 * NSEC_PER_MSEC)) {
  426. len = 125 * NSEC_PER_MSEC;
  427. q->stat_pkts_too_long++;
  428. }
  429. f->time_next_packet = now + len;
  430. }
  431. out:
  432. qdisc_bstats_update(sch, skb);
  433. qdisc_unthrottled(sch);
  434. return skb;
  435. }
  436. static void fq_reset(struct Qdisc *sch)
  437. {
  438. struct fq_sched_data *q = qdisc_priv(sch);
  439. struct rb_root *root;
  440. struct sk_buff *skb;
  441. struct rb_node *p;
  442. struct fq_flow *f;
  443. unsigned int idx;
  444. while ((skb = fq_dequeue_head(sch, &q->internal)) != NULL)
  445. kfree_skb(skb);
  446. if (!q->fq_root)
  447. return;
  448. for (idx = 0; idx < (1U << q->fq_trees_log); idx++) {
  449. root = &q->fq_root[idx];
  450. while ((p = rb_first(root)) != NULL) {
  451. f = container_of(p, struct fq_flow, fq_node);
  452. rb_erase(p, root);
  453. while ((skb = fq_dequeue_head(sch, f)) != NULL)
  454. kfree_skb(skb);
  455. kmem_cache_free(fq_flow_cachep, f);
  456. }
  457. }
  458. q->new_flows.first = NULL;
  459. q->old_flows.first = NULL;
  460. q->delayed = RB_ROOT;
  461. q->flows = 0;
  462. q->inactive_flows = 0;
  463. q->throttled_flows = 0;
  464. }
  465. static void fq_rehash(struct fq_sched_data *q,
  466. struct rb_root *old_array, u32 old_log,
  467. struct rb_root *new_array, u32 new_log)
  468. {
  469. struct rb_node *op, **np, *parent;
  470. struct rb_root *oroot, *nroot;
  471. struct fq_flow *of, *nf;
  472. int fcnt = 0;
  473. u32 idx;
  474. for (idx = 0; idx < (1U << old_log); idx++) {
  475. oroot = &old_array[idx];
  476. while ((op = rb_first(oroot)) != NULL) {
  477. rb_erase(op, oroot);
  478. of = container_of(op, struct fq_flow, fq_node);
  479. if (fq_gc_candidate(of)) {
  480. fcnt++;
  481. kmem_cache_free(fq_flow_cachep, of);
  482. continue;
  483. }
  484. nroot = &new_array[hash_32((u32)(long)of->sk, new_log)];
  485. np = &nroot->rb_node;
  486. parent = NULL;
  487. while (*np) {
  488. parent = *np;
  489. nf = container_of(parent, struct fq_flow, fq_node);
  490. BUG_ON(nf->sk == of->sk);
  491. if (nf->sk > of->sk)
  492. np = &parent->rb_right;
  493. else
  494. np = &parent->rb_left;
  495. }
  496. rb_link_node(&of->fq_node, parent, np);
  497. rb_insert_color(&of->fq_node, nroot);
  498. }
  499. }
  500. q->flows -= fcnt;
  501. q->inactive_flows -= fcnt;
  502. q->stat_gc_flows += fcnt;
  503. }
  504. static int fq_resize(struct fq_sched_data *q, u32 log)
  505. {
  506. struct rb_root *array;
  507. u32 idx;
  508. if (q->fq_root && log == q->fq_trees_log)
  509. return 0;
  510. array = kmalloc(sizeof(struct rb_root) << log, GFP_KERNEL);
  511. if (!array)
  512. return -ENOMEM;
  513. for (idx = 0; idx < (1U << log); idx++)
  514. array[idx] = RB_ROOT;
  515. if (q->fq_root) {
  516. fq_rehash(q, q->fq_root, q->fq_trees_log, array, log);
  517. kfree(q->fq_root);
  518. }
  519. q->fq_root = array;
  520. q->fq_trees_log = log;
  521. return 0;
  522. }
  523. static const struct nla_policy fq_policy[TCA_FQ_MAX + 1] = {
  524. [TCA_FQ_PLIMIT] = { .type = NLA_U32 },
  525. [TCA_FQ_FLOW_PLIMIT] = { .type = NLA_U32 },
  526. [TCA_FQ_QUANTUM] = { .type = NLA_U32 },
  527. [TCA_FQ_INITIAL_QUANTUM] = { .type = NLA_U32 },
  528. [TCA_FQ_RATE_ENABLE] = { .type = NLA_U32 },
  529. [TCA_FQ_FLOW_DEFAULT_RATE] = { .type = NLA_U32 },
  530. [TCA_FQ_FLOW_MAX_RATE] = { .type = NLA_U32 },
  531. [TCA_FQ_BUCKETS_LOG] = { .type = NLA_U32 },
  532. };
  533. static int fq_change(struct Qdisc *sch, struct nlattr *opt)
  534. {
  535. struct fq_sched_data *q = qdisc_priv(sch);
  536. struct nlattr *tb[TCA_FQ_MAX + 1];
  537. int err, drop_count = 0;
  538. u32 fq_log;
  539. if (!opt)
  540. return -EINVAL;
  541. err = nla_parse_nested(tb, TCA_FQ_MAX, opt, fq_policy);
  542. if (err < 0)
  543. return err;
  544. sch_tree_lock(sch);
  545. fq_log = q->fq_trees_log;
  546. if (tb[TCA_FQ_BUCKETS_LOG]) {
  547. u32 nval = nla_get_u32(tb[TCA_FQ_BUCKETS_LOG]);
  548. if (nval >= 1 && nval <= ilog2(256*1024))
  549. fq_log = nval;
  550. else
  551. err = -EINVAL;
  552. }
  553. if (tb[TCA_FQ_PLIMIT])
  554. sch->limit = nla_get_u32(tb[TCA_FQ_PLIMIT]);
  555. if (tb[TCA_FQ_FLOW_PLIMIT])
  556. q->flow_plimit = nla_get_u32(tb[TCA_FQ_FLOW_PLIMIT]);
  557. if (tb[TCA_FQ_QUANTUM])
  558. q->quantum = nla_get_u32(tb[TCA_FQ_QUANTUM]);
  559. if (tb[TCA_FQ_INITIAL_QUANTUM])
  560. q->initial_quantum = nla_get_u32(tb[TCA_FQ_INITIAL_QUANTUM]);
  561. if (tb[TCA_FQ_FLOW_DEFAULT_RATE])
  562. q->flow_default_rate = nla_get_u32(tb[TCA_FQ_FLOW_DEFAULT_RATE]);
  563. if (tb[TCA_FQ_FLOW_MAX_RATE])
  564. q->flow_max_rate = nla_get_u32(tb[TCA_FQ_FLOW_MAX_RATE]);
  565. if (tb[TCA_FQ_RATE_ENABLE]) {
  566. u32 enable = nla_get_u32(tb[TCA_FQ_RATE_ENABLE]);
  567. if (enable <= 1)
  568. q->rate_enable = enable;
  569. else
  570. err = -EINVAL;
  571. }
  572. if (!err)
  573. err = fq_resize(q, fq_log);
  574. while (sch->q.qlen > sch->limit) {
  575. struct sk_buff *skb = fq_dequeue(sch);
  576. if (!skb)
  577. break;
  578. kfree_skb(skb);
  579. drop_count++;
  580. }
  581. qdisc_tree_decrease_qlen(sch, drop_count);
  582. sch_tree_unlock(sch);
  583. return err;
  584. }
  585. static void fq_destroy(struct Qdisc *sch)
  586. {
  587. struct fq_sched_data *q = qdisc_priv(sch);
  588. fq_reset(sch);
  589. kfree(q->fq_root);
  590. qdisc_watchdog_cancel(&q->watchdog);
  591. }
  592. static int fq_init(struct Qdisc *sch, struct nlattr *opt)
  593. {
  594. struct fq_sched_data *q = qdisc_priv(sch);
  595. int err;
  596. sch->limit = 10000;
  597. q->flow_plimit = 100;
  598. q->quantum = 2 * psched_mtu(qdisc_dev(sch));
  599. q->initial_quantum = 10 * psched_mtu(qdisc_dev(sch));
  600. q->flow_default_rate = 0;
  601. q->flow_max_rate = ~0U;
  602. q->rate_enable = 1;
  603. q->new_flows.first = NULL;
  604. q->old_flows.first = NULL;
  605. q->delayed = RB_ROOT;
  606. q->fq_root = NULL;
  607. q->fq_trees_log = ilog2(1024);
  608. qdisc_watchdog_init(&q->watchdog, sch);
  609. if (opt)
  610. err = fq_change(sch, opt);
  611. else
  612. err = fq_resize(q, q->fq_trees_log);
  613. return err;
  614. }
  615. static int fq_dump(struct Qdisc *sch, struct sk_buff *skb)
  616. {
  617. struct fq_sched_data *q = qdisc_priv(sch);
  618. struct nlattr *opts;
  619. opts = nla_nest_start(skb, TCA_OPTIONS);
  620. if (opts == NULL)
  621. goto nla_put_failure;
  622. /* TCA_FQ_FLOW_DEFAULT_RATE is not used anymore,
  623. * do not bother giving its value
  624. */
  625. if (nla_put_u32(skb, TCA_FQ_PLIMIT, sch->limit) ||
  626. nla_put_u32(skb, TCA_FQ_FLOW_PLIMIT, q->flow_plimit) ||
  627. nla_put_u32(skb, TCA_FQ_QUANTUM, q->quantum) ||
  628. nla_put_u32(skb, TCA_FQ_INITIAL_QUANTUM, q->initial_quantum) ||
  629. nla_put_u32(skb, TCA_FQ_RATE_ENABLE, q->rate_enable) ||
  630. nla_put_u32(skb, TCA_FQ_FLOW_MAX_RATE, q->flow_max_rate) ||
  631. nla_put_u32(skb, TCA_FQ_BUCKETS_LOG, q->fq_trees_log))
  632. goto nla_put_failure;
  633. nla_nest_end(skb, opts);
  634. return skb->len;
  635. nla_put_failure:
  636. return -1;
  637. }
  638. static int fq_dump_stats(struct Qdisc *sch, struct gnet_dump *d)
  639. {
  640. struct fq_sched_data *q = qdisc_priv(sch);
  641. u64 now = ktime_to_ns(ktime_get());
  642. struct tc_fq_qd_stats st = {
  643. .gc_flows = q->stat_gc_flows,
  644. .highprio_packets = q->stat_internal_packets,
  645. .tcp_retrans = q->stat_tcp_retrans,
  646. .throttled = q->stat_throttled,
  647. .flows_plimit = q->stat_flows_plimit,
  648. .pkts_too_long = q->stat_pkts_too_long,
  649. .allocation_errors = q->stat_allocation_errors,
  650. .flows = q->flows,
  651. .inactive_flows = q->inactive_flows,
  652. .throttled_flows = q->throttled_flows,
  653. .time_next_delayed_flow = q->time_next_delayed_flow - now,
  654. };
  655. return gnet_stats_copy_app(d, &st, sizeof(st));
  656. }
  657. static struct Qdisc_ops fq_qdisc_ops __read_mostly = {
  658. .id = "fq",
  659. .priv_size = sizeof(struct fq_sched_data),
  660. .enqueue = fq_enqueue,
  661. .dequeue = fq_dequeue,
  662. .peek = qdisc_peek_dequeued,
  663. .init = fq_init,
  664. .reset = fq_reset,
  665. .destroy = fq_destroy,
  666. .change = fq_change,
  667. .dump = fq_dump,
  668. .dump_stats = fq_dump_stats,
  669. .owner = THIS_MODULE,
  670. };
  671. static int __init fq_module_init(void)
  672. {
  673. int ret;
  674. fq_flow_cachep = kmem_cache_create("fq_flow_cache",
  675. sizeof(struct fq_flow),
  676. 0, 0, NULL);
  677. if (!fq_flow_cachep)
  678. return -ENOMEM;
  679. ret = register_qdisc(&fq_qdisc_ops);
  680. if (ret)
  681. kmem_cache_destroy(fq_flow_cachep);
  682. return ret;
  683. }
  684. static void __exit fq_module_exit(void)
  685. {
  686. unregister_qdisc(&fq_qdisc_ops);
  687. kmem_cache_destroy(fq_flow_cachep);
  688. }
  689. module_init(fq_module_init)
  690. module_exit(fq_module_exit)
  691. MODULE_AUTHOR("Eric Dumazet");
  692. MODULE_LICENSE("GPL");