1453b329bSJens Axboe // SPDX-License-Identifier: GPL-2.0 2453b329bSJens Axboe #ifndef IOU_FILE_TABLE_H 3453b329bSJens Axboe #define IOU_FILE_TABLE_H 4453b329bSJens Axboe 5453b329bSJens Axboe struct io_ring_ctx; 6453b329bSJens Axboe 7453b329bSJens Axboe /* 8453b329bSJens Axboe * FFS_SCM is only available on 64-bit archs, for 32-bit we just define it as 0 9453b329bSJens Axboe * and define IO_URING_SCM_ALL. For this case, we use SCM for all files as we 10453b329bSJens Axboe * can't safely always dereference the file when the task has exited and ring 11453b329bSJens Axboe * cleanup is done. If a file is tracked and part of SCM, then unix gc on 12453b329bSJens Axboe * process exit may reap it before __io_sqe_files_unregister() is run. 13453b329bSJens Axboe */ 14453b329bSJens Axboe #define FFS_NOWAIT 0x1UL 15453b329bSJens Axboe #define FFS_ISREG 0x2UL 16453b329bSJens Axboe #if defined(CONFIG_64BIT) 17453b329bSJens Axboe #define FFS_SCM 0x4UL 18453b329bSJens Axboe #else 19453b329bSJens Axboe #define IO_URING_SCM_ALL 20453b329bSJens Axboe #define FFS_SCM 0x0UL 21453b329bSJens Axboe #endif 22453b329bSJens Axboe #define FFS_MASK ~(FFS_NOWAIT|FFS_ISREG|FFS_SCM) 23453b329bSJens Axboe 24453b329bSJens Axboe struct io_fixed_file { 25453b329bSJens Axboe /* file * with additional FFS_* flags */ 26453b329bSJens Axboe unsigned long file_ptr; 27453b329bSJens Axboe }; 28453b329bSJens Axboe 29453b329bSJens Axboe struct io_file_table { 30453b329bSJens Axboe struct io_fixed_file *files; 31453b329bSJens Axboe unsigned long *bitmap; 32453b329bSJens Axboe unsigned int alloc_hint; 33453b329bSJens Axboe }; 34453b329bSJens Axboe 35453b329bSJens Axboe bool io_alloc_file_tables(struct io_file_table *table, unsigned nr_files); 36453b329bSJens Axboe void io_free_file_tables(struct io_file_table *table); 37453b329bSJens Axboe int io_file_bitmap_get(struct io_ring_ctx *ctx); 38453b329bSJens Axboe 39*a4ad4f74SJens Axboe unsigned int io_file_get_flags(struct file *file); 40*a4ad4f74SJens Axboe 41453b329bSJens Axboe static inline void io_file_bitmap_clear(struct io_file_table *table, int bit) 42453b329bSJens Axboe { 43453b329bSJens Axboe __clear_bit(bit, table->bitmap); 44453b329bSJens Axboe table->alloc_hint = bit; 45453b329bSJens Axboe } 46453b329bSJens Axboe 47453b329bSJens Axboe static inline void io_file_bitmap_set(struct io_file_table *table, int bit) 48453b329bSJens Axboe { 49453b329bSJens Axboe WARN_ON_ONCE(test_bit(bit, table->bitmap)); 50453b329bSJens Axboe __set_bit(bit, table->bitmap); 51453b329bSJens Axboe table->alloc_hint = bit + 1; 52453b329bSJens Axboe } 53453b329bSJens Axboe 54453b329bSJens Axboe static inline struct io_fixed_file * 55453b329bSJens Axboe io_fixed_file_slot(struct io_file_table *table, unsigned i) 56453b329bSJens Axboe { 57453b329bSJens Axboe return &table->files[i]; 58453b329bSJens Axboe } 59453b329bSJens Axboe 60*a4ad4f74SJens Axboe static inline struct file *io_file_from_index(struct io_file_table *table, 61*a4ad4f74SJens Axboe int index) 62*a4ad4f74SJens Axboe { 63*a4ad4f74SJens Axboe struct io_fixed_file *slot = io_fixed_file_slot(table, index); 64*a4ad4f74SJens Axboe 65*a4ad4f74SJens Axboe return (struct file *) (slot->file_ptr & FFS_MASK); 66*a4ad4f74SJens Axboe } 67*a4ad4f74SJens Axboe 68*a4ad4f74SJens Axboe static inline void io_fixed_file_set(struct io_fixed_file *file_slot, 69*a4ad4f74SJens Axboe struct file *file) 70*a4ad4f74SJens Axboe { 71*a4ad4f74SJens Axboe unsigned long file_ptr = (unsigned long) file; 72*a4ad4f74SJens Axboe 73*a4ad4f74SJens Axboe file_ptr |= io_file_get_flags(file); 74*a4ad4f74SJens Axboe file_slot->file_ptr = file_ptr; 75*a4ad4f74SJens Axboe } 76*a4ad4f74SJens Axboe 77453b329bSJens Axboe #endif 78