blob: 23fe8dc69e2bfb820e9de322919bd315eac9249a [file] [log] [blame]
Jens Axboe906c8d72006-06-13 09:37:56 +02001/*
Jens Axboecb2c86f2006-10-26 13:48:34 +02002 * This file contains job initialization and setup functions.
Jens Axboe906c8d72006-06-13 09:37:56 +02003 */
Jens Axboeebac4652005-12-08 15:25:21 +01004#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>
Jens Axboeb4692822006-10-27 13:43:22 +020011#include <getopt.h>
Jens Axboeebac4652005-12-08 15:25:21 +010012#include <sys/ipc.h>
13#include <sys/shm.h>
14#include <sys/types.h>
15#include <sys/stat.h>
16
17#include "fio.h"
Jens Axboecb2c86f2006-10-26 13:48:34 +020018#include "parse.h"
Jens Axboeebac4652005-12-08 15:25:21 +010019
Jens Axboe214e1ec2007-03-15 10:48:13 +010020static char fio_version_string[] = "fio 1.14a";
21
Jens Axboeee738492007-01-10 11:23:16 +010022#define FIO_RANDSEED (0xb1899bedUL)
Jens Axboeebac4652005-12-08 15:25:21 +010023
Jens Axboe214e1ec2007-03-15 10:48:13 +010024static char **ini_file;
25static int max_jobs = MAX_JOBS;
Jens Axboee1f36502006-10-27 10:54:08 +020026
Jens Axboe214e1ec2007-03-15 10:48:13 +010027struct thread_data def_thread;
28struct thread_data *threads = NULL;
Jens Axboee1f36502006-10-27 10:54:08 +020029
Jens Axboe214e1ec2007-03-15 10:48:13 +010030int exitall_on_terminate = 0;
31int terse_output = 0;
32unsigned long long mlock_size = 0;
33FILE *f_out = NULL;
34FILE *f_err = NULL;
Jens Axboeee738492007-01-10 11:23:16 +010035
Jens Axboe214e1ec2007-03-15 10:48:13 +010036int write_bw_log = 0;
Jens Axboee1f36502006-10-27 10:54:08 +020037
Jens Axboe214e1ec2007-03-15 10:48:13 +010038static int def_timeout = 0;
39static int write_lat_log = 0;
40
41static int prev_group_jobs;
Jens Axboeb4692822006-10-27 13:43:22 +020042
43/*
44 * Command line options. These will contain the above, plus a few
45 * extra that only pertain to fio itself and not jobs.
46 */
Jens Axboe214e1ec2007-03-15 10:48:13 +010047static struct option long_options[FIO_NR_OPTIONS] = {
Jens Axboeb4692822006-10-27 13:43:22 +020048 {
49 .name = "output",
50 .has_arg = required_argument,
51 .val = 'o',
52 },
53 {
54 .name = "timeout",
55 .has_arg = required_argument,
56 .val = 't',
57 },
58 {
59 .name = "latency-log",
60 .has_arg = required_argument,
61 .val = 'l',
62 },
63 {
64 .name = "bandwidth-log",
65 .has_arg = required_argument,
66 .val = 'b',
67 },
68 {
69 .name = "minimal",
70 .has_arg = optional_argument,
71 .val = 'm',
72 },
73 {
74 .name = "version",
75 .has_arg = no_argument,
76 .val = 'v',
77 },
78 {
Jens Axboefd28ca42007-01-09 21:20:13 +010079 .name = "help",
80 .has_arg = no_argument,
81 .val = 'h',
82 },
83 {
84 .name = "cmdhelp",
Jens Axboe320beef2007-03-01 10:44:12 +010085 .has_arg = optional_argument,
Jens Axboefd28ca42007-01-09 21:20:13 +010086 .val = 'c',
87 },
88 {
Jens Axboeb4692822006-10-27 13:43:22 +020089 .name = NULL,
90 },
91};
92
Joel Becker9728ce32007-03-01 08:24:39 +010093FILE *get_f_out()
94{
95 return f_out;
96}
97
98FILE *get_f_err()
99{
100 return f_err;
101}
102
Jens Axboe906c8d72006-06-13 09:37:56 +0200103/*
104 * Return a free job structure.
105 */
Jens Axboeebac4652005-12-08 15:25:21 +0100106static struct thread_data *get_new_job(int global, struct thread_data *parent)
107{
108 struct thread_data *td;
109
110 if (global)
111 return &def_thread;
112 if (thread_number >= max_jobs)
113 return NULL;
114
115 td = &threads[thread_number++];
Jens Axboeddaeaa52006-06-08 11:00:58 +0200116 *td = *parent;
Jens Axboeebac4652005-12-08 15:25:21 +0100117
Jens Axboeebac4652005-12-08 15:25:21 +0100118 td->thread_number = thread_number;
Jens Axboeebac4652005-12-08 15:25:21 +0100119 return td;
120}
121
122static void put_job(struct thread_data *td)
123{
Jens Axboe549577a2006-10-30 12:23:41 +0100124 if (td == &def_thread)
125 return;
126
Jens Axboe16edf252007-02-10 14:59:22 +0100127 if (td->error)
Jens Axboe6d861442007-03-15 09:22:23 +0100128 log_info("fio: %s\n", td->verror);
Jens Axboe16edf252007-02-10 14:59:22 +0100129
Jens Axboeebac4652005-12-08 15:25:21 +0100130 memset(&threads[td->thread_number - 1], 0, sizeof(*td));
131 thread_number--;
132}
133
Jens Axboe127f6862007-03-15 12:09:57 +0100134static int setup_rate(struct thread_data *td)
135{
136 unsigned long nr_reads_per_msec;
137 unsigned long long rate;
138 unsigned int bs;
139
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100140 if (!td->o.rate && !td->o.rate_iops)
Jens Axboe127f6862007-03-15 12:09:57 +0100141 return 0;
142
143 if (td_rw(td))
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100144 bs = td->o.rw_min_bs;
Jens Axboe127f6862007-03-15 12:09:57 +0100145 else if (td_read(td))
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100146 bs = td->o.min_bs[DDIR_READ];
Jens Axboe127f6862007-03-15 12:09:57 +0100147 else
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100148 bs = td->o.min_bs[DDIR_WRITE];
Jens Axboe127f6862007-03-15 12:09:57 +0100149
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100150 if (td->o.rate) {
151 rate = td->o.rate;
Jens Axboe127f6862007-03-15 12:09:57 +0100152 nr_reads_per_msec = (rate * 1024 * 1000LL) / bs;
153 } else
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100154 nr_reads_per_msec = td->o.rate_iops * 1000UL;
Jens Axboe127f6862007-03-15 12:09:57 +0100155
156 if (!nr_reads_per_msec) {
157 log_err("rate lower than supported\n");
158 return -1;
159 }
160
161 td->rate_usec_cycle = 1000000000ULL / nr_reads_per_msec;
162 td->rate_pending_usleep = 0;
163 return 0;
164}
165
Jens Axboedad915e2006-10-27 11:10:18 +0200166/*
167 * Lazy way of fixing up options that depend on each other. We could also
168 * define option callback handlers, but this is easier.
169 */
Jens Axboe4e991c22007-03-15 11:41:11 +0100170static int fixup_options(struct thread_data *td)
Jens Axboee1f36502006-10-27 10:54:08 +0200171{
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100172 struct thread_options *o = &td->o;
Jens Axboedad915e2006-10-27 11:10:18 +0200173
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100174 if (!o->rwmixread && o->rwmixwrite)
175 o->rwmixread = 100 - o->rwmixwrite;
176
177 if (o->write_iolog_file && o->read_iolog_file) {
Jens Axboe076efc72006-10-27 11:24:25 +0200178 log_err("fio: read iolog overrides write_iolog\n");
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100179 free(o->write_iolog_file);
180 o->write_iolog_file = NULL;
Jens Axboe076efc72006-10-27 11:24:25 +0200181 }
Jens Axboe16b462a2006-10-30 12:35:18 +0100182
183 if (td->io_ops->flags & FIO_SYNCIO)
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100184 o->iodepth = 1;
Jens Axboe16b462a2006-10-30 12:35:18 +0100185 else {
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100186 if (!o->iodepth)
187 o->iodepth = o->open_files;
Jens Axboe16b462a2006-10-30 12:35:18 +0100188 }
189
190 /*
191 * only really works for sequential io for now, and with 1 file
192 */
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100193 if (o->zone_size && td_random(td) && o->open_files == 1)
194 o->zone_size = 0;
Jens Axboe16b462a2006-10-30 12:35:18 +0100195
196 /*
197 * Reads can do overwrites, we always need to pre-create the file
198 */
199 if (td_read(td) || td_rw(td))
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100200 o->overwrite = 1;
Jens Axboe16b462a2006-10-30 12:35:18 +0100201
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100202 if (!o->min_bs[DDIR_READ])
203 o->min_bs[DDIR_READ]= o->bs[DDIR_READ];
204 if (!o->max_bs[DDIR_READ])
205 o->max_bs[DDIR_READ] = o->bs[DDIR_READ];
206 if (!o->min_bs[DDIR_WRITE])
207 o->min_bs[DDIR_WRITE]= o->bs[DDIR_WRITE];
208 if (!o->max_bs[DDIR_WRITE])
209 o->max_bs[DDIR_WRITE] = o->bs[DDIR_WRITE];
Jens Axboea00735e2006-11-03 08:58:08 +0100210
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100211 o->rw_min_bs = min(o->min_bs[DDIR_READ], o->min_bs[DDIR_WRITE]);
Jens Axboea00735e2006-11-03 08:58:08 +0100212
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100213 if (!o->file_size_high)
214 o->file_size_high = o->file_size_low;
Jens Axboe9c60ce62007-03-15 09:14:47 +0100215
Jens Axboe16b462a2006-10-30 12:35:18 +0100216 if (td_read(td) && !td_rw(td))
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100217 o->verify = 0;
Jens Axboebb8895e2006-10-30 15:14:48 +0100218
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100219 if (o->norandommap && o->verify != VERIFY_NONE) {
Jens Axboebb8895e2006-10-30 15:14:48 +0100220 log_err("fio: norandommap given, verify disabled\n");
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100221 o->verify = VERIFY_NONE;
Jens Axboebb8895e2006-10-30 15:14:48 +0100222 }
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100223 if (o->bs_unaligned && (o->odirect || td->io_ops->flags & FIO_RAWIO))
Jens Axboe690adba2006-10-30 15:25:09 +0100224 log_err("fio: bs_unaligned may not work with raw io\n");
Jens Axboee0a22332006-12-20 12:54:25 +0100225
226 /*
Jens Axboe48097d52007-02-17 06:30:44 +0100227 * thinktime_spin must be less than thinktime
228 */
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100229 if (o->thinktime_spin > o->thinktime)
230 o->thinktime_spin = o->thinktime;
Jens Axboee916b392007-02-20 14:37:26 +0100231
232 /*
233 * The low water mark cannot be bigger than the iodepth
234 */
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100235 if (o->iodepth_low > o->iodepth || !o->iodepth_low) {
Jens Axboe9467b772007-02-27 19:56:43 +0100236 /*
237 * syslet work around - if the workload is sequential,
238 * we want to let the queue drain all the way down to
239 * avoid seeking between async threads
240 */
241 if (!strcmp(td->io_ops->name, "syslet-rw") && !td_random(td))
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100242 o->iodepth_low = 1;
Jens Axboe9467b772007-02-27 19:56:43 +0100243 else
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100244 o->iodepth_low = o->iodepth;
Jens Axboe9467b772007-02-27 19:56:43 +0100245 }
Jens Axboecb5ab512007-02-26 12:57:09 +0100246
247 /*
248 * If batch number isn't set, default to the same as iodepth
249 */
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100250 if (o->iodepth_batch > o->iodepth || !o->iodepth_batch)
251 o->iodepth_batch = o->iodepth;
Jens Axboeb5af8292007-03-08 12:43:13 +0100252
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100253 if (o->nr_files > td->files_index)
254 o->nr_files = td->files_index;
Jens Axboe9f9214f2007-03-13 14:02:16 +0100255
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100256 if (o->open_files > o->nr_files || !o->open_files)
257 o->open_files = o->nr_files;
Jens Axboe4e991c22007-03-15 11:41:11 +0100258
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100259 if ((o->rate && o->rate_iops) || (o->ratemin && o->rate_iops_min)) {
Jens Axboe4e991c22007-03-15 11:41:11 +0100260 log_err("fio: rate and rate_iops are mutually exclusive\n");
261 return 1;
262 }
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100263 if ((o->rate < o->ratemin) || (o->rate_iops < o->rate_iops_min)) {
Jens Axboe4e991c22007-03-15 11:41:11 +0100264 log_err("fio: minimum rate exceeds rate\n");
265 return 1;
266 }
267
268 return 0;
Jens Axboee1f36502006-10-27 10:54:08 +0200269}
270
Jens Axboe906c8d72006-06-13 09:37:56 +0200271/*
Jens Axboef8977ee2006-11-06 14:03:03 +0100272 * This function leaks the buffer
273 */
274static char *to_kmg(unsigned int val)
275{
276 char *buf = malloc(32);
Jens Axboef3502ba2007-02-14 01:27:09 +0100277 char post[] = { 0, 'K', 'M', 'G', 'P', 'E', 0 };
Jens Axboef8977ee2006-11-06 14:03:03 +0100278 char *p = post;
279
Jens Axboe245142f2006-11-08 12:49:21 +0100280 do {
Jens Axboef8977ee2006-11-06 14:03:03 +0100281 if (val & 1023)
282 break;
283
284 val >>= 10;
285 p++;
Jens Axboe245142f2006-11-08 12:49:21 +0100286 } while (*p);
Jens Axboef8977ee2006-11-06 14:03:03 +0100287
288 snprintf(buf, 31, "%u%c", val, *p);
289 return buf;
290}
291
Jens Axboe09629a92007-03-09 09:00:06 +0100292/* External engines are specified by "external:name.o") */
293static const char *get_engine_name(const char *str)
294{
295 char *p = strstr(str, ":");
296
297 if (!p)
298 return str;
299
300 p++;
301 strip_blank_front(&p);
302 strip_blank_end(p);
303 return p;
304}
305
Jens Axboee132cba2007-03-14 14:23:54 +0100306static int exists_and_not_file(const char *filename)
307{
308 struct stat sb;
309
310 if (lstat(filename, &sb) == -1)
311 return 0;
312
313 if (S_ISREG(sb.st_mode))
314 return 0;
315
316 return 1;
317}
318
Jens Axboef8977ee2006-11-06 14:03:03 +0100319/*
Jens Axboe9c60ce62007-03-15 09:14:47 +0100320 * Initialize the various random states we need (random io, block size ranges,
321 * read/write mix, etc).
322 */
323static int init_random_state(struct thread_data *td)
324{
325 unsigned long seeds[6];
Jens Axboe68727072007-03-15 20:49:25 +0100326 int fd;
Jens Axboe9c60ce62007-03-15 09:14:47 +0100327
328 fd = open("/dev/urandom", O_RDONLY);
329 if (fd == -1) {
330 td_verror(td, errno, "open");
331 return 1;
332 }
333
334 if (read(fd, seeds, sizeof(seeds)) < (int) sizeof(seeds)) {
335 td_verror(td, EIO, "read");
336 close(fd);
337 return 1;
338 }
339
340 close(fd);
341
342 os_random_seed(seeds[0], &td->bsrange_state);
343 os_random_seed(seeds[1], &td->verify_state);
344 os_random_seed(seeds[2], &td->rwmix_state);
345
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100346 if (td->o.file_service_type == FIO_FSERVICE_RANDOM)
Jens Axboe9c60ce62007-03-15 09:14:47 +0100347 os_random_seed(seeds[3], &td->next_file_state);
348
349 os_random_seed(seeds[5], &td->file_size_state);
350
351 if (!td_random(td))
352 return 0;
353
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100354 if (td->o.rand_repeatable)
Jens Axboe9c60ce62007-03-15 09:14:47 +0100355 seeds[4] = FIO_RANDSEED * td->thread_number;
356
Jens Axboe9c60ce62007-03-15 09:14:47 +0100357 os_random_seed(seeds[4], &td->random_state);
358 return 0;
359}
360
361
362/*
Jens Axboe906c8d72006-06-13 09:37:56 +0200363 * Adds a job to the list of things todo. Sanitizes the various options
364 * to make sure we don't have conflicts, and initializes various
365 * members of td.
366 */
Jens Axboe75154842006-06-01 13:56:09 +0200367static int add_job(struct thread_data *td, const char *jobname, int job_add_num)
Jens Axboeebac4652005-12-08 15:25:21 +0100368{
Jens Axboe413dd452007-02-23 09:26:09 +0100369 const char *ddir_str[] = { NULL, "read", "write", "rw", NULL,
370 "randread", "randwrite", "randrw" };
Jens Axboeaf52b342007-03-13 10:07:47 +0100371 unsigned int i;
Jens Axboe53cdc682006-10-18 11:50:58 +0200372 struct fio_file *f;
Jens Axboe09629a92007-03-09 09:00:06 +0100373 const char *engine;
Jens Axboeaf52b342007-03-13 10:07:47 +0100374 char fname[PATH_MAX];
Jens Axboee132cba2007-03-14 14:23:54 +0100375 int numjobs, file_alloced;
Jens Axboeebac4652005-12-08 15:25:21 +0100376
Jens Axboeebac4652005-12-08 15:25:21 +0100377 /*
378 * the def_thread is just for options, it's not a real job
379 */
380 if (td == &def_thread)
381 return 0;
382
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100383 engine = get_engine_name(td->o.ioengine);
Jens Axboe09629a92007-03-09 09:00:06 +0100384 td->io_ops = load_ioengine(td, engine);
385 if (!td->io_ops) {
386 log_err("fio: failed to load engine %s\n", engine);
Jens Axboe205927a2007-03-15 11:06:32 +0100387 goto err;
Jens Axboe09629a92007-03-09 09:00:06 +0100388 }
Jens Axboedf641192006-10-12 07:55:41 +0200389
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100390 if (td->o.use_thread)
Jens Axboe9cedf162007-03-12 11:29:30 +0100391 nr_thread++;
392 else
393 nr_process++;
394
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100395 if (td->o.odirect)
Jens Axboe690adba2006-10-30 15:25:09 +0100396 td->io_ops->flags |= FIO_RAWIO;
397
Jens Axboee132cba2007-03-14 14:23:54 +0100398 file_alloced = 0;
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100399 if (!td->o.filename && !td->files_index) {
Jens Axboee132cba2007-03-14 14:23:54 +0100400 file_alloced = 1;
Jens Axboe80be24f2007-03-12 11:01:25 +0100401
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100402 if (td->o.nr_files == 1 && exists_and_not_file(jobname))
Jens Axboee132cba2007-03-14 14:23:54 +0100403 add_file(td, jobname);
Jens Axboe7b05a212007-03-13 11:17:07 +0100404 else {
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100405 for (i = 0; i < td->o.nr_files; i++) {
Jens Axboee132cba2007-03-14 14:23:54 +0100406 sprintf(fname, "%s.%d.%d", jobname, td->thread_number, i);
Jens Axboe7b05a212007-03-13 11:17:07 +0100407 add_file(td, fname);
408 }
Jens Axboeaf52b342007-03-13 10:07:47 +0100409 }
Jens Axboe0af7b542006-02-17 10:10:12 +0100410 }
Jens Axboeebac4652005-12-08 15:25:21 +0100411
Jens Axboe4e991c22007-03-15 11:41:11 +0100412 if (fixup_options(td))
413 goto err;
Jens Axboee0a22332006-12-20 12:54:25 +0100414
Jens Axboe53cdc682006-10-18 11:50:58 +0200415 for_each_file(td, f, i) {
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100416 if (td->o.directory && f->filetype == FIO_TYPE_FILE) {
417 sprintf(fname, "%s/%s", td->o.directory, f->file_name);
Jens Axboeaf52b342007-03-13 10:07:47 +0100418 f->file_name = strdup(fname);
419 }
Jens Axboe53cdc682006-10-18 11:50:58 +0200420 }
421
Jens Axboe07739b52007-03-08 20:25:46 +0100422 td->mutex = fio_sem_init(0);
Jens Axboeebac4652005-12-08 15:25:21 +0100423
Jens Axboe756867b2007-03-06 15:19:24 +0100424 td->ts.clat_stat[0].min_val = td->ts.clat_stat[1].min_val = ULONG_MAX;
425 td->ts.slat_stat[0].min_val = td->ts.slat_stat[1].min_val = ULONG_MAX;
426 td->ts.bw_stat[0].min_val = td->ts.bw_stat[1].min_val = ULONG_MAX;
Jens Axboeebac4652005-12-08 15:25:21 +0100427
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100428 if ((td->o.stonewall || td->o.numjobs > 1) && prev_group_jobs) {
Jens Axboe3c5df6f2007-03-12 15:09:24 +0100429 prev_group_jobs = 0;
Jens Axboeebac4652005-12-08 15:25:21 +0100430 groupid++;
Jens Axboe3c5df6f2007-03-12 15:09:24 +0100431 }
Jens Axboeebac4652005-12-08 15:25:21 +0100432
433 td->groupid = groupid;
Jens Axboe3c5df6f2007-03-12 15:09:24 +0100434 prev_group_jobs++;
Jens Axboeebac4652005-12-08 15:25:21 +0100435
Jens Axboe9c60ce62007-03-15 09:14:47 +0100436 if (init_random_state(td))
437 goto err;
438
Jens Axboeebac4652005-12-08 15:25:21 +0100439 if (setup_rate(td))
440 goto err;
441
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100442 if (td->o.write_lat_log) {
Jens Axboe756867b2007-03-06 15:19:24 +0100443 setup_log(&td->ts.slat_log);
444 setup_log(&td->ts.clat_log);
Jens Axboeebac4652005-12-08 15:25:21 +0100445 }
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100446 if (td->o.write_bw_log)
Jens Axboe756867b2007-03-06 15:19:24 +0100447 setup_log(&td->ts.bw_log);
Jens Axboeebac4652005-12-08 15:25:21 +0100448
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100449 if (!td->o.name)
450 td->o.name = strdup(jobname);
Jens Axboe01452052006-06-07 10:29:47 +0200451
Jens Axboec6ae0a52006-06-12 11:04:44 +0200452 if (!terse_output) {
Jens Axboeb990b5c2006-09-14 09:48:22 +0200453 if (!job_add_num) {
Jens Axboeba0fbe12007-03-09 14:34:23 +0100454 if (!strcmp(td->io_ops->name, "cpuio"))
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100455 log_info("%s: ioengine=cpu, cpuload=%u, cpucycle=%u\n", td->o.name, td->o.cpuload, td->o.cpucycle);
Jens Axboef8977ee2006-11-06 14:03:03 +0100456 else {
457 char *c1, *c2, *c3, *c4;
458
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100459 c1 = to_kmg(td->o.min_bs[DDIR_READ]);
460 c2 = to_kmg(td->o.max_bs[DDIR_READ]);
461 c3 = to_kmg(td->o.min_bs[DDIR_WRITE]);
462 c4 = to_kmg(td->o.max_bs[DDIR_WRITE]);
Jens Axboef8977ee2006-11-06 14:03:03 +0100463
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100464 log_info("%s: (g=%d): rw=%s, bs=%s-%s/%s-%s, ioengine=%s, iodepth=%u\n", td->o.name, td->groupid, ddir_str[td->o.td_ddir], c1, c2, c3, c4, td->io_ops->name, td->o.iodepth);
Jens Axboef8977ee2006-11-06 14:03:03 +0100465
466 free(c1);
467 free(c2);
468 free(c3);
469 free(c4);
470 }
Jens Axboeb990b5c2006-09-14 09:48:22 +0200471 } else if (job_add_num == 1)
Jens Axboe6d861442007-03-15 09:22:23 +0100472 log_info("...\n");
Jens Axboec6ae0a52006-06-12 11:04:44 +0200473 }
Jens Axboeebac4652005-12-08 15:25:21 +0100474
475 /*
476 * recurse add identical jobs, clear numjobs and stonewall options
477 * as they don't apply to sub-jobs
478 */
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100479 numjobs = td->o.numjobs;
Jens Axboeebac4652005-12-08 15:25:21 +0100480 while (--numjobs) {
481 struct thread_data *td_new = get_new_job(0, td);
482
483 if (!td_new)
484 goto err;
485
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100486 td_new->o.numjobs = 1;
487 td_new->o.stonewall = 0;
Jens Axboee132cba2007-03-14 14:23:54 +0100488
489 if (file_alloced) {
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100490 td_new->o.filename = NULL;
Jens Axboee132cba2007-03-14 14:23:54 +0100491 td_new->files_index = 0;
492 td_new->files = NULL;
493 }
494
Jens Axboe75154842006-06-01 13:56:09 +0200495 job_add_num = numjobs - 1;
Jens Axboeebac4652005-12-08 15:25:21 +0100496
Jens Axboe75154842006-06-01 13:56:09 +0200497 if (add_job(td_new, jobname, job_add_num))
Jens Axboeebac4652005-12-08 15:25:21 +0100498 goto err;
499 }
Jens Axboe3c5df6f2007-03-12 15:09:24 +0100500
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100501 if (td->o.numjobs > 1) {
Jens Axboe3c5df6f2007-03-12 15:09:24 +0100502 groupid++;
503 prev_group_jobs = 0;
504 }
505
Jens Axboeebac4652005-12-08 15:25:21 +0100506 return 0;
507err:
508 put_job(td);
509 return -1;
510}
511
Jens Axboeebac4652005-12-08 15:25:21 +0100512static int is_empty_or_comment(char *line)
513{
514 unsigned int i;
515
516 for (i = 0; i < strlen(line); i++) {
517 if (line[i] == ';')
518 return 1;
Ingo Molnar5cc2da32007-02-20 10:19:44 +0100519 if (line[i] == '#')
520 return 1;
Jens Axboeebac4652005-12-08 15:25:21 +0100521 if (!isspace(line[i]) && !iscntrl(line[i]))
522 return 0;
523 }
524
525 return 1;
526}
527
Jens Axboe313cb202006-12-21 09:50:00 +0100528/*
Jens Axboe07261982006-06-07 10:51:12 +0200529 * This is our [ini] type file parser.
530 */
Jens Axboe1e97cce2006-12-05 11:44:16 +0100531static int parse_jobs_ini(char *file, int stonewall_flag)
Jens Axboeebac4652005-12-08 15:25:21 +0100532{
Jens Axboee1f36502006-10-27 10:54:08 +0200533 unsigned int global;
Jens Axboeebac4652005-12-08 15:25:21 +0100534 struct thread_data *td;
Jens Axboefee3bb42006-10-30 13:32:08 +0100535 char *string, *name;
Jens Axboeebac4652005-12-08 15:25:21 +0100536 fpos_t off;
537 FILE *f;
538 char *p;
Jens Axboe0c7e37a2006-06-13 14:53:38 +0200539 int ret = 0, stonewall;
Jens Axboeebac4652005-12-08 15:25:21 +0100540
541 f = fopen(file, "r");
542 if (!f) {
Jens Axboeaea47d42006-05-26 19:27:29 +0200543 perror("fopen job file");
Jens Axboeebac4652005-12-08 15:25:21 +0100544 return 1;
545 }
546
547 string = malloc(4096);
548 name = malloc(256);
Jens Axboefee3bb42006-10-30 13:32:08 +0100549 memset(name, 0, 256);
Jens Axboeebac4652005-12-08 15:25:21 +0100550
Jens Axboe0c7e37a2006-06-13 14:53:38 +0200551 stonewall = stonewall_flag;
Jens Axboe7c124ac2006-10-30 13:36:52 +0100552 do {
553 p = fgets(string, 4095, f);
554 if (!p)
Jens Axboe45410ac2006-06-07 11:10:39 +0200555 break;
Jens Axboeebac4652005-12-08 15:25:21 +0100556 if (is_empty_or_comment(p))
557 continue;
Jens Axboefee3bb42006-10-30 13:32:08 +0100558 if (sscanf(p, "[%255s]", name) != 1)
Jens Axboeebac4652005-12-08 15:25:21 +0100559 continue;
560
561 global = !strncmp(name, "global", 6);
562
563 name[strlen(name) - 1] = '\0';
564
565 td = get_new_job(global, &def_thread);
Jens Axboe45410ac2006-06-07 11:10:39 +0200566 if (!td) {
567 ret = 1;
568 break;
569 }
Jens Axboeebac4652005-12-08 15:25:21 +0100570
Jens Axboe972cfd22006-06-09 11:08:56 +0200571 /*
572 * Seperate multiple job files by a stonewall
573 */
Jens Axboef9481912006-06-09 11:37:28 +0200574 if (!global && stonewall) {
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100575 td->o.stonewall = stonewall;
Jens Axboe972cfd22006-06-09 11:08:56 +0200576 stonewall = 0;
577 }
578
Jens Axboeebac4652005-12-08 15:25:21 +0100579 fgetpos(f, &off);
580 while ((p = fgets(string, 4096, f)) != NULL) {
581 if (is_empty_or_comment(p))
582 continue;
Jens Axboee1f36502006-10-27 10:54:08 +0200583
Jens Axboeb6754f92006-05-30 14:29:32 +0200584 strip_blank_front(&p);
Jens Axboe7c124ac2006-10-30 13:36:52 +0100585
586 if (p[0] == '[')
587 break;
588
Jens Axboe4ae3f762006-05-30 13:35:14 +0200589 strip_blank_end(p);
Jens Axboeaea47d42006-05-26 19:27:29 +0200590
Jens Axboee1f36502006-10-27 10:54:08 +0200591 fgetpos(f, &off);
Jens Axboeebac4652005-12-08 15:25:21 +0100592
Jens Axboe45410ac2006-06-07 11:10:39 +0200593 /*
594 * Don't break here, continue parsing options so we
595 * dump all the bad ones. Makes trial/error fixups
596 * easier on the user.
597 */
Jens Axboe214e1ec2007-03-15 10:48:13 +0100598 ret |= fio_option_parse(td, p);
Jens Axboeebac4652005-12-08 15:25:21 +0100599 }
Jens Axboeebac4652005-12-08 15:25:21 +0100600
Jens Axboe45410ac2006-06-07 11:10:39 +0200601 if (!ret) {
602 fsetpos(f, &off);
603 ret = add_job(td, name, 0);
Jens Axboeb1508cf2006-11-02 09:12:40 +0100604 } else {
605 log_err("fio: job %s dropped\n", name);
606 put_job(td);
Jens Axboe45410ac2006-06-07 11:10:39 +0200607 }
Jens Axboe7c124ac2006-10-30 13:36:52 +0100608 } while (!ret);
Jens Axboeebac4652005-12-08 15:25:21 +0100609
610 free(string);
611 free(name);
612 fclose(f);
Jens Axboe45410ac2006-06-07 11:10:39 +0200613 return ret;
Jens Axboeebac4652005-12-08 15:25:21 +0100614}
615
616static int fill_def_thread(void)
617{
618 memset(&def_thread, 0, sizeof(def_thread));
619
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100620 if (fio_getaffinity(getpid(), &def_thread.o.cpumask) == -1) {
Jens Axboeebac4652005-12-08 15:25:21 +0100621 perror("sched_getaffinity");
622 return 1;
623 }
624
625 /*
Jens Axboeee738492007-01-10 11:23:16 +0100626 * fill default options
Jens Axboeebac4652005-12-08 15:25:21 +0100627 */
Jens Axboe214e1ec2007-03-15 10:48:13 +0100628 fio_fill_default_options(&def_thread);
Jens Axboeee738492007-01-10 11:23:16 +0100629
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100630 def_thread.o.timeout = def_timeout;
631 def_thread.o.write_bw_log = write_bw_log;
632 def_thread.o.write_lat_log = write_lat_log;
Jens Axboeee738492007-01-10 11:23:16 +0100633
Jens Axboeebac4652005-12-08 15:25:21 +0100634#ifdef FIO_HAVE_DISK_UTIL
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100635 def_thread.o.do_disk_util = 1;
Jens Axboeebac4652005-12-08 15:25:21 +0100636#endif
637
638 return 0;
639}
640
Jens Axboeebac4652005-12-08 15:25:21 +0100641static void free_shm(void)
642{
643 struct shmid_ds sbuf;
644
645 if (threads) {
Jens Axboe2c0ecd22006-06-08 13:25:41 +0200646 shmdt((void *) threads);
Jens Axboeebac4652005-12-08 15:25:21 +0100647 threads = NULL;
648 shmctl(shm_id, IPC_RMID, &sbuf);
649 }
650}
651
Jens Axboe906c8d72006-06-13 09:37:56 +0200652/*
653 * The thread area is shared between the main process and the job
654 * threads/processes. So setup a shared memory segment that will hold
655 * all the job info.
656 */
Jens Axboeebac4652005-12-08 15:25:21 +0100657static int setup_thread_area(void)
658{
659 /*
660 * 1024 is too much on some machines, scale max_jobs if
661 * we get a failure that looks like too large a shm segment
662 */
663 do {
Jens Axboe906c8d72006-06-13 09:37:56 +0200664 size_t size = max_jobs * sizeof(struct thread_data);
Jens Axboeebac4652005-12-08 15:25:21 +0100665
Jens Axboe906c8d72006-06-13 09:37:56 +0200666 shm_id = shmget(0, size, IPC_CREAT | 0600);
Jens Axboeebac4652005-12-08 15:25:21 +0100667 if (shm_id != -1)
668 break;
669 if (errno != EINVAL) {
670 perror("shmget");
671 break;
672 }
673
674 max_jobs >>= 1;
675 } while (max_jobs);
676
677 if (shm_id == -1)
678 return 1;
679
680 threads = shmat(shm_id, NULL, 0);
681 if (threads == (void *) -1) {
682 perror("shmat");
683 return 1;
684 }
685
686 atexit(free_shm);
687 return 0;
688}
689
Jens Axboe214e1ec2007-03-15 10:48:13 +0100690static void usage(void)
Jens Axboeb4692822006-10-27 13:43:22 +0200691{
Jens Axboe214e1ec2007-03-15 10:48:13 +0100692 printf("%s\n", fio_version_string);
693 printf("\t--output\tWrite output to file\n");
694 printf("\t--timeout\tRuntime in seconds\n");
695 printf("\t--latency-log\tGenerate per-job latency logs\n");
696 printf("\t--bandwidth-log\tGenerate per-job bandwidth logs\n");
697 printf("\t--minimal\tMinimal (terse) output\n");
698 printf("\t--version\tPrint version info and exit\n");
699 printf("\t--help\t\tPrint this page\n");
700 printf("\t--cmdhelp=cmd\tPrint command help, \"all\" for all of them\n");
Jens Axboeb4692822006-10-27 13:43:22 +0200701}
702
Jens Axboe214e1ec2007-03-15 10:48:13 +0100703static int parse_cmd_line(int argc, char *argv[])
704{
705 struct thread_data *td = NULL;
706 int c, ini_idx = 0, lidx, ret, dont_add_job = 0;
707
708 while ((c = getopt_long_only(argc, argv, "", long_options, &lidx)) != -1) {
709 switch (c) {
710 case 't':
711 def_timeout = atoi(optarg);
712 break;
713 case 'l':
714 write_lat_log = 1;
715 break;
716 case 'w':
717 write_bw_log = 1;
718 break;
719 case 'o':
720 f_out = fopen(optarg, "w+");
721 if (!f_out) {
722 perror("fopen output");
723 exit(1);
724 }
725 f_err = f_out;
726 break;
727 case 'm':
728 terse_output = 1;
729 break;
730 case 'h':
731 usage();
732 exit(0);
733 case 'c':
734 exit(fio_show_option_help(optarg));
735 case 'v':
736 printf("%s\n", fio_version_string);
737 exit(0);
738 case FIO_GETOPT_JOB: {
739 const char *opt = long_options[lidx].name;
740 char *val = optarg;
741
742 if (!strncmp(opt, "name", 4) && td) {
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100743 ret = add_job(td, td->o.name ?: "fio", 0);
Jens Axboe214e1ec2007-03-15 10:48:13 +0100744 if (ret) {
745 put_job(td);
746 return 0;
747 }
748 td = NULL;
749 }
750 if (!td) {
751 int global = !strncmp(val, "global", 6);
752
753 td = get_new_job(global, &def_thread);
754 if (!td)
755 return 0;
756 }
757
758 ret = fio_cmd_option_parse(td, opt, val);
759 if (ret)
760 dont_add_job = 1;
761 break;
762 }
763 default:
764 break;
765 }
766 }
767
768 if (td) {
769 if (dont_add_job)
770 put_job(td);
771 else {
Jens Axboe2dc1bbe2007-03-15 15:01:33 +0100772 ret = add_job(td, td->o.name ?: "fio", 0);
Jens Axboe214e1ec2007-03-15 10:48:13 +0100773 if (ret)
774 put_job(td);
775 }
776 }
777
778 while (optind < argc) {
779 ini_idx++;
780 ini_file = realloc(ini_file, ini_idx * sizeof(char *));
781 ini_file[ini_idx - 1] = strdup(argv[optind]);
782 optind++;
783 }
784
785 return ini_idx;
786}
787
788
Jens Axboeebac4652005-12-08 15:25:21 +0100789int parse_options(int argc, char *argv[])
790{
Jens Axboe972cfd22006-06-09 11:08:56 +0200791 int job_files, i;
792
Jens Axboeb4692822006-10-27 13:43:22 +0200793 f_out = stdout;
794 f_err = stderr;
795
Jens Axboe214e1ec2007-03-15 10:48:13 +0100796 fio_options_dup_and_init(long_options);
Jens Axboeb4692822006-10-27 13:43:22 +0200797
Jens Axboeebac4652005-12-08 15:25:21 +0100798 if (setup_thread_area())
799 return 1;
800 if (fill_def_thread())
801 return 1;
802
Jens Axboe972cfd22006-06-09 11:08:56 +0200803 job_files = parse_cmd_line(argc, argv);
Jens Axboeebac4652005-12-08 15:25:21 +0100804
Jens Axboe972cfd22006-06-09 11:08:56 +0200805 for (i = 0; i < job_files; i++) {
806 if (fill_def_thread())
807 return 1;
Jens Axboe0c7e37a2006-06-13 14:53:38 +0200808 if (parse_jobs_ini(ini_file[i], i))
Jens Axboe972cfd22006-06-09 11:08:56 +0200809 return 1;
Jens Axboe88c6ed82006-06-09 11:28:10 +0200810 free(ini_file[i]);
Jens Axboe972cfd22006-06-09 11:08:56 +0200811 }
Jens Axboeebac4652005-12-08 15:25:21 +0100812
Jens Axboe88c6ed82006-06-09 11:28:10 +0200813 free(ini_file);
Jens Axboeb4692822006-10-27 13:43:22 +0200814
815 if (!thread_number) {
816 log_err("No jobs defined(s)\n");
Jens Axboeb4692822006-10-27 13:43:22 +0200817 return 1;
818 }
819
Jens Axboeebac4652005-12-08 15:25:21 +0100820 return 0;
821}