#include <linux/netdevice.h>
 #include <linux/u64_stats_sync.h>
 #include <net/devlink.h>
+#include <net/udp_tunnel.h>
 #include <net/xdp.h>
 
 #define DRV_NAME       "netdevsim"
        struct {
                u32 inject_error;
                u32 sleep;
-               u32 ports[2][NSIM_UDP_TUNNEL_N_PORTS];
+               u32 __ports[2][NSIM_UDP_TUNNEL_N_PORTS];
+               u32 (*ports)[NSIM_UDP_TUNNEL_N_PORTS];
                struct debugfs_u32_array dfs_ports[2];
        } udp_ports;
 
        bool fail_trap_policer_set;
        bool fail_trap_policer_counter_get;
        struct {
+               struct udp_tunnel_nic_shared utn_shared;
+               u32 __ports[2][NSIM_UDP_TUNNEL_N_PORTS];
                bool sync_all;
                bool open_only;
                bool ipv4_only;
+               bool shared;
                u32 sleep;
        } udp_ports;
 };
 
        struct net_device *dev = file->private_data;
        struct netdevsim *ns = netdev_priv(dev);
 
-       memset(&ns->udp_ports.ports, 0, sizeof(ns->udp_ports.ports));
+       memset(ns->udp_ports.ports, 0, sizeof(ns->udp_ports.__ports));
        rtnl_lock();
        udp_tunnel_nic_reset_ntf(dev);
        rtnl_unlock();
        struct netdevsim *ns = netdev_priv(dev);
        struct udp_tunnel_nic_info *info;
 
+       if (nsim_dev->udp_ports.shared && nsim_dev->udp_ports.open_only) {
+               dev_err(&nsim_dev->nsim_bus_dev->dev,
+                       "shared can't be used in conjunction with open_only\n");
+               return -EINVAL;
+       }
+
+       if (!nsim_dev->udp_ports.shared)
+               ns->udp_ports.ports = ns->udp_ports.__ports;
+       else
+               ns->udp_ports.ports = nsim_dev->udp_ports.__ports;
+
        debugfs_create_u32("udp_ports_inject_error", 0600,
                           ns->nsim_dev_port->ddir,
                           &ns->udp_ports.inject_error);
                info->flags |= UDP_TUNNEL_NIC_INFO_OPEN_ONLY;
        if (nsim_dev->udp_ports.ipv4_only)
                info->flags |= UDP_TUNNEL_NIC_INFO_IPV4_ONLY;
+       if (nsim_dev->udp_ports.shared)
+               info->shared = &nsim_dev->udp_ports.utn_shared;
 
        dev->udp_tunnel_nic_info = info;
        return 0;
                            &nsim_dev->udp_ports.open_only);
        debugfs_create_bool("udp_ports_ipv4_only", 0600, nsim_dev->ddir,
                            &nsim_dev->udp_ports.ipv4_only);
+       debugfs_create_bool("udp_ports_shared", 0600, nsim_dev->ddir,
+                           &nsim_dev->udp_ports.shared);
        debugfs_create_u32("udp_ports_sleep", 0600, nsim_dev->ddir,
                           &nsim_dev->udp_ports.sleep);
 }