https://git.reactos.org/?p=reactos.git;a=commitdiff;h=45231fbbc55a91d67d4ebeec990edf9d9118355c
commit 45231fbbc55a91d67d4ebeec990edf9d9118355c
Author: Pierre Schweitzer <pie...@reactos.org>
AuthorDate: Sat Oct 13 14:23:38 2018 +0200
Commit: Pierre Schweitzer <pie...@reactos.org>
CommitDate: Sat Oct 13 14:24:52 2018 +0200
[KMTESTS:CC] Add a test showing our CcPinMappedData() implementation is
broken
---
modules/rostests/kmtests/CMakeLists.txt | 1 +
modules/rostests/kmtests/kmtest/testlist.c | 2 +
modules/rostests/kmtests/ntos_cc/CMakeLists.txt | 15 +
.../rostests/kmtests/ntos_cc/CcPinMappedData_drv.c | 366 +++++++++++++++++++++
.../kmtests/ntos_cc/CcPinMappedData_user.c | 32 ++
5 files changed, 416 insertions(+)
diff --git a/modules/rostests/kmtests/CMakeLists.txt
b/modules/rostests/kmtests/CMakeLists.txt
index 51407959c4..623188e913 100644
--- a/modules/rostests/kmtests/CMakeLists.txt
+++ b/modules/rostests/kmtests/CMakeLists.txt
@@ -137,6 +137,7 @@ list(APPEND KMTEST_SOURCE
kernel32/FindFile_user.c
ntos_cc/CcCopyRead_user.c
ntos_cc/CcMapData_user.c
+ ntos_cc/CcPinMappedData_user.c
ntos_cc/CcPinRead_user.c
ntos_io/IoCreateFile_user.c
ntos_io/IoDeviceObject_user.c
diff --git a/modules/rostests/kmtests/kmtest/testlist.c
b/modules/rostests/kmtests/kmtest/testlist.c
index a0ede71368..5b5b65364b 100644
--- a/modules/rostests/kmtests/kmtest/testlist.c
+++ b/modules/rostests/kmtests/kmtest/testlist.c
@@ -8,6 +8,7 @@
KMT_TESTFUNC Test_CcCopyRead;
KMT_TESTFUNC Test_CcMapData;
+KMT_TESTFUNC Test_CcPinMappedData;
KMT_TESTFUNC Test_CcPinRead;
KMT_TESTFUNC Test_Example;
KMT_TESTFUNC Test_FileAttributes;
@@ -38,6 +39,7 @@ const KMT_TEST TestList[] =
{
{ "CcCopyRead", Test_CcCopyRead },
{ "CcMapData", Test_CcMapData },
+ { "CcPinMappedData", Test_CcPinMappedData },
{ "CcPinRead", Test_CcPinRead },
{ "-Example", Test_Example },
{ "FileAttributes", Test_FileAttributes },
diff --git a/modules/rostests/kmtests/ntos_cc/CMakeLists.txt
b/modules/rostests/kmtests/ntos_cc/CMakeLists.txt
index 83d2317a75..3357f19e68 100644
--- a/modules/rostests/kmtests/ntos_cc/CMakeLists.txt
+++ b/modules/rostests/kmtests/ntos_cc/CMakeLists.txt
@@ -31,6 +31,21 @@ add_target_compile_definitions(ccmapdata_drv
KMT_STANDALONE_DRIVER)
#add_pch(ccmapdata_drv ../include/kmt_test.h)
add_rostests_file(TARGET ccmapdata_drv)
+#
+# CcPinMappedData
+#
+list(APPEND CCPINMAPPEDDATA_DRV_SOURCE
+ ../kmtest_drv/kmtest_standalone.c
+ CcPinMappedData_drv.c)
+
+add_library(ccpinmappeddata_drv SHARED ${CCPINMAPPEDDATA_DRV_SOURCE})
+set_module_type(ccpinmappeddata_drv kernelmodedriver)
+target_link_libraries(ccpinmappeddata_drv kmtest_printf ${PSEH_LIB})
+add_importlibs(ccpinmappeddata_drv ntoskrnl hal)
+add_target_compile_definitions(ccpinmappeddata_drv KMT_STANDALONE_DRIVER)
+#add_pch(ccpinmappeddata_drv ../include/kmt_test.h)
+add_rostests_file(TARGET ccpinmappeddata_drv)
+
#
# CcPinRead
#
diff --git a/modules/rostests/kmtests/ntos_cc/CcPinMappedData_drv.c
b/modules/rostests/kmtests/ntos_cc/CcPinMappedData_drv.c
new file mode 100644
index 0000000000..5f52301cce
--- /dev/null
+++ b/modules/rostests/kmtests/ntos_cc/CcPinMappedData_drv.c
@@ -0,0 +1,366 @@
+/*
+ * PROJECT: ReactOS kernel-mode tests
+ * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
+ * PURPOSE: Test driver for CcPinMappedData function
+ * PROGRAMMER: Pierre Schweitzer <pie...@reactos.org>
+ */
+
+#include <kmt_test.h>
+
+#define NDEBUG
+#include <debug.h>
+
+#define IOCTL_START_TEST 1
+#define IOCTL_FINISH_TEST 2
+
+typedef struct _TEST_FCB
+{
+ FSRTL_ADVANCED_FCB_HEADER Header;
+ SECTION_OBJECT_POINTERS SectionObjectPointers;
+ FAST_MUTEX HeaderMutex;
+} TEST_FCB, *PTEST_FCB;
+
+typedef struct _TEST_CONTEXT
+{
+ PVOID Bcb;
+ PVOID Buffer;
+ ULONG Length;
+} TEST_CONTEXT, *PTEST_CONTEXT;
+
+static ULONG TestTestId = -1;
+static PFILE_OBJECT TestFileObject;
+static PDEVICE_OBJECT TestDeviceObject;
+static KMT_IRP_HANDLER TestIrpHandler;
+static KMT_MESSAGE_HANDLER TestMessageHandler;
+
+NTSTATUS
+TestEntry(
+ _In_ PDRIVER_OBJECT DriverObject,
+ _In_ PCUNICODE_STRING RegistryPath,
+ _Out_ PCWSTR *DeviceName,
+ _Inout_ INT *Flags)
+{
+ NTSTATUS Status = STATUS_SUCCESS;
+
+ PAGED_CODE();
+
+ UNREFERENCED_PARAMETER(RegistryPath);
+
+ *DeviceName = L"CcPinMappedData";
+ *Flags = TESTENTRY_NO_EXCLUSIVE_DEVICE |
+ TESTENTRY_BUFFERED_IO_DEVICE |
+ TESTENTRY_NO_READONLY_DEVICE;
+
+ KmtRegisterIrpHandler(IRP_MJ_READ, NULL, TestIrpHandler);
+ KmtRegisterMessageHandler(0, NULL, TestMessageHandler);
+
+
+ return Status;
+}
+
+VOID
+TestUnload(
+ _In_ PDRIVER_OBJECT DriverObject)
+{
+ PAGED_CODE();
+}
+
+BOOLEAN
+NTAPI
+AcquireForLazyWrite(
+ _In_ PVOID Context,
+ _In_ BOOLEAN Wait)
+{
+ return TRUE;
+}
+
+VOID
+NTAPI
+ReleaseFromLazyWrite(
+ _In_ PVOID Context)
+{
+ return;
+}
+
+BOOLEAN
+NTAPI
+AcquireForReadAhead(
+ _In_ PVOID Context,
+ _In_ BOOLEAN Wait)
+{
+ return TRUE;
+}
+
+VOID
+NTAPI
+ReleaseFromReadAhead(
+ _In_ PVOID Context)
+{
+ return;
+}
+
+static CACHE_MANAGER_CALLBACKS Callbacks = {
+ AcquireForLazyWrite,
+ ReleaseFromLazyWrite,
+ AcquireForReadAhead,
+ ReleaseFromReadAhead,
+};
+
+static CC_FILE_SIZES FileSizes = {
+ RTL_CONSTANT_LARGE_INTEGER((LONGLONG)0x4000), // .AllocationSize
+ RTL_CONSTANT_LARGE_INTEGER((LONGLONG)0x4000), // .FileSize
+ RTL_CONSTANT_LARGE_INTEGER((LONGLONG)0x4000) // .ValidDataLength
+};
+
+static
+PVOID
+MapAndLockUserBuffer(
+ _In_ _Out_ PIRP Irp,
+ _In_ ULONG BufferLength)
+{
+ PMDL Mdl;
+
+ if (Irp->MdlAddress == NULL)
+ {
+ Mdl = IoAllocateMdl(Irp->UserBuffer, BufferLength, FALSE, FALSE, Irp);
+ if (Mdl == NULL)
+ {
+ return NULL;
+ }
+
+ _SEH2_TRY
+ {
+ MmProbeAndLockPages(Mdl, Irp->RequestorMode, IoWriteAccess);
+ }
+ _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
+ {
+ IoFreeMdl(Mdl);
+ Irp->MdlAddress = NULL;
+ _SEH2_YIELD(return NULL);
+ }
+ _SEH2_END;
+ }
+
+ return MmGetSystemAddressForMdlSafe(Irp->MdlAddress, NormalPagePriority);
+}
+
+static
+VOID
+PerformTest(
+ ULONG TestId,
+ PDEVICE_OBJECT DeviceObject)
+{
+ PVOID Bcb, PinBcb;
+ BOOLEAN Ret;
+ PULONG Buffer;
+ PTEST_FCB Fcb;
+ LARGE_INTEGER Offset;
+
+ ok_eq_pointer(TestFileObject, NULL);
+ ok_eq_pointer(TestDeviceObject, NULL);
+ ok_eq_ulong(TestTestId, -1);
+
+ TestDeviceObject = DeviceObject;
+ TestTestId = TestId;
+ TestFileObject = IoCreateStreamFileObject(NULL, DeviceObject);
+ if (!skip(TestFileObject != NULL, "Failed to allocate FO\n"))
+ {
+ Fcb = ExAllocatePool(NonPagedPool, sizeof(TEST_FCB));
+ if (!skip(Fcb != NULL, "ExAllocatePool failed\n"))
+ {
+ RtlZeroMemory(Fcb, sizeof(TEST_FCB));
+ ExInitializeFastMutex(&Fcb->HeaderMutex);
+ FsRtlSetupAdvancedHeader(&Fcb->Header, &Fcb->HeaderMutex);
+
+ TestFileObject->FsContext = Fcb;
+ TestFileObject->SectionObjectPointer = &Fcb->SectionObjectPointers;
+
+ KmtStartSeh();
+ CcInitializeCacheMap(TestFileObject, &FileSizes, TRUE, &Callbacks,
NULL);
+ KmtEndSeh(STATUS_SUCCESS);
+
+ if (!skip(CcIsFileCached(TestFileObject) == TRUE,
"CcInitializeCacheMap failed\n"))
+ {
+ if (TestId == 0)
+ {
+ Ret = FALSE;
+ Offset.QuadPart = TestId * 0x1000;
+ KmtStartSeh();
+ Ret = CcMapData(TestFileObject, &Offset,
FileSizes.FileSize.QuadPart - Offset.QuadPart, MAP_WAIT, &Bcb, (PVOID
*)&Buffer);
+ KmtEndSeh(STATUS_SUCCESS);
+
+ if (!skip(Ret == TRUE, "CcMapData failed\n"))
+ {
+ Ret = FALSE;
+ PinBcb = Bcb;
+ ok_eq_ulong(Buffer[0x3000 / sizeof(ULONG)],
0xDEADBABE);
+
+ KmtStartSeh();
+ Ret = CcPinMappedData(TestFileObject, &Offset,
FileSizes.FileSize.QuadPart - Offset.QuadPart, PIN_WAIT, &PinBcb);
+ KmtEndSeh(STATUS_SUCCESS);
+
+ if (!skip(Ret == TRUE, "CcPinMappedData failed\n"))
+ {
+ ok(Bcb != PinBcb, "Returned same BCB!\n");
+ ok(*(PUSHORT)PinBcb == 0x2FD, "Not a BCB: %x\n",
*(PUSHORT)PinBcb);
+
+ /* Previous BCB isn't valid anymore! */
+ Bcb = PinBcb;
+ }
+
+ CcUnpinData(Bcb);
+ }
+ }
+ }
+ }
+ }
+}
+
+
+static
+VOID
+CleanupTest(
+ ULONG TestId,
+ PDEVICE_OBJECT DeviceObject)
+{
+ LARGE_INTEGER Zero = RTL_CONSTANT_LARGE_INTEGER(0LL);
+ CACHE_UNINITIALIZE_EVENT CacheUninitEvent;
+
+ ok_eq_pointer(TestDeviceObject, DeviceObject);
+ ok_eq_ulong(TestTestId, TestId);
+
+ if (!skip(TestFileObject != NULL, "No test FO\n"))
+ {
+ if (CcIsFileCached(TestFileObject))
+ {
+ KeInitializeEvent(&CacheUninitEvent.Event, NotificationEvent,
FALSE);
+ CcUninitializeCacheMap(TestFileObject, &Zero, &CacheUninitEvent);
+ KeWaitForSingleObject(&CacheUninitEvent.Event, Executive,
KernelMode, FALSE, NULL);
+ }
+
+ if (TestFileObject->FsContext != NULL)
+ {
+ ExFreePool(TestFileObject->FsContext);
+ TestFileObject->FsContext = NULL;
+ TestFileObject->SectionObjectPointer = NULL;
+ }
+
+ ObDereferenceObject(TestFileObject);
+ }
+
+ TestFileObject = NULL;
+ TestDeviceObject = NULL;
+ TestTestId = -1;
+}
+
+
+static
+NTSTATUS
+TestMessageHandler(
+ _In_ PDEVICE_OBJECT DeviceObject,
+ _In_ ULONG ControlCode,
+ _In_opt_ PVOID Buffer,
+ _In_ SIZE_T InLength,
+ _Inout_ PSIZE_T OutLength)
+{
+ NTSTATUS Status = STATUS_SUCCESS;
+
+ FsRtlEnterFileSystem();
+
+ switch (ControlCode)
+ {
+ case IOCTL_START_TEST:
+ ok_eq_ulong((ULONG)InLength, sizeof(ULONG));
+ PerformTest(*(PULONG)Buffer, DeviceObject);
+ break;
+
+ case IOCTL_FINISH_TEST:
+ ok_eq_ulong((ULONG)InLength, sizeof(ULONG));
+ CleanupTest(*(PULONG)Buffer, DeviceObject);
+ break;
+
+ default:
+ Status = STATUS_NOT_IMPLEMENTED;
+ break;
+ }
+
+ FsRtlExitFileSystem();
+
+ return Status;
+}
+
+static
+NTSTATUS
+TestIrpHandler(
+ _In_ PDEVICE_OBJECT DeviceObject,
+ _In_ PIRP Irp,
+ _In_ PIO_STACK_LOCATION IoStack)
+{
+ NTSTATUS Status;
+
+ PAGED_CODE();
+
+ DPRINT("IRP %x/%x\n", IoStack->MajorFunction, IoStack->MinorFunction);
+ ASSERT(IoStack->MajorFunction == IRP_MJ_READ);
+
+ FsRtlEnterFileSystem();
+
+ Status = STATUS_NOT_SUPPORTED;
+ Irp->IoStatus.Information = 0;
+
+ if (IoStack->MajorFunction == IRP_MJ_READ)
+ {
+ PMDL Mdl;
+ ULONG Length;
+ PVOID Buffer;
+ LARGE_INTEGER Offset;
+
+ Offset = IoStack->Parameters.Read.ByteOffset;
+ Length = IoStack->Parameters.Read.Length;
+
+ ok_eq_pointer(DeviceObject, TestDeviceObject);
+ ok_eq_pointer(IoStack->FileObject, TestFileObject);
+
+ ok(FlagOn(Irp->Flags, IRP_NOCACHE), "Not coming from Cc\n");
+
+ ok_irql(APC_LEVEL);
+ ok((Offset.QuadPart % PAGE_SIZE == 0 || Offset.QuadPart == 0), "Offset
is not aligned: %I64i\n", Offset.QuadPart);
+ ok(Length % PAGE_SIZE == 0, "Length is not aligned: %I64i\n", Length);
+
+ ok(Irp->AssociatedIrp.SystemBuffer == NULL, "A SystemBuffer was
allocated!\n");
+ Buffer = MapAndLockUserBuffer(Irp, Length);
+ ok(Buffer != NULL, "Null pointer!\n");
+ RtlFillMemory(Buffer, Length, 0xBA);
+
+ Status = STATUS_SUCCESS;
+ if (Offset.QuadPart <= 0x3000 && Offset.QuadPart + Length > 0x3000)
+ {
+ *(PULONG)((ULONG_PTR)Buffer + (ULONG_PTR)(0x3000 -
Offset.QuadPart)) = 0xDEADBABE;
+ }
+
+ Mdl = Irp->MdlAddress;
+ ok(Mdl != NULL, "Null pointer for MDL!\n");
+ ok((Mdl->MdlFlags & MDL_PAGES_LOCKED) != 0, "MDL not locked\n");
+ ok((Mdl->MdlFlags & MDL_SOURCE_IS_NONPAGED_POOL) == 0, "MDL from non
paged\n");
+ ok((Mdl->MdlFlags & MDL_IO_PAGE_READ) != 0, "Non paging IO\n");
+ ok((Irp->Flags & IRP_PAGING_IO) != 0, "Non paging IO\n");
+
+ Irp->IoStatus.Information = Length;
+ }
+
+ if (Status == STATUS_PENDING)
+ {
+ IoMarkIrpPending(Irp);
+ IoCompleteRequest(Irp, IO_NO_INCREMENT);
+ Status = STATUS_PENDING;
+ }
+ else
+ {
+ Irp->IoStatus.Status = Status;
+ IoCompleteRequest(Irp, IO_NO_INCREMENT);
+ }
+
+ FsRtlExitFileSystem();
+
+ return Status;
+}
diff --git a/modules/rostests/kmtests/ntos_cc/CcPinMappedData_user.c
b/modules/rostests/kmtests/ntos_cc/CcPinMappedData_user.c
new file mode 100644
index 0000000000..df3cc320a2
--- /dev/null
+++ b/modules/rostests/kmtests/ntos_cc/CcPinMappedData_user.c
@@ -0,0 +1,32 @@
+/*
+ * PROJECT: ReactOS kernel-mode tests
+ * LICENSE: GPLv2+ - See COPYING in the top level directory
+ * PURPOSE: Kernel-Mode Test Suite CcPinMappedData test user-mode part
+ * PROGRAMMER: Pierre Schweitzer <pie...@reactos.org>
+ */
+
+#include <kmt_test.h>
+
+#define IOCTL_START_TEST 1
+#define IOCTL_FINISH_TEST 2
+
+START_TEST(CcPinMappedData)
+{
+ DWORD Ret;
+ ULONG TestId;
+
+ KmtLoadDriver(L"CcPinMappedData", FALSE);
+ KmtOpenDriver();
+
+ /* 1 basic test */
+ for (TestId = 0; TestId < 1; ++TestId)
+ {
+ Ret = KmtSendUlongToDriver(IOCTL_START_TEST, TestId);
+ ok(Ret == ERROR_SUCCESS, "KmtSendUlongToDriver failed: %lx\n", Ret);
+ Ret = KmtSendUlongToDriver(IOCTL_FINISH_TEST, TestId);
+ ok(Ret == ERROR_SUCCESS, "KmtSendUlongToDriver failed: %lx\n", Ret);
+ }
+
+ KmtCloseDriver();
+ KmtUnloadDriver();
+}
