2 * net/sched/sch_prio.c Simple 3-band priority "scheduler".
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.
9 * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
10 * Fixes: 19990609: J Hadi Salim <hadi@nortelnetworks.com>:
11 * Init -- EINVAL when opt undefined
14 #include <linux/config.h>
15 #include <linux/module.h>
16 #include <asm/uaccess.h>
17 #include <asm/system.h>
18 #include <asm/bitops.h>
19 #include <linux/types.h>
20 #include <linux/kernel.h>
21 #include <linux/sched.h>
22 #include <linux/string.h>
24 #include <linux/socket.h>
25 #include <linux/sockios.h>
27 #include <linux/errno.h>
28 #include <linux/interrupt.h>
29 #include <linux/if_ether.h>
30 #include <linux/inet.h>
31 #include <linux/netdevice.h>
32 #include <linux/etherdevice.h>
33 #include <linux/notifier.h>
35 #include <net/route.h>
36 #include <linux/skbuff.h>
38 #include <net/pkt_sched.h>
41 struct prio_sched_data
44 struct tcf_proto
*filter_list
;
45 u8 prio2band
[TC_PRIO_MAX
+1];
46 struct Qdisc
*queues
[TCQ_PRIO_BANDS
];
50 static __inline__
unsigned prio_classify(struct sk_buff
*skb
, struct Qdisc
*sch
)
52 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
53 struct tcf_result res
;
57 if (TC_H_MAJ(skb
->priority
) != sch
->handle
) {
58 if (!q
->filter_list
|| tc_classify(skb
, q
->filter_list
, &res
)) {
61 return q
->prio2band
[band
&TC_PRIO_MAX
];
65 band
= TC_H_MIN(band
) - 1;
66 return band
< q
->bands
? band
: q
->prio2band
[0];
70 prio_enqueue(struct sk_buff
*skb
, struct Qdisc
* sch
)
72 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
76 qdisc
= q
->queues
[prio_classify(skb
, sch
)];
78 if ((ret
= qdisc
->enqueue(skb
, qdisc
)) == 0) {
79 sch
->stats
.bytes
+= skb
->len
;
90 prio_requeue(struct sk_buff
*skb
, struct Qdisc
* sch
)
92 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
96 qdisc
= q
->queues
[prio_classify(skb
, sch
)];
98 if ((ret
= qdisc
->ops
->requeue(skb
, qdisc
)) == 0) {
107 static struct sk_buff
*
108 prio_dequeue(struct Qdisc
* sch
)
111 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
115 for (prio
= 0; prio
< q
->bands
; prio
++) {
116 qdisc
= q
->queues
[prio
];
117 skb
= qdisc
->dequeue(qdisc
);
128 prio_drop(struct Qdisc
* sch
)
130 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
134 for (prio
= q
->bands
-1; prio
>= 0; prio
--) {
135 qdisc
= q
->queues
[prio
];
136 if (qdisc
->ops
->drop(qdisc
)) {
146 prio_reset(struct Qdisc
* sch
)
149 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
151 for (prio
=0; prio
<q
->bands
; prio
++)
152 qdisc_reset(q
->queues
[prio
]);
157 prio_destroy(struct Qdisc
* sch
)
160 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
162 for (prio
=0; prio
<q
->bands
; prio
++) {
163 qdisc_destroy(q
->queues
[prio
]);
164 q
->queues
[prio
] = &noop_qdisc
;
169 static int prio_tune(struct Qdisc
*sch
, struct rtattr
*opt
)
171 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
172 struct tc_prio_qopt
*qopt
= RTA_DATA(opt
);
175 if (opt
->rta_len
< RTA_LENGTH(sizeof(*qopt
)))
177 if (qopt
->bands
> TCQ_PRIO_BANDS
|| qopt
->bands
< 2)
180 for (i
=0; i
<=TC_PRIO_MAX
; i
++) {
181 if (qopt
->priomap
[i
] >= qopt
->bands
)
186 q
->bands
= qopt
->bands
;
187 memcpy(q
->prio2band
, qopt
->priomap
, TC_PRIO_MAX
+1);
189 for (i
=q
->bands
; i
<TCQ_PRIO_BANDS
; i
++) {
190 struct Qdisc
*child
= xchg(&q
->queues
[i
], &noop_qdisc
);
191 if (child
!= &noop_qdisc
)
192 qdisc_destroy(child
);
194 sch_tree_unlock(sch
);
196 for (i
=0; i
<=TC_PRIO_MAX
; i
++) {
197 int band
= q
->prio2band
[i
];
198 if (q
->queues
[band
] == &noop_qdisc
) {
200 child
= qdisc_create_dflt(sch
->dev
, &pfifo_qdisc_ops
);
203 child
= xchg(&q
->queues
[band
], child
);
205 if (child
!= &noop_qdisc
)
206 qdisc_destroy(child
);
207 sch_tree_unlock(sch
);
214 static int prio_init(struct Qdisc
*sch
, struct rtattr
*opt
)
216 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
219 for (i
=0; i
<TCQ_PRIO_BANDS
; i
++)
220 q
->queues
[i
] = &noop_qdisc
;
227 if ((err
= prio_tune(sch
, opt
)) != 0)
234 #ifdef CONFIG_RTNETLINK
235 static int prio_dump(struct Qdisc
*sch
, struct sk_buff
*skb
)
237 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
238 unsigned char *b
= skb
->tail
;
239 struct tc_prio_qopt opt
;
241 opt
.bands
= q
->bands
;
242 memcpy(&opt
.priomap
, q
->prio2band
, TC_PRIO_MAX
+1);
243 RTA_PUT(skb
, TCA_OPTIONS
, sizeof(opt
), &opt
);
247 skb_trim(skb
, b
- skb
->data
);
252 static int prio_graft(struct Qdisc
*sch
, unsigned long arg
, struct Qdisc
*new,
255 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
256 unsigned long band
= arg
- 1;
258 if (band
>= q
->bands
)
265 *old
= q
->queues
[band
];
266 q
->queues
[band
] = new;
268 sch_tree_unlock(sch
);
273 static struct Qdisc
*
274 prio_leaf(struct Qdisc
*sch
, unsigned long arg
)
276 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
277 unsigned long band
= arg
- 1;
279 if (band
>= q
->bands
)
282 return q
->queues
[band
];
285 static unsigned long prio_get(struct Qdisc
*sch
, u32 classid
)
287 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
288 unsigned long band
= TC_H_MIN(classid
);
290 if (band
- 1 >= q
->bands
)
295 static unsigned long prio_bind(struct Qdisc
*sch
, unsigned long parent
, u32 classid
)
297 return prio_get(sch
, classid
);
301 static void prio_put(struct Qdisc
*q
, unsigned long cl
)
306 static int prio_change(struct Qdisc
*sch
, u32 handle
, u32 parent
, struct rtattr
**tca
, unsigned long *arg
)
308 unsigned long cl
= *arg
;
309 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
311 if (cl
- 1 > q
->bands
)
316 static int prio_delete(struct Qdisc
*sch
, unsigned long cl
)
318 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
319 if (cl
- 1 > q
->bands
)
325 #ifdef CONFIG_RTNETLINK
326 static int prio_dump_class(struct Qdisc
*sch
, unsigned long cl
, struct sk_buff
*skb
,
329 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
331 if (cl
- 1 > q
->bands
)
334 tcm
->tcm_info
= q
->queues
[cl
-1]->handle
;
339 static void prio_walk(struct Qdisc
*sch
, struct qdisc_walker
*arg
)
341 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
347 for (prio
= 0; prio
< q
->bands
; prio
++) {
348 if (arg
->count
< arg
->skip
) {
352 if (arg
->fn(sch
, prio
+1, arg
) < 0) {
360 static struct tcf_proto
** prio_find_tcf(struct Qdisc
*sch
, unsigned long cl
)
362 struct prio_sched_data
*q
= (struct prio_sched_data
*)sch
->data
;
366 return &q
->filter_list
;
369 static struct Qdisc_class_ops prio_class_ops
=
384 #ifdef CONFIG_RTNETLINK
389 struct Qdisc_ops prio_qdisc_ops
=
394 sizeof(struct prio_sched_data
),
406 #ifdef CONFIG_RTNETLINK
413 int init_module(void)
415 return register_qdisc(&prio_qdisc_ops
);
418 void cleanup_module(void)
420 unregister_qdisc(&prio_qdisc_ops
);