Add_hint_for_splitting_data_for_blake_with_no_encoding

Author: YairVaknin-starkware

vm/src/hint_processor/builtin_hint_processor/blake2s_utils.rs

@@ -268,51 +268,82 @@ pub fn is_less_than_63_bits_and_not_end(
     Ok(())
 }
 
-/* Implements Hint:
+pub enum BlakeEncodingMode {
+    /// 2 limbs if val < 2⁶³, otherwise 8 limbs after adding 2^255.
+    UseEncoding,
+    /// Always 8 limbs.
+    NoEncoding,
+}
+
+/* If mode is `BlakeEncodingMode::UseEncoding`, implements the following hint:
 offset = 0
 for i in range(ids.packed_values_len):
     val = (memory[ids.packed_values + i] % PRIME)
     val_len = 2 if val < 2**63 else 8
     if val_len == 8:
         val += 2**255
-    for i in range(val_len - 1, -1, -1):
+    for i in range(val_len):
+        val, memory[ids.unpacked_u32s + offset + i] = divmod(val, 2**32)
+    assert val == 0
+    offset += val_len
+
+   If mode is `BlakeEncodingMode::NoEncoding`, implements the following hint:
+offset = 0
+for i in range(ids.packed_values_len):
+    val = (memory[ids.packed_values + i] % PRIME)
+    for i in range(8):
         val, memory[ids.unpacked_u32s + offset + i] = divmod(val, 2**32)
     assert val == 0
     offset += val_len
 */
-pub fn blake2s_unpack_felts(
+pub fn split_packed_felts(
     vm: &mut VirtualMachine,
     ids_data: &HashMap<String, HintReference>,
     ap_tracking: &ApTracking,
+    mode: BlakeEncodingMode,
 ) -> Result<(), HintError> {
     let packed_values_len =
         get_integer_from_var_name("packed_values_len", vm, ids_data, ap_tracking)?;
     let packed_values = get_ptr_from_var_name("packed_values", vm, ids_data, ap_tracking)?;
     let unpacked_u32s = get_ptr_from_var_name("unpacked_u32s", vm, ids_data, ap_tracking)?;
 
     let vals = vm.get_integer_range(packed_values, felt_to_usize(&packed_values_len)?)?;
+
     let pow2_32 = BigUint::from(1_u32) << 32;
     let pow2_63 = BigUint::from(1_u32) << 63;
     let pow2_255 = BigUint::from(1_u32) << 255;
 
-    // Split value into either 2 or 8 32-bit limbs.
     let out: Vec<MaybeRelocatable> = vals
         .into_iter()
-        .map(|val| val.to_biguint())
-        .flat_map(|val| {
-            if val < pow2_63 {
-                let (high, low) = val.div_rem(&pow2_32);
-                vec![high, low]
-            } else {
+        .map(|v| v.to_biguint())
+        .flat_map(|mut val| match mode {
+            BlakeEncodingMode::NoEncoding => {
+                // Always 8 little‑endian limbs.
                 let mut limbs = vec![BigUint::from(0_u32); 8];
-                let mut val: BigUint = val + &pow2_255;
-                for limb in limbs.iter_mut().rev() {
+                for limb in &mut limbs {
                     let (q, r) = val.div_rem(&pow2_32);
                     *limb = r;
                     val = q;
                 }
                 limbs
             }
+            BlakeEncodingMode::UseEncoding => {
+                if val < pow2_63 {
+                    // 2 limbs: low, high.
+                    let (high, low) = val.div_rem(&pow2_32);
+                    vec![low, high]
+                } else {
+                    // 8 limbs after adding 2**255.
+                    val += &pow2_255;
+                    let mut limbs = vec![BigUint::from(0_u32); 8];
+                    for limb in &mut limbs {
+                        let (q, r) = val.div_rem(&pow2_32);
+                        *limb = r;
+                        val = q;
+                    }
+                    limbs
+                }
+            }
         })
         .map(Felt252::from)
         .map(MaybeRelocatable::from)
@@ -740,22 +771,28 @@ mod tests {
         check_memory![vm.segments.memory, ((1, 3), 0)];
     }
 
-    #[test]
-    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
-    fn blake2s_unpack_felts() {
-        let hint_code = hint_code::BLAKE2S_UNPACK_FELTS;
-        //Create vm
+    /// Builds a VM, writes the packed_values_len, packed_values, unpacked_u32s and the
+    /// actual small and big felts to the memory.
+    /// Returns the VM and the ids_data HashMap.
+    fn prepare_vm_for_splitting_felts_for_blake(
+        small_val: i128,
+        big_val: i128,
+    ) -> (VirtualMachine, HashMap<String, HintReference>) {
         let mut vm = vm!();
-        //Insert ids into memory
         vm.segments = segments![
+            // ids.packed_values_len = 2
             ((1, 0), 2),
+            // ids.packed_values = (1,3)
             ((1, 1), (1, 3)),
+            // ids.unpacked_u32s  = (2,0)
             ((1, 2), (2, 0)),
-            ((1, 3), 0x123456781234),
-            ((1, 4), 0x1234abcd5678efab1234abcd)
+            // packed small / big felts
+            ((1, 3), small_val),
+            ((1, 4), big_val)
         ];
         vm.set_fp(5);
         vm.set_ap(5);
+
         let ids_data = ids_data![
             "packed_values_len",
             "packed_values",
@@ -764,21 +801,64 @@ mod tests {
             "big_value"
         ];
         vm.segments.add();
-        //Execute the hint
-        assert_matches!(run_hint!(vm, ids_data, hint_code), Ok(()));
-        //Check data ptr
+        (vm, ids_data)
+    }
+
+    #[test]
+    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
+    fn blake2s_encode_and_split_felts() {
+        let (mut vm, ids_data) = prepare_vm_for_splitting_felts_for_blake(
+            0x123456781234u128 as i128,
+            0x1234abcd5678efab1234abcd,
+        );
+        assert_matches!(
+            run_hint!(vm, ids_data, hint_code::BLAKE2S_ENCODE_AND_SPLIT_FELTS),
+            Ok(())
+        );
+        check_memory![
+            vm.segments.memory,
+            ((2, 0), 0x56781234),
+            ((2, 1), 0x1234),
+            ((2, 2), 0x1234abcd),
+            ((2, 3), 0x5678efab),
+            ((2, 4), 0x1234abcd),
+            ((2, 5), 0),
+            ((2, 6), 0),
+            ((2, 7), 0),
+            ((2, 8), 0),
+            ((2, 9), 0x80000000),
+        ];
+    }
+
+    #[test]
+    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
+    fn blake2s_split_felts_to_u32s() {
+        let (mut vm, ids_data) = prepare_vm_for_splitting_felts_for_blake(
+            0x123456781234u128 as i128,
+            0x1234abcd5678efab1234abcd5678efab,
+        );
+        assert_matches!(
+            run_hint!(vm, ids_data, hint_code::BLAKE2S_SPLIT_FELTS_TO_U32S),
+            Ok(())
+        );
         check_memory![
             vm.segments.memory,
-            ((2, 0), 0x1234),
-            ((2, 1), 0x56781234),
-            ((2, 2), 0x80000000),
+            ((2, 0), 0x56781234),
+            ((2, 1), 0x1234),
+            ((2, 2), 0),
             ((2, 3), 0),
             ((2, 4), 0),
             ((2, 5), 0),
             ((2, 6), 0),
-            ((2, 7), 0x1234abcd),
+            ((2, 7), 0),
             ((2, 8), 0x5678efab),
-            ((2, 9), 0x1234abcd)
+            ((2, 9), 0x1234abcd),
+            ((2, 10), 0x5678efab),
+            ((2, 11), 0x1234abcd),
+            ((2, 12), 0),
+            ((2, 13), 0),
+            ((2, 14), 0),
+            ((2, 15), 0),
         ];
     }
 

vm/src/hint_processor/builtin_hint_processor/builtin_hint_processor_definition.rs

@@ -1,6 +1,9 @@
 use super::blake2s_utils::example_blake2s_compress;
 use super::{
-    blake2s_utils::{blake2s_unpack_felts, finalize_blake2s_v3, is_less_than_63_bits_and_not_end},
+    blake2s_utils::{
+        finalize_blake2s_v3, is_less_than_63_bits_and_not_end, split_packed_felts,
+        BlakeEncodingMode,
+    },
     ec_recover::{
         ec_recover_divmod_n_packed, ec_recover_product_div_m, ec_recover_product_mod,
         ec_recover_sub_a_b,
@@ -363,9 +366,18 @@ impl HintProcessorLogic for BuiltinHintProcessor {
             hint_code::IS_LESS_THAN_63_BITS_AND_NOT_END => {
                 is_less_than_63_bits_and_not_end(vm, &hint_data.ids_data, &hint_data.ap_tracking)
             }
-            hint_code::BLAKE2S_UNPACK_FELTS => {
-                blake2s_unpack_felts(vm, &hint_data.ids_data, &hint_data.ap_tracking)
-            }
+            hint_code::BLAKE2S_ENCODE_AND_SPLIT_FELTS => split_packed_felts(
+                vm,
+                &hint_data.ids_data,
+                &hint_data.ap_tracking,
+                BlakeEncodingMode::UseEncoding,
+            ),
+            hint_code::BLAKE2S_SPLIT_FELTS_TO_U32S => split_packed_felts(
+                vm,
+                &hint_data.ids_data,
+                &hint_data.ap_tracking,
+                BlakeEncodingMode::NoEncoding,
+            ),
             hint_code::UNSAFE_KECCAK => {
                 unsafe_keccak(vm, exec_scopes, &hint_data.ids_data, &hint_data.ap_tracking)
             }

vm/src/hint_processor/builtin_hint_processor/hint_code.rs

@@ -427,13 +427,20 @@ MASK = 2 ** 32 - 1
 segments.write_arg(ids.data, [(ids.high >> (B * (3 - i))) & MASK for i in range(4)])
 segments.write_arg(ids.data + 4, [(ids.low >> (B * (3 - i))) & MASK for i in range(4)])"#}),
 (IS_LESS_THAN_63_BITS_AND_NOT_END, indoc! {r#"memory[ap] = to_felt_or_relocatable((ids.end != ids.packed_values) and (memory[ids.packed_values] < 2**63))"#}),
-(BLAKE2S_UNPACK_FELTS, indoc! {r#"offset = 0
+(BLAKE2S_ENCODE_AND_SPLIT_FELTS, indoc! {r#"offset = 0
 for i in range(ids.packed_values_len):
     val = (memory[ids.packed_values + i] % PRIME)
     val_len = 2 if val < 2**63 else 8
     if val_len == 8:
         val += 2**255
-    for i in range(val_len - 1, -1, -1):
+    for i in range(val_len):
+        val, memory[ids.unpacked_u32s + offset + i] = divmod(val, 2**32)
+    assert val == 0
+    offset += val_len"#}),
+(BLAKE2S_SPLIT_FELTS_TO_U32S, indoc! {r#"offset = 0
+for i in range(ids.packed_values_len):
+    val = (memory[ids.packed_values + i] % PRIME)
+    for i in range(8):
         val, memory[ids.unpacked_u32s + offset + i] = divmod(val, 2**32)
     assert val == 0
     offset += val_len"#}),