fio.c

   1 /*
   2  * fio - the flexible io tester
   3  *
   4  * Copyright (C) 2005 Jens Axboe <axboe@suse.de>
   5  * Copyright (C) 2006-2012 Jens Axboe <axboe@kernel.dk>
   6  *
   7  * The license below covers all files distributed with fio unless otherwise
   8  * noted in the file itself.
   9  *
  10  *  This program is free software; you can redistribute it and/or modify
  11  *  it under the terms of the GNU General Public License version 2 as
  12  *  published by the Free Software Foundation.
  13  *
  14  *  This program is distributed in the hope that it will be useful,
  15  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  *  GNU General Public License for more details.
  18  *
  19  *  You should have received a copy of the GNU General Public License
  20  *  along with this program; if not, write to the Free Software
  21  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  22  *
  23  */
  24 #include <unistd.h>
  25 #include <fcntl.h>
  26 #include <string.h>
  27 #include <limits.h>
  28 #include <signal.h>
  29 #include <time.h>
  30 #include <locale.h>
  31 #include <assert.h>
  32 #include <time.h>
  33 #include <sys/stat.h>
  34 #include <sys/wait.h>
  35 #include <sys/ipc.h>
  36 #include <sys/shm.h>
  37 #include <sys/mman.h>
  38
  39 #include "fio.h"
  40 #include "hash.h"
  41 #include "smalloc.h"
  42 #include "verify.h"
  43 #include "trim.h"
  44 #include "diskutil.h"
  45 #include "cgroup.h"
  46 #include "profile.h"
  47 #include "lib/rand.h"
  48 #include "memalign.h"
  49 #include "server.h"
  50
  51 unsigned long page_mask;
  52 unsigned long page_size;
  53
  54 int groupid = 0;
  55 unsigned int thread_number = 0;
  56 unsigned int nr_process = 0;
  57 unsigned int nr_thread = 0;
  58 int shm_id = 0;
  59 int temp_stall_ts;
  60 unsigned long done_secs = 0;
  61
  62 /*
  63  * Just expose an empty list, if the OS does not support disk util stats
  64  */
  65 #ifndef FIO_HAVE_DISK_UTIL
  66 FLIST_HEAD(disk_list);
  67 #endif
  68
  69 unsigned long arch_flags = 0;
  70
  71 static int endian_check(void)
  72 {
  73         union {
  74                 uint8_t c[8];
  75                 uint64_t v;
  76         } u;
  77         int le = 0, be = 0;
  78
  79         u.v = 0x12;
  80         if (u.c[7] == 0x12)
  81                 be = 1;
  82         else if (u.c[0] == 0x12)
  83                 le = 1;
  84
  85 #if defined(FIO_LITTLE_ENDIAN)
  86         if (be)
  87                 return 1;
  88 #elif defined(FIO_BIG_ENDIAN)
  89         if (le)
  90                 return 1;
  91 #else
  92         return 1;
  93 #endif
  94
  95         if (!le && !be)
  96                 return 1;
  97
  98         return 0;
  99 }
 100
 101 int main(int argc, char *argv[], char *envp[])
 102 {
 103         long ps;
 104
 105         if (endian_check()) {
 106                 log_err("fio: endianness settings appear wrong.\n");
 107                 log_err("fio: please report this to fio@vger.kernel.org\n");
 108                 return 1;
 109         }
 110
 111         arch_init(envp);
 112
 113         sinit();
 114
 115         /*
 116          * We need locale for number printing, if it isn't set then just
 117          * go with the US format.
 118          */
 119         if (!getenv("LC_NUMERIC"))
 120                 setlocale(LC_NUMERIC, "en_US");
 121
 122         ps = sysconf(_SC_PAGESIZE);
 123         if (ps < 0) {
 124                 log_err("Failed to get page size\n");
 125                 return 1;
 126         }
 127
 128         page_size = ps;
 129         page_mask = ps - 1;
 130
 131         fio_keywords_init();
 132
 133         if (parse_options(argc, argv))
 134                 return 1;
 135
 136         if (nr_clients)
 137                 return fio_handle_clients();
 138         else
 139                 return fio_backend();
 140 }