add register validation and asserts

1 files changed, 22 insertions, 1 deletions

diff --git a/arch/x86_64/src/kernel/cpu/segment_register.cpp b/arch/x86_64/src/kernel/cpu/segment_register.cpp
index 9fb7433..cb367b6 100644
--- a/arch/x86_64/src/kernel/cpu/segment_register.cpp
+++ b/arch/x86_64/src/kernel/cpu/segment_register.cpp
@@ -1,6 +1,7 @@
 #include "arch/kernel/cpu/segment_register.hpp"
 
 #include "arch/context_switching/interrupt_descriptor_table/segment_selector.hpp"
+#include "arch/exception_handling/assert.hpp"
 
 namespace teachos::arch::kernel::cpu
 {
@@ -37,6 +38,26 @@ namespace teachos::arch::kernel::cpu
     return current_value;
   }
 
-  auto validate_segment_registers() -> context_switching::interrupt_descriptor_table::segment_selector {}
+  auto validate_data_segment_registers() -> context_switching::interrupt_descriptor_table::segment_selector
+  {
+    context_switching::interrupt_descriptor_table::segment_selector ss;
+    context_switching::interrupt_descriptor_table::segment_selector ds;
+    context_switching::interrupt_descriptor_table::segment_selector es;
+    context_switching::interrupt_descriptor_table::segment_selector fs;
+    context_switching::interrupt_descriptor_table::segment_selector gs;
+
+    asm volatile(
+        "mov %%ss, %[ss_output]\n"
+        "mov %%ds, %[ds_output]\n"
+        "mov %%es, %[es_output]\n"
+        "mov %%fs, %[fs_output]\n"
+        "mov %%gs, %[gs_output]\n"
+        : [ss_output] "=r"(ss), [ds_output] "=r"(ds), [es_output] "=r"(es), [fs_output] "=r"(fs), [gs_output] "=r"(gs));
+
+    auto result = ss == ds && ss == es && ss == fs && ss == gs;
+    exception_handling::assert(result, "[Segment Register] Values in data register are not the same.");
+
+    return ss;
+  }
 
 }  // namespace teachos::arch::kernel::cpu