blob: 12f09d0ec019b075625fd6a4f016337657c9e151 [file] [log] [blame]
Jens Axboeac9b9102011-10-01 15:01:08 -04001/*
2 * Code related to writing an iolog of what a thread is doing, and to
3 * later read that back and replay
4 */
5#include <stdio.h>
6#include <stdlib.h>
7#include <libgen.h>
8#include <assert.h>
9#include "flist.h"
10#include "fio.h"
11#include "verify.h"
12#include "trim.h"
13
14static const char iolog_ver2[] = "fio version 2 iolog";
15
16void queue_io_piece(struct thread_data *td, struct io_piece *ipo)
17{
18 flist_add_tail(&ipo->list, &td->io_log_list);
19 td->total_io_size += ipo->len;
20}
21
22void log_io_u(struct thread_data *td, struct io_u *io_u)
23{
24 const char *act[] = { "read", "write", "sync", "datasync",
25 "sync_file_range", "wait", "trim" };
26
27 assert(io_u->ddir <= 6);
28
29 if (!td->o.write_iolog_file)
30 return;
31
32 fprintf(td->iolog_f, "%s %s %llu %lu\n", io_u->file->file_name,
33 act[io_u->ddir], io_u->offset,
34 io_u->buflen);
35}
36
37void log_file(struct thread_data *td, struct fio_file *f,
38 enum file_log_act what)
39{
40 const char *act[] = { "add", "open", "close" };
41
42 assert(what < 3);
43
44 if (!td->o.write_iolog_file)
45 return;
46
47
48 /*
49 * this happens on the pre-open/close done before the job starts
50 */
51 if (!td->iolog_f)
52 return;
53
54 fprintf(td->iolog_f, "%s %s\n", f->file_name, act[what]);
55}
56
57static void iolog_delay(struct thread_data *td, unsigned long delay)
58{
59 unsigned long usec = utime_since_now(&td->last_issue);
60
61 if (delay < usec)
62 return;
63
64 delay -= usec;
65
66 /*
67 * less than 100 usec delay, just regard it as noise
68 */
69 if (delay < 100)
70 return;
71
72 usec_sleep(td, delay);
73}
74
75static int ipo_special(struct thread_data *td, struct io_piece *ipo)
76{
77 struct fio_file *f;
78 int ret;
79
80 /*
81 * Not a special ipo
82 */
83 if (ipo->ddir != DDIR_INVAL)
84 return 0;
85
86 f = td->files[ipo->fileno];
87
88 switch (ipo->file_action) {
89 case FIO_LOG_OPEN_FILE:
90 ret = td_io_open_file(td, f);
91 if (!ret)
92 break;
93 td_verror(td, ret, "iolog open file");
94 return -1;
95 case FIO_LOG_CLOSE_FILE:
96 td_io_close_file(td, f);
97 break;
98 case FIO_LOG_UNLINK_FILE:
99 unlink(f->file_name);
100 break;
101 default:
102 log_err("fio: bad file action %d\n", ipo->file_action);
103 break;
104 }
105
106 return 1;
107}
108
109int read_iolog_get(struct thread_data *td, struct io_u *io_u)
110{
111 struct io_piece *ipo;
112 unsigned long elapsed;
113
114 while (!flist_empty(&td->io_log_list)) {
115 int ret;
116
117 ipo = flist_entry(td->io_log_list.next, struct io_piece, list);
118 flist_del(&ipo->list);
119 remove_trim_entry(td, ipo);
120
121 ret = ipo_special(td, ipo);
122 if (ret < 0) {
123 free(ipo);
124 break;
125 } else if (ret > 0) {
126 free(ipo);
127 continue;
128 }
129
130 io_u->ddir = ipo->ddir;
131 if (ipo->ddir != DDIR_WAIT) {
132 io_u->offset = ipo->offset;
133 io_u->buflen = ipo->len;
134 io_u->file = td->files[ipo->fileno];
135 get_file(io_u->file);
136 dprint(FD_IO, "iolog: get %llu/%lu/%s\n", io_u->offset,
137 io_u->buflen, io_u->file->file_name);
138 if (ipo->delay)
139 iolog_delay(td, ipo->delay);
140 } else {
141 elapsed = mtime_since_genesis();
142 if (ipo->delay > elapsed)
143 usec_sleep(td, (ipo->delay - elapsed) * 1000);
144
145 }
146
147 free(ipo);
148
149 if (io_u->ddir != DDIR_WAIT)
150 return 0;
151 }
152
153 td->done = 1;
154 return 1;
155}
156
157void prune_io_piece_log(struct thread_data *td)
158{
159 struct io_piece *ipo;
160 struct rb_node *n;
161
162 while ((n = rb_first(&td->io_hist_tree)) != NULL) {
163 ipo = rb_entry(n, struct io_piece, rb_node);
164 rb_erase(n, &td->io_hist_tree);
165 remove_trim_entry(td, ipo);
166 td->io_hist_len--;
167 free(ipo);
168 }
169
170 while (!flist_empty(&td->io_hist_list)) {
171 ipo = flist_entry(td->io_hist_list.next, struct io_piece, list);
172 flist_del(&ipo->list);
173 remove_trim_entry(td, ipo);
174 td->io_hist_len--;
175 free(ipo);
176 }
177}
178
179/*
180 * log a successful write, so we can unwind the log for verify
181 */
182void log_io_piece(struct thread_data *td, struct io_u *io_u)
183{
184 struct rb_node **p, *parent;
185 struct io_piece *ipo, *__ipo;
186
187 ipo = malloc(sizeof(struct io_piece));
188 init_ipo(ipo);
189 ipo->file = io_u->file;
190 ipo->offset = io_u->offset;
191 ipo->len = io_u->buflen;
192
193 if (io_u_should_trim(td, io_u)) {
194 flist_add_tail(&ipo->trim_list, &td->trim_list);
195 td->trim_entries++;
196 }
197
198 /*
199 * We don't need to sort the entries, if:
200 *
201 * Sequential writes, or
202 * Random writes that lay out the file as it goes along
203 *
204 * For both these cases, just reading back data in the order we
205 * wrote it out is the fastest.
206 *
207 * One exception is if we don't have a random map AND we are doing
208 * verifies, in that case we need to check for duplicate blocks and
209 * drop the old one, which we rely on the rb insert/lookup for
210 * handling.
211 */
212 if ((!td_random(td) || !td->o.overwrite) &&
213 (file_randommap(td, ipo->file) || td->o.verify == VERIFY_NONE)) {
214 INIT_FLIST_HEAD(&ipo->list);
215 flist_add_tail(&ipo->list, &td->io_hist_list);
216 ipo->flags |= IP_F_ONLIST;
217 td->io_hist_len++;
218 return;
219 }
220
221 RB_CLEAR_NODE(&ipo->rb_node);
222
223 /*
224 * Sort the entry into the verification list
225 */
226restart:
227 p = &td->io_hist_tree.rb_node;
228 parent = NULL;
229 while (*p) {
230 parent = *p;
231
232 __ipo = rb_entry(parent, struct io_piece, rb_node);
233 if (ipo->file < __ipo->file)
234 p = &(*p)->rb_left;
235 else if (ipo->file > __ipo->file)
236 p = &(*p)->rb_right;
237 else if (ipo->offset < __ipo->offset)
238 p = &(*p)->rb_left;
239 else if (ipo->offset > __ipo->offset)
240 p = &(*p)->rb_right;
241 else {
Jens Axboe885ac622012-04-04 14:11:58 -0600242 dprint(FD_IO, "iolog: overlap %llu/%lu, %llu/%lu",
243 __ipo->offset, __ipo->len,
244 ipo->offset, ipo->len);
Jens Axboeac9b9102011-10-01 15:01:08 -0400245 td->io_hist_len--;
246 rb_erase(parent, &td->io_hist_tree);
247 remove_trim_entry(td, __ipo);
248 free(__ipo);
249 goto restart;
250 }
251 }
252
253 rb_link_node(&ipo->rb_node, parent, p);
254 rb_insert_color(&ipo->rb_node, &td->io_hist_tree);
255 ipo->flags |= IP_F_ONRB;
256 td->io_hist_len++;
257}
258
259void write_iolog_close(struct thread_data *td)
260{
261 fflush(td->iolog_f);
262 fclose(td->iolog_f);
263 free(td->iolog_buf);
264 td->iolog_f = NULL;
265 td->iolog_buf = NULL;
266}
267
268/*
269 * Read version 2 iolog data. It is enhanced to include per-file logging,
270 * syncs, etc.
271 */
272static int read_iolog2(struct thread_data *td, FILE *f)
273{
274 unsigned long long offset;
275 unsigned int bytes;
276 int reads, writes, waits, fileno = 0, file_action = 0; /* stupid gcc */
277 char *fname, *act;
278 char *str, *p;
279 enum fio_ddir rw;
280
281 free_release_files(td);
282
283 /*
284 * Read in the read iolog and store it, reuse the infrastructure
285 * for doing verifications.
286 */
287 str = malloc(4096);
288 fname = malloc(256+16);
289 act = malloc(256+16);
290
291 reads = writes = waits = 0;
292 while ((p = fgets(str, 4096, f)) != NULL) {
293 struct io_piece *ipo;
294 int r;
295
296 r = sscanf(p, "%256s %256s %llu %u", fname, act, &offset,
297 &bytes);
298 if (r == 4) {
299 /*
300 * Check action first
301 */
302 if (!strcmp(act, "wait"))
303 rw = DDIR_WAIT;
304 else if (!strcmp(act, "read"))
305 rw = DDIR_READ;
306 else if (!strcmp(act, "write"))
307 rw = DDIR_WRITE;
308 else if (!strcmp(act, "sync"))
309 rw = DDIR_SYNC;
310 else if (!strcmp(act, "datasync"))
311 rw = DDIR_DATASYNC;
312 else if (!strcmp(act, "trim"))
313 rw = DDIR_TRIM;
314 else {
315 log_err("fio: bad iolog file action: %s\n",
316 act);
317 continue;
318 }
319 } else if (r == 2) {
320 rw = DDIR_INVAL;
321 if (!strcmp(act, "add")) {
322 td->o.nr_files++;
323 fileno = add_file(td, fname);
324 file_action = FIO_LOG_ADD_FILE;
325 continue;
326 } else if (!strcmp(act, "open")) {
327 fileno = get_fileno(td, fname);
328 file_action = FIO_LOG_OPEN_FILE;
329 } else if (!strcmp(act, "close")) {
330 fileno = get_fileno(td, fname);
331 file_action = FIO_LOG_CLOSE_FILE;
332 } else {
333 log_err("fio: bad iolog file action: %s\n",
334 act);
335 continue;
336 }
337 } else {
338 log_err("bad iolog2: %s", p);
339 continue;
340 }
341
342 if (rw == DDIR_READ)
343 reads++;
344 else if (rw == DDIR_WRITE) {
345 /*
346 * Don't add a write for ro mode
347 */
348 if (read_only)
349 continue;
350 writes++;
351 } else if (rw == DDIR_WAIT) {
352 waits++;
353 } else if (rw == DDIR_INVAL) {
354 } else if (!ddir_sync(rw)) {
355 log_err("bad ddir: %d\n", rw);
356 continue;
357 }
358
359 /*
360 * Make note of file
361 */
362 ipo = malloc(sizeof(*ipo));
363 init_ipo(ipo);
364 ipo->ddir = rw;
365 if (rw == DDIR_WAIT) {
366 ipo->delay = offset;
367 } else {
368 ipo->offset = offset;
369 ipo->len = bytes;
370 if (bytes > td->o.max_bs[rw])
371 td->o.max_bs[rw] = bytes;
372 ipo->fileno = fileno;
373 ipo->file_action = file_action;
374 }
375
376 queue_io_piece(td, ipo);
377 }
378
379 free(str);
380 free(act);
381 free(fname);
382
383 if (writes && read_only) {
384 log_err("fio: <%s> skips replay of %d writes due to"
385 " read-only\n", td->o.name, writes);
386 writes = 0;
387 }
388
389 if (!reads && !writes && !waits)
390 return 1;
391 else if (reads && !writes)
392 td->o.td_ddir = TD_DDIR_READ;
393 else if (!reads && writes)
394 td->o.td_ddir = TD_DDIR_WRITE;
395 else
396 td->o.td_ddir = TD_DDIR_RW;
397
398 return 0;
399}
400
401/*
402 * open iolog, check version, and call appropriate parser
403 */
404static int init_iolog_read(struct thread_data *td)
405{
406 char buffer[256], *p;
407 FILE *f;
408 int ret;
409
410 f = fopen(td->o.read_iolog_file, "r");
411 if (!f) {
412 perror("fopen read iolog");
413 return 1;
414 }
415
416 p = fgets(buffer, sizeof(buffer), f);
417 if (!p) {
418 td_verror(td, errno, "iolog read");
419 log_err("fio: unable to read iolog\n");
420 fclose(f);
421 return 1;
422 }
423
424 /*
425 * version 2 of the iolog stores a specific string as the
426 * first line, check for that
427 */
428 if (!strncmp(iolog_ver2, buffer, strlen(iolog_ver2)))
429 ret = read_iolog2(td, f);
430 else {
431 log_err("fio: iolog version 1 is no longer supported\n");
432 ret = 1;
433 }
434
435 fclose(f);
436 return ret;
437}
438
439/*
440 * Set up a log for storing io patterns.
441 */
442static int init_iolog_write(struct thread_data *td)
443{
444 struct fio_file *ff;
445 FILE *f;
446 unsigned int i;
447
448 f = fopen(td->o.write_iolog_file, "a");
449 if (!f) {
450 perror("fopen write iolog");
451 return 1;
452 }
453
454 /*
455 * That's it for writing, setup a log buffer and we're done.
456 */
457 td->iolog_f = f;
458 td->iolog_buf = malloc(8192);
459 setvbuf(f, td->iolog_buf, _IOFBF, 8192);
460
461 /*
462 * write our version line
463 */
464 if (fprintf(f, "%s\n", iolog_ver2) < 0) {
465 perror("iolog init\n");
466 return 1;
467 }
468
469 /*
470 * add all known files
471 */
472 for_each_file(td, ff, i)
473 log_file(td, ff, FIO_LOG_ADD_FILE);
474
475 return 0;
476}
477
478int init_iolog(struct thread_data *td)
479{
480 int ret = 0;
481
482 if (td->o.read_iolog_file) {
483 /*
484 * Check if it's a blktrace file and load that if possible.
485 * Otherwise assume it's a normal log file and load that.
486 */
487 if (is_blktrace(td->o.read_iolog_file))
488 ret = load_blktrace(td, td->o.read_iolog_file);
489 else
490 ret = init_iolog_read(td);
491 } else if (td->o.write_iolog_file)
492 ret = init_iolog_write(td);
493
494 return ret;
495}
496
Jens Axboeb8bc8cb2011-12-01 09:04:31 +0100497void setup_log(struct io_log **log, unsigned long avg_msec)
Jens Axboeac9b9102011-10-01 15:01:08 -0400498{
499 struct io_log *l = malloc(sizeof(*l));
500
Jens Axboeb8bc8cb2011-12-01 09:04:31 +0100501 memset(l, 0, sizeof(*l));
Jens Axboeac9b9102011-10-01 15:01:08 -0400502 l->nr_samples = 0;
503 l->max_samples = 1024;
504 l->log = malloc(l->max_samples * sizeof(struct io_sample));
Jens Axboeb8bc8cb2011-12-01 09:04:31 +0100505 l->avg_msec = avg_msec;
Jens Axboeac9b9102011-10-01 15:01:08 -0400506 *log = l;
507}
508
509void __finish_log(struct io_log *log, const char *name)
510{
511 unsigned int i;
512 FILE *f;
513
514 f = fopen(name, "a");
515 if (!f) {
516 perror("fopen log");
517 return;
518 }
519
520 for (i = 0; i < log->nr_samples; i++) {
521 fprintf(f, "%lu, %lu, %u, %u\n", log->log[i].time,
522 log->log[i].val,
523 log->log[i].ddir,
524 log->log[i].bs);
525 }
526
527 fclose(f);
528 free(log->log);
529 free(log);
530}
531
532void finish_log_named(struct thread_data *td, struct io_log *log,
533 const char *prefix, const char *postfix)
534{
535 char file_name[256], *p;
536
537 snprintf(file_name, 200, "%s_%s.log", prefix, postfix);
538 p = basename(file_name);
539 __finish_log(log, p);
540}
541
542void finish_log(struct thread_data *td, struct io_log *log, const char *name)
543{
544 finish_log_named(td, log, td->o.name, name);
545}