ADD: Better code and tweaks to scheduler code.

1 files changed, 17 insertions, 9 deletions

diff --git a/dev/Kernel/src/UserProcessScheduler.cc b/dev/Kernel/src/UserProcessScheduler.cc
index 6cd7fb0b..2d3101f2 100644
--- a/dev/Kernel/src/UserProcessScheduler.cc
+++ b/dev/Kernel/src/UserProcessScheduler.cc
@@ -248,7 +248,7 @@ namespace Kernel
 		this->Image.fCode = nullptr;
 		this->StackFrame  = nullptr;
 
-		if (this->Kind == kExectuableDLLKind)
+		if (this->Kind == kExectuableDylibKind)
 		{
 			Bool success = false;
 
@@ -329,15 +329,29 @@ namespace Kernel
 		HAL::mm_map_page((VoidPtr)process.StackFrame, flags);
 #endif // __ZKA_VIRTUAL_MEMORY_SUPPORT__
 
-		// Create heap according to type of process.
-		if (process.Kind == UserThread::kExectuableDLLKind)
+		// React according to process kind.
+		switch (process.kind) 
+		{
+		case UserThread::kExectuableDylibKind:
 		{
 			process.DylibDelegate = rtl_init_dylib(process);
 			MUST_PASS(process.DylibDelegate);
 		}
+		default:
+		{
+			kcout << "Unknown process kind: " << number(process.Kind) << endl;
+			break;
+		}
+		}
 
 		process.StackReserve = new UInt8[process.StackSize];
 
+		if (!process.StackReserve)
+		{
+			process.Crash();
+			return kErrorProcessFault;
+		}
+
 #ifdef __ZKA_VIRTUAL_MEMORY_SUPPORT__
 		flags = HAL::kMMFlagsPresent;
 		flags |= HAL::kMMFlagsWr;
@@ -346,12 +360,6 @@ namespace Kernel
 		HAL::mm_map_page((VoidPtr)process.StackReserve, flags);
 #endif // __ZKA_VIRTUAL_MEMORY_SUPPORT__
 
-		if (!process.StackReserve)
-		{
-			process.Crash();
-			return kErrorProcessFault;
-		}
-
 		process.ProcessParentTeam = &mTeam;
 
 		process.ProcessId = pid;