git.oblomov.eu Git - wine/blob - dlls/kernel32/tests/virtual.c

   1 /*
   2  * Unit test suite for Virtual* family of APIs.
   3  *
   4  * Copyright 2004 Dmitry Timoshkov
   5  *
   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.
  10  *
  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.
  15  *
  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
  19  */
  20
  21 #include <stdarg.h>
  22 #include <stdio.h>
  23
  24 #include "ntstatus.h"
  25 #define WIN32_NO_STATUS
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winternl.h"
  29 #include "winerror.h"
  30 #include "wine/test.h"
  31
  32 #define NUM_THREADS 4
  33 #define MAPPING_SIZE 0x100000
  34
  35 static HINSTANCE hkernel32;
  36 static LPVOID (WINAPI *pVirtualAllocEx)(HANDLE, LPVOID, SIZE_T, DWORD, DWORD);
  37 static BOOL   (WINAPI *pVirtualFreeEx)(HANDLE, LPVOID, SIZE_T, DWORD);
  38 static UINT   (WINAPI *pGetWriteWatch)(DWORD,LPVOID,SIZE_T,LPVOID*,ULONG_PTR*,ULONG*);
  39 static UINT   (WINAPI *pResetWriteWatch)(LPVOID,SIZE_T);
  40 static NTSTATUS (WINAPI *pNtAreMappedFilesTheSame)(PVOID,PVOID);
  41 static NTSTATUS (WINAPI *pNtMapViewOfSection)(HANDLE, HANDLE, PVOID *, ULONG, SIZE_T, const LARGE_INTEGER *, SIZE_T *, ULONG, ULONG, ULONG);
  42 static DWORD (WINAPI *pNtUnmapViewOfSection)(HANDLE, PVOID);
  43
  44 /* ############################### */
  45
  46 static HANDLE create_target_process(const char *arg)
  47 {
  48     char **argv;
  49     char cmdline[MAX_PATH];
  50     PROCESS_INFORMATION pi;
  51     BOOL ret;
  52     STARTUPINFO si = { 0 };
  53     si.cb = sizeof(si);
  54
  55     winetest_get_mainargs( &argv );
  56     sprintf(cmdline, "%s %s %s", argv[0], argv[1], arg);
  57     ret = CreateProcess(NULL, cmdline, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi);
  58     ok(ret, "error: %u\n", GetLastError());
  59     ret = CloseHandle(pi.hThread);
  60     ok(ret, "error %u\n", GetLastError());
  61     return pi.hProcess;
  62 }
  63
  64 static void test_VirtualAllocEx(void)
  65 {
  66     const unsigned int alloc_size = 1<<15;
  67     char *src, *dst;
  68     SIZE_T bytes_written = 0, bytes_read = 0, i;
  69     void *addr1, *addr2;
  70     BOOL b;
  71     DWORD old_prot;
  72     MEMORY_BASIC_INFORMATION info;
  73     HANDLE hProcess;
  74
  75     /* not exported in all windows-versions  */
  76     if ((!pVirtualAllocEx) || (!pVirtualFreeEx)) {
  77         win_skip("Virtual{Alloc,Free}Ex not available\n");
  78         return;
  79     }
  80
  81     hProcess = create_target_process("sleep");
  82     ok(hProcess != NULL, "Can't start process\n");
  83
  84     SetLastError(0xdeadbeef);
  85     addr1 = pVirtualAllocEx(hProcess, NULL, alloc_size, MEM_COMMIT,
  86                            PAGE_EXECUTE_READWRITE);
  87     if (!addr1 && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
  88     {   /* Win9x */
  89         win_skip("VirtualAllocEx not implemented\n");
  90         TerminateProcess(hProcess, 0);
  91         CloseHandle(hProcess);
  92         return;
  93     }
  94
  95     src = VirtualAlloc( NULL, alloc_size, MEM_COMMIT, PAGE_READWRITE );
  96     dst = VirtualAlloc( NULL, alloc_size, MEM_COMMIT, PAGE_READWRITE );
  97     for (i = 0; i < alloc_size; i++)
  98         src[i] = i & 0xff;
  99
 100     ok(addr1 != NULL, "VirtualAllocEx error %u\n", GetLastError());
 101     b = WriteProcessMemory(hProcess, addr1, src, alloc_size, &bytes_written);
 102     ok(b && (bytes_written == alloc_size), "%lu bytes written\n",
 103        bytes_written);
 104     b = ReadProcessMemory(hProcess, addr1, dst, alloc_size, &bytes_read);
 105     ok(b && (bytes_read == alloc_size), "%lu bytes read\n", bytes_read);
 106     ok(!memcmp(src, dst, alloc_size), "Data from remote process differs\n");
 107
 108     /* test invalid source buffers */
 109
 110     b = VirtualProtect( src + 0x2000, 0x2000, PAGE_NOACCESS, &old_prot );
 111     ok( b, "VirtualProtect failed error %u\n", GetLastError() );
 112     b = WriteProcessMemory(hProcess, addr1, src, alloc_size, &bytes_written);
 113     ok( !b, "WriteProcessMemory succeeded\n" );
 114     ok( GetLastError() == ERROR_NOACCESS ||
 115         GetLastError() == ERROR_PARTIAL_COPY, /* vista */
 116         "wrong error %u\n", GetLastError() );
 117     ok( bytes_written == 0, "%lu bytes written\n", bytes_written );
 118     b = ReadProcessMemory(hProcess, addr1, src, alloc_size, &bytes_read);
 119     ok( !b, "ReadProcessMemory succeeded\n" );
 120     ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
 121     ok( bytes_read == 0, "%lu bytes written\n", bytes_read );
 122
 123     b = VirtualProtect( src, 0x2000, PAGE_NOACCESS, &old_prot );
 124     ok( b, "VirtualProtect failed error %u\n", GetLastError() );
 125     b = WriteProcessMemory(hProcess, addr1, src, alloc_size, &bytes_written);
 126     ok( !b, "WriteProcessMemory succeeded\n" );
 127     ok( GetLastError() == ERROR_NOACCESS ||
 128         GetLastError() == ERROR_PARTIAL_COPY, /* vista */
 129         "wrong error %u\n", GetLastError() );
 130     ok( bytes_written == 0, "%lu bytes written\n", bytes_written );
 131     b = ReadProcessMemory(hProcess, addr1, src, alloc_size, &bytes_read);
 132     ok( !b, "ReadProcessMemory succeeded\n" );
 133     ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
 134     ok( bytes_read == 0, "%lu bytes written\n", bytes_read );
 135
 136     b = pVirtualFreeEx(hProcess, addr1, 0, MEM_RELEASE);
 137     ok(b != 0, "VirtualFreeEx, error %u\n", GetLastError());
 138
 139     VirtualFree( src, 0, MEM_FREE );
 140     VirtualFree( dst, 0, MEM_FREE );
 141
 142     /*
 143      * The following tests parallel those in test_VirtualAlloc()
 144      */
 145
 146     SetLastError(0xdeadbeef);
 147     addr1 = pVirtualAllocEx(hProcess, 0, 0, MEM_RESERVE, PAGE_NOACCESS);
 148     ok(addr1 == NULL, "VirtualAllocEx should fail on zero-sized allocation\n");
 149     ok(GetLastError() == ERROR_INVALID_PARAMETER /* NT */ ||
 150        GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Win9x */
 151         "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 152
 153     addr1 = pVirtualAllocEx(hProcess, 0, 0xFFFC, MEM_RESERVE, PAGE_NOACCESS);
 154     ok(addr1 != NULL, "VirtualAllocEx failed\n");
 155
 156     /* test a not committed memory */
 157     memset(&info, 'q', sizeof(info));
 158     ok(VirtualQueryEx(hProcess, addr1, &info, sizeof(info)) == sizeof(info), "VirtualQueryEx failed\n");
 159     ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
 160     ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
 161     ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
 162     ok(info.RegionSize == 0x10000, "%lx != 0x10000\n", info.RegionSize);
 163     ok(info.State == MEM_RESERVE, "%x != MEM_RESERVE\n", info.State);
 164     /* NT reports Protect == 0 for a not committed memory block */
 165     ok(info.Protect == 0 /* NT */ ||
 166        info.Protect == PAGE_NOACCESS, /* Win9x */
 167         "%x != PAGE_NOACCESS\n", info.Protect);
 168     ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
 169
 170     SetLastError(0xdeadbeef);
 171     ok(!VirtualProtectEx(hProcess, addr1, 0xFFFC, PAGE_READONLY, &old_prot),
 172        "VirtualProtectEx should fail on a not committed memory\n");
 173     ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
 174        GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
 175         "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
 176
 177     addr2 = pVirtualAllocEx(hProcess, addr1, 0x1000, MEM_COMMIT, PAGE_NOACCESS);
 178     ok(addr1 == addr2, "VirtualAllocEx failed\n");
 179
 180     /* test a committed memory */
 181     ok(VirtualQueryEx(hProcess, addr1, &info, sizeof(info)) == sizeof(info),
 182         "VirtualQueryEx failed\n");
 183     ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
 184     ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
 185     ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
 186     ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
 187     ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
 188     /* this time NT reports PAGE_NOACCESS as well */
 189     ok(info.Protect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.Protect);
 190     ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
 191
 192     /* this should fail, since not the whole range is committed yet */
 193     SetLastError(0xdeadbeef);
 194     ok(!VirtualProtectEx(hProcess, addr1, 0xFFFC, PAGE_READONLY, &old_prot),
 195         "VirtualProtectEx should fail on a not committed memory\n");
 196     ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
 197        GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
 198         "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
 199
 200     old_prot = 0;
 201     ok(VirtualProtectEx(hProcess, addr1, 0x1000, PAGE_READONLY, &old_prot), "VirtualProtectEx failed\n");
 202     ok(old_prot == PAGE_NOACCESS, "wrong old protection: got %04x instead of PAGE_NOACCESS\n", old_prot);
 203
 204     old_prot = 0;
 205     ok(VirtualProtectEx(hProcess, addr1, 0x1000, PAGE_READWRITE, &old_prot), "VirtualProtectEx failed\n");
 206     ok(old_prot == PAGE_READONLY, "wrong old protection: got %04x instead of PAGE_READONLY\n", old_prot);
 207
 208     ok(!pVirtualFreeEx(hProcess, addr1, 0x10000, 0),
 209        "VirtualFreeEx should fail with type 0\n");
 210     ok(GetLastError() == ERROR_INVALID_PARAMETER,
 211         "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 212
 213     ok(pVirtualFreeEx(hProcess, addr1, 0x10000, MEM_DECOMMIT), "VirtualFreeEx failed\n");
 214
 215     /* if the type is MEM_RELEASE, size must be 0 */
 216     ok(!pVirtualFreeEx(hProcess, addr1, 1, MEM_RELEASE),
 217        "VirtualFreeEx should fail\n");
 218     ok(GetLastError() == ERROR_INVALID_PARAMETER,
 219         "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 220
 221     ok(pVirtualFreeEx(hProcess, addr1, 0, MEM_RELEASE), "VirtualFreeEx failed\n");
 222
 223     TerminateProcess(hProcess, 0);
 224     CloseHandle(hProcess);
 225 }
 226
 227 static void test_VirtualAlloc(void)
 228 {
 229     void *addr1, *addr2;
 230     DWORD old_prot;
 231     MEMORY_BASIC_INFORMATION info;
 232
 233     SetLastError(0xdeadbeef);
 234     addr1 = VirtualAlloc(0, 0, MEM_RESERVE, PAGE_NOACCESS);
 235     ok(addr1 == NULL, "VirtualAlloc should fail on zero-sized allocation\n");
 236     ok(GetLastError() == ERROR_INVALID_PARAMETER /* NT */ ||
 237        GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Win9x */
 238         "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 239
 240     addr1 = VirtualAlloc(0, 0xFFFC, MEM_RESERVE, PAGE_NOACCESS);
 241     ok(addr1 != NULL, "VirtualAlloc failed\n");
 242
 243     /* test a not committed memory */
 244     ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
 245         "VirtualQuery failed\n");
 246     ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
 247     ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
 248     ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
 249     ok(info.RegionSize == 0x10000, "%lx != 0x10000\n", info.RegionSize);
 250     ok(info.State == MEM_RESERVE, "%x != MEM_RESERVE\n", info.State);
 251     /* NT reports Protect == 0 for a not committed memory block */
 252     ok(info.Protect == 0 /* NT */ ||
 253        info.Protect == PAGE_NOACCESS, /* Win9x */
 254         "%x != PAGE_NOACCESS\n", info.Protect);
 255     ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
 256
 257     SetLastError(0xdeadbeef);
 258     ok(!VirtualProtect(addr1, 0xFFFC, PAGE_READONLY, &old_prot),
 259        "VirtualProtect should fail on a not committed memory\n");
 260     ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
 261        GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
 262         "got %d, expected ERROR_INVALID_ADDRESS\n", GetLastError());
 263
 264     addr2 = VirtualAlloc(addr1, 0x1000, MEM_COMMIT, PAGE_NOACCESS);
 265     ok(addr1 == addr2, "VirtualAlloc failed\n");
 266
 267     /* test a committed memory */
 268     ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
 269         "VirtualQuery failed\n");
 270     ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
 271     ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
 272     ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
 273     ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
 274     ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
 275     /* this time NT reports PAGE_NOACCESS as well */
 276     ok(info.Protect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.Protect);
 277     ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
 278
 279     /* this should fail, since not the whole range is committed yet */
 280     SetLastError(0xdeadbeef);
 281     ok(!VirtualProtect(addr1, 0xFFFC, PAGE_READONLY, &old_prot),
 282         "VirtualProtect should fail on a not committed memory\n");
 283     ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
 284        GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
 285         "got %d, expected ERROR_INVALID_ADDRESS\n", GetLastError());
 286
 287     ok(VirtualProtect(addr1, 0x1000, PAGE_READONLY, &old_prot), "VirtualProtect failed\n");
 288     ok(old_prot == PAGE_NOACCESS,
 289         "wrong old protection: got %04x instead of PAGE_NOACCESS\n", old_prot);
 290
 291     ok(VirtualProtect(addr1, 0x1000, PAGE_READWRITE, &old_prot), "VirtualProtect failed\n");
 292     ok(old_prot == PAGE_READONLY,
 293         "wrong old protection: got %04x instead of PAGE_READONLY\n", old_prot);
 294
 295     ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
 296         "VirtualQuery failed\n");
 297     ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
 298     ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
 299     ok(info.Protect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.Protect);
 300     memset( addr1, 0x55, 20 );
 301     ok( *(DWORD *)addr1 == 0x55555555, "wrong data %x\n", *(DWORD *)addr1 );
 302
 303     addr2 = VirtualAlloc( addr1, 0x1000, MEM_RESET, PAGE_NOACCESS );
 304     ok( addr2 == addr1 || broken( !addr2 && GetLastError() == ERROR_INVALID_PARAMETER), /* win9x */
 305         "VirtualAlloc failed err %u\n", GetLastError() );
 306     ok( *(DWORD *)addr1 == 0x55555555 || *(DWORD *)addr1 == 0, "wrong data %x\n", *(DWORD *)addr1 );
 307     if (addr2)
 308     {
 309         ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
 310            "VirtualQuery failed\n");
 311         ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
 312         ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
 313         ok(info.Protect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.Protect);
 314
 315         addr2 = VirtualAlloc( (char *)addr1 + 0x1000, 0x1000, MEM_RESET, PAGE_NOACCESS );
 316         ok( (char *)addr2 == (char *)addr1 + 0x1000, "VirtualAlloc failed\n" );
 317
 318         ok(VirtualQuery(addr2, &info, sizeof(info)) == sizeof(info),
 319            "VirtualQuery failed\n");
 320         ok(info.RegionSize == 0xf000, "%lx != 0xf000\n", info.RegionSize);
 321         ok(info.State == MEM_RESERVE, "%x != MEM_RESERVE\n", info.State);
 322         ok(info.Protect == 0, "%x != 0\n", info.Protect);
 323
 324         addr2 = VirtualAlloc( (char *)addr1 + 0xf000, 0x2000, MEM_RESET, PAGE_NOACCESS );
 325         ok( !addr2, "VirtualAlloc failed\n" );
 326         ok( GetLastError() == ERROR_INVALID_ADDRESS, "wrong error %u\n", GetLastError() );
 327     }
 328
 329     /* invalid protection values */
 330     SetLastError(0xdeadbeef);
 331     addr2 = VirtualAlloc(NULL, 0x1000, MEM_RESERVE, 0);
 332     ok(!addr2, "VirtualAlloc succeeded\n");
 333     ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
 334
 335     SetLastError(0xdeadbeef);
 336     addr2 = VirtualAlloc(NULL, 0x1000, MEM_COMMIT, 0);
 337     ok(!addr2, "VirtualAlloc succeeded\n");
 338     ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
 339
 340     SetLastError(0xdeadbeef);
 341     addr2 = VirtualAlloc(addr1, 0x1000, MEM_COMMIT, PAGE_READONLY | PAGE_EXECUTE);
 342     ok(!addr2, "VirtualAlloc succeeded\n");
 343     ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
 344
 345     SetLastError(0xdeadbeef);
 346     ok(!VirtualProtect(addr1, 0x1000, PAGE_READWRITE | PAGE_EXECUTE_WRITECOPY, &old_prot),
 347        "VirtualProtect succeeded\n");
 348     ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
 349
 350     SetLastError(0xdeadbeef);
 351     ok(!VirtualProtect(addr1, 0x1000, 0, &old_prot), "VirtualProtect succeeded\n");
 352     ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
 353
 354     SetLastError(0xdeadbeef);
 355     ok(!VirtualFree(addr1, 0x10000, 0), "VirtualFree should fail with type 0\n");
 356     ok(GetLastError() == ERROR_INVALID_PARAMETER,
 357         "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 358
 359     ok(VirtualFree(addr1, 0x10000, MEM_DECOMMIT), "VirtualFree failed\n");
 360
 361     /* if the type is MEM_RELEASE, size must be 0 */
 362     ok(!VirtualFree(addr1, 1, MEM_RELEASE), "VirtualFree should fail\n");
 363     ok(GetLastError() == ERROR_INVALID_PARAMETER,
 364         "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 365
 366     ok(VirtualFree(addr1, 0, MEM_RELEASE), "VirtualFree failed\n");
 367 }
 368
 369 static void test_MapViewOfFile(void)
 370 {
 371     static const char testfile[] = "testfile.xxx";
 372     const char *name;
 373     HANDLE file, mapping, map2;
 374     void *ptr, *ptr2, *addr;
 375     MEMORY_BASIC_INFORMATION info;
 376     BOOL ret;
 377
 378     SetLastError(0xdeadbeef);
 379     file = CreateFileA( testfile, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0 );
 380     ok( file != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
 381     SetFilePointer( file, 4096, NULL, FILE_BEGIN );
 382     SetEndOfFile( file );
 383
 384     /* read/write mapping */
 385
 386     SetLastError(0xdeadbeef);
 387     mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
 388     ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
 389
 390     SetLastError(0xdeadbeef);
 391     ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
 392     ok( ptr != NULL, "MapViewOfFile FILE_MAPE_READ error %u\n", GetLastError() );
 393     UnmapViewOfFile( ptr );
 394
 395     /* this fails on win9x but succeeds on NT */
 396     SetLastError(0xdeadbeef);
 397     ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
 398     if (ptr) UnmapViewOfFile( ptr );
 399     else ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
 400
 401     SetLastError(0xdeadbeef);
 402     ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
 403     ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
 404     UnmapViewOfFile( ptr );
 405
 406     SetLastError(0xdeadbeef);
 407     ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
 408     ok( ptr != NULL, "MapViewOfFile FILE_MAP_WRITE error %u\n", GetLastError() );
 409     UnmapViewOfFile( ptr );
 410
 411     ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2,
 412                            FILE_MAP_READ|FILE_MAP_WRITE, FALSE, 0 );
 413     ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
 414     ptr = MapViewOfFile( map2, FILE_MAP_WRITE, 0, 0, 4096 );
 415     ok( ptr != NULL, "MapViewOfFile FILE_MAP_WRITE error %u\n", GetLastError() );
 416     UnmapViewOfFile( ptr );
 417     CloseHandle( map2 );
 418
 419     ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2,
 420                            FILE_MAP_READ, FALSE, 0 );
 421     ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
 422     SetLastError(0xdeadbeef);
 423     ptr = MapViewOfFile( map2, FILE_MAP_WRITE, 0, 0, 4096 );
 424     if (!ptr)
 425     {
 426         ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 427         CloseHandle( map2 );
 428         ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2, 0, FALSE, 0 );
 429         ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
 430         SetLastError(0xdeadbeef);
 431         ptr = MapViewOfFile( map2, 0, 0, 0, 4096 );
 432         ok( !ptr, "MapViewOfFile succeeded\n" );
 433         ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 434         CloseHandle( map2 );
 435         ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2,
 436                                FILE_MAP_READ, FALSE, 0 );
 437         ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
 438         ptr = MapViewOfFile( map2, 0, 0, 0, 4096 );
 439         ok( ptr != NULL, "MapViewOfFile NO_ACCESS error %u\n", GetLastError() );
 440     }
 441     else win_skip( "no access checks on win9x\n" );
 442
 443     UnmapViewOfFile( ptr );
 444     CloseHandle( map2 );
 445     CloseHandle( mapping );
 446
 447     /* read-only mapping */
 448
 449     SetLastError(0xdeadbeef);
 450     mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL );
 451     ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
 452
 453     SetLastError(0xdeadbeef);
 454     ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
 455     ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 456     UnmapViewOfFile( ptr );
 457
 458     /* this fails on win9x but succeeds on NT */
 459     SetLastError(0xdeadbeef);
 460     ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
 461     if (ptr) UnmapViewOfFile( ptr );
 462     else ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
 463
 464     SetLastError(0xdeadbeef);
 465     ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
 466     ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
 467     UnmapViewOfFile( ptr );
 468
 469     SetLastError(0xdeadbeef);
 470     ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
 471     ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" );
 472     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 473         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 474     CloseHandle( mapping );
 475
 476     /* copy-on-write mapping */
 477
 478     SetLastError(0xdeadbeef);
 479     mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL );
 480     ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
 481
 482     SetLastError(0xdeadbeef);
 483     ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
 484     ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 485     UnmapViewOfFile( ptr );
 486
 487     SetLastError(0xdeadbeef);
 488     ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
 489     ok( ptr != NULL, "MapViewOfFile FILE_MAP_COPY error %u\n", GetLastError() );
 490     UnmapViewOfFile( ptr );
 491
 492     SetLastError(0xdeadbeef);
 493     ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
 494     ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
 495     UnmapViewOfFile( ptr );
 496
 497     SetLastError(0xdeadbeef);
 498     ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
 499     ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" );
 500     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 501         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 502     CloseHandle( mapping );
 503
 504     /* no access mapping */
 505
 506     SetLastError(0xdeadbeef);
 507     mapping = CreateFileMappingA( file, NULL, PAGE_NOACCESS, 0, 4096, NULL );
 508     /* fails on NT but succeeds on win9x */
 509     if (!mapping) ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
 510     else
 511     {
 512         SetLastError(0xdeadbeef);
 513         ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
 514         ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 515         UnmapViewOfFile( ptr );
 516
 517         SetLastError(0xdeadbeef);
 518         ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
 519         ok( !ptr, "MapViewOfFile FILE_MAP_COPY succeeded\n" );
 520         ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
 521
 522         SetLastError(0xdeadbeef);
 523         ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
 524         ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
 525         UnmapViewOfFile( ptr );
 526
 527         SetLastError(0xdeadbeef);
 528         ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
 529         ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" );
 530         ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
 531
 532         CloseHandle( mapping );
 533     }
 534
 535     CloseHandle( file );
 536
 537     /* now try read-only file */
 538
 539     SetLastError(0xdeadbeef);
 540     file = CreateFileA( testfile, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0 );
 541     ok( file != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
 542
 543     SetLastError(0xdeadbeef);
 544     mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
 545     ok( !mapping, "CreateFileMapping PAGE_READWRITE succeeded\n" );
 546     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 547         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 548
 549     SetLastError(0xdeadbeef);
 550     mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL );
 551     ok( mapping != 0, "CreateFileMapping PAGE_WRITECOPY error %u\n", GetLastError() );
 552     CloseHandle( mapping );
 553
 554     SetLastError(0xdeadbeef);
 555     mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL );
 556     ok( mapping != 0, "CreateFileMapping PAGE_READONLY error %u\n", GetLastError() );
 557     CloseHandle( mapping );
 558     CloseHandle( file );
 559
 560     /* now try no access file */
 561
 562     SetLastError(0xdeadbeef);
 563     file = CreateFileA( testfile, 0, 0, NULL, OPEN_EXISTING, 0, 0 );
 564     ok( file != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
 565
 566     SetLastError(0xdeadbeef);
 567     mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
 568     ok( !mapping, "CreateFileMapping PAGE_READWRITE succeeded\n" );
 569     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 570         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 571
 572     SetLastError(0xdeadbeef);
 573     mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL );
 574     ok( !mapping, "CreateFileMapping PAGE_WRITECOPY succeeded\n" );
 575     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 576         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 577
 578     SetLastError(0xdeadbeef);
 579     mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL );
 580     ok( !mapping, "CreateFileMapping PAGE_READONLY succeeded\n" );
 581     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 582         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 583
 584     CloseHandle( file );
 585     DeleteFileA( testfile );
 586
 587     SetLastError(0xdeadbeef);
 588     name = "Local\\Foo";
 589     file = CreateFileMapping( INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, 4096, name );
 590     /* nt4 doesn't have Local\\ */
 591     if (!file && GetLastError() == ERROR_PATH_NOT_FOUND)
 592     {
 593         name = "Foo";
 594         file = CreateFileMapping( INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, 4096, name );
 595     }
 596     ok( file != 0, "CreateFileMapping PAGE_READWRITE error %u\n", GetLastError() );
 597
 598     SetLastError(0xdeadbeef);
 599     mapping = OpenFileMapping( FILE_MAP_READ, FALSE, name );
 600     ok( mapping != 0, "OpenFileMapping FILE_MAP_READ error %u\n", GetLastError() );
 601     SetLastError(0xdeadbeef);
 602     ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 0 );
 603     if (!ptr)
 604     {
 605         SIZE_T size;
 606         ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 607         SetLastError(0xdeadbeef);
 608         ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 0 );
 609         ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 610         SetLastError(0xdeadbeef);
 611         size = VirtualQuery( ptr, &info, sizeof(info) );
 612         ok( size == sizeof(info),
 613             "VirtualQuery error %u\n", GetLastError() );
 614         ok( info.BaseAddress == ptr, "%p != %p\n", info.BaseAddress, ptr );
 615         ok( info.AllocationBase == ptr, "%p != %p\n", info.AllocationBase, ptr );
 616         ok( info.AllocationProtect == PAGE_READONLY, "%x != PAGE_READONLY\n", info.AllocationProtect );
 617         ok( info.RegionSize == 4096, "%lx != 4096\n", info.RegionSize );
 618         ok( info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State );
 619         ok( info.Protect == PAGE_READONLY, "%x != PAGE_READONLY\n", info.Protect );
 620     }
 621     else win_skip( "no access checks on win9x\n" );
 622     UnmapViewOfFile( ptr );
 623     CloseHandle( mapping );
 624
 625     SetLastError(0xdeadbeef);
 626     mapping = OpenFileMapping( FILE_MAP_WRITE, FALSE, name );
 627     ok( mapping != 0, "OpenFileMapping FILE_MAP_WRITE error %u\n", GetLastError() );
 628     SetLastError(0xdeadbeef);
 629     ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 0 );
 630     if (!ptr)
 631     {
 632         SIZE_T size;
 633         ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 634         SetLastError(0xdeadbeef);
 635         ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 0 );
 636         ok( ptr != NULL, "MapViewOfFile FILE_MAP_WRITE error %u\n", GetLastError() );
 637         SetLastError(0xdeadbeef);
 638         size = VirtualQuery( ptr, &info, sizeof(info) );
 639         ok( size == sizeof(info),
 640             "VirtualQuery error %u\n", GetLastError() );
 641         ok( info.BaseAddress == ptr, "%p != %p\n", info.BaseAddress, ptr );
 642         ok( info.AllocationBase == ptr, "%p != %p\n", info.AllocationBase, ptr );
 643         ok( info.AllocationProtect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.AllocationProtect );
 644         ok( info.RegionSize == 4096, "%lx != 4096\n", info.RegionSize );
 645         ok( info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State );
 646         ok( info.Protect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.Protect );
 647     }
 648     else win_skip( "no access checks on win9x\n" );
 649     UnmapViewOfFile( ptr );
 650     CloseHandle( mapping );
 651
 652     CloseHandle( file );
 653
 654     /* read/write mapping with SEC_RESERVE */
 655     mapping = CreateFileMappingA(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE | SEC_RESERVE, 0, MAPPING_SIZE, NULL);
 656     ok(mapping != INVALID_HANDLE_VALUE, "CreateFileMappingA failed with error %d\n", GetLastError());
 657
 658     ptr = MapViewOfFile(mapping, FILE_MAP_WRITE, 0, 0, 0);
 659     ok(ptr != NULL, "MapViewOfFile failed with error %d\n", GetLastError());
 660
 661     ptr2 = MapViewOfFile(mapping, FILE_MAP_WRITE, 0, 0, 0);
 662     /* on NT ptr != ptr2 but on Win9x ptr == ptr2 */
 663     ok(ptr2 != NULL, "MapViewOfFile failed with error %d\n", GetLastError());
 664     trace("mapping same section resulted in views %p and %p\n", ptr, ptr2);
 665
 666     ret = VirtualQuery(ptr, &info, sizeof(info));
 667     ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
 668     ok(info.BaseAddress == ptr, "BaseAddress should have been %p but was %p instead\n", ptr, info.BaseAddress);
 669     ok(info.AllocationBase == ptr, "AllocationBase should have been %p but was %p instead\n", ptr, info.AllocationBase);
 670     ok(info.RegionSize == MAPPING_SIZE, "RegionSize should have been 0x%x but was 0x%x\n", MAPPING_SIZE, (unsigned int)info.RegionSize);
 671     ok(info.State == MEM_RESERVE, "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
 672     if (info.Type == MEM_PRIVATE)  /* win9x is different for uncommitted mappings */
 673     {
 674         ok(info.AllocationProtect == PAGE_NOACCESS,
 675            "AllocationProtect should have been PAGE_NOACCESS but was 0x%x\n", info.AllocationProtect);
 676         ok(info.Protect == PAGE_NOACCESS,
 677            "Protect should have been PAGE_NOACCESS instead of 0x%x\n", info.Protect);
 678     }
 679     else
 680     {
 681         ok(info.AllocationProtect == PAGE_READWRITE,
 682            "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
 683         ok(info.Protect == 0, "Protect should have been 0 instead of 0x%x\n", info.Protect);
 684         ok(info.Type == MEM_MAPPED, "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
 685     }
 686
 687     if (ptr != ptr2)
 688     {
 689         ret = VirtualQuery(ptr2, &info, sizeof(info));
 690         ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
 691         ok(info.BaseAddress == ptr2,
 692            "BaseAddress should have been %p but was %p instead\n", ptr2, info.BaseAddress);
 693         ok(info.AllocationBase == ptr2,
 694            "AllocationBase should have been %p but was %p instead\n", ptr2, info.AllocationBase);
 695         ok(info.AllocationProtect == PAGE_READWRITE,
 696            "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
 697         ok(info.RegionSize == MAPPING_SIZE,
 698            "RegionSize should have been 0x%x but was 0x%x\n", MAPPING_SIZE, (unsigned int)info.RegionSize);
 699         ok(info.State == MEM_RESERVE,
 700            "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
 701         ok(info.Protect == 0,
 702            "Protect should have been 0 instead of 0x%x\n", info.Protect);
 703         ok(info.Type == MEM_MAPPED,
 704            "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
 705     }
 706
 707     ptr = VirtualAlloc(ptr, 0x10000, MEM_COMMIT, PAGE_READONLY);
 708     ok(ptr != NULL, "VirtualAlloc failed with error %d\n", GetLastError());
 709
 710     ret = VirtualQuery(ptr, &info, sizeof(info));
 711     ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
 712     ok(info.BaseAddress == ptr, "BaseAddress should have been %p but was %p instead\n", ptr, info.BaseAddress);
 713     ok(info.AllocationBase == ptr, "AllocationBase should have been %p but was %p instead\n", ptr, info.AllocationBase);
 714     ok(info.RegionSize == 0x10000, "RegionSize should have been 0x10000 but was 0x%x\n", (unsigned int)info.RegionSize);
 715     ok(info.State == MEM_COMMIT, "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
 716     ok(info.Protect == PAGE_READONLY, "Protect should have been 0 instead of 0x%x\n", info.Protect);
 717     if (info.Type == MEM_PRIVATE)  /* win9x is different for uncommitted mappings */
 718     {
 719         ok(info.AllocationProtect == PAGE_NOACCESS,
 720            "AllocationProtect should have been PAGE_NOACCESS but was 0x%x\n", info.AllocationProtect);
 721     }
 722     else
 723     {
 724         ok(info.AllocationProtect == PAGE_READWRITE,
 725            "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
 726         ok(info.Type == MEM_MAPPED, "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
 727     }
 728
 729     /* shows that the VirtualAlloc above affects the mapping, not just the
 730      * virtual memory in this process - it also affects all other processes
 731      * with a view of the mapping, but that isn't tested here */
 732     if (ptr != ptr2)
 733     {
 734         ret = VirtualQuery(ptr2, &info, sizeof(info));
 735         ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
 736         ok(info.BaseAddress == ptr2,
 737            "BaseAddress should have been %p but was %p instead\n", ptr2, info.BaseAddress);
 738         ok(info.AllocationBase == ptr2,
 739            "AllocationBase should have been %p but was %p instead\n", ptr2, info.AllocationBase);
 740         ok(info.AllocationProtect == PAGE_READWRITE,
 741            "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
 742         ok(info.RegionSize == 0x10000,
 743            "RegionSize should have been 0x10000 but was 0x%x\n", (unsigned int)info.RegionSize);
 744         ok(info.State == MEM_COMMIT,
 745            "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
 746         ok(info.Protect == PAGE_READWRITE,
 747            "Protect should have been 0 instead of 0x%x\n", info.Protect);
 748         ok(info.Type == MEM_MAPPED, "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
 749     }
 750
 751     addr = VirtualAlloc( ptr, MAPPING_SIZE, MEM_RESET, PAGE_READONLY );
 752     ok( addr == ptr || broken(!addr && GetLastError() == ERROR_INVALID_PARAMETER), /* win9x */
 753         "VirtualAlloc failed with error %u\n", GetLastError() );
 754
 755     ret = VirtualFree( ptr, 0x10000, MEM_DECOMMIT );
 756     ok( !ret || broken(ret) /* win9x */, "VirtualFree succeeded\n" );
 757     if (!ret)
 758         ok( GetLastError() == ERROR_INVALID_PARAMETER, "VirtualFree failed with %u\n", GetLastError() );
 759
 760     ret = UnmapViewOfFile(ptr2);
 761     ok(ret, "UnmapViewOfFile failed with error %d\n", GetLastError());
 762     ret = UnmapViewOfFile(ptr);
 763     ok(ret, "UnmapViewOfFile failed with error %d\n", GetLastError());
 764     CloseHandle(mapping);
 765
 766     addr = VirtualAlloc(NULL, 0x10000, MEM_COMMIT, PAGE_READONLY );
 767     ok( addr != NULL, "VirtualAlloc failed with error %u\n", GetLastError() );
 768
 769     SetLastError(0xdeadbeef);
 770     ok( !UnmapViewOfFile(addr), "UnmapViewOfFile should fail on VirtualAlloc mem\n" );
 771     ok( GetLastError() == ERROR_INVALID_ADDRESS,
 772         "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
 773     SetLastError(0xdeadbeef);
 774     ok( !UnmapViewOfFile((char *)addr + 0x3000), "UnmapViewOfFile should fail on VirtualAlloc mem\n" );
 775     ok( GetLastError() == ERROR_INVALID_ADDRESS,
 776         "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
 777     SetLastError(0xdeadbeef);
 778     ok( !UnmapViewOfFile((void *)0xdeadbeef), "UnmapViewOfFile should fail on VirtualAlloc mem\n" );
 779     ok( GetLastError() == ERROR_INVALID_ADDRESS,
 780        "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
 781
 782     ok( VirtualFree(addr, 0, MEM_RELEASE), "VirtualFree failed\n" );
 783 }
 784
 785 static void test_NtMapViewOfSection(void)
 786 {
 787     HANDLE hProcess;
 788
 789     static const char testfile[] = "testfile.xxx";
 790     static const char data[] = "test data for NtMapViewOfSection";
 791     char buffer[sizeof(data)];
 792     HANDLE file, mapping;
 793     void *ptr;
 794     BOOL ret;
 795     DWORD status, written;
 796     SIZE_T size, result;
 797     LARGE_INTEGER offset;
 798
 799     if (!pNtMapViewOfSection || !pNtUnmapViewOfSection)
 800     {
 801         win_skip( "NtMapViewOfSection not available\n" );
 802         return;
 803     }
 804
 805     file = CreateFileA( testfile, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0 );
 806     ok( file != INVALID_HANDLE_VALUE, "Failed to create test file\n" );
 807     WriteFile( file, data, sizeof(data), &written, NULL );
 808     SetFilePointer( file, 4096, NULL, FILE_BEGIN );
 809     SetEndOfFile( file );
 810
 811     /* read/write mapping */
 812
 813     mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
 814     ok( mapping != 0, "CreateFileMapping failed\n" );
 815
 816     hProcess = create_target_process("sleep");
 817     ok(hProcess != NULL, "Can't start process\n");
 818
 819     ptr = NULL;
 820     size = 0;
 821     offset.QuadPart = 0;
 822     status = pNtMapViewOfSection( mapping, hProcess, &ptr, 0, 0, &offset, &size, 1, 0, PAGE_READWRITE );
 823     ok( !status, "NtMapViewOfSection failed status %x\n", status );
 824
 825     ret = ReadProcessMemory( hProcess, ptr, buffer, sizeof(buffer), &result );
 826     ok( ret, "ReadProcessMemory failed\n" );
 827     ok( result == sizeof(buffer), "ReadProcessMemory didn't read all data (%lx)\n", result );
 828     ok( !memcmp( buffer, data, sizeof(buffer) ), "Wrong data read\n" );
 829
 830     status = pNtUnmapViewOfSection( hProcess, ptr );
 831     ok( !status, "NtUnmapViewOfSection failed status %x\n", status );
 832
 833     CloseHandle( mapping );
 834     CloseHandle( file );
 835     DeleteFileA( testfile );
 836
 837     TerminateProcess(hProcess, 0);
 838     CloseHandle(hProcess);
 839 }
 840
 841 static void test_NtAreMappedFilesTheSame(void)
 842 {
 843     static const char testfile[] = "testfile.xxx";
 844     HANDLE file, file2, mapping, map2;
 845     void *ptr, *ptr2;
 846     NTSTATUS status;
 847     char path[MAX_PATH];
 848
 849     if (!pNtAreMappedFilesTheSame)
 850     {
 851         win_skip( "NtAreMappedFilesTheSame not available\n" );
 852         return;
 853     }
 854
 855     file = CreateFileA( testfile, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE,
 856                         NULL, CREATE_ALWAYS, 0, 0 );
 857     ok( file != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
 858     SetFilePointer( file, 4096, NULL, FILE_BEGIN );
 859     SetEndOfFile( file );
 860
 861     mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
 862     ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
 863
 864     ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
 865     ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 866
 867     file2 = CreateFileA( testfile, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE,
 868                          NULL, OPEN_EXISTING, 0, 0 );
 869     ok( file2 != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
 870
 871     map2 = CreateFileMappingA( file2, NULL, PAGE_READONLY, 0, 4096, NULL );
 872     ok( map2 != 0, "CreateFileMapping error %u\n", GetLastError() );
 873     ptr2 = MapViewOfFile( map2, FILE_MAP_READ, 0, 0, 4096 );
 874     ok( ptr2 != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 875     status = pNtAreMappedFilesTheSame( ptr, ptr2 );
 876     ok( status == STATUS_NOT_SAME_DEVICE, "NtAreMappedFilesTheSame returned %x\n", status );
 877     UnmapViewOfFile( ptr2 );
 878
 879     ptr2 = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
 880     ok( ptr2 != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 881     status = pNtAreMappedFilesTheSame( ptr, ptr2 );
 882     ok( status == STATUS_NOT_SAME_DEVICE, "NtAreMappedFilesTheSame returned %x\n", status );
 883     UnmapViewOfFile( ptr2 );
 884     CloseHandle( map2 );
 885
 886     map2 = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL );
 887     ok( map2 != 0, "CreateFileMapping error %u\n", GetLastError() );
 888     ptr2 = MapViewOfFile( map2, FILE_MAP_READ, 0, 0, 4096 );
 889     ok( ptr2 != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 890     status = pNtAreMappedFilesTheSame( ptr, ptr2 );
 891     ok( status == STATUS_NOT_SAME_DEVICE, "NtAreMappedFilesTheSame returned %x\n", status );
 892     UnmapViewOfFile( ptr2 );
 893     CloseHandle( map2 );
 894     CloseHandle( file2 );
 895
 896     status = pNtAreMappedFilesTheSame( ptr, ptr );
 897     ok( status == STATUS_NOT_SAME_DEVICE, "NtAreMappedFilesTheSame returned %x\n", status );
 898
 899     status = pNtAreMappedFilesTheSame( ptr, (char *)ptr + 30 );
 900     ok( status == STATUS_NOT_SAME_DEVICE, "NtAreMappedFilesTheSame returned %x\n", status );
 901
 902     status = pNtAreMappedFilesTheSame( ptr, GetModuleHandleA("kernel32.dll") );
 903     ok( status == STATUS_NOT_SAME_DEVICE, "NtAreMappedFilesTheSame returned %x\n", status );
 904
 905     status = pNtAreMappedFilesTheSame( ptr, (void *)0xdeadbeef );
 906     ok( status == STATUS_CONFLICTING_ADDRESSES || status == STATUS_INVALID_ADDRESS,
 907         "NtAreMappedFilesTheSame returned %x\n", status );
 908
 909     status = pNtAreMappedFilesTheSame( ptr, NULL );
 910     ok( status == STATUS_INVALID_ADDRESS, "NtAreMappedFilesTheSame returned %x\n", status );
 911
 912     status = pNtAreMappedFilesTheSame( ptr, (void *)GetProcessHeap() );
 913     ok( status == STATUS_CONFLICTING_ADDRESSES, "NtAreMappedFilesTheSame returned %x\n", status );
 914
 915     status = pNtAreMappedFilesTheSame( NULL, NULL );
 916     ok( status == STATUS_INVALID_ADDRESS, "NtAreMappedFilesTheSame returned %x\n", status );
 917
 918     ptr2 = VirtualAlloc( NULL, 0x10000, MEM_COMMIT, PAGE_READWRITE );
 919     ok( ptr2 != NULL, "VirtualAlloc error %u\n", GetLastError() );
 920     status = pNtAreMappedFilesTheSame( ptr, ptr2 );
 921     ok( status == STATUS_CONFLICTING_ADDRESSES, "NtAreMappedFilesTheSame returned %x\n", status );
 922     VirtualFree( ptr2, 0, MEM_RELEASE );
 923
 924     UnmapViewOfFile( ptr );
 925     CloseHandle( mapping );
 926     CloseHandle( file );
 927
 928     status = pNtAreMappedFilesTheSame( GetModuleHandleA("ntdll.dll"),
 929                                        GetModuleHandleA("kernel32.dll") );
 930     ok( status == STATUS_NOT_SAME_DEVICE, "NtAreMappedFilesTheSame returned %x\n", status );
 931     status = pNtAreMappedFilesTheSame( GetModuleHandleA("kernel32.dll"),
 932                                        GetModuleHandleA("kernel32.dll") );
 933     ok( status == STATUS_SUCCESS, "NtAreMappedFilesTheSame returned %x\n", status );
 934     status = pNtAreMappedFilesTheSame( GetModuleHandleA("kernel32.dll"),
 935                                        (char *)GetModuleHandleA("kernel32.dll") + 4096 );
 936     ok( status == STATUS_SUCCESS, "NtAreMappedFilesTheSame returned %x\n", status );
 937
 938     GetSystemDirectoryA( path, MAX_PATH );
 939     strcat( path, "\\kernel32.dll" );
 940     file = CreateFileA( path, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, 0 );
 941     ok( file != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
 942
 943     mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL );
 944     ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
 945     ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
 946     ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 947     status = pNtAreMappedFilesTheSame( ptr, GetModuleHandleA("kernel32.dll") );
 948     ok( status == STATUS_NOT_SAME_DEVICE, "NtAreMappedFilesTheSame returned %x\n", status );
 949     UnmapViewOfFile( ptr );
 950     CloseHandle( mapping );
 951
 952     mapping = CreateFileMappingA( file, NULL, PAGE_READONLY | SEC_IMAGE, 0, 0, NULL );
 953     ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
 954     ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 0 );
 955     ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 956     status = pNtAreMappedFilesTheSame( ptr, GetModuleHandleA("kernel32.dll") );
 957     todo_wine
 958     ok( status == STATUS_SUCCESS, "NtAreMappedFilesTheSame returned %x\n", status );
 959
 960     file2 = CreateFileA( path, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, 0 );
 961     ok( file2 != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
 962     map2 = CreateFileMappingA( file2, NULL, PAGE_READONLY | SEC_IMAGE, 0, 0, NULL );
 963     ok( map2 != 0, "CreateFileMapping error %u\n", GetLastError() );
 964     ptr2 = MapViewOfFile( map2, FILE_MAP_READ, 0, 0, 0 );
 965     ok( ptr2 != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 966     status = pNtAreMappedFilesTheSame( ptr, ptr2 );
 967     ok( status == STATUS_SUCCESS, "NtAreMappedFilesTheSame returned %x\n", status );
 968     UnmapViewOfFile( ptr2 );
 969     CloseHandle( map2 );
 970     CloseHandle( file2 );
 971
 972     UnmapViewOfFile( ptr );
 973     CloseHandle( mapping );
 974
 975     CloseHandle( file );
 976     DeleteFileA( testfile );
 977 }
 978
 979 static void test_CreateFileMapping(void)
 980 {
 981     HANDLE handle, handle2;
 982
 983     /* test case sensitivity */
 984
 985     SetLastError(0xdeadbeef);
 986     handle = CreateFileMappingA( INVALID_HANDLE_VALUE, NULL, SEC_COMMIT | PAGE_READWRITE, 0, 0x1000,
 987                                  "Wine Test Mapping");
 988     ok( handle != NULL, "CreateFileMapping failed with error %u\n", GetLastError());
 989     ok( GetLastError() == 0, "wrong error %u\n", GetLastError());
 990
 991     SetLastError(0xdeadbeef);
 992     handle2 = CreateFileMappingA( INVALID_HANDLE_VALUE, NULL, SEC_COMMIT | PAGE_READWRITE, 0, 0x1000,
 993                                   "Wine Test Mapping");
 994     ok( handle2 != NULL, "CreateFileMapping failed with error %d\n", GetLastError());
 995     ok( GetLastError() == ERROR_ALREADY_EXISTS, "wrong error %u\n", GetLastError());
 996     CloseHandle( handle2 );
 997
 998     SetLastError(0xdeadbeef);
 999     handle2 = CreateFileMappingA( INVALID_HANDLE_VALUE, NULL, SEC_COMMIT | PAGE_READWRITE, 0, 0x1000,
1000                                  "WINE TEST MAPPING");
1001     ok( handle2 != NULL, "CreateFileMapping failed with error %d\n", GetLastError());
1002     ok( GetLastError() == 0, "wrong error %u\n", GetLastError());
1003     CloseHandle( handle2 );
1004
1005     SetLastError(0xdeadbeef);
1006     handle2 = OpenFileMappingA( FILE_MAP_ALL_ACCESS, FALSE, "Wine Test Mapping");
1007     ok( handle2 != NULL, "OpenFileMapping failed with error %d\n", GetLastError());
1008     CloseHandle( handle2 );
1009
1010     SetLastError(0xdeadbeef);
1011     handle2 = OpenFileMappingA( FILE_MAP_ALL_ACCESS, FALSE, "WINE TEST MAPPING");
1012     ok( !handle2, "OpenFileMapping succeeded\n");
1013     ok( GetLastError() == ERROR_FILE_NOT_FOUND || GetLastError() == ERROR_INVALID_NAME /* win9x */,
1014         "wrong error %u\n", GetLastError());
1015
1016     CloseHandle( handle );
1017 }
1018
1019 static void test_IsBadReadPtr(void)
1020 {
1021     BOOL ret;
1022     void *ptr = (void *)0xdeadbeef;
1023     char stackvar;
1024
1025     ret = IsBadReadPtr(NULL, 0);
1026     ok(ret == FALSE, "Expected IsBadReadPtr to return FALSE, got %d\n", ret);
1027
1028     ret = IsBadReadPtr(NULL, 1);
1029     ok(ret == TRUE, "Expected IsBadReadPtr to return TRUE, got %d\n", ret);
1030
1031     ret = IsBadReadPtr(ptr, 0);
1032     ok(ret == FALSE, "Expected IsBadReadPtr to return FALSE, got %d\n", ret);
1033
1034     ret = IsBadReadPtr(ptr, 1);
1035     ok(ret == TRUE, "Expected IsBadReadPtr to return TRUE, got %d\n", ret);
1036
1037     ret = IsBadReadPtr(&stackvar, 0);
1038     ok(ret == FALSE, "Expected IsBadReadPtr to return FALSE, got %d\n", ret);
1039
1040     ret = IsBadReadPtr(&stackvar, sizeof(char));
1041     ok(ret == FALSE, "Expected IsBadReadPtr to return FALSE, got %d\n", ret);
1042 }
1043
1044 static void test_IsBadWritePtr(void)
1045 {
1046     BOOL ret;
1047     void *ptr = (void *)0xdeadbeef;
1048     char stackval;
1049
1050     ret = IsBadWritePtr(NULL, 0);
1051     ok(ret == FALSE, "Expected IsBadWritePtr to return FALSE, got %d\n", ret);
1052
1053     ret = IsBadWritePtr(NULL, 1);
1054     ok(ret == TRUE, "Expected IsBadWritePtr to return TRUE, got %d\n", ret);
1055
1056     ret = IsBadWritePtr(ptr, 0);
1057     ok(ret == FALSE, "Expected IsBadWritePtr to return FALSE, got %d\n", ret);
1058
1059     ret = IsBadWritePtr(ptr, 1);
1060     ok(ret == TRUE, "Expected IsBadWritePtr to return TRUE, got %d\n", ret);
1061
1062     ret = IsBadWritePtr(&stackval, 0);
1063     ok(ret == FALSE, "Expected IsBadWritePtr to return FALSE, got %d\n", ret);
1064
1065     ret = IsBadWritePtr(&stackval, sizeof(char));
1066     ok(ret == FALSE, "Expected IsBadWritePtr to return FALSE, got %d\n", ret);
1067 }
1068
1069 static void test_IsBadCodePtr(void)
1070 {
1071     BOOL ret;
1072     void *ptr = (void *)0xdeadbeef;
1073     char stackval;
1074
1075     ret = IsBadCodePtr(NULL);
1076     ok(ret == TRUE, "Expected IsBadCodePtr to return TRUE, got %d\n", ret);
1077
1078     ret = IsBadCodePtr(ptr);
1079     ok(ret == TRUE, "Expected IsBadCodePtr to return TRUE, got %d\n", ret);
1080
1081     ret = IsBadCodePtr((void *)&stackval);
1082     ok(ret == FALSE, "Expected IsBadCodePtr to return FALSE, got %d\n", ret);
1083 }
1084
1085 static void test_write_watch(void)
1086 {
1087     char *base;
1088     DWORD ret, size, old_prot;
1089     MEMORY_BASIC_INFORMATION info;
1090     void *results[64];
1091     ULONG_PTR count;
1092     ULONG pagesize;
1093
1094     if (!pGetWriteWatch || !pResetWriteWatch)
1095     {
1096         win_skip( "GetWriteWatch not supported\n" );
1097         return;
1098     }
1099
1100     size = 0x10000;
1101     base = VirtualAlloc( 0, size, MEM_RESERVE | MEM_COMMIT | MEM_WRITE_WATCH, PAGE_READWRITE );
1102     if (!base &&
1103         (GetLastError() == ERROR_INVALID_PARAMETER || GetLastError() == ERROR_NOT_SUPPORTED))
1104     {
1105         win_skip( "MEM_WRITE_WATCH not supported\n" );
1106         return;
1107     }
1108     ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
1109     ret = VirtualQuery( base, &info, sizeof(info) );
1110     ok(ret, "VirtualQuery failed %u\n", GetLastError());
1111     ok( info.BaseAddress == base, "BaseAddress %p instead of %p\n", info.BaseAddress, base );
1112     ok( info.AllocationProtect == PAGE_READWRITE, "wrong AllocationProtect %x\n", info.AllocationProtect );
1113     ok( info.RegionSize == size, "wrong RegionSize 0x%lx\n", info.RegionSize );
1114     ok( info.State == MEM_COMMIT, "wrong State 0x%x\n", info.State );
1115     ok( info.Protect == PAGE_READWRITE, "wrong Protect 0x%x\n", info.Protect );
1116     ok( info.Type == MEM_PRIVATE, "wrong Type 0x%x\n", info.Type );
1117
1118     count = 64;
1119     SetLastError( 0xdeadbeef );
1120     ret = pGetWriteWatch( 0, NULL, size, results, &count, &pagesize );
1121     ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1122     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
1123         broken( GetLastError() == 0xdeadbeef ), /* win98 */
1124         "wrong error %u\n", GetLastError() );
1125
1126     SetLastError( 0xdeadbeef );
1127     ret = pGetWriteWatch( 0, GetModuleHandle(0), size, results, &count, &pagesize );
1128     if (ret)
1129     {
1130         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1131         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1132     }
1133     else  /* win98 */
1134     {
1135         ok( count == 0, "wrong count %lu\n", count );
1136     }
1137
1138     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1139     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1140     ok( count == 0, "wrong count %lu\n", count );
1141
1142     base[pagesize + 1] = 0x44;
1143
1144     count = 64;
1145     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1146     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1147     ok( count == 1, "wrong count %lu\n", count );
1148     ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
1149
1150     count = 64;
1151     ret = pGetWriteWatch( WRITE_WATCH_FLAG_RESET, base, size, results, &count, &pagesize );
1152     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1153     ok( count == 1, "wrong count %lu\n", count );
1154     ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
1155
1156     count = 64;
1157     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1158     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1159     ok( count == 0, "wrong count %lu\n", count );
1160
1161     base[2*pagesize + 3] = 0x11;
1162     base[4*pagesize + 8] = 0x11;
1163
1164     count = 64;
1165     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1166     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1167     ok( count == 2, "wrong count %lu\n", count );
1168     ok( results[0] == base + 2*pagesize, "wrong result %p\n", results[0] );
1169     ok( results[1] == base + 4*pagesize, "wrong result %p\n", results[1] );
1170
1171     count = 64;
1172     ret = pGetWriteWatch( 0, base + 3*pagesize, 2*pagesize, results, &count, &pagesize );
1173     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1174     ok( count == 1, "wrong count %lu\n", count );
1175     ok( results[0] == base + 4*pagesize, "wrong result %p\n", results[0] );
1176
1177     ret = pResetWriteWatch( base, 3*pagesize );
1178     ok( !ret, "pResetWriteWatch failed %u\n", GetLastError() );
1179
1180     count = 64;
1181     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1182     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1183     ok( count == 1, "wrong count %lu\n", count );
1184     ok( results[0] == base + 4*pagesize, "wrong result %p\n", results[0] );
1185
1186     *(DWORD *)(base + 2*pagesize - 2) = 0xdeadbeef;
1187
1188     count = 64;
1189     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1190     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1191     ok( count == 3, "wrong count %lu\n", count );
1192     ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
1193     ok( results[1] == base + 2*pagesize, "wrong result %p\n", results[1] );
1194     ok( results[2] == base + 4*pagesize, "wrong result %p\n", results[2] );
1195
1196     count = 1;
1197     ret = pGetWriteWatch( WRITE_WATCH_FLAG_RESET, base, size, results, &count, &pagesize );
1198     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1199     ok( count == 1, "wrong count %lu\n", count );
1200     ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
1201
1202     count = 64;
1203     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1204     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1205     ok( count == 2, "wrong count %lu\n", count );
1206     ok( results[0] == base + 2*pagesize, "wrong result %p\n", results[0] );
1207     ok( results[1] == base + 4*pagesize, "wrong result %p\n", results[1] );
1208
1209     /* changing protections doesn't affect watches */
1210
1211     ret = VirtualProtect( base, 3*pagesize, PAGE_READONLY, &old_prot );
1212     ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
1213     ok( old_prot == PAGE_READWRITE, "wrong old prot %x\n", old_prot );
1214
1215     ret = VirtualQuery( base, &info, sizeof(info) );
1216     ok(ret, "VirtualQuery failed %u\n", GetLastError());
1217     ok( info.BaseAddress == base, "BaseAddress %p instead of %p\n", info.BaseAddress, base );
1218     ok( info.RegionSize == 3*pagesize, "wrong RegionSize 0x%lx\n", info.RegionSize );
1219     ok( info.State == MEM_COMMIT, "wrong State 0x%x\n", info.State );
1220     ok( info.Protect == PAGE_READONLY, "wrong Protect 0x%x\n", info.Protect );
1221
1222     ret = VirtualProtect( base, 3*pagesize, PAGE_READWRITE, &old_prot );
1223     ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
1224     ok( old_prot == PAGE_READONLY, "wrong old prot %x\n", old_prot );
1225
1226     count = 64;
1227     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1228     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1229     ok( count == 2, "wrong count %lu\n", count );
1230     ok( results[0] == base + 2*pagesize, "wrong result %p\n", results[0] );
1231     ok( results[1] == base + 4*pagesize, "wrong result %p\n", results[1] );
1232
1233     ret = VirtualQuery( base, &info, sizeof(info) );
1234     ok(ret, "VirtualQuery failed %u\n", GetLastError());
1235     ok( info.BaseAddress == base, "BaseAddress %p instead of %p\n", info.BaseAddress, base );
1236     ok( info.RegionSize == size, "wrong RegionSize 0x%lx\n", info.RegionSize );
1237     ok( info.State == MEM_COMMIT, "wrong State 0x%x\n", info.State );
1238     ok( info.Protect == PAGE_READWRITE, "wrong Protect 0x%x\n", info.Protect );
1239
1240     /* some invalid parameter tests */
1241
1242     SetLastError( 0xdeadbeef );
1243     count = 0;
1244     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1245     if (ret)
1246     {
1247         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1248         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1249
1250         SetLastError( 0xdeadbeef );
1251         ret = pGetWriteWatch( 0, base, size, results, NULL, &pagesize );
1252         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1253         ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
1254
1255         SetLastError( 0xdeadbeef );
1256         count = 64;
1257         ret = pGetWriteWatch( 0, base, size, results, &count, NULL );
1258         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1259         ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
1260
1261         SetLastError( 0xdeadbeef );
1262         count = 64;
1263         ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1264         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1265         ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
1266
1267         SetLastError( 0xdeadbeef );
1268         count = 0;
1269         ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1270         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1271         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1272
1273         SetLastError( 0xdeadbeef );
1274         count = 64;
1275         ret = pGetWriteWatch( 0xdeadbeef, base, size, results, &count, &pagesize );
1276         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1277         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1278
1279         SetLastError( 0xdeadbeef );
1280         count = 64;
1281         ret = pGetWriteWatch( 0, base, 0, results, &count, &pagesize );
1282         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1283         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1284
1285         SetLastError( 0xdeadbeef );
1286         count = 64;
1287         ret = pGetWriteWatch( 0, base, size * 2, results, &count, &pagesize );
1288         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1289         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1290
1291         SetLastError( 0xdeadbeef );
1292         count = 64;
1293         ret = pGetWriteWatch( 0, base + size - pagesize, pagesize + 1, results, &count, &pagesize );
1294         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1295         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1296
1297         SetLastError( 0xdeadbeef );
1298         ret = pResetWriteWatch( base, 0 );
1299         ok( ret == ~0u, "ResetWriteWatch succeeded %u\n", ret );
1300         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1301
1302         SetLastError( 0xdeadbeef );
1303         ret = pResetWriteWatch( GetModuleHandle(0), size );
1304         ok( ret == ~0u, "ResetWriteWatch succeeded %u\n", ret );
1305         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1306     }
1307     else  /* win98 is completely different */
1308     {
1309         SetLastError( 0xdeadbeef );
1310         count = 64;
1311         ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1312         ok( ret == ERROR_INVALID_PARAMETER, "GetWriteWatch succeeded %u\n", ret );
1313         ok( GetLastError() == 0xdeadbeef, "wrong error %u\n", GetLastError() );
1314
1315         count = 0;
1316         ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1317         ok( !ret, "GetWriteWatch failed %u\n", ret );
1318
1319         count = 64;
1320         ret = pGetWriteWatch( 0xdeadbeef, base, size, results, &count, &pagesize );
1321         ok( !ret, "GetWriteWatch failed %u\n", ret );
1322
1323         count = 64;
1324         ret = pGetWriteWatch( 0, base, 0, results, &count, &pagesize );
1325         ok( !ret, "GetWriteWatch failed %u\n", ret );
1326
1327         ret = pResetWriteWatch( base, 0 );
1328         ok( !ret, "ResetWriteWatch failed %u\n", ret );
1329
1330         ret = pResetWriteWatch( GetModuleHandle(0), size );
1331         ok( !ret, "ResetWriteWatch failed %u\n", ret );
1332     }
1333
1334     VirtualFree( base, 0, MEM_FREE );
1335
1336     base = VirtualAlloc( 0, size, MEM_RESERVE | MEM_WRITE_WATCH, PAGE_READWRITE );
1337     ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
1338     VirtualFree( base, 0, MEM_FREE );
1339
1340     base = VirtualAlloc( 0, size, MEM_WRITE_WATCH, PAGE_READWRITE );
1341     ok( !base, "VirtualAlloc succeeded\n" );
1342     ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1343
1344     /* initial protect doesn't matter */
1345
1346     base = VirtualAlloc( 0, size, MEM_RESERVE | MEM_WRITE_WATCH, PAGE_NOACCESS );
1347     ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
1348     base = VirtualAlloc( base, size, MEM_COMMIT, PAGE_NOACCESS );
1349     ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
1350
1351     count = 64;
1352     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1353     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1354     ok( count == 0, "wrong count %lu\n", count );
1355
1356     ret = VirtualProtect( base, 6*pagesize, PAGE_READWRITE, &old_prot );
1357     ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
1358     ok( old_prot == PAGE_NOACCESS, "wrong old prot %x\n", old_prot );
1359
1360     base[5*pagesize + 200] = 3;
1361
1362     ret = VirtualProtect( base, 6*pagesize, PAGE_NOACCESS, &old_prot );
1363     ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
1364     ok( old_prot == PAGE_READWRITE, "wrong old prot %x\n", old_prot );
1365
1366     count = 64;
1367     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1368     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1369     ok( count == 1, "wrong count %lu\n", count );
1370     ok( results[0] == base + 5*pagesize, "wrong result %p\n", results[0] );
1371
1372     ret = VirtualFree( base, size, MEM_DECOMMIT );
1373     ok( ret, "VirtualFree failed %u\n", GetLastError() );
1374
1375     count = 64;
1376     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1377     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1378     ok( count == 1 || broken(count == 0), /* win98 */
1379         "wrong count %lu\n", count );
1380     if (count) ok( results[0] == base + 5*pagesize, "wrong result %p\n", results[0] );
1381
1382     VirtualFree( base, 0, MEM_FREE );
1383 }
1384
1385 static void test_VirtualProtect(void)
1386 {
1387     static const struct test_data
1388     {
1389         DWORD prot_set, prot_get;
1390     } td[] =
1391     {
1392         { 0, 0 }, /* 0x00 */
1393         { PAGE_NOACCESS, PAGE_NOACCESS }, /* 0x01 */
1394         { PAGE_READONLY, PAGE_READONLY }, /* 0x02 */
1395         { PAGE_READONLY | PAGE_NOACCESS, 0 }, /* 0x03 */
1396         { PAGE_READWRITE, PAGE_READWRITE }, /* 0x04 */
1397         { PAGE_READWRITE | PAGE_NOACCESS, 0 }, /* 0x05 */
1398         { PAGE_READWRITE | PAGE_READONLY, 0 }, /* 0x06 */
1399         { PAGE_READWRITE | PAGE_READONLY | PAGE_NOACCESS, 0 }, /* 0x07 */
1400         { PAGE_WRITECOPY, 0 }, /* 0x08 */
1401         { PAGE_WRITECOPY | PAGE_NOACCESS, 0 }, /* 0x09 */
1402         { PAGE_WRITECOPY | PAGE_READONLY, 0 }, /* 0x0a */
1403         { PAGE_WRITECOPY | PAGE_NOACCESS | PAGE_READONLY, 0 }, /* 0x0b */
1404         { PAGE_WRITECOPY | PAGE_READWRITE, 0 }, /* 0x0c */
1405         { PAGE_WRITECOPY | PAGE_READWRITE | PAGE_NOACCESS, 0 }, /* 0x0d */
1406         { PAGE_WRITECOPY | PAGE_READWRITE | PAGE_READONLY, 0 }, /* 0x0e */
1407         { PAGE_WRITECOPY | PAGE_READWRITE | PAGE_READONLY | PAGE_NOACCESS, 0 }, /* 0x0f */
1408
1409         { PAGE_EXECUTE, PAGE_EXECUTE }, /* 0x10 */
1410         { PAGE_EXECUTE_READ, PAGE_EXECUTE_READ }, /* 0x20 */
1411         { PAGE_EXECUTE_READ | PAGE_EXECUTE, 0 }, /* 0x30 */
1412         { PAGE_EXECUTE_READWRITE, PAGE_EXECUTE_READWRITE }, /* 0x40 */
1413         { PAGE_EXECUTE_READWRITE | PAGE_EXECUTE, 0 }, /* 0x50 */
1414         { PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ, 0 }, /* 0x60 */
1415         { PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ | PAGE_EXECUTE, 0 }, /* 0x70 */
1416         { PAGE_EXECUTE_WRITECOPY, 0 }, /* 0x80 */
1417         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE, 0 }, /* 0x90 */
1418         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READ, 0 }, /* 0xa0 */
1419         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READ | PAGE_EXECUTE, 0 }, /* 0xb0 */
1420         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE, 0 }, /* 0xc0 */
1421         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE | PAGE_EXECUTE, 0 }, /* 0xd0 */
1422         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ, 0 }, /* 0xe0 */
1423         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ | PAGE_EXECUTE, 0 } /* 0xf0 */
1424     };
1425     char *base;
1426     DWORD ret, old_prot, rw_prot, exec_prot, i, j;
1427     MEMORY_BASIC_INFORMATION info;
1428     SYSTEM_INFO si;
1429
1430     GetSystemInfo(&si);
1431     trace("system page size %#x\n", si.dwPageSize);
1432
1433     SetLastError(0xdeadbeef);
1434     base = VirtualAlloc(0, si.dwPageSize, MEM_RESERVE | MEM_COMMIT, PAGE_NOACCESS);
1435     ok(base != NULL, "VirtualAlloc failed %d\n", GetLastError());
1436
1437     for (i = 0; i < sizeof(td)/sizeof(td[0]); i++)
1438     {
1439         SetLastError(0xdeadbeef);
1440         ret = VirtualQuery(base, &info, sizeof(info));
1441         ok(ret, "VirtualQuery failed %d\n", GetLastError());
1442         ok(info.BaseAddress == base, "%d: got %p != expected %p\n", i, info.BaseAddress, base);
1443         ok(info.RegionSize == si.dwPageSize, "%d: got %#lx != expected %#x\n", i, info.RegionSize, si.dwPageSize);
1444         ok(info.Protect == PAGE_NOACCESS, "%d: got %#x != expected PAGE_NOACCESS\n", i, info.Protect);
1445         ok(info.AllocationBase == base, "%d: %p != %p\n", i, info.AllocationBase, base);
1446         ok(info.AllocationProtect == PAGE_NOACCESS, "%d: %#x != PAGE_NOACCESS\n", i, info.AllocationProtect);
1447         ok(info.State == MEM_COMMIT, "%d: %#x != MEM_COMMIT\n", i, info.State);
1448         ok(info.Type == MEM_PRIVATE, "%d: %#x != MEM_PRIVATE\n", i, info.Type);
1449
1450         old_prot = 0xdeadbeef;
1451         SetLastError(0xdeadbeef);
1452         ret = VirtualProtect(base, si.dwPageSize, td[i].prot_set, &old_prot);
1453         if (td[i].prot_get)
1454         {
1455             ok(ret, "%d: VirtualProtect error %d\n", i, GetLastError());
1456             ok(old_prot == PAGE_NOACCESS, "%d: got %#x != expected PAGE_NOACCESS\n", i, old_prot);
1457
1458             SetLastError(0xdeadbeef);
1459             ret = VirtualQuery(base, &info, sizeof(info));
1460             ok(ret, "VirtualQuery failed %d\n", GetLastError());
1461             ok(info.BaseAddress == base, "%d: got %p != expected %p\n", i, info.BaseAddress, base);
1462             ok(info.RegionSize == si.dwPageSize, "%d: got %#lx != expected %#x\n", i, info.RegionSize, si.dwPageSize);
1463             ok(info.Protect == td[i].prot_get, "%d: got %#x != expected %#x\n", i, info.Protect, td[i].prot_get);
1464             ok(info.AllocationBase == base, "%d: %p != %p\n", i, info.AllocationBase, base);
1465             ok(info.AllocationProtect == PAGE_NOACCESS, "%d: %#x != PAGE_NOACCESS\n", i, info.AllocationProtect);
1466             ok(info.State == MEM_COMMIT, "%d: %#x != MEM_COMMIT\n", i, info.State);
1467             ok(info.Type == MEM_PRIVATE, "%d: %#x != MEM_PRIVATE\n", i, info.Type);
1468         }
1469         else
1470         {
1471             ok(!ret, "%d: VirtualProtect should fail\n", i);
1472             ok(GetLastError() == ERROR_INVALID_PARAMETER, "%d: expected ERROR_INVALID_PARAMETER, got %d\n", i, GetLastError());
1473         }
1474
1475         old_prot = 0xdeadbeef;
1476         SetLastError(0xdeadbeef);
1477         ret = VirtualProtect(base, si.dwPageSize, PAGE_NOACCESS, &old_prot);
1478         ok(ret, "%d: VirtualProtect error %d\n", i, GetLastError());
1479         if (td[i].prot_get)
1480             ok(old_prot == td[i].prot_get, "%d: got %#x != expected %#x\n", i, old_prot, td[i].prot_get);
1481         else
1482             ok(old_prot == PAGE_NOACCESS, "%d: got %#x != expected PAGE_NOACCESS\n", i, old_prot);
1483     }
1484
1485     exec_prot = 0;
1486
1487     for (i = 0; i <= 4; i++)
1488     {
1489         rw_prot = 0;
1490
1491         for (j = 0; j <= 4; j++)
1492         {
1493             DWORD prot = exec_prot | rw_prot;
1494
1495             SetLastError(0xdeadbeef);
1496             ret = VirtualProtect(base, si.dwPageSize, prot, &old_prot);
1497             if ((rw_prot && exec_prot) || (!rw_prot && !exec_prot))
1498             {
1499                 ok(!ret, "VirtualProtect(%02x) should fail\n", prot);
1500                 ok(GetLastError() == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
1501             }
1502             else
1503             {
1504                 if (prot & (PAGE_WRITECOPY | PAGE_EXECUTE_WRITECOPY))
1505                 {
1506                     ok(!ret, "VirtualProtect(%02x) should fail\n", prot);
1507                     ok(GetLastError() == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
1508                 }
1509                 else
1510                     ok(ret, "VirtualProtect(%02x) error %d\n", prot, GetLastError());
1511             }
1512
1513             rw_prot = 1 << j;
1514         }
1515
1516         exec_prot = 1 << (i + 4);
1517     }
1518
1519     VirtualFree(base, 0, MEM_FREE);
1520 }
1521
1522 static BOOL is_mem_writable(DWORD prot)
1523 {
1524     switch (prot & 0xff)
1525     {
1526         case PAGE_READWRITE:
1527         case PAGE_WRITECOPY:
1528         case PAGE_EXECUTE_READWRITE:
1529         case PAGE_EXECUTE_WRITECOPY:
1530             return TRUE;
1531
1532         default:
1533             return FALSE;
1534     }
1535 }
1536
1537 static void test_VirtualAlloc_protection(void)
1538 {
1539     static const struct test_data
1540     {
1541         DWORD prot;
1542         BOOL success;
1543     } td[] =
1544     {
1545         { 0, FALSE }, /* 0x00 */
1546         { PAGE_NOACCESS, TRUE }, /* 0x01 */
1547         { PAGE_READONLY, TRUE }, /* 0x02 */
1548         { PAGE_READONLY | PAGE_NOACCESS, FALSE }, /* 0x03 */
1549         { PAGE_READWRITE, TRUE }, /* 0x04 */
1550         { PAGE_READWRITE | PAGE_NOACCESS, FALSE }, /* 0x05 */
1551         { PAGE_READWRITE | PAGE_READONLY, FALSE }, /* 0x06 */
1552         { PAGE_READWRITE | PAGE_READONLY | PAGE_NOACCESS, FALSE }, /* 0x07 */
1553         { PAGE_WRITECOPY, FALSE }, /* 0x08 */
1554         { PAGE_WRITECOPY | PAGE_NOACCESS, FALSE }, /* 0x09 */
1555         { PAGE_WRITECOPY | PAGE_READONLY, FALSE }, /* 0x0a */
1556         { PAGE_WRITECOPY | PAGE_NOACCESS | PAGE_READONLY, FALSE }, /* 0x0b */
1557         { PAGE_WRITECOPY | PAGE_READWRITE, FALSE }, /* 0x0c */
1558         { PAGE_WRITECOPY | PAGE_READWRITE | PAGE_NOACCESS, FALSE }, /* 0x0d */
1559         { PAGE_WRITECOPY | PAGE_READWRITE | PAGE_READONLY, FALSE }, /* 0x0e */
1560         { PAGE_WRITECOPY | PAGE_READWRITE | PAGE_READONLY | PAGE_NOACCESS, FALSE }, /* 0x0f */
1561
1562         { PAGE_EXECUTE, TRUE }, /* 0x10 */
1563         { PAGE_EXECUTE_READ, TRUE }, /* 0x20 */
1564         { PAGE_EXECUTE_READ | PAGE_EXECUTE, FALSE }, /* 0x30 */
1565         { PAGE_EXECUTE_READWRITE, TRUE }, /* 0x40 */
1566         { PAGE_EXECUTE_READWRITE | PAGE_EXECUTE, FALSE }, /* 0x50 */
1567         { PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ, FALSE }, /* 0x60 */
1568         { PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ | PAGE_EXECUTE, FALSE }, /* 0x70 */
1569         { PAGE_EXECUTE_WRITECOPY, FALSE }, /* 0x80 */
1570         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE, FALSE }, /* 0x90 */
1571         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READ, FALSE }, /* 0xa0 */
1572         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READ | PAGE_EXECUTE, FALSE }, /* 0xb0 */
1573         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE, FALSE }, /* 0xc0 */
1574         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE | PAGE_EXECUTE, FALSE }, /* 0xd0 */
1575         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ, FALSE }, /* 0xe0 */
1576         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ | PAGE_EXECUTE, FALSE } /* 0xf0 */
1577     };
1578     char *base;
1579     DWORD ret, i;
1580     MEMORY_BASIC_INFORMATION info;
1581     SYSTEM_INFO si;
1582
1583     GetSystemInfo(&si);
1584     trace("system page size %#x\n", si.dwPageSize);
1585
1586     for (i = 0; i < sizeof(td)/sizeof(td[0]); i++)
1587     {
1588         SetLastError(0xdeadbeef);
1589         base = VirtualAlloc(0, si.dwPageSize, MEM_COMMIT, td[i].prot);
1590
1591         if (td[i].success)
1592         {
1593             ok(base != NULL, "%d: VirtualAlloc failed %d\n", i, GetLastError());
1594
1595             SetLastError(0xdeadbeef);
1596             ret = VirtualQuery(base, &info, sizeof(info));
1597             ok(ret, "VirtualQuery failed %d\n", GetLastError());
1598             ok(info.BaseAddress == base, "%d: got %p != expected %p\n", i, info.BaseAddress, base);
1599             ok(info.RegionSize == si.dwPageSize, "%d: got %#lx != expected %#x\n", i, info.RegionSize, si.dwPageSize);
1600             ok(info.Protect == td[i].prot, "%d: got %#x != expected %#x\n", i, info.Protect, td[i].prot);
1601             ok(info.AllocationBase == base, "%d: %p != %p\n", i, info.AllocationBase, base);
1602             ok(info.AllocationProtect == td[i].prot, "%d: %#x != %#x\n", i, info.AllocationProtect, td[i].prot);
1603             ok(info.State == MEM_COMMIT, "%d: %#x != MEM_COMMIT\n", i, info.State);
1604             ok(info.Type == MEM_PRIVATE, "%d: %#x != MEM_PRIVATE\n", i, info.Type);
1605
1606             if (is_mem_writable(info.Protect))
1607             {
1608                 base[0] = 0xfe;
1609
1610                 SetLastError(0xdeadbeef);
1611                 ret = VirtualQuery(base, &info, sizeof(info));
1612                 ok(ret, "VirtualQuery failed %d\n", GetLastError());
1613                 ok(info.Protect == td[i].prot, "%d: got %#x != expected %#x\n", i, info.Protect, td[i].prot);
1614             }
1615
1616             VirtualFree(base, 0, MEM_FREE);
1617         }
1618         else
1619         {
1620             ok(!base, "%d: VirtualAlloc should fail\n", i);
1621             ok(GetLastError() == ERROR_INVALID_PARAMETER, "%d: expected ERROR_INVALID_PARAMETER, got %d\n", i, GetLastError());
1622         }
1623     }
1624 }
1625
1626 static void test_CreateFileMapping_protection(void)
1627 {
1628     static const struct test_data
1629     {
1630         DWORD prot;
1631         BOOL success;
1632         DWORD prot_after_write;
1633     } td[] =
1634     {
1635         { 0, FALSE, 0 }, /* 0x00 */
1636         { PAGE_NOACCESS, FALSE, PAGE_NOACCESS }, /* 0x01 */
1637         { PAGE_READONLY, TRUE, PAGE_READONLY }, /* 0x02 */
1638         { PAGE_READONLY | PAGE_NOACCESS, FALSE, PAGE_NOACCESS }, /* 0x03 */
1639         { PAGE_READWRITE, TRUE, PAGE_READWRITE }, /* 0x04 */
1640         { PAGE_READWRITE | PAGE_NOACCESS, FALSE, PAGE_NOACCESS }, /* 0x05 */
1641         { PAGE_READWRITE | PAGE_READONLY, FALSE, PAGE_NOACCESS }, /* 0x06 */
1642         { PAGE_READWRITE | PAGE_READONLY | PAGE_NOACCESS, FALSE, PAGE_NOACCESS }, /* 0x07 */
1643         { PAGE_WRITECOPY, TRUE, PAGE_READWRITE }, /* 0x08 */
1644         { PAGE_WRITECOPY | PAGE_NOACCESS, FALSE, PAGE_NOACCESS }, /* 0x09 */
1645         { PAGE_WRITECOPY | PAGE_READONLY, FALSE, PAGE_NOACCESS }, /* 0x0a */
1646         { PAGE_WRITECOPY | PAGE_NOACCESS | PAGE_READONLY, FALSE, PAGE_NOACCESS }, /* 0x0b */
1647         { PAGE_WRITECOPY | PAGE_READWRITE, FALSE, PAGE_NOACCESS }, /* 0x0c */
1648         { PAGE_WRITECOPY | PAGE_READWRITE | PAGE_NOACCESS, FALSE, PAGE_NOACCESS }, /* 0x0d */
1649         { PAGE_WRITECOPY | PAGE_READWRITE | PAGE_READONLY, FALSE, PAGE_NOACCESS }, /* 0x0e */
1650         { PAGE_WRITECOPY | PAGE_READWRITE | PAGE_READONLY | PAGE_NOACCESS, FALSE, PAGE_NOACCESS }, /* 0x0f */
1651
1652         { PAGE_EXECUTE, FALSE, PAGE_EXECUTE }, /* 0x10 */
1653         { PAGE_EXECUTE_READ, TRUE, PAGE_EXECUTE_READ }, /* 0x20 */
1654         { PAGE_EXECUTE_READ | PAGE_EXECUTE, FALSE, PAGE_EXECUTE_READ }, /* 0x30 */
1655         { PAGE_EXECUTE_READWRITE, TRUE, PAGE_EXECUTE_READWRITE }, /* 0x40 */
1656         { PAGE_EXECUTE_READWRITE | PAGE_EXECUTE, FALSE, PAGE_NOACCESS }, /* 0x50 */
1657         { PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ, FALSE, PAGE_NOACCESS }, /* 0x60 */
1658         { PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ | PAGE_EXECUTE, FALSE, PAGE_NOACCESS }, /* 0x70 */
1659         { PAGE_EXECUTE_WRITECOPY, TRUE, PAGE_EXECUTE_READWRITE }, /* 0x80 */
1660         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE, FALSE, PAGE_NOACCESS }, /* 0x90 */
1661         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READ, FALSE, PAGE_NOACCESS }, /* 0xa0 */
1662         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READ | PAGE_EXECUTE, FALSE, PAGE_NOACCESS }, /* 0xb0 */
1663         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE, FALSE, PAGE_NOACCESS }, /* 0xc0 */
1664         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE | PAGE_EXECUTE, FALSE, PAGE_NOACCESS }, /* 0xd0 */
1665         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ, FALSE, PAGE_NOACCESS }, /* 0xe0 */
1666         { PAGE_EXECUTE_WRITECOPY | PAGE_EXECUTE_READWRITE | PAGE_EXECUTE_READ | PAGE_EXECUTE, FALSE, PAGE_NOACCESS } /* 0xf0 */
1667     };
1668     char *base;
1669     DWORD ret, i, alloc_prot, prot, old_prot;
1670     MEMORY_BASIC_INFORMATION info;
1671     SYSTEM_INFO si;
1672     char temp_path[MAX_PATH];
1673     char file_name[MAX_PATH];
1674     HANDLE hfile, hmap;
1675     BOOL page_exec_supported = TRUE;
1676
1677     GetSystemInfo(&si);
1678     trace("system page size %#x\n", si.dwPageSize);
1679
1680     GetTempPath(MAX_PATH, temp_path);
1681     GetTempFileName(temp_path, "map", 0, file_name);
1682
1683     SetLastError(0xdeadbeef);
1684     hfile = CreateFile(file_name, GENERIC_READ|GENERIC_WRITE|GENERIC_EXECUTE, 0, NULL, CREATE_ALWAYS, 0, 0);
1685     ok(hfile != INVALID_HANDLE_VALUE, "CreateFile(%s) error %d\n", file_name, GetLastError());
1686     SetFilePointer(hfile, si.dwPageSize, NULL, FILE_BEGIN);
1687     SetEndOfFile(hfile);
1688
1689     for (i = 0; i < sizeof(td)/sizeof(td[0]); i++)
1690     {
1691         SetLastError(0xdeadbeef);
1692         hmap = CreateFileMapping(hfile, NULL, td[i].prot | SEC_COMMIT, 0, si.dwPageSize, NULL);
1693
1694         if (td[i].success)
1695         {
1696             if (!hmap)
1697             {
1698                 trace("%d: CreateFileMapping(%04x) failed: %d\n", i, td[i].prot, GetLastError());
1699                 /* NT4 and win2k don't support EXEC on file mappings */
1700                 if (td[i].prot == PAGE_EXECUTE_READ || td[i].prot == PAGE_EXECUTE_READWRITE)
1701                 {
1702                     page_exec_supported = FALSE;
1703                     ok(broken(!hmap), "%d: CreateFileMapping doesn't support PAGE_EXECUTE\n", i);
1704                     continue;
1705                 }
1706                 /* Vista+ supports PAGE_EXECUTE_WRITECOPY, earlier versions don't */
1707                 if (td[i].prot == PAGE_EXECUTE_WRITECOPY)
1708                 {
1709                     page_exec_supported = FALSE;
1710                     ok(broken(!hmap), "%d: CreateFileMapping doesn't support PAGE_EXECUTE_WRITECOPY\n", i);
1711                     continue;
1712                 }
1713             }
1714             ok(hmap != 0, "%d: CreateFileMapping(%04x) error %d\n", i, td[i].prot, GetLastError());
1715
1716             base = MapViewOfFile(hmap, FILE_MAP_READ, 0, 0, 0);
1717             ok(base != NULL, "%d: MapViewOfFile failed %d\n", i, GetLastError());
1718
1719             SetLastError(0xdeadbeef);
1720             ret = VirtualQuery(base, &info, sizeof(info));
1721             ok(ret, "VirtualQuery failed %d\n", GetLastError());
1722             ok(info.BaseAddress == base, "%d: got %p != expected %p\n", i, info.BaseAddress, base);
1723             ok(info.RegionSize == si.dwPageSize, "%d: got %#lx != expected %#x\n", i, info.RegionSize, si.dwPageSize);
1724             ok(info.Protect == PAGE_READONLY, "%d: got %#x != expected PAGE_READONLY\n", i, info.Protect);
1725             ok(info.AllocationBase == base, "%d: %p != %p\n", i, info.AllocationBase, base);
1726             ok(info.AllocationProtect == PAGE_READONLY, "%d: %#x != PAGE_READONLY\n", i, info.AllocationProtect);
1727             ok(info.State == MEM_COMMIT, "%d: %#x != MEM_COMMIT\n", i, info.State);
1728             ok(info.Type == MEM_MAPPED, "%d: %#x != MEM_MAPPED\n", i, info.Type);
1729
1730             if (is_mem_writable(info.Protect))
1731             {
1732                 base[0] = 0xfe;
1733
1734                 SetLastError(0xdeadbeef);
1735                 ret = VirtualQuery(base, &info, sizeof(info));
1736                 ok(ret, "VirtualQuery failed %d\n", GetLastError());
1737                 ok(info.Protect == td[i].prot, "%d: got %#x != expected %#x\n", i, info.Protect, td[i].prot);
1738             }
1739
1740             UnmapViewOfFile(base);
1741             CloseHandle(hmap);
1742         }
1743         else
1744         {
1745             ok(!hmap, "%d: CreateFileMapping should fail\n", i);
1746             ok(GetLastError() == ERROR_INVALID_PARAMETER, "%d: expected ERROR_INVALID_PARAMETER, got %d\n", i, GetLastError());
1747         }
1748     }
1749
1750     if (page_exec_supported) alloc_prot = PAGE_EXECUTE_READWRITE;
1751     else alloc_prot = PAGE_READWRITE;
1752     SetLastError(0xdeadbeef);
1753     hmap = CreateFileMapping(hfile, NULL, alloc_prot, 0, si.dwPageSize, NULL);
1754     ok(hmap != 0, "%d: CreateFileMapping error %d\n", i, GetLastError());
1755
1756     SetLastError(0xdeadbeef);
1757     base = MapViewOfFile(hmap, FILE_MAP_READ | FILE_MAP_WRITE | (page_exec_supported ? FILE_MAP_EXECUTE : 0), 0, 0, 0);
1758     ok(base != NULL, "MapViewOfFile failed %d\n", GetLastError());
1759
1760     old_prot = 0xdeadbeef;
1761     SetLastError(0xdeadbeef);
1762     ret = VirtualProtect(base, si.dwPageSize, PAGE_NOACCESS, &old_prot);
1763     ok(ret, "VirtualProtect error %d\n", GetLastError());
1764     ok(old_prot == alloc_prot, "got %#x != expected %#x\n", old_prot, alloc_prot);
1765
1766     for (i = 0; i < sizeof(td)/sizeof(td[0]); i++)
1767     {
1768         SetLastError(0xdeadbeef);
1769         ret = VirtualQuery(base, &info, sizeof(info));
1770         ok(ret, "VirtualQuery failed %d\n", GetLastError());
1771         ok(info.BaseAddress == base, "%d: got %p != expected %p\n", i, info.BaseAddress, base);
1772         ok(info.RegionSize == si.dwPageSize, "%d: got %#lx != expected %#x\n", i, info.RegionSize, si.dwPageSize);
1773         ok(info.Protect == PAGE_NOACCESS, "%d: got %#x != expected PAGE_NOACCESS\n", i, info.Protect);
1774         ok(info.AllocationBase == base, "%d: %p != %p\n", i, info.AllocationBase, base);
1775         ok(info.AllocationProtect == alloc_prot, "%d: %#x != %#x\n", i, info.AllocationProtect, alloc_prot);
1776         ok(info.State == MEM_COMMIT, "%d: %#x != MEM_COMMIT\n", i, info.State);
1777         ok(info.Type == MEM_MAPPED, "%d: %#x != MEM_MAPPED\n", i, info.Type);
1778
1779         old_prot = 0xdeadbeef;
1780         SetLastError(0xdeadbeef);
1781         ret = VirtualProtect(base, si.dwPageSize, td[i].prot, &old_prot);
1782         if (td[i].success || td[i].prot == PAGE_NOACCESS || td[i].prot == PAGE_EXECUTE)
1783         {
1784             if (!ret)
1785             {
1786                 /* win2k and XP don't support EXEC on file mappings */
1787                 if (td[i].prot == PAGE_EXECUTE)
1788                 {
1789                     ok(broken(!ret), "%d: VirtualProtect doesn't support PAGE_EXECUTE\n", i);
1790                     continue;
1791                 }
1792                 /* NT4 and win2k don't support EXEC on file mappings */
1793                 if (td[i].prot == PAGE_EXECUTE_READ || td[i].prot == PAGE_EXECUTE_READWRITE)
1794                 {
1795                     ok(broken(!ret), "%d: VirtualProtect doesn't support PAGE_EXECUTE\n", i);
1796                     continue;
1797                 }
1798                 /* Vista+ supports PAGE_EXECUTE_WRITECOPY, earlier versions don't */
1799                 if (td[i].prot == PAGE_EXECUTE_WRITECOPY)
1800                 {
1801                     ok(broken(!ret), "%d: VirtualProtect doesn't support PAGE_EXECUTE_WRITECOPY\n", i);
1802                     continue;
1803                 }
1804             }
1805
1806             ok(ret, "%d: VirtualProtect error %d\n", i, GetLastError());
1807             ok(old_prot == PAGE_NOACCESS, "%d: got %#x != expected PAGE_NOACCESS\n", i, old_prot);
1808
1809             prot = td[i].prot;
1810             /* looks strange but Windows doesn't do this for PAGE_WRITECOPY */
1811             if (prot == PAGE_EXECUTE_WRITECOPY) prot = PAGE_EXECUTE_READWRITE;
1812
1813             SetLastError(0xdeadbeef);
1814             ret = VirtualQuery(base, &info, sizeof(info));
1815             ok(ret, "VirtualQuery failed %d\n", GetLastError());
1816             ok(info.BaseAddress == base, "%d: got %p != expected %p\n", i, info.BaseAddress, base);
1817             ok(info.RegionSize == si.dwPageSize, "%d: got %#lx != expected %#x\n", i, info.RegionSize, si.dwPageSize);
1818             /* FIXME: remove the condition below once Wine is fixed */
1819             if (td[i].prot == PAGE_EXECUTE_WRITECOPY)
1820                 todo_wine ok(info.Protect == prot, "%d: got %#x != expected %#x\n", i, info.Protect, prot);
1821             else
1822                 ok(info.Protect == prot, "%d: got %#x != expected %#x\n", i, info.Protect, prot);
1823             ok(info.AllocationBase == base, "%d: %p != %p\n", i, info.AllocationBase, base);
1824             ok(info.AllocationProtect == alloc_prot, "%d: %#x != %#x\n", i, info.AllocationProtect, alloc_prot);
1825             ok(info.State == MEM_COMMIT, "%d: %#x != MEM_COMMIT\n", i, info.State);
1826             ok(info.Type == MEM_MAPPED, "%d: %#x != MEM_MAPPED\n", i, info.Type);
1827
1828             if (is_mem_writable(info.Protect))
1829             {
1830                 base[0] = 0xfe;
1831
1832                 SetLastError(0xdeadbeef);
1833                 ret = VirtualQuery(base, &info, sizeof(info));
1834                 ok(ret, "VirtualQuery failed %d\n", GetLastError());
1835                 /* FIXME: remove the condition below once Wine is fixed */
1836                 if (td[i].prot == PAGE_WRITECOPY || td[i].prot == PAGE_EXECUTE_WRITECOPY)
1837                     todo_wine ok(info.Protect == td[i].prot_after_write, "%d: got %#x != expected %#x\n", i, info.Protect, td[i].prot_after_write);
1838                 else
1839                     ok(info.Protect == td[i].prot_after_write, "%d: got %#x != expected %#x\n", i, info.Protect, td[i].prot_after_write);
1840             }
1841         }
1842         else
1843         {
1844             ok(!ret, "%d: VirtualProtect should fail\n", i);
1845             ok(GetLastError() == ERROR_INVALID_PARAMETER, "%d: expected ERROR_INVALID_PARAMETER, got %d\n", i, GetLastError());
1846             continue;
1847         }
1848
1849         old_prot = 0xdeadbeef;
1850         SetLastError(0xdeadbeef);
1851         ret = VirtualProtect(base, si.dwPageSize, PAGE_NOACCESS, &old_prot);
1852         ok(ret, "%d: VirtualProtect error %d\n", i, GetLastError());
1853         /* FIXME: remove the condition below once Wine is fixed */
1854         if (td[i].prot == PAGE_WRITECOPY || td[i].prot == PAGE_EXECUTE_WRITECOPY)
1855             todo_wine ok(old_prot == td[i].prot_after_write, "%d: got %#x != expected %#x\n", i, old_prot, td[i].prot_after_write);
1856         else
1857             ok(old_prot == td[i].prot_after_write, "%d: got %#x != expected %#x\n", i, old_prot, td[i].prot_after_write);
1858     }
1859
1860     UnmapViewOfFile(base);
1861     CloseHandle(hmap);
1862
1863     CloseHandle(hfile);
1864     DeleteFile(file_name);
1865 }
1866
1867 #define ACCESS_READ      0x01
1868 #define ACCESS_WRITE     0x02
1869 #define ACCESS_EXECUTE   0x04
1870 #define ACCESS_WRITECOPY 0x08
1871
1872 static DWORD page_prot_to_access(DWORD prot)
1873 {
1874     switch (prot)
1875     {
1876     case PAGE_READWRITE:
1877         return ACCESS_READ | ACCESS_WRITE;
1878
1879     case PAGE_EXECUTE:
1880     case PAGE_EXECUTE_READ:
1881         return ACCESS_READ | ACCESS_EXECUTE;
1882
1883     case PAGE_EXECUTE_READWRITE:
1884         return ACCESS_READ | ACCESS_WRITE | ACCESS_WRITECOPY | ACCESS_EXECUTE;
1885
1886     case PAGE_EXECUTE_WRITECOPY:
1887         return ACCESS_READ | ACCESS_WRITECOPY | ACCESS_EXECUTE;
1888
1889     case PAGE_READONLY:
1890         return ACCESS_READ;
1891
1892     case PAGE_WRITECOPY:
1893         return ACCESS_READ;
1894
1895     default:
1896         return 0;
1897     }
1898 }
1899
1900 static BOOL is_compatible_protection(DWORD map_prot, DWORD view_prot, DWORD prot)
1901 {
1902     DWORD map_access, view_access, prot_access;
1903
1904     map_access = page_prot_to_access(map_prot);
1905     view_access = page_prot_to_access(view_prot);
1906     prot_access = page_prot_to_access(prot);
1907
1908     if (view_access == prot_access) return TRUE;
1909     if (!view_access) return FALSE;
1910
1911     if ((view_access & prot_access) != prot_access) return FALSE;
1912     if ((map_access & prot_access) == prot_access) return TRUE;
1913
1914     return FALSE;
1915 }
1916
1917 static DWORD map_prot_to_access(DWORD prot)
1918 {
1919     switch (prot)
1920     {
1921     case PAGE_READWRITE:
1922     case PAGE_EXECUTE_READWRITE:
1923         return SECTION_MAP_READ | SECTION_MAP_WRITE | SECTION_MAP_EXECUTE | SECTION_MAP_EXECUTE_EXPLICIT | SECTION_QUERY;
1924     case PAGE_READONLY:
1925     case PAGE_WRITECOPY:
1926     case PAGE_EXECUTE:
1927     case PAGE_EXECUTE_READ:
1928     case PAGE_EXECUTE_WRITECOPY:
1929         return SECTION_MAP_READ | SECTION_MAP_EXECUTE | SECTION_MAP_EXECUTE_EXPLICIT | SECTION_QUERY;
1930     default:
1931         return 0;
1932     }
1933 }
1934
1935 static BOOL is_compatible_access(DWORD map_prot, DWORD view_prot)
1936 {
1937     DWORD access = map_prot_to_access(map_prot);
1938     if (!view_prot) view_prot = SECTION_MAP_READ;
1939     return (view_prot & access) == view_prot;
1940 }
1941
1942 static void *map_view_of_file(HANDLE handle, DWORD access)
1943 {
1944     NTSTATUS status;
1945     LARGE_INTEGER offset;
1946     SIZE_T count;
1947     ULONG protect;
1948     BOOL exec;
1949     void *addr;
1950
1951     if (!pNtMapViewOfSection) return NULL;
1952
1953     count = 0;
1954     offset.u.LowPart  = 0;
1955     offset.u.HighPart = 0;
1956
1957     exec = access & FILE_MAP_EXECUTE;
1958     access &= ~FILE_MAP_EXECUTE;
1959
1960     if (access == FILE_MAP_COPY)
1961     {
1962         if (exec)
1963             protect = PAGE_EXECUTE_WRITECOPY;
1964         else
1965             protect = PAGE_WRITECOPY;
1966     }
1967     else if (access & FILE_MAP_WRITE)
1968     {
1969         if (exec)
1970             protect = PAGE_EXECUTE_READWRITE;
1971         else
1972             protect = PAGE_READWRITE;
1973     }
1974     else if (access & FILE_MAP_READ)
1975     {
1976         if (exec)
1977             protect = PAGE_EXECUTE_READ;
1978         else
1979             protect = PAGE_READONLY;
1980     }
1981     else protect = PAGE_NOACCESS;
1982
1983     addr = NULL;
1984     status = pNtMapViewOfSection(handle, GetCurrentProcess(), &addr, 0, 0, &offset,
1985                                  &count, 1 /* ViewShare */, 0, protect);
1986     if (status)
1987     {
1988         /* for simplicity */
1989         SetLastError(ERROR_ACCESS_DENIED);
1990         addr = NULL;
1991     }
1992     return addr;
1993 }
1994
1995 static void test_mapping(void)
1996 {
1997     static const DWORD page_prot[] =
1998     {
1999         PAGE_NOACCESS, PAGE_READONLY, PAGE_READWRITE, PAGE_WRITECOPY,
2000         PAGE_EXECUTE_READ, PAGE_EXECUTE_READWRITE, PAGE_EXECUTE_WRITECOPY
2001     };
2002     static const struct
2003     {
2004         DWORD access, prot;
2005     } view[] =
2006     {
2007         { 0, PAGE_NOACCESS }, /* 0x00 */
2008         { FILE_MAP_COPY, PAGE_WRITECOPY }, /* 0x01 */
2009         { FILE_MAP_WRITE, PAGE_READWRITE }, /* 0x02 */
2010         { FILE_MAP_WRITE | FILE_MAP_COPY, PAGE_READWRITE }, /* 0x03 */
2011         { FILE_MAP_READ, PAGE_READONLY }, /* 0x04 */
2012         { FILE_MAP_READ | FILE_MAP_COPY, PAGE_READONLY }, /* 0x05 */
2013         { FILE_MAP_READ | FILE_MAP_WRITE, PAGE_READWRITE }, /* 0x06 */
2014         { FILE_MAP_READ | FILE_MAP_WRITE | FILE_MAP_COPY, PAGE_READWRITE }, /* 0x07 */
2015         { SECTION_MAP_EXECUTE, PAGE_NOACCESS }, /* 0x08 */
2016         { SECTION_MAP_EXECUTE | FILE_MAP_COPY, PAGE_NOACCESS }, /* 0x09 */
2017         { SECTION_MAP_EXECUTE | FILE_MAP_WRITE, PAGE_READWRITE }, /* 0x0a */
2018         { SECTION_MAP_EXECUTE | FILE_MAP_WRITE | FILE_MAP_COPY, PAGE_READWRITE }, /* 0x0b */
2019         { SECTION_MAP_EXECUTE | FILE_MAP_READ, PAGE_READONLY }, /* 0x0c */
2020         { SECTION_MAP_EXECUTE | FILE_MAP_READ | FILE_MAP_COPY, PAGE_READONLY }, /* 0x0d */
2021         { SECTION_MAP_EXECUTE | FILE_MAP_READ | FILE_MAP_WRITE, PAGE_READWRITE }, /* 0x0e */
2022         { SECTION_MAP_EXECUTE | FILE_MAP_READ | FILE_MAP_WRITE | FILE_MAP_COPY, PAGE_READWRITE }, /* 0x0f */
2023         { FILE_MAP_EXECUTE, PAGE_NOACCESS }, /* 0x20 */
2024         { FILE_MAP_EXECUTE | FILE_MAP_COPY, PAGE_EXECUTE_WRITECOPY }, /* 0x21 */
2025         { FILE_MAP_EXECUTE | FILE_MAP_WRITE, PAGE_EXECUTE_READWRITE }, /* 0x22 */
2026         { FILE_MAP_EXECUTE | FILE_MAP_WRITE | FILE_MAP_COPY, PAGE_EXECUTE_READWRITE }, /* 0x23 */
2027         { FILE_MAP_EXECUTE | FILE_MAP_READ, PAGE_EXECUTE_READ }, /* 0x24 */
2028         { FILE_MAP_EXECUTE | FILE_MAP_READ | FILE_MAP_COPY, PAGE_EXECUTE_READ }, /* 0x25 */
2029         { FILE_MAP_EXECUTE | FILE_MAP_READ | FILE_MAP_WRITE, PAGE_EXECUTE_READWRITE }, /* 0x26 */
2030         { FILE_MAP_EXECUTE | FILE_MAP_READ | FILE_MAP_WRITE | FILE_MAP_COPY, PAGE_EXECUTE_READWRITE }, /* 0x27 */
2031         { FILE_MAP_EXECUTE | SECTION_MAP_EXECUTE, PAGE_NOACCESS }, /* 0x28 */
2032         { FILE_MAP_EXECUTE | SECTION_MAP_EXECUTE | FILE_MAP_COPY, PAGE_NOACCESS }, /* 0x29 */
2033         { FILE_MAP_EXECUTE | SECTION_MAP_EXECUTE | FILE_MAP_WRITE, PAGE_EXECUTE_READWRITE }, /* 0x2a */
2034         { FILE_MAP_EXECUTE | SECTION_MAP_EXECUTE | FILE_MAP_WRITE | FILE_MAP_COPY, PAGE_EXECUTE_READWRITE }, /* 0x2b */
2035         { FILE_MAP_EXECUTE | SECTION_MAP_EXECUTE | FILE_MAP_READ, PAGE_EXECUTE_READ }, /* 0x2c */
2036         { FILE_MAP_EXECUTE | SECTION_MAP_EXECUTE | FILE_MAP_READ | FILE_MAP_COPY, PAGE_EXECUTE_READ }, /* 0x2d */
2037         { FILE_MAP_EXECUTE | SECTION_MAP_EXECUTE | FILE_MAP_READ | FILE_MAP_WRITE, PAGE_EXECUTE_READWRITE }, /* 0x2e */
2038         { FILE_MAP_EXECUTE | SECTION_MAP_EXECUTE | FILE_MAP_READ | FILE_MAP_WRITE | FILE_MAP_COPY, PAGE_EXECUTE_READWRITE } /* 0x2f */
2039     };
2040     void *base, *nt_base;
2041     DWORD i, j, k, ret, old_prot, prev_prot;
2042     SYSTEM_INFO si;
2043     char temp_path[MAX_PATH];
2044     char file_name[MAX_PATH];
2045     HANDLE hfile, hmap;
2046     MEMORY_BASIC_INFORMATION info, nt_info;
2047
2048     GetSystemInfo(&si);
2049     trace("system page size %#x\n", si.dwPageSize);
2050
2051     GetTempPath(MAX_PATH, temp_path);
2052     GetTempFileName(temp_path, "map", 0, file_name);
2053
2054     SetLastError(0xdeadbeef);
2055     hfile = CreateFile(file_name, GENERIC_READ|GENERIC_WRITE|GENERIC_EXECUTE, 0, NULL, CREATE_ALWAYS, 0, 0);
2056     ok(hfile != INVALID_HANDLE_VALUE, "CreateFile(%s) error %d\n", file_name, GetLastError());
2057     SetFilePointer(hfile, si.dwPageSize, NULL, FILE_BEGIN);
2058     SetEndOfFile(hfile);
2059
2060     for (i = 0; i < sizeof(page_prot)/sizeof(page_prot[0]); i++)
2061     {
2062         SetLastError(0xdeadbeef);
2063         hmap = CreateFileMapping(hfile, NULL, page_prot[i] | SEC_COMMIT, 0, si.dwPageSize, NULL);
2064
2065         if (page_prot[i] == PAGE_NOACCESS)
2066         {
2067             HANDLE hmap2;
2068
2069             ok(!hmap, "CreateFileMapping(PAGE_NOACCESS) should fail\n");
2070             ok(GetLastError() == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
2071
2072             /* A trick to create a not accessible mapping */
2073             SetLastError(0xdeadbeef);
2074             hmap = CreateFileMapping(hfile, NULL, PAGE_READWRITE | SEC_COMMIT, 0, si.dwPageSize, NULL);
2075             ok(hmap != 0, "CreateFileMapping(PAGE_READWRITE) error %d\n", GetLastError());
2076             SetLastError(0xdeadbeef);
2077             ret = DuplicateHandle(GetCurrentProcess(), hmap, GetCurrentProcess(), &hmap2, 0, FALSE, 0);
2078             ok(ret, "DuplicateHandle error %d\n", GetLastError());
2079             CloseHandle(hmap);
2080             hmap = hmap2;
2081         }
2082
2083         if (!hmap)
2084         {
2085             trace("%d: CreateFileMapping(%04x) failed: %d\n", i, page_prot[i], GetLastError());
2086
2087             /* NT4 and win2k don't support EXEC on file mappings */
2088             if (page_prot[i] == PAGE_EXECUTE_READ || page_prot[i] == PAGE_EXECUTE_READWRITE)
2089             {
2090                 ok(broken(!hmap), "%d: CreateFileMapping doesn't support PAGE_EXECUTE\n", i);
2091                 continue;
2092             }
2093             /* Vista+ supports PAGE_EXECUTE_WRITECOPY, earlier versions don't */
2094             if (page_prot[i] == PAGE_EXECUTE_WRITECOPY)
2095             {
2096                 ok(broken(!hmap), "%d: CreateFileMapping doesn't support PAGE_EXECUTE_WRITECOPY\n", i);
2097                 continue;
2098             }
2099         }
2100
2101         ok(hmap != 0, "%d: CreateFileMapping(%04x) error %d\n", i, page_prot[i], GetLastError());
2102
2103         for (j = 0; j < sizeof(view)/sizeof(view[0]); j++)
2104         {
2105             nt_base = map_view_of_file(hmap, view[j].access);
2106             if (nt_base)
2107             {
2108                 SetLastError(0xdeadbeef);
2109                 ret = VirtualQuery(nt_base, &nt_info, sizeof(nt_info));
2110                 ok(ret, "%d: VirtualQuery failed %d\n", j, GetLastError());
2111                 UnmapViewOfFile(nt_base);
2112             }
2113
2114             SetLastError(0xdeadbeef);
2115             base = MapViewOfFile(hmap, view[j].access, 0, 0, 0);
2116
2117             /* Vista+ supports FILE_MAP_EXECUTE properly, earlier versions don't */
2118             ok(!nt_base == !base ||
2119                broken((view[j].access & FILE_MAP_EXECUTE) && !nt_base != !base),
2120                "%d: (%04x/%04x) NT %p kernel %p\n", j, page_prot[i], view[j].access, nt_base, base);
2121
2122             if (!is_compatible_access(page_prot[i], view[j].access))
2123             {
2124                 ok(!base, "%d: MapViewOfFile(%04x/%04x) should fail\n", j, page_prot[i], view[j].access);
2125                 ok(GetLastError() == ERROR_ACCESS_DENIED, "wrong error %d\n", GetLastError());
2126                 continue;
2127             }
2128
2129             /* Vista+ properly supports FILE_MAP_EXECUTE, earlier versions don't */
2130             if (!base && (view[j].access & FILE_MAP_EXECUTE))
2131             {
2132                 ok(broken(!base), "%d: MapViewOfFile(%04x/%04x) failed %d\n", j, page_prot[i], view[j].access, GetLastError());
2133                 continue;
2134             }
2135
2136             ok(base != NULL, "%d: MapViewOfFile(%04x/%04x) failed %d\n", j, page_prot[i], view[j].access, GetLastError());
2137
2138             SetLastError(0xdeadbeef);
2139             ret = VirtualQuery(base, &info, sizeof(info));
2140             ok(ret, "%d: VirtualQuery failed %d\n", j, GetLastError());
2141             ok(info.BaseAddress == base, "%d: (%04x) got %p, expected %p\n", j, view[j].access, info.BaseAddress, base);
2142             ok(info.RegionSize == si.dwPageSize, "%d: (%04x) got %#lx != expected %#x\n", j, view[j].access, info.RegionSize, si.dwPageSize);
2143             ok(info.Protect == view[j].prot ||
2144                broken(view[j].prot == PAGE_EXECUTE_READ && info.Protect == PAGE_READONLY) || /* win2k */
2145                broken(view[j].prot == PAGE_EXECUTE_READWRITE && info.Protect == PAGE_READWRITE) || /* win2k */
2146                broken(view[j].prot == PAGE_EXECUTE_WRITECOPY && info.Protect == PAGE_NOACCESS), /* XP */
2147                "%d: (%04x) got %#x, expected %#x\n", j, view[j].access, info.Protect, view[j].prot);
2148             ok(info.AllocationBase == base, "%d: (%04x) got %p, expected %p\n", j, view[j].access, info.AllocationBase, base);
2149             ok(info.AllocationProtect == info.Protect, "%d: (%04x) got %#x, expected %#x\n", j, view[j].access, info.AllocationProtect, info.Protect);
2150             ok(info.State == MEM_COMMIT, "%d: (%04x) got %#x, expected MEM_COMMIT\n", j, view[j].access, info.State);
2151             ok(info.Type == MEM_MAPPED, "%d: (%04x) got %#x, expected MEM_MAPPED\n", j, view[j].access, info.Type);
2152
2153             if (nt_base && base)
2154             {
2155                 ok(nt_info.RegionSize == info.RegionSize, "%d: (%04x) got %#lx != expected %#lx\n", j, view[j].access, nt_info.RegionSize, info.RegionSize);
2156                 ok(nt_info.Protect == info.Protect /* Vista+ */ ||
2157                    broken(nt_info.AllocationProtect == PAGE_EXECUTE_WRITECOPY && info.Protect == PAGE_NOACCESS), /* XP */
2158                    "%d: (%04x) got %#x, expected %#x\n", j, view[j].access, nt_info.Protect, info.Protect);
2159                 ok(nt_info.AllocationProtect == info.AllocationProtect /* Vista+ */ ||
2160                    broken(nt_info.AllocationProtect == PAGE_EXECUTE_WRITECOPY && info.Protect == PAGE_NOACCESS), /* XP */
2161                    "%d: (%04x) got %#x, expected %#x\n", j, view[j].access, nt_info.AllocationProtect, info.AllocationProtect);
2162                 ok(nt_info.State == info.State, "%d: (%04x) got %#x, expected %#x\n", j, view[j].access, nt_info.State, info.State);
2163                 ok(nt_info.Type == info.Type, "%d: (%04x) got %#x, expected %#x\n", j, view[j].access, nt_info.Type, info.Type);
2164             }
2165
2166             prev_prot = info.Protect;
2167
2168             for (k = 0; k < sizeof(page_prot)/sizeof(page_prot[0]); k++)
2169             {
2170                 /*trace("map %#x, view %#x, requested prot %#x\n", page_prot[i], view[j].prot, page_prot[k]);*/
2171                 SetLastError(0xdeadbeef);
2172                 old_prot = 0xdeadbeef;
2173                 ret = VirtualProtect(base, si.dwPageSize, page_prot[k], &old_prot);
2174                 if (is_compatible_protection(page_prot[i], view[j].prot, page_prot[k]))
2175                 {
2176                     /* win2k and XP don't support EXEC on file mappings */
2177                     if (!ret && page_prot[k] == PAGE_EXECUTE)
2178                     {
2179                         ok(broken(!ret), "VirtualProtect doesn't support PAGE_EXECUTE\n");
2180                         continue;
2181                     }
2182                     /* NT4 and win2k don't support EXEC on file mappings */
2183                     if (!ret && (page_prot[k] == PAGE_EXECUTE_READ || page_prot[k] == PAGE_EXECUTE_READWRITE))
2184                     {
2185                         ok(broken(!ret), "VirtualProtect doesn't support PAGE_EXECUTE\n");
2186                         continue;
2187                     }
2188                     /* Vista+ supports PAGE_EXECUTE_WRITECOPY, earlier versions don't */
2189                     if (!ret && page_prot[k] == PAGE_EXECUTE_WRITECOPY)
2190                     {
2191                         ok(broken(!ret), "VirtualProtect doesn't support PAGE_EXECUTE_WRITECOPY\n");
2192                         continue;
2193                     }
2194                     /* win2k and XP don't support PAGE_EXECUTE_WRITECOPY views properly  */
2195                     if (!ret && view[j].prot == PAGE_EXECUTE_WRITECOPY)
2196                     {
2197                         ok(broken(!ret), "VirtualProtect doesn't support PAGE_EXECUTE_WRITECOPY view properly\n");
2198                         continue;
2199                     }
2200
2201                     ok(ret, "VirtualProtect error %d, map %#x, view %#x, requested prot %#x\n", GetLastError(), page_prot[i], view[j].prot, page_prot[k]);
2202                     ok(old_prot == prev_prot, "got %#x, expected %#x\n", old_prot, prev_prot);
2203                     prev_prot = page_prot[k];
2204                 }
2205                 else
2206                 {
2207                     /* NT4 doesn't fail on incompatible map and view */
2208                     if (ret)
2209                     {
2210                         ok(broken(ret), "VirtualProtect should fail, map %#x, view %#x, requested prot %#x\n", page_prot[i], view[j].prot, page_prot[k]);
2211                         skip("Incompatible map and view are not properly handled on this platform\n");
2212                         break; /* NT4 won't pass remaining tests */
2213                     }
2214
2215                     ok(!ret, "VirtualProtect should fail, map %#x, view %#x, requested prot %#x\n", page_prot[i], view[j].prot, page_prot[k]);
2216                     ok(GetLastError() == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
2217                 }
2218             }
2219
2220             UnmapViewOfFile(base);
2221         }
2222
2223         CloseHandle(hmap);
2224     }
2225
2226     CloseHandle(hfile);
2227     DeleteFile(file_name);
2228 }
2229
2230 START_TEST(virtual)
2231 {
2232     int argc;
2233     char **argv;
2234     argc = winetest_get_mainargs( &argv );
2235
2236     if (argc >= 3)
2237     {
2238         if (!strcmp(argv[2], "sleep"))
2239         {
2240             Sleep(5000); /* spawned process runs for at most 5 seconds */
2241             return;
2242         }
2243         while (1)
2244         {
2245             void *mem;
2246             BOOL ret;
2247             mem = VirtualAlloc(NULL, 1<<20, MEM_COMMIT|MEM_RESERVE,
2248                                PAGE_EXECUTE_READWRITE);
2249             ok(mem != NULL, "VirtualAlloc failed %u\n", GetLastError());
2250             if (mem == NULL) break;
2251             ret = VirtualFree(mem, 0, MEM_RELEASE);
2252             ok(ret, "VirtualFree failed %u\n", GetLastError());
2253             if (!ret) break;
2254         }
2255         return;
2256     }
2257
2258     hkernel32 = GetModuleHandleA("kernel32.dll");
2259     pVirtualAllocEx = (void *) GetProcAddress(hkernel32, "VirtualAllocEx");
2260     pVirtualFreeEx = (void *) GetProcAddress(hkernel32, "VirtualFreeEx");
2261     pGetWriteWatch = (void *) GetProcAddress(hkernel32, "GetWriteWatch");
2262     pResetWriteWatch = (void *) GetProcAddress(hkernel32, "ResetWriteWatch");
2263     pNtAreMappedFilesTheSame = (void *)GetProcAddress( GetModuleHandle("ntdll.dll"),
2264                                                        "NtAreMappedFilesTheSame" );
2265     pNtMapViewOfSection = (void *)GetProcAddress(GetModuleHandle("ntdll.dll"), "NtMapViewOfSection");
2266     pNtUnmapViewOfSection = (void *)GetProcAddress(GetModuleHandle("ntdll.dll"), "NtUnmapViewOfSection");
2267
2268     test_mapping();
2269     test_CreateFileMapping_protection();
2270     test_VirtualAlloc_protection();
2271     test_VirtualProtect();
2272     test_VirtualAllocEx();
2273     test_VirtualAlloc();
2274     test_MapViewOfFile();
2275     test_NtMapViewOfSection();
2276     test_NtAreMappedFilesTheSame();
2277     test_CreateFileMapping();
2278     test_IsBadReadPtr();
2279     test_IsBadWritePtr();
2280     test_IsBadCodePtr();
2281     test_write_watch();
2282 }