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; |
c3546b53 | 534 | uint64_t frand_max; |
54a21917 | 535 | unsigned long r; |
10ba535a | 536 | |
9ee1c647 | 537 | assert(ddir_rw(ddir)); |
6aca9b3d JA |
538 | |
539 | if (td->o.bs_is_seq_rand) | |
540 | ddir = is_random ? DDIR_WRITE: DDIR_READ; | |
ff58fced | 541 | |
f3059de1 JA |
542 | minbs = td->o.min_bs[ddir]; |
543 | maxbs = td->o.max_bs[ddir]; | |
544 | ||
79944128 JA |
545 | if (minbs == maxbs) |
546 | return minbs; | |
547 | ||
52c58027 JA |
548 | /* |
549 | * If we can't satisfy the min block size from here, then fail | |
550 | */ | |
551 | if (!io_u_fits(td, io_u, minbs)) | |
552 | return 0; | |
553 | ||
c3546b53 | 554 | frand_max = rand_max(&td->bsrange_state); |
79944128 | 555 | do { |
d6b72507 | 556 | r = __rand(&td->bsrange_state); |
4c07ad86 | 557 | |
720e84ad | 558 | if (!td->o.bssplit_nr[ddir]) { |
f3059de1 | 559 | buflen = 1 + (unsigned int) ((double) maxbs * |
c3546b53 | 560 | (r / (frand_max + 1.0))); |
f3059de1 JA |
561 | if (buflen < minbs) |
562 | buflen = minbs; | |
5ec10eaa | 563 | } else { |
564ca972 JA |
564 | long perc = 0; |
565 | unsigned int i; | |
566 | ||
720e84ad JA |
567 | for (i = 0; i < td->o.bssplit_nr[ddir]; i++) { |
568 | struct bssplit *bsp = &td->o.bssplit[ddir][i]; | |
564ca972 JA |
569 | |
570 | buflen = bsp->bs; | |
571 | perc += bsp->perc; | |
1ec4db15 | 572 | if ((r * 100UL <= frand_max * perc) && |
79944128 | 573 | io_u_fits(td, io_u, buflen)) |
564ca972 JA |
574 | break; |
575 | } | |
576 | } | |
79944128 | 577 | |
ced5db25 | 578 | if (td->o.verify != VERIFY_NONE) |
a9f70b1f JB |
579 | buflen = (buflen + td->o.verify_interval - 1) & |
580 | ~(td->o.verify_interval - 1); | |
581 | ||
f3059de1 | 582 | if (!td->o.bs_unaligned && is_power_of_2(minbs)) |
7c306bb1 | 583 | buflen &= ~(minbs - 1); |
10ba535a | 584 | |
79944128 | 585 | } while (!io_u_fits(td, io_u, buflen)); |
6a5e6884 | 586 | |
10ba535a JA |
587 | return buflen; |
588 | } | |
589 | ||
6aca9b3d JA |
590 | static unsigned int get_next_buflen(struct thread_data *td, struct io_u *io_u, |
591 | unsigned int is_random) | |
15dc1934 | 592 | { |
d72be545 JA |
593 | if (td->flags & TD_F_PROFILE_OPS) { |
594 | struct prof_io_ops *ops = &td->prof_io_ops; | |
7eb36574 | 595 | |
d72be545 | 596 | if (ops->fill_io_u_size) |
6aca9b3d | 597 | return ops->fill_io_u_size(td, io_u, is_random); |
d72be545 | 598 | } |
15dc1934 | 599 | |
6aca9b3d | 600 | return __get_next_buflen(td, io_u, is_random); |
15dc1934 JA |
601 | } |
602 | ||
afe24a5a JA |
603 | static void set_rwmix_bytes(struct thread_data *td) |
604 | { | |
afe24a5a JA |
605 | unsigned int diff; |
606 | ||
607 | /* | |
608 | * we do time or byte based switch. this is needed because | |
609 | * buffered writes may issue a lot quicker than they complete, | |
610 | * whereas reads do not. | |
611 | */ | |
e47f799f | 612 | diff = td->o.rwmix[td->rwmix_ddir ^ 1]; |
04c540d9 | 613 | td->rwmix_issues = (td->io_issues[td->rwmix_ddir] * diff) / 100; |
e47f799f JA |
614 | } |
615 | ||
616 | static inline enum fio_ddir get_rand_ddir(struct thread_data *td) | |
617 | { | |
618 | unsigned int v; | |
e47f799f | 619 | |
36dd3379 | 620 | v = rand32_between(&td->rwmix_state, 1, 100); |
4c07ad86 | 621 | |
04c540d9 | 622 | if (v <= td->o.rwmix[DDIR_READ]) |
e47f799f JA |
623 | return DDIR_READ; |
624 | ||
625 | return DDIR_WRITE; | |
afe24a5a JA |
626 | } |
627 | ||
0d593542 | 628 | int io_u_quiesce(struct thread_data *td) |
002e7183 | 629 | { |
0d593542 JA |
630 | int completed = 0; |
631 | ||
002e7183 JA |
632 | /* |
633 | * We are going to sleep, ensure that we flush anything pending as | |
634 | * not to skew our latency numbers. | |
635 | * | |
636 | * Changed to only monitor 'in flight' requests here instead of the | |
637 | * td->cur_depth, b/c td->cur_depth does not accurately represent | |
638 | * io's that have been actually submitted to an async engine, | |
639 | * and cur_depth is meaningless for sync engines. | |
640 | */ | |
9cc80b6d JA |
641 | if (td->io_u_queued || td->cur_depth) { |
642 | int fio_unused ret; | |
643 | ||
644 | ret = td_io_commit(td); | |
645 | } | |
646 | ||
002e7183 JA |
647 | while (td->io_u_in_flight) { |
648 | int fio_unused ret; | |
649 | ||
55312f9f | 650 | ret = io_u_queued_complete(td, 1); |
0d593542 JA |
651 | if (ret > 0) |
652 | completed += ret; | |
002e7183 | 653 | } |
0d593542 JA |
654 | |
655 | return completed; | |
002e7183 JA |
656 | } |
657 | ||
581e7141 JA |
658 | static enum fio_ddir rate_ddir(struct thread_data *td, enum fio_ddir ddir) |
659 | { | |
660 | enum fio_ddir odir = ddir ^ 1; | |
50a8ce86 | 661 | long usec, now; |
581e7141 | 662 | |
ff58fced | 663 | assert(ddir_rw(ddir)); |
50a8ce86 | 664 | now = utime_since_now(&td->start); |
ff58fced | 665 | |
50a8ce86 D |
666 | /* |
667 | * if rate_next_io_time is in the past, need to catch up to rate | |
668 | */ | |
669 | if (td->rate_next_io_time[ddir] <= now) | |
581e7141 JA |
670 | return ddir; |
671 | ||
672 | /* | |
50a8ce86 | 673 | * We are ahead of rate in this direction. See if we |
581e7141 JA |
674 | * should switch. |
675 | */ | |
315fcfec | 676 | if (td_rw(td) && td->o.rwmix[odir]) { |
581e7141 | 677 | /* |
50a8ce86 | 678 | * Other direction is behind rate, switch |
581e7141 | 679 | */ |
50a8ce86 | 680 | if (td->rate_next_io_time[odir] <= now) |
581e7141 JA |
681 | return odir; |
682 | ||
683 | /* | |
50a8ce86 D |
684 | * Both directions are ahead of rate. sleep the min |
685 | * switch if necissary | |
581e7141 | 686 | */ |
50a8ce86 D |
687 | if (td->rate_next_io_time[ddir] <= |
688 | td->rate_next_io_time[odir]) { | |
689 | usec = td->rate_next_io_time[ddir] - now; | |
581e7141 | 690 | } else { |
50a8ce86 | 691 | usec = td->rate_next_io_time[odir] - now; |
581e7141 JA |
692 | ddir = odir; |
693 | } | |
694 | } else | |
50a8ce86 | 695 | usec = td->rate_next_io_time[ddir] - now; |
581e7141 | 696 | |
a9da8ab2 JA |
697 | if (td->o.io_submit_mode == IO_MODE_INLINE) |
698 | io_u_quiesce(td); | |
78c1eda5 | 699 | |
9cc80b6d | 700 | usec = usec_sleep(td, usec); |
581e7141 | 701 | |
581e7141 JA |
702 | return ddir; |
703 | } | |
704 | ||
10ba535a JA |
705 | /* |
706 | * Return the data direction for the next io_u. If the job is a | |
707 | * mixed read/write workload, check the rwmix cycle and switch if | |
708 | * necessary. | |
709 | */ | |
1e97cce9 | 710 | static enum fio_ddir get_rw_ddir(struct thread_data *td) |
10ba535a | 711 | { |
581e7141 JA |
712 | enum fio_ddir ddir; |
713 | ||
5f9099ea JA |
714 | /* |
715 | * see if it's time to fsync | |
716 | */ | |
717 | if (td->o.fsync_blocks && | |
718 | !(td->io_issues[DDIR_WRITE] % td->o.fsync_blocks) && | |
719 | td->io_issues[DDIR_WRITE] && should_fsync(td)) | |
720 | return DDIR_SYNC; | |
721 | ||
722 | /* | |
723 | * see if it's time to fdatasync | |
724 | */ | |
725 | if (td->o.fdatasync_blocks && | |
726 | !(td->io_issues[DDIR_WRITE] % td->o.fdatasync_blocks) && | |
727 | td->io_issues[DDIR_WRITE] && should_fsync(td)) | |
728 | return DDIR_DATASYNC; | |
729 | ||
44f29692 JA |
730 | /* |
731 | * see if it's time to sync_file_range | |
732 | */ | |
733 | if (td->sync_file_range_nr && | |
734 | !(td->io_issues[DDIR_WRITE] % td->sync_file_range_nr) && | |
735 | td->io_issues[DDIR_WRITE] && should_fsync(td)) | |
736 | return DDIR_SYNC_FILE_RANGE; | |
737 | ||
10ba535a | 738 | if (td_rw(td)) { |
10ba535a JA |
739 | /* |
740 | * Check if it's time to seed a new data direction. | |
741 | */ | |
e4928662 | 742 | if (td->io_issues[td->rwmix_ddir] >= td->rwmix_issues) { |
e47f799f JA |
743 | /* |
744 | * Put a top limit on how many bytes we do for | |
745 | * one data direction, to avoid overflowing the | |
746 | * ranges too much | |
747 | */ | |
748 | ddir = get_rand_ddir(td); | |
e47f799f JA |
749 | |
750 | if (ddir != td->rwmix_ddir) | |
751 | set_rwmix_bytes(td); | |
752 | ||
753 | td->rwmix_ddir = ddir; | |
10ba535a | 754 | } |
581e7141 | 755 | ddir = td->rwmix_ddir; |
10ba535a | 756 | } else if (td_read(td)) |
581e7141 | 757 | ddir = DDIR_READ; |
6eaf09d6 | 758 | else if (td_write(td)) |
581e7141 | 759 | ddir = DDIR_WRITE; |
6eaf09d6 SL |
760 | else |
761 | ddir = DDIR_TRIM; | |
581e7141 JA |
762 | |
763 | td->rwmix_ddir = rate_ddir(td, ddir); | |
764 | return td->rwmix_ddir; | |
10ba535a JA |
765 | } |
766 | ||
1ef2b6be JA |
767 | static void set_rw_ddir(struct thread_data *td, struct io_u *io_u) |
768 | { | |
0e4dd95c DE |
769 | enum fio_ddir ddir = get_rw_ddir(td); |
770 | ||
82a90686 | 771 | if (td_trimwrite(td)) { |
0e4dd95c DE |
772 | struct fio_file *f = io_u->file; |
773 | if (f->last_pos[DDIR_WRITE] == f->last_pos[DDIR_TRIM]) | |
774 | ddir = DDIR_TRIM; | |
775 | else | |
776 | ddir = DDIR_WRITE; | |
777 | } | |
778 | ||
779 | io_u->ddir = io_u->acct_ddir = ddir; | |
1ef2b6be | 780 | |
9b87f09b | 781 | if (io_u->ddir == DDIR_WRITE && td_ioengine_flagged(td, FIO_BARRIER) && |
1ef2b6be JA |
782 | td->o.barrier_blocks && |
783 | !(td->io_issues[DDIR_WRITE] % td->o.barrier_blocks) && | |
784 | td->io_issues[DDIR_WRITE]) | |
1651e431 | 785 | io_u_set(td, io_u, IO_U_F_BARRIER); |
1ef2b6be JA |
786 | } |
787 | ||
e8462bd8 | 788 | void put_file_log(struct thread_data *td, struct fio_file *f) |
60f2c658 | 789 | { |
71b84caa | 790 | unsigned int ret = put_file(td, f); |
60f2c658 JA |
791 | |
792 | if (ret) | |
793 | td_verror(td, ret, "file close"); | |
794 | } | |
795 | ||
10ba535a JA |
796 | void put_io_u(struct thread_data *td, struct io_u *io_u) |
797 | { | |
a9da8ab2 JA |
798 | if (td->parent) |
799 | td = td->parent; | |
800 | ||
e8462bd8 JA |
801 | td_io_u_lock(td); |
802 | ||
f8b0bd10 | 803 | if (io_u->file && !(io_u->flags & IO_U_F_NO_FILE_PUT)) |
60f2c658 | 804 | put_file_log(td, io_u->file); |
f8b0bd10 | 805 | |
10ba535a | 806 | io_u->file = NULL; |
1651e431 | 807 | io_u_set(td, io_u, IO_U_F_FREE); |
d7ee2a7d | 808 | |
a9da8ab2 | 809 | if (io_u->flags & IO_U_F_IN_CUR_DEPTH) { |
0c41214f | 810 | td->cur_depth--; |
a9da8ab2 JA |
811 | assert(!(td->flags & TD_F_CHILD)); |
812 | } | |
2ae0b204 | 813 | io_u_qpush(&td->io_u_freelist, io_u); |
e8462bd8 JA |
814 | td_io_u_unlock(td); |
815 | td_io_u_free_notify(td); | |
10ba535a JA |
816 | } |
817 | ||
f2bba182 RR |
818 | void clear_io_u(struct thread_data *td, struct io_u *io_u) |
819 | { | |
1651e431 | 820 | io_u_clear(td, io_u, IO_U_F_FLIGHT); |
f2bba182 RR |
821 | put_io_u(td, io_u); |
822 | } | |
823 | ||
755200a3 JA |
824 | void requeue_io_u(struct thread_data *td, struct io_u **io_u) |
825 | { | |
826 | struct io_u *__io_u = *io_u; | |
bcd5abfa | 827 | enum fio_ddir ddir = acct_ddir(__io_u); |
755200a3 | 828 | |
465221b0 JA |
829 | dprint(FD_IO, "requeue %p\n", __io_u); |
830 | ||
a9da8ab2 JA |
831 | if (td->parent) |
832 | td = td->parent; | |
833 | ||
e8462bd8 JA |
834 | td_io_u_lock(td); |
835 | ||
1651e431 | 836 | io_u_set(td, __io_u, IO_U_F_FREE); |
bcd5abfa JA |
837 | if ((__io_u->flags & IO_U_F_FLIGHT) && ddir_rw(ddir)) |
838 | td->io_issues[ddir]--; | |
5ec10eaa | 839 | |
1651e431 | 840 | io_u_clear(td, __io_u, IO_U_F_FLIGHT); |
a9da8ab2 | 841 | if (__io_u->flags & IO_U_F_IN_CUR_DEPTH) { |
0c41214f | 842 | td->cur_depth--; |
a9da8ab2 JA |
843 | assert(!(td->flags & TD_F_CHILD)); |
844 | } | |
2ae0b204 JA |
845 | |
846 | io_u_rpush(&td->io_u_requeues, __io_u); | |
e8462bd8 | 847 | td_io_u_unlock(td); |
a9da8ab2 | 848 | td_io_u_free_notify(td); |
755200a3 JA |
849 | *io_u = NULL; |
850 | } | |
851 | ||
9bf2061e | 852 | static int fill_io_u(struct thread_data *td, struct io_u *io_u) |
10ba535a | 853 | { |
6aca9b3d JA |
854 | unsigned int is_random; |
855 | ||
9b87f09b | 856 | if (td_ioengine_flagged(td, FIO_NOIO)) |
b4c5e1ac JA |
857 | goto out; |
858 | ||
1ef2b6be | 859 | set_rw_ddir(td, io_u); |
5f9099ea | 860 | |
87dc1ab1 | 861 | /* |
ff58fced | 862 | * fsync() or fdatasync() or trim etc, we are done |
87dc1ab1 | 863 | */ |
ff58fced | 864 | if (!ddir_rw(io_u->ddir)) |
c38e9468 | 865 | goto out; |
a00735e6 | 866 | |
48f5abd3 JA |
867 | /* |
868 | * See if it's time to switch to a new zone | |
869 | */ | |
13af05ae | 870 | if (td->zone_bytes >= td->o.zone_size && td->o.zone_skip) { |
4c8be5b1 JA |
871 | struct fio_file *f = io_u->file; |
872 | ||
48f5abd3 | 873 | td->zone_bytes = 0; |
4c8be5b1 JA |
874 | f->file_offset += td->o.zone_range + td->o.zone_skip; |
875 | ||
876 | /* | |
877 | * Wrap from the beginning, if we exceed the file size | |
878 | */ | |
879 | if (f->file_offset >= f->real_file_size) | |
880 | f->file_offset = f->real_file_size - f->file_offset; | |
f1dfb668 | 881 | f->last_pos[io_u->ddir] = f->file_offset; |
48f5abd3 JA |
882 | td->io_skip_bytes += td->o.zone_skip; |
883 | } | |
884 | ||
10ba535a | 885 | /* |
c685b5b2 JA |
886 | * No log, let the seq/rand engine retrieve the next buflen and |
887 | * position. | |
10ba535a | 888 | */ |
6aca9b3d | 889 | if (get_next_offset(td, io_u, &is_random)) { |
2ba1c290 | 890 | dprint(FD_IO, "io_u %p, failed getting offset\n", io_u); |
bca4ed4d | 891 | return 1; |
2ba1c290 | 892 | } |
10ba535a | 893 | |
6aca9b3d | 894 | io_u->buflen = get_next_buflen(td, io_u, is_random); |
2ba1c290 JA |
895 | if (!io_u->buflen) { |
896 | dprint(FD_IO, "io_u %p, failed getting buflen\n", io_u); | |
bca4ed4d | 897 | return 1; |
2ba1c290 | 898 | } |
bca4ed4d | 899 | |
2ba1c290 JA |
900 | if (io_u->offset + io_u->buflen > io_u->file->real_file_size) { |
901 | dprint(FD_IO, "io_u %p, offset too large\n", io_u); | |
4b91ee8f JA |
902 | dprint(FD_IO, " off=%llu/%lu > %llu\n", |
903 | (unsigned long long) io_u->offset, io_u->buflen, | |
904 | (unsigned long long) io_u->file->real_file_size); | |
6a5e6884 | 905 | return 1; |
2ba1c290 | 906 | } |
6a5e6884 | 907 | |
bca4ed4d JA |
908 | /* |
909 | * mark entry before potentially trimming io_u | |
910 | */ | |
303032ae | 911 | if (td_random(td) && file_randommap(td, io_u->file)) |
9bf2061e | 912 | mark_random_map(td, io_u); |
bca4ed4d | 913 | |
c38e9468 | 914 | out: |
2ba1c290 | 915 | dprint_io_u(io_u, "fill_io_u"); |
d9d91e39 | 916 | td->zone_bytes += io_u->buflen; |
bca4ed4d | 917 | return 0; |
10ba535a JA |
918 | } |
919 | ||
838bc709 JA |
920 | static void __io_u_mark_map(unsigned int *map, unsigned int nr) |
921 | { | |
2b13e716 | 922 | int idx = 0; |
838bc709 JA |
923 | |
924 | switch (nr) { | |
925 | default: | |
2b13e716 | 926 | idx = 6; |
838bc709 JA |
927 | break; |
928 | case 33 ... 64: | |
2b13e716 | 929 | idx = 5; |
838bc709 JA |
930 | break; |
931 | case 17 ... 32: | |
2b13e716 | 932 | idx = 4; |
838bc709 JA |
933 | break; |
934 | case 9 ... 16: | |
2b13e716 | 935 | idx = 3; |
838bc709 JA |
936 | break; |
937 | case 5 ... 8: | |
2b13e716 | 938 | idx = 2; |
838bc709 JA |
939 | break; |
940 | case 1 ... 4: | |
2b13e716 | 941 | idx = 1; |
838bc709 JA |
942 | case 0: |
943 | break; | |
944 | } | |
945 | ||
2b13e716 | 946 | map[idx]++; |
838bc709 JA |
947 | } |
948 | ||
949 | void io_u_mark_submit(struct thread_data *td, unsigned int nr) | |
950 | { | |
951 | __io_u_mark_map(td->ts.io_u_submit, nr); | |
952 | td->ts.total_submit++; | |
953 | } | |
954 | ||
955 | void io_u_mark_complete(struct thread_data *td, unsigned int nr) | |
956 | { | |
957 | __io_u_mark_map(td->ts.io_u_complete, nr); | |
958 | td->ts.total_complete++; | |
959 | } | |
960 | ||
d8005759 | 961 | void io_u_mark_depth(struct thread_data *td, unsigned int nr) |
71619dc2 | 962 | { |
2b13e716 | 963 | int idx = 0; |
71619dc2 JA |
964 | |
965 | switch (td->cur_depth) { | |
966 | default: | |
2b13e716 | 967 | idx = 6; |
a783e61a | 968 | break; |
71619dc2 | 969 | case 32 ... 63: |
2b13e716 | 970 | idx = 5; |
a783e61a | 971 | break; |
71619dc2 | 972 | case 16 ... 31: |
2b13e716 | 973 | idx = 4; |
a783e61a | 974 | break; |
71619dc2 | 975 | case 8 ... 15: |
2b13e716 | 976 | idx = 3; |
a783e61a | 977 | break; |
71619dc2 | 978 | case 4 ... 7: |
2b13e716 | 979 | idx = 2; |
a783e61a | 980 | break; |
71619dc2 | 981 | case 2 ... 3: |
2b13e716 | 982 | idx = 1; |
71619dc2 JA |
983 | case 1: |
984 | break; | |
985 | } | |
986 | ||
2b13e716 | 987 | td->ts.io_u_map[idx] += nr; |
71619dc2 JA |
988 | } |
989 | ||
04a0feae JA |
990 | static void io_u_mark_lat_usec(struct thread_data *td, unsigned long usec) |
991 | { | |
2b13e716 | 992 | int idx = 0; |
04a0feae JA |
993 | |
994 | assert(usec < 1000); | |
995 | ||
996 | switch (usec) { | |
997 | case 750 ... 999: | |
2b13e716 | 998 | idx = 9; |
04a0feae JA |
999 | break; |
1000 | case 500 ... 749: | |
2b13e716 | 1001 | idx = 8; |
04a0feae JA |
1002 | break; |
1003 | case 250 ... 499: | |
2b13e716 | 1004 | idx = 7; |
04a0feae JA |
1005 | break; |
1006 | case 100 ... 249: | |
2b13e716 | 1007 | idx = 6; |
04a0feae JA |
1008 | break; |
1009 | case 50 ... 99: | |
2b13e716 | 1010 | idx = 5; |
04a0feae JA |
1011 | break; |
1012 | case 20 ... 49: | |
2b13e716 | 1013 | idx = 4; |
04a0feae JA |
1014 | break; |
1015 | case 10 ... 19: | |
2b13e716 | 1016 | idx = 3; |
04a0feae JA |
1017 | break; |
1018 | case 4 ... 9: | |
2b13e716 | 1019 | idx = 2; |
04a0feae JA |
1020 | break; |
1021 | case 2 ... 3: | |
2b13e716 | 1022 | idx = 1; |
04a0feae JA |
1023 | case 0 ... 1: |
1024 | break; | |
1025 | } | |
1026 | ||
2b13e716 JA |
1027 | assert(idx < FIO_IO_U_LAT_U_NR); |
1028 | td->ts.io_u_lat_u[idx]++; | |
04a0feae JA |
1029 | } |
1030 | ||
1031 | static void io_u_mark_lat_msec(struct thread_data *td, unsigned long msec) | |
ec118304 | 1032 | { |
2b13e716 | 1033 | int idx = 0; |
ec118304 JA |
1034 | |
1035 | switch (msec) { | |
1036 | default: | |
2b13e716 | 1037 | idx = 11; |
04a0feae | 1038 | break; |
8abdce66 | 1039 | case 1000 ... 1999: |
2b13e716 | 1040 | idx = 10; |
04a0feae | 1041 | break; |
8abdce66 | 1042 | case 750 ... 999: |
2b13e716 | 1043 | idx = 9; |
04a0feae | 1044 | break; |
8abdce66 | 1045 | case 500 ... 749: |
2b13e716 | 1046 | idx = 8; |
04a0feae | 1047 | break; |
8abdce66 | 1048 | case 250 ... 499: |
2b13e716 | 1049 | idx = 7; |
04a0feae | 1050 | break; |
8abdce66 | 1051 | case 100 ... 249: |
2b13e716 | 1052 | idx = 6; |
04a0feae | 1053 | break; |
8abdce66 | 1054 | case 50 ... 99: |
2b13e716 | 1055 | idx = 5; |
04a0feae | 1056 | break; |
8abdce66 | 1057 | case 20 ... 49: |
2b13e716 | 1058 | idx = 4; |
04a0feae | 1059 | break; |
8abdce66 | 1060 | case 10 ... 19: |
2b13e716 | 1061 | idx = 3; |
04a0feae | 1062 | break; |
8abdce66 | 1063 | case 4 ... 9: |
2b13e716 | 1064 | idx = 2; |
04a0feae | 1065 | break; |
ec118304 | 1066 | case 2 ... 3: |
2b13e716 | 1067 | idx = 1; |
ec118304 JA |
1068 | case 0 ... 1: |
1069 | break; | |
1070 | } | |
1071 | ||
2b13e716 JA |
1072 | assert(idx < FIO_IO_U_LAT_M_NR); |
1073 | td->ts.io_u_lat_m[idx]++; | |
04a0feae JA |
1074 | } |
1075 | ||
1076 | static void io_u_mark_latency(struct thread_data *td, unsigned long usec) | |
1077 | { | |
1078 | if (usec < 1000) | |
1079 | io_u_mark_lat_usec(td, usec); | |
1080 | else | |
1081 | io_u_mark_lat_msec(td, usec / 1000); | |
ec118304 JA |
1082 | } |
1083 | ||
8c07860d JA |
1084 | static unsigned int __get_next_fileno_rand(struct thread_data *td) |
1085 | { | |
1086 | unsigned long fileno; | |
1087 | ||
1088 | if (td->o.file_service_type == FIO_FSERVICE_RANDOM) { | |
1089 | uint64_t frand_max = rand_max(&td->next_file_state); | |
1090 | unsigned long r; | |
1091 | ||
1092 | r = __rand(&td->next_file_state); | |
1093 | return (unsigned int) ((double) td->o.nr_files | |
1094 | * (r / (frand_max + 1.0))); | |
1095 | } | |
1096 | ||
1097 | if (td->o.file_service_type == FIO_FSERVICE_ZIPF) | |
1098 | fileno = zipf_next(&td->next_file_zipf); | |
1099 | else if (td->o.file_service_type == FIO_FSERVICE_PARETO) | |
1100 | fileno = pareto_next(&td->next_file_zipf); | |
1101 | else if (td->o.file_service_type == FIO_FSERVICE_GAUSS) | |
1102 | fileno = gauss_next(&td->next_file_gauss); | |
1103 | else { | |
1104 | log_err("fio: bad file service type: %d\n", td->o.file_service_type); | |
1105 | assert(0); | |
1106 | return 0; | |
1107 | } | |
1108 | ||
1109 | return fileno >> FIO_FSERVICE_SHIFT; | |
1110 | } | |
1111 | ||
0aabe160 JA |
1112 | /* |
1113 | * Get next file to service by choosing one at random | |
1114 | */ | |
2cc52930 JA |
1115 | static struct fio_file *get_next_file_rand(struct thread_data *td, |
1116 | enum fio_file_flags goodf, | |
d6aed795 | 1117 | enum fio_file_flags badf) |
0aabe160 | 1118 | { |
0aabe160 | 1119 | struct fio_file *f; |
1c178180 | 1120 | int fno; |
0aabe160 JA |
1121 | |
1122 | do { | |
87b10676 | 1123 | int opened = 0; |
4c07ad86 | 1124 | |
8c07860d | 1125 | fno = __get_next_fileno_rand(td); |
7c83c089 | 1126 | |
126d65c6 | 1127 | f = td->files[fno]; |
d6aed795 | 1128 | if (fio_file_done(f)) |
059e63c0 | 1129 | continue; |
1c178180 | 1130 | |
d6aed795 | 1131 | if (!fio_file_open(f)) { |
87b10676 JA |
1132 | int err; |
1133 | ||
002fe734 JA |
1134 | if (td->nr_open_files >= td->o.open_files) |
1135 | return ERR_PTR(-EBUSY); | |
1136 | ||
87b10676 JA |
1137 | err = td_io_open_file(td, f); |
1138 | if (err) | |
1139 | continue; | |
1140 | opened = 1; | |
1141 | } | |
1142 | ||
2ba1c290 JA |
1143 | if ((!goodf || (f->flags & goodf)) && !(f->flags & badf)) { |
1144 | dprint(FD_FILE, "get_next_file_rand: %p\n", f); | |
0aabe160 | 1145 | return f; |
2ba1c290 | 1146 | } |
87b10676 JA |
1147 | if (opened) |
1148 | td_io_close_file(td, f); | |
0aabe160 JA |
1149 | } while (1); |
1150 | } | |
1151 | ||
1152 | /* | |
1153 | * Get next file to service by doing round robin between all available ones | |
1154 | */ | |
1c178180 JA |
1155 | static struct fio_file *get_next_file_rr(struct thread_data *td, int goodf, |
1156 | int badf) | |
3d7c391d JA |
1157 | { |
1158 | unsigned int old_next_file = td->next_file; | |
1159 | struct fio_file *f; | |
1160 | ||
1161 | do { | |
87b10676 JA |
1162 | int opened = 0; |
1163 | ||
126d65c6 | 1164 | f = td->files[td->next_file]; |
3d7c391d JA |
1165 | |
1166 | td->next_file++; | |
2dc1bbeb | 1167 | if (td->next_file >= td->o.nr_files) |
3d7c391d JA |
1168 | td->next_file = 0; |
1169 | ||
87b10676 | 1170 | dprint(FD_FILE, "trying file %s %x\n", f->file_name, f->flags); |
d6aed795 | 1171 | if (fio_file_done(f)) { |
d5ed68ea | 1172 | f = NULL; |
059e63c0 | 1173 | continue; |
d5ed68ea | 1174 | } |
059e63c0 | 1175 | |
d6aed795 | 1176 | if (!fio_file_open(f)) { |
87b10676 JA |
1177 | int err; |
1178 | ||
002fe734 JA |
1179 | if (td->nr_open_files >= td->o.open_files) |
1180 | return ERR_PTR(-EBUSY); | |
1181 | ||
87b10676 | 1182 | err = td_io_open_file(td, f); |
b5696bfc JA |
1183 | if (err) { |
1184 | dprint(FD_FILE, "error %d on open of %s\n", | |
1185 | err, f->file_name); | |
87c27b45 | 1186 | f = NULL; |
87b10676 | 1187 | continue; |
b5696bfc | 1188 | } |
87b10676 JA |
1189 | opened = 1; |
1190 | } | |
1191 | ||
0b9d69ec JA |
1192 | dprint(FD_FILE, "goodf=%x, badf=%x, ff=%x\n", goodf, badf, |
1193 | f->flags); | |
1c178180 | 1194 | if ((!goodf || (f->flags & goodf)) && !(f->flags & badf)) |
3d7c391d JA |
1195 | break; |
1196 | ||
87b10676 JA |
1197 | if (opened) |
1198 | td_io_close_file(td, f); | |
1199 | ||
3d7c391d JA |
1200 | f = NULL; |
1201 | } while (td->next_file != old_next_file); | |
1202 | ||
2ba1c290 | 1203 | dprint(FD_FILE, "get_next_file_rr: %p\n", f); |
3d7c391d JA |
1204 | return f; |
1205 | } | |
1206 | ||
7eb36574 | 1207 | static struct fio_file *__get_next_file(struct thread_data *td) |
bdb4e2e9 | 1208 | { |
1907dbc6 JA |
1209 | struct fio_file *f; |
1210 | ||
2dc1bbeb | 1211 | assert(td->o.nr_files <= td->files_index); |
1c178180 | 1212 | |
b5696bfc | 1213 | if (td->nr_done_files >= td->o.nr_files) { |
5ec10eaa JA |
1214 | dprint(FD_FILE, "get_next_file: nr_open=%d, nr_done=%d," |
1215 | " nr_files=%d\n", td->nr_open_files, | |
1216 | td->nr_done_files, | |
1217 | td->o.nr_files); | |
bdb4e2e9 | 1218 | return NULL; |
2ba1c290 | 1219 | } |
bdb4e2e9 | 1220 | |
1907dbc6 | 1221 | f = td->file_service_file; |
d6aed795 | 1222 | if (f && fio_file_open(f) && !fio_file_closing(f)) { |
a086c257 JA |
1223 | if (td->o.file_service_type == FIO_FSERVICE_SEQ) |
1224 | goto out; | |
1225 | if (td->file_service_left--) | |
1226 | goto out; | |
1227 | } | |
1907dbc6 | 1228 | |
a086c257 JA |
1229 | if (td->o.file_service_type == FIO_FSERVICE_RR || |
1230 | td->o.file_service_type == FIO_FSERVICE_SEQ) | |
d6aed795 | 1231 | f = get_next_file_rr(td, FIO_FILE_open, FIO_FILE_closing); |
bdb4e2e9 | 1232 | else |
d6aed795 | 1233 | f = get_next_file_rand(td, FIO_FILE_open, FIO_FILE_closing); |
1907dbc6 | 1234 | |
002fe734 JA |
1235 | if (IS_ERR(f)) |
1236 | return f; | |
1237 | ||
1907dbc6 JA |
1238 | td->file_service_file = f; |
1239 | td->file_service_left = td->file_service_nr - 1; | |
2ba1c290 | 1240 | out: |
0dac421f JA |
1241 | if (f) |
1242 | dprint(FD_FILE, "get_next_file: %p [%s]\n", f, f->file_name); | |
1243 | else | |
1244 | dprint(FD_FILE, "get_next_file: NULL\n"); | |
1907dbc6 | 1245 | return f; |
bdb4e2e9 JA |
1246 | } |
1247 | ||
7eb36574 JA |
1248 | static struct fio_file *get_next_file(struct thread_data *td) |
1249 | { | |
372d8962 | 1250 | if (td->flags & TD_F_PROFILE_OPS) { |
d72be545 | 1251 | struct prof_io_ops *ops = &td->prof_io_ops; |
7eb36574 | 1252 | |
d72be545 JA |
1253 | if (ops->get_next_file) |
1254 | return ops->get_next_file(td); | |
1255 | } | |
7eb36574 JA |
1256 | |
1257 | return __get_next_file(td); | |
1258 | } | |
1259 | ||
002fe734 | 1260 | static long set_io_u_file(struct thread_data *td, struct io_u *io_u) |
429f6675 JA |
1261 | { |
1262 | struct fio_file *f; | |
1263 | ||
1264 | do { | |
1265 | f = get_next_file(td); | |
002fe734 JA |
1266 | if (IS_ERR_OR_NULL(f)) |
1267 | return PTR_ERR(f); | |
429f6675 | 1268 | |
429f6675 JA |
1269 | io_u->file = f; |
1270 | get_file(f); | |
1271 | ||
1272 | if (!fill_io_u(td, io_u)) | |
1273 | break; | |
1274 | ||
b5696bfc | 1275 | put_file_log(td, f); |
429f6675 | 1276 | td_io_close_file(td, f); |
b5696bfc | 1277 | io_u->file = NULL; |
8c07860d JA |
1278 | if (td->o.file_service_type & __FIO_FSERVICE_NONUNIFORM) |
1279 | fio_file_reset(td, f); | |
1280 | else { | |
1281 | fio_file_set_done(f); | |
1282 | td->nr_done_files++; | |
1283 | dprint(FD_FILE, "%s: is done (%d of %d)\n", f->file_name, | |
0b9d69ec | 1284 | td->nr_done_files, td->o.nr_files); |
8c07860d | 1285 | } |
429f6675 JA |
1286 | } while (1); |
1287 | ||
1288 | return 0; | |
1289 | } | |
1290 | ||
3e260a46 JA |
1291 | static void lat_fatal(struct thread_data *td, struct io_completion_data *icd, |
1292 | unsigned long tusec, unsigned long max_usec) | |
1293 | { | |
1294 | if (!td->error) | |
1295 | log_err("fio: latency of %lu usec exceeds specified max (%lu usec)\n", tusec, max_usec); | |
1296 | td_verror(td, ETIMEDOUT, "max latency exceeded"); | |
1297 | icd->error = ETIMEDOUT; | |
1298 | } | |
1299 | ||
1300 | static void lat_new_cycle(struct thread_data *td) | |
1301 | { | |
1302 | fio_gettime(&td->latency_ts, NULL); | |
1303 | td->latency_ios = ddir_rw_sum(td->io_blocks); | |
1304 | td->latency_failed = 0; | |
1305 | } | |
1306 | ||
1307 | /* | |
1308 | * We had an IO outside the latency target. Reduce the queue depth. If we | |
1309 | * are at QD=1, then it's time to give up. | |
1310 | */ | |
e39c0676 | 1311 | static bool __lat_target_failed(struct thread_data *td) |
3e260a46 JA |
1312 | { |
1313 | if (td->latency_qd == 1) | |
e39c0676 | 1314 | return true; |
3e260a46 JA |
1315 | |
1316 | td->latency_qd_high = td->latency_qd; | |
6bb58215 JA |
1317 | |
1318 | if (td->latency_qd == td->latency_qd_low) | |
1319 | td->latency_qd_low--; | |
1320 | ||
3e260a46 JA |
1321 | td->latency_qd = (td->latency_qd + td->latency_qd_low) / 2; |
1322 | ||
1323 | dprint(FD_RATE, "Ramped down: %d %d %d\n", td->latency_qd_low, td->latency_qd, td->latency_qd_high); | |
1324 | ||
1325 | /* | |
1326 | * When we ramp QD down, quiesce existing IO to prevent | |
1327 | * a storm of ramp downs due to pending higher depth. | |
1328 | */ | |
1329 | io_u_quiesce(td); | |
1330 | lat_new_cycle(td); | |
e39c0676 | 1331 | return false; |
3e260a46 JA |
1332 | } |
1333 | ||
e39c0676 | 1334 | static bool lat_target_failed(struct thread_data *td) |
3e260a46 JA |
1335 | { |
1336 | if (td->o.latency_percentile.u.f == 100.0) | |
1337 | return __lat_target_failed(td); | |
1338 | ||
1339 | td->latency_failed++; | |
e39c0676 | 1340 | return false; |
3e260a46 JA |
1341 | } |
1342 | ||
1343 | void lat_target_init(struct thread_data *td) | |
1344 | { | |
6bb58215 JA |
1345 | td->latency_end_run = 0; |
1346 | ||
3e260a46 JA |
1347 | if (td->o.latency_target) { |
1348 | dprint(FD_RATE, "Latency target=%llu\n", td->o.latency_target); | |
1349 | fio_gettime(&td->latency_ts, NULL); | |
1350 | td->latency_qd = 1; | |
1351 | td->latency_qd_high = td->o.iodepth; | |
1352 | td->latency_qd_low = 1; | |
1353 | td->latency_ios = ddir_rw_sum(td->io_blocks); | |
1354 | } else | |
1355 | td->latency_qd = td->o.iodepth; | |
1356 | } | |
1357 | ||
6bb58215 JA |
1358 | void lat_target_reset(struct thread_data *td) |
1359 | { | |
1360 | if (!td->latency_end_run) | |
1361 | lat_target_init(td); | |
1362 | } | |
1363 | ||
3e260a46 JA |
1364 | static void lat_target_success(struct thread_data *td) |
1365 | { | |
1366 | const unsigned int qd = td->latency_qd; | |
6bb58215 | 1367 | struct thread_options *o = &td->o; |
3e260a46 JA |
1368 | |
1369 | td->latency_qd_low = td->latency_qd; | |
1370 | ||
1371 | /* | |
1372 | * If we haven't failed yet, we double up to a failing value instead | |
1373 | * of bisecting from highest possible queue depth. If we have set | |
1374 | * a limit other than td->o.iodepth, bisect between that. | |
1375 | */ | |
6bb58215 | 1376 | if (td->latency_qd_high != o->iodepth) |
3e260a46 JA |
1377 | td->latency_qd = (td->latency_qd + td->latency_qd_high) / 2; |
1378 | else | |
1379 | td->latency_qd *= 2; | |
1380 | ||
6bb58215 JA |
1381 | if (td->latency_qd > o->iodepth) |
1382 | td->latency_qd = o->iodepth; | |
3e260a46 JA |
1383 | |
1384 | dprint(FD_RATE, "Ramped up: %d %d %d\n", td->latency_qd_low, td->latency_qd, td->latency_qd_high); | |
6bb58215 | 1385 | |
3e260a46 | 1386 | /* |
6bb58215 JA |
1387 | * Same as last one, we are done. Let it run a latency cycle, so |
1388 | * we get only the results from the targeted depth. | |
3e260a46 | 1389 | */ |
6bb58215 JA |
1390 | if (td->latency_qd == qd) { |
1391 | if (td->latency_end_run) { | |
1392 | dprint(FD_RATE, "We are done\n"); | |
1393 | td->done = 1; | |
1394 | } else { | |
1395 | dprint(FD_RATE, "Quiesce and final run\n"); | |
1396 | io_u_quiesce(td); | |
1397 | td->latency_end_run = 1; | |
1398 | reset_all_stats(td); | |
1399 | reset_io_stats(td); | |
1400 | } | |
1401 | } | |
3e260a46 JA |
1402 | |
1403 | lat_new_cycle(td); | |
1404 | } | |
1405 | ||
1406 | /* | |
1407 | * Check if we can bump the queue depth | |
1408 | */ | |
1409 | void lat_target_check(struct thread_data *td) | |
1410 | { | |
1411 | uint64_t usec_window; | |
1412 | uint64_t ios; | |
1413 | double success_ios; | |
1414 | ||
1415 | usec_window = utime_since_now(&td->latency_ts); | |
1416 | if (usec_window < td->o.latency_window) | |
1417 | return; | |
1418 | ||
1419 | ios = ddir_rw_sum(td->io_blocks) - td->latency_ios; | |
1420 | success_ios = (double) (ios - td->latency_failed) / (double) ios; | |
1421 | success_ios *= 100.0; | |
1422 | ||
1423 | dprint(FD_RATE, "Success rate: %.2f%% (target %.2f%%)\n", success_ios, td->o.latency_percentile.u.f); | |
1424 | ||
1425 | if (success_ios >= td->o.latency_percentile.u.f) | |
1426 | lat_target_success(td); | |
1427 | else | |
1428 | __lat_target_failed(td); | |
1429 | } | |
1430 | ||
1431 | /* | |
1432 | * If latency target is enabled, we might be ramping up or down and not | |
1433 | * using the full queue depth available. | |
1434 | */ | |
e39c0676 | 1435 | bool queue_full(const struct thread_data *td) |
3e260a46 JA |
1436 | { |
1437 | const int qempty = io_u_qempty(&td->io_u_freelist); | |
1438 | ||
1439 | if (qempty) | |
e39c0676 | 1440 | return true; |
3e260a46 | 1441 | if (!td->o.latency_target) |
e39c0676 | 1442 | return false; |
3e260a46 JA |
1443 | |
1444 | return td->cur_depth >= td->latency_qd; | |
1445 | } | |
429f6675 | 1446 | |
10ba535a JA |
1447 | struct io_u *__get_io_u(struct thread_data *td) |
1448 | { | |
0cae66f6 | 1449 | struct io_u *io_u = NULL; |
10ba535a | 1450 | |
ca09be4b JA |
1451 | if (td->stop_io) |
1452 | return NULL; | |
1453 | ||
e8462bd8 JA |
1454 | td_io_u_lock(td); |
1455 | ||
1456 | again: | |
2ae0b204 JA |
1457 | if (!io_u_rempty(&td->io_u_requeues)) |
1458 | io_u = io_u_rpop(&td->io_u_requeues); | |
3e260a46 | 1459 | else if (!queue_full(td)) { |
2ae0b204 | 1460 | io_u = io_u_qpop(&td->io_u_freelist); |
10ba535a | 1461 | |
225ba9e3 | 1462 | io_u->file = NULL; |
6040dabc | 1463 | io_u->buflen = 0; |
10ba535a | 1464 | io_u->resid = 0; |
d7762cf8 | 1465 | io_u->end_io = NULL; |
755200a3 JA |
1466 | } |
1467 | ||
1468 | if (io_u) { | |
0c6e7517 | 1469 | assert(io_u->flags & IO_U_F_FREE); |
1651e431 | 1470 | io_u_clear(td, io_u, IO_U_F_FREE | IO_U_F_NO_FILE_PUT | |
f8b0bd10 JA |
1471 | IO_U_F_TRIMMED | IO_U_F_BARRIER | |
1472 | IO_U_F_VER_LIST); | |
0c6e7517 | 1473 | |
755200a3 | 1474 | io_u->error = 0; |
bcd5abfa | 1475 | io_u->acct_ddir = -1; |
10ba535a | 1476 | td->cur_depth++; |
a9da8ab2 | 1477 | assert(!(td->flags & TD_F_CHILD)); |
1651e431 | 1478 | io_u_set(td, io_u, IO_U_F_IN_CUR_DEPTH); |
f9401285 | 1479 | io_u->ipo = NULL; |
a9da8ab2 | 1480 | } else if (td_async_processing(td)) { |
1dec3e07 JA |
1481 | /* |
1482 | * We ran out, wait for async verify threads to finish and | |
1483 | * return one | |
1484 | */ | |
a9da8ab2 JA |
1485 | assert(!(td->flags & TD_F_CHILD)); |
1486 | assert(!pthread_cond_wait(&td->free_cond, &td->io_u_lock)); | |
1dec3e07 | 1487 | goto again; |
10ba535a JA |
1488 | } |
1489 | ||
e8462bd8 | 1490 | td_io_u_unlock(td); |
10ba535a JA |
1491 | return io_u; |
1492 | } | |
1493 | ||
e39c0676 | 1494 | static bool check_get_trim(struct thread_data *td, struct io_u *io_u) |
10ba535a | 1495 | { |
d72be545 | 1496 | if (!(td->flags & TD_F_TRIM_BACKLOG)) |
e39c0676 | 1497 | return false; |
d72be545 JA |
1498 | |
1499 | if (td->trim_entries) { | |
0d29de83 | 1500 | int get_trim = 0; |
10ba535a | 1501 | |
0d29de83 JA |
1502 | if (td->trim_batch) { |
1503 | td->trim_batch--; | |
1504 | get_trim = 1; | |
1505 | } else if (!(td->io_hist_len % td->o.trim_backlog) && | |
1506 | td->last_ddir != DDIR_READ) { | |
1507 | td->trim_batch = td->o.trim_batch; | |
1508 | if (!td->trim_batch) | |
1509 | td->trim_batch = td->o.trim_backlog; | |
1510 | get_trim = 1; | |
1511 | } | |
1512 | ||
1513 | if (get_trim && !get_next_trim(td, io_u)) | |
e39c0676 | 1514 | return true; |
2ba1c290 | 1515 | } |
10ba535a | 1516 | |
e39c0676 | 1517 | return false; |
0d29de83 JA |
1518 | } |
1519 | ||
e39c0676 | 1520 | static bool check_get_verify(struct thread_data *td, struct io_u *io_u) |
0d29de83 | 1521 | { |
d72be545 | 1522 | if (!(td->flags & TD_F_VER_BACKLOG)) |
e39c0676 | 1523 | return false; |
d72be545 JA |
1524 | |
1525 | if (td->io_hist_len) { | |
9e144189 JA |
1526 | int get_verify = 0; |
1527 | ||
d1ece0c7 | 1528 | if (td->verify_batch) |
9e144189 | 1529 | get_verify = 1; |
d1ece0c7 | 1530 | else if (!(td->io_hist_len % td->o.verify_backlog) && |
9e144189 JA |
1531 | td->last_ddir != DDIR_READ) { |
1532 | td->verify_batch = td->o.verify_batch; | |
f8a75c99 JA |
1533 | if (!td->verify_batch) |
1534 | td->verify_batch = td->o.verify_backlog; | |
9e144189 JA |
1535 | get_verify = 1; |
1536 | } | |
1537 | ||
d1ece0c7 JA |
1538 | if (get_verify && !get_next_verify(td, io_u)) { |
1539 | td->verify_batch--; | |
e39c0676 | 1540 | return true; |
d1ece0c7 | 1541 | } |
9e144189 JA |
1542 | } |
1543 | ||
e39c0676 | 1544 | return false; |
0d29de83 JA |
1545 | } |
1546 | ||
de789769 JA |
1547 | /* |
1548 | * Fill offset and start time into the buffer content, to prevent too | |
23f394d5 JA |
1549 | * easy compressible data for simple de-dupe attempts. Do this for every |
1550 | * 512b block in the range, since that should be the smallest block size | |
1551 | * we can expect from a device. | |
de789769 JA |
1552 | */ |
1553 | static void small_content_scramble(struct io_u *io_u) | |
1554 | { | |
23f394d5 | 1555 | unsigned int i, nr_blocks = io_u->buflen / 512; |
1ae83d45 | 1556 | uint64_t boffset; |
23f394d5 JA |
1557 | unsigned int offset; |
1558 | void *p, *end; | |
de789769 | 1559 | |
23f394d5 JA |
1560 | if (!nr_blocks) |
1561 | return; | |
1562 | ||
1563 | p = io_u->xfer_buf; | |
fba76ee8 | 1564 | boffset = io_u->offset; |
81f0366c | 1565 | io_u->buf_filled_len = 0; |
fad82f76 | 1566 | |
23f394d5 JA |
1567 | for (i = 0; i < nr_blocks; i++) { |
1568 | /* | |
1569 | * Fill the byte offset into a "random" start offset of | |
1570 | * the buffer, given by the product of the usec time | |
1571 | * and the actual offset. | |
1572 | */ | |
fad82f76 | 1573 | offset = (io_u->start_time.tv_usec ^ boffset) & 511; |
1ae83d45 JA |
1574 | offset &= ~(sizeof(uint64_t) - 1); |
1575 | if (offset >= 512 - sizeof(uint64_t)) | |
1576 | offset -= sizeof(uint64_t); | |
fba76ee8 | 1577 | memcpy(p + offset, &boffset, sizeof(boffset)); |
23f394d5 JA |
1578 | |
1579 | end = p + 512 - sizeof(io_u->start_time); | |
1580 | memcpy(end, &io_u->start_time, sizeof(io_u->start_time)); | |
1581 | p += 512; | |
fad82f76 | 1582 | boffset += 512; |
23f394d5 | 1583 | } |
de789769 JA |
1584 | } |
1585 | ||
0d29de83 JA |
1586 | /* |
1587 | * Return an io_u to be processed. Gets a buflen and offset, sets direction, | |
1588 | * etc. The returned io_u is fully ready to be prepped and submitted. | |
1589 | */ | |
1590 | struct io_u *get_io_u(struct thread_data *td) | |
1591 | { | |
1592 | struct fio_file *f; | |
1593 | struct io_u *io_u; | |
de789769 | 1594 | int do_scramble = 0; |
002fe734 | 1595 | long ret = 0; |
0d29de83 JA |
1596 | |
1597 | io_u = __get_io_u(td); | |
1598 | if (!io_u) { | |
1599 | dprint(FD_IO, "__get_io_u failed\n"); | |
1600 | return NULL; | |
1601 | } | |
1602 | ||
1603 | if (check_get_verify(td, io_u)) | |
1604 | goto out; | |
1605 | if (check_get_trim(td, io_u)) | |
1606 | goto out; | |
1607 | ||
755200a3 JA |
1608 | /* |
1609 | * from a requeue, io_u already setup | |
1610 | */ | |
1611 | if (io_u->file) | |
77f392bf | 1612 | goto out; |
755200a3 | 1613 | |
429f6675 JA |
1614 | /* |
1615 | * If using an iolog, grab next piece if any available. | |
1616 | */ | |
d72be545 | 1617 | if (td->flags & TD_F_READ_IOLOG) { |
429f6675 JA |
1618 | if (read_iolog_get(td, io_u)) |
1619 | goto err_put; | |
2ba1c290 | 1620 | } else if (set_io_u_file(td, io_u)) { |
002fe734 | 1621 | ret = -EBUSY; |
2ba1c290 | 1622 | dprint(FD_IO, "io_u %p, setting file failed\n", io_u); |
429f6675 | 1623 | goto err_put; |
2ba1c290 | 1624 | } |
5ec10eaa | 1625 | |
429f6675 | 1626 | f = io_u->file; |
002fe734 JA |
1627 | if (!f) { |
1628 | dprint(FD_IO, "io_u %p, setting file failed\n", io_u); | |
1629 | goto err_put; | |
1630 | } | |
1631 | ||
d6aed795 | 1632 | assert(fio_file_open(f)); |
97af62ce | 1633 | |
ff58fced | 1634 | if (ddir_rw(io_u->ddir)) { |
9b87f09b | 1635 | if (!io_u->buflen && !td_ioengine_flagged(td, FIO_NOIO)) { |
2ba1c290 | 1636 | dprint(FD_IO, "get_io_u: zero buflen on %p\n", io_u); |
429f6675 | 1637 | goto err_put; |
2ba1c290 | 1638 | } |
10ba535a | 1639 | |
f1dfb668 JA |
1640 | f->last_start[io_u->ddir] = io_u->offset; |
1641 | f->last_pos[io_u->ddir] = io_u->offset + io_u->buflen; | |
10ba535a | 1642 | |
fd68418e | 1643 | if (io_u->ddir == DDIR_WRITE) { |
d72be545 | 1644 | if (td->flags & TD_F_REFILL_BUFFERS) { |
9c42684e | 1645 | io_u_fill_buffer(td, io_u, |
1066358a | 1646 | td->o.min_bs[DDIR_WRITE], |
9e129577 | 1647 | io_u->buflen); |
ff441ae8 | 1648 | } else if ((td->flags & TD_F_SCRAMBLE_BUFFERS) && |
bedc9dc2 | 1649 | !(td->flags & TD_F_COMPRESS)) |
fd68418e | 1650 | do_scramble = 1; |
d72be545 | 1651 | if (td->flags & TD_F_VER_NONE) { |
629f1d71 JA |
1652 | populate_verify_io_u(td, io_u); |
1653 | do_scramble = 0; | |
1654 | } | |
fd68418e | 1655 | } else if (io_u->ddir == DDIR_READ) { |
cbe8d756 RR |
1656 | /* |
1657 | * Reset the buf_filled parameters so next time if the | |
1658 | * buffer is used for writes it is refilled. | |
1659 | */ | |
cbe8d756 RR |
1660 | io_u->buf_filled_len = 0; |
1661 | } | |
87dc1ab1 | 1662 | } |
10ba535a | 1663 | |
165faf16 JA |
1664 | /* |
1665 | * Set io data pointers. | |
1666 | */ | |
cec6b55d JA |
1667 | io_u->xfer_buf = io_u->buf; |
1668 | io_u->xfer_buflen = io_u->buflen; | |
5973cafb | 1669 | |
6ac7a331 | 1670 | out: |
0d29de83 | 1671 | assert(io_u->file); |
429f6675 | 1672 | if (!td_io_prep(td, io_u)) { |
3ecc8c67 | 1673 | if (!td->o.disable_lat) |
993bf48b | 1674 | fio_gettime(&io_u->start_time, NULL); |
de789769 JA |
1675 | if (do_scramble) |
1676 | small_content_scramble(io_u); | |
429f6675 | 1677 | return io_u; |
36167d82 | 1678 | } |
429f6675 | 1679 | err_put: |
2ba1c290 | 1680 | dprint(FD_IO, "get_io_u failed\n"); |
429f6675 | 1681 | put_io_u(td, io_u); |
002fe734 | 1682 | return ERR_PTR(ret); |
10ba535a JA |
1683 | } |
1684 | ||
a9da8ab2 | 1685 | static void __io_u_log_error(struct thread_data *td, struct io_u *io_u) |
5451792e | 1686 | { |
8b28bd41 | 1687 | enum error_type_bit eb = td_error_type(io_u->ddir, io_u->error); |
825f818e | 1688 | |
8b28bd41 DM |
1689 | if (td_non_fatal_error(td, eb, io_u->error) && !td->o.error_dump) |
1690 | return; | |
5451792e | 1691 | |
709c8313 RE |
1692 | log_err("fio: io_u error%s%s: %s: %s offset=%llu, buflen=%lu\n", |
1693 | io_u->file ? " on file " : "", | |
1694 | io_u->file ? io_u->file->file_name : "", | |
1695 | strerror(io_u->error), | |
1696 | io_ddir_name(io_u->ddir), | |
1697 | io_u->offset, io_u->xfer_buflen); | |
5451792e | 1698 | |
5ad7be56 KD |
1699 | if (td->io_ops->errdetails) { |
1700 | char *err = td->io_ops->errdetails(io_u); | |
1701 | ||
1702 | log_err("fio: %s\n", err); | |
1703 | free(err); | |
1704 | } | |
1705 | ||
5451792e JA |
1706 | if (!td->error) |
1707 | td_verror(td, io_u->error, "io_u error"); | |
1708 | } | |
1709 | ||
a9da8ab2 JA |
1710 | void io_u_log_error(struct thread_data *td, struct io_u *io_u) |
1711 | { | |
1712 | __io_u_log_error(td, io_u); | |
1713 | if (td->parent) | |
094e66cb | 1714 | __io_u_log_error(td->parent, io_u); |
a9da8ab2 JA |
1715 | } |
1716 | ||
e39c0676 | 1717 | static inline bool gtod_reduce(struct thread_data *td) |
aba6c951 | 1718 | { |
3ecc8c67 JA |
1719 | return (td->o.disable_clat && td->o.disable_slat && td->o.disable_bw) |
1720 | || td->o.gtod_reduce; | |
aba6c951 JA |
1721 | } |
1722 | ||
c8eeb9df JA |
1723 | static void account_io_completion(struct thread_data *td, struct io_u *io_u, |
1724 | struct io_completion_data *icd, | |
1725 | const enum fio_ddir idx, unsigned int bytes) | |
1726 | { | |
a9da8ab2 | 1727 | const int no_reduce = !gtod_reduce(td); |
24d23ca7 | 1728 | unsigned long lusec = 0; |
c8eeb9df | 1729 | |
75dc383e JA |
1730 | if (td->parent) |
1731 | td = td->parent; | |
1732 | ||
a9da8ab2 | 1733 | if (no_reduce) |
c8eeb9df JA |
1734 | lusec = utime_since(&io_u->issue_time, &icd->time); |
1735 | ||
1736 | if (!td->o.disable_lat) { | |
1737 | unsigned long tusec; | |
1738 | ||
1739 | tusec = utime_since(&io_u->start_time, &icd->time); | |
ae588852 | 1740 | add_lat_sample(td, idx, tusec, bytes, io_u->offset); |
15501535 | 1741 | |
d4afedfd JA |
1742 | if (td->flags & TD_F_PROFILE_OPS) { |
1743 | struct prof_io_ops *ops = &td->prof_io_ops; | |
1744 | ||
1745 | if (ops->io_u_lat) | |
1746 | icd->error = ops->io_u_lat(td, tusec); | |
1747 | } | |
1748 | ||
3e260a46 JA |
1749 | if (td->o.max_latency && tusec > td->o.max_latency) |
1750 | lat_fatal(td, icd, tusec, td->o.max_latency); | |
1751 | if (td->o.latency_target && tusec > td->o.latency_target) { | |
1752 | if (lat_target_failed(td)) | |
1753 | lat_fatal(td, icd, tusec, td->o.latency_target); | |
15501535 | 1754 | } |
c8eeb9df JA |
1755 | } |
1756 | ||
a47591e4 JA |
1757 | if (ddir_rw(idx)) { |
1758 | if (!td->o.disable_clat) { | |
1759 | add_clat_sample(td, idx, lusec, bytes, io_u->offset); | |
1760 | io_u_mark_latency(td, lusec); | |
1761 | } | |
c8eeb9df | 1762 | |
a47591e4 JA |
1763 | if (!td->o.disable_bw && per_unit_log(td->bw_log)) |
1764 | add_bw_sample(td, io_u, bytes, lusec); | |
c8eeb9df | 1765 | |
a47591e4 JA |
1766 | if (no_reduce && per_unit_log(td->iops_log)) |
1767 | add_iops_sample(td, io_u, bytes); | |
1768 | } | |
66347cfa DE |
1769 | |
1770 | if (td->ts.nr_block_infos && io_u->ddir == DDIR_TRIM) { | |
1771 | uint32_t *info = io_u_block_info(td, io_u); | |
1772 | if (BLOCK_INFO_STATE(*info) < BLOCK_STATE_TRIM_FAILURE) { | |
1773 | if (io_u->ddir == DDIR_TRIM) { | |
1774 | *info = BLOCK_INFO(BLOCK_STATE_TRIMMED, | |
1775 | BLOCK_INFO_TRIMS(*info) + 1); | |
1776 | } else if (io_u->ddir == DDIR_WRITE) { | |
1777 | *info = BLOCK_INFO_SET_STATE(BLOCK_STATE_WRITTEN, | |
1778 | *info); | |
1779 | } | |
1780 | } | |
1781 | } | |
c8eeb9df JA |
1782 | } |
1783 | ||
94a6e1bb JA |
1784 | static void file_log_write_comp(const struct thread_data *td, struct fio_file *f, |
1785 | uint64_t offset, unsigned int bytes) | |
1786 | { | |
1787 | int idx; | |
1788 | ||
639ad1ea JA |
1789 | if (!f) |
1790 | return; | |
1791 | ||
94a6e1bb JA |
1792 | if (f->first_write == -1ULL || offset < f->first_write) |
1793 | f->first_write = offset; | |
1794 | if (f->last_write == -1ULL || ((offset + bytes) > f->last_write)) | |
1795 | f->last_write = offset + bytes; | |
1796 | ||
1797 | if (!f->last_write_comp) | |
1798 | return; | |
1799 | ||
1800 | idx = f->last_write_idx++; | |
1801 | f->last_write_comp[idx] = offset; | |
1802 | if (f->last_write_idx == td->o.iodepth) | |
1803 | f->last_write_idx = 0; | |
1804 | } | |
1805 | ||
f8b0bd10 | 1806 | static void io_completed(struct thread_data *td, struct io_u **io_u_ptr, |
97601024 | 1807 | struct io_completion_data *icd) |
10ba535a | 1808 | { |
f8b0bd10 JA |
1809 | struct io_u *io_u = *io_u_ptr; |
1810 | enum fio_ddir ddir = io_u->ddir; | |
1811 | struct fio_file *f = io_u->file; | |
10ba535a | 1812 | |
2ba1c290 JA |
1813 | dprint_io_u(io_u, "io complete"); |
1814 | ||
0c6e7517 | 1815 | assert(io_u->flags & IO_U_F_FLIGHT); |
1651e431 | 1816 | io_u_clear(td, io_u, IO_U_F_FLIGHT | IO_U_F_BUSY_OK); |
f9401285 JA |
1817 | |
1818 | /* | |
1819 | * Mark IO ok to verify | |
1820 | */ | |
1821 | if (io_u->ipo) { | |
890b6656 JA |
1822 | /* |
1823 | * Remove errored entry from the verification list | |
1824 | */ | |
1825 | if (io_u->error) | |
1826 | unlog_io_piece(td, io_u); | |
1827 | else { | |
1828 | io_u->ipo->flags &= ~IP_F_IN_FLIGHT; | |
1829 | write_barrier(); | |
1830 | } | |
f9401285 JA |
1831 | } |
1832 | ||
f8b0bd10 | 1833 | if (ddir_sync(ddir)) { |
87dc1ab1 | 1834 | td->last_was_sync = 1; |
44f29692 JA |
1835 | if (f) { |
1836 | f->first_write = -1ULL; | |
1837 | f->last_write = -1ULL; | |
1838 | } | |
87dc1ab1 JA |
1839 | return; |
1840 | } | |
1841 | ||
1842 | td->last_was_sync = 0; | |
f8b0bd10 | 1843 | td->last_ddir = ddir; |
87dc1ab1 | 1844 | |
f8b0bd10 | 1845 | if (!io_u->error && ddir_rw(ddir)) { |
10ba535a | 1846 | unsigned int bytes = io_u->buflen - io_u->resid; |
b29ee5b3 | 1847 | int ret; |
10ba535a | 1848 | |
f8b0bd10 JA |
1849 | td->io_blocks[ddir]++; |
1850 | td->this_io_blocks[ddir]++; | |
1851 | td->io_bytes[ddir] += bytes; | |
ae2fafc8 | 1852 | |
1853 | if (!(io_u->flags & IO_U_F_VER_LIST)) | |
f8b0bd10 JA |
1854 | td->this_io_bytes[ddir] += bytes; |
1855 | ||
639ad1ea | 1856 | if (ddir == DDIR_WRITE) |
94a6e1bb | 1857 | file_log_write_comp(td, f, io_u->offset, bytes); |
44f29692 | 1858 | |
6b1190fd | 1859 | if (ramp_time_over(td) && (td->runstate == TD_RUNNING || |
50a8ce86 | 1860 | td->runstate == TD_VERIFYING)) |
f8b0bd10 | 1861 | account_io_completion(td, io_u, icd, ddir, bytes); |
40e1a6f0 | 1862 | |
f8b0bd10 | 1863 | icd->bytes_done[ddir] += bytes; |
3af6ef39 | 1864 | |
d7762cf8 | 1865 | if (io_u->end_io) { |
f8b0bd10 JA |
1866 | ret = io_u->end_io(td, io_u_ptr); |
1867 | io_u = *io_u_ptr; | |
3af6ef39 JA |
1868 | if (ret && !icd->error) |
1869 | icd->error = ret; | |
1870 | } | |
ff58fced | 1871 | } else if (io_u->error) { |
10ba535a | 1872 | icd->error = io_u->error; |
5451792e JA |
1873 | io_u_log_error(td, io_u); |
1874 | } | |
8b28bd41 | 1875 | if (icd->error) { |
f8b0bd10 JA |
1876 | enum error_type_bit eb = td_error_type(ddir, icd->error); |
1877 | ||
8b28bd41 DM |
1878 | if (!td_non_fatal_error(td, eb, icd->error)) |
1879 | return; | |
f8b0bd10 | 1880 | |
f2bba182 RR |
1881 | /* |
1882 | * If there is a non_fatal error, then add to the error count | |
1883 | * and clear all the errors. | |
1884 | */ | |
1885 | update_error_count(td, icd->error); | |
1886 | td_clear_error(td); | |
1887 | icd->error = 0; | |
f8b0bd10 JA |
1888 | if (io_u) |
1889 | io_u->error = 0; | |
f2bba182 | 1890 | } |
10ba535a JA |
1891 | } |
1892 | ||
9520ebb9 JA |
1893 | static void init_icd(struct thread_data *td, struct io_completion_data *icd, |
1894 | int nr) | |
10ba535a | 1895 | { |
6eaf09d6 | 1896 | int ddir; |
aba6c951 JA |
1897 | |
1898 | if (!gtod_reduce(td)) | |
9520ebb9 | 1899 | fio_gettime(&icd->time, NULL); |
02bcaa8c | 1900 | |
3af6ef39 JA |
1901 | icd->nr = nr; |
1902 | ||
10ba535a | 1903 | icd->error = 0; |
6eaf09d6 SL |
1904 | for (ddir = DDIR_READ; ddir < DDIR_RWDIR_CNT; ddir++) |
1905 | icd->bytes_done[ddir] = 0; | |
36167d82 JA |
1906 | } |
1907 | ||
97601024 JA |
1908 | static void ios_completed(struct thread_data *td, |
1909 | struct io_completion_data *icd) | |
36167d82 JA |
1910 | { |
1911 | struct io_u *io_u; | |
1912 | int i; | |
1913 | ||
10ba535a JA |
1914 | for (i = 0; i < icd->nr; i++) { |
1915 | io_u = td->io_ops->event(td, i); | |
1916 | ||
f8b0bd10 | 1917 | io_completed(td, &io_u, icd); |
e8462bd8 | 1918 | |
f8b0bd10 | 1919 | if (io_u) |
e8462bd8 | 1920 | put_io_u(td, io_u); |
10ba535a JA |
1921 | } |
1922 | } | |
97601024 | 1923 | |
e7e6cfb4 JA |
1924 | /* |
1925 | * Complete a single io_u for the sync engines. | |
1926 | */ | |
55312f9f | 1927 | int io_u_sync_complete(struct thread_data *td, struct io_u *io_u) |
97601024 JA |
1928 | { |
1929 | struct io_completion_data icd; | |
55312f9f | 1930 | int ddir; |
97601024 | 1931 | |
9520ebb9 | 1932 | init_icd(td, &icd, 1); |
f8b0bd10 | 1933 | io_completed(td, &io_u, &icd); |
e8462bd8 | 1934 | |
f8b0bd10 | 1935 | if (io_u) |
e8462bd8 | 1936 | put_io_u(td, io_u); |
97601024 | 1937 | |
581e7141 JA |
1938 | if (icd.error) { |
1939 | td_verror(td, icd.error, "io_u_sync_complete"); | |
1940 | return -1; | |
1941 | } | |
97601024 | 1942 | |
55312f9f JA |
1943 | for (ddir = DDIR_READ; ddir < DDIR_RWDIR_CNT; ddir++) |
1944 | td->bytes_done[ddir] += icd.bytes_done[ddir]; | |
581e7141 JA |
1945 | |
1946 | return 0; | |
97601024 JA |
1947 | } |
1948 | ||
e7e6cfb4 JA |
1949 | /* |
1950 | * Called to complete min_events number of io for the async engines. | |
1951 | */ | |
55312f9f | 1952 | int io_u_queued_complete(struct thread_data *td, int min_evts) |
97601024 | 1953 | { |
97601024 | 1954 | struct io_completion_data icd; |
00de55ef | 1955 | struct timespec *tvp = NULL; |
55312f9f | 1956 | int ret, ddir; |
4d06a338 | 1957 | struct timespec ts = { .tv_sec = 0, .tv_nsec = 0, }; |
97601024 | 1958 | |
4950421a | 1959 | dprint(FD_IO, "io_u_queued_completed: min=%d\n", min_evts); |
b271fe62 | 1960 | |
4950421a | 1961 | if (!min_evts) |
00de55ef | 1962 | tvp = &ts; |
5fb4b366 RE |
1963 | else if (min_evts > td->cur_depth) |
1964 | min_evts = td->cur_depth; | |
97601024 | 1965 | |
82407585 RP |
1966 | /* No worries, td_io_getevents fixes min and max if they are |
1967 | * set incorrectly */ | |
1968 | ret = td_io_getevents(td, min_evts, td->o.iodepth_batch_complete_max, tvp); | |
97601024 | 1969 | if (ret < 0) { |
e1161c32 | 1970 | td_verror(td, -ret, "td_io_getevents"); |
97601024 JA |
1971 | return ret; |
1972 | } else if (!ret) | |
1973 | return ret; | |
1974 | ||
9520ebb9 | 1975 | init_icd(td, &icd, ret); |
97601024 | 1976 | ios_completed(td, &icd); |
581e7141 JA |
1977 | if (icd.error) { |
1978 | td_verror(td, icd.error, "io_u_queued_complete"); | |
1979 | return -1; | |
1980 | } | |
97601024 | 1981 | |
55312f9f JA |
1982 | for (ddir = DDIR_READ; ddir < DDIR_RWDIR_CNT; ddir++) |
1983 | td->bytes_done[ddir] += icd.bytes_done[ddir]; | |
581e7141 | 1984 | |
0d593542 | 1985 | return ret; |
97601024 | 1986 | } |
7e77dd02 JA |
1987 | |
1988 | /* | |
1989 | * Call when io_u is really queued, to update the submission latency. | |
1990 | */ | |
1991 | void io_u_queued(struct thread_data *td, struct io_u *io_u) | |
1992 | { | |
9520ebb9 JA |
1993 | if (!td->o.disable_slat) { |
1994 | unsigned long slat_time; | |
7e77dd02 | 1995 | |
9520ebb9 | 1996 | slat_time = utime_since(&io_u->start_time, &io_u->issue_time); |
75dc383e JA |
1997 | |
1998 | if (td->parent) | |
1999 | td = td->parent; | |
2000 | ||
ae588852 JA |
2001 | add_slat_sample(td, io_u->ddir, slat_time, io_u->xfer_buflen, |
2002 | io_u->offset); | |
9520ebb9 | 2003 | } |
7e77dd02 | 2004 | } |
433afcb4 | 2005 | |
5c94b008 JA |
2006 | /* |
2007 | * See if we should reuse the last seed, if dedupe is enabled | |
2008 | */ | |
9451b93e | 2009 | static struct frand_state *get_buf_state(struct thread_data *td) |
5c94b008 JA |
2010 | { |
2011 | unsigned int v; | |
5c94b008 JA |
2012 | |
2013 | if (!td->o.dedupe_percentage) | |
2014 | return &td->buf_state; | |
732eedd0 | 2015 | else if (td->o.dedupe_percentage == 100) { |
9451b93e JA |
2016 | frand_copy(&td->buf_state_prev, &td->buf_state); |
2017 | return &td->buf_state; | |
732eedd0 | 2018 | } |
5c94b008 | 2019 | |
36dd3379 | 2020 | v = rand32_between(&td->dedupe_state, 1, 100); |
5c94b008 JA |
2021 | |
2022 | if (v <= td->o.dedupe_percentage) | |
2023 | return &td->buf_state_prev; | |
2024 | ||
2025 | return &td->buf_state; | |
2026 | } | |
2027 | ||
9451b93e | 2028 | static void save_buf_state(struct thread_data *td, struct frand_state *rs) |
5c94b008 | 2029 | { |
9451b93e JA |
2030 | if (td->o.dedupe_percentage == 100) |
2031 | frand_copy(rs, &td->buf_state_prev); | |
2032 | else if (rs == &td->buf_state) | |
5c94b008 JA |
2033 | frand_copy(&td->buf_state_prev, rs); |
2034 | } | |
2035 | ||
cc86c395 JA |
2036 | void fill_io_buffer(struct thread_data *td, void *buf, unsigned int min_write, |
2037 | unsigned int max_bs) | |
5973cafb | 2038 | { |
d1af2894 JA |
2039 | struct thread_options *o = &td->o; |
2040 | ||
4eff3e57 | 2041 | if (o->compress_percentage || o->dedupe_percentage) { |
9c42684e | 2042 | unsigned int perc = td->o.compress_percentage; |
5c94b008 | 2043 | struct frand_state *rs; |
1066358a | 2044 | unsigned int left = max_bs; |
1e7f82e2 | 2045 | unsigned int this_write; |
5c94b008 | 2046 | |
1066358a | 2047 | do { |
9451b93e | 2048 | rs = get_buf_state(td); |
9c42684e | 2049 | |
1066358a | 2050 | min_write = min(min_write, left); |
f97a43a1 | 2051 | |
1066358a | 2052 | if (perc) { |
4b157ac6 JA |
2053 | this_write = min_not_zero(min_write, |
2054 | td->o.compress_chunk); | |
1e7f82e2 JA |
2055 | |
2056 | fill_random_buf_percentage(rs, buf, perc, | |
2057 | this_write, this_write, | |
2058 | o->buffer_pattern, | |
2059 | o->buffer_pattern_bytes); | |
2060 | } else { | |
1066358a | 2061 | fill_random_buf(rs, buf, min_write); |
1e7f82e2 JA |
2062 | this_write = min_write; |
2063 | } | |
1066358a | 2064 | |
1e7f82e2 JA |
2065 | buf += this_write; |
2066 | left -= this_write; | |
9451b93e | 2067 | save_buf_state(td, rs); |
1066358a | 2068 | } while (left); |
d1af2894 JA |
2069 | } else if (o->buffer_pattern_bytes) |
2070 | fill_buffer_pattern(td, buf, max_bs); | |
999d245e | 2071 | else if (o->zero_buffers) |
cc86c395 | 2072 | memset(buf, 0, max_bs); |
999d245e | 2073 | else |
9451b93e | 2074 | fill_random_buf(get_buf_state(td), buf, max_bs); |
cc86c395 JA |
2075 | } |
2076 | ||
2077 | /* | |
2078 | * "randomly" fill the buffer contents | |
2079 | */ | |
2080 | void io_u_fill_buffer(struct thread_data *td, struct io_u *io_u, | |
2081 | unsigned int min_write, unsigned int max_bs) | |
2082 | { | |
2083 | io_u->buf_filled_len = 0; | |
2084 | fill_io_buffer(td, io_u->buf, min_write, max_bs); | |
5973cafb | 2085 | } |