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