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