Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-06-24 06:51:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NTOS:MM] Allow MiMapPageInHyperSpace to be called from DISPATCH_LEVEL

Browse source 
Also annotate it to show its behaviour.
This commit is contained in:
Jérôme Gardou 2021-04-07 23:05:51 +02:00
parent 81fd88e249
commit 8404d1a6ff
2 changed files with 41 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
ntoskrnl/include/internal/mm.h
View file

@ -1011,17 +1011,26 @@ MmZeroPageThread(
);
/* hypermap.c *****************************************************************/
_Acquires_lock_(Process->HyperSpaceLock)
_When_(OldIrql == 0, _IRQL_requires_(DISPATCH_LEVEL))
_When_(OldIrql != 0, _IRQL_requires_(PASSIVE_LEVEL))
_When_(OldIrql != 0, _At_(*OldIrql, IRQL_saves_))
_When_(OldIrql != 0, _IRQL_raises_(DISPATCH_LEVEL))
PVOID
NTAPI
MiMapPageInHyperSpace(IN PEPROCESS Process,
IN PFN_NUMBER Page,
IN PKIRQL OldIrql);
MiMapPageInHyperSpace(_In_ PEPROCESS Process,
_In_ PFN_NUMBER Page,
_Out_opt_ PKIRQL OldIrql);
_Requires_lock_held_(Process->HyperSpaceLock)
_Releases_lock_(Process->HyperSpaceLock)
_IRQL_requires_(DISPATCH_LEVEL)
_When_(OldIrql != MM_NOIRQL, _At_(OldIrql, _IRQL_restores_))
VOID
NTAPI
MiUnmapPageInHyperSpace(IN PEPROCESS Process,
IN PVOID Address,
IN KIRQL OldIrql);
MiUnmapPageInHyperSpace(_In_ PEPROCESS Process,
_In_ PVOID Address,
_In_ KIRQL OldIrql);
PVOID
NTAPI

34
ntoskrnl/mm/ARM3/hypermap.c
View file

@ -23,16 +23,24 @@ MMPTE HyperTemplatePte;
/* PRIVATE FUNCTIONS **********************************************************/
_Acquires_lock_(Process->HyperSpaceLock)
_When_(OldIrql == 0, _IRQL_requires_(DISPATCH_LEVEL))
_When_(OldIrql != 0, _IRQL_requires_(PASSIVE_LEVEL))
_When_(OldIrql != 0, _At_(*OldIrql, _IRQL_saves_))
_When_(OldIrql != 0, _IRQL_raises_(DISPATCH_LEVEL))
PVOID
NTAPI
MiMapPageInHyperSpace(IN PEPROCESS Process,
IN PFN_NUMBER Page,
IN PKIRQL OldIrql)
MiMapPageInHyperSpace(_In_ PEPROCESS Process,
_In_ PFN_NUMBER Page,
_Out_opt_ PKIRQL OldIrql)
{
MMPTE TempPte;
PMMPTE PointerPte;
PFN_NUMBER Offset;
ASSERT(((OldIrql != NULL) && (KeGetCurrentIrql() == PASSIVE_LEVEL))
|| ((OldIrql == NULL) && (KeGetCurrentIrql() == DISPATCH_LEVEL)));
//
// Never accept page 0 or non-physical pages
//
@ -54,7 +62,10 @@ MiMapPageInHyperSpace(IN PEPROCESS Process,
// Acquire the hyperlock
//
ASSERT(Process == PsGetCurrentProcess());
KeAcquireSpinLock(&Process->HyperSpaceLock, OldIrql);
if (OldIrql != NULL)
KeAcquireSpinLock(&Process->HyperSpaceLock, OldIrql);
else
KeAcquireSpinLockAtDpcLevel(&Process->HyperSpaceLock);
//
// Now get the first free PTE
@ -86,11 +97,15 @@ MiMapPageInHyperSpace(IN PEPROCESS Process,
return MiPteToAddress(PointerPte);
}
_Requires_lock_held_(Process->HyperSpaceLock)
_Releases_lock_(Process->HyperSpaceLock)
_IRQL_requires_(DISPATCH_LEVEL)
_When_(OldIrql != MM_NOIRQL, _At_(OldIrql, _IRQL_restores_))
VOID
NTAPI
MiUnmapPageInHyperSpace(IN PEPROCESS Process,
IN PVOID Address,
IN KIRQL OldIrql)
MiUnmapPageInHyperSpace(_In_ PEPROCESS Process,
_In_ PVOID Address,
_In_ KIRQL OldIrql)
{
ASSERT(Process == PsGetCurrentProcess());
@ -103,7 +118,10 @@ MiUnmapPageInHyperSpace(IN PEPROCESS Process,
// Release the hyperlock
//
ASSERT(KeGetCurrentIrql() == DISPATCH_LEVEL);
KeReleaseSpinLock(&Process->HyperSpaceLock, OldIrql);
if (OldIrql == MM_NOIRQL)
KeReleaseSpinLockFromDpcLevel(&Process->HyperSpaceLock);
else
KeReleaseSpinLock(&Process->HyperSpaceLock, OldIrql);
}
PVOID