JohnyTheCarrot/ableos
1
0
Fork 0
forked from AbleOS/ableos
Code Issues Pull requests Packages Projects Releases Wiki Activity

i pushed so i can work on it in windows thanks

Browse source
This commit is contained in:
MunirG05 2023-07-10 17:14:11 +05:30
parent 4252719a72
commit 359d62c1b0
3 changed files with 53 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
kernel/src/kmain.rs
View file

@ -6,6 +6,8 @@ use {
arch::logging::SERIAL_CONSOLE, arch::logging::SERIAL_CONSOLE,
bootmodules::{build_cmd, BootModules}, bootmodules::{build_cmd, BootModules},
device_tree::DeviceTree, device_tree::DeviceTree,
scheduler::Scheduler,
}, },
alloc::format, alloc::format,
log::{debug, info, trace}, log::{debug, info, trace},
@ -30,7 +32,7 @@ pub fn kmain(cmdline: &str, boot_modules: BootModules) -> ! {
let dt = DEVICE_TREE.lock(); let dt = DEVICE_TREE.lock();
info!("Device Tree{}", dt); info!("Device Tree: {}", dt);
info!("Boot complete. Moving to init_system"); info!("Boot complete. Moving to init_system");
// TODO: schedule the disk driver from the initramfs // TODO: schedule the disk driver from the initramfs
@ -59,13 +61,17 @@ pub fn kmain(cmdline: &str, boot_modules: BootModules) -> ! {
// log::error!("Program validation error: {e:?}"); // log::error!("Program validation error: {e:?}");
} }
Ok(_) => { Ok(_) => {
// log::info!("valid program"); // log::info!("valid program");
use {crate::host::TrapHandler, hbvm::vm::Vm}; // use {crate::host::TrapHandler, hbvm::vm::Vm};
let mut vm; let mut sched = Scheduler::new();
unsafe { sched.new_process(prog.clone());
vm = Vm::new_unchecked(&prog, TrapHandler); sched.scheduler_loop();
vm.memory.insert_test_page(); // let mut vm;
} // unsafe {
// vm = Vm::new_unchecked(&prog, TrapHandler);
// vm.memory.insert_test_page();
// }
// log::info!("Program interrupt: {:?}", vm.run()); // log::info!("Program interrupt: {:?}", vm.run());
// log::debug!("{:?}", vm.registers); // log::debug!("{:?}", vm.registers);
} }

55
kernel/src/scheduler.rs
View file

@ -1,22 +1,45 @@
use {crate::host::TrapHandler, alloc::vec::Vec, hbvm::vm::Vm}; use alloc::{collections::VecDeque, vec::Vec, rc::Rc, slice};
use hbvm::validate::validate;
use {crate::host::TrapHandler, hbvm::vm::Vm};
pub struct Scheduler<'a> { pub struct Scheduler<'a> {
data: Vec<Vm<'a, TrapHandler>>, data: VecDeque<Vm<'a, TrapHandler>>,
} }
impl Scheduler<'_> { impl Scheduler<'_> {
// fn new_process(&mut self, program: Vec<u8>) { pub fn new() -> Self {
// let prog = program.clone(); Self {
// #[allow(clippy::redundant_else)] data: VecDeque::new(),
// if let Err(e) = validate(&prog) { }
// log::error!("Program validation error: {e:?}"); }
// } else { pub fn new_process(&mut self, program: Vec<u8>) {
// log::info!("valid program"); let prog = program.clone();
// unsafe { let prog_arc = Rc::new(prog);
// let mut vm = Vm::new_unchecked(&prog, TrapHandler);
// vm.memory.insert_test_page(); let binding = Rc::try_unwrap(prog_arc).ok().unwrap();
// self.data.push(vm);
// }
// }
// } #[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 scheduler_loop(&mut self){
loop {
let mut prog = self.data.pop_front().unwrap();
prog.run().unwrap();
self.data.push_back(prog);
}
}
} }