windows: fix fio_cpu_count() definition
[fio.git] / os / os-windows.h
CommitLineData
93bcfd20
BC
1#ifndef FIO_OS_WINDOWS_H
2#define FIO_OS_WINDOWS_H
3
4#define FIO_OS os_windows
5
6#include <sys/types.h>
7#include <sys/shm.h>
8#include <sys/stat.h>
9#include <errno.h>
5aa23eb8 10#include <winsock2.h>
93bcfd20
BC
11#include <windows.h>
12#include <psapi.h>
13#include <stdlib.h>
14
15#include "../smalloc.h"
16#include "../file.h"
17#include "../log.h"
95db3079 18#include "../lib/hweight.h"
93bcfd20 19
1f81991e
JA
20#include "windows/posix.h"
21
f16b7405
BC
22#ifndef PTHREAD_STACK_MIN
23#define PTHREAD_STACK_MIN 65535
24#endif
25
93bcfd20
BC
26#define FIO_HAVE_ODIRECT
27#define FIO_HAVE_CPU_AFFINITY
28#define FIO_HAVE_CHARDEV_SIZE
93bcfd20 29#define FIO_HAVE_GETTID
93bcfd20
BC
30#define FIO_USE_GENERIC_RAND
31
32#define FIO_PREFERRED_ENGINE "windowsaio"
33#define FIO_PREFERRED_CLOCK_SOURCE CS_CGETTIME
34#define FIO_OS_PATH_SEPARATOR "\\"
35
36#define FIO_MAX_CPUS MAXIMUM_PROCESSORS
37
93bcfd20
BC
38#define OS_MAP_ANON MAP_ANON
39
93bcfd20
BC
40#define fio_swap16(x) _byteswap_ushort(x)
41#define fio_swap32(x) _byteswap_ulong(x)
42#define fio_swap64(x) _byteswap_uint64(x)
43
93bcfd20
BC
44typedef DWORD_PTR os_cpu_mask_t;
45
93bcfd20
BC
46#define _SC_PAGESIZE 0x1
47#define _SC_NPROCESSORS_ONLN 0x2
48#define _SC_PHYS_PAGES 0x4
49
50#define SA_RESTART 0
51#define SIGPIPE 0
52
53/*
54 * Windows doesn't have O_DIRECT or O_SYNC, so define them
55 * here so we can reject them at runtime when using the _open
56 * interface (windowsaio uses CreateFile)
57 */
58#define O_DIRECT 0x1000000
59#define O_SYNC 0x2000000
60
61/* Windows doesn't support madvise, so any values will work */
62#define POSIX_MADV_DONTNEED 0
63#define POSIX_MADV_SEQUENTIAL 0
64#define POSIX_MADV_RANDOM 0
65
66#define F_SETFL 0x1
67#define F_GETFL 0x2
68#define O_NONBLOCK FIONBIO
69
70/* Winsock doesn't support MSG_WAIT */
71#define OS_MSG_DONTWAIT 0
72
73#define POLLOUT 1
74#define POLLIN 2
75#define POLLERR 0
76#define POLLHUP 1
77
78#define SIGCONT 0
84306c1d 79#define SIGUSR1 1
2277d5d5 80#define SIGUSR2 2
93bcfd20
BC
81
82typedef int sigset_t;
83typedef int siginfo_t;
84
85struct sigaction
86{
87 void (*sa_handler)(int);
88 sigset_t sa_mask;
89 int sa_flags;
90 void* (*sa_sigaction)(int, siginfo_t *, void*);
91};
92
93bcfd20
BC
93long sysconf(int name);
94
95int kill(pid_t pid, int sig);
96pid_t setsid(void);
97int setgid(gid_t gid);
98int setuid(uid_t uid);
99int nice(int incr);
100int sigaction(int sig, const struct sigaction *act,
101 struct sigaction *oact);
102int fsync(int fildes);
103int fork(void);
104int fcntl(int fildes, int cmd, ...);
105int fdatasync(int fildes);
106int lstat(const char * path, struct stat * buf);
107uid_t geteuid(void);
108int nanosleep(const struct timespec *rqtp, struct timespec *rmtp);
109ssize_t pread(int fildes, void *buf, size_t nbyte, off_t offset);
110ssize_t pwrite(int fildes, const void *buf, size_t nbyte,
111 off_t offset);
112extern void td_fill_rand_seeds(struct thread_data *);
113
114static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)
115{
116 int rc = 0;
117 HANDLE hFile;
5aa23eb8
BC
118 GET_LENGTH_INFORMATION info;
119 DWORD outBytes;
93bcfd20
BC
120
121 if (f->hFile == NULL) {
122 hFile = CreateFile(f->file_name, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE,
123 NULL, OPEN_EXISTING, 0, NULL);
124 } else {
125 hFile = f->hFile;
126 }
127
93bcfd20
BC
128 if (DeviceIoControl(hFile, IOCTL_DISK_GET_LENGTH_INFO, NULL, 0, &info, sizeof(info), &outBytes, NULL))
129 *bytes = info.Length.QuadPart;
130 else
131 rc = EIO;
132
133 /* If we were passed a POSIX fd,
134 * close the HANDLE we created via CreateFile */
135 if (hFile != INVALID_HANDLE_VALUE && f->hFile == NULL)
136 CloseHandle(hFile);
137
138 return rc;
139}
140
141static inline int chardev_size(struct fio_file *f, unsigned long long *bytes)
142{
143 return blockdev_size(f, bytes);
144}
145
146static inline int blockdev_invalidate_cache(struct fio_file *f)
147{
148 /* There's no way to invalidate the cache in Windows
149 * so just pretend to succeed */
150 return 0;
151}
152
153static inline unsigned long long os_phys_mem(void)
154{
01d26955 155 long pagesize, pages;
93bcfd20 156
01d26955
BC
157 pagesize = sysconf(_SC_PAGESIZE);
158 pages = sysconf(_SC_PHYS_PAGES);
159 if (pages == -1 || pagesize == -1)
160 return 0;
161
162 return (unsigned long long) pages * (unsigned long long) pagesize;
93bcfd20
BC
163}
164
165static inline void os_get_tmpdir(char *path, int len)
166{
167 GetTempPath(len, path);
168}
169
170static inline int gettid(void)
171{
172 return GetCurrentThreadId();
173}
174
175static inline int fio_setaffinity(int pid, os_cpu_mask_t cpumask)
176{
177 HANDLE h;
178 BOOL bSuccess = FALSE;
179
180 h = OpenThread(THREAD_QUERY_INFORMATION | THREAD_SET_INFORMATION, TRUE, pid);
181 if (h != NULL) {
182 bSuccess = SetThreadAffinityMask(h, cpumask);
3cf1ff44
BC
183 if (!bSuccess)
184 log_err("fio_setaffinity failed: failed to set thread affinity (pid %d, mask %.16llx)\n", pid, cpumask);
185
93bcfd20 186 CloseHandle(h);
3cf1ff44
BC
187 } else {
188 log_err("fio_setaffinity failed: failed to get handle for pid %d\n", pid);
93bcfd20
BC
189 }
190
191 return (bSuccess)? 0 : -1;
192}
193
194static inline void fio_getaffinity(int pid, os_cpu_mask_t *mask)
195{
196 os_cpu_mask_t systemMask;
197
198 HANDLE h = OpenProcess(PROCESS_QUERY_INFORMATION, TRUE, pid);
199
200 if (h != NULL) {
201 GetProcessAffinityMask(h, mask, &systemMask);
202 CloseHandle(h);
203 } else {
204 log_err("fio_getaffinity failed: failed to get handle for pid %d\n", pid);
205 }
206}
207
208static inline void fio_cpu_clear(os_cpu_mask_t *mask, int cpu)
209{
210 *mask ^= 1 << (cpu-1);
211}
212
213static inline void fio_cpu_set(os_cpu_mask_t *mask, int cpu)
214{
3cf1ff44 215 *mask |= 1 << cpu;
93bcfd20
BC
216}
217
a1fc70fb 218static inline int fio_cpu_count(os_cpu_mask_t *mask)
c2acfbac
JA
219{
220 return hweight64(*mask);
221}
222
93bcfd20
BC
223static inline int fio_cpuset_init(os_cpu_mask_t *mask)
224{
225 *mask = 0;
226 return 0;
227}
228
229static inline int fio_cpuset_exit(os_cpu_mask_t *mask)
230{
231 return 0;
232}
233
234static inline int init_random_state(struct thread_data *td, unsigned long *rand_seeds, int size)
235{
236 HCRYPTPROV hCryptProv;
237
238 if (!CryptAcquireContext(&hCryptProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
239 {
240 errno = GetLastError();
241 log_err("CryptAcquireContext() failed: error %d\n", errno);
242 return 1;
243 }
244
245 if (!CryptGenRandom(hCryptProv, size, (BYTE*)rand_seeds)) {
246 errno = GetLastError();
247 log_err("CryptGenRandom() failed, error %d\n", errno);
248 CryptReleaseContext(hCryptProv, 0);
249 return 1;
250 }
251
252 CryptReleaseContext(hCryptProv, 0);
253 td_fill_rand_seeds(td);
254 return 0;
255}
256
257
f2a2ce0e
HL
258static inline int fio_set_sched_idle(void)
259{
260 /* SetThreadPriority returns nonzero for success */
261 return (SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_IDLE))? 0 : -1;
262}
263
264
93bcfd20 265#endif /* FIO_OS_WINDOWS_H */