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 "../file.h"
  17 #include "../log.h"
  18
  19 #include "windows/posix.h"
  20
  21 #ifndef PTHREAD_STACK_MIN
  22 #define PTHREAD_STACK_MIN 65535
  23 #endif
  24
  25 #define FIO_HAVE_ODIRECT
  26 #define FIO_HAVE_CPU_AFFINITY
  27 #define FIO_HAVE_CHARDEV_SIZE
  28 #define FIO_HAVE_GETTID
  29 #define FIO_USE_GENERIC_RAND
  30
  31 #define FIO_PREFERRED_ENGINE            "windowsaio"
  32 #define FIO_PREFERRED_CLOCK_SOURCE      CS_CGETTIME
  33 #define FIO_OS_PATH_SEPARATOR           "\\"
  34
  35 #define FIO_MAX_CPUS    MAXIMUM_PROCESSORS
  36
  37 #define OS_MAP_ANON             MAP_ANON
  38
  39 #define fio_swap16(x)   _byteswap_ushort(x)
  40 #define fio_swap32(x)   _byteswap_ulong(x)
  41 #define fio_swap64(x)   _byteswap_uint64(x)
  42
  43 typedef DWORD_PTR os_cpu_mask_t;
  44
  45 #define _SC_PAGESIZE                    0x1
  46 #define _SC_NPROCESSORS_ONLN    0x2
  47 #define _SC_PHYS_PAGES                  0x4
  48
  49 #define SA_RESTART      0
  50 #define SIGPIPE         0
  51
  52 /*
  53  * Windows doesn't have O_DIRECT or O_SYNC, so define them
  54  * here so we can reject them at runtime when using the _open
  55  * interface (windowsaio uses CreateFile)
  56  */
  57 #define O_DIRECT        0x1000000
  58 #define O_SYNC          0x2000000
  59
  60 /* Windows doesn't support madvise, so any values will work */
  61 #define POSIX_MADV_DONTNEED             0
  62 #define POSIX_MADV_SEQUENTIAL   0
  63 #define POSIX_MADV_RANDOM               0
  64
  65 #define F_SETFL                 0x1
  66 #define F_GETFL                 0x2
  67 #define O_NONBLOCK              FIONBIO
  68
  69 /* Winsock doesn't support MSG_WAIT */
  70 #define OS_MSG_DONTWAIT 0
  71
  72 #define POLLOUT 1
  73 #define POLLIN  2
  74 #define POLLERR 0
  75 #define POLLHUP 1
  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 int nanosleep(const struct timespec *rqtp, struct timespec *rmtp);
 108 ssize_t pread(int fildes, void *buf, size_t nbyte, off_t offset);
 109 ssize_t pwrite(int fildes, const void *buf, size_t nbyte,
 110                 off_t offset);
 111 extern void td_fill_rand_seeds(struct thread_data *);
 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         /* 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         long pagesize, pages;
 155
 156         pagesize = sysconf(_SC_PAGESIZE);
 157         pages = sysconf(_SC_PHYS_PAGES);
 158         if (pages == -1 || pagesize == -1)
 159                 return 0;
 160
 161         return (unsigned long long) pages * (unsigned long long) pagesize;
 162 }
 163
 164 static inline void os_get_tmpdir(char *path, int len)
 165 {
 166         GetTempPath(len, path);
 167 }
 168
 169 static inline int gettid(void)
 170 {
 171         return GetCurrentThreadId();
 172 }
 173
 174 static inline int fio_setaffinity(int pid, os_cpu_mask_t cpumask)
 175 {
 176         HANDLE h;
 177         BOOL bSuccess = FALSE;
 178
 179         h = OpenThread(THREAD_QUERY_INFORMATION | THREAD_SET_INFORMATION, TRUE, pid);
 180         if (h != NULL) {
 181                 bSuccess = SetThreadAffinityMask(h, cpumask);
 182                 if (!bSuccess)
 183                         log_err("fio_setaffinity failed: failed to set thread affinity (pid %d, mask %.16llx)\n", pid, cpumask);
 184
 185                 CloseHandle(h);
 186         } else {
 187                 log_err("fio_setaffinity failed: failed to get handle for pid %d\n", pid);
 188         }
 189
 190         return (bSuccess)? 0 : -1;
 191 }
 192
 193 static inline void fio_getaffinity(int pid, os_cpu_mask_t *mask)
 194 {
 195         os_cpu_mask_t systemMask;
 196
 197         HANDLE h = OpenProcess(PROCESS_QUERY_INFORMATION, TRUE, pid);
 198
 199         if (h != NULL) {
 200                 GetProcessAffinityMask(h, mask, &systemMask);
 201                 CloseHandle(h);
 202         } else {
 203                 log_err("fio_getaffinity failed: failed to get handle for pid %d\n", pid);
 204         }
 205 }
 206
 207 static inline void fio_cpu_clear(os_cpu_mask_t *mask, int cpu)
 208 {
 209         *mask ^= 1 << (cpu-1);
 210 }
 211
 212 static inline void fio_cpu_set(os_cpu_mask_t *mask, int cpu)
 213 {
 214         *mask |= 1 << cpu;
 215 }
 216
 217 static inline int fio_cpu_count(os_cpu_mask_t *mask, int cpu)
 218 {
 219         return hweight64(*mask);
 220 }
 221
 222 static inline int fio_cpuset_init(os_cpu_mask_t *mask)
 223 {
 224         *mask = 0;
 225         return 0;
 226 }
 227
 228 static inline int fio_cpuset_exit(os_cpu_mask_t *mask)
 229 {
 230         return 0;
 231 }
 232
 233 static inline int init_random_state(struct thread_data *td, unsigned long *rand_seeds, int size)
 234 {
 235         HCRYPTPROV hCryptProv;
 236
 237         if (!CryptAcquireContext(&hCryptProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
 238         {
 239                 errno = GetLastError();
 240                 log_err("CryptAcquireContext() failed: error %d\n", errno);
 241                 return 1;
 242         }
 243
 244         if (!CryptGenRandom(hCryptProv, size, (BYTE*)rand_seeds)) {
 245                 errno = GetLastError();
 246                 log_err("CryptGenRandom() failed, error %d\n", errno);
 247                 CryptReleaseContext(hCryptProv, 0);
 248                 return 1;
 249         }
 250
 251         CryptReleaseContext(hCryptProv, 0);
 252         td_fill_rand_seeds(td);
 253         return 0;
 254 }
 255
 256
 257 static inline int fio_set_sched_idle(void)
 258 {
 259         /* SetThreadPriority returns nonzero for success */
 260         return (SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_IDLE))? 0 : -1;
 261 }
 262
 263
 264 #endif /* FIO_OS_WINDOWS_H */