Generic endianness typo

[fio.git] / os / os-windows.h

diff --git a/os/os-windows.h b/os/os-windows.h
index f7712a1794f59a8fb8216222fd6f835ce2c52958..4af212d9d195a2a44e0f88bb7f9ce187bc14738c 100644 (file)
--- a/os/os-windows.h
+++ b/os/os-windows.h
@@ -4,24 +4,34 @@
 #include <sys/types.h>\r
 #include <errno.h>\r
 #include <windows.h>\r
+#include <psapi.h>\r
+#include <stdlib.h>\r
 \r
 #include "../smalloc.h"\r
 #include "../file.h"\r
 #include "../log.h"\r
 \r
 #define FIO_HAVE_ODIRECT\r
-#define FIO_USE_GENERIC_RAND\r
+#define FIO_HAVE_CPU_AFFINITY\r
 #define FIO_HAVE_CHARDEV_SIZE\r
-#define FIO_USE_GENERIC_RAND\r
-\r
 #define FIO_HAVE_FALLOCATE\r
 #define FIO_HAVE_FDATASYNC\r
 #define FIO_HAVE_WINDOWSAIO\r
+#define FIO_HAVE_GETTID\r
+\r
+#define FIO_USE_GENERIC_RAND\r
 \r
 #define OS_MAP_ANON            MAP_ANON\r
 \r
 #define OS_CLOCK CLOCK_REALTIME\r
 \r
+#define FIO_PREFERRED_ENGINE   "windowsaio"\r
+\r
+#define FIO_LITTLE_ENDIAN\r
+#define fio_swap16(x)  _byteswap_ushort(x)\r
+#define fio_swap32(x)  _byteswap_ulong(x)\r
+#define fio_swap64(x)  _byteswap_uint64(x)\r
+\r
 typedef off_t off64_t;\r
 \r
 typedef struct {\r
@@ -30,14 +40,16 @@ typedef struct {
 \r
 #define IOCTL_DISK_GET_LENGTH_INFO 0x7405C\r
 \r
+pid_t cygwin_winpid_to_pid(int winpid);\r
+\r
 static inline int blockdev_size(struct fio_file *f, unsigned long long *bytes)\r
 {\r
        int rc = 0;\r
        HANDLE hFile;\r
 \r
        if (f->hFile == NULL) {\r
-               hFile = CreateFile(f->file_name, (GENERIC_READ | GENERIC_WRITE),\r
-                       (FILE_SHARE_READ | FILE_SHARE_WRITE), NULL, OPEN_EXISTING, 0, NULL);\r
+               hFile = CreateFile(f->file_name, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE,\r
+                               NULL, OPEN_EXISTING, 0, NULL);\r
        } else {\r
                hFile = f->hFile;\r
        }\r
@@ -85,6 +97,117 @@ static inline void os_get_tmpdir(char *path, int len)
        GetTempPath(len, path);\r
 }\r
 \r
+typedef DWORD_PTR os_cpu_mask_t;\r
+\r
+static inline int gettid(void)\r
+{\r
+       return GetCurrentThreadId();\r
+}\r
+\r
+static inline int pid_to_winpid(int pid)\r
+{\r
+       int winpid = 0;\r
+       DWORD outbytes = 0;\r
+       DWORD *ids = NULL;\r
+       size_t allocsize;\r
+       \r
+       allocsize = sizeof(DWORD) * 1024;\r
+       \r
+       do {\r
+               if (allocsize == outbytes)\r
+                       allocsize *= 2;\r
+\r
+               ids = realloc(ids, allocsize);\r
+               EnumProcesses(ids, allocsize, &outbytes);\r
+       } while (allocsize == outbytes);\r
+       \r
+       for (int i = 0; i < (outbytes/sizeof(DWORD)); i++) {\r
+               if (cygwin_winpid_to_pid(ids[i]) == pid) {\r
+                       winpid = ids[i];\r
+                       break;\r
+               }\r
+       }\r
+       \r
+       free(ids);\r
+       return winpid;\r
+}\r
+\r
+HANDLE WINAPI OpenThread(\r
+    DWORD dwDesiredAccess,\r
+    BOOL bInheritHandle,\r
+    DWORD dwThreadId);\r
+    \r
+DWORD WINAPI GetProcessIdOfThread(HANDLE Thread);\r
+\r
+static inline int fio_setaffinity(int pid, os_cpu_mask_t cpumask)\r
+{\r
+       HANDLE h;\r
+       BOOL bSuccess;\r
+       int winpid;\r
+       \r
+       h = OpenThread(THREAD_QUERY_INFORMATION | THREAD_SET_INFORMATION, TRUE, pid);\r
+       if (h != NULL) {\r
+               bSuccess = SetThreadAffinityMask(h, cpumask);\r
+       } else {\r
+               // then we might have a process id instead of a thread id\r
+               winpid = pid_to_winpid(pid);\r
+               h = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_SET_INFORMATION, TRUE, winpid);\r
+               if (h == NULL)\r
+                       return -1;\r
+\r
+               bSuccess = SetProcessAffinityMask(h, cpumask);\r
+       }\r
+\r
+       CloseHandle(h);\r
+\r
+       return (bSuccess)? 0 : -1;\r
+}\r
+\r
+static inline void fio_getaffinity(int pid, os_cpu_mask_t *mask)\r
+{\r
+       os_cpu_mask_t systemMask;\r
+       int winpid;\r
+       \r
+       HANDLE h = OpenThread(THREAD_QUERY_INFORMATION, TRUE, pid);\r
+       if (h != NULL)\r
+               winpid = GetProcessIdOfThread(h);\r
+       else\r
+               winpid = pid_to_winpid(pid);\r
+       \r
+       h = OpenProcess(PROCESS_QUERY_INFORMATION, TRUE, winpid);\r
+\r
+       if (h != NULL) {\r
+               GetProcessAffinityMask(h, mask, &systemMask);\r
+               CloseHandle(h);\r
+       } else {\r
+               fprintf(stderr, "fio_getaffinity failed: failed to get handle for pid %d\n", pid);\r
+       }\r
+       \r
+}\r
+\r
+static inline void fio_cpu_clear(os_cpu_mask_t *mask, int cpu)\r
+{\r
+       *mask ^= 1 << (cpu-1);\r
+}\r
+\r
+static inline void fio_cpu_set(os_cpu_mask_t *mask, int cpu)\r
+{\r
+       *mask |= 1 << (cpu-1);\r
+}\r
+\r
+static inline int fio_cpuset_init(os_cpu_mask_t *mask)\r
+{\r
+       *mask = 0;\r
+       return 0;\r
+}\r
+\r
+static inline int fio_cpuset_exit(os_cpu_mask_t *mask)\r
+{\r
+       return 0;\r
+}\r
+\r
+#define FIO_MAX_CPUS                   MAXIMUM_PROCESSORS\r
+\r
 #ifdef MADV_FREE\r
 #define FIO_MADV_FREE  MADV_FREE\r
 #endif\r