chore: keccak256 in Noir

noir_stdlib/src/hash/keccak.nr

@@ -0,0 +1,122 @@
+global LIMBS_PER_BLOCK = 17; //BLOCK_SIZE / 8;
+global NUM_KECCAK_LANES = 25;
+global BLOCK_SIZE = 136; //(1600 - BITS * 2) / WORD_SIZE;
+global WORD_SIZE = 8;
+
+use crate::collections::bounded_vec::BoundedVec;
+
+#[foreign(keccakf1600)]
+fn keccakf1600(input: [u64; 25]) -> [u64; 25] {}
+
+fn keccak256<let N: u32>(input: [u8; N], message_size: u32) -> [u8; 32] {
+    // No var keccak for now
+    assert(N == message_size);
+
+    //1. format_input_lanes
+    let max_blocks = (N + BLOCK_SIZE) / BLOCK_SIZE;
+    //maximum number of bytes to hash
+    let max_blocks_length = (BLOCK_SIZE * (max_blocks));
+    assert(max_blocks_length < 1000);
+    let mut block_bytes :BoundedVec<u8,1000> = BoundedVec::from_array(input);
+    for i in N..N + BLOCK_SIZE {
+        let data = if i == N {
+            1
+        } else if i == max_blocks_length - 1 {
+            0x80
+        } else {
+            0
+        };
+        block_bytes.push(data);
+    }
+
+    // keccak lanes interpret memory as little-endian integers,
+    // means we need to swap our byte ordering
+    let num_limbs = max_blocks * LIMBS_PER_BLOCK; //max_blocks_length / WORD_SIZE;
+    for i in 0..num_limbs {
+        let mut temp = [0; 8];
+
+        for j in 0..8 {
+            temp[j] = block_bytes.get(8*i+j);
+        }
+        for j in 0..8 {
+            block_bytes.set(8 * i + j, temp[7 - j]);
+        }
+    }
+    let  byte_size = max_blocks_length;
+    assert(num_limbs < 1000);
+    let mut sliced_buffer = [0; 1000];
+    // populate a vector of 64-bit limbs from our byte array
+    for i in 0..num_limbs {
+        let mut sliced = 0;
+        if (i * WORD_SIZE + WORD_SIZE > byte_size) {
+            let slice_size = byte_size - (i * WORD_SIZE);
+            let byte_shift = (WORD_SIZE - slice_size) * 8;
+            let mut v = 1;
+            for k in 0..slice_size {
+                sliced += v * (block_bytes.get(i * WORD_SIZE+7-k) as Field);
+                v *= 256;
+            }
+            let w = 1 << (byte_shift as u8);
+            sliced *= w as Field;
+        } else {
+            let mut v = 1;
+            for k in 0..WORD_SIZE {
+                sliced += v * (block_bytes.get(i * WORD_SIZE+7-k) as Field);
+                v *= 256;
+            }
+        }
+        sliced_buffer[i] = sliced as u64;
+    }
+
+    //2. sponge_absorb
+    let num_blocks = max_blocks;
+    let mut state : [u64;NUM_KECCAK_LANES]= [0; NUM_KECCAK_LANES];
+
+    for i in 0..num_blocks {
+        if (i == 0) {
+            for j in 0..LIMBS_PER_BLOCK {
+                state[j] = sliced_buffer[j];
+            }
+        } else {
+            for j in 0..LIMBS_PER_BLOCK {
+                state[j] = state[j] ^ sliced_buffer[i * LIMBS_PER_BLOCK + j];
+            }
+        }
+        state = keccakf1600(state);
+    }
+
+    //3. sponge_squeeze
+    let mut result = [0; 32];
+    for i in 0..4 {
+        let lane = state[i] as Field;
+        let lane_le = lane.to_le_bytes(8);
+        for j in 0..8 {
+            result[8*i+j] = lane_le[j];
+        }
+    }
+    result
+}
+
+mod tests {
+    use crate::hash::keccak::keccak256;
+
+    #[test]
+    fn smoke_test() {
+        let input = [0xbd];
+        let result = [
+            0x5a, 0x50, 0x2f, 0x9f, 0xca, 0x46, 0x7b, 0x26, 0x6d, 0x5b, 0x78, 0x33, 0x65, 0x19, 0x37, 0xe8, 0x05, 0x27, 0x0c, 0xa3, 0xf3, 0xaf, 0x1c, 0x0d, 0xd2, 0x46, 0x2d, 0xca, 0x4b, 0x3b, 0x1a, 0xbf
+        ];
+        assert_eq(keccak256(input, input.len()), result);
+    }
+
+    #[test]
+    fn hash_hello_world() {
+        // "hello world"
+        let input = [72, 101, 108, 108, 111, 32, 119, 111, 114, 108, 100, 33];
+        let result = [
+            0xec, 0xd0, 0xe1, 0x8, 0xa9, 0x8e, 0x19, 0x2a, 0xf1, 0xd2, 0xc2, 0x50, 0x55, 0xf4, 0xe3, 0xbe, 0xd7, 0x84, 0xb5, 0xc8, 0x77, 0x20, 0x4e, 0x73, 0x21, 0x9a, 0x52, 0x3, 0x25, 0x1f, 0xea, 0xab
+        ];
+        assert_eq(keccak256(input, input.len()), result);
+    }
+}
+

noir_stdlib/src/hash/mod.nr

@@ -1,6 +1,7 @@
 mod poseidon;
 mod mimc;
 mod poseidon2;
+mod keccak;
 
 use crate::default::Default;
 use crate::uint128::U128;
@@ -61,18 +62,18 @@ pub fn pedersen_hash<let N: u32>(input: [Field; N]) -> Field
 }
 
 #[field(bn254)]
-fn derive_generators<let N: u32, let M: u32>(
-    domain_separator_bytes: [u8; M],
-    starting_index: u32
-) -> [EmbeddedCurvePoint; N] {
+fn derive_generators<let N: u32, let M: u32>(domain_separator_bytes: [u8; M], starting_index: u32) -> [EmbeddedCurvePoint; N] {
     crate::assert_constant(domain_separator_bytes);
     crate::assert_constant(starting_index);
     __derive_generators(domain_separator_bytes, starting_index)
 }
 
 #[builtin(derive_pedersen_generators)]
 #[field(bn254)]
-fn __derive_generators<let N: u32, let M: u32>(domain_separator_bytes: [u8; M], starting_index: u32) -> [EmbeddedCurvePoint; N] {}
+fn __derive_generators<let N: u32, let M: u32>(
+    domain_separator_bytes: [u8; M],
+    starting_index: u32
+) -> [EmbeddedCurvePoint; N] {}
 
 fn pedersen_hash_with_separator_noir<let N: u32>(input: [Field; N], separator: u32) -> Field {
     let v1 = pedersen_commitment_with_separator(input, separator);