Commit 13bf43f5 authored by Pavel Begunkov's avatar Pavel Begunkov Committed by Jens Axboe

io_uring: introduce ctx to tctx back map

For each pair tcxt-ctx create an object and chain it into ctx, so we
have a way to traverse all tctx that are using current ctx. Preparation
patch, will be used later.
Signed-off-by: default avatarPavel Begunkov <asml.silence@gmail.com>
Signed-off-by: default avatarJens Axboe <axboe@kernel.dk>
parent 2941267b
...@@ -454,6 +454,7 @@ struct io_ring_ctx { ...@@ -454,6 +454,7 @@ struct io_ring_ctx {
/* Keep this last, we don't need it for the fast path */ /* Keep this last, we don't need it for the fast path */
struct work_struct exit_work; struct work_struct exit_work;
struct list_head tctx_list;
}; };
/* /*
...@@ -805,6 +806,13 @@ struct io_kiocb { ...@@ -805,6 +806,13 @@ struct io_kiocb {
struct io_wq_work work; struct io_wq_work work;
}; };
struct io_tctx_node {
struct list_head ctx_node;
struct task_struct *task;
struct file *file;
struct io_ring_ctx *ctx;
};
struct io_defer_entry { struct io_defer_entry {
struct list_head list; struct list_head list;
struct io_kiocb *req; struct io_kiocb *req;
...@@ -1144,6 +1152,7 @@ static struct io_ring_ctx *io_ring_ctx_alloc(struct io_uring_params *p) ...@@ -1144,6 +1152,7 @@ static struct io_ring_ctx *io_ring_ctx_alloc(struct io_uring_params *p)
INIT_LIST_HEAD(&ctx->rsrc_ref_list); INIT_LIST_HEAD(&ctx->rsrc_ref_list);
INIT_DELAYED_WORK(&ctx->rsrc_put_work, io_rsrc_put_work); INIT_DELAYED_WORK(&ctx->rsrc_put_work, io_rsrc_put_work);
init_llist_head(&ctx->rsrc_put_llist); init_llist_head(&ctx->rsrc_put_llist);
INIT_LIST_HEAD(&ctx->tctx_list);
INIT_LIST_HEAD(&ctx->submit_state.comp.free_list); INIT_LIST_HEAD(&ctx->submit_state.comp.free_list);
INIT_LIST_HEAD(&ctx->submit_state.comp.locked_free_list); INIT_LIST_HEAD(&ctx->submit_state.comp.locked_free_list);
return ctx; return ctx;
...@@ -8748,6 +8757,7 @@ static void io_uring_cancel_task_requests(struct io_ring_ctx *ctx, ...@@ -8748,6 +8757,7 @@ static void io_uring_cancel_task_requests(struct io_ring_ctx *ctx,
static int io_uring_add_task_file(struct io_ring_ctx *ctx, struct file *file) static int io_uring_add_task_file(struct io_ring_ctx *ctx, struct file *file)
{ {
struct io_uring_task *tctx = current->io_uring; struct io_uring_task *tctx = current->io_uring;
struct io_tctx_node *node;
int ret; int ret;
if (unlikely(!tctx)) { if (unlikely(!tctx)) {
...@@ -8760,13 +8770,25 @@ static int io_uring_add_task_file(struct io_ring_ctx *ctx, struct file *file) ...@@ -8760,13 +8770,25 @@ static int io_uring_add_task_file(struct io_ring_ctx *ctx, struct file *file)
void *old = xa_load(&tctx->xa, (unsigned long)file); void *old = xa_load(&tctx->xa, (unsigned long)file);
if (!old) { if (!old) {
node = kmalloc(sizeof(*node), GFP_KERNEL);
if (!node)
return -ENOMEM;
node->ctx = ctx;
node->file = file;
node->task = current;
get_file(file); get_file(file);
ret = xa_err(xa_store(&tctx->xa, (unsigned long)file, ret = xa_err(xa_store(&tctx->xa, (unsigned long)file,
file, GFP_KERNEL)); node, GFP_KERNEL));
if (ret) { if (ret) {
fput(file); fput(file);
kfree(node);
return ret; return ret;
} }
mutex_lock(&ctx->uring_lock);
list_add(&node->ctx_node, &ctx->tctx_list);
mutex_unlock(&ctx->uring_lock);
} }
tctx->last = file; tctx->last = file;
} }
...@@ -8788,23 +8810,31 @@ static int io_uring_add_task_file(struct io_ring_ctx *ctx, struct file *file) ...@@ -8788,23 +8810,31 @@ static int io_uring_add_task_file(struct io_ring_ctx *ctx, struct file *file)
static void io_uring_del_task_file(unsigned long index) static void io_uring_del_task_file(unsigned long index)
{ {
struct io_uring_task *tctx = current->io_uring; struct io_uring_task *tctx = current->io_uring;
struct file *file; struct io_tctx_node *node;
file = xa_erase(&tctx->xa, index); node = xa_erase(&tctx->xa, index);
if (!file) if (!node)
return; return;
if (tctx->last == file) WARN_ON_ONCE(current != node->task);
WARN_ON_ONCE(list_empty(&node->ctx_node));
mutex_lock(&node->ctx->uring_lock);
list_del(&node->ctx_node);
mutex_unlock(&node->ctx->uring_lock);
if (tctx->last == node->file)
tctx->last = NULL; tctx->last = NULL;
fput(file); fput(node->file);
kfree(node);
} }
static void io_uring_clean_tctx(struct io_uring_task *tctx) static void io_uring_clean_tctx(struct io_uring_task *tctx)
{ {
struct file *file; struct io_tctx_node *node;
unsigned long index; unsigned long index;
xa_for_each(&tctx->xa, index, file) xa_for_each(&tctx->xa, index, node)
io_uring_del_task_file(index); io_uring_del_task_file(index);
if (tctx->io_wq) { if (tctx->io_wq) {
io_wq_put_and_exit(tctx->io_wq); io_wq_put_and_exit(tctx->io_wq);
...@@ -8815,13 +8845,13 @@ static void io_uring_clean_tctx(struct io_uring_task *tctx) ...@@ -8815,13 +8845,13 @@ static void io_uring_clean_tctx(struct io_uring_task *tctx)
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;
struct file *file; struct io_tctx_node *node;
unsigned long index; unsigned long index;
/* 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, node)
io_uring_cancel_task_requests(file->private_data, files); io_uring_cancel_task_requests(node->ctx, files);
atomic_dec(&tctx->in_idle); atomic_dec(&tctx->in_idle);
if (files) if (files)
...@@ -8884,11 +8914,11 @@ void __io_uring_task_cancel(void) ...@@ -8884,11 +8914,11 @@ void __io_uring_task_cancel(void)
atomic_inc(&tctx->in_idle); atomic_inc(&tctx->in_idle);
if (tctx->sqpoll) { if (tctx->sqpoll) {
struct file *file; struct io_tctx_node *node;
unsigned long index; unsigned long index;
xa_for_each(&tctx->xa, index, file) xa_for_each(&tctx->xa, index, node)
io_uring_cancel_sqpoll(file->private_data); io_uring_cancel_sqpoll(node->ctx);
} }
do { do {
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment