1// SPDX-License-Identifier: GPL-2.0
2
3struct io_tctx_node {
4	struct list_head	ctx_node;
5	struct task_struct	*task;
6	struct io_ring_ctx	*ctx;
7};
8
9int io_uring_alloc_task_context(struct task_struct *task,
10				struct io_ring_ctx *ctx);
11void io_uring_del_tctx_node(unsigned long index);
12int __io_uring_add_tctx_node(struct io_ring_ctx *ctx);
13int __io_uring_add_tctx_node_from_submit(struct io_ring_ctx *ctx);
14void io_uring_clean_tctx(struct io_uring_task *tctx);
15
16void io_uring_unreg_ringfd(void);
17int io_ringfd_register(struct io_ring_ctx *ctx, void __user *__arg,
18		       unsigned nr_args);
19int io_ringfd_unregister(struct io_ring_ctx *ctx, void __user *__arg,
20			 unsigned nr_args);
21
22/*
23 * Note that this task has used io_uring. We use it for cancelation purposes.
24 */
25static inline int io_uring_add_tctx_node(struct io_ring_ctx *ctx)
26{
27	struct io_uring_task *tctx = current->io_uring;
28
29	if (likely(tctx && tctx->last == ctx))
30		return 0;
31
32	return __io_uring_add_tctx_node_from_submit(ctx);
33}
34