CcPinRead_drv.c File Reference

#include <kmt_test.h>
#include <debug.h>

Include dependency graph for CcPinRead_drv.c:

Go to the source code of this file.

Classes
struct	_TEST_FCB
struct	_TEST_CONTEXT

Macros
#define	NDEBUG
#define	IOCTL_START_TEST   1
#define	IOCTL_FINISH_TEST   2
#define	ok_bcb(B, L, O)

Typedefs
typedef struct _TEST_FCB	TEST_FCB
typedef struct _TEST_FCB *	PTEST_FCB
typedef struct _TEST_CONTEXT	TEST_CONTEXT
typedef struct _TEST_CONTEXT *	PTEST_CONTEXT

Functions
NTSTATUS	TestEntry (_In_ PDRIVER_OBJECT DriverObject, _In_ PCUNICODE_STRING RegistryPath, _Out_ PCWSTR *DeviceName, _Inout_ INT *Flags)
VOID	TestUnload (_In_ PDRIVER_OBJECT DriverObject)
BOOLEAN NTAPI	AcquireForLazyWrite (_In_ PVOID Context, _In_ BOOLEAN Wait)
VOID NTAPI	ReleaseFromLazyWrite (_In_ PVOID Context)
BOOLEAN NTAPI	AcquireForReadAhead (_In_ PVOID Context, _In_ BOOLEAN Wait)
VOID NTAPI	ReleaseFromReadAhead (_In_ PVOID Context)
static PVOID	MapAndLockUserBuffer (_In_ _Out_ PIRP Irp, _In_ ULONG BufferLength)
static VOID NTAPI	PinInAnotherThread (IN PVOID Context)
static VOID NTAPI	PinInAnotherThreadExclusive (IN PVOID Context)
static VOID	PerformTest (ULONG TestId, PDEVICE_OBJECT DeviceObject)
static VOID	CleanupTest (ULONG TestId, PDEVICE_OBJECT DeviceObject)
static NTSTATUS	TestMessageHandler (_In_ PDEVICE_OBJECT DeviceObject, _In_ ULONG ControlCode, _In_opt_ PVOID Buffer, _In_ SIZE_T InLength, _Inout_ PSIZE_T OutLength)
static NTSTATUS	TestIrpHandler (_In_ PDEVICE_OBJECT DeviceObject, _In_ PIRP Irp, _In_ PIO_STACK_LOCATION IoStack)

Variables
static ULONG	TestTestId = -1
static PFILE_OBJECT	TestFileObject
static PDEVICE_OBJECT	TestDeviceObject
static KMT_IRP_HANDLER	TestIrpHandler
static KMT_MESSAGE_HANDLER	TestMessageHandler
static BOOLEAN	TestWriteCalled = FALSE
static ULONGLONG	Memory = 0
static BOOLEAN	TS = FALSE
LARGE_INTEGER	WriteOffset
ULONG	WriteLength
static CACHE_MANAGER_CALLBACKS	Callbacks
static CC_FILE_SIZES	FileSizes
static CC_FILE_SIZES	SmallFileSizes

Macro Definition Documentation

IOCTL_FINISH_TEST

#define IOCTL_FINISH_TEST   2

Definition at line 14 of file CcPinRead_drv.c.

IOCTL_START_TEST

#define IOCTL_START_TEST   1

Definition at line 13 of file CcPinRead_drv.c.

NDEBUG

#define NDEBUG

Definition at line 10 of file CcPinRead_drv.c.

ok_bcb

#define ok_bcb	(		B,
			L,
			O
	)

Value:

{                                                                                         \
    PPUBLIC_BCB public_bcb = (B);                                                         \
    ok(public_bcb->NodeTypeCode == 0x2FD, "Not a BCB: %04x\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));                           \
}

Definition at line 189 of file CcPinRead_drv.c.

Typedef Documentation

PTEST_CONTEXT

typedef struct _TEST_CONTEXT * PTEST_CONTEXT

PTEST_FCB

typedef struct _TEST_FCB * PTEST_FCB

TEST_CONTEXT

typedef struct _TEST_CONTEXT TEST_CONTEXT

TEST_FCB

typedef struct _TEST_FCB TEST_FCB

Function Documentation

AcquireForLazyWrite()

BOOLEAN NTAPI AcquireForLazyWrite	(	_In_ PVOID	Context,
		_In_ BOOLEAN	Wait
	)

Definition at line 106 of file CcPinRead_drv.c.

{
    return TRUE;
}

AcquireForReadAhead()

BOOLEAN NTAPI AcquireForReadAhead	(	_In_ PVOID	Context,
		_In_ BOOLEAN	Wait
	)

Definition at line 123 of file CcPinRead_drv.c.

{
    return TRUE;
}

CleanupTest()

static VOID CleanupTest ( ULONG  TestId, PDEVICE_OBJECT  DeviceObject  )

static

Definition at line 691 of file CcPinRead_drv.c.

{
    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;
        }
 
        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;
}

Referenced by TestMessageHandler().

MapAndLockUserBuffer()

static PVOID MapAndLockUserBuffer ( _In_ _Out_ PIRP  Irp, _In_ ULONG  BufferLength  )

static

Definition at line 159 of file CcPinRead_drv.c.

{
    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);
}

Referenced by TestIrpHandler().

PerformTest()

static VOID PerformTest ( ULONG  TestId, PDEVICE_OBJECT  DeviceObject  )

static

Definition at line 400 of file CcPinRead_drv.c.

{
    PVOID Bcb;
    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"))
        {
            BOOLEAN PinAccess = (TestId != 4);
 
            RtlZeroMemory(Fcb, sizeof(TEST_FCB));
            ExInitializeFastMutex(&Fcb->HeaderMutex);
            FsRtlSetupAdvancedHeader(&Fcb->Header, &Fcb->HeaderMutex);
 
            TestFileObject->FsContext = Fcb;
            TestFileObject->SectionObjectPointer = &Fcb->SectionObjectPointers;
 
            KmtStartSeh();
            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"))
            {
                if (TestId < 3)
                {
                    Ret = FALSE;
                    Offset.QuadPart = TestId * 0x1000;
                    KmtStartSeh();
                    Ret = CcPinRead(TestFileObject, &Offset, FileSizes.FileSize.QuadPart - Offset.QuadPart, PIN_WAIT, &Bcb, (PVOID *)&Buffer);
                    KmtEndSeh(STATUS_SUCCESS);
 
                    if (!skip(Ret == TRUE, "CcPinRead failed\n"))
                    {
                        ok_bcb(Bcb, ((4 - TestId) * 4096), Offset.QuadPart);
                        ok_eq_ulong(Buffer[(0x3000 - TestId * 0x1000) / sizeof(ULONG)], 0xDEADBABE);
 
                        CcUnpinData(Bcb);
                    }
                }
                else if (TestId == 3)
                {
                    PTEST_CONTEXT TestContext;
 
                    TestContext = ExAllocatePool(NonPagedPool, sizeof(TEST_CONTEXT));
                    if (!skip(TestContext != NULL, "ExAllocatePool failed\n"))
                    {
                        Ret = FALSE;
                        Offset.QuadPart = 0x1000;
 
                        /* Try enforce BCB first */
                        KmtStartSeh();
                        Ret = CcPinRead(TestFileObject, &Offset, FileSizes.FileSize.QuadPart - Offset.QuadPart, PIN_WAIT | PIN_IF_BCB, &Bcb, (PVOID *)&Buffer);
                        KmtEndSeh(STATUS_SUCCESS);
                        ok(Ret == FALSE, "CcPinRead succeed\n");
                        if (Ret)
                        {
                            CcUnpinData(Bcb);
                        }
 
                        KmtStartSeh();
                        Ret = CcPinRead(TestFileObject, &Offset, FileSizes.FileSize.QuadPart - Offset.QuadPart, PIN_WAIT, &TestContext->Bcb, &TestContext->Buffer);
                        KmtEndSeh(STATUS_SUCCESS);
 
                        if (!skip(Ret == TRUE, "CcPinRead failed\n"))
                        {
                            PKTHREAD ThreadHandle;
 
                            ok_bcb(TestContext->Bcb, 12288, Offset.QuadPart);
 
                            /* That's a bit rough but should do the job */
#ifdef _X86_
                            if (Memory >= 2 * 1024)
                            {
                                ok((TestContext->Buffer >= (PVOID)0xC1000000 && TestContext->Buffer < (PVOID)0xE0FFFFFF) ||
                                   (TestContext->Buffer >= (PVOID)0xA4000000 && TestContext->Buffer < (PVOID)0xBFFFFFFF),
                                   "Buffer %p not mapped in system space\n", TestContext->Buffer);
                            }
                            else if (TS)
                            {
                                ok(TestContext->Buffer >= (PVOID)0xC1000000 && TestContext->Buffer < (PVOID)0xDCFFFFFF,
                                   "Buffer %p not mapped in system space\n", TestContext->Buffer);
                            }
                            else
                            {
                                ok(TestContext->Buffer >= (PVOID)0xC1000000 && TestContext->Buffer < (PVOID)0xDBFFFFFF,
                                   "Buffer %p not mapped in system space\n", TestContext->Buffer);
                            }
#elif defined(_M_AMD64)
                            ok(TestContext->Buffer >= (PVOID)0xFFFFF98000000000 && TestContext->Buffer < (PVOID)0xFFFFFA8000000000,
                               "Buffer %p not mapped in system space\n", TestContext->Buffer);
#else
                            skip(FALSE, "System space mapping not defined\n");
#endif
 
                            TestContext->Length = FileSizes.FileSize.QuadPart - Offset.QuadPart;
                            ThreadHandle = KmtStartThread(PinInAnotherThread, TestContext);
                            KmtFinishThread(ThreadHandle, NULL);
 
                            TestContext->Length = FileSizes.FileSize.QuadPart - 2 * Offset.QuadPart;
                            ThreadHandle = KmtStartThread(PinInAnotherThread, TestContext);
                            KmtFinishThread(ThreadHandle, NULL);
 
                            CcUnpinData(TestContext->Bcb);
                        }
 
                        KmtStartSeh();
                        Ret = CcPinRead(TestFileObject, &Offset, FileSizes.FileSize.QuadPart - Offset.QuadPart, PIN_WAIT | PIN_EXCLUSIVE, &TestContext->Bcb, &TestContext->Buffer);
                        KmtEndSeh(STATUS_SUCCESS);
 
                        if (!skip(Ret == TRUE, "CcPinRead failed\n"))
                        {
                            PKTHREAD ThreadHandle;
 
                            ok_bcb(TestContext->Bcb, 12288, Offset.QuadPart);
 
                            TestContext->Length = FileSizes.FileSize.QuadPart - Offset.QuadPart;
                            ThreadHandle = KmtStartThread(PinInAnotherThreadExclusive, TestContext);
                            KmtFinishThread(ThreadHandle, NULL);
 
                            CcUnpinData(TestContext->Bcb);
                        }
 
                        ExFreePool(TestContext);
                    }
                }
                else if (TestId == 4)
                {
                    Ret = FALSE;
                    Offset.QuadPart = 0x1000;
                    KmtStartSeh();
                    Ret = CcPinRead(TestFileObject, &Offset, FileSizes.FileSize.QuadPart - Offset.QuadPart, PIN_WAIT, &Bcb, (PVOID *)&Buffer);
                    KmtEndSeh(STATUS_SUCCESS);
 
                    if (!skip(Ret == TRUE, "CcPinRead failed\n"))
                    {
                        ok_bcb(Bcb, 12288, Offset.QuadPart);
                        ok_eq_ulong(Buffer[0x2000 / sizeof(ULONG)], 0);
 
                        CcUnpinData(Bcb);
                    }
                }
                else if (TestId == 5)
                {
                    FileSizes.AllocationSize.QuadPart += VACB_MAPPING_GRANULARITY;
                    CcSetFileSizes(TestFileObject, &FileSizes);
 
                    /* Pin after EOF */
                    Ret = FALSE;
                    Offset.QuadPart = FileSizes.FileSize.QuadPart + 0x1000;
 
                    KmtStartSeh();
                    Ret = CcPinRead(TestFileObject, &Offset, 0x1000, PIN_WAIT, &Bcb, (PVOID *)&Buffer);
                    KmtEndSeh(STATUS_SUCCESS);
                    ok(Ret == TRUE, "CcPinRead failed\n");
                    if (Ret)
                    {
                        CcUnpinData(Bcb);
                    }
 
                    /* Pin a VACB after EOF */
                    Ret = FALSE;
                    Offset.QuadPart = FileSizes.FileSize.QuadPart + 0x1000 + VACB_MAPPING_GRANULARITY;
 
                    KmtStartSeh();
                    Ret = CcPinRead(TestFileObject, &Offset, 0x1000, PIN_WAIT, &Bcb, (PVOID *)&Buffer);
                    KmtEndSeh(STATUS_SUCCESS);
                    ok(Ret == TRUE, "CcPinRead failed\n");
                    if (Ret)
                    {
                        CcUnpinData(Bcb);
                    }
 
                    /* Pin after Allocation */
                    Ret = FALSE;
                    Offset.QuadPart = FileSizes.AllocationSize.QuadPart + 0x1000;
 
                    KmtStartSeh();
                    Ret = CcPinRead(TestFileObject, &Offset, 0x1000, PIN_WAIT, &Bcb, (PVOID *)&Buffer);
                    KmtEndSeh(STATUS_SUCCESS);
                    ok(Ret == TRUE, "CcPinRead failed\n");
                    if (Ret)
                    {
                        /* Set this one dirty */
                        CcSetDirtyPinnedData(Bcb, NULL);
 
                        CcUnpinData(Bcb);
 
                        WriteOffset.QuadPart = -1LL;
                        WriteLength = MAXULONG;
                        /* Flush to trigger the write */
                        CcFlushCache(TestFileObject->SectionObjectPointer, &Offset, 0x1000, NULL);
 
                        ok_eq_longlong(WriteOffset.QuadPart, Offset.QuadPart);
                        ok_eq_ulong(WriteLength, 0x1000);
                    }
 
                    /* Pin a VACB after Allocation */
                    Ret = FALSE;
                    Offset.QuadPart = FileSizes.AllocationSize.QuadPart + 0x1000 + VACB_MAPPING_GRANULARITY;
 
                    KmtStartSeh();
                    Ret = CcPinRead(TestFileObject, &Offset, 0x1000, PIN_WAIT, &Bcb, (PVOID *)&Buffer);
                    KmtEndSeh(STATUS_SUCCESS);
                    ok(Ret == TRUE, "CcPinRead failed\n");
                    if (Ret)
                    {
                        /* Set this one dirty */
                        CcSetDirtyPinnedData(Bcb, NULL);
 
                        CcUnpinData(Bcb);
 
                        WriteOffset.QuadPart = -1LL;
                        WriteLength = MAXULONG;
                        /* Flush to trigger the write */
                        CcFlushCache(TestFileObject->SectionObjectPointer, &Offset, 0x1000, NULL);
 
                        ok_eq_longlong(WriteOffset.QuadPart, Offset.QuadPart);
                        ok_eq_ulong(WriteLength, 0x1000);
                    }
 
                    /* Pin more than a VACB */
                    Ret = FALSE;
                    Offset.QuadPart = 0x0;
 
                    KmtStartSeh();
                    Ret = CcPinRead(TestFileObject, &Offset, 0x1000 + VACB_MAPPING_GRANULARITY, PIN_WAIT, &Bcb, (PVOID *)&Buffer);
                    KmtEndSeh(STATUS_SUCCESS);
                    ok(Ret == TRUE, "CcPinRead failed\n");
                    if (Ret)
                    {
                        /* Set this one dirty */
                        CcSetDirtyPinnedData(Bcb, NULL);
 
                        CcUnpinData(Bcb);
 
                        /* The data is dirtified through the VACB */
                        WriteOffset.QuadPart = -1LL;
                        WriteLength = MAXULONG;
                        CcFlushCache(TestFileObject->SectionObjectPointer, &Offset, VACB_MAPPING_GRANULARITY + 0x1000, NULL);
 
                        ok_eq_longlong(WriteOffset.QuadPart, Offset.QuadPart);
                        ok_eq_ulong(WriteLength, VACB_MAPPING_GRANULARITY + 0x1000);
                    }
                }
                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);
                    }
                }
            }
        }
    }
}

Referenced by TestMessageHandler().

PinInAnotherThread()

static VOID NTAPI PinInAnotherThread ( IN PVOID  Context )

static

Definition at line 201 of file CcPinRead_drv.c.

{
    BOOLEAN Ret;
    PULONG Buffer;
    PVOID Bcb;
    LARGE_INTEGER Offset;
    PTEST_CONTEXT TestContext;
 
    ok(TestFileObject != NULL, "Called in invalid context!\n");
    ok_eq_ulong(TestTestId, 3);
 
    TestContext = Context;
    ok(TestContext != NULL, "Called in invalid context!\n");
    ok(TestContext->Bcb != NULL, "Called in invalid context!\n");
    ok(TestContext->Buffer != NULL, "Called in invalid context!\n");
    ok(TestContext->Length != 0, "Called in invalid context!\n");
 
    Ret = FALSE;
    Offset.QuadPart = 0x1000;
    KmtStartSeh();
    Ret = CcPinRead(TestFileObject, &Offset, TestContext->Length, PIN_WAIT, &Bcb, (PVOID *)&Buffer);
    KmtEndSeh(STATUS_SUCCESS);
 
    if (!skip(Ret == TRUE, "CcPinRead failed\n"))
    {
        ok_bcb(Bcb, 12288, Offset.QuadPart);
        ok_eq_pointer(Bcb, TestContext->Bcb);
        ok_eq_pointer(Buffer, TestContext->Buffer);
 
        CcUnpinData(Bcb);
    }
 
    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_bcb(Bcb, 12288, Offset.QuadPart);
        ok_eq_pointer(Bcb, TestContext->Bcb);
        ok_eq_pointer(Buffer, TestContext->Buffer);
 
        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_bcb(Bcb, 12288, Offset.QuadPart);
        ok_eq_pointer(Bcb, TestContext->Bcb);
        ok_eq_pointer(Buffer, TestContext->Buffer);
 
        CcUnpinData(Bcb);
    }
 
    KmtStartSeh();
    Ret = CcMapData(TestFileObject, &Offset, TestContext->Length, MAP_WAIT, &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, TestContext->Buffer);
 
        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_bcb(Bcb, 12288, 4096);
        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_bcb(Bcb, 12288, 4096);
        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_bcb(Bcb, 12288, 4096);
        ok_eq_pointer(Bcb, TestContext->Bcb);
        ok_eq_pointer(Buffer, (PVOID)((ULONG_PTR)TestContext->Buffer + 0x500));
 
        CcUnpinData(Bcb);
    }
 
    return;
}

Referenced by PerformTest().

PinInAnotherThreadExclusive()

static VOID NTAPI PinInAnotherThreadExclusive ( IN PVOID  Context )

static

Definition at line 319 of file CcPinRead_drv.c.

{
    BOOLEAN Ret;
    PULONG Buffer;
    PVOID Bcb;
    LARGE_INTEGER Offset;
    PTEST_CONTEXT TestContext;
 
    ok(TestFileObject != NULL, "Called in invalid context!\n");
    ok_eq_ulong(TestTestId, 3);
 
    TestContext = Context;
    ok(TestContext != NULL, "Called in invalid context!\n");
    ok(TestContext->Bcb != NULL, "Called in invalid context!\n");
    ok(TestContext->Buffer != NULL, "Called in invalid context!\n");
    ok(TestContext->Length != 0, "Called in invalid context!\n");
 
    Ret = FALSE;
    Offset.QuadPart = 0x1000;
    KmtStartSeh();
    Ret = CcPinRead(TestFileObject, &Offset, TestContext->Length, PIN_EXCLUSIVE, &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, TestContext->Buffer);
 
        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;
}

Referenced by PerformTest().

ReleaseFromLazyWrite()

VOID NTAPI ReleaseFromLazyWrite

(

_In_ PVOID

Context

)

Definition at line 115 of file CcPinRead_drv.c.

{
    return;
}

ReleaseFromReadAhead()

VOID NTAPI ReleaseFromReadAhead

(

_In_ PVOID

Context

)

Definition at line 132 of file CcPinRead_drv.c.

{
    return;
}

TestEntry()

NTSTATUS TestEntry	(	_In_ PDRIVER_OBJECT	DriverObject,
		_In_ PCUNICODE_STRING	RegistryPath,
		_Out_ PCWSTR *	DeviceName,
		_Inout_ INT *	Flags
	)

Definition at line 43 of file CcPinRead_drv.c.

{
    ULONG Length;
    SYSTEM_BASIC_INFORMATION SBI;
    NTSTATUS Status = STATUS_SUCCESS;
    RTL_OSVERSIONINFOEXW VersionInfo;
 
    PAGED_CODE();
 
    UNREFERENCED_PARAMETER(RegistryPath);
 
    *DeviceName = L"CcPinRead";
    *Flags = TESTENTRY_NO_EXCLUSIVE_DEVICE |
             TESTENTRY_BUFFERED_IO_DEVICE |
             TESTENTRY_NO_READONLY_DEVICE;
 
    KmtRegisterIrpHandler(IRP_MJ_READ, NULL, TestIrpHandler);
    KmtRegisterIrpHandler(IRP_MJ_WRITE, NULL, TestIrpHandler);
    KmtRegisterMessageHandler(0, NULL, TestMessageHandler);
 
    Status = ZwQuerySystemInformation(SystemBasicInformation,
                                      &SBI,
                                      sizeof(SBI),
                                      &Length);
    if (NT_SUCCESS(Status))
    {
        Memory = (SBI.NumberOfPhysicalPages * SBI.PageSize) / 1024 / 1024;
    }
    else
    {
        Status = STATUS_SUCCESS;
    }
 
    VersionInfo.dwOSVersionInfoSize = sizeof(VersionInfo);
    Status = RtlGetVersion((PRTL_OSVERSIONINFOW)&VersionInfo);
    if (NT_SUCCESS(Status))
    {
        TS = BooleanFlagOn(VersionInfo.wSuiteMask, VER_SUITE_TERMINAL) &&
             !BooleanFlagOn(VersionInfo.wSuiteMask, VER_SUITE_SINGLEUSERTS);
    }
    else
    {
        Status = STATUS_SUCCESS;
    }
 
    trace("System with %I64dMb RAM and terminal services %S\n",  Memory, (TS ? L"enabled" : L"disabled"));
 
    return Status;
}

TestIrpHandler()

static NTSTATUS TestIrpHandler ( _In_ PDEVICE_OBJECT  DeviceObject, _In_ PIRP  Irp, _In_ PIO_STACK_LOCATION  IoStack  )

static

Definition at line 773 of file CcPinRead_drv.c.

{
    NTSTATUS Status;
 
    PAGED_CODE();
 
    DPRINT("IRP %x/%x\n", IoStack->MajorFunction, IoStack->MinorFunction);
    ASSERT(IoStack->MajorFunction == IRP_MJ_READ ||
           IoStack->MajorFunction == IRP_MJ_WRITE);
 
    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;
    }
    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 == 5) || (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);
 
        if (TestTestId == 5)
        {
            /* This assumes continuous writes. */
            if (WriteOffset.QuadPart != -1)
            {
                LONGLONG WriteEnd = WriteOffset.QuadPart + WriteLength;
 
                if (WriteOffset.QuadPart > Offset.QuadPart)
                    WriteOffset = Offset;
                if (WriteEnd < (Offset.QuadPart + Length))
                    WriteLength = (Offset.QuadPart + Length) - WriteOffset.QuadPart;
            }
            else
            {
                WriteOffset = Offset;
                WriteLength = Length;
            }
        }
        else
        {
            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)
    {
        IoMarkIrpPending(Irp);
        IoCompleteRequest(Irp, IO_NO_INCREMENT);
        Status = STATUS_PENDING;
    }
    else
    {
        Irp->IoStatus.Status = Status;
        IoCompleteRequest(Irp, IO_NO_INCREMENT);
    }
 
    FsRtlExitFileSystem();
 
    return Status;
}

TestMessageHandler()

static NTSTATUS TestMessageHandler ( _In_ PDEVICE_OBJECT  DeviceObject, _In_ ULONG  ControlCode, _In_opt_ PVOID Buffer  , _In_ SIZE_T  InLength, _Inout_ PSIZE_T  OutLength  )

static

Definition at line 738 of file CcPinRead_drv.c.

{
    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;
}

TestUnload()

VOID TestUnload

(

_In_ PDRIVER_OBJECT

DriverObject

)

Definition at line 98 of file CcPinRead_drv.c.

{
    PAGED_CODE();
}

Variable Documentation

Callbacks

CACHE_MANAGER_CALLBACKS Callbacks

static

Initial value:

= {
    AcquireForLazyWrite,
    ReleaseFromLazyWrite,
    AcquireForReadAhead,
    ReleaseFromReadAhead,
}

Definition at line 138 of file CcPinRead_drv.c.

Referenced by PerformTest().

FileSizes

CC_FILE_SIZES FileSizes

static

Initial value:

= {
    RTL_CONSTANT_LARGE_INTEGER((LONGLONG)0x4000), 
    RTL_CONSTANT_LARGE_INTEGER((LONGLONG)0x4000), 
  
}

Definition at line 145 of file CcPinRead_drv.c.

Referenced by PerformTest().

Memory

ULONGLONG Memory = 0

static

Definition at line 36 of file CcPinRead_drv.c.

SmallFileSizes

CC_FILE_SIZES SmallFileSizes

static

Initial value:

= {
    RTL_CONSTANT_LARGE_INTEGER((LONGLONG)512), 
    RTL_CONSTANT_LARGE_INTEGER((LONGLONG)496), 
  
}

Definition at line 151 of file CcPinRead_drv.c.

Referenced by PerformTest().

TestDeviceObject

PDEVICE_OBJECT TestDeviceObject

static

Definition at line 32 of file CcPinRead_drv.c.

Referenced by CleanupTest(), PerformTest(), and TestIrpHandler().

TestFileObject

PFILE_OBJECT TestFileObject

static

Definition at line 31 of file CcPinRead_drv.c.

Referenced by CleanupTest(), PerformTest(), PinInAnotherThread(), PinInAnotherThreadExclusive(), and TestIrpHandler().

TestIrpHandler

KMT_IRP_HANDLER TestIrpHandler

static

Definition at line 33 of file CcPinRead_drv.c.

Referenced by TestEntry().

TestMessageHandler

KMT_MESSAGE_HANDLER TestMessageHandler

static

Definition at line 34 of file CcPinRead_drv.c.

Referenced by TestEntry().

TestTestId

ULONG TestTestId = -1

static

Definition at line 30 of file CcPinRead_drv.c.

Referenced by CleanupTest(), PerformTest(), PinInAnotherThread(), PinInAnotherThreadExclusive(), and TestIrpHandler().

TestWriteCalled

BOOLEAN TestWriteCalled = FALSE

static

Definition at line 35 of file CcPinRead_drv.c.

Referenced by CleanupTest(), and TestIrpHandler().

TS

BOOLEAN TS = FALSE

static

Definition at line 37 of file CcPinRead_drv.c.

Referenced by PerformTest(), and TestEntry().

WriteLength

ULONG WriteLength

Definition at line 40 of file CcPinRead_drv.c.

Referenced by _Requires_lock_held_(), PerformTest(), RxCommonWrite(), TestIrpHandler(), UDFCommonWrite(), VfatWriteDisk(), VfatWriteDiskPartial(), and WriteAttribute().

WriteOffset

LARGE_INTEGER WriteOffset

Definition at line 39 of file CcPinRead_drv.c.