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

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