diff options
| author | Artemy Kovalyov <artemyko@mellanox.com> | 2017-01-02 11:37:48 +0200 |
|---|---|---|
| committer | David S. Miller <davem@davemloft.net> | 2017-01-02 15:51:20 -0500 |
| commit | aa8e08d2f523501c40b0e70f1c4ecacb97195931 (patch) | |
| tree | 2e7e1a80aa9aaaf328e1b1c64821bc76e27a487d /drivers/infiniband/hw/mlx5/odp.c | |
| parent | 17d2f88f92ce39b348f125f6b2e6eeb6b0906ac7 (diff) | |
| download | lwn-aa8e08d2f523501c40b0e70f1c4ecacb97195931.tar.gz lwn-aa8e08d2f523501c40b0e70f1c4ecacb97195931.zip | |
IB/mlx5: Improve MR check
Add "type" field to mlx5_core MKEY struct.
Check whether page fault happens on MKEY corresponding to MR.
Signed-off-by: Artemy Kovalyov <artemyko@mellanox.com>
Signed-off-by: Leon Romanovsky <leon@kernel.org>
Signed-off-by: Saeed Mahameed <saeedm@mellanox.com>
Signed-off-by: David S. Miller <davem@davemloft.net>
Diffstat (limited to 'drivers/infiniband/hw/mlx5/odp.c')
| -rw-r--r-- | drivers/infiniband/hw/mlx5/odp.c | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/drivers/infiniband/hw/mlx5/odp.c b/drivers/infiniband/hw/mlx5/odp.c index 971b2885f474..e5bc267aca73 100644 --- a/drivers/infiniband/hw/mlx5/odp.c +++ b/drivers/infiniband/hw/mlx5/odp.c @@ -155,9 +155,14 @@ static struct mlx5_ib_mr *mlx5_ib_odp_find_mr_lkey(struct mlx5_ib_dev *dev, { u32 base_key = mlx5_base_mkey(key); struct mlx5_core_mkey *mmkey = __mlx5_mr_lookup(dev->mdev, base_key); - struct mlx5_ib_mr *mr = container_of(mmkey, struct mlx5_ib_mr, mmkey); + struct mlx5_ib_mr *mr; + + if (!mmkey || mmkey->key != key || mmkey->type != MLX5_MKEY_MR) + return NULL; + + mr = container_of(mmkey, struct mlx5_ib_mr, mmkey); - if (!mmkey || mmkey->key != key || !mr->live) + if (!mr->live) return NULL; return container_of(mmkey, struct mlx5_ib_mr, mmkey); |