From af8a516fc22865abd80d6e26f1541fa3d6bebfdc Mon Sep 17 00:00:00 2001
From: Amlal El Mahrouss <amlal.elmahrouss@icloud.com>
Date: Thu, 9 May 2024 00:42:44 +0200
Subject: MHR-23: :boom:, refactors.

- Move NewBoot to /Boot, thus making Kernel directory only containing the kernel.

Signed-off-by: Amlal El Mahrouss <amlal.elmahrouss@icloud.com>
---
 Kernel/Source/ThreadLocalStorage.cxx | 58 ++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 Kernel/Source/ThreadLocalStorage.cxx

(limited to 'Kernel/Source/ThreadLocalStorage.cxx')

diff --git a/Kernel/Source/ThreadLocalStorage.cxx b/Kernel/Source/ThreadLocalStorage.cxx
new file mode 100644
index 00000000..c31ae1c2
--- /dev/null
+++ b/Kernel/Source/ThreadLocalStorage.cxx
@@ -0,0 +1,58 @@
+/*
+ * ========================================================
+ *
+ * NewOS
+ * Copyright SoftwareLabs, all rights reserved.
+ *
+ *  ========================================================
+ */
+
+#include <KernelKit/ProcessScheduler.hpp>
+#include <KernelKit/ThreadLocalStorage.hxx>
+
+///! BUGS: 0
+
+/***********************************************************************************/
+/// @file ThreadLocalStorage.cxx
+/// @brief TLS implementation in kernel.
+/***********************************************************************************/
+
+using namespace NewOS;
+
+/**
+ * @brief Check for cookie inside TIB.
+ * @param tib the TIB to check.
+ * @return if the cookie is enabled.
+ */
+
+Boolean tls_check_tib(ThreadInformationBlock* tib)
+{
+	if (!tib)
+		return false;
+
+	Encoder		encoder;
+	const char* tibAsBytes = encoder.AsBytes(tib);
+
+	kcout << "New OS: Checking for a valid cookie...\r";
+
+	return tibAsBytes[0] == kCookieMag0 && tibAsBytes[1] == kCookieMag1 &&
+		   tibAsBytes[2] == kCookieMag2;
+}
+
+/**
+ * @brief System call implementation of the TLS check.
+ * @param stackPtr The call frame.
+ * @return
+ */
+EXTERN_C Void tls_check_syscall_impl(NewOS::HAL::StackFramePtr stackPtr) noexcept
+{
+	ThreadInformationBlock* tib = (ThreadInformationBlock*)stackPtr->Gs;
+
+	if (!tls_check_tib(tib))
+	{
+		kcout << "New OS: Verification failed, Crashing...\r";
+		ProcessScheduler::Shared().Leak().GetCurrent().Leak().Crash();
+	}
+
+	kcout << "New OS: Verification succeeded! Keeping on...\r";
+}
-- 
cgit v1.2.3