src/interrupts/mod.rs

@@ -53,41 +53,7 @@ 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,4 +1,3 @@
-use crate::println;
 use crate::utils::mutex::AsyncMutex;
 
 use super::thread::{Thread, ThreadId};
@@ -24,7 +23,7 @@ pub const K_THREAD_ID: ThreadId = ThreadId(0); // Kernel main thread identifier
 
 struct ThreadStream {
     ids: ArrayQueue<ThreadId>,
-    waker: AtomicWaker,
+    waker: AtomicWaker
 }
 
 impl ThreadStream {
@@ -36,7 +35,9 @@ impl ThreadStream {
     }
 
     pub fn register(&mut self, id: ThreadId) {
-        self.ids.push(id).expect("Thread queue full");
+        self.ids
+            .push(id)
+            .expect("Thread queue full");
         self.waker.wake();
     }
 }
@@ -78,29 +79,19 @@ 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) {
+            if let Some(thread) = self.get_thread(id) { // Thread still exists
-                // Thread still exists
+                unsafe {
-                let blocked: bool;
+                    (&mut*thread.as_ptr()).run();
-                {
-                    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);
             }
         }
     }
@@ -115,21 +106,6 @@ 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();
     }
@@ -147,4 +123,4 @@ pub async fn scheduler_run() {
     let mut scheduler = SCHEDULER.lock().await;
     SCHEDULER.force_unlock();
     scheduler.run().await;
-}
+}

src/proc/thread/mod.rs

@@ -1,7 +1,7 @@
 use crate::println;
 use crate::utils::mutex::AsyncMutex;
 
-use super::scheduler::{K_THREAD_ID, SCHEDULER};
+use super::scheduler::{SCHEDULER, K_THREAD_ID};
 
 use core::arch::asm;
 use core::sync::atomic::{AtomicU64, Ordering};
@@ -25,22 +25,34 @@ 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.get_thread(K_THREAD_ID).unwrap().as_ptr();
+        k_thread = scheduler
+            .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");
-    crate::syscalls::syscall_routine(crate::syscalls::EXIT_ID); // Call exit
+    exit();
-    println!("SHOULD NEVER BE DISPLAYED");
 }
 
 pub struct Thread {
@@ -48,8 +60,7 @@ 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 {
@@ -62,17 +73,13 @@ impl Thread {
                 started: false,
                 rsp: stack_bottom + STACK_SIZE as u64,
                 base_stack: stack_bottom,
-                is_blocked: false
             }
         }
     }
 
     pub fn exit(&self) {
         unsafe {
-            dealloc(
+            dealloc(self.base_stack as *mut u8, Layout::new::<[u8; STACK_SIZE]>());
-                self.base_stack as *mut u8,
-                Layout::new::<[u8; STACK_SIZE]>(),
-            );
         }
     }
 
@@ -91,8 +98,8 @@ impl Thread {
                     out = out(reg) current_rsp, // Save thread rsp
                 );
                 current_thread.borrow_mut().rsp = current_rsp;
-            } else {
+            }
-                // Thread does not exists anymore
+            else { // 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

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

src/syscalls/mod.rs

@@ -1,71 +1,5 @@
-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 use ids::*;
-
-use alloc::sync::Arc;
-use core::cell::RefCell;
-
-pub mod ids;
 pub mod io;
-pub mod proc;
 
-pub type SyscallContextT = Arc<RefCell<SyscallContext>>;
+pub fn syscall_dispatcher() -> u64 {
-
+    0
-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

@@ -1,10 +0,0 @@
-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);
-}