1 #ifndef FIO_IOENGINE_H 2 #define FIO_IOENGINE_H 3 4 #include "compiler/compiler.h" 5 #include "os/os.h" 6 #include "log.h" 7 #include "io_ddir.h" 8 #include "debug.h" 9 #include "file.h" 10 11 #ifdef CONFIG_LIBAIO 12 #include <libaio.h> 13 #endif 14 #ifdef CONFIG_GUASI 15 #include <guasi.h> 16 #endif 17 18 #define FIO_IOOPS_VERSION 21 19 20 enum { 21 IO_U_F_FREE = 1 << 0, 22 IO_U_F_FLIGHT = 1 << 1, 23 IO_U_F_NO_FILE_PUT = 1 << 2, 24 IO_U_F_IN_CUR_DEPTH = 1 << 3, 25 IO_U_F_BUSY_OK = 1 << 4, 26 IO_U_F_TRIMMED = 1 << 5, 27 IO_U_F_BARRIER = 1 << 6, 28 IO_U_F_VER_LIST = 1 << 7, 29 }; 30 31 /* 32 * The io unit 33 */ 34 struct io_u { 35 struct timeval start_time; 36 struct timeval issue_time; 37 38 struct fio_file *file; 39 unsigned int flags; 40 enum fio_ddir ddir; 41 42 /* 43 * For replay workloads, we may want to account as a different 44 * IO type than what is being submitted. 45 */ 46 enum fio_ddir acct_ddir; 47 48 /* 49 * Allocated/set buffer and length 50 */ 51 unsigned long buflen; 52 unsigned long long offset; 53 unsigned short numberio; 54 void *buf; 55 56 /* 57 * Initial seed for generating the buffer contents 58 */ 59 uint64_t rand_seed; 60 61 /* 62 * IO engine state, may be different from above when we get 63 * partial transfers / residual data counts 64 */ 65 void *xfer_buf; 66 unsigned long xfer_buflen; 67 68 /* 69 * Parameter related to pre-filled buffers and 70 * their size to handle variable block sizes. 71 */ 72 unsigned long buf_filled_len; 73 74 struct io_piece *ipo; 75 76 unsigned int resid; 77 unsigned int error; 78 79 /* 80 * io engine private data 81 */ 82 union { 83 unsigned int index; 84 unsigned int seen; 85 void *engine_data; 86 }; 87 88 struct flist_head verify_list; 89 90 /* 91 * Callback for io completion 92 */ 93 int (*end_io)(struct thread_data *, struct io_u **); 94 95 union { 96 #ifdef CONFIG_LIBAIO 97 struct iocb iocb; 98 #endif 99 #ifdef CONFIG_POSIXAIO 100 os_aiocb_t aiocb; 101 #endif 102 #ifdef FIO_HAVE_SGIO 103 struct sg_io_hdr hdr; 104 #endif 105 #ifdef CONFIG_GUASI 106 guasi_req_t greq; 107 #endif 108 #ifdef CONFIG_SOLARISAIO 109 aio_result_t resultp; 110 #endif 111 #ifdef FIO_HAVE_BINJECT 112 struct b_user_cmd buc; 113 #endif 114 #ifdef CONFIG_RDMA 115 struct ibv_mr *mr; 116 #endif 117 void *mmap_data; 118 }; 119 }; 120 121 /* 122 * io_ops->queue() return values 123 */ 124 enum { 125 FIO_Q_COMPLETED = 0, /* completed sync */ 126 FIO_Q_QUEUED = 1, /* queued, will complete async */ 127 FIO_Q_BUSY = 2, /* no more room, call ->commit() */ 128 }; 129 130 struct ioengine_ops { 131 struct flist_head list; 132 char name[16]; 133 int version; 134 int flags; 135 int (*setup)(struct thread_data *); 136 int (*init)(struct thread_data *); 137 int (*prep)(struct thread_data *, struct io_u *); 138 int (*queue)(struct thread_data *, struct io_u *); 139 int (*commit)(struct thread_data *); 140 int (*getevents)(struct thread_data *, unsigned int, unsigned int, const struct timespec *); 141 struct io_u *(*event)(struct thread_data *, int); 142 int (*cancel)(struct thread_data *, struct io_u *); 143 void (*cleanup)(struct thread_data *); 144 int (*open_file)(struct thread_data *, struct fio_file *); 145 int (*close_file)(struct thread_data *, struct fio_file *); 146 int (*invalidate)(struct thread_data *, struct fio_file *); 147 int (*unlink_file)(struct thread_data *, struct fio_file *); 148 int (*get_file_size)(struct thread_data *, struct fio_file *); 149 void (*terminate)(struct thread_data *); 150 int (*io_u_init)(struct thread_data *, struct io_u *); 151 void (*io_u_free)(struct thread_data *, struct io_u *); 152 int option_struct_size; 153 struct fio_option *options; 154 void *data; 155 void *dlhandle; 156 }; 157 158 enum fio_ioengine_flags { 159 FIO_SYNCIO = 1 << 0, /* io engine has synchronous ->queue */ 160 FIO_RAWIO = 1 << 1, /* some sort of direct/raw io */ 161 FIO_DISKLESSIO = 1 << 2, /* no disk involved */ 162 FIO_NOEXTEND = 1 << 3, /* engine can't extend file */ 163 FIO_NODISKUTIL = 1 << 4, /* diskutil can't handle filename */ 164 FIO_UNIDIR = 1 << 5, /* engine is uni-directional */ 165 FIO_NOIO = 1 << 6, /* thread does only pseudo IO */ 166 FIO_PIPEIO = 1 << 7, /* input/output no seekable */ 167 FIO_BARRIER = 1 << 8, /* engine supports barriers */ 168 FIO_MEMALIGN = 1 << 9, /* engine wants aligned memory */ 169 FIO_BIT_BASED = 1 << 10, /* engine uses a bit base (e.g. uses Kbit as opposed to KB) */ 170 FIO_FAKEIO = 1 << 11, /* engine pretends to do IO */ 171 }; 172 173 /* 174 * External engine defined symbol to fill in the engine ops structure 175 */ 176 typedef void (*get_ioengine_t)(struct ioengine_ops **); 177 178 /* 179 * io engine entry points 180 */ 181 extern int __must_check td_io_init(struct thread_data *); 182 extern int __must_check td_io_prep(struct thread_data *, struct io_u *); 183 extern int __must_check td_io_queue(struct thread_data *, struct io_u *); 184 extern int __must_check td_io_sync(struct thread_data *, struct fio_file *); 185 extern int __must_check td_io_getevents(struct thread_data *, unsigned int, unsigned int, const struct timespec *); 186 extern int __must_check td_io_commit(struct thread_data *); 187 extern int __must_check td_io_open_file(struct thread_data *, struct fio_file *); 188 extern int td_io_close_file(struct thread_data *, struct fio_file *); 189 extern int td_io_unlink_file(struct thread_data *, struct fio_file *); 190 extern int __must_check td_io_get_file_size(struct thread_data *, struct fio_file *); 191 192 extern struct ioengine_ops *load_ioengine(struct thread_data *, const char *); 193 extern void register_ioengine(struct ioengine_ops *); 194 extern void unregister_ioengine(struct ioengine_ops *); 195 extern void free_ioengine(struct thread_data *); 196 extern void close_ioengine(struct thread_data *); 197 198 extern int fio_show_ioengine_help(const char *engine); 199 200 /* 201 * io unit handling 202 */ 203 extern struct io_u *__get_io_u(struct thread_data *); 204 extern struct io_u *get_io_u(struct thread_data *); 205 extern void put_io_u(struct thread_data *, struct io_u *); 206 extern void clear_io_u(struct thread_data *, struct io_u *); 207 extern void requeue_io_u(struct thread_data *, struct io_u **); 208 extern int __must_check io_u_sync_complete(struct thread_data *, struct io_u *, uint64_t *); 209 extern int __must_check io_u_queued_complete(struct thread_data *, int, uint64_t *); 210 extern void io_u_queued(struct thread_data *, struct io_u *); 211 extern void io_u_quiesce(struct thread_data *); 212 extern void io_u_log_error(struct thread_data *, struct io_u *); 213 extern void io_u_mark_depth(struct thread_data *, unsigned int); 214 extern void fill_io_buffer(struct thread_data *, void *, unsigned int, unsigned int); 215 extern void io_u_fill_buffer(struct thread_data *td, struct io_u *, unsigned int, unsigned int); 216 void io_u_mark_complete(struct thread_data *, unsigned int); 217 void io_u_mark_submit(struct thread_data *, unsigned int); 218 int queue_full(const struct thread_data *); 219 220 int do_io_u_sync(const struct thread_data *, struct io_u *); 221 int do_io_u_trim(const struct thread_data *, struct io_u *); 222 223 #ifdef FIO_INC_DEBUG 224 static inline void dprint_io_u(struct io_u *io_u, const char *p) 225 { 226 struct fio_file *f = io_u->file; 227 228 dprint(FD_IO, "%s: io_u %p: off=%llu/len=%lu/ddir=%d", p, io_u, 229 (unsigned long long) io_u->offset, 230 io_u->buflen, io_u->ddir); 231 if (fio_debug & (1 << FD_IO)) { 232 if (f) 233 log_info("/%s", f->file_name); 234 235 log_info("\n"); 236 } 237 } 238 #else 239 #define dprint_io_u(io_u, p) 240 #endif 241 242 static inline enum fio_ddir acct_ddir(struct io_u *io_u) 243 { 244 if (io_u->acct_ddir != -1) 245 return io_u->acct_ddir; 246 247 return io_u->ddir; 248 } 249 250 #endif 251
