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