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