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 "windef.h"
  25 #include "winbase.h"
  26 #include "winerror.h"
  27 #include "wine/test.h"
  28
  29 #define NUM_THREADS 4
  30 #define MAPPING_SIZE 0x100000
  31
  32 static HINSTANCE hkernel32;
  33 static LPVOID (WINAPI *pVirtualAllocEx)(HANDLE, LPVOID, SIZE_T, DWORD, DWORD);
  34 static BOOL   (WINAPI *pVirtualFreeEx)(HANDLE, LPVOID, SIZE_T, DWORD);
  35 static UINT   (WINAPI *pGetWriteWatch)(DWORD,LPVOID,SIZE_T,LPVOID*,ULONG_PTR*,ULONG*);
  36 static UINT   (WINAPI *pResetWriteWatch)(LPVOID,SIZE_T);
  37
  38 /* ############################### */
  39
  40 static HANDLE create_target_process(const char *arg)
  41 {
  42     char **argv;
  43     char cmdline[MAX_PATH];
  44     PROCESS_INFORMATION pi;
  45     STARTUPINFO si = { 0 };
  46     si.cb = sizeof(si);
  47
  48     winetest_get_mainargs( &argv );
  49     sprintf(cmdline, "%s %s %s", argv[0], argv[1], arg);
  50     ok(CreateProcess(NULL, cmdline, NULL, NULL, FALSE, 0, NULL, NULL,
  51                      &si, &pi) != 0, "error: %u\n", GetLastError());
  52     ok(CloseHandle(pi.hThread) != 0, "error %u\n", GetLastError());
  53     return pi.hProcess;
  54 }
  55
  56 static void test_VirtualAllocEx(void)
  57 {
  58     const unsigned int alloc_size = 1<<15;
  59     char *src, *dst;
  60     SIZE_T bytes_written = 0, bytes_read = 0, i;
  61     void *addr1, *addr2;
  62     BOOL b;
  63     DWORD old_prot;
  64     MEMORY_BASIC_INFORMATION info;
  65     HANDLE hProcess;
  66
  67     /* not exported in all windows-versions  */
  68     if ((!pVirtualAllocEx) || (!pVirtualFreeEx)) {
  69         win_skip("Virtual{Alloc,Free}Ex not available\n");
  70         return;
  71     }
  72
  73     hProcess = create_target_process("sleep");
  74     ok(hProcess != NULL, "Can't start process\n");
  75
  76     SetLastError(0xdeadbeef);
  77     addr1 = pVirtualAllocEx(hProcess, NULL, alloc_size, MEM_COMMIT,
  78                            PAGE_EXECUTE_READWRITE);
  79     if (!addr1 && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
  80     {   /* Win9x */
  81         win_skip("VirtualAllocEx not implemented\n");
  82         TerminateProcess(hProcess, 0);
  83         CloseHandle(hProcess);
  84         return;
  85     }
  86
  87     src = VirtualAlloc( NULL, alloc_size, MEM_COMMIT, PAGE_READWRITE );
  88     dst = VirtualAlloc( NULL, alloc_size, MEM_COMMIT, PAGE_READWRITE );
  89     for (i = 0; i < alloc_size; i++)
  90         src[i] = i & 0xff;
  91
  92     ok(addr1 != NULL, "VirtualAllocEx error %u\n", GetLastError());
  93     b = WriteProcessMemory(hProcess, addr1, src, alloc_size, &bytes_written);
  94     ok(b && (bytes_written == alloc_size), "%lu bytes written\n",
  95        bytes_written);
  96     b = ReadProcessMemory(hProcess, addr1, dst, alloc_size, &bytes_read);
  97     ok(b && (bytes_read == alloc_size), "%lu bytes read\n", bytes_read);
  98     ok(!memcmp(src, dst, alloc_size), "Data from remote process differs\n");
  99
 100     /* test invalid source buffers */
 101
 102     b = VirtualProtect( src + 0x2000, 0x2000, PAGE_NOACCESS, &old_prot );
 103     ok( b, "VirtualProtect failed error %u\n", GetLastError() );
 104     b = WriteProcessMemory(hProcess, addr1, src, alloc_size, &bytes_written);
 105     ok( !b, "WriteProcessMemory succeeded\n" );
 106     ok( GetLastError() == ERROR_NOACCESS ||
 107         GetLastError() == ERROR_PARTIAL_COPY, /* vista */
 108         "wrong error %u\n", GetLastError() );
 109     ok( bytes_written == 0, "%lu bytes written\n", bytes_written );
 110     b = ReadProcessMemory(hProcess, addr1, src, alloc_size, &bytes_read);
 111     ok( !b, "ReadProcessMemory succeeded\n" );
 112     ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
 113     ok( bytes_read == 0, "%lu bytes written\n", bytes_read );
 114
 115     b = VirtualProtect( src, 0x2000, PAGE_NOACCESS, &old_prot );
 116     ok( b, "VirtualProtect failed error %u\n", GetLastError() );
 117     b = WriteProcessMemory(hProcess, addr1, src, alloc_size, &bytes_written);
 118     ok( !b, "WriteProcessMemory succeeded\n" );
 119     ok( GetLastError() == ERROR_NOACCESS ||
 120         GetLastError() == ERROR_PARTIAL_COPY, /* vista */
 121         "wrong error %u\n", GetLastError() );
 122     ok( bytes_written == 0, "%lu bytes written\n", bytes_written );
 123     b = ReadProcessMemory(hProcess, addr1, src, alloc_size, &bytes_read);
 124     ok( !b, "ReadProcessMemory succeeded\n" );
 125     ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
 126     ok( bytes_read == 0, "%lu bytes written\n", bytes_read );
 127
 128     b = pVirtualFreeEx(hProcess, addr1, 0, MEM_RELEASE);
 129     ok(b != 0, "VirtualFreeEx, error %u\n", GetLastError());
 130
 131     VirtualFree( src, 0, MEM_FREE );
 132     VirtualFree( dst, 0, MEM_FREE );
 133
 134     /*
 135      * The following tests parallel those in test_VirtualAlloc()
 136      */
 137
 138     SetLastError(0xdeadbeef);
 139     addr1 = pVirtualAllocEx(hProcess, 0, 0, MEM_RESERVE, PAGE_NOACCESS);
 140     ok(addr1 == NULL, "VirtualAllocEx should fail on zero-sized allocation\n");
 141     ok(GetLastError() == ERROR_INVALID_PARAMETER /* NT */ ||
 142        GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Win9x */
 143         "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 144
 145     addr1 = pVirtualAllocEx(hProcess, 0, 0xFFFC, MEM_RESERVE, PAGE_NOACCESS);
 146     ok(addr1 != NULL, "VirtualAllocEx failed\n");
 147
 148     /* test a not committed memory */
 149     memset(&info, 'q', sizeof(info));
 150     ok(VirtualQueryEx(hProcess, addr1, &info, sizeof(info)) == sizeof(info), "VirtualQueryEx failed\n");
 151     ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
 152     ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
 153     ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
 154     ok(info.RegionSize == 0x10000, "%lx != 0x10000\n", info.RegionSize);
 155     ok(info.State == MEM_RESERVE, "%x != MEM_RESERVE\n", info.State);
 156     /* NT reports Protect == 0 for a not committed memory block */
 157     ok(info.Protect == 0 /* NT */ ||
 158        info.Protect == PAGE_NOACCESS, /* Win9x */
 159         "%x != PAGE_NOACCESS\n", info.Protect);
 160     ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
 161
 162     SetLastError(0xdeadbeef);
 163     ok(!VirtualProtectEx(hProcess, addr1, 0xFFFC, PAGE_READONLY, &old_prot),
 164        "VirtualProtectEx should fail on a not committed memory\n");
 165     ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
 166        GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
 167         "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
 168
 169     addr2 = pVirtualAllocEx(hProcess, addr1, 0x1000, MEM_COMMIT, PAGE_NOACCESS);
 170     ok(addr1 == addr2, "VirtualAllocEx failed\n");
 171
 172     /* test a committed memory */
 173     ok(VirtualQueryEx(hProcess, addr1, &info, sizeof(info)) == sizeof(info),
 174         "VirtualQueryEx failed\n");
 175     ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
 176     ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
 177     ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
 178     ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
 179     ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
 180     /* this time NT reports PAGE_NOACCESS as well */
 181     ok(info.Protect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.Protect);
 182     ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
 183
 184     /* this should fail, since not the whole range is committed yet */
 185     SetLastError(0xdeadbeef);
 186     ok(!VirtualProtectEx(hProcess, addr1, 0xFFFC, PAGE_READONLY, &old_prot),
 187         "VirtualProtectEx should fail on a not committed memory\n");
 188     ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
 189        GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
 190         "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError());
 191
 192     old_prot = 0;
 193     ok(VirtualProtectEx(hProcess, addr1, 0x1000, PAGE_READONLY, &old_prot), "VirtualProtectEx failed\n");
 194     ok(old_prot == PAGE_NOACCESS, "wrong old protection: got %04x instead of PAGE_NOACCESS\n", old_prot);
 195
 196     old_prot = 0;
 197     ok(VirtualProtectEx(hProcess, addr1, 0x1000, PAGE_READWRITE, &old_prot), "VirtualProtectEx failed\n");
 198     ok(old_prot == PAGE_READONLY, "wrong old protection: got %04x instead of PAGE_READONLY\n", old_prot);
 199
 200     ok(!pVirtualFreeEx(hProcess, addr1, 0x10000, 0),
 201        "VirtualFreeEx should fail with type 0\n");
 202     ok(GetLastError() == ERROR_INVALID_PARAMETER,
 203         "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 204
 205     ok(pVirtualFreeEx(hProcess, addr1, 0x10000, MEM_DECOMMIT), "VirtualFreeEx failed\n");
 206
 207     /* if the type is MEM_RELEASE, size must be 0 */
 208     ok(!pVirtualFreeEx(hProcess, addr1, 1, MEM_RELEASE),
 209        "VirtualFreeEx should fail\n");
 210     ok(GetLastError() == ERROR_INVALID_PARAMETER,
 211         "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 212
 213     ok(pVirtualFreeEx(hProcess, addr1, 0, MEM_RELEASE), "VirtualFreeEx failed\n");
 214
 215     TerminateProcess(hProcess, 0);
 216     CloseHandle(hProcess);
 217 }
 218
 219 static void test_VirtualAlloc(void)
 220 {
 221     void *addr1, *addr2;
 222     DWORD old_prot;
 223     MEMORY_BASIC_INFORMATION info;
 224
 225     SetLastError(0xdeadbeef);
 226     addr1 = VirtualAlloc(0, 0, MEM_RESERVE, PAGE_NOACCESS);
 227     ok(addr1 == NULL, "VirtualAlloc should fail on zero-sized allocation\n");
 228     ok(GetLastError() == ERROR_INVALID_PARAMETER /* NT */ ||
 229        GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Win9x */
 230         "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 231
 232     addr1 = VirtualAlloc(0, 0xFFFC, MEM_RESERVE, PAGE_NOACCESS);
 233     ok(addr1 != NULL, "VirtualAlloc failed\n");
 234
 235     /* test a not committed memory */
 236     ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
 237         "VirtualQuery failed\n");
 238     ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
 239     ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
 240     ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
 241     ok(info.RegionSize == 0x10000, "%lx != 0x10000\n", info.RegionSize);
 242     ok(info.State == MEM_RESERVE, "%x != MEM_RESERVE\n", info.State);
 243     /* NT reports Protect == 0 for a not committed memory block */
 244     ok(info.Protect == 0 /* NT */ ||
 245        info.Protect == PAGE_NOACCESS, /* Win9x */
 246         "%x != PAGE_NOACCESS\n", info.Protect);
 247     ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
 248
 249     SetLastError(0xdeadbeef);
 250     ok(!VirtualProtect(addr1, 0xFFFC, PAGE_READONLY, &old_prot),
 251        "VirtualProtect should fail on a not committed memory\n");
 252     ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
 253        GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
 254         "got %d, expected ERROR_INVALID_ADDRESS\n", GetLastError());
 255
 256     addr2 = VirtualAlloc(addr1, 0x1000, MEM_COMMIT, PAGE_NOACCESS);
 257     ok(addr1 == addr2, "VirtualAlloc failed\n");
 258
 259     /* test a committed memory */
 260     ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
 261         "VirtualQuery failed\n");
 262     ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1);
 263     ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1);
 264     ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect);
 265     ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
 266     ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
 267     /* this time NT reports PAGE_NOACCESS as well */
 268     ok(info.Protect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.Protect);
 269     ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type);
 270
 271     /* this should fail, since not the whole range is committed yet */
 272     SetLastError(0xdeadbeef);
 273     ok(!VirtualProtect(addr1, 0xFFFC, PAGE_READONLY, &old_prot),
 274         "VirtualProtect should fail on a not committed memory\n");
 275     ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ ||
 276        GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */
 277         "got %d, expected ERROR_INVALID_ADDRESS\n", GetLastError());
 278
 279     ok(VirtualProtect(addr1, 0x1000, PAGE_READONLY, &old_prot), "VirtualProtect failed\n");
 280     ok(old_prot == PAGE_NOACCESS,
 281         "wrong old protection: got %04x instead of PAGE_NOACCESS\n", old_prot);
 282
 283     ok(VirtualProtect(addr1, 0x1000, PAGE_READWRITE, &old_prot), "VirtualProtect failed\n");
 284     ok(old_prot == PAGE_READONLY,
 285         "wrong old protection: got %04x instead of PAGE_READONLY\n", old_prot);
 286
 287     ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
 288         "VirtualQuery failed\n");
 289     ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
 290     ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
 291     ok(info.Protect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.Protect);
 292     memset( addr1, 0x55, 20 );
 293     ok( *(DWORD *)addr1 == 0x55555555, "wrong data %x\n", *(DWORD *)addr1 );
 294
 295     addr2 = VirtualAlloc( addr1, 0x1000, MEM_RESET, PAGE_NOACCESS );
 296     ok( addr2 == addr1 || broken( !addr2 && GetLastError() == ERROR_INVALID_PARAMETER), /* win9x */
 297         "VirtualAlloc failed err %u\n", GetLastError() );
 298     ok( *(DWORD *)addr1 == 0x55555555 || *(DWORD *)addr1 == 0, "wrong data %x\n", *(DWORD *)addr1 );
 299     if (addr2)
 300     {
 301         ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info),
 302            "VirtualQuery failed\n");
 303         ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize);
 304         ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State);
 305         ok(info.Protect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.Protect);
 306
 307         addr2 = VirtualAlloc( (char *)addr1 + 0x1000, 0x1000, MEM_RESET, PAGE_NOACCESS );
 308         ok( (char *)addr2 == (char *)addr1 + 0x1000, "VirtualAlloc failed\n" );
 309
 310         ok(VirtualQuery(addr2, &info, sizeof(info)) == sizeof(info),
 311            "VirtualQuery failed\n");
 312         ok(info.RegionSize == 0xf000, "%lx != 0xf000\n", info.RegionSize);
 313         ok(info.State == MEM_RESERVE, "%x != MEM_RESERVE\n", info.State);
 314         ok(info.Protect == 0, "%x != 0\n", info.Protect);
 315
 316         addr2 = VirtualAlloc( (char *)addr1 + 0xf000, 0x2000, MEM_RESET, PAGE_NOACCESS );
 317         ok( !addr2, "VirtualAlloc failed\n" );
 318         ok( GetLastError() == ERROR_INVALID_ADDRESS, "wrong error %u\n", GetLastError() );
 319     }
 320
 321     /* invalid protection values */
 322     SetLastError(0xdeadbeef);
 323     addr2 = VirtualAlloc(NULL, 0x1000, MEM_RESERVE, 0);
 324     ok(!addr2, "VirtualAlloc succeeded\n");
 325     ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
 326
 327     SetLastError(0xdeadbeef);
 328     addr2 = VirtualAlloc(NULL, 0x1000, MEM_COMMIT, 0);
 329     ok(!addr2, "VirtualAlloc succeeded\n");
 330     ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
 331
 332     SetLastError(0xdeadbeef);
 333     addr2 = VirtualAlloc(addr1, 0x1000, MEM_COMMIT, PAGE_READONLY | PAGE_EXECUTE);
 334     ok(!addr2, "VirtualAlloc succeeded\n");
 335     ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
 336
 337     SetLastError(0xdeadbeef);
 338     ok(!VirtualProtect(addr1, 0x1000, PAGE_READWRITE | PAGE_EXECUTE_WRITECOPY, &old_prot),
 339        "VirtualProtect succeeded\n");
 340     ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
 341
 342     SetLastError(0xdeadbeef);
 343     ok(!VirtualProtect(addr1, 0x1000, 0, &old_prot), "VirtualProtect succeeded\n");
 344     ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
 345
 346     SetLastError(0xdeadbeef);
 347     ok(!VirtualFree(addr1, 0x10000, 0), "VirtualFree should fail with type 0\n");
 348     ok(GetLastError() == ERROR_INVALID_PARAMETER,
 349         "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 350
 351     ok(VirtualFree(addr1, 0x10000, MEM_DECOMMIT), "VirtualFree failed\n");
 352
 353     /* if the type is MEM_RELEASE, size must be 0 */
 354     ok(!VirtualFree(addr1, 1, MEM_RELEASE), "VirtualFree should fail\n");
 355     ok(GetLastError() == ERROR_INVALID_PARAMETER,
 356         "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError());
 357
 358     ok(VirtualFree(addr1, 0, MEM_RELEASE), "VirtualFree failed\n");
 359 }
 360
 361 static void test_MapViewOfFile(void)
 362 {
 363     static const char testfile[] = "testfile.xxx";
 364     const char *name;
 365     HANDLE file, mapping, map2;
 366     void *ptr, *ptr2, *addr;
 367     MEMORY_BASIC_INFORMATION info;
 368     BOOL ret;
 369
 370     SetLastError(0xdeadbeef);
 371     file = CreateFileA( testfile, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0 );
 372     ok( file != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
 373     SetFilePointer( file, 4096, NULL, FILE_BEGIN );
 374     SetEndOfFile( file );
 375
 376     /* read/write mapping */
 377
 378     SetLastError(0xdeadbeef);
 379     mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
 380     ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
 381
 382     SetLastError(0xdeadbeef);
 383     ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
 384     ok( ptr != NULL, "MapViewOfFile FILE_MAPE_READ error %u\n", GetLastError() );
 385     UnmapViewOfFile( ptr );
 386
 387     /* this fails on win9x but succeeds on NT */
 388     SetLastError(0xdeadbeef);
 389     ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
 390     if (ptr) UnmapViewOfFile( ptr );
 391     else ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
 392
 393     SetLastError(0xdeadbeef);
 394     ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
 395     ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
 396     UnmapViewOfFile( ptr );
 397
 398     SetLastError(0xdeadbeef);
 399     ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
 400     ok( ptr != NULL, "MapViewOfFile FILE_MAP_WRITE error %u\n", GetLastError() );
 401     UnmapViewOfFile( ptr );
 402
 403     ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2,
 404                            FILE_MAP_READ|FILE_MAP_WRITE, FALSE, 0 );
 405     ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
 406     ptr = MapViewOfFile( map2, FILE_MAP_WRITE, 0, 0, 4096 );
 407     ok( ptr != NULL, "MapViewOfFile FILE_MAP_WRITE error %u\n", GetLastError() );
 408     UnmapViewOfFile( ptr );
 409     CloseHandle( map2 );
 410
 411     ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2,
 412                            FILE_MAP_READ, FALSE, 0 );
 413     ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
 414     ptr = MapViewOfFile( map2, FILE_MAP_WRITE, 0, 0, 4096 );
 415     if (!ptr)
 416     {
 417         ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 418         CloseHandle( map2 );
 419         ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2, 0, FALSE, 0 );
 420         ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
 421         ptr = MapViewOfFile( map2, 0, 0, 0, 4096 );
 422         ok( !ptr, "MapViewOfFile succeeded\n" );
 423         ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 424         CloseHandle( map2 );
 425         ret = DuplicateHandle( GetCurrentProcess(), mapping, GetCurrentProcess(), &map2,
 426                                FILE_MAP_READ, FALSE, 0 );
 427         ok( ret, "DuplicateHandle failed error %u\n", GetLastError());
 428         ptr = MapViewOfFile( map2, 0, 0, 0, 4096 );
 429         ok( ptr != NULL, "MapViewOfFile NO_ACCESS error %u\n", GetLastError() );
 430     }
 431     else win_skip( "no access checks on win9x\n" );
 432
 433     UnmapViewOfFile( ptr );
 434     CloseHandle( map2 );
 435     CloseHandle( mapping );
 436
 437     /* read-only mapping */
 438
 439     SetLastError(0xdeadbeef);
 440     mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL );
 441     ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
 442
 443     SetLastError(0xdeadbeef);
 444     ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
 445     ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 446     UnmapViewOfFile( ptr );
 447
 448     /* this fails on win9x but succeeds on NT */
 449     SetLastError(0xdeadbeef);
 450     ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
 451     if (ptr) UnmapViewOfFile( ptr );
 452     else ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
 453
 454     SetLastError(0xdeadbeef);
 455     ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
 456     ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
 457     UnmapViewOfFile( ptr );
 458
 459     SetLastError(0xdeadbeef);
 460     ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
 461     ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" );
 462     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 463         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 464     CloseHandle( mapping );
 465
 466     /* copy-on-write mapping */
 467
 468     SetLastError(0xdeadbeef);
 469     mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL );
 470     ok( mapping != 0, "CreateFileMapping error %u\n", GetLastError() );
 471
 472     SetLastError(0xdeadbeef);
 473     ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
 474     ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 475     UnmapViewOfFile( ptr );
 476
 477     SetLastError(0xdeadbeef);
 478     ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
 479     ok( ptr != NULL, "MapViewOfFile FILE_MAP_COPY error %u\n", GetLastError() );
 480     UnmapViewOfFile( ptr );
 481
 482     SetLastError(0xdeadbeef);
 483     ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
 484     ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
 485     UnmapViewOfFile( ptr );
 486
 487     SetLastError(0xdeadbeef);
 488     ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
 489     ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" );
 490     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 491         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 492     CloseHandle( mapping );
 493
 494     /* no access mapping */
 495
 496     SetLastError(0xdeadbeef);
 497     mapping = CreateFileMappingA( file, NULL, PAGE_NOACCESS, 0, 4096, NULL );
 498     /* fails on NT but succeeds on win9x */
 499     if (!mapping) ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
 500     else
 501     {
 502         SetLastError(0xdeadbeef);
 503         ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 );
 504         ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 505         UnmapViewOfFile( ptr );
 506
 507         SetLastError(0xdeadbeef);
 508         ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 );
 509         ok( !ptr, "MapViewOfFile FILE_MAP_COPY succeeded\n" );
 510         ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
 511
 512         SetLastError(0xdeadbeef);
 513         ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 );
 514         ok( ptr != NULL, "MapViewOfFile 0 error %u\n", GetLastError() );
 515         UnmapViewOfFile( ptr );
 516
 517         SetLastError(0xdeadbeef);
 518         ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 );
 519         ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" );
 520         ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() );
 521
 522         CloseHandle( mapping );
 523     }
 524
 525     CloseHandle( file );
 526
 527     /* now try read-only file */
 528
 529     SetLastError(0xdeadbeef);
 530     file = CreateFileA( testfile, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0 );
 531     ok( file != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
 532
 533     SetLastError(0xdeadbeef);
 534     mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
 535     ok( !mapping, "CreateFileMapping PAGE_READWRITE succeeded\n" );
 536     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 537         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 538
 539     SetLastError(0xdeadbeef);
 540     mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL );
 541     ok( mapping != 0, "CreateFileMapping PAGE_WRITECOPY error %u\n", GetLastError() );
 542     CloseHandle( mapping );
 543
 544     SetLastError(0xdeadbeef);
 545     mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL );
 546     ok( mapping != 0, "CreateFileMapping PAGE_READONLY error %u\n", GetLastError() );
 547     CloseHandle( mapping );
 548     CloseHandle( file );
 549
 550     /* now try no access file */
 551
 552     SetLastError(0xdeadbeef);
 553     file = CreateFileA( testfile, 0, 0, NULL, OPEN_EXISTING, 0, 0 );
 554     ok( file != INVALID_HANDLE_VALUE, "CreateFile error %u\n", GetLastError() );
 555
 556     SetLastError(0xdeadbeef);
 557     mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
 558     ok( !mapping, "CreateFileMapping PAGE_READWRITE succeeded\n" );
 559     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 560         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 561
 562     SetLastError(0xdeadbeef);
 563     mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL );
 564     ok( !mapping, "CreateFileMapping PAGE_WRITECOPY succeeded\n" );
 565     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 566         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 567
 568     SetLastError(0xdeadbeef);
 569     mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL );
 570     ok( !mapping, "CreateFileMapping PAGE_READONLY succeeded\n" );
 571     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 572         GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 573
 574     CloseHandle( file );
 575     DeleteFileA( testfile );
 576
 577     SetLastError(0xdeadbeef);
 578     name = "Local\\Foo";
 579     file = CreateFileMapping( INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, 4096, name );
 580     /* nt4 doesn't have Local\\ */
 581     if (!file && GetLastError() == ERROR_PATH_NOT_FOUND)
 582     {
 583         name = "Foo";
 584         file = CreateFileMapping( INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, 4096, name );
 585     }
 586     ok( file != 0, "CreateFileMapping PAGE_READWRITE error %u\n", GetLastError() );
 587
 588     SetLastError(0xdeadbeef);
 589     mapping = OpenFileMapping( FILE_MAP_READ, FALSE, name );
 590     ok( mapping != 0, "OpenFileMapping FILE_MAP_READ error %u\n", GetLastError() );
 591     SetLastError(0xdeadbeef);
 592     ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 0 );
 593     if (!ptr)
 594     {
 595         ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 596         SetLastError(0xdeadbeef);
 597         ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 0 );
 598         ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ error %u\n", GetLastError() );
 599         SetLastError(0xdeadbeef);
 600         ok( VirtualQuery( ptr, &info, sizeof(info) ) == sizeof(info),
 601             "VirtualQuery error %u\n", GetLastError() );
 602         ok( info.BaseAddress == ptr, "%p != %p\n", info.BaseAddress, ptr );
 603         ok( info.AllocationBase == ptr, "%p != %p\n", info.AllocationBase, ptr );
 604         ok( info.AllocationProtect == PAGE_READONLY, "%x != PAGE_READONLY\n", info.AllocationProtect );
 605         ok( info.RegionSize == 4096, "%lx != 4096\n", info.RegionSize );
 606         ok( info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State );
 607         ok( info.Protect == PAGE_READONLY, "%x != PAGE_READONLY\n", info.Protect );
 608     }
 609     else win_skip( "no access checks on win9x\n" );
 610     UnmapViewOfFile( ptr );
 611     CloseHandle( mapping );
 612
 613     SetLastError(0xdeadbeef);
 614     mapping = OpenFileMapping( FILE_MAP_WRITE, FALSE, name );
 615     ok( mapping != 0, "OpenFileMapping FILE_MAP_WRITE error %u\n", GetLastError() );
 616     SetLastError(0xdeadbeef);
 617     ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 0 );
 618     if (!ptr)
 619     {
 620         ok( GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() );
 621         SetLastError(0xdeadbeef);
 622         ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 0 );
 623         ok( ptr != NULL, "MapViewOfFile FILE_MAP_WRITE error %u\n", GetLastError() );
 624         SetLastError(0xdeadbeef);
 625         ok( VirtualQuery( ptr, &info, sizeof(info) ) == sizeof(info),
 626             "VirtualQuery error %u\n", GetLastError() );
 627         ok( info.BaseAddress == ptr, "%p != %p\n", info.BaseAddress, ptr );
 628         ok( info.AllocationBase == ptr, "%p != %p\n", info.AllocationBase, ptr );
 629         ok( info.AllocationProtect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.AllocationProtect );
 630         ok( info.RegionSize == 4096, "%lx != 4096\n", info.RegionSize );
 631         ok( info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State );
 632         ok( info.Protect == PAGE_READWRITE, "%x != PAGE_READWRITE\n", info.Protect );
 633     }
 634     else win_skip( "no access checks on win9x\n" );
 635     UnmapViewOfFile( ptr );
 636     CloseHandle( mapping );
 637
 638     CloseHandle( file );
 639
 640     /* read/write mapping with SEC_RESERVE */
 641     mapping = CreateFileMappingA(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE | SEC_RESERVE, 0, MAPPING_SIZE, NULL);
 642     ok(mapping != INVALID_HANDLE_VALUE, "CreateFileMappingA failed with error %d\n", GetLastError());
 643
 644     ptr = MapViewOfFile(mapping, FILE_MAP_WRITE, 0, 0, 0);
 645     ok(ptr != NULL, "MapViewOfFile failed with error %d\n", GetLastError());
 646
 647     ptr2 = MapViewOfFile(mapping, FILE_MAP_WRITE, 0, 0, 0);
 648     /* on NT ptr != ptr2 but on Win9x ptr == ptr2 */
 649     ok(ptr2 != NULL, "MapViewOfFile failed with error %d\n", GetLastError());
 650     trace("mapping same section resulted in views %p and %p\n", ptr, ptr2);
 651
 652     ret = VirtualQuery(ptr, &info, sizeof(info));
 653     ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
 654     ok(info.BaseAddress == ptr, "BaseAddress should have been %p but was %p instead\n", ptr, info.BaseAddress);
 655     ok(info.AllocationBase == ptr, "AllocationBase should have been %p but was %p instead\n", ptr, info.AllocationBase);
 656     ok(info.RegionSize == MAPPING_SIZE, "RegionSize should have been 0x%x but was 0x%x\n", MAPPING_SIZE, (unsigned int)info.RegionSize);
 657     ok(info.State == MEM_RESERVE, "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
 658     if (info.Type == MEM_PRIVATE)  /* win9x is different for uncommitted mappings */
 659     {
 660         ok(info.AllocationProtect == PAGE_NOACCESS,
 661            "AllocationProtect should have been PAGE_NOACCESS but was 0x%x\n", info.AllocationProtect);
 662         ok(info.Protect == PAGE_NOACCESS,
 663            "Protect should have been PAGE_NOACCESS instead of 0x%x\n", info.Protect);
 664     }
 665     else
 666     {
 667         ok(info.AllocationProtect == PAGE_READWRITE,
 668            "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
 669         ok(info.Protect == 0, "Protect should have been 0 instead of 0x%x\n", info.Protect);
 670         ok(info.Type == MEM_MAPPED, "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
 671     }
 672
 673     if (ptr != ptr2)
 674     {
 675         ret = VirtualQuery(ptr2, &info, sizeof(info));
 676         ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
 677         ok(info.BaseAddress == ptr2,
 678            "BaseAddress should have been %p but was %p instead\n", ptr2, info.BaseAddress);
 679         ok(info.AllocationBase == ptr2,
 680            "AllocationBase should have been %p but was %p instead\n", ptr2, info.AllocationBase);
 681         ok(info.AllocationProtect == PAGE_READWRITE,
 682            "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
 683         ok(info.RegionSize == MAPPING_SIZE,
 684            "RegionSize should have been 0x%x but was 0x%x\n", MAPPING_SIZE, (unsigned int)info.RegionSize);
 685         ok(info.State == MEM_RESERVE,
 686            "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
 687         ok(info.Protect == 0,
 688            "Protect should have been 0 instead of 0x%x\n", info.Protect);
 689         ok(info.Type == MEM_MAPPED,
 690            "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
 691     }
 692
 693     ptr = VirtualAlloc(ptr, 0x10000, MEM_COMMIT, PAGE_READONLY);
 694     ok(ptr != NULL, "VirtualAlloc failed with error %d\n", GetLastError());
 695
 696     ret = VirtualQuery(ptr, &info, sizeof(info));
 697     ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
 698     ok(info.BaseAddress == ptr, "BaseAddress should have been %p but was %p instead\n", ptr, info.BaseAddress);
 699     ok(info.AllocationBase == ptr, "AllocationBase should have been %p but was %p instead\n", ptr, info.AllocationBase);
 700     ok(info.RegionSize == 0x10000, "RegionSize should have been 0x10000 but was 0x%x\n", (unsigned int)info.RegionSize);
 701     ok(info.State == MEM_COMMIT, "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
 702     ok(info.Protect == PAGE_READONLY, "Protect should have been 0 instead of 0x%x\n", info.Protect);
 703     if (info.Type == MEM_PRIVATE)  /* win9x is different for uncommitted mappings */
 704     {
 705         ok(info.AllocationProtect == PAGE_NOACCESS,
 706            "AllocationProtect should have been PAGE_NOACCESS but was 0x%x\n", info.AllocationProtect);
 707     }
 708     else
 709     {
 710         ok(info.AllocationProtect == PAGE_READWRITE,
 711            "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
 712         ok(info.Type == MEM_MAPPED, "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
 713     }
 714
 715     /* shows that the VirtualAlloc above affects the mapping, not just the
 716      * virtual memory in this process - it also affects all other processes
 717      * with a view of the mapping, but that isn't tested here */
 718     if (ptr != ptr2)
 719     {
 720         ret = VirtualQuery(ptr2, &info, sizeof(info));
 721         ok(ret, "VirtualQuery failed with error %d\n", GetLastError());
 722         ok(info.BaseAddress == ptr2,
 723            "BaseAddress should have been %p but was %p instead\n", ptr2, info.BaseAddress);
 724         ok(info.AllocationBase == ptr2,
 725            "AllocationBase should have been %p but was %p instead\n", ptr2, info.AllocationBase);
 726         ok(info.AllocationProtect == PAGE_READWRITE,
 727            "AllocationProtect should have been PAGE_READWRITE but was 0x%x\n", info.AllocationProtect);
 728         ok(info.RegionSize == 0x10000,
 729            "RegionSize should have been 0x10000 but was 0x%x\n", (unsigned int)info.RegionSize);
 730         ok(info.State == MEM_COMMIT,
 731            "State should have been MEM_RESERVE instead of 0x%x\n", info.State);
 732         ok(info.Protect == PAGE_READWRITE,
 733            "Protect should have been 0 instead of 0x%x\n", info.Protect);
 734         ok(info.Type == MEM_MAPPED, "Type should have been MEM_MAPPED instead of 0x%x\n", info.Type);
 735     }
 736
 737     addr = VirtualAlloc( ptr, MAPPING_SIZE, MEM_RESET, PAGE_READONLY );
 738     ok( addr == ptr || broken(!addr && GetLastError() == ERROR_INVALID_PARAMETER), /* win9x */
 739         "VirtualAlloc failed with error %u\n", GetLastError() );
 740
 741     ret = VirtualFree( ptr, 0x10000, MEM_DECOMMIT );
 742     ok( !ret || broken(ret) /* win9x */, "VirtualFree succeeded\n" );
 743     if (!ret)
 744         ok( GetLastError() == ERROR_INVALID_PARAMETER, "VirtualFree failed with %u\n", GetLastError() );
 745
 746     ret = UnmapViewOfFile(ptr2);
 747     ok(ret, "UnmapViewOfFile failed with error %d\n", GetLastError());
 748     ret = UnmapViewOfFile(ptr);
 749     ok(ret, "UnmapViewOfFile failed with error %d\n", GetLastError());
 750     CloseHandle(mapping);
 751 }
 752
 753 static DWORD (WINAPI *pNtMapViewOfSection)( HANDLE handle, HANDLE process, PVOID *addr_ptr,
 754                                             ULONG zero_bits, SIZE_T commit_size,
 755                                             const LARGE_INTEGER *offset_ptr, SIZE_T *size_ptr,
 756                                             ULONG inherit, ULONG alloc_type, ULONG protect );
 757 static DWORD (WINAPI *pNtUnmapViewOfSection)( HANDLE process, PVOID addr );
 758
 759 static void test_NtMapViewOfSection(void)
 760 {
 761     HANDLE hProcess;
 762
 763     static const char testfile[] = "testfile.xxx";
 764     static const char data[] = "test data for NtMapViewOfSection";
 765     char buffer[sizeof(data)];
 766     HANDLE file, mapping;
 767     void *ptr;
 768     BOOL ret;
 769     DWORD status, written;
 770     SIZE_T size, result;
 771     LARGE_INTEGER offset;
 772
 773     pNtMapViewOfSection = (void *)GetProcAddress( GetModuleHandle("ntdll.dll"), "NtMapViewOfSection" );
 774     pNtUnmapViewOfSection = (void *)GetProcAddress( GetModuleHandle("ntdll.dll"), "NtUnmapViewOfSection" );
 775     if (!pNtMapViewOfSection || !pNtUnmapViewOfSection)
 776     {
 777         win_skip( "NtMapViewOfSection not available\n" );
 778         return;
 779     }
 780
 781     file = CreateFileA( testfile, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0 );
 782     ok( file != INVALID_HANDLE_VALUE, "Failed to create test file\n" );
 783     WriteFile( file, data, sizeof(data), &written, NULL );
 784     SetFilePointer( file, 4096, NULL, FILE_BEGIN );
 785     SetEndOfFile( file );
 786
 787     /* read/write mapping */
 788
 789     mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL );
 790     ok( mapping != 0, "CreateFileMapping failed\n" );
 791
 792     hProcess = create_target_process("sleep");
 793     ok(hProcess != NULL, "Can't start process\n");
 794
 795     ptr = NULL;
 796     size = 0;
 797     offset.QuadPart = 0;
 798     status = pNtMapViewOfSection( mapping, hProcess, &ptr, 0, 0, &offset, &size, 1, 0, PAGE_READWRITE );
 799     ok( !status, "NtMapViewOfSection failed status %x\n", status );
 800
 801     ret = ReadProcessMemory( hProcess, ptr, buffer, sizeof(buffer), &result );
 802     ok( ret, "ReadProcessMemory failed\n" );
 803     ok( result == sizeof(buffer), "ReadProcessMemory didn't read all data (%lx)\n", result );
 804     ok( !memcmp( buffer, data, sizeof(buffer) ), "Wrong data read\n" );
 805
 806     status = pNtUnmapViewOfSection( hProcess, ptr );
 807     ok( !status, "NtUnmapViewOfSection failed status %x\n", status );
 808
 809     CloseHandle( mapping );
 810     CloseHandle( file );
 811     DeleteFileA( testfile );
 812
 813     TerminateProcess(hProcess, 0);
 814     CloseHandle(hProcess);
 815 }
 816
 817 static void test_CreateFileMapping(void)
 818 {
 819     HANDLE handle, handle2;
 820
 821     /* test case sensitivity */
 822
 823     SetLastError(0xdeadbeef);
 824     handle = CreateFileMappingA( INVALID_HANDLE_VALUE, NULL, SEC_COMMIT | PAGE_READWRITE, 0, 0x1000,
 825                                  "Wine Test Mapping");
 826     ok( handle != NULL, "CreateFileMapping failed with error %u\n", GetLastError());
 827     ok( GetLastError() == 0, "wrong error %u\n", GetLastError());
 828
 829     SetLastError(0xdeadbeef);
 830     handle2 = CreateFileMappingA( INVALID_HANDLE_VALUE, NULL, SEC_COMMIT | PAGE_READWRITE, 0, 0x1000,
 831                                   "Wine Test Mapping");
 832     ok( handle2 != NULL, "CreateFileMapping failed with error %d\n", GetLastError());
 833     ok( GetLastError() == ERROR_ALREADY_EXISTS, "wrong error %u\n", GetLastError());
 834     CloseHandle( handle2 );
 835
 836     SetLastError(0xdeadbeef);
 837     handle2 = CreateFileMappingA( INVALID_HANDLE_VALUE, NULL, SEC_COMMIT | PAGE_READWRITE, 0, 0x1000,
 838                                  "WINE TEST MAPPING");
 839     ok( handle2 != NULL, "CreateFileMapping failed with error %d\n", GetLastError());
 840     ok( GetLastError() == 0, "wrong error %u\n", GetLastError());
 841     CloseHandle( handle2 );
 842
 843     SetLastError(0xdeadbeef);
 844     handle2 = OpenFileMappingA( FILE_MAP_ALL_ACCESS, FALSE, "Wine Test Mapping");
 845     ok( handle2 != NULL, "OpenFileMapping failed with error %d\n", GetLastError());
 846     CloseHandle( handle2 );
 847
 848     SetLastError(0xdeadbeef);
 849     handle2 = OpenFileMappingA( FILE_MAP_ALL_ACCESS, FALSE, "WINE TEST MAPPING");
 850     ok( !handle2, "OpenFileMapping succeeded\n");
 851     ok( GetLastError() == ERROR_FILE_NOT_FOUND || GetLastError() == ERROR_INVALID_NAME /* win9x */,
 852         "wrong error %u\n", GetLastError());
 853
 854     CloseHandle( handle );
 855 }
 856
 857 static void test_BadPtr(void)
 858 {
 859     void *ptr = (void*)1;
 860     /* We assume address 1 is not mapped. */
 861     ok(IsBadReadPtr(ptr,1),"IsBadReadPtr(1) failed.\n");
 862     ok(IsBadWritePtr(ptr,1),"IsBadWritePtr(1) failed.\n");
 863     ok(IsBadCodePtr(ptr),"IsBadCodePtr(1) failed.\n");
 864 }
 865
 866 static void test_write_watch(void)
 867 {
 868     char *base;
 869     DWORD ret, size, old_prot;
 870     MEMORY_BASIC_INFORMATION info;
 871     void *results[64];
 872     ULONG_PTR count;
 873     ULONG pagesize;
 874
 875     if (!pGetWriteWatch || !pResetWriteWatch)
 876     {
 877         win_skip( "GetWriteWatch not supported\n" );
 878         return;
 879     }
 880
 881     size = 0x10000;
 882     base = VirtualAlloc( 0, size, MEM_RESERVE | MEM_COMMIT | MEM_WRITE_WATCH, PAGE_READWRITE );
 883     if (!base &&
 884         (GetLastError() == ERROR_INVALID_PARAMETER || GetLastError() == ERROR_NOT_SUPPORTED))
 885     {
 886         win_skip( "MEM_WRITE_WATCH not supported\n" );
 887         return;
 888     }
 889     ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
 890     ret = VirtualQuery( base, &info, sizeof(info) );
 891     ok(ret, "VirtualQuery failed %u\n", GetLastError());
 892     ok( info.BaseAddress == base, "BaseAddress %p instead of %p\n", info.BaseAddress, base );
 893     ok( info.AllocationProtect == PAGE_READWRITE, "wrong AllocationProtect %x\n", info.AllocationProtect );
 894     ok( info.RegionSize == size, "wrong RegionSize 0x%lx\n", info.RegionSize );
 895     ok( info.State == MEM_COMMIT, "wrong State 0x%x\n", info.State );
 896     ok( info.Protect == PAGE_READWRITE, "wrong Protect 0x%x\n", info.Protect );
 897     ok( info.Type == MEM_PRIVATE, "wrong Type 0x%x\n", info.Type );
 898
 899     count = 64;
 900     SetLastError( 0xdeadbeef );
 901     ret = pGetWriteWatch( 0, NULL, size, results, &count, &pagesize );
 902     ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
 903     ok( GetLastError() == ERROR_INVALID_PARAMETER ||
 904         broken( GetLastError() == 0xdeadbeef ), /* win98 */
 905         "wrong error %u\n", GetLastError() );
 906
 907     SetLastError( 0xdeadbeef );
 908     ret = pGetWriteWatch( 0, GetModuleHandle(0), size, results, &count, &pagesize );
 909     if (ret)
 910     {
 911         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
 912         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
 913     }
 914     else  /* win98 */
 915     {
 916         ok( count == 0, "wrong count %lu\n", count );
 917     }
 918
 919     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
 920     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
 921     ok( count == 0, "wrong count %lu\n", count );
 922
 923     base[pagesize + 1] = 0x44;
 924
 925     count = 64;
 926     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
 927     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
 928     ok( count == 1, "wrong count %lu\n", count );
 929     ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
 930
 931     count = 64;
 932     ret = pGetWriteWatch( WRITE_WATCH_FLAG_RESET, base, size, results, &count, &pagesize );
 933     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
 934     ok( count == 1, "wrong count %lu\n", count );
 935     ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
 936
 937     count = 64;
 938     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
 939     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
 940     ok( count == 0, "wrong count %lu\n", count );
 941
 942     base[2*pagesize + 3] = 0x11;
 943     base[4*pagesize + 8] = 0x11;
 944
 945     count = 64;
 946     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
 947     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
 948     ok( count == 2, "wrong count %lu\n", count );
 949     ok( results[0] == base + 2*pagesize, "wrong result %p\n", results[0] );
 950     ok( results[1] == base + 4*pagesize, "wrong result %p\n", results[1] );
 951
 952     count = 64;
 953     ret = pGetWriteWatch( 0, base + 3*pagesize, 2*pagesize, results, &count, &pagesize );
 954     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
 955     ok( count == 1, "wrong count %lu\n", count );
 956     ok( results[0] == base + 4*pagesize, "wrong result %p\n", results[0] );
 957
 958     ret = pResetWriteWatch( base, 3*pagesize );
 959     ok( !ret, "pResetWriteWatch failed %u\n", GetLastError() );
 960
 961     count = 64;
 962     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
 963     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
 964     ok( count == 1, "wrong count %lu\n", count );
 965     ok( results[0] == base + 4*pagesize, "wrong result %p\n", results[0] );
 966
 967     *(DWORD *)(base + 2*pagesize - 2) = 0xdeadbeef;
 968
 969     count = 64;
 970     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
 971     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
 972     ok( count == 3, "wrong count %lu\n", count );
 973     ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
 974     ok( results[1] == base + 2*pagesize, "wrong result %p\n", results[1] );
 975     ok( results[2] == base + 4*pagesize, "wrong result %p\n", results[2] );
 976
 977     count = 1;
 978     ret = pGetWriteWatch( WRITE_WATCH_FLAG_RESET, base, size, results, &count, &pagesize );
 979     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
 980     ok( count == 1, "wrong count %lu\n", count );
 981     ok( results[0] == base + pagesize, "wrong result %p\n", results[0] );
 982
 983     count = 64;
 984     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
 985     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
 986     ok( count == 2, "wrong count %lu\n", count );
 987     ok( results[0] == base + 2*pagesize, "wrong result %p\n", results[0] );
 988     ok( results[1] == base + 4*pagesize, "wrong result %p\n", results[1] );
 989
 990     /* changing protections doesn't affect watches */
 991
 992     ret = VirtualProtect( base, 3*pagesize, PAGE_READONLY, &old_prot );
 993     ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
 994     ok( old_prot == PAGE_READWRITE, "wrong old prot %x\n", old_prot );
 995
 996     ret = VirtualQuery( base, &info, sizeof(info) );
 997     ok(ret, "VirtualQuery failed %u\n", GetLastError());
 998     ok( info.BaseAddress == base, "BaseAddress %p instead of %p\n", info.BaseAddress, base );
 999     ok( info.RegionSize == 3*pagesize, "wrong RegionSize 0x%lx\n", info.RegionSize );
1000     ok( info.State == MEM_COMMIT, "wrong State 0x%x\n", info.State );
1001     ok( info.Protect == PAGE_READONLY, "wrong Protect 0x%x\n", info.Protect );
1002
1003     ret = VirtualProtect( base, 3*pagesize, PAGE_READWRITE, &old_prot );
1004     ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
1005     ok( old_prot == PAGE_READONLY, "wrong old prot %x\n", old_prot );
1006
1007     count = 64;
1008     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1009     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1010     ok( count == 2, "wrong count %lu\n", count );
1011     ok( results[0] == base + 2*pagesize, "wrong result %p\n", results[0] );
1012     ok( results[1] == base + 4*pagesize, "wrong result %p\n", results[1] );
1013
1014     ret = VirtualQuery( base, &info, sizeof(info) );
1015     ok(ret, "VirtualQuery failed %u\n", GetLastError());
1016     ok( info.BaseAddress == base, "BaseAddress %p instead of %p\n", info.BaseAddress, base );
1017     ok( info.RegionSize == size, "wrong RegionSize 0x%lx\n", info.RegionSize );
1018     ok( info.State == MEM_COMMIT, "wrong State 0x%x\n", info.State );
1019     ok( info.Protect == PAGE_READWRITE, "wrong Protect 0x%x\n", info.Protect );
1020
1021     /* some invalid parameter tests */
1022
1023     SetLastError( 0xdeadbeef );
1024     count = 0;
1025     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1026     if (ret)
1027     {
1028         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1029         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1030
1031         SetLastError( 0xdeadbeef );
1032         ret = pGetWriteWatch( 0, base, size, results, NULL, &pagesize );
1033         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1034         ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
1035
1036         SetLastError( 0xdeadbeef );
1037         count = 64;
1038         ret = pGetWriteWatch( 0, base, size, results, &count, NULL );
1039         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1040         ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
1041
1042         SetLastError( 0xdeadbeef );
1043         count = 64;
1044         ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1045         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1046         ok( GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError() );
1047
1048         SetLastError( 0xdeadbeef );
1049         count = 0;
1050         ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1051         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1052         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1053
1054         SetLastError( 0xdeadbeef );
1055         count = 64;
1056         ret = pGetWriteWatch( 0xdeadbeef, base, size, results, &count, &pagesize );
1057         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1058         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1059
1060         SetLastError( 0xdeadbeef );
1061         count = 64;
1062         ret = pGetWriteWatch( 0, base, 0, results, &count, &pagesize );
1063         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1064         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1065
1066         SetLastError( 0xdeadbeef );
1067         count = 64;
1068         ret = pGetWriteWatch( 0, base, size * 2, results, &count, &pagesize );
1069         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1070         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1071
1072         SetLastError( 0xdeadbeef );
1073         count = 64;
1074         ret = pGetWriteWatch( 0, base + size - pagesize, pagesize + 1, results, &count, &pagesize );
1075         ok( ret == ~0u, "GetWriteWatch succeeded %u\n", ret );
1076         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1077
1078         SetLastError( 0xdeadbeef );
1079         ret = pResetWriteWatch( base, 0 );
1080         ok( ret == ~0u, "ResetWriteWatch succeeded %u\n", ret );
1081         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1082
1083         SetLastError( 0xdeadbeef );
1084         ret = pResetWriteWatch( GetModuleHandle(0), size );
1085         ok( ret == ~0u, "ResetWriteWatch succeeded %u\n", ret );
1086         ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1087     }
1088     else  /* win98 is completely different */
1089     {
1090         SetLastError( 0xdeadbeef );
1091         count = 64;
1092         ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1093         ok( ret == ERROR_INVALID_PARAMETER, "GetWriteWatch succeeded %u\n", ret );
1094         ok( GetLastError() == 0xdeadbeef, "wrong error %u\n", GetLastError() );
1095
1096         count = 0;
1097         ret = pGetWriteWatch( 0, base, size, NULL, &count, &pagesize );
1098         ok( !ret, "GetWriteWatch failed %u\n", ret );
1099
1100         count = 64;
1101         ret = pGetWriteWatch( 0xdeadbeef, base, size, results, &count, &pagesize );
1102         ok( !ret, "GetWriteWatch failed %u\n", ret );
1103
1104         count = 64;
1105         ret = pGetWriteWatch( 0, base, 0, results, &count, &pagesize );
1106         ok( !ret, "GetWriteWatch failed %u\n", ret );
1107
1108         ret = pResetWriteWatch( base, 0 );
1109         ok( !ret, "ResetWriteWatch failed %u\n", ret );
1110
1111         ret = pResetWriteWatch( GetModuleHandle(0), size );
1112         ok( !ret, "ResetWriteWatch failed %u\n", ret );
1113     }
1114
1115     VirtualFree( base, 0, MEM_FREE );
1116
1117     base = VirtualAlloc( 0, size, MEM_RESERVE | MEM_WRITE_WATCH, PAGE_READWRITE );
1118     ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
1119     VirtualFree( base, 0, MEM_FREE );
1120
1121     base = VirtualAlloc( 0, size, MEM_WRITE_WATCH, PAGE_READWRITE );
1122     ok( !base, "VirtualAlloc succeeded\n" );
1123     ok( GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError() );
1124
1125     /* initial protect doesn't matter */
1126
1127     base = VirtualAlloc( 0, size, MEM_RESERVE | MEM_WRITE_WATCH, PAGE_NOACCESS );
1128     ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
1129     base = VirtualAlloc( base, size, MEM_COMMIT, PAGE_NOACCESS );
1130     ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
1131
1132     count = 64;
1133     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1134     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1135     ok( count == 0, "wrong count %lu\n", count );
1136
1137     ret = VirtualProtect( base, 6*pagesize, PAGE_READWRITE, &old_prot );
1138     ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
1139     ok( old_prot == PAGE_NOACCESS, "wrong old prot %x\n", old_prot );
1140
1141     base[5*pagesize + 200] = 3;
1142
1143     ret = VirtualProtect( base, 6*pagesize, PAGE_NOACCESS, &old_prot );
1144     ok( ret, "VirtualProtect failed error %u\n", GetLastError() );
1145     ok( old_prot == PAGE_READWRITE, "wrong old prot %x\n", old_prot );
1146
1147     count = 64;
1148     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1149     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1150     ok( count == 1, "wrong count %lu\n", count );
1151     ok( results[0] == base + 5*pagesize, "wrong result %p\n", results[0] );
1152
1153     ret = VirtualFree( base, size, MEM_DECOMMIT );
1154     ok( ret, "VirtualFree failed %u\n", GetLastError() );
1155
1156     count = 64;
1157     ret = pGetWriteWatch( 0, base, size, results, &count, &pagesize );
1158     ok( !ret, "GetWriteWatch failed %u\n", GetLastError() );
1159     ok( count == 1 || broken(count == 0), /* win98 */
1160         "wrong count %lu\n", count );
1161     if (count) ok( results[0] == base + 5*pagesize, "wrong result %p\n", results[0] );
1162
1163     VirtualFree( base, 0, MEM_FREE );
1164 }
1165
1166 START_TEST(virtual)
1167 {
1168     int argc;
1169     char **argv;
1170     argc = winetest_get_mainargs( &argv );
1171
1172     if (argc >= 3)
1173     {
1174         if (!strcmp(argv[2], "sleep"))
1175         {
1176             Sleep(5000); /* spawned process runs for at most 5 seconds */
1177             return;
1178         }
1179         while (1)
1180         {
1181             void *mem;
1182             BOOL ret;
1183             mem = VirtualAlloc(NULL, 1<<20, MEM_COMMIT|MEM_RESERVE,
1184                                PAGE_EXECUTE_READWRITE);
1185             ok(mem != NULL, "VirtualAlloc failed %u\n", GetLastError());
1186             if (mem == NULL) break;
1187             ret = VirtualFree(mem, 0, MEM_RELEASE);
1188             ok(ret, "VirtualFree failed %u\n", GetLastError());
1189             if (!ret) break;
1190         }
1191         return;
1192     }
1193
1194     hkernel32 = GetModuleHandleA("kernel32.dll");
1195     pVirtualAllocEx = (void *) GetProcAddress(hkernel32, "VirtualAllocEx");
1196     pVirtualFreeEx = (void *) GetProcAddress(hkernel32, "VirtualFreeEx");
1197     pGetWriteWatch = (void *) GetProcAddress(hkernel32, "GetWriteWatch");
1198     pResetWriteWatch = (void *) GetProcAddress(hkernel32, "ResetWriteWatch");
1199
1200     test_VirtualAllocEx();
1201     test_VirtualAlloc();
1202     test_MapViewOfFile();
1203     test_NtMapViewOfSection();
1204     test_CreateFileMapping();
1205     test_BadPtr();
1206     test_write_watch();
1207 }