forked from AbleOS/ableos
59 lines
1.7 KiB
Rust
59 lines
1.7 KiB
Rust
use {
|
|
alloc::{collections::VecDeque, rc::Rc, slice, vec::Vec},
|
|
hbvm::validate::validate,
|
|
};
|
|
|
|
use {crate::host::TrapHandler, hbvm::vm::Vm};
|
|
const TIMER_QUOTIENT: usize = 100;
|
|
|
|
pub struct Scheduler<'a> {
|
|
data: VecDeque<Vm<'a, TrapHandler, TIMER_QUOTIENT>>,
|
|
}
|
|
|
|
// 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 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);
|
|
}
|
|
}
|
|
}
|