From dc6c85c9344732a12f5e8f74e4cfc374b8bd391f Mon Sep 17 00:00:00 2001
From: Amlal <amlal@nekernel.org>
Date: Thu, 1 May 2025 12:07:58 +0200
Subject: kernel: scheduler and pef code mgr improvements.

Signed-off-by: Amlal <amlal@nekernel.org>
---
 dev/kernel/src/PEFCodeMgr.cc | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'dev/kernel/src/PEFCodeMgr.cc')

diff --git a/dev/kernel/src/PEFCodeMgr.cc b/dev/kernel/src/PEFCodeMgr.cc
index 632d5baa..40a0cff9 100644
--- a/dev/kernel/src/PEFCodeMgr.cc
+++ b/dev/kernel/src/PEFCodeMgr.cc
@@ -12,6 +12,7 @@
 #include <NewKit/KString.h>
 #include <NewKit/KernelPanic.h>
 #include <NewKit/OwnPtr.h>
+#include "KernelKit/UserProcessScheduler.h"
 
 /// @brief PEF stack size symbol.
 #define kPefStackSizeSymbol "__PEFSizeOfReserveStack"
@@ -242,7 +243,11 @@ namespace Utils {
       auto stacksym = exec.FindSymbol(kPefStackSizeSymbol, kPefData);
 
       if (!symname) {
-        stacksym = ErrorOr<VoidPtr>{(VoidPtr) new UIntPtr(mib_cast(16))};
+        stacksym = ErrorOr<VoidPtr>{(VoidPtr) new UIntPtr(kSchedMaxStackSz)};
+      }
+
+      if ((*(volatile UIntPtr*)stacksym.Leak().Leak()) > kSchedMaxStackSz) {
+        *(volatile UIntPtr*)stacksym.Leak().Leak() = kSchedMaxStackSz;
       }
 
       UserProcessScheduler::The().CurrentTeam().AsArray()[id].Kind = process_kind;
-- 
cgit v1.2.3