1ed29b0b4SJens Axboe // SPDX-License-Identifier: GPL-2.0 2ed29b0b4SJens Axboe /* 3ed29b0b4SJens Axboe * Shared application/kernel submission and completion ring pairs, for 4ed29b0b4SJens Axboe * supporting fast/efficient IO. 5ed29b0b4SJens Axboe * 6ed29b0b4SJens Axboe * A note on the read/write ordering memory barriers that are matched between 7ed29b0b4SJens Axboe * the application and kernel side. 8ed29b0b4SJens Axboe * 9ed29b0b4SJens Axboe * After the application reads the CQ ring tail, it must use an 10ed29b0b4SJens Axboe * appropriate smp_rmb() to pair with the smp_wmb() the kernel uses 11ed29b0b4SJens Axboe * before writing the tail (using smp_load_acquire to read the tail will 12ed29b0b4SJens Axboe * do). It also needs a smp_mb() before updating CQ head (ordering the 13ed29b0b4SJens Axboe * entry load(s) with the head store), pairing with an implicit barrier 14ed29b0b4SJens Axboe * through a control-dependency in io_get_cqe (smp_store_release to 15ed29b0b4SJens Axboe * store head will do). Failure to do so could lead to reading invalid 16ed29b0b4SJens Axboe * CQ entries. 17ed29b0b4SJens Axboe * 18ed29b0b4SJens Axboe * Likewise, the application must use an appropriate smp_wmb() before 19ed29b0b4SJens Axboe * writing the SQ tail (ordering SQ entry stores with the tail store), 20ed29b0b4SJens Axboe * which pairs with smp_load_acquire in io_get_sqring (smp_store_release 21ed29b0b4SJens Axboe * to store the tail will do). And it needs a barrier ordering the SQ 22ed29b0b4SJens Axboe * head load before writing new SQ entries (smp_load_acquire to read 23ed29b0b4SJens Axboe * head will do). 24ed29b0b4SJens Axboe * 25ed29b0b4SJens Axboe * When using the SQ poll thread (IORING_SETUP_SQPOLL), the application 26ed29b0b4SJens Axboe * needs to check the SQ flags for IORING_SQ_NEED_WAKEUP *after* 27ed29b0b4SJens Axboe * updating the SQ tail; a full memory barrier smp_mb() is needed 28ed29b0b4SJens Axboe * between. 29ed29b0b4SJens Axboe * 30ed29b0b4SJens Axboe * Also see the examples in the liburing library: 31ed29b0b4SJens Axboe * 32ed29b0b4SJens Axboe * git://git.kernel.dk/liburing 33ed29b0b4SJens Axboe * 34ed29b0b4SJens Axboe * io_uring also uses READ/WRITE_ONCE() for _any_ store or load that happens 35ed29b0b4SJens Axboe * from data shared between the kernel and application. This is done both 36ed29b0b4SJens Axboe * for ordering purposes, but also to ensure that once a value is loaded from 37ed29b0b4SJens Axboe * data that the application could potentially modify, it remains stable. 38ed29b0b4SJens Axboe * 39ed29b0b4SJens Axboe * Copyright (C) 2018-2019 Jens Axboe 40ed29b0b4SJens Axboe * Copyright (c) 2018-2019 Christoph Hellwig 41ed29b0b4SJens Axboe */ 42ed29b0b4SJens Axboe #include <linux/kernel.h> 43ed29b0b4SJens Axboe #include <linux/init.h> 44ed29b0b4SJens Axboe #include <linux/errno.h> 45ed29b0b4SJens Axboe #include <linux/syscalls.h> 46ed29b0b4SJens Axboe #include <linux/compat.h> 47ed29b0b4SJens Axboe #include <net/compat.h> 48ed29b0b4SJens Axboe #include <linux/refcount.h> 49ed29b0b4SJens Axboe #include <linux/uio.h> 50ed29b0b4SJens Axboe #include <linux/bits.h> 51ed29b0b4SJens Axboe 52ed29b0b4SJens Axboe #include <linux/sched/signal.h> 53ed29b0b4SJens Axboe #include <linux/fs.h> 54ed29b0b4SJens Axboe #include <linux/file.h> 55ed29b0b4SJens Axboe #include <linux/fdtable.h> 56ed29b0b4SJens Axboe #include <linux/mm.h> 57ed29b0b4SJens Axboe #include <linux/mman.h> 58ed29b0b4SJens Axboe #include <linux/percpu.h> 59ed29b0b4SJens Axboe #include <linux/slab.h> 60ed29b0b4SJens Axboe #include <linux/blk-mq.h> 61ed29b0b4SJens Axboe #include <linux/bvec.h> 62ed29b0b4SJens Axboe #include <linux/net.h> 63ed29b0b4SJens Axboe #include <net/sock.h> 64ed29b0b4SJens Axboe #include <net/af_unix.h> 65ed29b0b4SJens Axboe #include <net/scm.h> 66ed29b0b4SJens Axboe #include <linux/anon_inodes.h> 67ed29b0b4SJens Axboe #include <linux/sched/mm.h> 68ed29b0b4SJens Axboe #include <linux/uaccess.h> 69ed29b0b4SJens Axboe #include <linux/nospec.h> 70ed29b0b4SJens Axboe #include <linux/sizes.h> 71ed29b0b4SJens Axboe #include <linux/hugetlb.h> 72ed29b0b4SJens Axboe #include <linux/highmem.h> 73ed29b0b4SJens Axboe #include <linux/fsnotify.h> 74ed29b0b4SJens Axboe #include <linux/fadvise.h> 75ed29b0b4SJens Axboe #include <linux/eventpoll.h> 76ed29b0b4SJens Axboe #include <linux/task_work.h> 77ed29b0b4SJens Axboe #include <linux/pagemap.h> 78ed29b0b4SJens Axboe #include <linux/io_uring.h> 79ed29b0b4SJens Axboe #include <linux/audit.h> 80ed29b0b4SJens Axboe #include <linux/security.h> 81ed29b0b4SJens Axboe 82ed29b0b4SJens Axboe #define CREATE_TRACE_POINTS 83ed29b0b4SJens Axboe #include <trace/events/io_uring.h> 84ed29b0b4SJens Axboe 85ed29b0b4SJens Axboe #include <uapi/linux/io_uring.h> 86ed29b0b4SJens Axboe 87ed29b0b4SJens Axboe #include "io-wq.h" 88ed29b0b4SJens Axboe 89e27f928eSJens Axboe #include "io_uring_types.h" 90de23077eSJens Axboe #include "io_uring.h" 91e418bbc9SJens Axboe #include "refs.h" 92c9f06aa7SJens Axboe #include "tctx.h" 9317437f31SJens Axboe #include "sqpoll.h" 94a4ad4f74SJens Axboe #include "fdinfo.h" 95e27f928eSJens Axboe 965e2a18d9SJens Axboe #include "xattr.h" 97e28683bdSJens Axboe #include "nop.h" 9811aeb714SJens Axboe #include "fs.h" 99531113bbSJens Axboe #include "splice.h" 1000d584727SJens Axboe #include "sync.h" 101f4c163ddSJens Axboe #include "advise.h" 102cd40cae2SJens Axboe #include "openclose.h" 10399f15d8dSJens Axboe #include "uring_cmd.h" 104a9c210ceSJens Axboe #include "epoll.h" 105e0da14deSJens Axboe #include "statx.h" 106f9ead18cSJens Axboe #include "net.h" 10736404b09SJens Axboe #include "msg_ring.h" 10859915143SJens Axboe #include "timeout.h" 1095e2a18d9SJens Axboe 110ed29b0b4SJens Axboe #define IORING_MAX_ENTRIES 32768 111ed29b0b4SJens Axboe #define IORING_MAX_CQ_ENTRIES (2 * IORING_MAX_ENTRIES) 112ed29b0b4SJens Axboe 113ed29b0b4SJens Axboe /* only define max */ 114ed29b0b4SJens Axboe #define IORING_MAX_FIXED_FILES (1U << 20) 115ed29b0b4SJens Axboe #define IORING_MAX_RESTRICTIONS (IORING_RESTRICTION_LAST + \ 116ed29b0b4SJens Axboe IORING_REGISTER_LAST + IORING_OP_LAST) 117ed29b0b4SJens Axboe 118ed29b0b4SJens Axboe #define IO_RSRC_TAG_TABLE_SHIFT (PAGE_SHIFT - 3) 119ed29b0b4SJens Axboe #define IO_RSRC_TAG_TABLE_MAX (1U << IO_RSRC_TAG_TABLE_SHIFT) 120ed29b0b4SJens Axboe #define IO_RSRC_TAG_TABLE_MASK (IO_RSRC_TAG_TABLE_MAX - 1) 121ed29b0b4SJens Axboe 122ed29b0b4SJens Axboe #define IORING_MAX_REG_BUFFERS (1U << 14) 123ed29b0b4SJens Axboe 124ed29b0b4SJens Axboe #define SQE_COMMON_FLAGS (IOSQE_FIXED_FILE | IOSQE_IO_LINK | \ 125ed29b0b4SJens Axboe IOSQE_IO_HARDLINK | IOSQE_ASYNC) 126ed29b0b4SJens Axboe 127ed29b0b4SJens Axboe #define SQE_VALID_FLAGS (SQE_COMMON_FLAGS | IOSQE_BUFFER_SELECT | \ 128ed29b0b4SJens Axboe IOSQE_IO_DRAIN | IOSQE_CQE_SKIP_SUCCESS) 129ed29b0b4SJens Axboe 130ed29b0b4SJens Axboe #define IO_REQ_CLEAN_FLAGS (REQ_F_BUFFER_SELECTED | REQ_F_NEED_CLEANUP | \ 131ed29b0b4SJens Axboe REQ_F_POLLED | REQ_F_INFLIGHT | REQ_F_CREDS | \ 132ed29b0b4SJens Axboe REQ_F_ASYNC_DATA) 133ed29b0b4SJens Axboe 134ed29b0b4SJens Axboe #define IO_REQ_CLEAN_SLOW_FLAGS (REQ_F_REFCOUNT | REQ_F_LINK | REQ_F_HARDLINK |\ 135ed29b0b4SJens Axboe IO_REQ_CLEAN_FLAGS) 136ed29b0b4SJens Axboe 137ed29b0b4SJens Axboe #define IO_TCTX_REFS_CACHE_NR (1U << 10) 138ed29b0b4SJens Axboe 139ed29b0b4SJens Axboe struct io_rsrc_put { 140ed29b0b4SJens Axboe struct list_head list; 141ed29b0b4SJens Axboe u64 tag; 142ed29b0b4SJens Axboe union { 143ed29b0b4SJens Axboe void *rsrc; 144ed29b0b4SJens Axboe struct file *file; 145ed29b0b4SJens Axboe struct io_mapped_ubuf *buf; 146ed29b0b4SJens Axboe }; 147ed29b0b4SJens Axboe }; 148ed29b0b4SJens Axboe 149ed29b0b4SJens Axboe struct io_rsrc_node { 150ed29b0b4SJens Axboe struct percpu_ref refs; 151ed29b0b4SJens Axboe struct list_head node; 152ed29b0b4SJens Axboe struct list_head rsrc_list; 153ed29b0b4SJens Axboe struct io_rsrc_data *rsrc_data; 154ed29b0b4SJens Axboe struct llist_node llist; 155ed29b0b4SJens Axboe bool done; 156ed29b0b4SJens Axboe }; 157ed29b0b4SJens Axboe 158ed29b0b4SJens Axboe typedef void (rsrc_put_fn)(struct io_ring_ctx *ctx, struct io_rsrc_put *prsrc); 159ed29b0b4SJens Axboe 160ed29b0b4SJens Axboe struct io_rsrc_data { 161ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 162ed29b0b4SJens Axboe 163ed29b0b4SJens Axboe u64 **tags; 164ed29b0b4SJens Axboe unsigned int nr; 165ed29b0b4SJens Axboe rsrc_put_fn *do_put; 166ed29b0b4SJens Axboe atomic_t refs; 167ed29b0b4SJens Axboe struct completion done; 168ed29b0b4SJens Axboe bool quiesce; 169ed29b0b4SJens Axboe }; 170ed29b0b4SJens Axboe 171ed29b0b4SJens Axboe #define IO_BUFFER_LIST_BUF_PER_PAGE (PAGE_SIZE / sizeof(struct io_uring_buf)) 172ed29b0b4SJens Axboe struct io_buffer_list { 173ed29b0b4SJens Axboe /* 174ed29b0b4SJens Axboe * If ->buf_nr_pages is set, then buf_pages/buf_ring are used. If not, 175ed29b0b4SJens Axboe * then these are classic provided buffers and ->buf_list is used. 176ed29b0b4SJens Axboe */ 177ed29b0b4SJens Axboe union { 178ed29b0b4SJens Axboe struct list_head buf_list; 179ed29b0b4SJens Axboe struct { 180ed29b0b4SJens Axboe struct page **buf_pages; 181ed29b0b4SJens Axboe struct io_uring_buf_ring *buf_ring; 182ed29b0b4SJens Axboe }; 183ed29b0b4SJens Axboe }; 184ed29b0b4SJens Axboe __u16 bgid; 185ed29b0b4SJens Axboe 186ed29b0b4SJens Axboe /* below is for ring provided buffers */ 187ed29b0b4SJens Axboe __u16 buf_nr_pages; 188ed29b0b4SJens Axboe __u16 nr_entries; 189ed29b0b4SJens Axboe __u16 head; 190ed29b0b4SJens Axboe __u16 mask; 191ed29b0b4SJens Axboe }; 192ed29b0b4SJens Axboe 193ed29b0b4SJens Axboe struct io_buffer { 194ed29b0b4SJens Axboe struct list_head list; 195ed29b0b4SJens Axboe __u64 addr; 196ed29b0b4SJens Axboe __u32 len; 197ed29b0b4SJens Axboe __u16 bid; 198ed29b0b4SJens Axboe __u16 bgid; 199ed29b0b4SJens Axboe }; 200ed29b0b4SJens Axboe 201ed29b0b4SJens Axboe #define IO_COMPL_BATCH 32 202ed29b0b4SJens Axboe #define IO_REQ_CACHE_SIZE 32 203ed29b0b4SJens Axboe #define IO_REQ_ALLOC_BATCH 8 204ed29b0b4SJens Axboe 205ed29b0b4SJens Axboe #define BGID_ARRAY 64 206ed29b0b4SJens Axboe 207ed29b0b4SJens Axboe /* 208ed29b0b4SJens Axboe * First field must be the file pointer in all the 209ed29b0b4SJens Axboe * iocb unions! See also 'struct kiocb' in <linux/fs.h> 210ed29b0b4SJens Axboe */ 2118d4388d1SJens Axboe struct io_poll { 212ed29b0b4SJens Axboe struct file *file; 213ed29b0b4SJens Axboe struct wait_queue_head *head; 214ed29b0b4SJens Axboe __poll_t events; 215ed29b0b4SJens Axboe struct wait_queue_entry wait; 216ed29b0b4SJens Axboe }; 217ed29b0b4SJens Axboe 218ed29b0b4SJens Axboe struct io_poll_update { 219ed29b0b4SJens Axboe struct file *file; 220ed29b0b4SJens Axboe u64 old_user_data; 221ed29b0b4SJens Axboe u64 new_user_data; 222ed29b0b4SJens Axboe __poll_t events; 223ed29b0b4SJens Axboe bool update_events; 224ed29b0b4SJens Axboe bool update_user_data; 225ed29b0b4SJens Axboe }; 226ed29b0b4SJens Axboe 227ed29b0b4SJens Axboe struct io_cancel { 228ed29b0b4SJens Axboe struct file *file; 229ed29b0b4SJens Axboe u64 addr; 230ed29b0b4SJens Axboe u32 flags; 231ed29b0b4SJens Axboe s32 fd; 232ed29b0b4SJens Axboe }; 233ed29b0b4SJens Axboe 234ed29b0b4SJens Axboe struct io_rw { 235ed29b0b4SJens Axboe /* NOTE: kiocb has the file as the first member, so don't do it here */ 236ed29b0b4SJens Axboe struct kiocb kiocb; 237ed29b0b4SJens Axboe u64 addr; 238ed29b0b4SJens Axboe u32 len; 239ed29b0b4SJens Axboe rwf_t flags; 240ed29b0b4SJens Axboe }; 241ed29b0b4SJens Axboe 242ed29b0b4SJens Axboe struct io_rsrc_update { 243ed29b0b4SJens Axboe struct file *file; 244ed29b0b4SJens Axboe u64 arg; 245ed29b0b4SJens Axboe u32 nr_args; 246ed29b0b4SJens Axboe u32 offset; 247ed29b0b4SJens Axboe }; 248ed29b0b4SJens Axboe 249ed29b0b4SJens Axboe struct io_provide_buf { 250ed29b0b4SJens Axboe struct file *file; 251ed29b0b4SJens Axboe __u64 addr; 252ed29b0b4SJens Axboe __u32 len; 253ed29b0b4SJens Axboe __u32 bgid; 254ed29b0b4SJens Axboe __u16 nbufs; 255ed29b0b4SJens Axboe __u16 bid; 256ed29b0b4SJens Axboe }; 257ed29b0b4SJens Axboe 258ed29b0b4SJens Axboe struct io_rw_state { 259ed29b0b4SJens Axboe struct iov_iter iter; 260ed29b0b4SJens Axboe struct iov_iter_state iter_state; 261ed29b0b4SJens Axboe struct iovec fast_iov[UIO_FASTIOV]; 262ed29b0b4SJens Axboe }; 263ed29b0b4SJens Axboe 264ed29b0b4SJens Axboe struct io_async_rw { 265ed29b0b4SJens Axboe struct io_rw_state s; 266ed29b0b4SJens Axboe const struct iovec *free_iovec; 267ed29b0b4SJens Axboe size_t bytes_done; 268ed29b0b4SJens Axboe struct wait_page_queue wpq; 269ed29b0b4SJens Axboe }; 270ed29b0b4SJens Axboe 271ed29b0b4SJens Axboe struct async_poll { 2728d4388d1SJens Axboe struct io_poll poll; 2738d4388d1SJens Axboe struct io_poll *double_poll; 274ed29b0b4SJens Axboe }; 275ed29b0b4SJens Axboe 276ed29b0b4SJens Axboe enum { 277ed29b0b4SJens Axboe IORING_RSRC_FILE = 0, 278ed29b0b4SJens Axboe IORING_RSRC_BUFFER = 1, 279ed29b0b4SJens Axboe }; 280ed29b0b4SJens Axboe 281ed29b0b4SJens Axboe enum { 282ed29b0b4SJens Axboe IO_CHECK_CQ_OVERFLOW_BIT, 283ed29b0b4SJens Axboe IO_CHECK_CQ_DROPPED_BIT, 284ed29b0b4SJens Axboe }; 285ed29b0b4SJens Axboe 286ed29b0b4SJens Axboe struct io_defer_entry { 287ed29b0b4SJens Axboe struct list_head list; 288ed29b0b4SJens Axboe struct io_kiocb *req; 289ed29b0b4SJens Axboe u32 seq; 290ed29b0b4SJens Axboe }; 291ed29b0b4SJens Axboe 292ed29b0b4SJens Axboe struct io_op_def { 293ed29b0b4SJens Axboe /* needs req->file assigned */ 294ed29b0b4SJens Axboe unsigned needs_file : 1; 295ed29b0b4SJens Axboe /* should block plug */ 296ed29b0b4SJens Axboe unsigned plug : 1; 297ed29b0b4SJens Axboe /* hash wq insertion if file is a regular file */ 298ed29b0b4SJens Axboe unsigned hash_reg_file : 1; 299ed29b0b4SJens Axboe /* unbound wq insertion if file is a non-regular file */ 300ed29b0b4SJens Axboe unsigned unbound_nonreg_file : 1; 301ed29b0b4SJens Axboe /* set if opcode supports polled "wait" */ 302ed29b0b4SJens Axboe unsigned pollin : 1; 303ed29b0b4SJens Axboe unsigned pollout : 1; 304ed29b0b4SJens Axboe unsigned poll_exclusive : 1; 305ed29b0b4SJens Axboe /* op supports buffer selection */ 306ed29b0b4SJens Axboe unsigned buffer_select : 1; 307ed29b0b4SJens Axboe /* opcode is not supported by this kernel */ 308ed29b0b4SJens Axboe unsigned not_supported : 1; 309ed29b0b4SJens Axboe /* skip auditing */ 310ed29b0b4SJens Axboe unsigned audit_skip : 1; 311ed29b0b4SJens Axboe /* supports ioprio */ 312ed29b0b4SJens Axboe unsigned ioprio : 1; 313ed29b0b4SJens Axboe /* supports iopoll */ 314ed29b0b4SJens Axboe unsigned iopoll : 1; 315ed29b0b4SJens Axboe /* size of async data needed, if any */ 316ed29b0b4SJens Axboe unsigned short async_size; 317ed29b0b4SJens Axboe 318*cfd22e6bSJens Axboe const char *name; 319*cfd22e6bSJens Axboe 320ed29b0b4SJens Axboe int (*prep)(struct io_kiocb *, const struct io_uring_sqe *); 321ed29b0b4SJens Axboe int (*issue)(struct io_kiocb *, unsigned int); 322dc919cafSJens Axboe int (*prep_async)(struct io_kiocb *); 3234d4c9cffSJens Axboe void (*cleanup)(struct io_kiocb *); 324ed29b0b4SJens Axboe }; 325ed29b0b4SJens Axboe 326ed29b0b4SJens Axboe static const struct io_op_def io_op_defs[]; 327ed29b0b4SJens Axboe 328ed29b0b4SJens Axboe /* requests with any of those set should undergo io_disarm_next() */ 329ed29b0b4SJens Axboe #define IO_DISARM_MASK (REQ_F_ARM_LTIMEOUT | REQ_F_LINK_TIMEOUT | REQ_F_FAIL) 330ed29b0b4SJens Axboe #define IO_REQ_LINK_FLAGS (REQ_F_LINK | REQ_F_HARDLINK) 331ed29b0b4SJens Axboe 332ed29b0b4SJens Axboe static void io_uring_try_cancel_requests(struct io_ring_ctx *ctx, 333ed29b0b4SJens Axboe struct task_struct *task, 334ed29b0b4SJens Axboe bool cancel_all); 335ed29b0b4SJens Axboe 336ed29b0b4SJens Axboe static void io_dismantle_req(struct io_kiocb *req); 337ed29b0b4SJens Axboe static int __io_register_rsrc_update(struct io_ring_ctx *ctx, unsigned type, 338ed29b0b4SJens Axboe struct io_uring_rsrc_update2 *up, 339ed29b0b4SJens Axboe unsigned nr_args); 340ed29b0b4SJens Axboe static void io_clean_op(struct io_kiocb *req); 341ed29b0b4SJens Axboe static void io_queue_sqe(struct io_kiocb *req); 342ed29b0b4SJens Axboe static void io_rsrc_put_work(struct work_struct *work); 343ed29b0b4SJens Axboe 344ed29b0b4SJens Axboe static void io_req_task_queue(struct io_kiocb *req); 345ed29b0b4SJens Axboe static void __io_submit_flush_completions(struct io_ring_ctx *ctx); 346ed29b0b4SJens Axboe static int io_req_prep_async(struct io_kiocb *req); 347ed29b0b4SJens Axboe 348ed29b0b4SJens Axboe static void io_eventfd_signal(struct io_ring_ctx *ctx); 349ed29b0b4SJens Axboe 350ed29b0b4SJens Axboe static struct kmem_cache *req_cachep; 351ed29b0b4SJens Axboe 352ed29b0b4SJens Axboe const char *io_uring_get_opcode(u8 opcode) 353ed29b0b4SJens Axboe { 354*cfd22e6bSJens Axboe if (opcode < IORING_OP_LAST) 355*cfd22e6bSJens Axboe return io_op_defs[opcode].name; 356ed29b0b4SJens Axboe return "INVALID"; 357ed29b0b4SJens Axboe } 358ed29b0b4SJens Axboe 359ed29b0b4SJens Axboe struct sock *io_uring_get_socket(struct file *file) 360ed29b0b4SJens Axboe { 361ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 362cd40cae2SJens Axboe if (io_is_uring_fops(file)) { 363ed29b0b4SJens Axboe struct io_ring_ctx *ctx = file->private_data; 364ed29b0b4SJens Axboe 365ed29b0b4SJens Axboe return ctx->ring_sock->sk; 366ed29b0b4SJens Axboe } 367ed29b0b4SJens Axboe #endif 368ed29b0b4SJens Axboe return NULL; 369ed29b0b4SJens Axboe } 370ed29b0b4SJens Axboe EXPORT_SYMBOL(io_uring_get_socket); 371ed29b0b4SJens Axboe 372ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 373ed29b0b4SJens Axboe static inline bool io_file_need_scm(struct file *filp) 374ed29b0b4SJens Axboe { 375ed29b0b4SJens Axboe #if defined(IO_URING_SCM_ALL) 376ed29b0b4SJens Axboe return true; 377ed29b0b4SJens Axboe #else 378ed29b0b4SJens Axboe return !!unix_get_socket(filp); 379ed29b0b4SJens Axboe #endif 380ed29b0b4SJens Axboe } 381ed29b0b4SJens Axboe #else 382ed29b0b4SJens Axboe static inline bool io_file_need_scm(struct file *filp) 383ed29b0b4SJens Axboe { 384ed29b0b4SJens Axboe return false; 385ed29b0b4SJens Axboe } 386ed29b0b4SJens Axboe #endif 387ed29b0b4SJens Axboe 388ed29b0b4SJens Axboe static inline void io_tw_lock(struct io_ring_ctx *ctx, bool *locked) 389ed29b0b4SJens Axboe { 390ed29b0b4SJens Axboe if (!*locked) { 391ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 392ed29b0b4SJens Axboe *locked = true; 393ed29b0b4SJens Axboe } 394ed29b0b4SJens Axboe } 395ed29b0b4SJens Axboe 396ed29b0b4SJens Axboe static inline void io_submit_flush_completions(struct io_ring_ctx *ctx) 397ed29b0b4SJens Axboe { 398ed29b0b4SJens Axboe if (!wq_list_empty(&ctx->submit_state.compl_reqs)) 399ed29b0b4SJens Axboe __io_submit_flush_completions(ctx); 400ed29b0b4SJens Axboe } 401ed29b0b4SJens Axboe 402ed29b0b4SJens Axboe #define IO_RSRC_REF_BATCH 100 403ed29b0b4SJens Axboe 404ed29b0b4SJens Axboe static void io_rsrc_put_node(struct io_rsrc_node *node, int nr) 405ed29b0b4SJens Axboe { 406ed29b0b4SJens Axboe percpu_ref_put_many(&node->refs, nr); 407ed29b0b4SJens Axboe } 408ed29b0b4SJens Axboe 409ed29b0b4SJens Axboe static inline void io_req_put_rsrc_locked(struct io_kiocb *req, 410ed29b0b4SJens Axboe struct io_ring_ctx *ctx) 411ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 412ed29b0b4SJens Axboe { 413ed29b0b4SJens Axboe struct io_rsrc_node *node = req->rsrc_node; 414ed29b0b4SJens Axboe 415ed29b0b4SJens Axboe if (node) { 416ed29b0b4SJens Axboe if (node == ctx->rsrc_node) 417ed29b0b4SJens Axboe ctx->rsrc_cached_refs++; 418ed29b0b4SJens Axboe else 419ed29b0b4SJens Axboe io_rsrc_put_node(node, 1); 420ed29b0b4SJens Axboe } 421ed29b0b4SJens Axboe } 422ed29b0b4SJens Axboe 423ed29b0b4SJens Axboe static inline void io_req_put_rsrc(struct io_kiocb *req) 424ed29b0b4SJens Axboe { 425ed29b0b4SJens Axboe if (req->rsrc_node) 426ed29b0b4SJens Axboe io_rsrc_put_node(req->rsrc_node, 1); 427ed29b0b4SJens Axboe } 428ed29b0b4SJens Axboe 429ed29b0b4SJens Axboe static __cold void io_rsrc_refs_drop(struct io_ring_ctx *ctx) 430ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 431ed29b0b4SJens Axboe { 432ed29b0b4SJens Axboe if (ctx->rsrc_cached_refs) { 433ed29b0b4SJens Axboe io_rsrc_put_node(ctx->rsrc_node, ctx->rsrc_cached_refs); 434ed29b0b4SJens Axboe ctx->rsrc_cached_refs = 0; 435ed29b0b4SJens Axboe } 436ed29b0b4SJens Axboe } 437ed29b0b4SJens Axboe 438ed29b0b4SJens Axboe static void io_rsrc_refs_refill(struct io_ring_ctx *ctx) 439ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 440ed29b0b4SJens Axboe { 441ed29b0b4SJens Axboe ctx->rsrc_cached_refs += IO_RSRC_REF_BATCH; 442ed29b0b4SJens Axboe percpu_ref_get_many(&ctx->rsrc_node->refs, IO_RSRC_REF_BATCH); 443ed29b0b4SJens Axboe } 444ed29b0b4SJens Axboe 445ed29b0b4SJens Axboe static inline void io_req_set_rsrc_node(struct io_kiocb *req, 446ed29b0b4SJens Axboe struct io_ring_ctx *ctx, 447ed29b0b4SJens Axboe unsigned int issue_flags) 448ed29b0b4SJens Axboe { 449ed29b0b4SJens Axboe if (!req->rsrc_node) { 450ed29b0b4SJens Axboe req->rsrc_node = ctx->rsrc_node; 451ed29b0b4SJens Axboe 452ed29b0b4SJens Axboe if (!(issue_flags & IO_URING_F_UNLOCKED)) { 453ed29b0b4SJens Axboe lockdep_assert_held(&ctx->uring_lock); 454ed29b0b4SJens Axboe ctx->rsrc_cached_refs--; 455ed29b0b4SJens Axboe if (unlikely(ctx->rsrc_cached_refs < 0)) 456ed29b0b4SJens Axboe io_rsrc_refs_refill(ctx); 457ed29b0b4SJens Axboe } else { 458ed29b0b4SJens Axboe percpu_ref_get(&req->rsrc_node->refs); 459ed29b0b4SJens Axboe } 460ed29b0b4SJens Axboe } 461ed29b0b4SJens Axboe } 462ed29b0b4SJens Axboe 463ed29b0b4SJens Axboe static unsigned int __io_put_kbuf(struct io_kiocb *req, struct list_head *list) 464ed29b0b4SJens Axboe { 465ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_RING) { 466ed29b0b4SJens Axboe if (req->buf_list) 467ed29b0b4SJens Axboe req->buf_list->head++; 468ed29b0b4SJens Axboe req->flags &= ~REQ_F_BUFFER_RING; 469ed29b0b4SJens Axboe } else { 470ed29b0b4SJens Axboe list_add(&req->kbuf->list, list); 471ed29b0b4SJens Axboe req->flags &= ~REQ_F_BUFFER_SELECTED; 472ed29b0b4SJens Axboe } 473ed29b0b4SJens Axboe 474ed29b0b4SJens Axboe return IORING_CQE_F_BUFFER | (req->buf_index << IORING_CQE_BUFFER_SHIFT); 475ed29b0b4SJens Axboe } 476ed29b0b4SJens Axboe 477ed29b0b4SJens Axboe static inline unsigned int io_put_kbuf_comp(struct io_kiocb *req) 478ed29b0b4SJens Axboe { 479ed29b0b4SJens Axboe lockdep_assert_held(&req->ctx->completion_lock); 480ed29b0b4SJens Axboe 481ed29b0b4SJens Axboe if (!(req->flags & (REQ_F_BUFFER_SELECTED|REQ_F_BUFFER_RING))) 482ed29b0b4SJens Axboe return 0; 483ed29b0b4SJens Axboe return __io_put_kbuf(req, &req->ctx->io_buffers_comp); 484ed29b0b4SJens Axboe } 485ed29b0b4SJens Axboe 486f9ead18cSJens Axboe inline unsigned int io_put_kbuf(struct io_kiocb *req, unsigned issue_flags) 487ed29b0b4SJens Axboe { 488ed29b0b4SJens Axboe unsigned int cflags; 489ed29b0b4SJens Axboe 490ed29b0b4SJens Axboe if (!(req->flags & (REQ_F_BUFFER_SELECTED|REQ_F_BUFFER_RING))) 491ed29b0b4SJens Axboe return 0; 492ed29b0b4SJens Axboe 493ed29b0b4SJens Axboe /* 494ed29b0b4SJens Axboe * We can add this buffer back to two lists: 495ed29b0b4SJens Axboe * 496ed29b0b4SJens Axboe * 1) The io_buffers_cache list. This one is protected by the 497ed29b0b4SJens Axboe * ctx->uring_lock. If we already hold this lock, add back to this 498ed29b0b4SJens Axboe * list as we can grab it from issue as well. 499ed29b0b4SJens Axboe * 2) The io_buffers_comp list. This one is protected by the 500ed29b0b4SJens Axboe * ctx->completion_lock. 501ed29b0b4SJens Axboe * 502ed29b0b4SJens Axboe * We migrate buffers from the comp_list to the issue cache list 503ed29b0b4SJens Axboe * when we need one. 504ed29b0b4SJens Axboe */ 505ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_RING) { 506ed29b0b4SJens Axboe /* no buffers to recycle for this case */ 507ed29b0b4SJens Axboe cflags = __io_put_kbuf(req, NULL); 508ed29b0b4SJens Axboe } else if (issue_flags & IO_URING_F_UNLOCKED) { 509ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 510ed29b0b4SJens Axboe 511ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 512ed29b0b4SJens Axboe cflags = __io_put_kbuf(req, &ctx->io_buffers_comp); 513ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 514ed29b0b4SJens Axboe } else { 515ed29b0b4SJens Axboe lockdep_assert_held(&req->ctx->uring_lock); 516ed29b0b4SJens Axboe 517ed29b0b4SJens Axboe cflags = __io_put_kbuf(req, &req->ctx->io_buffers_cache); 518ed29b0b4SJens Axboe } 519ed29b0b4SJens Axboe 520ed29b0b4SJens Axboe return cflags; 521ed29b0b4SJens Axboe } 522ed29b0b4SJens Axboe 523ed29b0b4SJens Axboe static struct io_buffer_list *io_buffer_get_list(struct io_ring_ctx *ctx, 524ed29b0b4SJens Axboe unsigned int bgid) 525ed29b0b4SJens Axboe { 526ed29b0b4SJens Axboe if (ctx->io_bl && bgid < BGID_ARRAY) 527ed29b0b4SJens Axboe return &ctx->io_bl[bgid]; 528ed29b0b4SJens Axboe 529ed29b0b4SJens Axboe return xa_load(&ctx->io_bl_xa, bgid); 530ed29b0b4SJens Axboe } 531ed29b0b4SJens Axboe 532ed29b0b4SJens Axboe static void io_kbuf_recycle(struct io_kiocb *req, unsigned issue_flags) 533ed29b0b4SJens Axboe { 534ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 535ed29b0b4SJens Axboe struct io_buffer_list *bl; 536ed29b0b4SJens Axboe struct io_buffer *buf; 537ed29b0b4SJens Axboe 538ed29b0b4SJens Axboe if (!(req->flags & (REQ_F_BUFFER_SELECTED|REQ_F_BUFFER_RING))) 539ed29b0b4SJens Axboe return; 540ed29b0b4SJens Axboe /* 541ed29b0b4SJens Axboe * For legacy provided buffer mode, don't recycle if we already did 542ed29b0b4SJens Axboe * IO to this buffer. For ring-mapped provided buffer mode, we should 543ed29b0b4SJens Axboe * increment ring->head to explicitly monopolize the buffer to avoid 544ed29b0b4SJens Axboe * multiple use. 545ed29b0b4SJens Axboe */ 546ed29b0b4SJens Axboe if ((req->flags & REQ_F_BUFFER_SELECTED) && 547ed29b0b4SJens Axboe (req->flags & REQ_F_PARTIAL_IO)) 548ed29b0b4SJens Axboe return; 549ed29b0b4SJens Axboe 550ed29b0b4SJens Axboe /* 551ed29b0b4SJens Axboe * READV uses fields in `struct io_rw` (len/addr) to stash the selected 552ed29b0b4SJens Axboe * buffer data. However if that buffer is recycled the original request 553ed29b0b4SJens Axboe * data stored in addr is lost. Therefore forbid recycling for now. 554ed29b0b4SJens Axboe */ 555ed29b0b4SJens Axboe if (req->opcode == IORING_OP_READV) 556ed29b0b4SJens Axboe return; 557ed29b0b4SJens Axboe 558ed29b0b4SJens Axboe /* 559ed29b0b4SJens Axboe * We don't need to recycle for REQ_F_BUFFER_RING, we can just clear 560ed29b0b4SJens Axboe * the flag and hence ensure that bl->head doesn't get incremented. 561ed29b0b4SJens Axboe * If the tail has already been incremented, hang on to it. 562ed29b0b4SJens Axboe */ 563ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_RING) { 564ed29b0b4SJens Axboe if (req->buf_list) { 565ed29b0b4SJens Axboe if (req->flags & REQ_F_PARTIAL_IO) { 566ed29b0b4SJens Axboe req->buf_list->head++; 567ed29b0b4SJens Axboe req->buf_list = NULL; 568ed29b0b4SJens Axboe } else { 569ed29b0b4SJens Axboe req->buf_index = req->buf_list->bgid; 570ed29b0b4SJens Axboe req->flags &= ~REQ_F_BUFFER_RING; 571ed29b0b4SJens Axboe } 572ed29b0b4SJens Axboe } 573ed29b0b4SJens Axboe return; 574ed29b0b4SJens Axboe } 575ed29b0b4SJens Axboe 576ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 577ed29b0b4SJens Axboe 578ed29b0b4SJens Axboe buf = req->kbuf; 579ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, buf->bgid); 580ed29b0b4SJens Axboe list_add(&buf->list, &bl->buf_list); 581ed29b0b4SJens Axboe req->flags &= ~REQ_F_BUFFER_SELECTED; 582ed29b0b4SJens Axboe req->buf_index = buf->bgid; 583ed29b0b4SJens Axboe 584ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 585ed29b0b4SJens Axboe } 586ed29b0b4SJens Axboe 587ed29b0b4SJens Axboe static bool io_match_linked(struct io_kiocb *head) 588ed29b0b4SJens Axboe { 589ed29b0b4SJens Axboe struct io_kiocb *req; 590ed29b0b4SJens Axboe 591ed29b0b4SJens Axboe io_for_each_link(req, head) { 592ed29b0b4SJens Axboe if (req->flags & REQ_F_INFLIGHT) 593ed29b0b4SJens Axboe return true; 594ed29b0b4SJens Axboe } 595ed29b0b4SJens Axboe return false; 596ed29b0b4SJens Axboe } 597ed29b0b4SJens Axboe 598ed29b0b4SJens Axboe /* 599ed29b0b4SJens Axboe * As io_match_task() but protected against racing with linked timeouts. 600ed29b0b4SJens Axboe * User must not hold timeout_lock. 601ed29b0b4SJens Axboe */ 602ed29b0b4SJens Axboe static bool io_match_task_safe(struct io_kiocb *head, struct task_struct *task, 603ed29b0b4SJens Axboe bool cancel_all) 604ed29b0b4SJens Axboe { 605ed29b0b4SJens Axboe bool matched; 606ed29b0b4SJens Axboe 607ed29b0b4SJens Axboe if (task && head->task != task) 608ed29b0b4SJens Axboe return false; 609ed29b0b4SJens Axboe if (cancel_all) 610ed29b0b4SJens Axboe return true; 611ed29b0b4SJens Axboe 612ed29b0b4SJens Axboe if (head->flags & REQ_F_LINK_TIMEOUT) { 613ed29b0b4SJens Axboe struct io_ring_ctx *ctx = head->ctx; 614ed29b0b4SJens Axboe 615ed29b0b4SJens Axboe /* protect against races with linked timeouts */ 616ed29b0b4SJens Axboe spin_lock_irq(&ctx->timeout_lock); 617ed29b0b4SJens Axboe matched = io_match_linked(head); 618ed29b0b4SJens Axboe spin_unlock_irq(&ctx->timeout_lock); 619ed29b0b4SJens Axboe } else { 620ed29b0b4SJens Axboe matched = io_match_linked(head); 621ed29b0b4SJens Axboe } 622ed29b0b4SJens Axboe return matched; 623ed29b0b4SJens Axboe } 624ed29b0b4SJens Axboe 625ed29b0b4SJens Axboe static inline void req_fail_link_node(struct io_kiocb *req, int res) 626ed29b0b4SJens Axboe { 627ed29b0b4SJens Axboe req_set_fail(req); 62897b388d7SJens Axboe io_req_set_res(req, res, 0); 629ed29b0b4SJens Axboe } 630ed29b0b4SJens Axboe 631ed29b0b4SJens Axboe static inline void io_req_add_to_cache(struct io_kiocb *req, struct io_ring_ctx *ctx) 632ed29b0b4SJens Axboe { 633ed29b0b4SJens Axboe wq_stack_add_head(&req->comp_list, &ctx->submit_state.free_list); 634ed29b0b4SJens Axboe } 635ed29b0b4SJens Axboe 636ed29b0b4SJens Axboe static __cold void io_ring_ctx_ref_free(struct percpu_ref *ref) 637ed29b0b4SJens Axboe { 638ed29b0b4SJens Axboe struct io_ring_ctx *ctx = container_of(ref, struct io_ring_ctx, refs); 639ed29b0b4SJens Axboe 640ed29b0b4SJens Axboe complete(&ctx->ref_comp); 641ed29b0b4SJens Axboe } 642ed29b0b4SJens Axboe 643ed29b0b4SJens Axboe static __cold void io_fallback_req_func(struct work_struct *work) 644ed29b0b4SJens Axboe { 645ed29b0b4SJens Axboe struct io_ring_ctx *ctx = container_of(work, struct io_ring_ctx, 646ed29b0b4SJens Axboe fallback_work.work); 647ed29b0b4SJens Axboe struct llist_node *node = llist_del_all(&ctx->fallback_llist); 648ed29b0b4SJens Axboe struct io_kiocb *req, *tmp; 649ed29b0b4SJens Axboe bool locked = false; 650ed29b0b4SJens Axboe 651ed29b0b4SJens Axboe percpu_ref_get(&ctx->refs); 652ed29b0b4SJens Axboe llist_for_each_entry_safe(req, tmp, node, io_task_work.fallback_node) 653ed29b0b4SJens Axboe req->io_task_work.func(req, &locked); 654ed29b0b4SJens Axboe 655ed29b0b4SJens Axboe if (locked) { 656ed29b0b4SJens Axboe io_submit_flush_completions(ctx); 657ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 658ed29b0b4SJens Axboe } 659ed29b0b4SJens Axboe percpu_ref_put(&ctx->refs); 660ed29b0b4SJens Axboe } 661ed29b0b4SJens Axboe 662ed29b0b4SJens Axboe static __cold struct io_ring_ctx *io_ring_ctx_alloc(struct io_uring_params *p) 663ed29b0b4SJens Axboe { 664ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 665ed29b0b4SJens Axboe int hash_bits; 666ed29b0b4SJens Axboe 667ed29b0b4SJens Axboe ctx = kzalloc(sizeof(*ctx), GFP_KERNEL); 668ed29b0b4SJens Axboe if (!ctx) 669ed29b0b4SJens Axboe return NULL; 670ed29b0b4SJens Axboe 671ed29b0b4SJens Axboe xa_init(&ctx->io_bl_xa); 672ed29b0b4SJens Axboe 673ed29b0b4SJens Axboe /* 674ed29b0b4SJens Axboe * Use 5 bits less than the max cq entries, that should give us around 675ed29b0b4SJens Axboe * 32 entries per hash list if totally full and uniformly spread. 676ed29b0b4SJens Axboe */ 677ed29b0b4SJens Axboe hash_bits = ilog2(p->cq_entries); 678ed29b0b4SJens Axboe hash_bits -= 5; 679ed29b0b4SJens Axboe if (hash_bits <= 0) 680ed29b0b4SJens Axboe hash_bits = 1; 681ed29b0b4SJens Axboe ctx->cancel_hash_bits = hash_bits; 682ed29b0b4SJens Axboe ctx->cancel_hash = kmalloc((1U << hash_bits) * sizeof(struct hlist_head), 683ed29b0b4SJens Axboe GFP_KERNEL); 684ed29b0b4SJens Axboe if (!ctx->cancel_hash) 685ed29b0b4SJens Axboe goto err; 686ed29b0b4SJens Axboe __hash_init(ctx->cancel_hash, 1U << hash_bits); 687ed29b0b4SJens Axboe 688ed29b0b4SJens Axboe ctx->dummy_ubuf = kzalloc(sizeof(*ctx->dummy_ubuf), GFP_KERNEL); 689ed29b0b4SJens Axboe if (!ctx->dummy_ubuf) 690ed29b0b4SJens Axboe goto err; 691ed29b0b4SJens Axboe /* set invalid range, so io_import_fixed() fails meeting it */ 692ed29b0b4SJens Axboe ctx->dummy_ubuf->ubuf = -1UL; 693ed29b0b4SJens Axboe 694ed29b0b4SJens Axboe if (percpu_ref_init(&ctx->refs, io_ring_ctx_ref_free, 695ed29b0b4SJens Axboe PERCPU_REF_ALLOW_REINIT, GFP_KERNEL)) 696ed29b0b4SJens Axboe goto err; 697ed29b0b4SJens Axboe 698ed29b0b4SJens Axboe ctx->flags = p->flags; 699ed29b0b4SJens Axboe init_waitqueue_head(&ctx->sqo_sq_wait); 700ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->sqd_list); 701ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->cq_overflow_list); 702ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->io_buffers_cache); 703ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->apoll_cache); 704ed29b0b4SJens Axboe init_completion(&ctx->ref_comp); 705ed29b0b4SJens Axboe xa_init_flags(&ctx->personalities, XA_FLAGS_ALLOC1); 706ed29b0b4SJens Axboe mutex_init(&ctx->uring_lock); 707ed29b0b4SJens Axboe init_waitqueue_head(&ctx->cq_wait); 708ed29b0b4SJens Axboe spin_lock_init(&ctx->completion_lock); 709ed29b0b4SJens Axboe spin_lock_init(&ctx->timeout_lock); 710ed29b0b4SJens Axboe INIT_WQ_LIST(&ctx->iopoll_list); 711ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->io_buffers_pages); 712ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->io_buffers_comp); 713ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->defer_list); 714ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->timeout_list); 715ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->ltimeout_list); 716ed29b0b4SJens Axboe spin_lock_init(&ctx->rsrc_ref_lock); 717ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->rsrc_ref_list); 718ed29b0b4SJens Axboe INIT_DELAYED_WORK(&ctx->rsrc_put_work, io_rsrc_put_work); 719ed29b0b4SJens Axboe init_llist_head(&ctx->rsrc_put_llist); 720ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->tctx_list); 721ed29b0b4SJens Axboe ctx->submit_state.free_list.next = NULL; 722ed29b0b4SJens Axboe INIT_WQ_LIST(&ctx->locked_free_list); 723ed29b0b4SJens Axboe INIT_DELAYED_WORK(&ctx->fallback_work, io_fallback_req_func); 724ed29b0b4SJens Axboe INIT_WQ_LIST(&ctx->submit_state.compl_reqs); 725ed29b0b4SJens Axboe return ctx; 726ed29b0b4SJens Axboe err: 727ed29b0b4SJens Axboe kfree(ctx->dummy_ubuf); 728ed29b0b4SJens Axboe kfree(ctx->cancel_hash); 729ed29b0b4SJens Axboe kfree(ctx->io_bl); 730ed29b0b4SJens Axboe xa_destroy(&ctx->io_bl_xa); 731ed29b0b4SJens Axboe kfree(ctx); 732ed29b0b4SJens Axboe return NULL; 733ed29b0b4SJens Axboe } 734ed29b0b4SJens Axboe 735ed29b0b4SJens Axboe static void io_account_cq_overflow(struct io_ring_ctx *ctx) 736ed29b0b4SJens Axboe { 737ed29b0b4SJens Axboe struct io_rings *r = ctx->rings; 738ed29b0b4SJens Axboe 739ed29b0b4SJens Axboe WRITE_ONCE(r->cq_overflow, READ_ONCE(r->cq_overflow) + 1); 740ed29b0b4SJens Axboe ctx->cq_extra--; 741ed29b0b4SJens Axboe } 742ed29b0b4SJens Axboe 743ed29b0b4SJens Axboe static bool req_need_defer(struct io_kiocb *req, u32 seq) 744ed29b0b4SJens Axboe { 745ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_IO_DRAIN)) { 746ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 747ed29b0b4SJens Axboe 748ed29b0b4SJens Axboe return seq + READ_ONCE(ctx->cq_extra) != ctx->cached_cq_tail; 749ed29b0b4SJens Axboe } 750ed29b0b4SJens Axboe 751ed29b0b4SJens Axboe return false; 752ed29b0b4SJens Axboe } 753ed29b0b4SJens Axboe 754ed29b0b4SJens Axboe static inline bool io_req_ffs_set(struct io_kiocb *req) 755ed29b0b4SJens Axboe { 756ed29b0b4SJens Axboe return req->flags & REQ_F_FIXED_FILE; 757ed29b0b4SJens Axboe } 758ed29b0b4SJens Axboe 759ed29b0b4SJens Axboe static inline void io_req_track_inflight(struct io_kiocb *req) 760ed29b0b4SJens Axboe { 761ed29b0b4SJens Axboe if (!(req->flags & REQ_F_INFLIGHT)) { 762ed29b0b4SJens Axboe req->flags |= REQ_F_INFLIGHT; 763ed29b0b4SJens Axboe atomic_inc(&req->task->io_uring->inflight_tracked); 764ed29b0b4SJens Axboe } 765ed29b0b4SJens Axboe } 766ed29b0b4SJens Axboe 767ed29b0b4SJens Axboe static struct io_kiocb *__io_prep_linked_timeout(struct io_kiocb *req) 768ed29b0b4SJens Axboe { 769ed29b0b4SJens Axboe if (WARN_ON_ONCE(!req->link)) 770ed29b0b4SJens Axboe return NULL; 771ed29b0b4SJens Axboe 772ed29b0b4SJens Axboe req->flags &= ~REQ_F_ARM_LTIMEOUT; 773ed29b0b4SJens Axboe req->flags |= REQ_F_LINK_TIMEOUT; 774ed29b0b4SJens Axboe 775ed29b0b4SJens Axboe /* linked timeouts should have two refs once prep'ed */ 776ed29b0b4SJens Axboe io_req_set_refcount(req); 777ed29b0b4SJens Axboe __io_req_set_refcount(req->link, 2); 778ed29b0b4SJens Axboe return req->link; 779ed29b0b4SJens Axboe } 780ed29b0b4SJens Axboe 781ed29b0b4SJens Axboe static inline struct io_kiocb *io_prep_linked_timeout(struct io_kiocb *req) 782ed29b0b4SJens Axboe { 783ed29b0b4SJens Axboe if (likely(!(req->flags & REQ_F_ARM_LTIMEOUT))) 784ed29b0b4SJens Axboe return NULL; 785ed29b0b4SJens Axboe return __io_prep_linked_timeout(req); 786ed29b0b4SJens Axboe } 787ed29b0b4SJens Axboe 788ed29b0b4SJens Axboe static noinline void __io_arm_ltimeout(struct io_kiocb *req) 789ed29b0b4SJens Axboe { 790ed29b0b4SJens Axboe io_queue_linked_timeout(__io_prep_linked_timeout(req)); 791ed29b0b4SJens Axboe } 792ed29b0b4SJens Axboe 793ed29b0b4SJens Axboe static inline void io_arm_ltimeout(struct io_kiocb *req) 794ed29b0b4SJens Axboe { 795ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_ARM_LTIMEOUT)) 796ed29b0b4SJens Axboe __io_arm_ltimeout(req); 797ed29b0b4SJens Axboe } 798ed29b0b4SJens Axboe 799ed29b0b4SJens Axboe static void io_prep_async_work(struct io_kiocb *req) 800ed29b0b4SJens Axboe { 801ed29b0b4SJens Axboe const struct io_op_def *def = &io_op_defs[req->opcode]; 802ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 803ed29b0b4SJens Axboe 804ed29b0b4SJens Axboe if (!(req->flags & REQ_F_CREDS)) { 805ed29b0b4SJens Axboe req->flags |= REQ_F_CREDS; 806ed29b0b4SJens Axboe req->creds = get_current_cred(); 807ed29b0b4SJens Axboe } 808ed29b0b4SJens Axboe 809ed29b0b4SJens Axboe req->work.list.next = NULL; 810ed29b0b4SJens Axboe req->work.flags = 0; 811ed29b0b4SJens Axboe req->work.cancel_seq = atomic_read(&ctx->cancel_seq); 812ed29b0b4SJens Axboe if (req->flags & REQ_F_FORCE_ASYNC) 813ed29b0b4SJens Axboe req->work.flags |= IO_WQ_WORK_CONCURRENT; 814ed29b0b4SJens Axboe 815ed29b0b4SJens Axboe if (req->flags & REQ_F_ISREG) { 816ed29b0b4SJens Axboe if (def->hash_reg_file || (ctx->flags & IORING_SETUP_IOPOLL)) 817ed29b0b4SJens Axboe io_wq_hash_work(&req->work, file_inode(req->file)); 818ed29b0b4SJens Axboe } else if (!req->file || !S_ISBLK(file_inode(req->file)->i_mode)) { 819ed29b0b4SJens Axboe if (def->unbound_nonreg_file) 820ed29b0b4SJens Axboe req->work.flags |= IO_WQ_WORK_UNBOUND; 821ed29b0b4SJens Axboe } 822ed29b0b4SJens Axboe } 823ed29b0b4SJens Axboe 824ed29b0b4SJens Axboe static void io_prep_async_link(struct io_kiocb *req) 825ed29b0b4SJens Axboe { 826ed29b0b4SJens Axboe struct io_kiocb *cur; 827ed29b0b4SJens Axboe 828ed29b0b4SJens Axboe if (req->flags & REQ_F_LINK_TIMEOUT) { 829ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 830ed29b0b4SJens Axboe 831ed29b0b4SJens Axboe spin_lock_irq(&ctx->timeout_lock); 832ed29b0b4SJens Axboe io_for_each_link(cur, req) 833ed29b0b4SJens Axboe io_prep_async_work(cur); 834ed29b0b4SJens Axboe spin_unlock_irq(&ctx->timeout_lock); 835ed29b0b4SJens Axboe } else { 836ed29b0b4SJens Axboe io_for_each_link(cur, req) 837ed29b0b4SJens Axboe io_prep_async_work(cur); 838ed29b0b4SJens Axboe } 839ed29b0b4SJens Axboe } 840ed29b0b4SJens Axboe 841ed29b0b4SJens Axboe static inline void io_req_add_compl_list(struct io_kiocb *req) 842ed29b0b4SJens Axboe { 843ed29b0b4SJens Axboe struct io_submit_state *state = &req->ctx->submit_state; 844ed29b0b4SJens Axboe 845ed29b0b4SJens Axboe if (!(req->flags & REQ_F_CQE_SKIP)) 846ed29b0b4SJens Axboe state->flush_cqes = true; 847ed29b0b4SJens Axboe wq_list_add_tail(&req->comp_list, &state->compl_reqs); 848ed29b0b4SJens Axboe } 849ed29b0b4SJens Axboe 850ed29b0b4SJens Axboe static void io_queue_iowq(struct io_kiocb *req, bool *dont_use) 851ed29b0b4SJens Axboe { 852ed29b0b4SJens Axboe struct io_kiocb *link = io_prep_linked_timeout(req); 853ed29b0b4SJens Axboe struct io_uring_task *tctx = req->task->io_uring; 854ed29b0b4SJens Axboe 855ed29b0b4SJens Axboe BUG_ON(!tctx); 856ed29b0b4SJens Axboe BUG_ON(!tctx->io_wq); 857ed29b0b4SJens Axboe 858ed29b0b4SJens Axboe /* init ->work of the whole link before punting */ 859ed29b0b4SJens Axboe io_prep_async_link(req); 860ed29b0b4SJens Axboe 861ed29b0b4SJens Axboe /* 862ed29b0b4SJens Axboe * Not expected to happen, but if we do have a bug where this _can_ 863ed29b0b4SJens Axboe * happen, catch it here and ensure the request is marked as 864ed29b0b4SJens Axboe * canceled. That will make io-wq go through the usual work cancel 865ed29b0b4SJens Axboe * procedure rather than attempt to run this request (or create a new 866ed29b0b4SJens Axboe * worker for it). 867ed29b0b4SJens Axboe */ 868ed29b0b4SJens Axboe if (WARN_ON_ONCE(!same_thread_group(req->task, current))) 869ed29b0b4SJens Axboe req->work.flags |= IO_WQ_WORK_CANCEL; 870ed29b0b4SJens Axboe 871ed29b0b4SJens Axboe trace_io_uring_queue_async_work(req->ctx, req, req->cqe.user_data, 872ed29b0b4SJens Axboe req->opcode, req->flags, &req->work, 873ed29b0b4SJens Axboe io_wq_is_hashed(&req->work)); 874ed29b0b4SJens Axboe io_wq_enqueue(tctx->io_wq, &req->work); 875ed29b0b4SJens Axboe if (link) 876ed29b0b4SJens Axboe io_queue_linked_timeout(link); 877ed29b0b4SJens Axboe } 878ed29b0b4SJens Axboe 879ed29b0b4SJens Axboe static __cold void io_queue_deferred(struct io_ring_ctx *ctx) 880ed29b0b4SJens Axboe { 881ed29b0b4SJens Axboe while (!list_empty(&ctx->defer_list)) { 882ed29b0b4SJens Axboe struct io_defer_entry *de = list_first_entry(&ctx->defer_list, 883ed29b0b4SJens Axboe struct io_defer_entry, list); 884ed29b0b4SJens Axboe 885ed29b0b4SJens Axboe if (req_need_defer(de->req, de->seq)) 886ed29b0b4SJens Axboe break; 887ed29b0b4SJens Axboe list_del_init(&de->list); 888ed29b0b4SJens Axboe io_req_task_queue(de->req); 889ed29b0b4SJens Axboe kfree(de); 890ed29b0b4SJens Axboe } 891ed29b0b4SJens Axboe } 892ed29b0b4SJens Axboe 893ed29b0b4SJens Axboe static void __io_commit_cqring_flush(struct io_ring_ctx *ctx) 894ed29b0b4SJens Axboe { 895ed29b0b4SJens Axboe if (ctx->off_timeout_used || ctx->drain_active) { 896ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 897ed29b0b4SJens Axboe if (ctx->off_timeout_used) 898ed29b0b4SJens Axboe io_flush_timeouts(ctx); 899ed29b0b4SJens Axboe if (ctx->drain_active) 900ed29b0b4SJens Axboe io_queue_deferred(ctx); 901ed29b0b4SJens Axboe io_commit_cqring(ctx); 902ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 903ed29b0b4SJens Axboe } 904ed29b0b4SJens Axboe if (ctx->has_evfd) 905ed29b0b4SJens Axboe io_eventfd_signal(ctx); 906ed29b0b4SJens Axboe } 907ed29b0b4SJens Axboe 908ed29b0b4SJens Axboe static inline unsigned int __io_cqring_events(struct io_ring_ctx *ctx) 909ed29b0b4SJens Axboe { 910ed29b0b4SJens Axboe return ctx->cached_cq_tail - READ_ONCE(ctx->rings->cq.head); 911ed29b0b4SJens Axboe } 912ed29b0b4SJens Axboe 913ed29b0b4SJens Axboe /* 914ed29b0b4SJens Axboe * writes to the cq entry need to come after reading head; the 915ed29b0b4SJens Axboe * control dependency is enough as we're using WRITE_ONCE to 916ed29b0b4SJens Axboe * fill the cq entry 917ed29b0b4SJens Axboe */ 918ed29b0b4SJens Axboe static noinline struct io_uring_cqe *__io_get_cqe(struct io_ring_ctx *ctx) 919ed29b0b4SJens Axboe { 920ed29b0b4SJens Axboe struct io_rings *rings = ctx->rings; 921ed29b0b4SJens Axboe unsigned int off = ctx->cached_cq_tail & (ctx->cq_entries - 1); 922ed29b0b4SJens Axboe unsigned int shift = 0; 923ed29b0b4SJens Axboe unsigned int free, queued, len; 924ed29b0b4SJens Axboe 925ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_CQE32) 926ed29b0b4SJens Axboe shift = 1; 927ed29b0b4SJens Axboe 928ed29b0b4SJens Axboe /* userspace may cheat modifying the tail, be safe and do min */ 929ed29b0b4SJens Axboe queued = min(__io_cqring_events(ctx), ctx->cq_entries); 930ed29b0b4SJens Axboe free = ctx->cq_entries - queued; 931ed29b0b4SJens Axboe /* we need a contiguous range, limit based on the current array offset */ 932ed29b0b4SJens Axboe len = min(free, ctx->cq_entries - off); 933ed29b0b4SJens Axboe if (!len) 934ed29b0b4SJens Axboe return NULL; 935ed29b0b4SJens Axboe 936ed29b0b4SJens Axboe ctx->cached_cq_tail++; 937ed29b0b4SJens Axboe ctx->cqe_cached = &rings->cqes[off]; 938ed29b0b4SJens Axboe ctx->cqe_sentinel = ctx->cqe_cached + len; 939ed29b0b4SJens Axboe ctx->cqe_cached++; 940ed29b0b4SJens Axboe return &rings->cqes[off << shift]; 941ed29b0b4SJens Axboe } 942ed29b0b4SJens Axboe 943ed29b0b4SJens Axboe static inline struct io_uring_cqe *io_get_cqe(struct io_ring_ctx *ctx) 944ed29b0b4SJens Axboe { 945ed29b0b4SJens Axboe if (likely(ctx->cqe_cached < ctx->cqe_sentinel)) { 946ed29b0b4SJens Axboe struct io_uring_cqe *cqe = ctx->cqe_cached; 947ed29b0b4SJens Axboe 948ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_CQE32) { 949ed29b0b4SJens Axboe unsigned int off = ctx->cqe_cached - ctx->rings->cqes; 950ed29b0b4SJens Axboe 951ed29b0b4SJens Axboe cqe += off; 952ed29b0b4SJens Axboe } 953ed29b0b4SJens Axboe 954ed29b0b4SJens Axboe ctx->cached_cq_tail++; 955ed29b0b4SJens Axboe ctx->cqe_cached++; 956ed29b0b4SJens Axboe return cqe; 957ed29b0b4SJens Axboe } 958ed29b0b4SJens Axboe 959ed29b0b4SJens Axboe return __io_get_cqe(ctx); 960ed29b0b4SJens Axboe } 961ed29b0b4SJens Axboe 962ed29b0b4SJens Axboe static void io_eventfd_signal(struct io_ring_ctx *ctx) 963ed29b0b4SJens Axboe { 964ed29b0b4SJens Axboe struct io_ev_fd *ev_fd; 965ed29b0b4SJens Axboe 966ed29b0b4SJens Axboe rcu_read_lock(); 967ed29b0b4SJens Axboe /* 968ed29b0b4SJens Axboe * rcu_dereference ctx->io_ev_fd once and use it for both for checking 969ed29b0b4SJens Axboe * and eventfd_signal 970ed29b0b4SJens Axboe */ 971ed29b0b4SJens Axboe ev_fd = rcu_dereference(ctx->io_ev_fd); 972ed29b0b4SJens Axboe 973ed29b0b4SJens Axboe /* 974ed29b0b4SJens Axboe * Check again if ev_fd exists incase an io_eventfd_unregister call 975ed29b0b4SJens Axboe * completed between the NULL check of ctx->io_ev_fd at the start of 976ed29b0b4SJens Axboe * the function and rcu_read_lock. 977ed29b0b4SJens Axboe */ 978ed29b0b4SJens Axboe if (unlikely(!ev_fd)) 979ed29b0b4SJens Axboe goto out; 980ed29b0b4SJens Axboe if (READ_ONCE(ctx->rings->cq_flags) & IORING_CQ_EVENTFD_DISABLED) 981ed29b0b4SJens Axboe goto out; 982ed29b0b4SJens Axboe 983ed29b0b4SJens Axboe if (!ev_fd->eventfd_async || io_wq_current_is_worker()) 984ed29b0b4SJens Axboe eventfd_signal(ev_fd->cq_ev_fd, 1); 985ed29b0b4SJens Axboe out: 986ed29b0b4SJens Axboe rcu_read_unlock(); 987ed29b0b4SJens Axboe } 988ed29b0b4SJens Axboe 989ed29b0b4SJens Axboe static inline void io_cqring_wake(struct io_ring_ctx *ctx) 990ed29b0b4SJens Axboe { 991ed29b0b4SJens Axboe /* 992ed29b0b4SJens Axboe * wake_up_all() may seem excessive, but io_wake_function() and 993ed29b0b4SJens Axboe * io_should_wake() handle the termination of the loop and only 994ed29b0b4SJens Axboe * wake as many waiters as we need to. 995ed29b0b4SJens Axboe */ 996ed29b0b4SJens Axboe if (wq_has_sleeper(&ctx->cq_wait)) 997ed29b0b4SJens Axboe wake_up_all(&ctx->cq_wait); 998ed29b0b4SJens Axboe } 999ed29b0b4SJens Axboe 1000ed29b0b4SJens Axboe /* 1001ed29b0b4SJens Axboe * This should only get called when at least one event has been posted. 1002ed29b0b4SJens Axboe * Some applications rely on the eventfd notification count only changing 1003ed29b0b4SJens Axboe * IFF a new CQE has been added to the CQ ring. There's no depedency on 1004ed29b0b4SJens Axboe * 1:1 relationship between how many times this function is called (and 1005ed29b0b4SJens Axboe * hence the eventfd count) and number of CQEs posted to the CQ ring. 1006ed29b0b4SJens Axboe */ 1007f9ead18cSJens Axboe void io_cqring_ev_posted(struct io_ring_ctx *ctx) 1008ed29b0b4SJens Axboe { 1009ed29b0b4SJens Axboe if (unlikely(ctx->off_timeout_used || ctx->drain_active || 1010ed29b0b4SJens Axboe ctx->has_evfd)) 1011ed29b0b4SJens Axboe __io_commit_cqring_flush(ctx); 1012ed29b0b4SJens Axboe 1013ed29b0b4SJens Axboe io_cqring_wake(ctx); 1014ed29b0b4SJens Axboe } 1015ed29b0b4SJens Axboe 1016ed29b0b4SJens Axboe static void io_cqring_ev_posted_iopoll(struct io_ring_ctx *ctx) 1017ed29b0b4SJens Axboe { 1018ed29b0b4SJens Axboe if (unlikely(ctx->off_timeout_used || ctx->drain_active || 1019ed29b0b4SJens Axboe ctx->has_evfd)) 1020ed29b0b4SJens Axboe __io_commit_cqring_flush(ctx); 1021ed29b0b4SJens Axboe 1022ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQPOLL) 1023ed29b0b4SJens Axboe io_cqring_wake(ctx); 1024ed29b0b4SJens Axboe } 1025ed29b0b4SJens Axboe 1026ed29b0b4SJens Axboe /* Returns true if there are no backlogged entries after the flush */ 1027ed29b0b4SJens Axboe static bool __io_cqring_overflow_flush(struct io_ring_ctx *ctx, bool force) 1028ed29b0b4SJens Axboe { 1029ed29b0b4SJens Axboe bool all_flushed, posted; 1030ed29b0b4SJens Axboe size_t cqe_size = sizeof(struct io_uring_cqe); 1031ed29b0b4SJens Axboe 1032ed29b0b4SJens Axboe if (!force && __io_cqring_events(ctx) == ctx->cq_entries) 1033ed29b0b4SJens Axboe return false; 1034ed29b0b4SJens Axboe 1035ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_CQE32) 1036ed29b0b4SJens Axboe cqe_size <<= 1; 1037ed29b0b4SJens Axboe 1038ed29b0b4SJens Axboe posted = false; 1039ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 1040ed29b0b4SJens Axboe while (!list_empty(&ctx->cq_overflow_list)) { 1041ed29b0b4SJens Axboe struct io_uring_cqe *cqe = io_get_cqe(ctx); 1042ed29b0b4SJens Axboe struct io_overflow_cqe *ocqe; 1043ed29b0b4SJens Axboe 1044ed29b0b4SJens Axboe if (!cqe && !force) 1045ed29b0b4SJens Axboe break; 1046ed29b0b4SJens Axboe ocqe = list_first_entry(&ctx->cq_overflow_list, 1047ed29b0b4SJens Axboe struct io_overflow_cqe, list); 1048ed29b0b4SJens Axboe if (cqe) 1049ed29b0b4SJens Axboe memcpy(cqe, &ocqe->cqe, cqe_size); 1050ed29b0b4SJens Axboe else 1051ed29b0b4SJens Axboe io_account_cq_overflow(ctx); 1052ed29b0b4SJens Axboe 1053ed29b0b4SJens Axboe posted = true; 1054ed29b0b4SJens Axboe list_del(&ocqe->list); 1055ed29b0b4SJens Axboe kfree(ocqe); 1056ed29b0b4SJens Axboe } 1057ed29b0b4SJens Axboe 1058ed29b0b4SJens Axboe all_flushed = list_empty(&ctx->cq_overflow_list); 1059ed29b0b4SJens Axboe if (all_flushed) { 1060ed29b0b4SJens Axboe clear_bit(IO_CHECK_CQ_OVERFLOW_BIT, &ctx->check_cq); 1061ed29b0b4SJens Axboe atomic_andnot(IORING_SQ_CQ_OVERFLOW, &ctx->rings->sq_flags); 1062ed29b0b4SJens Axboe } 1063ed29b0b4SJens Axboe 1064ed29b0b4SJens Axboe io_commit_cqring(ctx); 1065ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 1066ed29b0b4SJens Axboe if (posted) 1067ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 1068ed29b0b4SJens Axboe return all_flushed; 1069ed29b0b4SJens Axboe } 1070ed29b0b4SJens Axboe 1071ed29b0b4SJens Axboe static bool io_cqring_overflow_flush(struct io_ring_ctx *ctx) 1072ed29b0b4SJens Axboe { 1073ed29b0b4SJens Axboe bool ret = true; 1074ed29b0b4SJens Axboe 1075ed29b0b4SJens Axboe if (test_bit(IO_CHECK_CQ_OVERFLOW_BIT, &ctx->check_cq)) { 1076ed29b0b4SJens Axboe /* iopoll syncs against uring_lock, not completion_lock */ 1077ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL) 1078ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 1079ed29b0b4SJens Axboe ret = __io_cqring_overflow_flush(ctx, false); 1080ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL) 1081ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 1082ed29b0b4SJens Axboe } 1083ed29b0b4SJens Axboe 1084ed29b0b4SJens Axboe return ret; 1085ed29b0b4SJens Axboe } 1086ed29b0b4SJens Axboe 1087ed29b0b4SJens Axboe static void __io_put_task(struct task_struct *task, int nr) 1088ed29b0b4SJens Axboe { 1089ed29b0b4SJens Axboe struct io_uring_task *tctx = task->io_uring; 1090ed29b0b4SJens Axboe 1091ed29b0b4SJens Axboe percpu_counter_sub(&tctx->inflight, nr); 1092ed29b0b4SJens Axboe if (unlikely(atomic_read(&tctx->in_idle))) 1093ed29b0b4SJens Axboe wake_up(&tctx->wait); 1094ed29b0b4SJens Axboe put_task_struct_many(task, nr); 1095ed29b0b4SJens Axboe } 1096ed29b0b4SJens Axboe 1097ed29b0b4SJens Axboe /* must to be called somewhat shortly after putting a request */ 1098ed29b0b4SJens Axboe static inline void io_put_task(struct task_struct *task, int nr) 1099ed29b0b4SJens Axboe { 1100ed29b0b4SJens Axboe if (likely(task == current)) 1101ed29b0b4SJens Axboe task->io_uring->cached_refs += nr; 1102ed29b0b4SJens Axboe else 1103ed29b0b4SJens Axboe __io_put_task(task, nr); 1104ed29b0b4SJens Axboe } 1105ed29b0b4SJens Axboe 1106ed29b0b4SJens Axboe static void io_task_refs_refill(struct io_uring_task *tctx) 1107ed29b0b4SJens Axboe { 1108ed29b0b4SJens Axboe unsigned int refill = -tctx->cached_refs + IO_TCTX_REFS_CACHE_NR; 1109ed29b0b4SJens Axboe 1110ed29b0b4SJens Axboe percpu_counter_add(&tctx->inflight, refill); 1111ed29b0b4SJens Axboe refcount_add(refill, ¤t->usage); 1112ed29b0b4SJens Axboe tctx->cached_refs += refill; 1113ed29b0b4SJens Axboe } 1114ed29b0b4SJens Axboe 1115ed29b0b4SJens Axboe static inline void io_get_task_refs(int nr) 1116ed29b0b4SJens Axboe { 1117ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 1118ed29b0b4SJens Axboe 1119ed29b0b4SJens Axboe tctx->cached_refs -= nr; 1120ed29b0b4SJens Axboe if (unlikely(tctx->cached_refs < 0)) 1121ed29b0b4SJens Axboe io_task_refs_refill(tctx); 1122ed29b0b4SJens Axboe } 1123ed29b0b4SJens Axboe 1124ed29b0b4SJens Axboe static __cold void io_uring_drop_tctx_refs(struct task_struct *task) 1125ed29b0b4SJens Axboe { 1126ed29b0b4SJens Axboe struct io_uring_task *tctx = task->io_uring; 1127ed29b0b4SJens Axboe unsigned int refs = tctx->cached_refs; 1128ed29b0b4SJens Axboe 1129ed29b0b4SJens Axboe if (refs) { 1130ed29b0b4SJens Axboe tctx->cached_refs = 0; 1131ed29b0b4SJens Axboe percpu_counter_sub(&tctx->inflight, refs); 1132ed29b0b4SJens Axboe put_task_struct_many(task, refs); 1133ed29b0b4SJens Axboe } 1134ed29b0b4SJens Axboe } 1135ed29b0b4SJens Axboe 1136ed29b0b4SJens Axboe static bool io_cqring_event_overflow(struct io_ring_ctx *ctx, u64 user_data, 1137ed29b0b4SJens Axboe s32 res, u32 cflags, u64 extra1, 1138ed29b0b4SJens Axboe u64 extra2) 1139ed29b0b4SJens Axboe { 1140ed29b0b4SJens Axboe struct io_overflow_cqe *ocqe; 1141ed29b0b4SJens Axboe size_t ocq_size = sizeof(struct io_overflow_cqe); 1142ed29b0b4SJens Axboe bool is_cqe32 = (ctx->flags & IORING_SETUP_CQE32); 1143ed29b0b4SJens Axboe 1144ed29b0b4SJens Axboe if (is_cqe32) 1145ed29b0b4SJens Axboe ocq_size += sizeof(struct io_uring_cqe); 1146ed29b0b4SJens Axboe 1147ed29b0b4SJens Axboe ocqe = kmalloc(ocq_size, GFP_ATOMIC | __GFP_ACCOUNT); 1148ed29b0b4SJens Axboe trace_io_uring_cqe_overflow(ctx, user_data, res, cflags, ocqe); 1149ed29b0b4SJens Axboe if (!ocqe) { 1150ed29b0b4SJens Axboe /* 1151ed29b0b4SJens Axboe * If we're in ring overflow flush mode, or in task cancel mode, 1152ed29b0b4SJens Axboe * or cannot allocate an overflow entry, then we need to drop it 1153ed29b0b4SJens Axboe * on the floor. 1154ed29b0b4SJens Axboe */ 1155ed29b0b4SJens Axboe io_account_cq_overflow(ctx); 1156ed29b0b4SJens Axboe set_bit(IO_CHECK_CQ_DROPPED_BIT, &ctx->check_cq); 1157ed29b0b4SJens Axboe return false; 1158ed29b0b4SJens Axboe } 1159ed29b0b4SJens Axboe if (list_empty(&ctx->cq_overflow_list)) { 1160ed29b0b4SJens Axboe set_bit(IO_CHECK_CQ_OVERFLOW_BIT, &ctx->check_cq); 1161ed29b0b4SJens Axboe atomic_or(IORING_SQ_CQ_OVERFLOW, &ctx->rings->sq_flags); 1162ed29b0b4SJens Axboe 1163ed29b0b4SJens Axboe } 1164ed29b0b4SJens Axboe ocqe->cqe.user_data = user_data; 1165ed29b0b4SJens Axboe ocqe->cqe.res = res; 1166ed29b0b4SJens Axboe ocqe->cqe.flags = cflags; 1167ed29b0b4SJens Axboe if (is_cqe32) { 1168ed29b0b4SJens Axboe ocqe->cqe.big_cqe[0] = extra1; 1169ed29b0b4SJens Axboe ocqe->cqe.big_cqe[1] = extra2; 1170ed29b0b4SJens Axboe } 1171ed29b0b4SJens Axboe list_add_tail(&ocqe->list, &ctx->cq_overflow_list); 1172ed29b0b4SJens Axboe return true; 1173ed29b0b4SJens Axboe } 1174ed29b0b4SJens Axboe 1175ed29b0b4SJens Axboe static inline bool __io_fill_cqe_req(struct io_ring_ctx *ctx, 1176ed29b0b4SJens Axboe struct io_kiocb *req) 1177ed29b0b4SJens Axboe { 1178ed29b0b4SJens Axboe struct io_uring_cqe *cqe; 1179ed29b0b4SJens Axboe 1180ed29b0b4SJens Axboe if (!(ctx->flags & IORING_SETUP_CQE32)) { 1181ed29b0b4SJens Axboe trace_io_uring_complete(req->ctx, req, req->cqe.user_data, 1182ed29b0b4SJens Axboe req->cqe.res, req->cqe.flags, 0, 0); 1183ed29b0b4SJens Axboe 1184ed29b0b4SJens Axboe /* 1185ed29b0b4SJens Axboe * If we can't get a cq entry, userspace overflowed the 1186ed29b0b4SJens Axboe * submission (by quite a lot). Increment the overflow count in 1187ed29b0b4SJens Axboe * the ring. 1188ed29b0b4SJens Axboe */ 1189ed29b0b4SJens Axboe cqe = io_get_cqe(ctx); 1190ed29b0b4SJens Axboe if (likely(cqe)) { 1191ed29b0b4SJens Axboe memcpy(cqe, &req->cqe, sizeof(*cqe)); 1192ed29b0b4SJens Axboe return true; 1193ed29b0b4SJens Axboe } 1194ed29b0b4SJens Axboe 1195ed29b0b4SJens Axboe return io_cqring_event_overflow(ctx, req->cqe.user_data, 1196ed29b0b4SJens Axboe req->cqe.res, req->cqe.flags, 1197ed29b0b4SJens Axboe 0, 0); 1198ed29b0b4SJens Axboe } else { 1199ed29b0b4SJens Axboe u64 extra1 = 0, extra2 = 0; 1200ed29b0b4SJens Axboe 1201ed29b0b4SJens Axboe if (req->flags & REQ_F_CQE32_INIT) { 1202ed29b0b4SJens Axboe extra1 = req->extra1; 1203ed29b0b4SJens Axboe extra2 = req->extra2; 1204ed29b0b4SJens Axboe } 1205ed29b0b4SJens Axboe 1206ed29b0b4SJens Axboe trace_io_uring_complete(req->ctx, req, req->cqe.user_data, 1207ed29b0b4SJens Axboe req->cqe.res, req->cqe.flags, extra1, extra2); 1208ed29b0b4SJens Axboe 1209ed29b0b4SJens Axboe /* 1210ed29b0b4SJens Axboe * If we can't get a cq entry, userspace overflowed the 1211ed29b0b4SJens Axboe * submission (by quite a lot). Increment the overflow count in 1212ed29b0b4SJens Axboe * the ring. 1213ed29b0b4SJens Axboe */ 1214ed29b0b4SJens Axboe cqe = io_get_cqe(ctx); 1215ed29b0b4SJens Axboe if (likely(cqe)) { 1216ed29b0b4SJens Axboe memcpy(cqe, &req->cqe, sizeof(struct io_uring_cqe)); 1217ed29b0b4SJens Axboe WRITE_ONCE(cqe->big_cqe[0], extra1); 1218ed29b0b4SJens Axboe WRITE_ONCE(cqe->big_cqe[1], extra2); 1219ed29b0b4SJens Axboe return true; 1220ed29b0b4SJens Axboe } 1221ed29b0b4SJens Axboe 1222ed29b0b4SJens Axboe return io_cqring_event_overflow(ctx, req->cqe.user_data, 1223ed29b0b4SJens Axboe req->cqe.res, req->cqe.flags, 1224ed29b0b4SJens Axboe extra1, extra2); 1225ed29b0b4SJens Axboe } 1226ed29b0b4SJens Axboe } 1227ed29b0b4SJens Axboe 1228f9ead18cSJens Axboe bool io_fill_cqe_aux(struct io_ring_ctx *ctx, u64 user_data, s32 res, 1229f9ead18cSJens Axboe u32 cflags) 1230ed29b0b4SJens Axboe { 1231ed29b0b4SJens Axboe struct io_uring_cqe *cqe; 1232ed29b0b4SJens Axboe 1233ed29b0b4SJens Axboe ctx->cq_extra++; 1234ed29b0b4SJens Axboe trace_io_uring_complete(ctx, NULL, user_data, res, cflags, 0, 0); 1235ed29b0b4SJens Axboe 1236ed29b0b4SJens Axboe /* 1237ed29b0b4SJens Axboe * If we can't get a cq entry, userspace overflowed the 1238ed29b0b4SJens Axboe * submission (by quite a lot). Increment the overflow count in 1239ed29b0b4SJens Axboe * the ring. 1240ed29b0b4SJens Axboe */ 1241ed29b0b4SJens Axboe cqe = io_get_cqe(ctx); 1242ed29b0b4SJens Axboe if (likely(cqe)) { 1243ed29b0b4SJens Axboe WRITE_ONCE(cqe->user_data, user_data); 1244ed29b0b4SJens Axboe WRITE_ONCE(cqe->res, res); 1245ed29b0b4SJens Axboe WRITE_ONCE(cqe->flags, cflags); 1246ed29b0b4SJens Axboe 1247ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_CQE32) { 1248ed29b0b4SJens Axboe WRITE_ONCE(cqe->big_cqe[0], 0); 1249ed29b0b4SJens Axboe WRITE_ONCE(cqe->big_cqe[1], 0); 1250ed29b0b4SJens Axboe } 1251ed29b0b4SJens Axboe return true; 1252ed29b0b4SJens Axboe } 1253ed29b0b4SJens Axboe return io_cqring_event_overflow(ctx, user_data, res, cflags, 0, 0); 1254ed29b0b4SJens Axboe } 1255ed29b0b4SJens Axboe 1256ed29b0b4SJens Axboe static void __io_req_complete_put(struct io_kiocb *req) 1257ed29b0b4SJens Axboe { 1258ed29b0b4SJens Axboe /* 1259ed29b0b4SJens Axboe * If we're the last reference to this request, add to our locked 1260ed29b0b4SJens Axboe * free_list cache. 1261ed29b0b4SJens Axboe */ 1262ed29b0b4SJens Axboe if (req_ref_put_and_test(req)) { 1263ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 1264ed29b0b4SJens Axboe 1265ed29b0b4SJens Axboe if (req->flags & IO_REQ_LINK_FLAGS) { 1266ed29b0b4SJens Axboe if (req->flags & IO_DISARM_MASK) 1267ed29b0b4SJens Axboe io_disarm_next(req); 1268ed29b0b4SJens Axboe if (req->link) { 1269ed29b0b4SJens Axboe io_req_task_queue(req->link); 1270ed29b0b4SJens Axboe req->link = NULL; 1271ed29b0b4SJens Axboe } 1272ed29b0b4SJens Axboe } 1273ed29b0b4SJens Axboe io_req_put_rsrc(req); 1274ed29b0b4SJens Axboe /* 1275ed29b0b4SJens Axboe * Selected buffer deallocation in io_clean_op() assumes that 1276ed29b0b4SJens Axboe * we don't hold ->completion_lock. Clean them here to avoid 1277ed29b0b4SJens Axboe * deadlocks. 1278ed29b0b4SJens Axboe */ 1279ed29b0b4SJens Axboe io_put_kbuf_comp(req); 1280ed29b0b4SJens Axboe io_dismantle_req(req); 1281ed29b0b4SJens Axboe io_put_task(req->task, 1); 1282ed29b0b4SJens Axboe wq_list_add_head(&req->comp_list, &ctx->locked_free_list); 1283ed29b0b4SJens Axboe ctx->locked_free_nr++; 1284ed29b0b4SJens Axboe } 1285ed29b0b4SJens Axboe } 1286ed29b0b4SJens Axboe 128759915143SJens Axboe void __io_req_complete_post(struct io_kiocb *req) 1288ed29b0b4SJens Axboe { 128997b388d7SJens Axboe if (!(req->flags & REQ_F_CQE_SKIP)) 1290ed29b0b4SJens Axboe __io_fill_cqe_req(req->ctx, req); 1291ed29b0b4SJens Axboe __io_req_complete_put(req); 1292ed29b0b4SJens Axboe } 1293ed29b0b4SJens Axboe 129459915143SJens Axboe void io_req_complete_post(struct io_kiocb *req) 1295ed29b0b4SJens Axboe { 1296ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 1297ed29b0b4SJens Axboe 1298ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 129997b388d7SJens Axboe __io_req_complete_post(req); 1300ed29b0b4SJens Axboe io_commit_cqring(ctx); 1301ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 1302ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 1303ed29b0b4SJens Axboe } 1304ed29b0b4SJens Axboe 130599f15d8dSJens Axboe inline void __io_req_complete(struct io_kiocb *req, unsigned issue_flags) 1306ed29b0b4SJens Axboe { 130797b388d7SJens Axboe if (issue_flags & IO_URING_F_COMPLETE_DEFER) 1308de23077eSJens Axboe req->flags |= REQ_F_COMPLETE_INLINE; 130997b388d7SJens Axboe else 131097b388d7SJens Axboe io_req_complete_post(req); 1311ed29b0b4SJens Axboe } 1312ed29b0b4SJens Axboe 1313ed29b0b4SJens Axboe static void io_req_complete_failed(struct io_kiocb *req, s32 res) 1314ed29b0b4SJens Axboe { 1315ed29b0b4SJens Axboe req_set_fail(req); 131697b388d7SJens Axboe io_req_set_res(req, res, io_put_kbuf(req, IO_URING_F_UNLOCKED)); 131797b388d7SJens Axboe io_req_complete_post(req); 1318ed29b0b4SJens Axboe } 1319ed29b0b4SJens Axboe 1320ed29b0b4SJens Axboe /* 1321ed29b0b4SJens Axboe * Don't initialise the fields below on every allocation, but do that in 1322ed29b0b4SJens Axboe * advance and keep them valid across allocations. 1323ed29b0b4SJens Axboe */ 1324ed29b0b4SJens Axboe static void io_preinit_req(struct io_kiocb *req, struct io_ring_ctx *ctx) 1325ed29b0b4SJens Axboe { 1326ed29b0b4SJens Axboe req->ctx = ctx; 1327ed29b0b4SJens Axboe req->link = NULL; 1328ed29b0b4SJens Axboe req->async_data = NULL; 1329ed29b0b4SJens Axboe /* not necessary, but safer to zero */ 1330ed29b0b4SJens Axboe req->cqe.res = 0; 1331ed29b0b4SJens Axboe } 1332ed29b0b4SJens Axboe 1333ed29b0b4SJens Axboe static void io_flush_cached_locked_reqs(struct io_ring_ctx *ctx, 1334ed29b0b4SJens Axboe struct io_submit_state *state) 1335ed29b0b4SJens Axboe { 1336ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 1337ed29b0b4SJens Axboe wq_list_splice(&ctx->locked_free_list, &state->free_list); 1338ed29b0b4SJens Axboe ctx->locked_free_nr = 0; 1339ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 1340ed29b0b4SJens Axboe } 1341ed29b0b4SJens Axboe 1342ed29b0b4SJens Axboe static inline bool io_req_cache_empty(struct io_ring_ctx *ctx) 1343ed29b0b4SJens Axboe { 1344ed29b0b4SJens Axboe return !ctx->submit_state.free_list.next; 1345ed29b0b4SJens Axboe } 1346ed29b0b4SJens Axboe 1347ed29b0b4SJens Axboe /* 1348ed29b0b4SJens Axboe * A request might get retired back into the request caches even before opcode 1349ed29b0b4SJens Axboe * handlers and io_issue_sqe() are done with it, e.g. inline completion path. 1350ed29b0b4SJens Axboe * Because of that, io_alloc_req() should be called only under ->uring_lock 1351ed29b0b4SJens Axboe * and with extra caution to not get a request that is still worked on. 1352ed29b0b4SJens Axboe */ 1353ed29b0b4SJens Axboe static __cold bool __io_alloc_req_refill(struct io_ring_ctx *ctx) 1354ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 1355ed29b0b4SJens Axboe { 1356ed29b0b4SJens Axboe gfp_t gfp = GFP_KERNEL | __GFP_NOWARN; 1357ed29b0b4SJens Axboe void *reqs[IO_REQ_ALLOC_BATCH]; 1358ed29b0b4SJens Axboe int ret, i; 1359ed29b0b4SJens Axboe 1360ed29b0b4SJens Axboe /* 1361ed29b0b4SJens Axboe * If we have more than a batch's worth of requests in our IRQ side 1362ed29b0b4SJens Axboe * locked cache, grab the lock and move them over to our submission 1363ed29b0b4SJens Axboe * side cache. 1364ed29b0b4SJens Axboe */ 1365ed29b0b4SJens Axboe if (data_race(ctx->locked_free_nr) > IO_COMPL_BATCH) { 1366ed29b0b4SJens Axboe io_flush_cached_locked_reqs(ctx, &ctx->submit_state); 1367ed29b0b4SJens Axboe if (!io_req_cache_empty(ctx)) 1368ed29b0b4SJens Axboe return true; 1369ed29b0b4SJens Axboe } 1370ed29b0b4SJens Axboe 1371ed29b0b4SJens Axboe ret = kmem_cache_alloc_bulk(req_cachep, gfp, ARRAY_SIZE(reqs), reqs); 1372ed29b0b4SJens Axboe 1373ed29b0b4SJens Axboe /* 1374ed29b0b4SJens Axboe * Bulk alloc is all-or-nothing. If we fail to get a batch, 1375ed29b0b4SJens Axboe * retry single alloc to be on the safe side. 1376ed29b0b4SJens Axboe */ 1377ed29b0b4SJens Axboe if (unlikely(ret <= 0)) { 1378ed29b0b4SJens Axboe reqs[0] = kmem_cache_alloc(req_cachep, gfp); 1379ed29b0b4SJens Axboe if (!reqs[0]) 1380ed29b0b4SJens Axboe return false; 1381ed29b0b4SJens Axboe ret = 1; 1382ed29b0b4SJens Axboe } 1383ed29b0b4SJens Axboe 1384ed29b0b4SJens Axboe percpu_ref_get_many(&ctx->refs, ret); 1385ed29b0b4SJens Axboe for (i = 0; i < ret; i++) { 1386ed29b0b4SJens Axboe struct io_kiocb *req = reqs[i]; 1387ed29b0b4SJens Axboe 1388ed29b0b4SJens Axboe io_preinit_req(req, ctx); 1389ed29b0b4SJens Axboe io_req_add_to_cache(req, ctx); 1390ed29b0b4SJens Axboe } 1391ed29b0b4SJens Axboe return true; 1392ed29b0b4SJens Axboe } 1393ed29b0b4SJens Axboe 1394ed29b0b4SJens Axboe static inline bool io_alloc_req_refill(struct io_ring_ctx *ctx) 1395ed29b0b4SJens Axboe { 1396ed29b0b4SJens Axboe if (unlikely(io_req_cache_empty(ctx))) 1397ed29b0b4SJens Axboe return __io_alloc_req_refill(ctx); 1398ed29b0b4SJens Axboe return true; 1399ed29b0b4SJens Axboe } 1400ed29b0b4SJens Axboe 1401ed29b0b4SJens Axboe static inline struct io_kiocb *io_alloc_req(struct io_ring_ctx *ctx) 1402ed29b0b4SJens Axboe { 1403ed29b0b4SJens Axboe struct io_wq_work_node *node; 1404ed29b0b4SJens Axboe 1405ed29b0b4SJens Axboe node = wq_stack_extract(&ctx->submit_state.free_list); 1406ed29b0b4SJens Axboe return container_of(node, struct io_kiocb, comp_list); 1407ed29b0b4SJens Axboe } 1408ed29b0b4SJens Axboe 1409ed29b0b4SJens Axboe static inline void io_dismantle_req(struct io_kiocb *req) 1410ed29b0b4SJens Axboe { 1411ed29b0b4SJens Axboe unsigned int flags = req->flags; 1412ed29b0b4SJens Axboe 1413ed29b0b4SJens Axboe if (unlikely(flags & IO_REQ_CLEAN_FLAGS)) 1414ed29b0b4SJens Axboe io_clean_op(req); 1415ed29b0b4SJens Axboe if (!(flags & REQ_F_FIXED_FILE)) 1416ed29b0b4SJens Axboe io_put_file(req->file); 1417ed29b0b4SJens Axboe } 1418ed29b0b4SJens Axboe 141959915143SJens Axboe __cold void io_free_req(struct io_kiocb *req) 1420ed29b0b4SJens Axboe { 1421ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 1422ed29b0b4SJens Axboe 1423ed29b0b4SJens Axboe io_req_put_rsrc(req); 1424ed29b0b4SJens Axboe io_dismantle_req(req); 1425ed29b0b4SJens Axboe io_put_task(req->task, 1); 1426ed29b0b4SJens Axboe 1427ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 1428ed29b0b4SJens Axboe wq_list_add_head(&req->comp_list, &ctx->locked_free_list); 1429ed29b0b4SJens Axboe ctx->locked_free_nr++; 1430ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 1431ed29b0b4SJens Axboe } 1432ed29b0b4SJens Axboe 1433ed29b0b4SJens Axboe static void __io_req_find_next_prep(struct io_kiocb *req) 1434ed29b0b4SJens Axboe { 1435ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 1436ed29b0b4SJens Axboe bool posted; 1437ed29b0b4SJens Axboe 1438ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 1439ed29b0b4SJens Axboe posted = io_disarm_next(req); 1440ed29b0b4SJens Axboe io_commit_cqring(ctx); 1441ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 1442ed29b0b4SJens Axboe if (posted) 1443ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 1444ed29b0b4SJens Axboe } 1445ed29b0b4SJens Axboe 1446ed29b0b4SJens Axboe static inline struct io_kiocb *io_req_find_next(struct io_kiocb *req) 1447ed29b0b4SJens Axboe { 1448ed29b0b4SJens Axboe struct io_kiocb *nxt; 1449ed29b0b4SJens Axboe 1450ed29b0b4SJens Axboe /* 1451ed29b0b4SJens Axboe * If LINK is set, we have dependent requests in this chain. If we 1452ed29b0b4SJens Axboe * didn't fail this request, queue the first one up, moving any other 1453ed29b0b4SJens Axboe * dependencies to the next request. In case of failure, fail the rest 1454ed29b0b4SJens Axboe * of the chain. 1455ed29b0b4SJens Axboe */ 1456ed29b0b4SJens Axboe if (unlikely(req->flags & IO_DISARM_MASK)) 1457ed29b0b4SJens Axboe __io_req_find_next_prep(req); 1458ed29b0b4SJens Axboe nxt = req->link; 1459ed29b0b4SJens Axboe req->link = NULL; 1460ed29b0b4SJens Axboe return nxt; 1461ed29b0b4SJens Axboe } 1462ed29b0b4SJens Axboe 1463ed29b0b4SJens Axboe static void ctx_flush_and_put(struct io_ring_ctx *ctx, bool *locked) 1464ed29b0b4SJens Axboe { 1465ed29b0b4SJens Axboe if (!ctx) 1466ed29b0b4SJens Axboe return; 1467ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_TASKRUN_FLAG) 1468ed29b0b4SJens Axboe atomic_andnot(IORING_SQ_TASKRUN, &ctx->rings->sq_flags); 1469ed29b0b4SJens Axboe if (*locked) { 1470ed29b0b4SJens Axboe io_submit_flush_completions(ctx); 1471ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 1472ed29b0b4SJens Axboe *locked = false; 1473ed29b0b4SJens Axboe } 1474ed29b0b4SJens Axboe percpu_ref_put(&ctx->refs); 1475ed29b0b4SJens Axboe } 1476ed29b0b4SJens Axboe 1477ed29b0b4SJens Axboe static inline void ctx_commit_and_unlock(struct io_ring_ctx *ctx) 1478ed29b0b4SJens Axboe { 1479ed29b0b4SJens Axboe io_commit_cqring(ctx); 1480ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 1481ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 1482ed29b0b4SJens Axboe } 1483ed29b0b4SJens Axboe 1484ed29b0b4SJens Axboe static void handle_prev_tw_list(struct io_wq_work_node *node, 1485ed29b0b4SJens Axboe struct io_ring_ctx **ctx, bool *uring_locked) 1486ed29b0b4SJens Axboe { 1487ed29b0b4SJens Axboe if (*ctx && !*uring_locked) 1488ed29b0b4SJens Axboe spin_lock(&(*ctx)->completion_lock); 1489ed29b0b4SJens Axboe 1490ed29b0b4SJens Axboe do { 1491ed29b0b4SJens Axboe struct io_wq_work_node *next = node->next; 1492ed29b0b4SJens Axboe struct io_kiocb *req = container_of(node, struct io_kiocb, 1493ed29b0b4SJens Axboe io_task_work.node); 1494ed29b0b4SJens Axboe 1495ed29b0b4SJens Axboe prefetch(container_of(next, struct io_kiocb, io_task_work.node)); 1496ed29b0b4SJens Axboe 1497ed29b0b4SJens Axboe if (req->ctx != *ctx) { 1498ed29b0b4SJens Axboe if (unlikely(!*uring_locked && *ctx)) 1499ed29b0b4SJens Axboe ctx_commit_and_unlock(*ctx); 1500ed29b0b4SJens Axboe 1501ed29b0b4SJens Axboe ctx_flush_and_put(*ctx, uring_locked); 1502ed29b0b4SJens Axboe *ctx = req->ctx; 1503ed29b0b4SJens Axboe /* if not contended, grab and improve batching */ 1504ed29b0b4SJens Axboe *uring_locked = mutex_trylock(&(*ctx)->uring_lock); 1505ed29b0b4SJens Axboe percpu_ref_get(&(*ctx)->refs); 1506ed29b0b4SJens Axboe if (unlikely(!*uring_locked)) 1507ed29b0b4SJens Axboe spin_lock(&(*ctx)->completion_lock); 1508ed29b0b4SJens Axboe } 150997b388d7SJens Axboe if (likely(*uring_locked)) { 1510ed29b0b4SJens Axboe req->io_task_work.func(req, uring_locked); 151197b388d7SJens Axboe } else { 151297b388d7SJens Axboe req->cqe.flags = io_put_kbuf_comp(req); 151397b388d7SJens Axboe __io_req_complete_post(req); 151497b388d7SJens Axboe } 1515ed29b0b4SJens Axboe node = next; 1516ed29b0b4SJens Axboe } while (node); 1517ed29b0b4SJens Axboe 1518ed29b0b4SJens Axboe if (unlikely(!*uring_locked)) 1519ed29b0b4SJens Axboe ctx_commit_and_unlock(*ctx); 1520ed29b0b4SJens Axboe } 1521ed29b0b4SJens Axboe 1522ed29b0b4SJens Axboe static void handle_tw_list(struct io_wq_work_node *node, 1523ed29b0b4SJens Axboe struct io_ring_ctx **ctx, bool *locked) 1524ed29b0b4SJens Axboe { 1525ed29b0b4SJens Axboe do { 1526ed29b0b4SJens Axboe struct io_wq_work_node *next = node->next; 1527ed29b0b4SJens Axboe struct io_kiocb *req = container_of(node, struct io_kiocb, 1528ed29b0b4SJens Axboe io_task_work.node); 1529ed29b0b4SJens Axboe 1530ed29b0b4SJens Axboe prefetch(container_of(next, struct io_kiocb, io_task_work.node)); 1531ed29b0b4SJens Axboe 1532ed29b0b4SJens Axboe if (req->ctx != *ctx) { 1533ed29b0b4SJens Axboe ctx_flush_and_put(*ctx, locked); 1534ed29b0b4SJens Axboe *ctx = req->ctx; 1535ed29b0b4SJens Axboe /* if not contended, grab and improve batching */ 1536ed29b0b4SJens Axboe *locked = mutex_trylock(&(*ctx)->uring_lock); 1537ed29b0b4SJens Axboe percpu_ref_get(&(*ctx)->refs); 1538ed29b0b4SJens Axboe } 1539ed29b0b4SJens Axboe req->io_task_work.func(req, locked); 1540ed29b0b4SJens Axboe node = next; 1541ed29b0b4SJens Axboe } while (node); 1542ed29b0b4SJens Axboe } 1543ed29b0b4SJens Axboe 1544c9f06aa7SJens Axboe void tctx_task_work(struct callback_head *cb) 1545ed29b0b4SJens Axboe { 1546ed29b0b4SJens Axboe bool uring_locked = false; 1547ed29b0b4SJens Axboe struct io_ring_ctx *ctx = NULL; 1548ed29b0b4SJens Axboe struct io_uring_task *tctx = container_of(cb, struct io_uring_task, 1549ed29b0b4SJens Axboe task_work); 1550ed29b0b4SJens Axboe 1551ed29b0b4SJens Axboe while (1) { 1552ed29b0b4SJens Axboe struct io_wq_work_node *node1, *node2; 1553ed29b0b4SJens Axboe 1554ed29b0b4SJens Axboe spin_lock_irq(&tctx->task_lock); 1555ed29b0b4SJens Axboe node1 = tctx->prio_task_list.first; 1556ed29b0b4SJens Axboe node2 = tctx->task_list.first; 1557ed29b0b4SJens Axboe INIT_WQ_LIST(&tctx->task_list); 1558ed29b0b4SJens Axboe INIT_WQ_LIST(&tctx->prio_task_list); 1559ed29b0b4SJens Axboe if (!node2 && !node1) 1560ed29b0b4SJens Axboe tctx->task_running = false; 1561ed29b0b4SJens Axboe spin_unlock_irq(&tctx->task_lock); 1562ed29b0b4SJens Axboe if (!node2 && !node1) 1563ed29b0b4SJens Axboe break; 1564ed29b0b4SJens Axboe 1565ed29b0b4SJens Axboe if (node1) 1566ed29b0b4SJens Axboe handle_prev_tw_list(node1, &ctx, &uring_locked); 1567ed29b0b4SJens Axboe if (node2) 1568ed29b0b4SJens Axboe handle_tw_list(node2, &ctx, &uring_locked); 1569ed29b0b4SJens Axboe cond_resched(); 1570ed29b0b4SJens Axboe 1571ed29b0b4SJens Axboe if (data_race(!tctx->task_list.first) && 1572ed29b0b4SJens Axboe data_race(!tctx->prio_task_list.first) && uring_locked) 1573ed29b0b4SJens Axboe io_submit_flush_completions(ctx); 1574ed29b0b4SJens Axboe } 1575ed29b0b4SJens Axboe 1576ed29b0b4SJens Axboe ctx_flush_and_put(ctx, &uring_locked); 1577ed29b0b4SJens Axboe 1578ed29b0b4SJens Axboe /* relaxed read is enough as only the task itself sets ->in_idle */ 1579ed29b0b4SJens Axboe if (unlikely(atomic_read(&tctx->in_idle))) 1580ed29b0b4SJens Axboe io_uring_drop_tctx_refs(current); 1581ed29b0b4SJens Axboe } 1582ed29b0b4SJens Axboe 1583ed29b0b4SJens Axboe static void __io_req_task_work_add(struct io_kiocb *req, 1584ed29b0b4SJens Axboe struct io_uring_task *tctx, 1585ed29b0b4SJens Axboe struct io_wq_work_list *list) 1586ed29b0b4SJens Axboe { 1587ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 1588ed29b0b4SJens Axboe struct io_wq_work_node *node; 1589ed29b0b4SJens Axboe unsigned long flags; 1590ed29b0b4SJens Axboe bool running; 1591ed29b0b4SJens Axboe 1592ed29b0b4SJens Axboe spin_lock_irqsave(&tctx->task_lock, flags); 1593ed29b0b4SJens Axboe wq_list_add_tail(&req->io_task_work.node, list); 1594ed29b0b4SJens Axboe running = tctx->task_running; 1595ed29b0b4SJens Axboe if (!running) 1596ed29b0b4SJens Axboe tctx->task_running = true; 1597ed29b0b4SJens Axboe spin_unlock_irqrestore(&tctx->task_lock, flags); 1598ed29b0b4SJens Axboe 1599ed29b0b4SJens Axboe /* task_work already pending, we're done */ 1600ed29b0b4SJens Axboe if (running) 1601ed29b0b4SJens Axboe return; 1602ed29b0b4SJens Axboe 1603ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_TASKRUN_FLAG) 1604ed29b0b4SJens Axboe atomic_or(IORING_SQ_TASKRUN, &ctx->rings->sq_flags); 1605ed29b0b4SJens Axboe 1606ed29b0b4SJens Axboe if (likely(!task_work_add(req->task, &tctx->task_work, ctx->notify_method))) 1607ed29b0b4SJens Axboe return; 1608ed29b0b4SJens Axboe 1609ed29b0b4SJens Axboe spin_lock_irqsave(&tctx->task_lock, flags); 1610ed29b0b4SJens Axboe tctx->task_running = false; 1611ed29b0b4SJens Axboe node = wq_list_merge(&tctx->prio_task_list, &tctx->task_list); 1612ed29b0b4SJens Axboe spin_unlock_irqrestore(&tctx->task_lock, flags); 1613ed29b0b4SJens Axboe 1614ed29b0b4SJens Axboe while (node) { 1615ed29b0b4SJens Axboe req = container_of(node, struct io_kiocb, io_task_work.node); 1616ed29b0b4SJens Axboe node = node->next; 1617ed29b0b4SJens Axboe if (llist_add(&req->io_task_work.fallback_node, 1618ed29b0b4SJens Axboe &req->ctx->fallback_llist)) 1619ed29b0b4SJens Axboe schedule_delayed_work(&req->ctx->fallback_work, 1); 1620ed29b0b4SJens Axboe } 1621ed29b0b4SJens Axboe } 1622ed29b0b4SJens Axboe 162399f15d8dSJens Axboe void io_req_task_work_add(struct io_kiocb *req) 1624ed29b0b4SJens Axboe { 1625ed29b0b4SJens Axboe struct io_uring_task *tctx = req->task->io_uring; 1626ed29b0b4SJens Axboe 1627ed29b0b4SJens Axboe __io_req_task_work_add(req, tctx, &tctx->task_list); 1628ed29b0b4SJens Axboe } 1629ed29b0b4SJens Axboe 1630ed29b0b4SJens Axboe static void io_req_task_prio_work_add(struct io_kiocb *req) 1631ed29b0b4SJens Axboe { 1632ed29b0b4SJens Axboe struct io_uring_task *tctx = req->task->io_uring; 1633ed29b0b4SJens Axboe 1634ed29b0b4SJens Axboe if (req->ctx->flags & IORING_SETUP_SQPOLL) 1635ed29b0b4SJens Axboe __io_req_task_work_add(req, tctx, &tctx->prio_task_list); 1636ed29b0b4SJens Axboe else 1637ed29b0b4SJens Axboe __io_req_task_work_add(req, tctx, &tctx->task_list); 1638ed29b0b4SJens Axboe } 1639ed29b0b4SJens Axboe 1640ed29b0b4SJens Axboe static void io_req_tw_post(struct io_kiocb *req, bool *locked) 1641ed29b0b4SJens Axboe { 164297b388d7SJens Axboe io_req_complete_post(req); 1643ed29b0b4SJens Axboe } 1644ed29b0b4SJens Axboe 164559915143SJens Axboe void io_req_tw_post_queue(struct io_kiocb *req, s32 res, u32 cflags) 1646ed29b0b4SJens Axboe { 164797b388d7SJens Axboe io_req_set_res(req, res, cflags); 1648ed29b0b4SJens Axboe req->io_task_work.func = io_req_tw_post; 1649ed29b0b4SJens Axboe io_req_task_work_add(req); 1650ed29b0b4SJens Axboe } 1651ed29b0b4SJens Axboe 1652ed29b0b4SJens Axboe static void io_req_task_cancel(struct io_kiocb *req, bool *locked) 1653ed29b0b4SJens Axboe { 1654ed29b0b4SJens Axboe /* not needed for normal modes, but SQPOLL depends on it */ 1655ed29b0b4SJens Axboe io_tw_lock(req->ctx, locked); 1656ed29b0b4SJens Axboe io_req_complete_failed(req, req->cqe.res); 1657ed29b0b4SJens Axboe } 1658ed29b0b4SJens Axboe 1659ed29b0b4SJens Axboe static void io_req_task_submit(struct io_kiocb *req, bool *locked) 1660ed29b0b4SJens Axboe { 1661ed29b0b4SJens Axboe io_tw_lock(req->ctx, locked); 1662ed29b0b4SJens Axboe /* req->task == current here, checking PF_EXITING is safe */ 1663ed29b0b4SJens Axboe if (likely(!(req->task->flags & PF_EXITING))) 1664ed29b0b4SJens Axboe io_queue_sqe(req); 1665ed29b0b4SJens Axboe else 1666ed29b0b4SJens Axboe io_req_complete_failed(req, -EFAULT); 1667ed29b0b4SJens Axboe } 1668ed29b0b4SJens Axboe 166959915143SJens Axboe void io_req_task_queue_fail(struct io_kiocb *req, int ret) 1670ed29b0b4SJens Axboe { 167197b388d7SJens Axboe io_req_set_res(req, ret, 0); 1672ed29b0b4SJens Axboe req->io_task_work.func = io_req_task_cancel; 1673ed29b0b4SJens Axboe io_req_task_work_add(req); 1674ed29b0b4SJens Axboe } 1675ed29b0b4SJens Axboe 1676ed29b0b4SJens Axboe static void io_req_task_queue(struct io_kiocb *req) 1677ed29b0b4SJens Axboe { 1678ed29b0b4SJens Axboe req->io_task_work.func = io_req_task_submit; 1679ed29b0b4SJens Axboe io_req_task_work_add(req); 1680ed29b0b4SJens Axboe } 1681ed29b0b4SJens Axboe 1682ed29b0b4SJens Axboe static void io_req_task_queue_reissue(struct io_kiocb *req) 1683ed29b0b4SJens Axboe { 1684ed29b0b4SJens Axboe req->io_task_work.func = io_queue_iowq; 1685ed29b0b4SJens Axboe io_req_task_work_add(req); 1686ed29b0b4SJens Axboe } 1687ed29b0b4SJens Axboe 168859915143SJens Axboe void io_queue_next(struct io_kiocb *req) 1689ed29b0b4SJens Axboe { 1690ed29b0b4SJens Axboe struct io_kiocb *nxt = io_req_find_next(req); 1691ed29b0b4SJens Axboe 1692ed29b0b4SJens Axboe if (nxt) 1693ed29b0b4SJens Axboe io_req_task_queue(nxt); 1694ed29b0b4SJens Axboe } 1695ed29b0b4SJens Axboe 1696ed29b0b4SJens Axboe static void io_free_batch_list(struct io_ring_ctx *ctx, 1697ed29b0b4SJens Axboe struct io_wq_work_node *node) 1698ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 1699ed29b0b4SJens Axboe { 1700ed29b0b4SJens Axboe struct task_struct *task = NULL; 1701ed29b0b4SJens Axboe int task_refs = 0; 1702ed29b0b4SJens Axboe 1703ed29b0b4SJens Axboe do { 1704ed29b0b4SJens Axboe struct io_kiocb *req = container_of(node, struct io_kiocb, 1705ed29b0b4SJens Axboe comp_list); 1706ed29b0b4SJens Axboe 1707ed29b0b4SJens Axboe if (unlikely(req->flags & IO_REQ_CLEAN_SLOW_FLAGS)) { 1708ed29b0b4SJens Axboe if (req->flags & REQ_F_REFCOUNT) { 1709ed29b0b4SJens Axboe node = req->comp_list.next; 1710ed29b0b4SJens Axboe if (!req_ref_put_and_test(req)) 1711ed29b0b4SJens Axboe continue; 1712ed29b0b4SJens Axboe } 1713ed29b0b4SJens Axboe if ((req->flags & REQ_F_POLLED) && req->apoll) { 1714ed29b0b4SJens Axboe struct async_poll *apoll = req->apoll; 1715ed29b0b4SJens Axboe 1716ed29b0b4SJens Axboe if (apoll->double_poll) 1717ed29b0b4SJens Axboe kfree(apoll->double_poll); 1718ed29b0b4SJens Axboe list_add(&apoll->poll.wait.entry, 1719ed29b0b4SJens Axboe &ctx->apoll_cache); 1720ed29b0b4SJens Axboe req->flags &= ~REQ_F_POLLED; 1721ed29b0b4SJens Axboe } 1722ed29b0b4SJens Axboe if (req->flags & IO_REQ_LINK_FLAGS) 1723ed29b0b4SJens Axboe io_queue_next(req); 1724ed29b0b4SJens Axboe if (unlikely(req->flags & IO_REQ_CLEAN_FLAGS)) 1725ed29b0b4SJens Axboe io_clean_op(req); 1726ed29b0b4SJens Axboe } 1727ed29b0b4SJens Axboe if (!(req->flags & REQ_F_FIXED_FILE)) 1728ed29b0b4SJens Axboe io_put_file(req->file); 1729ed29b0b4SJens Axboe 1730ed29b0b4SJens Axboe io_req_put_rsrc_locked(req, ctx); 1731ed29b0b4SJens Axboe 1732ed29b0b4SJens Axboe if (req->task != task) { 1733ed29b0b4SJens Axboe if (task) 1734ed29b0b4SJens Axboe io_put_task(task, task_refs); 1735ed29b0b4SJens Axboe task = req->task; 1736ed29b0b4SJens Axboe task_refs = 0; 1737ed29b0b4SJens Axboe } 1738ed29b0b4SJens Axboe task_refs++; 1739ed29b0b4SJens Axboe node = req->comp_list.next; 1740ed29b0b4SJens Axboe io_req_add_to_cache(req, ctx); 1741ed29b0b4SJens Axboe } while (node); 1742ed29b0b4SJens Axboe 1743ed29b0b4SJens Axboe if (task) 1744ed29b0b4SJens Axboe io_put_task(task, task_refs); 1745ed29b0b4SJens Axboe } 1746ed29b0b4SJens Axboe 1747ed29b0b4SJens Axboe static void __io_submit_flush_completions(struct io_ring_ctx *ctx) 1748ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 1749ed29b0b4SJens Axboe { 1750ed29b0b4SJens Axboe struct io_wq_work_node *node, *prev; 1751ed29b0b4SJens Axboe struct io_submit_state *state = &ctx->submit_state; 1752ed29b0b4SJens Axboe 1753ed29b0b4SJens Axboe if (state->flush_cqes) { 1754ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 1755ed29b0b4SJens Axboe wq_list_for_each(node, prev, &state->compl_reqs) { 1756ed29b0b4SJens Axboe struct io_kiocb *req = container_of(node, struct io_kiocb, 1757ed29b0b4SJens Axboe comp_list); 1758ed29b0b4SJens Axboe 1759ed29b0b4SJens Axboe if (!(req->flags & REQ_F_CQE_SKIP)) 1760ed29b0b4SJens Axboe __io_fill_cqe_req(ctx, req); 1761ed29b0b4SJens Axboe } 1762ed29b0b4SJens Axboe 1763ed29b0b4SJens Axboe io_commit_cqring(ctx); 1764ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 1765ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 1766ed29b0b4SJens Axboe state->flush_cqes = false; 1767ed29b0b4SJens Axboe } 1768ed29b0b4SJens Axboe 1769ed29b0b4SJens Axboe io_free_batch_list(ctx, state->compl_reqs.first); 1770ed29b0b4SJens Axboe INIT_WQ_LIST(&state->compl_reqs); 1771ed29b0b4SJens Axboe } 1772ed29b0b4SJens Axboe 1773ed29b0b4SJens Axboe /* 1774ed29b0b4SJens Axboe * Drop reference to request, return next in chain (if there is one) if this 1775ed29b0b4SJens Axboe * was the last reference to this request. 1776ed29b0b4SJens Axboe */ 1777ed29b0b4SJens Axboe static inline struct io_kiocb *io_put_req_find_next(struct io_kiocb *req) 1778ed29b0b4SJens Axboe { 1779ed29b0b4SJens Axboe struct io_kiocb *nxt = NULL; 1780ed29b0b4SJens Axboe 1781ed29b0b4SJens Axboe if (req_ref_put_and_test(req)) { 1782ed29b0b4SJens Axboe if (unlikely(req->flags & IO_REQ_LINK_FLAGS)) 1783ed29b0b4SJens Axboe nxt = io_req_find_next(req); 1784ed29b0b4SJens Axboe io_free_req(req); 1785ed29b0b4SJens Axboe } 1786ed29b0b4SJens Axboe return nxt; 1787ed29b0b4SJens Axboe } 1788ed29b0b4SJens Axboe 1789ed29b0b4SJens Axboe static unsigned io_cqring_events(struct io_ring_ctx *ctx) 1790ed29b0b4SJens Axboe { 1791ed29b0b4SJens Axboe /* See comment at the top of this file */ 1792ed29b0b4SJens Axboe smp_rmb(); 1793ed29b0b4SJens Axboe return __io_cqring_events(ctx); 1794ed29b0b4SJens Axboe } 1795ed29b0b4SJens Axboe 179617437f31SJens Axboe int io_do_iopoll(struct io_ring_ctx *ctx, bool force_nonspin) 1797ed29b0b4SJens Axboe { 1798ed29b0b4SJens Axboe struct io_wq_work_node *pos, *start, *prev; 1799ed29b0b4SJens Axboe unsigned int poll_flags = BLK_POLL_NOSLEEP; 1800ed29b0b4SJens Axboe DEFINE_IO_COMP_BATCH(iob); 1801ed29b0b4SJens Axboe int nr_events = 0; 1802ed29b0b4SJens Axboe 1803ed29b0b4SJens Axboe /* 1804ed29b0b4SJens Axboe * Only spin for completions if we don't have multiple devices hanging 1805ed29b0b4SJens Axboe * off our complete list. 1806ed29b0b4SJens Axboe */ 1807ed29b0b4SJens Axboe if (ctx->poll_multi_queue || force_nonspin) 1808ed29b0b4SJens Axboe poll_flags |= BLK_POLL_ONESHOT; 1809ed29b0b4SJens Axboe 1810ed29b0b4SJens Axboe wq_list_for_each(pos, start, &ctx->iopoll_list) { 1811ed29b0b4SJens Axboe struct io_kiocb *req = container_of(pos, struct io_kiocb, comp_list); 18123c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 1813ed29b0b4SJens Axboe int ret; 1814ed29b0b4SJens Axboe 1815ed29b0b4SJens Axboe /* 1816ed29b0b4SJens Axboe * Move completed and retryable entries to our local lists. 1817ed29b0b4SJens Axboe * If we find a request that requires polling, break out 1818ed29b0b4SJens Axboe * and complete those lists first, if we have entries there. 1819ed29b0b4SJens Axboe */ 1820ed29b0b4SJens Axboe if (READ_ONCE(req->iopoll_completed)) 1821ed29b0b4SJens Axboe break; 1822ed29b0b4SJens Axboe 18233c306fb2SJens Axboe ret = rw->kiocb.ki_filp->f_op->iopoll(&rw->kiocb, &iob, poll_flags); 1824ed29b0b4SJens Axboe if (unlikely(ret < 0)) 1825ed29b0b4SJens Axboe return ret; 1826ed29b0b4SJens Axboe else if (ret) 1827ed29b0b4SJens Axboe poll_flags |= BLK_POLL_ONESHOT; 1828ed29b0b4SJens Axboe 1829ed29b0b4SJens Axboe /* iopoll may have completed current req */ 1830ed29b0b4SJens Axboe if (!rq_list_empty(iob.req_list) || 1831ed29b0b4SJens Axboe READ_ONCE(req->iopoll_completed)) 1832ed29b0b4SJens Axboe break; 1833ed29b0b4SJens Axboe } 1834ed29b0b4SJens Axboe 1835ed29b0b4SJens Axboe if (!rq_list_empty(iob.req_list)) 1836ed29b0b4SJens Axboe iob.complete(&iob); 1837ed29b0b4SJens Axboe else if (!pos) 1838ed29b0b4SJens Axboe return 0; 1839ed29b0b4SJens Axboe 1840ed29b0b4SJens Axboe prev = start; 1841ed29b0b4SJens Axboe wq_list_for_each_resume(pos, prev) { 1842ed29b0b4SJens Axboe struct io_kiocb *req = container_of(pos, struct io_kiocb, comp_list); 1843ed29b0b4SJens Axboe 1844ed29b0b4SJens Axboe /* order with io_complete_rw_iopoll(), e.g. ->result updates */ 1845ed29b0b4SJens Axboe if (!smp_load_acquire(&req->iopoll_completed)) 1846ed29b0b4SJens Axboe break; 1847ed29b0b4SJens Axboe nr_events++; 1848ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_CQE_SKIP)) 1849ed29b0b4SJens Axboe continue; 1850ed29b0b4SJens Axboe 1851ed29b0b4SJens Axboe req->cqe.flags = io_put_kbuf(req, 0); 1852ed29b0b4SJens Axboe __io_fill_cqe_req(req->ctx, req); 1853ed29b0b4SJens Axboe } 1854ed29b0b4SJens Axboe 1855ed29b0b4SJens Axboe if (unlikely(!nr_events)) 1856ed29b0b4SJens Axboe return 0; 1857ed29b0b4SJens Axboe 1858ed29b0b4SJens Axboe io_commit_cqring(ctx); 1859ed29b0b4SJens Axboe io_cqring_ev_posted_iopoll(ctx); 1860ed29b0b4SJens Axboe pos = start ? start->next : ctx->iopoll_list.first; 1861ed29b0b4SJens Axboe wq_list_cut(&ctx->iopoll_list, prev, start); 1862ed29b0b4SJens Axboe io_free_batch_list(ctx, pos); 1863ed29b0b4SJens Axboe return nr_events; 1864ed29b0b4SJens Axboe } 1865ed29b0b4SJens Axboe 1866ed29b0b4SJens Axboe /* 1867ed29b0b4SJens Axboe * We can't just wait for polled events to come to us, we have to actively 1868ed29b0b4SJens Axboe * find and complete them. 1869ed29b0b4SJens Axboe */ 1870ed29b0b4SJens Axboe static __cold void io_iopoll_try_reap_events(struct io_ring_ctx *ctx) 1871ed29b0b4SJens Axboe { 1872ed29b0b4SJens Axboe if (!(ctx->flags & IORING_SETUP_IOPOLL)) 1873ed29b0b4SJens Axboe return; 1874ed29b0b4SJens Axboe 1875ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 1876ed29b0b4SJens Axboe while (!wq_list_empty(&ctx->iopoll_list)) { 1877ed29b0b4SJens Axboe /* let it sleep and repeat later if can't complete a request */ 1878ed29b0b4SJens Axboe if (io_do_iopoll(ctx, true) == 0) 1879ed29b0b4SJens Axboe break; 1880ed29b0b4SJens Axboe /* 1881ed29b0b4SJens Axboe * Ensure we allow local-to-the-cpu processing to take place, 1882ed29b0b4SJens Axboe * in this case we need to ensure that we reap all events. 1883ed29b0b4SJens Axboe * Also let task_work, etc. to progress by releasing the mutex 1884ed29b0b4SJens Axboe */ 1885ed29b0b4SJens Axboe if (need_resched()) { 1886ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 1887ed29b0b4SJens Axboe cond_resched(); 1888ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 1889ed29b0b4SJens Axboe } 1890ed29b0b4SJens Axboe } 1891ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 1892ed29b0b4SJens Axboe } 1893ed29b0b4SJens Axboe 1894ed29b0b4SJens Axboe static int io_iopoll_check(struct io_ring_ctx *ctx, long min) 1895ed29b0b4SJens Axboe { 1896ed29b0b4SJens Axboe unsigned int nr_events = 0; 1897ed29b0b4SJens Axboe int ret = 0; 1898ed29b0b4SJens Axboe unsigned long check_cq; 1899ed29b0b4SJens Axboe 1900ed29b0b4SJens Axboe /* 1901ed29b0b4SJens Axboe * Don't enter poll loop if we already have events pending. 1902ed29b0b4SJens Axboe * If we do, we can potentially be spinning for commands that 1903ed29b0b4SJens Axboe * already triggered a CQE (eg in error). 1904ed29b0b4SJens Axboe */ 1905ed29b0b4SJens Axboe check_cq = READ_ONCE(ctx->check_cq); 1906ed29b0b4SJens Axboe if (check_cq & BIT(IO_CHECK_CQ_OVERFLOW_BIT)) 1907ed29b0b4SJens Axboe __io_cqring_overflow_flush(ctx, false); 1908ed29b0b4SJens Axboe if (io_cqring_events(ctx)) 1909ed29b0b4SJens Axboe return 0; 1910ed29b0b4SJens Axboe 1911ed29b0b4SJens Axboe /* 1912ed29b0b4SJens Axboe * Similarly do not spin if we have not informed the user of any 1913ed29b0b4SJens Axboe * dropped CQE. 1914ed29b0b4SJens Axboe */ 1915ed29b0b4SJens Axboe if (unlikely(check_cq & BIT(IO_CHECK_CQ_DROPPED_BIT))) 1916ed29b0b4SJens Axboe return -EBADR; 1917ed29b0b4SJens Axboe 1918ed29b0b4SJens Axboe do { 1919ed29b0b4SJens Axboe /* 1920ed29b0b4SJens Axboe * If a submit got punted to a workqueue, we can have the 1921ed29b0b4SJens Axboe * application entering polling for a command before it gets 1922ed29b0b4SJens Axboe * issued. That app will hold the uring_lock for the duration 1923ed29b0b4SJens Axboe * of the poll right here, so we need to take a breather every 1924ed29b0b4SJens Axboe * now and then to ensure that the issue has a chance to add 1925ed29b0b4SJens Axboe * the poll to the issued list. Otherwise we can spin here 1926ed29b0b4SJens Axboe * forever, while the workqueue is stuck trying to acquire the 1927ed29b0b4SJens Axboe * very same mutex. 1928ed29b0b4SJens Axboe */ 1929ed29b0b4SJens Axboe if (wq_list_empty(&ctx->iopoll_list)) { 1930ed29b0b4SJens Axboe u32 tail = ctx->cached_cq_tail; 1931ed29b0b4SJens Axboe 1932ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 1933ed29b0b4SJens Axboe io_run_task_work(); 1934ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 1935ed29b0b4SJens Axboe 1936ed29b0b4SJens Axboe /* some requests don't go through iopoll_list */ 1937ed29b0b4SJens Axboe if (tail != ctx->cached_cq_tail || 1938ed29b0b4SJens Axboe wq_list_empty(&ctx->iopoll_list)) 1939ed29b0b4SJens Axboe break; 1940ed29b0b4SJens Axboe } 1941ed29b0b4SJens Axboe ret = io_do_iopoll(ctx, !min); 1942ed29b0b4SJens Axboe if (ret < 0) 1943ed29b0b4SJens Axboe break; 1944ed29b0b4SJens Axboe nr_events += ret; 1945ed29b0b4SJens Axboe ret = 0; 1946ed29b0b4SJens Axboe } while (nr_events < min && !need_resched()); 1947ed29b0b4SJens Axboe 1948ed29b0b4SJens Axboe return ret; 1949ed29b0b4SJens Axboe } 1950ed29b0b4SJens Axboe 1951ed29b0b4SJens Axboe static void kiocb_end_write(struct io_kiocb *req) 1952ed29b0b4SJens Axboe { 1953ed29b0b4SJens Axboe /* 1954ed29b0b4SJens Axboe * Tell lockdep we inherited freeze protection from submission 1955ed29b0b4SJens Axboe * thread. 1956ed29b0b4SJens Axboe */ 1957ed29b0b4SJens Axboe if (req->flags & REQ_F_ISREG) { 1958ed29b0b4SJens Axboe struct super_block *sb = file_inode(req->file)->i_sb; 1959ed29b0b4SJens Axboe 1960ed29b0b4SJens Axboe __sb_writers_acquired(sb, SB_FREEZE_WRITE); 1961ed29b0b4SJens Axboe sb_end_write(sb); 1962ed29b0b4SJens Axboe } 1963ed29b0b4SJens Axboe } 1964ed29b0b4SJens Axboe 1965ed29b0b4SJens Axboe #ifdef CONFIG_BLOCK 1966ed29b0b4SJens Axboe static bool io_resubmit_prep(struct io_kiocb *req) 1967ed29b0b4SJens Axboe { 19683c306fb2SJens Axboe struct io_async_rw *io = req->async_data; 1969ed29b0b4SJens Axboe 1970ed29b0b4SJens Axboe if (!req_has_async_data(req)) 1971ed29b0b4SJens Axboe return !io_req_prep_async(req); 19723c306fb2SJens Axboe iov_iter_restore(&io->s.iter, &io->s.iter_state); 1973ed29b0b4SJens Axboe return true; 1974ed29b0b4SJens Axboe } 1975ed29b0b4SJens Axboe 1976ed29b0b4SJens Axboe static bool io_rw_should_reissue(struct io_kiocb *req) 1977ed29b0b4SJens Axboe { 1978ed29b0b4SJens Axboe umode_t mode = file_inode(req->file)->i_mode; 1979ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 1980ed29b0b4SJens Axboe 1981ed29b0b4SJens Axboe if (!S_ISBLK(mode) && !S_ISREG(mode)) 1982ed29b0b4SJens Axboe return false; 1983ed29b0b4SJens Axboe if ((req->flags & REQ_F_NOWAIT) || (io_wq_current_is_worker() && 1984ed29b0b4SJens Axboe !(ctx->flags & IORING_SETUP_IOPOLL))) 1985ed29b0b4SJens Axboe return false; 1986ed29b0b4SJens Axboe /* 1987ed29b0b4SJens Axboe * If ref is dying, we might be running poll reap from the exit work. 1988ed29b0b4SJens Axboe * Don't attempt to reissue from that path, just let it fail with 1989ed29b0b4SJens Axboe * -EAGAIN. 1990ed29b0b4SJens Axboe */ 1991ed29b0b4SJens Axboe if (percpu_ref_is_dying(&ctx->refs)) 1992ed29b0b4SJens Axboe return false; 1993ed29b0b4SJens Axboe /* 1994ed29b0b4SJens Axboe * Play it safe and assume not safe to re-import and reissue if we're 1995ed29b0b4SJens Axboe * not in the original thread group (or in task context). 1996ed29b0b4SJens Axboe */ 1997ed29b0b4SJens Axboe if (!same_thread_group(req->task, current) || !in_task()) 1998ed29b0b4SJens Axboe return false; 1999ed29b0b4SJens Axboe return true; 2000ed29b0b4SJens Axboe } 2001ed29b0b4SJens Axboe #else 2002ed29b0b4SJens Axboe static bool io_resubmit_prep(struct io_kiocb *req) 2003ed29b0b4SJens Axboe { 2004ed29b0b4SJens Axboe return false; 2005ed29b0b4SJens Axboe } 2006ed29b0b4SJens Axboe static bool io_rw_should_reissue(struct io_kiocb *req) 2007ed29b0b4SJens Axboe { 2008ed29b0b4SJens Axboe return false; 2009ed29b0b4SJens Axboe } 2010ed29b0b4SJens Axboe #endif 2011ed29b0b4SJens Axboe 2012ed29b0b4SJens Axboe static bool __io_complete_rw_common(struct io_kiocb *req, long res) 2013ed29b0b4SJens Axboe { 20143c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 20153c306fb2SJens Axboe 20163c306fb2SJens Axboe if (rw->kiocb.ki_flags & IOCB_WRITE) { 2017ed29b0b4SJens Axboe kiocb_end_write(req); 2018ed29b0b4SJens Axboe fsnotify_modify(req->file); 2019ed29b0b4SJens Axboe } else { 2020ed29b0b4SJens Axboe fsnotify_access(req->file); 2021ed29b0b4SJens Axboe } 2022ed29b0b4SJens Axboe if (unlikely(res != req->cqe.res)) { 2023ed29b0b4SJens Axboe if ((res == -EAGAIN || res == -EOPNOTSUPP) && 2024ed29b0b4SJens Axboe io_rw_should_reissue(req)) { 2025ed29b0b4SJens Axboe req->flags |= REQ_F_REISSUE | REQ_F_PARTIAL_IO; 2026ed29b0b4SJens Axboe return true; 2027ed29b0b4SJens Axboe } 2028ed29b0b4SJens Axboe req_set_fail(req); 2029ed29b0b4SJens Axboe req->cqe.res = res; 2030ed29b0b4SJens Axboe } 2031ed29b0b4SJens Axboe return false; 2032ed29b0b4SJens Axboe } 2033ed29b0b4SJens Axboe 203459915143SJens Axboe inline void io_req_task_complete(struct io_kiocb *req, bool *locked) 2035ed29b0b4SJens Axboe { 2036ed29b0b4SJens Axboe if (*locked) { 203797b388d7SJens Axboe req->cqe.flags |= io_put_kbuf(req, 0); 2038de23077eSJens Axboe req->flags |= REQ_F_COMPLETE_INLINE; 2039ed29b0b4SJens Axboe io_req_add_compl_list(req); 2040ed29b0b4SJens Axboe } else { 204197b388d7SJens Axboe req->cqe.flags |= io_put_kbuf(req, IO_URING_F_UNLOCKED); 204297b388d7SJens Axboe io_req_complete_post(req); 2043ed29b0b4SJens Axboe } 2044ed29b0b4SJens Axboe } 2045ed29b0b4SJens Axboe 2046ed29b0b4SJens Axboe static void __io_complete_rw(struct io_kiocb *req, long res, 2047ed29b0b4SJens Axboe unsigned int issue_flags) 2048ed29b0b4SJens Axboe { 2049ed29b0b4SJens Axboe if (__io_complete_rw_common(req, res)) 2050ed29b0b4SJens Axboe return; 205197b388d7SJens Axboe io_req_set_res(req, req->cqe.res, io_put_kbuf(req, issue_flags)); 205297b388d7SJens Axboe __io_req_complete(req, issue_flags); 2053ed29b0b4SJens Axboe } 2054ed29b0b4SJens Axboe 2055ed29b0b4SJens Axboe static void io_complete_rw(struct kiocb *kiocb, long res) 2056ed29b0b4SJens Axboe { 20573c306fb2SJens Axboe struct io_rw *rw = container_of(kiocb, struct io_rw, kiocb); 20583c306fb2SJens Axboe struct io_kiocb *req = cmd_to_io_kiocb(rw); 2059ed29b0b4SJens Axboe 2060ed29b0b4SJens Axboe if (__io_complete_rw_common(req, res)) 2061ed29b0b4SJens Axboe return; 206297b388d7SJens Axboe io_req_set_res(req, res, 0); 2063ed29b0b4SJens Axboe req->io_task_work.func = io_req_task_complete; 2064ed29b0b4SJens Axboe io_req_task_prio_work_add(req); 2065ed29b0b4SJens Axboe } 2066ed29b0b4SJens Axboe 2067ed29b0b4SJens Axboe static void io_complete_rw_iopoll(struct kiocb *kiocb, long res) 2068ed29b0b4SJens Axboe { 20693c306fb2SJens Axboe struct io_rw *rw = container_of(kiocb, struct io_rw, kiocb); 20703c306fb2SJens Axboe struct io_kiocb *req = cmd_to_io_kiocb(rw); 2071ed29b0b4SJens Axboe 2072ed29b0b4SJens Axboe if (kiocb->ki_flags & IOCB_WRITE) 2073ed29b0b4SJens Axboe kiocb_end_write(req); 2074ed29b0b4SJens Axboe if (unlikely(res != req->cqe.res)) { 2075ed29b0b4SJens Axboe if (res == -EAGAIN && io_rw_should_reissue(req)) { 2076ed29b0b4SJens Axboe req->flags |= REQ_F_REISSUE | REQ_F_PARTIAL_IO; 2077ed29b0b4SJens Axboe return; 2078ed29b0b4SJens Axboe } 2079ed29b0b4SJens Axboe req->cqe.res = res; 2080ed29b0b4SJens Axboe } 2081ed29b0b4SJens Axboe 2082ed29b0b4SJens Axboe /* order with io_iopoll_complete() checking ->iopoll_completed */ 2083ed29b0b4SJens Axboe smp_store_release(&req->iopoll_completed, 1); 2084ed29b0b4SJens Axboe } 2085ed29b0b4SJens Axboe 2086ed29b0b4SJens Axboe /* 2087ed29b0b4SJens Axboe * After the iocb has been issued, it's safe to be found on the poll list. 2088ed29b0b4SJens Axboe * Adding the kiocb to the list AFTER submission ensures that we don't 2089ed29b0b4SJens Axboe * find it from a io_do_iopoll() thread before the issuer is done 2090ed29b0b4SJens Axboe * accessing the kiocb cookie. 2091ed29b0b4SJens Axboe */ 2092ed29b0b4SJens Axboe static void io_iopoll_req_issued(struct io_kiocb *req, unsigned int issue_flags) 2093ed29b0b4SJens Axboe { 2094ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 2095ed29b0b4SJens Axboe const bool needs_lock = issue_flags & IO_URING_F_UNLOCKED; 2096ed29b0b4SJens Axboe 2097ed29b0b4SJens Axboe /* workqueue context doesn't hold uring_lock, grab it now */ 2098ed29b0b4SJens Axboe if (unlikely(needs_lock)) 2099ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 2100ed29b0b4SJens Axboe 2101ed29b0b4SJens Axboe /* 2102ed29b0b4SJens Axboe * Track whether we have multiple files in our lists. This will impact 2103ed29b0b4SJens Axboe * how we do polling eventually, not spinning if we're on potentially 2104ed29b0b4SJens Axboe * different devices. 2105ed29b0b4SJens Axboe */ 2106ed29b0b4SJens Axboe if (wq_list_empty(&ctx->iopoll_list)) { 2107ed29b0b4SJens Axboe ctx->poll_multi_queue = false; 2108ed29b0b4SJens Axboe } else if (!ctx->poll_multi_queue) { 2109ed29b0b4SJens Axboe struct io_kiocb *list_req; 2110ed29b0b4SJens Axboe 2111ed29b0b4SJens Axboe list_req = container_of(ctx->iopoll_list.first, struct io_kiocb, 2112ed29b0b4SJens Axboe comp_list); 2113ed29b0b4SJens Axboe if (list_req->file != req->file) 2114ed29b0b4SJens Axboe ctx->poll_multi_queue = true; 2115ed29b0b4SJens Axboe } 2116ed29b0b4SJens Axboe 2117ed29b0b4SJens Axboe /* 2118ed29b0b4SJens Axboe * For fast devices, IO may have already completed. If it has, add 2119ed29b0b4SJens Axboe * it to the front so we find it first. 2120ed29b0b4SJens Axboe */ 2121ed29b0b4SJens Axboe if (READ_ONCE(req->iopoll_completed)) 2122ed29b0b4SJens Axboe wq_list_add_head(&req->comp_list, &ctx->iopoll_list); 2123ed29b0b4SJens Axboe else 2124ed29b0b4SJens Axboe wq_list_add_tail(&req->comp_list, &ctx->iopoll_list); 2125ed29b0b4SJens Axboe 2126ed29b0b4SJens Axboe if (unlikely(needs_lock)) { 2127ed29b0b4SJens Axboe /* 2128ed29b0b4SJens Axboe * If IORING_SETUP_SQPOLL is enabled, sqes are either handle 2129ed29b0b4SJens Axboe * in sq thread task context or in io worker task context. If 2130ed29b0b4SJens Axboe * current task context is sq thread, we don't need to check 2131ed29b0b4SJens Axboe * whether should wake up sq thread. 2132ed29b0b4SJens Axboe */ 2133ed29b0b4SJens Axboe if ((ctx->flags & IORING_SETUP_SQPOLL) && 2134ed29b0b4SJens Axboe wq_has_sleeper(&ctx->sq_data->wait)) 2135ed29b0b4SJens Axboe wake_up(&ctx->sq_data->wait); 2136ed29b0b4SJens Axboe 2137ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 2138ed29b0b4SJens Axboe } 2139ed29b0b4SJens Axboe } 2140ed29b0b4SJens Axboe 2141ed29b0b4SJens Axboe static bool io_bdev_nowait(struct block_device *bdev) 2142ed29b0b4SJens Axboe { 2143ed29b0b4SJens Axboe return !bdev || blk_queue_nowait(bdev_get_queue(bdev)); 2144ed29b0b4SJens Axboe } 2145ed29b0b4SJens Axboe 2146ed29b0b4SJens Axboe /* 2147ed29b0b4SJens Axboe * If we tracked the file through the SCM inflight mechanism, we could support 2148ed29b0b4SJens Axboe * any file. For now, just ensure that anything potentially problematic is done 2149ed29b0b4SJens Axboe * inline. 2150ed29b0b4SJens Axboe */ 2151ed29b0b4SJens Axboe static bool __io_file_supports_nowait(struct file *file, umode_t mode) 2152ed29b0b4SJens Axboe { 2153ed29b0b4SJens Axboe if (S_ISBLK(mode)) { 2154ed29b0b4SJens Axboe if (IS_ENABLED(CONFIG_BLOCK) && 2155ed29b0b4SJens Axboe io_bdev_nowait(I_BDEV(file->f_mapping->host))) 2156ed29b0b4SJens Axboe return true; 2157ed29b0b4SJens Axboe return false; 2158ed29b0b4SJens Axboe } 2159ed29b0b4SJens Axboe if (S_ISSOCK(mode)) 2160ed29b0b4SJens Axboe return true; 2161ed29b0b4SJens Axboe if (S_ISREG(mode)) { 2162ed29b0b4SJens Axboe if (IS_ENABLED(CONFIG_BLOCK) && 2163ed29b0b4SJens Axboe io_bdev_nowait(file->f_inode->i_sb->s_bdev) && 2164e5550a14SJens Axboe !io_is_uring_fops(file)) 2165ed29b0b4SJens Axboe return true; 2166ed29b0b4SJens Axboe return false; 2167ed29b0b4SJens Axboe } 2168ed29b0b4SJens Axboe 2169ed29b0b4SJens Axboe /* any ->read/write should understand O_NONBLOCK */ 2170ed29b0b4SJens Axboe if (file->f_flags & O_NONBLOCK) 2171ed29b0b4SJens Axboe return true; 2172ed29b0b4SJens Axboe return file->f_mode & FMODE_NOWAIT; 2173ed29b0b4SJens Axboe } 2174ed29b0b4SJens Axboe 2175ed29b0b4SJens Axboe /* 2176ed29b0b4SJens Axboe * If we tracked the file through the SCM inflight mechanism, we could support 2177ed29b0b4SJens Axboe * any file. For now, just ensure that anything potentially problematic is done 2178ed29b0b4SJens Axboe * inline. 2179ed29b0b4SJens Axboe */ 2180a4ad4f74SJens Axboe unsigned int io_file_get_flags(struct file *file) 2181ed29b0b4SJens Axboe { 2182ed29b0b4SJens Axboe umode_t mode = file_inode(file)->i_mode; 2183ed29b0b4SJens Axboe unsigned int res = 0; 2184ed29b0b4SJens Axboe 2185ed29b0b4SJens Axboe if (S_ISREG(mode)) 2186ed29b0b4SJens Axboe res |= FFS_ISREG; 2187ed29b0b4SJens Axboe if (__io_file_supports_nowait(file, mode)) 2188ed29b0b4SJens Axboe res |= FFS_NOWAIT; 2189ed29b0b4SJens Axboe if (io_file_need_scm(file)) 2190ed29b0b4SJens Axboe res |= FFS_SCM; 2191ed29b0b4SJens Axboe return res; 2192ed29b0b4SJens Axboe } 2193ed29b0b4SJens Axboe 2194ed29b0b4SJens Axboe static inline bool io_file_supports_nowait(struct io_kiocb *req) 2195ed29b0b4SJens Axboe { 2196ed29b0b4SJens Axboe return req->flags & REQ_F_SUPPORT_NOWAIT; 2197ed29b0b4SJens Axboe } 2198ed29b0b4SJens Axboe 2199ed29b0b4SJens Axboe static int io_prep_rw(struct io_kiocb *req, const struct io_uring_sqe *sqe) 2200ed29b0b4SJens Axboe { 22013c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 2202ed29b0b4SJens Axboe unsigned ioprio; 2203ed29b0b4SJens Axboe int ret; 2204ed29b0b4SJens Axboe 22053c306fb2SJens Axboe rw->kiocb.ki_pos = READ_ONCE(sqe->off); 2206ed29b0b4SJens Axboe /* used for fixed read/write too - just read unconditionally */ 2207ed29b0b4SJens Axboe req->buf_index = READ_ONCE(sqe->buf_index); 2208ed29b0b4SJens Axboe 2209ed29b0b4SJens Axboe if (req->opcode == IORING_OP_READ_FIXED || 2210ed29b0b4SJens Axboe req->opcode == IORING_OP_WRITE_FIXED) { 2211ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 2212ed29b0b4SJens Axboe u16 index; 2213ed29b0b4SJens Axboe 2214ed29b0b4SJens Axboe if (unlikely(req->buf_index >= ctx->nr_user_bufs)) 2215ed29b0b4SJens Axboe return -EFAULT; 2216ed29b0b4SJens Axboe index = array_index_nospec(req->buf_index, ctx->nr_user_bufs); 2217ed29b0b4SJens Axboe req->imu = ctx->user_bufs[index]; 2218ed29b0b4SJens Axboe io_req_set_rsrc_node(req, ctx, 0); 2219ed29b0b4SJens Axboe } 2220ed29b0b4SJens Axboe 2221ed29b0b4SJens Axboe ioprio = READ_ONCE(sqe->ioprio); 2222ed29b0b4SJens Axboe if (ioprio) { 2223ed29b0b4SJens Axboe ret = ioprio_check_cap(ioprio); 2224ed29b0b4SJens Axboe if (ret) 2225ed29b0b4SJens Axboe return ret; 2226ed29b0b4SJens Axboe 22273c306fb2SJens Axboe rw->kiocb.ki_ioprio = ioprio; 2228ed29b0b4SJens Axboe } else { 22293c306fb2SJens Axboe rw->kiocb.ki_ioprio = get_current_ioprio(); 2230ed29b0b4SJens Axboe } 2231ed29b0b4SJens Axboe 22323c306fb2SJens Axboe rw->addr = READ_ONCE(sqe->addr); 22333c306fb2SJens Axboe rw->len = READ_ONCE(sqe->len); 22343c306fb2SJens Axboe rw->flags = READ_ONCE(sqe->rw_flags); 2235ed29b0b4SJens Axboe return 0; 2236ed29b0b4SJens Axboe } 2237ed29b0b4SJens Axboe 22384d4c9cffSJens Axboe static void io_readv_writev_cleanup(struct io_kiocb *req) 22394d4c9cffSJens Axboe { 22404d4c9cffSJens Axboe struct io_async_rw *io = req->async_data; 22414d4c9cffSJens Axboe 22424d4c9cffSJens Axboe kfree(io->free_iovec); 22434d4c9cffSJens Axboe } 22444d4c9cffSJens Axboe 2245ed29b0b4SJens Axboe static inline void io_rw_done(struct kiocb *kiocb, ssize_t ret) 2246ed29b0b4SJens Axboe { 2247ed29b0b4SJens Axboe switch (ret) { 2248ed29b0b4SJens Axboe case -EIOCBQUEUED: 2249ed29b0b4SJens Axboe break; 2250ed29b0b4SJens Axboe case -ERESTARTSYS: 2251ed29b0b4SJens Axboe case -ERESTARTNOINTR: 2252ed29b0b4SJens Axboe case -ERESTARTNOHAND: 2253ed29b0b4SJens Axboe case -ERESTART_RESTARTBLOCK: 2254ed29b0b4SJens Axboe /* 2255ed29b0b4SJens Axboe * We can't just restart the syscall, since previously 2256ed29b0b4SJens Axboe * submitted sqes may already be in progress. Just fail this 2257ed29b0b4SJens Axboe * IO with EINTR. 2258ed29b0b4SJens Axboe */ 2259ed29b0b4SJens Axboe ret = -EINTR; 2260ed29b0b4SJens Axboe fallthrough; 2261ed29b0b4SJens Axboe default: 2262ed29b0b4SJens Axboe kiocb->ki_complete(kiocb, ret); 2263ed29b0b4SJens Axboe } 2264ed29b0b4SJens Axboe } 2265ed29b0b4SJens Axboe 2266ed29b0b4SJens Axboe static inline loff_t *io_kiocb_update_pos(struct io_kiocb *req) 2267ed29b0b4SJens Axboe { 22683c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 2269ed29b0b4SJens Axboe 22703c306fb2SJens Axboe if (rw->kiocb.ki_pos != -1) 22713c306fb2SJens Axboe return &rw->kiocb.ki_pos; 2272ed29b0b4SJens Axboe 2273ed29b0b4SJens Axboe if (!(req->file->f_mode & FMODE_STREAM)) { 2274ed29b0b4SJens Axboe req->flags |= REQ_F_CUR_POS; 22753c306fb2SJens Axboe rw->kiocb.ki_pos = req->file->f_pos; 22763c306fb2SJens Axboe return &rw->kiocb.ki_pos; 2277ed29b0b4SJens Axboe } 2278ed29b0b4SJens Axboe 22793c306fb2SJens Axboe rw->kiocb.ki_pos = 0; 2280ed29b0b4SJens Axboe return NULL; 2281ed29b0b4SJens Axboe } 2282ed29b0b4SJens Axboe 2283ed29b0b4SJens Axboe static void kiocb_done(struct io_kiocb *req, ssize_t ret, 2284ed29b0b4SJens Axboe unsigned int issue_flags) 2285ed29b0b4SJens Axboe { 2286ed29b0b4SJens Axboe struct io_async_rw *io = req->async_data; 22873c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 2288ed29b0b4SJens Axboe 2289ed29b0b4SJens Axboe /* add previously done IO, if any */ 2290ed29b0b4SJens Axboe if (req_has_async_data(req) && io->bytes_done > 0) { 2291ed29b0b4SJens Axboe if (ret < 0) 2292ed29b0b4SJens Axboe ret = io->bytes_done; 2293ed29b0b4SJens Axboe else 2294ed29b0b4SJens Axboe ret += io->bytes_done; 2295ed29b0b4SJens Axboe } 2296ed29b0b4SJens Axboe 2297ed29b0b4SJens Axboe if (req->flags & REQ_F_CUR_POS) 22983c306fb2SJens Axboe req->file->f_pos = rw->kiocb.ki_pos; 22993c306fb2SJens Axboe if (ret >= 0 && (rw->kiocb.ki_complete == io_complete_rw)) 2300ed29b0b4SJens Axboe __io_complete_rw(req, ret, issue_flags); 2301ed29b0b4SJens Axboe else 23023c306fb2SJens Axboe io_rw_done(&rw->kiocb, ret); 2303ed29b0b4SJens Axboe 2304ed29b0b4SJens Axboe if (req->flags & REQ_F_REISSUE) { 2305ed29b0b4SJens Axboe req->flags &= ~REQ_F_REISSUE; 2306ed29b0b4SJens Axboe if (io_resubmit_prep(req)) 2307ed29b0b4SJens Axboe io_req_task_queue_reissue(req); 2308ed29b0b4SJens Axboe else 2309ed29b0b4SJens Axboe io_req_task_queue_fail(req, ret); 2310ed29b0b4SJens Axboe } 2311ed29b0b4SJens Axboe } 2312ed29b0b4SJens Axboe 23133c306fb2SJens Axboe static int __io_import_fixed(struct io_kiocb *req, int ddir, 23143c306fb2SJens Axboe struct iov_iter *iter, struct io_mapped_ubuf *imu) 2315ed29b0b4SJens Axboe { 23163c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 23173c306fb2SJens Axboe size_t len = rw->len; 23183c306fb2SJens Axboe u64 buf_end, buf_addr = rw->addr; 2319ed29b0b4SJens Axboe size_t offset; 2320ed29b0b4SJens Axboe 2321ed29b0b4SJens Axboe if (unlikely(check_add_overflow(buf_addr, (u64)len, &buf_end))) 2322ed29b0b4SJens Axboe return -EFAULT; 2323ed29b0b4SJens Axboe /* not inside the mapped region */ 2324ed29b0b4SJens Axboe if (unlikely(buf_addr < imu->ubuf || buf_end > imu->ubuf_end)) 2325ed29b0b4SJens Axboe return -EFAULT; 2326ed29b0b4SJens Axboe 2327ed29b0b4SJens Axboe /* 2328ed29b0b4SJens Axboe * May not be a start of buffer, set size appropriately 2329ed29b0b4SJens Axboe * and advance us to the beginning. 2330ed29b0b4SJens Axboe */ 2331ed29b0b4SJens Axboe offset = buf_addr - imu->ubuf; 23323c306fb2SJens Axboe iov_iter_bvec(iter, ddir, imu->bvec, imu->nr_bvecs, offset + len); 2333ed29b0b4SJens Axboe 2334ed29b0b4SJens Axboe if (offset) { 2335ed29b0b4SJens Axboe /* 2336ed29b0b4SJens Axboe * Don't use iov_iter_advance() here, as it's really slow for 2337ed29b0b4SJens Axboe * using the latter parts of a big fixed buffer - it iterates 2338ed29b0b4SJens Axboe * over each segment manually. We can cheat a bit here, because 2339ed29b0b4SJens Axboe * we know that: 2340ed29b0b4SJens Axboe * 2341ed29b0b4SJens Axboe * 1) it's a BVEC iter, we set it up 2342ed29b0b4SJens Axboe * 2) all bvecs are PAGE_SIZE in size, except potentially the 2343ed29b0b4SJens Axboe * first and last bvec 2344ed29b0b4SJens Axboe * 2345ed29b0b4SJens Axboe * So just find our index, and adjust the iterator afterwards. 2346ed29b0b4SJens Axboe * If the offset is within the first bvec (or the whole first 2347ed29b0b4SJens Axboe * bvec, just use iov_iter_advance(). This makes it easier 2348ed29b0b4SJens Axboe * since we can just skip the first segment, which may not 2349ed29b0b4SJens Axboe * be PAGE_SIZE aligned. 2350ed29b0b4SJens Axboe */ 2351ed29b0b4SJens Axboe const struct bio_vec *bvec = imu->bvec; 2352ed29b0b4SJens Axboe 2353ed29b0b4SJens Axboe if (offset <= bvec->bv_len) { 2354ed29b0b4SJens Axboe iov_iter_advance(iter, offset); 2355ed29b0b4SJens Axboe } else { 2356ed29b0b4SJens Axboe unsigned long seg_skip; 2357ed29b0b4SJens Axboe 2358ed29b0b4SJens Axboe /* skip first vec */ 2359ed29b0b4SJens Axboe offset -= bvec->bv_len; 2360ed29b0b4SJens Axboe seg_skip = 1 + (offset >> PAGE_SHIFT); 2361ed29b0b4SJens Axboe 2362ed29b0b4SJens Axboe iter->bvec = bvec + seg_skip; 2363ed29b0b4SJens Axboe iter->nr_segs -= seg_skip; 2364ed29b0b4SJens Axboe iter->count -= bvec->bv_len + offset; 2365ed29b0b4SJens Axboe iter->iov_offset = offset & ~PAGE_MASK; 2366ed29b0b4SJens Axboe } 2367ed29b0b4SJens Axboe } 2368ed29b0b4SJens Axboe 2369ed29b0b4SJens Axboe return 0; 2370ed29b0b4SJens Axboe } 2371ed29b0b4SJens Axboe 2372ed29b0b4SJens Axboe static int io_import_fixed(struct io_kiocb *req, int rw, struct iov_iter *iter, 2373ed29b0b4SJens Axboe unsigned int issue_flags) 2374ed29b0b4SJens Axboe { 2375ed29b0b4SJens Axboe if (WARN_ON_ONCE(!req->imu)) 2376ed29b0b4SJens Axboe return -EFAULT; 2377ed29b0b4SJens Axboe return __io_import_fixed(req, rw, iter, req->imu); 2378ed29b0b4SJens Axboe } 2379ed29b0b4SJens Axboe 2380ed29b0b4SJens Axboe static int io_buffer_add_list(struct io_ring_ctx *ctx, 2381ed29b0b4SJens Axboe struct io_buffer_list *bl, unsigned int bgid) 2382ed29b0b4SJens Axboe { 2383ed29b0b4SJens Axboe bl->bgid = bgid; 2384ed29b0b4SJens Axboe if (bgid < BGID_ARRAY) 2385ed29b0b4SJens Axboe return 0; 2386ed29b0b4SJens Axboe 2387ed29b0b4SJens Axboe return xa_err(xa_store(&ctx->io_bl_xa, bgid, bl, GFP_KERNEL)); 2388ed29b0b4SJens Axboe } 2389ed29b0b4SJens Axboe 2390ed29b0b4SJens Axboe static void __user *io_provided_buffer_select(struct io_kiocb *req, size_t *len, 2391ed29b0b4SJens Axboe struct io_buffer_list *bl) 2392ed29b0b4SJens Axboe { 2393ed29b0b4SJens Axboe if (!list_empty(&bl->buf_list)) { 2394ed29b0b4SJens Axboe struct io_buffer *kbuf; 2395ed29b0b4SJens Axboe 2396ed29b0b4SJens Axboe kbuf = list_first_entry(&bl->buf_list, struct io_buffer, list); 2397ed29b0b4SJens Axboe list_del(&kbuf->list); 2398ed29b0b4SJens Axboe if (*len > kbuf->len) 2399ed29b0b4SJens Axboe *len = kbuf->len; 2400ed29b0b4SJens Axboe req->flags |= REQ_F_BUFFER_SELECTED; 2401ed29b0b4SJens Axboe req->kbuf = kbuf; 2402ed29b0b4SJens Axboe req->buf_index = kbuf->bid; 2403ed29b0b4SJens Axboe return u64_to_user_ptr(kbuf->addr); 2404ed29b0b4SJens Axboe } 2405ed29b0b4SJens Axboe return NULL; 2406ed29b0b4SJens Axboe } 2407ed29b0b4SJens Axboe 2408ed29b0b4SJens Axboe static void __user *io_ring_buffer_select(struct io_kiocb *req, size_t *len, 2409ed29b0b4SJens Axboe struct io_buffer_list *bl, 2410ed29b0b4SJens Axboe unsigned int issue_flags) 2411ed29b0b4SJens Axboe { 2412ed29b0b4SJens Axboe struct io_uring_buf_ring *br = bl->buf_ring; 2413ed29b0b4SJens Axboe struct io_uring_buf *buf; 2414ed29b0b4SJens Axboe __u16 head = bl->head; 2415ed29b0b4SJens Axboe 2416ed29b0b4SJens Axboe if (unlikely(smp_load_acquire(&br->tail) == head)) 2417ed29b0b4SJens Axboe return NULL; 2418ed29b0b4SJens Axboe 2419ed29b0b4SJens Axboe head &= bl->mask; 2420ed29b0b4SJens Axboe if (head < IO_BUFFER_LIST_BUF_PER_PAGE) { 2421ed29b0b4SJens Axboe buf = &br->bufs[head]; 2422ed29b0b4SJens Axboe } else { 2423ed29b0b4SJens Axboe int off = head & (IO_BUFFER_LIST_BUF_PER_PAGE - 1); 2424ed29b0b4SJens Axboe int index = head / IO_BUFFER_LIST_BUF_PER_PAGE; 2425ed29b0b4SJens Axboe buf = page_address(bl->buf_pages[index]); 2426ed29b0b4SJens Axboe buf += off; 2427ed29b0b4SJens Axboe } 2428ed29b0b4SJens Axboe if (*len > buf->len) 2429ed29b0b4SJens Axboe *len = buf->len; 2430ed29b0b4SJens Axboe req->flags |= REQ_F_BUFFER_RING; 2431ed29b0b4SJens Axboe req->buf_list = bl; 2432ed29b0b4SJens Axboe req->buf_index = buf->bid; 2433ed29b0b4SJens Axboe 2434ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_UNLOCKED || !file_can_poll(req->file)) { 2435ed29b0b4SJens Axboe /* 2436ed29b0b4SJens Axboe * If we came in unlocked, we have no choice but to consume the 2437ed29b0b4SJens Axboe * buffer here. This does mean it'll be pinned until the IO 2438ed29b0b4SJens Axboe * completes. But coming in unlocked means we're in io-wq 2439ed29b0b4SJens Axboe * context, hence there should be no further retry. For the 2440ed29b0b4SJens Axboe * locked case, the caller must ensure to call the commit when 2441ed29b0b4SJens Axboe * the transfer completes (or if we get -EAGAIN and must poll 2442ed29b0b4SJens Axboe * or retry). 2443ed29b0b4SJens Axboe */ 2444ed29b0b4SJens Axboe req->buf_list = NULL; 2445ed29b0b4SJens Axboe bl->head++; 2446ed29b0b4SJens Axboe } 2447ed29b0b4SJens Axboe return u64_to_user_ptr(buf->addr); 2448ed29b0b4SJens Axboe } 2449ed29b0b4SJens Axboe 2450f9ead18cSJens Axboe void __user *io_buffer_select(struct io_kiocb *req, size_t *len, 2451ed29b0b4SJens Axboe unsigned int issue_flags) 2452ed29b0b4SJens Axboe { 2453ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 2454ed29b0b4SJens Axboe struct io_buffer_list *bl; 2455ed29b0b4SJens Axboe void __user *ret = NULL; 2456ed29b0b4SJens Axboe 2457ed29b0b4SJens Axboe io_ring_submit_lock(req->ctx, issue_flags); 2458ed29b0b4SJens Axboe 2459ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, req->buf_index); 2460ed29b0b4SJens Axboe if (likely(bl)) { 2461ed29b0b4SJens Axboe if (bl->buf_nr_pages) 2462ed29b0b4SJens Axboe ret = io_ring_buffer_select(req, len, bl, issue_flags); 2463ed29b0b4SJens Axboe else 2464ed29b0b4SJens Axboe ret = io_provided_buffer_select(req, len, bl); 2465ed29b0b4SJens Axboe } 2466ed29b0b4SJens Axboe io_ring_submit_unlock(req->ctx, issue_flags); 2467ed29b0b4SJens Axboe return ret; 2468ed29b0b4SJens Axboe } 2469ed29b0b4SJens Axboe 2470ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 2471ed29b0b4SJens Axboe static ssize_t io_compat_import(struct io_kiocb *req, struct iovec *iov, 2472ed29b0b4SJens Axboe unsigned int issue_flags) 2473ed29b0b4SJens Axboe { 24743c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 2475ed29b0b4SJens Axboe struct compat_iovec __user *uiov; 2476ed29b0b4SJens Axboe compat_ssize_t clen; 2477ed29b0b4SJens Axboe void __user *buf; 2478ed29b0b4SJens Axboe size_t len; 2479ed29b0b4SJens Axboe 24803c306fb2SJens Axboe uiov = u64_to_user_ptr(rw->addr); 2481ed29b0b4SJens Axboe if (!access_ok(uiov, sizeof(*uiov))) 2482ed29b0b4SJens Axboe return -EFAULT; 2483ed29b0b4SJens Axboe if (__get_user(clen, &uiov->iov_len)) 2484ed29b0b4SJens Axboe return -EFAULT; 2485ed29b0b4SJens Axboe if (clen < 0) 2486ed29b0b4SJens Axboe return -EINVAL; 2487ed29b0b4SJens Axboe 2488ed29b0b4SJens Axboe len = clen; 2489ed29b0b4SJens Axboe buf = io_buffer_select(req, &len, issue_flags); 2490ed29b0b4SJens Axboe if (!buf) 2491ed29b0b4SJens Axboe return -ENOBUFS; 24923c306fb2SJens Axboe rw->addr = (unsigned long) buf; 2493ed29b0b4SJens Axboe iov[0].iov_base = buf; 24943c306fb2SJens Axboe rw->len = iov[0].iov_len = (compat_size_t) len; 2495ed29b0b4SJens Axboe return 0; 2496ed29b0b4SJens Axboe } 2497ed29b0b4SJens Axboe #endif 2498ed29b0b4SJens Axboe 2499ed29b0b4SJens Axboe static ssize_t __io_iov_buffer_select(struct io_kiocb *req, struct iovec *iov, 2500ed29b0b4SJens Axboe unsigned int issue_flags) 2501ed29b0b4SJens Axboe { 25023c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 25033c306fb2SJens Axboe struct iovec __user *uiov = u64_to_user_ptr(rw->addr); 2504ed29b0b4SJens Axboe void __user *buf; 2505ed29b0b4SJens Axboe ssize_t len; 2506ed29b0b4SJens Axboe 2507ed29b0b4SJens Axboe if (copy_from_user(iov, uiov, sizeof(*uiov))) 2508ed29b0b4SJens Axboe return -EFAULT; 2509ed29b0b4SJens Axboe 2510ed29b0b4SJens Axboe len = iov[0].iov_len; 2511ed29b0b4SJens Axboe if (len < 0) 2512ed29b0b4SJens Axboe return -EINVAL; 2513ed29b0b4SJens Axboe buf = io_buffer_select(req, &len, issue_flags); 2514ed29b0b4SJens Axboe if (!buf) 2515ed29b0b4SJens Axboe return -ENOBUFS; 25163c306fb2SJens Axboe rw->addr = (unsigned long) buf; 2517ed29b0b4SJens Axboe iov[0].iov_base = buf; 25183c306fb2SJens Axboe rw->len = iov[0].iov_len = len; 2519ed29b0b4SJens Axboe return 0; 2520ed29b0b4SJens Axboe } 2521ed29b0b4SJens Axboe 2522ed29b0b4SJens Axboe static ssize_t io_iov_buffer_select(struct io_kiocb *req, struct iovec *iov, 2523ed29b0b4SJens Axboe unsigned int issue_flags) 2524ed29b0b4SJens Axboe { 25253c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 25263c306fb2SJens Axboe 2527ed29b0b4SJens Axboe if (req->flags & (REQ_F_BUFFER_SELECTED|REQ_F_BUFFER_RING)) { 25283c306fb2SJens Axboe iov[0].iov_base = u64_to_user_ptr(rw->addr); 25293c306fb2SJens Axboe iov[0].iov_len = rw->len; 2530ed29b0b4SJens Axboe return 0; 2531ed29b0b4SJens Axboe } 25323c306fb2SJens Axboe if (rw->len != 1) 2533ed29b0b4SJens Axboe return -EINVAL; 2534ed29b0b4SJens Axboe 2535ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 2536ed29b0b4SJens Axboe if (req->ctx->compat) 2537ed29b0b4SJens Axboe return io_compat_import(req, iov, issue_flags); 2538ed29b0b4SJens Axboe #endif 2539ed29b0b4SJens Axboe 2540ed29b0b4SJens Axboe return __io_iov_buffer_select(req, iov, issue_flags); 2541ed29b0b4SJens Axboe } 2542ed29b0b4SJens Axboe 25433c306fb2SJens Axboe static struct iovec *__io_import_iovec(int ddir, struct io_kiocb *req, 2544ed29b0b4SJens Axboe struct io_rw_state *s, 2545ed29b0b4SJens Axboe unsigned int issue_flags) 2546ed29b0b4SJens Axboe { 25473c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 2548ed29b0b4SJens Axboe struct iov_iter *iter = &s->iter; 2549ed29b0b4SJens Axboe u8 opcode = req->opcode; 2550ed29b0b4SJens Axboe struct iovec *iovec; 2551ed29b0b4SJens Axboe void __user *buf; 2552ed29b0b4SJens Axboe size_t sqe_len; 2553ed29b0b4SJens Axboe ssize_t ret; 2554ed29b0b4SJens Axboe 2555ed29b0b4SJens Axboe if (opcode == IORING_OP_READ_FIXED || opcode == IORING_OP_WRITE_FIXED) { 25563c306fb2SJens Axboe ret = io_import_fixed(req, ddir, iter, issue_flags); 2557ed29b0b4SJens Axboe if (ret) 2558ed29b0b4SJens Axboe return ERR_PTR(ret); 2559ed29b0b4SJens Axboe return NULL; 2560ed29b0b4SJens Axboe } 2561ed29b0b4SJens Axboe 25623c306fb2SJens Axboe buf = u64_to_user_ptr(rw->addr); 25633c306fb2SJens Axboe sqe_len = rw->len; 2564ed29b0b4SJens Axboe 2565ed29b0b4SJens Axboe if (opcode == IORING_OP_READ || opcode == IORING_OP_WRITE) { 2566ed29b0b4SJens Axboe if (io_do_buffer_select(req)) { 2567ed29b0b4SJens Axboe buf = io_buffer_select(req, &sqe_len, issue_flags); 2568ed29b0b4SJens Axboe if (!buf) 2569ed29b0b4SJens Axboe return ERR_PTR(-ENOBUFS); 25703c306fb2SJens Axboe rw->addr = (unsigned long) buf; 25713c306fb2SJens Axboe rw->len = sqe_len; 2572ed29b0b4SJens Axboe } 2573ed29b0b4SJens Axboe 25743c306fb2SJens Axboe ret = import_single_range(ddir, buf, sqe_len, s->fast_iov, iter); 2575ed29b0b4SJens Axboe if (ret) 2576ed29b0b4SJens Axboe return ERR_PTR(ret); 2577ed29b0b4SJens Axboe return NULL; 2578ed29b0b4SJens Axboe } 2579ed29b0b4SJens Axboe 2580ed29b0b4SJens Axboe iovec = s->fast_iov; 2581ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_SELECT) { 2582ed29b0b4SJens Axboe ret = io_iov_buffer_select(req, iovec, issue_flags); 2583ed29b0b4SJens Axboe if (ret) 2584ed29b0b4SJens Axboe return ERR_PTR(ret); 25853c306fb2SJens Axboe iov_iter_init(iter, ddir, iovec, 1, iovec->iov_len); 2586ed29b0b4SJens Axboe return NULL; 2587ed29b0b4SJens Axboe } 2588ed29b0b4SJens Axboe 25893c306fb2SJens Axboe ret = __import_iovec(ddir, buf, sqe_len, UIO_FASTIOV, &iovec, iter, 2590ed29b0b4SJens Axboe req->ctx->compat); 2591ed29b0b4SJens Axboe if (unlikely(ret < 0)) 2592ed29b0b4SJens Axboe return ERR_PTR(ret); 2593ed29b0b4SJens Axboe return iovec; 2594ed29b0b4SJens Axboe } 2595ed29b0b4SJens Axboe 2596ed29b0b4SJens Axboe static inline int io_import_iovec(int rw, struct io_kiocb *req, 2597ed29b0b4SJens Axboe struct iovec **iovec, struct io_rw_state *s, 2598ed29b0b4SJens Axboe unsigned int issue_flags) 2599ed29b0b4SJens Axboe { 2600ed29b0b4SJens Axboe *iovec = __io_import_iovec(rw, req, s, issue_flags); 2601ed29b0b4SJens Axboe if (unlikely(IS_ERR(*iovec))) 2602ed29b0b4SJens Axboe return PTR_ERR(*iovec); 2603ed29b0b4SJens Axboe 2604ed29b0b4SJens Axboe iov_iter_save_state(&s->iter, &s->iter_state); 2605ed29b0b4SJens Axboe return 0; 2606ed29b0b4SJens Axboe } 2607ed29b0b4SJens Axboe 2608ed29b0b4SJens Axboe static inline loff_t *io_kiocb_ppos(struct kiocb *kiocb) 2609ed29b0b4SJens Axboe { 2610ed29b0b4SJens Axboe return (kiocb->ki_filp->f_mode & FMODE_STREAM) ? NULL : &kiocb->ki_pos; 2611ed29b0b4SJens Axboe } 2612ed29b0b4SJens Axboe 2613ed29b0b4SJens Axboe /* 2614ed29b0b4SJens Axboe * For files that don't have ->read_iter() and ->write_iter(), handle them 2615ed29b0b4SJens Axboe * by looping over ->read() or ->write() manually. 2616ed29b0b4SJens Axboe */ 26173c306fb2SJens Axboe static ssize_t loop_rw_iter(int ddir, struct io_rw *rw, struct iov_iter *iter) 2618ed29b0b4SJens Axboe { 26193c306fb2SJens Axboe struct kiocb *kiocb = &rw->kiocb; 26203c306fb2SJens Axboe struct file *file = kiocb->ki_filp; 2621ed29b0b4SJens Axboe ssize_t ret = 0; 2622ed29b0b4SJens Axboe loff_t *ppos; 2623ed29b0b4SJens Axboe 2624ed29b0b4SJens Axboe /* 2625ed29b0b4SJens Axboe * Don't support polled IO through this interface, and we can't 2626ed29b0b4SJens Axboe * support non-blocking either. For the latter, this just causes 2627ed29b0b4SJens Axboe * the kiocb to be handled from an async context. 2628ed29b0b4SJens Axboe */ 2629ed29b0b4SJens Axboe if (kiocb->ki_flags & IOCB_HIPRI) 2630ed29b0b4SJens Axboe return -EOPNOTSUPP; 2631ed29b0b4SJens Axboe if ((kiocb->ki_flags & IOCB_NOWAIT) && 2632ed29b0b4SJens Axboe !(kiocb->ki_filp->f_flags & O_NONBLOCK)) 2633ed29b0b4SJens Axboe return -EAGAIN; 2634ed29b0b4SJens Axboe 2635ed29b0b4SJens Axboe ppos = io_kiocb_ppos(kiocb); 2636ed29b0b4SJens Axboe 2637ed29b0b4SJens Axboe while (iov_iter_count(iter)) { 2638ed29b0b4SJens Axboe struct iovec iovec; 2639ed29b0b4SJens Axboe ssize_t nr; 2640ed29b0b4SJens Axboe 2641ed29b0b4SJens Axboe if (!iov_iter_is_bvec(iter)) { 2642ed29b0b4SJens Axboe iovec = iov_iter_iovec(iter); 2643ed29b0b4SJens Axboe } else { 26443c306fb2SJens Axboe iovec.iov_base = u64_to_user_ptr(rw->addr); 26453c306fb2SJens Axboe iovec.iov_len = rw->len; 2646ed29b0b4SJens Axboe } 2647ed29b0b4SJens Axboe 26483c306fb2SJens Axboe if (ddir == READ) { 2649ed29b0b4SJens Axboe nr = file->f_op->read(file, iovec.iov_base, 2650ed29b0b4SJens Axboe iovec.iov_len, ppos); 2651ed29b0b4SJens Axboe } else { 2652ed29b0b4SJens Axboe nr = file->f_op->write(file, iovec.iov_base, 2653ed29b0b4SJens Axboe iovec.iov_len, ppos); 2654ed29b0b4SJens Axboe } 2655ed29b0b4SJens Axboe 2656ed29b0b4SJens Axboe if (nr < 0) { 2657ed29b0b4SJens Axboe if (!ret) 2658ed29b0b4SJens Axboe ret = nr; 2659ed29b0b4SJens Axboe break; 2660ed29b0b4SJens Axboe } 2661ed29b0b4SJens Axboe ret += nr; 2662ed29b0b4SJens Axboe if (!iov_iter_is_bvec(iter)) { 2663ed29b0b4SJens Axboe iov_iter_advance(iter, nr); 2664ed29b0b4SJens Axboe } else { 26653c306fb2SJens Axboe rw->addr += nr; 26663c306fb2SJens Axboe rw->len -= nr; 26673c306fb2SJens Axboe if (!rw->len) 2668ed29b0b4SJens Axboe break; 2669ed29b0b4SJens Axboe } 2670ed29b0b4SJens Axboe if (nr != iovec.iov_len) 2671ed29b0b4SJens Axboe break; 2672ed29b0b4SJens Axboe } 2673ed29b0b4SJens Axboe 2674ed29b0b4SJens Axboe return ret; 2675ed29b0b4SJens Axboe } 2676ed29b0b4SJens Axboe 2677ed29b0b4SJens Axboe static void io_req_map_rw(struct io_kiocb *req, const struct iovec *iovec, 2678ed29b0b4SJens Axboe const struct iovec *fast_iov, struct iov_iter *iter) 2679ed29b0b4SJens Axboe { 26803c306fb2SJens Axboe struct io_async_rw *io = req->async_data; 2681ed29b0b4SJens Axboe 26823c306fb2SJens Axboe memcpy(&io->s.iter, iter, sizeof(*iter)); 26833c306fb2SJens Axboe io->free_iovec = iovec; 26843c306fb2SJens Axboe io->bytes_done = 0; 2685ed29b0b4SJens Axboe /* can only be fixed buffers, no need to do anything */ 2686ed29b0b4SJens Axboe if (iov_iter_is_bvec(iter)) 2687ed29b0b4SJens Axboe return; 2688ed29b0b4SJens Axboe if (!iovec) { 2689ed29b0b4SJens Axboe unsigned iov_off = 0; 2690ed29b0b4SJens Axboe 26913c306fb2SJens Axboe io->s.iter.iov = io->s.fast_iov; 2692ed29b0b4SJens Axboe if (iter->iov != fast_iov) { 2693ed29b0b4SJens Axboe iov_off = iter->iov - fast_iov; 26943c306fb2SJens Axboe io->s.iter.iov += iov_off; 2695ed29b0b4SJens Axboe } 26963c306fb2SJens Axboe if (io->s.fast_iov != fast_iov) 26973c306fb2SJens Axboe memcpy(io->s.fast_iov + iov_off, fast_iov + iov_off, 2698ed29b0b4SJens Axboe sizeof(struct iovec) * iter->nr_segs); 2699ed29b0b4SJens Axboe } else { 2700ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 2701ed29b0b4SJens Axboe } 2702ed29b0b4SJens Axboe } 2703ed29b0b4SJens Axboe 270499f15d8dSJens Axboe bool io_alloc_async_data(struct io_kiocb *req) 2705ed29b0b4SJens Axboe { 2706ed29b0b4SJens Axboe WARN_ON_ONCE(!io_op_defs[req->opcode].async_size); 2707ed29b0b4SJens Axboe req->async_data = kmalloc(io_op_defs[req->opcode].async_size, GFP_KERNEL); 2708ed29b0b4SJens Axboe if (req->async_data) { 2709ed29b0b4SJens Axboe req->flags |= REQ_F_ASYNC_DATA; 2710ed29b0b4SJens Axboe return false; 2711ed29b0b4SJens Axboe } 2712ed29b0b4SJens Axboe return true; 2713ed29b0b4SJens Axboe } 2714ed29b0b4SJens Axboe 2715ed29b0b4SJens Axboe static int io_setup_async_rw(struct io_kiocb *req, const struct iovec *iovec, 2716ed29b0b4SJens Axboe struct io_rw_state *s, bool force) 2717ed29b0b4SJens Axboe { 2718dc919cafSJens Axboe if (!force && !io_op_defs[req->opcode].prep_async) 2719ed29b0b4SJens Axboe return 0; 2720ed29b0b4SJens Axboe if (!req_has_async_data(req)) { 2721ed29b0b4SJens Axboe struct io_async_rw *iorw; 2722ed29b0b4SJens Axboe 2723ed29b0b4SJens Axboe if (io_alloc_async_data(req)) { 2724ed29b0b4SJens Axboe kfree(iovec); 2725ed29b0b4SJens Axboe return -ENOMEM; 2726ed29b0b4SJens Axboe } 2727ed29b0b4SJens Axboe 2728ed29b0b4SJens Axboe io_req_map_rw(req, iovec, s->fast_iov, &s->iter); 2729ed29b0b4SJens Axboe iorw = req->async_data; 2730ed29b0b4SJens Axboe /* we've copied and mapped the iter, ensure state is saved */ 2731ed29b0b4SJens Axboe iov_iter_save_state(&iorw->s.iter, &iorw->s.iter_state); 2732ed29b0b4SJens Axboe } 2733ed29b0b4SJens Axboe return 0; 2734ed29b0b4SJens Axboe } 2735ed29b0b4SJens Axboe 2736ed29b0b4SJens Axboe static inline int io_rw_prep_async(struct io_kiocb *req, int rw) 2737ed29b0b4SJens Axboe { 2738ed29b0b4SJens Axboe struct io_async_rw *iorw = req->async_data; 2739ed29b0b4SJens Axboe struct iovec *iov; 2740ed29b0b4SJens Axboe int ret; 2741ed29b0b4SJens Axboe 2742ed29b0b4SJens Axboe /* submission path, ->uring_lock should already be taken */ 2743ed29b0b4SJens Axboe ret = io_import_iovec(rw, req, &iov, &iorw->s, 0); 2744ed29b0b4SJens Axboe if (unlikely(ret < 0)) 2745ed29b0b4SJens Axboe return ret; 2746ed29b0b4SJens Axboe 2747ed29b0b4SJens Axboe iorw->bytes_done = 0; 2748ed29b0b4SJens Axboe iorw->free_iovec = iov; 2749ed29b0b4SJens Axboe if (iov) 2750ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 2751ed29b0b4SJens Axboe return 0; 2752ed29b0b4SJens Axboe } 2753ed29b0b4SJens Axboe 2754ed29b0b4SJens Axboe static int io_readv_prep_async(struct io_kiocb *req) 2755ed29b0b4SJens Axboe { 2756ed29b0b4SJens Axboe return io_rw_prep_async(req, READ); 2757ed29b0b4SJens Axboe } 2758ed29b0b4SJens Axboe 2759ed29b0b4SJens Axboe static int io_writev_prep_async(struct io_kiocb *req) 2760ed29b0b4SJens Axboe { 2761ed29b0b4SJens Axboe return io_rw_prep_async(req, WRITE); 2762ed29b0b4SJens Axboe } 2763ed29b0b4SJens Axboe 2764ed29b0b4SJens Axboe /* 2765ed29b0b4SJens Axboe * This is our waitqueue callback handler, registered through __folio_lock_async() 2766ed29b0b4SJens Axboe * when we initially tried to do the IO with the iocb armed our waitqueue. 2767ed29b0b4SJens Axboe * This gets called when the page is unlocked, and we generally expect that to 2768ed29b0b4SJens Axboe * happen when the page IO is completed and the page is now uptodate. This will 2769ed29b0b4SJens Axboe * queue a task_work based retry of the operation, attempting to copy the data 2770ed29b0b4SJens Axboe * again. If the latter fails because the page was NOT uptodate, then we will 2771ed29b0b4SJens Axboe * do a thread based blocking retry of the operation. That's the unexpected 2772ed29b0b4SJens Axboe * slow path. 2773ed29b0b4SJens Axboe */ 2774ed29b0b4SJens Axboe static int io_async_buf_func(struct wait_queue_entry *wait, unsigned mode, 2775ed29b0b4SJens Axboe int sync, void *arg) 2776ed29b0b4SJens Axboe { 2777ed29b0b4SJens Axboe struct wait_page_queue *wpq; 2778ed29b0b4SJens Axboe struct io_kiocb *req = wait->private; 27793c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 2780ed29b0b4SJens Axboe struct wait_page_key *key = arg; 2781ed29b0b4SJens Axboe 2782ed29b0b4SJens Axboe wpq = container_of(wait, struct wait_page_queue, wait); 2783ed29b0b4SJens Axboe 2784ed29b0b4SJens Axboe if (!wake_page_match(wpq, key)) 2785ed29b0b4SJens Axboe return 0; 2786ed29b0b4SJens Axboe 27873c306fb2SJens Axboe rw->kiocb.ki_flags &= ~IOCB_WAITQ; 2788ed29b0b4SJens Axboe list_del_init(&wait->entry); 2789ed29b0b4SJens Axboe io_req_task_queue(req); 2790ed29b0b4SJens Axboe return 1; 2791ed29b0b4SJens Axboe } 2792ed29b0b4SJens Axboe 2793ed29b0b4SJens Axboe /* 2794ed29b0b4SJens Axboe * This controls whether a given IO request should be armed for async page 2795ed29b0b4SJens Axboe * based retry. If we return false here, the request is handed to the async 2796ed29b0b4SJens Axboe * worker threads for retry. If we're doing buffered reads on a regular file, 2797ed29b0b4SJens Axboe * we prepare a private wait_page_queue entry and retry the operation. This 2798ed29b0b4SJens Axboe * will either succeed because the page is now uptodate and unlocked, or it 2799ed29b0b4SJens Axboe * will register a callback when the page is unlocked at IO completion. Through 2800ed29b0b4SJens Axboe * that callback, io_uring uses task_work to setup a retry of the operation. 2801ed29b0b4SJens Axboe * That retry will attempt the buffered read again. The retry will generally 2802ed29b0b4SJens Axboe * succeed, or in rare cases where it fails, we then fall back to using the 2803ed29b0b4SJens Axboe * async worker threads for a blocking retry. 2804ed29b0b4SJens Axboe */ 2805ed29b0b4SJens Axboe static bool io_rw_should_retry(struct io_kiocb *req) 2806ed29b0b4SJens Axboe { 28073c306fb2SJens Axboe struct io_async_rw *io = req->async_data; 28083c306fb2SJens Axboe struct wait_page_queue *wait = &io->wpq; 28093c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 28103c306fb2SJens Axboe struct kiocb *kiocb = &rw->kiocb; 2811ed29b0b4SJens Axboe 2812ed29b0b4SJens Axboe /* never retry for NOWAIT, we just complete with -EAGAIN */ 2813ed29b0b4SJens Axboe if (req->flags & REQ_F_NOWAIT) 2814ed29b0b4SJens Axboe return false; 2815ed29b0b4SJens Axboe 2816ed29b0b4SJens Axboe /* Only for buffered IO */ 2817ed29b0b4SJens Axboe if (kiocb->ki_flags & (IOCB_DIRECT | IOCB_HIPRI)) 2818ed29b0b4SJens Axboe return false; 2819ed29b0b4SJens Axboe 2820ed29b0b4SJens Axboe /* 2821ed29b0b4SJens Axboe * just use poll if we can, and don't attempt if the fs doesn't 2822ed29b0b4SJens Axboe * support callback based unlocks 2823ed29b0b4SJens Axboe */ 2824ed29b0b4SJens Axboe if (file_can_poll(req->file) || !(req->file->f_mode & FMODE_BUF_RASYNC)) 2825ed29b0b4SJens Axboe return false; 2826ed29b0b4SJens Axboe 2827ed29b0b4SJens Axboe wait->wait.func = io_async_buf_func; 2828ed29b0b4SJens Axboe wait->wait.private = req; 2829ed29b0b4SJens Axboe wait->wait.flags = 0; 2830ed29b0b4SJens Axboe INIT_LIST_HEAD(&wait->wait.entry); 2831ed29b0b4SJens Axboe kiocb->ki_flags |= IOCB_WAITQ; 2832ed29b0b4SJens Axboe kiocb->ki_flags &= ~IOCB_NOWAIT; 2833ed29b0b4SJens Axboe kiocb->ki_waitq = wait; 2834ed29b0b4SJens Axboe return true; 2835ed29b0b4SJens Axboe } 2836ed29b0b4SJens Axboe 28373c306fb2SJens Axboe static inline int io_iter_do_read(struct io_rw *rw, struct iov_iter *iter) 2838ed29b0b4SJens Axboe { 28393c306fb2SJens Axboe struct file *file = rw->kiocb.ki_filp; 28403c306fb2SJens Axboe 28413c306fb2SJens Axboe if (likely(file->f_op->read_iter)) 28423c306fb2SJens Axboe return call_read_iter(file, &rw->kiocb, iter); 28433c306fb2SJens Axboe else if (file->f_op->read) 28443c306fb2SJens Axboe return loop_rw_iter(READ, rw, iter); 2845ed29b0b4SJens Axboe else 2846ed29b0b4SJens Axboe return -EINVAL; 2847ed29b0b4SJens Axboe } 2848ed29b0b4SJens Axboe 2849ed29b0b4SJens Axboe static bool need_read_all(struct io_kiocb *req) 2850ed29b0b4SJens Axboe { 2851ed29b0b4SJens Axboe return req->flags & REQ_F_ISREG || 2852ed29b0b4SJens Axboe S_ISBLK(file_inode(req->file)->i_mode); 2853ed29b0b4SJens Axboe } 2854ed29b0b4SJens Axboe 2855ed29b0b4SJens Axboe static int io_rw_init_file(struct io_kiocb *req, fmode_t mode) 2856ed29b0b4SJens Axboe { 28573c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 28583c306fb2SJens Axboe struct kiocb *kiocb = &rw->kiocb; 2859ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 2860ed29b0b4SJens Axboe struct file *file = req->file; 2861ed29b0b4SJens Axboe int ret; 2862ed29b0b4SJens Axboe 2863ed29b0b4SJens Axboe if (unlikely(!file || !(file->f_mode & mode))) 2864ed29b0b4SJens Axboe return -EBADF; 2865ed29b0b4SJens Axboe 2866ed29b0b4SJens Axboe if (!io_req_ffs_set(req)) 2867ed29b0b4SJens Axboe req->flags |= io_file_get_flags(file) << REQ_F_SUPPORT_NOWAIT_BIT; 2868ed29b0b4SJens Axboe 2869ed29b0b4SJens Axboe kiocb->ki_flags = iocb_flags(file); 28703c306fb2SJens Axboe ret = kiocb_set_rw_flags(kiocb, rw->flags); 2871ed29b0b4SJens Axboe if (unlikely(ret)) 2872ed29b0b4SJens Axboe return ret; 2873ed29b0b4SJens Axboe 2874ed29b0b4SJens Axboe /* 2875ed29b0b4SJens Axboe * If the file is marked O_NONBLOCK, still allow retry for it if it 2876ed29b0b4SJens Axboe * supports async. Otherwise it's impossible to use O_NONBLOCK files 2877ed29b0b4SJens Axboe * reliably. If not, or it IOCB_NOWAIT is set, don't retry. 2878ed29b0b4SJens Axboe */ 2879ed29b0b4SJens Axboe if ((kiocb->ki_flags & IOCB_NOWAIT) || 2880ed29b0b4SJens Axboe ((file->f_flags & O_NONBLOCK) && !io_file_supports_nowait(req))) 2881ed29b0b4SJens Axboe req->flags |= REQ_F_NOWAIT; 2882ed29b0b4SJens Axboe 2883ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL) { 2884ed29b0b4SJens Axboe if (!(kiocb->ki_flags & IOCB_DIRECT) || !file->f_op->iopoll) 2885ed29b0b4SJens Axboe return -EOPNOTSUPP; 2886ed29b0b4SJens Axboe 2887ed29b0b4SJens Axboe kiocb->private = NULL; 2888ed29b0b4SJens Axboe kiocb->ki_flags |= IOCB_HIPRI | IOCB_ALLOC_CACHE; 2889ed29b0b4SJens Axboe kiocb->ki_complete = io_complete_rw_iopoll; 2890ed29b0b4SJens Axboe req->iopoll_completed = 0; 2891ed29b0b4SJens Axboe } else { 2892ed29b0b4SJens Axboe if (kiocb->ki_flags & IOCB_HIPRI) 2893ed29b0b4SJens Axboe return -EINVAL; 2894ed29b0b4SJens Axboe kiocb->ki_complete = io_complete_rw; 2895ed29b0b4SJens Axboe } 2896ed29b0b4SJens Axboe 2897ed29b0b4SJens Axboe return 0; 2898ed29b0b4SJens Axboe } 2899ed29b0b4SJens Axboe 2900ed29b0b4SJens Axboe static int io_read(struct io_kiocb *req, unsigned int issue_flags) 2901ed29b0b4SJens Axboe { 29023c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 2903ed29b0b4SJens Axboe struct io_rw_state __s, *s = &__s; 2904ed29b0b4SJens Axboe struct iovec *iovec; 29053c306fb2SJens Axboe struct kiocb *kiocb = &rw->kiocb; 2906ed29b0b4SJens Axboe bool force_nonblock = issue_flags & IO_URING_F_NONBLOCK; 29073c306fb2SJens Axboe struct io_async_rw *io; 2908ed29b0b4SJens Axboe ssize_t ret, ret2; 2909ed29b0b4SJens Axboe loff_t *ppos; 2910ed29b0b4SJens Axboe 2911ed29b0b4SJens Axboe if (!req_has_async_data(req)) { 2912ed29b0b4SJens Axboe ret = io_import_iovec(READ, req, &iovec, s, issue_flags); 2913ed29b0b4SJens Axboe if (unlikely(ret < 0)) 2914ed29b0b4SJens Axboe return ret; 2915ed29b0b4SJens Axboe } else { 29163c306fb2SJens Axboe io = req->async_data; 29173c306fb2SJens Axboe s = &io->s; 2918ed29b0b4SJens Axboe 2919ed29b0b4SJens Axboe /* 2920ed29b0b4SJens Axboe * Safe and required to re-import if we're using provided 2921ed29b0b4SJens Axboe * buffers, as we dropped the selected one before retry. 2922ed29b0b4SJens Axboe */ 2923ed29b0b4SJens Axboe if (io_do_buffer_select(req)) { 2924ed29b0b4SJens Axboe ret = io_import_iovec(READ, req, &iovec, s, issue_flags); 2925ed29b0b4SJens Axboe if (unlikely(ret < 0)) 2926ed29b0b4SJens Axboe return ret; 2927ed29b0b4SJens Axboe } 2928ed29b0b4SJens Axboe 2929ed29b0b4SJens Axboe /* 2930ed29b0b4SJens Axboe * We come here from an earlier attempt, restore our state to 2931ed29b0b4SJens Axboe * match in case it doesn't. It's cheap enough that we don't 2932ed29b0b4SJens Axboe * need to make this conditional. 2933ed29b0b4SJens Axboe */ 2934ed29b0b4SJens Axboe iov_iter_restore(&s->iter, &s->iter_state); 2935ed29b0b4SJens Axboe iovec = NULL; 2936ed29b0b4SJens Axboe } 2937ed29b0b4SJens Axboe ret = io_rw_init_file(req, FMODE_READ); 2938ed29b0b4SJens Axboe if (unlikely(ret)) { 2939ed29b0b4SJens Axboe kfree(iovec); 2940ed29b0b4SJens Axboe return ret; 2941ed29b0b4SJens Axboe } 2942ed29b0b4SJens Axboe req->cqe.res = iov_iter_count(&s->iter); 2943ed29b0b4SJens Axboe 2944ed29b0b4SJens Axboe if (force_nonblock) { 2945ed29b0b4SJens Axboe /* If the file doesn't support async, just async punt */ 2946ed29b0b4SJens Axboe if (unlikely(!io_file_supports_nowait(req))) { 2947ed29b0b4SJens Axboe ret = io_setup_async_rw(req, iovec, s, true); 2948ed29b0b4SJens Axboe return ret ?: -EAGAIN; 2949ed29b0b4SJens Axboe } 2950ed29b0b4SJens Axboe kiocb->ki_flags |= IOCB_NOWAIT; 2951ed29b0b4SJens Axboe } else { 2952ed29b0b4SJens Axboe /* Ensure we clear previously set non-block flag */ 2953ed29b0b4SJens Axboe kiocb->ki_flags &= ~IOCB_NOWAIT; 2954ed29b0b4SJens Axboe } 2955ed29b0b4SJens Axboe 2956ed29b0b4SJens Axboe ppos = io_kiocb_update_pos(req); 2957ed29b0b4SJens Axboe 2958ed29b0b4SJens Axboe ret = rw_verify_area(READ, req->file, ppos, req->cqe.res); 2959ed29b0b4SJens Axboe if (unlikely(ret)) { 2960ed29b0b4SJens Axboe kfree(iovec); 2961ed29b0b4SJens Axboe return ret; 2962ed29b0b4SJens Axboe } 2963ed29b0b4SJens Axboe 29643c306fb2SJens Axboe ret = io_iter_do_read(rw, &s->iter); 2965ed29b0b4SJens Axboe 2966ed29b0b4SJens Axboe if (ret == -EAGAIN || (req->flags & REQ_F_REISSUE)) { 2967ed29b0b4SJens Axboe req->flags &= ~REQ_F_REISSUE; 2968ed29b0b4SJens Axboe /* if we can poll, just do that */ 2969ed29b0b4SJens Axboe if (req->opcode == IORING_OP_READ && file_can_poll(req->file)) 2970ed29b0b4SJens Axboe return -EAGAIN; 2971ed29b0b4SJens Axboe /* IOPOLL retry should happen for io-wq threads */ 2972ed29b0b4SJens Axboe if (!force_nonblock && !(req->ctx->flags & IORING_SETUP_IOPOLL)) 2973ed29b0b4SJens Axboe goto done; 2974ed29b0b4SJens Axboe /* no retry on NONBLOCK nor RWF_NOWAIT */ 2975ed29b0b4SJens Axboe if (req->flags & REQ_F_NOWAIT) 2976ed29b0b4SJens Axboe goto done; 2977ed29b0b4SJens Axboe ret = 0; 2978ed29b0b4SJens Axboe } else if (ret == -EIOCBQUEUED) { 2979ed29b0b4SJens Axboe goto out_free; 2980ed29b0b4SJens Axboe } else if (ret == req->cqe.res || ret <= 0 || !force_nonblock || 2981ed29b0b4SJens Axboe (req->flags & REQ_F_NOWAIT) || !need_read_all(req)) { 2982ed29b0b4SJens Axboe /* read all, failed, already did sync or don't want to retry */ 2983ed29b0b4SJens Axboe goto done; 2984ed29b0b4SJens Axboe } 2985ed29b0b4SJens Axboe 2986ed29b0b4SJens Axboe /* 2987ed29b0b4SJens Axboe * Don't depend on the iter state matching what was consumed, or being 2988ed29b0b4SJens Axboe * untouched in case of error. Restore it and we'll advance it 2989ed29b0b4SJens Axboe * manually if we need to. 2990ed29b0b4SJens Axboe */ 2991ed29b0b4SJens Axboe iov_iter_restore(&s->iter, &s->iter_state); 2992ed29b0b4SJens Axboe 2993ed29b0b4SJens Axboe ret2 = io_setup_async_rw(req, iovec, s, true); 2994ed29b0b4SJens Axboe if (ret2) 2995ed29b0b4SJens Axboe return ret2; 2996ed29b0b4SJens Axboe 2997ed29b0b4SJens Axboe iovec = NULL; 29983c306fb2SJens Axboe io = req->async_data; 29993c306fb2SJens Axboe s = &io->s; 3000ed29b0b4SJens Axboe /* 3001ed29b0b4SJens Axboe * Now use our persistent iterator and state, if we aren't already. 3002ed29b0b4SJens Axboe * We've restored and mapped the iter to match. 3003ed29b0b4SJens Axboe */ 3004ed29b0b4SJens Axboe 3005ed29b0b4SJens Axboe do { 3006ed29b0b4SJens Axboe /* 3007ed29b0b4SJens Axboe * We end up here because of a partial read, either from 3008ed29b0b4SJens Axboe * above or inside this loop. Advance the iter by the bytes 3009ed29b0b4SJens Axboe * that were consumed. 3010ed29b0b4SJens Axboe */ 3011ed29b0b4SJens Axboe iov_iter_advance(&s->iter, ret); 3012ed29b0b4SJens Axboe if (!iov_iter_count(&s->iter)) 3013ed29b0b4SJens Axboe break; 30143c306fb2SJens Axboe io->bytes_done += ret; 3015ed29b0b4SJens Axboe iov_iter_save_state(&s->iter, &s->iter_state); 3016ed29b0b4SJens Axboe 3017ed29b0b4SJens Axboe /* if we can retry, do so with the callbacks armed */ 3018ed29b0b4SJens Axboe if (!io_rw_should_retry(req)) { 3019ed29b0b4SJens Axboe kiocb->ki_flags &= ~IOCB_WAITQ; 3020ed29b0b4SJens Axboe return -EAGAIN; 3021ed29b0b4SJens Axboe } 3022ed29b0b4SJens Axboe 3023ed29b0b4SJens Axboe /* 3024ed29b0b4SJens Axboe * Now retry read with the IOCB_WAITQ parts set in the iocb. If 3025ed29b0b4SJens Axboe * we get -EIOCBQUEUED, then we'll get a notification when the 3026ed29b0b4SJens Axboe * desired page gets unlocked. We can also get a partial read 3027ed29b0b4SJens Axboe * here, and if we do, then just retry at the new offset. 3028ed29b0b4SJens Axboe */ 30293c306fb2SJens Axboe ret = io_iter_do_read(rw, &s->iter); 3030ed29b0b4SJens Axboe if (ret == -EIOCBQUEUED) 303197b388d7SJens Axboe return IOU_ISSUE_SKIP_COMPLETE; 3032ed29b0b4SJens Axboe /* we got some bytes, but not all. retry. */ 3033ed29b0b4SJens Axboe kiocb->ki_flags &= ~IOCB_WAITQ; 3034ed29b0b4SJens Axboe iov_iter_restore(&s->iter, &s->iter_state); 3035ed29b0b4SJens Axboe } while (ret > 0); 3036ed29b0b4SJens Axboe done: 3037ed29b0b4SJens Axboe kiocb_done(req, ret, issue_flags); 3038ed29b0b4SJens Axboe out_free: 3039ed29b0b4SJens Axboe /* it's faster to check here then delegate to kfree */ 3040ed29b0b4SJens Axboe if (iovec) 3041ed29b0b4SJens Axboe kfree(iovec); 304297b388d7SJens Axboe return IOU_ISSUE_SKIP_COMPLETE; 3043ed29b0b4SJens Axboe } 3044ed29b0b4SJens Axboe 3045ed29b0b4SJens Axboe static int io_write(struct io_kiocb *req, unsigned int issue_flags) 3046ed29b0b4SJens Axboe { 30473c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 3048ed29b0b4SJens Axboe struct io_rw_state __s, *s = &__s; 3049ed29b0b4SJens Axboe struct iovec *iovec; 30503c306fb2SJens Axboe struct kiocb *kiocb = &rw->kiocb; 3051ed29b0b4SJens Axboe bool force_nonblock = issue_flags & IO_URING_F_NONBLOCK; 3052ed29b0b4SJens Axboe ssize_t ret, ret2; 3053ed29b0b4SJens Axboe loff_t *ppos; 3054ed29b0b4SJens Axboe 3055ed29b0b4SJens Axboe if (!req_has_async_data(req)) { 3056ed29b0b4SJens Axboe ret = io_import_iovec(WRITE, req, &iovec, s, issue_flags); 3057ed29b0b4SJens Axboe if (unlikely(ret < 0)) 3058ed29b0b4SJens Axboe return ret; 3059ed29b0b4SJens Axboe } else { 30603c306fb2SJens Axboe struct io_async_rw *io = req->async_data; 3061ed29b0b4SJens Axboe 30623c306fb2SJens Axboe s = &io->s; 3063ed29b0b4SJens Axboe iov_iter_restore(&s->iter, &s->iter_state); 3064ed29b0b4SJens Axboe iovec = NULL; 3065ed29b0b4SJens Axboe } 3066ed29b0b4SJens Axboe ret = io_rw_init_file(req, FMODE_WRITE); 3067ed29b0b4SJens Axboe if (unlikely(ret)) { 3068ed29b0b4SJens Axboe kfree(iovec); 3069ed29b0b4SJens Axboe return ret; 3070ed29b0b4SJens Axboe } 3071ed29b0b4SJens Axboe req->cqe.res = iov_iter_count(&s->iter); 3072ed29b0b4SJens Axboe 3073ed29b0b4SJens Axboe if (force_nonblock) { 3074ed29b0b4SJens Axboe /* If the file doesn't support async, just async punt */ 3075ed29b0b4SJens Axboe if (unlikely(!io_file_supports_nowait(req))) 3076ed29b0b4SJens Axboe goto copy_iov; 3077ed29b0b4SJens Axboe 3078ed29b0b4SJens Axboe /* file path doesn't support NOWAIT for non-direct_IO */ 3079ed29b0b4SJens Axboe if (force_nonblock && !(kiocb->ki_flags & IOCB_DIRECT) && 3080ed29b0b4SJens Axboe (req->flags & REQ_F_ISREG)) 3081ed29b0b4SJens Axboe goto copy_iov; 3082ed29b0b4SJens Axboe 3083ed29b0b4SJens Axboe kiocb->ki_flags |= IOCB_NOWAIT; 3084ed29b0b4SJens Axboe } else { 3085ed29b0b4SJens Axboe /* Ensure we clear previously set non-block flag */ 3086ed29b0b4SJens Axboe kiocb->ki_flags &= ~IOCB_NOWAIT; 3087ed29b0b4SJens Axboe } 3088ed29b0b4SJens Axboe 3089ed29b0b4SJens Axboe ppos = io_kiocb_update_pos(req); 3090ed29b0b4SJens Axboe 3091ed29b0b4SJens Axboe ret = rw_verify_area(WRITE, req->file, ppos, req->cqe.res); 3092ed29b0b4SJens Axboe if (unlikely(ret)) 3093ed29b0b4SJens Axboe goto out_free; 3094ed29b0b4SJens Axboe 3095ed29b0b4SJens Axboe /* 3096ed29b0b4SJens Axboe * Open-code file_start_write here to grab freeze protection, 3097ed29b0b4SJens Axboe * which will be released by another thread in 3098ed29b0b4SJens Axboe * io_complete_rw(). Fool lockdep by telling it the lock got 3099ed29b0b4SJens Axboe * released so that it doesn't complain about the held lock when 3100ed29b0b4SJens Axboe * we return to userspace. 3101ed29b0b4SJens Axboe */ 3102ed29b0b4SJens Axboe if (req->flags & REQ_F_ISREG) { 3103ed29b0b4SJens Axboe sb_start_write(file_inode(req->file)->i_sb); 3104ed29b0b4SJens Axboe __sb_writers_release(file_inode(req->file)->i_sb, 3105ed29b0b4SJens Axboe SB_FREEZE_WRITE); 3106ed29b0b4SJens Axboe } 3107ed29b0b4SJens Axboe kiocb->ki_flags |= IOCB_WRITE; 3108ed29b0b4SJens Axboe 3109ed29b0b4SJens Axboe if (likely(req->file->f_op->write_iter)) 3110ed29b0b4SJens Axboe ret2 = call_write_iter(req->file, kiocb, &s->iter); 3111ed29b0b4SJens Axboe else if (req->file->f_op->write) 31123c306fb2SJens Axboe ret2 = loop_rw_iter(WRITE, rw, &s->iter); 3113ed29b0b4SJens Axboe else 3114ed29b0b4SJens Axboe ret2 = -EINVAL; 3115ed29b0b4SJens Axboe 3116ed29b0b4SJens Axboe if (req->flags & REQ_F_REISSUE) { 3117ed29b0b4SJens Axboe req->flags &= ~REQ_F_REISSUE; 3118ed29b0b4SJens Axboe ret2 = -EAGAIN; 3119ed29b0b4SJens Axboe } 3120ed29b0b4SJens Axboe 3121ed29b0b4SJens Axboe /* 3122ed29b0b4SJens Axboe * Raw bdev writes will return -EOPNOTSUPP for IOCB_NOWAIT. Just 3123ed29b0b4SJens Axboe * retry them without IOCB_NOWAIT. 3124ed29b0b4SJens Axboe */ 3125ed29b0b4SJens Axboe if (ret2 == -EOPNOTSUPP && (kiocb->ki_flags & IOCB_NOWAIT)) 3126ed29b0b4SJens Axboe ret2 = -EAGAIN; 3127ed29b0b4SJens Axboe /* no retry on NONBLOCK nor RWF_NOWAIT */ 3128ed29b0b4SJens Axboe if (ret2 == -EAGAIN && (req->flags & REQ_F_NOWAIT)) 3129ed29b0b4SJens Axboe goto done; 3130ed29b0b4SJens Axboe if (!force_nonblock || ret2 != -EAGAIN) { 3131ed29b0b4SJens Axboe /* IOPOLL retry should happen for io-wq threads */ 3132ed29b0b4SJens Axboe if (ret2 == -EAGAIN && (req->ctx->flags & IORING_SETUP_IOPOLL)) 3133ed29b0b4SJens Axboe goto copy_iov; 3134ed29b0b4SJens Axboe done: 3135ed29b0b4SJens Axboe kiocb_done(req, ret2, issue_flags); 313697b388d7SJens Axboe ret = IOU_ISSUE_SKIP_COMPLETE; 3137ed29b0b4SJens Axboe } else { 3138ed29b0b4SJens Axboe copy_iov: 3139ed29b0b4SJens Axboe iov_iter_restore(&s->iter, &s->iter_state); 3140ed29b0b4SJens Axboe ret = io_setup_async_rw(req, iovec, s, false); 3141ed29b0b4SJens Axboe return ret ?: -EAGAIN; 3142ed29b0b4SJens Axboe } 3143ed29b0b4SJens Axboe out_free: 3144ed29b0b4SJens Axboe /* it's reportedly faster than delegating the null check to kfree() */ 3145ed29b0b4SJens Axboe if (iovec) 3146ed29b0b4SJens Axboe kfree(iovec); 3147ed29b0b4SJens Axboe return ret; 3148ed29b0b4SJens Axboe } 3149ed29b0b4SJens Axboe 3150ed29b0b4SJens Axboe /* 3151ed29b0b4SJens Axboe * Note when io_fixed_fd_install() returns error value, it will ensure 3152ed29b0b4SJens Axboe * fput() is called correspondingly. 3153ed29b0b4SJens Axboe */ 3154cd40cae2SJens Axboe int io_fixed_fd_install(struct io_kiocb *req, unsigned int issue_flags, 3155ed29b0b4SJens Axboe struct file *file, unsigned int file_slot) 3156ed29b0b4SJens Axboe { 3157ed29b0b4SJens Axboe bool alloc_slot = file_slot == IORING_FILE_INDEX_ALLOC; 3158ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3159ed29b0b4SJens Axboe int ret; 3160ed29b0b4SJens Axboe 3161ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 3162ed29b0b4SJens Axboe 3163ed29b0b4SJens Axboe if (alloc_slot) { 3164ed29b0b4SJens Axboe ret = io_file_bitmap_get(ctx); 3165ed29b0b4SJens Axboe if (unlikely(ret < 0)) 3166ed29b0b4SJens Axboe goto err; 3167ed29b0b4SJens Axboe file_slot = ret; 3168ed29b0b4SJens Axboe } else { 3169ed29b0b4SJens Axboe file_slot--; 3170ed29b0b4SJens Axboe } 3171ed29b0b4SJens Axboe 3172ed29b0b4SJens Axboe ret = io_install_fixed_file(req, file, issue_flags, file_slot); 3173ed29b0b4SJens Axboe if (!ret && alloc_slot) 3174ed29b0b4SJens Axboe ret = file_slot; 3175ed29b0b4SJens Axboe err: 3176ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 3177ed29b0b4SJens Axboe if (unlikely(ret < 0)) 3178ed29b0b4SJens Axboe fput(file); 3179ed29b0b4SJens Axboe return ret; 3180ed29b0b4SJens Axboe } 3181ed29b0b4SJens Axboe 3182ed29b0b4SJens Axboe static int io_remove_buffers_prep(struct io_kiocb *req, 3183ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 3184ed29b0b4SJens Axboe { 3185c1ee5595SJens Axboe struct io_provide_buf *p = io_kiocb_to_cmd(req); 3186ed29b0b4SJens Axboe u64 tmp; 3187ed29b0b4SJens Axboe 3188ed29b0b4SJens Axboe if (sqe->rw_flags || sqe->addr || sqe->len || sqe->off || 3189ed29b0b4SJens Axboe sqe->splice_fd_in) 3190ed29b0b4SJens Axboe return -EINVAL; 3191ed29b0b4SJens Axboe 3192ed29b0b4SJens Axboe tmp = READ_ONCE(sqe->fd); 3193ed29b0b4SJens Axboe if (!tmp || tmp > USHRT_MAX) 3194ed29b0b4SJens Axboe return -EINVAL; 3195ed29b0b4SJens Axboe 3196ed29b0b4SJens Axboe memset(p, 0, sizeof(*p)); 3197ed29b0b4SJens Axboe p->nbufs = tmp; 3198ed29b0b4SJens Axboe p->bgid = READ_ONCE(sqe->buf_group); 3199ed29b0b4SJens Axboe return 0; 3200ed29b0b4SJens Axboe } 3201ed29b0b4SJens Axboe 3202ed29b0b4SJens Axboe static int __io_remove_buffers(struct io_ring_ctx *ctx, 3203ed29b0b4SJens Axboe struct io_buffer_list *bl, unsigned nbufs) 3204ed29b0b4SJens Axboe { 3205ed29b0b4SJens Axboe unsigned i = 0; 3206ed29b0b4SJens Axboe 3207ed29b0b4SJens Axboe /* shouldn't happen */ 3208ed29b0b4SJens Axboe if (!nbufs) 3209ed29b0b4SJens Axboe return 0; 3210ed29b0b4SJens Axboe 3211ed29b0b4SJens Axboe if (bl->buf_nr_pages) { 3212ed29b0b4SJens Axboe int j; 3213ed29b0b4SJens Axboe 3214ed29b0b4SJens Axboe i = bl->buf_ring->tail - bl->head; 3215ed29b0b4SJens Axboe for (j = 0; j < bl->buf_nr_pages; j++) 3216ed29b0b4SJens Axboe unpin_user_page(bl->buf_pages[j]); 3217ed29b0b4SJens Axboe kvfree(bl->buf_pages); 3218ed29b0b4SJens Axboe bl->buf_pages = NULL; 3219ed29b0b4SJens Axboe bl->buf_nr_pages = 0; 3220ed29b0b4SJens Axboe /* make sure it's seen as empty */ 3221ed29b0b4SJens Axboe INIT_LIST_HEAD(&bl->buf_list); 3222ed29b0b4SJens Axboe return i; 3223ed29b0b4SJens Axboe } 3224ed29b0b4SJens Axboe 3225ed29b0b4SJens Axboe /* the head kbuf is the list itself */ 3226ed29b0b4SJens Axboe while (!list_empty(&bl->buf_list)) { 3227ed29b0b4SJens Axboe struct io_buffer *nxt; 3228ed29b0b4SJens Axboe 3229ed29b0b4SJens Axboe nxt = list_first_entry(&bl->buf_list, struct io_buffer, list); 3230ed29b0b4SJens Axboe list_del(&nxt->list); 3231ed29b0b4SJens Axboe if (++i == nbufs) 3232ed29b0b4SJens Axboe return i; 3233ed29b0b4SJens Axboe cond_resched(); 3234ed29b0b4SJens Axboe } 3235ed29b0b4SJens Axboe i++; 3236ed29b0b4SJens Axboe 3237ed29b0b4SJens Axboe return i; 3238ed29b0b4SJens Axboe } 3239ed29b0b4SJens Axboe 3240ed29b0b4SJens Axboe static int io_remove_buffers(struct io_kiocb *req, unsigned int issue_flags) 3241ed29b0b4SJens Axboe { 3242c1ee5595SJens Axboe struct io_provide_buf *p = io_kiocb_to_cmd(req); 3243ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3244ed29b0b4SJens Axboe struct io_buffer_list *bl; 3245ed29b0b4SJens Axboe int ret = 0; 3246ed29b0b4SJens Axboe 3247ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 3248ed29b0b4SJens Axboe 3249ed29b0b4SJens Axboe ret = -ENOENT; 3250ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, p->bgid); 3251ed29b0b4SJens Axboe if (bl) { 3252ed29b0b4SJens Axboe ret = -EINVAL; 3253ed29b0b4SJens Axboe /* can't use provide/remove buffers command on mapped buffers */ 3254ed29b0b4SJens Axboe if (!bl->buf_nr_pages) 3255ed29b0b4SJens Axboe ret = __io_remove_buffers(ctx, bl, p->nbufs); 3256ed29b0b4SJens Axboe } 3257ed29b0b4SJens Axboe if (ret < 0) 3258ed29b0b4SJens Axboe req_set_fail(req); 3259ed29b0b4SJens Axboe 3260ed29b0b4SJens Axboe /* complete before unlock, IOPOLL may need the lock */ 326197b388d7SJens Axboe io_req_set_res(req, ret, 0); 326297b388d7SJens Axboe __io_req_complete(req, issue_flags); 3263ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 326497b388d7SJens Axboe return IOU_ISSUE_SKIP_COMPLETE; 3265ed29b0b4SJens Axboe } 3266ed29b0b4SJens Axboe 3267ed29b0b4SJens Axboe static int io_provide_buffers_prep(struct io_kiocb *req, 3268ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 3269ed29b0b4SJens Axboe { 3270ed29b0b4SJens Axboe unsigned long size, tmp_check; 3271c1ee5595SJens Axboe struct io_provide_buf *p = io_kiocb_to_cmd(req); 3272ed29b0b4SJens Axboe u64 tmp; 3273ed29b0b4SJens Axboe 3274ed29b0b4SJens Axboe if (sqe->rw_flags || sqe->splice_fd_in) 3275ed29b0b4SJens Axboe return -EINVAL; 3276ed29b0b4SJens Axboe 3277ed29b0b4SJens Axboe tmp = READ_ONCE(sqe->fd); 3278ed29b0b4SJens Axboe if (!tmp || tmp > USHRT_MAX) 3279ed29b0b4SJens Axboe return -E2BIG; 3280ed29b0b4SJens Axboe p->nbufs = tmp; 3281ed29b0b4SJens Axboe p->addr = READ_ONCE(sqe->addr); 3282ed29b0b4SJens Axboe p->len = READ_ONCE(sqe->len); 3283ed29b0b4SJens Axboe 3284ed29b0b4SJens Axboe if (check_mul_overflow((unsigned long)p->len, (unsigned long)p->nbufs, 3285ed29b0b4SJens Axboe &size)) 3286ed29b0b4SJens Axboe return -EOVERFLOW; 3287ed29b0b4SJens Axboe if (check_add_overflow((unsigned long)p->addr, size, &tmp_check)) 3288ed29b0b4SJens Axboe return -EOVERFLOW; 3289ed29b0b4SJens Axboe 3290ed29b0b4SJens Axboe size = (unsigned long)p->len * p->nbufs; 3291ed29b0b4SJens Axboe if (!access_ok(u64_to_user_ptr(p->addr), size)) 3292ed29b0b4SJens Axboe return -EFAULT; 3293ed29b0b4SJens Axboe 3294ed29b0b4SJens Axboe p->bgid = READ_ONCE(sqe->buf_group); 3295ed29b0b4SJens Axboe tmp = READ_ONCE(sqe->off); 3296ed29b0b4SJens Axboe if (tmp > USHRT_MAX) 3297ed29b0b4SJens Axboe return -E2BIG; 3298ed29b0b4SJens Axboe p->bid = tmp; 3299ed29b0b4SJens Axboe return 0; 3300ed29b0b4SJens Axboe } 3301ed29b0b4SJens Axboe 3302ed29b0b4SJens Axboe static int io_refill_buffer_cache(struct io_ring_ctx *ctx) 3303ed29b0b4SJens Axboe { 3304ed29b0b4SJens Axboe struct io_buffer *buf; 3305ed29b0b4SJens Axboe struct page *page; 3306ed29b0b4SJens Axboe int bufs_in_page; 3307ed29b0b4SJens Axboe 3308ed29b0b4SJens Axboe /* 3309ed29b0b4SJens Axboe * Completions that don't happen inline (eg not under uring_lock) will 3310ed29b0b4SJens Axboe * add to ->io_buffers_comp. If we don't have any free buffers, check 3311ed29b0b4SJens Axboe * the completion list and splice those entries first. 3312ed29b0b4SJens Axboe */ 3313ed29b0b4SJens Axboe if (!list_empty_careful(&ctx->io_buffers_comp)) { 3314ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 3315ed29b0b4SJens Axboe if (!list_empty(&ctx->io_buffers_comp)) { 3316ed29b0b4SJens Axboe list_splice_init(&ctx->io_buffers_comp, 3317ed29b0b4SJens Axboe &ctx->io_buffers_cache); 3318ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 3319ed29b0b4SJens Axboe return 0; 3320ed29b0b4SJens Axboe } 3321ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 3322ed29b0b4SJens Axboe } 3323ed29b0b4SJens Axboe 3324ed29b0b4SJens Axboe /* 3325ed29b0b4SJens Axboe * No free buffers and no completion entries either. Allocate a new 3326ed29b0b4SJens Axboe * page worth of buffer entries and add those to our freelist. 3327ed29b0b4SJens Axboe */ 3328ed29b0b4SJens Axboe page = alloc_page(GFP_KERNEL_ACCOUNT); 3329ed29b0b4SJens Axboe if (!page) 3330ed29b0b4SJens Axboe return -ENOMEM; 3331ed29b0b4SJens Axboe 3332ed29b0b4SJens Axboe list_add(&page->lru, &ctx->io_buffers_pages); 3333ed29b0b4SJens Axboe 3334ed29b0b4SJens Axboe buf = page_address(page); 3335ed29b0b4SJens Axboe bufs_in_page = PAGE_SIZE / sizeof(*buf); 3336ed29b0b4SJens Axboe while (bufs_in_page) { 3337ed29b0b4SJens Axboe list_add_tail(&buf->list, &ctx->io_buffers_cache); 3338ed29b0b4SJens Axboe buf++; 3339ed29b0b4SJens Axboe bufs_in_page--; 3340ed29b0b4SJens Axboe } 3341ed29b0b4SJens Axboe 3342ed29b0b4SJens Axboe return 0; 3343ed29b0b4SJens Axboe } 3344ed29b0b4SJens Axboe 3345ed29b0b4SJens Axboe static int io_add_buffers(struct io_ring_ctx *ctx, struct io_provide_buf *pbuf, 3346ed29b0b4SJens Axboe struct io_buffer_list *bl) 3347ed29b0b4SJens Axboe { 3348ed29b0b4SJens Axboe struct io_buffer *buf; 3349ed29b0b4SJens Axboe u64 addr = pbuf->addr; 3350ed29b0b4SJens Axboe int i, bid = pbuf->bid; 3351ed29b0b4SJens Axboe 3352ed29b0b4SJens Axboe for (i = 0; i < pbuf->nbufs; i++) { 3353ed29b0b4SJens Axboe if (list_empty(&ctx->io_buffers_cache) && 3354ed29b0b4SJens Axboe io_refill_buffer_cache(ctx)) 3355ed29b0b4SJens Axboe break; 3356ed29b0b4SJens Axboe buf = list_first_entry(&ctx->io_buffers_cache, struct io_buffer, 3357ed29b0b4SJens Axboe list); 3358ed29b0b4SJens Axboe list_move_tail(&buf->list, &bl->buf_list); 3359ed29b0b4SJens Axboe buf->addr = addr; 3360ed29b0b4SJens Axboe buf->len = min_t(__u32, pbuf->len, MAX_RW_COUNT); 3361ed29b0b4SJens Axboe buf->bid = bid; 3362ed29b0b4SJens Axboe buf->bgid = pbuf->bgid; 3363ed29b0b4SJens Axboe addr += pbuf->len; 3364ed29b0b4SJens Axboe bid++; 3365ed29b0b4SJens Axboe cond_resched(); 3366ed29b0b4SJens Axboe } 3367ed29b0b4SJens Axboe 3368ed29b0b4SJens Axboe return i ? 0 : -ENOMEM; 3369ed29b0b4SJens Axboe } 3370ed29b0b4SJens Axboe 3371ed29b0b4SJens Axboe static __cold int io_init_bl_list(struct io_ring_ctx *ctx) 3372ed29b0b4SJens Axboe { 3373ed29b0b4SJens Axboe int i; 3374ed29b0b4SJens Axboe 3375ed29b0b4SJens Axboe ctx->io_bl = kcalloc(BGID_ARRAY, sizeof(struct io_buffer_list), 3376ed29b0b4SJens Axboe GFP_KERNEL); 3377ed29b0b4SJens Axboe if (!ctx->io_bl) 3378ed29b0b4SJens Axboe return -ENOMEM; 3379ed29b0b4SJens Axboe 3380ed29b0b4SJens Axboe for (i = 0; i < BGID_ARRAY; i++) { 3381ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->io_bl[i].buf_list); 3382ed29b0b4SJens Axboe ctx->io_bl[i].bgid = i; 3383ed29b0b4SJens Axboe } 3384ed29b0b4SJens Axboe 3385ed29b0b4SJens Axboe return 0; 3386ed29b0b4SJens Axboe } 3387ed29b0b4SJens Axboe 3388ed29b0b4SJens Axboe static int io_provide_buffers(struct io_kiocb *req, unsigned int issue_flags) 3389ed29b0b4SJens Axboe { 3390c1ee5595SJens Axboe struct io_provide_buf *p = io_kiocb_to_cmd(req); 3391ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3392ed29b0b4SJens Axboe struct io_buffer_list *bl; 3393ed29b0b4SJens Axboe int ret = 0; 3394ed29b0b4SJens Axboe 3395ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 3396ed29b0b4SJens Axboe 3397ed29b0b4SJens Axboe if (unlikely(p->bgid < BGID_ARRAY && !ctx->io_bl)) { 3398ed29b0b4SJens Axboe ret = io_init_bl_list(ctx); 3399ed29b0b4SJens Axboe if (ret) 3400ed29b0b4SJens Axboe goto err; 3401ed29b0b4SJens Axboe } 3402ed29b0b4SJens Axboe 3403ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, p->bgid); 3404ed29b0b4SJens Axboe if (unlikely(!bl)) { 3405ed29b0b4SJens Axboe bl = kzalloc(sizeof(*bl), GFP_KERNEL); 3406ed29b0b4SJens Axboe if (!bl) { 3407ed29b0b4SJens Axboe ret = -ENOMEM; 3408ed29b0b4SJens Axboe goto err; 3409ed29b0b4SJens Axboe } 3410ed29b0b4SJens Axboe INIT_LIST_HEAD(&bl->buf_list); 3411ed29b0b4SJens Axboe ret = io_buffer_add_list(ctx, bl, p->bgid); 3412ed29b0b4SJens Axboe if (ret) { 3413ed29b0b4SJens Axboe kfree(bl); 3414ed29b0b4SJens Axboe goto err; 3415ed29b0b4SJens Axboe } 3416ed29b0b4SJens Axboe } 3417ed29b0b4SJens Axboe /* can't add buffers via this command for a mapped buffer ring */ 3418ed29b0b4SJens Axboe if (bl->buf_nr_pages) { 3419ed29b0b4SJens Axboe ret = -EINVAL; 3420ed29b0b4SJens Axboe goto err; 3421ed29b0b4SJens Axboe } 3422ed29b0b4SJens Axboe 3423ed29b0b4SJens Axboe ret = io_add_buffers(ctx, p, bl); 3424ed29b0b4SJens Axboe err: 3425ed29b0b4SJens Axboe if (ret < 0) 3426ed29b0b4SJens Axboe req_set_fail(req); 3427ed29b0b4SJens Axboe /* complete before unlock, IOPOLL may need the lock */ 342897b388d7SJens Axboe io_req_set_res(req, ret, 0); 342997b388d7SJens Axboe __io_req_complete(req, issue_flags); 3430ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 343197b388d7SJens Axboe return IOU_ISSUE_SKIP_COMPLETE; 3432ed29b0b4SJens Axboe } 3433ed29b0b4SJens Axboe 34344cf90495SJens Axboe static __maybe_unused int io_eopnotsupp_prep(struct io_kiocb *kiocb, 34354cf90495SJens Axboe const struct io_uring_sqe *sqe) 34364cf90495SJens Axboe { 34374cf90495SJens Axboe return -EOPNOTSUPP; 34384cf90495SJens Axboe } 34394cf90495SJens Axboe 3440ed29b0b4SJens Axboe struct io_poll_table { 3441ed29b0b4SJens Axboe struct poll_table_struct pt; 3442ed29b0b4SJens Axboe struct io_kiocb *req; 3443ed29b0b4SJens Axboe int nr_entries; 3444ed29b0b4SJens Axboe int error; 3445ed29b0b4SJens Axboe }; 3446ed29b0b4SJens Axboe 3447ed29b0b4SJens Axboe #define IO_POLL_CANCEL_FLAG BIT(31) 3448ed29b0b4SJens Axboe #define IO_POLL_REF_MASK GENMASK(30, 0) 3449ed29b0b4SJens Axboe 3450ed29b0b4SJens Axboe /* 3451ed29b0b4SJens Axboe * If refs part of ->poll_refs (see IO_POLL_REF_MASK) is 0, it's free. We can 3452ed29b0b4SJens Axboe * bump it and acquire ownership. It's disallowed to modify requests while not 3453ed29b0b4SJens Axboe * owning it, that prevents from races for enqueueing task_work's and b/w 3454ed29b0b4SJens Axboe * arming poll and wakeups. 3455ed29b0b4SJens Axboe */ 3456ed29b0b4SJens Axboe static inline bool io_poll_get_ownership(struct io_kiocb *req) 3457ed29b0b4SJens Axboe { 3458ed29b0b4SJens Axboe return !(atomic_fetch_inc(&req->poll_refs) & IO_POLL_REF_MASK); 3459ed29b0b4SJens Axboe } 3460ed29b0b4SJens Axboe 3461ed29b0b4SJens Axboe static void io_poll_mark_cancelled(struct io_kiocb *req) 3462ed29b0b4SJens Axboe { 3463ed29b0b4SJens Axboe atomic_or(IO_POLL_CANCEL_FLAG, &req->poll_refs); 3464ed29b0b4SJens Axboe } 3465ed29b0b4SJens Axboe 34668d4388d1SJens Axboe static struct io_poll *io_poll_get_double(struct io_kiocb *req) 3467ed29b0b4SJens Axboe { 3468ed29b0b4SJens Axboe /* pure poll stashes this in ->async_data, poll driven retry elsewhere */ 3469ed29b0b4SJens Axboe if (req->opcode == IORING_OP_POLL_ADD) 3470ed29b0b4SJens Axboe return req->async_data; 3471ed29b0b4SJens Axboe return req->apoll->double_poll; 3472ed29b0b4SJens Axboe } 3473ed29b0b4SJens Axboe 34748d4388d1SJens Axboe static struct io_poll *io_poll_get_single(struct io_kiocb *req) 3475ed29b0b4SJens Axboe { 3476ed29b0b4SJens Axboe if (req->opcode == IORING_OP_POLL_ADD) 34778d4388d1SJens Axboe return io_kiocb_to_cmd(req); 3478ed29b0b4SJens Axboe return &req->apoll->poll; 3479ed29b0b4SJens Axboe } 3480ed29b0b4SJens Axboe 3481ed29b0b4SJens Axboe static void io_poll_req_insert(struct io_kiocb *req) 3482ed29b0b4SJens Axboe { 3483ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3484ed29b0b4SJens Axboe struct hlist_head *list; 3485ed29b0b4SJens Axboe 3486ed29b0b4SJens Axboe list = &ctx->cancel_hash[hash_long(req->cqe.user_data, ctx->cancel_hash_bits)]; 3487ed29b0b4SJens Axboe hlist_add_head(&req->hash_node, list); 3488ed29b0b4SJens Axboe } 3489ed29b0b4SJens Axboe 34908d4388d1SJens Axboe static void io_init_poll_iocb(struct io_poll *poll, __poll_t events, 3491ed29b0b4SJens Axboe wait_queue_func_t wake_func) 3492ed29b0b4SJens Axboe { 3493ed29b0b4SJens Axboe poll->head = NULL; 3494ed29b0b4SJens Axboe #define IO_POLL_UNMASK (EPOLLERR|EPOLLHUP|EPOLLNVAL|EPOLLRDHUP) 3495ed29b0b4SJens Axboe /* mask in events that we always want/need */ 3496ed29b0b4SJens Axboe poll->events = events | IO_POLL_UNMASK; 3497ed29b0b4SJens Axboe INIT_LIST_HEAD(&poll->wait.entry); 3498ed29b0b4SJens Axboe init_waitqueue_func_entry(&poll->wait, wake_func); 3499ed29b0b4SJens Axboe } 3500ed29b0b4SJens Axboe 35018d4388d1SJens Axboe static inline void io_poll_remove_entry(struct io_poll *poll) 3502ed29b0b4SJens Axboe { 3503ed29b0b4SJens Axboe struct wait_queue_head *head = smp_load_acquire(&poll->head); 3504ed29b0b4SJens Axboe 3505ed29b0b4SJens Axboe if (head) { 3506ed29b0b4SJens Axboe spin_lock_irq(&head->lock); 3507ed29b0b4SJens Axboe list_del_init(&poll->wait.entry); 3508ed29b0b4SJens Axboe poll->head = NULL; 3509ed29b0b4SJens Axboe spin_unlock_irq(&head->lock); 3510ed29b0b4SJens Axboe } 3511ed29b0b4SJens Axboe } 3512ed29b0b4SJens Axboe 3513ed29b0b4SJens Axboe static void io_poll_remove_entries(struct io_kiocb *req) 3514ed29b0b4SJens Axboe { 3515ed29b0b4SJens Axboe /* 3516ed29b0b4SJens Axboe * Nothing to do if neither of those flags are set. Avoid dipping 3517ed29b0b4SJens Axboe * into the poll/apoll/double cachelines if we can. 3518ed29b0b4SJens Axboe */ 3519ed29b0b4SJens Axboe if (!(req->flags & (REQ_F_SINGLE_POLL | REQ_F_DOUBLE_POLL))) 3520ed29b0b4SJens Axboe return; 3521ed29b0b4SJens Axboe 3522ed29b0b4SJens Axboe /* 3523ed29b0b4SJens Axboe * While we hold the waitqueue lock and the waitqueue is nonempty, 3524ed29b0b4SJens Axboe * wake_up_pollfree() will wait for us. However, taking the waitqueue 3525ed29b0b4SJens Axboe * lock in the first place can race with the waitqueue being freed. 3526ed29b0b4SJens Axboe * 3527ed29b0b4SJens Axboe * We solve this as eventpoll does: by taking advantage of the fact that 3528ed29b0b4SJens Axboe * all users of wake_up_pollfree() will RCU-delay the actual free. If 3529ed29b0b4SJens Axboe * we enter rcu_read_lock() and see that the pointer to the queue is 3530ed29b0b4SJens Axboe * non-NULL, we can then lock it without the memory being freed out from 3531ed29b0b4SJens Axboe * under us. 3532ed29b0b4SJens Axboe * 3533ed29b0b4SJens Axboe * Keep holding rcu_read_lock() as long as we hold the queue lock, in 3534ed29b0b4SJens Axboe * case the caller deletes the entry from the queue, leaving it empty. 3535ed29b0b4SJens Axboe * In that case, only RCU prevents the queue memory from being freed. 3536ed29b0b4SJens Axboe */ 3537ed29b0b4SJens Axboe rcu_read_lock(); 3538ed29b0b4SJens Axboe if (req->flags & REQ_F_SINGLE_POLL) 3539ed29b0b4SJens Axboe io_poll_remove_entry(io_poll_get_single(req)); 3540ed29b0b4SJens Axboe if (req->flags & REQ_F_DOUBLE_POLL) 3541ed29b0b4SJens Axboe io_poll_remove_entry(io_poll_get_double(req)); 3542ed29b0b4SJens Axboe rcu_read_unlock(); 3543ed29b0b4SJens Axboe } 3544ed29b0b4SJens Axboe 3545ed29b0b4SJens Axboe static int io_issue_sqe(struct io_kiocb *req, unsigned int issue_flags); 3546ed29b0b4SJens Axboe /* 3547ed29b0b4SJens Axboe * All poll tw should go through this. Checks for poll events, manages 3548ed29b0b4SJens Axboe * references, does rewait, etc. 3549ed29b0b4SJens Axboe * 3550ed29b0b4SJens Axboe * Returns a negative error on failure. >0 when no action require, which is 3551ed29b0b4SJens Axboe * either spurious wakeup or multishot CQE is served. 0 when it's done with 3552ed29b0b4SJens Axboe * the request, then the mask is stored in req->cqe.res. 3553ed29b0b4SJens Axboe */ 3554ed29b0b4SJens Axboe static int io_poll_check_events(struct io_kiocb *req, bool *locked) 3555ed29b0b4SJens Axboe { 3556ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3557ed29b0b4SJens Axboe int v, ret; 3558ed29b0b4SJens Axboe 3559ed29b0b4SJens Axboe /* req->task == current here, checking PF_EXITING is safe */ 3560ed29b0b4SJens Axboe if (unlikely(req->task->flags & PF_EXITING)) 3561ed29b0b4SJens Axboe return -ECANCELED; 3562ed29b0b4SJens Axboe 3563ed29b0b4SJens Axboe do { 3564ed29b0b4SJens Axboe v = atomic_read(&req->poll_refs); 3565ed29b0b4SJens Axboe 3566ed29b0b4SJens Axboe /* tw handler should be the owner, and so have some references */ 3567ed29b0b4SJens Axboe if (WARN_ON_ONCE(!(v & IO_POLL_REF_MASK))) 3568ed29b0b4SJens Axboe return 0; 3569ed29b0b4SJens Axboe if (v & IO_POLL_CANCEL_FLAG) 3570ed29b0b4SJens Axboe return -ECANCELED; 3571ed29b0b4SJens Axboe 3572ed29b0b4SJens Axboe if (!req->cqe.res) { 3573ed29b0b4SJens Axboe struct poll_table_struct pt = { ._key = req->apoll_events }; 3574ed29b0b4SJens Axboe req->cqe.res = vfs_poll(req->file, &pt) & req->apoll_events; 3575ed29b0b4SJens Axboe } 3576ed29b0b4SJens Axboe 3577ed29b0b4SJens Axboe if ((unlikely(!req->cqe.res))) 3578ed29b0b4SJens Axboe continue; 3579ed29b0b4SJens Axboe if (req->apoll_events & EPOLLONESHOT) 3580ed29b0b4SJens Axboe return 0; 3581ed29b0b4SJens Axboe 3582ed29b0b4SJens Axboe /* multishot, just fill a CQE and proceed */ 3583ed29b0b4SJens Axboe if (!(req->flags & REQ_F_APOLL_MULTISHOT)) { 3584ed29b0b4SJens Axboe __poll_t mask = mangle_poll(req->cqe.res & 3585ed29b0b4SJens Axboe req->apoll_events); 3586ed29b0b4SJens Axboe bool filled; 3587ed29b0b4SJens Axboe 3588ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 3589ed29b0b4SJens Axboe filled = io_fill_cqe_aux(ctx, req->cqe.user_data, 3590ed29b0b4SJens Axboe mask, IORING_CQE_F_MORE); 3591ed29b0b4SJens Axboe io_commit_cqring(ctx); 3592ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 3593ed29b0b4SJens Axboe if (filled) { 3594ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 3595ed29b0b4SJens Axboe continue; 3596ed29b0b4SJens Axboe } 3597ed29b0b4SJens Axboe return -ECANCELED; 3598ed29b0b4SJens Axboe } 3599ed29b0b4SJens Axboe 3600ed29b0b4SJens Axboe io_tw_lock(req->ctx, locked); 3601ed29b0b4SJens Axboe if (unlikely(req->task->flags & PF_EXITING)) 3602ed29b0b4SJens Axboe return -EFAULT; 3603ed29b0b4SJens Axboe ret = io_issue_sqe(req, 3604ed29b0b4SJens Axboe IO_URING_F_NONBLOCK|IO_URING_F_COMPLETE_DEFER); 3605ed29b0b4SJens Axboe if (ret) 3606ed29b0b4SJens Axboe return ret; 3607ed29b0b4SJens Axboe 3608ed29b0b4SJens Axboe /* 3609ed29b0b4SJens Axboe * Release all references, retry if someone tried to restart 3610ed29b0b4SJens Axboe * task_work while we were executing it. 3611ed29b0b4SJens Axboe */ 3612ed29b0b4SJens Axboe } while (atomic_sub_return(v & IO_POLL_REF_MASK, &req->poll_refs)); 3613ed29b0b4SJens Axboe 3614ed29b0b4SJens Axboe return 1; 3615ed29b0b4SJens Axboe } 3616ed29b0b4SJens Axboe 3617ed29b0b4SJens Axboe static void io_poll_task_func(struct io_kiocb *req, bool *locked) 3618ed29b0b4SJens Axboe { 3619ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3620ed29b0b4SJens Axboe int ret; 3621ed29b0b4SJens Axboe 3622ed29b0b4SJens Axboe ret = io_poll_check_events(req, locked); 3623ed29b0b4SJens Axboe if (ret > 0) 3624ed29b0b4SJens Axboe return; 3625ed29b0b4SJens Axboe 3626ed29b0b4SJens Axboe if (!ret) { 36278d4388d1SJens Axboe struct io_poll *poll = io_kiocb_to_cmd(req); 36288d4388d1SJens Axboe 36298d4388d1SJens Axboe req->cqe.res = mangle_poll(req->cqe.res & poll->events); 3630ed29b0b4SJens Axboe } else { 3631ed29b0b4SJens Axboe req->cqe.res = ret; 3632ed29b0b4SJens Axboe req_set_fail(req); 3633ed29b0b4SJens Axboe } 3634ed29b0b4SJens Axboe 3635ed29b0b4SJens Axboe io_poll_remove_entries(req); 3636ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 3637ed29b0b4SJens Axboe hash_del(&req->hash_node); 363897b388d7SJens Axboe req->cqe.flags = 0; 363997b388d7SJens Axboe __io_req_complete_post(req); 3640ed29b0b4SJens Axboe io_commit_cqring(ctx); 3641ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 3642ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 3643ed29b0b4SJens Axboe } 3644ed29b0b4SJens Axboe 3645ed29b0b4SJens Axboe static void io_apoll_task_func(struct io_kiocb *req, bool *locked) 3646ed29b0b4SJens Axboe { 3647ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3648ed29b0b4SJens Axboe int ret; 3649ed29b0b4SJens Axboe 3650ed29b0b4SJens Axboe ret = io_poll_check_events(req, locked); 3651ed29b0b4SJens Axboe if (ret > 0) 3652ed29b0b4SJens Axboe return; 3653ed29b0b4SJens Axboe 3654ed29b0b4SJens Axboe io_poll_remove_entries(req); 3655ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 3656ed29b0b4SJens Axboe hash_del(&req->hash_node); 3657ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 3658ed29b0b4SJens Axboe 3659ed29b0b4SJens Axboe if (!ret) 3660ed29b0b4SJens Axboe io_req_task_submit(req, locked); 3661ed29b0b4SJens Axboe else 3662ed29b0b4SJens Axboe io_req_complete_failed(req, ret); 3663ed29b0b4SJens Axboe } 3664ed29b0b4SJens Axboe 3665ed29b0b4SJens Axboe static void __io_poll_execute(struct io_kiocb *req, int mask, 3666ed29b0b4SJens Axboe __poll_t __maybe_unused events) 3667ed29b0b4SJens Axboe { 366897b388d7SJens Axboe io_req_set_res(req, mask, 0); 3669ed29b0b4SJens Axboe /* 3670ed29b0b4SJens Axboe * This is useful for poll that is armed on behalf of another 3671ed29b0b4SJens Axboe * request, and where the wakeup path could be on a different 3672ed29b0b4SJens Axboe * CPU. We want to avoid pulling in req->apoll->events for that 3673ed29b0b4SJens Axboe * case. 3674ed29b0b4SJens Axboe */ 3675ed29b0b4SJens Axboe if (req->opcode == IORING_OP_POLL_ADD) 3676ed29b0b4SJens Axboe req->io_task_work.func = io_poll_task_func; 3677ed29b0b4SJens Axboe else 3678ed29b0b4SJens Axboe req->io_task_work.func = io_apoll_task_func; 3679ed29b0b4SJens Axboe 3680ed29b0b4SJens Axboe trace_io_uring_task_add(req->ctx, req, req->cqe.user_data, req->opcode, mask); 3681ed29b0b4SJens Axboe io_req_task_work_add(req); 3682ed29b0b4SJens Axboe } 3683ed29b0b4SJens Axboe 3684ed29b0b4SJens Axboe static inline void io_poll_execute(struct io_kiocb *req, int res, 3685ed29b0b4SJens Axboe __poll_t events) 3686ed29b0b4SJens Axboe { 3687ed29b0b4SJens Axboe if (io_poll_get_ownership(req)) 3688ed29b0b4SJens Axboe __io_poll_execute(req, res, events); 3689ed29b0b4SJens Axboe } 3690ed29b0b4SJens Axboe 3691ed29b0b4SJens Axboe static void io_poll_cancel_req(struct io_kiocb *req) 3692ed29b0b4SJens Axboe { 3693ed29b0b4SJens Axboe io_poll_mark_cancelled(req); 3694ed29b0b4SJens Axboe /* kick tw, which should complete the request */ 3695ed29b0b4SJens Axboe io_poll_execute(req, 0, 0); 3696ed29b0b4SJens Axboe } 3697ed29b0b4SJens Axboe 3698ed29b0b4SJens Axboe #define wqe_to_req(wait) ((void *)((unsigned long) (wait)->private & ~1)) 3699ed29b0b4SJens Axboe #define wqe_is_double(wait) ((unsigned long) (wait)->private & 1) 3700ed29b0b4SJens Axboe #define IO_ASYNC_POLL_COMMON (EPOLLONESHOT | EPOLLPRI) 3701ed29b0b4SJens Axboe 3702ed29b0b4SJens Axboe static int io_poll_wake(struct wait_queue_entry *wait, unsigned mode, int sync, 3703ed29b0b4SJens Axboe void *key) 3704ed29b0b4SJens Axboe { 3705ed29b0b4SJens Axboe struct io_kiocb *req = wqe_to_req(wait); 37068d4388d1SJens Axboe struct io_poll *poll = container_of(wait, struct io_poll, wait); 3707ed29b0b4SJens Axboe __poll_t mask = key_to_poll(key); 3708ed29b0b4SJens Axboe 3709ed29b0b4SJens Axboe if (unlikely(mask & POLLFREE)) { 3710ed29b0b4SJens Axboe io_poll_mark_cancelled(req); 3711ed29b0b4SJens Axboe /* we have to kick tw in case it's not already */ 3712ed29b0b4SJens Axboe io_poll_execute(req, 0, poll->events); 3713ed29b0b4SJens Axboe 3714ed29b0b4SJens Axboe /* 3715ed29b0b4SJens Axboe * If the waitqueue is being freed early but someone is already 3716ed29b0b4SJens Axboe * holds ownership over it, we have to tear down the request as 3717ed29b0b4SJens Axboe * best we can. That means immediately removing the request from 3718ed29b0b4SJens Axboe * its waitqueue and preventing all further accesses to the 3719ed29b0b4SJens Axboe * waitqueue via the request. 3720ed29b0b4SJens Axboe */ 3721ed29b0b4SJens Axboe list_del_init(&poll->wait.entry); 3722ed29b0b4SJens Axboe 3723ed29b0b4SJens Axboe /* 3724ed29b0b4SJens Axboe * Careful: this *must* be the last step, since as soon 3725ed29b0b4SJens Axboe * as req->head is NULL'ed out, the request can be 3726ed29b0b4SJens Axboe * completed and freed, since aio_poll_complete_work() 3727ed29b0b4SJens Axboe * will no longer need to take the waitqueue lock. 3728ed29b0b4SJens Axboe */ 3729ed29b0b4SJens Axboe smp_store_release(&poll->head, NULL); 3730ed29b0b4SJens Axboe return 1; 3731ed29b0b4SJens Axboe } 3732ed29b0b4SJens Axboe 3733ed29b0b4SJens Axboe /* for instances that support it check for an event match first */ 3734ed29b0b4SJens Axboe if (mask && !(mask & (poll->events & ~IO_ASYNC_POLL_COMMON))) 3735ed29b0b4SJens Axboe return 0; 3736ed29b0b4SJens Axboe 3737ed29b0b4SJens Axboe if (io_poll_get_ownership(req)) { 3738ed29b0b4SJens Axboe /* optional, saves extra locking for removal in tw handler */ 3739ed29b0b4SJens Axboe if (mask && poll->events & EPOLLONESHOT) { 3740ed29b0b4SJens Axboe list_del_init(&poll->wait.entry); 3741ed29b0b4SJens Axboe poll->head = NULL; 3742ed29b0b4SJens Axboe if (wqe_is_double(wait)) 3743ed29b0b4SJens Axboe req->flags &= ~REQ_F_DOUBLE_POLL; 3744ed29b0b4SJens Axboe else 3745ed29b0b4SJens Axboe req->flags &= ~REQ_F_SINGLE_POLL; 3746ed29b0b4SJens Axboe } 3747ed29b0b4SJens Axboe __io_poll_execute(req, mask, poll->events); 3748ed29b0b4SJens Axboe } 3749ed29b0b4SJens Axboe return 1; 3750ed29b0b4SJens Axboe } 3751ed29b0b4SJens Axboe 37528d4388d1SJens Axboe static void __io_queue_proc(struct io_poll *poll, struct io_poll_table *pt, 3753ed29b0b4SJens Axboe struct wait_queue_head *head, 37548d4388d1SJens Axboe struct io_poll **poll_ptr) 3755ed29b0b4SJens Axboe { 3756ed29b0b4SJens Axboe struct io_kiocb *req = pt->req; 3757ed29b0b4SJens Axboe unsigned long wqe_private = (unsigned long) req; 3758ed29b0b4SJens Axboe 3759ed29b0b4SJens Axboe /* 3760ed29b0b4SJens Axboe * The file being polled uses multiple waitqueues for poll handling 37618d4388d1SJens Axboe * (e.g. one for read, one for write). Setup a separate io_poll 3762ed29b0b4SJens Axboe * if this happens. 3763ed29b0b4SJens Axboe */ 3764ed29b0b4SJens Axboe if (unlikely(pt->nr_entries)) { 37658d4388d1SJens Axboe struct io_poll *first = poll; 3766ed29b0b4SJens Axboe 3767ed29b0b4SJens Axboe /* double add on the same waitqueue head, ignore */ 3768ed29b0b4SJens Axboe if (first->head == head) 3769ed29b0b4SJens Axboe return; 3770ed29b0b4SJens Axboe /* already have a 2nd entry, fail a third attempt */ 3771ed29b0b4SJens Axboe if (*poll_ptr) { 3772ed29b0b4SJens Axboe if ((*poll_ptr)->head == head) 3773ed29b0b4SJens Axboe return; 3774ed29b0b4SJens Axboe pt->error = -EINVAL; 3775ed29b0b4SJens Axboe return; 3776ed29b0b4SJens Axboe } 3777ed29b0b4SJens Axboe 3778ed29b0b4SJens Axboe poll = kmalloc(sizeof(*poll), GFP_ATOMIC); 3779ed29b0b4SJens Axboe if (!poll) { 3780ed29b0b4SJens Axboe pt->error = -ENOMEM; 3781ed29b0b4SJens Axboe return; 3782ed29b0b4SJens Axboe } 3783ed29b0b4SJens Axboe /* mark as double wq entry */ 3784ed29b0b4SJens Axboe wqe_private |= 1; 3785ed29b0b4SJens Axboe req->flags |= REQ_F_DOUBLE_POLL; 3786ed29b0b4SJens Axboe io_init_poll_iocb(poll, first->events, first->wait.func); 3787ed29b0b4SJens Axboe *poll_ptr = poll; 3788ed29b0b4SJens Axboe if (req->opcode == IORING_OP_POLL_ADD) 3789ed29b0b4SJens Axboe req->flags |= REQ_F_ASYNC_DATA; 3790ed29b0b4SJens Axboe } 3791ed29b0b4SJens Axboe 3792ed29b0b4SJens Axboe req->flags |= REQ_F_SINGLE_POLL; 3793ed29b0b4SJens Axboe pt->nr_entries++; 3794ed29b0b4SJens Axboe poll->head = head; 3795ed29b0b4SJens Axboe poll->wait.private = (void *) wqe_private; 3796ed29b0b4SJens Axboe 3797ed29b0b4SJens Axboe if (poll->events & EPOLLEXCLUSIVE) 3798ed29b0b4SJens Axboe add_wait_queue_exclusive(head, &poll->wait); 3799ed29b0b4SJens Axboe else 3800ed29b0b4SJens Axboe add_wait_queue(head, &poll->wait); 3801ed29b0b4SJens Axboe } 3802ed29b0b4SJens Axboe 3803ed29b0b4SJens Axboe static void io_poll_queue_proc(struct file *file, struct wait_queue_head *head, 3804ed29b0b4SJens Axboe struct poll_table_struct *p) 3805ed29b0b4SJens Axboe { 3806ed29b0b4SJens Axboe struct io_poll_table *pt = container_of(p, struct io_poll_table, pt); 38078d4388d1SJens Axboe struct io_poll *poll = io_kiocb_to_cmd(pt->req); 3808ed29b0b4SJens Axboe 38098d4388d1SJens Axboe __io_queue_proc(poll, pt, head, 38108d4388d1SJens Axboe (struct io_poll **) &pt->req->async_data); 3811ed29b0b4SJens Axboe } 3812ed29b0b4SJens Axboe 3813ed29b0b4SJens Axboe static int __io_arm_poll_handler(struct io_kiocb *req, 38148d4388d1SJens Axboe struct io_poll *poll, 3815ed29b0b4SJens Axboe struct io_poll_table *ipt, __poll_t mask) 3816ed29b0b4SJens Axboe { 3817ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3818ed29b0b4SJens Axboe int v; 3819ed29b0b4SJens Axboe 3820ed29b0b4SJens Axboe INIT_HLIST_NODE(&req->hash_node); 3821ed29b0b4SJens Axboe req->work.cancel_seq = atomic_read(&ctx->cancel_seq); 3822ed29b0b4SJens Axboe io_init_poll_iocb(poll, mask, io_poll_wake); 3823ed29b0b4SJens Axboe poll->file = req->file; 3824ed29b0b4SJens Axboe 3825ed29b0b4SJens Axboe req->apoll_events = poll->events; 3826ed29b0b4SJens Axboe 3827ed29b0b4SJens Axboe ipt->pt._key = mask; 3828ed29b0b4SJens Axboe ipt->req = req; 3829ed29b0b4SJens Axboe ipt->error = 0; 3830ed29b0b4SJens Axboe ipt->nr_entries = 0; 3831ed29b0b4SJens Axboe 3832ed29b0b4SJens Axboe /* 3833ed29b0b4SJens Axboe * Take the ownership to delay any tw execution up until we're done 3834ed29b0b4SJens Axboe * with poll arming. see io_poll_get_ownership(). 3835ed29b0b4SJens Axboe */ 3836ed29b0b4SJens Axboe atomic_set(&req->poll_refs, 1); 3837ed29b0b4SJens Axboe mask = vfs_poll(req->file, &ipt->pt) & poll->events; 3838ed29b0b4SJens Axboe 3839ed29b0b4SJens Axboe if (mask && (poll->events & EPOLLONESHOT)) { 3840ed29b0b4SJens Axboe io_poll_remove_entries(req); 3841ed29b0b4SJens Axboe /* no one else has access to the req, forget about the ref */ 3842ed29b0b4SJens Axboe return mask; 3843ed29b0b4SJens Axboe } 3844ed29b0b4SJens Axboe if (!mask && unlikely(ipt->error || !ipt->nr_entries)) { 3845ed29b0b4SJens Axboe io_poll_remove_entries(req); 3846ed29b0b4SJens Axboe if (!ipt->error) 3847ed29b0b4SJens Axboe ipt->error = -EINVAL; 3848ed29b0b4SJens Axboe return 0; 3849ed29b0b4SJens Axboe } 3850ed29b0b4SJens Axboe 3851ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 3852ed29b0b4SJens Axboe io_poll_req_insert(req); 3853ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 3854ed29b0b4SJens Axboe 3855ed29b0b4SJens Axboe if (mask) { 3856ed29b0b4SJens Axboe /* can't multishot if failed, just queue the event we've got */ 3857ed29b0b4SJens Axboe if (unlikely(ipt->error || !ipt->nr_entries)) { 3858ed29b0b4SJens Axboe poll->events |= EPOLLONESHOT; 3859ed29b0b4SJens Axboe req->apoll_events |= EPOLLONESHOT; 3860ed29b0b4SJens Axboe ipt->error = 0; 3861ed29b0b4SJens Axboe } 3862ed29b0b4SJens Axboe __io_poll_execute(req, mask, poll->events); 3863ed29b0b4SJens Axboe return 0; 3864ed29b0b4SJens Axboe } 3865ed29b0b4SJens Axboe 3866ed29b0b4SJens Axboe /* 3867ed29b0b4SJens Axboe * Release ownership. If someone tried to queue a tw while it was 3868ed29b0b4SJens Axboe * locked, kick it off for them. 3869ed29b0b4SJens Axboe */ 3870ed29b0b4SJens Axboe v = atomic_dec_return(&req->poll_refs); 3871ed29b0b4SJens Axboe if (unlikely(v & IO_POLL_REF_MASK)) 3872ed29b0b4SJens Axboe __io_poll_execute(req, 0, poll->events); 3873ed29b0b4SJens Axboe return 0; 3874ed29b0b4SJens Axboe } 3875ed29b0b4SJens Axboe 3876ed29b0b4SJens Axboe static void io_async_queue_proc(struct file *file, struct wait_queue_head *head, 3877ed29b0b4SJens Axboe struct poll_table_struct *p) 3878ed29b0b4SJens Axboe { 3879ed29b0b4SJens Axboe struct io_poll_table *pt = container_of(p, struct io_poll_table, pt); 3880ed29b0b4SJens Axboe struct async_poll *apoll = pt->req->apoll; 3881ed29b0b4SJens Axboe 3882ed29b0b4SJens Axboe __io_queue_proc(&apoll->poll, pt, head, &apoll->double_poll); 3883ed29b0b4SJens Axboe } 3884ed29b0b4SJens Axboe 3885ed29b0b4SJens Axboe enum { 3886ed29b0b4SJens Axboe IO_APOLL_OK, 3887ed29b0b4SJens Axboe IO_APOLL_ABORTED, 3888ed29b0b4SJens Axboe IO_APOLL_READY 3889ed29b0b4SJens Axboe }; 3890ed29b0b4SJens Axboe 3891ed29b0b4SJens Axboe static int io_arm_poll_handler(struct io_kiocb *req, unsigned issue_flags) 3892ed29b0b4SJens Axboe { 3893ed29b0b4SJens Axboe const struct io_op_def *def = &io_op_defs[req->opcode]; 3894ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3895ed29b0b4SJens Axboe struct async_poll *apoll; 3896ed29b0b4SJens Axboe struct io_poll_table ipt; 3897ed29b0b4SJens Axboe __poll_t mask = POLLPRI | POLLERR; 3898ed29b0b4SJens Axboe int ret; 3899ed29b0b4SJens Axboe 3900ed29b0b4SJens Axboe if (!def->pollin && !def->pollout) 3901ed29b0b4SJens Axboe return IO_APOLL_ABORTED; 3902ed29b0b4SJens Axboe if (!file_can_poll(req->file)) 3903ed29b0b4SJens Axboe return IO_APOLL_ABORTED; 3904ed29b0b4SJens Axboe if ((req->flags & (REQ_F_POLLED|REQ_F_PARTIAL_IO)) == REQ_F_POLLED) 3905ed29b0b4SJens Axboe return IO_APOLL_ABORTED; 3906ed29b0b4SJens Axboe if (!(req->flags & REQ_F_APOLL_MULTISHOT)) 3907ed29b0b4SJens Axboe mask |= EPOLLONESHOT; 3908ed29b0b4SJens Axboe 3909ed29b0b4SJens Axboe if (def->pollin) { 3910ed29b0b4SJens Axboe mask |= EPOLLIN | EPOLLRDNORM; 3911ed29b0b4SJens Axboe 3912ed29b0b4SJens Axboe /* If reading from MSG_ERRQUEUE using recvmsg, ignore POLLIN */ 3913bd8587e4SJens Axboe if (req->flags & REQ_F_CLEAR_POLLIN) 3914ed29b0b4SJens Axboe mask &= ~EPOLLIN; 3915ed29b0b4SJens Axboe } else { 3916ed29b0b4SJens Axboe mask |= EPOLLOUT | EPOLLWRNORM; 3917ed29b0b4SJens Axboe } 3918ed29b0b4SJens Axboe if (def->poll_exclusive) 3919ed29b0b4SJens Axboe mask |= EPOLLEXCLUSIVE; 3920ed29b0b4SJens Axboe if (req->flags & REQ_F_POLLED) { 3921ed29b0b4SJens Axboe apoll = req->apoll; 3922ed29b0b4SJens Axboe kfree(apoll->double_poll); 3923ed29b0b4SJens Axboe } else if (!(issue_flags & IO_URING_F_UNLOCKED) && 3924ed29b0b4SJens Axboe !list_empty(&ctx->apoll_cache)) { 3925ed29b0b4SJens Axboe apoll = list_first_entry(&ctx->apoll_cache, struct async_poll, 3926ed29b0b4SJens Axboe poll.wait.entry); 3927ed29b0b4SJens Axboe list_del_init(&apoll->poll.wait.entry); 3928ed29b0b4SJens Axboe } else { 3929ed29b0b4SJens Axboe apoll = kmalloc(sizeof(*apoll), GFP_ATOMIC); 3930ed29b0b4SJens Axboe if (unlikely(!apoll)) 3931ed29b0b4SJens Axboe return IO_APOLL_ABORTED; 3932ed29b0b4SJens Axboe } 3933ed29b0b4SJens Axboe apoll->double_poll = NULL; 3934ed29b0b4SJens Axboe req->apoll = apoll; 3935ed29b0b4SJens Axboe req->flags |= REQ_F_POLLED; 3936ed29b0b4SJens Axboe ipt.pt._qproc = io_async_queue_proc; 3937ed29b0b4SJens Axboe 3938ed29b0b4SJens Axboe io_kbuf_recycle(req, issue_flags); 3939ed29b0b4SJens Axboe 3940ed29b0b4SJens Axboe ret = __io_arm_poll_handler(req, &apoll->poll, &ipt, mask); 3941ed29b0b4SJens Axboe if (ret || ipt.error) 3942ed29b0b4SJens Axboe return ret ? IO_APOLL_READY : IO_APOLL_ABORTED; 3943ed29b0b4SJens Axboe 3944ed29b0b4SJens Axboe trace_io_uring_poll_arm(ctx, req, req->cqe.user_data, req->opcode, 3945ed29b0b4SJens Axboe mask, apoll->poll.events); 3946ed29b0b4SJens Axboe return IO_APOLL_OK; 3947ed29b0b4SJens Axboe } 3948ed29b0b4SJens Axboe 3949ed29b0b4SJens Axboe /* 3950ed29b0b4SJens Axboe * Returns true if we found and killed one or more poll requests 3951ed29b0b4SJens Axboe */ 3952ed29b0b4SJens Axboe static __cold bool io_poll_remove_all(struct io_ring_ctx *ctx, 3953ed29b0b4SJens Axboe struct task_struct *tsk, bool cancel_all) 3954ed29b0b4SJens Axboe { 3955ed29b0b4SJens Axboe struct hlist_node *tmp; 3956ed29b0b4SJens Axboe struct io_kiocb *req; 3957ed29b0b4SJens Axboe bool found = false; 3958ed29b0b4SJens Axboe int i; 3959ed29b0b4SJens Axboe 3960ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 3961ed29b0b4SJens Axboe for (i = 0; i < (1U << ctx->cancel_hash_bits); i++) { 3962ed29b0b4SJens Axboe struct hlist_head *list; 3963ed29b0b4SJens Axboe 3964ed29b0b4SJens Axboe list = &ctx->cancel_hash[i]; 3965ed29b0b4SJens Axboe hlist_for_each_entry_safe(req, tmp, list, hash_node) { 3966ed29b0b4SJens Axboe if (io_match_task_safe(req, tsk, cancel_all)) { 3967ed29b0b4SJens Axboe hlist_del_init(&req->hash_node); 3968ed29b0b4SJens Axboe io_poll_cancel_req(req); 3969ed29b0b4SJens Axboe found = true; 3970ed29b0b4SJens Axboe } 3971ed29b0b4SJens Axboe } 3972ed29b0b4SJens Axboe } 3973ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 3974ed29b0b4SJens Axboe return found; 3975ed29b0b4SJens Axboe } 3976ed29b0b4SJens Axboe 3977ed29b0b4SJens Axboe static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only, 3978ed29b0b4SJens Axboe struct io_cancel_data *cd) 3979ed29b0b4SJens Axboe __must_hold(&ctx->completion_lock) 3980ed29b0b4SJens Axboe { 3981ed29b0b4SJens Axboe struct hlist_head *list; 3982ed29b0b4SJens Axboe struct io_kiocb *req; 3983ed29b0b4SJens Axboe 3984ed29b0b4SJens Axboe list = &ctx->cancel_hash[hash_long(cd->data, ctx->cancel_hash_bits)]; 3985ed29b0b4SJens Axboe hlist_for_each_entry(req, list, hash_node) { 3986ed29b0b4SJens Axboe if (cd->data != req->cqe.user_data) 3987ed29b0b4SJens Axboe continue; 3988ed29b0b4SJens Axboe if (poll_only && req->opcode != IORING_OP_POLL_ADD) 3989ed29b0b4SJens Axboe continue; 3990ed29b0b4SJens Axboe if (cd->flags & IORING_ASYNC_CANCEL_ALL) { 3991ed29b0b4SJens Axboe if (cd->seq == req->work.cancel_seq) 3992ed29b0b4SJens Axboe continue; 3993ed29b0b4SJens Axboe req->work.cancel_seq = cd->seq; 3994ed29b0b4SJens Axboe } 3995ed29b0b4SJens Axboe return req; 3996ed29b0b4SJens Axboe } 3997ed29b0b4SJens Axboe return NULL; 3998ed29b0b4SJens Axboe } 3999ed29b0b4SJens Axboe 4000ed29b0b4SJens Axboe static struct io_kiocb *io_poll_file_find(struct io_ring_ctx *ctx, 4001ed29b0b4SJens Axboe struct io_cancel_data *cd) 4002ed29b0b4SJens Axboe __must_hold(&ctx->completion_lock) 4003ed29b0b4SJens Axboe { 4004ed29b0b4SJens Axboe struct io_kiocb *req; 4005ed29b0b4SJens Axboe int i; 4006ed29b0b4SJens Axboe 4007ed29b0b4SJens Axboe for (i = 0; i < (1U << ctx->cancel_hash_bits); i++) { 4008ed29b0b4SJens Axboe struct hlist_head *list; 4009ed29b0b4SJens Axboe 4010ed29b0b4SJens Axboe list = &ctx->cancel_hash[i]; 4011ed29b0b4SJens Axboe hlist_for_each_entry(req, list, hash_node) { 4012ed29b0b4SJens Axboe if (!(cd->flags & IORING_ASYNC_CANCEL_ANY) && 4013ed29b0b4SJens Axboe req->file != cd->file) 4014ed29b0b4SJens Axboe continue; 4015ed29b0b4SJens Axboe if (cd->seq == req->work.cancel_seq) 4016ed29b0b4SJens Axboe continue; 4017ed29b0b4SJens Axboe req->work.cancel_seq = cd->seq; 4018ed29b0b4SJens Axboe return req; 4019ed29b0b4SJens Axboe } 4020ed29b0b4SJens Axboe } 4021ed29b0b4SJens Axboe return NULL; 4022ed29b0b4SJens Axboe } 4023ed29b0b4SJens Axboe 4024ed29b0b4SJens Axboe static bool io_poll_disarm(struct io_kiocb *req) 4025ed29b0b4SJens Axboe __must_hold(&ctx->completion_lock) 4026ed29b0b4SJens Axboe { 4027ed29b0b4SJens Axboe if (!io_poll_get_ownership(req)) 4028ed29b0b4SJens Axboe return false; 4029ed29b0b4SJens Axboe io_poll_remove_entries(req); 4030ed29b0b4SJens Axboe hash_del(&req->hash_node); 4031ed29b0b4SJens Axboe return true; 4032ed29b0b4SJens Axboe } 4033ed29b0b4SJens Axboe 4034ed29b0b4SJens Axboe static int io_poll_cancel(struct io_ring_ctx *ctx, struct io_cancel_data *cd) 4035ed29b0b4SJens Axboe __must_hold(&ctx->completion_lock) 4036ed29b0b4SJens Axboe { 4037ed29b0b4SJens Axboe struct io_kiocb *req; 4038ed29b0b4SJens Axboe 4039ed29b0b4SJens Axboe if (cd->flags & (IORING_ASYNC_CANCEL_FD|IORING_ASYNC_CANCEL_ANY)) 4040ed29b0b4SJens Axboe req = io_poll_file_find(ctx, cd); 4041ed29b0b4SJens Axboe else 4042ed29b0b4SJens Axboe req = io_poll_find(ctx, false, cd); 4043ed29b0b4SJens Axboe if (!req) 4044ed29b0b4SJens Axboe return -ENOENT; 4045ed29b0b4SJens Axboe io_poll_cancel_req(req); 4046ed29b0b4SJens Axboe return 0; 4047ed29b0b4SJens Axboe } 4048ed29b0b4SJens Axboe 4049ed29b0b4SJens Axboe static __poll_t io_poll_parse_events(const struct io_uring_sqe *sqe, 4050ed29b0b4SJens Axboe unsigned int flags) 4051ed29b0b4SJens Axboe { 4052ed29b0b4SJens Axboe u32 events; 4053ed29b0b4SJens Axboe 4054ed29b0b4SJens Axboe events = READ_ONCE(sqe->poll32_events); 4055ed29b0b4SJens Axboe #ifdef __BIG_ENDIAN 4056ed29b0b4SJens Axboe events = swahw32(events); 4057ed29b0b4SJens Axboe #endif 4058ed29b0b4SJens Axboe if (!(flags & IORING_POLL_ADD_MULTI)) 4059ed29b0b4SJens Axboe events |= EPOLLONESHOT; 4060ed29b0b4SJens Axboe return demangle_poll(events) | (events & (EPOLLEXCLUSIVE|EPOLLONESHOT)); 4061ed29b0b4SJens Axboe } 4062ed29b0b4SJens Axboe 4063ed29b0b4SJens Axboe static int io_poll_remove_prep(struct io_kiocb *req, 4064ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4065ed29b0b4SJens Axboe { 4066c24b1549SJens Axboe struct io_poll_update *upd = io_kiocb_to_cmd(req); 4067ed29b0b4SJens Axboe u32 flags; 4068ed29b0b4SJens Axboe 4069ed29b0b4SJens Axboe if (sqe->buf_index || sqe->splice_fd_in) 4070ed29b0b4SJens Axboe return -EINVAL; 4071ed29b0b4SJens Axboe flags = READ_ONCE(sqe->len); 4072ed29b0b4SJens Axboe if (flags & ~(IORING_POLL_UPDATE_EVENTS | IORING_POLL_UPDATE_USER_DATA | 4073ed29b0b4SJens Axboe IORING_POLL_ADD_MULTI)) 4074ed29b0b4SJens Axboe return -EINVAL; 4075ed29b0b4SJens Axboe /* meaningless without update */ 4076ed29b0b4SJens Axboe if (flags == IORING_POLL_ADD_MULTI) 4077ed29b0b4SJens Axboe return -EINVAL; 4078ed29b0b4SJens Axboe 4079ed29b0b4SJens Axboe upd->old_user_data = READ_ONCE(sqe->addr); 4080ed29b0b4SJens Axboe upd->update_events = flags & IORING_POLL_UPDATE_EVENTS; 4081ed29b0b4SJens Axboe upd->update_user_data = flags & IORING_POLL_UPDATE_USER_DATA; 4082ed29b0b4SJens Axboe 4083ed29b0b4SJens Axboe upd->new_user_data = READ_ONCE(sqe->off); 4084ed29b0b4SJens Axboe if (!upd->update_user_data && upd->new_user_data) 4085ed29b0b4SJens Axboe return -EINVAL; 4086ed29b0b4SJens Axboe if (upd->update_events) 4087ed29b0b4SJens Axboe upd->events = io_poll_parse_events(sqe, flags); 4088ed29b0b4SJens Axboe else if (sqe->poll32_events) 4089ed29b0b4SJens Axboe return -EINVAL; 4090ed29b0b4SJens Axboe 4091ed29b0b4SJens Axboe return 0; 4092ed29b0b4SJens Axboe } 4093ed29b0b4SJens Axboe 4094ed29b0b4SJens Axboe static int io_poll_add_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 4095ed29b0b4SJens Axboe { 40968d4388d1SJens Axboe struct io_poll *poll = io_kiocb_to_cmd(req); 4097ed29b0b4SJens Axboe u32 flags; 4098ed29b0b4SJens Axboe 4099ed29b0b4SJens Axboe if (sqe->buf_index || sqe->off || sqe->addr) 4100ed29b0b4SJens Axboe return -EINVAL; 4101ed29b0b4SJens Axboe flags = READ_ONCE(sqe->len); 4102ed29b0b4SJens Axboe if (flags & ~IORING_POLL_ADD_MULTI) 4103ed29b0b4SJens Axboe return -EINVAL; 4104ed29b0b4SJens Axboe if ((flags & IORING_POLL_ADD_MULTI) && (req->flags & REQ_F_CQE_SKIP)) 4105ed29b0b4SJens Axboe return -EINVAL; 4106ed29b0b4SJens Axboe 4107ed29b0b4SJens Axboe io_req_set_refcount(req); 4108ed29b0b4SJens Axboe poll->events = io_poll_parse_events(sqe, flags); 4109ed29b0b4SJens Axboe return 0; 4110ed29b0b4SJens Axboe } 4111ed29b0b4SJens Axboe 4112ed29b0b4SJens Axboe static int io_poll_add(struct io_kiocb *req, unsigned int issue_flags) 4113ed29b0b4SJens Axboe { 41148d4388d1SJens Axboe struct io_poll *poll = io_kiocb_to_cmd(req); 4115ed29b0b4SJens Axboe struct io_poll_table ipt; 4116ed29b0b4SJens Axboe int ret; 4117ed29b0b4SJens Axboe 4118ed29b0b4SJens Axboe ipt.pt._qproc = io_poll_queue_proc; 4119ed29b0b4SJens Axboe 41208d4388d1SJens Axboe ret = __io_arm_poll_handler(req, poll, &ipt, poll->events); 412197b388d7SJens Axboe if (ret) { 412297b388d7SJens Axboe io_req_set_res(req, ret, 0); 412397b388d7SJens Axboe return IOU_OK; 412497b388d7SJens Axboe } 412597b388d7SJens Axboe if (ipt.error) { 4126ed29b0b4SJens Axboe req_set_fail(req); 412797b388d7SJens Axboe return ipt.error; 412897b388d7SJens Axboe } 412997b388d7SJens Axboe 413097b388d7SJens Axboe return IOU_ISSUE_SKIP_COMPLETE; 4131ed29b0b4SJens Axboe } 4132ed29b0b4SJens Axboe 4133ed29b0b4SJens Axboe static int io_poll_remove(struct io_kiocb *req, unsigned int issue_flags) 4134ed29b0b4SJens Axboe { 4135c24b1549SJens Axboe struct io_poll_update *poll_update = io_kiocb_to_cmd(req); 4136c24b1549SJens Axboe struct io_cancel_data cd = { .data = poll_update->old_user_data, }; 4137ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 4138ed29b0b4SJens Axboe struct io_kiocb *preq; 4139ed29b0b4SJens Axboe int ret2, ret = 0; 4140ed29b0b4SJens Axboe bool locked; 4141ed29b0b4SJens Axboe 4142ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 4143ed29b0b4SJens Axboe preq = io_poll_find(ctx, true, &cd); 4144ed29b0b4SJens Axboe if (!preq || !io_poll_disarm(preq)) { 4145ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 4146ed29b0b4SJens Axboe ret = preq ? -EALREADY : -ENOENT; 4147ed29b0b4SJens Axboe goto out; 4148ed29b0b4SJens Axboe } 4149ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 4150ed29b0b4SJens Axboe 4151c24b1549SJens Axboe if (poll_update->update_events || poll_update->update_user_data) { 4152ed29b0b4SJens Axboe /* only mask one event flags, keep behavior flags */ 4153c24b1549SJens Axboe if (poll_update->update_events) { 41548d4388d1SJens Axboe struct io_poll *poll = io_kiocb_to_cmd(preq); 41558d4388d1SJens Axboe 41568d4388d1SJens Axboe poll->events &= ~0xffff; 4157c24b1549SJens Axboe poll->events |= poll_update->events & 0xffff; 41588d4388d1SJens Axboe poll->events |= IO_POLL_UNMASK; 4159ed29b0b4SJens Axboe } 4160c24b1549SJens Axboe if (poll_update->update_user_data) 4161c24b1549SJens Axboe preq->cqe.user_data = poll_update->new_user_data; 4162ed29b0b4SJens Axboe 4163ed29b0b4SJens Axboe ret2 = io_poll_add(preq, issue_flags); 4164ed29b0b4SJens Axboe /* successfully updated, don't complete poll request */ 416597b388d7SJens Axboe if (!ret2 || ret2 == -EIOCBQUEUED) 4166ed29b0b4SJens Axboe goto out; 4167ed29b0b4SJens Axboe } 4168ed29b0b4SJens Axboe 4169ed29b0b4SJens Axboe req_set_fail(preq); 417097b388d7SJens Axboe io_req_set_res(preq, -ECANCELED, 0); 4171ed29b0b4SJens Axboe locked = !(issue_flags & IO_URING_F_UNLOCKED); 4172ed29b0b4SJens Axboe io_req_task_complete(preq, &locked); 4173ed29b0b4SJens Axboe out: 417497b388d7SJens Axboe if (ret < 0) { 4175ed29b0b4SJens Axboe req_set_fail(req); 417697b388d7SJens Axboe return ret; 417797b388d7SJens Axboe } 4178ed29b0b4SJens Axboe /* complete update request, we're done with it */ 417997b388d7SJens Axboe io_req_set_res(req, ret, 0); 418097b388d7SJens Axboe return IOU_OK; 4181ed29b0b4SJens Axboe } 4182ed29b0b4SJens Axboe 4183ed29b0b4SJens Axboe static bool io_cancel_cb(struct io_wq_work *work, void *data) 4184ed29b0b4SJens Axboe { 4185ed29b0b4SJens Axboe struct io_kiocb *req = container_of(work, struct io_kiocb, work); 4186ed29b0b4SJens Axboe struct io_cancel_data *cd = data; 4187ed29b0b4SJens Axboe 4188ed29b0b4SJens Axboe if (req->ctx != cd->ctx) 4189ed29b0b4SJens Axboe return false; 4190ed29b0b4SJens Axboe if (cd->flags & IORING_ASYNC_CANCEL_ANY) { 4191ed29b0b4SJens Axboe ; 4192ed29b0b4SJens Axboe } else if (cd->flags & IORING_ASYNC_CANCEL_FD) { 4193ed29b0b4SJens Axboe if (req->file != cd->file) 4194ed29b0b4SJens Axboe return false; 4195ed29b0b4SJens Axboe } else { 4196ed29b0b4SJens Axboe if (req->cqe.user_data != cd->data) 4197ed29b0b4SJens Axboe return false; 4198ed29b0b4SJens Axboe } 4199ed29b0b4SJens Axboe if (cd->flags & (IORING_ASYNC_CANCEL_ALL|IORING_ASYNC_CANCEL_ANY)) { 4200ed29b0b4SJens Axboe if (cd->seq == req->work.cancel_seq) 4201ed29b0b4SJens Axboe return false; 4202ed29b0b4SJens Axboe req->work.cancel_seq = cd->seq; 4203ed29b0b4SJens Axboe } 4204ed29b0b4SJens Axboe return true; 4205ed29b0b4SJens Axboe } 4206ed29b0b4SJens Axboe 4207ed29b0b4SJens Axboe static int io_async_cancel_one(struct io_uring_task *tctx, 4208ed29b0b4SJens Axboe struct io_cancel_data *cd) 4209ed29b0b4SJens Axboe { 4210ed29b0b4SJens Axboe enum io_wq_cancel cancel_ret; 4211ed29b0b4SJens Axboe int ret = 0; 4212ed29b0b4SJens Axboe bool all; 4213ed29b0b4SJens Axboe 4214ed29b0b4SJens Axboe if (!tctx || !tctx->io_wq) 4215ed29b0b4SJens Axboe return -ENOENT; 4216ed29b0b4SJens Axboe 4217ed29b0b4SJens Axboe all = cd->flags & (IORING_ASYNC_CANCEL_ALL|IORING_ASYNC_CANCEL_ANY); 4218ed29b0b4SJens Axboe cancel_ret = io_wq_cancel_cb(tctx->io_wq, io_cancel_cb, cd, all); 4219ed29b0b4SJens Axboe switch (cancel_ret) { 4220ed29b0b4SJens Axboe case IO_WQ_CANCEL_OK: 4221ed29b0b4SJens Axboe ret = 0; 4222ed29b0b4SJens Axboe break; 4223ed29b0b4SJens Axboe case IO_WQ_CANCEL_RUNNING: 4224ed29b0b4SJens Axboe ret = -EALREADY; 4225ed29b0b4SJens Axboe break; 4226ed29b0b4SJens Axboe case IO_WQ_CANCEL_NOTFOUND: 4227ed29b0b4SJens Axboe ret = -ENOENT; 4228ed29b0b4SJens Axboe break; 4229ed29b0b4SJens Axboe } 4230ed29b0b4SJens Axboe 4231ed29b0b4SJens Axboe return ret; 4232ed29b0b4SJens Axboe } 4233ed29b0b4SJens Axboe 423459915143SJens Axboe int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd) 4235ed29b0b4SJens Axboe { 4236ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 4237ed29b0b4SJens Axboe int ret; 4238ed29b0b4SJens Axboe 4239ed29b0b4SJens Axboe WARN_ON_ONCE(!io_wq_current_is_worker() && req->task != current); 4240ed29b0b4SJens Axboe 4241ed29b0b4SJens Axboe ret = io_async_cancel_one(req->task->io_uring, cd); 4242ed29b0b4SJens Axboe /* 4243ed29b0b4SJens Axboe * Fall-through even for -EALREADY, as we may have poll armed 4244ed29b0b4SJens Axboe * that need unarming. 4245ed29b0b4SJens Axboe */ 4246ed29b0b4SJens Axboe if (!ret) 4247ed29b0b4SJens Axboe return 0; 4248ed29b0b4SJens Axboe 4249ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 4250ed29b0b4SJens Axboe ret = io_poll_cancel(ctx, cd); 4251ed29b0b4SJens Axboe if (ret != -ENOENT) 4252ed29b0b4SJens Axboe goto out; 4253ed29b0b4SJens Axboe if (!(cd->flags & IORING_ASYNC_CANCEL_FD)) 4254ed29b0b4SJens Axboe ret = io_timeout_cancel(ctx, cd); 4255ed29b0b4SJens Axboe out: 4256ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 4257ed29b0b4SJens Axboe return ret; 4258ed29b0b4SJens Axboe } 4259ed29b0b4SJens Axboe 4260ed29b0b4SJens Axboe #define CANCEL_FLAGS (IORING_ASYNC_CANCEL_ALL | IORING_ASYNC_CANCEL_FD | \ 4261ed29b0b4SJens Axboe IORING_ASYNC_CANCEL_ANY) 4262ed29b0b4SJens Axboe 4263ed29b0b4SJens Axboe static int io_async_cancel_prep(struct io_kiocb *req, 4264ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4265ed29b0b4SJens Axboe { 4266f38987f0SJens Axboe struct io_cancel *cancel = io_kiocb_to_cmd(req); 4267f38987f0SJens Axboe 4268ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_BUFFER_SELECT)) 4269ed29b0b4SJens Axboe return -EINVAL; 4270ed29b0b4SJens Axboe if (sqe->off || sqe->len || sqe->splice_fd_in) 4271ed29b0b4SJens Axboe return -EINVAL; 4272ed29b0b4SJens Axboe 4273f38987f0SJens Axboe cancel->addr = READ_ONCE(sqe->addr); 4274f38987f0SJens Axboe cancel->flags = READ_ONCE(sqe->cancel_flags); 4275f38987f0SJens Axboe if (cancel->flags & ~CANCEL_FLAGS) 4276ed29b0b4SJens Axboe return -EINVAL; 4277f38987f0SJens Axboe if (cancel->flags & IORING_ASYNC_CANCEL_FD) { 4278f38987f0SJens Axboe if (cancel->flags & IORING_ASYNC_CANCEL_ANY) 4279ed29b0b4SJens Axboe return -EINVAL; 4280f38987f0SJens Axboe cancel->fd = READ_ONCE(sqe->fd); 4281ed29b0b4SJens Axboe } 4282ed29b0b4SJens Axboe 4283ed29b0b4SJens Axboe return 0; 4284ed29b0b4SJens Axboe } 4285ed29b0b4SJens Axboe 4286ed29b0b4SJens Axboe static int __io_async_cancel(struct io_cancel_data *cd, struct io_kiocb *req, 4287ed29b0b4SJens Axboe unsigned int issue_flags) 4288ed29b0b4SJens Axboe { 4289ed29b0b4SJens Axboe bool all = cd->flags & (IORING_ASYNC_CANCEL_ALL|IORING_ASYNC_CANCEL_ANY); 4290ed29b0b4SJens Axboe struct io_ring_ctx *ctx = cd->ctx; 4291ed29b0b4SJens Axboe struct io_tctx_node *node; 4292ed29b0b4SJens Axboe int ret, nr = 0; 4293ed29b0b4SJens Axboe 4294ed29b0b4SJens Axboe do { 4295ed29b0b4SJens Axboe ret = io_try_cancel(req, cd); 4296ed29b0b4SJens Axboe if (ret == -ENOENT) 4297ed29b0b4SJens Axboe break; 4298ed29b0b4SJens Axboe if (!all) 4299ed29b0b4SJens Axboe return ret; 4300ed29b0b4SJens Axboe nr++; 4301ed29b0b4SJens Axboe } while (1); 4302ed29b0b4SJens Axboe 4303ed29b0b4SJens Axboe /* slow path, try all io-wq's */ 4304ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 4305ed29b0b4SJens Axboe ret = -ENOENT; 4306ed29b0b4SJens Axboe list_for_each_entry(node, &ctx->tctx_list, ctx_node) { 4307ed29b0b4SJens Axboe struct io_uring_task *tctx = node->task->io_uring; 4308ed29b0b4SJens Axboe 4309ed29b0b4SJens Axboe ret = io_async_cancel_one(tctx, cd); 4310ed29b0b4SJens Axboe if (ret != -ENOENT) { 4311ed29b0b4SJens Axboe if (!all) 4312ed29b0b4SJens Axboe break; 4313ed29b0b4SJens Axboe nr++; 4314ed29b0b4SJens Axboe } 4315ed29b0b4SJens Axboe } 4316ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 4317ed29b0b4SJens Axboe return all ? nr : ret; 4318ed29b0b4SJens Axboe } 4319ed29b0b4SJens Axboe 4320ed29b0b4SJens Axboe static int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags) 4321ed29b0b4SJens Axboe { 4322f38987f0SJens Axboe struct io_cancel *cancel = io_kiocb_to_cmd(req); 4323ed29b0b4SJens Axboe struct io_cancel_data cd = { 4324ed29b0b4SJens Axboe .ctx = req->ctx, 4325f38987f0SJens Axboe .data = cancel->addr, 4326f38987f0SJens Axboe .flags = cancel->flags, 4327ed29b0b4SJens Axboe .seq = atomic_inc_return(&req->ctx->cancel_seq), 4328ed29b0b4SJens Axboe }; 4329ed29b0b4SJens Axboe int ret; 4330ed29b0b4SJens Axboe 4331ed29b0b4SJens Axboe if (cd.flags & IORING_ASYNC_CANCEL_FD) { 4332ed29b0b4SJens Axboe if (req->flags & REQ_F_FIXED_FILE) 4333f38987f0SJens Axboe req->file = io_file_get_fixed(req, cancel->fd, 4334ed29b0b4SJens Axboe issue_flags); 4335ed29b0b4SJens Axboe else 4336f38987f0SJens Axboe req->file = io_file_get_normal(req, cancel->fd); 4337ed29b0b4SJens Axboe if (!req->file) { 4338ed29b0b4SJens Axboe ret = -EBADF; 4339ed29b0b4SJens Axboe goto done; 4340ed29b0b4SJens Axboe } 4341ed29b0b4SJens Axboe cd.file = req->file; 4342ed29b0b4SJens Axboe } 4343ed29b0b4SJens Axboe 4344ed29b0b4SJens Axboe ret = __io_async_cancel(&cd, req, issue_flags); 4345ed29b0b4SJens Axboe done: 4346ed29b0b4SJens Axboe if (ret < 0) 4347ed29b0b4SJens Axboe req_set_fail(req); 434897b388d7SJens Axboe io_req_set_res(req, ret, 0); 434997b388d7SJens Axboe return IOU_OK; 4350ed29b0b4SJens Axboe } 4351ed29b0b4SJens Axboe 4352ed29b0b4SJens Axboe static int io_files_update_prep(struct io_kiocb *req, 4353ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4354ed29b0b4SJens Axboe { 4355ea5af87dSJens Axboe struct io_rsrc_update *up = io_kiocb_to_cmd(req); 4356ea5af87dSJens Axboe 4357ed29b0b4SJens Axboe if (unlikely(req->flags & (REQ_F_FIXED_FILE | REQ_F_BUFFER_SELECT))) 4358ed29b0b4SJens Axboe return -EINVAL; 4359ed29b0b4SJens Axboe if (sqe->rw_flags || sqe->splice_fd_in) 4360ed29b0b4SJens Axboe return -EINVAL; 4361ed29b0b4SJens Axboe 4362ea5af87dSJens Axboe up->offset = READ_ONCE(sqe->off); 4363ea5af87dSJens Axboe up->nr_args = READ_ONCE(sqe->len); 4364ea5af87dSJens Axboe if (!up->nr_args) 4365ed29b0b4SJens Axboe return -EINVAL; 4366ea5af87dSJens Axboe up->arg = READ_ONCE(sqe->addr); 4367ed29b0b4SJens Axboe return 0; 4368ed29b0b4SJens Axboe } 4369ed29b0b4SJens Axboe 4370ed29b0b4SJens Axboe static int io_files_update_with_index_alloc(struct io_kiocb *req, 4371ed29b0b4SJens Axboe unsigned int issue_flags) 4372ed29b0b4SJens Axboe { 4373ea5af87dSJens Axboe struct io_rsrc_update *up = io_kiocb_to_cmd(req); 4374ea5af87dSJens Axboe __s32 __user *fds = u64_to_user_ptr(up->arg); 4375ed29b0b4SJens Axboe unsigned int done; 4376ed29b0b4SJens Axboe struct file *file; 4377ed29b0b4SJens Axboe int ret, fd; 4378ed29b0b4SJens Axboe 4379ed29b0b4SJens Axboe if (!req->ctx->file_data) 4380ed29b0b4SJens Axboe return -ENXIO; 4381ed29b0b4SJens Axboe 4382ea5af87dSJens Axboe for (done = 0; done < up->nr_args; done++) { 4383ed29b0b4SJens Axboe if (copy_from_user(&fd, &fds[done], sizeof(fd))) { 4384ed29b0b4SJens Axboe ret = -EFAULT; 4385ed29b0b4SJens Axboe break; 4386ed29b0b4SJens Axboe } 4387ed29b0b4SJens Axboe 4388ed29b0b4SJens Axboe file = fget(fd); 4389ed29b0b4SJens Axboe if (!file) { 4390ed29b0b4SJens Axboe ret = -EBADF; 4391ed29b0b4SJens Axboe break; 4392ed29b0b4SJens Axboe } 4393ed29b0b4SJens Axboe ret = io_fixed_fd_install(req, issue_flags, file, 4394ed29b0b4SJens Axboe IORING_FILE_INDEX_ALLOC); 4395ed29b0b4SJens Axboe if (ret < 0) 4396ed29b0b4SJens Axboe break; 4397ed29b0b4SJens Axboe if (copy_to_user(&fds[done], &ret, sizeof(ret))) { 4398ed29b0b4SJens Axboe __io_close_fixed(req, issue_flags, ret); 4399ed29b0b4SJens Axboe ret = -EFAULT; 4400ed29b0b4SJens Axboe break; 4401ed29b0b4SJens Axboe } 4402ed29b0b4SJens Axboe } 4403ed29b0b4SJens Axboe 4404ed29b0b4SJens Axboe if (done) 4405ed29b0b4SJens Axboe return done; 4406ed29b0b4SJens Axboe return ret; 4407ed29b0b4SJens Axboe } 4408ed29b0b4SJens Axboe 4409ed29b0b4SJens Axboe static int io_files_update(struct io_kiocb *req, unsigned int issue_flags) 4410ed29b0b4SJens Axboe { 4411ea5af87dSJens Axboe struct io_rsrc_update *up = io_kiocb_to_cmd(req); 4412ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 4413ea5af87dSJens Axboe struct io_uring_rsrc_update2 up2; 4414ed29b0b4SJens Axboe int ret; 4415ed29b0b4SJens Axboe 4416ea5af87dSJens Axboe up2.offset = up->offset; 4417ea5af87dSJens Axboe up2.data = up->arg; 4418ea5af87dSJens Axboe up2.nr = 0; 4419ea5af87dSJens Axboe up2.tags = 0; 4420ea5af87dSJens Axboe up2.resv = 0; 4421ea5af87dSJens Axboe up2.resv2 = 0; 4422ed29b0b4SJens Axboe 4423ea5af87dSJens Axboe if (up->offset == IORING_FILE_INDEX_ALLOC) { 4424ed29b0b4SJens Axboe ret = io_files_update_with_index_alloc(req, issue_flags); 4425ed29b0b4SJens Axboe } else { 4426ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 4427ed29b0b4SJens Axboe ret = __io_register_rsrc_update(ctx, IORING_RSRC_FILE, 4428ea5af87dSJens Axboe &up2, up->nr_args); 4429ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 4430ed29b0b4SJens Axboe } 4431ed29b0b4SJens Axboe 4432ed29b0b4SJens Axboe if (ret < 0) 4433ed29b0b4SJens Axboe req_set_fail(req); 443497b388d7SJens Axboe io_req_set_res(req, ret, 0); 443597b388d7SJens Axboe return IOU_OK; 4436ed29b0b4SJens Axboe } 4437ed29b0b4SJens Axboe 4438ed29b0b4SJens Axboe static int io_req_prep_async(struct io_kiocb *req) 4439ed29b0b4SJens Axboe { 4440ed29b0b4SJens Axboe const struct io_op_def *def = &io_op_defs[req->opcode]; 4441ed29b0b4SJens Axboe 4442ed29b0b4SJens Axboe /* assign early for deferred execution for non-fixed file */ 4443ed29b0b4SJens Axboe if (def->needs_file && !(req->flags & REQ_F_FIXED_FILE)) 4444ed29b0b4SJens Axboe req->file = io_file_get_normal(req, req->cqe.fd); 4445dc919cafSJens Axboe if (!def->prep_async) 4446ed29b0b4SJens Axboe return 0; 4447ed29b0b4SJens Axboe if (WARN_ON_ONCE(req_has_async_data(req))) 4448ed29b0b4SJens Axboe return -EFAULT; 4449ed29b0b4SJens Axboe if (io_alloc_async_data(req)) 4450ed29b0b4SJens Axboe return -EAGAIN; 4451ed29b0b4SJens Axboe 4452dc919cafSJens Axboe return def->prep_async(req); 4453ed29b0b4SJens Axboe } 4454ed29b0b4SJens Axboe 4455ed29b0b4SJens Axboe static u32 io_get_sequence(struct io_kiocb *req) 4456ed29b0b4SJens Axboe { 4457ed29b0b4SJens Axboe u32 seq = req->ctx->cached_sq_head; 4458ed29b0b4SJens Axboe struct io_kiocb *cur; 4459ed29b0b4SJens Axboe 4460ed29b0b4SJens Axboe /* need original cached_sq_head, but it was increased for each req */ 4461ed29b0b4SJens Axboe io_for_each_link(cur, req) 4462ed29b0b4SJens Axboe seq--; 4463ed29b0b4SJens Axboe return seq; 4464ed29b0b4SJens Axboe } 4465ed29b0b4SJens Axboe 4466ed29b0b4SJens Axboe static __cold void io_drain_req(struct io_kiocb *req) 4467ed29b0b4SJens Axboe { 4468ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 4469ed29b0b4SJens Axboe struct io_defer_entry *de; 4470ed29b0b4SJens Axboe int ret; 4471ed29b0b4SJens Axboe u32 seq = io_get_sequence(req); 4472ed29b0b4SJens Axboe 4473ed29b0b4SJens Axboe /* Still need defer if there is pending req in defer list. */ 4474ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 4475ed29b0b4SJens Axboe if (!req_need_defer(req, seq) && list_empty_careful(&ctx->defer_list)) { 4476ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 4477ed29b0b4SJens Axboe queue: 4478ed29b0b4SJens Axboe ctx->drain_active = false; 4479ed29b0b4SJens Axboe io_req_task_queue(req); 4480ed29b0b4SJens Axboe return; 4481ed29b0b4SJens Axboe } 4482ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 4483ed29b0b4SJens Axboe 4484ed29b0b4SJens Axboe ret = io_req_prep_async(req); 4485ed29b0b4SJens Axboe if (ret) { 4486ed29b0b4SJens Axboe fail: 4487ed29b0b4SJens Axboe io_req_complete_failed(req, ret); 4488ed29b0b4SJens Axboe return; 4489ed29b0b4SJens Axboe } 4490ed29b0b4SJens Axboe io_prep_async_link(req); 4491ed29b0b4SJens Axboe de = kmalloc(sizeof(*de), GFP_KERNEL); 4492ed29b0b4SJens Axboe if (!de) { 4493ed29b0b4SJens Axboe ret = -ENOMEM; 4494ed29b0b4SJens Axboe goto fail; 4495ed29b0b4SJens Axboe } 4496ed29b0b4SJens Axboe 4497ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 4498ed29b0b4SJens Axboe if (!req_need_defer(req, seq) && list_empty(&ctx->defer_list)) { 4499ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 4500ed29b0b4SJens Axboe kfree(de); 4501ed29b0b4SJens Axboe goto queue; 4502ed29b0b4SJens Axboe } 4503ed29b0b4SJens Axboe 4504ed29b0b4SJens Axboe trace_io_uring_defer(ctx, req, req->cqe.user_data, req->opcode); 4505ed29b0b4SJens Axboe de->req = req; 4506ed29b0b4SJens Axboe de->seq = seq; 4507ed29b0b4SJens Axboe list_add_tail(&de->list, &ctx->defer_list); 4508ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 4509ed29b0b4SJens Axboe } 4510ed29b0b4SJens Axboe 4511ed29b0b4SJens Axboe static void io_clean_op(struct io_kiocb *req) 4512ed29b0b4SJens Axboe { 4513ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_SELECTED) { 4514ed29b0b4SJens Axboe spin_lock(&req->ctx->completion_lock); 4515ed29b0b4SJens Axboe io_put_kbuf_comp(req); 4516ed29b0b4SJens Axboe spin_unlock(&req->ctx->completion_lock); 4517ed29b0b4SJens Axboe } 4518ed29b0b4SJens Axboe 4519ed29b0b4SJens Axboe if (req->flags & REQ_F_NEED_CLEANUP) { 45204d4c9cffSJens Axboe const struct io_op_def *def = &io_op_defs[req->opcode]; 4521ed29b0b4SJens Axboe 45224d4c9cffSJens Axboe if (def->cleanup) 45234d4c9cffSJens Axboe def->cleanup(req); 4524ed29b0b4SJens Axboe } 4525ed29b0b4SJens Axboe if ((req->flags & REQ_F_POLLED) && req->apoll) { 4526ed29b0b4SJens Axboe kfree(req->apoll->double_poll); 4527ed29b0b4SJens Axboe kfree(req->apoll); 4528ed29b0b4SJens Axboe req->apoll = NULL; 4529ed29b0b4SJens Axboe } 4530ed29b0b4SJens Axboe if (req->flags & REQ_F_INFLIGHT) { 4531ed29b0b4SJens Axboe struct io_uring_task *tctx = req->task->io_uring; 4532ed29b0b4SJens Axboe 4533ed29b0b4SJens Axboe atomic_dec(&tctx->inflight_tracked); 4534ed29b0b4SJens Axboe } 4535ed29b0b4SJens Axboe if (req->flags & REQ_F_CREDS) 4536ed29b0b4SJens Axboe put_cred(req->creds); 4537ed29b0b4SJens Axboe if (req->flags & REQ_F_ASYNC_DATA) { 4538ed29b0b4SJens Axboe kfree(req->async_data); 4539ed29b0b4SJens Axboe req->async_data = NULL; 4540ed29b0b4SJens Axboe } 4541ed29b0b4SJens Axboe req->flags &= ~IO_REQ_CLEAN_FLAGS; 4542ed29b0b4SJens Axboe } 4543ed29b0b4SJens Axboe 4544ed29b0b4SJens Axboe static bool io_assign_file(struct io_kiocb *req, unsigned int issue_flags) 4545ed29b0b4SJens Axboe { 4546ed29b0b4SJens Axboe if (req->file || !io_op_defs[req->opcode].needs_file) 4547ed29b0b4SJens Axboe return true; 4548ed29b0b4SJens Axboe 4549ed29b0b4SJens Axboe if (req->flags & REQ_F_FIXED_FILE) 4550ed29b0b4SJens Axboe req->file = io_file_get_fixed(req, req->cqe.fd, issue_flags); 4551ed29b0b4SJens Axboe else 4552ed29b0b4SJens Axboe req->file = io_file_get_normal(req, req->cqe.fd); 4553ed29b0b4SJens Axboe 4554ed29b0b4SJens Axboe return !!req->file; 4555ed29b0b4SJens Axboe } 4556ed29b0b4SJens Axboe 4557ed29b0b4SJens Axboe static int io_issue_sqe(struct io_kiocb *req, unsigned int issue_flags) 4558ed29b0b4SJens Axboe { 4559ed29b0b4SJens Axboe const struct io_op_def *def = &io_op_defs[req->opcode]; 4560ed29b0b4SJens Axboe const struct cred *creds = NULL; 4561ed29b0b4SJens Axboe int ret; 4562ed29b0b4SJens Axboe 4563ed29b0b4SJens Axboe if (unlikely(!io_assign_file(req, issue_flags))) 4564ed29b0b4SJens Axboe return -EBADF; 4565ed29b0b4SJens Axboe 4566ed29b0b4SJens Axboe if (unlikely((req->flags & REQ_F_CREDS) && req->creds != current_cred())) 4567ed29b0b4SJens Axboe creds = override_creds(req->creds); 4568ed29b0b4SJens Axboe 4569ed29b0b4SJens Axboe if (!def->audit_skip) 4570ed29b0b4SJens Axboe audit_uring_entry(req->opcode); 4571ed29b0b4SJens Axboe 4572ed29b0b4SJens Axboe ret = def->issue(req, issue_flags); 4573ed29b0b4SJens Axboe 4574ed29b0b4SJens Axboe if (!def->audit_skip) 4575ed29b0b4SJens Axboe audit_uring_exit(!ret, ret); 4576ed29b0b4SJens Axboe 4577ed29b0b4SJens Axboe if (creds) 4578ed29b0b4SJens Axboe revert_creds(creds); 457997b388d7SJens Axboe 458097b388d7SJens Axboe if (ret == IOU_OK) 458197b388d7SJens Axboe __io_req_complete(req, issue_flags); 458297b388d7SJens Axboe else if (ret != IOU_ISSUE_SKIP_COMPLETE) 4583ed29b0b4SJens Axboe return ret; 458497b388d7SJens Axboe 4585ed29b0b4SJens Axboe /* If the op doesn't have a file, we're not polling for it */ 4586ed29b0b4SJens Axboe if ((req->ctx->flags & IORING_SETUP_IOPOLL) && req->file) 4587ed29b0b4SJens Axboe io_iopoll_req_issued(req, issue_flags); 4588ed29b0b4SJens Axboe 4589ed29b0b4SJens Axboe return 0; 4590ed29b0b4SJens Axboe } 4591ed29b0b4SJens Axboe 4592c9f06aa7SJens Axboe struct io_wq_work *io_wq_free_work(struct io_wq_work *work) 4593ed29b0b4SJens Axboe { 4594ed29b0b4SJens Axboe struct io_kiocb *req = container_of(work, struct io_kiocb, work); 4595ed29b0b4SJens Axboe 4596ed29b0b4SJens Axboe req = io_put_req_find_next(req); 4597ed29b0b4SJens Axboe return req ? &req->work : NULL; 4598ed29b0b4SJens Axboe } 4599ed29b0b4SJens Axboe 4600c9f06aa7SJens Axboe void io_wq_submit_work(struct io_wq_work *work) 4601ed29b0b4SJens Axboe { 4602ed29b0b4SJens Axboe struct io_kiocb *req = container_of(work, struct io_kiocb, work); 4603ed29b0b4SJens Axboe const struct io_op_def *def = &io_op_defs[req->opcode]; 4604ed29b0b4SJens Axboe unsigned int issue_flags = IO_URING_F_UNLOCKED; 4605ed29b0b4SJens Axboe bool needs_poll = false; 4606ed29b0b4SJens Axboe int ret = 0, err = -ECANCELED; 4607ed29b0b4SJens Axboe 4608ed29b0b4SJens Axboe /* one will be dropped by ->io_free_work() after returning to io-wq */ 4609ed29b0b4SJens Axboe if (!(req->flags & REQ_F_REFCOUNT)) 4610ed29b0b4SJens Axboe __io_req_set_refcount(req, 2); 4611ed29b0b4SJens Axboe else 4612ed29b0b4SJens Axboe req_ref_get(req); 4613ed29b0b4SJens Axboe 4614ed29b0b4SJens Axboe io_arm_ltimeout(req); 4615ed29b0b4SJens Axboe 4616ed29b0b4SJens Axboe /* either cancelled or io-wq is dying, so don't touch tctx->iowq */ 4617ed29b0b4SJens Axboe if (work->flags & IO_WQ_WORK_CANCEL) { 4618ed29b0b4SJens Axboe fail: 4619ed29b0b4SJens Axboe io_req_task_queue_fail(req, err); 4620ed29b0b4SJens Axboe return; 4621ed29b0b4SJens Axboe } 4622ed29b0b4SJens Axboe if (!io_assign_file(req, issue_flags)) { 4623ed29b0b4SJens Axboe err = -EBADF; 4624ed29b0b4SJens Axboe work->flags |= IO_WQ_WORK_CANCEL; 4625ed29b0b4SJens Axboe goto fail; 4626ed29b0b4SJens Axboe } 4627ed29b0b4SJens Axboe 4628ed29b0b4SJens Axboe if (req->flags & REQ_F_FORCE_ASYNC) { 4629ed29b0b4SJens Axboe bool opcode_poll = def->pollin || def->pollout; 4630ed29b0b4SJens Axboe 4631ed29b0b4SJens Axboe if (opcode_poll && file_can_poll(req->file)) { 4632ed29b0b4SJens Axboe needs_poll = true; 4633ed29b0b4SJens Axboe issue_flags |= IO_URING_F_NONBLOCK; 4634ed29b0b4SJens Axboe } 4635ed29b0b4SJens Axboe } 4636ed29b0b4SJens Axboe 4637ed29b0b4SJens Axboe do { 4638ed29b0b4SJens Axboe ret = io_issue_sqe(req, issue_flags); 4639ed29b0b4SJens Axboe if (ret != -EAGAIN) 4640ed29b0b4SJens Axboe break; 4641ed29b0b4SJens Axboe /* 4642ed29b0b4SJens Axboe * We can get EAGAIN for iopolled IO even though we're 4643ed29b0b4SJens Axboe * forcing a sync submission from here, since we can't 4644ed29b0b4SJens Axboe * wait for request slots on the block side. 4645ed29b0b4SJens Axboe */ 4646ed29b0b4SJens Axboe if (!needs_poll) { 4647ed29b0b4SJens Axboe if (!(req->ctx->flags & IORING_SETUP_IOPOLL)) 4648ed29b0b4SJens Axboe break; 4649ed29b0b4SJens Axboe cond_resched(); 4650ed29b0b4SJens Axboe continue; 4651ed29b0b4SJens Axboe } 4652ed29b0b4SJens Axboe 4653ed29b0b4SJens Axboe if (io_arm_poll_handler(req, issue_flags) == IO_APOLL_OK) 4654ed29b0b4SJens Axboe return; 4655ed29b0b4SJens Axboe /* aborted or ready, in either case retry blocking */ 4656ed29b0b4SJens Axboe needs_poll = false; 4657ed29b0b4SJens Axboe issue_flags &= ~IO_URING_F_NONBLOCK; 4658ed29b0b4SJens Axboe } while (1); 4659ed29b0b4SJens Axboe 4660ed29b0b4SJens Axboe /* avoid locking problems by failing it from a clean context */ 466197b388d7SJens Axboe if (ret < 0) 4662ed29b0b4SJens Axboe io_req_task_queue_fail(req, ret); 4663ed29b0b4SJens Axboe } 4664ed29b0b4SJens Axboe 4665531113bbSJens Axboe inline struct file *io_file_get_fixed(struct io_kiocb *req, int fd, 4666ed29b0b4SJens Axboe unsigned int issue_flags) 4667ed29b0b4SJens Axboe { 4668ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 4669ed29b0b4SJens Axboe struct file *file = NULL; 4670ed29b0b4SJens Axboe unsigned long file_ptr; 4671ed29b0b4SJens Axboe 4672ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 4673ed29b0b4SJens Axboe 4674ed29b0b4SJens Axboe if (unlikely((unsigned int)fd >= ctx->nr_user_files)) 4675ed29b0b4SJens Axboe goto out; 4676ed29b0b4SJens Axboe fd = array_index_nospec(fd, ctx->nr_user_files); 4677ed29b0b4SJens Axboe file_ptr = io_fixed_file_slot(&ctx->file_table, fd)->file_ptr; 4678ed29b0b4SJens Axboe file = (struct file *) (file_ptr & FFS_MASK); 4679ed29b0b4SJens Axboe file_ptr &= ~FFS_MASK; 4680ed29b0b4SJens Axboe /* mask in overlapping REQ_F and FFS bits */ 4681ed29b0b4SJens Axboe req->flags |= (file_ptr << REQ_F_SUPPORT_NOWAIT_BIT); 4682ed29b0b4SJens Axboe io_req_set_rsrc_node(req, ctx, 0); 4683ed29b0b4SJens Axboe WARN_ON_ONCE(file && !test_bit(fd, ctx->file_table.bitmap)); 4684ed29b0b4SJens Axboe out: 4685ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 4686ed29b0b4SJens Axboe return file; 4687ed29b0b4SJens Axboe } 4688ed29b0b4SJens Axboe 4689531113bbSJens Axboe struct file *io_file_get_normal(struct io_kiocb *req, int fd) 4690ed29b0b4SJens Axboe { 4691ed29b0b4SJens Axboe struct file *file = fget(fd); 4692ed29b0b4SJens Axboe 4693ed29b0b4SJens Axboe trace_io_uring_file_get(req->ctx, req, req->cqe.user_data, fd); 4694ed29b0b4SJens Axboe 4695ed29b0b4SJens Axboe /* we don't allow fixed io_uring files */ 4696e5550a14SJens Axboe if (file && io_is_uring_fops(file)) 4697ed29b0b4SJens Axboe io_req_track_inflight(req); 4698ed29b0b4SJens Axboe return file; 4699ed29b0b4SJens Axboe } 4700ed29b0b4SJens Axboe 4701ed29b0b4SJens Axboe static void io_queue_async(struct io_kiocb *req, int ret) 4702ed29b0b4SJens Axboe __must_hold(&req->ctx->uring_lock) 4703ed29b0b4SJens Axboe { 4704ed29b0b4SJens Axboe struct io_kiocb *linked_timeout; 4705ed29b0b4SJens Axboe 4706ed29b0b4SJens Axboe if (ret != -EAGAIN || (req->flags & REQ_F_NOWAIT)) { 4707ed29b0b4SJens Axboe io_req_complete_failed(req, ret); 4708ed29b0b4SJens Axboe return; 4709ed29b0b4SJens Axboe } 4710ed29b0b4SJens Axboe 4711ed29b0b4SJens Axboe linked_timeout = io_prep_linked_timeout(req); 4712ed29b0b4SJens Axboe 4713ed29b0b4SJens Axboe switch (io_arm_poll_handler(req, 0)) { 4714ed29b0b4SJens Axboe case IO_APOLL_READY: 4715ed29b0b4SJens Axboe io_req_task_queue(req); 4716ed29b0b4SJens Axboe break; 4717ed29b0b4SJens Axboe case IO_APOLL_ABORTED: 4718ed29b0b4SJens Axboe /* 4719ed29b0b4SJens Axboe * Queued up for async execution, worker will release 4720ed29b0b4SJens Axboe * submit reference when the iocb is actually submitted. 4721ed29b0b4SJens Axboe */ 4722ed29b0b4SJens Axboe io_kbuf_recycle(req, 0); 4723ed29b0b4SJens Axboe io_queue_iowq(req, NULL); 4724ed29b0b4SJens Axboe break; 4725ed29b0b4SJens Axboe case IO_APOLL_OK: 4726ed29b0b4SJens Axboe break; 4727ed29b0b4SJens Axboe } 4728ed29b0b4SJens Axboe 4729ed29b0b4SJens Axboe if (linked_timeout) 4730ed29b0b4SJens Axboe io_queue_linked_timeout(linked_timeout); 4731ed29b0b4SJens Axboe } 4732ed29b0b4SJens Axboe 4733ed29b0b4SJens Axboe static inline void io_queue_sqe(struct io_kiocb *req) 4734ed29b0b4SJens Axboe __must_hold(&req->ctx->uring_lock) 4735ed29b0b4SJens Axboe { 4736ed29b0b4SJens Axboe int ret; 4737ed29b0b4SJens Axboe 4738ed29b0b4SJens Axboe ret = io_issue_sqe(req, IO_URING_F_NONBLOCK|IO_URING_F_COMPLETE_DEFER); 4739ed29b0b4SJens Axboe 4740ed29b0b4SJens Axboe if (req->flags & REQ_F_COMPLETE_INLINE) { 4741ed29b0b4SJens Axboe io_req_add_compl_list(req); 4742ed29b0b4SJens Axboe return; 4743ed29b0b4SJens Axboe } 4744ed29b0b4SJens Axboe /* 4745ed29b0b4SJens Axboe * We async punt it if the file wasn't marked NOWAIT, or if the file 4746ed29b0b4SJens Axboe * doesn't support non-blocking read/write attempts 4747ed29b0b4SJens Axboe */ 4748ed29b0b4SJens Axboe if (likely(!ret)) 4749ed29b0b4SJens Axboe io_arm_ltimeout(req); 4750ed29b0b4SJens Axboe else 4751ed29b0b4SJens Axboe io_queue_async(req, ret); 4752ed29b0b4SJens Axboe } 4753ed29b0b4SJens Axboe 4754ed29b0b4SJens Axboe static void io_queue_sqe_fallback(struct io_kiocb *req) 4755ed29b0b4SJens Axboe __must_hold(&req->ctx->uring_lock) 4756ed29b0b4SJens Axboe { 4757ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_FAIL)) { 4758ed29b0b4SJens Axboe /* 4759ed29b0b4SJens Axboe * We don't submit, fail them all, for that replace hardlinks 4760ed29b0b4SJens Axboe * with normal links. Extra REQ_F_LINK is tolerated. 4761ed29b0b4SJens Axboe */ 4762ed29b0b4SJens Axboe req->flags &= ~REQ_F_HARDLINK; 4763ed29b0b4SJens Axboe req->flags |= REQ_F_LINK; 4764ed29b0b4SJens Axboe io_req_complete_failed(req, req->cqe.res); 4765ed29b0b4SJens Axboe } else if (unlikely(req->ctx->drain_active)) { 4766ed29b0b4SJens Axboe io_drain_req(req); 4767ed29b0b4SJens Axboe } else { 4768ed29b0b4SJens Axboe int ret = io_req_prep_async(req); 4769ed29b0b4SJens Axboe 4770ed29b0b4SJens Axboe if (unlikely(ret)) 4771ed29b0b4SJens Axboe io_req_complete_failed(req, ret); 4772ed29b0b4SJens Axboe else 4773ed29b0b4SJens Axboe io_queue_iowq(req, NULL); 4774ed29b0b4SJens Axboe } 4775ed29b0b4SJens Axboe } 4776ed29b0b4SJens Axboe 4777ed29b0b4SJens Axboe /* 4778ed29b0b4SJens Axboe * Check SQE restrictions (opcode and flags). 4779ed29b0b4SJens Axboe * 4780ed29b0b4SJens Axboe * Returns 'true' if SQE is allowed, 'false' otherwise. 4781ed29b0b4SJens Axboe */ 4782ed29b0b4SJens Axboe static inline bool io_check_restriction(struct io_ring_ctx *ctx, 4783ed29b0b4SJens Axboe struct io_kiocb *req, 4784ed29b0b4SJens Axboe unsigned int sqe_flags) 4785ed29b0b4SJens Axboe { 4786ed29b0b4SJens Axboe if (!test_bit(req->opcode, ctx->restrictions.sqe_op)) 4787ed29b0b4SJens Axboe return false; 4788ed29b0b4SJens Axboe 4789ed29b0b4SJens Axboe if ((sqe_flags & ctx->restrictions.sqe_flags_required) != 4790ed29b0b4SJens Axboe ctx->restrictions.sqe_flags_required) 4791ed29b0b4SJens Axboe return false; 4792ed29b0b4SJens Axboe 4793ed29b0b4SJens Axboe if (sqe_flags & ~(ctx->restrictions.sqe_flags_allowed | 4794ed29b0b4SJens Axboe ctx->restrictions.sqe_flags_required)) 4795ed29b0b4SJens Axboe return false; 4796ed29b0b4SJens Axboe 4797ed29b0b4SJens Axboe return true; 4798ed29b0b4SJens Axboe } 4799ed29b0b4SJens Axboe 4800ed29b0b4SJens Axboe static void io_init_req_drain(struct io_kiocb *req) 4801ed29b0b4SJens Axboe { 4802ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 4803ed29b0b4SJens Axboe struct io_kiocb *head = ctx->submit_state.link.head; 4804ed29b0b4SJens Axboe 4805ed29b0b4SJens Axboe ctx->drain_active = true; 4806ed29b0b4SJens Axboe if (head) { 4807ed29b0b4SJens Axboe /* 4808ed29b0b4SJens Axboe * If we need to drain a request in the middle of a link, drain 4809ed29b0b4SJens Axboe * the head request and the next request/link after the current 4810ed29b0b4SJens Axboe * link. Considering sequential execution of links, 4811ed29b0b4SJens Axboe * REQ_F_IO_DRAIN will be maintained for every request of our 4812ed29b0b4SJens Axboe * link. 4813ed29b0b4SJens Axboe */ 4814ed29b0b4SJens Axboe head->flags |= REQ_F_IO_DRAIN | REQ_F_FORCE_ASYNC; 4815ed29b0b4SJens Axboe ctx->drain_next = true; 4816ed29b0b4SJens Axboe } 4817ed29b0b4SJens Axboe } 4818ed29b0b4SJens Axboe 4819ed29b0b4SJens Axboe static int io_init_req(struct io_ring_ctx *ctx, struct io_kiocb *req, 4820ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4821ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 4822ed29b0b4SJens Axboe { 4823ed29b0b4SJens Axboe const struct io_op_def *def; 4824ed29b0b4SJens Axboe unsigned int sqe_flags; 4825ed29b0b4SJens Axboe int personality; 4826ed29b0b4SJens Axboe u8 opcode; 4827ed29b0b4SJens Axboe 4828ed29b0b4SJens Axboe /* req is partially pre-initialised, see io_preinit_req() */ 4829ed29b0b4SJens Axboe req->opcode = opcode = READ_ONCE(sqe->opcode); 4830ed29b0b4SJens Axboe /* same numerical values with corresponding REQ_F_*, safe to copy */ 4831ed29b0b4SJens Axboe req->flags = sqe_flags = READ_ONCE(sqe->flags); 4832ed29b0b4SJens Axboe req->cqe.user_data = READ_ONCE(sqe->user_data); 4833ed29b0b4SJens Axboe req->file = NULL; 4834ed29b0b4SJens Axboe req->rsrc_node = NULL; 4835ed29b0b4SJens Axboe req->task = current; 4836ed29b0b4SJens Axboe 4837ed29b0b4SJens Axboe if (unlikely(opcode >= IORING_OP_LAST)) { 4838ed29b0b4SJens Axboe req->opcode = 0; 4839ed29b0b4SJens Axboe return -EINVAL; 4840ed29b0b4SJens Axboe } 4841ed29b0b4SJens Axboe def = &io_op_defs[opcode]; 4842ed29b0b4SJens Axboe if (unlikely(sqe_flags & ~SQE_COMMON_FLAGS)) { 4843ed29b0b4SJens Axboe /* enforce forwards compatibility on users */ 4844ed29b0b4SJens Axboe if (sqe_flags & ~SQE_VALID_FLAGS) 4845ed29b0b4SJens Axboe return -EINVAL; 4846ed29b0b4SJens Axboe if (sqe_flags & IOSQE_BUFFER_SELECT) { 4847ed29b0b4SJens Axboe if (!def->buffer_select) 4848ed29b0b4SJens Axboe return -EOPNOTSUPP; 4849ed29b0b4SJens Axboe req->buf_index = READ_ONCE(sqe->buf_group); 4850ed29b0b4SJens Axboe } 4851ed29b0b4SJens Axboe if (sqe_flags & IOSQE_CQE_SKIP_SUCCESS) 4852ed29b0b4SJens Axboe ctx->drain_disabled = true; 4853ed29b0b4SJens Axboe if (sqe_flags & IOSQE_IO_DRAIN) { 4854ed29b0b4SJens Axboe if (ctx->drain_disabled) 4855ed29b0b4SJens Axboe return -EOPNOTSUPP; 4856ed29b0b4SJens Axboe io_init_req_drain(req); 4857ed29b0b4SJens Axboe } 4858ed29b0b4SJens Axboe } 4859ed29b0b4SJens Axboe if (unlikely(ctx->restricted || ctx->drain_active || ctx->drain_next)) { 4860ed29b0b4SJens Axboe if (ctx->restricted && !io_check_restriction(ctx, req, sqe_flags)) 4861ed29b0b4SJens Axboe return -EACCES; 4862ed29b0b4SJens Axboe /* knock it to the slow queue path, will be drained there */ 4863ed29b0b4SJens Axboe if (ctx->drain_active) 4864ed29b0b4SJens Axboe req->flags |= REQ_F_FORCE_ASYNC; 4865ed29b0b4SJens Axboe /* if there is no link, we're at "next" request and need to drain */ 4866ed29b0b4SJens Axboe if (unlikely(ctx->drain_next) && !ctx->submit_state.link.head) { 4867ed29b0b4SJens Axboe ctx->drain_next = false; 4868ed29b0b4SJens Axboe ctx->drain_active = true; 4869ed29b0b4SJens Axboe req->flags |= REQ_F_IO_DRAIN | REQ_F_FORCE_ASYNC; 4870ed29b0b4SJens Axboe } 4871ed29b0b4SJens Axboe } 4872ed29b0b4SJens Axboe 4873ed29b0b4SJens Axboe if (!def->ioprio && sqe->ioprio) 4874ed29b0b4SJens Axboe return -EINVAL; 4875ed29b0b4SJens Axboe if (!def->iopoll && (ctx->flags & IORING_SETUP_IOPOLL)) 4876ed29b0b4SJens Axboe return -EINVAL; 4877ed29b0b4SJens Axboe 4878ed29b0b4SJens Axboe if (def->needs_file) { 4879ed29b0b4SJens Axboe struct io_submit_state *state = &ctx->submit_state; 4880ed29b0b4SJens Axboe 4881ed29b0b4SJens Axboe req->cqe.fd = READ_ONCE(sqe->fd); 4882ed29b0b4SJens Axboe 4883ed29b0b4SJens Axboe /* 4884ed29b0b4SJens Axboe * Plug now if we have more than 2 IO left after this, and the 4885ed29b0b4SJens Axboe * target is potentially a read/write to block based storage. 4886ed29b0b4SJens Axboe */ 4887ed29b0b4SJens Axboe if (state->need_plug && def->plug) { 4888ed29b0b4SJens Axboe state->plug_started = true; 4889ed29b0b4SJens Axboe state->need_plug = false; 4890ed29b0b4SJens Axboe blk_start_plug_nr_ios(&state->plug, state->submit_nr); 4891ed29b0b4SJens Axboe } 4892ed29b0b4SJens Axboe } 4893ed29b0b4SJens Axboe 4894ed29b0b4SJens Axboe personality = READ_ONCE(sqe->personality); 4895ed29b0b4SJens Axboe if (personality) { 4896ed29b0b4SJens Axboe int ret; 4897ed29b0b4SJens Axboe 4898ed29b0b4SJens Axboe req->creds = xa_load(&ctx->personalities, personality); 4899ed29b0b4SJens Axboe if (!req->creds) 4900ed29b0b4SJens Axboe return -EINVAL; 4901ed29b0b4SJens Axboe get_cred(req->creds); 4902ed29b0b4SJens Axboe ret = security_uring_override_creds(req->creds); 4903ed29b0b4SJens Axboe if (ret) { 4904ed29b0b4SJens Axboe put_cred(req->creds); 4905ed29b0b4SJens Axboe return ret; 4906ed29b0b4SJens Axboe } 4907ed29b0b4SJens Axboe req->flags |= REQ_F_CREDS; 4908ed29b0b4SJens Axboe } 4909ed29b0b4SJens Axboe 4910ed29b0b4SJens Axboe return def->prep(req, sqe); 4911ed29b0b4SJens Axboe } 4912ed29b0b4SJens Axboe 4913ed29b0b4SJens Axboe static __cold int io_submit_fail_init(const struct io_uring_sqe *sqe, 4914ed29b0b4SJens Axboe struct io_kiocb *req, int ret) 4915ed29b0b4SJens Axboe { 4916ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 4917ed29b0b4SJens Axboe struct io_submit_link *link = &ctx->submit_state.link; 4918ed29b0b4SJens Axboe struct io_kiocb *head = link->head; 4919ed29b0b4SJens Axboe 4920ed29b0b4SJens Axboe trace_io_uring_req_failed(sqe, ctx, req, ret); 4921ed29b0b4SJens Axboe 4922ed29b0b4SJens Axboe /* 4923ed29b0b4SJens Axboe * Avoid breaking links in the middle as it renders links with SQPOLL 4924ed29b0b4SJens Axboe * unusable. Instead of failing eagerly, continue assembling the link if 4925ed29b0b4SJens Axboe * applicable and mark the head with REQ_F_FAIL. The link flushing code 4926ed29b0b4SJens Axboe * should find the flag and handle the rest. 4927ed29b0b4SJens Axboe */ 4928ed29b0b4SJens Axboe req_fail_link_node(req, ret); 4929ed29b0b4SJens Axboe if (head && !(head->flags & REQ_F_FAIL)) 4930ed29b0b4SJens Axboe req_fail_link_node(head, -ECANCELED); 4931ed29b0b4SJens Axboe 4932ed29b0b4SJens Axboe if (!(req->flags & IO_REQ_LINK_FLAGS)) { 4933ed29b0b4SJens Axboe if (head) { 4934ed29b0b4SJens Axboe link->last->link = req; 4935ed29b0b4SJens Axboe link->head = NULL; 4936ed29b0b4SJens Axboe req = head; 4937ed29b0b4SJens Axboe } 4938ed29b0b4SJens Axboe io_queue_sqe_fallback(req); 4939ed29b0b4SJens Axboe return ret; 4940ed29b0b4SJens Axboe } 4941ed29b0b4SJens Axboe 4942ed29b0b4SJens Axboe if (head) 4943ed29b0b4SJens Axboe link->last->link = req; 4944ed29b0b4SJens Axboe else 4945ed29b0b4SJens Axboe link->head = req; 4946ed29b0b4SJens Axboe link->last = req; 4947ed29b0b4SJens Axboe return 0; 4948ed29b0b4SJens Axboe } 4949ed29b0b4SJens Axboe 4950ed29b0b4SJens Axboe static inline int io_submit_sqe(struct io_ring_ctx *ctx, struct io_kiocb *req, 4951ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4952ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 4953ed29b0b4SJens Axboe { 4954ed29b0b4SJens Axboe struct io_submit_link *link = &ctx->submit_state.link; 4955ed29b0b4SJens Axboe int ret; 4956ed29b0b4SJens Axboe 4957ed29b0b4SJens Axboe ret = io_init_req(ctx, req, sqe); 4958ed29b0b4SJens Axboe if (unlikely(ret)) 4959ed29b0b4SJens Axboe return io_submit_fail_init(sqe, req, ret); 4960ed29b0b4SJens Axboe 4961ed29b0b4SJens Axboe /* don't need @sqe from now on */ 4962ed29b0b4SJens Axboe trace_io_uring_submit_sqe(ctx, req, req->cqe.user_data, req->opcode, 4963ed29b0b4SJens Axboe req->flags, true, 4964ed29b0b4SJens Axboe ctx->flags & IORING_SETUP_SQPOLL); 4965ed29b0b4SJens Axboe 4966ed29b0b4SJens Axboe /* 4967ed29b0b4SJens Axboe * If we already have a head request, queue this one for async 4968ed29b0b4SJens Axboe * submittal once the head completes. If we don't have a head but 4969ed29b0b4SJens Axboe * IOSQE_IO_LINK is set in the sqe, start a new head. This one will be 4970ed29b0b4SJens Axboe * submitted sync once the chain is complete. If none of those 4971ed29b0b4SJens Axboe * conditions are true (normal request), then just queue it. 4972ed29b0b4SJens Axboe */ 4973ed29b0b4SJens Axboe if (unlikely(link->head)) { 4974ed29b0b4SJens Axboe ret = io_req_prep_async(req); 4975ed29b0b4SJens Axboe if (unlikely(ret)) 4976ed29b0b4SJens Axboe return io_submit_fail_init(sqe, req, ret); 4977ed29b0b4SJens Axboe 4978ed29b0b4SJens Axboe trace_io_uring_link(ctx, req, link->head); 4979ed29b0b4SJens Axboe link->last->link = req; 4980ed29b0b4SJens Axboe link->last = req; 4981ed29b0b4SJens Axboe 4982ed29b0b4SJens Axboe if (req->flags & IO_REQ_LINK_FLAGS) 4983ed29b0b4SJens Axboe return 0; 4984ed29b0b4SJens Axboe /* last request of the link, flush it */ 4985ed29b0b4SJens Axboe req = link->head; 4986ed29b0b4SJens Axboe link->head = NULL; 4987ed29b0b4SJens Axboe if (req->flags & (REQ_F_FORCE_ASYNC | REQ_F_FAIL)) 4988ed29b0b4SJens Axboe goto fallback; 4989ed29b0b4SJens Axboe 4990ed29b0b4SJens Axboe } else if (unlikely(req->flags & (IO_REQ_LINK_FLAGS | 4991ed29b0b4SJens Axboe REQ_F_FORCE_ASYNC | REQ_F_FAIL))) { 4992ed29b0b4SJens Axboe if (req->flags & IO_REQ_LINK_FLAGS) { 4993ed29b0b4SJens Axboe link->head = req; 4994ed29b0b4SJens Axboe link->last = req; 4995ed29b0b4SJens Axboe } else { 4996ed29b0b4SJens Axboe fallback: 4997ed29b0b4SJens Axboe io_queue_sqe_fallback(req); 4998ed29b0b4SJens Axboe } 4999ed29b0b4SJens Axboe return 0; 5000ed29b0b4SJens Axboe } 5001ed29b0b4SJens Axboe 5002ed29b0b4SJens Axboe io_queue_sqe(req); 5003ed29b0b4SJens Axboe return 0; 5004ed29b0b4SJens Axboe } 5005ed29b0b4SJens Axboe 5006ed29b0b4SJens Axboe /* 5007ed29b0b4SJens Axboe * Batched submission is done, ensure local IO is flushed out. 5008ed29b0b4SJens Axboe */ 5009ed29b0b4SJens Axboe static void io_submit_state_end(struct io_ring_ctx *ctx) 5010ed29b0b4SJens Axboe { 5011ed29b0b4SJens Axboe struct io_submit_state *state = &ctx->submit_state; 5012ed29b0b4SJens Axboe 5013ed29b0b4SJens Axboe if (unlikely(state->link.head)) 5014ed29b0b4SJens Axboe io_queue_sqe_fallback(state->link.head); 5015ed29b0b4SJens Axboe /* flush only after queuing links as they can generate completions */ 5016ed29b0b4SJens Axboe io_submit_flush_completions(ctx); 5017ed29b0b4SJens Axboe if (state->plug_started) 5018ed29b0b4SJens Axboe blk_finish_plug(&state->plug); 5019ed29b0b4SJens Axboe } 5020ed29b0b4SJens Axboe 5021ed29b0b4SJens Axboe /* 5022ed29b0b4SJens Axboe * Start submission side cache. 5023ed29b0b4SJens Axboe */ 5024ed29b0b4SJens Axboe static void io_submit_state_start(struct io_submit_state *state, 5025ed29b0b4SJens Axboe unsigned int max_ios) 5026ed29b0b4SJens Axboe { 5027ed29b0b4SJens Axboe state->plug_started = false; 5028ed29b0b4SJens Axboe state->need_plug = max_ios > 2; 5029ed29b0b4SJens Axboe state->submit_nr = max_ios; 5030ed29b0b4SJens Axboe /* set only head, no need to init link_last in advance */ 5031ed29b0b4SJens Axboe state->link.head = NULL; 5032ed29b0b4SJens Axboe } 5033ed29b0b4SJens Axboe 5034ed29b0b4SJens Axboe static void io_commit_sqring(struct io_ring_ctx *ctx) 5035ed29b0b4SJens Axboe { 5036ed29b0b4SJens Axboe struct io_rings *rings = ctx->rings; 5037ed29b0b4SJens Axboe 5038ed29b0b4SJens Axboe /* 5039ed29b0b4SJens Axboe * Ensure any loads from the SQEs are done at this point, 5040ed29b0b4SJens Axboe * since once we write the new head, the application could 5041ed29b0b4SJens Axboe * write new data to them. 5042ed29b0b4SJens Axboe */ 5043ed29b0b4SJens Axboe smp_store_release(&rings->sq.head, ctx->cached_sq_head); 5044ed29b0b4SJens Axboe } 5045ed29b0b4SJens Axboe 5046ed29b0b4SJens Axboe /* 5047ed29b0b4SJens Axboe * Fetch an sqe, if one is available. Note this returns a pointer to memory 5048ed29b0b4SJens Axboe * that is mapped by userspace. This means that care needs to be taken to 5049ed29b0b4SJens Axboe * ensure that reads are stable, as we cannot rely on userspace always 5050ed29b0b4SJens Axboe * being a good citizen. If members of the sqe are validated and then later 5051ed29b0b4SJens Axboe * used, it's important that those reads are done through READ_ONCE() to 5052ed29b0b4SJens Axboe * prevent a re-load down the line. 5053ed29b0b4SJens Axboe */ 5054ed29b0b4SJens Axboe static const struct io_uring_sqe *io_get_sqe(struct io_ring_ctx *ctx) 5055ed29b0b4SJens Axboe { 5056ed29b0b4SJens Axboe unsigned head, mask = ctx->sq_entries - 1; 5057ed29b0b4SJens Axboe unsigned sq_idx = ctx->cached_sq_head++ & mask; 5058ed29b0b4SJens Axboe 5059ed29b0b4SJens Axboe /* 5060ed29b0b4SJens Axboe * The cached sq head (or cq tail) serves two purposes: 5061ed29b0b4SJens Axboe * 5062ed29b0b4SJens Axboe * 1) allows us to batch the cost of updating the user visible 5063ed29b0b4SJens Axboe * head updates. 5064ed29b0b4SJens Axboe * 2) allows the kernel side to track the head on its own, even 5065ed29b0b4SJens Axboe * though the application is the one updating it. 5066ed29b0b4SJens Axboe */ 5067ed29b0b4SJens Axboe head = READ_ONCE(ctx->sq_array[sq_idx]); 5068ed29b0b4SJens Axboe if (likely(head < ctx->sq_entries)) { 5069ed29b0b4SJens Axboe /* double index for 128-byte SQEs, twice as long */ 5070ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQE128) 5071ed29b0b4SJens Axboe head <<= 1; 5072ed29b0b4SJens Axboe return &ctx->sq_sqes[head]; 5073ed29b0b4SJens Axboe } 5074ed29b0b4SJens Axboe 5075ed29b0b4SJens Axboe /* drop invalid entries */ 5076ed29b0b4SJens Axboe ctx->cq_extra--; 5077ed29b0b4SJens Axboe WRITE_ONCE(ctx->rings->sq_dropped, 5078ed29b0b4SJens Axboe READ_ONCE(ctx->rings->sq_dropped) + 1); 5079ed29b0b4SJens Axboe return NULL; 5080ed29b0b4SJens Axboe } 5081ed29b0b4SJens Axboe 508217437f31SJens Axboe int io_submit_sqes(struct io_ring_ctx *ctx, unsigned int nr) 5083ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 5084ed29b0b4SJens Axboe { 5085ed29b0b4SJens Axboe unsigned int entries = io_sqring_entries(ctx); 5086ed29b0b4SJens Axboe unsigned int left; 5087ed29b0b4SJens Axboe int ret; 5088ed29b0b4SJens Axboe 5089ed29b0b4SJens Axboe if (unlikely(!entries)) 5090ed29b0b4SJens Axboe return 0; 5091ed29b0b4SJens Axboe /* make sure SQ entry isn't read before tail */ 5092ed29b0b4SJens Axboe ret = left = min3(nr, ctx->sq_entries, entries); 5093ed29b0b4SJens Axboe io_get_task_refs(left); 5094ed29b0b4SJens Axboe io_submit_state_start(&ctx->submit_state, left); 5095ed29b0b4SJens Axboe 5096ed29b0b4SJens Axboe do { 5097ed29b0b4SJens Axboe const struct io_uring_sqe *sqe; 5098ed29b0b4SJens Axboe struct io_kiocb *req; 5099ed29b0b4SJens Axboe 5100ed29b0b4SJens Axboe if (unlikely(!io_alloc_req_refill(ctx))) 5101ed29b0b4SJens Axboe break; 5102ed29b0b4SJens Axboe req = io_alloc_req(ctx); 5103ed29b0b4SJens Axboe sqe = io_get_sqe(ctx); 5104ed29b0b4SJens Axboe if (unlikely(!sqe)) { 5105ed29b0b4SJens Axboe io_req_add_to_cache(req, ctx); 5106ed29b0b4SJens Axboe break; 5107ed29b0b4SJens Axboe } 5108ed29b0b4SJens Axboe 5109ed29b0b4SJens Axboe /* 5110ed29b0b4SJens Axboe * Continue submitting even for sqe failure if the 5111ed29b0b4SJens Axboe * ring was setup with IORING_SETUP_SUBMIT_ALL 5112ed29b0b4SJens Axboe */ 5113ed29b0b4SJens Axboe if (unlikely(io_submit_sqe(ctx, req, sqe)) && 5114ed29b0b4SJens Axboe !(ctx->flags & IORING_SETUP_SUBMIT_ALL)) { 5115ed29b0b4SJens Axboe left--; 5116ed29b0b4SJens Axboe break; 5117ed29b0b4SJens Axboe } 5118ed29b0b4SJens Axboe } while (--left); 5119ed29b0b4SJens Axboe 5120ed29b0b4SJens Axboe if (unlikely(left)) { 5121ed29b0b4SJens Axboe ret -= left; 5122ed29b0b4SJens Axboe /* try again if it submitted nothing and can't allocate a req */ 5123ed29b0b4SJens Axboe if (!ret && io_req_cache_empty(ctx)) 5124ed29b0b4SJens Axboe ret = -EAGAIN; 5125ed29b0b4SJens Axboe current->io_uring->cached_refs += left; 5126ed29b0b4SJens Axboe } 5127ed29b0b4SJens Axboe 5128ed29b0b4SJens Axboe io_submit_state_end(ctx); 5129ed29b0b4SJens Axboe /* Commit SQ ring head once we've consumed and submitted all SQEs */ 5130ed29b0b4SJens Axboe io_commit_sqring(ctx); 5131ed29b0b4SJens Axboe return ret; 5132ed29b0b4SJens Axboe } 5133ed29b0b4SJens Axboe 5134ed29b0b4SJens Axboe struct io_wait_queue { 5135ed29b0b4SJens Axboe struct wait_queue_entry wq; 5136ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 5137ed29b0b4SJens Axboe unsigned cq_tail; 5138ed29b0b4SJens Axboe unsigned nr_timeouts; 5139ed29b0b4SJens Axboe }; 5140ed29b0b4SJens Axboe 5141ed29b0b4SJens Axboe static inline bool io_should_wake(struct io_wait_queue *iowq) 5142ed29b0b4SJens Axboe { 5143ed29b0b4SJens Axboe struct io_ring_ctx *ctx = iowq->ctx; 5144ed29b0b4SJens Axboe int dist = ctx->cached_cq_tail - (int) iowq->cq_tail; 5145ed29b0b4SJens Axboe 5146ed29b0b4SJens Axboe /* 5147ed29b0b4SJens Axboe * Wake up if we have enough events, or if a timeout occurred since we 5148ed29b0b4SJens Axboe * started waiting. For timeouts, we always want to return to userspace, 5149ed29b0b4SJens Axboe * regardless of event count. 5150ed29b0b4SJens Axboe */ 5151ed29b0b4SJens Axboe return dist >= 0 || atomic_read(&ctx->cq_timeouts) != iowq->nr_timeouts; 5152ed29b0b4SJens Axboe } 5153ed29b0b4SJens Axboe 5154ed29b0b4SJens Axboe static int io_wake_function(struct wait_queue_entry *curr, unsigned int mode, 5155ed29b0b4SJens Axboe int wake_flags, void *key) 5156ed29b0b4SJens Axboe { 5157ed29b0b4SJens Axboe struct io_wait_queue *iowq = container_of(curr, struct io_wait_queue, 5158ed29b0b4SJens Axboe wq); 5159ed29b0b4SJens Axboe 5160ed29b0b4SJens Axboe /* 5161ed29b0b4SJens Axboe * Cannot safely flush overflowed CQEs from here, ensure we wake up 5162ed29b0b4SJens Axboe * the task, and the next invocation will do it. 5163ed29b0b4SJens Axboe */ 5164ed29b0b4SJens Axboe if (io_should_wake(iowq) || 5165ed29b0b4SJens Axboe test_bit(IO_CHECK_CQ_OVERFLOW_BIT, &iowq->ctx->check_cq)) 5166ed29b0b4SJens Axboe return autoremove_wake_function(curr, mode, wake_flags, key); 5167ed29b0b4SJens Axboe return -1; 5168ed29b0b4SJens Axboe } 5169ed29b0b4SJens Axboe 5170ed29b0b4SJens Axboe static int io_run_task_work_sig(void) 5171ed29b0b4SJens Axboe { 5172ed29b0b4SJens Axboe if (io_run_task_work()) 5173ed29b0b4SJens Axboe return 1; 5174ed29b0b4SJens Axboe if (test_thread_flag(TIF_NOTIFY_SIGNAL)) 5175ed29b0b4SJens Axboe return -ERESTARTSYS; 5176ed29b0b4SJens Axboe if (task_sigpending(current)) 5177ed29b0b4SJens Axboe return -EINTR; 5178ed29b0b4SJens Axboe return 0; 5179ed29b0b4SJens Axboe } 5180ed29b0b4SJens Axboe 5181ed29b0b4SJens Axboe /* when returns >0, the caller should retry */ 5182ed29b0b4SJens Axboe static inline int io_cqring_wait_schedule(struct io_ring_ctx *ctx, 5183ed29b0b4SJens Axboe struct io_wait_queue *iowq, 5184ed29b0b4SJens Axboe ktime_t timeout) 5185ed29b0b4SJens Axboe { 5186ed29b0b4SJens Axboe int ret; 5187ed29b0b4SJens Axboe unsigned long check_cq; 5188ed29b0b4SJens Axboe 5189ed29b0b4SJens Axboe /* make sure we run task_work before checking for signals */ 5190ed29b0b4SJens Axboe ret = io_run_task_work_sig(); 5191ed29b0b4SJens Axboe if (ret || io_should_wake(iowq)) 5192ed29b0b4SJens Axboe return ret; 5193ed29b0b4SJens Axboe check_cq = READ_ONCE(ctx->check_cq); 5194ed29b0b4SJens Axboe /* let the caller flush overflows, retry */ 5195ed29b0b4SJens Axboe if (check_cq & BIT(IO_CHECK_CQ_OVERFLOW_BIT)) 5196ed29b0b4SJens Axboe return 1; 5197ed29b0b4SJens Axboe if (unlikely(check_cq & BIT(IO_CHECK_CQ_DROPPED_BIT))) 5198ed29b0b4SJens Axboe return -EBADR; 5199ed29b0b4SJens Axboe if (!schedule_hrtimeout(&timeout, HRTIMER_MODE_ABS)) 5200ed29b0b4SJens Axboe return -ETIME; 5201ed29b0b4SJens Axboe return 1; 5202ed29b0b4SJens Axboe } 5203ed29b0b4SJens Axboe 5204ed29b0b4SJens Axboe /* 5205ed29b0b4SJens Axboe * Wait until events become available, if we don't already have some. The 5206ed29b0b4SJens Axboe * application must reap them itself, as they reside on the shared cq ring. 5207ed29b0b4SJens Axboe */ 5208ed29b0b4SJens Axboe static int io_cqring_wait(struct io_ring_ctx *ctx, int min_events, 5209ed29b0b4SJens Axboe const sigset_t __user *sig, size_t sigsz, 5210ed29b0b4SJens Axboe struct __kernel_timespec __user *uts) 5211ed29b0b4SJens Axboe { 5212ed29b0b4SJens Axboe struct io_wait_queue iowq; 5213ed29b0b4SJens Axboe struct io_rings *rings = ctx->rings; 5214ed29b0b4SJens Axboe ktime_t timeout = KTIME_MAX; 5215ed29b0b4SJens Axboe int ret; 5216ed29b0b4SJens Axboe 5217ed29b0b4SJens Axboe do { 5218ed29b0b4SJens Axboe io_cqring_overflow_flush(ctx); 5219ed29b0b4SJens Axboe if (io_cqring_events(ctx) >= min_events) 5220ed29b0b4SJens Axboe return 0; 5221ed29b0b4SJens Axboe if (!io_run_task_work()) 5222ed29b0b4SJens Axboe break; 5223ed29b0b4SJens Axboe } while (1); 5224ed29b0b4SJens Axboe 5225ed29b0b4SJens Axboe if (sig) { 5226ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 5227ed29b0b4SJens Axboe if (in_compat_syscall()) 5228ed29b0b4SJens Axboe ret = set_compat_user_sigmask((const compat_sigset_t __user *)sig, 5229ed29b0b4SJens Axboe sigsz); 5230ed29b0b4SJens Axboe else 5231ed29b0b4SJens Axboe #endif 5232ed29b0b4SJens Axboe ret = set_user_sigmask(sig, sigsz); 5233ed29b0b4SJens Axboe 5234ed29b0b4SJens Axboe if (ret) 5235ed29b0b4SJens Axboe return ret; 5236ed29b0b4SJens Axboe } 5237ed29b0b4SJens Axboe 5238ed29b0b4SJens Axboe if (uts) { 5239ed29b0b4SJens Axboe struct timespec64 ts; 5240ed29b0b4SJens Axboe 5241ed29b0b4SJens Axboe if (get_timespec64(&ts, uts)) 5242ed29b0b4SJens Axboe return -EFAULT; 5243ed29b0b4SJens Axboe timeout = ktime_add_ns(timespec64_to_ktime(ts), ktime_get_ns()); 5244ed29b0b4SJens Axboe } 5245ed29b0b4SJens Axboe 5246ed29b0b4SJens Axboe init_waitqueue_func_entry(&iowq.wq, io_wake_function); 5247ed29b0b4SJens Axboe iowq.wq.private = current; 5248ed29b0b4SJens Axboe INIT_LIST_HEAD(&iowq.wq.entry); 5249ed29b0b4SJens Axboe iowq.ctx = ctx; 5250ed29b0b4SJens Axboe iowq.nr_timeouts = atomic_read(&ctx->cq_timeouts); 5251ed29b0b4SJens Axboe iowq.cq_tail = READ_ONCE(ctx->rings->cq.head) + min_events; 5252ed29b0b4SJens Axboe 5253ed29b0b4SJens Axboe trace_io_uring_cqring_wait(ctx, min_events); 5254ed29b0b4SJens Axboe do { 5255ed29b0b4SJens Axboe /* if we can't even flush overflow, don't wait for more */ 5256ed29b0b4SJens Axboe if (!io_cqring_overflow_flush(ctx)) { 5257ed29b0b4SJens Axboe ret = -EBUSY; 5258ed29b0b4SJens Axboe break; 5259ed29b0b4SJens Axboe } 5260ed29b0b4SJens Axboe prepare_to_wait_exclusive(&ctx->cq_wait, &iowq.wq, 5261ed29b0b4SJens Axboe TASK_INTERRUPTIBLE); 5262ed29b0b4SJens Axboe ret = io_cqring_wait_schedule(ctx, &iowq, timeout); 5263ed29b0b4SJens Axboe cond_resched(); 5264ed29b0b4SJens Axboe } while (ret > 0); 5265ed29b0b4SJens Axboe 5266ed29b0b4SJens Axboe finish_wait(&ctx->cq_wait, &iowq.wq); 5267ed29b0b4SJens Axboe restore_saved_sigmask_unless(ret == -EINTR); 5268ed29b0b4SJens Axboe 5269ed29b0b4SJens Axboe return READ_ONCE(rings->cq.head) == READ_ONCE(rings->cq.tail) ? ret : 0; 5270ed29b0b4SJens Axboe } 5271ed29b0b4SJens Axboe 5272ed29b0b4SJens Axboe static void io_free_page_table(void **table, size_t size) 5273ed29b0b4SJens Axboe { 5274ed29b0b4SJens Axboe unsigned i, nr_tables = DIV_ROUND_UP(size, PAGE_SIZE); 5275ed29b0b4SJens Axboe 5276ed29b0b4SJens Axboe for (i = 0; i < nr_tables; i++) 5277ed29b0b4SJens Axboe kfree(table[i]); 5278ed29b0b4SJens Axboe kfree(table); 5279ed29b0b4SJens Axboe } 5280ed29b0b4SJens Axboe 5281ed29b0b4SJens Axboe static __cold void **io_alloc_page_table(size_t size) 5282ed29b0b4SJens Axboe { 5283ed29b0b4SJens Axboe unsigned i, nr_tables = DIV_ROUND_UP(size, PAGE_SIZE); 5284ed29b0b4SJens Axboe size_t init_size = size; 5285ed29b0b4SJens Axboe void **table; 5286ed29b0b4SJens Axboe 5287ed29b0b4SJens Axboe table = kcalloc(nr_tables, sizeof(*table), GFP_KERNEL_ACCOUNT); 5288ed29b0b4SJens Axboe if (!table) 5289ed29b0b4SJens Axboe return NULL; 5290ed29b0b4SJens Axboe 5291ed29b0b4SJens Axboe for (i = 0; i < nr_tables; i++) { 5292ed29b0b4SJens Axboe unsigned int this_size = min_t(size_t, size, PAGE_SIZE); 5293ed29b0b4SJens Axboe 5294ed29b0b4SJens Axboe table[i] = kzalloc(this_size, GFP_KERNEL_ACCOUNT); 5295ed29b0b4SJens Axboe if (!table[i]) { 5296ed29b0b4SJens Axboe io_free_page_table(table, init_size); 5297ed29b0b4SJens Axboe return NULL; 5298ed29b0b4SJens Axboe } 5299ed29b0b4SJens Axboe size -= this_size; 5300ed29b0b4SJens Axboe } 5301ed29b0b4SJens Axboe return table; 5302ed29b0b4SJens Axboe } 5303ed29b0b4SJens Axboe 5304ed29b0b4SJens Axboe static void io_rsrc_node_destroy(struct io_rsrc_node *ref_node) 5305ed29b0b4SJens Axboe { 5306ed29b0b4SJens Axboe percpu_ref_exit(&ref_node->refs); 5307ed29b0b4SJens Axboe kfree(ref_node); 5308ed29b0b4SJens Axboe } 5309ed29b0b4SJens Axboe 5310ed29b0b4SJens Axboe static __cold void io_rsrc_node_ref_zero(struct percpu_ref *ref) 5311ed29b0b4SJens Axboe { 5312ed29b0b4SJens Axboe struct io_rsrc_node *node = container_of(ref, struct io_rsrc_node, refs); 5313ed29b0b4SJens Axboe struct io_ring_ctx *ctx = node->rsrc_data->ctx; 5314ed29b0b4SJens Axboe unsigned long flags; 5315ed29b0b4SJens Axboe bool first_add = false; 5316ed29b0b4SJens Axboe unsigned long delay = HZ; 5317ed29b0b4SJens Axboe 5318ed29b0b4SJens Axboe spin_lock_irqsave(&ctx->rsrc_ref_lock, flags); 5319ed29b0b4SJens Axboe node->done = true; 5320ed29b0b4SJens Axboe 5321ed29b0b4SJens Axboe /* if we are mid-quiesce then do not delay */ 5322ed29b0b4SJens Axboe if (node->rsrc_data->quiesce) 5323ed29b0b4SJens Axboe delay = 0; 5324ed29b0b4SJens Axboe 5325ed29b0b4SJens Axboe while (!list_empty(&ctx->rsrc_ref_list)) { 5326ed29b0b4SJens Axboe node = list_first_entry(&ctx->rsrc_ref_list, 5327ed29b0b4SJens Axboe struct io_rsrc_node, node); 5328ed29b0b4SJens Axboe /* recycle ref nodes in order */ 5329ed29b0b4SJens Axboe if (!node->done) 5330ed29b0b4SJens Axboe break; 5331ed29b0b4SJens Axboe list_del(&node->node); 5332ed29b0b4SJens Axboe first_add |= llist_add(&node->llist, &ctx->rsrc_put_llist); 5333ed29b0b4SJens Axboe } 5334ed29b0b4SJens Axboe spin_unlock_irqrestore(&ctx->rsrc_ref_lock, flags); 5335ed29b0b4SJens Axboe 5336ed29b0b4SJens Axboe if (first_add) 5337ed29b0b4SJens Axboe mod_delayed_work(system_wq, &ctx->rsrc_put_work, delay); 5338ed29b0b4SJens Axboe } 5339ed29b0b4SJens Axboe 5340ed29b0b4SJens Axboe static struct io_rsrc_node *io_rsrc_node_alloc(void) 5341ed29b0b4SJens Axboe { 5342ed29b0b4SJens Axboe struct io_rsrc_node *ref_node; 5343ed29b0b4SJens Axboe 5344ed29b0b4SJens Axboe ref_node = kzalloc(sizeof(*ref_node), GFP_KERNEL); 5345ed29b0b4SJens Axboe if (!ref_node) 5346ed29b0b4SJens Axboe return NULL; 5347ed29b0b4SJens Axboe 5348ed29b0b4SJens Axboe if (percpu_ref_init(&ref_node->refs, io_rsrc_node_ref_zero, 5349ed29b0b4SJens Axboe 0, GFP_KERNEL)) { 5350ed29b0b4SJens Axboe kfree(ref_node); 5351ed29b0b4SJens Axboe return NULL; 5352ed29b0b4SJens Axboe } 5353ed29b0b4SJens Axboe INIT_LIST_HEAD(&ref_node->node); 5354ed29b0b4SJens Axboe INIT_LIST_HEAD(&ref_node->rsrc_list); 5355ed29b0b4SJens Axboe ref_node->done = false; 5356ed29b0b4SJens Axboe return ref_node; 5357ed29b0b4SJens Axboe } 5358ed29b0b4SJens Axboe 5359cd40cae2SJens Axboe void io_rsrc_node_switch(struct io_ring_ctx *ctx, 5360ed29b0b4SJens Axboe struct io_rsrc_data *data_to_kill) 5361ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 5362ed29b0b4SJens Axboe { 5363ed29b0b4SJens Axboe WARN_ON_ONCE(!ctx->rsrc_backup_node); 5364ed29b0b4SJens Axboe WARN_ON_ONCE(data_to_kill && !ctx->rsrc_node); 5365ed29b0b4SJens Axboe 5366ed29b0b4SJens Axboe io_rsrc_refs_drop(ctx); 5367ed29b0b4SJens Axboe 5368ed29b0b4SJens Axboe if (data_to_kill) { 5369ed29b0b4SJens Axboe struct io_rsrc_node *rsrc_node = ctx->rsrc_node; 5370ed29b0b4SJens Axboe 5371ed29b0b4SJens Axboe rsrc_node->rsrc_data = data_to_kill; 5372ed29b0b4SJens Axboe spin_lock_irq(&ctx->rsrc_ref_lock); 5373ed29b0b4SJens Axboe list_add_tail(&rsrc_node->node, &ctx->rsrc_ref_list); 5374ed29b0b4SJens Axboe spin_unlock_irq(&ctx->rsrc_ref_lock); 5375ed29b0b4SJens Axboe 5376ed29b0b4SJens Axboe atomic_inc(&data_to_kill->refs); 5377ed29b0b4SJens Axboe percpu_ref_kill(&rsrc_node->refs); 5378ed29b0b4SJens Axboe ctx->rsrc_node = NULL; 5379ed29b0b4SJens Axboe } 5380ed29b0b4SJens Axboe 5381ed29b0b4SJens Axboe if (!ctx->rsrc_node) { 5382ed29b0b4SJens Axboe ctx->rsrc_node = ctx->rsrc_backup_node; 5383ed29b0b4SJens Axboe ctx->rsrc_backup_node = NULL; 5384ed29b0b4SJens Axboe } 5385ed29b0b4SJens Axboe } 5386ed29b0b4SJens Axboe 5387cd40cae2SJens Axboe int io_rsrc_node_switch_start(struct io_ring_ctx *ctx) 5388ed29b0b4SJens Axboe { 5389ed29b0b4SJens Axboe if (ctx->rsrc_backup_node) 5390ed29b0b4SJens Axboe return 0; 5391ed29b0b4SJens Axboe ctx->rsrc_backup_node = io_rsrc_node_alloc(); 5392ed29b0b4SJens Axboe return ctx->rsrc_backup_node ? 0 : -ENOMEM; 5393ed29b0b4SJens Axboe } 5394ed29b0b4SJens Axboe 5395ed29b0b4SJens Axboe static __cold int io_rsrc_ref_quiesce(struct io_rsrc_data *data, 5396ed29b0b4SJens Axboe struct io_ring_ctx *ctx) 5397ed29b0b4SJens Axboe { 5398ed29b0b4SJens Axboe int ret; 5399ed29b0b4SJens Axboe 5400ed29b0b4SJens Axboe /* As we may drop ->uring_lock, other task may have started quiesce */ 5401ed29b0b4SJens Axboe if (data->quiesce) 5402ed29b0b4SJens Axboe return -ENXIO; 5403ed29b0b4SJens Axboe 5404ed29b0b4SJens Axboe data->quiesce = true; 5405ed29b0b4SJens Axboe do { 5406ed29b0b4SJens Axboe ret = io_rsrc_node_switch_start(ctx); 5407ed29b0b4SJens Axboe if (ret) 5408ed29b0b4SJens Axboe break; 5409ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, data); 5410ed29b0b4SJens Axboe 5411ed29b0b4SJens Axboe /* kill initial ref, already quiesced if zero */ 5412ed29b0b4SJens Axboe if (atomic_dec_and_test(&data->refs)) 5413ed29b0b4SJens Axboe break; 5414ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 5415ed29b0b4SJens Axboe flush_delayed_work(&ctx->rsrc_put_work); 5416ed29b0b4SJens Axboe ret = wait_for_completion_interruptible(&data->done); 5417ed29b0b4SJens Axboe if (!ret) { 5418ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 5419ed29b0b4SJens Axboe if (atomic_read(&data->refs) > 0) { 5420ed29b0b4SJens Axboe /* 5421ed29b0b4SJens Axboe * it has been revived by another thread while 5422ed29b0b4SJens Axboe * we were unlocked 5423ed29b0b4SJens Axboe */ 5424ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 5425ed29b0b4SJens Axboe } else { 5426ed29b0b4SJens Axboe break; 5427ed29b0b4SJens Axboe } 5428ed29b0b4SJens Axboe } 5429ed29b0b4SJens Axboe 5430ed29b0b4SJens Axboe atomic_inc(&data->refs); 5431ed29b0b4SJens Axboe /* wait for all works potentially completing data->done */ 5432ed29b0b4SJens Axboe flush_delayed_work(&ctx->rsrc_put_work); 5433ed29b0b4SJens Axboe reinit_completion(&data->done); 5434ed29b0b4SJens Axboe 5435ed29b0b4SJens Axboe ret = io_run_task_work_sig(); 5436ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 5437ed29b0b4SJens Axboe } while (ret >= 0); 5438ed29b0b4SJens Axboe data->quiesce = false; 5439ed29b0b4SJens Axboe 5440ed29b0b4SJens Axboe return ret; 5441ed29b0b4SJens Axboe } 5442ed29b0b4SJens Axboe 5443ed29b0b4SJens Axboe static u64 *io_get_tag_slot(struct io_rsrc_data *data, unsigned int idx) 5444ed29b0b4SJens Axboe { 5445ed29b0b4SJens Axboe unsigned int off = idx & IO_RSRC_TAG_TABLE_MASK; 5446ed29b0b4SJens Axboe unsigned int table_idx = idx >> IO_RSRC_TAG_TABLE_SHIFT; 5447ed29b0b4SJens Axboe 5448ed29b0b4SJens Axboe return &data->tags[table_idx][off]; 5449ed29b0b4SJens Axboe } 5450ed29b0b4SJens Axboe 5451ed29b0b4SJens Axboe static void io_rsrc_data_free(struct io_rsrc_data *data) 5452ed29b0b4SJens Axboe { 5453ed29b0b4SJens Axboe size_t size = data->nr * sizeof(data->tags[0][0]); 5454ed29b0b4SJens Axboe 5455ed29b0b4SJens Axboe if (data->tags) 5456ed29b0b4SJens Axboe io_free_page_table((void **)data->tags, size); 5457ed29b0b4SJens Axboe kfree(data); 5458ed29b0b4SJens Axboe } 5459ed29b0b4SJens Axboe 5460ed29b0b4SJens Axboe static __cold int io_rsrc_data_alloc(struct io_ring_ctx *ctx, rsrc_put_fn *do_put, 5461ed29b0b4SJens Axboe u64 __user *utags, unsigned nr, 5462ed29b0b4SJens Axboe struct io_rsrc_data **pdata) 5463ed29b0b4SJens Axboe { 5464ed29b0b4SJens Axboe struct io_rsrc_data *data; 5465ed29b0b4SJens Axboe int ret = -ENOMEM; 5466ed29b0b4SJens Axboe unsigned i; 5467ed29b0b4SJens Axboe 5468ed29b0b4SJens Axboe data = kzalloc(sizeof(*data), GFP_KERNEL); 5469ed29b0b4SJens Axboe if (!data) 5470ed29b0b4SJens Axboe return -ENOMEM; 5471ed29b0b4SJens Axboe data->tags = (u64 **)io_alloc_page_table(nr * sizeof(data->tags[0][0])); 5472ed29b0b4SJens Axboe if (!data->tags) { 5473ed29b0b4SJens Axboe kfree(data); 5474ed29b0b4SJens Axboe return -ENOMEM; 5475ed29b0b4SJens Axboe } 5476ed29b0b4SJens Axboe 5477ed29b0b4SJens Axboe data->nr = nr; 5478ed29b0b4SJens Axboe data->ctx = ctx; 5479ed29b0b4SJens Axboe data->do_put = do_put; 5480ed29b0b4SJens Axboe if (utags) { 5481ed29b0b4SJens Axboe ret = -EFAULT; 5482ed29b0b4SJens Axboe for (i = 0; i < nr; i++) { 5483ed29b0b4SJens Axboe u64 *tag_slot = io_get_tag_slot(data, i); 5484ed29b0b4SJens Axboe 5485ed29b0b4SJens Axboe if (copy_from_user(tag_slot, &utags[i], 5486ed29b0b4SJens Axboe sizeof(*tag_slot))) 5487ed29b0b4SJens Axboe goto fail; 5488ed29b0b4SJens Axboe } 5489ed29b0b4SJens Axboe } 5490ed29b0b4SJens Axboe 5491ed29b0b4SJens Axboe atomic_set(&data->refs, 1); 5492ed29b0b4SJens Axboe init_completion(&data->done); 5493ed29b0b4SJens Axboe *pdata = data; 5494ed29b0b4SJens Axboe return 0; 5495ed29b0b4SJens Axboe fail: 5496ed29b0b4SJens Axboe io_rsrc_data_free(data); 5497ed29b0b4SJens Axboe return ret; 5498ed29b0b4SJens Axboe } 5499ed29b0b4SJens Axboe 5500ed29b0b4SJens Axboe static void __io_sqe_files_unregister(struct io_ring_ctx *ctx) 5501ed29b0b4SJens Axboe { 5502ed29b0b4SJens Axboe #if !defined(IO_URING_SCM_ALL) 5503ed29b0b4SJens Axboe int i; 5504ed29b0b4SJens Axboe 5505ed29b0b4SJens Axboe for (i = 0; i < ctx->nr_user_files; i++) { 5506a4ad4f74SJens Axboe struct file *file = io_file_from_index(&ctx->file_table, i); 5507ed29b0b4SJens Axboe 5508ed29b0b4SJens Axboe if (!file) 5509ed29b0b4SJens Axboe continue; 5510ed29b0b4SJens Axboe if (io_fixed_file_slot(&ctx->file_table, i)->file_ptr & FFS_SCM) 5511ed29b0b4SJens Axboe continue; 5512ed29b0b4SJens Axboe io_file_bitmap_clear(&ctx->file_table, i); 5513ed29b0b4SJens Axboe fput(file); 5514ed29b0b4SJens Axboe } 5515ed29b0b4SJens Axboe #endif 5516ed29b0b4SJens Axboe 5517ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 5518ed29b0b4SJens Axboe if (ctx->ring_sock) { 5519ed29b0b4SJens Axboe struct sock *sock = ctx->ring_sock->sk; 5520ed29b0b4SJens Axboe struct sk_buff *skb; 5521ed29b0b4SJens Axboe 5522ed29b0b4SJens Axboe while ((skb = skb_dequeue(&sock->sk_receive_queue)) != NULL) 5523ed29b0b4SJens Axboe kfree_skb(skb); 5524ed29b0b4SJens Axboe } 5525ed29b0b4SJens Axboe #endif 5526ed29b0b4SJens Axboe io_free_file_tables(&ctx->file_table); 5527ed29b0b4SJens Axboe io_rsrc_data_free(ctx->file_data); 5528ed29b0b4SJens Axboe ctx->file_data = NULL; 5529ed29b0b4SJens Axboe ctx->nr_user_files = 0; 5530ed29b0b4SJens Axboe } 5531ed29b0b4SJens Axboe 5532ed29b0b4SJens Axboe static int io_sqe_files_unregister(struct io_ring_ctx *ctx) 5533ed29b0b4SJens Axboe { 5534ed29b0b4SJens Axboe unsigned nr = ctx->nr_user_files; 5535ed29b0b4SJens Axboe int ret; 5536ed29b0b4SJens Axboe 5537ed29b0b4SJens Axboe if (!ctx->file_data) 5538ed29b0b4SJens Axboe return -ENXIO; 5539ed29b0b4SJens Axboe 5540ed29b0b4SJens Axboe /* 5541ed29b0b4SJens Axboe * Quiesce may unlock ->uring_lock, and while it's not held 5542ed29b0b4SJens Axboe * prevent new requests using the table. 5543ed29b0b4SJens Axboe */ 5544ed29b0b4SJens Axboe ctx->nr_user_files = 0; 5545ed29b0b4SJens Axboe ret = io_rsrc_ref_quiesce(ctx->file_data, ctx); 5546ed29b0b4SJens Axboe ctx->nr_user_files = nr; 5547ed29b0b4SJens Axboe if (!ret) 5548ed29b0b4SJens Axboe __io_sqe_files_unregister(ctx); 5549ed29b0b4SJens Axboe return ret; 5550ed29b0b4SJens Axboe } 5551ed29b0b4SJens Axboe 5552ed29b0b4SJens Axboe /* 5553ed29b0b4SJens Axboe * Ensure the UNIX gc is aware of our file set, so we are certain that 5554ed29b0b4SJens Axboe * the io_uring can be safely unregistered on process exit, even if we have 5555ed29b0b4SJens Axboe * loops in the file referencing. We account only files that can hold other 5556ed29b0b4SJens Axboe * files because otherwise they can't form a loop and so are not interesting 5557ed29b0b4SJens Axboe * for GC. 5558ed29b0b4SJens Axboe */ 5559ed29b0b4SJens Axboe static int io_scm_file_account(struct io_ring_ctx *ctx, struct file *file) 5560ed29b0b4SJens Axboe { 5561ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 5562ed29b0b4SJens Axboe struct sock *sk = ctx->ring_sock->sk; 5563ed29b0b4SJens Axboe struct sk_buff_head *head = &sk->sk_receive_queue; 5564ed29b0b4SJens Axboe struct scm_fp_list *fpl; 5565ed29b0b4SJens Axboe struct sk_buff *skb; 5566ed29b0b4SJens Axboe 5567ed29b0b4SJens Axboe if (likely(!io_file_need_scm(file))) 5568ed29b0b4SJens Axboe return 0; 5569ed29b0b4SJens Axboe 5570ed29b0b4SJens Axboe /* 5571ed29b0b4SJens Axboe * See if we can merge this file into an existing skb SCM_RIGHTS 5572ed29b0b4SJens Axboe * file set. If there's no room, fall back to allocating a new skb 5573ed29b0b4SJens Axboe * and filling it in. 5574ed29b0b4SJens Axboe */ 5575ed29b0b4SJens Axboe spin_lock_irq(&head->lock); 5576ed29b0b4SJens Axboe skb = skb_peek(head); 5577ed29b0b4SJens Axboe if (skb && UNIXCB(skb).fp->count < SCM_MAX_FD) 5578ed29b0b4SJens Axboe __skb_unlink(skb, head); 5579ed29b0b4SJens Axboe else 5580ed29b0b4SJens Axboe skb = NULL; 5581ed29b0b4SJens Axboe spin_unlock_irq(&head->lock); 5582ed29b0b4SJens Axboe 5583ed29b0b4SJens Axboe if (!skb) { 5584ed29b0b4SJens Axboe fpl = kzalloc(sizeof(*fpl), GFP_KERNEL); 5585ed29b0b4SJens Axboe if (!fpl) 5586ed29b0b4SJens Axboe return -ENOMEM; 5587ed29b0b4SJens Axboe 5588ed29b0b4SJens Axboe skb = alloc_skb(0, GFP_KERNEL); 5589ed29b0b4SJens Axboe if (!skb) { 5590ed29b0b4SJens Axboe kfree(fpl); 5591ed29b0b4SJens Axboe return -ENOMEM; 5592ed29b0b4SJens Axboe } 5593ed29b0b4SJens Axboe 5594ed29b0b4SJens Axboe fpl->user = get_uid(current_user()); 5595ed29b0b4SJens Axboe fpl->max = SCM_MAX_FD; 5596ed29b0b4SJens Axboe fpl->count = 0; 5597ed29b0b4SJens Axboe 5598ed29b0b4SJens Axboe UNIXCB(skb).fp = fpl; 5599ed29b0b4SJens Axboe skb->sk = sk; 5600ed29b0b4SJens Axboe skb->destructor = unix_destruct_scm; 5601ed29b0b4SJens Axboe refcount_add(skb->truesize, &sk->sk_wmem_alloc); 5602ed29b0b4SJens Axboe } 5603ed29b0b4SJens Axboe 5604ed29b0b4SJens Axboe fpl = UNIXCB(skb).fp; 5605ed29b0b4SJens Axboe fpl->fp[fpl->count++] = get_file(file); 5606ed29b0b4SJens Axboe unix_inflight(fpl->user, file); 5607ed29b0b4SJens Axboe skb_queue_head(head, skb); 5608ed29b0b4SJens Axboe fput(file); 5609ed29b0b4SJens Axboe #endif 5610ed29b0b4SJens Axboe return 0; 5611ed29b0b4SJens Axboe } 5612ed29b0b4SJens Axboe 5613ed29b0b4SJens Axboe static void io_rsrc_file_put(struct io_ring_ctx *ctx, struct io_rsrc_put *prsrc) 5614ed29b0b4SJens Axboe { 5615ed29b0b4SJens Axboe struct file *file = prsrc->file; 5616ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 5617ed29b0b4SJens Axboe struct sock *sock = ctx->ring_sock->sk; 5618ed29b0b4SJens Axboe struct sk_buff_head list, *head = &sock->sk_receive_queue; 5619ed29b0b4SJens Axboe struct sk_buff *skb; 5620ed29b0b4SJens Axboe int i; 5621ed29b0b4SJens Axboe 5622ed29b0b4SJens Axboe if (!io_file_need_scm(file)) { 5623ed29b0b4SJens Axboe fput(file); 5624ed29b0b4SJens Axboe return; 5625ed29b0b4SJens Axboe } 5626ed29b0b4SJens Axboe 5627ed29b0b4SJens Axboe __skb_queue_head_init(&list); 5628ed29b0b4SJens Axboe 5629ed29b0b4SJens Axboe /* 5630ed29b0b4SJens Axboe * Find the skb that holds this file in its SCM_RIGHTS. When found, 5631ed29b0b4SJens Axboe * remove this entry and rearrange the file array. 5632ed29b0b4SJens Axboe */ 5633ed29b0b4SJens Axboe skb = skb_dequeue(head); 5634ed29b0b4SJens Axboe while (skb) { 5635ed29b0b4SJens Axboe struct scm_fp_list *fp; 5636ed29b0b4SJens Axboe 5637ed29b0b4SJens Axboe fp = UNIXCB(skb).fp; 5638ed29b0b4SJens Axboe for (i = 0; i < fp->count; i++) { 5639ed29b0b4SJens Axboe int left; 5640ed29b0b4SJens Axboe 5641ed29b0b4SJens Axboe if (fp->fp[i] != file) 5642ed29b0b4SJens Axboe continue; 5643ed29b0b4SJens Axboe 5644ed29b0b4SJens Axboe unix_notinflight(fp->user, fp->fp[i]); 5645ed29b0b4SJens Axboe left = fp->count - 1 - i; 5646ed29b0b4SJens Axboe if (left) { 5647ed29b0b4SJens Axboe memmove(&fp->fp[i], &fp->fp[i + 1], 5648ed29b0b4SJens Axboe left * sizeof(struct file *)); 5649ed29b0b4SJens Axboe } 5650ed29b0b4SJens Axboe fp->count--; 5651ed29b0b4SJens Axboe if (!fp->count) { 5652ed29b0b4SJens Axboe kfree_skb(skb); 5653ed29b0b4SJens Axboe skb = NULL; 5654ed29b0b4SJens Axboe } else { 5655ed29b0b4SJens Axboe __skb_queue_tail(&list, skb); 5656ed29b0b4SJens Axboe } 5657ed29b0b4SJens Axboe fput(file); 5658ed29b0b4SJens Axboe file = NULL; 5659ed29b0b4SJens Axboe break; 5660ed29b0b4SJens Axboe } 5661ed29b0b4SJens Axboe 5662ed29b0b4SJens Axboe if (!file) 5663ed29b0b4SJens Axboe break; 5664ed29b0b4SJens Axboe 5665ed29b0b4SJens Axboe __skb_queue_tail(&list, skb); 5666ed29b0b4SJens Axboe 5667ed29b0b4SJens Axboe skb = skb_dequeue(head); 5668ed29b0b4SJens Axboe } 5669ed29b0b4SJens Axboe 5670ed29b0b4SJens Axboe if (skb_peek(&list)) { 5671ed29b0b4SJens Axboe spin_lock_irq(&head->lock); 5672ed29b0b4SJens Axboe while ((skb = __skb_dequeue(&list)) != NULL) 5673ed29b0b4SJens Axboe __skb_queue_tail(head, skb); 5674ed29b0b4SJens Axboe spin_unlock_irq(&head->lock); 5675ed29b0b4SJens Axboe } 5676ed29b0b4SJens Axboe #else 5677ed29b0b4SJens Axboe fput(file); 5678ed29b0b4SJens Axboe #endif 5679ed29b0b4SJens Axboe } 5680ed29b0b4SJens Axboe 5681ed29b0b4SJens Axboe static void __io_rsrc_put_work(struct io_rsrc_node *ref_node) 5682ed29b0b4SJens Axboe { 5683ed29b0b4SJens Axboe struct io_rsrc_data *rsrc_data = ref_node->rsrc_data; 5684ed29b0b4SJens Axboe struct io_ring_ctx *ctx = rsrc_data->ctx; 5685ed29b0b4SJens Axboe struct io_rsrc_put *prsrc, *tmp; 5686ed29b0b4SJens Axboe 5687ed29b0b4SJens Axboe list_for_each_entry_safe(prsrc, tmp, &ref_node->rsrc_list, list) { 5688ed29b0b4SJens Axboe list_del(&prsrc->list); 5689ed29b0b4SJens Axboe 5690ed29b0b4SJens Axboe if (prsrc->tag) { 5691ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL) 5692ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 5693ed29b0b4SJens Axboe 5694ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 5695ed29b0b4SJens Axboe io_fill_cqe_aux(ctx, prsrc->tag, 0, 0); 5696ed29b0b4SJens Axboe io_commit_cqring(ctx); 5697ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 5698ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 5699ed29b0b4SJens Axboe 5700ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL) 5701ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 5702ed29b0b4SJens Axboe } 5703ed29b0b4SJens Axboe 5704ed29b0b4SJens Axboe rsrc_data->do_put(ctx, prsrc); 5705ed29b0b4SJens Axboe kfree(prsrc); 5706ed29b0b4SJens Axboe } 5707ed29b0b4SJens Axboe 5708ed29b0b4SJens Axboe io_rsrc_node_destroy(ref_node); 5709ed29b0b4SJens Axboe if (atomic_dec_and_test(&rsrc_data->refs)) 5710ed29b0b4SJens Axboe complete(&rsrc_data->done); 5711ed29b0b4SJens Axboe } 5712ed29b0b4SJens Axboe 5713ed29b0b4SJens Axboe static void io_rsrc_put_work(struct work_struct *work) 5714ed29b0b4SJens Axboe { 5715ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 5716ed29b0b4SJens Axboe struct llist_node *node; 5717ed29b0b4SJens Axboe 5718ed29b0b4SJens Axboe ctx = container_of(work, struct io_ring_ctx, rsrc_put_work.work); 5719ed29b0b4SJens Axboe node = llist_del_all(&ctx->rsrc_put_llist); 5720ed29b0b4SJens Axboe 5721ed29b0b4SJens Axboe while (node) { 5722ed29b0b4SJens Axboe struct io_rsrc_node *ref_node; 5723ed29b0b4SJens Axboe struct llist_node *next = node->next; 5724ed29b0b4SJens Axboe 5725ed29b0b4SJens Axboe ref_node = llist_entry(node, struct io_rsrc_node, llist); 5726ed29b0b4SJens Axboe __io_rsrc_put_work(ref_node); 5727ed29b0b4SJens Axboe node = next; 5728ed29b0b4SJens Axboe } 5729ed29b0b4SJens Axboe } 5730ed29b0b4SJens Axboe 5731ed29b0b4SJens Axboe static int io_sqe_files_register(struct io_ring_ctx *ctx, void __user *arg, 5732ed29b0b4SJens Axboe unsigned nr_args, u64 __user *tags) 5733ed29b0b4SJens Axboe { 5734ed29b0b4SJens Axboe __s32 __user *fds = (__s32 __user *) arg; 5735ed29b0b4SJens Axboe struct file *file; 5736ed29b0b4SJens Axboe int fd, ret; 5737ed29b0b4SJens Axboe unsigned i; 5738ed29b0b4SJens Axboe 5739ed29b0b4SJens Axboe if (ctx->file_data) 5740ed29b0b4SJens Axboe return -EBUSY; 5741ed29b0b4SJens Axboe if (!nr_args) 5742ed29b0b4SJens Axboe return -EINVAL; 5743ed29b0b4SJens Axboe if (nr_args > IORING_MAX_FIXED_FILES) 5744ed29b0b4SJens Axboe return -EMFILE; 5745ed29b0b4SJens Axboe if (nr_args > rlimit(RLIMIT_NOFILE)) 5746ed29b0b4SJens Axboe return -EMFILE; 5747ed29b0b4SJens Axboe ret = io_rsrc_node_switch_start(ctx); 5748ed29b0b4SJens Axboe if (ret) 5749ed29b0b4SJens Axboe return ret; 5750ed29b0b4SJens Axboe ret = io_rsrc_data_alloc(ctx, io_rsrc_file_put, tags, nr_args, 5751ed29b0b4SJens Axboe &ctx->file_data); 5752ed29b0b4SJens Axboe if (ret) 5753ed29b0b4SJens Axboe return ret; 5754ed29b0b4SJens Axboe 5755ed29b0b4SJens Axboe if (!io_alloc_file_tables(&ctx->file_table, nr_args)) { 5756ed29b0b4SJens Axboe io_rsrc_data_free(ctx->file_data); 5757ed29b0b4SJens Axboe ctx->file_data = NULL; 5758ed29b0b4SJens Axboe return -ENOMEM; 5759ed29b0b4SJens Axboe } 5760ed29b0b4SJens Axboe 5761ed29b0b4SJens Axboe for (i = 0; i < nr_args; i++, ctx->nr_user_files++) { 5762ed29b0b4SJens Axboe struct io_fixed_file *file_slot; 5763ed29b0b4SJens Axboe 5764ed29b0b4SJens Axboe if (fds && copy_from_user(&fd, &fds[i], sizeof(fd))) { 5765ed29b0b4SJens Axboe ret = -EFAULT; 5766ed29b0b4SJens Axboe goto fail; 5767ed29b0b4SJens Axboe } 5768ed29b0b4SJens Axboe /* allow sparse sets */ 5769ed29b0b4SJens Axboe if (!fds || fd == -1) { 5770ed29b0b4SJens Axboe ret = -EINVAL; 5771ed29b0b4SJens Axboe if (unlikely(*io_get_tag_slot(ctx->file_data, i))) 5772ed29b0b4SJens Axboe goto fail; 5773ed29b0b4SJens Axboe continue; 5774ed29b0b4SJens Axboe } 5775ed29b0b4SJens Axboe 5776ed29b0b4SJens Axboe file = fget(fd); 5777ed29b0b4SJens Axboe ret = -EBADF; 5778ed29b0b4SJens Axboe if (unlikely(!file)) 5779ed29b0b4SJens Axboe goto fail; 5780ed29b0b4SJens Axboe 5781ed29b0b4SJens Axboe /* 5782ed29b0b4SJens Axboe * Don't allow io_uring instances to be registered. If UNIX 5783ed29b0b4SJens Axboe * isn't enabled, then this causes a reference cycle and this 5784ed29b0b4SJens Axboe * instance can never get freed. If UNIX is enabled we'll 5785ed29b0b4SJens Axboe * handle it just fine, but there's still no point in allowing 5786ed29b0b4SJens Axboe * a ring fd as it doesn't support regular read/write anyway. 5787ed29b0b4SJens Axboe */ 5788e5550a14SJens Axboe if (io_is_uring_fops(file)) { 5789ed29b0b4SJens Axboe fput(file); 5790ed29b0b4SJens Axboe goto fail; 5791ed29b0b4SJens Axboe } 5792ed29b0b4SJens Axboe ret = io_scm_file_account(ctx, file); 5793ed29b0b4SJens Axboe if (ret) { 5794ed29b0b4SJens Axboe fput(file); 5795ed29b0b4SJens Axboe goto fail; 5796ed29b0b4SJens Axboe } 5797ed29b0b4SJens Axboe file_slot = io_fixed_file_slot(&ctx->file_table, i); 5798ed29b0b4SJens Axboe io_fixed_file_set(file_slot, file); 5799ed29b0b4SJens Axboe io_file_bitmap_set(&ctx->file_table, i); 5800ed29b0b4SJens Axboe } 5801ed29b0b4SJens Axboe 5802ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, NULL); 5803ed29b0b4SJens Axboe return 0; 5804ed29b0b4SJens Axboe fail: 5805ed29b0b4SJens Axboe __io_sqe_files_unregister(ctx); 5806ed29b0b4SJens Axboe return ret; 5807ed29b0b4SJens Axboe } 5808ed29b0b4SJens Axboe 5809cd40cae2SJens Axboe int io_queue_rsrc_removal(struct io_rsrc_data *data, unsigned idx, 5810ed29b0b4SJens Axboe struct io_rsrc_node *node, void *rsrc) 5811ed29b0b4SJens Axboe { 5812ed29b0b4SJens Axboe u64 *tag_slot = io_get_tag_slot(data, idx); 5813ed29b0b4SJens Axboe struct io_rsrc_put *prsrc; 5814ed29b0b4SJens Axboe 5815ed29b0b4SJens Axboe prsrc = kzalloc(sizeof(*prsrc), GFP_KERNEL); 5816ed29b0b4SJens Axboe if (!prsrc) 5817ed29b0b4SJens Axboe return -ENOMEM; 5818ed29b0b4SJens Axboe 5819ed29b0b4SJens Axboe prsrc->tag = *tag_slot; 5820ed29b0b4SJens Axboe *tag_slot = 0; 5821ed29b0b4SJens Axboe prsrc->rsrc = rsrc; 5822ed29b0b4SJens Axboe list_add(&prsrc->list, &node->rsrc_list); 5823ed29b0b4SJens Axboe return 0; 5824ed29b0b4SJens Axboe } 5825ed29b0b4SJens Axboe 5826f9ead18cSJens Axboe int io_install_fixed_file(struct io_kiocb *req, struct file *file, 5827ed29b0b4SJens Axboe unsigned int issue_flags, u32 slot_index) 5828ed29b0b4SJens Axboe __must_hold(&req->ctx->uring_lock) 5829ed29b0b4SJens Axboe { 5830ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 5831ed29b0b4SJens Axboe bool needs_switch = false; 5832ed29b0b4SJens Axboe struct io_fixed_file *file_slot; 5833ed29b0b4SJens Axboe int ret; 5834ed29b0b4SJens Axboe 5835e5550a14SJens Axboe if (io_is_uring_fops(file)) 5836ed29b0b4SJens Axboe return -EBADF; 5837ed29b0b4SJens Axboe if (!ctx->file_data) 5838ed29b0b4SJens Axboe return -ENXIO; 5839ed29b0b4SJens Axboe if (slot_index >= ctx->nr_user_files) 5840ed29b0b4SJens Axboe return -EINVAL; 5841ed29b0b4SJens Axboe 5842ed29b0b4SJens Axboe slot_index = array_index_nospec(slot_index, ctx->nr_user_files); 5843ed29b0b4SJens Axboe file_slot = io_fixed_file_slot(&ctx->file_table, slot_index); 5844ed29b0b4SJens Axboe 5845ed29b0b4SJens Axboe if (file_slot->file_ptr) { 5846ed29b0b4SJens Axboe struct file *old_file; 5847ed29b0b4SJens Axboe 5848ed29b0b4SJens Axboe ret = io_rsrc_node_switch_start(ctx); 5849ed29b0b4SJens Axboe if (ret) 5850ed29b0b4SJens Axboe goto err; 5851ed29b0b4SJens Axboe 5852ed29b0b4SJens Axboe old_file = (struct file *)(file_slot->file_ptr & FFS_MASK); 5853ed29b0b4SJens Axboe ret = io_queue_rsrc_removal(ctx->file_data, slot_index, 5854ed29b0b4SJens Axboe ctx->rsrc_node, old_file); 5855ed29b0b4SJens Axboe if (ret) 5856ed29b0b4SJens Axboe goto err; 5857ed29b0b4SJens Axboe file_slot->file_ptr = 0; 5858ed29b0b4SJens Axboe io_file_bitmap_clear(&ctx->file_table, slot_index); 5859ed29b0b4SJens Axboe needs_switch = true; 5860ed29b0b4SJens Axboe } 5861ed29b0b4SJens Axboe 5862ed29b0b4SJens Axboe ret = io_scm_file_account(ctx, file); 5863ed29b0b4SJens Axboe if (!ret) { 5864ed29b0b4SJens Axboe *io_get_tag_slot(ctx->file_data, slot_index) = 0; 5865ed29b0b4SJens Axboe io_fixed_file_set(file_slot, file); 5866ed29b0b4SJens Axboe io_file_bitmap_set(&ctx->file_table, slot_index); 5867ed29b0b4SJens Axboe } 5868ed29b0b4SJens Axboe err: 5869ed29b0b4SJens Axboe if (needs_switch) 5870ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, ctx->file_data); 5871ed29b0b4SJens Axboe if (ret) 5872ed29b0b4SJens Axboe fput(file); 5873ed29b0b4SJens Axboe return ret; 5874ed29b0b4SJens Axboe } 5875ed29b0b4SJens Axboe 5876ed29b0b4SJens Axboe static int __io_sqe_files_update(struct io_ring_ctx *ctx, 5877ed29b0b4SJens Axboe struct io_uring_rsrc_update2 *up, 5878ed29b0b4SJens Axboe unsigned nr_args) 5879ed29b0b4SJens Axboe { 5880ed29b0b4SJens Axboe u64 __user *tags = u64_to_user_ptr(up->tags); 5881ed29b0b4SJens Axboe __s32 __user *fds = u64_to_user_ptr(up->data); 5882ed29b0b4SJens Axboe struct io_rsrc_data *data = ctx->file_data; 5883ed29b0b4SJens Axboe struct io_fixed_file *file_slot; 5884ed29b0b4SJens Axboe struct file *file; 5885ed29b0b4SJens Axboe int fd, i, err = 0; 5886ed29b0b4SJens Axboe unsigned int done; 5887ed29b0b4SJens Axboe bool needs_switch = false; 5888ed29b0b4SJens Axboe 5889ed29b0b4SJens Axboe if (!ctx->file_data) 5890ed29b0b4SJens Axboe return -ENXIO; 5891ed29b0b4SJens Axboe if (up->offset + nr_args > ctx->nr_user_files) 5892ed29b0b4SJens Axboe return -EINVAL; 5893ed29b0b4SJens Axboe 5894ed29b0b4SJens Axboe for (done = 0; done < nr_args; done++) { 5895ed29b0b4SJens Axboe u64 tag = 0; 5896ed29b0b4SJens Axboe 5897ed29b0b4SJens Axboe if ((tags && copy_from_user(&tag, &tags[done], sizeof(tag))) || 5898ed29b0b4SJens Axboe copy_from_user(&fd, &fds[done], sizeof(fd))) { 5899ed29b0b4SJens Axboe err = -EFAULT; 5900ed29b0b4SJens Axboe break; 5901ed29b0b4SJens Axboe } 5902ed29b0b4SJens Axboe if ((fd == IORING_REGISTER_FILES_SKIP || fd == -1) && tag) { 5903ed29b0b4SJens Axboe err = -EINVAL; 5904ed29b0b4SJens Axboe break; 5905ed29b0b4SJens Axboe } 5906ed29b0b4SJens Axboe if (fd == IORING_REGISTER_FILES_SKIP) 5907ed29b0b4SJens Axboe continue; 5908ed29b0b4SJens Axboe 5909ed29b0b4SJens Axboe i = array_index_nospec(up->offset + done, ctx->nr_user_files); 5910ed29b0b4SJens Axboe file_slot = io_fixed_file_slot(&ctx->file_table, i); 5911ed29b0b4SJens Axboe 5912ed29b0b4SJens Axboe if (file_slot->file_ptr) { 5913ed29b0b4SJens Axboe file = (struct file *)(file_slot->file_ptr & FFS_MASK); 5914ed29b0b4SJens Axboe err = io_queue_rsrc_removal(data, i, ctx->rsrc_node, file); 5915ed29b0b4SJens Axboe if (err) 5916ed29b0b4SJens Axboe break; 5917ed29b0b4SJens Axboe file_slot->file_ptr = 0; 5918ed29b0b4SJens Axboe io_file_bitmap_clear(&ctx->file_table, i); 5919ed29b0b4SJens Axboe needs_switch = true; 5920ed29b0b4SJens Axboe } 5921ed29b0b4SJens Axboe if (fd != -1) { 5922ed29b0b4SJens Axboe file = fget(fd); 5923ed29b0b4SJens Axboe if (!file) { 5924ed29b0b4SJens Axboe err = -EBADF; 5925ed29b0b4SJens Axboe break; 5926ed29b0b4SJens Axboe } 5927ed29b0b4SJens Axboe /* 5928ed29b0b4SJens Axboe * Don't allow io_uring instances to be registered. If 5929ed29b0b4SJens Axboe * UNIX isn't enabled, then this causes a reference 5930ed29b0b4SJens Axboe * cycle and this instance can never get freed. If UNIX 5931ed29b0b4SJens Axboe * is enabled we'll handle it just fine, but there's 5932ed29b0b4SJens Axboe * still no point in allowing a ring fd as it doesn't 5933ed29b0b4SJens Axboe * support regular read/write anyway. 5934ed29b0b4SJens Axboe */ 5935e5550a14SJens Axboe if (io_is_uring_fops(file)) { 5936ed29b0b4SJens Axboe fput(file); 5937ed29b0b4SJens Axboe err = -EBADF; 5938ed29b0b4SJens Axboe break; 5939ed29b0b4SJens Axboe } 5940ed29b0b4SJens Axboe err = io_scm_file_account(ctx, file); 5941ed29b0b4SJens Axboe if (err) { 5942ed29b0b4SJens Axboe fput(file); 5943ed29b0b4SJens Axboe break; 5944ed29b0b4SJens Axboe } 5945ed29b0b4SJens Axboe *io_get_tag_slot(data, i) = tag; 5946ed29b0b4SJens Axboe io_fixed_file_set(file_slot, file); 5947ed29b0b4SJens Axboe io_file_bitmap_set(&ctx->file_table, i); 5948ed29b0b4SJens Axboe } 5949ed29b0b4SJens Axboe } 5950ed29b0b4SJens Axboe 5951ed29b0b4SJens Axboe if (needs_switch) 5952ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, data); 5953ed29b0b4SJens Axboe return done ? done : err; 5954ed29b0b4SJens Axboe } 5955ed29b0b4SJens Axboe 5956ed29b0b4SJens Axboe static inline void __io_unaccount_mem(struct user_struct *user, 5957ed29b0b4SJens Axboe unsigned long nr_pages) 5958ed29b0b4SJens Axboe { 5959ed29b0b4SJens Axboe atomic_long_sub(nr_pages, &user->locked_vm); 5960ed29b0b4SJens Axboe } 5961ed29b0b4SJens Axboe 5962ed29b0b4SJens Axboe static inline int __io_account_mem(struct user_struct *user, 5963ed29b0b4SJens Axboe unsigned long nr_pages) 5964ed29b0b4SJens Axboe { 5965ed29b0b4SJens Axboe unsigned long page_limit, cur_pages, new_pages; 5966ed29b0b4SJens Axboe 5967ed29b0b4SJens Axboe /* Don't allow more pages than we can safely lock */ 5968ed29b0b4SJens Axboe page_limit = rlimit(RLIMIT_MEMLOCK) >> PAGE_SHIFT; 5969ed29b0b4SJens Axboe 5970ed29b0b4SJens Axboe do { 5971ed29b0b4SJens Axboe cur_pages = atomic_long_read(&user->locked_vm); 5972ed29b0b4SJens Axboe new_pages = cur_pages + nr_pages; 5973ed29b0b4SJens Axboe if (new_pages > page_limit) 5974ed29b0b4SJens Axboe return -ENOMEM; 5975ed29b0b4SJens Axboe } while (atomic_long_cmpxchg(&user->locked_vm, cur_pages, 5976ed29b0b4SJens Axboe new_pages) != cur_pages); 5977ed29b0b4SJens Axboe 5978ed29b0b4SJens Axboe return 0; 5979ed29b0b4SJens Axboe } 5980ed29b0b4SJens Axboe 5981ed29b0b4SJens Axboe static void io_unaccount_mem(struct io_ring_ctx *ctx, unsigned long nr_pages) 5982ed29b0b4SJens Axboe { 5983ed29b0b4SJens Axboe if (ctx->user) 5984ed29b0b4SJens Axboe __io_unaccount_mem(ctx->user, nr_pages); 5985ed29b0b4SJens Axboe 5986ed29b0b4SJens Axboe if (ctx->mm_account) 5987ed29b0b4SJens Axboe atomic64_sub(nr_pages, &ctx->mm_account->pinned_vm); 5988ed29b0b4SJens Axboe } 5989ed29b0b4SJens Axboe 5990ed29b0b4SJens Axboe static int io_account_mem(struct io_ring_ctx *ctx, unsigned long nr_pages) 5991ed29b0b4SJens Axboe { 5992ed29b0b4SJens Axboe int ret; 5993ed29b0b4SJens Axboe 5994ed29b0b4SJens Axboe if (ctx->user) { 5995ed29b0b4SJens Axboe ret = __io_account_mem(ctx->user, nr_pages); 5996ed29b0b4SJens Axboe if (ret) 5997ed29b0b4SJens Axboe return ret; 5998ed29b0b4SJens Axboe } 5999ed29b0b4SJens Axboe 6000ed29b0b4SJens Axboe if (ctx->mm_account) 6001ed29b0b4SJens Axboe atomic64_add(nr_pages, &ctx->mm_account->pinned_vm); 6002ed29b0b4SJens Axboe 6003ed29b0b4SJens Axboe return 0; 6004ed29b0b4SJens Axboe } 6005ed29b0b4SJens Axboe 6006ed29b0b4SJens Axboe static void io_mem_free(void *ptr) 6007ed29b0b4SJens Axboe { 6008ed29b0b4SJens Axboe struct page *page; 6009ed29b0b4SJens Axboe 6010ed29b0b4SJens Axboe if (!ptr) 6011ed29b0b4SJens Axboe return; 6012ed29b0b4SJens Axboe 6013ed29b0b4SJens Axboe page = virt_to_head_page(ptr); 6014ed29b0b4SJens Axboe if (put_page_testzero(page)) 6015ed29b0b4SJens Axboe free_compound_page(page); 6016ed29b0b4SJens Axboe } 6017ed29b0b4SJens Axboe 6018ed29b0b4SJens Axboe static void *io_mem_alloc(size_t size) 6019ed29b0b4SJens Axboe { 6020ed29b0b4SJens Axboe gfp_t gfp = GFP_KERNEL_ACCOUNT | __GFP_ZERO | __GFP_NOWARN | __GFP_COMP; 6021ed29b0b4SJens Axboe 6022ed29b0b4SJens Axboe return (void *) __get_free_pages(gfp, get_order(size)); 6023ed29b0b4SJens Axboe } 6024ed29b0b4SJens Axboe 6025ed29b0b4SJens Axboe static unsigned long rings_size(struct io_ring_ctx *ctx, unsigned int sq_entries, 6026ed29b0b4SJens Axboe unsigned int cq_entries, size_t *sq_offset) 6027ed29b0b4SJens Axboe { 6028ed29b0b4SJens Axboe struct io_rings *rings; 6029ed29b0b4SJens Axboe size_t off, sq_array_size; 6030ed29b0b4SJens Axboe 6031ed29b0b4SJens Axboe off = struct_size(rings, cqes, cq_entries); 6032ed29b0b4SJens Axboe if (off == SIZE_MAX) 6033ed29b0b4SJens Axboe return SIZE_MAX; 6034ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_CQE32) { 6035ed29b0b4SJens Axboe if (check_shl_overflow(off, 1, &off)) 6036ed29b0b4SJens Axboe return SIZE_MAX; 6037ed29b0b4SJens Axboe } 6038ed29b0b4SJens Axboe 6039ed29b0b4SJens Axboe #ifdef CONFIG_SMP 6040ed29b0b4SJens Axboe off = ALIGN(off, SMP_CACHE_BYTES); 6041ed29b0b4SJens Axboe if (off == 0) 6042ed29b0b4SJens Axboe return SIZE_MAX; 6043ed29b0b4SJens Axboe #endif 6044ed29b0b4SJens Axboe 6045ed29b0b4SJens Axboe if (sq_offset) 6046ed29b0b4SJens Axboe *sq_offset = off; 6047ed29b0b4SJens Axboe 6048ed29b0b4SJens Axboe sq_array_size = array_size(sizeof(u32), sq_entries); 6049ed29b0b4SJens Axboe if (sq_array_size == SIZE_MAX) 6050ed29b0b4SJens Axboe return SIZE_MAX; 6051ed29b0b4SJens Axboe 6052ed29b0b4SJens Axboe if (check_add_overflow(off, sq_array_size, &off)) 6053ed29b0b4SJens Axboe return SIZE_MAX; 6054ed29b0b4SJens Axboe 6055ed29b0b4SJens Axboe return off; 6056ed29b0b4SJens Axboe } 6057ed29b0b4SJens Axboe 6058ed29b0b4SJens Axboe static void io_buffer_unmap(struct io_ring_ctx *ctx, struct io_mapped_ubuf **slot) 6059ed29b0b4SJens Axboe { 6060ed29b0b4SJens Axboe struct io_mapped_ubuf *imu = *slot; 6061ed29b0b4SJens Axboe unsigned int i; 6062ed29b0b4SJens Axboe 6063ed29b0b4SJens Axboe if (imu != ctx->dummy_ubuf) { 6064ed29b0b4SJens Axboe for (i = 0; i < imu->nr_bvecs; i++) 6065ed29b0b4SJens Axboe unpin_user_page(imu->bvec[i].bv_page); 6066ed29b0b4SJens Axboe if (imu->acct_pages) 6067ed29b0b4SJens Axboe io_unaccount_mem(ctx, imu->acct_pages); 6068ed29b0b4SJens Axboe kvfree(imu); 6069ed29b0b4SJens Axboe } 6070ed29b0b4SJens Axboe *slot = NULL; 6071ed29b0b4SJens Axboe } 6072ed29b0b4SJens Axboe 6073ed29b0b4SJens Axboe static void io_rsrc_buf_put(struct io_ring_ctx *ctx, struct io_rsrc_put *prsrc) 6074ed29b0b4SJens Axboe { 6075ed29b0b4SJens Axboe io_buffer_unmap(ctx, &prsrc->buf); 6076ed29b0b4SJens Axboe prsrc->buf = NULL; 6077ed29b0b4SJens Axboe } 6078ed29b0b4SJens Axboe 6079ed29b0b4SJens Axboe static void __io_sqe_buffers_unregister(struct io_ring_ctx *ctx) 6080ed29b0b4SJens Axboe { 6081ed29b0b4SJens Axboe unsigned int i; 6082ed29b0b4SJens Axboe 6083ed29b0b4SJens Axboe for (i = 0; i < ctx->nr_user_bufs; i++) 6084ed29b0b4SJens Axboe io_buffer_unmap(ctx, &ctx->user_bufs[i]); 6085ed29b0b4SJens Axboe kfree(ctx->user_bufs); 6086ed29b0b4SJens Axboe io_rsrc_data_free(ctx->buf_data); 6087ed29b0b4SJens Axboe ctx->user_bufs = NULL; 6088ed29b0b4SJens Axboe ctx->buf_data = NULL; 6089ed29b0b4SJens Axboe ctx->nr_user_bufs = 0; 6090ed29b0b4SJens Axboe } 6091ed29b0b4SJens Axboe 6092ed29b0b4SJens Axboe static int io_sqe_buffers_unregister(struct io_ring_ctx *ctx) 6093ed29b0b4SJens Axboe { 6094ed29b0b4SJens Axboe unsigned nr = ctx->nr_user_bufs; 6095ed29b0b4SJens Axboe int ret; 6096ed29b0b4SJens Axboe 6097ed29b0b4SJens Axboe if (!ctx->buf_data) 6098ed29b0b4SJens Axboe return -ENXIO; 6099ed29b0b4SJens Axboe 6100ed29b0b4SJens Axboe /* 6101ed29b0b4SJens Axboe * Quiesce may unlock ->uring_lock, and while it's not held 6102ed29b0b4SJens Axboe * prevent new requests using the table. 6103ed29b0b4SJens Axboe */ 6104ed29b0b4SJens Axboe ctx->nr_user_bufs = 0; 6105ed29b0b4SJens Axboe ret = io_rsrc_ref_quiesce(ctx->buf_data, ctx); 6106ed29b0b4SJens Axboe ctx->nr_user_bufs = nr; 6107ed29b0b4SJens Axboe if (!ret) 6108ed29b0b4SJens Axboe __io_sqe_buffers_unregister(ctx); 6109ed29b0b4SJens Axboe return ret; 6110ed29b0b4SJens Axboe } 6111ed29b0b4SJens Axboe 6112ed29b0b4SJens Axboe static int io_copy_iov(struct io_ring_ctx *ctx, struct iovec *dst, 6113ed29b0b4SJens Axboe void __user *arg, unsigned index) 6114ed29b0b4SJens Axboe { 6115ed29b0b4SJens Axboe struct iovec __user *src; 6116ed29b0b4SJens Axboe 6117ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 6118ed29b0b4SJens Axboe if (ctx->compat) { 6119ed29b0b4SJens Axboe struct compat_iovec __user *ciovs; 6120ed29b0b4SJens Axboe struct compat_iovec ciov; 6121ed29b0b4SJens Axboe 6122ed29b0b4SJens Axboe ciovs = (struct compat_iovec __user *) arg; 6123ed29b0b4SJens Axboe if (copy_from_user(&ciov, &ciovs[index], sizeof(ciov))) 6124ed29b0b4SJens Axboe return -EFAULT; 6125ed29b0b4SJens Axboe 6126ed29b0b4SJens Axboe dst->iov_base = u64_to_user_ptr((u64)ciov.iov_base); 6127ed29b0b4SJens Axboe dst->iov_len = ciov.iov_len; 6128ed29b0b4SJens Axboe return 0; 6129ed29b0b4SJens Axboe } 6130ed29b0b4SJens Axboe #endif 6131ed29b0b4SJens Axboe src = (struct iovec __user *) arg; 6132ed29b0b4SJens Axboe if (copy_from_user(dst, &src[index], sizeof(*dst))) 6133ed29b0b4SJens Axboe return -EFAULT; 6134ed29b0b4SJens Axboe return 0; 6135ed29b0b4SJens Axboe } 6136ed29b0b4SJens Axboe 6137ed29b0b4SJens Axboe /* 6138ed29b0b4SJens Axboe * Not super efficient, but this is just a registration time. And we do cache 6139ed29b0b4SJens Axboe * the last compound head, so generally we'll only do a full search if we don't 6140ed29b0b4SJens Axboe * match that one. 6141ed29b0b4SJens Axboe * 6142ed29b0b4SJens Axboe * We check if the given compound head page has already been accounted, to 6143ed29b0b4SJens Axboe * avoid double accounting it. This allows us to account the full size of the 6144ed29b0b4SJens Axboe * page, not just the constituent pages of a huge page. 6145ed29b0b4SJens Axboe */ 6146ed29b0b4SJens Axboe static bool headpage_already_acct(struct io_ring_ctx *ctx, struct page **pages, 6147ed29b0b4SJens Axboe int nr_pages, struct page *hpage) 6148ed29b0b4SJens Axboe { 6149ed29b0b4SJens Axboe int i, j; 6150ed29b0b4SJens Axboe 6151ed29b0b4SJens Axboe /* check current page array */ 6152ed29b0b4SJens Axboe for (i = 0; i < nr_pages; i++) { 6153ed29b0b4SJens Axboe if (!PageCompound(pages[i])) 6154ed29b0b4SJens Axboe continue; 6155ed29b0b4SJens Axboe if (compound_head(pages[i]) == hpage) 6156ed29b0b4SJens Axboe return true; 6157ed29b0b4SJens Axboe } 6158ed29b0b4SJens Axboe 6159ed29b0b4SJens Axboe /* check previously registered pages */ 6160ed29b0b4SJens Axboe for (i = 0; i < ctx->nr_user_bufs; i++) { 6161ed29b0b4SJens Axboe struct io_mapped_ubuf *imu = ctx->user_bufs[i]; 6162ed29b0b4SJens Axboe 6163ed29b0b4SJens Axboe for (j = 0; j < imu->nr_bvecs; j++) { 6164ed29b0b4SJens Axboe if (!PageCompound(imu->bvec[j].bv_page)) 6165ed29b0b4SJens Axboe continue; 6166ed29b0b4SJens Axboe if (compound_head(imu->bvec[j].bv_page) == hpage) 6167ed29b0b4SJens Axboe return true; 6168ed29b0b4SJens Axboe } 6169ed29b0b4SJens Axboe } 6170ed29b0b4SJens Axboe 6171ed29b0b4SJens Axboe return false; 6172ed29b0b4SJens Axboe } 6173ed29b0b4SJens Axboe 6174ed29b0b4SJens Axboe static int io_buffer_account_pin(struct io_ring_ctx *ctx, struct page **pages, 6175ed29b0b4SJens Axboe int nr_pages, struct io_mapped_ubuf *imu, 6176ed29b0b4SJens Axboe struct page **last_hpage) 6177ed29b0b4SJens Axboe { 6178ed29b0b4SJens Axboe int i, ret; 6179ed29b0b4SJens Axboe 6180ed29b0b4SJens Axboe imu->acct_pages = 0; 6181ed29b0b4SJens Axboe for (i = 0; i < nr_pages; i++) { 6182ed29b0b4SJens Axboe if (!PageCompound(pages[i])) { 6183ed29b0b4SJens Axboe imu->acct_pages++; 6184ed29b0b4SJens Axboe } else { 6185ed29b0b4SJens Axboe struct page *hpage; 6186ed29b0b4SJens Axboe 6187ed29b0b4SJens Axboe hpage = compound_head(pages[i]); 6188ed29b0b4SJens Axboe if (hpage == *last_hpage) 6189ed29b0b4SJens Axboe continue; 6190ed29b0b4SJens Axboe *last_hpage = hpage; 6191ed29b0b4SJens Axboe if (headpage_already_acct(ctx, pages, i, hpage)) 6192ed29b0b4SJens Axboe continue; 6193ed29b0b4SJens Axboe imu->acct_pages += page_size(hpage) >> PAGE_SHIFT; 6194ed29b0b4SJens Axboe } 6195ed29b0b4SJens Axboe } 6196ed29b0b4SJens Axboe 6197ed29b0b4SJens Axboe if (!imu->acct_pages) 6198ed29b0b4SJens Axboe return 0; 6199ed29b0b4SJens Axboe 6200ed29b0b4SJens Axboe ret = io_account_mem(ctx, imu->acct_pages); 6201ed29b0b4SJens Axboe if (ret) 6202ed29b0b4SJens Axboe imu->acct_pages = 0; 6203ed29b0b4SJens Axboe return ret; 6204ed29b0b4SJens Axboe } 6205ed29b0b4SJens Axboe 6206ed29b0b4SJens Axboe static struct page **io_pin_pages(unsigned long ubuf, unsigned long len, 6207ed29b0b4SJens Axboe int *npages) 6208ed29b0b4SJens Axboe { 6209ed29b0b4SJens Axboe unsigned long start, end, nr_pages; 6210ed29b0b4SJens Axboe struct vm_area_struct **vmas = NULL; 6211ed29b0b4SJens Axboe struct page **pages = NULL; 6212ed29b0b4SJens Axboe int i, pret, ret = -ENOMEM; 6213ed29b0b4SJens Axboe 6214ed29b0b4SJens Axboe end = (ubuf + len + PAGE_SIZE - 1) >> PAGE_SHIFT; 6215ed29b0b4SJens Axboe start = ubuf >> PAGE_SHIFT; 6216ed29b0b4SJens Axboe nr_pages = end - start; 6217ed29b0b4SJens Axboe 6218ed29b0b4SJens Axboe pages = kvmalloc_array(nr_pages, sizeof(struct page *), GFP_KERNEL); 6219ed29b0b4SJens Axboe if (!pages) 6220ed29b0b4SJens Axboe goto done; 6221ed29b0b4SJens Axboe 6222ed29b0b4SJens Axboe vmas = kvmalloc_array(nr_pages, sizeof(struct vm_area_struct *), 6223ed29b0b4SJens Axboe GFP_KERNEL); 6224ed29b0b4SJens Axboe if (!vmas) 6225ed29b0b4SJens Axboe goto done; 6226ed29b0b4SJens Axboe 6227ed29b0b4SJens Axboe ret = 0; 6228ed29b0b4SJens Axboe mmap_read_lock(current->mm); 6229ed29b0b4SJens Axboe pret = pin_user_pages(ubuf, nr_pages, FOLL_WRITE | FOLL_LONGTERM, 6230ed29b0b4SJens Axboe pages, vmas); 6231ed29b0b4SJens Axboe if (pret == nr_pages) { 6232ed29b0b4SJens Axboe /* don't support file backed memory */ 6233ed29b0b4SJens Axboe for (i = 0; i < nr_pages; i++) { 6234ed29b0b4SJens Axboe struct vm_area_struct *vma = vmas[i]; 6235ed29b0b4SJens Axboe 6236ed29b0b4SJens Axboe if (vma_is_shmem(vma)) 6237ed29b0b4SJens Axboe continue; 6238ed29b0b4SJens Axboe if (vma->vm_file && 6239ed29b0b4SJens Axboe !is_file_hugepages(vma->vm_file)) { 6240ed29b0b4SJens Axboe ret = -EOPNOTSUPP; 6241ed29b0b4SJens Axboe break; 6242ed29b0b4SJens Axboe } 6243ed29b0b4SJens Axboe } 6244ed29b0b4SJens Axboe *npages = nr_pages; 6245ed29b0b4SJens Axboe } else { 6246ed29b0b4SJens Axboe ret = pret < 0 ? pret : -EFAULT; 6247ed29b0b4SJens Axboe } 6248ed29b0b4SJens Axboe mmap_read_unlock(current->mm); 6249ed29b0b4SJens Axboe if (ret) { 6250ed29b0b4SJens Axboe /* 6251ed29b0b4SJens Axboe * if we did partial map, or found file backed vmas, 6252ed29b0b4SJens Axboe * release any pages we did get 6253ed29b0b4SJens Axboe */ 6254ed29b0b4SJens Axboe if (pret > 0) 6255ed29b0b4SJens Axboe unpin_user_pages(pages, pret); 6256ed29b0b4SJens Axboe goto done; 6257ed29b0b4SJens Axboe } 6258ed29b0b4SJens Axboe ret = 0; 6259ed29b0b4SJens Axboe done: 6260ed29b0b4SJens Axboe kvfree(vmas); 6261ed29b0b4SJens Axboe if (ret < 0) { 6262ed29b0b4SJens Axboe kvfree(pages); 6263ed29b0b4SJens Axboe pages = ERR_PTR(ret); 6264ed29b0b4SJens Axboe } 6265ed29b0b4SJens Axboe return pages; 6266ed29b0b4SJens Axboe } 6267ed29b0b4SJens Axboe 6268ed29b0b4SJens Axboe static int io_sqe_buffer_register(struct io_ring_ctx *ctx, struct iovec *iov, 6269ed29b0b4SJens Axboe struct io_mapped_ubuf **pimu, 6270ed29b0b4SJens Axboe struct page **last_hpage) 6271ed29b0b4SJens Axboe { 6272ed29b0b4SJens Axboe struct io_mapped_ubuf *imu = NULL; 6273ed29b0b4SJens Axboe struct page **pages = NULL; 6274ed29b0b4SJens Axboe unsigned long off; 6275ed29b0b4SJens Axboe size_t size; 6276ed29b0b4SJens Axboe int ret, nr_pages, i; 6277ed29b0b4SJens Axboe 6278ed29b0b4SJens Axboe if (!iov->iov_base) { 6279ed29b0b4SJens Axboe *pimu = ctx->dummy_ubuf; 6280ed29b0b4SJens Axboe return 0; 6281ed29b0b4SJens Axboe } 6282ed29b0b4SJens Axboe 6283ed29b0b4SJens Axboe *pimu = NULL; 6284ed29b0b4SJens Axboe ret = -ENOMEM; 6285ed29b0b4SJens Axboe 6286ed29b0b4SJens Axboe pages = io_pin_pages((unsigned long) iov->iov_base, iov->iov_len, 6287ed29b0b4SJens Axboe &nr_pages); 6288ed29b0b4SJens Axboe if (IS_ERR(pages)) { 6289ed29b0b4SJens Axboe ret = PTR_ERR(pages); 6290ed29b0b4SJens Axboe pages = NULL; 6291ed29b0b4SJens Axboe goto done; 6292ed29b0b4SJens Axboe } 6293ed29b0b4SJens Axboe 6294ed29b0b4SJens Axboe imu = kvmalloc(struct_size(imu, bvec, nr_pages), GFP_KERNEL); 6295ed29b0b4SJens Axboe if (!imu) 6296ed29b0b4SJens Axboe goto done; 6297ed29b0b4SJens Axboe 6298ed29b0b4SJens Axboe ret = io_buffer_account_pin(ctx, pages, nr_pages, imu, last_hpage); 6299ed29b0b4SJens Axboe if (ret) { 6300ed29b0b4SJens Axboe unpin_user_pages(pages, nr_pages); 6301ed29b0b4SJens Axboe goto done; 6302ed29b0b4SJens Axboe } 6303ed29b0b4SJens Axboe 6304ed29b0b4SJens Axboe off = (unsigned long) iov->iov_base & ~PAGE_MASK; 6305ed29b0b4SJens Axboe size = iov->iov_len; 6306ed29b0b4SJens Axboe for (i = 0; i < nr_pages; i++) { 6307ed29b0b4SJens Axboe size_t vec_len; 6308ed29b0b4SJens Axboe 6309ed29b0b4SJens Axboe vec_len = min_t(size_t, size, PAGE_SIZE - off); 6310ed29b0b4SJens Axboe imu->bvec[i].bv_page = pages[i]; 6311ed29b0b4SJens Axboe imu->bvec[i].bv_len = vec_len; 6312ed29b0b4SJens Axboe imu->bvec[i].bv_offset = off; 6313ed29b0b4SJens Axboe off = 0; 6314ed29b0b4SJens Axboe size -= vec_len; 6315ed29b0b4SJens Axboe } 6316ed29b0b4SJens Axboe /* store original address for later verification */ 6317ed29b0b4SJens Axboe imu->ubuf = (unsigned long) iov->iov_base; 6318ed29b0b4SJens Axboe imu->ubuf_end = imu->ubuf + iov->iov_len; 6319ed29b0b4SJens Axboe imu->nr_bvecs = nr_pages; 6320ed29b0b4SJens Axboe *pimu = imu; 6321ed29b0b4SJens Axboe ret = 0; 6322ed29b0b4SJens Axboe done: 6323ed29b0b4SJens Axboe if (ret) 6324ed29b0b4SJens Axboe kvfree(imu); 6325ed29b0b4SJens Axboe kvfree(pages); 6326ed29b0b4SJens Axboe return ret; 6327ed29b0b4SJens Axboe } 6328ed29b0b4SJens Axboe 6329ed29b0b4SJens Axboe static int io_buffers_map_alloc(struct io_ring_ctx *ctx, unsigned int nr_args) 6330ed29b0b4SJens Axboe { 6331ed29b0b4SJens Axboe ctx->user_bufs = kcalloc(nr_args, sizeof(*ctx->user_bufs), GFP_KERNEL); 6332ed29b0b4SJens Axboe return ctx->user_bufs ? 0 : -ENOMEM; 6333ed29b0b4SJens Axboe } 6334ed29b0b4SJens Axboe 6335ed29b0b4SJens Axboe static int io_buffer_validate(struct iovec *iov) 6336ed29b0b4SJens Axboe { 6337ed29b0b4SJens Axboe unsigned long tmp, acct_len = iov->iov_len + (PAGE_SIZE - 1); 6338ed29b0b4SJens Axboe 6339ed29b0b4SJens Axboe /* 6340ed29b0b4SJens Axboe * Don't impose further limits on the size and buffer 6341ed29b0b4SJens Axboe * constraints here, we'll -EINVAL later when IO is 6342ed29b0b4SJens Axboe * submitted if they are wrong. 6343ed29b0b4SJens Axboe */ 6344ed29b0b4SJens Axboe if (!iov->iov_base) 6345ed29b0b4SJens Axboe return iov->iov_len ? -EFAULT : 0; 6346ed29b0b4SJens Axboe if (!iov->iov_len) 6347ed29b0b4SJens Axboe return -EFAULT; 6348ed29b0b4SJens Axboe 6349ed29b0b4SJens Axboe /* arbitrary limit, but we need something */ 6350ed29b0b4SJens Axboe if (iov->iov_len > SZ_1G) 6351ed29b0b4SJens Axboe return -EFAULT; 6352ed29b0b4SJens Axboe 6353ed29b0b4SJens Axboe if (check_add_overflow((unsigned long)iov->iov_base, acct_len, &tmp)) 6354ed29b0b4SJens Axboe return -EOVERFLOW; 6355ed29b0b4SJens Axboe 6356ed29b0b4SJens Axboe return 0; 6357ed29b0b4SJens Axboe } 6358ed29b0b4SJens Axboe 6359ed29b0b4SJens Axboe static int io_sqe_buffers_register(struct io_ring_ctx *ctx, void __user *arg, 6360ed29b0b4SJens Axboe unsigned int nr_args, u64 __user *tags) 6361ed29b0b4SJens Axboe { 6362ed29b0b4SJens Axboe struct page *last_hpage = NULL; 6363ed29b0b4SJens Axboe struct io_rsrc_data *data; 6364ed29b0b4SJens Axboe int i, ret; 6365ed29b0b4SJens Axboe struct iovec iov; 6366ed29b0b4SJens Axboe 6367ed29b0b4SJens Axboe if (ctx->user_bufs) 6368ed29b0b4SJens Axboe return -EBUSY; 6369ed29b0b4SJens Axboe if (!nr_args || nr_args > IORING_MAX_REG_BUFFERS) 6370ed29b0b4SJens Axboe return -EINVAL; 6371ed29b0b4SJens Axboe ret = io_rsrc_node_switch_start(ctx); 6372ed29b0b4SJens Axboe if (ret) 6373ed29b0b4SJens Axboe return ret; 6374ed29b0b4SJens Axboe ret = io_rsrc_data_alloc(ctx, io_rsrc_buf_put, tags, nr_args, &data); 6375ed29b0b4SJens Axboe if (ret) 6376ed29b0b4SJens Axboe return ret; 6377ed29b0b4SJens Axboe ret = io_buffers_map_alloc(ctx, nr_args); 6378ed29b0b4SJens Axboe if (ret) { 6379ed29b0b4SJens Axboe io_rsrc_data_free(data); 6380ed29b0b4SJens Axboe return ret; 6381ed29b0b4SJens Axboe } 6382ed29b0b4SJens Axboe 6383ed29b0b4SJens Axboe for (i = 0; i < nr_args; i++, ctx->nr_user_bufs++) { 6384ed29b0b4SJens Axboe if (arg) { 6385ed29b0b4SJens Axboe ret = io_copy_iov(ctx, &iov, arg, i); 6386ed29b0b4SJens Axboe if (ret) 6387ed29b0b4SJens Axboe break; 6388ed29b0b4SJens Axboe ret = io_buffer_validate(&iov); 6389ed29b0b4SJens Axboe if (ret) 6390ed29b0b4SJens Axboe break; 6391ed29b0b4SJens Axboe } else { 6392ed29b0b4SJens Axboe memset(&iov, 0, sizeof(iov)); 6393ed29b0b4SJens Axboe } 6394ed29b0b4SJens Axboe 6395ed29b0b4SJens Axboe if (!iov.iov_base && *io_get_tag_slot(data, i)) { 6396ed29b0b4SJens Axboe ret = -EINVAL; 6397ed29b0b4SJens Axboe break; 6398ed29b0b4SJens Axboe } 6399ed29b0b4SJens Axboe 6400ed29b0b4SJens Axboe ret = io_sqe_buffer_register(ctx, &iov, &ctx->user_bufs[i], 6401ed29b0b4SJens Axboe &last_hpage); 6402ed29b0b4SJens Axboe if (ret) 6403ed29b0b4SJens Axboe break; 6404ed29b0b4SJens Axboe } 6405ed29b0b4SJens Axboe 6406ed29b0b4SJens Axboe WARN_ON_ONCE(ctx->buf_data); 6407ed29b0b4SJens Axboe 6408ed29b0b4SJens Axboe ctx->buf_data = data; 6409ed29b0b4SJens Axboe if (ret) 6410ed29b0b4SJens Axboe __io_sqe_buffers_unregister(ctx); 6411ed29b0b4SJens Axboe else 6412ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, NULL); 6413ed29b0b4SJens Axboe return ret; 6414ed29b0b4SJens Axboe } 6415ed29b0b4SJens Axboe 6416ed29b0b4SJens Axboe static int __io_sqe_buffers_update(struct io_ring_ctx *ctx, 6417ed29b0b4SJens Axboe struct io_uring_rsrc_update2 *up, 6418ed29b0b4SJens Axboe unsigned int nr_args) 6419ed29b0b4SJens Axboe { 6420ed29b0b4SJens Axboe u64 __user *tags = u64_to_user_ptr(up->tags); 6421ed29b0b4SJens Axboe struct iovec iov, __user *iovs = u64_to_user_ptr(up->data); 6422ed29b0b4SJens Axboe struct page *last_hpage = NULL; 6423ed29b0b4SJens Axboe bool needs_switch = false; 6424ed29b0b4SJens Axboe __u32 done; 6425ed29b0b4SJens Axboe int i, err; 6426ed29b0b4SJens Axboe 6427ed29b0b4SJens Axboe if (!ctx->buf_data) 6428ed29b0b4SJens Axboe return -ENXIO; 6429ed29b0b4SJens Axboe if (up->offset + nr_args > ctx->nr_user_bufs) 6430ed29b0b4SJens Axboe return -EINVAL; 6431ed29b0b4SJens Axboe 6432ed29b0b4SJens Axboe for (done = 0; done < nr_args; done++) { 6433ed29b0b4SJens Axboe struct io_mapped_ubuf *imu; 6434ed29b0b4SJens Axboe int offset = up->offset + done; 6435ed29b0b4SJens Axboe u64 tag = 0; 6436ed29b0b4SJens Axboe 6437ed29b0b4SJens Axboe err = io_copy_iov(ctx, &iov, iovs, done); 6438ed29b0b4SJens Axboe if (err) 6439ed29b0b4SJens Axboe break; 6440ed29b0b4SJens Axboe if (tags && copy_from_user(&tag, &tags[done], sizeof(tag))) { 6441ed29b0b4SJens Axboe err = -EFAULT; 6442ed29b0b4SJens Axboe break; 6443ed29b0b4SJens Axboe } 6444ed29b0b4SJens Axboe err = io_buffer_validate(&iov); 6445ed29b0b4SJens Axboe if (err) 6446ed29b0b4SJens Axboe break; 6447ed29b0b4SJens Axboe if (!iov.iov_base && tag) { 6448ed29b0b4SJens Axboe err = -EINVAL; 6449ed29b0b4SJens Axboe break; 6450ed29b0b4SJens Axboe } 6451ed29b0b4SJens Axboe err = io_sqe_buffer_register(ctx, &iov, &imu, &last_hpage); 6452ed29b0b4SJens Axboe if (err) 6453ed29b0b4SJens Axboe break; 6454ed29b0b4SJens Axboe 6455ed29b0b4SJens Axboe i = array_index_nospec(offset, ctx->nr_user_bufs); 6456ed29b0b4SJens Axboe if (ctx->user_bufs[i] != ctx->dummy_ubuf) { 6457ed29b0b4SJens Axboe err = io_queue_rsrc_removal(ctx->buf_data, i, 6458ed29b0b4SJens Axboe ctx->rsrc_node, ctx->user_bufs[i]); 6459ed29b0b4SJens Axboe if (unlikely(err)) { 6460ed29b0b4SJens Axboe io_buffer_unmap(ctx, &imu); 6461ed29b0b4SJens Axboe break; 6462ed29b0b4SJens Axboe } 6463ed29b0b4SJens Axboe ctx->user_bufs[i] = NULL; 6464ed29b0b4SJens Axboe needs_switch = true; 6465ed29b0b4SJens Axboe } 6466ed29b0b4SJens Axboe 6467ed29b0b4SJens Axboe ctx->user_bufs[i] = imu; 6468ed29b0b4SJens Axboe *io_get_tag_slot(ctx->buf_data, offset) = tag; 6469ed29b0b4SJens Axboe } 6470ed29b0b4SJens Axboe 6471ed29b0b4SJens Axboe if (needs_switch) 6472ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, ctx->buf_data); 6473ed29b0b4SJens Axboe return done ? done : err; 6474ed29b0b4SJens Axboe } 6475ed29b0b4SJens Axboe 6476ed29b0b4SJens Axboe static int io_eventfd_register(struct io_ring_ctx *ctx, void __user *arg, 6477ed29b0b4SJens Axboe unsigned int eventfd_async) 6478ed29b0b4SJens Axboe { 6479ed29b0b4SJens Axboe struct io_ev_fd *ev_fd; 6480ed29b0b4SJens Axboe __s32 __user *fds = arg; 6481ed29b0b4SJens Axboe int fd; 6482ed29b0b4SJens Axboe 6483ed29b0b4SJens Axboe ev_fd = rcu_dereference_protected(ctx->io_ev_fd, 6484ed29b0b4SJens Axboe lockdep_is_held(&ctx->uring_lock)); 6485ed29b0b4SJens Axboe if (ev_fd) 6486ed29b0b4SJens Axboe return -EBUSY; 6487ed29b0b4SJens Axboe 6488ed29b0b4SJens Axboe if (copy_from_user(&fd, fds, sizeof(*fds))) 6489ed29b0b4SJens Axboe return -EFAULT; 6490ed29b0b4SJens Axboe 6491ed29b0b4SJens Axboe ev_fd = kmalloc(sizeof(*ev_fd), GFP_KERNEL); 6492ed29b0b4SJens Axboe if (!ev_fd) 6493ed29b0b4SJens Axboe return -ENOMEM; 6494ed29b0b4SJens Axboe 6495ed29b0b4SJens Axboe ev_fd->cq_ev_fd = eventfd_ctx_fdget(fd); 6496ed29b0b4SJens Axboe if (IS_ERR(ev_fd->cq_ev_fd)) { 6497ed29b0b4SJens Axboe int ret = PTR_ERR(ev_fd->cq_ev_fd); 6498ed29b0b4SJens Axboe kfree(ev_fd); 6499ed29b0b4SJens Axboe return ret; 6500ed29b0b4SJens Axboe } 6501ed29b0b4SJens Axboe ev_fd->eventfd_async = eventfd_async; 6502ed29b0b4SJens Axboe ctx->has_evfd = true; 6503ed29b0b4SJens Axboe rcu_assign_pointer(ctx->io_ev_fd, ev_fd); 6504ed29b0b4SJens Axboe return 0; 6505ed29b0b4SJens Axboe } 6506ed29b0b4SJens Axboe 6507ed29b0b4SJens Axboe static void io_eventfd_put(struct rcu_head *rcu) 6508ed29b0b4SJens Axboe { 6509ed29b0b4SJens Axboe struct io_ev_fd *ev_fd = container_of(rcu, struct io_ev_fd, rcu); 6510ed29b0b4SJens Axboe 6511ed29b0b4SJens Axboe eventfd_ctx_put(ev_fd->cq_ev_fd); 6512ed29b0b4SJens Axboe kfree(ev_fd); 6513ed29b0b4SJens Axboe } 6514ed29b0b4SJens Axboe 6515ed29b0b4SJens Axboe static int io_eventfd_unregister(struct io_ring_ctx *ctx) 6516ed29b0b4SJens Axboe { 6517ed29b0b4SJens Axboe struct io_ev_fd *ev_fd; 6518ed29b0b4SJens Axboe 6519ed29b0b4SJens Axboe ev_fd = rcu_dereference_protected(ctx->io_ev_fd, 6520ed29b0b4SJens Axboe lockdep_is_held(&ctx->uring_lock)); 6521ed29b0b4SJens Axboe if (ev_fd) { 6522ed29b0b4SJens Axboe ctx->has_evfd = false; 6523ed29b0b4SJens Axboe rcu_assign_pointer(ctx->io_ev_fd, NULL); 6524ed29b0b4SJens Axboe call_rcu(&ev_fd->rcu, io_eventfd_put); 6525ed29b0b4SJens Axboe return 0; 6526ed29b0b4SJens Axboe } 6527ed29b0b4SJens Axboe 6528ed29b0b4SJens Axboe return -ENXIO; 6529ed29b0b4SJens Axboe } 6530ed29b0b4SJens Axboe 6531ed29b0b4SJens Axboe static void io_destroy_buffers(struct io_ring_ctx *ctx) 6532ed29b0b4SJens Axboe { 6533ed29b0b4SJens Axboe struct io_buffer_list *bl; 6534ed29b0b4SJens Axboe unsigned long index; 6535ed29b0b4SJens Axboe int i; 6536ed29b0b4SJens Axboe 6537ed29b0b4SJens Axboe for (i = 0; i < BGID_ARRAY; i++) { 6538ed29b0b4SJens Axboe if (!ctx->io_bl) 6539ed29b0b4SJens Axboe break; 6540ed29b0b4SJens Axboe __io_remove_buffers(ctx, &ctx->io_bl[i], -1U); 6541ed29b0b4SJens Axboe } 6542ed29b0b4SJens Axboe 6543ed29b0b4SJens Axboe xa_for_each(&ctx->io_bl_xa, index, bl) { 6544ed29b0b4SJens Axboe xa_erase(&ctx->io_bl_xa, bl->bgid); 6545ed29b0b4SJens Axboe __io_remove_buffers(ctx, bl, -1U); 6546ed29b0b4SJens Axboe kfree(bl); 6547ed29b0b4SJens Axboe } 6548ed29b0b4SJens Axboe 6549ed29b0b4SJens Axboe while (!list_empty(&ctx->io_buffers_pages)) { 6550ed29b0b4SJens Axboe struct page *page; 6551ed29b0b4SJens Axboe 6552ed29b0b4SJens Axboe page = list_first_entry(&ctx->io_buffers_pages, struct page, lru); 6553ed29b0b4SJens Axboe list_del_init(&page->lru); 6554ed29b0b4SJens Axboe __free_page(page); 6555ed29b0b4SJens Axboe } 6556ed29b0b4SJens Axboe } 6557ed29b0b4SJens Axboe 6558ed29b0b4SJens Axboe static void io_req_caches_free(struct io_ring_ctx *ctx) 6559ed29b0b4SJens Axboe { 6560ed29b0b4SJens Axboe struct io_submit_state *state = &ctx->submit_state; 6561ed29b0b4SJens Axboe int nr = 0; 6562ed29b0b4SJens Axboe 6563ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 6564ed29b0b4SJens Axboe io_flush_cached_locked_reqs(ctx, state); 6565ed29b0b4SJens Axboe 6566ed29b0b4SJens Axboe while (!io_req_cache_empty(ctx)) { 6567ed29b0b4SJens Axboe struct io_wq_work_node *node; 6568ed29b0b4SJens Axboe struct io_kiocb *req; 6569ed29b0b4SJens Axboe 6570ed29b0b4SJens Axboe node = wq_stack_extract(&state->free_list); 6571ed29b0b4SJens Axboe req = container_of(node, struct io_kiocb, comp_list); 6572ed29b0b4SJens Axboe kmem_cache_free(req_cachep, req); 6573ed29b0b4SJens Axboe nr++; 6574ed29b0b4SJens Axboe } 6575ed29b0b4SJens Axboe if (nr) 6576ed29b0b4SJens Axboe percpu_ref_put_many(&ctx->refs, nr); 6577ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 6578ed29b0b4SJens Axboe } 6579ed29b0b4SJens Axboe 6580ed29b0b4SJens Axboe static void io_wait_rsrc_data(struct io_rsrc_data *data) 6581ed29b0b4SJens Axboe { 6582ed29b0b4SJens Axboe if (data && !atomic_dec_and_test(&data->refs)) 6583ed29b0b4SJens Axboe wait_for_completion(&data->done); 6584ed29b0b4SJens Axboe } 6585ed29b0b4SJens Axboe 6586ed29b0b4SJens Axboe static void io_flush_apoll_cache(struct io_ring_ctx *ctx) 6587ed29b0b4SJens Axboe { 6588ed29b0b4SJens Axboe struct async_poll *apoll; 6589ed29b0b4SJens Axboe 6590ed29b0b4SJens Axboe while (!list_empty(&ctx->apoll_cache)) { 6591ed29b0b4SJens Axboe apoll = list_first_entry(&ctx->apoll_cache, struct async_poll, 6592ed29b0b4SJens Axboe poll.wait.entry); 6593ed29b0b4SJens Axboe list_del(&apoll->poll.wait.entry); 6594ed29b0b4SJens Axboe kfree(apoll); 6595ed29b0b4SJens Axboe } 6596ed29b0b4SJens Axboe } 6597ed29b0b4SJens Axboe 6598ed29b0b4SJens Axboe static __cold void io_ring_ctx_free(struct io_ring_ctx *ctx) 6599ed29b0b4SJens Axboe { 6600ed29b0b4SJens Axboe io_sq_thread_finish(ctx); 6601ed29b0b4SJens Axboe 6602ed29b0b4SJens Axboe if (ctx->mm_account) { 6603ed29b0b4SJens Axboe mmdrop(ctx->mm_account); 6604ed29b0b4SJens Axboe ctx->mm_account = NULL; 6605ed29b0b4SJens Axboe } 6606ed29b0b4SJens Axboe 6607ed29b0b4SJens Axboe io_rsrc_refs_drop(ctx); 6608ed29b0b4SJens Axboe /* __io_rsrc_put_work() may need uring_lock to progress, wait w/o it */ 6609ed29b0b4SJens Axboe io_wait_rsrc_data(ctx->buf_data); 6610ed29b0b4SJens Axboe io_wait_rsrc_data(ctx->file_data); 6611ed29b0b4SJens Axboe 6612ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 6613ed29b0b4SJens Axboe if (ctx->buf_data) 6614ed29b0b4SJens Axboe __io_sqe_buffers_unregister(ctx); 6615ed29b0b4SJens Axboe if (ctx->file_data) 6616ed29b0b4SJens Axboe __io_sqe_files_unregister(ctx); 6617ed29b0b4SJens Axboe if (ctx->rings) 6618ed29b0b4SJens Axboe __io_cqring_overflow_flush(ctx, true); 6619ed29b0b4SJens Axboe io_eventfd_unregister(ctx); 6620ed29b0b4SJens Axboe io_flush_apoll_cache(ctx); 6621ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 6622ed29b0b4SJens Axboe io_destroy_buffers(ctx); 6623ed29b0b4SJens Axboe if (ctx->sq_creds) 6624ed29b0b4SJens Axboe put_cred(ctx->sq_creds); 6625ed29b0b4SJens Axboe 6626ed29b0b4SJens Axboe /* there are no registered resources left, nobody uses it */ 6627ed29b0b4SJens Axboe if (ctx->rsrc_node) 6628ed29b0b4SJens Axboe io_rsrc_node_destroy(ctx->rsrc_node); 6629ed29b0b4SJens Axboe if (ctx->rsrc_backup_node) 6630ed29b0b4SJens Axboe io_rsrc_node_destroy(ctx->rsrc_backup_node); 6631ed29b0b4SJens Axboe flush_delayed_work(&ctx->rsrc_put_work); 6632ed29b0b4SJens Axboe flush_delayed_work(&ctx->fallback_work); 6633ed29b0b4SJens Axboe 6634ed29b0b4SJens Axboe WARN_ON_ONCE(!list_empty(&ctx->rsrc_ref_list)); 6635ed29b0b4SJens Axboe WARN_ON_ONCE(!llist_empty(&ctx->rsrc_put_llist)); 6636ed29b0b4SJens Axboe 6637ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 6638ed29b0b4SJens Axboe if (ctx->ring_sock) { 6639ed29b0b4SJens Axboe ctx->ring_sock->file = NULL; /* so that iput() is called */ 6640ed29b0b4SJens Axboe sock_release(ctx->ring_sock); 6641ed29b0b4SJens Axboe } 6642ed29b0b4SJens Axboe #endif 6643ed29b0b4SJens Axboe WARN_ON_ONCE(!list_empty(&ctx->ltimeout_list)); 6644ed29b0b4SJens Axboe 6645ed29b0b4SJens Axboe io_mem_free(ctx->rings); 6646ed29b0b4SJens Axboe io_mem_free(ctx->sq_sqes); 6647ed29b0b4SJens Axboe 6648ed29b0b4SJens Axboe percpu_ref_exit(&ctx->refs); 6649ed29b0b4SJens Axboe free_uid(ctx->user); 6650ed29b0b4SJens Axboe io_req_caches_free(ctx); 6651ed29b0b4SJens Axboe if (ctx->hash_map) 6652ed29b0b4SJens Axboe io_wq_put_hash(ctx->hash_map); 6653ed29b0b4SJens Axboe kfree(ctx->cancel_hash); 6654ed29b0b4SJens Axboe kfree(ctx->dummy_ubuf); 6655ed29b0b4SJens Axboe kfree(ctx->io_bl); 6656ed29b0b4SJens Axboe xa_destroy(&ctx->io_bl_xa); 6657ed29b0b4SJens Axboe kfree(ctx); 6658ed29b0b4SJens Axboe } 6659ed29b0b4SJens Axboe 6660ed29b0b4SJens Axboe static __poll_t io_uring_poll(struct file *file, poll_table *wait) 6661ed29b0b4SJens Axboe { 6662ed29b0b4SJens Axboe struct io_ring_ctx *ctx = file->private_data; 6663ed29b0b4SJens Axboe __poll_t mask = 0; 6664ed29b0b4SJens Axboe 6665ed29b0b4SJens Axboe poll_wait(file, &ctx->cq_wait, wait); 6666ed29b0b4SJens Axboe /* 6667ed29b0b4SJens Axboe * synchronizes with barrier from wq_has_sleeper call in 6668ed29b0b4SJens Axboe * io_commit_cqring 6669ed29b0b4SJens Axboe */ 6670ed29b0b4SJens Axboe smp_rmb(); 6671ed29b0b4SJens Axboe if (!io_sqring_full(ctx)) 6672ed29b0b4SJens Axboe mask |= EPOLLOUT | EPOLLWRNORM; 6673ed29b0b4SJens Axboe 6674ed29b0b4SJens Axboe /* 6675ed29b0b4SJens Axboe * Don't flush cqring overflow list here, just do a simple check. 6676ed29b0b4SJens Axboe * Otherwise there could possible be ABBA deadlock: 6677ed29b0b4SJens Axboe * CPU0 CPU1 6678ed29b0b4SJens Axboe * ---- ---- 6679ed29b0b4SJens Axboe * lock(&ctx->uring_lock); 6680ed29b0b4SJens Axboe * lock(&ep->mtx); 6681ed29b0b4SJens Axboe * lock(&ctx->uring_lock); 6682ed29b0b4SJens Axboe * lock(&ep->mtx); 6683ed29b0b4SJens Axboe * 6684ed29b0b4SJens Axboe * Users may get EPOLLIN meanwhile seeing nothing in cqring, this 6685ed29b0b4SJens Axboe * pushs them to do the flush. 6686ed29b0b4SJens Axboe */ 6687ed29b0b4SJens Axboe if (io_cqring_events(ctx) || 6688ed29b0b4SJens Axboe test_bit(IO_CHECK_CQ_OVERFLOW_BIT, &ctx->check_cq)) 6689ed29b0b4SJens Axboe mask |= EPOLLIN | EPOLLRDNORM; 6690ed29b0b4SJens Axboe 6691ed29b0b4SJens Axboe return mask; 6692ed29b0b4SJens Axboe } 6693ed29b0b4SJens Axboe 6694ed29b0b4SJens Axboe static int io_unregister_personality(struct io_ring_ctx *ctx, unsigned id) 6695ed29b0b4SJens Axboe { 6696ed29b0b4SJens Axboe const struct cred *creds; 6697ed29b0b4SJens Axboe 6698ed29b0b4SJens Axboe creds = xa_erase(&ctx->personalities, id); 6699ed29b0b4SJens Axboe if (creds) { 6700ed29b0b4SJens Axboe put_cred(creds); 6701ed29b0b4SJens Axboe return 0; 6702ed29b0b4SJens Axboe } 6703ed29b0b4SJens Axboe 6704ed29b0b4SJens Axboe return -EINVAL; 6705ed29b0b4SJens Axboe } 6706ed29b0b4SJens Axboe 6707ed29b0b4SJens Axboe struct io_tctx_exit { 6708ed29b0b4SJens Axboe struct callback_head task_work; 6709ed29b0b4SJens Axboe struct completion completion; 6710ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 6711ed29b0b4SJens Axboe }; 6712ed29b0b4SJens Axboe 6713ed29b0b4SJens Axboe static __cold void io_tctx_exit_cb(struct callback_head *cb) 6714ed29b0b4SJens Axboe { 6715ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 6716ed29b0b4SJens Axboe struct io_tctx_exit *work; 6717ed29b0b4SJens Axboe 6718ed29b0b4SJens Axboe work = container_of(cb, struct io_tctx_exit, task_work); 6719ed29b0b4SJens Axboe /* 6720ed29b0b4SJens Axboe * When @in_idle, we're in cancellation and it's racy to remove the 6721ed29b0b4SJens Axboe * node. It'll be removed by the end of cancellation, just ignore it. 6722ed29b0b4SJens Axboe */ 6723ed29b0b4SJens Axboe if (!atomic_read(&tctx->in_idle)) 6724ed29b0b4SJens Axboe io_uring_del_tctx_node((unsigned long)work->ctx); 6725ed29b0b4SJens Axboe complete(&work->completion); 6726ed29b0b4SJens Axboe } 6727ed29b0b4SJens Axboe 6728ed29b0b4SJens Axboe static __cold bool io_cancel_ctx_cb(struct io_wq_work *work, void *data) 6729ed29b0b4SJens Axboe { 6730ed29b0b4SJens Axboe struct io_kiocb *req = container_of(work, struct io_kiocb, work); 6731ed29b0b4SJens Axboe 6732ed29b0b4SJens Axboe return req->ctx == data; 6733ed29b0b4SJens Axboe } 6734ed29b0b4SJens Axboe 6735ed29b0b4SJens Axboe static __cold void io_ring_exit_work(struct work_struct *work) 6736ed29b0b4SJens Axboe { 6737ed29b0b4SJens Axboe struct io_ring_ctx *ctx = container_of(work, struct io_ring_ctx, exit_work); 6738ed29b0b4SJens Axboe unsigned long timeout = jiffies + HZ * 60 * 5; 6739ed29b0b4SJens Axboe unsigned long interval = HZ / 20; 6740ed29b0b4SJens Axboe struct io_tctx_exit exit; 6741ed29b0b4SJens Axboe struct io_tctx_node *node; 6742ed29b0b4SJens Axboe int ret; 6743ed29b0b4SJens Axboe 6744ed29b0b4SJens Axboe /* 6745ed29b0b4SJens Axboe * If we're doing polled IO and end up having requests being 6746ed29b0b4SJens Axboe * submitted async (out-of-line), then completions can come in while 6747ed29b0b4SJens Axboe * we're waiting for refs to drop. We need to reap these manually, 6748ed29b0b4SJens Axboe * as nobody else will be looking for them. 6749ed29b0b4SJens Axboe */ 6750ed29b0b4SJens Axboe do { 6751ed29b0b4SJens Axboe io_uring_try_cancel_requests(ctx, NULL, true); 6752ed29b0b4SJens Axboe if (ctx->sq_data) { 6753ed29b0b4SJens Axboe struct io_sq_data *sqd = ctx->sq_data; 6754ed29b0b4SJens Axboe struct task_struct *tsk; 6755ed29b0b4SJens Axboe 6756ed29b0b4SJens Axboe io_sq_thread_park(sqd); 6757ed29b0b4SJens Axboe tsk = sqd->thread; 6758ed29b0b4SJens Axboe if (tsk && tsk->io_uring && tsk->io_uring->io_wq) 6759ed29b0b4SJens Axboe io_wq_cancel_cb(tsk->io_uring->io_wq, 6760ed29b0b4SJens Axboe io_cancel_ctx_cb, ctx, true); 6761ed29b0b4SJens Axboe io_sq_thread_unpark(sqd); 6762ed29b0b4SJens Axboe } 6763ed29b0b4SJens Axboe 6764ed29b0b4SJens Axboe io_req_caches_free(ctx); 6765ed29b0b4SJens Axboe 6766ed29b0b4SJens Axboe if (WARN_ON_ONCE(time_after(jiffies, timeout))) { 6767ed29b0b4SJens Axboe /* there is little hope left, don't run it too often */ 6768ed29b0b4SJens Axboe interval = HZ * 60; 6769ed29b0b4SJens Axboe } 6770ed29b0b4SJens Axboe } while (!wait_for_completion_timeout(&ctx->ref_comp, interval)); 6771ed29b0b4SJens Axboe 6772ed29b0b4SJens Axboe init_completion(&exit.completion); 6773ed29b0b4SJens Axboe init_task_work(&exit.task_work, io_tctx_exit_cb); 6774ed29b0b4SJens Axboe exit.ctx = ctx; 6775ed29b0b4SJens Axboe /* 6776ed29b0b4SJens Axboe * Some may use context even when all refs and requests have been put, 6777ed29b0b4SJens Axboe * and they are free to do so while still holding uring_lock or 6778ed29b0b4SJens Axboe * completion_lock, see io_req_task_submit(). Apart from other work, 6779ed29b0b4SJens Axboe * this lock/unlock section also waits them to finish. 6780ed29b0b4SJens Axboe */ 6781ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 6782ed29b0b4SJens Axboe while (!list_empty(&ctx->tctx_list)) { 6783ed29b0b4SJens Axboe WARN_ON_ONCE(time_after(jiffies, timeout)); 6784ed29b0b4SJens Axboe 6785ed29b0b4SJens Axboe node = list_first_entry(&ctx->tctx_list, struct io_tctx_node, 6786ed29b0b4SJens Axboe ctx_node); 6787ed29b0b4SJens Axboe /* don't spin on a single task if cancellation failed */ 6788ed29b0b4SJens Axboe list_rotate_left(&ctx->tctx_list); 6789ed29b0b4SJens Axboe ret = task_work_add(node->task, &exit.task_work, TWA_SIGNAL); 6790ed29b0b4SJens Axboe if (WARN_ON_ONCE(ret)) 6791ed29b0b4SJens Axboe continue; 6792ed29b0b4SJens Axboe 6793ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 6794ed29b0b4SJens Axboe wait_for_completion(&exit.completion); 6795ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 6796ed29b0b4SJens Axboe } 6797ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 6798ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 6799ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 6800ed29b0b4SJens Axboe 6801ed29b0b4SJens Axboe io_ring_ctx_free(ctx); 6802ed29b0b4SJens Axboe } 6803ed29b0b4SJens Axboe 6804ed29b0b4SJens Axboe static __cold void io_ring_ctx_wait_and_kill(struct io_ring_ctx *ctx) 6805ed29b0b4SJens Axboe { 6806ed29b0b4SJens Axboe unsigned long index; 6807ed29b0b4SJens Axboe struct creds *creds; 6808ed29b0b4SJens Axboe 6809ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 6810ed29b0b4SJens Axboe percpu_ref_kill(&ctx->refs); 6811ed29b0b4SJens Axboe if (ctx->rings) 6812ed29b0b4SJens Axboe __io_cqring_overflow_flush(ctx, true); 6813ed29b0b4SJens Axboe xa_for_each(&ctx->personalities, index, creds) 6814ed29b0b4SJens Axboe io_unregister_personality(ctx, index); 6815ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 6816ed29b0b4SJens Axboe 6817ed29b0b4SJens Axboe /* failed during ring init, it couldn't have issued any requests */ 6818ed29b0b4SJens Axboe if (ctx->rings) { 6819ed29b0b4SJens Axboe io_kill_timeouts(ctx, NULL, true); 6820ed29b0b4SJens Axboe io_poll_remove_all(ctx, NULL, true); 6821ed29b0b4SJens Axboe /* if we failed setting up the ctx, we might not have any rings */ 6822ed29b0b4SJens Axboe io_iopoll_try_reap_events(ctx); 6823ed29b0b4SJens Axboe } 6824ed29b0b4SJens Axboe 6825ed29b0b4SJens Axboe INIT_WORK(&ctx->exit_work, io_ring_exit_work); 6826ed29b0b4SJens Axboe /* 6827ed29b0b4SJens Axboe * Use system_unbound_wq to avoid spawning tons of event kworkers 6828ed29b0b4SJens Axboe * if we're exiting a ton of rings at the same time. It just adds 6829ed29b0b4SJens Axboe * noise and overhead, there's no discernable change in runtime 6830ed29b0b4SJens Axboe * over using system_wq. 6831ed29b0b4SJens Axboe */ 6832ed29b0b4SJens Axboe queue_work(system_unbound_wq, &ctx->exit_work); 6833ed29b0b4SJens Axboe } 6834ed29b0b4SJens Axboe 6835ed29b0b4SJens Axboe static int io_uring_release(struct inode *inode, struct file *file) 6836ed29b0b4SJens Axboe { 6837ed29b0b4SJens Axboe struct io_ring_ctx *ctx = file->private_data; 6838ed29b0b4SJens Axboe 6839ed29b0b4SJens Axboe file->private_data = NULL; 6840ed29b0b4SJens Axboe io_ring_ctx_wait_and_kill(ctx); 6841ed29b0b4SJens Axboe return 0; 6842ed29b0b4SJens Axboe } 6843ed29b0b4SJens Axboe 6844ed29b0b4SJens Axboe struct io_task_cancel { 6845ed29b0b4SJens Axboe struct task_struct *task; 6846ed29b0b4SJens Axboe bool all; 6847ed29b0b4SJens Axboe }; 6848ed29b0b4SJens Axboe 6849ed29b0b4SJens Axboe static bool io_cancel_task_cb(struct io_wq_work *work, void *data) 6850ed29b0b4SJens Axboe { 6851ed29b0b4SJens Axboe struct io_kiocb *req = container_of(work, struct io_kiocb, work); 6852ed29b0b4SJens Axboe struct io_task_cancel *cancel = data; 6853ed29b0b4SJens Axboe 6854ed29b0b4SJens Axboe return io_match_task_safe(req, cancel->task, cancel->all); 6855ed29b0b4SJens Axboe } 6856ed29b0b4SJens Axboe 6857ed29b0b4SJens Axboe static __cold bool io_cancel_defer_files(struct io_ring_ctx *ctx, 6858ed29b0b4SJens Axboe struct task_struct *task, 6859ed29b0b4SJens Axboe bool cancel_all) 6860ed29b0b4SJens Axboe { 6861ed29b0b4SJens Axboe struct io_defer_entry *de; 6862ed29b0b4SJens Axboe LIST_HEAD(list); 6863ed29b0b4SJens Axboe 6864ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 6865ed29b0b4SJens Axboe list_for_each_entry_reverse(de, &ctx->defer_list, list) { 6866ed29b0b4SJens Axboe if (io_match_task_safe(de->req, task, cancel_all)) { 6867ed29b0b4SJens Axboe list_cut_position(&list, &ctx->defer_list, &de->list); 6868ed29b0b4SJens Axboe break; 6869ed29b0b4SJens Axboe } 6870ed29b0b4SJens Axboe } 6871ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 6872ed29b0b4SJens Axboe if (list_empty(&list)) 6873ed29b0b4SJens Axboe return false; 6874ed29b0b4SJens Axboe 6875ed29b0b4SJens Axboe while (!list_empty(&list)) { 6876ed29b0b4SJens Axboe de = list_first_entry(&list, struct io_defer_entry, list); 6877ed29b0b4SJens Axboe list_del_init(&de->list); 6878ed29b0b4SJens Axboe io_req_complete_failed(de->req, -ECANCELED); 6879ed29b0b4SJens Axboe kfree(de); 6880ed29b0b4SJens Axboe } 6881ed29b0b4SJens Axboe return true; 6882ed29b0b4SJens Axboe } 6883ed29b0b4SJens Axboe 6884ed29b0b4SJens Axboe static __cold bool io_uring_try_cancel_iowq(struct io_ring_ctx *ctx) 6885ed29b0b4SJens Axboe { 6886ed29b0b4SJens Axboe struct io_tctx_node *node; 6887ed29b0b4SJens Axboe enum io_wq_cancel cret; 6888ed29b0b4SJens Axboe bool ret = false; 6889ed29b0b4SJens Axboe 6890ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 6891ed29b0b4SJens Axboe list_for_each_entry(node, &ctx->tctx_list, ctx_node) { 6892ed29b0b4SJens Axboe struct io_uring_task *tctx = node->task->io_uring; 6893ed29b0b4SJens Axboe 6894ed29b0b4SJens Axboe /* 6895ed29b0b4SJens Axboe * io_wq will stay alive while we hold uring_lock, because it's 6896ed29b0b4SJens Axboe * killed after ctx nodes, which requires to take the lock. 6897ed29b0b4SJens Axboe */ 6898ed29b0b4SJens Axboe if (!tctx || !tctx->io_wq) 6899ed29b0b4SJens Axboe continue; 6900ed29b0b4SJens Axboe cret = io_wq_cancel_cb(tctx->io_wq, io_cancel_ctx_cb, ctx, true); 6901ed29b0b4SJens Axboe ret |= (cret != IO_WQ_CANCEL_NOTFOUND); 6902ed29b0b4SJens Axboe } 6903ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 6904ed29b0b4SJens Axboe 6905ed29b0b4SJens Axboe return ret; 6906ed29b0b4SJens Axboe } 6907ed29b0b4SJens Axboe 6908ed29b0b4SJens Axboe static __cold void io_uring_try_cancel_requests(struct io_ring_ctx *ctx, 6909ed29b0b4SJens Axboe struct task_struct *task, 6910ed29b0b4SJens Axboe bool cancel_all) 6911ed29b0b4SJens Axboe { 6912ed29b0b4SJens Axboe struct io_task_cancel cancel = { .task = task, .all = cancel_all, }; 6913ed29b0b4SJens Axboe struct io_uring_task *tctx = task ? task->io_uring : NULL; 6914ed29b0b4SJens Axboe 6915ed29b0b4SJens Axboe /* failed during ring init, it couldn't have issued any requests */ 6916ed29b0b4SJens Axboe if (!ctx->rings) 6917ed29b0b4SJens Axboe return; 6918ed29b0b4SJens Axboe 6919ed29b0b4SJens Axboe while (1) { 6920ed29b0b4SJens Axboe enum io_wq_cancel cret; 6921ed29b0b4SJens Axboe bool ret = false; 6922ed29b0b4SJens Axboe 6923ed29b0b4SJens Axboe if (!task) { 6924ed29b0b4SJens Axboe ret |= io_uring_try_cancel_iowq(ctx); 6925ed29b0b4SJens Axboe } else if (tctx && tctx->io_wq) { 6926ed29b0b4SJens Axboe /* 6927ed29b0b4SJens Axboe * Cancels requests of all rings, not only @ctx, but 6928ed29b0b4SJens Axboe * it's fine as the task is in exit/exec. 6929ed29b0b4SJens Axboe */ 6930ed29b0b4SJens Axboe cret = io_wq_cancel_cb(tctx->io_wq, io_cancel_task_cb, 6931ed29b0b4SJens Axboe &cancel, true); 6932ed29b0b4SJens Axboe ret |= (cret != IO_WQ_CANCEL_NOTFOUND); 6933ed29b0b4SJens Axboe } 6934ed29b0b4SJens Axboe 6935ed29b0b4SJens Axboe /* SQPOLL thread does its own polling */ 6936ed29b0b4SJens Axboe if ((!(ctx->flags & IORING_SETUP_SQPOLL) && cancel_all) || 6937ed29b0b4SJens Axboe (ctx->sq_data && ctx->sq_data->thread == current)) { 6938ed29b0b4SJens Axboe while (!wq_list_empty(&ctx->iopoll_list)) { 6939ed29b0b4SJens Axboe io_iopoll_try_reap_events(ctx); 6940ed29b0b4SJens Axboe ret = true; 6941ed29b0b4SJens Axboe } 6942ed29b0b4SJens Axboe } 6943ed29b0b4SJens Axboe 6944ed29b0b4SJens Axboe ret |= io_cancel_defer_files(ctx, task, cancel_all); 6945ed29b0b4SJens Axboe ret |= io_poll_remove_all(ctx, task, cancel_all); 6946ed29b0b4SJens Axboe ret |= io_kill_timeouts(ctx, task, cancel_all); 6947ed29b0b4SJens Axboe if (task) 6948ed29b0b4SJens Axboe ret |= io_run_task_work(); 6949ed29b0b4SJens Axboe if (!ret) 6950ed29b0b4SJens Axboe break; 6951ed29b0b4SJens Axboe cond_resched(); 6952ed29b0b4SJens Axboe } 6953ed29b0b4SJens Axboe } 6954ed29b0b4SJens Axboe 6955ed29b0b4SJens Axboe static s64 tctx_inflight(struct io_uring_task *tctx, bool tracked) 6956ed29b0b4SJens Axboe { 6957ed29b0b4SJens Axboe if (tracked) 6958ed29b0b4SJens Axboe return atomic_read(&tctx->inflight_tracked); 6959ed29b0b4SJens Axboe return percpu_counter_sum(&tctx->inflight); 6960ed29b0b4SJens Axboe } 6961ed29b0b4SJens Axboe 6962ed29b0b4SJens Axboe /* 6963ed29b0b4SJens Axboe * Find any io_uring ctx that this task has registered or done IO on, and cancel 6964ed29b0b4SJens Axboe * requests. @sqd should be not-null IFF it's an SQPOLL thread cancellation. 6965ed29b0b4SJens Axboe */ 696617437f31SJens Axboe __cold void io_uring_cancel_generic(bool cancel_all, struct io_sq_data *sqd) 6967ed29b0b4SJens Axboe { 6968ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 6969ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 6970ed29b0b4SJens Axboe s64 inflight; 6971ed29b0b4SJens Axboe DEFINE_WAIT(wait); 6972ed29b0b4SJens Axboe 6973ed29b0b4SJens Axboe WARN_ON_ONCE(sqd && sqd->thread != current); 6974ed29b0b4SJens Axboe 6975ed29b0b4SJens Axboe if (!current->io_uring) 6976ed29b0b4SJens Axboe return; 6977ed29b0b4SJens Axboe if (tctx->io_wq) 6978ed29b0b4SJens Axboe io_wq_exit_start(tctx->io_wq); 6979ed29b0b4SJens Axboe 6980ed29b0b4SJens Axboe atomic_inc(&tctx->in_idle); 6981ed29b0b4SJens Axboe do { 6982ed29b0b4SJens Axboe io_uring_drop_tctx_refs(current); 6983ed29b0b4SJens Axboe /* read completions before cancelations */ 6984ed29b0b4SJens Axboe inflight = tctx_inflight(tctx, !cancel_all); 6985ed29b0b4SJens Axboe if (!inflight) 6986ed29b0b4SJens Axboe break; 6987ed29b0b4SJens Axboe 6988ed29b0b4SJens Axboe if (!sqd) { 6989ed29b0b4SJens Axboe struct io_tctx_node *node; 6990ed29b0b4SJens Axboe unsigned long index; 6991ed29b0b4SJens Axboe 6992ed29b0b4SJens Axboe xa_for_each(&tctx->xa, index, node) { 6993ed29b0b4SJens Axboe /* sqpoll task will cancel all its requests */ 6994ed29b0b4SJens Axboe if (node->ctx->sq_data) 6995ed29b0b4SJens Axboe continue; 6996ed29b0b4SJens Axboe io_uring_try_cancel_requests(node->ctx, current, 6997ed29b0b4SJens Axboe cancel_all); 6998ed29b0b4SJens Axboe } 6999ed29b0b4SJens Axboe } else { 7000ed29b0b4SJens Axboe list_for_each_entry(ctx, &sqd->ctx_list, sqd_list) 7001ed29b0b4SJens Axboe io_uring_try_cancel_requests(ctx, current, 7002ed29b0b4SJens Axboe cancel_all); 7003ed29b0b4SJens Axboe } 7004ed29b0b4SJens Axboe 7005ed29b0b4SJens Axboe prepare_to_wait(&tctx->wait, &wait, TASK_INTERRUPTIBLE); 7006ed29b0b4SJens Axboe io_run_task_work(); 7007ed29b0b4SJens Axboe io_uring_drop_tctx_refs(current); 7008ed29b0b4SJens Axboe 7009ed29b0b4SJens Axboe /* 7010ed29b0b4SJens Axboe * If we've seen completions, retry without waiting. This 7011ed29b0b4SJens Axboe * avoids a race where a completion comes in before we did 7012ed29b0b4SJens Axboe * prepare_to_wait(). 7013ed29b0b4SJens Axboe */ 7014ed29b0b4SJens Axboe if (inflight == tctx_inflight(tctx, !cancel_all)) 7015ed29b0b4SJens Axboe schedule(); 7016ed29b0b4SJens Axboe finish_wait(&tctx->wait, &wait); 7017ed29b0b4SJens Axboe } while (1); 7018ed29b0b4SJens Axboe 7019ed29b0b4SJens Axboe io_uring_clean_tctx(tctx); 7020ed29b0b4SJens Axboe if (cancel_all) { 7021ed29b0b4SJens Axboe /* 7022ed29b0b4SJens Axboe * We shouldn't run task_works after cancel, so just leave 7023ed29b0b4SJens Axboe * ->in_idle set for normal exit. 7024ed29b0b4SJens Axboe */ 7025ed29b0b4SJens Axboe atomic_dec(&tctx->in_idle); 7026ed29b0b4SJens Axboe /* for exec all current's requests should be gone, kill tctx */ 7027ed29b0b4SJens Axboe __io_uring_free(current); 7028ed29b0b4SJens Axboe } 7029ed29b0b4SJens Axboe } 7030ed29b0b4SJens Axboe 7031ed29b0b4SJens Axboe void __io_uring_cancel(bool cancel_all) 7032ed29b0b4SJens Axboe { 7033ed29b0b4SJens Axboe io_uring_cancel_generic(cancel_all, NULL); 7034ed29b0b4SJens Axboe } 7035ed29b0b4SJens Axboe 7036ed29b0b4SJens Axboe static void *io_uring_validate_mmap_request(struct file *file, 7037ed29b0b4SJens Axboe loff_t pgoff, size_t sz) 7038ed29b0b4SJens Axboe { 7039ed29b0b4SJens Axboe struct io_ring_ctx *ctx = file->private_data; 7040ed29b0b4SJens Axboe loff_t offset = pgoff << PAGE_SHIFT; 7041ed29b0b4SJens Axboe struct page *page; 7042ed29b0b4SJens Axboe void *ptr; 7043ed29b0b4SJens Axboe 7044ed29b0b4SJens Axboe switch (offset) { 7045ed29b0b4SJens Axboe case IORING_OFF_SQ_RING: 7046ed29b0b4SJens Axboe case IORING_OFF_CQ_RING: 7047ed29b0b4SJens Axboe ptr = ctx->rings; 7048ed29b0b4SJens Axboe break; 7049ed29b0b4SJens Axboe case IORING_OFF_SQES: 7050ed29b0b4SJens Axboe ptr = ctx->sq_sqes; 7051ed29b0b4SJens Axboe break; 7052ed29b0b4SJens Axboe default: 7053ed29b0b4SJens Axboe return ERR_PTR(-EINVAL); 7054ed29b0b4SJens Axboe } 7055ed29b0b4SJens Axboe 7056ed29b0b4SJens Axboe page = virt_to_head_page(ptr); 7057ed29b0b4SJens Axboe if (sz > page_size(page)) 7058ed29b0b4SJens Axboe return ERR_PTR(-EINVAL); 7059ed29b0b4SJens Axboe 7060ed29b0b4SJens Axboe return ptr; 7061ed29b0b4SJens Axboe } 7062ed29b0b4SJens Axboe 7063ed29b0b4SJens Axboe #ifdef CONFIG_MMU 7064ed29b0b4SJens Axboe 7065ed29b0b4SJens Axboe static __cold int io_uring_mmap(struct file *file, struct vm_area_struct *vma) 7066ed29b0b4SJens Axboe { 7067ed29b0b4SJens Axboe size_t sz = vma->vm_end - vma->vm_start; 7068ed29b0b4SJens Axboe unsigned long pfn; 7069ed29b0b4SJens Axboe void *ptr; 7070ed29b0b4SJens Axboe 7071ed29b0b4SJens Axboe ptr = io_uring_validate_mmap_request(file, vma->vm_pgoff, sz); 7072ed29b0b4SJens Axboe if (IS_ERR(ptr)) 7073ed29b0b4SJens Axboe return PTR_ERR(ptr); 7074ed29b0b4SJens Axboe 7075ed29b0b4SJens Axboe pfn = virt_to_phys(ptr) >> PAGE_SHIFT; 7076ed29b0b4SJens Axboe return remap_pfn_range(vma, vma->vm_start, pfn, sz, vma->vm_page_prot); 7077ed29b0b4SJens Axboe } 7078ed29b0b4SJens Axboe 7079ed29b0b4SJens Axboe #else /* !CONFIG_MMU */ 7080ed29b0b4SJens Axboe 7081ed29b0b4SJens Axboe static int io_uring_mmap(struct file *file, struct vm_area_struct *vma) 7082ed29b0b4SJens Axboe { 7083ed29b0b4SJens Axboe return vma->vm_flags & (VM_SHARED | VM_MAYSHARE) ? 0 : -EINVAL; 7084ed29b0b4SJens Axboe } 7085ed29b0b4SJens Axboe 7086ed29b0b4SJens Axboe static unsigned int io_uring_nommu_mmap_capabilities(struct file *file) 7087ed29b0b4SJens Axboe { 7088ed29b0b4SJens Axboe return NOMMU_MAP_DIRECT | NOMMU_MAP_READ | NOMMU_MAP_WRITE; 7089ed29b0b4SJens Axboe } 7090ed29b0b4SJens Axboe 7091ed29b0b4SJens Axboe static unsigned long io_uring_nommu_get_unmapped_area(struct file *file, 7092ed29b0b4SJens Axboe unsigned long addr, unsigned long len, 7093ed29b0b4SJens Axboe unsigned long pgoff, unsigned long flags) 7094ed29b0b4SJens Axboe { 7095ed29b0b4SJens Axboe void *ptr; 7096ed29b0b4SJens Axboe 7097ed29b0b4SJens Axboe ptr = io_uring_validate_mmap_request(file, pgoff, len); 7098ed29b0b4SJens Axboe if (IS_ERR(ptr)) 7099ed29b0b4SJens Axboe return PTR_ERR(ptr); 7100ed29b0b4SJens Axboe 7101ed29b0b4SJens Axboe return (unsigned long) ptr; 7102ed29b0b4SJens Axboe } 7103ed29b0b4SJens Axboe 7104ed29b0b4SJens Axboe #endif /* !CONFIG_MMU */ 7105ed29b0b4SJens Axboe 7106ed29b0b4SJens Axboe static int io_validate_ext_arg(unsigned flags, const void __user *argp, size_t argsz) 7107ed29b0b4SJens Axboe { 7108ed29b0b4SJens Axboe if (flags & IORING_ENTER_EXT_ARG) { 7109ed29b0b4SJens Axboe struct io_uring_getevents_arg arg; 7110ed29b0b4SJens Axboe 7111ed29b0b4SJens Axboe if (argsz != sizeof(arg)) 7112ed29b0b4SJens Axboe return -EINVAL; 7113ed29b0b4SJens Axboe if (copy_from_user(&arg, argp, sizeof(arg))) 7114ed29b0b4SJens Axboe return -EFAULT; 7115ed29b0b4SJens Axboe } 7116ed29b0b4SJens Axboe return 0; 7117ed29b0b4SJens Axboe } 7118ed29b0b4SJens Axboe 7119ed29b0b4SJens Axboe static int io_get_ext_arg(unsigned flags, const void __user *argp, size_t *argsz, 7120ed29b0b4SJens Axboe struct __kernel_timespec __user **ts, 7121ed29b0b4SJens Axboe const sigset_t __user **sig) 7122ed29b0b4SJens Axboe { 7123ed29b0b4SJens Axboe struct io_uring_getevents_arg arg; 7124ed29b0b4SJens Axboe 7125ed29b0b4SJens Axboe /* 7126ed29b0b4SJens Axboe * If EXT_ARG isn't set, then we have no timespec and the argp pointer 7127ed29b0b4SJens Axboe * is just a pointer to the sigset_t. 7128ed29b0b4SJens Axboe */ 7129ed29b0b4SJens Axboe if (!(flags & IORING_ENTER_EXT_ARG)) { 7130ed29b0b4SJens Axboe *sig = (const sigset_t __user *) argp; 7131ed29b0b4SJens Axboe *ts = NULL; 7132ed29b0b4SJens Axboe return 0; 7133ed29b0b4SJens Axboe } 7134ed29b0b4SJens Axboe 7135ed29b0b4SJens Axboe /* 7136ed29b0b4SJens Axboe * EXT_ARG is set - ensure we agree on the size of it and copy in our 7137ed29b0b4SJens Axboe * timespec and sigset_t pointers if good. 7138ed29b0b4SJens Axboe */ 7139ed29b0b4SJens Axboe if (*argsz != sizeof(arg)) 7140ed29b0b4SJens Axboe return -EINVAL; 7141ed29b0b4SJens Axboe if (copy_from_user(&arg, argp, sizeof(arg))) 7142ed29b0b4SJens Axboe return -EFAULT; 7143ed29b0b4SJens Axboe if (arg.pad) 7144ed29b0b4SJens Axboe return -EINVAL; 7145ed29b0b4SJens Axboe *sig = u64_to_user_ptr(arg.sigmask); 7146ed29b0b4SJens Axboe *argsz = arg.sigmask_sz; 7147ed29b0b4SJens Axboe *ts = u64_to_user_ptr(arg.ts); 7148ed29b0b4SJens Axboe return 0; 7149ed29b0b4SJens Axboe } 7150ed29b0b4SJens Axboe 7151ed29b0b4SJens Axboe SYSCALL_DEFINE6(io_uring_enter, unsigned int, fd, u32, to_submit, 7152ed29b0b4SJens Axboe u32, min_complete, u32, flags, const void __user *, argp, 7153ed29b0b4SJens Axboe size_t, argsz) 7154ed29b0b4SJens Axboe { 7155ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 7156ed29b0b4SJens Axboe struct fd f; 7157ed29b0b4SJens Axboe long ret; 7158ed29b0b4SJens Axboe 7159ed29b0b4SJens Axboe io_run_task_work(); 7160ed29b0b4SJens Axboe 7161ed29b0b4SJens Axboe if (unlikely(flags & ~(IORING_ENTER_GETEVENTS | IORING_ENTER_SQ_WAKEUP | 7162ed29b0b4SJens Axboe IORING_ENTER_SQ_WAIT | IORING_ENTER_EXT_ARG | 7163ed29b0b4SJens Axboe IORING_ENTER_REGISTERED_RING))) 7164ed29b0b4SJens Axboe return -EINVAL; 7165ed29b0b4SJens Axboe 7166ed29b0b4SJens Axboe /* 7167ed29b0b4SJens Axboe * Ring fd has been registered via IORING_REGISTER_RING_FDS, we 7168ed29b0b4SJens Axboe * need only dereference our task private array to find it. 7169ed29b0b4SJens Axboe */ 7170ed29b0b4SJens Axboe if (flags & IORING_ENTER_REGISTERED_RING) { 7171ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 7172ed29b0b4SJens Axboe 7173ed29b0b4SJens Axboe if (!tctx || fd >= IO_RINGFD_REG_MAX) 7174ed29b0b4SJens Axboe return -EINVAL; 7175ed29b0b4SJens Axboe fd = array_index_nospec(fd, IO_RINGFD_REG_MAX); 7176ed29b0b4SJens Axboe f.file = tctx->registered_rings[fd]; 7177ed29b0b4SJens Axboe f.flags = 0; 7178ed29b0b4SJens Axboe } else { 7179ed29b0b4SJens Axboe f = fdget(fd); 7180ed29b0b4SJens Axboe } 7181ed29b0b4SJens Axboe 7182ed29b0b4SJens Axboe if (unlikely(!f.file)) 7183ed29b0b4SJens Axboe return -EBADF; 7184ed29b0b4SJens Axboe 7185ed29b0b4SJens Axboe ret = -EOPNOTSUPP; 7186e5550a14SJens Axboe if (unlikely(!io_is_uring_fops(f.file))) 7187ed29b0b4SJens Axboe goto out_fput; 7188ed29b0b4SJens Axboe 7189ed29b0b4SJens Axboe ret = -ENXIO; 7190ed29b0b4SJens Axboe ctx = f.file->private_data; 7191ed29b0b4SJens Axboe if (unlikely(!percpu_ref_tryget(&ctx->refs))) 7192ed29b0b4SJens Axboe goto out_fput; 7193ed29b0b4SJens Axboe 7194ed29b0b4SJens Axboe ret = -EBADFD; 7195ed29b0b4SJens Axboe if (unlikely(ctx->flags & IORING_SETUP_R_DISABLED)) 7196ed29b0b4SJens Axboe goto out; 7197ed29b0b4SJens Axboe 7198ed29b0b4SJens Axboe /* 7199ed29b0b4SJens Axboe * For SQ polling, the thread will do all submissions and completions. 7200ed29b0b4SJens Axboe * Just return the requested submit count, and wake the thread if 7201ed29b0b4SJens Axboe * we were asked to. 7202ed29b0b4SJens Axboe */ 7203ed29b0b4SJens Axboe ret = 0; 7204ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQPOLL) { 7205ed29b0b4SJens Axboe io_cqring_overflow_flush(ctx); 7206ed29b0b4SJens Axboe 7207ed29b0b4SJens Axboe if (unlikely(ctx->sq_data->thread == NULL)) { 7208ed29b0b4SJens Axboe ret = -EOWNERDEAD; 7209ed29b0b4SJens Axboe goto out; 7210ed29b0b4SJens Axboe } 7211ed29b0b4SJens Axboe if (flags & IORING_ENTER_SQ_WAKEUP) 7212ed29b0b4SJens Axboe wake_up(&ctx->sq_data->wait); 7213ed29b0b4SJens Axboe if (flags & IORING_ENTER_SQ_WAIT) { 7214ed29b0b4SJens Axboe ret = io_sqpoll_wait_sq(ctx); 7215ed29b0b4SJens Axboe if (ret) 7216ed29b0b4SJens Axboe goto out; 7217ed29b0b4SJens Axboe } 7218ed29b0b4SJens Axboe ret = to_submit; 7219ed29b0b4SJens Axboe } else if (to_submit) { 7220ed29b0b4SJens Axboe ret = io_uring_add_tctx_node(ctx); 7221ed29b0b4SJens Axboe if (unlikely(ret)) 7222ed29b0b4SJens Axboe goto out; 7223ed29b0b4SJens Axboe 7224ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 7225ed29b0b4SJens Axboe ret = io_submit_sqes(ctx, to_submit); 7226ed29b0b4SJens Axboe if (ret != to_submit) { 7227ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 7228ed29b0b4SJens Axboe goto out; 7229ed29b0b4SJens Axboe } 7230ed29b0b4SJens Axboe if ((flags & IORING_ENTER_GETEVENTS) && ctx->syscall_iopoll) 7231ed29b0b4SJens Axboe goto iopoll_locked; 7232ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 7233ed29b0b4SJens Axboe } 7234ed29b0b4SJens Axboe if (flags & IORING_ENTER_GETEVENTS) { 7235ed29b0b4SJens Axboe int ret2; 7236ed29b0b4SJens Axboe if (ctx->syscall_iopoll) { 7237ed29b0b4SJens Axboe /* 7238ed29b0b4SJens Axboe * We disallow the app entering submit/complete with 7239ed29b0b4SJens Axboe * polling, but we still need to lock the ring to 7240ed29b0b4SJens Axboe * prevent racing with polled issue that got punted to 7241ed29b0b4SJens Axboe * a workqueue. 7242ed29b0b4SJens Axboe */ 7243ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 7244ed29b0b4SJens Axboe iopoll_locked: 7245ed29b0b4SJens Axboe ret2 = io_validate_ext_arg(flags, argp, argsz); 7246ed29b0b4SJens Axboe if (likely(!ret2)) { 7247ed29b0b4SJens Axboe min_complete = min(min_complete, 7248ed29b0b4SJens Axboe ctx->cq_entries); 7249ed29b0b4SJens Axboe ret2 = io_iopoll_check(ctx, min_complete); 7250ed29b0b4SJens Axboe } 7251ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 7252ed29b0b4SJens Axboe } else { 7253ed29b0b4SJens Axboe const sigset_t __user *sig; 7254ed29b0b4SJens Axboe struct __kernel_timespec __user *ts; 7255ed29b0b4SJens Axboe 7256ed29b0b4SJens Axboe ret2 = io_get_ext_arg(flags, argp, &argsz, &ts, &sig); 7257ed29b0b4SJens Axboe if (likely(!ret2)) { 7258ed29b0b4SJens Axboe min_complete = min(min_complete, 7259ed29b0b4SJens Axboe ctx->cq_entries); 7260ed29b0b4SJens Axboe ret2 = io_cqring_wait(ctx, min_complete, sig, 7261ed29b0b4SJens Axboe argsz, ts); 7262ed29b0b4SJens Axboe } 7263ed29b0b4SJens Axboe } 7264ed29b0b4SJens Axboe 7265ed29b0b4SJens Axboe if (!ret) { 7266ed29b0b4SJens Axboe ret = ret2; 7267ed29b0b4SJens Axboe 7268ed29b0b4SJens Axboe /* 7269ed29b0b4SJens Axboe * EBADR indicates that one or more CQE were dropped. 7270ed29b0b4SJens Axboe * Once the user has been informed we can clear the bit 7271ed29b0b4SJens Axboe * as they are obviously ok with those drops. 7272ed29b0b4SJens Axboe */ 7273ed29b0b4SJens Axboe if (unlikely(ret2 == -EBADR)) 7274ed29b0b4SJens Axboe clear_bit(IO_CHECK_CQ_DROPPED_BIT, 7275ed29b0b4SJens Axboe &ctx->check_cq); 7276ed29b0b4SJens Axboe } 7277ed29b0b4SJens Axboe } 7278ed29b0b4SJens Axboe 7279ed29b0b4SJens Axboe out: 7280ed29b0b4SJens Axboe percpu_ref_put(&ctx->refs); 7281ed29b0b4SJens Axboe out_fput: 7282ed29b0b4SJens Axboe fdput(f); 7283ed29b0b4SJens Axboe return ret; 7284ed29b0b4SJens Axboe } 7285ed29b0b4SJens Axboe 7286ed29b0b4SJens Axboe static const struct file_operations io_uring_fops = { 7287ed29b0b4SJens Axboe .release = io_uring_release, 7288ed29b0b4SJens Axboe .mmap = io_uring_mmap, 7289ed29b0b4SJens Axboe #ifndef CONFIG_MMU 7290ed29b0b4SJens Axboe .get_unmapped_area = io_uring_nommu_get_unmapped_area, 7291ed29b0b4SJens Axboe .mmap_capabilities = io_uring_nommu_mmap_capabilities, 7292ed29b0b4SJens Axboe #endif 7293ed29b0b4SJens Axboe .poll = io_uring_poll, 7294ed29b0b4SJens Axboe #ifdef CONFIG_PROC_FS 7295ed29b0b4SJens Axboe .show_fdinfo = io_uring_show_fdinfo, 7296ed29b0b4SJens Axboe #endif 7297ed29b0b4SJens Axboe }; 7298ed29b0b4SJens Axboe 729992ac8beaSJens Axboe bool io_is_uring_fops(struct file *file) 730092ac8beaSJens Axboe { 730192ac8beaSJens Axboe return file->f_op == &io_uring_fops; 730292ac8beaSJens Axboe } 730392ac8beaSJens Axboe 7304ed29b0b4SJens Axboe static __cold int io_allocate_scq_urings(struct io_ring_ctx *ctx, 7305ed29b0b4SJens Axboe struct io_uring_params *p) 7306ed29b0b4SJens Axboe { 7307ed29b0b4SJens Axboe struct io_rings *rings; 7308ed29b0b4SJens Axboe size_t size, sq_array_offset; 7309ed29b0b4SJens Axboe 7310ed29b0b4SJens Axboe /* make sure these are sane, as we already accounted them */ 7311ed29b0b4SJens Axboe ctx->sq_entries = p->sq_entries; 7312ed29b0b4SJens Axboe ctx->cq_entries = p->cq_entries; 7313ed29b0b4SJens Axboe 7314ed29b0b4SJens Axboe size = rings_size(ctx, p->sq_entries, p->cq_entries, &sq_array_offset); 7315ed29b0b4SJens Axboe if (size == SIZE_MAX) 7316ed29b0b4SJens Axboe return -EOVERFLOW; 7317ed29b0b4SJens Axboe 7318ed29b0b4SJens Axboe rings = io_mem_alloc(size); 7319ed29b0b4SJens Axboe if (!rings) 7320ed29b0b4SJens Axboe return -ENOMEM; 7321ed29b0b4SJens Axboe 7322ed29b0b4SJens Axboe ctx->rings = rings; 7323ed29b0b4SJens Axboe ctx->sq_array = (u32 *)((char *)rings + sq_array_offset); 7324ed29b0b4SJens Axboe rings->sq_ring_mask = p->sq_entries - 1; 7325ed29b0b4SJens Axboe rings->cq_ring_mask = p->cq_entries - 1; 7326ed29b0b4SJens Axboe rings->sq_ring_entries = p->sq_entries; 7327ed29b0b4SJens Axboe rings->cq_ring_entries = p->cq_entries; 7328ed29b0b4SJens Axboe 7329ed29b0b4SJens Axboe if (p->flags & IORING_SETUP_SQE128) 7330ed29b0b4SJens Axboe size = array_size(2 * sizeof(struct io_uring_sqe), p->sq_entries); 7331ed29b0b4SJens Axboe else 7332ed29b0b4SJens Axboe size = array_size(sizeof(struct io_uring_sqe), p->sq_entries); 7333ed29b0b4SJens Axboe if (size == SIZE_MAX) { 7334ed29b0b4SJens Axboe io_mem_free(ctx->rings); 7335ed29b0b4SJens Axboe ctx->rings = NULL; 7336ed29b0b4SJens Axboe return -EOVERFLOW; 7337ed29b0b4SJens Axboe } 7338ed29b0b4SJens Axboe 7339ed29b0b4SJens Axboe ctx->sq_sqes = io_mem_alloc(size); 7340ed29b0b4SJens Axboe if (!ctx->sq_sqes) { 7341ed29b0b4SJens Axboe io_mem_free(ctx->rings); 7342ed29b0b4SJens Axboe ctx->rings = NULL; 7343ed29b0b4SJens Axboe return -ENOMEM; 7344ed29b0b4SJens Axboe } 7345ed29b0b4SJens Axboe 7346ed29b0b4SJens Axboe return 0; 7347ed29b0b4SJens Axboe } 7348ed29b0b4SJens Axboe 7349ed29b0b4SJens Axboe static int io_uring_install_fd(struct io_ring_ctx *ctx, struct file *file) 7350ed29b0b4SJens Axboe { 7351ed29b0b4SJens Axboe int ret, fd; 7352ed29b0b4SJens Axboe 7353ed29b0b4SJens Axboe fd = get_unused_fd_flags(O_RDWR | O_CLOEXEC); 7354ed29b0b4SJens Axboe if (fd < 0) 7355ed29b0b4SJens Axboe return fd; 7356ed29b0b4SJens Axboe 7357ed29b0b4SJens Axboe ret = io_uring_add_tctx_node(ctx); 7358ed29b0b4SJens Axboe if (ret) { 7359ed29b0b4SJens Axboe put_unused_fd(fd); 7360ed29b0b4SJens Axboe return ret; 7361ed29b0b4SJens Axboe } 7362ed29b0b4SJens Axboe fd_install(fd, file); 7363ed29b0b4SJens Axboe return fd; 7364ed29b0b4SJens Axboe } 7365ed29b0b4SJens Axboe 7366ed29b0b4SJens Axboe /* 7367ed29b0b4SJens Axboe * Allocate an anonymous fd, this is what constitutes the application 7368ed29b0b4SJens Axboe * visible backing of an io_uring instance. The application mmaps this 7369ed29b0b4SJens Axboe * fd to gain access to the SQ/CQ ring details. If UNIX sockets are enabled, 7370ed29b0b4SJens Axboe * we have to tie this fd to a socket for file garbage collection purposes. 7371ed29b0b4SJens Axboe */ 7372ed29b0b4SJens Axboe static struct file *io_uring_get_file(struct io_ring_ctx *ctx) 7373ed29b0b4SJens Axboe { 7374ed29b0b4SJens Axboe struct file *file; 7375ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 7376ed29b0b4SJens Axboe int ret; 7377ed29b0b4SJens Axboe 7378ed29b0b4SJens Axboe ret = sock_create_kern(&init_net, PF_UNIX, SOCK_RAW, IPPROTO_IP, 7379ed29b0b4SJens Axboe &ctx->ring_sock); 7380ed29b0b4SJens Axboe if (ret) 7381ed29b0b4SJens Axboe return ERR_PTR(ret); 7382ed29b0b4SJens Axboe #endif 7383ed29b0b4SJens Axboe 7384ed29b0b4SJens Axboe file = anon_inode_getfile_secure("[io_uring]", &io_uring_fops, ctx, 7385ed29b0b4SJens Axboe O_RDWR | O_CLOEXEC, NULL); 7386ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 7387ed29b0b4SJens Axboe if (IS_ERR(file)) { 7388ed29b0b4SJens Axboe sock_release(ctx->ring_sock); 7389ed29b0b4SJens Axboe ctx->ring_sock = NULL; 7390ed29b0b4SJens Axboe } else { 7391ed29b0b4SJens Axboe ctx->ring_sock->file = file; 7392ed29b0b4SJens Axboe } 7393ed29b0b4SJens Axboe #endif 7394ed29b0b4SJens Axboe return file; 7395ed29b0b4SJens Axboe } 7396ed29b0b4SJens Axboe 7397ed29b0b4SJens Axboe static __cold int io_uring_create(unsigned entries, struct io_uring_params *p, 7398ed29b0b4SJens Axboe struct io_uring_params __user *params) 7399ed29b0b4SJens Axboe { 7400ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 7401ed29b0b4SJens Axboe struct file *file; 7402ed29b0b4SJens Axboe int ret; 7403ed29b0b4SJens Axboe 7404ed29b0b4SJens Axboe if (!entries) 7405ed29b0b4SJens Axboe return -EINVAL; 7406ed29b0b4SJens Axboe if (entries > IORING_MAX_ENTRIES) { 7407ed29b0b4SJens Axboe if (!(p->flags & IORING_SETUP_CLAMP)) 7408ed29b0b4SJens Axboe return -EINVAL; 7409ed29b0b4SJens Axboe entries = IORING_MAX_ENTRIES; 7410ed29b0b4SJens Axboe } 7411ed29b0b4SJens Axboe 7412ed29b0b4SJens Axboe /* 7413ed29b0b4SJens Axboe * Use twice as many entries for the CQ ring. It's possible for the 7414ed29b0b4SJens Axboe * application to drive a higher depth than the size of the SQ ring, 7415ed29b0b4SJens Axboe * since the sqes are only used at submission time. This allows for 7416ed29b0b4SJens Axboe * some flexibility in overcommitting a bit. If the application has 7417ed29b0b4SJens Axboe * set IORING_SETUP_CQSIZE, it will have passed in the desired number 7418ed29b0b4SJens Axboe * of CQ ring entries manually. 7419ed29b0b4SJens Axboe */ 7420ed29b0b4SJens Axboe p->sq_entries = roundup_pow_of_two(entries); 7421ed29b0b4SJens Axboe if (p->flags & IORING_SETUP_CQSIZE) { 7422ed29b0b4SJens Axboe /* 7423ed29b0b4SJens Axboe * If IORING_SETUP_CQSIZE is set, we do the same roundup 7424ed29b0b4SJens Axboe * to a power-of-two, if it isn't already. We do NOT impose 7425ed29b0b4SJens Axboe * any cq vs sq ring sizing. 7426ed29b0b4SJens Axboe */ 7427ed29b0b4SJens Axboe if (!p->cq_entries) 7428ed29b0b4SJens Axboe return -EINVAL; 7429ed29b0b4SJens Axboe if (p->cq_entries > IORING_MAX_CQ_ENTRIES) { 7430ed29b0b4SJens Axboe if (!(p->flags & IORING_SETUP_CLAMP)) 7431ed29b0b4SJens Axboe return -EINVAL; 7432ed29b0b4SJens Axboe p->cq_entries = IORING_MAX_CQ_ENTRIES; 7433ed29b0b4SJens Axboe } 7434ed29b0b4SJens Axboe p->cq_entries = roundup_pow_of_two(p->cq_entries); 7435ed29b0b4SJens Axboe if (p->cq_entries < p->sq_entries) 7436ed29b0b4SJens Axboe return -EINVAL; 7437ed29b0b4SJens Axboe } else { 7438ed29b0b4SJens Axboe p->cq_entries = 2 * p->sq_entries; 7439ed29b0b4SJens Axboe } 7440ed29b0b4SJens Axboe 7441ed29b0b4SJens Axboe ctx = io_ring_ctx_alloc(p); 7442ed29b0b4SJens Axboe if (!ctx) 7443ed29b0b4SJens Axboe return -ENOMEM; 7444ed29b0b4SJens Axboe 7445ed29b0b4SJens Axboe /* 7446ed29b0b4SJens Axboe * When SETUP_IOPOLL and SETUP_SQPOLL are both enabled, user 7447ed29b0b4SJens Axboe * space applications don't need to do io completion events 7448ed29b0b4SJens Axboe * polling again, they can rely on io_sq_thread to do polling 7449ed29b0b4SJens Axboe * work, which can reduce cpu usage and uring_lock contention. 7450ed29b0b4SJens Axboe */ 7451ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL && 7452ed29b0b4SJens Axboe !(ctx->flags & IORING_SETUP_SQPOLL)) 7453ed29b0b4SJens Axboe ctx->syscall_iopoll = 1; 7454ed29b0b4SJens Axboe 7455ed29b0b4SJens Axboe ctx->compat = in_compat_syscall(); 7456ed29b0b4SJens Axboe if (!capable(CAP_IPC_LOCK)) 7457ed29b0b4SJens Axboe ctx->user = get_uid(current_user()); 7458ed29b0b4SJens Axboe 7459ed29b0b4SJens Axboe /* 7460ed29b0b4SJens Axboe * For SQPOLL, we just need a wakeup, always. For !SQPOLL, if 7461ed29b0b4SJens Axboe * COOP_TASKRUN is set, then IPIs are never needed by the app. 7462ed29b0b4SJens Axboe */ 7463ed29b0b4SJens Axboe ret = -EINVAL; 7464ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQPOLL) { 7465ed29b0b4SJens Axboe /* IPI related flags don't make sense with SQPOLL */ 7466ed29b0b4SJens Axboe if (ctx->flags & (IORING_SETUP_COOP_TASKRUN | 7467ed29b0b4SJens Axboe IORING_SETUP_TASKRUN_FLAG)) 7468ed29b0b4SJens Axboe goto err; 7469ed29b0b4SJens Axboe ctx->notify_method = TWA_SIGNAL_NO_IPI; 7470ed29b0b4SJens Axboe } else if (ctx->flags & IORING_SETUP_COOP_TASKRUN) { 7471ed29b0b4SJens Axboe ctx->notify_method = TWA_SIGNAL_NO_IPI; 7472ed29b0b4SJens Axboe } else { 7473ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_TASKRUN_FLAG) 7474ed29b0b4SJens Axboe goto err; 7475ed29b0b4SJens Axboe ctx->notify_method = TWA_SIGNAL; 7476ed29b0b4SJens Axboe } 7477ed29b0b4SJens Axboe 7478ed29b0b4SJens Axboe /* 7479ed29b0b4SJens Axboe * This is just grabbed for accounting purposes. When a process exits, 7480ed29b0b4SJens Axboe * the mm is exited and dropped before the files, hence we need to hang 7481ed29b0b4SJens Axboe * on to this mm purely for the purposes of being able to unaccount 7482ed29b0b4SJens Axboe * memory (locked/pinned vm). It's not used for anything else. 7483ed29b0b4SJens Axboe */ 7484ed29b0b4SJens Axboe mmgrab(current->mm); 7485ed29b0b4SJens Axboe ctx->mm_account = current->mm; 7486ed29b0b4SJens Axboe 7487ed29b0b4SJens Axboe ret = io_allocate_scq_urings(ctx, p); 7488ed29b0b4SJens Axboe if (ret) 7489ed29b0b4SJens Axboe goto err; 7490ed29b0b4SJens Axboe 7491ed29b0b4SJens Axboe ret = io_sq_offload_create(ctx, p); 7492ed29b0b4SJens Axboe if (ret) 7493ed29b0b4SJens Axboe goto err; 7494ed29b0b4SJens Axboe /* always set a rsrc node */ 7495ed29b0b4SJens Axboe ret = io_rsrc_node_switch_start(ctx); 7496ed29b0b4SJens Axboe if (ret) 7497ed29b0b4SJens Axboe goto err; 7498ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, NULL); 7499ed29b0b4SJens Axboe 7500ed29b0b4SJens Axboe memset(&p->sq_off, 0, sizeof(p->sq_off)); 7501ed29b0b4SJens Axboe p->sq_off.head = offsetof(struct io_rings, sq.head); 7502ed29b0b4SJens Axboe p->sq_off.tail = offsetof(struct io_rings, sq.tail); 7503ed29b0b4SJens Axboe p->sq_off.ring_mask = offsetof(struct io_rings, sq_ring_mask); 7504ed29b0b4SJens Axboe p->sq_off.ring_entries = offsetof(struct io_rings, sq_ring_entries); 7505ed29b0b4SJens Axboe p->sq_off.flags = offsetof(struct io_rings, sq_flags); 7506ed29b0b4SJens Axboe p->sq_off.dropped = offsetof(struct io_rings, sq_dropped); 7507ed29b0b4SJens Axboe p->sq_off.array = (char *)ctx->sq_array - (char *)ctx->rings; 7508ed29b0b4SJens Axboe 7509ed29b0b4SJens Axboe memset(&p->cq_off, 0, sizeof(p->cq_off)); 7510ed29b0b4SJens Axboe p->cq_off.head = offsetof(struct io_rings, cq.head); 7511ed29b0b4SJens Axboe p->cq_off.tail = offsetof(struct io_rings, cq.tail); 7512ed29b0b4SJens Axboe p->cq_off.ring_mask = offsetof(struct io_rings, cq_ring_mask); 7513ed29b0b4SJens Axboe p->cq_off.ring_entries = offsetof(struct io_rings, cq_ring_entries); 7514ed29b0b4SJens Axboe p->cq_off.overflow = offsetof(struct io_rings, cq_overflow); 7515ed29b0b4SJens Axboe p->cq_off.cqes = offsetof(struct io_rings, cqes); 7516ed29b0b4SJens Axboe p->cq_off.flags = offsetof(struct io_rings, cq_flags); 7517ed29b0b4SJens Axboe 7518ed29b0b4SJens Axboe p->features = IORING_FEAT_SINGLE_MMAP | IORING_FEAT_NODROP | 7519ed29b0b4SJens Axboe IORING_FEAT_SUBMIT_STABLE | IORING_FEAT_RW_CUR_POS | 7520ed29b0b4SJens Axboe IORING_FEAT_CUR_PERSONALITY | IORING_FEAT_FAST_POLL | 7521ed29b0b4SJens Axboe IORING_FEAT_POLL_32BITS | IORING_FEAT_SQPOLL_NONFIXED | 7522ed29b0b4SJens Axboe IORING_FEAT_EXT_ARG | IORING_FEAT_NATIVE_WORKERS | 7523ed29b0b4SJens Axboe IORING_FEAT_RSRC_TAGS | IORING_FEAT_CQE_SKIP | 7524ed29b0b4SJens Axboe IORING_FEAT_LINKED_FILE; 7525ed29b0b4SJens Axboe 7526ed29b0b4SJens Axboe if (copy_to_user(params, p, sizeof(*p))) { 7527ed29b0b4SJens Axboe ret = -EFAULT; 7528ed29b0b4SJens Axboe goto err; 7529ed29b0b4SJens Axboe } 7530ed29b0b4SJens Axboe 7531ed29b0b4SJens Axboe file = io_uring_get_file(ctx); 7532ed29b0b4SJens Axboe if (IS_ERR(file)) { 7533ed29b0b4SJens Axboe ret = PTR_ERR(file); 7534ed29b0b4SJens Axboe goto err; 7535ed29b0b4SJens Axboe } 7536ed29b0b4SJens Axboe 7537ed29b0b4SJens Axboe /* 7538ed29b0b4SJens Axboe * Install ring fd as the very last thing, so we don't risk someone 7539ed29b0b4SJens Axboe * having closed it before we finish setup 7540ed29b0b4SJens Axboe */ 7541ed29b0b4SJens Axboe ret = io_uring_install_fd(ctx, file); 7542ed29b0b4SJens Axboe if (ret < 0) { 7543ed29b0b4SJens Axboe /* fput will clean it up */ 7544ed29b0b4SJens Axboe fput(file); 7545ed29b0b4SJens Axboe return ret; 7546ed29b0b4SJens Axboe } 7547ed29b0b4SJens Axboe 7548ed29b0b4SJens Axboe trace_io_uring_create(ret, ctx, p->sq_entries, p->cq_entries, p->flags); 7549ed29b0b4SJens Axboe return ret; 7550ed29b0b4SJens Axboe err: 7551ed29b0b4SJens Axboe io_ring_ctx_wait_and_kill(ctx); 7552ed29b0b4SJens Axboe return ret; 7553ed29b0b4SJens Axboe } 7554ed29b0b4SJens Axboe 7555ed29b0b4SJens Axboe /* 7556ed29b0b4SJens Axboe * Sets up an aio uring context, and returns the fd. Applications asks for a 7557ed29b0b4SJens Axboe * ring size, we return the actual sq/cq ring sizes (among other things) in the 7558ed29b0b4SJens Axboe * params structure passed in. 7559ed29b0b4SJens Axboe */ 7560ed29b0b4SJens Axboe static long io_uring_setup(u32 entries, struct io_uring_params __user *params) 7561ed29b0b4SJens Axboe { 7562ed29b0b4SJens Axboe struct io_uring_params p; 7563ed29b0b4SJens Axboe int i; 7564ed29b0b4SJens Axboe 7565ed29b0b4SJens Axboe if (copy_from_user(&p, params, sizeof(p))) 7566ed29b0b4SJens Axboe return -EFAULT; 7567ed29b0b4SJens Axboe for (i = 0; i < ARRAY_SIZE(p.resv); i++) { 7568ed29b0b4SJens Axboe if (p.resv[i]) 7569ed29b0b4SJens Axboe return -EINVAL; 7570ed29b0b4SJens Axboe } 7571ed29b0b4SJens Axboe 7572ed29b0b4SJens Axboe if (p.flags & ~(IORING_SETUP_IOPOLL | IORING_SETUP_SQPOLL | 7573ed29b0b4SJens Axboe IORING_SETUP_SQ_AFF | IORING_SETUP_CQSIZE | 7574ed29b0b4SJens Axboe IORING_SETUP_CLAMP | IORING_SETUP_ATTACH_WQ | 7575ed29b0b4SJens Axboe IORING_SETUP_R_DISABLED | IORING_SETUP_SUBMIT_ALL | 7576ed29b0b4SJens Axboe IORING_SETUP_COOP_TASKRUN | IORING_SETUP_TASKRUN_FLAG | 7577ed29b0b4SJens Axboe IORING_SETUP_SQE128 | IORING_SETUP_CQE32)) 7578ed29b0b4SJens Axboe return -EINVAL; 7579ed29b0b4SJens Axboe 7580ed29b0b4SJens Axboe return io_uring_create(entries, &p, params); 7581ed29b0b4SJens Axboe } 7582ed29b0b4SJens Axboe 7583ed29b0b4SJens Axboe SYSCALL_DEFINE2(io_uring_setup, u32, entries, 7584ed29b0b4SJens Axboe struct io_uring_params __user *, params) 7585ed29b0b4SJens Axboe { 7586ed29b0b4SJens Axboe return io_uring_setup(entries, params); 7587ed29b0b4SJens Axboe } 7588ed29b0b4SJens Axboe 7589ed29b0b4SJens Axboe static __cold int io_probe(struct io_ring_ctx *ctx, void __user *arg, 7590ed29b0b4SJens Axboe unsigned nr_args) 7591ed29b0b4SJens Axboe { 7592ed29b0b4SJens Axboe struct io_uring_probe *p; 7593ed29b0b4SJens Axboe size_t size; 7594ed29b0b4SJens Axboe int i, ret; 7595ed29b0b4SJens Axboe 7596ed29b0b4SJens Axboe size = struct_size(p, ops, nr_args); 7597ed29b0b4SJens Axboe if (size == SIZE_MAX) 7598ed29b0b4SJens Axboe return -EOVERFLOW; 7599ed29b0b4SJens Axboe p = kzalloc(size, GFP_KERNEL); 7600ed29b0b4SJens Axboe if (!p) 7601ed29b0b4SJens Axboe return -ENOMEM; 7602ed29b0b4SJens Axboe 7603ed29b0b4SJens Axboe ret = -EFAULT; 7604ed29b0b4SJens Axboe if (copy_from_user(p, arg, size)) 7605ed29b0b4SJens Axboe goto out; 7606ed29b0b4SJens Axboe ret = -EINVAL; 7607ed29b0b4SJens Axboe if (memchr_inv(p, 0, size)) 7608ed29b0b4SJens Axboe goto out; 7609ed29b0b4SJens Axboe 7610ed29b0b4SJens Axboe p->last_op = IORING_OP_LAST - 1; 7611ed29b0b4SJens Axboe if (nr_args > IORING_OP_LAST) 7612ed29b0b4SJens Axboe nr_args = IORING_OP_LAST; 7613ed29b0b4SJens Axboe 7614ed29b0b4SJens Axboe for (i = 0; i < nr_args; i++) { 7615ed29b0b4SJens Axboe p->ops[i].op = i; 7616ed29b0b4SJens Axboe if (!io_op_defs[i].not_supported) 7617ed29b0b4SJens Axboe p->ops[i].flags = IO_URING_OP_SUPPORTED; 7618ed29b0b4SJens Axboe } 7619ed29b0b4SJens Axboe p->ops_len = i; 7620ed29b0b4SJens Axboe 7621ed29b0b4SJens Axboe ret = 0; 7622ed29b0b4SJens Axboe if (copy_to_user(arg, p, size)) 7623ed29b0b4SJens Axboe ret = -EFAULT; 7624ed29b0b4SJens Axboe out: 7625ed29b0b4SJens Axboe kfree(p); 7626ed29b0b4SJens Axboe return ret; 7627ed29b0b4SJens Axboe } 7628ed29b0b4SJens Axboe 7629ed29b0b4SJens Axboe static int io_register_personality(struct io_ring_ctx *ctx) 7630ed29b0b4SJens Axboe { 7631ed29b0b4SJens Axboe const struct cred *creds; 7632ed29b0b4SJens Axboe u32 id; 7633ed29b0b4SJens Axboe int ret; 7634ed29b0b4SJens Axboe 7635ed29b0b4SJens Axboe creds = get_current_cred(); 7636ed29b0b4SJens Axboe 7637ed29b0b4SJens Axboe ret = xa_alloc_cyclic(&ctx->personalities, &id, (void *)creds, 7638ed29b0b4SJens Axboe XA_LIMIT(0, USHRT_MAX), &ctx->pers_next, GFP_KERNEL); 7639ed29b0b4SJens Axboe if (ret < 0) { 7640ed29b0b4SJens Axboe put_cred(creds); 7641ed29b0b4SJens Axboe return ret; 7642ed29b0b4SJens Axboe } 7643ed29b0b4SJens Axboe return id; 7644ed29b0b4SJens Axboe } 7645ed29b0b4SJens Axboe 7646ed29b0b4SJens Axboe static __cold int io_register_restrictions(struct io_ring_ctx *ctx, 7647ed29b0b4SJens Axboe void __user *arg, unsigned int nr_args) 7648ed29b0b4SJens Axboe { 7649ed29b0b4SJens Axboe struct io_uring_restriction *res; 7650ed29b0b4SJens Axboe size_t size; 7651ed29b0b4SJens Axboe int i, ret; 7652ed29b0b4SJens Axboe 7653ed29b0b4SJens Axboe /* Restrictions allowed only if rings started disabled */ 7654ed29b0b4SJens Axboe if (!(ctx->flags & IORING_SETUP_R_DISABLED)) 7655ed29b0b4SJens Axboe return -EBADFD; 7656ed29b0b4SJens Axboe 7657ed29b0b4SJens Axboe /* We allow only a single restrictions registration */ 7658ed29b0b4SJens Axboe if (ctx->restrictions.registered) 7659ed29b0b4SJens Axboe return -EBUSY; 7660ed29b0b4SJens Axboe 7661ed29b0b4SJens Axboe if (!arg || nr_args > IORING_MAX_RESTRICTIONS) 7662ed29b0b4SJens Axboe return -EINVAL; 7663ed29b0b4SJens Axboe 7664ed29b0b4SJens Axboe size = array_size(nr_args, sizeof(*res)); 7665ed29b0b4SJens Axboe if (size == SIZE_MAX) 7666ed29b0b4SJens Axboe return -EOVERFLOW; 7667ed29b0b4SJens Axboe 7668ed29b0b4SJens Axboe res = memdup_user(arg, size); 7669ed29b0b4SJens Axboe if (IS_ERR(res)) 7670ed29b0b4SJens Axboe return PTR_ERR(res); 7671ed29b0b4SJens Axboe 7672ed29b0b4SJens Axboe ret = 0; 7673ed29b0b4SJens Axboe 7674ed29b0b4SJens Axboe for (i = 0; i < nr_args; i++) { 7675ed29b0b4SJens Axboe switch (res[i].opcode) { 7676ed29b0b4SJens Axboe case IORING_RESTRICTION_REGISTER_OP: 7677ed29b0b4SJens Axboe if (res[i].register_op >= IORING_REGISTER_LAST) { 7678ed29b0b4SJens Axboe ret = -EINVAL; 7679ed29b0b4SJens Axboe goto out; 7680ed29b0b4SJens Axboe } 7681ed29b0b4SJens Axboe 7682ed29b0b4SJens Axboe __set_bit(res[i].register_op, 7683ed29b0b4SJens Axboe ctx->restrictions.register_op); 7684ed29b0b4SJens Axboe break; 7685ed29b0b4SJens Axboe case IORING_RESTRICTION_SQE_OP: 7686ed29b0b4SJens Axboe if (res[i].sqe_op >= IORING_OP_LAST) { 7687ed29b0b4SJens Axboe ret = -EINVAL; 7688ed29b0b4SJens Axboe goto out; 7689ed29b0b4SJens Axboe } 7690ed29b0b4SJens Axboe 7691ed29b0b4SJens Axboe __set_bit(res[i].sqe_op, ctx->restrictions.sqe_op); 7692ed29b0b4SJens Axboe break; 7693ed29b0b4SJens Axboe case IORING_RESTRICTION_SQE_FLAGS_ALLOWED: 7694ed29b0b4SJens Axboe ctx->restrictions.sqe_flags_allowed = res[i].sqe_flags; 7695ed29b0b4SJens Axboe break; 7696ed29b0b4SJens Axboe case IORING_RESTRICTION_SQE_FLAGS_REQUIRED: 7697ed29b0b4SJens Axboe ctx->restrictions.sqe_flags_required = res[i].sqe_flags; 7698ed29b0b4SJens Axboe break; 7699ed29b0b4SJens Axboe default: 7700ed29b0b4SJens Axboe ret = -EINVAL; 7701ed29b0b4SJens Axboe goto out; 7702ed29b0b4SJens Axboe } 7703ed29b0b4SJens Axboe } 7704ed29b0b4SJens Axboe 7705ed29b0b4SJens Axboe out: 7706ed29b0b4SJens Axboe /* Reset all restrictions if an error happened */ 7707ed29b0b4SJens Axboe if (ret != 0) 7708ed29b0b4SJens Axboe memset(&ctx->restrictions, 0, sizeof(ctx->restrictions)); 7709ed29b0b4SJens Axboe else 7710ed29b0b4SJens Axboe ctx->restrictions.registered = true; 7711ed29b0b4SJens Axboe 7712ed29b0b4SJens Axboe kfree(res); 7713ed29b0b4SJens Axboe return ret; 7714ed29b0b4SJens Axboe } 7715ed29b0b4SJens Axboe 7716ed29b0b4SJens Axboe static int io_register_enable_rings(struct io_ring_ctx *ctx) 7717ed29b0b4SJens Axboe { 7718ed29b0b4SJens Axboe if (!(ctx->flags & IORING_SETUP_R_DISABLED)) 7719ed29b0b4SJens Axboe return -EBADFD; 7720ed29b0b4SJens Axboe 7721ed29b0b4SJens Axboe if (ctx->restrictions.registered) 7722ed29b0b4SJens Axboe ctx->restricted = 1; 7723ed29b0b4SJens Axboe 7724ed29b0b4SJens Axboe ctx->flags &= ~IORING_SETUP_R_DISABLED; 7725ed29b0b4SJens Axboe if (ctx->sq_data && wq_has_sleeper(&ctx->sq_data->wait)) 7726ed29b0b4SJens Axboe wake_up(&ctx->sq_data->wait); 7727ed29b0b4SJens Axboe return 0; 7728ed29b0b4SJens Axboe } 7729ed29b0b4SJens Axboe 7730ed29b0b4SJens Axboe static int __io_register_rsrc_update(struct io_ring_ctx *ctx, unsigned type, 7731ed29b0b4SJens Axboe struct io_uring_rsrc_update2 *up, 7732ed29b0b4SJens Axboe unsigned nr_args) 7733ed29b0b4SJens Axboe { 7734ed29b0b4SJens Axboe __u32 tmp; 7735ed29b0b4SJens Axboe int err; 7736ed29b0b4SJens Axboe 7737ed29b0b4SJens Axboe if (check_add_overflow(up->offset, nr_args, &tmp)) 7738ed29b0b4SJens Axboe return -EOVERFLOW; 7739ed29b0b4SJens Axboe err = io_rsrc_node_switch_start(ctx); 7740ed29b0b4SJens Axboe if (err) 7741ed29b0b4SJens Axboe return err; 7742ed29b0b4SJens Axboe 7743ed29b0b4SJens Axboe switch (type) { 7744ed29b0b4SJens Axboe case IORING_RSRC_FILE: 7745ed29b0b4SJens Axboe return __io_sqe_files_update(ctx, up, nr_args); 7746ed29b0b4SJens Axboe case IORING_RSRC_BUFFER: 7747ed29b0b4SJens Axboe return __io_sqe_buffers_update(ctx, up, nr_args); 7748ed29b0b4SJens Axboe } 7749ed29b0b4SJens Axboe return -EINVAL; 7750ed29b0b4SJens Axboe } 7751ed29b0b4SJens Axboe 7752ed29b0b4SJens Axboe static int io_register_files_update(struct io_ring_ctx *ctx, void __user *arg, 7753ed29b0b4SJens Axboe unsigned nr_args) 7754ed29b0b4SJens Axboe { 7755ed29b0b4SJens Axboe struct io_uring_rsrc_update2 up; 7756ed29b0b4SJens Axboe 7757ed29b0b4SJens Axboe if (!nr_args) 7758ed29b0b4SJens Axboe return -EINVAL; 7759ed29b0b4SJens Axboe memset(&up, 0, sizeof(up)); 7760ed29b0b4SJens Axboe if (copy_from_user(&up, arg, sizeof(struct io_uring_rsrc_update))) 7761ed29b0b4SJens Axboe return -EFAULT; 7762ed29b0b4SJens Axboe if (up.resv || up.resv2) 7763ed29b0b4SJens Axboe return -EINVAL; 7764ed29b0b4SJens Axboe return __io_register_rsrc_update(ctx, IORING_RSRC_FILE, &up, nr_args); 7765ed29b0b4SJens Axboe } 7766ed29b0b4SJens Axboe 7767ed29b0b4SJens Axboe static int io_register_rsrc_update(struct io_ring_ctx *ctx, void __user *arg, 7768ed29b0b4SJens Axboe unsigned size, unsigned type) 7769ed29b0b4SJens Axboe { 7770ed29b0b4SJens Axboe struct io_uring_rsrc_update2 up; 7771ed29b0b4SJens Axboe 7772ed29b0b4SJens Axboe if (size != sizeof(up)) 7773ed29b0b4SJens Axboe return -EINVAL; 7774ed29b0b4SJens Axboe if (copy_from_user(&up, arg, sizeof(up))) 7775ed29b0b4SJens Axboe return -EFAULT; 7776ed29b0b4SJens Axboe if (!up.nr || up.resv || up.resv2) 7777ed29b0b4SJens Axboe return -EINVAL; 7778ed29b0b4SJens Axboe return __io_register_rsrc_update(ctx, type, &up, up.nr); 7779ed29b0b4SJens Axboe } 7780ed29b0b4SJens Axboe 7781ed29b0b4SJens Axboe static __cold int io_register_rsrc(struct io_ring_ctx *ctx, void __user *arg, 7782ed29b0b4SJens Axboe unsigned int size, unsigned int type) 7783ed29b0b4SJens Axboe { 7784ed29b0b4SJens Axboe struct io_uring_rsrc_register rr; 7785ed29b0b4SJens Axboe 7786ed29b0b4SJens Axboe /* keep it extendible */ 7787ed29b0b4SJens Axboe if (size != sizeof(rr)) 7788ed29b0b4SJens Axboe return -EINVAL; 7789ed29b0b4SJens Axboe 7790ed29b0b4SJens Axboe memset(&rr, 0, sizeof(rr)); 7791ed29b0b4SJens Axboe if (copy_from_user(&rr, arg, size)) 7792ed29b0b4SJens Axboe return -EFAULT; 7793ed29b0b4SJens Axboe if (!rr.nr || rr.resv2) 7794ed29b0b4SJens Axboe return -EINVAL; 7795ed29b0b4SJens Axboe if (rr.flags & ~IORING_RSRC_REGISTER_SPARSE) 7796ed29b0b4SJens Axboe return -EINVAL; 7797ed29b0b4SJens Axboe 7798ed29b0b4SJens Axboe switch (type) { 7799ed29b0b4SJens Axboe case IORING_RSRC_FILE: 7800ed29b0b4SJens Axboe if (rr.flags & IORING_RSRC_REGISTER_SPARSE && rr.data) 7801ed29b0b4SJens Axboe break; 7802ed29b0b4SJens Axboe return io_sqe_files_register(ctx, u64_to_user_ptr(rr.data), 7803ed29b0b4SJens Axboe rr.nr, u64_to_user_ptr(rr.tags)); 7804ed29b0b4SJens Axboe case IORING_RSRC_BUFFER: 7805ed29b0b4SJens Axboe if (rr.flags & IORING_RSRC_REGISTER_SPARSE && rr.data) 7806ed29b0b4SJens Axboe break; 7807ed29b0b4SJens Axboe return io_sqe_buffers_register(ctx, u64_to_user_ptr(rr.data), 7808ed29b0b4SJens Axboe rr.nr, u64_to_user_ptr(rr.tags)); 7809ed29b0b4SJens Axboe } 7810ed29b0b4SJens Axboe return -EINVAL; 7811ed29b0b4SJens Axboe } 7812ed29b0b4SJens Axboe 7813ed29b0b4SJens Axboe static __cold int io_register_iowq_aff(struct io_ring_ctx *ctx, 7814ed29b0b4SJens Axboe void __user *arg, unsigned len) 7815ed29b0b4SJens Axboe { 7816ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 7817ed29b0b4SJens Axboe cpumask_var_t new_mask; 7818ed29b0b4SJens Axboe int ret; 7819ed29b0b4SJens Axboe 7820ed29b0b4SJens Axboe if (!tctx || !tctx->io_wq) 7821ed29b0b4SJens Axboe return -EINVAL; 7822ed29b0b4SJens Axboe 7823ed29b0b4SJens Axboe if (!alloc_cpumask_var(&new_mask, GFP_KERNEL)) 7824ed29b0b4SJens Axboe return -ENOMEM; 7825ed29b0b4SJens Axboe 7826ed29b0b4SJens Axboe cpumask_clear(new_mask); 7827ed29b0b4SJens Axboe if (len > cpumask_size()) 7828ed29b0b4SJens Axboe len = cpumask_size(); 7829ed29b0b4SJens Axboe 7830ed29b0b4SJens Axboe if (in_compat_syscall()) { 7831ed29b0b4SJens Axboe ret = compat_get_bitmap(cpumask_bits(new_mask), 7832ed29b0b4SJens Axboe (const compat_ulong_t __user *)arg, 7833ed29b0b4SJens Axboe len * 8 /* CHAR_BIT */); 7834ed29b0b4SJens Axboe } else { 7835ed29b0b4SJens Axboe ret = copy_from_user(new_mask, arg, len); 7836ed29b0b4SJens Axboe } 7837ed29b0b4SJens Axboe 7838ed29b0b4SJens Axboe if (ret) { 7839ed29b0b4SJens Axboe free_cpumask_var(new_mask); 7840ed29b0b4SJens Axboe return -EFAULT; 7841ed29b0b4SJens Axboe } 7842ed29b0b4SJens Axboe 7843ed29b0b4SJens Axboe ret = io_wq_cpu_affinity(tctx->io_wq, new_mask); 7844ed29b0b4SJens Axboe free_cpumask_var(new_mask); 7845ed29b0b4SJens Axboe return ret; 7846ed29b0b4SJens Axboe } 7847ed29b0b4SJens Axboe 7848ed29b0b4SJens Axboe static __cold int io_unregister_iowq_aff(struct io_ring_ctx *ctx) 7849ed29b0b4SJens Axboe { 7850ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 7851ed29b0b4SJens Axboe 7852ed29b0b4SJens Axboe if (!tctx || !tctx->io_wq) 7853ed29b0b4SJens Axboe return -EINVAL; 7854ed29b0b4SJens Axboe 7855ed29b0b4SJens Axboe return io_wq_cpu_affinity(tctx->io_wq, NULL); 7856ed29b0b4SJens Axboe } 7857ed29b0b4SJens Axboe 7858ed29b0b4SJens Axboe static __cold int io_register_iowq_max_workers(struct io_ring_ctx *ctx, 7859ed29b0b4SJens Axboe void __user *arg) 7860ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 7861ed29b0b4SJens Axboe { 7862ed29b0b4SJens Axboe struct io_tctx_node *node; 7863ed29b0b4SJens Axboe struct io_uring_task *tctx = NULL; 7864ed29b0b4SJens Axboe struct io_sq_data *sqd = NULL; 7865ed29b0b4SJens Axboe __u32 new_count[2]; 7866ed29b0b4SJens Axboe int i, ret; 7867ed29b0b4SJens Axboe 7868ed29b0b4SJens Axboe if (copy_from_user(new_count, arg, sizeof(new_count))) 7869ed29b0b4SJens Axboe return -EFAULT; 7870ed29b0b4SJens Axboe for (i = 0; i < ARRAY_SIZE(new_count); i++) 7871ed29b0b4SJens Axboe if (new_count[i] > INT_MAX) 7872ed29b0b4SJens Axboe return -EINVAL; 7873ed29b0b4SJens Axboe 7874ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQPOLL) { 7875ed29b0b4SJens Axboe sqd = ctx->sq_data; 7876ed29b0b4SJens Axboe if (sqd) { 7877ed29b0b4SJens Axboe /* 7878ed29b0b4SJens Axboe * Observe the correct sqd->lock -> ctx->uring_lock 7879ed29b0b4SJens Axboe * ordering. Fine to drop uring_lock here, we hold 7880ed29b0b4SJens Axboe * a ref to the ctx. 7881ed29b0b4SJens Axboe */ 7882ed29b0b4SJens Axboe refcount_inc(&sqd->refs); 7883ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 7884ed29b0b4SJens Axboe mutex_lock(&sqd->lock); 7885ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 7886ed29b0b4SJens Axboe if (sqd->thread) 7887ed29b0b4SJens Axboe tctx = sqd->thread->io_uring; 7888ed29b0b4SJens Axboe } 7889ed29b0b4SJens Axboe } else { 7890ed29b0b4SJens Axboe tctx = current->io_uring; 7891ed29b0b4SJens Axboe } 7892ed29b0b4SJens Axboe 7893ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(new_count) != sizeof(ctx->iowq_limits)); 7894ed29b0b4SJens Axboe 7895ed29b0b4SJens Axboe for (i = 0; i < ARRAY_SIZE(new_count); i++) 7896ed29b0b4SJens Axboe if (new_count[i]) 7897ed29b0b4SJens Axboe ctx->iowq_limits[i] = new_count[i]; 7898ed29b0b4SJens Axboe ctx->iowq_limits_set = true; 7899ed29b0b4SJens Axboe 7900ed29b0b4SJens Axboe if (tctx && tctx->io_wq) { 7901ed29b0b4SJens Axboe ret = io_wq_max_workers(tctx->io_wq, new_count); 7902ed29b0b4SJens Axboe if (ret) 7903ed29b0b4SJens Axboe goto err; 7904ed29b0b4SJens Axboe } else { 7905ed29b0b4SJens Axboe memset(new_count, 0, sizeof(new_count)); 7906ed29b0b4SJens Axboe } 7907ed29b0b4SJens Axboe 7908ed29b0b4SJens Axboe if (sqd) { 7909ed29b0b4SJens Axboe mutex_unlock(&sqd->lock); 7910ed29b0b4SJens Axboe io_put_sq_data(sqd); 7911ed29b0b4SJens Axboe } 7912ed29b0b4SJens Axboe 7913ed29b0b4SJens Axboe if (copy_to_user(arg, new_count, sizeof(new_count))) 7914ed29b0b4SJens Axboe return -EFAULT; 7915ed29b0b4SJens Axboe 7916ed29b0b4SJens Axboe /* that's it for SQPOLL, only the SQPOLL task creates requests */ 7917ed29b0b4SJens Axboe if (sqd) 7918ed29b0b4SJens Axboe return 0; 7919ed29b0b4SJens Axboe 7920ed29b0b4SJens Axboe /* now propagate the restriction to all registered users */ 7921ed29b0b4SJens Axboe list_for_each_entry(node, &ctx->tctx_list, ctx_node) { 7922ed29b0b4SJens Axboe struct io_uring_task *tctx = node->task->io_uring; 7923ed29b0b4SJens Axboe 7924ed29b0b4SJens Axboe if (WARN_ON_ONCE(!tctx->io_wq)) 7925ed29b0b4SJens Axboe continue; 7926ed29b0b4SJens Axboe 7927ed29b0b4SJens Axboe for (i = 0; i < ARRAY_SIZE(new_count); i++) 7928ed29b0b4SJens Axboe new_count[i] = ctx->iowq_limits[i]; 7929ed29b0b4SJens Axboe /* ignore errors, it always returns zero anyway */ 7930ed29b0b4SJens Axboe (void)io_wq_max_workers(tctx->io_wq, new_count); 7931ed29b0b4SJens Axboe } 7932ed29b0b4SJens Axboe return 0; 7933ed29b0b4SJens Axboe err: 7934ed29b0b4SJens Axboe if (sqd) { 7935ed29b0b4SJens Axboe mutex_unlock(&sqd->lock); 7936ed29b0b4SJens Axboe io_put_sq_data(sqd); 7937ed29b0b4SJens Axboe } 7938ed29b0b4SJens Axboe return ret; 7939ed29b0b4SJens Axboe } 7940ed29b0b4SJens Axboe 7941ed29b0b4SJens Axboe static int io_register_pbuf_ring(struct io_ring_ctx *ctx, void __user *arg) 7942ed29b0b4SJens Axboe { 7943ed29b0b4SJens Axboe struct io_uring_buf_ring *br; 7944ed29b0b4SJens Axboe struct io_uring_buf_reg reg; 7945ed29b0b4SJens Axboe struct io_buffer_list *bl, *free_bl = NULL; 7946ed29b0b4SJens Axboe struct page **pages; 7947ed29b0b4SJens Axboe int nr_pages; 7948ed29b0b4SJens Axboe 7949ed29b0b4SJens Axboe if (copy_from_user(®, arg, sizeof(reg))) 7950ed29b0b4SJens Axboe return -EFAULT; 7951ed29b0b4SJens Axboe 7952ed29b0b4SJens Axboe if (reg.pad || reg.resv[0] || reg.resv[1] || reg.resv[2]) 7953ed29b0b4SJens Axboe return -EINVAL; 7954ed29b0b4SJens Axboe if (!reg.ring_addr) 7955ed29b0b4SJens Axboe return -EFAULT; 7956ed29b0b4SJens Axboe if (reg.ring_addr & ~PAGE_MASK) 7957ed29b0b4SJens Axboe return -EINVAL; 7958ed29b0b4SJens Axboe if (!is_power_of_2(reg.ring_entries)) 7959ed29b0b4SJens Axboe return -EINVAL; 7960ed29b0b4SJens Axboe 7961ed29b0b4SJens Axboe /* cannot disambiguate full vs empty due to head/tail size */ 7962ed29b0b4SJens Axboe if (reg.ring_entries >= 65536) 7963ed29b0b4SJens Axboe return -EINVAL; 7964ed29b0b4SJens Axboe 7965ed29b0b4SJens Axboe if (unlikely(reg.bgid < BGID_ARRAY && !ctx->io_bl)) { 7966ed29b0b4SJens Axboe int ret = io_init_bl_list(ctx); 7967ed29b0b4SJens Axboe if (ret) 7968ed29b0b4SJens Axboe return ret; 7969ed29b0b4SJens Axboe } 7970ed29b0b4SJens Axboe 7971ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, reg.bgid); 7972ed29b0b4SJens Axboe if (bl) { 7973ed29b0b4SJens Axboe /* if mapped buffer ring OR classic exists, don't allow */ 7974ed29b0b4SJens Axboe if (bl->buf_nr_pages || !list_empty(&bl->buf_list)) 7975ed29b0b4SJens Axboe return -EEXIST; 7976ed29b0b4SJens Axboe } else { 7977ed29b0b4SJens Axboe free_bl = bl = kzalloc(sizeof(*bl), GFP_KERNEL); 7978ed29b0b4SJens Axboe if (!bl) 7979ed29b0b4SJens Axboe return -ENOMEM; 7980ed29b0b4SJens Axboe } 7981ed29b0b4SJens Axboe 7982ed29b0b4SJens Axboe pages = io_pin_pages(reg.ring_addr, 7983ed29b0b4SJens Axboe struct_size(br, bufs, reg.ring_entries), 7984ed29b0b4SJens Axboe &nr_pages); 7985ed29b0b4SJens Axboe if (IS_ERR(pages)) { 7986ed29b0b4SJens Axboe kfree(free_bl); 7987ed29b0b4SJens Axboe return PTR_ERR(pages); 7988ed29b0b4SJens Axboe } 7989ed29b0b4SJens Axboe 7990ed29b0b4SJens Axboe br = page_address(pages[0]); 7991ed29b0b4SJens Axboe bl->buf_pages = pages; 7992ed29b0b4SJens Axboe bl->buf_nr_pages = nr_pages; 7993ed29b0b4SJens Axboe bl->nr_entries = reg.ring_entries; 7994ed29b0b4SJens Axboe bl->buf_ring = br; 7995ed29b0b4SJens Axboe bl->mask = reg.ring_entries - 1; 7996ed29b0b4SJens Axboe io_buffer_add_list(ctx, bl, reg.bgid); 7997ed29b0b4SJens Axboe return 0; 7998ed29b0b4SJens Axboe } 7999ed29b0b4SJens Axboe 8000ed29b0b4SJens Axboe static int io_unregister_pbuf_ring(struct io_ring_ctx *ctx, void __user *arg) 8001ed29b0b4SJens Axboe { 8002ed29b0b4SJens Axboe struct io_uring_buf_reg reg; 8003ed29b0b4SJens Axboe struct io_buffer_list *bl; 8004ed29b0b4SJens Axboe 8005ed29b0b4SJens Axboe if (copy_from_user(®, arg, sizeof(reg))) 8006ed29b0b4SJens Axboe return -EFAULT; 8007ed29b0b4SJens Axboe if (reg.pad || reg.resv[0] || reg.resv[1] || reg.resv[2]) 8008ed29b0b4SJens Axboe return -EINVAL; 8009ed29b0b4SJens Axboe 8010ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, reg.bgid); 8011ed29b0b4SJens Axboe if (!bl) 8012ed29b0b4SJens Axboe return -ENOENT; 8013ed29b0b4SJens Axboe if (!bl->buf_nr_pages) 8014ed29b0b4SJens Axboe return -EINVAL; 8015ed29b0b4SJens Axboe 8016ed29b0b4SJens Axboe __io_remove_buffers(ctx, bl, -1U); 8017ed29b0b4SJens Axboe if (bl->bgid >= BGID_ARRAY) { 8018ed29b0b4SJens Axboe xa_erase(&ctx->io_bl_xa, bl->bgid); 8019ed29b0b4SJens Axboe kfree(bl); 8020ed29b0b4SJens Axboe } 8021ed29b0b4SJens Axboe return 0; 8022ed29b0b4SJens Axboe } 8023ed29b0b4SJens Axboe 8024ed29b0b4SJens Axboe static int __io_uring_register(struct io_ring_ctx *ctx, unsigned opcode, 8025ed29b0b4SJens Axboe void __user *arg, unsigned nr_args) 8026ed29b0b4SJens Axboe __releases(ctx->uring_lock) 8027ed29b0b4SJens Axboe __acquires(ctx->uring_lock) 8028ed29b0b4SJens Axboe { 8029ed29b0b4SJens Axboe int ret; 8030ed29b0b4SJens Axboe 8031ed29b0b4SJens Axboe /* 8032ed29b0b4SJens Axboe * We're inside the ring mutex, if the ref is already dying, then 8033ed29b0b4SJens Axboe * someone else killed the ctx or is already going through 8034ed29b0b4SJens Axboe * io_uring_register(). 8035ed29b0b4SJens Axboe */ 8036ed29b0b4SJens Axboe if (percpu_ref_is_dying(&ctx->refs)) 8037ed29b0b4SJens Axboe return -ENXIO; 8038ed29b0b4SJens Axboe 8039ed29b0b4SJens Axboe if (ctx->restricted) { 8040ed29b0b4SJens Axboe if (opcode >= IORING_REGISTER_LAST) 8041ed29b0b4SJens Axboe return -EINVAL; 8042ed29b0b4SJens Axboe opcode = array_index_nospec(opcode, IORING_REGISTER_LAST); 8043ed29b0b4SJens Axboe if (!test_bit(opcode, ctx->restrictions.register_op)) 8044ed29b0b4SJens Axboe return -EACCES; 8045ed29b0b4SJens Axboe } 8046ed29b0b4SJens Axboe 8047ed29b0b4SJens Axboe switch (opcode) { 8048ed29b0b4SJens Axboe case IORING_REGISTER_BUFFERS: 8049ed29b0b4SJens Axboe ret = -EFAULT; 8050ed29b0b4SJens Axboe if (!arg) 8051ed29b0b4SJens Axboe break; 8052ed29b0b4SJens Axboe ret = io_sqe_buffers_register(ctx, arg, nr_args, NULL); 8053ed29b0b4SJens Axboe break; 8054ed29b0b4SJens Axboe case IORING_UNREGISTER_BUFFERS: 8055ed29b0b4SJens Axboe ret = -EINVAL; 8056ed29b0b4SJens Axboe if (arg || nr_args) 8057ed29b0b4SJens Axboe break; 8058ed29b0b4SJens Axboe ret = io_sqe_buffers_unregister(ctx); 8059ed29b0b4SJens Axboe break; 8060ed29b0b4SJens Axboe case IORING_REGISTER_FILES: 8061ed29b0b4SJens Axboe ret = -EFAULT; 8062ed29b0b4SJens Axboe if (!arg) 8063ed29b0b4SJens Axboe break; 8064ed29b0b4SJens Axboe ret = io_sqe_files_register(ctx, arg, nr_args, NULL); 8065ed29b0b4SJens Axboe break; 8066ed29b0b4SJens Axboe case IORING_UNREGISTER_FILES: 8067ed29b0b4SJens Axboe ret = -EINVAL; 8068ed29b0b4SJens Axboe if (arg || nr_args) 8069ed29b0b4SJens Axboe break; 8070ed29b0b4SJens Axboe ret = io_sqe_files_unregister(ctx); 8071ed29b0b4SJens Axboe break; 8072ed29b0b4SJens Axboe case IORING_REGISTER_FILES_UPDATE: 8073ed29b0b4SJens Axboe ret = io_register_files_update(ctx, arg, nr_args); 8074ed29b0b4SJens Axboe break; 8075ed29b0b4SJens Axboe case IORING_REGISTER_EVENTFD: 8076ed29b0b4SJens Axboe ret = -EINVAL; 8077ed29b0b4SJens Axboe if (nr_args != 1) 8078ed29b0b4SJens Axboe break; 8079ed29b0b4SJens Axboe ret = io_eventfd_register(ctx, arg, 0); 8080ed29b0b4SJens Axboe break; 8081ed29b0b4SJens Axboe case IORING_REGISTER_EVENTFD_ASYNC: 8082ed29b0b4SJens Axboe ret = -EINVAL; 8083ed29b0b4SJens Axboe if (nr_args != 1) 8084ed29b0b4SJens Axboe break; 8085ed29b0b4SJens Axboe ret = io_eventfd_register(ctx, arg, 1); 8086ed29b0b4SJens Axboe break; 8087ed29b0b4SJens Axboe case IORING_UNREGISTER_EVENTFD: 8088ed29b0b4SJens Axboe ret = -EINVAL; 8089ed29b0b4SJens Axboe if (arg || nr_args) 8090ed29b0b4SJens Axboe break; 8091ed29b0b4SJens Axboe ret = io_eventfd_unregister(ctx); 8092ed29b0b4SJens Axboe break; 8093ed29b0b4SJens Axboe case IORING_REGISTER_PROBE: 8094ed29b0b4SJens Axboe ret = -EINVAL; 8095ed29b0b4SJens Axboe if (!arg || nr_args > 256) 8096ed29b0b4SJens Axboe break; 8097ed29b0b4SJens Axboe ret = io_probe(ctx, arg, nr_args); 8098ed29b0b4SJens Axboe break; 8099ed29b0b4SJens Axboe case IORING_REGISTER_PERSONALITY: 8100ed29b0b4SJens Axboe ret = -EINVAL; 8101ed29b0b4SJens Axboe if (arg || nr_args) 8102ed29b0b4SJens Axboe break; 8103ed29b0b4SJens Axboe ret = io_register_personality(ctx); 8104ed29b0b4SJens Axboe break; 8105ed29b0b4SJens Axboe case IORING_UNREGISTER_PERSONALITY: 8106ed29b0b4SJens Axboe ret = -EINVAL; 8107ed29b0b4SJens Axboe if (arg) 8108ed29b0b4SJens Axboe break; 8109ed29b0b4SJens Axboe ret = io_unregister_personality(ctx, nr_args); 8110ed29b0b4SJens Axboe break; 8111ed29b0b4SJens Axboe case IORING_REGISTER_ENABLE_RINGS: 8112ed29b0b4SJens Axboe ret = -EINVAL; 8113ed29b0b4SJens Axboe if (arg || nr_args) 8114ed29b0b4SJens Axboe break; 8115ed29b0b4SJens Axboe ret = io_register_enable_rings(ctx); 8116ed29b0b4SJens Axboe break; 8117ed29b0b4SJens Axboe case IORING_REGISTER_RESTRICTIONS: 8118ed29b0b4SJens Axboe ret = io_register_restrictions(ctx, arg, nr_args); 8119ed29b0b4SJens Axboe break; 8120ed29b0b4SJens Axboe case IORING_REGISTER_FILES2: 8121ed29b0b4SJens Axboe ret = io_register_rsrc(ctx, arg, nr_args, IORING_RSRC_FILE); 8122ed29b0b4SJens Axboe break; 8123ed29b0b4SJens Axboe case IORING_REGISTER_FILES_UPDATE2: 8124ed29b0b4SJens Axboe ret = io_register_rsrc_update(ctx, arg, nr_args, 8125ed29b0b4SJens Axboe IORING_RSRC_FILE); 8126ed29b0b4SJens Axboe break; 8127ed29b0b4SJens Axboe case IORING_REGISTER_BUFFERS2: 8128ed29b0b4SJens Axboe ret = io_register_rsrc(ctx, arg, nr_args, IORING_RSRC_BUFFER); 8129ed29b0b4SJens Axboe break; 8130ed29b0b4SJens Axboe case IORING_REGISTER_BUFFERS_UPDATE: 8131ed29b0b4SJens Axboe ret = io_register_rsrc_update(ctx, arg, nr_args, 8132ed29b0b4SJens Axboe IORING_RSRC_BUFFER); 8133ed29b0b4SJens Axboe break; 8134ed29b0b4SJens Axboe case IORING_REGISTER_IOWQ_AFF: 8135ed29b0b4SJens Axboe ret = -EINVAL; 8136ed29b0b4SJens Axboe if (!arg || !nr_args) 8137ed29b0b4SJens Axboe break; 8138ed29b0b4SJens Axboe ret = io_register_iowq_aff(ctx, arg, nr_args); 8139ed29b0b4SJens Axboe break; 8140ed29b0b4SJens Axboe case IORING_UNREGISTER_IOWQ_AFF: 8141ed29b0b4SJens Axboe ret = -EINVAL; 8142ed29b0b4SJens Axboe if (arg || nr_args) 8143ed29b0b4SJens Axboe break; 8144ed29b0b4SJens Axboe ret = io_unregister_iowq_aff(ctx); 8145ed29b0b4SJens Axboe break; 8146ed29b0b4SJens Axboe case IORING_REGISTER_IOWQ_MAX_WORKERS: 8147ed29b0b4SJens Axboe ret = -EINVAL; 8148ed29b0b4SJens Axboe if (!arg || nr_args != 2) 8149ed29b0b4SJens Axboe break; 8150ed29b0b4SJens Axboe ret = io_register_iowq_max_workers(ctx, arg); 8151ed29b0b4SJens Axboe break; 8152ed29b0b4SJens Axboe case IORING_REGISTER_RING_FDS: 8153ed29b0b4SJens Axboe ret = io_ringfd_register(ctx, arg, nr_args); 8154ed29b0b4SJens Axboe break; 8155ed29b0b4SJens Axboe case IORING_UNREGISTER_RING_FDS: 8156ed29b0b4SJens Axboe ret = io_ringfd_unregister(ctx, arg, nr_args); 8157ed29b0b4SJens Axboe break; 8158ed29b0b4SJens Axboe case IORING_REGISTER_PBUF_RING: 8159ed29b0b4SJens Axboe ret = -EINVAL; 8160ed29b0b4SJens Axboe if (!arg || nr_args != 1) 8161ed29b0b4SJens Axboe break; 8162ed29b0b4SJens Axboe ret = io_register_pbuf_ring(ctx, arg); 8163ed29b0b4SJens Axboe break; 8164ed29b0b4SJens Axboe case IORING_UNREGISTER_PBUF_RING: 8165ed29b0b4SJens Axboe ret = -EINVAL; 8166ed29b0b4SJens Axboe if (!arg || nr_args != 1) 8167ed29b0b4SJens Axboe break; 8168ed29b0b4SJens Axboe ret = io_unregister_pbuf_ring(ctx, arg); 8169ed29b0b4SJens Axboe break; 8170ed29b0b4SJens Axboe default: 8171ed29b0b4SJens Axboe ret = -EINVAL; 8172ed29b0b4SJens Axboe break; 8173ed29b0b4SJens Axboe } 8174ed29b0b4SJens Axboe 8175ed29b0b4SJens Axboe return ret; 8176ed29b0b4SJens Axboe } 8177ed29b0b4SJens Axboe 8178ed29b0b4SJens Axboe SYSCALL_DEFINE4(io_uring_register, unsigned int, fd, unsigned int, opcode, 8179ed29b0b4SJens Axboe void __user *, arg, unsigned int, nr_args) 8180ed29b0b4SJens Axboe { 8181ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 8182ed29b0b4SJens Axboe long ret = -EBADF; 8183ed29b0b4SJens Axboe struct fd f; 8184ed29b0b4SJens Axboe 8185ed29b0b4SJens Axboe f = fdget(fd); 8186ed29b0b4SJens Axboe if (!f.file) 8187ed29b0b4SJens Axboe return -EBADF; 8188ed29b0b4SJens Axboe 8189ed29b0b4SJens Axboe ret = -EOPNOTSUPP; 8190e5550a14SJens Axboe if (!io_is_uring_fops(f.file)) 8191ed29b0b4SJens Axboe goto out_fput; 8192ed29b0b4SJens Axboe 8193ed29b0b4SJens Axboe ctx = f.file->private_data; 8194ed29b0b4SJens Axboe 8195ed29b0b4SJens Axboe io_run_task_work(); 8196ed29b0b4SJens Axboe 8197ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 8198ed29b0b4SJens Axboe ret = __io_uring_register(ctx, opcode, arg, nr_args); 8199ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 8200ed29b0b4SJens Axboe trace_io_uring_register(ctx, opcode, ctx->nr_user_files, ctx->nr_user_bufs, ret); 8201ed29b0b4SJens Axboe out_fput: 8202ed29b0b4SJens Axboe fdput(f); 8203ed29b0b4SJens Axboe return ret; 8204ed29b0b4SJens Axboe } 8205ed29b0b4SJens Axboe 8206ed29b0b4SJens Axboe static int io_no_issue(struct io_kiocb *req, unsigned int issue_flags) 8207ed29b0b4SJens Axboe { 8208ed29b0b4SJens Axboe WARN_ON_ONCE(1); 8209ed29b0b4SJens Axboe return -ECANCELED; 8210ed29b0b4SJens Axboe } 8211ed29b0b4SJens Axboe 8212ed29b0b4SJens Axboe static const struct io_op_def io_op_defs[] = { 8213ed29b0b4SJens Axboe [IORING_OP_NOP] = { 8214ed29b0b4SJens Axboe .audit_skip = 1, 8215ed29b0b4SJens Axboe .iopoll = 1, 8216*cfd22e6bSJens Axboe .name = "NOP", 8217ed29b0b4SJens Axboe .prep = io_nop_prep, 8218ed29b0b4SJens Axboe .issue = io_nop, 8219ed29b0b4SJens Axboe }, 8220ed29b0b4SJens Axboe [IORING_OP_READV] = { 8221ed29b0b4SJens Axboe .needs_file = 1, 8222ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8223ed29b0b4SJens Axboe .pollin = 1, 8224ed29b0b4SJens Axboe .buffer_select = 1, 8225ed29b0b4SJens Axboe .plug = 1, 8226ed29b0b4SJens Axboe .audit_skip = 1, 8227ed29b0b4SJens Axboe .ioprio = 1, 8228ed29b0b4SJens Axboe .iopoll = 1, 8229ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 8230*cfd22e6bSJens Axboe .name = "READV", 8231ed29b0b4SJens Axboe .prep = io_prep_rw, 8232ed29b0b4SJens Axboe .issue = io_read, 8233dc919cafSJens Axboe .prep_async = io_readv_prep_async, 82344d4c9cffSJens Axboe .cleanup = io_readv_writev_cleanup, 8235ed29b0b4SJens Axboe }, 8236ed29b0b4SJens Axboe [IORING_OP_WRITEV] = { 8237ed29b0b4SJens Axboe .needs_file = 1, 8238ed29b0b4SJens Axboe .hash_reg_file = 1, 8239ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8240ed29b0b4SJens Axboe .pollout = 1, 8241ed29b0b4SJens Axboe .plug = 1, 8242ed29b0b4SJens Axboe .audit_skip = 1, 8243ed29b0b4SJens Axboe .ioprio = 1, 8244ed29b0b4SJens Axboe .iopoll = 1, 8245ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 8246*cfd22e6bSJens Axboe .name = "WRITEV", 8247ed29b0b4SJens Axboe .prep = io_prep_rw, 8248ed29b0b4SJens Axboe .issue = io_write, 8249dc919cafSJens Axboe .prep_async = io_writev_prep_async, 82504d4c9cffSJens Axboe .cleanup = io_readv_writev_cleanup, 8251ed29b0b4SJens Axboe }, 8252ed29b0b4SJens Axboe [IORING_OP_FSYNC] = { 8253ed29b0b4SJens Axboe .needs_file = 1, 8254ed29b0b4SJens Axboe .audit_skip = 1, 8255*cfd22e6bSJens Axboe .name = "FSYNC", 8256ed29b0b4SJens Axboe .prep = io_fsync_prep, 8257ed29b0b4SJens Axboe .issue = io_fsync, 8258ed29b0b4SJens Axboe }, 8259ed29b0b4SJens Axboe [IORING_OP_READ_FIXED] = { 8260ed29b0b4SJens Axboe .needs_file = 1, 8261ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8262ed29b0b4SJens Axboe .pollin = 1, 8263ed29b0b4SJens Axboe .plug = 1, 8264ed29b0b4SJens Axboe .audit_skip = 1, 8265ed29b0b4SJens Axboe .ioprio = 1, 8266ed29b0b4SJens Axboe .iopoll = 1, 8267ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 8268*cfd22e6bSJens Axboe .name = "READ_FIXED", 8269ed29b0b4SJens Axboe .prep = io_prep_rw, 8270ed29b0b4SJens Axboe .issue = io_read, 8271ed29b0b4SJens Axboe }, 8272ed29b0b4SJens Axboe [IORING_OP_WRITE_FIXED] = { 8273ed29b0b4SJens Axboe .needs_file = 1, 8274ed29b0b4SJens Axboe .hash_reg_file = 1, 8275ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8276ed29b0b4SJens Axboe .pollout = 1, 8277ed29b0b4SJens Axboe .plug = 1, 8278ed29b0b4SJens Axboe .audit_skip = 1, 8279ed29b0b4SJens Axboe .ioprio = 1, 8280ed29b0b4SJens Axboe .iopoll = 1, 8281ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 8282*cfd22e6bSJens Axboe .name = "WRITE_FIXED", 8283ed29b0b4SJens Axboe .prep = io_prep_rw, 8284ed29b0b4SJens Axboe .issue = io_write, 8285ed29b0b4SJens Axboe }, 8286ed29b0b4SJens Axboe [IORING_OP_POLL_ADD] = { 8287ed29b0b4SJens Axboe .needs_file = 1, 8288ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8289ed29b0b4SJens Axboe .audit_skip = 1, 8290*cfd22e6bSJens Axboe .name = "POLL_ADD", 8291ed29b0b4SJens Axboe .prep = io_poll_add_prep, 8292ed29b0b4SJens Axboe .issue = io_poll_add, 8293ed29b0b4SJens Axboe }, 8294ed29b0b4SJens Axboe [IORING_OP_POLL_REMOVE] = { 8295ed29b0b4SJens Axboe .audit_skip = 1, 8296*cfd22e6bSJens Axboe .name = "POLL_REMOVE", 8297ed29b0b4SJens Axboe .prep = io_poll_remove_prep, 8298ed29b0b4SJens Axboe .issue = io_poll_remove, 8299ed29b0b4SJens Axboe }, 8300ed29b0b4SJens Axboe [IORING_OP_SYNC_FILE_RANGE] = { 8301ed29b0b4SJens Axboe .needs_file = 1, 8302ed29b0b4SJens Axboe .audit_skip = 1, 8303*cfd22e6bSJens Axboe .name = "SYNC_FILE_RANGE", 8304ed29b0b4SJens Axboe .prep = io_sfr_prep, 8305ed29b0b4SJens Axboe .issue = io_sync_file_range, 8306ed29b0b4SJens Axboe }, 8307ed29b0b4SJens Axboe [IORING_OP_SENDMSG] = { 8308ed29b0b4SJens Axboe .needs_file = 1, 8309ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8310ed29b0b4SJens Axboe .pollout = 1, 8311ed29b0b4SJens Axboe .ioprio = 1, 8312*cfd22e6bSJens Axboe .name = "SENDMSG", 8313f9ead18cSJens Axboe #if defined(CONFIG_NET) 8314ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_msghdr), 8315ed29b0b4SJens Axboe .prep = io_sendmsg_prep, 8316ed29b0b4SJens Axboe .issue = io_sendmsg, 8317dc919cafSJens Axboe .prep_async = io_sendmsg_prep_async, 83184d4c9cffSJens Axboe .cleanup = io_sendmsg_recvmsg_cleanup, 8319f9ead18cSJens Axboe #else 8320f9ead18cSJens Axboe .prep = io_eopnotsupp_prep, 83214d4c9cffSJens Axboe #endif 8322ed29b0b4SJens Axboe }, 8323ed29b0b4SJens Axboe [IORING_OP_RECVMSG] = { 8324ed29b0b4SJens Axboe .needs_file = 1, 8325ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8326ed29b0b4SJens Axboe .pollin = 1, 8327ed29b0b4SJens Axboe .buffer_select = 1, 8328ed29b0b4SJens Axboe .ioprio = 1, 8329*cfd22e6bSJens Axboe .name = "RECVMSG", 8330f9ead18cSJens Axboe #if defined(CONFIG_NET) 8331ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_msghdr), 8332ed29b0b4SJens Axboe .prep = io_recvmsg_prep, 8333ed29b0b4SJens Axboe .issue = io_recvmsg, 8334dc919cafSJens Axboe .prep_async = io_recvmsg_prep_async, 83354d4c9cffSJens Axboe .cleanup = io_sendmsg_recvmsg_cleanup, 8336f9ead18cSJens Axboe #else 8337f9ead18cSJens Axboe .prep = io_eopnotsupp_prep, 83384d4c9cffSJens Axboe #endif 8339ed29b0b4SJens Axboe }, 8340ed29b0b4SJens Axboe [IORING_OP_TIMEOUT] = { 8341ed29b0b4SJens Axboe .audit_skip = 1, 8342ed29b0b4SJens Axboe .async_size = sizeof(struct io_timeout_data), 8343*cfd22e6bSJens Axboe .name = "TIMEOUT", 8344ed29b0b4SJens Axboe .prep = io_timeout_prep, 8345ed29b0b4SJens Axboe .issue = io_timeout, 8346ed29b0b4SJens Axboe }, 8347ed29b0b4SJens Axboe [IORING_OP_TIMEOUT_REMOVE] = { 8348ed29b0b4SJens Axboe /* used by timeout updates' prep() */ 8349ed29b0b4SJens Axboe .audit_skip = 1, 8350*cfd22e6bSJens Axboe .name = "TIMEOUT_REMOVE", 8351ed29b0b4SJens Axboe .prep = io_timeout_remove_prep, 8352ed29b0b4SJens Axboe .issue = io_timeout_remove, 8353ed29b0b4SJens Axboe }, 8354ed29b0b4SJens Axboe [IORING_OP_ACCEPT] = { 8355ed29b0b4SJens Axboe .needs_file = 1, 8356ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8357ed29b0b4SJens Axboe .pollin = 1, 8358ed29b0b4SJens Axboe .poll_exclusive = 1, 8359ed29b0b4SJens Axboe .ioprio = 1, /* used for flags */ 8360*cfd22e6bSJens Axboe .name = "ACCEPT", 8361f9ead18cSJens Axboe #if defined(CONFIG_NET) 8362ed29b0b4SJens Axboe .prep = io_accept_prep, 8363ed29b0b4SJens Axboe .issue = io_accept, 8364f9ead18cSJens Axboe #else 8365f9ead18cSJens Axboe .prep = io_eopnotsupp_prep, 8366f9ead18cSJens Axboe #endif 8367ed29b0b4SJens Axboe }, 8368ed29b0b4SJens Axboe [IORING_OP_ASYNC_CANCEL] = { 8369ed29b0b4SJens Axboe .audit_skip = 1, 8370*cfd22e6bSJens Axboe .name = "ASYNC_CANCEL", 8371ed29b0b4SJens Axboe .prep = io_async_cancel_prep, 8372ed29b0b4SJens Axboe .issue = io_async_cancel, 8373ed29b0b4SJens Axboe }, 8374ed29b0b4SJens Axboe [IORING_OP_LINK_TIMEOUT] = { 8375ed29b0b4SJens Axboe .audit_skip = 1, 8376ed29b0b4SJens Axboe .async_size = sizeof(struct io_timeout_data), 8377*cfd22e6bSJens Axboe .name = "LINK_TIMEOUT", 8378ed29b0b4SJens Axboe .prep = io_link_timeout_prep, 8379ed29b0b4SJens Axboe .issue = io_no_issue, 8380ed29b0b4SJens Axboe }, 8381ed29b0b4SJens Axboe [IORING_OP_CONNECT] = { 8382ed29b0b4SJens Axboe .needs_file = 1, 8383ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8384ed29b0b4SJens Axboe .pollout = 1, 8385*cfd22e6bSJens Axboe .name = "CONNECT", 8386f9ead18cSJens Axboe #if defined(CONFIG_NET) 8387ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_connect), 8388ed29b0b4SJens Axboe .prep = io_connect_prep, 8389ed29b0b4SJens Axboe .issue = io_connect, 8390dc919cafSJens Axboe .prep_async = io_connect_prep_async, 8391f9ead18cSJens Axboe #else 8392f9ead18cSJens Axboe .prep = io_eopnotsupp_prep, 8393f9ead18cSJens Axboe #endif 8394ed29b0b4SJens Axboe }, 8395ed29b0b4SJens Axboe [IORING_OP_FALLOCATE] = { 8396ed29b0b4SJens Axboe .needs_file = 1, 8397*cfd22e6bSJens Axboe .name = "FALLOCATE", 8398ed29b0b4SJens Axboe .prep = io_fallocate_prep, 8399ed29b0b4SJens Axboe .issue = io_fallocate, 8400ed29b0b4SJens Axboe }, 8401ed29b0b4SJens Axboe [IORING_OP_OPENAT] = { 8402*cfd22e6bSJens Axboe .name = "OPENAT", 8403ed29b0b4SJens Axboe .prep = io_openat_prep, 8404ed29b0b4SJens Axboe .issue = io_openat, 84054d4c9cffSJens Axboe .cleanup = io_open_cleanup, 8406ed29b0b4SJens Axboe }, 8407ed29b0b4SJens Axboe [IORING_OP_CLOSE] = { 8408*cfd22e6bSJens Axboe .name = "CLOSE", 8409ed29b0b4SJens Axboe .prep = io_close_prep, 8410ed29b0b4SJens Axboe .issue = io_close, 8411ed29b0b4SJens Axboe }, 8412ed29b0b4SJens Axboe [IORING_OP_FILES_UPDATE] = { 8413ed29b0b4SJens Axboe .audit_skip = 1, 8414ed29b0b4SJens Axboe .iopoll = 1, 8415*cfd22e6bSJens Axboe .name = "FILES_UPDATE", 8416ed29b0b4SJens Axboe .prep = io_files_update_prep, 8417ed29b0b4SJens Axboe .issue = io_files_update, 8418ed29b0b4SJens Axboe }, 8419ed29b0b4SJens Axboe [IORING_OP_STATX] = { 8420ed29b0b4SJens Axboe .audit_skip = 1, 8421*cfd22e6bSJens Axboe .name = "STATX", 8422ed29b0b4SJens Axboe .prep = io_statx_prep, 8423ed29b0b4SJens Axboe .issue = io_statx, 84244d4c9cffSJens Axboe .cleanup = io_statx_cleanup, 8425ed29b0b4SJens Axboe }, 8426ed29b0b4SJens Axboe [IORING_OP_READ] = { 8427ed29b0b4SJens Axboe .needs_file = 1, 8428ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8429ed29b0b4SJens Axboe .pollin = 1, 8430ed29b0b4SJens Axboe .buffer_select = 1, 8431ed29b0b4SJens Axboe .plug = 1, 8432ed29b0b4SJens Axboe .audit_skip = 1, 8433ed29b0b4SJens Axboe .ioprio = 1, 8434ed29b0b4SJens Axboe .iopoll = 1, 8435ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 8436*cfd22e6bSJens Axboe .name = "READ", 8437ed29b0b4SJens Axboe .prep = io_prep_rw, 8438ed29b0b4SJens Axboe .issue = io_read, 8439ed29b0b4SJens Axboe }, 8440ed29b0b4SJens Axboe [IORING_OP_WRITE] = { 8441ed29b0b4SJens Axboe .needs_file = 1, 8442ed29b0b4SJens Axboe .hash_reg_file = 1, 8443ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8444ed29b0b4SJens Axboe .pollout = 1, 8445ed29b0b4SJens Axboe .plug = 1, 8446ed29b0b4SJens Axboe .audit_skip = 1, 8447ed29b0b4SJens Axboe .ioprio = 1, 8448ed29b0b4SJens Axboe .iopoll = 1, 8449ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 8450*cfd22e6bSJens Axboe .name = "WRITE", 8451ed29b0b4SJens Axboe .prep = io_prep_rw, 8452ed29b0b4SJens Axboe .issue = io_write, 8453ed29b0b4SJens Axboe }, 8454ed29b0b4SJens Axboe [IORING_OP_FADVISE] = { 8455ed29b0b4SJens Axboe .needs_file = 1, 8456ed29b0b4SJens Axboe .audit_skip = 1, 8457*cfd22e6bSJens Axboe .name = "FADVISE", 8458ed29b0b4SJens Axboe .prep = io_fadvise_prep, 8459ed29b0b4SJens Axboe .issue = io_fadvise, 8460ed29b0b4SJens Axboe }, 8461ed29b0b4SJens Axboe [IORING_OP_MADVISE] = { 8462*cfd22e6bSJens Axboe .name = "MADVISE", 8463ed29b0b4SJens Axboe .prep = io_madvise_prep, 8464ed29b0b4SJens Axboe .issue = io_madvise, 8465ed29b0b4SJens Axboe }, 8466ed29b0b4SJens Axboe [IORING_OP_SEND] = { 8467ed29b0b4SJens Axboe .needs_file = 1, 8468ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8469ed29b0b4SJens Axboe .pollout = 1, 8470ed29b0b4SJens Axboe .audit_skip = 1, 8471ed29b0b4SJens Axboe .ioprio = 1, 8472*cfd22e6bSJens Axboe .name = "SEND", 8473f9ead18cSJens Axboe #if defined(CONFIG_NET) 8474ed29b0b4SJens Axboe .prep = io_sendmsg_prep, 8475ed29b0b4SJens Axboe .issue = io_send, 8476f9ead18cSJens Axboe #else 8477f9ead18cSJens Axboe .prep = io_eopnotsupp_prep, 8478f9ead18cSJens Axboe #endif 8479ed29b0b4SJens Axboe }, 8480ed29b0b4SJens Axboe [IORING_OP_RECV] = { 8481ed29b0b4SJens Axboe .needs_file = 1, 8482ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8483ed29b0b4SJens Axboe .pollin = 1, 8484ed29b0b4SJens Axboe .buffer_select = 1, 8485ed29b0b4SJens Axboe .audit_skip = 1, 8486ed29b0b4SJens Axboe .ioprio = 1, 8487*cfd22e6bSJens Axboe .name = "RECV", 8488f9ead18cSJens Axboe #if defined(CONFIG_NET) 8489ed29b0b4SJens Axboe .prep = io_recvmsg_prep, 8490ed29b0b4SJens Axboe .issue = io_recv, 8491f9ead18cSJens Axboe #else 8492f9ead18cSJens Axboe .prep = io_eopnotsupp_prep, 8493f9ead18cSJens Axboe #endif 8494ed29b0b4SJens Axboe }, 8495ed29b0b4SJens Axboe [IORING_OP_OPENAT2] = { 8496*cfd22e6bSJens Axboe .name = "OPENAT2", 8497ed29b0b4SJens Axboe .prep = io_openat2_prep, 8498ed29b0b4SJens Axboe .issue = io_openat2, 84994d4c9cffSJens Axboe .cleanup = io_open_cleanup, 8500ed29b0b4SJens Axboe }, 8501ed29b0b4SJens Axboe [IORING_OP_EPOLL_CTL] = { 8502ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8503ed29b0b4SJens Axboe .audit_skip = 1, 8504*cfd22e6bSJens Axboe .name = "EPOLL", 85054cf90495SJens Axboe #if defined(CONFIG_EPOLL) 8506ed29b0b4SJens Axboe .prep = io_epoll_ctl_prep, 8507ed29b0b4SJens Axboe .issue = io_epoll_ctl, 85084cf90495SJens Axboe #else 85094cf90495SJens Axboe .prep = io_eopnotsupp_prep, 85104cf90495SJens Axboe #endif 8511ed29b0b4SJens Axboe }, 8512ed29b0b4SJens Axboe [IORING_OP_SPLICE] = { 8513ed29b0b4SJens Axboe .needs_file = 1, 8514ed29b0b4SJens Axboe .hash_reg_file = 1, 8515ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8516ed29b0b4SJens Axboe .audit_skip = 1, 8517*cfd22e6bSJens Axboe .name = "SPLICE", 8518ed29b0b4SJens Axboe .prep = io_splice_prep, 8519ed29b0b4SJens Axboe .issue = io_splice, 8520ed29b0b4SJens Axboe }, 8521ed29b0b4SJens Axboe [IORING_OP_PROVIDE_BUFFERS] = { 8522ed29b0b4SJens Axboe .audit_skip = 1, 8523ed29b0b4SJens Axboe .iopoll = 1, 8524*cfd22e6bSJens Axboe .name = "PROVIDE_BUFFERS", 8525ed29b0b4SJens Axboe .prep = io_provide_buffers_prep, 8526ed29b0b4SJens Axboe .issue = io_provide_buffers, 8527ed29b0b4SJens Axboe }, 8528ed29b0b4SJens Axboe [IORING_OP_REMOVE_BUFFERS] = { 8529ed29b0b4SJens Axboe .audit_skip = 1, 8530ed29b0b4SJens Axboe .iopoll = 1, 8531*cfd22e6bSJens Axboe .name = "REMOVE_BUFFERS", 8532ed29b0b4SJens Axboe .prep = io_remove_buffers_prep, 8533ed29b0b4SJens Axboe .issue = io_remove_buffers, 8534ed29b0b4SJens Axboe }, 8535ed29b0b4SJens Axboe [IORING_OP_TEE] = { 8536ed29b0b4SJens Axboe .needs_file = 1, 8537ed29b0b4SJens Axboe .hash_reg_file = 1, 8538ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 8539ed29b0b4SJens Axboe .audit_skip = 1, 8540*cfd22e6bSJens Axboe .name = "TEE", 8541ed29b0b4SJens Axboe .prep = io_tee_prep, 8542ed29b0b4SJens Axboe .issue = io_tee, 8543ed29b0b4SJens Axboe }, 8544ed29b0b4SJens Axboe [IORING_OP_SHUTDOWN] = { 8545ed29b0b4SJens Axboe .needs_file = 1, 8546*cfd22e6bSJens Axboe .name = "SHUTDOWN", 8547f9ead18cSJens Axboe #if defined(CONFIG_NET) 8548ed29b0b4SJens Axboe .prep = io_shutdown_prep, 8549ed29b0b4SJens Axboe .issue = io_shutdown, 8550f9ead18cSJens Axboe #else 8551f9ead18cSJens Axboe .prep = io_eopnotsupp_prep, 8552f9ead18cSJens Axboe #endif 8553ed29b0b4SJens Axboe }, 8554ed29b0b4SJens Axboe [IORING_OP_RENAMEAT] = { 8555*cfd22e6bSJens Axboe .name = "RENAMEAT", 8556ed29b0b4SJens Axboe .prep = io_renameat_prep, 8557ed29b0b4SJens Axboe .issue = io_renameat, 85584d4c9cffSJens Axboe .cleanup = io_renameat_cleanup, 8559ed29b0b4SJens Axboe }, 8560ed29b0b4SJens Axboe [IORING_OP_UNLINKAT] = { 8561*cfd22e6bSJens Axboe .name = "UNLINKAT", 8562ed29b0b4SJens Axboe .prep = io_unlinkat_prep, 8563ed29b0b4SJens Axboe .issue = io_unlinkat, 85644d4c9cffSJens Axboe .cleanup = io_unlinkat_cleanup, 8565ed29b0b4SJens Axboe }, 8566ed29b0b4SJens Axboe [IORING_OP_MKDIRAT] = { 8567*cfd22e6bSJens Axboe .name = "MKDIRAT", 8568ed29b0b4SJens Axboe .prep = io_mkdirat_prep, 8569ed29b0b4SJens Axboe .issue = io_mkdirat, 85704d4c9cffSJens Axboe .cleanup = io_mkdirat_cleanup, 8571ed29b0b4SJens Axboe }, 8572ed29b0b4SJens Axboe [IORING_OP_SYMLINKAT] = { 8573*cfd22e6bSJens Axboe .name = "SYMLINKAT", 8574ed29b0b4SJens Axboe .prep = io_symlinkat_prep, 8575ed29b0b4SJens Axboe .issue = io_symlinkat, 85764d4c9cffSJens Axboe .cleanup = io_link_cleanup, 8577ed29b0b4SJens Axboe }, 8578ed29b0b4SJens Axboe [IORING_OP_LINKAT] = { 8579*cfd22e6bSJens Axboe .name = "LINKAT", 8580ed29b0b4SJens Axboe .prep = io_linkat_prep, 8581ed29b0b4SJens Axboe .issue = io_linkat, 85824d4c9cffSJens Axboe .cleanup = io_link_cleanup, 8583ed29b0b4SJens Axboe }, 8584ed29b0b4SJens Axboe [IORING_OP_MSG_RING] = { 8585ed29b0b4SJens Axboe .needs_file = 1, 8586ed29b0b4SJens Axboe .iopoll = 1, 8587*cfd22e6bSJens Axboe .name = "MSG_RING", 8588ed29b0b4SJens Axboe .prep = io_msg_ring_prep, 8589ed29b0b4SJens Axboe .issue = io_msg_ring, 8590ed29b0b4SJens Axboe }, 8591ed29b0b4SJens Axboe [IORING_OP_FSETXATTR] = { 8592ed29b0b4SJens Axboe .needs_file = 1, 8593*cfd22e6bSJens Axboe .name = "FSETXATTR", 8594ed29b0b4SJens Axboe .prep = io_fsetxattr_prep, 8595ed29b0b4SJens Axboe .issue = io_fsetxattr, 85964d4c9cffSJens Axboe .cleanup = io_xattr_cleanup, 8597ed29b0b4SJens Axboe }, 8598ed29b0b4SJens Axboe [IORING_OP_SETXATTR] = { 8599*cfd22e6bSJens Axboe .name = "SETXATTR", 8600ed29b0b4SJens Axboe .prep = io_setxattr_prep, 8601ed29b0b4SJens Axboe .issue = io_setxattr, 86024d4c9cffSJens Axboe .cleanup = io_xattr_cleanup, 8603ed29b0b4SJens Axboe }, 8604ed29b0b4SJens Axboe [IORING_OP_FGETXATTR] = { 8605ed29b0b4SJens Axboe .needs_file = 1, 8606*cfd22e6bSJens Axboe .name = "FGETXATTR", 8607ed29b0b4SJens Axboe .prep = io_fgetxattr_prep, 8608ed29b0b4SJens Axboe .issue = io_fgetxattr, 86094d4c9cffSJens Axboe .cleanup = io_xattr_cleanup, 8610ed29b0b4SJens Axboe }, 8611ed29b0b4SJens Axboe [IORING_OP_GETXATTR] = { 8612*cfd22e6bSJens Axboe .name = "GETXATTR", 8613ed29b0b4SJens Axboe .prep = io_getxattr_prep, 8614ed29b0b4SJens Axboe .issue = io_getxattr, 86154d4c9cffSJens Axboe .cleanup = io_xattr_cleanup, 8616ed29b0b4SJens Axboe }, 8617ed29b0b4SJens Axboe [IORING_OP_SOCKET] = { 8618ed29b0b4SJens Axboe .audit_skip = 1, 8619*cfd22e6bSJens Axboe .name = "SOCKET", 8620f9ead18cSJens Axboe #if defined(CONFIG_NET) 8621ed29b0b4SJens Axboe .prep = io_socket_prep, 8622ed29b0b4SJens Axboe .issue = io_socket, 8623f9ead18cSJens Axboe #else 8624f9ead18cSJens Axboe .prep = io_eopnotsupp_prep, 8625f9ead18cSJens Axboe #endif 8626ed29b0b4SJens Axboe }, 8627ed29b0b4SJens Axboe [IORING_OP_URING_CMD] = { 8628ed29b0b4SJens Axboe .needs_file = 1, 8629ed29b0b4SJens Axboe .plug = 1, 8630*cfd22e6bSJens Axboe .name = "URING_CMD", 8631ed29b0b4SJens Axboe .async_size = uring_cmd_pdu_size(1), 8632ed29b0b4SJens Axboe .prep = io_uring_cmd_prep, 8633ed29b0b4SJens Axboe .issue = io_uring_cmd, 8634dc919cafSJens Axboe .prep_async = io_uring_cmd_prep_async, 8635ed29b0b4SJens Axboe }, 8636ed29b0b4SJens Axboe }; 8637ed29b0b4SJens Axboe 8638ed29b0b4SJens Axboe static int __init io_uring_init(void) 8639ed29b0b4SJens Axboe { 8640ed29b0b4SJens Axboe int i; 8641ed29b0b4SJens Axboe 8642ed29b0b4SJens Axboe #define __BUILD_BUG_VERIFY_ELEMENT(stype, eoffset, etype, ename) do { \ 8643ed29b0b4SJens Axboe BUILD_BUG_ON(offsetof(stype, ename) != eoffset); \ 8644ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(etype) != sizeof_field(stype, ename)); \ 8645ed29b0b4SJens Axboe } while (0) 8646ed29b0b4SJens Axboe 8647ed29b0b4SJens Axboe #define BUILD_BUG_SQE_ELEM(eoffset, etype, ename) \ 8648ed29b0b4SJens Axboe __BUILD_BUG_VERIFY_ELEMENT(struct io_uring_sqe, eoffset, etype, ename) 8649ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(struct io_uring_sqe) != 64); 8650ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(0, __u8, opcode); 8651ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(1, __u8, flags); 8652ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(2, __u16, ioprio); 8653ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(4, __s32, fd); 8654ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(8, __u64, off); 8655ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(8, __u64, addr2); 8656ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(16, __u64, addr); 8657ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(16, __u64, splice_off_in); 8658ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(24, __u32, len); 8659ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __kernel_rwf_t, rw_flags); 8660ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, /* compat */ int, rw_flags); 8661ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, /* compat */ __u32, rw_flags); 8662ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, fsync_flags); 8663ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, /* compat */ __u16, poll_events); 8664ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, poll32_events); 8665ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, sync_range_flags); 8666ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, msg_flags); 8667ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, timeout_flags); 8668ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, accept_flags); 8669ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, cancel_flags); 8670ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, open_flags); 8671ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, statx_flags); 8672ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, fadvise_advice); 8673ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, splice_flags); 8674ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(32, __u64, user_data); 8675ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(40, __u16, buf_index); 8676ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(40, __u16, buf_group); 8677ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(42, __u16, personality); 8678ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(44, __s32, splice_fd_in); 8679ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(44, __u32, file_index); 8680ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(48, __u64, addr3); 8681ed29b0b4SJens Axboe 8682ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(struct io_uring_files_update) != 8683ed29b0b4SJens Axboe sizeof(struct io_uring_rsrc_update)); 8684ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(struct io_uring_rsrc_update) > 8685ed29b0b4SJens Axboe sizeof(struct io_uring_rsrc_update2)); 8686ed29b0b4SJens Axboe 8687ed29b0b4SJens Axboe /* ->buf_index is u16 */ 8688ed29b0b4SJens Axboe BUILD_BUG_ON(IORING_MAX_REG_BUFFERS >= (1u << 16)); 8689ed29b0b4SJens Axboe BUILD_BUG_ON(BGID_ARRAY * sizeof(struct io_buffer_list) > PAGE_SIZE); 8690ed29b0b4SJens Axboe BUILD_BUG_ON(offsetof(struct io_uring_buf_ring, bufs) != 0); 8691ed29b0b4SJens Axboe BUILD_BUG_ON(offsetof(struct io_uring_buf, resv) != 8692ed29b0b4SJens Axboe offsetof(struct io_uring_buf_ring, tail)); 8693ed29b0b4SJens Axboe 8694ed29b0b4SJens Axboe /* should fit into one byte */ 8695ed29b0b4SJens Axboe BUILD_BUG_ON(SQE_VALID_FLAGS >= (1 << 8)); 8696ed29b0b4SJens Axboe BUILD_BUG_ON(SQE_COMMON_FLAGS >= (1 << 8)); 8697ed29b0b4SJens Axboe BUILD_BUG_ON((SQE_VALID_FLAGS | SQE_COMMON_FLAGS) != SQE_VALID_FLAGS); 8698ed29b0b4SJens Axboe 8699ed29b0b4SJens Axboe BUILD_BUG_ON(ARRAY_SIZE(io_op_defs) != IORING_OP_LAST); 8700ed29b0b4SJens Axboe BUILD_BUG_ON(__REQ_F_LAST_BIT > 8 * sizeof(int)); 8701ed29b0b4SJens Axboe 8702ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(atomic_t) != sizeof(u32)); 8703ed29b0b4SJens Axboe 8704ed29b0b4SJens Axboe for (i = 0; i < ARRAY_SIZE(io_op_defs); i++) { 8705ed29b0b4SJens Axboe BUG_ON(!io_op_defs[i].prep); 87064cf90495SJens Axboe if (io_op_defs[i].prep != io_eopnotsupp_prep) 8707ed29b0b4SJens Axboe BUG_ON(!io_op_defs[i].issue); 8708*cfd22e6bSJens Axboe WARN_ON_ONCE(!io_op_defs[i].name); 8709ed29b0b4SJens Axboe } 8710ed29b0b4SJens Axboe 8711ed29b0b4SJens Axboe req_cachep = KMEM_CACHE(io_kiocb, SLAB_HWCACHE_ALIGN | SLAB_PANIC | 8712ed29b0b4SJens Axboe SLAB_ACCOUNT); 8713ed29b0b4SJens Axboe return 0; 8714ed29b0b4SJens Axboe }; 8715ed29b0b4SJens Axboe __initcall(io_uring_init); 8716