[ros-diffs] 01/01: [KMTESTS:CC] Introduce a macro to test public BCB

[Pierre Schweitzer]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=3d68c016640c58a49d9e66...

commit 3d68c016640c58a49d9e660de47c57f35a01f7c9
Author:     Pierre Schweitzer [e-mail]
AuthorDate: Sun Oct 28 22:37:59 2018 +0100
Commit:     Pierre Schweitzer [e-mail]
CommitDate: Sun Oct 28 22:37:59 2018 +0100

[KMTESTS:CC] Introduce a macro to test public BCB

For now, it's local, but it may be moved to some header
    to be shared between tests.
---
 modules/rostests/kmtests/ntos_cc/CcPinRead_drv.c | 29 ++++++++++++++++--------
 1 file changed, 19 insertions(+), 10 deletions(-)

diff --git a/modules/rostests/kmtests/ntos_cc/CcPinRead_drv.c b/modules/rostests/kmtests/ntos_cc/CcPinRead_drv.c
index 369d105742..c242f442a2 100644
--- a/modules/rostests/kmtests/ntos_cc/CcPinRead_drv.c
+++ b/modules/rostests/kmtests/ntos_cc/CcPinRead_drv.c
@@ -144,6 +144,15 @@ MapAndLockUserBuffer(
     return MmGetSystemAddressForMdlSafe(Irp->MdlAddress, NormalPagePriority);
 }

+#define ok_bcb(B, L, O)                                                                 \
+{                                                                                       \
+    PPUBLIC_BCB public_bcb = (B);                                                       \
+    ok(public_bcb->NodeTypeCode == 0x2FD, "Not a BCB: %x\n", public_bcb->NodeTypeCode); \
+    ok(public_bcb->NodeByteSize == 0, "Invalid size: %d\n", public_bcb->NodeByteSize);  \
+    ok_eq_ulong(public_bcb->MappedLength, (L));                                         \
+    ok_eq_longlong(public_bcb->MappedFileOffset.QuadPart, (O));                         \
+}
+
 static
 VOID
 NTAPI
@@ -172,7 +181,7 @@ PinInAnotherThread(IN PVOID Context)

if (!skip(Ret == TRUE, "CcPinRead failed\n"))
     {
-        ok(*(PUSHORT)Bcb == 0x2FD, "Not a BCB: %x\n", *(PUSHORT)Bcb);
+        ok_bcb(Bcb, 12288, Offset.QuadPart);
         ok_eq_pointer(Bcb, TestContext->Bcb);
         ok_eq_pointer(Buffer, TestContext->Buffer);

@@ -185,7 +194,7 @@ PinInAnotherThread(IN PVOID Context)

if (!skip(Ret == TRUE, "CcPinRead failed\n"))
     {
-        ok(*(PUSHORT)Bcb == 0x2FD, "Not a BCB: %x\n", *(PUSHORT)Bcb);
+        ok_bcb(Bcb, 12288, Offset.QuadPart);
         ok_eq_pointer(Bcb, TestContext->Bcb);
         ok_eq_pointer(Buffer, TestContext->Buffer);

@@ -198,7 +207,7 @@ PinInAnotherThread(IN PVOID Context)

if (!skip(Ret == TRUE, "CcPinRead failed\n"))
     {
-        ok(*(PUSHORT)Bcb == 0x2FD, "Not a BCB: %x\n", *(PUSHORT)Bcb);
+        ok_bcb(Bcb, 12288, Offset.QuadPart);
         ok_eq_pointer(Bcb, TestContext->Bcb);
         ok_eq_pointer(Buffer, TestContext->Buffer);

@@ -226,7 +235,7 @@ PinInAnotherThread(IN PVOID Context)

if (!skip(Ret == TRUE, "CcPinRead failed\n"))
     {
-        ok(*(PUSHORT)Bcb == 0x2FD, "Not a BCB: %x\n", *(PUSHORT)Bcb);
+        ok_bcb(Bcb, 12288, 4096);
         ok_eq_pointer(Bcb, TestContext->Bcb);
         ok_eq_pointer(Buffer, (PVOID)((ULONG_PTR)TestContext->Buffer + 0x500));

@@ -239,7 +248,7 @@ PinInAnotherThread(IN PVOID Context)

if (!skip(Ret == TRUE, "CcPinRead failed\n"))
     {
-        ok(*(PUSHORT)Bcb == 0x2FD, "Not a BCB: %x\n", *(PUSHORT)Bcb);
+        ok_bcb(Bcb, 12288, 4096);
         ok_eq_pointer(Bcb, TestContext->Bcb);
         ok_eq_pointer(Buffer, (PVOID)((ULONG_PTR)TestContext->Buffer + 0x500));

@@ -252,7 +261,7 @@ PinInAnotherThread(IN PVOID Context)

if (!skip(Ret == TRUE, "CcPinRead failed\n"))
     {
-        ok(*(PUSHORT)Bcb == 0x2FD, "Not a BCB: %x\n", *(PUSHORT)Bcb);
+        ok_bcb(Bcb, 12288, 4096);
         ok_eq_pointer(Bcb, TestContext->Bcb);
         ok_eq_pointer(Buffer, (PVOID)((ULONG_PTR)TestContext->Buffer + 0x500));

@@ -393,7 +402,7 @@ PerformTest(

if (!skip(Ret == TRUE, "CcPinRead failed\n"))
                     {
-                        ok(*(PUSHORT)Bcb == 0x2FD, "Not a BCB: %x\n", *(PUSHORT)Bcb);
+                        ok_bcb(Bcb, ((4 - TestId) * 4096), Offset.QuadPart);
                         ok_eq_ulong(Buffer[(0x3000 - TestId * 0x1000) / sizeof(ULONG)], 0xDEADBABE);

CcUnpinData(Bcb);
@@ -427,7 +436,7 @@ PerformTest(
                         {
                             PKTHREAD ThreadHandle;

-                            ok(*(PUSHORT)TestContext->Bcb == 0x2FD, "Not a BCB: %x\n", *(PUSHORT)TestContext->Bcb);
+                            ok_bcb(TestContext->Bcb, 12288, Offset.QuadPart);

#ifdef _X86_
                             /* FIXME: Should be fixed, will fail under certains conditions */
@@ -460,7 +469,7 @@ PerformTest(
                         {
                             PKTHREAD ThreadHandle;

-                            ok(*(PUSHORT)TestContext->Bcb == 0x2FD, "Not a BCB: %x\n", *(PUSHORT)TestContext->Bcb);
+                            ok_bcb(TestContext->Bcb, 12288, Offset.QuadPart);

TestContext->Length = FileSizes.FileSize.QuadPart - Offset.QuadPart;
                             ThreadHandle = KmtStartThread(PinInAnotherThreadExclusive, TestContext);
@@ -482,7 +491,7 @@ PerformTest(

if (!skip(Ret == TRUE, "CcPinRead failed\n"))
                     {
-                        ok(*(PUSHORT)Bcb == 0x2FD, "Not a BCB: %x\n", *(PUSHORT)Bcb);
+                        ok_bcb(Bcb, 12288, Offset.QuadPart);
                         ok_eq_ulong(Buffer[0x2000 / sizeof(ULONG)], 0);

CcUnpinData(Bcb);