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