Initialize TLS for idle task

Author: jounathaen

src/arch/x86_64/kernel/scheduler.rs

@@ -16,7 +16,9 @@ use crate::arch::x86_64::mm::paging::{
 use crate::config::*;
 use crate::env;
 use crate::mm::{FrameAlloc, PageAlloc, PageRangeAllocator};
+use crate::processor::writefs;
 use crate::scheduler::PerCoreSchedulerExt;
+use crate::scheduler::task::tls::Tls;
 use crate::scheduler::task::{Task, TaskFrame};
 
 #[repr(C, packed)]
@@ -332,3 +334,8 @@ pub fn install_timer_handler() {
 	}
 	interrupts::add_irq_name(apic::TIMER_INTERRUPT_NUMBER - 32, "Timer");
 }
+
+/// Initializes the TLS for the init/boot task and sets FS register to the respective value
+pub fn set_init_task_tls() -> Option<Tls> {
+	Tls::from_env().inspect(|tls| writefs(tls.thread_ptr().addr()))
+}

src/scheduler/task/mod.rs

@@ -19,7 +19,7 @@ use memory_addresses::VirtAddr;
 #[cfg(not(feature = "common-os"))]
 use self::tls::Tls;
 use crate::arch::core_local::*;
-use crate::arch::scheduler::TaskStacks;
+use crate::arch::scheduler::{TaskStacks, set_init_task_tls};
 use crate::fd::stdio::*;
 use crate::fd::{ObjectInterface, RawFd, STDERR_FILENO, STDIN_FILENO, STDOUT_FILENO};
 use crate::scheduler::CoreId;
@@ -488,7 +488,7 @@ impl Task {
 			stacks: TaskStacks::from_boot_stacks(),
 			object_map: OBJECT_MAP.get().unwrap().clone(),
 			#[cfg(not(feature = "common-os"))]
-			tls: None,
+			tls: set_init_task_tls(),
 			#[cfg(all(target_arch = "x86_64", feature = "common-os"))]
 			root_page_table: *crate::scheduler::BOOT_ROOT_PAGE_TABLE.get().unwrap(),
 		}