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 { 24 u32 limit; 25 }; 26 27 static int bfifo_enqueue(struct sk_buff *skb, struct Qdisc* sch) 28 { 29 struct fifo_sched_data *q = qdisc_priv(sch); 30 31 if (likely(sch->qstats.backlog + qdisc_pkt_len(skb) <= q->limit)) 32 return qdisc_enqueue_tail(skb, sch); 33 34 return qdisc_reshape_fail(skb, sch); 35 } 36 37 static int pfifo_enqueue(struct sk_buff *skb, struct Qdisc* sch) 38 { 39 struct fifo_sched_data *q = qdisc_priv(sch); 40 41 if (likely(skb_queue_len(&sch->q) < q->limit)) 42 return qdisc_enqueue_tail(skb, sch); 43 44 return qdisc_reshape_fail(skb, sch); 45 } 46 47 static int pfifo_tail_enqueue(struct sk_buff *skb, struct Qdisc* sch) 48 { 49 struct sk_buff *skb_head; 50 struct fifo_sched_data *q = qdisc_priv(sch); 51 52 if (likely(skb_queue_len(&sch->q) < q->limit)) 53 return qdisc_enqueue_tail(skb, sch); 54 55 /* queue full, remove one skb to fulfill the limit */ 56 skb_head = qdisc_dequeue_head(sch); 57 sch->qstats.drops++; 58 kfree_skb(skb_head); 59 60 qdisc_enqueue_tail(skb, sch); 61 62 return NET_XMIT_CN; 63 } 64 65 static int fifo_init(struct Qdisc *sch, struct nlattr *opt) 66 { 67 struct fifo_sched_data *q = qdisc_priv(sch); 68 69 if (opt == NULL) { 70 u32 limit = qdisc_dev(sch)->tx_queue_len ? : 1; 71 72 if (sch->ops == &bfifo_qdisc_ops) 73 limit *= psched_mtu(qdisc_dev(sch)); 74 75 q->limit = limit; 76 } else { 77 struct tc_fifo_qopt *ctl = nla_data(opt); 78 79 if (nla_len(opt) < sizeof(*ctl)) 80 return -EINVAL; 81 82 q->limit = ctl->limit; 83 } 84 85 return 0; 86 } 87 88 static int fifo_dump(struct Qdisc *sch, struct sk_buff *skb) 89 { 90 struct fifo_sched_data *q = qdisc_priv(sch); 91 struct tc_fifo_qopt opt = { .limit = q->limit }; 92 93 NLA_PUT(skb, TCA_OPTIONS, sizeof(opt), &opt); 94 return skb->len; 95 96 nla_put_failure: 97 return -1; 98 } 99 100 struct Qdisc_ops pfifo_qdisc_ops __read_mostly = { 101 .id = "pfifo", 102 .priv_size = sizeof(struct fifo_sched_data), 103 .enqueue = pfifo_enqueue, 104 .dequeue = qdisc_dequeue_head, 105 .peek = qdisc_peek_head, 106 .drop = qdisc_queue_drop, 107 .init = fifo_init, 108 .reset = qdisc_reset_queue, 109 .change = fifo_init, 110 .dump = fifo_dump, 111 .owner = THIS_MODULE, 112 }; 113 EXPORT_SYMBOL(pfifo_qdisc_ops); 114 115 struct Qdisc_ops bfifo_qdisc_ops __read_mostly = { 116 .id = "bfifo", 117 .priv_size = sizeof(struct fifo_sched_data), 118 .enqueue = bfifo_enqueue, 119 .dequeue = qdisc_dequeue_head, 120 .peek = qdisc_peek_head, 121 .drop = qdisc_queue_drop, 122 .init = fifo_init, 123 .reset = qdisc_reset_queue, 124 .change = fifo_init, 125 .dump = fifo_dump, 126 .owner = THIS_MODULE, 127 }; 128 EXPORT_SYMBOL(bfifo_qdisc_ops); 129 130 struct Qdisc_ops pfifo_head_drop_qdisc_ops __read_mostly = { 131 .id = "pfifo_head_drop", 132 .priv_size = sizeof(struct fifo_sched_data), 133 .enqueue = pfifo_tail_enqueue, 134 .dequeue = qdisc_dequeue_head, 135 .peek = qdisc_peek_head, 136 .drop = qdisc_queue_drop_head, 137 .init = fifo_init, 138 .reset = qdisc_reset_queue, 139 .change = fifo_init, 140 .dump = fifo_dump, 141 .owner = THIS_MODULE, 142 }; 143 144 /* Pass size change message down to embedded FIFO */ 145 int fifo_set_limit(struct Qdisc *q, unsigned int limit) 146 { 147 struct nlattr *nla; 148 int ret = -ENOMEM; 149 150 /* Hack to avoid sending change message to non-FIFO */ 151 if (strncmp(q->ops->id + 1, "fifo", 4) != 0) 152 return 0; 153 154 nla = kmalloc(nla_attr_size(sizeof(struct tc_fifo_qopt)), GFP_KERNEL); 155 if (nla) { 156 nla->nla_type = RTM_NEWQDISC; 157 nla->nla_len = nla_attr_size(sizeof(struct tc_fifo_qopt)); 158 ((struct tc_fifo_qopt *)nla_data(nla))->limit = limit; 159 160 ret = q->ops->change(q, nla); 161 kfree(nla); 162 } 163 return ret; 164 } 165 EXPORT_SYMBOL(fifo_set_limit); 166 167 struct Qdisc *fifo_create_dflt(struct Qdisc *sch, struct Qdisc_ops *ops, 168 unsigned int limit) 169 { 170 struct Qdisc *q; 171 int err = -ENOMEM; 172 173 q = qdisc_create_dflt(sch->dev_queue, ops, TC_H_MAKE(sch->handle, 1)); 174 if (q) { 175 err = fifo_set_limit(q, limit); 176 if (err < 0) { 177 qdisc_destroy(q); 178 q = NULL; 179 } 180 } 181 182 return q ? : ERR_PTR(err); 183 } 184 EXPORT_SYMBOL(fifo_create_dflt); 185