use { alloc::{collections::VecDeque, rc::Rc, slice, vec::Vec}, hbvm::validate::validate, }; use {crate::host::TrapHandler, hbvm::vm::Vm}; const TIMER_QUOTIENT: usize = 10_000_000; pub struct Scheduler<'a> { data: VecDeque>, } // 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) { 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 then sleep. if self.data.is_empty() { use crate::arch::sloop; sloop(); } let mut prog = self.data.pop_front().unwrap(); prog.run().unwrap(); // log::info!("VM registers {:?}", prog.registers); log::info!("Scheduled program"); self.data.push_back(prog); } } }