logos-storage/plonky2
1
0
Fork 0
mirror of https://github.com/logos-storage/plonky2.git synced 2026-01-03 22:33:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

clean up and format

Browse Source
This commit is contained in:
Dmitry Vagner 2022-10-05 12:11:10 -07:00
parent 0afb9b0cf4
commit 7c7084a8d3
1 changed files with 18 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
evm/src/cpu/kernel/tests/hash.rs
View File

@ -23,27 +23,24 @@ fn ripemd(input: Vec<u8>) -> U256 {
U256::from(&hasher.finalize()[..])
}
fn make_random_input() -> (u32, Vec<u8>) {
fn make_random_input() -> Vec<u8> {
// Generate a random message, between 0 and 9999 bytes.
let mut rng = thread_rng();
let num_bytes = rng.gen_range(0..10000);
(num_bytes, (0..num_bytes).map(|_| rng.gen()).collect())
(0..num_bytes).map(|_| rng.gen()).collect()
}
fn make_custom_input() -> (u32, Vec<u8>) {
fn make_custom_input() -> Vec<u8> {
// Hardcode a custom message
(
48,
vec![
86, 124, 206, 245, 74, 57, 250, 43, 60, 30, 254, 43, 143, 144, 242, 215, 13, 103, 237,
61, 90, 105, 123, 250, 189, 181, 110, 192, 227, 57, 145, 46, 221, 238, 7, 181, 146,
111, 209, 150, 31, 157, 229, 126, 206, 105, 37, 17,
],
)
vec![
86, 124, 206, 245, 74, 57, 250, 43, 60, 30, 254, 43, 143, 144, 242, 215, 13, 103, 237, 61,
90, 105, 123, 250, 189, 181, 110, 192, 227, 57, 145, 46, 221, 238, 7, 181, 146, 111, 209,
150, 31, 157, 229, 126, 206, 105, 37, 17,
]
}
fn make_input_stack(length: u32, message: Vec<u8>) -> Vec<U256> {
let mut initial_stack = vec![U256::from(length)];
fn make_input_stack(message: Vec<u8>) -> Vec<U256> {
let mut initial_stack = vec![U256::from(message.len())];
let bytes: Vec<U256> = message.iter().map(|&x| U256::from(x as u32)).collect();
initial_stack.extend(bytes);
initial_stack.push(U256::from_str("0xdeadbeef").unwrap());
@ -52,17 +49,17 @@ fn make_input_stack(length: u32, message: Vec<u8>) -> Vec<U256> {
}
fn test_hash(hash_fn_label: &str, standard_implementation: &dyn Fn(Vec<u8>) -> U256) -> Result<()> {
// Make the inputs.
let (length_random, message_random) = make_random_input();
let (length_custom, message_custom) = make_custom_input();
// Make the input.
let message_random = make_random_input();
let message_custom = make_custom_input();
// Hash the message using a standard implementation.
let expected_random = standard_implementation(message_random.clone());
let expected_custom = standard_implementation(message_custom.clone());
// Load the message onto the stack.
let initial_stack_random = make_input_stack(length_random, message_random);
let initial_stack_custom = make_input_stack(length_custom, message_custom);
let initial_stack_random = make_input_stack(message_random);
let initial_stack_custom = make_input_stack(message_custom);
// Make the kernel.
let kernel = combined_kernel();
@ -72,15 +69,12 @@ fn test_hash(hash_fn_label: &str, standard_implementation: &dyn Fn(Vec<u8>) -> U
let result_random = run_with_kernel(&kernel, kernel_function, initial_stack_random)?;
let result_custom = run_with_kernel(&kernel, kernel_function, initial_stack_custom)?;
// Print stack for debugging
// let printable_stack: Vec <String> = result.stack().iter().map(|x| format!("{:x}", x)).collect();
// println!("{:#?}", printable_stack);
// Extract the final output.
let actual_random = result_random.stack()[0];
let actual_custom = result_custom.stack()[0];
// Check that the result is correct.
let actual_random = result_random.stack()[0];
assert_eq!(expected_random, actual_random);
let actual_custom = result_custom.stack()[0];
assert_eq!(expected_custom, actual_custom);
Ok(())