Root/target/linux/generic/patches-3.6/620-sched_esfq.patch

1--- a/include/linux/pkt_sched.h
2+++ b/include/linux/pkt_sched.h
3@@ -214,6 +214,33 @@ struct tc_sfq_xstats {
4     __s32 allot;
5 };
6 
7+/* ESFQ section */
8+
9+enum
10+{
11+ /* traditional */
12+ TCA_SFQ_HASH_CLASSIC,
13+ TCA_SFQ_HASH_DST,
14+ TCA_SFQ_HASH_SRC,
15+ TCA_SFQ_HASH_FWMARK,
16+ /* conntrack */
17+ TCA_SFQ_HASH_CTORIGDST,
18+ TCA_SFQ_HASH_CTORIGSRC,
19+ TCA_SFQ_HASH_CTREPLDST,
20+ TCA_SFQ_HASH_CTREPLSRC,
21+ TCA_SFQ_HASH_CTNATCHG,
22+};
23+
24+struct tc_esfq_qopt
25+{
26+ unsigned quantum; /* Bytes per round allocated to flow */
27+ int perturb_period; /* Period of hash perturbation */
28+ __u32 limit; /* Maximal packets in queue */
29+ unsigned divisor; /* Hash divisor */
30+ unsigned flows; /* Maximal number of flows */
31+ unsigned hash_kind; /* Hash function to use for flow identification */
32+};
33+
34 /* RED section */
35 
36 enum {
37--- a/net/sched/Kconfig
38+++ b/net/sched/Kconfig
39@@ -148,6 +148,37 @@ config NET_SCH_SFQ
40       To compile this code as a module, choose M here: the
41       module will be called sch_sfq.
42 
43+config NET_SCH_ESFQ
44+ tristate "Enhanced Stochastic Fairness Queueing (ESFQ)"
45+ ---help---
46+ Say Y here if you want to use the Enhanced Stochastic Fairness
47+ Queueing (ESFQ) packet scheduling algorithm for some of your network
48+ devices or as a leaf discipline for a classful qdisc such as HTB or
49+ CBQ (see the top of <file:net/sched/sch_esfq.c> for details and
50+ references to the SFQ algorithm).
51+
52+ This is an enchanced SFQ version which allows you to control some
53+ hardcoded values in the SFQ scheduler.
54+
55+ ESFQ also adds control of the hash function used to identify packet
56+ flows. The original SFQ discipline hashes by connection; ESFQ add
57+ several other hashing methods, such as by src IP or by dst IP, which
58+ can be more fair to users in some networking situations.
59+
60+ To compile this code as a module, choose M here: the
61+ module will be called sch_esfq.
62+
63+config NET_SCH_ESFQ_NFCT
64+ bool "Connection Tracking Hash Types"
65+ depends on NET_SCH_ESFQ && NF_CONNTRACK
66+ ---help---
67+ Say Y here to enable support for hashing based on netfilter connection
68+ tracking information. This is useful for a router that is also using
69+ NAT to connect privately-addressed hosts to the Internet. If you want
70+ to provide fair distribution of upstream bandwidth, ESFQ must use
71+ connection tracking information, since all outgoing packets will share
72+ the same source address.
73+
74 config NET_SCH_TEQL
75     tristate "True Link Equalizer (TEQL)"
76     ---help---
77--- a/net/sched/Makefile
78+++ b/net/sched/Makefile
79@@ -26,6 +26,7 @@ obj-$(CONFIG_NET_SCH_INGRESS) += sch_ing
80 obj-$(CONFIG_NET_SCH_DSMARK) += sch_dsmark.o
81 obj-$(CONFIG_NET_SCH_SFB) += sch_sfb.o
82 obj-$(CONFIG_NET_SCH_SFQ) += sch_sfq.o
83+obj-$(CONFIG_NET_SCH_ESFQ) += sch_esfq.o
84 obj-$(CONFIG_NET_SCH_TBF) += sch_tbf.o
85 obj-$(CONFIG_NET_SCH_TEQL) += sch_teql.o
86 obj-$(CONFIG_NET_SCH_PRIO) += sch_prio.o
87--- /dev/null
88+++ b/net/sched/sch_esfq.c
89@@ -0,0 +1,702 @@
90+/*
91+ * net/sched/sch_esfq.c Extended Stochastic Fairness Queueing discipline.
92+ *
93+ * This program is free software; you can redistribute it and/or
94+ * modify it under the terms of the GNU General Public License
95+ * as published by the Free Software Foundation; either version
96+ * 2 of the License, or (at your option) any later version.
97+ *
98+ * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
99+ *
100+ * Changes: Alexander Atanasov, <alex@ssi.bg>
101+ * Added dynamic depth,limit,divisor,hash_kind options.
102+ * Added dst and src hashes.
103+ *
104+ * Alexander Clouter, <alex@digriz.org.uk>
105+ * Ported ESFQ to Linux 2.6.
106+ *
107+ * Corey Hickey, <bugfood-c@fatooh.org>
108+ * Maintenance of the Linux 2.6 port.
109+ * Added fwmark hash (thanks to Robert Kurjata).
110+ * Added usage of jhash.
111+ * Added conntrack support.
112+ * Added ctnatchg hash (thanks to Ben Pfountz).
113+ */
114+
115+#include <linux/module.h>
116+#include <asm/uaccess.h>
117+#include <linux/bitops.h>
118+#include <linux/types.h>
119+#include <linux/kernel.h>
120+#include <linux/jiffies.h>
121+#include <linux/string.h>
122+#include <linux/mm.h>
123+#include <linux/socket.h>
124+#include <linux/sockios.h>
125+#include <linux/in.h>
126+#include <linux/errno.h>
127+#include <linux/interrupt.h>
128+#include <linux/if_ether.h>
129+#include <linux/inet.h>
130+#include <linux/netdevice.h>
131+#include <linux/etherdevice.h>
132+#include <linux/notifier.h>
133+#include <linux/init.h>
134+#include <net/ip.h>
135+#include <net/netlink.h>
136+#include <linux/ipv6.h>
137+#include <net/route.h>
138+#include <linux/skbuff.h>
139+#include <net/sock.h>
140+#include <net/pkt_sched.h>
141+#include <linux/jhash.h>
142+#ifdef CONFIG_NET_SCH_ESFQ_NFCT
143+#include <net/netfilter/nf_conntrack.h>
144+#endif
145+
146+/* Stochastic Fairness Queuing algorithm.
147+ For more comments look at sch_sfq.c.
148+ The difference is that you can change limit, depth,
149+ hash table size and choose alternate hash types.
150+
151+ classic: same as in sch_sfq.c
152+ dst: destination IP address
153+ src: source IP address
154+ fwmark: netfilter mark value
155+ ctorigdst: original destination IP address
156+ ctorigsrc: original source IP address
157+ ctrepldst: reply destination IP address
158+ ctreplsrc: reply source IP
159+
160+*/
161+
162+#define ESFQ_HEAD 0
163+#define ESFQ_TAIL 1
164+
165+/* This type should contain at least SFQ_DEPTH*2 values */
166+typedef unsigned int esfq_index;
167+
168+struct esfq_head
169+{
170+ esfq_index next;
171+ esfq_index prev;
172+};
173+
174+struct esfq_sched_data
175+{
176+/* Parameters */
177+ int perturb_period;
178+ unsigned quantum; /* Allotment per round: MUST BE >= MTU */
179+ int limit;
180+ unsigned depth;
181+ unsigned hash_divisor;
182+ unsigned hash_kind;
183+/* Variables */
184+ struct timer_list perturb_timer;
185+ int perturbation;
186+ esfq_index tail; /* Index of current slot in round */
187+ esfq_index max_depth; /* Maximal depth */
188+
189+ esfq_index *ht; /* Hash table */
190+ esfq_index *next; /* Active slots link */
191+ short *allot; /* Current allotment per slot */
192+ unsigned short *hash; /* Hash value indexed by slots */
193+ struct sk_buff_head *qs; /* Slot queue */
194+ struct esfq_head *dep; /* Linked list of slots, indexed by depth */
195+};
196+
197+/* This contains the info we will hash. */
198+struct esfq_packet_info
199+{
200+ u32 proto; /* protocol or port */
201+ u32 src; /* source from packet header */
202+ u32 dst; /* destination from packet header */
203+ u32 ctorigsrc; /* original source from conntrack */
204+ u32 ctorigdst; /* original destination from conntrack */
205+ u32 ctreplsrc; /* reply source from conntrack */
206+ u32 ctrepldst; /* reply destination from conntrack */
207+ u32 mark; /* netfilter mark (fwmark) */
208+};
209+
210+static __inline__ unsigned esfq_jhash_1word(struct esfq_sched_data *q,u32 a)
211+{
212+ return jhash_1word(a, q->perturbation) & (q->hash_divisor-1);
213+}
214+
215+static __inline__ unsigned esfq_jhash_2words(struct esfq_sched_data *q, u32 a, u32 b)
216+{
217+ return jhash_2words(a, b, q->perturbation) & (q->hash_divisor-1);
218+}
219+
220+static __inline__ unsigned esfq_jhash_3words(struct esfq_sched_data *q, u32 a, u32 b, u32 c)
221+{
222+ return jhash_3words(a, b, c, q->perturbation) & (q->hash_divisor-1);
223+}
224+
225+static unsigned esfq_hash(struct esfq_sched_data *q, struct sk_buff *skb)
226+{
227+ struct esfq_packet_info info;
228+#ifdef CONFIG_NET_SCH_ESFQ_NFCT
229+ enum ip_conntrack_info ctinfo;
230+ struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
231+#endif
232+
233+ switch (skb->protocol) {
234+ case __constant_htons(ETH_P_IP):
235+ {
236+ struct iphdr *iph = ip_hdr(skb);
237+ info.dst = iph->daddr;
238+ info.src = iph->saddr;
239+ if (!(iph->frag_off&htons(IP_MF|IP_OFFSET)) &&
240+ (iph->protocol == IPPROTO_TCP ||
241+ iph->protocol == IPPROTO_UDP ||
242+ iph->protocol == IPPROTO_SCTP ||
243+ iph->protocol == IPPROTO_DCCP ||
244+ iph->protocol == IPPROTO_ESP))
245+ info.proto = *(((u32*)iph) + iph->ihl);
246+ else
247+ info.proto = iph->protocol;
248+ break;
249+ }
250+ case __constant_htons(ETH_P_IPV6):
251+ {
252+ struct ipv6hdr *iph = ipv6_hdr(skb);
253+ /* Hash ipv6 addresses into a u32. This isn't ideal,
254+ * but the code is simple. */
255+ info.dst = jhash2(iph->daddr.s6_addr32, 4, q->perturbation);
256+ info.src = jhash2(iph->saddr.s6_addr32, 4, q->perturbation);
257+ if (iph->nexthdr == IPPROTO_TCP ||
258+ iph->nexthdr == IPPROTO_UDP ||
259+ iph->nexthdr == IPPROTO_SCTP ||
260+ iph->nexthdr == IPPROTO_DCCP ||
261+ iph->nexthdr == IPPROTO_ESP)
262+ info.proto = *(u32*)&iph[1];
263+ else
264+ info.proto = iph->nexthdr;
265+ break;
266+ }
267+ default:
268+ info.dst = (u32)(unsigned long)skb_dst(skb);
269+ info.src = (u32)(unsigned long)skb->sk;
270+ info.proto = skb->protocol;
271+ }
272+
273+ info.mark = skb->mark;
274+
275+#ifdef CONFIG_NET_SCH_ESFQ_NFCT
276+ /* defaults if there is no conntrack info */
277+ info.ctorigsrc = info.src;
278+ info.ctorigdst = info.dst;
279+ info.ctreplsrc = info.dst;
280+ info.ctrepldst = info.src;
281+ /* collect conntrack info */
282+ if (ct && ct != &nf_conntrack_untracked) {
283+ if (skb->protocol == __constant_htons(ETH_P_IP)) {
284+ info.ctorigsrc = ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.u3.ip;
285+ info.ctorigdst = ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.dst.u3.ip;
286+ info.ctreplsrc = ct->tuplehash[IP_CT_DIR_REPLY].tuple.src.u3.ip;
287+ info.ctrepldst = ct->tuplehash[IP_CT_DIR_REPLY].tuple.dst.u3.ip;
288+ }
289+ else if (skb->protocol == __constant_htons(ETH_P_IPV6)) {
290+ /* Again, hash ipv6 addresses into a single u32. */
291+ info.ctorigsrc = jhash2(ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.u3.ip6, 4, q->perturbation);
292+ info.ctorigdst = jhash2(ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.dst.u3.ip6, 4, q->perturbation);
293+ info.ctreplsrc = jhash2(ct->tuplehash[IP_CT_DIR_REPLY].tuple.src.u3.ip6, 4, q->perturbation);
294+ info.ctrepldst = jhash2(ct->tuplehash[IP_CT_DIR_REPLY].tuple.dst.u3.ip6, 4, q->perturbation);
295+ }
296+
297+ }
298+#endif
299+
300+ switch(q->hash_kind) {
301+ case TCA_SFQ_HASH_CLASSIC:
302+ return esfq_jhash_3words(q, info.dst, info.src, info.proto);
303+ case TCA_SFQ_HASH_DST:
304+ return esfq_jhash_1word(q, info.dst);
305+ case TCA_SFQ_HASH_SRC:
306+ return esfq_jhash_1word(q, info.src);
307+ case TCA_SFQ_HASH_FWMARK:
308+ return esfq_jhash_1word(q, info.mark);
309+#ifdef CONFIG_NET_SCH_ESFQ_NFCT
310+ case TCA_SFQ_HASH_CTORIGDST:
311+ return esfq_jhash_1word(q, info.ctorigdst);
312+ case TCA_SFQ_HASH_CTORIGSRC:
313+ return esfq_jhash_1word(q, info.ctorigsrc);
314+ case TCA_SFQ_HASH_CTREPLDST:
315+ return esfq_jhash_1word(q, info.ctrepldst);
316+ case TCA_SFQ_HASH_CTREPLSRC:
317+ return esfq_jhash_1word(q, info.ctreplsrc);
318+ case TCA_SFQ_HASH_CTNATCHG:
319+ {
320+ if (info.ctorigdst == info.ctreplsrc)
321+ return esfq_jhash_1word(q, info.ctorigsrc);
322+ return esfq_jhash_1word(q, info.ctreplsrc);
323+ }
324+#endif
325+ default:
326+ if (net_ratelimit())
327+ printk(KERN_WARNING "ESFQ: Unknown hash method. Falling back to classic.\n");
328+ }
329+ return esfq_jhash_3words(q, info.dst, info.src, info.proto);
330+}
331+
332+static inline void esfq_link(struct esfq_sched_data *q, esfq_index x)
333+{
334+ esfq_index p, n;
335+ int d = q->qs[x].qlen + q->depth;
336+
337+ p = d;
338+ n = q->dep[d].next;
339+ q->dep[x].next = n;
340+ q->dep[x].prev = p;
341+ q->dep[p].next = q->dep[n].prev = x;
342+}
343+
344+static inline void esfq_dec(struct esfq_sched_data *q, esfq_index x)
345+{
346+ esfq_index p, n;
347+
348+ n = q->dep[x].next;
349+ p = q->dep[x].prev;
350+ q->dep[p].next = n;
351+ q->dep[n].prev = p;
352+
353+ if (n == p && q->max_depth == q->qs[x].qlen + 1)
354+ q->max_depth--;
355+
356+ esfq_link(q, x);
357+}
358+
359+static inline void esfq_inc(struct esfq_sched_data *q, esfq_index x)
360+{
361+ esfq_index p, n;
362+ int d;
363+
364+ n = q->dep[x].next;
365+ p = q->dep[x].prev;
366+ q->dep[p].next = n;
367+ q->dep[n].prev = p;
368+ d = q->qs[x].qlen;
369+ if (q->max_depth < d)
370+ q->max_depth = d;
371+
372+ esfq_link(q, x);
373+}
374+
375+static unsigned int esfq_drop(struct Qdisc *sch)
376+{
377+ struct esfq_sched_data *q = qdisc_priv(sch);
378+ esfq_index d = q->max_depth;
379+ struct sk_buff *skb;
380+ unsigned int len;
381+
382+ /* Queue is full! Find the longest slot and
383+ drop a packet from it */
384+
385+ if (d > 1) {
386+ esfq_index x = q->dep[d+q->depth].next;
387+ skb = q->qs[x].prev;
388+ len = skb->len;
389+ __skb_unlink(skb, &q->qs[x]);
390+ kfree_skb(skb);
391+ esfq_dec(q, x);
392+ sch->q.qlen--;
393+ sch->qstats.drops++;
394+ sch->qstats.backlog -= len;
395+ return len;
396+ }
397+
398+ if (d == 1) {
399+ /* It is difficult to believe, but ALL THE SLOTS HAVE LENGTH 1. */
400+ d = q->next[q->tail];
401+ q->next[q->tail] = q->next[d];
402+ q->allot[q->next[d]] += q->quantum;
403+ skb = q->qs[d].prev;
404+ len = skb->len;
405+ __skb_unlink(skb, &q->qs[d]);
406+ kfree_skb(skb);
407+ esfq_dec(q, d);
408+ sch->q.qlen--;
409+ q->ht[q->hash[d]] = q->depth;
410+ sch->qstats.drops++;
411+ sch->qstats.backlog -= len;
412+ return len;
413+ }
414+
415+ return 0;
416+}
417+
418+static void esfq_q_enqueue(struct sk_buff *skb, struct esfq_sched_data *q, unsigned int end)
419+{
420+ unsigned hash = esfq_hash(q, skb);
421+ unsigned depth = q->depth;
422+ esfq_index x;
423+
424+ x = q->ht[hash];
425+ if (x == depth) {
426+ q->ht[hash] = x = q->dep[depth].next;
427+ q->hash[x] = hash;
428+ }
429+
430+ if (end == ESFQ_TAIL)
431+ __skb_queue_tail(&q->qs[x], skb);
432+ else
433+ __skb_queue_head(&q->qs[x], skb);
434+
435+ esfq_inc(q, x);
436+ if (q->qs[x].qlen == 1) { /* The flow is new */
437+ if (q->tail == depth) { /* It is the first flow */
438+ q->tail = x;
439+ q->next[x] = x;
440+ q->allot[x] = q->quantum;
441+ } else {
442+ q->next[x] = q->next[q->tail];
443+ q->next[q->tail] = x;
444+ q->tail = x;
445+ }
446+ }
447+}
448+
449+static int esfq_enqueue(struct sk_buff *skb, struct Qdisc* sch)
450+{
451+ struct esfq_sched_data *q = qdisc_priv(sch);
452+ esfq_q_enqueue(skb, q, ESFQ_TAIL);
453+ sch->qstats.backlog += skb->len;
454+ if (++sch->q.qlen < q->limit-1) {
455+ sch->bstats.bytes += skb->len;
456+ sch->bstats.packets++;
457+ return 0;
458+ }
459+
460+ sch->qstats.drops++;
461+ esfq_drop(sch);
462+ return NET_XMIT_CN;
463+}
464+
465+static struct sk_buff *esfq_peek(struct Qdisc* sch)
466+{
467+ struct esfq_sched_data *q = qdisc_priv(sch);
468+ esfq_index a;
469+
470+ /* No active slots */
471+ if (q->tail == q->depth)
472+ return NULL;
473+
474+ a = q->next[q->tail];
475+ return skb_peek(&q->qs[a]);
476+}
477+
478+static struct sk_buff *esfq_q_dequeue(struct esfq_sched_data *q)
479+{
480+ struct sk_buff *skb;
481+ unsigned depth = q->depth;
482+ esfq_index a, old_a;
483+
484+ /* No active slots */
485+ if (q->tail == depth)
486+ return NULL;
487+
488+ a = old_a = q->next[q->tail];
489+
490+ /* Grab packet */
491+ skb = __skb_dequeue(&q->qs[a]);
492+ esfq_dec(q, a);
493+
494+ /* Is the slot empty? */
495+ if (q->qs[a].qlen == 0) {
496+ q->ht[q->hash[a]] = depth;
497+ a = q->next[a];
498+ if (a == old_a) {
499+ q->tail = depth;
500+ return skb;
501+ }
502+ q->next[q->tail] = a;
503+ q->allot[a] += q->quantum;
504+ } else if ((q->allot[a] -= skb->len) <= 0) {
505+ q->tail = a;
506+ a = q->next[a];
507+ q->allot[a] += q->quantum;
508+ }
509+
510+ return skb;
511+}
512+
513+static struct sk_buff *esfq_dequeue(struct Qdisc* sch)
514+{
515+ struct esfq_sched_data *q = qdisc_priv(sch);
516+ struct sk_buff *skb;
517+
518+ skb = esfq_q_dequeue(q);
519+ if (skb == NULL)
520+ return NULL;
521+ sch->q.qlen--;
522+ sch->qstats.backlog -= skb->len;
523+ return skb;
524+}
525+
526+static void esfq_q_destroy(struct esfq_sched_data *q)
527+{
528+ del_timer(&q->perturb_timer);
529+ if(q->ht)
530+ kfree(q->ht);
531+ if(q->dep)
532+ kfree(q->dep);
533+ if(q->next)
534+ kfree(q->next);
535+ if(q->allot)
536+ kfree(q->allot);
537+ if(q->hash)
538+ kfree(q->hash);
539+ if(q->qs)
540+ kfree(q->qs);
541+}
542+
543+static void esfq_destroy(struct Qdisc *sch)
544+{
545+ struct esfq_sched_data *q = qdisc_priv(sch);
546+ esfq_q_destroy(q);
547+}
548+
549+
550+static void esfq_reset(struct Qdisc* sch)
551+{
552+ struct sk_buff *skb;
553+
554+ while ((skb = esfq_dequeue(sch)) != NULL)
555+ kfree_skb(skb);
556+}
557+
558+static void esfq_perturbation(unsigned long arg)
559+{
560+ struct Qdisc *sch = (struct Qdisc*)arg;
561+ struct esfq_sched_data *q = qdisc_priv(sch);
562+
563+ q->perturbation = net_random()&0x1F;
564+
565+ if (q->perturb_period) {
566+ q->perturb_timer.expires = jiffies + q->perturb_period;
567+ add_timer(&q->perturb_timer);
568+ }
569+}
570+
571+static unsigned int esfq_check_hash(unsigned int kind)
572+{
573+ switch (kind) {
574+ case TCA_SFQ_HASH_CTORIGDST:
575+ case TCA_SFQ_HASH_CTORIGSRC:
576+ case TCA_SFQ_HASH_CTREPLDST:
577+ case TCA_SFQ_HASH_CTREPLSRC:
578+ case TCA_SFQ_HASH_CTNATCHG:
579+#ifndef CONFIG_NET_SCH_ESFQ_NFCT
580+ {
581+ if (net_ratelimit())
582+ printk(KERN_WARNING "ESFQ: Conntrack hash types disabled in kernel config. Falling back to classic.\n");
583+ return TCA_SFQ_HASH_CLASSIC;
584+ }
585+#endif
586+ case TCA_SFQ_HASH_CLASSIC:
587+ case TCA_SFQ_HASH_DST:
588+ case TCA_SFQ_HASH_SRC:
589+ case TCA_SFQ_HASH_FWMARK:
590+ return kind;
591+ default:
592+ {
593+ if (net_ratelimit())
594+ printk(KERN_WARNING "ESFQ: Unknown hash type. Falling back to classic.\n");
595+ return TCA_SFQ_HASH_CLASSIC;
596+ }
597+ }
598+}
599+
600+static int esfq_q_init(struct esfq_sched_data *q, struct nlattr *opt)
601+{
602+ struct tc_esfq_qopt *ctl = nla_data(opt);
603+ esfq_index p = ~0U/2;
604+ int i;
605+
606+ if (opt && opt->nla_len < nla_attr_size(sizeof(*ctl)))
607+ return -EINVAL;
608+
609+ q->perturbation = 0;
610+ q->hash_kind = TCA_SFQ_HASH_CLASSIC;
611+ q->max_depth = 0;
612+ if (opt == NULL) {
613+ q->perturb_period = 0;
614+ q->hash_divisor = 1024;
615+ q->tail = q->limit = q->depth = 128;
616+
617+ } else {
618+ struct tc_esfq_qopt *ctl = nla_data(opt);
619+ if (ctl->quantum)
620+ q->quantum = ctl->quantum;
621+ q->perturb_period = ctl->perturb_period*HZ;
622+ q->hash_divisor = ctl->divisor ? : 1024;
623+ q->tail = q->limit = q->depth = ctl->flows ? : 128;
624+
625+ if ( q->depth > p - 1 )
626+ return -EINVAL;
627+
628+ if (ctl->limit)
629+ q->limit = min_t(u32, ctl->limit, q->depth);
630+
631+ if (ctl->hash_kind) {
632+ q->hash_kind = esfq_check_hash(ctl->hash_kind);
633+ }
634+ }
635+
636+ q->ht = kmalloc(q->hash_divisor*sizeof(esfq_index), GFP_KERNEL);
637+ if (!q->ht)
638+ goto err_case;
639+ q->dep = kmalloc((1+q->depth*2)*sizeof(struct esfq_head), GFP_KERNEL);
640+ if (!q->dep)
641+ goto err_case;
642+ q->next = kmalloc(q->depth*sizeof(esfq_index), GFP_KERNEL);
643+ if (!q->next)
644+ goto err_case;
645+ q->allot = kmalloc(q->depth*sizeof(short), GFP_KERNEL);
646+ if (!q->allot)
647+ goto err_case;
648+ q->hash = kmalloc(q->depth*sizeof(unsigned short), GFP_KERNEL);
649+ if (!q->hash)
650+ goto err_case;
651+ q->qs = kmalloc(q->depth*sizeof(struct sk_buff_head), GFP_KERNEL);
652+ if (!q->qs)
653+ goto err_case;
654+
655+ for (i=0; i< q->hash_divisor; i++)
656+ q->ht[i] = q->depth;
657+ for (i=0; i<q->depth; i++) {
658+ skb_queue_head_init(&q->qs[i]);
659+ q->dep[i+q->depth].next = i+q->depth;
660+ q->dep[i+q->depth].prev = i+q->depth;
661+ }
662+
663+ for (i=0; i<q->depth; i++)
664+ esfq_link(q, i);
665+ return 0;
666+err_case:
667+ esfq_q_destroy(q);
668+ return -ENOBUFS;
669+}
670+
671+static int esfq_init(struct Qdisc *sch, struct nlattr *opt)
672+{
673+ struct esfq_sched_data *q = qdisc_priv(sch);
674+ int err;
675+
676+ q->quantum = psched_mtu(qdisc_dev(sch)); /* default */
677+ if ((err = esfq_q_init(q, opt)))
678+ return err;
679+
680+ init_timer(&q->perturb_timer);
681+ q->perturb_timer.data = (unsigned long)sch;
682+ q->perturb_timer.function = esfq_perturbation;
683+ if (q->perturb_period) {
684+ q->perturb_timer.expires = jiffies + q->perturb_period;
685+ add_timer(&q->perturb_timer);
686+ }
687+
688+ return 0;
689+}
690+
691+static int esfq_change(struct Qdisc *sch, struct nlattr *opt)
692+{
693+ struct esfq_sched_data *q = qdisc_priv(sch);
694+ struct esfq_sched_data new;
695+ struct sk_buff *skb;
696+ int err;
697+
698+ /* set up new queue */
699+ memset(&new, 0, sizeof(struct esfq_sched_data));
700+ new.quantum = psched_mtu(qdisc_dev(sch)); /* default */
701+ if ((err = esfq_q_init(&new, opt)))
702+ return err;
703+
704+ /* copy all packets from the old queue to the new queue */
705+ sch_tree_lock(sch);
706+ while ((skb = esfq_q_dequeue(q)) != NULL)
707+ esfq_q_enqueue(skb, &new, ESFQ_TAIL);
708+
709+ /* clean up the old queue */
710+ esfq_q_destroy(q);
711+
712+ /* copy elements of the new queue into the old queue */
713+ q->perturb_period = new.perturb_period;
714+ q->quantum = new.quantum;
715+ q->limit = new.limit;
716+ q->depth = new.depth;
717+ q->hash_divisor = new.hash_divisor;
718+ q->hash_kind = new.hash_kind;
719+ q->tail = new.tail;
720+ q->max_depth = new.max_depth;
721+ q->ht = new.ht;
722+ q->dep = new.dep;
723+ q->next = new.next;
724+ q->allot = new.allot;
725+ q->hash = new.hash;
726+ q->qs = new.qs;
727+
728+ /* finish up */
729+ if (q->perturb_period) {
730+ q->perturb_timer.expires = jiffies + q->perturb_period;
731+ add_timer(&q->perturb_timer);
732+ } else {
733+ q->perturbation = 0;
734+ }
735+ sch_tree_unlock(sch);
736+ return 0;
737+}
738+
739+static int esfq_dump(struct Qdisc *sch, struct sk_buff *skb)
740+{
741+ struct esfq_sched_data *q = qdisc_priv(sch);
742+ unsigned char *b = skb_tail_pointer(skb);
743+ struct tc_esfq_qopt opt;
744+
745+ opt.quantum = q->quantum;
746+ opt.perturb_period = q->perturb_period/HZ;
747+
748+ opt.limit = q->limit;
749+ opt.divisor = q->hash_divisor;
750+ opt.flows = q->depth;
751+ opt.hash_kind = q->hash_kind;
752+
753+ if (nla_put(skb, TCA_OPTIONS, sizeof(opt), &opt))
754+ goto nla_put_failure;
755+
756+ return skb->len;
757+
758+nla_put_failure:
759+ nlmsg_trim(skb, b);
760+ return -1;
761+}
762+
763+static struct Qdisc_ops esfq_qdisc_ops =
764+{
765+ .next = NULL,
766+ .cl_ops = NULL,
767+ .id = "esfq",
768+ .priv_size = sizeof(struct esfq_sched_data),
769+ .enqueue = esfq_enqueue,
770+ .dequeue = esfq_dequeue,
771+ .peek = esfq_peek,
772+ .drop = esfq_drop,
773+ .init = esfq_init,
774+ .reset = esfq_reset,
775+ .destroy = esfq_destroy,
776+ .change = esfq_change,
777+ .dump = esfq_dump,
778+ .owner = THIS_MODULE,
779+};
780+
781+static int __init esfq_module_init(void)
782+{
783+ return register_qdisc(&esfq_qdisc_ops);
784+}
785+static void __exit esfq_module_exit(void)
786+{
787+ unregister_qdisc(&esfq_qdisc_ops);
788+}
789+module_init(esfq_module_init)
790+module_exit(esfq_module_exit)
791+MODULE_LICENSE("GPL");
792

Archive Download this file

Branches:
backfire
history
master
release_2010-11-17
release_2010-12-14
release_2011-02-23
release_2011-05-28
release_2011-08-27
release_2011-11-13
release_2012-03-18
release_2012-04-09
release_2012-10-11
uclibc_nptl



interactive