RDMA/nldev: Add checks for nla_nest_start() in fill_stat_counter_qps()

[ Upstream commit df579bdc5679557601955578cea4676bd532b00d ]

As the nla_nest_start() may fail with NULL returned, the return value needs
to be checked.

Fixes: f41bb7eb11d4 ("RDMA/netlink: Implement counter dumpit calback")
Signed-off-by: Yuan Can <yuancan@huawei.com>
Link: https://lore.kernel.org/r/20221126043410.85632-1-yuancan@huawei.com
Signed-off-by: Leon Romanovsky <leon@kernel.org>
Signed-off-by: Sasha Levin <sashal@kernel.org>

diff --git a/drivers/infiniband/core/nldev.c b/drivers/infiniband/core/nldev.c
index f1e0755cd56ecaf9ed5af13f50b339fb9bb0edbe..54d4e693e03bf4da0a881a2633fbbc052856e6ac 100644 (file)
--- a/drivers/infiniband/core/nldev.c
+++ b/drivers/infiniband/core/nldev.c
@@ -894,6 +894,8 @@ static int fill_stat_counter_qps(struct sk_buff *msg,
        int ret = 0;
 
        table_attr = nla_nest_start(msg, RDMA_NLDEV_ATTR_RES_QP);
+       if (!table_attr)
+               return -EMSGSIZE;
 
        rt = &counter->device->res[RDMA_RESTRACK_QP];
        xa_lock(&rt->xa);