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