https://git.reactos.org/?p=reactos.git;a=commitdiff;h=7375173b7fc050ce36ed5796cf4d46530af87379
commit 7375173b7fc050ce36ed5796cf4d46530af87379
Author: Pierre Schweitzer <pie...@reactos.org>
AuthorDate: Mon Dec 17 20:29:31 2018 +0100
Commit: Pierre Schweitzer <pie...@reactos.org>
CommitDate: Mon Dec 17 20:31:24 2018 +0100
[KMTESTS:CC] Add some tests regarding small pinning size (< PAGE_SIZE)
CORE-15384
---
modules/rostests/kmtests/ntos_cc/CcPinRead_drv.c | 84 ++++++++++++++++++++++-
modules/rostests/kmtests/ntos_cc/CcPinRead_user.c | 3 +-
2 files changed, 84 insertions(+), 3 deletions(-)
diff --git a/modules/rostests/kmtests/ntos_cc/CcPinRead_drv.c
b/modules/rostests/kmtests/ntos_cc/CcPinRead_drv.c
index c242f442a2..d6edb96d15 100644
--- a/modules/rostests/kmtests/ntos_cc/CcPinRead_drv.c
+++ b/modules/rostests/kmtests/ntos_cc/CcPinRead_drv.c
@@ -32,6 +32,7 @@ static PFILE_OBJECT TestFileObject;
static PDEVICE_OBJECT TestDeviceObject;
static KMT_IRP_HANDLER TestIrpHandler;
static KMT_MESSAGE_HANDLER TestMessageHandler;
+static BOOLEAN TestWriteCalled = FALSE;
NTSTATUS
TestEntry(
@@ -52,6 +53,7 @@ TestEntry(
TESTENTRY_NO_READONLY_DEVICE;
KmtRegisterIrpHandler(IRP_MJ_READ, NULL, TestIrpHandler);
+ KmtRegisterIrpHandler(IRP_MJ_WRITE, NULL, TestIrpHandler);
KmtRegisterMessageHandler(0, NULL, TestMessageHandler);
@@ -112,6 +114,12 @@ static CC_FILE_SIZES FileSizes = {
RTL_CONSTANT_LARGE_INTEGER((LONGLONG)0x4000) // .ValidDataLength
};
+static CC_FILE_SIZES SmallFileSizes = {
+ RTL_CONSTANT_LARGE_INTEGER((LONGLONG)512), // .AllocationSize
+ RTL_CONSTANT_LARGE_INTEGER((LONGLONG)496), // .FileSize
+ RTL_CONSTANT_LARGE_INTEGER((LONGLONG)496) // .ValidDataLength
+};
+
static
PVOID
MapAndLockUserBuffer(
@@ -387,7 +395,14 @@ PerformTest(
TestFileObject->SectionObjectPointer = &Fcb->SectionObjectPointers;
KmtStartSeh();
- CcInitializeCacheMap(TestFileObject, &FileSizes, PinAccess,
&Callbacks, NULL);
+ if (TestId < 6)
+ {
+ CcInitializeCacheMap(TestFileObject, &FileSizes, PinAccess,
&Callbacks, NULL);
+ }
+ else
+ {
+ CcInitializeCacheMap(TestFileObject, &SmallFileSizes,
PinAccess, &Callbacks, NULL);
+ }
KmtEndSeh(STATUS_SUCCESS);
if (!skip(CcIsFileCached(TestFileObject) == TRUE,
"CcInitializeCacheMap failed\n"))
@@ -541,6 +556,24 @@ PerformTest(
CcUnpinData(Bcb);
}
}
+ else if (TestId == 6)
+ {
+ Ret = FALSE;
+ Offset.QuadPart = 0;
+
+ KmtStartSeh();
+ Ret = CcPinRead(TestFileObject, &Offset,
FileSizes.FileSize.QuadPart, PIN_WAIT, &Bcb, (PVOID *)&Buffer);
+ KmtEndSeh(STATUS_SUCCESS);
+
+ if (!skip(Ret == TRUE, "CcPinRead failed\n"))
+ {
+ ok_bcb(Bcb, PAGE_SIZE * 4, Offset.QuadPart);
+ RtlFillMemory(Buffer, 0xbd,
FileSizes.FileSize.LowPart);
+ CcSetDirtyPinnedData(Bcb, NULL);
+
+ CcUnpinData(Bcb);
+ }
+ }
}
}
}
@@ -575,12 +608,22 @@ CleanupTest(
TestFileObject->SectionObjectPointer = NULL;
}
+ if (TestTestId == 6)
+ {
+ ok_bool_true(TestWriteCalled, "Write was not called!\n");
+ }
+ else
+ {
+ ok_bool_false(TestWriteCalled, "Write was unexpectedly called\n");
+ }
+
ObDereferenceObject(TestFileObject);
}
TestFileObject = NULL;
TestDeviceObject = NULL;
TestTestId = -1;
+ TestWriteCalled = FALSE;
}
@@ -631,7 +674,8 @@ TestIrpHandler(
PAGED_CODE();
DPRINT("IRP %x/%x\n", IoStack->MajorFunction, IoStack->MinorFunction);
- ASSERT(IoStack->MajorFunction == IRP_MJ_READ);
+ ASSERT(IoStack->MajorFunction == IRP_MJ_READ ||
+ IoStack->MajorFunction == IRP_MJ_WRITE);
FsRtlEnterFileSystem();
@@ -677,6 +721,42 @@ TestIrpHandler(
Irp->IoStatus.Information = Length;
}
+ else if (IoStack->MajorFunction == IRP_MJ_WRITE)
+ {
+ PMDL Mdl;
+ ULONG Length;
+ PVOID Buffer;
+ LARGE_INTEGER Offset;
+
+ Offset = IoStack->Parameters.Write.ByteOffset;
+ Length = IoStack->Parameters.Write.Length;
+
+ ok(TestTestId == 6, "Unexpected test id: %d\n", TestTestId);
+ ok_eq_pointer(DeviceObject, TestDeviceObject);
+ ok_eq_pointer(IoStack->FileObject, TestFileObject);
+
+ ok(FlagOn(Irp->Flags, IRP_NOCACHE), "Not coming from Cc\n");
+
+ ok_irql(PASSIVE_LEVEL);
+ ok(Offset.QuadPart == 0, "Offset is not null: %I64i\n",
Offset.QuadPart);
+ ok(Length % PAGE_SIZE == 0, "Length is not aligned: %I64i\n", Length);
+ ok(Length == PAGE_SIZE * 4, "Length is not MappedLength-sized:
%I64i\n", Length);
+
+ Buffer = MapAndLockUserBuffer(Irp, Length);
+ ok(Buffer != NULL, "Null pointer!\n");
+
+ 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((Irp->Flags & IRP_PAGING_IO) != 0, "Non paging IO\n");
+
+ ok_bool_false(TestWriteCalled, "Write has been unexpectedly called
twice!\n");
+ TestWriteCalled = TRUE;
+
+ Status = STATUS_SUCCESS;
+ Irp->IoStatus.Information = Length;
+ }
if (Status == STATUS_PENDING)
{
diff --git a/modules/rostests/kmtests/ntos_cc/CcPinRead_user.c
b/modules/rostests/kmtests/ntos_cc/CcPinRead_user.c
index e8e78cbc44..c0878435fe 100644
--- a/modules/rostests/kmtests/ntos_cc/CcPinRead_user.c
+++ b/modules/rostests/kmtests/ntos_cc/CcPinRead_user.c
@@ -22,8 +22,9 @@ START_TEST(CcPinRead)
* 1 test for BCB
* 1 test for pinning access
* 1 test for length/offset
+ * 1 test for read/write size
*/
- for (TestId = 0; TestId < 6; ++TestId)
+ for (TestId = 0; TestId < 7; ++TestId)
{
Ret = KmtSendUlongToDriver(IOCTL_START_TEST, TestId);
ok(Ret == ERROR_SUCCESS, "KmtSendUlongToDriver failed: %lx\n", Ret);
