os/os.h

   1 #ifndef FIO_OS_H
   2 #define FIO_OS_H
   3
   4 #include <sys/types.h>
   5 #include <sys/socket.h>
   6 #include <fcntl.h>
   7 #include <pthread.h>
   8 #include <unistd.h>
   9 #include <stdlib.h>
  10
  11 #include "../arch/arch.h"
  12
  13 enum {
  14         os_linux = 1,
  15         os_aix,
  16         os_freebsd,
  17         os_hpux,
  18         os_mac,
  19         os_netbsd,
  20         os_openbsd,
  21         os_solaris,
  22         os_windows,
  23         os_android,
  24         os_dragonfly,
  25
  26         os_nr,
  27 };
  28
  29 #if defined(__ANDROID__)
  30 #include "os-android.h"
  31 #elif defined(__linux__)
  32 #include "os-linux.h"
  33 #elif defined(__FreeBSD__)
  34 #include "os-freebsd.h"
  35 #elif defined(__OpenBSD__)
  36 #include "os-openbsd.h"
  37 #elif defined(__NetBSD__)
  38 #include "os-netbsd.h"
  39 #elif defined(__sun__)
  40 #include "os-solaris.h"
  41 #elif defined(__APPLE__)
  42 #include "os-mac.h"
  43 #elif defined(_AIX)
  44 #include "os-aix.h"
  45 #elif defined(__hpux)
  46 #include "os-hpux.h"
  47 #elif defined(WIN32)
  48 #include "os-windows.h"
  49 #elif defined (__DragonFly__)
  50 #include "os-dragonfly.h"
  51 #else
  52 #error "unsupported os"
  53 #endif
  54
  55 #ifdef CONFIG_POSIXAIO
  56 #include <aio.h>
  57 #ifndef FIO_OS_HAVE_AIOCB_TYPEDEF
  58 typedef struct aiocb os_aiocb_t;
  59 #endif
  60 #endif
  61
  62 #ifdef FIO_HAVE_SGIO
  63 #include <linux/fs.h>
  64 #include <scsi/sg.h>
  65 #endif
  66
  67 #ifndef CONFIG_STRSEP
  68 #include "../oslib/strsep.h"
  69 #endif
  70
  71 #ifndef CONFIG_STRLCAT
  72 #include "../oslib/strlcat.h"
  73 #endif
  74
  75 #ifdef MSG_DONTWAIT
  76 #define OS_MSG_DONTWAIT MSG_DONTWAIT
  77 #endif
  78
  79 #ifndef POSIX_FADV_DONTNEED
  80 #define POSIX_FADV_DONTNEED     (0)
  81 #define POSIX_FADV_SEQUENTIAL   (0)
  82 #define POSIX_FADV_RANDOM       (0)
  83 #endif
  84
  85 #ifndef FIO_HAVE_CPU_AFFINITY
  86 #define fio_getaffinity(pid, mask)      do { } while (0)
  87 #define fio_cpu_clear(mask, cpu)        do { } while (0)
  88 typedef unsigned long os_cpu_mask_t;
  89
  90 static inline int fio_setaffinity(int pid, os_cpu_mask_t cpumask)
  91 {
  92         return 0;
  93 }
  94
  95 static inline int fio_cpuset_exit(os_cpu_mask_t *mask)
  96 {
  97         return -1;
  98 }
  99
 100 static inline int fio_cpus_split(os_cpu_mask_t *mask, unsigned int cpu_index)
 101 {
 102         return 0;
 103 }
 104 #else
 105 extern int fio_cpus_split(os_cpu_mask_t *mask, unsigned int cpu);
 106 #endif
 107
 108 #ifndef FIO_HAVE_IOPRIO
 109 #define ioprio_set(which, who, prioclass, prio) (0)
 110 #endif
 111
 112 #ifndef FIO_HAVE_ODIRECT
 113 #define OS_O_DIRECT                     0
 114 #else
 115 #define OS_O_DIRECT                     O_DIRECT
 116 #endif
 117
 118 #ifdef OS_O_ATOMIC
 119 #define FIO_O_ATOMIC                    OS_O_ATOMIC
 120 #else
 121 #define FIO_O_ATOMIC                    0
 122 #endif
 123
 124 #ifndef FIO_HAVE_HUGETLB
 125 #define SHM_HUGETLB                     0
 126 #define MAP_HUGETLB                     0
 127 #ifndef FIO_HUGE_PAGE
 128 #define FIO_HUGE_PAGE                   0
 129 #endif
 130 #else
 131 #ifndef FIO_HUGE_PAGE
 132 #define FIO_HUGE_PAGE                   4194304
 133 #endif
 134 #endif
 135
 136 #ifndef FIO_HAVE_MMAP_HUGE
 137 #define MAP_HUGETLB                     0
 138 #endif
 139
 140 #ifndef FIO_O_NOATIME
 141 #define FIO_O_NOATIME                   0
 142 #endif
 143
 144 #ifndef OS_RAND_MAX
 145 #define OS_RAND_MAX                     RAND_MAX
 146 #endif
 147
 148 #ifndef FIO_HAVE_RAWBIND
 149 #define fio_lookup_raw(dev, majdev, mindev)     1
 150 #endif
 151
 152 #ifndef FIO_PREFERRED_ENGINE
 153 #define FIO_PREFERRED_ENGINE    "sync"
 154 #endif
 155
 156 #ifndef FIO_OS_PATH_SEPARATOR
 157 #define FIO_OS_PATH_SEPARATOR   "/"
 158 #endif
 159
 160 #ifndef FIO_PREFERRED_CLOCK_SOURCE
 161 #ifdef CONFIG_CLOCK_GETTIME
 162 #define FIO_PREFERRED_CLOCK_SOURCE      CS_CGETTIME
 163 #else
 164 #define FIO_PREFERRED_CLOCK_SOURCE      CS_GTOD
 165 #endif
 166 #endif
 167
 168 #ifndef FIO_MAX_JOBS
 169 #define FIO_MAX_JOBS            2048
 170 #endif
 171
 172 #ifndef CONFIG_SOCKLEN_T
 173 typedef unsigned int socklen_t;
 174 #endif
 175
 176 #ifndef FIO_OS_HAS_CTIME_R
 177 #define os_ctime_r(x, y, z)     (void) ctime_r((x), (y))
 178 #endif
 179
 180 #ifdef FIO_USE_GENERIC_SWAP
 181 static inline uint16_t fio_swap16(uint16_t val)
 182 {
 183         return (val << 8) | (val >> 8);
 184 }
 185
 186 static inline uint32_t fio_swap32(uint32_t val)
 187 {
 188         val = ((val & 0xff00ff00UL) >> 8) | ((val & 0x00ff00ffUL) << 8);
 189
 190         return (val >> 16) | (val << 16);
 191 }
 192
 193 static inline uint64_t fio_swap64(uint64_t val)
 194 {
 195         val = ((val & 0xff00ff00ff00ff00ULL) >> 8) |
 196               ((val & 0x00ff00ff00ff00ffULL) << 8);
 197         val = ((val & 0xffff0000ffff0000ULL) >> 16) |
 198               ((val & 0x0000ffff0000ffffULL) << 16);
 199
 200         return (val >> 32) | (val << 32);
 201 }
 202 #endif
 203
 204 #ifndef FIO_HAVE_BYTEORDER_FUNCS
 205 #ifdef CONFIG_LITTLE_ENDIAN
 206 #define __le16_to_cpu(x)                (x)
 207 #define __le32_to_cpu(x)                (x)
 208 #define __le64_to_cpu(x)                (x)
 209 #define __cpu_to_le16(x)                (x)
 210 #define __cpu_to_le32(x)                (x)
 211 #define __cpu_to_le64(x)                (x)
 212 #else
 213 #define __le16_to_cpu(x)                fio_swap16(x)
 214 #define __le32_to_cpu(x)                fio_swap32(x)
 215 #define __le64_to_cpu(x)                fio_swap64(x)
 216 #define __cpu_to_le16(x)                fio_swap16(x)
 217 #define __cpu_to_le32(x)                fio_swap32(x)
 218 #define __cpu_to_le64(x)                fio_swap64(x)
 219 #endif
 220 #endif /* FIO_HAVE_BYTEORDER_FUNCS */
 221
 222 #ifdef FIO_INTERNAL
 223 #define le16_to_cpu(val) ({                     \
 224         typecheck(uint16_t, val);               \
 225         __le16_to_cpu(val);                     \
 226 })
 227 #define le32_to_cpu(val) ({                     \
 228         typecheck(uint32_t, val);               \
 229         __le32_to_cpu(val);                     \
 230 })
 231 #define le64_to_cpu(val) ({                     \
 232         typecheck(uint64_t, val);               \
 233         __le64_to_cpu(val);                     \
 234 })
 235 #endif
 236
 237 #define cpu_to_le16(val) ({                     \
 238         typecheck(uint16_t, val);               \
 239         __cpu_to_le16(val);                     \
 240 })
 241 #define cpu_to_le32(val) ({                     \
 242         typecheck(uint32_t, val);               \
 243         __cpu_to_le32(val);                     \
 244 })
 245 #define cpu_to_le64(val) ({                     \
 246         typecheck(uint64_t, val);               \
 247         __cpu_to_le64(val);                     \
 248 })
 249
 250 #ifndef FIO_HAVE_BLKTRACE
 251 static inline int is_blktrace(const char *fname, int *need_swap)
 252 {
 253         return 0;
 254 }
 255 struct thread_data;
 256 static inline int load_blktrace(struct thread_data *td, const char *fname,
 257                                 int need_swap)
 258 {
 259         return 1;
 260 }
 261 #endif
 262
 263 #define FIO_DEF_CL_SIZE         128
 264
 265 static inline int os_cache_line_size(void)
 266 {
 267 #ifdef FIO_HAVE_CL_SIZE
 268         int ret = arch_cache_line_size();
 269
 270         if (ret <= 0)
 271                 return FIO_DEF_CL_SIZE;
 272
 273         return ret;
 274 #else
 275         return FIO_DEF_CL_SIZE;
 276 #endif
 277 }
 278
 279 #ifdef FIO_USE_GENERIC_BDEV_SIZE
 280 static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)
 281 {
 282         off_t end;
 283
 284         *bytes = 0;
 285
 286         end = lseek(f->fd, 0, SEEK_END);
 287         if (end < 0)
 288                 return errno;
 289
 290         *bytes = end;
 291         return 0;
 292 }
 293 #endif
 294
 295 #ifdef FIO_USE_GENERIC_RAND
 296 typedef unsigned int os_random_state_t;
 297
 298 static inline void os_random_seed(unsigned long seed, os_random_state_t *rs)
 299 {
 300         srand(seed);
 301 }
 302
 303 static inline long os_random_long(os_random_state_t *rs)
 304 {
 305         long val;
 306
 307         val = rand_r(rs);
 308         return val;
 309 }
 310 #endif
 311
 312 #ifdef FIO_USE_GENERIC_INIT_RANDOM_STATE
 313 extern void td_fill_rand_seeds(struct thread_data *td);
 314 /*
 315  * Initialize the various random states we need (random io, block size ranges,
 316  * read/write mix, etc).
 317  */
 318 static inline int init_random_state(struct thread_data *td, unsigned long *rand_seeds, int size)
 319 {
 320         int fd;
 321
 322         fd = open("/dev/urandom", O_RDONLY);
 323         if (fd == -1) {
 324                 return 1;
 325         }
 326
 327         if (read(fd, rand_seeds, size) < size) {
 328                 close(fd);
 329                 return 1;
 330         }
 331
 332         close(fd);
 333         td_fill_rand_seeds(td);
 334         return 0;
 335 }
 336 #endif
 337
 338 #ifndef FIO_HAVE_FS_STAT
 339 static inline unsigned long long get_fs_free_size(const char *path)
 340 {
 341         return 0;
 342 }
 343 #endif
 344
 345 #ifndef FIO_HAVE_CPU_ONLINE_SYSCONF
 346 static inline unsigned int cpus_online(void)
 347 {
 348         return sysconf(_SC_NPROCESSORS_ONLN);
 349 }
 350 #endif
 351
 352 #ifndef CPU_COUNT
 353 #ifdef FIO_HAVE_CPU_AFFINITY
 354 static inline int CPU_COUNT(os_cpu_mask_t *mask)
 355 {
 356         int max_cpus = cpus_online();
 357         int nr_cpus, i;
 358
 359         for (i = 0, nr_cpus = 0; i < max_cpus; i++)
 360                 if (fio_cpu_isset(mask, i))
 361                         nr_cpus++;
 362
 363         return nr_cpus;
 364 }
 365 #endif
 366 #endif
 367
 368 #ifndef FIO_HAVE_GETTID
 369 static inline int gettid(void)
 370 {
 371         return getpid();
 372 }
 373 #endif
 374
 375 #endif