windows: target Windows 7 and add support for more than 64 CPUs
[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 <winsock2.h>
11 #include <windows.h>
12 #include <psapi.h>
13 #include <stdlib.h>
14
15 #include "../smalloc.h"
16 #include "../debug.h"
17 #include "../file.h"
18 #include "../log.h"
19 #include "../lib/hweight.h"
20 #include "../oslib/strcasestr.h"
21 #include "../lib/types.h"
22
23 #include "windows/posix.h"
24
25 /* MinGW won't declare rand_r unless _POSIX is defined */
26 #if defined(WIN32) && !defined(rand_r)
27 int rand_r(unsigned *);
28 #endif
29
30 #ifndef PTHREAD_STACK_MIN
31 #define PTHREAD_STACK_MIN 65535
32 #endif
33
34 #define FIO_HAVE_ODIRECT
35 #define FIO_HAVE_CPU_AFFINITY
36 #define FIO_HAVE_CHARDEV_SIZE
37 #define FIO_HAVE_GETTID
38 #define FIO_USE_GENERIC_RAND
39
40 #define FIO_PREFERRED_ENGINE            "windowsaio"
41 #define FIO_PREFERRED_CLOCK_SOURCE      CS_CGETTIME
42 #define FIO_OS_PATH_SEPARATOR           '\\'
43
44 #define OS_MAP_ANON             MAP_ANON
45
46 #define fio_swap16(x)   _byteswap_ushort(x)
47 #define fio_swap32(x)   _byteswap_ulong(x)
48 #define fio_swap64(x)   _byteswap_uint64(x)
49
50 #define _SC_PAGESIZE                    0x1
51 #define _SC_NPROCESSORS_ONLN    0x2
52 #define _SC_PHYS_PAGES                  0x4
53
54 #define SA_RESTART      0
55 #define SIGPIPE         0
56
57 /*
58  * Windows doesn't have O_DIRECT or O_SYNC, so define them
59  * here so we can reject them at runtime when using the _open
60  * interface (windowsaio uses CreateFile)
61  */
62 #define O_DIRECT        0x1000000
63 #define O_SYNC          0x2000000
64
65 /* Windows doesn't support madvise, so any values will work */
66 #define POSIX_MADV_DONTNEED             0
67 #define POSIX_MADV_SEQUENTIAL   0
68 #define POSIX_MADV_RANDOM               0
69
70 #define F_SETFL                 0x1
71 #define F_GETFL                 0x2
72 #define O_NONBLOCK              FIONBIO
73
74 /* Winsock doesn't support MSG_WAIT */
75 #define OS_MSG_DONTWAIT 0
76
77 #define SIGCONT 0
78 #define SIGUSR1 1
79 #define SIGUSR2 2
80
81 typedef int sigset_t;
82 typedef int siginfo_t;
83
84 struct sigaction
85 {
86         void (*sa_handler)(int);
87         sigset_t sa_mask;
88         int sa_flags;
89         void* (*sa_sigaction)(int, siginfo_t *, void*);
90 };
91
92 long sysconf(int name);
93
94 int kill(pid_t pid, int sig);
95 pid_t setsid(void);
96 int setgid(gid_t gid);
97 int setuid(uid_t uid);
98 int nice(int incr);
99 int sigaction(int sig, const struct sigaction *act,
100                 struct sigaction *oact);
101 int fsync(int fildes);
102 int fork(void);
103 int fcntl(int fildes, int cmd, ...);
104 int fdatasync(int fildes);
105 int lstat(const char * path, struct stat * buf);
106 uid_t geteuid(void);
107 char* ctime_r(const time_t *t, char *buf);
108 int nanosleep(const struct timespec *rqtp, struct timespec *rmtp);
109 ssize_t pread(int fildes, void *buf, size_t nbyte, off_t offset);
110 ssize_t pwrite(int fildes, const void *buf, size_t nbyte,
111                 off_t offset);
112
113 static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)
114 {
115         int rc = 0;
116         HANDLE hFile;
117         GET_LENGTH_INFORMATION info;
118         DWORD outBytes;
119
120         if (f->hFile == NULL) {
121                 hFile = CreateFile(f->file_name, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE,
122                                 NULL, OPEN_EXISTING, 0, NULL);
123         } else {
124                 hFile = f->hFile;
125         }
126
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         return ENOTSUP;
148 }
149
150 static inline unsigned long long os_phys_mem(void)
151 {
152         long pagesize, pages;
153
154         pagesize = sysconf(_SC_PAGESIZE);
155         pages = sysconf(_SC_PHYS_PAGES);
156         if (pages == -1 || pagesize == -1)
157                 return 0;
158
159         return (unsigned long long) pages * (unsigned long long) pagesize;
160 }
161
162 static inline int gettid(void)
163 {
164         return GetCurrentThreadId();
165 }
166
167 static inline int init_random_seeds(unsigned long *rand_seeds, int size)
168 {
169         HCRYPTPROV hCryptProv;
170
171         if (!CryptAcquireContext(&hCryptProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
172         {
173                 errno = GetLastError();
174                 log_err("CryptAcquireContext() failed: error %d\n", errno);
175                 return 1;
176         }
177
178         if (!CryptGenRandom(hCryptProv, size, (BYTE*)rand_seeds)) {
179                 errno = GetLastError();
180                 log_err("CryptGenRandom() failed, error %d\n", errno);
181                 CryptReleaseContext(hCryptProv, 0);
182                 return 1;
183         }
184
185         CryptReleaseContext(hCryptProv, 0);
186         return 0;
187 }
188
189 static inline int fio_set_sched_idle(void)
190 {
191         /* SetThreadPriority returns nonzero for success */
192         return (SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_IDLE))? 0 : -1;
193 }
194
195 #ifdef CONFIG_WINDOWS_XP
196 #include "os-windows-xp.h"
197 #else
198 #include "os-windows-7.h"
199 #endif
200
201 #endif /* FIO_OS_WINDOWS_H */