arch,traps: set default stack segment selector

Signed-off-by: Pawel Wieczorkiewicz <[email protected]>
KernelTestFramework · Nov 7, 2023 · 6f9a740 · 6f9a740
1 parent a4be3d2
commit 6f9a740
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/arch/x86/traps.c b/arch/x86/traps.c
@@ -112,6 +112,7 @@ static void init_gdt(percpu_t *percpu) {
     barrier();
     lgdt(&percpu->gdt_ptr);
 
+    write_ss(__KERN_DS);
     write_gs(GDT_PERCPU << 3);
     wrmsr(MSR_KERNEL_GS_BASE, 0x0);
 

diff --git a/include/lib.h b/include/lib.h
@@ -239,6 +239,10 @@ static inline unsigned long read_gs(void) {
     return gs;
 }
 
+static inline void write_ss(unsigned long ss) {
+    asm volatile("mov %0, %%ss" ::"r"(ss));
+}
+
 static inline void write_gs(unsigned long gs) {
     asm volatile("mov %0, %%gs" ::"r"(gs));
 }