os/os-windows.h

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