[KMTESTS:CC] Add tests for BCB extension

This commit is contained in:
Pierre Schweitzer 2018-09-01 21:52:39 +02:00
parent 04a192fa30
commit 9ab21d67ae
No known key found for this signature in database
GPG key ID: 7545556C3D585B0B
2 changed files with 89 additions and 0 deletions

View file

@ -212,6 +212,21 @@ MapInAnotherThread(IN PVOID Context)
CcUnpinData(Bcb);
}
Offset.QuadPart = 0x1500;
TestContext->Length -= 0x500;
KmtStartSeh();
Ret = CcMapData(TestFileObject, &Offset, TestContext->Length, MAP_WAIT, &Bcb, (PVOID *)&Buffer);
KmtEndSeh(STATUS_SUCCESS);
if (!skip(Ret == TRUE, "CcMapData failed\n"))
{
ok_eq_pointer(Bcb, TestContext->Bcb);
ok_eq_pointer(Buffer, (PVOID)((ULONG_PTR)TestContext->Buffer + 0x500));
CcUnpinData(Bcb);
}
return;
}

View file

@ -214,6 +214,45 @@ PinInAnotherThread(IN PVOID Context)
CcUnpinData(Bcb);
}
Offset.QuadPart = 0x1500;
TestContext->Length -= 0x500;
KmtStartSeh();
Ret = CcPinRead(TestFileObject, &Offset, TestContext->Length, PIN_WAIT | PIN_IF_BCB, &Bcb, (PVOID *)&Buffer);
KmtEndSeh(STATUS_SUCCESS);
if (!skip(Ret == TRUE, "CcPinRead failed\n"))
{
ok_eq_pointer(Bcb, TestContext->Bcb);
ok_eq_pointer(Buffer, (PVOID)((ULONG_PTR)TestContext->Buffer + 0x500));
CcUnpinData(Bcb);
}
KmtStartSeh();
Ret = CcPinRead(TestFileObject, &Offset, TestContext->Length, PIN_WAIT, &Bcb, (PVOID *)&Buffer);
KmtEndSeh(STATUS_SUCCESS);
if (!skip(Ret == TRUE, "CcPinRead failed\n"))
{
ok_eq_pointer(Bcb, TestContext->Bcb);
ok_eq_pointer(Buffer, (PVOID)((ULONG_PTR)TestContext->Buffer + 0x500));
CcUnpinData(Bcb);
}
KmtStartSeh();
Ret = CcPinRead(TestFileObject, &Offset, TestContext->Length, PIN_EXCLUSIVE, &Bcb, (PVOID *)&Buffer);
KmtEndSeh(STATUS_SUCCESS);
if (!skip(Ret == TRUE, "CcPinRead failed\n"))
{
ok_eq_pointer(Bcb, TestContext->Bcb);
ok_eq_pointer(Buffer, (PVOID)((ULONG_PTR)TestContext->Buffer + 0x500));
CcUnpinData(Bcb);
}
return;
}
@ -261,6 +300,41 @@ PinInAnotherThreadExclusive(IN PVOID Context)
CcUnpinData(Bcb);
}
Offset.QuadPart = 0x1500;
TestContext->Length -= 0x500;
KmtStartSeh();
Ret = CcPinRead(TestFileObject, &Offset, TestContext->Length, PIN_IF_BCB, &Bcb, (PVOID *)&Buffer);
KmtEndSeh(STATUS_SUCCESS);
ok(Ret == FALSE, "CcPinRead succeed\n");
if (Ret)
{
CcUnpinData(Bcb);
}
KmtStartSeh();
Ret = CcPinRead(TestFileObject, &Offset, TestContext->Length, 0, &Bcb, (PVOID *)&Buffer);
KmtEndSeh(STATUS_SUCCESS);
ok(Ret == FALSE, "CcPinRead succeed\n");
if (Ret)
{
CcUnpinData(Bcb);
}
KmtStartSeh();
Ret = CcMapData(TestFileObject, &Offset, TestContext->Length, 0, &Bcb, (PVOID *)&Buffer);
KmtEndSeh(STATUS_SUCCESS);
if (!skip(Ret == TRUE, "CcMapData failed\n"))
{
ok(Bcb != TestContext->Bcb, "Returned same BCB!\n");
ok_eq_pointer(Buffer, (PVOID)((ULONG_PTR)TestContext->Buffer + 0x500));
CcUnpinData(Bcb);
}
return;
}