X-Git-Url: https://git.kernel.dk/?p=fio.git;a=blobdiff_plain;f=os%2Fos-windows.h;h=4af212d9d195a2a44e0f88bb7f9ce187bc14738c;hp=f7712a1794f59a8fb8216222fd6f835ce2c52958;hb=9677bec37d1b9d2949116cdc3d691a4d3e534cbe;hpb=63a582671fe6db03f0a0be81843f2b72bbc8ab33

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