mirror of
https://github.com/logos-storage/plonky2.git
synced 2026-01-11 18:23:09 +00:00
files
This commit is contained in:
Nicholas Ward
parent
87e06946b9
commit
baa4bd4ea1
171
evm/src/cpu/kernel/asm/sha2_memory.asm
Normal file
171
evm/src/cpu/kernel/asm/sha2_memory.asm
Normal file
@ -0,0 +1,171 @@
|
||||
// Load a single byte from kernel general memory.
|
||||
%macro mload_kernel_general
|
||||
// stack: offset
|
||||
PUSH @SEGMENT_KERNEL_GENERAL
|
||||
// stack: segment, offset
|
||||
PUSH 0 // kernel has context 0
|
||||
// stack: context, segment, offset
|
||||
MLOAD_GENERAL
|
||||
// stack: value
|
||||
%endmacro
|
||||
|
||||
// Load a big-endian u32, consisting of 4 bytes (c_3, c_2, c_1, c_0),
|
||||
// from kernel general memory.
|
||||
%macro mload_kernel_general_u32
|
||||
// stack: offset
|
||||
DUP1
|
||||
%mload_kernel_general
|
||||
// stack: c_3, offset
|
||||
%shl_const(8)
|
||||
// stack: c_3 << 8, offset
|
||||
DUP2
|
||||
%add_const(1)
|
||||
%mload_kernel_general
|
||||
OR
|
||||
// stack: (c_3 << 8) | c_2, offset
|
||||
%shl_const(8)
|
||||
// stack: ((c_3 << 8) | c_2) << 8, offset
|
||||
DUP2
|
||||
%add_const(2)
|
||||
%mload_kernel_general
|
||||
OR
|
||||
// stack: (((c_3 << 8) | c_2) << 8) | c_1, offset
|
||||
%shl_const(8)
|
||||
// stack: ((((c_3 << 8) | c_2) << 8) | c_1) << 8, offset
|
||||
SWAP1
|
||||
%add_const(3)
|
||||
%mload_kernel_general
|
||||
OR
|
||||
// stack: (((((c_3 << 8) | c_2) << 8) | c_1) << 8) | c_0
|
||||
%endmacro
|
||||
|
||||
// Load 256 bits (half of a 512-bit SHA-2 block) from general kernel memory.
|
||||
%macro mload_kernel_general_u256
|
||||
// stack: offset
|
||||
DUP1
|
||||
%mload_kernel_code_u32
|
||||
// stack: c_7, offset
|
||||
%shl_const(32)
|
||||
// stack: c7 << 32, offset
|
||||
DUP2
|
||||
%add_const(1)
|
||||
%mload_kernel_general_u32
|
||||
OR
|
||||
// stack: (c_7 << 32) | c_6, offset
|
||||
%shl_const(32)
|
||||
// stack: ((c_7 << 32) | c_6) << 32, offset
|
||||
DUP2
|
||||
%add_const(2)
|
||||
%mload_kernel_general_u32
|
||||
OR
|
||||
// stack: (c_7 << 64) | (c_6 << 32) | c_5, offset
|
||||
%shl_const(32)
|
||||
// stack: ((c_7 << 64) | (c_6 << 32) | c_5) << 32, offset
|
||||
DUP2
|
||||
%add_const(3)
|
||||
%mload_kernel_general_u32
|
||||
OR
|
||||
// stack: (c_7 << 96) | (c_6 << 64) | (c_5 << 32) | c_4, offset
|
||||
%shl_const(32)
|
||||
// stack: ((c_7 << 96) | (c_6 << 64) | (c_5 << 32) | c_4) << 32, offset
|
||||
DUP2
|
||||
%add_const(4)
|
||||
%mload_kernel_general_u32
|
||||
OR
|
||||
// stack: (c_7 << 128) | (c_6 << 96) | (c_5 << 64) | (c_4 << 32) | c_3, offset
|
||||
%shl_const(32)
|
||||
// stack: ((c_7 << 128) | (c_6 << 96) | (c_5 << 64) | (c_4 << 32) | c_3) << 32, offset
|
||||
DUP2
|
||||
%add_const(5)
|
||||
%mload_kernel_general_u32
|
||||
OR
|
||||
// stack: (c_7 << 160) | (c_6 << 128) | (c_5 << 96) | (c_4 << 64) | (c_3 << 32) | c_2, offset
|
||||
%shl_const(32)
|
||||
// stack: ((c_7 << 160) | (c_6 << 128) | (c_5 << 96) | (c_4 << 64) | (c_3 << 32) | c_2) << 32, offset
|
||||
DUP2
|
||||
%add_const(6)
|
||||
%mload_kernel_general_u32
|
||||
OR
|
||||
// stack: (c_7 << 192) | (c_6 << 160) | (c_5 << 128) | (c_4 << 96) | (c_3 << 64) | (c_2 << 32) | c_1, offset
|
||||
%shl_const(32)
|
||||
// stack: ((c_7 << 192) | (c_6 << 160) | (c_5 << 128) | (c_4 << 96) | (c_3 << 64) | (c_2 << 32) | c_1) << 32, offset
|
||||
DUP2
|
||||
%add_const(7)
|
||||
%mload_kernel_general_u32
|
||||
OR
|
||||
// stack: (c_7 << 224) | (c_6 << 192) | (c_5 << 160) | (c_4 << 128) | (c_3 << 96) | (c_2 << 64) | (c_1 << 32) | c_0, offset
|
||||
%endmacro
|
||||
|
||||
// Store a single byte to kernel general memory.
|
||||
%macro mstore_kernel_general
|
||||
// stack: offset, value
|
||||
PUSH @SEGMENT_KERNEL_GENERAL
|
||||
// stack: segment, offset
|
||||
PUSH 0 // kernel has context 0
|
||||
// stack: context, segment, offset, value
|
||||
MSTORE_GENERAL
|
||||
%endmacro
|
||||
|
||||
// Store a big-endian u32, consisting of 4 bytes (c_3, c_2, c_1, c_0),
|
||||
// to kernel general memory.
|
||||
%macro mstore_kernel_general_u32
|
||||
// stack: offset, value
|
||||
swap1
|
||||
// stack: value, offset
|
||||
push 1
|
||||
push 8
|
||||
shl
|
||||
// stack: 1 << 8, value, offset
|
||||
swap1
|
||||
// stack: value, 1 << 8, offset
|
||||
dup2
|
||||
dup2
|
||||
// stack: value, 1 << 8, value, 1 << 8, offset
|
||||
mod
|
||||
// stack: c_0 = value % (1 << 8), value, 1 << 8, offset
|
||||
swap2
|
||||
swap1
|
||||
// stack: value, 1 << 8, c_0, offset
|
||||
push 8
|
||||
shr
|
||||
// stack: value >> 8, 1 << 8, c_0, offset
|
||||
dup2
|
||||
dup2
|
||||
// stack: value >> 8, 1 << 8, value >> 8, 1 << 8, c_0, offset
|
||||
mod
|
||||
// stack: c_1 = (value >> 8) % (1 << 8), value >> 8, 1 << 8, c_0, offset
|
||||
swap2
|
||||
swap1
|
||||
// stack: value >> 8, 1 << 8, c_1, c_0, offset
|
||||
push 8
|
||||
shr
|
||||
// stack: value >> 16, 1 << 8, c_1, c_0, offset
|
||||
dup2
|
||||
dup2
|
||||
// stack: value >> 16, 1 << 8, value >> 16, 1 << 8, c_1, c_0, offset
|
||||
mod
|
||||
// stack: c_2 = (value >> 16) % (1 << 8), value >> 16, 1 << 8, c_1, c_0, offset
|
||||
swap2
|
||||
swap1
|
||||
// stack: value >> 16, 1 << 8, c_2, c_1, c_0, offset
|
||||
push 8
|
||||
shr
|
||||
// stack: value >> 24, 1 << 8, c_2, c_1, c_0, offset
|
||||
mod
|
||||
// stack: c_3 = (value >> 24) % (1 << 8), c_2, c_1, c_0, offset
|
||||
dup5
|
||||
// stack: offset, c_3, c_2, c_1, c_0, offset
|
||||
%mstore_kernel_general
|
||||
// stack: c_2, c_1, c_0, offset
|
||||
dup4
|
||||
// stack: offset, c_2, c_1, c_0, offset
|
||||
%mstore_kernel_general
|
||||
// stack: c_1, c_0, offset
|
||||
dup3
|
||||
// stack: offset, c_1, c_0, offset
|
||||
%mstore_kernel_general
|
||||
// stack: c_0, offset
|
||||
swap1
|
||||
// stack: offset, c_0
|
||||
%mstore_kernel_general
|
||||
%endmacro
|
||||
53
evm/src/cpu/kernel/tests/sha2.rs
Normal file
53
evm/src/cpu/kernel/tests/sha2.rs
Normal file
@ -0,0 +1,53 @@
|
||||
use std::str::FromStr;
|
||||
|
||||
use anyhow::Result;
|
||||
use ethereum_types::U256;
|
||||
use rand::{thread_rng, Rng};
|
||||
|
||||
use crate::cpu::kernel::aggregator::combined_kernel;
|
||||
use crate::cpu::kernel::interpreter::run;
|
||||
|
||||
#[test]
|
||||
fn test_sha2_store() -> Result<()> {
|
||||
let kernel = combined_kernel();
|
||||
let sha2_store = kernel.global_labels["sha2_store"];
|
||||
let mut rng = thread_rng();
|
||||
let num_bytes = rng.gen_range(0..20);
|
||||
let mut bytes: Vec<U256> = Vec::with_capacity(num_bytes);
|
||||
for _ in 0..num_bytes {
|
||||
let byte: u8 = rng.gen();
|
||||
let mut v = vec![0; 31];
|
||||
v.push(byte);
|
||||
let v2: [u8; 32] = v.try_into().unwrap();
|
||||
bytes.push(U256::from(v2));
|
||||
}
|
||||
|
||||
let mut initial_stack = vec![U256::from(num_bytes)];
|
||||
initial_stack.extend(bytes);
|
||||
let stack_with_kernel = run(&kernel.code, sha2_store, initial_stack)?.stack;
|
||||
|
||||
// let expected_stack = todo!();
|
||||
// assert_eq!(stack_with_kernel, expected_stack);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/*#[test]
|
||||
fn test_sha2() -> Result<()> {
|
||||
let kernel = combined_kernel();
|
||||
let sha2_store = kernel.global_labels["sha2_store"];
|
||||
let sha2_pad = kernel.global_labels["sha2_pad"];
|
||||
let mut rng = thread_rng();
|
||||
let a = U256([0; 4].map(|_| rng.gen()));
|
||||
let b = U256([0; 4].map(|_| rng.gen()));
|
||||
|
||||
let initial_stack = vec![U256::from_str("0xdeadbeef")?, b, a];
|
||||
let stack_with_kernel = run(&kernel.code, exp, initial_stack)?.stack;
|
||||
let initial_stack = vec![b, a];
|
||||
let code = [0xa, 0x63, 0xde, 0xad, 0xbe, 0xef, 0x56]; // EXP, PUSH4 deadbeef, JUMP
|
||||
|
||||
let expected_stack = todo!();
|
||||
assert_eq!(stack_with_kernel, expected_stack);
|
||||
|
||||
Ok(())
|
||||
}*/
|
||||
Loading…
x
Reference in New Issue
Block a user