RDMA/hns: Add illegal hop_num judgement

[ Upstream commit 26f63b9c33ceda12fb9136a1d0c80e03c9ebb514 ]

When hop_num is more than three, it need to return -EINVAL.  This patch
fixes it.

Signed-off-by: Lijun Ou <oulijun@huawei.com>
Signed-off-by: Jason Gunthorpe <jgg@mellanox.com>
Signed-off-by: Sasha Levin <alexander.levin@microsoft.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

diff --git a/drivers/infiniband/hw/hns/hns_roce_hem.c b/drivers/infiniband/hw/hns/hns_roce_hem.c
index 63b5b3edabcbad7b9ea6a002334a62433fef8a6a..8dc336a85128f4eccdb92115dcd8ea72a42cbdd3 100644 (file)
--- a/drivers/infiniband/hw/hns/hns_roce_hem.c
+++ b/drivers/infiniband/hw/hns/hns_roce_hem.c
@@ -494,6 +494,9 @@ static int hns_roce_table_mhop_get(struct hns_roce_dev *hr_dev,
                        step_idx = 1;
                } else if (hop_num == HNS_ROCE_HOP_NUM_0) {
                        step_idx = 0;
+               } else {
+                       ret = -EINVAL;
+                       goto err_dma_alloc_l1;
                }
 
                /* set HEM base address to hardware */