fix padlength issue

This commit is contained in:
Dmitry Vagner 2022-10-05 09:45:38 -07:00
parent 1275bcca5b
commit 9ebbc033bc
2 changed files with 50 additions and 19 deletions

View File

@ -88,20 +88,20 @@ global process:
/// def padlength(length):
/// t = length % 64
/// return 56 + 64*(t > 47) - t
/// t = length % 64
/// return 56 + 64*(t > 55) - t
%macro padlength
// stack: count
%mod_const(64)
// stack: t = count % 64
PUSH 47
PUSH 55
DUP2
// stack: t , 47 , t
// stack: t , 55 , t
GT
// stack: t > 47 , t
// stack: t > 55 , t
%mul_const(64)
%add_const(56)
// stack: 56 + 64*(t > 47), t
// stack: 56 + 64*(t > 55), t
SUB
%endmacro

View File

@ -23,31 +23,62 @@ fn ripemd(input: Vec<u8>) -> U256 {
U256::from(&hasher.finalize()[..])
}
fn test_hash(hash_fn_label: &str, standard_implementation: &dyn Fn(Vec<u8>) -> U256) -> Result<()> {
let kernel = combined_kernel();
let mut rng = thread_rng();
fn make_random_input() -> (u32, Vec<u8>) {
// Generate a random message, between 0 and 9999 bytes.
let mut rng = thread_rng();
let num_bytes = rng.gen_range(0..10000);
let message: Vec<u8> = (0..num_bytes).map(|_| rng.gen()).collect();
(num_bytes, (0..num_bytes).map(|_| rng.gen()).collect())
}
// Hash the message using a standard implementation.
let expected = standard_implementation(message.clone());
fn make_custom_input() -> (u32, 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,
])
}
// Load the message onto the stack.
let mut initial_stack = vec![U256::from(num_bytes)];
fn make_input_stack(length: u32, message: Vec<u8>) -> Vec<U256> {
let mut initial_stack = vec![U256::from(length)];
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());
initial_stack.reverse();
initial_stack
}
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();
// 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);
// Make the kernel.
let kernel = combined_kernel();
let kernel_function = kernel.global_labels[hash_fn_label];
// Run the kernel code.
let kernel_function = kernel.global_labels[hash_fn_label];
let result = run_with_kernel(&kernel, kernel_function, initial_stack)?;
let actual = result.stack()[0];
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);
// Check that the result is correct.
assert_eq!(expected, actual);
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(())
}