2 files changed, 186 insertions, 45 deletions

diff --git a/include/uapi/linux/fou.h b/include/uapi/linux/fou.h
index c303588bb767..d2947c52dc67 100644
--- a/include/uapi/linux/fou.h
+++ b/include/uapi/linux/fou.h
@@ -25,6 +25,7 @@ enum {
         FOU_CMD_UNSPEC,
         FOU_CMD_ADD,
         FOU_CMD_DEL,
+        FOU_CMD_GET,
 
         __FOU_CMD_MAX,
 };
diff --git a/net/ipv4/fou.c b/net/ipv4/fou.c
index ff069f6597ac..263710259774 100644
--- a/net/ipv4/fou.c
+++ b/net/ipv4/fou.c
@@ -16,14 +16,12 @@
 #include <uapi/linux/fou.h>
 #include <uapi/linux/genetlink.h>
 
-static DEFINE_SPINLOCK(fou_lock);
-static LIST_HEAD(fou_list);
-
 struct fou {
         struct socket *sock;
         u8 protocol;
         u8 flags;
-        u16 port;
+        __be16 port;
+        u16 type;
         struct udp_offload udp_offloads;
         struct list_head list;
 };
@@ -37,6 +35,13 @@ struct fou_cfg {
         struct udp_port_cfg udp_config;
 };
 
+static unsigned int fou_net_id;
+
+struct fou_net {
+        struct list_head fou_list;
+        struct mutex fou_lock;
+};
+
 static inline struct fou *fou_from_sock(struct sock *sk)
 {
         return sk->sk_user_data;
@@ -387,20 +392,21 @@ out_unlock:
         return err;
 }
 
-static int fou_add_to_port_list(struct fou *fou)
+static int fou_add_to_port_list(struct net *net, struct fou *fou)
 {
+        struct fou_net *fn = net_generic(net, fou_net_id);
         struct fou *fout;
 
-        spin_lock(&fou_lock);
-        list_for_each_entry(fout, &fou_list, list) {
+        mutex_lock(&fn->fou_lock);
+        list_for_each_entry(fout, &fn->fou_list, list) {
                 if (fou->port == fout->port) {
-                        spin_unlock(&fou_lock);
+                        mutex_unlock(&fn->fou_lock);
                         return -EALREADY;
                 }
         }
 
-        list_add(&fou->list, &fou_list);
-        spin_unlock(&fou_lock);
+        list_add(&fou->list, &fn->fou_list);
+        mutex_unlock(&fn->fou_lock);
 
         return 0;
 }
@@ -410,14 +416,10 @@ static void fou_release(struct fou *fou)
         struct socket *sock = fou->sock;
         struct sock *sk = sock->sk;
 
-        udp_del_offload(&fou->udp_offloads);
-
+        if (sk->sk_family == AF_INET)
+                udp_del_offload(&fou->udp_offloads);
         list_del(&fou->list);
-
-        /* Remove hooks into tunnel socket */
-        sk->sk_user_data = NULL;
-
-        sock_release(sock);
+        udp_tunnel_sock_release(sock);
 
         kfree(fou);
 }
@@ -447,10 +449,10 @@ static int gue_encap_init(struct sock *sk, struct fou *fou, struct fou_cfg *cfg)
 static int fou_create(struct net *net, struct fou_cfg *cfg,
                       struct socket **sockp)
 {
-        struct fou *fou = NULL;
-        int err;
         struct socket *sock = NULL;
+        struct fou *fou = NULL;
         struct sock *sk;
+        int err;
 
         /* Open UDP socket */
         err = udp_sock_create(net, &cfg->udp_config, &sock);
@@ -486,6 +488,8 @@ static int fou_create(struct net *net, struct fou_cfg *cfg,
                 goto error;
         }
 
+        fou->type = cfg->type;
+
         udp_sk(sk)->encap_type = 1;
         udp_encap_enable();
 
@@ -502,7 +506,7 @@ static int fou_create(struct net *net, struct fou_cfg *cfg,
                         goto error;
         }
 
-        err = fou_add_to_port_list(fou);
+        err = fou_add_to_port_list(net, fou);
         if (err)
                 goto error;
 
@@ -514,27 +518,27 @@ static int fou_create(struct net *net, struct fou_cfg *cfg,
 error:
         kfree(fou);
         if (sock)
-                sock_release(sock);
+                udp_tunnel_sock_release(sock);
 
         return err;
 }
 
 static int fou_destroy(struct net *net, struct fou_cfg *cfg)
 {
-        struct fou *fou;
-        u16 port = cfg->udp_config.local_udp_port;
+        struct fou_net *fn = net_generic(net, fou_net_id);
+        __be16 port = cfg->udp_config.local_udp_port;
         int err = -EINVAL;
+        struct fou *fou;
 
-        spin_lock(&fou_lock);
-        list_for_each_entry(fou, &fou_list, list) {
+        mutex_lock(&fn->fou_lock);
+        list_for_each_entry(fou, &fn->fou_list, list) {
                 if (fou->port == port) {
-                        udp_del_offload(&fou->udp_offloads);
                         fou_release(fou);
                         err = 0;
                         break;
                 }
         }
-        spin_unlock(&fou_lock);
+        mutex_unlock(&fn->fou_lock);
 
         return err;
 }
@@ -573,7 +577,7 @@ static int parse_nl_config(struct genl_info *info,
         }
 
         if (info->attrs[FOU_ATTR_PORT]) {
-                u16 port = nla_get_u16(info->attrs[FOU_ATTR_PORT]);
+                __be16 port = nla_get_be16(info->attrs[FOU_ATTR_PORT]);
 
                 cfg->udp_config.local_udp_port = port;
         }
@@ -592,6 +596,7 @@ static int parse_nl_config(struct genl_info *info,
 
 static int fou_nl_cmd_add_port(struct sk_buff *skb, struct genl_info *info)
 {
+        struct net *net = genl_info_net(info);
         struct fou_cfg cfg;
         int err;
 
@@ -599,16 +604,120 @@ static int fou_nl_cmd_add_port(struct sk_buff *skb, struct genl_info *info)
         if (err)
                 return err;
 
-        return fou_create(&init_net, &cfg, NULL);
+        return fou_create(net, &cfg, NULL);