|
@@ -381,18 +381,12 @@ static int vmci_host_do_send_datagram(struct vmci_host_dev *vmci_host_dev,
|
|
|
return -EINVAL;
|
|
|
}
|
|
|
|
|
|
- dg = kmalloc(send_info.len, GFP_KERNEL);
|
|
|
- if (!dg) {
|
|
|
+ dg = memdup_user((void __user *)(uintptr_t)send_info.addr,
|
|
|
+ send_info.len);
|
|
|
+ if (IS_ERR(dg)) {
|
|
|
vmci_ioctl_err(
|
|
|
"cannot allocate memory to dispatch datagram\n");
|
|
|
- return -ENOMEM;
|
|
|
- }
|
|
|
-
|
|
|
- if (copy_from_user(dg, (void __user *)(uintptr_t)send_info.addr,
|
|
|
- send_info.len)) {
|
|
|
- vmci_ioctl_err("error getting datagram\n");
|
|
|
- kfree(dg);
|
|
|
- return -EFAULT;
|
|
|
+ return PTR_ERR(dg);
|
|
|
}
|
|
|
|
|
|
if (VMCI_DG_SIZE(dg) != send_info.len) {
|