| /* | |
| * Unit test suite for Virtual* family of APIs. | |
| * | |
| * Copyright 2004 Dmitry Timoshkov | |
| * | |
| * This library is free software; you can redistribute it and/or | |
| * modify it under the terms of the GNU Lesser General Public | |
| * License as published by the Free Software Foundation; either | |
| * version 2.1 of the License, or (at your option) any later version. | |
| * | |
| * This library is distributed in the hope that it will be useful, | |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| * Lesser General Public License for more details. | |
| * | |
| * You should have received a copy of the GNU Lesser General Public | |
| * License along with this library; if not, write to the Free Software | |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA | |
| */ | |
| #include <stdarg.h> | |
| #include <stdio.h> | |
| #include "windef.h" | |
| #include "winbase.h" | |
| #include "winerror.h" | |
| #include "wine/test.h" | |
| #define NUM_THREADS 4 | |
| static HINSTANCE hkernel32; | |
| static LPVOID (WINAPI *pVirtualAllocEx)(HANDLE, LPVOID, SIZE_T, DWORD, DWORD); | |
| static BOOL (WINAPI *pVirtualFreeEx)(HANDLE, LPVOID, SIZE_T, DWORD); | |
| /* ############################### */ | |
| static HANDLE create_target_process(const char *arg) | |
| { | |
| char **argv; | |
| char cmdline[MAX_PATH]; | |
| PROCESS_INFORMATION pi; | |
| STARTUPINFO si = { 0 }; | |
| si.cb = sizeof(si); | |
| winetest_get_mainargs( &argv ); | |
| sprintf(cmdline, "%s %s %s", argv[0], argv[1], arg); | |
| ok(CreateProcess(NULL, cmdline, NULL, NULL, FALSE, 0, NULL, NULL, | |
| &si, &pi) != 0, "error: %u\n", GetLastError()); | |
| ok(CloseHandle(pi.hThread) != 0, "error %u\n", GetLastError()); | |
| return pi.hProcess; | |
| } | |
| static void test_VirtualAllocEx(void) | |
| { | |
| const unsigned int alloc_size = 1<<15; | |
| char *src, *dst; | |
| unsigned long bytes_written = 0, bytes_read = 0, i; | |
| void *addr1, *addr2; | |
| BOOL b; | |
| DWORD old_prot; | |
| MEMORY_BASIC_INFORMATION info; | |
| HANDLE hProcess; | |
| /* not exported in all windows-versions */ | |
| if ((!pVirtualAllocEx) || (!pVirtualFreeEx)) { | |
| skip("VirtualAllocEx not found\n"); | |
| return; | |
| } | |
| hProcess = create_target_process("sleep"); | |
| ok(hProcess != NULL, "Can't start process\n"); | |
| SetLastError(0xdeadbeef); | |
| addr1 = pVirtualAllocEx(hProcess, NULL, alloc_size, MEM_COMMIT, | |
| PAGE_EXECUTE_READWRITE); | |
| if (!addr1 && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) | |
| { /* Win9x */ | |
| skip("VirtualAllocEx not implemented\n"); | |
| TerminateProcess(hProcess, 0); | |
| CloseHandle(hProcess); | |
| return; | |
| } | |
| src = HeapAlloc( GetProcessHeap(), 0, alloc_size ); | |
| dst = HeapAlloc( GetProcessHeap(), 0, alloc_size ); | |
| for (i = 0; i < alloc_size; i++) | |
| src[i] = i & 0xff; | |
| ok(addr1 != NULL, "VirtualAllocEx error %u\n", GetLastError()); | |
| b = WriteProcessMemory(hProcess, addr1, src, alloc_size, &bytes_written); | |
| ok(b && (bytes_written == alloc_size), "%lu bytes written\n", | |
| bytes_written); | |
| b = ReadProcessMemory(hProcess, addr1, dst, alloc_size, &bytes_read); | |
| ok(b && (bytes_read == alloc_size), "%lu bytes read\n", bytes_read); | |
| ok(!memcmp(src, dst, alloc_size), "Data from remote process differs\n"); | |
| b = pVirtualFreeEx(hProcess, addr1, 0, MEM_RELEASE); | |
| ok(b != 0, "VirtualFreeEx, error %u\n", GetLastError()); | |
| HeapFree( GetProcessHeap(), 0, src ); | |
| HeapFree( GetProcessHeap(), 0, dst ); | |
| /* | |
| * The following tests parallel those in test_VirtualAlloc() | |
| */ | |
| SetLastError(0xdeadbeef); | |
| addr1 = pVirtualAllocEx(hProcess, 0, 0, MEM_RESERVE, PAGE_NOACCESS); | |
| ok(addr1 == NULL, "VirtualAllocEx should fail on zero-sized allocation\n"); | |
| ok(GetLastError() == ERROR_INVALID_PARAMETER /* NT */ || | |
| GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Win9x */ | |
| "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError()); | |
| addr1 = pVirtualAllocEx(hProcess, 0, 0xFFFC, MEM_RESERVE, PAGE_NOACCESS); | |
| ok(addr1 != NULL, "VirtualAllocEx failed\n"); | |
| /* test a not committed memory */ | |
| memset(&info, 'q', sizeof(info)); | |
| ok(VirtualQueryEx(hProcess, addr1, &info, sizeof(info)) == sizeof(info), "VirtualQueryEx failed\n"); | |
| ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1); | |
| ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1); | |
| ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect); | |
| ok(info.RegionSize == 0x10000, "%lx != 0x10000\n", info.RegionSize); | |
| ok(info.State == MEM_RESERVE, "%x != MEM_RESERVE\n", info.State); | |
| /* NT reports Protect == 0 for a not committed memory block */ | |
| ok(info.Protect == 0 /* NT */ || | |
| info.Protect == PAGE_NOACCESS, /* Win9x */ | |
| "%x != PAGE_NOACCESS\n", info.Protect); | |
| ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type); | |
| SetLastError(0xdeadbeef); | |
| ok(!VirtualProtectEx(hProcess, addr1, 0xFFFC, PAGE_READONLY, &old_prot), | |
| "VirtualProtectEx should fail on a not committed memory\n"); | |
| ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ || | |
| GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */ | |
| "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError()); | |
| addr2 = pVirtualAllocEx(hProcess, addr1, 0x1000, MEM_COMMIT, PAGE_NOACCESS); | |
| ok(addr1 == addr2, "VirtualAllocEx failed\n"); | |
| /* test a committed memory */ | |
| ok(VirtualQueryEx(hProcess, addr1, &info, sizeof(info)) == sizeof(info), | |
| "VirtualQueryEx failed\n"); | |
| ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1); | |
| ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1); | |
| ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect); | |
| ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize); | |
| ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State); | |
| /* this time NT reports PAGE_NOACCESS as well */ | |
| ok(info.Protect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.Protect); | |
| ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type); | |
| /* this should fail, since not the whole range is committed yet */ | |
| SetLastError(0xdeadbeef); | |
| ok(!VirtualProtectEx(hProcess, addr1, 0xFFFC, PAGE_READONLY, &old_prot), | |
| "VirtualProtectEx should fail on a not committed memory\n"); | |
| ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ || | |
| GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */ | |
| "got %u, expected ERROR_INVALID_ADDRESS\n", GetLastError()); | |
| old_prot = 0; | |
| ok(VirtualProtectEx(hProcess, addr1, 0x1000, PAGE_READONLY, &old_prot), "VirtualProtectEx failed\n"); | |
| ok(old_prot == PAGE_NOACCESS, "wrong old protection: got %04x instead of PAGE_NOACCESS\n", old_prot); | |
| old_prot = 0; | |
| ok(VirtualProtectEx(hProcess, addr1, 0x1000, PAGE_READWRITE, &old_prot), "VirtualProtectEx failed\n"); | |
| ok(old_prot == PAGE_READONLY, "wrong old protection: got %04x instead of PAGE_READONLY\n", old_prot); | |
| ok(!pVirtualFreeEx(hProcess, addr1, 0x10000, 0), | |
| "VirtualFreeEx should fail with type 0\n"); | |
| ok(GetLastError() == ERROR_INVALID_PARAMETER, | |
| "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError()); | |
| ok(pVirtualFreeEx(hProcess, addr1, 0x10000, MEM_DECOMMIT), "VirtualFreeEx failed\n"); | |
| /* if the type is MEM_RELEASE, size must be 0 */ | |
| ok(!pVirtualFreeEx(hProcess, addr1, 1, MEM_RELEASE), | |
| "VirtualFreeEx should fail\n"); | |
| ok(GetLastError() == ERROR_INVALID_PARAMETER, | |
| "got %u, expected ERROR_INVALID_PARAMETER\n", GetLastError()); | |
| ok(pVirtualFreeEx(hProcess, addr1, 0, MEM_RELEASE), "VirtualFreeEx failed\n"); | |
| TerminateProcess(hProcess, 0); | |
| CloseHandle(hProcess); | |
| } | |
| static void test_VirtualAlloc(void) | |
| { | |
| void *addr1, *addr2; | |
| DWORD old_prot; | |
| MEMORY_BASIC_INFORMATION info; | |
| SetLastError(0xdeadbeef); | |
| addr1 = VirtualAlloc(0, 0, MEM_RESERVE, PAGE_NOACCESS); | |
| ok(addr1 == NULL, "VirtualAlloc should fail on zero-sized allocation\n"); | |
| ok(GetLastError() == ERROR_INVALID_PARAMETER /* NT */ || | |
| GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Win9x */ | |
| "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError()); | |
| addr1 = VirtualAlloc(0, 0xFFFC, MEM_RESERVE, PAGE_NOACCESS); | |
| ok(addr1 != NULL, "VirtualAlloc failed\n"); | |
| /* test a not committed memory */ | |
| ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info), | |
| "VirtualQuery failed\n"); | |
| ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1); | |
| ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1); | |
| ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect); | |
| ok(info.RegionSize == 0x10000, "%lx != 0x10000\n", info.RegionSize); | |
| ok(info.State == MEM_RESERVE, "%x != MEM_RESERVE\n", info.State); | |
| /* NT reports Protect == 0 for a not committed memory block */ | |
| ok(info.Protect == 0 /* NT */ || | |
| info.Protect == PAGE_NOACCESS, /* Win9x */ | |
| "%x != PAGE_NOACCESS\n", info.Protect); | |
| ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type); | |
| SetLastError(0xdeadbeef); | |
| ok(!VirtualProtect(addr1, 0xFFFC, PAGE_READONLY, &old_prot), | |
| "VirtualProtect should fail on a not committed memory\n"); | |
| ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ || | |
| GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */ | |
| "got %d, expected ERROR_INVALID_ADDRESS\n", GetLastError()); | |
| addr2 = VirtualAlloc(addr1, 0x1000, MEM_COMMIT, PAGE_NOACCESS); | |
| ok(addr1 == addr2, "VirtualAlloc failed\n"); | |
| /* test a committed memory */ | |
| ok(VirtualQuery(addr1, &info, sizeof(info)) == sizeof(info), | |
| "VirtualQuery failed\n"); | |
| ok(info.BaseAddress == addr1, "%p != %p\n", info.BaseAddress, addr1); | |
| ok(info.AllocationBase == addr1, "%p != %p\n", info.AllocationBase, addr1); | |
| ok(info.AllocationProtect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.AllocationProtect); | |
| ok(info.RegionSize == 0x1000, "%lx != 0x1000\n", info.RegionSize); | |
| ok(info.State == MEM_COMMIT, "%x != MEM_COMMIT\n", info.State); | |
| /* this time NT reports PAGE_NOACCESS as well */ | |
| ok(info.Protect == PAGE_NOACCESS, "%x != PAGE_NOACCESS\n", info.Protect); | |
| ok(info.Type == MEM_PRIVATE, "%x != MEM_PRIVATE\n", info.Type); | |
| /* this should fail, since not the whole range is committed yet */ | |
| SetLastError(0xdeadbeef); | |
| ok(!VirtualProtect(addr1, 0xFFFC, PAGE_READONLY, &old_prot), | |
| "VirtualProtect should fail on a not committed memory\n"); | |
| ok(GetLastError() == ERROR_INVALID_ADDRESS /* NT */ || | |
| GetLastError() == ERROR_INVALID_PARAMETER, /* Win9x */ | |
| "got %d, expected ERROR_INVALID_ADDRESS\n", GetLastError()); | |
| ok(VirtualProtect(addr1, 0x1000, PAGE_READONLY, &old_prot), "VirtualProtect failed\n"); | |
| ok(old_prot == PAGE_NOACCESS, | |
| "wrong old protection: got %04x instead of PAGE_NOACCESS\n", old_prot); | |
| ok(VirtualProtect(addr1, 0x1000, PAGE_READWRITE, &old_prot), "VirtualProtect failed\n"); | |
| ok(old_prot == PAGE_READONLY, | |
| "wrong old protection: got %04x instead of PAGE_READONLY\n", old_prot); | |
| ok(!VirtualFree(addr1, 0x10000, 0), "VirtualFree should fail with type 0\n"); | |
| ok(GetLastError() == ERROR_INVALID_PARAMETER, | |
| "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError()); | |
| ok(VirtualFree(addr1, 0x10000, MEM_DECOMMIT), "VirtualFree failed\n"); | |
| /* if the type is MEM_RELEASE, size must be 0 */ | |
| ok(!VirtualFree(addr1, 1, MEM_RELEASE), "VirtualFree should fail\n"); | |
| ok(GetLastError() == ERROR_INVALID_PARAMETER, | |
| "got %d, expected ERROR_INVALID_PARAMETER\n", GetLastError()); | |
| ok(VirtualFree(addr1, 0, MEM_RELEASE), "VirtualFree failed\n"); | |
| } | |
| static void test_MapViewOfFile(void) | |
| { | |
| static const char testfile[] = "testfile.xxx"; | |
| HANDLE file, mapping; | |
| void *ptr; | |
| file = CreateFileA( testfile, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0 ); | |
| ok( file != INVALID_HANDLE_VALUE, "Failed to create test file\n" ); | |
| SetFilePointer( file, 4096, NULL, FILE_BEGIN ); | |
| SetEndOfFile( file ); | |
| /* read/write mapping */ | |
| mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL ); | |
| ok( mapping != 0, "CreateFileMapping failed\n" ); | |
| ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 ); | |
| ok( ptr != NULL, "MapViewOfFile FILE_MAPE_READ failed\n" ); | |
| UnmapViewOfFile( ptr ); | |
| /* this fails on win9x but succeeds on NT */ | |
| ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 ); | |
| if (ptr) UnmapViewOfFile( ptr ); | |
| else ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() ); | |
| ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 ); | |
| ok( ptr != NULL, "MapViewOfFile 0 failed\n" ); | |
| UnmapViewOfFile( ptr ); | |
| ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 ); | |
| ok( ptr != NULL, "MapViewOfFile FILE_MAP_WRITE failed\n" ); | |
| UnmapViewOfFile( ptr ); | |
| CloseHandle( mapping ); | |
| /* read-only mapping */ | |
| mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL ); | |
| ok( mapping != 0, "CreateFileMapping failed\n" ); | |
| ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 ); | |
| ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ failed\n" ); | |
| UnmapViewOfFile( ptr ); | |
| /* this fails on win9x but succeeds on NT */ | |
| ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 ); | |
| if (ptr) UnmapViewOfFile( ptr ); | |
| else ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() ); | |
| ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 ); | |
| ok( ptr != NULL, "MapViewOfFile 0 failed\n" ); | |
| UnmapViewOfFile( ptr ); | |
| ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 ); | |
| ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" ); | |
| ok( GetLastError() == ERROR_INVALID_PARAMETER || | |
| GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() ); | |
| CloseHandle( mapping ); | |
| /* copy-on-write mapping */ | |
| mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL ); | |
| ok( mapping != 0, "CreateFileMapping failed\n" ); | |
| ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 ); | |
| ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ failed\n" ); | |
| UnmapViewOfFile( ptr ); | |
| ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 ); | |
| ok( ptr != NULL, "MapViewOfFile FILE_MAP_COPY failed\n" ); | |
| UnmapViewOfFile( ptr ); | |
| ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 ); | |
| ok( ptr != NULL, "MapViewOfFile 0 failed\n" ); | |
| UnmapViewOfFile( ptr ); | |
| ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 ); | |
| ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" ); | |
| ok( GetLastError() == ERROR_INVALID_PARAMETER || | |
| GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() ); | |
| CloseHandle( mapping ); | |
| /* no access mapping */ | |
| mapping = CreateFileMappingA( file, NULL, PAGE_NOACCESS, 0, 4096, NULL ); | |
| /* fails on NT but succeeds on win9x */ | |
| if (!mapping) ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() ); | |
| else | |
| { | |
| ptr = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 4096 ); | |
| ok( ptr != NULL, "MapViewOfFile FILE_MAP_READ failed\n" ); | |
| UnmapViewOfFile( ptr ); | |
| ptr = MapViewOfFile( mapping, FILE_MAP_COPY, 0, 0, 4096 ); | |
| ok( !ptr, "MapViewOfFile FILE_MAP_COPY succeeded\n" ); | |
| ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() ); | |
| ptr = MapViewOfFile( mapping, 0, 0, 0, 4096 ); | |
| ok( ptr != NULL, "MapViewOfFile 0 failed\n" ); | |
| UnmapViewOfFile( ptr ); | |
| ptr = MapViewOfFile( mapping, FILE_MAP_WRITE, 0, 0, 4096 ); | |
| ok( !ptr, "MapViewOfFile FILE_MAP_WRITE succeeded\n" ); | |
| ok( GetLastError() == ERROR_INVALID_PARAMETER, "Wrong error %d\n", GetLastError() ); | |
| CloseHandle( mapping ); | |
| } | |
| CloseHandle( file ); | |
| /* now try read-only file */ | |
| file = CreateFileA( testfile, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0 ); | |
| ok( file != INVALID_HANDLE_VALUE, "Failed to create test file\n" ); | |
| mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL ); | |
| ok( !mapping, "CreateFileMapping PAGE_READWRITE succeeded\n" ); | |
| ok( GetLastError() == ERROR_INVALID_PARAMETER || | |
| GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() ); | |
| mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL ); | |
| ok( mapping != 0, "CreateFileMapping PAGE_WRITECOPY failed\n" ); | |
| CloseHandle( mapping ); | |
| mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL ); | |
| ok( mapping != 0, "CreateFileMapping PAGE_READONLY failed\n" ); | |
| CloseHandle( mapping ); | |
| CloseHandle( file ); | |
| /* now try no access file */ | |
| file = CreateFileA( testfile, 0, 0, NULL, OPEN_EXISTING, 0, 0 ); | |
| ok( file != INVALID_HANDLE_VALUE, "Failed to create test file\n" ); | |
| mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL ); | |
| ok( !mapping, "CreateFileMapping PAGE_READWRITE succeeded\n" ); | |
| ok( GetLastError() == ERROR_INVALID_PARAMETER || | |
| GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() ); | |
| mapping = CreateFileMappingA( file, NULL, PAGE_WRITECOPY, 0, 4096, NULL ); | |
| ok( !mapping, "CreateFileMapping PAGE_WRITECOPY succeeded\n" ); | |
| ok( GetLastError() == ERROR_INVALID_PARAMETER || | |
| GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() ); | |
| mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 4096, NULL ); | |
| ok( !mapping, "CreateFileMapping PAGE_READONLY succeeded\n" ); | |
| ok( GetLastError() == ERROR_INVALID_PARAMETER || | |
| GetLastError() == ERROR_ACCESS_DENIED, "Wrong error %d\n", GetLastError() ); | |
| CloseHandle( file ); | |
| DeleteFileA( testfile ); | |
| } | |
| static DWORD (WINAPI *pNtMapViewOfSection)( HANDLE handle, HANDLE process, PVOID *addr_ptr, | |
| ULONG zero_bits, SIZE_T commit_size, | |
| const LARGE_INTEGER *offset_ptr, SIZE_T *size_ptr, | |
| ULONG inherit, ULONG alloc_type, ULONG protect ); | |
| static DWORD (WINAPI *pNtUnmapViewOfSection)( HANDLE process, PVOID addr ); | |
| static void test_NtMapViewOfSection(void) | |
| { | |
| HANDLE hProcess; | |
| static const char testfile[] = "testfile.xxx"; | |
| static const char data[] = "test data for NtMapViewOfSection"; | |
| char buffer[sizeof(data)]; | |
| HANDLE file, mapping; | |
| void *ptr; | |
| BOOL ret; | |
| DWORD status, written; | |
| SIZE_T size, result; | |
| LARGE_INTEGER offset; | |
| pNtMapViewOfSection = (void *)GetProcAddress( GetModuleHandle("ntdll.dll"), "NtMapViewOfSection" ); | |
| pNtUnmapViewOfSection = (void *)GetProcAddress( GetModuleHandle("ntdll.dll"), "NtUnmapViewOfSection" ); | |
| if (!pNtMapViewOfSection || !pNtUnmapViewOfSection) | |
| { | |
| skip( "NtMapViewOfSection not found\n" ); | |
| return; | |
| } | |
| file = CreateFileA( testfile, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0 ); | |
| ok( file != INVALID_HANDLE_VALUE, "Failed to create test file\n" ); | |
| WriteFile( file, data, sizeof(data), &written, NULL ); | |
| SetFilePointer( file, 4096, NULL, FILE_BEGIN ); | |
| SetEndOfFile( file ); | |
| /* read/write mapping */ | |
| mapping = CreateFileMappingA( file, NULL, PAGE_READWRITE, 0, 4096, NULL ); | |
| ok( mapping != 0, "CreateFileMapping failed\n" ); | |
| hProcess = create_target_process("sleep"); | |
| ok(hProcess != NULL, "Can't start process\n"); | |
| ptr = NULL; | |
| size = 0; | |
| offset.QuadPart = 0; | |
| status = pNtMapViewOfSection( mapping, hProcess, &ptr, 0, 0, &offset, &size, 1, 0, PAGE_READWRITE ); | |
| ok( !status, "NtMapViewOfSection failed status %x\n", status ); | |
| ret = ReadProcessMemory( hProcess, ptr, buffer, sizeof(buffer), &result ); | |
| ok( ret, "ReadProcessMemory failed\n" ); | |
| ok( result == sizeof(buffer), "ReadProcessMemory didn't read all data (%lx)\n", result ); | |
| ok( !memcmp( buffer, data, sizeof(buffer) ), "Wrong data read\n" ); | |
| status = pNtUnmapViewOfSection( hProcess, ptr ); | |
| ok( !status, "NtUnmapViewOfSection failed status %x\n", status ); | |
| CloseHandle( mapping ); | |
| CloseHandle( file ); | |
| DeleteFileA( testfile ); | |
| } | |
| START_TEST(virtual) | |
| { | |
| int argc; | |
| char **argv; | |
| argc = winetest_get_mainargs( &argv ); | |
| if (argc >= 3) | |
| { | |
| if (!strcmp(argv[2], "sleep")) | |
| { | |
| Sleep(5000); /* spawned process runs for at most 5 seconds */ | |
| return; | |
| } | |
| while (1) | |
| { | |
| void *mem; | |
| BOOL ret; | |
| mem = VirtualAlloc(NULL, 1<<20, MEM_COMMIT|MEM_RESERVE, | |
| PAGE_EXECUTE_READWRITE); | |
| ok(mem != NULL, "VirtualAlloc failed %u\n", GetLastError()); | |
| if (mem == NULL) break; | |
| ret = VirtualFree(mem, 0, MEM_RELEASE); | |
| ok(ret, "VirtualFree failed %u\n", GetLastError()); | |
| if (!ret) break; | |
| } | |
| return; | |
| } | |
| hkernel32 = GetModuleHandleA("kernel32.dll"); | |
| pVirtualAllocEx = (void *) GetProcAddress(hkernel32, "VirtualAllocEx"); | |
| pVirtualFreeEx = (void *) GetProcAddress(hkernel32, "VirtualFreeEx"); | |
| test_VirtualAllocEx(); | |
| test_VirtualAlloc(); | |
| test_MapViewOfFile(); | |
| test_NtMapViewOfSection(); | |
| } |