Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 1 | #ifndef FIO_OS_LINUX_H |
| 2 | #define FIO_OS_LINUX_H |
| 3 | |
Jens Axboe | cca8464 | 2011-10-07 12:47:57 +0200 | [diff] [blame] | 4 | #define FIO_OS os_linux |
| 5 | |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 6 | #include <sys/ioctl.h> |
Jens Axboe | 8756e4d | 2006-05-27 20:24:53 +0200 | [diff] [blame] | 7 | #include <sys/uio.h> |
Jens Axboe | 3c39a37 | 2006-06-06 20:56:12 +0200 | [diff] [blame] | 8 | #include <sys/syscall.h> |
Jens Axboe | 2e3bd4c | 2010-05-17 12:29:57 +0200 | [diff] [blame] | 9 | #include <sys/vfs.h> |
Jens Axboe | 39b9356 | 2012-11-27 08:27:50 +0100 | [diff] [blame] | 10 | #include <sys/mman.h> |
Jens Axboe | 3c39a37 | 2006-06-06 20:56:12 +0200 | [diff] [blame] | 11 | #include <unistd.h> |
Jens Axboe | 97fbdfa | 2006-10-05 13:44:15 +0200 | [diff] [blame] | 12 | #include <fcntl.h> |
Jens Axboe | 3c2d93e | 2009-01-05 19:04:15 +0100 | [diff] [blame] | 13 | #include <errno.h> |
Jens Axboe | d09913c | 2009-12-16 19:23:10 +0100 | [diff] [blame] | 14 | #include <sched.h> |
Jens Axboe | ea42179 | 2006-07-11 09:11:07 +0200 | [diff] [blame] | 15 | #include <linux/unistd.h> |
Jens Axboe | 07e5b26 | 2007-04-02 14:46:07 +0200 | [diff] [blame] | 16 | #include <linux/raw.h> |
| 17 | #include <linux/major.h> |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 18 | |
Jens Axboe | 79a4318 | 2010-09-07 13:28:58 +0200 | [diff] [blame] | 19 | #include "binject.h" |
Jens Axboe | e2e5888 | 2011-01-04 08:36:06 +0100 | [diff] [blame] | 20 | #include "../file.h" |
Jens Axboe | f3de88a | 2008-02-24 21:36:00 +0100 | [diff] [blame] | 21 | |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 22 | #define FIO_HAVE_CPU_AFFINITY |
| 23 | #define FIO_HAVE_DISK_UTIL |
| 24 | #define FIO_HAVE_SGIO |
Jens Axboe | ba4f892 | 2006-01-26 13:45:54 -0800 | [diff] [blame] | 25 | #define FIO_HAVE_IOPRIO |
Jens Axboe | 22f78b3 | 2006-06-07 11:30:07 +0200 | [diff] [blame] | 26 | #define FIO_HAVE_IOSCHED_SWITCH |
Jens Axboe | 2c0ecd2 | 2006-06-08 13:25:41 +0200 | [diff] [blame] | 27 | #define FIO_HAVE_ODIRECT |
Jens Axboe | 74b025b | 2006-12-19 15:18:14 +0100 | [diff] [blame] | 28 | #define FIO_HAVE_HUGETLB |
Jens Axboe | 07e5b26 | 2007-04-02 14:46:07 +0200 | [diff] [blame] | 29 | #define FIO_HAVE_RAWBIND |
Jens Axboe | 5e62c22 | 2007-05-22 13:27:30 +0200 | [diff] [blame] | 30 | #define FIO_HAVE_BLKTRACE |
Jens Axboe | f356d01 | 2009-01-05 09:56:29 +0100 | [diff] [blame] | 31 | #define FIO_HAVE_PSHARED_MUTEX |
Jens Axboe | eb7ccf3 | 2009-04-29 09:48:04 +0200 | [diff] [blame] | 32 | #define FIO_HAVE_CL_SIZE |
Jens Axboe | a696fa2 | 2009-12-04 10:05:02 +0100 | [diff] [blame] | 33 | #define FIO_HAVE_CGROUPS |
Jens Axboe | 2e3bd4c | 2010-05-17 12:29:57 +0200 | [diff] [blame] | 34 | #define FIO_HAVE_FS_STAT |
Jens Axboe | a5f3027 | 2010-07-19 16:19:55 -0600 | [diff] [blame] | 35 | #define FIO_HAVE_TRIM |
Jens Axboe | 79a4318 | 2010-09-07 13:28:58 +0200 | [diff] [blame] | 36 | #define FIO_HAVE_BINJECT |
Jens Axboe | 47f767c | 2011-07-12 21:17:49 +0200 | [diff] [blame] | 37 | #define FIO_HAVE_GETTID |
Bruce Cran | 93bcfd2 | 2012-02-20 20:18:19 +0100 | [diff] [blame] | 38 | #define FIO_USE_GENERIC_INIT_RANDOM_STATE |
Jens Axboe | b6959b5 | 2007-02-14 21:40:27 +0000 | [diff] [blame] | 39 | |
Jens Axboe | 39b9356 | 2012-11-27 08:27:50 +0100 | [diff] [blame] | 40 | #ifdef MAP_HUGETLB |
| 41 | #define FIO_HAVE_MMAP_HUGE |
| 42 | #endif |
| 43 | |
Jens Axboe | dc873b6 | 2008-06-04 20:13:04 +0200 | [diff] [blame] | 44 | #define OS_MAP_ANON MAP_ANONYMOUS |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 45 | |
| 46 | typedef cpu_set_t os_cpu_mask_t; |
Jens Axboe | 4e78e40 | 2008-12-12 20:40:27 +0100 | [diff] [blame] | 47 | |
Jens Axboe | 6dfd46b | 2006-06-07 13:57:06 +0200 | [diff] [blame] | 48 | typedef struct drand48_data os_random_state_t; |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 49 | |
Jens Axboe | 67bf982 | 2013-01-10 11:23:19 +0100 | [diff] [blame] | 50 | #ifdef CONFIG_3ARG_AFFINITY |
Jens Axboe | e8462bd | 2009-07-06 12:59:04 +0200 | [diff] [blame] | 51 | #define fio_setaffinity(pid, cpumask) \ |
| 52 | sched_setaffinity((pid), sizeof(cpumask), &(cpumask)) |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 53 | #define fio_getaffinity(pid, ptr) \ |
| 54 | sched_getaffinity((pid), sizeof(cpu_set_t), (ptr)) |
Jens Axboe | 67bf982 | 2013-01-10 11:23:19 +0100 | [diff] [blame] | 55 | #elif defined(CONFIG_2ARG_AFFINITY) |
Jens Axboe | e8462bd | 2009-07-06 12:59:04 +0200 | [diff] [blame] | 56 | #define fio_setaffinity(pid, cpumask) \ |
| 57 | sched_setaffinity((pid), &(cpumask)) |
Jens Axboe | c8f025f | 2007-05-22 17:06:24 +0200 | [diff] [blame] | 58 | #define fio_getaffinity(pid, ptr) \ |
| 59 | sched_getaffinity((pid), (ptr)) |
| 60 | #endif |
Jens Axboe | be4ecfd | 2008-12-08 14:10:52 +0100 | [diff] [blame] | 61 | |
Jens Axboe | 1f4c0a4 | 2011-10-07 13:26:27 +0200 | [diff] [blame] | 62 | #define fio_cpu_clear(mask, cpu) (void) CPU_CLR((cpu), (mask)) |
| 63 | #define fio_cpu_set(mask, cpu) (void) CPU_SET((cpu), (mask)) |
Jens Axboe | d2ce18b | 2008-12-12 20:51:40 +0100 | [diff] [blame] | 64 | |
| 65 | static inline int fio_cpuset_init(os_cpu_mask_t *mask) |
| 66 | { |
| 67 | CPU_ZERO(mask); |
| 68 | return 0; |
| 69 | } |
| 70 | |
| 71 | static inline int fio_cpuset_exit(os_cpu_mask_t *mask) |
| 72 | { |
| 73 | return 0; |
| 74 | } |
Jens Axboe | 6d459ee | 2008-12-12 20:02:58 +0100 | [diff] [blame] | 75 | |
| 76 | #define FIO_MAX_CPUS CPU_SETSIZE |
Shawn Lewis | 4d8947d | 2007-07-25 07:51:58 +0200 | [diff] [blame] | 77 | |
Jens Axboe | 28727df | 2012-03-29 08:33:15 +0200 | [diff] [blame] | 78 | enum { |
| 79 | IOPRIO_CLASS_NONE, |
| 80 | IOPRIO_CLASS_RT, |
| 81 | IOPRIO_CLASS_BE, |
| 82 | IOPRIO_CLASS_IDLE, |
| 83 | }; |
| 84 | |
| 85 | enum { |
| 86 | IOPRIO_WHO_PROCESS = 1, |
| 87 | IOPRIO_WHO_PGRP, |
| 88 | IOPRIO_WHO_USER, |
| 89 | }; |
| 90 | |
| 91 | #define IOPRIO_BITS 16 |
| 92 | #define IOPRIO_CLASS_SHIFT 13 |
| 93 | |
| 94 | static inline int ioprio_set(int which, int who, int ioprio_class, int ioprio) |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 95 | { |
Jens Axboe | 28727df | 2012-03-29 08:33:15 +0200 | [diff] [blame] | 96 | /* |
| 97 | * If no class is set, assume BE |
| 98 | */ |
| 99 | if (!ioprio_class) |
| 100 | ioprio_class = IOPRIO_CLASS_BE; |
| 101 | |
| 102 | ioprio |= ioprio_class << IOPRIO_CLASS_SHIFT; |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 103 | return syscall(__NR_ioprio_set, which, who, ioprio); |
| 104 | } |
| 105 | |
Jens Axboe | 47f767c | 2011-07-12 21:17:49 +0200 | [diff] [blame] | 106 | static inline int gettid(void) |
| 107 | { |
| 108 | return syscall(__NR_gettid); |
| 109 | } |
| 110 | |
Jens Axboe | 3feedc6 | 2006-05-28 20:13:51 +0200 | [diff] [blame] | 111 | #define SPLICE_DEF_SIZE (64*1024) |
| 112 | |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 113 | #ifndef BLKGETSIZE64 |
| 114 | #define BLKGETSIZE64 _IOR(0x12,114,size_t) |
| 115 | #endif |
| 116 | |
Jens Axboe | e5b401d | 2006-10-18 16:03:40 +0200 | [diff] [blame] | 117 | #ifndef BLKFLSBUF |
| 118 | #define BLKFLSBUF _IO(0x12,97) |
| 119 | #endif |
| 120 | |
Jens Axboe | a5f3027 | 2010-07-19 16:19:55 -0600 | [diff] [blame] | 121 | #ifndef BLKDISCARD |
| 122 | #define BLKDISCARD _IO(0x12,119) |
| 123 | #endif |
| 124 | |
Jens Axboe | e2e5888 | 2011-01-04 08:36:06 +0100 | [diff] [blame] | 125 | static inline int blockdev_invalidate_cache(struct fio_file *f) |
Jens Axboe | e5b401d | 2006-10-18 16:03:40 +0200 | [diff] [blame] | 126 | { |
Bruce Cran | ecc314b | 2011-01-04 10:59:30 +0100 | [diff] [blame] | 127 | return ioctl(f->fd, BLKFLSBUF); |
Jens Axboe | e5b401d | 2006-10-18 16:03:40 +0200 | [diff] [blame] | 128 | } |
| 129 | |
Bruce Cran | ecc314b | 2011-01-04 10:59:30 +0100 | [diff] [blame] | 130 | static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes) |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 131 | { |
Bruce Cran | ecc314b | 2011-01-04 10:59:30 +0100 | [diff] [blame] | 132 | if (!ioctl(f->fd, BLKGETSIZE64, bytes)) |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 133 | return 0; |
| 134 | |
| 135 | return errno; |
| 136 | } |
| 137 | |
Jens Axboe | 32cd46a | 2006-06-07 13:40:40 +0200 | [diff] [blame] | 138 | static inline unsigned long long os_phys_mem(void) |
| 139 | { |
| 140 | long pagesize, pages; |
| 141 | |
| 142 | pagesize = sysconf(_SC_PAGESIZE); |
| 143 | pages = sysconf(_SC_PHYS_PAGES); |
| 144 | if (pages == -1 || pagesize == -1) |
| 145 | return 0; |
| 146 | |
| 147 | return (unsigned long long) pages * (unsigned long long) pagesize; |
| 148 | } |
| 149 | |
Jens Axboe | 6dfd46b | 2006-06-07 13:57:06 +0200 | [diff] [blame] | 150 | static inline void os_random_seed(unsigned long seed, os_random_state_t *rs) |
| 151 | { |
| 152 | srand48_r(seed, rs); |
| 153 | } |
| 154 | |
| 155 | static inline long os_random_long(os_random_state_t *rs) |
| 156 | { |
| 157 | long val; |
| 158 | |
| 159 | lrand48_r(rs, &val); |
| 160 | return val; |
| 161 | } |
| 162 | |
Jens Axboe | 8cc7afa | 2007-04-17 09:06:43 +0200 | [diff] [blame] | 163 | static inline int fio_lookup_raw(dev_t dev, int *majdev, int *mindev) |
Jens Axboe | 07e5b26 | 2007-04-02 14:46:07 +0200 | [diff] [blame] | 164 | { |
| 165 | struct raw_config_request rq; |
| 166 | int fd; |
| 167 | |
| 168 | if (major(dev) != RAW_MAJOR) |
Jens Axboe | 8cc7afa | 2007-04-17 09:06:43 +0200 | [diff] [blame] | 169 | return 1; |
Jens Axboe | 07e5b26 | 2007-04-02 14:46:07 +0200 | [diff] [blame] | 170 | |
| 171 | /* |
| 172 | * we should be able to find /dev/rawctl or /dev/raw/rawctl |
| 173 | */ |
| 174 | fd = open("/dev/rawctl", O_RDONLY); |
| 175 | if (fd < 0) { |
| 176 | fd = open("/dev/raw/rawctl", O_RDONLY); |
| 177 | if (fd < 0) |
Jens Axboe | 8cc7afa | 2007-04-17 09:06:43 +0200 | [diff] [blame] | 178 | return 1; |
Jens Axboe | 07e5b26 | 2007-04-02 14:46:07 +0200 | [diff] [blame] | 179 | } |
| 180 | |
| 181 | rq.raw_minor = minor(dev); |
| 182 | if (ioctl(fd, RAW_GETBIND, &rq) < 0) { |
| 183 | close(fd); |
Jens Axboe | 8cc7afa | 2007-04-17 09:06:43 +0200 | [diff] [blame] | 184 | return 1; |
Jens Axboe | 07e5b26 | 2007-04-02 14:46:07 +0200 | [diff] [blame] | 185 | } |
| 186 | |
| 187 | close(fd); |
| 188 | *majdev = rq.block_major; |
| 189 | *mindev = rq.block_minor; |
Jens Axboe | 8cc7afa | 2007-04-17 09:06:43 +0200 | [diff] [blame] | 190 | return 0; |
Jens Axboe | 07e5b26 | 2007-04-02 14:46:07 +0200 | [diff] [blame] | 191 | } |
| 192 | |
Jens Axboe | 5b6f5c6 | 2008-06-11 10:21:43 +0200 | [diff] [blame] | 193 | #ifdef O_NOATIME |
Jens Axboe | 5921e80 | 2008-05-30 15:02:38 +0200 | [diff] [blame] | 194 | #define FIO_O_NOATIME O_NOATIME |
Jens Axboe | 5b6f5c6 | 2008-06-11 10:21:43 +0200 | [diff] [blame] | 195 | #else |
| 196 | #define FIO_O_NOATIME 0 |
| 197 | #endif |
Jens Axboe | 5921e80 | 2008-05-30 15:02:38 +0200 | [diff] [blame] | 198 | |
Jens Axboe | a1c5807 | 2009-08-04 23:17:02 +0200 | [diff] [blame] | 199 | #ifdef MADV_REMOVE |
| 200 | #define FIO_MADV_FREE MADV_REMOVE |
| 201 | #endif |
| 202 | |
Jens Axboe | ff24519 | 2011-10-04 14:30:33 +0200 | [diff] [blame] | 203 | #define fio_swap16(x) __bswap_16(x) |
| 204 | #define fio_swap32(x) __bswap_32(x) |
| 205 | #define fio_swap64(x) __bswap_64(x) |
| 206 | |
Jens Axboe | eb7ccf3 | 2009-04-29 09:48:04 +0200 | [diff] [blame] | 207 | #define CACHE_LINE_FILE \ |
| 208 | "/sys/devices/system/cpu/cpu0/cache/index0/coherency_line_size" |
| 209 | |
| 210 | static inline int arch_cache_line_size(void) |
| 211 | { |
| 212 | char size[32]; |
| 213 | int fd, ret; |
| 214 | |
| 215 | fd = open(CACHE_LINE_FILE, O_RDONLY); |
| 216 | if (fd < 0) |
| 217 | return -1; |
| 218 | |
| 219 | ret = read(fd, size, sizeof(size)); |
Jens Axboe | ec76f15 | 2009-06-09 13:05:32 +0200 | [diff] [blame] | 220 | |
| 221 | close(fd); |
| 222 | |
Jens Axboe | eb7ccf3 | 2009-04-29 09:48:04 +0200 | [diff] [blame] | 223 | if (ret <= 0) |
| 224 | return -1; |
Jens Axboe | ec76f15 | 2009-06-09 13:05:32 +0200 | [diff] [blame] | 225 | else |
| 226 | return atoi(size); |
Jens Axboe | eb7ccf3 | 2009-04-29 09:48:04 +0200 | [diff] [blame] | 227 | } |
| 228 | |
Jens Axboe | 2e3bd4c | 2010-05-17 12:29:57 +0200 | [diff] [blame] | 229 | static inline unsigned long long get_fs_size(const char *path) |
| 230 | { |
| 231 | unsigned long long ret; |
| 232 | struct statfs s; |
| 233 | |
| 234 | if (statfs(path, &s) < 0) |
| 235 | return -1ULL; |
| 236 | |
| 237 | ret = s.f_bsize; |
| 238 | ret *= (unsigned long long) s.f_bfree; |
| 239 | return ret; |
| 240 | } |
| 241 | |
Jens Axboe | a5f3027 | 2010-07-19 16:19:55 -0600 | [diff] [blame] | 242 | static inline int os_trim(int fd, unsigned long long start, |
| 243 | unsigned long long len) |
| 244 | { |
| 245 | uint64_t range[2]; |
| 246 | |
| 247 | range[0] = start; |
| 248 | range[1] = len; |
| 249 | |
| 250 | if (!ioctl(fd, BLKDISCARD, range)) |
| 251 | return 0; |
| 252 | |
| 253 | return errno; |
| 254 | } |
| 255 | |
Jens Axboe | 7e09a9f | 2013-01-30 13:58:58 +0100 | [diff] [blame] | 256 | #ifdef CONFIG_SCHED_IDLE |
Huadong Liu | f2a2ce0 | 2013-01-30 13:22:24 +0100 | [diff] [blame] | 257 | static inline int fio_set_sched_idle(void) |
| 258 | { |
| 259 | struct sched_param p = { .sched_priority = 0, }; |
| 260 | return sched_setscheduler(gettid(), SCHED_IDLE, &p); |
| 261 | } |
Jens Axboe | 7e09a9f | 2013-01-30 13:58:58 +0100 | [diff] [blame] | 262 | #endif |
Huadong Liu | f2a2ce0 | 2013-01-30 13:22:24 +0100 | [diff] [blame] | 263 | |
Jens Axboe | ebac465 | 2005-12-08 15:25:21 +0100 | [diff] [blame] | 264 | #endif |