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