[reactos] 02/02: [NTOSKRNL] Properly check for address validity in !poolfind

[Pierre Schweitzer]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=28193399ee039b62a03f8a...

commit 28193399ee039b62a03f8ad8c6b23d6bb04facfe
Author:     Pierre Schweitzer [e-mail]
AuthorDate: Sun Jan 6 12:58:14 2019 +0100
Commit:     Pierre Schweitzer [e-mail]
CommitDate: Sun Jan 6 12:58:14 2019 +0100

[NTOSKRNL] Properly check for address validity in !poolfind
---
 ntoskrnl/mm/ARM3/expool.c | 54 ++++++++++++++++++-----------------------------
 1 file changed, 21 insertions(+), 33 deletions(-)

diff --git a/ntoskrnl/mm/ARM3/expool.c b/ntoskrnl/mm/ARM3/expool.c
index e03645387a..e69be03cbf 100644
--- a/ntoskrnl/mm/ARM3/expool.c
+++ b/ntoskrnl/mm/ARM3/expool.c
@@ -3163,7 +3163,6 @@ ExpKdbgExtPoolFindPagedPool(
     ULONG i = 0;
     PPOOL_HEADER Entry;
     PVOID BaseVa;
-    PMMPTE PointerPte;
     PMMPDE PointerPde;

KdbpPrint("Searching Paged pool (%p : %p) for Tag: %.4s\n", MmPagedPoolStart, MmPagedPoolEnd, (PCHAR)&Tag);
@@ -3192,13 +3191,7 @@ ExpKdbgExtPoolFindPagedPool(
         }

/* Check if allocation is valid */
-        PointerPte = MiAddressToPte(BaseVa);
-        if ((ULONG_PTR)PointerPte > PTE_TOP)
-        {
-            break;
-        }
-
-        if (PointerPte->u.Hard.Valid)
+        if (MmIsAddressValid(BaseVa))
         {
             for (Entry = BaseVa;
                  (ULONG_PTR)Entry + sizeof(POOL_HEADER) < (ULONG_PTR)BaseVa + PAGE_SIZE;
@@ -3243,7 +3236,6 @@ ExpKdbgExtPoolFindNonPagedPool(
 {
     PPOOL_HEADER Entry;
     PVOID BaseVa;
-    PMMPTE PointerPte;

KdbpPrint("Searching NonPaged pool (%p : %p) for Tag: %.4s\n", MmNonPagedPoolStart, MmNonPagedPoolEnd0, (PCHAR)&Tag);

@@ -3261,38 +3253,34 @@ ExpKdbgExtPoolFindNonPagedPool(
         }

/* Check if allocation is valid */
-        PointerPte = MiAddressToPte(BaseVa);
-        if ((ULONG_PTR)PointerPte > PTE_TOP)
+        if (!MmIsAddressValid(BaseVa))
         {
-            break;
+            continue;
         }

-        if (PointerPte->u.Hard.Valid)
+        for (Entry = BaseVa;
+             (ULONG_PTR)Entry + sizeof(POOL_HEADER) < (ULONG_PTR)BaseVa + PAGE_SIZE;
+             Entry = (PVOID)((ULONG_PTR)Entry + 8))
         {
-            for (Entry = BaseVa;
-                 (ULONG_PTR)Entry + sizeof(POOL_HEADER) < (ULONG_PTR)BaseVa + PAGE_SIZE;
-                 Entry = (PVOID)((ULONG_PTR)Entry + 8))
+            /* Try to find whether we have a pool entry */
+            if (!ExpKdbgExtValidatePoolHeader(BaseVa, Entry, NonPagedPool))
             {
-                /* Try to find whether we have a pool entry */
-                if (!ExpKdbgExtValidatePoolHeader(BaseVa, Entry, NonPagedPool))
+                continue;
+            }
+
+            if ((Entry->PoolTag & Mask) == (Tag & Mask))
+            {
+                if (FoundCallback != NULL)
                 {
-                    continue;
+                    FoundCallback(Entry, CallbackContext);
                 }
-
-                if ((Entry->PoolTag & Mask) == (Tag & Mask))
+                else
                 {
-                    if (FoundCallback != NULL)
-                    {
-                        FoundCallback(Entry, CallbackContext);
-                    }
-                    else
-                    {
-                        /* Print the line */
-                        KdbpPrint("%p size: %4d previous size: %4d  %s  %.4s\n",
-                                  Entry, Entry->BlockSize, Entry->PreviousSize,
-                                  Entry->PoolType ? "(Allocated)" : "(Free)     ",
-                                  (PCHAR)&Entry->PoolTag);
-                    }
+                    /* Print the line */
+                    KdbpPrint("%p size: %4d previous size: %4d  %s  %.4s\n",
+                              Entry, Entry->BlockSize, Entry->PreviousSize,
+                              Entry->PoolType ? "(Allocated)" : "(Free)     ",
+                              (PCHAR)&Entry->PoolTag);
                 }
             }
         }