47 lines
1.8 KiB
Rust
47 lines
1.8 KiB
Rust
use prometeu_bytecode::isa::core::CoreOpCode;
|
|
use prometeu_bytecode::{assemble, disassemble};
|
|
|
|
fn emit(op: CoreOpCode, imm: Option<&[u8]>, out: &mut Vec<u8>) {
|
|
out.extend_from_slice(&(op as u16).to_le_bytes());
|
|
if let Some(bytes) = imm {
|
|
out.extend_from_slice(bytes);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn roundtrip_disasm_assemble_byte_equal_with_closures_and_coroutines() {
|
|
// Program: PUSH_I32 7; MAKE_CLOSURE fn=1,captures=0; CALL_CLOSURE argc=1;
|
|
// SPAWN fn=2,argc=1; YIELD; SLEEP 3; HOSTCALL 2; SYSCALL 0x1003; FRAME_SYNC; HALT
|
|
let mut prog = Vec::new();
|
|
|
|
emit(CoreOpCode::PushI32, Some(&7i32.to_le_bytes()), &mut prog);
|
|
// MAKE_CLOSURE (fn=1, captures=0)
|
|
let mut mc = [0u8; 8];
|
|
mc[0..4].copy_from_slice(&1u32.to_le_bytes());
|
|
mc[4..8].copy_from_slice(&0u32.to_le_bytes());
|
|
emit(CoreOpCode::MakeClosure, Some(&mc), &mut prog);
|
|
// CALL_CLOSURE argc=1
|
|
emit(CoreOpCode::CallClosure, Some(&1u32.to_le_bytes()), &mut prog);
|
|
// SPAWN (fn=2, argc=1)
|
|
let mut sp = [0u8; 8];
|
|
sp[0..4].copy_from_slice(&2u32.to_le_bytes());
|
|
sp[4..8].copy_from_slice(&1u32.to_le_bytes());
|
|
emit(CoreOpCode::Spawn, Some(&sp), &mut prog);
|
|
// YIELD
|
|
emit(CoreOpCode::Yield, None, &mut prog);
|
|
// SLEEP 3
|
|
emit(CoreOpCode::Sleep, Some(&3u32.to_le_bytes()), &mut prog);
|
|
// HOSTCALL sysc[2]
|
|
emit(CoreOpCode::Hostcall, Some(&2u32.to_le_bytes()), &mut prog);
|
|
// SYSCALL gfx.draw_line (0x1003)
|
|
emit(CoreOpCode::Syscall, Some(&0x1003u32.to_le_bytes()), &mut prog);
|
|
// FRAME_SYNC
|
|
emit(CoreOpCode::FrameSync, None, &mut prog);
|
|
// HALT
|
|
emit(CoreOpCode::Halt, None, &mut prog);
|
|
|
|
let text = disassemble(&prog).expect("disasm ok");
|
|
let rebuilt = assemble(&text).expect("assemble ok");
|
|
assert_eq!(rebuilt, prog, "re-assembled bytes must match original");
|
|
}
|