[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_OS_HAVE_SOCKLEN_T\r
typedef int fio_socklen_t;\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
72ddbe79 BC	24	#define FIO_OS_HAVE_SOCKLEN_T\r
	25	typedef int fio_socklen_t;\r
	26	\r
0f7ac781 BC	27	#define FIO_USE_GENERIC_RAND\r
0f7ac781 BC	28	\r
03e20d68 BC	29	#define OS_MAP_ANON MAP_ANON\r
03e20d68 BC	30	\r
ecc314ba	31	#define OS_CLOCK CLOCK_REALTIME\r
03e20d68	32	\r
58483fa4 JA	33	#define FIO_PREFERRED_ENGINE "windowsaio"\r
58483fa4 JA	34	\r
232f9b73 JA	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
ecc314ba	40	typedef off_t off64_t;\r
03e20d68 BC	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
0f7ac781 BC	48	pid_t cygwin_winpid_to_pid(int winpid);\r
0f7ac781 BC	49	\r
ecc314ba	50	static inline int blockdev_size(struct fio_file f, unsigned long long bytes)\r
03e20d68 BC	51	{\r
03e20d68 BC	52	int rc = 0;\r
ecc314ba BC	53	HANDLE hFile;\r
	54	\r
	55	if (f->hFile == NULL) {\r
9b836561 BC	56	hFile = CreateFile(f->file_name, GENERIC_READ, FILE_SHARE_READ \| FILE_SHARE_WRITE,\r
9b836561 BC	57	NULL, OPEN_EXISTING, 0, NULL);\r
ecc314ba BC	58	} else {\r
ecc314ba BC	59	hFile = f->hFile;\r
03e20d68 BC	60	}\r
03e20d68 BC	61	\r
ecc314ba BC	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
ecc314ba BC	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
03e20d68 BC	77	}\r
03e20d68 BC	78	\r
ecc314ba	79	static inline int chardev_size(struct fio_file f, unsigned long long bytes)\r
03e20d68	80	{\r
ecc314ba	81	return blockdev_size(f, bytes);\r
03e20d68 BC	82	}\r
03e20d68 BC	83	\r
ecc314ba BC	84	static inline int blockdev_invalidate_cache(struct fio_file *f)\r
ecc314ba BC	85	{\r
e4db9fec BC	86	/* There's no way to invalidate the cache in Windows\r
e4db9fec BC	87	* so just pretend to succeed */\r
ecc314ba	88	return 0;\r
03e20d68 BC	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
0f7ac781 BC	105	typedef DWORD_PTR os_cpu_mask_t;\r
0f7ac781 BC	106	\r
1332757c BC	107	static inline int gettid(void)\r
	108	{\r
	109	return GetCurrentThreadId();\r
	110	}\r
	111	\r
0f7ac781 BC	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
1332757c BC	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
0f7ac781 BC	147	static inline int fio_setaffinity(int pid, os_cpu_mask_t cpumask)\r
0f7ac781 BC	148	{\r
1332757c BC	149	HANDLE h;\r
	150	BOOL bSuccess;\r
	151	int winpid;\r
0f7ac781	152	\r
1332757c BC	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
0f7ac781	166	CloseHandle(h);\r
1332757c	167	\r
0f7ac781 BC	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
1332757c BC	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
0f7ac781 BC	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
03e20d68 BC	216	#ifdef MADV_FREE\r
	217	#define FIO_MADV_FREE MADV_FREE\r
	218	#endif\r
	219	\r
03e20d68	220	#endif /* FIO_OS_WINDOWS_H */\r