[PATCH] Fix buflen trimming
[fio.git] / os-linux.h
CommitLineData
ebac4655
JA
1#ifndef FIO_OS_LINUX_H
2#define FIO_OS_LINUX_H
3
4#include <sys/ioctl.h>
8756e4d4 5#include <sys/uio.h>
3c39a379
JA
6#include <sys/syscall.h>
7#include <unistd.h>
97fbdfa1 8#include <fcntl.h>
ea421790 9#include <linux/unistd.h>
ebac4655
JA
10
11#define FIO_HAVE_LIBAIO
12#define FIO_HAVE_POSIXAIO
13#define FIO_HAVE_FADVISE
14#define FIO_HAVE_CPU_AFFINITY
15#define FIO_HAVE_DISK_UTIL
16#define FIO_HAVE_SGIO
ba4f8923 17#define FIO_HAVE_IOPRIO
8756e4d4 18#define FIO_HAVE_SPLICE
22f78b32 19#define FIO_HAVE_IOSCHED_SWITCH
2c0ecd28 20#define FIO_HAVE_ODIRECT
74b025b0 21#define FIO_HAVE_HUGETLB
ebac4655
JA
22
23#define OS_MAP_ANON (MAP_ANONYMOUS)
24
25typedef cpu_set_t os_cpu_mask_t;
6dfd46b9 26typedef struct drand48_data os_random_state_t;
ebac4655
JA
27
28/*
29 * we want fadvise64 really, but it's so tangled... later
30 */
31#define fadvise(fd, off, len, advice) \
32 posix_fadvise((fd), (off_t)(off), (len), (advice))
33
34#define fio_setaffinity(td) \
35 sched_setaffinity((td)->pid, sizeof((td)->cpumask), &(td)->cpumask)
36#define fio_getaffinity(pid, ptr) \
37 sched_getaffinity((pid), sizeof(cpu_set_t), (ptr))
38
39static inline int ioprio_set(int which, int who, int ioprio)
40{
41 return syscall(__NR_ioprio_set, which, who, ioprio);
42}
43
97fbdfa1
JA
44/*
45 * Just check for SPLICE_F_MOVE, if that isn't there, assume the others
46 * aren't either.
47 */
48#ifndef SPLICE_F_MOVE
49#define SPLICE_F_MOVE (0x01) /* move pages instead of copying */
50#define SPLICE_F_NONBLOCK (0x02) /* don't block on the pipe splicing (but */
51 /* we may still block on the fd we splice */
52 /* from/to, of course */
53#define SPLICE_F_MORE (0x04) /* expect more data */
54#define SPLICE_F_GIFT (0x08) /* pages passed in are a gift */
8756e4d4
JA
55
56static inline int splice(int fdin, loff_t *off_in, int fdout, loff_t *off_out,
57 size_t len, unsigned long flags)
58{
97fbdfa1 59 return syscall(__NR_sys_splice, fdin, off_in, fdout, off_out, len, flags);
8756e4d4
JA
60}
61
62static inline int tee(int fdin, int fdout, size_t len, unsigned int flags)
63{
97fbdfa1 64 return syscall(__NR_sys_tee, fdin, fdout, len, flags);
8756e4d4
JA
65}
66
67static inline int vmsplice(int fd, const struct iovec *iov,
68 unsigned long nr_segs, unsigned int flags)
69{
97fbdfa1 70 return syscall(__NR_sys_vmsplice, fd, iov, nr_segs, flags);
8756e4d4 71}
97fbdfa1 72#endif
8756e4d4 73
3feedc60
JA
74#define SPLICE_DEF_SIZE (64*1024)
75
ebac4655
JA
76enum {
77 IOPRIO_WHO_PROCESS = 1,
78 IOPRIO_WHO_PGRP,
79 IOPRIO_WHO_USER,
80};
81
82#define IOPRIO_CLASS_SHIFT 13
83
84#ifndef BLKGETSIZE64
85#define BLKGETSIZE64 _IOR(0x12,114,size_t)
86#endif
87
e5b401d4
JA
88#ifndef BLKFLSBUF
89#define BLKFLSBUF _IO(0x12,97)
90#endif
91
92static inline int blockdev_invalidate_cache(int fd)
93{
94 if (!ioctl(fd, BLKFLSBUF))
95 return 0;
96
97 return errno;
98}
99
9104f874 100static inline int blockdev_size(int fd, unsigned long long *bytes)
ebac4655
JA
101{
102 if (!ioctl(fd, BLKGETSIZE64, bytes))
103 return 0;
104
105 return errno;
106}
107
32cd46a0
JA
108static inline unsigned long long os_phys_mem(void)
109{
110 long pagesize, pages;
111
112 pagesize = sysconf(_SC_PAGESIZE);
113 pages = sysconf(_SC_PHYS_PAGES);
114 if (pages == -1 || pagesize == -1)
115 return 0;
116
117 return (unsigned long long) pages * (unsigned long long) pagesize;
118}
119
6dfd46b9
JA
120static inline void os_random_seed(unsigned long seed, os_random_state_t *rs)
121{
122 srand48_r(seed, rs);
123}
124
125static inline long os_random_long(os_random_state_t *rs)
126{
127 long val;
128
129 lrand48_r(rs, &val);
130 return val;
131}
132
133static inline double os_random_double(os_random_state_t *rs)
134{
135 double val;
136
137 drand48_r(rs, &val);
138 return val;
139}
140
ebac4655 141#endif