X-Git-Url: https://git.kernel.dk/?p=fio.git;a=blobdiff_plain;f=init.c;h=cf66ec0e8ee9875ded959cb616be5c633c4e13fc;hp=bbf6f2a8f3bb9b5452855e8ed5ba0c694610383d;hb=ebb1415f729c123b8a13bcbd667bf4b4cc95b4d4;hpb=756867bd127cfa1fb27ae2d4ba973af8cb341f13 diff --git a/init.c b/init.c index bbf6f2a8..cf66ec0e 100644 --- a/init.c +++ b/init.c @@ -22,7 +22,6 @@ #define td_var_offset(var) ((size_t) &((struct thread_data *)0)->var) -static int str_ioengine_cb(void *, const char *); static int str_mem_cb(void *, const char *); static int str_lockmem_cb(void *, unsigned long *); #ifdef FIO_HAVE_IOPRIO @@ -31,6 +30,9 @@ static int str_prioclass_cb(void *, unsigned int *); #endif static int str_exitall_cb(void); static int str_cpumask_cb(void *, unsigned int *); +static int str_fst_cb(void *, const char *); +static int str_filename_cb(void *, const char *); +static int str_directory_cb(void *, const char *); #define __stringify_1(x) #x #define __stringify(x) __stringify_1(x) @@ -55,13 +57,15 @@ static struct fio_option options[] = { .name = "directory", .type = FIO_OPT_STR_STORE, .off1 = td_var_offset(directory), + .cb = str_directory_cb, .help = "Directory to store files in", }, { .name = "filename", .type = FIO_OPT_STR_STORE, .off1 = td_var_offset(filename), - .help = "Force the use of a specific file", + .cb = str_filename_cb, + .help = "File(s) to use for the workload", }, { .name = "rw", @@ -80,8 +84,8 @@ static struct fio_option options[] = { }, { .name = "ioengine", - .type = FIO_OPT_STR, - .cb = str_ioengine_cb, + .type = FIO_OPT_STR_STORE, + .off1 = td_var_offset(ioengine), .help = "IO engine to use", .def = "sync", .posval = { @@ -103,6 +107,7 @@ static struct fio_option options[] = { #ifdef FIO_HAVE_SYSLET { .ival = "syslet-rw", }, #endif + { .ival = "cpuio", }, { .ival = "external", }, }, }, @@ -181,9 +186,16 @@ static struct fio_option options[] = { .help = "Split job workload between this number of files", .def = "1", }, + { + .name = "openfiles", + .type = FIO_OPT_INT, + .off1 = td_var_offset(open_files), + .help = "Number of files to keep open at the same time", + }, { .name = "file_service_type", .type = FIO_OPT_STR, + .cb = str_fst_cb, .off1 = td_var_offset(file_service_type), .help = "How to select which file to service next", .def = "roundrobin", @@ -469,7 +481,8 @@ static struct fio_option options[] = { .name = "cpuchunks", .type = FIO_OPT_INT, .off1 = td_var_offset(cpucycle), - .help = "Length of the CPU burn cycles", + .help = "Length of the CPU burn cycles (usecs)", + .def = "50000", }, #ifdef FIO_HAVE_CPU_AFFINITY { @@ -486,6 +499,13 @@ static struct fio_option options[] = { .help = "Include fsync at the end of job", .def = "0", }, + { + .name = "fsync_on_close", + .type = FIO_OPT_BOOL, + .off1 = td_var_offset(fsync_on_close), + .help = "fsync files on close", + .def = "0", + }, { .name = "unlink", .type = FIO_OPT_BOOL, @@ -614,6 +634,8 @@ FILE *f_err = NULL; static int write_lat_log = 0; int write_bw_log = 0; +static int prev_group_jobs; + FILE *get_f_out() { return f_out; @@ -674,13 +696,13 @@ static void fixup_options(struct thread_data *td) td->iodepth = 1; else { if (!td->iodepth) - td->iodepth = td->nr_files; + td->iodepth = td->open_files; } /* * only really works for sequential io for now, and with 1 file */ - if (td->zone_size && td_random(td) && td->nr_files == 1) + if (td->zone_size && td_random(td) && td->open_files == 1) td->zone_size = 0; /* @@ -710,12 +732,6 @@ static void fixup_options(struct thread_data *td) if (td->bs_unaligned && (td->odirect || td->io_ops->flags & FIO_RAWIO)) log_err("fio: bs_unaligned may not work with raw io\n"); - /* - * O_DIRECT and char doesn't mix, clear that flag if necessary. - */ - if (td->filetype == FIO_TYPE_CHAR && td->odirect) - td->odirect = 0; - /* * thinktime_spin must be less than thinktime */ @@ -742,6 +758,14 @@ static void fixup_options(struct thread_data *td) */ if (td->iodepth_batch > td->iodepth || !td->iodepth_batch) td->iodepth_batch = td->iodepth; + + if (!td->nr_files) + td->nr_files = td->files_index; + else if (td->nr_files > td->files_index) + td->nr_files = td->files_index; + + if (td->open_files > td->nr_files || !td->open_files) + td->open_files = td->nr_files; } /* @@ -765,6 +789,20 @@ static char *to_kmg(unsigned int val) return buf; } +/* External engines are specified by "external:name.o") */ +static const char *get_engine_name(const char *str) +{ + char *p = strstr(str, ":"); + + if (!p) + return str; + + p++; + strip_blank_front(&p); + strip_blank_end(p); + return p; +} + /* * Adds a job to the list of things todo. Sanitizes the various options * to make sure we don't have conflicts, and initializes various @@ -774,9 +812,11 @@ static int add_job(struct thread_data *td, const char *jobname, int job_add_num) { const char *ddir_str[] = { NULL, "read", "write", "rw", NULL, "randread", "randwrite", "randrw" }; - struct stat sb; - int numjobs, i; + unsigned int i; struct fio_file *f; + const char *engine; + char fname[PATH_MAX]; + int numjobs; /* * the def_thread is just for options, it's not a real job @@ -784,80 +824,56 @@ static int add_job(struct thread_data *td, const char *jobname, int job_add_num) if (td == &def_thread) return 0; - assert(td->io_ops); - - if (td->odirect) - td->io_ops->flags |= FIO_RAWIO; - - td->filetype = FIO_TYPE_FILE; - if (td->filename && !lstat(td->filename, &sb)) { - if (S_ISBLK(sb.st_mode)) - td->filetype = FIO_TYPE_BD; - else if (S_ISCHR(sb.st_mode)) - td->filetype = FIO_TYPE_CHAR; + engine = get_engine_name(td->ioengine); + td->io_ops = load_ioengine(td, engine); + if (!td->io_ops) { + log_err("fio: failed to load engine %s\n", engine); + return 1; } - fixup_options(td); - - if (td->filename) - td->nr_uniq_files = 1; + if (td->use_thread) + nr_thread++; else - td->nr_uniq_files = td->nr_files; - - if (td->filetype == FIO_TYPE_FILE || td->filename) { - char tmp[PATH_MAX]; - int len = 0; + nr_process++; - if (td->directory && td->directory[0] != '\0') { - if (lstat(td->directory, &sb) < 0) { - log_err("fio: %s is not a directory\n", td->directory); - td_verror(td, errno, "lstat"); - return 1; - } - if (!S_ISDIR(sb.st_mode)) { - log_err("fio: %s is not a directory\n", td->directory); - return 1; - } - len = sprintf(tmp, "%s/", td->directory); - } + if (td->odirect) + td->io_ops->flags |= FIO_RAWIO; - td->files = malloc(sizeof(struct fio_file) * td->nr_files); + if (!td->filename) { + td->filename = strdup(jobname); - for_each_file(td, f, i) { - memset(f, 0, sizeof(*f)); - f->fd = -1; - - if (td->filename) - sprintf(tmp + len, "%s", td->filename); - else - sprintf(tmp + len, "%s.%d.%d", jobname, td->thread_number, i); - f->file_name = strdup(tmp); + if (td->nr_files == 1) + add_file(td, td->filename); + else { + for (i = 0; i < td->nr_files; i++) { + sprintf(fname, "%s.%d.%d", td->filename, td->thread_number, i); + add_file(td, fname); + } } - } else { - td->nr_files = 1; - td->files = malloc(sizeof(struct fio_file)); - f = &td->files[0]; - - memset(f, 0, sizeof(*f)); - f->fd = -1; - f->file_name = strdup(jobname); } + fixup_options(td); + for_each_file(td, f, i) { - f->file_size = td->total_file_size / td->nr_files; - f->file_offset = td->start_offset; + if (td->directory && f->filetype == FIO_TYPE_FILE) { + sprintf(fname, "%s/%s", td->directory, f->file_name); + f->file_name = strdup(fname); + } } - fio_sem_init(&td->mutex, 0); + td->mutex = fio_sem_init(0); td->ts.clat_stat[0].min_val = td->ts.clat_stat[1].min_val = ULONG_MAX; td->ts.slat_stat[0].min_val = td->ts.slat_stat[1].min_val = ULONG_MAX; td->ts.bw_stat[0].min_val = td->ts.bw_stat[1].min_val = ULONG_MAX; - if (td->stonewall && td->thread_number > 1) + if ((td->stonewall || td->numjobs > 1) && prev_group_jobs) { + prev_group_jobs = 0; groupid++; + } td->groupid = groupid; + prev_group_jobs++; if (setup_rate(td)) goto err; @@ -874,7 +890,7 @@ static int add_job(struct thread_data *td, const char *jobname, int job_add_num) if (!terse_output) { if (!job_add_num) { - if (td->io_ops->flags & FIO_CPUIO) + if (!strcmp(td->io_ops->name, "cpuio")) fprintf(f_out, "%s: ioengine=cpu, cpuload=%u, cpucycle=%u\n", td->name, td->cpuload, td->cpucycle); else { char *c1, *c2, *c3, *c4; @@ -913,6 +929,12 @@ static int add_job(struct thread_data *td, const char *jobname, int job_add_num) if (add_job(td_new, jobname, job_add_num)) goto err; } + + if (td->numjobs > 1) { + groupid++; + prev_group_jobs = 0; + } + return 0; err: put_job(td); @@ -926,10 +948,11 @@ err: int init_random_state(struct thread_data *td) { unsigned long seeds[5]; - int fd, num_maps, blocks, i; + int fd, num_maps, blocks; struct fio_file *f; + unsigned int i; - if (td->io_ops->flags & FIO_CPUIO) + if (td->io_ops->flags & FIO_DISKLESSIO) return 0; fd = open("/dev/urandom", O_RDONLY); @@ -964,6 +987,10 @@ int init_random_state(struct thread_data *td) blocks = (f->real_file_size + td->rw_min_bs - 1) / td->rw_min_bs; num_maps = (blocks + BLOCKS_PER_MAP-1)/ BLOCKS_PER_MAP; f->file_map = malloc(num_maps * sizeof(long)); + if (!f->file_map) { + log_err("fio: failed allocating random map. If running a large number of jobs, try the 'norandommap' option\n"); + return 1; + } f->num_maps = num_maps; memset(f->file_map, 0, num_maps * sizeof(long)); } @@ -1006,7 +1033,7 @@ static int is_empty_or_comment(char *line) /* * Check if mmap/mmaphuge has a :/foo/bar/file at the end. If so, return that. */ -static char *get_mmap_file(const char *str) +static char *get_opt_postfix(const char *str) { char *p = strstr(str, ":"); @@ -1024,7 +1051,7 @@ static int str_mem_cb(void *data, const char *mem) struct thread_data *td = data; if (td->mem_type == MEM_MMAPHUGE || td->mem_type == MEM_MMAP) { - td->mmapfile = get_mmap_file(mem); + td->mmapfile = get_opt_postfix(mem); if (td->mem_type == MEM_MMAPHUGE && !td->mmapfile) { log_err("fio: mmaphuge:/path/to/file\n"); return 1; @@ -1034,32 +1061,6 @@ static int str_mem_cb(void *data, const char *mem) return 0; } -/* External engines are specified by "external:name.o") */ -static const char *get_engine_name(const char *str) -{ - char *p = strstr(str, ":"); - - if (!p) - return str; - - p++; - strip_blank_front(&p); - strip_blank_end(p); - return p; -} - -static int str_ioengine_cb(void *data, const char *str) -{ - struct thread_data *td = data; - const char *name = get_engine_name(str); - - td->io_ops = load_ioengine(td, name); - if (td->io_ops) - return 0; - - return 1; -} - static int str_lockmem_cb(void fio_unused *data, unsigned long *val) { mlock_size = *val; @@ -1098,6 +1099,56 @@ static int str_cpumask_cb(void *data, unsigned int *val) return 0; } +static int str_fst_cb(void *data, const char *str) +{ + struct thread_data *td = data; + char *nr = get_opt_postfix(str); + + td->file_service_nr = 1; + if (nr) + td->file_service_nr = atoi(nr); + + return 0; +} + +static int str_filename_cb(void *data, const char *input) +{ + struct thread_data *td = data; + char *fname, *str, *p; + + p = str = strdup(input); + + strip_blank_front(&str); + strip_blank_end(str); + + while ((fname = strsep(&str, ":")) != NULL) { + if (!strlen(fname)) + break; + add_file(td, fname); + } + + free(p); + return 0; +} + +static int str_directory_cb(void *data, const char fio_unused *str) +{ + struct thread_data *td = data; + struct stat sb; + + if (lstat(td->directory, &sb) < 0) { + log_err("fio: %s is not a directory\n", td->directory); + td_verror(td, errno, "lstat"); + return 1; + } + if (!S_ISDIR(sb.st_mode)) { + log_err("fio: %s is not a directory\n", td->directory); + return 1; + } + + return 0; +} + /* * This is our [ini] type file parser. */