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 #ifndef PTHREAD_STACK_MIN
  26 #define PTHREAD_STACK_MIN 65535
  27 #endif
  28
  29 #define FIO_HAVE_ODIRECT
  30 #define FIO_HAVE_CPU_AFFINITY
  31 #define FIO_HAVE_CHARDEV_SIZE
  32 #define FIO_HAVE_GETTID
  33 #define FIO_EMULATED_MKDIR_TWO
  34
  35 #define FIO_PREFERRED_ENGINE            "windowsaio"
  36 #define FIO_PREFERRED_CLOCK_SOURCE      CS_CGETTIME
  37 #define FIO_OS_PATH_SEPARATOR           '\\'
  38
  39 #define OS_MAP_ANON             MAP_ANON
  40
  41 #define fio_swap16(x)   _byteswap_ushort(x)
  42 #define fio_swap32(x)   _byteswap_ulong(x)
  43 #define fio_swap64(x)   _byteswap_uint64(x)
  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 #ifndef S_ISSOCK
  73 #define S_ISSOCK(x) 0
  74 #endif
  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 char* ctime_r(const time_t *t, char *buf);
 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
 112 static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)
 113 {
 114         int rc = 0;
 115         HANDLE hFile;
 116         GET_LENGTH_INFORMATION info;
 117         DWORD outBytes;
 118
 119         if (f->hFile == NULL) {
 120                 hFile = CreateFile(f->file_name, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE,
 121                                 NULL, OPEN_EXISTING, 0, NULL);
 122         } else {
 123                 hFile = f->hFile;
 124         }
 125
 126         if (DeviceIoControl(hFile, IOCTL_DISK_GET_LENGTH_INFO, NULL, 0, &info, sizeof(info), &outBytes, NULL))
 127                 *bytes = info.Length.QuadPart;
 128         else
 129                 rc = EIO;
 130
 131         /* If we were passed a POSIX fd,
 132          * close the HANDLE we created via CreateFile */
 133         if (hFile != INVALID_HANDLE_VALUE && f->hFile == NULL)
 134                 CloseHandle(hFile);
 135
 136         return rc;
 137 }
 138
 139 static inline int chardev_size(struct fio_file *f, unsigned long long *bytes)
 140 {
 141         return blockdev_size(f, bytes);
 142 }
 143
 144 static inline int blockdev_invalidate_cache(struct fio_file *f)
 145 {
 146         return ENOTSUP;
 147 }
 148
 149 static inline unsigned long long os_phys_mem(void)
 150 {
 151         long pagesize, pages;
 152
 153         pagesize = sysconf(_SC_PAGESIZE);
 154         pages = sysconf(_SC_PHYS_PAGES);
 155         if (pages == -1 || pagesize == -1)
 156                 return 0;
 157
 158         return (unsigned long long) pages * (unsigned long long) pagesize;
 159 }
 160
 161 #ifndef CONFIG_HAVE_GETTID
 162 static inline int gettid(void)
 163 {
 164         return GetCurrentThreadId();
 165 }
 166 #endif
 167
 168 static inline int init_random_seeds(uint64_t *rand_seeds, int size)
 169 {
 170         HCRYPTPROV hCryptProv;
 171
 172         if (!CryptAcquireContext(&hCryptProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
 173         {
 174                 errno = GetLastError();
 175                 log_err("CryptAcquireContext() failed: error %d\n", errno);
 176                 return 1;
 177         }
 178
 179         if (!CryptGenRandom(hCryptProv, size, (BYTE*)rand_seeds)) {
 180                 errno = GetLastError();
 181                 log_err("CryptGenRandom() failed, error %d\n", errno);
 182                 CryptReleaseContext(hCryptProv, 0);
 183                 return 1;
 184         }
 185
 186         CryptReleaseContext(hCryptProv, 0);
 187         return 0;
 188 }
 189
 190 static inline int fio_set_sched_idle(void)
 191 {
 192         /* SetThreadPriority returns nonzero for success */
 193         return (SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_IDLE))? 0 : -1;
 194 }
 195
 196 static inline int fio_mkdir(const char *path, mode_t mode) {
 197         DWORD dwAttr = GetFileAttributesA(path);
 198
 199         if (dwAttr != INVALID_FILE_ATTRIBUTES &&
 200             (dwAttr & FILE_ATTRIBUTE_DIRECTORY)) {
 201                 errno = EEXIST;
 202                 return -1;
 203         }
 204
 205         if (CreateDirectoryA(path, NULL) == 0) {
 206                 /* Ignore errors if path is a device namespace */
 207                 if (strcmp(path, "\\\\.") == 0) {
 208                         errno = EEXIST;
 209                         return -1;
 210                 }
 211                 errno = win_to_posix_error(GetLastError());
 212                 return -1;
 213         }
 214
 215         return 0;
 216 }
 217
 218 #ifdef CONFIG_WINDOWS_XP
 219 #include "os-windows-xp.h"
 220 #else
 221 #define FIO_HAVE_CPU_ONLINE_SYSCONF
 222 unsigned int cpus_online(void);
 223 #include "os-windows-7.h"
 224 #endif
 225
 226 int first_set_cpu(os_cpu_mask_t *cpumask);
 227 int fio_setaffinity(int pid, os_cpu_mask_t cpumask);
 228 int fio_cpuset_init(os_cpu_mask_t *mask);
 229 int fio_getaffinity(int pid, os_cpu_mask_t *mask);
 230 void fio_cpu_clear(os_cpu_mask_t *mask, int cpu);
 231 void fio_cpu_set(os_cpu_mask_t *mask, int cpu);
 232 int fio_cpu_isset(os_cpu_mask_t *mask, int cpu);
 233 int fio_cpu_count(os_cpu_mask_t *mask);
 234 int fio_cpuset_exit(os_cpu_mask_t *mask);
 235
 236 #endif /* FIO_OS_WINDOWS_H */