[tfaber] 53810: [KMTESTS/MM] - Extend MmSection test to include passing different files and mapping a view
Author: tfaber Date: Fri Sep 23 11:43:02 2011 New Revision: 53810
URL: http://svn.reactos.org/svn/reactos?rev=53810&view=rev Log: [KMTESTS/MM] - Extend MmSection test to include passing different files and mapping a view
Modified: trunk/rostests/kmtests/ntos_mm/MmSection.c
Modified: trunk/rostests/kmtests/ntos_mm/MmSection.c URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/ntos_mm/MmSection.... ============================================================================== --- trunk/rostests/kmtests/ntos_mm/MmSection.c [iso-8859-1] (original) +++ trunk/rostests/kmtests/ntos_mm/MmSection.c [iso-8859-1] Fri Sep 23 11:43:02 2011 @@ -18,6 +18,57 @@ ok_eq_hex(Status, STATUS_SUCCESS); \ ok_eq_ulong(ObjectInfo.PointerCount, Pointers); \ ok_eq_ulong(ObjectInfo.HandleCount, Handles); \ +} while (0) + +#define CheckSection(SectionObject, SectionFlag) do \ +{ \ + SECTION_BASIC_INFORMATION Sbi; \ + HANDLE SectionHandle = NULL; \ + NTSTATUS Status; \ + if (skip(SectionObject != NULL && \ + SectionObject != (PVOID)0x5555555555555555ULL, \ + "blah\n")) \ + break; \ + Status = ObOpenObjectByPointer(SectionObject, OBJ_KERNEL_HANDLE, \ + NULL, 0, MmSectionObjectType, \ + KernelMode, &SectionHandle); \ + ok_eq_hex(Status, STATUS_SUCCESS); \ + ok(SectionHandle != NULL, "Section handle null\n"); \ + if (!skip(NT_SUCCESS(Status) && SectionHandle, \ + "No section handle\n")) \ + { \ + Status = ZwQuerySection(SectionHandle, SectionBasicInformation, \ + &Sbi, sizeof Sbi, NULL); \ + ok_eq_hex(Status, STATUS_SUCCESS); \ + ok_eq_pointer(Sbi.BaseAddress, NULL); \ + ok_eq_longlong(Sbi.Size.QuadPart, 1LL); \ + ok_eq_hex(Sbi.Attributes, SectionFlag | SEC_FILE); \ + ZwClose(SectionHandle); \ + } \ +} while (0) + +#define TestMapView(SectionObject, ExpectAtBase, ExpectM) do \ +{ \ + NTSTATUS Status; \ + PVOID BaseAddress = NULL; \ + SIZE_T ViewSize = 0; \ + LARGE_INTEGER SectionOffset; \ + if (skip(SectionObject != NULL && \ + SectionObject != (PVOID)0x5555555555555555ULL, \ + "No section object\n")) \ + break; \ + \ + SectionOffset.QuadPart = 0; \ + Status = MmMapViewOfSection(SectionObject, PsGetCurrentProcess(), \ + &BaseAddress, 0, 1, &SectionOffset, \ + &ViewSize, ViewUnmap, 0, PAGE_READONLY); \ + ok_eq_hex(Status, ExpectAtBase ? STATUS_SUCCESS : STATUS_IMAGE_NOT_AT_BASE);\ + if (!skip(NT_SUCCESS(Status), "Section not mapped\n")) \ + { \ + ok_eq_uint(*(PUCHAR)BaseAddress, ExpectM ? 'M' : 0); \ + Status = MmUnmapViewOfSection(PsGetCurrentProcess(), BaseAddress); \ + ok_eq_hex(Status, STATUS_SUCCESS); \ + } \ } while (0)
static @@ -149,6 +200,8 @@ ok(SectionObject != InvalidPointer, "Section object pointer untouched\n"); ok(SectionObject != NULL, "Section object pointer NULL\n"); CheckObject(FileHandle2, PointerCount2, 1L); + CheckSection(SectionObject, SEC_IMAGE); + TestMapView(SectionObject, FALSE, TRUE);
if (SectionObject && SectionObject != InvalidPointer) ObDereferenceObject(SectionObject); @@ -165,6 +218,8 @@ ok(SectionObject != NULL, "Section object pointer NULL\n"); ++PointerCount2; CheckObject(FileHandle2, PointerCount2, 1L); + CheckSection(SectionObject, 0); + TestMapView(SectionObject, TRUE, TRUE);
if (SectionObject && SectionObject != InvalidPointer) ObDereferenceObject(SectionObject); @@ -181,6 +236,8 @@ ok(SectionObject != InvalidPointer, "Section object pointer untouched\n"); ok(SectionObject != NULL, "Section object pointer NULL\n"); CheckObject(FileHandle2, PointerCount2, 1L); + CheckSection(SectionObject, 0); + TestMapView(SectionObject, TRUE, TRUE);
if (SectionObject && SectionObject != InvalidPointer) ObDereferenceObject(SectionObject); @@ -212,6 +269,8 @@ ok(SectionObject != NULL, "Section object pointer NULL\n"); ++PointerCount1; CheckObject(FileHandle1, PointerCount1, 1L); + CheckSection(SectionObject, 0); + TestMapView(SectionObject, TRUE, FALSE);
if (SectionObject && SectionObject != InvalidPointer) ObDereferenceObject(SectionObject); @@ -228,6 +287,45 @@ ok(SectionObject != InvalidPointer, "Section object pointer untouched\n"); ok(SectionObject != NULL, "Section object pointer NULL\n"); CheckObject(FileHandle1, PointerCount1, 1L); + CheckSection(SectionObject, 0); + TestMapView(SectionObject, TRUE, FALSE); + + if (SectionObject && SectionObject != InvalidPointer) + ObDereferenceObject(SectionObject); + + /* image section with two different files */ + CheckObject(FileHandle1, PointerCount1, 1L); + SectionObject = InvalidPointer; + MaximumSize.QuadPart = 1; + StartSeh() + Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, FileHandle1, FileObject2); + EndSeh(STATUS_SUCCESS); + ok_eq_hex(Status, STATUS_SUCCESS); + ok_eq_longlong(MaximumSize.QuadPart, 1LL); + ok(SectionObject != InvalidPointer, "Section object pointer untouched\n"); + ok(SectionObject != NULL, "Section object pointer NULL\n"); + CheckObject(FileHandle1, PointerCount1, 1L); + CheckObject(FileHandle2, PointerCount2, 1L); + CheckSection(SectionObject, 0); + TestMapView(SectionObject, TRUE, TRUE); + + if (SectionObject && SectionObject != InvalidPointer) + ObDereferenceObject(SectionObject); + + CheckObject(FileHandle1, PointerCount1, 1L); + SectionObject = InvalidPointer; + MaximumSize.QuadPart = 1; + StartSeh() + Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, FileHandle2, FileObject1); + EndSeh(STATUS_SUCCESS); + ok_eq_hex(Status, STATUS_SUCCESS); + ok_eq_longlong(MaximumSize.QuadPart, 1LL); + ok(SectionObject != InvalidPointer, "Section object pointer untouched\n"); + ok(SectionObject != NULL, "Section object pointer NULL\n"); + CheckObject(FileHandle1, PointerCount1, 1L); + CheckObject(FileHandle2, PointerCount2, 1L); + CheckSection(SectionObject, 0); + TestMapView(SectionObject, TRUE, FALSE);
if (SectionObject && SectionObject != InvalidPointer) ObDereferenceObject(SectionObject); @@ -244,6 +342,8 @@ ok(SectionObject != InvalidPointer, "Section object pointer untouched\n"); ok(SectionObject != NULL, "Section object pointer NULL\n"); CheckObject(FileHandle1, PointerCount1, 1L); + CheckSection(SectionObject, 0); + TestMapView(SectionObject, TRUE, FALSE);
if (SectionObject && SectionObject != InvalidPointer) ObDereferenceObject(SectionObject); @@ -259,6 +359,8 @@ ok(SectionObject != InvalidPointer, "Section object pointer untouched\n"); ok(SectionObject != NULL, "Section object pointer NULL\n"); CheckObject(FileHandle1, PointerCount1, 1L); + CheckSection(SectionObject, 0); + TestMapView(SectionObject, TRUE, FALSE);
if (SectionObject && SectionObject != InvalidPointer) ObDereferenceObject(SectionObject); @@ -274,6 +376,8 @@ ok(SectionObject != InvalidPointer, "Section object pointer untouched\n"); ok(SectionObject != NULL, "Section object pointer NULL\n"); CheckObject(FileHandle1, PointerCount1, 1L); + CheckSection(SectionObject, 0); + TestMapView(SectionObject, TRUE, FALSE);
if (SectionObject && SectionObject != InvalidPointer) ObDereferenceObject(SectionObject); @@ -294,6 +398,7 @@ LARGE_INTEGER FileOffset; UCHAR FileData = 0;
+ ok(ExGetPreviousMode() == UserMode, "Previous mode is kernel mode\n"); /* create a one-byte file that we can use */ InitializeObjectAttributes(&ObjectAttributes, &FileName1, OBJ_CASE_INSENSITIVE, NULL, NULL); Status = ZwCreateFile(&FileHandle1, GENERIC_ALL, &ObjectAttributes, &IoStatusBlock, NULL, FILE_ATTRIBUTE_NORMAL, FILE_SHARE_READ, FILE_SUPERSEDE, FILE_NON_DIRECTORY_FILE, NULL, 0);
-
tfaber＠svn.reactos.org