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 #include "../lib/hweight.h"
  19
  20 #include "windows/posix.h"
  21
  22 #ifndef PTHREAD_STACK_MIN
  23 #define PTHREAD_STACK_MIN 65535
  24 #endif
  25
  26 #define FIO_HAVE_ODIRECT
  27 #define FIO_HAVE_CPU_AFFINITY
  28 #define FIO_HAVE_CHARDEV_SIZE
  29 #define FIO_HAVE_GETTID
  30 #define FIO_USE_GENERIC_RAND
  31
  32 #define FIO_PREFERRED_ENGINE            "windowsaio"
  33 #define FIO_PREFERRED_CLOCK_SOURCE      CS_CGETTIME
  34 #define FIO_OS_PATH_SEPARATOR           "\\"
  35
  36 #define FIO_MAX_CPUS    MAXIMUM_PROCESSORS
  37
  38 #define OS_MAP_ANON             MAP_ANON
  39
  40 #define fio_swap16(x)   _byteswap_ushort(x)
  41 #define fio_swap32(x)   _byteswap_ulong(x)
  42 #define fio_swap64(x)   _byteswap_uint64(x)
  43
  44 typedef DWORD_PTR os_cpu_mask_t;
  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 #define POLLOUT 1
  74 #define POLLIN  2
  75 #define POLLERR 0
  76 #define POLLHUP 1
  77
  78 #define SIGCONT 0
  79 #define SIGUSR1 1
  80 #define SIGUSR2 2
  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 extern void td_fill_rand_seeds(struct thread_data *);
 114
 115 static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)
 116 {
 117         int rc = 0;
 118         HANDLE hFile;
 119         GET_LENGTH_INFORMATION info;
 120         DWORD outBytes;
 121
 122         if (f->hFile == NULL) {
 123                 hFile = CreateFile(f->file_name, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE,
 124                                 NULL, OPEN_EXISTING, 0, NULL);
 125         } else {
 126                 hFile = f->hFile;
 127         }
 128
 129         if (DeviceIoControl(hFile, IOCTL_DISK_GET_LENGTH_INFO, NULL, 0, &info, sizeof(info), &outBytes, NULL))
 130                 *bytes = info.Length.QuadPart;
 131         else
 132                 rc = EIO;
 133
 134         /* If we were passed a POSIX fd,
 135          * close the HANDLE we created via CreateFile */
 136         if (hFile != INVALID_HANDLE_VALUE && f->hFile == NULL)
 137                 CloseHandle(hFile);
 138
 139         return rc;
 140 }
 141
 142 static inline int chardev_size(struct fio_file *f, unsigned long long *bytes)
 143 {
 144         return blockdev_size(f, bytes);
 145 }
 146
 147 static inline int blockdev_invalidate_cache(struct fio_file *f)
 148 {
 149         /* There's no way to invalidate the cache in Windows
 150          * so just pretend to succeed */
 151         return 0;
 152 }
 153
 154 static inline unsigned long long os_phys_mem(void)
 155 {
 156         long pagesize, pages;
 157
 158         pagesize = sysconf(_SC_PAGESIZE);
 159         pages = sysconf(_SC_PHYS_PAGES);
 160         if (pages == -1 || pagesize == -1)
 161                 return 0;
 162
 163         return (unsigned long long) pages * (unsigned long long) pagesize;
 164 }
 165
 166 static inline int gettid(void)
 167 {
 168         return GetCurrentThreadId();
 169 }
 170
 171 static inline int fio_setaffinity(int pid, os_cpu_mask_t cpumask)
 172 {
 173         HANDLE h;
 174         BOOL bSuccess = FALSE;
 175
 176         h = OpenThread(THREAD_QUERY_INFORMATION | THREAD_SET_INFORMATION, TRUE, pid);
 177         if (h != NULL) {
 178                 bSuccess = SetThreadAffinityMask(h, cpumask);
 179                 if (!bSuccess)
 180                         log_err("fio_setaffinity failed: failed to set thread affinity (pid %d, mask %.16llx)\n", pid, cpumask);
 181
 182                 CloseHandle(h);
 183         } else {
 184                 log_err("fio_setaffinity failed: failed to get handle for pid %d\n", pid);
 185         }
 186
 187         return (bSuccess)? 0 : -1;
 188 }
 189
 190 static inline void fio_getaffinity(int pid, os_cpu_mask_t *mask)
 191 {
 192         os_cpu_mask_t systemMask;
 193
 194         HANDLE h = OpenProcess(PROCESS_QUERY_INFORMATION, TRUE, pid);
 195
 196         if (h != NULL) {
 197                 GetProcessAffinityMask(h, mask, &systemMask);
 198                 CloseHandle(h);
 199         } else {
 200                 log_err("fio_getaffinity failed: failed to get handle for pid %d\n", pid);
 201         }
 202 }
 203
 204 static inline void fio_cpu_clear(os_cpu_mask_t *mask, int cpu)
 205 {
 206         *mask ^= 1 << (cpu-1);
 207 }
 208
 209 static inline void fio_cpu_set(os_cpu_mask_t *mask, int cpu)
 210 {
 211         *mask |= 1 << cpu;
 212 }
 213
 214 static inline int fio_cpu_isset(os_cpu_mask_t *mask, int cpu)
 215 {
 216         return (*mask & (1U << cpu));
 217 }
 218
 219 static inline int fio_cpu_count(os_cpu_mask_t *mask)
 220 {
 221         return hweight64(*mask);
 222 }
 223
 224 static inline int fio_cpuset_init(os_cpu_mask_t *mask)
 225 {
 226         *mask = 0;
 227         return 0;
 228 }
 229
 230 static inline int fio_cpuset_exit(os_cpu_mask_t *mask)
 231 {
 232         return 0;
 233 }
 234
 235 static inline int init_random_state(struct thread_data *td, unsigned long *rand_seeds, int size)
 236 {
 237         HCRYPTPROV hCryptProv;
 238
 239         if (!CryptAcquireContext(&hCryptProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
 240         {
 241                 errno = GetLastError();
 242                 log_err("CryptAcquireContext() failed: error %d\n", errno);
 243                 return 1;
 244         }
 245
 246         if (!CryptGenRandom(hCryptProv, size, (BYTE*)rand_seeds)) {
 247                 errno = GetLastError();
 248                 log_err("CryptGenRandom() failed, error %d\n", errno);
 249                 CryptReleaseContext(hCryptProv, 0);
 250                 return 1;
 251         }
 252
 253         CryptReleaseContext(hCryptProv, 0);
 254         td_fill_rand_seeds(td);
 255         return 0;
 256 }
 257
 258
 259 static inline int fio_set_sched_idle(void)
 260 {
 261         /* SetThreadPriority returns nonzero for success */
 262         return (SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_IDLE))? 0 : -1;
 263 }
 264
 265
 266 #endif /* FIO_OS_WINDOWS_H */