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
dprint_io_u(struct io_u * io_u,const char * p)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
acct_ddir(struct io_u * io_u)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