1 // SPDX-License-Identifier: GPL-2.0 2 #ifndef IOU_FILE_TABLE_H 3 #define IOU_FILE_TABLE_H 4 5 #include <linux/file.h> 6 #include <linux/io_uring_types.h> 7 8 #define FFS_NOWAIT 0x1UL 9 #define FFS_ISREG 0x2UL 10 #define FFS_MASK ~(FFS_NOWAIT|FFS_ISREG) 11 12 bool io_alloc_file_tables(struct io_file_table *table, unsigned nr_files); 13 void io_free_file_tables(struct io_file_table *table); 14 15 int io_fixed_fd_install(struct io_kiocb *req, unsigned int issue_flags, 16 struct file *file, unsigned int file_slot); 17 int __io_fixed_fd_install(struct io_ring_ctx *ctx, struct file *file, 18 unsigned int file_slot); 19 int io_fixed_fd_remove(struct io_ring_ctx *ctx, unsigned int offset); 20 21 int io_register_file_alloc_range(struct io_ring_ctx *ctx, 22 struct io_uring_file_index_range __user *arg); 23 24 unsigned int io_file_get_flags(struct file *file); 25 26 static inline void io_file_bitmap_clear(struct io_file_table *table, int bit) 27 { 28 WARN_ON_ONCE(!test_bit(bit, table->bitmap)); 29 __clear_bit(bit, table->bitmap); 30 table->alloc_hint = bit; 31 } 32 33 static inline void io_file_bitmap_set(struct io_file_table *table, int bit) 34 { 35 WARN_ON_ONCE(test_bit(bit, table->bitmap)); 36 __set_bit(bit, table->bitmap); 37 table->alloc_hint = bit + 1; 38 } 39 40 static inline struct io_fixed_file * 41 io_fixed_file_slot(struct io_file_table *table, unsigned i) 42 { 43 return &table->files[i]; 44 } 45 46 static inline struct file *io_file_from_index(struct io_file_table *table, 47 int index) 48 { 49 struct io_fixed_file *slot = io_fixed_file_slot(table, index); 50 51 return (struct file *) (slot->file_ptr & FFS_MASK); 52 } 53 54 static inline void io_fixed_file_set(struct io_fixed_file *file_slot, 55 struct file *file) 56 { 57 file_slot->file_ptr = (unsigned long)file | 58 (io_file_get_flags(file) >> REQ_F_SUPPORT_NOWAIT_BIT); 59 } 60 61 static inline void io_reset_alloc_hint(struct io_ring_ctx *ctx) 62 { 63 ctx->file_table.alloc_hint = ctx->file_alloc_start; 64 } 65 66 static inline void io_file_table_set_alloc_range(struct io_ring_ctx *ctx, 67 unsigned off, unsigned len) 68 { 69 ctx->file_alloc_start = off; 70 ctx->file_alloc_end = off + len; 71 io_reset_alloc_hint(ctx); 72 } 73 74 #endif 75