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