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