/* * fio - the flexible io tester * * Copyright (C) 2005 Jens Axboe * Copyright (C) 2006-2012 Jens Axboe * * The license below covers all files distributed with fio unless otherwise * noted in the file itself. * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License version 2 as * published by the Free Software Foundation. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * */ #include #include #include #include #include #include #include "fio.h" #include "smalloc.h" #include "os/os.h" #include "filelock.h" /* * Just expose an empty list, if the OS does not support disk util stats */ #ifndef FIO_HAVE_DISK_UTIL FLIST_HEAD(disk_list); #endif unsigned long arch_flags = 0; uintptr_t page_mask = 0; uintptr_t page_size = 0; static const char *fio_os_strings[os_nr] = { "Invalid", "Linux", "AIX", "FreeBSD", "HP-UX", "OSX", "NetBSD", "OpenBSD", "Solaris", "Windows", "Android", }; static const char *fio_arch_strings[arch_nr] = { "Invalid", "x86-64", "x86", "ppc", "ia64", "s390", "alpha", "sparc", "sparc64", "arm", "sh", "hppa", "generic" }; static void reset_io_counters(struct thread_data *td) { int ddir; for (ddir = 0; ddir < DDIR_RWDIR_CNT; ddir++) { td->stat_io_bytes[ddir] = 0; td->this_io_bytes[ddir] = 0; td->stat_io_blocks[ddir] = 0; td->this_io_blocks[ddir] = 0; td->rate_bytes[ddir] = 0; td->rate_blocks[ddir] = 0; } td->zone_bytes = 0; td->last_was_sync = 0; td->rwmix_issues = 0; /* * reset file done count if we are to start over */ if (td->o.time_based || td->o.loops || td->o.do_verify) td->nr_done_files = 0; } void clear_io_state(struct thread_data *td) { struct fio_file *f; unsigned int i; reset_io_counters(td); close_files(td); for_each_file(td, f, i) fio_file_clear_done(f); /* * Set the same seed to get repeatable runs */ td_fill_rand_seeds(td); } void reset_all_stats(struct thread_data *td) { struct timeval tv; int i; reset_io_counters(td); for (i = 0; i < DDIR_RWDIR_CNT; i++) { td->io_bytes[i] = 0; td->io_blocks[i] = 0; td->io_issues[i] = 0; td->ts.total_io_u[i] = 0; td->ts.runtime[i] = 0; td->rwmix_issues = 0; } fio_gettime(&tv, NULL); memcpy(&td->epoch, &tv, sizeof(tv)); memcpy(&td->start, &tv, sizeof(tv)); lat_target_reset(td); } void reset_fio_state(void) { groupid = 0; thread_number = 0; stat_number = 0; done_secs = 0; } const char *fio_get_os_string(int nr) { if (nr < os_nr) return fio_os_strings[nr]; return NULL; } const char *fio_get_arch_string(int nr) { if (nr < arch_nr) return fio_arch_strings[nr]; return NULL; } void td_set_runstate(struct thread_data *td, int runstate) { if (td->runstate == runstate) return; dprint(FD_PROCESS, "pid=%d: runstate %d -> %d\n", (int) td->pid, td->runstate, runstate); td->runstate = runstate; } int td_bump_runstate(struct thread_data *td, int new_state) { int old_state = td->runstate; td_set_runstate(td, new_state); return old_state; } void td_restore_runstate(struct thread_data *td, int old_state) { td_set_runstate(td, old_state); } void fio_mark_td_terminate(struct thread_data *td) { fio_gettime(&td->terminate_time, NULL); write_barrier(); td->terminate = 1; } void fio_terminate_threads(int group_id) { struct thread_data *td; pid_t pid = getpid(); int i; dprint(FD_PROCESS, "terminate group_id=%d\n", group_id); for_each_td(td, i) { if (group_id == TERMINATE_ALL || groupid == td->groupid) { dprint(FD_PROCESS, "setting terminate on %s/%d\n", td->o.name, (int) td->pid); if (td->terminate) continue; fio_mark_td_terminate(td); td->o.start_delay = 0; /* * if the thread is running, just let it exit */ if (!td->pid || pid == td->pid) continue; else if (td->runstate < TD_RAMP) kill(td->pid, SIGTERM); else { struct ioengine_ops *ops = td->io_ops; if (ops && ops->terminate) ops->terminate(td); } } } } int fio_running_or_pending_io_threads(void) { struct thread_data *td; int i; for_each_td(td, i) { if (td->flags & TD_F_NOIO) continue; if (td->runstate < TD_EXITED) return 1; } return 0; } int fio_set_fd_nonblocking(int fd, const char *who) { int flags; flags = fcntl(fd, F_GETFL); if (flags < 0) log_err("fio: %s failed to get file flags: %s\n", who, strerror(errno)); else { int new_flags = flags | O_NONBLOCK; new_flags = fcntl(fd, F_SETFL, new_flags); if (new_flags < 0) log_err("fio: %s failed to get file flags: %s\n", who, strerror(errno)); } return flags; } static int endian_check(void) { union { uint8_t c[8]; uint64_t v; } u; int le = 0, be = 0; u.v = 0x12; if (u.c[7] == 0x12) be = 1; else if (u.c[0] == 0x12) le = 1; #if defined(CONFIG_LITTLE_ENDIAN) if (be) return 1; #elif defined(CONFIG_BIG_ENDIAN) if (le) return 1; #else return 1; #endif if (!le && !be) return 1; return 0; } int initialize_fio(char *envp[]) { long ps; if (endian_check()) { log_err("fio: endianness settings appear wrong.\n"); log_err("fio: please report this to fio@vger.kernel.org\n"); return 1; } #if !defined(CONFIG_GETTIMEOFDAY) && !defined(CONFIG_CLOCK_GETTIME) #error "No available clock source!" #endif arch_init(envp); sinit(); if (fio_filelock_init()) { log_err("fio: failed initializing filelock subsys\n"); return 1; } /* * We need locale for number printing, if it isn't set then just * go with the US format. */ if (!getenv("LC_NUMERIC")) setlocale(LC_NUMERIC, "en_US"); ps = sysconf(_SC_PAGESIZE); if (ps < 0) { log_err("Failed to get page size\n"); return 1; } page_size = ps; page_mask = ps - 1; fio_keywords_init(); return 0; }