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