Catch allocation errors in hvutil_transport_send.

Fixes: 14b50f80c32d ('Drivers: hv: util: introduce hv_utils_transport 
abstraction')

Signed-off-by: Olaf Hering <o...@aepfle.de>
---
 drivers/hv/hv_utils_transport.c | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/drivers/hv/hv_utils_transport.c b/drivers/hv/hv_utils_transport.c
index 6a9d80a..1505ee6 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;
--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to