From baa4bd4ea10eafafac8b72755a020acfd33228aa Mon Sep 17 00:00:00 2001
From: Nicholas Ward <npward@berkeley.edu>
Date: Tue, 26 Jul 2022 11:02:12 -0700
Subject: [PATCH] files

---
 evm/src/cpu/kernel/asm/sha2_memory.asm | 171 +++++++++++++++++++++++++
 evm/src/cpu/kernel/tests/sha2.rs       |  53 ++++++++
 2 files changed, 224 insertions(+)
 create mode 100644 evm/src/cpu/kernel/asm/sha2_memory.asm
 create mode 100644 evm/src/cpu/kernel/tests/sha2.rs

diff --git a/evm/src/cpu/kernel/asm/sha2_memory.asm b/evm/src/cpu/kernel/asm/sha2_memory.asm
new file mode 100644
index 00000000..42fb5849
--- /dev/null
+++ b/evm/src/cpu/kernel/asm/sha2_memory.asm
@@ -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
diff --git a/evm/src/cpu/kernel/tests/sha2.rs b/evm/src/cpu/kernel/tests/sha2.rs
new file mode 100644
index 00000000..b46fd6cc
--- /dev/null
+++ b/evm/src/cpu/kernel/tests/sha2.rs
@@ -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(())
+}*/