mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/stable/linux.git
synced 2024-08-26 02:39:48 +00:00
![Leon Romanovsky](/assets/img/avatar_default.png)
Radix tree lookup can be performed without locking.
Fixes: 225c7b1fee
("IB/mlx4: Add a driver Mellanox ConnectX InfiniBand adapters")
Signed-off-by: Leon Romanovsky <leonro@mellanox.com>
Suggested-by: Sagi Grimberg <sagi@grimberg.me>
Signed-off-by: Tariq Toukan <tariqt@mellanox.com>
Signed-off-by: David S. Miller <davem@davemloft.net>
309 lines
8.3 KiB
C
309 lines
8.3 KiB
C
/*
|
|
* Copyright (c) 2006, 2007 Cisco Systems, Inc. All rights reserved.
|
|
* Copyright (c) 2007, 2008 Mellanox Technologies. All rights reserved.
|
|
*
|
|
* This software is available to you under a choice of one of two
|
|
* licenses. You may choose to be licensed under the terms of the GNU
|
|
* General Public License (GPL) Version 2, available from the file
|
|
* COPYING in the main directory of this source tree, or the
|
|
* OpenIB.org BSD license below:
|
|
*
|
|
* Redistribution and use in source and binary forms, with or
|
|
* without modification, are permitted provided that the following
|
|
* conditions are met:
|
|
*
|
|
* - Redistributions of source code must retain the above
|
|
* copyright notice, this list of conditions and the following
|
|
* disclaimer.
|
|
*
|
|
* - Redistributions in binary form must reproduce the above
|
|
* copyright notice, this list of conditions and the following
|
|
* disclaimer in the documentation and/or other materials
|
|
* provided with the distribution.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
|
|
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
|
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*/
|
|
|
|
|
|
#include <linux/mlx4/cmd.h>
|
|
#include <linux/mlx4/srq.h>
|
|
#include <linux/export.h>
|
|
#include <linux/gfp.h>
|
|
|
|
#include "mlx4.h"
|
|
#include "icm.h"
|
|
|
|
void mlx4_srq_event(struct mlx4_dev *dev, u32 srqn, int event_type)
|
|
{
|
|
struct mlx4_srq_table *srq_table = &mlx4_priv(dev)->srq_table;
|
|
struct mlx4_srq *srq;
|
|
|
|
rcu_read_lock();
|
|
srq = radix_tree_lookup(&srq_table->tree, srqn & (dev->caps.num_srqs - 1));
|
|
rcu_read_unlock();
|
|
if (srq)
|
|
atomic_inc(&srq->refcount);
|
|
else {
|
|
mlx4_warn(dev, "Async event for bogus SRQ %08x\n", srqn);
|
|
return;
|
|
}
|
|
|
|
srq->event(srq, event_type);
|
|
|
|
if (atomic_dec_and_test(&srq->refcount))
|
|
complete(&srq->free);
|
|
}
|
|
|
|
static int mlx4_SW2HW_SRQ(struct mlx4_dev *dev, struct mlx4_cmd_mailbox *mailbox,
|
|
int srq_num)
|
|
{
|
|
return mlx4_cmd(dev, mailbox->dma, srq_num, 0,
|
|
MLX4_CMD_SW2HW_SRQ, MLX4_CMD_TIME_CLASS_A,
|
|
MLX4_CMD_WRAPPED);
|
|
}
|
|
|
|
static int mlx4_HW2SW_SRQ(struct mlx4_dev *dev, struct mlx4_cmd_mailbox *mailbox,
|
|
int srq_num)
|
|
{
|
|
return mlx4_cmd_box(dev, 0, mailbox ? mailbox->dma : 0, srq_num,
|
|
mailbox ? 0 : 1, MLX4_CMD_HW2SW_SRQ,
|
|
MLX4_CMD_TIME_CLASS_A, MLX4_CMD_WRAPPED);
|
|
}
|
|
|
|
static int mlx4_ARM_SRQ(struct mlx4_dev *dev, int srq_num, int limit_watermark)
|
|
{
|
|
return mlx4_cmd(dev, limit_watermark, srq_num, 0, MLX4_CMD_ARM_SRQ,
|
|
MLX4_CMD_TIME_CLASS_B, MLX4_CMD_WRAPPED);
|
|
}
|
|
|
|
static int mlx4_QUERY_SRQ(struct mlx4_dev *dev, struct mlx4_cmd_mailbox *mailbox,
|
|
int srq_num)
|
|
{
|
|
return mlx4_cmd_box(dev, 0, mailbox->dma, srq_num, 0, MLX4_CMD_QUERY_SRQ,
|
|
MLX4_CMD_TIME_CLASS_A, MLX4_CMD_WRAPPED);
|
|
}
|
|
|
|
int __mlx4_srq_alloc_icm(struct mlx4_dev *dev, int *srqn)
|
|
{
|
|
struct mlx4_srq_table *srq_table = &mlx4_priv(dev)->srq_table;
|
|
int err;
|
|
|
|
|
|
*srqn = mlx4_bitmap_alloc(&srq_table->bitmap);
|
|
if (*srqn == -1)
|
|
return -ENOMEM;
|
|
|
|
err = mlx4_table_get(dev, &srq_table->table, *srqn, GFP_KERNEL);
|
|
if (err)
|
|
goto err_out;
|
|
|
|
err = mlx4_table_get(dev, &srq_table->cmpt_table, *srqn, GFP_KERNEL);
|
|
if (err)
|
|
goto err_put;
|
|
return 0;
|
|
|
|
err_put:
|
|
mlx4_table_put(dev, &srq_table->table, *srqn);
|
|
|
|
err_out:
|
|
mlx4_bitmap_free(&srq_table->bitmap, *srqn, MLX4_NO_RR);
|
|
return err;
|
|
}
|
|
|
|
static int mlx4_srq_alloc_icm(struct mlx4_dev *dev, int *srqn)
|
|
{
|
|
u64 out_param;
|
|
int err;
|
|
|
|
if (mlx4_is_mfunc(dev)) {
|
|
err = mlx4_cmd_imm(dev, 0, &out_param, RES_SRQ,
|
|
RES_OP_RESERVE_AND_MAP,
|
|
MLX4_CMD_ALLOC_RES,
|
|
MLX4_CMD_TIME_CLASS_A, MLX4_CMD_WRAPPED);
|
|
if (!err)
|
|
*srqn = get_param_l(&out_param);
|
|
|
|
return err;
|
|
}
|
|
return __mlx4_srq_alloc_icm(dev, srqn);
|
|
}
|
|
|
|
void __mlx4_srq_free_icm(struct mlx4_dev *dev, int srqn)
|
|
{
|
|
struct mlx4_srq_table *srq_table = &mlx4_priv(dev)->srq_table;
|
|
|
|
mlx4_table_put(dev, &srq_table->cmpt_table, srqn);
|
|
mlx4_table_put(dev, &srq_table->table, srqn);
|
|
mlx4_bitmap_free(&srq_table->bitmap, srqn, MLX4_NO_RR);
|
|
}
|
|
|
|
static void mlx4_srq_free_icm(struct mlx4_dev *dev, int srqn)
|
|
{
|
|
u64 in_param = 0;
|
|
|
|
if (mlx4_is_mfunc(dev)) {
|
|
set_param_l(&in_param, srqn);
|
|
if (mlx4_cmd(dev, in_param, RES_SRQ, RES_OP_RESERVE_AND_MAP,
|
|
MLX4_CMD_FREE_RES,
|
|
MLX4_CMD_TIME_CLASS_A, MLX4_CMD_WRAPPED))
|
|
mlx4_warn(dev, "Failed freeing cq:%d\n", srqn);
|
|
return;
|
|
}
|
|
__mlx4_srq_free_icm(dev, srqn);
|
|
}
|
|
|
|
int mlx4_srq_alloc(struct mlx4_dev *dev, u32 pdn, u32 cqn, u16 xrcd,
|
|
struct mlx4_mtt *mtt, u64 db_rec, struct mlx4_srq *srq)
|
|
{
|
|
struct mlx4_srq_table *srq_table = &mlx4_priv(dev)->srq_table;
|
|
struct mlx4_cmd_mailbox *mailbox;
|
|
struct mlx4_srq_context *srq_context;
|
|
u64 mtt_addr;
|
|
int err;
|
|
|
|
err = mlx4_srq_alloc_icm(dev, &srq->srqn);
|
|
if (err)
|
|
return err;
|
|
|
|
spin_lock_irq(&srq_table->lock);
|
|
err = radix_tree_insert(&srq_table->tree, srq->srqn, srq);
|
|
spin_unlock_irq(&srq_table->lock);
|
|
if (err)
|
|
goto err_icm;
|
|
|
|
mailbox = mlx4_alloc_cmd_mailbox(dev);
|
|
if (IS_ERR(mailbox)) {
|
|
err = PTR_ERR(mailbox);
|
|
goto err_radix;
|
|
}
|
|
|
|
srq_context = mailbox->buf;
|
|
srq_context->state_logsize_srqn = cpu_to_be32((ilog2(srq->max) << 24) |
|
|
srq->srqn);
|
|
srq_context->logstride = srq->wqe_shift - 4;
|
|
srq_context->xrcd = cpu_to_be16(xrcd);
|
|
srq_context->pg_offset_cqn = cpu_to_be32(cqn & 0xffffff);
|
|
srq_context->log_page_size = mtt->page_shift - MLX4_ICM_PAGE_SHIFT;
|
|
|
|
mtt_addr = mlx4_mtt_addr(dev, mtt);
|
|
srq_context->mtt_base_addr_h = mtt_addr >> 32;
|
|
srq_context->mtt_base_addr_l = cpu_to_be32(mtt_addr & 0xffffffff);
|
|
srq_context->pd = cpu_to_be32(pdn);
|
|
srq_context->db_rec_addr = cpu_to_be64(db_rec);
|
|
|
|
err = mlx4_SW2HW_SRQ(dev, mailbox, srq->srqn);
|
|
mlx4_free_cmd_mailbox(dev, mailbox);
|
|
if (err)
|
|
goto err_radix;
|
|
|
|
atomic_set(&srq->refcount, 1);
|
|
init_completion(&srq->free);
|
|
|
|
return 0;
|
|
|
|
err_radix:
|
|
spin_lock_irq(&srq_table->lock);
|
|
radix_tree_delete(&srq_table->tree, srq->srqn);
|
|
spin_unlock_irq(&srq_table->lock);
|
|
|
|
err_icm:
|
|
mlx4_srq_free_icm(dev, srq->srqn);
|
|
return err;
|
|
}
|
|
EXPORT_SYMBOL_GPL(mlx4_srq_alloc);
|
|
|
|
void mlx4_srq_free(struct mlx4_dev *dev, struct mlx4_srq *srq)
|
|
{
|
|
struct mlx4_srq_table *srq_table = &mlx4_priv(dev)->srq_table;
|
|
int err;
|
|
|
|
err = mlx4_HW2SW_SRQ(dev, NULL, srq->srqn);
|
|
if (err)
|
|
mlx4_warn(dev, "HW2SW_SRQ failed (%d) for SRQN %06x\n", err, srq->srqn);
|
|
|
|
spin_lock_irq(&srq_table->lock);
|
|
radix_tree_delete(&srq_table->tree, srq->srqn);
|
|
spin_unlock_irq(&srq_table->lock);
|
|
|
|
if (atomic_dec_and_test(&srq->refcount))
|
|
complete(&srq->free);
|
|
wait_for_completion(&srq->free);
|
|
|
|
mlx4_srq_free_icm(dev, srq->srqn);
|
|
}
|
|
EXPORT_SYMBOL_GPL(mlx4_srq_free);
|
|
|
|
int mlx4_srq_arm(struct mlx4_dev *dev, struct mlx4_srq *srq, int limit_watermark)
|
|
{
|
|
return mlx4_ARM_SRQ(dev, srq->srqn, limit_watermark);
|
|
}
|
|
EXPORT_SYMBOL_GPL(mlx4_srq_arm);
|
|
|
|
int mlx4_srq_query(struct mlx4_dev *dev, struct mlx4_srq *srq, int *limit_watermark)
|
|
{
|
|
struct mlx4_cmd_mailbox *mailbox;
|
|
struct mlx4_srq_context *srq_context;
|
|
int err;
|
|
|
|
mailbox = mlx4_alloc_cmd_mailbox(dev);
|
|
if (IS_ERR(mailbox))
|
|
return PTR_ERR(mailbox);
|
|
|
|
srq_context = mailbox->buf;
|
|
|
|
err = mlx4_QUERY_SRQ(dev, mailbox, srq->srqn);
|
|
if (err)
|
|
goto err_out;
|
|
*limit_watermark = be16_to_cpu(srq_context->limit_watermark);
|
|
|
|
err_out:
|
|
mlx4_free_cmd_mailbox(dev, mailbox);
|
|
return err;
|
|
}
|
|
EXPORT_SYMBOL_GPL(mlx4_srq_query);
|
|
|
|
int mlx4_init_srq_table(struct mlx4_dev *dev)
|
|
{
|
|
struct mlx4_srq_table *srq_table = &mlx4_priv(dev)->srq_table;
|
|
int err;
|
|
|
|
spin_lock_init(&srq_table->lock);
|
|
INIT_RADIX_TREE(&srq_table->tree, GFP_ATOMIC);
|
|
if (mlx4_is_slave(dev))
|
|
return 0;
|
|
|
|
err = mlx4_bitmap_init(&srq_table->bitmap, dev->caps.num_srqs,
|
|
dev->caps.num_srqs - 1, dev->caps.reserved_srqs, 0);
|
|
if (err)
|
|
return err;
|
|
|
|
return 0;
|
|
}
|
|
|
|
void mlx4_cleanup_srq_table(struct mlx4_dev *dev)
|
|
{
|
|
if (mlx4_is_slave(dev))
|
|
return;
|
|
mlx4_bitmap_cleanup(&mlx4_priv(dev)->srq_table.bitmap);
|
|
}
|
|
|
|
struct mlx4_srq *mlx4_srq_lookup(struct mlx4_dev *dev, u32 srqn)
|
|
{
|
|
struct mlx4_srq_table *srq_table = &mlx4_priv(dev)->srq_table;
|
|
struct mlx4_srq *srq;
|
|
|
|
rcu_read_lock();
|
|
srq = radix_tree_lookup(&srq_table->tree,
|
|
srqn & (dev->caps.num_srqs - 1));
|
|
rcu_read_unlock();
|
|
|
|
return srq;
|
|
}
|
|
EXPORT_SYMBOL_GPL(mlx4_srq_lookup);
|