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
  25 /* MinGW won't declare rand_r unless _POSIX is defined */
  26 #if defined(WIN32) && !defined(rand_r)
  27 int rand_r(unsigned *);
  28 #endif
  29
  30 #ifndef PTHREAD_STACK_MIN
  31 #define PTHREAD_STACK_MIN 65535
  32 #endif
  33
  34 #define FIO_HAVE_ODIRECT
  35 #define FIO_HAVE_CPU_AFFINITY
  36 #define FIO_HAVE_CHARDEV_SIZE
  37 #define FIO_HAVE_GETTID
  38
  39 #define FIO_PREFERRED_ENGINE            "windowsaio"
  40 #define FIO_PREFERRED_CLOCK_SOURCE      CS_CGETTIME
  41 #define FIO_OS_PATH_SEPARATOR           '\\'
  42
  43 #define OS_MAP_ANON             MAP_ANON
  44
  45 #define fio_swap16(x)   _byteswap_ushort(x)
  46 #define fio_swap32(x)   _byteswap_ulong(x)
  47 #define fio_swap64(x)   _byteswap_uint64(x)
  48
  49 #define _SC_PAGESIZE                    0x1
  50 #define _SC_NPROCESSORS_ONLN    0x2
  51 #define _SC_PHYS_PAGES                  0x4
  52
  53 #define SA_RESTART      0
  54 #define SIGPIPE         0
  55
  56 /*
  57  * Windows doesn't have O_DIRECT or O_SYNC, so define them
  58  * here so we can reject them at runtime when using the _open
  59  * interface (windowsaio uses CreateFile)
  60  */
  61 #define O_DIRECT        0x1000000
  62 #define O_SYNC          0x2000000
  63
  64 /* Windows doesn't support madvise, so any values will work */
  65 #define POSIX_MADV_DONTNEED             0
  66 #define POSIX_MADV_SEQUENTIAL   0
  67 #define POSIX_MADV_RANDOM               0
  68
  69 #define F_SETFL                 0x1
  70 #define F_GETFL                 0x2
  71 #define O_NONBLOCK              FIONBIO
  72
  73 /* Winsock doesn't support MSG_WAIT */
  74 #define OS_MSG_DONTWAIT 0
  75
  76 #ifndef S_ISSOCK
  77 #define S_ISSOCK(x) 0
  78 #endif
  79
  80 #define SIGCONT 0
  81 #define SIGUSR1 1
  82 #define SIGUSR2 2
  83
  84 typedef int sigset_t;
  85 typedef int siginfo_t;
  86
  87 struct sigaction
  88 {
  89         void (*sa_handler)(int);
  90         sigset_t sa_mask;
  91         int sa_flags;
  92         void* (*sa_sigaction)(int, siginfo_t *, void*);
  93 };
  94
  95 long sysconf(int name);
  96
  97 int kill(pid_t pid, int sig);
  98 pid_t setsid(void);
  99 int setgid(gid_t gid);
 100 int setuid(uid_t uid);
 101 int nice(int incr);
 102 int sigaction(int sig, const struct sigaction *act,
 103                 struct sigaction *oact);
 104 int fsync(int fildes);
 105 int fork(void);
 106 int fcntl(int fildes, int cmd, ...);
 107 int fdatasync(int fildes);
 108 int lstat(const char * path, struct stat * buf);
 109 uid_t geteuid(void);
 110 char* ctime_r(const time_t *t, char *buf);
 111 int nanosleep(const struct timespec *rqtp, struct timespec *rmtp);
 112 ssize_t pread(int fildes, void *buf, size_t nbyte, off_t offset);
 113 ssize_t pwrite(int fildes, const void *buf, size_t nbyte,
 114                 off_t offset);
 115
 116 static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)
 117 {
 118         int rc = 0;
 119         HANDLE hFile;
 120         GET_LENGTH_INFORMATION info;
 121         DWORD outBytes;
 122
 123         if (f->hFile == NULL) {
 124                 hFile = CreateFile(f->file_name, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE,
 125                                 NULL, OPEN_EXISTING, 0, NULL);
 126         } else {
 127                 hFile = f->hFile;
 128         }
 129
 130         if (DeviceIoControl(hFile, IOCTL_DISK_GET_LENGTH_INFO, NULL, 0, &info, sizeof(info), &outBytes, NULL))
 131                 *bytes = info.Length.QuadPart;
 132         else
 133                 rc = EIO;
 134
 135         /* If we were passed a POSIX fd,
 136          * close the HANDLE we created via CreateFile */
 137         if (hFile != INVALID_HANDLE_VALUE && f->hFile == NULL)
 138                 CloseHandle(hFile);
 139
 140         return rc;
 141 }
 142
 143 static inline int chardev_size(struct fio_file *f, unsigned long long *bytes)
 144 {
 145         return blockdev_size(f, bytes);
 146 }
 147
 148 static inline int blockdev_invalidate_cache(struct fio_file *f)
 149 {
 150         return ENOTSUP;
 151 }
 152
 153 static inline unsigned long long os_phys_mem(void)
 154 {
 155         long pagesize, pages;
 156
 157         pagesize = sysconf(_SC_PAGESIZE);
 158         pages = sysconf(_SC_PHYS_PAGES);
 159         if (pages == -1 || pagesize == -1)
 160                 return 0;
 161
 162         return (unsigned long long) pages * (unsigned long long) pagesize;
 163 }
 164
 165 #ifndef CONFIG_HAVE_GETTID
 166 static inline int gettid(void)
 167 {
 168         return GetCurrentThreadId();
 169 }
 170 #endif
 171
 172 static inline int init_random_seeds(uint64_t *rand_seeds, int size)
 173 {
 174         HCRYPTPROV hCryptProv;
 175
 176         if (!CryptAcquireContext(&hCryptProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
 177         {
 178                 errno = GetLastError();
 179                 log_err("CryptAcquireContext() failed: error %d\n", errno);
 180                 return 1;
 181         }
 182
 183         if (!CryptGenRandom(hCryptProv, size, (BYTE*)rand_seeds)) {
 184                 errno = GetLastError();
 185                 log_err("CryptGenRandom() failed, error %d\n", errno);
 186                 CryptReleaseContext(hCryptProv, 0);
 187                 return 1;
 188         }
 189
 190         CryptReleaseContext(hCryptProv, 0);
 191         return 0;
 192 }
 193
 194 static inline int fio_set_sched_idle(void)
 195 {
 196         /* SetThreadPriority returns nonzero for success */
 197         return (SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_IDLE))? 0 : -1;
 198 }
 199
 200 #ifdef CONFIG_WINDOWS_XP
 201 #include "os-windows-xp.h"
 202 #else
 203 #include "os-windows-7.h"
 204 #endif
 205
 206 #endif /* FIO_OS_WINDOWS_H */