Add ioprio_set() support for DragonFlyBSD
[fio.git] / os / os-dragonfly.h
1 #ifndef FIO_OS_DRAGONFLY_H
2 #define FIO_OS_DRAGONFLY_H
3
4 #define FIO_OS  os_dragonfly
5
6 #include <errno.h>
7 #include <unistd.h>
8 #include <sys/param.h>
9 #include <sys/sysctl.h>
10 #include <sys/statvfs.h>
11 #include <sys/diskslice.h>
12 #include <sys/ioctl_compat.h>
13 #include <sys/usched.h>
14 #include <sys/resource.h>
15
16 #include "../file.h"
17
18 #define FIO_HAVE_ODIRECT
19 #define FIO_USE_GENERIC_RAND
20 #define FIO_USE_GENERIC_INIT_RANDOM_STATE
21 #define FIO_HAVE_FS_STAT
22 #define FIO_HAVE_TRIM
23 #define FIO_HAVE_CHARDEV_SIZE
24 #define FIO_HAVE_GETTID
25 #define FIO_HAVE_CPU_AFFINITY
26 #define FIO_HAVE_IOPRIO
27
28 #define OS_MAP_ANON             MAP_ANON
29
30 #ifndef PTHREAD_STACK_MIN
31 #define PTHREAD_STACK_MIN 4096
32 #endif
33
34 #define fio_swap16(x)   bswap16(x)
35 #define fio_swap32(x)   bswap32(x)
36 #define fio_swap64(x)   bswap64(x)
37
38 /* This is supposed to equal (sizeof(cpumask_t)*8) */
39 #define FIO_MAX_CPUS    SMP_MAXCPU
40
41 typedef off_t off64_t;
42 typedef cpumask_t os_cpu_mask_t;
43
44 /*
45  * These macros are copied from sys/cpu/x86_64/include/types.h.
46  * It's okay to copy from arch dependent header because x86_64 is the only
47  * supported arch, and no other arch is going to be supported any time soon.
48  *
49  * These are supposed to be able to be included from userspace by defining
50  * _KERNEL_STRUCTURES, however this scheme is badly broken that enabling it
51  * causes compile-time conflicts with other headers. Although the current
52  * upstream code no longer requires _KERNEL_STRUCTURES, they should be kept
53  * here for compatibility with older versions.
54  */
55 #ifndef CPUMASK_SIMPLE
56 #define CPUMASK_SIMPLE(cpu)             ((uint64_t)1 << (cpu))
57 #define CPUMASK_TESTBIT(val, i)         ((val).ary[((i) >> 6) & 3] & \
58                                          CPUMASK_SIMPLE((i) & 63))
59 #define CPUMASK_ORBIT(mask, i)          ((mask).ary[((i) >> 6) & 3] |= \
60                                          CPUMASK_SIMPLE((i) & 63))
61 #define CPUMASK_NANDBIT(mask, i)        ((mask).ary[((i) >> 6) & 3] &= \
62                                          ~CPUMASK_SIMPLE((i) & 63))
63 #define CPUMASK_ASSZERO(mask)           do {                            \
64                                         (mask).ary[0] = 0;              \
65                                         (mask).ary[1] = 0;              \
66                                         (mask).ary[2] = 0;              \
67                                         (mask).ary[3] = 0;              \
68                                         } while(0)
69 #endif
70
71 /*
72  * Define USCHED_GET_CPUMASK as the macro didn't exist until release 4.5.
73  * usched_set(2) returns EINVAL if the kernel doesn't support it, though
74  * fio_getaffinity() returns void.
75  *
76  * Also note usched_set(2) works only for the current thread regardless of
77  * the command type. It doesn't work against another thread regardless of
78  * a caller's privilege. A caller would generally specify 0 for pid for the
79  * current thread though that's the only choice. See BUGS in usched_set(2).
80  */
81 #ifndef USCHED_GET_CPUMASK
82 #define USCHED_GET_CPUMASK      5
83 #endif
84
85 static inline int fio_cpuset_init(os_cpu_mask_t *mask)
86 {
87         CPUMASK_ASSZERO(*mask);
88         return 0;
89 }
90
91 static inline int fio_cpuset_exit(os_cpu_mask_t *mask)
92 {
93         return 0;
94 }
95
96 static inline void fio_cpu_clear(os_cpu_mask_t *mask, int cpu)
97 {
98         CPUMASK_NANDBIT(*mask, cpu);
99 }
100
101 static inline void fio_cpu_set(os_cpu_mask_t *mask, int cpu)
102 {
103         CPUMASK_ORBIT(*mask, cpu);
104 }
105
106 static inline int fio_cpu_isset(os_cpu_mask_t *mask, int cpu)
107 {
108         if (CPUMASK_TESTBIT(*mask, cpu))
109                 return 1;
110
111         return 0;
112 }
113
114 static inline int fio_cpu_count(os_cpu_mask_t *mask)
115 {
116         int i, n = 0;
117
118         for (i = 0; i < FIO_MAX_CPUS; i++)
119                 if (CPUMASK_TESTBIT(*mask, i))
120                         n++;
121
122         return n;
123 }
124
125 static inline int fio_setaffinity(int pid, os_cpu_mask_t mask)
126 {
127         int i, firstcall = 1;
128
129         /* 0 for the current thread, see BUGS in usched_set(2) */
130         pid = 0;
131
132         for (i = 0; i < FIO_MAX_CPUS; i++) {
133                 if (!CPUMASK_TESTBIT(mask, i))
134                         continue;
135                 if (firstcall) {
136                         if (usched_set(pid, USCHED_SET_CPU, &i, sizeof(int)))
137                                 return -1;
138                         firstcall = 0;
139                 } else {
140                         if (usched_set(pid, USCHED_ADD_CPU, &i, sizeof(int)))
141                                 return -1;
142                 }
143         }
144
145         return 0;
146 }
147
148 static inline void fio_getaffinity(int pid, os_cpu_mask_t *mask)
149 {
150         /* 0 for the current thread, see BUGS in usched_set(2) */
151         pid = 0;
152
153         usched_set(pid, USCHED_GET_CPUMASK, mask, sizeof(*mask));
154 }
155
156 /* fio code is Linux based, so rename macros to Linux style */
157 #define IOPRIO_WHO_PROCESS      PRIO_PROCESS
158 #define IOPRIO_WHO_PGRP         PRIO_PGRP
159 #define IOPRIO_WHO_USER         PRIO_USER
160
161 #define IOPRIO_MIN_PRIO         1       /* lowest priority */
162 #define IOPRIO_MAX_PRIO         10      /* highest priority */
163
164 /*
165  * Prototypes declared in sys/sys/resource.h are preventing from defining
166  * ioprio_set() with 4 arguments, so define fio's ioprio_set() as a macro.
167  * Note that there is no idea of class within ioprio_set(2) unlike Linux.
168  */
169 #define ioprio_set(which, who, ioprio_class, ioprio)    \
170         ioprio_set(which, who, ioprio)
171
172 static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)
173 {
174         struct partinfo pi;
175
176         if (!ioctl(f->fd, DIOCGPART, &pi)) {
177                 *bytes = (unsigned long long) pi.media_size;
178                 return 0;
179         }
180
181         *bytes = 0;
182         return errno;
183 }
184
185 static inline int chardev_size(struct fio_file *f, unsigned long long *bytes)
186 {
187         return blockdev_size(f, bytes);
188 }
189
190 static inline int blockdev_invalidate_cache(struct fio_file *f)
191 {
192         return EINVAL;
193 }
194
195 static inline unsigned long long os_phys_mem(void)
196 {
197         int mib[2] = { CTL_HW, HW_PHYSMEM };
198         uint64_t mem;
199         size_t len = sizeof(mem);
200
201         sysctl(mib, 2, &mem, &len, NULL, 0);
202         return mem;
203 }
204
205 static inline int gettid(void)
206 {
207         return (int) lwp_gettid();
208 }
209
210 static inline unsigned long long get_fs_free_size(const char *path)
211 {
212         unsigned long long ret;
213         struct statvfs s;
214
215         if (statvfs(path, &s) < 0)
216                 return -1ULL;
217
218         ret = s.f_frsize;
219         ret *= (unsigned long long) s.f_bfree;
220         return ret;
221 }
222
223 static inline int os_trim(int fd, unsigned long long start,
224                           unsigned long long len)
225 {
226         off_t range[2];
227
228         range[0] = start;
229         range[1] = len;
230
231         if (!ioctl(fd, IOCTLTRIM, range))
232                 return 0;
233
234         return errno;
235 }
236
237 #ifdef MADV_FREE
238 #define FIO_MADV_FREE   MADV_FREE
239 #endif
240
241 #endif