mirror of
https://github.com/reactos/reactos.git
synced 2024-08-29 22:58:34 +00:00
b41af56f6e
- Add some tests for MmCreateSection involving file objects svn path=/trunk/; revision=53785
355 lines
15 KiB
C
355 lines
15 KiB
C
/*
|
|
* PROJECT: ReactOS kernel-mode tests
|
|
* LICENSE: GPLv2+ - See COPYING in the top level directory
|
|
* PURPOSE: Kernel-Mode Test Suite Section Object test
|
|
* PROGRAMMER: Thomas Faber <thfabba@gmx.de>
|
|
*/
|
|
|
|
#include <kmt_test.h>
|
|
|
|
#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)
|
|
|
|
#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);
|
|
EndSeh(STATUS_SUCCESS);
|
|
ok_eq_hex(Status, STATUS_INVALID_PAGE_PROTECTION);
|
|
|
|
if (!KmtIsCheckedBuild)
|
|
{
|
|
/* PAGE_NOACCESS and missing SEC_RESERVE/SEC_COMMIT/SEC_IMAGE assert */
|
|
StartSeh()
|
|
Status = MmCreateSection(NULL, 0, NULL, NULL, PAGE_NOACCESS, SEC_RESERVE, NULL, NULL);
|
|
EndSeh(STATUS_ACCESS_VIOLATION);
|
|
|
|
StartSeh()
|
|
Status = MmCreateSection(NULL, 0, NULL, NULL, PAGE_NOACCESS, 0, NULL, NULL);
|
|
EndSeh(STATUS_ACCESS_VIOLATION);
|
|
}
|
|
|
|
SectionObject = InvalidPointer;
|
|
StartSeh()
|
|
Status = MmCreateSection(&SectionObject, 0, NULL, NULL, 0, SEC_RESERVE, NULL, NULL);
|
|
EndSeh(STATUS_SUCCESS);
|
|
ok_eq_hex(Status, STATUS_INVALID_PAGE_PROTECTION);
|
|
ok_eq_pointer(SectionObject, InvalidPointer);
|
|
|
|
if (SectionObject && SectionObject != InvalidPointer)
|
|
ObDereferenceObject(SectionObject);
|
|
|
|
StartSeh()
|
|
Status = MmCreateSection(NULL, 0, NULL, NULL, PAGE_READONLY, SEC_RESERVE, NULL, NULL);
|
|
EndSeh(STATUS_ACCESS_VIOLATION);
|
|
|
|
SectionObject = InvalidPointer;
|
|
StartSeh()
|
|
Status = MmCreateSection(&SectionObject, 0, NULL, NULL, PAGE_READONLY, SEC_RESERVE, NULL, NULL);
|
|
EndSeh(STATUS_ACCESS_VIOLATION);
|
|
ok_eq_pointer(SectionObject, InvalidPointer);
|
|
|
|
if (SectionObject && SectionObject != InvalidPointer)
|
|
ObDereferenceObject(SectionObject);
|
|
|
|
SectionObject = InvalidPointer;
|
|
MaximumSize.QuadPart = 0;
|
|
StartSeh()
|
|
Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, NULL, NULL);
|
|
EndSeh(STATUS_SUCCESS);
|
|
ok_eq_hex(Status, STATUS_INVALID_FILE_FOR_SECTION);
|
|
ok_eq_longlong(MaximumSize.QuadPart, 0LL);
|
|
ok_eq_pointer(SectionObject, InvalidPointer);
|
|
|
|
if (SectionObject && SectionObject != InvalidPointer)
|
|
ObDereferenceObject(SectionObject);
|
|
|
|
MaximumSize.QuadPart = 0;
|
|
StartSeh()
|
|
Status = MmCreateSection(NULL, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_RESERVE, NULL, NULL);
|
|
EndSeh(STATUS_SUCCESS);
|
|
ok_eq_hex(Status, STATUS_INVALID_PARAMETER_4);
|
|
ok_eq_longlong(MaximumSize.QuadPart, 0LL);
|
|
|
|
if (SectionObject && SectionObject != InvalidPointer)
|
|
ObDereferenceObject(SectionObject);
|
|
|
|
MaximumSize.QuadPart = 1;
|
|
StartSeh()
|
|
Status = MmCreateSection(NULL, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_RESERVE, NULL, NULL);
|
|
EndSeh(STATUS_ACCESS_VIOLATION);
|
|
ok_eq_longlong(MaximumSize.QuadPart, 1LL);
|
|
|
|
if (SectionObject && SectionObject != InvalidPointer)
|
|
ObDereferenceObject(SectionObject);
|
|
|
|
SectionObject = InvalidPointer;
|
|
MaximumSize.QuadPart = 0;
|
|
StartSeh()
|
|
Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_RESERVE, NULL, NULL);
|
|
EndSeh(STATUS_SUCCESS);
|
|
ok_eq_hex(Status, STATUS_INVALID_PARAMETER_4);
|
|
ok_eq_longlong(MaximumSize.QuadPart, 0LL);
|
|
ok_eq_pointer(SectionObject, InvalidPointer);
|
|
|
|
if (SectionObject && SectionObject != InvalidPointer)
|
|
ObDereferenceObject(SectionObject);
|
|
|
|
/* page file section */
|
|
SectionObject = InvalidPointer;
|
|
MaximumSize.QuadPart = 1;
|
|
StartSeh()
|
|
Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_RESERVE, NULL, 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");
|
|
|
|
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);
|
|
}
|