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 __clear_bit(bit, table->bitmap); 29 table->alloc_hint = bit; 30 } 31 32 static inline void io_file_bitmap_set(struct io_file_table *table, int bit) 33 { 34 WARN_ON_ONCE(test_bit(bit, table->bitmap)); 35 __set_bit(bit, table->bitmap); 36 table->alloc_hint = bit + 1; 37 } 38 39 static inline struct io_fixed_file * 40 io_fixed_file_slot(struct io_file_table *table, unsigned i) 41 { 42 return &table->files[i]; 43 } 44 45 static inline struct file *io_file_from_index(struct io_file_table *table, 46 int index) 47 { 48 struct io_fixed_file *slot = io_fixed_file_slot(table, index); 49 50 return (struct file *) (slot->file_ptr & FFS_MASK); 51 } 52 53 static inline void io_fixed_file_set(struct io_fixed_file *file_slot, 54 struct file *file) 55 { 56 unsigned long file_ptr = (unsigned long) file; 57 58 file_ptr |= io_file_get_flags(file); 59 file_slot->file_ptr = file_ptr; 60 } 61 62 static inline void io_reset_alloc_hint(struct io_ring_ctx *ctx) 63 { 64 ctx->file_table.alloc_hint = ctx->file_alloc_start; 65 } 66 67 static inline void io_file_table_set_alloc_range(struct io_ring_ctx *ctx, 68 unsigned off, unsigned len) 69 { 70 ctx->file_alloc_start = off; 71 ctx->file_alloc_end = off + len; 72 io_reset_alloc_hint(ctx); 73 } 74 75 #endif 76