mirror of
https://github.com/reactos/reactos.git
synced 2025-07-31 09:11:42 +00:00
[KMTESTS/MM]
- Add some tests for MmCreateSection involving file objects svn path=/trunk/; revision=53785
This commit is contained in:
Thomas Faber
parent
1f95a6dd13
commit
b41af56f6e
1 changed files with 239 additions and 1 deletions
|
|
@ -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 @@ START_TEST(MmSection)
|
|||
if (SectionObject && SectionObject != InvalidPointer)
|
||||
ObDereferenceObject(SectionObject);
|
||||
|
||||
/* page file section */
|
||||
SectionObject = InvalidPointer;
|
||||
MaximumSize.QuadPart = 1;
|
||||
StartSeh()
|
||||
|
|
@ -113,4 +131,224 @@ START_TEST(MmSection)
|
|||
|
||||
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);
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue