Don't ever send frames on OpenFlow devices to the host stack.
[openvswitch] / datapath / datapath.c
1 /*
2  * Distributed under the terms of the GNU GPL version 2.
3  * Copyright (c) 2007, 2008 The Board of Trustees of The Leland 
4  * Stanford Junior University
5  */
6
7 /* Functions for managing the dp interface/device. */
8
9 #include <linux/init.h>
10 #include <linux/module.h>
11 #include <linux/if_arp.h>
12 #include <linux/if_bridge.h>
13 #include <linux/if_vlan.h>
14 #include <linux/in.h>
15 #include <net/genetlink.h>
16 #include <linux/ip.h>
17 #include <linux/delay.h>
18 #include <linux/etherdevice.h>
19 #include <linux/kernel.h>
20 #include <linux/kthread.h>
21 #include <linux/mutex.h>
22 #include <linux/rtnetlink.h>
23 #include <linux/rcupdate.h>
24 #include <linux/version.h>
25 #include <linux/ethtool.h>
26 #include <linux/random.h>
27 #include <asm/system.h>
28 #include <linux/netfilter_bridge.h>
29 #include <linux/inetdevice.h>
30 #include <linux/list.h>
31
32 #include "openflow-netlink.h"
33 #include "datapath.h"
34 #include "table.h"
35 #include "chain.h"
36 #include "forward.h"
37 #include "flow.h"
38 #include "datapath_t.h"
39
40 #include "compat.h"
41
42
43 /* Number of milliseconds between runs of the maintenance thread. */
44 #define MAINT_SLEEP_MSECS 1000
45
46 #define BRIDGE_PORT_NO_FLOOD    0x00000001 
47
48 #define UINT32_MAX                        4294967295U
49 #define UINT16_MAX                        65535
50 #define MAX(X, Y) ((X) > (Y) ? (X) : (Y))
51
52 struct net_bridge_port {
53         u16     port_no;
54         u32 flags;
55         struct datapath *dp;
56         struct net_device *dev;
57         struct list_head node; /* Element in datapath.ports. */
58 };
59
60 static struct genl_family dp_genl_family;
61 static struct genl_multicast_group mc_group;
62
63 int dp_dev_setup(struct net_device *dev);  
64
65 /* It's hard to imagine wanting more than one datapath, but... */
66 #define DP_MAX 32
67
68 /* datapaths.  Protected on the read side by rcu_read_lock, on the write side
69  * by dp_mutex.
70  *
71  * It is safe to access the datapath and net_bridge_port structures with just
72  * the dp_mutex, but to access the chain you need to take the rcu_read_lock
73  * also (because dp_mutex doesn't prevent flows from being destroyed).
74  */
75 static struct datapath *dps[DP_MAX];
76 static DEFINE_MUTEX(dp_mutex);
77
78 static int dp_maint_func(void *data);
79 static int send_port_status(struct net_bridge_port *p, uint8_t status);
80 static int dp_genl_openflow_done(struct netlink_callback *);
81
82 /* nla_shrink - reduce amount of space reserved by nla_reserve
83  * @skb: socket buffer from which to recover room
84  * @nla: netlink attribute to adjust
85  * @len: new length of attribute payload
86  *
87  * Reduces amount of space reserved by a call to nla_reserve.
88  *
89  * No other attributes may be added between calling nla_reserve and this
90  * function, since it will create a hole in the message.
91  */
92 void nla_shrink(struct sk_buff *skb, struct nlattr *nla, int len)
93 {
94         int delta = nla_total_size(len) - nla_total_size(nla_len(nla));
95         BUG_ON(delta > 0);
96         skb->tail += delta;
97         skb->len  += delta;
98         nla->nla_len = nla_attr_size(len);
99 }
100
101 /* Puts a set of openflow headers for a message of the given 'type' into 'skb'.
102  * If 'sender' is nonnull, then it is used as the message's destination.  'dp'
103  * must specify the datapath to use.
104  *
105  * '*max_openflow_len' receives the maximum number of bytes that are available
106  * for the embedded OpenFlow message.  The caller must call
107  * resize_openflow_skb() to set the actual size of the message to this number
108  * of bytes or less.
109  *
110  * Returns the openflow header if successful, otherwise (if 'skb' is too small)
111  * an error code. */
112 static void *
113 put_openflow_headers(struct datapath *dp, struct sk_buff *skb, uint8_t type,
114                      const struct sender *sender, int *max_openflow_len)
115 {
116         struct ofp_header *oh;
117         struct nlattr *attr;
118         int openflow_len;
119
120         /* Assemble the Generic Netlink wrapper. */
121         if (!genlmsg_put(skb,
122                          sender ? sender->pid : 0,
123                          sender ? sender->seq : 0,
124                          &dp_genl_family, 0, DP_GENL_C_OPENFLOW))
125                 return ERR_PTR(-ENOBUFS);
126         if (nla_put_u32(skb, DP_GENL_A_DP_IDX, dp->dp_idx) < 0)
127                 return ERR_PTR(-ENOBUFS);
128         openflow_len = (skb_tailroom(skb) - NLA_HDRLEN) & ~(NLA_ALIGNTO - 1);
129         if (openflow_len < sizeof *oh)
130                 return ERR_PTR(-ENOBUFS);
131         *max_openflow_len = openflow_len;
132         attr = nla_reserve(skb, DP_GENL_A_OPENFLOW, openflow_len);
133         BUG_ON(!attr);
134
135         /* Fill in the header.  The caller is responsible for the length. */
136         oh = nla_data(attr);
137         oh->version = OFP_VERSION;
138         oh->type = type;
139         oh->xid = sender ? sender->xid : 0;
140
141         return oh;
142 }
143
144 /* Resizes OpenFlow header 'oh', which must be at the tail end of 'skb', to new
145  * length 'new_length' (in bytes), adjusting pointers and size values as
146  * necessary. */
147 static void
148 resize_openflow_skb(struct sk_buff *skb,
149                     struct ofp_header *oh, size_t new_length)
150 {
151         struct nlattr *attr = ((void *) oh) - NLA_HDRLEN;
152         nla_shrink(skb, attr, new_length);
153         oh->length = htons(new_length);
154         nlmsg_end(skb, (struct nlmsghdr *) skb->data);
155 }
156
157 /* Allocates a new skb to contain an OpenFlow message 'openflow_len' bytes in
158  * length.  Returns a null pointer if memory is unavailable, otherwise returns
159  * the OpenFlow header and stores a pointer to the skb in '*pskb'. 
160  *
161  * 'type' is the OpenFlow message type.  If 'sender' is nonnull, then it is
162  * used as the message's destination.  'dp' must specify the datapath to
163  * use.  */
164 static void *
165 alloc_openflow_skb(struct datapath *dp, size_t openflow_len, uint8_t type,
166                    const struct sender *sender, struct sk_buff **pskb) 
167 {
168         struct ofp_header *oh;
169         size_t genl_len;
170         struct sk_buff *skb;
171         int max_openflow_len;
172
173         if ((openflow_len + sizeof(struct ofp_header)) > UINT16_MAX) {
174                 if (net_ratelimit())
175                         printk("alloc_openflow_skb: openflow message too large: %zu\n", 
176                                         openflow_len);
177                 return NULL;
178         }
179
180         genl_len = nlmsg_total_size(GENL_HDRLEN + dp_genl_family.hdrsize);
181         genl_len += nla_total_size(sizeof(uint32_t)); /* DP_GENL_A_DP_IDX */
182         genl_len += nla_total_size(openflow_len);    /* DP_GENL_A_OPENFLOW */
183         skb = *pskb = genlmsg_new(genl_len, GFP_ATOMIC);
184         if (!skb) {
185                 if (net_ratelimit())
186                         printk("alloc_openflow_skb: genlmsg_new failed\n");
187                 return NULL;
188         }
189
190         oh = put_openflow_headers(dp, skb, type, sender, &max_openflow_len);
191         BUG_ON(!oh || IS_ERR(oh));
192         resize_openflow_skb(skb, oh, openflow_len);
193
194         return oh;
195 }
196
197 /* Sends 'skb' to 'sender' if it is nonnull, otherwise multicasts 'skb' to all
198  * listeners. */
199 static int
200 send_openflow_skb(struct sk_buff *skb, const struct sender *sender) 
201 {
202         int err = (sender
203                    ? genlmsg_unicast(skb, sender->pid)
204                    : genlmsg_multicast(skb, 0, mc_group.id, GFP_ATOMIC));
205         if (err && net_ratelimit())
206                 printk(KERN_WARNING "send_openflow_skb: send failed: %d\n",
207                        err);
208         return err;
209 }
210
211 /* Generates a unique datapath id.  It incorporates the datapath index
212  * and a hardware address, if available.  If not, it generates a random
213  * one.
214  */
215 static 
216 uint64_t gen_datapath_id(uint16_t dp_idx)
217 {
218         uint64_t id;
219         int i;
220         struct net_device *dev;
221
222         /* The top 16 bits are used to identify the datapath.  The lower 48 bits
223          * use an interface address.  */
224         id = (uint64_t)dp_idx << 48;
225         if ((dev = dev_get_by_name(&init_net, "ctl0")) 
226                         || (dev = dev_get_by_name(&init_net, "eth0"))) {
227                 for (i=0; i<ETH_ALEN; i++) {
228                         id |= (uint64_t)dev->dev_addr[i] << (8*(ETH_ALEN-1 - i));
229                 }
230                 dev_put(dev);
231         } else {
232                 /* Randomly choose the lower 48 bits if we cannot find an
233                  * address and mark the most significant bit to indicate that
234                  * this was randomly generated. */
235                 uint8_t rand[ETH_ALEN];
236                 get_random_bytes(rand, ETH_ALEN);
237                 id |= (uint64_t)1 << 63;
238                 for (i=0; i<ETH_ALEN; i++) {
239                         id |= (uint64_t)rand[i] << (8*(ETH_ALEN-1 - i));
240                 }
241         }
242
243         return id;
244 }
245
246 /* Creates a new datapath numbered 'dp_idx'.  Returns 0 for success or a
247  * negative error code.
248  *
249  * Not called with any locks. */
250 static int new_dp(int dp_idx)
251 {
252         struct datapath *dp;
253         int err;
254
255         if (dp_idx < 0 || dp_idx >= DP_MAX)
256                 return -EINVAL;
257
258         if (!try_module_get(THIS_MODULE))
259                 return -ENODEV;
260
261         mutex_lock(&dp_mutex);
262         dp = rcu_dereference(dps[dp_idx]);
263         if (dp != NULL) {
264                 err = -EEXIST;
265                 goto err_unlock;
266         }
267
268         err = -ENOMEM;
269         dp = kzalloc(sizeof *dp, GFP_KERNEL);
270         if (dp == NULL)
271                 goto err_unlock;
272
273         dp->dp_idx = dp_idx;
274         dp->id = gen_datapath_id(dp_idx);
275         dp->chain = chain_create(dp);
276         if (dp->chain == NULL)
277                 goto err_free_dp;
278         INIT_LIST_HEAD(&dp->port_list);
279
280 #if 0
281         /* Setup our "of" device */
282         dp->dev.priv = dp;
283         rtnl_lock();
284         err = dp_dev_setup(&dp->dev);
285         rtnl_unlock();
286         if (err != 0) 
287                 printk("datapath: problem setting up 'of' device\n");
288 #endif
289
290         dp->flags = 0;
291         dp->miss_send_len = OFP_DEFAULT_MISS_SEND_LEN;
292
293         dp->dp_task = kthread_run(dp_maint_func, dp, "dp%d", dp_idx);
294         if (IS_ERR(dp->dp_task))
295                 goto err_free_dp;
296
297         rcu_assign_pointer(dps[dp_idx], dp);
298         mutex_unlock(&dp_mutex);
299
300         return 0;
301
302 err_free_dp:
303         kfree(dp);
304 err_unlock:
305         mutex_unlock(&dp_mutex);
306         module_put(THIS_MODULE);
307                 return err;
308 }
309
310 /* Find and return a free port number under 'dp'.  Called under dp_mutex. */
311 static int find_portno(struct datapath *dp)
312 {
313         int i;
314         for (i = 0; i < OFPP_MAX; i++)
315                 if (dp->ports[i] == NULL)
316                         return i;
317         return -EXFULL;
318 }
319
320 static struct net_bridge_port *new_nbp(struct datapath *dp,
321                                                                            struct net_device *dev)
322 {
323         struct net_bridge_port *p;
324         int port_no;
325
326         port_no = find_portno(dp);
327         if (port_no < 0)
328                 return ERR_PTR(port_no);
329
330         p = kzalloc(sizeof(*p), GFP_KERNEL);
331         if (p == NULL)
332                 return ERR_PTR(-ENOMEM);
333
334         p->dp = dp;
335         dev_hold(dev);
336         p->dev = dev;
337         p->port_no = port_no;
338
339         return p;
340 }
341
342 /* Called with dp_mutex. */
343 int add_switch_port(struct datapath *dp, struct net_device *dev)
344 {
345         struct net_bridge_port *p;
346
347         if (dev->flags & IFF_LOOPBACK || dev->type != ARPHRD_ETHER)
348                 return -EINVAL;
349
350         if (dev->br_port != NULL)
351                 return -EBUSY;
352
353         p = new_nbp(dp, dev);
354         if (IS_ERR(p))
355                 return PTR_ERR(p);
356
357         dev_hold(dev);
358         rcu_assign_pointer(dev->br_port, p);
359         rtnl_lock();
360         dev_set_promiscuity(dev, 1);
361         rtnl_unlock();
362
363         rcu_assign_pointer(dp->ports[p->port_no], p);
364         list_add_rcu(&p->node, &dp->port_list);
365
366         /* Notify the ctlpath that this port has been added */
367         send_port_status(p, OFPPR_ADD);
368
369         return 0;
370 }
371
372 /* Delete 'p' from switch.
373  * Called with dp_mutex. */
374 static int del_switch_port(struct net_bridge_port *p)
375 {
376         /* First drop references to device. */
377         rtnl_lock();
378         dev_set_promiscuity(p->dev, -1);
379         rtnl_unlock();
380         list_del_rcu(&p->node);
381         rcu_assign_pointer(p->dp->ports[p->port_no], NULL);
382         rcu_assign_pointer(p->dev->br_port, NULL);
383
384         /* Then wait until no one is still using it, and destroy it. */
385         synchronize_rcu();
386
387         /* Notify the ctlpath that this port no longer exists */
388         send_port_status(p, OFPPR_DELETE);
389
390         dev_put(p->dev);
391         kfree(p);
392
393         return 0;
394 }
395
396 /* Called with dp_mutex. */
397 static void del_dp(struct datapath *dp)
398 {
399         struct net_bridge_port *p, *n;
400
401 #if 0
402         /* Unregister the "of" device of this dp */
403         rtnl_lock();
404         unregister_netdevice(&dp->dev);
405         rtnl_unlock();
406 #endif
407
408         kthread_stop(dp->dp_task);
409
410         /* Drop references to DP. */
411         list_for_each_entry_safe (p, n, &dp->port_list, node)
412                 del_switch_port(p);
413         rcu_assign_pointer(dps[dp->dp_idx], NULL);
414
415         /* Wait until no longer in use, then destroy it. */
416         synchronize_rcu();
417         chain_destroy(dp->chain);
418         kfree(dp);
419         module_put(THIS_MODULE);
420 }
421
422 static int dp_maint_func(void *data)
423 {
424         struct datapath *dp = (struct datapath *) data;
425
426         while (!kthread_should_stop()) {
427 #if 1
428                 chain_timeout(dp->chain);
429 #else
430                 int count = chain_timeout(dp->chain);
431                 chain_print_stats(dp->chain);
432                 if (count)
433                         printk("%d flows timed out\n", count);
434 #endif
435                 msleep_interruptible(MAINT_SLEEP_MSECS);
436         }
437                 
438         return 0;
439 }
440
441 static void
442 do_port_input(struct net_bridge_port *p, struct sk_buff *skb) 
443 {
444         /* Push the Ethernet header back on. */
445         if (skb->protocol == htons(ETH_P_8021Q))
446                 skb_push(skb, VLAN_ETH_HLEN);
447         else
448                 skb_push(skb, ETH_HLEN);
449         fwd_port_input(p->dp->chain, skb, p->port_no);
450 }
451
452 /*
453  * Used as br_handle_frame_hook.  (Cannot run bridge at the same time, even on
454  * different set of devices!)
455  */
456 #if LINUX_VERSION_CODE >= KERNEL_VERSION(2,6,22)
457 /* Called with rcu_read_lock. */
458 static struct sk_buff *dp_frame_hook(struct net_bridge_port *p,
459                                          struct sk_buff *skb)
460 {
461         do_port_input(p, skb);
462         return NULL;
463 }
464 #elif LINUX_VERSION_CODE >= KERNEL_VERSION(2,6,0)
465 static int dp_frame_hook(struct net_bridge_port *p, struct sk_buff **pskb)
466 {
467         do_port_input(p, *pskb);
468         return 1;
469 }
470 #else
471 /* NB: This has only been tested on 2.4.35 */
472
473 /* Called without any locks (?) */
474 static void dp_frame_hook(struct sk_buff *skb)
475 {
476         struct net_bridge_port *p = skb->dev->br_port;
477         if (p) {
478                 rcu_read_lock();
479                 do_port_input(p, skb);
480                 rcu_read_unlock();
481         } else
482                 kfree_skb(skb);
483 }
484 #endif
485
486 /* Forwarding output path.
487  * Based on net/bridge/br_forward.c. */
488
489 /* Don't forward packets to originating port.  If we're flooding,
490  * then don't send out ports with flooding disabled.
491  */
492 static inline int should_deliver(const struct net_bridge_port *p,
493                         const struct sk_buff *skb, int flood)
494 {
495         if (skb->dev == p->dev)
496                 return 0;
497
498         if (flood && (p->flags & BRIDGE_PORT_NO_FLOOD))
499                 return 0;
500
501         return 1;
502 }
503
504 static inline unsigned packet_length(const struct sk_buff *skb)
505 {
506         int length = skb->len - ETH_HLEN;
507         if (skb->protocol == htons(ETH_P_8021Q))
508                 length -= VLAN_HLEN;
509         return length;
510 }
511
512 /* Send packets out all the ports except the originating one.  If the
513  * "flood" argument is set, only send along the minimum spanning tree.
514  */
515 static int
516 output_all(struct datapath *dp, struct sk_buff *skb, int flood)
517 {
518         struct net_bridge_port *p;
519         int prev_port;
520
521         prev_port = -1;
522         list_for_each_entry_rcu (p, &dp->port_list, node) {
523                 if (!should_deliver(p, skb, flood))
524                         continue;
525                 if (prev_port != -1) {
526                         struct sk_buff *clone = skb_clone(skb, GFP_ATOMIC);
527                         if (!clone) {
528                                 kfree_skb(skb);
529                                 return -ENOMEM;
530                         }
531                         dp_output_port(dp, clone, prev_port); 
532                 }
533                 prev_port = p->port_no;
534         }
535         if (prev_port != -1)
536                 dp_output_port(dp, skb, prev_port);
537         else
538                 kfree_skb(skb);
539
540         return 0;
541 }
542
543 /* Marks 'skb' as having originated from 'in_port' in 'dp'.
544    FIXME: how are devices reference counted? */
545 int dp_set_origin(struct datapath *dp, uint16_t in_port,
546                            struct sk_buff *skb)
547 {
548         if (in_port < OFPP_MAX && dp->ports[in_port]) {
549                 skb->dev = dp->ports[in_port]->dev;
550                 return 0;
551         }
552         return -ENOENT;
553 }
554
555 /* Takes ownership of 'skb' and transmits it to 'out_port' on 'dp'.
556  */
557 int dp_output_port(struct datapath *dp, struct sk_buff *skb, int out_port)
558 {
559         struct net_bridge_port *p;
560         int len = skb->len;
561
562         BUG_ON(!skb);
563         if (out_port == OFPP_FLOOD)
564                 return output_all(dp, skb, 1);
565         else if (out_port == OFPP_ALL)
566                 return output_all(dp, skb, 0);
567         else if (out_port == OFPP_CONTROLLER)
568                 return dp_output_control(dp, skb, fwd_save_skb(skb), 0,
569                                                   OFPR_ACTION);
570         else if (out_port == OFPP_TABLE) {
571                 struct sw_flow_key key;
572                 struct sw_flow *flow;
573
574                 flow_extract(skb, skb->dev->br_port->port_no, &key);
575                 flow = chain_lookup(dp->chain, &key);
576                 if (likely(flow != NULL)) {
577                         flow_used(flow, skb);
578                         execute_actions(dp, skb, &key, flow->actions, flow->n_actions);
579                         return 0;
580                 }
581                 return -ESRCH;
582         } else if (out_port >= OFPP_MAX)
583                 goto bad_port;
584
585         p = dp->ports[out_port];
586         if (p == NULL)
587                 goto bad_port;
588
589         skb->dev = p->dev;
590         if (packet_length(skb) > skb->dev->mtu) {
591                 printk("dropped over-mtu packet: %d > %d\n",
592                                         packet_length(skb), skb->dev->mtu);
593                 kfree_skb(skb);
594                 return -E2BIG;
595         }
596
597         dev_queue_xmit(skb);
598
599         return len;
600
601 bad_port:
602         kfree_skb(skb);
603         if (net_ratelimit())
604                 printk("can't forward to bad port %d\n", out_port);
605         return -ENOENT;
606 }
607
608 /* Takes ownership of 'skb' and transmits it to 'dp''s control path.  If
609  * 'buffer_id' != -1, then only the first 64 bytes of 'skb' are sent;
610  * otherwise, all of 'skb' is sent.  'reason' indicates why 'skb' is being
611  * sent. 'max_len' sets the maximum number of bytes that the caller
612  * wants to be sent; a value of 0 indicates the entire packet should be
613  * sent. */
614 int
615 dp_output_control(struct datapath *dp, struct sk_buff *skb,
616                            uint32_t buffer_id, size_t max_len, int reason)
617 {
618         /* FIXME?  Can we avoid creating a new skbuff in the case where we
619          * forward the whole packet? */
620         struct sk_buff *f_skb;
621         struct ofp_packet_in *opi;
622         size_t fwd_len, opi_len;
623         int err;
624
625         fwd_len = skb->len;
626         if ((buffer_id != (uint32_t) -1) && max_len)
627                 fwd_len = min(fwd_len, max_len);
628
629         opi_len = offsetof(struct ofp_packet_in, data) + fwd_len;
630         opi = alloc_openflow_skb(dp, opi_len, OFPT_PACKET_IN, NULL, &f_skb);
631         if (!opi) {
632                 err = -ENOMEM;
633                 goto out;
634         }
635         opi->buffer_id      = htonl(buffer_id);
636         opi->total_len      = htons(skb->len);
637         opi->in_port        = htons(skb->dev->br_port->port_no);
638         opi->reason         = reason;
639         opi->pad            = 0;
640         memcpy(opi->data, skb_mac_header(skb), fwd_len);
641         err = send_openflow_skb(f_skb, NULL);
642
643 out:
644         kfree_skb(skb);
645         return err;
646 }
647
648 static void fill_port_desc(struct net_bridge_port *p, struct ofp_phy_port *desc)
649 {
650         desc->port_no = htons(p->port_no);
651         strncpy(desc->name, p->dev->name, OFP_MAX_PORT_NAME_LEN);
652         desc->name[OFP_MAX_PORT_NAME_LEN-1] = '\0';
653         memcpy(desc->hw_addr, p->dev->dev_addr, ETH_ALEN);
654         desc->flags = htonl(p->flags);
655         desc->features = 0;
656         desc->speed = 0;
657
658 #if LINUX_VERSION_CODE >= KERNEL_VERSION(2,4,24)
659         if (p->dev->ethtool_ops && p->dev->ethtool_ops->get_settings) {
660                 struct ethtool_cmd ecmd = { .cmd = ETHTOOL_GSET };
661
662                 if (!p->dev->ethtool_ops->get_settings(p->dev, &ecmd)) {
663                         if (ecmd.supported & SUPPORTED_10baseT_Half) 
664                                 desc->features |= OFPPF_10MB_HD;
665                         if (ecmd.supported & SUPPORTED_10baseT_Full)
666                                 desc->features |= OFPPF_10MB_FD;
667                         if (ecmd.supported & SUPPORTED_100baseT_Half) 
668                                 desc->features |= OFPPF_100MB_HD;
669                         if (ecmd.supported & SUPPORTED_100baseT_Full)
670                                 desc->features |= OFPPF_100MB_FD;
671                         if (ecmd.supported & SUPPORTED_1000baseT_Half)
672                                 desc->features |= OFPPF_1GB_HD;
673                         if (ecmd.supported & SUPPORTED_1000baseT_Full)
674                                 desc->features |= OFPPF_1GB_FD;
675                         /* 10Gbps half-duplex doesn't exist... */
676                         if (ecmd.supported & SUPPORTED_10000baseT_Full)
677                                 desc->features |= OFPPF_10GB_FD;
678
679                         desc->features = htonl(desc->features);
680                         desc->speed = htonl(ecmd.speed);
681                 }
682         }
683 #endif
684 }
685
686 static int 
687 fill_features_reply(struct datapath *dp, struct ofp_switch_features *ofr)
688 {
689         struct net_bridge_port *p;
690         int port_count = 0;
691
692         ofr->datapath_id    = cpu_to_be64(dp->id); 
693
694         ofr->n_exact        = htonl(2 * TABLE_HASH_MAX_FLOWS);
695         ofr->n_compression  = 0;                                           /* Not supported */
696         ofr->n_general      = htonl(TABLE_LINEAR_MAX_FLOWS);
697         ofr->buffer_mb      = htonl(UINT32_MAX);
698         ofr->n_buffers      = htonl(N_PKT_BUFFERS);
699         ofr->capabilities   = htonl(OFP_SUPPORTED_CAPABILITIES);
700         ofr->actions        = htonl(OFP_SUPPORTED_ACTIONS);
701
702         list_for_each_entry_rcu (p, &dp->port_list, node) {
703                 fill_port_desc(p, &ofr->ports[port_count]);
704                 port_count++;
705         }
706
707         return port_count;
708 }
709
710 int
711 dp_send_features_reply(struct datapath *dp, const struct sender *sender)
712 {
713         struct sk_buff *skb;
714         struct ofp_switch_features *ofr;
715         size_t ofr_len, port_max_len;
716         int port_count;
717
718         /* Overallocate. */
719         port_max_len = sizeof(struct ofp_phy_port) * OFPP_MAX;
720         ofr = alloc_openflow_skb(dp, sizeof(*ofr) + port_max_len,
721                                  OFPT_FEATURES_REPLY, sender, &skb);
722         if (!ofr)
723                 return -ENOMEM;
724
725         /* Fill. */
726         port_count = fill_features_reply(dp, ofr);
727
728         /* Shrink to fit. */
729         ofr_len = sizeof(*ofr) + (sizeof(struct ofp_phy_port) * port_count);
730         resize_openflow_skb(skb, &ofr->header, ofr_len);
731         return send_openflow_skb(skb, sender);
732 }
733
734 int
735 dp_send_config_reply(struct datapath *dp, const struct sender *sender)
736 {
737         struct sk_buff *skb;
738         struct ofp_switch_config *osc;
739
740         osc = alloc_openflow_skb(dp, sizeof *osc, OFPT_GET_CONFIG_REPLY, sender,
741                                  &skb);
742         if (!osc)
743                 return -ENOMEM;
744
745         osc->flags = htons(dp->flags);
746         osc->miss_send_len = htons(dp->miss_send_len);
747
748         return send_openflow_skb(skb, sender);
749 }
750
751 int
752 dp_update_port_flags(struct datapath *dp, const struct ofp_phy_port *opp)
753 {
754         struct net_bridge_port *p;
755
756         p = dp->ports[htons(opp->port_no)];
757
758         /* Make sure the port id hasn't changed since this was sent */
759         if (!p || memcmp(opp->hw_addr, p->dev->dev_addr, ETH_ALEN) != 0) 
760                 return -1;
761         
762         p->flags = htonl(opp->flags);
763
764         return 0;
765 }
766
767
768 static int
769 send_port_status(struct net_bridge_port *p, uint8_t status)
770 {
771         struct sk_buff *skb;
772         struct ofp_port_status *ops;
773
774         ops = alloc_openflow_skb(p->dp, sizeof *ops, OFPT_PORT_STATUS, NULL,
775                                  &skb);
776         if (!ops)
777                 return -ENOMEM;
778         ops->reason = status;
779         memset(ops->pad, 0, sizeof ops->pad);
780         fill_port_desc(p, &ops->desc);
781
782         return send_openflow_skb(skb, NULL);
783 }
784
785 int 
786 dp_send_flow_expired(struct datapath *dp, struct sw_flow *flow)
787 {
788         struct sk_buff *skb;
789         struct ofp_flow_expired *ofe;
790         unsigned long duration_j;
791
792         ofe = alloc_openflow_skb(dp, sizeof *ofe, OFPT_FLOW_EXPIRED, 0, &skb);
793         if (!ofe)
794                 return -ENOMEM;
795
796         flow_fill_match(&ofe->match, &flow->key);
797
798         memset(ofe->pad, 0, sizeof ofe->pad);
799         ofe->priority = htons(flow->priority);
800
801         duration_j = (flow->timeout - HZ * flow->max_idle) - flow->init_time;
802         ofe->duration     = htonl(duration_j / HZ);
803         ofe->packet_count = cpu_to_be64(flow->packet_count);
804         ofe->byte_count   = cpu_to_be64(flow->byte_count);
805
806         return send_openflow_skb(skb, NULL);
807 }
808 EXPORT_SYMBOL(dp_send_flow_expired);
809
810 int
811 dp_send_error_msg(struct datapath *dp, const struct sender *sender, 
812                 uint16_t type, uint16_t code, const uint8_t *data, size_t len)
813 {
814         struct sk_buff *skb;
815         struct ofp_error_msg *oem;
816
817
818         oem = alloc_openflow_skb(dp, sizeof(*oem)+len, OFPT_ERROR_MSG, 
819                         sender, &skb);
820         if (!oem)
821                 return -ENOMEM;
822
823         oem->type = htons(type);
824         oem->code = htons(code);
825         memcpy(oem->data, data, len);
826
827         return send_openflow_skb(skb, sender);
828 }
829
830 /* Generic Netlink interface.
831  *
832  * See netlink(7) for an introduction to netlink.  See
833  * http://linux-net.osdl.org/index.php/Netlink for more information and
834  * pointers on how to work with netlink and Generic Netlink in the kernel and
835  * in userspace. */
836
837 static struct genl_family dp_genl_family = {
838         .id = GENL_ID_GENERATE,
839         .hdrsize = 0,
840         .name = DP_GENL_FAMILY_NAME,
841         .version = 1,
842         .maxattr = DP_GENL_A_MAX,
843 };
844
845 /* Attribute policy: what each attribute may contain.  */
846 static struct nla_policy dp_genl_policy[DP_GENL_A_MAX + 1] = {
847         [DP_GENL_A_DP_IDX] = { .type = NLA_U32 },
848         [DP_GENL_A_MC_GROUP] = { .type = NLA_U32 },
849         [DP_GENL_A_PORTNAME] = { .type = NLA_STRING }
850 };
851
852 static int dp_genl_add(struct sk_buff *skb, struct genl_info *info)
853 {
854         if (!info->attrs[DP_GENL_A_DP_IDX])
855                 return -EINVAL;
856
857         return new_dp(nla_get_u32(info->attrs[DP_GENL_A_DP_IDX]));
858 }
859
860 static struct genl_ops dp_genl_ops_add_dp = {
861         .cmd = DP_GENL_C_ADD_DP,
862         .flags = GENL_ADMIN_PERM, /* Requires CAP_NET_ADMIN privilege. */
863         .policy = dp_genl_policy,
864         .doit = dp_genl_add,
865         .dumpit = NULL,
866 };
867
868 struct datapath *dp_get(int dp_idx)
869 {
870         if (dp_idx < 0 || dp_idx > DP_MAX)
871                 return NULL;
872         return rcu_dereference(dps[dp_idx]);
873 }
874
875 static int dp_genl_del(struct sk_buff *skb, struct genl_info *info)
876 {
877         struct datapath *dp;
878         int err;
879
880         if (!info->attrs[DP_GENL_A_DP_IDX])
881                 return -EINVAL;
882
883         mutex_lock(&dp_mutex);
884         dp = dp_get(nla_get_u32((info->attrs[DP_GENL_A_DP_IDX])));
885         if (!dp)
886                 err = -ENOENT;
887         else {
888                 del_dp(dp);
889                 err = 0;
890         }
891         mutex_unlock(&dp_mutex);
892         return err;
893 }
894
895 static struct genl_ops dp_genl_ops_del_dp = {
896         .cmd = DP_GENL_C_DEL_DP,
897         .flags = GENL_ADMIN_PERM, /* Requires CAP_NET_ADMIN privilege. */
898         .policy = dp_genl_policy,
899         .doit = dp_genl_del,
900         .dumpit = NULL,
901 };
902
903 /* Queries a datapath for related information.  Currently the only relevant
904  * information is the datapath's multicast group ID.  Really we want one
905  * multicast group per datapath, but because of locking issues[*] we can't
906  * easily get one.  Thus, every datapath will currently return the same
907  * global multicast group ID, but in the future it would be nice to fix that.
908  *
909  * [*] dp_genl_add, to add a new datapath, is called under the genl_lock
910  *       mutex, and genl_register_mc_group, called to acquire a new multicast
911  *       group ID, also acquires genl_lock, thus deadlock.
912  */
913 static int dp_genl_query(struct sk_buff *skb, struct genl_info *info)
914 {
915         struct datapath *dp;
916         struct sk_buff *ans_skb = NULL;
917         int dp_idx;
918         int err = -ENOMEM;
919
920         if (!info->attrs[DP_GENL_A_DP_IDX])
921                 return -EINVAL;
922
923         rcu_read_lock();
924         dp_idx = nla_get_u32((info->attrs[DP_GENL_A_DP_IDX]));
925         dp = dp_get(dp_idx);
926         if (!dp)
927                 err = -ENOENT;
928         else {
929                 void *data;
930                 ans_skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_ATOMIC);
931                 if (!ans_skb) {
932                         err = -ENOMEM;
933                         goto err;
934                 }
935                 data = genlmsg_put_reply(ans_skb, info, &dp_genl_family,
936                                          0, DP_GENL_C_QUERY_DP);
937                 if (data == NULL) {
938                         err = -ENOMEM;
939                         goto err;
940                 }
941                 NLA_PUT_U32(ans_skb, DP_GENL_A_DP_IDX, dp_idx);
942                 NLA_PUT_U32(ans_skb, DP_GENL_A_MC_GROUP, mc_group.id);
943
944                 genlmsg_end(ans_skb, data);
945                 err = genlmsg_reply(ans_skb, info);
946                 if (!err)
947                         ans_skb = NULL;
948         }
949 err:
950 nla_put_failure:
951         if (ans_skb)
952                 kfree_skb(ans_skb);
953         rcu_read_unlock();
954         return err;
955 }
956
957 static struct genl_ops dp_genl_ops_query_dp = {
958         .cmd = DP_GENL_C_QUERY_DP,
959         .flags = GENL_ADMIN_PERM, /* Requires CAP_NET_ADMIN privilege. */
960         .policy = dp_genl_policy,
961         .doit = dp_genl_query,
962         .dumpit = NULL,
963 };
964
965 static int dp_genl_add_del_port(struct sk_buff *skb, struct genl_info *info)
966 {
967         struct datapath *dp;
968         struct net_device *port;
969         int err;
970
971         if (!info->attrs[DP_GENL_A_DP_IDX] || !info->attrs[DP_GENL_A_PORTNAME])
972                 return -EINVAL;
973
974         /* Get datapath. */
975         mutex_lock(&dp_mutex);
976         dp = dp_get(nla_get_u32(info->attrs[DP_GENL_A_DP_IDX]));
977         if (!dp) {
978                 err = -ENOENT;
979                 goto out;
980         }
981
982         /* Get interface to add/remove. */
983         port = dev_get_by_name(&init_net, 
984                         nla_data(info->attrs[DP_GENL_A_PORTNAME]));
985         if (!port) {
986                 err = -ENOENT;
987                 goto out;
988         }
989
990         /* Execute operation. */
991         if (info->genlhdr->cmd == DP_GENL_C_ADD_PORT)
992                 err = add_switch_port(dp, port);
993         else {
994                 if (port->br_port == NULL || port->br_port->dp != dp) {
995                         err = -ENOENT;
996                         goto out_put;
997                 }
998                 err = del_switch_port(port->br_port);
999         }
1000
1001 out_put:
1002         dev_put(port);
1003 out:
1004         mutex_unlock(&dp_mutex);
1005         return err;
1006 }
1007
1008 static struct genl_ops dp_genl_ops_add_port = {
1009         .cmd = DP_GENL_C_ADD_PORT,
1010         .flags = GENL_ADMIN_PERM, /* Requires CAP_NET_ADMIN privilege. */
1011         .policy = dp_genl_policy,
1012         .doit = dp_genl_add_del_port,
1013         .dumpit = NULL,
1014 };
1015
1016 static struct genl_ops dp_genl_ops_del_port = {
1017         .cmd = DP_GENL_C_DEL_PORT,
1018         .flags = GENL_ADMIN_PERM, /* Requires CAP_NET_ADMIN privilege. */
1019         .policy = dp_genl_policy,
1020         .doit = dp_genl_add_del_port,
1021         .dumpit = NULL,
1022 };
1023
1024 static int dp_genl_openflow(struct sk_buff *skb, struct genl_info *info)
1025 {
1026         struct nlattr *va = info->attrs[DP_GENL_A_OPENFLOW];
1027         struct datapath *dp;
1028         struct ofp_header *oh;
1029         struct sender sender;
1030         int err;
1031
1032         if (!info->attrs[DP_GENL_A_DP_IDX] || !va)
1033                 return -EINVAL;
1034
1035         rcu_read_lock();
1036         dp = dp_get(nla_get_u32(info->attrs[DP_GENL_A_DP_IDX]));
1037         if (!dp) {
1038                 err = -ENOENT;
1039                 goto out;
1040         }
1041
1042         if (nla_len(va) < sizeof(struct ofp_header)) {
1043                 err = -EINVAL;
1044                 goto out;
1045         }
1046         oh = nla_data(va);
1047
1048         sender.xid = oh->xid;
1049         sender.pid = info->snd_pid;
1050         sender.seq = info->snd_seq;
1051         err = fwd_control_input(dp->chain, &sender, nla_data(va), nla_len(va));
1052
1053 out:
1054         rcu_read_unlock();
1055         return err;
1056 }
1057
1058 static struct nla_policy dp_genl_openflow_policy[DP_GENL_A_MAX + 1] = {
1059         [DP_GENL_A_DP_IDX] = { .type = NLA_U32 },
1060 };
1061
1062 struct flow_stats_state {
1063         int table_idx;
1064         struct sw_table_position position;
1065         const struct ofp_flow_stats_request *rq;
1066
1067         void *body;
1068         int bytes_used, bytes_allocated;
1069 };
1070
1071 static int flow_stats_init(struct datapath *dp, const void *body, int body_len,
1072                            void **state)
1073 {
1074         const struct ofp_flow_stats_request *fsr = body;
1075         struct flow_stats_state *s = kmalloc(sizeof *s, GFP_ATOMIC);
1076         if (!s)
1077                 return -ENOMEM;
1078         s->table_idx = fsr->table_id == 0xff ? 0 : fsr->table_id;
1079         memset(&s->position, 0, sizeof s->position);
1080         s->rq = fsr;
1081         *state = s;
1082         return 0;
1083 }
1084
1085 static int flow_stats_dump_callback(struct sw_flow *flow, void *private)
1086 {
1087         struct flow_stats_state *s = private;
1088         struct ofp_flow_stats *ofs;
1089         int actions_length;
1090         int length;
1091
1092         actions_length = sizeof *ofs->actions * flow->n_actions;
1093         length = sizeof *ofs + sizeof *ofs->actions * flow->n_actions;
1094         if (length + s->bytes_used > s->bytes_allocated)
1095                 return 1;
1096
1097         ofs = s->body + s->bytes_used;
1098         ofs->length          = htons(length);
1099         ofs->table_id        = s->table_idx;
1100         ofs->pad             = 0;
1101         ofs->match.wildcards = htons(flow->key.wildcards);
1102         ofs->match.in_port   = flow->key.in_port;
1103         memcpy(ofs->match.dl_src, flow->key.dl_src, ETH_ALEN);
1104         memcpy(ofs->match.dl_dst, flow->key.dl_dst, ETH_ALEN);
1105         ofs->match.dl_vlan   = flow->key.dl_vlan;
1106         ofs->match.dl_type   = flow->key.dl_type;
1107         ofs->match.nw_src    = flow->key.nw_src;
1108         ofs->match.nw_dst    = flow->key.nw_dst;
1109         ofs->match.nw_proto  = flow->key.nw_proto;
1110         memset(ofs->match.pad, 0, sizeof ofs->match.pad);
1111         ofs->match.tp_src    = flow->key.tp_src;
1112         ofs->match.tp_dst    = flow->key.tp_dst;
1113         ofs->duration        = htonl((jiffies - flow->init_time) / HZ);
1114         ofs->packet_count    = cpu_to_be64(flow->packet_count);
1115         ofs->byte_count      = cpu_to_be64(flow->byte_count);
1116         ofs->priority        = htons(flow->priority);
1117         ofs->max_idle        = htons(flow->max_idle);
1118         memcpy(ofs->actions, flow->actions, actions_length);
1119
1120         s->bytes_used += length;
1121         return 0;
1122 }
1123
1124 static int flow_stats_dump(struct datapath *dp, void *state,
1125                            void *body, int *body_len)
1126 {
1127         struct flow_stats_state *s = state;
1128         struct sw_flow_key match_key;
1129         int error = 0;
1130
1131         s->bytes_used = 0;
1132         s->bytes_allocated = *body_len;
1133         s->body = body;
1134
1135         flow_extract_match(&match_key, &s->rq->match);
1136         while (s->table_idx < dp->chain->n_tables
1137                && (s->rq->table_id == 0xff || s->rq->table_id == s->table_idx))
1138         {
1139                 struct sw_table *table = dp->chain->tables[s->table_idx];
1140
1141                 error = table->iterate(table, &match_key, &s->position,
1142                                        flow_stats_dump_callback, s);
1143                 if (error)
1144                         break;
1145
1146                 s->table_idx++;
1147                 memset(&s->position, 0, sizeof s->position);
1148         }
1149         *body_len = s->bytes_used;
1150
1151         /* If error is 0, we're done.
1152          * Otherwise, if some bytes were used, there are more flows to come.
1153          * Otherwise, we were not able to fit even a single flow in the body,
1154          * which indicates that we have a single flow with too many actions to
1155          * fit.  We won't ever make any progress at that rate, so give up. */
1156         return !error ? 0 : s->bytes_used ? 1 : -ENOMEM;
1157 }
1158
1159 static void flow_stats_done(void *state)
1160 {
1161         kfree(state);
1162 }
1163
1164 static int aggregate_stats_init(struct datapath *dp,
1165                                 const void *body, int body_len,
1166                                 void **state)
1167 {
1168         *state = (void *)body;
1169         return 0;
1170 }
1171
1172 static int aggregate_stats_dump_callback(struct sw_flow *flow, void *private)
1173 {
1174         struct ofp_aggregate_stats_reply *rpy = private;
1175         rpy->packet_count += flow->packet_count;
1176         rpy->byte_count += flow->byte_count;
1177         rpy->flow_count++;
1178         return 0;
1179 }
1180
1181 static int aggregate_stats_dump(struct datapath *dp, void *state,
1182                                 void *body, int *body_len)
1183 {
1184         struct ofp_aggregate_stats_request *rq = state;
1185         struct ofp_aggregate_stats_reply *rpy;
1186         struct sw_table_position position;
1187         struct sw_flow_key match_key;
1188         int table_idx;
1189
1190         if (*body_len < sizeof *rpy)
1191                 return -ENOBUFS;
1192         rpy = body;
1193         *body_len = sizeof *rpy;
1194
1195         memset(rpy, 0, sizeof *rpy);
1196
1197         flow_extract_match(&match_key, &rq->match);
1198         table_idx = rq->table_id == 0xff ? 0 : rq->table_id;
1199         memset(&position, 0, sizeof position);
1200         while (table_idx < dp->chain->n_tables
1201                && (rq->table_id == 0xff || rq->table_id == table_idx))
1202         {
1203                 struct sw_table *table = dp->chain->tables[table_idx];
1204                 int error;
1205
1206                 error = table->iterate(table, &match_key, &position,
1207                                        aggregate_stats_dump_callback, rpy);
1208                 if (error)
1209                         return error;
1210
1211                 table_idx++;
1212                 memset(&position, 0, sizeof position);
1213         }
1214
1215         rpy->packet_count = cpu_to_be64(rpy->packet_count);
1216         rpy->byte_count = cpu_to_be64(rpy->byte_count);
1217         rpy->flow_count = htonl(rpy->flow_count);
1218         return 0;
1219 }
1220
1221 static int table_stats_dump(struct datapath *dp, void *state,
1222                             void *body, int *body_len)
1223 {
1224         struct ofp_table_stats *ots;
1225         int nbytes = dp->chain->n_tables * sizeof *ots;
1226         int i;
1227         if (nbytes > *body_len)
1228                 return -ENOBUFS;
1229         *body_len = nbytes;
1230         for (i = 0, ots = body; i < dp->chain->n_tables; i++, ots++) {
1231                 struct sw_table_stats stats;
1232                 dp->chain->tables[i]->stats(dp->chain->tables[i], &stats);
1233                 strncpy(ots->name, stats.name, sizeof ots->name);
1234                 ots->table_id = i;
1235                 memset(ots->pad, 0, sizeof ots->pad);
1236                 ots->max_entries = htonl(stats.max_flows);
1237                 ots->active_count = htonl(stats.n_flows);
1238                 ots->matched_count = cpu_to_be64(0); /* FIXME */
1239         }
1240         return 0;
1241 }
1242
1243 struct port_stats_state {
1244         int port;
1245 };
1246
1247 static int port_stats_init(struct datapath *dp, const void *body, int body_len,
1248                            void **state)
1249 {
1250         struct port_stats_state *s = kmalloc(sizeof *s, GFP_ATOMIC);
1251         if (!s)
1252                 return -ENOMEM;
1253         s->port = 0;
1254         *state = s;
1255         return 0;
1256 }
1257
1258 static int port_stats_dump(struct datapath *dp, void *state,
1259                            void *body, int *body_len)
1260 {
1261         struct port_stats_state *s = state;
1262         struct ofp_port_stats *ops;
1263         int n_ports, max_ports;
1264         int i;
1265
1266         max_ports = *body_len / sizeof *ops;
1267         if (!max_ports)
1268                 return -ENOMEM;
1269         ops = body;
1270
1271         n_ports = 0;
1272         for (i = s->port; i < OFPP_MAX && n_ports < max_ports; i++) {
1273                 struct net_bridge_port *p = dp->ports[i];
1274                 struct net_device_stats *stats;
1275                 if (!p)
1276                         continue;
1277                 stats = p->dev->get_stats(p->dev);
1278                 ops->port_no = htons(p->port_no);
1279                 memset(ops->pad, 0, sizeof ops->pad);
1280                 ops->rx_count = cpu_to_be64(stats->rx_packets);
1281                 ops->tx_count = cpu_to_be64(stats->tx_packets);
1282                 ops->drop_count = cpu_to_be64(stats->rx_dropped
1283                                               + stats->tx_dropped);
1284                 n_ports++;
1285                 ops++;
1286         }
1287         s->port = i;
1288         *body_len = n_ports * sizeof *ops;
1289         return n_ports >= max_ports;
1290 }
1291
1292 static void port_stats_done(void *state)
1293 {
1294         kfree(state);
1295 }
1296
1297 struct stats_type {
1298         /* Minimum and maximum acceptable number of bytes in body member of
1299          * struct ofp_stats_request. */
1300         size_t min_body, max_body;
1301
1302         /* Prepares to dump some kind of statistics on 'dp'.  'body' and
1303          * 'body_len' are the 'body' member of the struct ofp_stats_request.
1304          * Returns zero if successful, otherwise a negative error code.
1305          * May initialize '*state' to state information.  May be null if no
1306          * initialization is required.*/
1307         int (*init)(struct datapath *dp, const void *body, int body_len,
1308                     void **state);
1309
1310         /* Dumps statistics for 'dp' into the '*body_len' bytes at 'body', and
1311          * modifies '*body_len' to reflect the number of bytes actually used.
1312          * ('body' will be transmitted as the 'body' member of struct
1313          * ofp_stats_reply.) */
1314         int (*dump)(struct datapath *dp, void *state,
1315                     void *body, int *body_len);
1316
1317         /* Cleans any state created by the init or dump functions.  May be null
1318          * if no cleanup is required. */
1319         void (*done)(void *state);
1320 };
1321
1322 static const struct stats_type stats[] = {
1323         [OFPST_FLOW] = {
1324                 sizeof(struct ofp_flow_stats_request),
1325                 sizeof(struct ofp_flow_stats_request),
1326                 flow_stats_init,
1327                 flow_stats_dump,
1328                 flow_stats_done
1329         },
1330         [OFPST_AGGREGATE] = {
1331                 sizeof(struct ofp_aggregate_stats_request),
1332                 sizeof(struct ofp_aggregate_stats_request),
1333                 aggregate_stats_init,
1334                 aggregate_stats_dump,
1335                 NULL
1336         },
1337         [OFPST_TABLE] = {
1338                 0,
1339                 0,
1340                 NULL,
1341                 table_stats_dump,
1342                 NULL
1343         },
1344         [OFPST_PORT] = {
1345                 0,
1346                 0,
1347                 port_stats_init,
1348                 port_stats_dump,
1349                 port_stats_done
1350         },
1351 };
1352
1353 static int
1354 dp_genl_openflow_dumpit(struct sk_buff *skb, struct netlink_callback *cb)
1355 {
1356         struct datapath *dp;
1357         struct sender sender;
1358         const struct stats_type *s;
1359         struct ofp_stats_reply *osr;
1360         int dp_idx;
1361         int max_openflow_len, body_len;
1362         void *body;
1363         int err;
1364
1365         /* Set up the cleanup function for this dump.  Linux 2.6.20 and later
1366          * support setting up cleanup functions via the .doneit member of
1367          * struct genl_ops.  This kluge supports earlier versions also. */
1368         cb->done = dp_genl_openflow_done;
1369
1370         rcu_read_lock();
1371         if (!cb->args[0]) {
1372                 struct nlattr *attrs[DP_GENL_A_MAX + 1];
1373                 struct ofp_stats_request *rq;
1374                 struct nlattr *va;
1375                 size_t len, body_len;
1376                 int type;
1377
1378                 err = nlmsg_parse(cb->nlh, GENL_HDRLEN, attrs, DP_GENL_A_MAX,
1379                                   dp_genl_openflow_policy);
1380                 if (err < 0)
1381                         return err;
1382
1383                 err = -EINVAL;
1384
1385                 if (!attrs[DP_GENL_A_DP_IDX])
1386                         goto out;
1387                 dp_idx = nla_get_u16(attrs[DP_GENL_A_DP_IDX]);
1388                 dp = dp_get(dp_idx);
1389                 if (!dp) {
1390                         err = -ENOENT;
1391                         goto out;
1392                 }
1393
1394                 va = attrs[DP_GENL_A_OPENFLOW];
1395                 len = nla_len(va);
1396                 if (!va || len < sizeof *rq)
1397                         goto out;
1398
1399                 rq = nla_data(va);
1400                 type = ntohs(rq->type);
1401                 if (rq->header.version != OFP_VERSION
1402                     || rq->header.type != OFPT_STATS_REQUEST
1403                     || ntohs(rq->header.length) != len
1404                     || type >= ARRAY_SIZE(stats)
1405                     || !stats[type].dump)
1406                         goto out;
1407
1408                 s = &stats[type];
1409                 body_len = len - offsetof(struct ofp_stats_request, body);
1410                 if (body_len < s->min_body || body_len > s->max_body)
1411                         goto out;
1412
1413                 cb->args[0] = 1;
1414                 cb->args[1] = dp_idx;
1415                 cb->args[2] = type;
1416                 cb->args[3] = rq->header.xid;
1417                 if (s->init) {
1418                         void *state;
1419                         err = s->init(dp, rq->body, body_len, &state);
1420                         if (err)
1421                                 goto out;
1422                         cb->args[4] = (long) state;
1423                 }
1424         } else if (cb->args[0] == 1) {
1425                 dp_idx = cb->args[1];
1426                 s = &stats[cb->args[2]];
1427
1428                 dp = dp_get(dp_idx);
1429                 if (!dp) {
1430                         err = -ENOENT;
1431                         goto out;
1432                 }
1433         } else {
1434                 err = 0;
1435                 goto out;
1436         }
1437
1438         sender.xid = cb->args[3];
1439         sender.pid = NETLINK_CB(cb->skb).pid;
1440         sender.seq = cb->nlh->nlmsg_seq;
1441
1442         osr = put_openflow_headers(dp, skb, OFPT_STATS_REPLY, &sender,
1443                                    &max_openflow_len);
1444         if (IS_ERR(osr)) {
1445                 err = PTR_ERR(osr);
1446                 goto out;
1447         }
1448         osr->type = htons(s - stats);
1449         osr->flags = 0;
1450         resize_openflow_skb(skb, &osr->header, max_openflow_len);
1451         body = osr->body;
1452         body_len = max_openflow_len - offsetof(struct ofp_stats_reply, body);
1453
1454         err = s->dump(dp, (void *) cb->args[4], body, &body_len);
1455         if (err >= 0) {
1456                 if (!err)
1457                         cb->args[0] = 2;
1458                 else
1459                         osr->flags = ntohs(OFPSF_REPLY_MORE);
1460                 resize_openflow_skb(skb, &osr->header,
1461                                     (offsetof(struct ofp_stats_reply, body)
1462                                      + body_len));
1463                 err = skb->len;
1464         }
1465
1466 out:
1467         rcu_read_unlock();
1468         return err;
1469 }
1470
1471 static int
1472 dp_genl_openflow_done(struct netlink_callback *cb)
1473 {
1474         if (cb->args[0]) {
1475                 const struct stats_type *s = &stats[cb->args[2]];
1476                 if (s->done)
1477                         s->done((void *) cb->args[4]);
1478         }
1479         return 0;
1480 }
1481
1482 static struct genl_ops dp_genl_ops_openflow = {
1483         .cmd = DP_GENL_C_OPENFLOW,
1484         .flags = GENL_ADMIN_PERM, /* Requires CAP_NET_ADMIN privilege. */
1485         .policy = dp_genl_openflow_policy,
1486         .doit = dp_genl_openflow,
1487         .dumpit = dp_genl_openflow_dumpit,
1488 };
1489
1490 static struct nla_policy dp_genl_benchmark_policy[DP_GENL_A_MAX + 1] = {
1491         [DP_GENL_A_DP_IDX] = { .type = NLA_U32 },
1492         [DP_GENL_A_NPACKETS] = { .type = NLA_U32 },
1493         [DP_GENL_A_PSIZE] = { .type = NLA_U32 },
1494 };
1495
1496 static struct genl_ops dp_genl_ops_benchmark_nl = {
1497         .cmd = DP_GENL_C_BENCHMARK_NL,
1498         .flags = GENL_ADMIN_PERM, /* Requires CAP_NET_ADMIN privilege. */
1499         .policy = dp_genl_benchmark_policy,
1500         .doit = dp_genl_benchmark_nl,
1501         .dumpit = NULL,
1502 };
1503
1504 static struct genl_ops *dp_genl_all_ops[] = {
1505         /* Keep this operation first.  Generic Netlink dispatching
1506          * looks up operations with linear search, so we want it at the
1507          * front. */
1508         &dp_genl_ops_openflow,
1509
1510         &dp_genl_ops_add_dp,
1511         &dp_genl_ops_del_dp,
1512         &dp_genl_ops_query_dp,
1513         &dp_genl_ops_add_port,
1514         &dp_genl_ops_del_port,
1515         &dp_genl_ops_benchmark_nl,
1516 };
1517
1518 static int dp_init_netlink(void)
1519 {
1520         int err;
1521         int i;
1522
1523         err = genl_register_family(&dp_genl_family);
1524         if (err)
1525                 return err;
1526
1527         for (i = 0; i < ARRAY_SIZE(dp_genl_all_ops); i++) {
1528                 err = genl_register_ops(&dp_genl_family, dp_genl_all_ops[i]);
1529                 if (err)
1530                         goto err_unregister;
1531         }
1532
1533         strcpy(mc_group.name, "openflow");
1534         err = genl_register_mc_group(&dp_genl_family, &mc_group);
1535         if (err < 0)
1536                 goto err_unregister;
1537
1538         return 0;
1539
1540 err_unregister:
1541         genl_unregister_family(&dp_genl_family);
1542                 return err;
1543 }
1544
1545 static void dp_uninit_netlink(void)
1546 {
1547         genl_unregister_family(&dp_genl_family);
1548 }
1549
1550 #define DRV_NAME                "openflow"
1551 #define DRV_VERSION      VERSION
1552 #define DRV_DESCRIPTION "OpenFlow switching datapath implementation"
1553 #define DRV_COPYRIGHT   "Copyright (c) 2007, 2008 The Board of Trustees of The Leland Stanford Junior University"
1554
1555
1556 static int __init dp_init(void)
1557 {
1558         int err;
1559
1560         printk(KERN_INFO DRV_NAME ": " DRV_DESCRIPTION "\n");
1561         printk(KERN_INFO DRV_NAME ": " VERSION" built on "__DATE__" "__TIME__"\n");
1562         printk(KERN_INFO DRV_NAME ": " DRV_COPYRIGHT "\n");
1563
1564         err = flow_init();
1565         if (err)
1566                 goto error;
1567
1568         err = dp_init_netlink();
1569         if (err)
1570                 goto error_flow_exit;
1571
1572         /* Hook into callback used by the bridge to intercept packets.
1573          * Parasites we are. */
1574         if (br_handle_frame_hook)
1575                 printk("openflow: hijacking bridge hook\n");
1576         br_handle_frame_hook = dp_frame_hook;
1577
1578         return 0;
1579
1580 error_flow_exit:
1581         flow_exit();
1582 error:
1583         printk(KERN_EMERG "openflow: failed to install!");
1584         return err;
1585 }
1586
1587 static void dp_cleanup(void)
1588 {
1589         fwd_exit();
1590         dp_uninit_netlink();
1591         flow_exit();
1592         br_handle_frame_hook = NULL;
1593 }
1594
1595 module_init(dp_init);
1596 module_exit(dp_cleanup);
1597
1598 MODULE_DESCRIPTION(DRV_DESCRIPTION);
1599 MODULE_AUTHOR(DRV_COPYRIGHT);
1600 MODULE_LICENSE("GPL");