1
2
3
4
5
6
7
8
9
10
11
12#include <linux/module.h>
13#include <linux/types.h>
14#include <linux/kernel.h>
15#include <linux/errno.h>
16#include <linux/skbuff.h>
17#include <net/pkt_sched.h>
18
19
20
21struct fifo_sched_data
22{
23 u32 limit;
24};
25
26static int bfifo_enqueue(struct sk_buff *skb, struct Qdisc* sch)
27{
28 struct fifo_sched_data *q = qdisc_priv(sch);
29
30 if (likely(sch->qstats.backlog + skb->len <= q->limit))
31 return qdisc_enqueue_tail(skb, sch);
32
33 return qdisc_reshape_fail(skb, sch);
34}
35
36static int pfifo_enqueue(struct sk_buff *skb, struct Qdisc* sch)
37{
38 struct fifo_sched_data *q = qdisc_priv(sch);
39
40 if (likely(skb_queue_len(&sch->q) < q->limit))
41 return qdisc_enqueue_tail(skb, sch);
42
43 return qdisc_reshape_fail(skb, sch);
44}
45
46static int fifo_init(struct Qdisc *sch, struct rtattr *opt)
47{
48 struct fifo_sched_data *q = qdisc_priv(sch);
49
50 if (opt == NULL) {
51 u32 limit = sch->dev->tx_queue_len ? : 1;
52
53 if (sch->ops == &bfifo_qdisc_ops)
54 limit *= sch->dev->mtu;
55
56 q->limit = limit;
57 } else {
58 struct tc_fifo_qopt *ctl = RTA_DATA(opt);
59
60 if (RTA_PAYLOAD(opt) < sizeof(*ctl))
61 return -EINVAL;
62
63 q->limit = ctl->limit;
64 }
65
66 return 0;
67}
68
69static int fifo_dump(struct Qdisc *sch, struct sk_buff *skb)
70{
71 struct fifo_sched_data *q = qdisc_priv(sch);
72 struct tc_fifo_qopt opt = { .limit = q->limit };
73
74 RTA_PUT(skb, TCA_OPTIONS, sizeof(opt), &opt);
75 return skb->len;
76
77rtattr_failure:
78 return -1;
79}
80
81struct Qdisc_ops pfifo_qdisc_ops = {
82 .id = "pfifo",
83 .priv_size = sizeof(struct fifo_sched_data),
84 .enqueue = pfifo_enqueue,
85 .dequeue = qdisc_dequeue_head,
86 .requeue = qdisc_requeue,
87 .drop = qdisc_queue_drop,
88 .init = fifo_init,
89 .reset = qdisc_reset_queue,
90 .change = fifo_init,
91 .dump = fifo_dump,
92 .owner = THIS_MODULE,
93};
94
95struct Qdisc_ops bfifo_qdisc_ops = {
96 .id = "bfifo",
97 .priv_size = sizeof(struct fifo_sched_data),
98 .enqueue = bfifo_enqueue,
99 .dequeue = qdisc_dequeue_head,
100 .requeue = qdisc_requeue,
101 .drop = qdisc_queue_drop,
102 .init = fifo_init,
103 .reset = qdisc_reset_queue,
104 .change = fifo_init,
105 .dump = fifo_dump,
106 .owner = THIS_MODULE,
107};
108
109EXPORT_SYMBOL(bfifo_qdisc_ops);
110EXPORT_SYMBOL(pfifo_qdisc_ops);
111