io_uring: drop file refs after task cancel

[ Upstream commit de7f1d9e99d8b99e4e494ad8fcd91f0c4c5c9357 ]

io_uring fds marked O_CLOEXEC and we explicitly cancel all requests
before going through exec, so we don't want to leave task's file
references to not our anymore io_uring instances.

Signed-off-by: Pavel Begunkov <asml.silence@gmail.com>
Signed-off-by: Jens Axboe <axboe@kernel.dk>
Signed-off-by: Sasha Levin <sashal@kernel.org>
This commit is contained in:
Pavel Begunkov 2021-01-04 20:43:29 +00:00 committed by Greg Kroah-Hartman
parent 501e1875da
commit 94dbb87fc0

View File

@ -8821,6 +8821,15 @@ static void io_uring_attempt_task_drop(struct file *file)
io_uring_del_task_file(file); io_uring_del_task_file(file);
} }
static void io_uring_remove_task_files(struct io_uring_task *tctx)
{
struct file *file;
unsigned long index;
xa_for_each(&tctx->xa, index, file)
io_uring_del_task_file(file);
}
void __io_uring_files_cancel(struct files_struct *files) void __io_uring_files_cancel(struct files_struct *files)
{ {
struct io_uring_task *tctx = current->io_uring; struct io_uring_task *tctx = current->io_uring;
@ -8829,16 +8838,12 @@ void __io_uring_files_cancel(struct files_struct *files)
/* make sure overflow events are dropped */ /* make sure overflow events are dropped */
atomic_inc(&tctx->in_idle); atomic_inc(&tctx->in_idle);
xa_for_each(&tctx->xa, index, file)
xa_for_each(&tctx->xa, index, file) { io_uring_cancel_task_requests(file->private_data, files);
struct io_ring_ctx *ctx = file->private_data;
io_uring_cancel_task_requests(ctx, files);
if (files)
io_uring_del_task_file(file);
}
atomic_dec(&tctx->in_idle); atomic_dec(&tctx->in_idle);
if (files)
io_uring_remove_task_files(tctx);
} }
static s64 tctx_inflight(struct io_uring_task *tctx) static s64 tctx_inflight(struct io_uring_task *tctx)
@ -8901,6 +8906,8 @@ void __io_uring_task_cancel(void)
finish_wait(&tctx->wait, &wait); finish_wait(&tctx->wait, &wait);
atomic_dec(&tctx->in_idle); atomic_dec(&tctx->in_idle);
io_uring_remove_task_files(tctx);
} }
static int io_uring_flush(struct file *file, void *data) static int io_uring_flush(struct file *file, void *data)