blob: 7f52a243fbdbc8e2b12706c1f45c52716729a8ab [file] [log] [blame]
Jens Axboe10ba5352006-10-20 11:39:27 +02001#include <unistd.h>
2#include <fcntl.h>
3#include <string.h>
4#include <signal.h>
5#include <time.h>
Jens Axboe0c6e7512007-02-22 11:19:39 +01006#include <assert.h>
Jens Axboe10ba5352006-10-20 11:39:27 +02007
8#include "fio.h"
Jens Axboe10ba5352006-10-20 11:39:27 +02009
Jens Axboe5945b9b2007-02-22 20:33:01 +010010/*
11 * Change this define to play with the timeout handling
12 */
13#undef FIO_USE_TIMEOUT
14
Jens Axboe97601022007-02-18 12:47:29 +010015struct io_completion_data {
16 int nr; /* input */
Jens Axboe97601022007-02-18 12:47:29 +010017
18 int error; /* output */
19 unsigned long bytes_done[2]; /* output */
20 struct timeval time; /* output */
21};
22
Jens Axboe10ba5352006-10-20 11:39:27 +020023/*
24 * The ->file_map[] contains a map of blocks we have or have not done io
25 * to yet. Used to make sure we cover the entire range in a fair fashion.
26 */
Jens Axboeaec2de22008-04-24 12:44:42 +020027static int random_map_free(struct fio_file *f, const unsigned long long block)
Jens Axboe10ba5352006-10-20 11:39:27 +020028{
Jens Axboeaec2de22008-04-24 12:44:42 +020029 unsigned int idx = RAND_MAP_IDX(f, block);
30 unsigned int bit = RAND_MAP_BIT(f, block);
Jens Axboe10ba5352006-10-20 11:39:27 +020031
Jens Axboe84422ac2008-02-19 20:10:26 +010032 dprint(FD_RANDOM, "free: b=%llu, idx=%u, bit=%u\n", block, idx, bit);
33
Jens Axboe10ba5352006-10-20 11:39:27 +020034 return (f->file_map[idx] & (1UL << bit)) == 0;
35}
36
37/*
Jens Axboedf415582006-10-20 11:41:03 +020038 * Mark a given offset as used in the map.
39 */
Jens Axboe9bf20612007-03-01 09:33:57 +010040static void mark_random_map(struct thread_data *td, struct io_u *io_u)
Jens Axboedf415582006-10-20 11:41:03 +020041{
Jens Axboe2dc1bbe2007-03-15 15:01:33 +010042 unsigned int min_bs = td->o.rw_min_bs;
Jens Axboe9bf20612007-03-01 09:33:57 +010043 struct fio_file *f = io_u->file;
Jens Axboea00735e2006-11-03 08:58:08 +010044 unsigned long long block;
45 unsigned int blocks;
Jens Axboec685b5b2007-02-10 20:02:28 +010046 unsigned int nr_blocks;
Jens Axboedf415582006-10-20 11:41:03 +020047
Jens Axboeb9c5b642008-02-18 14:13:08 +010048 block = (io_u->offset - f->file_offset) / (unsigned long long) min_bs;
Jens Axboea00735e2006-11-03 08:58:08 +010049 blocks = 0;
Jens Axboec685b5b2007-02-10 20:02:28 +010050 nr_blocks = (io_u->buflen + min_bs - 1) / min_bs;
51
52 while (blocks < nr_blocks) {
Jens Axboedf415582006-10-20 11:41:03 +020053 unsigned int idx, bit;
54
Jens Axboe1e3d53a2007-03-22 19:01:48 +010055 /*
56 * If we have a mixed random workload, we may
57 * encounter blocks we already did IO to.
58 */
Jens Axboeaec2de22008-04-24 12:44:42 +020059 if ((td->o.ddir_nr == 1) && !random_map_free(f, block))
Jens Axboedf415582006-10-20 11:41:03 +020060 break;
61
Jens Axboeaec2de22008-04-24 12:44:42 +020062 idx = RAND_MAP_IDX(f, block);
63 bit = RAND_MAP_BIT(f, block);
Jens Axboedf415582006-10-20 11:41:03 +020064
Jens Axboe0032bf92007-02-13 17:39:56 +010065 fio_assert(td, idx < f->num_maps);
Jens Axboedf415582006-10-20 11:41:03 +020066
67 f->file_map[idx] |= (1UL << bit);
68 block++;
69 blocks++;
70 }
71
Jens Axboea00735e2006-11-03 08:58:08 +010072 if ((blocks * min_bs) < io_u->buflen)
73 io_u->buflen = blocks * min_bs;
Jens Axboedf415582006-10-20 11:41:03 +020074}
75
Jens Axboe2ba1c292008-02-01 13:16:38 +010076static inline unsigned long long last_block(struct thread_data *td,
Jens Axboe4ba66132008-02-05 10:05:50 +010077 struct fio_file *f,
78 enum fio_ddir ddir)
Jens Axboe2ba1c292008-02-01 13:16:38 +010079{
80 unsigned long long max_blocks;
81
Jens Axboe84422ac2008-02-19 20:10:26 +010082 max_blocks = f->io_size / (unsigned long long) td->o.min_bs[ddir];
Jens Axboe2ba1c292008-02-01 13:16:38 +010083 if (!max_blocks)
84 return 0;
85
Jens Axboe67778e82008-05-15 09:20:08 +020086 return max_blocks;
Jens Axboe2ba1c292008-02-01 13:16:38 +010087}
88
Jens Axboedf415582006-10-20 11:41:03 +020089/*
Jens Axboe10ba5352006-10-20 11:39:27 +020090 * Return the next free block in the map.
91 */
92static int get_next_free_block(struct thread_data *td, struct fio_file *f,
Jens Axboe4ba66132008-02-05 10:05:50 +010093 enum fio_ddir ddir, unsigned long long *b)
Jens Axboe10ba5352006-10-20 11:39:27 +020094{
Jens Axboe84422ac2008-02-19 20:10:26 +010095 unsigned long long min_bs = td->o.rw_min_bs;
Jens Axboe10ba5352006-10-20 11:39:27 +020096 int i;
97
Jens Axboec685b5b2007-02-10 20:02:28 +010098 i = f->last_free_lookup;
99 *b = (i * BLOCKS_PER_MAP);
Jens Axboe84422ac2008-02-19 20:10:26 +0100100 while ((*b) * min_bs < f->real_file_size) {
Jens Axboe10ba5352006-10-20 11:39:27 +0200101 if (f->file_map[i] != -1UL) {
Jens Axboeb12ebc62007-05-21 11:42:33 +0200102 *b += fio_ffz(f->file_map[i]);
Jens Axboe4ba66132008-02-05 10:05:50 +0100103 if (*b > last_block(td, f, ddir))
Jens Axboe2ba1c292008-02-01 13:16:38 +0100104 break;
Jens Axboec685b5b2007-02-10 20:02:28 +0100105 f->last_free_lookup = i;
Jens Axboe10ba5352006-10-20 11:39:27 +0200106 return 0;
107 }
108
109 *b += BLOCKS_PER_MAP;
110 i++;
111 }
112
Jens Axboe2ba1c292008-02-01 13:16:38 +0100113 dprint(FD_IO, "failed finding a free block\n");
Jens Axboe10ba5352006-10-20 11:39:27 +0200114 return 1;
115}
116
Jens Axboeec4015d2007-03-23 08:04:27 +0100117static int get_next_rand_offset(struct thread_data *td, struct fio_file *f,
Jens Axboe4ba66132008-02-05 10:05:50 +0100118 enum fio_ddir ddir, unsigned long long *b)
Jens Axboeec4015d2007-03-23 08:04:27 +0100119{
Jens Axboe84422ac2008-02-19 20:10:26 +0100120 unsigned long long r;
Jens Axboeec4015d2007-03-23 08:04:27 +0100121 int loops = 5;
122
123 do {
124 r = os_random_long(&td->random_state);
Jens Axboe84422ac2008-02-19 20:10:26 +0100125 dprint(FD_RANDOM, "off rand %llu\n", r);
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100126 *b = (last_block(td, f, ddir) - 1)
127 * (r / ((unsigned long long) RAND_MAX + 1.0));
Jens Axboe2ba1c292008-02-01 13:16:38 +0100128
Jens Axboe43c63a72007-05-21 13:23:30 +0200129 /*
130 * if we are not maintaining a random map, we are done.
131 */
Jens Axboe303032a2008-03-26 10:11:10 +0100132 if (!file_randommap(td, f))
Jens Axboe43c63a72007-05-21 13:23:30 +0200133 return 0;
134
135 /*
Jens Axboe84422ac2008-02-19 20:10:26 +0100136 * calculate map offset and check if it's free
Jens Axboe43c63a72007-05-21 13:23:30 +0200137 */
Jens Axboeaec2de22008-04-24 12:44:42 +0200138 if (random_map_free(f, *b))
Jens Axboe43c63a72007-05-21 13:23:30 +0200139 return 0;
140
Jens Axboe84422ac2008-02-19 20:10:26 +0100141 dprint(FD_RANDOM, "get_next_rand_offset: offset %llu busy\n",
142 *b);
Jens Axboe43c63a72007-05-21 13:23:30 +0200143 } while (--loops);
Jens Axboeec4015d2007-03-23 08:04:27 +0100144
145 /*
Jens Axboe43c63a72007-05-21 13:23:30 +0200146 * we get here, if we didn't suceed in looking up a block. generate
147 * a random start offset into the filemap, and find the first free
148 * block from there.
Jens Axboeec4015d2007-03-23 08:04:27 +0100149 */
Jens Axboe43c63a72007-05-21 13:23:30 +0200150 loops = 10;
151 do {
152 f->last_free_lookup = (f->num_maps - 1) * (r / (RAND_MAX+1.0));
Jens Axboe4ba66132008-02-05 10:05:50 +0100153 if (!get_next_free_block(td, f, ddir, b))
Jens Axboe43c63a72007-05-21 13:23:30 +0200154 return 0;
Jens Axboeec4015d2007-03-23 08:04:27 +0100155
Jens Axboe43c63a72007-05-21 13:23:30 +0200156 r = os_random_long(&td->random_state);
157 } while (--loops);
158
159 /*
160 * that didn't work either, try exhaustive search from the start
161 */
162 f->last_free_lookup = 0;
Jens Axboe4ba66132008-02-05 10:05:50 +0100163 return get_next_free_block(td, f, ddir, b);
Jens Axboeec4015d2007-03-23 08:04:27 +0100164}
165
Jens Axboe10ba5352006-10-20 11:39:27 +0200166/*
167 * For random io, generate a random new block and see if it's used. Repeat
168 * until we find a free one. For sequential io, just return the end of
169 * the last io issued.
170 */
Jens Axboe9bf20612007-03-01 09:33:57 +0100171static int get_next_offset(struct thread_data *td, struct io_u *io_u)
Jens Axboe10ba5352006-10-20 11:39:27 +0200172{
Jens Axboe9bf20612007-03-01 09:33:57 +0100173 struct fio_file *f = io_u->file;
Jens Axboeec4015d2007-03-23 08:04:27 +0100174 unsigned long long b;
Jens Axboe4ba66132008-02-05 10:05:50 +0100175 enum fio_ddir ddir = io_u->ddir;
Jens Axboe10ba5352006-10-20 11:39:27 +0200176
Jens Axboeec4015d2007-03-23 08:04:27 +0100177 if (td_random(td) && (td->o.ddir_nr && !--td->ddir_nr)) {
178 td->ddir_nr = td->o.ddir_nr;
Jens Axboe10ba5352006-10-20 11:39:27 +0200179
Jens Axboe4ba66132008-02-05 10:05:50 +0100180 if (get_next_rand_offset(td, f, ddir, &b))
Jens Axboebca4ed42007-02-12 05:13:23 +0100181 return 1;
Jens Axboe43063a12007-03-27 20:21:24 +0200182 } else {
ljzhang,Yaxin Hu,Jianchao Tang0c3d7682007-07-26 10:59:25 +0200183 if (f->last_pos >= f->real_file_size) {
Jens Axboe4ba66132008-02-05 10:05:50 +0100184 if (!td_random(td) ||
185 get_next_rand_offset(td, f, ddir, &b))
ljzhang,Yaxin Hu,Jianchao Tang0c3d7682007-07-26 10:59:25 +0200186 return 1;
ljzhang,Yaxin Hu,Jianchao Tangbcdedd02007-07-27 13:28:26 +0200187 } else
Jens Axboe4ba66132008-02-05 10:05:50 +0100188 b = (f->last_pos - f->file_offset) / td->o.min_bs[ddir];
Jens Axboe43063a12007-03-27 20:21:24 +0200189 }
Jens Axboe10ba5352006-10-20 11:39:27 +0200190
Jens Axboe4ba66132008-02-05 10:05:50 +0100191 io_u->offset = (b * td->o.min_bs[ddir]) + f->file_offset;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100192 if (io_u->offset >= f->real_file_size) {
193 dprint(FD_IO, "get_next_offset: offset %llu >= size %llu\n",
194 io_u->offset, f->real_file_size);
Jens Axboe10ba5352006-10-20 11:39:27 +0200195 return 1;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100196 }
Jens Axboe10ba5352006-10-20 11:39:27 +0200197
198 return 0;
199}
200
Jens Axboe9bf20612007-03-01 09:33:57 +0100201static unsigned int get_next_buflen(struct thread_data *td, struct io_u *io_u)
Jens Axboe10ba5352006-10-20 11:39:27 +0200202{
Jens Axboebca4ed42007-02-12 05:13:23 +0100203 const int ddir = io_u->ddir;
Jens Axboe10ba5352006-10-20 11:39:27 +0200204 unsigned int buflen;
205 long r;
206
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100207 if (td->o.min_bs[ddir] == td->o.max_bs[ddir])
208 buflen = td->o.min_bs[ddir];
Jens Axboe10ba5352006-10-20 11:39:27 +0200209 else {
210 r = os_random_long(&td->bsrange_state);
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100211 if (!td->o.bssplit_nr) {
212 buflen = (unsigned int)
213 (1 + (double) (td->o.max_bs[ddir] - 1)
214 * r / (RAND_MAX + 1.0));
215 } else {
Jens Axboe564ca972007-12-14 12:21:19 +0100216 long perc = 0;
217 unsigned int i;
218
219 for (i = 0; i < td->o.bssplit_nr; i++) {
220 struct bssplit *bsp = &td->o.bssplit[i];
221
222 buflen = bsp->bs;
223 perc += bsp->perc;
224 if (r <= ((LONG_MAX / 100L) * perc))
225 break;
226 }
227 }
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100228 if (!td->o.bs_unaligned) {
229 buflen = (buflen + td->o.min_bs[ddir] - 1)
230 & ~(td->o.min_bs[ddir] - 1);
231 }
Jens Axboe10ba5352006-10-20 11:39:27 +0200232 }
233
Jens Axboe4ba66132008-02-05 10:05:50 +0100234 if (io_u->offset + buflen > io_u->file->real_file_size) {
235 dprint(FD_IO, "lower buflen %u -> %u (ddir=%d)\n", buflen,
236 td->o.min_bs[ddir], ddir);
Jens Axboe6a5e6882007-07-26 10:47:51 +0200237 buflen = td->o.min_bs[ddir];
Jens Axboe4ba66132008-02-05 10:05:50 +0100238 }
Jens Axboe6a5e6882007-07-26 10:47:51 +0200239
Jens Axboe10ba5352006-10-20 11:39:27 +0200240 return buflen;
241}
242
Jens Axboeafe24a52007-03-16 20:27:27 +0100243static void set_rwmix_bytes(struct thread_data *td)
244{
Jens Axboee4928662008-04-07 09:19:46 +0200245 unsigned long issues;
Jens Axboeafe24a52007-03-16 20:27:27 +0100246 unsigned int diff;
247
248 /*
249 * we do time or byte based switch. this is needed because
250 * buffered writes may issue a lot quicker than they complete,
251 * whereas reads do not.
252 */
Jens Axboee4928662008-04-07 09:19:46 +0200253 issues = td->io_issues[td->rwmix_ddir] - td->rwmix_issues;
Jens Axboee47f7992007-03-21 14:05:39 +0100254 diff = td->o.rwmix[td->rwmix_ddir ^ 1];
Jens Axboeafe24a52007-03-16 20:27:27 +0100255
Jens Axboee4928662008-04-07 09:19:46 +0200256 td->rwmix_issues = td->io_issues[td->rwmix_ddir]
257 + (issues * ((100 - diff)) / diff);
Jens Axboee47f7992007-03-21 14:05:39 +0100258}
259
260static inline enum fio_ddir get_rand_ddir(struct thread_data *td)
261{
262 unsigned int v;
263 long r;
264
265 r = os_random_long(&td->rwmix_state);
266 v = 1 + (int) (100.0 * (r / (RAND_MAX + 1.0)));
267 if (v < td->o.rwmix[DDIR_READ])
268 return DDIR_READ;
269
270 return DDIR_WRITE;
Jens Axboeafe24a52007-03-16 20:27:27 +0100271}
272
Jens Axboe10ba5352006-10-20 11:39:27 +0200273/*
274 * Return the data direction for the next io_u. If the job is a
275 * mixed read/write workload, check the rwmix cycle and switch if
276 * necessary.
277 */
Jens Axboe1e97cce2006-12-05 11:44:16 +0100278static enum fio_ddir get_rw_ddir(struct thread_data *td)
Jens Axboe10ba5352006-10-20 11:39:27 +0200279{
280 if (td_rw(td)) {
Jens Axboe10ba5352006-10-20 11:39:27 +0200281 /*
282 * Check if it's time to seed a new data direction.
283 */
Jens Axboee4928662008-04-07 09:19:46 +0200284 if (td->io_issues[td->rwmix_ddir] >= td->rwmix_issues) {
Jens Axboee47f7992007-03-21 14:05:39 +0100285 unsigned long long max_bytes;
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100286 enum fio_ddir ddir;
Jens Axboe10ba5352006-10-20 11:39:27 +0200287
Jens Axboee47f7992007-03-21 14:05:39 +0100288 /*
289 * Put a top limit on how many bytes we do for
290 * one data direction, to avoid overflowing the
291 * ranges too much
292 */
293 ddir = get_rand_ddir(td);
294 max_bytes = td->this_io_bytes[ddir];
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100295 if (max_bytes >=
296 (td->o.size * td->o.rwmix[ddir] / 100)) {
Jens Axboee4928662008-04-07 09:19:46 +0200297 if (!td->rw_end_set[ddir])
Jens Axboe38d77ca2007-03-21 14:20:20 +0100298 td->rw_end_set[ddir] = 1;
Jens Axboee4928662008-04-07 09:19:46 +0200299
Jens Axboee47f7992007-03-21 14:05:39 +0100300 ddir ^= 1;
Jens Axboe38d77ca2007-03-21 14:20:20 +0100301 }
Jens Axboee47f7992007-03-21 14:05:39 +0100302
303 if (ddir != td->rwmix_ddir)
304 set_rwmix_bytes(td);
305
306 td->rwmix_ddir = ddir;
Jens Axboe10ba5352006-10-20 11:39:27 +0200307 }
308 return td->rwmix_ddir;
309 } else if (td_read(td))
310 return DDIR_READ;
311 else
312 return DDIR_WRITE;
313}
314
Jens Axboe10ba5352006-10-20 11:39:27 +0200315void put_io_u(struct thread_data *td, struct io_u *io_u)
316{
Jens Axboe0c6e7512007-02-22 11:19:39 +0100317 assert((io_u->flags & IO_U_F_FREE) == 0);
318 io_u->flags |= IO_U_F_FREE;
319
Jens Axboe6977bcd2008-03-01 15:55:36 +0100320 if (io_u->file) {
321 int ret = put_file(td, io_u->file);
322
323 if (ret)
324 td_verror(td, ret, "file close");
325 }
Jens Axboe2dbdab72007-04-26 10:21:15 +0200326
Jens Axboe10ba5352006-10-20 11:39:27 +0200327 io_u->file = NULL;
328 list_del(&io_u->list);
329 list_add(&io_u->list, &td->io_u_freelist);
330 td->cur_depth--;
331}
332
Jens Axboe755200a2007-02-19 13:08:12 +0100333void requeue_io_u(struct thread_data *td, struct io_u **io_u)
334{
335 struct io_u *__io_u = *io_u;
336
Jens Axboe4d2e0f42007-02-24 13:31:57 +0100337 __io_u->flags |= IO_U_F_FREE;
Jens Axboee4f54ad2008-02-04 10:56:07 +0100338 if ((__io_u->flags & IO_U_F_FLIGHT) && (__io_u->ddir != DDIR_SYNC))
339 td->io_issues[__io_u->ddir]--;
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100340
Jens Axboe4d2e0f42007-02-24 13:31:57 +0100341 __io_u->flags &= ~IO_U_F_FLIGHT;
342
Jens Axboe755200a2007-02-19 13:08:12 +0100343 list_del(&__io_u->list);
344 list_add_tail(&__io_u->list, &td->io_u_requeues);
345 td->cur_depth--;
346 *io_u = NULL;
347}
348
Jens Axboe9bf20612007-03-01 09:33:57 +0100349static int fill_io_u(struct thread_data *td, struct io_u *io_u)
Jens Axboe10ba5352006-10-20 11:39:27 +0200350{
Jens Axboeb4c5e1a2007-10-25 18:44:45 +0200351 if (td->io_ops->flags & FIO_NOIO)
352 goto out;
353
Jens Axboe10ba5352006-10-20 11:39:27 +0200354 /*
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200355 * see if it's time to sync
356 */
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100357 if (td->o.fsync_blocks &&
358 !(td->io_issues[DDIR_WRITE] % td->o.fsync_blocks) &&
359 td->io_issues[DDIR_WRITE] && should_fsync(td)) {
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200360 io_u->ddir = DDIR_SYNC;
Jens Axboec38e9462007-03-27 08:48:48 +0200361 goto out;
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200362 }
363
Jens Axboea00735e2006-11-03 08:58:08 +0100364 io_u->ddir = get_rw_ddir(td);
365
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200366 /*
Jens Axboe48f5abd2007-07-20 13:25:04 +0200367 * See if it's time to switch to a new zone
368 */
369 if (td->zone_bytes >= td->o.zone_size) {
370 td->zone_bytes = 0;
371 io_u->file->last_pos += td->o.zone_skip;
372 td->io_skip_bytes += td->o.zone_skip;
373 }
374
375 /*
Jens Axboec685b5b2007-02-10 20:02:28 +0100376 * No log, let the seq/rand engine retrieve the next buflen and
377 * position.
Jens Axboe10ba5352006-10-20 11:39:27 +0200378 */
Jens Axboe2ba1c292008-02-01 13:16:38 +0100379 if (get_next_offset(td, io_u)) {
380 dprint(FD_IO, "io_u %p, failed getting offset\n", io_u);
Jens Axboebca4ed42007-02-12 05:13:23 +0100381 return 1;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100382 }
Jens Axboec685b5b2007-02-10 20:02:28 +0100383
Jens Axboe9bf20612007-03-01 09:33:57 +0100384 io_u->buflen = get_next_buflen(td, io_u);
Jens Axboe2ba1c292008-02-01 13:16:38 +0100385 if (!io_u->buflen) {
386 dprint(FD_IO, "io_u %p, failed getting buflen\n", io_u);
Jens Axboebca4ed42007-02-12 05:13:23 +0100387 return 1;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100388 }
Jens Axboe10ba5352006-10-20 11:39:27 +0200389
Jens Axboe2ba1c292008-02-01 13:16:38 +0100390 if (io_u->offset + io_u->buflen > io_u->file->real_file_size) {
391 dprint(FD_IO, "io_u %p, offset too large\n", io_u);
Jens Axboe4ba66132008-02-05 10:05:50 +0100392 dprint(FD_IO, " off=%llu/%lu > %llu\n", io_u->offset,
393 io_u->buflen, io_u->file->real_file_size);
Jens Axboe6a5e6882007-07-26 10:47:51 +0200394 return 1;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100395 }
Jens Axboe6a5e6882007-07-26 10:47:51 +0200396
Jens Axboebca4ed42007-02-12 05:13:23 +0100397 /*
398 * mark entry before potentially trimming io_u
399 */
Jens Axboe303032a2008-03-26 10:11:10 +0100400 if (td_random(td) && file_randommap(td, io_u->file))
Jens Axboe9bf20612007-03-01 09:33:57 +0100401 mark_random_map(td, io_u);
Jens Axboe10ba5352006-10-20 11:39:27 +0200402
Jens Axboebca4ed42007-02-12 05:13:23 +0100403 /*
404 * If using a write iolog, store this entry.
405 */
Jens Axboec38e9462007-03-27 08:48:48 +0200406out:
Jens Axboe2ba1c292008-02-01 13:16:38 +0100407 dprint_io_u(io_u, "fill_io_u");
Jens Axboed9d91e32008-01-31 13:25:42 +0100408 td->zone_bytes += io_u->buflen;
Jens Axboef29b25a2007-07-23 08:56:43 +0200409 log_io_u(td, io_u);
Jens Axboebca4ed42007-02-12 05:13:23 +0100410 return 0;
Jens Axboe10ba5352006-10-20 11:39:27 +0200411}
412
Jens Axboed8005752008-05-15 09:49:09 +0200413void io_u_mark_depth(struct thread_data *td, unsigned int nr)
Jens Axboe71619dc2007-01-13 23:56:33 +0100414{
415 int index = 0;
416
417 switch (td->cur_depth) {
418 default:
Jens Axboea783e612007-06-19 09:50:28 +0200419 index = 6;
420 break;
Jens Axboe71619dc2007-01-13 23:56:33 +0100421 case 32 ... 63:
Jens Axboea783e612007-06-19 09:50:28 +0200422 index = 5;
423 break;
Jens Axboe71619dc2007-01-13 23:56:33 +0100424 case 16 ... 31:
Jens Axboea783e612007-06-19 09:50:28 +0200425 index = 4;
426 break;
Jens Axboe71619dc2007-01-13 23:56:33 +0100427 case 8 ... 15:
Jens Axboea783e612007-06-19 09:50:28 +0200428 index = 3;
429 break;
Jens Axboe71619dc2007-01-13 23:56:33 +0100430 case 4 ... 7:
Jens Axboea783e612007-06-19 09:50:28 +0200431 index = 2;
432 break;
Jens Axboe71619dc2007-01-13 23:56:33 +0100433 case 2 ... 3:
Jens Axboea783e612007-06-19 09:50:28 +0200434 index = 1;
Jens Axboe71619dc2007-01-13 23:56:33 +0100435 case 1:
436 break;
437 }
438
Jens Axboe3bec7ae2008-05-14 21:08:37 +0200439 td->ts.io_u_map[index] += nr;
Jens Axboe71619dc2007-01-13 23:56:33 +0100440}
441
Jens Axboe04a0fea2007-06-19 12:48:41 +0200442static void io_u_mark_lat_usec(struct thread_data *td, unsigned long usec)
443{
444 int index = 0;
445
446 assert(usec < 1000);
447
448 switch (usec) {
449 case 750 ... 999:
450 index = 9;
451 break;
452 case 500 ... 749:
453 index = 8;
454 break;
455 case 250 ... 499:
456 index = 7;
457 break;
458 case 100 ... 249:
459 index = 6;
460 break;
461 case 50 ... 99:
462 index = 5;
463 break;
464 case 20 ... 49:
465 index = 4;
466 break;
467 case 10 ... 19:
468 index = 3;
469 break;
470 case 4 ... 9:
471 index = 2;
472 break;
473 case 2 ... 3:
474 index = 1;
475 case 0 ... 1:
476 break;
477 }
478
479 assert(index < FIO_IO_U_LAT_U_NR);
480 td->ts.io_u_lat_u[index]++;
481}
482
483static void io_u_mark_lat_msec(struct thread_data *td, unsigned long msec)
Jens Axboeec118302007-02-17 04:38:20 +0100484{
485 int index = 0;
486
487 switch (msec) {
488 default:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200489 index = 11;
490 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100491 case 1000 ... 1999:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200492 index = 10;
493 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100494 case 750 ... 999:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200495 index = 9;
496 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100497 case 500 ... 749:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200498 index = 8;
499 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100500 case 250 ... 499:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200501 index = 7;
502 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100503 case 100 ... 249:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200504 index = 6;
505 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100506 case 50 ... 99:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200507 index = 5;
508 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100509 case 20 ... 49:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200510 index = 4;
511 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100512 case 10 ... 19:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200513 index = 3;
514 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100515 case 4 ... 9:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200516 index = 2;
517 break;
Jens Axboeec118302007-02-17 04:38:20 +0100518 case 2 ... 3:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200519 index = 1;
Jens Axboeec118302007-02-17 04:38:20 +0100520 case 0 ... 1:
521 break;
522 }
523
Jens Axboe04a0fea2007-06-19 12:48:41 +0200524 assert(index < FIO_IO_U_LAT_M_NR);
525 td->ts.io_u_lat_m[index]++;
526}
527
528static void io_u_mark_latency(struct thread_data *td, unsigned long usec)
529{
530 if (usec < 1000)
531 io_u_mark_lat_usec(td, usec);
532 else
533 io_u_mark_lat_msec(td, usec / 1000);
Jens Axboeec118302007-02-17 04:38:20 +0100534}
535
Jens Axboe0aabe162007-02-23 08:45:55 +0100536/*
537 * Get next file to service by choosing one at random
538 */
Jens Axboe1c178182007-03-13 13:25:18 +0100539static struct fio_file *get_next_file_rand(struct thread_data *td, int goodf,
540 int badf)
Jens Axboe0aabe162007-02-23 08:45:55 +0100541{
Jens Axboe0aabe162007-02-23 08:45:55 +0100542 struct fio_file *f;
Jens Axboe1c178182007-03-13 13:25:18 +0100543 int fno;
Jens Axboe0aabe162007-02-23 08:45:55 +0100544
545 do {
Jens Axboe7c83c082007-03-01 10:04:15 +0100546 long r = os_random_long(&td->next_file_state);
547
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100548 fno = (unsigned int) ((double) td->o.nr_files
549 * (r / (RAND_MAX + 1.0)));
Jens Axboe126d65c2008-03-01 18:04:31 +0100550 f = td->files[fno];
Jens Axboe059e63c2007-03-27 20:34:47 +0200551 if (f->flags & FIO_FILE_DONE)
552 continue;
Jens Axboe1c178182007-03-13 13:25:18 +0100553
Jens Axboe2ba1c292008-02-01 13:16:38 +0100554 if ((!goodf || (f->flags & goodf)) && !(f->flags & badf)) {
555 dprint(FD_FILE, "get_next_file_rand: %p\n", f);
Jens Axboe0aabe162007-02-23 08:45:55 +0100556 return f;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100557 }
Jens Axboe0aabe162007-02-23 08:45:55 +0100558 } while (1);
559}
560
561/*
562 * Get next file to service by doing round robin between all available ones
563 */
Jens Axboe1c178182007-03-13 13:25:18 +0100564static struct fio_file *get_next_file_rr(struct thread_data *td, int goodf,
565 int badf)
Jens Axboe3d7c3912007-02-19 13:16:12 +0100566{
567 unsigned int old_next_file = td->next_file;
568 struct fio_file *f;
569
570 do {
Jens Axboe126d65c2008-03-01 18:04:31 +0100571 f = td->files[td->next_file];
Jens Axboe3d7c3912007-02-19 13:16:12 +0100572
573 td->next_file++;
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100574 if (td->next_file >= td->o.nr_files)
Jens Axboe3d7c3912007-02-19 13:16:12 +0100575 td->next_file = 0;
576
Jens Axboed5ed68e2007-03-28 09:33:43 +0200577 if (f->flags & FIO_FILE_DONE) {
578 f = NULL;
Jens Axboe059e63c2007-03-27 20:34:47 +0200579 continue;
Jens Axboed5ed68e2007-03-28 09:33:43 +0200580 }
Jens Axboe059e63c2007-03-27 20:34:47 +0200581
Jens Axboe1c178182007-03-13 13:25:18 +0100582 if ((!goodf || (f->flags & goodf)) && !(f->flags & badf))
Jens Axboe3d7c3912007-02-19 13:16:12 +0100583 break;
584
585 f = NULL;
586 } while (td->next_file != old_next_file);
587
Jens Axboe2ba1c292008-02-01 13:16:38 +0100588 dprint(FD_FILE, "get_next_file_rr: %p\n", f);
Jens Axboe3d7c3912007-02-19 13:16:12 +0100589 return f;
590}
591
Jens Axboebdb4e2e2007-03-01 09:54:57 +0100592static struct fio_file *get_next_file(struct thread_data *td)
593{
Jens Axboe1907dbc2007-03-12 11:44:28 +0100594 struct fio_file *f;
595
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100596 assert(td->o.nr_files <= td->files_index);
Jens Axboe1c178182007-03-13 13:25:18 +0100597
Jens Axboe2ba1c292008-02-01 13:16:38 +0100598 if (!td->nr_open_files || td->nr_done_files >= td->o.nr_files) {
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100599 dprint(FD_FILE, "get_next_file: nr_open=%d, nr_done=%d,"
600 " nr_files=%d\n", td->nr_open_files,
601 td->nr_done_files,
602 td->o.nr_files);
Jens Axboebdb4e2e2007-03-01 09:54:57 +0100603 return NULL;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100604 }
Jens Axboebdb4e2e2007-03-01 09:54:57 +0100605
Jens Axboe1907dbc2007-03-12 11:44:28 +0100606 f = td->file_service_file;
Jens Axboef11bd942007-03-13 10:16:34 +0100607 if (f && (f->flags & FIO_FILE_OPEN) && td->file_service_left--)
Jens Axboe2ba1c292008-02-01 13:16:38 +0100608 goto out;
Jens Axboe1907dbc2007-03-12 11:44:28 +0100609
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100610 if (td->o.file_service_type == FIO_FSERVICE_RR)
Jens Axboe1c178182007-03-13 13:25:18 +0100611 f = get_next_file_rr(td, FIO_FILE_OPEN, FIO_FILE_CLOSING);
Jens Axboebdb4e2e2007-03-01 09:54:57 +0100612 else
Jens Axboe1c178182007-03-13 13:25:18 +0100613 f = get_next_file_rand(td, FIO_FILE_OPEN, FIO_FILE_CLOSING);
Jens Axboe1907dbc2007-03-12 11:44:28 +0100614
615 td->file_service_file = f;
616 td->file_service_left = td->file_service_nr - 1;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100617out:
618 dprint(FD_FILE, "get_next_file: %p\n", f);
Jens Axboe1907dbc2007-03-12 11:44:28 +0100619 return f;
Jens Axboebdb4e2e2007-03-01 09:54:57 +0100620}
621
Jens Axboe1c178182007-03-13 13:25:18 +0100622static struct fio_file *find_next_new_file(struct thread_data *td)
623{
624 struct fio_file *f;
625
Jens Axboe1020a132007-03-29 11:15:06 +0200626 if (!td->nr_open_files || td->nr_done_files >= td->o.nr_files)
627 return NULL;
628
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100629 if (td->o.file_service_type == FIO_FSERVICE_RR)
Jens Axboe1c178182007-03-13 13:25:18 +0100630 f = get_next_file_rr(td, 0, FIO_FILE_OPEN);
631 else
632 f = get_next_file_rand(td, 0, FIO_FILE_OPEN);
633
634 return f;
635}
636
Jens Axboe429f6672007-07-23 10:38:43 +0200637static int set_io_u_file(struct thread_data *td, struct io_u *io_u)
638{
639 struct fio_file *f;
640
641 do {
642 f = get_next_file(td);
643 if (!f)
644 return 1;
645
646set_file:
647 io_u->file = f;
648 get_file(f);
649
650 if (!fill_io_u(td, io_u))
651 break;
652
653 /*
654 * td_io_close() does a put_file() as well, so no need to
655 * do that here.
656 */
657 io_u->file = NULL;
658 td_io_close_file(td, f);
659 f->flags |= FIO_FILE_DONE;
660 td->nr_done_files++;
661
662 /*
663 * probably not the right place to do this, but see
664 * if we need to open a new file
665 */
666 if (td->nr_open_files < td->o.open_files &&
667 td->o.open_files != td->o.nr_files) {
668 f = find_next_new_file(td);
669
670 if (!f || td_io_open_file(td, f))
671 return 1;
672
673 goto set_file;
674 }
675 } while (1);
676
677 return 0;
678}
679
680
Jens Axboe10ba5352006-10-20 11:39:27 +0200681struct io_u *__get_io_u(struct thread_data *td)
682{
683 struct io_u *io_u = NULL;
684
Jens Axboe755200a2007-02-19 13:08:12 +0100685 if (!list_empty(&td->io_u_requeues))
686 io_u = list_entry(td->io_u_requeues.next, struct io_u, list);
687 else if (!queue_full(td)) {
Jens Axboe10ba5352006-10-20 11:39:27 +0200688 io_u = list_entry(td->io_u_freelist.next, struct io_u, list);
689
Jens Axboe6040dab2006-10-24 19:38:15 +0200690 io_u->buflen = 0;
Jens Axboe10ba5352006-10-20 11:39:27 +0200691 io_u->resid = 0;
Jens Axboe755200a2007-02-19 13:08:12 +0100692 io_u->file = NULL;
Jens Axboed7762cf2007-02-23 12:34:57 +0100693 io_u->end_io = NULL;
Jens Axboe755200a2007-02-19 13:08:12 +0100694 }
695
696 if (io_u) {
Jens Axboe0c6e7512007-02-22 11:19:39 +0100697 assert(io_u->flags & IO_U_F_FREE);
698 io_u->flags &= ~IO_U_F_FREE;
699
Jens Axboe755200a2007-02-19 13:08:12 +0100700 io_u->error = 0;
Jens Axboe10ba5352006-10-20 11:39:27 +0200701 list_del(&io_u->list);
702 list_add(&io_u->list, &td->io_u_busylist);
703 td->cur_depth++;
704 }
705
706 return io_u;
707}
708
709/*
710 * Return an io_u to be processed. Gets a buflen and offset, sets direction,
711 * etc. The returned io_u is fully ready to be prepped and submitted.
712 */
Jens Axboe3d7c3912007-02-19 13:16:12 +0100713struct io_u *get_io_u(struct thread_data *td)
Jens Axboe10ba5352006-10-20 11:39:27 +0200714{
Jens Axboe3d7c3912007-02-19 13:16:12 +0100715 struct fio_file *f;
Jens Axboe10ba5352006-10-20 11:39:27 +0200716 struct io_u *io_u;
717
718 io_u = __get_io_u(td);
Jens Axboe2ba1c292008-02-01 13:16:38 +0100719 if (!io_u) {
720 dprint(FD_IO, "__get_io_u failed\n");
Jens Axboe10ba5352006-10-20 11:39:27 +0200721 return NULL;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100722 }
Jens Axboe10ba5352006-10-20 11:39:27 +0200723
Jens Axboe755200a2007-02-19 13:08:12 +0100724 /*
725 * from a requeue, io_u already setup
726 */
727 if (io_u->file)
Jens Axboe77f392b2007-02-19 20:13:09 +0100728 goto out;
Jens Axboe755200a2007-02-19 13:08:12 +0100729
Jens Axboe429f6672007-07-23 10:38:43 +0200730 /*
731 * If using an iolog, grab next piece if any available.
732 */
733 if (td->o.read_iolog_file) {
734 if (read_iolog_get(td, io_u))
735 goto err_put;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100736 } else if (set_io_u_file(td, io_u)) {
737 dprint(FD_IO, "io_u %p, setting file failed\n", io_u);
Jens Axboe429f6672007-07-23 10:38:43 +0200738 goto err_put;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100739 }
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100740
Jens Axboe429f6672007-07-23 10:38:43 +0200741 f = io_u->file;
742 assert(f->flags & FIO_FILE_OPEN);
Jens Axboe97af62c2007-05-22 11:12:13 +0200743
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200744 if (io_u->ddir != DDIR_SYNC) {
Jens Axboed0656a92008-02-01 18:33:23 +0100745 if (!io_u->buflen && !(td->io_ops->flags & FIO_NOIO)) {
Jens Axboe2ba1c292008-02-01 13:16:38 +0100746 dprint(FD_IO, "get_io_u: zero buflen on %p\n", io_u);
Jens Axboe429f6672007-07-23 10:38:43 +0200747 goto err_put;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100748 }
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200749
Jens Axboe36167d82007-02-18 05:41:31 +0100750 f->last_pos = io_u->offset + io_u->buflen;
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200751
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100752 if (td->o.verify != VERIFY_NONE)
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200753 populate_verify_io_u(td, io_u);
Jens Axboe10ba5352006-10-20 11:39:27 +0200754 }
755
Jens Axboe165faf12007-02-07 11:30:37 +0100756 /*
757 * Set io data pointers.
758 */
Jens Axboed460eb32007-04-16 21:39:33 +0200759 io_u->endpos = io_u->offset + io_u->buflen;
Jens Axboecec6b552007-02-06 20:15:38 +0100760 io_u->xfer_buf = io_u->buf;
761 io_u->xfer_buflen = io_u->buflen;
Jens Axboe6ac7a332008-03-01 15:22:32 +0100762out:
Jens Axboe429f6672007-07-23 10:38:43 +0200763 if (!td_io_prep(td, io_u)) {
764 fio_gettime(&io_u->start_time, NULL);
765 return io_u;
Jens Axboe36167d82007-02-18 05:41:31 +0100766 }
Jens Axboe429f6672007-07-23 10:38:43 +0200767err_put:
Jens Axboe2ba1c292008-02-01 13:16:38 +0100768 dprint(FD_IO, "get_io_u failed\n");
Jens Axboe429f6672007-07-23 10:38:43 +0200769 put_io_u(td, io_u);
770 return NULL;
Jens Axboe10ba5352006-10-20 11:39:27 +0200771}
772
Jens Axboe54517922007-03-05 10:06:06 +0100773void io_u_log_error(struct thread_data *td, struct io_u *io_u)
774{
775 const char *msg[] = { "read", "write", "sync" };
776
777 log_err("fio: io_u error");
778
779 if (io_u->file)
780 log_err(" on file %s", io_u->file->file_name);
781
782 log_err(": %s\n", strerror(io_u->error));
783
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100784 log_err(" %s offset=%llu, buflen=%lu\n", msg[io_u->ddir],
785 io_u->offset, io_u->xfer_buflen);
Jens Axboe54517922007-03-05 10:06:06 +0100786
787 if (!td->error)
788 td_verror(td, io_u->error, "io_u error");
789}
790
Jens Axboe97601022007-02-18 12:47:29 +0100791static void io_completed(struct thread_data *td, struct io_u *io_u,
792 struct io_completion_data *icd)
Jens Axboe10ba5352006-10-20 11:39:27 +0200793{
Jens Axboed85f5112007-06-18 09:41:23 +0200794 unsigned long usec;
Jens Axboe10ba5352006-10-20 11:39:27 +0200795
Jens Axboe2ba1c292008-02-01 13:16:38 +0100796 dprint_io_u(io_u, "io complete");
797
Jens Axboe0c6e7512007-02-22 11:19:39 +0100798 assert(io_u->flags & IO_U_F_FLIGHT);
799 io_u->flags &= ~IO_U_F_FLIGHT;
800
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200801 if (io_u->ddir == DDIR_SYNC) {
802 td->last_was_sync = 1;
803 return;
804 }
805
806 td->last_was_sync = 0;
807
Jens Axboe10ba5352006-10-20 11:39:27 +0200808 if (!io_u->error) {
809 unsigned int bytes = io_u->buflen - io_u->resid;
Jens Axboe1e97cce2006-12-05 11:44:16 +0100810 const enum fio_ddir idx = io_u->ddir;
Jens Axboe3af6ef32007-02-18 06:57:43 +0100811 int ret;
Jens Axboe10ba5352006-10-20 11:39:27 +0200812
813 td->io_blocks[idx]++;
814 td->io_bytes[idx] += bytes;
Jens Axboe10ba5352006-10-20 11:39:27 +0200815 td->this_io_bytes[idx] += bytes;
816
Jens Axboed85f5112007-06-18 09:41:23 +0200817 usec = utime_since(&io_u->issue_time, &icd->time);
Jens Axboe10ba5352006-10-20 11:39:27 +0200818
Jens Axboed85f5112007-06-18 09:41:23 +0200819 add_clat_sample(td, idx, usec);
Jens Axboe02bcaa82006-11-24 10:42:00 +0100820 add_bw_sample(td, idx, &icd->time);
Jens Axboe04a0fea2007-06-19 12:48:41 +0200821 io_u_mark_latency(td, usec);
Jens Axboe10ba5352006-10-20 11:39:27 +0200822
Jens Axboe660a1cb2007-04-17 15:37:47 +0200823 if (td_write(td) && idx == DDIR_WRITE &&
Shawn Lewis86705792007-11-21 09:35:41 +0100824 td->o.do_verify &&
Jens Axboe41128402007-03-27 08:32:48 +0200825 td->o.verify != VERIFY_NONE)
Jens Axboe10ba5352006-10-20 11:39:27 +0200826 log_io_piece(td, io_u);
827
828 icd->bytes_done[idx] += bytes;
Jens Axboe3af6ef32007-02-18 06:57:43 +0100829
Jens Axboed7762cf2007-02-23 12:34:57 +0100830 if (io_u->end_io) {
Jens Axboe36690c92007-03-26 10:23:34 +0200831 ret = io_u->end_io(td, io_u);
Jens Axboe3af6ef32007-02-18 06:57:43 +0100832 if (ret && !icd->error)
833 icd->error = ret;
834 }
Jens Axboe54517922007-03-05 10:06:06 +0100835 } else {
Jens Axboe10ba5352006-10-20 11:39:27 +0200836 icd->error = io_u->error;
Jens Axboe54517922007-03-05 10:06:06 +0100837 io_u_log_error(td, io_u);
838 }
Jens Axboe10ba5352006-10-20 11:39:27 +0200839}
840
Jens Axboed7762cf2007-02-23 12:34:57 +0100841static void init_icd(struct io_completion_data *icd, int nr)
Jens Axboe36167d82007-02-18 05:41:31 +0100842{
843 fio_gettime(&icd->time, NULL);
844
Jens Axboe3af6ef32007-02-18 06:57:43 +0100845 icd->nr = nr;
846
Jens Axboe36167d82007-02-18 05:41:31 +0100847 icd->error = 0;
848 icd->bytes_done[0] = icd->bytes_done[1] = 0;
849}
850
Jens Axboe97601022007-02-18 12:47:29 +0100851static void ios_completed(struct thread_data *td,
852 struct io_completion_data *icd)
Jens Axboe10ba5352006-10-20 11:39:27 +0200853{
854 struct io_u *io_u;
855 int i;
856
Jens Axboe10ba5352006-10-20 11:39:27 +0200857 for (i = 0; i < icd->nr; i++) {
858 io_u = td->io_ops->event(td, i);
859
860 io_completed(td, io_u, icd);
861 put_io_u(td, io_u);
862 }
863}
Jens Axboe97601022007-02-18 12:47:29 +0100864
Jens Axboee7e6cfb2007-02-20 10:58:34 +0100865/*
866 * Complete a single io_u for the sync engines.
867 */
Jens Axboed7762cf2007-02-23 12:34:57 +0100868long io_u_sync_complete(struct thread_data *td, struct io_u *io_u)
Jens Axboe97601022007-02-18 12:47:29 +0100869{
870 struct io_completion_data icd;
871
Jens Axboed7762cf2007-02-23 12:34:57 +0100872 init_icd(&icd, 1);
Jens Axboe97601022007-02-18 12:47:29 +0100873 io_completed(td, io_u, &icd);
874 put_io_u(td, io_u);
875
876 if (!icd.error)
877 return icd.bytes_done[0] + icd.bytes_done[1];
878
Jens Axboe37e974a2007-03-02 15:16:45 +0100879 td_verror(td, icd.error, "io_u_sync_complete");
Jens Axboe97601022007-02-18 12:47:29 +0100880 return -1;
881}
882
Jens Axboee7e6cfb2007-02-20 10:58:34 +0100883/*
884 * Called to complete min_events number of io for the async engines.
885 */
Jens Axboed7762cf2007-02-23 12:34:57 +0100886long io_u_queued_complete(struct thread_data *td, int min_events)
Jens Axboe97601022007-02-18 12:47:29 +0100887{
Jens Axboe97601022007-02-18 12:47:29 +0100888 struct io_completion_data icd;
Jens Axboe00de55e2007-02-20 10:45:57 +0100889 struct timespec *tvp = NULL;
Jens Axboe97601022007-02-18 12:47:29 +0100890 int ret;
Davide Libenzi4d06a332007-03-22 07:43:50 +0100891 struct timespec ts = { .tv_sec = 0, .tv_nsec = 0, };
Jens Axboe97601022007-02-18 12:47:29 +0100892
Jens Axboeb271fe62008-02-04 10:49:41 +0100893 dprint(FD_IO, "io_u_queued_completed: min=%d\n", min_events);
894
Davide Libenzi4d06a332007-03-22 07:43:50 +0100895 if (!min_events)
Jens Axboe00de55e2007-02-20 10:45:57 +0100896 tvp = &ts;
Jens Axboe97601022007-02-18 12:47:29 +0100897
Jens Axboe00de55e2007-02-20 10:45:57 +0100898 ret = td_io_getevents(td, min_events, td->cur_depth, tvp);
Jens Axboe97601022007-02-18 12:47:29 +0100899 if (ret < 0) {
Jens Axboee1161c32007-02-22 19:36:48 +0100900 td_verror(td, -ret, "td_io_getevents");
Jens Axboe97601022007-02-18 12:47:29 +0100901 return ret;
902 } else if (!ret)
903 return ret;
904
Jens Axboed7762cf2007-02-23 12:34:57 +0100905 init_icd(&icd, ret);
Jens Axboe97601022007-02-18 12:47:29 +0100906 ios_completed(td, &icd);
907 if (!icd.error)
908 return icd.bytes_done[0] + icd.bytes_done[1];
909
Jens Axboe37e974a2007-03-02 15:16:45 +0100910 td_verror(td, icd.error, "io_u_queued_complete");
Jens Axboe97601022007-02-18 12:47:29 +0100911 return -1;
912}
Jens Axboe7e77dd02007-02-20 10:57:34 +0100913
914/*
915 * Call when io_u is really queued, to update the submission latency.
916 */
917void io_u_queued(struct thread_data *td, struct io_u *io_u)
918{
919 unsigned long slat_time;
920
Jens Axboed85f5112007-06-18 09:41:23 +0200921 slat_time = utime_since(&io_u->start_time, &io_u->issue_time);
Jens Axboe7e77dd02007-02-20 10:57:34 +0100922 add_slat_sample(td, io_u->ddir, slat_time);
923}
Jens Axboe433afcb2007-02-22 10:39:01 +0100924
Jens Axboe55bc9722007-02-22 11:30:05 +0100925#ifdef FIO_USE_TIMEOUT
Jens Axboe433afcb2007-02-22 10:39:01 +0100926void io_u_set_timeout(struct thread_data *td)
927{
928 assert(td->cur_depth);
929
930 td->timer.it_interval.tv_sec = 0;
931 td->timer.it_interval.tv_usec = 0;
932 td->timer.it_value.tv_sec = IO_U_TIMEOUT + IO_U_TIMEOUT_INC;
933 td->timer.it_value.tv_usec = 0;
934 setitimer(ITIMER_REAL, &td->timer, NULL);
935 fio_gettime(&td->timeout_end, NULL);
936}
Jens Axboe5945b9b2007-02-22 20:33:01 +0100937
938static void io_u_dump(struct io_u *io_u)
939{
940 unsigned long t_start = mtime_since_now(&io_u->start_time);
941 unsigned long t_issue = mtime_since_now(&io_u->issue_time);
942
943 log_err("io_u=%p, t_start=%lu, t_issue=%lu\n", io_u, t_start, t_issue);
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100944 log_err(" buf=%p/%p, len=%lu/%lu, offset=%llu\n", io_u->buf,
945 io_u->xfer_buf, io_u->buflen,
946 io_u->xfer_buflen,
947 io_u->offset);
Jens Axboe5945b9b2007-02-22 20:33:01 +0100948 log_err(" ddir=%d, fname=%s\n", io_u->ddir, io_u->file->file_name);
949}
Jens Axboe55bc9722007-02-22 11:30:05 +0100950#else
951void io_u_set_timeout(struct thread_data fio_unused *td)
952{
953}
954#endif
Jens Axboe433afcb2007-02-22 10:39:01 +0100955
Jens Axboe55bc9722007-02-22 11:30:05 +0100956#ifdef FIO_USE_TIMEOUT
Jens Axboe433afcb2007-02-22 10:39:01 +0100957static void io_u_timeout_handler(int fio_unused sig)
958{
959 struct thread_data *td, *__td;
960 pid_t pid = getpid();
Jens Axboe5945b9b2007-02-22 20:33:01 +0100961 struct list_head *entry;
962 struct io_u *io_u;
Jens Axboe433afcb2007-02-22 10:39:01 +0100963 int i;
964
965 log_err("fio: io_u timeout\n");
966
967 /*
968 * TLS would be nice...
969 */
970 td = NULL;
971 for_each_td(__td, i) {
972 if (__td->pid == pid) {
973 td = __td;
974 break;
975 }
976 }
977
978 if (!td) {
979 log_err("fio: io_u timeout, can't find job\n");
980 exit(1);
981 }
982
983 if (!td->cur_depth) {
984 log_err("fio: timeout without pending work?\n");
985 return;
986 }
987
Jens Axboe15506d02007-03-15 15:04:43 +0100988 log_err("fio: io_u timeout: job=%s, pid=%d\n", td->o.name, td->pid);
Jens Axboe5945b9b2007-02-22 20:33:01 +0100989
990 list_for_each(entry, &td->io_u_busylist) {
991 io_u = list_entry(entry, struct io_u, list);
992
993 io_u_dump(io_u);
994 }
995
996 td_verror(td, ETIMEDOUT, "io_u timeout");
Jens Axboe433afcb2007-02-22 10:39:01 +0100997 exit(1);
998}
Jens Axboe55bc9722007-02-22 11:30:05 +0100999#endif
Jens Axboe433afcb2007-02-22 10:39:01 +01001000
1001void io_u_init_timeout(void)
1002{
Jens Axboe55bc9722007-02-22 11:30:05 +01001003#ifdef FIO_USE_TIMEOUT
Jens Axboe433afcb2007-02-22 10:39:01 +01001004 signal(SIGALRM, io_u_timeout_handler);
Jens Axboe55bc9722007-02-22 11:30:05 +01001005#endif
Jens Axboe433afcb2007-02-22 10:39:01 +01001006}