server: fix wrong use of exit() in server when daemonized
[fio.git] / init.c
... / ...
CommitLineData
1/*
2 * This file contains job initialization and setup functions.
3 */
4#include <stdio.h>
5#include <stdlib.h>
6#include <unistd.h>
7#include <fcntl.h>
8#include <ctype.h>
9#include <string.h>
10#include <errno.h>
11#include <sys/ipc.h>
12#include <sys/types.h>
13#include <sys/stat.h>
14
15#include "fio.h"
16#ifndef FIO_NO_HAVE_SHM_H
17#include <sys/shm.h>
18#endif
19
20#include "parse.h"
21#include "smalloc.h"
22#include "filehash.h"
23#include "verify.h"
24#include "profile.h"
25#include "server.h"
26#include "idletime.h"
27#include "filelock.h"
28
29#include "lib/getopt.h"
30#include "lib/strcasestr.h"
31
32#include "crc/test.h"
33
34const char fio_version_string[] = FIO_VERSION;
35
36#define FIO_RANDSEED (0xb1899bedUL)
37
38static char **ini_file;
39static int max_jobs = FIO_MAX_JOBS;
40static int dump_cmdline;
41static long long def_timeout;
42static int parse_only;
43
44static struct thread_data def_thread;
45struct thread_data *threads = NULL;
46static char **job_sections;
47static int nr_job_sections;
48
49int exitall_on_terminate = 0;
50int output_format = FIO_OUTPUT_NORMAL;
51int append_terse_output = 0;
52int eta_print = FIO_ETA_AUTO;
53int eta_new_line = 0;
54FILE *f_out = NULL;
55FILE *f_err = NULL;
56char *exec_profile = NULL;
57int warnings_fatal = 0;
58int terse_version = 3;
59int is_backend = 0;
60int nr_clients = 0;
61int log_syslog = 0;
62
63int write_bw_log = 0;
64int read_only = 0;
65int status_interval = 0;
66
67static int prev_group_jobs;
68
69unsigned long fio_debug = 0;
70unsigned int fio_debug_jobno = -1;
71unsigned int *fio_debug_jobp = NULL;
72
73static char cmd_optstr[256];
74static int did_arg;
75
76#define FIO_CLIENT_FLAG (1 << 16)
77
78/*
79 * Command line options. These will contain the above, plus a few
80 * extra that only pertain to fio itself and not jobs.
81 */
82static struct option l_opts[FIO_NR_OPTIONS] = {
83 {
84 .name = (char *) "output",
85 .has_arg = required_argument,
86 .val = 'o' | FIO_CLIENT_FLAG,
87 },
88 {
89 .name = (char *) "timeout",
90 .has_arg = required_argument,
91 .val = 't' | FIO_CLIENT_FLAG,
92 },
93 {
94 .name = (char *) "latency-log",
95 .has_arg = required_argument,
96 .val = 'l' | FIO_CLIENT_FLAG,
97 },
98 {
99 .name = (char *) "bandwidth-log",
100 .has_arg = required_argument,
101 .val = 'b' | FIO_CLIENT_FLAG,
102 },
103 {
104 .name = (char *) "minimal",
105 .has_arg = no_argument,
106 .val = 'm' | FIO_CLIENT_FLAG,
107 },
108 {
109 .name = (char *) "output-format",
110 .has_arg = optional_argument,
111 .val = 'F' | FIO_CLIENT_FLAG,
112 },
113 {
114 .name = (char *) "append-terse",
115 .has_arg = optional_argument,
116 .val = 'f',
117 },
118 {
119 .name = (char *) "version",
120 .has_arg = no_argument,
121 .val = 'v' | FIO_CLIENT_FLAG,
122 },
123 {
124 .name = (char *) "help",
125 .has_arg = no_argument,
126 .val = 'h' | FIO_CLIENT_FLAG,
127 },
128 {
129 .name = (char *) "cmdhelp",
130 .has_arg = optional_argument,
131 .val = 'c' | FIO_CLIENT_FLAG,
132 },
133 {
134 .name = (char *) "enghelp",
135 .has_arg = optional_argument,
136 .val = 'i' | FIO_CLIENT_FLAG,
137 },
138 {
139 .name = (char *) "showcmd",
140 .has_arg = no_argument,
141 .val = 's' | FIO_CLIENT_FLAG,
142 },
143 {
144 .name = (char *) "readonly",
145 .has_arg = no_argument,
146 .val = 'r' | FIO_CLIENT_FLAG,
147 },
148 {
149 .name = (char *) "eta",
150 .has_arg = required_argument,
151 .val = 'e' | FIO_CLIENT_FLAG,
152 },
153 {
154 .name = (char *) "eta-newline",
155 .has_arg = required_argument,
156 .val = 'E' | FIO_CLIENT_FLAG,
157 },
158 {
159 .name = (char *) "debug",
160 .has_arg = required_argument,
161 .val = 'd' | FIO_CLIENT_FLAG,
162 },
163 {
164 .name = (char *) "parse-only",
165 .has_arg = no_argument,
166 .val = 'P' | FIO_CLIENT_FLAG,
167 },
168 {
169 .name = (char *) "section",
170 .has_arg = required_argument,
171 .val = 'x' | FIO_CLIENT_FLAG,
172 },
173#ifdef CONFIG_ZLIB
174 {
175 .name = (char *) "inflate-log",
176 .has_arg = required_argument,
177 .val = 'X' | FIO_CLIENT_FLAG,
178 },
179#endif
180 {
181 .name = (char *) "alloc-size",
182 .has_arg = required_argument,
183 .val = 'a' | FIO_CLIENT_FLAG,
184 },
185 {
186 .name = (char *) "profile",
187 .has_arg = required_argument,
188 .val = 'p' | FIO_CLIENT_FLAG,
189 },
190 {
191 .name = (char *) "warnings-fatal",
192 .has_arg = no_argument,
193 .val = 'w' | FIO_CLIENT_FLAG,
194 },
195 {
196 .name = (char *) "max-jobs",
197 .has_arg = required_argument,
198 .val = 'j' | FIO_CLIENT_FLAG,
199 },
200 {
201 .name = (char *) "terse-version",
202 .has_arg = required_argument,
203 .val = 'V' | FIO_CLIENT_FLAG,
204 },
205 {
206 .name = (char *) "server",
207 .has_arg = optional_argument,
208 .val = 'S',
209 },
210 { .name = (char *) "daemonize",
211 .has_arg = required_argument,
212 .val = 'D',
213 },
214 {
215 .name = (char *) "client",
216 .has_arg = required_argument,
217 .val = 'C',
218 },
219 {
220 .name = (char *) "cpuclock-test",
221 .has_arg = no_argument,
222 .val = 'T',
223 },
224 {
225 .name = (char *) "crctest",
226 .has_arg = optional_argument,
227 .val = 'G',
228 },
229 {
230 .name = (char *) "idle-prof",
231 .has_arg = required_argument,
232 .val = 'I',
233 },
234 {
235 .name = (char *) "status-interval",
236 .has_arg = required_argument,
237 .val = 'L',
238 },
239 {
240 .name = NULL,
241 },
242};
243
244void free_threads_shm(void)
245{
246 if (threads) {
247 void *tp = threads;
248#ifndef CONFIG_NO_SHM
249 struct shmid_ds sbuf;
250
251 threads = NULL;
252 shmdt(tp);
253 shmctl(shm_id, IPC_RMID, &sbuf);
254 shm_id = -1;
255#else
256 threads = NULL;
257 free(tp);
258#endif
259 }
260}
261
262static void free_shm(void)
263{
264 if (threads) {
265 file_hash_exit();
266 flow_exit();
267 fio_debug_jobp = NULL;
268 free_threads_shm();
269 }
270
271 options_free(fio_options, &def_thread);
272 fio_filelock_exit();
273 scleanup();
274}
275
276/*
277 * The thread area is shared between the main process and the job
278 * threads/processes. So setup a shared memory segment that will hold
279 * all the job info. We use the end of the region for keeping track of
280 * open files across jobs, for file sharing.
281 */
282static int setup_thread_area(void)
283{
284 void *hash;
285
286 if (threads)
287 return 0;
288
289 /*
290 * 1024 is too much on some machines, scale max_jobs if
291 * we get a failure that looks like too large a shm segment
292 */
293 do {
294 size_t size = max_jobs * sizeof(struct thread_data);
295
296 size += file_hash_size;
297 size += sizeof(unsigned int);
298
299#ifndef CONFIG_NO_SHM
300 shm_id = shmget(0, size, IPC_CREAT | 0600);
301 if (shm_id != -1)
302 break;
303 if (errno != EINVAL && errno != ENOMEM && errno != ENOSPC) {
304 perror("shmget");
305 break;
306 }
307#else
308 threads = malloc(size);
309 if (threads)
310 break;
311#endif
312
313 max_jobs >>= 1;
314 } while (max_jobs);
315
316#ifndef CONFIG_NO_SHM
317 if (shm_id == -1)
318 return 1;
319
320 threads = shmat(shm_id, NULL, 0);
321 if (threads == (void *) -1) {
322 perror("shmat");
323 return 1;
324 }
325#endif
326
327 memset(threads, 0, max_jobs * sizeof(struct thread_data));
328 hash = (void *) threads + max_jobs * sizeof(struct thread_data);
329 fio_debug_jobp = (void *) hash + file_hash_size;
330 *fio_debug_jobp = -1;
331 file_hash_init(hash);
332
333 flow_init();
334
335 return 0;
336}
337
338static void set_cmd_options(struct thread_data *td)
339{
340 struct thread_options *o = &td->o;
341
342 if (!o->timeout)
343 o->timeout = def_timeout;
344}
345
346/*
347 * Return a free job structure.
348 */
349static struct thread_data *get_new_job(int global, struct thread_data *parent,
350 int preserve_eo, const char *jobname)
351{
352 struct thread_data *td;
353
354 if (global) {
355 set_cmd_options(&def_thread);
356 return &def_thread;
357 }
358 if (setup_thread_area()) {
359 log_err("error: failed to setup shm segment\n");
360 return NULL;
361 }
362 if (thread_number >= max_jobs) {
363 log_err("error: maximum number of jobs (%d) reached.\n",
364 max_jobs);
365 return NULL;
366 }
367
368 td = &threads[thread_number++];
369 *td = *parent;
370
371 td->io_ops = NULL;
372 if (!preserve_eo)
373 td->eo = NULL;
374
375 td->o.uid = td->o.gid = -1U;
376
377 dup_files(td, parent);
378 fio_options_mem_dupe(td);
379
380 profile_add_hooks(td);
381
382 td->thread_number = thread_number;
383 td->subjob_number = 0;
384
385 if (jobname)
386 td->o.name = strdup(jobname);
387
388 if (!parent->o.group_reporting)
389 stat_number++;
390
391 set_cmd_options(td);
392 return td;
393}
394
395static void put_job(struct thread_data *td)
396{
397 if (td == &def_thread)
398 return;
399
400 profile_td_exit(td);
401 flow_exit_job(td);
402
403 if (td->error)
404 log_info("fio: %s\n", td->verror);
405
406 fio_options_free(td);
407 if (td->io_ops)
408 free_ioengine(td);
409
410 if (td->o.name)
411 free(td->o.name);
412
413 memset(&threads[td->thread_number - 1], 0, sizeof(*td));
414 thread_number--;
415}
416
417static int __setup_rate(struct thread_data *td, enum fio_ddir ddir)
418{
419 unsigned int bs = td->o.min_bs[ddir];
420
421 assert(ddir_rw(ddir));
422
423 if (td->o.rate[ddir])
424 td->rate_bps[ddir] = td->o.rate[ddir];
425 else
426 td->rate_bps[ddir] = td->o.rate_iops[ddir] * bs;
427
428 if (!td->rate_bps[ddir]) {
429 log_err("rate lower than supported\n");
430 return -1;
431 }
432
433 td->rate_pending_usleep[ddir] = 0;
434 return 0;
435}
436
437static int setup_rate(struct thread_data *td)
438{
439 int ret = 0;
440
441 if (td->o.rate[DDIR_READ] || td->o.rate_iops[DDIR_READ])
442 ret = __setup_rate(td, DDIR_READ);
443 if (td->o.rate[DDIR_WRITE] || td->o.rate_iops[DDIR_WRITE])
444 ret |= __setup_rate(td, DDIR_WRITE);
445 if (td->o.rate[DDIR_TRIM] || td->o.rate_iops[DDIR_TRIM])
446 ret |= __setup_rate(td, DDIR_TRIM);
447
448 return ret;
449}
450
451static int fixed_block_size(struct thread_options *o)
452{
453 return o->min_bs[DDIR_READ] == o->max_bs[DDIR_READ] &&
454 o->min_bs[DDIR_WRITE] == o->max_bs[DDIR_WRITE] &&
455 o->min_bs[DDIR_TRIM] == o->max_bs[DDIR_TRIM] &&
456 o->min_bs[DDIR_READ] == o->min_bs[DDIR_WRITE] &&
457 o->min_bs[DDIR_READ] == o->min_bs[DDIR_TRIM];
458}
459
460
461static unsigned long long get_rand_start_delay(struct thread_data *td)
462{
463 unsigned long long delayrange;
464 unsigned long r;
465
466 delayrange = td->o.start_delay_high - td->o.start_delay;
467
468 if (td->o.use_os_rand) {
469 r = os_random_long(&td->delay_state);
470 delayrange = (unsigned long long) ((double) delayrange * (r / (OS_RAND_MAX + 1.0)));
471 } else {
472 r = __rand(&td->__delay_state);
473 delayrange = (unsigned long long) ((double) delayrange * (r / (FRAND_MAX + 1.0)));
474 }
475
476 delayrange += td->o.start_delay;
477 return delayrange;
478}
479
480/*
481 * Lazy way of fixing up options that depend on each other. We could also
482 * define option callback handlers, but this is easier.
483 */
484static int fixup_options(struct thread_data *td)
485{
486 struct thread_options *o = &td->o;
487 int ret = 0;
488
489#ifndef FIO_HAVE_PSHARED_MUTEX
490 if (!o->use_thread) {
491 log_info("fio: this platform does not support process shared"
492 " mutexes, forcing use of threads. Use the 'thread'"
493 " option to get rid of this warning.\n");
494 o->use_thread = 1;
495 ret = warnings_fatal;
496 }
497#endif
498
499 if (o->write_iolog_file && o->read_iolog_file) {
500 log_err("fio: read iolog overrides write_iolog\n");
501 free(o->write_iolog_file);
502 o->write_iolog_file = NULL;
503 ret = warnings_fatal;
504 }
505
506 /*
507 * only really works with 1 file
508 */
509 if (o->zone_size && o->open_files > 1)
510 o->zone_size = 0;
511
512 /*
513 * If zone_range isn't specified, backward compatibility dictates it
514 * should be made equal to zone_size.
515 */
516 if (o->zone_size && !o->zone_range)
517 o->zone_range = o->zone_size;
518
519 /*
520 * Reads can do overwrites, we always need to pre-create the file
521 */
522 if (td_read(td) || td_rw(td))
523 o->overwrite = 1;
524
525 if (!o->min_bs[DDIR_READ])
526 o->min_bs[DDIR_READ] = o->bs[DDIR_READ];
527 if (!o->max_bs[DDIR_READ])
528 o->max_bs[DDIR_READ] = o->bs[DDIR_READ];
529 if (!o->min_bs[DDIR_WRITE])
530 o->min_bs[DDIR_WRITE] = o->bs[DDIR_WRITE];
531 if (!o->max_bs[DDIR_WRITE])
532 o->max_bs[DDIR_WRITE] = o->bs[DDIR_WRITE];
533 if (!o->min_bs[DDIR_TRIM])
534 o->min_bs[DDIR_TRIM] = o->bs[DDIR_TRIM];
535 if (!o->max_bs[DDIR_TRIM])
536 o->max_bs[DDIR_TRIM] = o->bs[DDIR_TRIM];
537
538
539 o->rw_min_bs = min(o->min_bs[DDIR_READ], o->min_bs[DDIR_WRITE]);
540 o->rw_min_bs = min(o->min_bs[DDIR_TRIM], o->rw_min_bs);
541
542 /*
543 * For random IO, allow blockalign offset other than min_bs.
544 */
545 if (!o->ba[DDIR_READ] || !td_random(td))
546 o->ba[DDIR_READ] = o->min_bs[DDIR_READ];
547 if (!o->ba[DDIR_WRITE] || !td_random(td))
548 o->ba[DDIR_WRITE] = o->min_bs[DDIR_WRITE];
549 if (!o->ba[DDIR_TRIM] || !td_random(td))
550 o->ba[DDIR_TRIM] = o->min_bs[DDIR_TRIM];
551
552 if ((o->ba[DDIR_READ] != o->min_bs[DDIR_READ] ||
553 o->ba[DDIR_WRITE] != o->min_bs[DDIR_WRITE] ||
554 o->ba[DDIR_TRIM] != o->min_bs[DDIR_TRIM]) &&
555 !o->norandommap) {
556 log_err("fio: Any use of blockalign= turns off randommap\n");
557 o->norandommap = 1;
558 ret = warnings_fatal;
559 }
560
561 if (!o->file_size_high)
562 o->file_size_high = o->file_size_low;
563
564 if (o->start_delay_high)
565 o->start_delay = get_rand_start_delay(td);
566
567 if (o->norandommap && o->verify != VERIFY_NONE
568 && !fixed_block_size(o)) {
569 log_err("fio: norandommap given for variable block sizes, "
570 "verify disabled\n");
571 o->verify = VERIFY_NONE;
572 ret = warnings_fatal;
573 }
574 if (o->bs_unaligned && (o->odirect || td->io_ops->flags & FIO_RAWIO))
575 log_err("fio: bs_unaligned may not work with raw io\n");
576
577 /*
578 * thinktime_spin must be less than thinktime
579 */
580 if (o->thinktime_spin > o->thinktime)
581 o->thinktime_spin = o->thinktime;
582
583 /*
584 * The low water mark cannot be bigger than the iodepth
585 */
586 if (o->iodepth_low > o->iodepth || !o->iodepth_low)
587 o->iodepth_low = o->iodepth;
588
589 /*
590 * If batch number isn't set, default to the same as iodepth
591 */
592 if (o->iodepth_batch > o->iodepth || !o->iodepth_batch)
593 o->iodepth_batch = o->iodepth;
594
595 if (o->nr_files > td->files_index)
596 o->nr_files = td->files_index;
597
598 if (o->open_files > o->nr_files || !o->open_files)
599 o->open_files = o->nr_files;
600
601 if (((o->rate[DDIR_READ] + o->rate[DDIR_WRITE] + o->rate[DDIR_TRIM]) &&
602 (o->rate_iops[DDIR_READ] + o->rate_iops[DDIR_WRITE] + o->rate_iops[DDIR_TRIM])) ||
603 ((o->ratemin[DDIR_READ] + o->ratemin[DDIR_WRITE] + o->ratemin[DDIR_TRIM]) &&
604 (o->rate_iops_min[DDIR_READ] + o->rate_iops_min[DDIR_WRITE] + o->rate_iops_min[DDIR_TRIM]))) {
605 log_err("fio: rate and rate_iops are mutually exclusive\n");
606 ret = 1;
607 }
608 if ((o->rate[DDIR_READ] < o->ratemin[DDIR_READ]) ||
609 (o->rate[DDIR_WRITE] < o->ratemin[DDIR_WRITE]) ||
610 (o->rate[DDIR_TRIM] < o->ratemin[DDIR_TRIM]) ||
611 (o->rate_iops[DDIR_READ] < o->rate_iops_min[DDIR_READ]) ||
612 (o->rate_iops[DDIR_WRITE] < o->rate_iops_min[DDIR_WRITE]) ||
613 (o->rate_iops[DDIR_TRIM] < o->rate_iops_min[DDIR_TRIM])) {
614 log_err("fio: minimum rate exceeds rate\n");
615 ret = 1;
616 }
617
618 if (!o->timeout && o->time_based) {
619 log_err("fio: time_based requires a runtime/timeout setting\n");
620 o->time_based = 0;
621 ret = warnings_fatal;
622 }
623
624 if (o->fill_device && !o->size)
625 o->size = -1ULL;
626
627 if (o->verify != VERIFY_NONE) {
628 if (td_write(td) && o->do_verify && o->numjobs > 1) {
629 log_info("Multiple writers may overwrite blocks that "
630 "belong to other jobs. This can cause "
631 "verification failures.\n");
632 ret = warnings_fatal;
633 }
634
635 o->refill_buffers = 1;
636 if (o->max_bs[DDIR_WRITE] != o->min_bs[DDIR_WRITE] &&
637 !o->verify_interval)
638 o->verify_interval = o->min_bs[DDIR_WRITE];
639
640 /*
641 * Verify interval must be smaller or equal to the
642 * write size.
643 */
644 if (o->verify_interval > o->min_bs[DDIR_WRITE])
645 o->verify_interval = o->min_bs[DDIR_WRITE];
646 else if (td_read(td) && o->verify_interval > o->min_bs[DDIR_READ])
647 o->verify_interval = o->min_bs[DDIR_READ];
648 }
649
650 if (o->pre_read) {
651 o->invalidate_cache = 0;
652 if (td->io_ops->flags & FIO_PIPEIO) {
653 log_info("fio: cannot pre-read files with an IO engine"
654 " that isn't seekable. Pre-read disabled.\n");
655 ret = warnings_fatal;
656 }
657 }
658
659 if (!o->unit_base) {
660 if (td->io_ops->flags & FIO_BIT_BASED)
661 o->unit_base = 1;
662 else
663 o->unit_base = 8;
664 }
665
666#ifndef CONFIG_FDATASYNC
667 if (o->fdatasync_blocks) {
668 log_info("fio: this platform does not support fdatasync()"
669 " falling back to using fsync(). Use the 'fsync'"
670 " option instead of 'fdatasync' to get rid of"
671 " this warning\n");
672 o->fsync_blocks = o->fdatasync_blocks;
673 o->fdatasync_blocks = 0;
674 ret = warnings_fatal;
675 }
676#endif
677
678#ifdef WIN32
679 /*
680 * Windows doesn't support O_DIRECT or O_SYNC with the _open interface,
681 * so fail if we're passed those flags
682 */
683 if ((td->io_ops->flags & FIO_SYNCIO) && (td->o.odirect || td->o.sync_io)) {
684 log_err("fio: Windows does not support direct or non-buffered io with"
685 " the synchronous ioengines. Use the 'windowsaio' ioengine"
686 " with 'direct=1' and 'iodepth=1' instead.\n");
687 ret = 1;
688 }
689#endif
690
691 /*
692 * For fully compressible data, just zero them at init time.
693 * It's faster than repeatedly filling it.
694 */
695 if (td->o.compress_percentage == 100) {
696 td->o.zero_buffers = 1;
697 td->o.compress_percentage = 0;
698 }
699
700 /*
701 * Using a non-uniform random distribution excludes usage of
702 * a random map
703 */
704 if (td->o.random_distribution != FIO_RAND_DIST_RANDOM)
705 td->o.norandommap = 1;
706
707 /*
708 * If size is set but less than the min block size, complain
709 */
710 if (o->size && o->size < td_min_bs(td)) {
711 log_err("fio: size too small, must be larger than the IO size: %llu\n", (unsigned long long) o->size);
712 ret = 1;
713 }
714
715 /*
716 * O_ATOMIC implies O_DIRECT
717 */
718 if (td->o.oatomic)
719 td->o.odirect = 1;
720
721 /*
722 * If randseed is set, that overrides randrepeat
723 */
724 if (td->o.rand_seed)
725 td->o.rand_repeatable = 0;
726
727 if ((td->io_ops->flags & FIO_NOEXTEND) && td->o.file_append) {
728 log_err("fio: can't append/extent with IO engine %s\n", td->io_ops->name);
729 ret = 1;
730 }
731
732 return ret;
733}
734
735/*
736 * This function leaks the buffer
737 */
738char *fio_uint_to_kmg(unsigned int val)
739{
740 char *buf = malloc(32);
741 char post[] = { 0, 'K', 'M', 'G', 'P', 'E', 0 };
742 char *p = post;
743
744 do {
745 if (val & 1023)
746 break;
747
748 val >>= 10;
749 p++;
750 } while (*p);
751
752 snprintf(buf, 32, "%u%c", val, *p);
753 return buf;
754}
755
756/* External engines are specified by "external:name.o") */
757static const char *get_engine_name(const char *str)
758{
759 char *p = strstr(str, ":");
760
761 if (!p)
762 return str;
763
764 p++;
765 strip_blank_front(&p);
766 strip_blank_end(p);
767 return p;
768}
769
770static int exists_and_not_file(const char *filename)
771{
772 struct stat sb;
773
774 if (lstat(filename, &sb) == -1)
775 return 0;
776
777 /* \\.\ is the device namespace in Windows, where every file
778 * is a device node */
779 if (S_ISREG(sb.st_mode) && strncmp(filename, "\\\\.\\", 4) != 0)
780 return 0;
781
782 return 1;
783}
784
785static void td_fill_rand_seeds_os(struct thread_data *td)
786{
787 os_random_seed(td->rand_seeds[FIO_RAND_BS_OFF], &td->bsrange_state);
788 os_random_seed(td->rand_seeds[FIO_RAND_VER_OFF], &td->verify_state);
789 os_random_seed(td->rand_seeds[FIO_RAND_MIX_OFF], &td->rwmix_state);
790
791 if (td->o.file_service_type == FIO_FSERVICE_RANDOM)
792 os_random_seed(td->rand_seeds[FIO_RAND_FILE_OFF], &td->next_file_state);
793
794 os_random_seed(td->rand_seeds[FIO_RAND_FILE_SIZE_OFF], &td->file_size_state);
795 os_random_seed(td->rand_seeds[FIO_RAND_TRIM_OFF], &td->trim_state);
796 os_random_seed(td->rand_seeds[FIO_RAND_START_DELAY], &td->delay_state);
797
798 if (!td_random(td))
799 return;
800
801 if (td->o.rand_repeatable)
802 td->rand_seeds[FIO_RAND_BLOCK_OFF] = FIO_RANDSEED * td->thread_number;
803
804 os_random_seed(td->rand_seeds[FIO_RAND_BLOCK_OFF], &td->random_state);
805
806 os_random_seed(td->rand_seeds[FIO_RAND_SEQ_RAND_READ_OFF], &td->seq_rand_state[DDIR_READ]);
807 os_random_seed(td->rand_seeds[FIO_RAND_SEQ_RAND_WRITE_OFF], &td->seq_rand_state[DDIR_WRITE]);
808 os_random_seed(td->rand_seeds[FIO_RAND_SEQ_RAND_TRIM_OFF], &td->seq_rand_state[DDIR_TRIM]);
809}
810
811static void td_fill_rand_seeds_internal(struct thread_data *td)
812{
813 init_rand_seed(&td->__bsrange_state, td->rand_seeds[FIO_RAND_BS_OFF]);
814 init_rand_seed(&td->__verify_state, td->rand_seeds[FIO_RAND_VER_OFF]);
815 init_rand_seed(&td->__rwmix_state, td->rand_seeds[FIO_RAND_MIX_OFF]);
816
817 if (td->o.file_service_type == FIO_FSERVICE_RANDOM)
818 init_rand_seed(&td->__next_file_state, td->rand_seeds[FIO_RAND_FILE_OFF]);
819
820 init_rand_seed(&td->__file_size_state, td->rand_seeds[FIO_RAND_FILE_SIZE_OFF]);
821 init_rand_seed(&td->__trim_state, td->rand_seeds[FIO_RAND_TRIM_OFF]);
822 init_rand_seed(&td->__delay_state, td->rand_seeds[FIO_RAND_START_DELAY]);
823
824 if (!td_random(td))
825 return;
826
827 if (td->o.rand_repeatable)
828 td->rand_seeds[FIO_RAND_BLOCK_OFF] = FIO_RANDSEED * td->thread_number;
829
830 init_rand_seed(&td->__random_state, td->rand_seeds[FIO_RAND_BLOCK_OFF]);
831 init_rand_seed(&td->__seq_rand_state[DDIR_READ], td->rand_seeds[FIO_RAND_SEQ_RAND_READ_OFF]);
832 init_rand_seed(&td->__seq_rand_state[DDIR_WRITE], td->rand_seeds[FIO_RAND_SEQ_RAND_WRITE_OFF]);
833 init_rand_seed(&td->__seq_rand_state[DDIR_TRIM], td->rand_seeds[FIO_RAND_SEQ_RAND_TRIM_OFF]);
834}
835
836void td_fill_rand_seeds(struct thread_data *td)
837{
838 if (td->o.allrand_repeatable) {
839 for (int i = 0; i < FIO_RAND_NR_OFFS; i++)
840 td->rand_seeds[i] = FIO_RANDSEED * td->thread_number
841 + i;
842 }
843
844 if (td->o.use_os_rand)
845 td_fill_rand_seeds_os(td);
846 else
847 td_fill_rand_seeds_internal(td);
848
849 init_rand_seed(&td->buf_state, td->rand_seeds[FIO_RAND_BUF_OFF]);
850}
851
852/*
853 * Initializes the ioengine configured for a job, if it has not been done so
854 * already.
855 */
856int ioengine_load(struct thread_data *td)
857{
858 const char *engine;
859
860 /*
861 * Engine has already been loaded.
862 */
863 if (td->io_ops)
864 return 0;
865 if (!td->o.ioengine) {
866 log_err("fio: internal fault, no IO engine specified\n");
867 return 1;
868 }
869
870 engine = get_engine_name(td->o.ioengine);
871 td->io_ops = load_ioengine(td, engine);
872 if (!td->io_ops) {
873 log_err("fio: failed to load engine %s\n", engine);
874 return 1;
875 }
876
877 if (td->io_ops->option_struct_size && td->io_ops->options) {
878 /*
879 * In cases where td->eo is set, clone it for a child thread.
880 * This requires that the parent thread has the same ioengine,
881 * but that requirement must be enforced by the code which
882 * cloned the thread.
883 */
884 void *origeo = td->eo;
885 /*
886 * Otherwise use the default thread options.
887 */
888 if (!origeo && td != &def_thread && def_thread.eo &&
889 def_thread.io_ops->options == td->io_ops->options)
890 origeo = def_thread.eo;
891
892 options_init(td->io_ops->options);
893 td->eo = malloc(td->io_ops->option_struct_size);
894 /*
895 * Use the default thread as an option template if this uses the
896 * same options structure and there are non-default options
897 * used.
898 */
899 if (origeo) {
900 memcpy(td->eo, origeo, td->io_ops->option_struct_size);
901 options_mem_dupe(td->eo, td->io_ops->options);
902 } else {
903 memset(td->eo, 0, td->io_ops->option_struct_size);
904 fill_default_options(td->eo, td->io_ops->options);
905 }
906 *(struct thread_data **)td->eo = td;
907 }
908
909 return 0;
910}
911
912static void init_flags(struct thread_data *td)
913{
914 struct thread_options *o = &td->o;
915
916 if (o->verify_backlog)
917 td->flags |= TD_F_VER_BACKLOG;
918 if (o->trim_backlog)
919 td->flags |= TD_F_TRIM_BACKLOG;
920 if (o->read_iolog_file)
921 td->flags |= TD_F_READ_IOLOG;
922 if (o->refill_buffers)
923 td->flags |= TD_F_REFILL_BUFFERS;
924 if (o->scramble_buffers)
925 td->flags |= TD_F_SCRAMBLE_BUFFERS;
926 if (o->verify != VERIFY_NONE)
927 td->flags |= TD_F_VER_NONE;
928}
929
930static int setup_random_seeds(struct thread_data *td)
931{
932 unsigned long seed;
933 unsigned int i;
934
935 if (!td->o.rand_repeatable && !td->o.rand_seed)
936 return init_random_state(td, td->rand_seeds, sizeof(td->rand_seeds));
937
938 if (!td->o.rand_seed)
939 seed = 0x89;
940 else
941 seed = td->o.rand_seed;
942
943 for (i = 0; i < 4; i++)
944 seed *= 0x9e370001UL;
945
946 for (i = 0; i < FIO_RAND_NR_OFFS; i++) {
947 td->rand_seeds[i] = seed;
948 seed *= 0x9e370001UL;
949 }
950
951 td_fill_rand_seeds(td);
952 return 0;
953}
954
955enum {
956 FPRE_NONE = 0,
957 FPRE_JOBNAME,
958 FPRE_JOBNUM,
959 FPRE_FILENUM
960};
961
962static struct fpre_keyword {
963 const char *keyword;
964 size_t strlen;
965 int key;
966} fpre_keywords[] = {
967 { .keyword = "$jobname", .key = FPRE_JOBNAME, },
968 { .keyword = "$jobnum", .key = FPRE_JOBNUM, },
969 { .keyword = "$filenum", .key = FPRE_FILENUM, },
970 { .keyword = NULL, },
971 };
972
973static char *make_filename(char *buf, size_t buf_size,struct thread_options *o,
974 const char *jobname, int jobnum, int filenum)
975{
976 struct fpre_keyword *f;
977 char copy[PATH_MAX];
978 size_t dst_left = PATH_MAX - 1;
979
980 if (!o->filename_format || !strlen(o->filename_format)) {
981 sprintf(buf, "%s.%d.%d", jobname, jobnum, filenum);
982 return NULL;
983 }
984
985 for (f = &fpre_keywords[0]; f->keyword; f++)
986 f->strlen = strlen(f->keyword);
987
988 buf[buf_size - 1] = '\0';
989 strncpy(buf, o->filename_format, buf_size - 1);
990
991 memset(copy, 0, sizeof(copy));
992 for (f = &fpre_keywords[0]; f->keyword; f++) {
993 do {
994 size_t pre_len, post_start = 0;
995 char *str, *dst = copy;
996
997 str = strcasestr(buf, f->keyword);
998 if (!str)
999 break;
1000
1001 pre_len = str - buf;
1002 if (strlen(str) != f->strlen)
1003 post_start = pre_len + f->strlen;
1004
1005 if (pre_len) {
1006 strncpy(dst, buf, pre_len);
1007 dst += pre_len;
1008 dst_left -= pre_len;
1009 }
1010
1011 switch (f->key) {
1012 case FPRE_JOBNAME: {
1013 int ret;
1014
1015 ret = snprintf(dst, dst_left, "%s", jobname);
1016 if (ret < 0)
1017 break;
1018 dst += ret;
1019 dst_left -= ret;
1020 break;
1021 }
1022 case FPRE_JOBNUM: {
1023 int ret;
1024
1025 ret = snprintf(dst, dst_left, "%d", jobnum);
1026 if (ret < 0)
1027 break;
1028 dst += ret;
1029 dst_left -= ret;
1030 break;
1031 }
1032 case FPRE_FILENUM: {
1033 int ret;
1034
1035 ret = snprintf(dst, dst_left, "%d", filenum);
1036 if (ret < 0)
1037 break;
1038 dst += ret;
1039 dst_left -= ret;
1040 break;
1041 }
1042 default:
1043 assert(0);
1044 break;
1045 }
1046
1047 if (post_start)
1048 strncpy(dst, buf + post_start, dst_left);
1049
1050 strncpy(buf, copy, buf_size - 1);
1051 } while (1);
1052 }
1053
1054 return buf;
1055}
1056
1057int parse_dryrun(void)
1058{
1059 return dump_cmdline || parse_only;
1060}
1061
1062/*
1063 * Adds a job to the list of things todo. Sanitizes the various options
1064 * to make sure we don't have conflicts, and initializes various
1065 * members of td.
1066 */
1067static int add_job(struct thread_data *td, const char *jobname, int job_add_num,
1068 int recursed, int client_type)
1069{
1070 unsigned int i;
1071 char fname[PATH_MAX];
1072 int numjobs, file_alloced;
1073 struct thread_options *o = &td->o;
1074 char logname[PATH_MAX + 32];
1075
1076 /*
1077 * the def_thread is just for options, it's not a real job
1078 */
1079 if (td == &def_thread)
1080 return 0;
1081
1082 init_flags(td);
1083
1084 /*
1085 * if we are just dumping the output command line, don't add the job
1086 */
1087 if (parse_dryrun()) {
1088 put_job(td);
1089 return 0;
1090 }
1091
1092 td->client_type = client_type;
1093
1094 if (profile_td_init(td))
1095 goto err;
1096
1097 if (ioengine_load(td))
1098 goto err;
1099
1100 if (o->odirect)
1101 td->io_ops->flags |= FIO_RAWIO;
1102
1103 file_alloced = 0;
1104 if (!o->filename && !td->files_index && !o->read_iolog_file) {
1105 file_alloced = 1;
1106
1107 if (o->nr_files == 1 && exists_and_not_file(jobname))
1108 add_file(td, jobname, job_add_num, 0);
1109 else {
1110 for (i = 0; i < o->nr_files; i++)
1111 add_file(td, make_filename(fname, sizeof(fname), o, jobname, job_add_num, i), job_add_num, 0);
1112 }
1113 }
1114
1115 if (fixup_options(td))
1116 goto err;
1117
1118 flow_init_job(td);
1119
1120 /*
1121 * IO engines only need this for option callbacks, and the address may
1122 * change in subprocesses.
1123 */
1124 if (td->eo)
1125 *(struct thread_data **)td->eo = NULL;
1126
1127 if (td->io_ops->flags & FIO_DISKLESSIO) {
1128 struct fio_file *f;
1129
1130 for_each_file(td, f, i)
1131 f->real_file_size = -1ULL;
1132 }
1133
1134 td->mutex = fio_mutex_init(FIO_MUTEX_LOCKED);
1135
1136 td->ts.clat_percentiles = o->clat_percentiles;
1137 td->ts.percentile_precision = o->percentile_precision;
1138 memcpy(td->ts.percentile_list, o->percentile_list, sizeof(o->percentile_list));
1139
1140 for (i = 0; i < DDIR_RWDIR_CNT; i++) {
1141 td->ts.clat_stat[i].min_val = ULONG_MAX;
1142 td->ts.slat_stat[i].min_val = ULONG_MAX;
1143 td->ts.lat_stat[i].min_val = ULONG_MAX;
1144 td->ts.bw_stat[i].min_val = ULONG_MAX;
1145 }
1146 td->ddir_seq_nr = o->ddir_seq_nr;
1147
1148 if ((o->stonewall || o->new_group) && prev_group_jobs) {
1149 prev_group_jobs = 0;
1150 groupid++;
1151 }
1152
1153 td->groupid = groupid;
1154 prev_group_jobs++;
1155
1156 if (setup_random_seeds(td)) {
1157 td_verror(td, errno, "init_random_state");
1158 goto err;
1159 }
1160
1161 if (setup_rate(td))
1162 goto err;
1163
1164 if (o->lat_log_file) {
1165 struct log_params p = {
1166 .td = td,
1167 .avg_msec = o->log_avg_msec,
1168 .log_type = IO_LOG_TYPE_LAT,
1169 .log_offset = o->log_offset,
1170 .log_gz = o->log_gz,
1171 .log_gz_store = o->log_gz_store,
1172 };
1173 const char *suf;
1174
1175 if (p.log_gz_store)
1176 suf = "log.fz";
1177 else
1178 suf = "log";
1179
1180 snprintf(logname, sizeof(logname), "%s_lat.%d.%s",
1181 o->lat_log_file, td->thread_number, suf);
1182 setup_log(&td->lat_log, &p, logname);
1183 snprintf(logname, sizeof(logname), "%s_slat.%d.%s",
1184 o->lat_log_file, td->thread_number, suf);
1185 setup_log(&td->slat_log, &p, logname);
1186 snprintf(logname, sizeof(logname), "%s_clat.%d.%s",
1187 o->lat_log_file, td->thread_number, suf);
1188 setup_log(&td->clat_log, &p, logname);
1189 }
1190 if (o->bw_log_file) {
1191 struct log_params p = {
1192 .td = td,
1193 .avg_msec = o->log_avg_msec,
1194 .log_type = IO_LOG_TYPE_BW,
1195 .log_offset = o->log_offset,
1196 .log_gz = o->log_gz,
1197 .log_gz_store = o->log_gz_store,
1198 };
1199 const char *suf;
1200
1201 if (p.log_gz_store)
1202 suf = "log.fz";
1203 else
1204 suf = "log";
1205
1206 snprintf(logname, sizeof(logname), "%s_bw.%d.%s",
1207 o->bw_log_file, td->thread_number, suf);
1208 setup_log(&td->bw_log, &p, logname);
1209 }
1210 if (o->iops_log_file) {
1211 struct log_params p = {
1212 .td = td,
1213 .avg_msec = o->log_avg_msec,
1214 .log_type = IO_LOG_TYPE_IOPS,
1215 .log_offset = o->log_offset,
1216 .log_gz = o->log_gz,
1217 .log_gz_store = o->log_gz_store,
1218 };
1219 const char *suf;
1220
1221 if (p.log_gz_store)
1222 suf = "log.fz";
1223 else
1224 suf = "log";
1225
1226 snprintf(logname, sizeof(logname), "%s_iops.%d.%s",
1227 o->iops_log_file, td->thread_number, suf);
1228 setup_log(&td->iops_log, &p, logname);
1229 }
1230
1231 if (!o->name)
1232 o->name = strdup(jobname);
1233
1234 if (output_format == FIO_OUTPUT_NORMAL) {
1235 if (!job_add_num) {
1236 if (is_backend && !recursed)
1237 fio_server_send_add_job(td);
1238
1239 if (!(td->io_ops->flags & FIO_NOIO)) {
1240 char *c1, *c2, *c3, *c4;
1241 char *c5 = NULL, *c6 = NULL;
1242
1243 c1 = fio_uint_to_kmg(o->min_bs[DDIR_READ]);
1244 c2 = fio_uint_to_kmg(o->max_bs[DDIR_READ]);
1245 c3 = fio_uint_to_kmg(o->min_bs[DDIR_WRITE]);
1246 c4 = fio_uint_to_kmg(o->max_bs[DDIR_WRITE]);
1247
1248 if (!o->bs_is_seq_rand) {
1249 c5 = fio_uint_to_kmg(o->min_bs[DDIR_TRIM]);
1250 c6 = fio_uint_to_kmg(o->max_bs[DDIR_TRIM]);
1251 }
1252
1253 log_info("%s: (g=%d): rw=%s, ", td->o.name,
1254 td->groupid,
1255 ddir_str(o->td_ddir));
1256
1257 if (o->bs_is_seq_rand)
1258 log_info("bs(seq/rand)=%s-%s/%s-%s, ",
1259 c1, c2, c3, c4);
1260 else
1261 log_info("bs=%s-%s/%s-%s/%s-%s, ",
1262 c1, c2, c3, c4, c5, c6);
1263
1264 log_info("ioengine=%s, iodepth=%u\n",
1265 td->io_ops->name, o->iodepth);
1266
1267 free(c1);
1268 free(c2);
1269 free(c3);
1270 free(c4);
1271 free(c5);
1272 free(c6);
1273 }
1274 } else if (job_add_num == 1)
1275 log_info("...\n");
1276 }
1277
1278 /*
1279 * recurse add identical jobs, clear numjobs and stonewall options
1280 * as they don't apply to sub-jobs
1281 */
1282 numjobs = o->numjobs;
1283 while (--numjobs) {
1284 struct thread_data *td_new = get_new_job(0, td, 1, jobname);
1285
1286 if (!td_new)
1287 goto err;
1288
1289 td_new->o.numjobs = 1;
1290 td_new->o.stonewall = 0;
1291 td_new->o.new_group = 0;
1292 td_new->subjob_number = numjobs;
1293
1294 if (file_alloced) {
1295 if (td_new->files) {
1296 struct fio_file *f;
1297 for_each_file(td_new, f, i) {
1298 if (f->file_name)
1299 sfree(f->file_name);
1300 sfree(f);
1301 }
1302 free(td_new->files);
1303 td_new->files = NULL;
1304 }
1305 td_new->files_index = 0;
1306 td_new->files_size = 0;
1307 if (td_new->o.filename) {
1308 free(td_new->o.filename);
1309 td_new->o.filename = NULL;
1310 }
1311 }
1312
1313 if (add_job(td_new, jobname, numjobs, 1, client_type))
1314 goto err;
1315 }
1316
1317 return 0;
1318err:
1319 put_job(td);
1320 return -1;
1321}
1322
1323/*
1324 * Parse as if 'o' was a command line
1325 */
1326void add_job_opts(const char **o, int client_type)
1327{
1328 struct thread_data *td, *td_parent;
1329 int i, in_global = 1;
1330 char jobname[32];
1331
1332 i = 0;
1333 td_parent = td = NULL;
1334 while (o[i]) {
1335 if (!strncmp(o[i], "name", 4)) {
1336 in_global = 0;
1337 if (td)
1338 add_job(td, jobname, 0, 0, client_type);
1339 td = NULL;
1340 sprintf(jobname, "%s", o[i] + 5);
1341 }
1342 if (in_global && !td_parent)
1343 td_parent = get_new_job(1, &def_thread, 0, jobname);
1344 else if (!in_global && !td) {
1345 if (!td_parent)
1346 td_parent = &def_thread;
1347 td = get_new_job(0, td_parent, 0, jobname);
1348 }
1349 if (in_global)
1350 fio_options_parse(td_parent, (char **) &o[i], 1, 0);
1351 else
1352 fio_options_parse(td, (char **) &o[i], 1, 0);
1353 i++;
1354 }
1355
1356 if (td)
1357 add_job(td, jobname, 0, 0, client_type);
1358}
1359
1360static int skip_this_section(const char *name)
1361{
1362 int i;
1363
1364 if (!nr_job_sections)
1365 return 0;
1366 if (!strncmp(name, "global", 6))
1367 return 0;
1368
1369 for (i = 0; i < nr_job_sections; i++)
1370 if (!strcmp(job_sections[i], name))
1371 return 0;
1372
1373 return 1;
1374}
1375
1376static int is_empty_or_comment(char *line)
1377{
1378 unsigned int i;
1379
1380 for (i = 0; i < strlen(line); i++) {
1381 if (line[i] == ';')
1382 return 1;
1383 if (line[i] == '#')
1384 return 1;
1385 if (!isspace((int) line[i]) && !iscntrl((int) line[i]))
1386 return 0;
1387 }
1388
1389 return 1;
1390}
1391
1392/*
1393 * This is our [ini] type file parser.
1394 */
1395int parse_jobs_ini(char *file, int is_buf, int stonewall_flag, int type)
1396{
1397 unsigned int global;
1398 struct thread_data *td;
1399 char *string, *name;
1400 FILE *f;
1401 char *p;
1402 int ret = 0, stonewall;
1403 int first_sect = 1;
1404 int skip_fgets = 0;
1405 int inside_skip = 0;
1406 char **opts;
1407 int i, alloc_opts, num_opts;
1408
1409 if (is_buf)
1410 f = NULL;
1411 else {
1412 if (!strcmp(file, "-"))
1413 f = stdin;
1414 else
1415 f = fopen(file, "r");
1416
1417 if (!f) {
1418 perror("fopen job file");
1419 return 1;
1420 }
1421 }
1422
1423 string = malloc(4096);
1424
1425 /*
1426 * it's really 256 + small bit, 280 should suffice
1427 */
1428 name = malloc(280);
1429 memset(name, 0, 280);
1430
1431 alloc_opts = 8;
1432 opts = malloc(sizeof(char *) * alloc_opts);
1433 num_opts = 0;
1434
1435 stonewall = stonewall_flag;
1436 do {
1437 /*
1438 * if skip_fgets is set, we already have loaded a line we
1439 * haven't handled.
1440 */
1441 if (!skip_fgets) {
1442 if (is_buf)
1443 p = strsep(&file, "\n");
1444 else
1445 p = fgets(string, 4096, f);
1446 if (!p)
1447 break;
1448 }
1449
1450 skip_fgets = 0;
1451 strip_blank_front(&p);
1452 strip_blank_end(p);
1453
1454 if (is_empty_or_comment(p))
1455 continue;
1456 if (sscanf(p, "[%255[^\n]]", name) != 1) {
1457 if (inside_skip)
1458 continue;
1459 log_err("fio: option <%s> outside of [] job section\n",
1460 p);
1461 break;
1462 }
1463
1464 name[strlen(name) - 1] = '\0';
1465
1466 if (skip_this_section(name)) {
1467 inside_skip = 1;
1468 continue;
1469 } else
1470 inside_skip = 0;
1471
1472 global = !strncmp(name, "global", 6);
1473
1474 if (dump_cmdline) {
1475 if (first_sect)
1476 log_info("fio ");
1477 if (!global)
1478 log_info("--name=%s ", name);
1479 first_sect = 0;
1480 }
1481
1482 td = get_new_job(global, &def_thread, 0, name);
1483 if (!td) {
1484 ret = 1;
1485 break;
1486 }
1487
1488 /*
1489 * Separate multiple job files by a stonewall
1490 */
1491 if (!global && stonewall) {
1492 td->o.stonewall = stonewall;
1493 stonewall = 0;
1494 }
1495
1496 num_opts = 0;
1497 memset(opts, 0, alloc_opts * sizeof(char *));
1498
1499 while (1) {
1500 if (is_buf)
1501 p = strsep(&file, "\n");
1502 else
1503 p = fgets(string, 4096, f);
1504 if (!p)
1505 break;
1506
1507 if (is_empty_or_comment(p))
1508 continue;
1509
1510 strip_blank_front(&p);
1511
1512 /*
1513 * new section, break out and make sure we don't
1514 * fgets() a new line at the top.
1515 */
1516 if (p[0] == '[') {
1517 skip_fgets = 1;
1518 break;
1519 }
1520
1521 strip_blank_end(p);
1522
1523 if (num_opts == alloc_opts) {
1524 alloc_opts <<= 1;
1525 opts = realloc(opts,
1526 alloc_opts * sizeof(char *));
1527 }
1528
1529 opts[num_opts] = strdup(p);
1530 num_opts++;
1531 }
1532
1533 ret = fio_options_parse(td, opts, num_opts, dump_cmdline);
1534 if (!ret)
1535 ret = add_job(td, name, 0, 0, type);
1536 else {
1537 log_err("fio: job %s dropped\n", name);
1538 put_job(td);
1539 }
1540
1541 for (i = 0; i < num_opts; i++)
1542 free(opts[i]);
1543 num_opts = 0;
1544 } while (!ret);
1545
1546 if (dump_cmdline)
1547 log_info("\n");
1548
1549 i = 0;
1550 while (i < nr_job_sections) {
1551 free(job_sections[i]);
1552 i++;
1553 }
1554
1555 free(string);
1556 free(name);
1557 free(opts);
1558 if (!is_buf && f != stdin)
1559 fclose(f);
1560 return ret;
1561}
1562
1563static int fill_def_thread(void)
1564{
1565 memset(&def_thread, 0, sizeof(def_thread));
1566
1567 fio_getaffinity(getpid(), &def_thread.o.cpumask);
1568 def_thread.o.error_dump = 1;
1569
1570 /*
1571 * fill default options
1572 */
1573 fio_fill_default_options(&def_thread);
1574 return 0;
1575}
1576
1577static void usage(const char *name)
1578{
1579 printf("%s\n", fio_version_string);
1580 printf("%s [options] [job options] <job file(s)>\n", name);
1581 printf(" --debug=options\tEnable debug logging. May be one/more of:\n"
1582 "\t\t\tprocess,file,io,mem,blktrace,verify,random,parse,\n"
1583 "\t\t\tdiskutil,job,mutex,profile,time,net,rate,compress\n");
1584 printf(" --parse-only\t\tParse options only, don't start any IO\n");
1585 printf(" --output\t\tWrite output to file\n");
1586 printf(" --runtime\t\tRuntime in seconds\n");
1587 printf(" --bandwidth-log\tGenerate per-job bandwidth logs\n");
1588 printf(" --minimal\t\tMinimal (terse) output\n");
1589 printf(" --output-format=x\tOutput format (terse,json,normal)\n");
1590 printf(" --terse-version=x\tSet terse version output format to 'x'\n");
1591 printf(" --version\t\tPrint version info and exit\n");
1592 printf(" --help\t\tPrint this page\n");
1593 printf(" --cpuclock-test\tPerform test/validation of CPU clock\n");
1594 printf(" --crctest\t\tTest speed of checksum functions\n");
1595 printf(" --cmdhelp=cmd\t\tPrint command help, \"all\" for all of"
1596 " them\n");
1597 printf(" --enghelp=engine\tPrint ioengine help, or list"
1598 " available ioengines\n");
1599 printf(" --enghelp=engine,cmd\tPrint help for an ioengine"
1600 " cmd\n");
1601 printf(" --showcmd\t\tTurn a job file into command line options\n");
1602 printf(" --eta=when\t\tWhen ETA estimate should be printed\n");
1603 printf(" \t\tMay be \"always\", \"never\" or \"auto\"\n");
1604 printf(" --eta-newline=time\tForce a new line for every 'time'");
1605 printf(" period passed\n");
1606 printf(" --status-interval=t\tForce full status dump every");
1607 printf(" 't' period passed\n");
1608 printf(" --readonly\t\tTurn on safety read-only checks, preventing"
1609 " writes\n");
1610 printf(" --section=name\tOnly run specified section in job file\n");
1611 printf(" --alloc-size=kb\tSet smalloc pool to this size in kb"
1612 " (def 1024)\n");
1613 printf(" --warnings-fatal\tFio parser warnings are fatal\n");
1614 printf(" --max-jobs=nr\t\tMaximum number of threads/processes to support\n");
1615 printf(" --server=args\t\tStart a backend fio server\n");
1616 printf(" --daemonize=pidfile\tBackground fio server, write pid to file\n");
1617 printf(" --client=hostname\tTalk to remote backend fio server at hostname\n");
1618 printf(" --idle-prof=option\tReport cpu idleness on a system or percpu basis\n"
1619 "\t\t\t(option=system,percpu) or run unit work\n"
1620 "\t\t\tcalibration only (option=calibrate)\n");
1621#ifdef CONFIG_ZLIB
1622 printf(" --inflate-log=log\tInflate and output compressed log\n");
1623#endif
1624 printf("\nFio was written by Jens Axboe <jens.axboe@oracle.com>");
1625 printf("\n Jens Axboe <jaxboe@fusionio.com>");
1626 printf("\n Jens Axboe <axboe@fb.com>\n");
1627}
1628
1629#ifdef FIO_INC_DEBUG
1630struct debug_level debug_levels[] = {
1631 { .name = "process",
1632 .help = "Process creation/exit logging",
1633 .shift = FD_PROCESS,
1634 },
1635 { .name = "file",
1636 .help = "File related action logging",
1637 .shift = FD_FILE,
1638 },
1639 { .name = "io",
1640 .help = "IO and IO engine action logging (offsets, queue, completions, etc)",
1641 .shift = FD_IO,
1642 },
1643 { .name = "mem",
1644 .help = "Memory allocation/freeing logging",
1645 .shift = FD_MEM,
1646 },
1647 { .name = "blktrace",
1648 .help = "blktrace action logging",
1649 .shift = FD_BLKTRACE,
1650 },
1651 { .name = "verify",
1652 .help = "IO verification action logging",
1653 .shift = FD_VERIFY,
1654 },
1655 { .name = "random",
1656 .help = "Random generation logging",
1657 .shift = FD_RANDOM,
1658 },
1659 { .name = "parse",
1660 .help = "Parser logging",
1661 .shift = FD_PARSE,
1662 },
1663 { .name = "diskutil",
1664 .help = "Disk utility logging actions",
1665 .shift = FD_DISKUTIL,
1666 },
1667 { .name = "job",
1668 .help = "Logging related to creating/destroying jobs",
1669 .shift = FD_JOB,
1670 },
1671 { .name = "mutex",
1672 .help = "Mutex logging",
1673 .shift = FD_MUTEX
1674 },
1675 { .name = "profile",
1676 .help = "Logging related to profiles",
1677 .shift = FD_PROFILE,
1678 },
1679 { .name = "time",
1680 .help = "Logging related to time keeping functions",
1681 .shift = FD_TIME,
1682 },
1683 { .name = "net",
1684 .help = "Network logging",
1685 .shift = FD_NET,
1686 },
1687 { .name = "rate",
1688 .help = "Rate logging",
1689 .shift = FD_RATE,
1690 },
1691 { .name = "compress",
1692 .help = "Log compression logging",
1693 .shift = FD_COMPRESS,
1694 },
1695 { .name = NULL, },
1696};
1697
1698static int set_debug(const char *string)
1699{
1700 struct debug_level *dl;
1701 char *p = (char *) string;
1702 char *opt;
1703 int i;
1704
1705 if (!strcmp(string, "?") || !strcmp(string, "help")) {
1706 log_info("fio: dumping debug options:");
1707 for (i = 0; debug_levels[i].name; i++) {
1708 dl = &debug_levels[i];
1709 log_info("%s,", dl->name);
1710 }
1711 log_info("all\n");
1712 return 1;
1713 }
1714
1715 while ((opt = strsep(&p, ",")) != NULL) {
1716 int found = 0;
1717
1718 if (!strncmp(opt, "all", 3)) {
1719 log_info("fio: set all debug options\n");
1720 fio_debug = ~0UL;
1721 continue;
1722 }
1723
1724 for (i = 0; debug_levels[i].name; i++) {
1725 dl = &debug_levels[i];
1726 found = !strncmp(opt, dl->name, strlen(dl->name));
1727 if (!found)
1728 continue;
1729
1730 if (dl->shift == FD_JOB) {
1731 opt = strchr(opt, ':');
1732 if (!opt) {
1733 log_err("fio: missing job number\n");
1734 break;
1735 }
1736 opt++;
1737 fio_debug_jobno = atoi(opt);
1738 log_info("fio: set debug jobno %d\n",
1739 fio_debug_jobno);
1740 } else {
1741 log_info("fio: set debug option %s\n", opt);
1742 fio_debug |= (1UL << dl->shift);
1743 }
1744 break;
1745 }
1746
1747 if (!found)
1748 log_err("fio: debug mask %s not found\n", opt);
1749 }
1750 return 0;
1751}
1752#else
1753static int set_debug(const char *string)
1754{
1755 log_err("fio: debug tracing not included in build\n");
1756 return 1;
1757}
1758#endif
1759
1760static void fio_options_fill_optstring(void)
1761{
1762 char *ostr = cmd_optstr;
1763 int i, c;
1764
1765 c = i = 0;
1766 while (l_opts[i].name) {
1767 ostr[c++] = l_opts[i].val;
1768 if (l_opts[i].has_arg == required_argument)
1769 ostr[c++] = ':';
1770 else if (l_opts[i].has_arg == optional_argument) {
1771 ostr[c++] = ':';
1772 ostr[c++] = ':';
1773 }
1774 i++;
1775 }
1776 ostr[c] = '\0';
1777}
1778
1779static int client_flag_set(char c)
1780{
1781 int i;
1782
1783 i = 0;
1784 while (l_opts[i].name) {
1785 int val = l_opts[i].val;
1786
1787 if (c == (val & 0xff))
1788 return (val & FIO_CLIENT_FLAG);
1789
1790 i++;
1791 }
1792
1793 return 0;
1794}
1795
1796static void parse_cmd_client(void *client, char *opt)
1797{
1798 fio_client_add_cmd_option(client, opt);
1799}
1800
1801int parse_cmd_line(int argc, char *argv[], int client_type)
1802{
1803 struct thread_data *td = NULL;
1804 int c, ini_idx = 0, lidx, ret = 0, do_exit = 0, exit_val = 0;
1805 char *ostr = cmd_optstr;
1806 void *pid_file = NULL;
1807 void *cur_client = NULL;
1808 int backend = 0;
1809
1810 /*
1811 * Reset optind handling, since we may call this multiple times
1812 * for the backend.
1813 */
1814 optind = 1;
1815
1816 while ((c = getopt_long_only(argc, argv, ostr, l_opts, &lidx)) != -1) {
1817 if ((c & FIO_CLIENT_FLAG) || client_flag_set(c)) {
1818 parse_cmd_client(cur_client, argv[optind - 1]);
1819 c &= ~FIO_CLIENT_FLAG;
1820 }
1821
1822 switch (c) {
1823 case 'a':
1824 smalloc_pool_size = atoi(optarg);
1825 break;
1826 case 't':
1827 if (check_str_time(optarg, &def_timeout, 1)) {
1828 log_err("fio: failed parsing time %s\n", optarg);
1829 do_exit++;
1830 exit_val = 1;
1831 }
1832 break;
1833 case 'l':
1834 log_err("fio: --latency-log is deprecated. Use per-job latency log options.\n");
1835 do_exit++;
1836 exit_val = 1;
1837 break;
1838 case 'b':
1839 write_bw_log = 1;
1840 break;
1841 case 'o':
1842 if (f_out && f_out != stdout)
1843 fclose(f_out);
1844
1845 f_out = fopen(optarg, "w+");
1846 if (!f_out) {
1847 perror("fopen output");
1848 exit(1);
1849 }
1850 f_err = f_out;
1851 break;
1852 case 'm':
1853 output_format = FIO_OUTPUT_TERSE;
1854 break;
1855 case 'F':
1856 if (!optarg) {
1857 log_err("fio: missing --output-format argument\n");
1858 exit_val = 1;
1859 do_exit++;
1860 break;
1861 }
1862 if (!strcmp(optarg, "minimal") ||
1863 !strcmp(optarg, "terse") ||
1864 !strcmp(optarg, "csv"))
1865 output_format = FIO_OUTPUT_TERSE;
1866 else if (!strcmp(optarg, "json"))
1867 output_format = FIO_OUTPUT_JSON;
1868 else
1869 output_format = FIO_OUTPUT_NORMAL;
1870 break;
1871 case 'f':
1872 append_terse_output = 1;
1873 break;
1874 case 'h':
1875 did_arg = 1;
1876 if (!cur_client) {
1877 usage(argv[0]);
1878 do_exit++;
1879 }
1880 break;
1881 case 'c':
1882 did_arg = 1;
1883 if (!cur_client) {
1884 fio_show_option_help(optarg);
1885 do_exit++;
1886 }
1887 break;
1888 case 'i':
1889 did_arg = 1;
1890 if (!cur_client) {
1891 fio_show_ioengine_help(optarg);
1892 do_exit++;
1893 }
1894 break;
1895 case 's':
1896 did_arg = 1;
1897 dump_cmdline = 1;
1898 break;
1899 case 'r':
1900 read_only = 1;
1901 break;
1902 case 'v':
1903 did_arg = 1;
1904 if (!cur_client) {
1905 log_info("%s\n", fio_version_string);
1906 do_exit++;
1907 }
1908 break;
1909 case 'V':
1910 terse_version = atoi(optarg);
1911 if (!(terse_version == 2 || terse_version == 3 ||
1912 terse_version == 4)) {
1913 log_err("fio: bad terse version format\n");
1914 exit_val = 1;
1915 do_exit++;
1916 }
1917 break;
1918 case 'e':
1919 if (!strcmp("always", optarg))
1920 eta_print = FIO_ETA_ALWAYS;
1921 else if (!strcmp("never", optarg))
1922 eta_print = FIO_ETA_NEVER;
1923 break;
1924 case 'E': {
1925 long long t = 0;
1926
1927 if (str_to_decimal(optarg, &t, 0, NULL, 1)) {
1928 log_err("fio: failed parsing eta time %s\n", optarg);
1929 exit_val = 1;
1930 do_exit++;
1931 }
1932 eta_new_line = t;
1933 break;
1934 }
1935 case 'd':
1936 if (set_debug(optarg))
1937 do_exit++;
1938 break;
1939 case 'P':
1940 did_arg = 1;
1941 parse_only = 1;
1942 break;
1943 case 'x': {
1944 size_t new_size;
1945
1946 if (!strcmp(optarg, "global")) {
1947 log_err("fio: can't use global as only "
1948 "section\n");
1949 do_exit++;
1950 exit_val = 1;
1951 break;
1952 }
1953 new_size = (nr_job_sections + 1) * sizeof(char *);
1954 job_sections = realloc(job_sections, new_size);
1955 job_sections[nr_job_sections] = strdup(optarg);
1956 nr_job_sections++;
1957 break;
1958 }
1959#ifdef CONFIG_ZLIB
1960 case 'X':
1961 exit_val = iolog_file_inflate(optarg);
1962 did_arg++;
1963 do_exit++;
1964 break;
1965#endif
1966 case 'p':
1967 did_arg = 1;
1968 if (exec_profile)
1969 free(exec_profile);
1970 exec_profile = strdup(optarg);
1971 break;
1972 case FIO_GETOPT_JOB: {
1973 const char *opt = l_opts[lidx].name;
1974 char *val = optarg;
1975
1976 if (!strncmp(opt, "name", 4) && td) {
1977 ret = add_job(td, td->o.name ?: "fio", 0, 0, client_type);
1978 if (ret)
1979 goto out_free;
1980 td = NULL;
1981 did_arg = 1;
1982 }
1983 if (!td) {
1984 int is_section = !strncmp(opt, "name", 4);
1985 int global = 0;
1986
1987 if (!is_section || !strncmp(val, "global", 6))
1988 global = 1;
1989
1990 if (is_section && skip_this_section(val))
1991 continue;
1992
1993 td = get_new_job(global, &def_thread, 1, NULL);
1994 if (!td || ioengine_load(td)) {
1995 if (td) {
1996 put_job(td);
1997 td = NULL;
1998 }
1999 do_exit++;
2000 break;
2001 }
2002 fio_options_set_ioengine_opts(l_opts, td);
2003 }
2004
2005 if ((!val || !strlen(val)) &&
2006 l_opts[lidx].has_arg == required_argument) {
2007 log_err("fio: option %s requires an argument\n", opt);
2008 ret = 1;
2009 } else
2010 ret = fio_cmd_option_parse(td, opt, val);
2011
2012 if (ret) {
2013 if (td) {
2014 put_job(td);
2015 td = NULL;
2016 }
2017 do_exit++;
2018 }
2019
2020 if (!ret && !strcmp(opt, "ioengine")) {
2021 free_ioengine(td);
2022 if (ioengine_load(td)) {
2023 put_job(td);
2024 td = NULL;
2025 do_exit++;
2026 break;
2027 }
2028 fio_options_set_ioengine_opts(l_opts, td);
2029 }
2030 break;
2031 }
2032 case FIO_GETOPT_IOENGINE: {
2033 const char *opt = l_opts[lidx].name;
2034 char *val = optarg;
2035
2036 if (!td)
2037 break;
2038
2039 ret = fio_cmd_ioengine_option_parse(td, opt, val);
2040 break;
2041 }
2042 case 'w':
2043 warnings_fatal = 1;
2044 break;
2045 case 'j':
2046 max_jobs = atoi(optarg);
2047 if (!max_jobs || max_jobs > REAL_MAX_JOBS) {
2048 log_err("fio: invalid max jobs: %d\n", max_jobs);
2049 do_exit++;
2050 exit_val = 1;
2051 }
2052 break;
2053 case 'S':
2054 did_arg = 1;
2055#ifndef CONFIG_NO_SHM
2056 if (nr_clients) {
2057 log_err("fio: can't be both client and server\n");
2058 do_exit++;
2059 exit_val = 1;
2060 break;
2061 }
2062 if (optarg)
2063 fio_server_set_arg(optarg);
2064 is_backend = 1;
2065 backend = 1;
2066#else
2067 log_err("fio: client/server requires SHM support\n");
2068 do_exit++;
2069 exit_val = 1;
2070#endif
2071 break;
2072 case 'D':
2073 if (pid_file)
2074 free(pid_file);
2075 pid_file = strdup(optarg);
2076 break;
2077 case 'I':
2078 if ((ret = fio_idle_prof_parse_opt(optarg))) {
2079 /* exit on error and calibration only */
2080 did_arg = 1;
2081 do_exit++;
2082 if (ret == -1)
2083 exit_val = 1;
2084 }
2085 break;
2086 case 'C':
2087 did_arg = 1;
2088 if (is_backend) {
2089 log_err("fio: can't be both client and server\n");
2090 do_exit++;
2091 exit_val = 1;
2092 break;
2093 }
2094 if (fio_client_add(&fio_client_ops, optarg, &cur_client)) {
2095 log_err("fio: failed adding client %s\n", optarg);
2096 do_exit++;
2097 exit_val = 1;
2098 break;
2099 }
2100 /*
2101 * If the next argument exists and isn't an option,
2102 * assume it's a job file for this client only.
2103 */
2104 while (optind < argc) {
2105 if (!strncmp(argv[optind], "--", 2) ||
2106 !strncmp(argv[optind], "-", 1))
2107 break;
2108
2109 fio_client_add_ini_file(cur_client, argv[optind]);
2110 optind++;
2111 }
2112 break;
2113 case 'T':
2114 did_arg = 1;
2115 do_exit++;
2116 exit_val = fio_monotonic_clocktest();
2117 break;
2118 case 'G':
2119 did_arg = 1;
2120 do_exit++;
2121 exit_val = fio_crctest(optarg);
2122 break;
2123 case 'L': {
2124 long long val;
2125
2126 if (check_str_time(optarg, &val, 0)) {
2127 log_err("fio: failed parsing time %s\n", optarg);
2128 do_exit++;
2129 exit_val = 1;
2130 break;
2131 }
2132 status_interval = val * 1000;
2133 break;
2134 }
2135 case '?':
2136 log_err("%s: unrecognized option '%s'\n", argv[0],
2137 argv[optind - 1]);
2138 default:
2139 do_exit++;
2140 exit_val = 1;
2141 break;
2142 }
2143 if (do_exit)
2144 break;
2145 }
2146
2147 if (do_exit && !(is_backend || nr_clients))
2148 exit(exit_val);
2149
2150 if (nr_clients && fio_clients_connect())
2151 exit(1);
2152
2153 if (is_backend && backend)
2154 return fio_start_server(pid_file);
2155 else if (pid_file)
2156 free(pid_file);
2157
2158 if (td) {
2159 if (!ret) {
2160 ret = add_job(td, td->o.name ?: "fio", 0, 0, client_type);
2161 if (ret)
2162 did_arg = 1;
2163 }
2164 }
2165
2166 while (!ret && optind < argc) {
2167 ini_idx++;
2168 ini_file = realloc(ini_file, ini_idx * sizeof(char *));
2169 ini_file[ini_idx - 1] = strdup(argv[optind]);
2170 optind++;
2171 }
2172
2173out_free:
2174 if (pid_file)
2175 free(pid_file);
2176
2177 return ini_idx;
2178}
2179
2180int fio_init_options(void)
2181{
2182 f_out = stdout;
2183 f_err = stderr;
2184
2185 fio_options_fill_optstring();
2186 fio_options_dup_and_init(l_opts);
2187
2188 atexit(free_shm);
2189
2190 if (fill_def_thread())
2191 return 1;
2192
2193 return 0;
2194}
2195
2196extern int fio_check_options(struct thread_options *);
2197
2198int parse_options(int argc, char *argv[])
2199{
2200 const int type = FIO_CLIENT_TYPE_CLI;
2201 int job_files, i;
2202
2203 if (fio_init_options())
2204 return 1;
2205 if (fio_test_cconv(&def_thread.o))
2206 log_err("fio: failed internal cconv test\n");
2207
2208 job_files = parse_cmd_line(argc, argv, type);
2209
2210 if (job_files > 0) {
2211 for (i = 0; i < job_files; i++) {
2212 if (i && fill_def_thread())
2213 return 1;
2214 if (nr_clients) {
2215 if (fio_clients_send_ini(ini_file[i]))
2216 return 1;
2217 free(ini_file[i]);
2218 } else if (!is_backend) {
2219 if (parse_jobs_ini(ini_file[i], 0, i, type))
2220 return 1;
2221 free(ini_file[i]);
2222 }
2223 }
2224 } else if (nr_clients) {
2225 if (fill_def_thread())
2226 return 1;
2227 if (fio_clients_send_ini(NULL))
2228 return 1;
2229 }
2230
2231 free(ini_file);
2232 fio_options_free(&def_thread);
2233 filesetup_mem_free();
2234
2235 if (!thread_number) {
2236 if (parse_dryrun())
2237 return 0;
2238 if (exec_profile)
2239 return 0;
2240 if (is_backend || nr_clients)
2241 return 0;
2242 if (did_arg)
2243 return 0;
2244
2245 log_err("No jobs(s) defined\n\n");
2246
2247 if (!did_arg) {
2248 usage(argv[0]);
2249 return 1;
2250 }
2251
2252 return 0;
2253 }
2254
2255 if (def_thread.o.gtod_offload) {
2256 fio_gtod_init();
2257 fio_gtod_offload = 1;
2258 fio_gtod_cpu = def_thread.o.gtod_cpu;
2259 }
2260
2261 if (output_format == FIO_OUTPUT_NORMAL)
2262 log_info("%s\n", fio_version_string);
2263
2264 return 0;
2265}
2266
2267void options_default_fill(struct thread_options *o)
2268{
2269 memcpy(o, &def_thread.o, sizeof(*o));
2270}