1 /* SPDX-License-Identifier: GPL-2.0-or-later */
2 #ifndef _LINUX_IO_URING_H
3 #define _LINUX_IO_URING_H
5 #include <linux/sched.h>
6 #include <linux/xarray.h>
7 #include <uapi/linux/io_uring.h>
9 enum io_uring_cmd_flags {
10 IO_URING_F_COMPLETE_DEFER = 1,
11 IO_URING_F_UNLOCKED = 2,
12 /* the request is executed from poll, it should not be freed */
13 IO_URING_F_MULTISHOT = 4,
14 /* executed by io-wq */
16 /* int's last bit, sign checks are usually faster than a bit test */
17 IO_URING_F_NONBLOCK = INT_MIN,
19 /* ctx state flags, for URING_CMD */
20 IO_URING_F_SQE128 = (1 << 8),
21 IO_URING_F_CQE32 = (1 << 9),
22 IO_URING_F_IOPOLL = (1 << 10),
29 /* callback to defer completions to task context */
30 void (*task_work_cb)(struct io_uring_cmd *cmd, unsigned);
31 /* used for polled completion */
36 u8 pdu[32]; /* available inline for free use */
39 #if defined(CONFIG_IO_URING)
40 int io_uring_cmd_import_fixed(u64 ubuf, unsigned long len, int rw,
41 struct iov_iter *iter, void *ioucmd);
42 void io_uring_cmd_done(struct io_uring_cmd *cmd, ssize_t ret, ssize_t res2,
43 unsigned issue_flags);
44 void io_uring_cmd_complete_in_task(struct io_uring_cmd *ioucmd,
45 void (*task_work_cb)(struct io_uring_cmd *, unsigned));
46 struct sock *io_uring_get_socket(struct file *file);
47 void __io_uring_cancel(bool cancel_all);
48 void __io_uring_free(struct task_struct *tsk);
49 void io_uring_unreg_ringfd(void);
50 const char *io_uring_get_opcode(u8 opcode);
52 static inline void io_uring_files_cancel(void)
54 if (current->io_uring) {
55 io_uring_unreg_ringfd();
56 __io_uring_cancel(false);
59 static inline void io_uring_task_cancel(void)
61 if (current->io_uring)
62 __io_uring_cancel(true);
64 static inline void io_uring_free(struct task_struct *tsk)
70 static inline int io_uring_cmd_import_fixed(u64 ubuf, unsigned long len, int rw,
71 struct iov_iter *iter, void *ioucmd)
75 static inline void io_uring_cmd_done(struct io_uring_cmd *cmd, ssize_t ret,
76 ssize_t ret2, unsigned issue_flags)
79 static inline void io_uring_cmd_complete_in_task(struct io_uring_cmd *ioucmd,
80 void (*task_work_cb)(struct io_uring_cmd *, unsigned))
83 static inline struct sock *io_uring_get_socket(struct file *file)
87 static inline void io_uring_task_cancel(void)
90 static inline void io_uring_files_cancel(void)
93 static inline void io_uring_free(struct task_struct *tsk)
96 static inline const char *io_uring_get_opcode(u8 opcode)