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

[NTOSKRNL:CC] MDL Handler Refactored for using RAII

Browse Source
This commit is contained in:
Dibyamartanda Samanta 2024-05-24 10:42:12 +02:00
parent ac5f558605
commit 2300440e3d
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
NTOSKRNL/CC/ccmdlhandler.cpp
View File

@ -43,7 +43,6 @@ CcMdlWriteComplete2(IN PFILE_OBJECT FileObject,
{
PLARGE_INTEGER FOffset = nullptr;
KIRQL CurrentIrql = {0};
NTSTATUS Exception = {0};
PIO_STATUS_BLOCK IoStatus = nullptr;
@ -71,9 +70,9 @@ for( auto it = begin(MdlChain); it != end(MdlChain);++it)
if (!(MdlChain->MdlFlags & 2) == 0))
{
CurrentIrql = KeAcquireQueuedSpinLock(LockQueueMasterLock);
SpinLockGuard lockGuard(LockQueueMasterLock);
CcDecrementOpenCount(SharedCacheMap);
KeReleaseQueuedSpinLock(LockQueueMasterLock,CurrentIrql);
}
if (!NT_SUCCESS(Exception))
{
@ -191,9 +190,9 @@ InterlockedExchange64(&PrivateCachemap->BeyondLastByte2.Quadpart,FinalLength);
/* Update Shared Cachemap Flags */
if ( (SharedCacheMap->Flags & 8) != 0 )
{
KIRQL LAST_IRQL = KeAcquireQueuedSpinLock(LockQueueMasterLock);
SpinLockGuard lockGuard(LockQueueMasterLock);
SharedCacheMap->Flags &= ~8;
KeReleaseQueuedSpinLock(LockQueueMasterLock, LAST_IRQL);
}
/* Update IRP Bytes */
@ -234,7 +233,6 @@ NTAPI
CcMdlWriteAbort(IN PFILE_OBJECT FileObject,
IN PMDL MdlChain)
{
KIRQL CurrentIrql = {0};
PSHARED_CACHE_MAP SharedCacheMap = FileObject->SectionObjectPointer->SharedCacheMap;
@ -248,9 +246,9 @@ for( auto it = begin(MdlChain); it != end(MdlChain);++it)
}
if ( (MdlChain->MdlFlags & 2) != 0 )
{
CurrentIrql = KeAcquireQueuedSpinLock(LockQueueMasterLock);
SpinLockGuard lockGuard(LockQueueMasterLock);
CcDecrementOpenCount(SharedCacheMap);
KeReleaseQueuedSpinLock(LockQueueMasterLock, CurrentIrql);
}
}