return jhash_2words(key, vni, vxlan_salt) & (FDB_HASH_SIZE - 1);
 }
 
+static u32 fdb_head_index(struct vxlan_dev *vxlan, const u8 *mac, __be32 vni)
+{
+       if (vxlan->cfg.flags & VXLAN_F_COLLECT_METADATA)
+               return eth_vni_hash(mac, vni);
+       else
+               return eth_hash(mac);
+}
+
 /* Hash chain to use given mac address */
 static inline struct hlist_head *vxlan_fdb_head(struct vxlan_dev *vxlan,
                                                const u8 *mac, __be32 vni)
 {
-       if (vxlan->cfg.flags & VXLAN_F_COLLECT_METADATA)
-               return &vxlan->fdb_head[eth_vni_hash(mac, vni)];
-       else
-               return &vxlan->fdb_head[eth_hash(mac)];
+       return &vxlan->fdb_head[fdb_head_index(vxlan, mac, vni)];
 }
 
 /* Look up Ethernet address in forwarding table */
                return -EINVAL;
        vxlan = netdev_priv(dev);
 
-       spin_lock_bh(&vxlan->hash_lock);
        for (h = 0; h < FDB_HASH_SIZE; ++h) {
+               spin_lock_bh(&vxlan->hash_lock[h]);
                hlist_for_each_entry(f, &vxlan->fdb_head[h], hlist) {
                        if (f->vni == vni) {
                                list_for_each_entry(rdst, &f->remotes, list) {
                                                                  f, rdst,
                                                                  extack);
                                        if (rc)
-                                               goto out;
+                                               goto unlock;
                                }
                        }
                }
+               spin_unlock_bh(&vxlan->hash_lock[h]);
        }
+       return 0;
 
-out:
-       spin_unlock_bh(&vxlan->hash_lock);
+unlock:
+       spin_unlock_bh(&vxlan->hash_lock[h]);
        return rc;
 }
 EXPORT_SYMBOL_GPL(vxlan_fdb_replay);
                return;
        vxlan = netdev_priv(dev);
 
-       spin_lock_bh(&vxlan->hash_lock);
        for (h = 0; h < FDB_HASH_SIZE; ++h) {
+               spin_lock_bh(&vxlan->hash_lock[h]);
                hlist_for_each_entry(f, &vxlan->fdb_head[h], hlist)
                        if (f->vni == vni)
                                list_for_each_entry(rdst, &f->remotes, list)
                                        rdst->offloaded = false;
+               spin_unlock_bh(&vxlan->hash_lock[h]);
        }
-       spin_unlock_bh(&vxlan->hash_lock);
+
 }
 EXPORT_SYMBOL_GPL(vxlan_fdb_clear_offload);
 
        __be16 port;
        __be32 src_vni, vni;
        u32 ifindex;
+       u32 hash_index;
        int err;
 
        if (!(ndm->ndm_state & (NUD_PERMANENT|NUD_REACHABLE))) {
        if (vxlan->default_dst.remote_ip.sa.sa_family != ip.sa.sa_family)
                return -EAFNOSUPPORT;
 
-       spin_lock_bh(&vxlan->hash_lock);
+       hash_index = fdb_head_index(vxlan, addr, src_vni);
+       spin_lock_bh(&vxlan->hash_lock[hash_index]);
        err = vxlan_fdb_update(vxlan, addr, &ip, ndm->ndm_state, flags,
                               port, src_vni, vni, ifindex,
                               ndm->ndm_flags | NTF_VXLAN_ADDED_BY_USER,
                               true, extack);
-       spin_unlock_bh(&vxlan->hash_lock);
+       spin_unlock_bh(&vxlan->hash_lock[hash_index]);
 
        return err;
 }
        __be32 src_vni, vni;
        __be16 port;
        u32 ifindex;
+       u32 hash_index;
        int err;
 
        err = vxlan_fdb_parse(tb, vxlan, &ip, &port, &src_vni, &vni, &ifindex);
        if (err)
                return err;
 
-       spin_lock_bh(&vxlan->hash_lock);
+       hash_index = fdb_head_index(vxlan, addr, src_vni);
+       spin_lock_bh(&vxlan->hash_lock[hash_index]);
        err = __vxlan_fdb_delete(vxlan, addr, ip, port, src_vni, vni, ifindex,
                                 true);
-       spin_unlock_bh(&vxlan->hash_lock);
+       spin_unlock_bh(&vxlan->hash_lock[hash_index]);
 
        return err;
 }
                f->updated = jiffies;
                vxlan_fdb_notify(vxlan, f, rdst, RTM_NEWNEIGH, true, NULL);
        } else {
+               u32 hash_index = fdb_head_index(vxlan, src_mac, vni);
+
                /* learned new entry */
-               spin_lock(&vxlan->hash_lock);
+               spin_lock(&vxlan->hash_lock[hash_index]);
 
                /* close off race between vxlan_flush and incoming packets */
                if (netif_running(dev))
                                         vni,
                                         vxlan->default_dst.remote_vni,
                                         ifindex, NTF_SELF, true, NULL);
-               spin_unlock(&vxlan->hash_lock);
+               spin_unlock(&vxlan->hash_lock[hash_index]);
        }
 
        return false;
        for (h = 0; h < FDB_HASH_SIZE; ++h) {
                struct hlist_node *p, *n;
 
-               spin_lock(&vxlan->hash_lock);
+               spin_lock(&vxlan->hash_lock[h]);
                hlist_for_each_safe(p, n, &vxlan->fdb_head[h]) {
                        struct vxlan_fdb *f
                                = container_of(p, struct vxlan_fdb, hlist);
                        } else if (time_before(timeout, next_timer))
                                next_timer = timeout;
                }
-               spin_unlock(&vxlan->hash_lock);
+               spin_unlock(&vxlan->hash_lock[h]);
        }
 
        mod_timer(&vxlan->age_timer, next_timer);
 static void vxlan_fdb_delete_default(struct vxlan_dev *vxlan, __be32 vni)
 {
        struct vxlan_fdb *f;
+       u32 hash_index = fdb_head_index(vxlan, all_zeros_mac, vni);
 
-       spin_lock_bh(&vxlan->hash_lock);
+       spin_lock_bh(&vxlan->hash_lock[hash_index]);
        f = __vxlan_find_mac(vxlan, all_zeros_mac, vni);
        if (f)
                vxlan_fdb_destroy(vxlan, f, true, true);
-       spin_unlock_bh(&vxlan->hash_lock);
+       spin_unlock_bh(&vxlan->hash_lock[hash_index]);
 }
 
 static void vxlan_uninit(struct net_device *dev)
 {
        unsigned int h;
 
-       spin_lock_bh(&vxlan->hash_lock);
        for (h = 0; h < FDB_HASH_SIZE; ++h) {
                struct hlist_node *p, *n;
+
+               spin_lock_bh(&vxlan->hash_lock[h]);
                hlist_for_each_safe(p, n, &vxlan->fdb_head[h]) {
                        struct vxlan_fdb *f
                                = container_of(p, struct vxlan_fdb, hlist);
                        if (!is_zero_ether_addr(f->eth_addr))
                                vxlan_fdb_destroy(vxlan, f, true, true);
                }
+               spin_unlock_bh(&vxlan->hash_lock[h]);
        }
-       spin_unlock_bh(&vxlan->hash_lock);
 }
 
 /* Cleanup timer and forwarding table on shutdown */
        dev->max_mtu = ETH_MAX_MTU;
 
        INIT_LIST_HEAD(&vxlan->next);
-       spin_lock_init(&vxlan->hash_lock);
 
        timer_setup(&vxlan->age_timer, vxlan_cleanup, TIMER_DEFERRABLE);
 
 
        gro_cells_init(&vxlan->gro_cells, dev);
 
-       for (h = 0; h < FDB_HASH_SIZE; ++h)
+       for (h = 0; h < FDB_HASH_SIZE; ++h) {
+               spin_lock_init(&vxlan->hash_lock[h]);
                INIT_HLIST_HEAD(&vxlan->fdb_head[h]);
+       }
 }
 
 static void vxlan_ether_setup(struct net_device *dev)
 
        /* handle default dst entry */
        if (!vxlan_addr_equal(&conf.remote_ip, &dst->remote_ip)) {
-               spin_lock_bh(&vxlan->hash_lock);
+               u32 hash_index = fdb_head_index(vxlan, all_zeros_mac, conf.vni);
+
+               spin_lock_bh(&vxlan->hash_lock[hash_index]);
                if (!vxlan_addr_any(&conf.remote_ip)) {
                        err = vxlan_fdb_update(vxlan, all_zeros_mac,
                                               &conf.remote_ip,
                                               conf.remote_ifindex,
                                               NTF_SELF, true, extack);
                        if (err) {
-                               spin_unlock_bh(&vxlan->hash_lock);
+                               spin_unlock_bh(&vxlan->hash_lock[hash_index]);
                                return err;
                        }
                }
                                           dst->remote_vni,
                                           dst->remote_ifindex,
                                           true);
-               spin_unlock_bh(&vxlan->hash_lock);
+               spin_unlock_bh(&vxlan->hash_lock[hash_index]);
        }
 
        if (conf.age_interval != vxlan->cfg.age_interval)
        struct vxlan_dev *vxlan = netdev_priv(dev);
        struct vxlan_rdst *rdst;
        struct vxlan_fdb *f;
+       u32 hash_index;
+
+       hash_index = fdb_head_index(vxlan, fdb_info->eth_addr, fdb_info->vni);
 
-       spin_lock_bh(&vxlan->hash_lock);
+       spin_lock_bh(&vxlan->hash_lock[hash_index]);
 
        f = vxlan_find_mac(vxlan, fdb_info->eth_addr, fdb_info->vni);
        if (!f)
        rdst->offloaded = fdb_info->offloaded;
 
 out:
-       spin_unlock_bh(&vxlan->hash_lock);
+       spin_unlock_bh(&vxlan->hash_lock[hash_index]);
 }
 
 static int
 {
        struct vxlan_dev *vxlan = netdev_priv(dev);
        struct netlink_ext_ack *extack;
+       u32 hash_index;
        int err;
 
+       hash_index = fdb_head_index(vxlan, fdb_info->eth_addr, fdb_info->vni);
        extack = switchdev_notifier_info_to_extack(&fdb_info->info);
 
-       spin_lock_bh(&vxlan->hash_lock);
+       spin_lock_bh(&vxlan->hash_lock[hash_index]);
        err = vxlan_fdb_update(vxlan, fdb_info->eth_addr, &fdb_info->remote_ip,
                               NUD_REACHABLE,
                               NLM_F_CREATE | NLM_F_REPLACE,
                               fdb_info->remote_ifindex,
                               NTF_USE | NTF_SELF | NTF_EXT_LEARNED,
                               false, extack);
-       spin_unlock_bh(&vxlan->hash_lock);
+       spin_unlock_bh(&vxlan->hash_lock[hash_index]);
 
        return err;
 }
 {
        struct vxlan_dev *vxlan = netdev_priv(dev);
        struct vxlan_fdb *f;
+       u32 hash_index;
        int err = 0;
 
-       spin_lock_bh(&vxlan->hash_lock);
+       hash_index = fdb_head_index(vxlan, fdb_info->eth_addr, fdb_info->vni);
+       spin_lock_bh(&vxlan->hash_lock[hash_index]);
 
        f = vxlan_find_mac(vxlan, fdb_info->eth_addr, fdb_info->vni);
        if (!f)
                                         fdb_info->remote_ifindex,
                                         false);
 
-       spin_unlock_bh(&vxlan->hash_lock);
+       spin_unlock_bh(&vxlan->hash_lock[hash_index]);
 
        return err;
 }