mirror of
https://github.com/logos-storage/plonky2.git
synced 2026-01-12 18:53:11 +00:00
115 lines
3.9 KiB
Rust
115 lines
3.9 KiB
Rust
use anyhow::Result;
|
|
|
|
use crate::cpu::kernel::aggregator::KERNEL;
|
|
use crate::cpu::kernel::interpreter::Interpreter;
|
|
|
|
#[test]
|
|
fn test_decode_rlp_string_len_short() -> Result<()> {
|
|
let decode_rlp_string_len = KERNEL.global_labels["decode_rlp_string_len"];
|
|
|
|
let initial_stack = vec![0xDEADBEEFu32.into(), 2.into()];
|
|
let mut interpreter = Interpreter::new_with_kernel(decode_rlp_string_len, initial_stack);
|
|
|
|
// A couple dummy bytes, followed by "0x70" which is its own encoding.
|
|
interpreter.set_rlp_memory(vec![123, 234, 0x70]);
|
|
|
|
interpreter.run()?;
|
|
let expected_stack = vec![1.into(), 2.into()]; // len, pos
|
|
assert_eq!(interpreter.stack(), expected_stack);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_decode_rlp_string_len_medium() -> Result<()> {
|
|
let decode_rlp_string_len = KERNEL.global_labels["decode_rlp_string_len"];
|
|
|
|
let initial_stack = vec![0xDEADBEEFu32.into(), 2.into()];
|
|
let mut interpreter = Interpreter::new_with_kernel(decode_rlp_string_len, initial_stack);
|
|
|
|
// A couple dummy bytes, followed by the RLP encoding of "1 2 3 4 5".
|
|
interpreter.set_rlp_memory(vec![123, 234, 0x85, 1, 2, 3, 4, 5]);
|
|
|
|
interpreter.run()?;
|
|
let expected_stack = vec![5.into(), 3.into()]; // len, pos
|
|
assert_eq!(interpreter.stack(), expected_stack);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_decode_rlp_string_len_long() -> Result<()> {
|
|
let decode_rlp_string_len = KERNEL.global_labels["decode_rlp_string_len"];
|
|
|
|
let initial_stack = vec![0xDEADBEEFu32.into(), 2.into()];
|
|
let mut interpreter = Interpreter::new_with_kernel(decode_rlp_string_len, initial_stack);
|
|
|
|
// The RLP encoding of the string "1 2 3 ... 56".
|
|
interpreter.set_rlp_memory(vec![
|
|
123, 234, 0xb8, 56, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20,
|
|
21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43,
|
|
44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56,
|
|
]);
|
|
|
|
interpreter.run()?;
|
|
let expected_stack = vec![56.into(), 4.into()]; // len, pos
|
|
assert_eq!(interpreter.stack(), expected_stack);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_decode_rlp_list_len_short() -> Result<()> {
|
|
let decode_rlp_list_len = KERNEL.global_labels["decode_rlp_list_len"];
|
|
|
|
let initial_stack = vec![0xDEADBEEFu32.into(), 0.into()];
|
|
let mut interpreter = Interpreter::new_with_kernel(decode_rlp_list_len, initial_stack);
|
|
|
|
// The RLP encoding of [1, 2, [3, 4]].
|
|
interpreter.set_rlp_memory(vec![0xc5, 1, 2, 0xc2, 3, 4]);
|
|
|
|
interpreter.run()?;
|
|
let expected_stack = vec![5.into(), 1.into()]; // len, pos
|
|
assert_eq!(interpreter.stack(), expected_stack);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_decode_rlp_list_len_long() -> Result<()> {
|
|
let decode_rlp_list_len = KERNEL.global_labels["decode_rlp_list_len"];
|
|
|
|
let initial_stack = vec![0xDEADBEEFu32.into(), 0.into()];
|
|
let mut interpreter = Interpreter::new_with_kernel(decode_rlp_list_len, initial_stack);
|
|
|
|
// The RLP encoding of [1, ..., 56].
|
|
interpreter.set_rlp_memory(vec![
|
|
0xf8, 56, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22,
|
|
23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45,
|
|
46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56,
|
|
]);
|
|
|
|
interpreter.run()?;
|
|
let expected_stack = vec![56.into(), 2.into()]; // len, pos
|
|
assert_eq!(interpreter.stack(), expected_stack);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_decode_rlp_scalar() -> Result<()> {
|
|
let decode_rlp_scalar = KERNEL.global_labels["decode_rlp_scalar"];
|
|
|
|
let initial_stack = vec![0xDEADBEEFu32.into(), 0.into()];
|
|
let mut interpreter = Interpreter::new_with_kernel(decode_rlp_scalar, initial_stack);
|
|
|
|
// The RLP encoding of "12 34 56".
|
|
interpreter.set_rlp_memory(vec![0x83, 0x12, 0x34, 0x56]);
|
|
|
|
interpreter.run()?;
|
|
let expected_stack = vec![0x123456.into(), 4.into()]; // scalar, pos
|
|
assert_eq!(interpreter.stack(), expected_stack);
|
|
|
|
Ok(())
|
|
}
|