Prepare Open vSwitch 1.1.2 release.
[sliver-openvswitch.git] / datapath / flow.c
index 09bdcd1..d670925 100644 (file)
 #include <linux/if_arp.h>
 #include <linux/if_ether.h>
 #include <linux/ip.h>
+#include <linux/ipv6.h>
 #include <linux/tcp.h>
 #include <linux/udp.h>
 #include <linux/icmp.h>
+#include <linux/icmpv6.h>
 #include <net/inet_ecn.h>
 #include <net/ip.h>
+#include <net/ipv6.h>
+#include <net/ndisc.h>
+
+#include "vlan.h"
 
 static struct kmem_cache *flow_cache;
 static unsigned int hash_seed __read_mostly;
@@ -82,6 +88,63 @@ static inline bool icmphdr_ok(struct sk_buff *skb)
        return skb->len >= skb_transport_offset(skb) + sizeof(struct icmphdr);
 }
 
+u64 flow_used_time(unsigned long flow_jiffies)
+{
+       struct timespec cur_ts;
+       u64 cur_ms, idle_ms;
+
+       ktime_get_ts(&cur_ts);
+       idle_ms = jiffies_to_msecs(jiffies - flow_jiffies);
+       cur_ms = (u64)cur_ts.tv_sec * MSEC_PER_SEC +
+                cur_ts.tv_nsec / NSEC_PER_MSEC;
+
+       return cur_ms - idle_ms;
+}
+
+static int parse_ipv6hdr(struct sk_buff *skb, struct sw_flow_key *key)
+{
+       unsigned int nh_ofs = skb_network_offset(skb);
+       unsigned int nh_len;
+       int payload_ofs;
+       struct ipv6hdr *nh;
+       uint8_t nexthdr;
+
+       if (unlikely(skb->len < nh_ofs + sizeof(*nh)))
+               return -EINVAL;
+
+       nh = ipv6_hdr(skb);
+       nexthdr = nh->nexthdr;
+       payload_ofs = (u8 *)(nh + 1) - skb->data;
+
+       ipv6_addr_copy(&key->ipv6_src, &nh->saddr);
+       ipv6_addr_copy(&key->ipv6_dst, &nh->daddr);
+       key->nw_tos = ipv6_get_dsfield(nh) & ~INET_ECN_MASK;
+       key->nw_proto = NEXTHDR_NONE;
+
+       payload_ofs = ipv6_skip_exthdr(skb, payload_ofs, &nexthdr);
+       if (unlikely(payload_ofs < 0))
+               return -EINVAL;
+
+       nh_len = payload_ofs - nh_ofs;
+
+       /* Pull enough header bytes to account for the IP header plus the
+        * longest transport header that we parse, currently 20 bytes for TCP.
+        * To dig deeper than the transport header, transport parsers may need
+        * to pull more header bytes.
+        */
+       if (unlikely(!pskb_may_pull(skb, min(nh_ofs + nh_len + 20, skb->len))))
+               return -ENOMEM;
+
+       skb_set_transport_header(skb, nh_ofs + nh_len);
+       key->nw_proto = nexthdr;
+       return nh_len;
+}
+
+static bool icmp6hdr_ok(struct sk_buff *skb)
+{
+       return skb->len >= skb_transport_offset(skb) + sizeof(struct icmp6hdr);
+}
+
 #define TCP_FLAGS_OFFSET 13
 #define TCP_FLAG_MASK 0x3f
 
@@ -103,13 +166,11 @@ void flow_used(struct sw_flow *flow, struct sk_buff *skb)
        spin_unlock_bh(&flow->lock);
 }
 
-struct sw_flow_actions *flow_actions_alloc(u32 actions_len)
+struct sw_flow_actions *flow_actions_alloc(const struct nlattr *actions)
 {
+       int actions_len = nla_len(actions);
        struct sw_flow_actions *sfa;
 
-       if (actions_len % NLA_ALIGNTO)
-               return ERR_PTR(-EINVAL);
-
        /* At least DP_MAX_PORTS actions are required to be able to flood a
         * packet to every port.  Factor of 2 allows for setting VLAN tags,
         * etc. */
@@ -121,6 +182,7 @@ struct sw_flow_actions *flow_actions_alloc(u32 actions_len)
                return ERR_PTR(-ENOMEM);
 
        sfa->actions_len = actions_len;
+       memcpy(sfa->actions, nla_data(actions), actions_len);
        return sfa;
 }
 
@@ -134,6 +196,7 @@ struct sw_flow *flow_alloc(void)
 
        spin_lock_init(&flow->lock);
        atomic_set(&flow->refcnt, 1);
+       flow->sf_acts = NULL;
        flow->dead = false;
 
        return flow;
@@ -241,6 +304,77 @@ static __be16 parse_ethertype(struct sk_buff *skb)
        return llc->ethertype;
 }
 
+static int parse_icmpv6(struct sk_buff *skb, struct sw_flow_key *key,
+                       int nh_len)
+{
+       struct icmp6hdr *icmp = icmp6_hdr(skb);
+
+       /* The ICMPv6 type and code fields use the 16-bit transport port
+        * fields, so we need to store them in 16-bit network byte order.
+        */
+       key->tp_src = htons(icmp->icmp6_type);
+       key->tp_dst = htons(icmp->icmp6_code);
+
+       if (icmp->icmp6_code == 0 &&
+           (icmp->icmp6_type == NDISC_NEIGHBOUR_SOLICITATION ||
+            icmp->icmp6_type == NDISC_NEIGHBOUR_ADVERTISEMENT)) {
+               int icmp_len = skb->len - skb_transport_offset(skb);
+               struct nd_msg *nd;
+               int offset;
+
+               /* In order to process neighbor discovery options, we need the
+                * entire packet.
+                */
+               if (unlikely(icmp_len < sizeof(*nd)))
+                       return 0;
+               if (unlikely(skb_linearize(skb)))
+                       return -ENOMEM;
+
+               nd = (struct nd_msg *)skb_transport_header(skb);
+               ipv6_addr_copy(&key->nd_target, &nd->target);
+
+               icmp_len -= sizeof(*nd);
+               offset = 0;
+               while (icmp_len >= 8) {
+                       struct nd_opt_hdr *nd_opt = (struct nd_opt_hdr *)(nd->opt + offset);
+                       int opt_len = nd_opt->nd_opt_len * 8;
+
+                       if (unlikely(!opt_len || opt_len > icmp_len))
+                               goto invalid;
+
+                       /* Store the link layer address if the appropriate
+                        * option is provided.  It is considered an error if
+                        * the same link layer option is specified twice.
+                        */
+                       if (nd_opt->nd_opt_type == ND_OPT_SOURCE_LL_ADDR
+                           && opt_len == 8) {
+                               if (unlikely(!is_zero_ether_addr(key->arp_sha)))
+                                       goto invalid;
+                               memcpy(key->arp_sha,
+                                   &nd->opt[offset+sizeof(*nd_opt)], ETH_ALEN);
+                       } else if (nd_opt->nd_opt_type == ND_OPT_TARGET_LL_ADDR
+                                  && opt_len == 8) {
+                               if (unlikely(!is_zero_ether_addr(key->arp_tha)))
+                                       goto invalid;
+                               memcpy(key->arp_tha,
+                                   &nd->opt[offset+sizeof(*nd_opt)], ETH_ALEN);
+                       }
+
+                       icmp_len -= opt_len;
+                       offset += opt_len;
+               }
+       }
+
+       return 0;
+
+invalid:
+       memset(&key->nd_target, 0, sizeof(key->nd_target));
+       memset(key->arp_sha, 0, sizeof(key->arp_sha));
+       memset(key->arp_tha, 0, sizeof(key->arp_tha));
+
+       return 0;
+}
+
 /**
  * flow_extract - extracts a flow key from an Ethernet frame.
  * @skb: sk_buff that contains the frame, with skb->data pointing to the
@@ -261,10 +395,10 @@ static __be16 parse_ethertype(struct sk_buff *skb)
  *    - skb->network_header: just past the Ethernet header, or just past the
  *      VLAN header, to the first byte of the Ethernet payload.
  *
- *    - skb->transport_header: If key->dl_type is ETH_P_IP on output, then just
- *      past the IPv4 header, if one is present and of a correct length,
- *      otherwise the same as skb->network_header.  For other key->dl_type
- *      values it is left untouched.
+ *    - skb->transport_header: If key->dl_type is ETH_P_IP or ETH_P_IPV6
+ *      on output, then just past the IP header, if one is present and
+ *      of a correct length, otherwise the same as skb->network_header.
+ *      For other key->dl_type values it is left untouched.
  */
 int flow_extract(struct sk_buff *skb, u16 in_port, struct sw_flow_key *key,
                 bool *is_frag)
@@ -278,7 +412,8 @@ int flow_extract(struct sk_buff *skb, u16 in_port, struct sw_flow_key *key,
 
        /*
         * We would really like to pull as many bytes as we could possibly
-        * want to parse into the linear data area.  Currently that is:
+        * want to parse into the linear data area.  Currently, for IPv4,
+        * that is:
         *
         *    14     Ethernet header
         *     4     VLAN header
@@ -305,8 +440,12 @@ int flow_extract(struct sk_buff *skb, u16 in_port, struct sw_flow_key *key,
 
        /* dl_type, dl_vlan, dl_vlan_pcp. */
        __skb_pull(skb, 2 * ETH_ALEN);
-       if (eth->h_proto == htons(ETH_P_8021Q))
+
+       if (vlan_tx_tag_present(skb))
+               key->dl_tci = htons(vlan_get_tci(skb));
+       else if (eth->h_proto == htons(ETH_P_8021Q))
                parse_vlan(skb, key);
+
        key->dl_type = parse_ethertype(skb);
        skb_reset_network_header(skb);
        __skb_push(skb, skb->data - (unsigned char *)eth);
@@ -326,8 +465,8 @@ int flow_extract(struct sk_buff *skb, u16 in_port, struct sw_flow_key *key,
                }
 
                nh = ip_hdr(skb);
-               key->nw_src = nh->saddr;
-               key->nw_dst = nh->daddr;
+               key->ipv4_src = nh->saddr;
+               key->ipv4_dst = nh->daddr;
                key->nw_tos = nh->tos & ~INET_ECN_MASK;
                key->nw_proto = nh->protocol;
 
@@ -375,8 +514,42 @@ int flow_extract(struct sk_buff *skb, u16 in_port, struct sw_flow_key *key,
 
                        if (key->nw_proto == ARPOP_REQUEST
                                        || key->nw_proto == ARPOP_REPLY) {
-                               memcpy(&key->nw_src, arp->ar_sip, sizeof(key->nw_src));
-                               memcpy(&key->nw_dst, arp->ar_tip, sizeof(key->nw_dst));
+                               memcpy(&key->ipv4_src, arp->ar_sip, sizeof(key->ipv4_src));
+                               memcpy(&key->ipv4_dst, arp->ar_tip, sizeof(key->ipv4_dst));
+                               memcpy(key->arp_sha, arp->ar_sha, ETH_ALEN);
+                               memcpy(key->arp_tha, arp->ar_tha, ETH_ALEN);
+                       }
+               }
+       } else if (key->dl_type == htons(ETH_P_IPV6)) {
+               int nh_len;             /* IPv6 Header + Extensions */
+
+               nh_len = parse_ipv6hdr(skb, key);
+               if (unlikely(nh_len < 0)) {
+                       if (nh_len == -EINVAL) {
+                               skb->transport_header = skb->network_header;
+                               return 0;
+                       }
+                       return nh_len;
+               }
+
+               /* Transport layer. */
+               if (key->nw_proto == NEXTHDR_TCP) {
+                       if (tcphdr_ok(skb)) {
+                               struct tcphdr *tcp = tcp_hdr(skb);
+                               key->tp_src = tcp->source;
+                               key->tp_dst = tcp->dest;
+                       }
+               } else if (key->nw_proto == NEXTHDR_UDP) {
+                       if (udphdr_ok(skb)) {
+                               struct udphdr *udp = udp_hdr(skb);
+                               key->tp_src = udp->source;
+                               key->tp_dst = udp->dest;
+                       }
+               } else if (key->nw_proto == NEXTHDR_ICMP) {
+                       if (icmp6hdr_ok(skb)) {
+                               int error = parse_icmpv6(skb, key, nh_len);
+                               if (error < 0)
+                                       return error;
                        }
                }
        }
@@ -399,15 +572,16 @@ int flow_cmp(const struct tbl_node *node, void *key2_)
 /**
  * flow_from_nlattrs - parses Netlink attributes into a flow key.
  * @swkey: receives the extracted flow key.
- * @key: start of %ODP_KEY_ATTR_* Netlink attribute sequence.
- * @key_len: number of bytes in @key.
+ * @key: Netlink attribute holding nested %ODP_KEY_ATTR_* Netlink attribute
+ * sequence.
  *
  * This state machine accepts the following forms, with [] for optional
  * elements and | for alternatives:
  *
- * [tun_id] in_port ethernet [8021q] [ethertype [IP [TCP|UDP|ICMP] | ARP]
+ * [tun_id] in_port ethernet [8021q] [ethertype \
+ *              [IPv4 [TCP|UDP|ICMP] | IPv6 [TCP|UDP|ICMPv6 [ND]] | ARP]]
  */
-static int flow_from_nlattrs(struct sw_flow_key *swkey, const struct nlattr *key, u32 key_len)
+int flow_from_nlattrs(struct sw_flow_key *swkey, const struct nlattr *attr)
 {
        const struct nlattr *nla;
        u16 prev_type;
@@ -417,7 +591,7 @@ static int flow_from_nlattrs(struct sw_flow_key *swkey, const struct nlattr *key
        swkey->dl_type = htons(ETH_P_802_2);
 
        prev_type = ODP_KEY_ATTR_UNSPEC;
-       nla_for_each_attr(nla, key, key_len, rem) {
+       nla_for_each_nested(nla, attr, rem) {
                static const u32 key_lens[ODP_KEY_ATTR_MAX + 1] = {
                        [ODP_KEY_ATTR_TUN_ID] = 8,
                        [ODP_KEY_ATTR_IN_PORT] = 4,
@@ -425,19 +599,25 @@ static int flow_from_nlattrs(struct sw_flow_key *swkey, const struct nlattr *key
                        [ODP_KEY_ATTR_8021Q] = sizeof(struct odp_key_8021q),
                        [ODP_KEY_ATTR_ETHERTYPE] = 2,
                        [ODP_KEY_ATTR_IPV4] = sizeof(struct odp_key_ipv4),
+                       [ODP_KEY_ATTR_IPV6] = sizeof(struct odp_key_ipv6),
                        [ODP_KEY_ATTR_TCP] = sizeof(struct odp_key_tcp),
                        [ODP_KEY_ATTR_UDP] = sizeof(struct odp_key_udp),
                        [ODP_KEY_ATTR_ICMP] = sizeof(struct odp_key_icmp),
+                       [ODP_KEY_ATTR_ICMPV6] = sizeof(struct odp_key_icmpv6),
                        [ODP_KEY_ATTR_ARP] = sizeof(struct odp_key_arp),
+                       [ODP_KEY_ATTR_ND] = sizeof(struct odp_key_nd),
                };
 
                const struct odp_key_ethernet *eth_key;
                const struct odp_key_8021q *q_key;
                const struct odp_key_ipv4 *ipv4_key;
+               const struct odp_key_ipv6 *ipv6_key;
                const struct odp_key_tcp *tcp_key;
                const struct odp_key_udp *udp_key;
                const struct odp_key_icmp *icmp_key;
+               const struct odp_key_icmpv6 *icmpv6_key;
                const struct odp_key_arp *arp_key;
+               const struct odp_key_nd *nd_key;
 
                 int type = nla_type(nla);
 
@@ -484,15 +664,30 @@ static int flow_from_nlattrs(struct sw_flow_key *swkey, const struct nlattr *key
                        if (swkey->dl_type != htons(ETH_P_IP))
                                return -EINVAL;
                        ipv4_key = nla_data(nla);
-                       swkey->nw_src = ipv4_key->ipv4_src;
-                       swkey->nw_dst = ipv4_key->ipv4_dst;
+                       swkey->ipv4_src = ipv4_key->ipv4_src;
+                       swkey->ipv4_dst = ipv4_key->ipv4_dst;
                        swkey->nw_proto = ipv4_key->ipv4_proto;
                        swkey->nw_tos = ipv4_key->ipv4_tos;
                        if (swkey->nw_tos & INET_ECN_MASK)
                                return -EINVAL;
                        break;
 
+               case TRANSITION(ODP_KEY_ATTR_ETHERTYPE, ODP_KEY_ATTR_IPV6):
+                       if (swkey->dl_type != htons(ETH_P_IPV6))
+                               return -EINVAL;
+                       ipv6_key = nla_data(nla);
+                       memcpy(&swkey->ipv6_src, ipv6_key->ipv6_src,
+                                       sizeof(swkey->ipv6_src));
+                       memcpy(&swkey->ipv6_dst, ipv6_key->ipv6_dst,
+                                       sizeof(swkey->ipv6_dst));
+                       swkey->nw_proto = ipv6_key->ipv6_proto;
+                       swkey->nw_tos = ipv6_key->ipv6_tos;
+                       if (swkey->nw_tos & INET_ECN_MASK)
+                               return -EINVAL;
+                       break;
+
                case TRANSITION(ODP_KEY_ATTR_IPV4, ODP_KEY_ATTR_TCP):
+               case TRANSITION(ODP_KEY_ATTR_IPV6, ODP_KEY_ATTR_TCP):
                        if (swkey->nw_proto != IPPROTO_TCP)
                                return -EINVAL;
                        tcp_key = nla_data(nla);
@@ -501,6 +696,7 @@ static int flow_from_nlattrs(struct sw_flow_key *swkey, const struct nlattr *key
                        break;
 
                case TRANSITION(ODP_KEY_ATTR_IPV4, ODP_KEY_ATTR_UDP):
+               case TRANSITION(ODP_KEY_ATTR_IPV6, ODP_KEY_ATTR_UDP):
                        if (swkey->nw_proto != IPPROTO_UDP)
                                return -EINVAL;
                        udp_key = nla_data(nla);
@@ -516,15 +712,36 @@ static int flow_from_nlattrs(struct sw_flow_key *swkey, const struct nlattr *key
                        swkey->tp_dst = htons(icmp_key->icmp_code);
                        break;
 
+               case TRANSITION(ODP_KEY_ATTR_IPV6, ODP_KEY_ATTR_ICMPV6):
+                       if (swkey->nw_proto != IPPROTO_ICMPV6)
+                               return -EINVAL;
+                       icmpv6_key = nla_data(nla);
+                       swkey->tp_src = htons(icmpv6_key->icmpv6_type);
+                       swkey->tp_dst = htons(icmpv6_key->icmpv6_code);
+                       break;
+
                case TRANSITION(ODP_KEY_ATTR_ETHERTYPE, ODP_KEY_ATTR_ARP):
                        if (swkey->dl_type != htons(ETH_P_ARP))
                                return -EINVAL;
                        arp_key = nla_data(nla);
-                       swkey->nw_src = arp_key->arp_sip;
-                       swkey->nw_dst = arp_key->arp_tip;
+                       swkey->ipv4_src = arp_key->arp_sip;
+                       swkey->ipv4_dst = arp_key->arp_tip;
                        if (arp_key->arp_op & htons(0xff00))
                                return -EINVAL;
                        swkey->nw_proto = ntohs(arp_key->arp_op);
+                       memcpy(swkey->arp_sha, arp_key->arp_sha, ETH_ALEN);
+                       memcpy(swkey->arp_tha, arp_key->arp_tha, ETH_ALEN);
+                       break;
+
+               case TRANSITION(ODP_KEY_ATTR_ICMPV6, ODP_KEY_ATTR_ND):
+                       if (swkey->tp_src != htons(NDISC_NEIGHBOUR_SOLICITATION)
+                           && swkey->tp_src != htons(NDISC_NEIGHBOUR_ADVERTISEMENT))
+                               return -EINVAL;
+                       nd_key = nla_data(nla);
+                       memcpy(&swkey->nd_target, nd_key->nd_target,
+                                       sizeof(swkey->nd_target));
+                       memcpy(swkey->arp_sha, nd_key->nd_sll, ETH_ALEN);
+                       memcpy(swkey->arp_tha, nd_key->nd_tll, ETH_ALEN);
                        break;
 
                default:
@@ -561,10 +778,24 @@ static int flow_from_nlattrs(struct sw_flow_key *swkey, const struct nlattr *key
                        return -EINVAL;
                return 0;
 
+       case ODP_KEY_ATTR_IPV6:
+               if (swkey->nw_proto == IPPROTO_TCP ||
+                   swkey->nw_proto == IPPROTO_UDP ||
+                   swkey->nw_proto == IPPROTO_ICMPV6)
+                       return -EINVAL;
+               return 0;
+
+       case ODP_KEY_ATTR_ICMPV6:
+               if (swkey->tp_src == htons(NDISC_NEIGHBOUR_SOLICITATION) ||
+                   swkey->tp_src == htons(NDISC_NEIGHBOUR_ADVERTISEMENT))
+                       return -EINVAL;
+               return 0;
+
        case ODP_KEY_ATTR_TCP:
        case ODP_KEY_ATTR_UDP:
        case ODP_KEY_ATTR_ICMP:
        case ODP_KEY_ATTR_ARP:
+       case ODP_KEY_ATTR_ND:
                return 0;
        }
 
@@ -572,104 +803,145 @@ static int flow_from_nlattrs(struct sw_flow_key *swkey, const struct nlattr *key
        return -EINVAL;
 }
 
-static u32 flow_to_nlattrs(const struct sw_flow_key *swkey, struct sk_buff *skb)
+int flow_to_nlattrs(const struct sw_flow_key *swkey, struct sk_buff *skb)
 {
        struct odp_key_ethernet *eth_key;
+       struct nlattr *nla;
 
-       if (skb_tailroom(skb) < FLOW_BUFSIZE)
-               return -EMSGSIZE;
+       /* This is an imperfect sanity-check that FLOW_BUFSIZE doesn't need
+        * to be updated, but will at least raise awareness when new ODP key
+        * types are added. */
+       BUILD_BUG_ON(__ODP_KEY_ATTR_MAX != 14);
 
        if (swkey->tun_id != cpu_to_be64(0))
-               nla_put_be64(skb, ODP_KEY_ATTR_TUN_ID, swkey->tun_id);
+               NLA_PUT_BE64(skb, ODP_KEY_ATTR_TUN_ID, swkey->tun_id);
 
-       nla_put_u32(skb, ODP_KEY_ATTR_IN_PORT, swkey->in_port);
+       NLA_PUT_U32(skb, ODP_KEY_ATTR_IN_PORT, swkey->in_port);
 
-       eth_key = nla_data(__nla_reserve(skb, ODP_KEY_ATTR_ETHERNET, sizeof(*eth_key)));
+       nla = nla_reserve(skb, ODP_KEY_ATTR_ETHERNET, sizeof(*eth_key));
+       if (!nla)
+               goto nla_put_failure;
+       eth_key = nla_data(nla);
        memcpy(eth_key->eth_src, swkey->dl_src, ETH_ALEN);
        memcpy(eth_key->eth_dst, swkey->dl_dst, ETH_ALEN);
 
        if (swkey->dl_tci != htons(0)) {
-               struct odp_key_8021q *q_key;
+               struct odp_key_8021q q_key;
 
-               q_key = nla_data(__nla_reserve(skb, ODP_KEY_ATTR_8021Q, sizeof(*q_key)));
-               q_key->q_tpid = htons(ETH_P_8021Q);
-               q_key->q_tci = swkey->dl_tci & ~htons(VLAN_TAG_PRESENT);
+               q_key.q_tpid = htons(ETH_P_8021Q);
+               q_key.q_tci = swkey->dl_tci & ~htons(VLAN_TAG_PRESENT);
+               NLA_PUT(skb, ODP_KEY_ATTR_8021Q, sizeof(q_key), &q_key);
        }
 
        if (swkey->dl_type == htons(ETH_P_802_2))
-               goto exit;
+               return 0;
 
-       nla_put_be16(skb, ODP_KEY_ATTR_ETHERTYPE, swkey->dl_type);
+       NLA_PUT_BE16(skb, ODP_KEY_ATTR_ETHERTYPE, swkey->dl_type);
 
        if (swkey->dl_type == htons(ETH_P_IP)) {
                struct odp_key_ipv4 *ipv4_key;
 
-               ipv4_key = nla_data(__nla_reserve(skb, ODP_KEY_ATTR_IPV4, sizeof(*ipv4_key)));
-               ipv4_key->ipv4_src = swkey->nw_src;
-               ipv4_key->ipv4_dst = swkey->nw_dst;
+               nla = nla_reserve(skb, ODP_KEY_ATTR_IPV4, sizeof(*ipv4_key));
+               if (!nla)
+                       goto nla_put_failure;
+               ipv4_key = nla_data(nla);
+               memset(ipv4_key, 0, sizeof(struct odp_key_ipv4));
+               ipv4_key->ipv4_src = swkey->ipv4_src;
+               ipv4_key->ipv4_dst = swkey->ipv4_dst;
                ipv4_key->ipv4_proto = swkey->nw_proto;
                ipv4_key->ipv4_tos = swkey->nw_tos;
+       } else if (swkey->dl_type == htons(ETH_P_IPV6)) {
+               struct odp_key_ipv6 *ipv6_key;
+
+               nla = nla_reserve(skb, ODP_KEY_ATTR_IPV6, sizeof(*ipv6_key));
+               if (!nla)
+                       goto nla_put_failure;
+               ipv6_key = nla_data(nla);
+               memset(ipv6_key, 0, sizeof(struct odp_key_ipv6));
+               memcpy(ipv6_key->ipv6_src, &swkey->ipv6_src,
+                               sizeof(ipv6_key->ipv6_src));
+               memcpy(ipv6_key->ipv6_dst, &swkey->ipv6_dst,
+                               sizeof(ipv6_key->ipv6_dst));
+               ipv6_key->ipv6_proto = swkey->nw_proto;
+               ipv6_key->ipv6_tos = swkey->nw_tos;
+       } else if (swkey->dl_type == htons(ETH_P_ARP)) {
+               struct odp_key_arp *arp_key;
+
+               nla = nla_reserve(skb, ODP_KEY_ATTR_ARP, sizeof(*arp_key));
+               if (!nla)
+                       goto nla_put_failure;
+               arp_key = nla_data(nla);
+               memset(arp_key, 0, sizeof(struct odp_key_arp));
+               arp_key->arp_sip = swkey->ipv4_src;
+               arp_key->arp_tip = swkey->ipv4_dst;
+               arp_key->arp_op = htons(swkey->nw_proto);
+               memcpy(arp_key->arp_sha, swkey->arp_sha, ETH_ALEN);
+               memcpy(arp_key->arp_tha, swkey->arp_tha, ETH_ALEN);
+       }
+
+       if (swkey->dl_type == htons(ETH_P_IP) ||
+           swkey->dl_type == htons(ETH_P_IPV6)) {
 
                if (swkey->nw_proto == IPPROTO_TCP) {
                        struct odp_key_tcp *tcp_key;
 
-                       tcp_key = nla_data(__nla_reserve(skb, ODP_KEY_ATTR_TCP, sizeof(*tcp_key)));
+                       nla = nla_reserve(skb, ODP_KEY_ATTR_TCP, sizeof(*tcp_key));
+                       if (!nla)
+                               goto nla_put_failure;
+                       tcp_key = nla_data(nla);
                        tcp_key->tcp_src = swkey->tp_src;
                        tcp_key->tcp_dst = swkey->tp_dst;
                } else if (swkey->nw_proto == IPPROTO_UDP) {
                        struct odp_key_udp *udp_key;
 
-                       udp_key = nla_data(__nla_reserve(skb, ODP_KEY_ATTR_UDP, sizeof(*udp_key)));
+                       nla = nla_reserve(skb, ODP_KEY_ATTR_UDP, sizeof(*udp_key));
+                       if (!nla)
+                               goto nla_put_failure;
+                       udp_key = nla_data(nla);
                        udp_key->udp_src = swkey->tp_src;
                        udp_key->udp_dst = swkey->tp_dst;
-               } else if (swkey->nw_proto == IPPROTO_ICMP) {
+               } else if (swkey->dl_type == htons(ETH_P_IP) &&
+                          swkey->nw_proto == IPPROTO_ICMP) {
                        struct odp_key_icmp *icmp_key;
 
-                       icmp_key = nla_data(__nla_reserve(skb, ODP_KEY_ATTR_ICMP, sizeof(*icmp_key)));
+                       nla = nla_reserve(skb, ODP_KEY_ATTR_ICMP, sizeof(*icmp_key));
+                       if (!nla)
+                               goto nla_put_failure;
+                       icmp_key = nla_data(nla);
                        icmp_key->icmp_type = ntohs(swkey->tp_src);
                        icmp_key->icmp_code = ntohs(swkey->tp_dst);
+               } else if (swkey->dl_type == htons(ETH_P_IPV6) &&
+                          swkey->nw_proto == IPPROTO_ICMPV6) {
+                       struct odp_key_icmpv6 *icmpv6_key;
+
+                       nla = nla_reserve(skb, ODP_KEY_ATTR_ICMPV6,
+                                               sizeof(*icmpv6_key));
+                       if (!nla)
+                               goto nla_put_failure;
+                       icmpv6_key = nla_data(nla);
+                       icmpv6_key->icmpv6_type = ntohs(swkey->tp_src);
+                       icmpv6_key->icmpv6_code = ntohs(swkey->tp_dst);
+
+                       if (icmpv6_key->icmpv6_type == NDISC_NEIGHBOUR_SOLICITATION ||
+                           icmpv6_key->icmpv6_type == NDISC_NEIGHBOUR_ADVERTISEMENT) {
+                               struct odp_key_nd *nd_key;
+
+                               nla = nla_reserve(skb, ODP_KEY_ATTR_ND, sizeof(*nd_key));
+                               if (!nla)
+                                       goto nla_put_failure;
+                               nd_key = nla_data(nla);
+                               memcpy(nd_key->nd_target, &swkey->nd_target,
+                                                       sizeof(nd_key->nd_target));
+                               memcpy(nd_key->nd_sll, swkey->arp_sha, ETH_ALEN);
+                               memcpy(nd_key->nd_tll, swkey->arp_tha, ETH_ALEN);
+                       }
                }
-       } else if (swkey->dl_type == htons(ETH_P_ARP)) {
-               struct odp_key_arp *arp_key;
-
-               arp_key = nla_data(__nla_reserve(skb, ODP_KEY_ATTR_ARP, sizeof(*arp_key)));
-               arp_key->arp_sip = swkey->nw_src;
-               arp_key->arp_tip = swkey->nw_dst;
-               arp_key->arp_op = htons(swkey->nw_proto);
        }
 
-exit:
-       return skb->len;
-}
-
-int flow_copy_from_user(struct sw_flow_key *swkey, const struct nlattr __user *ukey, u32 ukey_len)
-{
-       char key[FLOW_BUFSIZE] __aligned(NLA_ALIGNTO);
-
-       if (ukey_len > FLOW_BUFSIZE || ukey_len % NLA_ALIGNTO)
-               return -EINVAL;
-
-       if (copy_from_user(key, ukey, ukey_len))
-               return -EFAULT;
-
-       return flow_from_nlattrs(swkey, (const struct nlattr *)key, ukey_len);
-}
-
-int flow_copy_to_user(struct nlattr __user *ukey, const struct sw_flow_key *swkey, u32 ukey_len)
-{
-       struct sk_buff *skb;
-       int retval;
-
-       skb = alloc_skb(FLOW_BUFSIZE, GFP_KERNEL);
-       if (!skb)
-               return -ENOMEM;
-
-       retval = flow_to_nlattrs(swkey, skb);
-       if (copy_to_user(ukey, skb->data, min(skb->len, ukey_len)))
-               retval = -EFAULT;
-       kfree_skb(skb);
+       return 0;
 
-       return retval;
+nla_put_failure:
+       return -EMSGSIZE;
 }
 
 /* Initializes the flow module.