Drivers: hv: util: catch allocation errors

Catch allocation errors in hvutil_transport_send. Fixes: 14b50f8 ('Drivers: hv: util: introduce hv_utils_transport abstraction') Signed-off-by: Olaf Hering <[email protected]> Signed-off-by: K. Y. Srinivasan <[email protected]> Signed-off-by: Greg Kroah-Hartman <[email protected]>
iegor · Dec 15, 2015 · cdc0c0c · cdc0c0c
1 parent 6dfb867
commit cdc0c0c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git 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;