From 1a7296bb39e031bbb7d13d2de8d30f26811a76f9 Mon Sep 17 00:00:00 2001
From: Roel Messiant <roelmessiant@gmail.com>
Date: Sat, 17 Sep 2011 20:26:31 +0000
Subject: [PATCH] [NTOSKRNL] Disable normal APC delivery while
 acquiring/releasing a heap lock. Also, BOOLEAN != NTSTATUS.

svn path=/trunk/; revision=53729
---
 reactos/ntoskrnl/rtl/libsupp.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/reactos/ntoskrnl/rtl/libsupp.c b/reactos/ntoskrnl/rtl/libsupp.c
index 5f9e0c10d5a..7171284d61a 100644
--- a/reactos/ntoskrnl/rtl/libsupp.c
+++ b/reactos/ntoskrnl/rtl/libsupp.c
@@ -168,7 +168,10 @@ NTAPI
 RtlEnterHeapLock(
     PHEAP_LOCK Lock)
 {
-    return ExAcquireResourceExclusive(&Lock->Resource, TRUE);
+    KeEnterCriticalRegion();
+    ExAcquireResourceExclusive(&Lock->Resource, TRUE);
+    KeLeaveCriticalRegion();
+    return STATUS_SUCCESS;
 }
 
 NTSTATUS
@@ -185,7 +188,9 @@ NTAPI
 RtlLeaveHeapLock(
     PHEAP_LOCK Lock)
 {
+    KeEnterCriticalRegion();
     ExReleaseResource(&Lock->Resource);
+    KeLeaveCriticalRegion();
     return STATUS_SUCCESS;
 }