Add softrandommap and --alloc-size options
[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 <getopt.h>
12#include <sys/ipc.h>
13#include <sys/shm.h>
14#include <sys/types.h>
15#include <sys/stat.h>
16
17#include "fio.h"
18#include "parse.h"
19#include "smalloc.h"
20#include "filehash.h"
21
22static char fio_version_string[] = "fio 1.20-rc3";
23
24#define FIO_RANDSEED (0xb1899bedUL)
25
26static char **ini_file;
27static int max_jobs = MAX_JOBS;
28static int dump_cmdline;
29
30struct thread_data def_thread;
31struct thread_data *threads = NULL;
32
33int exitall_on_terminate = 0;
34int terse_output = 0;
35int eta_print;
36unsigned long long mlock_size = 0;
37FILE *f_out = NULL;
38FILE *f_err = NULL;
39char *job_section = NULL;
40
41int write_bw_log = 0;
42int read_only = 0;
43
44static int def_timeout;
45static int write_lat_log;
46
47static int prev_group_jobs;
48
49unsigned long fio_debug = 0;
50
51/*
52 * Command line options. These will contain the above, plus a few
53 * extra that only pertain to fio itself and not jobs.
54 */
55static struct option l_opts[FIO_NR_OPTIONS] = {
56 {
57 .name = "output",
58 .has_arg = required_argument,
59 .val = 'o',
60 },
61 {
62 .name = "timeout",
63 .has_arg = required_argument,
64 .val = 't',
65 },
66 {
67 .name = "latency-log",
68 .has_arg = required_argument,
69 .val = 'l',
70 },
71 {
72 .name = "bandwidth-log",
73 .has_arg = required_argument,
74 .val = 'b',
75 },
76 {
77 .name = "minimal",
78 .has_arg = optional_argument,
79 .val = 'm',
80 },
81 {
82 .name = "version",
83 .has_arg = no_argument,
84 .val = 'v',
85 },
86 {
87 .name = "help",
88 .has_arg = no_argument,
89 .val = 'h',
90 },
91 {
92 .name = "cmdhelp",
93 .has_arg = optional_argument,
94 .val = 'c',
95 },
96 {
97 .name = "showcmd",
98 .has_arg = no_argument,
99 .val = 's',
100 },
101 {
102 .name = "readonly",
103 .has_arg = no_argument,
104 .val = 'r',
105 },
106 {
107 .name = "eta",
108 .has_arg = required_argument,
109 .val = 'e',
110 },
111 {
112 .name = "debug",
113 .has_arg = required_argument,
114 .val = 'd',
115 },
116 {
117 .name = "section",
118 .has_arg = required_argument,
119 .val = 'x',
120 },
121 {
122 .name = "alloc-size",
123 .has_arg = required_argument,
124 .val = 'a',
125 },
126 {
127 .name = NULL,
128 },
129};
130
131FILE *get_f_out()
132{
133 return f_out;
134}
135
136FILE *get_f_err()
137{
138 return f_err;
139}
140
141/*
142 * Return a free job structure.
143 */
144static struct thread_data *get_new_job(int global, struct thread_data *parent)
145{
146 struct thread_data *td;
147
148 if (global)
149 return &def_thread;
150 if (thread_number >= max_jobs)
151 return NULL;
152
153 td = &threads[thread_number++];
154 *td = *parent;
155
156 dup_files(td, parent);
157 options_mem_dupe(td);
158
159 td->thread_number = thread_number;
160 return td;
161}
162
163static void put_job(struct thread_data *td)
164{
165 if (td == &def_thread)
166 return;
167
168 if (td->error)
169 log_info("fio: %s\n", td->verror);
170
171 memset(&threads[td->thread_number - 1], 0, sizeof(*td));
172 thread_number--;
173}
174
175static int setup_rate(struct thread_data *td)
176{
177 unsigned long nr_reads_per_msec;
178 unsigned long long rate;
179 unsigned int bs;
180
181 if (!td->o.rate && !td->o.rate_iops)
182 return 0;
183
184 if (td_rw(td))
185 bs = td->o.rw_min_bs;
186 else if (td_read(td))
187 bs = td->o.min_bs[DDIR_READ];
188 else
189 bs = td->o.min_bs[DDIR_WRITE];
190
191 if (td->o.rate) {
192 rate = td->o.rate;
193 nr_reads_per_msec = (rate * 1024 * 1000LL) / bs;
194 } else
195 nr_reads_per_msec = td->o.rate_iops * 1000UL;
196
197 if (!nr_reads_per_msec) {
198 log_err("rate lower than supported\n");
199 return -1;
200 }
201
202 td->rate_usec_cycle = 1000000000ULL / nr_reads_per_msec;
203 td->rate_pending_usleep = 0;
204 return 0;
205}
206
207/*
208 * Lazy way of fixing up options that depend on each other. We could also
209 * define option callback handlers, but this is easier.
210 */
211static int fixup_options(struct thread_data *td)
212{
213 struct thread_options *o = &td->o;
214
215 if (read_only && td_write(td)) {
216 log_err("fio: job <%s> has write bit set, but fio is in"
217 " read-only mode\n", td->o.name);
218 return 1;
219 }
220
221 if (o->rwmix[DDIR_READ] + o->rwmix[DDIR_WRITE] > 100)
222 o->rwmix[DDIR_WRITE] = 100 - o->rwmix[DDIR_READ];
223
224 if (o->write_iolog_file && o->read_iolog_file) {
225 log_err("fio: read iolog overrides write_iolog\n");
226 free(o->write_iolog_file);
227 o->write_iolog_file = NULL;
228 }
229
230 /*
231 * only really works for sequential io for now, and with 1 file
232 */
233 if (o->zone_size && td_random(td) && o->open_files == 1)
234 o->zone_size = 0;
235
236 /*
237 * Reads can do overwrites, we always need to pre-create the file
238 */
239 if (td_read(td) || td_rw(td))
240 o->overwrite = 1;
241
242 if (!o->min_bs[DDIR_READ])
243 o->min_bs[DDIR_READ] = o->bs[DDIR_READ];
244 if (!o->max_bs[DDIR_READ])
245 o->max_bs[DDIR_READ] = o->bs[DDIR_READ];
246 if (!o->min_bs[DDIR_WRITE])
247 o->min_bs[DDIR_WRITE] = o->bs[DDIR_WRITE];
248 if (!o->max_bs[DDIR_WRITE])
249 o->max_bs[DDIR_WRITE] = o->bs[DDIR_WRITE];
250
251 o->rw_min_bs = min(o->min_bs[DDIR_READ], o->min_bs[DDIR_WRITE]);
252
253 if (!o->file_size_high)
254 o->file_size_high = o->file_size_low;
255
256 if (o->norandommap && o->verify != VERIFY_NONE) {
257 log_err("fio: norandommap given, verify disabled\n");
258 o->verify = VERIFY_NONE;
259 }
260 if (o->bs_unaligned && (o->odirect || td->io_ops->flags & FIO_RAWIO))
261 log_err("fio: bs_unaligned may not work with raw io\n");
262
263 /*
264 * thinktime_spin must be less than thinktime
265 */
266 if (o->thinktime_spin > o->thinktime)
267 o->thinktime_spin = o->thinktime;
268
269 /*
270 * The low water mark cannot be bigger than the iodepth
271 */
272 if (o->iodepth_low > o->iodepth || !o->iodepth_low) {
273 /*
274 * syslet work around - if the workload is sequential,
275 * we want to let the queue drain all the way down to
276 * avoid seeking between async threads
277 */
278 if (!strcmp(td->io_ops->name, "syslet-rw") && !td_random(td))
279 o->iodepth_low = 1;
280 else
281 o->iodepth_low = o->iodepth;
282 }
283
284 /*
285 * If batch number isn't set, default to the same as iodepth
286 */
287 if (o->iodepth_batch > o->iodepth || !o->iodepth_batch)
288 o->iodepth_batch = o->iodepth;
289
290 if (o->nr_files > td->files_index)
291 o->nr_files = td->files_index;
292
293 if (o->open_files > o->nr_files || !o->open_files)
294 o->open_files = o->nr_files;
295
296 if ((o->rate && o->rate_iops) || (o->ratemin && o->rate_iops_min)) {
297 log_err("fio: rate and rate_iops are mutually exclusive\n");
298 return 1;
299 }
300 if ((o->rate < o->ratemin) || (o->rate_iops < o->rate_iops_min)) {
301 log_err("fio: minimum rate exceeds rate\n");
302 return 1;
303 }
304
305 if (!o->timeout && o->time_based) {
306 log_err("fio: time_based requires a runtime/timeout setting\n");
307 o->time_based = 0;
308 }
309
310 if (o->fill_device && !o->size)
311 o->size = ULONG_LONG_MAX;
312
313 if (td_rw(td) && td->o.verify != VERIFY_NONE)
314 log_info("fio: mixed read/write workload with verify. May not "
315 "work as expected, unless you pre-populated the file\n");
316
317 return 0;
318}
319
320/*
321 * This function leaks the buffer
322 */
323static char *to_kmg(unsigned int val)
324{
325 char *buf = malloc(32);
326 char post[] = { 0, 'K', 'M', 'G', 'P', 'E', 0 };
327 char *p = post;
328
329 do {
330 if (val & 1023)
331 break;
332
333 val >>= 10;
334 p++;
335 } while (*p);
336
337 snprintf(buf, 31, "%u%c", val, *p);
338 return buf;
339}
340
341/* External engines are specified by "external:name.o") */
342static const char *get_engine_name(const char *str)
343{
344 char *p = strstr(str, ":");
345
346 if (!p)
347 return str;
348
349 p++;
350 strip_blank_front(&p);
351 strip_blank_end(p);
352 return p;
353}
354
355static int exists_and_not_file(const char *filename)
356{
357 struct stat sb;
358
359 if (lstat(filename, &sb) == -1)
360 return 0;
361
362 if (S_ISREG(sb.st_mode))
363 return 0;
364
365 return 1;
366}
367
368/*
369 * Initialize the various random states we need (random io, block size ranges,
370 * read/write mix, etc).
371 */
372static int init_random_state(struct thread_data *td)
373{
374 unsigned long seeds[6];
375 int fd;
376
377 fd = open("/dev/urandom", O_RDONLY);
378 if (fd == -1) {
379 td_verror(td, errno, "open");
380 return 1;
381 }
382
383 if (read(fd, seeds, sizeof(seeds)) < (int) sizeof(seeds)) {
384 td_verror(td, EIO, "read");
385 close(fd);
386 return 1;
387 }
388
389 close(fd);
390
391 os_random_seed(seeds[0], &td->bsrange_state);
392 os_random_seed(seeds[1], &td->verify_state);
393 os_random_seed(seeds[2], &td->rwmix_state);
394
395 if (td->o.file_service_type == FIO_FSERVICE_RANDOM)
396 os_random_seed(seeds[3], &td->next_file_state);
397
398 os_random_seed(seeds[5], &td->file_size_state);
399
400 if (!td_random(td))
401 return 0;
402
403 if (td->o.rand_repeatable)
404 seeds[4] = FIO_RANDSEED * td->thread_number;
405
406 os_random_seed(seeds[4], &td->random_state);
407 return 0;
408}
409
410/*
411 * Adds a job to the list of things todo. Sanitizes the various options
412 * to make sure we don't have conflicts, and initializes various
413 * members of td.
414 */
415static int add_job(struct thread_data *td, const char *jobname, int job_add_num)
416{
417 const char *ddir_str[] = { NULL, "read", "write", "rw", NULL,
418 "randread", "randwrite", "randrw" };
419 unsigned int i;
420 const char *engine;
421 char fname[PATH_MAX];
422 int numjobs, file_alloced;
423
424 /*
425 * the def_thread is just for options, it's not a real job
426 */
427 if (td == &def_thread)
428 return 0;
429
430 /*
431 * if we are just dumping the output command line, don't add the job
432 */
433 if (dump_cmdline) {
434 put_job(td);
435 return 0;
436 }
437
438 engine = get_engine_name(td->o.ioengine);
439 td->io_ops = load_ioengine(td, engine);
440 if (!td->io_ops) {
441 log_err("fio: failed to load engine %s\n", engine);
442 goto err;
443 }
444
445 if (td->o.use_thread)
446 nr_thread++;
447 else
448 nr_process++;
449
450 if (td->o.odirect)
451 td->io_ops->flags |= FIO_RAWIO;
452
453 file_alloced = 0;
454 if (!td->o.filename && !td->files_index && !td->o.read_iolog_file) {
455 file_alloced = 1;
456
457 if (td->o.nr_files == 1 && exists_and_not_file(jobname))
458 add_file(td, jobname);
459 else {
460 for (i = 0; i < td->o.nr_files; i++) {
461 sprintf(fname, "%s.%d.%d", jobname,
462 td->thread_number, i);
463 add_file(td, fname);
464 }
465 }
466 }
467
468 if (fixup_options(td))
469 goto err;
470
471 if (td->io_ops->flags & FIO_DISKLESSIO) {
472 struct fio_file *f;
473
474 for_each_file(td, f, i)
475 f->real_file_size = -1ULL;
476 }
477
478 td->mutex = fio_mutex_init(0);
479
480 td->ts.clat_stat[0].min_val = td->ts.clat_stat[1].min_val = ULONG_MAX;
481 td->ts.slat_stat[0].min_val = td->ts.slat_stat[1].min_val = ULONG_MAX;
482 td->ts.bw_stat[0].min_val = td->ts.bw_stat[1].min_val = ULONG_MAX;
483 td->ddir_nr = td->o.ddir_nr;
484
485 if ((td->o.stonewall || td->o.numjobs > 1 || td->o.new_group)
486 && prev_group_jobs) {
487 prev_group_jobs = 0;
488 groupid++;
489 }
490
491 td->groupid = groupid;
492 prev_group_jobs++;
493
494 if (init_random_state(td))
495 goto err;
496
497 if (setup_rate(td))
498 goto err;
499
500 if (td->o.write_lat_log) {
501 setup_log(&td->ts.slat_log);
502 setup_log(&td->ts.clat_log);
503 }
504 if (td->o.write_bw_log)
505 setup_log(&td->ts.bw_log);
506
507 if (!td->o.name)
508 td->o.name = strdup(jobname);
509
510 if (!terse_output) {
511 if (!job_add_num) {
512 if (!strcmp(td->io_ops->name, "cpuio")) {
513 log_info("%s: ioengine=cpu, cpuload=%u,"
514 " cpucycle=%u\n", td->o.name,
515 td->o.cpuload,
516 td->o.cpucycle);
517 } else {
518 char *c1, *c2, *c3, *c4;
519
520 c1 = to_kmg(td->o.min_bs[DDIR_READ]);
521 c2 = to_kmg(td->o.max_bs[DDIR_READ]);
522 c3 = to_kmg(td->o.min_bs[DDIR_WRITE]);
523 c4 = to_kmg(td->o.max_bs[DDIR_WRITE]);
524
525 log_info("%s: (g=%d): rw=%s, bs=%s-%s/%s-%s,"
526 " ioengine=%s, iodepth=%u\n",
527 td->o.name, td->groupid,
528 ddir_str[td->o.td_ddir],
529 c1, c2, c3, c4,
530 td->io_ops->name,
531 td->o.iodepth);
532
533 free(c1);
534 free(c2);
535 free(c3);
536 free(c4);
537 }
538 } else if (job_add_num == 1)
539 log_info("...\n");
540 }
541
542 /*
543 * recurse add identical jobs, clear numjobs and stonewall options
544 * as they don't apply to sub-jobs
545 */
546 numjobs = td->o.numjobs;
547 while (--numjobs) {
548 struct thread_data *td_new = get_new_job(0, td);
549
550 if (!td_new)
551 goto err;
552
553 td_new->o.numjobs = 1;
554 td_new->o.stonewall = 0;
555 td_new->o.new_group = 0;
556
557 if (file_alloced) {
558 td_new->o.filename = NULL;
559 td_new->files_index = 0;
560 td_new->files = NULL;
561 }
562
563 job_add_num = numjobs - 1;
564
565 if (add_job(td_new, jobname, job_add_num))
566 goto err;
567 }
568
569 return 0;
570err:
571 put_job(td);
572 return -1;
573}
574
575static int skip_this_section(const char *name)
576{
577 if (!job_section)
578 return 0;
579 if (!strncmp(name, "global", 6))
580 return 0;
581
582 return strncmp(job_section, name, strlen(job_section));
583}
584
585static int is_empty_or_comment(char *line)
586{
587 unsigned int i;
588
589 for (i = 0; i < strlen(line); i++) {
590 if (line[i] == ';')
591 return 1;
592 if (line[i] == '#')
593 return 1;
594 if (!isspace(line[i]) && !iscntrl(line[i]))
595 return 0;
596 }
597
598 return 1;
599}
600
601/*
602 * This is our [ini] type file parser.
603 */
604static int parse_jobs_ini(char *file, int stonewall_flag)
605{
606 unsigned int global;
607 struct thread_data *td;
608 char *string, *name;
609 FILE *f;
610 char *p;
611 int ret = 0, stonewall;
612 int first_sect = 1;
613 int skip_fgets = 0;
614 int inside_skip = 0;
615
616 if (!strcmp(file, "-"))
617 f = stdin;
618 else
619 f = fopen(file, "r");
620
621 if (!f) {
622 perror("fopen job file");
623 return 1;
624 }
625
626 string = malloc(4096);
627
628 /*
629 * it's really 256 + small bit, 280 should suffice
630 */
631 name = malloc(280);
632 memset(name, 0, 280);
633
634 stonewall = stonewall_flag;
635 do {
636 /*
637 * if skip_fgets is set, we already have loaded a line we
638 * haven't handled.
639 */
640 if (!skip_fgets) {
641 p = fgets(string, 4095, f);
642 if (!p)
643 break;
644 }
645
646 skip_fgets = 0;
647 strip_blank_front(&p);
648 strip_blank_end(p);
649
650 if (is_empty_or_comment(p))
651 continue;
652 if (sscanf(p, "[%255s]", name) != 1) {
653 if (inside_skip)
654 continue;
655 log_err("fio: option <%s> outside of [] job section\n",
656 p);
657 break;
658 }
659
660 if (skip_this_section(name)) {
661 inside_skip = 1;
662 continue;
663 } else
664 inside_skip = 0;
665
666 global = !strncmp(name, "global", 6);
667
668 name[strlen(name) - 1] = '\0';
669
670 if (dump_cmdline) {
671 if (first_sect)
672 log_info("fio ");
673 if (!global)
674 log_info("--name=%s ", name);
675 first_sect = 0;
676 }
677
678 td = get_new_job(global, &def_thread);
679 if (!td) {
680 ret = 1;
681 break;
682 }
683
684 /*
685 * Seperate multiple job files by a stonewall
686 */
687 if (!global && stonewall) {
688 td->o.stonewall = stonewall;
689 stonewall = 0;
690 }
691
692 while ((p = fgets(string, 4096, f)) != NULL) {
693 if (is_empty_or_comment(p))
694 continue;
695
696 strip_blank_front(&p);
697
698 /*
699 * new section, break out and make sure we don't
700 * fgets() a new line at the top.
701 */
702 if (p[0] == '[') {
703 skip_fgets = 1;
704 break;
705 }
706
707 strip_blank_end(p);
708
709 /*
710 * Don't break here, continue parsing options so we
711 * dump all the bad ones. Makes trial/error fixups
712 * easier on the user.
713 */
714 ret |= fio_option_parse(td, p);
715 if (!ret && dump_cmdline)
716 log_info("--%s ", p);
717 }
718
719 if (!ret)
720 ret = add_job(td, name, 0);
721 else {
722 log_err("fio: job %s dropped\n", name);
723 put_job(td);
724 }
725 } while (!ret);
726
727 if (dump_cmdline)
728 log_info("\n");
729
730 free(string);
731 free(name);
732 if (f != stdin)
733 fclose(f);
734 return ret;
735}
736
737static int fill_def_thread(void)
738{
739 memset(&def_thread, 0, sizeof(def_thread));
740
741 fio_getaffinity(getpid(), &def_thread.o.cpumask);
742
743 /*
744 * fill default options
745 */
746 fio_fill_default_options(&def_thread);
747
748 def_thread.o.timeout = def_timeout;
749 def_thread.o.write_bw_log = write_bw_log;
750 def_thread.o.write_lat_log = write_lat_log;
751
752 return 0;
753}
754
755static void free_shm(void)
756{
757 struct shmid_ds sbuf;
758
759 if (threads) {
760 shmdt((void *) threads);
761 threads = NULL;
762 shmctl(shm_id, IPC_RMID, &sbuf);
763 }
764
765 scleanup();
766}
767
768/*
769 * The thread area is shared between the main process and the job
770 * threads/processes. So setup a shared memory segment that will hold
771 * all the job info. We use the end of the region for keeping track of
772 * open files across jobs, for file sharing.
773 */
774static int setup_thread_area(void)
775{
776 void *hash;
777
778 /*
779 * 1024 is too much on some machines, scale max_jobs if
780 * we get a failure that looks like too large a shm segment
781 */
782 do {
783 size_t size = max_jobs * sizeof(struct thread_data);
784
785 size += file_hash_size;
786
787 shm_id = shmget(0, size, IPC_CREAT | 0600);
788 if (shm_id != -1)
789 break;
790 if (errno != EINVAL) {
791 perror("shmget");
792 break;
793 }
794
795 max_jobs >>= 1;
796 } while (max_jobs);
797
798 if (shm_id == -1)
799 return 1;
800
801 threads = shmat(shm_id, NULL, 0);
802 if (threads == (void *) -1) {
803 perror("shmat");
804 return 1;
805 }
806
807 memset(threads, 0, max_jobs * sizeof(struct thread_data));
808 hash = (void *) threads + max_jobs * sizeof(struct thread_data);
809 file_hash_init(hash);
810 atexit(free_shm);
811 return 0;
812}
813
814static void usage(const char *name)
815{
816 printf("%s\n", fio_version_string);
817 printf("%s [options] [job options] <job file(s)>\n", name);
818 printf("\t--debug=options\tEnable debug logging\n");
819 printf("\t--output\tWrite output to file\n");
820 printf("\t--timeout\tRuntime in seconds\n");
821 printf("\t--latency-log\tGenerate per-job latency logs\n");
822 printf("\t--bandwidth-log\tGenerate per-job bandwidth logs\n");
823 printf("\t--minimal\tMinimal (terse) output\n");
824 printf("\t--version\tPrint version info and exit\n");
825 printf("\t--help\t\tPrint this page\n");
826 printf("\t--cmdhelp=cmd\tPrint command help, \"all\" for all of"
827 " them\n");
828 printf("\t--showcmd\tTurn a job file into command line options\n");
829 printf("\t--eta=when\tWhen ETA estimate should be printed\n");
830 printf("\t \tMay be \"always\", \"never\" or \"auto\"\n");
831 printf("\t--readonly\tTurn on safety read-only checks, preventing"
832 " writes\n");
833 printf("\t--section=name\tOnly run specified section in job file\n");
834 printf("\t--alloc-size=kb\tSet smalloc pool to this size in kb"
835 " (def 1024)\n");
836}
837
838#ifdef FIO_INC_DEBUG
839struct debug_level debug_levels[] = {
840 { .name = "process", .shift = FD_PROCESS, },
841 { .name = "file", .shift = FD_FILE, },
842 { .name = "io", .shift = FD_IO, },
843 { .name = "mem", .shift = FD_MEM, },
844 { .name = "blktrace", .shift = FD_BLKTRACE },
845 { .name = "verify", .shift = FD_VERIFY },
846 { .name = "random", .shift = FD_RANDOM },
847 { .name = "parse", .shift = FD_PARSE },
848 { .name = "diskutil", .shift = FD_DISKUTIL },
849 { },
850};
851
852static int set_debug(const char *string)
853{
854 struct debug_level *dl;
855 char *p = (char *) string;
856 char *opt;
857 int i;
858
859 if (!strcmp(string, "?") || !strcmp(string, "help")) {
860 int i;
861
862 log_info("fio: dumping debug options:");
863 for (i = 0; debug_levels[i].name; i++) {
864 dl = &debug_levels[i];
865 log_info("%s,", dl->name);
866 }
867 log_info("all\n");
868 return 1;
869 } else if (!strcmp(string, "all")) {
870 fio_debug = ~0UL;
871 return 0;
872 }
873
874 while ((opt = strsep(&p, ",")) != NULL) {
875 int found = 0;
876
877 for (i = 0; debug_levels[i].name; i++) {
878 dl = &debug_levels[i];
879 if (!strncmp(opt, dl->name, strlen(opt))) {
880 log_info("fio: set debug option %s\n", opt);
881 found = 1;
882 fio_debug |= (1UL << dl->shift);
883 break;
884 }
885 }
886
887 if (!found)
888 log_err("fio: debug mask %s not found\n", opt);
889 }
890 return 0;
891}
892#else
893static void set_debug(const char *string)
894{
895 log_err("fio: debug tracing not included in build\n");
896 return 1;
897}
898#endif
899
900static int parse_cmd_line(int argc, char *argv[])
901{
902 struct thread_data *td = NULL;
903 int c, ini_idx = 0, lidx, ret = 0, do_exit = 0, exit_val = 0;
904
905 while ((c = getopt_long_only(argc, argv, "", l_opts, &lidx)) != -1) {
906 switch (c) {
907 case 'a':
908 smalloc_pool_size = atoi(optarg);
909 break;
910 case 't':
911 def_timeout = atoi(optarg);
912 break;
913 case 'l':
914 write_lat_log = 1;
915 break;
916 case 'w':
917 write_bw_log = 1;
918 break;
919 case 'o':
920 f_out = fopen(optarg, "w+");
921 if (!f_out) {
922 perror("fopen output");
923 exit(1);
924 }
925 f_err = f_out;
926 break;
927 case 'm':
928 terse_output = 1;
929 break;
930 case 'h':
931 usage(argv[0]);
932 exit(0);
933 case 'c':
934 exit(fio_show_option_help(optarg));
935 case 's':
936 dump_cmdline = 1;
937 break;
938 case 'r':
939 read_only = 1;
940 break;
941 case 'v':
942 printf("%s\n", fio_version_string);
943 exit(0);
944 case 'e':
945 if (!strcmp("always", optarg))
946 eta_print = FIO_ETA_ALWAYS;
947 else if (!strcmp("never", optarg))
948 eta_print = FIO_ETA_NEVER;
949 break;
950 case 'd':
951 if (set_debug(optarg))
952 do_exit++;
953 break;
954 case 'x':
955 if (!strcmp(optarg, "global")) {
956 log_err("fio: can't use global as only "
957 "section\n");
958 do_exit++;
959 exit_val = 1;
960 break;
961 }
962 if (job_section)
963 free(job_section);
964 job_section = strdup(optarg);
965 break;
966 case FIO_GETOPT_JOB: {
967 const char *opt = l_opts[lidx].name;
968 char *val = optarg;
969
970 if (!strncmp(opt, "name", 4) && td) {
971 ret = add_job(td, td->o.name ?: "fio", 0);
972 if (ret) {
973 put_job(td);
974 return 0;
975 }
976 td = NULL;
977 }
978 if (!td) {
979 int is_section = !strncmp(opt, "name", 4);
980 int global = 0;
981
982 if (!is_section || !strncmp(val, "global", 6))
983 global = 1;
984
985 if (is_section && skip_this_section(val))
986 continue;
987
988 td = get_new_job(global, &def_thread);
989 if (!td)
990 return 0;
991 }
992
993 ret = fio_cmd_option_parse(td, opt, val);
994 break;
995 }
996 default:
997 do_exit++;
998 exit_val = 1;
999 break;
1000 }
1001 }
1002
1003 if (do_exit)
1004 exit(exit_val);
1005
1006 if (td) {
1007 if (!ret)
1008 ret = add_job(td, td->o.name ?: "fio", 0);
1009 if (ret)
1010 put_job(td);
1011 }
1012
1013 while (optind < argc) {
1014 ini_idx++;
1015 ini_file = realloc(ini_file, ini_idx * sizeof(char *));
1016 ini_file[ini_idx - 1] = strdup(argv[optind]);
1017 optind++;
1018 }
1019
1020 return ini_idx;
1021}
1022
1023int parse_options(int argc, char *argv[])
1024{
1025 int job_files, i;
1026
1027 f_out = stdout;
1028 f_err = stderr;
1029
1030 fio_options_dup_and_init(l_opts);
1031
1032 if (setup_thread_area())
1033 return 1;
1034 if (fill_def_thread())
1035 return 1;
1036
1037 job_files = parse_cmd_line(argc, argv);
1038
1039 for (i = 0; i < job_files; i++) {
1040 if (fill_def_thread())
1041 return 1;
1042 if (parse_jobs_ini(ini_file[i], i))
1043 return 1;
1044 free(ini_file[i]);
1045 }
1046
1047 free(ini_file);
1048 options_mem_free(&def_thread);
1049
1050 if (!thread_number) {
1051 if (dump_cmdline)
1052 return 0;
1053
1054 log_err("No jobs defined(s)\n");
1055 usage(argv[0]);
1056 return 1;
1057 }
1058
1059 return 0;
1060}