Merge branch 'patch-1' of https://github.com/avlasov-mos-de/fio
[fio.git] / os / os-linux.h
... / ...
CommitLineData
1#ifndef FIO_OS_LINUX_H
2#define FIO_OS_LINUX_H
3
4#define FIO_OS os_linux
5
6#include <sys/ioctl.h>
7#include <sys/uio.h>
8#include <sys/syscall.h>
9#include <sys/sysmacros.h>
10#include <sys/vfs.h>
11#include <sys/mman.h>
12#include <unistd.h>
13#include <fcntl.h>
14#include <errno.h>
15#include <sched.h>
16#include <linux/unistd.h>
17#include <linux/raw.h>
18#include <linux/major.h>
19#include <linux/fs.h>
20#include <scsi/sg.h>
21
22#ifdef ARCH_HAVE_CRC_CRYPTO
23#include <sys/auxv.h>
24#ifndef HWCAP_CRC32
25#define HWCAP_CRC32 (1 << 7)
26#endif /* HWCAP_CRC32 */
27#endif /* ARCH_HAVE_CRC_CRYPTO */
28
29#include "./os-linux-syscall.h"
30#include "../file.h"
31
32#ifndef __has_builtin // Optional of course.
33 #define __has_builtin(x) 0 // Compatibility with non-clang compilers.
34#endif
35
36#define FIO_HAVE_CPU_AFFINITY
37#define FIO_HAVE_DISK_UTIL
38#define FIO_HAVE_SGIO
39#define FIO_HAVE_IOPRIO
40#define FIO_HAVE_IOPRIO_CLASS
41#define FIO_HAVE_IOSCHED_SWITCH
42#define FIO_HAVE_ODIRECT
43#define FIO_HAVE_HUGETLB
44#define FIO_HAVE_RAWBIND
45#define FIO_HAVE_BLKTRACE
46#define FIO_HAVE_CL_SIZE
47#define FIO_HAVE_CGROUPS
48#define FIO_HAVE_FS_STAT
49#define FIO_HAVE_TRIM
50#define FIO_HAVE_GETTID
51#define FIO_USE_GENERIC_INIT_RANDOM_STATE
52#define FIO_HAVE_PWRITEV2
53#define FIO_HAVE_SHM_ATTACH_REMOVED
54
55#ifdef MAP_HUGETLB
56#define FIO_HAVE_MMAP_HUGE
57#endif
58
59#define OS_MAP_ANON MAP_ANONYMOUS
60
61#define FIO_EXT_ENG_DIR "/usr/lib/fio"
62
63typedef cpu_set_t os_cpu_mask_t;
64
65#ifdef CONFIG_3ARG_AFFINITY
66#define fio_setaffinity(pid, cpumask) \
67 sched_setaffinity((pid), sizeof(cpumask), &(cpumask))
68#define fio_getaffinity(pid, ptr) \
69 sched_getaffinity((pid), sizeof(cpu_set_t), (ptr))
70#elif defined(CONFIG_2ARG_AFFINITY)
71#define fio_setaffinity(pid, cpumask) \
72 sched_setaffinity((pid), &(cpumask))
73#define fio_getaffinity(pid, ptr) \
74 sched_getaffinity((pid), (ptr))
75#endif
76
77#define fio_cpu_clear(mask, cpu) (void) CPU_CLR((cpu), (mask))
78#define fio_cpu_set(mask, cpu) (void) CPU_SET((cpu), (mask))
79#define fio_cpu_isset(mask, cpu) (CPU_ISSET((cpu), (mask)) != 0)
80#define fio_cpu_count(mask) CPU_COUNT((mask))
81
82static inline int fio_cpuset_init(os_cpu_mask_t *mask)
83{
84 CPU_ZERO(mask);
85 return 0;
86}
87
88static inline int fio_cpuset_exit(os_cpu_mask_t *mask)
89{
90 return 0;
91}
92
93#define FIO_MAX_CPUS CPU_SETSIZE
94
95enum {
96 IOPRIO_CLASS_NONE,
97 IOPRIO_CLASS_RT,
98 IOPRIO_CLASS_BE,
99 IOPRIO_CLASS_IDLE,
100};
101
102enum {
103 IOPRIO_WHO_PROCESS = 1,
104 IOPRIO_WHO_PGRP,
105 IOPRIO_WHO_USER,
106};
107
108#define IOPRIO_BITS 16
109#define IOPRIO_CLASS_SHIFT 13
110
111#define IOPRIO_MIN_PRIO 0 /* highest priority */
112#define IOPRIO_MAX_PRIO 7 /* lowest priority */
113
114#define IOPRIO_MIN_PRIO_CLASS 0
115#define IOPRIO_MAX_PRIO_CLASS 3
116
117static inline int ioprio_set(int which, int who, int ioprio_class, int ioprio)
118{
119 /*
120 * If no class is set, assume BE
121 */
122 if (!ioprio_class)
123 ioprio_class = IOPRIO_CLASS_BE;
124
125 ioprio |= ioprio_class << IOPRIO_CLASS_SHIFT;
126 return syscall(__NR_ioprio_set, which, who, ioprio);
127}
128
129#ifndef CONFIG_HAVE_GETTID
130static inline int gettid(void)
131{
132 return syscall(__NR_gettid);
133}
134#endif
135
136#define SPLICE_DEF_SIZE (64*1024)
137
138#ifndef BLKGETSIZE64
139#define BLKGETSIZE64 _IOR(0x12,114,size_t)
140#endif
141
142#ifndef BLKFLSBUF
143#define BLKFLSBUF _IO(0x12,97)
144#endif
145
146#ifndef BLKDISCARD
147#define BLKDISCARD _IO(0x12,119)
148#endif
149
150static inline int blockdev_invalidate_cache(struct fio_file *f)
151{
152 return ioctl(f->fd, BLKFLSBUF);
153}
154
155static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)
156{
157 if (!ioctl(f->fd, BLKGETSIZE64, bytes))
158 return 0;
159
160 return errno;
161}
162
163static inline unsigned long long os_phys_mem(void)
164{
165 long pagesize, pages;
166
167 pagesize = sysconf(_SC_PAGESIZE);
168 pages = sysconf(_SC_PHYS_PAGES);
169 if (pages == -1 || pagesize == -1)
170 return 0;
171
172 return (unsigned long long) pages * (unsigned long long) pagesize;
173}
174
175static inline int fio_lookup_raw(dev_t dev, int *majdev, int *mindev)
176{
177 struct raw_config_request rq;
178 int fd;
179
180 if (major(dev) != RAW_MAJOR)
181 return 1;
182
183 /*
184 * we should be able to find /dev/rawctl or /dev/raw/rawctl
185 */
186 fd = open("/dev/rawctl", O_RDONLY);
187 if (fd < 0) {
188 fd = open("/dev/raw/rawctl", O_RDONLY);
189 if (fd < 0)
190 return 1;
191 }
192
193 rq.raw_minor = minor(dev);
194 if (ioctl(fd, RAW_GETBIND, &rq) < 0) {
195 close(fd);
196 return 1;
197 }
198
199 close(fd);
200 *majdev = rq.block_major;
201 *mindev = rq.block_minor;
202 return 0;
203}
204
205#ifdef O_NOATIME
206#define FIO_O_NOATIME O_NOATIME
207#else
208#define FIO_O_NOATIME 0
209#endif
210
211#ifdef O_ATOMIC
212#define OS_O_ATOMIC O_ATOMIC
213#else
214#define OS_O_ATOMIC 040000000
215#endif
216
217#ifdef MADV_REMOVE
218#define FIO_MADV_FREE MADV_REMOVE
219#endif
220
221/* Check for GCC or Clang byte swap intrinsics */
222#if (__has_builtin(__builtin_bswap16) && __has_builtin(__builtin_bswap32) \
223 && __has_builtin(__builtin_bswap64)) || (__GNUC__ > 4 \
224 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 8)) /* fio_swapN */
225#define fio_swap16(x) __builtin_bswap16(x)
226#define fio_swap32(x) __builtin_bswap32(x)
227#define fio_swap64(x) __builtin_bswap64(x)
228#else
229#include <byteswap.h>
230#define fio_swap16(x) bswap_16(x)
231#define fio_swap32(x) bswap_32(x)
232#define fio_swap64(x) bswap_64(x)
233#endif /* fio_swapN */
234
235#define CACHE_LINE_FILE \
236 "/sys/devices/system/cpu/cpu0/cache/index0/coherency_line_size"
237
238static inline int arch_cache_line_size(void)
239{
240 char size[32];
241 int fd, ret;
242
243 fd = open(CACHE_LINE_FILE, O_RDONLY);
244 if (fd < 0)
245 return -1;
246
247 ret = read(fd, size, sizeof(size));
248
249 close(fd);
250
251 if (ret <= 0)
252 return -1;
253 else
254 return atoi(size);
255}
256
257#ifdef __powerpc64__
258#define FIO_HAVE_CPU_ONLINE_SYSCONF
259static inline unsigned int cpus_online(void)
260{
261 return sysconf(_SC_NPROCESSORS_CONF);
262}
263#endif
264
265static inline unsigned long long get_fs_free_size(const char *path)
266{
267 unsigned long long ret;
268 struct statfs s;
269
270 if (statfs(path, &s) < 0)
271 return -1ULL;
272
273 ret = s.f_bsize;
274 ret *= (unsigned long long) s.f_bfree;
275 return ret;
276}
277
278static inline int os_trim(struct fio_file *f, unsigned long long start,
279 unsigned long long len)
280{
281 uint64_t range[2];
282
283 range[0] = start;
284 range[1] = len;
285
286 if (!ioctl(f->fd, BLKDISCARD, range))
287 return 0;
288
289 return errno;
290}
291
292#ifdef CONFIG_SCHED_IDLE
293static inline int fio_set_sched_idle(void)
294{
295 struct sched_param p = { .sched_priority = 0, };
296 return sched_setscheduler(gettid(), SCHED_IDLE, &p);
297}
298#endif
299
300#ifndef F_GET_RW_HINT
301#ifndef F_LINUX_SPECIFIC_BASE
302#define F_LINUX_SPECIFIC_BASE 1024
303#endif
304#define F_GET_RW_HINT (F_LINUX_SPECIFIC_BASE + 11)
305#define F_SET_RW_HINT (F_LINUX_SPECIFIC_BASE + 12)
306#define F_GET_FILE_RW_HINT (F_LINUX_SPECIFIC_BASE + 13)
307#define F_SET_FILE_RW_HINT (F_LINUX_SPECIFIC_BASE + 14)
308#endif
309
310#ifndef RWH_WRITE_LIFE_NONE
311#define RWH_WRITE_LIFE_NOT_SET 0
312#define RWH_WRITE_LIFE_NONE 1
313#define RWH_WRITE_LIFE_SHORT 2
314#define RWH_WRITE_LIFE_MEDIUM 3
315#define RWH_WRITE_LIFE_LONG 4
316#define RWH_WRITE_LIFE_EXTREME 5
317#endif
318
319#define FIO_HAVE_WRITE_HINT
320
321#ifndef RWF_HIPRI
322#define RWF_HIPRI 0x00000001
323#endif
324#ifndef RWF_DSYNC
325#define RWF_DSYNC 0x00000002
326#endif
327#ifndef RWF_SYNC
328#define RWF_SYNC 0x00000004
329#endif
330#ifndef RWF_NOWAIT
331#define RWF_NOWAIT 0x00000008
332#endif
333
334#ifndef RWF_UNCACHED
335#define RWF_UNCACHED 0x00000040
336#endif
337
338#ifndef RWF_WRITE_LIFE_SHIFT
339#define RWF_WRITE_LIFE_SHIFT 4
340#define RWF_WRITE_LIFE_SHORT (1 << RWF_WRITE_LIFE_SHIFT)
341#define RWF_WRITE_LIFE_MEDIUM (2 << RWF_WRITE_LIFE_SHIFT)
342#define RWF_WRITE_LIFE_LONG (3 << RWF_WRITE_LIFE_SHIFT)
343#define RWF_WRITE_LIFE_EXTREME (4 << RWF_WRITE_LIFE_SHIFT)
344#endif
345
346#ifndef CONFIG_PWRITEV2
347#ifdef __NR_preadv2
348static inline void make_pos_h_l(unsigned long *pos_h, unsigned long *pos_l,
349 off_t offset)
350{
351#if BITS_PER_LONG == 64
352 *pos_l = offset;
353 *pos_h = 0;
354#else
355 *pos_l = offset & 0xffffffff;
356 *pos_h = ((uint64_t) offset) >> 32;
357#endif
358}
359static inline ssize_t preadv2(int fd, const struct iovec *iov, int iovcnt,
360 off_t offset, unsigned int flags)
361{
362 unsigned long pos_l, pos_h;
363
364 make_pos_h_l(&pos_h, &pos_l, offset);
365 return syscall(__NR_preadv2, fd, iov, iovcnt, pos_l, pos_h, flags);
366}
367static inline ssize_t pwritev2(int fd, const struct iovec *iov, int iovcnt,
368 off_t offset, unsigned int flags)
369{
370 unsigned long pos_l, pos_h;
371
372 make_pos_h_l(&pos_h, &pos_l, offset);
373 return syscall(__NR_pwritev2, fd, iov, iovcnt, pos_l, pos_h, flags);
374}
375#else
376static inline ssize_t preadv2(int fd, const struct iovec *iov, int iovcnt,
377 off_t offset, unsigned int flags)
378{
379 errno = ENOSYS;
380 return -1;
381}
382static inline ssize_t pwritev2(int fd, const struct iovec *iov, int iovcnt,
383 off_t offset, unsigned int flags)
384{
385 errno = ENOSYS;
386 return -1;
387}
388#endif /* __NR_preadv2 */
389#endif /* CONFIG_PWRITEV2 */
390
391static inline int shm_attach_to_open_removed(void)
392{
393 return 1;
394}
395
396#ifdef CONFIG_LINUX_FALLOCATE
397#define FIO_HAVE_NATIVE_FALLOCATE
398static inline bool fio_fallocate(struct fio_file *f, uint64_t offset,
399 uint64_t len)
400{
401 int ret;
402 ret = fallocate(f->fd, 0, offset, len);
403 if (ret == 0)
404 return true;
405
406 /* Work around buggy old glibc versions... */
407 if (ret > 0)
408 errno = ret;
409
410 return false;
411}
412#endif
413
414#define FIO_HAVE_CPU_HAS
415static inline bool os_cpu_has(cpu_features feature)
416{
417 bool have_feature;
418 unsigned long fio_unused hwcap;
419
420 switch (feature) {
421#ifdef ARCH_HAVE_CRC_CRYPTO
422 case CPU_ARM64_CRC32C:
423 hwcap = getauxval(AT_HWCAP);
424 have_feature = (hwcap & HWCAP_CRC32) != 0;
425 break;
426#endif
427 default:
428 have_feature = false;
429 }
430
431 return have_feature;
432}
433
434#endif