[tfaber] 53785: [KMTESTS/MM] - Add some tests for MmCreateSection involving file objects

Author: tfaber
Date: Wed Sep 21 10:35:33 2011
New Revision: 53785

URL: http://svn.reactos.org/svn/reactos?rev=53785&view=rev
Log:
[KMTESTS/MM]
- Add some tests for MmCreateSection involving file objects

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] Wed Sep 21 10:35:33 2011
@@ -10,13 +10,30 @@
 #define StartSeh() ExceptionStatus = STATUS_SUCCESS; _SEH2_TRY {
 #define EndSeh(ExpectedStatus) } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
ExceptionStatus = _SEH2_GetExceptionCode(); } _SEH2_END; ok_eq_hex(ExceptionStatus,
ExpectedStatus)
 
-START_TEST(MmSection)
+#define CheckObject(Handle, Pointers, Handles) do                   \
+{                                                                   \
+    PUBLIC_OBJECT_BASIC_INFORMATION ObjectInfo;                     \
+    Status = ZwQueryObject(Handle, ObjectBasicInformation,          \
+                            &ObjectInfo, sizeof ObjectInfo, NULL);  \
+    ok_eq_hex(Status, STATUS_SUCCESS);                              \
+    ok_eq_ulong(ObjectInfo.PointerCount, Pointers);                 \
+    ok_eq_ulong(ObjectInfo.HandleCount, Handles);                   \
+} while (0)
+
+static
+VOID
+TestCreateSection(
+    IN HANDLE FileHandle1,
+    IN PFILE_OBJECT FileObject1,
+    IN HANDLE FileHandle2,
+    IN PFILE_OBJECT FileObject2)
 {
     NTSTATUS Status = STATUS_SUCCESS;
     NTSTATUS ExceptionStatus;
     const PVOID InvalidPointer = (PVOID)0x5555555555555555ULL;
     PVOID SectionObject;
     LARGE_INTEGER MaximumSize;
+    ULONG PointerCount1, PointerCount2;
 
     StartSeh()
         Status = MmCreateSection(NULL, 0, NULL, NULL, 0, SEC_RESERVE, NULL, NULL);
@@ -101,6 +118,7 @@
     if (SectionObject && SectionObject != InvalidPointer)
         ObDereferenceObject(SectionObject);
 
+    /* page file section */
     SectionObject = InvalidPointer;
     MaximumSize.QuadPart = 1;
     StartSeh()
@@ -113,4 +131,224 @@
 
     if (SectionObject && SectionObject != InvalidPointer)
         ObDereferenceObject(SectionObject);
+
+    if (!skip(FileHandle1 != NULL && FileObject1 != NULL &&
+              FileHandle2 != NULL && FileObject2 != NULL, "No file handle or
object\n"))
+    {
+        PointerCount1 = 3;
+        PointerCount2 = 3;
+        /* image section */
+        CheckObject(FileHandle2, PointerCount2, 1L);
+        SectionObject = InvalidPointer;
+        MaximumSize.QuadPart = 1;
+        StartSeh()
+            Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize,
PAGE_READONLY, SEC_IMAGE, FileHandle2, NULL);
+        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(FileHandle2, PointerCount2, 1L);
+
+        if (SectionObject && SectionObject != InvalidPointer)
+            ObDereferenceObject(SectionObject);
+
+        CheckObject(FileHandle2, PointerCount2, 1L);
+        SectionObject = InvalidPointer;
+        MaximumSize.QuadPart = 1;
+        StartSeh()
+            Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize,
PAGE_READONLY, SEC_IMAGE, NULL, 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");
+        ++PointerCount2;
+        CheckObject(FileHandle2, PointerCount2, 1L);
+
+        if (SectionObject && SectionObject != InvalidPointer)
+            ObDereferenceObject(SectionObject);
+        //--PointerCount2;  // ????
+
+        CheckObject(FileHandle2, PointerCount2, 1L);
+        SectionObject = InvalidPointer;
+        MaximumSize.QuadPart = 1;
+        StartSeh()
+            Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize,
PAGE_READONLY, SEC_IMAGE, FileHandle2, 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(FileHandle2, PointerCount2, 1L);
+
+        if (SectionObject && SectionObject != InvalidPointer)
+            ObDereferenceObject(SectionObject);
+
+        /* image section with inappropriate file */
+        CheckObject(FileHandle1, PointerCount1, 1L);
+        SectionObject = InvalidPointer;
+        MaximumSize.QuadPart = 1;
+        StartSeh()
+            Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize,
PAGE_READONLY, SEC_IMAGE, FileHandle1, NULL);
+        EndSeh(STATUS_SUCCESS);
+        ok_eq_hex(Status, STATUS_INVALID_IMAGE_NOT_MZ);
+        ok_eq_longlong(MaximumSize.QuadPart, 1LL);
+        ok_eq_pointer(SectionObject, InvalidPointer);
+        CheckObject(FileHandle1, PointerCount1, 1L);
+
+        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, NULL, 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");
+        ++PointerCount1;
+        CheckObject(FileHandle1, PointerCount1, 1L);
+
+        if (SectionObject && SectionObject != InvalidPointer)
+            ObDereferenceObject(SectionObject);
+        //--PointerCount1; // ????
+
+        CheckObject(FileHandle1, PointerCount1, 1L);
+        SectionObject = InvalidPointer;
+        MaximumSize.QuadPart = 1;
+        StartSeh()
+            Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize,
PAGE_READONLY, SEC_IMAGE, FileHandle1, 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);
+
+        if (SectionObject && SectionObject != InvalidPointer)
+            ObDereferenceObject(SectionObject);
+
+        /* data file section */
+        CheckObject(FileHandle1, PointerCount1, 1L);
+        SectionObject = InvalidPointer;
+        MaximumSize.QuadPart = 1;
+        StartSeh()
+            Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize,
PAGE_READONLY, SEC_RESERVE, FileHandle1, NULL);
+        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);
+
+        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_RESERVE, NULL, 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);
+
+        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_RESERVE, FileHandle1, 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);
+
+        if (SectionObject && SectionObject != InvalidPointer)
+            ObDereferenceObject(SectionObject);
+
+        CheckObject(FileHandle1, PointerCount1, 1L);
+    }
 }
+
+START_TEST(MmSection)
+{
+    NTSTATUS Status;
+    HANDLE FileHandle1 = NULL, FileHandle2 = NULL;
+    PFILE_OBJECT FileObject1 = NULL, FileObject2 = NULL;
+    OBJECT_ATTRIBUTES ObjectAttributes;
+    IO_STATUS_BLOCK IoStatusBlock;
+    UNICODE_STRING FileName1 =
RTL_CONSTANT_STRING(L"\\SystemRoot\\kmtest-MmSection.txt");
+    UNICODE_STRING FileName2 =
RTL_CONSTANT_STRING(L"\\SystemRoot\\system32\\ntdll.dll");
+    LARGE_INTEGER FileOffset;
+    UCHAR FileData = 0;
+
+    /* 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);
+    ok_eq_hex(Status, STATUS_SUCCESS);
+    ok_eq_ulongptr(IoStatusBlock.Information, FILE_CREATED);
+    ok(FileHandle1 != NULL, "FileHandle1 is NULL\n");
+    if (FileHandle1)
+    {
+        FileOffset.QuadPart = 0;
+        Status = ZwWriteFile(FileHandle1, NULL, NULL, NULL, &IoStatusBlock,
&FileData, sizeof FileData, &FileOffset, NULL);
+        ok_eq_hex(Status, STATUS_SUCCESS);
+        ok_eq_ulongptr(IoStatusBlock.Information, 1);
+        Status = ZwClose(FileHandle1);
+        ok_eq_hex(Status, STATUS_SUCCESS);
+        FileHandle1 = NULL;
+    }
+
+    InitializeObjectAttributes(&ObjectAttributes, &FileName1,
OBJ_CASE_INSENSITIVE, NULL, NULL);
+    Status = ZwCreateFile(&FileHandle1, GENERIC_ALL, &ObjectAttributes,
&IoStatusBlock, NULL, FILE_ATTRIBUTE_NORMAL, FILE_SHARE_READ, FILE_OPEN,
FILE_NON_DIRECTORY_FILE | FILE_DELETE_ON_CLOSE, NULL, 0);
+    ok_eq_hex(Status, STATUS_SUCCESS);
+    ok_eq_ulongptr(IoStatusBlock.Information, FILE_OPENED);
+    ok(FileHandle1 != NULL, "FileHandle1 is NULL\n");
+    CheckObject(FileHandle1, 2L, 1L);
+
+    InitializeObjectAttributes(&ObjectAttributes, &FileName2,
OBJ_CASE_INSENSITIVE, NULL, NULL);
+    Status = ZwCreateFile(&FileHandle2, GENERIC_READ, &ObjectAttributes,
&IoStatusBlock, NULL, FILE_ATTRIBUTE_NORMAL, FILE_SHARE_READ, FILE_OPEN,
FILE_NON_DIRECTORY_FILE, NULL, 0);
+    ok_eq_hex(Status, STATUS_SUCCESS);
+    ok_eq_ulongptr(IoStatusBlock.Information, FILE_OPENED);
+    ok(FileHandle2 != NULL, "FileHandle2 is NULL\n");
+
+    if (!skip(Status == STATUS_SUCCESS && FileHandle1 != NULL, "Failed to
open file 1\n"))
+    {
+        Status = ObReferenceObjectByHandle(FileHandle1, FILE_READ_DATA | FILE_WRITE_DATA,
IoFileObjectType, KernelMode, (PVOID *)&FileObject1, NULL);
+        ok_eq_hex(Status, STATUS_SUCCESS);
+        ok(FileObject1 != NULL, "FileObject1 is NULL\n");
+        CheckObject(FileHandle1, 3L, 1L);
+    }
+
+    if (!skip(Status == STATUS_SUCCESS && FileHandle2 != NULL, "Failed to
open file 2\n"))
+    {
+        Status = ObReferenceObjectByHandle(FileHandle2, FILE_READ_DATA | FILE_WRITE_DATA,
IoFileObjectType, KernelMode, (PVOID *)&FileObject2, NULL);
+        ok_eq_hex(Status, STATUS_SUCCESS);
+        ok(FileObject2 != NULL, "FileObject2 is NULL\n");
+    }
+
+    trace("FileHandle1=%p, FileObject1=%p\n", FileHandle1, FileObject1);
+    trace("FileHandle2=%p, FileObject2=%p\n", FileHandle2, FileObject2);
+    TestCreateSection(FileHandle1, FileObject1, FileHandle2, FileObject2);
+
+    if (FileObject2)
+        ObDereferenceObject(FileObject2);
+    if (FileObject1)
+        ObDereferenceObject(FileObject1);
+    if (FileHandle2)
+        ZwClose(FileHandle2);
+    if (FileHandle1)
+        ZwClose(FileHandle1);
+}