From 5fe0740c2ec80e97e8f05bdf04c7e4b9028507f5 Mon Sep 17 00:00:00 2001
From: Aiken Harris <harraiken91@gmail.com>
Date: Tue, 10 Mar 2026 23:09:40 +0100
Subject: [PATCH] Initialize system PTE pool for non-paged expansion pool

---
 xtoskrnl/mm/alloc.cc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/xtoskrnl/mm/alloc.cc b/xtoskrnl/mm/alloc.cc
index 26a46c4..d610a8e 100644
--- a/xtoskrnl/mm/alloc.cc
+++ b/xtoskrnl/mm/alloc.cc
@@ -548,6 +548,11 @@ MM::Allocator::InitializeNonPagedPool(VOID)
     /* Store first and last allocated non-paged pool page */
     NonPagedPoolFrameStart = MM::Paging::GetPageFrameNumber(MM::Paging::GetPteAddress(MemoryLayout->NonPagedPoolStart));
     NonPagedPoolFrameEnd = MM::Paging::GetPageFrameNumber(MM::Paging::GetPteAddress(MemoryLayout->NonPagedPoolEnd));
+
+    /* Initialize system PTE pool for the non-paged expansion pool */
+    Pte::InitializeSystemPtePool(Paging::GetNextPte(Paging::GetPteAddress(MemoryLayout->NonPagedExpansionPoolStart)),
+                                                                          MemoryLayout->NonPagedExpansionPoolSize - 2,
+                                                                          NonPagedPoolExpansion);
 }
 
 /**