Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 1 | /* |
| 2 | * The io parts of the fio tool, includes workers for sync and mmap'ed |
| 3 | * io, as well as both posix and linux libaio support. |
| 4 | * |
| 5 | * sync io is implemented on top of aio. |
| 6 | * |
| 7 | * This is not really specific to fio, if the get_io_u/put_io_u and |
| 8 | * structures was pulled into this as well it would be a perfectly |
| 9 | * generic io engine that could be used for other projects. |
| 10 | * |
| 11 | */ |
| 12 | #include <stdio.h> |
| 13 | #include <stdlib.h> |
| 14 | #include <unistd.h> |
Jens Axboe | 5c4e1db | 2006-06-07 14:17:08 +0200 | [diff] [blame] | 15 | #include <string.h> |
Jens Axboe | 2866c82 | 2006-10-09 15:57:48 +0200 | [diff] [blame] | 16 | #include <dlfcn.h> |
Jens Axboe | 0c6e751 | 2007-02-22 11:19:39 +0100 | [diff] [blame] | 17 | #include <assert.h> |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 18 | |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 19 | #include "fio.h" |
Jens Axboe | 7c9b1bc | 2009-06-03 09:25:57 +0200 | [diff] [blame] | 20 | #include "diskutil.h" |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 21 | |
Jens Axboe | 01743ee | 2008-06-02 12:19:19 +0200 | [diff] [blame] | 22 | static FLIST_HEAD(engine_list); |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 23 | |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 24 | static int check_engine_ops(struct ioengine_ops *ops) |
| 25 | { |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 26 | if (ops->version != FIO_IOOPS_VERSION) { |
Jens Axboe | 5ec10ea | 2008-03-06 15:42:00 +0100 | [diff] [blame] | 27 | log_err("bad ioops version %d (want %d)\n", ops->version, |
| 28 | FIO_IOOPS_VERSION); |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 29 | return 1; |
| 30 | } |
| 31 | |
Jens Axboe | 36167d8 | 2007-02-18 05:41:31 +0100 | [diff] [blame] | 32 | if (!ops->queue) { |
| 33 | log_err("%s: no queue handler\n", ops->name); |
| 34 | return 1; |
| 35 | } |
| 36 | |
| 37 | /* |
| 38 | * sync engines only need a ->queue() |
| 39 | */ |
| 40 | if (ops->flags & FIO_SYNCIO) |
| 41 | return 0; |
Jens Axboe | 5ec10ea | 2008-03-06 15:42:00 +0100 | [diff] [blame] | 42 | |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 43 | if (!ops->event) { |
Jens Axboe | 36167d8 | 2007-02-18 05:41:31 +0100 | [diff] [blame] | 44 | log_err("%s: no event handler\n", ops->name); |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 45 | return 1; |
| 46 | } |
| 47 | if (!ops->getevents) { |
Jens Axboe | 36167d8 | 2007-02-18 05:41:31 +0100 | [diff] [blame] | 48 | log_err("%s: no getevents handler\n", ops->name); |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 49 | return 1; |
| 50 | } |
| 51 | if (!ops->queue) { |
Jens Axboe | 36167d8 | 2007-02-18 05:41:31 +0100 | [diff] [blame] | 52 | log_err("%s: no queue handler\n", ops->name); |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 53 | return 1; |
| 54 | } |
Jens Axboe | 5ec10ea | 2008-03-06 15:42:00 +0100 | [diff] [blame] | 55 | |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 56 | return 0; |
| 57 | } |
| 58 | |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 59 | void unregister_ioengine(struct ioengine_ops *ops) |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 60 | { |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 61 | dprint(FD_IO, "ioengine %s unregistered\n", ops->name); |
Jens Axboe | 01743ee | 2008-06-02 12:19:19 +0200 | [diff] [blame] | 62 | flist_del(&ops->list); |
| 63 | INIT_FLIST_HEAD(&ops->list); |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 64 | } |
| 65 | |
Jens Axboe | b2fdda4 | 2007-02-20 20:52:51 +0100 | [diff] [blame] | 66 | void register_ioengine(struct ioengine_ops *ops) |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 67 | { |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 68 | dprint(FD_IO, "ioengine %s registered\n", ops->name); |
Jens Axboe | 01743ee | 2008-06-02 12:19:19 +0200 | [diff] [blame] | 69 | INIT_FLIST_HEAD(&ops->list); |
| 70 | flist_add_tail(&ops->list, &engine_list); |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 71 | } |
| 72 | |
| 73 | static struct ioengine_ops *find_ioengine(const char *name) |
| 74 | { |
| 75 | struct ioengine_ops *ops; |
Jens Axboe | 01743ee | 2008-06-02 12:19:19 +0200 | [diff] [blame] | 76 | struct flist_head *entry; |
Jens Axboe | 2866c82 | 2006-10-09 15:57:48 +0200 | [diff] [blame] | 77 | |
Jens Axboe | 01743ee | 2008-06-02 12:19:19 +0200 | [diff] [blame] | 78 | flist_for_each(entry, &engine_list) { |
| 79 | ops = flist_entry(entry, struct ioengine_ops, list); |
Jens Axboe | bc5b77a | 2007-02-28 09:32:54 +0100 | [diff] [blame] | 80 | if (!strcmp(name, ops->name)) |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 81 | return ops; |
| 82 | } |
| 83 | |
| 84 | return NULL; |
| 85 | } |
| 86 | |
| 87 | static struct ioengine_ops *dlopen_ioengine(struct thread_data *td, |
| 88 | const char *engine_lib) |
| 89 | { |
| 90 | struct ioengine_ops *ops; |
| 91 | void *dlhandle; |
| 92 | |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 93 | dprint(FD_IO, "dload engine %s\n", engine_lib); |
| 94 | |
Jens Axboe | 2866c82 | 2006-10-09 15:57:48 +0200 | [diff] [blame] | 95 | dlerror(); |
| 96 | dlhandle = dlopen(engine_lib, RTLD_LAZY); |
Jens Axboe | d4dbaaa | 2006-10-09 16:01:27 +0200 | [diff] [blame] | 97 | if (!dlhandle) { |
Jens Axboe | e1161c3 | 2007-02-22 19:36:48 +0100 | [diff] [blame] | 98 | td_vmsg(td, -1, dlerror(), "dlopen"); |
Jens Axboe | d4dbaaa | 2006-10-09 16:01:27 +0200 | [diff] [blame] | 99 | return NULL; |
| 100 | } |
Jens Axboe | 2866c82 | 2006-10-09 15:57:48 +0200 | [diff] [blame] | 101 | |
Jens Axboe | da51c05 | 2006-11-07 16:02:11 +0100 | [diff] [blame] | 102 | /* |
| 103 | * Unlike the included modules, external engines should have a |
| 104 | * non-static ioengine structure that we can reference. |
| 105 | */ |
Jens Axboe | 2866c82 | 2006-10-09 15:57:48 +0200 | [diff] [blame] | 106 | ops = dlsym(dlhandle, "ioengine"); |
Jens Axboe | d4dbaaa | 2006-10-09 16:01:27 +0200 | [diff] [blame] | 107 | if (!ops) { |
Jens Axboe | e1161c3 | 2007-02-22 19:36:48 +0100 | [diff] [blame] | 108 | td_vmsg(td, -1, dlerror(), "dlsym"); |
Jens Axboe | d4dbaaa | 2006-10-09 16:01:27 +0200 | [diff] [blame] | 109 | dlclose(dlhandle); |
| 110 | return NULL; |
| 111 | } |
Jens Axboe | 2866c82 | 2006-10-09 15:57:48 +0200 | [diff] [blame] | 112 | |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 113 | ops->dlhandle = dlhandle; |
| 114 | return ops; |
| 115 | } |
| 116 | |
| 117 | struct ioengine_ops *load_ioengine(struct thread_data *td, const char *name) |
| 118 | { |
| 119 | struct ioengine_ops *ops, *ret; |
| 120 | char engine[16]; |
| 121 | |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 122 | dprint(FD_IO, "load ioengine %s\n", name); |
| 123 | |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 124 | strncpy(engine, name, sizeof(engine) - 1); |
| 125 | |
| 126 | /* |
| 127 | * linux libaio has alias names, so convert to what we want |
| 128 | */ |
| 129 | if (!strncmp(engine, "linuxaio", 8) || !strncmp(engine, "aio", 3)) |
| 130 | strcpy(engine, "libaio"); |
| 131 | |
| 132 | ops = find_ioengine(engine); |
| 133 | if (!ops) |
| 134 | ops = dlopen_ioengine(td, name); |
| 135 | |
| 136 | if (!ops) { |
| 137 | log_err("fio: engine %s not loadable\n", name); |
Jens Axboe | b902ceb | 2006-10-09 16:11:45 +0200 | [diff] [blame] | 138 | return NULL; |
| 139 | } |
| 140 | |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 141 | /* |
| 142 | * Check that the required methods are there. |
| 143 | */ |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 144 | if (check_engine_ops(ops)) |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 145 | return NULL; |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 146 | |
Jens Axboe | 8458500 | 2006-10-19 20:26:22 +0200 | [diff] [blame] | 147 | ret = malloc(sizeof(*ret)); |
| 148 | memcpy(ret, ops, sizeof(*ret)); |
| 149 | ret->data = NULL; |
Jens Axboe | 8458500 | 2006-10-19 20:26:22 +0200 | [diff] [blame] | 150 | |
| 151 | return ret; |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 152 | } |
| 153 | |
Jens Axboe | 2866c82 | 2006-10-09 15:57:48 +0200 | [diff] [blame] | 154 | void close_ioengine(struct thread_data *td) |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 155 | { |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 156 | dprint(FD_IO, "close ioengine %s\n", td->io_ops->name); |
| 157 | |
Jens Axboe | 2992b05 | 2008-05-30 22:07:12 +0200 | [diff] [blame] | 158 | if (td->io_ops->cleanup) { |
Jens Axboe | 2866c82 | 2006-10-09 15:57:48 +0200 | [diff] [blame] | 159 | td->io_ops->cleanup(td); |
Jens Axboe | 2992b05 | 2008-05-30 22:07:12 +0200 | [diff] [blame] | 160 | td->io_ops->data = NULL; |
| 161 | } |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 162 | |
Jens Axboe | 5f35095 | 2006-11-07 15:20:59 +0100 | [diff] [blame] | 163 | if (td->io_ops->dlhandle) |
| 164 | dlclose(td->io_ops->dlhandle); |
| 165 | |
Jens Axboe | 8458500 | 2006-10-19 20:26:22 +0200 | [diff] [blame] | 166 | free(td->io_ops); |
| 167 | td->io_ops = NULL; |
Jens Axboe | b990b5c | 2006-09-14 09:48:22 +0200 | [diff] [blame] | 168 | } |
Jens Axboe | 10ba535 | 2006-10-20 11:39:27 +0200 | [diff] [blame] | 169 | |
| 170 | int td_io_prep(struct thread_data *td, struct io_u *io_u) |
| 171 | { |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 172 | dprint_io_u(io_u, "prep"); |
Jens Axboe | 7101d9c | 2007-09-12 13:12:39 +0200 | [diff] [blame] | 173 | fio_ro_check(td, io_u); |
| 174 | |
Jens Axboe | 4d4e80f | 2008-03-04 10:18:56 +0100 | [diff] [blame] | 175 | lock_file(td, io_u->file, io_u->ddir); |
Jens Axboe | b2bd2bd | 2008-03-01 18:19:52 +0100 | [diff] [blame] | 176 | |
Jens Axboe | 2ba1c29 | 2008-02-01 13:16:38 +0100 | [diff] [blame] | 177 | if (td->io_ops->prep) { |
| 178 | int ret = td->io_ops->prep(td, io_u); |
| 179 | |
| 180 | dprint(FD_IO, "->prep(%p)=%d\n", io_u, ret); |
Jens Axboe | b2bd2bd | 2008-03-01 18:19:52 +0100 | [diff] [blame] | 181 | if (ret) |
Jens Axboe | 4d4e80f | 2008-03-04 10:18:56 +0100 | [diff] [blame] | 182 | unlock_file(td, io_u->file); |
Jens Axboe | 2ba1c29 | 2008-02-01 13:16:38 +0100 | [diff] [blame] | 183 | return ret; |
| 184 | } |
Jens Axboe | 10ba535 | 2006-10-20 11:39:27 +0200 | [diff] [blame] | 185 | |
| 186 | return 0; |
| 187 | } |
| 188 | |
Jens Axboe | e7d2e61 | 2007-12-11 10:49:39 +0100 | [diff] [blame] | 189 | int td_io_getevents(struct thread_data *td, unsigned int min, unsigned int max, |
Jens Axboe | 10ba535 | 2006-10-20 11:39:27 +0200 | [diff] [blame] | 190 | struct timespec *t) |
| 191 | { |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 192 | int r = 0; |
| 193 | |
Jens Axboe | face81b | 2007-02-26 10:40:03 +0100 | [diff] [blame] | 194 | if (min > 0 && td->io_ops->commit) { |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 195 | r = td->io_ops->commit(td); |
Jens Axboe | face81b | 2007-02-26 10:40:03 +0100 | [diff] [blame] | 196 | if (r < 0) |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 197 | goto out; |
Jens Axboe | face81b | 2007-02-26 10:40:03 +0100 | [diff] [blame] | 198 | } |
Jens Axboe | 4950421 | 2008-06-05 09:03:30 +0200 | [diff] [blame] | 199 | if (max > td->cur_depth) |
| 200 | max = td->cur_depth; |
| 201 | if (min > max) |
| 202 | max = min; |
Jens Axboe | 36167d8 | 2007-02-18 05:41:31 +0100 | [diff] [blame] | 203 | |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 204 | r = 0; |
Jens Axboe | 4950421 | 2008-06-05 09:03:30 +0200 | [diff] [blame] | 205 | if (max && td->io_ops->getevents) |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 206 | r = td->io_ops->getevents(td, min, max, t); |
| 207 | out: |
Jens Axboe | 838bc70 | 2008-05-22 13:08:23 +0200 | [diff] [blame] | 208 | if (r >= 0) |
| 209 | io_u_mark_complete(td, r); |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 210 | dprint(FD_IO, "getevents: %d\n", r); |
| 211 | return r; |
Jens Axboe | 10ba535 | 2006-10-20 11:39:27 +0200 | [diff] [blame] | 212 | } |
| 213 | |
| 214 | int td_io_queue(struct thread_data *td, struct io_u *io_u) |
| 215 | { |
Jens Axboe | 7e77dd0 | 2007-02-20 10:57:34 +0100 | [diff] [blame] | 216 | int ret; |
| 217 | |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 218 | dprint_io_u(io_u, "queue"); |
Jens Axboe | 7101d9c | 2007-09-12 13:12:39 +0200 | [diff] [blame] | 219 | fio_ro_check(td, io_u); |
| 220 | |
Jens Axboe | 0c6e751 | 2007-02-22 11:19:39 +0100 | [diff] [blame] | 221 | assert((io_u->flags & IO_U_F_FLIGHT) == 0); |
| 222 | io_u->flags |= IO_U_F_FLIGHT; |
| 223 | |
Jens Axboe | d6aed79 | 2009-06-03 08:41:15 +0200 | [diff] [blame] | 224 | assert(fio_file_open(io_u->file)); |
Jens Axboe | 3d7b485 | 2007-03-13 14:50:28 +0100 | [diff] [blame] | 225 | |
Jens Axboe | 1178680 | 2007-03-05 18:33:22 +0100 | [diff] [blame] | 226 | io_u->error = 0; |
| 227 | io_u->resid = 0; |
| 228 | |
Jens Axboe | 433afcb | 2007-02-22 10:39:01 +0100 | [diff] [blame] | 229 | if (td->io_ops->flags & FIO_SYNCIO) { |
Jens Axboe | 12d9d84 | 2008-10-16 21:26:10 +0200 | [diff] [blame] | 230 | if (fio_fill_issue_time(td)) |
Jens Axboe | 9520ebb | 2008-10-16 21:03:27 +0200 | [diff] [blame] | 231 | fio_gettime(&io_u->issue_time, NULL); |
Jens Axboe | d0c1532 | 2008-10-16 20:33:51 +0200 | [diff] [blame] | 232 | |
| 233 | /* |
| 234 | * only used for iolog |
| 235 | */ |
| 236 | if (td->o.read_iolog_file) |
| 237 | memcpy(&td->last_issue, &io_u->issue_time, |
Jens Axboe | 5ec10ea | 2008-03-06 15:42:00 +0100 | [diff] [blame] | 238 | sizeof(struct timeval)); |
Jens Axboe | 433afcb | 2007-02-22 10:39:01 +0100 | [diff] [blame] | 239 | } |
| 240 | |
Jens Axboe | 5f9099e | 2009-06-16 22:40:26 +0200 | [diff] [blame^] | 241 | if (!ddir_sync(io_u->ddir)) |
Jens Axboe | 755200a | 2007-02-19 13:08:12 +0100 | [diff] [blame] | 242 | td->io_issues[io_u->ddir]++; |
| 243 | |
Jens Axboe | 7e77dd0 | 2007-02-20 10:57:34 +0100 | [diff] [blame] | 244 | ret = td->io_ops->queue(td, io_u); |
Jens Axboe | 5aeb77d | 2007-02-20 11:34:54 +0100 | [diff] [blame] | 245 | |
Jens Axboe | 4d4e80f | 2008-03-04 10:18:56 +0100 | [diff] [blame] | 246 | unlock_file(td, io_u->file); |
Jens Axboe | b2bd2bd | 2008-03-01 18:19:52 +0100 | [diff] [blame] | 247 | |
Jens Axboe | 838bc70 | 2008-05-22 13:08:23 +0200 | [diff] [blame] | 248 | if (!td->io_ops->commit) { |
| 249 | io_u_mark_submit(td, 1); |
| 250 | io_u_mark_complete(td, 1); |
| 251 | } |
| 252 | |
Jens Axboe | d800575 | 2008-05-15 09:49:09 +0200 | [diff] [blame] | 253 | if (ret == FIO_Q_COMPLETED) { |
Jens Axboe | 5f9099e | 2009-06-16 22:40:26 +0200 | [diff] [blame^] | 254 | if (!ddir_sync(io_u->ddir)) { |
Jens Axboe | d800575 | 2008-05-15 09:49:09 +0200 | [diff] [blame] | 255 | io_u_mark_depth(td, 1); |
| 256 | td->ts.total_io_u[io_u->ddir]++; |
| 257 | } |
| 258 | } else if (ret == FIO_Q_QUEUED) { |
Jens Axboe | eb7c8ae | 2007-02-26 14:45:12 +0100 | [diff] [blame] | 259 | int r; |
| 260 | |
Jens Axboe | 5f9099e | 2009-06-16 22:40:26 +0200 | [diff] [blame^] | 261 | if (!ddir_sync(io_u->ddir)) { |
Jens Axboe | d800575 | 2008-05-15 09:49:09 +0200 | [diff] [blame] | 262 | td->io_u_queued++; |
| 263 | td->ts.total_io_u[io_u->ddir]++; |
| 264 | } |
| 265 | |
| 266 | if (td->io_u_queued >= td->o.iodepth_batch) { |
Jens Axboe | eb7c8ae | 2007-02-26 14:45:12 +0100 | [diff] [blame] | 267 | r = td_io_commit(td); |
| 268 | if (r < 0) |
| 269 | return r; |
| 270 | } |
| 271 | } |
Jens Axboe | cb5ab51 | 2007-02-26 12:57:09 +0100 | [diff] [blame] | 272 | |
Jens Axboe | 433afcb | 2007-02-22 10:39:01 +0100 | [diff] [blame] | 273 | if ((td->io_ops->flags & FIO_SYNCIO) == 0) { |
Jens Axboe | 12d9d84 | 2008-10-16 21:26:10 +0200 | [diff] [blame] | 274 | if (fio_fill_issue_time(td)) |
Jens Axboe | 9520ebb | 2008-10-16 21:03:27 +0200 | [diff] [blame] | 275 | fio_gettime(&io_u->issue_time, NULL); |
Jens Axboe | d0c1532 | 2008-10-16 20:33:51 +0200 | [diff] [blame] | 276 | |
| 277 | /* |
| 278 | * only used for iolog |
| 279 | */ |
| 280 | if (td->o.read_iolog_file) |
| 281 | memcpy(&td->last_issue, &io_u->issue_time, |
| 282 | sizeof(struct timeval)); |
Jens Axboe | 433afcb | 2007-02-22 10:39:01 +0100 | [diff] [blame] | 283 | } |
| 284 | |
Jens Axboe | 7e77dd0 | 2007-02-20 10:57:34 +0100 | [diff] [blame] | 285 | return ret; |
Jens Axboe | 10ba535 | 2006-10-20 11:39:27 +0200 | [diff] [blame] | 286 | } |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 287 | |
| 288 | int td_io_init(struct thread_data *td) |
| 289 | { |
Jens Axboe | eeb1216 | 2007-03-20 10:47:45 +0100 | [diff] [blame] | 290 | int ret = 0; |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 291 | |
Jens Axboe | eeb1216 | 2007-03-20 10:47:45 +0100 | [diff] [blame] | 292 | if (td->io_ops->init) { |
| 293 | ret = td->io_ops->init(td); |
Jens Axboe | 5ec10ea | 2008-03-06 15:42:00 +0100 | [diff] [blame] | 294 | if (ret && td->o.iodepth > 1) { |
| 295 | log_err("fio: io engine init failed. Perhaps try" |
| 296 | " reducing io depth?\n"); |
| 297 | } |
Jens Axboe | eeb1216 | 2007-03-20 10:47:45 +0100 | [diff] [blame] | 298 | } |
| 299 | |
| 300 | return ret; |
Jens Axboe | 8c16d84 | 2006-10-20 11:48:33 +0200 | [diff] [blame] | 301 | } |
Jens Axboe | 755200a | 2007-02-19 13:08:12 +0100 | [diff] [blame] | 302 | |
| 303 | int td_io_commit(struct thread_data *td) |
| 304 | { |
Jens Axboe | ee56ad5 | 2008-02-01 10:30:20 +0100 | [diff] [blame] | 305 | dprint(FD_IO, "calling ->commit(), depth %d\n", td->cur_depth); |
| 306 | |
Jens Axboe | d800575 | 2008-05-15 09:49:09 +0200 | [diff] [blame] | 307 | if (!td->cur_depth || !td->io_u_queued) |
Jens Axboe | e1161c3 | 2007-02-22 19:36:48 +0100 | [diff] [blame] | 308 | return 0; |
Jens Axboe | cb5ab51 | 2007-02-26 12:57:09 +0100 | [diff] [blame] | 309 | |
Jens Axboe | d800575 | 2008-05-15 09:49:09 +0200 | [diff] [blame] | 310 | io_u_mark_depth(td, td->io_u_queued); |
Jens Axboe | cb5ab51 | 2007-02-26 12:57:09 +0100 | [diff] [blame] | 311 | td->io_u_queued = 0; |
Jens Axboe | d800575 | 2008-05-15 09:49:09 +0200 | [diff] [blame] | 312 | |
Jens Axboe | 755200a | 2007-02-19 13:08:12 +0100 | [diff] [blame] | 313 | if (td->io_ops->commit) |
| 314 | return td->io_ops->commit(td); |
| 315 | |
| 316 | return 0; |
| 317 | } |
Jens Axboe | b5af829 | 2007-03-08 12:43:13 +0100 | [diff] [blame] | 318 | |
| 319 | int td_io_open_file(struct thread_data *td, struct fio_file *f) |
| 320 | { |
Jens Axboe | 22a57ba | 2009-06-09 14:34:35 +0200 | [diff] [blame] | 321 | assert(!fio_file_open(f)); |
| 322 | assert(f->fd == -1); |
| 323 | |
Jens Axboe | 413d669 | 2007-03-27 15:38:21 +0200 | [diff] [blame] | 324 | if (td->io_ops->open_file(td, f)) { |
| 325 | if (td->error == EINVAL && td->o.odirect) |
| 326 | log_err("fio: destination does not support O_DIRECT\n"); |
Jens Axboe | 5ec10ea | 2008-03-06 15:42:00 +0100 | [diff] [blame] | 327 | if (td->error == EMFILE) { |
| 328 | log_err("fio: try reducing/setting openfiles (failed" |
| 329 | " at %u of %u)\n", td->nr_open_files, |
| 330 | td->o.nr_files); |
| 331 | } |
Jens Axboe | b5af829 | 2007-03-08 12:43:13 +0100 | [diff] [blame] | 332 | |
Jens Axboe | 22a57ba | 2009-06-09 14:34:35 +0200 | [diff] [blame] | 333 | assert(f->fd == -1); |
| 334 | assert(!fio_file_open(f)); |
Jens Axboe | 413d669 | 2007-03-27 15:38:21 +0200 | [diff] [blame] | 335 | return 1; |
Jens Axboe | 7bb48f8 | 2007-03-27 15:30:28 +0200 | [diff] [blame] | 336 | } |
Jens Axboe | a978ba6 | 2007-03-08 14:29:03 +0100 | [diff] [blame] | 337 | |
Jens Axboe | d5707a3 | 2008-06-04 15:13:02 +0200 | [diff] [blame] | 338 | fio_file_reset(f); |
Jens Axboe | d6aed79 | 2009-06-03 08:41:15 +0200 | [diff] [blame] | 339 | fio_file_set_open(f); |
| 340 | fio_file_clear_closing(f); |
Jens Axboe | c97bd0f | 2009-05-27 13:11:20 +0200 | [diff] [blame] | 341 | disk_util_inc(f->du); |
Jens Axboe | d5707a3 | 2008-06-04 15:13:02 +0200 | [diff] [blame] | 342 | |
| 343 | td->nr_open_files++; |
| 344 | get_file(f); |
| 345 | |
Jens Axboe | 6615982 | 2007-04-16 21:54:24 +0200 | [diff] [blame] | 346 | if (f->filetype == FIO_TYPE_PIPE) { |
| 347 | if (td_random(td)) { |
| 348 | log_err("fio: can't seek on pipes (no random io)\n"); |
| 349 | goto err; |
| 350 | } |
| 351 | } |
| 352 | |
Jens Axboe | 413d669 | 2007-03-27 15:38:21 +0200 | [diff] [blame] | 353 | if (td->io_ops->flags & FIO_DISKLESSIO) |
| 354 | goto done; |
| 355 | |
| 356 | if (td->o.invalidate_cache && file_invalidate_cache(td, f)) |
| 357 | goto err; |
| 358 | |
Jens Axboe | 6615982 | 2007-04-16 21:54:24 +0200 | [diff] [blame] | 359 | if (td->o.fadvise_hint && |
| 360 | (f->filetype == FIO_TYPE_BD || f->filetype == FIO_TYPE_FILE)) { |
Jens Axboe | 413d669 | 2007-03-27 15:38:21 +0200 | [diff] [blame] | 361 | int flags; |
| 362 | |
| 363 | if (td_random(td)) |
| 364 | flags = POSIX_FADV_RANDOM; |
| 365 | else |
| 366 | flags = POSIX_FADV_SEQUENTIAL; |
| 367 | |
| 368 | if (fadvise(f->fd, f->file_offset, f->io_size, flags) < 0) { |
| 369 | td_verror(td, errno, "fadvise"); |
| 370 | goto err; |
| 371 | } |
| 372 | } |
| 373 | |
Jens Axboe | e116f2b | 2008-06-04 15:14:24 +0200 | [diff] [blame] | 374 | #ifdef FIO_OS_DIRECTIO |
| 375 | /* |
| 376 | * Some OS's have a distinct call to mark the file non-buffered, |
| 377 | * instead of using O_DIRECT (Solaris) |
| 378 | */ |
| 379 | if (td->o.odirect) { |
| 380 | int ret = fio_set_odirect(f->fd); |
| 381 | |
| 382 | if (ret) { |
| 383 | td_verror(td, ret, "fio_set_odirect"); |
| 384 | goto err; |
| 385 | } |
| 386 | } |
| 387 | #endif |
| 388 | |
Jens Axboe | 413d669 | 2007-03-27 15:38:21 +0200 | [diff] [blame] | 389 | done: |
Jens Axboe | f29b25a | 2007-07-23 08:56:43 +0200 | [diff] [blame] | 390 | log_file(td, f, FIO_LOG_OPEN_FILE); |
Jens Axboe | 413d669 | 2007-03-27 15:38:21 +0200 | [diff] [blame] | 391 | return 0; |
| 392 | err: |
Jens Axboe | c97bd0f | 2009-05-27 13:11:20 +0200 | [diff] [blame] | 393 | disk_util_dec(f->du); |
Jens Axboe | b284075 | 2007-04-12 12:57:27 +0200 | [diff] [blame] | 394 | if (td->io_ops->close_file) |
| 395 | td->io_ops->close_file(td, f); |
Jens Axboe | 7bb48f8 | 2007-03-27 15:30:28 +0200 | [diff] [blame] | 396 | return 1; |
Jens Axboe | b5af829 | 2007-03-08 12:43:13 +0100 | [diff] [blame] | 397 | } |
| 398 | |
Jens Axboe | 6977bcd | 2008-03-01 15:55:36 +0100 | [diff] [blame] | 399 | int td_io_close_file(struct thread_data *td, struct fio_file *f) |
Jens Axboe | b5af829 | 2007-03-08 12:43:13 +0100 | [diff] [blame] | 400 | { |
Jens Axboe | d6aed79 | 2009-06-03 08:41:15 +0200 | [diff] [blame] | 401 | if (!fio_file_closing(f)) |
Jens Axboe | f29b25a | 2007-07-23 08:56:43 +0200 | [diff] [blame] | 402 | log_file(td, f, FIO_LOG_CLOSE_FILE); |
| 403 | |
Jens Axboe | 0ad920e | 2007-03-13 11:06:45 +0100 | [diff] [blame] | 404 | /* |
| 405 | * mark as closing, do real close when last io on it has completed |
| 406 | */ |
Jens Axboe | d6aed79 | 2009-06-03 08:41:15 +0200 | [diff] [blame] | 407 | fio_file_set_closing(f); |
Jens Axboe | 0ad920e | 2007-03-13 11:06:45 +0100 | [diff] [blame] | 408 | |
Jens Axboe | c97bd0f | 2009-05-27 13:11:20 +0200 | [diff] [blame] | 409 | disk_util_dec(f->du); |
Jens Axboe | 4d4e80f | 2008-03-04 10:18:56 +0100 | [diff] [blame] | 410 | unlock_file_all(td, f); |
Jens Axboe | 29c1349 | 2008-03-01 19:25:20 +0100 | [diff] [blame] | 411 | |
Jens Axboe | 6977bcd | 2008-03-01 15:55:36 +0100 | [diff] [blame] | 412 | return put_file(td, f); |
Jens Axboe | b5af829 | 2007-03-08 12:43:13 +0100 | [diff] [blame] | 413 | } |
Jens Axboe | df9c26b | 2009-03-05 10:13:58 +0100 | [diff] [blame] | 414 | |
| 415 | int td_io_get_file_size(struct thread_data *td, struct fio_file *f) |
| 416 | { |
| 417 | if (!td->io_ops->get_file_size) |
| 418 | return 0; |
| 419 | |
| 420 | return td->io_ops->get_file_size(td, f); |
| 421 | } |