forked from AbleOS/holey-bytes
78 lines
2 KiB
Rust
78 lines
2 KiB
Rust
use {
|
|
hbbytecode::valider::validate,
|
|
hbvm::{
|
|
softpaging::{paging::PageTable, HandlePageFault, PageSize, SoftPagedMem},
|
|
MemoryAccessReason, Vm,
|
|
},
|
|
std::io::{stdin, Read},
|
|
};
|
|
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let mut prog = vec![];
|
|
stdin().read_to_end(&mut prog)?;
|
|
|
|
if let Err(e) = validate(&prog) {
|
|
eprintln!("Program validation error: {e:?}");
|
|
return Ok(());
|
|
} else {
|
|
unsafe {
|
|
let mut vm = Vm::<_, 0>::new(
|
|
SoftPagedMem {
|
|
pf_handler: TestTrapHandler,
|
|
program: &prog,
|
|
root_pt: Box::into_raw(Default::default()),
|
|
},
|
|
4,
|
|
);
|
|
let data = {
|
|
let ptr = std::alloc::alloc_zeroed(std::alloc::Layout::from_size_align_unchecked(
|
|
4096, 4096,
|
|
));
|
|
if ptr.is_null() {
|
|
panic!("Alloc error tbhl");
|
|
}
|
|
ptr
|
|
};
|
|
|
|
vm.memory
|
|
.map(
|
|
data,
|
|
8192,
|
|
hbvm::softpaging::paging::Permission::Write,
|
|
PageSize::Size4K,
|
|
)
|
|
.unwrap();
|
|
|
|
println!("Program interrupt: {:?}", vm.run());
|
|
println!("{:?}", vm.registers);
|
|
|
|
println!("{:?}", core::slice::from_raw_parts(data, 4096));
|
|
std::alloc::dealloc(
|
|
data,
|
|
std::alloc::Layout::from_size_align_unchecked(4096, 4096),
|
|
);
|
|
vm.memory.unmap(8192).unwrap();
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
pub fn time() -> u32 {
|
|
9
|
|
}
|
|
|
|
#[derive(Default)]
|
|
struct TestTrapHandler;
|
|
impl HandlePageFault for TestTrapHandler {
|
|
fn page_fault(
|
|
&mut self,
|
|
_: MemoryAccessReason,
|
|
_: &mut PageTable,
|
|
_: u64,
|
|
_: PageSize,
|
|
_: *mut u8,
|
|
) -> bool {
|
|
false
|
|
}
|
|
}
|