[PATCH] Support for splice IO engine
[fio.git] / os-linux.h
1 #ifndef FIO_OS_LINUX_H
2 #define FIO_OS_LINUX_H
3
4 #include <sys/ioctl.h>
5 #include <sys/uio.h>
6 #include <asm/unistd.h>
7
8 #define FIO_HAVE_LIBAIO
9 #define FIO_HAVE_POSIXAIO
10 #define FIO_HAVE_FADVISE
11 #define FIO_HAVE_CPU_AFFINITY
12 #define FIO_HAVE_DISK_UTIL
13 #define FIO_HAVE_SGIO
14 #define FIO_HAVE_IOPRIO
15 #define FIO_HAVE_SPLICE
16
17 #define OS_MAP_ANON             (MAP_ANONYMOUS)
18
19 typedef cpu_set_t os_cpu_mask_t;
20
21 /*
22  * we want fadvise64 really, but it's so tangled... later
23  */
24 #define fadvise(fd, off, len, advice)   \
25         posix_fadvise((fd), (off_t)(off), (len), (advice))
26
27 #define fio_setaffinity(td)             \
28         sched_setaffinity((td)->pid, sizeof((td)->cpumask), &(td)->cpumask)
29 #define fio_getaffinity(pid, ptr)       \
30         sched_getaffinity((pid), sizeof(cpu_set_t), (ptr))
31
32 static inline int ioprio_set(int which, int who, int ioprio)
33 {
34         return syscall(__NR_ioprio_set, which, who, ioprio);
35 }
36
37 static _syscall6(int, sys_splice, int, fdin, loff_t *, off_in, int, fdout, loff_t *, off_out, size_t, len, unsigned int, flags);
38 static _syscall4(int, sys_vmsplice, int, fd, const struct iovec *, iov, unsigned long, nr_segs, unsigned int, flags);
39 static _syscall4(int, sys_tee, int, fdin, int, fdout, size_t, len, unsigned int, flags);
40
41 static inline int splice(int fdin, loff_t *off_in, int fdout, loff_t *off_out,
42                          size_t len, unsigned long flags)
43 {
44         return sys_splice(fdin, off_in, fdout, off_out, len, flags);
45 }
46
47 static inline int tee(int fdin, int fdout, size_t len, unsigned int flags)
48 {
49         return sys_tee(fdin, fdout, len, flags);
50 }
51
52 static inline int vmsplice(int fd, const struct iovec *iov,
53                            unsigned long nr_segs, unsigned int flags)
54 {
55         return sys_vmsplice(fd, iov, nr_segs, flags);
56 }
57
58 #define SPLICE_F_MOVE   (0x01)  /* move pages instead of copying */
59 #define SPLICE_F_NONBLOCK (0x02) /* don't block on the pipe splicing (but */
60                                  /* we may still block on the fd we splice */
61                                  /* from/to, of course */
62 #define SPLICE_F_MORE   (0x04)  /* expect more data */
63 #define SPLICE_F_GIFT   (0x08)  /* pages passed in are a gift */
64
65 enum {
66         IOPRIO_WHO_PROCESS = 1,
67         IOPRIO_WHO_PGRP,
68         IOPRIO_WHO_USER,
69 };
70
71 #define IOPRIO_CLASS_SHIFT      13
72
73 #ifndef BLKGETSIZE64
74 #define BLKGETSIZE64    _IOR(0x12,114,size_t)
75 #endif
76
77 static inline int blockdev_size(int fd, unsigned long long *bytes)
78 {
79         if (!ioctl(fd, BLKGETSIZE64, bytes))
80                 return 0;
81
82         return errno;
83 }
84
85 #endif