Merge branch 'master' of https://github.com/ben-ihelputech/fio
[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 #include "os-windows-7.h"
25
26 #ifndef PTHREAD_STACK_MIN
27 #define PTHREAD_STACK_MIN 65535
28 #endif
29
30 #define FIO_HAVE_ODIRECT
31 #define FIO_HAVE_CPU_AFFINITY
32 #define FIO_HAVE_CHARDEV_SIZE
33 #define FIO_HAVE_GETTID
34 #define FIO_EMULATED_MKDIR_TWO
35
36 #define FIO_PREFERRED_ENGINE            "windowsaio"
37 #define FIO_PREFERRED_CLOCK_SOURCE      CS_CGETTIME
38 #define FIO_OS_PATH_SEPARATOR           '\\'
39
40 #define OS_MAP_ANON             MAP_ANON
41
42 #define fio_swap16(x)   _byteswap_ushort(x)
43 #define fio_swap32(x)   _byteswap_ulong(x)
44 #define fio_swap64(x)   _byteswap_uint64(x)
45
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 #ifndef S_ISSOCK
74 #define S_ISSOCK(x) 0
75 #endif
76
77 #define SIGCONT 0
78 #define SIGUSR1 1
79 #define SIGUSR2 2
80 #define SIGKILL 15 /* SIGKILL doesn't exists, let's use SIGTERM */
81
82 typedef int sigset_t;
83 typedef int siginfo_t;
84
85 struct 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
93 long sysconf(int name);
94
95 int kill(pid_t pid, int sig);
96 pid_t setsid(void);
97 int setgid(gid_t gid);
98 int setuid(uid_t uid);
99 int nice(int incr);
100 int sigaction(int sig, const struct sigaction *act,
101                 struct sigaction *oact);
102 int fsync(int fildes);
103 int fork(void);
104 int fcntl(int fildes, int cmd, ...);
105 int fdatasync(int fildes);
106 int lstat(const char * path, struct stat * buf);
107 uid_t geteuid(void);
108 char* ctime_r(const time_t *t, char *buf);
109 int nanosleep(const struct timespec *rqtp, struct timespec *rmtp);
110 ssize_t pread(int fildes, void *buf, size_t nbyte, off_t offset);
111 ssize_t pwrite(int fildes, const void *buf, size_t nbyte,
112                 off_t offset);
113
114 static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)
115 {
116         int rc = 0;
117         HANDLE hFile;
118         GET_LENGTH_INFORMATION info;
119         DWORD outBytes;
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
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
141 static inline int chardev_size(struct fio_file *f, unsigned long long *bytes)
142 {
143         return blockdev_size(f, bytes);
144 }
145
146 static inline int blockdev_invalidate_cache(struct fio_file *f)
147 {
148         return ENOTSUP;
149 }
150
151 static inline unsigned long long os_phys_mem(void)
152 {
153         long pagesize, pages;
154
155         pagesize = sysconf(_SC_PAGESIZE);
156         pages = sysconf(_SC_PHYS_PAGES);
157         if (pages == -1 || pagesize == -1)
158                 return 0;
159
160         return (unsigned long long) pages * (unsigned long long) pagesize;
161 }
162
163 #ifndef CONFIG_HAVE_GETTID
164 static inline int gettid(void)
165 {
166         return GetCurrentThreadId();
167 }
168 #endif
169
170 static inline int init_random_seeds(uint64_t *rand_seeds, int size)
171 {
172         HCRYPTPROV hCryptProv;
173
174         if (!CryptAcquireContext(&hCryptProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
175         {
176                 errno = GetLastError();
177                 log_err("CryptAcquireContext() failed: error %d\n", errno);
178                 return 1;
179         }
180
181         if (!CryptGenRandom(hCryptProv, size, (BYTE*)rand_seeds)) {
182                 errno = GetLastError();
183                 log_err("CryptGenRandom() failed, error %d\n", errno);
184                 CryptReleaseContext(hCryptProv, 0);
185                 return 1;
186         }
187
188         CryptReleaseContext(hCryptProv, 0);
189         return 0;
190 }
191
192 static inline int fio_set_sched_idle(void)
193 {
194         /* SetThreadPriority returns nonzero for success */
195         return (SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_IDLE))? 0 : -1;
196 }
197
198 static inline int fio_mkdir(const char *path, mode_t mode) {
199         DWORD dwAttr = GetFileAttributesA(path);
200
201         if (dwAttr != INVALID_FILE_ATTRIBUTES &&
202             (dwAttr & FILE_ATTRIBUTE_DIRECTORY)) {
203                 errno = EEXIST;
204                 return -1;
205         }
206
207         if (CreateDirectoryA(path, NULL) == 0) {
208                 /* Ignore errors if path is a device namespace */
209                 if (strcmp(path, "\\\\.") == 0) {
210                         errno = EEXIST;
211                         return -1;
212                 }
213                 errno = win_to_posix_error(GetLastError());
214                 return -1;
215         }
216
217         return 0;
218 }
219
220 #define FIO_HAVE_CPU_ONLINE_SYSCONF
221 unsigned int cpus_online(void);
222
223 int first_set_cpu(os_cpu_mask_t *cpumask);
224 int fio_setaffinity(int pid, os_cpu_mask_t cpumask);
225 int fio_cpuset_init(os_cpu_mask_t *mask);
226 int fio_getaffinity(int pid, os_cpu_mask_t *mask);
227 void fio_cpu_clear(os_cpu_mask_t *mask, int cpu);
228 void fio_cpu_set(os_cpu_mask_t *mask, int cpu);
229 int fio_cpu_isset(os_cpu_mask_t *mask, int cpu);
230 int fio_cpu_count(os_cpu_mask_t *mask);
231 int fio_cpuset_exit(os_cpu_mask_t *mask);
232
233 #endif /* FIO_OS_WINDOWS_H */