ableos/kernel/src/scheduler.rs

use {
    alloc::{collections::VecDeque, rc::Rc, slice, vec::Vec},
    hbvm::validate::validate,
};

use {crate::host::TrapHandler, hbvm::vm::Vm};

pub struct Scheduler<'a> {
    data: VecDeque<Vm<'a, TrapHandler>>,
}

// NOTE: This is a very simple schduler and it sucks and should be replaced with a better one
// Written By Yours Truly: Munir

impl Scheduler<'_> {
    pub fn new() -> Self {
        Self {
            data: VecDeque::new(),
        }
    }
    pub fn new_process(&mut self, program: Vec<u8>) {
        let prog = program.clone();
        let prog_arc = Rc::new(prog);

        let binding = Rc::try_unwrap(prog_arc).ok().unwrap();

        #[allow(clippy::redundant_else)]
        if let Err(e) = validate(&program.as_slice()) {
            log::error!("Program validation error: {e:?}");
        } else {
            log::info!("valid program");
            unsafe {
                let slice = slice::from_raw_parts(binding.as_ptr(), binding.len());
                let mut vm = Vm::new_unchecked(&*slice, TrapHandler);
                vm.memory.insert_test_page();
                self.data.push_front(vm);
            }
        }
    }

    pub fn run(&mut self) -> ! {
        loop {
            // If there are no programs to run sleep
            if self.data.is_empty() {
                use crate::arch::sloop;
                sloop();
            }

            let mut prog = self.data.pop_front().unwrap();
            prog.run().unwrap();
            log::trace!("VM registers {:?}", prog.registers);
            self.data.push_back(prog);
        }
    }
}