blob: 5ed8c6073603d4edb0b381fb356694cce600e04b [file] [log] [blame]
/*
* fio - the flexible io tester
*
* Copyright (C) 2005 Jens Axboe <axboe@suse.de>
* Copyright (C) 2006-2012 Jens Axboe <axboe@kernel.dk>
*
* 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 <string.h>
#include <sys/types.h>
#include <signal.h>
#include <stdint.h>
#include <locale.h>
#include <fcntl.h>
#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_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);
td->terminate = 1;
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;
}
void 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 {
flags |= O_NONBLOCK;
flags = fcntl(fd, F_SETFL, flags);
if (flags < 0)
log_err("fio: %s failed to get file flags: %s\n", who, strerror(errno));
}
}
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;
}