kernel32/tests: Add tests for IsBadReadPtr.
[wine/wine-gecko.git] / dlls / kernel32 / tests / virtual.c
blobfb084e8176961c3aff0cee5db7ca33f5a91dbc34
1 /*
2 * Unit test suite for Virtual* family of APIs.
4 * Copyright 2004 Dmitry Timoshkov
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include <stdarg.h>
22 #include <stdio.h>
24 #include "windef.h"
25 #include "winbase.h"
26 #include "winerror.h"
27 #include "wine/test.h"
29 #define NUM_THREADS 4
30 #define MAPPING_SIZE 0x100000
32 static HINSTANCE hkernel32;
33 static LPVOID (WINAPI *pVirtualAllocEx)(HANDLE, LPVOID, SIZE_T, DWORD, DWORD);
34 static BOOL (WINAPI *pVirtualFreeEx)(HANDLE, LPVOID, SIZE_T, DWORD);
35 static UINT (WINAPI *pGetWriteWatch)(DWORD,LPVOID,SIZE_T,LPVOID*,ULONG_PTR*,ULONG*);
36 static UINT (WINAPI *pResetWriteWatch)(LPVOID,SIZE_T);
38 /* ############################### */
40 static HANDLE create_target_process(const char *arg)
42 char **argv;
43 char cmdline[MAX_PATH];
44 PROCESS_INFORMATION pi;
45 STARTUPINFO si = { 0 };
46 si.cb = sizeof(si);
48 winetest_get_mainargs( &argv );
49 sprintf(cmdline, "%s %s %s", argv[0], argv[1], arg);
50 ok(CreateProcess(NULL, cmdline, NULL, NULL, FALSE, 0, NULL, NULL,
51 &si, &pi) != 0, "error: %u\n", GetLastError());
52 ok(CloseHandle(pi.hThread) != 0, "error %u\n", GetLastError());
53 return pi.hProcess;
56 static void test_VirtualAllocEx(void)
58 const unsigned int alloc_size = 1<<15;
59 char *src, *dst;
60 SIZE_T bytes_written = 0, bytes_read = 0, i;
61 void *addr1, *addr2;
62 BOOL b;
63 DWORD old_prot;
64 MEMORY_BASIC_INFORMATION info;
65 HANDLE hProcess;
67 /* not exported in all windows-versions */
68 if ((!pVirtualAllocEx) || (!pVirtualFreeEx)) {
69 win_skip("Virtual{Alloc,Free}Ex not available\n");
70 return;
73 hProcess = create_target_process("sleep");
74 ok(hProcess != NULL, "Can't start process\n");
76 SetLastError(0xdeadbeef);
77 addr1 = pVirtualAllocEx(hProcess, NULL, alloc_size, MEM_COMMIT,
78 PAGE_EXECUTE_READWRITE);
79 if (!addr1 && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
80 { /* Win9x */
81 win_skip("VirtualAllocEx not implemented\n");
82 TerminateProcess(hProcess, 0);
83 CloseHandle(hProcess);
84 return;
87 src = VirtualAlloc( NULL, alloc_size, MEM_COMMIT, PAGE_READWRITE );
88 dst = VirtualAlloc( NULL, alloc_size, MEM_COMMIT, PAGE_READWRITE );
89 for (i = 0; i < alloc_size; i++)
90 src[i] = i & 0xff;
92 ok(addr1 != NULL, "VirtualAllocEx error %u\n", GetLastError());
93 b = WriteProcessMemory(hProcess, addr1, src, alloc_size, &bytes_written);
94 ok(b && (bytes_written == alloc_size), "%lu bytes written\n",
95 bytes_written);
96 b = ReadProcessMemory(hProcess, addr1, dst, alloc_size, &bytes_read);
97 ok(b && (bytes_read == alloc_size), "%lu bytes read\n", bytes_read);
98 ok(!memcmp(src, dst, alloc_size), "Data from remote process differs\n");
100 /* test invalid source buffers */
102 b = VirtualProtect( src + 0x2000, 0x2000, PAGE_NOACCESS, &old_prot );
103 ok( b, "VirtualProtect failed error %u\n", GetLastError() );
104 b = WriteProcessMemory(hProcess, addr1, src, alloc_size, &bytes_written);
105 ok( !b, "WriteProcessMemory succeeded\n" );
106 ok( GetLastError() == ERROR_NOACCESS ||
107 GetLastError() == ERROR_PARTIAL_COPY, /* vista */
108 "wrong error %u\n", GetLastError() );
109 ok( bytes_written == 0, "%lu bytes written\n", bytes_written );
110 b = ReadProcessMemory(hProcess, addr1, src, alloc_size, &bytes_read);
111 ok( !b, "ReadProcessMemory succeeded\n" );
112 ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
113 ok( bytes_read == 0, "%lu bytes written\n", bytes_read );
115 b = VirtualProtect( src, 0x2000, PAGE_NOACCESS, &old_prot );
116 ok( b, "VirtualProtect failed error %u\n", GetLastError() );
117 b = WriteProcessMemory(hProcess, addr1, src, alloc_size, &bytes_written);
118 ok( !b, "WriteProcessMemory succeeded\n" );
119 ok( GetLastError() == ERROR_NOACCESS ||
120 GetLastError() == ERROR_PARTIAL_COPY, /* vista */
121 "wrong error %u\n", GetLastError() );
122 ok( bytes_written == 0, "%lu bytes written\n", bytes_written );
123 b = ReadProcessMemory(hProcess, addr1, src, alloc_size, &bytes_read);
124 ok( !b, "ReadProcessMemory succeeded\n" );
125 ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
126 ok( bytes_read == 0, "%lu bytes written\n", bytes_read );
128 b = pVirtualFreeEx(hProcess, addr1, 0, MEM_RELEASE);
129 ok(b != 0, "VirtualFreeEx, error %u\n", GetLastError());
131 VirtualFree( src, 0, MEM_FREE );
132 VirtualFree( dst, 0, MEM_FREE );
135 * The following tests parallel those in test_VirtualAlloc()
138 SetLastError(0xdeadbeef);
139 addr1 = pVirtualAllocEx(hProcess, 0, 0, MEM_RESERVE, PAGE_NOACCESS);
140 ok(addr1 == NULL, "VirtualAllocEx should fail on zero-sized allocation\n");
141 ok(GetLastError() == ERROR_INVALID_PARAMETER /* NT */ ||
142 GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Win9x */
143 "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError());
145 addr1 = pVirtualAllocEx(hProcess, 0, 0xFFFC, MEM_RESERVE, PAGE_NOACCESS);
146 ok(addr1 != NULL, "VirtualAllocEx failed\n");
148 /* test a not committed memory */
149 memset(&info, 'q', sizeof(info));
150 ok(VirtualQueryEx(hProcess, addr1, &info, sizeof(info)) == sizeof(info), "VirtualQueryEx failed\n");
151 ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
152 ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
153 ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
154 ok(info.RegionSize == 0x10000, "%lx != 0x10000\n", info.RegionSize);
155 ok(info.State == MEM_RESERVE, "%x != MEM_RESERVE\n", info.State);
156 /* NT reports Protect == 0 for a not committed memory block */
157 ok(info.Protect == 0 /* NT */ ||
158 info.Protect == PAGE_NOACCESS, /* Win9x */
159 "%x != PAGE_NOACCESS\n", info.Protect);
160 ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
162 SetLastError(0xdeadbeef);
163 ok(!VirtualProtectEx(hProcess, addr1, 0xFFFC, PAGE_READONLY, &old_prot),
164 "VirtualProtectEx should fail on a not committed memory\n");
165 ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
166 GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
167 "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
169 addr2 = pVirtualAllocEx(hProcess, addr1, 0x1000, MEM_COMMIT, PAGE_NOACCESS);
170 ok(addr1 == addr2, "VirtualAllocEx failed\n");
172 /* test a committed memory */
173 ok(VirtualQueryEx(hProcess, addr1, &info, sizeof(info)) == sizeof(info),
174 "VirtualQueryEx failed\n");
175 ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
176 ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
177 ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
178 ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
179 ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
180 /* this time NT reports PAGE_NOACCESS as well */
181 ok(info.Protect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.Protect);
182 ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
184 /* this should fail, since not the whole range is committed yet */
185 SetLastError(0xdeadbeef);
186 ok(!VirtualProtectEx(hProcess, addr1, 0xFFFC, PAGE_READONLY, &old_prot),
187 "VirtualProtectEx should fail on a not committed memory\n");
188 ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
189 GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
190 "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
192 old_prot = 0;
193 ok(VirtualProtectEx(hProcess, addr1, 0x1000, PAGE_READONLY, &old_prot), "VirtualProtectEx failed\n");
194 ok(old_prot == PAGE_NOACCESS, "wrong old protection: got %04x instead of PAGE_NOACCESS\n", old_prot);
196 old_prot = 0;
197 ok(VirtualProtectEx(hProcess, addr1, 0x1000, PAGE_READWRITE, &old_prot), "VirtualProtectEx failed\n");
198 ok(old_prot == PAGE_READONLY, "wrong old protection: got %04x instead of PAGE_READONLY\n", old_prot);
200 ok(!pVirtualFreeEx(hProcess, addr1, 0x10000, 0),
201 "VirtualFreeEx should fail with type 0\n");
202 ok(GetLastError() == ERROR_INVALID_PARAMETER,
203 "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError());
205 ok(pVirtualFreeEx(hProcess, addr1, 0x10000, MEM_DECOMMIT), "VirtualFreeEx failed\n");
207 /* if the type is MEM_RELEASE, size must be 0 */
208 ok(!pVirtualFreeEx(hProcess, addr1, 1, MEM_RELEASE),
209 "VirtualFreeEx should fail\n");
210 ok(GetLastError() == ERROR_INVALID_PARAMETER,
211 "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError());
213 ok(pVirtualFreeEx(hProcess, addr1, 0, MEM_RELEASE), "VirtualFreeEx failed\n");
215 TerminateProcess(hProcess, 0);
216 CloseHandle(hProcess);
219 static void test_VirtualAlloc(void)
221 void *addr1, *addr2;
222 DWORD old_prot;
223 MEMORY_BASIC_INFORMATION info;
225 SetLastError(0xdeadbeef);
226 addr1 = VirtualAlloc(0, 0, MEM_RESERVE, PAGE_NOACCESS);
227 ok(addr1 == NULL, "VirtualAlloc should fail on zero-sized allocation\n");
228 ok(GetLastError() == ERROR_INVALID_PARAMETER /* NT */ ||
229 GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Win9x */
230 "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError());
232 addr1 = VirtualAlloc(0, 0xFFFC, MEM_RESERVE, PAGE_NOACCESS);
233 ok(addr1 != NULL, "VirtualAlloc failed\n");
235 /* test a not committed memory */
236 ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
237 "VirtualQuery failed\n");
238 ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
239 ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
240 ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
241 ok(info.RegionSize == 0x10000, "%lx != 0x10000\n", info.RegionSize);
242 ok(info.State == MEM_RESERVE, "%x != MEM_RESERVE\n", info.State);
243 /* NT reports Protect == 0 for a not committed memory block */
244 ok(info.Protect == 0 /* NT */ ||
245 info.Protect == PAGE_NOACCESS, /* Win9x */
246 "%x != PAGE_NOACCESS\n", info.Protect);
247 ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
249 SetLastError(0xdeadbeef);
250 ok(!VirtualProtect(addr1, 0xFFFC, PAGE_READONLY, &old_prot),
251 "VirtualProtect should fail on a not committed memory\n");
252 ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
253 GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
254 "got %d, expected ERROR_INVALID_ADDRESS\n", GetLastError());
256 addr2 = VirtualAlloc(addr1, 0x1000, MEM_COMMIT, PAGE_NOACCESS);
257 ok(addr1 == addr2, "VirtualAlloc failed\n");
259 /* test a committed memory */
260 ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
261 "VirtualQuery failed\n");
262 ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
263 ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
264 ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
265 ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
266 ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
267 /* this time NT reports PAGE_NOACCESS as well */
268 ok(info.Protect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.Protect);
269 ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
271 /* this should fail, since not the whole range is committed yet */
272 SetLastError(0xdeadbeef);
273 ok(!VirtualProtect(addr1, 0xFFFC, PAGE_READONLY, &old_prot),
274 "VirtualProtect should fail on a not committed memory\n");
275 ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
276 GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
277 "got %d, expected ERROR_INVALID_ADDRESS\n", GetLastError());
279 ok(VirtualProtect(addr1, 0x1000, PAGE_READONLY, &old_prot), "VirtualProtect failed\n");
280 ok(old_prot == PAGE_NOACCESS,
281 "wrong old protection: got %04x instead of PAGE_NOACCESS\n", old_prot);
283 ok(VirtualProtect(addr1, 0x1000, PAGE_READWRITE, &old_prot), "VirtualProtect failed\n");
284 ok(old_prot == PAGE_READONLY,
285 "wrong old protection: got %04x instead of PAGE_READONLY\n", old_prot);
287 ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
288 "VirtualQuery failed\n");
289 ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
290 ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
291 ok(info.Protect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.Protect);
292 memset( addr1, 0x55, 20 );
293 ok( *(DWORD *)addr1 == 0x55555555, "wrong data %x\n", *(DWORD *)addr1 );
295 addr2 = VirtualAlloc( addr1, 0x1000, MEM_RESET, PAGE_NOACCESS );
296 ok( addr2 == addr1 || broken( !addr2 && GetLastError() == ERROR_INVALID_PARAMETER), /* win9x */
297 "VirtualAlloc failed err %u\n", GetLastError() );
298 ok( *(DWORD *)addr1 == 0x55555555 || *(DWORD *)addr1 == 0, "wrong data %x\n", *(DWORD *)addr1 );
299 if (addr2)
301 ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
302 "VirtualQuery failed\n");
303 ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
304 ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
305 ok(info.Protect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.Protect);
307 addr2 = VirtualAlloc( (char *)addr1 + 0x1000, 0x1000, MEM_RESET, PAGE_NOACCESS );
308 ok( (char *)addr2 == (char *)addr1 + 0x1000, "VirtualAlloc failed\n" );
310 ok(VirtualQuery(addr2, &info, sizeof(info)) == sizeof(info),
311 "VirtualQuery failed\n");
312 ok(info.RegionSize == 0xf000, "%lx != 0xf000\n", info.RegionSize);
313 ok(info.State == MEM_RESERVE, "%x != MEM_RESERVE\n", info.State);
314 ok(info.Protect == 0, "%x != 0\n", info.Protect);
316 addr2 = VirtualAlloc( (char *)addr1 + 0xf000, 0x2000, MEM_RESET, PAGE_NOACCESS );
317 ok( !addr2, "VirtualAlloc failed\n" );
318 ok( GetLastError() == ERROR_INVALID_ADDRESS, "wrong error %u\n", GetLastError() );
321 /* invalid protection values */
322 SetLastError(0xdeadbeef);
323 addr2 = VirtualAlloc(NULL, 0x1000, MEM_RESERVE, 0);
324 ok(!addr2, "VirtualAlloc succeeded\n");
325 ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
327 SetLastError(0xdeadbeef);
328 addr2 = VirtualAlloc(NULL, 0x1000, MEM_COMMIT, 0);
329 ok(!addr2, "VirtualAlloc succeeded\n");
330 ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
332 SetLastError(0xdeadbeef);
333 addr2 = VirtualAlloc(addr1, 0x1000, MEM_COMMIT, PAGE_READONLY | PAGE_EXECUTE);
334 ok(!addr2, "VirtualAlloc succeeded\n");
335 ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
337 SetLastError(0xdeadbeef);
338 ok(!VirtualProtect(addr1, 0x1000, PAGE_READWRITE | PAGE_EXECUTE_WRITECOPY, &old_prot),
339 "VirtualProtect succeeded\n");
340 ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
342 SetLastError(0xdeadbeef);
343 ok(!VirtualProtect(addr1, 0x1000, 0, &old_prot), "VirtualProtect succeeded\n");
344 ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
346 SetLastError(0xdeadbeef);
347 ok(!VirtualFree(addr1, 0x10000, 0), "VirtualFree should fail with type 0\n");
348 ok(GetLastError() == ERROR_INVALID_PARAMETER,
349 "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError());
351 ok(VirtualFree(addr1, 0x10000, MEM_DECOMMIT), "VirtualFree failed\n");
353 /* if the type is MEM_RELEASE, size must be 0 */
354 ok(!VirtualFree(addr1, 1, MEM_RELEASE), "VirtualFree should fail\n");
355 ok(GetLastError() == ERROR_INVALID_PARAMETER,
356 "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError());
358 ok(VirtualFree(addr1, 0, MEM_RELEASE), "VirtualFree failed\n");
361 static void test_MapViewOfFile(void)
363 static const char testfile[] = "testfile.xxx";
364 const char *name;
365 HANDLE file, mapping, map2;
366 void *ptr, *ptr2, *addr;
367 MEMORY_BASIC_INFORMATION info;
368 BOOL ret;
370 SetLastError(0xdeadbeef);
371 file = CreateFileA( testfile, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0 );
372 ok( file != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
373 SetFilePointer( file, 4096, NULL, FILE_BEGIN );
374 SetEndOfFile( file );
376 /* read/write mapping */
378 SetLastError(0xdeadbeef);
379 mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
380 ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
382 SetLastError(0xdeadbeef);
383 ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
384 ok( ptr != NULL, "MapViewOfFile FILE_MAPE_READ error %u\n", GetLastError() );
385 UnmapViewOfFile( ptr );
387 /* this fails on win9x but succeeds on NT */
388 SetLastError(0xdeadbeef);
389 ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
390 if (ptr) UnmapViewOfFile( ptr );
391 else ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
393 SetLastError(0xdeadbeef);
394 ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
395 ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
396 UnmapViewOfFile( ptr );
398 SetLastError(0xdeadbeef);
399 ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
400 ok( ptr != NULL, "MapViewOfFile FILE_MAP_WRITE error %u\n", GetLastError() );
401 UnmapViewOfFile( ptr );
403 ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2,
404 FILE_MAP_READ|FILE_MAP_WRITE, FALSE, 0 );
405 ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
406 ptr = MapViewOfFile( map2, FILE_MAP_WRITE, 0, 0, 4096 );
407 ok( ptr != NULL, "MapViewOfFile FILE_MAP_WRITE error %u\n", GetLastError() );
408 UnmapViewOfFile( ptr );
409 CloseHandle( map2 );
411 ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2,
412 FILE_MAP_READ, FALSE, 0 );
413 ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
414 ptr = MapViewOfFile( map2, FILE_MAP_WRITE, 0, 0, 4096 );
415 if (!ptr)
417 ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
418 CloseHandle( map2 );
419 ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2, 0, FALSE, 0 );
420 ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
421 ptr = MapViewOfFile( map2, 0, 0, 0, 4096 );
422 ok( !ptr, "MapViewOfFile succeeded\n" );
423 ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
424 CloseHandle( map2 );
425 ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2,
426 FILE_MAP_READ, FALSE, 0 );
427 ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
428 ptr = MapViewOfFile( map2, 0, 0, 0, 4096 );
429 ok( ptr != NULL, "MapViewOfFile NO_ACCESS error %u\n", GetLastError() );
431 else win_skip( "no access checks on win9x\n" );
433 UnmapViewOfFile( ptr );
434 CloseHandle( map2 );
435 CloseHandle( mapping );
437 /* read-only mapping */
439 SetLastError(0xdeadbeef);
440 mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL );
441 ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
443 SetLastError(0xdeadbeef);
444 ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
445 ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
446 UnmapViewOfFile( ptr );
448 /* this fails on win9x but succeeds on NT */
449 SetLastError(0xdeadbeef);
450 ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
451 if (ptr) UnmapViewOfFile( ptr );
452 else ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
454 SetLastError(0xdeadbeef);
455 ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
456 ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
457 UnmapViewOfFile( ptr );
459 SetLastError(0xdeadbeef);
460 ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
461 ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" );
462 ok( GetLastError() == ERROR_INVALID_PARAMETER ||
463 GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
464 CloseHandle( mapping );
466 /* copy-on-write mapping */
468 SetLastError(0xdeadbeef);
469 mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL );
470 ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
472 SetLastError(0xdeadbeef);
473 ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
474 ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
475 UnmapViewOfFile( ptr );
477 SetLastError(0xdeadbeef);
478 ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
479 ok( ptr != NULL, "MapViewOfFile FILE_MAP_COPY error %u\n", GetLastError() );
480 UnmapViewOfFile( ptr );
482 SetLastError(0xdeadbeef);
483 ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
484 ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
485 UnmapViewOfFile( ptr );
487 SetLastError(0xdeadbeef);
488 ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
489 ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" );
490 ok( GetLastError() == ERROR_INVALID_PARAMETER ||
491 GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
492 CloseHandle( mapping );
494 /* no access mapping */
496 SetLastError(0xdeadbeef);
497 mapping = CreateFileMappingA( file, NULL, PAGE_NOACCESS, 0, 4096, NULL );
498 /* fails on NT but succeeds on win9x */
499 if (!mapping) ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
500 else
502 SetLastError(0xdeadbeef);
503 ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
504 ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
505 UnmapViewOfFile( ptr );
507 SetLastError(0xdeadbeef);
508 ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
509 ok( !ptr, "MapViewOfFile FILE_MAP_COPY succeeded\n" );
510 ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
512 SetLastError(0xdeadbeef);
513 ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
514 ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
515 UnmapViewOfFile( ptr );
517 SetLastError(0xdeadbeef);
518 ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
519 ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" );
520 ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
522 CloseHandle( mapping );
525 CloseHandle( file );
527 /* now try read-only file */
529 SetLastError(0xdeadbeef);
530 file = CreateFileA( testfile, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0 );
531 ok( file != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
533 SetLastError(0xdeadbeef);
534 mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
535 ok( !mapping, "CreateFileMapping PAGE_READWRITE succeeded\n" );
536 ok( GetLastError() == ERROR_INVALID_PARAMETER ||
537 GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
539 SetLastError(0xdeadbeef);
540 mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL );
541 ok( mapping != 0, "CreateFileMapping PAGE_WRITECOPY error %u\n", GetLastError() );
542 CloseHandle( mapping );
544 SetLastError(0xdeadbeef);
545 mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL );
546 ok( mapping != 0, "CreateFileMapping PAGE_READONLY error %u\n", GetLastError() );
547 CloseHandle( mapping );
548 CloseHandle( file );
550 /* now try no access file */
552 SetLastError(0xdeadbeef);
553 file = CreateFileA( testfile, 0, 0, NULL, OPEN_EXISTING, 0, 0 );
554 ok( file != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
556 SetLastError(0xdeadbeef);
557 mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
558 ok( !mapping, "CreateFileMapping PAGE_READWRITE succeeded\n" );
559 ok( GetLastError() == ERROR_INVALID_PARAMETER ||
560 GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
562 SetLastError(0xdeadbeef);
563 mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL );
564 ok( !mapping, "CreateFileMapping PAGE_WRITECOPY succeeded\n" );
565 ok( GetLastError() == ERROR_INVALID_PARAMETER ||
566 GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
568 SetLastError(0xdeadbeef);
569 mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL );
570 ok( !mapping, "CreateFileMapping PAGE_READONLY succeeded\n" );
571 ok( GetLastError() == ERROR_INVALID_PARAMETER ||
572 GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
574 CloseHandle( file );
575 DeleteFileA( testfile );
577 SetLastError(0xdeadbeef);
578 name = "Local\\Foo";
579 file = CreateFileMapping( INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, 4096, name );
580 /* nt4 doesn't have Local\\ */
581 if (!file && GetLastError() == ERROR_PATH_NOT_FOUND)
583 name = "Foo";
584 file = CreateFileMapping( INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, 4096, name );
586 ok( file != 0, "CreateFileMapping PAGE_READWRITE error %u\n", GetLastError() );
588 SetLastError(0xdeadbeef);
589 mapping = OpenFileMapping( FILE_MAP_READ, FALSE, name );
590 ok( mapping != 0, "OpenFileMapping FILE_MAP_READ error %u\n", GetLastError() );
591 SetLastError(0xdeadbeef);
592 ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 0 );
593 if (!ptr)
595 ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
596 SetLastError(0xdeadbeef);
597 ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 0 );
598 ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
599 SetLastError(0xdeadbeef);
600 ok( VirtualQuery( ptr, &info, sizeof(info) ) == sizeof(info),
601 "VirtualQuery error %u\n", GetLastError() );
602 ok( info.BaseAddress == ptr, "%p != %p\n", info.BaseAddress, ptr );
603 ok( info.AllocationBase == ptr, "%p != %p\n", info.AllocationBase, ptr );
604 ok( info.AllocationProtect == PAGE_READONLY, "%x != PAGE_READONLY\n", info.AllocationProtect );
605 ok( info.RegionSize == 4096, "%lx != 4096\n", info.RegionSize );
606 ok( info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State );
607 ok( info.Protect == PAGE_READONLY, "%x != PAGE_READONLY\n", info.Protect );
609 else win_skip( "no access checks on win9x\n" );
610 UnmapViewOfFile( ptr );
611 CloseHandle( mapping );
613 SetLastError(0xdeadbeef);
614 mapping = OpenFileMapping( FILE_MAP_WRITE, FALSE, name );
615 ok( mapping != 0, "OpenFileMapping FILE_MAP_WRITE error %u\n", GetLastError() );
616 SetLastError(0xdeadbeef);
617 ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 0 );
618 if (!ptr)
620 ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
621 SetLastError(0xdeadbeef);
622 ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 0 );
623 ok( ptr != NULL, "MapViewOfFile FILE_MAP_WRITE error %u\n", GetLastError() );
624 SetLastError(0xdeadbeef);
625 ok( VirtualQuery( ptr, &info, sizeof(info) ) == sizeof(info),
626 "VirtualQuery error %u\n", GetLastError() );
627 ok( info.BaseAddress == ptr, "%p != %p\n", info.BaseAddress, ptr );
628 ok( info.AllocationBase == ptr, "%p != %p\n", info.AllocationBase, ptr );
629 ok( info.AllocationProtect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.AllocationProtect );
630 ok( info.RegionSize == 4096, "%lx != 4096\n", info.RegionSize );
631 ok( info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State );
632 ok( info.Protect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.Protect );
634 else win_skip( "no access checks on win9x\n" );
635 UnmapViewOfFile( ptr );
636 CloseHandle( mapping );
638 CloseHandle( file );
640 /* read/write mapping with SEC_RESERVE */
641 mapping = CreateFileMappingA(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE | SEC_RESERVE, 0, MAPPING_SIZE, NULL);
642 ok(mapping != INVALID_HANDLE_VALUE, "CreateFileMappingA failed with error %d\n", GetLastError());
644 ptr = MapViewOfFile(mapping, FILE_MAP_WRITE, 0, 0, 0);
645 ok(ptr != NULL, "MapViewOfFile failed with error %d\n", GetLastError());
647 ptr2 = MapViewOfFile(mapping, FILE_MAP_WRITE, 0, 0, 0);
648 /* on NT ptr != ptr2 but on Win9x ptr == ptr2 */
649 ok(ptr2 != NULL, "MapViewOfFile failed with error %d\n", GetLastError());
650 trace("mapping same section resulted in views %p and %p\n", ptr, ptr2);
652 ret = VirtualQuery(ptr, &info, sizeof(info));
653 ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
654 ok(info.BaseAddress == ptr, "BaseAddress should have been %p but was %p instead\n", ptr, info.BaseAddress);
655 ok(info.AllocationBase == ptr, "AllocationBase should have been %p but was %p instead\n", ptr, info.AllocationBase);
656 ok(info.RegionSize == MAPPING_SIZE, "RegionSize should have been 0x%x but was 0x%x\n", MAPPING_SIZE, (unsigned int)info.RegionSize);
657 ok(info.State == MEM_RESERVE, "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
658 if (info.Type == MEM_PRIVATE) /* win9x is different for uncommitted mappings */
660 ok(info.AllocationProtect == PAGE_NOACCESS,
661 "AllocationProtect should have been PAGE_NOACCESS but was 0x%x\n", info.AllocationProtect);
662 ok(info.Protect == PAGE_NOACCESS,
663 "Protect should have been PAGE_NOACCESS instead of 0x%x\n", info.Protect);
665 else
667 ok(info.AllocationProtect == PAGE_READWRITE,
668 "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
669 ok(info.Protect == 0, "Protect should have been 0 instead of 0x%x\n", info.Protect);
670 ok(info.Type == MEM_MAPPED, "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
673 if (ptr != ptr2)
675 ret = VirtualQuery(ptr2, &info, sizeof(info));
676 ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
677 ok(info.BaseAddress == ptr2,
678 "BaseAddress should have been %p but was %p instead\n", ptr2, info.BaseAddress);
679 ok(info.AllocationBase == ptr2,
680 "AllocationBase should have been %p but was %p instead\n", ptr2, info.AllocationBase);
681 ok(info.AllocationProtect == PAGE_READWRITE,
682 "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
683 ok(info.RegionSize == MAPPING_SIZE,
684 "RegionSize should have been 0x%x but was 0x%x\n", MAPPING_SIZE, (unsigned int)info.RegionSize);
685 ok(info.State == MEM_RESERVE,
686 "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
687 ok(info.Protect == 0,
688 "Protect should have been 0 instead of 0x%x\n", info.Protect);
689 ok(info.Type == MEM_MAPPED,
690 "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
693 ptr = VirtualAlloc(ptr, 0x10000, MEM_COMMIT, PAGE_READONLY);
694 ok(ptr != NULL, "VirtualAlloc failed with error %d\n", GetLastError());
696 ret = VirtualQuery(ptr, &info, sizeof(info));
697 ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
698 ok(info.BaseAddress == ptr, "BaseAddress should have been %p but was %p instead\n", ptr, info.BaseAddress);
699 ok(info.AllocationBase == ptr, "AllocationBase should have been %p but was %p instead\n", ptr, info.AllocationBase);
700 ok(info.RegionSize == 0x10000, "RegionSize should have been 0x10000 but was 0x%x\n", (unsigned int)info.RegionSize);
701 ok(info.State == MEM_COMMIT, "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
702 ok(info.Protect == PAGE_READONLY, "Protect should have been 0 instead of 0x%x\n", info.Protect);
703 if (info.Type == MEM_PRIVATE) /* win9x is different for uncommitted mappings */
705 ok(info.AllocationProtect == PAGE_NOACCESS,
706 "AllocationProtect should have been PAGE_NOACCESS but was 0x%x\n", info.AllocationProtect);
708 else
710 ok(info.AllocationProtect == PAGE_READWRITE,
711 "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
712 ok(info.Type == MEM_MAPPED, "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
715 /* shows that the VirtualAlloc above affects the mapping, not just the
716 * virtual memory in this process - it also affects all other processes
717 * with a view of the mapping, but that isn't tested here */
718 if (ptr != ptr2)
720 ret = VirtualQuery(ptr2, &info, sizeof(info));
721 ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
722 ok(info.BaseAddress == ptr2,
723 "BaseAddress should have been %p but was %p instead\n", ptr2, info.BaseAddress);
724 ok(info.AllocationBase == ptr2,
725 "AllocationBase should have been %p but was %p instead\n", ptr2, info.AllocationBase);
726 ok(info.AllocationProtect == PAGE_READWRITE,
727 "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
728 ok(info.RegionSize == 0x10000,
729 "RegionSize should have been 0x10000 but was 0x%x\n", (unsigned int)info.RegionSize);
730 ok(info.State == MEM_COMMIT,
731 "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
732 ok(info.Protect == PAGE_READWRITE,
733 "Protect should have been 0 instead of 0x%x\n", info.Protect);
734 ok(info.Type == MEM_MAPPED, "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
737 addr = VirtualAlloc( ptr, MAPPING_SIZE, MEM_RESET, PAGE_READONLY );
738 ok( addr == ptr || broken(!addr && GetLastError() == ERROR_INVALID_PARAMETER), /* win9x */
739 "VirtualAlloc failed with error %u\n", GetLastError() );
741 ret = VirtualFree( ptr, 0x10000, MEM_DECOMMIT );
742 ok( !ret || broken(ret) /* win9x */, "VirtualFree succeeded\n" );
743 if (!ret)
744 ok( GetLastError() == ERROR_INVALID_PARAMETER, "VirtualFree failed with %u\n", GetLastError() );
746 ret = UnmapViewOfFile(ptr2);
747 ok(ret, "UnmapViewOfFile failed with error %d\n", GetLastError());
748 ret = UnmapViewOfFile(ptr);
749 ok(ret, "UnmapViewOfFile failed with error %d\n", GetLastError());
750 CloseHandle(mapping);
752 addr = VirtualAlloc(NULL, 0x10000, MEM_COMMIT, PAGE_READONLY );
753 ok( addr != NULL, "VirtualAlloc failed with error %u\n", GetLastError() );
755 SetLastError(0xdeadbeef);
756 ok( !UnmapViewOfFile(addr), "UnmapViewOfFile should fail on VirtualAlloc mem\n" );
757 ok( GetLastError() == ERROR_INVALID_ADDRESS,
758 "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
759 SetLastError(0xdeadbeef);
760 ok( !UnmapViewOfFile((char *)addr + 0x3000), "UnmapViewOfFile should fail on VirtualAlloc mem\n" );
761 ok( GetLastError() == ERROR_INVALID_ADDRESS,
762 "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
763 SetLastError(0xdeadbeef);
764 ok( !UnmapViewOfFile((void *)0xdeadbeef), "UnmapViewOfFile should fail on VirtualAlloc mem\n" );
765 ok( GetLastError() == ERROR_INVALID_ADDRESS,
766 "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
768 ok( VirtualFree(addr, 0, MEM_RELEASE), "VirtualFree failed\n" );
771 static DWORD (WINAPI *pNtMapViewOfSection)( HANDLE handle, HANDLE process, PVOID *addr_ptr,
772 ULONG zero_bits, SIZE_T commit_size,
773 const LARGE_INTEGER *offset_ptr, SIZE_T *size_ptr,
774 ULONG inherit, ULONG alloc_type, ULONG protect );
775 static DWORD (WINAPI *pNtUnmapViewOfSection)( HANDLE process, PVOID addr );
777 static void test_NtMapViewOfSection(void)
779 HANDLE hProcess;
781 static const char testfile[] = "testfile.xxx";
782 static const char data[] = "test data for NtMapViewOfSection";
783 char buffer[sizeof(data)];
784 HANDLE file, mapping;
785 void *ptr;
786 BOOL ret;
787 DWORD status, written;
788 SIZE_T size, result;
789 LARGE_INTEGER offset;
791 pNtMapViewOfSection = (void *)GetProcAddress( GetModuleHandle("ntdll.dll"), "NtMapViewOfSection" );
792 pNtUnmapViewOfSection = (void *)GetProcAddress( GetModuleHandle("ntdll.dll"), "NtUnmapViewOfSection" );
793 if (!pNtMapViewOfSection || !pNtUnmapViewOfSection)
795 win_skip( "NtMapViewOfSection not available\n" );
796 return;
799 file = CreateFileA( testfile, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0 );
800 ok( file != INVALID_HANDLE_VALUE, "Failed to create test file\n" );
801 WriteFile( file, data, sizeof(data), &written, NULL );
802 SetFilePointer( file, 4096, NULL, FILE_BEGIN );
803 SetEndOfFile( file );
805 /* read/write mapping */
807 mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
808 ok( mapping != 0, "CreateFileMapping failed\n" );
810 hProcess = create_target_process("sleep");
811 ok(hProcess != NULL, "Can't start process\n");
813 ptr = NULL;
814 size = 0;
815 offset.QuadPart = 0;
816 status = pNtMapViewOfSection( mapping, hProcess, &ptr, 0, 0, &offset, &size, 1, 0, PAGE_READWRITE );
817 ok( !status, "NtMapViewOfSection failed status %x\n", status );
819 ret = ReadProcessMemory( hProcess, ptr, buffer, sizeof(buffer), &result );
820 ok( ret, "ReadProcessMemory failed\n" );
821 ok( result == sizeof(buffer), "ReadProcessMemory didn't read all data (%lx)\n", result );
822 ok( !memcmp( buffer, data, sizeof(buffer) ), "Wrong data read\n" );
824 status = pNtUnmapViewOfSection( hProcess, ptr );
825 ok( !status, "NtUnmapViewOfSection failed status %x\n", status );
827 CloseHandle( mapping );
828 CloseHandle( file );
829 DeleteFileA( testfile );
831 TerminateProcess(hProcess, 0);
832 CloseHandle(hProcess);
835 static void test_CreateFileMapping(void)
837 HANDLE handle, handle2;
839 /* test case sensitivity */
841 SetLastError(0xdeadbeef);
842 handle = CreateFileMappingA( INVALID_HANDLE_VALUE, NULL, SEC_COMMIT | PAGE_READWRITE, 0, 0x1000,
843 "Wine Test Mapping");
844 ok( handle != NULL, "CreateFileMapping failed with error %u\n", GetLastError());
845 ok( GetLastError() == 0, "wrong error %u\n", GetLastError());
847 SetLastError(0xdeadbeef);
848 handle2 = CreateFileMappingA( INVALID_HANDLE_VALUE, NULL, SEC_COMMIT | PAGE_READWRITE, 0, 0x1000,
849 "Wine Test Mapping");
850 ok( handle2 != NULL, "CreateFileMapping failed with error %d\n", GetLastError());
851 ok( GetLastError() == ERROR_ALREADY_EXISTS, "wrong error %u\n", GetLastError());
852 CloseHandle( handle2 );
854 SetLastError(0xdeadbeef);
855 handle2 = CreateFileMappingA( INVALID_HANDLE_VALUE, NULL, SEC_COMMIT | PAGE_READWRITE, 0, 0x1000,
856 "WINE TEST MAPPING");
857 ok( handle2 != NULL, "CreateFileMapping failed with error %d\n", GetLastError());
858 ok( GetLastError() == 0, "wrong error %u\n", GetLastError());
859 CloseHandle( handle2 );
861 SetLastError(0xdeadbeef);
862 handle2 = OpenFileMappingA( FILE_MAP_ALL_ACCESS, FALSE, "Wine Test Mapping");
863 ok( handle2 != NULL, "OpenFileMapping failed with error %d\n", GetLastError());
864 CloseHandle( handle2 );
866 SetLastError(0xdeadbeef);
867 handle2 = OpenFileMappingA( FILE_MAP_ALL_ACCESS, FALSE, "WINE TEST MAPPING");
868 ok( !handle2, "OpenFileMapping succeeded\n");
869 ok( GetLastError() == ERROR_FILE_NOT_FOUND || GetLastError() == ERROR_INVALID_NAME /* win9x */,
870 "wrong error %u\n", GetLastError());
872 CloseHandle( handle );
875 static void test_IsBadReadPtr(void)
877 BOOL ret;
878 void *ptr = (void *)0xdeadbeef;
879 char stackvar;
881 ret = IsBadReadPtr(NULL, 0);
882 ok(ret == FALSE, "Expected IsBadReadPtr to return FALSE, got %d\n", ret);
884 ret = IsBadReadPtr(NULL, 1);
885 ok(ret == TRUE, "Expected IsBadReadPtr to return TRUE, got %d\n", ret);
887 ret = IsBadReadPtr(ptr, 0);
888 ok(ret == FALSE, "Expected IsBadReadPtr to return FALSE, got %d\n", ret);
890 ret = IsBadReadPtr(ptr, 1);
891 ok(ret == TRUE, "Expected IsBadReadPtr to return TRUE, got %d\n", ret);
893 ret = IsBadReadPtr(&stackvar, 0);
894 ok(ret == FALSE, "Expected IsBadReadPtr to return FALSE, got %d\n", ret);
896 ret = IsBadReadPtr(&stackvar, sizeof(char));
897 ok(ret == FALSE, "Expected IsBadReadPtr to return FALSE, got %d\n", ret);
900 static void test_BadPtr(void)
902 void *ptr = (void*)1;
903 /* We assume address 1 is not mapped. */
904 ok(IsBadWritePtr(ptr,1),"IsBadWritePtr(1) failed.\n");
905 ok(IsBadCodePtr(ptr),"IsBadCodePtr(1) failed.\n");
908 static void test_write_watch(void)
910 char *base;
911 DWORD ret, size, old_prot;
912 MEMORY_BASIC_INFORMATION info;
913 void *results[64];
914 ULONG_PTR count;
915 ULONG pagesize;
917 if (!pGetWriteWatch || !pResetWriteWatch)
919 win_skip( "GetWriteWatch not supported\n" );
920 return;
923 size = 0x10000;
924 base = VirtualAlloc( 0, size, MEM_RESERVE | MEM_COMMIT | MEM_WRITE_WATCH, PAGE_READWRITE );
925 if (!base &&
926 (GetLastError() == ERROR_INVALID_PARAMETER || GetLastError() == ERROR_NOT_SUPPORTED))
928 win_skip( "MEM_WRITE_WATCH not supported\n" );
929 return;
931 ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
932 ret = VirtualQuery( base, &info, sizeof(info) );
933 ok(ret, "VirtualQuery failed %u\n", GetLastError());
934 ok( info.BaseAddress == base, "BaseAddress %p instead of %p\n", info.BaseAddress, base );
935 ok( info.AllocationProtect == PAGE_READWRITE, "wrong AllocationProtect %x\n", info.AllocationProtect );
936 ok( info.RegionSize == size, "wrong RegionSize 0x%lx\n", info.RegionSize );
937 ok( info.State == MEM_COMMIT, "wrong State 0x%x\n", info.State );
938 ok( info.Protect == PAGE_READWRITE, "wrong Protect 0x%x\n", info.Protect );
939 ok( info.Type == MEM_PRIVATE, "wrong Type 0x%x\n", info.Type );
941 count = 64;
942 SetLastError( 0xdeadbeef );
943 ret = pGetWriteWatch( 0, NULL, size, results, &count, &pagesize );
944 ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
945 ok( GetLastError() == ERROR_INVALID_PARAMETER ||
946 broken( GetLastError() == 0xdeadbeef ), /* win98 */
947 "wrong error %u\n", GetLastError() );
949 SetLastError( 0xdeadbeef );
950 ret = pGetWriteWatch( 0, GetModuleHandle(0), size, results, &count, &pagesize );
951 if (ret)
953 ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
954 ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
956 else /* win98 */
958 ok( count == 0, "wrong count %lu\n", count );
961 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
962 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
963 ok( count == 0, "wrong count %lu\n", count );
965 base[pagesize + 1] = 0x44;
967 count = 64;
968 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
969 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
970 ok( count == 1, "wrong count %lu\n", count );
971 ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
973 count = 64;
974 ret = pGetWriteWatch( WRITE_WATCH_FLAG_RESET, base, size, results, &count, &pagesize );
975 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
976 ok( count == 1, "wrong count %lu\n", count );
977 ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
979 count = 64;
980 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
981 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
982 ok( count == 0, "wrong count %lu\n", count );
984 base[2*pagesize + 3] = 0x11;
985 base[4*pagesize + 8] = 0x11;
987 count = 64;
988 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
989 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
990 ok( count == 2, "wrong count %lu\n", count );
991 ok( results[0] == base + 2*pagesize, "wrong result %p\n", results[0] );
992 ok( results[1] == base + 4*pagesize, "wrong result %p\n", results[1] );
994 count = 64;
995 ret = pGetWriteWatch( 0, base + 3*pagesize, 2*pagesize, results, &count, &pagesize );
996 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
997 ok( count == 1, "wrong count %lu\n", count );
998 ok( results[0] == base + 4*pagesize, "wrong result %p\n", results[0] );
1000 ret = pResetWriteWatch( base, 3*pagesize );
1001 ok( !ret, "pResetWriteWatch failed %u\n", GetLastError() );
1003 count = 64;
1004 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1005 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1006 ok( count == 1, "wrong count %lu\n", count );
1007 ok( results[0] == base + 4*pagesize, "wrong result %p\n", results[0] );
1009 *(DWORD *)(base + 2*pagesize - 2) = 0xdeadbeef;
1011 count = 64;
1012 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1013 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1014 ok( count == 3, "wrong count %lu\n", count );
1015 ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
1016 ok( results[1] == base + 2*pagesize, "wrong result %p\n", results[1] );
1017 ok( results[2] == base + 4*pagesize, "wrong result %p\n", results[2] );
1019 count = 1;
1020 ret = pGetWriteWatch( WRITE_WATCH_FLAG_RESET, base, size, results, &count, &pagesize );
1021 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1022 ok( count == 1, "wrong count %lu\n", count );
1023 ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
1025 count = 64;
1026 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1027 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1028 ok( count == 2, "wrong count %lu\n", count );
1029 ok( results[0] == base + 2*pagesize, "wrong result %p\n", results[0] );
1030 ok( results[1] == base + 4*pagesize, "wrong result %p\n", results[1] );
1032 /* changing protections doesn't affect watches */
1034 ret = VirtualProtect( base, 3*pagesize, PAGE_READONLY, &old_prot );
1035 ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
1036 ok( old_prot == PAGE_READWRITE, "wrong old prot %x\n", old_prot );
1038 ret = VirtualQuery( base, &info, sizeof(info) );
1039 ok(ret, "VirtualQuery failed %u\n", GetLastError());
1040 ok( info.BaseAddress == base, "BaseAddress %p instead of %p\n", info.BaseAddress, base );
1041 ok( info.RegionSize == 3*pagesize, "wrong RegionSize 0x%lx\n", info.RegionSize );
1042 ok( info.State == MEM_COMMIT, "wrong State 0x%x\n", info.State );
1043 ok( info.Protect == PAGE_READONLY, "wrong Protect 0x%x\n", info.Protect );
1045 ret = VirtualProtect( base, 3*pagesize, PAGE_READWRITE, &old_prot );
1046 ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
1047 ok( old_prot == PAGE_READONLY, "wrong old prot %x\n", old_prot );
1049 count = 64;
1050 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1051 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1052 ok( count == 2, "wrong count %lu\n", count );
1053 ok( results[0] == base + 2*pagesize, "wrong result %p\n", results[0] );
1054 ok( results[1] == base + 4*pagesize, "wrong result %p\n", results[1] );
1056 ret = VirtualQuery( base, &info, sizeof(info) );
1057 ok(ret, "VirtualQuery failed %u\n", GetLastError());
1058 ok( info.BaseAddress == base, "BaseAddress %p instead of %p\n", info.BaseAddress, base );
1059 ok( info.RegionSize == size, "wrong RegionSize 0x%lx\n", info.RegionSize );
1060 ok( info.State == MEM_COMMIT, "wrong State 0x%x\n", info.State );
1061 ok( info.Protect == PAGE_READWRITE, "wrong Protect 0x%x\n", info.Protect );
1063 /* some invalid parameter tests */
1065 SetLastError( 0xdeadbeef );
1066 count = 0;
1067 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1068 if (ret)
1070 ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1071 ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1073 SetLastError( 0xdeadbeef );
1074 ret = pGetWriteWatch( 0, base, size, results, NULL, &pagesize );
1075 ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1076 ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
1078 SetLastError( 0xdeadbeef );
1079 count = 64;
1080 ret = pGetWriteWatch( 0, base, size, results, &count, NULL );
1081 ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1082 ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
1084 SetLastError( 0xdeadbeef );
1085 count = 64;
1086 ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1087 ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1088 ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
1090 SetLastError( 0xdeadbeef );
1091 count = 0;
1092 ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1093 ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1094 ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1096 SetLastError( 0xdeadbeef );
1097 count = 64;
1098 ret = pGetWriteWatch( 0xdeadbeef, base, size, results, &count, &pagesize );
1099 ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1100 ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1102 SetLastError( 0xdeadbeef );
1103 count = 64;
1104 ret = pGetWriteWatch( 0, base, 0, results, &count, &pagesize );
1105 ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1106 ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1108 SetLastError( 0xdeadbeef );
1109 count = 64;
1110 ret = pGetWriteWatch( 0, base, size * 2, results, &count, &pagesize );
1111 ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1112 ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1114 SetLastError( 0xdeadbeef );
1115 count = 64;
1116 ret = pGetWriteWatch( 0, base + size - pagesize, pagesize + 1, results, &count, &pagesize );
1117 ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1118 ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1120 SetLastError( 0xdeadbeef );
1121 ret = pResetWriteWatch( base, 0 );
1122 ok( ret == ~0u, "ResetWriteWatch succeeded %u\n", ret );
1123 ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1125 SetLastError( 0xdeadbeef );
1126 ret = pResetWriteWatch( GetModuleHandle(0), size );
1127 ok( ret == ~0u, "ResetWriteWatch succeeded %u\n", ret );
1128 ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1130 else /* win98 is completely different */
1132 SetLastError( 0xdeadbeef );
1133 count = 64;
1134 ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1135 ok( ret == ERROR_INVALID_PARAMETER, "GetWriteWatch succeeded %u\n", ret );
1136 ok( GetLastError() == 0xdeadbeef, "wrong error %u\n", GetLastError() );
1138 count = 0;
1139 ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1140 ok( !ret, "GetWriteWatch failed %u\n", ret );
1142 count = 64;
1143 ret = pGetWriteWatch( 0xdeadbeef, base, size, results, &count, &pagesize );
1144 ok( !ret, "GetWriteWatch failed %u\n", ret );
1146 count = 64;
1147 ret = pGetWriteWatch( 0, base, 0, results, &count, &pagesize );
1148 ok( !ret, "GetWriteWatch failed %u\n", ret );
1150 ret = pResetWriteWatch( base, 0 );
1151 ok( !ret, "ResetWriteWatch failed %u\n", ret );
1153 ret = pResetWriteWatch( GetModuleHandle(0), size );
1154 ok( !ret, "ResetWriteWatch failed %u\n", ret );
1157 VirtualFree( base, 0, MEM_FREE );
1159 base = VirtualAlloc( 0, size, MEM_RESERVE | MEM_WRITE_WATCH, PAGE_READWRITE );
1160 ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
1161 VirtualFree( base, 0, MEM_FREE );
1163 base = VirtualAlloc( 0, size, MEM_WRITE_WATCH, PAGE_READWRITE );
1164 ok( !base, "VirtualAlloc succeeded\n" );
1165 ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1167 /* initial protect doesn't matter */
1169 base = VirtualAlloc( 0, size, MEM_RESERVE | MEM_WRITE_WATCH, PAGE_NOACCESS );
1170 ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
1171 base = VirtualAlloc( base, size, MEM_COMMIT, PAGE_NOACCESS );
1172 ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
1174 count = 64;
1175 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1176 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1177 ok( count == 0, "wrong count %lu\n", count );
1179 ret = VirtualProtect( base, 6*pagesize, PAGE_READWRITE, &old_prot );
1180 ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
1181 ok( old_prot == PAGE_NOACCESS, "wrong old prot %x\n", old_prot );
1183 base[5*pagesize + 200] = 3;
1185 ret = VirtualProtect( base, 6*pagesize, PAGE_NOACCESS, &old_prot );
1186 ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
1187 ok( old_prot == PAGE_READWRITE, "wrong old prot %x\n", old_prot );
1189 count = 64;
1190 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1191 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1192 ok( count == 1, "wrong count %lu\n", count );
1193 ok( results[0] == base + 5*pagesize, "wrong result %p\n", results[0] );
1195 ret = VirtualFree( base, size, MEM_DECOMMIT );
1196 ok( ret, "VirtualFree failed %u\n", GetLastError() );
1198 count = 64;
1199 ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1200 ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1201 ok( count == 1 || broken(count == 0), /* win98 */
1202 "wrong count %lu\n", count );
1203 if (count) ok( results[0] == base + 5*pagesize, "wrong result %p\n", results[0] );
1205 VirtualFree( base, 0, MEM_FREE );
1208 START_TEST(virtual)
1210 int argc;
1211 char **argv;
1212 argc = winetest_get_mainargs( &argv );
1214 if (argc >= 3)
1216 if (!strcmp(argv[2], "sleep"))
1218 Sleep(5000); /* spawned process runs for at most 5 seconds */
1219 return;
1221 while (1)
1223 void *mem;
1224 BOOL ret;
1225 mem = VirtualAlloc(NULL, 1<<20, MEM_COMMIT|MEM_RESERVE,
1226 PAGE_EXECUTE_READWRITE);
1227 ok(mem != NULL, "VirtualAlloc failed %u\n", GetLastError());
1228 if (mem == NULL) break;
1229 ret = VirtualFree(mem, 0, MEM_RELEASE);
1230 ok(ret, "VirtualFree failed %u\n", GetLastError());
1231 if (!ret) break;
1233 return;
1236 hkernel32 = GetModuleHandleA("kernel32.dll");
1237 pVirtualAllocEx = (void *) GetProcAddress(hkernel32, "VirtualAllocEx");
1238 pVirtualFreeEx = (void *) GetProcAddress(hkernel32, "VirtualFreeEx");
1239 pGetWriteWatch = (void *) GetProcAddress(hkernel32, "GetWriteWatch");
1240 pResetWriteWatch = (void *) GetProcAddress(hkernel32, "ResetWriteWatch");
1242 test_VirtualAllocEx();
1243 test_VirtualAlloc();
1244 test_MapViewOfFile();
1245 test_NtMapViewOfSection();
1246 test_CreateFileMapping();
1247 test_IsBadReadPtr();
1248 test_BadPtr();
1249 test_write_watch();