Commit | Line | Data |
---|---|---|
10ba535a JA |
1 | #include <unistd.h> |
2 | #include <fcntl.h> | |
3 | #include <string.h> | |
4 | #include <signal.h> | |
5 | #include <time.h> | |
0c6e7517 | 6 | #include <assert.h> |
10ba535a JA |
7 | |
8 | #include "fio.h" | |
5973cafb | 9 | #include "hash.h" |
4f5af7b2 | 10 | #include "verify.h" |
0d29de83 | 11 | #include "trim.h" |
1fbbf72e | 12 | #include "lib/rand.h" |
7ebd796f | 13 | #include "lib/axmap.h" |
002fe734 | 14 | #include "err.h" |
0f38bbef | 15 | #include "lib/pow2.h" |
4b157ac6 | 16 | #include "minmax.h" |
10ba535a | 17 | |
97601024 JA |
18 | struct io_completion_data { |
19 | int nr; /* input */ | |
97601024 JA |
20 | |
21 | int error; /* output */ | |
100f49f1 | 22 | uint64_t bytes_done[DDIR_RWDIR_CNT]; /* output */ |
97601024 JA |
23 | struct timeval time; /* output */ |
24 | }; | |
25 | ||
10ba535a | 26 | /* |
7ebd796f | 27 | * The ->io_axmap contains a map of blocks we have or have not done io |
10ba535a JA |
28 | * to yet. Used to make sure we cover the entire range in a fair fashion. |
29 | */ | |
e39c0676 | 30 | static bool random_map_free(struct fio_file *f, const uint64_t block) |
10ba535a | 31 | { |
7ebd796f | 32 | return !axmap_isset(f->io_axmap, block); |
10ba535a JA |
33 | } |
34 | ||
df415585 JA |
35 | /* |
36 | * Mark a given offset as used in the map. | |
37 | */ | |
9bf2061e | 38 | static void mark_random_map(struct thread_data *td, struct io_u *io_u) |
df415585 | 39 | { |
2dc1bbeb | 40 | unsigned int min_bs = td->o.rw_min_bs; |
9bf2061e | 41 | struct fio_file *f = io_u->file; |
51ede0b1 | 42 | unsigned int nr_blocks; |
1ae83d45 | 43 | uint64_t block; |
df415585 | 44 | |
1ae83d45 | 45 | block = (io_u->offset - f->file_offset) / (uint64_t) min_bs; |
c685b5b2 | 46 | nr_blocks = (io_u->buflen + min_bs - 1) / min_bs; |
df415585 | 47 | |
2ab9e98b | 48 | if (!(io_u->flags & IO_U_F_BUSY_OK)) |
7ebd796f | 49 | nr_blocks = axmap_set_nr(f->io_axmap, block, nr_blocks); |
df415585 | 50 | |
51ede0b1 JA |
51 | if ((nr_blocks * min_bs) < io_u->buflen) |
52 | io_u->buflen = nr_blocks * min_bs; | |
df415585 JA |
53 | } |
54 | ||
74776733 JA |
55 | static uint64_t last_block(struct thread_data *td, struct fio_file *f, |
56 | enum fio_ddir ddir) | |
2ba1c290 | 57 | { |
74776733 JA |
58 | uint64_t max_blocks; |
59 | uint64_t max_size; | |
2ba1c290 | 60 | |
ff58fced JA |
61 | assert(ddir_rw(ddir)); |
62 | ||
d9dd70f7 JA |
63 | /* |
64 | * Hmm, should we make sure that ->io_size <= ->real_file_size? | |
65 | */ | |
66 | max_size = f->io_size; | |
67 | if (max_size > f->real_file_size) | |
68 | max_size = f->real_file_size; | |
69 | ||
ed335855 SN |
70 | if (td->o.zone_range) |
71 | max_size = td->o.zone_range; | |
72 | ||
0412d12e JE |
73 | if (td->o.min_bs[ddir] > td->o.ba[ddir]) |
74 | max_size -= td->o.min_bs[ddir] - td->o.ba[ddir]; | |
75 | ||
1ae83d45 | 76 | max_blocks = max_size / (uint64_t) td->o.ba[ddir]; |
2ba1c290 JA |
77 | if (!max_blocks) |
78 | return 0; | |
79 | ||
67778e88 | 80 | return max_blocks; |
2ba1c290 JA |
81 | } |
82 | ||
1ae83d45 JA |
83 | struct rand_off { |
84 | struct flist_head list; | |
85 | uint64_t off; | |
86 | }; | |
87 | ||
e25839d4 | 88 | static int __get_next_rand_offset(struct thread_data *td, struct fio_file *f, |
e0a04ac1 JA |
89 | enum fio_ddir ddir, uint64_t *b, |
90 | uint64_t lastb) | |
ec4015da | 91 | { |
6f49f8bc | 92 | uint64_t r; |
5e0baa7f | 93 | |
c3546b53 JA |
94 | if (td->o.random_generator == FIO_RAND_GEN_TAUSWORTHE || |
95 | td->o.random_generator == FIO_RAND_GEN_TAUSWORTHE64) { | |
6f49f8bc | 96 | |
d6b72507 | 97 | r = __rand(&td->random_state); |
8055e41d | 98 | |
4b91ee8f | 99 | dprint(FD_RANDOM, "off rand %llu\n", (unsigned long long) r); |
8055e41d | 100 | |
e0a04ac1 | 101 | *b = lastb * (r / (rand_max(&td->random_state) + 1.0)); |
51ede0b1 | 102 | } else { |
8055e41d | 103 | uint64_t off = 0; |
43c63a78 | 104 | |
967d1b63 JA |
105 | assert(fio_file_lfsr(f)); |
106 | ||
6f49f8bc | 107 | if (lfsr_next(&f->lfsr, &off)) |
8055e41d | 108 | return 1; |
ec4015da | 109 | |
8055e41d JA |
110 | *b = off; |
111 | } | |
0ce8b119 | 112 | |
ec4015da | 113 | /* |
51ede0b1 | 114 | * if we are not maintaining a random map, we are done. |
ec4015da | 115 | */ |
51ede0b1 JA |
116 | if (!file_randommap(td, f)) |
117 | goto ret; | |
43c63a78 JA |
118 | |
119 | /* | |
51ede0b1 | 120 | * calculate map offset and check if it's free |
43c63a78 | 121 | */ |
51ede0b1 JA |
122 | if (random_map_free(f, *b)) |
123 | goto ret; | |
124 | ||
4b91ee8f JA |
125 | dprint(FD_RANDOM, "get_next_rand_offset: offset %llu busy\n", |
126 | (unsigned long long) *b); | |
51ede0b1 | 127 | |
7ebd796f | 128 | *b = axmap_next_free(f->io_axmap, *b); |
51ede0b1 JA |
129 | if (*b == (uint64_t) -1ULL) |
130 | return 1; | |
0ce8b119 JA |
131 | ret: |
132 | return 0; | |
ec4015da JA |
133 | } |
134 | ||
925fee33 JA |
135 | static int __get_next_rand_offset_zipf(struct thread_data *td, |
136 | struct fio_file *f, enum fio_ddir ddir, | |
1ae83d45 | 137 | uint64_t *b) |
e25839d4 | 138 | { |
9c6f6316 | 139 | *b = zipf_next(&f->zipf); |
e25839d4 JA |
140 | return 0; |
141 | } | |
142 | ||
925fee33 JA |
143 | static int __get_next_rand_offset_pareto(struct thread_data *td, |
144 | struct fio_file *f, enum fio_ddir ddir, | |
1ae83d45 | 145 | uint64_t *b) |
925fee33 | 146 | { |
9c6f6316 | 147 | *b = pareto_next(&f->zipf); |
925fee33 JA |
148 | return 0; |
149 | } | |
150 | ||
56d9fa4b JA |
151 | static int __get_next_rand_offset_gauss(struct thread_data *td, |
152 | struct fio_file *f, enum fio_ddir ddir, | |
153 | uint64_t *b) | |
154 | { | |
155 | *b = gauss_next(&f->gauss); | |
156 | return 0; | |
157 | } | |
158 | ||
e0a04ac1 JA |
159 | static int __get_next_rand_offset_zoned(struct thread_data *td, |
160 | struct fio_file *f, enum fio_ddir ddir, | |
161 | uint64_t *b) | |
162 | { | |
163 | unsigned int v, send, stotal; | |
164 | uint64_t offset, lastb; | |
165 | static int warned; | |
166 | struct zone_split_index *zsi; | |
167 | ||
168 | lastb = last_block(td, f, ddir); | |
169 | if (!lastb) | |
170 | return 1; | |
171 | ||
172 | if (!td->o.zone_split_nr[ddir]) { | |
173 | bail: | |
174 | return __get_next_rand_offset(td, f, ddir, b, lastb); | |
175 | } | |
176 | ||
177 | /* | |
178 | * Generate a value, v, between 1 and 100, both inclusive | |
179 | */ | |
36dd3379 | 180 | v = rand32_between(&td->zone_state, 1, 100); |
e0a04ac1 JA |
181 | |
182 | zsi = &td->zone_state_index[ddir][v - 1]; | |
183 | stotal = zsi->size_perc_prev; | |
184 | send = zsi->size_perc; | |
185 | ||
186 | /* | |
187 | * Should never happen | |
188 | */ | |
189 | if (send == -1U) { | |
190 | if (!warned) { | |
191 | log_err("fio: bug in zoned generation\n"); | |
192 | warned = 1; | |
193 | } | |
194 | goto bail; | |
195 | } | |
196 | ||
197 | /* | |
198 | * 'send' is some percentage below or equal to 100 that | |
199 | * marks the end of the current IO range. 'stotal' marks | |
200 | * the start, in percent. | |
201 | */ | |
202 | if (stotal) | |
203 | offset = stotal * lastb / 100ULL; | |
204 | else | |
205 | offset = 0; | |
206 | ||
207 | lastb = lastb * (send - stotal) / 100ULL; | |
208 | ||
209 | /* | |
210 | * Generate index from 0..send-of-lastb | |
211 | */ | |
212 | if (__get_next_rand_offset(td, f, ddir, b, lastb) == 1) | |
213 | return 1; | |
214 | ||
215 | /* | |
216 | * Add our start offset, if any | |
217 | */ | |
218 | if (offset) | |
219 | *b += offset; | |
220 | ||
221 | return 0; | |
222 | } | |
56d9fa4b | 223 | |
1ae83d45 JA |
224 | static int flist_cmp(void *data, struct flist_head *a, struct flist_head *b) |
225 | { | |
226 | struct rand_off *r1 = flist_entry(a, struct rand_off, list); | |
227 | struct rand_off *r2 = flist_entry(b, struct rand_off, list); | |
228 | ||
229 | return r1->off - r2->off; | |
230 | } | |
231 | ||
232 | static int get_off_from_method(struct thread_data *td, struct fio_file *f, | |
233 | enum fio_ddir ddir, uint64_t *b) | |
e25839d4 | 234 | { |
e0a04ac1 JA |
235 | if (td->o.random_distribution == FIO_RAND_DIST_RANDOM) { |
236 | uint64_t lastb; | |
237 | ||
238 | lastb = last_block(td, f, ddir); | |
239 | if (!lastb) | |
240 | return 1; | |
241 | ||
242 | return __get_next_rand_offset(td, f, ddir, b, lastb); | |
243 | } else if (td->o.random_distribution == FIO_RAND_DIST_ZIPF) | |
e25839d4 | 244 | return __get_next_rand_offset_zipf(td, f, ddir, b); |
925fee33 JA |
245 | else if (td->o.random_distribution == FIO_RAND_DIST_PARETO) |
246 | return __get_next_rand_offset_pareto(td, f, ddir, b); | |
56d9fa4b JA |
247 | else if (td->o.random_distribution == FIO_RAND_DIST_GAUSS) |
248 | return __get_next_rand_offset_gauss(td, f, ddir, b); | |
e0a04ac1 JA |
249 | else if (td->o.random_distribution == FIO_RAND_DIST_ZONED) |
250 | return __get_next_rand_offset_zoned(td, f, ddir, b); | |
e25839d4 JA |
251 | |
252 | log_err("fio: unknown random distribution: %d\n", td->o.random_distribution); | |
253 | return 1; | |
254 | } | |
255 | ||
bcd5abfa JA |
256 | /* |
257 | * Sort the reads for a verify phase in batches of verifysort_nr, if | |
258 | * specified. | |
259 | */ | |
e39c0676 | 260 | static inline bool should_sort_io(struct thread_data *td) |
bcd5abfa JA |
261 | { |
262 | if (!td->o.verifysort_nr || !td->o.do_verify) | |
e39c0676 | 263 | return false; |
bcd5abfa | 264 | if (!td_random(td)) |
e39c0676 | 265 | return false; |
bcd5abfa | 266 | if (td->runstate != TD_VERIFYING) |
e39c0676 | 267 | return false; |
c3546b53 JA |
268 | if (td->o.random_generator == FIO_RAND_GEN_TAUSWORTHE || |
269 | td->o.random_generator == FIO_RAND_GEN_TAUSWORTHE64) | |
e39c0676 | 270 | return false; |
bcd5abfa | 271 | |
e39c0676 | 272 | return true; |
bcd5abfa JA |
273 | } |
274 | ||
e39c0676 | 275 | static bool should_do_random(struct thread_data *td, enum fio_ddir ddir) |
211c9b89 JA |
276 | { |
277 | unsigned int v; | |
211c9b89 | 278 | |
d9472271 | 279 | if (td->o.perc_rand[ddir] == 100) |
e39c0676 | 280 | return true; |
211c9b89 | 281 | |
36dd3379 | 282 | v = rand32_between(&td->seq_rand_state[ddir], 1, 100); |
211c9b89 | 283 | |
d9472271 | 284 | return v <= td->o.perc_rand[ddir]; |
211c9b89 JA |
285 | } |
286 | ||
1ae83d45 JA |
287 | static int get_next_rand_offset(struct thread_data *td, struct fio_file *f, |
288 | enum fio_ddir ddir, uint64_t *b) | |
289 | { | |
290 | struct rand_off *r; | |
291 | int i, ret = 1; | |
292 | ||
bcd5abfa | 293 | if (!should_sort_io(td)) |
1ae83d45 JA |
294 | return get_off_from_method(td, f, ddir, b); |
295 | ||
296 | if (!flist_empty(&td->next_rand_list)) { | |
1ae83d45 | 297 | fetch: |
9342d5f8 | 298 | r = flist_first_entry(&td->next_rand_list, struct rand_off, list); |
1ae83d45 JA |
299 | flist_del(&r->list); |
300 | *b = r->off; | |
301 | free(r); | |
302 | return 0; | |
303 | } | |
304 | ||
305 | for (i = 0; i < td->o.verifysort_nr; i++) { | |
306 | r = malloc(sizeof(*r)); | |
307 | ||
308 | ret = get_off_from_method(td, f, ddir, &r->off); | |
309 | if (ret) { | |
310 | free(r); | |
311 | break; | |
312 | } | |
313 | ||
314 | flist_add(&r->list, &td->next_rand_list); | |
315 | } | |
316 | ||
317 | if (ret && !i) | |
318 | return ret; | |
319 | ||
320 | assert(!flist_empty(&td->next_rand_list)); | |
321 | flist_sort(NULL, &td->next_rand_list, flist_cmp); | |
322 | goto fetch; | |
323 | } | |
324 | ||
38dad62d | 325 | static int get_next_rand_block(struct thread_data *td, struct fio_file *f, |
1ae83d45 | 326 | enum fio_ddir ddir, uint64_t *b) |
38dad62d | 327 | { |
c04e4661 DE |
328 | if (!get_next_rand_offset(td, f, ddir, b)) |
329 | return 0; | |
330 | ||
8c07860d JA |
331 | if (td->o.time_based || |
332 | (td->o.file_service_type & __FIO_FSERVICE_NONUNIFORM)) { | |
33c48814 | 333 | fio_file_reset(td, f); |
c04e4661 DE |
334 | if (!get_next_rand_offset(td, f, ddir, b)) |
335 | return 0; | |
38dad62d JA |
336 | } |
337 | ||
c04e4661 | 338 | dprint(FD_IO, "%s: rand offset failed, last=%llu, size=%llu\n", |
f1dfb668 | 339 | f->file_name, (unsigned long long) f->last_pos[ddir], |
4b91ee8f | 340 | (unsigned long long) f->real_file_size); |
c04e4661 | 341 | return 1; |
38dad62d JA |
342 | } |
343 | ||
37cf9e3c | 344 | static int get_next_seq_offset(struct thread_data *td, struct fio_file *f, |
1ae83d45 | 345 | enum fio_ddir ddir, uint64_t *offset) |
38dad62d | 346 | { |
ac002339 JA |
347 | struct thread_options *o = &td->o; |
348 | ||
ff58fced JA |
349 | assert(ddir_rw(ddir)); |
350 | ||
f1dfb668 | 351 | if (f->last_pos[ddir] >= f->io_size + get_start_offset(td, f) && |
19ddc35b JA |
352 | o->time_based) { |
353 | struct thread_options *o = &td->o; | |
354 | uint64_t io_size = f->io_size + (f->io_size % o->min_bs[ddir]); | |
355 | ||
356 | if (io_size > f->last_pos[ddir]) | |
357 | f->last_pos[ddir] = 0; | |
358 | else | |
359 | f->last_pos[ddir] = f->last_pos[ddir] - io_size; | |
360 | } | |
c04e4661 | 361 | |
f1dfb668 | 362 | if (f->last_pos[ddir] < f->real_file_size) { |
1ae83d45 | 363 | uint64_t pos; |
059b0802 | 364 | |
c22825bb JA |
365 | if (f->last_pos[ddir] == f->file_offset && o->ddir_seq_add < 0) { |
366 | if (f->real_file_size > f->io_size) | |
367 | f->last_pos[ddir] = f->io_size; | |
368 | else | |
369 | f->last_pos[ddir] = f->real_file_size; | |
370 | } | |
a66da7a2 | 371 | |
f1dfb668 | 372 | pos = f->last_pos[ddir] - f->file_offset; |
ac002339 JA |
373 | if (pos && o->ddir_seq_add) { |
374 | pos += o->ddir_seq_add; | |
375 | ||
376 | /* | |
377 | * If we reach beyond the end of the file | |
378 | * with holed IO, wrap around to the | |
b0a84f48 JA |
379 | * beginning again. If we're doing backwards IO, |
380 | * wrap to the end. | |
ac002339 | 381 | */ |
b0a84f48 JA |
382 | if (pos >= f->real_file_size) { |
383 | if (o->ddir_seq_add > 0) | |
384 | pos = f->file_offset; | |
c22825bb JA |
385 | else { |
386 | if (f->real_file_size > f->io_size) | |
387 | pos = f->io_size; | |
388 | else | |
389 | pos = f->real_file_size; | |
390 | ||
391 | pos += o->ddir_seq_add; | |
392 | } | |
b0a84f48 | 393 | } |
ac002339 | 394 | } |
059b0802 | 395 | |
37cf9e3c | 396 | *offset = pos; |
38dad62d JA |
397 | return 0; |
398 | } | |
399 | ||
400 | return 1; | |
401 | } | |
402 | ||
403 | static int get_next_block(struct thread_data *td, struct io_u *io_u, | |
6aca9b3d JA |
404 | enum fio_ddir ddir, int rw_seq, |
405 | unsigned int *is_random) | |
38dad62d JA |
406 | { |
407 | struct fio_file *f = io_u->file; | |
1ae83d45 | 408 | uint64_t b, offset; |
38dad62d JA |
409 | int ret; |
410 | ||
ff58fced JA |
411 | assert(ddir_rw(ddir)); |
412 | ||
37cf9e3c JA |
413 | b = offset = -1ULL; |
414 | ||
38dad62d | 415 | if (rw_seq) { |
211c9b89 | 416 | if (td_random(td)) { |
6aca9b3d | 417 | if (should_do_random(td, ddir)) { |
211c9b89 | 418 | ret = get_next_rand_block(td, f, ddir, &b); |
6aca9b3d JA |
419 | *is_random = 1; |
420 | } else { | |
421 | *is_random = 0; | |
1651e431 | 422 | io_u_set(td, io_u, IO_U_F_BUSY_OK); |
211c9b89 JA |
423 | ret = get_next_seq_offset(td, f, ddir, &offset); |
424 | if (ret) | |
425 | ret = get_next_rand_block(td, f, ddir, &b); | |
426 | } | |
6aca9b3d JA |
427 | } else { |
428 | *is_random = 0; | |
37cf9e3c | 429 | ret = get_next_seq_offset(td, f, ddir, &offset); |
6aca9b3d | 430 | } |
38dad62d | 431 | } else { |
1651e431 | 432 | io_u_set(td, io_u, IO_U_F_BUSY_OK); |
6aca9b3d | 433 | *is_random = 0; |
38dad62d JA |
434 | |
435 | if (td->o.rw_seq == RW_SEQ_SEQ) { | |
37cf9e3c | 436 | ret = get_next_seq_offset(td, f, ddir, &offset); |
6aca9b3d | 437 | if (ret) { |
37cf9e3c | 438 | ret = get_next_rand_block(td, f, ddir, &b); |
6aca9b3d JA |
439 | *is_random = 0; |
440 | } | |
38dad62d | 441 | } else if (td->o.rw_seq == RW_SEQ_IDENT) { |
f1dfb668 JA |
442 | if (f->last_start[ddir] != -1ULL) |
443 | offset = f->last_start[ddir] - f->file_offset; | |
38dad62d | 444 | else |
37cf9e3c | 445 | offset = 0; |
38dad62d JA |
446 | ret = 0; |
447 | } else { | |
448 | log_err("fio: unknown rw_seq=%d\n", td->o.rw_seq); | |
449 | ret = 1; | |
450 | } | |
451 | } | |
6d68b997 | 452 | |
37cf9e3c JA |
453 | if (!ret) { |
454 | if (offset != -1ULL) | |
455 | io_u->offset = offset; | |
456 | else if (b != -1ULL) | |
457 | io_u->offset = b * td->o.ba[ddir]; | |
458 | else { | |
4e0a8fa2 | 459 | log_err("fio: bug in offset generation: offset=%llu, b=%llu\n", (unsigned long long) offset, (unsigned long long) b); |
37cf9e3c JA |
460 | ret = 1; |
461 | } | |
462 | } | |
463 | ||
38dad62d JA |
464 | return ret; |
465 | } | |
466 | ||
10ba535a JA |
467 | /* |
468 | * For random io, generate a random new block and see if it's used. Repeat | |
469 | * until we find a free one. For sequential io, just return the end of | |
470 | * the last io issued. | |
471 | */ | |
6aca9b3d JA |
472 | static int __get_next_offset(struct thread_data *td, struct io_u *io_u, |
473 | unsigned int *is_random) | |
10ba535a | 474 | { |
9bf2061e | 475 | struct fio_file *f = io_u->file; |
4ba66134 | 476 | enum fio_ddir ddir = io_u->ddir; |
38dad62d | 477 | int rw_seq_hit = 0; |
10ba535a | 478 | |
ff58fced JA |
479 | assert(ddir_rw(ddir)); |
480 | ||
38dad62d JA |
481 | if (td->o.ddir_seq_nr && !--td->ddir_seq_nr) { |
482 | rw_seq_hit = 1; | |
5736c10d | 483 | td->ddir_seq_nr = td->o.ddir_seq_nr; |
38dad62d | 484 | } |
211097b2 | 485 | |
6aca9b3d | 486 | if (get_next_block(td, io_u, ddir, rw_seq_hit, is_random)) |
38dad62d | 487 | return 1; |
10ba535a | 488 | |
009bd847 JA |
489 | if (io_u->offset >= f->io_size) { |
490 | dprint(FD_IO, "get_next_offset: offset %llu >= io_size %llu\n", | |
4b91ee8f JA |
491 | (unsigned long long) io_u->offset, |
492 | (unsigned long long) f->io_size); | |
009bd847 JA |
493 | return 1; |
494 | } | |
495 | ||
496 | io_u->offset += f->file_offset; | |
2ba1c290 JA |
497 | if (io_u->offset >= f->real_file_size) { |
498 | dprint(FD_IO, "get_next_offset: offset %llu >= size %llu\n", | |
4b91ee8f JA |
499 | (unsigned long long) io_u->offset, |
500 | (unsigned long long) f->real_file_size); | |
10ba535a | 501 | return 1; |
2ba1c290 | 502 | } |
10ba535a JA |
503 | |
504 | return 0; | |
505 | } | |
506 | ||
6aca9b3d JA |
507 | static int get_next_offset(struct thread_data *td, struct io_u *io_u, |
508 | unsigned int *is_random) | |
15dc1934 | 509 | { |
d72be545 JA |
510 | if (td->flags & TD_F_PROFILE_OPS) { |
511 | struct prof_io_ops *ops = &td->prof_io_ops; | |
7eb36574 | 512 | |
d72be545 | 513 | if (ops->fill_io_u_off) |
6aca9b3d | 514 | return ops->fill_io_u_off(td, io_u, is_random); |
d72be545 | 515 | } |
15dc1934 | 516 | |
6aca9b3d | 517 | return __get_next_offset(td, io_u, is_random); |
15dc1934 JA |
518 | } |
519 | ||
e39c0676 JA |
520 | static inline bool io_u_fits(struct thread_data *td, struct io_u *io_u, |
521 | unsigned int buflen) | |
79944128 JA |
522 | { |
523 | struct fio_file *f = io_u->file; | |
524 | ||
bedc9dc2 | 525 | return io_u->offset + buflen <= f->io_size + get_start_offset(td, f); |
79944128 JA |
526 | } |
527 | ||
6aca9b3d JA |
528 | static unsigned int __get_next_buflen(struct thread_data *td, struct io_u *io_u, |
529 | unsigned int is_random) | |
10ba535a | 530 | { |
6aca9b3d | 531 | int ddir = io_u->ddir; |
24d23ca7 | 532 | unsigned int buflen = 0; |
f3059de1 | 533 | unsigned int minbs, maxbs; |
3dd29f7c | 534 | uint64_t frand_max, r; |
10ba535a | 535 | |
9ee1c647 | 536 | assert(ddir_rw(ddir)); |
6aca9b3d JA |
537 | |
538 | if (td->o.bs_is_seq_rand) | |
539 | ddir = is_random ? DDIR_WRITE: DDIR_READ; | |
ff58fced | 540 | |
f3059de1 JA |
541 | minbs = td->o.min_bs[ddir]; |
542 | maxbs = td->o.max_bs[ddir]; | |
543 | ||
79944128 JA |
544 | if (minbs == maxbs) |
545 | return minbs; | |
546 | ||
52c58027 JA |
547 | /* |
548 | * If we can't satisfy the min block size from here, then fail | |
549 | */ | |
550 | if (!io_u_fits(td, io_u, minbs)) | |
551 | return 0; | |
552 | ||
c3546b53 | 553 | frand_max = rand_max(&td->bsrange_state); |
79944128 | 554 | do { |
d6b72507 | 555 | r = __rand(&td->bsrange_state); |
4c07ad86 | 556 | |
720e84ad | 557 | if (!td->o.bssplit_nr[ddir]) { |
f3059de1 | 558 | buflen = 1 + (unsigned int) ((double) maxbs * |
c3546b53 | 559 | (r / (frand_max + 1.0))); |
f3059de1 JA |
560 | if (buflen < minbs) |
561 | buflen = minbs; | |
5ec10eaa | 562 | } else { |
3dd29f7c | 563 | long long perc = 0; |
564ca972 JA |
564 | unsigned int i; |
565 | ||
720e84ad JA |
566 | for (i = 0; i < td->o.bssplit_nr[ddir]; i++) { |
567 | struct bssplit *bsp = &td->o.bssplit[ddir][i]; | |
564ca972 JA |
568 | |
569 | buflen = bsp->bs; | |
570 | perc += bsp->perc; | |
3dd29f7c JA |
571 | if (!perc) |
572 | break; | |
573 | if ((r / perc <= frand_max / 100ULL) && | |
79944128 | 574 | io_u_fits(td, io_u, buflen)) |
564ca972 JA |
575 | break; |
576 | } | |
577 | } | |
79944128 | 578 | |
ced5db25 | 579 | if (td->o.verify != VERIFY_NONE) |
a9f70b1f JB |
580 | buflen = (buflen + td->o.verify_interval - 1) & |
581 | ~(td->o.verify_interval - 1); | |
582 | ||
f3059de1 | 583 | if (!td->o.bs_unaligned && is_power_of_2(minbs)) |
7c306bb1 | 584 | buflen &= ~(minbs - 1); |
10ba535a | 585 | |
79944128 | 586 | } while (!io_u_fits(td, io_u, buflen)); |
6a5e6884 | 587 | |
10ba535a JA |
588 | return buflen; |
589 | } | |
590 | ||
6aca9b3d JA |
591 | static unsigned int get_next_buflen(struct thread_data *td, struct io_u *io_u, |
592 | unsigned int is_random) | |
15dc1934 | 593 | { |
d72be545 JA |
594 | if (td->flags & TD_F_PROFILE_OPS) { |
595 | struct prof_io_ops *ops = &td->prof_io_ops; | |
7eb36574 | 596 | |
d72be545 | 597 | if (ops->fill_io_u_size) |
6aca9b3d | 598 | return ops->fill_io_u_size(td, io_u, is_random); |
d72be545 | 599 | } |
15dc1934 | 600 | |
6aca9b3d | 601 | return __get_next_buflen(td, io_u, is_random); |
15dc1934 JA |
602 | } |
603 | ||
afe24a5a JA |
604 | static void set_rwmix_bytes(struct thread_data *td) |
605 | { | |
afe24a5a JA |
606 | unsigned int diff; |
607 | ||
608 | /* | |
609 | * we do time or byte based switch. this is needed because | |
610 | * buffered writes may issue a lot quicker than they complete, | |
611 | * whereas reads do not. | |
612 | */ | |
e47f799f | 613 | diff = td->o.rwmix[td->rwmix_ddir ^ 1]; |
04c540d9 | 614 | td->rwmix_issues = (td->io_issues[td->rwmix_ddir] * diff) / 100; |
e47f799f JA |
615 | } |
616 | ||
617 | static inline enum fio_ddir get_rand_ddir(struct thread_data *td) | |
618 | { | |
619 | unsigned int v; | |
e47f799f | 620 | |
36dd3379 | 621 | v = rand32_between(&td->rwmix_state, 1, 100); |
4c07ad86 | 622 | |
04c540d9 | 623 | if (v <= td->o.rwmix[DDIR_READ]) |
e47f799f JA |
624 | return DDIR_READ; |
625 | ||
626 | return DDIR_WRITE; | |
afe24a5a JA |
627 | } |
628 | ||
0d593542 | 629 | int io_u_quiesce(struct thread_data *td) |
002e7183 | 630 | { |
0d593542 JA |
631 | int completed = 0; |
632 | ||
002e7183 JA |
633 | /* |
634 | * We are going to sleep, ensure that we flush anything pending as | |
635 | * not to skew our latency numbers. | |
636 | * | |
637 | * Changed to only monitor 'in flight' requests here instead of the | |
638 | * td->cur_depth, b/c td->cur_depth does not accurately represent | |
639 | * io's that have been actually submitted to an async engine, | |
640 | * and cur_depth is meaningless for sync engines. | |
641 | */ | |
9cc80b6d JA |
642 | if (td->io_u_queued || td->cur_depth) { |
643 | int fio_unused ret; | |
644 | ||
645 | ret = td_io_commit(td); | |
646 | } | |
647 | ||
002e7183 JA |
648 | while (td->io_u_in_flight) { |
649 | int fio_unused ret; | |
650 | ||
55312f9f | 651 | ret = io_u_queued_complete(td, 1); |
0d593542 JA |
652 | if (ret > 0) |
653 | completed += ret; | |
002e7183 | 654 | } |
0d593542 | 655 | |
6be06c46 JA |
656 | if (td->flags & TD_F_REGROW_LOGS) |
657 | regrow_logs(td); | |
658 | ||
0d593542 | 659 | return completed; |
002e7183 JA |
660 | } |
661 | ||
581e7141 JA |
662 | static enum fio_ddir rate_ddir(struct thread_data *td, enum fio_ddir ddir) |
663 | { | |
664 | enum fio_ddir odir = ddir ^ 1; | |
90eff1c9 SW |
665 | long usec; |
666 | uint64_t now; | |
581e7141 | 667 | |
ff58fced | 668 | assert(ddir_rw(ddir)); |
50a8ce86 | 669 | now = utime_since_now(&td->start); |
ff58fced | 670 | |
50a8ce86 D |
671 | /* |
672 | * if rate_next_io_time is in the past, need to catch up to rate | |
673 | */ | |
674 | if (td->rate_next_io_time[ddir] <= now) | |
581e7141 JA |
675 | return ddir; |
676 | ||
677 | /* | |
50a8ce86 | 678 | * We are ahead of rate in this direction. See if we |
581e7141 JA |
679 | * should switch. |
680 | */ | |
315fcfec | 681 | if (td_rw(td) && td->o.rwmix[odir]) { |
581e7141 | 682 | /* |
50a8ce86 | 683 | * Other direction is behind rate, switch |
581e7141 | 684 | */ |
50a8ce86 | 685 | if (td->rate_next_io_time[odir] <= now) |
581e7141 JA |
686 | return odir; |
687 | ||
688 | /* | |
50a8ce86 D |
689 | * Both directions are ahead of rate. sleep the min |
690 | * switch if necissary | |
581e7141 | 691 | */ |
50a8ce86 D |
692 | if (td->rate_next_io_time[ddir] <= |
693 | td->rate_next_io_time[odir]) { | |
694 | usec = td->rate_next_io_time[ddir] - now; | |
581e7141 | 695 | } else { |
50a8ce86 | 696 | usec = td->rate_next_io_time[odir] - now; |
581e7141 JA |
697 | ddir = odir; |
698 | } | |
699 | } else | |
50a8ce86 | 700 | usec = td->rate_next_io_time[ddir] - now; |
581e7141 | 701 | |
a9da8ab2 JA |
702 | if (td->o.io_submit_mode == IO_MODE_INLINE) |
703 | io_u_quiesce(td); | |
78c1eda5 | 704 | |
9cc80b6d | 705 | usec = usec_sleep(td, usec); |
581e7141 | 706 | |
581e7141 JA |
707 | return ddir; |
708 | } | |
709 | ||
10ba535a JA |
710 | /* |
711 | * Return the data direction for the next io_u. If the job is a | |
712 | * mixed read/write workload, check the rwmix cycle and switch if | |
713 | * necessary. | |
714 | */ | |
1e97cce9 | 715 | static enum fio_ddir get_rw_ddir(struct thread_data *td) |
10ba535a | 716 | { |
581e7141 JA |
717 | enum fio_ddir ddir; |
718 | ||
5f9099ea JA |
719 | /* |
720 | * see if it's time to fsync | |
721 | */ | |
722 | if (td->o.fsync_blocks && | |
723 | !(td->io_issues[DDIR_WRITE] % td->o.fsync_blocks) && | |
724 | td->io_issues[DDIR_WRITE] && should_fsync(td)) | |
725 | return DDIR_SYNC; | |
726 | ||
727 | /* | |
728 | * see if it's time to fdatasync | |
729 | */ | |
730 | if (td->o.fdatasync_blocks && | |
731 | !(td->io_issues[DDIR_WRITE] % td->o.fdatasync_blocks) && | |
732 | td->io_issues[DDIR_WRITE] && should_fsync(td)) | |
733 | return DDIR_DATASYNC; | |
734 | ||
44f29692 JA |
735 | /* |
736 | * see if it's time to sync_file_range | |
737 | */ | |
738 | if (td->sync_file_range_nr && | |
739 | !(td->io_issues[DDIR_WRITE] % td->sync_file_range_nr) && | |
740 | td->io_issues[DDIR_WRITE] && should_fsync(td)) | |
741 | return DDIR_SYNC_FILE_RANGE; | |
742 | ||
10ba535a | 743 | if (td_rw(td)) { |
10ba535a JA |
744 | /* |
745 | * Check if it's time to seed a new data direction. | |
746 | */ | |
e4928662 | 747 | if (td->io_issues[td->rwmix_ddir] >= td->rwmix_issues) { |
e47f799f JA |
748 | /* |
749 | * Put a top limit on how many bytes we do for | |
750 | * one data direction, to avoid overflowing the | |
751 | * ranges too much | |
752 | */ | |
753 | ddir = get_rand_ddir(td); | |
e47f799f JA |
754 | |
755 | if (ddir != td->rwmix_ddir) | |
756 | set_rwmix_bytes(td); | |
757 | ||
758 | td->rwmix_ddir = ddir; | |
10ba535a | 759 | } |
581e7141 | 760 | ddir = td->rwmix_ddir; |
10ba535a | 761 | } else if (td_read(td)) |
581e7141 | 762 | ddir = DDIR_READ; |
6eaf09d6 | 763 | else if (td_write(td)) |
581e7141 | 764 | ddir = DDIR_WRITE; |
6eaf09d6 SL |
765 | else |
766 | ddir = DDIR_TRIM; | |
581e7141 JA |
767 | |
768 | td->rwmix_ddir = rate_ddir(td, ddir); | |
769 | return td->rwmix_ddir; | |
10ba535a JA |
770 | } |
771 | ||
1ef2b6be JA |
772 | static void set_rw_ddir(struct thread_data *td, struct io_u *io_u) |
773 | { | |
0e4dd95c DE |
774 | enum fio_ddir ddir = get_rw_ddir(td); |
775 | ||
82a90686 | 776 | if (td_trimwrite(td)) { |
0e4dd95c DE |
777 | struct fio_file *f = io_u->file; |
778 | if (f->last_pos[DDIR_WRITE] == f->last_pos[DDIR_TRIM]) | |
779 | ddir = DDIR_TRIM; | |
780 | else | |
781 | ddir = DDIR_WRITE; | |
782 | } | |
783 | ||
784 | io_u->ddir = io_u->acct_ddir = ddir; | |
1ef2b6be | 785 | |
9b87f09b | 786 | if (io_u->ddir == DDIR_WRITE && td_ioengine_flagged(td, FIO_BARRIER) && |
1ef2b6be JA |
787 | td->o.barrier_blocks && |
788 | !(td->io_issues[DDIR_WRITE] % td->o.barrier_blocks) && | |
789 | td->io_issues[DDIR_WRITE]) | |
1651e431 | 790 | io_u_set(td, io_u, IO_U_F_BARRIER); |
1ef2b6be JA |
791 | } |
792 | ||
e8462bd8 | 793 | void put_file_log(struct thread_data *td, struct fio_file *f) |
60f2c658 | 794 | { |
71b84caa | 795 | unsigned int ret = put_file(td, f); |
60f2c658 JA |
796 | |
797 | if (ret) | |
798 | td_verror(td, ret, "file close"); | |
799 | } | |
800 | ||
10ba535a JA |
801 | void put_io_u(struct thread_data *td, struct io_u *io_u) |
802 | { | |
a9da8ab2 JA |
803 | if (td->parent) |
804 | td = td->parent; | |
805 | ||
e8462bd8 JA |
806 | td_io_u_lock(td); |
807 | ||
f8b0bd10 | 808 | if (io_u->file && !(io_u->flags & IO_U_F_NO_FILE_PUT)) |
60f2c658 | 809 | put_file_log(td, io_u->file); |
f8b0bd10 | 810 | |
10ba535a | 811 | io_u->file = NULL; |
1651e431 | 812 | io_u_set(td, io_u, IO_U_F_FREE); |
d7ee2a7d | 813 | |
a9da8ab2 | 814 | if (io_u->flags & IO_U_F_IN_CUR_DEPTH) { |
0c41214f | 815 | td->cur_depth--; |
a9da8ab2 JA |
816 | assert(!(td->flags & TD_F_CHILD)); |
817 | } | |
2ae0b204 | 818 | io_u_qpush(&td->io_u_freelist, io_u); |
e8462bd8 JA |
819 | td_io_u_unlock(td); |
820 | td_io_u_free_notify(td); | |
10ba535a JA |
821 | } |
822 | ||
f2bba182 RR |
823 | void clear_io_u(struct thread_data *td, struct io_u *io_u) |
824 | { | |
1651e431 | 825 | io_u_clear(td, io_u, IO_U_F_FLIGHT); |
f2bba182 RR |
826 | put_io_u(td, io_u); |
827 | } | |
828 | ||
755200a3 JA |
829 | void requeue_io_u(struct thread_data *td, struct io_u **io_u) |
830 | { | |
831 | struct io_u *__io_u = *io_u; | |
bcd5abfa | 832 | enum fio_ddir ddir = acct_ddir(__io_u); |
755200a3 | 833 | |
465221b0 JA |
834 | dprint(FD_IO, "requeue %p\n", __io_u); |
835 | ||
a9da8ab2 JA |
836 | if (td->parent) |
837 | td = td->parent; | |
838 | ||
e8462bd8 JA |
839 | td_io_u_lock(td); |
840 | ||
1651e431 | 841 | io_u_set(td, __io_u, IO_U_F_FREE); |
bcd5abfa JA |
842 | if ((__io_u->flags & IO_U_F_FLIGHT) && ddir_rw(ddir)) |
843 | td->io_issues[ddir]--; | |
5ec10eaa | 844 | |
1651e431 | 845 | io_u_clear(td, __io_u, IO_U_F_FLIGHT); |
a9da8ab2 | 846 | if (__io_u->flags & IO_U_F_IN_CUR_DEPTH) { |
0c41214f | 847 | td->cur_depth--; |
a9da8ab2 JA |
848 | assert(!(td->flags & TD_F_CHILD)); |
849 | } | |
2ae0b204 JA |
850 | |
851 | io_u_rpush(&td->io_u_requeues, __io_u); | |
e8462bd8 | 852 | td_io_u_unlock(td); |
a9da8ab2 | 853 | td_io_u_free_notify(td); |
755200a3 JA |
854 | *io_u = NULL; |
855 | } | |
856 | ||
9bf2061e | 857 | static int fill_io_u(struct thread_data *td, struct io_u *io_u) |
10ba535a | 858 | { |
6aca9b3d JA |
859 | unsigned int is_random; |
860 | ||
9b87f09b | 861 | if (td_ioengine_flagged(td, FIO_NOIO)) |
b4c5e1ac JA |
862 | goto out; |
863 | ||
1ef2b6be | 864 | set_rw_ddir(td, io_u); |
5f9099ea | 865 | |
87dc1ab1 | 866 | /* |
ff58fced | 867 | * fsync() or fdatasync() or trim etc, we are done |
87dc1ab1 | 868 | */ |
ff58fced | 869 | if (!ddir_rw(io_u->ddir)) |
c38e9468 | 870 | goto out; |
a00735e6 | 871 | |
48f5abd3 JA |
872 | /* |
873 | * See if it's time to switch to a new zone | |
874 | */ | |
13af05ae | 875 | if (td->zone_bytes >= td->o.zone_size && td->o.zone_skip) { |
4c8be5b1 JA |
876 | struct fio_file *f = io_u->file; |
877 | ||
48f5abd3 | 878 | td->zone_bytes = 0; |
4c8be5b1 JA |
879 | f->file_offset += td->o.zone_range + td->o.zone_skip; |
880 | ||
881 | /* | |
882 | * Wrap from the beginning, if we exceed the file size | |
883 | */ | |
884 | if (f->file_offset >= f->real_file_size) | |
885 | f->file_offset = f->real_file_size - f->file_offset; | |
f1dfb668 | 886 | f->last_pos[io_u->ddir] = f->file_offset; |
48f5abd3 JA |
887 | td->io_skip_bytes += td->o.zone_skip; |
888 | } | |
889 | ||
10ba535a | 890 | /* |
c685b5b2 JA |
891 | * No log, let the seq/rand engine retrieve the next buflen and |
892 | * position. | |
10ba535a | 893 | */ |
6aca9b3d | 894 | if (get_next_offset(td, io_u, &is_random)) { |
2ba1c290 | 895 | dprint(FD_IO, "io_u %p, failed getting offset\n", io_u); |
bca4ed4d | 896 | return 1; |
2ba1c290 | 897 | } |
10ba535a | 898 | |
6aca9b3d | 899 | io_u->buflen = get_next_buflen(td, io_u, is_random); |
2ba1c290 JA |
900 | if (!io_u->buflen) { |
901 | dprint(FD_IO, "io_u %p, failed getting buflen\n", io_u); | |
bca4ed4d | 902 | return 1; |
2ba1c290 | 903 | } |
bca4ed4d | 904 | |
2ba1c290 JA |
905 | if (io_u->offset + io_u->buflen > io_u->file->real_file_size) { |
906 | dprint(FD_IO, "io_u %p, offset too large\n", io_u); | |
4b91ee8f JA |
907 | dprint(FD_IO, " off=%llu/%lu > %llu\n", |
908 | (unsigned long long) io_u->offset, io_u->buflen, | |
909 | (unsigned long long) io_u->file->real_file_size); | |
6a5e6884 | 910 | return 1; |
2ba1c290 | 911 | } |
6a5e6884 | 912 | |
bca4ed4d JA |
913 | /* |
914 | * mark entry before potentially trimming io_u | |
915 | */ | |
303032ae | 916 | if (td_random(td) && file_randommap(td, io_u->file)) |
9bf2061e | 917 | mark_random_map(td, io_u); |
bca4ed4d | 918 | |
c38e9468 | 919 | out: |
2ba1c290 | 920 | dprint_io_u(io_u, "fill_io_u"); |
d9d91e39 | 921 | td->zone_bytes += io_u->buflen; |
bca4ed4d | 922 | return 0; |
10ba535a JA |
923 | } |
924 | ||
838bc709 JA |
925 | static void __io_u_mark_map(unsigned int *map, unsigned int nr) |
926 | { | |
2b13e716 | 927 | int idx = 0; |
838bc709 JA |
928 | |
929 | switch (nr) { | |
930 | default: | |
2b13e716 | 931 | idx = 6; |
838bc709 JA |
932 | break; |
933 | case 33 ... 64: | |
2b13e716 | 934 | idx = 5; |
838bc709 JA |
935 | break; |
936 | case 17 ... 32: | |
2b13e716 | 937 | idx = 4; |
838bc709 JA |
938 | break; |
939 | case 9 ... 16: | |
2b13e716 | 940 | idx = 3; |
838bc709 JA |
941 | break; |
942 | case 5 ... 8: | |
2b13e716 | 943 | idx = 2; |
838bc709 JA |
944 | break; |
945 | case 1 ... 4: | |
2b13e716 | 946 | idx = 1; |
838bc709 JA |
947 | case 0: |
948 | break; | |
949 | } | |
950 | ||
2b13e716 | 951 | map[idx]++; |
838bc709 JA |
952 | } |
953 | ||
954 | void io_u_mark_submit(struct thread_data *td, unsigned int nr) | |
955 | { | |
956 | __io_u_mark_map(td->ts.io_u_submit, nr); | |
957 | td->ts.total_submit++; | |
958 | } | |
959 | ||
960 | void io_u_mark_complete(struct thread_data *td, unsigned int nr) | |
961 | { | |
962 | __io_u_mark_map(td->ts.io_u_complete, nr); | |
963 | td->ts.total_complete++; | |
964 | } | |
965 | ||
d8005759 | 966 | void io_u_mark_depth(struct thread_data *td, unsigned int nr) |
71619dc2 | 967 | { |
2b13e716 | 968 | int idx = 0; |
71619dc2 JA |
969 | |
970 | switch (td->cur_depth) { | |
971 | default: | |
2b13e716 | 972 | idx = 6; |
a783e61a | 973 | break; |
71619dc2 | 974 | case 32 ... 63: |
2b13e716 | 975 | idx = 5; |
a783e61a | 976 | break; |
71619dc2 | 977 | case 16 ... 31: |
2b13e716 | 978 | idx = 4; |
a783e61a | 979 | break; |
71619dc2 | 980 | case 8 ... 15: |
2b13e716 | 981 | idx = 3; |
a783e61a | 982 | break; |
71619dc2 | 983 | case 4 ... 7: |
2b13e716 | 984 | idx = 2; |
a783e61a | 985 | break; |
71619dc2 | 986 | case 2 ... 3: |
2b13e716 | 987 | idx = 1; |
71619dc2 JA |
988 | case 1: |
989 | break; | |
990 | } | |
991 | ||
2b13e716 | 992 | td->ts.io_u_map[idx] += nr; |
71619dc2 JA |
993 | } |
994 | ||
04a0feae JA |
995 | static void io_u_mark_lat_usec(struct thread_data *td, unsigned long usec) |
996 | { | |
2b13e716 | 997 | int idx = 0; |
04a0feae JA |
998 | |
999 | assert(usec < 1000); | |
1000 | ||
1001 | switch (usec) { | |
1002 | case 750 ... 999: | |
2b13e716 | 1003 | idx = 9; |
04a0feae JA |
1004 | break; |
1005 | case 500 ... 749: | |
2b13e716 | 1006 | idx = 8; |
04a0feae JA |
1007 | break; |
1008 | case 250 ... 499: | |
2b13e716 | 1009 | idx = 7; |
04a0feae JA |
1010 | break; |
1011 | case 100 ... 249: | |
2b13e716 | 1012 | idx = 6; |
04a0feae JA |
1013 | break; |
1014 | case 50 ... 99: | |
2b13e716 | 1015 | idx = 5; |
04a0feae JA |
1016 | break; |
1017 | case 20 ... 49: | |
2b13e716 | 1018 | idx = 4; |
04a0feae JA |
1019 | break; |
1020 | case 10 ... 19: | |
2b13e716 | 1021 | idx = 3; |
04a0feae JA |
1022 | break; |
1023 | case 4 ... 9: | |
2b13e716 | 1024 | idx = 2; |
04a0feae JA |
1025 | break; |
1026 | case 2 ... 3: | |
2b13e716 | 1027 | idx = 1; |
04a0feae JA |
1028 | case 0 ... 1: |
1029 | break; | |
1030 | } | |
1031 | ||
2b13e716 JA |
1032 | assert(idx < FIO_IO_U_LAT_U_NR); |
1033 | td->ts.io_u_lat_u[idx]++; | |
04a0feae JA |
1034 | } |
1035 | ||
1036 | static void io_u_mark_lat_msec(struct thread_data *td, unsigned long msec) | |
ec118304 | 1037 | { |
2b13e716 | 1038 | int idx = 0; |
ec118304 JA |
1039 | |
1040 | switch (msec) { | |
1041 | default: | |
2b13e716 | 1042 | idx = 11; |
04a0feae | 1043 | break; |
8abdce66 | 1044 | case 1000 ... 1999: |
2b13e716 | 1045 | idx = 10; |
04a0feae | 1046 | break; |
8abdce66 | 1047 | case 750 ... 999: |
2b13e716 | 1048 | idx = 9; |
04a0feae | 1049 | break; |
8abdce66 | 1050 | case 500 ... 749: |
2b13e716 | 1051 | idx = 8; |
04a0feae | 1052 | break; |
8abdce66 | 1053 | case 250 ... 499: |
2b13e716 | 1054 | idx = 7; |
04a0feae | 1055 | break; |
8abdce66 | 1056 | case 100 ... 249: |
2b13e716 | 1057 | idx = 6; |
04a0feae | 1058 | break; |
8abdce66 | 1059 | case 50 ... 99: |
2b13e716 | 1060 | idx = 5; |
04a0feae | 1061 | break; |
8abdce66 | 1062 | case 20 ... 49: |
2b13e716 | 1063 | idx = 4; |
04a0feae | 1064 | break; |
8abdce66 | 1065 | case 10 ... 19: |
2b13e716 | 1066 | idx = 3; |
04a0feae | 1067 | break; |
8abdce66 | 1068 | case 4 ... 9: |
2b13e716 | 1069 | idx = 2; |
04a0feae | 1070 | break; |
ec118304 | 1071 | case 2 ... 3: |
2b13e716 | 1072 | idx = 1; |
ec118304 JA |
1073 | case 0 ... 1: |
1074 | break; | |
1075 | } | |
1076 | ||
2b13e716 JA |
1077 | assert(idx < FIO_IO_U_LAT_M_NR); |
1078 | td->ts.io_u_lat_m[idx]++; | |
04a0feae JA |
1079 | } |
1080 | ||
1081 | static void io_u_mark_latency(struct thread_data *td, unsigned long usec) | |
1082 | { | |
1083 | if (usec < 1000) | |
1084 | io_u_mark_lat_usec(td, usec); | |
1085 | else | |
1086 | io_u_mark_lat_msec(td, usec / 1000); | |
ec118304 JA |
1087 | } |
1088 | ||
8c07860d JA |
1089 | static unsigned int __get_next_fileno_rand(struct thread_data *td) |
1090 | { | |
1091 | unsigned long fileno; | |
1092 | ||
1093 | if (td->o.file_service_type == FIO_FSERVICE_RANDOM) { | |
1094 | uint64_t frand_max = rand_max(&td->next_file_state); | |
1095 | unsigned long r; | |
1096 | ||
1097 | r = __rand(&td->next_file_state); | |
1098 | return (unsigned int) ((double) td->o.nr_files | |
1099 | * (r / (frand_max + 1.0))); | |
1100 | } | |
1101 | ||
1102 | if (td->o.file_service_type == FIO_FSERVICE_ZIPF) | |
1103 | fileno = zipf_next(&td->next_file_zipf); | |
1104 | else if (td->o.file_service_type == FIO_FSERVICE_PARETO) | |
1105 | fileno = pareto_next(&td->next_file_zipf); | |
1106 | else if (td->o.file_service_type == FIO_FSERVICE_GAUSS) | |
1107 | fileno = gauss_next(&td->next_file_gauss); | |
1108 | else { | |
1109 | log_err("fio: bad file service type: %d\n", td->o.file_service_type); | |
1110 | assert(0); | |
1111 | return 0; | |
1112 | } | |
1113 | ||
1114 | return fileno >> FIO_FSERVICE_SHIFT; | |
1115 | } | |
1116 | ||
0aabe160 JA |
1117 | /* |
1118 | * Get next file to service by choosing one at random | |
1119 | */ | |
2cc52930 JA |
1120 | static struct fio_file *get_next_file_rand(struct thread_data *td, |
1121 | enum fio_file_flags goodf, | |
d6aed795 | 1122 | enum fio_file_flags badf) |
0aabe160 | 1123 | { |
0aabe160 | 1124 | struct fio_file *f; |
1c178180 | 1125 | int fno; |
0aabe160 JA |
1126 | |
1127 | do { | |
87b10676 | 1128 | int opened = 0; |
4c07ad86 | 1129 | |
8c07860d | 1130 | fno = __get_next_fileno_rand(td); |
7c83c089 | 1131 | |
126d65c6 | 1132 | f = td->files[fno]; |
d6aed795 | 1133 | if (fio_file_done(f)) |
059e63c0 | 1134 | continue; |
1c178180 | 1135 | |
d6aed795 | 1136 | if (!fio_file_open(f)) { |
87b10676 JA |
1137 | int err; |
1138 | ||
002fe734 JA |
1139 | if (td->nr_open_files >= td->o.open_files) |
1140 | return ERR_PTR(-EBUSY); | |
1141 | ||
87b10676 JA |
1142 | err = td_io_open_file(td, f); |
1143 | if (err) | |
1144 | continue; | |
1145 | opened = 1; | |
1146 | } | |
1147 | ||
2ba1c290 JA |
1148 | if ((!goodf || (f->flags & goodf)) && !(f->flags & badf)) { |
1149 | dprint(FD_FILE, "get_next_file_rand: %p\n", f); | |
0aabe160 | 1150 | return f; |
2ba1c290 | 1151 | } |
87b10676 JA |
1152 | if (opened) |
1153 | td_io_close_file(td, f); | |
0aabe160 JA |
1154 | } while (1); |
1155 | } | |
1156 | ||
1157 | /* | |
1158 | * Get next file to service by doing round robin between all available ones | |
1159 | */ | |
1c178180 JA |
1160 | static struct fio_file *get_next_file_rr(struct thread_data *td, int goodf, |
1161 | int badf) | |
3d7c391d JA |
1162 | { |
1163 | unsigned int old_next_file = td->next_file; | |
1164 | struct fio_file *f; | |
1165 | ||
1166 | do { | |
87b10676 JA |
1167 | int opened = 0; |
1168 | ||
126d65c6 | 1169 | f = td->files[td->next_file]; |
3d7c391d JA |
1170 | |
1171 | td->next_file++; | |
2dc1bbeb | 1172 | if (td->next_file >= td->o.nr_files) |
3d7c391d JA |
1173 | td->next_file = 0; |
1174 | ||
87b10676 | 1175 | dprint(FD_FILE, "trying file %s %x\n", f->file_name, f->flags); |
d6aed795 | 1176 | if (fio_file_done(f)) { |
d5ed68ea | 1177 | f = NULL; |
059e63c0 | 1178 | continue; |
d5ed68ea | 1179 | } |
059e63c0 | 1180 | |
d6aed795 | 1181 | if (!fio_file_open(f)) { |
87b10676 JA |
1182 | int err; |
1183 | ||
002fe734 JA |
1184 | if (td->nr_open_files >= td->o.open_files) |
1185 | return ERR_PTR(-EBUSY); | |
1186 | ||
87b10676 | 1187 | err = td_io_open_file(td, f); |
b5696bfc JA |
1188 | if (err) { |
1189 | dprint(FD_FILE, "error %d on open of %s\n", | |
1190 | err, f->file_name); | |
87c27b45 | 1191 | f = NULL; |
87b10676 | 1192 | continue; |
b5696bfc | 1193 | } |
87b10676 JA |
1194 | opened = 1; |
1195 | } | |
1196 | ||
0b9d69ec JA |
1197 | dprint(FD_FILE, "goodf=%x, badf=%x, ff=%x\n", goodf, badf, |
1198 | f->flags); | |
1c178180 | 1199 | if ((!goodf || (f->flags & goodf)) && !(f->flags & badf)) |
3d7c391d JA |
1200 | break; |
1201 | ||
87b10676 JA |
1202 | if (opened) |
1203 | td_io_close_file(td, f); | |
1204 | ||
3d7c391d JA |
1205 | f = NULL; |
1206 | } while (td->next_file != old_next_file); | |
1207 | ||
2ba1c290 | 1208 | dprint(FD_FILE, "get_next_file_rr: %p\n", f); |
3d7c391d JA |
1209 | return f; |
1210 | } | |
1211 | ||
7eb36574 | 1212 | static struct fio_file *__get_next_file(struct thread_data *td) |
bdb4e2e9 | 1213 | { |
1907dbc6 JA |
1214 | struct fio_file *f; |
1215 | ||
2dc1bbeb | 1216 | assert(td->o.nr_files <= td->files_index); |
1c178180 | 1217 | |
b5696bfc | 1218 | if (td->nr_done_files >= td->o.nr_files) { |
5ec10eaa JA |
1219 | dprint(FD_FILE, "get_next_file: nr_open=%d, nr_done=%d," |
1220 | " nr_files=%d\n", td->nr_open_files, | |
1221 | td->nr_done_files, | |
1222 | td->o.nr_files); | |
bdb4e2e9 | 1223 | return NULL; |
2ba1c290 | 1224 | } |
bdb4e2e9 | 1225 | |
1907dbc6 | 1226 | f = td->file_service_file; |
d6aed795 | 1227 | if (f && fio_file_open(f) && !fio_file_closing(f)) { |
a086c257 JA |
1228 | if (td->o.file_service_type == FIO_FSERVICE_SEQ) |
1229 | goto out; | |
1230 | if (td->file_service_left--) | |
1231 | goto out; | |
1232 | } | |
1907dbc6 | 1233 | |
a086c257 JA |
1234 | if (td->o.file_service_type == FIO_FSERVICE_RR || |
1235 | td->o.file_service_type == FIO_FSERVICE_SEQ) | |
d6aed795 | 1236 | f = get_next_file_rr(td, FIO_FILE_open, FIO_FILE_closing); |
bdb4e2e9 | 1237 | else |
d6aed795 | 1238 | f = get_next_file_rand(td, FIO_FILE_open, FIO_FILE_closing); |
1907dbc6 | 1239 | |
002fe734 JA |
1240 | if (IS_ERR(f)) |
1241 | return f; | |
1242 | ||
1907dbc6 JA |
1243 | td->file_service_file = f; |
1244 | td->file_service_left = td->file_service_nr - 1; | |
2ba1c290 | 1245 | out: |
0dac421f JA |
1246 | if (f) |
1247 | dprint(FD_FILE, "get_next_file: %p [%s]\n", f, f->file_name); | |
1248 | else | |
1249 | dprint(FD_FILE, "get_next_file: NULL\n"); | |
1907dbc6 | 1250 | return f; |
bdb4e2e9 JA |
1251 | } |
1252 | ||
7eb36574 JA |
1253 | static struct fio_file *get_next_file(struct thread_data *td) |
1254 | { | |
372d8962 | 1255 | if (td->flags & TD_F_PROFILE_OPS) { |
d72be545 | 1256 | struct prof_io_ops *ops = &td->prof_io_ops; |
7eb36574 | 1257 | |
d72be545 JA |
1258 | if (ops->get_next_file) |
1259 | return ops->get_next_file(td); | |
1260 | } | |
7eb36574 JA |
1261 | |
1262 | return __get_next_file(td); | |
1263 | } | |
1264 | ||
002fe734 | 1265 | static long set_io_u_file(struct thread_data *td, struct io_u *io_u) |
429f6675 JA |
1266 | { |
1267 | struct fio_file *f; | |
1268 | ||
1269 | do { | |
1270 | f = get_next_file(td); | |
002fe734 JA |
1271 | if (IS_ERR_OR_NULL(f)) |
1272 | return PTR_ERR(f); | |
429f6675 | 1273 | |
429f6675 JA |
1274 | io_u->file = f; |
1275 | get_file(f); | |
1276 | ||
1277 | if (!fill_io_u(td, io_u)) | |
1278 | break; | |
1279 | ||
b5696bfc | 1280 | put_file_log(td, f); |
429f6675 | 1281 | td_io_close_file(td, f); |
b5696bfc | 1282 | io_u->file = NULL; |
8c07860d JA |
1283 | if (td->o.file_service_type & __FIO_FSERVICE_NONUNIFORM) |
1284 | fio_file_reset(td, f); | |
1285 | else { | |
1286 | fio_file_set_done(f); | |
1287 | td->nr_done_files++; | |
1288 | dprint(FD_FILE, "%s: is done (%d of %d)\n", f->file_name, | |
0b9d69ec | 1289 | td->nr_done_files, td->o.nr_files); |
8c07860d | 1290 | } |
429f6675 JA |
1291 | } while (1); |
1292 | ||
1293 | return 0; | |
1294 | } | |
1295 | ||
3e260a46 JA |
1296 | static void lat_fatal(struct thread_data *td, struct io_completion_data *icd, |
1297 | unsigned long tusec, unsigned long max_usec) | |
1298 | { | |
1299 | if (!td->error) | |
1300 | log_err("fio: latency of %lu usec exceeds specified max (%lu usec)\n", tusec, max_usec); | |
1301 | td_verror(td, ETIMEDOUT, "max latency exceeded"); | |
1302 | icd->error = ETIMEDOUT; | |
1303 | } | |
1304 | ||
1305 | static void lat_new_cycle(struct thread_data *td) | |
1306 | { | |
1307 | fio_gettime(&td->latency_ts, NULL); | |
1308 | td->latency_ios = ddir_rw_sum(td->io_blocks); | |
1309 | td->latency_failed = 0; | |
1310 | } | |
1311 | ||
1312 | /* | |
1313 | * We had an IO outside the latency target. Reduce the queue depth. If we | |
1314 | * are at QD=1, then it's time to give up. | |
1315 | */ | |
e39c0676 | 1316 | static bool __lat_target_failed(struct thread_data *td) |
3e260a46 JA |
1317 | { |
1318 | if (td->latency_qd == 1) | |
e39c0676 | 1319 | return true; |
3e260a46 JA |
1320 | |
1321 | td->latency_qd_high = td->latency_qd; | |
6bb58215 JA |
1322 | |
1323 | if (td->latency_qd == td->latency_qd_low) | |
1324 | td->latency_qd_low--; | |
1325 | ||
3e260a46 JA |
1326 | td->latency_qd = (td->latency_qd + td->latency_qd_low) / 2; |
1327 | ||
1328 | dprint(FD_RATE, "Ramped down: %d %d %d\n", td->latency_qd_low, td->latency_qd, td->latency_qd_high); | |
1329 | ||
1330 | /* | |
1331 | * When we ramp QD down, quiesce existing IO to prevent | |
1332 | * a storm of ramp downs due to pending higher depth. | |
1333 | */ | |
1334 | io_u_quiesce(td); | |
1335 | lat_new_cycle(td); | |
e39c0676 | 1336 | return false; |
3e260a46 JA |
1337 | } |
1338 | ||
e39c0676 | 1339 | static bool lat_target_failed(struct thread_data *td) |
3e260a46 JA |
1340 | { |
1341 | if (td->o.latency_percentile.u.f == 100.0) | |
1342 | return __lat_target_failed(td); | |
1343 | ||
1344 | td->latency_failed++; | |
e39c0676 | 1345 | return false; |
3e260a46 JA |
1346 | } |
1347 | ||
1348 | void lat_target_init(struct thread_data *td) | |
1349 | { | |
6bb58215 JA |
1350 | td->latency_end_run = 0; |
1351 | ||
3e260a46 JA |
1352 | if (td->o.latency_target) { |
1353 | dprint(FD_RATE, "Latency target=%llu\n", td->o.latency_target); | |
1354 | fio_gettime(&td->latency_ts, NULL); | |
1355 | td->latency_qd = 1; | |
1356 | td->latency_qd_high = td->o.iodepth; | |
1357 | td->latency_qd_low = 1; | |
1358 | td->latency_ios = ddir_rw_sum(td->io_blocks); | |
1359 | } else | |
1360 | td->latency_qd = td->o.iodepth; | |
1361 | } | |
1362 | ||
6bb58215 JA |
1363 | void lat_target_reset(struct thread_data *td) |
1364 | { | |
1365 | if (!td->latency_end_run) | |
1366 | lat_target_init(td); | |
1367 | } | |
1368 | ||
3e260a46 JA |
1369 | static void lat_target_success(struct thread_data *td) |
1370 | { | |
1371 | const unsigned int qd = td->latency_qd; | |
6bb58215 | 1372 | struct thread_options *o = &td->o; |
3e260a46 JA |
1373 | |
1374 | td->latency_qd_low = td->latency_qd; | |
1375 | ||
1376 | /* | |
1377 | * If we haven't failed yet, we double up to a failing value instead | |
1378 | * of bisecting from highest possible queue depth. If we have set | |
1379 | * a limit other than td->o.iodepth, bisect between that. | |
1380 | */ | |
6bb58215 | 1381 | if (td->latency_qd_high != o->iodepth) |
3e260a46 JA |
1382 | td->latency_qd = (td->latency_qd + td->latency_qd_high) / 2; |
1383 | else | |
1384 | td->latency_qd *= 2; | |
1385 | ||
6bb58215 JA |
1386 | if (td->latency_qd > o->iodepth) |
1387 | td->latency_qd = o->iodepth; | |
3e260a46 JA |
1388 | |
1389 | dprint(FD_RATE, "Ramped up: %d %d %d\n", td->latency_qd_low, td->latency_qd, td->latency_qd_high); | |
6bb58215 | 1390 | |
3e260a46 | 1391 | /* |
6bb58215 JA |
1392 | * Same as last one, we are done. Let it run a latency cycle, so |
1393 | * we get only the results from the targeted depth. | |
3e260a46 | 1394 | */ |
6bb58215 JA |
1395 | if (td->latency_qd == qd) { |
1396 | if (td->latency_end_run) { | |
1397 | dprint(FD_RATE, "We are done\n"); | |
1398 | td->done = 1; | |
1399 | } else { | |
1400 | dprint(FD_RATE, "Quiesce and final run\n"); | |
1401 | io_u_quiesce(td); | |
1402 | td->latency_end_run = 1; | |
1403 | reset_all_stats(td); | |
1404 | reset_io_stats(td); | |
1405 | } | |
1406 | } | |
3e260a46 JA |
1407 | |
1408 | lat_new_cycle(td); | |
1409 | } | |
1410 | ||
1411 | /* | |
1412 | * Check if we can bump the queue depth | |
1413 | */ | |
1414 | void lat_target_check(struct thread_data *td) | |
1415 | { | |
1416 | uint64_t usec_window; | |
1417 | uint64_t ios; | |
1418 | double success_ios; | |
1419 | ||
1420 | usec_window = utime_since_now(&td->latency_ts); | |
1421 | if (usec_window < td->o.latency_window) | |
1422 | return; | |
1423 | ||
1424 | ios = ddir_rw_sum(td->io_blocks) - td->latency_ios; | |
1425 | success_ios = (double) (ios - td->latency_failed) / (double) ios; | |
1426 | success_ios *= 100.0; | |
1427 | ||
1428 | dprint(FD_RATE, "Success rate: %.2f%% (target %.2f%%)\n", success_ios, td->o.latency_percentile.u.f); | |
1429 | ||
1430 | if (success_ios >= td->o.latency_percentile.u.f) | |
1431 | lat_target_success(td); | |
1432 | else | |
1433 | __lat_target_failed(td); | |
1434 | } | |
1435 | ||
1436 | /* | |
1437 | * If latency target is enabled, we might be ramping up or down and not | |
1438 | * using the full queue depth available. | |
1439 | */ | |
e39c0676 | 1440 | bool queue_full(const struct thread_data *td) |
3e260a46 JA |
1441 | { |
1442 | const int qempty = io_u_qempty(&td->io_u_freelist); | |
1443 | ||
1444 | if (qempty) | |
e39c0676 | 1445 | return true; |
3e260a46 | 1446 | if (!td->o.latency_target) |
e39c0676 | 1447 | return false; |
3e260a46 JA |
1448 | |
1449 | return td->cur_depth >= td->latency_qd; | |
1450 | } | |
429f6675 | 1451 | |
10ba535a JA |
1452 | struct io_u *__get_io_u(struct thread_data *td) |
1453 | { | |
0cae66f6 | 1454 | struct io_u *io_u = NULL; |
10ba535a | 1455 | |
ca09be4b JA |
1456 | if (td->stop_io) |
1457 | return NULL; | |
1458 | ||
e8462bd8 JA |
1459 | td_io_u_lock(td); |
1460 | ||
1461 | again: | |
2ae0b204 JA |
1462 | if (!io_u_rempty(&td->io_u_requeues)) |
1463 | io_u = io_u_rpop(&td->io_u_requeues); | |
3e260a46 | 1464 | else if (!queue_full(td)) { |
2ae0b204 | 1465 | io_u = io_u_qpop(&td->io_u_freelist); |
10ba535a | 1466 | |
225ba9e3 | 1467 | io_u->file = NULL; |
6040dabc | 1468 | io_u->buflen = 0; |
10ba535a | 1469 | io_u->resid = 0; |
d7762cf8 | 1470 | io_u->end_io = NULL; |
755200a3 JA |
1471 | } |
1472 | ||
1473 | if (io_u) { | |
0c6e7517 | 1474 | assert(io_u->flags & IO_U_F_FREE); |
1651e431 | 1475 | io_u_clear(td, io_u, IO_U_F_FREE | IO_U_F_NO_FILE_PUT | |
f8b0bd10 JA |
1476 | IO_U_F_TRIMMED | IO_U_F_BARRIER | |
1477 | IO_U_F_VER_LIST); | |
0c6e7517 | 1478 | |
755200a3 | 1479 | io_u->error = 0; |
bcd5abfa | 1480 | io_u->acct_ddir = -1; |
10ba535a | 1481 | td->cur_depth++; |
a9da8ab2 | 1482 | assert(!(td->flags & TD_F_CHILD)); |
1651e431 | 1483 | io_u_set(td, io_u, IO_U_F_IN_CUR_DEPTH); |
f9401285 | 1484 | io_u->ipo = NULL; |
a9da8ab2 | 1485 | } else if (td_async_processing(td)) { |
1dec3e07 JA |
1486 | /* |
1487 | * We ran out, wait for async verify threads to finish and | |
1488 | * return one | |
1489 | */ | |
a9da8ab2 JA |
1490 | assert(!(td->flags & TD_F_CHILD)); |
1491 | assert(!pthread_cond_wait(&td->free_cond, &td->io_u_lock)); | |
1dec3e07 | 1492 | goto again; |
10ba535a JA |
1493 | } |
1494 | ||
e8462bd8 | 1495 | td_io_u_unlock(td); |
10ba535a JA |
1496 | return io_u; |
1497 | } | |
1498 | ||
e39c0676 | 1499 | static bool check_get_trim(struct thread_data *td, struct io_u *io_u) |
10ba535a | 1500 | { |
d72be545 | 1501 | if (!(td->flags & TD_F_TRIM_BACKLOG)) |
e39c0676 | 1502 | return false; |
d72be545 JA |
1503 | |
1504 | if (td->trim_entries) { | |
0d29de83 | 1505 | int get_trim = 0; |
10ba535a | 1506 | |
0d29de83 JA |
1507 | if (td->trim_batch) { |
1508 | td->trim_batch--; | |
1509 | get_trim = 1; | |
1510 | } else if (!(td->io_hist_len % td->o.trim_backlog) && | |
1511 | td->last_ddir != DDIR_READ) { | |
1512 | td->trim_batch = td->o.trim_batch; | |
1513 | if (!td->trim_batch) | |
1514 | td->trim_batch = td->o.trim_backlog; | |
1515 | get_trim = 1; | |
1516 | } | |
1517 | ||
917c64b3 | 1518 | if (get_trim && get_next_trim(td, io_u)) |
e39c0676 | 1519 | return true; |
2ba1c290 | 1520 | } |
10ba535a | 1521 | |
e39c0676 | 1522 | return false; |
0d29de83 JA |
1523 | } |
1524 | ||
e39c0676 | 1525 | static bool check_get_verify(struct thread_data *td, struct io_u *io_u) |
0d29de83 | 1526 | { |
d72be545 | 1527 | if (!(td->flags & TD_F_VER_BACKLOG)) |
e39c0676 | 1528 | return false; |
d72be545 JA |
1529 | |
1530 | if (td->io_hist_len) { | |
9e144189 JA |
1531 | int get_verify = 0; |
1532 | ||
d1ece0c7 | 1533 | if (td->verify_batch) |
9e144189 | 1534 | get_verify = 1; |
d1ece0c7 | 1535 | else if (!(td->io_hist_len % td->o.verify_backlog) && |
9e144189 JA |
1536 | td->last_ddir != DDIR_READ) { |
1537 | td->verify_batch = td->o.verify_batch; | |
f8a75c99 JA |
1538 | if (!td->verify_batch) |
1539 | td->verify_batch = td->o.verify_backlog; | |
9e144189 JA |
1540 | get_verify = 1; |
1541 | } | |
1542 | ||
d1ece0c7 JA |
1543 | if (get_verify && !get_next_verify(td, io_u)) { |
1544 | td->verify_batch--; | |
e39c0676 | 1545 | return true; |
d1ece0c7 | 1546 | } |
9e144189 JA |
1547 | } |
1548 | ||
e39c0676 | 1549 | return false; |
0d29de83 JA |
1550 | } |
1551 | ||
de789769 JA |
1552 | /* |
1553 | * Fill offset and start time into the buffer content, to prevent too | |
23f394d5 JA |
1554 | * easy compressible data for simple de-dupe attempts. Do this for every |
1555 | * 512b block in the range, since that should be the smallest block size | |
1556 | * we can expect from a device. | |
de789769 JA |
1557 | */ |
1558 | static void small_content_scramble(struct io_u *io_u) | |
1559 | { | |
23f394d5 | 1560 | unsigned int i, nr_blocks = io_u->buflen / 512; |
1ae83d45 | 1561 | uint64_t boffset; |
23f394d5 JA |
1562 | unsigned int offset; |
1563 | void *p, *end; | |
de789769 | 1564 | |
23f394d5 JA |
1565 | if (!nr_blocks) |
1566 | return; | |
1567 | ||
1568 | p = io_u->xfer_buf; | |
fba76ee8 | 1569 | boffset = io_u->offset; |
81f0366c | 1570 | io_u->buf_filled_len = 0; |
fad82f76 | 1571 | |
23f394d5 JA |
1572 | for (i = 0; i < nr_blocks; i++) { |
1573 | /* | |
1574 | * Fill the byte offset into a "random" start offset of | |
1575 | * the buffer, given by the product of the usec time | |
1576 | * and the actual offset. | |
1577 | */ | |
fad82f76 | 1578 | offset = (io_u->start_time.tv_usec ^ boffset) & 511; |
1ae83d45 JA |
1579 | offset &= ~(sizeof(uint64_t) - 1); |
1580 | if (offset >= 512 - sizeof(uint64_t)) | |
1581 | offset -= sizeof(uint64_t); | |
fba76ee8 | 1582 | memcpy(p + offset, &boffset, sizeof(boffset)); |
23f394d5 JA |
1583 | |
1584 | end = p + 512 - sizeof(io_u->start_time); | |
1585 | memcpy(end, &io_u->start_time, sizeof(io_u->start_time)); | |
1586 | p += 512; | |
fad82f76 | 1587 | boffset += 512; |
23f394d5 | 1588 | } |
de789769 JA |
1589 | } |
1590 | ||
0d29de83 JA |
1591 | /* |
1592 | * Return an io_u to be processed. Gets a buflen and offset, sets direction, | |
1593 | * etc. The returned io_u is fully ready to be prepped and submitted. | |
1594 | */ | |
1595 | struct io_u *get_io_u(struct thread_data *td) | |
1596 | { | |
1597 | struct fio_file *f; | |
1598 | struct io_u *io_u; | |
de789769 | 1599 | int do_scramble = 0; |
002fe734 | 1600 | long ret = 0; |
0d29de83 JA |
1601 | |
1602 | io_u = __get_io_u(td); | |
1603 | if (!io_u) { | |
1604 | dprint(FD_IO, "__get_io_u failed\n"); | |
1605 | return NULL; | |
1606 | } | |
1607 | ||
1608 | if (check_get_verify(td, io_u)) | |
1609 | goto out; | |
1610 | if (check_get_trim(td, io_u)) | |
1611 | goto out; | |
1612 | ||
755200a3 JA |
1613 | /* |
1614 | * from a requeue, io_u already setup | |
1615 | */ | |
1616 | if (io_u->file) | |
77f392bf | 1617 | goto out; |
755200a3 | 1618 | |
429f6675 JA |
1619 | /* |
1620 | * If using an iolog, grab next piece if any available. | |
1621 | */ | |
d72be545 | 1622 | if (td->flags & TD_F_READ_IOLOG) { |
429f6675 JA |
1623 | if (read_iolog_get(td, io_u)) |
1624 | goto err_put; | |
2ba1c290 | 1625 | } else if (set_io_u_file(td, io_u)) { |
002fe734 | 1626 | ret = -EBUSY; |
2ba1c290 | 1627 | dprint(FD_IO, "io_u %p, setting file failed\n", io_u); |
429f6675 | 1628 | goto err_put; |
2ba1c290 | 1629 | } |
5ec10eaa | 1630 | |
429f6675 | 1631 | f = io_u->file; |
002fe734 JA |
1632 | if (!f) { |
1633 | dprint(FD_IO, "io_u %p, setting file failed\n", io_u); | |
1634 | goto err_put; | |
1635 | } | |
1636 | ||
d6aed795 | 1637 | assert(fio_file_open(f)); |
97af62ce | 1638 | |
ff58fced | 1639 | if (ddir_rw(io_u->ddir)) { |
9b87f09b | 1640 | if (!io_u->buflen && !td_ioengine_flagged(td, FIO_NOIO)) { |
2ba1c290 | 1641 | dprint(FD_IO, "get_io_u: zero buflen on %p\n", io_u); |
429f6675 | 1642 | goto err_put; |
2ba1c290 | 1643 | } |
10ba535a | 1644 | |
f1dfb668 JA |
1645 | f->last_start[io_u->ddir] = io_u->offset; |
1646 | f->last_pos[io_u->ddir] = io_u->offset + io_u->buflen; | |
10ba535a | 1647 | |
fd68418e | 1648 | if (io_u->ddir == DDIR_WRITE) { |
d72be545 | 1649 | if (td->flags & TD_F_REFILL_BUFFERS) { |
9c42684e | 1650 | io_u_fill_buffer(td, io_u, |
1066358a | 1651 | td->o.min_bs[DDIR_WRITE], |
9e129577 | 1652 | io_u->buflen); |
ff441ae8 | 1653 | } else if ((td->flags & TD_F_SCRAMBLE_BUFFERS) && |
bedc9dc2 | 1654 | !(td->flags & TD_F_COMPRESS)) |
fd68418e | 1655 | do_scramble = 1; |
d72be545 | 1656 | if (td->flags & TD_F_VER_NONE) { |
629f1d71 JA |
1657 | populate_verify_io_u(td, io_u); |
1658 | do_scramble = 0; | |
1659 | } | |
fd68418e | 1660 | } else if (io_u->ddir == DDIR_READ) { |
cbe8d756 RR |
1661 | /* |
1662 | * Reset the buf_filled parameters so next time if the | |
1663 | * buffer is used for writes it is refilled. | |
1664 | */ | |
cbe8d756 RR |
1665 | io_u->buf_filled_len = 0; |
1666 | } | |
87dc1ab1 | 1667 | } |
10ba535a | 1668 | |
165faf16 JA |
1669 | /* |
1670 | * Set io data pointers. | |
1671 | */ | |
cec6b55d JA |
1672 | io_u->xfer_buf = io_u->buf; |
1673 | io_u->xfer_buflen = io_u->buflen; | |
5973cafb | 1674 | |
6ac7a331 | 1675 | out: |
0d29de83 | 1676 | assert(io_u->file); |
429f6675 | 1677 | if (!td_io_prep(td, io_u)) { |
3ecc8c67 | 1678 | if (!td->o.disable_lat) |
993bf48b | 1679 | fio_gettime(&io_u->start_time, NULL); |
de789769 JA |
1680 | if (do_scramble) |
1681 | small_content_scramble(io_u); | |
429f6675 | 1682 | return io_u; |
36167d82 | 1683 | } |
429f6675 | 1684 | err_put: |
2ba1c290 | 1685 | dprint(FD_IO, "get_io_u failed\n"); |
429f6675 | 1686 | put_io_u(td, io_u); |
002fe734 | 1687 | return ERR_PTR(ret); |
10ba535a JA |
1688 | } |
1689 | ||
a9da8ab2 | 1690 | static void __io_u_log_error(struct thread_data *td, struct io_u *io_u) |
5451792e | 1691 | { |
8b28bd41 | 1692 | enum error_type_bit eb = td_error_type(io_u->ddir, io_u->error); |
825f818e | 1693 | |
8b28bd41 DM |
1694 | if (td_non_fatal_error(td, eb, io_u->error) && !td->o.error_dump) |
1695 | return; | |
5451792e | 1696 | |
709c8313 RE |
1697 | log_err("fio: io_u error%s%s: %s: %s offset=%llu, buflen=%lu\n", |
1698 | io_u->file ? " on file " : "", | |
1699 | io_u->file ? io_u->file->file_name : "", | |
1700 | strerror(io_u->error), | |
1701 | io_ddir_name(io_u->ddir), | |
1702 | io_u->offset, io_u->xfer_buflen); | |
5451792e | 1703 | |
5ad7be56 KD |
1704 | if (td->io_ops->errdetails) { |
1705 | char *err = td->io_ops->errdetails(io_u); | |
1706 | ||
1707 | log_err("fio: %s\n", err); | |
1708 | free(err); | |
1709 | } | |
1710 | ||
5451792e JA |
1711 | if (!td->error) |
1712 | td_verror(td, io_u->error, "io_u error"); | |
1713 | } | |
1714 | ||
a9da8ab2 JA |
1715 | void io_u_log_error(struct thread_data *td, struct io_u *io_u) |
1716 | { | |
1717 | __io_u_log_error(td, io_u); | |
1718 | if (td->parent) | |
094e66cb | 1719 | __io_u_log_error(td->parent, io_u); |
a9da8ab2 JA |
1720 | } |
1721 | ||
e39c0676 | 1722 | static inline bool gtod_reduce(struct thread_data *td) |
aba6c951 | 1723 | { |
3ecc8c67 JA |
1724 | return (td->o.disable_clat && td->o.disable_slat && td->o.disable_bw) |
1725 | || td->o.gtod_reduce; | |
aba6c951 JA |
1726 | } |
1727 | ||
c8eeb9df JA |
1728 | static void account_io_completion(struct thread_data *td, struct io_u *io_u, |
1729 | struct io_completion_data *icd, | |
1730 | const enum fio_ddir idx, unsigned int bytes) | |
1731 | { | |
a9da8ab2 | 1732 | const int no_reduce = !gtod_reduce(td); |
24d23ca7 | 1733 | unsigned long lusec = 0; |
c8eeb9df | 1734 | |
75dc383e JA |
1735 | if (td->parent) |
1736 | td = td->parent; | |
1737 | ||
a9da8ab2 | 1738 | if (no_reduce) |
c8eeb9df JA |
1739 | lusec = utime_since(&io_u->issue_time, &icd->time); |
1740 | ||
1741 | if (!td->o.disable_lat) { | |
1742 | unsigned long tusec; | |
1743 | ||
1744 | tusec = utime_since(&io_u->start_time, &icd->time); | |
ae588852 | 1745 | add_lat_sample(td, idx, tusec, bytes, io_u->offset); |
15501535 | 1746 | |
d4afedfd JA |
1747 | if (td->flags & TD_F_PROFILE_OPS) { |
1748 | struct prof_io_ops *ops = &td->prof_io_ops; | |
1749 | ||
1750 | if (ops->io_u_lat) | |
1751 | icd->error = ops->io_u_lat(td, tusec); | |
1752 | } | |
1753 | ||
3e260a46 JA |
1754 | if (td->o.max_latency && tusec > td->o.max_latency) |
1755 | lat_fatal(td, icd, tusec, td->o.max_latency); | |
1756 | if (td->o.latency_target && tusec > td->o.latency_target) { | |
1757 | if (lat_target_failed(td)) | |
1758 | lat_fatal(td, icd, tusec, td->o.latency_target); | |
15501535 | 1759 | } |
c8eeb9df JA |
1760 | } |
1761 | ||
a47591e4 JA |
1762 | if (ddir_rw(idx)) { |
1763 | if (!td->o.disable_clat) { | |
1764 | add_clat_sample(td, idx, lusec, bytes, io_u->offset); | |
1765 | io_u_mark_latency(td, lusec); | |
1766 | } | |
c8eeb9df | 1767 | |
a47591e4 JA |
1768 | if (!td->o.disable_bw && per_unit_log(td->bw_log)) |
1769 | add_bw_sample(td, io_u, bytes, lusec); | |
c8eeb9df | 1770 | |
a47591e4 JA |
1771 | if (no_reduce && per_unit_log(td->iops_log)) |
1772 | add_iops_sample(td, io_u, bytes); | |
1773 | } | |
66347cfa DE |
1774 | |
1775 | if (td->ts.nr_block_infos && io_u->ddir == DDIR_TRIM) { | |
1776 | uint32_t *info = io_u_block_info(td, io_u); | |
1777 | if (BLOCK_INFO_STATE(*info) < BLOCK_STATE_TRIM_FAILURE) { | |
1778 | if (io_u->ddir == DDIR_TRIM) { | |
1779 | *info = BLOCK_INFO(BLOCK_STATE_TRIMMED, | |
1780 | BLOCK_INFO_TRIMS(*info) + 1); | |
1781 | } else if (io_u->ddir == DDIR_WRITE) { | |
1782 | *info = BLOCK_INFO_SET_STATE(BLOCK_STATE_WRITTEN, | |
1783 | *info); | |
1784 | } | |
1785 | } | |
1786 | } | |
c8eeb9df JA |
1787 | } |
1788 | ||
94a6e1bb JA |
1789 | static void file_log_write_comp(const struct thread_data *td, struct fio_file *f, |
1790 | uint64_t offset, unsigned int bytes) | |
1791 | { | |
1792 | int idx; | |
1793 | ||
639ad1ea JA |
1794 | if (!f) |
1795 | return; | |
1796 | ||
94a6e1bb JA |
1797 | if (f->first_write == -1ULL || offset < f->first_write) |
1798 | f->first_write = offset; | |
1799 | if (f->last_write == -1ULL || ((offset + bytes) > f->last_write)) | |
1800 | f->last_write = offset + bytes; | |
1801 | ||
1802 | if (!f->last_write_comp) | |
1803 | return; | |
1804 | ||
1805 | idx = f->last_write_idx++; | |
1806 | f->last_write_comp[idx] = offset; | |
1807 | if (f->last_write_idx == td->o.iodepth) | |
1808 | f->last_write_idx = 0; | |
1809 | } | |
1810 | ||
f8b0bd10 | 1811 | static void io_completed(struct thread_data *td, struct io_u **io_u_ptr, |
97601024 | 1812 | struct io_completion_data *icd) |
10ba535a | 1813 | { |
f8b0bd10 JA |
1814 | struct io_u *io_u = *io_u_ptr; |
1815 | enum fio_ddir ddir = io_u->ddir; | |
1816 | struct fio_file *f = io_u->file; | |
10ba535a | 1817 | |
2ba1c290 JA |
1818 | dprint_io_u(io_u, "io complete"); |
1819 | ||
0c6e7517 | 1820 | assert(io_u->flags & IO_U_F_FLIGHT); |
1651e431 | 1821 | io_u_clear(td, io_u, IO_U_F_FLIGHT | IO_U_F_BUSY_OK); |
f9401285 JA |
1822 | |
1823 | /* | |
1824 | * Mark IO ok to verify | |
1825 | */ | |
1826 | if (io_u->ipo) { | |
890b6656 JA |
1827 | /* |
1828 | * Remove errored entry from the verification list | |
1829 | */ | |
1830 | if (io_u->error) | |
1831 | unlog_io_piece(td, io_u); | |
1832 | else { | |
1833 | io_u->ipo->flags &= ~IP_F_IN_FLIGHT; | |
1834 | write_barrier(); | |
1835 | } | |
f9401285 JA |
1836 | } |
1837 | ||
f8b0bd10 | 1838 | if (ddir_sync(ddir)) { |
87dc1ab1 | 1839 | td->last_was_sync = 1; |
44f29692 JA |
1840 | if (f) { |
1841 | f->first_write = -1ULL; | |
1842 | f->last_write = -1ULL; | |
1843 | } | |
87dc1ab1 JA |
1844 | return; |
1845 | } | |
1846 | ||
1847 | td->last_was_sync = 0; | |
f8b0bd10 | 1848 | td->last_ddir = ddir; |
87dc1ab1 | 1849 | |
f8b0bd10 | 1850 | if (!io_u->error && ddir_rw(ddir)) { |
10ba535a | 1851 | unsigned int bytes = io_u->buflen - io_u->resid; |
b29ee5b3 | 1852 | int ret; |
10ba535a | 1853 | |
f8b0bd10 JA |
1854 | td->io_blocks[ddir]++; |
1855 | td->this_io_blocks[ddir]++; | |
1856 | td->io_bytes[ddir] += bytes; | |
ae2fafc8 | 1857 | |
1858 | if (!(io_u->flags & IO_U_F_VER_LIST)) | |
f8b0bd10 JA |
1859 | td->this_io_bytes[ddir] += bytes; |
1860 | ||
639ad1ea | 1861 | if (ddir == DDIR_WRITE) |
94a6e1bb | 1862 | file_log_write_comp(td, f, io_u->offset, bytes); |
44f29692 | 1863 | |
6b1190fd | 1864 | if (ramp_time_over(td) && (td->runstate == TD_RUNNING || |
50a8ce86 | 1865 | td->runstate == TD_VERIFYING)) |
f8b0bd10 | 1866 | account_io_completion(td, io_u, icd, ddir, bytes); |
40e1a6f0 | 1867 | |
f8b0bd10 | 1868 | icd->bytes_done[ddir] += bytes; |
3af6ef39 | 1869 | |
d7762cf8 | 1870 | if (io_u->end_io) { |
f8b0bd10 JA |
1871 | ret = io_u->end_io(td, io_u_ptr); |
1872 | io_u = *io_u_ptr; | |
3af6ef39 JA |
1873 | if (ret && !icd->error) |
1874 | icd->error = ret; | |
1875 | } | |
ff58fced | 1876 | } else if (io_u->error) { |
10ba535a | 1877 | icd->error = io_u->error; |
5451792e JA |
1878 | io_u_log_error(td, io_u); |
1879 | } | |
8b28bd41 | 1880 | if (icd->error) { |
f8b0bd10 JA |
1881 | enum error_type_bit eb = td_error_type(ddir, icd->error); |
1882 | ||
8b28bd41 DM |
1883 | if (!td_non_fatal_error(td, eb, icd->error)) |
1884 | return; | |
f8b0bd10 | 1885 | |
f2bba182 RR |
1886 | /* |
1887 | * If there is a non_fatal error, then add to the error count | |
1888 | * and clear all the errors. | |
1889 | */ | |
1890 | update_error_count(td, icd->error); | |
1891 | td_clear_error(td); | |
1892 | icd->error = 0; | |
f8b0bd10 JA |
1893 | if (io_u) |
1894 | io_u->error = 0; | |
f2bba182 | 1895 | } |
10ba535a JA |
1896 | } |
1897 | ||
9520ebb9 JA |
1898 | static void init_icd(struct thread_data *td, struct io_completion_data *icd, |
1899 | int nr) | |
10ba535a | 1900 | { |
6eaf09d6 | 1901 | int ddir; |
aba6c951 JA |
1902 | |
1903 | if (!gtod_reduce(td)) | |
9520ebb9 | 1904 | fio_gettime(&icd->time, NULL); |
02bcaa8c | 1905 | |
3af6ef39 JA |
1906 | icd->nr = nr; |
1907 | ||
10ba535a | 1908 | icd->error = 0; |
6eaf09d6 SL |
1909 | for (ddir = DDIR_READ; ddir < DDIR_RWDIR_CNT; ddir++) |
1910 | icd->bytes_done[ddir] = 0; | |
36167d82 JA |
1911 | } |
1912 | ||
97601024 JA |
1913 | static void ios_completed(struct thread_data *td, |
1914 | struct io_completion_data *icd) | |
36167d82 JA |
1915 | { |
1916 | struct io_u *io_u; | |
1917 | int i; | |
1918 | ||
10ba535a JA |
1919 | for (i = 0; i < icd->nr; i++) { |
1920 | io_u = td->io_ops->event(td, i); | |
1921 | ||
f8b0bd10 | 1922 | io_completed(td, &io_u, icd); |
e8462bd8 | 1923 | |
f8b0bd10 | 1924 | if (io_u) |
e8462bd8 | 1925 | put_io_u(td, io_u); |
10ba535a JA |
1926 | } |
1927 | } | |
97601024 | 1928 | |
e7e6cfb4 JA |
1929 | /* |
1930 | * Complete a single io_u for the sync engines. | |
1931 | */ | |
55312f9f | 1932 | int io_u_sync_complete(struct thread_data *td, struct io_u *io_u) |
97601024 JA |
1933 | { |
1934 | struct io_completion_data icd; | |
55312f9f | 1935 | int ddir; |
97601024 | 1936 | |
9520ebb9 | 1937 | init_icd(td, &icd, 1); |
f8b0bd10 | 1938 | io_completed(td, &io_u, &icd); |
e8462bd8 | 1939 | |
f8b0bd10 | 1940 | if (io_u) |
e8462bd8 | 1941 | put_io_u(td, io_u); |
97601024 | 1942 | |
581e7141 JA |
1943 | if (icd.error) { |
1944 | td_verror(td, icd.error, "io_u_sync_complete"); | |
1945 | return -1; | |
1946 | } | |
97601024 | 1947 | |
55312f9f JA |
1948 | for (ddir = DDIR_READ; ddir < DDIR_RWDIR_CNT; ddir++) |
1949 | td->bytes_done[ddir] += icd.bytes_done[ddir]; | |
581e7141 JA |
1950 | |
1951 | return 0; | |
97601024 JA |
1952 | } |
1953 | ||
e7e6cfb4 JA |
1954 | /* |
1955 | * Called to complete min_events number of io for the async engines. | |
1956 | */ | |
55312f9f | 1957 | int io_u_queued_complete(struct thread_data *td, int min_evts) |
97601024 | 1958 | { |
97601024 | 1959 | struct io_completion_data icd; |
00de55ef | 1960 | struct timespec *tvp = NULL; |
55312f9f | 1961 | int ret, ddir; |
4d06a338 | 1962 | struct timespec ts = { .tv_sec = 0, .tv_nsec = 0, }; |
97601024 | 1963 | |
4950421a | 1964 | dprint(FD_IO, "io_u_queued_completed: min=%d\n", min_evts); |
b271fe62 | 1965 | |
4950421a | 1966 | if (!min_evts) |
00de55ef | 1967 | tvp = &ts; |
5fb4b366 RE |
1968 | else if (min_evts > td->cur_depth) |
1969 | min_evts = td->cur_depth; | |
97601024 | 1970 | |
82407585 RP |
1971 | /* No worries, td_io_getevents fixes min and max if they are |
1972 | * set incorrectly */ | |
1973 | ret = td_io_getevents(td, min_evts, td->o.iodepth_batch_complete_max, tvp); | |
97601024 | 1974 | if (ret < 0) { |
e1161c32 | 1975 | td_verror(td, -ret, "td_io_getevents"); |
97601024 JA |
1976 | return ret; |
1977 | } else if (!ret) | |
1978 | return ret; | |
1979 | ||
9520ebb9 | 1980 | init_icd(td, &icd, ret); |
97601024 | 1981 | ios_completed(td, &icd); |
581e7141 JA |
1982 | if (icd.error) { |
1983 | td_verror(td, icd.error, "io_u_queued_complete"); | |
1984 | return -1; | |
1985 | } | |
97601024 | 1986 | |
55312f9f JA |
1987 | for (ddir = DDIR_READ; ddir < DDIR_RWDIR_CNT; ddir++) |
1988 | td->bytes_done[ddir] += icd.bytes_done[ddir]; | |
581e7141 | 1989 | |
0d593542 | 1990 | return ret; |
97601024 | 1991 | } |
7e77dd02 JA |
1992 | |
1993 | /* | |
1994 | * Call when io_u is really queued, to update the submission latency. | |
1995 | */ | |
1996 | void io_u_queued(struct thread_data *td, struct io_u *io_u) | |
1997 | { | |
9520ebb9 JA |
1998 | if (!td->o.disable_slat) { |
1999 | unsigned long slat_time; | |
7e77dd02 | 2000 | |
9520ebb9 | 2001 | slat_time = utime_since(&io_u->start_time, &io_u->issue_time); |
75dc383e JA |
2002 | |
2003 | if (td->parent) | |
2004 | td = td->parent; | |
2005 | ||
ae588852 JA |
2006 | add_slat_sample(td, io_u->ddir, slat_time, io_u->xfer_buflen, |
2007 | io_u->offset); | |
9520ebb9 | 2008 | } |
7e77dd02 | 2009 | } |
433afcb4 | 2010 | |
5c94b008 JA |
2011 | /* |
2012 | * See if we should reuse the last seed, if dedupe is enabled | |
2013 | */ | |
9451b93e | 2014 | static struct frand_state *get_buf_state(struct thread_data *td) |
5c94b008 JA |
2015 | { |
2016 | unsigned int v; | |
5c94b008 JA |
2017 | |
2018 | if (!td->o.dedupe_percentage) | |
2019 | return &td->buf_state; | |
732eedd0 | 2020 | else if (td->o.dedupe_percentage == 100) { |
9451b93e JA |
2021 | frand_copy(&td->buf_state_prev, &td->buf_state); |
2022 | return &td->buf_state; | |
732eedd0 | 2023 | } |
5c94b008 | 2024 | |
36dd3379 | 2025 | v = rand32_between(&td->dedupe_state, 1, 100); |
5c94b008 JA |
2026 | |
2027 | if (v <= td->o.dedupe_percentage) | |
2028 | return &td->buf_state_prev; | |
2029 | ||
2030 | return &td->buf_state; | |
2031 | } | |
2032 | ||
9451b93e | 2033 | static void save_buf_state(struct thread_data *td, struct frand_state *rs) |
5c94b008 | 2034 | { |
9451b93e JA |
2035 | if (td->o.dedupe_percentage == 100) |
2036 | frand_copy(rs, &td->buf_state_prev); | |
2037 | else if (rs == &td->buf_state) | |
5c94b008 JA |
2038 | frand_copy(&td->buf_state_prev, rs); |
2039 | } | |
2040 | ||
cc86c395 JA |
2041 | void fill_io_buffer(struct thread_data *td, void *buf, unsigned int min_write, |
2042 | unsigned int max_bs) | |
5973cafb | 2043 | { |
d1af2894 JA |
2044 | struct thread_options *o = &td->o; |
2045 | ||
4eff3e57 | 2046 | if (o->compress_percentage || o->dedupe_percentage) { |
9c42684e | 2047 | unsigned int perc = td->o.compress_percentage; |
5c94b008 | 2048 | struct frand_state *rs; |
1066358a | 2049 | unsigned int left = max_bs; |
1e7f82e2 | 2050 | unsigned int this_write; |
5c94b008 | 2051 | |
1066358a | 2052 | do { |
9451b93e | 2053 | rs = get_buf_state(td); |
9c42684e | 2054 | |
1066358a | 2055 | min_write = min(min_write, left); |
f97a43a1 | 2056 | |
1066358a | 2057 | if (perc) { |
4b157ac6 JA |
2058 | this_write = min_not_zero(min_write, |
2059 | td->o.compress_chunk); | |
1e7f82e2 JA |
2060 | |
2061 | fill_random_buf_percentage(rs, buf, perc, | |
2062 | this_write, this_write, | |
2063 | o->buffer_pattern, | |
2064 | o->buffer_pattern_bytes); | |
2065 | } else { | |
1066358a | 2066 | fill_random_buf(rs, buf, min_write); |
1e7f82e2 JA |
2067 | this_write = min_write; |
2068 | } | |
1066358a | 2069 | |
1e7f82e2 JA |
2070 | buf += this_write; |
2071 | left -= this_write; | |
9451b93e | 2072 | save_buf_state(td, rs); |
1066358a | 2073 | } while (left); |
d1af2894 JA |
2074 | } else if (o->buffer_pattern_bytes) |
2075 | fill_buffer_pattern(td, buf, max_bs); | |
999d245e | 2076 | else if (o->zero_buffers) |
cc86c395 | 2077 | memset(buf, 0, max_bs); |
999d245e | 2078 | else |
9451b93e | 2079 | fill_random_buf(get_buf_state(td), buf, max_bs); |
cc86c395 JA |
2080 | } |
2081 | ||
2082 | /* | |
2083 | * "randomly" fill the buffer contents | |
2084 | */ | |
2085 | void io_u_fill_buffer(struct thread_data *td, struct io_u *io_u, | |
2086 | unsigned int min_write, unsigned int max_bs) | |
2087 | { | |
2088 | io_u->buf_filled_len = 0; | |
2089 | fill_io_buffer(td, io_u->buf, min_write, max_bs); | |
5973cafb | 2090 | } |