Drivers: hv: util: catch allocation errors

Catch allocation errors in hvutil_transport_send.

Fixes: 14b50f80c32d ('Drivers: hv: util: introduce hv_utils_transport abstraction')
Signed-off-by: Olaf Hering <olaf@aepfle.de>
Signed-off-by: K. Y. Srinivasan <kys@microsoft.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>
Orabug: 27426063
(cherry picked from commit cdc0c0c94e4e6dfa371d497a3130f83349b6ead6)
Signed-off-by: Jack Vogel <jack.vogel@oracle.com>
Reviewed-by: Ashok Vairavan <ashok.vairavan@oracle.com>

diff --git a/drivers/hv/hv_utils_transport.c b/drivers/hv/hv_utils_transport.c
index ea7ba5ef16a90c2f834378c7675193033309671c..ac5eae5eb76c1642b4dbae1b0a10a074d4e0f387 100644 (file)
--- a/drivers/hv/hv_utils_transport.c
+++ b/drivers/hv/hv_utils_transport.c
@@ -204,9 +204,12 @@ int hvutil_transport_send(struct hvutil_transport *hvt, void *msg, int len)
                goto out_unlock;
        }
        hvt->outmsg = kzalloc(len, GFP_KERNEL);
-       memcpy(hvt->outmsg, msg, len);
-       hvt->outmsg_len = len;
-       wake_up_interruptible(&hvt->outmsg_q);
+       if (hvt->outmsg) {
+               memcpy(hvt->outmsg, msg, len);
+               hvt->outmsg_len = len;
+               wake_up_interruptible(&hvt->outmsg_q);
+       } else
+               ret = -ENOMEM;
 out_unlock:
        mutex_unlock(&hvt->outmsg_lock);
        return ret;