os/os-windows.h

   1 #ifndef FIO_OS_WINDOWS_H\r
   2 #define FIO_OS_WINDOWS_H\r
   3 \r
   4 #include <sys/types.h>\r
   5 #include <errno.h>\r
   6 #include <windows.h>\r
   7 #include <psapi.h>\r
   8 #include <stdlib.h>\r
   9 \r
  10 #include "../smalloc.h"\r
  11 #include "../file.h"\r
  12 #include "../log.h"\r
  13 \r
  14 #define FIO_HAVE_ODIRECT\r
  15 #define FIO_HAVE_CPU_AFFINITY\r
  16 #define FIO_HAVE_CHARDEV_SIZE\r
  17 #define FIO_HAVE_FALLOCATE\r
  18 #define FIO_HAVE_FDATASYNC\r
  19 #define FIO_HAVE_WINDOWSAIO\r
  20 #define FIO_HAVE_GETTID\r
  21 \r
  22 #define FIO_USE_GENERIC_RAND\r
  23 \r
  24 #define OS_MAP_ANON             MAP_ANON\r
  25 \r
  26 #define OS_CLOCK CLOCK_REALTIME\r
  27 \r
  28 #define FIO_PREFERRED_ENGINE    "windowsaio"\r
  29 \r
  30 #define FIO_LITTLE_ENDIAN\r
  31 #define fio_swap16(x)   _byteswap_ushort(x)\r
  32 #define fio_swap32(x)   _byteswap_ulong(x)\r
  33 #define fio_swap64(x)   _byteswap_uint64(x)\r
  34 \r
  35 typedef off_t off64_t;\r
  36 \r
  37 typedef struct {\r
  38   LARGE_INTEGER Length;\r
  39 } GET_LENGTH_INFORMATION;\r
  40 \r
  41 #define IOCTL_DISK_GET_LENGTH_INFO 0x7405C\r
  42 \r
  43 pid_t cygwin_winpid_to_pid(int winpid);\r
  44 \r
  45 static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)\r
  46 {\r
  47         int rc = 0;\r
  48         HANDLE hFile;\r
  49 \r
  50         if (f->hFile == NULL) {\r
  51                 hFile = CreateFile(f->file_name, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE,\r
  52                                 NULL, OPEN_EXISTING, 0, NULL);\r
  53         } else {\r
  54                 hFile = f->hFile;\r
  55         }\r
  56 \r
  57         GET_LENGTH_INFORMATION info;\r
  58         DWORD outBytes;\r
  59         LARGE_INTEGER size;\r
  60         size.QuadPart = 0;\r
  61         if (DeviceIoControl(hFile, IOCTL_DISK_GET_LENGTH_INFO, NULL, 0, &info, sizeof(info), &outBytes, NULL))\r
  62                 *bytes = info.Length.QuadPart;\r
  63         else\r
  64                 rc = EIO;\r
  65 \r
  66         /* If we were passed a POSIX fd,\r
  67          * close the HANDLE we created via CreateFile */\r
  68         if (hFile != INVALID_HANDLE_VALUE && f->hFile == NULL)\r
  69                 CloseHandle(hFile);\r
  70 \r
  71         return rc;\r
  72 }\r
  73 \r
  74 static inline int chardev_size(struct fio_file *f, unsigned long long *bytes)\r
  75 {\r
  76         return blockdev_size(f, bytes);\r
  77 }\r
  78 \r
  79 static inline int blockdev_invalidate_cache(struct fio_file *f)\r
  80 {\r
  81         /* There's no way to invalidate the cache in Windows\r
  82          * so just pretend to succeed */\r
  83         return 0;\r
  84 }\r
  85 \r
  86 static inline unsigned long long os_phys_mem(void)\r
  87 {\r
  88         SYSTEM_INFO sysInfo;\r
  89         unsigned long addr;\r
  90         GetSystemInfo(&sysInfo);\r
  91         addr = (unsigned long)sysInfo.lpMaximumApplicationAddress;\r
  92         return addr;\r
  93 }\r
  94 \r
  95 static inline void os_get_tmpdir(char *path, int len)\r
  96 {\r
  97         GetTempPath(len, path);\r
  98 }\r
  99 \r
 100 typedef DWORD_PTR os_cpu_mask_t;\r
 101 \r
 102 static inline int gettid(void)\r
 103 {\r
 104         return GetCurrentThreadId();\r
 105 }\r
 106 \r
 107 static inline int pid_to_winpid(int pid)\r
 108 {\r
 109         int winpid = 0;\r
 110         DWORD outbytes = 0;\r
 111         DWORD *ids = NULL;\r
 112         size_t allocsize;\r
 113         \r
 114         allocsize = sizeof(DWORD) * 1024;\r
 115         \r
 116         do {\r
 117                 if (allocsize == outbytes)\r
 118                         allocsize *= 2;\r
 119 \r
 120                 ids = realloc(ids, allocsize);\r
 121                 EnumProcesses(ids, allocsize, &outbytes);\r
 122         } while (allocsize == outbytes);\r
 123         \r
 124         for (int i = 0; i < (outbytes/sizeof(DWORD)); i++) {\r
 125                 if (cygwin_winpid_to_pid(ids[i]) == pid) {\r
 126                         winpid = ids[i];\r
 127                         break;\r
 128                 }\r
 129         }\r
 130         \r
 131         free(ids);\r
 132         return winpid;\r
 133 }\r
 134 \r
 135 HANDLE WINAPI OpenThread(\r
 136     DWORD dwDesiredAccess,\r
 137     BOOL bInheritHandle,\r
 138     DWORD dwThreadId);\r
 139     \r
 140 DWORD WINAPI GetProcessIdOfThread(HANDLE Thread);\r
 141 \r
 142 static inline int fio_setaffinity(int pid, os_cpu_mask_t cpumask)\r
 143 {\r
 144         HANDLE h;\r
 145         BOOL bSuccess;\r
 146         int winpid;\r
 147         \r
 148         h = OpenThread(THREAD_QUERY_INFORMATION | THREAD_SET_INFORMATION, TRUE, pid);\r
 149         if (h != NULL) {\r
 150                 bSuccess = SetThreadAffinityMask(h, cpumask);\r
 151         } else {\r
 152                 // then we might have a process id instead of a thread id\r
 153                 winpid = pid_to_winpid(pid);\r
 154                 h = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_SET_INFORMATION, TRUE, winpid);\r
 155                 if (h == NULL)\r
 156                         return -1;\r
 157 \r
 158                 bSuccess = SetProcessAffinityMask(h, cpumask);\r
 159         }\r
 160 \r
 161         CloseHandle(h);\r
 162 \r
 163         return (bSuccess)? 0 : -1;\r
 164 }\r
 165 \r
 166 static inline void fio_getaffinity(int pid, os_cpu_mask_t *mask)\r
 167 {\r
 168         os_cpu_mask_t systemMask;\r
 169         int winpid;\r
 170         \r
 171         HANDLE h = OpenThread(THREAD_QUERY_INFORMATION, TRUE, pid);\r
 172         if (h != NULL)\r
 173                 winpid = GetProcessIdOfThread(h);\r
 174         else\r
 175                 winpid = pid_to_winpid(pid);\r
 176         \r
 177         h = OpenProcess(PROCESS_QUERY_INFORMATION, TRUE, winpid);\r
 178 \r
 179         if (h != NULL) {\r
 180                 GetProcessAffinityMask(h, mask, &systemMask);\r
 181                 CloseHandle(h);\r
 182         } else {\r
 183                 fprintf(stderr, "fio_getaffinity failed: failed to get handle for pid %d\n", pid);\r
 184         }\r
 185         \r
 186 }\r
 187 \r
 188 static inline void fio_cpu_clear(os_cpu_mask_t *mask, int cpu)\r
 189 {\r
 190         *mask ^= 1 << (cpu-1);\r
 191 }\r
 192 \r
 193 static inline void fio_cpu_set(os_cpu_mask_t *mask, int cpu)\r
 194 {\r
 195         *mask |= 1 << (cpu-1);\r
 196 }\r
 197 \r
 198 static inline int fio_cpuset_init(os_cpu_mask_t *mask)\r
 199 {\r
 200         *mask = 0;\r
 201         return 0;\r
 202 }\r
 203 \r
 204 static inline int fio_cpuset_exit(os_cpu_mask_t *mask)\r
 205 {\r
 206         return 0;\r
 207 }\r
 208 \r
 209 #define FIO_MAX_CPUS                    MAXIMUM_PROCESSORS\r
 210 \r
 211 #ifdef MADV_FREE\r
 212 #define FIO_MADV_FREE   MADV_FREE\r
 213 #endif\r
 214 \r
 215 #endif /* FIO_OS_WINDOWS_H */\r