Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198/* * PROJECT: ReactOS api tests * LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later) * PURPOSE: Test for CAtlFileMapping * COPYRIGHT: Copyright 2019 Mark Jansen (mark.jansen@reactos.org) */
#include <atlfile.h>
#ifdef HAVE_APITEST #include <apitest.h>#else #include "atltest.h"#endif
struct TestData{ int data[4];};
static void test_SharedMem(){ CAtlFileMapping<TestData> test1, test2; CAtlFileMappingBase test3; BOOL bAlreadyExisted; HRESULT hr;
ok(test1.GetData() == NULL, "Expected NULL, got %p\n", test1.GetData()); ok(test3.GetData() == NULL, "Expected NULL, got %p\n", test3.GetData()); ok(test1.GetHandle() == NULL, "Expected NULL, got %p\n", test1.GetHandle()); ok(test3.GetHandle() == NULL, "Expected NULL, got %p\n", test3.GetHandle()); ok(test1.GetMappingSize() == 0, "Expected 0, got %lu\n", test1.GetMappingSize()); ok(test3.GetMappingSize() == 0, "Expected 0, got %lu\n", test3.GetMappingSize());
test1 = test1; //test1 = test2; // Asserts on orig.m_pData != NULL, same with CopyFrom //test1 = test3; // does not compile hr = test1.Unmap(); ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr);
// Asserts on name == NULL hr = test1.OpenMapping(_T("TEST_MAPPING"), 123, 0, FILE_MAP_ALL_ACCESS); ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), "Expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%lx\n", hr); ok(test1.GetData() == NULL, "Expected NULL, got %p\n", test1.GetData()); ok(test1.GetHandle() == NULL, "Expected NULL, got %p\n", test1.GetHandle()); ok(test1.GetMappingSize() == 123, "Expected 123, got %lu\n", test1.GetMappingSize());
hr = test1.Unmap(); ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr); ok(test1.GetData() == NULL, "Expected NULL, got %p\n", test1.GetData()); ok(test1.GetHandle() == NULL, "Expected NULL, got %p\n", test1.GetHandle()); ok(test1.GetMappingSize() == 123, "Expected 123, got %lu\n", test1.GetMappingSize());
bAlreadyExisted = 123; hr = test1.MapSharedMem(sizeof(TestData), _T("TEST_MAPPING"), &bAlreadyExisted, (LPSECURITY_ATTRIBUTES)0, PAGE_READWRITE, FILE_MAP_ALL_ACCESS); ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr); ok(test1.GetData() != NULL, "Expected ptr, got %p\n", test1.GetData()); ok(test1.GetHandle() != NULL, "Expected handle, got %p\n", test1.GetHandle()); ok(test1.GetMappingSize() == sizeof(TestData), "Expected sizeof(TestData), got %lu\n", test1.GetMappingSize()); ok(bAlreadyExisted == FALSE, "Expected FALSE, got %u\n", bAlreadyExisted);
if (test1.GetData()) { memset(test1.GetData(), 0x35, sizeof(TestData)); }
hr = test2.CopyFrom(test1); ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr); ok(test2.GetData() != NULL, "Expected ptr, got %p\n", test2.GetData()); ok(test2.GetHandle() != NULL, "Expected handle, got %p\n", test2.GetHandle()); ok(test2.GetMappingSize() == sizeof(TestData), "Expected sizeof(TestData), got %lu\n", test2.GetMappingSize());
// test1 is not closed: ok(test1.GetData() != NULL, "Expected ptr, got %p\n", test1.GetData()); ok(test1.GetHandle() != NULL, "Expected handle, got %p\n", test1.GetHandle()); ok(test1.GetMappingSize() == sizeof(TestData), "Expected sizeof(TestData), got %lu\n", test1.GetMappingSize());
// test2 does not equal test1 ok(test1.GetData() != test2.GetData(), "Expected different ptrs\n"); ok(test1.GetHandle() != test2.GetHandle(), "Expected different handles\n");
TestData* t1 = test1; TestData* t2 = test2; if (t1 && t2) { ok(t1->data[0] == 0x35353535, "Expected 0x35353535, got 0x%x\n", t1->data[0]); ok(t2->data[0] == 0x35353535, "Expected 0x35353535, got 0x%x\n", t2->data[0]);
t1->data[0] = 0xbeefbeef; ok(t1->data[0] == (int)0xbeefbeef, "Expected 0xbeefbeef, got 0x%x\n", t1->data[0]); ok(t2->data[0] == (int)0xbeefbeef, "Expected 0xbeefbeef, got 0x%x\n", t2->data[0]); }
hr = test3.OpenMapping(_T("TEST_MAPPING"), sizeof(TestData), offsetof(TestData, data[1])); ok(hr == HRESULT_FROM_WIN32(ERROR_MAPPED_ALIGNMENT), "Expected HRESULT_FROM_WIN32(ERROR_MAPPED_ALIGNMENT), got 0x%lx\n", hr); ok(test3.GetData() == NULL, "Expected NULL, got %p\n", test3.GetData()); ok(test3.GetHandle() == NULL, "Expected NULL, got %p\n", test3.GetHandle()); ok(test3.GetMappingSize() == sizeof(TestData), "Expected sizeof(TestData), got %lu\n", test3.GetMappingSize());
hr = test2.Unmap(); ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr); ok(test2.GetData() == NULL, "Expected NULL, got %p\n", test2.GetData()); ok(test2.GetHandle() == NULL, "Expected NULL, got %p\n", test2.GetHandle()); ok(test2.GetMappingSize() == sizeof(TestData), "Expected sizeof(TestData), got %lu\n", test2.GetMappingSize());
bAlreadyExisted = 123; // Wrong access flag hr = test2.MapSharedMem(sizeof(TestData), _T("TEST_MAPPING"), &bAlreadyExisted, (LPSECURITY_ATTRIBUTES)0, PAGE_EXECUTE_READ, FILE_MAP_ALL_ACCESS); ok(hr == E_ACCESSDENIED, "Expected E_ACCESSDENIED, got 0x%lx\n", hr); ok(test2.GetData() == NULL, "Expected NULL, got %p\n", test2.GetData()); ok(test2.GetHandle() == NULL, "Expected NULL, got %p\n", test2.GetHandle()); ok(test2.GetMappingSize() == sizeof(TestData), "Expected sizeof(TestData), got %lu\n", test2.GetMappingSize()); ok(bAlreadyExisted == TRUE, "Expected TRUE, got %u\n", bAlreadyExisted);
bAlreadyExisted = 123; hr = test2.MapSharedMem(sizeof(TestData), _T("TEST_MAPPING"), &bAlreadyExisted, (LPSECURITY_ATTRIBUTES)0, PAGE_READWRITE, FILE_MAP_ALL_ACCESS); ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr); ok(test2.GetData() != NULL, "Expected ptr, got %p\n", test2.GetData()); ok(test2.GetHandle() != NULL, "Expected handle, got %p\n", test2.GetHandle()); ok(test2.GetMappingSize() == sizeof(TestData), "Expected sizeof(TestData), got %lu\n", test2.GetMappingSize()); ok(bAlreadyExisted == TRUE, "Expected TRUE, got %u\n", bAlreadyExisted);
// test2 does not equal test1 ok(test1.GetData() != test2.GetData(), "Expected different ptrs\n"); ok(test1.GetHandle() != test2.GetHandle(), "Expected different handles\n");
t2 = test2; if (t1 && t2) { ok(t1->data[0] == (int)0xbeefbeef, "Expected 0xbeefbeef, got 0x%x\n", t1->data[0]); ok(t2->data[0] == (int)0xbeefbeef, "Expected 0xbeefbeef, got 0x%x\n", t2->data[0]);
t1->data[0] = 0xdeaddead; ok(t1->data[0] == (int)0xdeaddead, "Expected 0xdeaddead, got 0x%x\n", t1->data[0]); ok(t2->data[0] == (int)0xdeaddead, "Expected 0xdeaddead, got 0x%x\n", t2->data[0]); }}
static void test_FileMapping(){ WCHAR Buf[MAX_PATH] = {0}; ULARGE_INTEGER FileSize;
GetModuleFileNameW(NULL, Buf, _countof(Buf));
CAtlFileMapping<IMAGE_DOS_HEADER> test1, test2; HRESULT hr;
{ CHandle hFile(CreateFileW(Buf, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL)); ok(hFile != INVALID_HANDLE_VALUE, "Could not open %S, aborting test\n", Buf); if (hFile == INVALID_HANDLE_VALUE) return;
FileSize.LowPart = GetFileSize(hFile, &FileSize.HighPart);
hr = test1.MapFile(hFile, 0, 0, PAGE_READONLY, FILE_MAP_READ); ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr); ok(test1.GetData() != NULL, "Expected ptr, got %p\n", test1.GetData()); ok(test1.GetHandle() != NULL, "Expected handle, got %p\n", test1.GetHandle()); ok(test1.GetMappingSize() == FileSize.LowPart, "Expected %lu, got %lu\n", FileSize.LowPart, test1.GetMappingSize());
hr = test2.MapFile(hFile, 0, 0x10000, PAGE_READONLY, FILE_MAP_READ); ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr); ok(test2.GetData() != NULL, "Expected ptr, got %p\n", test2.GetData()); ok(test2.GetHandle() != NULL, "Expected handle, got %p\n", test2.GetHandle()); // Offset is subtracted ok(test2.GetMappingSize() == FileSize.LowPart - 0x10000, "Expected %lu, got %lu\n", FileSize.LowPart-0x10000, test2.GetMappingSize());
hr = test1.Unmap(); ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr); ok(test1.GetData() == NULL, "Expected NULL, got %p\n", test1.GetData()); ok(test1.GetHandle() == NULL, "Expected NULL, got %p\n", test1.GetHandle()); ok(test1.GetMappingSize() == FileSize.LowPart, "Expected %lu, got %lu\n", FileSize.LowPart, test1.GetMappingSize());
hr = test1.MapFile(hFile, 0x1000); ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr); ok(test1.GetData() != NULL, "Expected ptr, got %p\n", test1.GetData()); ok(test1.GetHandle() != NULL, "Expected handle, got %p\n", test1.GetHandle()); ok(test1.GetMappingSize() == 0x1000, "Expected 0x1000, got %lu\n", test1.GetMappingSize()); }
// We can still access it after the file is closed PIMAGE_DOS_HEADER dos = test1; if (dos) { ok(dos->e_magic == IMAGE_DOS_SIGNATURE, "Expected IMAGE_DOS_SIGNATURE, got 0x%x\n", dos->e_magic); }}
START_TEST(CAtlFileMapping){ test_SharedMem(); test_FileMapping();}