Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-07-01 10:20:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NTDLL_APITEST] Show that NtMapViewOfSection does not properly NULL out the remainder of the mapped memory. CORE-11206

Browse source 
svn path=/trunk/; revision=71289
This commit is contained in:
Mark Jansen 2016-05-08 17:01:15 +00:00
parent da19f382ab
commit 0f8226d756
1 changed files with 96 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

96
rostests/apitests/ntdll/NtMapViewOfSection.c
View file

@ -1398,6 +1398,101 @@ Test_EmptyFile(VOID)
DeleteFileW(FileName);
}
// CORE-11206
static void
Test_Truncate(VOID)
{
WCHAR TempPath[MAX_PATH];
WCHAR FileName[MAX_PATH];
NTSTATUS Status;
SIZE_T ViewSize = 0;
HANDLE Handle;
HANDLE SectionHandle;
ULONG Length;
BOOL Success;
DWORD Written, Error;
VOID* BaseAddress;
Length = GetTempPathW(MAX_PATH, TempPath);
ok(Length != 0, "GetTempPathW failed with %lu\n", GetLastError());
Length = GetTempFileNameW(TempPath, L"nta", 0, FileName);
ok(Length != 0, "GetTempFileNameW failed with %lu\n", GetLastError());
Handle = CreateFileW(FileName, FILE_ALL_ACCESS, 0, NULL, CREATE_ALWAYS, 0, NULL);
Success = WriteFile(Handle, "TESTDATA", 8, &Written, NULL);
ok(Success == TRUE, "WriteFile failed with %lu\n", GetLastError());
ok(Written == 8, "WriteFile wrote %lu bytes\n", Written);
Written = SetFilePointer(Handle, 6, NULL, FILE_BEGIN);
ok(Written == 6, "SetFilePointer returned %lu bytes\n", Written);
Success = SetEndOfFile(Handle);
ok(Success == TRUE, "SetEndOfFile failed with %lu\n", GetLastError());
Status = NtCreateSection(&SectionHandle,
STANDARD_RIGHTS_REQUIRED | SECTION_QUERY | SECTION_MAP_READ,
0, 0, PAGE_READONLY, SEC_COMMIT, Handle);
ok_ntstatus(Status, STATUS_SUCCESS);
BaseAddress = NULL;
ViewSize = 0;
Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0,
0, 0, &ViewSize, ViewShare, 0, PAGE_READONLY);
ok_ntstatus(Status, STATUS_SUCCESS);
if (BaseAddress)
{
// First we test data that was truncated even before the file mapping was opened
Length = strlen((char*)BaseAddress);
ok(Length == 6, "Old data was not properly erased! (Length=%lu)\n", Length);
}
// Now we truncate the file on disk some more
Written = SetFilePointer(Handle, 4, NULL, FILE_BEGIN);
ok(Written == 4, "SetFilePointer returned %lu bytes\n", Written);
Success = SetEndOfFile(Handle);
Error = GetLastError();
ok(Success == FALSE, "SetEndOfFile succeeded\n");
ok(Error == ERROR_USER_MAPPED_FILE, "SetEndOfFile did not set error to ERROR_USER_MAPPED_FILE (%lu)\n", Error);
if (BaseAddress)
{
Length = strlen((char*)BaseAddress);
ok(Length == 6, "Length should not have changed! (Length=%lu)\n", Length);
}
// Unmap and set the end shorter.
Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress);
ok_ntstatus(Status, STATUS_SUCCESS);
Success = CloseHandle(SectionHandle);
ok(Success == TRUE, "CloseHandle failed with %lu\n", GetLastError());
Written = SetFilePointer(Handle, 4, NULL, FILE_BEGIN);
ok(Written == 4, "SetFilePointer returned %lu bytes\n", Written);
Success = SetEndOfFile(Handle);
ok(Success == TRUE, "SetEndOfFile failed with %lu\n", GetLastError());
Status = NtCreateSection(&SectionHandle,
STANDARD_RIGHTS_REQUIRED | SECTION_QUERY | SECTION_MAP_READ,
0, 0, PAGE_READONLY, SEC_COMMIT, Handle);
ok_ntstatus(Status, STATUS_SUCCESS);
BaseAddress = NULL;
ViewSize = 0;
Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0,
0, 0, &ViewSize, ViewShare, 0, PAGE_READONLY);
ok_ntstatus(Status, STATUS_SUCCESS);
// CLEANUP
Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress);
ok_ntstatus(Status, STATUS_SUCCESS);
Success = CloseHandle(SectionHandle);
ok(Success == TRUE, "CloseHandle failed with %lu\n", GetLastError());
Success = CloseHandle(Handle);
ok(Success == TRUE, "CloseHandle failed with %lu\n", GetLastError());
Success = DeleteFileW(FileName);
ok(Success == TRUE, "DeleteFileW failed with %lu\n", GetLastError());
}
START_TEST(NtMapViewOfSection)
{
Test_PageFileSection();
@ -1406,4 +1501,5 @@ START_TEST(NtMapViewOfSection)
Test_NoLoadSection(FALSE);
Test_NoLoadSection(TRUE);
Test_EmptyFile();
Test_Truncate();
}