net/sched/sch_fifo.c

   1 /*
   2  * net/sched/sch_fifo.c The simplest FIFO queue.
   3  *
   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.
   8  *
   9  * Authors:     Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
  10  */
  11
  12 #include <linux/module.h>
  13 #include <linux/slab.h>
  14 #include <linux/types.h>
  15 #include <linux/kernel.h>
  16 #include <linux/errno.h>
  17 #include <linux/skbuff.h>
  18 #include <net/pkt_sched.h>
  19
  20 /* 1 band FIFO pseudo-"scheduler" */
  21
  22 struct fifo_sched_data {
  23         u32 limit;
  24 };
  25
  26 static 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 + qdisc_pkt_len(skb) <= q->limit))
  31                 return qdisc_enqueue_tail(skb, sch);
  32
  33         return qdisc_reshape_fail(skb, sch);
  34 }
  35
  36 static 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
  46 static int pfifo_tail_enqueue(struct sk_buff *skb, struct Qdisc *sch)
  47 {
  48         struct fifo_sched_data *q = qdisc_priv(sch);
  49
  50         if (likely(skb_queue_len(&sch->q) < q->limit))
  51                 return qdisc_enqueue_tail(skb, sch);
  52
  53         /* queue full, remove one skb to fulfill the limit */
  54         __qdisc_queue_drop_head(sch, &sch->q);
  55         sch->qstats.drops++;
  56         qdisc_enqueue_tail(skb, sch);
  57
  58         return NET_XMIT_CN;
  59 }
  60
  61 static int fifo_init(struct Qdisc *sch, struct nlattr *opt)
  62 {
  63         struct fifo_sched_data *q = qdisc_priv(sch);
  64         bool bypass;
  65         bool is_bfifo = sch->ops == &bfifo_qdisc_ops;
  66
  67         if (opt == NULL) {
  68                 u32 limit = qdisc_dev(sch)->tx_queue_len ? : 1;
  69
  70                 if (is_bfifo)
  71                         limit *= psched_mtu(qdisc_dev(sch));
  72
  73                 q->limit = limit;
  74         } else {
  75                 struct tc_fifo_qopt *ctl = nla_data(opt);
  76
  77                 if (nla_len(opt) < sizeof(*ctl))
  78                         return -EINVAL;
  79
  80                 q->limit = ctl->limit;
  81         }
  82
  83         if (is_bfifo)
  84                 bypass = q->limit >= psched_mtu(qdisc_dev(sch));
  85         else
  86                 bypass = q->limit >= 1;
  87
  88         if (bypass)
  89                 sch->flags |= TCQ_F_CAN_BYPASS;
  90         else
  91                 sch->flags &= ~TCQ_F_CAN_BYPASS;
  92         return 0;
  93 }
  94
  95 static int fifo_dump(struct Qdisc *sch, struct sk_buff *skb)
  96 {
  97         struct fifo_sched_data *q = qdisc_priv(sch);
  98         struct tc_fifo_qopt opt = { .limit = q->limit };
  99
 100         NLA_PUT(skb, TCA_OPTIONS, sizeof(opt), &opt);
 101         return skb->len;
 102
 103 nla_put_failure:
 104         return -1;
 105 }
 106
 107 struct Qdisc_ops pfifo_qdisc_ops __read_mostly = {
 108         .id             =       "pfifo",
 109         .priv_size      =       sizeof(struct fifo_sched_data),
 110         .enqueue        =       pfifo_enqueue,
 111         .dequeue        =       qdisc_dequeue_head,
 112         .peek           =       qdisc_peek_head,
 113         .drop           =       qdisc_queue_drop,
 114         .init           =       fifo_init,
 115         .reset          =       qdisc_reset_queue,
 116         .change         =       fifo_init,
 117         .dump           =       fifo_dump,
 118         .owner          =       THIS_MODULE,
 119 };
 120 EXPORT_SYMBOL(pfifo_qdisc_ops);
 121
 122 struct Qdisc_ops bfifo_qdisc_ops __read_mostly = {
 123         .id             =       "bfifo",
 124         .priv_size      =       sizeof(struct fifo_sched_data),
 125         .enqueue        =       bfifo_enqueue,
 126         .dequeue        =       qdisc_dequeue_head,
 127         .peek           =       qdisc_peek_head,
 128         .drop           =       qdisc_queue_drop,
 129         .init           =       fifo_init,
 130         .reset          =       qdisc_reset_queue,
 131         .change         =       fifo_init,
 132         .dump           =       fifo_dump,
 133         .owner          =       THIS_MODULE,
 134 };
 135 EXPORT_SYMBOL(bfifo_qdisc_ops);
 136
 137 struct Qdisc_ops pfifo_head_drop_qdisc_ops __read_mostly = {
 138         .id             =       "pfifo_head_drop",
 139         .priv_size      =       sizeof(struct fifo_sched_data),
 140         .enqueue        =       pfifo_tail_enqueue,
 141         .dequeue        =       qdisc_dequeue_head,
 142         .peek           =       qdisc_peek_head,
 143         .drop           =       qdisc_queue_drop_head,
 144         .init           =       fifo_init,
 145         .reset          =       qdisc_reset_queue,
 146         .change         =       fifo_init,
 147         .dump           =       fifo_dump,
 148         .owner          =       THIS_MODULE,
 149 };
 150
 151 /* Pass size change message down to embedded FIFO */
 152 int fifo_set_limit(struct Qdisc *q, unsigned int limit)
 153 {
 154         struct nlattr *nla;
 155         int ret = -ENOMEM;
 156
 157         /* Hack to avoid sending change message to non-FIFO */
 158         if (strncmp(q->ops->id + 1, "fifo", 4) != 0)
 159                 return 0;
 160
 161         nla = kmalloc(nla_attr_size(sizeof(struct tc_fifo_qopt)), GFP_KERNEL);
 162         if (nla) {
 163                 nla->nla_type = RTM_NEWQDISC;
 164                 nla->nla_len = nla_attr_size(sizeof(struct tc_fifo_qopt));
 165                 ((struct tc_fifo_qopt *)nla_data(nla))->limit = limit;
 166
 167                 ret = q->ops->change(q, nla);
 168                 kfree(nla);
 169         }
 170         return ret;
 171 }
 172 EXPORT_SYMBOL(fifo_set_limit);
 173
 174 struct Qdisc *fifo_create_dflt(struct Qdisc *sch, struct Qdisc_ops *ops,
 175                                unsigned int limit)
 176 {
 177         struct Qdisc *q;
 178         int err = -ENOMEM;
 179
 180         q = qdisc_create_dflt(sch->dev_queue, ops, TC_H_MAKE(sch->handle, 1));
 181         if (q) {
 182                 err = fifo_set_limit(q, limit);
 183                 if (err < 0) {
 184                         qdisc_destroy(q);
 185                         q = NULL;
 186                 }
 187         }
 188
 189         return q ? : ERR_PTR(err);
 190 }
 191 EXPORT_SYMBOL(fifo_create_dflt);