From 2d5bb5193d72ec03cbe568d9f1358508abc33e2a Mon Sep 17 00:00:00 2001
From: EnderIce2 <enderice2@users.noreply.github.com>
Date: Wed, 13 Mar 2024 18:45:07 +0200
Subject: [PATCH] Refactor BaseBufferStackError function

---
 core/panic/handler.cpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/core/panic/handler.cpp b/core/panic/handler.cpp
index b78c1c8..1cba9b2 100644
--- a/core/panic/handler.cpp
+++ b/core/panic/handler.cpp
@@ -301,7 +301,7 @@ ExceptionExit:
 	ExceptionLock.store(false, std::memory_order_release);
 }
 
-nsa void BaseBufferStackError()
+nsa void BaseBufferStackError(bool Stack)
 {
 	/* We don't need to restore the page table
 		because the ExceptionHandlerStub will
@@ -326,7 +326,7 @@ nsa void BaseBufferStackError()
 	ForceUnlock = true;
 
 	debug("-----------------------------------------------------------------------------------");
-	error("Buffer overflow detected");
+	error("%s", Stack ? "Stack smashing detected" : "Buffer overflow detected");
 	debug("%ld MiB / %ld MiB (%ld MiB Reserved)",
 		  TO_MiB(KernelAllocator.GetUsedMemory()),
 		  TO_MiB(KernelAllocator.GetTotalMemory()),
@@ -335,14 +335,14 @@ nsa void BaseBufferStackError()
 
 nsa __noreturn void HandleStackSmashing()
 {
-	BaseBufferStackError();
+	BaseBufferStackError(true);
 	DisplayStackSmashing();
 	CPU::Stop();
 }
 
 nsa __noreturn void HandleBufferOverflow()
 {
-	BaseBufferStackError();
+	BaseBufferStackError(false);
 	DisplayBufferOverflow();
 	CPU::Stop();
 }