add statefull scheduler and syscall routine

Signed-off-by: Julien CLEMENT <julien.clement@epita.fr>

src/interrupts/mod.rs

@@ -53,7 +53,41 @@ extern "x86-interrupt" fn breakpoint_handler(stack_frame: InterruptStackFrame) {
 }
 
 extern "x86-interrupt" fn syscall_handler_32(_stack_frame: InterruptStackFrame) {
+    let rax: u64;
+    let rbx: u64;
+    let rcx: u64;
+    let rdx: u64;
+    let rsi: u64;
+    let rdi: u64;
+    let rbp: u64;
+    unsafe {
+        asm!(
+            "push rax",
+            "push rbx",
+            "push rcx",
+            "push rdx",
+            "push rsi",
+            "push rdi",
+            "push rbp",
+
+            "pop {rbp}",
+            "pop {rdi}",
+            "pop {rsi}",
+            "pop {rdx}",
+            "pop {rcx}",
+            "pop {rbx}",
+            "pop {rax}",
+            rax = out(reg) rax,
+            rbx = out(reg) rbx,
+            rcx = out(reg) rcx,
+            rdx = out(reg) rdx,
+            rsi = out(reg) rsi,
+            rdi = out(reg) rdi,
+            rbp = out(reg) rbp,
+        )
+    }
     println!("Received syscall");
+    crate::syscalls::syscall_routine(rax);
 }
 
 extern "x86-interrupt" fn page_fault_handler(

src/proc/scheduler/mod.rs

@@ -1,3 +1,4 @@
+use crate::println;
 use crate::utils::mutex::AsyncMutex;
 
 use super::thread::{Thread, ThreadId};
@@ -23,7 +24,7 @@ pub const K_THREAD_ID: ThreadId = ThreadId(0); // Kernel main thread identifier
 
 struct ThreadStream {
     ids: ArrayQueue<ThreadId>,
-    waker: AtomicWaker
+    waker: AtomicWaker,
 }
 
 impl ThreadStream {
@@ -35,9 +36,7 @@ impl ThreadStream {
     }
 
     pub fn register(&mut self, id: ThreadId) {
-        self.ids
+        self.ids.push(id).expect("Thread queue full");
-            .push(id)
-            .expect("Thread queue full");
         self.waker.wake();
     }
 }
@@ -79,19 +78,29 @@ impl Scheduler {
             started: true,
             rsp: 0,
             base_stack: 0,
+            is_blocked: false,
         };
         res.register(Arc::new(RefCell::new(k_thread)));
         res
     }
 
-
     pub async fn run(&mut self) {
         while let Some(id) = self.thread_queue.next().await {
-            if let Some(thread) = self.get_thread(id) { // Thread still exists
+            if let Some(thread) = self.get_thread(id) {
-                unsafe {
+                // Thread still exists
-                    (&mut*thread.as_ptr()).run();
+                let blocked: bool;
+                {
+                    blocked = thread.borrow().is_blocked;
+                } // Drop thread borrow
+                if !blocked {
+                    unsafe {
+                        (&mut *thread.as_ptr()).run();
+                    }
+
+                    if !thread.borrow().is_blocked {
+                        self.thread_queue.register(id);
+                    }
                 }
-                self.thread_queue.register(id);
             }
         }
     }
@@ -106,6 +115,21 @@ impl Scheduler {
         }
     }
 
+    pub fn block(&mut self, id: ThreadId) {
+        if let Some(thread) = self.get_thread(id) {
+            thread.borrow_mut().is_blocked = true;
+            println!("Blocked thread {:?}", id);
+        }
+    }
+
+    pub fn unblock(&mut self, id: ThreadId) {
+        if let Some(thread) = self.get_thread(id) {
+            thread.borrow_mut().is_blocked = false;
+            self.thread_queue.register(id);
+            println!("Unblocked thread {:?}", id);
+        }
+    }
+
     pub fn exit(&mut self, id: ThreadId) {
         self.threads.remove(&id).unwrap().borrow().exit();
     }

src/proc/thread/mod.rs

@@ -1,7 +1,7 @@
 use crate::println;
 use crate::utils::mutex::AsyncMutex;
 
-use super::scheduler::{SCHEDULER, K_THREAD_ID};
+use super::scheduler::{K_THREAD_ID, SCHEDULER};
 
 use core::arch::asm;
 use core::sync::atomic::{AtomicU64, Ordering};
@@ -25,34 +25,22 @@ impl ThreadId {
     }
 }
 
-pub fn exit() {
-    println!("Exiting thread");
-    {
-        let mut scheduler = SCHEDULER.try_lock().unwrap();
-        scheduler.exit(*RUNNING_THREAD.try_lock().unwrap());
-    } // Drop scheduler mutex guard
-
-    resume_k_thread();
-}
-
 pub fn resume_k_thread() {
     let k_thread: *mut Thread;
     {
         let mut scheduler = SCHEDULER.try_lock().unwrap();
-        k_thread = scheduler
+        k_thread = scheduler.get_thread(K_THREAD_ID).unwrap().as_ptr();
-            .get_thread(K_THREAD_ID)
-            .unwrap()
-            .as_ptr();
     } // Drop scheduler mutex guard
 
     unsafe {
-        (&mut* k_thread).run();
+        (&mut *k_thread).run();
     }
 }
 
 pub fn routine() {
     println!("Routine executed");
-    exit();
+    crate::syscalls::syscall_routine(crate::syscalls::EXIT_ID); // Call exit
+    println!("SHOULD NEVER BE DISPLAYED");
 }
 
 pub struct Thread {
@@ -60,7 +48,8 @@ pub struct Thread {
     pub entry_point: u64,
     pub started: bool,
     pub rsp: u64,
-    pub base_stack: u64
+    pub base_stack: u64,
+    pub is_blocked: bool,
 }
 
 impl Thread {
@@ -73,13 +62,17 @@ impl Thread {
                 started: false,
                 rsp: stack_bottom + STACK_SIZE as u64,
                 base_stack: stack_bottom,
+                is_blocked: false
             }
         }
     }
 
     pub fn exit(&self) {
         unsafe {
-            dealloc(self.base_stack as *mut u8, Layout::new::<[u8; STACK_SIZE]>());
+            dealloc(
+                self.base_stack as *mut u8,
+                Layout::new::<[u8; STACK_SIZE]>(),
+            );
         }
     }
 
@@ -98,8 +91,8 @@ impl Thread {
                     out = out(reg) current_rsp, // Save thread rsp
                 );
                 current_thread.borrow_mut().rsp = current_rsp;
-            }
+            } else {
-            else { // Thread does not exists anymore
+                // Thread does not exists anymore
                 *current_thread_guard = self.id; // change running thread
                 asm!( // Just switch to new thead without saving registers
                     "push {rsp}", // Set stack pointer to the new thread

src/syscalls/ids.rs

@@ -0,0 +1,3 @@
+pub type SyscallId = u64;
+
+pub const EXIT_ID: u64 = 0;

src/syscalls/mod.rs

@@ -1,5 +1,71 @@
-pub mod io;
+use crate::println;
+use crate::proc::thread::{resume_k_thread, RUNNING_THREAD};
+use crate::proc::scheduler::SCHEDULER;
+use crate::task::executor::EXECUTOR;
+use crate::task::Task;
 
-pub fn syscall_dispatcher() -> u64 {
+pub use ids::*;
-    0
+
+use alloc::sync::Arc;
+use core::cell::RefCell;
+
+pub mod ids;
+pub mod io;
+pub mod proc;
+
+pub type SyscallContextT = Arc<RefCell<SyscallContext>>;
+
+pub struct SyscallContext {
+    id: SyscallId,
+    res: u64,
+    thread_id: crate::proc::thread::ThreadId,
 }
+
+impl SyscallContext {
+    pub async fn run(&mut self) {
+        println!("Running async syscall runner");
+        self.dispatch().await;
+        println!("Syscall end, unblocking thread");
+        SCHEDULER.lock().await.unblock(self.thread_id);
+    }
+
+    pub async fn dispatch(&mut self) {
+        match self.id {
+            EXIT_ID => proc::exit(self).await,
+            _ => bad_syscall().await,
+        }
+    }
+}
+
+
+async fn syscall_runner(context: SyscallContextT) {
+    context.borrow_mut().run().await;
+}
+
+pub fn syscall_routine(syscall_id: SyscallId) -> u64 {
+    println!("Running syscall");
+    let context: SyscallContextT = Arc::new(RefCell::new(SyscallContext {
+        id: syscall_id,
+        res: 0,
+        thread_id: *RUNNING_THREAD.try_lock().unwrap(),
+    }));
+    
+    println!("Spawning async syscall runner");
+    EXECUTOR
+        .try_lock()
+        .unwrap()
+        .spawn(Task::new(syscall_runner(context.clone())));
+
+    println!("Blocking thread");
+    SCHEDULER
+        .try_lock()
+        .unwrap()
+        .block(context.borrow().thread_id);
+    println!("Returning to scheduler");
+    resume_k_thread();
+
+    let res = context.borrow().res;
+    res
+}
+
+async fn bad_syscall() {}

src/syscalls/proc.rs

@@ -0,0 +1,10 @@
+use crate::println;
+use crate::proc::scheduler::SCHEDULER;
+
+use super::SyscallContext;
+
+pub async fn exit(context: &SyscallContext) {
+    println!("Running exit(2)");
+    let mut scheduler = SCHEDULER.lock().await;
+    scheduler.exit(context.thread_id);
+}