[NTOSKRNL:CC] Bug Fix

This commit is contained in:
Dibyamartanda Samanta 2024-05-25 09:42:40 +02:00
parent 5a5ebcae89
commit 785ff4b644

View File

@ -10,39 +10,51 @@
#include <ntoskrnl.hpp>
BOOL
NTAPI
IsGoToNextMap(IN PSHARED_CACHE_MAP SharedMap, IN ULONG TargetPages)
BOOLEAN NTAPI IsGoToNextMap(IN PSHARED_CACHE_MAP SharedMap, IN ULONG TargetPages)
{
BOOLEAN Skip = FALSE;
if (SharedMap->Flags & (0x20 | 0x800))
// If the map is marked for teardown or modified without writing, return TRUE
if (SharedMap->Flags & (SHARE_FL_WAITING_TEARDOWN | SHARE_FL_MODIFIED_NO_WRITE)) {
return TRUE;
}
if ((SharedMap->OpenCount || SharedMap->DirtyPages) && SharedMap->FileSize.QuadPart)
{
// Determine if we should skip based on open count and dirty pages
if ((SharedMap->OpenCount || SharedMap->DirtyPages) && SharedMap->FileSize.QuadPart != 0) {
Skip = TRUE;
}
if (!SharedMap->DirtyPages && Skip)
// Skip if no dirty pages but `Skip` is `TRUE`
if (!SharedMap->DirtyPages && Skip) {
return TRUE;
}
if (SharedMap->Flags & SHARE_FL_WAITING_TEARDOWN)
// If the map is waiting for teardown, return FALSE
if (SharedMap->Flags & SHARE_FL_WAITING_TEARDOWN) {
return FALSE;
}
if (!TargetPages && Skip)
// Skip if no target pages and `Skip` is `TRUE`
if (TargetPages == 0 && Skip) {
return TRUE;
}
// Increment the lazy write pass count
SharedMap->LazyWritePassCount++;
if ((SharedMap->LazyWritePassCount & 0xF) && (SharedMap->Flags & SHARE_FL_MODIFIED_NO_WRITE) &&
CcCapturedSystemSize != MmSmallSystem && SharedMap->DirtyPages < 0x40 && Skip)
// Skip based on lazy write pass count and other conditions
if ((SharedMap->LazyWritePassCount & 0xF) &&
(SharedMap->Flags & SHARE_FL_MODIFIED_NO_WRITE) &&
CcCapturedSystemSize != MmSmallSystem &&
SharedMap->DirtyPages < 0x40 && Skip)
{
return TRUE;
}
if ((SharedMap->FileObject->Flags & 0x8000) && SharedMap->OpenCount &&
CcCanIWrite(SharedMap->FileObject, 0x40000, 0, 0xFF) && Skip)
// Skip if file object is marked for direct I/O and other conditions
if ((SharedMap->FileObject->Flags & FO_DIRECT_IO) &&
SharedMap->OpenCount != 0 &&
CcCanIWrite(SharedMap->FileObject, 0x40000, FALSE, 0xFF) && Skip)
{
return TRUE;
}