[fio.git] / os / os.h

#ifndef FIO_OS_H
#define FIO_OS_H

#include <sys/types.h>
#include <sys/socket.h>
#include <fcntl.h>
#include <pthread.h>
#include <unistd.h>
#include <stdlib.h>

#include "../arch/arch.h"
#include "../lib/types.h"

enum {
	os_linux = 1,
	os_aix,
	os_freebsd,
	os_hpux,
	os_mac,
	os_netbsd,
	os_openbsd,
	os_solaris,
	os_windows,
	os_android,
	os_dragonfly,

	os_nr,
};

#if defined(__ANDROID__)
#include "os-android.h"
#elif defined(__linux__)
#include "os-linux.h"
#elif defined(__FreeBSD__)
#include "os-freebsd.h"
#elif defined(__OpenBSD__)
#include "os-openbsd.h"
#elif defined(__NetBSD__)
#include "os-netbsd.h"
#elif defined(__sun__)
#include "os-solaris.h"
#elif defined(__APPLE__)
#include "os-mac.h"
#elif defined(_AIX)
#include "os-aix.h"
#elif defined(__hpux)
#include "os-hpux.h"
#elif defined(WIN32)
#include "os-windows.h"
#elif defined (__DragonFly__)
#include "os-dragonfly.h"
#else
#error "unsupported os"
#endif

#ifdef CONFIG_POSIXAIO
#include <aio.h>
#ifndef FIO_OS_HAVE_AIOCB_TYPEDEF
typedef struct aiocb os_aiocb_t;
#endif
#endif

#ifdef FIO_HAVE_SGIO
#include <linux/fs.h>
#include <scsi/sg.h>
#endif

#ifndef CONFIG_STRSEP
#include "../oslib/strsep.h"
#endif

#ifndef CONFIG_STRLCAT
#include "../oslib/strlcat.h"
#endif

#ifdef MSG_DONTWAIT
#define OS_MSG_DONTWAIT	MSG_DONTWAIT
#endif

#ifndef POSIX_FADV_DONTNEED
#define POSIX_FADV_DONTNEED	(0)
#define POSIX_FADV_SEQUENTIAL	(0)
#define POSIX_FADV_RANDOM	(0)
#endif

#ifndef FIO_HAVE_CPU_AFFINITY
#define fio_cpu_clear(mask, cpu)	do { } while (0)
typedef unsigned long os_cpu_mask_t;

static inline int fio_setaffinity(int pid, os_cpu_mask_t cpumask)
{
	return 0;
}

static inline int fio_getaffinity(int pid, os_cpu_mask_t *cpumask)
{
	return -1;
}

static inline int fio_cpuset_exit(os_cpu_mask_t *mask)
{
	return -1;
}

static inline int fio_cpus_split(os_cpu_mask_t *mask, unsigned int cpu_index)
{
	return 0;
}
#else
extern int fio_cpus_split(os_cpu_mask_t *mask, unsigned int cpu);
#endif

#ifndef FIO_HAVE_IOPRIO
#define ioprio_set(which, who, prioclass, prio)	(0)
#endif

#ifndef FIO_HAVE_ODIRECT
#define OS_O_DIRECT			0
#else
#define OS_O_DIRECT			O_DIRECT
#endif

#ifdef OS_O_ATOMIC
#define FIO_O_ATOMIC			OS_O_ATOMIC
#else
#define FIO_O_ATOMIC			0
#endif

#ifndef FIO_HAVE_HUGETLB
#define SHM_HUGETLB			0
#define MAP_HUGETLB			0
#ifndef FIO_HUGE_PAGE
#define FIO_HUGE_PAGE			0
#endif
#else
#ifndef FIO_HUGE_PAGE
#define FIO_HUGE_PAGE			4194304
#endif
#endif

#ifndef FIO_HAVE_MMAP_HUGE
#define MAP_HUGETLB			0
#endif

#ifndef FIO_O_NOATIME
#define FIO_O_NOATIME			0
#endif

#ifndef OS_RAND_MAX
#define OS_RAND_MAX			RAND_MAX
#endif

#ifndef FIO_HAVE_RAWBIND
#define fio_lookup_raw(dev, majdev, mindev)	1
#endif

#ifndef FIO_PREFERRED_ENGINE
#define FIO_PREFERRED_ENGINE	"psync"
#endif

#ifndef FIO_OS_PATH_SEPARATOR
#define FIO_OS_PATH_SEPARATOR	"/"
#endif

#ifndef FIO_PREFERRED_CLOCK_SOURCE
#ifdef CONFIG_CLOCK_GETTIME
#define FIO_PREFERRED_CLOCK_SOURCE	CS_CGETTIME
#else
#define FIO_PREFERRED_CLOCK_SOURCE	CS_GTOD
#endif
#endif

#ifndef FIO_MAX_JOBS
#define FIO_MAX_JOBS		4096
#endif

#ifndef CONFIG_SOCKLEN_T
typedef unsigned int socklen_t;
#endif

#ifndef FIO_OS_HAS_CTIME_R
#define os_ctime_r(x, y, z)     (void) ctime_r((x), (y))
#endif

#ifdef FIO_USE_GENERIC_SWAP
static inline uint16_t fio_swap16(uint16_t val)
{
	return (val << 8) | (val >> 8);
}

static inline uint32_t fio_swap32(uint32_t val)
{
	val = ((val & 0xff00ff00UL) >> 8) | ((val & 0x00ff00ffUL) << 8);

	return (val >> 16) | (val << 16);
}

static inline uint64_t fio_swap64(uint64_t val)
{
	val = ((val & 0xff00ff00ff00ff00ULL) >> 8) |
	      ((val & 0x00ff00ff00ff00ffULL) << 8);
	val = ((val & 0xffff0000ffff0000ULL) >> 16) |
	      ((val & 0x0000ffff0000ffffULL) << 16);

	return (val >> 32) | (val << 32);
}
#endif

#ifndef FIO_HAVE_BYTEORDER_FUNCS
#ifdef CONFIG_LITTLE_ENDIAN
#define __le16_to_cpu(x)		(x)
#define __le32_to_cpu(x)		(x)
#define __le64_to_cpu(x)		(x)
#define __cpu_to_le16(x)		(x)
#define __cpu_to_le32(x)		(x)
#define __cpu_to_le64(x)		(x)
#else
#define __le16_to_cpu(x)		fio_swap16(x)
#define __le32_to_cpu(x)		fio_swap32(x)
#define __le64_to_cpu(x)		fio_swap64(x)
#define __cpu_to_le16(x)		fio_swap16(x)
#define __cpu_to_le32(x)		fio_swap32(x)
#define __cpu_to_le64(x)		fio_swap64(x)
#endif
#endif /* FIO_HAVE_BYTEORDER_FUNCS */

#ifdef FIO_INTERNAL
#define le16_to_cpu(val) ({			\
	typecheck(uint16_t, val);		\
	__le16_to_cpu(val);			\
})
#define le32_to_cpu(val) ({			\
	typecheck(uint32_t, val);		\
	__le32_to_cpu(val);			\
})
#define le64_to_cpu(val) ({			\
	typecheck(uint64_t, val);		\
	__le64_to_cpu(val);			\
})
#endif

#define cpu_to_le16(val) ({			\
	typecheck(uint16_t, val);		\
	__cpu_to_le16(val);			\
})
#define cpu_to_le32(val) ({			\
	typecheck(uint32_t, val);		\
	__cpu_to_le32(val);			\
})
#define cpu_to_le64(val) ({			\
	typecheck(uint64_t, val);		\
	__cpu_to_le64(val);			\
})

#ifndef FIO_HAVE_BLKTRACE
static inline int is_blktrace(const char *fname, int *need_swap)
{
	return 0;
}
struct thread_data;
static inline int load_blktrace(struct thread_data *td, const char *fname,
				int need_swap)
{
	return 1;
}
#endif

#define FIO_DEF_CL_SIZE		128

static inline int os_cache_line_size(void)
{
#ifdef FIO_HAVE_CL_SIZE
	int ret = arch_cache_line_size();

	if (ret <= 0)
		return FIO_DEF_CL_SIZE;

	return ret;
#else
	return FIO_DEF_CL_SIZE;
#endif
}

#ifdef FIO_USE_GENERIC_BDEV_SIZE
static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)
{
	off_t end;

	*bytes = 0;

	end = lseek(f->fd, 0, SEEK_END);
	if (end < 0)
		return errno;

	*bytes = end;
	return 0;
}
#endif

#ifdef FIO_USE_GENERIC_RAND
typedef unsigned int os_random_state_t;

static inline void os_random_seed(unsigned long seed, os_random_state_t *rs)
{
	srand(seed);
}

static inline long os_random_long(os_random_state_t *rs)
{
	long val;

	val = rand_r(rs);
	return val;
}
#endif

#ifdef FIO_USE_GENERIC_INIT_RANDOM_STATE
extern void td_fill_rand_seeds(struct thread_data *td);
/*
 * Initialize the various random states we need (random io, block size ranges,
 * read/write mix, etc).
 */
static inline int init_random_state(struct thread_data *td, unsigned long *rand_seeds, int size)
{
	int fd;

	fd = open("/dev/urandom", O_RDONLY);
	if (fd == -1) {
		return 1;
	}

	if (read(fd, rand_seeds, size) < size) {
		close(fd);
		return 1;
	}

	close(fd);
	td_fill_rand_seeds(td);
	return 0;
}
#endif

#ifndef FIO_HAVE_FS_STAT
static inline unsigned long long get_fs_free_size(const char *path)
{
	return 0;
}
#endif

#ifdef __powerpc64__
#define FIO_HAVE_CPU_ONLINE_SYSCONF
static inline unsigned int cpus_online(void)
{
        return sysconf(_SC_NPROCESSORS_CONF);
}
#endif

#ifndef FIO_HAVE_CPU_ONLINE_SYSCONF
static inline unsigned int cpus_online(void)
{
	return sysconf(_SC_NPROCESSORS_ONLN);
}
#endif

#ifndef CPU_COUNT
#ifdef FIO_HAVE_CPU_AFFINITY
static inline int CPU_COUNT(os_cpu_mask_t *mask)
{
	int max_cpus = cpus_online();
	int nr_cpus, i;

	for (i = 0, nr_cpus = 0; i < max_cpus; i++)
		if (fio_cpu_isset(mask, i))
			nr_cpus++;

	return nr_cpus;
}
#endif
#endif

#ifndef FIO_HAVE_GETTID
static inline int gettid(void)
{
	return getpid();
}
#endif

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