net/mlx5: Check device memory pointer before usage

[ Upstream commit 70f238c902b8c0461ae6fbb8d1a0bbddc4350eea ]

Add a NULL check before accessing device memory to prevent a crash if
dev->dm allocation in mlx5_init_once() fails.

Fixes: c9b9dcb430b3 ("net/mlx5: Move device memory management to mlx5_core")
Signed-off-by: Stav Aviram <saviram@nvidia.com>
Link: https://patch.msgid.link/c88711327f4d74d5cebc730dc629607e989ca187.1751370035.git.leon@kernel.org
Signed-off-by: Leon Romanovsky <leon@kernel.org>
Signed-off-by: Sasha Levin <sashal@kernel.org>

diff --git a/drivers/infiniband/hw/mlx5/dm.c b/drivers/infiniband/hw/mlx5/dm.c
index b4c97fb62abfccbce92a926bf8dd64bbab64c9f1..9ded2b7c1e319969371fcbf3696b68188e9547c0 100644 (file)
--- a/drivers/infiniband/hw/mlx5/dm.c
+++ b/drivers/infiniband/hw/mlx5/dm.c
@@ -282,7 +282,7 @@ static struct ib_dm *handle_alloc_dm_memic(struct ib_ucontext *ctx,
        int err;
        u64 address;
 
-       if (!MLX5_CAP_DEV_MEM(dm_db->dev, memic))
+       if (!dm_db || !MLX5_CAP_DEV_MEM(dm_db->dev, memic))
                return ERR_PTR(-EOPNOTSUPP);
 
        dm = kzalloc(sizeof(*dm), GFP_KERNEL);

diff --git a/drivers/net/ethernet/mellanox/mlx5/core/lib/dm.c b/drivers/net/ethernet/mellanox/mlx5/core/lib/dm.c
index 7c5516b0a844948402ec8c77b931f564ada940c0..8115071c34a4ae4d259affa6eee2dd8ec505df87 100644 (file)
--- a/drivers/net/ethernet/mellanox/mlx5/core/lib/dm.c
+++ b/drivers/net/ethernet/mellanox/mlx5/core/lib/dm.c
@@ -30,7 +30,7 @@ struct mlx5_dm *mlx5_dm_create(struct mlx5_core_dev *dev)
 
        dm = kzalloc(sizeof(*dm), GFP_KERNEL);
        if (!dm)
-               return ERR_PTR(-ENOMEM);
+               return NULL;
 
        spin_lock_init(&dm->lock);
 
@@ -96,7 +96,7 @@ err_modify_hdr:
 err_steering:
        kfree(dm);
 
-       return ERR_PTR(-ENOMEM);
+       return NULL;
 }
 
 void mlx5_dm_cleanup(struct mlx5_core_dev *dev)

diff --git a/drivers/net/ethernet/mellanox/mlx5/core/main.c b/drivers/net/ethernet/mellanox/mlx5/core/main.c
index 9c1504d29d34c35d480d8bd08c3ff58914ff6ca1..e7bcd0f0a70979b64114f3b070a86d2332df8a5b 100644 (file)
--- a/drivers/net/ethernet/mellanox/mlx5/core/main.c
+++ b/drivers/net/ethernet/mellanox/mlx5/core/main.c
@@ -1102,9 +1102,6 @@ static int mlx5_init_once(struct mlx5_core_dev *dev)
        }
 
        dev->dm = mlx5_dm_create(dev);
-       if (IS_ERR(dev->dm))
-               mlx5_core_warn(dev, "Failed to init device memory %ld\n", PTR_ERR(dev->dm));
-
        dev->tracer = mlx5_fw_tracer_create(dev);
        dev->hv_vhca = mlx5_hv_vhca_create(dev);
        dev->rsc_dump = mlx5_rsc_dump_create(dev);