1 /* drivers/net/ifb.c:
2
3 The purpose of this driver is to provide a device that allows
4 for sharing of resources:
5
6 1) qdiscs/policies that are per device as opposed to system wide.
7 ifb allows for a device which can be redirected to thus providing
8 an impression of sharing.
9
10 2) Allows for queueing incoming traffic for shaping instead of
11 dropping.
12
13 The original concept is based on what is known as the IMQ
14 driver initially written by Martin Devera, later rewritten
15 by Patrick McHardy and then maintained by Andre Correa.
16
17 You need the tc action mirror or redirect to feed this device
18 packets.
19
20 This program is free software; you can redistribute it and/or
21 modify it under the terms of the GNU General Public License
22 as published by the Free Software Foundation; either version
23 2 of the License, or (at your option) any later version.
24
25 Authors: Jamal Hadi Salim (2005)
26
27 */
28
29
30 #include <linux/module.h>
31 #include <linux/kernel.h>
32 #include <linux/netdevice.h>
33 #include <linux/etherdevice.h>
34 #include <linux/init.h>
35 #include <linux/moduleparam.h>
36 #include <net/pkt_sched.h>
37 #include <net/net_namespace.h>
38
39 #define TX_Q_LIMIT 32
40 struct ifb_private {
41 struct tasklet_struct ifb_tasklet;
42 int tasklet_pending;
43 struct sk_buff_head rq;
44 struct sk_buff_head tq;
45 };
46
47 static int numifbs = 2;
48
49 static void ri_tasklet(unsigned long dev);
50 static netdev_tx_t ifb_xmit(struct sk_buff *skb, struct net_device *dev);
51 static int ifb_open(struct net_device *dev);
52 static int ifb_close(struct net_device *dev);
53
ri_tasklet(unsigned long dev)54 static void ri_tasklet(unsigned long dev)
55 {
56
57 struct net_device *_dev = (struct net_device *)dev;
58 struct ifb_private *dp = netdev_priv(_dev);
59 struct net_device_stats *stats = &_dev->stats;
60 struct netdev_queue *txq;
61 struct sk_buff *skb;
62
63 txq = netdev_get_tx_queue(_dev, 0);
64 if ((skb = skb_peek(&dp->tq)) == NULL) {
65 if (__netif_tx_trylock(txq)) {
66 skb_queue_splice_tail_init(&dp->rq, &dp->tq);
67 __netif_tx_unlock(txq);
68 } else {
69 /* reschedule */
70 goto resched;
71 }
72 }
73
74 while ((skb = __skb_dequeue(&dp->tq)) != NULL) {
75 u32 from = G_TC_FROM(skb->tc_verd);
76
77 skb->tc_verd = 0;
78 skb->tc_verd = SET_TC_NCLS(skb->tc_verd);
79 stats->tx_packets++;
80 stats->tx_bytes +=skb->len;
81
82 rcu_read_lock();
83 skb->dev = dev_get_by_index_rcu(&init_net, skb->skb_iif);
84 if (!skb->dev) {
85 rcu_read_unlock();
86 dev_kfree_skb(skb);
87 stats->tx_dropped++;
88 if (skb_queue_len(&dp->tq) != 0)
89 goto resched;
90 break;
91 }
92 rcu_read_unlock();
93 skb->skb_iif = _dev->ifindex;
94
95 if (from & AT_EGRESS) {
96 dev_queue_xmit(skb);
97 } else if (from & AT_INGRESS) {
98 skb_pull(skb, skb->dev->hard_header_len);
99 netif_receive_skb(skb);
100 } else
101 BUG();
102 }
103
104 if (__netif_tx_trylock(txq)) {
105 if ((skb = skb_peek(&dp->rq)) == NULL) {
106 dp->tasklet_pending = 0;
107 if (netif_queue_stopped(_dev))
108 netif_wake_queue(_dev);
109 } else {
110 __netif_tx_unlock(txq);
111 goto resched;
112 }
113 __netif_tx_unlock(txq);
114 } else {
115 resched:
116 dp->tasklet_pending = 1;
117 tasklet_schedule(&dp->ifb_tasklet);
118 }
119
120 }
121
122 static const struct net_device_ops ifb_netdev_ops = {
123 .ndo_open = ifb_open,
124 .ndo_stop = ifb_close,
125 .ndo_start_xmit = ifb_xmit,
126 .ndo_validate_addr = eth_validate_addr,
127 };
128
129 #define IFB_FEATURES (NETIF_F_NO_CSUM | NETIF_F_SG | NETIF_F_FRAGLIST | \
130 NETIF_F_TSO_ECN | NETIF_F_TSO | NETIF_F_TSO6 | \
131 NETIF_F_HIGHDMA | NETIF_F_HW_VLAN_TX)
132
ifb_setup(struct net_device * dev)133 static void ifb_setup(struct net_device *dev)
134 {
135 /* Initialize the device structure. */
136 dev->destructor = free_netdev;
137 dev->netdev_ops = &ifb_netdev_ops;
138
139 /* Fill in device structure with ethernet-generic values. */
140 ether_setup(dev);
141 dev->tx_queue_len = TX_Q_LIMIT;
142
143 dev->features |= IFB_FEATURES;
144 dev->vlan_features |= IFB_FEATURES;
145
146 dev->flags |= IFF_NOARP;
147 dev->flags &= ~IFF_MULTICAST;
148 dev->priv_flags &= ~IFF_XMIT_DST_RELEASE;
149 random_ether_addr(dev->dev_addr);
150 }
151
ifb_xmit(struct sk_buff * skb,struct net_device * dev)152 static netdev_tx_t ifb_xmit(struct sk_buff *skb, struct net_device *dev)
153 {
154 struct ifb_private *dp = netdev_priv(dev);
155 struct net_device_stats *stats = &dev->stats;
156 u32 from = G_TC_FROM(skb->tc_verd);
157
158 stats->rx_packets++;
159 stats->rx_bytes+=skb->len;
160
161 if (!(from & (AT_INGRESS|AT_EGRESS)) || !skb->skb_iif) {
162 dev_kfree_skb(skb);
163 stats->rx_dropped++;
164 return NETDEV_TX_OK;
165 }
166
167 if (skb_queue_len(&dp->rq) >= dev->tx_queue_len) {
168 netif_stop_queue(dev);
169 }
170
171 __skb_queue_tail(&dp->rq, skb);
172 if (!dp->tasklet_pending) {
173 dp->tasklet_pending = 1;
174 tasklet_schedule(&dp->ifb_tasklet);
175 }
176
177 return NETDEV_TX_OK;
178 }
179
ifb_close(struct net_device * dev)180 static int ifb_close(struct net_device *dev)
181 {
182 struct ifb_private *dp = netdev_priv(dev);
183
184 tasklet_kill(&dp->ifb_tasklet);
185 netif_stop_queue(dev);
186 __skb_queue_purge(&dp->rq);
187 __skb_queue_purge(&dp->tq);
188 return 0;
189 }
190
ifb_open(struct net_device * dev)191 static int ifb_open(struct net_device *dev)
192 {
193 struct ifb_private *dp = netdev_priv(dev);
194
195 tasklet_init(&dp->ifb_tasklet, ri_tasklet, (unsigned long)dev);
196 __skb_queue_head_init(&dp->rq);
197 __skb_queue_head_init(&dp->tq);
198 netif_start_queue(dev);
199
200 return 0;
201 }
202
ifb_validate(struct nlattr * tb[],struct nlattr * data[])203 static int ifb_validate(struct nlattr *tb[], struct nlattr *data[])
204 {
205 if (tb[IFLA_ADDRESS]) {
206 if (nla_len(tb[IFLA_ADDRESS]) != ETH_ALEN)
207 return -EINVAL;
208 if (!is_valid_ether_addr(nla_data(tb[IFLA_ADDRESS])))
209 return -EADDRNOTAVAIL;
210 }
211 return 0;
212 }
213
214 static struct rtnl_link_ops ifb_link_ops __read_mostly = {
215 .kind = "ifb",
216 .priv_size = sizeof(struct ifb_private),
217 .setup = ifb_setup,
218 .validate = ifb_validate,
219 };
220
221 /* Number of ifb devices to be set up by this module. */
222 module_param(numifbs, int, 0);
223 MODULE_PARM_DESC(numifbs, "Number of ifb devices");
224
ifb_init_one(int index)225 static int __init ifb_init_one(int index)
226 {
227 struct net_device *dev_ifb;
228 int err;
229
230 dev_ifb = alloc_netdev(sizeof(struct ifb_private),
231 "ifb%d", ifb_setup);
232
233 if (!dev_ifb)
234 return -ENOMEM;
235
236 err = dev_alloc_name(dev_ifb, dev_ifb->name);
237 if (err < 0)
238 goto err;
239
240 dev_ifb->rtnl_link_ops = &ifb_link_ops;
241 err = register_netdevice(dev_ifb);
242 if (err < 0)
243 goto err;
244
245 return 0;
246
247 err:
248 free_netdev(dev_ifb);
249 return err;
250 }
251
ifb_init_module(void)252 static int __init ifb_init_module(void)
253 {
254 int i, err;
255
256 rtnl_lock();
257 err = __rtnl_link_register(&ifb_link_ops);
258
259 for (i = 0; i < numifbs && !err; i++)
260 err = ifb_init_one(i);
261 if (err)
262 __rtnl_link_unregister(&ifb_link_ops);
263 rtnl_unlock();
264
265 return err;
266 }
267
ifb_cleanup_module(void)268 static void __exit ifb_cleanup_module(void)
269 {
270 rtnl_link_unregister(&ifb_link_ops);
271 }
272
273 module_init(ifb_init_module);
274 module_exit(ifb_cleanup_module);
275 MODULE_LICENSE("GPL");
276 MODULE_AUTHOR("Jamal Hadi Salim");
277 MODULE_ALIAS_RTNL_LINK("ifb");
278