From 3440678be4651c0b41699a9121951efbfb56ef9d Mon Sep 17 00:00:00 2001 From: Thomas Faber Date: Sat, 19 Oct 2013 17:59:29 +0000 Subject: [PATCH] [NTDLL_APITEST] - Add test for Rtl MemoryStream functions. Patch by David Quintana. CORE-7492 svn path=/trunk/; revision=60707 --- rostests/apitests/ntdll/CMakeLists.txt | 3 +- rostests/apitests/ntdll/RtlMemoryStream.c | 525 ++++++++++++++++++++++ rostests/apitests/ntdll/testlist.c | 2 + 3 files changed, 529 insertions(+), 1 deletion(-) create mode 100644 rostests/apitests/ntdll/RtlMemoryStream.c diff --git a/rostests/apitests/ntdll/CMakeLists.txt b/rostests/apitests/ntdll/CMakeLists.txt index 4fb7da1ee99..9404d89918f 100644 --- a/rostests/apitests/ntdll/CMakeLists.txt +++ b/rostests/apitests/ntdll/CMakeLists.txt @@ -21,6 +21,7 @@ list(APPEND SOURCE RtlGetLengthWithoutTrailingPathSeperators.c RtlGetLongestNtPathLength.c RtlInitializeBitMap.c + RtlMemoryStream.c SystemInfo.c Timer.c testlist.c) @@ -30,7 +31,7 @@ if(ARCH STREQUAL "i386") endif() add_executable(ntdll_apitest ${SOURCE} ${ntdll_apitest_asm}) -target_link_libraries(ntdll_apitest wine ${PSEH_LIB}) +target_link_libraries(ntdll_apitest wine uuid ${PSEH_LIB}) set_module_type(ntdll_apitest win32cui) add_importlibs(ntdll_apitest msvcrt advapi32 kernel32 ntdll) diff --git a/rostests/apitests/ntdll/RtlMemoryStream.c b/rostests/apitests/ntdll/RtlMemoryStream.c new file mode 100644 index 00000000000..e2ae475d74c --- /dev/null +++ b/rostests/apitests/ntdll/RtlMemoryStream.c @@ -0,0 +1,525 @@ +/* + * PROJECT: ReactOS api tests + * LICENSE: GPLv2+ - See COPYING in the top level directory + * PURPOSE: Test for the Rtl*MemoryStream* series of functions + * PROGRAMMER: David Quintana + */ + +#define COBJMACROS + +#include +#include +#include +#include + +ULONG finalReleaseCallCount = 0; + +VOID +NTAPI +CustomFinalReleaseMemoryStream(PRTL_MEMORY_STREAM stream) +{ + finalReleaseCallCount++; + trace("FinalRelease CALLED.\n"); +} + +VOID +NTAPI +CustomFinalReleaseOutOfProcessMemoryStream(PRTL_MEMORY_STREAM stream) +{ + finalReleaseCallCount++; + trace("FinalRelease CALLED.\n"); + RtlFinalReleaseOutOfProcessMemoryStream(stream); +} + +BOOL CompareStructsAndSaveForLater(PRTL_MEMORY_STREAM pold, PRTL_MEMORY_STREAM pnew, PSTR at) +{ + BOOL equal = TRUE; + + // Compare + if (pold->Vtbl != pnew->Vtbl) { if (equal) { trace("%s: \n", at); equal = FALSE; } trace("Vtbl changed from %p to %p\n", pold->Vtbl, pnew->Vtbl);} + if (pold->RefCount != pnew->RefCount) { if (equal) { trace("%s: \n", at); equal = FALSE; } trace("RefCount changed from %p to %p\n", pold->RefCount, pnew->RefCount); } + if (pold->Unk1 != pnew->Unk1) { if (equal) { trace("%s: \n", at); equal = FALSE; } trace("Unk1 changed from %p to %p\n", pold->Unk1, pnew->Unk1); } + if (pold->Current != pnew->Current) { if (equal) { trace("%s: \n", at); equal = FALSE; } trace("Current changed from %p to %p\n", pold->Current, pnew->Current); } + if (pold->Start != pnew->Start) { if (equal) { trace("%s: \n", at); equal = FALSE; } trace("Start changed from %p to %p\n", pold->Start, pnew->Start); } + if (pold->End != pnew->End) { if (equal) { trace("%s: \n", at); equal = FALSE; } trace("End changed from %p to %p\n", pold->End, pnew->End); } + if (pold->FinalRelease != pnew->FinalRelease) { if (equal) { trace("%s: \n", at); equal = FALSE; } trace("FinalRelease changed from %p to %p\n", pold->FinalRelease, pnew->FinalRelease); } + if (pold->ProcessHandle != pnew->ProcessHandle) { if (equal) { trace("%s: \n", at); equal = FALSE; } trace("ProcessHandle changed from %p to %p\n", pold->ProcessHandle, pnew->ProcessHandle); } + + // Save + pold->Vtbl = pnew->Vtbl; + pold->RefCount = pnew->RefCount; + pold->Unk1 = pnew->Unk1; + pold->Current = pnew->Current; + pold->Start = pnew->Start; + pold->End = pnew->End; + pold->FinalRelease = pnew->FinalRelease; + pold->ProcessHandle = pnew->ProcessHandle; + + return equal; +} + +void test_InProcess() +{ + LARGE_INTEGER move; + ULARGE_INTEGER size; + HRESULT res; + ULONG i; + + RTL_MEMORY_STREAM stream; + RTL_MEMORY_STREAM previous; + + IStream * istream; + + UCHAR buffer[80]; + UCHAR buffer2[180]; + ULONG bytesRead; + + STATSTG stat; + + finalReleaseCallCount = 0; + + for (i = 0; i < sizeof(buffer2); i++) + { + buffer2[i] = i % UCHAR_MAX; + } + + memset(&stream, 0x90, sizeof(stream)); + memset(&previous, 0x00, sizeof(previous)); + + StartSeh() + RtlInitMemoryStream(NULL); + EndSeh(STATUS_ACCESS_VIOLATION); + + StartSeh() + RtlInitMemoryStream(&stream); + EndSeh(STATUS_SUCCESS); + + CompareStructsAndSaveForLater(&previous, &stream, "After init"); + + ok(stream.RefCount == 0, "RefCount has a wrong value: %d (expected %d).\n", stream.RefCount, 0); + + stream.Current = buffer2; + stream.Start = buffer2; + stream.End = buffer2 + sizeof(buffer2); + stream.FinalRelease = CustomFinalReleaseMemoryStream; + + CompareStructsAndSaveForLater(&previous, &stream, "After assigning"); + + StartSeh() + res = IStream_QueryInterface((struct IStream*)&stream, NULL, NULL); + ok(res == E_INVALIDARG, "QueryInterface to IStream returned wrong hResult: 0x%08x.\n", res); + ok(stream.RefCount == 0, "RefCount has a wrong value: %d (expected %d).\n", stream.RefCount, 2); + EndSeh(STATUS_ACCESS_VIOLATION); + + StartSeh() + res = IStream_QueryInterface((struct IStream*)&stream, &IID_IStream, NULL); + ok(res == E_INVALIDARG, "QueryInterface to IStream returned wrong hResult: 0x%08x.\n", res); + ok(stream.RefCount == 1, "RefCount has a wrong value: %d (expected %d).\n", stream.RefCount, 2); + EndSeh(STATUS_ACCESS_VIOLATION); + + StartSeh() + res = IStream_QueryInterface((struct IStream*)&stream, NULL, (void**)&istream); + ok(res == E_INVALIDARG, "QueryInterface to IStream returned wrong hResult: 0x%08x.\n", res); + ok(stream.RefCount == 1, "RefCount has a wrong value: %d (expected %d).\n", stream.RefCount, 2); + EndSeh(STATUS_ACCESS_VIOLATION); + + StartSeh() + res = IStream_QueryInterface((struct IStream*)&stream, &IID_IStream, (void**)&istream); + ok(res == S_OK, "QueryInterface to IStream returned wrong hResult: 0x%08x.\n", res); + ok(stream.RefCount == 2, "RefCount has a wrong value: %d (expected %d).\n", stream.RefCount, 2); + EndSeh(STATUS_SUCCESS); + + CompareStructsAndSaveForLater(&previous, &stream, "After QueryInterface"); + + StartSeh() + res = IStream_Stat(istream, NULL, 0); + ok(res == STG_E_INVALIDPOINTER, "Stat to IStream returned wrong hResult: 0x%08x.\n", res); + EndSeh(STATUS_SUCCESS); + + StartSeh() + res = IStream_Stat(istream, &stat, STATFLAG_NONAME); + ok(res == S_OK, "Stat to IStream returned wrong hResult: 0x%08x.\n", res); + EndSeh(STATUS_SUCCESS); + + ok(stream.Current == buffer2, + "stream.Current points to the wrong address 0x%p (expected 0x%p)\n", + stream.Current, buffer2); + ok(stream.Start == buffer2, "stream.Start was changed unexpectedly\n"); + ok(stream.End == buffer2 + sizeof(buffer2), "stream.End was changed unexpectedly\n"); + + ok(stat.cbSize.QuadPart == ((PUCHAR)stream.End - (PUCHAR)stream.Start), + "stat.cbSize has the wrong value %lld (expected %d)\n", + stat.cbSize.QuadPart, (PUCHAR)stream.End - (PUCHAR)stream.Start); + + CompareStructsAndSaveForLater(&previous, &stream, "After Stat"); + + StartSeh() + res = IStream_AddRef(istream); + ok(res == 3, "AddRef to IStream returned wrong hResult: %d.\n", res); + EndSeh(STATUS_SUCCESS); + + StartSeh() + res = IStream_AddRef(istream); + ok(res == 4, "AddRef to IStream returned wrong hResult: %d.\n", res); + EndSeh(STATUS_SUCCESS); + + StartSeh() + res = IStream_Release(istream); + ok(res == 3, "Release to IStream returned wrong hResult: %d.\n", res); + EndSeh(STATUS_SUCCESS); + + StartSeh() + res = IStream_AddRef(istream); + ok(res == 4, "AddRef to IStream returned wrong hResult: %d.\n", res); + EndSeh(STATUS_SUCCESS); + + StartSeh() + res = IStream_Release(istream); + ok(res == 3, "Release to IStream returned wrong hResult: %d.\n", res); + EndSeh(STATUS_SUCCESS); + + StartSeh() + res = IStream_Release(istream); + ok(res == 2, "Release to IStream returned wrong hResult: %d.\n", res); + EndSeh(STATUS_SUCCESS); + + CompareStructsAndSaveForLater(&previous, &stream, "After AddRef"); + + StartSeh() + res = IStream_Read(istream, NULL, 0, &bytesRead); + ok(res == S_OK, "Read to IStream returned wrong hResult: 0x%08x.\n", res); + EndSeh(STATUS_SUCCESS); + + StartSeh() + res = IStream_Read(istream, buffer, 40, NULL); + ok(res == S_OK, "Read to IStream returned wrong hResult: 0x%08x.\n", res); + EndSeh(STATUS_ACCESS_VIOLATION); + + StartSeh() + res = IStream_Read(istream, buffer + 40, 39, &bytesRead); + ok(res == S_OK, "Read to IStream returned wrong hResult: 0x%08x.\n", res); + EndSeh(STATUS_SUCCESS); + + if (SUCCEEDED(res)) + { + bytesRead += 40; + for (i = 0; i < bytesRead; i++) + { + ok(buffer[i] == i, "Buffer[%d] contains a wrong number %d (expected %d).\n", i, buffer[i], i); + } + } + + ok(stream.Current == buffer2 + 79, + "stream.Current points to the wrong address 0x%p (expected 0x%p)\n", + stream.Current, buffer2); + ok(stream.Start == buffer2, "stream.Start was changed unexpectedly\n"); + ok(stream.End == buffer2 + sizeof(buffer2), "stream.End was changed unexpectedly\n"); + + CompareStructsAndSaveForLater(&previous, &stream, "After Read 1"); + + size.QuadPart = 0x9090909090909090ull; + + StartSeh() + move.QuadPart = -1; + res = IStream_Seek(istream, move, STREAM_SEEK_END, &size); + ok(res == STG_E_INVALIDPOINTER, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == 0x9090909090909090ull, "Seek modified the new location in an error (0x%08x,0x%08x).\n", size.HighPart, size.LowPart); + EndSeh(STATUS_SUCCESS); + + StartSeh() + move.QuadPart = 0; + res = IStream_Seek(istream, move, STREAM_SEEK_END, &size); + ok(res == S_OK, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == (PUCHAR)stream.End - (PUCHAR)stream.Start, "Seek new location unexpected value: 0x%08x.\n", size.LowPart); + EndSeh(STATUS_SUCCESS); + + size.QuadPart = 0x9090909090909090ull; + + StartSeh() + move.QuadPart = 1; + res = IStream_Seek(istream, move, STREAM_SEEK_END, &size); + ok(res == S_OK, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == (PUCHAR)stream.End - (PUCHAR)stream.Start - 1, "Seek new location unexpected value: 0x%08x.\n", size.LowPart); + EndSeh(STATUS_SUCCESS); + + size.QuadPart = 0x9090909090909090ull; + + StartSeh() + move.QuadPart = 2; + res = IStream_Seek(istream, move, STREAM_SEEK_END, &size); + ok(res == S_OK, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == (PUCHAR)stream.End - (PUCHAR)stream.Start - 2, "Seek new location unexpected value: 0x%08x.\n", size.LowPart); + EndSeh(STATUS_SUCCESS); + + size.QuadPart = 0x9090909090909090ull; + + StartSeh() + move.QuadPart = -20; + res = IStream_Seek(istream, move, STREAM_SEEK_SET, &size); + ok(res == STG_E_INVALIDPOINTER, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == 0x9090909090909090ull, "Seek modified the new location in an error.\n"); + EndSeh(STATUS_SUCCESS); + + StartSeh() + move.QuadPart = 4000; + res = IStream_Seek(istream, move, STREAM_SEEK_SET, &size); + ok(res == STG_E_INVALIDPOINTER, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == 0x9090909090909090ull, "Seek modified the new location in an error.\n"); + EndSeh(STATUS_SUCCESS); + + StartSeh() + move.QuadPart = 0x100000000ull; + res = IStream_Seek(istream, move, STREAM_SEEK_SET, &size); +#ifdef _WIN64 + ok(res == STG_E_INVALIDPOINTER, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == 0x9090909090909090ull, "Seek modified the new location in an error (0x%08x,0x%08x).\n", size.HighPart, size.LowPart); +#else + ok(res == S_OK, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == 0, "Seek new location unexpected value: 0x%08x.\n", size.LowPart); +#endif + EndSeh(STATUS_SUCCESS); + +#ifdef _WIN64 + StartSeh() + move.QuadPart = 0; + res = IStream_Seek(istream, move, STREAM_SEEK_SET, &size); + ok(res == S_OK, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == 0, "Seek new location unexpected value: 0x%08x.\n", size.LowPart); + EndSeh(STATUS_SUCCESS); +#endif + + size.QuadPart = 0x9090909090909090ull; + + StartSeh() + move.QuadPart = -20; + res = IStream_Seek(istream, move, STREAM_SEEK_CUR, &size); + ok(res == STG_E_INVALIDPOINTER, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == 0x9090909090909090ull, "Seek modified the new location in an error (0x%08x,0x%08x).\n", size.HighPart, size.LowPart); + EndSeh(STATUS_SUCCESS); + + StartSeh() + move.QuadPart = 0x100000000ull; + res = IStream_Seek(istream, move, STREAM_SEEK_CUR, &size); +#ifdef _WIN64 + ok(res == STG_E_INVALIDPOINTER, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == 0x9090909090909090ull, "Seek modified the new location in an error (0x%08x,0x%08x).\n", size.HighPart, size.LowPart); +#else + ok(res == S_OK, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + ok(size.QuadPart == 0, "Seek new location unexpected value: 0x%08x.\n", size.LowPart); +#endif + EndSeh(STATUS_SUCCESS); + + StartSeh() + move.QuadPart = 40; + res = IStream_Seek(istream, move, STREAM_SEEK_SET, &size); + ok(res == S_OK, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + EndSeh(STATUS_SUCCESS); + + ok(size.QuadPart == 40, + "Seek returned wrong offset %lld (expected %d)\n", + size.QuadPart, 40); + + ok(stream.Current == buffer2 + 40, + "stream.Current points to the wrong address 0x%p (expected 0x%p)\n", + stream.Current, buffer2); + ok(stream.Start == buffer2, "stream.Start was changed unexpectedly\n"); + ok(stream.End == buffer2 + sizeof(buffer2), "stream.End was changed unexpectedly\n"); + + CompareStructsAndSaveForLater(&previous, &stream, "After Seek"); + + res = IStream_Read(istream, buffer, sizeof(buffer), &bytesRead); + + ok(res == S_OK, "Read to IStream returned wrong hResult: 0x%08x.\n", res); + + if (SUCCEEDED(res)) + { + for (i = 0; i < bytesRead; i++) + { + ok(buffer[i] == (i + 40), "Buffer[%d] contains a wrong number %d (expected %d).\n", i, buffer[i], i + 40); + } + } + + ok(stream.Current == buffer2 + 40 + sizeof(buffer), + "stream.Current points to the wrong address 0x%p (expected 0x%p)\n", + stream.Current, buffer2); + ok(stream.Start == buffer2, "stream.Start was changed unexpectedly\n"); + ok(stream.End == buffer2 + sizeof(buffer2), "stream.End was changed unexpectedly\n"); + + CompareStructsAndSaveForLater(&previous, &stream, "After Read 2"); + + res = IStream_Release(istream); + + ok(res == 1, "Release to IStream returned wrong hResult: 0x%08x.\n", res); + + ok(stream.RefCount == 1, "RefCount has a wrong value: %d (expected %d).\n", stream.RefCount, 1); + + res = IStream_Release(istream); + + ok(res == S_OK, "Release to IStream returned wrong hResult: 0x%08x.\n", res); + + ok(stream.RefCount == 0, "RefCount has a wrong value: %d (expected %d).\n", stream.RefCount, 0); + + ok(finalReleaseCallCount == 1, "FinalRelease was called %d times instead of 1.\n", finalReleaseCallCount); +} + +void test_OutOfProcess() +{ + LARGE_INTEGER move; + ULARGE_INTEGER size; + HRESULT res; + HANDLE process; + ULONG i; + + RTL_MEMORY_STREAM stream; + RTL_MEMORY_STREAM previous; + + IStream * istream; + + UCHAR buffer[80]; + UCHAR buffer2[180]; + ULONG bytesRead; + + STATSTG stat; + + finalReleaseCallCount = 0; + + for (i = 0; i < sizeof(buffer2); i++) + { + buffer2[i] = i % UCHAR_MAX; + } + + memset(&stream, 0x90, sizeof(stream)); + memset(&previous, 0x00, sizeof(previous)); + + process = GetCurrentProcess(); + + RtlInitOutOfProcessMemoryStream(&stream); + + ok(stream.FinalRelease == RtlFinalReleaseOutOfProcessMemoryStream, + "stream.FinalRelease unexpected %p != %p.\n", + stream.FinalRelease, RtlFinalReleaseOutOfProcessMemoryStream); + + ok(stream.RefCount == 0, "RefCount has a wrong value: %d (expected %d).\n", stream.RefCount, 0); + + CompareStructsAndSaveForLater(&previous, &stream, "After init"); + + stream.Current = buffer2; + stream.Start = buffer2; + stream.End = buffer2 + sizeof(buffer2); + stream.ProcessHandle = process; + stream.FinalRelease = CustomFinalReleaseOutOfProcessMemoryStream; + + CompareStructsAndSaveForLater(&previous, &stream, "After assigning"); + + res = IStream_QueryInterface((struct IStream*)&stream, &IID_IStream, (void**)&istream); + + ok(res == S_OK, "QueryInterface to IStream returned wrong hResult: 0x%08x.\n", res); + + ok(stream.RefCount == 1, "RefCount has a wrong value: %d (expected %d).\n", stream.RefCount, 1); + + ok(stream.ProcessHandle == process, + "ProcessHandle changed unexpectedly: 0x%08x (expected 0x%p)\n", + stream.ProcessHandle, process); + + CompareStructsAndSaveForLater(&previous, &stream, "After QueryInterface"); + + res = IStream_Stat(istream, &stat, STATFLAG_NONAME); + + ok(res == S_OK, "Stat to IStream returned wrong hResult: 0x%08x.\n", res); + + ok(stream.Current == buffer2, + "stream.Current points to the wrong address 0x%p (expected 0x%p)\n", + stream.Current, buffer2); + ok(stream.Start == buffer2, "stream.Start was changed unexpectedly\n"); + ok(stream.End == buffer2 + sizeof(buffer2), "stream.End was changed unexpectedly\n"); + ok(stream.ProcessHandle == process, + "ProcessHandle changed unexpectedly: 0x%08x (expected 0x%p)\n", + stream.ProcessHandle, process); + + ok(stat.cbSize.QuadPart == ((PUCHAR)stream.End - (PUCHAR)stream.Start), + "stat.cbSize has the wrong value %lld (expected %d)\n", + stat.cbSize.QuadPart, (PUCHAR)stream.End - (PUCHAR)stream.Start); + + CompareStructsAndSaveForLater(&previous, &stream, "After Stat"); + + res = IStream_Read(istream, buffer, sizeof(buffer), &bytesRead); + + ok(res == S_OK, "Read to IStream returned wrong hResult: 0x%08x.\n", res); + + if (SUCCEEDED(res)) + { + for (i = 0; i < bytesRead; i++) + { + ok(buffer[i] == i, "Buffer[%d] contains a wrong number %d (expected %d).\n", i, buffer[i], i); + } + } + + ok(stream.Current == buffer2 + sizeof(buffer), + "stream.Current points to the wrong address 0x%p (expected 0x%p)\n", + stream.Current, buffer2); + ok(stream.Start == buffer2, "stream.Start was changed unexpectedly\n"); + ok(stream.End == buffer2 + sizeof(buffer2), "stream.End was changed unexpectedly\n"); + ok(stream.ProcessHandle == process, + "ProcessHandle changed unexpectedly: 0x%08x (expected 0x%p)\n", + stream.ProcessHandle, process); + + CompareStructsAndSaveForLater(&previous, &stream, "After Read 1"); + + move.QuadPart = 40; + + res = IStream_Seek(istream, move, STREAM_SEEK_SET, &size); + + ok(res == S_OK, "Seek to IStream returned wrong hResult: 0x%08x.\n", res); + + ok(size.QuadPart == 40, + "Seek returned wrong offset %lld (expected %d)\n", + size.QuadPart, 40); + + ok(stream.Current == buffer2 + 40, + "stream.Current points to the wrong address 0x%p (expected 0x%p)\n", + stream.Current, buffer2); + ok(stream.Start == buffer2, "stream.Start was changed unexpectedly\n"); + ok(stream.End == buffer2 + sizeof(buffer2), "stream.End was changed unexpectedly\n"); + ok(stream.ProcessHandle == process, + "ProcessHandle changed unexpectedly: 0x%08x (expected 0x%p)\n", + stream.ProcessHandle, process); + + CompareStructsAndSaveForLater(&previous, &stream, "After Seek"); + + res = IStream_Read(istream, buffer, sizeof(buffer), &bytesRead); + + ok(res == S_OK, "Read to IStream returned wrong hResult: 0x%08x.\n", res); + + if (SUCCEEDED(res)) + { + for (i = 0; i < bytesRead; i++) + { + ok(buffer[i] == (i + 40), "Buffer[%d] contains a wrong number %d (expected %d).\n", i, buffer[i], i + 40); + } + } + + ok(stream.Current == buffer2 + 40 + sizeof(buffer), + "stream.Current points to the wrong address 0x%p (expected 0x%p)\n", + stream.Current, buffer2); + ok(stream.Start == buffer2, "stream.Start was changed unexpectedly\n"); + ok(stream.End == buffer2 + sizeof(buffer2), "stream.End was changed unexpectedly\n"); + ok(stream.ProcessHandle == process, + "ProcessHandle changed unexpectedly: 0x%08x (expected 0x%p)\n", + stream.ProcessHandle, process); + + CompareStructsAndSaveForLater(&previous, &stream, "After Read 2"); + + res = IStream_Release(istream); + + ok(res == S_OK, "Release to IStream returned wrong hResult: 0x%08x.\n", res); + + ok(stream.RefCount == 0, "RefCount has a wrong value: %d (expected %d).\n", stream.RefCount, 0); + + ok(finalReleaseCallCount == 1, "FinalRelease was called %d times instead of 1.\n", finalReleaseCallCount); +} + +START_TEST(RtlMemoryStream) +{ + test_InProcess(); + test_OutOfProcess(); +} diff --git a/rostests/apitests/ntdll/testlist.c b/rostests/apitests/ntdll/testlist.c index 8dfa718cee7..0f45d260e6b 100644 --- a/rostests/apitests/ntdll/testlist.c +++ b/rostests/apitests/ntdll/testlist.c @@ -25,6 +25,7 @@ extern void func_RtlGetFullPathName_UstrEx(void); extern void func_RtlGetLengthWithoutTrailingPathSeperators(void); extern void func_RtlGetLongestNtPathLength(void); extern void func_RtlInitializeBitMap(void); +extern void func_RtlMemoryStream(void); extern void func_TimerResolution(void); const struct test winetest_testlist[] = @@ -51,6 +52,7 @@ const struct test winetest_testlist[] = { "RtlGetLengthWithoutTrailingPathSeperators", func_RtlGetLengthWithoutTrailingPathSeperators }, { "RtlGetLongestNtPathLength", func_RtlGetLongestNtPathLength }, { "RtlInitializeBitMap", func_RtlInitializeBitMap }, + { "RtlMemoryStream", func_RtlMemoryStream }, { "TimerResolution", func_TimerResolution }, { 0, 0 }