1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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
40struct 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
47static int numifbs = 2;
48
49static void ri_tasklet(unsigned long dev);
50static netdev_tx_t ifb_xmit(struct sk_buff *skb, struct net_device *dev);
51static int ifb_open(struct net_device *dev);
52static int ifb_close(struct net_device *dev);
53
54static 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
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 {
115resched:
116 dp->tasklet_pending = 1;
117 tasklet_schedule(&dp->ifb_tasklet);
118 }
119
120}
121
122static 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
133static void ifb_setup(struct net_device *dev)
134{
135
136 dev->destructor = free_netdev;
137 dev->netdev_ops = &ifb_netdev_ops;
138
139
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
152static 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
180static 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
191static 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
203static 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
214static 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
222module_param(numifbs, int, 0);
223MODULE_PARM_DESC(numifbs, "Number of ifb devices");
224
225static 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
247err:
248 free_netdev(dev_ifb);
249 return err;
250}
251
252static 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
268static void __exit ifb_cleanup_module(void)
269{
270 rtnl_link_unregister(&ifb_link_ops);
271}
272
273module_init(ifb_init_module);
274module_exit(ifb_cleanup_module);
275MODULE_LICENSE("GPL");
276MODULE_AUTHOR("Jamal Hadi Salim");
277MODULE_ALIAS_RTNL_LINK("ifb");
278