2 * Unit test suite for PSAPI
4 * Copyright (C) 2005 Felix Nawothnig
5 * Copyright (C) 2012 Dmitry Timoshkov
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
24 /* 0x0600 makes PROCESS_ALL_ACCESS not compatible with pre-Vista versions */
25 #define _WIN32_WINNT 0x0500
28 #define WIN32_NO_STATUS
37 #include "wine/test.h"
39 #define PSAPI_GET_PROC(func) \
40 p ## func = (void*)GetProcAddress(hpsapi, #func); \
42 ok(0, "GetProcAddress(%s) failed\n", #func); \
43 FreeLibrary(hpsapi); \
47 static BOOL (WINAPI
*pEmptyWorkingSet
)(HANDLE
);
48 static BOOL (WINAPI
*pEnumProcesses
)(DWORD
*, DWORD
, DWORD
*);
49 static BOOL (WINAPI
*pEnumProcessModules
)(HANDLE
, HMODULE
*, DWORD
, LPDWORD
);
50 static DWORD (WINAPI
*pGetModuleBaseNameA
)(HANDLE
, HMODULE
, LPSTR
, DWORD
);
51 static DWORD (WINAPI
*pGetModuleFileNameExA
)(HANDLE
, HMODULE
, LPSTR
, DWORD
);
52 static BOOL (WINAPI
*pGetModuleInformation
)(HANDLE
, HMODULE
, LPMODULEINFO
, DWORD
);
53 static DWORD (WINAPI
*pGetMappedFileNameA
)(HANDLE
, LPVOID
, LPSTR
, DWORD
);
54 static DWORD (WINAPI
*pGetMappedFileNameW
)(HANDLE
, LPVOID
, LPWSTR
, DWORD
);
55 static DWORD (WINAPI
*pGetProcessImageFileNameA
)(HANDLE
, LPSTR
, DWORD
);
56 static DWORD (WINAPI
*pGetProcessImageFileNameW
)(HANDLE
, LPWSTR
, DWORD
);
57 static BOOL (WINAPI
*pGetProcessMemoryInfo
)(HANDLE
, PPROCESS_MEMORY_COUNTERS
, DWORD
);
58 static BOOL (WINAPI
*pGetWsChanges
)(HANDLE
, PPSAPI_WS_WATCH_INFORMATION
, DWORD
);
59 static BOOL (WINAPI
*pInitializeProcessForWsWatch
)(HANDLE
);
60 static BOOL (WINAPI
*pQueryWorkingSet
)(HANDLE
, PVOID
, DWORD
);
61 static NTSTATUS (WINAPI
*pNtQueryVirtualMemory
)(HANDLE
, LPCVOID
, ULONG
, PVOID
, SIZE_T
, SIZE_T
*);
63 static BOOL
InitFunctionPtrs(HMODULE hpsapi
)
65 PSAPI_GET_PROC(EmptyWorkingSet
);
66 PSAPI_GET_PROC(EnumProcessModules
);
67 PSAPI_GET_PROC(EnumProcesses
);
68 PSAPI_GET_PROC(GetModuleBaseNameA
);
69 PSAPI_GET_PROC(GetModuleFileNameExA
);
70 PSAPI_GET_PROC(GetModuleInformation
);
71 PSAPI_GET_PROC(GetMappedFileNameA
);
72 PSAPI_GET_PROC(GetMappedFileNameW
);
73 PSAPI_GET_PROC(GetProcessMemoryInfo
);
74 PSAPI_GET_PROC(GetWsChanges
);
75 PSAPI_GET_PROC(InitializeProcessForWsWatch
);
76 PSAPI_GET_PROC(QueryWorkingSet
);
77 /* GetProcessImageFileName is not exported on NT4 */
78 pGetProcessImageFileNameA
=
79 (void *)GetProcAddress(hpsapi
, "GetProcessImageFileNameA");
80 pGetProcessImageFileNameW
=
81 (void *)GetProcAddress(hpsapi
, "GetProcessImageFileNameW");
82 pNtQueryVirtualMemory
= (void *)GetProcAddress(GetModuleHandle("ntdll.dll"), "NtQueryVirtualMemory");
86 static HANDLE hpSR
, hpQI
, hpVR
, hpQV
, hpAA
;
87 static const HANDLE hBad
= (HANDLE
)0xdeadbeef;
89 static void test_EnumProcesses(void)
91 DWORD pid
, ret
, cbUsed
= 0xdeadbeef;
93 SetLastError(0xdeadbeef);
94 ret
= pEnumProcesses(NULL
, 0, &cbUsed
);
95 ok(ret
== 1, "failed with %d\n", GetLastError());
96 ok(cbUsed
== 0, "cbUsed=%d\n", cbUsed
);
98 SetLastError(0xdeadbeef);
99 ret
= pEnumProcesses(&pid
, 4, &cbUsed
);
100 ok(ret
== 1, "failed with %d\n", GetLastError());
101 ok(cbUsed
== 4, "cbUsed=%d\n", cbUsed
);
104 static void test_EnumProcessModules(void)
106 HMODULE hMod
= GetModuleHandle(NULL
);
107 DWORD ret
, cbNeeded
= 0xdeadbeef;
109 SetLastError(0xdeadbeef);
110 pEnumProcessModules(NULL
, NULL
, 0, &cbNeeded
);
111 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
113 SetLastError(0xdeadbeef);
114 pEnumProcessModules(hpQI
, NULL
, 0, &cbNeeded
);
115 ok(GetLastError() == ERROR_ACCESS_DENIED
, "expected error=ERROR_ACCESS_DENIED but got %d\n", GetLastError());
117 SetLastError(0xdeadbeef);
118 ret
= pEnumProcessModules(hpQV
, NULL
, 0, &cbNeeded
);
119 ok(ret
== 1, "failed with %d\n", GetLastError());
121 SetLastError(0xdeadbeef);
122 ret
= pEnumProcessModules(hpQV
, &hMod
, sizeof(HMODULE
), &cbNeeded
);
125 ok(hMod
== GetModuleHandle(NULL
),
126 "hMod=%p GetModuleHandle(NULL)=%p\n", hMod
, GetModuleHandle(NULL
));
127 ok(cbNeeded
% sizeof(hMod
) == 0, "not a multiple of sizeof(HMODULE) cbNeeded=%d\n", cbNeeded
);
128 /* Windows sometimes has a bunch of extra dlls, presumably brought in by
131 if (cbNeeded
< 4 * sizeof(HMODULE
) || cbNeeded
> 30 * sizeof(HMODULE
))
135 ok(0, "cbNeeded=%d\n", cbNeeded
);
137 pEnumProcessModules(hpQV
, hmods
, sizeof(hmods
), &cbNeeded
);
138 for (i
= 0 ; i
< cbNeeded
/sizeof(*hmods
); i
++)
141 GetModuleFileNameA(hmods
[i
], path
, sizeof(path
));
142 trace("i=%d hmod=%p path=[%s]\n", i
, hmods
[i
], path
);
147 static void test_GetModuleInformation(void)
149 HMODULE hMod
= GetModuleHandle(NULL
);
153 SetLastError(0xdeadbeef);
154 pGetModuleInformation(NULL
, hMod
, &info
, sizeof(info
));
155 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
157 SetLastError(0xdeadbeef);
158 pGetModuleInformation(hpQI
, hMod
, &info
, sizeof(info
));
159 ok(GetLastError() == ERROR_ACCESS_DENIED
, "expected error=ERROR_ACCESS_DENIED but got %d\n", GetLastError());
161 SetLastError(0xdeadbeef);
162 pGetModuleInformation(hpQV
, hBad
, &info
, sizeof(info
));
163 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
165 SetLastError(0xdeadbeef);
166 pGetModuleInformation(hpQV
, hMod
, &info
, sizeof(info
)-1);
167 ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER
, "expected error=ERROR_INSUFFICIENT_BUFFER but got %d\n", GetLastError());
169 SetLastError(0xdeadbeef);
170 ret
= pGetModuleInformation(hpQV
, hMod
, &info
, sizeof(info
));
171 ok(ret
== 1, "failed with %d\n", GetLastError());
172 ok(info
.lpBaseOfDll
== hMod
, "lpBaseOfDll=%p hMod=%p\n", info
.lpBaseOfDll
, hMod
);
175 static void test_GetProcessMemoryInfo(void)
177 PROCESS_MEMORY_COUNTERS pmc
;
180 SetLastError(0xdeadbeef);
181 ret
= pGetProcessMemoryInfo(NULL
, &pmc
, sizeof(pmc
));
182 ok(!ret
, "GetProcessMemoryInfo should fail\n");
183 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
185 SetLastError(0xdeadbeef);
186 ret
= pGetProcessMemoryInfo(hpSR
, &pmc
, sizeof(pmc
));
188 ok(!ret
, "GetProcessMemoryInfo should fail\n");
190 ok(GetLastError() == ERROR_ACCESS_DENIED
, "expected error=ERROR_ACCESS_DENIED but got %d\n", GetLastError());
192 SetLastError(0xdeadbeef);
193 ret
= pGetProcessMemoryInfo(hpQI
, &pmc
, sizeof(pmc
)-1);
194 ok(!ret
, "GetProcessMemoryInfo should fail\n");
195 ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER
, "expected error=ERROR_INSUFFICIENT_BUFFER but got %d\n", GetLastError());
197 SetLastError(0xdeadbeef);
198 ret
= pGetProcessMemoryInfo(hpQI
, &pmc
, sizeof(pmc
));
199 ok(ret
== 1, "failed with %d\n", GetLastError());
202 static BOOL
nt_get_mapped_file_name(HANDLE process
, LPVOID addr
, LPWSTR name
, DWORD len
)
204 MEMORY_SECTION_NAME
*section_name
;
206 SIZE_T buf_len
, ret_len
;
209 if (!pNtQueryVirtualMemory
) return FALSE
;
211 buf_len
= len
* sizeof(WCHAR
) + sizeof(MEMORY_SECTION_NAME
);
212 buf
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, buf_len
);
214 ret_len
= 0xdeadbeef;
215 status
= pNtQueryVirtualMemory(process
, addr
, MemorySectionName
, buf
, buf_len
, &ret_len
);
217 ok(!status
, "NtQueryVirtualMemory error %x\n", status
);
218 /* FIXME: remove once Wine is fixed */
219 if (status
) return FALSE
;
221 section_name
= (MEMORY_SECTION_NAME
*)buf
;
222 ok(ret_len
== section_name
->SectionFileName
.MaximumLength
+ sizeof(*section_name
), "got %lu, %u\n",
223 ret_len
, section_name
->SectionFileName
.MaximumLength
);
224 ok((char *)section_name
->SectionFileName
.Buffer
== (char *)section_name
+ sizeof(*section_name
), "got %p, %p\n",
225 section_name
, section_name
->SectionFileName
.Buffer
);
226 ok(section_name
->SectionFileName
.MaximumLength
== section_name
->SectionFileName
.Length
+ sizeof(WCHAR
), "got %u, %u\n",
227 section_name
->SectionFileName
.MaximumLength
, section_name
->SectionFileName
.Length
);
228 ok(section_name
->SectionFileName
.Length
== lstrlenW(section_name
->SectionFileName
.Buffer
) * sizeof(WCHAR
), "got %u, %u\n",
229 section_name
->SectionFileName
.Length
, lstrlenW(section_name
->SectionFileName
.Buffer
));
231 memcpy(name
, section_name
->SectionFileName
.Buffer
, section_name
->SectionFileName
.MaximumLength
);
232 HeapFree(GetProcessHeap(), 0, buf
);
236 static void test_GetMappedFileName(void)
238 HMODULE hMod
= GetModuleHandle(NULL
);
239 char szMapPath
[MAX_PATH
], szModPath
[MAX_PATH
], *szMapBaseName
;
242 char temp_path
[MAX_PATH
], file_name
[MAX_PATH
], map_name
[MAX_PATH
], device_name
[MAX_PATH
], drive
[3];
243 WCHAR map_nameW
[MAX_PATH
], nt_map_name
[MAX_PATH
];
246 SetLastError(0xdeadbeef);
247 ret
= pGetMappedFileNameA(NULL
, hMod
, szMapPath
, sizeof(szMapPath
));
248 ok(!ret
, "GetMappedFileName should fail\n");
250 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
252 SetLastError(0xdeadbeef);
253 ret
= pGetMappedFileNameA(hpSR
, hMod
, szMapPath
, sizeof(szMapPath
));
254 ok(!ret
, "GetMappedFileName should fail\n");
256 ok(GetLastError() == ERROR_ACCESS_DENIED
, "expected error=ERROR_ACCESS_DENIED but got %d\n", GetLastError());
258 SetLastError( 0xdeadbeef );
259 ret
= pGetMappedFileNameA(hpQI
, hMod
, szMapPath
, sizeof(szMapPath
));
261 ok( ret
|| broken(GetLastError() == ERROR_UNEXP_NET_ERR
), /* win2k */
262 "GetMappedFileNameA failed with error %u\n", GetLastError() );
265 ok(ret
== strlen(szMapPath
), "szMapPath=\"%s\" ret=%d\n", szMapPath
, ret
);
267 ok(szMapPath
[0] == '\\', "szMapPath=\"%s\"\n", szMapPath
);
268 szMapBaseName
= strrchr(szMapPath
, '\\'); /* That's close enough for us */
270 ok(szMapBaseName
&& *szMapBaseName
, "szMapPath=\"%s\"\n", szMapPath
);
273 GetModuleFileNameA(NULL
, szModPath
, sizeof(szModPath
));
274 ok(!strcmp(strrchr(szModPath
, '\\'), szMapBaseName
),
275 "szModPath=\"%s\" szMapBaseName=\"%s\"\n", szModPath
, szMapBaseName
);
279 GetTempPath(MAX_PATH
, temp_path
);
280 GetTempFileName(temp_path
, "map", 0, file_name
);
282 drive
[0] = file_name
[0];
285 SetLastError(0xdeadbeef);
286 ret
= QueryDosDevice(drive
, device_name
, sizeof(device_name
));
287 ok(ret
, "QueryDosDevice error %d\n", GetLastError());
288 trace("%s -> %s\n", drive
, device_name
);
290 SetLastError(0xdeadbeef);
291 hfile
= CreateFile(file_name
, GENERIC_READ
| GENERIC_WRITE
, 0, NULL
, CREATE_ALWAYS
, 0, 0);
292 ok(hfile
!= INVALID_HANDLE_VALUE
, "CreateFile(%s) error %d\n", file_name
, GetLastError());
293 SetFilePointer(hfile
, 0x4000, NULL
, FILE_BEGIN
);
296 SetLastError(0xdeadbeef);
297 hmap
= CreateFileMapping(hfile
, NULL
, PAGE_READONLY
| SEC_COMMIT
, 0, 0, NULL
);
298 ok(hmap
!= 0, "CreateFileMapping error %d\n", GetLastError());
300 SetLastError(0xdeadbeef);
301 base
= MapViewOfFile(hmap
, FILE_MAP_READ
, 0, 0, 0);
302 ok(base
!= NULL
, "MapViewOfFile error %d\n", GetLastError());
304 SetLastError(0xdeadbeef);
305 ret
= pGetMappedFileNameA(GetCurrentProcess(), base
, map_name
, 0);
306 ok(!ret
, "GetMappedFileName should fail\n");
308 ok(GetLastError() == ERROR_INVALID_PARAMETER
, "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
310 SetLastError(0xdeadbeef);
311 ret
= pGetMappedFileNameA(GetCurrentProcess(), base
, 0, sizeof(map_name
));
312 ok(!ret
, "GetMappedFileName should fail\n");
314 ok(GetLastError() == ERROR_INVALID_PARAMETER
, "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
316 SetLastError(0xdeadbeef);
317 ret
= pGetMappedFileNameA(GetCurrentProcess(), base
, map_name
, 1);
319 ok(ret
== 1, "GetMappedFileName error %d\n", GetLastError());
320 ok(!map_name
[0] || broken(map_name
[0] == device_name
[0]) /* before win2k */, "expected 0, got %c\n", map_name
[0]);
322 SetLastError(0xdeadbeef);
323 ret
= pGetMappedFileNameA(GetCurrentProcess(), base
, map_name
, sizeof(map_name
));
325 ok(ret
, "GetMappedFileName error %d\n", GetLastError());
326 ok(ret
> strlen(device_name
), "map_name should be longer than device_name\n");
327 ok(memcmp(map_name
, device_name
, strlen(device_name
)) == 0, "map name does not start with a device name: %s\n", map_name
);
330 SetLastError(0xdeadbeef);
331 ret
= pGetMappedFileNameW(GetCurrentProcess(), base
, map_nameW
, sizeof(map_nameW
)/sizeof(map_nameW
[0]));
333 ok(ret
, "GetMappedFileNameW error %d\n", GetLastError());
334 ok(ret
> strlen(device_name
), "map_name should be longer than device_name\n");
336 if (nt_get_mapped_file_name(GetCurrentProcess(), base
, nt_map_name
, sizeof(nt_map_name
)/sizeof(nt_map_name
[0])))
338 ok(memcmp(map_nameW
, nt_map_name
, lstrlenW(map_nameW
)) == 0, "map name does not start with a device name: %s\n", map_name
);
339 WideCharToMultiByte(CP_ACP
, 0, map_nameW
, -1, map_name
, MAX_PATH
, NULL
, NULL
);
340 ok(memcmp(map_name
, device_name
, strlen(device_name
)) == 0, "map name does not start with a device name: %s\n", map_name
);
343 SetLastError(0xdeadbeef);
344 ret
= pGetMappedFileNameA(GetCurrentProcess(), base
+ 0x2000, map_name
, sizeof(map_name
));
346 ok(ret
, "GetMappedFileName error %d\n", GetLastError());
347 ok(ret
> strlen(device_name
), "map_name should be longer than device_name\n");
348 ok(memcmp(map_name
, device_name
, strlen(device_name
)) == 0, "map name does not start with a device name: %s\n", map_name
);
351 SetLastError(0xdeadbeef);
352 ret
= pGetMappedFileNameA(GetCurrentProcess(), base
+ 0x4000, map_name
, sizeof(map_name
));
353 ok(!ret
, "GetMappedFileName should fail\n");
355 ok(GetLastError() == ERROR_UNEXP_NET_ERR
, "expected ERROR_UNEXP_NET_ERR, got %d\n", GetLastError());
357 SetLastError(0xdeadbeef);
358 ret
= pGetMappedFileNameA(GetCurrentProcess(), NULL
, map_name
, sizeof(map_name
));
359 ok(!ret
, "GetMappedFileName should fail\n");
361 ok(GetLastError() == ERROR_UNEXP_NET_ERR
, "expected ERROR_UNEXP_NET_ERR, got %d\n", GetLastError());
363 SetLastError(0xdeadbeef);
364 ret
= pGetMappedFileNameA(0, base
, map_name
, sizeof(map_name
));
365 ok(!ret
, "GetMappedFileName should fail\n");
367 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected ERROR_INVALID_HANDLE, got %d\n", GetLastError());
369 UnmapViewOfFile(base
);
372 DeleteFile(file_name
);
374 SetLastError(0xdeadbeef);
375 hmap
= CreateFileMapping(INVALID_HANDLE_VALUE
, NULL
, PAGE_READONLY
| SEC_COMMIT
, 0, 4096, NULL
);
376 ok(hmap
!= 0, "CreateFileMapping error %d\n", GetLastError());
378 SetLastError(0xdeadbeef);
379 base
= MapViewOfFile(hmap
, FILE_MAP_READ
, 0, 0, 0);
380 ok(base
!= NULL
, "MapViewOfFile error %d\n", GetLastError());
382 SetLastError(0xdeadbeef);
383 ret
= pGetMappedFileNameA(GetCurrentProcess(), base
, map_name
, sizeof(map_name
));
384 ok(!ret
, "GetMappedFileName should fail\n");
386 ok(GetLastError() == ERROR_FILE_INVALID
, "expected ERROR_FILE_INVALID, got %d\n", GetLastError());
388 UnmapViewOfFile(base
);
392 static void test_GetProcessImageFileName(void)
394 HMODULE hMod
= GetModuleHandle(NULL
);
395 char szImgPath
[MAX_PATH
], szMapPath
[MAX_PATH
];
396 WCHAR szImgPathW
[MAX_PATH
];
399 if(pGetProcessImageFileNameA
== NULL
)
402 /* This function is available on WinXP+ only */
403 SetLastError(0xdeadbeef);
404 if(!pGetProcessImageFileNameA(hpQI
, szImgPath
, sizeof(szImgPath
)))
406 if(GetLastError() == ERROR_INVALID_FUNCTION
) {
407 win_skip("GetProcessImageFileName not implemented\n");
411 if(GetLastError() == 0xdeadbeef)
412 todo_wine
ok(0, "failed without error code\n");
414 todo_wine
ok(0, "failed with %d\n", GetLastError());
417 SetLastError(0xdeadbeef);
418 pGetProcessImageFileNameA(NULL
, szImgPath
, sizeof(szImgPath
));
419 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
421 SetLastError(0xdeadbeef);
422 pGetProcessImageFileNameA(hpSR
, szImgPath
, sizeof(szImgPath
));
423 ok(GetLastError() == ERROR_ACCESS_DENIED
, "expected error=ERROR_ACCESS_DENIED but got %d\n", GetLastError());
425 SetLastError(0xdeadbeef);
426 pGetProcessImageFileNameA(hpQI
, szImgPath
, 0);
427 ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER
, "expected error=ERROR_INSUFFICIENT_BUFFER but got %d\n", GetLastError());
429 ret
= pGetProcessImageFileNameA(hpQI
, szImgPath
, sizeof(szImgPath
));
430 ret1
= pGetMappedFileNameA(hpQV
, hMod
, szMapPath
, sizeof(szMapPath
));
433 /* Windows returns 2*strlen-1 */
434 todo_wine
ok(ret
>= strlen(szImgPath
), "szImgPath=\"%s\" ret=%d\n", szImgPath
, ret
);
435 todo_wine
ok(!strcmp(szImgPath
, szMapPath
), "szImgPath=\"%s\" szMapPath=\"%s\"\n", szImgPath
, szMapPath
);
438 SetLastError(0xdeadbeef);
439 pGetProcessImageFileNameW(NULL
, szImgPathW
, sizeof(szImgPathW
));
440 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
442 /* no information about correct buffer size returned: */
443 SetLastError(0xdeadbeef);
444 pGetProcessImageFileNameW(hpQI
, szImgPathW
, 0);
445 ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER
, "expected error=ERROR_INSUFFICIENT_BUFFER but got %d\n", GetLastError());
447 SetLastError(0xdeadbeef);
448 pGetProcessImageFileNameW(hpQI
, NULL
, 0);
449 ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER
, "expected error=ERROR_INSUFFICIENT_BUFFER but got %d\n", GetLastError());
452 memset(szImgPathW
, 0xff, sizeof(szImgPathW
));
453 ret
= pGetProcessImageFileNameW(hpQI
, szImgPathW
, sizeof(szImgPathW
)/sizeof(WCHAR
));
454 ok(ret
> 0, "GetProcessImageFileNameW should have succeeded.\n");
455 ok(szImgPathW
[0] == '\\', "GetProcessImageFileNameW should have returned an NT path.\n");
456 ok(lstrlenW(szImgPathW
) == ret
, "Expected length to be %d, got %d\n", ret
, lstrlenW(szImgPathW
));
458 /* boundary values of 'size' */
459 SetLastError(0xdeadbeef);
460 pGetProcessImageFileNameW(hpQI
, szImgPathW
, ret
);
461 ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER
, "expected error=ERROR_INSUFFICIENT_BUFFER but got %d\n", GetLastError());
463 memset(szImgPathW
, 0xff, sizeof(szImgPathW
));
464 ret
= pGetProcessImageFileNameW(hpQI
, szImgPathW
, ret
+ 1);
465 ok(ret
> 0, "GetProcessImageFileNameW should have succeeded.\n");
466 ok(szImgPathW
[0] == '\\', "GetProcessImageFileNameW should have returned an NT path.\n");
467 ok(lstrlenW(szImgPathW
) == ret
, "Expected length to be %d, got %d\n", ret
, lstrlenW(szImgPathW
));
470 static void test_GetModuleFileNameEx(void)
472 HMODULE hMod
= GetModuleHandle(NULL
);
473 char szModExPath
[MAX_PATH
+1], szModPath
[MAX_PATH
+1];
476 SetLastError(0xdeadbeef);
477 pGetModuleFileNameExA(NULL
, hMod
, szModExPath
, sizeof(szModExPath
));
478 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
480 SetLastError(0xdeadbeef);
481 pGetModuleFileNameExA(hpQI
, hMod
, szModExPath
, sizeof(szModExPath
));
482 ok(GetLastError() == ERROR_ACCESS_DENIED
, "expected error=ERROR_ACCESS_DENIED but got %d\n", GetLastError());
484 SetLastError(0xdeadbeef);
485 pGetModuleFileNameExA(hpQV
, hBad
, szModExPath
, sizeof(szModExPath
));
486 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
488 ret
= pGetModuleFileNameExA(hpQV
, NULL
, szModExPath
, sizeof(szModExPath
));
491 ok(ret
== strlen(szModExPath
), "szModExPath=\"%s\" ret=%d\n", szModExPath
, ret
);
492 GetModuleFileNameA(NULL
, szModPath
, sizeof(szModPath
));
493 ok(!strncmp(szModExPath
, szModPath
, MAX_PATH
),
494 "szModExPath=\"%s\" szModPath=\"%s\"\n", szModExPath
, szModPath
);
497 static void test_GetModuleBaseName(void)
499 HMODULE hMod
= GetModuleHandle(NULL
);
500 char szModPath
[MAX_PATH
], szModBaseName
[MAX_PATH
];
503 SetLastError(0xdeadbeef);
504 pGetModuleBaseNameA(NULL
, hMod
, szModBaseName
, sizeof(szModBaseName
));
505 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
507 SetLastError(0xdeadbeef);
508 pGetModuleBaseNameA(hpQI
, hMod
, szModBaseName
, sizeof(szModBaseName
));
509 ok(GetLastError() == ERROR_ACCESS_DENIED
, "expected error=ERROR_ACCESS_DENIED but got %d\n", GetLastError());
511 SetLastError(0xdeadbeef);
512 pGetModuleBaseNameA(hpQV
, hBad
, szModBaseName
, sizeof(szModBaseName
));
513 ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
515 ret
= pGetModuleBaseNameA(hpQV
, NULL
, szModBaseName
, sizeof(szModBaseName
));
518 ok(ret
== strlen(szModBaseName
), "szModBaseName=\"%s\" ret=%d\n", szModBaseName
, ret
);
519 GetModuleFileNameA(NULL
, szModPath
, sizeof(szModPath
));
520 ok(!strcmp(strrchr(szModPath
, '\\') + 1, szModBaseName
),
521 "szModPath=\"%s\" szModBaseName=\"%s\"\n", szModPath
, szModBaseName
);
524 static void test_ws_functions(void)
526 PSAPI_WS_WATCH_INFORMATION wswi
[4096];
527 ULONG_PTR pages
[4096];
532 SetLastError(0xdeadbeef);
533 pEmptyWorkingSet(NULL
);
534 todo_wine
ok(GetLastError() == ERROR_INVALID_HANDLE
, "expected error=ERROR_INVALID_HANDLE but got %d\n", GetLastError());
536 SetLastError(0xdeadbeef);
537 pEmptyWorkingSet(hpSR
);
538 todo_wine
ok(GetLastError() == ERROR_ACCESS_DENIED
, "expected error=ERROR_ACCESS_DENIED but got %d\n", GetLastError());
540 SetLastError(0xdeadbeef);
541 ret
= pEmptyWorkingSet(hpAA
);
542 ok(ret
== 1, "failed with %d\n", GetLastError());
544 SetLastError( 0xdeadbeef );
545 ret
= pInitializeProcessForWsWatch( NULL
);
546 todo_wine
ok( !ret
, "InitializeProcessForWsWatch succeeded\n" );
549 if (GetLastError() == ERROR_INVALID_FUNCTION
) /* not supported on xp in wow64 mode */
551 trace( "InitializeProcessForWsWatch not supported\n" );
554 ok( GetLastError() == ERROR_INVALID_HANDLE
, "wrong error %u\n", GetLastError() );
556 SetLastError(0xdeadbeef);
557 ret
= pInitializeProcessForWsWatch(hpAA
);
558 ok(ret
== 1, "failed with %d\n", GetLastError());
560 addr
= VirtualAlloc(NULL
, 1, MEM_COMMIT
, PAGE_READWRITE
);
564 *addr
= 0; /* make sure it's paged in (needed on wow64) */
565 if(!VirtualLock(addr
, 1))
567 trace("locking failed (error=%d) - skipping test\n", GetLastError());
571 SetLastError(0xdeadbeef);
572 ret
= pQueryWorkingSet(hpQI
, pages
, 4096 * sizeof(ULONG_PTR
));
573 todo_wine
ok(ret
== 1, "failed with %d\n", GetLastError());
576 for(i
= 0; i
< pages
[0]; i
++)
577 if((pages
[i
+1] & ~0xfffL
) == (ULONG_PTR
)addr
)
579 todo_wine
ok(ret
== 1, "QueryWorkingSet found our page\n");
583 todo_wine
ok(0, "QueryWorkingSet didn't find our page\n");
587 SetLastError(0xdeadbeef);
588 ret
= pGetWsChanges(hpQI
, wswi
, sizeof(wswi
));
589 todo_wine
ok(ret
== 1, "failed with %d\n", GetLastError());
592 for(i
= 0; wswi
[i
].FaultingVa
; i
++)
593 if(((ULONG_PTR
)wswi
[i
].FaultingVa
& ~0xfffL
) == (ULONG_PTR
)addr
)
595 todo_wine
ok(ret
== 1, "GetWsChanges found our page\n");
599 todo_wine
ok(0, "GetWsChanges didn't find our page\n");
603 VirtualFree(addr
, 0, MEM_RELEASE
);
606 START_TEST(psapi_main
)
608 HMODULE hpsapi
= LoadLibraryA("psapi.dll");
612 win_skip("Could not load psapi.dll\n");
616 if(InitFunctionPtrs(hpsapi
))
618 DWORD pid
= GetCurrentProcessId();
620 hpSR
= OpenProcess(STANDARD_RIGHTS_REQUIRED
, FALSE
, pid
);
621 hpQI
= OpenProcess(PROCESS_QUERY_INFORMATION
, FALSE
, pid
);
622 hpVR
= OpenProcess(PROCESS_VM_READ
, FALSE
, pid
);
623 hpQV
= OpenProcess(PROCESS_QUERY_INFORMATION
| PROCESS_VM_READ
, FALSE
, pid
);
624 hpAA
= OpenProcess(PROCESS_ALL_ACCESS
, FALSE
, pid
);
626 if(hpSR
&& hpQI
&& hpVR
&& hpQV
&& hpAA
)
628 test_EnumProcesses();
629 test_EnumProcessModules();
630 test_GetModuleInformation();
631 test_GetProcessMemoryInfo();
632 test_GetMappedFileName();
633 test_GetProcessImageFileName();
634 test_GetModuleFileNameEx();
635 test_GetModuleBaseName();