From 3e5c2bcacd96b5e10213caacf37f33a39c8ad46a Mon Sep 17 00:00:00 2001
From: Alex <enderice2@users.noreply.github.com>
Date: Mon, 7 Nov 2022 11:27:35 +0200
Subject: [PATCH] If empty, set default

---
 Architecture/amd64/cpu/GlobalDescriptorTable.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/Architecture/amd64/cpu/GlobalDescriptorTable.cpp b/Architecture/amd64/cpu/GlobalDescriptorTable.cpp
index 647e996..eadfc25 100644
--- a/Architecture/amd64/cpu/GlobalDescriptorTable.cpp
+++ b/Architecture/amd64/cpu/GlobalDescriptorTable.cpp
@@ -145,6 +145,9 @@ namespace GlobalDescriptorTable
 
     __attribute__((no_stack_protector)) void SetKernelStack(void *Stack)
     {
-        tss[GetCurrentCPU()->ID].StackPointer[0] = (uint64_t)Stack;
+        if (Stack)
+            tss[GetCurrentCPU()->ID].StackPointer[0] = (uint64_t)Stack;
+        else
+            tss[GetCurrentCPU()->ID].StackPointer[0] = (uint64_t)CPUStackPointer[GetCurrentCPU()->ID] + STACK_SIZE;
     }
 }