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>

1 files changed, 6 insertions, 3 deletions

diff --git a/drivers/hv/hv_utils_transport.c b/drivers/hv/hv_utils_transport.c
index 6a9d80a5332d..1505ee6e6605 100644
--- 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;