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/namei.h> 74ed29b0b4SJens Axboe #include <linux/fsnotify.h> 75ed29b0b4SJens Axboe #include <linux/fadvise.h> 76ed29b0b4SJens Axboe #include <linux/eventpoll.h> 77ed29b0b4SJens Axboe #include <linux/splice.h> 78ed29b0b4SJens Axboe #include <linux/task_work.h> 79ed29b0b4SJens Axboe #include <linux/pagemap.h> 80ed29b0b4SJens Axboe #include <linux/io_uring.h> 81ed29b0b4SJens Axboe #include <linux/audit.h> 82ed29b0b4SJens Axboe #include <linux/security.h> 83ed29b0b4SJens Axboe #include <linux/xattr.h> 84ed29b0b4SJens Axboe 85ed29b0b4SJens Axboe #define CREATE_TRACE_POINTS 86ed29b0b4SJens Axboe #include <trace/events/io_uring.h> 87ed29b0b4SJens Axboe 88ed29b0b4SJens Axboe #include <uapi/linux/io_uring.h> 89ed29b0b4SJens Axboe 90ed29b0b4SJens Axboe #include "../fs/internal.h" 91ed29b0b4SJens Axboe #include "io-wq.h" 92ed29b0b4SJens Axboe 93ed29b0b4SJens Axboe #define IORING_MAX_ENTRIES 32768 94ed29b0b4SJens Axboe #define IORING_MAX_CQ_ENTRIES (2 * IORING_MAX_ENTRIES) 95ed29b0b4SJens Axboe #define IORING_SQPOLL_CAP_ENTRIES_VALUE 8 96ed29b0b4SJens Axboe 97ed29b0b4SJens Axboe /* only define max */ 98ed29b0b4SJens Axboe #define IORING_MAX_FIXED_FILES (1U << 20) 99ed29b0b4SJens Axboe #define IORING_MAX_RESTRICTIONS (IORING_RESTRICTION_LAST + \ 100ed29b0b4SJens Axboe IORING_REGISTER_LAST + IORING_OP_LAST) 101ed29b0b4SJens Axboe 102ed29b0b4SJens Axboe #define IO_RSRC_TAG_TABLE_SHIFT (PAGE_SHIFT - 3) 103ed29b0b4SJens Axboe #define IO_RSRC_TAG_TABLE_MAX (1U << IO_RSRC_TAG_TABLE_SHIFT) 104ed29b0b4SJens Axboe #define IO_RSRC_TAG_TABLE_MASK (IO_RSRC_TAG_TABLE_MAX - 1) 105ed29b0b4SJens Axboe 106ed29b0b4SJens Axboe #define IORING_MAX_REG_BUFFERS (1U << 14) 107ed29b0b4SJens Axboe 108ed29b0b4SJens Axboe #define SQE_COMMON_FLAGS (IOSQE_FIXED_FILE | IOSQE_IO_LINK | \ 109ed29b0b4SJens Axboe IOSQE_IO_HARDLINK | IOSQE_ASYNC) 110ed29b0b4SJens Axboe 111ed29b0b4SJens Axboe #define SQE_VALID_FLAGS (SQE_COMMON_FLAGS | IOSQE_BUFFER_SELECT | \ 112ed29b0b4SJens Axboe IOSQE_IO_DRAIN | IOSQE_CQE_SKIP_SUCCESS) 113ed29b0b4SJens Axboe 114ed29b0b4SJens Axboe #define IO_REQ_CLEAN_FLAGS (REQ_F_BUFFER_SELECTED | REQ_F_NEED_CLEANUP | \ 115ed29b0b4SJens Axboe REQ_F_POLLED | REQ_F_INFLIGHT | REQ_F_CREDS | \ 116ed29b0b4SJens Axboe REQ_F_ASYNC_DATA) 117ed29b0b4SJens Axboe 118ed29b0b4SJens Axboe #define IO_REQ_CLEAN_SLOW_FLAGS (REQ_F_REFCOUNT | REQ_F_LINK | REQ_F_HARDLINK |\ 119ed29b0b4SJens Axboe IO_REQ_CLEAN_FLAGS) 120ed29b0b4SJens Axboe 121ed29b0b4SJens Axboe #define IO_APOLL_MULTI_POLLED (REQ_F_APOLL_MULTISHOT | REQ_F_POLLED) 122ed29b0b4SJens Axboe 123ed29b0b4SJens Axboe #define IO_TCTX_REFS_CACHE_NR (1U << 10) 124ed29b0b4SJens Axboe 125ed29b0b4SJens Axboe struct io_uring { 126ed29b0b4SJens Axboe u32 head ____cacheline_aligned_in_smp; 127ed29b0b4SJens Axboe u32 tail ____cacheline_aligned_in_smp; 128ed29b0b4SJens Axboe }; 129ed29b0b4SJens Axboe 130ed29b0b4SJens Axboe /* 131ed29b0b4SJens Axboe * This data is shared with the application through the mmap at offsets 132ed29b0b4SJens Axboe * IORING_OFF_SQ_RING and IORING_OFF_CQ_RING. 133ed29b0b4SJens Axboe * 134ed29b0b4SJens Axboe * The offsets to the member fields are published through struct 135ed29b0b4SJens Axboe * io_sqring_offsets when calling io_uring_setup. 136ed29b0b4SJens Axboe */ 137ed29b0b4SJens Axboe struct io_rings { 138ed29b0b4SJens Axboe /* 139ed29b0b4SJens Axboe * Head and tail offsets into the ring; the offsets need to be 140ed29b0b4SJens Axboe * masked to get valid indices. 141ed29b0b4SJens Axboe * 142ed29b0b4SJens Axboe * The kernel controls head of the sq ring and the tail of the cq ring, 143ed29b0b4SJens Axboe * and the application controls tail of the sq ring and the head of the 144ed29b0b4SJens Axboe * cq ring. 145ed29b0b4SJens Axboe */ 146ed29b0b4SJens Axboe struct io_uring sq, cq; 147ed29b0b4SJens Axboe /* 148ed29b0b4SJens Axboe * Bitmasks to apply to head and tail offsets (constant, equals 149ed29b0b4SJens Axboe * ring_entries - 1) 150ed29b0b4SJens Axboe */ 151ed29b0b4SJens Axboe u32 sq_ring_mask, cq_ring_mask; 152ed29b0b4SJens Axboe /* Ring sizes (constant, power of 2) */ 153ed29b0b4SJens Axboe u32 sq_ring_entries, cq_ring_entries; 154ed29b0b4SJens Axboe /* 155ed29b0b4SJens Axboe * Number of invalid entries dropped by the kernel due to 156ed29b0b4SJens Axboe * invalid index stored in array 157ed29b0b4SJens Axboe * 158ed29b0b4SJens Axboe * Written by the kernel, shouldn't be modified by the 159ed29b0b4SJens Axboe * application (i.e. get number of "new events" by comparing to 160ed29b0b4SJens Axboe * cached value). 161ed29b0b4SJens Axboe * 162ed29b0b4SJens Axboe * After a new SQ head value was read by the application this 163ed29b0b4SJens Axboe * counter includes all submissions that were dropped reaching 164ed29b0b4SJens Axboe * the new SQ head (and possibly more). 165ed29b0b4SJens Axboe */ 166ed29b0b4SJens Axboe u32 sq_dropped; 167ed29b0b4SJens Axboe /* 168ed29b0b4SJens Axboe * Runtime SQ flags 169ed29b0b4SJens Axboe * 170ed29b0b4SJens Axboe * Written by the kernel, shouldn't be modified by the 171ed29b0b4SJens Axboe * application. 172ed29b0b4SJens Axboe * 173ed29b0b4SJens Axboe * The application needs a full memory barrier before checking 174ed29b0b4SJens Axboe * for IORING_SQ_NEED_WAKEUP after updating the sq tail. 175ed29b0b4SJens Axboe */ 176ed29b0b4SJens Axboe atomic_t sq_flags; 177ed29b0b4SJens Axboe /* 178ed29b0b4SJens Axboe * Runtime CQ flags 179ed29b0b4SJens Axboe * 180ed29b0b4SJens Axboe * Written by the application, shouldn't be modified by the 181ed29b0b4SJens Axboe * kernel. 182ed29b0b4SJens Axboe */ 183ed29b0b4SJens Axboe u32 cq_flags; 184ed29b0b4SJens Axboe /* 185ed29b0b4SJens Axboe * Number of completion events lost because the queue was full; 186ed29b0b4SJens Axboe * this should be avoided by the application by making sure 187ed29b0b4SJens Axboe * there are not more requests pending than there is space in 188ed29b0b4SJens Axboe * the completion queue. 189ed29b0b4SJens Axboe * 190ed29b0b4SJens Axboe * Written by the kernel, shouldn't be modified by the 191ed29b0b4SJens Axboe * application (i.e. get number of "new events" by comparing to 192ed29b0b4SJens Axboe * cached value). 193ed29b0b4SJens Axboe * 194ed29b0b4SJens Axboe * As completion events come in out of order this counter is not 195ed29b0b4SJens Axboe * ordered with any other data. 196ed29b0b4SJens Axboe */ 197ed29b0b4SJens Axboe u32 cq_overflow; 198ed29b0b4SJens Axboe /* 199ed29b0b4SJens Axboe * Ring buffer of completion events. 200ed29b0b4SJens Axboe * 201ed29b0b4SJens Axboe * The kernel writes completion events fresh every time they are 202ed29b0b4SJens Axboe * produced, so the application is allowed to modify pending 203ed29b0b4SJens Axboe * entries. 204ed29b0b4SJens Axboe */ 205ed29b0b4SJens Axboe struct io_uring_cqe cqes[] ____cacheline_aligned_in_smp; 206ed29b0b4SJens Axboe }; 207ed29b0b4SJens Axboe 208ed29b0b4SJens Axboe struct io_mapped_ubuf { 209ed29b0b4SJens Axboe u64 ubuf; 210ed29b0b4SJens Axboe u64 ubuf_end; 211ed29b0b4SJens Axboe unsigned int nr_bvecs; 212ed29b0b4SJens Axboe unsigned long acct_pages; 213ed29b0b4SJens Axboe struct bio_vec bvec[]; 214ed29b0b4SJens Axboe }; 215ed29b0b4SJens Axboe 216ed29b0b4SJens Axboe struct io_ring_ctx; 217ed29b0b4SJens Axboe 218ed29b0b4SJens Axboe struct io_overflow_cqe { 219ed29b0b4SJens Axboe struct list_head list; 220ed29b0b4SJens Axboe struct io_uring_cqe cqe; 221ed29b0b4SJens Axboe }; 222ed29b0b4SJens Axboe 223ed29b0b4SJens Axboe /* 224ed29b0b4SJens Axboe * FFS_SCM is only available on 64-bit archs, for 32-bit we just define it as 0 225ed29b0b4SJens Axboe * and define IO_URING_SCM_ALL. For this case, we use SCM for all files as we 226ed29b0b4SJens Axboe * can't safely always dereference the file when the task has exited and ring 227ed29b0b4SJens Axboe * cleanup is done. If a file is tracked and part of SCM, then unix gc on 228ed29b0b4SJens Axboe * process exit may reap it before __io_sqe_files_unregister() is run. 229ed29b0b4SJens Axboe */ 230ed29b0b4SJens Axboe #define FFS_NOWAIT 0x1UL 231ed29b0b4SJens Axboe #define FFS_ISREG 0x2UL 232ed29b0b4SJens Axboe #if defined(CONFIG_64BIT) 233ed29b0b4SJens Axboe #define FFS_SCM 0x4UL 234ed29b0b4SJens Axboe #else 235ed29b0b4SJens Axboe #define IO_URING_SCM_ALL 236ed29b0b4SJens Axboe #define FFS_SCM 0x0UL 237ed29b0b4SJens Axboe #endif 238ed29b0b4SJens Axboe #define FFS_MASK ~(FFS_NOWAIT|FFS_ISREG|FFS_SCM) 239ed29b0b4SJens Axboe 240ed29b0b4SJens Axboe struct io_fixed_file { 241ed29b0b4SJens Axboe /* file * with additional FFS_* flags */ 242ed29b0b4SJens Axboe unsigned long file_ptr; 243ed29b0b4SJens Axboe }; 244ed29b0b4SJens Axboe 245ed29b0b4SJens Axboe struct io_rsrc_put { 246ed29b0b4SJens Axboe struct list_head list; 247ed29b0b4SJens Axboe u64 tag; 248ed29b0b4SJens Axboe union { 249ed29b0b4SJens Axboe void *rsrc; 250ed29b0b4SJens Axboe struct file *file; 251ed29b0b4SJens Axboe struct io_mapped_ubuf *buf; 252ed29b0b4SJens Axboe }; 253ed29b0b4SJens Axboe }; 254ed29b0b4SJens Axboe 255ed29b0b4SJens Axboe struct io_file_table { 256ed29b0b4SJens Axboe struct io_fixed_file *files; 257ed29b0b4SJens Axboe unsigned long *bitmap; 258ed29b0b4SJens Axboe unsigned int alloc_hint; 259ed29b0b4SJens Axboe }; 260ed29b0b4SJens Axboe 261ed29b0b4SJens Axboe struct io_rsrc_node { 262ed29b0b4SJens Axboe struct percpu_ref refs; 263ed29b0b4SJens Axboe struct list_head node; 264ed29b0b4SJens Axboe struct list_head rsrc_list; 265ed29b0b4SJens Axboe struct io_rsrc_data *rsrc_data; 266ed29b0b4SJens Axboe struct llist_node llist; 267ed29b0b4SJens Axboe bool done; 268ed29b0b4SJens Axboe }; 269ed29b0b4SJens Axboe 270ed29b0b4SJens Axboe typedef void (rsrc_put_fn)(struct io_ring_ctx *ctx, struct io_rsrc_put *prsrc); 271ed29b0b4SJens Axboe 272ed29b0b4SJens Axboe struct io_rsrc_data { 273ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 274ed29b0b4SJens Axboe 275ed29b0b4SJens Axboe u64 **tags; 276ed29b0b4SJens Axboe unsigned int nr; 277ed29b0b4SJens Axboe rsrc_put_fn *do_put; 278ed29b0b4SJens Axboe atomic_t refs; 279ed29b0b4SJens Axboe struct completion done; 280ed29b0b4SJens Axboe bool quiesce; 281ed29b0b4SJens Axboe }; 282ed29b0b4SJens Axboe 283ed29b0b4SJens Axboe #define IO_BUFFER_LIST_BUF_PER_PAGE (PAGE_SIZE / sizeof(struct io_uring_buf)) 284ed29b0b4SJens Axboe struct io_buffer_list { 285ed29b0b4SJens Axboe /* 286ed29b0b4SJens Axboe * If ->buf_nr_pages is set, then buf_pages/buf_ring are used. If not, 287ed29b0b4SJens Axboe * then these are classic provided buffers and ->buf_list is used. 288ed29b0b4SJens Axboe */ 289ed29b0b4SJens Axboe union { 290ed29b0b4SJens Axboe struct list_head buf_list; 291ed29b0b4SJens Axboe struct { 292ed29b0b4SJens Axboe struct page **buf_pages; 293ed29b0b4SJens Axboe struct io_uring_buf_ring *buf_ring; 294ed29b0b4SJens Axboe }; 295ed29b0b4SJens Axboe }; 296ed29b0b4SJens Axboe __u16 bgid; 297ed29b0b4SJens Axboe 298ed29b0b4SJens Axboe /* below is for ring provided buffers */ 299ed29b0b4SJens Axboe __u16 buf_nr_pages; 300ed29b0b4SJens Axboe __u16 nr_entries; 301ed29b0b4SJens Axboe __u16 head; 302ed29b0b4SJens Axboe __u16 mask; 303ed29b0b4SJens Axboe }; 304ed29b0b4SJens Axboe 305ed29b0b4SJens Axboe struct io_buffer { 306ed29b0b4SJens Axboe struct list_head list; 307ed29b0b4SJens Axboe __u64 addr; 308ed29b0b4SJens Axboe __u32 len; 309ed29b0b4SJens Axboe __u16 bid; 310ed29b0b4SJens Axboe __u16 bgid; 311ed29b0b4SJens Axboe }; 312ed29b0b4SJens Axboe 313ed29b0b4SJens Axboe struct io_restriction { 314ed29b0b4SJens Axboe DECLARE_BITMAP(register_op, IORING_REGISTER_LAST); 315ed29b0b4SJens Axboe DECLARE_BITMAP(sqe_op, IORING_OP_LAST); 316ed29b0b4SJens Axboe u8 sqe_flags_allowed; 317ed29b0b4SJens Axboe u8 sqe_flags_required; 318ed29b0b4SJens Axboe bool registered; 319ed29b0b4SJens Axboe }; 320ed29b0b4SJens Axboe 321ed29b0b4SJens Axboe enum { 322ed29b0b4SJens Axboe IO_SQ_THREAD_SHOULD_STOP = 0, 323ed29b0b4SJens Axboe IO_SQ_THREAD_SHOULD_PARK, 324ed29b0b4SJens Axboe }; 325ed29b0b4SJens Axboe 326ed29b0b4SJens Axboe struct io_sq_data { 327ed29b0b4SJens Axboe refcount_t refs; 328ed29b0b4SJens Axboe atomic_t park_pending; 329ed29b0b4SJens Axboe struct mutex lock; 330ed29b0b4SJens Axboe 331ed29b0b4SJens Axboe /* ctx's that are using this sqd */ 332ed29b0b4SJens Axboe struct list_head ctx_list; 333ed29b0b4SJens Axboe 334ed29b0b4SJens Axboe struct task_struct *thread; 335ed29b0b4SJens Axboe struct wait_queue_head wait; 336ed29b0b4SJens Axboe 337ed29b0b4SJens Axboe unsigned sq_thread_idle; 338ed29b0b4SJens Axboe int sq_cpu; 339ed29b0b4SJens Axboe pid_t task_pid; 340ed29b0b4SJens Axboe pid_t task_tgid; 341ed29b0b4SJens Axboe 342ed29b0b4SJens Axboe unsigned long state; 343ed29b0b4SJens Axboe struct completion exited; 344ed29b0b4SJens Axboe }; 345ed29b0b4SJens Axboe 346ed29b0b4SJens Axboe #define IO_COMPL_BATCH 32 347ed29b0b4SJens Axboe #define IO_REQ_CACHE_SIZE 32 348ed29b0b4SJens Axboe #define IO_REQ_ALLOC_BATCH 8 349ed29b0b4SJens Axboe 350ed29b0b4SJens Axboe struct io_submit_link { 351ed29b0b4SJens Axboe struct io_kiocb *head; 352ed29b0b4SJens Axboe struct io_kiocb *last; 353ed29b0b4SJens Axboe }; 354ed29b0b4SJens Axboe 355ed29b0b4SJens Axboe struct io_submit_state { 356ed29b0b4SJens Axboe /* inline/task_work completion list, under ->uring_lock */ 357ed29b0b4SJens Axboe struct io_wq_work_node free_list; 358ed29b0b4SJens Axboe /* batch completion logic */ 359ed29b0b4SJens Axboe struct io_wq_work_list compl_reqs; 360ed29b0b4SJens Axboe struct io_submit_link link; 361ed29b0b4SJens Axboe 362ed29b0b4SJens Axboe bool plug_started; 363ed29b0b4SJens Axboe bool need_plug; 364ed29b0b4SJens Axboe bool flush_cqes; 365ed29b0b4SJens Axboe unsigned short submit_nr; 366ed29b0b4SJens Axboe struct blk_plug plug; 367ed29b0b4SJens Axboe }; 368ed29b0b4SJens Axboe 369ed29b0b4SJens Axboe struct io_ev_fd { 370ed29b0b4SJens Axboe struct eventfd_ctx *cq_ev_fd; 371ed29b0b4SJens Axboe unsigned int eventfd_async: 1; 372ed29b0b4SJens Axboe struct rcu_head rcu; 373ed29b0b4SJens Axboe }; 374ed29b0b4SJens Axboe 375ed29b0b4SJens Axboe #define BGID_ARRAY 64 376ed29b0b4SJens Axboe 377ed29b0b4SJens Axboe struct io_ring_ctx { 378ed29b0b4SJens Axboe /* const or read-mostly hot data */ 379ed29b0b4SJens Axboe struct { 380ed29b0b4SJens Axboe struct percpu_ref refs; 381ed29b0b4SJens Axboe 382ed29b0b4SJens Axboe struct io_rings *rings; 383ed29b0b4SJens Axboe unsigned int flags; 384ed29b0b4SJens Axboe enum task_work_notify_mode notify_method; 385ed29b0b4SJens Axboe unsigned int compat: 1; 386ed29b0b4SJens Axboe unsigned int drain_next: 1; 387ed29b0b4SJens Axboe unsigned int restricted: 1; 388ed29b0b4SJens Axboe unsigned int off_timeout_used: 1; 389ed29b0b4SJens Axboe unsigned int drain_active: 1; 390ed29b0b4SJens Axboe unsigned int drain_disabled: 1; 391ed29b0b4SJens Axboe unsigned int has_evfd: 1; 392ed29b0b4SJens Axboe unsigned int syscall_iopoll: 1; 393ed29b0b4SJens Axboe } ____cacheline_aligned_in_smp; 394ed29b0b4SJens Axboe 395ed29b0b4SJens Axboe /* submission data */ 396ed29b0b4SJens Axboe struct { 397ed29b0b4SJens Axboe struct mutex uring_lock; 398ed29b0b4SJens Axboe 399ed29b0b4SJens Axboe /* 400ed29b0b4SJens Axboe * Ring buffer of indices into array of io_uring_sqe, which is 401ed29b0b4SJens Axboe * mmapped by the application using the IORING_OFF_SQES offset. 402ed29b0b4SJens Axboe * 403ed29b0b4SJens Axboe * This indirection could e.g. be used to assign fixed 404ed29b0b4SJens Axboe * io_uring_sqe entries to operations and only submit them to 405ed29b0b4SJens Axboe * the queue when needed. 406ed29b0b4SJens Axboe * 407ed29b0b4SJens Axboe * The kernel modifies neither the indices array nor the entries 408ed29b0b4SJens Axboe * array. 409ed29b0b4SJens Axboe */ 410ed29b0b4SJens Axboe u32 *sq_array; 411ed29b0b4SJens Axboe struct io_uring_sqe *sq_sqes; 412ed29b0b4SJens Axboe unsigned cached_sq_head; 413ed29b0b4SJens Axboe unsigned sq_entries; 414ed29b0b4SJens Axboe struct list_head defer_list; 415ed29b0b4SJens Axboe 416ed29b0b4SJens Axboe /* 417ed29b0b4SJens Axboe * Fixed resources fast path, should be accessed only under 418ed29b0b4SJens Axboe * uring_lock, and updated through io_uring_register(2) 419ed29b0b4SJens Axboe */ 420ed29b0b4SJens Axboe struct io_rsrc_node *rsrc_node; 421ed29b0b4SJens Axboe int rsrc_cached_refs; 422ed29b0b4SJens Axboe atomic_t cancel_seq; 423ed29b0b4SJens Axboe struct io_file_table file_table; 424ed29b0b4SJens Axboe unsigned nr_user_files; 425ed29b0b4SJens Axboe unsigned nr_user_bufs; 426ed29b0b4SJens Axboe struct io_mapped_ubuf **user_bufs; 427ed29b0b4SJens Axboe 428ed29b0b4SJens Axboe struct io_submit_state submit_state; 429ed29b0b4SJens Axboe 430ed29b0b4SJens Axboe struct io_buffer_list *io_bl; 431ed29b0b4SJens Axboe struct xarray io_bl_xa; 432ed29b0b4SJens Axboe struct list_head io_buffers_cache; 433ed29b0b4SJens Axboe 434ed29b0b4SJens Axboe struct list_head timeout_list; 435ed29b0b4SJens Axboe struct list_head ltimeout_list; 436ed29b0b4SJens Axboe struct list_head cq_overflow_list; 437ed29b0b4SJens Axboe struct list_head apoll_cache; 438ed29b0b4SJens Axboe struct xarray personalities; 439ed29b0b4SJens Axboe u32 pers_next; 440ed29b0b4SJens Axboe unsigned sq_thread_idle; 441ed29b0b4SJens Axboe } ____cacheline_aligned_in_smp; 442ed29b0b4SJens Axboe 443ed29b0b4SJens Axboe /* IRQ completion list, under ->completion_lock */ 444ed29b0b4SJens Axboe struct io_wq_work_list locked_free_list; 445ed29b0b4SJens Axboe unsigned int locked_free_nr; 446ed29b0b4SJens Axboe 447ed29b0b4SJens Axboe const struct cred *sq_creds; /* cred used for __io_sq_thread() */ 448ed29b0b4SJens Axboe struct io_sq_data *sq_data; /* if using sq thread polling */ 449ed29b0b4SJens Axboe 450ed29b0b4SJens Axboe struct wait_queue_head sqo_sq_wait; 451ed29b0b4SJens Axboe struct list_head sqd_list; 452ed29b0b4SJens Axboe 453ed29b0b4SJens Axboe unsigned long check_cq; 454ed29b0b4SJens Axboe 455ed29b0b4SJens Axboe struct { 456ed29b0b4SJens Axboe /* 457ed29b0b4SJens Axboe * We cache a range of free CQEs we can use, once exhausted it 458ed29b0b4SJens Axboe * should go through a slower range setup, see __io_get_cqe() 459ed29b0b4SJens Axboe */ 460ed29b0b4SJens Axboe struct io_uring_cqe *cqe_cached; 461ed29b0b4SJens Axboe struct io_uring_cqe *cqe_sentinel; 462ed29b0b4SJens Axboe 463ed29b0b4SJens Axboe unsigned cached_cq_tail; 464ed29b0b4SJens Axboe unsigned cq_entries; 465ed29b0b4SJens Axboe struct io_ev_fd __rcu *io_ev_fd; 466ed29b0b4SJens Axboe struct wait_queue_head cq_wait; 467ed29b0b4SJens Axboe unsigned cq_extra; 468ed29b0b4SJens Axboe atomic_t cq_timeouts; 469ed29b0b4SJens Axboe unsigned cq_last_tm_flush; 470ed29b0b4SJens Axboe } ____cacheline_aligned_in_smp; 471ed29b0b4SJens Axboe 472ed29b0b4SJens Axboe struct { 473ed29b0b4SJens Axboe spinlock_t completion_lock; 474ed29b0b4SJens Axboe 475ed29b0b4SJens Axboe spinlock_t timeout_lock; 476ed29b0b4SJens Axboe 477ed29b0b4SJens Axboe /* 478ed29b0b4SJens Axboe * ->iopoll_list is protected by the ctx->uring_lock for 479ed29b0b4SJens Axboe * io_uring instances that don't use IORING_SETUP_SQPOLL. 480ed29b0b4SJens Axboe * For SQPOLL, only the single threaded io_sq_thread() will 481ed29b0b4SJens Axboe * manipulate the list, hence no extra locking is needed there. 482ed29b0b4SJens Axboe */ 483ed29b0b4SJens Axboe struct io_wq_work_list iopoll_list; 484ed29b0b4SJens Axboe struct hlist_head *cancel_hash; 485ed29b0b4SJens Axboe unsigned cancel_hash_bits; 486ed29b0b4SJens Axboe bool poll_multi_queue; 487ed29b0b4SJens Axboe 488ed29b0b4SJens Axboe struct list_head io_buffers_comp; 489ed29b0b4SJens Axboe } ____cacheline_aligned_in_smp; 490ed29b0b4SJens Axboe 491ed29b0b4SJens Axboe struct io_restriction restrictions; 492ed29b0b4SJens Axboe 493ed29b0b4SJens Axboe /* slow path rsrc auxilary data, used by update/register */ 494ed29b0b4SJens Axboe struct { 495ed29b0b4SJens Axboe struct io_rsrc_node *rsrc_backup_node; 496ed29b0b4SJens Axboe struct io_mapped_ubuf *dummy_ubuf; 497ed29b0b4SJens Axboe struct io_rsrc_data *file_data; 498ed29b0b4SJens Axboe struct io_rsrc_data *buf_data; 499ed29b0b4SJens Axboe 500ed29b0b4SJens Axboe struct delayed_work rsrc_put_work; 501ed29b0b4SJens Axboe struct llist_head rsrc_put_llist; 502ed29b0b4SJens Axboe struct list_head rsrc_ref_list; 503ed29b0b4SJens Axboe spinlock_t rsrc_ref_lock; 504ed29b0b4SJens Axboe 505ed29b0b4SJens Axboe struct list_head io_buffers_pages; 506ed29b0b4SJens Axboe }; 507ed29b0b4SJens Axboe 508ed29b0b4SJens Axboe /* Keep this last, we don't need it for the fast path */ 509ed29b0b4SJens Axboe struct { 510ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 511ed29b0b4SJens Axboe struct socket *ring_sock; 512ed29b0b4SJens Axboe #endif 513ed29b0b4SJens Axboe /* hashed buffered write serialization */ 514ed29b0b4SJens Axboe struct io_wq_hash *hash_map; 515ed29b0b4SJens Axboe 516ed29b0b4SJens Axboe /* Only used for accounting purposes */ 517ed29b0b4SJens Axboe struct user_struct *user; 518ed29b0b4SJens Axboe struct mm_struct *mm_account; 519ed29b0b4SJens Axboe 520ed29b0b4SJens Axboe /* ctx exit and cancelation */ 521ed29b0b4SJens Axboe struct llist_head fallback_llist; 522ed29b0b4SJens Axboe struct delayed_work fallback_work; 523ed29b0b4SJens Axboe struct work_struct exit_work; 524ed29b0b4SJens Axboe struct list_head tctx_list; 525ed29b0b4SJens Axboe struct completion ref_comp; 526ed29b0b4SJens Axboe u32 iowq_limits[2]; 527ed29b0b4SJens Axboe bool iowq_limits_set; 528ed29b0b4SJens Axboe }; 529ed29b0b4SJens Axboe }; 530ed29b0b4SJens Axboe 531ed29b0b4SJens Axboe /* 532ed29b0b4SJens Axboe * Arbitrary limit, can be raised if need be 533ed29b0b4SJens Axboe */ 534ed29b0b4SJens Axboe #define IO_RINGFD_REG_MAX 16 535ed29b0b4SJens Axboe 536ed29b0b4SJens Axboe struct io_uring_task { 537ed29b0b4SJens Axboe /* submission side */ 538ed29b0b4SJens Axboe int cached_refs; 539ed29b0b4SJens Axboe struct xarray xa; 540ed29b0b4SJens Axboe struct wait_queue_head wait; 541ed29b0b4SJens Axboe const struct io_ring_ctx *last; 542ed29b0b4SJens Axboe struct io_wq *io_wq; 543ed29b0b4SJens Axboe struct percpu_counter inflight; 544ed29b0b4SJens Axboe atomic_t inflight_tracked; 545ed29b0b4SJens Axboe atomic_t in_idle; 546ed29b0b4SJens Axboe 547ed29b0b4SJens Axboe spinlock_t task_lock; 548ed29b0b4SJens Axboe struct io_wq_work_list task_list; 549ed29b0b4SJens Axboe struct io_wq_work_list prio_task_list; 550ed29b0b4SJens Axboe struct callback_head task_work; 551ed29b0b4SJens Axboe struct file **registered_rings; 552ed29b0b4SJens Axboe bool task_running; 553ed29b0b4SJens Axboe }; 554ed29b0b4SJens Axboe 555ed29b0b4SJens Axboe /* 556ed29b0b4SJens Axboe * First field must be the file pointer in all the 557ed29b0b4SJens Axboe * iocb unions! See also 'struct kiocb' in <linux/fs.h> 558ed29b0b4SJens Axboe */ 5598d4388d1SJens Axboe struct io_poll { 560ed29b0b4SJens Axboe struct file *file; 561ed29b0b4SJens Axboe struct wait_queue_head *head; 562ed29b0b4SJens Axboe __poll_t events; 563ed29b0b4SJens Axboe struct wait_queue_entry wait; 564ed29b0b4SJens Axboe }; 565ed29b0b4SJens Axboe 566ed29b0b4SJens Axboe struct io_poll_update { 567ed29b0b4SJens Axboe struct file *file; 568ed29b0b4SJens Axboe u64 old_user_data; 569ed29b0b4SJens Axboe u64 new_user_data; 570ed29b0b4SJens Axboe __poll_t events; 571ed29b0b4SJens Axboe bool update_events; 572ed29b0b4SJens Axboe bool update_user_data; 573ed29b0b4SJens Axboe }; 574ed29b0b4SJens Axboe 575ed29b0b4SJens Axboe struct io_close { 576ed29b0b4SJens Axboe struct file *file; 577ed29b0b4SJens Axboe int fd; 578ed29b0b4SJens Axboe u32 file_slot; 579ed29b0b4SJens Axboe }; 580ed29b0b4SJens Axboe 581ed29b0b4SJens Axboe struct io_timeout_data { 582ed29b0b4SJens Axboe struct io_kiocb *req; 583ed29b0b4SJens Axboe struct hrtimer timer; 584ed29b0b4SJens Axboe struct timespec64 ts; 585ed29b0b4SJens Axboe enum hrtimer_mode mode; 586ed29b0b4SJens Axboe u32 flags; 587ed29b0b4SJens Axboe }; 588ed29b0b4SJens Axboe 589ed29b0b4SJens Axboe struct io_accept { 590ed29b0b4SJens Axboe struct file *file; 591ed29b0b4SJens Axboe struct sockaddr __user *addr; 592ed29b0b4SJens Axboe int __user *addr_len; 593ed29b0b4SJens Axboe int flags; 594ed29b0b4SJens Axboe u32 file_slot; 595ed29b0b4SJens Axboe unsigned long nofile; 596ed29b0b4SJens Axboe }; 597ed29b0b4SJens Axboe 598ed29b0b4SJens Axboe struct io_socket { 599ed29b0b4SJens Axboe struct file *file; 600ed29b0b4SJens Axboe int domain; 601ed29b0b4SJens Axboe int type; 602ed29b0b4SJens Axboe int protocol; 603ed29b0b4SJens Axboe int flags; 604ed29b0b4SJens Axboe u32 file_slot; 605ed29b0b4SJens Axboe unsigned long nofile; 606ed29b0b4SJens Axboe }; 607ed29b0b4SJens Axboe 608ed29b0b4SJens Axboe struct io_sync { 609ed29b0b4SJens Axboe struct file *file; 610ed29b0b4SJens Axboe loff_t len; 611ed29b0b4SJens Axboe loff_t off; 612ed29b0b4SJens Axboe int flags; 613ed29b0b4SJens Axboe int mode; 614ed29b0b4SJens Axboe }; 615ed29b0b4SJens Axboe 616ed29b0b4SJens Axboe struct io_cancel { 617ed29b0b4SJens Axboe struct file *file; 618ed29b0b4SJens Axboe u64 addr; 619ed29b0b4SJens Axboe u32 flags; 620ed29b0b4SJens Axboe s32 fd; 621ed29b0b4SJens Axboe }; 622ed29b0b4SJens Axboe 623ed29b0b4SJens Axboe struct io_timeout { 624ed29b0b4SJens Axboe struct file *file; 625ed29b0b4SJens Axboe u32 off; 626ed29b0b4SJens Axboe u32 target_seq; 627ed29b0b4SJens Axboe struct list_head list; 628ed29b0b4SJens Axboe /* head of the link, used by linked timeouts only */ 629ed29b0b4SJens Axboe struct io_kiocb *head; 630ed29b0b4SJens Axboe /* for linked completions */ 631ed29b0b4SJens Axboe struct io_kiocb *prev; 632ed29b0b4SJens Axboe }; 633ed29b0b4SJens Axboe 634ed29b0b4SJens Axboe struct io_timeout_rem { 635ed29b0b4SJens Axboe struct file *file; 636ed29b0b4SJens Axboe u64 addr; 637ed29b0b4SJens Axboe 638ed29b0b4SJens Axboe /* timeout update */ 639ed29b0b4SJens Axboe struct timespec64 ts; 640ed29b0b4SJens Axboe u32 flags; 641ed29b0b4SJens Axboe bool ltimeout; 642ed29b0b4SJens Axboe }; 643ed29b0b4SJens Axboe 644ed29b0b4SJens Axboe struct io_rw { 645ed29b0b4SJens Axboe /* NOTE: kiocb has the file as the first member, so don't do it here */ 646ed29b0b4SJens Axboe struct kiocb kiocb; 647ed29b0b4SJens Axboe u64 addr; 648ed29b0b4SJens Axboe u32 len; 649ed29b0b4SJens Axboe rwf_t flags; 650ed29b0b4SJens Axboe }; 651ed29b0b4SJens Axboe 652ed29b0b4SJens Axboe struct io_connect { 653ed29b0b4SJens Axboe struct file *file; 654ed29b0b4SJens Axboe struct sockaddr __user *addr; 655ed29b0b4SJens Axboe int addr_len; 656ed29b0b4SJens Axboe }; 657ed29b0b4SJens Axboe 658ed29b0b4SJens Axboe struct io_sr_msg { 659ed29b0b4SJens Axboe struct file *file; 660ed29b0b4SJens Axboe union { 661ed29b0b4SJens Axboe struct compat_msghdr __user *umsg_compat; 662ed29b0b4SJens Axboe struct user_msghdr __user *umsg; 663ed29b0b4SJens Axboe void __user *buf; 664ed29b0b4SJens Axboe }; 665ed29b0b4SJens Axboe int msg_flags; 666ed29b0b4SJens Axboe size_t len; 667ed29b0b4SJens Axboe size_t done_io; 668ed29b0b4SJens Axboe unsigned int flags; 669ed29b0b4SJens Axboe }; 670ed29b0b4SJens Axboe 671ed29b0b4SJens Axboe struct io_open { 672ed29b0b4SJens Axboe struct file *file; 673ed29b0b4SJens Axboe int dfd; 674ed29b0b4SJens Axboe u32 file_slot; 675ed29b0b4SJens Axboe struct filename *filename; 676ed29b0b4SJens Axboe struct open_how how; 677ed29b0b4SJens Axboe unsigned long nofile; 678ed29b0b4SJens Axboe }; 679ed29b0b4SJens Axboe 680ed29b0b4SJens Axboe struct io_rsrc_update { 681ed29b0b4SJens Axboe struct file *file; 682ed29b0b4SJens Axboe u64 arg; 683ed29b0b4SJens Axboe u32 nr_args; 684ed29b0b4SJens Axboe u32 offset; 685ed29b0b4SJens Axboe }; 686ed29b0b4SJens Axboe 687ed29b0b4SJens Axboe struct io_fadvise { 688ed29b0b4SJens Axboe struct file *file; 689ed29b0b4SJens Axboe u64 offset; 690ed29b0b4SJens Axboe u32 len; 691ed29b0b4SJens Axboe u32 advice; 692ed29b0b4SJens Axboe }; 693ed29b0b4SJens Axboe 694ed29b0b4SJens Axboe struct io_madvise { 695ed29b0b4SJens Axboe struct file *file; 696ed29b0b4SJens Axboe u64 addr; 697ed29b0b4SJens Axboe u32 len; 698ed29b0b4SJens Axboe u32 advice; 699ed29b0b4SJens Axboe }; 700ed29b0b4SJens Axboe 701ed29b0b4SJens Axboe struct io_epoll { 702ed29b0b4SJens Axboe struct file *file; 703ed29b0b4SJens Axboe int epfd; 704ed29b0b4SJens Axboe int op; 705ed29b0b4SJens Axboe int fd; 706ed29b0b4SJens Axboe struct epoll_event event; 707ed29b0b4SJens Axboe }; 708ed29b0b4SJens Axboe 709ed29b0b4SJens Axboe struct io_splice { 710ed29b0b4SJens Axboe struct file *file_out; 711ed29b0b4SJens Axboe loff_t off_out; 712ed29b0b4SJens Axboe loff_t off_in; 713ed29b0b4SJens Axboe u64 len; 714ed29b0b4SJens Axboe int splice_fd_in; 715ed29b0b4SJens Axboe unsigned int flags; 716ed29b0b4SJens Axboe }; 717ed29b0b4SJens Axboe 718ed29b0b4SJens Axboe struct io_provide_buf { 719ed29b0b4SJens Axboe struct file *file; 720ed29b0b4SJens Axboe __u64 addr; 721ed29b0b4SJens Axboe __u32 len; 722ed29b0b4SJens Axboe __u32 bgid; 723ed29b0b4SJens Axboe __u16 nbufs; 724ed29b0b4SJens Axboe __u16 bid; 725ed29b0b4SJens Axboe }; 726ed29b0b4SJens Axboe 727ed29b0b4SJens Axboe struct io_statx { 728ed29b0b4SJens Axboe struct file *file; 729ed29b0b4SJens Axboe int dfd; 730ed29b0b4SJens Axboe unsigned int mask; 731ed29b0b4SJens Axboe unsigned int flags; 732ed29b0b4SJens Axboe struct filename *filename; 733ed29b0b4SJens Axboe struct statx __user *buffer; 734ed29b0b4SJens Axboe }; 735ed29b0b4SJens Axboe 736ed29b0b4SJens Axboe struct io_shutdown { 737ed29b0b4SJens Axboe struct file *file; 738ed29b0b4SJens Axboe int how; 739ed29b0b4SJens Axboe }; 740ed29b0b4SJens Axboe 741ed29b0b4SJens Axboe struct io_rename { 742ed29b0b4SJens Axboe struct file *file; 743ed29b0b4SJens Axboe int old_dfd; 744ed29b0b4SJens Axboe int new_dfd; 745ed29b0b4SJens Axboe struct filename *oldpath; 746ed29b0b4SJens Axboe struct filename *newpath; 747ed29b0b4SJens Axboe int flags; 748ed29b0b4SJens Axboe }; 749ed29b0b4SJens Axboe 750ed29b0b4SJens Axboe struct io_unlink { 751ed29b0b4SJens Axboe struct file *file; 752ed29b0b4SJens Axboe int dfd; 753ed29b0b4SJens Axboe int flags; 754ed29b0b4SJens Axboe struct filename *filename; 755ed29b0b4SJens Axboe }; 756ed29b0b4SJens Axboe 757ed29b0b4SJens Axboe struct io_mkdir { 758ed29b0b4SJens Axboe struct file *file; 759ed29b0b4SJens Axboe int dfd; 760ed29b0b4SJens Axboe umode_t mode; 761ed29b0b4SJens Axboe struct filename *filename; 762ed29b0b4SJens Axboe }; 763ed29b0b4SJens Axboe 764ed29b0b4SJens Axboe struct io_symlink { 765ed29b0b4SJens Axboe struct file *file; 766ed29b0b4SJens Axboe int new_dfd; 767ed29b0b4SJens Axboe struct filename *oldpath; 768ed29b0b4SJens Axboe struct filename *newpath; 769ed29b0b4SJens Axboe }; 770ed29b0b4SJens Axboe 771ed29b0b4SJens Axboe struct io_hardlink { 772ed29b0b4SJens Axboe struct file *file; 773ed29b0b4SJens Axboe int old_dfd; 774ed29b0b4SJens Axboe int new_dfd; 775ed29b0b4SJens Axboe struct filename *oldpath; 776ed29b0b4SJens Axboe struct filename *newpath; 777ed29b0b4SJens Axboe int flags; 778ed29b0b4SJens Axboe }; 779ed29b0b4SJens Axboe 780ed29b0b4SJens Axboe struct io_msg { 781ed29b0b4SJens Axboe struct file *file; 782ed29b0b4SJens Axboe u64 user_data; 783ed29b0b4SJens Axboe u32 len; 784ed29b0b4SJens Axboe }; 785ed29b0b4SJens Axboe 786ed29b0b4SJens Axboe struct io_async_connect { 787ed29b0b4SJens Axboe struct sockaddr_storage address; 788ed29b0b4SJens Axboe }; 789ed29b0b4SJens Axboe 790ed29b0b4SJens Axboe struct io_async_msghdr { 791ed29b0b4SJens Axboe struct iovec fast_iov[UIO_FASTIOV]; 792ed29b0b4SJens Axboe /* points to an allocated iov, if NULL we use fast_iov instead */ 793ed29b0b4SJens Axboe struct iovec *free_iov; 794ed29b0b4SJens Axboe struct sockaddr __user *uaddr; 795ed29b0b4SJens Axboe struct msghdr msg; 796ed29b0b4SJens Axboe struct sockaddr_storage addr; 797ed29b0b4SJens Axboe }; 798ed29b0b4SJens Axboe 799ed29b0b4SJens Axboe struct io_rw_state { 800ed29b0b4SJens Axboe struct iov_iter iter; 801ed29b0b4SJens Axboe struct iov_iter_state iter_state; 802ed29b0b4SJens Axboe struct iovec fast_iov[UIO_FASTIOV]; 803ed29b0b4SJens Axboe }; 804ed29b0b4SJens Axboe 805ed29b0b4SJens Axboe struct io_async_rw { 806ed29b0b4SJens Axboe struct io_rw_state s; 807ed29b0b4SJens Axboe const struct iovec *free_iovec; 808ed29b0b4SJens Axboe size_t bytes_done; 809ed29b0b4SJens Axboe struct wait_page_queue wpq; 810ed29b0b4SJens Axboe }; 811ed29b0b4SJens Axboe 812ed29b0b4SJens Axboe struct io_xattr { 813ed29b0b4SJens Axboe struct file *file; 814ed29b0b4SJens Axboe struct xattr_ctx ctx; 815ed29b0b4SJens Axboe struct filename *filename; 816ed29b0b4SJens Axboe }; 817ed29b0b4SJens Axboe 818ed29b0b4SJens Axboe enum { 819ed29b0b4SJens Axboe REQ_F_FIXED_FILE_BIT = IOSQE_FIXED_FILE_BIT, 820ed29b0b4SJens Axboe REQ_F_IO_DRAIN_BIT = IOSQE_IO_DRAIN_BIT, 821ed29b0b4SJens Axboe REQ_F_LINK_BIT = IOSQE_IO_LINK_BIT, 822ed29b0b4SJens Axboe REQ_F_HARDLINK_BIT = IOSQE_IO_HARDLINK_BIT, 823ed29b0b4SJens Axboe REQ_F_FORCE_ASYNC_BIT = IOSQE_ASYNC_BIT, 824ed29b0b4SJens Axboe REQ_F_BUFFER_SELECT_BIT = IOSQE_BUFFER_SELECT_BIT, 825ed29b0b4SJens Axboe REQ_F_CQE_SKIP_BIT = IOSQE_CQE_SKIP_SUCCESS_BIT, 826ed29b0b4SJens Axboe 827ed29b0b4SJens Axboe /* first byte is taken by user flags, shift it to not overlap */ 828ed29b0b4SJens Axboe REQ_F_FAIL_BIT = 8, 829ed29b0b4SJens Axboe REQ_F_INFLIGHT_BIT, 830ed29b0b4SJens Axboe REQ_F_CUR_POS_BIT, 831ed29b0b4SJens Axboe REQ_F_NOWAIT_BIT, 832ed29b0b4SJens Axboe REQ_F_LINK_TIMEOUT_BIT, 833ed29b0b4SJens Axboe REQ_F_NEED_CLEANUP_BIT, 834ed29b0b4SJens Axboe REQ_F_POLLED_BIT, 835ed29b0b4SJens Axboe REQ_F_BUFFER_SELECTED_BIT, 836ed29b0b4SJens Axboe REQ_F_BUFFER_RING_BIT, 837ed29b0b4SJens Axboe REQ_F_COMPLETE_INLINE_BIT, 838ed29b0b4SJens Axboe REQ_F_REISSUE_BIT, 839ed29b0b4SJens Axboe REQ_F_CREDS_BIT, 840ed29b0b4SJens Axboe REQ_F_REFCOUNT_BIT, 841ed29b0b4SJens Axboe REQ_F_ARM_LTIMEOUT_BIT, 842ed29b0b4SJens Axboe REQ_F_ASYNC_DATA_BIT, 843ed29b0b4SJens Axboe REQ_F_SKIP_LINK_CQES_BIT, 844ed29b0b4SJens Axboe REQ_F_SINGLE_POLL_BIT, 845ed29b0b4SJens Axboe REQ_F_DOUBLE_POLL_BIT, 846ed29b0b4SJens Axboe REQ_F_PARTIAL_IO_BIT, 847ed29b0b4SJens Axboe REQ_F_CQE32_INIT_BIT, 848ed29b0b4SJens Axboe REQ_F_APOLL_MULTISHOT_BIT, 849bd8587e4SJens Axboe REQ_F_CLEAR_POLLIN_BIT, 850ed29b0b4SJens Axboe /* keep async read/write and isreg together and in order */ 851ed29b0b4SJens Axboe REQ_F_SUPPORT_NOWAIT_BIT, 852ed29b0b4SJens Axboe REQ_F_ISREG_BIT, 853ed29b0b4SJens Axboe 854ed29b0b4SJens Axboe /* not a real bit, just to check we're not overflowing the space */ 855ed29b0b4SJens Axboe __REQ_F_LAST_BIT, 856ed29b0b4SJens Axboe }; 857ed29b0b4SJens Axboe 858ed29b0b4SJens Axboe enum { 859ed29b0b4SJens Axboe /* ctx owns file */ 860ed29b0b4SJens Axboe REQ_F_FIXED_FILE = BIT(REQ_F_FIXED_FILE_BIT), 861ed29b0b4SJens Axboe /* drain existing IO first */ 862ed29b0b4SJens Axboe REQ_F_IO_DRAIN = BIT(REQ_F_IO_DRAIN_BIT), 863ed29b0b4SJens Axboe /* linked sqes */ 864ed29b0b4SJens Axboe REQ_F_LINK = BIT(REQ_F_LINK_BIT), 865ed29b0b4SJens Axboe /* doesn't sever on completion < 0 */ 866ed29b0b4SJens Axboe REQ_F_HARDLINK = BIT(REQ_F_HARDLINK_BIT), 867ed29b0b4SJens Axboe /* IOSQE_ASYNC */ 868ed29b0b4SJens Axboe REQ_F_FORCE_ASYNC = BIT(REQ_F_FORCE_ASYNC_BIT), 869ed29b0b4SJens Axboe /* IOSQE_BUFFER_SELECT */ 870ed29b0b4SJens Axboe REQ_F_BUFFER_SELECT = BIT(REQ_F_BUFFER_SELECT_BIT), 871ed29b0b4SJens Axboe /* IOSQE_CQE_SKIP_SUCCESS */ 872ed29b0b4SJens Axboe REQ_F_CQE_SKIP = BIT(REQ_F_CQE_SKIP_BIT), 873ed29b0b4SJens Axboe 874ed29b0b4SJens Axboe /* fail rest of links */ 875ed29b0b4SJens Axboe REQ_F_FAIL = BIT(REQ_F_FAIL_BIT), 876ed29b0b4SJens Axboe /* on inflight list, should be cancelled and waited on exit reliably */ 877ed29b0b4SJens Axboe REQ_F_INFLIGHT = BIT(REQ_F_INFLIGHT_BIT), 878ed29b0b4SJens Axboe /* read/write uses file position */ 879ed29b0b4SJens Axboe REQ_F_CUR_POS = BIT(REQ_F_CUR_POS_BIT), 880ed29b0b4SJens Axboe /* must not punt to workers */ 881ed29b0b4SJens Axboe REQ_F_NOWAIT = BIT(REQ_F_NOWAIT_BIT), 882ed29b0b4SJens Axboe /* has or had linked timeout */ 883ed29b0b4SJens Axboe REQ_F_LINK_TIMEOUT = BIT(REQ_F_LINK_TIMEOUT_BIT), 884ed29b0b4SJens Axboe /* needs cleanup */ 885ed29b0b4SJens Axboe REQ_F_NEED_CLEANUP = BIT(REQ_F_NEED_CLEANUP_BIT), 886ed29b0b4SJens Axboe /* already went through poll handler */ 887ed29b0b4SJens Axboe REQ_F_POLLED = BIT(REQ_F_POLLED_BIT), 888ed29b0b4SJens Axboe /* buffer already selected */ 889ed29b0b4SJens Axboe REQ_F_BUFFER_SELECTED = BIT(REQ_F_BUFFER_SELECTED_BIT), 890ed29b0b4SJens Axboe /* buffer selected from ring, needs commit */ 891ed29b0b4SJens Axboe REQ_F_BUFFER_RING = BIT(REQ_F_BUFFER_RING_BIT), 892ed29b0b4SJens Axboe /* completion is deferred through io_comp_state */ 893ed29b0b4SJens Axboe REQ_F_COMPLETE_INLINE = BIT(REQ_F_COMPLETE_INLINE_BIT), 894ed29b0b4SJens Axboe /* caller should reissue async */ 895ed29b0b4SJens Axboe REQ_F_REISSUE = BIT(REQ_F_REISSUE_BIT), 896ed29b0b4SJens Axboe /* supports async reads/writes */ 897ed29b0b4SJens Axboe REQ_F_SUPPORT_NOWAIT = BIT(REQ_F_SUPPORT_NOWAIT_BIT), 898ed29b0b4SJens Axboe /* regular file */ 899ed29b0b4SJens Axboe REQ_F_ISREG = BIT(REQ_F_ISREG_BIT), 900ed29b0b4SJens Axboe /* has creds assigned */ 901ed29b0b4SJens Axboe REQ_F_CREDS = BIT(REQ_F_CREDS_BIT), 902ed29b0b4SJens Axboe /* skip refcounting if not set */ 903ed29b0b4SJens Axboe REQ_F_REFCOUNT = BIT(REQ_F_REFCOUNT_BIT), 904ed29b0b4SJens Axboe /* there is a linked timeout that has to be armed */ 905ed29b0b4SJens Axboe REQ_F_ARM_LTIMEOUT = BIT(REQ_F_ARM_LTIMEOUT_BIT), 906ed29b0b4SJens Axboe /* ->async_data allocated */ 907ed29b0b4SJens Axboe REQ_F_ASYNC_DATA = BIT(REQ_F_ASYNC_DATA_BIT), 908ed29b0b4SJens Axboe /* don't post CQEs while failing linked requests */ 909ed29b0b4SJens Axboe REQ_F_SKIP_LINK_CQES = BIT(REQ_F_SKIP_LINK_CQES_BIT), 910ed29b0b4SJens Axboe /* single poll may be active */ 911ed29b0b4SJens Axboe REQ_F_SINGLE_POLL = BIT(REQ_F_SINGLE_POLL_BIT), 912ed29b0b4SJens Axboe /* double poll may active */ 913ed29b0b4SJens Axboe REQ_F_DOUBLE_POLL = BIT(REQ_F_DOUBLE_POLL_BIT), 914ed29b0b4SJens Axboe /* request has already done partial IO */ 915ed29b0b4SJens Axboe REQ_F_PARTIAL_IO = BIT(REQ_F_PARTIAL_IO_BIT), 916ed29b0b4SJens Axboe /* fast poll multishot mode */ 917ed29b0b4SJens Axboe REQ_F_APOLL_MULTISHOT = BIT(REQ_F_APOLL_MULTISHOT_BIT), 918ed29b0b4SJens Axboe /* ->extra1 and ->extra2 are initialised */ 919ed29b0b4SJens Axboe REQ_F_CQE32_INIT = BIT(REQ_F_CQE32_INIT_BIT), 920bd8587e4SJens Axboe /* recvmsg special flag, clear EPOLLIN */ 921bd8587e4SJens Axboe REQ_F_CLEAR_POLLIN = BIT(REQ_F_CLEAR_POLLIN_BIT), 922ed29b0b4SJens Axboe }; 923ed29b0b4SJens Axboe 924ed29b0b4SJens Axboe struct async_poll { 9258d4388d1SJens Axboe struct io_poll poll; 9268d4388d1SJens Axboe struct io_poll *double_poll; 927ed29b0b4SJens Axboe }; 928ed29b0b4SJens Axboe 929ed29b0b4SJens Axboe typedef void (*io_req_tw_func_t)(struct io_kiocb *req, bool *locked); 930ed29b0b4SJens Axboe 931ed29b0b4SJens Axboe struct io_task_work { 932ed29b0b4SJens Axboe union { 933ed29b0b4SJens Axboe struct io_wq_work_node node; 934ed29b0b4SJens Axboe struct llist_node fallback_node; 935ed29b0b4SJens Axboe }; 936ed29b0b4SJens Axboe io_req_tw_func_t func; 937ed29b0b4SJens Axboe }; 938ed29b0b4SJens Axboe 939ed29b0b4SJens Axboe enum { 940ed29b0b4SJens Axboe IORING_RSRC_FILE = 0, 941ed29b0b4SJens Axboe IORING_RSRC_BUFFER = 1, 942ed29b0b4SJens Axboe }; 943ed29b0b4SJens Axboe 944ed29b0b4SJens Axboe struct io_cqe { 945ed29b0b4SJens Axboe __u64 user_data; 946ed29b0b4SJens Axboe __s32 res; 947ed29b0b4SJens Axboe /* fd initially, then cflags for completion */ 948ed29b0b4SJens Axboe union { 949ed29b0b4SJens Axboe __u32 flags; 950ed29b0b4SJens Axboe int fd; 951ed29b0b4SJens Axboe }; 952ed29b0b4SJens Axboe }; 953ed29b0b4SJens Axboe 954ed29b0b4SJens Axboe enum { 955ed29b0b4SJens Axboe IO_CHECK_CQ_OVERFLOW_BIT, 956ed29b0b4SJens Axboe IO_CHECK_CQ_DROPPED_BIT, 957ed29b0b4SJens Axboe }; 958ed29b0b4SJens Axboe 959ed29b0b4SJens Axboe /* 960f49eca21SJens Axboe * Each request type overlays its private data structure on top of this one. 961f49eca21SJens Axboe * They must not exceed this one in size. 962f49eca21SJens Axboe */ 963f49eca21SJens Axboe struct io_cmd_data { 964f49eca21SJens Axboe struct file *file; 965f49eca21SJens Axboe /* each command gets 56 bytes of data */ 966f49eca21SJens Axboe __u8 data[56]; 967f49eca21SJens Axboe }; 968f49eca21SJens Axboe 969f49eca21SJens Axboe #define io_kiocb_to_cmd(req) ((void *) &(req)->cmd) 9703c306fb2SJens Axboe #define cmd_to_io_kiocb(ptr) ((struct io_kiocb *) ptr) 971f49eca21SJens Axboe 972f49eca21SJens Axboe struct io_kiocb { 973f49eca21SJens Axboe union { 974f49eca21SJens Axboe /* 975f49eca21SJens Axboe * NOTE! Each of the io_kiocb union members has the file pointer 976ed29b0b4SJens Axboe * as the first entry in their struct definition. So you can 977ed29b0b4SJens Axboe * access the file pointer through any of the sub-structs, 978ed29b0b4SJens Axboe * or directly as just 'file' in this struct. 979ed29b0b4SJens Axboe */ 980ed29b0b4SJens Axboe struct file *file; 981f49eca21SJens Axboe struct io_cmd_data cmd; 982ed29b0b4SJens Axboe struct io_rsrc_update rsrc_update; 983ed29b0b4SJens Axboe struct io_xattr xattr; 984ed29b0b4SJens Axboe struct io_uring_cmd uring_cmd; 985ed29b0b4SJens Axboe }; 986ed29b0b4SJens Axboe 987ed29b0b4SJens Axboe u8 opcode; 988ed29b0b4SJens Axboe /* polled IO has completed */ 989ed29b0b4SJens Axboe u8 iopoll_completed; 990ed29b0b4SJens Axboe /* 991ed29b0b4SJens Axboe * Can be either a fixed buffer index, or used with provided buffers. 992ed29b0b4SJens Axboe * For the latter, before issue it points to the buffer group ID, 993ed29b0b4SJens Axboe * and after selection it points to the buffer ID itself. 994ed29b0b4SJens Axboe */ 995ed29b0b4SJens Axboe u16 buf_index; 996ed29b0b4SJens Axboe unsigned int flags; 997ed29b0b4SJens Axboe 998ed29b0b4SJens Axboe struct io_cqe cqe; 999ed29b0b4SJens Axboe 1000ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 1001ed29b0b4SJens Axboe struct task_struct *task; 1002ed29b0b4SJens Axboe 1003ed29b0b4SJens Axboe struct io_rsrc_node *rsrc_node; 1004ed29b0b4SJens Axboe 1005ed29b0b4SJens Axboe union { 1006ed29b0b4SJens Axboe /* store used ubuf, so we can prevent reloading */ 1007ed29b0b4SJens Axboe struct io_mapped_ubuf *imu; 1008ed29b0b4SJens Axboe 1009ed29b0b4SJens Axboe /* stores selected buf, valid IFF REQ_F_BUFFER_SELECTED is set */ 1010ed29b0b4SJens Axboe struct io_buffer *kbuf; 1011ed29b0b4SJens Axboe 1012ed29b0b4SJens Axboe /* 1013ed29b0b4SJens Axboe * stores buffer ID for ring provided buffers, valid IFF 1014ed29b0b4SJens Axboe * REQ_F_BUFFER_RING is set. 1015ed29b0b4SJens Axboe */ 1016ed29b0b4SJens Axboe struct io_buffer_list *buf_list; 1017ed29b0b4SJens Axboe }; 1018ed29b0b4SJens Axboe 1019ed29b0b4SJens Axboe union { 1020ed29b0b4SJens Axboe /* used by request caches, completion batching and iopoll */ 1021ed29b0b4SJens Axboe struct io_wq_work_node comp_list; 1022ed29b0b4SJens Axboe /* cache ->apoll->events */ 1023ed29b0b4SJens Axboe __poll_t apoll_events; 1024ed29b0b4SJens Axboe }; 1025ed29b0b4SJens Axboe atomic_t refs; 1026ed29b0b4SJens Axboe atomic_t poll_refs; 1027ed29b0b4SJens Axboe struct io_task_work io_task_work; 1028ed29b0b4SJens Axboe /* for polled requests, i.e. IORING_OP_POLL_ADD and async armed poll */ 1029ed29b0b4SJens Axboe union { 1030ed29b0b4SJens Axboe struct hlist_node hash_node; 1031ed29b0b4SJens Axboe struct { 1032ed29b0b4SJens Axboe u64 extra1; 1033ed29b0b4SJens Axboe u64 extra2; 1034ed29b0b4SJens Axboe }; 1035ed29b0b4SJens Axboe }; 1036ed29b0b4SJens Axboe /* internal polling, see IORING_FEAT_FAST_POLL */ 1037ed29b0b4SJens Axboe struct async_poll *apoll; 1038ed29b0b4SJens Axboe /* opcode allocated if it needs to store data for async defer */ 1039ed29b0b4SJens Axboe void *async_data; 1040ed29b0b4SJens Axboe /* linked requests, IFF REQ_F_HARDLINK or REQ_F_LINK are set */ 1041ed29b0b4SJens Axboe struct io_kiocb *link; 1042ed29b0b4SJens Axboe /* custom credentials, valid IFF REQ_F_CREDS is set */ 1043ed29b0b4SJens Axboe const struct cred *creds; 1044ed29b0b4SJens Axboe struct io_wq_work work; 1045ed29b0b4SJens Axboe }; 1046ed29b0b4SJens Axboe 1047ed29b0b4SJens Axboe struct io_tctx_node { 1048ed29b0b4SJens Axboe struct list_head ctx_node; 1049ed29b0b4SJens Axboe struct task_struct *task; 1050ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 1051ed29b0b4SJens Axboe }; 1052ed29b0b4SJens Axboe 1053ed29b0b4SJens Axboe struct io_defer_entry { 1054ed29b0b4SJens Axboe struct list_head list; 1055ed29b0b4SJens Axboe struct io_kiocb *req; 1056ed29b0b4SJens Axboe u32 seq; 1057ed29b0b4SJens Axboe }; 1058ed29b0b4SJens Axboe 1059ed29b0b4SJens Axboe struct io_cancel_data { 1060ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 1061ed29b0b4SJens Axboe union { 1062ed29b0b4SJens Axboe u64 data; 1063ed29b0b4SJens Axboe struct file *file; 1064ed29b0b4SJens Axboe }; 1065ed29b0b4SJens Axboe u32 flags; 1066ed29b0b4SJens Axboe int seq; 1067ed29b0b4SJens Axboe }; 1068ed29b0b4SJens Axboe 1069ed29b0b4SJens Axboe /* 1070ed29b0b4SJens Axboe * The URING_CMD payload starts at 'cmd' in the first sqe, and continues into 1071ed29b0b4SJens Axboe * the following sqe if SQE128 is used. 1072ed29b0b4SJens Axboe */ 1073ed29b0b4SJens Axboe #define uring_cmd_pdu_size(is_sqe128) \ 1074ed29b0b4SJens Axboe ((1 + !!(is_sqe128)) * sizeof(struct io_uring_sqe) - \ 1075ed29b0b4SJens Axboe offsetof(struct io_uring_sqe, cmd)) 1076ed29b0b4SJens Axboe 1077ed29b0b4SJens Axboe struct io_op_def { 1078ed29b0b4SJens Axboe /* needs req->file assigned */ 1079ed29b0b4SJens Axboe unsigned needs_file : 1; 1080ed29b0b4SJens Axboe /* should block plug */ 1081ed29b0b4SJens Axboe unsigned plug : 1; 1082ed29b0b4SJens Axboe /* hash wq insertion if file is a regular file */ 1083ed29b0b4SJens Axboe unsigned hash_reg_file : 1; 1084ed29b0b4SJens Axboe /* unbound wq insertion if file is a non-regular file */ 1085ed29b0b4SJens Axboe unsigned unbound_nonreg_file : 1; 1086ed29b0b4SJens Axboe /* set if opcode supports polled "wait" */ 1087ed29b0b4SJens Axboe unsigned pollin : 1; 1088ed29b0b4SJens Axboe unsigned pollout : 1; 1089ed29b0b4SJens Axboe unsigned poll_exclusive : 1; 1090ed29b0b4SJens Axboe /* op supports buffer selection */ 1091ed29b0b4SJens Axboe unsigned buffer_select : 1; 1092ed29b0b4SJens Axboe /* opcode is not supported by this kernel */ 1093ed29b0b4SJens Axboe unsigned not_supported : 1; 1094ed29b0b4SJens Axboe /* skip auditing */ 1095ed29b0b4SJens Axboe unsigned audit_skip : 1; 1096ed29b0b4SJens Axboe /* supports ioprio */ 1097ed29b0b4SJens Axboe unsigned ioprio : 1; 1098ed29b0b4SJens Axboe /* supports iopoll */ 1099ed29b0b4SJens Axboe unsigned iopoll : 1; 1100ed29b0b4SJens Axboe /* size of async data needed, if any */ 1101ed29b0b4SJens Axboe unsigned short async_size; 1102ed29b0b4SJens Axboe 1103ed29b0b4SJens Axboe int (*prep)(struct io_kiocb *, const struct io_uring_sqe *); 1104ed29b0b4SJens Axboe int (*issue)(struct io_kiocb *, unsigned int); 1105dc919cafSJens Axboe int (*prep_async)(struct io_kiocb *); 1106ed29b0b4SJens Axboe }; 1107ed29b0b4SJens Axboe 1108ed29b0b4SJens Axboe static const struct io_op_def io_op_defs[]; 1109ed29b0b4SJens Axboe 1110ed29b0b4SJens Axboe /* requests with any of those set should undergo io_disarm_next() */ 1111ed29b0b4SJens Axboe #define IO_DISARM_MASK (REQ_F_ARM_LTIMEOUT | REQ_F_LINK_TIMEOUT | REQ_F_FAIL) 1112ed29b0b4SJens Axboe #define IO_REQ_LINK_FLAGS (REQ_F_LINK | REQ_F_HARDLINK) 1113ed29b0b4SJens Axboe 1114ed29b0b4SJens Axboe static bool io_disarm_next(struct io_kiocb *req); 1115ed29b0b4SJens Axboe static void io_uring_del_tctx_node(unsigned long index); 1116ed29b0b4SJens Axboe static void io_uring_try_cancel_requests(struct io_ring_ctx *ctx, 1117ed29b0b4SJens Axboe struct task_struct *task, 1118ed29b0b4SJens Axboe bool cancel_all); 1119ed29b0b4SJens Axboe static void io_uring_cancel_generic(bool cancel_all, struct io_sq_data *sqd); 1120ed29b0b4SJens Axboe 1121ed29b0b4SJens Axboe static void __io_req_complete_post(struct io_kiocb *req, s32 res, u32 cflags); 1122ed29b0b4SJens Axboe static void io_dismantle_req(struct io_kiocb *req); 1123ed29b0b4SJens Axboe static void io_queue_linked_timeout(struct io_kiocb *req); 1124ed29b0b4SJens Axboe static int __io_register_rsrc_update(struct io_ring_ctx *ctx, unsigned type, 1125ed29b0b4SJens Axboe struct io_uring_rsrc_update2 *up, 1126ed29b0b4SJens Axboe unsigned nr_args); 1127ed29b0b4SJens Axboe static void io_clean_op(struct io_kiocb *req); 1128ed29b0b4SJens Axboe static inline struct file *io_file_get_fixed(struct io_kiocb *req, int fd, 1129ed29b0b4SJens Axboe unsigned issue_flags); 1130ed29b0b4SJens Axboe static struct file *io_file_get_normal(struct io_kiocb *req, int fd); 1131ed29b0b4SJens Axboe static void io_queue_sqe(struct io_kiocb *req); 1132ed29b0b4SJens Axboe static void io_rsrc_put_work(struct work_struct *work); 1133ed29b0b4SJens Axboe 1134ed29b0b4SJens Axboe static void io_req_task_queue(struct io_kiocb *req); 1135ed29b0b4SJens Axboe static void __io_submit_flush_completions(struct io_ring_ctx *ctx); 1136ed29b0b4SJens Axboe static int io_req_prep_async(struct io_kiocb *req); 1137ed29b0b4SJens Axboe 1138ed29b0b4SJens Axboe static int io_install_fixed_file(struct io_kiocb *req, struct file *file, 1139ed29b0b4SJens Axboe unsigned int issue_flags, u32 slot_index); 1140ed29b0b4SJens Axboe static int __io_close_fixed(struct io_kiocb *req, unsigned int issue_flags, 1141ed29b0b4SJens Axboe unsigned int offset); 1142ed29b0b4SJens Axboe static inline int io_close_fixed(struct io_kiocb *req, unsigned int issue_flags); 1143ed29b0b4SJens Axboe 1144ed29b0b4SJens Axboe static enum hrtimer_restart io_link_timeout_fn(struct hrtimer *timer); 1145ed29b0b4SJens Axboe static void io_eventfd_signal(struct io_ring_ctx *ctx); 1146ed29b0b4SJens Axboe static void io_req_tw_post_queue(struct io_kiocb *req, s32 res, u32 cflags); 1147ed29b0b4SJens Axboe 1148ed29b0b4SJens Axboe static struct kmem_cache *req_cachep; 1149ed29b0b4SJens Axboe 1150ed29b0b4SJens Axboe static const struct file_operations io_uring_fops; 1151ed29b0b4SJens Axboe 1152ed29b0b4SJens Axboe const char *io_uring_get_opcode(u8 opcode) 1153ed29b0b4SJens Axboe { 1154ed29b0b4SJens Axboe switch ((enum io_uring_op)opcode) { 1155ed29b0b4SJens Axboe case IORING_OP_NOP: 1156ed29b0b4SJens Axboe return "NOP"; 1157ed29b0b4SJens Axboe case IORING_OP_READV: 1158ed29b0b4SJens Axboe return "READV"; 1159ed29b0b4SJens Axboe case IORING_OP_WRITEV: 1160ed29b0b4SJens Axboe return "WRITEV"; 1161ed29b0b4SJens Axboe case IORING_OP_FSYNC: 1162ed29b0b4SJens Axboe return "FSYNC"; 1163ed29b0b4SJens Axboe case IORING_OP_READ_FIXED: 1164ed29b0b4SJens Axboe return "READ_FIXED"; 1165ed29b0b4SJens Axboe case IORING_OP_WRITE_FIXED: 1166ed29b0b4SJens Axboe return "WRITE_FIXED"; 1167ed29b0b4SJens Axboe case IORING_OP_POLL_ADD: 1168ed29b0b4SJens Axboe return "POLL_ADD"; 1169ed29b0b4SJens Axboe case IORING_OP_POLL_REMOVE: 1170ed29b0b4SJens Axboe return "POLL_REMOVE"; 1171ed29b0b4SJens Axboe case IORING_OP_SYNC_FILE_RANGE: 1172ed29b0b4SJens Axboe return "SYNC_FILE_RANGE"; 1173ed29b0b4SJens Axboe case IORING_OP_SENDMSG: 1174ed29b0b4SJens Axboe return "SENDMSG"; 1175ed29b0b4SJens Axboe case IORING_OP_RECVMSG: 1176ed29b0b4SJens Axboe return "RECVMSG"; 1177ed29b0b4SJens Axboe case IORING_OP_TIMEOUT: 1178ed29b0b4SJens Axboe return "TIMEOUT"; 1179ed29b0b4SJens Axboe case IORING_OP_TIMEOUT_REMOVE: 1180ed29b0b4SJens Axboe return "TIMEOUT_REMOVE"; 1181ed29b0b4SJens Axboe case IORING_OP_ACCEPT: 1182ed29b0b4SJens Axboe return "ACCEPT"; 1183ed29b0b4SJens Axboe case IORING_OP_ASYNC_CANCEL: 1184ed29b0b4SJens Axboe return "ASYNC_CANCEL"; 1185ed29b0b4SJens Axboe case IORING_OP_LINK_TIMEOUT: 1186ed29b0b4SJens Axboe return "LINK_TIMEOUT"; 1187ed29b0b4SJens Axboe case IORING_OP_CONNECT: 1188ed29b0b4SJens Axboe return "CONNECT"; 1189ed29b0b4SJens Axboe case IORING_OP_FALLOCATE: 1190ed29b0b4SJens Axboe return "FALLOCATE"; 1191ed29b0b4SJens Axboe case IORING_OP_OPENAT: 1192ed29b0b4SJens Axboe return "OPENAT"; 1193ed29b0b4SJens Axboe case IORING_OP_CLOSE: 1194ed29b0b4SJens Axboe return "CLOSE"; 1195ed29b0b4SJens Axboe case IORING_OP_FILES_UPDATE: 1196ed29b0b4SJens Axboe return "FILES_UPDATE"; 1197ed29b0b4SJens Axboe case IORING_OP_STATX: 1198ed29b0b4SJens Axboe return "STATX"; 1199ed29b0b4SJens Axboe case IORING_OP_READ: 1200ed29b0b4SJens Axboe return "READ"; 1201ed29b0b4SJens Axboe case IORING_OP_WRITE: 1202ed29b0b4SJens Axboe return "WRITE"; 1203ed29b0b4SJens Axboe case IORING_OP_FADVISE: 1204ed29b0b4SJens Axboe return "FADVISE"; 1205ed29b0b4SJens Axboe case IORING_OP_MADVISE: 1206ed29b0b4SJens Axboe return "MADVISE"; 1207ed29b0b4SJens Axboe case IORING_OP_SEND: 1208ed29b0b4SJens Axboe return "SEND"; 1209ed29b0b4SJens Axboe case IORING_OP_RECV: 1210ed29b0b4SJens Axboe return "RECV"; 1211ed29b0b4SJens Axboe case IORING_OP_OPENAT2: 1212ed29b0b4SJens Axboe return "OPENAT2"; 1213ed29b0b4SJens Axboe case IORING_OP_EPOLL_CTL: 1214ed29b0b4SJens Axboe return "EPOLL_CTL"; 1215ed29b0b4SJens Axboe case IORING_OP_SPLICE: 1216ed29b0b4SJens Axboe return "SPLICE"; 1217ed29b0b4SJens Axboe case IORING_OP_PROVIDE_BUFFERS: 1218ed29b0b4SJens Axboe return "PROVIDE_BUFFERS"; 1219ed29b0b4SJens Axboe case IORING_OP_REMOVE_BUFFERS: 1220ed29b0b4SJens Axboe return "REMOVE_BUFFERS"; 1221ed29b0b4SJens Axboe case IORING_OP_TEE: 1222ed29b0b4SJens Axboe return "TEE"; 1223ed29b0b4SJens Axboe case IORING_OP_SHUTDOWN: 1224ed29b0b4SJens Axboe return "SHUTDOWN"; 1225ed29b0b4SJens Axboe case IORING_OP_RENAMEAT: 1226ed29b0b4SJens Axboe return "RENAMEAT"; 1227ed29b0b4SJens Axboe case IORING_OP_UNLINKAT: 1228ed29b0b4SJens Axboe return "UNLINKAT"; 1229ed29b0b4SJens Axboe case IORING_OP_MKDIRAT: 1230ed29b0b4SJens Axboe return "MKDIRAT"; 1231ed29b0b4SJens Axboe case IORING_OP_SYMLINKAT: 1232ed29b0b4SJens Axboe return "SYMLINKAT"; 1233ed29b0b4SJens Axboe case IORING_OP_LINKAT: 1234ed29b0b4SJens Axboe return "LINKAT"; 1235ed29b0b4SJens Axboe case IORING_OP_MSG_RING: 1236ed29b0b4SJens Axboe return "MSG_RING"; 1237ed29b0b4SJens Axboe case IORING_OP_FSETXATTR: 1238ed29b0b4SJens Axboe return "FSETXATTR"; 1239ed29b0b4SJens Axboe case IORING_OP_SETXATTR: 1240ed29b0b4SJens Axboe return "SETXATTR"; 1241ed29b0b4SJens Axboe case IORING_OP_FGETXATTR: 1242ed29b0b4SJens Axboe return "FGETXATTR"; 1243ed29b0b4SJens Axboe case IORING_OP_GETXATTR: 1244ed29b0b4SJens Axboe return "GETXATTR"; 1245ed29b0b4SJens Axboe case IORING_OP_SOCKET: 1246ed29b0b4SJens Axboe return "SOCKET"; 1247ed29b0b4SJens Axboe case IORING_OP_URING_CMD: 1248ed29b0b4SJens Axboe return "URING_CMD"; 1249ed29b0b4SJens Axboe case IORING_OP_LAST: 1250ed29b0b4SJens Axboe return "INVALID"; 1251ed29b0b4SJens Axboe } 1252ed29b0b4SJens Axboe return "INVALID"; 1253ed29b0b4SJens Axboe } 1254ed29b0b4SJens Axboe 1255ed29b0b4SJens Axboe struct sock *io_uring_get_socket(struct file *file) 1256ed29b0b4SJens Axboe { 1257ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 1258ed29b0b4SJens Axboe if (file->f_op == &io_uring_fops) { 1259ed29b0b4SJens Axboe struct io_ring_ctx *ctx = file->private_data; 1260ed29b0b4SJens Axboe 1261ed29b0b4SJens Axboe return ctx->ring_sock->sk; 1262ed29b0b4SJens Axboe } 1263ed29b0b4SJens Axboe #endif 1264ed29b0b4SJens Axboe return NULL; 1265ed29b0b4SJens Axboe } 1266ed29b0b4SJens Axboe EXPORT_SYMBOL(io_uring_get_socket); 1267ed29b0b4SJens Axboe 1268ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 1269ed29b0b4SJens Axboe static inline bool io_file_need_scm(struct file *filp) 1270ed29b0b4SJens Axboe { 1271ed29b0b4SJens Axboe #if defined(IO_URING_SCM_ALL) 1272ed29b0b4SJens Axboe return true; 1273ed29b0b4SJens Axboe #else 1274ed29b0b4SJens Axboe return !!unix_get_socket(filp); 1275ed29b0b4SJens Axboe #endif 1276ed29b0b4SJens Axboe } 1277ed29b0b4SJens Axboe #else 1278ed29b0b4SJens Axboe static inline bool io_file_need_scm(struct file *filp) 1279ed29b0b4SJens Axboe { 1280ed29b0b4SJens Axboe return false; 1281ed29b0b4SJens Axboe } 1282ed29b0b4SJens Axboe #endif 1283ed29b0b4SJens Axboe 1284ed29b0b4SJens Axboe static void io_ring_submit_unlock(struct io_ring_ctx *ctx, unsigned issue_flags) 1285ed29b0b4SJens Axboe { 1286ed29b0b4SJens Axboe lockdep_assert_held(&ctx->uring_lock); 1287ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_UNLOCKED) 1288ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 1289ed29b0b4SJens Axboe } 1290ed29b0b4SJens Axboe 1291ed29b0b4SJens Axboe static void io_ring_submit_lock(struct io_ring_ctx *ctx, unsigned issue_flags) 1292ed29b0b4SJens Axboe { 1293ed29b0b4SJens Axboe /* 1294ed29b0b4SJens Axboe * "Normal" inline submissions always hold the uring_lock, since we 1295ed29b0b4SJens Axboe * grab it from the system call. Same is true for the SQPOLL offload. 1296ed29b0b4SJens Axboe * The only exception is when we've detached the request and issue it 1297ed29b0b4SJens Axboe * from an async worker thread, grab the lock for that case. 1298ed29b0b4SJens Axboe */ 1299ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_UNLOCKED) 1300ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 1301ed29b0b4SJens Axboe lockdep_assert_held(&ctx->uring_lock); 1302ed29b0b4SJens Axboe } 1303ed29b0b4SJens Axboe 1304ed29b0b4SJens Axboe static inline void io_tw_lock(struct io_ring_ctx *ctx, bool *locked) 1305ed29b0b4SJens Axboe { 1306ed29b0b4SJens Axboe if (!*locked) { 1307ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 1308ed29b0b4SJens Axboe *locked = true; 1309ed29b0b4SJens Axboe } 1310ed29b0b4SJens Axboe } 1311ed29b0b4SJens Axboe 1312ed29b0b4SJens Axboe #define io_for_each_link(pos, head) \ 1313ed29b0b4SJens Axboe for (pos = (head); pos; pos = pos->link) 1314ed29b0b4SJens Axboe 1315ed29b0b4SJens Axboe /* 1316ed29b0b4SJens Axboe * Shamelessly stolen from the mm implementation of page reference checking, 1317ed29b0b4SJens Axboe * see commit f958d7b528b1 for details. 1318ed29b0b4SJens Axboe */ 1319ed29b0b4SJens Axboe #define req_ref_zero_or_close_to_overflow(req) \ 1320ed29b0b4SJens Axboe ((unsigned int) atomic_read(&(req->refs)) + 127u <= 127u) 1321ed29b0b4SJens Axboe 1322ed29b0b4SJens Axboe static inline bool req_ref_inc_not_zero(struct io_kiocb *req) 1323ed29b0b4SJens Axboe { 1324ed29b0b4SJens Axboe WARN_ON_ONCE(!(req->flags & REQ_F_REFCOUNT)); 1325ed29b0b4SJens Axboe return atomic_inc_not_zero(&req->refs); 1326ed29b0b4SJens Axboe } 1327ed29b0b4SJens Axboe 1328ed29b0b4SJens Axboe static inline bool req_ref_put_and_test(struct io_kiocb *req) 1329ed29b0b4SJens Axboe { 1330ed29b0b4SJens Axboe if (likely(!(req->flags & REQ_F_REFCOUNT))) 1331ed29b0b4SJens Axboe return true; 1332ed29b0b4SJens Axboe 1333ed29b0b4SJens Axboe WARN_ON_ONCE(req_ref_zero_or_close_to_overflow(req)); 1334ed29b0b4SJens Axboe return atomic_dec_and_test(&req->refs); 1335ed29b0b4SJens Axboe } 1336ed29b0b4SJens Axboe 1337ed29b0b4SJens Axboe static inline void req_ref_get(struct io_kiocb *req) 1338ed29b0b4SJens Axboe { 1339ed29b0b4SJens Axboe WARN_ON_ONCE(!(req->flags & REQ_F_REFCOUNT)); 1340ed29b0b4SJens Axboe WARN_ON_ONCE(req_ref_zero_or_close_to_overflow(req)); 1341ed29b0b4SJens Axboe atomic_inc(&req->refs); 1342ed29b0b4SJens Axboe } 1343ed29b0b4SJens Axboe 1344ed29b0b4SJens Axboe static inline void io_submit_flush_completions(struct io_ring_ctx *ctx) 1345ed29b0b4SJens Axboe { 1346ed29b0b4SJens Axboe if (!wq_list_empty(&ctx->submit_state.compl_reqs)) 1347ed29b0b4SJens Axboe __io_submit_flush_completions(ctx); 1348ed29b0b4SJens Axboe } 1349ed29b0b4SJens Axboe 1350ed29b0b4SJens Axboe static inline void __io_req_set_refcount(struct io_kiocb *req, int nr) 1351ed29b0b4SJens Axboe { 1352ed29b0b4SJens Axboe if (!(req->flags & REQ_F_REFCOUNT)) { 1353ed29b0b4SJens Axboe req->flags |= REQ_F_REFCOUNT; 1354ed29b0b4SJens Axboe atomic_set(&req->refs, nr); 1355ed29b0b4SJens Axboe } 1356ed29b0b4SJens Axboe } 1357ed29b0b4SJens Axboe 1358ed29b0b4SJens Axboe static inline void io_req_set_refcount(struct io_kiocb *req) 1359ed29b0b4SJens Axboe { 1360ed29b0b4SJens Axboe __io_req_set_refcount(req, 1); 1361ed29b0b4SJens Axboe } 1362ed29b0b4SJens Axboe 1363ed29b0b4SJens Axboe #define IO_RSRC_REF_BATCH 100 1364ed29b0b4SJens Axboe 1365ed29b0b4SJens Axboe static void io_rsrc_put_node(struct io_rsrc_node *node, int nr) 1366ed29b0b4SJens Axboe { 1367ed29b0b4SJens Axboe percpu_ref_put_many(&node->refs, nr); 1368ed29b0b4SJens Axboe } 1369ed29b0b4SJens Axboe 1370ed29b0b4SJens Axboe static inline void io_req_put_rsrc_locked(struct io_kiocb *req, 1371ed29b0b4SJens Axboe struct io_ring_ctx *ctx) 1372ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 1373ed29b0b4SJens Axboe { 1374ed29b0b4SJens Axboe struct io_rsrc_node *node = req->rsrc_node; 1375ed29b0b4SJens Axboe 1376ed29b0b4SJens Axboe if (node) { 1377ed29b0b4SJens Axboe if (node == ctx->rsrc_node) 1378ed29b0b4SJens Axboe ctx->rsrc_cached_refs++; 1379ed29b0b4SJens Axboe else 1380ed29b0b4SJens Axboe io_rsrc_put_node(node, 1); 1381ed29b0b4SJens Axboe } 1382ed29b0b4SJens Axboe } 1383ed29b0b4SJens Axboe 1384ed29b0b4SJens Axboe static inline void io_req_put_rsrc(struct io_kiocb *req) 1385ed29b0b4SJens Axboe { 1386ed29b0b4SJens Axboe if (req->rsrc_node) 1387ed29b0b4SJens Axboe io_rsrc_put_node(req->rsrc_node, 1); 1388ed29b0b4SJens Axboe } 1389ed29b0b4SJens Axboe 1390ed29b0b4SJens Axboe static __cold void io_rsrc_refs_drop(struct io_ring_ctx *ctx) 1391ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 1392ed29b0b4SJens Axboe { 1393ed29b0b4SJens Axboe if (ctx->rsrc_cached_refs) { 1394ed29b0b4SJens Axboe io_rsrc_put_node(ctx->rsrc_node, ctx->rsrc_cached_refs); 1395ed29b0b4SJens Axboe ctx->rsrc_cached_refs = 0; 1396ed29b0b4SJens Axboe } 1397ed29b0b4SJens Axboe } 1398ed29b0b4SJens Axboe 1399ed29b0b4SJens Axboe static void io_rsrc_refs_refill(struct io_ring_ctx *ctx) 1400ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 1401ed29b0b4SJens Axboe { 1402ed29b0b4SJens Axboe ctx->rsrc_cached_refs += IO_RSRC_REF_BATCH; 1403ed29b0b4SJens Axboe percpu_ref_get_many(&ctx->rsrc_node->refs, IO_RSRC_REF_BATCH); 1404ed29b0b4SJens Axboe } 1405ed29b0b4SJens Axboe 1406ed29b0b4SJens Axboe static inline void io_req_set_rsrc_node(struct io_kiocb *req, 1407ed29b0b4SJens Axboe struct io_ring_ctx *ctx, 1408ed29b0b4SJens Axboe unsigned int issue_flags) 1409ed29b0b4SJens Axboe { 1410ed29b0b4SJens Axboe if (!req->rsrc_node) { 1411ed29b0b4SJens Axboe req->rsrc_node = ctx->rsrc_node; 1412ed29b0b4SJens Axboe 1413ed29b0b4SJens Axboe if (!(issue_flags & IO_URING_F_UNLOCKED)) { 1414ed29b0b4SJens Axboe lockdep_assert_held(&ctx->uring_lock); 1415ed29b0b4SJens Axboe ctx->rsrc_cached_refs--; 1416ed29b0b4SJens Axboe if (unlikely(ctx->rsrc_cached_refs < 0)) 1417ed29b0b4SJens Axboe io_rsrc_refs_refill(ctx); 1418ed29b0b4SJens Axboe } else { 1419ed29b0b4SJens Axboe percpu_ref_get(&req->rsrc_node->refs); 1420ed29b0b4SJens Axboe } 1421ed29b0b4SJens Axboe } 1422ed29b0b4SJens Axboe } 1423ed29b0b4SJens Axboe 1424ed29b0b4SJens Axboe static unsigned int __io_put_kbuf(struct io_kiocb *req, struct list_head *list) 1425ed29b0b4SJens Axboe { 1426ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_RING) { 1427ed29b0b4SJens Axboe if (req->buf_list) 1428ed29b0b4SJens Axboe req->buf_list->head++; 1429ed29b0b4SJens Axboe req->flags &= ~REQ_F_BUFFER_RING; 1430ed29b0b4SJens Axboe } else { 1431ed29b0b4SJens Axboe list_add(&req->kbuf->list, list); 1432ed29b0b4SJens Axboe req->flags &= ~REQ_F_BUFFER_SELECTED; 1433ed29b0b4SJens Axboe } 1434ed29b0b4SJens Axboe 1435ed29b0b4SJens Axboe return IORING_CQE_F_BUFFER | (req->buf_index << IORING_CQE_BUFFER_SHIFT); 1436ed29b0b4SJens Axboe } 1437ed29b0b4SJens Axboe 1438ed29b0b4SJens Axboe static inline unsigned int io_put_kbuf_comp(struct io_kiocb *req) 1439ed29b0b4SJens Axboe { 1440ed29b0b4SJens Axboe lockdep_assert_held(&req->ctx->completion_lock); 1441ed29b0b4SJens Axboe 1442ed29b0b4SJens Axboe if (!(req->flags & (REQ_F_BUFFER_SELECTED|REQ_F_BUFFER_RING))) 1443ed29b0b4SJens Axboe return 0; 1444ed29b0b4SJens Axboe return __io_put_kbuf(req, &req->ctx->io_buffers_comp); 1445ed29b0b4SJens Axboe } 1446ed29b0b4SJens Axboe 1447ed29b0b4SJens Axboe static inline unsigned int io_put_kbuf(struct io_kiocb *req, 1448ed29b0b4SJens Axboe unsigned issue_flags) 1449ed29b0b4SJens Axboe { 1450ed29b0b4SJens Axboe unsigned int cflags; 1451ed29b0b4SJens Axboe 1452ed29b0b4SJens Axboe if (!(req->flags & (REQ_F_BUFFER_SELECTED|REQ_F_BUFFER_RING))) 1453ed29b0b4SJens Axboe return 0; 1454ed29b0b4SJens Axboe 1455ed29b0b4SJens Axboe /* 1456ed29b0b4SJens Axboe * We can add this buffer back to two lists: 1457ed29b0b4SJens Axboe * 1458ed29b0b4SJens Axboe * 1) The io_buffers_cache list. This one is protected by the 1459ed29b0b4SJens Axboe * ctx->uring_lock. If we already hold this lock, add back to this 1460ed29b0b4SJens Axboe * list as we can grab it from issue as well. 1461ed29b0b4SJens Axboe * 2) The io_buffers_comp list. This one is protected by the 1462ed29b0b4SJens Axboe * ctx->completion_lock. 1463ed29b0b4SJens Axboe * 1464ed29b0b4SJens Axboe * We migrate buffers from the comp_list to the issue cache list 1465ed29b0b4SJens Axboe * when we need one. 1466ed29b0b4SJens Axboe */ 1467ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_RING) { 1468ed29b0b4SJens Axboe /* no buffers to recycle for this case */ 1469ed29b0b4SJens Axboe cflags = __io_put_kbuf(req, NULL); 1470ed29b0b4SJens Axboe } else if (issue_flags & IO_URING_F_UNLOCKED) { 1471ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 1472ed29b0b4SJens Axboe 1473ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 1474ed29b0b4SJens Axboe cflags = __io_put_kbuf(req, &ctx->io_buffers_comp); 1475ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 1476ed29b0b4SJens Axboe } else { 1477ed29b0b4SJens Axboe lockdep_assert_held(&req->ctx->uring_lock); 1478ed29b0b4SJens Axboe 1479ed29b0b4SJens Axboe cflags = __io_put_kbuf(req, &req->ctx->io_buffers_cache); 1480ed29b0b4SJens Axboe } 1481ed29b0b4SJens Axboe 1482ed29b0b4SJens Axboe return cflags; 1483ed29b0b4SJens Axboe } 1484ed29b0b4SJens Axboe 1485ed29b0b4SJens Axboe static struct io_buffer_list *io_buffer_get_list(struct io_ring_ctx *ctx, 1486ed29b0b4SJens Axboe unsigned int bgid) 1487ed29b0b4SJens Axboe { 1488ed29b0b4SJens Axboe if (ctx->io_bl && bgid < BGID_ARRAY) 1489ed29b0b4SJens Axboe return &ctx->io_bl[bgid]; 1490ed29b0b4SJens Axboe 1491ed29b0b4SJens Axboe return xa_load(&ctx->io_bl_xa, bgid); 1492ed29b0b4SJens Axboe } 1493ed29b0b4SJens Axboe 1494ed29b0b4SJens Axboe static void io_kbuf_recycle(struct io_kiocb *req, unsigned issue_flags) 1495ed29b0b4SJens Axboe { 1496ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 1497ed29b0b4SJens Axboe struct io_buffer_list *bl; 1498ed29b0b4SJens Axboe struct io_buffer *buf; 1499ed29b0b4SJens Axboe 1500ed29b0b4SJens Axboe if (!(req->flags & (REQ_F_BUFFER_SELECTED|REQ_F_BUFFER_RING))) 1501ed29b0b4SJens Axboe return; 1502ed29b0b4SJens Axboe /* 1503ed29b0b4SJens Axboe * For legacy provided buffer mode, don't recycle if we already did 1504ed29b0b4SJens Axboe * IO to this buffer. For ring-mapped provided buffer mode, we should 1505ed29b0b4SJens Axboe * increment ring->head to explicitly monopolize the buffer to avoid 1506ed29b0b4SJens Axboe * multiple use. 1507ed29b0b4SJens Axboe */ 1508ed29b0b4SJens Axboe if ((req->flags & REQ_F_BUFFER_SELECTED) && 1509ed29b0b4SJens Axboe (req->flags & REQ_F_PARTIAL_IO)) 1510ed29b0b4SJens Axboe return; 1511ed29b0b4SJens Axboe 1512ed29b0b4SJens Axboe /* 1513ed29b0b4SJens Axboe * READV uses fields in `struct io_rw` (len/addr) to stash the selected 1514ed29b0b4SJens Axboe * buffer data. However if that buffer is recycled the original request 1515ed29b0b4SJens Axboe * data stored in addr is lost. Therefore forbid recycling for now. 1516ed29b0b4SJens Axboe */ 1517ed29b0b4SJens Axboe if (req->opcode == IORING_OP_READV) 1518ed29b0b4SJens Axboe return; 1519ed29b0b4SJens Axboe 1520ed29b0b4SJens Axboe /* 1521ed29b0b4SJens Axboe * We don't need to recycle for REQ_F_BUFFER_RING, we can just clear 1522ed29b0b4SJens Axboe * the flag and hence ensure that bl->head doesn't get incremented. 1523ed29b0b4SJens Axboe * If the tail has already been incremented, hang on to it. 1524ed29b0b4SJens Axboe */ 1525ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_RING) { 1526ed29b0b4SJens Axboe if (req->buf_list) { 1527ed29b0b4SJens Axboe if (req->flags & REQ_F_PARTIAL_IO) { 1528ed29b0b4SJens Axboe req->buf_list->head++; 1529ed29b0b4SJens Axboe req->buf_list = NULL; 1530ed29b0b4SJens Axboe } else { 1531ed29b0b4SJens Axboe req->buf_index = req->buf_list->bgid; 1532ed29b0b4SJens Axboe req->flags &= ~REQ_F_BUFFER_RING; 1533ed29b0b4SJens Axboe } 1534ed29b0b4SJens Axboe } 1535ed29b0b4SJens Axboe return; 1536ed29b0b4SJens Axboe } 1537ed29b0b4SJens Axboe 1538ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 1539ed29b0b4SJens Axboe 1540ed29b0b4SJens Axboe buf = req->kbuf; 1541ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, buf->bgid); 1542ed29b0b4SJens Axboe list_add(&buf->list, &bl->buf_list); 1543ed29b0b4SJens Axboe req->flags &= ~REQ_F_BUFFER_SELECTED; 1544ed29b0b4SJens Axboe req->buf_index = buf->bgid; 1545ed29b0b4SJens Axboe 1546ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 1547ed29b0b4SJens Axboe } 1548ed29b0b4SJens Axboe 1549ed29b0b4SJens Axboe static bool io_match_task(struct io_kiocb *head, struct task_struct *task, 1550ed29b0b4SJens Axboe bool cancel_all) 1551ed29b0b4SJens Axboe __must_hold(&req->ctx->timeout_lock) 1552ed29b0b4SJens Axboe { 1553ed29b0b4SJens Axboe struct io_kiocb *req; 1554ed29b0b4SJens Axboe 1555ed29b0b4SJens Axboe if (task && head->task != task) 1556ed29b0b4SJens Axboe return false; 1557ed29b0b4SJens Axboe if (cancel_all) 1558ed29b0b4SJens Axboe return true; 1559ed29b0b4SJens Axboe 1560ed29b0b4SJens Axboe io_for_each_link(req, head) { 1561ed29b0b4SJens Axboe if (req->flags & REQ_F_INFLIGHT) 1562ed29b0b4SJens Axboe return true; 1563ed29b0b4SJens Axboe } 1564ed29b0b4SJens Axboe return false; 1565ed29b0b4SJens Axboe } 1566ed29b0b4SJens Axboe 1567ed29b0b4SJens Axboe static bool io_match_linked(struct io_kiocb *head) 1568ed29b0b4SJens Axboe { 1569ed29b0b4SJens Axboe struct io_kiocb *req; 1570ed29b0b4SJens Axboe 1571ed29b0b4SJens Axboe io_for_each_link(req, head) { 1572ed29b0b4SJens Axboe if (req->flags & REQ_F_INFLIGHT) 1573ed29b0b4SJens Axboe return true; 1574ed29b0b4SJens Axboe } 1575ed29b0b4SJens Axboe return false; 1576ed29b0b4SJens Axboe } 1577ed29b0b4SJens Axboe 1578ed29b0b4SJens Axboe /* 1579ed29b0b4SJens Axboe * As io_match_task() but protected against racing with linked timeouts. 1580ed29b0b4SJens Axboe * User must not hold timeout_lock. 1581ed29b0b4SJens Axboe */ 1582ed29b0b4SJens Axboe static bool io_match_task_safe(struct io_kiocb *head, struct task_struct *task, 1583ed29b0b4SJens Axboe bool cancel_all) 1584ed29b0b4SJens Axboe { 1585ed29b0b4SJens Axboe bool matched; 1586ed29b0b4SJens Axboe 1587ed29b0b4SJens Axboe if (task && head->task != task) 1588ed29b0b4SJens Axboe return false; 1589ed29b0b4SJens Axboe if (cancel_all) 1590ed29b0b4SJens Axboe return true; 1591ed29b0b4SJens Axboe 1592ed29b0b4SJens Axboe if (head->flags & REQ_F_LINK_TIMEOUT) { 1593ed29b0b4SJens Axboe struct io_ring_ctx *ctx = head->ctx; 1594ed29b0b4SJens Axboe 1595ed29b0b4SJens Axboe /* protect against races with linked timeouts */ 1596ed29b0b4SJens Axboe spin_lock_irq(&ctx->timeout_lock); 1597ed29b0b4SJens Axboe matched = io_match_linked(head); 1598ed29b0b4SJens Axboe spin_unlock_irq(&ctx->timeout_lock); 1599ed29b0b4SJens Axboe } else { 1600ed29b0b4SJens Axboe matched = io_match_linked(head); 1601ed29b0b4SJens Axboe } 1602ed29b0b4SJens Axboe return matched; 1603ed29b0b4SJens Axboe } 1604ed29b0b4SJens Axboe 1605ed29b0b4SJens Axboe static inline bool req_has_async_data(struct io_kiocb *req) 1606ed29b0b4SJens Axboe { 1607ed29b0b4SJens Axboe return req->flags & REQ_F_ASYNC_DATA; 1608ed29b0b4SJens Axboe } 1609ed29b0b4SJens Axboe 1610ed29b0b4SJens Axboe static inline void req_set_fail(struct io_kiocb *req) 1611ed29b0b4SJens Axboe { 1612ed29b0b4SJens Axboe req->flags |= REQ_F_FAIL; 1613ed29b0b4SJens Axboe if (req->flags & REQ_F_CQE_SKIP) { 1614ed29b0b4SJens Axboe req->flags &= ~REQ_F_CQE_SKIP; 1615ed29b0b4SJens Axboe req->flags |= REQ_F_SKIP_LINK_CQES; 1616ed29b0b4SJens Axboe } 1617ed29b0b4SJens Axboe } 1618ed29b0b4SJens Axboe 1619ed29b0b4SJens Axboe static inline void req_fail_link_node(struct io_kiocb *req, int res) 1620ed29b0b4SJens Axboe { 1621ed29b0b4SJens Axboe req_set_fail(req); 1622ed29b0b4SJens Axboe req->cqe.res = res; 1623ed29b0b4SJens Axboe } 1624ed29b0b4SJens Axboe 1625ed29b0b4SJens Axboe static inline void io_req_add_to_cache(struct io_kiocb *req, struct io_ring_ctx *ctx) 1626ed29b0b4SJens Axboe { 1627ed29b0b4SJens Axboe wq_stack_add_head(&req->comp_list, &ctx->submit_state.free_list); 1628ed29b0b4SJens Axboe } 1629ed29b0b4SJens Axboe 1630ed29b0b4SJens Axboe static __cold void io_ring_ctx_ref_free(struct percpu_ref *ref) 1631ed29b0b4SJens Axboe { 1632ed29b0b4SJens Axboe struct io_ring_ctx *ctx = container_of(ref, struct io_ring_ctx, refs); 1633ed29b0b4SJens Axboe 1634ed29b0b4SJens Axboe complete(&ctx->ref_comp); 1635ed29b0b4SJens Axboe } 1636ed29b0b4SJens Axboe 1637ed29b0b4SJens Axboe static inline bool io_is_timeout_noseq(struct io_kiocb *req) 1638ed29b0b4SJens Axboe { 1639a43714acSJens Axboe struct io_timeout *timeout = io_kiocb_to_cmd(req); 1640a43714acSJens Axboe 1641a43714acSJens Axboe return !timeout->off; 1642ed29b0b4SJens Axboe } 1643ed29b0b4SJens Axboe 1644ed29b0b4SJens Axboe static __cold void io_fallback_req_func(struct work_struct *work) 1645ed29b0b4SJens Axboe { 1646ed29b0b4SJens Axboe struct io_ring_ctx *ctx = container_of(work, struct io_ring_ctx, 1647ed29b0b4SJens Axboe fallback_work.work); 1648ed29b0b4SJens Axboe struct llist_node *node = llist_del_all(&ctx->fallback_llist); 1649ed29b0b4SJens Axboe struct io_kiocb *req, *tmp; 1650ed29b0b4SJens Axboe bool locked = false; 1651ed29b0b4SJens Axboe 1652ed29b0b4SJens Axboe percpu_ref_get(&ctx->refs); 1653ed29b0b4SJens Axboe llist_for_each_entry_safe(req, tmp, node, io_task_work.fallback_node) 1654ed29b0b4SJens Axboe req->io_task_work.func(req, &locked); 1655ed29b0b4SJens Axboe 1656ed29b0b4SJens Axboe if (locked) { 1657ed29b0b4SJens Axboe io_submit_flush_completions(ctx); 1658ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 1659ed29b0b4SJens Axboe } 1660ed29b0b4SJens Axboe percpu_ref_put(&ctx->refs); 1661ed29b0b4SJens Axboe } 1662ed29b0b4SJens Axboe 1663ed29b0b4SJens Axboe static __cold struct io_ring_ctx *io_ring_ctx_alloc(struct io_uring_params *p) 1664ed29b0b4SJens Axboe { 1665ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 1666ed29b0b4SJens Axboe int hash_bits; 1667ed29b0b4SJens Axboe 1668ed29b0b4SJens Axboe ctx = kzalloc(sizeof(*ctx), GFP_KERNEL); 1669ed29b0b4SJens Axboe if (!ctx) 1670ed29b0b4SJens Axboe return NULL; 1671ed29b0b4SJens Axboe 1672ed29b0b4SJens Axboe xa_init(&ctx->io_bl_xa); 1673ed29b0b4SJens Axboe 1674ed29b0b4SJens Axboe /* 1675ed29b0b4SJens Axboe * Use 5 bits less than the max cq entries, that should give us around 1676ed29b0b4SJens Axboe * 32 entries per hash list if totally full and uniformly spread. 1677ed29b0b4SJens Axboe */ 1678ed29b0b4SJens Axboe hash_bits = ilog2(p->cq_entries); 1679ed29b0b4SJens Axboe hash_bits -= 5; 1680ed29b0b4SJens Axboe if (hash_bits <= 0) 1681ed29b0b4SJens Axboe hash_bits = 1; 1682ed29b0b4SJens Axboe ctx->cancel_hash_bits = hash_bits; 1683ed29b0b4SJens Axboe ctx->cancel_hash = kmalloc((1U << hash_bits) * sizeof(struct hlist_head), 1684ed29b0b4SJens Axboe GFP_KERNEL); 1685ed29b0b4SJens Axboe if (!ctx->cancel_hash) 1686ed29b0b4SJens Axboe goto err; 1687ed29b0b4SJens Axboe __hash_init(ctx->cancel_hash, 1U << hash_bits); 1688ed29b0b4SJens Axboe 1689ed29b0b4SJens Axboe ctx->dummy_ubuf = kzalloc(sizeof(*ctx->dummy_ubuf), GFP_KERNEL); 1690ed29b0b4SJens Axboe if (!ctx->dummy_ubuf) 1691ed29b0b4SJens Axboe goto err; 1692ed29b0b4SJens Axboe /* set invalid range, so io_import_fixed() fails meeting it */ 1693ed29b0b4SJens Axboe ctx->dummy_ubuf->ubuf = -1UL; 1694ed29b0b4SJens Axboe 1695ed29b0b4SJens Axboe if (percpu_ref_init(&ctx->refs, io_ring_ctx_ref_free, 1696ed29b0b4SJens Axboe PERCPU_REF_ALLOW_REINIT, GFP_KERNEL)) 1697ed29b0b4SJens Axboe goto err; 1698ed29b0b4SJens Axboe 1699ed29b0b4SJens Axboe ctx->flags = p->flags; 1700ed29b0b4SJens Axboe init_waitqueue_head(&ctx->sqo_sq_wait); 1701ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->sqd_list); 1702ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->cq_overflow_list); 1703ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->io_buffers_cache); 1704ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->apoll_cache); 1705ed29b0b4SJens Axboe init_completion(&ctx->ref_comp); 1706ed29b0b4SJens Axboe xa_init_flags(&ctx->personalities, XA_FLAGS_ALLOC1); 1707ed29b0b4SJens Axboe mutex_init(&ctx->uring_lock); 1708ed29b0b4SJens Axboe init_waitqueue_head(&ctx->cq_wait); 1709ed29b0b4SJens Axboe spin_lock_init(&ctx->completion_lock); 1710ed29b0b4SJens Axboe spin_lock_init(&ctx->timeout_lock); 1711ed29b0b4SJens Axboe INIT_WQ_LIST(&ctx->iopoll_list); 1712ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->io_buffers_pages); 1713ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->io_buffers_comp); 1714ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->defer_list); 1715ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->timeout_list); 1716ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->ltimeout_list); 1717ed29b0b4SJens Axboe spin_lock_init(&ctx->rsrc_ref_lock); 1718ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->rsrc_ref_list); 1719ed29b0b4SJens Axboe INIT_DELAYED_WORK(&ctx->rsrc_put_work, io_rsrc_put_work); 1720ed29b0b4SJens Axboe init_llist_head(&ctx->rsrc_put_llist); 1721ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->tctx_list); 1722ed29b0b4SJens Axboe ctx->submit_state.free_list.next = NULL; 1723ed29b0b4SJens Axboe INIT_WQ_LIST(&ctx->locked_free_list); 1724ed29b0b4SJens Axboe INIT_DELAYED_WORK(&ctx->fallback_work, io_fallback_req_func); 1725ed29b0b4SJens Axboe INIT_WQ_LIST(&ctx->submit_state.compl_reqs); 1726ed29b0b4SJens Axboe return ctx; 1727ed29b0b4SJens Axboe err: 1728ed29b0b4SJens Axboe kfree(ctx->dummy_ubuf); 1729ed29b0b4SJens Axboe kfree(ctx->cancel_hash); 1730ed29b0b4SJens Axboe kfree(ctx->io_bl); 1731ed29b0b4SJens Axboe xa_destroy(&ctx->io_bl_xa); 1732ed29b0b4SJens Axboe kfree(ctx); 1733ed29b0b4SJens Axboe return NULL; 1734ed29b0b4SJens Axboe } 1735ed29b0b4SJens Axboe 1736ed29b0b4SJens Axboe static void io_account_cq_overflow(struct io_ring_ctx *ctx) 1737ed29b0b4SJens Axboe { 1738ed29b0b4SJens Axboe struct io_rings *r = ctx->rings; 1739ed29b0b4SJens Axboe 1740ed29b0b4SJens Axboe WRITE_ONCE(r->cq_overflow, READ_ONCE(r->cq_overflow) + 1); 1741ed29b0b4SJens Axboe ctx->cq_extra--; 1742ed29b0b4SJens Axboe } 1743ed29b0b4SJens Axboe 1744ed29b0b4SJens Axboe static bool req_need_defer(struct io_kiocb *req, u32 seq) 1745ed29b0b4SJens Axboe { 1746ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_IO_DRAIN)) { 1747ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 1748ed29b0b4SJens Axboe 1749ed29b0b4SJens Axboe return seq + READ_ONCE(ctx->cq_extra) != ctx->cached_cq_tail; 1750ed29b0b4SJens Axboe } 1751ed29b0b4SJens Axboe 1752ed29b0b4SJens Axboe return false; 1753ed29b0b4SJens Axboe } 1754ed29b0b4SJens Axboe 1755ed29b0b4SJens Axboe static inline bool io_req_ffs_set(struct io_kiocb *req) 1756ed29b0b4SJens Axboe { 1757ed29b0b4SJens Axboe return req->flags & REQ_F_FIXED_FILE; 1758ed29b0b4SJens Axboe } 1759ed29b0b4SJens Axboe 1760ed29b0b4SJens Axboe static inline void io_req_track_inflight(struct io_kiocb *req) 1761ed29b0b4SJens Axboe { 1762ed29b0b4SJens Axboe if (!(req->flags & REQ_F_INFLIGHT)) { 1763ed29b0b4SJens Axboe req->flags |= REQ_F_INFLIGHT; 1764ed29b0b4SJens Axboe atomic_inc(&req->task->io_uring->inflight_tracked); 1765ed29b0b4SJens Axboe } 1766ed29b0b4SJens Axboe } 1767ed29b0b4SJens Axboe 1768ed29b0b4SJens Axboe static struct io_kiocb *__io_prep_linked_timeout(struct io_kiocb *req) 1769ed29b0b4SJens Axboe { 1770ed29b0b4SJens Axboe if (WARN_ON_ONCE(!req->link)) 1771ed29b0b4SJens Axboe return NULL; 1772ed29b0b4SJens Axboe 1773ed29b0b4SJens Axboe req->flags &= ~REQ_F_ARM_LTIMEOUT; 1774ed29b0b4SJens Axboe req->flags |= REQ_F_LINK_TIMEOUT; 1775ed29b0b4SJens Axboe 1776ed29b0b4SJens Axboe /* linked timeouts should have two refs once prep'ed */ 1777ed29b0b4SJens Axboe io_req_set_refcount(req); 1778ed29b0b4SJens Axboe __io_req_set_refcount(req->link, 2); 1779ed29b0b4SJens Axboe return req->link; 1780ed29b0b4SJens Axboe } 1781ed29b0b4SJens Axboe 1782ed29b0b4SJens Axboe static inline struct io_kiocb *io_prep_linked_timeout(struct io_kiocb *req) 1783ed29b0b4SJens Axboe { 1784ed29b0b4SJens Axboe if (likely(!(req->flags & REQ_F_ARM_LTIMEOUT))) 1785ed29b0b4SJens Axboe return NULL; 1786ed29b0b4SJens Axboe return __io_prep_linked_timeout(req); 1787ed29b0b4SJens Axboe } 1788ed29b0b4SJens Axboe 1789ed29b0b4SJens Axboe static noinline void __io_arm_ltimeout(struct io_kiocb *req) 1790ed29b0b4SJens Axboe { 1791ed29b0b4SJens Axboe io_queue_linked_timeout(__io_prep_linked_timeout(req)); 1792ed29b0b4SJens Axboe } 1793ed29b0b4SJens Axboe 1794ed29b0b4SJens Axboe static inline void io_arm_ltimeout(struct io_kiocb *req) 1795ed29b0b4SJens Axboe { 1796ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_ARM_LTIMEOUT)) 1797ed29b0b4SJens Axboe __io_arm_ltimeout(req); 1798ed29b0b4SJens Axboe } 1799ed29b0b4SJens Axboe 1800ed29b0b4SJens Axboe static void io_prep_async_work(struct io_kiocb *req) 1801ed29b0b4SJens Axboe { 1802ed29b0b4SJens Axboe const struct io_op_def *def = &io_op_defs[req->opcode]; 1803ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 1804ed29b0b4SJens Axboe 1805ed29b0b4SJens Axboe if (!(req->flags & REQ_F_CREDS)) { 1806ed29b0b4SJens Axboe req->flags |= REQ_F_CREDS; 1807ed29b0b4SJens Axboe req->creds = get_current_cred(); 1808ed29b0b4SJens Axboe } 1809ed29b0b4SJens Axboe 1810ed29b0b4SJens Axboe req->work.list.next = NULL; 1811ed29b0b4SJens Axboe req->work.flags = 0; 1812ed29b0b4SJens Axboe req->work.cancel_seq = atomic_read(&ctx->cancel_seq); 1813ed29b0b4SJens Axboe if (req->flags & REQ_F_FORCE_ASYNC) 1814ed29b0b4SJens Axboe req->work.flags |= IO_WQ_WORK_CONCURRENT; 1815ed29b0b4SJens Axboe 1816ed29b0b4SJens Axboe if (req->flags & REQ_F_ISREG) { 1817ed29b0b4SJens Axboe if (def->hash_reg_file || (ctx->flags & IORING_SETUP_IOPOLL)) 1818ed29b0b4SJens Axboe io_wq_hash_work(&req->work, file_inode(req->file)); 1819ed29b0b4SJens Axboe } else if (!req->file || !S_ISBLK(file_inode(req->file)->i_mode)) { 1820ed29b0b4SJens Axboe if (def->unbound_nonreg_file) 1821ed29b0b4SJens Axboe req->work.flags |= IO_WQ_WORK_UNBOUND; 1822ed29b0b4SJens Axboe } 1823ed29b0b4SJens Axboe } 1824ed29b0b4SJens Axboe 1825ed29b0b4SJens Axboe static void io_prep_async_link(struct io_kiocb *req) 1826ed29b0b4SJens Axboe { 1827ed29b0b4SJens Axboe struct io_kiocb *cur; 1828ed29b0b4SJens Axboe 1829ed29b0b4SJens Axboe if (req->flags & REQ_F_LINK_TIMEOUT) { 1830ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 1831ed29b0b4SJens Axboe 1832ed29b0b4SJens Axboe spin_lock_irq(&ctx->timeout_lock); 1833ed29b0b4SJens Axboe io_for_each_link(cur, req) 1834ed29b0b4SJens Axboe io_prep_async_work(cur); 1835ed29b0b4SJens Axboe spin_unlock_irq(&ctx->timeout_lock); 1836ed29b0b4SJens Axboe } else { 1837ed29b0b4SJens Axboe io_for_each_link(cur, req) 1838ed29b0b4SJens Axboe io_prep_async_work(cur); 1839ed29b0b4SJens Axboe } 1840ed29b0b4SJens Axboe } 1841ed29b0b4SJens Axboe 1842ed29b0b4SJens Axboe static inline void io_req_add_compl_list(struct io_kiocb *req) 1843ed29b0b4SJens Axboe { 1844ed29b0b4SJens Axboe struct io_submit_state *state = &req->ctx->submit_state; 1845ed29b0b4SJens Axboe 1846ed29b0b4SJens Axboe if (!(req->flags & REQ_F_CQE_SKIP)) 1847ed29b0b4SJens Axboe state->flush_cqes = true; 1848ed29b0b4SJens Axboe wq_list_add_tail(&req->comp_list, &state->compl_reqs); 1849ed29b0b4SJens Axboe } 1850ed29b0b4SJens Axboe 1851ed29b0b4SJens Axboe static void io_queue_iowq(struct io_kiocb *req, bool *dont_use) 1852ed29b0b4SJens Axboe { 1853ed29b0b4SJens Axboe struct io_kiocb *link = io_prep_linked_timeout(req); 1854ed29b0b4SJens Axboe struct io_uring_task *tctx = req->task->io_uring; 1855ed29b0b4SJens Axboe 1856ed29b0b4SJens Axboe BUG_ON(!tctx); 1857ed29b0b4SJens Axboe BUG_ON(!tctx->io_wq); 1858ed29b0b4SJens Axboe 1859ed29b0b4SJens Axboe /* init ->work of the whole link before punting */ 1860ed29b0b4SJens Axboe io_prep_async_link(req); 1861ed29b0b4SJens Axboe 1862ed29b0b4SJens Axboe /* 1863ed29b0b4SJens Axboe * Not expected to happen, but if we do have a bug where this _can_ 1864ed29b0b4SJens Axboe * happen, catch it here and ensure the request is marked as 1865ed29b0b4SJens Axboe * canceled. That will make io-wq go through the usual work cancel 1866ed29b0b4SJens Axboe * procedure rather than attempt to run this request (or create a new 1867ed29b0b4SJens Axboe * worker for it). 1868ed29b0b4SJens Axboe */ 1869ed29b0b4SJens Axboe if (WARN_ON_ONCE(!same_thread_group(req->task, current))) 1870ed29b0b4SJens Axboe req->work.flags |= IO_WQ_WORK_CANCEL; 1871ed29b0b4SJens Axboe 1872ed29b0b4SJens Axboe trace_io_uring_queue_async_work(req->ctx, req, req->cqe.user_data, 1873ed29b0b4SJens Axboe req->opcode, req->flags, &req->work, 1874ed29b0b4SJens Axboe io_wq_is_hashed(&req->work)); 1875ed29b0b4SJens Axboe io_wq_enqueue(tctx->io_wq, &req->work); 1876ed29b0b4SJens Axboe if (link) 1877ed29b0b4SJens Axboe io_queue_linked_timeout(link); 1878ed29b0b4SJens Axboe } 1879ed29b0b4SJens Axboe 1880ed29b0b4SJens Axboe static void io_kill_timeout(struct io_kiocb *req, int status) 1881ed29b0b4SJens Axboe __must_hold(&req->ctx->completion_lock) 1882ed29b0b4SJens Axboe __must_hold(&req->ctx->timeout_lock) 1883ed29b0b4SJens Axboe { 1884ed29b0b4SJens Axboe struct io_timeout_data *io = req->async_data; 1885ed29b0b4SJens Axboe 1886ed29b0b4SJens Axboe if (hrtimer_try_to_cancel(&io->timer) != -1) { 1887a43714acSJens Axboe struct io_timeout *timeout = io_kiocb_to_cmd(req); 1888a43714acSJens Axboe 1889ed29b0b4SJens Axboe if (status) 1890ed29b0b4SJens Axboe req_set_fail(req); 1891ed29b0b4SJens Axboe atomic_set(&req->ctx->cq_timeouts, 1892ed29b0b4SJens Axboe atomic_read(&req->ctx->cq_timeouts) + 1); 1893a43714acSJens Axboe list_del_init(&timeout->list); 1894ed29b0b4SJens Axboe io_req_tw_post_queue(req, status, 0); 1895ed29b0b4SJens Axboe } 1896ed29b0b4SJens Axboe } 1897ed29b0b4SJens Axboe 1898ed29b0b4SJens Axboe static __cold void io_queue_deferred(struct io_ring_ctx *ctx) 1899ed29b0b4SJens Axboe { 1900ed29b0b4SJens Axboe while (!list_empty(&ctx->defer_list)) { 1901ed29b0b4SJens Axboe struct io_defer_entry *de = list_first_entry(&ctx->defer_list, 1902ed29b0b4SJens Axboe struct io_defer_entry, list); 1903ed29b0b4SJens Axboe 1904ed29b0b4SJens Axboe if (req_need_defer(de->req, de->seq)) 1905ed29b0b4SJens Axboe break; 1906ed29b0b4SJens Axboe list_del_init(&de->list); 1907ed29b0b4SJens Axboe io_req_task_queue(de->req); 1908ed29b0b4SJens Axboe kfree(de); 1909ed29b0b4SJens Axboe } 1910ed29b0b4SJens Axboe } 1911ed29b0b4SJens Axboe 1912ed29b0b4SJens Axboe static __cold void io_flush_timeouts(struct io_ring_ctx *ctx) 1913ed29b0b4SJens Axboe __must_hold(&ctx->completion_lock) 1914ed29b0b4SJens Axboe { 1915ed29b0b4SJens Axboe u32 seq = ctx->cached_cq_tail - atomic_read(&ctx->cq_timeouts); 1916a43714acSJens Axboe struct io_timeout *timeout, *tmp; 1917ed29b0b4SJens Axboe 1918ed29b0b4SJens Axboe spin_lock_irq(&ctx->timeout_lock); 1919a43714acSJens Axboe list_for_each_entry_safe(timeout, tmp, &ctx->timeout_list, list) { 1920a43714acSJens Axboe struct io_kiocb *req = cmd_to_io_kiocb(timeout); 1921ed29b0b4SJens Axboe u32 events_needed, events_got; 1922ed29b0b4SJens Axboe 1923ed29b0b4SJens Axboe if (io_is_timeout_noseq(req)) 1924ed29b0b4SJens Axboe break; 1925ed29b0b4SJens Axboe 1926ed29b0b4SJens Axboe /* 1927ed29b0b4SJens Axboe * Since seq can easily wrap around over time, subtract 1928ed29b0b4SJens Axboe * the last seq at which timeouts were flushed before comparing. 1929ed29b0b4SJens Axboe * Assuming not more than 2^31-1 events have happened since, 1930ed29b0b4SJens Axboe * these subtractions won't have wrapped, so we can check if 1931ed29b0b4SJens Axboe * target is in [last_seq, current_seq] by comparing the two. 1932ed29b0b4SJens Axboe */ 1933a43714acSJens Axboe events_needed = timeout->target_seq - ctx->cq_last_tm_flush; 1934ed29b0b4SJens Axboe events_got = seq - ctx->cq_last_tm_flush; 1935ed29b0b4SJens Axboe if (events_got < events_needed) 1936ed29b0b4SJens Axboe break; 1937ed29b0b4SJens Axboe 1938ed29b0b4SJens Axboe io_kill_timeout(req, 0); 1939ed29b0b4SJens Axboe } 1940ed29b0b4SJens Axboe ctx->cq_last_tm_flush = seq; 1941ed29b0b4SJens Axboe spin_unlock_irq(&ctx->timeout_lock); 1942ed29b0b4SJens Axboe } 1943ed29b0b4SJens Axboe 1944ed29b0b4SJens Axboe static inline void io_commit_cqring(struct io_ring_ctx *ctx) 1945ed29b0b4SJens Axboe { 1946ed29b0b4SJens Axboe /* order cqe stores with ring update */ 1947ed29b0b4SJens Axboe smp_store_release(&ctx->rings->cq.tail, ctx->cached_cq_tail); 1948ed29b0b4SJens Axboe } 1949ed29b0b4SJens Axboe 1950ed29b0b4SJens Axboe static void __io_commit_cqring_flush(struct io_ring_ctx *ctx) 1951ed29b0b4SJens Axboe { 1952ed29b0b4SJens Axboe if (ctx->off_timeout_used || ctx->drain_active) { 1953ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 1954ed29b0b4SJens Axboe if (ctx->off_timeout_used) 1955ed29b0b4SJens Axboe io_flush_timeouts(ctx); 1956ed29b0b4SJens Axboe if (ctx->drain_active) 1957ed29b0b4SJens Axboe io_queue_deferred(ctx); 1958ed29b0b4SJens Axboe io_commit_cqring(ctx); 1959ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 1960ed29b0b4SJens Axboe } 1961ed29b0b4SJens Axboe if (ctx->has_evfd) 1962ed29b0b4SJens Axboe io_eventfd_signal(ctx); 1963ed29b0b4SJens Axboe } 1964ed29b0b4SJens Axboe 1965ed29b0b4SJens Axboe static inline bool io_sqring_full(struct io_ring_ctx *ctx) 1966ed29b0b4SJens Axboe { 1967ed29b0b4SJens Axboe struct io_rings *r = ctx->rings; 1968ed29b0b4SJens Axboe 1969ed29b0b4SJens Axboe return READ_ONCE(r->sq.tail) - ctx->cached_sq_head == ctx->sq_entries; 1970ed29b0b4SJens Axboe } 1971ed29b0b4SJens Axboe 1972ed29b0b4SJens Axboe static inline unsigned int __io_cqring_events(struct io_ring_ctx *ctx) 1973ed29b0b4SJens Axboe { 1974ed29b0b4SJens Axboe return ctx->cached_cq_tail - READ_ONCE(ctx->rings->cq.head); 1975ed29b0b4SJens Axboe } 1976ed29b0b4SJens Axboe 1977ed29b0b4SJens Axboe /* 1978ed29b0b4SJens Axboe * writes to the cq entry need to come after reading head; the 1979ed29b0b4SJens Axboe * control dependency is enough as we're using WRITE_ONCE to 1980ed29b0b4SJens Axboe * fill the cq entry 1981ed29b0b4SJens Axboe */ 1982ed29b0b4SJens Axboe static noinline struct io_uring_cqe *__io_get_cqe(struct io_ring_ctx *ctx) 1983ed29b0b4SJens Axboe { 1984ed29b0b4SJens Axboe struct io_rings *rings = ctx->rings; 1985ed29b0b4SJens Axboe unsigned int off = ctx->cached_cq_tail & (ctx->cq_entries - 1); 1986ed29b0b4SJens Axboe unsigned int shift = 0; 1987ed29b0b4SJens Axboe unsigned int free, queued, len; 1988ed29b0b4SJens Axboe 1989ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_CQE32) 1990ed29b0b4SJens Axboe shift = 1; 1991ed29b0b4SJens Axboe 1992ed29b0b4SJens Axboe /* userspace may cheat modifying the tail, be safe and do min */ 1993ed29b0b4SJens Axboe queued = min(__io_cqring_events(ctx), ctx->cq_entries); 1994ed29b0b4SJens Axboe free = ctx->cq_entries - queued; 1995ed29b0b4SJens Axboe /* we need a contiguous range, limit based on the current array offset */ 1996ed29b0b4SJens Axboe len = min(free, ctx->cq_entries - off); 1997ed29b0b4SJens Axboe if (!len) 1998ed29b0b4SJens Axboe return NULL; 1999ed29b0b4SJens Axboe 2000ed29b0b4SJens Axboe ctx->cached_cq_tail++; 2001ed29b0b4SJens Axboe ctx->cqe_cached = &rings->cqes[off]; 2002ed29b0b4SJens Axboe ctx->cqe_sentinel = ctx->cqe_cached + len; 2003ed29b0b4SJens Axboe ctx->cqe_cached++; 2004ed29b0b4SJens Axboe return &rings->cqes[off << shift]; 2005ed29b0b4SJens Axboe } 2006ed29b0b4SJens Axboe 2007ed29b0b4SJens Axboe static inline struct io_uring_cqe *io_get_cqe(struct io_ring_ctx *ctx) 2008ed29b0b4SJens Axboe { 2009ed29b0b4SJens Axboe if (likely(ctx->cqe_cached < ctx->cqe_sentinel)) { 2010ed29b0b4SJens Axboe struct io_uring_cqe *cqe = ctx->cqe_cached; 2011ed29b0b4SJens Axboe 2012ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_CQE32) { 2013ed29b0b4SJens Axboe unsigned int off = ctx->cqe_cached - ctx->rings->cqes; 2014ed29b0b4SJens Axboe 2015ed29b0b4SJens Axboe cqe += off; 2016ed29b0b4SJens Axboe } 2017ed29b0b4SJens Axboe 2018ed29b0b4SJens Axboe ctx->cached_cq_tail++; 2019ed29b0b4SJens Axboe ctx->cqe_cached++; 2020ed29b0b4SJens Axboe return cqe; 2021ed29b0b4SJens Axboe } 2022ed29b0b4SJens Axboe 2023ed29b0b4SJens Axboe return __io_get_cqe(ctx); 2024ed29b0b4SJens Axboe } 2025ed29b0b4SJens Axboe 2026ed29b0b4SJens Axboe static void io_eventfd_signal(struct io_ring_ctx *ctx) 2027ed29b0b4SJens Axboe { 2028ed29b0b4SJens Axboe struct io_ev_fd *ev_fd; 2029ed29b0b4SJens Axboe 2030ed29b0b4SJens Axboe rcu_read_lock(); 2031ed29b0b4SJens Axboe /* 2032ed29b0b4SJens Axboe * rcu_dereference ctx->io_ev_fd once and use it for both for checking 2033ed29b0b4SJens Axboe * and eventfd_signal 2034ed29b0b4SJens Axboe */ 2035ed29b0b4SJens Axboe ev_fd = rcu_dereference(ctx->io_ev_fd); 2036ed29b0b4SJens Axboe 2037ed29b0b4SJens Axboe /* 2038ed29b0b4SJens Axboe * Check again if ev_fd exists incase an io_eventfd_unregister call 2039ed29b0b4SJens Axboe * completed between the NULL check of ctx->io_ev_fd at the start of 2040ed29b0b4SJens Axboe * the function and rcu_read_lock. 2041ed29b0b4SJens Axboe */ 2042ed29b0b4SJens Axboe if (unlikely(!ev_fd)) 2043ed29b0b4SJens Axboe goto out; 2044ed29b0b4SJens Axboe if (READ_ONCE(ctx->rings->cq_flags) & IORING_CQ_EVENTFD_DISABLED) 2045ed29b0b4SJens Axboe goto out; 2046ed29b0b4SJens Axboe 2047ed29b0b4SJens Axboe if (!ev_fd->eventfd_async || io_wq_current_is_worker()) 2048ed29b0b4SJens Axboe eventfd_signal(ev_fd->cq_ev_fd, 1); 2049ed29b0b4SJens Axboe out: 2050ed29b0b4SJens Axboe rcu_read_unlock(); 2051ed29b0b4SJens Axboe } 2052ed29b0b4SJens Axboe 2053ed29b0b4SJens Axboe static inline void io_cqring_wake(struct io_ring_ctx *ctx) 2054ed29b0b4SJens Axboe { 2055ed29b0b4SJens Axboe /* 2056ed29b0b4SJens Axboe * wake_up_all() may seem excessive, but io_wake_function() and 2057ed29b0b4SJens Axboe * io_should_wake() handle the termination of the loop and only 2058ed29b0b4SJens Axboe * wake as many waiters as we need to. 2059ed29b0b4SJens Axboe */ 2060ed29b0b4SJens Axboe if (wq_has_sleeper(&ctx->cq_wait)) 2061ed29b0b4SJens Axboe wake_up_all(&ctx->cq_wait); 2062ed29b0b4SJens Axboe } 2063ed29b0b4SJens Axboe 2064ed29b0b4SJens Axboe /* 2065ed29b0b4SJens Axboe * This should only get called when at least one event has been posted. 2066ed29b0b4SJens Axboe * Some applications rely on the eventfd notification count only changing 2067ed29b0b4SJens Axboe * IFF a new CQE has been added to the CQ ring. There's no depedency on 2068ed29b0b4SJens Axboe * 1:1 relationship between how many times this function is called (and 2069ed29b0b4SJens Axboe * hence the eventfd count) and number of CQEs posted to the CQ ring. 2070ed29b0b4SJens Axboe */ 2071ed29b0b4SJens Axboe static inline void io_cqring_ev_posted(struct io_ring_ctx *ctx) 2072ed29b0b4SJens Axboe { 2073ed29b0b4SJens Axboe if (unlikely(ctx->off_timeout_used || ctx->drain_active || 2074ed29b0b4SJens Axboe ctx->has_evfd)) 2075ed29b0b4SJens Axboe __io_commit_cqring_flush(ctx); 2076ed29b0b4SJens Axboe 2077ed29b0b4SJens Axboe io_cqring_wake(ctx); 2078ed29b0b4SJens Axboe } 2079ed29b0b4SJens Axboe 2080ed29b0b4SJens Axboe static void io_cqring_ev_posted_iopoll(struct io_ring_ctx *ctx) 2081ed29b0b4SJens Axboe { 2082ed29b0b4SJens Axboe if (unlikely(ctx->off_timeout_used || ctx->drain_active || 2083ed29b0b4SJens Axboe ctx->has_evfd)) 2084ed29b0b4SJens Axboe __io_commit_cqring_flush(ctx); 2085ed29b0b4SJens Axboe 2086ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQPOLL) 2087ed29b0b4SJens Axboe io_cqring_wake(ctx); 2088ed29b0b4SJens Axboe } 2089ed29b0b4SJens Axboe 2090ed29b0b4SJens Axboe /* Returns true if there are no backlogged entries after the flush */ 2091ed29b0b4SJens Axboe static bool __io_cqring_overflow_flush(struct io_ring_ctx *ctx, bool force) 2092ed29b0b4SJens Axboe { 2093ed29b0b4SJens Axboe bool all_flushed, posted; 2094ed29b0b4SJens Axboe size_t cqe_size = sizeof(struct io_uring_cqe); 2095ed29b0b4SJens Axboe 2096ed29b0b4SJens Axboe if (!force && __io_cqring_events(ctx) == ctx->cq_entries) 2097ed29b0b4SJens Axboe return false; 2098ed29b0b4SJens Axboe 2099ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_CQE32) 2100ed29b0b4SJens Axboe cqe_size <<= 1; 2101ed29b0b4SJens Axboe 2102ed29b0b4SJens Axboe posted = false; 2103ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 2104ed29b0b4SJens Axboe while (!list_empty(&ctx->cq_overflow_list)) { 2105ed29b0b4SJens Axboe struct io_uring_cqe *cqe = io_get_cqe(ctx); 2106ed29b0b4SJens Axboe struct io_overflow_cqe *ocqe; 2107ed29b0b4SJens Axboe 2108ed29b0b4SJens Axboe if (!cqe && !force) 2109ed29b0b4SJens Axboe break; 2110ed29b0b4SJens Axboe ocqe = list_first_entry(&ctx->cq_overflow_list, 2111ed29b0b4SJens Axboe struct io_overflow_cqe, list); 2112ed29b0b4SJens Axboe if (cqe) 2113ed29b0b4SJens Axboe memcpy(cqe, &ocqe->cqe, cqe_size); 2114ed29b0b4SJens Axboe else 2115ed29b0b4SJens Axboe io_account_cq_overflow(ctx); 2116ed29b0b4SJens Axboe 2117ed29b0b4SJens Axboe posted = true; 2118ed29b0b4SJens Axboe list_del(&ocqe->list); 2119ed29b0b4SJens Axboe kfree(ocqe); 2120ed29b0b4SJens Axboe } 2121ed29b0b4SJens Axboe 2122ed29b0b4SJens Axboe all_flushed = list_empty(&ctx->cq_overflow_list); 2123ed29b0b4SJens Axboe if (all_flushed) { 2124ed29b0b4SJens Axboe clear_bit(IO_CHECK_CQ_OVERFLOW_BIT, &ctx->check_cq); 2125ed29b0b4SJens Axboe atomic_andnot(IORING_SQ_CQ_OVERFLOW, &ctx->rings->sq_flags); 2126ed29b0b4SJens Axboe } 2127ed29b0b4SJens Axboe 2128ed29b0b4SJens Axboe io_commit_cqring(ctx); 2129ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 2130ed29b0b4SJens Axboe if (posted) 2131ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 2132ed29b0b4SJens Axboe return all_flushed; 2133ed29b0b4SJens Axboe } 2134ed29b0b4SJens Axboe 2135ed29b0b4SJens Axboe static bool io_cqring_overflow_flush(struct io_ring_ctx *ctx) 2136ed29b0b4SJens Axboe { 2137ed29b0b4SJens Axboe bool ret = true; 2138ed29b0b4SJens Axboe 2139ed29b0b4SJens Axboe if (test_bit(IO_CHECK_CQ_OVERFLOW_BIT, &ctx->check_cq)) { 2140ed29b0b4SJens Axboe /* iopoll syncs against uring_lock, not completion_lock */ 2141ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL) 2142ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 2143ed29b0b4SJens Axboe ret = __io_cqring_overflow_flush(ctx, false); 2144ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL) 2145ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 2146ed29b0b4SJens Axboe } 2147ed29b0b4SJens Axboe 2148ed29b0b4SJens Axboe return ret; 2149ed29b0b4SJens Axboe } 2150ed29b0b4SJens Axboe 2151ed29b0b4SJens Axboe static void __io_put_task(struct task_struct *task, int nr) 2152ed29b0b4SJens Axboe { 2153ed29b0b4SJens Axboe struct io_uring_task *tctx = task->io_uring; 2154ed29b0b4SJens Axboe 2155ed29b0b4SJens Axboe percpu_counter_sub(&tctx->inflight, nr); 2156ed29b0b4SJens Axboe if (unlikely(atomic_read(&tctx->in_idle))) 2157ed29b0b4SJens Axboe wake_up(&tctx->wait); 2158ed29b0b4SJens Axboe put_task_struct_many(task, nr); 2159ed29b0b4SJens Axboe } 2160ed29b0b4SJens Axboe 2161ed29b0b4SJens Axboe /* must to be called somewhat shortly after putting a request */ 2162ed29b0b4SJens Axboe static inline void io_put_task(struct task_struct *task, int nr) 2163ed29b0b4SJens Axboe { 2164ed29b0b4SJens Axboe if (likely(task == current)) 2165ed29b0b4SJens Axboe task->io_uring->cached_refs += nr; 2166ed29b0b4SJens Axboe else 2167ed29b0b4SJens Axboe __io_put_task(task, nr); 2168ed29b0b4SJens Axboe } 2169ed29b0b4SJens Axboe 2170ed29b0b4SJens Axboe static void io_task_refs_refill(struct io_uring_task *tctx) 2171ed29b0b4SJens Axboe { 2172ed29b0b4SJens Axboe unsigned int refill = -tctx->cached_refs + IO_TCTX_REFS_CACHE_NR; 2173ed29b0b4SJens Axboe 2174ed29b0b4SJens Axboe percpu_counter_add(&tctx->inflight, refill); 2175ed29b0b4SJens Axboe refcount_add(refill, ¤t->usage); 2176ed29b0b4SJens Axboe tctx->cached_refs += refill; 2177ed29b0b4SJens Axboe } 2178ed29b0b4SJens Axboe 2179ed29b0b4SJens Axboe static inline void io_get_task_refs(int nr) 2180ed29b0b4SJens Axboe { 2181ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 2182ed29b0b4SJens Axboe 2183ed29b0b4SJens Axboe tctx->cached_refs -= nr; 2184ed29b0b4SJens Axboe if (unlikely(tctx->cached_refs < 0)) 2185ed29b0b4SJens Axboe io_task_refs_refill(tctx); 2186ed29b0b4SJens Axboe } 2187ed29b0b4SJens Axboe 2188ed29b0b4SJens Axboe static __cold void io_uring_drop_tctx_refs(struct task_struct *task) 2189ed29b0b4SJens Axboe { 2190ed29b0b4SJens Axboe struct io_uring_task *tctx = task->io_uring; 2191ed29b0b4SJens Axboe unsigned int refs = tctx->cached_refs; 2192ed29b0b4SJens Axboe 2193ed29b0b4SJens Axboe if (refs) { 2194ed29b0b4SJens Axboe tctx->cached_refs = 0; 2195ed29b0b4SJens Axboe percpu_counter_sub(&tctx->inflight, refs); 2196ed29b0b4SJens Axboe put_task_struct_many(task, refs); 2197ed29b0b4SJens Axboe } 2198ed29b0b4SJens Axboe } 2199ed29b0b4SJens Axboe 2200ed29b0b4SJens Axboe static bool io_cqring_event_overflow(struct io_ring_ctx *ctx, u64 user_data, 2201ed29b0b4SJens Axboe s32 res, u32 cflags, u64 extra1, 2202ed29b0b4SJens Axboe u64 extra2) 2203ed29b0b4SJens Axboe { 2204ed29b0b4SJens Axboe struct io_overflow_cqe *ocqe; 2205ed29b0b4SJens Axboe size_t ocq_size = sizeof(struct io_overflow_cqe); 2206ed29b0b4SJens Axboe bool is_cqe32 = (ctx->flags & IORING_SETUP_CQE32); 2207ed29b0b4SJens Axboe 2208ed29b0b4SJens Axboe if (is_cqe32) 2209ed29b0b4SJens Axboe ocq_size += sizeof(struct io_uring_cqe); 2210ed29b0b4SJens Axboe 2211ed29b0b4SJens Axboe ocqe = kmalloc(ocq_size, GFP_ATOMIC | __GFP_ACCOUNT); 2212ed29b0b4SJens Axboe trace_io_uring_cqe_overflow(ctx, user_data, res, cflags, ocqe); 2213ed29b0b4SJens Axboe if (!ocqe) { 2214ed29b0b4SJens Axboe /* 2215ed29b0b4SJens Axboe * If we're in ring overflow flush mode, or in task cancel mode, 2216ed29b0b4SJens Axboe * or cannot allocate an overflow entry, then we need to drop it 2217ed29b0b4SJens Axboe * on the floor. 2218ed29b0b4SJens Axboe */ 2219ed29b0b4SJens Axboe io_account_cq_overflow(ctx); 2220ed29b0b4SJens Axboe set_bit(IO_CHECK_CQ_DROPPED_BIT, &ctx->check_cq); 2221ed29b0b4SJens Axboe return false; 2222ed29b0b4SJens Axboe } 2223ed29b0b4SJens Axboe if (list_empty(&ctx->cq_overflow_list)) { 2224ed29b0b4SJens Axboe set_bit(IO_CHECK_CQ_OVERFLOW_BIT, &ctx->check_cq); 2225ed29b0b4SJens Axboe atomic_or(IORING_SQ_CQ_OVERFLOW, &ctx->rings->sq_flags); 2226ed29b0b4SJens Axboe 2227ed29b0b4SJens Axboe } 2228ed29b0b4SJens Axboe ocqe->cqe.user_data = user_data; 2229ed29b0b4SJens Axboe ocqe->cqe.res = res; 2230ed29b0b4SJens Axboe ocqe->cqe.flags = cflags; 2231ed29b0b4SJens Axboe if (is_cqe32) { 2232ed29b0b4SJens Axboe ocqe->cqe.big_cqe[0] = extra1; 2233ed29b0b4SJens Axboe ocqe->cqe.big_cqe[1] = extra2; 2234ed29b0b4SJens Axboe } 2235ed29b0b4SJens Axboe list_add_tail(&ocqe->list, &ctx->cq_overflow_list); 2236ed29b0b4SJens Axboe return true; 2237ed29b0b4SJens Axboe } 2238ed29b0b4SJens Axboe 2239ed29b0b4SJens Axboe static inline bool __io_fill_cqe_req(struct io_ring_ctx *ctx, 2240ed29b0b4SJens Axboe struct io_kiocb *req) 2241ed29b0b4SJens Axboe { 2242ed29b0b4SJens Axboe struct io_uring_cqe *cqe; 2243ed29b0b4SJens Axboe 2244ed29b0b4SJens Axboe if (!(ctx->flags & IORING_SETUP_CQE32)) { 2245ed29b0b4SJens Axboe trace_io_uring_complete(req->ctx, req, req->cqe.user_data, 2246ed29b0b4SJens Axboe req->cqe.res, req->cqe.flags, 0, 0); 2247ed29b0b4SJens Axboe 2248ed29b0b4SJens Axboe /* 2249ed29b0b4SJens Axboe * If we can't get a cq entry, userspace overflowed the 2250ed29b0b4SJens Axboe * submission (by quite a lot). Increment the overflow count in 2251ed29b0b4SJens Axboe * the ring. 2252ed29b0b4SJens Axboe */ 2253ed29b0b4SJens Axboe cqe = io_get_cqe(ctx); 2254ed29b0b4SJens Axboe if (likely(cqe)) { 2255ed29b0b4SJens Axboe memcpy(cqe, &req->cqe, sizeof(*cqe)); 2256ed29b0b4SJens Axboe return true; 2257ed29b0b4SJens Axboe } 2258ed29b0b4SJens Axboe 2259ed29b0b4SJens Axboe return io_cqring_event_overflow(ctx, req->cqe.user_data, 2260ed29b0b4SJens Axboe req->cqe.res, req->cqe.flags, 2261ed29b0b4SJens Axboe 0, 0); 2262ed29b0b4SJens Axboe } else { 2263ed29b0b4SJens Axboe u64 extra1 = 0, extra2 = 0; 2264ed29b0b4SJens Axboe 2265ed29b0b4SJens Axboe if (req->flags & REQ_F_CQE32_INIT) { 2266ed29b0b4SJens Axboe extra1 = req->extra1; 2267ed29b0b4SJens Axboe extra2 = req->extra2; 2268ed29b0b4SJens Axboe } 2269ed29b0b4SJens Axboe 2270ed29b0b4SJens Axboe trace_io_uring_complete(req->ctx, req, req->cqe.user_data, 2271ed29b0b4SJens Axboe req->cqe.res, req->cqe.flags, extra1, extra2); 2272ed29b0b4SJens Axboe 2273ed29b0b4SJens Axboe /* 2274ed29b0b4SJens Axboe * If we can't get a cq entry, userspace overflowed the 2275ed29b0b4SJens Axboe * submission (by quite a lot). Increment the overflow count in 2276ed29b0b4SJens Axboe * the ring. 2277ed29b0b4SJens Axboe */ 2278ed29b0b4SJens Axboe cqe = io_get_cqe(ctx); 2279ed29b0b4SJens Axboe if (likely(cqe)) { 2280ed29b0b4SJens Axboe memcpy(cqe, &req->cqe, sizeof(struct io_uring_cqe)); 2281ed29b0b4SJens Axboe WRITE_ONCE(cqe->big_cqe[0], extra1); 2282ed29b0b4SJens Axboe WRITE_ONCE(cqe->big_cqe[1], extra2); 2283ed29b0b4SJens Axboe return true; 2284ed29b0b4SJens Axboe } 2285ed29b0b4SJens Axboe 2286ed29b0b4SJens Axboe return io_cqring_event_overflow(ctx, req->cqe.user_data, 2287ed29b0b4SJens Axboe req->cqe.res, req->cqe.flags, 2288ed29b0b4SJens Axboe extra1, extra2); 2289ed29b0b4SJens Axboe } 2290ed29b0b4SJens Axboe } 2291ed29b0b4SJens Axboe 2292ed29b0b4SJens Axboe static noinline bool io_fill_cqe_aux(struct io_ring_ctx *ctx, u64 user_data, 2293ed29b0b4SJens Axboe s32 res, u32 cflags) 2294ed29b0b4SJens Axboe { 2295ed29b0b4SJens Axboe struct io_uring_cqe *cqe; 2296ed29b0b4SJens Axboe 2297ed29b0b4SJens Axboe ctx->cq_extra++; 2298ed29b0b4SJens Axboe trace_io_uring_complete(ctx, NULL, user_data, res, cflags, 0, 0); 2299ed29b0b4SJens Axboe 2300ed29b0b4SJens Axboe /* 2301ed29b0b4SJens Axboe * If we can't get a cq entry, userspace overflowed the 2302ed29b0b4SJens Axboe * submission (by quite a lot). Increment the overflow count in 2303ed29b0b4SJens Axboe * the ring. 2304ed29b0b4SJens Axboe */ 2305ed29b0b4SJens Axboe cqe = io_get_cqe(ctx); 2306ed29b0b4SJens Axboe if (likely(cqe)) { 2307ed29b0b4SJens Axboe WRITE_ONCE(cqe->user_data, user_data); 2308ed29b0b4SJens Axboe WRITE_ONCE(cqe->res, res); 2309ed29b0b4SJens Axboe WRITE_ONCE(cqe->flags, cflags); 2310ed29b0b4SJens Axboe 2311ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_CQE32) { 2312ed29b0b4SJens Axboe WRITE_ONCE(cqe->big_cqe[0], 0); 2313ed29b0b4SJens Axboe WRITE_ONCE(cqe->big_cqe[1], 0); 2314ed29b0b4SJens Axboe } 2315ed29b0b4SJens Axboe return true; 2316ed29b0b4SJens Axboe } 2317ed29b0b4SJens Axboe return io_cqring_event_overflow(ctx, user_data, res, cflags, 0, 0); 2318ed29b0b4SJens Axboe } 2319ed29b0b4SJens Axboe 2320ed29b0b4SJens Axboe static void __io_req_complete_put(struct io_kiocb *req) 2321ed29b0b4SJens Axboe { 2322ed29b0b4SJens Axboe /* 2323ed29b0b4SJens Axboe * If we're the last reference to this request, add to our locked 2324ed29b0b4SJens Axboe * free_list cache. 2325ed29b0b4SJens Axboe */ 2326ed29b0b4SJens Axboe if (req_ref_put_and_test(req)) { 2327ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 2328ed29b0b4SJens Axboe 2329ed29b0b4SJens Axboe if (req->flags & IO_REQ_LINK_FLAGS) { 2330ed29b0b4SJens Axboe if (req->flags & IO_DISARM_MASK) 2331ed29b0b4SJens Axboe io_disarm_next(req); 2332ed29b0b4SJens Axboe if (req->link) { 2333ed29b0b4SJens Axboe io_req_task_queue(req->link); 2334ed29b0b4SJens Axboe req->link = NULL; 2335ed29b0b4SJens Axboe } 2336ed29b0b4SJens Axboe } 2337ed29b0b4SJens Axboe io_req_put_rsrc(req); 2338ed29b0b4SJens Axboe /* 2339ed29b0b4SJens Axboe * Selected buffer deallocation in io_clean_op() assumes that 2340ed29b0b4SJens Axboe * we don't hold ->completion_lock. Clean them here to avoid 2341ed29b0b4SJens Axboe * deadlocks. 2342ed29b0b4SJens Axboe */ 2343ed29b0b4SJens Axboe io_put_kbuf_comp(req); 2344ed29b0b4SJens Axboe io_dismantle_req(req); 2345ed29b0b4SJens Axboe io_put_task(req->task, 1); 2346ed29b0b4SJens Axboe wq_list_add_head(&req->comp_list, &ctx->locked_free_list); 2347ed29b0b4SJens Axboe ctx->locked_free_nr++; 2348ed29b0b4SJens Axboe } 2349ed29b0b4SJens Axboe } 2350ed29b0b4SJens Axboe 2351ed29b0b4SJens Axboe static void __io_req_complete_post(struct io_kiocb *req, s32 res, 2352ed29b0b4SJens Axboe u32 cflags) 2353ed29b0b4SJens Axboe { 2354ed29b0b4SJens Axboe if (!(req->flags & REQ_F_CQE_SKIP)) { 2355ed29b0b4SJens Axboe req->cqe.res = res; 2356ed29b0b4SJens Axboe req->cqe.flags = cflags; 2357ed29b0b4SJens Axboe __io_fill_cqe_req(req->ctx, req); 2358ed29b0b4SJens Axboe } 2359ed29b0b4SJens Axboe __io_req_complete_put(req); 2360ed29b0b4SJens Axboe } 2361ed29b0b4SJens Axboe 2362ed29b0b4SJens Axboe static void io_req_complete_post(struct io_kiocb *req, s32 res, u32 cflags) 2363ed29b0b4SJens Axboe { 2364ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 2365ed29b0b4SJens Axboe 2366ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 2367ed29b0b4SJens Axboe __io_req_complete_post(req, res, cflags); 2368ed29b0b4SJens Axboe io_commit_cqring(ctx); 2369ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 2370ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 2371ed29b0b4SJens Axboe } 2372ed29b0b4SJens Axboe 2373ed29b0b4SJens Axboe static inline void io_req_complete_state(struct io_kiocb *req, s32 res, 2374ed29b0b4SJens Axboe u32 cflags) 2375ed29b0b4SJens Axboe { 2376ed29b0b4SJens Axboe req->cqe.res = res; 2377ed29b0b4SJens Axboe req->cqe.flags = cflags; 2378ed29b0b4SJens Axboe req->flags |= REQ_F_COMPLETE_INLINE; 2379ed29b0b4SJens Axboe } 2380ed29b0b4SJens Axboe 2381ed29b0b4SJens Axboe static inline void __io_req_complete(struct io_kiocb *req, unsigned issue_flags, 2382ed29b0b4SJens Axboe s32 res, u32 cflags) 2383ed29b0b4SJens Axboe { 2384ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_COMPLETE_DEFER) 2385ed29b0b4SJens Axboe io_req_complete_state(req, res, cflags); 2386ed29b0b4SJens Axboe else 2387ed29b0b4SJens Axboe io_req_complete_post(req, res, cflags); 2388ed29b0b4SJens Axboe } 2389ed29b0b4SJens Axboe 2390ed29b0b4SJens Axboe static inline void io_req_complete(struct io_kiocb *req, s32 res) 2391ed29b0b4SJens Axboe { 2392ed29b0b4SJens Axboe if (res < 0) 2393ed29b0b4SJens Axboe req_set_fail(req); 2394ed29b0b4SJens Axboe __io_req_complete(req, 0, res, 0); 2395ed29b0b4SJens Axboe } 2396ed29b0b4SJens Axboe 2397ed29b0b4SJens Axboe static void io_req_complete_failed(struct io_kiocb *req, s32 res) 2398ed29b0b4SJens Axboe { 2399ed29b0b4SJens Axboe req_set_fail(req); 2400ed29b0b4SJens Axboe io_req_complete_post(req, res, io_put_kbuf(req, IO_URING_F_UNLOCKED)); 2401ed29b0b4SJens Axboe } 2402ed29b0b4SJens Axboe 2403ed29b0b4SJens Axboe /* 2404ed29b0b4SJens Axboe * Don't initialise the fields below on every allocation, but do that in 2405ed29b0b4SJens Axboe * advance and keep them valid across allocations. 2406ed29b0b4SJens Axboe */ 2407ed29b0b4SJens Axboe static void io_preinit_req(struct io_kiocb *req, struct io_ring_ctx *ctx) 2408ed29b0b4SJens Axboe { 2409ed29b0b4SJens Axboe req->ctx = ctx; 2410ed29b0b4SJens Axboe req->link = NULL; 2411ed29b0b4SJens Axboe req->async_data = NULL; 2412ed29b0b4SJens Axboe /* not necessary, but safer to zero */ 2413ed29b0b4SJens Axboe req->cqe.res = 0; 2414ed29b0b4SJens Axboe } 2415ed29b0b4SJens Axboe 2416ed29b0b4SJens Axboe static void io_flush_cached_locked_reqs(struct io_ring_ctx *ctx, 2417ed29b0b4SJens Axboe struct io_submit_state *state) 2418ed29b0b4SJens Axboe { 2419ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 2420ed29b0b4SJens Axboe wq_list_splice(&ctx->locked_free_list, &state->free_list); 2421ed29b0b4SJens Axboe ctx->locked_free_nr = 0; 2422ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 2423ed29b0b4SJens Axboe } 2424ed29b0b4SJens Axboe 2425ed29b0b4SJens Axboe static inline bool io_req_cache_empty(struct io_ring_ctx *ctx) 2426ed29b0b4SJens Axboe { 2427ed29b0b4SJens Axboe return !ctx->submit_state.free_list.next; 2428ed29b0b4SJens Axboe } 2429ed29b0b4SJens Axboe 2430ed29b0b4SJens Axboe /* 2431ed29b0b4SJens Axboe * A request might get retired back into the request caches even before opcode 2432ed29b0b4SJens Axboe * handlers and io_issue_sqe() are done with it, e.g. inline completion path. 2433ed29b0b4SJens Axboe * Because of that, io_alloc_req() should be called only under ->uring_lock 2434ed29b0b4SJens Axboe * and with extra caution to not get a request that is still worked on. 2435ed29b0b4SJens Axboe */ 2436ed29b0b4SJens Axboe static __cold bool __io_alloc_req_refill(struct io_ring_ctx *ctx) 2437ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 2438ed29b0b4SJens Axboe { 2439ed29b0b4SJens Axboe gfp_t gfp = GFP_KERNEL | __GFP_NOWARN; 2440ed29b0b4SJens Axboe void *reqs[IO_REQ_ALLOC_BATCH]; 2441ed29b0b4SJens Axboe int ret, i; 2442ed29b0b4SJens Axboe 2443ed29b0b4SJens Axboe /* 2444ed29b0b4SJens Axboe * If we have more than a batch's worth of requests in our IRQ side 2445ed29b0b4SJens Axboe * locked cache, grab the lock and move them over to our submission 2446ed29b0b4SJens Axboe * side cache. 2447ed29b0b4SJens Axboe */ 2448ed29b0b4SJens Axboe if (data_race(ctx->locked_free_nr) > IO_COMPL_BATCH) { 2449ed29b0b4SJens Axboe io_flush_cached_locked_reqs(ctx, &ctx->submit_state); 2450ed29b0b4SJens Axboe if (!io_req_cache_empty(ctx)) 2451ed29b0b4SJens Axboe return true; 2452ed29b0b4SJens Axboe } 2453ed29b0b4SJens Axboe 2454ed29b0b4SJens Axboe ret = kmem_cache_alloc_bulk(req_cachep, gfp, ARRAY_SIZE(reqs), reqs); 2455ed29b0b4SJens Axboe 2456ed29b0b4SJens Axboe /* 2457ed29b0b4SJens Axboe * Bulk alloc is all-or-nothing. If we fail to get a batch, 2458ed29b0b4SJens Axboe * retry single alloc to be on the safe side. 2459ed29b0b4SJens Axboe */ 2460ed29b0b4SJens Axboe if (unlikely(ret <= 0)) { 2461ed29b0b4SJens Axboe reqs[0] = kmem_cache_alloc(req_cachep, gfp); 2462ed29b0b4SJens Axboe if (!reqs[0]) 2463ed29b0b4SJens Axboe return false; 2464ed29b0b4SJens Axboe ret = 1; 2465ed29b0b4SJens Axboe } 2466ed29b0b4SJens Axboe 2467ed29b0b4SJens Axboe percpu_ref_get_many(&ctx->refs, ret); 2468ed29b0b4SJens Axboe for (i = 0; i < ret; i++) { 2469ed29b0b4SJens Axboe struct io_kiocb *req = reqs[i]; 2470ed29b0b4SJens Axboe 2471ed29b0b4SJens Axboe io_preinit_req(req, ctx); 2472ed29b0b4SJens Axboe io_req_add_to_cache(req, ctx); 2473ed29b0b4SJens Axboe } 2474ed29b0b4SJens Axboe return true; 2475ed29b0b4SJens Axboe } 2476ed29b0b4SJens Axboe 2477ed29b0b4SJens Axboe static inline bool io_alloc_req_refill(struct io_ring_ctx *ctx) 2478ed29b0b4SJens Axboe { 2479ed29b0b4SJens Axboe if (unlikely(io_req_cache_empty(ctx))) 2480ed29b0b4SJens Axboe return __io_alloc_req_refill(ctx); 2481ed29b0b4SJens Axboe return true; 2482ed29b0b4SJens Axboe } 2483ed29b0b4SJens Axboe 2484ed29b0b4SJens Axboe static inline struct io_kiocb *io_alloc_req(struct io_ring_ctx *ctx) 2485ed29b0b4SJens Axboe { 2486ed29b0b4SJens Axboe struct io_wq_work_node *node; 2487ed29b0b4SJens Axboe 2488ed29b0b4SJens Axboe node = wq_stack_extract(&ctx->submit_state.free_list); 2489ed29b0b4SJens Axboe return container_of(node, struct io_kiocb, comp_list); 2490ed29b0b4SJens Axboe } 2491ed29b0b4SJens Axboe 2492ed29b0b4SJens Axboe static inline void io_put_file(struct file *file) 2493ed29b0b4SJens Axboe { 2494ed29b0b4SJens Axboe if (file) 2495ed29b0b4SJens Axboe fput(file); 2496ed29b0b4SJens Axboe } 2497ed29b0b4SJens Axboe 2498ed29b0b4SJens Axboe static inline void io_dismantle_req(struct io_kiocb *req) 2499ed29b0b4SJens Axboe { 2500ed29b0b4SJens Axboe unsigned int flags = req->flags; 2501ed29b0b4SJens Axboe 2502ed29b0b4SJens Axboe if (unlikely(flags & IO_REQ_CLEAN_FLAGS)) 2503ed29b0b4SJens Axboe io_clean_op(req); 2504ed29b0b4SJens Axboe if (!(flags & REQ_F_FIXED_FILE)) 2505ed29b0b4SJens Axboe io_put_file(req->file); 2506ed29b0b4SJens Axboe } 2507ed29b0b4SJens Axboe 2508ed29b0b4SJens Axboe static __cold void io_free_req(struct io_kiocb *req) 2509ed29b0b4SJens Axboe { 2510ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 2511ed29b0b4SJens Axboe 2512ed29b0b4SJens Axboe io_req_put_rsrc(req); 2513ed29b0b4SJens Axboe io_dismantle_req(req); 2514ed29b0b4SJens Axboe io_put_task(req->task, 1); 2515ed29b0b4SJens Axboe 2516ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 2517ed29b0b4SJens Axboe wq_list_add_head(&req->comp_list, &ctx->locked_free_list); 2518ed29b0b4SJens Axboe ctx->locked_free_nr++; 2519ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 2520ed29b0b4SJens Axboe } 2521ed29b0b4SJens Axboe 2522ed29b0b4SJens Axboe static inline void io_remove_next_linked(struct io_kiocb *req) 2523ed29b0b4SJens Axboe { 2524ed29b0b4SJens Axboe struct io_kiocb *nxt = req->link; 2525ed29b0b4SJens Axboe 2526ed29b0b4SJens Axboe req->link = nxt->link; 2527ed29b0b4SJens Axboe nxt->link = NULL; 2528ed29b0b4SJens Axboe } 2529ed29b0b4SJens Axboe 2530ed29b0b4SJens Axboe static struct io_kiocb *io_disarm_linked_timeout(struct io_kiocb *req) 2531ed29b0b4SJens Axboe __must_hold(&req->ctx->completion_lock) 2532ed29b0b4SJens Axboe __must_hold(&req->ctx->timeout_lock) 2533ed29b0b4SJens Axboe { 2534ed29b0b4SJens Axboe struct io_kiocb *link = req->link; 2535ed29b0b4SJens Axboe 2536ed29b0b4SJens Axboe if (link && link->opcode == IORING_OP_LINK_TIMEOUT) { 2537ed29b0b4SJens Axboe struct io_timeout_data *io = link->async_data; 2538a43714acSJens Axboe struct io_timeout *timeout = io_kiocb_to_cmd(link); 2539ed29b0b4SJens Axboe 2540ed29b0b4SJens Axboe io_remove_next_linked(req); 2541a43714acSJens Axboe timeout->head = NULL; 2542ed29b0b4SJens Axboe if (hrtimer_try_to_cancel(&io->timer) != -1) { 2543a43714acSJens Axboe list_del(&timeout->list); 2544ed29b0b4SJens Axboe return link; 2545ed29b0b4SJens Axboe } 2546ed29b0b4SJens Axboe } 2547ed29b0b4SJens Axboe return NULL; 2548ed29b0b4SJens Axboe } 2549ed29b0b4SJens Axboe 2550ed29b0b4SJens Axboe static void io_fail_links(struct io_kiocb *req) 2551ed29b0b4SJens Axboe __must_hold(&req->ctx->completion_lock) 2552ed29b0b4SJens Axboe { 2553ed29b0b4SJens Axboe struct io_kiocb *nxt, *link = req->link; 2554ed29b0b4SJens Axboe bool ignore_cqes = req->flags & REQ_F_SKIP_LINK_CQES; 2555ed29b0b4SJens Axboe 2556ed29b0b4SJens Axboe req->link = NULL; 2557ed29b0b4SJens Axboe while (link) { 2558ed29b0b4SJens Axboe long res = -ECANCELED; 2559ed29b0b4SJens Axboe 2560ed29b0b4SJens Axboe if (link->flags & REQ_F_FAIL) 2561ed29b0b4SJens Axboe res = link->cqe.res; 2562ed29b0b4SJens Axboe 2563ed29b0b4SJens Axboe nxt = link->link; 2564ed29b0b4SJens Axboe link->link = NULL; 2565ed29b0b4SJens Axboe 2566ed29b0b4SJens Axboe trace_io_uring_fail_link(req->ctx, req, req->cqe.user_data, 2567ed29b0b4SJens Axboe req->opcode, link); 2568ed29b0b4SJens Axboe 2569ed29b0b4SJens Axboe if (ignore_cqes) 2570ed29b0b4SJens Axboe link->flags |= REQ_F_CQE_SKIP; 2571ed29b0b4SJens Axboe else 2572ed29b0b4SJens Axboe link->flags &= ~REQ_F_CQE_SKIP; 2573ed29b0b4SJens Axboe __io_req_complete_post(link, res, 0); 2574ed29b0b4SJens Axboe link = nxt; 2575ed29b0b4SJens Axboe } 2576ed29b0b4SJens Axboe } 2577ed29b0b4SJens Axboe 2578ed29b0b4SJens Axboe static bool io_disarm_next(struct io_kiocb *req) 2579ed29b0b4SJens Axboe __must_hold(&req->ctx->completion_lock) 2580ed29b0b4SJens Axboe { 2581ed29b0b4SJens Axboe struct io_kiocb *link = NULL; 2582ed29b0b4SJens Axboe bool posted = false; 2583ed29b0b4SJens Axboe 2584ed29b0b4SJens Axboe if (req->flags & REQ_F_ARM_LTIMEOUT) { 2585ed29b0b4SJens Axboe link = req->link; 2586ed29b0b4SJens Axboe req->flags &= ~REQ_F_ARM_LTIMEOUT; 2587ed29b0b4SJens Axboe if (link && link->opcode == IORING_OP_LINK_TIMEOUT) { 2588ed29b0b4SJens Axboe io_remove_next_linked(req); 2589ed29b0b4SJens Axboe io_req_tw_post_queue(link, -ECANCELED, 0); 2590ed29b0b4SJens Axboe posted = true; 2591ed29b0b4SJens Axboe } 2592ed29b0b4SJens Axboe } else if (req->flags & REQ_F_LINK_TIMEOUT) { 2593ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 2594ed29b0b4SJens Axboe 2595ed29b0b4SJens Axboe spin_lock_irq(&ctx->timeout_lock); 2596ed29b0b4SJens Axboe link = io_disarm_linked_timeout(req); 2597ed29b0b4SJens Axboe spin_unlock_irq(&ctx->timeout_lock); 2598ed29b0b4SJens Axboe if (link) { 2599ed29b0b4SJens Axboe posted = true; 2600ed29b0b4SJens Axboe io_req_tw_post_queue(link, -ECANCELED, 0); 2601ed29b0b4SJens Axboe } 2602ed29b0b4SJens Axboe } 2603ed29b0b4SJens Axboe if (unlikely((req->flags & REQ_F_FAIL) && 2604ed29b0b4SJens Axboe !(req->flags & REQ_F_HARDLINK))) { 2605ed29b0b4SJens Axboe posted |= (req->link != NULL); 2606ed29b0b4SJens Axboe io_fail_links(req); 2607ed29b0b4SJens Axboe } 2608ed29b0b4SJens Axboe return posted; 2609ed29b0b4SJens Axboe } 2610ed29b0b4SJens Axboe 2611ed29b0b4SJens Axboe static void __io_req_find_next_prep(struct io_kiocb *req) 2612ed29b0b4SJens Axboe { 2613ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 2614ed29b0b4SJens Axboe bool posted; 2615ed29b0b4SJens Axboe 2616ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 2617ed29b0b4SJens Axboe posted = io_disarm_next(req); 2618ed29b0b4SJens Axboe io_commit_cqring(ctx); 2619ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 2620ed29b0b4SJens Axboe if (posted) 2621ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 2622ed29b0b4SJens Axboe } 2623ed29b0b4SJens Axboe 2624ed29b0b4SJens Axboe static inline struct io_kiocb *io_req_find_next(struct io_kiocb *req) 2625ed29b0b4SJens Axboe { 2626ed29b0b4SJens Axboe struct io_kiocb *nxt; 2627ed29b0b4SJens Axboe 2628ed29b0b4SJens Axboe /* 2629ed29b0b4SJens Axboe * If LINK is set, we have dependent requests in this chain. If we 2630ed29b0b4SJens Axboe * didn't fail this request, queue the first one up, moving any other 2631ed29b0b4SJens Axboe * dependencies to the next request. In case of failure, fail the rest 2632ed29b0b4SJens Axboe * of the chain. 2633ed29b0b4SJens Axboe */ 2634ed29b0b4SJens Axboe if (unlikely(req->flags & IO_DISARM_MASK)) 2635ed29b0b4SJens Axboe __io_req_find_next_prep(req); 2636ed29b0b4SJens Axboe nxt = req->link; 2637ed29b0b4SJens Axboe req->link = NULL; 2638ed29b0b4SJens Axboe return nxt; 2639ed29b0b4SJens Axboe } 2640ed29b0b4SJens Axboe 2641ed29b0b4SJens Axboe static void ctx_flush_and_put(struct io_ring_ctx *ctx, bool *locked) 2642ed29b0b4SJens Axboe { 2643ed29b0b4SJens Axboe if (!ctx) 2644ed29b0b4SJens Axboe return; 2645ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_TASKRUN_FLAG) 2646ed29b0b4SJens Axboe atomic_andnot(IORING_SQ_TASKRUN, &ctx->rings->sq_flags); 2647ed29b0b4SJens Axboe if (*locked) { 2648ed29b0b4SJens Axboe io_submit_flush_completions(ctx); 2649ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 2650ed29b0b4SJens Axboe *locked = false; 2651ed29b0b4SJens Axboe } 2652ed29b0b4SJens Axboe percpu_ref_put(&ctx->refs); 2653ed29b0b4SJens Axboe } 2654ed29b0b4SJens Axboe 2655ed29b0b4SJens Axboe static inline void ctx_commit_and_unlock(struct io_ring_ctx *ctx) 2656ed29b0b4SJens Axboe { 2657ed29b0b4SJens Axboe io_commit_cqring(ctx); 2658ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 2659ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 2660ed29b0b4SJens Axboe } 2661ed29b0b4SJens Axboe 2662ed29b0b4SJens Axboe static void handle_prev_tw_list(struct io_wq_work_node *node, 2663ed29b0b4SJens Axboe struct io_ring_ctx **ctx, bool *uring_locked) 2664ed29b0b4SJens Axboe { 2665ed29b0b4SJens Axboe if (*ctx && !*uring_locked) 2666ed29b0b4SJens Axboe spin_lock(&(*ctx)->completion_lock); 2667ed29b0b4SJens Axboe 2668ed29b0b4SJens Axboe do { 2669ed29b0b4SJens Axboe struct io_wq_work_node *next = node->next; 2670ed29b0b4SJens Axboe struct io_kiocb *req = container_of(node, struct io_kiocb, 2671ed29b0b4SJens Axboe io_task_work.node); 2672ed29b0b4SJens Axboe 2673ed29b0b4SJens Axboe prefetch(container_of(next, struct io_kiocb, io_task_work.node)); 2674ed29b0b4SJens Axboe 2675ed29b0b4SJens Axboe if (req->ctx != *ctx) { 2676ed29b0b4SJens Axboe if (unlikely(!*uring_locked && *ctx)) 2677ed29b0b4SJens Axboe ctx_commit_and_unlock(*ctx); 2678ed29b0b4SJens Axboe 2679ed29b0b4SJens Axboe ctx_flush_and_put(*ctx, uring_locked); 2680ed29b0b4SJens Axboe *ctx = req->ctx; 2681ed29b0b4SJens Axboe /* if not contended, grab and improve batching */ 2682ed29b0b4SJens Axboe *uring_locked = mutex_trylock(&(*ctx)->uring_lock); 2683ed29b0b4SJens Axboe percpu_ref_get(&(*ctx)->refs); 2684ed29b0b4SJens Axboe if (unlikely(!*uring_locked)) 2685ed29b0b4SJens Axboe spin_lock(&(*ctx)->completion_lock); 2686ed29b0b4SJens Axboe } 2687ed29b0b4SJens Axboe if (likely(*uring_locked)) 2688ed29b0b4SJens Axboe req->io_task_work.func(req, uring_locked); 2689ed29b0b4SJens Axboe else 2690ed29b0b4SJens Axboe __io_req_complete_post(req, req->cqe.res, 2691ed29b0b4SJens Axboe io_put_kbuf_comp(req)); 2692ed29b0b4SJens Axboe node = next; 2693ed29b0b4SJens Axboe } while (node); 2694ed29b0b4SJens Axboe 2695ed29b0b4SJens Axboe if (unlikely(!*uring_locked)) 2696ed29b0b4SJens Axboe ctx_commit_and_unlock(*ctx); 2697ed29b0b4SJens Axboe } 2698ed29b0b4SJens Axboe 2699ed29b0b4SJens Axboe static void handle_tw_list(struct io_wq_work_node *node, 2700ed29b0b4SJens Axboe struct io_ring_ctx **ctx, bool *locked) 2701ed29b0b4SJens Axboe { 2702ed29b0b4SJens Axboe do { 2703ed29b0b4SJens Axboe struct io_wq_work_node *next = node->next; 2704ed29b0b4SJens Axboe struct io_kiocb *req = container_of(node, struct io_kiocb, 2705ed29b0b4SJens Axboe io_task_work.node); 2706ed29b0b4SJens Axboe 2707ed29b0b4SJens Axboe prefetch(container_of(next, struct io_kiocb, io_task_work.node)); 2708ed29b0b4SJens Axboe 2709ed29b0b4SJens Axboe if (req->ctx != *ctx) { 2710ed29b0b4SJens Axboe ctx_flush_and_put(*ctx, locked); 2711ed29b0b4SJens Axboe *ctx = req->ctx; 2712ed29b0b4SJens Axboe /* if not contended, grab and improve batching */ 2713ed29b0b4SJens Axboe *locked = mutex_trylock(&(*ctx)->uring_lock); 2714ed29b0b4SJens Axboe percpu_ref_get(&(*ctx)->refs); 2715ed29b0b4SJens Axboe } 2716ed29b0b4SJens Axboe req->io_task_work.func(req, locked); 2717ed29b0b4SJens Axboe node = next; 2718ed29b0b4SJens Axboe } while (node); 2719ed29b0b4SJens Axboe } 2720ed29b0b4SJens Axboe 2721ed29b0b4SJens Axboe static void tctx_task_work(struct callback_head *cb) 2722ed29b0b4SJens Axboe { 2723ed29b0b4SJens Axboe bool uring_locked = false; 2724ed29b0b4SJens Axboe struct io_ring_ctx *ctx = NULL; 2725ed29b0b4SJens Axboe struct io_uring_task *tctx = container_of(cb, struct io_uring_task, 2726ed29b0b4SJens Axboe task_work); 2727ed29b0b4SJens Axboe 2728ed29b0b4SJens Axboe while (1) { 2729ed29b0b4SJens Axboe struct io_wq_work_node *node1, *node2; 2730ed29b0b4SJens Axboe 2731ed29b0b4SJens Axboe spin_lock_irq(&tctx->task_lock); 2732ed29b0b4SJens Axboe node1 = tctx->prio_task_list.first; 2733ed29b0b4SJens Axboe node2 = tctx->task_list.first; 2734ed29b0b4SJens Axboe INIT_WQ_LIST(&tctx->task_list); 2735ed29b0b4SJens Axboe INIT_WQ_LIST(&tctx->prio_task_list); 2736ed29b0b4SJens Axboe if (!node2 && !node1) 2737ed29b0b4SJens Axboe tctx->task_running = false; 2738ed29b0b4SJens Axboe spin_unlock_irq(&tctx->task_lock); 2739ed29b0b4SJens Axboe if (!node2 && !node1) 2740ed29b0b4SJens Axboe break; 2741ed29b0b4SJens Axboe 2742ed29b0b4SJens Axboe if (node1) 2743ed29b0b4SJens Axboe handle_prev_tw_list(node1, &ctx, &uring_locked); 2744ed29b0b4SJens Axboe if (node2) 2745ed29b0b4SJens Axboe handle_tw_list(node2, &ctx, &uring_locked); 2746ed29b0b4SJens Axboe cond_resched(); 2747ed29b0b4SJens Axboe 2748ed29b0b4SJens Axboe if (data_race(!tctx->task_list.first) && 2749ed29b0b4SJens Axboe data_race(!tctx->prio_task_list.first) && uring_locked) 2750ed29b0b4SJens Axboe io_submit_flush_completions(ctx); 2751ed29b0b4SJens Axboe } 2752ed29b0b4SJens Axboe 2753ed29b0b4SJens Axboe ctx_flush_and_put(ctx, &uring_locked); 2754ed29b0b4SJens Axboe 2755ed29b0b4SJens Axboe /* relaxed read is enough as only the task itself sets ->in_idle */ 2756ed29b0b4SJens Axboe if (unlikely(atomic_read(&tctx->in_idle))) 2757ed29b0b4SJens Axboe io_uring_drop_tctx_refs(current); 2758ed29b0b4SJens Axboe } 2759ed29b0b4SJens Axboe 2760ed29b0b4SJens Axboe static void __io_req_task_work_add(struct io_kiocb *req, 2761ed29b0b4SJens Axboe struct io_uring_task *tctx, 2762ed29b0b4SJens Axboe struct io_wq_work_list *list) 2763ed29b0b4SJens Axboe { 2764ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 2765ed29b0b4SJens Axboe struct io_wq_work_node *node; 2766ed29b0b4SJens Axboe unsigned long flags; 2767ed29b0b4SJens Axboe bool running; 2768ed29b0b4SJens Axboe 2769ed29b0b4SJens Axboe spin_lock_irqsave(&tctx->task_lock, flags); 2770ed29b0b4SJens Axboe wq_list_add_tail(&req->io_task_work.node, list); 2771ed29b0b4SJens Axboe running = tctx->task_running; 2772ed29b0b4SJens Axboe if (!running) 2773ed29b0b4SJens Axboe tctx->task_running = true; 2774ed29b0b4SJens Axboe spin_unlock_irqrestore(&tctx->task_lock, flags); 2775ed29b0b4SJens Axboe 2776ed29b0b4SJens Axboe /* task_work already pending, we're done */ 2777ed29b0b4SJens Axboe if (running) 2778ed29b0b4SJens Axboe return; 2779ed29b0b4SJens Axboe 2780ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_TASKRUN_FLAG) 2781ed29b0b4SJens Axboe atomic_or(IORING_SQ_TASKRUN, &ctx->rings->sq_flags); 2782ed29b0b4SJens Axboe 2783ed29b0b4SJens Axboe if (likely(!task_work_add(req->task, &tctx->task_work, ctx->notify_method))) 2784ed29b0b4SJens Axboe return; 2785ed29b0b4SJens Axboe 2786ed29b0b4SJens Axboe spin_lock_irqsave(&tctx->task_lock, flags); 2787ed29b0b4SJens Axboe tctx->task_running = false; 2788ed29b0b4SJens Axboe node = wq_list_merge(&tctx->prio_task_list, &tctx->task_list); 2789ed29b0b4SJens Axboe spin_unlock_irqrestore(&tctx->task_lock, flags); 2790ed29b0b4SJens Axboe 2791ed29b0b4SJens Axboe while (node) { 2792ed29b0b4SJens Axboe req = container_of(node, struct io_kiocb, io_task_work.node); 2793ed29b0b4SJens Axboe node = node->next; 2794ed29b0b4SJens Axboe if (llist_add(&req->io_task_work.fallback_node, 2795ed29b0b4SJens Axboe &req->ctx->fallback_llist)) 2796ed29b0b4SJens Axboe schedule_delayed_work(&req->ctx->fallback_work, 1); 2797ed29b0b4SJens Axboe } 2798ed29b0b4SJens Axboe } 2799ed29b0b4SJens Axboe 2800ed29b0b4SJens Axboe static void io_req_task_work_add(struct io_kiocb *req) 2801ed29b0b4SJens Axboe { 2802ed29b0b4SJens Axboe struct io_uring_task *tctx = req->task->io_uring; 2803ed29b0b4SJens Axboe 2804ed29b0b4SJens Axboe __io_req_task_work_add(req, tctx, &tctx->task_list); 2805ed29b0b4SJens Axboe } 2806ed29b0b4SJens Axboe 2807ed29b0b4SJens Axboe static void io_req_task_prio_work_add(struct io_kiocb *req) 2808ed29b0b4SJens Axboe { 2809ed29b0b4SJens Axboe struct io_uring_task *tctx = req->task->io_uring; 2810ed29b0b4SJens Axboe 2811ed29b0b4SJens Axboe if (req->ctx->flags & IORING_SETUP_SQPOLL) 2812ed29b0b4SJens Axboe __io_req_task_work_add(req, tctx, &tctx->prio_task_list); 2813ed29b0b4SJens Axboe else 2814ed29b0b4SJens Axboe __io_req_task_work_add(req, tctx, &tctx->task_list); 2815ed29b0b4SJens Axboe } 2816ed29b0b4SJens Axboe 2817ed29b0b4SJens Axboe static void io_req_tw_post(struct io_kiocb *req, bool *locked) 2818ed29b0b4SJens Axboe { 2819ed29b0b4SJens Axboe io_req_complete_post(req, req->cqe.res, req->cqe.flags); 2820ed29b0b4SJens Axboe } 2821ed29b0b4SJens Axboe 2822ed29b0b4SJens Axboe static void io_req_tw_post_queue(struct io_kiocb *req, s32 res, u32 cflags) 2823ed29b0b4SJens Axboe { 2824ed29b0b4SJens Axboe req->cqe.res = res; 2825ed29b0b4SJens Axboe req->cqe.flags = cflags; 2826ed29b0b4SJens Axboe req->io_task_work.func = io_req_tw_post; 2827ed29b0b4SJens Axboe io_req_task_work_add(req); 2828ed29b0b4SJens Axboe } 2829ed29b0b4SJens Axboe 2830ed29b0b4SJens Axboe static void io_req_task_cancel(struct io_kiocb *req, bool *locked) 2831ed29b0b4SJens Axboe { 2832ed29b0b4SJens Axboe /* not needed for normal modes, but SQPOLL depends on it */ 2833ed29b0b4SJens Axboe io_tw_lock(req->ctx, locked); 2834ed29b0b4SJens Axboe io_req_complete_failed(req, req->cqe.res); 2835ed29b0b4SJens Axboe } 2836ed29b0b4SJens Axboe 2837ed29b0b4SJens Axboe static void io_req_task_submit(struct io_kiocb *req, bool *locked) 2838ed29b0b4SJens Axboe { 2839ed29b0b4SJens Axboe io_tw_lock(req->ctx, locked); 2840ed29b0b4SJens Axboe /* req->task == current here, checking PF_EXITING is safe */ 2841ed29b0b4SJens Axboe if (likely(!(req->task->flags & PF_EXITING))) 2842ed29b0b4SJens Axboe io_queue_sqe(req); 2843ed29b0b4SJens Axboe else 2844ed29b0b4SJens Axboe io_req_complete_failed(req, -EFAULT); 2845ed29b0b4SJens Axboe } 2846ed29b0b4SJens Axboe 2847ed29b0b4SJens Axboe static void io_req_task_queue_fail(struct io_kiocb *req, int ret) 2848ed29b0b4SJens Axboe { 2849ed29b0b4SJens Axboe req->cqe.res = ret; 2850ed29b0b4SJens Axboe req->io_task_work.func = io_req_task_cancel; 2851ed29b0b4SJens Axboe io_req_task_work_add(req); 2852ed29b0b4SJens Axboe } 2853ed29b0b4SJens Axboe 2854ed29b0b4SJens Axboe static void io_req_task_queue(struct io_kiocb *req) 2855ed29b0b4SJens Axboe { 2856ed29b0b4SJens Axboe req->io_task_work.func = io_req_task_submit; 2857ed29b0b4SJens Axboe io_req_task_work_add(req); 2858ed29b0b4SJens Axboe } 2859ed29b0b4SJens Axboe 2860ed29b0b4SJens Axboe static void io_req_task_queue_reissue(struct io_kiocb *req) 2861ed29b0b4SJens Axboe { 2862ed29b0b4SJens Axboe req->io_task_work.func = io_queue_iowq; 2863ed29b0b4SJens Axboe io_req_task_work_add(req); 2864ed29b0b4SJens Axboe } 2865ed29b0b4SJens Axboe 2866ed29b0b4SJens Axboe static void io_queue_next(struct io_kiocb *req) 2867ed29b0b4SJens Axboe { 2868ed29b0b4SJens Axboe struct io_kiocb *nxt = io_req_find_next(req); 2869ed29b0b4SJens Axboe 2870ed29b0b4SJens Axboe if (nxt) 2871ed29b0b4SJens Axboe io_req_task_queue(nxt); 2872ed29b0b4SJens Axboe } 2873ed29b0b4SJens Axboe 2874ed29b0b4SJens Axboe static void io_free_batch_list(struct io_ring_ctx *ctx, 2875ed29b0b4SJens Axboe struct io_wq_work_node *node) 2876ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 2877ed29b0b4SJens Axboe { 2878ed29b0b4SJens Axboe struct task_struct *task = NULL; 2879ed29b0b4SJens Axboe int task_refs = 0; 2880ed29b0b4SJens Axboe 2881ed29b0b4SJens Axboe do { 2882ed29b0b4SJens Axboe struct io_kiocb *req = container_of(node, struct io_kiocb, 2883ed29b0b4SJens Axboe comp_list); 2884ed29b0b4SJens Axboe 2885ed29b0b4SJens Axboe if (unlikely(req->flags & IO_REQ_CLEAN_SLOW_FLAGS)) { 2886ed29b0b4SJens Axboe if (req->flags & REQ_F_REFCOUNT) { 2887ed29b0b4SJens Axboe node = req->comp_list.next; 2888ed29b0b4SJens Axboe if (!req_ref_put_and_test(req)) 2889ed29b0b4SJens Axboe continue; 2890ed29b0b4SJens Axboe } 2891ed29b0b4SJens Axboe if ((req->flags & REQ_F_POLLED) && req->apoll) { 2892ed29b0b4SJens Axboe struct async_poll *apoll = req->apoll; 2893ed29b0b4SJens Axboe 2894ed29b0b4SJens Axboe if (apoll->double_poll) 2895ed29b0b4SJens Axboe kfree(apoll->double_poll); 2896ed29b0b4SJens Axboe list_add(&apoll->poll.wait.entry, 2897ed29b0b4SJens Axboe &ctx->apoll_cache); 2898ed29b0b4SJens Axboe req->flags &= ~REQ_F_POLLED; 2899ed29b0b4SJens Axboe } 2900ed29b0b4SJens Axboe if (req->flags & IO_REQ_LINK_FLAGS) 2901ed29b0b4SJens Axboe io_queue_next(req); 2902ed29b0b4SJens Axboe if (unlikely(req->flags & IO_REQ_CLEAN_FLAGS)) 2903ed29b0b4SJens Axboe io_clean_op(req); 2904ed29b0b4SJens Axboe } 2905ed29b0b4SJens Axboe if (!(req->flags & REQ_F_FIXED_FILE)) 2906ed29b0b4SJens Axboe io_put_file(req->file); 2907ed29b0b4SJens Axboe 2908ed29b0b4SJens Axboe io_req_put_rsrc_locked(req, ctx); 2909ed29b0b4SJens Axboe 2910ed29b0b4SJens Axboe if (req->task != task) { 2911ed29b0b4SJens Axboe if (task) 2912ed29b0b4SJens Axboe io_put_task(task, task_refs); 2913ed29b0b4SJens Axboe task = req->task; 2914ed29b0b4SJens Axboe task_refs = 0; 2915ed29b0b4SJens Axboe } 2916ed29b0b4SJens Axboe task_refs++; 2917ed29b0b4SJens Axboe node = req->comp_list.next; 2918ed29b0b4SJens Axboe io_req_add_to_cache(req, ctx); 2919ed29b0b4SJens Axboe } while (node); 2920ed29b0b4SJens Axboe 2921ed29b0b4SJens Axboe if (task) 2922ed29b0b4SJens Axboe io_put_task(task, task_refs); 2923ed29b0b4SJens Axboe } 2924ed29b0b4SJens Axboe 2925ed29b0b4SJens Axboe static void __io_submit_flush_completions(struct io_ring_ctx *ctx) 2926ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 2927ed29b0b4SJens Axboe { 2928ed29b0b4SJens Axboe struct io_wq_work_node *node, *prev; 2929ed29b0b4SJens Axboe struct io_submit_state *state = &ctx->submit_state; 2930ed29b0b4SJens Axboe 2931ed29b0b4SJens Axboe if (state->flush_cqes) { 2932ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 2933ed29b0b4SJens Axboe wq_list_for_each(node, prev, &state->compl_reqs) { 2934ed29b0b4SJens Axboe struct io_kiocb *req = container_of(node, struct io_kiocb, 2935ed29b0b4SJens Axboe comp_list); 2936ed29b0b4SJens Axboe 2937ed29b0b4SJens Axboe if (!(req->flags & REQ_F_CQE_SKIP)) 2938ed29b0b4SJens Axboe __io_fill_cqe_req(ctx, req); 2939ed29b0b4SJens Axboe } 2940ed29b0b4SJens Axboe 2941ed29b0b4SJens Axboe io_commit_cqring(ctx); 2942ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 2943ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 2944ed29b0b4SJens Axboe state->flush_cqes = false; 2945ed29b0b4SJens Axboe } 2946ed29b0b4SJens Axboe 2947ed29b0b4SJens Axboe io_free_batch_list(ctx, state->compl_reqs.first); 2948ed29b0b4SJens Axboe INIT_WQ_LIST(&state->compl_reqs); 2949ed29b0b4SJens Axboe } 2950ed29b0b4SJens Axboe 2951ed29b0b4SJens Axboe /* 2952ed29b0b4SJens Axboe * Drop reference to request, return next in chain (if there is one) if this 2953ed29b0b4SJens Axboe * was the last reference to this request. 2954ed29b0b4SJens Axboe */ 2955ed29b0b4SJens Axboe static inline struct io_kiocb *io_put_req_find_next(struct io_kiocb *req) 2956ed29b0b4SJens Axboe { 2957ed29b0b4SJens Axboe struct io_kiocb *nxt = NULL; 2958ed29b0b4SJens Axboe 2959ed29b0b4SJens Axboe if (req_ref_put_and_test(req)) { 2960ed29b0b4SJens Axboe if (unlikely(req->flags & IO_REQ_LINK_FLAGS)) 2961ed29b0b4SJens Axboe nxt = io_req_find_next(req); 2962ed29b0b4SJens Axboe io_free_req(req); 2963ed29b0b4SJens Axboe } 2964ed29b0b4SJens Axboe return nxt; 2965ed29b0b4SJens Axboe } 2966ed29b0b4SJens Axboe 2967ed29b0b4SJens Axboe static inline void io_put_req(struct io_kiocb *req) 2968ed29b0b4SJens Axboe { 2969ed29b0b4SJens Axboe if (req_ref_put_and_test(req)) { 2970ed29b0b4SJens Axboe io_queue_next(req); 2971ed29b0b4SJens Axboe io_free_req(req); 2972ed29b0b4SJens Axboe } 2973ed29b0b4SJens Axboe } 2974ed29b0b4SJens Axboe 2975ed29b0b4SJens Axboe static unsigned io_cqring_events(struct io_ring_ctx *ctx) 2976ed29b0b4SJens Axboe { 2977ed29b0b4SJens Axboe /* See comment at the top of this file */ 2978ed29b0b4SJens Axboe smp_rmb(); 2979ed29b0b4SJens Axboe return __io_cqring_events(ctx); 2980ed29b0b4SJens Axboe } 2981ed29b0b4SJens Axboe 2982ed29b0b4SJens Axboe static inline unsigned int io_sqring_entries(struct io_ring_ctx *ctx) 2983ed29b0b4SJens Axboe { 2984ed29b0b4SJens Axboe struct io_rings *rings = ctx->rings; 2985ed29b0b4SJens Axboe 2986ed29b0b4SJens Axboe /* make sure SQ entry isn't read before tail */ 2987ed29b0b4SJens Axboe return smp_load_acquire(&rings->sq.tail) - ctx->cached_sq_head; 2988ed29b0b4SJens Axboe } 2989ed29b0b4SJens Axboe 2990ed29b0b4SJens Axboe static inline bool io_run_task_work(void) 2991ed29b0b4SJens Axboe { 2992ed29b0b4SJens Axboe if (test_thread_flag(TIF_NOTIFY_SIGNAL) || task_work_pending(current)) { 2993ed29b0b4SJens Axboe __set_current_state(TASK_RUNNING); 2994ed29b0b4SJens Axboe clear_notify_signal(); 2995ed29b0b4SJens Axboe if (task_work_pending(current)) 2996ed29b0b4SJens Axboe task_work_run(); 2997ed29b0b4SJens Axboe return true; 2998ed29b0b4SJens Axboe } 2999ed29b0b4SJens Axboe 3000ed29b0b4SJens Axboe return false; 3001ed29b0b4SJens Axboe } 3002ed29b0b4SJens Axboe 3003ed29b0b4SJens Axboe static int io_do_iopoll(struct io_ring_ctx *ctx, bool force_nonspin) 3004ed29b0b4SJens Axboe { 3005ed29b0b4SJens Axboe struct io_wq_work_node *pos, *start, *prev; 3006ed29b0b4SJens Axboe unsigned int poll_flags = BLK_POLL_NOSLEEP; 3007ed29b0b4SJens Axboe DEFINE_IO_COMP_BATCH(iob); 3008ed29b0b4SJens Axboe int nr_events = 0; 3009ed29b0b4SJens Axboe 3010ed29b0b4SJens Axboe /* 3011ed29b0b4SJens Axboe * Only spin for completions if we don't have multiple devices hanging 3012ed29b0b4SJens Axboe * off our complete list. 3013ed29b0b4SJens Axboe */ 3014ed29b0b4SJens Axboe if (ctx->poll_multi_queue || force_nonspin) 3015ed29b0b4SJens Axboe poll_flags |= BLK_POLL_ONESHOT; 3016ed29b0b4SJens Axboe 3017ed29b0b4SJens Axboe wq_list_for_each(pos, start, &ctx->iopoll_list) { 3018ed29b0b4SJens Axboe struct io_kiocb *req = container_of(pos, struct io_kiocb, comp_list); 30193c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 3020ed29b0b4SJens Axboe int ret; 3021ed29b0b4SJens Axboe 3022ed29b0b4SJens Axboe /* 3023ed29b0b4SJens Axboe * Move completed and retryable entries to our local lists. 3024ed29b0b4SJens Axboe * If we find a request that requires polling, break out 3025ed29b0b4SJens Axboe * and complete those lists first, if we have entries there. 3026ed29b0b4SJens Axboe */ 3027ed29b0b4SJens Axboe if (READ_ONCE(req->iopoll_completed)) 3028ed29b0b4SJens Axboe break; 3029ed29b0b4SJens Axboe 30303c306fb2SJens Axboe ret = rw->kiocb.ki_filp->f_op->iopoll(&rw->kiocb, &iob, poll_flags); 3031ed29b0b4SJens Axboe if (unlikely(ret < 0)) 3032ed29b0b4SJens Axboe return ret; 3033ed29b0b4SJens Axboe else if (ret) 3034ed29b0b4SJens Axboe poll_flags |= BLK_POLL_ONESHOT; 3035ed29b0b4SJens Axboe 3036ed29b0b4SJens Axboe /* iopoll may have completed current req */ 3037ed29b0b4SJens Axboe if (!rq_list_empty(iob.req_list) || 3038ed29b0b4SJens Axboe READ_ONCE(req->iopoll_completed)) 3039ed29b0b4SJens Axboe break; 3040ed29b0b4SJens Axboe } 3041ed29b0b4SJens Axboe 3042ed29b0b4SJens Axboe if (!rq_list_empty(iob.req_list)) 3043ed29b0b4SJens Axboe iob.complete(&iob); 3044ed29b0b4SJens Axboe else if (!pos) 3045ed29b0b4SJens Axboe return 0; 3046ed29b0b4SJens Axboe 3047ed29b0b4SJens Axboe prev = start; 3048ed29b0b4SJens Axboe wq_list_for_each_resume(pos, prev) { 3049ed29b0b4SJens Axboe struct io_kiocb *req = container_of(pos, struct io_kiocb, comp_list); 3050ed29b0b4SJens Axboe 3051ed29b0b4SJens Axboe /* order with io_complete_rw_iopoll(), e.g. ->result updates */ 3052ed29b0b4SJens Axboe if (!smp_load_acquire(&req->iopoll_completed)) 3053ed29b0b4SJens Axboe break; 3054ed29b0b4SJens Axboe nr_events++; 3055ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_CQE_SKIP)) 3056ed29b0b4SJens Axboe continue; 3057ed29b0b4SJens Axboe 3058ed29b0b4SJens Axboe req->cqe.flags = io_put_kbuf(req, 0); 3059ed29b0b4SJens Axboe __io_fill_cqe_req(req->ctx, req); 3060ed29b0b4SJens Axboe } 3061ed29b0b4SJens Axboe 3062ed29b0b4SJens Axboe if (unlikely(!nr_events)) 3063ed29b0b4SJens Axboe return 0; 3064ed29b0b4SJens Axboe 3065ed29b0b4SJens Axboe io_commit_cqring(ctx); 3066ed29b0b4SJens Axboe io_cqring_ev_posted_iopoll(ctx); 3067ed29b0b4SJens Axboe pos = start ? start->next : ctx->iopoll_list.first; 3068ed29b0b4SJens Axboe wq_list_cut(&ctx->iopoll_list, prev, start); 3069ed29b0b4SJens Axboe io_free_batch_list(ctx, pos); 3070ed29b0b4SJens Axboe return nr_events; 3071ed29b0b4SJens Axboe } 3072ed29b0b4SJens Axboe 3073ed29b0b4SJens Axboe /* 3074ed29b0b4SJens Axboe * We can't just wait for polled events to come to us, we have to actively 3075ed29b0b4SJens Axboe * find and complete them. 3076ed29b0b4SJens Axboe */ 3077ed29b0b4SJens Axboe static __cold void io_iopoll_try_reap_events(struct io_ring_ctx *ctx) 3078ed29b0b4SJens Axboe { 3079ed29b0b4SJens Axboe if (!(ctx->flags & IORING_SETUP_IOPOLL)) 3080ed29b0b4SJens Axboe return; 3081ed29b0b4SJens Axboe 3082ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 3083ed29b0b4SJens Axboe while (!wq_list_empty(&ctx->iopoll_list)) { 3084ed29b0b4SJens Axboe /* let it sleep and repeat later if can't complete a request */ 3085ed29b0b4SJens Axboe if (io_do_iopoll(ctx, true) == 0) 3086ed29b0b4SJens Axboe break; 3087ed29b0b4SJens Axboe /* 3088ed29b0b4SJens Axboe * Ensure we allow local-to-the-cpu processing to take place, 3089ed29b0b4SJens Axboe * in this case we need to ensure that we reap all events. 3090ed29b0b4SJens Axboe * Also let task_work, etc. to progress by releasing the mutex 3091ed29b0b4SJens Axboe */ 3092ed29b0b4SJens Axboe if (need_resched()) { 3093ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 3094ed29b0b4SJens Axboe cond_resched(); 3095ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 3096ed29b0b4SJens Axboe } 3097ed29b0b4SJens Axboe } 3098ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 3099ed29b0b4SJens Axboe } 3100ed29b0b4SJens Axboe 3101ed29b0b4SJens Axboe static int io_iopoll_check(struct io_ring_ctx *ctx, long min) 3102ed29b0b4SJens Axboe { 3103ed29b0b4SJens Axboe unsigned int nr_events = 0; 3104ed29b0b4SJens Axboe int ret = 0; 3105ed29b0b4SJens Axboe unsigned long check_cq; 3106ed29b0b4SJens Axboe 3107ed29b0b4SJens Axboe /* 3108ed29b0b4SJens Axboe * Don't enter poll loop if we already have events pending. 3109ed29b0b4SJens Axboe * If we do, we can potentially be spinning for commands that 3110ed29b0b4SJens Axboe * already triggered a CQE (eg in error). 3111ed29b0b4SJens Axboe */ 3112ed29b0b4SJens Axboe check_cq = READ_ONCE(ctx->check_cq); 3113ed29b0b4SJens Axboe if (check_cq & BIT(IO_CHECK_CQ_OVERFLOW_BIT)) 3114ed29b0b4SJens Axboe __io_cqring_overflow_flush(ctx, false); 3115ed29b0b4SJens Axboe if (io_cqring_events(ctx)) 3116ed29b0b4SJens Axboe return 0; 3117ed29b0b4SJens Axboe 3118ed29b0b4SJens Axboe /* 3119ed29b0b4SJens Axboe * Similarly do not spin if we have not informed the user of any 3120ed29b0b4SJens Axboe * dropped CQE. 3121ed29b0b4SJens Axboe */ 3122ed29b0b4SJens Axboe if (unlikely(check_cq & BIT(IO_CHECK_CQ_DROPPED_BIT))) 3123ed29b0b4SJens Axboe return -EBADR; 3124ed29b0b4SJens Axboe 3125ed29b0b4SJens Axboe do { 3126ed29b0b4SJens Axboe /* 3127ed29b0b4SJens Axboe * If a submit got punted to a workqueue, we can have the 3128ed29b0b4SJens Axboe * application entering polling for a command before it gets 3129ed29b0b4SJens Axboe * issued. That app will hold the uring_lock for the duration 3130ed29b0b4SJens Axboe * of the poll right here, so we need to take a breather every 3131ed29b0b4SJens Axboe * now and then to ensure that the issue has a chance to add 3132ed29b0b4SJens Axboe * the poll to the issued list. Otherwise we can spin here 3133ed29b0b4SJens Axboe * forever, while the workqueue is stuck trying to acquire the 3134ed29b0b4SJens Axboe * very same mutex. 3135ed29b0b4SJens Axboe */ 3136ed29b0b4SJens Axboe if (wq_list_empty(&ctx->iopoll_list)) { 3137ed29b0b4SJens Axboe u32 tail = ctx->cached_cq_tail; 3138ed29b0b4SJens Axboe 3139ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 3140ed29b0b4SJens Axboe io_run_task_work(); 3141ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 3142ed29b0b4SJens Axboe 3143ed29b0b4SJens Axboe /* some requests don't go through iopoll_list */ 3144ed29b0b4SJens Axboe if (tail != ctx->cached_cq_tail || 3145ed29b0b4SJens Axboe wq_list_empty(&ctx->iopoll_list)) 3146ed29b0b4SJens Axboe break; 3147ed29b0b4SJens Axboe } 3148ed29b0b4SJens Axboe ret = io_do_iopoll(ctx, !min); 3149ed29b0b4SJens Axboe if (ret < 0) 3150ed29b0b4SJens Axboe break; 3151ed29b0b4SJens Axboe nr_events += ret; 3152ed29b0b4SJens Axboe ret = 0; 3153ed29b0b4SJens Axboe } while (nr_events < min && !need_resched()); 3154ed29b0b4SJens Axboe 3155ed29b0b4SJens Axboe return ret; 3156ed29b0b4SJens Axboe } 3157ed29b0b4SJens Axboe 3158ed29b0b4SJens Axboe static void kiocb_end_write(struct io_kiocb *req) 3159ed29b0b4SJens Axboe { 3160ed29b0b4SJens Axboe /* 3161ed29b0b4SJens Axboe * Tell lockdep we inherited freeze protection from submission 3162ed29b0b4SJens Axboe * thread. 3163ed29b0b4SJens Axboe */ 3164ed29b0b4SJens Axboe if (req->flags & REQ_F_ISREG) { 3165ed29b0b4SJens Axboe struct super_block *sb = file_inode(req->file)->i_sb; 3166ed29b0b4SJens Axboe 3167ed29b0b4SJens Axboe __sb_writers_acquired(sb, SB_FREEZE_WRITE); 3168ed29b0b4SJens Axboe sb_end_write(sb); 3169ed29b0b4SJens Axboe } 3170ed29b0b4SJens Axboe } 3171ed29b0b4SJens Axboe 3172ed29b0b4SJens Axboe #ifdef CONFIG_BLOCK 3173ed29b0b4SJens Axboe static bool io_resubmit_prep(struct io_kiocb *req) 3174ed29b0b4SJens Axboe { 31753c306fb2SJens Axboe struct io_async_rw *io = req->async_data; 3176ed29b0b4SJens Axboe 3177ed29b0b4SJens Axboe if (!req_has_async_data(req)) 3178ed29b0b4SJens Axboe return !io_req_prep_async(req); 31793c306fb2SJens Axboe iov_iter_restore(&io->s.iter, &io->s.iter_state); 3180ed29b0b4SJens Axboe return true; 3181ed29b0b4SJens Axboe } 3182ed29b0b4SJens Axboe 3183ed29b0b4SJens Axboe static bool io_rw_should_reissue(struct io_kiocb *req) 3184ed29b0b4SJens Axboe { 3185ed29b0b4SJens Axboe umode_t mode = file_inode(req->file)->i_mode; 3186ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3187ed29b0b4SJens Axboe 3188ed29b0b4SJens Axboe if (!S_ISBLK(mode) && !S_ISREG(mode)) 3189ed29b0b4SJens Axboe return false; 3190ed29b0b4SJens Axboe if ((req->flags & REQ_F_NOWAIT) || (io_wq_current_is_worker() && 3191ed29b0b4SJens Axboe !(ctx->flags & IORING_SETUP_IOPOLL))) 3192ed29b0b4SJens Axboe return false; 3193ed29b0b4SJens Axboe /* 3194ed29b0b4SJens Axboe * If ref is dying, we might be running poll reap from the exit work. 3195ed29b0b4SJens Axboe * Don't attempt to reissue from that path, just let it fail with 3196ed29b0b4SJens Axboe * -EAGAIN. 3197ed29b0b4SJens Axboe */ 3198ed29b0b4SJens Axboe if (percpu_ref_is_dying(&ctx->refs)) 3199ed29b0b4SJens Axboe return false; 3200ed29b0b4SJens Axboe /* 3201ed29b0b4SJens Axboe * Play it safe and assume not safe to re-import and reissue if we're 3202ed29b0b4SJens Axboe * not in the original thread group (or in task context). 3203ed29b0b4SJens Axboe */ 3204ed29b0b4SJens Axboe if (!same_thread_group(req->task, current) || !in_task()) 3205ed29b0b4SJens Axboe return false; 3206ed29b0b4SJens Axboe return true; 3207ed29b0b4SJens Axboe } 3208ed29b0b4SJens Axboe #else 3209ed29b0b4SJens Axboe static bool io_resubmit_prep(struct io_kiocb *req) 3210ed29b0b4SJens Axboe { 3211ed29b0b4SJens Axboe return false; 3212ed29b0b4SJens Axboe } 3213ed29b0b4SJens Axboe static bool io_rw_should_reissue(struct io_kiocb *req) 3214ed29b0b4SJens Axboe { 3215ed29b0b4SJens Axboe return false; 3216ed29b0b4SJens Axboe } 3217ed29b0b4SJens Axboe #endif 3218ed29b0b4SJens Axboe 3219ed29b0b4SJens Axboe static bool __io_complete_rw_common(struct io_kiocb *req, long res) 3220ed29b0b4SJens Axboe { 32213c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 32223c306fb2SJens Axboe 32233c306fb2SJens Axboe if (rw->kiocb.ki_flags & IOCB_WRITE) { 3224ed29b0b4SJens Axboe kiocb_end_write(req); 3225ed29b0b4SJens Axboe fsnotify_modify(req->file); 3226ed29b0b4SJens Axboe } else { 3227ed29b0b4SJens Axboe fsnotify_access(req->file); 3228ed29b0b4SJens Axboe } 3229ed29b0b4SJens Axboe if (unlikely(res != req->cqe.res)) { 3230ed29b0b4SJens Axboe if ((res == -EAGAIN || res == -EOPNOTSUPP) && 3231ed29b0b4SJens Axboe io_rw_should_reissue(req)) { 3232ed29b0b4SJens Axboe req->flags |= REQ_F_REISSUE | REQ_F_PARTIAL_IO; 3233ed29b0b4SJens Axboe return true; 3234ed29b0b4SJens Axboe } 3235ed29b0b4SJens Axboe req_set_fail(req); 3236ed29b0b4SJens Axboe req->cqe.res = res; 3237ed29b0b4SJens Axboe } 3238ed29b0b4SJens Axboe return false; 3239ed29b0b4SJens Axboe } 3240ed29b0b4SJens Axboe 3241ed29b0b4SJens Axboe static inline void io_req_task_complete(struct io_kiocb *req, bool *locked) 3242ed29b0b4SJens Axboe { 3243ed29b0b4SJens Axboe int res = req->cqe.res; 3244ed29b0b4SJens Axboe 3245ed29b0b4SJens Axboe if (*locked) { 3246ed29b0b4SJens Axboe io_req_complete_state(req, res, io_put_kbuf(req, 0)); 3247ed29b0b4SJens Axboe io_req_add_compl_list(req); 3248ed29b0b4SJens Axboe } else { 3249ed29b0b4SJens Axboe io_req_complete_post(req, res, 3250ed29b0b4SJens Axboe io_put_kbuf(req, IO_URING_F_UNLOCKED)); 3251ed29b0b4SJens Axboe } 3252ed29b0b4SJens Axboe } 3253ed29b0b4SJens Axboe 3254ed29b0b4SJens Axboe static void __io_complete_rw(struct io_kiocb *req, long res, 3255ed29b0b4SJens Axboe unsigned int issue_flags) 3256ed29b0b4SJens Axboe { 3257ed29b0b4SJens Axboe if (__io_complete_rw_common(req, res)) 3258ed29b0b4SJens Axboe return; 3259ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, req->cqe.res, 3260ed29b0b4SJens Axboe io_put_kbuf(req, issue_flags)); 3261ed29b0b4SJens Axboe } 3262ed29b0b4SJens Axboe 3263ed29b0b4SJens Axboe static void io_complete_rw(struct kiocb *kiocb, long res) 3264ed29b0b4SJens Axboe { 32653c306fb2SJens Axboe struct io_rw *rw = container_of(kiocb, struct io_rw, kiocb); 32663c306fb2SJens Axboe struct io_kiocb *req = cmd_to_io_kiocb(rw); 3267ed29b0b4SJens Axboe 3268ed29b0b4SJens Axboe if (__io_complete_rw_common(req, res)) 3269ed29b0b4SJens Axboe return; 3270ed29b0b4SJens Axboe req->cqe.res = res; 3271ed29b0b4SJens Axboe req->io_task_work.func = io_req_task_complete; 3272ed29b0b4SJens Axboe io_req_task_prio_work_add(req); 3273ed29b0b4SJens Axboe } 3274ed29b0b4SJens Axboe 3275ed29b0b4SJens Axboe static void io_complete_rw_iopoll(struct kiocb *kiocb, long res) 3276ed29b0b4SJens Axboe { 32773c306fb2SJens Axboe struct io_rw *rw = container_of(kiocb, struct io_rw, kiocb); 32783c306fb2SJens Axboe struct io_kiocb *req = cmd_to_io_kiocb(rw); 3279ed29b0b4SJens Axboe 3280ed29b0b4SJens Axboe if (kiocb->ki_flags & IOCB_WRITE) 3281ed29b0b4SJens Axboe kiocb_end_write(req); 3282ed29b0b4SJens Axboe if (unlikely(res != req->cqe.res)) { 3283ed29b0b4SJens Axboe if (res == -EAGAIN && io_rw_should_reissue(req)) { 3284ed29b0b4SJens Axboe req->flags |= REQ_F_REISSUE | REQ_F_PARTIAL_IO; 3285ed29b0b4SJens Axboe return; 3286ed29b0b4SJens Axboe } 3287ed29b0b4SJens Axboe req->cqe.res = res; 3288ed29b0b4SJens Axboe } 3289ed29b0b4SJens Axboe 3290ed29b0b4SJens Axboe /* order with io_iopoll_complete() checking ->iopoll_completed */ 3291ed29b0b4SJens Axboe smp_store_release(&req->iopoll_completed, 1); 3292ed29b0b4SJens Axboe } 3293ed29b0b4SJens Axboe 3294ed29b0b4SJens Axboe /* 3295ed29b0b4SJens Axboe * After the iocb has been issued, it's safe to be found on the poll list. 3296ed29b0b4SJens Axboe * Adding the kiocb to the list AFTER submission ensures that we don't 3297ed29b0b4SJens Axboe * find it from a io_do_iopoll() thread before the issuer is done 3298ed29b0b4SJens Axboe * accessing the kiocb cookie. 3299ed29b0b4SJens Axboe */ 3300ed29b0b4SJens Axboe static void io_iopoll_req_issued(struct io_kiocb *req, unsigned int issue_flags) 3301ed29b0b4SJens Axboe { 3302ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3303ed29b0b4SJens Axboe const bool needs_lock = issue_flags & IO_URING_F_UNLOCKED; 3304ed29b0b4SJens Axboe 3305ed29b0b4SJens Axboe /* workqueue context doesn't hold uring_lock, grab it now */ 3306ed29b0b4SJens Axboe if (unlikely(needs_lock)) 3307ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 3308ed29b0b4SJens Axboe 3309ed29b0b4SJens Axboe /* 3310ed29b0b4SJens Axboe * Track whether we have multiple files in our lists. This will impact 3311ed29b0b4SJens Axboe * how we do polling eventually, not spinning if we're on potentially 3312ed29b0b4SJens Axboe * different devices. 3313ed29b0b4SJens Axboe */ 3314ed29b0b4SJens Axboe if (wq_list_empty(&ctx->iopoll_list)) { 3315ed29b0b4SJens Axboe ctx->poll_multi_queue = false; 3316ed29b0b4SJens Axboe } else if (!ctx->poll_multi_queue) { 3317ed29b0b4SJens Axboe struct io_kiocb *list_req; 3318ed29b0b4SJens Axboe 3319ed29b0b4SJens Axboe list_req = container_of(ctx->iopoll_list.first, struct io_kiocb, 3320ed29b0b4SJens Axboe comp_list); 3321ed29b0b4SJens Axboe if (list_req->file != req->file) 3322ed29b0b4SJens Axboe ctx->poll_multi_queue = true; 3323ed29b0b4SJens Axboe } 3324ed29b0b4SJens Axboe 3325ed29b0b4SJens Axboe /* 3326ed29b0b4SJens Axboe * For fast devices, IO may have already completed. If it has, add 3327ed29b0b4SJens Axboe * it to the front so we find it first. 3328ed29b0b4SJens Axboe */ 3329ed29b0b4SJens Axboe if (READ_ONCE(req->iopoll_completed)) 3330ed29b0b4SJens Axboe wq_list_add_head(&req->comp_list, &ctx->iopoll_list); 3331ed29b0b4SJens Axboe else 3332ed29b0b4SJens Axboe wq_list_add_tail(&req->comp_list, &ctx->iopoll_list); 3333ed29b0b4SJens Axboe 3334ed29b0b4SJens Axboe if (unlikely(needs_lock)) { 3335ed29b0b4SJens Axboe /* 3336ed29b0b4SJens Axboe * If IORING_SETUP_SQPOLL is enabled, sqes are either handle 3337ed29b0b4SJens Axboe * in sq thread task context or in io worker task context. If 3338ed29b0b4SJens Axboe * current task context is sq thread, we don't need to check 3339ed29b0b4SJens Axboe * whether should wake up sq thread. 3340ed29b0b4SJens Axboe */ 3341ed29b0b4SJens Axboe if ((ctx->flags & IORING_SETUP_SQPOLL) && 3342ed29b0b4SJens Axboe wq_has_sleeper(&ctx->sq_data->wait)) 3343ed29b0b4SJens Axboe wake_up(&ctx->sq_data->wait); 3344ed29b0b4SJens Axboe 3345ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 3346ed29b0b4SJens Axboe } 3347ed29b0b4SJens Axboe } 3348ed29b0b4SJens Axboe 3349ed29b0b4SJens Axboe static bool io_bdev_nowait(struct block_device *bdev) 3350ed29b0b4SJens Axboe { 3351ed29b0b4SJens Axboe return !bdev || blk_queue_nowait(bdev_get_queue(bdev)); 3352ed29b0b4SJens Axboe } 3353ed29b0b4SJens Axboe 3354ed29b0b4SJens Axboe /* 3355ed29b0b4SJens Axboe * If we tracked the file through the SCM inflight mechanism, we could support 3356ed29b0b4SJens Axboe * any file. For now, just ensure that anything potentially problematic is done 3357ed29b0b4SJens Axboe * inline. 3358ed29b0b4SJens Axboe */ 3359ed29b0b4SJens Axboe static bool __io_file_supports_nowait(struct file *file, umode_t mode) 3360ed29b0b4SJens Axboe { 3361ed29b0b4SJens Axboe if (S_ISBLK(mode)) { 3362ed29b0b4SJens Axboe if (IS_ENABLED(CONFIG_BLOCK) && 3363ed29b0b4SJens Axboe io_bdev_nowait(I_BDEV(file->f_mapping->host))) 3364ed29b0b4SJens Axboe return true; 3365ed29b0b4SJens Axboe return false; 3366ed29b0b4SJens Axboe } 3367ed29b0b4SJens Axboe if (S_ISSOCK(mode)) 3368ed29b0b4SJens Axboe return true; 3369ed29b0b4SJens Axboe if (S_ISREG(mode)) { 3370ed29b0b4SJens Axboe if (IS_ENABLED(CONFIG_BLOCK) && 3371ed29b0b4SJens Axboe io_bdev_nowait(file->f_inode->i_sb->s_bdev) && 3372ed29b0b4SJens Axboe file->f_op != &io_uring_fops) 3373ed29b0b4SJens Axboe return true; 3374ed29b0b4SJens Axboe return false; 3375ed29b0b4SJens Axboe } 3376ed29b0b4SJens Axboe 3377ed29b0b4SJens Axboe /* any ->read/write should understand O_NONBLOCK */ 3378ed29b0b4SJens Axboe if (file->f_flags & O_NONBLOCK) 3379ed29b0b4SJens Axboe return true; 3380ed29b0b4SJens Axboe return file->f_mode & FMODE_NOWAIT; 3381ed29b0b4SJens Axboe } 3382ed29b0b4SJens Axboe 3383ed29b0b4SJens Axboe /* 3384ed29b0b4SJens Axboe * If we tracked the file through the SCM inflight mechanism, we could support 3385ed29b0b4SJens Axboe * any file. For now, just ensure that anything potentially problematic is done 3386ed29b0b4SJens Axboe * inline. 3387ed29b0b4SJens Axboe */ 3388ed29b0b4SJens Axboe static unsigned int io_file_get_flags(struct file *file) 3389ed29b0b4SJens Axboe { 3390ed29b0b4SJens Axboe umode_t mode = file_inode(file)->i_mode; 3391ed29b0b4SJens Axboe unsigned int res = 0; 3392ed29b0b4SJens Axboe 3393ed29b0b4SJens Axboe if (S_ISREG(mode)) 3394ed29b0b4SJens Axboe res |= FFS_ISREG; 3395ed29b0b4SJens Axboe if (__io_file_supports_nowait(file, mode)) 3396ed29b0b4SJens Axboe res |= FFS_NOWAIT; 3397ed29b0b4SJens Axboe if (io_file_need_scm(file)) 3398ed29b0b4SJens Axboe res |= FFS_SCM; 3399ed29b0b4SJens Axboe return res; 3400ed29b0b4SJens Axboe } 3401ed29b0b4SJens Axboe 3402ed29b0b4SJens Axboe static inline bool io_file_supports_nowait(struct io_kiocb *req) 3403ed29b0b4SJens Axboe { 3404ed29b0b4SJens Axboe return req->flags & REQ_F_SUPPORT_NOWAIT; 3405ed29b0b4SJens Axboe } 3406ed29b0b4SJens Axboe 3407ed29b0b4SJens Axboe static int io_prep_rw(struct io_kiocb *req, const struct io_uring_sqe *sqe) 3408ed29b0b4SJens Axboe { 34093c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 3410ed29b0b4SJens Axboe unsigned ioprio; 3411ed29b0b4SJens Axboe int ret; 3412ed29b0b4SJens Axboe 34133c306fb2SJens Axboe rw->kiocb.ki_pos = READ_ONCE(sqe->off); 3414ed29b0b4SJens Axboe /* used for fixed read/write too - just read unconditionally */ 3415ed29b0b4SJens Axboe req->buf_index = READ_ONCE(sqe->buf_index); 3416ed29b0b4SJens Axboe 3417ed29b0b4SJens Axboe if (req->opcode == IORING_OP_READ_FIXED || 3418ed29b0b4SJens Axboe req->opcode == IORING_OP_WRITE_FIXED) { 3419ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3420ed29b0b4SJens Axboe u16 index; 3421ed29b0b4SJens Axboe 3422ed29b0b4SJens Axboe if (unlikely(req->buf_index >= ctx->nr_user_bufs)) 3423ed29b0b4SJens Axboe return -EFAULT; 3424ed29b0b4SJens Axboe index = array_index_nospec(req->buf_index, ctx->nr_user_bufs); 3425ed29b0b4SJens Axboe req->imu = ctx->user_bufs[index]; 3426ed29b0b4SJens Axboe io_req_set_rsrc_node(req, ctx, 0); 3427ed29b0b4SJens Axboe } 3428ed29b0b4SJens Axboe 3429ed29b0b4SJens Axboe ioprio = READ_ONCE(sqe->ioprio); 3430ed29b0b4SJens Axboe if (ioprio) { 3431ed29b0b4SJens Axboe ret = ioprio_check_cap(ioprio); 3432ed29b0b4SJens Axboe if (ret) 3433ed29b0b4SJens Axboe return ret; 3434ed29b0b4SJens Axboe 34353c306fb2SJens Axboe rw->kiocb.ki_ioprio = ioprio; 3436ed29b0b4SJens Axboe } else { 34373c306fb2SJens Axboe rw->kiocb.ki_ioprio = get_current_ioprio(); 3438ed29b0b4SJens Axboe } 3439ed29b0b4SJens Axboe 34403c306fb2SJens Axboe rw->addr = READ_ONCE(sqe->addr); 34413c306fb2SJens Axboe rw->len = READ_ONCE(sqe->len); 34423c306fb2SJens Axboe rw->flags = READ_ONCE(sqe->rw_flags); 3443ed29b0b4SJens Axboe return 0; 3444ed29b0b4SJens Axboe } 3445ed29b0b4SJens Axboe 3446ed29b0b4SJens Axboe static inline void io_rw_done(struct kiocb *kiocb, ssize_t ret) 3447ed29b0b4SJens Axboe { 3448ed29b0b4SJens Axboe switch (ret) { 3449ed29b0b4SJens Axboe case -EIOCBQUEUED: 3450ed29b0b4SJens Axboe break; 3451ed29b0b4SJens Axboe case -ERESTARTSYS: 3452ed29b0b4SJens Axboe case -ERESTARTNOINTR: 3453ed29b0b4SJens Axboe case -ERESTARTNOHAND: 3454ed29b0b4SJens Axboe case -ERESTART_RESTARTBLOCK: 3455ed29b0b4SJens Axboe /* 3456ed29b0b4SJens Axboe * We can't just restart the syscall, since previously 3457ed29b0b4SJens Axboe * submitted sqes may already be in progress. Just fail this 3458ed29b0b4SJens Axboe * IO with EINTR. 3459ed29b0b4SJens Axboe */ 3460ed29b0b4SJens Axboe ret = -EINTR; 3461ed29b0b4SJens Axboe fallthrough; 3462ed29b0b4SJens Axboe default: 3463ed29b0b4SJens Axboe kiocb->ki_complete(kiocb, ret); 3464ed29b0b4SJens Axboe } 3465ed29b0b4SJens Axboe } 3466ed29b0b4SJens Axboe 3467ed29b0b4SJens Axboe static inline loff_t *io_kiocb_update_pos(struct io_kiocb *req) 3468ed29b0b4SJens Axboe { 34693c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 3470ed29b0b4SJens Axboe 34713c306fb2SJens Axboe if (rw->kiocb.ki_pos != -1) 34723c306fb2SJens Axboe return &rw->kiocb.ki_pos; 3473ed29b0b4SJens Axboe 3474ed29b0b4SJens Axboe if (!(req->file->f_mode & FMODE_STREAM)) { 3475ed29b0b4SJens Axboe req->flags |= REQ_F_CUR_POS; 34763c306fb2SJens Axboe rw->kiocb.ki_pos = req->file->f_pos; 34773c306fb2SJens Axboe return &rw->kiocb.ki_pos; 3478ed29b0b4SJens Axboe } 3479ed29b0b4SJens Axboe 34803c306fb2SJens Axboe rw->kiocb.ki_pos = 0; 3481ed29b0b4SJens Axboe return NULL; 3482ed29b0b4SJens Axboe } 3483ed29b0b4SJens Axboe 3484ed29b0b4SJens Axboe static void kiocb_done(struct io_kiocb *req, ssize_t ret, 3485ed29b0b4SJens Axboe unsigned int issue_flags) 3486ed29b0b4SJens Axboe { 3487ed29b0b4SJens Axboe struct io_async_rw *io = req->async_data; 34883c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 3489ed29b0b4SJens Axboe 3490ed29b0b4SJens Axboe /* add previously done IO, if any */ 3491ed29b0b4SJens Axboe if (req_has_async_data(req) && io->bytes_done > 0) { 3492ed29b0b4SJens Axboe if (ret < 0) 3493ed29b0b4SJens Axboe ret = io->bytes_done; 3494ed29b0b4SJens Axboe else 3495ed29b0b4SJens Axboe ret += io->bytes_done; 3496ed29b0b4SJens Axboe } 3497ed29b0b4SJens Axboe 3498ed29b0b4SJens Axboe if (req->flags & REQ_F_CUR_POS) 34993c306fb2SJens Axboe req->file->f_pos = rw->kiocb.ki_pos; 35003c306fb2SJens Axboe if (ret >= 0 && (rw->kiocb.ki_complete == io_complete_rw)) 3501ed29b0b4SJens Axboe __io_complete_rw(req, ret, issue_flags); 3502ed29b0b4SJens Axboe else 35033c306fb2SJens Axboe io_rw_done(&rw->kiocb, ret); 3504ed29b0b4SJens Axboe 3505ed29b0b4SJens Axboe if (req->flags & REQ_F_REISSUE) { 3506ed29b0b4SJens Axboe req->flags &= ~REQ_F_REISSUE; 3507ed29b0b4SJens Axboe if (io_resubmit_prep(req)) 3508ed29b0b4SJens Axboe io_req_task_queue_reissue(req); 3509ed29b0b4SJens Axboe else 3510ed29b0b4SJens Axboe io_req_task_queue_fail(req, ret); 3511ed29b0b4SJens Axboe } 3512ed29b0b4SJens Axboe } 3513ed29b0b4SJens Axboe 35143c306fb2SJens Axboe static int __io_import_fixed(struct io_kiocb *req, int ddir, 35153c306fb2SJens Axboe struct iov_iter *iter, struct io_mapped_ubuf *imu) 3516ed29b0b4SJens Axboe { 35173c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 35183c306fb2SJens Axboe size_t len = rw->len; 35193c306fb2SJens Axboe u64 buf_end, buf_addr = rw->addr; 3520ed29b0b4SJens Axboe size_t offset; 3521ed29b0b4SJens Axboe 3522ed29b0b4SJens Axboe if (unlikely(check_add_overflow(buf_addr, (u64)len, &buf_end))) 3523ed29b0b4SJens Axboe return -EFAULT; 3524ed29b0b4SJens Axboe /* not inside the mapped region */ 3525ed29b0b4SJens Axboe if (unlikely(buf_addr < imu->ubuf || buf_end > imu->ubuf_end)) 3526ed29b0b4SJens Axboe return -EFAULT; 3527ed29b0b4SJens Axboe 3528ed29b0b4SJens Axboe /* 3529ed29b0b4SJens Axboe * May not be a start of buffer, set size appropriately 3530ed29b0b4SJens Axboe * and advance us to the beginning. 3531ed29b0b4SJens Axboe */ 3532ed29b0b4SJens Axboe offset = buf_addr - imu->ubuf; 35333c306fb2SJens Axboe iov_iter_bvec(iter, ddir, imu->bvec, imu->nr_bvecs, offset + len); 3534ed29b0b4SJens Axboe 3535ed29b0b4SJens Axboe if (offset) { 3536ed29b0b4SJens Axboe /* 3537ed29b0b4SJens Axboe * Don't use iov_iter_advance() here, as it's really slow for 3538ed29b0b4SJens Axboe * using the latter parts of a big fixed buffer - it iterates 3539ed29b0b4SJens Axboe * over each segment manually. We can cheat a bit here, because 3540ed29b0b4SJens Axboe * we know that: 3541ed29b0b4SJens Axboe * 3542ed29b0b4SJens Axboe * 1) it's a BVEC iter, we set it up 3543ed29b0b4SJens Axboe * 2) all bvecs are PAGE_SIZE in size, except potentially the 3544ed29b0b4SJens Axboe * first and last bvec 3545ed29b0b4SJens Axboe * 3546ed29b0b4SJens Axboe * So just find our index, and adjust the iterator afterwards. 3547ed29b0b4SJens Axboe * If the offset is within the first bvec (or the whole first 3548ed29b0b4SJens Axboe * bvec, just use iov_iter_advance(). This makes it easier 3549ed29b0b4SJens Axboe * since we can just skip the first segment, which may not 3550ed29b0b4SJens Axboe * be PAGE_SIZE aligned. 3551ed29b0b4SJens Axboe */ 3552ed29b0b4SJens Axboe const struct bio_vec *bvec = imu->bvec; 3553ed29b0b4SJens Axboe 3554ed29b0b4SJens Axboe if (offset <= bvec->bv_len) { 3555ed29b0b4SJens Axboe iov_iter_advance(iter, offset); 3556ed29b0b4SJens Axboe } else { 3557ed29b0b4SJens Axboe unsigned long seg_skip; 3558ed29b0b4SJens Axboe 3559ed29b0b4SJens Axboe /* skip first vec */ 3560ed29b0b4SJens Axboe offset -= bvec->bv_len; 3561ed29b0b4SJens Axboe seg_skip = 1 + (offset >> PAGE_SHIFT); 3562ed29b0b4SJens Axboe 3563ed29b0b4SJens Axboe iter->bvec = bvec + seg_skip; 3564ed29b0b4SJens Axboe iter->nr_segs -= seg_skip; 3565ed29b0b4SJens Axboe iter->count -= bvec->bv_len + offset; 3566ed29b0b4SJens Axboe iter->iov_offset = offset & ~PAGE_MASK; 3567ed29b0b4SJens Axboe } 3568ed29b0b4SJens Axboe } 3569ed29b0b4SJens Axboe 3570ed29b0b4SJens Axboe return 0; 3571ed29b0b4SJens Axboe } 3572ed29b0b4SJens Axboe 3573ed29b0b4SJens Axboe static int io_import_fixed(struct io_kiocb *req, int rw, struct iov_iter *iter, 3574ed29b0b4SJens Axboe unsigned int issue_flags) 3575ed29b0b4SJens Axboe { 3576ed29b0b4SJens Axboe if (WARN_ON_ONCE(!req->imu)) 3577ed29b0b4SJens Axboe return -EFAULT; 3578ed29b0b4SJens Axboe return __io_import_fixed(req, rw, iter, req->imu); 3579ed29b0b4SJens Axboe } 3580ed29b0b4SJens Axboe 3581ed29b0b4SJens Axboe static int io_buffer_add_list(struct io_ring_ctx *ctx, 3582ed29b0b4SJens Axboe struct io_buffer_list *bl, unsigned int bgid) 3583ed29b0b4SJens Axboe { 3584ed29b0b4SJens Axboe bl->bgid = bgid; 3585ed29b0b4SJens Axboe if (bgid < BGID_ARRAY) 3586ed29b0b4SJens Axboe return 0; 3587ed29b0b4SJens Axboe 3588ed29b0b4SJens Axboe return xa_err(xa_store(&ctx->io_bl_xa, bgid, bl, GFP_KERNEL)); 3589ed29b0b4SJens Axboe } 3590ed29b0b4SJens Axboe 3591ed29b0b4SJens Axboe static void __user *io_provided_buffer_select(struct io_kiocb *req, size_t *len, 3592ed29b0b4SJens Axboe struct io_buffer_list *bl) 3593ed29b0b4SJens Axboe { 3594ed29b0b4SJens Axboe if (!list_empty(&bl->buf_list)) { 3595ed29b0b4SJens Axboe struct io_buffer *kbuf; 3596ed29b0b4SJens Axboe 3597ed29b0b4SJens Axboe kbuf = list_first_entry(&bl->buf_list, struct io_buffer, list); 3598ed29b0b4SJens Axboe list_del(&kbuf->list); 3599ed29b0b4SJens Axboe if (*len > kbuf->len) 3600ed29b0b4SJens Axboe *len = kbuf->len; 3601ed29b0b4SJens Axboe req->flags |= REQ_F_BUFFER_SELECTED; 3602ed29b0b4SJens Axboe req->kbuf = kbuf; 3603ed29b0b4SJens Axboe req->buf_index = kbuf->bid; 3604ed29b0b4SJens Axboe return u64_to_user_ptr(kbuf->addr); 3605ed29b0b4SJens Axboe } 3606ed29b0b4SJens Axboe return NULL; 3607ed29b0b4SJens Axboe } 3608ed29b0b4SJens Axboe 3609ed29b0b4SJens Axboe static void __user *io_ring_buffer_select(struct io_kiocb *req, size_t *len, 3610ed29b0b4SJens Axboe struct io_buffer_list *bl, 3611ed29b0b4SJens Axboe unsigned int issue_flags) 3612ed29b0b4SJens Axboe { 3613ed29b0b4SJens Axboe struct io_uring_buf_ring *br = bl->buf_ring; 3614ed29b0b4SJens Axboe struct io_uring_buf *buf; 3615ed29b0b4SJens Axboe __u16 head = bl->head; 3616ed29b0b4SJens Axboe 3617ed29b0b4SJens Axboe if (unlikely(smp_load_acquire(&br->tail) == head)) 3618ed29b0b4SJens Axboe return NULL; 3619ed29b0b4SJens Axboe 3620ed29b0b4SJens Axboe head &= bl->mask; 3621ed29b0b4SJens Axboe if (head < IO_BUFFER_LIST_BUF_PER_PAGE) { 3622ed29b0b4SJens Axboe buf = &br->bufs[head]; 3623ed29b0b4SJens Axboe } else { 3624ed29b0b4SJens Axboe int off = head & (IO_BUFFER_LIST_BUF_PER_PAGE - 1); 3625ed29b0b4SJens Axboe int index = head / IO_BUFFER_LIST_BUF_PER_PAGE; 3626ed29b0b4SJens Axboe buf = page_address(bl->buf_pages[index]); 3627ed29b0b4SJens Axboe buf += off; 3628ed29b0b4SJens Axboe } 3629ed29b0b4SJens Axboe if (*len > buf->len) 3630ed29b0b4SJens Axboe *len = buf->len; 3631ed29b0b4SJens Axboe req->flags |= REQ_F_BUFFER_RING; 3632ed29b0b4SJens Axboe req->buf_list = bl; 3633ed29b0b4SJens Axboe req->buf_index = buf->bid; 3634ed29b0b4SJens Axboe 3635ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_UNLOCKED || !file_can_poll(req->file)) { 3636ed29b0b4SJens Axboe /* 3637ed29b0b4SJens Axboe * If we came in unlocked, we have no choice but to consume the 3638ed29b0b4SJens Axboe * buffer here. This does mean it'll be pinned until the IO 3639ed29b0b4SJens Axboe * completes. But coming in unlocked means we're in io-wq 3640ed29b0b4SJens Axboe * context, hence there should be no further retry. For the 3641ed29b0b4SJens Axboe * locked case, the caller must ensure to call the commit when 3642ed29b0b4SJens Axboe * the transfer completes (or if we get -EAGAIN and must poll 3643ed29b0b4SJens Axboe * or retry). 3644ed29b0b4SJens Axboe */ 3645ed29b0b4SJens Axboe req->buf_list = NULL; 3646ed29b0b4SJens Axboe bl->head++; 3647ed29b0b4SJens Axboe } 3648ed29b0b4SJens Axboe return u64_to_user_ptr(buf->addr); 3649ed29b0b4SJens Axboe } 3650ed29b0b4SJens Axboe 3651ed29b0b4SJens Axboe static void __user *io_buffer_select(struct io_kiocb *req, size_t *len, 3652ed29b0b4SJens Axboe unsigned int issue_flags) 3653ed29b0b4SJens Axboe { 3654ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 3655ed29b0b4SJens Axboe struct io_buffer_list *bl; 3656ed29b0b4SJens Axboe void __user *ret = NULL; 3657ed29b0b4SJens Axboe 3658ed29b0b4SJens Axboe io_ring_submit_lock(req->ctx, issue_flags); 3659ed29b0b4SJens Axboe 3660ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, req->buf_index); 3661ed29b0b4SJens Axboe if (likely(bl)) { 3662ed29b0b4SJens Axboe if (bl->buf_nr_pages) 3663ed29b0b4SJens Axboe ret = io_ring_buffer_select(req, len, bl, issue_flags); 3664ed29b0b4SJens Axboe else 3665ed29b0b4SJens Axboe ret = io_provided_buffer_select(req, len, bl); 3666ed29b0b4SJens Axboe } 3667ed29b0b4SJens Axboe io_ring_submit_unlock(req->ctx, issue_flags); 3668ed29b0b4SJens Axboe return ret; 3669ed29b0b4SJens Axboe } 3670ed29b0b4SJens Axboe 3671ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 3672ed29b0b4SJens Axboe static ssize_t io_compat_import(struct io_kiocb *req, struct iovec *iov, 3673ed29b0b4SJens Axboe unsigned int issue_flags) 3674ed29b0b4SJens Axboe { 36753c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 3676ed29b0b4SJens Axboe struct compat_iovec __user *uiov; 3677ed29b0b4SJens Axboe compat_ssize_t clen; 3678ed29b0b4SJens Axboe void __user *buf; 3679ed29b0b4SJens Axboe size_t len; 3680ed29b0b4SJens Axboe 36813c306fb2SJens Axboe uiov = u64_to_user_ptr(rw->addr); 3682ed29b0b4SJens Axboe if (!access_ok(uiov, sizeof(*uiov))) 3683ed29b0b4SJens Axboe return -EFAULT; 3684ed29b0b4SJens Axboe if (__get_user(clen, &uiov->iov_len)) 3685ed29b0b4SJens Axboe return -EFAULT; 3686ed29b0b4SJens Axboe if (clen < 0) 3687ed29b0b4SJens Axboe return -EINVAL; 3688ed29b0b4SJens Axboe 3689ed29b0b4SJens Axboe len = clen; 3690ed29b0b4SJens Axboe buf = io_buffer_select(req, &len, issue_flags); 3691ed29b0b4SJens Axboe if (!buf) 3692ed29b0b4SJens Axboe return -ENOBUFS; 36933c306fb2SJens Axboe rw->addr = (unsigned long) buf; 3694ed29b0b4SJens Axboe iov[0].iov_base = buf; 36953c306fb2SJens Axboe rw->len = iov[0].iov_len = (compat_size_t) len; 3696ed29b0b4SJens Axboe return 0; 3697ed29b0b4SJens Axboe } 3698ed29b0b4SJens Axboe #endif 3699ed29b0b4SJens Axboe 3700ed29b0b4SJens Axboe static ssize_t __io_iov_buffer_select(struct io_kiocb *req, struct iovec *iov, 3701ed29b0b4SJens Axboe unsigned int issue_flags) 3702ed29b0b4SJens Axboe { 37033c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 37043c306fb2SJens Axboe struct iovec __user *uiov = u64_to_user_ptr(rw->addr); 3705ed29b0b4SJens Axboe void __user *buf; 3706ed29b0b4SJens Axboe ssize_t len; 3707ed29b0b4SJens Axboe 3708ed29b0b4SJens Axboe if (copy_from_user(iov, uiov, sizeof(*uiov))) 3709ed29b0b4SJens Axboe return -EFAULT; 3710ed29b0b4SJens Axboe 3711ed29b0b4SJens Axboe len = iov[0].iov_len; 3712ed29b0b4SJens Axboe if (len < 0) 3713ed29b0b4SJens Axboe return -EINVAL; 3714ed29b0b4SJens Axboe buf = io_buffer_select(req, &len, issue_flags); 3715ed29b0b4SJens Axboe if (!buf) 3716ed29b0b4SJens Axboe return -ENOBUFS; 37173c306fb2SJens Axboe rw->addr = (unsigned long) buf; 3718ed29b0b4SJens Axboe iov[0].iov_base = buf; 37193c306fb2SJens Axboe rw->len = iov[0].iov_len = len; 3720ed29b0b4SJens Axboe return 0; 3721ed29b0b4SJens Axboe } 3722ed29b0b4SJens Axboe 3723ed29b0b4SJens Axboe static ssize_t io_iov_buffer_select(struct io_kiocb *req, struct iovec *iov, 3724ed29b0b4SJens Axboe unsigned int issue_flags) 3725ed29b0b4SJens Axboe { 37263c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 37273c306fb2SJens Axboe 3728ed29b0b4SJens Axboe if (req->flags & (REQ_F_BUFFER_SELECTED|REQ_F_BUFFER_RING)) { 37293c306fb2SJens Axboe iov[0].iov_base = u64_to_user_ptr(rw->addr); 37303c306fb2SJens Axboe iov[0].iov_len = rw->len; 3731ed29b0b4SJens Axboe return 0; 3732ed29b0b4SJens Axboe } 37333c306fb2SJens Axboe if (rw->len != 1) 3734ed29b0b4SJens Axboe return -EINVAL; 3735ed29b0b4SJens Axboe 3736ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 3737ed29b0b4SJens Axboe if (req->ctx->compat) 3738ed29b0b4SJens Axboe return io_compat_import(req, iov, issue_flags); 3739ed29b0b4SJens Axboe #endif 3740ed29b0b4SJens Axboe 3741ed29b0b4SJens Axboe return __io_iov_buffer_select(req, iov, issue_flags); 3742ed29b0b4SJens Axboe } 3743ed29b0b4SJens Axboe 3744ed29b0b4SJens Axboe static inline bool io_do_buffer_select(struct io_kiocb *req) 3745ed29b0b4SJens Axboe { 3746ed29b0b4SJens Axboe if (!(req->flags & REQ_F_BUFFER_SELECT)) 3747ed29b0b4SJens Axboe return false; 3748ed29b0b4SJens Axboe return !(req->flags & (REQ_F_BUFFER_SELECTED|REQ_F_BUFFER_RING)); 3749ed29b0b4SJens Axboe } 3750ed29b0b4SJens Axboe 37513c306fb2SJens Axboe static struct iovec *__io_import_iovec(int ddir, struct io_kiocb *req, 3752ed29b0b4SJens Axboe struct io_rw_state *s, 3753ed29b0b4SJens Axboe unsigned int issue_flags) 3754ed29b0b4SJens Axboe { 37553c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 3756ed29b0b4SJens Axboe struct iov_iter *iter = &s->iter; 3757ed29b0b4SJens Axboe u8 opcode = req->opcode; 3758ed29b0b4SJens Axboe struct iovec *iovec; 3759ed29b0b4SJens Axboe void __user *buf; 3760ed29b0b4SJens Axboe size_t sqe_len; 3761ed29b0b4SJens Axboe ssize_t ret; 3762ed29b0b4SJens Axboe 3763ed29b0b4SJens Axboe if (opcode == IORING_OP_READ_FIXED || opcode == IORING_OP_WRITE_FIXED) { 37643c306fb2SJens Axboe ret = io_import_fixed(req, ddir, iter, issue_flags); 3765ed29b0b4SJens Axboe if (ret) 3766ed29b0b4SJens Axboe return ERR_PTR(ret); 3767ed29b0b4SJens Axboe return NULL; 3768ed29b0b4SJens Axboe } 3769ed29b0b4SJens Axboe 37703c306fb2SJens Axboe buf = u64_to_user_ptr(rw->addr); 37713c306fb2SJens Axboe sqe_len = rw->len; 3772ed29b0b4SJens Axboe 3773ed29b0b4SJens Axboe if (opcode == IORING_OP_READ || opcode == IORING_OP_WRITE) { 3774ed29b0b4SJens Axboe if (io_do_buffer_select(req)) { 3775ed29b0b4SJens Axboe buf = io_buffer_select(req, &sqe_len, issue_flags); 3776ed29b0b4SJens Axboe if (!buf) 3777ed29b0b4SJens Axboe return ERR_PTR(-ENOBUFS); 37783c306fb2SJens Axboe rw->addr = (unsigned long) buf; 37793c306fb2SJens Axboe rw->len = sqe_len; 3780ed29b0b4SJens Axboe } 3781ed29b0b4SJens Axboe 37823c306fb2SJens Axboe ret = import_single_range(ddir, buf, sqe_len, s->fast_iov, iter); 3783ed29b0b4SJens Axboe if (ret) 3784ed29b0b4SJens Axboe return ERR_PTR(ret); 3785ed29b0b4SJens Axboe return NULL; 3786ed29b0b4SJens Axboe } 3787ed29b0b4SJens Axboe 3788ed29b0b4SJens Axboe iovec = s->fast_iov; 3789ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_SELECT) { 3790ed29b0b4SJens Axboe ret = io_iov_buffer_select(req, iovec, issue_flags); 3791ed29b0b4SJens Axboe if (ret) 3792ed29b0b4SJens Axboe return ERR_PTR(ret); 37933c306fb2SJens Axboe iov_iter_init(iter, ddir, iovec, 1, iovec->iov_len); 3794ed29b0b4SJens Axboe return NULL; 3795ed29b0b4SJens Axboe } 3796ed29b0b4SJens Axboe 37973c306fb2SJens Axboe ret = __import_iovec(ddir, buf, sqe_len, UIO_FASTIOV, &iovec, iter, 3798ed29b0b4SJens Axboe req->ctx->compat); 3799ed29b0b4SJens Axboe if (unlikely(ret < 0)) 3800ed29b0b4SJens Axboe return ERR_PTR(ret); 3801ed29b0b4SJens Axboe return iovec; 3802ed29b0b4SJens Axboe } 3803ed29b0b4SJens Axboe 3804ed29b0b4SJens Axboe static inline int io_import_iovec(int rw, struct io_kiocb *req, 3805ed29b0b4SJens Axboe struct iovec **iovec, struct io_rw_state *s, 3806ed29b0b4SJens Axboe unsigned int issue_flags) 3807ed29b0b4SJens Axboe { 3808ed29b0b4SJens Axboe *iovec = __io_import_iovec(rw, req, s, issue_flags); 3809ed29b0b4SJens Axboe if (unlikely(IS_ERR(*iovec))) 3810ed29b0b4SJens Axboe return PTR_ERR(*iovec); 3811ed29b0b4SJens Axboe 3812ed29b0b4SJens Axboe iov_iter_save_state(&s->iter, &s->iter_state); 3813ed29b0b4SJens Axboe return 0; 3814ed29b0b4SJens Axboe } 3815ed29b0b4SJens Axboe 3816ed29b0b4SJens Axboe static inline loff_t *io_kiocb_ppos(struct kiocb *kiocb) 3817ed29b0b4SJens Axboe { 3818ed29b0b4SJens Axboe return (kiocb->ki_filp->f_mode & FMODE_STREAM) ? NULL : &kiocb->ki_pos; 3819ed29b0b4SJens Axboe } 3820ed29b0b4SJens Axboe 3821ed29b0b4SJens Axboe /* 3822ed29b0b4SJens Axboe * For files that don't have ->read_iter() and ->write_iter(), handle them 3823ed29b0b4SJens Axboe * by looping over ->read() or ->write() manually. 3824ed29b0b4SJens Axboe */ 38253c306fb2SJens Axboe static ssize_t loop_rw_iter(int ddir, struct io_rw *rw, struct iov_iter *iter) 3826ed29b0b4SJens Axboe { 38273c306fb2SJens Axboe struct kiocb *kiocb = &rw->kiocb; 38283c306fb2SJens Axboe struct file *file = kiocb->ki_filp; 3829ed29b0b4SJens Axboe ssize_t ret = 0; 3830ed29b0b4SJens Axboe loff_t *ppos; 3831ed29b0b4SJens Axboe 3832ed29b0b4SJens Axboe /* 3833ed29b0b4SJens Axboe * Don't support polled IO through this interface, and we can't 3834ed29b0b4SJens Axboe * support non-blocking either. For the latter, this just causes 3835ed29b0b4SJens Axboe * the kiocb to be handled from an async context. 3836ed29b0b4SJens Axboe */ 3837ed29b0b4SJens Axboe if (kiocb->ki_flags & IOCB_HIPRI) 3838ed29b0b4SJens Axboe return -EOPNOTSUPP; 3839ed29b0b4SJens Axboe if ((kiocb->ki_flags & IOCB_NOWAIT) && 3840ed29b0b4SJens Axboe !(kiocb->ki_filp->f_flags & O_NONBLOCK)) 3841ed29b0b4SJens Axboe return -EAGAIN; 3842ed29b0b4SJens Axboe 3843ed29b0b4SJens Axboe ppos = io_kiocb_ppos(kiocb); 3844ed29b0b4SJens Axboe 3845ed29b0b4SJens Axboe while (iov_iter_count(iter)) { 3846ed29b0b4SJens Axboe struct iovec iovec; 3847ed29b0b4SJens Axboe ssize_t nr; 3848ed29b0b4SJens Axboe 3849ed29b0b4SJens Axboe if (!iov_iter_is_bvec(iter)) { 3850ed29b0b4SJens Axboe iovec = iov_iter_iovec(iter); 3851ed29b0b4SJens Axboe } else { 38523c306fb2SJens Axboe iovec.iov_base = u64_to_user_ptr(rw->addr); 38533c306fb2SJens Axboe iovec.iov_len = rw->len; 3854ed29b0b4SJens Axboe } 3855ed29b0b4SJens Axboe 38563c306fb2SJens Axboe if (ddir == READ) { 3857ed29b0b4SJens Axboe nr = file->f_op->read(file, iovec.iov_base, 3858ed29b0b4SJens Axboe iovec.iov_len, ppos); 3859ed29b0b4SJens Axboe } else { 3860ed29b0b4SJens Axboe nr = file->f_op->write(file, iovec.iov_base, 3861ed29b0b4SJens Axboe iovec.iov_len, ppos); 3862ed29b0b4SJens Axboe } 3863ed29b0b4SJens Axboe 3864ed29b0b4SJens Axboe if (nr < 0) { 3865ed29b0b4SJens Axboe if (!ret) 3866ed29b0b4SJens Axboe ret = nr; 3867ed29b0b4SJens Axboe break; 3868ed29b0b4SJens Axboe } 3869ed29b0b4SJens Axboe ret += nr; 3870ed29b0b4SJens Axboe if (!iov_iter_is_bvec(iter)) { 3871ed29b0b4SJens Axboe iov_iter_advance(iter, nr); 3872ed29b0b4SJens Axboe } else { 38733c306fb2SJens Axboe rw->addr += nr; 38743c306fb2SJens Axboe rw->len -= nr; 38753c306fb2SJens Axboe if (!rw->len) 3876ed29b0b4SJens Axboe break; 3877ed29b0b4SJens Axboe } 3878ed29b0b4SJens Axboe if (nr != iovec.iov_len) 3879ed29b0b4SJens Axboe break; 3880ed29b0b4SJens Axboe } 3881ed29b0b4SJens Axboe 3882ed29b0b4SJens Axboe return ret; 3883ed29b0b4SJens Axboe } 3884ed29b0b4SJens Axboe 3885ed29b0b4SJens Axboe static void io_req_map_rw(struct io_kiocb *req, const struct iovec *iovec, 3886ed29b0b4SJens Axboe const struct iovec *fast_iov, struct iov_iter *iter) 3887ed29b0b4SJens Axboe { 38883c306fb2SJens Axboe struct io_async_rw *io = req->async_data; 3889ed29b0b4SJens Axboe 38903c306fb2SJens Axboe memcpy(&io->s.iter, iter, sizeof(*iter)); 38913c306fb2SJens Axboe io->free_iovec = iovec; 38923c306fb2SJens Axboe io->bytes_done = 0; 3893ed29b0b4SJens Axboe /* can only be fixed buffers, no need to do anything */ 3894ed29b0b4SJens Axboe if (iov_iter_is_bvec(iter)) 3895ed29b0b4SJens Axboe return; 3896ed29b0b4SJens Axboe if (!iovec) { 3897ed29b0b4SJens Axboe unsigned iov_off = 0; 3898ed29b0b4SJens Axboe 38993c306fb2SJens Axboe io->s.iter.iov = io->s.fast_iov; 3900ed29b0b4SJens Axboe if (iter->iov != fast_iov) { 3901ed29b0b4SJens Axboe iov_off = iter->iov - fast_iov; 39023c306fb2SJens Axboe io->s.iter.iov += iov_off; 3903ed29b0b4SJens Axboe } 39043c306fb2SJens Axboe if (io->s.fast_iov != fast_iov) 39053c306fb2SJens Axboe memcpy(io->s.fast_iov + iov_off, fast_iov + iov_off, 3906ed29b0b4SJens Axboe sizeof(struct iovec) * iter->nr_segs); 3907ed29b0b4SJens Axboe } else { 3908ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 3909ed29b0b4SJens Axboe } 3910ed29b0b4SJens Axboe } 3911ed29b0b4SJens Axboe 3912ed29b0b4SJens Axboe static inline bool io_alloc_async_data(struct io_kiocb *req) 3913ed29b0b4SJens Axboe { 3914ed29b0b4SJens Axboe WARN_ON_ONCE(!io_op_defs[req->opcode].async_size); 3915ed29b0b4SJens Axboe req->async_data = kmalloc(io_op_defs[req->opcode].async_size, GFP_KERNEL); 3916ed29b0b4SJens Axboe if (req->async_data) { 3917ed29b0b4SJens Axboe req->flags |= REQ_F_ASYNC_DATA; 3918ed29b0b4SJens Axboe return false; 3919ed29b0b4SJens Axboe } 3920ed29b0b4SJens Axboe return true; 3921ed29b0b4SJens Axboe } 3922ed29b0b4SJens Axboe 3923ed29b0b4SJens Axboe static int io_setup_async_rw(struct io_kiocb *req, const struct iovec *iovec, 3924ed29b0b4SJens Axboe struct io_rw_state *s, bool force) 3925ed29b0b4SJens Axboe { 3926dc919cafSJens Axboe if (!force && !io_op_defs[req->opcode].prep_async) 3927ed29b0b4SJens Axboe return 0; 3928ed29b0b4SJens Axboe if (!req_has_async_data(req)) { 3929ed29b0b4SJens Axboe struct io_async_rw *iorw; 3930ed29b0b4SJens Axboe 3931ed29b0b4SJens Axboe if (io_alloc_async_data(req)) { 3932ed29b0b4SJens Axboe kfree(iovec); 3933ed29b0b4SJens Axboe return -ENOMEM; 3934ed29b0b4SJens Axboe } 3935ed29b0b4SJens Axboe 3936ed29b0b4SJens Axboe io_req_map_rw(req, iovec, s->fast_iov, &s->iter); 3937ed29b0b4SJens Axboe iorw = req->async_data; 3938ed29b0b4SJens Axboe /* we've copied and mapped the iter, ensure state is saved */ 3939ed29b0b4SJens Axboe iov_iter_save_state(&iorw->s.iter, &iorw->s.iter_state); 3940ed29b0b4SJens Axboe } 3941ed29b0b4SJens Axboe return 0; 3942ed29b0b4SJens Axboe } 3943ed29b0b4SJens Axboe 3944ed29b0b4SJens Axboe static inline int io_rw_prep_async(struct io_kiocb *req, int rw) 3945ed29b0b4SJens Axboe { 3946ed29b0b4SJens Axboe struct io_async_rw *iorw = req->async_data; 3947ed29b0b4SJens Axboe struct iovec *iov; 3948ed29b0b4SJens Axboe int ret; 3949ed29b0b4SJens Axboe 3950ed29b0b4SJens Axboe /* submission path, ->uring_lock should already be taken */ 3951ed29b0b4SJens Axboe ret = io_import_iovec(rw, req, &iov, &iorw->s, 0); 3952ed29b0b4SJens Axboe if (unlikely(ret < 0)) 3953ed29b0b4SJens Axboe return ret; 3954ed29b0b4SJens Axboe 3955ed29b0b4SJens Axboe iorw->bytes_done = 0; 3956ed29b0b4SJens Axboe iorw->free_iovec = iov; 3957ed29b0b4SJens Axboe if (iov) 3958ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 3959ed29b0b4SJens Axboe return 0; 3960ed29b0b4SJens Axboe } 3961ed29b0b4SJens Axboe 3962ed29b0b4SJens Axboe static int io_readv_prep_async(struct io_kiocb *req) 3963ed29b0b4SJens Axboe { 3964ed29b0b4SJens Axboe return io_rw_prep_async(req, READ); 3965ed29b0b4SJens Axboe } 3966ed29b0b4SJens Axboe 3967ed29b0b4SJens Axboe static int io_writev_prep_async(struct io_kiocb *req) 3968ed29b0b4SJens Axboe { 3969ed29b0b4SJens Axboe return io_rw_prep_async(req, WRITE); 3970ed29b0b4SJens Axboe } 3971ed29b0b4SJens Axboe 3972ed29b0b4SJens Axboe /* 3973ed29b0b4SJens Axboe * This is our waitqueue callback handler, registered through __folio_lock_async() 3974ed29b0b4SJens Axboe * when we initially tried to do the IO with the iocb armed our waitqueue. 3975ed29b0b4SJens Axboe * This gets called when the page is unlocked, and we generally expect that to 3976ed29b0b4SJens Axboe * happen when the page IO is completed and the page is now uptodate. This will 3977ed29b0b4SJens Axboe * queue a task_work based retry of the operation, attempting to copy the data 3978ed29b0b4SJens Axboe * again. If the latter fails because the page was NOT uptodate, then we will 3979ed29b0b4SJens Axboe * do a thread based blocking retry of the operation. That's the unexpected 3980ed29b0b4SJens Axboe * slow path. 3981ed29b0b4SJens Axboe */ 3982ed29b0b4SJens Axboe static int io_async_buf_func(struct wait_queue_entry *wait, unsigned mode, 3983ed29b0b4SJens Axboe int sync, void *arg) 3984ed29b0b4SJens Axboe { 3985ed29b0b4SJens Axboe struct wait_page_queue *wpq; 3986ed29b0b4SJens Axboe struct io_kiocb *req = wait->private; 39873c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 3988ed29b0b4SJens Axboe struct wait_page_key *key = arg; 3989ed29b0b4SJens Axboe 3990ed29b0b4SJens Axboe wpq = container_of(wait, struct wait_page_queue, wait); 3991ed29b0b4SJens Axboe 3992ed29b0b4SJens Axboe if (!wake_page_match(wpq, key)) 3993ed29b0b4SJens Axboe return 0; 3994ed29b0b4SJens Axboe 39953c306fb2SJens Axboe rw->kiocb.ki_flags &= ~IOCB_WAITQ; 3996ed29b0b4SJens Axboe list_del_init(&wait->entry); 3997ed29b0b4SJens Axboe io_req_task_queue(req); 3998ed29b0b4SJens Axboe return 1; 3999ed29b0b4SJens Axboe } 4000ed29b0b4SJens Axboe 4001ed29b0b4SJens Axboe /* 4002ed29b0b4SJens Axboe * This controls whether a given IO request should be armed for async page 4003ed29b0b4SJens Axboe * based retry. If we return false here, the request is handed to the async 4004ed29b0b4SJens Axboe * worker threads for retry. If we're doing buffered reads on a regular file, 4005ed29b0b4SJens Axboe * we prepare a private wait_page_queue entry and retry the operation. This 4006ed29b0b4SJens Axboe * will either succeed because the page is now uptodate and unlocked, or it 4007ed29b0b4SJens Axboe * will register a callback when the page is unlocked at IO completion. Through 4008ed29b0b4SJens Axboe * that callback, io_uring uses task_work to setup a retry of the operation. 4009ed29b0b4SJens Axboe * That retry will attempt the buffered read again. The retry will generally 4010ed29b0b4SJens Axboe * succeed, or in rare cases where it fails, we then fall back to using the 4011ed29b0b4SJens Axboe * async worker threads for a blocking retry. 4012ed29b0b4SJens Axboe */ 4013ed29b0b4SJens Axboe static bool io_rw_should_retry(struct io_kiocb *req) 4014ed29b0b4SJens Axboe { 40153c306fb2SJens Axboe struct io_async_rw *io = req->async_data; 40163c306fb2SJens Axboe struct wait_page_queue *wait = &io->wpq; 40173c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 40183c306fb2SJens Axboe struct kiocb *kiocb = &rw->kiocb; 4019ed29b0b4SJens Axboe 4020ed29b0b4SJens Axboe /* never retry for NOWAIT, we just complete with -EAGAIN */ 4021ed29b0b4SJens Axboe if (req->flags & REQ_F_NOWAIT) 4022ed29b0b4SJens Axboe return false; 4023ed29b0b4SJens Axboe 4024ed29b0b4SJens Axboe /* Only for buffered IO */ 4025ed29b0b4SJens Axboe if (kiocb->ki_flags & (IOCB_DIRECT | IOCB_HIPRI)) 4026ed29b0b4SJens Axboe return false; 4027ed29b0b4SJens Axboe 4028ed29b0b4SJens Axboe /* 4029ed29b0b4SJens Axboe * just use poll if we can, and don't attempt if the fs doesn't 4030ed29b0b4SJens Axboe * support callback based unlocks 4031ed29b0b4SJens Axboe */ 4032ed29b0b4SJens Axboe if (file_can_poll(req->file) || !(req->file->f_mode & FMODE_BUF_RASYNC)) 4033ed29b0b4SJens Axboe return false; 4034ed29b0b4SJens Axboe 4035ed29b0b4SJens Axboe wait->wait.func = io_async_buf_func; 4036ed29b0b4SJens Axboe wait->wait.private = req; 4037ed29b0b4SJens Axboe wait->wait.flags = 0; 4038ed29b0b4SJens Axboe INIT_LIST_HEAD(&wait->wait.entry); 4039ed29b0b4SJens Axboe kiocb->ki_flags |= IOCB_WAITQ; 4040ed29b0b4SJens Axboe kiocb->ki_flags &= ~IOCB_NOWAIT; 4041ed29b0b4SJens Axboe kiocb->ki_waitq = wait; 4042ed29b0b4SJens Axboe return true; 4043ed29b0b4SJens Axboe } 4044ed29b0b4SJens Axboe 40453c306fb2SJens Axboe static inline int io_iter_do_read(struct io_rw *rw, struct iov_iter *iter) 4046ed29b0b4SJens Axboe { 40473c306fb2SJens Axboe struct file *file = rw->kiocb.ki_filp; 40483c306fb2SJens Axboe 40493c306fb2SJens Axboe if (likely(file->f_op->read_iter)) 40503c306fb2SJens Axboe return call_read_iter(file, &rw->kiocb, iter); 40513c306fb2SJens Axboe else if (file->f_op->read) 40523c306fb2SJens Axboe return loop_rw_iter(READ, rw, iter); 4053ed29b0b4SJens Axboe else 4054ed29b0b4SJens Axboe return -EINVAL; 4055ed29b0b4SJens Axboe } 4056ed29b0b4SJens Axboe 4057ed29b0b4SJens Axboe static bool need_read_all(struct io_kiocb *req) 4058ed29b0b4SJens Axboe { 4059ed29b0b4SJens Axboe return req->flags & REQ_F_ISREG || 4060ed29b0b4SJens Axboe S_ISBLK(file_inode(req->file)->i_mode); 4061ed29b0b4SJens Axboe } 4062ed29b0b4SJens Axboe 4063ed29b0b4SJens Axboe static int io_rw_init_file(struct io_kiocb *req, fmode_t mode) 4064ed29b0b4SJens Axboe { 40653c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 40663c306fb2SJens Axboe struct kiocb *kiocb = &rw->kiocb; 4067ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 4068ed29b0b4SJens Axboe struct file *file = req->file; 4069ed29b0b4SJens Axboe int ret; 4070ed29b0b4SJens Axboe 4071ed29b0b4SJens Axboe if (unlikely(!file || !(file->f_mode & mode))) 4072ed29b0b4SJens Axboe return -EBADF; 4073ed29b0b4SJens Axboe 4074ed29b0b4SJens Axboe if (!io_req_ffs_set(req)) 4075ed29b0b4SJens Axboe req->flags |= io_file_get_flags(file) << REQ_F_SUPPORT_NOWAIT_BIT; 4076ed29b0b4SJens Axboe 4077ed29b0b4SJens Axboe kiocb->ki_flags = iocb_flags(file); 40783c306fb2SJens Axboe ret = kiocb_set_rw_flags(kiocb, rw->flags); 4079ed29b0b4SJens Axboe if (unlikely(ret)) 4080ed29b0b4SJens Axboe return ret; 4081ed29b0b4SJens Axboe 4082ed29b0b4SJens Axboe /* 4083ed29b0b4SJens Axboe * If the file is marked O_NONBLOCK, still allow retry for it if it 4084ed29b0b4SJens Axboe * supports async. Otherwise it's impossible to use O_NONBLOCK files 4085ed29b0b4SJens Axboe * reliably. If not, or it IOCB_NOWAIT is set, don't retry. 4086ed29b0b4SJens Axboe */ 4087ed29b0b4SJens Axboe if ((kiocb->ki_flags & IOCB_NOWAIT) || 4088ed29b0b4SJens Axboe ((file->f_flags & O_NONBLOCK) && !io_file_supports_nowait(req))) 4089ed29b0b4SJens Axboe req->flags |= REQ_F_NOWAIT; 4090ed29b0b4SJens Axboe 4091ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL) { 4092ed29b0b4SJens Axboe if (!(kiocb->ki_flags & IOCB_DIRECT) || !file->f_op->iopoll) 4093ed29b0b4SJens Axboe return -EOPNOTSUPP; 4094ed29b0b4SJens Axboe 4095ed29b0b4SJens Axboe kiocb->private = NULL; 4096ed29b0b4SJens Axboe kiocb->ki_flags |= IOCB_HIPRI | IOCB_ALLOC_CACHE; 4097ed29b0b4SJens Axboe kiocb->ki_complete = io_complete_rw_iopoll; 4098ed29b0b4SJens Axboe req->iopoll_completed = 0; 4099ed29b0b4SJens Axboe } else { 4100ed29b0b4SJens Axboe if (kiocb->ki_flags & IOCB_HIPRI) 4101ed29b0b4SJens Axboe return -EINVAL; 4102ed29b0b4SJens Axboe kiocb->ki_complete = io_complete_rw; 4103ed29b0b4SJens Axboe } 4104ed29b0b4SJens Axboe 4105ed29b0b4SJens Axboe return 0; 4106ed29b0b4SJens Axboe } 4107ed29b0b4SJens Axboe 4108ed29b0b4SJens Axboe static int io_read(struct io_kiocb *req, unsigned int issue_flags) 4109ed29b0b4SJens Axboe { 41103c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 4111ed29b0b4SJens Axboe struct io_rw_state __s, *s = &__s; 4112ed29b0b4SJens Axboe struct iovec *iovec; 41133c306fb2SJens Axboe struct kiocb *kiocb = &rw->kiocb; 4114ed29b0b4SJens Axboe bool force_nonblock = issue_flags & IO_URING_F_NONBLOCK; 41153c306fb2SJens Axboe struct io_async_rw *io; 4116ed29b0b4SJens Axboe ssize_t ret, ret2; 4117ed29b0b4SJens Axboe loff_t *ppos; 4118ed29b0b4SJens Axboe 4119ed29b0b4SJens Axboe if (!req_has_async_data(req)) { 4120ed29b0b4SJens Axboe ret = io_import_iovec(READ, req, &iovec, s, issue_flags); 4121ed29b0b4SJens Axboe if (unlikely(ret < 0)) 4122ed29b0b4SJens Axboe return ret; 4123ed29b0b4SJens Axboe } else { 41243c306fb2SJens Axboe io = req->async_data; 41253c306fb2SJens Axboe s = &io->s; 4126ed29b0b4SJens Axboe 4127ed29b0b4SJens Axboe /* 4128ed29b0b4SJens Axboe * Safe and required to re-import if we're using provided 4129ed29b0b4SJens Axboe * buffers, as we dropped the selected one before retry. 4130ed29b0b4SJens Axboe */ 4131ed29b0b4SJens Axboe if (io_do_buffer_select(req)) { 4132ed29b0b4SJens Axboe ret = io_import_iovec(READ, req, &iovec, s, issue_flags); 4133ed29b0b4SJens Axboe if (unlikely(ret < 0)) 4134ed29b0b4SJens Axboe return ret; 4135ed29b0b4SJens Axboe } 4136ed29b0b4SJens Axboe 4137ed29b0b4SJens Axboe /* 4138ed29b0b4SJens Axboe * We come here from an earlier attempt, restore our state to 4139ed29b0b4SJens Axboe * match in case it doesn't. It's cheap enough that we don't 4140ed29b0b4SJens Axboe * need to make this conditional. 4141ed29b0b4SJens Axboe */ 4142ed29b0b4SJens Axboe iov_iter_restore(&s->iter, &s->iter_state); 4143ed29b0b4SJens Axboe iovec = NULL; 4144ed29b0b4SJens Axboe } 4145ed29b0b4SJens Axboe ret = io_rw_init_file(req, FMODE_READ); 4146ed29b0b4SJens Axboe if (unlikely(ret)) { 4147ed29b0b4SJens Axboe kfree(iovec); 4148ed29b0b4SJens Axboe return ret; 4149ed29b0b4SJens Axboe } 4150ed29b0b4SJens Axboe req->cqe.res = iov_iter_count(&s->iter); 4151ed29b0b4SJens Axboe 4152ed29b0b4SJens Axboe if (force_nonblock) { 4153ed29b0b4SJens Axboe /* If the file doesn't support async, just async punt */ 4154ed29b0b4SJens Axboe if (unlikely(!io_file_supports_nowait(req))) { 4155ed29b0b4SJens Axboe ret = io_setup_async_rw(req, iovec, s, true); 4156ed29b0b4SJens Axboe return ret ?: -EAGAIN; 4157ed29b0b4SJens Axboe } 4158ed29b0b4SJens Axboe kiocb->ki_flags |= IOCB_NOWAIT; 4159ed29b0b4SJens Axboe } else { 4160ed29b0b4SJens Axboe /* Ensure we clear previously set non-block flag */ 4161ed29b0b4SJens Axboe kiocb->ki_flags &= ~IOCB_NOWAIT; 4162ed29b0b4SJens Axboe } 4163ed29b0b4SJens Axboe 4164ed29b0b4SJens Axboe ppos = io_kiocb_update_pos(req); 4165ed29b0b4SJens Axboe 4166ed29b0b4SJens Axboe ret = rw_verify_area(READ, req->file, ppos, req->cqe.res); 4167ed29b0b4SJens Axboe if (unlikely(ret)) { 4168ed29b0b4SJens Axboe kfree(iovec); 4169ed29b0b4SJens Axboe return ret; 4170ed29b0b4SJens Axboe } 4171ed29b0b4SJens Axboe 41723c306fb2SJens Axboe ret = io_iter_do_read(rw, &s->iter); 4173ed29b0b4SJens Axboe 4174ed29b0b4SJens Axboe if (ret == -EAGAIN || (req->flags & REQ_F_REISSUE)) { 4175ed29b0b4SJens Axboe req->flags &= ~REQ_F_REISSUE; 4176ed29b0b4SJens Axboe /* if we can poll, just do that */ 4177ed29b0b4SJens Axboe if (req->opcode == IORING_OP_READ && file_can_poll(req->file)) 4178ed29b0b4SJens Axboe return -EAGAIN; 4179ed29b0b4SJens Axboe /* IOPOLL retry should happen for io-wq threads */ 4180ed29b0b4SJens Axboe if (!force_nonblock && !(req->ctx->flags & IORING_SETUP_IOPOLL)) 4181ed29b0b4SJens Axboe goto done; 4182ed29b0b4SJens Axboe /* no retry on NONBLOCK nor RWF_NOWAIT */ 4183ed29b0b4SJens Axboe if (req->flags & REQ_F_NOWAIT) 4184ed29b0b4SJens Axboe goto done; 4185ed29b0b4SJens Axboe ret = 0; 4186ed29b0b4SJens Axboe } else if (ret == -EIOCBQUEUED) { 4187ed29b0b4SJens Axboe goto out_free; 4188ed29b0b4SJens Axboe } else if (ret == req->cqe.res || ret <= 0 || !force_nonblock || 4189ed29b0b4SJens Axboe (req->flags & REQ_F_NOWAIT) || !need_read_all(req)) { 4190ed29b0b4SJens Axboe /* read all, failed, already did sync or don't want to retry */ 4191ed29b0b4SJens Axboe goto done; 4192ed29b0b4SJens Axboe } 4193ed29b0b4SJens Axboe 4194ed29b0b4SJens Axboe /* 4195ed29b0b4SJens Axboe * Don't depend on the iter state matching what was consumed, or being 4196ed29b0b4SJens Axboe * untouched in case of error. Restore it and we'll advance it 4197ed29b0b4SJens Axboe * manually if we need to. 4198ed29b0b4SJens Axboe */ 4199ed29b0b4SJens Axboe iov_iter_restore(&s->iter, &s->iter_state); 4200ed29b0b4SJens Axboe 4201ed29b0b4SJens Axboe ret2 = io_setup_async_rw(req, iovec, s, true); 4202ed29b0b4SJens Axboe if (ret2) 4203ed29b0b4SJens Axboe return ret2; 4204ed29b0b4SJens Axboe 4205ed29b0b4SJens Axboe iovec = NULL; 42063c306fb2SJens Axboe io = req->async_data; 42073c306fb2SJens Axboe s = &io->s; 4208ed29b0b4SJens Axboe /* 4209ed29b0b4SJens Axboe * Now use our persistent iterator and state, if we aren't already. 4210ed29b0b4SJens Axboe * We've restored and mapped the iter to match. 4211ed29b0b4SJens Axboe */ 4212ed29b0b4SJens Axboe 4213ed29b0b4SJens Axboe do { 4214ed29b0b4SJens Axboe /* 4215ed29b0b4SJens Axboe * We end up here because of a partial read, either from 4216ed29b0b4SJens Axboe * above or inside this loop. Advance the iter by the bytes 4217ed29b0b4SJens Axboe * that were consumed. 4218ed29b0b4SJens Axboe */ 4219ed29b0b4SJens Axboe iov_iter_advance(&s->iter, ret); 4220ed29b0b4SJens Axboe if (!iov_iter_count(&s->iter)) 4221ed29b0b4SJens Axboe break; 42223c306fb2SJens Axboe io->bytes_done += ret; 4223ed29b0b4SJens Axboe iov_iter_save_state(&s->iter, &s->iter_state); 4224ed29b0b4SJens Axboe 4225ed29b0b4SJens Axboe /* if we can retry, do so with the callbacks armed */ 4226ed29b0b4SJens Axboe if (!io_rw_should_retry(req)) { 4227ed29b0b4SJens Axboe kiocb->ki_flags &= ~IOCB_WAITQ; 4228ed29b0b4SJens Axboe return -EAGAIN; 4229ed29b0b4SJens Axboe } 4230ed29b0b4SJens Axboe 4231ed29b0b4SJens Axboe /* 4232ed29b0b4SJens Axboe * Now retry read with the IOCB_WAITQ parts set in the iocb. If 4233ed29b0b4SJens Axboe * we get -EIOCBQUEUED, then we'll get a notification when the 4234ed29b0b4SJens Axboe * desired page gets unlocked. We can also get a partial read 4235ed29b0b4SJens Axboe * here, and if we do, then just retry at the new offset. 4236ed29b0b4SJens Axboe */ 42373c306fb2SJens Axboe ret = io_iter_do_read(rw, &s->iter); 4238ed29b0b4SJens Axboe if (ret == -EIOCBQUEUED) 4239ed29b0b4SJens Axboe return 0; 4240ed29b0b4SJens Axboe /* we got some bytes, but not all. retry. */ 4241ed29b0b4SJens Axboe kiocb->ki_flags &= ~IOCB_WAITQ; 4242ed29b0b4SJens Axboe iov_iter_restore(&s->iter, &s->iter_state); 4243ed29b0b4SJens Axboe } while (ret > 0); 4244ed29b0b4SJens Axboe done: 4245ed29b0b4SJens Axboe kiocb_done(req, ret, issue_flags); 4246ed29b0b4SJens Axboe out_free: 4247ed29b0b4SJens Axboe /* it's faster to check here then delegate to kfree */ 4248ed29b0b4SJens Axboe if (iovec) 4249ed29b0b4SJens Axboe kfree(iovec); 4250ed29b0b4SJens Axboe return 0; 4251ed29b0b4SJens Axboe } 4252ed29b0b4SJens Axboe 4253ed29b0b4SJens Axboe static int io_write(struct io_kiocb *req, unsigned int issue_flags) 4254ed29b0b4SJens Axboe { 42553c306fb2SJens Axboe struct io_rw *rw = io_kiocb_to_cmd(req); 4256ed29b0b4SJens Axboe struct io_rw_state __s, *s = &__s; 4257ed29b0b4SJens Axboe struct iovec *iovec; 42583c306fb2SJens Axboe struct kiocb *kiocb = &rw->kiocb; 4259ed29b0b4SJens Axboe bool force_nonblock = issue_flags & IO_URING_F_NONBLOCK; 4260ed29b0b4SJens Axboe ssize_t ret, ret2; 4261ed29b0b4SJens Axboe loff_t *ppos; 4262ed29b0b4SJens Axboe 4263ed29b0b4SJens Axboe if (!req_has_async_data(req)) { 4264ed29b0b4SJens Axboe ret = io_import_iovec(WRITE, req, &iovec, s, issue_flags); 4265ed29b0b4SJens Axboe if (unlikely(ret < 0)) 4266ed29b0b4SJens Axboe return ret; 4267ed29b0b4SJens Axboe } else { 42683c306fb2SJens Axboe struct io_async_rw *io = req->async_data; 4269ed29b0b4SJens Axboe 42703c306fb2SJens Axboe s = &io->s; 4271ed29b0b4SJens Axboe iov_iter_restore(&s->iter, &s->iter_state); 4272ed29b0b4SJens Axboe iovec = NULL; 4273ed29b0b4SJens Axboe } 4274ed29b0b4SJens Axboe ret = io_rw_init_file(req, FMODE_WRITE); 4275ed29b0b4SJens Axboe if (unlikely(ret)) { 4276ed29b0b4SJens Axboe kfree(iovec); 4277ed29b0b4SJens Axboe return ret; 4278ed29b0b4SJens Axboe } 4279ed29b0b4SJens Axboe req->cqe.res = iov_iter_count(&s->iter); 4280ed29b0b4SJens Axboe 4281ed29b0b4SJens Axboe if (force_nonblock) { 4282ed29b0b4SJens Axboe /* If the file doesn't support async, just async punt */ 4283ed29b0b4SJens Axboe if (unlikely(!io_file_supports_nowait(req))) 4284ed29b0b4SJens Axboe goto copy_iov; 4285ed29b0b4SJens Axboe 4286ed29b0b4SJens Axboe /* file path doesn't support NOWAIT for non-direct_IO */ 4287ed29b0b4SJens Axboe if (force_nonblock && !(kiocb->ki_flags & IOCB_DIRECT) && 4288ed29b0b4SJens Axboe (req->flags & REQ_F_ISREG)) 4289ed29b0b4SJens Axboe goto copy_iov; 4290ed29b0b4SJens Axboe 4291ed29b0b4SJens Axboe kiocb->ki_flags |= IOCB_NOWAIT; 4292ed29b0b4SJens Axboe } else { 4293ed29b0b4SJens Axboe /* Ensure we clear previously set non-block flag */ 4294ed29b0b4SJens Axboe kiocb->ki_flags &= ~IOCB_NOWAIT; 4295ed29b0b4SJens Axboe } 4296ed29b0b4SJens Axboe 4297ed29b0b4SJens Axboe ppos = io_kiocb_update_pos(req); 4298ed29b0b4SJens Axboe 4299ed29b0b4SJens Axboe ret = rw_verify_area(WRITE, req->file, ppos, req->cqe.res); 4300ed29b0b4SJens Axboe if (unlikely(ret)) 4301ed29b0b4SJens Axboe goto out_free; 4302ed29b0b4SJens Axboe 4303ed29b0b4SJens Axboe /* 4304ed29b0b4SJens Axboe * Open-code file_start_write here to grab freeze protection, 4305ed29b0b4SJens Axboe * which will be released by another thread in 4306ed29b0b4SJens Axboe * io_complete_rw(). Fool lockdep by telling it the lock got 4307ed29b0b4SJens Axboe * released so that it doesn't complain about the held lock when 4308ed29b0b4SJens Axboe * we return to userspace. 4309ed29b0b4SJens Axboe */ 4310ed29b0b4SJens Axboe if (req->flags & REQ_F_ISREG) { 4311ed29b0b4SJens Axboe sb_start_write(file_inode(req->file)->i_sb); 4312ed29b0b4SJens Axboe __sb_writers_release(file_inode(req->file)->i_sb, 4313ed29b0b4SJens Axboe SB_FREEZE_WRITE); 4314ed29b0b4SJens Axboe } 4315ed29b0b4SJens Axboe kiocb->ki_flags |= IOCB_WRITE; 4316ed29b0b4SJens Axboe 4317ed29b0b4SJens Axboe if (likely(req->file->f_op->write_iter)) 4318ed29b0b4SJens Axboe ret2 = call_write_iter(req->file, kiocb, &s->iter); 4319ed29b0b4SJens Axboe else if (req->file->f_op->write) 43203c306fb2SJens Axboe ret2 = loop_rw_iter(WRITE, rw, &s->iter); 4321ed29b0b4SJens Axboe else 4322ed29b0b4SJens Axboe ret2 = -EINVAL; 4323ed29b0b4SJens Axboe 4324ed29b0b4SJens Axboe if (req->flags & REQ_F_REISSUE) { 4325ed29b0b4SJens Axboe req->flags &= ~REQ_F_REISSUE; 4326ed29b0b4SJens Axboe ret2 = -EAGAIN; 4327ed29b0b4SJens Axboe } 4328ed29b0b4SJens Axboe 4329ed29b0b4SJens Axboe /* 4330ed29b0b4SJens Axboe * Raw bdev writes will return -EOPNOTSUPP for IOCB_NOWAIT. Just 4331ed29b0b4SJens Axboe * retry them without IOCB_NOWAIT. 4332ed29b0b4SJens Axboe */ 4333ed29b0b4SJens Axboe if (ret2 == -EOPNOTSUPP && (kiocb->ki_flags & IOCB_NOWAIT)) 4334ed29b0b4SJens Axboe ret2 = -EAGAIN; 4335ed29b0b4SJens Axboe /* no retry on NONBLOCK nor RWF_NOWAIT */ 4336ed29b0b4SJens Axboe if (ret2 == -EAGAIN && (req->flags & REQ_F_NOWAIT)) 4337ed29b0b4SJens Axboe goto done; 4338ed29b0b4SJens Axboe if (!force_nonblock || ret2 != -EAGAIN) { 4339ed29b0b4SJens Axboe /* IOPOLL retry should happen for io-wq threads */ 4340ed29b0b4SJens Axboe if (ret2 == -EAGAIN && (req->ctx->flags & IORING_SETUP_IOPOLL)) 4341ed29b0b4SJens Axboe goto copy_iov; 4342ed29b0b4SJens Axboe done: 4343ed29b0b4SJens Axboe kiocb_done(req, ret2, issue_flags); 4344ed29b0b4SJens Axboe } else { 4345ed29b0b4SJens Axboe copy_iov: 4346ed29b0b4SJens Axboe iov_iter_restore(&s->iter, &s->iter_state); 4347ed29b0b4SJens Axboe ret = io_setup_async_rw(req, iovec, s, false); 4348ed29b0b4SJens Axboe return ret ?: -EAGAIN; 4349ed29b0b4SJens Axboe } 4350ed29b0b4SJens Axboe out_free: 4351ed29b0b4SJens Axboe /* it's reportedly faster than delegating the null check to kfree() */ 4352ed29b0b4SJens Axboe if (iovec) 4353ed29b0b4SJens Axboe kfree(iovec); 4354ed29b0b4SJens Axboe return ret; 4355ed29b0b4SJens Axboe } 4356ed29b0b4SJens Axboe 4357ed29b0b4SJens Axboe static int io_renameat_prep(struct io_kiocb *req, 4358ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4359ed29b0b4SJens Axboe { 4360bb040a21SJens Axboe struct io_rename *ren = io_kiocb_to_cmd(req); 4361ed29b0b4SJens Axboe const char __user *oldf, *newf; 4362ed29b0b4SJens Axboe 4363ed29b0b4SJens Axboe if (sqe->buf_index || sqe->splice_fd_in) 4364ed29b0b4SJens Axboe return -EINVAL; 4365ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_FIXED_FILE)) 4366ed29b0b4SJens Axboe return -EBADF; 4367ed29b0b4SJens Axboe 4368ed29b0b4SJens Axboe ren->old_dfd = READ_ONCE(sqe->fd); 4369ed29b0b4SJens Axboe oldf = u64_to_user_ptr(READ_ONCE(sqe->addr)); 4370ed29b0b4SJens Axboe newf = u64_to_user_ptr(READ_ONCE(sqe->addr2)); 4371ed29b0b4SJens Axboe ren->new_dfd = READ_ONCE(sqe->len); 4372ed29b0b4SJens Axboe ren->flags = READ_ONCE(sqe->rename_flags); 4373ed29b0b4SJens Axboe 4374ed29b0b4SJens Axboe ren->oldpath = getname(oldf); 4375ed29b0b4SJens Axboe if (IS_ERR(ren->oldpath)) 4376ed29b0b4SJens Axboe return PTR_ERR(ren->oldpath); 4377ed29b0b4SJens Axboe 4378ed29b0b4SJens Axboe ren->newpath = getname(newf); 4379ed29b0b4SJens Axboe if (IS_ERR(ren->newpath)) { 4380ed29b0b4SJens Axboe putname(ren->oldpath); 4381ed29b0b4SJens Axboe return PTR_ERR(ren->newpath); 4382ed29b0b4SJens Axboe } 4383ed29b0b4SJens Axboe 4384ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 4385ed29b0b4SJens Axboe return 0; 4386ed29b0b4SJens Axboe } 4387ed29b0b4SJens Axboe 4388ed29b0b4SJens Axboe static int io_renameat(struct io_kiocb *req, unsigned int issue_flags) 4389ed29b0b4SJens Axboe { 4390bb040a21SJens Axboe struct io_rename *ren = io_kiocb_to_cmd(req); 4391ed29b0b4SJens Axboe int ret; 4392ed29b0b4SJens Axboe 4393ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 4394ed29b0b4SJens Axboe return -EAGAIN; 4395ed29b0b4SJens Axboe 4396ed29b0b4SJens Axboe ret = do_renameat2(ren->old_dfd, ren->oldpath, ren->new_dfd, 4397ed29b0b4SJens Axboe ren->newpath, ren->flags); 4398ed29b0b4SJens Axboe 4399ed29b0b4SJens Axboe req->flags &= ~REQ_F_NEED_CLEANUP; 4400ed29b0b4SJens Axboe io_req_complete(req, ret); 4401ed29b0b4SJens Axboe return 0; 4402ed29b0b4SJens Axboe } 4403ed29b0b4SJens Axboe 4404ed29b0b4SJens Axboe static inline void __io_xattr_finish(struct io_kiocb *req) 4405ed29b0b4SJens Axboe { 4406ed29b0b4SJens Axboe struct io_xattr *ix = &req->xattr; 4407ed29b0b4SJens Axboe 4408ed29b0b4SJens Axboe if (ix->filename) 4409ed29b0b4SJens Axboe putname(ix->filename); 4410ed29b0b4SJens Axboe 4411ed29b0b4SJens Axboe kfree(ix->ctx.kname); 4412ed29b0b4SJens Axboe kvfree(ix->ctx.kvalue); 4413ed29b0b4SJens Axboe } 4414ed29b0b4SJens Axboe 4415ed29b0b4SJens Axboe static void io_xattr_finish(struct io_kiocb *req, int ret) 4416ed29b0b4SJens Axboe { 4417ed29b0b4SJens Axboe req->flags &= ~REQ_F_NEED_CLEANUP; 4418ed29b0b4SJens Axboe 4419ed29b0b4SJens Axboe __io_xattr_finish(req); 4420ed29b0b4SJens Axboe io_req_complete(req, ret); 4421ed29b0b4SJens Axboe } 4422ed29b0b4SJens Axboe 4423ed29b0b4SJens Axboe static int __io_getxattr_prep(struct io_kiocb *req, 4424ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4425ed29b0b4SJens Axboe { 4426ed29b0b4SJens Axboe struct io_xattr *ix = &req->xattr; 4427ed29b0b4SJens Axboe const char __user *name; 4428ed29b0b4SJens Axboe int ret; 4429ed29b0b4SJens Axboe 4430ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_FIXED_FILE)) 4431ed29b0b4SJens Axboe return -EBADF; 4432ed29b0b4SJens Axboe 4433ed29b0b4SJens Axboe ix->filename = NULL; 4434ed29b0b4SJens Axboe ix->ctx.kvalue = NULL; 4435ed29b0b4SJens Axboe name = u64_to_user_ptr(READ_ONCE(sqe->addr)); 4436ed29b0b4SJens Axboe ix->ctx.cvalue = u64_to_user_ptr(READ_ONCE(sqe->addr2)); 4437ed29b0b4SJens Axboe ix->ctx.size = READ_ONCE(sqe->len); 4438ed29b0b4SJens Axboe ix->ctx.flags = READ_ONCE(sqe->xattr_flags); 4439ed29b0b4SJens Axboe 4440ed29b0b4SJens Axboe if (ix->ctx.flags) 4441ed29b0b4SJens Axboe return -EINVAL; 4442ed29b0b4SJens Axboe 4443ed29b0b4SJens Axboe ix->ctx.kname = kmalloc(sizeof(*ix->ctx.kname), GFP_KERNEL); 4444ed29b0b4SJens Axboe if (!ix->ctx.kname) 4445ed29b0b4SJens Axboe return -ENOMEM; 4446ed29b0b4SJens Axboe 4447ed29b0b4SJens Axboe ret = strncpy_from_user(ix->ctx.kname->name, name, 4448ed29b0b4SJens Axboe sizeof(ix->ctx.kname->name)); 4449ed29b0b4SJens Axboe if (!ret || ret == sizeof(ix->ctx.kname->name)) 4450ed29b0b4SJens Axboe ret = -ERANGE; 4451ed29b0b4SJens Axboe if (ret < 0) { 4452ed29b0b4SJens Axboe kfree(ix->ctx.kname); 4453ed29b0b4SJens Axboe return ret; 4454ed29b0b4SJens Axboe } 4455ed29b0b4SJens Axboe 4456ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 4457ed29b0b4SJens Axboe return 0; 4458ed29b0b4SJens Axboe } 4459ed29b0b4SJens Axboe 4460ed29b0b4SJens Axboe static int io_fgetxattr_prep(struct io_kiocb *req, 4461ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4462ed29b0b4SJens Axboe { 4463ed29b0b4SJens Axboe return __io_getxattr_prep(req, sqe); 4464ed29b0b4SJens Axboe } 4465ed29b0b4SJens Axboe 4466ed29b0b4SJens Axboe static int io_getxattr_prep(struct io_kiocb *req, 4467ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4468ed29b0b4SJens Axboe { 4469ed29b0b4SJens Axboe struct io_xattr *ix = &req->xattr; 4470ed29b0b4SJens Axboe const char __user *path; 4471ed29b0b4SJens Axboe int ret; 4472ed29b0b4SJens Axboe 4473ed29b0b4SJens Axboe ret = __io_getxattr_prep(req, sqe); 4474ed29b0b4SJens Axboe if (ret) 4475ed29b0b4SJens Axboe return ret; 4476ed29b0b4SJens Axboe 4477ed29b0b4SJens Axboe path = u64_to_user_ptr(READ_ONCE(sqe->addr3)); 4478ed29b0b4SJens Axboe 4479ed29b0b4SJens Axboe ix->filename = getname_flags(path, LOOKUP_FOLLOW, NULL); 4480ed29b0b4SJens Axboe if (IS_ERR(ix->filename)) { 4481ed29b0b4SJens Axboe ret = PTR_ERR(ix->filename); 4482ed29b0b4SJens Axboe ix->filename = NULL; 4483ed29b0b4SJens Axboe } 4484ed29b0b4SJens Axboe 4485ed29b0b4SJens Axboe return ret; 4486ed29b0b4SJens Axboe } 4487ed29b0b4SJens Axboe 4488ed29b0b4SJens Axboe static int io_fgetxattr(struct io_kiocb *req, unsigned int issue_flags) 4489ed29b0b4SJens Axboe { 4490ed29b0b4SJens Axboe struct io_xattr *ix = &req->xattr; 4491ed29b0b4SJens Axboe int ret; 4492ed29b0b4SJens Axboe 4493ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 4494ed29b0b4SJens Axboe return -EAGAIN; 4495ed29b0b4SJens Axboe 4496ed29b0b4SJens Axboe ret = do_getxattr(mnt_user_ns(req->file->f_path.mnt), 4497ed29b0b4SJens Axboe req->file->f_path.dentry, 4498ed29b0b4SJens Axboe &ix->ctx); 4499ed29b0b4SJens Axboe 4500ed29b0b4SJens Axboe io_xattr_finish(req, ret); 4501ed29b0b4SJens Axboe return 0; 4502ed29b0b4SJens Axboe } 4503ed29b0b4SJens Axboe 4504ed29b0b4SJens Axboe static int io_getxattr(struct io_kiocb *req, unsigned int issue_flags) 4505ed29b0b4SJens Axboe { 4506ed29b0b4SJens Axboe struct io_xattr *ix = &req->xattr; 4507ed29b0b4SJens Axboe unsigned int lookup_flags = LOOKUP_FOLLOW; 4508ed29b0b4SJens Axboe struct path path; 4509ed29b0b4SJens Axboe int ret; 4510ed29b0b4SJens Axboe 4511ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 4512ed29b0b4SJens Axboe return -EAGAIN; 4513ed29b0b4SJens Axboe 4514ed29b0b4SJens Axboe retry: 4515ed29b0b4SJens Axboe ret = filename_lookup(AT_FDCWD, ix->filename, lookup_flags, &path, NULL); 4516ed29b0b4SJens Axboe if (!ret) { 4517ed29b0b4SJens Axboe ret = do_getxattr(mnt_user_ns(path.mnt), 4518ed29b0b4SJens Axboe path.dentry, 4519ed29b0b4SJens Axboe &ix->ctx); 4520ed29b0b4SJens Axboe 4521ed29b0b4SJens Axboe path_put(&path); 4522ed29b0b4SJens Axboe if (retry_estale(ret, lookup_flags)) { 4523ed29b0b4SJens Axboe lookup_flags |= LOOKUP_REVAL; 4524ed29b0b4SJens Axboe goto retry; 4525ed29b0b4SJens Axboe } 4526ed29b0b4SJens Axboe } 4527ed29b0b4SJens Axboe 4528ed29b0b4SJens Axboe io_xattr_finish(req, ret); 4529ed29b0b4SJens Axboe return 0; 4530ed29b0b4SJens Axboe } 4531ed29b0b4SJens Axboe 4532ed29b0b4SJens Axboe static int __io_setxattr_prep(struct io_kiocb *req, 4533ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4534ed29b0b4SJens Axboe { 4535ed29b0b4SJens Axboe struct io_xattr *ix = &req->xattr; 4536ed29b0b4SJens Axboe const char __user *name; 4537ed29b0b4SJens Axboe int ret; 4538ed29b0b4SJens Axboe 4539ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_FIXED_FILE)) 4540ed29b0b4SJens Axboe return -EBADF; 4541ed29b0b4SJens Axboe 4542ed29b0b4SJens Axboe ix->filename = NULL; 4543ed29b0b4SJens Axboe name = u64_to_user_ptr(READ_ONCE(sqe->addr)); 4544ed29b0b4SJens Axboe ix->ctx.cvalue = u64_to_user_ptr(READ_ONCE(sqe->addr2)); 4545ed29b0b4SJens Axboe ix->ctx.kvalue = NULL; 4546ed29b0b4SJens Axboe ix->ctx.size = READ_ONCE(sqe->len); 4547ed29b0b4SJens Axboe ix->ctx.flags = READ_ONCE(sqe->xattr_flags); 4548ed29b0b4SJens Axboe 4549ed29b0b4SJens Axboe ix->ctx.kname = kmalloc(sizeof(*ix->ctx.kname), GFP_KERNEL); 4550ed29b0b4SJens Axboe if (!ix->ctx.kname) 4551ed29b0b4SJens Axboe return -ENOMEM; 4552ed29b0b4SJens Axboe 4553ed29b0b4SJens Axboe ret = setxattr_copy(name, &ix->ctx); 4554ed29b0b4SJens Axboe if (ret) { 4555ed29b0b4SJens Axboe kfree(ix->ctx.kname); 4556ed29b0b4SJens Axboe return ret; 4557ed29b0b4SJens Axboe } 4558ed29b0b4SJens Axboe 4559ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 4560ed29b0b4SJens Axboe return 0; 4561ed29b0b4SJens Axboe } 4562ed29b0b4SJens Axboe 4563ed29b0b4SJens Axboe static int io_setxattr_prep(struct io_kiocb *req, 4564ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4565ed29b0b4SJens Axboe { 4566ed29b0b4SJens Axboe struct io_xattr *ix = &req->xattr; 4567ed29b0b4SJens Axboe const char __user *path; 4568ed29b0b4SJens Axboe int ret; 4569ed29b0b4SJens Axboe 4570ed29b0b4SJens Axboe ret = __io_setxattr_prep(req, sqe); 4571ed29b0b4SJens Axboe if (ret) 4572ed29b0b4SJens Axboe return ret; 4573ed29b0b4SJens Axboe 4574ed29b0b4SJens Axboe path = u64_to_user_ptr(READ_ONCE(sqe->addr3)); 4575ed29b0b4SJens Axboe 4576ed29b0b4SJens Axboe ix->filename = getname_flags(path, LOOKUP_FOLLOW, NULL); 4577ed29b0b4SJens Axboe if (IS_ERR(ix->filename)) { 4578ed29b0b4SJens Axboe ret = PTR_ERR(ix->filename); 4579ed29b0b4SJens Axboe ix->filename = NULL; 4580ed29b0b4SJens Axboe } 4581ed29b0b4SJens Axboe 4582ed29b0b4SJens Axboe return ret; 4583ed29b0b4SJens Axboe } 4584ed29b0b4SJens Axboe 4585ed29b0b4SJens Axboe static int io_fsetxattr_prep(struct io_kiocb *req, 4586ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4587ed29b0b4SJens Axboe { 4588ed29b0b4SJens Axboe return __io_setxattr_prep(req, sqe); 4589ed29b0b4SJens Axboe } 4590ed29b0b4SJens Axboe 4591ed29b0b4SJens Axboe static int __io_setxattr(struct io_kiocb *req, unsigned int issue_flags, 4592ed29b0b4SJens Axboe struct path *path) 4593ed29b0b4SJens Axboe { 4594ed29b0b4SJens Axboe struct io_xattr *ix = &req->xattr; 4595ed29b0b4SJens Axboe int ret; 4596ed29b0b4SJens Axboe 4597ed29b0b4SJens Axboe ret = mnt_want_write(path->mnt); 4598ed29b0b4SJens Axboe if (!ret) { 4599ed29b0b4SJens Axboe ret = do_setxattr(mnt_user_ns(path->mnt), path->dentry, &ix->ctx); 4600ed29b0b4SJens Axboe mnt_drop_write(path->mnt); 4601ed29b0b4SJens Axboe } 4602ed29b0b4SJens Axboe 4603ed29b0b4SJens Axboe return ret; 4604ed29b0b4SJens Axboe } 4605ed29b0b4SJens Axboe 4606ed29b0b4SJens Axboe static int io_fsetxattr(struct io_kiocb *req, unsigned int issue_flags) 4607ed29b0b4SJens Axboe { 4608ed29b0b4SJens Axboe int ret; 4609ed29b0b4SJens Axboe 4610ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 4611ed29b0b4SJens Axboe return -EAGAIN; 4612ed29b0b4SJens Axboe 4613ed29b0b4SJens Axboe ret = __io_setxattr(req, issue_flags, &req->file->f_path); 4614ed29b0b4SJens Axboe io_xattr_finish(req, ret); 4615ed29b0b4SJens Axboe 4616ed29b0b4SJens Axboe return 0; 4617ed29b0b4SJens Axboe } 4618ed29b0b4SJens Axboe 4619ed29b0b4SJens Axboe static int io_setxattr(struct io_kiocb *req, unsigned int issue_flags) 4620ed29b0b4SJens Axboe { 4621ed29b0b4SJens Axboe struct io_xattr *ix = &req->xattr; 4622ed29b0b4SJens Axboe unsigned int lookup_flags = LOOKUP_FOLLOW; 4623ed29b0b4SJens Axboe struct path path; 4624ed29b0b4SJens Axboe int ret; 4625ed29b0b4SJens Axboe 4626ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 4627ed29b0b4SJens Axboe return -EAGAIN; 4628ed29b0b4SJens Axboe 4629ed29b0b4SJens Axboe retry: 4630ed29b0b4SJens Axboe ret = filename_lookup(AT_FDCWD, ix->filename, lookup_flags, &path, NULL); 4631ed29b0b4SJens Axboe if (!ret) { 4632ed29b0b4SJens Axboe ret = __io_setxattr(req, issue_flags, &path); 4633ed29b0b4SJens Axboe path_put(&path); 4634ed29b0b4SJens Axboe if (retry_estale(ret, lookup_flags)) { 4635ed29b0b4SJens Axboe lookup_flags |= LOOKUP_REVAL; 4636ed29b0b4SJens Axboe goto retry; 4637ed29b0b4SJens Axboe } 4638ed29b0b4SJens Axboe } 4639ed29b0b4SJens Axboe 4640ed29b0b4SJens Axboe io_xattr_finish(req, ret); 4641ed29b0b4SJens Axboe return 0; 4642ed29b0b4SJens Axboe } 4643ed29b0b4SJens Axboe 4644ed29b0b4SJens Axboe static int io_unlinkat_prep(struct io_kiocb *req, 4645ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4646ed29b0b4SJens Axboe { 4647bb040a21SJens Axboe struct io_unlink *un = io_kiocb_to_cmd(req); 4648ed29b0b4SJens Axboe const char __user *fname; 4649ed29b0b4SJens Axboe 4650ed29b0b4SJens Axboe if (sqe->off || sqe->len || sqe->buf_index || sqe->splice_fd_in) 4651ed29b0b4SJens Axboe return -EINVAL; 4652ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_FIXED_FILE)) 4653ed29b0b4SJens Axboe return -EBADF; 4654ed29b0b4SJens Axboe 4655ed29b0b4SJens Axboe un->dfd = READ_ONCE(sqe->fd); 4656ed29b0b4SJens Axboe 4657ed29b0b4SJens Axboe un->flags = READ_ONCE(sqe->unlink_flags); 4658ed29b0b4SJens Axboe if (un->flags & ~AT_REMOVEDIR) 4659ed29b0b4SJens Axboe return -EINVAL; 4660ed29b0b4SJens Axboe 4661ed29b0b4SJens Axboe fname = u64_to_user_ptr(READ_ONCE(sqe->addr)); 4662ed29b0b4SJens Axboe un->filename = getname(fname); 4663ed29b0b4SJens Axboe if (IS_ERR(un->filename)) 4664ed29b0b4SJens Axboe return PTR_ERR(un->filename); 4665ed29b0b4SJens Axboe 4666ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 4667ed29b0b4SJens Axboe return 0; 4668ed29b0b4SJens Axboe } 4669ed29b0b4SJens Axboe 4670ed29b0b4SJens Axboe static int io_unlinkat(struct io_kiocb *req, unsigned int issue_flags) 4671ed29b0b4SJens Axboe { 4672bb040a21SJens Axboe struct io_unlink *un = io_kiocb_to_cmd(req); 4673ed29b0b4SJens Axboe int ret; 4674ed29b0b4SJens Axboe 4675ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 4676ed29b0b4SJens Axboe return -EAGAIN; 4677ed29b0b4SJens Axboe 4678ed29b0b4SJens Axboe if (un->flags & AT_REMOVEDIR) 4679ed29b0b4SJens Axboe ret = do_rmdir(un->dfd, un->filename); 4680ed29b0b4SJens Axboe else 4681ed29b0b4SJens Axboe ret = do_unlinkat(un->dfd, un->filename); 4682ed29b0b4SJens Axboe 4683ed29b0b4SJens Axboe req->flags &= ~REQ_F_NEED_CLEANUP; 4684ed29b0b4SJens Axboe io_req_complete(req, ret); 4685ed29b0b4SJens Axboe return 0; 4686ed29b0b4SJens Axboe } 4687ed29b0b4SJens Axboe 4688ed29b0b4SJens Axboe static int io_mkdirat_prep(struct io_kiocb *req, 4689ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4690ed29b0b4SJens Axboe { 4691bb040a21SJens Axboe struct io_mkdir *mkd = io_kiocb_to_cmd(req); 4692ed29b0b4SJens Axboe const char __user *fname; 4693ed29b0b4SJens Axboe 4694ed29b0b4SJens Axboe if (sqe->off || sqe->rw_flags || sqe->buf_index || sqe->splice_fd_in) 4695ed29b0b4SJens Axboe return -EINVAL; 4696ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_FIXED_FILE)) 4697ed29b0b4SJens Axboe return -EBADF; 4698ed29b0b4SJens Axboe 4699ed29b0b4SJens Axboe mkd->dfd = READ_ONCE(sqe->fd); 4700ed29b0b4SJens Axboe mkd->mode = READ_ONCE(sqe->len); 4701ed29b0b4SJens Axboe 4702ed29b0b4SJens Axboe fname = u64_to_user_ptr(READ_ONCE(sqe->addr)); 4703ed29b0b4SJens Axboe mkd->filename = getname(fname); 4704ed29b0b4SJens Axboe if (IS_ERR(mkd->filename)) 4705ed29b0b4SJens Axboe return PTR_ERR(mkd->filename); 4706ed29b0b4SJens Axboe 4707ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 4708ed29b0b4SJens Axboe return 0; 4709ed29b0b4SJens Axboe } 4710ed29b0b4SJens Axboe 4711ed29b0b4SJens Axboe static int io_mkdirat(struct io_kiocb *req, unsigned int issue_flags) 4712ed29b0b4SJens Axboe { 4713bb040a21SJens Axboe struct io_mkdir *mkd = io_kiocb_to_cmd(req); 4714ed29b0b4SJens Axboe int ret; 4715ed29b0b4SJens Axboe 4716ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 4717ed29b0b4SJens Axboe return -EAGAIN; 4718ed29b0b4SJens Axboe 4719ed29b0b4SJens Axboe ret = do_mkdirat(mkd->dfd, mkd->filename, mkd->mode); 4720ed29b0b4SJens Axboe 4721ed29b0b4SJens Axboe req->flags &= ~REQ_F_NEED_CLEANUP; 4722ed29b0b4SJens Axboe io_req_complete(req, ret); 4723ed29b0b4SJens Axboe return 0; 4724ed29b0b4SJens Axboe } 4725ed29b0b4SJens Axboe 4726ed29b0b4SJens Axboe static int io_symlinkat_prep(struct io_kiocb *req, 4727ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4728ed29b0b4SJens Axboe { 4729bb040a21SJens Axboe struct io_symlink *sl = io_kiocb_to_cmd(req); 4730ed29b0b4SJens Axboe const char __user *oldpath, *newpath; 4731ed29b0b4SJens Axboe 4732ed29b0b4SJens Axboe if (sqe->len || sqe->rw_flags || sqe->buf_index || sqe->splice_fd_in) 4733ed29b0b4SJens Axboe return -EINVAL; 4734ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_FIXED_FILE)) 4735ed29b0b4SJens Axboe return -EBADF; 4736ed29b0b4SJens Axboe 4737ed29b0b4SJens Axboe sl->new_dfd = READ_ONCE(sqe->fd); 4738ed29b0b4SJens Axboe oldpath = u64_to_user_ptr(READ_ONCE(sqe->addr)); 4739ed29b0b4SJens Axboe newpath = u64_to_user_ptr(READ_ONCE(sqe->addr2)); 4740ed29b0b4SJens Axboe 4741ed29b0b4SJens Axboe sl->oldpath = getname(oldpath); 4742ed29b0b4SJens Axboe if (IS_ERR(sl->oldpath)) 4743ed29b0b4SJens Axboe return PTR_ERR(sl->oldpath); 4744ed29b0b4SJens Axboe 4745ed29b0b4SJens Axboe sl->newpath = getname(newpath); 4746ed29b0b4SJens Axboe if (IS_ERR(sl->newpath)) { 4747ed29b0b4SJens Axboe putname(sl->oldpath); 4748ed29b0b4SJens Axboe return PTR_ERR(sl->newpath); 4749ed29b0b4SJens Axboe } 4750ed29b0b4SJens Axboe 4751ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 4752ed29b0b4SJens Axboe return 0; 4753ed29b0b4SJens Axboe } 4754ed29b0b4SJens Axboe 4755ed29b0b4SJens Axboe static int io_symlinkat(struct io_kiocb *req, unsigned int issue_flags) 4756ed29b0b4SJens Axboe { 4757bb040a21SJens Axboe struct io_symlink *sl = io_kiocb_to_cmd(req); 4758ed29b0b4SJens Axboe int ret; 4759ed29b0b4SJens Axboe 4760ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 4761ed29b0b4SJens Axboe return -EAGAIN; 4762ed29b0b4SJens Axboe 4763ed29b0b4SJens Axboe ret = do_symlinkat(sl->oldpath, sl->new_dfd, sl->newpath); 4764ed29b0b4SJens Axboe 4765ed29b0b4SJens Axboe req->flags &= ~REQ_F_NEED_CLEANUP; 4766ed29b0b4SJens Axboe io_req_complete(req, ret); 4767ed29b0b4SJens Axboe return 0; 4768ed29b0b4SJens Axboe } 4769ed29b0b4SJens Axboe 4770ed29b0b4SJens Axboe static int io_linkat_prep(struct io_kiocb *req, 4771ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4772ed29b0b4SJens Axboe { 4773bb040a21SJens Axboe struct io_hardlink *lnk = io_kiocb_to_cmd(req); 4774ed29b0b4SJens Axboe const char __user *oldf, *newf; 4775ed29b0b4SJens Axboe 4776ed29b0b4SJens Axboe if (sqe->rw_flags || sqe->buf_index || sqe->splice_fd_in) 4777ed29b0b4SJens Axboe return -EINVAL; 4778ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_FIXED_FILE)) 4779ed29b0b4SJens Axboe return -EBADF; 4780ed29b0b4SJens Axboe 4781ed29b0b4SJens Axboe lnk->old_dfd = READ_ONCE(sqe->fd); 4782ed29b0b4SJens Axboe lnk->new_dfd = READ_ONCE(sqe->len); 4783ed29b0b4SJens Axboe oldf = u64_to_user_ptr(READ_ONCE(sqe->addr)); 4784ed29b0b4SJens Axboe newf = u64_to_user_ptr(READ_ONCE(sqe->addr2)); 4785ed29b0b4SJens Axboe lnk->flags = READ_ONCE(sqe->hardlink_flags); 4786ed29b0b4SJens Axboe 4787ed29b0b4SJens Axboe lnk->oldpath = getname(oldf); 4788ed29b0b4SJens Axboe if (IS_ERR(lnk->oldpath)) 4789ed29b0b4SJens Axboe return PTR_ERR(lnk->oldpath); 4790ed29b0b4SJens Axboe 4791ed29b0b4SJens Axboe lnk->newpath = getname(newf); 4792ed29b0b4SJens Axboe if (IS_ERR(lnk->newpath)) { 4793ed29b0b4SJens Axboe putname(lnk->oldpath); 4794ed29b0b4SJens Axboe return PTR_ERR(lnk->newpath); 4795ed29b0b4SJens Axboe } 4796ed29b0b4SJens Axboe 4797ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 4798ed29b0b4SJens Axboe return 0; 4799ed29b0b4SJens Axboe } 4800ed29b0b4SJens Axboe 4801ed29b0b4SJens Axboe static int io_linkat(struct io_kiocb *req, unsigned int issue_flags) 4802ed29b0b4SJens Axboe { 4803bb040a21SJens Axboe struct io_hardlink *lnk = io_kiocb_to_cmd(req); 4804ed29b0b4SJens Axboe int ret; 4805ed29b0b4SJens Axboe 4806ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 4807ed29b0b4SJens Axboe return -EAGAIN; 4808ed29b0b4SJens Axboe 4809ed29b0b4SJens Axboe ret = do_linkat(lnk->old_dfd, lnk->oldpath, lnk->new_dfd, 4810ed29b0b4SJens Axboe lnk->newpath, lnk->flags); 4811ed29b0b4SJens Axboe 4812ed29b0b4SJens Axboe req->flags &= ~REQ_F_NEED_CLEANUP; 4813ed29b0b4SJens Axboe io_req_complete(req, ret); 4814ed29b0b4SJens Axboe return 0; 4815ed29b0b4SJens Axboe } 4816ed29b0b4SJens Axboe 4817ed29b0b4SJens Axboe static void io_uring_cmd_work(struct io_kiocb *req, bool *locked) 4818ed29b0b4SJens Axboe { 4819ed29b0b4SJens Axboe req->uring_cmd.task_work_cb(&req->uring_cmd); 4820ed29b0b4SJens Axboe } 4821ed29b0b4SJens Axboe 4822ed29b0b4SJens Axboe void io_uring_cmd_complete_in_task(struct io_uring_cmd *ioucmd, 4823ed29b0b4SJens Axboe void (*task_work_cb)(struct io_uring_cmd *)) 4824ed29b0b4SJens Axboe { 4825ed29b0b4SJens Axboe struct io_kiocb *req = container_of(ioucmd, struct io_kiocb, uring_cmd); 4826ed29b0b4SJens Axboe 4827ed29b0b4SJens Axboe req->uring_cmd.task_work_cb = task_work_cb; 4828ed29b0b4SJens Axboe req->io_task_work.func = io_uring_cmd_work; 4829ed29b0b4SJens Axboe io_req_task_work_add(req); 4830ed29b0b4SJens Axboe } 4831ed29b0b4SJens Axboe EXPORT_SYMBOL_GPL(io_uring_cmd_complete_in_task); 4832ed29b0b4SJens Axboe 4833ed29b0b4SJens Axboe static inline void io_req_set_cqe32_extra(struct io_kiocb *req, 4834ed29b0b4SJens Axboe u64 extra1, u64 extra2) 4835ed29b0b4SJens Axboe { 4836ed29b0b4SJens Axboe req->extra1 = extra1; 4837ed29b0b4SJens Axboe req->extra2 = extra2; 4838ed29b0b4SJens Axboe req->flags |= REQ_F_CQE32_INIT; 4839ed29b0b4SJens Axboe } 4840ed29b0b4SJens Axboe 4841ed29b0b4SJens Axboe /* 4842ed29b0b4SJens Axboe * Called by consumers of io_uring_cmd, if they originally returned 4843ed29b0b4SJens Axboe * -EIOCBQUEUED upon receiving the command. 4844ed29b0b4SJens Axboe */ 4845ed29b0b4SJens Axboe void io_uring_cmd_done(struct io_uring_cmd *ioucmd, ssize_t ret, ssize_t res2) 4846ed29b0b4SJens Axboe { 4847ed29b0b4SJens Axboe struct io_kiocb *req = container_of(ioucmd, struct io_kiocb, uring_cmd); 4848ed29b0b4SJens Axboe 4849ed29b0b4SJens Axboe if (ret < 0) 4850ed29b0b4SJens Axboe req_set_fail(req); 4851ed29b0b4SJens Axboe 4852ed29b0b4SJens Axboe if (req->ctx->flags & IORING_SETUP_CQE32) 4853ed29b0b4SJens Axboe io_req_set_cqe32_extra(req, res2, 0); 4854ed29b0b4SJens Axboe io_req_complete(req, ret); 4855ed29b0b4SJens Axboe } 4856ed29b0b4SJens Axboe EXPORT_SYMBOL_GPL(io_uring_cmd_done); 4857ed29b0b4SJens Axboe 4858ed29b0b4SJens Axboe static int io_uring_cmd_prep_async(struct io_kiocb *req) 4859ed29b0b4SJens Axboe { 4860ed29b0b4SJens Axboe size_t cmd_size; 4861ed29b0b4SJens Axboe 4862ed29b0b4SJens Axboe cmd_size = uring_cmd_pdu_size(req->ctx->flags & IORING_SETUP_SQE128); 4863ed29b0b4SJens Axboe 4864ed29b0b4SJens Axboe memcpy(req->async_data, req->uring_cmd.cmd, cmd_size); 4865ed29b0b4SJens Axboe return 0; 4866ed29b0b4SJens Axboe } 4867ed29b0b4SJens Axboe 4868ed29b0b4SJens Axboe static int io_uring_cmd_prep(struct io_kiocb *req, 4869ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4870ed29b0b4SJens Axboe { 4871ed29b0b4SJens Axboe struct io_uring_cmd *ioucmd = &req->uring_cmd; 4872ed29b0b4SJens Axboe 4873ed29b0b4SJens Axboe if (sqe->rw_flags || sqe->__pad1) 4874ed29b0b4SJens Axboe return -EINVAL; 4875ed29b0b4SJens Axboe ioucmd->cmd = sqe->cmd; 4876ed29b0b4SJens Axboe ioucmd->cmd_op = READ_ONCE(sqe->cmd_op); 4877ed29b0b4SJens Axboe return 0; 4878ed29b0b4SJens Axboe } 4879ed29b0b4SJens Axboe 4880ed29b0b4SJens Axboe static int io_uring_cmd(struct io_kiocb *req, unsigned int issue_flags) 4881ed29b0b4SJens Axboe { 4882ed29b0b4SJens Axboe struct io_uring_cmd *ioucmd = &req->uring_cmd; 4883ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 4884ed29b0b4SJens Axboe struct file *file = req->file; 4885ed29b0b4SJens Axboe int ret; 4886ed29b0b4SJens Axboe 4887ed29b0b4SJens Axboe if (!req->file->f_op->uring_cmd) 4888ed29b0b4SJens Axboe return -EOPNOTSUPP; 4889ed29b0b4SJens Axboe 4890ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQE128) 4891ed29b0b4SJens Axboe issue_flags |= IO_URING_F_SQE128; 4892ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_CQE32) 4893ed29b0b4SJens Axboe issue_flags |= IO_URING_F_CQE32; 4894ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL) 4895ed29b0b4SJens Axboe issue_flags |= IO_URING_F_IOPOLL; 4896ed29b0b4SJens Axboe 4897ed29b0b4SJens Axboe if (req_has_async_data(req)) 4898ed29b0b4SJens Axboe ioucmd->cmd = req->async_data; 4899ed29b0b4SJens Axboe 4900ed29b0b4SJens Axboe ret = file->f_op->uring_cmd(ioucmd, issue_flags); 4901ed29b0b4SJens Axboe if (ret == -EAGAIN) { 4902ed29b0b4SJens Axboe if (!req_has_async_data(req)) { 4903ed29b0b4SJens Axboe if (io_alloc_async_data(req)) 4904ed29b0b4SJens Axboe return -ENOMEM; 4905ed29b0b4SJens Axboe io_uring_cmd_prep_async(req); 4906ed29b0b4SJens Axboe } 4907ed29b0b4SJens Axboe return -EAGAIN; 4908ed29b0b4SJens Axboe } 4909ed29b0b4SJens Axboe 4910ed29b0b4SJens Axboe if (ret != -EIOCBQUEUED) 4911ed29b0b4SJens Axboe io_uring_cmd_done(ioucmd, ret, 0); 4912ed29b0b4SJens Axboe return 0; 4913ed29b0b4SJens Axboe } 4914ed29b0b4SJens Axboe 4915ed29b0b4SJens Axboe static int __io_splice_prep(struct io_kiocb *req, 4916ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4917ed29b0b4SJens Axboe { 49182511d303SJens Axboe struct io_splice *sp = io_kiocb_to_cmd(req); 4919ed29b0b4SJens Axboe unsigned int valid_flags = SPLICE_F_FD_IN_FIXED | SPLICE_F_ALL; 4920ed29b0b4SJens Axboe 4921ed29b0b4SJens Axboe sp->len = READ_ONCE(sqe->len); 4922ed29b0b4SJens Axboe sp->flags = READ_ONCE(sqe->splice_flags); 4923ed29b0b4SJens Axboe if (unlikely(sp->flags & ~valid_flags)) 4924ed29b0b4SJens Axboe return -EINVAL; 4925ed29b0b4SJens Axboe sp->splice_fd_in = READ_ONCE(sqe->splice_fd_in); 4926ed29b0b4SJens Axboe return 0; 4927ed29b0b4SJens Axboe } 4928ed29b0b4SJens Axboe 4929ed29b0b4SJens Axboe static int io_tee_prep(struct io_kiocb *req, 4930ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 4931ed29b0b4SJens Axboe { 4932ed29b0b4SJens Axboe if (READ_ONCE(sqe->splice_off_in) || READ_ONCE(sqe->off)) 4933ed29b0b4SJens Axboe return -EINVAL; 4934ed29b0b4SJens Axboe return __io_splice_prep(req, sqe); 4935ed29b0b4SJens Axboe } 4936ed29b0b4SJens Axboe 4937ed29b0b4SJens Axboe static int io_tee(struct io_kiocb *req, unsigned int issue_flags) 4938ed29b0b4SJens Axboe { 49392511d303SJens Axboe struct io_splice *sp = io_kiocb_to_cmd(req); 4940ed29b0b4SJens Axboe struct file *out = sp->file_out; 4941ed29b0b4SJens Axboe unsigned int flags = sp->flags & ~SPLICE_F_FD_IN_FIXED; 4942ed29b0b4SJens Axboe struct file *in; 4943ed29b0b4SJens Axboe long ret = 0; 4944ed29b0b4SJens Axboe 4945ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 4946ed29b0b4SJens Axboe return -EAGAIN; 4947ed29b0b4SJens Axboe 4948ed29b0b4SJens Axboe if (sp->flags & SPLICE_F_FD_IN_FIXED) 4949ed29b0b4SJens Axboe in = io_file_get_fixed(req, sp->splice_fd_in, issue_flags); 4950ed29b0b4SJens Axboe else 4951ed29b0b4SJens Axboe in = io_file_get_normal(req, sp->splice_fd_in); 4952ed29b0b4SJens Axboe if (!in) { 4953ed29b0b4SJens Axboe ret = -EBADF; 4954ed29b0b4SJens Axboe goto done; 4955ed29b0b4SJens Axboe } 4956ed29b0b4SJens Axboe 4957ed29b0b4SJens Axboe if (sp->len) 4958ed29b0b4SJens Axboe ret = do_tee(in, out, sp->len, flags); 4959ed29b0b4SJens Axboe 4960ed29b0b4SJens Axboe if (!(sp->flags & SPLICE_F_FD_IN_FIXED)) 4961ed29b0b4SJens Axboe io_put_file(in); 4962ed29b0b4SJens Axboe done: 4963ed29b0b4SJens Axboe if (ret != sp->len) 4964ed29b0b4SJens Axboe req_set_fail(req); 4965ed29b0b4SJens Axboe __io_req_complete(req, 0, ret, 0); 4966ed29b0b4SJens Axboe return 0; 4967ed29b0b4SJens Axboe } 4968ed29b0b4SJens Axboe 4969ed29b0b4SJens Axboe static int io_splice_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 4970ed29b0b4SJens Axboe { 49712511d303SJens Axboe struct io_splice *sp = io_kiocb_to_cmd(req); 4972ed29b0b4SJens Axboe 4973ed29b0b4SJens Axboe sp->off_in = READ_ONCE(sqe->splice_off_in); 4974ed29b0b4SJens Axboe sp->off_out = READ_ONCE(sqe->off); 4975ed29b0b4SJens Axboe return __io_splice_prep(req, sqe); 4976ed29b0b4SJens Axboe } 4977ed29b0b4SJens Axboe 4978ed29b0b4SJens Axboe static int io_splice(struct io_kiocb *req, unsigned int issue_flags) 4979ed29b0b4SJens Axboe { 49802511d303SJens Axboe struct io_splice *sp = io_kiocb_to_cmd(req); 4981ed29b0b4SJens Axboe struct file *out = sp->file_out; 4982ed29b0b4SJens Axboe unsigned int flags = sp->flags & ~SPLICE_F_FD_IN_FIXED; 4983ed29b0b4SJens Axboe loff_t *poff_in, *poff_out; 4984ed29b0b4SJens Axboe struct file *in; 4985ed29b0b4SJens Axboe long ret = 0; 4986ed29b0b4SJens Axboe 4987ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 4988ed29b0b4SJens Axboe return -EAGAIN; 4989ed29b0b4SJens Axboe 4990ed29b0b4SJens Axboe if (sp->flags & SPLICE_F_FD_IN_FIXED) 4991ed29b0b4SJens Axboe in = io_file_get_fixed(req, sp->splice_fd_in, issue_flags); 4992ed29b0b4SJens Axboe else 4993ed29b0b4SJens Axboe in = io_file_get_normal(req, sp->splice_fd_in); 4994ed29b0b4SJens Axboe if (!in) { 4995ed29b0b4SJens Axboe ret = -EBADF; 4996ed29b0b4SJens Axboe goto done; 4997ed29b0b4SJens Axboe } 4998ed29b0b4SJens Axboe 4999ed29b0b4SJens Axboe poff_in = (sp->off_in == -1) ? NULL : &sp->off_in; 5000ed29b0b4SJens Axboe poff_out = (sp->off_out == -1) ? NULL : &sp->off_out; 5001ed29b0b4SJens Axboe 5002ed29b0b4SJens Axboe if (sp->len) 5003ed29b0b4SJens Axboe ret = do_splice(in, poff_in, out, poff_out, sp->len, flags); 5004ed29b0b4SJens Axboe 5005ed29b0b4SJens Axboe if (!(sp->flags & SPLICE_F_FD_IN_FIXED)) 5006ed29b0b4SJens Axboe io_put_file(in); 5007ed29b0b4SJens Axboe done: 5008ed29b0b4SJens Axboe if (ret != sp->len) 5009ed29b0b4SJens Axboe req_set_fail(req); 5010ed29b0b4SJens Axboe __io_req_complete(req, 0, ret, 0); 5011ed29b0b4SJens Axboe return 0; 5012ed29b0b4SJens Axboe } 5013ed29b0b4SJens Axboe 5014ed29b0b4SJens Axboe static int io_nop_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 5015ed29b0b4SJens Axboe { 5016ed29b0b4SJens Axboe return 0; 5017ed29b0b4SJens Axboe } 5018ed29b0b4SJens Axboe 5019ed29b0b4SJens Axboe /* 5020ed29b0b4SJens Axboe * IORING_OP_NOP just posts a completion event, nothing else. 5021ed29b0b4SJens Axboe */ 5022ed29b0b4SJens Axboe static int io_nop(struct io_kiocb *req, unsigned int issue_flags) 5023ed29b0b4SJens Axboe { 5024ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, 0, 0); 5025ed29b0b4SJens Axboe return 0; 5026ed29b0b4SJens Axboe } 5027ed29b0b4SJens Axboe 5028ed29b0b4SJens Axboe static int io_msg_ring_prep(struct io_kiocb *req, 5029ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 5030ed29b0b4SJens Axboe { 5031*c1ee5595SJens Axboe struct io_msg *msg = io_kiocb_to_cmd(req); 5032*c1ee5595SJens Axboe 5033ed29b0b4SJens Axboe if (unlikely(sqe->addr || sqe->rw_flags || sqe->splice_fd_in || 5034ed29b0b4SJens Axboe sqe->buf_index || sqe->personality)) 5035ed29b0b4SJens Axboe return -EINVAL; 5036ed29b0b4SJens Axboe 5037*c1ee5595SJens Axboe msg->user_data = READ_ONCE(sqe->off); 5038*c1ee5595SJens Axboe msg->len = READ_ONCE(sqe->len); 5039ed29b0b4SJens Axboe return 0; 5040ed29b0b4SJens Axboe } 5041ed29b0b4SJens Axboe 5042ed29b0b4SJens Axboe static int io_msg_ring(struct io_kiocb *req, unsigned int issue_flags) 5043ed29b0b4SJens Axboe { 5044*c1ee5595SJens Axboe struct io_msg *msg = io_kiocb_to_cmd(req); 5045ed29b0b4SJens Axboe struct io_ring_ctx *target_ctx; 5046ed29b0b4SJens Axboe bool filled; 5047ed29b0b4SJens Axboe int ret; 5048ed29b0b4SJens Axboe 5049ed29b0b4SJens Axboe ret = -EBADFD; 5050ed29b0b4SJens Axboe if (req->file->f_op != &io_uring_fops) 5051ed29b0b4SJens Axboe goto done; 5052ed29b0b4SJens Axboe 5053ed29b0b4SJens Axboe ret = -EOVERFLOW; 5054ed29b0b4SJens Axboe target_ctx = req->file->private_data; 5055ed29b0b4SJens Axboe 5056ed29b0b4SJens Axboe spin_lock(&target_ctx->completion_lock); 5057ed29b0b4SJens Axboe filled = io_fill_cqe_aux(target_ctx, msg->user_data, msg->len, 0); 5058ed29b0b4SJens Axboe io_commit_cqring(target_ctx); 5059ed29b0b4SJens Axboe spin_unlock(&target_ctx->completion_lock); 5060ed29b0b4SJens Axboe 5061ed29b0b4SJens Axboe if (filled) { 5062ed29b0b4SJens Axboe io_cqring_ev_posted(target_ctx); 5063ed29b0b4SJens Axboe ret = 0; 5064ed29b0b4SJens Axboe } 5065ed29b0b4SJens Axboe 5066ed29b0b4SJens Axboe done: 5067ed29b0b4SJens Axboe if (ret < 0) 5068ed29b0b4SJens Axboe req_set_fail(req); 5069ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 5070ed29b0b4SJens Axboe /* put file to avoid an attempt to IOPOLL the req */ 5071ed29b0b4SJens Axboe io_put_file(req->file); 5072ed29b0b4SJens Axboe req->file = NULL; 5073ed29b0b4SJens Axboe return 0; 5074ed29b0b4SJens Axboe } 5075ed29b0b4SJens Axboe 5076ed29b0b4SJens Axboe static int io_fsync_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 5077ed29b0b4SJens Axboe { 5078e4a71006SJens Axboe struct io_sync *sync = io_kiocb_to_cmd(req); 5079e4a71006SJens Axboe 5080ed29b0b4SJens Axboe if (unlikely(sqe->addr || sqe->buf_index || sqe->splice_fd_in)) 5081ed29b0b4SJens Axboe return -EINVAL; 5082ed29b0b4SJens Axboe 5083e4a71006SJens Axboe sync->flags = READ_ONCE(sqe->fsync_flags); 5084e4a71006SJens Axboe if (unlikely(sync->flags & ~IORING_FSYNC_DATASYNC)) 5085ed29b0b4SJens Axboe return -EINVAL; 5086ed29b0b4SJens Axboe 5087e4a71006SJens Axboe sync->off = READ_ONCE(sqe->off); 5088e4a71006SJens Axboe sync->len = READ_ONCE(sqe->len); 5089ed29b0b4SJens Axboe return 0; 5090ed29b0b4SJens Axboe } 5091ed29b0b4SJens Axboe 5092ed29b0b4SJens Axboe static int io_fsync(struct io_kiocb *req, unsigned int issue_flags) 5093ed29b0b4SJens Axboe { 5094e4a71006SJens Axboe struct io_sync *sync = io_kiocb_to_cmd(req); 5095e4a71006SJens Axboe loff_t end = sync->off + sync->len; 5096ed29b0b4SJens Axboe int ret; 5097ed29b0b4SJens Axboe 5098ed29b0b4SJens Axboe /* fsync always requires a blocking context */ 5099ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 5100ed29b0b4SJens Axboe return -EAGAIN; 5101ed29b0b4SJens Axboe 5102e4a71006SJens Axboe ret = vfs_fsync_range(req->file, sync->off, end > 0 ? end : LLONG_MAX, 5103e4a71006SJens Axboe sync->flags & IORING_FSYNC_DATASYNC); 5104ed29b0b4SJens Axboe io_req_complete(req, ret); 5105ed29b0b4SJens Axboe return 0; 5106ed29b0b4SJens Axboe } 5107ed29b0b4SJens Axboe 5108ed29b0b4SJens Axboe static int io_fallocate_prep(struct io_kiocb *req, 5109ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 5110ed29b0b4SJens Axboe { 5111e4a71006SJens Axboe struct io_sync *sync = io_kiocb_to_cmd(req); 5112e4a71006SJens Axboe 5113ed29b0b4SJens Axboe if (sqe->buf_index || sqe->rw_flags || sqe->splice_fd_in) 5114ed29b0b4SJens Axboe return -EINVAL; 5115ed29b0b4SJens Axboe 5116e4a71006SJens Axboe sync->off = READ_ONCE(sqe->off); 5117e4a71006SJens Axboe sync->len = READ_ONCE(sqe->addr); 5118e4a71006SJens Axboe sync->mode = READ_ONCE(sqe->len); 5119ed29b0b4SJens Axboe return 0; 5120ed29b0b4SJens Axboe } 5121ed29b0b4SJens Axboe 5122ed29b0b4SJens Axboe static int io_fallocate(struct io_kiocb *req, unsigned int issue_flags) 5123ed29b0b4SJens Axboe { 5124e4a71006SJens Axboe struct io_sync *sync = io_kiocb_to_cmd(req); 5125ed29b0b4SJens Axboe int ret; 5126ed29b0b4SJens Axboe 5127ed29b0b4SJens Axboe /* fallocate always requiring blocking context */ 5128ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 5129ed29b0b4SJens Axboe return -EAGAIN; 5130e4a71006SJens Axboe ret = vfs_fallocate(req->file, sync->mode, sync->off, sync->len); 5131ed29b0b4SJens Axboe if (ret >= 0) 5132ed29b0b4SJens Axboe fsnotify_modify(req->file); 5133ed29b0b4SJens Axboe io_req_complete(req, ret); 5134ed29b0b4SJens Axboe return 0; 5135ed29b0b4SJens Axboe } 5136ed29b0b4SJens Axboe 5137ed29b0b4SJens Axboe static int __io_openat_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 5138ed29b0b4SJens Axboe { 5139dd752582SJens Axboe struct io_open *open = io_kiocb_to_cmd(req); 5140ed29b0b4SJens Axboe const char __user *fname; 5141ed29b0b4SJens Axboe int ret; 5142ed29b0b4SJens Axboe 5143ed29b0b4SJens Axboe if (unlikely(sqe->buf_index)) 5144ed29b0b4SJens Axboe return -EINVAL; 5145ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_FIXED_FILE)) 5146ed29b0b4SJens Axboe return -EBADF; 5147ed29b0b4SJens Axboe 5148ed29b0b4SJens Axboe /* open.how should be already initialised */ 5149dd752582SJens Axboe if (!(open->how.flags & O_PATH) && force_o_largefile()) 5150dd752582SJens Axboe open->how.flags |= O_LARGEFILE; 5151ed29b0b4SJens Axboe 5152dd752582SJens Axboe open->dfd = READ_ONCE(sqe->fd); 5153ed29b0b4SJens Axboe fname = u64_to_user_ptr(READ_ONCE(sqe->addr)); 5154dd752582SJens Axboe open->filename = getname(fname); 5155dd752582SJens Axboe if (IS_ERR(open->filename)) { 5156dd752582SJens Axboe ret = PTR_ERR(open->filename); 5157dd752582SJens Axboe open->filename = NULL; 5158ed29b0b4SJens Axboe return ret; 5159ed29b0b4SJens Axboe } 5160ed29b0b4SJens Axboe 5161dd752582SJens Axboe open->file_slot = READ_ONCE(sqe->file_index); 5162dd752582SJens Axboe if (open->file_slot && (open->how.flags & O_CLOEXEC)) 5163ed29b0b4SJens Axboe return -EINVAL; 5164ed29b0b4SJens Axboe 5165dd752582SJens Axboe open->nofile = rlimit(RLIMIT_NOFILE); 5166ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 5167ed29b0b4SJens Axboe return 0; 5168ed29b0b4SJens Axboe } 5169ed29b0b4SJens Axboe 5170ed29b0b4SJens Axboe static int io_openat_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 5171ed29b0b4SJens Axboe { 5172dd752582SJens Axboe struct io_open *open = io_kiocb_to_cmd(req); 5173ed29b0b4SJens Axboe u64 mode = READ_ONCE(sqe->len); 5174ed29b0b4SJens Axboe u64 flags = READ_ONCE(sqe->open_flags); 5175ed29b0b4SJens Axboe 5176dd752582SJens Axboe open->how = build_open_how(flags, mode); 5177ed29b0b4SJens Axboe return __io_openat_prep(req, sqe); 5178ed29b0b4SJens Axboe } 5179ed29b0b4SJens Axboe 5180ed29b0b4SJens Axboe static int io_openat2_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 5181ed29b0b4SJens Axboe { 5182dd752582SJens Axboe struct io_open *open = io_kiocb_to_cmd(req); 5183ed29b0b4SJens Axboe struct open_how __user *how; 5184ed29b0b4SJens Axboe size_t len; 5185ed29b0b4SJens Axboe int ret; 5186ed29b0b4SJens Axboe 5187ed29b0b4SJens Axboe how = u64_to_user_ptr(READ_ONCE(sqe->addr2)); 5188ed29b0b4SJens Axboe len = READ_ONCE(sqe->len); 5189ed29b0b4SJens Axboe if (len < OPEN_HOW_SIZE_VER0) 5190ed29b0b4SJens Axboe return -EINVAL; 5191ed29b0b4SJens Axboe 5192dd752582SJens Axboe ret = copy_struct_from_user(&open->how, sizeof(open->how), how, len); 5193ed29b0b4SJens Axboe if (ret) 5194ed29b0b4SJens Axboe return ret; 5195ed29b0b4SJens Axboe 5196ed29b0b4SJens Axboe return __io_openat_prep(req, sqe); 5197ed29b0b4SJens Axboe } 5198ed29b0b4SJens Axboe 5199ed29b0b4SJens Axboe static int io_file_bitmap_get(struct io_ring_ctx *ctx) 5200ed29b0b4SJens Axboe { 5201ed29b0b4SJens Axboe struct io_file_table *table = &ctx->file_table; 5202ed29b0b4SJens Axboe unsigned long nr = ctx->nr_user_files; 5203ed29b0b4SJens Axboe int ret; 5204ed29b0b4SJens Axboe 5205ed29b0b4SJens Axboe do { 5206ed29b0b4SJens Axboe ret = find_next_zero_bit(table->bitmap, nr, table->alloc_hint); 5207ed29b0b4SJens Axboe if (ret != nr) 5208ed29b0b4SJens Axboe return ret; 5209ed29b0b4SJens Axboe 5210ed29b0b4SJens Axboe if (!table->alloc_hint) 5211ed29b0b4SJens Axboe break; 5212ed29b0b4SJens Axboe 5213ed29b0b4SJens Axboe nr = table->alloc_hint; 5214ed29b0b4SJens Axboe table->alloc_hint = 0; 5215ed29b0b4SJens Axboe } while (1); 5216ed29b0b4SJens Axboe 5217ed29b0b4SJens Axboe return -ENFILE; 5218ed29b0b4SJens Axboe } 5219ed29b0b4SJens Axboe 5220ed29b0b4SJens Axboe /* 5221ed29b0b4SJens Axboe * Note when io_fixed_fd_install() returns error value, it will ensure 5222ed29b0b4SJens Axboe * fput() is called correspondingly. 5223ed29b0b4SJens Axboe */ 5224ed29b0b4SJens Axboe static int io_fixed_fd_install(struct io_kiocb *req, unsigned int issue_flags, 5225ed29b0b4SJens Axboe struct file *file, unsigned int file_slot) 5226ed29b0b4SJens Axboe { 5227ed29b0b4SJens Axboe bool alloc_slot = file_slot == IORING_FILE_INDEX_ALLOC; 5228ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 5229ed29b0b4SJens Axboe int ret; 5230ed29b0b4SJens Axboe 5231ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 5232ed29b0b4SJens Axboe 5233ed29b0b4SJens Axboe if (alloc_slot) { 5234ed29b0b4SJens Axboe ret = io_file_bitmap_get(ctx); 5235ed29b0b4SJens Axboe if (unlikely(ret < 0)) 5236ed29b0b4SJens Axboe goto err; 5237ed29b0b4SJens Axboe file_slot = ret; 5238ed29b0b4SJens Axboe } else { 5239ed29b0b4SJens Axboe file_slot--; 5240ed29b0b4SJens Axboe } 5241ed29b0b4SJens Axboe 5242ed29b0b4SJens Axboe ret = io_install_fixed_file(req, file, issue_flags, file_slot); 5243ed29b0b4SJens Axboe if (!ret && alloc_slot) 5244ed29b0b4SJens Axboe ret = file_slot; 5245ed29b0b4SJens Axboe err: 5246ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 5247ed29b0b4SJens Axboe if (unlikely(ret < 0)) 5248ed29b0b4SJens Axboe fput(file); 5249ed29b0b4SJens Axboe return ret; 5250ed29b0b4SJens Axboe } 5251ed29b0b4SJens Axboe 5252ed29b0b4SJens Axboe static int io_openat2(struct io_kiocb *req, unsigned int issue_flags) 5253ed29b0b4SJens Axboe { 5254dd752582SJens Axboe struct io_open *open = io_kiocb_to_cmd(req); 5255ed29b0b4SJens Axboe struct open_flags op; 5256ed29b0b4SJens Axboe struct file *file; 5257ed29b0b4SJens Axboe bool resolve_nonblock, nonblock_set; 5258dd752582SJens Axboe bool fixed = !!open->file_slot; 5259ed29b0b4SJens Axboe int ret; 5260ed29b0b4SJens Axboe 5261dd752582SJens Axboe ret = build_open_flags(&open->how, &op); 5262ed29b0b4SJens Axboe if (ret) 5263ed29b0b4SJens Axboe goto err; 5264ed29b0b4SJens Axboe nonblock_set = op.open_flag & O_NONBLOCK; 5265dd752582SJens Axboe resolve_nonblock = open->how.resolve & RESOLVE_CACHED; 5266ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) { 5267ed29b0b4SJens Axboe /* 5268ed29b0b4SJens Axboe * Don't bother trying for O_TRUNC, O_CREAT, or O_TMPFILE open, 5269ed29b0b4SJens Axboe * it'll always -EAGAIN 5270ed29b0b4SJens Axboe */ 5271dd752582SJens Axboe if (open->how.flags & (O_TRUNC | O_CREAT | O_TMPFILE)) 5272ed29b0b4SJens Axboe return -EAGAIN; 5273ed29b0b4SJens Axboe op.lookup_flags |= LOOKUP_CACHED; 5274ed29b0b4SJens Axboe op.open_flag |= O_NONBLOCK; 5275ed29b0b4SJens Axboe } 5276ed29b0b4SJens Axboe 5277ed29b0b4SJens Axboe if (!fixed) { 5278dd752582SJens Axboe ret = __get_unused_fd_flags(open->how.flags, open->nofile); 5279ed29b0b4SJens Axboe if (ret < 0) 5280ed29b0b4SJens Axboe goto err; 5281ed29b0b4SJens Axboe } 5282ed29b0b4SJens Axboe 5283dd752582SJens Axboe file = do_filp_open(open->dfd, open->filename, &op); 5284ed29b0b4SJens Axboe if (IS_ERR(file)) { 5285ed29b0b4SJens Axboe /* 5286ed29b0b4SJens Axboe * We could hang on to this 'fd' on retrying, but seems like 5287ed29b0b4SJens Axboe * marginal gain for something that is now known to be a slower 5288ed29b0b4SJens Axboe * path. So just put it, and we'll get a new one when we retry. 5289ed29b0b4SJens Axboe */ 5290ed29b0b4SJens Axboe if (!fixed) 5291ed29b0b4SJens Axboe put_unused_fd(ret); 5292ed29b0b4SJens Axboe 5293ed29b0b4SJens Axboe ret = PTR_ERR(file); 5294ed29b0b4SJens Axboe /* only retry if RESOLVE_CACHED wasn't already set by application */ 5295ed29b0b4SJens Axboe if (ret == -EAGAIN && 5296ed29b0b4SJens Axboe (!resolve_nonblock && (issue_flags & IO_URING_F_NONBLOCK))) 5297ed29b0b4SJens Axboe return -EAGAIN; 5298ed29b0b4SJens Axboe goto err; 5299ed29b0b4SJens Axboe } 5300ed29b0b4SJens Axboe 5301ed29b0b4SJens Axboe if ((issue_flags & IO_URING_F_NONBLOCK) && !nonblock_set) 5302ed29b0b4SJens Axboe file->f_flags &= ~O_NONBLOCK; 5303ed29b0b4SJens Axboe fsnotify_open(file); 5304ed29b0b4SJens Axboe 5305ed29b0b4SJens Axboe if (!fixed) 5306ed29b0b4SJens Axboe fd_install(ret, file); 5307ed29b0b4SJens Axboe else 5308ed29b0b4SJens Axboe ret = io_fixed_fd_install(req, issue_flags, file, 5309dd752582SJens Axboe open->file_slot); 5310ed29b0b4SJens Axboe err: 5311dd752582SJens Axboe putname(open->filename); 5312ed29b0b4SJens Axboe req->flags &= ~REQ_F_NEED_CLEANUP; 5313ed29b0b4SJens Axboe if (ret < 0) 5314ed29b0b4SJens Axboe req_set_fail(req); 5315ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 5316ed29b0b4SJens Axboe return 0; 5317ed29b0b4SJens Axboe } 5318ed29b0b4SJens Axboe 5319ed29b0b4SJens Axboe static int io_openat(struct io_kiocb *req, unsigned int issue_flags) 5320ed29b0b4SJens Axboe { 5321ed29b0b4SJens Axboe return io_openat2(req, issue_flags); 5322ed29b0b4SJens Axboe } 5323ed29b0b4SJens Axboe 5324ed29b0b4SJens Axboe static int io_remove_buffers_prep(struct io_kiocb *req, 5325ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 5326ed29b0b4SJens Axboe { 5327*c1ee5595SJens Axboe struct io_provide_buf *p = io_kiocb_to_cmd(req); 5328ed29b0b4SJens Axboe u64 tmp; 5329ed29b0b4SJens Axboe 5330ed29b0b4SJens Axboe if (sqe->rw_flags || sqe->addr || sqe->len || sqe->off || 5331ed29b0b4SJens Axboe sqe->splice_fd_in) 5332ed29b0b4SJens Axboe return -EINVAL; 5333ed29b0b4SJens Axboe 5334ed29b0b4SJens Axboe tmp = READ_ONCE(sqe->fd); 5335ed29b0b4SJens Axboe if (!tmp || tmp > USHRT_MAX) 5336ed29b0b4SJens Axboe return -EINVAL; 5337ed29b0b4SJens Axboe 5338ed29b0b4SJens Axboe memset(p, 0, sizeof(*p)); 5339ed29b0b4SJens Axboe p->nbufs = tmp; 5340ed29b0b4SJens Axboe p->bgid = READ_ONCE(sqe->buf_group); 5341ed29b0b4SJens Axboe return 0; 5342ed29b0b4SJens Axboe } 5343ed29b0b4SJens Axboe 5344ed29b0b4SJens Axboe static int __io_remove_buffers(struct io_ring_ctx *ctx, 5345ed29b0b4SJens Axboe struct io_buffer_list *bl, unsigned nbufs) 5346ed29b0b4SJens Axboe { 5347ed29b0b4SJens Axboe unsigned i = 0; 5348ed29b0b4SJens Axboe 5349ed29b0b4SJens Axboe /* shouldn't happen */ 5350ed29b0b4SJens Axboe if (!nbufs) 5351ed29b0b4SJens Axboe return 0; 5352ed29b0b4SJens Axboe 5353ed29b0b4SJens Axboe if (bl->buf_nr_pages) { 5354ed29b0b4SJens Axboe int j; 5355ed29b0b4SJens Axboe 5356ed29b0b4SJens Axboe i = bl->buf_ring->tail - bl->head; 5357ed29b0b4SJens Axboe for (j = 0; j < bl->buf_nr_pages; j++) 5358ed29b0b4SJens Axboe unpin_user_page(bl->buf_pages[j]); 5359ed29b0b4SJens Axboe kvfree(bl->buf_pages); 5360ed29b0b4SJens Axboe bl->buf_pages = NULL; 5361ed29b0b4SJens Axboe bl->buf_nr_pages = 0; 5362ed29b0b4SJens Axboe /* make sure it's seen as empty */ 5363ed29b0b4SJens Axboe INIT_LIST_HEAD(&bl->buf_list); 5364ed29b0b4SJens Axboe return i; 5365ed29b0b4SJens Axboe } 5366ed29b0b4SJens Axboe 5367ed29b0b4SJens Axboe /* the head kbuf is the list itself */ 5368ed29b0b4SJens Axboe while (!list_empty(&bl->buf_list)) { 5369ed29b0b4SJens Axboe struct io_buffer *nxt; 5370ed29b0b4SJens Axboe 5371ed29b0b4SJens Axboe nxt = list_first_entry(&bl->buf_list, struct io_buffer, list); 5372ed29b0b4SJens Axboe list_del(&nxt->list); 5373ed29b0b4SJens Axboe if (++i == nbufs) 5374ed29b0b4SJens Axboe return i; 5375ed29b0b4SJens Axboe cond_resched(); 5376ed29b0b4SJens Axboe } 5377ed29b0b4SJens Axboe i++; 5378ed29b0b4SJens Axboe 5379ed29b0b4SJens Axboe return i; 5380ed29b0b4SJens Axboe } 5381ed29b0b4SJens Axboe 5382ed29b0b4SJens Axboe static int io_remove_buffers(struct io_kiocb *req, unsigned int issue_flags) 5383ed29b0b4SJens Axboe { 5384*c1ee5595SJens Axboe struct io_provide_buf *p = io_kiocb_to_cmd(req); 5385ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 5386ed29b0b4SJens Axboe struct io_buffer_list *bl; 5387ed29b0b4SJens Axboe int ret = 0; 5388ed29b0b4SJens Axboe 5389ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 5390ed29b0b4SJens Axboe 5391ed29b0b4SJens Axboe ret = -ENOENT; 5392ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, p->bgid); 5393ed29b0b4SJens Axboe if (bl) { 5394ed29b0b4SJens Axboe ret = -EINVAL; 5395ed29b0b4SJens Axboe /* can't use provide/remove buffers command on mapped buffers */ 5396ed29b0b4SJens Axboe if (!bl->buf_nr_pages) 5397ed29b0b4SJens Axboe ret = __io_remove_buffers(ctx, bl, p->nbufs); 5398ed29b0b4SJens Axboe } 5399ed29b0b4SJens Axboe if (ret < 0) 5400ed29b0b4SJens Axboe req_set_fail(req); 5401ed29b0b4SJens Axboe 5402ed29b0b4SJens Axboe /* complete before unlock, IOPOLL may need the lock */ 5403ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 5404ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 5405ed29b0b4SJens Axboe return 0; 5406ed29b0b4SJens Axboe } 5407ed29b0b4SJens Axboe 5408ed29b0b4SJens Axboe static int io_provide_buffers_prep(struct io_kiocb *req, 5409ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 5410ed29b0b4SJens Axboe { 5411ed29b0b4SJens Axboe unsigned long size, tmp_check; 5412*c1ee5595SJens Axboe struct io_provide_buf *p = io_kiocb_to_cmd(req); 5413ed29b0b4SJens Axboe u64 tmp; 5414ed29b0b4SJens Axboe 5415ed29b0b4SJens Axboe if (sqe->rw_flags || sqe->splice_fd_in) 5416ed29b0b4SJens Axboe return -EINVAL; 5417ed29b0b4SJens Axboe 5418ed29b0b4SJens Axboe tmp = READ_ONCE(sqe->fd); 5419ed29b0b4SJens Axboe if (!tmp || tmp > USHRT_MAX) 5420ed29b0b4SJens Axboe return -E2BIG; 5421ed29b0b4SJens Axboe p->nbufs = tmp; 5422ed29b0b4SJens Axboe p->addr = READ_ONCE(sqe->addr); 5423ed29b0b4SJens Axboe p->len = READ_ONCE(sqe->len); 5424ed29b0b4SJens Axboe 5425ed29b0b4SJens Axboe if (check_mul_overflow((unsigned long)p->len, (unsigned long)p->nbufs, 5426ed29b0b4SJens Axboe &size)) 5427ed29b0b4SJens Axboe return -EOVERFLOW; 5428ed29b0b4SJens Axboe if (check_add_overflow((unsigned long)p->addr, size, &tmp_check)) 5429ed29b0b4SJens Axboe return -EOVERFLOW; 5430ed29b0b4SJens Axboe 5431ed29b0b4SJens Axboe size = (unsigned long)p->len * p->nbufs; 5432ed29b0b4SJens Axboe if (!access_ok(u64_to_user_ptr(p->addr), size)) 5433ed29b0b4SJens Axboe return -EFAULT; 5434ed29b0b4SJens Axboe 5435ed29b0b4SJens Axboe p->bgid = READ_ONCE(sqe->buf_group); 5436ed29b0b4SJens Axboe tmp = READ_ONCE(sqe->off); 5437ed29b0b4SJens Axboe if (tmp > USHRT_MAX) 5438ed29b0b4SJens Axboe return -E2BIG; 5439ed29b0b4SJens Axboe p->bid = tmp; 5440ed29b0b4SJens Axboe return 0; 5441ed29b0b4SJens Axboe } 5442ed29b0b4SJens Axboe 5443ed29b0b4SJens Axboe static int io_refill_buffer_cache(struct io_ring_ctx *ctx) 5444ed29b0b4SJens Axboe { 5445ed29b0b4SJens Axboe struct io_buffer *buf; 5446ed29b0b4SJens Axboe struct page *page; 5447ed29b0b4SJens Axboe int bufs_in_page; 5448ed29b0b4SJens Axboe 5449ed29b0b4SJens Axboe /* 5450ed29b0b4SJens Axboe * Completions that don't happen inline (eg not under uring_lock) will 5451ed29b0b4SJens Axboe * add to ->io_buffers_comp. If we don't have any free buffers, check 5452ed29b0b4SJens Axboe * the completion list and splice those entries first. 5453ed29b0b4SJens Axboe */ 5454ed29b0b4SJens Axboe if (!list_empty_careful(&ctx->io_buffers_comp)) { 5455ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 5456ed29b0b4SJens Axboe if (!list_empty(&ctx->io_buffers_comp)) { 5457ed29b0b4SJens Axboe list_splice_init(&ctx->io_buffers_comp, 5458ed29b0b4SJens Axboe &ctx->io_buffers_cache); 5459ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 5460ed29b0b4SJens Axboe return 0; 5461ed29b0b4SJens Axboe } 5462ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 5463ed29b0b4SJens Axboe } 5464ed29b0b4SJens Axboe 5465ed29b0b4SJens Axboe /* 5466ed29b0b4SJens Axboe * No free buffers and no completion entries either. Allocate a new 5467ed29b0b4SJens Axboe * page worth of buffer entries and add those to our freelist. 5468ed29b0b4SJens Axboe */ 5469ed29b0b4SJens Axboe page = alloc_page(GFP_KERNEL_ACCOUNT); 5470ed29b0b4SJens Axboe if (!page) 5471ed29b0b4SJens Axboe return -ENOMEM; 5472ed29b0b4SJens Axboe 5473ed29b0b4SJens Axboe list_add(&page->lru, &ctx->io_buffers_pages); 5474ed29b0b4SJens Axboe 5475ed29b0b4SJens Axboe buf = page_address(page); 5476ed29b0b4SJens Axboe bufs_in_page = PAGE_SIZE / sizeof(*buf); 5477ed29b0b4SJens Axboe while (bufs_in_page) { 5478ed29b0b4SJens Axboe list_add_tail(&buf->list, &ctx->io_buffers_cache); 5479ed29b0b4SJens Axboe buf++; 5480ed29b0b4SJens Axboe bufs_in_page--; 5481ed29b0b4SJens Axboe } 5482ed29b0b4SJens Axboe 5483ed29b0b4SJens Axboe return 0; 5484ed29b0b4SJens Axboe } 5485ed29b0b4SJens Axboe 5486ed29b0b4SJens Axboe static int io_add_buffers(struct io_ring_ctx *ctx, struct io_provide_buf *pbuf, 5487ed29b0b4SJens Axboe struct io_buffer_list *bl) 5488ed29b0b4SJens Axboe { 5489ed29b0b4SJens Axboe struct io_buffer *buf; 5490ed29b0b4SJens Axboe u64 addr = pbuf->addr; 5491ed29b0b4SJens Axboe int i, bid = pbuf->bid; 5492ed29b0b4SJens Axboe 5493ed29b0b4SJens Axboe for (i = 0; i < pbuf->nbufs; i++) { 5494ed29b0b4SJens Axboe if (list_empty(&ctx->io_buffers_cache) && 5495ed29b0b4SJens Axboe io_refill_buffer_cache(ctx)) 5496ed29b0b4SJens Axboe break; 5497ed29b0b4SJens Axboe buf = list_first_entry(&ctx->io_buffers_cache, struct io_buffer, 5498ed29b0b4SJens Axboe list); 5499ed29b0b4SJens Axboe list_move_tail(&buf->list, &bl->buf_list); 5500ed29b0b4SJens Axboe buf->addr = addr; 5501ed29b0b4SJens Axboe buf->len = min_t(__u32, pbuf->len, MAX_RW_COUNT); 5502ed29b0b4SJens Axboe buf->bid = bid; 5503ed29b0b4SJens Axboe buf->bgid = pbuf->bgid; 5504ed29b0b4SJens Axboe addr += pbuf->len; 5505ed29b0b4SJens Axboe bid++; 5506ed29b0b4SJens Axboe cond_resched(); 5507ed29b0b4SJens Axboe } 5508ed29b0b4SJens Axboe 5509ed29b0b4SJens Axboe return i ? 0 : -ENOMEM; 5510ed29b0b4SJens Axboe } 5511ed29b0b4SJens Axboe 5512ed29b0b4SJens Axboe static __cold int io_init_bl_list(struct io_ring_ctx *ctx) 5513ed29b0b4SJens Axboe { 5514ed29b0b4SJens Axboe int i; 5515ed29b0b4SJens Axboe 5516ed29b0b4SJens Axboe ctx->io_bl = kcalloc(BGID_ARRAY, sizeof(struct io_buffer_list), 5517ed29b0b4SJens Axboe GFP_KERNEL); 5518ed29b0b4SJens Axboe if (!ctx->io_bl) 5519ed29b0b4SJens Axboe return -ENOMEM; 5520ed29b0b4SJens Axboe 5521ed29b0b4SJens Axboe for (i = 0; i < BGID_ARRAY; i++) { 5522ed29b0b4SJens Axboe INIT_LIST_HEAD(&ctx->io_bl[i].buf_list); 5523ed29b0b4SJens Axboe ctx->io_bl[i].bgid = i; 5524ed29b0b4SJens Axboe } 5525ed29b0b4SJens Axboe 5526ed29b0b4SJens Axboe return 0; 5527ed29b0b4SJens Axboe } 5528ed29b0b4SJens Axboe 5529ed29b0b4SJens Axboe static int io_provide_buffers(struct io_kiocb *req, unsigned int issue_flags) 5530ed29b0b4SJens Axboe { 5531*c1ee5595SJens Axboe struct io_provide_buf *p = io_kiocb_to_cmd(req); 5532ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 5533ed29b0b4SJens Axboe struct io_buffer_list *bl; 5534ed29b0b4SJens Axboe int ret = 0; 5535ed29b0b4SJens Axboe 5536ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 5537ed29b0b4SJens Axboe 5538ed29b0b4SJens Axboe if (unlikely(p->bgid < BGID_ARRAY && !ctx->io_bl)) { 5539ed29b0b4SJens Axboe ret = io_init_bl_list(ctx); 5540ed29b0b4SJens Axboe if (ret) 5541ed29b0b4SJens Axboe goto err; 5542ed29b0b4SJens Axboe } 5543ed29b0b4SJens Axboe 5544ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, p->bgid); 5545ed29b0b4SJens Axboe if (unlikely(!bl)) { 5546ed29b0b4SJens Axboe bl = kzalloc(sizeof(*bl), GFP_KERNEL); 5547ed29b0b4SJens Axboe if (!bl) { 5548ed29b0b4SJens Axboe ret = -ENOMEM; 5549ed29b0b4SJens Axboe goto err; 5550ed29b0b4SJens Axboe } 5551ed29b0b4SJens Axboe INIT_LIST_HEAD(&bl->buf_list); 5552ed29b0b4SJens Axboe ret = io_buffer_add_list(ctx, bl, p->bgid); 5553ed29b0b4SJens Axboe if (ret) { 5554ed29b0b4SJens Axboe kfree(bl); 5555ed29b0b4SJens Axboe goto err; 5556ed29b0b4SJens Axboe } 5557ed29b0b4SJens Axboe } 5558ed29b0b4SJens Axboe /* can't add buffers via this command for a mapped buffer ring */ 5559ed29b0b4SJens Axboe if (bl->buf_nr_pages) { 5560ed29b0b4SJens Axboe ret = -EINVAL; 5561ed29b0b4SJens Axboe goto err; 5562ed29b0b4SJens Axboe } 5563ed29b0b4SJens Axboe 5564ed29b0b4SJens Axboe ret = io_add_buffers(ctx, p, bl); 5565ed29b0b4SJens Axboe err: 5566ed29b0b4SJens Axboe if (ret < 0) 5567ed29b0b4SJens Axboe req_set_fail(req); 5568ed29b0b4SJens Axboe /* complete before unlock, IOPOLL may need the lock */ 5569ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 5570ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 5571ed29b0b4SJens Axboe return 0; 5572ed29b0b4SJens Axboe } 5573ed29b0b4SJens Axboe 5574ed29b0b4SJens Axboe static int io_epoll_ctl_prep(struct io_kiocb *req, 5575ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 5576ed29b0b4SJens Axboe { 5577ed29b0b4SJens Axboe #if defined(CONFIG_EPOLL) 55783e93a357SJens Axboe struct io_epoll *epoll = io_kiocb_to_cmd(req); 55793e93a357SJens Axboe 5580ed29b0b4SJens Axboe if (sqe->buf_index || sqe->splice_fd_in) 5581ed29b0b4SJens Axboe return -EINVAL; 5582ed29b0b4SJens Axboe 55833e93a357SJens Axboe epoll->epfd = READ_ONCE(sqe->fd); 55843e93a357SJens Axboe epoll->op = READ_ONCE(sqe->len); 55853e93a357SJens Axboe epoll->fd = READ_ONCE(sqe->off); 5586ed29b0b4SJens Axboe 55873e93a357SJens Axboe if (ep_op_has_event(epoll->op)) { 5588ed29b0b4SJens Axboe struct epoll_event __user *ev; 5589ed29b0b4SJens Axboe 5590ed29b0b4SJens Axboe ev = u64_to_user_ptr(READ_ONCE(sqe->addr)); 55913e93a357SJens Axboe if (copy_from_user(&epoll->event, ev, sizeof(*ev))) 5592ed29b0b4SJens Axboe return -EFAULT; 5593ed29b0b4SJens Axboe } 5594ed29b0b4SJens Axboe 5595ed29b0b4SJens Axboe return 0; 5596ed29b0b4SJens Axboe #else 5597ed29b0b4SJens Axboe return -EOPNOTSUPP; 5598ed29b0b4SJens Axboe #endif 5599ed29b0b4SJens Axboe } 5600ed29b0b4SJens Axboe 5601ed29b0b4SJens Axboe static int io_epoll_ctl(struct io_kiocb *req, unsigned int issue_flags) 5602ed29b0b4SJens Axboe { 5603ed29b0b4SJens Axboe #if defined(CONFIG_EPOLL) 56043e93a357SJens Axboe struct io_epoll *ie = io_kiocb_to_cmd(req); 5605ed29b0b4SJens Axboe int ret; 5606ed29b0b4SJens Axboe bool force_nonblock = issue_flags & IO_URING_F_NONBLOCK; 5607ed29b0b4SJens Axboe 5608ed29b0b4SJens Axboe ret = do_epoll_ctl(ie->epfd, ie->op, ie->fd, &ie->event, force_nonblock); 5609ed29b0b4SJens Axboe if (force_nonblock && ret == -EAGAIN) 5610ed29b0b4SJens Axboe return -EAGAIN; 5611ed29b0b4SJens Axboe 5612ed29b0b4SJens Axboe if (ret < 0) 5613ed29b0b4SJens Axboe req_set_fail(req); 5614ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 5615ed29b0b4SJens Axboe return 0; 5616ed29b0b4SJens Axboe #else 5617ed29b0b4SJens Axboe return -EOPNOTSUPP; 5618ed29b0b4SJens Axboe #endif 5619ed29b0b4SJens Axboe } 5620ed29b0b4SJens Axboe 5621ed29b0b4SJens Axboe static int io_madvise_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 5622ed29b0b4SJens Axboe { 5623ed29b0b4SJens Axboe #if defined(CONFIG_ADVISE_SYSCALLS) && defined(CONFIG_MMU) 562437d4842fSJens Axboe struct io_madvise *ma = io_kiocb_to_cmd(req); 562537d4842fSJens Axboe 5626ed29b0b4SJens Axboe if (sqe->buf_index || sqe->off || sqe->splice_fd_in) 5627ed29b0b4SJens Axboe return -EINVAL; 5628ed29b0b4SJens Axboe 562937d4842fSJens Axboe ma->addr = READ_ONCE(sqe->addr); 563037d4842fSJens Axboe ma->len = READ_ONCE(sqe->len); 563137d4842fSJens Axboe ma->advice = READ_ONCE(sqe->fadvise_advice); 5632ed29b0b4SJens Axboe return 0; 5633ed29b0b4SJens Axboe #else 5634ed29b0b4SJens Axboe return -EOPNOTSUPP; 5635ed29b0b4SJens Axboe #endif 5636ed29b0b4SJens Axboe } 5637ed29b0b4SJens Axboe 5638ed29b0b4SJens Axboe static int io_madvise(struct io_kiocb *req, unsigned int issue_flags) 5639ed29b0b4SJens Axboe { 5640ed29b0b4SJens Axboe #if defined(CONFIG_ADVISE_SYSCALLS) && defined(CONFIG_MMU) 564137d4842fSJens Axboe struct io_madvise *ma = io_kiocb_to_cmd(req); 5642ed29b0b4SJens Axboe int ret; 5643ed29b0b4SJens Axboe 5644ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 5645ed29b0b4SJens Axboe return -EAGAIN; 5646ed29b0b4SJens Axboe 5647ed29b0b4SJens Axboe ret = do_madvise(current->mm, ma->addr, ma->len, ma->advice); 5648ed29b0b4SJens Axboe io_req_complete(req, ret); 5649ed29b0b4SJens Axboe return 0; 5650ed29b0b4SJens Axboe #else 5651ed29b0b4SJens Axboe return -EOPNOTSUPP; 5652ed29b0b4SJens Axboe #endif 5653ed29b0b4SJens Axboe } 5654ed29b0b4SJens Axboe 5655ed29b0b4SJens Axboe static int io_fadvise_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 5656ed29b0b4SJens Axboe { 565737d4842fSJens Axboe struct io_fadvise *fa = io_kiocb_to_cmd(req); 565837d4842fSJens Axboe 5659ed29b0b4SJens Axboe if (sqe->buf_index || sqe->addr || sqe->splice_fd_in) 5660ed29b0b4SJens Axboe return -EINVAL; 5661ed29b0b4SJens Axboe 566237d4842fSJens Axboe fa->offset = READ_ONCE(sqe->off); 566337d4842fSJens Axboe fa->len = READ_ONCE(sqe->len); 566437d4842fSJens Axboe fa->advice = READ_ONCE(sqe->fadvise_advice); 5665ed29b0b4SJens Axboe return 0; 5666ed29b0b4SJens Axboe } 5667ed29b0b4SJens Axboe 5668ed29b0b4SJens Axboe static int io_fadvise(struct io_kiocb *req, unsigned int issue_flags) 5669ed29b0b4SJens Axboe { 567037d4842fSJens Axboe struct io_fadvise *fa = io_kiocb_to_cmd(req); 5671ed29b0b4SJens Axboe int ret; 5672ed29b0b4SJens Axboe 5673ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) { 5674ed29b0b4SJens Axboe switch (fa->advice) { 5675ed29b0b4SJens Axboe case POSIX_FADV_NORMAL: 5676ed29b0b4SJens Axboe case POSIX_FADV_RANDOM: 5677ed29b0b4SJens Axboe case POSIX_FADV_SEQUENTIAL: 5678ed29b0b4SJens Axboe break; 5679ed29b0b4SJens Axboe default: 5680ed29b0b4SJens Axboe return -EAGAIN; 5681ed29b0b4SJens Axboe } 5682ed29b0b4SJens Axboe } 5683ed29b0b4SJens Axboe 5684ed29b0b4SJens Axboe ret = vfs_fadvise(req->file, fa->offset, fa->len, fa->advice); 5685ed29b0b4SJens Axboe if (ret < 0) 5686ed29b0b4SJens Axboe req_set_fail(req); 5687ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 5688ed29b0b4SJens Axboe return 0; 5689ed29b0b4SJens Axboe } 5690ed29b0b4SJens Axboe 5691ed29b0b4SJens Axboe static int io_statx_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 5692ed29b0b4SJens Axboe { 5693bb040a21SJens Axboe struct io_statx *sx = io_kiocb_to_cmd(req); 5694ed29b0b4SJens Axboe const char __user *path; 5695ed29b0b4SJens Axboe 5696ed29b0b4SJens Axboe if (sqe->buf_index || sqe->splice_fd_in) 5697ed29b0b4SJens Axboe return -EINVAL; 5698ed29b0b4SJens Axboe if (req->flags & REQ_F_FIXED_FILE) 5699ed29b0b4SJens Axboe return -EBADF; 5700ed29b0b4SJens Axboe 5701bb040a21SJens Axboe sx->dfd = READ_ONCE(sqe->fd); 5702bb040a21SJens Axboe sx->mask = READ_ONCE(sqe->len); 5703ed29b0b4SJens Axboe path = u64_to_user_ptr(READ_ONCE(sqe->addr)); 5704bb040a21SJens Axboe sx->buffer = u64_to_user_ptr(READ_ONCE(sqe->addr2)); 5705bb040a21SJens Axboe sx->flags = READ_ONCE(sqe->statx_flags); 5706ed29b0b4SJens Axboe 5707bb040a21SJens Axboe sx->filename = getname_flags(path, 5708bb040a21SJens Axboe getname_statx_lookup_flags(sx->flags), 5709ed29b0b4SJens Axboe NULL); 5710ed29b0b4SJens Axboe 5711bb040a21SJens Axboe if (IS_ERR(sx->filename)) { 5712bb040a21SJens Axboe int ret = PTR_ERR(sx->filename); 5713ed29b0b4SJens Axboe 5714bb040a21SJens Axboe sx->filename = NULL; 5715ed29b0b4SJens Axboe return ret; 5716ed29b0b4SJens Axboe } 5717ed29b0b4SJens Axboe 5718ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 5719ed29b0b4SJens Axboe return 0; 5720ed29b0b4SJens Axboe } 5721ed29b0b4SJens Axboe 5722ed29b0b4SJens Axboe static int io_statx(struct io_kiocb *req, unsigned int issue_flags) 5723ed29b0b4SJens Axboe { 5724bb040a21SJens Axboe struct io_statx *sx = io_kiocb_to_cmd(req); 5725ed29b0b4SJens Axboe int ret; 5726ed29b0b4SJens Axboe 5727ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 5728ed29b0b4SJens Axboe return -EAGAIN; 5729ed29b0b4SJens Axboe 5730bb040a21SJens Axboe ret = do_statx(sx->dfd, sx->filename, sx->flags, sx->mask, sx->buffer); 5731ed29b0b4SJens Axboe io_req_complete(req, ret); 5732ed29b0b4SJens Axboe return 0; 5733ed29b0b4SJens Axboe } 5734ed29b0b4SJens Axboe 5735ed29b0b4SJens Axboe static int io_close_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 5736ed29b0b4SJens Axboe { 5737dd752582SJens Axboe struct io_close *close = io_kiocb_to_cmd(req); 5738dd752582SJens Axboe 5739ed29b0b4SJens Axboe if (sqe->off || sqe->addr || sqe->len || sqe->rw_flags || sqe->buf_index) 5740ed29b0b4SJens Axboe return -EINVAL; 5741ed29b0b4SJens Axboe if (req->flags & REQ_F_FIXED_FILE) 5742ed29b0b4SJens Axboe return -EBADF; 5743ed29b0b4SJens Axboe 5744dd752582SJens Axboe close->fd = READ_ONCE(sqe->fd); 5745dd752582SJens Axboe close->file_slot = READ_ONCE(sqe->file_index); 5746dd752582SJens Axboe if (close->file_slot && close->fd) 5747ed29b0b4SJens Axboe return -EINVAL; 5748ed29b0b4SJens Axboe 5749ed29b0b4SJens Axboe return 0; 5750ed29b0b4SJens Axboe } 5751ed29b0b4SJens Axboe 5752ed29b0b4SJens Axboe static int io_close(struct io_kiocb *req, unsigned int issue_flags) 5753ed29b0b4SJens Axboe { 5754ed29b0b4SJens Axboe struct files_struct *files = current->files; 5755dd752582SJens Axboe struct io_close *close = io_kiocb_to_cmd(req); 5756ed29b0b4SJens Axboe struct fdtable *fdt; 5757ed29b0b4SJens Axboe struct file *file; 5758ed29b0b4SJens Axboe int ret = -EBADF; 5759ed29b0b4SJens Axboe 5760dd752582SJens Axboe if (close->file_slot) { 5761ed29b0b4SJens Axboe ret = io_close_fixed(req, issue_flags); 5762ed29b0b4SJens Axboe goto err; 5763ed29b0b4SJens Axboe } 5764ed29b0b4SJens Axboe 5765ed29b0b4SJens Axboe spin_lock(&files->file_lock); 5766ed29b0b4SJens Axboe fdt = files_fdtable(files); 5767ed29b0b4SJens Axboe if (close->fd >= fdt->max_fds) { 5768ed29b0b4SJens Axboe spin_unlock(&files->file_lock); 5769ed29b0b4SJens Axboe goto err; 5770ed29b0b4SJens Axboe } 5771ed29b0b4SJens Axboe file = rcu_dereference_protected(fdt->fd[close->fd], 5772ed29b0b4SJens Axboe lockdep_is_held(&files->file_lock)); 5773ed29b0b4SJens Axboe if (!file || file->f_op == &io_uring_fops) { 5774ed29b0b4SJens Axboe spin_unlock(&files->file_lock); 5775ed29b0b4SJens Axboe goto err; 5776ed29b0b4SJens Axboe } 5777ed29b0b4SJens Axboe 5778ed29b0b4SJens Axboe /* if the file has a flush method, be safe and punt to async */ 5779ed29b0b4SJens Axboe if (file->f_op->flush && (issue_flags & IO_URING_F_NONBLOCK)) { 5780ed29b0b4SJens Axboe spin_unlock(&files->file_lock); 5781ed29b0b4SJens Axboe return -EAGAIN; 5782ed29b0b4SJens Axboe } 5783ed29b0b4SJens Axboe 5784ed29b0b4SJens Axboe file = __close_fd_get_file(close->fd); 5785ed29b0b4SJens Axboe spin_unlock(&files->file_lock); 5786ed29b0b4SJens Axboe if (!file) 5787ed29b0b4SJens Axboe goto err; 5788ed29b0b4SJens Axboe 5789ed29b0b4SJens Axboe /* No ->flush() or already async, safely close from here */ 5790ed29b0b4SJens Axboe ret = filp_close(file, current->files); 5791ed29b0b4SJens Axboe err: 5792ed29b0b4SJens Axboe if (ret < 0) 5793ed29b0b4SJens Axboe req_set_fail(req); 5794ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 5795ed29b0b4SJens Axboe return 0; 5796ed29b0b4SJens Axboe } 5797ed29b0b4SJens Axboe 5798ed29b0b4SJens Axboe static int io_sfr_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 5799ed29b0b4SJens Axboe { 5800e4a71006SJens Axboe struct io_sync *sync = io_kiocb_to_cmd(req); 5801e4a71006SJens Axboe 5802ed29b0b4SJens Axboe if (unlikely(sqe->addr || sqe->buf_index || sqe->splice_fd_in)) 5803ed29b0b4SJens Axboe return -EINVAL; 5804ed29b0b4SJens Axboe 5805e4a71006SJens Axboe sync->off = READ_ONCE(sqe->off); 5806e4a71006SJens Axboe sync->len = READ_ONCE(sqe->len); 5807e4a71006SJens Axboe sync->flags = READ_ONCE(sqe->sync_range_flags); 5808ed29b0b4SJens Axboe return 0; 5809ed29b0b4SJens Axboe } 5810ed29b0b4SJens Axboe 5811ed29b0b4SJens Axboe static int io_sync_file_range(struct io_kiocb *req, unsigned int issue_flags) 5812ed29b0b4SJens Axboe { 5813e4a71006SJens Axboe struct io_sync *sync = io_kiocb_to_cmd(req); 5814ed29b0b4SJens Axboe int ret; 5815ed29b0b4SJens Axboe 5816ed29b0b4SJens Axboe /* sync_file_range always requires a blocking context */ 5817ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 5818ed29b0b4SJens Axboe return -EAGAIN; 5819ed29b0b4SJens Axboe 5820e4a71006SJens Axboe ret = sync_file_range(req->file, sync->off, sync->len, sync->flags); 5821ed29b0b4SJens Axboe io_req_complete(req, ret); 5822ed29b0b4SJens Axboe return 0; 5823ed29b0b4SJens Axboe } 5824ed29b0b4SJens Axboe 5825ed29b0b4SJens Axboe #if defined(CONFIG_NET) 5826ed29b0b4SJens Axboe static int io_shutdown_prep(struct io_kiocb *req, 5827ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 5828ed29b0b4SJens Axboe { 58298ff86d85SJens Axboe struct io_shutdown *shutdown = io_kiocb_to_cmd(req); 58308ff86d85SJens Axboe 5831ed29b0b4SJens Axboe if (unlikely(sqe->off || sqe->addr || sqe->rw_flags || 5832ed29b0b4SJens Axboe sqe->buf_index || sqe->splice_fd_in)) 5833ed29b0b4SJens Axboe return -EINVAL; 5834ed29b0b4SJens Axboe 58358ff86d85SJens Axboe shutdown->how = READ_ONCE(sqe->len); 5836ed29b0b4SJens Axboe return 0; 5837ed29b0b4SJens Axboe } 5838ed29b0b4SJens Axboe 5839ed29b0b4SJens Axboe static int io_shutdown(struct io_kiocb *req, unsigned int issue_flags) 5840ed29b0b4SJens Axboe { 58418ff86d85SJens Axboe struct io_shutdown *shutdown = io_kiocb_to_cmd(req); 5842ed29b0b4SJens Axboe struct socket *sock; 5843ed29b0b4SJens Axboe int ret; 5844ed29b0b4SJens Axboe 5845ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 5846ed29b0b4SJens Axboe return -EAGAIN; 5847ed29b0b4SJens Axboe 5848ed29b0b4SJens Axboe sock = sock_from_file(req->file); 5849ed29b0b4SJens Axboe if (unlikely(!sock)) 5850ed29b0b4SJens Axboe return -ENOTSOCK; 5851ed29b0b4SJens Axboe 58528ff86d85SJens Axboe ret = __sys_shutdown_sock(sock, shutdown->how); 5853ed29b0b4SJens Axboe io_req_complete(req, ret); 5854ed29b0b4SJens Axboe return 0; 5855ed29b0b4SJens Axboe } 5856ed29b0b4SJens Axboe 5857ed29b0b4SJens Axboe static bool io_net_retry(struct socket *sock, int flags) 5858ed29b0b4SJens Axboe { 5859ed29b0b4SJens Axboe if (!(flags & MSG_WAITALL)) 5860ed29b0b4SJens Axboe return false; 5861ed29b0b4SJens Axboe return sock->type == SOCK_STREAM || sock->type == SOCK_SEQPACKET; 5862ed29b0b4SJens Axboe } 5863ed29b0b4SJens Axboe 5864ed29b0b4SJens Axboe static int io_setup_async_msg(struct io_kiocb *req, 5865ed29b0b4SJens Axboe struct io_async_msghdr *kmsg) 5866ed29b0b4SJens Axboe { 5867ed29b0b4SJens Axboe struct io_async_msghdr *async_msg = req->async_data; 5868ed29b0b4SJens Axboe 5869ed29b0b4SJens Axboe if (async_msg) 5870ed29b0b4SJens Axboe return -EAGAIN; 5871ed29b0b4SJens Axboe if (io_alloc_async_data(req)) { 5872ed29b0b4SJens Axboe kfree(kmsg->free_iov); 5873ed29b0b4SJens Axboe return -ENOMEM; 5874ed29b0b4SJens Axboe } 5875ed29b0b4SJens Axboe async_msg = req->async_data; 5876ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 5877ed29b0b4SJens Axboe memcpy(async_msg, kmsg, sizeof(*kmsg)); 5878ed29b0b4SJens Axboe async_msg->msg.msg_name = &async_msg->addr; 5879ed29b0b4SJens Axboe /* if were using fast_iov, set it to the new one */ 5880ed29b0b4SJens Axboe if (!async_msg->free_iov) 5881ed29b0b4SJens Axboe async_msg->msg.msg_iter.iov = async_msg->fast_iov; 5882ed29b0b4SJens Axboe 5883ed29b0b4SJens Axboe return -EAGAIN; 5884ed29b0b4SJens Axboe } 5885ed29b0b4SJens Axboe 5886ed29b0b4SJens Axboe static int io_sendmsg_copy_hdr(struct io_kiocb *req, 5887ed29b0b4SJens Axboe struct io_async_msghdr *iomsg) 5888ed29b0b4SJens Axboe { 58898ff86d85SJens Axboe struct io_sr_msg *sr = io_kiocb_to_cmd(req); 58908ff86d85SJens Axboe 5891ed29b0b4SJens Axboe iomsg->msg.msg_name = &iomsg->addr; 5892ed29b0b4SJens Axboe iomsg->free_iov = iomsg->fast_iov; 58938ff86d85SJens Axboe return sendmsg_copy_msghdr(&iomsg->msg, sr->umsg, sr->msg_flags, 58948ff86d85SJens Axboe &iomsg->free_iov); 5895ed29b0b4SJens Axboe } 5896ed29b0b4SJens Axboe 5897ed29b0b4SJens Axboe static int io_sendmsg_prep_async(struct io_kiocb *req) 5898ed29b0b4SJens Axboe { 5899ed29b0b4SJens Axboe int ret; 5900ed29b0b4SJens Axboe 5901ed29b0b4SJens Axboe ret = io_sendmsg_copy_hdr(req, req->async_data); 5902ed29b0b4SJens Axboe if (!ret) 5903ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 5904ed29b0b4SJens Axboe return ret; 5905ed29b0b4SJens Axboe } 5906ed29b0b4SJens Axboe 5907ed29b0b4SJens Axboe static int io_sendmsg_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 5908ed29b0b4SJens Axboe { 59098ff86d85SJens Axboe struct io_sr_msg *sr = io_kiocb_to_cmd(req); 5910ed29b0b4SJens Axboe 5911ed29b0b4SJens Axboe if (unlikely(sqe->file_index || sqe->addr2)) 5912ed29b0b4SJens Axboe return -EINVAL; 5913ed29b0b4SJens Axboe 5914ed29b0b4SJens Axboe sr->umsg = u64_to_user_ptr(READ_ONCE(sqe->addr)); 5915ed29b0b4SJens Axboe sr->len = READ_ONCE(sqe->len); 5916ed29b0b4SJens Axboe sr->flags = READ_ONCE(sqe->ioprio); 5917ed29b0b4SJens Axboe if (sr->flags & ~IORING_RECVSEND_POLL_FIRST) 5918ed29b0b4SJens Axboe return -EINVAL; 5919ed29b0b4SJens Axboe sr->msg_flags = READ_ONCE(sqe->msg_flags) | MSG_NOSIGNAL; 5920ed29b0b4SJens Axboe if (sr->msg_flags & MSG_DONTWAIT) 5921ed29b0b4SJens Axboe req->flags |= REQ_F_NOWAIT; 5922ed29b0b4SJens Axboe 5923ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 5924ed29b0b4SJens Axboe if (req->ctx->compat) 5925ed29b0b4SJens Axboe sr->msg_flags |= MSG_CMSG_COMPAT; 5926ed29b0b4SJens Axboe #endif 5927ed29b0b4SJens Axboe sr->done_io = 0; 5928ed29b0b4SJens Axboe return 0; 5929ed29b0b4SJens Axboe } 5930ed29b0b4SJens Axboe 5931ed29b0b4SJens Axboe static int io_sendmsg(struct io_kiocb *req, unsigned int issue_flags) 5932ed29b0b4SJens Axboe { 59338ff86d85SJens Axboe struct io_sr_msg *sr = io_kiocb_to_cmd(req); 5934ed29b0b4SJens Axboe struct io_async_msghdr iomsg, *kmsg; 5935ed29b0b4SJens Axboe struct socket *sock; 5936ed29b0b4SJens Axboe unsigned flags; 5937ed29b0b4SJens Axboe int min_ret = 0; 5938ed29b0b4SJens Axboe int ret; 5939ed29b0b4SJens Axboe 5940ed29b0b4SJens Axboe sock = sock_from_file(req->file); 5941ed29b0b4SJens Axboe if (unlikely(!sock)) 5942ed29b0b4SJens Axboe return -ENOTSOCK; 5943ed29b0b4SJens Axboe 5944ed29b0b4SJens Axboe if (req_has_async_data(req)) { 5945ed29b0b4SJens Axboe kmsg = req->async_data; 5946ed29b0b4SJens Axboe } else { 5947ed29b0b4SJens Axboe ret = io_sendmsg_copy_hdr(req, &iomsg); 5948ed29b0b4SJens Axboe if (ret) 5949ed29b0b4SJens Axboe return ret; 5950ed29b0b4SJens Axboe kmsg = &iomsg; 5951ed29b0b4SJens Axboe } 5952ed29b0b4SJens Axboe 5953ed29b0b4SJens Axboe if (!(req->flags & REQ_F_POLLED) && 5954ed29b0b4SJens Axboe (sr->flags & IORING_RECVSEND_POLL_FIRST)) 5955ed29b0b4SJens Axboe return io_setup_async_msg(req, kmsg); 5956ed29b0b4SJens Axboe 5957ed29b0b4SJens Axboe flags = sr->msg_flags; 5958ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 5959ed29b0b4SJens Axboe flags |= MSG_DONTWAIT; 5960ed29b0b4SJens Axboe if (flags & MSG_WAITALL) 5961ed29b0b4SJens Axboe min_ret = iov_iter_count(&kmsg->msg.msg_iter); 5962ed29b0b4SJens Axboe 5963ed29b0b4SJens Axboe ret = __sys_sendmsg_sock(sock, &kmsg->msg, flags); 5964ed29b0b4SJens Axboe 5965ed29b0b4SJens Axboe if (ret < min_ret) { 5966ed29b0b4SJens Axboe if (ret == -EAGAIN && (issue_flags & IO_URING_F_NONBLOCK)) 5967ed29b0b4SJens Axboe return io_setup_async_msg(req, kmsg); 5968ed29b0b4SJens Axboe if (ret == -ERESTARTSYS) 5969ed29b0b4SJens Axboe ret = -EINTR; 5970ed29b0b4SJens Axboe if (ret > 0 && io_net_retry(sock, flags)) { 5971ed29b0b4SJens Axboe sr->done_io += ret; 5972ed29b0b4SJens Axboe req->flags |= REQ_F_PARTIAL_IO; 5973ed29b0b4SJens Axboe return io_setup_async_msg(req, kmsg); 5974ed29b0b4SJens Axboe } 5975ed29b0b4SJens Axboe req_set_fail(req); 5976ed29b0b4SJens Axboe } 5977ed29b0b4SJens Axboe /* fast path, check for non-NULL to avoid function call */ 5978ed29b0b4SJens Axboe if (kmsg->free_iov) 5979ed29b0b4SJens Axboe kfree(kmsg->free_iov); 5980ed29b0b4SJens Axboe req->flags &= ~REQ_F_NEED_CLEANUP; 5981ed29b0b4SJens Axboe if (ret >= 0) 5982ed29b0b4SJens Axboe ret += sr->done_io; 5983ed29b0b4SJens Axboe else if (sr->done_io) 5984ed29b0b4SJens Axboe ret = sr->done_io; 5985ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 5986ed29b0b4SJens Axboe return 0; 5987ed29b0b4SJens Axboe } 5988ed29b0b4SJens Axboe 5989ed29b0b4SJens Axboe static int io_send(struct io_kiocb *req, unsigned int issue_flags) 5990ed29b0b4SJens Axboe { 59918ff86d85SJens Axboe struct io_sr_msg *sr = io_kiocb_to_cmd(req); 5992ed29b0b4SJens Axboe struct msghdr msg; 5993ed29b0b4SJens Axboe struct iovec iov; 5994ed29b0b4SJens Axboe struct socket *sock; 5995ed29b0b4SJens Axboe unsigned flags; 5996ed29b0b4SJens Axboe int min_ret = 0; 5997ed29b0b4SJens Axboe int ret; 5998ed29b0b4SJens Axboe 5999ed29b0b4SJens Axboe if (!(req->flags & REQ_F_POLLED) && 6000ed29b0b4SJens Axboe (sr->flags & IORING_RECVSEND_POLL_FIRST)) 6001ed29b0b4SJens Axboe return -EAGAIN; 6002ed29b0b4SJens Axboe 6003ed29b0b4SJens Axboe sock = sock_from_file(req->file); 6004ed29b0b4SJens Axboe if (unlikely(!sock)) 6005ed29b0b4SJens Axboe return -ENOTSOCK; 6006ed29b0b4SJens Axboe 6007ed29b0b4SJens Axboe ret = import_single_range(WRITE, sr->buf, sr->len, &iov, &msg.msg_iter); 6008ed29b0b4SJens Axboe if (unlikely(ret)) 6009ed29b0b4SJens Axboe return ret; 6010ed29b0b4SJens Axboe 6011ed29b0b4SJens Axboe msg.msg_name = NULL; 6012ed29b0b4SJens Axboe msg.msg_control = NULL; 6013ed29b0b4SJens Axboe msg.msg_controllen = 0; 6014ed29b0b4SJens Axboe msg.msg_namelen = 0; 6015ed29b0b4SJens Axboe 6016ed29b0b4SJens Axboe flags = sr->msg_flags; 6017ed29b0b4SJens Axboe if (issue_flags & IO_URING_F_NONBLOCK) 6018ed29b0b4SJens Axboe flags |= MSG_DONTWAIT; 6019ed29b0b4SJens Axboe if (flags & MSG_WAITALL) 6020ed29b0b4SJens Axboe min_ret = iov_iter_count(&msg.msg_iter); 6021ed29b0b4SJens Axboe 6022ed29b0b4SJens Axboe msg.msg_flags = flags; 6023ed29b0b4SJens Axboe ret = sock_sendmsg(sock, &msg); 6024ed29b0b4SJens Axboe if (ret < min_ret) { 6025ed29b0b4SJens Axboe if (ret == -EAGAIN && (issue_flags & IO_URING_F_NONBLOCK)) 6026ed29b0b4SJens Axboe return -EAGAIN; 6027ed29b0b4SJens Axboe if (ret == -ERESTARTSYS) 6028ed29b0b4SJens Axboe ret = -EINTR; 6029ed29b0b4SJens Axboe if (ret > 0 && io_net_retry(sock, flags)) { 6030ed29b0b4SJens Axboe sr->len -= ret; 6031ed29b0b4SJens Axboe sr->buf += ret; 6032ed29b0b4SJens Axboe sr->done_io += ret; 6033ed29b0b4SJens Axboe req->flags |= REQ_F_PARTIAL_IO; 6034ed29b0b4SJens Axboe return -EAGAIN; 6035ed29b0b4SJens Axboe } 6036ed29b0b4SJens Axboe req_set_fail(req); 6037ed29b0b4SJens Axboe } 6038ed29b0b4SJens Axboe if (ret >= 0) 6039ed29b0b4SJens Axboe ret += sr->done_io; 6040ed29b0b4SJens Axboe else if (sr->done_io) 6041ed29b0b4SJens Axboe ret = sr->done_io; 6042ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 6043ed29b0b4SJens Axboe return 0; 6044ed29b0b4SJens Axboe } 6045ed29b0b4SJens Axboe 6046ed29b0b4SJens Axboe static int __io_recvmsg_copy_hdr(struct io_kiocb *req, 6047ed29b0b4SJens Axboe struct io_async_msghdr *iomsg) 6048ed29b0b4SJens Axboe { 60498ff86d85SJens Axboe struct io_sr_msg *sr = io_kiocb_to_cmd(req); 6050ed29b0b4SJens Axboe struct iovec __user *uiov; 6051ed29b0b4SJens Axboe size_t iov_len; 6052ed29b0b4SJens Axboe int ret; 6053ed29b0b4SJens Axboe 6054ed29b0b4SJens Axboe ret = __copy_msghdr_from_user(&iomsg->msg, sr->umsg, 6055ed29b0b4SJens Axboe &iomsg->uaddr, &uiov, &iov_len); 6056ed29b0b4SJens Axboe if (ret) 6057ed29b0b4SJens Axboe return ret; 6058ed29b0b4SJens Axboe 6059ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_SELECT) { 6060ed29b0b4SJens Axboe if (iov_len > 1) 6061ed29b0b4SJens Axboe return -EINVAL; 6062ed29b0b4SJens Axboe if (copy_from_user(iomsg->fast_iov, uiov, sizeof(*uiov))) 6063ed29b0b4SJens Axboe return -EFAULT; 6064ed29b0b4SJens Axboe sr->len = iomsg->fast_iov[0].iov_len; 6065ed29b0b4SJens Axboe iomsg->free_iov = NULL; 6066ed29b0b4SJens Axboe } else { 6067ed29b0b4SJens Axboe iomsg->free_iov = iomsg->fast_iov; 6068ed29b0b4SJens Axboe ret = __import_iovec(READ, uiov, iov_len, UIO_FASTIOV, 6069ed29b0b4SJens Axboe &iomsg->free_iov, &iomsg->msg.msg_iter, 6070ed29b0b4SJens Axboe false); 6071ed29b0b4SJens Axboe if (ret > 0) 6072ed29b0b4SJens Axboe ret = 0; 6073ed29b0b4SJens Axboe } 6074ed29b0b4SJens Axboe 6075ed29b0b4SJens Axboe return ret; 6076ed29b0b4SJens Axboe } 6077ed29b0b4SJens Axboe 6078ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 6079ed29b0b4SJens Axboe static int __io_compat_recvmsg_copy_hdr(struct io_kiocb *req, 6080ed29b0b4SJens Axboe struct io_async_msghdr *iomsg) 6081ed29b0b4SJens Axboe { 60828ff86d85SJens Axboe struct io_sr_msg *sr = io_kiocb_to_cmd(req); 6083ed29b0b4SJens Axboe struct compat_iovec __user *uiov; 6084ed29b0b4SJens Axboe compat_uptr_t ptr; 6085ed29b0b4SJens Axboe compat_size_t len; 6086ed29b0b4SJens Axboe int ret; 6087ed29b0b4SJens Axboe 6088ed29b0b4SJens Axboe ret = __get_compat_msghdr(&iomsg->msg, sr->umsg_compat, &iomsg->uaddr, 6089ed29b0b4SJens Axboe &ptr, &len); 6090ed29b0b4SJens Axboe if (ret) 6091ed29b0b4SJens Axboe return ret; 6092ed29b0b4SJens Axboe 6093ed29b0b4SJens Axboe uiov = compat_ptr(ptr); 6094ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_SELECT) { 6095ed29b0b4SJens Axboe compat_ssize_t clen; 6096ed29b0b4SJens Axboe 6097ed29b0b4SJens Axboe if (len > 1) 6098ed29b0b4SJens Axboe return -EINVAL; 6099ed29b0b4SJens Axboe if (!access_ok(uiov, sizeof(*uiov))) 6100ed29b0b4SJens Axboe return -EFAULT; 6101ed29b0b4SJens Axboe if (__get_user(clen, &uiov->iov_len)) 6102ed29b0b4SJens Axboe return -EFAULT; 6103ed29b0b4SJens Axboe if (clen < 0) 6104ed29b0b4SJens Axboe return -EINVAL; 6105ed29b0b4SJens Axboe sr->len = clen; 6106ed29b0b4SJens Axboe iomsg->free_iov = NULL; 6107ed29b0b4SJens Axboe } else { 6108ed29b0b4SJens Axboe iomsg->free_iov = iomsg->fast_iov; 6109ed29b0b4SJens Axboe ret = __import_iovec(READ, (struct iovec __user *)uiov, len, 6110ed29b0b4SJens Axboe UIO_FASTIOV, &iomsg->free_iov, 6111ed29b0b4SJens Axboe &iomsg->msg.msg_iter, true); 6112ed29b0b4SJens Axboe if (ret < 0) 6113ed29b0b4SJens Axboe return ret; 6114ed29b0b4SJens Axboe } 6115ed29b0b4SJens Axboe 6116ed29b0b4SJens Axboe return 0; 6117ed29b0b4SJens Axboe } 6118ed29b0b4SJens Axboe #endif 6119ed29b0b4SJens Axboe 6120ed29b0b4SJens Axboe static int io_recvmsg_copy_hdr(struct io_kiocb *req, 6121ed29b0b4SJens Axboe struct io_async_msghdr *iomsg) 6122ed29b0b4SJens Axboe { 6123ed29b0b4SJens Axboe iomsg->msg.msg_name = &iomsg->addr; 6124ed29b0b4SJens Axboe 6125ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 6126ed29b0b4SJens Axboe if (req->ctx->compat) 6127ed29b0b4SJens Axboe return __io_compat_recvmsg_copy_hdr(req, iomsg); 6128ed29b0b4SJens Axboe #endif 6129ed29b0b4SJens Axboe 6130ed29b0b4SJens Axboe return __io_recvmsg_copy_hdr(req, iomsg); 6131ed29b0b4SJens Axboe } 6132ed29b0b4SJens Axboe 6133ed29b0b4SJens Axboe static int io_recvmsg_prep_async(struct io_kiocb *req) 6134ed29b0b4SJens Axboe { 6135ed29b0b4SJens Axboe int ret; 6136ed29b0b4SJens Axboe 6137ed29b0b4SJens Axboe ret = io_recvmsg_copy_hdr(req, req->async_data); 6138ed29b0b4SJens Axboe if (!ret) 6139ed29b0b4SJens Axboe req->flags |= REQ_F_NEED_CLEANUP; 6140ed29b0b4SJens Axboe return ret; 6141ed29b0b4SJens Axboe } 6142ed29b0b4SJens Axboe 6143ed29b0b4SJens Axboe static int io_recvmsg_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 6144ed29b0b4SJens Axboe { 61458ff86d85SJens Axboe struct io_sr_msg *sr = io_kiocb_to_cmd(req); 6146ed29b0b4SJens Axboe 6147ed29b0b4SJens Axboe if (unlikely(sqe->file_index || sqe->addr2)) 6148ed29b0b4SJens Axboe return -EINVAL; 6149ed29b0b4SJens Axboe 6150ed29b0b4SJens Axboe sr->umsg = u64_to_user_ptr(READ_ONCE(sqe->addr)); 6151ed29b0b4SJens Axboe sr->len = READ_ONCE(sqe->len); 6152ed29b0b4SJens Axboe sr->flags = READ_ONCE(sqe->ioprio); 6153ed29b0b4SJens Axboe if (sr->flags & ~IORING_RECVSEND_POLL_FIRST) 6154ed29b0b4SJens Axboe return -EINVAL; 6155ed29b0b4SJens Axboe sr->msg_flags = READ_ONCE(sqe->msg_flags) | MSG_NOSIGNAL; 6156ed29b0b4SJens Axboe if (sr->msg_flags & MSG_DONTWAIT) 6157ed29b0b4SJens Axboe req->flags |= REQ_F_NOWAIT; 6158bd8587e4SJens Axboe if (sr->msg_flags & MSG_ERRQUEUE) 6159bd8587e4SJens Axboe req->flags |= REQ_F_CLEAR_POLLIN; 6160ed29b0b4SJens Axboe 6161ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 6162ed29b0b4SJens Axboe if (req->ctx->compat) 6163ed29b0b4SJens Axboe sr->msg_flags |= MSG_CMSG_COMPAT; 6164ed29b0b4SJens Axboe #endif 6165ed29b0b4SJens Axboe sr->done_io = 0; 6166ed29b0b4SJens Axboe return 0; 6167ed29b0b4SJens Axboe } 6168ed29b0b4SJens Axboe 6169ed29b0b4SJens Axboe static int io_recvmsg(struct io_kiocb *req, unsigned int issue_flags) 6170ed29b0b4SJens Axboe { 61718ff86d85SJens Axboe struct io_sr_msg *sr = io_kiocb_to_cmd(req); 6172ed29b0b4SJens Axboe struct io_async_msghdr iomsg, *kmsg; 6173ed29b0b4SJens Axboe struct socket *sock; 6174ed29b0b4SJens Axboe unsigned int cflags; 6175ed29b0b4SJens Axboe unsigned flags; 6176ed29b0b4SJens Axboe int ret, min_ret = 0; 6177ed29b0b4SJens Axboe bool force_nonblock = issue_flags & IO_URING_F_NONBLOCK; 6178ed29b0b4SJens Axboe 6179ed29b0b4SJens Axboe sock = sock_from_file(req->file); 6180ed29b0b4SJens Axboe if (unlikely(!sock)) 6181ed29b0b4SJens Axboe return -ENOTSOCK; 6182ed29b0b4SJens Axboe 6183ed29b0b4SJens Axboe if (req_has_async_data(req)) { 6184ed29b0b4SJens Axboe kmsg = req->async_data; 6185ed29b0b4SJens Axboe } else { 6186ed29b0b4SJens Axboe ret = io_recvmsg_copy_hdr(req, &iomsg); 6187ed29b0b4SJens Axboe if (ret) 6188ed29b0b4SJens Axboe return ret; 6189ed29b0b4SJens Axboe kmsg = &iomsg; 6190ed29b0b4SJens Axboe } 6191ed29b0b4SJens Axboe 6192ed29b0b4SJens Axboe if (!(req->flags & REQ_F_POLLED) && 6193ed29b0b4SJens Axboe (sr->flags & IORING_RECVSEND_POLL_FIRST)) 6194ed29b0b4SJens Axboe return io_setup_async_msg(req, kmsg); 6195ed29b0b4SJens Axboe 6196ed29b0b4SJens Axboe if (io_do_buffer_select(req)) { 6197ed29b0b4SJens Axboe void __user *buf; 6198ed29b0b4SJens Axboe 6199ed29b0b4SJens Axboe buf = io_buffer_select(req, &sr->len, issue_flags); 6200ed29b0b4SJens Axboe if (!buf) 6201ed29b0b4SJens Axboe return -ENOBUFS; 6202ed29b0b4SJens Axboe kmsg->fast_iov[0].iov_base = buf; 6203ed29b0b4SJens Axboe kmsg->fast_iov[0].iov_len = sr->len; 6204ed29b0b4SJens Axboe iov_iter_init(&kmsg->msg.msg_iter, READ, kmsg->fast_iov, 1, 6205ed29b0b4SJens Axboe sr->len); 6206ed29b0b4SJens Axboe } 6207ed29b0b4SJens Axboe 6208ed29b0b4SJens Axboe flags = sr->msg_flags; 6209ed29b0b4SJens Axboe if (force_nonblock) 6210ed29b0b4SJens Axboe flags |= MSG_DONTWAIT; 6211ed29b0b4SJens Axboe if (flags & MSG_WAITALL) 6212ed29b0b4SJens Axboe min_ret = iov_iter_count(&kmsg->msg.msg_iter); 6213ed29b0b4SJens Axboe 6214ed29b0b4SJens Axboe kmsg->msg.msg_get_inq = 1; 6215ed29b0b4SJens Axboe ret = __sys_recvmsg_sock(sock, &kmsg->msg, sr->umsg, kmsg->uaddr, flags); 6216ed29b0b4SJens Axboe if (ret < min_ret) { 6217ed29b0b4SJens Axboe if (ret == -EAGAIN && force_nonblock) 6218ed29b0b4SJens Axboe return io_setup_async_msg(req, kmsg); 6219ed29b0b4SJens Axboe if (ret == -ERESTARTSYS) 6220ed29b0b4SJens Axboe ret = -EINTR; 6221ed29b0b4SJens Axboe if (ret > 0 && io_net_retry(sock, flags)) { 6222ed29b0b4SJens Axboe sr->done_io += ret; 6223ed29b0b4SJens Axboe req->flags |= REQ_F_PARTIAL_IO; 6224ed29b0b4SJens Axboe return io_setup_async_msg(req, kmsg); 6225ed29b0b4SJens Axboe } 6226ed29b0b4SJens Axboe req_set_fail(req); 6227ed29b0b4SJens Axboe } else if ((flags & MSG_WAITALL) && (kmsg->msg.msg_flags & (MSG_TRUNC | MSG_CTRUNC))) { 6228ed29b0b4SJens Axboe req_set_fail(req); 6229ed29b0b4SJens Axboe } 6230ed29b0b4SJens Axboe 6231ed29b0b4SJens Axboe /* fast path, check for non-NULL to avoid function call */ 6232ed29b0b4SJens Axboe if (kmsg->free_iov) 6233ed29b0b4SJens Axboe kfree(kmsg->free_iov); 6234ed29b0b4SJens Axboe req->flags &= ~REQ_F_NEED_CLEANUP; 6235ed29b0b4SJens Axboe if (ret >= 0) 6236ed29b0b4SJens Axboe ret += sr->done_io; 6237ed29b0b4SJens Axboe else if (sr->done_io) 6238ed29b0b4SJens Axboe ret = sr->done_io; 6239ed29b0b4SJens Axboe cflags = io_put_kbuf(req, issue_flags); 6240ed29b0b4SJens Axboe if (kmsg->msg.msg_inq) 6241ed29b0b4SJens Axboe cflags |= IORING_CQE_F_SOCK_NONEMPTY; 6242ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, cflags); 6243ed29b0b4SJens Axboe return 0; 6244ed29b0b4SJens Axboe } 6245ed29b0b4SJens Axboe 6246ed29b0b4SJens Axboe static int io_recv(struct io_kiocb *req, unsigned int issue_flags) 6247ed29b0b4SJens Axboe { 62488ff86d85SJens Axboe struct io_sr_msg *sr = io_kiocb_to_cmd(req); 6249ed29b0b4SJens Axboe struct msghdr msg; 6250ed29b0b4SJens Axboe struct socket *sock; 6251ed29b0b4SJens Axboe struct iovec iov; 6252ed29b0b4SJens Axboe unsigned int cflags; 6253ed29b0b4SJens Axboe unsigned flags; 6254ed29b0b4SJens Axboe int ret, min_ret = 0; 6255ed29b0b4SJens Axboe bool force_nonblock = issue_flags & IO_URING_F_NONBLOCK; 6256ed29b0b4SJens Axboe 6257ed29b0b4SJens Axboe if (!(req->flags & REQ_F_POLLED) && 6258ed29b0b4SJens Axboe (sr->flags & IORING_RECVSEND_POLL_FIRST)) 6259ed29b0b4SJens Axboe return -EAGAIN; 6260ed29b0b4SJens Axboe 6261ed29b0b4SJens Axboe sock = sock_from_file(req->file); 6262ed29b0b4SJens Axboe if (unlikely(!sock)) 6263ed29b0b4SJens Axboe return -ENOTSOCK; 6264ed29b0b4SJens Axboe 6265ed29b0b4SJens Axboe if (io_do_buffer_select(req)) { 6266ed29b0b4SJens Axboe void __user *buf; 6267ed29b0b4SJens Axboe 6268ed29b0b4SJens Axboe buf = io_buffer_select(req, &sr->len, issue_flags); 6269ed29b0b4SJens Axboe if (!buf) 6270ed29b0b4SJens Axboe return -ENOBUFS; 6271ed29b0b4SJens Axboe sr->buf = buf; 6272ed29b0b4SJens Axboe } 6273ed29b0b4SJens Axboe 6274ed29b0b4SJens Axboe ret = import_single_range(READ, sr->buf, sr->len, &iov, &msg.msg_iter); 6275ed29b0b4SJens Axboe if (unlikely(ret)) 6276ed29b0b4SJens Axboe goto out_free; 6277ed29b0b4SJens Axboe 6278ed29b0b4SJens Axboe msg.msg_name = NULL; 6279ed29b0b4SJens Axboe msg.msg_namelen = 0; 6280ed29b0b4SJens Axboe msg.msg_control = NULL; 6281ed29b0b4SJens Axboe msg.msg_get_inq = 1; 6282ed29b0b4SJens Axboe msg.msg_flags = 0; 6283ed29b0b4SJens Axboe msg.msg_controllen = 0; 6284ed29b0b4SJens Axboe msg.msg_iocb = NULL; 6285ed29b0b4SJens Axboe 6286ed29b0b4SJens Axboe flags = sr->msg_flags; 6287ed29b0b4SJens Axboe if (force_nonblock) 6288ed29b0b4SJens Axboe flags |= MSG_DONTWAIT; 6289ed29b0b4SJens Axboe if (flags & MSG_WAITALL) 6290ed29b0b4SJens Axboe min_ret = iov_iter_count(&msg.msg_iter); 6291ed29b0b4SJens Axboe 6292ed29b0b4SJens Axboe ret = sock_recvmsg(sock, &msg, flags); 6293ed29b0b4SJens Axboe if (ret < min_ret) { 6294ed29b0b4SJens Axboe if (ret == -EAGAIN && force_nonblock) 6295ed29b0b4SJens Axboe return -EAGAIN; 6296ed29b0b4SJens Axboe if (ret == -ERESTARTSYS) 6297ed29b0b4SJens Axboe ret = -EINTR; 6298ed29b0b4SJens Axboe if (ret > 0 && io_net_retry(sock, flags)) { 6299ed29b0b4SJens Axboe sr->len -= ret; 6300ed29b0b4SJens Axboe sr->buf += ret; 6301ed29b0b4SJens Axboe sr->done_io += ret; 6302ed29b0b4SJens Axboe req->flags |= REQ_F_PARTIAL_IO; 6303ed29b0b4SJens Axboe return -EAGAIN; 6304ed29b0b4SJens Axboe } 6305ed29b0b4SJens Axboe req_set_fail(req); 6306ed29b0b4SJens Axboe } else if ((flags & MSG_WAITALL) && (msg.msg_flags & (MSG_TRUNC | MSG_CTRUNC))) { 6307ed29b0b4SJens Axboe out_free: 6308ed29b0b4SJens Axboe req_set_fail(req); 6309ed29b0b4SJens Axboe } 6310ed29b0b4SJens Axboe 6311ed29b0b4SJens Axboe if (ret >= 0) 6312ed29b0b4SJens Axboe ret += sr->done_io; 6313ed29b0b4SJens Axboe else if (sr->done_io) 6314ed29b0b4SJens Axboe ret = sr->done_io; 6315ed29b0b4SJens Axboe cflags = io_put_kbuf(req, issue_flags); 6316ed29b0b4SJens Axboe if (msg.msg_inq) 6317ed29b0b4SJens Axboe cflags |= IORING_CQE_F_SOCK_NONEMPTY; 6318ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, cflags); 6319ed29b0b4SJens Axboe return 0; 6320ed29b0b4SJens Axboe } 6321ed29b0b4SJens Axboe 6322ed29b0b4SJens Axboe static int io_accept_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 6323ed29b0b4SJens Axboe { 63248ff86d85SJens Axboe struct io_accept *accept = io_kiocb_to_cmd(req); 6325ed29b0b4SJens Axboe unsigned flags; 6326ed29b0b4SJens Axboe 6327ed29b0b4SJens Axboe if (sqe->len || sqe->buf_index) 6328ed29b0b4SJens Axboe return -EINVAL; 6329ed29b0b4SJens Axboe 6330ed29b0b4SJens Axboe accept->addr = u64_to_user_ptr(READ_ONCE(sqe->addr)); 6331ed29b0b4SJens Axboe accept->addr_len = u64_to_user_ptr(READ_ONCE(sqe->addr2)); 6332ed29b0b4SJens Axboe accept->flags = READ_ONCE(sqe->accept_flags); 6333ed29b0b4SJens Axboe accept->nofile = rlimit(RLIMIT_NOFILE); 6334ed29b0b4SJens Axboe flags = READ_ONCE(sqe->ioprio); 6335ed29b0b4SJens Axboe if (flags & ~IORING_ACCEPT_MULTISHOT) 6336ed29b0b4SJens Axboe return -EINVAL; 6337ed29b0b4SJens Axboe 6338ed29b0b4SJens Axboe accept->file_slot = READ_ONCE(sqe->file_index); 6339ed29b0b4SJens Axboe if (accept->file_slot) { 6340ed29b0b4SJens Axboe if (accept->flags & SOCK_CLOEXEC) 6341ed29b0b4SJens Axboe return -EINVAL; 6342ed29b0b4SJens Axboe if (flags & IORING_ACCEPT_MULTISHOT && 6343ed29b0b4SJens Axboe accept->file_slot != IORING_FILE_INDEX_ALLOC) 6344ed29b0b4SJens Axboe return -EINVAL; 6345ed29b0b4SJens Axboe } 6346ed29b0b4SJens Axboe if (accept->flags & ~(SOCK_CLOEXEC | SOCK_NONBLOCK)) 6347ed29b0b4SJens Axboe return -EINVAL; 6348ed29b0b4SJens Axboe if (SOCK_NONBLOCK != O_NONBLOCK && (accept->flags & SOCK_NONBLOCK)) 6349ed29b0b4SJens Axboe accept->flags = (accept->flags & ~SOCK_NONBLOCK) | O_NONBLOCK; 6350ed29b0b4SJens Axboe if (flags & IORING_ACCEPT_MULTISHOT) 6351ed29b0b4SJens Axboe req->flags |= REQ_F_APOLL_MULTISHOT; 6352ed29b0b4SJens Axboe return 0; 6353ed29b0b4SJens Axboe } 6354ed29b0b4SJens Axboe 6355ed29b0b4SJens Axboe static int io_accept(struct io_kiocb *req, unsigned int issue_flags) 6356ed29b0b4SJens Axboe { 6357ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 63588ff86d85SJens Axboe struct io_accept *accept = io_kiocb_to_cmd(req); 6359ed29b0b4SJens Axboe bool force_nonblock = issue_flags & IO_URING_F_NONBLOCK; 6360ed29b0b4SJens Axboe unsigned int file_flags = force_nonblock ? O_NONBLOCK : 0; 6361ed29b0b4SJens Axboe bool fixed = !!accept->file_slot; 6362ed29b0b4SJens Axboe struct file *file; 6363ed29b0b4SJens Axboe int ret, fd; 6364ed29b0b4SJens Axboe 6365ed29b0b4SJens Axboe retry: 6366ed29b0b4SJens Axboe if (!fixed) { 6367ed29b0b4SJens Axboe fd = __get_unused_fd_flags(accept->flags, accept->nofile); 6368ed29b0b4SJens Axboe if (unlikely(fd < 0)) 6369ed29b0b4SJens Axboe return fd; 6370ed29b0b4SJens Axboe } 6371ed29b0b4SJens Axboe file = do_accept(req->file, file_flags, accept->addr, accept->addr_len, 6372ed29b0b4SJens Axboe accept->flags); 6373ed29b0b4SJens Axboe if (IS_ERR(file)) { 6374ed29b0b4SJens Axboe if (!fixed) 6375ed29b0b4SJens Axboe put_unused_fd(fd); 6376ed29b0b4SJens Axboe ret = PTR_ERR(file); 6377ed29b0b4SJens Axboe if (ret == -EAGAIN && force_nonblock) { 6378ed29b0b4SJens Axboe /* 6379ed29b0b4SJens Axboe * if it's multishot and polled, we don't need to 6380ed29b0b4SJens Axboe * return EAGAIN to arm the poll infra since it 6381ed29b0b4SJens Axboe * has already been done 6382ed29b0b4SJens Axboe */ 6383ed29b0b4SJens Axboe if ((req->flags & IO_APOLL_MULTI_POLLED) == 6384ed29b0b4SJens Axboe IO_APOLL_MULTI_POLLED) 6385ed29b0b4SJens Axboe ret = 0; 6386ed29b0b4SJens Axboe return ret; 6387ed29b0b4SJens Axboe } 6388ed29b0b4SJens Axboe if (ret == -ERESTARTSYS) 6389ed29b0b4SJens Axboe ret = -EINTR; 6390ed29b0b4SJens Axboe req_set_fail(req); 6391ed29b0b4SJens Axboe } else if (!fixed) { 6392ed29b0b4SJens Axboe fd_install(fd, file); 6393ed29b0b4SJens Axboe ret = fd; 6394ed29b0b4SJens Axboe } else { 6395ed29b0b4SJens Axboe ret = io_fixed_fd_install(req, issue_flags, file, 6396ed29b0b4SJens Axboe accept->file_slot); 6397ed29b0b4SJens Axboe } 6398ed29b0b4SJens Axboe 6399ed29b0b4SJens Axboe if (!(req->flags & REQ_F_APOLL_MULTISHOT)) { 6400ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 6401ed29b0b4SJens Axboe return 0; 6402ed29b0b4SJens Axboe } 6403ed29b0b4SJens Axboe if (ret >= 0) { 6404ed29b0b4SJens Axboe bool filled; 6405ed29b0b4SJens Axboe 6406ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 6407ed29b0b4SJens Axboe filled = io_fill_cqe_aux(ctx, req->cqe.user_data, ret, 6408ed29b0b4SJens Axboe IORING_CQE_F_MORE); 6409ed29b0b4SJens Axboe io_commit_cqring(ctx); 6410ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 6411ed29b0b4SJens Axboe if (filled) { 6412ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 6413ed29b0b4SJens Axboe goto retry; 6414ed29b0b4SJens Axboe } 6415ed29b0b4SJens Axboe ret = -ECANCELED; 6416ed29b0b4SJens Axboe } 6417ed29b0b4SJens Axboe 6418ed29b0b4SJens Axboe return ret; 6419ed29b0b4SJens Axboe } 6420ed29b0b4SJens Axboe 6421ed29b0b4SJens Axboe static int io_socket_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 6422ed29b0b4SJens Axboe { 64238ff86d85SJens Axboe struct io_socket *sock = io_kiocb_to_cmd(req); 6424ed29b0b4SJens Axboe 6425ed29b0b4SJens Axboe if (sqe->addr || sqe->rw_flags || sqe->buf_index) 6426ed29b0b4SJens Axboe return -EINVAL; 6427ed29b0b4SJens Axboe 6428ed29b0b4SJens Axboe sock->domain = READ_ONCE(sqe->fd); 6429ed29b0b4SJens Axboe sock->type = READ_ONCE(sqe->off); 6430ed29b0b4SJens Axboe sock->protocol = READ_ONCE(sqe->len); 6431ed29b0b4SJens Axboe sock->file_slot = READ_ONCE(sqe->file_index); 6432ed29b0b4SJens Axboe sock->nofile = rlimit(RLIMIT_NOFILE); 6433ed29b0b4SJens Axboe 6434ed29b0b4SJens Axboe sock->flags = sock->type & ~SOCK_TYPE_MASK; 6435ed29b0b4SJens Axboe if (sock->file_slot && (sock->flags & SOCK_CLOEXEC)) 6436ed29b0b4SJens Axboe return -EINVAL; 6437ed29b0b4SJens Axboe if (sock->flags & ~(SOCK_CLOEXEC | SOCK_NONBLOCK)) 6438ed29b0b4SJens Axboe return -EINVAL; 6439ed29b0b4SJens Axboe return 0; 6440ed29b0b4SJens Axboe } 6441ed29b0b4SJens Axboe 6442ed29b0b4SJens Axboe static int io_socket(struct io_kiocb *req, unsigned int issue_flags) 6443ed29b0b4SJens Axboe { 64448ff86d85SJens Axboe struct io_socket *sock = io_kiocb_to_cmd(req); 6445ed29b0b4SJens Axboe bool fixed = !!sock->file_slot; 6446ed29b0b4SJens Axboe struct file *file; 6447ed29b0b4SJens Axboe int ret, fd; 6448ed29b0b4SJens Axboe 6449ed29b0b4SJens Axboe if (!fixed) { 6450ed29b0b4SJens Axboe fd = __get_unused_fd_flags(sock->flags, sock->nofile); 6451ed29b0b4SJens Axboe if (unlikely(fd < 0)) 6452ed29b0b4SJens Axboe return fd; 6453ed29b0b4SJens Axboe } 6454ed29b0b4SJens Axboe file = __sys_socket_file(sock->domain, sock->type, sock->protocol); 6455ed29b0b4SJens Axboe if (IS_ERR(file)) { 6456ed29b0b4SJens Axboe if (!fixed) 6457ed29b0b4SJens Axboe put_unused_fd(fd); 6458ed29b0b4SJens Axboe ret = PTR_ERR(file); 6459ed29b0b4SJens Axboe if (ret == -EAGAIN && (issue_flags & IO_URING_F_NONBLOCK)) 6460ed29b0b4SJens Axboe return -EAGAIN; 6461ed29b0b4SJens Axboe if (ret == -ERESTARTSYS) 6462ed29b0b4SJens Axboe ret = -EINTR; 6463ed29b0b4SJens Axboe req_set_fail(req); 6464ed29b0b4SJens Axboe } else if (!fixed) { 6465ed29b0b4SJens Axboe fd_install(fd, file); 6466ed29b0b4SJens Axboe ret = fd; 6467ed29b0b4SJens Axboe } else { 6468ed29b0b4SJens Axboe ret = io_fixed_fd_install(req, issue_flags, file, 6469ed29b0b4SJens Axboe sock->file_slot); 6470ed29b0b4SJens Axboe } 6471ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 6472ed29b0b4SJens Axboe return 0; 6473ed29b0b4SJens Axboe } 6474ed29b0b4SJens Axboe 6475ed29b0b4SJens Axboe static int io_connect_prep_async(struct io_kiocb *req) 6476ed29b0b4SJens Axboe { 6477ed29b0b4SJens Axboe struct io_async_connect *io = req->async_data; 64788ff86d85SJens Axboe struct io_connect *conn = io_kiocb_to_cmd(req); 6479ed29b0b4SJens Axboe 6480ed29b0b4SJens Axboe return move_addr_to_kernel(conn->addr, conn->addr_len, &io->address); 6481ed29b0b4SJens Axboe } 6482ed29b0b4SJens Axboe 6483ed29b0b4SJens Axboe static int io_connect_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 6484ed29b0b4SJens Axboe { 64858ff86d85SJens Axboe struct io_connect *conn = io_kiocb_to_cmd(req); 6486ed29b0b4SJens Axboe 6487ed29b0b4SJens Axboe if (sqe->len || sqe->buf_index || sqe->rw_flags || sqe->splice_fd_in) 6488ed29b0b4SJens Axboe return -EINVAL; 6489ed29b0b4SJens Axboe 6490ed29b0b4SJens Axboe conn->addr = u64_to_user_ptr(READ_ONCE(sqe->addr)); 6491ed29b0b4SJens Axboe conn->addr_len = READ_ONCE(sqe->addr2); 6492ed29b0b4SJens Axboe return 0; 6493ed29b0b4SJens Axboe } 6494ed29b0b4SJens Axboe 6495ed29b0b4SJens Axboe static int io_connect(struct io_kiocb *req, unsigned int issue_flags) 6496ed29b0b4SJens Axboe { 64978ff86d85SJens Axboe struct io_connect *connect = io_kiocb_to_cmd(req); 6498ed29b0b4SJens Axboe struct io_async_connect __io, *io; 6499ed29b0b4SJens Axboe unsigned file_flags; 6500ed29b0b4SJens Axboe int ret; 6501ed29b0b4SJens Axboe bool force_nonblock = issue_flags & IO_URING_F_NONBLOCK; 6502ed29b0b4SJens Axboe 6503ed29b0b4SJens Axboe if (req_has_async_data(req)) { 6504ed29b0b4SJens Axboe io = req->async_data; 6505ed29b0b4SJens Axboe } else { 65068ff86d85SJens Axboe ret = move_addr_to_kernel(connect->addr, 65078ff86d85SJens Axboe connect->addr_len, 6508ed29b0b4SJens Axboe &__io.address); 6509ed29b0b4SJens Axboe if (ret) 6510ed29b0b4SJens Axboe goto out; 6511ed29b0b4SJens Axboe io = &__io; 6512ed29b0b4SJens Axboe } 6513ed29b0b4SJens Axboe 6514ed29b0b4SJens Axboe file_flags = force_nonblock ? O_NONBLOCK : 0; 6515ed29b0b4SJens Axboe 6516ed29b0b4SJens Axboe ret = __sys_connect_file(req->file, &io->address, 65178ff86d85SJens Axboe connect->addr_len, file_flags); 6518ed29b0b4SJens Axboe if ((ret == -EAGAIN || ret == -EINPROGRESS) && force_nonblock) { 6519ed29b0b4SJens Axboe if (req_has_async_data(req)) 6520ed29b0b4SJens Axboe return -EAGAIN; 6521ed29b0b4SJens Axboe if (io_alloc_async_data(req)) { 6522ed29b0b4SJens Axboe ret = -ENOMEM; 6523ed29b0b4SJens Axboe goto out; 6524ed29b0b4SJens Axboe } 6525ed29b0b4SJens Axboe memcpy(req->async_data, &__io, sizeof(__io)); 6526ed29b0b4SJens Axboe return -EAGAIN; 6527ed29b0b4SJens Axboe } 6528ed29b0b4SJens Axboe if (ret == -ERESTARTSYS) 6529ed29b0b4SJens Axboe ret = -EINTR; 6530ed29b0b4SJens Axboe out: 6531ed29b0b4SJens Axboe if (ret < 0) 6532ed29b0b4SJens Axboe req_set_fail(req); 6533ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 6534ed29b0b4SJens Axboe return 0; 6535ed29b0b4SJens Axboe } 6536ed29b0b4SJens Axboe #else /* !CONFIG_NET */ 6537ed29b0b4SJens Axboe #define IO_NETOP_FN(op) \ 6538ed29b0b4SJens Axboe static int io_##op(struct io_kiocb *req, unsigned int issue_flags) \ 6539ed29b0b4SJens Axboe { \ 6540ed29b0b4SJens Axboe return -EOPNOTSUPP; \ 6541ed29b0b4SJens Axboe } 6542ed29b0b4SJens Axboe 6543ed29b0b4SJens Axboe #define IO_NETOP_PREP(op) \ 6544ed29b0b4SJens Axboe IO_NETOP_FN(op) \ 6545ed29b0b4SJens Axboe static int io_##op##_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) \ 6546ed29b0b4SJens Axboe { \ 6547ed29b0b4SJens Axboe return -EOPNOTSUPP; \ 6548ed29b0b4SJens Axboe } \ 6549ed29b0b4SJens Axboe 6550ed29b0b4SJens Axboe #define IO_NETOP_PREP_ASYNC(op) \ 6551ed29b0b4SJens Axboe IO_NETOP_PREP(op) \ 6552ed29b0b4SJens Axboe static int io_##op##_prep_async(struct io_kiocb *req) \ 6553ed29b0b4SJens Axboe { \ 6554ed29b0b4SJens Axboe return -EOPNOTSUPP; \ 6555ed29b0b4SJens Axboe } 6556ed29b0b4SJens Axboe 6557ed29b0b4SJens Axboe IO_NETOP_PREP_ASYNC(sendmsg); 6558ed29b0b4SJens Axboe IO_NETOP_PREP_ASYNC(recvmsg); 6559ed29b0b4SJens Axboe IO_NETOP_PREP_ASYNC(connect); 6560ed29b0b4SJens Axboe IO_NETOP_PREP(accept); 6561ed29b0b4SJens Axboe IO_NETOP_PREP(socket); 6562ed29b0b4SJens Axboe IO_NETOP_PREP(shutdown); 6563ed29b0b4SJens Axboe IO_NETOP_FN(send); 6564ed29b0b4SJens Axboe IO_NETOP_FN(recv); 6565ed29b0b4SJens Axboe #endif /* CONFIG_NET */ 6566ed29b0b4SJens Axboe 6567ed29b0b4SJens Axboe struct io_poll_table { 6568ed29b0b4SJens Axboe struct poll_table_struct pt; 6569ed29b0b4SJens Axboe struct io_kiocb *req; 6570ed29b0b4SJens Axboe int nr_entries; 6571ed29b0b4SJens Axboe int error; 6572ed29b0b4SJens Axboe }; 6573ed29b0b4SJens Axboe 6574ed29b0b4SJens Axboe #define IO_POLL_CANCEL_FLAG BIT(31) 6575ed29b0b4SJens Axboe #define IO_POLL_REF_MASK GENMASK(30, 0) 6576ed29b0b4SJens Axboe 6577ed29b0b4SJens Axboe /* 6578ed29b0b4SJens Axboe * If refs part of ->poll_refs (see IO_POLL_REF_MASK) is 0, it's free. We can 6579ed29b0b4SJens Axboe * bump it and acquire ownership. It's disallowed to modify requests while not 6580ed29b0b4SJens Axboe * owning it, that prevents from races for enqueueing task_work's and b/w 6581ed29b0b4SJens Axboe * arming poll and wakeups. 6582ed29b0b4SJens Axboe */ 6583ed29b0b4SJens Axboe static inline bool io_poll_get_ownership(struct io_kiocb *req) 6584ed29b0b4SJens Axboe { 6585ed29b0b4SJens Axboe return !(atomic_fetch_inc(&req->poll_refs) & IO_POLL_REF_MASK); 6586ed29b0b4SJens Axboe } 6587ed29b0b4SJens Axboe 6588ed29b0b4SJens Axboe static void io_poll_mark_cancelled(struct io_kiocb *req) 6589ed29b0b4SJens Axboe { 6590ed29b0b4SJens Axboe atomic_or(IO_POLL_CANCEL_FLAG, &req->poll_refs); 6591ed29b0b4SJens Axboe } 6592ed29b0b4SJens Axboe 65938d4388d1SJens Axboe static struct io_poll *io_poll_get_double(struct io_kiocb *req) 6594ed29b0b4SJens Axboe { 6595ed29b0b4SJens Axboe /* pure poll stashes this in ->async_data, poll driven retry elsewhere */ 6596ed29b0b4SJens Axboe if (req->opcode == IORING_OP_POLL_ADD) 6597ed29b0b4SJens Axboe return req->async_data; 6598ed29b0b4SJens Axboe return req->apoll->double_poll; 6599ed29b0b4SJens Axboe } 6600ed29b0b4SJens Axboe 66018d4388d1SJens Axboe static struct io_poll *io_poll_get_single(struct io_kiocb *req) 6602ed29b0b4SJens Axboe { 6603ed29b0b4SJens Axboe if (req->opcode == IORING_OP_POLL_ADD) 66048d4388d1SJens Axboe return io_kiocb_to_cmd(req); 6605ed29b0b4SJens Axboe return &req->apoll->poll; 6606ed29b0b4SJens Axboe } 6607ed29b0b4SJens Axboe 6608ed29b0b4SJens Axboe static void io_poll_req_insert(struct io_kiocb *req) 6609ed29b0b4SJens Axboe { 6610ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 6611ed29b0b4SJens Axboe struct hlist_head *list; 6612ed29b0b4SJens Axboe 6613ed29b0b4SJens Axboe list = &ctx->cancel_hash[hash_long(req->cqe.user_data, ctx->cancel_hash_bits)]; 6614ed29b0b4SJens Axboe hlist_add_head(&req->hash_node, list); 6615ed29b0b4SJens Axboe } 6616ed29b0b4SJens Axboe 66178d4388d1SJens Axboe static void io_init_poll_iocb(struct io_poll *poll, __poll_t events, 6618ed29b0b4SJens Axboe wait_queue_func_t wake_func) 6619ed29b0b4SJens Axboe { 6620ed29b0b4SJens Axboe poll->head = NULL; 6621ed29b0b4SJens Axboe #define IO_POLL_UNMASK (EPOLLERR|EPOLLHUP|EPOLLNVAL|EPOLLRDHUP) 6622ed29b0b4SJens Axboe /* mask in events that we always want/need */ 6623ed29b0b4SJens Axboe poll->events = events | IO_POLL_UNMASK; 6624ed29b0b4SJens Axboe INIT_LIST_HEAD(&poll->wait.entry); 6625ed29b0b4SJens Axboe init_waitqueue_func_entry(&poll->wait, wake_func); 6626ed29b0b4SJens Axboe } 6627ed29b0b4SJens Axboe 66288d4388d1SJens Axboe static inline void io_poll_remove_entry(struct io_poll *poll) 6629ed29b0b4SJens Axboe { 6630ed29b0b4SJens Axboe struct wait_queue_head *head = smp_load_acquire(&poll->head); 6631ed29b0b4SJens Axboe 6632ed29b0b4SJens Axboe if (head) { 6633ed29b0b4SJens Axboe spin_lock_irq(&head->lock); 6634ed29b0b4SJens Axboe list_del_init(&poll->wait.entry); 6635ed29b0b4SJens Axboe poll->head = NULL; 6636ed29b0b4SJens Axboe spin_unlock_irq(&head->lock); 6637ed29b0b4SJens Axboe } 6638ed29b0b4SJens Axboe } 6639ed29b0b4SJens Axboe 6640ed29b0b4SJens Axboe static void io_poll_remove_entries(struct io_kiocb *req) 6641ed29b0b4SJens Axboe { 6642ed29b0b4SJens Axboe /* 6643ed29b0b4SJens Axboe * Nothing to do if neither of those flags are set. Avoid dipping 6644ed29b0b4SJens Axboe * into the poll/apoll/double cachelines if we can. 6645ed29b0b4SJens Axboe */ 6646ed29b0b4SJens Axboe if (!(req->flags & (REQ_F_SINGLE_POLL | REQ_F_DOUBLE_POLL))) 6647ed29b0b4SJens Axboe return; 6648ed29b0b4SJens Axboe 6649ed29b0b4SJens Axboe /* 6650ed29b0b4SJens Axboe * While we hold the waitqueue lock and the waitqueue is nonempty, 6651ed29b0b4SJens Axboe * wake_up_pollfree() will wait for us. However, taking the waitqueue 6652ed29b0b4SJens Axboe * lock in the first place can race with the waitqueue being freed. 6653ed29b0b4SJens Axboe * 6654ed29b0b4SJens Axboe * We solve this as eventpoll does: by taking advantage of the fact that 6655ed29b0b4SJens Axboe * all users of wake_up_pollfree() will RCU-delay the actual free. If 6656ed29b0b4SJens Axboe * we enter rcu_read_lock() and see that the pointer to the queue is 6657ed29b0b4SJens Axboe * non-NULL, we can then lock it without the memory being freed out from 6658ed29b0b4SJens Axboe * under us. 6659ed29b0b4SJens Axboe * 6660ed29b0b4SJens Axboe * Keep holding rcu_read_lock() as long as we hold the queue lock, in 6661ed29b0b4SJens Axboe * case the caller deletes the entry from the queue, leaving it empty. 6662ed29b0b4SJens Axboe * In that case, only RCU prevents the queue memory from being freed. 6663ed29b0b4SJens Axboe */ 6664ed29b0b4SJens Axboe rcu_read_lock(); 6665ed29b0b4SJens Axboe if (req->flags & REQ_F_SINGLE_POLL) 6666ed29b0b4SJens Axboe io_poll_remove_entry(io_poll_get_single(req)); 6667ed29b0b4SJens Axboe if (req->flags & REQ_F_DOUBLE_POLL) 6668ed29b0b4SJens Axboe io_poll_remove_entry(io_poll_get_double(req)); 6669ed29b0b4SJens Axboe rcu_read_unlock(); 6670ed29b0b4SJens Axboe } 6671ed29b0b4SJens Axboe 6672ed29b0b4SJens Axboe static int io_issue_sqe(struct io_kiocb *req, unsigned int issue_flags); 6673ed29b0b4SJens Axboe /* 6674ed29b0b4SJens Axboe * All poll tw should go through this. Checks for poll events, manages 6675ed29b0b4SJens Axboe * references, does rewait, etc. 6676ed29b0b4SJens Axboe * 6677ed29b0b4SJens Axboe * Returns a negative error on failure. >0 when no action require, which is 6678ed29b0b4SJens Axboe * either spurious wakeup or multishot CQE is served. 0 when it's done with 6679ed29b0b4SJens Axboe * the request, then the mask is stored in req->cqe.res. 6680ed29b0b4SJens Axboe */ 6681ed29b0b4SJens Axboe static int io_poll_check_events(struct io_kiocb *req, bool *locked) 6682ed29b0b4SJens Axboe { 6683ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 6684ed29b0b4SJens Axboe int v, ret; 6685ed29b0b4SJens Axboe 6686ed29b0b4SJens Axboe /* req->task == current here, checking PF_EXITING is safe */ 6687ed29b0b4SJens Axboe if (unlikely(req->task->flags & PF_EXITING)) 6688ed29b0b4SJens Axboe return -ECANCELED; 6689ed29b0b4SJens Axboe 6690ed29b0b4SJens Axboe do { 6691ed29b0b4SJens Axboe v = atomic_read(&req->poll_refs); 6692ed29b0b4SJens Axboe 6693ed29b0b4SJens Axboe /* tw handler should be the owner, and so have some references */ 6694ed29b0b4SJens Axboe if (WARN_ON_ONCE(!(v & IO_POLL_REF_MASK))) 6695ed29b0b4SJens Axboe return 0; 6696ed29b0b4SJens Axboe if (v & IO_POLL_CANCEL_FLAG) 6697ed29b0b4SJens Axboe return -ECANCELED; 6698ed29b0b4SJens Axboe 6699ed29b0b4SJens Axboe if (!req->cqe.res) { 6700ed29b0b4SJens Axboe struct poll_table_struct pt = { ._key = req->apoll_events }; 6701ed29b0b4SJens Axboe req->cqe.res = vfs_poll(req->file, &pt) & req->apoll_events; 6702ed29b0b4SJens Axboe } 6703ed29b0b4SJens Axboe 6704ed29b0b4SJens Axboe if ((unlikely(!req->cqe.res))) 6705ed29b0b4SJens Axboe continue; 6706ed29b0b4SJens Axboe if (req->apoll_events & EPOLLONESHOT) 6707ed29b0b4SJens Axboe return 0; 6708ed29b0b4SJens Axboe 6709ed29b0b4SJens Axboe /* multishot, just fill a CQE and proceed */ 6710ed29b0b4SJens Axboe if (!(req->flags & REQ_F_APOLL_MULTISHOT)) { 6711ed29b0b4SJens Axboe __poll_t mask = mangle_poll(req->cqe.res & 6712ed29b0b4SJens Axboe req->apoll_events); 6713ed29b0b4SJens Axboe bool filled; 6714ed29b0b4SJens Axboe 6715ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 6716ed29b0b4SJens Axboe filled = io_fill_cqe_aux(ctx, req->cqe.user_data, 6717ed29b0b4SJens Axboe mask, IORING_CQE_F_MORE); 6718ed29b0b4SJens Axboe io_commit_cqring(ctx); 6719ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 6720ed29b0b4SJens Axboe if (filled) { 6721ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 6722ed29b0b4SJens Axboe continue; 6723ed29b0b4SJens Axboe } 6724ed29b0b4SJens Axboe return -ECANCELED; 6725ed29b0b4SJens Axboe } 6726ed29b0b4SJens Axboe 6727ed29b0b4SJens Axboe io_tw_lock(req->ctx, locked); 6728ed29b0b4SJens Axboe if (unlikely(req->task->flags & PF_EXITING)) 6729ed29b0b4SJens Axboe return -EFAULT; 6730ed29b0b4SJens Axboe ret = io_issue_sqe(req, 6731ed29b0b4SJens Axboe IO_URING_F_NONBLOCK|IO_URING_F_COMPLETE_DEFER); 6732ed29b0b4SJens Axboe if (ret) 6733ed29b0b4SJens Axboe return ret; 6734ed29b0b4SJens Axboe 6735ed29b0b4SJens Axboe /* 6736ed29b0b4SJens Axboe * Release all references, retry if someone tried to restart 6737ed29b0b4SJens Axboe * task_work while we were executing it. 6738ed29b0b4SJens Axboe */ 6739ed29b0b4SJens Axboe } while (atomic_sub_return(v & IO_POLL_REF_MASK, &req->poll_refs)); 6740ed29b0b4SJens Axboe 6741ed29b0b4SJens Axboe return 1; 6742ed29b0b4SJens Axboe } 6743ed29b0b4SJens Axboe 6744ed29b0b4SJens Axboe static void io_poll_task_func(struct io_kiocb *req, bool *locked) 6745ed29b0b4SJens Axboe { 6746ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 6747ed29b0b4SJens Axboe int ret; 6748ed29b0b4SJens Axboe 6749ed29b0b4SJens Axboe ret = io_poll_check_events(req, locked); 6750ed29b0b4SJens Axboe if (ret > 0) 6751ed29b0b4SJens Axboe return; 6752ed29b0b4SJens Axboe 6753ed29b0b4SJens Axboe if (!ret) { 67548d4388d1SJens Axboe struct io_poll *poll = io_kiocb_to_cmd(req); 67558d4388d1SJens Axboe 67568d4388d1SJens Axboe req->cqe.res = mangle_poll(req->cqe.res & poll->events); 6757ed29b0b4SJens Axboe } else { 6758ed29b0b4SJens Axboe req->cqe.res = ret; 6759ed29b0b4SJens Axboe req_set_fail(req); 6760ed29b0b4SJens Axboe } 6761ed29b0b4SJens Axboe 6762ed29b0b4SJens Axboe io_poll_remove_entries(req); 6763ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 6764ed29b0b4SJens Axboe hash_del(&req->hash_node); 6765ed29b0b4SJens Axboe __io_req_complete_post(req, req->cqe.res, 0); 6766ed29b0b4SJens Axboe io_commit_cqring(ctx); 6767ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 6768ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 6769ed29b0b4SJens Axboe } 6770ed29b0b4SJens Axboe 6771ed29b0b4SJens Axboe static void io_apoll_task_func(struct io_kiocb *req, bool *locked) 6772ed29b0b4SJens Axboe { 6773ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 6774ed29b0b4SJens Axboe int ret; 6775ed29b0b4SJens Axboe 6776ed29b0b4SJens Axboe ret = io_poll_check_events(req, locked); 6777ed29b0b4SJens Axboe if (ret > 0) 6778ed29b0b4SJens Axboe return; 6779ed29b0b4SJens Axboe 6780ed29b0b4SJens Axboe io_poll_remove_entries(req); 6781ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 6782ed29b0b4SJens Axboe hash_del(&req->hash_node); 6783ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 6784ed29b0b4SJens Axboe 6785ed29b0b4SJens Axboe if (!ret) 6786ed29b0b4SJens Axboe io_req_task_submit(req, locked); 6787ed29b0b4SJens Axboe else 6788ed29b0b4SJens Axboe io_req_complete_failed(req, ret); 6789ed29b0b4SJens Axboe } 6790ed29b0b4SJens Axboe 6791ed29b0b4SJens Axboe static void __io_poll_execute(struct io_kiocb *req, int mask, 6792ed29b0b4SJens Axboe __poll_t __maybe_unused events) 6793ed29b0b4SJens Axboe { 6794ed29b0b4SJens Axboe req->cqe.res = mask; 6795ed29b0b4SJens Axboe /* 6796ed29b0b4SJens Axboe * This is useful for poll that is armed on behalf of another 6797ed29b0b4SJens Axboe * request, and where the wakeup path could be on a different 6798ed29b0b4SJens Axboe * CPU. We want to avoid pulling in req->apoll->events for that 6799ed29b0b4SJens Axboe * case. 6800ed29b0b4SJens Axboe */ 6801ed29b0b4SJens Axboe if (req->opcode == IORING_OP_POLL_ADD) 6802ed29b0b4SJens Axboe req->io_task_work.func = io_poll_task_func; 6803ed29b0b4SJens Axboe else 6804ed29b0b4SJens Axboe req->io_task_work.func = io_apoll_task_func; 6805ed29b0b4SJens Axboe 6806ed29b0b4SJens Axboe trace_io_uring_task_add(req->ctx, req, req->cqe.user_data, req->opcode, mask); 6807ed29b0b4SJens Axboe io_req_task_work_add(req); 6808ed29b0b4SJens Axboe } 6809ed29b0b4SJens Axboe 6810ed29b0b4SJens Axboe static inline void io_poll_execute(struct io_kiocb *req, int res, 6811ed29b0b4SJens Axboe __poll_t events) 6812ed29b0b4SJens Axboe { 6813ed29b0b4SJens Axboe if (io_poll_get_ownership(req)) 6814ed29b0b4SJens Axboe __io_poll_execute(req, res, events); 6815ed29b0b4SJens Axboe } 6816ed29b0b4SJens Axboe 6817ed29b0b4SJens Axboe static void io_poll_cancel_req(struct io_kiocb *req) 6818ed29b0b4SJens Axboe { 6819ed29b0b4SJens Axboe io_poll_mark_cancelled(req); 6820ed29b0b4SJens Axboe /* kick tw, which should complete the request */ 6821ed29b0b4SJens Axboe io_poll_execute(req, 0, 0); 6822ed29b0b4SJens Axboe } 6823ed29b0b4SJens Axboe 6824ed29b0b4SJens Axboe #define wqe_to_req(wait) ((void *)((unsigned long) (wait)->private & ~1)) 6825ed29b0b4SJens Axboe #define wqe_is_double(wait) ((unsigned long) (wait)->private & 1) 6826ed29b0b4SJens Axboe #define IO_ASYNC_POLL_COMMON (EPOLLONESHOT | EPOLLPRI) 6827ed29b0b4SJens Axboe 6828ed29b0b4SJens Axboe static int io_poll_wake(struct wait_queue_entry *wait, unsigned mode, int sync, 6829ed29b0b4SJens Axboe void *key) 6830ed29b0b4SJens Axboe { 6831ed29b0b4SJens Axboe struct io_kiocb *req = wqe_to_req(wait); 68328d4388d1SJens Axboe struct io_poll *poll = container_of(wait, struct io_poll, wait); 6833ed29b0b4SJens Axboe __poll_t mask = key_to_poll(key); 6834ed29b0b4SJens Axboe 6835ed29b0b4SJens Axboe if (unlikely(mask & POLLFREE)) { 6836ed29b0b4SJens Axboe io_poll_mark_cancelled(req); 6837ed29b0b4SJens Axboe /* we have to kick tw in case it's not already */ 6838ed29b0b4SJens Axboe io_poll_execute(req, 0, poll->events); 6839ed29b0b4SJens Axboe 6840ed29b0b4SJens Axboe /* 6841ed29b0b4SJens Axboe * If the waitqueue is being freed early but someone is already 6842ed29b0b4SJens Axboe * holds ownership over it, we have to tear down the request as 6843ed29b0b4SJens Axboe * best we can. That means immediately removing the request from 6844ed29b0b4SJens Axboe * its waitqueue and preventing all further accesses to the 6845ed29b0b4SJens Axboe * waitqueue via the request. 6846ed29b0b4SJens Axboe */ 6847ed29b0b4SJens Axboe list_del_init(&poll->wait.entry); 6848ed29b0b4SJens Axboe 6849ed29b0b4SJens Axboe /* 6850ed29b0b4SJens Axboe * Careful: this *must* be the last step, since as soon 6851ed29b0b4SJens Axboe * as req->head is NULL'ed out, the request can be 6852ed29b0b4SJens Axboe * completed and freed, since aio_poll_complete_work() 6853ed29b0b4SJens Axboe * will no longer need to take the waitqueue lock. 6854ed29b0b4SJens Axboe */ 6855ed29b0b4SJens Axboe smp_store_release(&poll->head, NULL); 6856ed29b0b4SJens Axboe return 1; 6857ed29b0b4SJens Axboe } 6858ed29b0b4SJens Axboe 6859ed29b0b4SJens Axboe /* for instances that support it check for an event match first */ 6860ed29b0b4SJens Axboe if (mask && !(mask & (poll->events & ~IO_ASYNC_POLL_COMMON))) 6861ed29b0b4SJens Axboe return 0; 6862ed29b0b4SJens Axboe 6863ed29b0b4SJens Axboe if (io_poll_get_ownership(req)) { 6864ed29b0b4SJens Axboe /* optional, saves extra locking for removal in tw handler */ 6865ed29b0b4SJens Axboe if (mask && poll->events & EPOLLONESHOT) { 6866ed29b0b4SJens Axboe list_del_init(&poll->wait.entry); 6867ed29b0b4SJens Axboe poll->head = NULL; 6868ed29b0b4SJens Axboe if (wqe_is_double(wait)) 6869ed29b0b4SJens Axboe req->flags &= ~REQ_F_DOUBLE_POLL; 6870ed29b0b4SJens Axboe else 6871ed29b0b4SJens Axboe req->flags &= ~REQ_F_SINGLE_POLL; 6872ed29b0b4SJens Axboe } 6873ed29b0b4SJens Axboe __io_poll_execute(req, mask, poll->events); 6874ed29b0b4SJens Axboe } 6875ed29b0b4SJens Axboe return 1; 6876ed29b0b4SJens Axboe } 6877ed29b0b4SJens Axboe 68788d4388d1SJens Axboe static void __io_queue_proc(struct io_poll *poll, struct io_poll_table *pt, 6879ed29b0b4SJens Axboe struct wait_queue_head *head, 68808d4388d1SJens Axboe struct io_poll **poll_ptr) 6881ed29b0b4SJens Axboe { 6882ed29b0b4SJens Axboe struct io_kiocb *req = pt->req; 6883ed29b0b4SJens Axboe unsigned long wqe_private = (unsigned long) req; 6884ed29b0b4SJens Axboe 6885ed29b0b4SJens Axboe /* 6886ed29b0b4SJens Axboe * The file being polled uses multiple waitqueues for poll handling 68878d4388d1SJens Axboe * (e.g. one for read, one for write). Setup a separate io_poll 6888ed29b0b4SJens Axboe * if this happens. 6889ed29b0b4SJens Axboe */ 6890ed29b0b4SJens Axboe if (unlikely(pt->nr_entries)) { 68918d4388d1SJens Axboe struct io_poll *first = poll; 6892ed29b0b4SJens Axboe 6893ed29b0b4SJens Axboe /* double add on the same waitqueue head, ignore */ 6894ed29b0b4SJens Axboe if (first->head == head) 6895ed29b0b4SJens Axboe return; 6896ed29b0b4SJens Axboe /* already have a 2nd entry, fail a third attempt */ 6897ed29b0b4SJens Axboe if (*poll_ptr) { 6898ed29b0b4SJens Axboe if ((*poll_ptr)->head == head) 6899ed29b0b4SJens Axboe return; 6900ed29b0b4SJens Axboe pt->error = -EINVAL; 6901ed29b0b4SJens Axboe return; 6902ed29b0b4SJens Axboe } 6903ed29b0b4SJens Axboe 6904ed29b0b4SJens Axboe poll = kmalloc(sizeof(*poll), GFP_ATOMIC); 6905ed29b0b4SJens Axboe if (!poll) { 6906ed29b0b4SJens Axboe pt->error = -ENOMEM; 6907ed29b0b4SJens Axboe return; 6908ed29b0b4SJens Axboe } 6909ed29b0b4SJens Axboe /* mark as double wq entry */ 6910ed29b0b4SJens Axboe wqe_private |= 1; 6911ed29b0b4SJens Axboe req->flags |= REQ_F_DOUBLE_POLL; 6912ed29b0b4SJens Axboe io_init_poll_iocb(poll, first->events, first->wait.func); 6913ed29b0b4SJens Axboe *poll_ptr = poll; 6914ed29b0b4SJens Axboe if (req->opcode == IORING_OP_POLL_ADD) 6915ed29b0b4SJens Axboe req->flags |= REQ_F_ASYNC_DATA; 6916ed29b0b4SJens Axboe } 6917ed29b0b4SJens Axboe 6918ed29b0b4SJens Axboe req->flags |= REQ_F_SINGLE_POLL; 6919ed29b0b4SJens Axboe pt->nr_entries++; 6920ed29b0b4SJens Axboe poll->head = head; 6921ed29b0b4SJens Axboe poll->wait.private = (void *) wqe_private; 6922ed29b0b4SJens Axboe 6923ed29b0b4SJens Axboe if (poll->events & EPOLLEXCLUSIVE) 6924ed29b0b4SJens Axboe add_wait_queue_exclusive(head, &poll->wait); 6925ed29b0b4SJens Axboe else 6926ed29b0b4SJens Axboe add_wait_queue(head, &poll->wait); 6927ed29b0b4SJens Axboe } 6928ed29b0b4SJens Axboe 6929ed29b0b4SJens Axboe static void io_poll_queue_proc(struct file *file, struct wait_queue_head *head, 6930ed29b0b4SJens Axboe struct poll_table_struct *p) 6931ed29b0b4SJens Axboe { 6932ed29b0b4SJens Axboe struct io_poll_table *pt = container_of(p, struct io_poll_table, pt); 69338d4388d1SJens Axboe struct io_poll *poll = io_kiocb_to_cmd(pt->req); 6934ed29b0b4SJens Axboe 69358d4388d1SJens Axboe __io_queue_proc(poll, pt, head, 69368d4388d1SJens Axboe (struct io_poll **) &pt->req->async_data); 6937ed29b0b4SJens Axboe } 6938ed29b0b4SJens Axboe 6939ed29b0b4SJens Axboe static int __io_arm_poll_handler(struct io_kiocb *req, 69408d4388d1SJens Axboe struct io_poll *poll, 6941ed29b0b4SJens Axboe struct io_poll_table *ipt, __poll_t mask) 6942ed29b0b4SJens Axboe { 6943ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 6944ed29b0b4SJens Axboe int v; 6945ed29b0b4SJens Axboe 6946ed29b0b4SJens Axboe INIT_HLIST_NODE(&req->hash_node); 6947ed29b0b4SJens Axboe req->work.cancel_seq = atomic_read(&ctx->cancel_seq); 6948ed29b0b4SJens Axboe io_init_poll_iocb(poll, mask, io_poll_wake); 6949ed29b0b4SJens Axboe poll->file = req->file; 6950ed29b0b4SJens Axboe 6951ed29b0b4SJens Axboe req->apoll_events = poll->events; 6952ed29b0b4SJens Axboe 6953ed29b0b4SJens Axboe ipt->pt._key = mask; 6954ed29b0b4SJens Axboe ipt->req = req; 6955ed29b0b4SJens Axboe ipt->error = 0; 6956ed29b0b4SJens Axboe ipt->nr_entries = 0; 6957ed29b0b4SJens Axboe 6958ed29b0b4SJens Axboe /* 6959ed29b0b4SJens Axboe * Take the ownership to delay any tw execution up until we're done 6960ed29b0b4SJens Axboe * with poll arming. see io_poll_get_ownership(). 6961ed29b0b4SJens Axboe */ 6962ed29b0b4SJens Axboe atomic_set(&req->poll_refs, 1); 6963ed29b0b4SJens Axboe mask = vfs_poll(req->file, &ipt->pt) & poll->events; 6964ed29b0b4SJens Axboe 6965ed29b0b4SJens Axboe if (mask && (poll->events & EPOLLONESHOT)) { 6966ed29b0b4SJens Axboe io_poll_remove_entries(req); 6967ed29b0b4SJens Axboe /* no one else has access to the req, forget about the ref */ 6968ed29b0b4SJens Axboe return mask; 6969ed29b0b4SJens Axboe } 6970ed29b0b4SJens Axboe if (!mask && unlikely(ipt->error || !ipt->nr_entries)) { 6971ed29b0b4SJens Axboe io_poll_remove_entries(req); 6972ed29b0b4SJens Axboe if (!ipt->error) 6973ed29b0b4SJens Axboe ipt->error = -EINVAL; 6974ed29b0b4SJens Axboe return 0; 6975ed29b0b4SJens Axboe } 6976ed29b0b4SJens Axboe 6977ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 6978ed29b0b4SJens Axboe io_poll_req_insert(req); 6979ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 6980ed29b0b4SJens Axboe 6981ed29b0b4SJens Axboe if (mask) { 6982ed29b0b4SJens Axboe /* can't multishot if failed, just queue the event we've got */ 6983ed29b0b4SJens Axboe if (unlikely(ipt->error || !ipt->nr_entries)) { 6984ed29b0b4SJens Axboe poll->events |= EPOLLONESHOT; 6985ed29b0b4SJens Axboe req->apoll_events |= EPOLLONESHOT; 6986ed29b0b4SJens Axboe ipt->error = 0; 6987ed29b0b4SJens Axboe } 6988ed29b0b4SJens Axboe __io_poll_execute(req, mask, poll->events); 6989ed29b0b4SJens Axboe return 0; 6990ed29b0b4SJens Axboe } 6991ed29b0b4SJens Axboe 6992ed29b0b4SJens Axboe /* 6993ed29b0b4SJens Axboe * Release ownership. If someone tried to queue a tw while it was 6994ed29b0b4SJens Axboe * locked, kick it off for them. 6995ed29b0b4SJens Axboe */ 6996ed29b0b4SJens Axboe v = atomic_dec_return(&req->poll_refs); 6997ed29b0b4SJens Axboe if (unlikely(v & IO_POLL_REF_MASK)) 6998ed29b0b4SJens Axboe __io_poll_execute(req, 0, poll->events); 6999ed29b0b4SJens Axboe return 0; 7000ed29b0b4SJens Axboe } 7001ed29b0b4SJens Axboe 7002ed29b0b4SJens Axboe static void io_async_queue_proc(struct file *file, struct wait_queue_head *head, 7003ed29b0b4SJens Axboe struct poll_table_struct *p) 7004ed29b0b4SJens Axboe { 7005ed29b0b4SJens Axboe struct io_poll_table *pt = container_of(p, struct io_poll_table, pt); 7006ed29b0b4SJens Axboe struct async_poll *apoll = pt->req->apoll; 7007ed29b0b4SJens Axboe 7008ed29b0b4SJens Axboe __io_queue_proc(&apoll->poll, pt, head, &apoll->double_poll); 7009ed29b0b4SJens Axboe } 7010ed29b0b4SJens Axboe 7011ed29b0b4SJens Axboe enum { 7012ed29b0b4SJens Axboe IO_APOLL_OK, 7013ed29b0b4SJens Axboe IO_APOLL_ABORTED, 7014ed29b0b4SJens Axboe IO_APOLL_READY 7015ed29b0b4SJens Axboe }; 7016ed29b0b4SJens Axboe 7017ed29b0b4SJens Axboe static int io_arm_poll_handler(struct io_kiocb *req, unsigned issue_flags) 7018ed29b0b4SJens Axboe { 7019ed29b0b4SJens Axboe const struct io_op_def *def = &io_op_defs[req->opcode]; 7020ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 7021ed29b0b4SJens Axboe struct async_poll *apoll; 7022ed29b0b4SJens Axboe struct io_poll_table ipt; 7023ed29b0b4SJens Axboe __poll_t mask = POLLPRI | POLLERR; 7024ed29b0b4SJens Axboe int ret; 7025ed29b0b4SJens Axboe 7026ed29b0b4SJens Axboe if (!def->pollin && !def->pollout) 7027ed29b0b4SJens Axboe return IO_APOLL_ABORTED; 7028ed29b0b4SJens Axboe if (!file_can_poll(req->file)) 7029ed29b0b4SJens Axboe return IO_APOLL_ABORTED; 7030ed29b0b4SJens Axboe if ((req->flags & (REQ_F_POLLED|REQ_F_PARTIAL_IO)) == REQ_F_POLLED) 7031ed29b0b4SJens Axboe return IO_APOLL_ABORTED; 7032ed29b0b4SJens Axboe if (!(req->flags & REQ_F_APOLL_MULTISHOT)) 7033ed29b0b4SJens Axboe mask |= EPOLLONESHOT; 7034ed29b0b4SJens Axboe 7035ed29b0b4SJens Axboe if (def->pollin) { 7036ed29b0b4SJens Axboe mask |= EPOLLIN | EPOLLRDNORM; 7037ed29b0b4SJens Axboe 7038ed29b0b4SJens Axboe /* If reading from MSG_ERRQUEUE using recvmsg, ignore POLLIN */ 7039bd8587e4SJens Axboe if (req->flags & REQ_F_CLEAR_POLLIN) 7040ed29b0b4SJens Axboe mask &= ~EPOLLIN; 7041ed29b0b4SJens Axboe } else { 7042ed29b0b4SJens Axboe mask |= EPOLLOUT | EPOLLWRNORM; 7043ed29b0b4SJens Axboe } 7044ed29b0b4SJens Axboe if (def->poll_exclusive) 7045ed29b0b4SJens Axboe mask |= EPOLLEXCLUSIVE; 7046ed29b0b4SJens Axboe if (req->flags & REQ_F_POLLED) { 7047ed29b0b4SJens Axboe apoll = req->apoll; 7048ed29b0b4SJens Axboe kfree(apoll->double_poll); 7049ed29b0b4SJens Axboe } else if (!(issue_flags & IO_URING_F_UNLOCKED) && 7050ed29b0b4SJens Axboe !list_empty(&ctx->apoll_cache)) { 7051ed29b0b4SJens Axboe apoll = list_first_entry(&ctx->apoll_cache, struct async_poll, 7052ed29b0b4SJens Axboe poll.wait.entry); 7053ed29b0b4SJens Axboe list_del_init(&apoll->poll.wait.entry); 7054ed29b0b4SJens Axboe } else { 7055ed29b0b4SJens Axboe apoll = kmalloc(sizeof(*apoll), GFP_ATOMIC); 7056ed29b0b4SJens Axboe if (unlikely(!apoll)) 7057ed29b0b4SJens Axboe return IO_APOLL_ABORTED; 7058ed29b0b4SJens Axboe } 7059ed29b0b4SJens Axboe apoll->double_poll = NULL; 7060ed29b0b4SJens Axboe req->apoll = apoll; 7061ed29b0b4SJens Axboe req->flags |= REQ_F_POLLED; 7062ed29b0b4SJens Axboe ipt.pt._qproc = io_async_queue_proc; 7063ed29b0b4SJens Axboe 7064ed29b0b4SJens Axboe io_kbuf_recycle(req, issue_flags); 7065ed29b0b4SJens Axboe 7066ed29b0b4SJens Axboe ret = __io_arm_poll_handler(req, &apoll->poll, &ipt, mask); 7067ed29b0b4SJens Axboe if (ret || ipt.error) 7068ed29b0b4SJens Axboe return ret ? IO_APOLL_READY : IO_APOLL_ABORTED; 7069ed29b0b4SJens Axboe 7070ed29b0b4SJens Axboe trace_io_uring_poll_arm(ctx, req, req->cqe.user_data, req->opcode, 7071ed29b0b4SJens Axboe mask, apoll->poll.events); 7072ed29b0b4SJens Axboe return IO_APOLL_OK; 7073ed29b0b4SJens Axboe } 7074ed29b0b4SJens Axboe 7075ed29b0b4SJens Axboe /* 7076ed29b0b4SJens Axboe * Returns true if we found and killed one or more poll requests 7077ed29b0b4SJens Axboe */ 7078ed29b0b4SJens Axboe static __cold bool io_poll_remove_all(struct io_ring_ctx *ctx, 7079ed29b0b4SJens Axboe struct task_struct *tsk, bool cancel_all) 7080ed29b0b4SJens Axboe { 7081ed29b0b4SJens Axboe struct hlist_node *tmp; 7082ed29b0b4SJens Axboe struct io_kiocb *req; 7083ed29b0b4SJens Axboe bool found = false; 7084ed29b0b4SJens Axboe int i; 7085ed29b0b4SJens Axboe 7086ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 7087ed29b0b4SJens Axboe for (i = 0; i < (1U << ctx->cancel_hash_bits); i++) { 7088ed29b0b4SJens Axboe struct hlist_head *list; 7089ed29b0b4SJens Axboe 7090ed29b0b4SJens Axboe list = &ctx->cancel_hash[i]; 7091ed29b0b4SJens Axboe hlist_for_each_entry_safe(req, tmp, list, hash_node) { 7092ed29b0b4SJens Axboe if (io_match_task_safe(req, tsk, cancel_all)) { 7093ed29b0b4SJens Axboe hlist_del_init(&req->hash_node); 7094ed29b0b4SJens Axboe io_poll_cancel_req(req); 7095ed29b0b4SJens Axboe found = true; 7096ed29b0b4SJens Axboe } 7097ed29b0b4SJens Axboe } 7098ed29b0b4SJens Axboe } 7099ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 7100ed29b0b4SJens Axboe return found; 7101ed29b0b4SJens Axboe } 7102ed29b0b4SJens Axboe 7103ed29b0b4SJens Axboe static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only, 7104ed29b0b4SJens Axboe struct io_cancel_data *cd) 7105ed29b0b4SJens Axboe __must_hold(&ctx->completion_lock) 7106ed29b0b4SJens Axboe { 7107ed29b0b4SJens Axboe struct hlist_head *list; 7108ed29b0b4SJens Axboe struct io_kiocb *req; 7109ed29b0b4SJens Axboe 7110ed29b0b4SJens Axboe list = &ctx->cancel_hash[hash_long(cd->data, ctx->cancel_hash_bits)]; 7111ed29b0b4SJens Axboe hlist_for_each_entry(req, list, hash_node) { 7112ed29b0b4SJens Axboe if (cd->data != req->cqe.user_data) 7113ed29b0b4SJens Axboe continue; 7114ed29b0b4SJens Axboe if (poll_only && req->opcode != IORING_OP_POLL_ADD) 7115ed29b0b4SJens Axboe continue; 7116ed29b0b4SJens Axboe if (cd->flags & IORING_ASYNC_CANCEL_ALL) { 7117ed29b0b4SJens Axboe if (cd->seq == req->work.cancel_seq) 7118ed29b0b4SJens Axboe continue; 7119ed29b0b4SJens Axboe req->work.cancel_seq = cd->seq; 7120ed29b0b4SJens Axboe } 7121ed29b0b4SJens Axboe return req; 7122ed29b0b4SJens Axboe } 7123ed29b0b4SJens Axboe return NULL; 7124ed29b0b4SJens Axboe } 7125ed29b0b4SJens Axboe 7126ed29b0b4SJens Axboe static struct io_kiocb *io_poll_file_find(struct io_ring_ctx *ctx, 7127ed29b0b4SJens Axboe struct io_cancel_data *cd) 7128ed29b0b4SJens Axboe __must_hold(&ctx->completion_lock) 7129ed29b0b4SJens Axboe { 7130ed29b0b4SJens Axboe struct io_kiocb *req; 7131ed29b0b4SJens Axboe int i; 7132ed29b0b4SJens Axboe 7133ed29b0b4SJens Axboe for (i = 0; i < (1U << ctx->cancel_hash_bits); i++) { 7134ed29b0b4SJens Axboe struct hlist_head *list; 7135ed29b0b4SJens Axboe 7136ed29b0b4SJens Axboe list = &ctx->cancel_hash[i]; 7137ed29b0b4SJens Axboe hlist_for_each_entry(req, list, hash_node) { 7138ed29b0b4SJens Axboe if (!(cd->flags & IORING_ASYNC_CANCEL_ANY) && 7139ed29b0b4SJens Axboe req->file != cd->file) 7140ed29b0b4SJens Axboe continue; 7141ed29b0b4SJens Axboe if (cd->seq == req->work.cancel_seq) 7142ed29b0b4SJens Axboe continue; 7143ed29b0b4SJens Axboe req->work.cancel_seq = cd->seq; 7144ed29b0b4SJens Axboe return req; 7145ed29b0b4SJens Axboe } 7146ed29b0b4SJens Axboe } 7147ed29b0b4SJens Axboe return NULL; 7148ed29b0b4SJens Axboe } 7149ed29b0b4SJens Axboe 7150ed29b0b4SJens Axboe static bool io_poll_disarm(struct io_kiocb *req) 7151ed29b0b4SJens Axboe __must_hold(&ctx->completion_lock) 7152ed29b0b4SJens Axboe { 7153ed29b0b4SJens Axboe if (!io_poll_get_ownership(req)) 7154ed29b0b4SJens Axboe return false; 7155ed29b0b4SJens Axboe io_poll_remove_entries(req); 7156ed29b0b4SJens Axboe hash_del(&req->hash_node); 7157ed29b0b4SJens Axboe return true; 7158ed29b0b4SJens Axboe } 7159ed29b0b4SJens Axboe 7160ed29b0b4SJens Axboe static int io_poll_cancel(struct io_ring_ctx *ctx, struct io_cancel_data *cd) 7161ed29b0b4SJens Axboe __must_hold(&ctx->completion_lock) 7162ed29b0b4SJens Axboe { 7163ed29b0b4SJens Axboe struct io_kiocb *req; 7164ed29b0b4SJens Axboe 7165ed29b0b4SJens Axboe if (cd->flags & (IORING_ASYNC_CANCEL_FD|IORING_ASYNC_CANCEL_ANY)) 7166ed29b0b4SJens Axboe req = io_poll_file_find(ctx, cd); 7167ed29b0b4SJens Axboe else 7168ed29b0b4SJens Axboe req = io_poll_find(ctx, false, cd); 7169ed29b0b4SJens Axboe if (!req) 7170ed29b0b4SJens Axboe return -ENOENT; 7171ed29b0b4SJens Axboe io_poll_cancel_req(req); 7172ed29b0b4SJens Axboe return 0; 7173ed29b0b4SJens Axboe } 7174ed29b0b4SJens Axboe 7175ed29b0b4SJens Axboe static __poll_t io_poll_parse_events(const struct io_uring_sqe *sqe, 7176ed29b0b4SJens Axboe unsigned int flags) 7177ed29b0b4SJens Axboe { 7178ed29b0b4SJens Axboe u32 events; 7179ed29b0b4SJens Axboe 7180ed29b0b4SJens Axboe events = READ_ONCE(sqe->poll32_events); 7181ed29b0b4SJens Axboe #ifdef __BIG_ENDIAN 7182ed29b0b4SJens Axboe events = swahw32(events); 7183ed29b0b4SJens Axboe #endif 7184ed29b0b4SJens Axboe if (!(flags & IORING_POLL_ADD_MULTI)) 7185ed29b0b4SJens Axboe events |= EPOLLONESHOT; 7186ed29b0b4SJens Axboe return demangle_poll(events) | (events & (EPOLLEXCLUSIVE|EPOLLONESHOT)); 7187ed29b0b4SJens Axboe } 7188ed29b0b4SJens Axboe 7189ed29b0b4SJens Axboe static int io_poll_remove_prep(struct io_kiocb *req, 7190ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 7191ed29b0b4SJens Axboe { 7192c24b1549SJens Axboe struct io_poll_update *upd = io_kiocb_to_cmd(req); 7193ed29b0b4SJens Axboe u32 flags; 7194ed29b0b4SJens Axboe 7195ed29b0b4SJens Axboe if (sqe->buf_index || sqe->splice_fd_in) 7196ed29b0b4SJens Axboe return -EINVAL; 7197ed29b0b4SJens Axboe flags = READ_ONCE(sqe->len); 7198ed29b0b4SJens Axboe if (flags & ~(IORING_POLL_UPDATE_EVENTS | IORING_POLL_UPDATE_USER_DATA | 7199ed29b0b4SJens Axboe IORING_POLL_ADD_MULTI)) 7200ed29b0b4SJens Axboe return -EINVAL; 7201ed29b0b4SJens Axboe /* meaningless without update */ 7202ed29b0b4SJens Axboe if (flags == IORING_POLL_ADD_MULTI) 7203ed29b0b4SJens Axboe return -EINVAL; 7204ed29b0b4SJens Axboe 7205ed29b0b4SJens Axboe upd->old_user_data = READ_ONCE(sqe->addr); 7206ed29b0b4SJens Axboe upd->update_events = flags & IORING_POLL_UPDATE_EVENTS; 7207ed29b0b4SJens Axboe upd->update_user_data = flags & IORING_POLL_UPDATE_USER_DATA; 7208ed29b0b4SJens Axboe 7209ed29b0b4SJens Axboe upd->new_user_data = READ_ONCE(sqe->off); 7210ed29b0b4SJens Axboe if (!upd->update_user_data && upd->new_user_data) 7211ed29b0b4SJens Axboe return -EINVAL; 7212ed29b0b4SJens Axboe if (upd->update_events) 7213ed29b0b4SJens Axboe upd->events = io_poll_parse_events(sqe, flags); 7214ed29b0b4SJens Axboe else if (sqe->poll32_events) 7215ed29b0b4SJens Axboe return -EINVAL; 7216ed29b0b4SJens Axboe 7217ed29b0b4SJens Axboe return 0; 7218ed29b0b4SJens Axboe } 7219ed29b0b4SJens Axboe 7220ed29b0b4SJens Axboe static int io_poll_add_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) 7221ed29b0b4SJens Axboe { 72228d4388d1SJens Axboe struct io_poll *poll = io_kiocb_to_cmd(req); 7223ed29b0b4SJens Axboe u32 flags; 7224ed29b0b4SJens Axboe 7225ed29b0b4SJens Axboe if (sqe->buf_index || sqe->off || sqe->addr) 7226ed29b0b4SJens Axboe return -EINVAL; 7227ed29b0b4SJens Axboe flags = READ_ONCE(sqe->len); 7228ed29b0b4SJens Axboe if (flags & ~IORING_POLL_ADD_MULTI) 7229ed29b0b4SJens Axboe return -EINVAL; 7230ed29b0b4SJens Axboe if ((flags & IORING_POLL_ADD_MULTI) && (req->flags & REQ_F_CQE_SKIP)) 7231ed29b0b4SJens Axboe return -EINVAL; 7232ed29b0b4SJens Axboe 7233ed29b0b4SJens Axboe io_req_set_refcount(req); 7234ed29b0b4SJens Axboe poll->events = io_poll_parse_events(sqe, flags); 7235ed29b0b4SJens Axboe return 0; 7236ed29b0b4SJens Axboe } 7237ed29b0b4SJens Axboe 7238ed29b0b4SJens Axboe static int io_poll_add(struct io_kiocb *req, unsigned int issue_flags) 7239ed29b0b4SJens Axboe { 72408d4388d1SJens Axboe struct io_poll *poll = io_kiocb_to_cmd(req); 7241ed29b0b4SJens Axboe struct io_poll_table ipt; 7242ed29b0b4SJens Axboe int ret; 7243ed29b0b4SJens Axboe 7244ed29b0b4SJens Axboe ipt.pt._qproc = io_poll_queue_proc; 7245ed29b0b4SJens Axboe 72468d4388d1SJens Axboe ret = __io_arm_poll_handler(req, poll, &ipt, poll->events); 7247ed29b0b4SJens Axboe if (!ret && ipt.error) 7248ed29b0b4SJens Axboe req_set_fail(req); 7249ed29b0b4SJens Axboe ret = ret ?: ipt.error; 7250ed29b0b4SJens Axboe if (ret) 7251ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 7252ed29b0b4SJens Axboe return 0; 7253ed29b0b4SJens Axboe } 7254ed29b0b4SJens Axboe 7255ed29b0b4SJens Axboe static int io_poll_remove(struct io_kiocb *req, unsigned int issue_flags) 7256ed29b0b4SJens Axboe { 7257c24b1549SJens Axboe struct io_poll_update *poll_update = io_kiocb_to_cmd(req); 7258c24b1549SJens Axboe struct io_cancel_data cd = { .data = poll_update->old_user_data, }; 7259ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 7260ed29b0b4SJens Axboe struct io_kiocb *preq; 7261ed29b0b4SJens Axboe int ret2, ret = 0; 7262ed29b0b4SJens Axboe bool locked; 7263ed29b0b4SJens Axboe 7264ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 7265ed29b0b4SJens Axboe preq = io_poll_find(ctx, true, &cd); 7266ed29b0b4SJens Axboe if (!preq || !io_poll_disarm(preq)) { 7267ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 7268ed29b0b4SJens Axboe ret = preq ? -EALREADY : -ENOENT; 7269ed29b0b4SJens Axboe goto out; 7270ed29b0b4SJens Axboe } 7271ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 7272ed29b0b4SJens Axboe 7273c24b1549SJens Axboe if (poll_update->update_events || poll_update->update_user_data) { 7274ed29b0b4SJens Axboe /* only mask one event flags, keep behavior flags */ 7275c24b1549SJens Axboe if (poll_update->update_events) { 72768d4388d1SJens Axboe struct io_poll *poll = io_kiocb_to_cmd(preq); 72778d4388d1SJens Axboe 72788d4388d1SJens Axboe poll->events &= ~0xffff; 7279c24b1549SJens Axboe poll->events |= poll_update->events & 0xffff; 72808d4388d1SJens Axboe poll->events |= IO_POLL_UNMASK; 7281ed29b0b4SJens Axboe } 7282c24b1549SJens Axboe if (poll_update->update_user_data) 7283c24b1549SJens Axboe preq->cqe.user_data = poll_update->new_user_data; 7284ed29b0b4SJens Axboe 7285ed29b0b4SJens Axboe ret2 = io_poll_add(preq, issue_flags); 7286ed29b0b4SJens Axboe /* successfully updated, don't complete poll request */ 7287ed29b0b4SJens Axboe if (!ret2) 7288ed29b0b4SJens Axboe goto out; 7289ed29b0b4SJens Axboe } 7290ed29b0b4SJens Axboe 7291ed29b0b4SJens Axboe req_set_fail(preq); 7292ed29b0b4SJens Axboe preq->cqe.res = -ECANCELED; 7293ed29b0b4SJens Axboe locked = !(issue_flags & IO_URING_F_UNLOCKED); 7294ed29b0b4SJens Axboe io_req_task_complete(preq, &locked); 7295ed29b0b4SJens Axboe out: 7296ed29b0b4SJens Axboe if (ret < 0) 7297ed29b0b4SJens Axboe req_set_fail(req); 7298ed29b0b4SJens Axboe /* complete update request, we're done with it */ 7299ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 7300ed29b0b4SJens Axboe return 0; 7301ed29b0b4SJens Axboe } 7302ed29b0b4SJens Axboe 7303ed29b0b4SJens Axboe static enum hrtimer_restart io_timeout_fn(struct hrtimer *timer) 7304ed29b0b4SJens Axboe { 7305ed29b0b4SJens Axboe struct io_timeout_data *data = container_of(timer, 7306ed29b0b4SJens Axboe struct io_timeout_data, timer); 7307ed29b0b4SJens Axboe struct io_kiocb *req = data->req; 7308a43714acSJens Axboe struct io_timeout *timeout = io_kiocb_to_cmd(req); 7309ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 7310ed29b0b4SJens Axboe unsigned long flags; 7311ed29b0b4SJens Axboe 7312ed29b0b4SJens Axboe spin_lock_irqsave(&ctx->timeout_lock, flags); 7313a43714acSJens Axboe list_del_init(&timeout->list); 7314ed29b0b4SJens Axboe atomic_set(&req->ctx->cq_timeouts, 7315ed29b0b4SJens Axboe atomic_read(&req->ctx->cq_timeouts) + 1); 7316ed29b0b4SJens Axboe spin_unlock_irqrestore(&ctx->timeout_lock, flags); 7317ed29b0b4SJens Axboe 7318ed29b0b4SJens Axboe if (!(data->flags & IORING_TIMEOUT_ETIME_SUCCESS)) 7319ed29b0b4SJens Axboe req_set_fail(req); 7320ed29b0b4SJens Axboe 7321ed29b0b4SJens Axboe req->cqe.res = -ETIME; 7322ed29b0b4SJens Axboe req->io_task_work.func = io_req_task_complete; 7323ed29b0b4SJens Axboe io_req_task_work_add(req); 7324ed29b0b4SJens Axboe return HRTIMER_NORESTART; 7325ed29b0b4SJens Axboe } 7326ed29b0b4SJens Axboe 7327ed29b0b4SJens Axboe static struct io_kiocb *io_timeout_extract(struct io_ring_ctx *ctx, 7328ed29b0b4SJens Axboe struct io_cancel_data *cd) 7329ed29b0b4SJens Axboe __must_hold(&ctx->timeout_lock) 7330ed29b0b4SJens Axboe { 7331a43714acSJens Axboe struct io_timeout *timeout; 7332ed29b0b4SJens Axboe struct io_timeout_data *io; 7333a43714acSJens Axboe struct io_kiocb *req = NULL; 7334ed29b0b4SJens Axboe 7335a43714acSJens Axboe list_for_each_entry(timeout, &ctx->timeout_list, list) { 7336a43714acSJens Axboe struct io_kiocb *tmp = cmd_to_io_kiocb(timeout); 7337a43714acSJens Axboe 7338ed29b0b4SJens Axboe if (!(cd->flags & IORING_ASYNC_CANCEL_ANY) && 7339a43714acSJens Axboe cd->data != tmp->cqe.user_data) 7340ed29b0b4SJens Axboe continue; 7341ed29b0b4SJens Axboe if (cd->flags & (IORING_ASYNC_CANCEL_ALL|IORING_ASYNC_CANCEL_ANY)) { 7342a43714acSJens Axboe if (cd->seq == tmp->work.cancel_seq) 7343ed29b0b4SJens Axboe continue; 7344a43714acSJens Axboe tmp->work.cancel_seq = cd->seq; 7345ed29b0b4SJens Axboe } 7346a43714acSJens Axboe req = tmp; 7347ed29b0b4SJens Axboe break; 7348ed29b0b4SJens Axboe } 7349a43714acSJens Axboe if (!req) 7350ed29b0b4SJens Axboe return ERR_PTR(-ENOENT); 7351ed29b0b4SJens Axboe 7352ed29b0b4SJens Axboe io = req->async_data; 7353ed29b0b4SJens Axboe if (hrtimer_try_to_cancel(&io->timer) == -1) 7354ed29b0b4SJens Axboe return ERR_PTR(-EALREADY); 7355a43714acSJens Axboe timeout = io_kiocb_to_cmd(req); 7356a43714acSJens Axboe list_del_init(&timeout->list); 7357ed29b0b4SJens Axboe return req; 7358ed29b0b4SJens Axboe } 7359ed29b0b4SJens Axboe 7360ed29b0b4SJens Axboe static int io_timeout_cancel(struct io_ring_ctx *ctx, struct io_cancel_data *cd) 7361ed29b0b4SJens Axboe __must_hold(&ctx->completion_lock) 7362ed29b0b4SJens Axboe { 7363ed29b0b4SJens Axboe struct io_kiocb *req; 7364ed29b0b4SJens Axboe 7365ed29b0b4SJens Axboe spin_lock_irq(&ctx->timeout_lock); 7366ed29b0b4SJens Axboe req = io_timeout_extract(ctx, cd); 7367ed29b0b4SJens Axboe spin_unlock_irq(&ctx->timeout_lock); 7368ed29b0b4SJens Axboe 7369ed29b0b4SJens Axboe if (IS_ERR(req)) 7370ed29b0b4SJens Axboe return PTR_ERR(req); 7371ed29b0b4SJens Axboe io_req_task_queue_fail(req, -ECANCELED); 7372ed29b0b4SJens Axboe return 0; 7373ed29b0b4SJens Axboe } 7374ed29b0b4SJens Axboe 7375ed29b0b4SJens Axboe static clockid_t io_timeout_get_clock(struct io_timeout_data *data) 7376ed29b0b4SJens Axboe { 7377ed29b0b4SJens Axboe switch (data->flags & IORING_TIMEOUT_CLOCK_MASK) { 7378ed29b0b4SJens Axboe case IORING_TIMEOUT_BOOTTIME: 7379ed29b0b4SJens Axboe return CLOCK_BOOTTIME; 7380ed29b0b4SJens Axboe case IORING_TIMEOUT_REALTIME: 7381ed29b0b4SJens Axboe return CLOCK_REALTIME; 7382ed29b0b4SJens Axboe default: 7383ed29b0b4SJens Axboe /* can't happen, vetted at prep time */ 7384ed29b0b4SJens Axboe WARN_ON_ONCE(1); 7385ed29b0b4SJens Axboe fallthrough; 7386ed29b0b4SJens Axboe case 0: 7387ed29b0b4SJens Axboe return CLOCK_MONOTONIC; 7388ed29b0b4SJens Axboe } 7389ed29b0b4SJens Axboe } 7390ed29b0b4SJens Axboe 7391ed29b0b4SJens Axboe static int io_linked_timeout_update(struct io_ring_ctx *ctx, __u64 user_data, 7392ed29b0b4SJens Axboe struct timespec64 *ts, enum hrtimer_mode mode) 7393ed29b0b4SJens Axboe __must_hold(&ctx->timeout_lock) 7394ed29b0b4SJens Axboe { 7395ed29b0b4SJens Axboe struct io_timeout_data *io; 7396a43714acSJens Axboe struct io_timeout *timeout; 7397a43714acSJens Axboe struct io_kiocb *req = NULL; 7398ed29b0b4SJens Axboe 7399a43714acSJens Axboe list_for_each_entry(timeout, &ctx->ltimeout_list, list) { 7400a43714acSJens Axboe struct io_kiocb *tmp = cmd_to_io_kiocb(timeout); 7401a43714acSJens Axboe 7402a43714acSJens Axboe if (user_data == tmp->cqe.user_data) { 7403a43714acSJens Axboe req = tmp; 7404ed29b0b4SJens Axboe break; 7405ed29b0b4SJens Axboe } 7406a43714acSJens Axboe } 7407a43714acSJens Axboe if (!req) 7408ed29b0b4SJens Axboe return -ENOENT; 7409ed29b0b4SJens Axboe 7410ed29b0b4SJens Axboe io = req->async_data; 7411ed29b0b4SJens Axboe if (hrtimer_try_to_cancel(&io->timer) == -1) 7412ed29b0b4SJens Axboe return -EALREADY; 7413ed29b0b4SJens Axboe hrtimer_init(&io->timer, io_timeout_get_clock(io), mode); 7414ed29b0b4SJens Axboe io->timer.function = io_link_timeout_fn; 7415ed29b0b4SJens Axboe hrtimer_start(&io->timer, timespec64_to_ktime(*ts), mode); 7416ed29b0b4SJens Axboe return 0; 7417ed29b0b4SJens Axboe } 7418ed29b0b4SJens Axboe 7419ed29b0b4SJens Axboe static int io_timeout_update(struct io_ring_ctx *ctx, __u64 user_data, 7420ed29b0b4SJens Axboe struct timespec64 *ts, enum hrtimer_mode mode) 7421ed29b0b4SJens Axboe __must_hold(&ctx->timeout_lock) 7422ed29b0b4SJens Axboe { 7423ed29b0b4SJens Axboe struct io_cancel_data cd = { .data = user_data, }; 7424ed29b0b4SJens Axboe struct io_kiocb *req = io_timeout_extract(ctx, &cd); 7425a43714acSJens Axboe struct io_timeout *timeout = io_kiocb_to_cmd(req); 7426ed29b0b4SJens Axboe struct io_timeout_data *data; 7427ed29b0b4SJens Axboe 7428ed29b0b4SJens Axboe if (IS_ERR(req)) 7429ed29b0b4SJens Axboe return PTR_ERR(req); 7430ed29b0b4SJens Axboe 7431a43714acSJens Axboe timeout->off = 0; /* noseq */ 7432ed29b0b4SJens Axboe data = req->async_data; 7433a43714acSJens Axboe list_add_tail(&timeout->list, &ctx->timeout_list); 7434ed29b0b4SJens Axboe hrtimer_init(&data->timer, io_timeout_get_clock(data), mode); 7435ed29b0b4SJens Axboe data->timer.function = io_timeout_fn; 7436ed29b0b4SJens Axboe hrtimer_start(&data->timer, timespec64_to_ktime(*ts), mode); 7437ed29b0b4SJens Axboe return 0; 7438ed29b0b4SJens Axboe } 7439ed29b0b4SJens Axboe 7440ed29b0b4SJens Axboe static int io_timeout_remove_prep(struct io_kiocb *req, 7441ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 7442ed29b0b4SJens Axboe { 7443a43714acSJens Axboe struct io_timeout_rem *tr = io_kiocb_to_cmd(req); 7444ed29b0b4SJens Axboe 7445ed29b0b4SJens Axboe if (unlikely(req->flags & (REQ_F_FIXED_FILE | REQ_F_BUFFER_SELECT))) 7446ed29b0b4SJens Axboe return -EINVAL; 7447ed29b0b4SJens Axboe if (sqe->buf_index || sqe->len || sqe->splice_fd_in) 7448ed29b0b4SJens Axboe return -EINVAL; 7449ed29b0b4SJens Axboe 7450ed29b0b4SJens Axboe tr->ltimeout = false; 7451ed29b0b4SJens Axboe tr->addr = READ_ONCE(sqe->addr); 7452ed29b0b4SJens Axboe tr->flags = READ_ONCE(sqe->timeout_flags); 7453ed29b0b4SJens Axboe if (tr->flags & IORING_TIMEOUT_UPDATE_MASK) { 7454ed29b0b4SJens Axboe if (hweight32(tr->flags & IORING_TIMEOUT_CLOCK_MASK) > 1) 7455ed29b0b4SJens Axboe return -EINVAL; 7456ed29b0b4SJens Axboe if (tr->flags & IORING_LINK_TIMEOUT_UPDATE) 7457ed29b0b4SJens Axboe tr->ltimeout = true; 7458ed29b0b4SJens Axboe if (tr->flags & ~(IORING_TIMEOUT_UPDATE_MASK|IORING_TIMEOUT_ABS)) 7459ed29b0b4SJens Axboe return -EINVAL; 7460ed29b0b4SJens Axboe if (get_timespec64(&tr->ts, u64_to_user_ptr(sqe->addr2))) 7461ed29b0b4SJens Axboe return -EFAULT; 7462ed29b0b4SJens Axboe if (tr->ts.tv_sec < 0 || tr->ts.tv_nsec < 0) 7463ed29b0b4SJens Axboe return -EINVAL; 7464ed29b0b4SJens Axboe } else if (tr->flags) { 7465ed29b0b4SJens Axboe /* timeout removal doesn't support flags */ 7466ed29b0b4SJens Axboe return -EINVAL; 7467ed29b0b4SJens Axboe } 7468ed29b0b4SJens Axboe 7469ed29b0b4SJens Axboe return 0; 7470ed29b0b4SJens Axboe } 7471ed29b0b4SJens Axboe 7472ed29b0b4SJens Axboe static inline enum hrtimer_mode io_translate_timeout_mode(unsigned int flags) 7473ed29b0b4SJens Axboe { 7474ed29b0b4SJens Axboe return (flags & IORING_TIMEOUT_ABS) ? HRTIMER_MODE_ABS 7475ed29b0b4SJens Axboe : HRTIMER_MODE_REL; 7476ed29b0b4SJens Axboe } 7477ed29b0b4SJens Axboe 7478ed29b0b4SJens Axboe /* 7479ed29b0b4SJens Axboe * Remove or update an existing timeout command 7480ed29b0b4SJens Axboe */ 7481ed29b0b4SJens Axboe static int io_timeout_remove(struct io_kiocb *req, unsigned int issue_flags) 7482ed29b0b4SJens Axboe { 7483a43714acSJens Axboe struct io_timeout_rem *tr = io_kiocb_to_cmd(req); 7484ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 7485ed29b0b4SJens Axboe int ret; 7486ed29b0b4SJens Axboe 7487a43714acSJens Axboe if (!(tr->flags & IORING_TIMEOUT_UPDATE)) { 7488ed29b0b4SJens Axboe struct io_cancel_data cd = { .data = tr->addr, }; 7489ed29b0b4SJens Axboe 7490ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 7491ed29b0b4SJens Axboe ret = io_timeout_cancel(ctx, &cd); 7492ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 7493ed29b0b4SJens Axboe } else { 7494ed29b0b4SJens Axboe enum hrtimer_mode mode = io_translate_timeout_mode(tr->flags); 7495ed29b0b4SJens Axboe 7496ed29b0b4SJens Axboe spin_lock_irq(&ctx->timeout_lock); 7497ed29b0b4SJens Axboe if (tr->ltimeout) 7498ed29b0b4SJens Axboe ret = io_linked_timeout_update(ctx, tr->addr, &tr->ts, mode); 7499ed29b0b4SJens Axboe else 7500ed29b0b4SJens Axboe ret = io_timeout_update(ctx, tr->addr, &tr->ts, mode); 7501ed29b0b4SJens Axboe spin_unlock_irq(&ctx->timeout_lock); 7502ed29b0b4SJens Axboe } 7503ed29b0b4SJens Axboe 7504ed29b0b4SJens Axboe if (ret < 0) 7505ed29b0b4SJens Axboe req_set_fail(req); 7506ed29b0b4SJens Axboe io_req_complete_post(req, ret, 0); 7507ed29b0b4SJens Axboe return 0; 7508ed29b0b4SJens Axboe } 7509ed29b0b4SJens Axboe 7510ed29b0b4SJens Axboe static int __io_timeout_prep(struct io_kiocb *req, 7511ed29b0b4SJens Axboe const struct io_uring_sqe *sqe, 7512ed29b0b4SJens Axboe bool is_timeout_link) 7513ed29b0b4SJens Axboe { 7514a43714acSJens Axboe struct io_timeout *timeout = io_kiocb_to_cmd(req); 7515ed29b0b4SJens Axboe struct io_timeout_data *data; 7516ed29b0b4SJens Axboe unsigned flags; 7517ed29b0b4SJens Axboe u32 off = READ_ONCE(sqe->off); 7518ed29b0b4SJens Axboe 7519ed29b0b4SJens Axboe if (sqe->buf_index || sqe->len != 1 || sqe->splice_fd_in) 7520ed29b0b4SJens Axboe return -EINVAL; 7521ed29b0b4SJens Axboe if (off && is_timeout_link) 7522ed29b0b4SJens Axboe return -EINVAL; 7523ed29b0b4SJens Axboe flags = READ_ONCE(sqe->timeout_flags); 7524ed29b0b4SJens Axboe if (flags & ~(IORING_TIMEOUT_ABS | IORING_TIMEOUT_CLOCK_MASK | 7525ed29b0b4SJens Axboe IORING_TIMEOUT_ETIME_SUCCESS)) 7526ed29b0b4SJens Axboe return -EINVAL; 7527ed29b0b4SJens Axboe /* more than one clock specified is invalid, obviously */ 7528ed29b0b4SJens Axboe if (hweight32(flags & IORING_TIMEOUT_CLOCK_MASK) > 1) 7529ed29b0b4SJens Axboe return -EINVAL; 7530ed29b0b4SJens Axboe 7531a43714acSJens Axboe INIT_LIST_HEAD(&timeout->list); 7532a43714acSJens Axboe timeout->off = off; 7533ed29b0b4SJens Axboe if (unlikely(off && !req->ctx->off_timeout_used)) 7534ed29b0b4SJens Axboe req->ctx->off_timeout_used = true; 7535ed29b0b4SJens Axboe 7536ed29b0b4SJens Axboe if (WARN_ON_ONCE(req_has_async_data(req))) 7537ed29b0b4SJens Axboe return -EFAULT; 7538ed29b0b4SJens Axboe if (io_alloc_async_data(req)) 7539ed29b0b4SJens Axboe return -ENOMEM; 7540ed29b0b4SJens Axboe 7541ed29b0b4SJens Axboe data = req->async_data; 7542ed29b0b4SJens Axboe data->req = req; 7543ed29b0b4SJens Axboe data->flags = flags; 7544ed29b0b4SJens Axboe 7545ed29b0b4SJens Axboe if (get_timespec64(&data->ts, u64_to_user_ptr(sqe->addr))) 7546ed29b0b4SJens Axboe return -EFAULT; 7547ed29b0b4SJens Axboe 7548ed29b0b4SJens Axboe if (data->ts.tv_sec < 0 || data->ts.tv_nsec < 0) 7549ed29b0b4SJens Axboe return -EINVAL; 7550ed29b0b4SJens Axboe 7551a43714acSJens Axboe INIT_LIST_HEAD(&timeout->list); 7552ed29b0b4SJens Axboe data->mode = io_translate_timeout_mode(flags); 7553ed29b0b4SJens Axboe hrtimer_init(&data->timer, io_timeout_get_clock(data), data->mode); 7554ed29b0b4SJens Axboe 7555ed29b0b4SJens Axboe if (is_timeout_link) { 7556ed29b0b4SJens Axboe struct io_submit_link *link = &req->ctx->submit_state.link; 7557ed29b0b4SJens Axboe 7558ed29b0b4SJens Axboe if (!link->head) 7559ed29b0b4SJens Axboe return -EINVAL; 7560ed29b0b4SJens Axboe if (link->last->opcode == IORING_OP_LINK_TIMEOUT) 7561ed29b0b4SJens Axboe return -EINVAL; 7562a43714acSJens Axboe timeout->head = link->last; 7563ed29b0b4SJens Axboe link->last->flags |= REQ_F_ARM_LTIMEOUT; 7564ed29b0b4SJens Axboe } 7565ed29b0b4SJens Axboe return 0; 7566ed29b0b4SJens Axboe } 7567ed29b0b4SJens Axboe 7568ed29b0b4SJens Axboe static int io_timeout_prep(struct io_kiocb *req, 7569ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 7570ed29b0b4SJens Axboe { 7571ed29b0b4SJens Axboe return __io_timeout_prep(req, sqe, false); 7572ed29b0b4SJens Axboe } 7573ed29b0b4SJens Axboe 7574ed29b0b4SJens Axboe static int io_link_timeout_prep(struct io_kiocb *req, 7575ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 7576ed29b0b4SJens Axboe { 7577ed29b0b4SJens Axboe return __io_timeout_prep(req, sqe, true); 7578ed29b0b4SJens Axboe } 7579ed29b0b4SJens Axboe 7580ed29b0b4SJens Axboe static int io_timeout(struct io_kiocb *req, unsigned int issue_flags) 7581ed29b0b4SJens Axboe { 7582a43714acSJens Axboe struct io_timeout *timeout = io_kiocb_to_cmd(req); 7583ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 7584ed29b0b4SJens Axboe struct io_timeout_data *data = req->async_data; 7585ed29b0b4SJens Axboe struct list_head *entry; 7586a43714acSJens Axboe u32 tail, off = timeout->off; 7587ed29b0b4SJens Axboe 7588ed29b0b4SJens Axboe spin_lock_irq(&ctx->timeout_lock); 7589ed29b0b4SJens Axboe 7590ed29b0b4SJens Axboe /* 7591ed29b0b4SJens Axboe * sqe->off holds how many events that need to occur for this 7592ed29b0b4SJens Axboe * timeout event to be satisfied. If it isn't set, then this is 7593ed29b0b4SJens Axboe * a pure timeout request, sequence isn't used. 7594ed29b0b4SJens Axboe */ 7595ed29b0b4SJens Axboe if (io_is_timeout_noseq(req)) { 7596ed29b0b4SJens Axboe entry = ctx->timeout_list.prev; 7597ed29b0b4SJens Axboe goto add; 7598ed29b0b4SJens Axboe } 7599ed29b0b4SJens Axboe 7600ed29b0b4SJens Axboe tail = ctx->cached_cq_tail - atomic_read(&ctx->cq_timeouts); 7601a43714acSJens Axboe timeout->target_seq = tail + off; 7602ed29b0b4SJens Axboe 7603ed29b0b4SJens Axboe /* Update the last seq here in case io_flush_timeouts() hasn't. 7604ed29b0b4SJens Axboe * This is safe because ->completion_lock is held, and submissions 7605ed29b0b4SJens Axboe * and completions are never mixed in the same ->completion_lock section. 7606ed29b0b4SJens Axboe */ 7607ed29b0b4SJens Axboe ctx->cq_last_tm_flush = tail; 7608ed29b0b4SJens Axboe 7609ed29b0b4SJens Axboe /* 7610ed29b0b4SJens Axboe * Insertion sort, ensuring the first entry in the list is always 7611ed29b0b4SJens Axboe * the one we need first. 7612ed29b0b4SJens Axboe */ 7613ed29b0b4SJens Axboe list_for_each_prev(entry, &ctx->timeout_list) { 7614a43714acSJens Axboe struct io_timeout *nextt = list_entry(entry, struct io_timeout, list); 7615a43714acSJens Axboe struct io_kiocb *nxt = cmd_to_io_kiocb(nextt); 7616ed29b0b4SJens Axboe 7617ed29b0b4SJens Axboe if (io_is_timeout_noseq(nxt)) 7618ed29b0b4SJens Axboe continue; 7619ed29b0b4SJens Axboe /* nxt.seq is behind @tail, otherwise would've been completed */ 7620a43714acSJens Axboe if (off >= nextt->target_seq - tail) 7621ed29b0b4SJens Axboe break; 7622ed29b0b4SJens Axboe } 7623ed29b0b4SJens Axboe add: 7624a43714acSJens Axboe list_add(&timeout->list, entry); 7625ed29b0b4SJens Axboe data->timer.function = io_timeout_fn; 7626ed29b0b4SJens Axboe hrtimer_start(&data->timer, timespec64_to_ktime(data->ts), data->mode); 7627ed29b0b4SJens Axboe spin_unlock_irq(&ctx->timeout_lock); 7628ed29b0b4SJens Axboe return 0; 7629ed29b0b4SJens Axboe } 7630ed29b0b4SJens Axboe 7631ed29b0b4SJens Axboe static bool io_cancel_cb(struct io_wq_work *work, void *data) 7632ed29b0b4SJens Axboe { 7633ed29b0b4SJens Axboe struct io_kiocb *req = container_of(work, struct io_kiocb, work); 7634ed29b0b4SJens Axboe struct io_cancel_data *cd = data; 7635ed29b0b4SJens Axboe 7636ed29b0b4SJens Axboe if (req->ctx != cd->ctx) 7637ed29b0b4SJens Axboe return false; 7638ed29b0b4SJens Axboe if (cd->flags & IORING_ASYNC_CANCEL_ANY) { 7639ed29b0b4SJens Axboe ; 7640ed29b0b4SJens Axboe } else if (cd->flags & IORING_ASYNC_CANCEL_FD) { 7641ed29b0b4SJens Axboe if (req->file != cd->file) 7642ed29b0b4SJens Axboe return false; 7643ed29b0b4SJens Axboe } else { 7644ed29b0b4SJens Axboe if (req->cqe.user_data != cd->data) 7645ed29b0b4SJens Axboe return false; 7646ed29b0b4SJens Axboe } 7647ed29b0b4SJens Axboe if (cd->flags & (IORING_ASYNC_CANCEL_ALL|IORING_ASYNC_CANCEL_ANY)) { 7648ed29b0b4SJens Axboe if (cd->seq == req->work.cancel_seq) 7649ed29b0b4SJens Axboe return false; 7650ed29b0b4SJens Axboe req->work.cancel_seq = cd->seq; 7651ed29b0b4SJens Axboe } 7652ed29b0b4SJens Axboe return true; 7653ed29b0b4SJens Axboe } 7654ed29b0b4SJens Axboe 7655ed29b0b4SJens Axboe static int io_async_cancel_one(struct io_uring_task *tctx, 7656ed29b0b4SJens Axboe struct io_cancel_data *cd) 7657ed29b0b4SJens Axboe { 7658ed29b0b4SJens Axboe enum io_wq_cancel cancel_ret; 7659ed29b0b4SJens Axboe int ret = 0; 7660ed29b0b4SJens Axboe bool all; 7661ed29b0b4SJens Axboe 7662ed29b0b4SJens Axboe if (!tctx || !tctx->io_wq) 7663ed29b0b4SJens Axboe return -ENOENT; 7664ed29b0b4SJens Axboe 7665ed29b0b4SJens Axboe all = cd->flags & (IORING_ASYNC_CANCEL_ALL|IORING_ASYNC_CANCEL_ANY); 7666ed29b0b4SJens Axboe cancel_ret = io_wq_cancel_cb(tctx->io_wq, io_cancel_cb, cd, all); 7667ed29b0b4SJens Axboe switch (cancel_ret) { 7668ed29b0b4SJens Axboe case IO_WQ_CANCEL_OK: 7669ed29b0b4SJens Axboe ret = 0; 7670ed29b0b4SJens Axboe break; 7671ed29b0b4SJens Axboe case IO_WQ_CANCEL_RUNNING: 7672ed29b0b4SJens Axboe ret = -EALREADY; 7673ed29b0b4SJens Axboe break; 7674ed29b0b4SJens Axboe case IO_WQ_CANCEL_NOTFOUND: 7675ed29b0b4SJens Axboe ret = -ENOENT; 7676ed29b0b4SJens Axboe break; 7677ed29b0b4SJens Axboe } 7678ed29b0b4SJens Axboe 7679ed29b0b4SJens Axboe return ret; 7680ed29b0b4SJens Axboe } 7681ed29b0b4SJens Axboe 7682ed29b0b4SJens Axboe static int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd) 7683ed29b0b4SJens Axboe { 7684ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 7685ed29b0b4SJens Axboe int ret; 7686ed29b0b4SJens Axboe 7687ed29b0b4SJens Axboe WARN_ON_ONCE(!io_wq_current_is_worker() && req->task != current); 7688ed29b0b4SJens Axboe 7689ed29b0b4SJens Axboe ret = io_async_cancel_one(req->task->io_uring, cd); 7690ed29b0b4SJens Axboe /* 7691ed29b0b4SJens Axboe * Fall-through even for -EALREADY, as we may have poll armed 7692ed29b0b4SJens Axboe * that need unarming. 7693ed29b0b4SJens Axboe */ 7694ed29b0b4SJens Axboe if (!ret) 7695ed29b0b4SJens Axboe return 0; 7696ed29b0b4SJens Axboe 7697ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 7698ed29b0b4SJens Axboe ret = io_poll_cancel(ctx, cd); 7699ed29b0b4SJens Axboe if (ret != -ENOENT) 7700ed29b0b4SJens Axboe goto out; 7701ed29b0b4SJens Axboe if (!(cd->flags & IORING_ASYNC_CANCEL_FD)) 7702ed29b0b4SJens Axboe ret = io_timeout_cancel(ctx, cd); 7703ed29b0b4SJens Axboe out: 7704ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 7705ed29b0b4SJens Axboe return ret; 7706ed29b0b4SJens Axboe } 7707ed29b0b4SJens Axboe 7708ed29b0b4SJens Axboe #define CANCEL_FLAGS (IORING_ASYNC_CANCEL_ALL | IORING_ASYNC_CANCEL_FD | \ 7709ed29b0b4SJens Axboe IORING_ASYNC_CANCEL_ANY) 7710ed29b0b4SJens Axboe 7711ed29b0b4SJens Axboe static int io_async_cancel_prep(struct io_kiocb *req, 7712ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 7713ed29b0b4SJens Axboe { 7714f38987f0SJens Axboe struct io_cancel *cancel = io_kiocb_to_cmd(req); 7715f38987f0SJens Axboe 7716ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_BUFFER_SELECT)) 7717ed29b0b4SJens Axboe return -EINVAL; 7718ed29b0b4SJens Axboe if (sqe->off || sqe->len || sqe->splice_fd_in) 7719ed29b0b4SJens Axboe return -EINVAL; 7720ed29b0b4SJens Axboe 7721f38987f0SJens Axboe cancel->addr = READ_ONCE(sqe->addr); 7722f38987f0SJens Axboe cancel->flags = READ_ONCE(sqe->cancel_flags); 7723f38987f0SJens Axboe if (cancel->flags & ~CANCEL_FLAGS) 7724ed29b0b4SJens Axboe return -EINVAL; 7725f38987f0SJens Axboe if (cancel->flags & IORING_ASYNC_CANCEL_FD) { 7726f38987f0SJens Axboe if (cancel->flags & IORING_ASYNC_CANCEL_ANY) 7727ed29b0b4SJens Axboe return -EINVAL; 7728f38987f0SJens Axboe cancel->fd = READ_ONCE(sqe->fd); 7729ed29b0b4SJens Axboe } 7730ed29b0b4SJens Axboe 7731ed29b0b4SJens Axboe return 0; 7732ed29b0b4SJens Axboe } 7733ed29b0b4SJens Axboe 7734ed29b0b4SJens Axboe static int __io_async_cancel(struct io_cancel_data *cd, struct io_kiocb *req, 7735ed29b0b4SJens Axboe unsigned int issue_flags) 7736ed29b0b4SJens Axboe { 7737ed29b0b4SJens Axboe bool all = cd->flags & (IORING_ASYNC_CANCEL_ALL|IORING_ASYNC_CANCEL_ANY); 7738ed29b0b4SJens Axboe struct io_ring_ctx *ctx = cd->ctx; 7739ed29b0b4SJens Axboe struct io_tctx_node *node; 7740ed29b0b4SJens Axboe int ret, nr = 0; 7741ed29b0b4SJens Axboe 7742ed29b0b4SJens Axboe do { 7743ed29b0b4SJens Axboe ret = io_try_cancel(req, cd); 7744ed29b0b4SJens Axboe if (ret == -ENOENT) 7745ed29b0b4SJens Axboe break; 7746ed29b0b4SJens Axboe if (!all) 7747ed29b0b4SJens Axboe return ret; 7748ed29b0b4SJens Axboe nr++; 7749ed29b0b4SJens Axboe } while (1); 7750ed29b0b4SJens Axboe 7751ed29b0b4SJens Axboe /* slow path, try all io-wq's */ 7752ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 7753ed29b0b4SJens Axboe ret = -ENOENT; 7754ed29b0b4SJens Axboe list_for_each_entry(node, &ctx->tctx_list, ctx_node) { 7755ed29b0b4SJens Axboe struct io_uring_task *tctx = node->task->io_uring; 7756ed29b0b4SJens Axboe 7757ed29b0b4SJens Axboe ret = io_async_cancel_one(tctx, cd); 7758ed29b0b4SJens Axboe if (ret != -ENOENT) { 7759ed29b0b4SJens Axboe if (!all) 7760ed29b0b4SJens Axboe break; 7761ed29b0b4SJens Axboe nr++; 7762ed29b0b4SJens Axboe } 7763ed29b0b4SJens Axboe } 7764ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 7765ed29b0b4SJens Axboe return all ? nr : ret; 7766ed29b0b4SJens Axboe } 7767ed29b0b4SJens Axboe 7768ed29b0b4SJens Axboe static int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags) 7769ed29b0b4SJens Axboe { 7770f38987f0SJens Axboe struct io_cancel *cancel = io_kiocb_to_cmd(req); 7771ed29b0b4SJens Axboe struct io_cancel_data cd = { 7772ed29b0b4SJens Axboe .ctx = req->ctx, 7773f38987f0SJens Axboe .data = cancel->addr, 7774f38987f0SJens Axboe .flags = cancel->flags, 7775ed29b0b4SJens Axboe .seq = atomic_inc_return(&req->ctx->cancel_seq), 7776ed29b0b4SJens Axboe }; 7777ed29b0b4SJens Axboe int ret; 7778ed29b0b4SJens Axboe 7779ed29b0b4SJens Axboe if (cd.flags & IORING_ASYNC_CANCEL_FD) { 7780ed29b0b4SJens Axboe if (req->flags & REQ_F_FIXED_FILE) 7781f38987f0SJens Axboe req->file = io_file_get_fixed(req, cancel->fd, 7782ed29b0b4SJens Axboe issue_flags); 7783ed29b0b4SJens Axboe else 7784f38987f0SJens Axboe req->file = io_file_get_normal(req, cancel->fd); 7785ed29b0b4SJens Axboe if (!req->file) { 7786ed29b0b4SJens Axboe ret = -EBADF; 7787ed29b0b4SJens Axboe goto done; 7788ed29b0b4SJens Axboe } 7789ed29b0b4SJens Axboe cd.file = req->file; 7790ed29b0b4SJens Axboe } 7791ed29b0b4SJens Axboe 7792ed29b0b4SJens Axboe ret = __io_async_cancel(&cd, req, issue_flags); 7793ed29b0b4SJens Axboe done: 7794ed29b0b4SJens Axboe if (ret < 0) 7795ed29b0b4SJens Axboe req_set_fail(req); 7796ed29b0b4SJens Axboe io_req_complete_post(req, ret, 0); 7797ed29b0b4SJens Axboe return 0; 7798ed29b0b4SJens Axboe } 7799ed29b0b4SJens Axboe 7800ed29b0b4SJens Axboe static int io_files_update_prep(struct io_kiocb *req, 7801ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 7802ed29b0b4SJens Axboe { 7803ed29b0b4SJens Axboe if (unlikely(req->flags & (REQ_F_FIXED_FILE | REQ_F_BUFFER_SELECT))) 7804ed29b0b4SJens Axboe return -EINVAL; 7805ed29b0b4SJens Axboe if (sqe->rw_flags || sqe->splice_fd_in) 7806ed29b0b4SJens Axboe return -EINVAL; 7807ed29b0b4SJens Axboe 7808ed29b0b4SJens Axboe req->rsrc_update.offset = READ_ONCE(sqe->off); 7809ed29b0b4SJens Axboe req->rsrc_update.nr_args = READ_ONCE(sqe->len); 7810ed29b0b4SJens Axboe if (!req->rsrc_update.nr_args) 7811ed29b0b4SJens Axboe return -EINVAL; 7812ed29b0b4SJens Axboe req->rsrc_update.arg = READ_ONCE(sqe->addr); 7813ed29b0b4SJens Axboe return 0; 7814ed29b0b4SJens Axboe } 7815ed29b0b4SJens Axboe 7816ed29b0b4SJens Axboe static int io_files_update_with_index_alloc(struct io_kiocb *req, 7817ed29b0b4SJens Axboe unsigned int issue_flags) 7818ed29b0b4SJens Axboe { 7819ed29b0b4SJens Axboe __s32 __user *fds = u64_to_user_ptr(req->rsrc_update.arg); 7820ed29b0b4SJens Axboe unsigned int done; 7821ed29b0b4SJens Axboe struct file *file; 7822ed29b0b4SJens Axboe int ret, fd; 7823ed29b0b4SJens Axboe 7824ed29b0b4SJens Axboe if (!req->ctx->file_data) 7825ed29b0b4SJens Axboe return -ENXIO; 7826ed29b0b4SJens Axboe 7827ed29b0b4SJens Axboe for (done = 0; done < req->rsrc_update.nr_args; done++) { 7828ed29b0b4SJens Axboe if (copy_from_user(&fd, &fds[done], sizeof(fd))) { 7829ed29b0b4SJens Axboe ret = -EFAULT; 7830ed29b0b4SJens Axboe break; 7831ed29b0b4SJens Axboe } 7832ed29b0b4SJens Axboe 7833ed29b0b4SJens Axboe file = fget(fd); 7834ed29b0b4SJens Axboe if (!file) { 7835ed29b0b4SJens Axboe ret = -EBADF; 7836ed29b0b4SJens Axboe break; 7837ed29b0b4SJens Axboe } 7838ed29b0b4SJens Axboe ret = io_fixed_fd_install(req, issue_flags, file, 7839ed29b0b4SJens Axboe IORING_FILE_INDEX_ALLOC); 7840ed29b0b4SJens Axboe if (ret < 0) 7841ed29b0b4SJens Axboe break; 7842ed29b0b4SJens Axboe if (copy_to_user(&fds[done], &ret, sizeof(ret))) { 7843ed29b0b4SJens Axboe __io_close_fixed(req, issue_flags, ret); 7844ed29b0b4SJens Axboe ret = -EFAULT; 7845ed29b0b4SJens Axboe break; 7846ed29b0b4SJens Axboe } 7847ed29b0b4SJens Axboe } 7848ed29b0b4SJens Axboe 7849ed29b0b4SJens Axboe if (done) 7850ed29b0b4SJens Axboe return done; 7851ed29b0b4SJens Axboe return ret; 7852ed29b0b4SJens Axboe } 7853ed29b0b4SJens Axboe 7854ed29b0b4SJens Axboe static int io_files_update(struct io_kiocb *req, unsigned int issue_flags) 7855ed29b0b4SJens Axboe { 7856ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 7857ed29b0b4SJens Axboe struct io_uring_rsrc_update2 up; 7858ed29b0b4SJens Axboe int ret; 7859ed29b0b4SJens Axboe 7860ed29b0b4SJens Axboe up.offset = req->rsrc_update.offset; 7861ed29b0b4SJens Axboe up.data = req->rsrc_update.arg; 7862ed29b0b4SJens Axboe up.nr = 0; 7863ed29b0b4SJens Axboe up.tags = 0; 7864ed29b0b4SJens Axboe up.resv = 0; 7865ed29b0b4SJens Axboe up.resv2 = 0; 7866ed29b0b4SJens Axboe 7867ed29b0b4SJens Axboe if (req->rsrc_update.offset == IORING_FILE_INDEX_ALLOC) { 7868ed29b0b4SJens Axboe ret = io_files_update_with_index_alloc(req, issue_flags); 7869ed29b0b4SJens Axboe } else { 7870ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 7871ed29b0b4SJens Axboe ret = __io_register_rsrc_update(ctx, IORING_RSRC_FILE, 7872ed29b0b4SJens Axboe &up, req->rsrc_update.nr_args); 7873ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 7874ed29b0b4SJens Axboe } 7875ed29b0b4SJens Axboe 7876ed29b0b4SJens Axboe if (ret < 0) 7877ed29b0b4SJens Axboe req_set_fail(req); 7878ed29b0b4SJens Axboe __io_req_complete(req, issue_flags, ret, 0); 7879ed29b0b4SJens Axboe return 0; 7880ed29b0b4SJens Axboe } 7881ed29b0b4SJens Axboe 7882ed29b0b4SJens Axboe static int io_req_prep_async(struct io_kiocb *req) 7883ed29b0b4SJens Axboe { 7884ed29b0b4SJens Axboe const struct io_op_def *def = &io_op_defs[req->opcode]; 7885ed29b0b4SJens Axboe 7886ed29b0b4SJens Axboe /* assign early for deferred execution for non-fixed file */ 7887ed29b0b4SJens Axboe if (def->needs_file && !(req->flags & REQ_F_FIXED_FILE)) 7888ed29b0b4SJens Axboe req->file = io_file_get_normal(req, req->cqe.fd); 7889dc919cafSJens Axboe if (!def->prep_async) 7890ed29b0b4SJens Axboe return 0; 7891ed29b0b4SJens Axboe if (WARN_ON_ONCE(req_has_async_data(req))) 7892ed29b0b4SJens Axboe return -EFAULT; 7893ed29b0b4SJens Axboe if (io_alloc_async_data(req)) 7894ed29b0b4SJens Axboe return -EAGAIN; 7895ed29b0b4SJens Axboe 7896dc919cafSJens Axboe return def->prep_async(req); 7897ed29b0b4SJens Axboe } 7898ed29b0b4SJens Axboe 7899ed29b0b4SJens Axboe static u32 io_get_sequence(struct io_kiocb *req) 7900ed29b0b4SJens Axboe { 7901ed29b0b4SJens Axboe u32 seq = req->ctx->cached_sq_head; 7902ed29b0b4SJens Axboe struct io_kiocb *cur; 7903ed29b0b4SJens Axboe 7904ed29b0b4SJens Axboe /* need original cached_sq_head, but it was increased for each req */ 7905ed29b0b4SJens Axboe io_for_each_link(cur, req) 7906ed29b0b4SJens Axboe seq--; 7907ed29b0b4SJens Axboe return seq; 7908ed29b0b4SJens Axboe } 7909ed29b0b4SJens Axboe 7910ed29b0b4SJens Axboe static __cold void io_drain_req(struct io_kiocb *req) 7911ed29b0b4SJens Axboe { 7912ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 7913ed29b0b4SJens Axboe struct io_defer_entry *de; 7914ed29b0b4SJens Axboe int ret; 7915ed29b0b4SJens Axboe u32 seq = io_get_sequence(req); 7916ed29b0b4SJens Axboe 7917ed29b0b4SJens Axboe /* Still need defer if there is pending req in defer list. */ 7918ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 7919ed29b0b4SJens Axboe if (!req_need_defer(req, seq) && list_empty_careful(&ctx->defer_list)) { 7920ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 7921ed29b0b4SJens Axboe queue: 7922ed29b0b4SJens Axboe ctx->drain_active = false; 7923ed29b0b4SJens Axboe io_req_task_queue(req); 7924ed29b0b4SJens Axboe return; 7925ed29b0b4SJens Axboe } 7926ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 7927ed29b0b4SJens Axboe 7928ed29b0b4SJens Axboe ret = io_req_prep_async(req); 7929ed29b0b4SJens Axboe if (ret) { 7930ed29b0b4SJens Axboe fail: 7931ed29b0b4SJens Axboe io_req_complete_failed(req, ret); 7932ed29b0b4SJens Axboe return; 7933ed29b0b4SJens Axboe } 7934ed29b0b4SJens Axboe io_prep_async_link(req); 7935ed29b0b4SJens Axboe de = kmalloc(sizeof(*de), GFP_KERNEL); 7936ed29b0b4SJens Axboe if (!de) { 7937ed29b0b4SJens Axboe ret = -ENOMEM; 7938ed29b0b4SJens Axboe goto fail; 7939ed29b0b4SJens Axboe } 7940ed29b0b4SJens Axboe 7941ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 7942ed29b0b4SJens Axboe if (!req_need_defer(req, seq) && list_empty(&ctx->defer_list)) { 7943ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 7944ed29b0b4SJens Axboe kfree(de); 7945ed29b0b4SJens Axboe goto queue; 7946ed29b0b4SJens Axboe } 7947ed29b0b4SJens Axboe 7948ed29b0b4SJens Axboe trace_io_uring_defer(ctx, req, req->cqe.user_data, req->opcode); 7949ed29b0b4SJens Axboe de->req = req; 7950ed29b0b4SJens Axboe de->seq = seq; 7951ed29b0b4SJens Axboe list_add_tail(&de->list, &ctx->defer_list); 7952ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 7953ed29b0b4SJens Axboe } 7954ed29b0b4SJens Axboe 7955ed29b0b4SJens Axboe static void io_clean_op(struct io_kiocb *req) 7956ed29b0b4SJens Axboe { 7957ed29b0b4SJens Axboe if (req->flags & REQ_F_BUFFER_SELECTED) { 7958ed29b0b4SJens Axboe spin_lock(&req->ctx->completion_lock); 7959ed29b0b4SJens Axboe io_put_kbuf_comp(req); 7960ed29b0b4SJens Axboe spin_unlock(&req->ctx->completion_lock); 7961ed29b0b4SJens Axboe } 7962ed29b0b4SJens Axboe 7963ed29b0b4SJens Axboe if (req->flags & REQ_F_NEED_CLEANUP) { 7964ed29b0b4SJens Axboe switch (req->opcode) { 7965ed29b0b4SJens Axboe case IORING_OP_READV: 7966ed29b0b4SJens Axboe case IORING_OP_READ_FIXED: 7967ed29b0b4SJens Axboe case IORING_OP_READ: 7968ed29b0b4SJens Axboe case IORING_OP_WRITEV: 7969ed29b0b4SJens Axboe case IORING_OP_WRITE_FIXED: 7970ed29b0b4SJens Axboe case IORING_OP_WRITE: { 7971ed29b0b4SJens Axboe struct io_async_rw *io = req->async_data; 7972ed29b0b4SJens Axboe 7973ed29b0b4SJens Axboe kfree(io->free_iovec); 7974ed29b0b4SJens Axboe break; 7975ed29b0b4SJens Axboe } 7976ed29b0b4SJens Axboe case IORING_OP_RECVMSG: 7977ed29b0b4SJens Axboe case IORING_OP_SENDMSG: { 7978ed29b0b4SJens Axboe struct io_async_msghdr *io = req->async_data; 7979ed29b0b4SJens Axboe 7980ed29b0b4SJens Axboe kfree(io->free_iov); 7981ed29b0b4SJens Axboe break; 7982ed29b0b4SJens Axboe } 7983ed29b0b4SJens Axboe case IORING_OP_OPENAT: 7984dd752582SJens Axboe case IORING_OP_OPENAT2: { 7985dd752582SJens Axboe struct io_open *open = io_kiocb_to_cmd(req); 7986dd752582SJens Axboe 7987dd752582SJens Axboe if (open->filename) 7988dd752582SJens Axboe putname(open->filename); 7989ed29b0b4SJens Axboe break; 7990dd752582SJens Axboe } 7991bb040a21SJens Axboe case IORING_OP_RENAMEAT: { 7992bb040a21SJens Axboe struct io_rename *ren = io_kiocb_to_cmd(req); 7993bb040a21SJens Axboe 7994bb040a21SJens Axboe putname(ren->oldpath); 7995bb040a21SJens Axboe putname(ren->newpath); 7996ed29b0b4SJens Axboe break; 7997bb040a21SJens Axboe } 7998bb040a21SJens Axboe case IORING_OP_UNLINKAT: { 7999bb040a21SJens Axboe struct io_unlink *ul = io_kiocb_to_cmd(req); 8000bb040a21SJens Axboe 8001bb040a21SJens Axboe putname(ul->filename); 8002ed29b0b4SJens Axboe break; 8003bb040a21SJens Axboe } 8004bb040a21SJens Axboe case IORING_OP_MKDIRAT: { 8005bb040a21SJens Axboe struct io_mkdir *md = io_kiocb_to_cmd(req); 8006bb040a21SJens Axboe 8007bb040a21SJens Axboe putname(md->filename); 8008ed29b0b4SJens Axboe break; 8009bb040a21SJens Axboe } 8010bb040a21SJens Axboe case IORING_OP_SYMLINKAT: { 8011bb040a21SJens Axboe struct io_symlink *sl = io_kiocb_to_cmd(req); 8012bb040a21SJens Axboe 8013bb040a21SJens Axboe putname(sl->oldpath); 8014bb040a21SJens Axboe putname(sl->newpath); 8015ed29b0b4SJens Axboe break; 8016bb040a21SJens Axboe } 8017bb040a21SJens Axboe case IORING_OP_LINKAT: { 8018bb040a21SJens Axboe struct io_hardlink *hl = io_kiocb_to_cmd(req); 8019bb040a21SJens Axboe 8020bb040a21SJens Axboe putname(hl->oldpath); 8021bb040a21SJens Axboe putname(hl->newpath); 8022ed29b0b4SJens Axboe break; 8023bb040a21SJens Axboe } 8024bb040a21SJens Axboe case IORING_OP_STATX: { 8025bb040a21SJens Axboe struct io_statx *sx = io_kiocb_to_cmd(req); 8026bb040a21SJens Axboe 8027bb040a21SJens Axboe if (sx->filename) 8028bb040a21SJens Axboe putname(sx->filename); 8029ed29b0b4SJens Axboe break; 8030bb040a21SJens Axboe } 8031ed29b0b4SJens Axboe case IORING_OP_SETXATTR: 8032ed29b0b4SJens Axboe case IORING_OP_FSETXATTR: 8033ed29b0b4SJens Axboe case IORING_OP_GETXATTR: 8034ed29b0b4SJens Axboe case IORING_OP_FGETXATTR: 8035ed29b0b4SJens Axboe __io_xattr_finish(req); 8036ed29b0b4SJens Axboe break; 8037ed29b0b4SJens Axboe } 8038ed29b0b4SJens Axboe } 8039ed29b0b4SJens Axboe if ((req->flags & REQ_F_POLLED) && req->apoll) { 8040ed29b0b4SJens Axboe kfree(req->apoll->double_poll); 8041ed29b0b4SJens Axboe kfree(req->apoll); 8042ed29b0b4SJens Axboe req->apoll = NULL; 8043ed29b0b4SJens Axboe } 8044ed29b0b4SJens Axboe if (req->flags & REQ_F_INFLIGHT) { 8045ed29b0b4SJens Axboe struct io_uring_task *tctx = req->task->io_uring; 8046ed29b0b4SJens Axboe 8047ed29b0b4SJens Axboe atomic_dec(&tctx->inflight_tracked); 8048ed29b0b4SJens Axboe } 8049ed29b0b4SJens Axboe if (req->flags & REQ_F_CREDS) 8050ed29b0b4SJens Axboe put_cred(req->creds); 8051ed29b0b4SJens Axboe if (req->flags & REQ_F_ASYNC_DATA) { 8052ed29b0b4SJens Axboe kfree(req->async_data); 8053ed29b0b4SJens Axboe req->async_data = NULL; 8054ed29b0b4SJens Axboe } 8055ed29b0b4SJens Axboe req->flags &= ~IO_REQ_CLEAN_FLAGS; 8056ed29b0b4SJens Axboe } 8057ed29b0b4SJens Axboe 8058ed29b0b4SJens Axboe static bool io_assign_file(struct io_kiocb *req, unsigned int issue_flags) 8059ed29b0b4SJens Axboe { 8060ed29b0b4SJens Axboe if (req->file || !io_op_defs[req->opcode].needs_file) 8061ed29b0b4SJens Axboe return true; 8062ed29b0b4SJens Axboe 8063ed29b0b4SJens Axboe if (req->flags & REQ_F_FIXED_FILE) 8064ed29b0b4SJens Axboe req->file = io_file_get_fixed(req, req->cqe.fd, issue_flags); 8065ed29b0b4SJens Axboe else 8066ed29b0b4SJens Axboe req->file = io_file_get_normal(req, req->cqe.fd); 8067ed29b0b4SJens Axboe 8068ed29b0b4SJens Axboe return !!req->file; 8069ed29b0b4SJens Axboe } 8070ed29b0b4SJens Axboe 8071ed29b0b4SJens Axboe static int io_issue_sqe(struct io_kiocb *req, unsigned int issue_flags) 8072ed29b0b4SJens Axboe { 8073ed29b0b4SJens Axboe const struct io_op_def *def = &io_op_defs[req->opcode]; 8074ed29b0b4SJens Axboe const struct cred *creds = NULL; 8075ed29b0b4SJens Axboe int ret; 8076ed29b0b4SJens Axboe 8077ed29b0b4SJens Axboe if (unlikely(!io_assign_file(req, issue_flags))) 8078ed29b0b4SJens Axboe return -EBADF; 8079ed29b0b4SJens Axboe 8080ed29b0b4SJens Axboe if (unlikely((req->flags & REQ_F_CREDS) && req->creds != current_cred())) 8081ed29b0b4SJens Axboe creds = override_creds(req->creds); 8082ed29b0b4SJens Axboe 8083ed29b0b4SJens Axboe if (!def->audit_skip) 8084ed29b0b4SJens Axboe audit_uring_entry(req->opcode); 8085ed29b0b4SJens Axboe 8086ed29b0b4SJens Axboe ret = def->issue(req, issue_flags); 8087ed29b0b4SJens Axboe 8088ed29b0b4SJens Axboe if (!def->audit_skip) 8089ed29b0b4SJens Axboe audit_uring_exit(!ret, ret); 8090ed29b0b4SJens Axboe 8091ed29b0b4SJens Axboe if (creds) 8092ed29b0b4SJens Axboe revert_creds(creds); 8093ed29b0b4SJens Axboe if (ret) 8094ed29b0b4SJens Axboe return ret; 8095ed29b0b4SJens Axboe /* If the op doesn't have a file, we're not polling for it */ 8096ed29b0b4SJens Axboe if ((req->ctx->flags & IORING_SETUP_IOPOLL) && req->file) 8097ed29b0b4SJens Axboe io_iopoll_req_issued(req, issue_flags); 8098ed29b0b4SJens Axboe 8099ed29b0b4SJens Axboe return 0; 8100ed29b0b4SJens Axboe } 8101ed29b0b4SJens Axboe 8102ed29b0b4SJens Axboe static struct io_wq_work *io_wq_free_work(struct io_wq_work *work) 8103ed29b0b4SJens Axboe { 8104ed29b0b4SJens Axboe struct io_kiocb *req = container_of(work, struct io_kiocb, work); 8105ed29b0b4SJens Axboe 8106ed29b0b4SJens Axboe req = io_put_req_find_next(req); 8107ed29b0b4SJens Axboe return req ? &req->work : NULL; 8108ed29b0b4SJens Axboe } 8109ed29b0b4SJens Axboe 8110ed29b0b4SJens Axboe static void io_wq_submit_work(struct io_wq_work *work) 8111ed29b0b4SJens Axboe { 8112ed29b0b4SJens Axboe struct io_kiocb *req = container_of(work, struct io_kiocb, work); 8113ed29b0b4SJens Axboe const struct io_op_def *def = &io_op_defs[req->opcode]; 8114ed29b0b4SJens Axboe unsigned int issue_flags = IO_URING_F_UNLOCKED; 8115ed29b0b4SJens Axboe bool needs_poll = false; 8116ed29b0b4SJens Axboe int ret = 0, err = -ECANCELED; 8117ed29b0b4SJens Axboe 8118ed29b0b4SJens Axboe /* one will be dropped by ->io_free_work() after returning to io-wq */ 8119ed29b0b4SJens Axboe if (!(req->flags & REQ_F_REFCOUNT)) 8120ed29b0b4SJens Axboe __io_req_set_refcount(req, 2); 8121ed29b0b4SJens Axboe else 8122ed29b0b4SJens Axboe req_ref_get(req); 8123ed29b0b4SJens Axboe 8124ed29b0b4SJens Axboe io_arm_ltimeout(req); 8125ed29b0b4SJens Axboe 8126ed29b0b4SJens Axboe /* either cancelled or io-wq is dying, so don't touch tctx->iowq */ 8127ed29b0b4SJens Axboe if (work->flags & IO_WQ_WORK_CANCEL) { 8128ed29b0b4SJens Axboe fail: 8129ed29b0b4SJens Axboe io_req_task_queue_fail(req, err); 8130ed29b0b4SJens Axboe return; 8131ed29b0b4SJens Axboe } 8132ed29b0b4SJens Axboe if (!io_assign_file(req, issue_flags)) { 8133ed29b0b4SJens Axboe err = -EBADF; 8134ed29b0b4SJens Axboe work->flags |= IO_WQ_WORK_CANCEL; 8135ed29b0b4SJens Axboe goto fail; 8136ed29b0b4SJens Axboe } 8137ed29b0b4SJens Axboe 8138ed29b0b4SJens Axboe if (req->flags & REQ_F_FORCE_ASYNC) { 8139ed29b0b4SJens Axboe bool opcode_poll = def->pollin || def->pollout; 8140ed29b0b4SJens Axboe 8141ed29b0b4SJens Axboe if (opcode_poll && file_can_poll(req->file)) { 8142ed29b0b4SJens Axboe needs_poll = true; 8143ed29b0b4SJens Axboe issue_flags |= IO_URING_F_NONBLOCK; 8144ed29b0b4SJens Axboe } 8145ed29b0b4SJens Axboe } 8146ed29b0b4SJens Axboe 8147ed29b0b4SJens Axboe do { 8148ed29b0b4SJens Axboe ret = io_issue_sqe(req, issue_flags); 8149ed29b0b4SJens Axboe if (ret != -EAGAIN) 8150ed29b0b4SJens Axboe break; 8151ed29b0b4SJens Axboe /* 8152ed29b0b4SJens Axboe * We can get EAGAIN for iopolled IO even though we're 8153ed29b0b4SJens Axboe * forcing a sync submission from here, since we can't 8154ed29b0b4SJens Axboe * wait for request slots on the block side. 8155ed29b0b4SJens Axboe */ 8156ed29b0b4SJens Axboe if (!needs_poll) { 8157ed29b0b4SJens Axboe if (!(req->ctx->flags & IORING_SETUP_IOPOLL)) 8158ed29b0b4SJens Axboe break; 8159ed29b0b4SJens Axboe cond_resched(); 8160ed29b0b4SJens Axboe continue; 8161ed29b0b4SJens Axboe } 8162ed29b0b4SJens Axboe 8163ed29b0b4SJens Axboe if (io_arm_poll_handler(req, issue_flags) == IO_APOLL_OK) 8164ed29b0b4SJens Axboe return; 8165ed29b0b4SJens Axboe /* aborted or ready, in either case retry blocking */ 8166ed29b0b4SJens Axboe needs_poll = false; 8167ed29b0b4SJens Axboe issue_flags &= ~IO_URING_F_NONBLOCK; 8168ed29b0b4SJens Axboe } while (1); 8169ed29b0b4SJens Axboe 8170ed29b0b4SJens Axboe /* avoid locking problems by failing it from a clean context */ 8171ed29b0b4SJens Axboe if (ret) 8172ed29b0b4SJens Axboe io_req_task_queue_fail(req, ret); 8173ed29b0b4SJens Axboe } 8174ed29b0b4SJens Axboe 8175ed29b0b4SJens Axboe static inline struct io_fixed_file *io_fixed_file_slot(struct io_file_table *table, 8176ed29b0b4SJens Axboe unsigned i) 8177ed29b0b4SJens Axboe { 8178ed29b0b4SJens Axboe return &table->files[i]; 8179ed29b0b4SJens Axboe } 8180ed29b0b4SJens Axboe 8181ed29b0b4SJens Axboe static inline struct file *io_file_from_index(struct io_ring_ctx *ctx, 8182ed29b0b4SJens Axboe int index) 8183ed29b0b4SJens Axboe { 8184ed29b0b4SJens Axboe struct io_fixed_file *slot = io_fixed_file_slot(&ctx->file_table, index); 8185ed29b0b4SJens Axboe 8186ed29b0b4SJens Axboe return (struct file *) (slot->file_ptr & FFS_MASK); 8187ed29b0b4SJens Axboe } 8188ed29b0b4SJens Axboe 8189ed29b0b4SJens Axboe static void io_fixed_file_set(struct io_fixed_file *file_slot, struct file *file) 8190ed29b0b4SJens Axboe { 8191ed29b0b4SJens Axboe unsigned long file_ptr = (unsigned long) file; 8192ed29b0b4SJens Axboe 8193ed29b0b4SJens Axboe file_ptr |= io_file_get_flags(file); 8194ed29b0b4SJens Axboe file_slot->file_ptr = file_ptr; 8195ed29b0b4SJens Axboe } 8196ed29b0b4SJens Axboe 8197ed29b0b4SJens Axboe static inline struct file *io_file_get_fixed(struct io_kiocb *req, int fd, 8198ed29b0b4SJens Axboe unsigned int issue_flags) 8199ed29b0b4SJens Axboe { 8200ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 8201ed29b0b4SJens Axboe struct file *file = NULL; 8202ed29b0b4SJens Axboe unsigned long file_ptr; 8203ed29b0b4SJens Axboe 8204ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 8205ed29b0b4SJens Axboe 8206ed29b0b4SJens Axboe if (unlikely((unsigned int)fd >= ctx->nr_user_files)) 8207ed29b0b4SJens Axboe goto out; 8208ed29b0b4SJens Axboe fd = array_index_nospec(fd, ctx->nr_user_files); 8209ed29b0b4SJens Axboe file_ptr = io_fixed_file_slot(&ctx->file_table, fd)->file_ptr; 8210ed29b0b4SJens Axboe file = (struct file *) (file_ptr & FFS_MASK); 8211ed29b0b4SJens Axboe file_ptr &= ~FFS_MASK; 8212ed29b0b4SJens Axboe /* mask in overlapping REQ_F and FFS bits */ 8213ed29b0b4SJens Axboe req->flags |= (file_ptr << REQ_F_SUPPORT_NOWAIT_BIT); 8214ed29b0b4SJens Axboe io_req_set_rsrc_node(req, ctx, 0); 8215ed29b0b4SJens Axboe WARN_ON_ONCE(file && !test_bit(fd, ctx->file_table.bitmap)); 8216ed29b0b4SJens Axboe out: 8217ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 8218ed29b0b4SJens Axboe return file; 8219ed29b0b4SJens Axboe } 8220ed29b0b4SJens Axboe 8221ed29b0b4SJens Axboe static struct file *io_file_get_normal(struct io_kiocb *req, int fd) 8222ed29b0b4SJens Axboe { 8223ed29b0b4SJens Axboe struct file *file = fget(fd); 8224ed29b0b4SJens Axboe 8225ed29b0b4SJens Axboe trace_io_uring_file_get(req->ctx, req, req->cqe.user_data, fd); 8226ed29b0b4SJens Axboe 8227ed29b0b4SJens Axboe /* we don't allow fixed io_uring files */ 8228ed29b0b4SJens Axboe if (file && file->f_op == &io_uring_fops) 8229ed29b0b4SJens Axboe io_req_track_inflight(req); 8230ed29b0b4SJens Axboe return file; 8231ed29b0b4SJens Axboe } 8232ed29b0b4SJens Axboe 8233ed29b0b4SJens Axboe static void io_req_task_link_timeout(struct io_kiocb *req, bool *locked) 8234ed29b0b4SJens Axboe { 8235a43714acSJens Axboe struct io_timeout *timeout = io_kiocb_to_cmd(req); 8236a43714acSJens Axboe struct io_kiocb *prev = timeout->prev; 8237ed29b0b4SJens Axboe int ret = -ENOENT; 8238ed29b0b4SJens Axboe 8239ed29b0b4SJens Axboe if (prev) { 8240ed29b0b4SJens Axboe if (!(req->task->flags & PF_EXITING)) { 8241ed29b0b4SJens Axboe struct io_cancel_data cd = { 8242ed29b0b4SJens Axboe .ctx = req->ctx, 8243ed29b0b4SJens Axboe .data = prev->cqe.user_data, 8244ed29b0b4SJens Axboe }; 8245ed29b0b4SJens Axboe 8246ed29b0b4SJens Axboe ret = io_try_cancel(req, &cd); 8247ed29b0b4SJens Axboe } 8248ed29b0b4SJens Axboe io_req_complete_post(req, ret ?: -ETIME, 0); 8249ed29b0b4SJens Axboe io_put_req(prev); 8250ed29b0b4SJens Axboe } else { 8251ed29b0b4SJens Axboe io_req_complete_post(req, -ETIME, 0); 8252ed29b0b4SJens Axboe } 8253ed29b0b4SJens Axboe } 8254ed29b0b4SJens Axboe 8255ed29b0b4SJens Axboe static enum hrtimer_restart io_link_timeout_fn(struct hrtimer *timer) 8256ed29b0b4SJens Axboe { 8257ed29b0b4SJens Axboe struct io_timeout_data *data = container_of(timer, 8258ed29b0b4SJens Axboe struct io_timeout_data, timer); 8259ed29b0b4SJens Axboe struct io_kiocb *prev, *req = data->req; 8260a43714acSJens Axboe struct io_timeout *timeout = io_kiocb_to_cmd(req); 8261ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 8262ed29b0b4SJens Axboe unsigned long flags; 8263ed29b0b4SJens Axboe 8264ed29b0b4SJens Axboe spin_lock_irqsave(&ctx->timeout_lock, flags); 8265a43714acSJens Axboe prev = timeout->head; 8266a43714acSJens Axboe timeout->head = NULL; 8267ed29b0b4SJens Axboe 8268ed29b0b4SJens Axboe /* 8269ed29b0b4SJens Axboe * We don't expect the list to be empty, that will only happen if we 8270ed29b0b4SJens Axboe * race with the completion of the linked work. 8271ed29b0b4SJens Axboe */ 8272ed29b0b4SJens Axboe if (prev) { 8273ed29b0b4SJens Axboe io_remove_next_linked(prev); 8274ed29b0b4SJens Axboe if (!req_ref_inc_not_zero(prev)) 8275ed29b0b4SJens Axboe prev = NULL; 8276ed29b0b4SJens Axboe } 8277a43714acSJens Axboe list_del(&timeout->list); 8278a43714acSJens Axboe timeout->prev = prev; 8279ed29b0b4SJens Axboe spin_unlock_irqrestore(&ctx->timeout_lock, flags); 8280ed29b0b4SJens Axboe 8281ed29b0b4SJens Axboe req->io_task_work.func = io_req_task_link_timeout; 8282ed29b0b4SJens Axboe io_req_task_work_add(req); 8283ed29b0b4SJens Axboe return HRTIMER_NORESTART; 8284ed29b0b4SJens Axboe } 8285ed29b0b4SJens Axboe 8286ed29b0b4SJens Axboe static void io_queue_linked_timeout(struct io_kiocb *req) 8287ed29b0b4SJens Axboe { 8288a43714acSJens Axboe struct io_timeout *timeout = io_kiocb_to_cmd(req); 8289ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 8290ed29b0b4SJens Axboe 8291ed29b0b4SJens Axboe spin_lock_irq(&ctx->timeout_lock); 8292ed29b0b4SJens Axboe /* 8293ed29b0b4SJens Axboe * If the back reference is NULL, then our linked request finished 8294ed29b0b4SJens Axboe * before we got a chance to setup the timer 8295ed29b0b4SJens Axboe */ 8296a43714acSJens Axboe if (timeout->head) { 8297ed29b0b4SJens Axboe struct io_timeout_data *data = req->async_data; 8298ed29b0b4SJens Axboe 8299ed29b0b4SJens Axboe data->timer.function = io_link_timeout_fn; 8300ed29b0b4SJens Axboe hrtimer_start(&data->timer, timespec64_to_ktime(data->ts), 8301ed29b0b4SJens Axboe data->mode); 8302a43714acSJens Axboe list_add_tail(&timeout->list, &ctx->ltimeout_list); 8303ed29b0b4SJens Axboe } 8304ed29b0b4SJens Axboe spin_unlock_irq(&ctx->timeout_lock); 8305ed29b0b4SJens Axboe /* drop submission reference */ 8306ed29b0b4SJens Axboe io_put_req(req); 8307ed29b0b4SJens Axboe } 8308ed29b0b4SJens Axboe 8309ed29b0b4SJens Axboe static void io_queue_async(struct io_kiocb *req, int ret) 8310ed29b0b4SJens Axboe __must_hold(&req->ctx->uring_lock) 8311ed29b0b4SJens Axboe { 8312ed29b0b4SJens Axboe struct io_kiocb *linked_timeout; 8313ed29b0b4SJens Axboe 8314ed29b0b4SJens Axboe if (ret != -EAGAIN || (req->flags & REQ_F_NOWAIT)) { 8315ed29b0b4SJens Axboe io_req_complete_failed(req, ret); 8316ed29b0b4SJens Axboe return; 8317ed29b0b4SJens Axboe } 8318ed29b0b4SJens Axboe 8319ed29b0b4SJens Axboe linked_timeout = io_prep_linked_timeout(req); 8320ed29b0b4SJens Axboe 8321ed29b0b4SJens Axboe switch (io_arm_poll_handler(req, 0)) { 8322ed29b0b4SJens Axboe case IO_APOLL_READY: 8323ed29b0b4SJens Axboe io_req_task_queue(req); 8324ed29b0b4SJens Axboe break; 8325ed29b0b4SJens Axboe case IO_APOLL_ABORTED: 8326ed29b0b4SJens Axboe /* 8327ed29b0b4SJens Axboe * Queued up for async execution, worker will release 8328ed29b0b4SJens Axboe * submit reference when the iocb is actually submitted. 8329ed29b0b4SJens Axboe */ 8330ed29b0b4SJens Axboe io_kbuf_recycle(req, 0); 8331ed29b0b4SJens Axboe io_queue_iowq(req, NULL); 8332ed29b0b4SJens Axboe break; 8333ed29b0b4SJens Axboe case IO_APOLL_OK: 8334ed29b0b4SJens Axboe break; 8335ed29b0b4SJens Axboe } 8336ed29b0b4SJens Axboe 8337ed29b0b4SJens Axboe if (linked_timeout) 8338ed29b0b4SJens Axboe io_queue_linked_timeout(linked_timeout); 8339ed29b0b4SJens Axboe } 8340ed29b0b4SJens Axboe 8341ed29b0b4SJens Axboe static inline void io_queue_sqe(struct io_kiocb *req) 8342ed29b0b4SJens Axboe __must_hold(&req->ctx->uring_lock) 8343ed29b0b4SJens Axboe { 8344ed29b0b4SJens Axboe int ret; 8345ed29b0b4SJens Axboe 8346ed29b0b4SJens Axboe ret = io_issue_sqe(req, IO_URING_F_NONBLOCK|IO_URING_F_COMPLETE_DEFER); 8347ed29b0b4SJens Axboe 8348ed29b0b4SJens Axboe if (req->flags & REQ_F_COMPLETE_INLINE) { 8349ed29b0b4SJens Axboe io_req_add_compl_list(req); 8350ed29b0b4SJens Axboe return; 8351ed29b0b4SJens Axboe } 8352ed29b0b4SJens Axboe /* 8353ed29b0b4SJens Axboe * We async punt it if the file wasn't marked NOWAIT, or if the file 8354ed29b0b4SJens Axboe * doesn't support non-blocking read/write attempts 8355ed29b0b4SJens Axboe */ 8356ed29b0b4SJens Axboe if (likely(!ret)) 8357ed29b0b4SJens Axboe io_arm_ltimeout(req); 8358ed29b0b4SJens Axboe else 8359ed29b0b4SJens Axboe io_queue_async(req, ret); 8360ed29b0b4SJens Axboe } 8361ed29b0b4SJens Axboe 8362ed29b0b4SJens Axboe static void io_queue_sqe_fallback(struct io_kiocb *req) 8363ed29b0b4SJens Axboe __must_hold(&req->ctx->uring_lock) 8364ed29b0b4SJens Axboe { 8365ed29b0b4SJens Axboe if (unlikely(req->flags & REQ_F_FAIL)) { 8366ed29b0b4SJens Axboe /* 8367ed29b0b4SJens Axboe * We don't submit, fail them all, for that replace hardlinks 8368ed29b0b4SJens Axboe * with normal links. Extra REQ_F_LINK is tolerated. 8369ed29b0b4SJens Axboe */ 8370ed29b0b4SJens Axboe req->flags &= ~REQ_F_HARDLINK; 8371ed29b0b4SJens Axboe req->flags |= REQ_F_LINK; 8372ed29b0b4SJens Axboe io_req_complete_failed(req, req->cqe.res); 8373ed29b0b4SJens Axboe } else if (unlikely(req->ctx->drain_active)) { 8374ed29b0b4SJens Axboe io_drain_req(req); 8375ed29b0b4SJens Axboe } else { 8376ed29b0b4SJens Axboe int ret = io_req_prep_async(req); 8377ed29b0b4SJens Axboe 8378ed29b0b4SJens Axboe if (unlikely(ret)) 8379ed29b0b4SJens Axboe io_req_complete_failed(req, ret); 8380ed29b0b4SJens Axboe else 8381ed29b0b4SJens Axboe io_queue_iowq(req, NULL); 8382ed29b0b4SJens Axboe } 8383ed29b0b4SJens Axboe } 8384ed29b0b4SJens Axboe 8385ed29b0b4SJens Axboe /* 8386ed29b0b4SJens Axboe * Check SQE restrictions (opcode and flags). 8387ed29b0b4SJens Axboe * 8388ed29b0b4SJens Axboe * Returns 'true' if SQE is allowed, 'false' otherwise. 8389ed29b0b4SJens Axboe */ 8390ed29b0b4SJens Axboe static inline bool io_check_restriction(struct io_ring_ctx *ctx, 8391ed29b0b4SJens Axboe struct io_kiocb *req, 8392ed29b0b4SJens Axboe unsigned int sqe_flags) 8393ed29b0b4SJens Axboe { 8394ed29b0b4SJens Axboe if (!test_bit(req->opcode, ctx->restrictions.sqe_op)) 8395ed29b0b4SJens Axboe return false; 8396ed29b0b4SJens Axboe 8397ed29b0b4SJens Axboe if ((sqe_flags & ctx->restrictions.sqe_flags_required) != 8398ed29b0b4SJens Axboe ctx->restrictions.sqe_flags_required) 8399ed29b0b4SJens Axboe return false; 8400ed29b0b4SJens Axboe 8401ed29b0b4SJens Axboe if (sqe_flags & ~(ctx->restrictions.sqe_flags_allowed | 8402ed29b0b4SJens Axboe ctx->restrictions.sqe_flags_required)) 8403ed29b0b4SJens Axboe return false; 8404ed29b0b4SJens Axboe 8405ed29b0b4SJens Axboe return true; 8406ed29b0b4SJens Axboe } 8407ed29b0b4SJens Axboe 8408ed29b0b4SJens Axboe static void io_init_req_drain(struct io_kiocb *req) 8409ed29b0b4SJens Axboe { 8410ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 8411ed29b0b4SJens Axboe struct io_kiocb *head = ctx->submit_state.link.head; 8412ed29b0b4SJens Axboe 8413ed29b0b4SJens Axboe ctx->drain_active = true; 8414ed29b0b4SJens Axboe if (head) { 8415ed29b0b4SJens Axboe /* 8416ed29b0b4SJens Axboe * If we need to drain a request in the middle of a link, drain 8417ed29b0b4SJens Axboe * the head request and the next request/link after the current 8418ed29b0b4SJens Axboe * link. Considering sequential execution of links, 8419ed29b0b4SJens Axboe * REQ_F_IO_DRAIN will be maintained for every request of our 8420ed29b0b4SJens Axboe * link. 8421ed29b0b4SJens Axboe */ 8422ed29b0b4SJens Axboe head->flags |= REQ_F_IO_DRAIN | REQ_F_FORCE_ASYNC; 8423ed29b0b4SJens Axboe ctx->drain_next = true; 8424ed29b0b4SJens Axboe } 8425ed29b0b4SJens Axboe } 8426ed29b0b4SJens Axboe 8427ed29b0b4SJens Axboe static int io_init_req(struct io_ring_ctx *ctx, struct io_kiocb *req, 8428ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 8429ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 8430ed29b0b4SJens Axboe { 8431ed29b0b4SJens Axboe const struct io_op_def *def; 8432ed29b0b4SJens Axboe unsigned int sqe_flags; 8433ed29b0b4SJens Axboe int personality; 8434ed29b0b4SJens Axboe u8 opcode; 8435ed29b0b4SJens Axboe 8436ed29b0b4SJens Axboe /* req is partially pre-initialised, see io_preinit_req() */ 8437ed29b0b4SJens Axboe req->opcode = opcode = READ_ONCE(sqe->opcode); 8438ed29b0b4SJens Axboe /* same numerical values with corresponding REQ_F_*, safe to copy */ 8439ed29b0b4SJens Axboe req->flags = sqe_flags = READ_ONCE(sqe->flags); 8440ed29b0b4SJens Axboe req->cqe.user_data = READ_ONCE(sqe->user_data); 8441ed29b0b4SJens Axboe req->file = NULL; 8442ed29b0b4SJens Axboe req->rsrc_node = NULL; 8443ed29b0b4SJens Axboe req->task = current; 8444ed29b0b4SJens Axboe 8445ed29b0b4SJens Axboe if (unlikely(opcode >= IORING_OP_LAST)) { 8446ed29b0b4SJens Axboe req->opcode = 0; 8447ed29b0b4SJens Axboe return -EINVAL; 8448ed29b0b4SJens Axboe } 8449ed29b0b4SJens Axboe def = &io_op_defs[opcode]; 8450ed29b0b4SJens Axboe if (unlikely(sqe_flags & ~SQE_COMMON_FLAGS)) { 8451ed29b0b4SJens Axboe /* enforce forwards compatibility on users */ 8452ed29b0b4SJens Axboe if (sqe_flags & ~SQE_VALID_FLAGS) 8453ed29b0b4SJens Axboe return -EINVAL; 8454ed29b0b4SJens Axboe if (sqe_flags & IOSQE_BUFFER_SELECT) { 8455ed29b0b4SJens Axboe if (!def->buffer_select) 8456ed29b0b4SJens Axboe return -EOPNOTSUPP; 8457ed29b0b4SJens Axboe req->buf_index = READ_ONCE(sqe->buf_group); 8458ed29b0b4SJens Axboe } 8459ed29b0b4SJens Axboe if (sqe_flags & IOSQE_CQE_SKIP_SUCCESS) 8460ed29b0b4SJens Axboe ctx->drain_disabled = true; 8461ed29b0b4SJens Axboe if (sqe_flags & IOSQE_IO_DRAIN) { 8462ed29b0b4SJens Axboe if (ctx->drain_disabled) 8463ed29b0b4SJens Axboe return -EOPNOTSUPP; 8464ed29b0b4SJens Axboe io_init_req_drain(req); 8465ed29b0b4SJens Axboe } 8466ed29b0b4SJens Axboe } 8467ed29b0b4SJens Axboe if (unlikely(ctx->restricted || ctx->drain_active || ctx->drain_next)) { 8468ed29b0b4SJens Axboe if (ctx->restricted && !io_check_restriction(ctx, req, sqe_flags)) 8469ed29b0b4SJens Axboe return -EACCES; 8470ed29b0b4SJens Axboe /* knock it to the slow queue path, will be drained there */ 8471ed29b0b4SJens Axboe if (ctx->drain_active) 8472ed29b0b4SJens Axboe req->flags |= REQ_F_FORCE_ASYNC; 8473ed29b0b4SJens Axboe /* if there is no link, we're at "next" request and need to drain */ 8474ed29b0b4SJens Axboe if (unlikely(ctx->drain_next) && !ctx->submit_state.link.head) { 8475ed29b0b4SJens Axboe ctx->drain_next = false; 8476ed29b0b4SJens Axboe ctx->drain_active = true; 8477ed29b0b4SJens Axboe req->flags |= REQ_F_IO_DRAIN | REQ_F_FORCE_ASYNC; 8478ed29b0b4SJens Axboe } 8479ed29b0b4SJens Axboe } 8480ed29b0b4SJens Axboe 8481ed29b0b4SJens Axboe if (!def->ioprio && sqe->ioprio) 8482ed29b0b4SJens Axboe return -EINVAL; 8483ed29b0b4SJens Axboe if (!def->iopoll && (ctx->flags & IORING_SETUP_IOPOLL)) 8484ed29b0b4SJens Axboe return -EINVAL; 8485ed29b0b4SJens Axboe 8486ed29b0b4SJens Axboe if (def->needs_file) { 8487ed29b0b4SJens Axboe struct io_submit_state *state = &ctx->submit_state; 8488ed29b0b4SJens Axboe 8489ed29b0b4SJens Axboe req->cqe.fd = READ_ONCE(sqe->fd); 8490ed29b0b4SJens Axboe 8491ed29b0b4SJens Axboe /* 8492ed29b0b4SJens Axboe * Plug now if we have more than 2 IO left after this, and the 8493ed29b0b4SJens Axboe * target is potentially a read/write to block based storage. 8494ed29b0b4SJens Axboe */ 8495ed29b0b4SJens Axboe if (state->need_plug && def->plug) { 8496ed29b0b4SJens Axboe state->plug_started = true; 8497ed29b0b4SJens Axboe state->need_plug = false; 8498ed29b0b4SJens Axboe blk_start_plug_nr_ios(&state->plug, state->submit_nr); 8499ed29b0b4SJens Axboe } 8500ed29b0b4SJens Axboe } 8501ed29b0b4SJens Axboe 8502ed29b0b4SJens Axboe personality = READ_ONCE(sqe->personality); 8503ed29b0b4SJens Axboe if (personality) { 8504ed29b0b4SJens Axboe int ret; 8505ed29b0b4SJens Axboe 8506ed29b0b4SJens Axboe req->creds = xa_load(&ctx->personalities, personality); 8507ed29b0b4SJens Axboe if (!req->creds) 8508ed29b0b4SJens Axboe return -EINVAL; 8509ed29b0b4SJens Axboe get_cred(req->creds); 8510ed29b0b4SJens Axboe ret = security_uring_override_creds(req->creds); 8511ed29b0b4SJens Axboe if (ret) { 8512ed29b0b4SJens Axboe put_cred(req->creds); 8513ed29b0b4SJens Axboe return ret; 8514ed29b0b4SJens Axboe } 8515ed29b0b4SJens Axboe req->flags |= REQ_F_CREDS; 8516ed29b0b4SJens Axboe } 8517ed29b0b4SJens Axboe 8518ed29b0b4SJens Axboe return def->prep(req, sqe); 8519ed29b0b4SJens Axboe } 8520ed29b0b4SJens Axboe 8521ed29b0b4SJens Axboe static __cold int io_submit_fail_init(const struct io_uring_sqe *sqe, 8522ed29b0b4SJens Axboe struct io_kiocb *req, int ret) 8523ed29b0b4SJens Axboe { 8524ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 8525ed29b0b4SJens Axboe struct io_submit_link *link = &ctx->submit_state.link; 8526ed29b0b4SJens Axboe struct io_kiocb *head = link->head; 8527ed29b0b4SJens Axboe 8528ed29b0b4SJens Axboe trace_io_uring_req_failed(sqe, ctx, req, ret); 8529ed29b0b4SJens Axboe 8530ed29b0b4SJens Axboe /* 8531ed29b0b4SJens Axboe * Avoid breaking links in the middle as it renders links with SQPOLL 8532ed29b0b4SJens Axboe * unusable. Instead of failing eagerly, continue assembling the link if 8533ed29b0b4SJens Axboe * applicable and mark the head with REQ_F_FAIL. The link flushing code 8534ed29b0b4SJens Axboe * should find the flag and handle the rest. 8535ed29b0b4SJens Axboe */ 8536ed29b0b4SJens Axboe req_fail_link_node(req, ret); 8537ed29b0b4SJens Axboe if (head && !(head->flags & REQ_F_FAIL)) 8538ed29b0b4SJens Axboe req_fail_link_node(head, -ECANCELED); 8539ed29b0b4SJens Axboe 8540ed29b0b4SJens Axboe if (!(req->flags & IO_REQ_LINK_FLAGS)) { 8541ed29b0b4SJens Axboe if (head) { 8542ed29b0b4SJens Axboe link->last->link = req; 8543ed29b0b4SJens Axboe link->head = NULL; 8544ed29b0b4SJens Axboe req = head; 8545ed29b0b4SJens Axboe } 8546ed29b0b4SJens Axboe io_queue_sqe_fallback(req); 8547ed29b0b4SJens Axboe return ret; 8548ed29b0b4SJens Axboe } 8549ed29b0b4SJens Axboe 8550ed29b0b4SJens Axboe if (head) 8551ed29b0b4SJens Axboe link->last->link = req; 8552ed29b0b4SJens Axboe else 8553ed29b0b4SJens Axboe link->head = req; 8554ed29b0b4SJens Axboe link->last = req; 8555ed29b0b4SJens Axboe return 0; 8556ed29b0b4SJens Axboe } 8557ed29b0b4SJens Axboe 8558ed29b0b4SJens Axboe static inline int io_submit_sqe(struct io_ring_ctx *ctx, struct io_kiocb *req, 8559ed29b0b4SJens Axboe const struct io_uring_sqe *sqe) 8560ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 8561ed29b0b4SJens Axboe { 8562ed29b0b4SJens Axboe struct io_submit_link *link = &ctx->submit_state.link; 8563ed29b0b4SJens Axboe int ret; 8564ed29b0b4SJens Axboe 8565ed29b0b4SJens Axboe ret = io_init_req(ctx, req, sqe); 8566ed29b0b4SJens Axboe if (unlikely(ret)) 8567ed29b0b4SJens Axboe return io_submit_fail_init(sqe, req, ret); 8568ed29b0b4SJens Axboe 8569ed29b0b4SJens Axboe /* don't need @sqe from now on */ 8570ed29b0b4SJens Axboe trace_io_uring_submit_sqe(ctx, req, req->cqe.user_data, req->opcode, 8571ed29b0b4SJens Axboe req->flags, true, 8572ed29b0b4SJens Axboe ctx->flags & IORING_SETUP_SQPOLL); 8573ed29b0b4SJens Axboe 8574ed29b0b4SJens Axboe /* 8575ed29b0b4SJens Axboe * If we already have a head request, queue this one for async 8576ed29b0b4SJens Axboe * submittal once the head completes. If we don't have a head but 8577ed29b0b4SJens Axboe * IOSQE_IO_LINK is set in the sqe, start a new head. This one will be 8578ed29b0b4SJens Axboe * submitted sync once the chain is complete. If none of those 8579ed29b0b4SJens Axboe * conditions are true (normal request), then just queue it. 8580ed29b0b4SJens Axboe */ 8581ed29b0b4SJens Axboe if (unlikely(link->head)) { 8582ed29b0b4SJens Axboe ret = io_req_prep_async(req); 8583ed29b0b4SJens Axboe if (unlikely(ret)) 8584ed29b0b4SJens Axboe return io_submit_fail_init(sqe, req, ret); 8585ed29b0b4SJens Axboe 8586ed29b0b4SJens Axboe trace_io_uring_link(ctx, req, link->head); 8587ed29b0b4SJens Axboe link->last->link = req; 8588ed29b0b4SJens Axboe link->last = req; 8589ed29b0b4SJens Axboe 8590ed29b0b4SJens Axboe if (req->flags & IO_REQ_LINK_FLAGS) 8591ed29b0b4SJens Axboe return 0; 8592ed29b0b4SJens Axboe /* last request of the link, flush it */ 8593ed29b0b4SJens Axboe req = link->head; 8594ed29b0b4SJens Axboe link->head = NULL; 8595ed29b0b4SJens Axboe if (req->flags & (REQ_F_FORCE_ASYNC | REQ_F_FAIL)) 8596ed29b0b4SJens Axboe goto fallback; 8597ed29b0b4SJens Axboe 8598ed29b0b4SJens Axboe } else if (unlikely(req->flags & (IO_REQ_LINK_FLAGS | 8599ed29b0b4SJens Axboe REQ_F_FORCE_ASYNC | REQ_F_FAIL))) { 8600ed29b0b4SJens Axboe if (req->flags & IO_REQ_LINK_FLAGS) { 8601ed29b0b4SJens Axboe link->head = req; 8602ed29b0b4SJens Axboe link->last = req; 8603ed29b0b4SJens Axboe } else { 8604ed29b0b4SJens Axboe fallback: 8605ed29b0b4SJens Axboe io_queue_sqe_fallback(req); 8606ed29b0b4SJens Axboe } 8607ed29b0b4SJens Axboe return 0; 8608ed29b0b4SJens Axboe } 8609ed29b0b4SJens Axboe 8610ed29b0b4SJens Axboe io_queue_sqe(req); 8611ed29b0b4SJens Axboe return 0; 8612ed29b0b4SJens Axboe } 8613ed29b0b4SJens Axboe 8614ed29b0b4SJens Axboe /* 8615ed29b0b4SJens Axboe * Batched submission is done, ensure local IO is flushed out. 8616ed29b0b4SJens Axboe */ 8617ed29b0b4SJens Axboe static void io_submit_state_end(struct io_ring_ctx *ctx) 8618ed29b0b4SJens Axboe { 8619ed29b0b4SJens Axboe struct io_submit_state *state = &ctx->submit_state; 8620ed29b0b4SJens Axboe 8621ed29b0b4SJens Axboe if (unlikely(state->link.head)) 8622ed29b0b4SJens Axboe io_queue_sqe_fallback(state->link.head); 8623ed29b0b4SJens Axboe /* flush only after queuing links as they can generate completions */ 8624ed29b0b4SJens Axboe io_submit_flush_completions(ctx); 8625ed29b0b4SJens Axboe if (state->plug_started) 8626ed29b0b4SJens Axboe blk_finish_plug(&state->plug); 8627ed29b0b4SJens Axboe } 8628ed29b0b4SJens Axboe 8629ed29b0b4SJens Axboe /* 8630ed29b0b4SJens Axboe * Start submission side cache. 8631ed29b0b4SJens Axboe */ 8632ed29b0b4SJens Axboe static void io_submit_state_start(struct io_submit_state *state, 8633ed29b0b4SJens Axboe unsigned int max_ios) 8634ed29b0b4SJens Axboe { 8635ed29b0b4SJens Axboe state->plug_started = false; 8636ed29b0b4SJens Axboe state->need_plug = max_ios > 2; 8637ed29b0b4SJens Axboe state->submit_nr = max_ios; 8638ed29b0b4SJens Axboe /* set only head, no need to init link_last in advance */ 8639ed29b0b4SJens Axboe state->link.head = NULL; 8640ed29b0b4SJens Axboe } 8641ed29b0b4SJens Axboe 8642ed29b0b4SJens Axboe static void io_commit_sqring(struct io_ring_ctx *ctx) 8643ed29b0b4SJens Axboe { 8644ed29b0b4SJens Axboe struct io_rings *rings = ctx->rings; 8645ed29b0b4SJens Axboe 8646ed29b0b4SJens Axboe /* 8647ed29b0b4SJens Axboe * Ensure any loads from the SQEs are done at this point, 8648ed29b0b4SJens Axboe * since once we write the new head, the application could 8649ed29b0b4SJens Axboe * write new data to them. 8650ed29b0b4SJens Axboe */ 8651ed29b0b4SJens Axboe smp_store_release(&rings->sq.head, ctx->cached_sq_head); 8652ed29b0b4SJens Axboe } 8653ed29b0b4SJens Axboe 8654ed29b0b4SJens Axboe /* 8655ed29b0b4SJens Axboe * Fetch an sqe, if one is available. Note this returns a pointer to memory 8656ed29b0b4SJens Axboe * that is mapped by userspace. This means that care needs to be taken to 8657ed29b0b4SJens Axboe * ensure that reads are stable, as we cannot rely on userspace always 8658ed29b0b4SJens Axboe * being a good citizen. If members of the sqe are validated and then later 8659ed29b0b4SJens Axboe * used, it's important that those reads are done through READ_ONCE() to 8660ed29b0b4SJens Axboe * prevent a re-load down the line. 8661ed29b0b4SJens Axboe */ 8662ed29b0b4SJens Axboe static const struct io_uring_sqe *io_get_sqe(struct io_ring_ctx *ctx) 8663ed29b0b4SJens Axboe { 8664ed29b0b4SJens Axboe unsigned head, mask = ctx->sq_entries - 1; 8665ed29b0b4SJens Axboe unsigned sq_idx = ctx->cached_sq_head++ & mask; 8666ed29b0b4SJens Axboe 8667ed29b0b4SJens Axboe /* 8668ed29b0b4SJens Axboe * The cached sq head (or cq tail) serves two purposes: 8669ed29b0b4SJens Axboe * 8670ed29b0b4SJens Axboe * 1) allows us to batch the cost of updating the user visible 8671ed29b0b4SJens Axboe * head updates. 8672ed29b0b4SJens Axboe * 2) allows the kernel side to track the head on its own, even 8673ed29b0b4SJens Axboe * though the application is the one updating it. 8674ed29b0b4SJens Axboe */ 8675ed29b0b4SJens Axboe head = READ_ONCE(ctx->sq_array[sq_idx]); 8676ed29b0b4SJens Axboe if (likely(head < ctx->sq_entries)) { 8677ed29b0b4SJens Axboe /* double index for 128-byte SQEs, twice as long */ 8678ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQE128) 8679ed29b0b4SJens Axboe head <<= 1; 8680ed29b0b4SJens Axboe return &ctx->sq_sqes[head]; 8681ed29b0b4SJens Axboe } 8682ed29b0b4SJens Axboe 8683ed29b0b4SJens Axboe /* drop invalid entries */ 8684ed29b0b4SJens Axboe ctx->cq_extra--; 8685ed29b0b4SJens Axboe WRITE_ONCE(ctx->rings->sq_dropped, 8686ed29b0b4SJens Axboe READ_ONCE(ctx->rings->sq_dropped) + 1); 8687ed29b0b4SJens Axboe return NULL; 8688ed29b0b4SJens Axboe } 8689ed29b0b4SJens Axboe 8690ed29b0b4SJens Axboe static int io_submit_sqes(struct io_ring_ctx *ctx, unsigned int nr) 8691ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 8692ed29b0b4SJens Axboe { 8693ed29b0b4SJens Axboe unsigned int entries = io_sqring_entries(ctx); 8694ed29b0b4SJens Axboe unsigned int left; 8695ed29b0b4SJens Axboe int ret; 8696ed29b0b4SJens Axboe 8697ed29b0b4SJens Axboe if (unlikely(!entries)) 8698ed29b0b4SJens Axboe return 0; 8699ed29b0b4SJens Axboe /* make sure SQ entry isn't read before tail */ 8700ed29b0b4SJens Axboe ret = left = min3(nr, ctx->sq_entries, entries); 8701ed29b0b4SJens Axboe io_get_task_refs(left); 8702ed29b0b4SJens Axboe io_submit_state_start(&ctx->submit_state, left); 8703ed29b0b4SJens Axboe 8704ed29b0b4SJens Axboe do { 8705ed29b0b4SJens Axboe const struct io_uring_sqe *sqe; 8706ed29b0b4SJens Axboe struct io_kiocb *req; 8707ed29b0b4SJens Axboe 8708ed29b0b4SJens Axboe if (unlikely(!io_alloc_req_refill(ctx))) 8709ed29b0b4SJens Axboe break; 8710ed29b0b4SJens Axboe req = io_alloc_req(ctx); 8711ed29b0b4SJens Axboe sqe = io_get_sqe(ctx); 8712ed29b0b4SJens Axboe if (unlikely(!sqe)) { 8713ed29b0b4SJens Axboe io_req_add_to_cache(req, ctx); 8714ed29b0b4SJens Axboe break; 8715ed29b0b4SJens Axboe } 8716ed29b0b4SJens Axboe 8717ed29b0b4SJens Axboe /* 8718ed29b0b4SJens Axboe * Continue submitting even for sqe failure if the 8719ed29b0b4SJens Axboe * ring was setup with IORING_SETUP_SUBMIT_ALL 8720ed29b0b4SJens Axboe */ 8721ed29b0b4SJens Axboe if (unlikely(io_submit_sqe(ctx, req, sqe)) && 8722ed29b0b4SJens Axboe !(ctx->flags & IORING_SETUP_SUBMIT_ALL)) { 8723ed29b0b4SJens Axboe left--; 8724ed29b0b4SJens Axboe break; 8725ed29b0b4SJens Axboe } 8726ed29b0b4SJens Axboe } while (--left); 8727ed29b0b4SJens Axboe 8728ed29b0b4SJens Axboe if (unlikely(left)) { 8729ed29b0b4SJens Axboe ret -= left; 8730ed29b0b4SJens Axboe /* try again if it submitted nothing and can't allocate a req */ 8731ed29b0b4SJens Axboe if (!ret && io_req_cache_empty(ctx)) 8732ed29b0b4SJens Axboe ret = -EAGAIN; 8733ed29b0b4SJens Axboe current->io_uring->cached_refs += left; 8734ed29b0b4SJens Axboe } 8735ed29b0b4SJens Axboe 8736ed29b0b4SJens Axboe io_submit_state_end(ctx); 8737ed29b0b4SJens Axboe /* Commit SQ ring head once we've consumed and submitted all SQEs */ 8738ed29b0b4SJens Axboe io_commit_sqring(ctx); 8739ed29b0b4SJens Axboe return ret; 8740ed29b0b4SJens Axboe } 8741ed29b0b4SJens Axboe 8742ed29b0b4SJens Axboe static inline bool io_sqd_events_pending(struct io_sq_data *sqd) 8743ed29b0b4SJens Axboe { 8744ed29b0b4SJens Axboe return READ_ONCE(sqd->state); 8745ed29b0b4SJens Axboe } 8746ed29b0b4SJens Axboe 8747ed29b0b4SJens Axboe static int __io_sq_thread(struct io_ring_ctx *ctx, bool cap_entries) 8748ed29b0b4SJens Axboe { 8749ed29b0b4SJens Axboe unsigned int to_submit; 8750ed29b0b4SJens Axboe int ret = 0; 8751ed29b0b4SJens Axboe 8752ed29b0b4SJens Axboe to_submit = io_sqring_entries(ctx); 8753ed29b0b4SJens Axboe /* if we're handling multiple rings, cap submit size for fairness */ 8754ed29b0b4SJens Axboe if (cap_entries && to_submit > IORING_SQPOLL_CAP_ENTRIES_VALUE) 8755ed29b0b4SJens Axboe to_submit = IORING_SQPOLL_CAP_ENTRIES_VALUE; 8756ed29b0b4SJens Axboe 8757ed29b0b4SJens Axboe if (!wq_list_empty(&ctx->iopoll_list) || to_submit) { 8758ed29b0b4SJens Axboe const struct cred *creds = NULL; 8759ed29b0b4SJens Axboe 8760ed29b0b4SJens Axboe if (ctx->sq_creds != current_cred()) 8761ed29b0b4SJens Axboe creds = override_creds(ctx->sq_creds); 8762ed29b0b4SJens Axboe 8763ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 8764ed29b0b4SJens Axboe if (!wq_list_empty(&ctx->iopoll_list)) 8765ed29b0b4SJens Axboe io_do_iopoll(ctx, true); 8766ed29b0b4SJens Axboe 8767ed29b0b4SJens Axboe /* 8768ed29b0b4SJens Axboe * Don't submit if refs are dying, good for io_uring_register(), 8769ed29b0b4SJens Axboe * but also it is relied upon by io_ring_exit_work() 8770ed29b0b4SJens Axboe */ 8771ed29b0b4SJens Axboe if (to_submit && likely(!percpu_ref_is_dying(&ctx->refs)) && 8772ed29b0b4SJens Axboe !(ctx->flags & IORING_SETUP_R_DISABLED)) 8773ed29b0b4SJens Axboe ret = io_submit_sqes(ctx, to_submit); 8774ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 8775ed29b0b4SJens Axboe 8776ed29b0b4SJens Axboe if (to_submit && wq_has_sleeper(&ctx->sqo_sq_wait)) 8777ed29b0b4SJens Axboe wake_up(&ctx->sqo_sq_wait); 8778ed29b0b4SJens Axboe if (creds) 8779ed29b0b4SJens Axboe revert_creds(creds); 8780ed29b0b4SJens Axboe } 8781ed29b0b4SJens Axboe 8782ed29b0b4SJens Axboe return ret; 8783ed29b0b4SJens Axboe } 8784ed29b0b4SJens Axboe 8785ed29b0b4SJens Axboe static __cold void io_sqd_update_thread_idle(struct io_sq_data *sqd) 8786ed29b0b4SJens Axboe { 8787ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 8788ed29b0b4SJens Axboe unsigned sq_thread_idle = 0; 8789ed29b0b4SJens Axboe 8790ed29b0b4SJens Axboe list_for_each_entry(ctx, &sqd->ctx_list, sqd_list) 8791ed29b0b4SJens Axboe sq_thread_idle = max(sq_thread_idle, ctx->sq_thread_idle); 8792ed29b0b4SJens Axboe sqd->sq_thread_idle = sq_thread_idle; 8793ed29b0b4SJens Axboe } 8794ed29b0b4SJens Axboe 8795ed29b0b4SJens Axboe static bool io_sqd_handle_event(struct io_sq_data *sqd) 8796ed29b0b4SJens Axboe { 8797ed29b0b4SJens Axboe bool did_sig = false; 8798ed29b0b4SJens Axboe struct ksignal ksig; 8799ed29b0b4SJens Axboe 8800ed29b0b4SJens Axboe if (test_bit(IO_SQ_THREAD_SHOULD_PARK, &sqd->state) || 8801ed29b0b4SJens Axboe signal_pending(current)) { 8802ed29b0b4SJens Axboe mutex_unlock(&sqd->lock); 8803ed29b0b4SJens Axboe if (signal_pending(current)) 8804ed29b0b4SJens Axboe did_sig = get_signal(&ksig); 8805ed29b0b4SJens Axboe cond_resched(); 8806ed29b0b4SJens Axboe mutex_lock(&sqd->lock); 8807ed29b0b4SJens Axboe } 8808ed29b0b4SJens Axboe return did_sig || test_bit(IO_SQ_THREAD_SHOULD_STOP, &sqd->state); 8809ed29b0b4SJens Axboe } 8810ed29b0b4SJens Axboe 8811ed29b0b4SJens Axboe static int io_sq_thread(void *data) 8812ed29b0b4SJens Axboe { 8813ed29b0b4SJens Axboe struct io_sq_data *sqd = data; 8814ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 8815ed29b0b4SJens Axboe unsigned long timeout = 0; 8816ed29b0b4SJens Axboe char buf[TASK_COMM_LEN]; 8817ed29b0b4SJens Axboe DEFINE_WAIT(wait); 8818ed29b0b4SJens Axboe 8819ed29b0b4SJens Axboe snprintf(buf, sizeof(buf), "iou-sqp-%d", sqd->task_pid); 8820ed29b0b4SJens Axboe set_task_comm(current, buf); 8821ed29b0b4SJens Axboe 8822ed29b0b4SJens Axboe if (sqd->sq_cpu != -1) 8823ed29b0b4SJens Axboe set_cpus_allowed_ptr(current, cpumask_of(sqd->sq_cpu)); 8824ed29b0b4SJens Axboe else 8825ed29b0b4SJens Axboe set_cpus_allowed_ptr(current, cpu_online_mask); 8826ed29b0b4SJens Axboe current->flags |= PF_NO_SETAFFINITY; 8827ed29b0b4SJens Axboe 8828ed29b0b4SJens Axboe audit_alloc_kernel(current); 8829ed29b0b4SJens Axboe 8830ed29b0b4SJens Axboe mutex_lock(&sqd->lock); 8831ed29b0b4SJens Axboe while (1) { 8832ed29b0b4SJens Axboe bool cap_entries, sqt_spin = false; 8833ed29b0b4SJens Axboe 8834ed29b0b4SJens Axboe if (io_sqd_events_pending(sqd) || signal_pending(current)) { 8835ed29b0b4SJens Axboe if (io_sqd_handle_event(sqd)) 8836ed29b0b4SJens Axboe break; 8837ed29b0b4SJens Axboe timeout = jiffies + sqd->sq_thread_idle; 8838ed29b0b4SJens Axboe } 8839ed29b0b4SJens Axboe 8840ed29b0b4SJens Axboe cap_entries = !list_is_singular(&sqd->ctx_list); 8841ed29b0b4SJens Axboe list_for_each_entry(ctx, &sqd->ctx_list, sqd_list) { 8842ed29b0b4SJens Axboe int ret = __io_sq_thread(ctx, cap_entries); 8843ed29b0b4SJens Axboe 8844ed29b0b4SJens Axboe if (!sqt_spin && (ret > 0 || !wq_list_empty(&ctx->iopoll_list))) 8845ed29b0b4SJens Axboe sqt_spin = true; 8846ed29b0b4SJens Axboe } 8847ed29b0b4SJens Axboe if (io_run_task_work()) 8848ed29b0b4SJens Axboe sqt_spin = true; 8849ed29b0b4SJens Axboe 8850ed29b0b4SJens Axboe if (sqt_spin || !time_after(jiffies, timeout)) { 8851ed29b0b4SJens Axboe cond_resched(); 8852ed29b0b4SJens Axboe if (sqt_spin) 8853ed29b0b4SJens Axboe timeout = jiffies + sqd->sq_thread_idle; 8854ed29b0b4SJens Axboe continue; 8855ed29b0b4SJens Axboe } 8856ed29b0b4SJens Axboe 8857ed29b0b4SJens Axboe prepare_to_wait(&sqd->wait, &wait, TASK_INTERRUPTIBLE); 8858ed29b0b4SJens Axboe if (!io_sqd_events_pending(sqd) && !task_work_pending(current)) { 8859ed29b0b4SJens Axboe bool needs_sched = true; 8860ed29b0b4SJens Axboe 8861ed29b0b4SJens Axboe list_for_each_entry(ctx, &sqd->ctx_list, sqd_list) { 8862ed29b0b4SJens Axboe atomic_or(IORING_SQ_NEED_WAKEUP, 8863ed29b0b4SJens Axboe &ctx->rings->sq_flags); 8864ed29b0b4SJens Axboe if ((ctx->flags & IORING_SETUP_IOPOLL) && 8865ed29b0b4SJens Axboe !wq_list_empty(&ctx->iopoll_list)) { 8866ed29b0b4SJens Axboe needs_sched = false; 8867ed29b0b4SJens Axboe break; 8868ed29b0b4SJens Axboe } 8869ed29b0b4SJens Axboe 8870ed29b0b4SJens Axboe /* 8871ed29b0b4SJens Axboe * Ensure the store of the wakeup flag is not 8872ed29b0b4SJens Axboe * reordered with the load of the SQ tail 8873ed29b0b4SJens Axboe */ 8874ed29b0b4SJens Axboe smp_mb__after_atomic(); 8875ed29b0b4SJens Axboe 8876ed29b0b4SJens Axboe if (io_sqring_entries(ctx)) { 8877ed29b0b4SJens Axboe needs_sched = false; 8878ed29b0b4SJens Axboe break; 8879ed29b0b4SJens Axboe } 8880ed29b0b4SJens Axboe } 8881ed29b0b4SJens Axboe 8882ed29b0b4SJens Axboe if (needs_sched) { 8883ed29b0b4SJens Axboe mutex_unlock(&sqd->lock); 8884ed29b0b4SJens Axboe schedule(); 8885ed29b0b4SJens Axboe mutex_lock(&sqd->lock); 8886ed29b0b4SJens Axboe } 8887ed29b0b4SJens Axboe list_for_each_entry(ctx, &sqd->ctx_list, sqd_list) 8888ed29b0b4SJens Axboe atomic_andnot(IORING_SQ_NEED_WAKEUP, 8889ed29b0b4SJens Axboe &ctx->rings->sq_flags); 8890ed29b0b4SJens Axboe } 8891ed29b0b4SJens Axboe 8892ed29b0b4SJens Axboe finish_wait(&sqd->wait, &wait); 8893ed29b0b4SJens Axboe timeout = jiffies + sqd->sq_thread_idle; 8894ed29b0b4SJens Axboe } 8895ed29b0b4SJens Axboe 8896ed29b0b4SJens Axboe io_uring_cancel_generic(true, sqd); 8897ed29b0b4SJens Axboe sqd->thread = NULL; 8898ed29b0b4SJens Axboe list_for_each_entry(ctx, &sqd->ctx_list, sqd_list) 8899ed29b0b4SJens Axboe atomic_or(IORING_SQ_NEED_WAKEUP, &ctx->rings->sq_flags); 8900ed29b0b4SJens Axboe io_run_task_work(); 8901ed29b0b4SJens Axboe mutex_unlock(&sqd->lock); 8902ed29b0b4SJens Axboe 8903ed29b0b4SJens Axboe audit_free(current); 8904ed29b0b4SJens Axboe 8905ed29b0b4SJens Axboe complete(&sqd->exited); 8906ed29b0b4SJens Axboe do_exit(0); 8907ed29b0b4SJens Axboe } 8908ed29b0b4SJens Axboe 8909ed29b0b4SJens Axboe struct io_wait_queue { 8910ed29b0b4SJens Axboe struct wait_queue_entry wq; 8911ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 8912ed29b0b4SJens Axboe unsigned cq_tail; 8913ed29b0b4SJens Axboe unsigned nr_timeouts; 8914ed29b0b4SJens Axboe }; 8915ed29b0b4SJens Axboe 8916ed29b0b4SJens Axboe static inline bool io_should_wake(struct io_wait_queue *iowq) 8917ed29b0b4SJens Axboe { 8918ed29b0b4SJens Axboe struct io_ring_ctx *ctx = iowq->ctx; 8919ed29b0b4SJens Axboe int dist = ctx->cached_cq_tail - (int) iowq->cq_tail; 8920ed29b0b4SJens Axboe 8921ed29b0b4SJens Axboe /* 8922ed29b0b4SJens Axboe * Wake up if we have enough events, or if a timeout occurred since we 8923ed29b0b4SJens Axboe * started waiting. For timeouts, we always want to return to userspace, 8924ed29b0b4SJens Axboe * regardless of event count. 8925ed29b0b4SJens Axboe */ 8926ed29b0b4SJens Axboe return dist >= 0 || atomic_read(&ctx->cq_timeouts) != iowq->nr_timeouts; 8927ed29b0b4SJens Axboe } 8928ed29b0b4SJens Axboe 8929ed29b0b4SJens Axboe static int io_wake_function(struct wait_queue_entry *curr, unsigned int mode, 8930ed29b0b4SJens Axboe int wake_flags, void *key) 8931ed29b0b4SJens Axboe { 8932ed29b0b4SJens Axboe struct io_wait_queue *iowq = container_of(curr, struct io_wait_queue, 8933ed29b0b4SJens Axboe wq); 8934ed29b0b4SJens Axboe 8935ed29b0b4SJens Axboe /* 8936ed29b0b4SJens Axboe * Cannot safely flush overflowed CQEs from here, ensure we wake up 8937ed29b0b4SJens Axboe * the task, and the next invocation will do it. 8938ed29b0b4SJens Axboe */ 8939ed29b0b4SJens Axboe if (io_should_wake(iowq) || 8940ed29b0b4SJens Axboe test_bit(IO_CHECK_CQ_OVERFLOW_BIT, &iowq->ctx->check_cq)) 8941ed29b0b4SJens Axboe return autoremove_wake_function(curr, mode, wake_flags, key); 8942ed29b0b4SJens Axboe return -1; 8943ed29b0b4SJens Axboe } 8944ed29b0b4SJens Axboe 8945ed29b0b4SJens Axboe static int io_run_task_work_sig(void) 8946ed29b0b4SJens Axboe { 8947ed29b0b4SJens Axboe if (io_run_task_work()) 8948ed29b0b4SJens Axboe return 1; 8949ed29b0b4SJens Axboe if (test_thread_flag(TIF_NOTIFY_SIGNAL)) 8950ed29b0b4SJens Axboe return -ERESTARTSYS; 8951ed29b0b4SJens Axboe if (task_sigpending(current)) 8952ed29b0b4SJens Axboe return -EINTR; 8953ed29b0b4SJens Axboe return 0; 8954ed29b0b4SJens Axboe } 8955ed29b0b4SJens Axboe 8956ed29b0b4SJens Axboe /* when returns >0, the caller should retry */ 8957ed29b0b4SJens Axboe static inline int io_cqring_wait_schedule(struct io_ring_ctx *ctx, 8958ed29b0b4SJens Axboe struct io_wait_queue *iowq, 8959ed29b0b4SJens Axboe ktime_t timeout) 8960ed29b0b4SJens Axboe { 8961ed29b0b4SJens Axboe int ret; 8962ed29b0b4SJens Axboe unsigned long check_cq; 8963ed29b0b4SJens Axboe 8964ed29b0b4SJens Axboe /* make sure we run task_work before checking for signals */ 8965ed29b0b4SJens Axboe ret = io_run_task_work_sig(); 8966ed29b0b4SJens Axboe if (ret || io_should_wake(iowq)) 8967ed29b0b4SJens Axboe return ret; 8968ed29b0b4SJens Axboe check_cq = READ_ONCE(ctx->check_cq); 8969ed29b0b4SJens Axboe /* let the caller flush overflows, retry */ 8970ed29b0b4SJens Axboe if (check_cq & BIT(IO_CHECK_CQ_OVERFLOW_BIT)) 8971ed29b0b4SJens Axboe return 1; 8972ed29b0b4SJens Axboe if (unlikely(check_cq & BIT(IO_CHECK_CQ_DROPPED_BIT))) 8973ed29b0b4SJens Axboe return -EBADR; 8974ed29b0b4SJens Axboe if (!schedule_hrtimeout(&timeout, HRTIMER_MODE_ABS)) 8975ed29b0b4SJens Axboe return -ETIME; 8976ed29b0b4SJens Axboe return 1; 8977ed29b0b4SJens Axboe } 8978ed29b0b4SJens Axboe 8979ed29b0b4SJens Axboe /* 8980ed29b0b4SJens Axboe * Wait until events become available, if we don't already have some. The 8981ed29b0b4SJens Axboe * application must reap them itself, as they reside on the shared cq ring. 8982ed29b0b4SJens Axboe */ 8983ed29b0b4SJens Axboe static int io_cqring_wait(struct io_ring_ctx *ctx, int min_events, 8984ed29b0b4SJens Axboe const sigset_t __user *sig, size_t sigsz, 8985ed29b0b4SJens Axboe struct __kernel_timespec __user *uts) 8986ed29b0b4SJens Axboe { 8987ed29b0b4SJens Axboe struct io_wait_queue iowq; 8988ed29b0b4SJens Axboe struct io_rings *rings = ctx->rings; 8989ed29b0b4SJens Axboe ktime_t timeout = KTIME_MAX; 8990ed29b0b4SJens Axboe int ret; 8991ed29b0b4SJens Axboe 8992ed29b0b4SJens Axboe do { 8993ed29b0b4SJens Axboe io_cqring_overflow_flush(ctx); 8994ed29b0b4SJens Axboe if (io_cqring_events(ctx) >= min_events) 8995ed29b0b4SJens Axboe return 0; 8996ed29b0b4SJens Axboe if (!io_run_task_work()) 8997ed29b0b4SJens Axboe break; 8998ed29b0b4SJens Axboe } while (1); 8999ed29b0b4SJens Axboe 9000ed29b0b4SJens Axboe if (sig) { 9001ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 9002ed29b0b4SJens Axboe if (in_compat_syscall()) 9003ed29b0b4SJens Axboe ret = set_compat_user_sigmask((const compat_sigset_t __user *)sig, 9004ed29b0b4SJens Axboe sigsz); 9005ed29b0b4SJens Axboe else 9006ed29b0b4SJens Axboe #endif 9007ed29b0b4SJens Axboe ret = set_user_sigmask(sig, sigsz); 9008ed29b0b4SJens Axboe 9009ed29b0b4SJens Axboe if (ret) 9010ed29b0b4SJens Axboe return ret; 9011ed29b0b4SJens Axboe } 9012ed29b0b4SJens Axboe 9013ed29b0b4SJens Axboe if (uts) { 9014ed29b0b4SJens Axboe struct timespec64 ts; 9015ed29b0b4SJens Axboe 9016ed29b0b4SJens Axboe if (get_timespec64(&ts, uts)) 9017ed29b0b4SJens Axboe return -EFAULT; 9018ed29b0b4SJens Axboe timeout = ktime_add_ns(timespec64_to_ktime(ts), ktime_get_ns()); 9019ed29b0b4SJens Axboe } 9020ed29b0b4SJens Axboe 9021ed29b0b4SJens Axboe init_waitqueue_func_entry(&iowq.wq, io_wake_function); 9022ed29b0b4SJens Axboe iowq.wq.private = current; 9023ed29b0b4SJens Axboe INIT_LIST_HEAD(&iowq.wq.entry); 9024ed29b0b4SJens Axboe iowq.ctx = ctx; 9025ed29b0b4SJens Axboe iowq.nr_timeouts = atomic_read(&ctx->cq_timeouts); 9026ed29b0b4SJens Axboe iowq.cq_tail = READ_ONCE(ctx->rings->cq.head) + min_events; 9027ed29b0b4SJens Axboe 9028ed29b0b4SJens Axboe trace_io_uring_cqring_wait(ctx, min_events); 9029ed29b0b4SJens Axboe do { 9030ed29b0b4SJens Axboe /* if we can't even flush overflow, don't wait for more */ 9031ed29b0b4SJens Axboe if (!io_cqring_overflow_flush(ctx)) { 9032ed29b0b4SJens Axboe ret = -EBUSY; 9033ed29b0b4SJens Axboe break; 9034ed29b0b4SJens Axboe } 9035ed29b0b4SJens Axboe prepare_to_wait_exclusive(&ctx->cq_wait, &iowq.wq, 9036ed29b0b4SJens Axboe TASK_INTERRUPTIBLE); 9037ed29b0b4SJens Axboe ret = io_cqring_wait_schedule(ctx, &iowq, timeout); 9038ed29b0b4SJens Axboe cond_resched(); 9039ed29b0b4SJens Axboe } while (ret > 0); 9040ed29b0b4SJens Axboe 9041ed29b0b4SJens Axboe finish_wait(&ctx->cq_wait, &iowq.wq); 9042ed29b0b4SJens Axboe restore_saved_sigmask_unless(ret == -EINTR); 9043ed29b0b4SJens Axboe 9044ed29b0b4SJens Axboe return READ_ONCE(rings->cq.head) == READ_ONCE(rings->cq.tail) ? ret : 0; 9045ed29b0b4SJens Axboe } 9046ed29b0b4SJens Axboe 9047ed29b0b4SJens Axboe static void io_free_page_table(void **table, size_t size) 9048ed29b0b4SJens Axboe { 9049ed29b0b4SJens Axboe unsigned i, nr_tables = DIV_ROUND_UP(size, PAGE_SIZE); 9050ed29b0b4SJens Axboe 9051ed29b0b4SJens Axboe for (i = 0; i < nr_tables; i++) 9052ed29b0b4SJens Axboe kfree(table[i]); 9053ed29b0b4SJens Axboe kfree(table); 9054ed29b0b4SJens Axboe } 9055ed29b0b4SJens Axboe 9056ed29b0b4SJens Axboe static __cold void **io_alloc_page_table(size_t size) 9057ed29b0b4SJens Axboe { 9058ed29b0b4SJens Axboe unsigned i, nr_tables = DIV_ROUND_UP(size, PAGE_SIZE); 9059ed29b0b4SJens Axboe size_t init_size = size; 9060ed29b0b4SJens Axboe void **table; 9061ed29b0b4SJens Axboe 9062ed29b0b4SJens Axboe table = kcalloc(nr_tables, sizeof(*table), GFP_KERNEL_ACCOUNT); 9063ed29b0b4SJens Axboe if (!table) 9064ed29b0b4SJens Axboe return NULL; 9065ed29b0b4SJens Axboe 9066ed29b0b4SJens Axboe for (i = 0; i < nr_tables; i++) { 9067ed29b0b4SJens Axboe unsigned int this_size = min_t(size_t, size, PAGE_SIZE); 9068ed29b0b4SJens Axboe 9069ed29b0b4SJens Axboe table[i] = kzalloc(this_size, GFP_KERNEL_ACCOUNT); 9070ed29b0b4SJens Axboe if (!table[i]) { 9071ed29b0b4SJens Axboe io_free_page_table(table, init_size); 9072ed29b0b4SJens Axboe return NULL; 9073ed29b0b4SJens Axboe } 9074ed29b0b4SJens Axboe size -= this_size; 9075ed29b0b4SJens Axboe } 9076ed29b0b4SJens Axboe return table; 9077ed29b0b4SJens Axboe } 9078ed29b0b4SJens Axboe 9079ed29b0b4SJens Axboe static void io_rsrc_node_destroy(struct io_rsrc_node *ref_node) 9080ed29b0b4SJens Axboe { 9081ed29b0b4SJens Axboe percpu_ref_exit(&ref_node->refs); 9082ed29b0b4SJens Axboe kfree(ref_node); 9083ed29b0b4SJens Axboe } 9084ed29b0b4SJens Axboe 9085ed29b0b4SJens Axboe static __cold void io_rsrc_node_ref_zero(struct percpu_ref *ref) 9086ed29b0b4SJens Axboe { 9087ed29b0b4SJens Axboe struct io_rsrc_node *node = container_of(ref, struct io_rsrc_node, refs); 9088ed29b0b4SJens Axboe struct io_ring_ctx *ctx = node->rsrc_data->ctx; 9089ed29b0b4SJens Axboe unsigned long flags; 9090ed29b0b4SJens Axboe bool first_add = false; 9091ed29b0b4SJens Axboe unsigned long delay = HZ; 9092ed29b0b4SJens Axboe 9093ed29b0b4SJens Axboe spin_lock_irqsave(&ctx->rsrc_ref_lock, flags); 9094ed29b0b4SJens Axboe node->done = true; 9095ed29b0b4SJens Axboe 9096ed29b0b4SJens Axboe /* if we are mid-quiesce then do not delay */ 9097ed29b0b4SJens Axboe if (node->rsrc_data->quiesce) 9098ed29b0b4SJens Axboe delay = 0; 9099ed29b0b4SJens Axboe 9100ed29b0b4SJens Axboe while (!list_empty(&ctx->rsrc_ref_list)) { 9101ed29b0b4SJens Axboe node = list_first_entry(&ctx->rsrc_ref_list, 9102ed29b0b4SJens Axboe struct io_rsrc_node, node); 9103ed29b0b4SJens Axboe /* recycle ref nodes in order */ 9104ed29b0b4SJens Axboe if (!node->done) 9105ed29b0b4SJens Axboe break; 9106ed29b0b4SJens Axboe list_del(&node->node); 9107ed29b0b4SJens Axboe first_add |= llist_add(&node->llist, &ctx->rsrc_put_llist); 9108ed29b0b4SJens Axboe } 9109ed29b0b4SJens Axboe spin_unlock_irqrestore(&ctx->rsrc_ref_lock, flags); 9110ed29b0b4SJens Axboe 9111ed29b0b4SJens Axboe if (first_add) 9112ed29b0b4SJens Axboe mod_delayed_work(system_wq, &ctx->rsrc_put_work, delay); 9113ed29b0b4SJens Axboe } 9114ed29b0b4SJens Axboe 9115ed29b0b4SJens Axboe static struct io_rsrc_node *io_rsrc_node_alloc(void) 9116ed29b0b4SJens Axboe { 9117ed29b0b4SJens Axboe struct io_rsrc_node *ref_node; 9118ed29b0b4SJens Axboe 9119ed29b0b4SJens Axboe ref_node = kzalloc(sizeof(*ref_node), GFP_KERNEL); 9120ed29b0b4SJens Axboe if (!ref_node) 9121ed29b0b4SJens Axboe return NULL; 9122ed29b0b4SJens Axboe 9123ed29b0b4SJens Axboe if (percpu_ref_init(&ref_node->refs, io_rsrc_node_ref_zero, 9124ed29b0b4SJens Axboe 0, GFP_KERNEL)) { 9125ed29b0b4SJens Axboe kfree(ref_node); 9126ed29b0b4SJens Axboe return NULL; 9127ed29b0b4SJens Axboe } 9128ed29b0b4SJens Axboe INIT_LIST_HEAD(&ref_node->node); 9129ed29b0b4SJens Axboe INIT_LIST_HEAD(&ref_node->rsrc_list); 9130ed29b0b4SJens Axboe ref_node->done = false; 9131ed29b0b4SJens Axboe return ref_node; 9132ed29b0b4SJens Axboe } 9133ed29b0b4SJens Axboe 9134ed29b0b4SJens Axboe static void io_rsrc_node_switch(struct io_ring_ctx *ctx, 9135ed29b0b4SJens Axboe struct io_rsrc_data *data_to_kill) 9136ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 9137ed29b0b4SJens Axboe { 9138ed29b0b4SJens Axboe WARN_ON_ONCE(!ctx->rsrc_backup_node); 9139ed29b0b4SJens Axboe WARN_ON_ONCE(data_to_kill && !ctx->rsrc_node); 9140ed29b0b4SJens Axboe 9141ed29b0b4SJens Axboe io_rsrc_refs_drop(ctx); 9142ed29b0b4SJens Axboe 9143ed29b0b4SJens Axboe if (data_to_kill) { 9144ed29b0b4SJens Axboe struct io_rsrc_node *rsrc_node = ctx->rsrc_node; 9145ed29b0b4SJens Axboe 9146ed29b0b4SJens Axboe rsrc_node->rsrc_data = data_to_kill; 9147ed29b0b4SJens Axboe spin_lock_irq(&ctx->rsrc_ref_lock); 9148ed29b0b4SJens Axboe list_add_tail(&rsrc_node->node, &ctx->rsrc_ref_list); 9149ed29b0b4SJens Axboe spin_unlock_irq(&ctx->rsrc_ref_lock); 9150ed29b0b4SJens Axboe 9151ed29b0b4SJens Axboe atomic_inc(&data_to_kill->refs); 9152ed29b0b4SJens Axboe percpu_ref_kill(&rsrc_node->refs); 9153ed29b0b4SJens Axboe ctx->rsrc_node = NULL; 9154ed29b0b4SJens Axboe } 9155ed29b0b4SJens Axboe 9156ed29b0b4SJens Axboe if (!ctx->rsrc_node) { 9157ed29b0b4SJens Axboe ctx->rsrc_node = ctx->rsrc_backup_node; 9158ed29b0b4SJens Axboe ctx->rsrc_backup_node = NULL; 9159ed29b0b4SJens Axboe } 9160ed29b0b4SJens Axboe } 9161ed29b0b4SJens Axboe 9162ed29b0b4SJens Axboe static int io_rsrc_node_switch_start(struct io_ring_ctx *ctx) 9163ed29b0b4SJens Axboe { 9164ed29b0b4SJens Axboe if (ctx->rsrc_backup_node) 9165ed29b0b4SJens Axboe return 0; 9166ed29b0b4SJens Axboe ctx->rsrc_backup_node = io_rsrc_node_alloc(); 9167ed29b0b4SJens Axboe return ctx->rsrc_backup_node ? 0 : -ENOMEM; 9168ed29b0b4SJens Axboe } 9169ed29b0b4SJens Axboe 9170ed29b0b4SJens Axboe static __cold int io_rsrc_ref_quiesce(struct io_rsrc_data *data, 9171ed29b0b4SJens Axboe struct io_ring_ctx *ctx) 9172ed29b0b4SJens Axboe { 9173ed29b0b4SJens Axboe int ret; 9174ed29b0b4SJens Axboe 9175ed29b0b4SJens Axboe /* As we may drop ->uring_lock, other task may have started quiesce */ 9176ed29b0b4SJens Axboe if (data->quiesce) 9177ed29b0b4SJens Axboe return -ENXIO; 9178ed29b0b4SJens Axboe 9179ed29b0b4SJens Axboe data->quiesce = true; 9180ed29b0b4SJens Axboe do { 9181ed29b0b4SJens Axboe ret = io_rsrc_node_switch_start(ctx); 9182ed29b0b4SJens Axboe if (ret) 9183ed29b0b4SJens Axboe break; 9184ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, data); 9185ed29b0b4SJens Axboe 9186ed29b0b4SJens Axboe /* kill initial ref, already quiesced if zero */ 9187ed29b0b4SJens Axboe if (atomic_dec_and_test(&data->refs)) 9188ed29b0b4SJens Axboe break; 9189ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 9190ed29b0b4SJens Axboe flush_delayed_work(&ctx->rsrc_put_work); 9191ed29b0b4SJens Axboe ret = wait_for_completion_interruptible(&data->done); 9192ed29b0b4SJens Axboe if (!ret) { 9193ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 9194ed29b0b4SJens Axboe if (atomic_read(&data->refs) > 0) { 9195ed29b0b4SJens Axboe /* 9196ed29b0b4SJens Axboe * it has been revived by another thread while 9197ed29b0b4SJens Axboe * we were unlocked 9198ed29b0b4SJens Axboe */ 9199ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 9200ed29b0b4SJens Axboe } else { 9201ed29b0b4SJens Axboe break; 9202ed29b0b4SJens Axboe } 9203ed29b0b4SJens Axboe } 9204ed29b0b4SJens Axboe 9205ed29b0b4SJens Axboe atomic_inc(&data->refs); 9206ed29b0b4SJens Axboe /* wait for all works potentially completing data->done */ 9207ed29b0b4SJens Axboe flush_delayed_work(&ctx->rsrc_put_work); 9208ed29b0b4SJens Axboe reinit_completion(&data->done); 9209ed29b0b4SJens Axboe 9210ed29b0b4SJens Axboe ret = io_run_task_work_sig(); 9211ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 9212ed29b0b4SJens Axboe } while (ret >= 0); 9213ed29b0b4SJens Axboe data->quiesce = false; 9214ed29b0b4SJens Axboe 9215ed29b0b4SJens Axboe return ret; 9216ed29b0b4SJens Axboe } 9217ed29b0b4SJens Axboe 9218ed29b0b4SJens Axboe static u64 *io_get_tag_slot(struct io_rsrc_data *data, unsigned int idx) 9219ed29b0b4SJens Axboe { 9220ed29b0b4SJens Axboe unsigned int off = idx & IO_RSRC_TAG_TABLE_MASK; 9221ed29b0b4SJens Axboe unsigned int table_idx = idx >> IO_RSRC_TAG_TABLE_SHIFT; 9222ed29b0b4SJens Axboe 9223ed29b0b4SJens Axboe return &data->tags[table_idx][off]; 9224ed29b0b4SJens Axboe } 9225ed29b0b4SJens Axboe 9226ed29b0b4SJens Axboe static void io_rsrc_data_free(struct io_rsrc_data *data) 9227ed29b0b4SJens Axboe { 9228ed29b0b4SJens Axboe size_t size = data->nr * sizeof(data->tags[0][0]); 9229ed29b0b4SJens Axboe 9230ed29b0b4SJens Axboe if (data->tags) 9231ed29b0b4SJens Axboe io_free_page_table((void **)data->tags, size); 9232ed29b0b4SJens Axboe kfree(data); 9233ed29b0b4SJens Axboe } 9234ed29b0b4SJens Axboe 9235ed29b0b4SJens Axboe static __cold int io_rsrc_data_alloc(struct io_ring_ctx *ctx, rsrc_put_fn *do_put, 9236ed29b0b4SJens Axboe u64 __user *utags, unsigned nr, 9237ed29b0b4SJens Axboe struct io_rsrc_data **pdata) 9238ed29b0b4SJens Axboe { 9239ed29b0b4SJens Axboe struct io_rsrc_data *data; 9240ed29b0b4SJens Axboe int ret = -ENOMEM; 9241ed29b0b4SJens Axboe unsigned i; 9242ed29b0b4SJens Axboe 9243ed29b0b4SJens Axboe data = kzalloc(sizeof(*data), GFP_KERNEL); 9244ed29b0b4SJens Axboe if (!data) 9245ed29b0b4SJens Axboe return -ENOMEM; 9246ed29b0b4SJens Axboe data->tags = (u64 **)io_alloc_page_table(nr * sizeof(data->tags[0][0])); 9247ed29b0b4SJens Axboe if (!data->tags) { 9248ed29b0b4SJens Axboe kfree(data); 9249ed29b0b4SJens Axboe return -ENOMEM; 9250ed29b0b4SJens Axboe } 9251ed29b0b4SJens Axboe 9252ed29b0b4SJens Axboe data->nr = nr; 9253ed29b0b4SJens Axboe data->ctx = ctx; 9254ed29b0b4SJens Axboe data->do_put = do_put; 9255ed29b0b4SJens Axboe if (utags) { 9256ed29b0b4SJens Axboe ret = -EFAULT; 9257ed29b0b4SJens Axboe for (i = 0; i < nr; i++) { 9258ed29b0b4SJens Axboe u64 *tag_slot = io_get_tag_slot(data, i); 9259ed29b0b4SJens Axboe 9260ed29b0b4SJens Axboe if (copy_from_user(tag_slot, &utags[i], 9261ed29b0b4SJens Axboe sizeof(*tag_slot))) 9262ed29b0b4SJens Axboe goto fail; 9263ed29b0b4SJens Axboe } 9264ed29b0b4SJens Axboe } 9265ed29b0b4SJens Axboe 9266ed29b0b4SJens Axboe atomic_set(&data->refs, 1); 9267ed29b0b4SJens Axboe init_completion(&data->done); 9268ed29b0b4SJens Axboe *pdata = data; 9269ed29b0b4SJens Axboe return 0; 9270ed29b0b4SJens Axboe fail: 9271ed29b0b4SJens Axboe io_rsrc_data_free(data); 9272ed29b0b4SJens Axboe return ret; 9273ed29b0b4SJens Axboe } 9274ed29b0b4SJens Axboe 9275ed29b0b4SJens Axboe static bool io_alloc_file_tables(struct io_file_table *table, unsigned nr_files) 9276ed29b0b4SJens Axboe { 9277ed29b0b4SJens Axboe table->files = kvcalloc(nr_files, sizeof(table->files[0]), 9278ed29b0b4SJens Axboe GFP_KERNEL_ACCOUNT); 9279ed29b0b4SJens Axboe if (unlikely(!table->files)) 9280ed29b0b4SJens Axboe return false; 9281ed29b0b4SJens Axboe 9282ed29b0b4SJens Axboe table->bitmap = bitmap_zalloc(nr_files, GFP_KERNEL_ACCOUNT); 9283ed29b0b4SJens Axboe if (unlikely(!table->bitmap)) { 9284ed29b0b4SJens Axboe kvfree(table->files); 9285ed29b0b4SJens Axboe return false; 9286ed29b0b4SJens Axboe } 9287ed29b0b4SJens Axboe 9288ed29b0b4SJens Axboe return true; 9289ed29b0b4SJens Axboe } 9290ed29b0b4SJens Axboe 9291ed29b0b4SJens Axboe static void io_free_file_tables(struct io_file_table *table) 9292ed29b0b4SJens Axboe { 9293ed29b0b4SJens Axboe kvfree(table->files); 9294ed29b0b4SJens Axboe bitmap_free(table->bitmap); 9295ed29b0b4SJens Axboe table->files = NULL; 9296ed29b0b4SJens Axboe table->bitmap = NULL; 9297ed29b0b4SJens Axboe } 9298ed29b0b4SJens Axboe 9299ed29b0b4SJens Axboe static inline void io_file_bitmap_set(struct io_file_table *table, int bit) 9300ed29b0b4SJens Axboe { 9301ed29b0b4SJens Axboe WARN_ON_ONCE(test_bit(bit, table->bitmap)); 9302ed29b0b4SJens Axboe __set_bit(bit, table->bitmap); 9303ed29b0b4SJens Axboe table->alloc_hint = bit + 1; 9304ed29b0b4SJens Axboe } 9305ed29b0b4SJens Axboe 9306ed29b0b4SJens Axboe static inline void io_file_bitmap_clear(struct io_file_table *table, int bit) 9307ed29b0b4SJens Axboe { 9308ed29b0b4SJens Axboe __clear_bit(bit, table->bitmap); 9309ed29b0b4SJens Axboe table->alloc_hint = bit; 9310ed29b0b4SJens Axboe } 9311ed29b0b4SJens Axboe 9312ed29b0b4SJens Axboe static void __io_sqe_files_unregister(struct io_ring_ctx *ctx) 9313ed29b0b4SJens Axboe { 9314ed29b0b4SJens Axboe #if !defined(IO_URING_SCM_ALL) 9315ed29b0b4SJens Axboe int i; 9316ed29b0b4SJens Axboe 9317ed29b0b4SJens Axboe for (i = 0; i < ctx->nr_user_files; i++) { 9318ed29b0b4SJens Axboe struct file *file = io_file_from_index(ctx, i); 9319ed29b0b4SJens Axboe 9320ed29b0b4SJens Axboe if (!file) 9321ed29b0b4SJens Axboe continue; 9322ed29b0b4SJens Axboe if (io_fixed_file_slot(&ctx->file_table, i)->file_ptr & FFS_SCM) 9323ed29b0b4SJens Axboe continue; 9324ed29b0b4SJens Axboe io_file_bitmap_clear(&ctx->file_table, i); 9325ed29b0b4SJens Axboe fput(file); 9326ed29b0b4SJens Axboe } 9327ed29b0b4SJens Axboe #endif 9328ed29b0b4SJens Axboe 9329ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 9330ed29b0b4SJens Axboe if (ctx->ring_sock) { 9331ed29b0b4SJens Axboe struct sock *sock = ctx->ring_sock->sk; 9332ed29b0b4SJens Axboe struct sk_buff *skb; 9333ed29b0b4SJens Axboe 9334ed29b0b4SJens Axboe while ((skb = skb_dequeue(&sock->sk_receive_queue)) != NULL) 9335ed29b0b4SJens Axboe kfree_skb(skb); 9336ed29b0b4SJens Axboe } 9337ed29b0b4SJens Axboe #endif 9338ed29b0b4SJens Axboe io_free_file_tables(&ctx->file_table); 9339ed29b0b4SJens Axboe io_rsrc_data_free(ctx->file_data); 9340ed29b0b4SJens Axboe ctx->file_data = NULL; 9341ed29b0b4SJens Axboe ctx->nr_user_files = 0; 9342ed29b0b4SJens Axboe } 9343ed29b0b4SJens Axboe 9344ed29b0b4SJens Axboe static int io_sqe_files_unregister(struct io_ring_ctx *ctx) 9345ed29b0b4SJens Axboe { 9346ed29b0b4SJens Axboe unsigned nr = ctx->nr_user_files; 9347ed29b0b4SJens Axboe int ret; 9348ed29b0b4SJens Axboe 9349ed29b0b4SJens Axboe if (!ctx->file_data) 9350ed29b0b4SJens Axboe return -ENXIO; 9351ed29b0b4SJens Axboe 9352ed29b0b4SJens Axboe /* 9353ed29b0b4SJens Axboe * Quiesce may unlock ->uring_lock, and while it's not held 9354ed29b0b4SJens Axboe * prevent new requests using the table. 9355ed29b0b4SJens Axboe */ 9356ed29b0b4SJens Axboe ctx->nr_user_files = 0; 9357ed29b0b4SJens Axboe ret = io_rsrc_ref_quiesce(ctx->file_data, ctx); 9358ed29b0b4SJens Axboe ctx->nr_user_files = nr; 9359ed29b0b4SJens Axboe if (!ret) 9360ed29b0b4SJens Axboe __io_sqe_files_unregister(ctx); 9361ed29b0b4SJens Axboe return ret; 9362ed29b0b4SJens Axboe } 9363ed29b0b4SJens Axboe 9364ed29b0b4SJens Axboe static void io_sq_thread_unpark(struct io_sq_data *sqd) 9365ed29b0b4SJens Axboe __releases(&sqd->lock) 9366ed29b0b4SJens Axboe { 9367ed29b0b4SJens Axboe WARN_ON_ONCE(sqd->thread == current); 9368ed29b0b4SJens Axboe 9369ed29b0b4SJens Axboe /* 9370ed29b0b4SJens Axboe * Do the dance but not conditional clear_bit() because it'd race with 9371ed29b0b4SJens Axboe * other threads incrementing park_pending and setting the bit. 9372ed29b0b4SJens Axboe */ 9373ed29b0b4SJens Axboe clear_bit(IO_SQ_THREAD_SHOULD_PARK, &sqd->state); 9374ed29b0b4SJens Axboe if (atomic_dec_return(&sqd->park_pending)) 9375ed29b0b4SJens Axboe set_bit(IO_SQ_THREAD_SHOULD_PARK, &sqd->state); 9376ed29b0b4SJens Axboe mutex_unlock(&sqd->lock); 9377ed29b0b4SJens Axboe } 9378ed29b0b4SJens Axboe 9379ed29b0b4SJens Axboe static void io_sq_thread_park(struct io_sq_data *sqd) 9380ed29b0b4SJens Axboe __acquires(&sqd->lock) 9381ed29b0b4SJens Axboe { 9382ed29b0b4SJens Axboe WARN_ON_ONCE(sqd->thread == current); 9383ed29b0b4SJens Axboe 9384ed29b0b4SJens Axboe atomic_inc(&sqd->park_pending); 9385ed29b0b4SJens Axboe set_bit(IO_SQ_THREAD_SHOULD_PARK, &sqd->state); 9386ed29b0b4SJens Axboe mutex_lock(&sqd->lock); 9387ed29b0b4SJens Axboe if (sqd->thread) 9388ed29b0b4SJens Axboe wake_up_process(sqd->thread); 9389ed29b0b4SJens Axboe } 9390ed29b0b4SJens Axboe 9391ed29b0b4SJens Axboe static void io_sq_thread_stop(struct io_sq_data *sqd) 9392ed29b0b4SJens Axboe { 9393ed29b0b4SJens Axboe WARN_ON_ONCE(sqd->thread == current); 9394ed29b0b4SJens Axboe WARN_ON_ONCE(test_bit(IO_SQ_THREAD_SHOULD_STOP, &sqd->state)); 9395ed29b0b4SJens Axboe 9396ed29b0b4SJens Axboe set_bit(IO_SQ_THREAD_SHOULD_STOP, &sqd->state); 9397ed29b0b4SJens Axboe mutex_lock(&sqd->lock); 9398ed29b0b4SJens Axboe if (sqd->thread) 9399ed29b0b4SJens Axboe wake_up_process(sqd->thread); 9400ed29b0b4SJens Axboe mutex_unlock(&sqd->lock); 9401ed29b0b4SJens Axboe wait_for_completion(&sqd->exited); 9402ed29b0b4SJens Axboe } 9403ed29b0b4SJens Axboe 9404ed29b0b4SJens Axboe static void io_put_sq_data(struct io_sq_data *sqd) 9405ed29b0b4SJens Axboe { 9406ed29b0b4SJens Axboe if (refcount_dec_and_test(&sqd->refs)) { 9407ed29b0b4SJens Axboe WARN_ON_ONCE(atomic_read(&sqd->park_pending)); 9408ed29b0b4SJens Axboe 9409ed29b0b4SJens Axboe io_sq_thread_stop(sqd); 9410ed29b0b4SJens Axboe kfree(sqd); 9411ed29b0b4SJens Axboe } 9412ed29b0b4SJens Axboe } 9413ed29b0b4SJens Axboe 9414ed29b0b4SJens Axboe static void io_sq_thread_finish(struct io_ring_ctx *ctx) 9415ed29b0b4SJens Axboe { 9416ed29b0b4SJens Axboe struct io_sq_data *sqd = ctx->sq_data; 9417ed29b0b4SJens Axboe 9418ed29b0b4SJens Axboe if (sqd) { 9419ed29b0b4SJens Axboe io_sq_thread_park(sqd); 9420ed29b0b4SJens Axboe list_del_init(&ctx->sqd_list); 9421ed29b0b4SJens Axboe io_sqd_update_thread_idle(sqd); 9422ed29b0b4SJens Axboe io_sq_thread_unpark(sqd); 9423ed29b0b4SJens Axboe 9424ed29b0b4SJens Axboe io_put_sq_data(sqd); 9425ed29b0b4SJens Axboe ctx->sq_data = NULL; 9426ed29b0b4SJens Axboe } 9427ed29b0b4SJens Axboe } 9428ed29b0b4SJens Axboe 9429ed29b0b4SJens Axboe static struct io_sq_data *io_attach_sq_data(struct io_uring_params *p) 9430ed29b0b4SJens Axboe { 9431ed29b0b4SJens Axboe struct io_ring_ctx *ctx_attach; 9432ed29b0b4SJens Axboe struct io_sq_data *sqd; 9433ed29b0b4SJens Axboe struct fd f; 9434ed29b0b4SJens Axboe 9435ed29b0b4SJens Axboe f = fdget(p->wq_fd); 9436ed29b0b4SJens Axboe if (!f.file) 9437ed29b0b4SJens Axboe return ERR_PTR(-ENXIO); 9438ed29b0b4SJens Axboe if (f.file->f_op != &io_uring_fops) { 9439ed29b0b4SJens Axboe fdput(f); 9440ed29b0b4SJens Axboe return ERR_PTR(-EINVAL); 9441ed29b0b4SJens Axboe } 9442ed29b0b4SJens Axboe 9443ed29b0b4SJens Axboe ctx_attach = f.file->private_data; 9444ed29b0b4SJens Axboe sqd = ctx_attach->sq_data; 9445ed29b0b4SJens Axboe if (!sqd) { 9446ed29b0b4SJens Axboe fdput(f); 9447ed29b0b4SJens Axboe return ERR_PTR(-EINVAL); 9448ed29b0b4SJens Axboe } 9449ed29b0b4SJens Axboe if (sqd->task_tgid != current->tgid) { 9450ed29b0b4SJens Axboe fdput(f); 9451ed29b0b4SJens Axboe return ERR_PTR(-EPERM); 9452ed29b0b4SJens Axboe } 9453ed29b0b4SJens Axboe 9454ed29b0b4SJens Axboe refcount_inc(&sqd->refs); 9455ed29b0b4SJens Axboe fdput(f); 9456ed29b0b4SJens Axboe return sqd; 9457ed29b0b4SJens Axboe } 9458ed29b0b4SJens Axboe 9459ed29b0b4SJens Axboe static struct io_sq_data *io_get_sq_data(struct io_uring_params *p, 9460ed29b0b4SJens Axboe bool *attached) 9461ed29b0b4SJens Axboe { 9462ed29b0b4SJens Axboe struct io_sq_data *sqd; 9463ed29b0b4SJens Axboe 9464ed29b0b4SJens Axboe *attached = false; 9465ed29b0b4SJens Axboe if (p->flags & IORING_SETUP_ATTACH_WQ) { 9466ed29b0b4SJens Axboe sqd = io_attach_sq_data(p); 9467ed29b0b4SJens Axboe if (!IS_ERR(sqd)) { 9468ed29b0b4SJens Axboe *attached = true; 9469ed29b0b4SJens Axboe return sqd; 9470ed29b0b4SJens Axboe } 9471ed29b0b4SJens Axboe /* fall through for EPERM case, setup new sqd/task */ 9472ed29b0b4SJens Axboe if (PTR_ERR(sqd) != -EPERM) 9473ed29b0b4SJens Axboe return sqd; 9474ed29b0b4SJens Axboe } 9475ed29b0b4SJens Axboe 9476ed29b0b4SJens Axboe sqd = kzalloc(sizeof(*sqd), GFP_KERNEL); 9477ed29b0b4SJens Axboe if (!sqd) 9478ed29b0b4SJens Axboe return ERR_PTR(-ENOMEM); 9479ed29b0b4SJens Axboe 9480ed29b0b4SJens Axboe atomic_set(&sqd->park_pending, 0); 9481ed29b0b4SJens Axboe refcount_set(&sqd->refs, 1); 9482ed29b0b4SJens Axboe INIT_LIST_HEAD(&sqd->ctx_list); 9483ed29b0b4SJens Axboe mutex_init(&sqd->lock); 9484ed29b0b4SJens Axboe init_waitqueue_head(&sqd->wait); 9485ed29b0b4SJens Axboe init_completion(&sqd->exited); 9486ed29b0b4SJens Axboe return sqd; 9487ed29b0b4SJens Axboe } 9488ed29b0b4SJens Axboe 9489ed29b0b4SJens Axboe /* 9490ed29b0b4SJens Axboe * Ensure the UNIX gc is aware of our file set, so we are certain that 9491ed29b0b4SJens Axboe * the io_uring can be safely unregistered on process exit, even if we have 9492ed29b0b4SJens Axboe * loops in the file referencing. We account only files that can hold other 9493ed29b0b4SJens Axboe * files because otherwise they can't form a loop and so are not interesting 9494ed29b0b4SJens Axboe * for GC. 9495ed29b0b4SJens Axboe */ 9496ed29b0b4SJens Axboe static int io_scm_file_account(struct io_ring_ctx *ctx, struct file *file) 9497ed29b0b4SJens Axboe { 9498ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 9499ed29b0b4SJens Axboe struct sock *sk = ctx->ring_sock->sk; 9500ed29b0b4SJens Axboe struct sk_buff_head *head = &sk->sk_receive_queue; 9501ed29b0b4SJens Axboe struct scm_fp_list *fpl; 9502ed29b0b4SJens Axboe struct sk_buff *skb; 9503ed29b0b4SJens Axboe 9504ed29b0b4SJens Axboe if (likely(!io_file_need_scm(file))) 9505ed29b0b4SJens Axboe return 0; 9506ed29b0b4SJens Axboe 9507ed29b0b4SJens Axboe /* 9508ed29b0b4SJens Axboe * See if we can merge this file into an existing skb SCM_RIGHTS 9509ed29b0b4SJens Axboe * file set. If there's no room, fall back to allocating a new skb 9510ed29b0b4SJens Axboe * and filling it in. 9511ed29b0b4SJens Axboe */ 9512ed29b0b4SJens Axboe spin_lock_irq(&head->lock); 9513ed29b0b4SJens Axboe skb = skb_peek(head); 9514ed29b0b4SJens Axboe if (skb && UNIXCB(skb).fp->count < SCM_MAX_FD) 9515ed29b0b4SJens Axboe __skb_unlink(skb, head); 9516ed29b0b4SJens Axboe else 9517ed29b0b4SJens Axboe skb = NULL; 9518ed29b0b4SJens Axboe spin_unlock_irq(&head->lock); 9519ed29b0b4SJens Axboe 9520ed29b0b4SJens Axboe if (!skb) { 9521ed29b0b4SJens Axboe fpl = kzalloc(sizeof(*fpl), GFP_KERNEL); 9522ed29b0b4SJens Axboe if (!fpl) 9523ed29b0b4SJens Axboe return -ENOMEM; 9524ed29b0b4SJens Axboe 9525ed29b0b4SJens Axboe skb = alloc_skb(0, GFP_KERNEL); 9526ed29b0b4SJens Axboe if (!skb) { 9527ed29b0b4SJens Axboe kfree(fpl); 9528ed29b0b4SJens Axboe return -ENOMEM; 9529ed29b0b4SJens Axboe } 9530ed29b0b4SJens Axboe 9531ed29b0b4SJens Axboe fpl->user = get_uid(current_user()); 9532ed29b0b4SJens Axboe fpl->max = SCM_MAX_FD; 9533ed29b0b4SJens Axboe fpl->count = 0; 9534ed29b0b4SJens Axboe 9535ed29b0b4SJens Axboe UNIXCB(skb).fp = fpl; 9536ed29b0b4SJens Axboe skb->sk = sk; 9537ed29b0b4SJens Axboe skb->destructor = unix_destruct_scm; 9538ed29b0b4SJens Axboe refcount_add(skb->truesize, &sk->sk_wmem_alloc); 9539ed29b0b4SJens Axboe } 9540ed29b0b4SJens Axboe 9541ed29b0b4SJens Axboe fpl = UNIXCB(skb).fp; 9542ed29b0b4SJens Axboe fpl->fp[fpl->count++] = get_file(file); 9543ed29b0b4SJens Axboe unix_inflight(fpl->user, file); 9544ed29b0b4SJens Axboe skb_queue_head(head, skb); 9545ed29b0b4SJens Axboe fput(file); 9546ed29b0b4SJens Axboe #endif 9547ed29b0b4SJens Axboe return 0; 9548ed29b0b4SJens Axboe } 9549ed29b0b4SJens Axboe 9550ed29b0b4SJens Axboe static void io_rsrc_file_put(struct io_ring_ctx *ctx, struct io_rsrc_put *prsrc) 9551ed29b0b4SJens Axboe { 9552ed29b0b4SJens Axboe struct file *file = prsrc->file; 9553ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 9554ed29b0b4SJens Axboe struct sock *sock = ctx->ring_sock->sk; 9555ed29b0b4SJens Axboe struct sk_buff_head list, *head = &sock->sk_receive_queue; 9556ed29b0b4SJens Axboe struct sk_buff *skb; 9557ed29b0b4SJens Axboe int i; 9558ed29b0b4SJens Axboe 9559ed29b0b4SJens Axboe if (!io_file_need_scm(file)) { 9560ed29b0b4SJens Axboe fput(file); 9561ed29b0b4SJens Axboe return; 9562ed29b0b4SJens Axboe } 9563ed29b0b4SJens Axboe 9564ed29b0b4SJens Axboe __skb_queue_head_init(&list); 9565ed29b0b4SJens Axboe 9566ed29b0b4SJens Axboe /* 9567ed29b0b4SJens Axboe * Find the skb that holds this file in its SCM_RIGHTS. When found, 9568ed29b0b4SJens Axboe * remove this entry and rearrange the file array. 9569ed29b0b4SJens Axboe */ 9570ed29b0b4SJens Axboe skb = skb_dequeue(head); 9571ed29b0b4SJens Axboe while (skb) { 9572ed29b0b4SJens Axboe struct scm_fp_list *fp; 9573ed29b0b4SJens Axboe 9574ed29b0b4SJens Axboe fp = UNIXCB(skb).fp; 9575ed29b0b4SJens Axboe for (i = 0; i < fp->count; i++) { 9576ed29b0b4SJens Axboe int left; 9577ed29b0b4SJens Axboe 9578ed29b0b4SJens Axboe if (fp->fp[i] != file) 9579ed29b0b4SJens Axboe continue; 9580ed29b0b4SJens Axboe 9581ed29b0b4SJens Axboe unix_notinflight(fp->user, fp->fp[i]); 9582ed29b0b4SJens Axboe left = fp->count - 1 - i; 9583ed29b0b4SJens Axboe if (left) { 9584ed29b0b4SJens Axboe memmove(&fp->fp[i], &fp->fp[i + 1], 9585ed29b0b4SJens Axboe left * sizeof(struct file *)); 9586ed29b0b4SJens Axboe } 9587ed29b0b4SJens Axboe fp->count--; 9588ed29b0b4SJens Axboe if (!fp->count) { 9589ed29b0b4SJens Axboe kfree_skb(skb); 9590ed29b0b4SJens Axboe skb = NULL; 9591ed29b0b4SJens Axboe } else { 9592ed29b0b4SJens Axboe __skb_queue_tail(&list, skb); 9593ed29b0b4SJens Axboe } 9594ed29b0b4SJens Axboe fput(file); 9595ed29b0b4SJens Axboe file = NULL; 9596ed29b0b4SJens Axboe break; 9597ed29b0b4SJens Axboe } 9598ed29b0b4SJens Axboe 9599ed29b0b4SJens Axboe if (!file) 9600ed29b0b4SJens Axboe break; 9601ed29b0b4SJens Axboe 9602ed29b0b4SJens Axboe __skb_queue_tail(&list, skb); 9603ed29b0b4SJens Axboe 9604ed29b0b4SJens Axboe skb = skb_dequeue(head); 9605ed29b0b4SJens Axboe } 9606ed29b0b4SJens Axboe 9607ed29b0b4SJens Axboe if (skb_peek(&list)) { 9608ed29b0b4SJens Axboe spin_lock_irq(&head->lock); 9609ed29b0b4SJens Axboe while ((skb = __skb_dequeue(&list)) != NULL) 9610ed29b0b4SJens Axboe __skb_queue_tail(head, skb); 9611ed29b0b4SJens Axboe spin_unlock_irq(&head->lock); 9612ed29b0b4SJens Axboe } 9613ed29b0b4SJens Axboe #else 9614ed29b0b4SJens Axboe fput(file); 9615ed29b0b4SJens Axboe #endif 9616ed29b0b4SJens Axboe } 9617ed29b0b4SJens Axboe 9618ed29b0b4SJens Axboe static void __io_rsrc_put_work(struct io_rsrc_node *ref_node) 9619ed29b0b4SJens Axboe { 9620ed29b0b4SJens Axboe struct io_rsrc_data *rsrc_data = ref_node->rsrc_data; 9621ed29b0b4SJens Axboe struct io_ring_ctx *ctx = rsrc_data->ctx; 9622ed29b0b4SJens Axboe struct io_rsrc_put *prsrc, *tmp; 9623ed29b0b4SJens Axboe 9624ed29b0b4SJens Axboe list_for_each_entry_safe(prsrc, tmp, &ref_node->rsrc_list, list) { 9625ed29b0b4SJens Axboe list_del(&prsrc->list); 9626ed29b0b4SJens Axboe 9627ed29b0b4SJens Axboe if (prsrc->tag) { 9628ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL) 9629ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 9630ed29b0b4SJens Axboe 9631ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 9632ed29b0b4SJens Axboe io_fill_cqe_aux(ctx, prsrc->tag, 0, 0); 9633ed29b0b4SJens Axboe io_commit_cqring(ctx); 9634ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 9635ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 9636ed29b0b4SJens Axboe 9637ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL) 9638ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 9639ed29b0b4SJens Axboe } 9640ed29b0b4SJens Axboe 9641ed29b0b4SJens Axboe rsrc_data->do_put(ctx, prsrc); 9642ed29b0b4SJens Axboe kfree(prsrc); 9643ed29b0b4SJens Axboe } 9644ed29b0b4SJens Axboe 9645ed29b0b4SJens Axboe io_rsrc_node_destroy(ref_node); 9646ed29b0b4SJens Axboe if (atomic_dec_and_test(&rsrc_data->refs)) 9647ed29b0b4SJens Axboe complete(&rsrc_data->done); 9648ed29b0b4SJens Axboe } 9649ed29b0b4SJens Axboe 9650ed29b0b4SJens Axboe static void io_rsrc_put_work(struct work_struct *work) 9651ed29b0b4SJens Axboe { 9652ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 9653ed29b0b4SJens Axboe struct llist_node *node; 9654ed29b0b4SJens Axboe 9655ed29b0b4SJens Axboe ctx = container_of(work, struct io_ring_ctx, rsrc_put_work.work); 9656ed29b0b4SJens Axboe node = llist_del_all(&ctx->rsrc_put_llist); 9657ed29b0b4SJens Axboe 9658ed29b0b4SJens Axboe while (node) { 9659ed29b0b4SJens Axboe struct io_rsrc_node *ref_node; 9660ed29b0b4SJens Axboe struct llist_node *next = node->next; 9661ed29b0b4SJens Axboe 9662ed29b0b4SJens Axboe ref_node = llist_entry(node, struct io_rsrc_node, llist); 9663ed29b0b4SJens Axboe __io_rsrc_put_work(ref_node); 9664ed29b0b4SJens Axboe node = next; 9665ed29b0b4SJens Axboe } 9666ed29b0b4SJens Axboe } 9667ed29b0b4SJens Axboe 9668ed29b0b4SJens Axboe static int io_sqe_files_register(struct io_ring_ctx *ctx, void __user *arg, 9669ed29b0b4SJens Axboe unsigned nr_args, u64 __user *tags) 9670ed29b0b4SJens Axboe { 9671ed29b0b4SJens Axboe __s32 __user *fds = (__s32 __user *) arg; 9672ed29b0b4SJens Axboe struct file *file; 9673ed29b0b4SJens Axboe int fd, ret; 9674ed29b0b4SJens Axboe unsigned i; 9675ed29b0b4SJens Axboe 9676ed29b0b4SJens Axboe if (ctx->file_data) 9677ed29b0b4SJens Axboe return -EBUSY; 9678ed29b0b4SJens Axboe if (!nr_args) 9679ed29b0b4SJens Axboe return -EINVAL; 9680ed29b0b4SJens Axboe if (nr_args > IORING_MAX_FIXED_FILES) 9681ed29b0b4SJens Axboe return -EMFILE; 9682ed29b0b4SJens Axboe if (nr_args > rlimit(RLIMIT_NOFILE)) 9683ed29b0b4SJens Axboe return -EMFILE; 9684ed29b0b4SJens Axboe ret = io_rsrc_node_switch_start(ctx); 9685ed29b0b4SJens Axboe if (ret) 9686ed29b0b4SJens Axboe return ret; 9687ed29b0b4SJens Axboe ret = io_rsrc_data_alloc(ctx, io_rsrc_file_put, tags, nr_args, 9688ed29b0b4SJens Axboe &ctx->file_data); 9689ed29b0b4SJens Axboe if (ret) 9690ed29b0b4SJens Axboe return ret; 9691ed29b0b4SJens Axboe 9692ed29b0b4SJens Axboe if (!io_alloc_file_tables(&ctx->file_table, nr_args)) { 9693ed29b0b4SJens Axboe io_rsrc_data_free(ctx->file_data); 9694ed29b0b4SJens Axboe ctx->file_data = NULL; 9695ed29b0b4SJens Axboe return -ENOMEM; 9696ed29b0b4SJens Axboe } 9697ed29b0b4SJens Axboe 9698ed29b0b4SJens Axboe for (i = 0; i < nr_args; i++, ctx->nr_user_files++) { 9699ed29b0b4SJens Axboe struct io_fixed_file *file_slot; 9700ed29b0b4SJens Axboe 9701ed29b0b4SJens Axboe if (fds && copy_from_user(&fd, &fds[i], sizeof(fd))) { 9702ed29b0b4SJens Axboe ret = -EFAULT; 9703ed29b0b4SJens Axboe goto fail; 9704ed29b0b4SJens Axboe } 9705ed29b0b4SJens Axboe /* allow sparse sets */ 9706ed29b0b4SJens Axboe if (!fds || fd == -1) { 9707ed29b0b4SJens Axboe ret = -EINVAL; 9708ed29b0b4SJens Axboe if (unlikely(*io_get_tag_slot(ctx->file_data, i))) 9709ed29b0b4SJens Axboe goto fail; 9710ed29b0b4SJens Axboe continue; 9711ed29b0b4SJens Axboe } 9712ed29b0b4SJens Axboe 9713ed29b0b4SJens Axboe file = fget(fd); 9714ed29b0b4SJens Axboe ret = -EBADF; 9715ed29b0b4SJens Axboe if (unlikely(!file)) 9716ed29b0b4SJens Axboe goto fail; 9717ed29b0b4SJens Axboe 9718ed29b0b4SJens Axboe /* 9719ed29b0b4SJens Axboe * Don't allow io_uring instances to be registered. If UNIX 9720ed29b0b4SJens Axboe * isn't enabled, then this causes a reference cycle and this 9721ed29b0b4SJens Axboe * instance can never get freed. If UNIX is enabled we'll 9722ed29b0b4SJens Axboe * handle it just fine, but there's still no point in allowing 9723ed29b0b4SJens Axboe * a ring fd as it doesn't support regular read/write anyway. 9724ed29b0b4SJens Axboe */ 9725ed29b0b4SJens Axboe if (file->f_op == &io_uring_fops) { 9726ed29b0b4SJens Axboe fput(file); 9727ed29b0b4SJens Axboe goto fail; 9728ed29b0b4SJens Axboe } 9729ed29b0b4SJens Axboe ret = io_scm_file_account(ctx, file); 9730ed29b0b4SJens Axboe if (ret) { 9731ed29b0b4SJens Axboe fput(file); 9732ed29b0b4SJens Axboe goto fail; 9733ed29b0b4SJens Axboe } 9734ed29b0b4SJens Axboe file_slot = io_fixed_file_slot(&ctx->file_table, i); 9735ed29b0b4SJens Axboe io_fixed_file_set(file_slot, file); 9736ed29b0b4SJens Axboe io_file_bitmap_set(&ctx->file_table, i); 9737ed29b0b4SJens Axboe } 9738ed29b0b4SJens Axboe 9739ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, NULL); 9740ed29b0b4SJens Axboe return 0; 9741ed29b0b4SJens Axboe fail: 9742ed29b0b4SJens Axboe __io_sqe_files_unregister(ctx); 9743ed29b0b4SJens Axboe return ret; 9744ed29b0b4SJens Axboe } 9745ed29b0b4SJens Axboe 9746ed29b0b4SJens Axboe static int io_queue_rsrc_removal(struct io_rsrc_data *data, unsigned idx, 9747ed29b0b4SJens Axboe struct io_rsrc_node *node, void *rsrc) 9748ed29b0b4SJens Axboe { 9749ed29b0b4SJens Axboe u64 *tag_slot = io_get_tag_slot(data, idx); 9750ed29b0b4SJens Axboe struct io_rsrc_put *prsrc; 9751ed29b0b4SJens Axboe 9752ed29b0b4SJens Axboe prsrc = kzalloc(sizeof(*prsrc), GFP_KERNEL); 9753ed29b0b4SJens Axboe if (!prsrc) 9754ed29b0b4SJens Axboe return -ENOMEM; 9755ed29b0b4SJens Axboe 9756ed29b0b4SJens Axboe prsrc->tag = *tag_slot; 9757ed29b0b4SJens Axboe *tag_slot = 0; 9758ed29b0b4SJens Axboe prsrc->rsrc = rsrc; 9759ed29b0b4SJens Axboe list_add(&prsrc->list, &node->rsrc_list); 9760ed29b0b4SJens Axboe return 0; 9761ed29b0b4SJens Axboe } 9762ed29b0b4SJens Axboe 9763ed29b0b4SJens Axboe static int io_install_fixed_file(struct io_kiocb *req, struct file *file, 9764ed29b0b4SJens Axboe unsigned int issue_flags, u32 slot_index) 9765ed29b0b4SJens Axboe __must_hold(&req->ctx->uring_lock) 9766ed29b0b4SJens Axboe { 9767ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 9768ed29b0b4SJens Axboe bool needs_switch = false; 9769ed29b0b4SJens Axboe struct io_fixed_file *file_slot; 9770ed29b0b4SJens Axboe int ret; 9771ed29b0b4SJens Axboe 9772ed29b0b4SJens Axboe if (file->f_op == &io_uring_fops) 9773ed29b0b4SJens Axboe return -EBADF; 9774ed29b0b4SJens Axboe if (!ctx->file_data) 9775ed29b0b4SJens Axboe return -ENXIO; 9776ed29b0b4SJens Axboe if (slot_index >= ctx->nr_user_files) 9777ed29b0b4SJens Axboe return -EINVAL; 9778ed29b0b4SJens Axboe 9779ed29b0b4SJens Axboe slot_index = array_index_nospec(slot_index, ctx->nr_user_files); 9780ed29b0b4SJens Axboe file_slot = io_fixed_file_slot(&ctx->file_table, slot_index); 9781ed29b0b4SJens Axboe 9782ed29b0b4SJens Axboe if (file_slot->file_ptr) { 9783ed29b0b4SJens Axboe struct file *old_file; 9784ed29b0b4SJens Axboe 9785ed29b0b4SJens Axboe ret = io_rsrc_node_switch_start(ctx); 9786ed29b0b4SJens Axboe if (ret) 9787ed29b0b4SJens Axboe goto err; 9788ed29b0b4SJens Axboe 9789ed29b0b4SJens Axboe old_file = (struct file *)(file_slot->file_ptr & FFS_MASK); 9790ed29b0b4SJens Axboe ret = io_queue_rsrc_removal(ctx->file_data, slot_index, 9791ed29b0b4SJens Axboe ctx->rsrc_node, old_file); 9792ed29b0b4SJens Axboe if (ret) 9793ed29b0b4SJens Axboe goto err; 9794ed29b0b4SJens Axboe file_slot->file_ptr = 0; 9795ed29b0b4SJens Axboe io_file_bitmap_clear(&ctx->file_table, slot_index); 9796ed29b0b4SJens Axboe needs_switch = true; 9797ed29b0b4SJens Axboe } 9798ed29b0b4SJens Axboe 9799ed29b0b4SJens Axboe ret = io_scm_file_account(ctx, file); 9800ed29b0b4SJens Axboe if (!ret) { 9801ed29b0b4SJens Axboe *io_get_tag_slot(ctx->file_data, slot_index) = 0; 9802ed29b0b4SJens Axboe io_fixed_file_set(file_slot, file); 9803ed29b0b4SJens Axboe io_file_bitmap_set(&ctx->file_table, slot_index); 9804ed29b0b4SJens Axboe } 9805ed29b0b4SJens Axboe err: 9806ed29b0b4SJens Axboe if (needs_switch) 9807ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, ctx->file_data); 9808ed29b0b4SJens Axboe if (ret) 9809ed29b0b4SJens Axboe fput(file); 9810ed29b0b4SJens Axboe return ret; 9811ed29b0b4SJens Axboe } 9812ed29b0b4SJens Axboe 9813ed29b0b4SJens Axboe static int __io_close_fixed(struct io_kiocb *req, unsigned int issue_flags, 9814ed29b0b4SJens Axboe unsigned int offset) 9815ed29b0b4SJens Axboe { 9816ed29b0b4SJens Axboe struct io_ring_ctx *ctx = req->ctx; 9817ed29b0b4SJens Axboe struct io_fixed_file *file_slot; 9818ed29b0b4SJens Axboe struct file *file; 9819ed29b0b4SJens Axboe int ret; 9820ed29b0b4SJens Axboe 9821ed29b0b4SJens Axboe io_ring_submit_lock(ctx, issue_flags); 9822ed29b0b4SJens Axboe ret = -ENXIO; 9823ed29b0b4SJens Axboe if (unlikely(!ctx->file_data)) 9824ed29b0b4SJens Axboe goto out; 9825ed29b0b4SJens Axboe ret = -EINVAL; 9826ed29b0b4SJens Axboe if (offset >= ctx->nr_user_files) 9827ed29b0b4SJens Axboe goto out; 9828ed29b0b4SJens Axboe ret = io_rsrc_node_switch_start(ctx); 9829ed29b0b4SJens Axboe if (ret) 9830ed29b0b4SJens Axboe goto out; 9831ed29b0b4SJens Axboe 9832ed29b0b4SJens Axboe offset = array_index_nospec(offset, ctx->nr_user_files); 9833ed29b0b4SJens Axboe file_slot = io_fixed_file_slot(&ctx->file_table, offset); 9834ed29b0b4SJens Axboe ret = -EBADF; 9835ed29b0b4SJens Axboe if (!file_slot->file_ptr) 9836ed29b0b4SJens Axboe goto out; 9837ed29b0b4SJens Axboe 9838ed29b0b4SJens Axboe file = (struct file *)(file_slot->file_ptr & FFS_MASK); 9839ed29b0b4SJens Axboe ret = io_queue_rsrc_removal(ctx->file_data, offset, ctx->rsrc_node, file); 9840ed29b0b4SJens Axboe if (ret) 9841ed29b0b4SJens Axboe goto out; 9842ed29b0b4SJens Axboe 9843ed29b0b4SJens Axboe file_slot->file_ptr = 0; 9844ed29b0b4SJens Axboe io_file_bitmap_clear(&ctx->file_table, offset); 9845ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, ctx->file_data); 9846ed29b0b4SJens Axboe ret = 0; 9847ed29b0b4SJens Axboe out: 9848ed29b0b4SJens Axboe io_ring_submit_unlock(ctx, issue_flags); 9849ed29b0b4SJens Axboe return ret; 9850ed29b0b4SJens Axboe } 9851ed29b0b4SJens Axboe 9852ed29b0b4SJens Axboe static inline int io_close_fixed(struct io_kiocb *req, unsigned int issue_flags) 9853ed29b0b4SJens Axboe { 9854dd752582SJens Axboe struct io_close *close = io_kiocb_to_cmd(req); 9855dd752582SJens Axboe 9856dd752582SJens Axboe return __io_close_fixed(req, issue_flags, close->file_slot - 1); 9857ed29b0b4SJens Axboe } 9858ed29b0b4SJens Axboe 9859ed29b0b4SJens Axboe static int __io_sqe_files_update(struct io_ring_ctx *ctx, 9860ed29b0b4SJens Axboe struct io_uring_rsrc_update2 *up, 9861ed29b0b4SJens Axboe unsigned nr_args) 9862ed29b0b4SJens Axboe { 9863ed29b0b4SJens Axboe u64 __user *tags = u64_to_user_ptr(up->tags); 9864ed29b0b4SJens Axboe __s32 __user *fds = u64_to_user_ptr(up->data); 9865ed29b0b4SJens Axboe struct io_rsrc_data *data = ctx->file_data; 9866ed29b0b4SJens Axboe struct io_fixed_file *file_slot; 9867ed29b0b4SJens Axboe struct file *file; 9868ed29b0b4SJens Axboe int fd, i, err = 0; 9869ed29b0b4SJens Axboe unsigned int done; 9870ed29b0b4SJens Axboe bool needs_switch = false; 9871ed29b0b4SJens Axboe 9872ed29b0b4SJens Axboe if (!ctx->file_data) 9873ed29b0b4SJens Axboe return -ENXIO; 9874ed29b0b4SJens Axboe if (up->offset + nr_args > ctx->nr_user_files) 9875ed29b0b4SJens Axboe return -EINVAL; 9876ed29b0b4SJens Axboe 9877ed29b0b4SJens Axboe for (done = 0; done < nr_args; done++) { 9878ed29b0b4SJens Axboe u64 tag = 0; 9879ed29b0b4SJens Axboe 9880ed29b0b4SJens Axboe if ((tags && copy_from_user(&tag, &tags[done], sizeof(tag))) || 9881ed29b0b4SJens Axboe copy_from_user(&fd, &fds[done], sizeof(fd))) { 9882ed29b0b4SJens Axboe err = -EFAULT; 9883ed29b0b4SJens Axboe break; 9884ed29b0b4SJens Axboe } 9885ed29b0b4SJens Axboe if ((fd == IORING_REGISTER_FILES_SKIP || fd == -1) && tag) { 9886ed29b0b4SJens Axboe err = -EINVAL; 9887ed29b0b4SJens Axboe break; 9888ed29b0b4SJens Axboe } 9889ed29b0b4SJens Axboe if (fd == IORING_REGISTER_FILES_SKIP) 9890ed29b0b4SJens Axboe continue; 9891ed29b0b4SJens Axboe 9892ed29b0b4SJens Axboe i = array_index_nospec(up->offset + done, ctx->nr_user_files); 9893ed29b0b4SJens Axboe file_slot = io_fixed_file_slot(&ctx->file_table, i); 9894ed29b0b4SJens Axboe 9895ed29b0b4SJens Axboe if (file_slot->file_ptr) { 9896ed29b0b4SJens Axboe file = (struct file *)(file_slot->file_ptr & FFS_MASK); 9897ed29b0b4SJens Axboe err = io_queue_rsrc_removal(data, i, ctx->rsrc_node, file); 9898ed29b0b4SJens Axboe if (err) 9899ed29b0b4SJens Axboe break; 9900ed29b0b4SJens Axboe file_slot->file_ptr = 0; 9901ed29b0b4SJens Axboe io_file_bitmap_clear(&ctx->file_table, i); 9902ed29b0b4SJens Axboe needs_switch = true; 9903ed29b0b4SJens Axboe } 9904ed29b0b4SJens Axboe if (fd != -1) { 9905ed29b0b4SJens Axboe file = fget(fd); 9906ed29b0b4SJens Axboe if (!file) { 9907ed29b0b4SJens Axboe err = -EBADF; 9908ed29b0b4SJens Axboe break; 9909ed29b0b4SJens Axboe } 9910ed29b0b4SJens Axboe /* 9911ed29b0b4SJens Axboe * Don't allow io_uring instances to be registered. If 9912ed29b0b4SJens Axboe * UNIX isn't enabled, then this causes a reference 9913ed29b0b4SJens Axboe * cycle and this instance can never get freed. If UNIX 9914ed29b0b4SJens Axboe * is enabled we'll handle it just fine, but there's 9915ed29b0b4SJens Axboe * still no point in allowing a ring fd as it doesn't 9916ed29b0b4SJens Axboe * support regular read/write anyway. 9917ed29b0b4SJens Axboe */ 9918ed29b0b4SJens Axboe if (file->f_op == &io_uring_fops) { 9919ed29b0b4SJens Axboe fput(file); 9920ed29b0b4SJens Axboe err = -EBADF; 9921ed29b0b4SJens Axboe break; 9922ed29b0b4SJens Axboe } 9923ed29b0b4SJens Axboe err = io_scm_file_account(ctx, file); 9924ed29b0b4SJens Axboe if (err) { 9925ed29b0b4SJens Axboe fput(file); 9926ed29b0b4SJens Axboe break; 9927ed29b0b4SJens Axboe } 9928ed29b0b4SJens Axboe *io_get_tag_slot(data, i) = tag; 9929ed29b0b4SJens Axboe io_fixed_file_set(file_slot, file); 9930ed29b0b4SJens Axboe io_file_bitmap_set(&ctx->file_table, i); 9931ed29b0b4SJens Axboe } 9932ed29b0b4SJens Axboe } 9933ed29b0b4SJens Axboe 9934ed29b0b4SJens Axboe if (needs_switch) 9935ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, data); 9936ed29b0b4SJens Axboe return done ? done : err; 9937ed29b0b4SJens Axboe } 9938ed29b0b4SJens Axboe 9939ed29b0b4SJens Axboe static struct io_wq *io_init_wq_offload(struct io_ring_ctx *ctx, 9940ed29b0b4SJens Axboe struct task_struct *task) 9941ed29b0b4SJens Axboe { 9942ed29b0b4SJens Axboe struct io_wq_hash *hash; 9943ed29b0b4SJens Axboe struct io_wq_data data; 9944ed29b0b4SJens Axboe unsigned int concurrency; 9945ed29b0b4SJens Axboe 9946ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 9947ed29b0b4SJens Axboe hash = ctx->hash_map; 9948ed29b0b4SJens Axboe if (!hash) { 9949ed29b0b4SJens Axboe hash = kzalloc(sizeof(*hash), GFP_KERNEL); 9950ed29b0b4SJens Axboe if (!hash) { 9951ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 9952ed29b0b4SJens Axboe return ERR_PTR(-ENOMEM); 9953ed29b0b4SJens Axboe } 9954ed29b0b4SJens Axboe refcount_set(&hash->refs, 1); 9955ed29b0b4SJens Axboe init_waitqueue_head(&hash->wait); 9956ed29b0b4SJens Axboe ctx->hash_map = hash; 9957ed29b0b4SJens Axboe } 9958ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 9959ed29b0b4SJens Axboe 9960ed29b0b4SJens Axboe data.hash = hash; 9961ed29b0b4SJens Axboe data.task = task; 9962ed29b0b4SJens Axboe data.free_work = io_wq_free_work; 9963ed29b0b4SJens Axboe data.do_work = io_wq_submit_work; 9964ed29b0b4SJens Axboe 9965ed29b0b4SJens Axboe /* Do QD, or 4 * CPUS, whatever is smallest */ 9966ed29b0b4SJens Axboe concurrency = min(ctx->sq_entries, 4 * num_online_cpus()); 9967ed29b0b4SJens Axboe 9968ed29b0b4SJens Axboe return io_wq_create(concurrency, &data); 9969ed29b0b4SJens Axboe } 9970ed29b0b4SJens Axboe 9971ed29b0b4SJens Axboe static __cold int io_uring_alloc_task_context(struct task_struct *task, 9972ed29b0b4SJens Axboe struct io_ring_ctx *ctx) 9973ed29b0b4SJens Axboe { 9974ed29b0b4SJens Axboe struct io_uring_task *tctx; 9975ed29b0b4SJens Axboe int ret; 9976ed29b0b4SJens Axboe 9977ed29b0b4SJens Axboe tctx = kzalloc(sizeof(*tctx), GFP_KERNEL); 9978ed29b0b4SJens Axboe if (unlikely(!tctx)) 9979ed29b0b4SJens Axboe return -ENOMEM; 9980ed29b0b4SJens Axboe 9981ed29b0b4SJens Axboe tctx->registered_rings = kcalloc(IO_RINGFD_REG_MAX, 9982ed29b0b4SJens Axboe sizeof(struct file *), GFP_KERNEL); 9983ed29b0b4SJens Axboe if (unlikely(!tctx->registered_rings)) { 9984ed29b0b4SJens Axboe kfree(tctx); 9985ed29b0b4SJens Axboe return -ENOMEM; 9986ed29b0b4SJens Axboe } 9987ed29b0b4SJens Axboe 9988ed29b0b4SJens Axboe ret = percpu_counter_init(&tctx->inflight, 0, GFP_KERNEL); 9989ed29b0b4SJens Axboe if (unlikely(ret)) { 9990ed29b0b4SJens Axboe kfree(tctx->registered_rings); 9991ed29b0b4SJens Axboe kfree(tctx); 9992ed29b0b4SJens Axboe return ret; 9993ed29b0b4SJens Axboe } 9994ed29b0b4SJens Axboe 9995ed29b0b4SJens Axboe tctx->io_wq = io_init_wq_offload(ctx, task); 9996ed29b0b4SJens Axboe if (IS_ERR(tctx->io_wq)) { 9997ed29b0b4SJens Axboe ret = PTR_ERR(tctx->io_wq); 9998ed29b0b4SJens Axboe percpu_counter_destroy(&tctx->inflight); 9999ed29b0b4SJens Axboe kfree(tctx->registered_rings); 10000ed29b0b4SJens Axboe kfree(tctx); 10001ed29b0b4SJens Axboe return ret; 10002ed29b0b4SJens Axboe } 10003ed29b0b4SJens Axboe 10004ed29b0b4SJens Axboe xa_init(&tctx->xa); 10005ed29b0b4SJens Axboe init_waitqueue_head(&tctx->wait); 10006ed29b0b4SJens Axboe atomic_set(&tctx->in_idle, 0); 10007ed29b0b4SJens Axboe atomic_set(&tctx->inflight_tracked, 0); 10008ed29b0b4SJens Axboe task->io_uring = tctx; 10009ed29b0b4SJens Axboe spin_lock_init(&tctx->task_lock); 10010ed29b0b4SJens Axboe INIT_WQ_LIST(&tctx->task_list); 10011ed29b0b4SJens Axboe INIT_WQ_LIST(&tctx->prio_task_list); 10012ed29b0b4SJens Axboe init_task_work(&tctx->task_work, tctx_task_work); 10013ed29b0b4SJens Axboe return 0; 10014ed29b0b4SJens Axboe } 10015ed29b0b4SJens Axboe 10016ed29b0b4SJens Axboe void __io_uring_free(struct task_struct *tsk) 10017ed29b0b4SJens Axboe { 10018ed29b0b4SJens Axboe struct io_uring_task *tctx = tsk->io_uring; 10019ed29b0b4SJens Axboe 10020ed29b0b4SJens Axboe WARN_ON_ONCE(!xa_empty(&tctx->xa)); 10021ed29b0b4SJens Axboe WARN_ON_ONCE(tctx->io_wq); 10022ed29b0b4SJens Axboe WARN_ON_ONCE(tctx->cached_refs); 10023ed29b0b4SJens Axboe 10024ed29b0b4SJens Axboe kfree(tctx->registered_rings); 10025ed29b0b4SJens Axboe percpu_counter_destroy(&tctx->inflight); 10026ed29b0b4SJens Axboe kfree(tctx); 10027ed29b0b4SJens Axboe tsk->io_uring = NULL; 10028ed29b0b4SJens Axboe } 10029ed29b0b4SJens Axboe 10030ed29b0b4SJens Axboe static __cold int io_sq_offload_create(struct io_ring_ctx *ctx, 10031ed29b0b4SJens Axboe struct io_uring_params *p) 10032ed29b0b4SJens Axboe { 10033ed29b0b4SJens Axboe int ret; 10034ed29b0b4SJens Axboe 10035ed29b0b4SJens Axboe /* Retain compatibility with failing for an invalid attach attempt */ 10036ed29b0b4SJens Axboe if ((ctx->flags & (IORING_SETUP_ATTACH_WQ | IORING_SETUP_SQPOLL)) == 10037ed29b0b4SJens Axboe IORING_SETUP_ATTACH_WQ) { 10038ed29b0b4SJens Axboe struct fd f; 10039ed29b0b4SJens Axboe 10040ed29b0b4SJens Axboe f = fdget(p->wq_fd); 10041ed29b0b4SJens Axboe if (!f.file) 10042ed29b0b4SJens Axboe return -ENXIO; 10043ed29b0b4SJens Axboe if (f.file->f_op != &io_uring_fops) { 10044ed29b0b4SJens Axboe fdput(f); 10045ed29b0b4SJens Axboe return -EINVAL; 10046ed29b0b4SJens Axboe } 10047ed29b0b4SJens Axboe fdput(f); 10048ed29b0b4SJens Axboe } 10049ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQPOLL) { 10050ed29b0b4SJens Axboe struct task_struct *tsk; 10051ed29b0b4SJens Axboe struct io_sq_data *sqd; 10052ed29b0b4SJens Axboe bool attached; 10053ed29b0b4SJens Axboe 10054ed29b0b4SJens Axboe ret = security_uring_sqpoll(); 10055ed29b0b4SJens Axboe if (ret) 10056ed29b0b4SJens Axboe return ret; 10057ed29b0b4SJens Axboe 10058ed29b0b4SJens Axboe sqd = io_get_sq_data(p, &attached); 10059ed29b0b4SJens Axboe if (IS_ERR(sqd)) { 10060ed29b0b4SJens Axboe ret = PTR_ERR(sqd); 10061ed29b0b4SJens Axboe goto err; 10062ed29b0b4SJens Axboe } 10063ed29b0b4SJens Axboe 10064ed29b0b4SJens Axboe ctx->sq_creds = get_current_cred(); 10065ed29b0b4SJens Axboe ctx->sq_data = sqd; 10066ed29b0b4SJens Axboe ctx->sq_thread_idle = msecs_to_jiffies(p->sq_thread_idle); 10067ed29b0b4SJens Axboe if (!ctx->sq_thread_idle) 10068ed29b0b4SJens Axboe ctx->sq_thread_idle = HZ; 10069ed29b0b4SJens Axboe 10070ed29b0b4SJens Axboe io_sq_thread_park(sqd); 10071ed29b0b4SJens Axboe list_add(&ctx->sqd_list, &sqd->ctx_list); 10072ed29b0b4SJens Axboe io_sqd_update_thread_idle(sqd); 10073ed29b0b4SJens Axboe /* don't attach to a dying SQPOLL thread, would be racy */ 10074ed29b0b4SJens Axboe ret = (attached && !sqd->thread) ? -ENXIO : 0; 10075ed29b0b4SJens Axboe io_sq_thread_unpark(sqd); 10076ed29b0b4SJens Axboe 10077ed29b0b4SJens Axboe if (ret < 0) 10078ed29b0b4SJens Axboe goto err; 10079ed29b0b4SJens Axboe if (attached) 10080ed29b0b4SJens Axboe return 0; 10081ed29b0b4SJens Axboe 10082ed29b0b4SJens Axboe if (p->flags & IORING_SETUP_SQ_AFF) { 10083ed29b0b4SJens Axboe int cpu = p->sq_thread_cpu; 10084ed29b0b4SJens Axboe 10085ed29b0b4SJens Axboe ret = -EINVAL; 10086ed29b0b4SJens Axboe if (cpu >= nr_cpu_ids || !cpu_online(cpu)) 10087ed29b0b4SJens Axboe goto err_sqpoll; 10088ed29b0b4SJens Axboe sqd->sq_cpu = cpu; 10089ed29b0b4SJens Axboe } else { 10090ed29b0b4SJens Axboe sqd->sq_cpu = -1; 10091ed29b0b4SJens Axboe } 10092ed29b0b4SJens Axboe 10093ed29b0b4SJens Axboe sqd->task_pid = current->pid; 10094ed29b0b4SJens Axboe sqd->task_tgid = current->tgid; 10095ed29b0b4SJens Axboe tsk = create_io_thread(io_sq_thread, sqd, NUMA_NO_NODE); 10096ed29b0b4SJens Axboe if (IS_ERR(tsk)) { 10097ed29b0b4SJens Axboe ret = PTR_ERR(tsk); 10098ed29b0b4SJens Axboe goto err_sqpoll; 10099ed29b0b4SJens Axboe } 10100ed29b0b4SJens Axboe 10101ed29b0b4SJens Axboe sqd->thread = tsk; 10102ed29b0b4SJens Axboe ret = io_uring_alloc_task_context(tsk, ctx); 10103ed29b0b4SJens Axboe wake_up_new_task(tsk); 10104ed29b0b4SJens Axboe if (ret) 10105ed29b0b4SJens Axboe goto err; 10106ed29b0b4SJens Axboe } else if (p->flags & IORING_SETUP_SQ_AFF) { 10107ed29b0b4SJens Axboe /* Can't have SQ_AFF without SQPOLL */ 10108ed29b0b4SJens Axboe ret = -EINVAL; 10109ed29b0b4SJens Axboe goto err; 10110ed29b0b4SJens Axboe } 10111ed29b0b4SJens Axboe 10112ed29b0b4SJens Axboe return 0; 10113ed29b0b4SJens Axboe err_sqpoll: 10114ed29b0b4SJens Axboe complete(&ctx->sq_data->exited); 10115ed29b0b4SJens Axboe err: 10116ed29b0b4SJens Axboe io_sq_thread_finish(ctx); 10117ed29b0b4SJens Axboe return ret; 10118ed29b0b4SJens Axboe } 10119ed29b0b4SJens Axboe 10120ed29b0b4SJens Axboe static inline void __io_unaccount_mem(struct user_struct *user, 10121ed29b0b4SJens Axboe unsigned long nr_pages) 10122ed29b0b4SJens Axboe { 10123ed29b0b4SJens Axboe atomic_long_sub(nr_pages, &user->locked_vm); 10124ed29b0b4SJens Axboe } 10125ed29b0b4SJens Axboe 10126ed29b0b4SJens Axboe static inline int __io_account_mem(struct user_struct *user, 10127ed29b0b4SJens Axboe unsigned long nr_pages) 10128ed29b0b4SJens Axboe { 10129ed29b0b4SJens Axboe unsigned long page_limit, cur_pages, new_pages; 10130ed29b0b4SJens Axboe 10131ed29b0b4SJens Axboe /* Don't allow more pages than we can safely lock */ 10132ed29b0b4SJens Axboe page_limit = rlimit(RLIMIT_MEMLOCK) >> PAGE_SHIFT; 10133ed29b0b4SJens Axboe 10134ed29b0b4SJens Axboe do { 10135ed29b0b4SJens Axboe cur_pages = atomic_long_read(&user->locked_vm); 10136ed29b0b4SJens Axboe new_pages = cur_pages + nr_pages; 10137ed29b0b4SJens Axboe if (new_pages > page_limit) 10138ed29b0b4SJens Axboe return -ENOMEM; 10139ed29b0b4SJens Axboe } while (atomic_long_cmpxchg(&user->locked_vm, cur_pages, 10140ed29b0b4SJens Axboe new_pages) != cur_pages); 10141ed29b0b4SJens Axboe 10142ed29b0b4SJens Axboe return 0; 10143ed29b0b4SJens Axboe } 10144ed29b0b4SJens Axboe 10145ed29b0b4SJens Axboe static void io_unaccount_mem(struct io_ring_ctx *ctx, unsigned long nr_pages) 10146ed29b0b4SJens Axboe { 10147ed29b0b4SJens Axboe if (ctx->user) 10148ed29b0b4SJens Axboe __io_unaccount_mem(ctx->user, nr_pages); 10149ed29b0b4SJens Axboe 10150ed29b0b4SJens Axboe if (ctx->mm_account) 10151ed29b0b4SJens Axboe atomic64_sub(nr_pages, &ctx->mm_account->pinned_vm); 10152ed29b0b4SJens Axboe } 10153ed29b0b4SJens Axboe 10154ed29b0b4SJens Axboe static int io_account_mem(struct io_ring_ctx *ctx, unsigned long nr_pages) 10155ed29b0b4SJens Axboe { 10156ed29b0b4SJens Axboe int ret; 10157ed29b0b4SJens Axboe 10158ed29b0b4SJens Axboe if (ctx->user) { 10159ed29b0b4SJens Axboe ret = __io_account_mem(ctx->user, nr_pages); 10160ed29b0b4SJens Axboe if (ret) 10161ed29b0b4SJens Axboe return ret; 10162ed29b0b4SJens Axboe } 10163ed29b0b4SJens Axboe 10164ed29b0b4SJens Axboe if (ctx->mm_account) 10165ed29b0b4SJens Axboe atomic64_add(nr_pages, &ctx->mm_account->pinned_vm); 10166ed29b0b4SJens Axboe 10167ed29b0b4SJens Axboe return 0; 10168ed29b0b4SJens Axboe } 10169ed29b0b4SJens Axboe 10170ed29b0b4SJens Axboe static void io_mem_free(void *ptr) 10171ed29b0b4SJens Axboe { 10172ed29b0b4SJens Axboe struct page *page; 10173ed29b0b4SJens Axboe 10174ed29b0b4SJens Axboe if (!ptr) 10175ed29b0b4SJens Axboe return; 10176ed29b0b4SJens Axboe 10177ed29b0b4SJens Axboe page = virt_to_head_page(ptr); 10178ed29b0b4SJens Axboe if (put_page_testzero(page)) 10179ed29b0b4SJens Axboe free_compound_page(page); 10180ed29b0b4SJens Axboe } 10181ed29b0b4SJens Axboe 10182ed29b0b4SJens Axboe static void *io_mem_alloc(size_t size) 10183ed29b0b4SJens Axboe { 10184ed29b0b4SJens Axboe gfp_t gfp = GFP_KERNEL_ACCOUNT | __GFP_ZERO | __GFP_NOWARN | __GFP_COMP; 10185ed29b0b4SJens Axboe 10186ed29b0b4SJens Axboe return (void *) __get_free_pages(gfp, get_order(size)); 10187ed29b0b4SJens Axboe } 10188ed29b0b4SJens Axboe 10189ed29b0b4SJens Axboe static unsigned long rings_size(struct io_ring_ctx *ctx, unsigned int sq_entries, 10190ed29b0b4SJens Axboe unsigned int cq_entries, size_t *sq_offset) 10191ed29b0b4SJens Axboe { 10192ed29b0b4SJens Axboe struct io_rings *rings; 10193ed29b0b4SJens Axboe size_t off, sq_array_size; 10194ed29b0b4SJens Axboe 10195ed29b0b4SJens Axboe off = struct_size(rings, cqes, cq_entries); 10196ed29b0b4SJens Axboe if (off == SIZE_MAX) 10197ed29b0b4SJens Axboe return SIZE_MAX; 10198ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_CQE32) { 10199ed29b0b4SJens Axboe if (check_shl_overflow(off, 1, &off)) 10200ed29b0b4SJens Axboe return SIZE_MAX; 10201ed29b0b4SJens Axboe } 10202ed29b0b4SJens Axboe 10203ed29b0b4SJens Axboe #ifdef CONFIG_SMP 10204ed29b0b4SJens Axboe off = ALIGN(off, SMP_CACHE_BYTES); 10205ed29b0b4SJens Axboe if (off == 0) 10206ed29b0b4SJens Axboe return SIZE_MAX; 10207ed29b0b4SJens Axboe #endif 10208ed29b0b4SJens Axboe 10209ed29b0b4SJens Axboe if (sq_offset) 10210ed29b0b4SJens Axboe *sq_offset = off; 10211ed29b0b4SJens Axboe 10212ed29b0b4SJens Axboe sq_array_size = array_size(sizeof(u32), sq_entries); 10213ed29b0b4SJens Axboe if (sq_array_size == SIZE_MAX) 10214ed29b0b4SJens Axboe return SIZE_MAX; 10215ed29b0b4SJens Axboe 10216ed29b0b4SJens Axboe if (check_add_overflow(off, sq_array_size, &off)) 10217ed29b0b4SJens Axboe return SIZE_MAX; 10218ed29b0b4SJens Axboe 10219ed29b0b4SJens Axboe return off; 10220ed29b0b4SJens Axboe } 10221ed29b0b4SJens Axboe 10222ed29b0b4SJens Axboe static void io_buffer_unmap(struct io_ring_ctx *ctx, struct io_mapped_ubuf **slot) 10223ed29b0b4SJens Axboe { 10224ed29b0b4SJens Axboe struct io_mapped_ubuf *imu = *slot; 10225ed29b0b4SJens Axboe unsigned int i; 10226ed29b0b4SJens Axboe 10227ed29b0b4SJens Axboe if (imu != ctx->dummy_ubuf) { 10228ed29b0b4SJens Axboe for (i = 0; i < imu->nr_bvecs; i++) 10229ed29b0b4SJens Axboe unpin_user_page(imu->bvec[i].bv_page); 10230ed29b0b4SJens Axboe if (imu->acct_pages) 10231ed29b0b4SJens Axboe io_unaccount_mem(ctx, imu->acct_pages); 10232ed29b0b4SJens Axboe kvfree(imu); 10233ed29b0b4SJens Axboe } 10234ed29b0b4SJens Axboe *slot = NULL; 10235ed29b0b4SJens Axboe } 10236ed29b0b4SJens Axboe 10237ed29b0b4SJens Axboe static void io_rsrc_buf_put(struct io_ring_ctx *ctx, struct io_rsrc_put *prsrc) 10238ed29b0b4SJens Axboe { 10239ed29b0b4SJens Axboe io_buffer_unmap(ctx, &prsrc->buf); 10240ed29b0b4SJens Axboe prsrc->buf = NULL; 10241ed29b0b4SJens Axboe } 10242ed29b0b4SJens Axboe 10243ed29b0b4SJens Axboe static void __io_sqe_buffers_unregister(struct io_ring_ctx *ctx) 10244ed29b0b4SJens Axboe { 10245ed29b0b4SJens Axboe unsigned int i; 10246ed29b0b4SJens Axboe 10247ed29b0b4SJens Axboe for (i = 0; i < ctx->nr_user_bufs; i++) 10248ed29b0b4SJens Axboe io_buffer_unmap(ctx, &ctx->user_bufs[i]); 10249ed29b0b4SJens Axboe kfree(ctx->user_bufs); 10250ed29b0b4SJens Axboe io_rsrc_data_free(ctx->buf_data); 10251ed29b0b4SJens Axboe ctx->user_bufs = NULL; 10252ed29b0b4SJens Axboe ctx->buf_data = NULL; 10253ed29b0b4SJens Axboe ctx->nr_user_bufs = 0; 10254ed29b0b4SJens Axboe } 10255ed29b0b4SJens Axboe 10256ed29b0b4SJens Axboe static int io_sqe_buffers_unregister(struct io_ring_ctx *ctx) 10257ed29b0b4SJens Axboe { 10258ed29b0b4SJens Axboe unsigned nr = ctx->nr_user_bufs; 10259ed29b0b4SJens Axboe int ret; 10260ed29b0b4SJens Axboe 10261ed29b0b4SJens Axboe if (!ctx->buf_data) 10262ed29b0b4SJens Axboe return -ENXIO; 10263ed29b0b4SJens Axboe 10264ed29b0b4SJens Axboe /* 10265ed29b0b4SJens Axboe * Quiesce may unlock ->uring_lock, and while it's not held 10266ed29b0b4SJens Axboe * prevent new requests using the table. 10267ed29b0b4SJens Axboe */ 10268ed29b0b4SJens Axboe ctx->nr_user_bufs = 0; 10269ed29b0b4SJens Axboe ret = io_rsrc_ref_quiesce(ctx->buf_data, ctx); 10270ed29b0b4SJens Axboe ctx->nr_user_bufs = nr; 10271ed29b0b4SJens Axboe if (!ret) 10272ed29b0b4SJens Axboe __io_sqe_buffers_unregister(ctx); 10273ed29b0b4SJens Axboe return ret; 10274ed29b0b4SJens Axboe } 10275ed29b0b4SJens Axboe 10276ed29b0b4SJens Axboe static int io_copy_iov(struct io_ring_ctx *ctx, struct iovec *dst, 10277ed29b0b4SJens Axboe void __user *arg, unsigned index) 10278ed29b0b4SJens Axboe { 10279ed29b0b4SJens Axboe struct iovec __user *src; 10280ed29b0b4SJens Axboe 10281ed29b0b4SJens Axboe #ifdef CONFIG_COMPAT 10282ed29b0b4SJens Axboe if (ctx->compat) { 10283ed29b0b4SJens Axboe struct compat_iovec __user *ciovs; 10284ed29b0b4SJens Axboe struct compat_iovec ciov; 10285ed29b0b4SJens Axboe 10286ed29b0b4SJens Axboe ciovs = (struct compat_iovec __user *) arg; 10287ed29b0b4SJens Axboe if (copy_from_user(&ciov, &ciovs[index], sizeof(ciov))) 10288ed29b0b4SJens Axboe return -EFAULT; 10289ed29b0b4SJens Axboe 10290ed29b0b4SJens Axboe dst->iov_base = u64_to_user_ptr((u64)ciov.iov_base); 10291ed29b0b4SJens Axboe dst->iov_len = ciov.iov_len; 10292ed29b0b4SJens Axboe return 0; 10293ed29b0b4SJens Axboe } 10294ed29b0b4SJens Axboe #endif 10295ed29b0b4SJens Axboe src = (struct iovec __user *) arg; 10296ed29b0b4SJens Axboe if (copy_from_user(dst, &src[index], sizeof(*dst))) 10297ed29b0b4SJens Axboe return -EFAULT; 10298ed29b0b4SJens Axboe return 0; 10299ed29b0b4SJens Axboe } 10300ed29b0b4SJens Axboe 10301ed29b0b4SJens Axboe /* 10302ed29b0b4SJens Axboe * Not super efficient, but this is just a registration time. And we do cache 10303ed29b0b4SJens Axboe * the last compound head, so generally we'll only do a full search if we don't 10304ed29b0b4SJens Axboe * match that one. 10305ed29b0b4SJens Axboe * 10306ed29b0b4SJens Axboe * We check if the given compound head page has already been accounted, to 10307ed29b0b4SJens Axboe * avoid double accounting it. This allows us to account the full size of the 10308ed29b0b4SJens Axboe * page, not just the constituent pages of a huge page. 10309ed29b0b4SJens Axboe */ 10310ed29b0b4SJens Axboe static bool headpage_already_acct(struct io_ring_ctx *ctx, struct page **pages, 10311ed29b0b4SJens Axboe int nr_pages, struct page *hpage) 10312ed29b0b4SJens Axboe { 10313ed29b0b4SJens Axboe int i, j; 10314ed29b0b4SJens Axboe 10315ed29b0b4SJens Axboe /* check current page array */ 10316ed29b0b4SJens Axboe for (i = 0; i < nr_pages; i++) { 10317ed29b0b4SJens Axboe if (!PageCompound(pages[i])) 10318ed29b0b4SJens Axboe continue; 10319ed29b0b4SJens Axboe if (compound_head(pages[i]) == hpage) 10320ed29b0b4SJens Axboe return true; 10321ed29b0b4SJens Axboe } 10322ed29b0b4SJens Axboe 10323ed29b0b4SJens Axboe /* check previously registered pages */ 10324ed29b0b4SJens Axboe for (i = 0; i < ctx->nr_user_bufs; i++) { 10325ed29b0b4SJens Axboe struct io_mapped_ubuf *imu = ctx->user_bufs[i]; 10326ed29b0b4SJens Axboe 10327ed29b0b4SJens Axboe for (j = 0; j < imu->nr_bvecs; j++) { 10328ed29b0b4SJens Axboe if (!PageCompound(imu->bvec[j].bv_page)) 10329ed29b0b4SJens Axboe continue; 10330ed29b0b4SJens Axboe if (compound_head(imu->bvec[j].bv_page) == hpage) 10331ed29b0b4SJens Axboe return true; 10332ed29b0b4SJens Axboe } 10333ed29b0b4SJens Axboe } 10334ed29b0b4SJens Axboe 10335ed29b0b4SJens Axboe return false; 10336ed29b0b4SJens Axboe } 10337ed29b0b4SJens Axboe 10338ed29b0b4SJens Axboe static int io_buffer_account_pin(struct io_ring_ctx *ctx, struct page **pages, 10339ed29b0b4SJens Axboe int nr_pages, struct io_mapped_ubuf *imu, 10340ed29b0b4SJens Axboe struct page **last_hpage) 10341ed29b0b4SJens Axboe { 10342ed29b0b4SJens Axboe int i, ret; 10343ed29b0b4SJens Axboe 10344ed29b0b4SJens Axboe imu->acct_pages = 0; 10345ed29b0b4SJens Axboe for (i = 0; i < nr_pages; i++) { 10346ed29b0b4SJens Axboe if (!PageCompound(pages[i])) { 10347ed29b0b4SJens Axboe imu->acct_pages++; 10348ed29b0b4SJens Axboe } else { 10349ed29b0b4SJens Axboe struct page *hpage; 10350ed29b0b4SJens Axboe 10351ed29b0b4SJens Axboe hpage = compound_head(pages[i]); 10352ed29b0b4SJens Axboe if (hpage == *last_hpage) 10353ed29b0b4SJens Axboe continue; 10354ed29b0b4SJens Axboe *last_hpage = hpage; 10355ed29b0b4SJens Axboe if (headpage_already_acct(ctx, pages, i, hpage)) 10356ed29b0b4SJens Axboe continue; 10357ed29b0b4SJens Axboe imu->acct_pages += page_size(hpage) >> PAGE_SHIFT; 10358ed29b0b4SJens Axboe } 10359ed29b0b4SJens Axboe } 10360ed29b0b4SJens Axboe 10361ed29b0b4SJens Axboe if (!imu->acct_pages) 10362ed29b0b4SJens Axboe return 0; 10363ed29b0b4SJens Axboe 10364ed29b0b4SJens Axboe ret = io_account_mem(ctx, imu->acct_pages); 10365ed29b0b4SJens Axboe if (ret) 10366ed29b0b4SJens Axboe imu->acct_pages = 0; 10367ed29b0b4SJens Axboe return ret; 10368ed29b0b4SJens Axboe } 10369ed29b0b4SJens Axboe 10370ed29b0b4SJens Axboe static struct page **io_pin_pages(unsigned long ubuf, unsigned long len, 10371ed29b0b4SJens Axboe int *npages) 10372ed29b0b4SJens Axboe { 10373ed29b0b4SJens Axboe unsigned long start, end, nr_pages; 10374ed29b0b4SJens Axboe struct vm_area_struct **vmas = NULL; 10375ed29b0b4SJens Axboe struct page **pages = NULL; 10376ed29b0b4SJens Axboe int i, pret, ret = -ENOMEM; 10377ed29b0b4SJens Axboe 10378ed29b0b4SJens Axboe end = (ubuf + len + PAGE_SIZE - 1) >> PAGE_SHIFT; 10379ed29b0b4SJens Axboe start = ubuf >> PAGE_SHIFT; 10380ed29b0b4SJens Axboe nr_pages = end - start; 10381ed29b0b4SJens Axboe 10382ed29b0b4SJens Axboe pages = kvmalloc_array(nr_pages, sizeof(struct page *), GFP_KERNEL); 10383ed29b0b4SJens Axboe if (!pages) 10384ed29b0b4SJens Axboe goto done; 10385ed29b0b4SJens Axboe 10386ed29b0b4SJens Axboe vmas = kvmalloc_array(nr_pages, sizeof(struct vm_area_struct *), 10387ed29b0b4SJens Axboe GFP_KERNEL); 10388ed29b0b4SJens Axboe if (!vmas) 10389ed29b0b4SJens Axboe goto done; 10390ed29b0b4SJens Axboe 10391ed29b0b4SJens Axboe ret = 0; 10392ed29b0b4SJens Axboe mmap_read_lock(current->mm); 10393ed29b0b4SJens Axboe pret = pin_user_pages(ubuf, nr_pages, FOLL_WRITE | FOLL_LONGTERM, 10394ed29b0b4SJens Axboe pages, vmas); 10395ed29b0b4SJens Axboe if (pret == nr_pages) { 10396ed29b0b4SJens Axboe /* don't support file backed memory */ 10397ed29b0b4SJens Axboe for (i = 0; i < nr_pages; i++) { 10398ed29b0b4SJens Axboe struct vm_area_struct *vma = vmas[i]; 10399ed29b0b4SJens Axboe 10400ed29b0b4SJens Axboe if (vma_is_shmem(vma)) 10401ed29b0b4SJens Axboe continue; 10402ed29b0b4SJens Axboe if (vma->vm_file && 10403ed29b0b4SJens Axboe !is_file_hugepages(vma->vm_file)) { 10404ed29b0b4SJens Axboe ret = -EOPNOTSUPP; 10405ed29b0b4SJens Axboe break; 10406ed29b0b4SJens Axboe } 10407ed29b0b4SJens Axboe } 10408ed29b0b4SJens Axboe *npages = nr_pages; 10409ed29b0b4SJens Axboe } else { 10410ed29b0b4SJens Axboe ret = pret < 0 ? pret : -EFAULT; 10411ed29b0b4SJens Axboe } 10412ed29b0b4SJens Axboe mmap_read_unlock(current->mm); 10413ed29b0b4SJens Axboe if (ret) { 10414ed29b0b4SJens Axboe /* 10415ed29b0b4SJens Axboe * if we did partial map, or found file backed vmas, 10416ed29b0b4SJens Axboe * release any pages we did get 10417ed29b0b4SJens Axboe */ 10418ed29b0b4SJens Axboe if (pret > 0) 10419ed29b0b4SJens Axboe unpin_user_pages(pages, pret); 10420ed29b0b4SJens Axboe goto done; 10421ed29b0b4SJens Axboe } 10422ed29b0b4SJens Axboe ret = 0; 10423ed29b0b4SJens Axboe done: 10424ed29b0b4SJens Axboe kvfree(vmas); 10425ed29b0b4SJens Axboe if (ret < 0) { 10426ed29b0b4SJens Axboe kvfree(pages); 10427ed29b0b4SJens Axboe pages = ERR_PTR(ret); 10428ed29b0b4SJens Axboe } 10429ed29b0b4SJens Axboe return pages; 10430ed29b0b4SJens Axboe } 10431ed29b0b4SJens Axboe 10432ed29b0b4SJens Axboe static int io_sqe_buffer_register(struct io_ring_ctx *ctx, struct iovec *iov, 10433ed29b0b4SJens Axboe struct io_mapped_ubuf **pimu, 10434ed29b0b4SJens Axboe struct page **last_hpage) 10435ed29b0b4SJens Axboe { 10436ed29b0b4SJens Axboe struct io_mapped_ubuf *imu = NULL; 10437ed29b0b4SJens Axboe struct page **pages = NULL; 10438ed29b0b4SJens Axboe unsigned long off; 10439ed29b0b4SJens Axboe size_t size; 10440ed29b0b4SJens Axboe int ret, nr_pages, i; 10441ed29b0b4SJens Axboe 10442ed29b0b4SJens Axboe if (!iov->iov_base) { 10443ed29b0b4SJens Axboe *pimu = ctx->dummy_ubuf; 10444ed29b0b4SJens Axboe return 0; 10445ed29b0b4SJens Axboe } 10446ed29b0b4SJens Axboe 10447ed29b0b4SJens Axboe *pimu = NULL; 10448ed29b0b4SJens Axboe ret = -ENOMEM; 10449ed29b0b4SJens Axboe 10450ed29b0b4SJens Axboe pages = io_pin_pages((unsigned long) iov->iov_base, iov->iov_len, 10451ed29b0b4SJens Axboe &nr_pages); 10452ed29b0b4SJens Axboe if (IS_ERR(pages)) { 10453ed29b0b4SJens Axboe ret = PTR_ERR(pages); 10454ed29b0b4SJens Axboe pages = NULL; 10455ed29b0b4SJens Axboe goto done; 10456ed29b0b4SJens Axboe } 10457ed29b0b4SJens Axboe 10458ed29b0b4SJens Axboe imu = kvmalloc(struct_size(imu, bvec, nr_pages), GFP_KERNEL); 10459ed29b0b4SJens Axboe if (!imu) 10460ed29b0b4SJens Axboe goto done; 10461ed29b0b4SJens Axboe 10462ed29b0b4SJens Axboe ret = io_buffer_account_pin(ctx, pages, nr_pages, imu, last_hpage); 10463ed29b0b4SJens Axboe if (ret) { 10464ed29b0b4SJens Axboe unpin_user_pages(pages, nr_pages); 10465ed29b0b4SJens Axboe goto done; 10466ed29b0b4SJens Axboe } 10467ed29b0b4SJens Axboe 10468ed29b0b4SJens Axboe off = (unsigned long) iov->iov_base & ~PAGE_MASK; 10469ed29b0b4SJens Axboe size = iov->iov_len; 10470ed29b0b4SJens Axboe for (i = 0; i < nr_pages; i++) { 10471ed29b0b4SJens Axboe size_t vec_len; 10472ed29b0b4SJens Axboe 10473ed29b0b4SJens Axboe vec_len = min_t(size_t, size, PAGE_SIZE - off); 10474ed29b0b4SJens Axboe imu->bvec[i].bv_page = pages[i]; 10475ed29b0b4SJens Axboe imu->bvec[i].bv_len = vec_len; 10476ed29b0b4SJens Axboe imu->bvec[i].bv_offset = off; 10477ed29b0b4SJens Axboe off = 0; 10478ed29b0b4SJens Axboe size -= vec_len; 10479ed29b0b4SJens Axboe } 10480ed29b0b4SJens Axboe /* store original address for later verification */ 10481ed29b0b4SJens Axboe imu->ubuf = (unsigned long) iov->iov_base; 10482ed29b0b4SJens Axboe imu->ubuf_end = imu->ubuf + iov->iov_len; 10483ed29b0b4SJens Axboe imu->nr_bvecs = nr_pages; 10484ed29b0b4SJens Axboe *pimu = imu; 10485ed29b0b4SJens Axboe ret = 0; 10486ed29b0b4SJens Axboe done: 10487ed29b0b4SJens Axboe if (ret) 10488ed29b0b4SJens Axboe kvfree(imu); 10489ed29b0b4SJens Axboe kvfree(pages); 10490ed29b0b4SJens Axboe return ret; 10491ed29b0b4SJens Axboe } 10492ed29b0b4SJens Axboe 10493ed29b0b4SJens Axboe static int io_buffers_map_alloc(struct io_ring_ctx *ctx, unsigned int nr_args) 10494ed29b0b4SJens Axboe { 10495ed29b0b4SJens Axboe ctx->user_bufs = kcalloc(nr_args, sizeof(*ctx->user_bufs), GFP_KERNEL); 10496ed29b0b4SJens Axboe return ctx->user_bufs ? 0 : -ENOMEM; 10497ed29b0b4SJens Axboe } 10498ed29b0b4SJens Axboe 10499ed29b0b4SJens Axboe static int io_buffer_validate(struct iovec *iov) 10500ed29b0b4SJens Axboe { 10501ed29b0b4SJens Axboe unsigned long tmp, acct_len = iov->iov_len + (PAGE_SIZE - 1); 10502ed29b0b4SJens Axboe 10503ed29b0b4SJens Axboe /* 10504ed29b0b4SJens Axboe * Don't impose further limits on the size and buffer 10505ed29b0b4SJens Axboe * constraints here, we'll -EINVAL later when IO is 10506ed29b0b4SJens Axboe * submitted if they are wrong. 10507ed29b0b4SJens Axboe */ 10508ed29b0b4SJens Axboe if (!iov->iov_base) 10509ed29b0b4SJens Axboe return iov->iov_len ? -EFAULT : 0; 10510ed29b0b4SJens Axboe if (!iov->iov_len) 10511ed29b0b4SJens Axboe return -EFAULT; 10512ed29b0b4SJens Axboe 10513ed29b0b4SJens Axboe /* arbitrary limit, but we need something */ 10514ed29b0b4SJens Axboe if (iov->iov_len > SZ_1G) 10515ed29b0b4SJens Axboe return -EFAULT; 10516ed29b0b4SJens Axboe 10517ed29b0b4SJens Axboe if (check_add_overflow((unsigned long)iov->iov_base, acct_len, &tmp)) 10518ed29b0b4SJens Axboe return -EOVERFLOW; 10519ed29b0b4SJens Axboe 10520ed29b0b4SJens Axboe return 0; 10521ed29b0b4SJens Axboe } 10522ed29b0b4SJens Axboe 10523ed29b0b4SJens Axboe static int io_sqe_buffers_register(struct io_ring_ctx *ctx, void __user *arg, 10524ed29b0b4SJens Axboe unsigned int nr_args, u64 __user *tags) 10525ed29b0b4SJens Axboe { 10526ed29b0b4SJens Axboe struct page *last_hpage = NULL; 10527ed29b0b4SJens Axboe struct io_rsrc_data *data; 10528ed29b0b4SJens Axboe int i, ret; 10529ed29b0b4SJens Axboe struct iovec iov; 10530ed29b0b4SJens Axboe 10531ed29b0b4SJens Axboe if (ctx->user_bufs) 10532ed29b0b4SJens Axboe return -EBUSY; 10533ed29b0b4SJens Axboe if (!nr_args || nr_args > IORING_MAX_REG_BUFFERS) 10534ed29b0b4SJens Axboe return -EINVAL; 10535ed29b0b4SJens Axboe ret = io_rsrc_node_switch_start(ctx); 10536ed29b0b4SJens Axboe if (ret) 10537ed29b0b4SJens Axboe return ret; 10538ed29b0b4SJens Axboe ret = io_rsrc_data_alloc(ctx, io_rsrc_buf_put, tags, nr_args, &data); 10539ed29b0b4SJens Axboe if (ret) 10540ed29b0b4SJens Axboe return ret; 10541ed29b0b4SJens Axboe ret = io_buffers_map_alloc(ctx, nr_args); 10542ed29b0b4SJens Axboe if (ret) { 10543ed29b0b4SJens Axboe io_rsrc_data_free(data); 10544ed29b0b4SJens Axboe return ret; 10545ed29b0b4SJens Axboe } 10546ed29b0b4SJens Axboe 10547ed29b0b4SJens Axboe for (i = 0; i < nr_args; i++, ctx->nr_user_bufs++) { 10548ed29b0b4SJens Axboe if (arg) { 10549ed29b0b4SJens Axboe ret = io_copy_iov(ctx, &iov, arg, i); 10550ed29b0b4SJens Axboe if (ret) 10551ed29b0b4SJens Axboe break; 10552ed29b0b4SJens Axboe ret = io_buffer_validate(&iov); 10553ed29b0b4SJens Axboe if (ret) 10554ed29b0b4SJens Axboe break; 10555ed29b0b4SJens Axboe } else { 10556ed29b0b4SJens Axboe memset(&iov, 0, sizeof(iov)); 10557ed29b0b4SJens Axboe } 10558ed29b0b4SJens Axboe 10559ed29b0b4SJens Axboe if (!iov.iov_base && *io_get_tag_slot(data, i)) { 10560ed29b0b4SJens Axboe ret = -EINVAL; 10561ed29b0b4SJens Axboe break; 10562ed29b0b4SJens Axboe } 10563ed29b0b4SJens Axboe 10564ed29b0b4SJens Axboe ret = io_sqe_buffer_register(ctx, &iov, &ctx->user_bufs[i], 10565ed29b0b4SJens Axboe &last_hpage); 10566ed29b0b4SJens Axboe if (ret) 10567ed29b0b4SJens Axboe break; 10568ed29b0b4SJens Axboe } 10569ed29b0b4SJens Axboe 10570ed29b0b4SJens Axboe WARN_ON_ONCE(ctx->buf_data); 10571ed29b0b4SJens Axboe 10572ed29b0b4SJens Axboe ctx->buf_data = data; 10573ed29b0b4SJens Axboe if (ret) 10574ed29b0b4SJens Axboe __io_sqe_buffers_unregister(ctx); 10575ed29b0b4SJens Axboe else 10576ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, NULL); 10577ed29b0b4SJens Axboe return ret; 10578ed29b0b4SJens Axboe } 10579ed29b0b4SJens Axboe 10580ed29b0b4SJens Axboe static int __io_sqe_buffers_update(struct io_ring_ctx *ctx, 10581ed29b0b4SJens Axboe struct io_uring_rsrc_update2 *up, 10582ed29b0b4SJens Axboe unsigned int nr_args) 10583ed29b0b4SJens Axboe { 10584ed29b0b4SJens Axboe u64 __user *tags = u64_to_user_ptr(up->tags); 10585ed29b0b4SJens Axboe struct iovec iov, __user *iovs = u64_to_user_ptr(up->data); 10586ed29b0b4SJens Axboe struct page *last_hpage = NULL; 10587ed29b0b4SJens Axboe bool needs_switch = false; 10588ed29b0b4SJens Axboe __u32 done; 10589ed29b0b4SJens Axboe int i, err; 10590ed29b0b4SJens Axboe 10591ed29b0b4SJens Axboe if (!ctx->buf_data) 10592ed29b0b4SJens Axboe return -ENXIO; 10593ed29b0b4SJens Axboe if (up->offset + nr_args > ctx->nr_user_bufs) 10594ed29b0b4SJens Axboe return -EINVAL; 10595ed29b0b4SJens Axboe 10596ed29b0b4SJens Axboe for (done = 0; done < nr_args; done++) { 10597ed29b0b4SJens Axboe struct io_mapped_ubuf *imu; 10598ed29b0b4SJens Axboe int offset = up->offset + done; 10599ed29b0b4SJens Axboe u64 tag = 0; 10600ed29b0b4SJens Axboe 10601ed29b0b4SJens Axboe err = io_copy_iov(ctx, &iov, iovs, done); 10602ed29b0b4SJens Axboe if (err) 10603ed29b0b4SJens Axboe break; 10604ed29b0b4SJens Axboe if (tags && copy_from_user(&tag, &tags[done], sizeof(tag))) { 10605ed29b0b4SJens Axboe err = -EFAULT; 10606ed29b0b4SJens Axboe break; 10607ed29b0b4SJens Axboe } 10608ed29b0b4SJens Axboe err = io_buffer_validate(&iov); 10609ed29b0b4SJens Axboe if (err) 10610ed29b0b4SJens Axboe break; 10611ed29b0b4SJens Axboe if (!iov.iov_base && tag) { 10612ed29b0b4SJens Axboe err = -EINVAL; 10613ed29b0b4SJens Axboe break; 10614ed29b0b4SJens Axboe } 10615ed29b0b4SJens Axboe err = io_sqe_buffer_register(ctx, &iov, &imu, &last_hpage); 10616ed29b0b4SJens Axboe if (err) 10617ed29b0b4SJens Axboe break; 10618ed29b0b4SJens Axboe 10619ed29b0b4SJens Axboe i = array_index_nospec(offset, ctx->nr_user_bufs); 10620ed29b0b4SJens Axboe if (ctx->user_bufs[i] != ctx->dummy_ubuf) { 10621ed29b0b4SJens Axboe err = io_queue_rsrc_removal(ctx->buf_data, i, 10622ed29b0b4SJens Axboe ctx->rsrc_node, ctx->user_bufs[i]); 10623ed29b0b4SJens Axboe if (unlikely(err)) { 10624ed29b0b4SJens Axboe io_buffer_unmap(ctx, &imu); 10625ed29b0b4SJens Axboe break; 10626ed29b0b4SJens Axboe } 10627ed29b0b4SJens Axboe ctx->user_bufs[i] = NULL; 10628ed29b0b4SJens Axboe needs_switch = true; 10629ed29b0b4SJens Axboe } 10630ed29b0b4SJens Axboe 10631ed29b0b4SJens Axboe ctx->user_bufs[i] = imu; 10632ed29b0b4SJens Axboe *io_get_tag_slot(ctx->buf_data, offset) = tag; 10633ed29b0b4SJens Axboe } 10634ed29b0b4SJens Axboe 10635ed29b0b4SJens Axboe if (needs_switch) 10636ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, ctx->buf_data); 10637ed29b0b4SJens Axboe return done ? done : err; 10638ed29b0b4SJens Axboe } 10639ed29b0b4SJens Axboe 10640ed29b0b4SJens Axboe static int io_eventfd_register(struct io_ring_ctx *ctx, void __user *arg, 10641ed29b0b4SJens Axboe unsigned int eventfd_async) 10642ed29b0b4SJens Axboe { 10643ed29b0b4SJens Axboe struct io_ev_fd *ev_fd; 10644ed29b0b4SJens Axboe __s32 __user *fds = arg; 10645ed29b0b4SJens Axboe int fd; 10646ed29b0b4SJens Axboe 10647ed29b0b4SJens Axboe ev_fd = rcu_dereference_protected(ctx->io_ev_fd, 10648ed29b0b4SJens Axboe lockdep_is_held(&ctx->uring_lock)); 10649ed29b0b4SJens Axboe if (ev_fd) 10650ed29b0b4SJens Axboe return -EBUSY; 10651ed29b0b4SJens Axboe 10652ed29b0b4SJens Axboe if (copy_from_user(&fd, fds, sizeof(*fds))) 10653ed29b0b4SJens Axboe return -EFAULT; 10654ed29b0b4SJens Axboe 10655ed29b0b4SJens Axboe ev_fd = kmalloc(sizeof(*ev_fd), GFP_KERNEL); 10656ed29b0b4SJens Axboe if (!ev_fd) 10657ed29b0b4SJens Axboe return -ENOMEM; 10658ed29b0b4SJens Axboe 10659ed29b0b4SJens Axboe ev_fd->cq_ev_fd = eventfd_ctx_fdget(fd); 10660ed29b0b4SJens Axboe if (IS_ERR(ev_fd->cq_ev_fd)) { 10661ed29b0b4SJens Axboe int ret = PTR_ERR(ev_fd->cq_ev_fd); 10662ed29b0b4SJens Axboe kfree(ev_fd); 10663ed29b0b4SJens Axboe return ret; 10664ed29b0b4SJens Axboe } 10665ed29b0b4SJens Axboe ev_fd->eventfd_async = eventfd_async; 10666ed29b0b4SJens Axboe ctx->has_evfd = true; 10667ed29b0b4SJens Axboe rcu_assign_pointer(ctx->io_ev_fd, ev_fd); 10668ed29b0b4SJens Axboe return 0; 10669ed29b0b4SJens Axboe } 10670ed29b0b4SJens Axboe 10671ed29b0b4SJens Axboe static void io_eventfd_put(struct rcu_head *rcu) 10672ed29b0b4SJens Axboe { 10673ed29b0b4SJens Axboe struct io_ev_fd *ev_fd = container_of(rcu, struct io_ev_fd, rcu); 10674ed29b0b4SJens Axboe 10675ed29b0b4SJens Axboe eventfd_ctx_put(ev_fd->cq_ev_fd); 10676ed29b0b4SJens Axboe kfree(ev_fd); 10677ed29b0b4SJens Axboe } 10678ed29b0b4SJens Axboe 10679ed29b0b4SJens Axboe static int io_eventfd_unregister(struct io_ring_ctx *ctx) 10680ed29b0b4SJens Axboe { 10681ed29b0b4SJens Axboe struct io_ev_fd *ev_fd; 10682ed29b0b4SJens Axboe 10683ed29b0b4SJens Axboe ev_fd = rcu_dereference_protected(ctx->io_ev_fd, 10684ed29b0b4SJens Axboe lockdep_is_held(&ctx->uring_lock)); 10685ed29b0b4SJens Axboe if (ev_fd) { 10686ed29b0b4SJens Axboe ctx->has_evfd = false; 10687ed29b0b4SJens Axboe rcu_assign_pointer(ctx->io_ev_fd, NULL); 10688ed29b0b4SJens Axboe call_rcu(&ev_fd->rcu, io_eventfd_put); 10689ed29b0b4SJens Axboe return 0; 10690ed29b0b4SJens Axboe } 10691ed29b0b4SJens Axboe 10692ed29b0b4SJens Axboe return -ENXIO; 10693ed29b0b4SJens Axboe } 10694ed29b0b4SJens Axboe 10695ed29b0b4SJens Axboe static void io_destroy_buffers(struct io_ring_ctx *ctx) 10696ed29b0b4SJens Axboe { 10697ed29b0b4SJens Axboe struct io_buffer_list *bl; 10698ed29b0b4SJens Axboe unsigned long index; 10699ed29b0b4SJens Axboe int i; 10700ed29b0b4SJens Axboe 10701ed29b0b4SJens Axboe for (i = 0; i < BGID_ARRAY; i++) { 10702ed29b0b4SJens Axboe if (!ctx->io_bl) 10703ed29b0b4SJens Axboe break; 10704ed29b0b4SJens Axboe __io_remove_buffers(ctx, &ctx->io_bl[i], -1U); 10705ed29b0b4SJens Axboe } 10706ed29b0b4SJens Axboe 10707ed29b0b4SJens Axboe xa_for_each(&ctx->io_bl_xa, index, bl) { 10708ed29b0b4SJens Axboe xa_erase(&ctx->io_bl_xa, bl->bgid); 10709ed29b0b4SJens Axboe __io_remove_buffers(ctx, bl, -1U); 10710ed29b0b4SJens Axboe kfree(bl); 10711ed29b0b4SJens Axboe } 10712ed29b0b4SJens Axboe 10713ed29b0b4SJens Axboe while (!list_empty(&ctx->io_buffers_pages)) { 10714ed29b0b4SJens Axboe struct page *page; 10715ed29b0b4SJens Axboe 10716ed29b0b4SJens Axboe page = list_first_entry(&ctx->io_buffers_pages, struct page, lru); 10717ed29b0b4SJens Axboe list_del_init(&page->lru); 10718ed29b0b4SJens Axboe __free_page(page); 10719ed29b0b4SJens Axboe } 10720ed29b0b4SJens Axboe } 10721ed29b0b4SJens Axboe 10722ed29b0b4SJens Axboe static void io_req_caches_free(struct io_ring_ctx *ctx) 10723ed29b0b4SJens Axboe { 10724ed29b0b4SJens Axboe struct io_submit_state *state = &ctx->submit_state; 10725ed29b0b4SJens Axboe int nr = 0; 10726ed29b0b4SJens Axboe 10727ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 10728ed29b0b4SJens Axboe io_flush_cached_locked_reqs(ctx, state); 10729ed29b0b4SJens Axboe 10730ed29b0b4SJens Axboe while (!io_req_cache_empty(ctx)) { 10731ed29b0b4SJens Axboe struct io_wq_work_node *node; 10732ed29b0b4SJens Axboe struct io_kiocb *req; 10733ed29b0b4SJens Axboe 10734ed29b0b4SJens Axboe node = wq_stack_extract(&state->free_list); 10735ed29b0b4SJens Axboe req = container_of(node, struct io_kiocb, comp_list); 10736ed29b0b4SJens Axboe kmem_cache_free(req_cachep, req); 10737ed29b0b4SJens Axboe nr++; 10738ed29b0b4SJens Axboe } 10739ed29b0b4SJens Axboe if (nr) 10740ed29b0b4SJens Axboe percpu_ref_put_many(&ctx->refs, nr); 10741ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 10742ed29b0b4SJens Axboe } 10743ed29b0b4SJens Axboe 10744ed29b0b4SJens Axboe static void io_wait_rsrc_data(struct io_rsrc_data *data) 10745ed29b0b4SJens Axboe { 10746ed29b0b4SJens Axboe if (data && !atomic_dec_and_test(&data->refs)) 10747ed29b0b4SJens Axboe wait_for_completion(&data->done); 10748ed29b0b4SJens Axboe } 10749ed29b0b4SJens Axboe 10750ed29b0b4SJens Axboe static void io_flush_apoll_cache(struct io_ring_ctx *ctx) 10751ed29b0b4SJens Axboe { 10752ed29b0b4SJens Axboe struct async_poll *apoll; 10753ed29b0b4SJens Axboe 10754ed29b0b4SJens Axboe while (!list_empty(&ctx->apoll_cache)) { 10755ed29b0b4SJens Axboe apoll = list_first_entry(&ctx->apoll_cache, struct async_poll, 10756ed29b0b4SJens Axboe poll.wait.entry); 10757ed29b0b4SJens Axboe list_del(&apoll->poll.wait.entry); 10758ed29b0b4SJens Axboe kfree(apoll); 10759ed29b0b4SJens Axboe } 10760ed29b0b4SJens Axboe } 10761ed29b0b4SJens Axboe 10762ed29b0b4SJens Axboe static __cold void io_ring_ctx_free(struct io_ring_ctx *ctx) 10763ed29b0b4SJens Axboe { 10764ed29b0b4SJens Axboe io_sq_thread_finish(ctx); 10765ed29b0b4SJens Axboe 10766ed29b0b4SJens Axboe if (ctx->mm_account) { 10767ed29b0b4SJens Axboe mmdrop(ctx->mm_account); 10768ed29b0b4SJens Axboe ctx->mm_account = NULL; 10769ed29b0b4SJens Axboe } 10770ed29b0b4SJens Axboe 10771ed29b0b4SJens Axboe io_rsrc_refs_drop(ctx); 10772ed29b0b4SJens Axboe /* __io_rsrc_put_work() may need uring_lock to progress, wait w/o it */ 10773ed29b0b4SJens Axboe io_wait_rsrc_data(ctx->buf_data); 10774ed29b0b4SJens Axboe io_wait_rsrc_data(ctx->file_data); 10775ed29b0b4SJens Axboe 10776ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 10777ed29b0b4SJens Axboe if (ctx->buf_data) 10778ed29b0b4SJens Axboe __io_sqe_buffers_unregister(ctx); 10779ed29b0b4SJens Axboe if (ctx->file_data) 10780ed29b0b4SJens Axboe __io_sqe_files_unregister(ctx); 10781ed29b0b4SJens Axboe if (ctx->rings) 10782ed29b0b4SJens Axboe __io_cqring_overflow_flush(ctx, true); 10783ed29b0b4SJens Axboe io_eventfd_unregister(ctx); 10784ed29b0b4SJens Axboe io_flush_apoll_cache(ctx); 10785ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 10786ed29b0b4SJens Axboe io_destroy_buffers(ctx); 10787ed29b0b4SJens Axboe if (ctx->sq_creds) 10788ed29b0b4SJens Axboe put_cred(ctx->sq_creds); 10789ed29b0b4SJens Axboe 10790ed29b0b4SJens Axboe /* there are no registered resources left, nobody uses it */ 10791ed29b0b4SJens Axboe if (ctx->rsrc_node) 10792ed29b0b4SJens Axboe io_rsrc_node_destroy(ctx->rsrc_node); 10793ed29b0b4SJens Axboe if (ctx->rsrc_backup_node) 10794ed29b0b4SJens Axboe io_rsrc_node_destroy(ctx->rsrc_backup_node); 10795ed29b0b4SJens Axboe flush_delayed_work(&ctx->rsrc_put_work); 10796ed29b0b4SJens Axboe flush_delayed_work(&ctx->fallback_work); 10797ed29b0b4SJens Axboe 10798ed29b0b4SJens Axboe WARN_ON_ONCE(!list_empty(&ctx->rsrc_ref_list)); 10799ed29b0b4SJens Axboe WARN_ON_ONCE(!llist_empty(&ctx->rsrc_put_llist)); 10800ed29b0b4SJens Axboe 10801ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 10802ed29b0b4SJens Axboe if (ctx->ring_sock) { 10803ed29b0b4SJens Axboe ctx->ring_sock->file = NULL; /* so that iput() is called */ 10804ed29b0b4SJens Axboe sock_release(ctx->ring_sock); 10805ed29b0b4SJens Axboe } 10806ed29b0b4SJens Axboe #endif 10807ed29b0b4SJens Axboe WARN_ON_ONCE(!list_empty(&ctx->ltimeout_list)); 10808ed29b0b4SJens Axboe 10809ed29b0b4SJens Axboe io_mem_free(ctx->rings); 10810ed29b0b4SJens Axboe io_mem_free(ctx->sq_sqes); 10811ed29b0b4SJens Axboe 10812ed29b0b4SJens Axboe percpu_ref_exit(&ctx->refs); 10813ed29b0b4SJens Axboe free_uid(ctx->user); 10814ed29b0b4SJens Axboe io_req_caches_free(ctx); 10815ed29b0b4SJens Axboe if (ctx->hash_map) 10816ed29b0b4SJens Axboe io_wq_put_hash(ctx->hash_map); 10817ed29b0b4SJens Axboe kfree(ctx->cancel_hash); 10818ed29b0b4SJens Axboe kfree(ctx->dummy_ubuf); 10819ed29b0b4SJens Axboe kfree(ctx->io_bl); 10820ed29b0b4SJens Axboe xa_destroy(&ctx->io_bl_xa); 10821ed29b0b4SJens Axboe kfree(ctx); 10822ed29b0b4SJens Axboe } 10823ed29b0b4SJens Axboe 10824ed29b0b4SJens Axboe static __poll_t io_uring_poll(struct file *file, poll_table *wait) 10825ed29b0b4SJens Axboe { 10826ed29b0b4SJens Axboe struct io_ring_ctx *ctx = file->private_data; 10827ed29b0b4SJens Axboe __poll_t mask = 0; 10828ed29b0b4SJens Axboe 10829ed29b0b4SJens Axboe poll_wait(file, &ctx->cq_wait, wait); 10830ed29b0b4SJens Axboe /* 10831ed29b0b4SJens Axboe * synchronizes with barrier from wq_has_sleeper call in 10832ed29b0b4SJens Axboe * io_commit_cqring 10833ed29b0b4SJens Axboe */ 10834ed29b0b4SJens Axboe smp_rmb(); 10835ed29b0b4SJens Axboe if (!io_sqring_full(ctx)) 10836ed29b0b4SJens Axboe mask |= EPOLLOUT | EPOLLWRNORM; 10837ed29b0b4SJens Axboe 10838ed29b0b4SJens Axboe /* 10839ed29b0b4SJens Axboe * Don't flush cqring overflow list here, just do a simple check. 10840ed29b0b4SJens Axboe * Otherwise there could possible be ABBA deadlock: 10841ed29b0b4SJens Axboe * CPU0 CPU1 10842ed29b0b4SJens Axboe * ---- ---- 10843ed29b0b4SJens Axboe * lock(&ctx->uring_lock); 10844ed29b0b4SJens Axboe * lock(&ep->mtx); 10845ed29b0b4SJens Axboe * lock(&ctx->uring_lock); 10846ed29b0b4SJens Axboe * lock(&ep->mtx); 10847ed29b0b4SJens Axboe * 10848ed29b0b4SJens Axboe * Users may get EPOLLIN meanwhile seeing nothing in cqring, this 10849ed29b0b4SJens Axboe * pushs them to do the flush. 10850ed29b0b4SJens Axboe */ 10851ed29b0b4SJens Axboe if (io_cqring_events(ctx) || 10852ed29b0b4SJens Axboe test_bit(IO_CHECK_CQ_OVERFLOW_BIT, &ctx->check_cq)) 10853ed29b0b4SJens Axboe mask |= EPOLLIN | EPOLLRDNORM; 10854ed29b0b4SJens Axboe 10855ed29b0b4SJens Axboe return mask; 10856ed29b0b4SJens Axboe } 10857ed29b0b4SJens Axboe 10858ed29b0b4SJens Axboe static int io_unregister_personality(struct io_ring_ctx *ctx, unsigned id) 10859ed29b0b4SJens Axboe { 10860ed29b0b4SJens Axboe const struct cred *creds; 10861ed29b0b4SJens Axboe 10862ed29b0b4SJens Axboe creds = xa_erase(&ctx->personalities, id); 10863ed29b0b4SJens Axboe if (creds) { 10864ed29b0b4SJens Axboe put_cred(creds); 10865ed29b0b4SJens Axboe return 0; 10866ed29b0b4SJens Axboe } 10867ed29b0b4SJens Axboe 10868ed29b0b4SJens Axboe return -EINVAL; 10869ed29b0b4SJens Axboe } 10870ed29b0b4SJens Axboe 10871ed29b0b4SJens Axboe struct io_tctx_exit { 10872ed29b0b4SJens Axboe struct callback_head task_work; 10873ed29b0b4SJens Axboe struct completion completion; 10874ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 10875ed29b0b4SJens Axboe }; 10876ed29b0b4SJens Axboe 10877ed29b0b4SJens Axboe static __cold void io_tctx_exit_cb(struct callback_head *cb) 10878ed29b0b4SJens Axboe { 10879ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 10880ed29b0b4SJens Axboe struct io_tctx_exit *work; 10881ed29b0b4SJens Axboe 10882ed29b0b4SJens Axboe work = container_of(cb, struct io_tctx_exit, task_work); 10883ed29b0b4SJens Axboe /* 10884ed29b0b4SJens Axboe * When @in_idle, we're in cancellation and it's racy to remove the 10885ed29b0b4SJens Axboe * node. It'll be removed by the end of cancellation, just ignore it. 10886ed29b0b4SJens Axboe */ 10887ed29b0b4SJens Axboe if (!atomic_read(&tctx->in_idle)) 10888ed29b0b4SJens Axboe io_uring_del_tctx_node((unsigned long)work->ctx); 10889ed29b0b4SJens Axboe complete(&work->completion); 10890ed29b0b4SJens Axboe } 10891ed29b0b4SJens Axboe 10892ed29b0b4SJens Axboe static __cold bool io_cancel_ctx_cb(struct io_wq_work *work, void *data) 10893ed29b0b4SJens Axboe { 10894ed29b0b4SJens Axboe struct io_kiocb *req = container_of(work, struct io_kiocb, work); 10895ed29b0b4SJens Axboe 10896ed29b0b4SJens Axboe return req->ctx == data; 10897ed29b0b4SJens Axboe } 10898ed29b0b4SJens Axboe 10899ed29b0b4SJens Axboe static __cold void io_ring_exit_work(struct work_struct *work) 10900ed29b0b4SJens Axboe { 10901ed29b0b4SJens Axboe struct io_ring_ctx *ctx = container_of(work, struct io_ring_ctx, exit_work); 10902ed29b0b4SJens Axboe unsigned long timeout = jiffies + HZ * 60 * 5; 10903ed29b0b4SJens Axboe unsigned long interval = HZ / 20; 10904ed29b0b4SJens Axboe struct io_tctx_exit exit; 10905ed29b0b4SJens Axboe struct io_tctx_node *node; 10906ed29b0b4SJens Axboe int ret; 10907ed29b0b4SJens Axboe 10908ed29b0b4SJens Axboe /* 10909ed29b0b4SJens Axboe * If we're doing polled IO and end up having requests being 10910ed29b0b4SJens Axboe * submitted async (out-of-line), then completions can come in while 10911ed29b0b4SJens Axboe * we're waiting for refs to drop. We need to reap these manually, 10912ed29b0b4SJens Axboe * as nobody else will be looking for them. 10913ed29b0b4SJens Axboe */ 10914ed29b0b4SJens Axboe do { 10915ed29b0b4SJens Axboe io_uring_try_cancel_requests(ctx, NULL, true); 10916ed29b0b4SJens Axboe if (ctx->sq_data) { 10917ed29b0b4SJens Axboe struct io_sq_data *sqd = ctx->sq_data; 10918ed29b0b4SJens Axboe struct task_struct *tsk; 10919ed29b0b4SJens Axboe 10920ed29b0b4SJens Axboe io_sq_thread_park(sqd); 10921ed29b0b4SJens Axboe tsk = sqd->thread; 10922ed29b0b4SJens Axboe if (tsk && tsk->io_uring && tsk->io_uring->io_wq) 10923ed29b0b4SJens Axboe io_wq_cancel_cb(tsk->io_uring->io_wq, 10924ed29b0b4SJens Axboe io_cancel_ctx_cb, ctx, true); 10925ed29b0b4SJens Axboe io_sq_thread_unpark(sqd); 10926ed29b0b4SJens Axboe } 10927ed29b0b4SJens Axboe 10928ed29b0b4SJens Axboe io_req_caches_free(ctx); 10929ed29b0b4SJens Axboe 10930ed29b0b4SJens Axboe if (WARN_ON_ONCE(time_after(jiffies, timeout))) { 10931ed29b0b4SJens Axboe /* there is little hope left, don't run it too often */ 10932ed29b0b4SJens Axboe interval = HZ * 60; 10933ed29b0b4SJens Axboe } 10934ed29b0b4SJens Axboe } while (!wait_for_completion_timeout(&ctx->ref_comp, interval)); 10935ed29b0b4SJens Axboe 10936ed29b0b4SJens Axboe init_completion(&exit.completion); 10937ed29b0b4SJens Axboe init_task_work(&exit.task_work, io_tctx_exit_cb); 10938ed29b0b4SJens Axboe exit.ctx = ctx; 10939ed29b0b4SJens Axboe /* 10940ed29b0b4SJens Axboe * Some may use context even when all refs and requests have been put, 10941ed29b0b4SJens Axboe * and they are free to do so while still holding uring_lock or 10942ed29b0b4SJens Axboe * completion_lock, see io_req_task_submit(). Apart from other work, 10943ed29b0b4SJens Axboe * this lock/unlock section also waits them to finish. 10944ed29b0b4SJens Axboe */ 10945ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 10946ed29b0b4SJens Axboe while (!list_empty(&ctx->tctx_list)) { 10947ed29b0b4SJens Axboe WARN_ON_ONCE(time_after(jiffies, timeout)); 10948ed29b0b4SJens Axboe 10949ed29b0b4SJens Axboe node = list_first_entry(&ctx->tctx_list, struct io_tctx_node, 10950ed29b0b4SJens Axboe ctx_node); 10951ed29b0b4SJens Axboe /* don't spin on a single task if cancellation failed */ 10952ed29b0b4SJens Axboe list_rotate_left(&ctx->tctx_list); 10953ed29b0b4SJens Axboe ret = task_work_add(node->task, &exit.task_work, TWA_SIGNAL); 10954ed29b0b4SJens Axboe if (WARN_ON_ONCE(ret)) 10955ed29b0b4SJens Axboe continue; 10956ed29b0b4SJens Axboe 10957ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 10958ed29b0b4SJens Axboe wait_for_completion(&exit.completion); 10959ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 10960ed29b0b4SJens Axboe } 10961ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 10962ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 10963ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 10964ed29b0b4SJens Axboe 10965ed29b0b4SJens Axboe io_ring_ctx_free(ctx); 10966ed29b0b4SJens Axboe } 10967ed29b0b4SJens Axboe 10968ed29b0b4SJens Axboe /* Returns true if we found and killed one or more timeouts */ 10969ed29b0b4SJens Axboe static __cold bool io_kill_timeouts(struct io_ring_ctx *ctx, 10970ed29b0b4SJens Axboe struct task_struct *tsk, bool cancel_all) 10971ed29b0b4SJens Axboe { 10972a43714acSJens Axboe struct io_timeout *timeout, *tmp; 10973ed29b0b4SJens Axboe int canceled = 0; 10974ed29b0b4SJens Axboe 10975ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 10976ed29b0b4SJens Axboe spin_lock_irq(&ctx->timeout_lock); 10977a43714acSJens Axboe list_for_each_entry_safe(timeout, tmp, &ctx->timeout_list, list) { 10978a43714acSJens Axboe struct io_kiocb *req = cmd_to_io_kiocb(timeout); 10979a43714acSJens Axboe 10980ed29b0b4SJens Axboe if (io_match_task(req, tsk, cancel_all)) { 10981ed29b0b4SJens Axboe io_kill_timeout(req, -ECANCELED); 10982ed29b0b4SJens Axboe canceled++; 10983ed29b0b4SJens Axboe } 10984ed29b0b4SJens Axboe } 10985ed29b0b4SJens Axboe spin_unlock_irq(&ctx->timeout_lock); 10986ed29b0b4SJens Axboe io_commit_cqring(ctx); 10987ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 10988ed29b0b4SJens Axboe if (canceled != 0) 10989ed29b0b4SJens Axboe io_cqring_ev_posted(ctx); 10990ed29b0b4SJens Axboe return canceled != 0; 10991ed29b0b4SJens Axboe } 10992ed29b0b4SJens Axboe 10993ed29b0b4SJens Axboe static __cold void io_ring_ctx_wait_and_kill(struct io_ring_ctx *ctx) 10994ed29b0b4SJens Axboe { 10995ed29b0b4SJens Axboe unsigned long index; 10996ed29b0b4SJens Axboe struct creds *creds; 10997ed29b0b4SJens Axboe 10998ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 10999ed29b0b4SJens Axboe percpu_ref_kill(&ctx->refs); 11000ed29b0b4SJens Axboe if (ctx->rings) 11001ed29b0b4SJens Axboe __io_cqring_overflow_flush(ctx, true); 11002ed29b0b4SJens Axboe xa_for_each(&ctx->personalities, index, creds) 11003ed29b0b4SJens Axboe io_unregister_personality(ctx, index); 11004ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 11005ed29b0b4SJens Axboe 11006ed29b0b4SJens Axboe /* failed during ring init, it couldn't have issued any requests */ 11007ed29b0b4SJens Axboe if (ctx->rings) { 11008ed29b0b4SJens Axboe io_kill_timeouts(ctx, NULL, true); 11009ed29b0b4SJens Axboe io_poll_remove_all(ctx, NULL, true); 11010ed29b0b4SJens Axboe /* if we failed setting up the ctx, we might not have any rings */ 11011ed29b0b4SJens Axboe io_iopoll_try_reap_events(ctx); 11012ed29b0b4SJens Axboe } 11013ed29b0b4SJens Axboe 11014ed29b0b4SJens Axboe INIT_WORK(&ctx->exit_work, io_ring_exit_work); 11015ed29b0b4SJens Axboe /* 11016ed29b0b4SJens Axboe * Use system_unbound_wq to avoid spawning tons of event kworkers 11017ed29b0b4SJens Axboe * if we're exiting a ton of rings at the same time. It just adds 11018ed29b0b4SJens Axboe * noise and overhead, there's no discernable change in runtime 11019ed29b0b4SJens Axboe * over using system_wq. 11020ed29b0b4SJens Axboe */ 11021ed29b0b4SJens Axboe queue_work(system_unbound_wq, &ctx->exit_work); 11022ed29b0b4SJens Axboe } 11023ed29b0b4SJens Axboe 11024ed29b0b4SJens Axboe static int io_uring_release(struct inode *inode, struct file *file) 11025ed29b0b4SJens Axboe { 11026ed29b0b4SJens Axboe struct io_ring_ctx *ctx = file->private_data; 11027ed29b0b4SJens Axboe 11028ed29b0b4SJens Axboe file->private_data = NULL; 11029ed29b0b4SJens Axboe io_ring_ctx_wait_and_kill(ctx); 11030ed29b0b4SJens Axboe return 0; 11031ed29b0b4SJens Axboe } 11032ed29b0b4SJens Axboe 11033ed29b0b4SJens Axboe struct io_task_cancel { 11034ed29b0b4SJens Axboe struct task_struct *task; 11035ed29b0b4SJens Axboe bool all; 11036ed29b0b4SJens Axboe }; 11037ed29b0b4SJens Axboe 11038ed29b0b4SJens Axboe static bool io_cancel_task_cb(struct io_wq_work *work, void *data) 11039ed29b0b4SJens Axboe { 11040ed29b0b4SJens Axboe struct io_kiocb *req = container_of(work, struct io_kiocb, work); 11041ed29b0b4SJens Axboe struct io_task_cancel *cancel = data; 11042ed29b0b4SJens Axboe 11043ed29b0b4SJens Axboe return io_match_task_safe(req, cancel->task, cancel->all); 11044ed29b0b4SJens Axboe } 11045ed29b0b4SJens Axboe 11046ed29b0b4SJens Axboe static __cold bool io_cancel_defer_files(struct io_ring_ctx *ctx, 11047ed29b0b4SJens Axboe struct task_struct *task, 11048ed29b0b4SJens Axboe bool cancel_all) 11049ed29b0b4SJens Axboe { 11050ed29b0b4SJens Axboe struct io_defer_entry *de; 11051ed29b0b4SJens Axboe LIST_HEAD(list); 11052ed29b0b4SJens Axboe 11053ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 11054ed29b0b4SJens Axboe list_for_each_entry_reverse(de, &ctx->defer_list, list) { 11055ed29b0b4SJens Axboe if (io_match_task_safe(de->req, task, cancel_all)) { 11056ed29b0b4SJens Axboe list_cut_position(&list, &ctx->defer_list, &de->list); 11057ed29b0b4SJens Axboe break; 11058ed29b0b4SJens Axboe } 11059ed29b0b4SJens Axboe } 11060ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 11061ed29b0b4SJens Axboe if (list_empty(&list)) 11062ed29b0b4SJens Axboe return false; 11063ed29b0b4SJens Axboe 11064ed29b0b4SJens Axboe while (!list_empty(&list)) { 11065ed29b0b4SJens Axboe de = list_first_entry(&list, struct io_defer_entry, list); 11066ed29b0b4SJens Axboe list_del_init(&de->list); 11067ed29b0b4SJens Axboe io_req_complete_failed(de->req, -ECANCELED); 11068ed29b0b4SJens Axboe kfree(de); 11069ed29b0b4SJens Axboe } 11070ed29b0b4SJens Axboe return true; 11071ed29b0b4SJens Axboe } 11072ed29b0b4SJens Axboe 11073ed29b0b4SJens Axboe static __cold bool io_uring_try_cancel_iowq(struct io_ring_ctx *ctx) 11074ed29b0b4SJens Axboe { 11075ed29b0b4SJens Axboe struct io_tctx_node *node; 11076ed29b0b4SJens Axboe enum io_wq_cancel cret; 11077ed29b0b4SJens Axboe bool ret = false; 11078ed29b0b4SJens Axboe 11079ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 11080ed29b0b4SJens Axboe list_for_each_entry(node, &ctx->tctx_list, ctx_node) { 11081ed29b0b4SJens Axboe struct io_uring_task *tctx = node->task->io_uring; 11082ed29b0b4SJens Axboe 11083ed29b0b4SJens Axboe /* 11084ed29b0b4SJens Axboe * io_wq will stay alive while we hold uring_lock, because it's 11085ed29b0b4SJens Axboe * killed after ctx nodes, which requires to take the lock. 11086ed29b0b4SJens Axboe */ 11087ed29b0b4SJens Axboe if (!tctx || !tctx->io_wq) 11088ed29b0b4SJens Axboe continue; 11089ed29b0b4SJens Axboe cret = io_wq_cancel_cb(tctx->io_wq, io_cancel_ctx_cb, ctx, true); 11090ed29b0b4SJens Axboe ret |= (cret != IO_WQ_CANCEL_NOTFOUND); 11091ed29b0b4SJens Axboe } 11092ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 11093ed29b0b4SJens Axboe 11094ed29b0b4SJens Axboe return ret; 11095ed29b0b4SJens Axboe } 11096ed29b0b4SJens Axboe 11097ed29b0b4SJens Axboe static __cold void io_uring_try_cancel_requests(struct io_ring_ctx *ctx, 11098ed29b0b4SJens Axboe struct task_struct *task, 11099ed29b0b4SJens Axboe bool cancel_all) 11100ed29b0b4SJens Axboe { 11101ed29b0b4SJens Axboe struct io_task_cancel cancel = { .task = task, .all = cancel_all, }; 11102ed29b0b4SJens Axboe struct io_uring_task *tctx = task ? task->io_uring : NULL; 11103ed29b0b4SJens Axboe 11104ed29b0b4SJens Axboe /* failed during ring init, it couldn't have issued any requests */ 11105ed29b0b4SJens Axboe if (!ctx->rings) 11106ed29b0b4SJens Axboe return; 11107ed29b0b4SJens Axboe 11108ed29b0b4SJens Axboe while (1) { 11109ed29b0b4SJens Axboe enum io_wq_cancel cret; 11110ed29b0b4SJens Axboe bool ret = false; 11111ed29b0b4SJens Axboe 11112ed29b0b4SJens Axboe if (!task) { 11113ed29b0b4SJens Axboe ret |= io_uring_try_cancel_iowq(ctx); 11114ed29b0b4SJens Axboe } else if (tctx && tctx->io_wq) { 11115ed29b0b4SJens Axboe /* 11116ed29b0b4SJens Axboe * Cancels requests of all rings, not only @ctx, but 11117ed29b0b4SJens Axboe * it's fine as the task is in exit/exec. 11118ed29b0b4SJens Axboe */ 11119ed29b0b4SJens Axboe cret = io_wq_cancel_cb(tctx->io_wq, io_cancel_task_cb, 11120ed29b0b4SJens Axboe &cancel, true); 11121ed29b0b4SJens Axboe ret |= (cret != IO_WQ_CANCEL_NOTFOUND); 11122ed29b0b4SJens Axboe } 11123ed29b0b4SJens Axboe 11124ed29b0b4SJens Axboe /* SQPOLL thread does its own polling */ 11125ed29b0b4SJens Axboe if ((!(ctx->flags & IORING_SETUP_SQPOLL) && cancel_all) || 11126ed29b0b4SJens Axboe (ctx->sq_data && ctx->sq_data->thread == current)) { 11127ed29b0b4SJens Axboe while (!wq_list_empty(&ctx->iopoll_list)) { 11128ed29b0b4SJens Axboe io_iopoll_try_reap_events(ctx); 11129ed29b0b4SJens Axboe ret = true; 11130ed29b0b4SJens Axboe } 11131ed29b0b4SJens Axboe } 11132ed29b0b4SJens Axboe 11133ed29b0b4SJens Axboe ret |= io_cancel_defer_files(ctx, task, cancel_all); 11134ed29b0b4SJens Axboe ret |= io_poll_remove_all(ctx, task, cancel_all); 11135ed29b0b4SJens Axboe ret |= io_kill_timeouts(ctx, task, cancel_all); 11136ed29b0b4SJens Axboe if (task) 11137ed29b0b4SJens Axboe ret |= io_run_task_work(); 11138ed29b0b4SJens Axboe if (!ret) 11139ed29b0b4SJens Axboe break; 11140ed29b0b4SJens Axboe cond_resched(); 11141ed29b0b4SJens Axboe } 11142ed29b0b4SJens Axboe } 11143ed29b0b4SJens Axboe 11144ed29b0b4SJens Axboe static int __io_uring_add_tctx_node(struct io_ring_ctx *ctx) 11145ed29b0b4SJens Axboe { 11146ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 11147ed29b0b4SJens Axboe struct io_tctx_node *node; 11148ed29b0b4SJens Axboe int ret; 11149ed29b0b4SJens Axboe 11150ed29b0b4SJens Axboe if (unlikely(!tctx)) { 11151ed29b0b4SJens Axboe ret = io_uring_alloc_task_context(current, ctx); 11152ed29b0b4SJens Axboe if (unlikely(ret)) 11153ed29b0b4SJens Axboe return ret; 11154ed29b0b4SJens Axboe 11155ed29b0b4SJens Axboe tctx = current->io_uring; 11156ed29b0b4SJens Axboe if (ctx->iowq_limits_set) { 11157ed29b0b4SJens Axboe unsigned int limits[2] = { ctx->iowq_limits[0], 11158ed29b0b4SJens Axboe ctx->iowq_limits[1], }; 11159ed29b0b4SJens Axboe 11160ed29b0b4SJens Axboe ret = io_wq_max_workers(tctx->io_wq, limits); 11161ed29b0b4SJens Axboe if (ret) 11162ed29b0b4SJens Axboe return ret; 11163ed29b0b4SJens Axboe } 11164ed29b0b4SJens Axboe } 11165ed29b0b4SJens Axboe if (!xa_load(&tctx->xa, (unsigned long)ctx)) { 11166ed29b0b4SJens Axboe node = kmalloc(sizeof(*node), GFP_KERNEL); 11167ed29b0b4SJens Axboe if (!node) 11168ed29b0b4SJens Axboe return -ENOMEM; 11169ed29b0b4SJens Axboe node->ctx = ctx; 11170ed29b0b4SJens Axboe node->task = current; 11171ed29b0b4SJens Axboe 11172ed29b0b4SJens Axboe ret = xa_err(xa_store(&tctx->xa, (unsigned long)ctx, 11173ed29b0b4SJens Axboe node, GFP_KERNEL)); 11174ed29b0b4SJens Axboe if (ret) { 11175ed29b0b4SJens Axboe kfree(node); 11176ed29b0b4SJens Axboe return ret; 11177ed29b0b4SJens Axboe } 11178ed29b0b4SJens Axboe 11179ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 11180ed29b0b4SJens Axboe list_add(&node->ctx_node, &ctx->tctx_list); 11181ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 11182ed29b0b4SJens Axboe } 11183ed29b0b4SJens Axboe tctx->last = ctx; 11184ed29b0b4SJens Axboe return 0; 11185ed29b0b4SJens Axboe } 11186ed29b0b4SJens Axboe 11187ed29b0b4SJens Axboe /* 11188ed29b0b4SJens Axboe * Note that this task has used io_uring. We use it for cancelation purposes. 11189ed29b0b4SJens Axboe */ 11190ed29b0b4SJens Axboe static inline int io_uring_add_tctx_node(struct io_ring_ctx *ctx) 11191ed29b0b4SJens Axboe { 11192ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 11193ed29b0b4SJens Axboe 11194ed29b0b4SJens Axboe if (likely(tctx && tctx->last == ctx)) 11195ed29b0b4SJens Axboe return 0; 11196ed29b0b4SJens Axboe return __io_uring_add_tctx_node(ctx); 11197ed29b0b4SJens Axboe } 11198ed29b0b4SJens Axboe 11199ed29b0b4SJens Axboe /* 11200ed29b0b4SJens Axboe * Remove this io_uring_file -> task mapping. 11201ed29b0b4SJens Axboe */ 11202ed29b0b4SJens Axboe static __cold void io_uring_del_tctx_node(unsigned long index) 11203ed29b0b4SJens Axboe { 11204ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 11205ed29b0b4SJens Axboe struct io_tctx_node *node; 11206ed29b0b4SJens Axboe 11207ed29b0b4SJens Axboe if (!tctx) 11208ed29b0b4SJens Axboe return; 11209ed29b0b4SJens Axboe node = xa_erase(&tctx->xa, index); 11210ed29b0b4SJens Axboe if (!node) 11211ed29b0b4SJens Axboe return; 11212ed29b0b4SJens Axboe 11213ed29b0b4SJens Axboe WARN_ON_ONCE(current != node->task); 11214ed29b0b4SJens Axboe WARN_ON_ONCE(list_empty(&node->ctx_node)); 11215ed29b0b4SJens Axboe 11216ed29b0b4SJens Axboe mutex_lock(&node->ctx->uring_lock); 11217ed29b0b4SJens Axboe list_del(&node->ctx_node); 11218ed29b0b4SJens Axboe mutex_unlock(&node->ctx->uring_lock); 11219ed29b0b4SJens Axboe 11220ed29b0b4SJens Axboe if (tctx->last == node->ctx) 11221ed29b0b4SJens Axboe tctx->last = NULL; 11222ed29b0b4SJens Axboe kfree(node); 11223ed29b0b4SJens Axboe } 11224ed29b0b4SJens Axboe 11225ed29b0b4SJens Axboe static __cold void io_uring_clean_tctx(struct io_uring_task *tctx) 11226ed29b0b4SJens Axboe { 11227ed29b0b4SJens Axboe struct io_wq *wq = tctx->io_wq; 11228ed29b0b4SJens Axboe struct io_tctx_node *node; 11229ed29b0b4SJens Axboe unsigned long index; 11230ed29b0b4SJens Axboe 11231ed29b0b4SJens Axboe xa_for_each(&tctx->xa, index, node) { 11232ed29b0b4SJens Axboe io_uring_del_tctx_node(index); 11233ed29b0b4SJens Axboe cond_resched(); 11234ed29b0b4SJens Axboe } 11235ed29b0b4SJens Axboe if (wq) { 11236ed29b0b4SJens Axboe /* 11237ed29b0b4SJens Axboe * Must be after io_uring_del_tctx_node() (removes nodes under 11238ed29b0b4SJens Axboe * uring_lock) to avoid race with io_uring_try_cancel_iowq(). 11239ed29b0b4SJens Axboe */ 11240ed29b0b4SJens Axboe io_wq_put_and_exit(wq); 11241ed29b0b4SJens Axboe tctx->io_wq = NULL; 11242ed29b0b4SJens Axboe } 11243ed29b0b4SJens Axboe } 11244ed29b0b4SJens Axboe 11245ed29b0b4SJens Axboe static s64 tctx_inflight(struct io_uring_task *tctx, bool tracked) 11246ed29b0b4SJens Axboe { 11247ed29b0b4SJens Axboe if (tracked) 11248ed29b0b4SJens Axboe return atomic_read(&tctx->inflight_tracked); 11249ed29b0b4SJens Axboe return percpu_counter_sum(&tctx->inflight); 11250ed29b0b4SJens Axboe } 11251ed29b0b4SJens Axboe 11252ed29b0b4SJens Axboe /* 11253ed29b0b4SJens Axboe * Find any io_uring ctx that this task has registered or done IO on, and cancel 11254ed29b0b4SJens Axboe * requests. @sqd should be not-null IFF it's an SQPOLL thread cancellation. 11255ed29b0b4SJens Axboe */ 11256ed29b0b4SJens Axboe static __cold void io_uring_cancel_generic(bool cancel_all, 11257ed29b0b4SJens Axboe struct io_sq_data *sqd) 11258ed29b0b4SJens Axboe { 11259ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 11260ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 11261ed29b0b4SJens Axboe s64 inflight; 11262ed29b0b4SJens Axboe DEFINE_WAIT(wait); 11263ed29b0b4SJens Axboe 11264ed29b0b4SJens Axboe WARN_ON_ONCE(sqd && sqd->thread != current); 11265ed29b0b4SJens Axboe 11266ed29b0b4SJens Axboe if (!current->io_uring) 11267ed29b0b4SJens Axboe return; 11268ed29b0b4SJens Axboe if (tctx->io_wq) 11269ed29b0b4SJens Axboe io_wq_exit_start(tctx->io_wq); 11270ed29b0b4SJens Axboe 11271ed29b0b4SJens Axboe atomic_inc(&tctx->in_idle); 11272ed29b0b4SJens Axboe do { 11273ed29b0b4SJens Axboe io_uring_drop_tctx_refs(current); 11274ed29b0b4SJens Axboe /* read completions before cancelations */ 11275ed29b0b4SJens Axboe inflight = tctx_inflight(tctx, !cancel_all); 11276ed29b0b4SJens Axboe if (!inflight) 11277ed29b0b4SJens Axboe break; 11278ed29b0b4SJens Axboe 11279ed29b0b4SJens Axboe if (!sqd) { 11280ed29b0b4SJens Axboe struct io_tctx_node *node; 11281ed29b0b4SJens Axboe unsigned long index; 11282ed29b0b4SJens Axboe 11283ed29b0b4SJens Axboe xa_for_each(&tctx->xa, index, node) { 11284ed29b0b4SJens Axboe /* sqpoll task will cancel all its requests */ 11285ed29b0b4SJens Axboe if (node->ctx->sq_data) 11286ed29b0b4SJens Axboe continue; 11287ed29b0b4SJens Axboe io_uring_try_cancel_requests(node->ctx, current, 11288ed29b0b4SJens Axboe cancel_all); 11289ed29b0b4SJens Axboe } 11290ed29b0b4SJens Axboe } else { 11291ed29b0b4SJens Axboe list_for_each_entry(ctx, &sqd->ctx_list, sqd_list) 11292ed29b0b4SJens Axboe io_uring_try_cancel_requests(ctx, current, 11293ed29b0b4SJens Axboe cancel_all); 11294ed29b0b4SJens Axboe } 11295ed29b0b4SJens Axboe 11296ed29b0b4SJens Axboe prepare_to_wait(&tctx->wait, &wait, TASK_INTERRUPTIBLE); 11297ed29b0b4SJens Axboe io_run_task_work(); 11298ed29b0b4SJens Axboe io_uring_drop_tctx_refs(current); 11299ed29b0b4SJens Axboe 11300ed29b0b4SJens Axboe /* 11301ed29b0b4SJens Axboe * If we've seen completions, retry without waiting. This 11302ed29b0b4SJens Axboe * avoids a race where a completion comes in before we did 11303ed29b0b4SJens Axboe * prepare_to_wait(). 11304ed29b0b4SJens Axboe */ 11305ed29b0b4SJens Axboe if (inflight == tctx_inflight(tctx, !cancel_all)) 11306ed29b0b4SJens Axboe schedule(); 11307ed29b0b4SJens Axboe finish_wait(&tctx->wait, &wait); 11308ed29b0b4SJens Axboe } while (1); 11309ed29b0b4SJens Axboe 11310ed29b0b4SJens Axboe io_uring_clean_tctx(tctx); 11311ed29b0b4SJens Axboe if (cancel_all) { 11312ed29b0b4SJens Axboe /* 11313ed29b0b4SJens Axboe * We shouldn't run task_works after cancel, so just leave 11314ed29b0b4SJens Axboe * ->in_idle set for normal exit. 11315ed29b0b4SJens Axboe */ 11316ed29b0b4SJens Axboe atomic_dec(&tctx->in_idle); 11317ed29b0b4SJens Axboe /* for exec all current's requests should be gone, kill tctx */ 11318ed29b0b4SJens Axboe __io_uring_free(current); 11319ed29b0b4SJens Axboe } 11320ed29b0b4SJens Axboe } 11321ed29b0b4SJens Axboe 11322ed29b0b4SJens Axboe void __io_uring_cancel(bool cancel_all) 11323ed29b0b4SJens Axboe { 11324ed29b0b4SJens Axboe io_uring_cancel_generic(cancel_all, NULL); 11325ed29b0b4SJens Axboe } 11326ed29b0b4SJens Axboe 11327ed29b0b4SJens Axboe void io_uring_unreg_ringfd(void) 11328ed29b0b4SJens Axboe { 11329ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 11330ed29b0b4SJens Axboe int i; 11331ed29b0b4SJens Axboe 11332ed29b0b4SJens Axboe for (i = 0; i < IO_RINGFD_REG_MAX; i++) { 11333ed29b0b4SJens Axboe if (tctx->registered_rings[i]) { 11334ed29b0b4SJens Axboe fput(tctx->registered_rings[i]); 11335ed29b0b4SJens Axboe tctx->registered_rings[i] = NULL; 11336ed29b0b4SJens Axboe } 11337ed29b0b4SJens Axboe } 11338ed29b0b4SJens Axboe } 11339ed29b0b4SJens Axboe 11340ed29b0b4SJens Axboe static int io_ring_add_registered_fd(struct io_uring_task *tctx, int fd, 11341ed29b0b4SJens Axboe int start, int end) 11342ed29b0b4SJens Axboe { 11343ed29b0b4SJens Axboe struct file *file; 11344ed29b0b4SJens Axboe int offset; 11345ed29b0b4SJens Axboe 11346ed29b0b4SJens Axboe for (offset = start; offset < end; offset++) { 11347ed29b0b4SJens Axboe offset = array_index_nospec(offset, IO_RINGFD_REG_MAX); 11348ed29b0b4SJens Axboe if (tctx->registered_rings[offset]) 11349ed29b0b4SJens Axboe continue; 11350ed29b0b4SJens Axboe 11351ed29b0b4SJens Axboe file = fget(fd); 11352ed29b0b4SJens Axboe if (!file) { 11353ed29b0b4SJens Axboe return -EBADF; 11354ed29b0b4SJens Axboe } else if (file->f_op != &io_uring_fops) { 11355ed29b0b4SJens Axboe fput(file); 11356ed29b0b4SJens Axboe return -EOPNOTSUPP; 11357ed29b0b4SJens Axboe } 11358ed29b0b4SJens Axboe tctx->registered_rings[offset] = file; 11359ed29b0b4SJens Axboe return offset; 11360ed29b0b4SJens Axboe } 11361ed29b0b4SJens Axboe 11362ed29b0b4SJens Axboe return -EBUSY; 11363ed29b0b4SJens Axboe } 11364ed29b0b4SJens Axboe 11365ed29b0b4SJens Axboe /* 11366ed29b0b4SJens Axboe * Register a ring fd to avoid fdget/fdput for each io_uring_enter() 11367ed29b0b4SJens Axboe * invocation. User passes in an array of struct io_uring_rsrc_update 11368ed29b0b4SJens Axboe * with ->data set to the ring_fd, and ->offset given for the desired 11369ed29b0b4SJens Axboe * index. If no index is desired, application may set ->offset == -1U 11370ed29b0b4SJens Axboe * and we'll find an available index. Returns number of entries 11371ed29b0b4SJens Axboe * successfully processed, or < 0 on error if none were processed. 11372ed29b0b4SJens Axboe */ 11373ed29b0b4SJens Axboe static int io_ringfd_register(struct io_ring_ctx *ctx, void __user *__arg, 11374ed29b0b4SJens Axboe unsigned nr_args) 11375ed29b0b4SJens Axboe { 11376ed29b0b4SJens Axboe struct io_uring_rsrc_update __user *arg = __arg; 11377ed29b0b4SJens Axboe struct io_uring_rsrc_update reg; 11378ed29b0b4SJens Axboe struct io_uring_task *tctx; 11379ed29b0b4SJens Axboe int ret, i; 11380ed29b0b4SJens Axboe 11381ed29b0b4SJens Axboe if (!nr_args || nr_args > IO_RINGFD_REG_MAX) 11382ed29b0b4SJens Axboe return -EINVAL; 11383ed29b0b4SJens Axboe 11384ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 11385ed29b0b4SJens Axboe ret = io_uring_add_tctx_node(ctx); 11386ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 11387ed29b0b4SJens Axboe if (ret) 11388ed29b0b4SJens Axboe return ret; 11389ed29b0b4SJens Axboe 11390ed29b0b4SJens Axboe tctx = current->io_uring; 11391ed29b0b4SJens Axboe for (i = 0; i < nr_args; i++) { 11392ed29b0b4SJens Axboe int start, end; 11393ed29b0b4SJens Axboe 11394ed29b0b4SJens Axboe if (copy_from_user(®, &arg[i], sizeof(reg))) { 11395ed29b0b4SJens Axboe ret = -EFAULT; 11396ed29b0b4SJens Axboe break; 11397ed29b0b4SJens Axboe } 11398ed29b0b4SJens Axboe 11399ed29b0b4SJens Axboe if (reg.resv) { 11400ed29b0b4SJens Axboe ret = -EINVAL; 11401ed29b0b4SJens Axboe break; 11402ed29b0b4SJens Axboe } 11403ed29b0b4SJens Axboe 11404ed29b0b4SJens Axboe if (reg.offset == -1U) { 11405ed29b0b4SJens Axboe start = 0; 11406ed29b0b4SJens Axboe end = IO_RINGFD_REG_MAX; 11407ed29b0b4SJens Axboe } else { 11408ed29b0b4SJens Axboe if (reg.offset >= IO_RINGFD_REG_MAX) { 11409ed29b0b4SJens Axboe ret = -EINVAL; 11410ed29b0b4SJens Axboe break; 11411ed29b0b4SJens Axboe } 11412ed29b0b4SJens Axboe start = reg.offset; 11413ed29b0b4SJens Axboe end = start + 1; 11414ed29b0b4SJens Axboe } 11415ed29b0b4SJens Axboe 11416ed29b0b4SJens Axboe ret = io_ring_add_registered_fd(tctx, reg.data, start, end); 11417ed29b0b4SJens Axboe if (ret < 0) 11418ed29b0b4SJens Axboe break; 11419ed29b0b4SJens Axboe 11420ed29b0b4SJens Axboe reg.offset = ret; 11421ed29b0b4SJens Axboe if (copy_to_user(&arg[i], ®, sizeof(reg))) { 11422ed29b0b4SJens Axboe fput(tctx->registered_rings[reg.offset]); 11423ed29b0b4SJens Axboe tctx->registered_rings[reg.offset] = NULL; 11424ed29b0b4SJens Axboe ret = -EFAULT; 11425ed29b0b4SJens Axboe break; 11426ed29b0b4SJens Axboe } 11427ed29b0b4SJens Axboe } 11428ed29b0b4SJens Axboe 11429ed29b0b4SJens Axboe return i ? i : ret; 11430ed29b0b4SJens Axboe } 11431ed29b0b4SJens Axboe 11432ed29b0b4SJens Axboe static int io_ringfd_unregister(struct io_ring_ctx *ctx, void __user *__arg, 11433ed29b0b4SJens Axboe unsigned nr_args) 11434ed29b0b4SJens Axboe { 11435ed29b0b4SJens Axboe struct io_uring_rsrc_update __user *arg = __arg; 11436ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 11437ed29b0b4SJens Axboe struct io_uring_rsrc_update reg; 11438ed29b0b4SJens Axboe int ret = 0, i; 11439ed29b0b4SJens Axboe 11440ed29b0b4SJens Axboe if (!nr_args || nr_args > IO_RINGFD_REG_MAX) 11441ed29b0b4SJens Axboe return -EINVAL; 11442ed29b0b4SJens Axboe if (!tctx) 11443ed29b0b4SJens Axboe return 0; 11444ed29b0b4SJens Axboe 11445ed29b0b4SJens Axboe for (i = 0; i < nr_args; i++) { 11446ed29b0b4SJens Axboe if (copy_from_user(®, &arg[i], sizeof(reg))) { 11447ed29b0b4SJens Axboe ret = -EFAULT; 11448ed29b0b4SJens Axboe break; 11449ed29b0b4SJens Axboe } 11450ed29b0b4SJens Axboe if (reg.resv || reg.data || reg.offset >= IO_RINGFD_REG_MAX) { 11451ed29b0b4SJens Axboe ret = -EINVAL; 11452ed29b0b4SJens Axboe break; 11453ed29b0b4SJens Axboe } 11454ed29b0b4SJens Axboe 11455ed29b0b4SJens Axboe reg.offset = array_index_nospec(reg.offset, IO_RINGFD_REG_MAX); 11456ed29b0b4SJens Axboe if (tctx->registered_rings[reg.offset]) { 11457ed29b0b4SJens Axboe fput(tctx->registered_rings[reg.offset]); 11458ed29b0b4SJens Axboe tctx->registered_rings[reg.offset] = NULL; 11459ed29b0b4SJens Axboe } 11460ed29b0b4SJens Axboe } 11461ed29b0b4SJens Axboe 11462ed29b0b4SJens Axboe return i ? i : ret; 11463ed29b0b4SJens Axboe } 11464ed29b0b4SJens Axboe 11465ed29b0b4SJens Axboe static void *io_uring_validate_mmap_request(struct file *file, 11466ed29b0b4SJens Axboe loff_t pgoff, size_t sz) 11467ed29b0b4SJens Axboe { 11468ed29b0b4SJens Axboe struct io_ring_ctx *ctx = file->private_data; 11469ed29b0b4SJens Axboe loff_t offset = pgoff << PAGE_SHIFT; 11470ed29b0b4SJens Axboe struct page *page; 11471ed29b0b4SJens Axboe void *ptr; 11472ed29b0b4SJens Axboe 11473ed29b0b4SJens Axboe switch (offset) { 11474ed29b0b4SJens Axboe case IORING_OFF_SQ_RING: 11475ed29b0b4SJens Axboe case IORING_OFF_CQ_RING: 11476ed29b0b4SJens Axboe ptr = ctx->rings; 11477ed29b0b4SJens Axboe break; 11478ed29b0b4SJens Axboe case IORING_OFF_SQES: 11479ed29b0b4SJens Axboe ptr = ctx->sq_sqes; 11480ed29b0b4SJens Axboe break; 11481ed29b0b4SJens Axboe default: 11482ed29b0b4SJens Axboe return ERR_PTR(-EINVAL); 11483ed29b0b4SJens Axboe } 11484ed29b0b4SJens Axboe 11485ed29b0b4SJens Axboe page = virt_to_head_page(ptr); 11486ed29b0b4SJens Axboe if (sz > page_size(page)) 11487ed29b0b4SJens Axboe return ERR_PTR(-EINVAL); 11488ed29b0b4SJens Axboe 11489ed29b0b4SJens Axboe return ptr; 11490ed29b0b4SJens Axboe } 11491ed29b0b4SJens Axboe 11492ed29b0b4SJens Axboe #ifdef CONFIG_MMU 11493ed29b0b4SJens Axboe 11494ed29b0b4SJens Axboe static __cold int io_uring_mmap(struct file *file, struct vm_area_struct *vma) 11495ed29b0b4SJens Axboe { 11496ed29b0b4SJens Axboe size_t sz = vma->vm_end - vma->vm_start; 11497ed29b0b4SJens Axboe unsigned long pfn; 11498ed29b0b4SJens Axboe void *ptr; 11499ed29b0b4SJens Axboe 11500ed29b0b4SJens Axboe ptr = io_uring_validate_mmap_request(file, vma->vm_pgoff, sz); 11501ed29b0b4SJens Axboe if (IS_ERR(ptr)) 11502ed29b0b4SJens Axboe return PTR_ERR(ptr); 11503ed29b0b4SJens Axboe 11504ed29b0b4SJens Axboe pfn = virt_to_phys(ptr) >> PAGE_SHIFT; 11505ed29b0b4SJens Axboe return remap_pfn_range(vma, vma->vm_start, pfn, sz, vma->vm_page_prot); 11506ed29b0b4SJens Axboe } 11507ed29b0b4SJens Axboe 11508ed29b0b4SJens Axboe #else /* !CONFIG_MMU */ 11509ed29b0b4SJens Axboe 11510ed29b0b4SJens Axboe static int io_uring_mmap(struct file *file, struct vm_area_struct *vma) 11511ed29b0b4SJens Axboe { 11512ed29b0b4SJens Axboe return vma->vm_flags & (VM_SHARED | VM_MAYSHARE) ? 0 : -EINVAL; 11513ed29b0b4SJens Axboe } 11514ed29b0b4SJens Axboe 11515ed29b0b4SJens Axboe static unsigned int io_uring_nommu_mmap_capabilities(struct file *file) 11516ed29b0b4SJens Axboe { 11517ed29b0b4SJens Axboe return NOMMU_MAP_DIRECT | NOMMU_MAP_READ | NOMMU_MAP_WRITE; 11518ed29b0b4SJens Axboe } 11519ed29b0b4SJens Axboe 11520ed29b0b4SJens Axboe static unsigned long io_uring_nommu_get_unmapped_area(struct file *file, 11521ed29b0b4SJens Axboe unsigned long addr, unsigned long len, 11522ed29b0b4SJens Axboe unsigned long pgoff, unsigned long flags) 11523ed29b0b4SJens Axboe { 11524ed29b0b4SJens Axboe void *ptr; 11525ed29b0b4SJens Axboe 11526ed29b0b4SJens Axboe ptr = io_uring_validate_mmap_request(file, pgoff, len); 11527ed29b0b4SJens Axboe if (IS_ERR(ptr)) 11528ed29b0b4SJens Axboe return PTR_ERR(ptr); 11529ed29b0b4SJens Axboe 11530ed29b0b4SJens Axboe return (unsigned long) ptr; 11531ed29b0b4SJens Axboe } 11532ed29b0b4SJens Axboe 11533ed29b0b4SJens Axboe #endif /* !CONFIG_MMU */ 11534ed29b0b4SJens Axboe 11535ed29b0b4SJens Axboe static int io_sqpoll_wait_sq(struct io_ring_ctx *ctx) 11536ed29b0b4SJens Axboe { 11537ed29b0b4SJens Axboe DEFINE_WAIT(wait); 11538ed29b0b4SJens Axboe 11539ed29b0b4SJens Axboe do { 11540ed29b0b4SJens Axboe if (!io_sqring_full(ctx)) 11541ed29b0b4SJens Axboe break; 11542ed29b0b4SJens Axboe prepare_to_wait(&ctx->sqo_sq_wait, &wait, TASK_INTERRUPTIBLE); 11543ed29b0b4SJens Axboe 11544ed29b0b4SJens Axboe if (!io_sqring_full(ctx)) 11545ed29b0b4SJens Axboe break; 11546ed29b0b4SJens Axboe schedule(); 11547ed29b0b4SJens Axboe } while (!signal_pending(current)); 11548ed29b0b4SJens Axboe 11549ed29b0b4SJens Axboe finish_wait(&ctx->sqo_sq_wait, &wait); 11550ed29b0b4SJens Axboe return 0; 11551ed29b0b4SJens Axboe } 11552ed29b0b4SJens Axboe 11553ed29b0b4SJens Axboe static int io_validate_ext_arg(unsigned flags, const void __user *argp, size_t argsz) 11554ed29b0b4SJens Axboe { 11555ed29b0b4SJens Axboe if (flags & IORING_ENTER_EXT_ARG) { 11556ed29b0b4SJens Axboe struct io_uring_getevents_arg arg; 11557ed29b0b4SJens Axboe 11558ed29b0b4SJens Axboe if (argsz != sizeof(arg)) 11559ed29b0b4SJens Axboe return -EINVAL; 11560ed29b0b4SJens Axboe if (copy_from_user(&arg, argp, sizeof(arg))) 11561ed29b0b4SJens Axboe return -EFAULT; 11562ed29b0b4SJens Axboe } 11563ed29b0b4SJens Axboe return 0; 11564ed29b0b4SJens Axboe } 11565ed29b0b4SJens Axboe 11566ed29b0b4SJens Axboe static int io_get_ext_arg(unsigned flags, const void __user *argp, size_t *argsz, 11567ed29b0b4SJens Axboe struct __kernel_timespec __user **ts, 11568ed29b0b4SJens Axboe const sigset_t __user **sig) 11569ed29b0b4SJens Axboe { 11570ed29b0b4SJens Axboe struct io_uring_getevents_arg arg; 11571ed29b0b4SJens Axboe 11572ed29b0b4SJens Axboe /* 11573ed29b0b4SJens Axboe * If EXT_ARG isn't set, then we have no timespec and the argp pointer 11574ed29b0b4SJens Axboe * is just a pointer to the sigset_t. 11575ed29b0b4SJens Axboe */ 11576ed29b0b4SJens Axboe if (!(flags & IORING_ENTER_EXT_ARG)) { 11577ed29b0b4SJens Axboe *sig = (const sigset_t __user *) argp; 11578ed29b0b4SJens Axboe *ts = NULL; 11579ed29b0b4SJens Axboe return 0; 11580ed29b0b4SJens Axboe } 11581ed29b0b4SJens Axboe 11582ed29b0b4SJens Axboe /* 11583ed29b0b4SJens Axboe * EXT_ARG is set - ensure we agree on the size of it and copy in our 11584ed29b0b4SJens Axboe * timespec and sigset_t pointers if good. 11585ed29b0b4SJens Axboe */ 11586ed29b0b4SJens Axboe if (*argsz != sizeof(arg)) 11587ed29b0b4SJens Axboe return -EINVAL; 11588ed29b0b4SJens Axboe if (copy_from_user(&arg, argp, sizeof(arg))) 11589ed29b0b4SJens Axboe return -EFAULT; 11590ed29b0b4SJens Axboe if (arg.pad) 11591ed29b0b4SJens Axboe return -EINVAL; 11592ed29b0b4SJens Axboe *sig = u64_to_user_ptr(arg.sigmask); 11593ed29b0b4SJens Axboe *argsz = arg.sigmask_sz; 11594ed29b0b4SJens Axboe *ts = u64_to_user_ptr(arg.ts); 11595ed29b0b4SJens Axboe return 0; 11596ed29b0b4SJens Axboe } 11597ed29b0b4SJens Axboe 11598ed29b0b4SJens Axboe SYSCALL_DEFINE6(io_uring_enter, unsigned int, fd, u32, to_submit, 11599ed29b0b4SJens Axboe u32, min_complete, u32, flags, const void __user *, argp, 11600ed29b0b4SJens Axboe size_t, argsz) 11601ed29b0b4SJens Axboe { 11602ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 11603ed29b0b4SJens Axboe struct fd f; 11604ed29b0b4SJens Axboe long ret; 11605ed29b0b4SJens Axboe 11606ed29b0b4SJens Axboe io_run_task_work(); 11607ed29b0b4SJens Axboe 11608ed29b0b4SJens Axboe if (unlikely(flags & ~(IORING_ENTER_GETEVENTS | IORING_ENTER_SQ_WAKEUP | 11609ed29b0b4SJens Axboe IORING_ENTER_SQ_WAIT | IORING_ENTER_EXT_ARG | 11610ed29b0b4SJens Axboe IORING_ENTER_REGISTERED_RING))) 11611ed29b0b4SJens Axboe return -EINVAL; 11612ed29b0b4SJens Axboe 11613ed29b0b4SJens Axboe /* 11614ed29b0b4SJens Axboe * Ring fd has been registered via IORING_REGISTER_RING_FDS, we 11615ed29b0b4SJens Axboe * need only dereference our task private array to find it. 11616ed29b0b4SJens Axboe */ 11617ed29b0b4SJens Axboe if (flags & IORING_ENTER_REGISTERED_RING) { 11618ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 11619ed29b0b4SJens Axboe 11620ed29b0b4SJens Axboe if (!tctx || fd >= IO_RINGFD_REG_MAX) 11621ed29b0b4SJens Axboe return -EINVAL; 11622ed29b0b4SJens Axboe fd = array_index_nospec(fd, IO_RINGFD_REG_MAX); 11623ed29b0b4SJens Axboe f.file = tctx->registered_rings[fd]; 11624ed29b0b4SJens Axboe f.flags = 0; 11625ed29b0b4SJens Axboe } else { 11626ed29b0b4SJens Axboe f = fdget(fd); 11627ed29b0b4SJens Axboe } 11628ed29b0b4SJens Axboe 11629ed29b0b4SJens Axboe if (unlikely(!f.file)) 11630ed29b0b4SJens Axboe return -EBADF; 11631ed29b0b4SJens Axboe 11632ed29b0b4SJens Axboe ret = -EOPNOTSUPP; 11633ed29b0b4SJens Axboe if (unlikely(f.file->f_op != &io_uring_fops)) 11634ed29b0b4SJens Axboe goto out_fput; 11635ed29b0b4SJens Axboe 11636ed29b0b4SJens Axboe ret = -ENXIO; 11637ed29b0b4SJens Axboe ctx = f.file->private_data; 11638ed29b0b4SJens Axboe if (unlikely(!percpu_ref_tryget(&ctx->refs))) 11639ed29b0b4SJens Axboe goto out_fput; 11640ed29b0b4SJens Axboe 11641ed29b0b4SJens Axboe ret = -EBADFD; 11642ed29b0b4SJens Axboe if (unlikely(ctx->flags & IORING_SETUP_R_DISABLED)) 11643ed29b0b4SJens Axboe goto out; 11644ed29b0b4SJens Axboe 11645ed29b0b4SJens Axboe /* 11646ed29b0b4SJens Axboe * For SQ polling, the thread will do all submissions and completions. 11647ed29b0b4SJens Axboe * Just return the requested submit count, and wake the thread if 11648ed29b0b4SJens Axboe * we were asked to. 11649ed29b0b4SJens Axboe */ 11650ed29b0b4SJens Axboe ret = 0; 11651ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQPOLL) { 11652ed29b0b4SJens Axboe io_cqring_overflow_flush(ctx); 11653ed29b0b4SJens Axboe 11654ed29b0b4SJens Axboe if (unlikely(ctx->sq_data->thread == NULL)) { 11655ed29b0b4SJens Axboe ret = -EOWNERDEAD; 11656ed29b0b4SJens Axboe goto out; 11657ed29b0b4SJens Axboe } 11658ed29b0b4SJens Axboe if (flags & IORING_ENTER_SQ_WAKEUP) 11659ed29b0b4SJens Axboe wake_up(&ctx->sq_data->wait); 11660ed29b0b4SJens Axboe if (flags & IORING_ENTER_SQ_WAIT) { 11661ed29b0b4SJens Axboe ret = io_sqpoll_wait_sq(ctx); 11662ed29b0b4SJens Axboe if (ret) 11663ed29b0b4SJens Axboe goto out; 11664ed29b0b4SJens Axboe } 11665ed29b0b4SJens Axboe ret = to_submit; 11666ed29b0b4SJens Axboe } else if (to_submit) { 11667ed29b0b4SJens Axboe ret = io_uring_add_tctx_node(ctx); 11668ed29b0b4SJens Axboe if (unlikely(ret)) 11669ed29b0b4SJens Axboe goto out; 11670ed29b0b4SJens Axboe 11671ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 11672ed29b0b4SJens Axboe ret = io_submit_sqes(ctx, to_submit); 11673ed29b0b4SJens Axboe if (ret != to_submit) { 11674ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 11675ed29b0b4SJens Axboe goto out; 11676ed29b0b4SJens Axboe } 11677ed29b0b4SJens Axboe if ((flags & IORING_ENTER_GETEVENTS) && ctx->syscall_iopoll) 11678ed29b0b4SJens Axboe goto iopoll_locked; 11679ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 11680ed29b0b4SJens Axboe } 11681ed29b0b4SJens Axboe if (flags & IORING_ENTER_GETEVENTS) { 11682ed29b0b4SJens Axboe int ret2; 11683ed29b0b4SJens Axboe if (ctx->syscall_iopoll) { 11684ed29b0b4SJens Axboe /* 11685ed29b0b4SJens Axboe * We disallow the app entering submit/complete with 11686ed29b0b4SJens Axboe * polling, but we still need to lock the ring to 11687ed29b0b4SJens Axboe * prevent racing with polled issue that got punted to 11688ed29b0b4SJens Axboe * a workqueue. 11689ed29b0b4SJens Axboe */ 11690ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 11691ed29b0b4SJens Axboe iopoll_locked: 11692ed29b0b4SJens Axboe ret2 = io_validate_ext_arg(flags, argp, argsz); 11693ed29b0b4SJens Axboe if (likely(!ret2)) { 11694ed29b0b4SJens Axboe min_complete = min(min_complete, 11695ed29b0b4SJens Axboe ctx->cq_entries); 11696ed29b0b4SJens Axboe ret2 = io_iopoll_check(ctx, min_complete); 11697ed29b0b4SJens Axboe } 11698ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 11699ed29b0b4SJens Axboe } else { 11700ed29b0b4SJens Axboe const sigset_t __user *sig; 11701ed29b0b4SJens Axboe struct __kernel_timespec __user *ts; 11702ed29b0b4SJens Axboe 11703ed29b0b4SJens Axboe ret2 = io_get_ext_arg(flags, argp, &argsz, &ts, &sig); 11704ed29b0b4SJens Axboe if (likely(!ret2)) { 11705ed29b0b4SJens Axboe min_complete = min(min_complete, 11706ed29b0b4SJens Axboe ctx->cq_entries); 11707ed29b0b4SJens Axboe ret2 = io_cqring_wait(ctx, min_complete, sig, 11708ed29b0b4SJens Axboe argsz, ts); 11709ed29b0b4SJens Axboe } 11710ed29b0b4SJens Axboe } 11711ed29b0b4SJens Axboe 11712ed29b0b4SJens Axboe if (!ret) { 11713ed29b0b4SJens Axboe ret = ret2; 11714ed29b0b4SJens Axboe 11715ed29b0b4SJens Axboe /* 11716ed29b0b4SJens Axboe * EBADR indicates that one or more CQE were dropped. 11717ed29b0b4SJens Axboe * Once the user has been informed we can clear the bit 11718ed29b0b4SJens Axboe * as they are obviously ok with those drops. 11719ed29b0b4SJens Axboe */ 11720ed29b0b4SJens Axboe if (unlikely(ret2 == -EBADR)) 11721ed29b0b4SJens Axboe clear_bit(IO_CHECK_CQ_DROPPED_BIT, 11722ed29b0b4SJens Axboe &ctx->check_cq); 11723ed29b0b4SJens Axboe } 11724ed29b0b4SJens Axboe } 11725ed29b0b4SJens Axboe 11726ed29b0b4SJens Axboe out: 11727ed29b0b4SJens Axboe percpu_ref_put(&ctx->refs); 11728ed29b0b4SJens Axboe out_fput: 11729ed29b0b4SJens Axboe fdput(f); 11730ed29b0b4SJens Axboe return ret; 11731ed29b0b4SJens Axboe } 11732ed29b0b4SJens Axboe 11733ed29b0b4SJens Axboe #ifdef CONFIG_PROC_FS 11734ed29b0b4SJens Axboe static __cold int io_uring_show_cred(struct seq_file *m, unsigned int id, 11735ed29b0b4SJens Axboe const struct cred *cred) 11736ed29b0b4SJens Axboe { 11737ed29b0b4SJens Axboe struct user_namespace *uns = seq_user_ns(m); 11738ed29b0b4SJens Axboe struct group_info *gi; 11739ed29b0b4SJens Axboe kernel_cap_t cap; 11740ed29b0b4SJens Axboe unsigned __capi; 11741ed29b0b4SJens Axboe int g; 11742ed29b0b4SJens Axboe 11743ed29b0b4SJens Axboe seq_printf(m, "%5d\n", id); 11744ed29b0b4SJens Axboe seq_put_decimal_ull(m, "\tUid:\t", from_kuid_munged(uns, cred->uid)); 11745ed29b0b4SJens Axboe seq_put_decimal_ull(m, "\t\t", from_kuid_munged(uns, cred->euid)); 11746ed29b0b4SJens Axboe seq_put_decimal_ull(m, "\t\t", from_kuid_munged(uns, cred->suid)); 11747ed29b0b4SJens Axboe seq_put_decimal_ull(m, "\t\t", from_kuid_munged(uns, cred->fsuid)); 11748ed29b0b4SJens Axboe seq_put_decimal_ull(m, "\n\tGid:\t", from_kgid_munged(uns, cred->gid)); 11749ed29b0b4SJens Axboe seq_put_decimal_ull(m, "\t\t", from_kgid_munged(uns, cred->egid)); 11750ed29b0b4SJens Axboe seq_put_decimal_ull(m, "\t\t", from_kgid_munged(uns, cred->sgid)); 11751ed29b0b4SJens Axboe seq_put_decimal_ull(m, "\t\t", from_kgid_munged(uns, cred->fsgid)); 11752ed29b0b4SJens Axboe seq_puts(m, "\n\tGroups:\t"); 11753ed29b0b4SJens Axboe gi = cred->group_info; 11754ed29b0b4SJens Axboe for (g = 0; g < gi->ngroups; g++) { 11755ed29b0b4SJens Axboe seq_put_decimal_ull(m, g ? " " : "", 11756ed29b0b4SJens Axboe from_kgid_munged(uns, gi->gid[g])); 11757ed29b0b4SJens Axboe } 11758ed29b0b4SJens Axboe seq_puts(m, "\n\tCapEff:\t"); 11759ed29b0b4SJens Axboe cap = cred->cap_effective; 11760ed29b0b4SJens Axboe CAP_FOR_EACH_U32(__capi) 11761ed29b0b4SJens Axboe seq_put_hex_ll(m, NULL, cap.cap[CAP_LAST_U32 - __capi], 8); 11762ed29b0b4SJens Axboe seq_putc(m, '\n'); 11763ed29b0b4SJens Axboe return 0; 11764ed29b0b4SJens Axboe } 11765ed29b0b4SJens Axboe 11766ed29b0b4SJens Axboe static __cold void __io_uring_show_fdinfo(struct io_ring_ctx *ctx, 11767ed29b0b4SJens Axboe struct seq_file *m) 11768ed29b0b4SJens Axboe { 11769ed29b0b4SJens Axboe struct io_sq_data *sq = NULL; 11770ed29b0b4SJens Axboe struct io_overflow_cqe *ocqe; 11771ed29b0b4SJens Axboe struct io_rings *r = ctx->rings; 11772ed29b0b4SJens Axboe unsigned int sq_mask = ctx->sq_entries - 1, cq_mask = ctx->cq_entries - 1; 11773ed29b0b4SJens Axboe unsigned int sq_head = READ_ONCE(r->sq.head); 11774ed29b0b4SJens Axboe unsigned int sq_tail = READ_ONCE(r->sq.tail); 11775ed29b0b4SJens Axboe unsigned int cq_head = READ_ONCE(r->cq.head); 11776ed29b0b4SJens Axboe unsigned int cq_tail = READ_ONCE(r->cq.tail); 11777ed29b0b4SJens Axboe unsigned int cq_shift = 0; 11778ed29b0b4SJens Axboe unsigned int sq_entries, cq_entries; 11779ed29b0b4SJens Axboe bool has_lock; 11780ed29b0b4SJens Axboe bool is_cqe32 = (ctx->flags & IORING_SETUP_CQE32); 11781ed29b0b4SJens Axboe unsigned int i; 11782ed29b0b4SJens Axboe 11783ed29b0b4SJens Axboe if (is_cqe32) 11784ed29b0b4SJens Axboe cq_shift = 1; 11785ed29b0b4SJens Axboe 11786ed29b0b4SJens Axboe /* 11787ed29b0b4SJens Axboe * we may get imprecise sqe and cqe info if uring is actively running 11788ed29b0b4SJens Axboe * since we get cached_sq_head and cached_cq_tail without uring_lock 11789ed29b0b4SJens Axboe * and sq_tail and cq_head are changed by userspace. But it's ok since 11790ed29b0b4SJens Axboe * we usually use these info when it is stuck. 11791ed29b0b4SJens Axboe */ 11792ed29b0b4SJens Axboe seq_printf(m, "SqMask:\t0x%x\n", sq_mask); 11793ed29b0b4SJens Axboe seq_printf(m, "SqHead:\t%u\n", sq_head); 11794ed29b0b4SJens Axboe seq_printf(m, "SqTail:\t%u\n", sq_tail); 11795ed29b0b4SJens Axboe seq_printf(m, "CachedSqHead:\t%u\n", ctx->cached_sq_head); 11796ed29b0b4SJens Axboe seq_printf(m, "CqMask:\t0x%x\n", cq_mask); 11797ed29b0b4SJens Axboe seq_printf(m, "CqHead:\t%u\n", cq_head); 11798ed29b0b4SJens Axboe seq_printf(m, "CqTail:\t%u\n", cq_tail); 11799ed29b0b4SJens Axboe seq_printf(m, "CachedCqTail:\t%u\n", ctx->cached_cq_tail); 11800ed29b0b4SJens Axboe seq_printf(m, "SQEs:\t%u\n", sq_tail - ctx->cached_sq_head); 11801ed29b0b4SJens Axboe sq_entries = min(sq_tail - sq_head, ctx->sq_entries); 11802ed29b0b4SJens Axboe for (i = 0; i < sq_entries; i++) { 11803ed29b0b4SJens Axboe unsigned int entry = i + sq_head; 11804ed29b0b4SJens Axboe unsigned int sq_idx = READ_ONCE(ctx->sq_array[entry & sq_mask]); 11805ed29b0b4SJens Axboe struct io_uring_sqe *sqe; 11806ed29b0b4SJens Axboe 11807ed29b0b4SJens Axboe if (sq_idx > sq_mask) 11808ed29b0b4SJens Axboe continue; 11809ed29b0b4SJens Axboe sqe = &ctx->sq_sqes[sq_idx]; 11810ed29b0b4SJens Axboe seq_printf(m, "%5u: opcode:%d, fd:%d, flags:%x, user_data:%llu\n", 11811ed29b0b4SJens Axboe sq_idx, sqe->opcode, sqe->fd, sqe->flags, 11812ed29b0b4SJens Axboe sqe->user_data); 11813ed29b0b4SJens Axboe } 11814ed29b0b4SJens Axboe seq_printf(m, "CQEs:\t%u\n", cq_tail - cq_head); 11815ed29b0b4SJens Axboe cq_entries = min(cq_tail - cq_head, ctx->cq_entries); 11816ed29b0b4SJens Axboe for (i = 0; i < cq_entries; i++) { 11817ed29b0b4SJens Axboe unsigned int entry = i + cq_head; 11818ed29b0b4SJens Axboe struct io_uring_cqe *cqe = &r->cqes[(entry & cq_mask) << cq_shift]; 11819ed29b0b4SJens Axboe 11820ed29b0b4SJens Axboe if (!is_cqe32) { 11821ed29b0b4SJens Axboe seq_printf(m, "%5u: user_data:%llu, res:%d, flag:%x\n", 11822ed29b0b4SJens Axboe entry & cq_mask, cqe->user_data, cqe->res, 11823ed29b0b4SJens Axboe cqe->flags); 11824ed29b0b4SJens Axboe } else { 11825ed29b0b4SJens Axboe seq_printf(m, "%5u: user_data:%llu, res:%d, flag:%x, " 11826ed29b0b4SJens Axboe "extra1:%llu, extra2:%llu\n", 11827ed29b0b4SJens Axboe entry & cq_mask, cqe->user_data, cqe->res, 11828ed29b0b4SJens Axboe cqe->flags, cqe->big_cqe[0], cqe->big_cqe[1]); 11829ed29b0b4SJens Axboe } 11830ed29b0b4SJens Axboe } 11831ed29b0b4SJens Axboe 11832ed29b0b4SJens Axboe /* 11833ed29b0b4SJens Axboe * Avoid ABBA deadlock between the seq lock and the io_uring mutex, 11834ed29b0b4SJens Axboe * since fdinfo case grabs it in the opposite direction of normal use 11835ed29b0b4SJens Axboe * cases. If we fail to get the lock, we just don't iterate any 11836ed29b0b4SJens Axboe * structures that could be going away outside the io_uring mutex. 11837ed29b0b4SJens Axboe */ 11838ed29b0b4SJens Axboe has_lock = mutex_trylock(&ctx->uring_lock); 11839ed29b0b4SJens Axboe 11840ed29b0b4SJens Axboe if (has_lock && (ctx->flags & IORING_SETUP_SQPOLL)) { 11841ed29b0b4SJens Axboe sq = ctx->sq_data; 11842ed29b0b4SJens Axboe if (!sq->thread) 11843ed29b0b4SJens Axboe sq = NULL; 11844ed29b0b4SJens Axboe } 11845ed29b0b4SJens Axboe 11846ed29b0b4SJens Axboe seq_printf(m, "SqThread:\t%d\n", sq ? task_pid_nr(sq->thread) : -1); 11847ed29b0b4SJens Axboe seq_printf(m, "SqThreadCpu:\t%d\n", sq ? task_cpu(sq->thread) : -1); 11848ed29b0b4SJens Axboe seq_printf(m, "UserFiles:\t%u\n", ctx->nr_user_files); 11849ed29b0b4SJens Axboe for (i = 0; has_lock && i < ctx->nr_user_files; i++) { 11850ed29b0b4SJens Axboe struct file *f = io_file_from_index(ctx, i); 11851ed29b0b4SJens Axboe 11852ed29b0b4SJens Axboe if (f) 11853ed29b0b4SJens Axboe seq_printf(m, "%5u: %s\n", i, file_dentry(f)->d_iname); 11854ed29b0b4SJens Axboe else 11855ed29b0b4SJens Axboe seq_printf(m, "%5u: <none>\n", i); 11856ed29b0b4SJens Axboe } 11857ed29b0b4SJens Axboe seq_printf(m, "UserBufs:\t%u\n", ctx->nr_user_bufs); 11858ed29b0b4SJens Axboe for (i = 0; has_lock && i < ctx->nr_user_bufs; i++) { 11859ed29b0b4SJens Axboe struct io_mapped_ubuf *buf = ctx->user_bufs[i]; 11860ed29b0b4SJens Axboe unsigned int len = buf->ubuf_end - buf->ubuf; 11861ed29b0b4SJens Axboe 11862ed29b0b4SJens Axboe seq_printf(m, "%5u: 0x%llx/%u\n", i, buf->ubuf, len); 11863ed29b0b4SJens Axboe } 11864ed29b0b4SJens Axboe if (has_lock && !xa_empty(&ctx->personalities)) { 11865ed29b0b4SJens Axboe unsigned long index; 11866ed29b0b4SJens Axboe const struct cred *cred; 11867ed29b0b4SJens Axboe 11868ed29b0b4SJens Axboe seq_printf(m, "Personalities:\n"); 11869ed29b0b4SJens Axboe xa_for_each(&ctx->personalities, index, cred) 11870ed29b0b4SJens Axboe io_uring_show_cred(m, index, cred); 11871ed29b0b4SJens Axboe } 11872ed29b0b4SJens Axboe if (has_lock) 11873ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 11874ed29b0b4SJens Axboe 11875ed29b0b4SJens Axboe seq_puts(m, "PollList:\n"); 11876ed29b0b4SJens Axboe spin_lock(&ctx->completion_lock); 11877ed29b0b4SJens Axboe for (i = 0; i < (1U << ctx->cancel_hash_bits); i++) { 11878ed29b0b4SJens Axboe struct hlist_head *list = &ctx->cancel_hash[i]; 11879ed29b0b4SJens Axboe struct io_kiocb *req; 11880ed29b0b4SJens Axboe 11881ed29b0b4SJens Axboe hlist_for_each_entry(req, list, hash_node) 11882ed29b0b4SJens Axboe seq_printf(m, " op=%d, task_works=%d\n", req->opcode, 11883ed29b0b4SJens Axboe task_work_pending(req->task)); 11884ed29b0b4SJens Axboe } 11885ed29b0b4SJens Axboe 11886ed29b0b4SJens Axboe seq_puts(m, "CqOverflowList:\n"); 11887ed29b0b4SJens Axboe list_for_each_entry(ocqe, &ctx->cq_overflow_list, list) { 11888ed29b0b4SJens Axboe struct io_uring_cqe *cqe = &ocqe->cqe; 11889ed29b0b4SJens Axboe 11890ed29b0b4SJens Axboe seq_printf(m, " user_data=%llu, res=%d, flags=%x\n", 11891ed29b0b4SJens Axboe cqe->user_data, cqe->res, cqe->flags); 11892ed29b0b4SJens Axboe 11893ed29b0b4SJens Axboe } 11894ed29b0b4SJens Axboe 11895ed29b0b4SJens Axboe spin_unlock(&ctx->completion_lock); 11896ed29b0b4SJens Axboe } 11897ed29b0b4SJens Axboe 11898ed29b0b4SJens Axboe static __cold void io_uring_show_fdinfo(struct seq_file *m, struct file *f) 11899ed29b0b4SJens Axboe { 11900ed29b0b4SJens Axboe struct io_ring_ctx *ctx = f->private_data; 11901ed29b0b4SJens Axboe 11902ed29b0b4SJens Axboe if (percpu_ref_tryget(&ctx->refs)) { 11903ed29b0b4SJens Axboe __io_uring_show_fdinfo(ctx, m); 11904ed29b0b4SJens Axboe percpu_ref_put(&ctx->refs); 11905ed29b0b4SJens Axboe } 11906ed29b0b4SJens Axboe } 11907ed29b0b4SJens Axboe #endif 11908ed29b0b4SJens Axboe 11909ed29b0b4SJens Axboe static const struct file_operations io_uring_fops = { 11910ed29b0b4SJens Axboe .release = io_uring_release, 11911ed29b0b4SJens Axboe .mmap = io_uring_mmap, 11912ed29b0b4SJens Axboe #ifndef CONFIG_MMU 11913ed29b0b4SJens Axboe .get_unmapped_area = io_uring_nommu_get_unmapped_area, 11914ed29b0b4SJens Axboe .mmap_capabilities = io_uring_nommu_mmap_capabilities, 11915ed29b0b4SJens Axboe #endif 11916ed29b0b4SJens Axboe .poll = io_uring_poll, 11917ed29b0b4SJens Axboe #ifdef CONFIG_PROC_FS 11918ed29b0b4SJens Axboe .show_fdinfo = io_uring_show_fdinfo, 11919ed29b0b4SJens Axboe #endif 11920ed29b0b4SJens Axboe }; 11921ed29b0b4SJens Axboe 11922ed29b0b4SJens Axboe static __cold int io_allocate_scq_urings(struct io_ring_ctx *ctx, 11923ed29b0b4SJens Axboe struct io_uring_params *p) 11924ed29b0b4SJens Axboe { 11925ed29b0b4SJens Axboe struct io_rings *rings; 11926ed29b0b4SJens Axboe size_t size, sq_array_offset; 11927ed29b0b4SJens Axboe 11928ed29b0b4SJens Axboe /* make sure these are sane, as we already accounted them */ 11929ed29b0b4SJens Axboe ctx->sq_entries = p->sq_entries; 11930ed29b0b4SJens Axboe ctx->cq_entries = p->cq_entries; 11931ed29b0b4SJens Axboe 11932ed29b0b4SJens Axboe size = rings_size(ctx, p->sq_entries, p->cq_entries, &sq_array_offset); 11933ed29b0b4SJens Axboe if (size == SIZE_MAX) 11934ed29b0b4SJens Axboe return -EOVERFLOW; 11935ed29b0b4SJens Axboe 11936ed29b0b4SJens Axboe rings = io_mem_alloc(size); 11937ed29b0b4SJens Axboe if (!rings) 11938ed29b0b4SJens Axboe return -ENOMEM; 11939ed29b0b4SJens Axboe 11940ed29b0b4SJens Axboe ctx->rings = rings; 11941ed29b0b4SJens Axboe ctx->sq_array = (u32 *)((char *)rings + sq_array_offset); 11942ed29b0b4SJens Axboe rings->sq_ring_mask = p->sq_entries - 1; 11943ed29b0b4SJens Axboe rings->cq_ring_mask = p->cq_entries - 1; 11944ed29b0b4SJens Axboe rings->sq_ring_entries = p->sq_entries; 11945ed29b0b4SJens Axboe rings->cq_ring_entries = p->cq_entries; 11946ed29b0b4SJens Axboe 11947ed29b0b4SJens Axboe if (p->flags & IORING_SETUP_SQE128) 11948ed29b0b4SJens Axboe size = array_size(2 * sizeof(struct io_uring_sqe), p->sq_entries); 11949ed29b0b4SJens Axboe else 11950ed29b0b4SJens Axboe size = array_size(sizeof(struct io_uring_sqe), p->sq_entries); 11951ed29b0b4SJens Axboe if (size == SIZE_MAX) { 11952ed29b0b4SJens Axboe io_mem_free(ctx->rings); 11953ed29b0b4SJens Axboe ctx->rings = NULL; 11954ed29b0b4SJens Axboe return -EOVERFLOW; 11955ed29b0b4SJens Axboe } 11956ed29b0b4SJens Axboe 11957ed29b0b4SJens Axboe ctx->sq_sqes = io_mem_alloc(size); 11958ed29b0b4SJens Axboe if (!ctx->sq_sqes) { 11959ed29b0b4SJens Axboe io_mem_free(ctx->rings); 11960ed29b0b4SJens Axboe ctx->rings = NULL; 11961ed29b0b4SJens Axboe return -ENOMEM; 11962ed29b0b4SJens Axboe } 11963ed29b0b4SJens Axboe 11964ed29b0b4SJens Axboe return 0; 11965ed29b0b4SJens Axboe } 11966ed29b0b4SJens Axboe 11967ed29b0b4SJens Axboe static int io_uring_install_fd(struct io_ring_ctx *ctx, struct file *file) 11968ed29b0b4SJens Axboe { 11969ed29b0b4SJens Axboe int ret, fd; 11970ed29b0b4SJens Axboe 11971ed29b0b4SJens Axboe fd = get_unused_fd_flags(O_RDWR | O_CLOEXEC); 11972ed29b0b4SJens Axboe if (fd < 0) 11973ed29b0b4SJens Axboe return fd; 11974ed29b0b4SJens Axboe 11975ed29b0b4SJens Axboe ret = io_uring_add_tctx_node(ctx); 11976ed29b0b4SJens Axboe if (ret) { 11977ed29b0b4SJens Axboe put_unused_fd(fd); 11978ed29b0b4SJens Axboe return ret; 11979ed29b0b4SJens Axboe } 11980ed29b0b4SJens Axboe fd_install(fd, file); 11981ed29b0b4SJens Axboe return fd; 11982ed29b0b4SJens Axboe } 11983ed29b0b4SJens Axboe 11984ed29b0b4SJens Axboe /* 11985ed29b0b4SJens Axboe * Allocate an anonymous fd, this is what constitutes the application 11986ed29b0b4SJens Axboe * visible backing of an io_uring instance. The application mmaps this 11987ed29b0b4SJens Axboe * fd to gain access to the SQ/CQ ring details. If UNIX sockets are enabled, 11988ed29b0b4SJens Axboe * we have to tie this fd to a socket for file garbage collection purposes. 11989ed29b0b4SJens Axboe */ 11990ed29b0b4SJens Axboe static struct file *io_uring_get_file(struct io_ring_ctx *ctx) 11991ed29b0b4SJens Axboe { 11992ed29b0b4SJens Axboe struct file *file; 11993ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 11994ed29b0b4SJens Axboe int ret; 11995ed29b0b4SJens Axboe 11996ed29b0b4SJens Axboe ret = sock_create_kern(&init_net, PF_UNIX, SOCK_RAW, IPPROTO_IP, 11997ed29b0b4SJens Axboe &ctx->ring_sock); 11998ed29b0b4SJens Axboe if (ret) 11999ed29b0b4SJens Axboe return ERR_PTR(ret); 12000ed29b0b4SJens Axboe #endif 12001ed29b0b4SJens Axboe 12002ed29b0b4SJens Axboe file = anon_inode_getfile_secure("[io_uring]", &io_uring_fops, ctx, 12003ed29b0b4SJens Axboe O_RDWR | O_CLOEXEC, NULL); 12004ed29b0b4SJens Axboe #if defined(CONFIG_UNIX) 12005ed29b0b4SJens Axboe if (IS_ERR(file)) { 12006ed29b0b4SJens Axboe sock_release(ctx->ring_sock); 12007ed29b0b4SJens Axboe ctx->ring_sock = NULL; 12008ed29b0b4SJens Axboe } else { 12009ed29b0b4SJens Axboe ctx->ring_sock->file = file; 12010ed29b0b4SJens Axboe } 12011ed29b0b4SJens Axboe #endif 12012ed29b0b4SJens Axboe return file; 12013ed29b0b4SJens Axboe } 12014ed29b0b4SJens Axboe 12015ed29b0b4SJens Axboe static __cold int io_uring_create(unsigned entries, struct io_uring_params *p, 12016ed29b0b4SJens Axboe struct io_uring_params __user *params) 12017ed29b0b4SJens Axboe { 12018ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 12019ed29b0b4SJens Axboe struct file *file; 12020ed29b0b4SJens Axboe int ret; 12021ed29b0b4SJens Axboe 12022ed29b0b4SJens Axboe if (!entries) 12023ed29b0b4SJens Axboe return -EINVAL; 12024ed29b0b4SJens Axboe if (entries > IORING_MAX_ENTRIES) { 12025ed29b0b4SJens Axboe if (!(p->flags & IORING_SETUP_CLAMP)) 12026ed29b0b4SJens Axboe return -EINVAL; 12027ed29b0b4SJens Axboe entries = IORING_MAX_ENTRIES; 12028ed29b0b4SJens Axboe } 12029ed29b0b4SJens Axboe 12030ed29b0b4SJens Axboe /* 12031ed29b0b4SJens Axboe * Use twice as many entries for the CQ ring. It's possible for the 12032ed29b0b4SJens Axboe * application to drive a higher depth than the size of the SQ ring, 12033ed29b0b4SJens Axboe * since the sqes are only used at submission time. This allows for 12034ed29b0b4SJens Axboe * some flexibility in overcommitting a bit. If the application has 12035ed29b0b4SJens Axboe * set IORING_SETUP_CQSIZE, it will have passed in the desired number 12036ed29b0b4SJens Axboe * of CQ ring entries manually. 12037ed29b0b4SJens Axboe */ 12038ed29b0b4SJens Axboe p->sq_entries = roundup_pow_of_two(entries); 12039ed29b0b4SJens Axboe if (p->flags & IORING_SETUP_CQSIZE) { 12040ed29b0b4SJens Axboe /* 12041ed29b0b4SJens Axboe * If IORING_SETUP_CQSIZE is set, we do the same roundup 12042ed29b0b4SJens Axboe * to a power-of-two, if it isn't already. We do NOT impose 12043ed29b0b4SJens Axboe * any cq vs sq ring sizing. 12044ed29b0b4SJens Axboe */ 12045ed29b0b4SJens Axboe if (!p->cq_entries) 12046ed29b0b4SJens Axboe return -EINVAL; 12047ed29b0b4SJens Axboe if (p->cq_entries > IORING_MAX_CQ_ENTRIES) { 12048ed29b0b4SJens Axboe if (!(p->flags & IORING_SETUP_CLAMP)) 12049ed29b0b4SJens Axboe return -EINVAL; 12050ed29b0b4SJens Axboe p->cq_entries = IORING_MAX_CQ_ENTRIES; 12051ed29b0b4SJens Axboe } 12052ed29b0b4SJens Axboe p->cq_entries = roundup_pow_of_two(p->cq_entries); 12053ed29b0b4SJens Axboe if (p->cq_entries < p->sq_entries) 12054ed29b0b4SJens Axboe return -EINVAL; 12055ed29b0b4SJens Axboe } else { 12056ed29b0b4SJens Axboe p->cq_entries = 2 * p->sq_entries; 12057ed29b0b4SJens Axboe } 12058ed29b0b4SJens Axboe 12059ed29b0b4SJens Axboe ctx = io_ring_ctx_alloc(p); 12060ed29b0b4SJens Axboe if (!ctx) 12061ed29b0b4SJens Axboe return -ENOMEM; 12062ed29b0b4SJens Axboe 12063ed29b0b4SJens Axboe /* 12064ed29b0b4SJens Axboe * When SETUP_IOPOLL and SETUP_SQPOLL are both enabled, user 12065ed29b0b4SJens Axboe * space applications don't need to do io completion events 12066ed29b0b4SJens Axboe * polling again, they can rely on io_sq_thread to do polling 12067ed29b0b4SJens Axboe * work, which can reduce cpu usage and uring_lock contention. 12068ed29b0b4SJens Axboe */ 12069ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_IOPOLL && 12070ed29b0b4SJens Axboe !(ctx->flags & IORING_SETUP_SQPOLL)) 12071ed29b0b4SJens Axboe ctx->syscall_iopoll = 1; 12072ed29b0b4SJens Axboe 12073ed29b0b4SJens Axboe ctx->compat = in_compat_syscall(); 12074ed29b0b4SJens Axboe if (!capable(CAP_IPC_LOCK)) 12075ed29b0b4SJens Axboe ctx->user = get_uid(current_user()); 12076ed29b0b4SJens Axboe 12077ed29b0b4SJens Axboe /* 12078ed29b0b4SJens Axboe * For SQPOLL, we just need a wakeup, always. For !SQPOLL, if 12079ed29b0b4SJens Axboe * COOP_TASKRUN is set, then IPIs are never needed by the app. 12080ed29b0b4SJens Axboe */ 12081ed29b0b4SJens Axboe ret = -EINVAL; 12082ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQPOLL) { 12083ed29b0b4SJens Axboe /* IPI related flags don't make sense with SQPOLL */ 12084ed29b0b4SJens Axboe if (ctx->flags & (IORING_SETUP_COOP_TASKRUN | 12085ed29b0b4SJens Axboe IORING_SETUP_TASKRUN_FLAG)) 12086ed29b0b4SJens Axboe goto err; 12087ed29b0b4SJens Axboe ctx->notify_method = TWA_SIGNAL_NO_IPI; 12088ed29b0b4SJens Axboe } else if (ctx->flags & IORING_SETUP_COOP_TASKRUN) { 12089ed29b0b4SJens Axboe ctx->notify_method = TWA_SIGNAL_NO_IPI; 12090ed29b0b4SJens Axboe } else { 12091ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_TASKRUN_FLAG) 12092ed29b0b4SJens Axboe goto err; 12093ed29b0b4SJens Axboe ctx->notify_method = TWA_SIGNAL; 12094ed29b0b4SJens Axboe } 12095ed29b0b4SJens Axboe 12096ed29b0b4SJens Axboe /* 12097ed29b0b4SJens Axboe * This is just grabbed for accounting purposes. When a process exits, 12098ed29b0b4SJens Axboe * the mm is exited and dropped before the files, hence we need to hang 12099ed29b0b4SJens Axboe * on to this mm purely for the purposes of being able to unaccount 12100ed29b0b4SJens Axboe * memory (locked/pinned vm). It's not used for anything else. 12101ed29b0b4SJens Axboe */ 12102ed29b0b4SJens Axboe mmgrab(current->mm); 12103ed29b0b4SJens Axboe ctx->mm_account = current->mm; 12104ed29b0b4SJens Axboe 12105ed29b0b4SJens Axboe ret = io_allocate_scq_urings(ctx, p); 12106ed29b0b4SJens Axboe if (ret) 12107ed29b0b4SJens Axboe goto err; 12108ed29b0b4SJens Axboe 12109ed29b0b4SJens Axboe ret = io_sq_offload_create(ctx, p); 12110ed29b0b4SJens Axboe if (ret) 12111ed29b0b4SJens Axboe goto err; 12112ed29b0b4SJens Axboe /* always set a rsrc node */ 12113ed29b0b4SJens Axboe ret = io_rsrc_node_switch_start(ctx); 12114ed29b0b4SJens Axboe if (ret) 12115ed29b0b4SJens Axboe goto err; 12116ed29b0b4SJens Axboe io_rsrc_node_switch(ctx, NULL); 12117ed29b0b4SJens Axboe 12118ed29b0b4SJens Axboe memset(&p->sq_off, 0, sizeof(p->sq_off)); 12119ed29b0b4SJens Axboe p->sq_off.head = offsetof(struct io_rings, sq.head); 12120ed29b0b4SJens Axboe p->sq_off.tail = offsetof(struct io_rings, sq.tail); 12121ed29b0b4SJens Axboe p->sq_off.ring_mask = offsetof(struct io_rings, sq_ring_mask); 12122ed29b0b4SJens Axboe p->sq_off.ring_entries = offsetof(struct io_rings, sq_ring_entries); 12123ed29b0b4SJens Axboe p->sq_off.flags = offsetof(struct io_rings, sq_flags); 12124ed29b0b4SJens Axboe p->sq_off.dropped = offsetof(struct io_rings, sq_dropped); 12125ed29b0b4SJens Axboe p->sq_off.array = (char *)ctx->sq_array - (char *)ctx->rings; 12126ed29b0b4SJens Axboe 12127ed29b0b4SJens Axboe memset(&p->cq_off, 0, sizeof(p->cq_off)); 12128ed29b0b4SJens Axboe p->cq_off.head = offsetof(struct io_rings, cq.head); 12129ed29b0b4SJens Axboe p->cq_off.tail = offsetof(struct io_rings, cq.tail); 12130ed29b0b4SJens Axboe p->cq_off.ring_mask = offsetof(struct io_rings, cq_ring_mask); 12131ed29b0b4SJens Axboe p->cq_off.ring_entries = offsetof(struct io_rings, cq_ring_entries); 12132ed29b0b4SJens Axboe p->cq_off.overflow = offsetof(struct io_rings, cq_overflow); 12133ed29b0b4SJens Axboe p->cq_off.cqes = offsetof(struct io_rings, cqes); 12134ed29b0b4SJens Axboe p->cq_off.flags = offsetof(struct io_rings, cq_flags); 12135ed29b0b4SJens Axboe 12136ed29b0b4SJens Axboe p->features = IORING_FEAT_SINGLE_MMAP | IORING_FEAT_NODROP | 12137ed29b0b4SJens Axboe IORING_FEAT_SUBMIT_STABLE | IORING_FEAT_RW_CUR_POS | 12138ed29b0b4SJens Axboe IORING_FEAT_CUR_PERSONALITY | IORING_FEAT_FAST_POLL | 12139ed29b0b4SJens Axboe IORING_FEAT_POLL_32BITS | IORING_FEAT_SQPOLL_NONFIXED | 12140ed29b0b4SJens Axboe IORING_FEAT_EXT_ARG | IORING_FEAT_NATIVE_WORKERS | 12141ed29b0b4SJens Axboe IORING_FEAT_RSRC_TAGS | IORING_FEAT_CQE_SKIP | 12142ed29b0b4SJens Axboe IORING_FEAT_LINKED_FILE; 12143ed29b0b4SJens Axboe 12144ed29b0b4SJens Axboe if (copy_to_user(params, p, sizeof(*p))) { 12145ed29b0b4SJens Axboe ret = -EFAULT; 12146ed29b0b4SJens Axboe goto err; 12147ed29b0b4SJens Axboe } 12148ed29b0b4SJens Axboe 12149ed29b0b4SJens Axboe file = io_uring_get_file(ctx); 12150ed29b0b4SJens Axboe if (IS_ERR(file)) { 12151ed29b0b4SJens Axboe ret = PTR_ERR(file); 12152ed29b0b4SJens Axboe goto err; 12153ed29b0b4SJens Axboe } 12154ed29b0b4SJens Axboe 12155ed29b0b4SJens Axboe /* 12156ed29b0b4SJens Axboe * Install ring fd as the very last thing, so we don't risk someone 12157ed29b0b4SJens Axboe * having closed it before we finish setup 12158ed29b0b4SJens Axboe */ 12159ed29b0b4SJens Axboe ret = io_uring_install_fd(ctx, file); 12160ed29b0b4SJens Axboe if (ret < 0) { 12161ed29b0b4SJens Axboe /* fput will clean it up */ 12162ed29b0b4SJens Axboe fput(file); 12163ed29b0b4SJens Axboe return ret; 12164ed29b0b4SJens Axboe } 12165ed29b0b4SJens Axboe 12166ed29b0b4SJens Axboe trace_io_uring_create(ret, ctx, p->sq_entries, p->cq_entries, p->flags); 12167ed29b0b4SJens Axboe return ret; 12168ed29b0b4SJens Axboe err: 12169ed29b0b4SJens Axboe io_ring_ctx_wait_and_kill(ctx); 12170ed29b0b4SJens Axboe return ret; 12171ed29b0b4SJens Axboe } 12172ed29b0b4SJens Axboe 12173ed29b0b4SJens Axboe /* 12174ed29b0b4SJens Axboe * Sets up an aio uring context, and returns the fd. Applications asks for a 12175ed29b0b4SJens Axboe * ring size, we return the actual sq/cq ring sizes (among other things) in the 12176ed29b0b4SJens Axboe * params structure passed in. 12177ed29b0b4SJens Axboe */ 12178ed29b0b4SJens Axboe static long io_uring_setup(u32 entries, struct io_uring_params __user *params) 12179ed29b0b4SJens Axboe { 12180ed29b0b4SJens Axboe struct io_uring_params p; 12181ed29b0b4SJens Axboe int i; 12182ed29b0b4SJens Axboe 12183ed29b0b4SJens Axboe if (copy_from_user(&p, params, sizeof(p))) 12184ed29b0b4SJens Axboe return -EFAULT; 12185ed29b0b4SJens Axboe for (i = 0; i < ARRAY_SIZE(p.resv); i++) { 12186ed29b0b4SJens Axboe if (p.resv[i]) 12187ed29b0b4SJens Axboe return -EINVAL; 12188ed29b0b4SJens Axboe } 12189ed29b0b4SJens Axboe 12190ed29b0b4SJens Axboe if (p.flags & ~(IORING_SETUP_IOPOLL | IORING_SETUP_SQPOLL | 12191ed29b0b4SJens Axboe IORING_SETUP_SQ_AFF | IORING_SETUP_CQSIZE | 12192ed29b0b4SJens Axboe IORING_SETUP_CLAMP | IORING_SETUP_ATTACH_WQ | 12193ed29b0b4SJens Axboe IORING_SETUP_R_DISABLED | IORING_SETUP_SUBMIT_ALL | 12194ed29b0b4SJens Axboe IORING_SETUP_COOP_TASKRUN | IORING_SETUP_TASKRUN_FLAG | 12195ed29b0b4SJens Axboe IORING_SETUP_SQE128 | IORING_SETUP_CQE32)) 12196ed29b0b4SJens Axboe return -EINVAL; 12197ed29b0b4SJens Axboe 12198ed29b0b4SJens Axboe return io_uring_create(entries, &p, params); 12199ed29b0b4SJens Axboe } 12200ed29b0b4SJens Axboe 12201ed29b0b4SJens Axboe SYSCALL_DEFINE2(io_uring_setup, u32, entries, 12202ed29b0b4SJens Axboe struct io_uring_params __user *, params) 12203ed29b0b4SJens Axboe { 12204ed29b0b4SJens Axboe return io_uring_setup(entries, params); 12205ed29b0b4SJens Axboe } 12206ed29b0b4SJens Axboe 12207ed29b0b4SJens Axboe static __cold int io_probe(struct io_ring_ctx *ctx, void __user *arg, 12208ed29b0b4SJens Axboe unsigned nr_args) 12209ed29b0b4SJens Axboe { 12210ed29b0b4SJens Axboe struct io_uring_probe *p; 12211ed29b0b4SJens Axboe size_t size; 12212ed29b0b4SJens Axboe int i, ret; 12213ed29b0b4SJens Axboe 12214ed29b0b4SJens Axboe size = struct_size(p, ops, nr_args); 12215ed29b0b4SJens Axboe if (size == SIZE_MAX) 12216ed29b0b4SJens Axboe return -EOVERFLOW; 12217ed29b0b4SJens Axboe p = kzalloc(size, GFP_KERNEL); 12218ed29b0b4SJens Axboe if (!p) 12219ed29b0b4SJens Axboe return -ENOMEM; 12220ed29b0b4SJens Axboe 12221ed29b0b4SJens Axboe ret = -EFAULT; 12222ed29b0b4SJens Axboe if (copy_from_user(p, arg, size)) 12223ed29b0b4SJens Axboe goto out; 12224ed29b0b4SJens Axboe ret = -EINVAL; 12225ed29b0b4SJens Axboe if (memchr_inv(p, 0, size)) 12226ed29b0b4SJens Axboe goto out; 12227ed29b0b4SJens Axboe 12228ed29b0b4SJens Axboe p->last_op = IORING_OP_LAST - 1; 12229ed29b0b4SJens Axboe if (nr_args > IORING_OP_LAST) 12230ed29b0b4SJens Axboe nr_args = IORING_OP_LAST; 12231ed29b0b4SJens Axboe 12232ed29b0b4SJens Axboe for (i = 0; i < nr_args; i++) { 12233ed29b0b4SJens Axboe p->ops[i].op = i; 12234ed29b0b4SJens Axboe if (!io_op_defs[i].not_supported) 12235ed29b0b4SJens Axboe p->ops[i].flags = IO_URING_OP_SUPPORTED; 12236ed29b0b4SJens Axboe } 12237ed29b0b4SJens Axboe p->ops_len = i; 12238ed29b0b4SJens Axboe 12239ed29b0b4SJens Axboe ret = 0; 12240ed29b0b4SJens Axboe if (copy_to_user(arg, p, size)) 12241ed29b0b4SJens Axboe ret = -EFAULT; 12242ed29b0b4SJens Axboe out: 12243ed29b0b4SJens Axboe kfree(p); 12244ed29b0b4SJens Axboe return ret; 12245ed29b0b4SJens Axboe } 12246ed29b0b4SJens Axboe 12247ed29b0b4SJens Axboe static int io_register_personality(struct io_ring_ctx *ctx) 12248ed29b0b4SJens Axboe { 12249ed29b0b4SJens Axboe const struct cred *creds; 12250ed29b0b4SJens Axboe u32 id; 12251ed29b0b4SJens Axboe int ret; 12252ed29b0b4SJens Axboe 12253ed29b0b4SJens Axboe creds = get_current_cred(); 12254ed29b0b4SJens Axboe 12255ed29b0b4SJens Axboe ret = xa_alloc_cyclic(&ctx->personalities, &id, (void *)creds, 12256ed29b0b4SJens Axboe XA_LIMIT(0, USHRT_MAX), &ctx->pers_next, GFP_KERNEL); 12257ed29b0b4SJens Axboe if (ret < 0) { 12258ed29b0b4SJens Axboe put_cred(creds); 12259ed29b0b4SJens Axboe return ret; 12260ed29b0b4SJens Axboe } 12261ed29b0b4SJens Axboe return id; 12262ed29b0b4SJens Axboe } 12263ed29b0b4SJens Axboe 12264ed29b0b4SJens Axboe static __cold int io_register_restrictions(struct io_ring_ctx *ctx, 12265ed29b0b4SJens Axboe void __user *arg, unsigned int nr_args) 12266ed29b0b4SJens Axboe { 12267ed29b0b4SJens Axboe struct io_uring_restriction *res; 12268ed29b0b4SJens Axboe size_t size; 12269ed29b0b4SJens Axboe int i, ret; 12270ed29b0b4SJens Axboe 12271ed29b0b4SJens Axboe /* Restrictions allowed only if rings started disabled */ 12272ed29b0b4SJens Axboe if (!(ctx->flags & IORING_SETUP_R_DISABLED)) 12273ed29b0b4SJens Axboe return -EBADFD; 12274ed29b0b4SJens Axboe 12275ed29b0b4SJens Axboe /* We allow only a single restrictions registration */ 12276ed29b0b4SJens Axboe if (ctx->restrictions.registered) 12277ed29b0b4SJens Axboe return -EBUSY; 12278ed29b0b4SJens Axboe 12279ed29b0b4SJens Axboe if (!arg || nr_args > IORING_MAX_RESTRICTIONS) 12280ed29b0b4SJens Axboe return -EINVAL; 12281ed29b0b4SJens Axboe 12282ed29b0b4SJens Axboe size = array_size(nr_args, sizeof(*res)); 12283ed29b0b4SJens Axboe if (size == SIZE_MAX) 12284ed29b0b4SJens Axboe return -EOVERFLOW; 12285ed29b0b4SJens Axboe 12286ed29b0b4SJens Axboe res = memdup_user(arg, size); 12287ed29b0b4SJens Axboe if (IS_ERR(res)) 12288ed29b0b4SJens Axboe return PTR_ERR(res); 12289ed29b0b4SJens Axboe 12290ed29b0b4SJens Axboe ret = 0; 12291ed29b0b4SJens Axboe 12292ed29b0b4SJens Axboe for (i = 0; i < nr_args; i++) { 12293ed29b0b4SJens Axboe switch (res[i].opcode) { 12294ed29b0b4SJens Axboe case IORING_RESTRICTION_REGISTER_OP: 12295ed29b0b4SJens Axboe if (res[i].register_op >= IORING_REGISTER_LAST) { 12296ed29b0b4SJens Axboe ret = -EINVAL; 12297ed29b0b4SJens Axboe goto out; 12298ed29b0b4SJens Axboe } 12299ed29b0b4SJens Axboe 12300ed29b0b4SJens Axboe __set_bit(res[i].register_op, 12301ed29b0b4SJens Axboe ctx->restrictions.register_op); 12302ed29b0b4SJens Axboe break; 12303ed29b0b4SJens Axboe case IORING_RESTRICTION_SQE_OP: 12304ed29b0b4SJens Axboe if (res[i].sqe_op >= IORING_OP_LAST) { 12305ed29b0b4SJens Axboe ret = -EINVAL; 12306ed29b0b4SJens Axboe goto out; 12307ed29b0b4SJens Axboe } 12308ed29b0b4SJens Axboe 12309ed29b0b4SJens Axboe __set_bit(res[i].sqe_op, ctx->restrictions.sqe_op); 12310ed29b0b4SJens Axboe break; 12311ed29b0b4SJens Axboe case IORING_RESTRICTION_SQE_FLAGS_ALLOWED: 12312ed29b0b4SJens Axboe ctx->restrictions.sqe_flags_allowed = res[i].sqe_flags; 12313ed29b0b4SJens Axboe break; 12314ed29b0b4SJens Axboe case IORING_RESTRICTION_SQE_FLAGS_REQUIRED: 12315ed29b0b4SJens Axboe ctx->restrictions.sqe_flags_required = res[i].sqe_flags; 12316ed29b0b4SJens Axboe break; 12317ed29b0b4SJens Axboe default: 12318ed29b0b4SJens Axboe ret = -EINVAL; 12319ed29b0b4SJens Axboe goto out; 12320ed29b0b4SJens Axboe } 12321ed29b0b4SJens Axboe } 12322ed29b0b4SJens Axboe 12323ed29b0b4SJens Axboe out: 12324ed29b0b4SJens Axboe /* Reset all restrictions if an error happened */ 12325ed29b0b4SJens Axboe if (ret != 0) 12326ed29b0b4SJens Axboe memset(&ctx->restrictions, 0, sizeof(ctx->restrictions)); 12327ed29b0b4SJens Axboe else 12328ed29b0b4SJens Axboe ctx->restrictions.registered = true; 12329ed29b0b4SJens Axboe 12330ed29b0b4SJens Axboe kfree(res); 12331ed29b0b4SJens Axboe return ret; 12332ed29b0b4SJens Axboe } 12333ed29b0b4SJens Axboe 12334ed29b0b4SJens Axboe static int io_register_enable_rings(struct io_ring_ctx *ctx) 12335ed29b0b4SJens Axboe { 12336ed29b0b4SJens Axboe if (!(ctx->flags & IORING_SETUP_R_DISABLED)) 12337ed29b0b4SJens Axboe return -EBADFD; 12338ed29b0b4SJens Axboe 12339ed29b0b4SJens Axboe if (ctx->restrictions.registered) 12340ed29b0b4SJens Axboe ctx->restricted = 1; 12341ed29b0b4SJens Axboe 12342ed29b0b4SJens Axboe ctx->flags &= ~IORING_SETUP_R_DISABLED; 12343ed29b0b4SJens Axboe if (ctx->sq_data && wq_has_sleeper(&ctx->sq_data->wait)) 12344ed29b0b4SJens Axboe wake_up(&ctx->sq_data->wait); 12345ed29b0b4SJens Axboe return 0; 12346ed29b0b4SJens Axboe } 12347ed29b0b4SJens Axboe 12348ed29b0b4SJens Axboe static int __io_register_rsrc_update(struct io_ring_ctx *ctx, unsigned type, 12349ed29b0b4SJens Axboe struct io_uring_rsrc_update2 *up, 12350ed29b0b4SJens Axboe unsigned nr_args) 12351ed29b0b4SJens Axboe { 12352ed29b0b4SJens Axboe __u32 tmp; 12353ed29b0b4SJens Axboe int err; 12354ed29b0b4SJens Axboe 12355ed29b0b4SJens Axboe if (check_add_overflow(up->offset, nr_args, &tmp)) 12356ed29b0b4SJens Axboe return -EOVERFLOW; 12357ed29b0b4SJens Axboe err = io_rsrc_node_switch_start(ctx); 12358ed29b0b4SJens Axboe if (err) 12359ed29b0b4SJens Axboe return err; 12360ed29b0b4SJens Axboe 12361ed29b0b4SJens Axboe switch (type) { 12362ed29b0b4SJens Axboe case IORING_RSRC_FILE: 12363ed29b0b4SJens Axboe return __io_sqe_files_update(ctx, up, nr_args); 12364ed29b0b4SJens Axboe case IORING_RSRC_BUFFER: 12365ed29b0b4SJens Axboe return __io_sqe_buffers_update(ctx, up, nr_args); 12366ed29b0b4SJens Axboe } 12367ed29b0b4SJens Axboe return -EINVAL; 12368ed29b0b4SJens Axboe } 12369ed29b0b4SJens Axboe 12370ed29b0b4SJens Axboe static int io_register_files_update(struct io_ring_ctx *ctx, void __user *arg, 12371ed29b0b4SJens Axboe unsigned nr_args) 12372ed29b0b4SJens Axboe { 12373ed29b0b4SJens Axboe struct io_uring_rsrc_update2 up; 12374ed29b0b4SJens Axboe 12375ed29b0b4SJens Axboe if (!nr_args) 12376ed29b0b4SJens Axboe return -EINVAL; 12377ed29b0b4SJens Axboe memset(&up, 0, sizeof(up)); 12378ed29b0b4SJens Axboe if (copy_from_user(&up, arg, sizeof(struct io_uring_rsrc_update))) 12379ed29b0b4SJens Axboe return -EFAULT; 12380ed29b0b4SJens Axboe if (up.resv || up.resv2) 12381ed29b0b4SJens Axboe return -EINVAL; 12382ed29b0b4SJens Axboe return __io_register_rsrc_update(ctx, IORING_RSRC_FILE, &up, nr_args); 12383ed29b0b4SJens Axboe } 12384ed29b0b4SJens Axboe 12385ed29b0b4SJens Axboe static int io_register_rsrc_update(struct io_ring_ctx *ctx, void __user *arg, 12386ed29b0b4SJens Axboe unsigned size, unsigned type) 12387ed29b0b4SJens Axboe { 12388ed29b0b4SJens Axboe struct io_uring_rsrc_update2 up; 12389ed29b0b4SJens Axboe 12390ed29b0b4SJens Axboe if (size != sizeof(up)) 12391ed29b0b4SJens Axboe return -EINVAL; 12392ed29b0b4SJens Axboe if (copy_from_user(&up, arg, sizeof(up))) 12393ed29b0b4SJens Axboe return -EFAULT; 12394ed29b0b4SJens Axboe if (!up.nr || up.resv || up.resv2) 12395ed29b0b4SJens Axboe return -EINVAL; 12396ed29b0b4SJens Axboe return __io_register_rsrc_update(ctx, type, &up, up.nr); 12397ed29b0b4SJens Axboe } 12398ed29b0b4SJens Axboe 12399ed29b0b4SJens Axboe static __cold int io_register_rsrc(struct io_ring_ctx *ctx, void __user *arg, 12400ed29b0b4SJens Axboe unsigned int size, unsigned int type) 12401ed29b0b4SJens Axboe { 12402ed29b0b4SJens Axboe struct io_uring_rsrc_register rr; 12403ed29b0b4SJens Axboe 12404ed29b0b4SJens Axboe /* keep it extendible */ 12405ed29b0b4SJens Axboe if (size != sizeof(rr)) 12406ed29b0b4SJens Axboe return -EINVAL; 12407ed29b0b4SJens Axboe 12408ed29b0b4SJens Axboe memset(&rr, 0, sizeof(rr)); 12409ed29b0b4SJens Axboe if (copy_from_user(&rr, arg, size)) 12410ed29b0b4SJens Axboe return -EFAULT; 12411ed29b0b4SJens Axboe if (!rr.nr || rr.resv2) 12412ed29b0b4SJens Axboe return -EINVAL; 12413ed29b0b4SJens Axboe if (rr.flags & ~IORING_RSRC_REGISTER_SPARSE) 12414ed29b0b4SJens Axboe return -EINVAL; 12415ed29b0b4SJens Axboe 12416ed29b0b4SJens Axboe switch (type) { 12417ed29b0b4SJens Axboe case IORING_RSRC_FILE: 12418ed29b0b4SJens Axboe if (rr.flags & IORING_RSRC_REGISTER_SPARSE && rr.data) 12419ed29b0b4SJens Axboe break; 12420ed29b0b4SJens Axboe return io_sqe_files_register(ctx, u64_to_user_ptr(rr.data), 12421ed29b0b4SJens Axboe rr.nr, u64_to_user_ptr(rr.tags)); 12422ed29b0b4SJens Axboe case IORING_RSRC_BUFFER: 12423ed29b0b4SJens Axboe if (rr.flags & IORING_RSRC_REGISTER_SPARSE && rr.data) 12424ed29b0b4SJens Axboe break; 12425ed29b0b4SJens Axboe return io_sqe_buffers_register(ctx, u64_to_user_ptr(rr.data), 12426ed29b0b4SJens Axboe rr.nr, u64_to_user_ptr(rr.tags)); 12427ed29b0b4SJens Axboe } 12428ed29b0b4SJens Axboe return -EINVAL; 12429ed29b0b4SJens Axboe } 12430ed29b0b4SJens Axboe 12431ed29b0b4SJens Axboe static __cold int io_register_iowq_aff(struct io_ring_ctx *ctx, 12432ed29b0b4SJens Axboe void __user *arg, unsigned len) 12433ed29b0b4SJens Axboe { 12434ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 12435ed29b0b4SJens Axboe cpumask_var_t new_mask; 12436ed29b0b4SJens Axboe int ret; 12437ed29b0b4SJens Axboe 12438ed29b0b4SJens Axboe if (!tctx || !tctx->io_wq) 12439ed29b0b4SJens Axboe return -EINVAL; 12440ed29b0b4SJens Axboe 12441ed29b0b4SJens Axboe if (!alloc_cpumask_var(&new_mask, GFP_KERNEL)) 12442ed29b0b4SJens Axboe return -ENOMEM; 12443ed29b0b4SJens Axboe 12444ed29b0b4SJens Axboe cpumask_clear(new_mask); 12445ed29b0b4SJens Axboe if (len > cpumask_size()) 12446ed29b0b4SJens Axboe len = cpumask_size(); 12447ed29b0b4SJens Axboe 12448ed29b0b4SJens Axboe if (in_compat_syscall()) { 12449ed29b0b4SJens Axboe ret = compat_get_bitmap(cpumask_bits(new_mask), 12450ed29b0b4SJens Axboe (const compat_ulong_t __user *)arg, 12451ed29b0b4SJens Axboe len * 8 /* CHAR_BIT */); 12452ed29b0b4SJens Axboe } else { 12453ed29b0b4SJens Axboe ret = copy_from_user(new_mask, arg, len); 12454ed29b0b4SJens Axboe } 12455ed29b0b4SJens Axboe 12456ed29b0b4SJens Axboe if (ret) { 12457ed29b0b4SJens Axboe free_cpumask_var(new_mask); 12458ed29b0b4SJens Axboe return -EFAULT; 12459ed29b0b4SJens Axboe } 12460ed29b0b4SJens Axboe 12461ed29b0b4SJens Axboe ret = io_wq_cpu_affinity(tctx->io_wq, new_mask); 12462ed29b0b4SJens Axboe free_cpumask_var(new_mask); 12463ed29b0b4SJens Axboe return ret; 12464ed29b0b4SJens Axboe } 12465ed29b0b4SJens Axboe 12466ed29b0b4SJens Axboe static __cold int io_unregister_iowq_aff(struct io_ring_ctx *ctx) 12467ed29b0b4SJens Axboe { 12468ed29b0b4SJens Axboe struct io_uring_task *tctx = current->io_uring; 12469ed29b0b4SJens Axboe 12470ed29b0b4SJens Axboe if (!tctx || !tctx->io_wq) 12471ed29b0b4SJens Axboe return -EINVAL; 12472ed29b0b4SJens Axboe 12473ed29b0b4SJens Axboe return io_wq_cpu_affinity(tctx->io_wq, NULL); 12474ed29b0b4SJens Axboe } 12475ed29b0b4SJens Axboe 12476ed29b0b4SJens Axboe static __cold int io_register_iowq_max_workers(struct io_ring_ctx *ctx, 12477ed29b0b4SJens Axboe void __user *arg) 12478ed29b0b4SJens Axboe __must_hold(&ctx->uring_lock) 12479ed29b0b4SJens Axboe { 12480ed29b0b4SJens Axboe struct io_tctx_node *node; 12481ed29b0b4SJens Axboe struct io_uring_task *tctx = NULL; 12482ed29b0b4SJens Axboe struct io_sq_data *sqd = NULL; 12483ed29b0b4SJens Axboe __u32 new_count[2]; 12484ed29b0b4SJens Axboe int i, ret; 12485ed29b0b4SJens Axboe 12486ed29b0b4SJens Axboe if (copy_from_user(new_count, arg, sizeof(new_count))) 12487ed29b0b4SJens Axboe return -EFAULT; 12488ed29b0b4SJens Axboe for (i = 0; i < ARRAY_SIZE(new_count); i++) 12489ed29b0b4SJens Axboe if (new_count[i] > INT_MAX) 12490ed29b0b4SJens Axboe return -EINVAL; 12491ed29b0b4SJens Axboe 12492ed29b0b4SJens Axboe if (ctx->flags & IORING_SETUP_SQPOLL) { 12493ed29b0b4SJens Axboe sqd = ctx->sq_data; 12494ed29b0b4SJens Axboe if (sqd) { 12495ed29b0b4SJens Axboe /* 12496ed29b0b4SJens Axboe * Observe the correct sqd->lock -> ctx->uring_lock 12497ed29b0b4SJens Axboe * ordering. Fine to drop uring_lock here, we hold 12498ed29b0b4SJens Axboe * a ref to the ctx. 12499ed29b0b4SJens Axboe */ 12500ed29b0b4SJens Axboe refcount_inc(&sqd->refs); 12501ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 12502ed29b0b4SJens Axboe mutex_lock(&sqd->lock); 12503ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 12504ed29b0b4SJens Axboe if (sqd->thread) 12505ed29b0b4SJens Axboe tctx = sqd->thread->io_uring; 12506ed29b0b4SJens Axboe } 12507ed29b0b4SJens Axboe } else { 12508ed29b0b4SJens Axboe tctx = current->io_uring; 12509ed29b0b4SJens Axboe } 12510ed29b0b4SJens Axboe 12511ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(new_count) != sizeof(ctx->iowq_limits)); 12512ed29b0b4SJens Axboe 12513ed29b0b4SJens Axboe for (i = 0; i < ARRAY_SIZE(new_count); i++) 12514ed29b0b4SJens Axboe if (new_count[i]) 12515ed29b0b4SJens Axboe ctx->iowq_limits[i] = new_count[i]; 12516ed29b0b4SJens Axboe ctx->iowq_limits_set = true; 12517ed29b0b4SJens Axboe 12518ed29b0b4SJens Axboe if (tctx && tctx->io_wq) { 12519ed29b0b4SJens Axboe ret = io_wq_max_workers(tctx->io_wq, new_count); 12520ed29b0b4SJens Axboe if (ret) 12521ed29b0b4SJens Axboe goto err; 12522ed29b0b4SJens Axboe } else { 12523ed29b0b4SJens Axboe memset(new_count, 0, sizeof(new_count)); 12524ed29b0b4SJens Axboe } 12525ed29b0b4SJens Axboe 12526ed29b0b4SJens Axboe if (sqd) { 12527ed29b0b4SJens Axboe mutex_unlock(&sqd->lock); 12528ed29b0b4SJens Axboe io_put_sq_data(sqd); 12529ed29b0b4SJens Axboe } 12530ed29b0b4SJens Axboe 12531ed29b0b4SJens Axboe if (copy_to_user(arg, new_count, sizeof(new_count))) 12532ed29b0b4SJens Axboe return -EFAULT; 12533ed29b0b4SJens Axboe 12534ed29b0b4SJens Axboe /* that's it for SQPOLL, only the SQPOLL task creates requests */ 12535ed29b0b4SJens Axboe if (sqd) 12536ed29b0b4SJens Axboe return 0; 12537ed29b0b4SJens Axboe 12538ed29b0b4SJens Axboe /* now propagate the restriction to all registered users */ 12539ed29b0b4SJens Axboe list_for_each_entry(node, &ctx->tctx_list, ctx_node) { 12540ed29b0b4SJens Axboe struct io_uring_task *tctx = node->task->io_uring; 12541ed29b0b4SJens Axboe 12542ed29b0b4SJens Axboe if (WARN_ON_ONCE(!tctx->io_wq)) 12543ed29b0b4SJens Axboe continue; 12544ed29b0b4SJens Axboe 12545ed29b0b4SJens Axboe for (i = 0; i < ARRAY_SIZE(new_count); i++) 12546ed29b0b4SJens Axboe new_count[i] = ctx->iowq_limits[i]; 12547ed29b0b4SJens Axboe /* ignore errors, it always returns zero anyway */ 12548ed29b0b4SJens Axboe (void)io_wq_max_workers(tctx->io_wq, new_count); 12549ed29b0b4SJens Axboe } 12550ed29b0b4SJens Axboe return 0; 12551ed29b0b4SJens Axboe err: 12552ed29b0b4SJens Axboe if (sqd) { 12553ed29b0b4SJens Axboe mutex_unlock(&sqd->lock); 12554ed29b0b4SJens Axboe io_put_sq_data(sqd); 12555ed29b0b4SJens Axboe } 12556ed29b0b4SJens Axboe return ret; 12557ed29b0b4SJens Axboe } 12558ed29b0b4SJens Axboe 12559ed29b0b4SJens Axboe static int io_register_pbuf_ring(struct io_ring_ctx *ctx, void __user *arg) 12560ed29b0b4SJens Axboe { 12561ed29b0b4SJens Axboe struct io_uring_buf_ring *br; 12562ed29b0b4SJens Axboe struct io_uring_buf_reg reg; 12563ed29b0b4SJens Axboe struct io_buffer_list *bl, *free_bl = NULL; 12564ed29b0b4SJens Axboe struct page **pages; 12565ed29b0b4SJens Axboe int nr_pages; 12566ed29b0b4SJens Axboe 12567ed29b0b4SJens Axboe if (copy_from_user(®, arg, sizeof(reg))) 12568ed29b0b4SJens Axboe return -EFAULT; 12569ed29b0b4SJens Axboe 12570ed29b0b4SJens Axboe if (reg.pad || reg.resv[0] || reg.resv[1] || reg.resv[2]) 12571ed29b0b4SJens Axboe return -EINVAL; 12572ed29b0b4SJens Axboe if (!reg.ring_addr) 12573ed29b0b4SJens Axboe return -EFAULT; 12574ed29b0b4SJens Axboe if (reg.ring_addr & ~PAGE_MASK) 12575ed29b0b4SJens Axboe return -EINVAL; 12576ed29b0b4SJens Axboe if (!is_power_of_2(reg.ring_entries)) 12577ed29b0b4SJens Axboe return -EINVAL; 12578ed29b0b4SJens Axboe 12579ed29b0b4SJens Axboe /* cannot disambiguate full vs empty due to head/tail size */ 12580ed29b0b4SJens Axboe if (reg.ring_entries >= 65536) 12581ed29b0b4SJens Axboe return -EINVAL; 12582ed29b0b4SJens Axboe 12583ed29b0b4SJens Axboe if (unlikely(reg.bgid < BGID_ARRAY && !ctx->io_bl)) { 12584ed29b0b4SJens Axboe int ret = io_init_bl_list(ctx); 12585ed29b0b4SJens Axboe if (ret) 12586ed29b0b4SJens Axboe return ret; 12587ed29b0b4SJens Axboe } 12588ed29b0b4SJens Axboe 12589ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, reg.bgid); 12590ed29b0b4SJens Axboe if (bl) { 12591ed29b0b4SJens Axboe /* if mapped buffer ring OR classic exists, don't allow */ 12592ed29b0b4SJens Axboe if (bl->buf_nr_pages || !list_empty(&bl->buf_list)) 12593ed29b0b4SJens Axboe return -EEXIST; 12594ed29b0b4SJens Axboe } else { 12595ed29b0b4SJens Axboe free_bl = bl = kzalloc(sizeof(*bl), GFP_KERNEL); 12596ed29b0b4SJens Axboe if (!bl) 12597ed29b0b4SJens Axboe return -ENOMEM; 12598ed29b0b4SJens Axboe } 12599ed29b0b4SJens Axboe 12600ed29b0b4SJens Axboe pages = io_pin_pages(reg.ring_addr, 12601ed29b0b4SJens Axboe struct_size(br, bufs, reg.ring_entries), 12602ed29b0b4SJens Axboe &nr_pages); 12603ed29b0b4SJens Axboe if (IS_ERR(pages)) { 12604ed29b0b4SJens Axboe kfree(free_bl); 12605ed29b0b4SJens Axboe return PTR_ERR(pages); 12606ed29b0b4SJens Axboe } 12607ed29b0b4SJens Axboe 12608ed29b0b4SJens Axboe br = page_address(pages[0]); 12609ed29b0b4SJens Axboe bl->buf_pages = pages; 12610ed29b0b4SJens Axboe bl->buf_nr_pages = nr_pages; 12611ed29b0b4SJens Axboe bl->nr_entries = reg.ring_entries; 12612ed29b0b4SJens Axboe bl->buf_ring = br; 12613ed29b0b4SJens Axboe bl->mask = reg.ring_entries - 1; 12614ed29b0b4SJens Axboe io_buffer_add_list(ctx, bl, reg.bgid); 12615ed29b0b4SJens Axboe return 0; 12616ed29b0b4SJens Axboe } 12617ed29b0b4SJens Axboe 12618ed29b0b4SJens Axboe static int io_unregister_pbuf_ring(struct io_ring_ctx *ctx, void __user *arg) 12619ed29b0b4SJens Axboe { 12620ed29b0b4SJens Axboe struct io_uring_buf_reg reg; 12621ed29b0b4SJens Axboe struct io_buffer_list *bl; 12622ed29b0b4SJens Axboe 12623ed29b0b4SJens Axboe if (copy_from_user(®, arg, sizeof(reg))) 12624ed29b0b4SJens Axboe return -EFAULT; 12625ed29b0b4SJens Axboe if (reg.pad || reg.resv[0] || reg.resv[1] || reg.resv[2]) 12626ed29b0b4SJens Axboe return -EINVAL; 12627ed29b0b4SJens Axboe 12628ed29b0b4SJens Axboe bl = io_buffer_get_list(ctx, reg.bgid); 12629ed29b0b4SJens Axboe if (!bl) 12630ed29b0b4SJens Axboe return -ENOENT; 12631ed29b0b4SJens Axboe if (!bl->buf_nr_pages) 12632ed29b0b4SJens Axboe return -EINVAL; 12633ed29b0b4SJens Axboe 12634ed29b0b4SJens Axboe __io_remove_buffers(ctx, bl, -1U); 12635ed29b0b4SJens Axboe if (bl->bgid >= BGID_ARRAY) { 12636ed29b0b4SJens Axboe xa_erase(&ctx->io_bl_xa, bl->bgid); 12637ed29b0b4SJens Axboe kfree(bl); 12638ed29b0b4SJens Axboe } 12639ed29b0b4SJens Axboe return 0; 12640ed29b0b4SJens Axboe } 12641ed29b0b4SJens Axboe 12642ed29b0b4SJens Axboe static int __io_uring_register(struct io_ring_ctx *ctx, unsigned opcode, 12643ed29b0b4SJens Axboe void __user *arg, unsigned nr_args) 12644ed29b0b4SJens Axboe __releases(ctx->uring_lock) 12645ed29b0b4SJens Axboe __acquires(ctx->uring_lock) 12646ed29b0b4SJens Axboe { 12647ed29b0b4SJens Axboe int ret; 12648ed29b0b4SJens Axboe 12649ed29b0b4SJens Axboe /* 12650ed29b0b4SJens Axboe * We're inside the ring mutex, if the ref is already dying, then 12651ed29b0b4SJens Axboe * someone else killed the ctx or is already going through 12652ed29b0b4SJens Axboe * io_uring_register(). 12653ed29b0b4SJens Axboe */ 12654ed29b0b4SJens Axboe if (percpu_ref_is_dying(&ctx->refs)) 12655ed29b0b4SJens Axboe return -ENXIO; 12656ed29b0b4SJens Axboe 12657ed29b0b4SJens Axboe if (ctx->restricted) { 12658ed29b0b4SJens Axboe if (opcode >= IORING_REGISTER_LAST) 12659ed29b0b4SJens Axboe return -EINVAL; 12660ed29b0b4SJens Axboe opcode = array_index_nospec(opcode, IORING_REGISTER_LAST); 12661ed29b0b4SJens Axboe if (!test_bit(opcode, ctx->restrictions.register_op)) 12662ed29b0b4SJens Axboe return -EACCES; 12663ed29b0b4SJens Axboe } 12664ed29b0b4SJens Axboe 12665ed29b0b4SJens Axboe switch (opcode) { 12666ed29b0b4SJens Axboe case IORING_REGISTER_BUFFERS: 12667ed29b0b4SJens Axboe ret = -EFAULT; 12668ed29b0b4SJens Axboe if (!arg) 12669ed29b0b4SJens Axboe break; 12670ed29b0b4SJens Axboe ret = io_sqe_buffers_register(ctx, arg, nr_args, NULL); 12671ed29b0b4SJens Axboe break; 12672ed29b0b4SJens Axboe case IORING_UNREGISTER_BUFFERS: 12673ed29b0b4SJens Axboe ret = -EINVAL; 12674ed29b0b4SJens Axboe if (arg || nr_args) 12675ed29b0b4SJens Axboe break; 12676ed29b0b4SJens Axboe ret = io_sqe_buffers_unregister(ctx); 12677ed29b0b4SJens Axboe break; 12678ed29b0b4SJens Axboe case IORING_REGISTER_FILES: 12679ed29b0b4SJens Axboe ret = -EFAULT; 12680ed29b0b4SJens Axboe if (!arg) 12681ed29b0b4SJens Axboe break; 12682ed29b0b4SJens Axboe ret = io_sqe_files_register(ctx, arg, nr_args, NULL); 12683ed29b0b4SJens Axboe break; 12684ed29b0b4SJens Axboe case IORING_UNREGISTER_FILES: 12685ed29b0b4SJens Axboe ret = -EINVAL; 12686ed29b0b4SJens Axboe if (arg || nr_args) 12687ed29b0b4SJens Axboe break; 12688ed29b0b4SJens Axboe ret = io_sqe_files_unregister(ctx); 12689ed29b0b4SJens Axboe break; 12690ed29b0b4SJens Axboe case IORING_REGISTER_FILES_UPDATE: 12691ed29b0b4SJens Axboe ret = io_register_files_update(ctx, arg, nr_args); 12692ed29b0b4SJens Axboe break; 12693ed29b0b4SJens Axboe case IORING_REGISTER_EVENTFD: 12694ed29b0b4SJens Axboe ret = -EINVAL; 12695ed29b0b4SJens Axboe if (nr_args != 1) 12696ed29b0b4SJens Axboe break; 12697ed29b0b4SJens Axboe ret = io_eventfd_register(ctx, arg, 0); 12698ed29b0b4SJens Axboe break; 12699ed29b0b4SJens Axboe case IORING_REGISTER_EVENTFD_ASYNC: 12700ed29b0b4SJens Axboe ret = -EINVAL; 12701ed29b0b4SJens Axboe if (nr_args != 1) 12702ed29b0b4SJens Axboe break; 12703ed29b0b4SJens Axboe ret = io_eventfd_register(ctx, arg, 1); 12704ed29b0b4SJens Axboe break; 12705ed29b0b4SJens Axboe case IORING_UNREGISTER_EVENTFD: 12706ed29b0b4SJens Axboe ret = -EINVAL; 12707ed29b0b4SJens Axboe if (arg || nr_args) 12708ed29b0b4SJens Axboe break; 12709ed29b0b4SJens Axboe ret = io_eventfd_unregister(ctx); 12710ed29b0b4SJens Axboe break; 12711ed29b0b4SJens Axboe case IORING_REGISTER_PROBE: 12712ed29b0b4SJens Axboe ret = -EINVAL; 12713ed29b0b4SJens Axboe if (!arg || nr_args > 256) 12714ed29b0b4SJens Axboe break; 12715ed29b0b4SJens Axboe ret = io_probe(ctx, arg, nr_args); 12716ed29b0b4SJens Axboe break; 12717ed29b0b4SJens Axboe case IORING_REGISTER_PERSONALITY: 12718ed29b0b4SJens Axboe ret = -EINVAL; 12719ed29b0b4SJens Axboe if (arg || nr_args) 12720ed29b0b4SJens Axboe break; 12721ed29b0b4SJens Axboe ret = io_register_personality(ctx); 12722ed29b0b4SJens Axboe break; 12723ed29b0b4SJens Axboe case IORING_UNREGISTER_PERSONALITY: 12724ed29b0b4SJens Axboe ret = -EINVAL; 12725ed29b0b4SJens Axboe if (arg) 12726ed29b0b4SJens Axboe break; 12727ed29b0b4SJens Axboe ret = io_unregister_personality(ctx, nr_args); 12728ed29b0b4SJens Axboe break; 12729ed29b0b4SJens Axboe case IORING_REGISTER_ENABLE_RINGS: 12730ed29b0b4SJens Axboe ret = -EINVAL; 12731ed29b0b4SJens Axboe if (arg || nr_args) 12732ed29b0b4SJens Axboe break; 12733ed29b0b4SJens Axboe ret = io_register_enable_rings(ctx); 12734ed29b0b4SJens Axboe break; 12735ed29b0b4SJens Axboe case IORING_REGISTER_RESTRICTIONS: 12736ed29b0b4SJens Axboe ret = io_register_restrictions(ctx, arg, nr_args); 12737ed29b0b4SJens Axboe break; 12738ed29b0b4SJens Axboe case IORING_REGISTER_FILES2: 12739ed29b0b4SJens Axboe ret = io_register_rsrc(ctx, arg, nr_args, IORING_RSRC_FILE); 12740ed29b0b4SJens Axboe break; 12741ed29b0b4SJens Axboe case IORING_REGISTER_FILES_UPDATE2: 12742ed29b0b4SJens Axboe ret = io_register_rsrc_update(ctx, arg, nr_args, 12743ed29b0b4SJens Axboe IORING_RSRC_FILE); 12744ed29b0b4SJens Axboe break; 12745ed29b0b4SJens Axboe case IORING_REGISTER_BUFFERS2: 12746ed29b0b4SJens Axboe ret = io_register_rsrc(ctx, arg, nr_args, IORING_RSRC_BUFFER); 12747ed29b0b4SJens Axboe break; 12748ed29b0b4SJens Axboe case IORING_REGISTER_BUFFERS_UPDATE: 12749ed29b0b4SJens Axboe ret = io_register_rsrc_update(ctx, arg, nr_args, 12750ed29b0b4SJens Axboe IORING_RSRC_BUFFER); 12751ed29b0b4SJens Axboe break; 12752ed29b0b4SJens Axboe case IORING_REGISTER_IOWQ_AFF: 12753ed29b0b4SJens Axboe ret = -EINVAL; 12754ed29b0b4SJens Axboe if (!arg || !nr_args) 12755ed29b0b4SJens Axboe break; 12756ed29b0b4SJens Axboe ret = io_register_iowq_aff(ctx, arg, nr_args); 12757ed29b0b4SJens Axboe break; 12758ed29b0b4SJens Axboe case IORING_UNREGISTER_IOWQ_AFF: 12759ed29b0b4SJens Axboe ret = -EINVAL; 12760ed29b0b4SJens Axboe if (arg || nr_args) 12761ed29b0b4SJens Axboe break; 12762ed29b0b4SJens Axboe ret = io_unregister_iowq_aff(ctx); 12763ed29b0b4SJens Axboe break; 12764ed29b0b4SJens Axboe case IORING_REGISTER_IOWQ_MAX_WORKERS: 12765ed29b0b4SJens Axboe ret = -EINVAL; 12766ed29b0b4SJens Axboe if (!arg || nr_args != 2) 12767ed29b0b4SJens Axboe break; 12768ed29b0b4SJens Axboe ret = io_register_iowq_max_workers(ctx, arg); 12769ed29b0b4SJens Axboe break; 12770ed29b0b4SJens Axboe case IORING_REGISTER_RING_FDS: 12771ed29b0b4SJens Axboe ret = io_ringfd_register(ctx, arg, nr_args); 12772ed29b0b4SJens Axboe break; 12773ed29b0b4SJens Axboe case IORING_UNREGISTER_RING_FDS: 12774ed29b0b4SJens Axboe ret = io_ringfd_unregister(ctx, arg, nr_args); 12775ed29b0b4SJens Axboe break; 12776ed29b0b4SJens Axboe case IORING_REGISTER_PBUF_RING: 12777ed29b0b4SJens Axboe ret = -EINVAL; 12778ed29b0b4SJens Axboe if (!arg || nr_args != 1) 12779ed29b0b4SJens Axboe break; 12780ed29b0b4SJens Axboe ret = io_register_pbuf_ring(ctx, arg); 12781ed29b0b4SJens Axboe break; 12782ed29b0b4SJens Axboe case IORING_UNREGISTER_PBUF_RING: 12783ed29b0b4SJens Axboe ret = -EINVAL; 12784ed29b0b4SJens Axboe if (!arg || nr_args != 1) 12785ed29b0b4SJens Axboe break; 12786ed29b0b4SJens Axboe ret = io_unregister_pbuf_ring(ctx, arg); 12787ed29b0b4SJens Axboe break; 12788ed29b0b4SJens Axboe default: 12789ed29b0b4SJens Axboe ret = -EINVAL; 12790ed29b0b4SJens Axboe break; 12791ed29b0b4SJens Axboe } 12792ed29b0b4SJens Axboe 12793ed29b0b4SJens Axboe return ret; 12794ed29b0b4SJens Axboe } 12795ed29b0b4SJens Axboe 12796ed29b0b4SJens Axboe SYSCALL_DEFINE4(io_uring_register, unsigned int, fd, unsigned int, opcode, 12797ed29b0b4SJens Axboe void __user *, arg, unsigned int, nr_args) 12798ed29b0b4SJens Axboe { 12799ed29b0b4SJens Axboe struct io_ring_ctx *ctx; 12800ed29b0b4SJens Axboe long ret = -EBADF; 12801ed29b0b4SJens Axboe struct fd f; 12802ed29b0b4SJens Axboe 12803ed29b0b4SJens Axboe f = fdget(fd); 12804ed29b0b4SJens Axboe if (!f.file) 12805ed29b0b4SJens Axboe return -EBADF; 12806ed29b0b4SJens Axboe 12807ed29b0b4SJens Axboe ret = -EOPNOTSUPP; 12808ed29b0b4SJens Axboe if (f.file->f_op != &io_uring_fops) 12809ed29b0b4SJens Axboe goto out_fput; 12810ed29b0b4SJens Axboe 12811ed29b0b4SJens Axboe ctx = f.file->private_data; 12812ed29b0b4SJens Axboe 12813ed29b0b4SJens Axboe io_run_task_work(); 12814ed29b0b4SJens Axboe 12815ed29b0b4SJens Axboe mutex_lock(&ctx->uring_lock); 12816ed29b0b4SJens Axboe ret = __io_uring_register(ctx, opcode, arg, nr_args); 12817ed29b0b4SJens Axboe mutex_unlock(&ctx->uring_lock); 12818ed29b0b4SJens Axboe trace_io_uring_register(ctx, opcode, ctx->nr_user_files, ctx->nr_user_bufs, ret); 12819ed29b0b4SJens Axboe out_fput: 12820ed29b0b4SJens Axboe fdput(f); 12821ed29b0b4SJens Axboe return ret; 12822ed29b0b4SJens Axboe } 12823ed29b0b4SJens Axboe 12824ed29b0b4SJens Axboe static int io_no_issue(struct io_kiocb *req, unsigned int issue_flags) 12825ed29b0b4SJens Axboe { 12826ed29b0b4SJens Axboe WARN_ON_ONCE(1); 12827ed29b0b4SJens Axboe return -ECANCELED; 12828ed29b0b4SJens Axboe } 12829ed29b0b4SJens Axboe 12830ed29b0b4SJens Axboe static const struct io_op_def io_op_defs[] = { 12831ed29b0b4SJens Axboe [IORING_OP_NOP] = { 12832ed29b0b4SJens Axboe .audit_skip = 1, 12833ed29b0b4SJens Axboe .iopoll = 1, 12834ed29b0b4SJens Axboe .prep = io_nop_prep, 12835ed29b0b4SJens Axboe .issue = io_nop, 12836ed29b0b4SJens Axboe }, 12837ed29b0b4SJens Axboe [IORING_OP_READV] = { 12838ed29b0b4SJens Axboe .needs_file = 1, 12839ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 12840ed29b0b4SJens Axboe .pollin = 1, 12841ed29b0b4SJens Axboe .buffer_select = 1, 12842ed29b0b4SJens Axboe .plug = 1, 12843ed29b0b4SJens Axboe .audit_skip = 1, 12844ed29b0b4SJens Axboe .ioprio = 1, 12845ed29b0b4SJens Axboe .iopoll = 1, 12846ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 12847ed29b0b4SJens Axboe .prep = io_prep_rw, 12848ed29b0b4SJens Axboe .issue = io_read, 12849dc919cafSJens Axboe .prep_async = io_readv_prep_async, 12850ed29b0b4SJens Axboe }, 12851ed29b0b4SJens Axboe [IORING_OP_WRITEV] = { 12852ed29b0b4SJens Axboe .needs_file = 1, 12853ed29b0b4SJens Axboe .hash_reg_file = 1, 12854ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 12855ed29b0b4SJens Axboe .pollout = 1, 12856ed29b0b4SJens Axboe .plug = 1, 12857ed29b0b4SJens Axboe .audit_skip = 1, 12858ed29b0b4SJens Axboe .ioprio = 1, 12859ed29b0b4SJens Axboe .iopoll = 1, 12860ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 12861ed29b0b4SJens Axboe .prep = io_prep_rw, 12862ed29b0b4SJens Axboe .issue = io_write, 12863dc919cafSJens Axboe .prep_async = io_writev_prep_async, 12864ed29b0b4SJens Axboe }, 12865ed29b0b4SJens Axboe [IORING_OP_FSYNC] = { 12866ed29b0b4SJens Axboe .needs_file = 1, 12867ed29b0b4SJens Axboe .audit_skip = 1, 12868ed29b0b4SJens Axboe .prep = io_fsync_prep, 12869ed29b0b4SJens Axboe .issue = io_fsync, 12870ed29b0b4SJens Axboe }, 12871ed29b0b4SJens Axboe [IORING_OP_READ_FIXED] = { 12872ed29b0b4SJens Axboe .needs_file = 1, 12873ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 12874ed29b0b4SJens Axboe .pollin = 1, 12875ed29b0b4SJens Axboe .plug = 1, 12876ed29b0b4SJens Axboe .audit_skip = 1, 12877ed29b0b4SJens Axboe .ioprio = 1, 12878ed29b0b4SJens Axboe .iopoll = 1, 12879ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 12880ed29b0b4SJens Axboe .prep = io_prep_rw, 12881ed29b0b4SJens Axboe .issue = io_read, 12882ed29b0b4SJens Axboe }, 12883ed29b0b4SJens Axboe [IORING_OP_WRITE_FIXED] = { 12884ed29b0b4SJens Axboe .needs_file = 1, 12885ed29b0b4SJens Axboe .hash_reg_file = 1, 12886ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 12887ed29b0b4SJens Axboe .pollout = 1, 12888ed29b0b4SJens Axboe .plug = 1, 12889ed29b0b4SJens Axboe .audit_skip = 1, 12890ed29b0b4SJens Axboe .ioprio = 1, 12891ed29b0b4SJens Axboe .iopoll = 1, 12892ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 12893ed29b0b4SJens Axboe .prep = io_prep_rw, 12894ed29b0b4SJens Axboe .issue = io_write, 12895ed29b0b4SJens Axboe }, 12896ed29b0b4SJens Axboe [IORING_OP_POLL_ADD] = { 12897ed29b0b4SJens Axboe .needs_file = 1, 12898ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 12899ed29b0b4SJens Axboe .audit_skip = 1, 12900ed29b0b4SJens Axboe .prep = io_poll_add_prep, 12901ed29b0b4SJens Axboe .issue = io_poll_add, 12902ed29b0b4SJens Axboe }, 12903ed29b0b4SJens Axboe [IORING_OP_POLL_REMOVE] = { 12904ed29b0b4SJens Axboe .audit_skip = 1, 12905ed29b0b4SJens Axboe .prep = io_poll_remove_prep, 12906ed29b0b4SJens Axboe .issue = io_poll_remove, 12907ed29b0b4SJens Axboe }, 12908ed29b0b4SJens Axboe [IORING_OP_SYNC_FILE_RANGE] = { 12909ed29b0b4SJens Axboe .needs_file = 1, 12910ed29b0b4SJens Axboe .audit_skip = 1, 12911ed29b0b4SJens Axboe .prep = io_sfr_prep, 12912ed29b0b4SJens Axboe .issue = io_sync_file_range, 12913ed29b0b4SJens Axboe }, 12914ed29b0b4SJens Axboe [IORING_OP_SENDMSG] = { 12915ed29b0b4SJens Axboe .needs_file = 1, 12916ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 12917ed29b0b4SJens Axboe .pollout = 1, 12918ed29b0b4SJens Axboe .ioprio = 1, 12919ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_msghdr), 12920ed29b0b4SJens Axboe .prep = io_sendmsg_prep, 12921ed29b0b4SJens Axboe .issue = io_sendmsg, 12922dc919cafSJens Axboe .prep_async = io_sendmsg_prep_async, 12923ed29b0b4SJens Axboe }, 12924ed29b0b4SJens Axboe [IORING_OP_RECVMSG] = { 12925ed29b0b4SJens Axboe .needs_file = 1, 12926ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 12927ed29b0b4SJens Axboe .pollin = 1, 12928ed29b0b4SJens Axboe .buffer_select = 1, 12929ed29b0b4SJens Axboe .ioprio = 1, 12930ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_msghdr), 12931ed29b0b4SJens Axboe .prep = io_recvmsg_prep, 12932ed29b0b4SJens Axboe .issue = io_recvmsg, 12933dc919cafSJens Axboe .prep_async = io_recvmsg_prep_async, 12934ed29b0b4SJens Axboe }, 12935ed29b0b4SJens Axboe [IORING_OP_TIMEOUT] = { 12936ed29b0b4SJens Axboe .audit_skip = 1, 12937ed29b0b4SJens Axboe .async_size = sizeof(struct io_timeout_data), 12938ed29b0b4SJens Axboe .prep = io_timeout_prep, 12939ed29b0b4SJens Axboe .issue = io_timeout, 12940ed29b0b4SJens Axboe }, 12941ed29b0b4SJens Axboe [IORING_OP_TIMEOUT_REMOVE] = { 12942ed29b0b4SJens Axboe /* used by timeout updates' prep() */ 12943ed29b0b4SJens Axboe .audit_skip = 1, 12944ed29b0b4SJens Axboe .prep = io_timeout_remove_prep, 12945ed29b0b4SJens Axboe .issue = io_timeout_remove, 12946ed29b0b4SJens Axboe }, 12947ed29b0b4SJens Axboe [IORING_OP_ACCEPT] = { 12948ed29b0b4SJens Axboe .needs_file = 1, 12949ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 12950ed29b0b4SJens Axboe .pollin = 1, 12951ed29b0b4SJens Axboe .poll_exclusive = 1, 12952ed29b0b4SJens Axboe .ioprio = 1, /* used for flags */ 12953ed29b0b4SJens Axboe .prep = io_accept_prep, 12954ed29b0b4SJens Axboe .issue = io_accept, 12955ed29b0b4SJens Axboe }, 12956ed29b0b4SJens Axboe [IORING_OP_ASYNC_CANCEL] = { 12957ed29b0b4SJens Axboe .audit_skip = 1, 12958ed29b0b4SJens Axboe .prep = io_async_cancel_prep, 12959ed29b0b4SJens Axboe .issue = io_async_cancel, 12960ed29b0b4SJens Axboe }, 12961ed29b0b4SJens Axboe [IORING_OP_LINK_TIMEOUT] = { 12962ed29b0b4SJens Axboe .audit_skip = 1, 12963ed29b0b4SJens Axboe .async_size = sizeof(struct io_timeout_data), 12964ed29b0b4SJens Axboe .prep = io_link_timeout_prep, 12965ed29b0b4SJens Axboe .issue = io_no_issue, 12966ed29b0b4SJens Axboe }, 12967ed29b0b4SJens Axboe [IORING_OP_CONNECT] = { 12968ed29b0b4SJens Axboe .needs_file = 1, 12969ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 12970ed29b0b4SJens Axboe .pollout = 1, 12971ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_connect), 12972ed29b0b4SJens Axboe .prep = io_connect_prep, 12973ed29b0b4SJens Axboe .issue = io_connect, 12974dc919cafSJens Axboe .prep_async = io_connect_prep_async, 12975ed29b0b4SJens Axboe }, 12976ed29b0b4SJens Axboe [IORING_OP_FALLOCATE] = { 12977ed29b0b4SJens Axboe .needs_file = 1, 12978ed29b0b4SJens Axboe .prep = io_fallocate_prep, 12979ed29b0b4SJens Axboe .issue = io_fallocate, 12980ed29b0b4SJens Axboe }, 12981ed29b0b4SJens Axboe [IORING_OP_OPENAT] = { 12982ed29b0b4SJens Axboe .prep = io_openat_prep, 12983ed29b0b4SJens Axboe .issue = io_openat, 12984ed29b0b4SJens Axboe }, 12985ed29b0b4SJens Axboe [IORING_OP_CLOSE] = { 12986ed29b0b4SJens Axboe .prep = io_close_prep, 12987ed29b0b4SJens Axboe .issue = io_close, 12988ed29b0b4SJens Axboe }, 12989ed29b0b4SJens Axboe [IORING_OP_FILES_UPDATE] = { 12990ed29b0b4SJens Axboe .audit_skip = 1, 12991ed29b0b4SJens Axboe .iopoll = 1, 12992ed29b0b4SJens Axboe .prep = io_files_update_prep, 12993ed29b0b4SJens Axboe .issue = io_files_update, 12994ed29b0b4SJens Axboe }, 12995ed29b0b4SJens Axboe [IORING_OP_STATX] = { 12996ed29b0b4SJens Axboe .audit_skip = 1, 12997ed29b0b4SJens Axboe .prep = io_statx_prep, 12998ed29b0b4SJens Axboe .issue = io_statx, 12999ed29b0b4SJens Axboe }, 13000ed29b0b4SJens Axboe [IORING_OP_READ] = { 13001ed29b0b4SJens Axboe .needs_file = 1, 13002ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 13003ed29b0b4SJens Axboe .pollin = 1, 13004ed29b0b4SJens Axboe .buffer_select = 1, 13005ed29b0b4SJens Axboe .plug = 1, 13006ed29b0b4SJens Axboe .audit_skip = 1, 13007ed29b0b4SJens Axboe .ioprio = 1, 13008ed29b0b4SJens Axboe .iopoll = 1, 13009ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 13010ed29b0b4SJens Axboe .prep = io_prep_rw, 13011ed29b0b4SJens Axboe .issue = io_read, 13012ed29b0b4SJens Axboe }, 13013ed29b0b4SJens Axboe [IORING_OP_WRITE] = { 13014ed29b0b4SJens Axboe .needs_file = 1, 13015ed29b0b4SJens Axboe .hash_reg_file = 1, 13016ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 13017ed29b0b4SJens Axboe .pollout = 1, 13018ed29b0b4SJens Axboe .plug = 1, 13019ed29b0b4SJens Axboe .audit_skip = 1, 13020ed29b0b4SJens Axboe .ioprio = 1, 13021ed29b0b4SJens Axboe .iopoll = 1, 13022ed29b0b4SJens Axboe .async_size = sizeof(struct io_async_rw), 13023ed29b0b4SJens Axboe .prep = io_prep_rw, 13024ed29b0b4SJens Axboe .issue = io_write, 13025ed29b0b4SJens Axboe }, 13026ed29b0b4SJens Axboe [IORING_OP_FADVISE] = { 13027ed29b0b4SJens Axboe .needs_file = 1, 13028ed29b0b4SJens Axboe .audit_skip = 1, 13029ed29b0b4SJens Axboe .prep = io_fadvise_prep, 13030ed29b0b4SJens Axboe .issue = io_fadvise, 13031ed29b0b4SJens Axboe }, 13032ed29b0b4SJens Axboe [IORING_OP_MADVISE] = { 13033ed29b0b4SJens Axboe .prep = io_madvise_prep, 13034ed29b0b4SJens Axboe .issue = io_madvise, 13035ed29b0b4SJens Axboe }, 13036ed29b0b4SJens Axboe [IORING_OP_SEND] = { 13037ed29b0b4SJens Axboe .needs_file = 1, 13038ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 13039ed29b0b4SJens Axboe .pollout = 1, 13040ed29b0b4SJens Axboe .audit_skip = 1, 13041ed29b0b4SJens Axboe .ioprio = 1, 13042ed29b0b4SJens Axboe .prep = io_sendmsg_prep, 13043ed29b0b4SJens Axboe .issue = io_send, 13044ed29b0b4SJens Axboe }, 13045ed29b0b4SJens Axboe [IORING_OP_RECV] = { 13046ed29b0b4SJens Axboe .needs_file = 1, 13047ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 13048ed29b0b4SJens Axboe .pollin = 1, 13049ed29b0b4SJens Axboe .buffer_select = 1, 13050ed29b0b4SJens Axboe .audit_skip = 1, 13051ed29b0b4SJens Axboe .ioprio = 1, 13052ed29b0b4SJens Axboe .prep = io_recvmsg_prep, 13053ed29b0b4SJens Axboe .issue = io_recv, 13054ed29b0b4SJens Axboe }, 13055ed29b0b4SJens Axboe [IORING_OP_OPENAT2] = { 13056ed29b0b4SJens Axboe .prep = io_openat2_prep, 13057ed29b0b4SJens Axboe .issue = io_openat2, 13058ed29b0b4SJens Axboe }, 13059ed29b0b4SJens Axboe [IORING_OP_EPOLL_CTL] = { 13060ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 13061ed29b0b4SJens Axboe .audit_skip = 1, 13062ed29b0b4SJens Axboe .prep = io_epoll_ctl_prep, 13063ed29b0b4SJens Axboe .issue = io_epoll_ctl, 13064ed29b0b4SJens Axboe }, 13065ed29b0b4SJens Axboe [IORING_OP_SPLICE] = { 13066ed29b0b4SJens Axboe .needs_file = 1, 13067ed29b0b4SJens Axboe .hash_reg_file = 1, 13068ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 13069ed29b0b4SJens Axboe .audit_skip = 1, 13070ed29b0b4SJens Axboe .prep = io_splice_prep, 13071ed29b0b4SJens Axboe .issue = io_splice, 13072ed29b0b4SJens Axboe }, 13073ed29b0b4SJens Axboe [IORING_OP_PROVIDE_BUFFERS] = { 13074ed29b0b4SJens Axboe .audit_skip = 1, 13075ed29b0b4SJens Axboe .iopoll = 1, 13076ed29b0b4SJens Axboe .prep = io_provide_buffers_prep, 13077ed29b0b4SJens Axboe .issue = io_provide_buffers, 13078ed29b0b4SJens Axboe }, 13079ed29b0b4SJens Axboe [IORING_OP_REMOVE_BUFFERS] = { 13080ed29b0b4SJens Axboe .audit_skip = 1, 13081ed29b0b4SJens Axboe .iopoll = 1, 13082ed29b0b4SJens Axboe .prep = io_remove_buffers_prep, 13083ed29b0b4SJens Axboe .issue = io_remove_buffers, 13084ed29b0b4SJens Axboe }, 13085ed29b0b4SJens Axboe [IORING_OP_TEE] = { 13086ed29b0b4SJens Axboe .needs_file = 1, 13087ed29b0b4SJens Axboe .hash_reg_file = 1, 13088ed29b0b4SJens Axboe .unbound_nonreg_file = 1, 13089ed29b0b4SJens Axboe .audit_skip = 1, 13090ed29b0b4SJens Axboe .prep = io_tee_prep, 13091ed29b0b4SJens Axboe .issue = io_tee, 13092ed29b0b4SJens Axboe }, 13093ed29b0b4SJens Axboe [IORING_OP_SHUTDOWN] = { 13094ed29b0b4SJens Axboe .needs_file = 1, 13095ed29b0b4SJens Axboe .prep = io_shutdown_prep, 13096ed29b0b4SJens Axboe .issue = io_shutdown, 13097ed29b0b4SJens Axboe }, 13098ed29b0b4SJens Axboe [IORING_OP_RENAMEAT] = { 13099ed29b0b4SJens Axboe .prep = io_renameat_prep, 13100ed29b0b4SJens Axboe .issue = io_renameat, 13101ed29b0b4SJens Axboe }, 13102ed29b0b4SJens Axboe [IORING_OP_UNLINKAT] = { 13103ed29b0b4SJens Axboe .prep = io_unlinkat_prep, 13104ed29b0b4SJens Axboe .issue = io_unlinkat, 13105ed29b0b4SJens Axboe }, 13106ed29b0b4SJens Axboe [IORING_OP_MKDIRAT] = { 13107ed29b0b4SJens Axboe .prep = io_mkdirat_prep, 13108ed29b0b4SJens Axboe .issue = io_mkdirat, 13109ed29b0b4SJens Axboe }, 13110ed29b0b4SJens Axboe [IORING_OP_SYMLINKAT] = { 13111ed29b0b4SJens Axboe .prep = io_symlinkat_prep, 13112ed29b0b4SJens Axboe .issue = io_symlinkat, 13113ed29b0b4SJens Axboe }, 13114ed29b0b4SJens Axboe [IORING_OP_LINKAT] = { 13115ed29b0b4SJens Axboe .prep = io_linkat_prep, 13116ed29b0b4SJens Axboe .issue = io_linkat, 13117ed29b0b4SJens Axboe }, 13118ed29b0b4SJens Axboe [IORING_OP_MSG_RING] = { 13119ed29b0b4SJens Axboe .needs_file = 1, 13120ed29b0b4SJens Axboe .iopoll = 1, 13121ed29b0b4SJens Axboe .prep = io_msg_ring_prep, 13122ed29b0b4SJens Axboe .issue = io_msg_ring, 13123ed29b0b4SJens Axboe }, 13124ed29b0b4SJens Axboe [IORING_OP_FSETXATTR] = { 13125ed29b0b4SJens Axboe .needs_file = 1, 13126ed29b0b4SJens Axboe .prep = io_fsetxattr_prep, 13127ed29b0b4SJens Axboe .issue = io_fsetxattr, 13128ed29b0b4SJens Axboe }, 13129ed29b0b4SJens Axboe [IORING_OP_SETXATTR] = { 13130ed29b0b4SJens Axboe .prep = io_setxattr_prep, 13131ed29b0b4SJens Axboe .issue = io_setxattr, 13132ed29b0b4SJens Axboe }, 13133ed29b0b4SJens Axboe [IORING_OP_FGETXATTR] = { 13134ed29b0b4SJens Axboe .needs_file = 1, 13135ed29b0b4SJens Axboe .prep = io_fgetxattr_prep, 13136ed29b0b4SJens Axboe .issue = io_fgetxattr, 13137ed29b0b4SJens Axboe }, 13138ed29b0b4SJens Axboe [IORING_OP_GETXATTR] = { 13139ed29b0b4SJens Axboe .prep = io_getxattr_prep, 13140ed29b0b4SJens Axboe .issue = io_getxattr, 13141ed29b0b4SJens Axboe }, 13142ed29b0b4SJens Axboe [IORING_OP_SOCKET] = { 13143ed29b0b4SJens Axboe .audit_skip = 1, 13144ed29b0b4SJens Axboe .prep = io_socket_prep, 13145ed29b0b4SJens Axboe .issue = io_socket, 13146ed29b0b4SJens Axboe }, 13147ed29b0b4SJens Axboe [IORING_OP_URING_CMD] = { 13148ed29b0b4SJens Axboe .needs_file = 1, 13149ed29b0b4SJens Axboe .plug = 1, 13150ed29b0b4SJens Axboe .async_size = uring_cmd_pdu_size(1), 13151ed29b0b4SJens Axboe .prep = io_uring_cmd_prep, 13152ed29b0b4SJens Axboe .issue = io_uring_cmd, 13153dc919cafSJens Axboe .prep_async = io_uring_cmd_prep_async, 13154ed29b0b4SJens Axboe }, 13155ed29b0b4SJens Axboe }; 13156ed29b0b4SJens Axboe 13157ed29b0b4SJens Axboe static int __init io_uring_init(void) 13158ed29b0b4SJens Axboe { 13159ed29b0b4SJens Axboe int i; 13160ed29b0b4SJens Axboe 13161ed29b0b4SJens Axboe #define __BUILD_BUG_VERIFY_ELEMENT(stype, eoffset, etype, ename) do { \ 13162ed29b0b4SJens Axboe BUILD_BUG_ON(offsetof(stype, ename) != eoffset); \ 13163ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(etype) != sizeof_field(stype, ename)); \ 13164ed29b0b4SJens Axboe } while (0) 13165ed29b0b4SJens Axboe 13166ed29b0b4SJens Axboe #define BUILD_BUG_SQE_ELEM(eoffset, etype, ename) \ 13167ed29b0b4SJens Axboe __BUILD_BUG_VERIFY_ELEMENT(struct io_uring_sqe, eoffset, etype, ename) 13168ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(struct io_uring_sqe) != 64); 13169ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(0, __u8, opcode); 13170ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(1, __u8, flags); 13171ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(2, __u16, ioprio); 13172ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(4, __s32, fd); 13173ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(8, __u64, off); 13174ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(8, __u64, addr2); 13175ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(16, __u64, addr); 13176ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(16, __u64, splice_off_in); 13177ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(24, __u32, len); 13178ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __kernel_rwf_t, rw_flags); 13179ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, /* compat */ int, rw_flags); 13180ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, /* compat */ __u32, rw_flags); 13181ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, fsync_flags); 13182ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, /* compat */ __u16, poll_events); 13183ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, poll32_events); 13184ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, sync_range_flags); 13185ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, msg_flags); 13186ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, timeout_flags); 13187ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, accept_flags); 13188ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, cancel_flags); 13189ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, open_flags); 13190ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, statx_flags); 13191ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, fadvise_advice); 13192ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(28, __u32, splice_flags); 13193ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(32, __u64, user_data); 13194ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(40, __u16, buf_index); 13195ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(40, __u16, buf_group); 13196ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(42, __u16, personality); 13197ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(44, __s32, splice_fd_in); 13198ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(44, __u32, file_index); 13199ed29b0b4SJens Axboe BUILD_BUG_SQE_ELEM(48, __u64, addr3); 13200ed29b0b4SJens Axboe 13201ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(struct io_uring_files_update) != 13202ed29b0b4SJens Axboe sizeof(struct io_uring_rsrc_update)); 13203ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(struct io_uring_rsrc_update) > 13204ed29b0b4SJens Axboe sizeof(struct io_uring_rsrc_update2)); 13205ed29b0b4SJens Axboe 13206ed29b0b4SJens Axboe /* ->buf_index is u16 */ 13207ed29b0b4SJens Axboe BUILD_BUG_ON(IORING_MAX_REG_BUFFERS >= (1u << 16)); 13208ed29b0b4SJens Axboe BUILD_BUG_ON(BGID_ARRAY * sizeof(struct io_buffer_list) > PAGE_SIZE); 13209ed29b0b4SJens Axboe BUILD_BUG_ON(offsetof(struct io_uring_buf_ring, bufs) != 0); 13210ed29b0b4SJens Axboe BUILD_BUG_ON(offsetof(struct io_uring_buf, resv) != 13211ed29b0b4SJens Axboe offsetof(struct io_uring_buf_ring, tail)); 13212ed29b0b4SJens Axboe 13213ed29b0b4SJens Axboe /* should fit into one byte */ 13214ed29b0b4SJens Axboe BUILD_BUG_ON(SQE_VALID_FLAGS >= (1 << 8)); 13215ed29b0b4SJens Axboe BUILD_BUG_ON(SQE_COMMON_FLAGS >= (1 << 8)); 13216ed29b0b4SJens Axboe BUILD_BUG_ON((SQE_VALID_FLAGS | SQE_COMMON_FLAGS) != SQE_VALID_FLAGS); 13217ed29b0b4SJens Axboe 13218ed29b0b4SJens Axboe BUILD_BUG_ON(ARRAY_SIZE(io_op_defs) != IORING_OP_LAST); 13219ed29b0b4SJens Axboe BUILD_BUG_ON(__REQ_F_LAST_BIT > 8 * sizeof(int)); 13220ed29b0b4SJens Axboe 13221ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(atomic_t) != sizeof(u32)); 13222ed29b0b4SJens Axboe 13223ed29b0b4SJens Axboe BUILD_BUG_ON(sizeof(struct io_uring_cmd) > 64); 13224ed29b0b4SJens Axboe 13225ed29b0b4SJens Axboe for (i = 0; i < ARRAY_SIZE(io_op_defs); i++) { 13226ed29b0b4SJens Axboe BUG_ON(!io_op_defs[i].prep); 13227ed29b0b4SJens Axboe BUG_ON(!io_op_defs[i].issue); 13228ed29b0b4SJens Axboe } 13229ed29b0b4SJens Axboe 13230ed29b0b4SJens Axboe req_cachep = KMEM_CACHE(io_kiocb, SLAB_HWCACHE_ALIGN | SLAB_PANIC | 13231ed29b0b4SJens Axboe SLAB_ACCOUNT); 13232ed29b0b4SJens Axboe return 0; 13233ed29b0b4SJens Axboe }; 13234ed29b0b4SJens Axboe __initcall(io_uring_init); 13235