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