blob: 2e9dac0db6b4aa0adae673a3b21d26173e48da3e [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 Axboe5973caf2008-05-21 19:52:35 +02009#include "hash.h"
Jens Axboe4f5af7b2009-06-03 08:45:40 +020010#include "verify.h"
Jens Axboe10ba5352006-10-20 11:39:27 +020011
Jens Axboe97601022007-02-18 12:47:29 +010012struct io_completion_data {
13 int nr; /* input */
Jens Axboe97601022007-02-18 12:47:29 +010014
15 int error; /* output */
16 unsigned long bytes_done[2]; /* output */
17 struct timeval time; /* output */
18};
19
Jens Axboe10ba5352006-10-20 11:39:27 +020020/*
21 * The ->file_map[] contains a map of blocks we have or have not done io
22 * to yet. Used to make sure we cover the entire range in a fair fashion.
23 */
Jens Axboeaec2de22008-04-24 12:44:42 +020024static int random_map_free(struct fio_file *f, const unsigned long long block)
Jens Axboe10ba5352006-10-20 11:39:27 +020025{
Jens Axboeaec2de22008-04-24 12:44:42 +020026 unsigned int idx = RAND_MAP_IDX(f, block);
27 unsigned int bit = RAND_MAP_BIT(f, block);
Jens Axboe10ba5352006-10-20 11:39:27 +020028
Jens Axboe84422ac2008-02-19 20:10:26 +010029 dprint(FD_RANDOM, "free: b=%llu, idx=%u, bit=%u\n", block, idx, bit);
30
Jens Axboede605662008-05-31 00:21:12 +020031 return (f->file_map[idx] & (1 << bit)) == 0;
Jens Axboe10ba5352006-10-20 11:39:27 +020032}
33
34/*
Jens Axboedf415582006-10-20 11:41:03 +020035 * Mark a given offset as used in the map.
36 */
Jens Axboe9bf20612007-03-01 09:33:57 +010037static void mark_random_map(struct thread_data *td, struct io_u *io_u)
Jens Axboedf415582006-10-20 11:41:03 +020038{
Jens Axboe2dc1bbe2007-03-15 15:01:33 +010039 unsigned int min_bs = td->o.rw_min_bs;
Jens Axboe9bf20612007-03-01 09:33:57 +010040 struct fio_file *f = io_u->file;
Jens Axboea00735e2006-11-03 08:58:08 +010041 unsigned long long block;
Jens Axboe3e3357b2008-06-02 09:53:05 +020042 unsigned int blocks, nr_blocks;
Jens Axboedf415582006-10-20 11:41:03 +020043
Jens Axboeb9c5b642008-02-18 14:13:08 +010044 block = (io_u->offset - f->file_offset) / (unsigned long long) min_bs;
Jens Axboec685b5b2007-02-10 20:02:28 +010045 nr_blocks = (io_u->buflen + min_bs - 1) / min_bs;
Jens Axboe3e3357b2008-06-02 09:53:05 +020046 blocks = 0;
Jens Axboec685b5b2007-02-10 20:02:28 +010047
Jens Axboe3e3357b2008-06-02 09:53:05 +020048 while (nr_blocks) {
49 unsigned int this_blocks, mask;
Jens Axboedf415582006-10-20 11:41:03 +020050 unsigned int idx, bit;
51
Jens Axboe1e3d53a2007-03-22 19:01:48 +010052 /*
53 * If we have a mixed random workload, we may
54 * encounter blocks we already did IO to.
55 */
Jens Axboe8b113b42008-06-02 10:15:48 +020056 if ((td->o.ddir_nr == 1) && !random_map_free(f, block)) {
57 if (!blocks)
58 blocks = 1;
Jens Axboedf415582006-10-20 11:41:03 +020059 break;
Jens Axboe8b113b42008-06-02 10:15:48 +020060 }
Jens Axboedf415582006-10-20 11:41:03 +020061
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
Jens Axboe3e3357b2008-06-02 09:53:05 +020067 this_blocks = nr_blocks;
68 if (this_blocks + bit > BLOCKS_PER_MAP)
69 this_blocks = BLOCKS_PER_MAP - bit;
70
71 if (this_blocks == BLOCKS_PER_MAP)
72 mask = -1U;
73 else
74 mask = ((1U << this_blocks) - 1) << bit;
75
Jens Axboe3e3357b2008-06-02 09:53:05 +020076 f->file_map[idx] |= mask;
77 nr_blocks -= this_blocks;
78 blocks += this_blocks;
Jens Axboec9dd34b2008-06-02 10:09:43 +020079 block += this_blocks;
Jens Axboedf415582006-10-20 11:41:03 +020080 }
81
Jens Axboea00735e2006-11-03 08:58:08 +010082 if ((blocks * min_bs) < io_u->buflen)
83 io_u->buflen = blocks * min_bs;
Jens Axboedf415582006-10-20 11:41:03 +020084}
85
Jens Axboe3e3357b2008-06-02 09:53:05 +020086static unsigned long long last_block(struct thread_data *td, struct fio_file *f,
87 enum fio_ddir ddir)
Jens Axboe2ba1c292008-02-01 13:16:38 +010088{
89 unsigned long long max_blocks;
Jens Axboed9dd70f2008-05-23 12:37:23 +020090 unsigned long long max_size;
Jens Axboe2ba1c292008-02-01 13:16:38 +010091
Jens Axboed9dd70f2008-05-23 12:37:23 +020092 /*
93 * Hmm, should we make sure that ->io_size <= ->real_file_size?
94 */
95 max_size = f->io_size;
96 if (max_size > f->real_file_size)
97 max_size = f->real_file_size;
98
Jens Axboe2b7a01d2009-03-11 11:00:13 +010099 max_blocks = max_size / (unsigned long long) td->o.ba[ddir];
Jens Axboe2ba1c292008-02-01 13:16:38 +0100100 if (!max_blocks)
101 return 0;
102
Jens Axboe67778e82008-05-15 09:20:08 +0200103 return max_blocks;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100104}
105
Jens Axboedf415582006-10-20 11:41:03 +0200106/*
Jens Axboe10ba5352006-10-20 11:39:27 +0200107 * Return the next free block in the map.
108 */
109static int get_next_free_block(struct thread_data *td, struct fio_file *f,
Jens Axboe4ba66132008-02-05 10:05:50 +0100110 enum fio_ddir ddir, unsigned long long *b)
Jens Axboe10ba5352006-10-20 11:39:27 +0200111{
Jens Axboe84422ac2008-02-19 20:10:26 +0100112 unsigned long long min_bs = td->o.rw_min_bs;
Jens Axboe10ba5352006-10-20 11:39:27 +0200113 int i;
114
Jens Axboec685b5b2007-02-10 20:02:28 +0100115 i = f->last_free_lookup;
116 *b = (i * BLOCKS_PER_MAP);
Radha Ramachandran899c29f2009-04-22 08:17:42 +0200117 while ((*b) * min_bs < f->real_file_size &&
118 (*b) * min_bs < f->io_size) {
Jens Axboede605662008-05-31 00:21:12 +0200119 if (f->file_map[i] != (unsigned int) -1) {
Jens Axboe697a6062008-05-31 00:04:45 +0200120 *b += ffz(f->file_map[i]);
Jens Axboe4ba66132008-02-05 10:05:50 +0100121 if (*b > last_block(td, f, ddir))
Jens Axboe2ba1c292008-02-01 13:16:38 +0100122 break;
Jens Axboec685b5b2007-02-10 20:02:28 +0100123 f->last_free_lookup = i;
Jens Axboe10ba5352006-10-20 11:39:27 +0200124 return 0;
125 }
126
127 *b += BLOCKS_PER_MAP;
128 i++;
129 }
130
Jens Axboe2ba1c292008-02-01 13:16:38 +0100131 dprint(FD_IO, "failed finding a free block\n");
Jens Axboe10ba5352006-10-20 11:39:27 +0200132 return 1;
133}
134
Jens Axboeec4015d2007-03-23 08:04:27 +0100135static int get_next_rand_offset(struct thread_data *td, struct fio_file *f,
Jens Axboe4ba66132008-02-05 10:05:50 +0100136 enum fio_ddir ddir, unsigned long long *b)
Jens Axboeec4015d2007-03-23 08:04:27 +0100137{
Jens Axboe84422ac2008-02-19 20:10:26 +0100138 unsigned long long r;
Jens Axboeec4015d2007-03-23 08:04:27 +0100139 int loops = 5;
140
141 do {
142 r = os_random_long(&td->random_state);
Jens Axboe84422ac2008-02-19 20:10:26 +0100143 dprint(FD_RANDOM, "off rand %llu\n", r);
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100144 *b = (last_block(td, f, ddir) - 1)
Jens Axboedc873b62008-06-04 20:13:04 +0200145 * (r / ((unsigned long long) OS_RAND_MAX + 1.0));
Jens Axboe2ba1c292008-02-01 13:16:38 +0100146
Jens Axboe43c63a72007-05-21 13:23:30 +0200147 /*
148 * if we are not maintaining a random map, we are done.
149 */
Jens Axboe303032a2008-03-26 10:11:10 +0100150 if (!file_randommap(td, f))
Jens Axboe43c63a72007-05-21 13:23:30 +0200151 return 0;
152
153 /*
Jens Axboe84422ac2008-02-19 20:10:26 +0100154 * calculate map offset and check if it's free
Jens Axboe43c63a72007-05-21 13:23:30 +0200155 */
Jens Axboeaec2de22008-04-24 12:44:42 +0200156 if (random_map_free(f, *b))
Jens Axboe43c63a72007-05-21 13:23:30 +0200157 return 0;
158
Jens Axboe84422ac2008-02-19 20:10:26 +0100159 dprint(FD_RANDOM, "get_next_rand_offset: offset %llu busy\n",
160 *b);
Jens Axboe43c63a72007-05-21 13:23:30 +0200161 } while (--loops);
Jens Axboeec4015d2007-03-23 08:04:27 +0100162
163 /*
Jens Axboe43c63a72007-05-21 13:23:30 +0200164 * we get here, if we didn't suceed in looking up a block. generate
165 * a random start offset into the filemap, and find the first free
166 * block from there.
Jens Axboeec4015d2007-03-23 08:04:27 +0100167 */
Jens Axboe43c63a72007-05-21 13:23:30 +0200168 loops = 10;
169 do {
Jens Axboedc873b62008-06-04 20:13:04 +0200170 f->last_free_lookup = (f->num_maps - 1) *
171 (r / (OS_RAND_MAX + 1.0));
Jens Axboe4ba66132008-02-05 10:05:50 +0100172 if (!get_next_free_block(td, f, ddir, b))
Jens Axboe43c63a72007-05-21 13:23:30 +0200173 return 0;
Jens Axboeec4015d2007-03-23 08:04:27 +0100174
Jens Axboe43c63a72007-05-21 13:23:30 +0200175 r = os_random_long(&td->random_state);
176 } while (--loops);
177
178 /*
179 * that didn't work either, try exhaustive search from the start
180 */
181 f->last_free_lookup = 0;
Jens Axboe4ba66132008-02-05 10:05:50 +0100182 return get_next_free_block(td, f, ddir, b);
Jens Axboeec4015d2007-03-23 08:04:27 +0100183}
184
Jens Axboe10ba5352006-10-20 11:39:27 +0200185/*
186 * For random io, generate a random new block and see if it's used. Repeat
187 * until we find a free one. For sequential io, just return the end of
188 * the last io issued.
189 */
Jens Axboe9bf20612007-03-01 09:33:57 +0100190static int get_next_offset(struct thread_data *td, struct io_u *io_u)
Jens Axboe10ba5352006-10-20 11:39:27 +0200191{
Jens Axboe9bf20612007-03-01 09:33:57 +0100192 struct fio_file *f = io_u->file;
Jens Axboeec4015d2007-03-23 08:04:27 +0100193 unsigned long long b;
Jens Axboe4ba66132008-02-05 10:05:50 +0100194 enum fio_ddir ddir = io_u->ddir;
Jens Axboe10ba5352006-10-20 11:39:27 +0200195
Jens Axboeec4015d2007-03-23 08:04:27 +0100196 if (td_random(td) && (td->o.ddir_nr && !--td->ddir_nr)) {
197 td->ddir_nr = td->o.ddir_nr;
Jens Axboe10ba5352006-10-20 11:39:27 +0200198
Jens Axboe683023e2009-03-04 08:41:31 +0100199 if (get_next_rand_offset(td, f, ddir, &b)) {
200 dprint(FD_IO, "%s: getting rand offset failed\n",
201 f->file_name);
Jens Axboebca4ed42007-02-12 05:13:23 +0100202 return 1;
Jens Axboe683023e2009-03-04 08:41:31 +0100203 }
Jens Axboe43063a12007-03-27 20:21:24 +0200204 } else {
ljzhang,Yaxin Hu,Jianchao Tang0c3d7682007-07-26 10:59:25 +0200205 if (f->last_pos >= f->real_file_size) {
Jens Axboe4ba66132008-02-05 10:05:50 +0100206 if (!td_random(td) ||
Jens Axboe683023e2009-03-04 08:41:31 +0100207 get_next_rand_offset(td, f, ddir, &b)) {
208 dprint(FD_IO, "%s: pos %llu > size %llu\n",
209 f->file_name, f->last_pos,
210 f->real_file_size);
ljzhang,Yaxin Hu,Jianchao Tang0c3d7682007-07-26 10:59:25 +0200211 return 1;
Jens Axboe683023e2009-03-04 08:41:31 +0100212 }
ljzhang,Yaxin Hu,Jianchao Tangbcdedd02007-07-27 13:28:26 +0200213 } else
Jens Axboe4ba66132008-02-05 10:05:50 +0100214 b = (f->last_pos - f->file_offset) / td->o.min_bs[ddir];
Jens Axboe43063a12007-03-27 20:21:24 +0200215 }
Jens Axboe10ba5352006-10-20 11:39:27 +0200216
Jens Axboe2b7a01d2009-03-11 11:00:13 +0100217 io_u->offset = b * td->o.ba[ddir];
Jens Axboe009bd842008-05-15 10:19:46 +0200218 if (io_u->offset >= f->io_size) {
219 dprint(FD_IO, "get_next_offset: offset %llu >= io_size %llu\n",
220 io_u->offset, f->io_size);
221 return 1;
222 }
223
224 io_u->offset += f->file_offset;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100225 if (io_u->offset >= f->real_file_size) {
226 dprint(FD_IO, "get_next_offset: offset %llu >= size %llu\n",
227 io_u->offset, f->real_file_size);
Jens Axboe10ba5352006-10-20 11:39:27 +0200228 return 1;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100229 }
Jens Axboe10ba5352006-10-20 11:39:27 +0200230
231 return 0;
232}
233
Jens Axboe9bf20612007-03-01 09:33:57 +0100234static unsigned int get_next_buflen(struct thread_data *td, struct io_u *io_u)
Jens Axboe10ba5352006-10-20 11:39:27 +0200235{
Jens Axboebca4ed42007-02-12 05:13:23 +0100236 const int ddir = io_u->ddir;
Jens Axboef3f552b2008-06-13 10:27:05 +0200237 unsigned int uninitialized_var(buflen);
Jens Axboef3059de2008-06-11 15:37:32 +0200238 unsigned int minbs, maxbs;
Jens Axboe10ba5352006-10-20 11:39:27 +0200239 long r;
240
Jens Axboef3059de2008-06-11 15:37:32 +0200241 minbs = td->o.min_bs[ddir];
242 maxbs = td->o.max_bs[ddir];
243
244 if (minbs == maxbs)
245 buflen = minbs;
Jens Axboe10ba5352006-10-20 11:39:27 +0200246 else {
247 r = os_random_long(&td->bsrange_state);
Jens Axboe720e84a2009-04-21 08:29:55 +0200248 if (!td->o.bssplit_nr[ddir]) {
Jens Axboef3059de2008-06-11 15:37:32 +0200249 buflen = 1 + (unsigned int) ((double) maxbs *
250 (r / (OS_RAND_MAX + 1.0)));
251 if (buflen < minbs)
252 buflen = minbs;
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100253 } else {
Jens Axboe564ca972007-12-14 12:21:19 +0100254 long perc = 0;
255 unsigned int i;
256
Jens Axboe720e84a2009-04-21 08:29:55 +0200257 for (i = 0; i < td->o.bssplit_nr[ddir]; i++) {
258 struct bssplit *bsp = &td->o.bssplit[ddir][i];
Jens Axboe564ca972007-12-14 12:21:19 +0100259
260 buflen = bsp->bs;
261 perc += bsp->perc;
Jens Axboef3059de2008-06-11 15:37:32 +0200262 if (r <= ((OS_RAND_MAX / 100L) * perc))
Jens Axboe564ca972007-12-14 12:21:19 +0100263 break;
264 }
265 }
Jens Axboef3059de2008-06-11 15:37:32 +0200266 if (!td->o.bs_unaligned && is_power_of_2(minbs))
267 buflen = (buflen + minbs - 1) & ~(minbs - 1);
Jens Axboe10ba5352006-10-20 11:39:27 +0200268 }
269
Jens Axboe4ba66132008-02-05 10:05:50 +0100270 if (io_u->offset + buflen > io_u->file->real_file_size) {
271 dprint(FD_IO, "lower buflen %u -> %u (ddir=%d)\n", buflen,
Jens Axboef3059de2008-06-11 15:37:32 +0200272 minbs, ddir);
273 buflen = minbs;
Jens Axboe4ba66132008-02-05 10:05:50 +0100274 }
Jens Axboe6a5e6882007-07-26 10:47:51 +0200275
Jens Axboe10ba5352006-10-20 11:39:27 +0200276 return buflen;
277}
278
Jens Axboeafe24a52007-03-16 20:27:27 +0100279static void set_rwmix_bytes(struct thread_data *td)
280{
Jens Axboeafe24a52007-03-16 20:27:27 +0100281 unsigned int diff;
282
283 /*
284 * we do time or byte based switch. this is needed because
285 * buffered writes may issue a lot quicker than they complete,
286 * whereas reads do not.
287 */
Jens Axboee47f7992007-03-21 14:05:39 +0100288 diff = td->o.rwmix[td->rwmix_ddir ^ 1];
Jens Axboe04c540d2008-05-28 10:35:26 +0200289 td->rwmix_issues = (td->io_issues[td->rwmix_ddir] * diff) / 100;
Jens Axboee47f7992007-03-21 14:05:39 +0100290}
291
292static inline enum fio_ddir get_rand_ddir(struct thread_data *td)
293{
294 unsigned int v;
295 long r;
296
297 r = os_random_long(&td->rwmix_state);
Jens Axboedc873b62008-06-04 20:13:04 +0200298 v = 1 + (int) (100.0 * (r / (OS_RAND_MAX + 1.0)));
Jens Axboe04c540d2008-05-28 10:35:26 +0200299 if (v <= td->o.rwmix[DDIR_READ])
Jens Axboee47f7992007-03-21 14:05:39 +0100300 return DDIR_READ;
301
302 return DDIR_WRITE;
Jens Axboeafe24a52007-03-16 20:27:27 +0100303}
304
Jens Axboe581e7142009-06-09 12:47:16 +0200305static enum fio_ddir rate_ddir(struct thread_data *td, enum fio_ddir ddir)
306{
307 enum fio_ddir odir = ddir ^ 1;
308 struct timeval t;
309 long usec;
310
311 if (td->rate_pending_usleep[ddir] <= 0)
312 return ddir;
313
314 /*
315 * We have too much pending sleep in this direction. See if we
316 * should switch.
317 */
318 if (td_rw(td)) {
319 /*
320 * Other direction does not have too much pending, switch
321 */
322 if (td->rate_pending_usleep[odir] < 100000)
323 return odir;
324
325 /*
326 * Both directions have pending sleep. Sleep the minimum time
327 * and deduct from both.
328 */
329 if (td->rate_pending_usleep[ddir] <=
330 td->rate_pending_usleep[odir]) {
331 usec = td->rate_pending_usleep[ddir];
332 } else {
333 usec = td->rate_pending_usleep[odir];
334 ddir = odir;
335 }
336 } else
337 usec = td->rate_pending_usleep[ddir];
338
339 fio_gettime(&t, NULL);
340 usec_sleep(td, usec);
341 usec = utime_since_now(&t);
342
343 td->rate_pending_usleep[ddir] -= usec;
344
345 odir = ddir ^ 1;
346 if (td_rw(td) && __should_check_rate(td, odir))
347 td->rate_pending_usleep[odir] -= usec;
348
349 return ddir;
350}
351
Jens Axboe10ba5352006-10-20 11:39:27 +0200352/*
353 * Return the data direction for the next io_u. If the job is a
354 * mixed read/write workload, check the rwmix cycle and switch if
355 * necessary.
356 */
Jens Axboe1e97cce2006-12-05 11:44:16 +0100357static enum fio_ddir get_rw_ddir(struct thread_data *td)
Jens Axboe10ba5352006-10-20 11:39:27 +0200358{
Jens Axboe581e7142009-06-09 12:47:16 +0200359 enum fio_ddir ddir;
360
Jens Axboe5f9099e2009-06-16 22:40:26 +0200361 /*
362 * see if it's time to fsync
363 */
364 if (td->o.fsync_blocks &&
365 !(td->io_issues[DDIR_WRITE] % td->o.fsync_blocks) &&
366 td->io_issues[DDIR_WRITE] && should_fsync(td))
367 return DDIR_SYNC;
368
369 /*
370 * see if it's time to fdatasync
371 */
372 if (td->o.fdatasync_blocks &&
373 !(td->io_issues[DDIR_WRITE] % td->o.fdatasync_blocks) &&
374 td->io_issues[DDIR_WRITE] && should_fsync(td))
375 return DDIR_DATASYNC;
376
Jens Axboe10ba5352006-10-20 11:39:27 +0200377 if (td_rw(td)) {
Jens Axboe10ba5352006-10-20 11:39:27 +0200378 /*
379 * Check if it's time to seed a new data direction.
380 */
Jens Axboee4928662008-04-07 09:19:46 +0200381 if (td->io_issues[td->rwmix_ddir] >= td->rwmix_issues) {
Jens Axboee47f7992007-03-21 14:05:39 +0100382 /*
383 * Put a top limit on how many bytes we do for
384 * one data direction, to avoid overflowing the
385 * ranges too much
386 */
387 ddir = get_rand_ddir(td);
Jens Axboee47f7992007-03-21 14:05:39 +0100388
389 if (ddir != td->rwmix_ddir)
390 set_rwmix_bytes(td);
391
392 td->rwmix_ddir = ddir;
Jens Axboe10ba5352006-10-20 11:39:27 +0200393 }
Jens Axboe581e7142009-06-09 12:47:16 +0200394 ddir = td->rwmix_ddir;
Jens Axboe10ba5352006-10-20 11:39:27 +0200395 } else if (td_read(td))
Jens Axboe581e7142009-06-09 12:47:16 +0200396 ddir = DDIR_READ;
Jens Axboe10ba5352006-10-20 11:39:27 +0200397 else
Jens Axboe581e7142009-06-09 12:47:16 +0200398 ddir = DDIR_WRITE;
399
400 td->rwmix_ddir = rate_ddir(td, ddir);
401 return td->rwmix_ddir;
Jens Axboe10ba5352006-10-20 11:39:27 +0200402}
403
Jens Axboe60f2c652008-05-16 12:31:36 +0200404static void put_file_log(struct thread_data *td, struct fio_file *f)
405{
406 int ret = put_file(td, f);
407
408 if (ret)
409 td_verror(td, ret, "file close");
410}
411
Jens Axboe10ba5352006-10-20 11:39:27 +0200412void put_io_u(struct thread_data *td, struct io_u *io_u)
413{
Jens Axboe0c6e7512007-02-22 11:19:39 +0100414 assert((io_u->flags & IO_U_F_FREE) == 0);
415 io_u->flags |= IO_U_F_FREE;
416
Jens Axboe60f2c652008-05-16 12:31:36 +0200417 if (io_u->file)
418 put_file_log(td, io_u->file);
Jens Axboe2dbdab72007-04-26 10:21:15 +0200419
Jens Axboe10ba5352006-10-20 11:39:27 +0200420 io_u->file = NULL;
Jens Axboe01743ee2008-06-02 12:19:19 +0200421 flist_del(&io_u->list);
422 flist_add(&io_u->list, &td->io_u_freelist);
Jens Axboe10ba5352006-10-20 11:39:27 +0200423 td->cur_depth--;
424}
425
Radha Ramachandranf2bba182009-06-15 08:40:16 +0200426void clear_io_u(struct thread_data *td, struct io_u *io_u)
427{
428 io_u->flags &= ~IO_U_F_FLIGHT;
429 put_io_u(td, io_u);
430}
431
Jens Axboe755200a2007-02-19 13:08:12 +0100432void requeue_io_u(struct thread_data *td, struct io_u **io_u)
433{
434 struct io_u *__io_u = *io_u;
435
Jens Axboe465221b2008-05-30 22:07:49 +0200436 dprint(FD_IO, "requeue %p\n", __io_u);
437
Jens Axboe4d2e0f42007-02-24 13:31:57 +0100438 __io_u->flags |= IO_U_F_FREE;
Jens Axboe5f9099e2009-06-16 22:40:26 +0200439 if ((__io_u->flags & IO_U_F_FLIGHT) && !ddir_sync(__io_u->ddir))
Jens Axboee4f54ad2008-02-04 10:56:07 +0100440 td->io_issues[__io_u->ddir]--;
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100441
Jens Axboe4d2e0f42007-02-24 13:31:57 +0100442 __io_u->flags &= ~IO_U_F_FLIGHT;
443
Jens Axboe01743ee2008-06-02 12:19:19 +0200444 flist_del(&__io_u->list);
445 flist_add_tail(&__io_u->list, &td->io_u_requeues);
Jens Axboe755200a2007-02-19 13:08:12 +0100446 td->cur_depth--;
447 *io_u = NULL;
448}
449
Jens Axboe9bf20612007-03-01 09:33:57 +0100450static int fill_io_u(struct thread_data *td, struct io_u *io_u)
Jens Axboe10ba5352006-10-20 11:39:27 +0200451{
Jens Axboeb4c5e1a2007-10-25 18:44:45 +0200452 if (td->io_ops->flags & FIO_NOIO)
453 goto out;
454
Jens Axboea00735e2006-11-03 08:58:08 +0100455 io_u->ddir = get_rw_ddir(td);
456
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200457 /*
Jens Axboe5f9099e2009-06-16 22:40:26 +0200458 * fsync() or fdatasync(), we are done
459 */
460 if (ddir_sync(io_u->ddir))
461 goto out;
462
463 /*
Jens Axboe48f5abd2007-07-20 13:25:04 +0200464 * See if it's time to switch to a new zone
465 */
466 if (td->zone_bytes >= td->o.zone_size) {
467 td->zone_bytes = 0;
468 io_u->file->last_pos += td->o.zone_skip;
469 td->io_skip_bytes += td->o.zone_skip;
470 }
471
472 /*
Jens Axboec685b5b2007-02-10 20:02:28 +0100473 * No log, let the seq/rand engine retrieve the next buflen and
474 * position.
Jens Axboe10ba5352006-10-20 11:39:27 +0200475 */
Jens Axboe2ba1c292008-02-01 13:16:38 +0100476 if (get_next_offset(td, io_u)) {
477 dprint(FD_IO, "io_u %p, failed getting offset\n", io_u);
Jens Axboebca4ed42007-02-12 05:13:23 +0100478 return 1;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100479 }
Jens Axboec685b5b2007-02-10 20:02:28 +0100480
Jens Axboe9bf20612007-03-01 09:33:57 +0100481 io_u->buflen = get_next_buflen(td, io_u);
Jens Axboe2ba1c292008-02-01 13:16:38 +0100482 if (!io_u->buflen) {
483 dprint(FD_IO, "io_u %p, failed getting buflen\n", io_u);
Jens Axboebca4ed42007-02-12 05:13:23 +0100484 return 1;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100485 }
Jens Axboe10ba5352006-10-20 11:39:27 +0200486
Jens Axboe2ba1c292008-02-01 13:16:38 +0100487 if (io_u->offset + io_u->buflen > io_u->file->real_file_size) {
488 dprint(FD_IO, "io_u %p, offset too large\n", io_u);
Jens Axboe4ba66132008-02-05 10:05:50 +0100489 dprint(FD_IO, " off=%llu/%lu > %llu\n", io_u->offset,
490 io_u->buflen, io_u->file->real_file_size);
Jens Axboe6a5e6882007-07-26 10:47:51 +0200491 return 1;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100492 }
Jens Axboe6a5e6882007-07-26 10:47:51 +0200493
Jens Axboebca4ed42007-02-12 05:13:23 +0100494 /*
495 * mark entry before potentially trimming io_u
496 */
Jens Axboe303032a2008-03-26 10:11:10 +0100497 if (td_random(td) && file_randommap(td, io_u->file))
Jens Axboe9bf20612007-03-01 09:33:57 +0100498 mark_random_map(td, io_u);
Jens Axboe10ba5352006-10-20 11:39:27 +0200499
Jens Axboebca4ed42007-02-12 05:13:23 +0100500 /*
501 * If using a write iolog, store this entry.
502 */
Jens Axboec38e9462007-03-27 08:48:48 +0200503out:
Jens Axboe2ba1c292008-02-01 13:16:38 +0100504 dprint_io_u(io_u, "fill_io_u");
Jens Axboed9d91e32008-01-31 13:25:42 +0100505 td->zone_bytes += io_u->buflen;
Jens Axboef29b25a2007-07-23 08:56:43 +0200506 log_io_u(td, io_u);
Jens Axboebca4ed42007-02-12 05:13:23 +0100507 return 0;
Jens Axboe10ba5352006-10-20 11:39:27 +0200508}
509
Jens Axboe838bc702008-05-22 13:08:23 +0200510static void __io_u_mark_map(unsigned int *map, unsigned int nr)
511{
512 int index = 0;
513
514 switch (nr) {
515 default:
516 index = 6;
517 break;
518 case 33 ... 64:
519 index = 5;
520 break;
521 case 17 ... 32:
522 index = 4;
523 break;
524 case 9 ... 16:
525 index = 3;
526 break;
527 case 5 ... 8:
528 index = 2;
529 break;
530 case 1 ... 4:
531 index = 1;
532 case 0:
533 break;
534 }
535
536 map[index]++;
537}
538
539void io_u_mark_submit(struct thread_data *td, unsigned int nr)
540{
541 __io_u_mark_map(td->ts.io_u_submit, nr);
542 td->ts.total_submit++;
543}
544
545void io_u_mark_complete(struct thread_data *td, unsigned int nr)
546{
547 __io_u_mark_map(td->ts.io_u_complete, nr);
548 td->ts.total_complete++;
549}
550
Jens Axboed8005752008-05-15 09:49:09 +0200551void io_u_mark_depth(struct thread_data *td, unsigned int nr)
Jens Axboe71619dc2007-01-13 23:56:33 +0100552{
553 int index = 0;
554
555 switch (td->cur_depth) {
556 default:
Jens Axboea783e612007-06-19 09:50:28 +0200557 index = 6;
558 break;
Jens Axboe71619dc2007-01-13 23:56:33 +0100559 case 32 ... 63:
Jens Axboea783e612007-06-19 09:50:28 +0200560 index = 5;
561 break;
Jens Axboe71619dc2007-01-13 23:56:33 +0100562 case 16 ... 31:
Jens Axboea783e612007-06-19 09:50:28 +0200563 index = 4;
564 break;
Jens Axboe71619dc2007-01-13 23:56:33 +0100565 case 8 ... 15:
Jens Axboea783e612007-06-19 09:50:28 +0200566 index = 3;
567 break;
Jens Axboe71619dc2007-01-13 23:56:33 +0100568 case 4 ... 7:
Jens Axboea783e612007-06-19 09:50:28 +0200569 index = 2;
570 break;
Jens Axboe71619dc2007-01-13 23:56:33 +0100571 case 2 ... 3:
Jens Axboea783e612007-06-19 09:50:28 +0200572 index = 1;
Jens Axboe71619dc2007-01-13 23:56:33 +0100573 case 1:
574 break;
575 }
576
Jens Axboe3bec7ae2008-05-14 21:08:37 +0200577 td->ts.io_u_map[index] += nr;
Jens Axboe71619dc2007-01-13 23:56:33 +0100578}
579
Jens Axboe04a0fea2007-06-19 12:48:41 +0200580static void io_u_mark_lat_usec(struct thread_data *td, unsigned long usec)
581{
582 int index = 0;
583
584 assert(usec < 1000);
585
586 switch (usec) {
587 case 750 ... 999:
588 index = 9;
589 break;
590 case 500 ... 749:
591 index = 8;
592 break;
593 case 250 ... 499:
594 index = 7;
595 break;
596 case 100 ... 249:
597 index = 6;
598 break;
599 case 50 ... 99:
600 index = 5;
601 break;
602 case 20 ... 49:
603 index = 4;
604 break;
605 case 10 ... 19:
606 index = 3;
607 break;
608 case 4 ... 9:
609 index = 2;
610 break;
611 case 2 ... 3:
612 index = 1;
613 case 0 ... 1:
614 break;
615 }
616
617 assert(index < FIO_IO_U_LAT_U_NR);
618 td->ts.io_u_lat_u[index]++;
619}
620
621static void io_u_mark_lat_msec(struct thread_data *td, unsigned long msec)
Jens Axboeec118302007-02-17 04:38:20 +0100622{
623 int index = 0;
624
625 switch (msec) {
626 default:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200627 index = 11;
628 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100629 case 1000 ... 1999:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200630 index = 10;
631 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100632 case 750 ... 999:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200633 index = 9;
634 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100635 case 500 ... 749:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200636 index = 8;
637 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100638 case 250 ... 499:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200639 index = 7;
640 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100641 case 100 ... 249:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200642 index = 6;
643 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100644 case 50 ... 99:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200645 index = 5;
646 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100647 case 20 ... 49:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200648 index = 4;
649 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100650 case 10 ... 19:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200651 index = 3;
652 break;
Jens Axboe8abdce62007-02-21 10:22:55 +0100653 case 4 ... 9:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200654 index = 2;
655 break;
Jens Axboeec118302007-02-17 04:38:20 +0100656 case 2 ... 3:
Jens Axboe04a0fea2007-06-19 12:48:41 +0200657 index = 1;
Jens Axboeec118302007-02-17 04:38:20 +0100658 case 0 ... 1:
659 break;
660 }
661
Jens Axboe04a0fea2007-06-19 12:48:41 +0200662 assert(index < FIO_IO_U_LAT_M_NR);
663 td->ts.io_u_lat_m[index]++;
664}
665
666static void io_u_mark_latency(struct thread_data *td, unsigned long usec)
667{
668 if (usec < 1000)
669 io_u_mark_lat_usec(td, usec);
670 else
671 io_u_mark_lat_msec(td, usec / 1000);
Jens Axboeec118302007-02-17 04:38:20 +0100672}
673
Jens Axboe0aabe162007-02-23 08:45:55 +0100674/*
675 * Get next file to service by choosing one at random
676 */
Jens Axboe2cc52932009-06-09 14:14:20 +0200677static struct fio_file *get_next_file_rand(struct thread_data *td,
678 enum fio_file_flags goodf,
Jens Axboed6aed792009-06-03 08:41:15 +0200679 enum fio_file_flags badf)
Jens Axboe0aabe162007-02-23 08:45:55 +0100680{
Jens Axboe0aabe162007-02-23 08:45:55 +0100681 struct fio_file *f;
Jens Axboe1c178182007-03-13 13:25:18 +0100682 int fno;
Jens Axboe0aabe162007-02-23 08:45:55 +0100683
684 do {
Jens Axboe7c83c082007-03-01 10:04:15 +0100685 long r = os_random_long(&td->next_file_state);
Jens Axboe87b10672009-03-04 09:39:47 +0100686 int opened = 0;
Jens Axboe7c83c082007-03-01 10:04:15 +0100687
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100688 fno = (unsigned int) ((double) td->o.nr_files
Jens Axboedc873b62008-06-04 20:13:04 +0200689 * (r / (OS_RAND_MAX + 1.0)));
Jens Axboe126d65c2008-03-01 18:04:31 +0100690 f = td->files[fno];
Jens Axboed6aed792009-06-03 08:41:15 +0200691 if (fio_file_done(f))
Jens Axboe059e63c2007-03-27 20:34:47 +0200692 continue;
Jens Axboe1c178182007-03-13 13:25:18 +0100693
Jens Axboed6aed792009-06-03 08:41:15 +0200694 if (!fio_file_open(f)) {
Jens Axboe87b10672009-03-04 09:39:47 +0100695 int err;
696
697 err = td_io_open_file(td, f);
698 if (err)
699 continue;
700 opened = 1;
701 }
702
Jens Axboe2ba1c292008-02-01 13:16:38 +0100703 if ((!goodf || (f->flags & goodf)) && !(f->flags & badf)) {
704 dprint(FD_FILE, "get_next_file_rand: %p\n", f);
Jens Axboe0aabe162007-02-23 08:45:55 +0100705 return f;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100706 }
Jens Axboe87b10672009-03-04 09:39:47 +0100707 if (opened)
708 td_io_close_file(td, f);
Jens Axboe0aabe162007-02-23 08:45:55 +0100709 } while (1);
710}
711
712/*
713 * Get next file to service by doing round robin between all available ones
714 */
Jens Axboe1c178182007-03-13 13:25:18 +0100715static struct fio_file *get_next_file_rr(struct thread_data *td, int goodf,
716 int badf)
Jens Axboe3d7c3912007-02-19 13:16:12 +0100717{
718 unsigned int old_next_file = td->next_file;
719 struct fio_file *f;
720
721 do {
Jens Axboe87b10672009-03-04 09:39:47 +0100722 int opened = 0;
723
Jens Axboe126d65c2008-03-01 18:04:31 +0100724 f = td->files[td->next_file];
Jens Axboe3d7c3912007-02-19 13:16:12 +0100725
726 td->next_file++;
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100727 if (td->next_file >= td->o.nr_files)
Jens Axboe3d7c3912007-02-19 13:16:12 +0100728 td->next_file = 0;
729
Jens Axboe87b10672009-03-04 09:39:47 +0100730 dprint(FD_FILE, "trying file %s %x\n", f->file_name, f->flags);
Jens Axboed6aed792009-06-03 08:41:15 +0200731 if (fio_file_done(f)) {
Jens Axboed5ed68e2007-03-28 09:33:43 +0200732 f = NULL;
Jens Axboe059e63c2007-03-27 20:34:47 +0200733 continue;
Jens Axboed5ed68e2007-03-28 09:33:43 +0200734 }
Jens Axboe059e63c2007-03-27 20:34:47 +0200735
Jens Axboed6aed792009-06-03 08:41:15 +0200736 if (!fio_file_open(f)) {
Jens Axboe87b10672009-03-04 09:39:47 +0100737 int err;
738
739 err = td_io_open_file(td, f);
Jens Axboeb5696bf2009-03-04 16:03:49 +0100740 if (err) {
741 dprint(FD_FILE, "error %d on open of %s\n",
742 err, f->file_name);
Jens Axboe87c27b42009-05-20 10:45:12 +0200743 f = NULL;
Jens Axboe87b10672009-03-04 09:39:47 +0100744 continue;
Jens Axboeb5696bf2009-03-04 16:03:49 +0100745 }
Jens Axboe87b10672009-03-04 09:39:47 +0100746 opened = 1;
747 }
748
Jens Axboeb5696bf2009-03-04 16:03:49 +0100749 dprint(FD_FILE, "goodf=%x, badf=%x, ff=%x\n", goodf, badf, f->flags);
Jens Axboe1c178182007-03-13 13:25:18 +0100750 if ((!goodf || (f->flags & goodf)) && !(f->flags & badf))
Jens Axboe3d7c3912007-02-19 13:16:12 +0100751 break;
752
Jens Axboe87b10672009-03-04 09:39:47 +0100753 if (opened)
754 td_io_close_file(td, f);
755
Jens Axboe3d7c3912007-02-19 13:16:12 +0100756 f = NULL;
757 } while (td->next_file != old_next_file);
758
Jens Axboe2ba1c292008-02-01 13:16:38 +0100759 dprint(FD_FILE, "get_next_file_rr: %p\n", f);
Jens Axboe3d7c3912007-02-19 13:16:12 +0100760 return f;
761}
762
Jens Axboebdb4e2e2007-03-01 09:54:57 +0100763static struct fio_file *get_next_file(struct thread_data *td)
764{
Jens Axboe1907dbc2007-03-12 11:44:28 +0100765 struct fio_file *f;
766
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100767 assert(td->o.nr_files <= td->files_index);
Jens Axboe1c178182007-03-13 13:25:18 +0100768
Jens Axboeb5696bf2009-03-04 16:03:49 +0100769 if (td->nr_done_files >= td->o.nr_files) {
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100770 dprint(FD_FILE, "get_next_file: nr_open=%d, nr_done=%d,"
771 " nr_files=%d\n", td->nr_open_files,
772 td->nr_done_files,
773 td->o.nr_files);
Jens Axboebdb4e2e2007-03-01 09:54:57 +0100774 return NULL;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100775 }
Jens Axboebdb4e2e2007-03-01 09:54:57 +0100776
Jens Axboe1907dbc2007-03-12 11:44:28 +0100777 f = td->file_service_file;
Jens Axboed6aed792009-06-03 08:41:15 +0200778 if (f && fio_file_open(f) && !fio_file_closing(f)) {
Jens Axboea086c252009-03-04 08:27:37 +0100779 if (td->o.file_service_type == FIO_FSERVICE_SEQ)
780 goto out;
781 if (td->file_service_left--)
782 goto out;
783 }
Jens Axboe1907dbc2007-03-12 11:44:28 +0100784
Jens Axboea086c252009-03-04 08:27:37 +0100785 if (td->o.file_service_type == FIO_FSERVICE_RR ||
786 td->o.file_service_type == FIO_FSERVICE_SEQ)
Jens Axboed6aed792009-06-03 08:41:15 +0200787 f = get_next_file_rr(td, FIO_FILE_open, FIO_FILE_closing);
Jens Axboebdb4e2e2007-03-01 09:54:57 +0100788 else
Jens Axboed6aed792009-06-03 08:41:15 +0200789 f = get_next_file_rand(td, FIO_FILE_open, FIO_FILE_closing);
Jens Axboe1907dbc2007-03-12 11:44:28 +0100790
791 td->file_service_file = f;
792 td->file_service_left = td->file_service_nr - 1;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100793out:
Jens Axboe683023e2009-03-04 08:41:31 +0100794 dprint(FD_FILE, "get_next_file: %p [%s]\n", f, f->file_name);
Jens Axboe1907dbc2007-03-12 11:44:28 +0100795 return f;
Jens Axboebdb4e2e2007-03-01 09:54:57 +0100796}
797
Jens Axboe429f6672007-07-23 10:38:43 +0200798static int set_io_u_file(struct thread_data *td, struct io_u *io_u)
799{
800 struct fio_file *f;
801
802 do {
803 f = get_next_file(td);
804 if (!f)
805 return 1;
806
Jens Axboe429f6672007-07-23 10:38:43 +0200807 io_u->file = f;
808 get_file(f);
809
810 if (!fill_io_u(td, io_u))
811 break;
812
Jens Axboeb5696bf2009-03-04 16:03:49 +0100813 put_file_log(td, f);
Jens Axboe429f6672007-07-23 10:38:43 +0200814 td_io_close_file(td, f);
Jens Axboeb5696bf2009-03-04 16:03:49 +0100815 io_u->file = NULL;
Jens Axboed6aed792009-06-03 08:41:15 +0200816 fio_file_set_done(f);
Jens Axboe429f6672007-07-23 10:38:43 +0200817 td->nr_done_files++;
Jens Axboe87b10672009-03-04 09:39:47 +0100818 dprint(FD_FILE, "%s: is done (%d of %d)\n", f->file_name, td->nr_done_files, td->o.nr_files);
Jens Axboe429f6672007-07-23 10:38:43 +0200819 } while (1);
820
821 return 0;
822}
823
824
Jens Axboe10ba5352006-10-20 11:39:27 +0200825struct io_u *__get_io_u(struct thread_data *td)
826{
827 struct io_u *io_u = NULL;
828
Jens Axboe01743ee2008-06-02 12:19:19 +0200829 if (!flist_empty(&td->io_u_requeues))
830 io_u = flist_entry(td->io_u_requeues.next, struct io_u, list);
Jens Axboe755200a2007-02-19 13:08:12 +0100831 else if (!queue_full(td)) {
Jens Axboe01743ee2008-06-02 12:19:19 +0200832 io_u = flist_entry(td->io_u_freelist.next, struct io_u, list);
Jens Axboe10ba5352006-10-20 11:39:27 +0200833
Jens Axboe6040dab2006-10-24 19:38:15 +0200834 io_u->buflen = 0;
Jens Axboe10ba5352006-10-20 11:39:27 +0200835 io_u->resid = 0;
Jens Axboe755200a2007-02-19 13:08:12 +0100836 io_u->file = NULL;
Jens Axboed7762cf2007-02-23 12:34:57 +0100837 io_u->end_io = NULL;
Jens Axboe755200a2007-02-19 13:08:12 +0100838 }
839
840 if (io_u) {
Jens Axboe0c6e7512007-02-22 11:19:39 +0100841 assert(io_u->flags & IO_U_F_FREE);
842 io_u->flags &= ~IO_U_F_FREE;
843
Jens Axboe755200a2007-02-19 13:08:12 +0100844 io_u->error = 0;
Jens Axboe01743ee2008-06-02 12:19:19 +0200845 flist_del(&io_u->list);
846 flist_add(&io_u->list, &td->io_u_busylist);
Jens Axboe10ba5352006-10-20 11:39:27 +0200847 td->cur_depth++;
848 }
849
850 return io_u;
851}
852
853/*
854 * Return an io_u to be processed. Gets a buflen and offset, sets direction,
855 * etc. The returned io_u is fully ready to be prepped and submitted.
856 */
Jens Axboe3d7c3912007-02-19 13:16:12 +0100857struct io_u *get_io_u(struct thread_data *td)
Jens Axboe10ba5352006-10-20 11:39:27 +0200858{
Jens Axboe3d7c3912007-02-19 13:16:12 +0100859 struct fio_file *f;
Jens Axboe10ba5352006-10-20 11:39:27 +0200860 struct io_u *io_u;
861
862 io_u = __get_io_u(td);
Jens Axboe2ba1c292008-02-01 13:16:38 +0100863 if (!io_u) {
864 dprint(FD_IO, "__get_io_u failed\n");
Jens Axboe10ba5352006-10-20 11:39:27 +0200865 return NULL;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100866 }
Jens Axboe10ba5352006-10-20 11:39:27 +0200867
Jens Axboe755200a2007-02-19 13:08:12 +0100868 /*
869 * from a requeue, io_u already setup
870 */
871 if (io_u->file)
Jens Axboe77f392b2007-02-19 20:13:09 +0100872 goto out;
Jens Axboe755200a2007-02-19 13:08:12 +0100873
Jens Axboe429f6672007-07-23 10:38:43 +0200874 /*
875 * If using an iolog, grab next piece if any available.
876 */
877 if (td->o.read_iolog_file) {
878 if (read_iolog_get(td, io_u))
879 goto err_put;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100880 } else if (set_io_u_file(td, io_u)) {
881 dprint(FD_IO, "io_u %p, setting file failed\n", io_u);
Jens Axboe429f6672007-07-23 10:38:43 +0200882 goto err_put;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100883 }
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100884
Jens Axboe429f6672007-07-23 10:38:43 +0200885 f = io_u->file;
Jens Axboed6aed792009-06-03 08:41:15 +0200886 assert(fio_file_open(f));
Jens Axboe97af62c2007-05-22 11:12:13 +0200887
Jens Axboe5f9099e2009-06-16 22:40:26 +0200888 if (!ddir_sync(io_u->ddir)) {
Jens Axboed0656a92008-02-01 18:33:23 +0100889 if (!io_u->buflen && !(td->io_ops->flags & FIO_NOIO)) {
Jens Axboe2ba1c292008-02-01 13:16:38 +0100890 dprint(FD_IO, "get_io_u: zero buflen on %p\n", io_u);
Jens Axboe429f6672007-07-23 10:38:43 +0200891 goto err_put;
Jens Axboe2ba1c292008-02-01 13:16:38 +0100892 }
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200893
Jens Axboe36167d82007-02-18 05:41:31 +0100894 f->last_pos = io_u->offset + io_u->buflen;
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200895
Radha Ramachandranc311cd22009-04-28 08:12:56 +0200896 if (td->o.verify != VERIFY_NONE && io_u->ddir == DDIR_WRITE)
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200897 populate_verify_io_u(td, io_u);
Jens Axboee4dad9c2008-05-28 10:53:44 +0200898 else if (td->o.refill_buffers && io_u->ddir == DDIR_WRITE)
899 io_u_fill_buffer(td, io_u, io_u->xfer_buflen);
Jens Axboe10ba5352006-10-20 11:39:27 +0200900 }
901
Jens Axboe165faf12007-02-07 11:30:37 +0100902 /*
903 * Set io data pointers.
904 */
Jens Axboecec6b552007-02-06 20:15:38 +0100905 io_u->xfer_buf = io_u->buf;
906 io_u->xfer_buflen = io_u->buflen;
Jens Axboe5973caf2008-05-21 19:52:35 +0200907
Jens Axboe6ac7a332008-03-01 15:22:32 +0100908out:
Jens Axboe429f6672007-07-23 10:38:43 +0200909 if (!td_io_prep(td, io_u)) {
Jens Axboe993bf482008-11-14 13:04:53 +0100910 if (!td->o.disable_slat)
911 fio_gettime(&io_u->start_time, NULL);
Jens Axboe429f6672007-07-23 10:38:43 +0200912 return io_u;
Jens Axboe36167d82007-02-18 05:41:31 +0100913 }
Jens Axboe429f6672007-07-23 10:38:43 +0200914err_put:
Jens Axboe2ba1c292008-02-01 13:16:38 +0100915 dprint(FD_IO, "get_io_u failed\n");
Jens Axboe429f6672007-07-23 10:38:43 +0200916 put_io_u(td, io_u);
917 return NULL;
Jens Axboe10ba5352006-10-20 11:39:27 +0200918}
919
Jens Axboe54517922007-03-05 10:06:06 +0100920void io_u_log_error(struct thread_data *td, struct io_u *io_u)
921{
922 const char *msg[] = { "read", "write", "sync" };
923
924 log_err("fio: io_u error");
925
926 if (io_u->file)
927 log_err(" on file %s", io_u->file->file_name);
928
929 log_err(": %s\n", strerror(io_u->error));
930
Jens Axboe5ec10ea2008-03-06 15:42:00 +0100931 log_err(" %s offset=%llu, buflen=%lu\n", msg[io_u->ddir],
932 io_u->offset, io_u->xfer_buflen);
Jens Axboe54517922007-03-05 10:06:06 +0100933
934 if (!td->error)
935 td_verror(td, io_u->error, "io_u error");
936}
937
Jens Axboe97601022007-02-18 12:47:29 +0100938static void io_completed(struct thread_data *td, struct io_u *io_u,
939 struct io_completion_data *icd)
Jens Axboe10ba5352006-10-20 11:39:27 +0200940{
Jens Axboedbad30b2008-12-05 13:17:16 +0100941 /*
942 * Older gcc's are too dumb to realize that usec is always used
943 * initialized, silence that warning.
944 */
945 unsigned long uninitialized_var(usec);
Jens Axboe10ba5352006-10-20 11:39:27 +0200946
Jens Axboe2ba1c292008-02-01 13:16:38 +0100947 dprint_io_u(io_u, "io complete");
948
Jens Axboe0c6e7512007-02-22 11:19:39 +0100949 assert(io_u->flags & IO_U_F_FLIGHT);
950 io_u->flags &= ~IO_U_F_FLIGHT;
951
Jens Axboe5f9099e2009-06-16 22:40:26 +0200952 if (ddir_sync(io_u->ddir)) {
Jens Axboe87dc1ab2006-10-24 14:41:26 +0200953 td->last_was_sync = 1;
954 return;
955 }
956
957 td->last_was_sync = 0;
958
Jens Axboe10ba5352006-10-20 11:39:27 +0200959 if (!io_u->error) {
960 unsigned int bytes = io_u->buflen - io_u->resid;
Jens Axboe1e97cce2006-12-05 11:44:16 +0100961 const enum fio_ddir idx = io_u->ddir;
Jens Axboeb29ee5b2008-09-11 10:17:26 +0200962 int ret;
Jens Axboe10ba5352006-10-20 11:39:27 +0200963
Jens Axboeb29ee5b2008-09-11 10:17:26 +0200964 td->io_blocks[idx]++;
965 td->io_bytes[idx] += bytes;
966 td->this_io_bytes[idx] += bytes;
Jens Axboe10ba5352006-10-20 11:39:27 +0200967
Jens Axboeb29ee5b2008-09-11 10:17:26 +0200968 if (ramp_time_over(td)) {
Jens Axboe40e1a6f2009-06-11 10:55:39 +0200969 unsigned long uninitialized_var(lusec);
970 unsigned long uninitialized_var(rusec);
971
972 if (!td->o.disable_clat || !td->o.disable_bw)
973 lusec = utime_since(&io_u->issue_time,
974 &icd->time);
975 if (__should_check_rate(td, idx) ||
976 __should_check_rate(td, idx ^ 1))
977 rusec = utime_since(&io_u->start_time,
Jens Axboe9520ebb2008-10-16 21:03:27 +0200978 &icd->time);
Jens Axboe721938a2008-09-10 09:46:16 +0200979
Jens Axboe9520ebb2008-10-16 21:03:27 +0200980 if (!td->o.disable_clat) {
Jens Axboe306ddc92009-05-18 13:08:12 +0200981 add_clat_sample(td, idx, usec, bytes);
Jens Axboe40e1a6f2009-06-11 10:55:39 +0200982 io_u_mark_latency(td, lusec);
Jens Axboe9520ebb2008-10-16 21:03:27 +0200983 }
984 if (!td->o.disable_bw)
Jens Axboe306ddc92009-05-18 13:08:12 +0200985 add_bw_sample(td, idx, bytes, &icd->time);
Jens Axboeb23b6a22009-06-11 22:06:23 +0200986 if (__should_check_rate(td, idx)) {
987 td->rate_pending_usleep[idx] +=
988 (long) td->rate_usec_cycle[idx] - rusec;
989 }
Jens Axboe581e7142009-06-09 12:47:16 +0200990 if (__should_check_rate(td, idx ^ 1))
Jens Axboeb23b6a22009-06-11 22:06:23 +0200991 td->rate_pending_usleep[idx ^ 1] -= rusec;
Jens Axboe721938a2008-09-10 09:46:16 +0200992 }
Jens Axboe10ba5352006-10-20 11:39:27 +0200993
Jens Axboe660a1cb2007-04-17 15:37:47 +0200994 if (td_write(td) && idx == DDIR_WRITE &&
Shawn Lewis86705792007-11-21 09:35:41 +0100995 td->o.do_verify &&
Jens Axboe41128402007-03-27 08:32:48 +0200996 td->o.verify != VERIFY_NONE)
Jens Axboe10ba5352006-10-20 11:39:27 +0200997 log_io_piece(td, io_u);
998
Jens Axboeb29ee5b2008-09-11 10:17:26 +0200999 icd->bytes_done[idx] += bytes;
Jens Axboe3af6ef32007-02-18 06:57:43 +01001000
Jens Axboed7762cf2007-02-23 12:34:57 +01001001 if (io_u->end_io) {
Jens Axboe36690c92007-03-26 10:23:34 +02001002 ret = io_u->end_io(td, io_u);
Jens Axboe3af6ef32007-02-18 06:57:43 +01001003 if (ret && !icd->error)
1004 icd->error = ret;
1005 }
Jens Axboe54517922007-03-05 10:06:06 +01001006 } else {
Jens Axboe10ba5352006-10-20 11:39:27 +02001007 icd->error = io_u->error;
Jens Axboe54517922007-03-05 10:06:06 +01001008 io_u_log_error(td, io_u);
1009 }
Radha Ramachandranf2bba182009-06-15 08:40:16 +02001010 if (td->o.continue_on_error && icd->error &&
1011 td_non_fatal_error(icd->error)) {
1012 /*
1013 * If there is a non_fatal error, then add to the error count
1014 * and clear all the errors.
1015 */
1016 update_error_count(td, icd->error);
1017 td_clear_error(td);
1018 icd->error = 0;
1019 io_u->error = 0;
1020 }
Jens Axboe10ba5352006-10-20 11:39:27 +02001021}
1022
Jens Axboe9520ebb2008-10-16 21:03:27 +02001023static void init_icd(struct thread_data *td, struct io_completion_data *icd,
1024 int nr)
Jens Axboe36167d82007-02-18 05:41:31 +01001025{
Jens Axboe9520ebb2008-10-16 21:03:27 +02001026 if (!td->o.disable_clat || !td->o.disable_bw)
1027 fio_gettime(&icd->time, NULL);
Jens Axboe36167d82007-02-18 05:41:31 +01001028
Jens Axboe3af6ef32007-02-18 06:57:43 +01001029 icd->nr = nr;
1030
Jens Axboe36167d82007-02-18 05:41:31 +01001031 icd->error = 0;
1032 icd->bytes_done[0] = icd->bytes_done[1] = 0;
1033}
1034
Jens Axboe97601022007-02-18 12:47:29 +01001035static void ios_completed(struct thread_data *td,
1036 struct io_completion_data *icd)
Jens Axboe10ba5352006-10-20 11:39:27 +02001037{
1038 struct io_u *io_u;
1039 int i;
1040
Jens Axboe10ba5352006-10-20 11:39:27 +02001041 for (i = 0; i < icd->nr; i++) {
1042 io_u = td->io_ops->event(td, i);
1043
1044 io_completed(td, io_u, icd);
1045 put_io_u(td, io_u);
1046 }
1047}
Jens Axboe97601022007-02-18 12:47:29 +01001048
Jens Axboee7e6cfb2007-02-20 10:58:34 +01001049/*
1050 * Complete a single io_u for the sync engines.
1051 */
Jens Axboe581e7142009-06-09 12:47:16 +02001052int io_u_sync_complete(struct thread_data *td, struct io_u *io_u,
1053 unsigned long *bytes)
Jens Axboe97601022007-02-18 12:47:29 +01001054{
1055 struct io_completion_data icd;
1056
Jens Axboe9520ebb2008-10-16 21:03:27 +02001057 init_icd(td, &icd, 1);
Jens Axboe97601022007-02-18 12:47:29 +01001058 io_completed(td, io_u, &icd);
1059 put_io_u(td, io_u);
1060
Jens Axboe581e7142009-06-09 12:47:16 +02001061 if (icd.error) {
1062 td_verror(td, icd.error, "io_u_sync_complete");
1063 return -1;
1064 }
Jens Axboe97601022007-02-18 12:47:29 +01001065
Jens Axboe581e7142009-06-09 12:47:16 +02001066 if (bytes) {
1067 bytes[0] += icd.bytes_done[0];
1068 bytes[1] += icd.bytes_done[1];
1069 }
1070
1071 return 0;
Jens Axboe97601022007-02-18 12:47:29 +01001072}
1073
Jens Axboee7e6cfb2007-02-20 10:58:34 +01001074/*
1075 * Called to complete min_events number of io for the async engines.
1076 */
Jens Axboe581e7142009-06-09 12:47:16 +02001077int io_u_queued_complete(struct thread_data *td, int min_evts,
1078 unsigned long *bytes)
Jens Axboe97601022007-02-18 12:47:29 +01001079{
Jens Axboe97601022007-02-18 12:47:29 +01001080 struct io_completion_data icd;
Jens Axboe00de55e2007-02-20 10:45:57 +01001081 struct timespec *tvp = NULL;
Jens Axboe97601022007-02-18 12:47:29 +01001082 int ret;
Davide Libenzi4d06a332007-03-22 07:43:50 +01001083 struct timespec ts = { .tv_sec = 0, .tv_nsec = 0, };
Jens Axboe97601022007-02-18 12:47:29 +01001084
Jens Axboe49504212008-06-05 09:03:30 +02001085 dprint(FD_IO, "io_u_queued_completed: min=%d\n", min_evts);
Jens Axboeb271fe62008-02-04 10:49:41 +01001086
Jens Axboe49504212008-06-05 09:03:30 +02001087 if (!min_evts)
Jens Axboe00de55e2007-02-20 10:45:57 +01001088 tvp = &ts;
Jens Axboe97601022007-02-18 12:47:29 +01001089
Jens Axboe49504212008-06-05 09:03:30 +02001090 ret = td_io_getevents(td, min_evts, td->o.iodepth_batch_complete, tvp);
Jens Axboe97601022007-02-18 12:47:29 +01001091 if (ret < 0) {
Jens Axboee1161c32007-02-22 19:36:48 +01001092 td_verror(td, -ret, "td_io_getevents");
Jens Axboe97601022007-02-18 12:47:29 +01001093 return ret;
1094 } else if (!ret)
1095 return ret;
1096
Jens Axboe9520ebb2008-10-16 21:03:27 +02001097 init_icd(td, &icd, ret);
Jens Axboe97601022007-02-18 12:47:29 +01001098 ios_completed(td, &icd);
Jens Axboe581e7142009-06-09 12:47:16 +02001099 if (icd.error) {
1100 td_verror(td, icd.error, "io_u_queued_complete");
1101 return -1;
1102 }
Jens Axboe97601022007-02-18 12:47:29 +01001103
Jens Axboe581e7142009-06-09 12:47:16 +02001104 if (bytes) {
1105 bytes[0] += icd.bytes_done[0];
1106 bytes[1] += icd.bytes_done[1];
1107 }
1108
1109 return 0;
Jens Axboe97601022007-02-18 12:47:29 +01001110}
Jens Axboe7e77dd02007-02-20 10:57:34 +01001111
1112/*
1113 * Call when io_u is really queued, to update the submission latency.
1114 */
1115void io_u_queued(struct thread_data *td, struct io_u *io_u)
1116{
Jens Axboe9520ebb2008-10-16 21:03:27 +02001117 if (!td->o.disable_slat) {
1118 unsigned long slat_time;
Jens Axboe7e77dd02007-02-20 10:57:34 +01001119
Jens Axboe9520ebb2008-10-16 21:03:27 +02001120 slat_time = utime_since(&io_u->start_time, &io_u->issue_time);
Jens Axboe29a90dd2009-06-10 06:57:47 +02001121 add_slat_sample(td, io_u->ddir, slat_time, io_u->xfer_buflen);
Jens Axboe9520ebb2008-10-16 21:03:27 +02001122 }
Jens Axboe7e77dd02007-02-20 10:57:34 +01001123}
Jens Axboe433afcb2007-02-22 10:39:01 +01001124
Jens Axboe5973caf2008-05-21 19:52:35 +02001125/*
1126 * "randomly" fill the buffer contents
1127 */
1128void io_u_fill_buffer(struct thread_data *td, struct io_u *io_u,
1129 unsigned int max_bs)
1130{
1131 long *ptr = io_u->buf;
1132
1133 if (!td->o.zero_buffers) {
1134 while ((void *) ptr - io_u->buf < max_bs) {
1135 *ptr = rand() * GOLDEN_RATIO_PRIME;
1136 ptr++;
1137 }
1138 } else
1139 memset(ptr, 0, max_bs);
1140}