devlink: Add missing check of nlmsg_put

nlmsg_put may fail, this fix add a check of its return value.

Signed-off-by: YueHaibing <yuehaibing@huawei.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

diff --git a/net/core/devlink.c b/net/core/devlink.c
index 60248a53c0ad08ff676d4dd497148c86204f2bc5..059575c85ddeb7d3da9a50233c63f524c222eef0 100644 (file)
--- a/net/core/devlink.c
+++ b/net/core/devlink.c
@@ -4086,6 +4086,9 @@ devlink_health_buffer_snd(struct genl_info *info,
                return -ENOMEM;
        nlh = nlmsg_put(skb, info->snd_portid, info->snd_seq,
                        NLMSG_DONE, 0, flags | NLM_F_MULTI);
+       if (!nlh)
+               goto nla_put_failure;
+
        err = genlmsg_reply(skb, info);
        if (err)
                return err;