mlx5: Fix cleanup flow when DMA mapping fails

If DMA mapping fails, the driver cleared the object that holds the
previously DMA mapped pages. Fix this by allocating a new object for
the command that reports back to firmware that pages can't be
supplied.

Signed-off-by: Eli Cohen <eli@mellanox.com>
Signed-off-by: Roland Dreier <roland@purestorage.com>

1 files changed, 13 insertions, 5 deletions

diff --git a/drivers/net/ethernet/mellanox/mlx5/core/pagealloc.c b/drivers/net/ethernet/mellanox/mlx5/core/pagealloc.c
index 7b12acf210f8..a0d0da35578c 100644
--- a/drivers/net/ethernet/mellanox/mlx5/core/pagealloc.c
+++ b/drivers/net/ethernet/mellanox/mlx5/core/pagealloc.c
@@ -181,6 +181,7 @@ static int give_pages(struct mlx5_core_dev *dev, u16 func_id, int npages,
 {
         struct mlx5_manage_pages_inbox *in;
         struct mlx5_manage_pages_outbox out;
+        struct mlx5_manage_pages_inbox *nin;
         struct page *page;
         int inlen;
         u64 addr;
@@ -247,13 +248,20 @@ static int give_pages(struct mlx5_core_dev *dev, u16 func_id, int npages,
 
 out_alloc:
         if (notify_fail) {
-                memset(in, 0, inlen);
+                nin = kzalloc(sizeof(*nin), GFP_KERNEL);
+                if (!nin) {
+                        mlx5_core_warn(dev, "allocation failed\n");
+                        goto unmap;
+                }
                 memset(&out, 0, sizeof(out));
-                in->hdr.opcode = cpu_to_be16(MLX5_CMD_OP_MANAGE_PAGES);
-                in->hdr.opmod = cpu_to_be16(MLX5_PAGES_CANT_GIVE);
-                if (mlx5_cmd_exec(dev, in, sizeof(*in), &out, sizeof(out)))
-                        mlx5_core_warn(dev, "\n");
+                nin->hdr.opcode = cpu_to_be16(MLX5_CMD_OP_MANAGE_PAGES);
+                nin->hdr.opmod = cpu_to_be16(MLX5_PAGES_CANT_GIVE);
+                if (mlx5_cmd_exec(dev, nin, sizeof(*nin), &out, sizeof(out)))
+                        mlx5_core_warn(dev, "page notify failed\n");
+                kfree(nin);
         }
+
+unmap:
         for (i--; i >= 0; i--) {
                 addr = be64_to_cpu(in->pas[i]);
                 page = remove_page(dev, addr);