drivers/net/ethernet/mellanox/mlx5/core/vxlan.c

   1 /*
   2  * Copyright (c) 2016, Mellanox Technologies, Ltd.  All rights reserved.
   3  *
   4  * This software is available to you under a choice of one of two
   5  * licenses.  You may choose to be licensed under the terms of the GNU
   6  * General Public License (GPL) Version 2, available from the file
   7  * COPYING in the main directory of this source tree, or the
   8  * OpenIB.org BSD license below:
   9  *
  10  *     Redistribution and use in source and binary forms, with or
  11  *     without modification, are permitted provided that the following
  12  *     conditions are met:
  13  *
  14  *      - Redistributions of source code must retain the above
  15  *        copyright notice, this list of conditions and the following
  16  *        disclaimer.
  17  *
  18  *      - Redistributions in binary form must reproduce the above
  19  *        copyright notice, this list of conditions and the following
  20  *        disclaimer in the documentation and/or other materials
  21  *        provided with the distribution.
  22  *
  23  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  24  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  25  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  26  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  27  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  28  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  29  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  30  * SOFTWARE.
  31  */
  32
  33 #include <linux/kernel.h>
  34 #include <linux/module.h>
  35 #include <linux/mlx5/driver.h>
  36 #include "mlx5_core.h"
  37 #include "vxlan.h"
  38
  39 void mlx5e_vxlan_init(struct mlx5e_priv *priv)
  40 {
  41         struct mlx5e_vxlan_db *vxlan_db = &priv->vxlan;
  42
  43         spin_lock_init(&vxlan_db->lock);
  44         INIT_RADIX_TREE(&vxlan_db->tree, GFP_ATOMIC);
  45 }
  46
  47 static int mlx5e_vxlan_core_add_port_cmd(struct mlx5_core_dev *mdev, u16 port)
  48 {
  49         u32 in[MLX5_ST_SZ_DW(add_vxlan_udp_dport_in)]   = {0};
  50         u32 out[MLX5_ST_SZ_DW(add_vxlan_udp_dport_out)] = {0};
  51
  52         MLX5_SET(add_vxlan_udp_dport_in, in, opcode,
  53                  MLX5_CMD_OP_ADD_VXLAN_UDP_DPORT);
  54         MLX5_SET(add_vxlan_udp_dport_in, in, vxlan_udp_port, port);
  55         return mlx5_cmd_exec(mdev, in, sizeof(in), out, sizeof(out));
  56 }
  57
  58 static int mlx5e_vxlan_core_del_port_cmd(struct mlx5_core_dev *mdev, u16 port)
  59 {
  60         u32 in[MLX5_ST_SZ_DW(delete_vxlan_udp_dport_in)]   = {0};
  61         u32 out[MLX5_ST_SZ_DW(delete_vxlan_udp_dport_out)] = {0};
  62
  63         MLX5_SET(delete_vxlan_udp_dport_in, in, opcode,
  64                  MLX5_CMD_OP_DELETE_VXLAN_UDP_DPORT);
  65         MLX5_SET(delete_vxlan_udp_dport_in, in, vxlan_udp_port, port);
  66         return mlx5_cmd_exec(mdev, in, sizeof(in), out, sizeof(out));
  67 }
  68
  69 struct mlx5e_vxlan *mlx5e_vxlan_lookup_port(struct mlx5e_priv *priv, u16 port)
  70 {
  71         struct mlx5e_vxlan_db *vxlan_db = &priv->vxlan;
  72         struct mlx5e_vxlan *vxlan;
  73
  74         spin_lock(&vxlan_db->lock);
  75         vxlan = radix_tree_lookup(&vxlan_db->tree, port);
  76         spin_unlock(&vxlan_db->lock);
  77
  78         return vxlan;
  79 }
  80
  81 static void mlx5e_vxlan_add_port(struct work_struct *work)
  82 {
  83         struct mlx5e_vxlan_work *vxlan_work =
  84                 container_of(work, struct mlx5e_vxlan_work, work);
  85         struct mlx5e_priv *priv = vxlan_work->priv;
  86         struct mlx5e_vxlan_db *vxlan_db = &priv->vxlan;
  87         u16 port = vxlan_work->port;
  88         struct mlx5e_vxlan *vxlan;
  89         int err;
  90
  91         if (mlx5e_vxlan_lookup_port(priv, port))
  92                 goto free_work;
  93
  94         if (mlx5e_vxlan_core_add_port_cmd(priv->mdev, port))
  95                 goto free_work;
  96
  97         vxlan = kzalloc(sizeof(*vxlan), GFP_KERNEL);
  98         if (!vxlan)
  99                 goto err_delete_port;
 100
 101         vxlan->udp_port = port;
 102
 103         spin_lock_irq(&vxlan_db->lock);
 104         err = radix_tree_insert(&vxlan_db->tree, vxlan->udp_port, vxlan);
 105         spin_unlock_irq(&vxlan_db->lock);
 106         if (err)
 107                 goto err_free;
 108
 109         goto free_work;
 110
 111 err_free:
 112         kfree(vxlan);
 113 err_delete_port:
 114         mlx5e_vxlan_core_del_port_cmd(priv->mdev, port);
 115 free_work:
 116         kfree(vxlan_work);
 117 }
 118
 119 static void __mlx5e_vxlan_core_del_port(struct mlx5e_priv *priv, u16 port)
 120 {
 121         struct mlx5e_vxlan_db *vxlan_db = &priv->vxlan;
 122         struct mlx5e_vxlan *vxlan;
 123
 124         spin_lock_irq(&vxlan_db->lock);
 125         vxlan = radix_tree_delete(&vxlan_db->tree, port);
 126         spin_unlock_irq(&vxlan_db->lock);
 127
 128         if (!vxlan)
 129                 return;
 130
 131         mlx5e_vxlan_core_del_port_cmd(priv->mdev, vxlan->udp_port);
 132
 133         kfree(vxlan);
 134 }
 135
 136 static void mlx5e_vxlan_del_port(struct work_struct *work)
 137 {
 138         struct mlx5e_vxlan_work *vxlan_work =
 139                 container_of(work, struct mlx5e_vxlan_work, work);
 140         struct mlx5e_priv *priv = vxlan_work->priv;
 141         u16 port = vxlan_work->port;
 142
 143         __mlx5e_vxlan_core_del_port(priv, port);
 144
 145         kfree(vxlan_work);
 146 }
 147
 148 void mlx5e_vxlan_queue_work(struct mlx5e_priv *priv, sa_family_t sa_family,
 149                             u16 port, int add)
 150 {
 151         struct mlx5e_vxlan_work *vxlan_work;
 152
 153         vxlan_work = kmalloc(sizeof(*vxlan_work), GFP_ATOMIC);
 154         if (!vxlan_work)
 155                 return;
 156
 157         if (add)
 158                 INIT_WORK(&vxlan_work->work, mlx5e_vxlan_add_port);
 159         else
 160                 INIT_WORK(&vxlan_work->work, mlx5e_vxlan_del_port);
 161
 162         vxlan_work->priv = priv;
 163         vxlan_work->port = port;
 164         vxlan_work->sa_family = sa_family;
 165         queue_work(priv->wq, &vxlan_work->work);
 166 }
 167
 168 void mlx5e_vxlan_cleanup(struct mlx5e_priv *priv)
 169 {
 170         struct mlx5e_vxlan_db *vxlan_db = &priv->vxlan;
 171         struct mlx5e_vxlan *vxlan;
 172         unsigned int port = 0;
 173
 174         spin_lock_irq(&vxlan_db->lock);
 175         while (radix_tree_gang_lookup(&vxlan_db->tree, (void **)&vxlan, port, 1)) {
 176                 port = vxlan->udp_port;
 177                 spin_unlock_irq(&vxlan_db->lock);
 178                 __mlx5e_vxlan_core_del_port(priv, (u16)port);
 179                 spin_lock_irq(&vxlan_db->lock);
 180         }
 181         spin_unlock_irq(&vxlan_db->lock);
 182 }