summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorPavel Begunkov <asml.silence@gmail.com>2023-01-22 10:24:20 -0700
committerGreg Kroah-Hartman <gregkh@linuxfoundation.org>2023-01-24 07:22:49 +0100
commit124fb13cc75777d3903a1ea6094498c3ad477ef3 (patch)
treed41b74d222dfad93f951ef19e7665d1d17393840
parente944f1e37b971cf96216fd842e1b0399d541a5d8 (diff)
io_uring: fix double poll leak on repolling
commit c0737fa9a5a5cf5a053bcc983f72d58919b997c6 upstream. We have re-polling for partial IO, so a request can be polled twice. If it used two poll entries the first time then on the second io_arm_poll_handler() it will find the old apoll entry and NULL kmalloc()'ed second entry, i.e. apoll->double_poll, so leaking it. Fixes: 10c873334feba ("io_uring: allow re-poll if we made progress") Signed-off-by: Pavel Begunkov <asml.silence@gmail.com> Link: https://lore.kernel.org/r/fee2452494222ecc7f1f88c8fb659baef971414a.1655852245.git.asml.silence@gmail.com Signed-off-by: Jens Axboe <axboe@kernel.dk> Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>
-rw-r--r--io_uring/io_uring.c6
1 files changed, 4 insertions, 2 deletions
diff --git a/io_uring/io_uring.c b/io_uring/io_uring.c
index d892b842b07c..e851c6855895 100644
--- a/io_uring/io_uring.c
+++ b/io_uring/io_uring.c
@@ -5896,10 +5896,12 @@ static int io_arm_poll_handler(struct io_kiocb *req)
mask |= POLLOUT | POLLWRNORM;
}
- if (req->flags & REQ_F_POLLED)
+ if (req->flags & REQ_F_POLLED) {
apoll = req->apoll;
- else
+ kfree(apoll->double_poll);
+ } else {
apoll = kmalloc(sizeof(*apoll), GFP_ATOMIC);
+ }
if (unlikely(!apoll))
return IO_APOLL_ABORTED;
apoll->double_poll = NULL;