DibyaXP/alcyone
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[NTOSKRNL:CC] CC LOG: Added CcGetLsnForFileObject

Browse Source 
Added CcGetLsnForFileObject  and fixed some Synchronization code,
This commit is contained in:
Dibyamartanda Samanta 2024-05-18 08:22:19 +02:00
parent 90e470627f
commit 5b5000d7f1
1 changed files with 56 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
NTOSKRNL/CC/ccloghandler.cpp
View File

@ -58,7 +58,8 @@ CcGetDirtyPages(IN PVOID LogHandle,
FileObject = CcReferenceSharedCacheMapFileObject(SharedCacheMap); FileObject = CcReferenceSharedCacheMapFileObject(SharedCacheMap);
/* Acquire the BCB Lock */ /* Acquire the BCB Lock */
oldIrql = ExAcquireSpinLockShared (&SharedCacheMap->BcbSpinLock); KLOCK_QUEUE_HANDLE LockHandle;
KeAcquireInStackQueuedSpinLock(&SharedCacheMap->BcbLock, &LockHandle);
/* Iterate over BCB List */ /* Iterate over BCB List */
for (auto it = ListEntryIterator::begin(&SharedCacheMap->BcbList); it != ListEntryIterator::end(&SharedCacheMap->BcbList); ++it) for (auto it = ListEntryIterator::begin(&SharedCacheMap->BcbList); it != ListEntryIterator::end(&SharedCacheMap->BcbList); ++it)
@ -75,7 +76,7 @@ CcGetDirtyPages(IN PVOID LogHandle,
Bcb->PinCount++; Bcb->PinCount++;
/*Release the BCB lock temporarily for the callback*/ /*Release the BCB lock temporarily for the callback*/
ExReleaseSpinLockShared (&SharedCacheMap->BcbSpinLock,oldIrql); KeReleaseInStackQueuedSpinLock(&LockHandle);
/*Unpin any previously pinned BCB*/ /*Unpin any previously pinned BCB*/
if (BcbToUnpin) if (BcbToUnpin)
@ -99,7 +100,7 @@ CcGetDirtyPages(IN PVOID LogHandle,
} }
// Reacquire the BCB lock // Reacquire the BCB lock
oldIrql = ExAcquireSpinLockShared (&SharedCacheMap->BcbSpinLock); KeAcquireInStackQueuedSpinLock(&SharedCacheMap->BcbLock, &LockHandle);
// Decrement the PinCount or mark for unpinning // Decrement the PinCount or mark for unpinning
if (Bcb->PinCount > 1) if (Bcb->PinCount > 1)
{ {
@ -114,7 +115,7 @@ CcGetDirtyPages(IN PVOID LogHandle,
} }
/*Release the BCB lock*/ /*Release the BCB lock*/
ExReleaseSpinLockShared (&SharedCacheMap->BcbSpinLock,oldIrql); KeReleaseInStackQueuedSpinLock(&LockHandle);
/* Dereference the shared cache map file object*/ /* Dereference the shared cache map file object*/
CcDereferenceSharedCacheMapFileObject(SharedCacheMap, FileObject); CcDereferenceSharedCacheMapFileObject(SharedCacheMap, FileObject);
@ -143,7 +144,57 @@ NTAPI
CcGetLsnForFileObject(IN PFILE_OBJECT FileObject, CcGetLsnForFileObject(IN PFILE_OBJECT FileObject,
OUT OPTIONAL PLARGE_INTEGER OldestLsn) OUT OPTIONAL PLARGE_INTEGER OldestLsn)
{ {
/* IMPLEMENTATION INCOMPLETE*/ PSHARED_CACHE_MAP SharedCacheMap = nullptr;
PMBCB Bcb = nullptr;
KLOCK_QUEUE_HANDLE LockHandle;
LARGE_INTEGER OldestLsnValue = {0};
LARGE_INTEGER Result = {0};
BITMAP_RANGE OldRangeBasePage = {0};
if (OldestLsn)
{
OldestLsn->QuadPart = 0;
}
SharedCacheMap = FileObject->SectionObjectPointer->SharedCacheMap;
if (!SharedCacheMap || !SharedCacheMap->BcbList.Flink) {
return 0;
}
/* Acquire the spin lock*/
KeAcquireInStackQueuedSpinLock(&SharedCacheMap->BcbLock, &LockHandle);
for(auto it = ListEntryIterator::begin(&SharedCacheMap->BcbList); it != ListEntryIterator::end(&SharedCacheMap->BcbList); ++it)
{
Bcb = CONTAINING_RECORD(*it, _MBCB, BitmapRange1);
if (Bcb->NodeTypeCode == NODE_TYPE_CODE_BCB && Bcb->NodeIsInZone)
{
auto MostRecentlyDirtiedPage = Bcb->MostRecentlyDirtiedPage.QuadPart;
if (MostRecentlyDirtiedPage && (!OldestLsnValue.QuadPart || MostRecentlyDirtiedPage < OldestLsnValue.QuadPart))
{
OldestLsnValue = MostRecentlyDirtiedPage;
}
if(Bcb->BitmapRange1.BasePage > OldRangeBasePage.BasePage)
{
OldRangeBasePage = Bcb->BitmapRange1;
}
}
}
}
KeReleaseInStackQueuedSpinLock(&LockHandle);
/* Set the OldestLsn value if requested*/
if (OldestLsn)
{
OldestLsn->QuadPart = OldestLsnValue;
}
Result.QuadPart = OldRangeBasePage.BasePage;
return Result;
} }