Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add tests to invalid dynamic jumps #145

Open
wants to merge 11 commits into
base: master
Choose a base branch
from
Open

add tests to invalid dynamic jumps #145

Changes from 1 commit
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
69242ef
Test: Add test to verify invalid jump triggers ExecutionError::Trap
alk0x1 Jul 10, 2024
6940e77
test: test correct trap of dynamic jumps
alk0x1 Jul 12, 2024
42b929d
zygote: handle invalid dynamic jumps as a ExecutionError::Trap
alk0x1 Jul 14, 2024
12474f0
tests: add test to invalid dynamic jumps
alk0x1 Jul 18, 2024
349282b
Update main.rs
alk0x1 Jul 18, 2024
0a7192d
Update main.rs
alk0x1 Jul 18, 2024
7a85176
Update crates/polkavm/src/tests.rs
alk0x1 Jul 22, 2024
1effdbb
Update crates/polkavm/src/tests.rs
alk0x1 Jul 22, 2024
801cdb0
Update crates/polkavm/src/tests.rs
alk0x1 Jul 22, 2024
3c0e8d3
Update tests.rs
alk0x1 Jul 24, 2024
514be6a
Update tests.rs
alk0x1 Jul 24, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
42 changes: 42 additions & 0 deletions crates/polkavm/src/tests.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -517,6 +517,46 @@ impl TestInstance {
}
}

fn basic_test_blob_invalid_jump() -> ProgramBlob {
let memory_map = MemoryMap::new(0x4000, 0, 0x4000, 0).unwrap();
let mut builder = ProgramBlobBuilder::new();
builder.set_rw_data_size(0x4000);
builder.add_export_by_basic_block(0, b"main");
builder.set_code(
&[
asm::store_imm_u32(memory_map.rw_data_address(), 0x12345678),
asm::add(S0, A0, A1),
asm::jump(0xFFFFFFFF), // This should be an invalid jump address.
asm::add(A0, A0, S0),
asm::ret(),
],
&[],
);
ProgramBlob::parse(builder.into_vec().into()).unwrap()
}

fn test_invalid_jump_trap(config: Config) {
let _ = env_logger::try_init();
let engine = Engine::new(&config).unwrap();
let linker: Linker<()> = Linker::new(&engine);

let invalid_jump_blob = basic_test_blob_invalid_jump();

let module = Module::from_blob(&engine, &Default::default(), invalid_jump_blob).unwrap();
let instance_pre = linker.instantiate_pre(&module).unwrap();
let instance = instance_pre.instantiate().unwrap();

let state_args = StateArgs::default();
let ext_main = instance.module().lookup_export("main").unwrap();

let result = instance.call(state_args, CallArgs::new(&mut (), ext_main));

match result {
Ok(()) => panic!("Expected ExecutionError::Trap, but got Ok"),
Err(e) => assert!(matches!(e, ExecutionError::Trap(_)), "Expected ExecutionError::Trap, but got {:?}", e),
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

More idiomatic Rust:

if let Err(e) == instance.call(state_args, CallArgs::new(&mut (), ext_main)) {
    assert!(matches!(e, ExecutionError::Trap(_)), "Expected ExecutionError::Trap, but got {:?}", e);
} else {
    panic!("Expected ExecutionError::Trap, but got Ok");
}

}

fn test_blob_basic_test(config: Config) {
let i = TestInstance::new(&config);
assert_eq!(i.call::<(), u32>("push_one_to_global_vec", ()).unwrap(), 1);
Expand Down Expand Up @@ -956,6 +996,8 @@ run_tests! {
doom_o3_dwarf2
pinky

test_invalid_jump_trap

test_blob_basic_test
test_blob_atomic_fetch_add
test_blob_atomic_fetch_swap
Expand Down