From 1a8cf7342c0e7954285c3751bc54d2b685599cdd Mon Sep 17 00:00:00 2001
From: Aleksey Bragin <aleksey@reactos.org>
Date: Sun, 9 Mar 2008 21:15:12 +0000
Subject: [PATCH] - Move MiGetPfnEntry into arch-independent mm.h header, along
 with a few defines, structures, and function prototypes which are going to be
 used with the new pool implementation in future. - Add two new fields into
 PHYSICAL_PAGE structure, they will be used (in the future new pool
 implementation) to find end/start of a non-paged pool allocation without
 storing the actual pointers (based on NTs own PFN entry optimization for
 this) - thanks to Alex for providing this great idea!

svn path=/trunk/; revision=32630
---
 reactos/ntoskrnl/include/internal/mm.h | 72 ++++++++++++++++++++++++++
 reactos/ntoskrnl/mm/freelist.c         | 24 +--------
 2 files changed, 73 insertions(+), 23 deletions(-)

diff --git a/reactos/ntoskrnl/include/internal/mm.h b/reactos/ntoskrnl/include/internal/mm.h
index b897fb77d30..fcc52a9007a 100644
--- a/reactos/ntoskrnl/include/internal/mm.h
+++ b/reactos/ntoskrnl/include/internal/mm.h
@@ -56,6 +56,12 @@ typedef ULONG PFN_TYPE, *PPFN_TYPE;
 #define MM_PAGEOP_PAGESYNCH                 (3)
 #define MM_PAGEOP_ACCESSFAULT               (4)
 
+/* Number of list heads to use */
+#define MI_FREE_POOL_LISTS 4
+
+/* Signature of free pool blocks */
+#define MM_FREE_POOL_TAG    TAG('F', 'r', 'p', 'l')
+
 #define PAGE_TO_SECTION_PAGE_DIRECTORY_OFFSET(x) \
     ((x) / (4*1024*1024))
 
@@ -275,6 +281,8 @@ typedef struct _PHYSICAL_PAGE
             ULONG Type: 2;
             ULONG Consumer: 3;
             ULONG Zero: 1;
+            ULONG StartOfAllocation: 1;
+            ULONG EndOfAllocation: 1;
         }
         Flags;
         ULONG AllFlags;
@@ -337,6 +345,28 @@ typedef struct _MM_REGION
     LIST_ENTRY RegionListEntry;
 } MM_REGION, *PMM_REGION;
 
+/* Entry describing free pool memory */
+typedef struct _MMFREE_POOL_ENTRY
+{
+    LIST_ENTRY List;
+    PFN_NUMBER Size;
+    ULONG Signature;
+    struct _MMFREE_POOL_ENTRY *Owner;
+} MMFREE_POOL_ENTRY, *PMMFREE_POOL_ENTRY;
+
+/* Paged pool information */
+typedef struct _MM_PAGED_POOL_INFO
+{  
+    PRTL_BITMAP PagedPoolAllocationMap;
+    PRTL_BITMAP EndOfPagedPoolBitmap;
+    PMMPTE FirstPteForPagedPool;
+    PMMPTE LastPteForPagedPool;
+    PMMPTE NextPdeForPagedPoolExpansion;
+    ULONG PagedPoolHint;
+    SIZE_T PagedPoolCommit;
+    SIZE_T AllocatedPagedPool;
+} MM_PAGED_POOL_INFO, *PMM_PAGED_POOL_INFO;
+
 extern MM_MEMORY_CONSUMER MiMemoryConsumers[MC_MAXIMUM];
 
 typedef VOID
@@ -487,6 +517,25 @@ VOID
 NTAPI
 MiInitializeNonPagedPool(VOID);
 
+PVOID
+NTAPI
+MiAllocatePoolPages(
+    IN POOL_TYPE PoolType,
+    IN SIZE_T SizeInBytes
+);
+
+POOL_TYPE
+NTAPI
+MmDeterminePoolType(
+    IN PVOID VirtualAddress
+);
+
+ULONG
+NTAPI
+MiFreePoolPages(
+    IN PVOID StartingAddress
+);
+
 PVOID
 NTAPI
 MmGetMdlPageAddress(
@@ -936,6 +985,29 @@ MmPageOutPhysicalAddress(PFN_TYPE Page);
 
 /* freelist.c **********************************************************/
 
+#define ASSERT_PFN(x) ASSERT((x)->Flags.Type != 0)
+
+FORCEINLINE
+PPHYSICAL_PAGE
+MiGetPfnEntry(IN PFN_TYPE Pfn)
+{
+    PPHYSICAL_PAGE Page;
+    extern PPHYSICAL_PAGE MmPageArray;
+    extern ULONG MmPageArraySize;
+
+    /* Make sure the PFN number is valid */
+    ASSERT(Pfn <= MmPageArraySize);
+
+    /* Get the entry */
+    Page = &MmPageArray[Pfn];
+
+    /* Make sure it's valid */
+    ASSERT_PFN(Page);
+
+    /* Return it */
+    return Page;
+};
+
 PFN_TYPE
 NTAPI
 MmGetLRUNextUserPage(PFN_TYPE PreviousPage);
diff --git a/reactos/ntoskrnl/mm/freelist.c b/reactos/ntoskrnl/mm/freelist.c
index 3d1bc323381..c0567fa21ce 100644
--- a/reactos/ntoskrnl/mm/freelist.c
+++ b/reactos/ntoskrnl/mm/freelist.c
@@ -25,12 +25,9 @@
 #define MM_PHYSICAL_PAGE_USED    (0x2)
 #define MM_PHYSICAL_PAGE_BIOS    (0x3)
 
-
-#define ASSERT_PFN(x) ASSERT((x)->Flags.Type != 0)
-
 /* GLOBALS ****************************************************************/
 
-static PPHYSICAL_PAGE MmPageArray;
+PPHYSICAL_PAGE MmPageArray;
 ULONG MmPageArraySize;
 
 static KSPIN_LOCK PageListLock;
@@ -45,25 +42,6 @@ static ULONG UnzeroedPageCount = 0;
 
 /* FUNCTIONS *************************************************************/
 
-FORCEINLINE
-PPHYSICAL_PAGE
-MiGetPfnEntry(IN PFN_TYPE Pfn)
-{
-    PPHYSICAL_PAGE Page;
-
-    /* Make sure the PFN number is valid */
-    ASSERT(Pfn <= MmPageArraySize);
-
-    /* Get the entry */
-    Page = &MmPageArray[Pfn];
-
-    /* Make sure it's valid */
-    ASSERT_PFN(Page);
-
-    /* Return it */
-    return Page;
-}
-
 PFN_TYPE
 NTAPI
 MmGetLRUFirstUserPage(VOID)