diff options
| -rw-r--r-- | io_uring/register.c | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/io_uring/register.c b/io_uring/register.c index 9c31a8afb83d..f4c76db27683 100644 --- a/io_uring/register.c +++ b/io_uring/register.c @@ -47,13 +47,9 @@ static __cold int io_probe(struct io_ring_ctx *ctx, void __user *arg, nr_args = IORING_OP_LAST; size = struct_size(p, ops, nr_args); - p = kzalloc(size, GFP_KERNEL); - if (!p) - return -ENOMEM; - - ret = -EFAULT; - if (copy_from_user(p, arg, size)) - goto out; + p = memdup_user(arg, size); + if (IS_ERR(p)) + return PTR_ERR(p); ret = -EINVAL; if (memchr_inv(p, 0, size)) goto out; |