Token-based flow control
[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 <pthread.h>
7 #include <unistd.h>
8 #include <stdlib.h>
9
10 enum {
11         os_linux = 1,
12         os_aix,
13         os_freebsd,
14         os_hpux,
15         os_mac,
16         os_netbsd,
17         os_solaris,
18         os_windows,
19
20         os_nr,
21 };
22
23 #if defined(__linux__)
24 #include "os-linux.h"
25 #elif defined(__FreeBSD__)
26 #include "os-freebsd.h"
27 #elif defined(__NetBSD__)
28 #include "os-netbsd.h"
29 #elif defined(__sun__)
30 #include "os-solaris.h"
31 #elif defined(__APPLE__)
32 #include "os-mac.h"
33 #elif defined(_AIX)
34 #include "os-aix.h"
35 #elif defined(__hpux)
36 #include "os-hpux.h"
37 #elif defined(__CYGWIN__)
38 #include "os-windows.h"
39 #else
40 #error "unsupported os"
41 #endif
42
43 #ifdef FIO_HAVE_LIBAIO
44 #include <libaio.h>
45 #endif
46
47 #ifdef FIO_HAVE_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 FIO_HAVE_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 FIO_HAVE_FADVISE
68 #define posix_fadvise(fd, off, len, advice)     (0)
69
70 #ifndef POSIX_FADV_DONTNEED
71 #define POSIX_FADV_DONTNEED     (0)
72 #define POSIX_FADV_SEQUENTIAL   (0)
73 #define POSIX_FADV_RANDOM       (0)
74 #endif
75 #endif /* FIO_HAVE_FADVISE */
76
77 #ifndef FIO_HAVE_CPU_AFFINITY
78 #define fio_setaffinity(pid, mask)      (0)
79 #define fio_getaffinity(pid, mask)      do { } while (0)
80 #define fio_cpu_clear(mask, cpu)        do { } while (0)
81 #define fio_cpuset_exit(mask)           (-1)
82 typedef unsigned long os_cpu_mask_t;
83 #endif
84
85 #ifndef FIO_HAVE_IOPRIO
86 #define ioprio_set(which, who, prio)    (0)
87 #endif
88
89 #ifndef FIO_HAVE_ODIRECT
90 #define OS_O_DIRECT                     0
91 #else
92 #define OS_O_DIRECT                     O_DIRECT
93 #endif
94
95 #ifndef FIO_HAVE_HUGETLB
96 #define SHM_HUGETLB                     0
97 #ifndef FIO_HUGE_PAGE
98 #define FIO_HUGE_PAGE                   0
99 #endif
100 #else
101 #ifndef FIO_HUGE_PAGE
102 #define FIO_HUGE_PAGE                   4194304
103 #endif
104 #endif
105
106 #ifndef FIO_O_NOATIME
107 #define FIO_O_NOATIME                   0
108 #endif
109
110 #ifndef OS_RAND_MAX
111 #define OS_RAND_MAX                     RAND_MAX
112 #endif
113
114 #ifdef FIO_HAVE_CLOCK_MONOTONIC
115 #define FIO_TIMER_CLOCK CLOCK_MONOTONIC
116 #else
117 #define FIO_TIMER_CLOCK CLOCK_REALTIME
118 #endif
119
120 #ifndef FIO_HAVE_RAWBIND
121 #define fio_lookup_raw(dev, majdev, mindev)     1
122 #endif
123
124 #ifndef FIO_PREFERRED_ENGINE
125 #define FIO_PREFERRED_ENGINE    "sync"
126 #endif
127
128 #ifndef FIO_MAX_JOBS
129 #define FIO_MAX_JOBS            2048
130 #endif
131
132 #ifndef FIO_OS_HAVE_SOCKLEN_T
133 typedef socklen_t fio_socklen_t;
134 #endif
135
136 #ifdef FIO_USE_GENERIC_SWAP
137 static inline uint16_t fio_swap16(uint16_t val)
138 {
139         return (val << 8) | (val >> 8);
140 }
141
142 static inline uint32_t fio_swap32(uint32_t val)
143 {
144         val = ((val & 0xff00ff00UL) >> 8) | ((val & 0x00ff00ffUL) << 8);
145
146         return (val >> 16) | (val << 16);
147 }
148
149 static inline uint64_t fio_swap64(uint64_t val)
150 {
151         val = ((val & 0xff00ff00ff00ff00ULL) >> 8) |
152               ((val & 0x00ff00ff00ff00ffULL) << 8);
153         val = ((val & 0xffff0000ffff0000ULL) >> 16) |
154               ((val & 0x0000ffff0000ffffULL) << 16);
155
156         return (val >> 32) | (val << 32);
157 }
158 #endif
159
160 #ifdef FIO_LITTLE_ENDIAN
161 #define __le16_to_cpu(x)                (x)
162 #define __le32_to_cpu(x)                (x)
163 #define __le64_to_cpu(x)                (x)
164 #define __cpu_to_le16(x)                (x)
165 #define __cpu_to_le32(x)                (x)
166 #define __cpu_to_le64(x)                (x)
167 #else
168 #define __le16_to_cpu(x)                fio_swap16(x)
169 #define __le32_to_cpu(x)                fio_swap32(x)
170 #define __le64_to_cpu(x)                fio_swap64(x)
171 #define __cpu_to_le16(x)                fio_swap16(x)
172 #define __cpu_to_le32(x)                fio_swap32(x)
173 #define __cpu_to_le64(x)                fio_swap64(x)
174 #endif
175
176 #define le16_to_cpu(val) ({                     \
177         uint16_t *__val = &(val);               \
178         __le16_to_cpu(*__val);                  \
179 })
180 #define le32_to_cpu(val) ({                     \
181         uint32_t *__val = &(val);               \
182         __le32_to_cpu(*__val);                  \
183 })
184 #define le64_to_cpu(val) ({                     \
185         uint64_t *__val = &(val);               \
186         __le64_to_cpu(*__val);                  \
187 })
188 #define cpu_to_le16(val) ({                     \
189         uint16_t *__val = &(val);               \
190         __cpu_to_le16(*__val);                  \
191 })
192 #define cpu_to_le32(val) ({                     \
193         uint32_t *__val = &(val);               \
194         __cpu_to_le32(*__val);                  \
195 })
196 #define cpu_to_le64(val) ({                     \
197         uint64_t *__val = &(val);               \
198         __cpu_to_le64(*__val);                  \
199 })
200
201 #ifndef FIO_HAVE_BLKTRACE
202 static inline int is_blktrace(const char *fname)
203 {
204         return 0;
205 }
206 struct thread_data;
207 static inline int load_blktrace(struct thread_data *td, const char *fname)
208 {
209         return 1;
210 }
211 #endif
212
213 #define FIO_DEF_CL_SIZE         128
214
215 static inline int os_cache_line_size(void)
216 {
217 #ifdef FIO_HAVE_CL_SIZE
218         int ret = arch_cache_line_size();
219
220         if (ret <= 0)
221                 return FIO_DEF_CL_SIZE;
222
223         return ret;
224 #else
225         return FIO_DEF_CL_SIZE;
226 #endif
227 }
228
229 #ifdef FIO_USE_GENERIC_BDEV_SIZE
230 static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)
231 {
232         off_t end;
233
234         *bytes = 0;
235
236         end = lseek(f->fd, 0, SEEK_END);
237         if (end < 0)
238                 return errno;
239
240         *bytes = end;
241         return 0;
242 }
243 #endif
244
245 #ifdef FIO_USE_GENERIC_RAND
246 typedef unsigned int os_random_state_t;
247
248 static inline void os_random_seed(unsigned long seed, os_random_state_t *rs)
249 {
250         srand(seed);
251 }
252
253 static inline long os_random_long(os_random_state_t *rs)
254 {
255         long val;
256
257         val = rand_r(rs);
258         return val;
259 }
260 #endif
261
262 #ifndef FIO_HAVE_FS_STAT
263 static inline unsigned long long get_fs_size(const char *path)
264 {
265         return 0;
266 }
267 #endif
268
269 #ifndef FIO_HAVE_CPU_ONLINE_SYSCONF
270 static inline unsigned int cpus_online(void)
271 {
272         return sysconf(_SC_NPROCESSORS_ONLN);
273 }
274 #endif
275
276 #ifndef FIO_HAVE_GETTID
277 static inline int gettid(void)
278 {
279         return getpid();
280 }
281 #endif
282
283 #endif