mirror of
https://github.com/reactos/reactos.git
synced 2024-11-02 12:53:33 +00:00
a800fa7bb2
- Add "atltest.h" header file for common use in atl_apitest. - Delete the duplicated and simplify.
198 lines
9 KiB
C++
198 lines
9 KiB
C++
/*
|
|
* 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();
|
|
}
|