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