Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558/* * PROJECT: ReactOS kernel-mode tests * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory * PURPOSE: Test driver for CcSetFileSizes function * PROGRAMMER: Pierre Schweitzer <pierre@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;
static ULONG TestTestId = -1;static PFILE_OBJECT TestFileObject;static PDEVICE_OBJECT TestDeviceObject;static KMT_IRP_HANDLER TestIrpHandler;static KMT_MESSAGE_HANDLER TestMessageHandler;static BOOLEAN TestUnpin = FALSE;static BOOLEAN TestSizing = FALSE;static BOOLEAN TestDirtying = FALSE;static BOOLEAN TestUncaching = FALSE;static BOOLEAN TestWritten = FALSE;
NTSTATUSTestEntry( _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"CcSetFileSizes"; *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);
return Status;}
VOIDTestUnload( _In_ PDRIVER_OBJECT DriverObject){ PAGED_CODE();}
BOOLEANNTAPIAcquireForLazyWrite( _In_ PVOID Context, _In_ BOOLEAN Wait){ return TRUE;}
VOIDNTAPIReleaseFromLazyWrite( _In_ PVOID Context){ return;}
BOOLEANNTAPIAcquireForReadAhead( _In_ PVOID Context, _In_ BOOLEAN Wait){ return TRUE;}
VOIDNTAPIReleaseFromReadAhead( _In_ PVOID Context){ return;}
static CACHE_MANAGER_CALLBACKS Callbacks = { AcquireForLazyWrite, ReleaseFromLazyWrite, AcquireForReadAhead, ReleaseFromReadAhead,};
static CC_FILE_SIZES NewFileSizes = { RTL_CONSTANT_LARGE_INTEGER((LONGLONG)VACB_MAPPING_GRANULARITY), // .AllocationSize RTL_CONSTANT_LARGE_INTEGER((LONGLONG)VACB_MAPPING_GRANULARITY), // .FileSize RTL_CONSTANT_LARGE_INTEGER((LONGLONG)VACB_MAPPING_GRANULARITY) // .ValidDataLength};
staticPVOIDMapAndLockUserBuffer( _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);}
staticVOIDPerformTest( ULONG TestId, PDEVICE_OBJECT DeviceObject){ PVOID Bcb; BOOLEAN Ret; PULONG Buffer; PTEST_FCB Fcb; LARGE_INTEGER Offset; IO_STATUS_BLOCK IoStatus;
ok_eq_pointer(TestFileObject, NULL); ok_eq_pointer(TestDeviceObject, NULL); ok_eq_ulong(TestTestId, -1);
TestWritten = FALSE; 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; Fcb->Header.AllocationSize.QuadPart = VACB_MAPPING_GRANULARITY; Fcb->Header.FileSize.QuadPart = VACB_MAPPING_GRANULARITY - PAGE_SIZE; Fcb->Header.ValidDataLength.QuadPart = VACB_MAPPING_GRANULARITY - PAGE_SIZE;
if ((TestId > 1 && TestId < 4) || TestId >= 5) { Fcb->Header.AllocationSize.QuadPart = VACB_MAPPING_GRANULARITY - PAGE_SIZE; }
KmtStartSeh(); CcInitializeCacheMap(TestFileObject, (PCC_FILE_SIZES)&Fcb->Header.AllocationSize, TRUE, &Callbacks, NULL); KmtEndSeh(STATUS_SUCCESS);
if (!skip(CcIsFileCached(TestFileObject) == TRUE, "CcInitializeCacheMap failed\n")) { trace("Starting test: %d\n", TestId);
if (TestId == 0 || TestId == 2) { Offset.QuadPart = 0; KmtStartSeh(); Ret = CcMapData(TestFileObject, &Offset, VACB_MAPPING_GRANULARITY - PAGE_SIZE, MAP_WAIT, &Bcb, (PVOID *)&Buffer); KmtEndSeh(STATUS_SUCCESS);
if (!skip(Ret == TRUE, "CcMapData failed\n")) { ok_eq_ulong(Buffer[(VACB_MAPPING_GRANULARITY - PAGE_SIZE - sizeof(ULONG)) / sizeof(ULONG)], 0xBABABABA); CcUnpinData(Bcb); }
KmtStartSeh(); CcSetFileSizes(TestFileObject, &NewFileSizes); KmtEndSeh(STATUS_SUCCESS);
Fcb->Header.AllocationSize.QuadPart = VACB_MAPPING_GRANULARITY; Fcb->Header.FileSize.QuadPart = VACB_MAPPING_GRANULARITY;
Offset.QuadPart = 0; KmtStartSeh(); Ret = CcMapData(TestFileObject, &Offset, VACB_MAPPING_GRANULARITY, MAP_WAIT, &Bcb, (PVOID *)&Buffer); KmtEndSeh(STATUS_SUCCESS);
if (!skip(Ret == TRUE, "CcMapData failed\n")) { ok_eq_ulong(Buffer[(VACB_MAPPING_GRANULARITY - sizeof(ULONG)) / sizeof(ULONG)], 0xBABABABA);
CcUnpinData(Bcb); } } else if (TestId == 1 || TestId == 3) { Buffer = ExAllocatePool(NonPagedPool, PAGE_SIZE); if (!skip(Buffer != NULL, "ExAllocatePool failed\n")) { Ret = FALSE; Offset.QuadPart = VACB_MAPPING_GRANULARITY - 2 * PAGE_SIZE;
KmtStartSeh(); Ret = CcCopyRead(TestFileObject, &Offset, PAGE_SIZE, TRUE, Buffer, &IoStatus); KmtEndSeh(STATUS_SUCCESS);
ok_eq_ulong(Buffer[(PAGE_SIZE - sizeof(ULONG)) / sizeof(ULONG)], 0xBABABABA);
KmtStartSeh(); CcSetFileSizes(TestFileObject, &NewFileSizes); KmtEndSeh(STATUS_SUCCESS);
Fcb->Header.AllocationSize.QuadPart = VACB_MAPPING_GRANULARITY; Fcb->Header.FileSize.QuadPart = VACB_MAPPING_GRANULARITY; RtlZeroMemory(Buffer, PAGE_SIZE);
Offset.QuadPart = VACB_MAPPING_GRANULARITY - PAGE_SIZE;
KmtStartSeh(); Ret = CcCopyRead(TestFileObject, &Offset, PAGE_SIZE, TRUE, Buffer, &IoStatus); KmtEndSeh(STATUS_SUCCESS);
ok_eq_ulong(Buffer[(PAGE_SIZE - sizeof(ULONG)) / sizeof(ULONG)], 0xBABABABA);
ExFreePool(Buffer); } } else if (TestId == 4 || TestId == 5) { /* Kill lazy writer */ CcSetAdditionalCacheAttributes(TestFileObject, FALSE, TRUE);
Offset.QuadPart = 0; KmtStartSeh(); Ret = CcPinRead(TestFileObject, &Offset, VACB_MAPPING_GRANULARITY - PAGE_SIZE, MAP_WAIT, &Bcb, (PVOID *)&Buffer); KmtEndSeh(STATUS_SUCCESS);
if (!skip(Ret == TRUE, "CcPinRead failed\n")) { LARGE_INTEGER Flushed;
ok_eq_ulong(Buffer[(VACB_MAPPING_GRANULARITY - PAGE_SIZE - sizeof(ULONG)) / sizeof(ULONG)], 0xBABABABA); Buffer[(VACB_MAPPING_GRANULARITY - PAGE_SIZE - sizeof(ULONG)) / sizeof(ULONG)] = 0xDADADADA;
TestDirtying = TRUE; CcSetDirtyPinnedData(Bcb, NULL); TestDirtying = FALSE;
ok_bool_false(TestWritten, "Dirty VACB has been unexpectedly written!\n");
TestSizing = TRUE; KmtStartSeh(); CcSetFileSizes(TestFileObject, &NewFileSizes); KmtEndSeh(STATUS_SUCCESS); TestSizing = FALSE;
ok_bool_false(TestWritten, "Dirty VACB has been unexpectedly written!\n");
Fcb->Header.AllocationSize.QuadPart = VACB_MAPPING_GRANULARITY; Fcb->Header.FileSize.QuadPart = VACB_MAPPING_GRANULARITY;
Flushed = CcGetFlushedValidData(TestFileObject->SectionObjectPointer, FALSE); ok(Flushed.QuadPart == 0, "Flushed: %I64d\n", Flushed.QuadPart);
TestUnpin = TRUE; CcUnpinData(Bcb); TestUnpin = FALSE;
ok_bool_false(TestWritten, "Dirty VACB has been unexpectedly written!\n");
Offset.QuadPart = 0; KmtStartSeh(); Ret = CcMapData(TestFileObject, &Offset, VACB_MAPPING_GRANULARITY, MAP_WAIT, &Bcb, (PVOID *)&Buffer); KmtEndSeh(STATUS_SUCCESS);
if (!skip(Ret == TRUE, "CcMapData failed\n")) { ok_eq_ulong(Buffer[(VACB_MAPPING_GRANULARITY - PAGE_SIZE - sizeof(ULONG)) / sizeof(ULONG)], 0xDADADADA); ok_eq_ulong(Buffer[(VACB_MAPPING_GRANULARITY - sizeof(ULONG)) / sizeof(ULONG)], 0xBABABABA);
CcUnpinData(Bcb);
ok_bool_false(TestWritten, "Dirty VACB has been unexpectedly written!\n"); } } } else if (TestId == 6) { Offset.QuadPart = 0; KmtStartSeh(); Ret = CcMapData(TestFileObject, &Offset, VACB_MAPPING_GRANULARITY - PAGE_SIZE, MAP_WAIT, &Bcb, (PVOID *)&Buffer); KmtEndSeh(STATUS_SUCCESS);
if (!skip(Ret == TRUE, "CcMapData failed\n")) { ok_eq_ulong(Buffer[(VACB_MAPPING_GRANULARITY - PAGE_SIZE - sizeof(ULONG)) / sizeof(ULONG)], 0xBABABABA); }
KmtStartSeh(); CcSetFileSizes(TestFileObject, &NewFileSizes); KmtEndSeh(STATUS_SUCCESS);
if (Ret == TRUE) CcUnpinData(Bcb); } } } }}
staticVOIDCleanupTest( 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)) { TestUncaching = TRUE; KeInitializeEvent(&CacheUninitEvent.Event, NotificationEvent, FALSE); CcUninitializeCacheMap(TestFileObject, &Zero, &CacheUninitEvent); KeWaitForSingleObject(&CacheUninitEvent.Event, Executive, KernelMode, FALSE, NULL); TestUncaching = FALSE; }
if (TestFileObject->FsContext != NULL) { ExFreePool(TestFileObject->FsContext); TestFileObject->FsContext = NULL; TestFileObject->SectionObjectPointer = NULL; }
ObDereferenceObject(TestFileObject); }
TestFileObject = NULL; TestDeviceObject = NULL; TestTestId = -1;}
staticNTSTATUSTestMessageHandler( _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;}
staticNTSTATUSTestIrpHandler( _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 || IoStack->MajorFunction == IRP_MJ_WRITE);
FsRtlEnterFileSystem();
Status = STATUS_NOT_SUPPORTED; Irp->IoStatus.Information = 0;
if (IoStack->MajorFunction == IRP_MJ_READ) { PMDL Mdl; ULONG Length; PTEST_FCB Fcb; LARGE_INTEGER Offset; PVOID Buffer, OrigBuffer;
Offset = IoStack->Parameters.Read.ByteOffset; Length = IoStack->Parameters.Read.Length; Fcb = IoStack->FileObject->FsContext;
ok_eq_pointer(DeviceObject, TestDeviceObject); ok_eq_pointer(IoStack->FileObject, TestFileObject); ok(Fcb != NULL, "Null FCB\n");
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"); OrigBuffer = Buffer = MapAndLockUserBuffer(Irp, Length); ok(Buffer != NULL, "Null pointer!\n"); if (Buffer == NULL) { Status = STATUS_UNSUCCESSFUL; goto Exit; }
if (Offset.QuadPart < Fcb->Header.FileSize.QuadPart) { RtlFillMemory(Buffer, min(Length, Fcb->Header.FileSize.QuadPart - Offset.QuadPart), 0xBA); Buffer = (PVOID)((ULONG_PTR)Buffer + (ULONG_PTR)min(Length, Fcb->Header.FileSize.QuadPart - Offset.QuadPart));
if (Length > (Fcb->Header.FileSize.QuadPart - Offset.QuadPart)) { RtlFillMemory(Buffer, Length - (Fcb->Header.FileSize.QuadPart - Offset.QuadPart), 0xBD); } } else { RtlFillMemory(Buffer, Length, 0xBD); }
if ((TestTestId == 4 || TestTestId == 5) && TestWritten && Offset.QuadPart <= VACB_MAPPING_GRANULARITY - PAGE_SIZE - sizeof(ULONG) && Offset.QuadPart + Length >= VACB_MAPPING_GRANULARITY - PAGE_SIZE) { Buffer = (PVOID)((ULONG_PTR)OrigBuffer + (VACB_MAPPING_GRANULARITY - PAGE_SIZE - sizeof(ULONG))); RtlFillMemory(Buffer, sizeof(ULONG), 0xDA); }
Status = STATUS_SUCCESS;
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 == 4 || TestTestId == 5), "Unexpected test id: %d\n", TestTestId); ok_eq_pointer(DeviceObject, TestDeviceObject); ok_eq_pointer(IoStack->FileObject, TestFileObject);
ok_bool_false(TestUnpin, "Write triggered while unpinning!\n"); ok_bool_false(TestSizing, "Write triggered while sizing!\n"); ok_bool_false(TestDirtying, "Write triggered while dirtying!\n"); ok_bool_true(TestUncaching, "Write not triggered while uncaching!\n");
ok(FlagOn(Irp->Flags, IRP_NOCACHE), "Not coming from Cc\n");
ok_irql(PASSIVE_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);
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");
TestWritten = 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); }
Exit: FsRtlExitFileSystem();
return Status;}