feat(starknet_os): integrate execute_transactions hints

crates/starknet_os/src/hints.rs

@@ -6,5 +6,6 @@ pub mod deprecated_compiled_class;
 pub mod enum_definition;
 pub mod enum_generation;
 pub mod error;
+pub mod execute_transactions;
 pub mod stateless_compression;
 pub mod types;

crates/starknet_os/src/hints/enum_definition.rs

@@ -38,6 +38,14 @@ use crate::hints::deprecated_compiled_class::{
     load_deprecated_class_inner,
 };
 use crate::hints::error::{HintExtensionResult, HintResult, OsHintError};
+use crate::hints::execute_transactions::{
+    fill_holes_in_rc96_segment,
+    log_remaining_txs,
+    set_component_hashes,
+    set_sha256_segment_in_syscall_handler,
+    sha2_finalize,
+    start_tx_validate_declare_execution_context,
+};
 use crate::hints::stateless_compression::{
     compression_hint,
     dictionary_from_bucket,
@@ -296,6 +304,68 @@ define_hint_enum!(
     ids.compiled_class = segments.gen_arg(cairo_contract)"#
         }
     ),
+    (
+        StartTxValidateDeclareExecutionContext,
+        start_tx_validate_declare_execution_context,
+        indoc! {r#"
+    execution_helper.start_tx(
+        tx_info_ptr=ids.validate_declare_execution_context.deprecated_tx_info.address_
+    )"#
+        }
+    ),
+    (
+        SetSha256SegmentInSyscallHandler,
+        set_sha256_segment_in_syscall_handler,
+        indoc! {r#"syscall_handler.sha256_segment = ids.sha256_ptr"#}
+    ),
+    (
+        LogRemainingTxs,
+        log_remaining_txs,
+        indoc! {r#"print(f"execute_transactions_inner: {ids.n_txs} transactions remaining.")"#}
+    ),
+    (
+        FillHolesInRc96Segment,
+        fill_holes_in_rc96_segment,
+        indoc! {r#"
+rc96_ptr = ids.range_check96_ptr
+segment_size = rc96_ptr.offset
+base = rc96_ptr - segment_size
+
+for i in range(segment_size):
+    memory.setdefault(base + i, 0)"#}
+    ),
+    (
+        SetComponentHashes,
+        set_component_hashes,
+        indoc! {r#"
+class_component_hashes = component_hashes[tx.class_hash]
+assert (
+    len(class_component_hashes) == ids.ContractClassComponentHashes.SIZE
+), "Wrong number of class component hashes."
+ids.contract_class_component_hashes = segments.gen_arg(class_component_hashes)"#
+        }
+    ),
+    (
+        Sha2Finalize,
+        sha2_finalize,
+        indoc! {r#"# Add dummy pairs of input and output.
+from starkware.cairo.common.cairo_sha256.sha256_utils import (
+    IV,
+    compute_message_schedule,
+    sha2_compress_function,
+)
+
+number_of_missing_blocks = (-ids.n) % ids.BATCH_SIZE
+assert 0 <= number_of_missing_blocks < 20
+_sha256_input_chunk_size_felts = ids.SHA256_INPUT_CHUNK_SIZE_FELTS
+assert 0 <= _sha256_input_chunk_size_felts < 100
+
+message = [0] * _sha256_input_chunk_size_felts
+w = compute_message_schedule(message)
+output = sha2_compress_function(IV, w)
+padding = (message + IV + output) * number_of_missing_blocks
+segments.write_arg(ids.sha256_ptr_end, padding)"#}
+    )
 );
 
 define_hint_extension_enum!(

crates/starknet_os/src/hints/execute_transactions.rs

@@ -0,0 +1,69 @@
+use std::collections::HashMap;
+
+use cairo_vm::hint_processor::hint_processor_definition::HintReference;
+use cairo_vm::serde::deserialize_program::ApTracking;
+use cairo_vm::types::exec_scope::ExecutionScopes;
+use cairo_vm::vm::vm_core::VirtualMachine;
+use starknet_types_core::felt::Felt;
+
+use crate::hints::error::HintResult;
+
+pub fn start_tx_validate_declare_execution_context(
+    _vm: &mut VirtualMachine,
+    _exec_scopes: &mut ExecutionScopes,
+    _ids_data: &HashMap<String, HintReference>,
+    _ap_tracking: &ApTracking,
+    _constants: &HashMap<String, Felt>,
+) -> HintResult {
+    todo!()
+}
+
+pub fn set_sha256_segment_in_syscall_handler(
+    _vm: &mut VirtualMachine,
+    _exec_scopes: &mut ExecutionScopes,
+    _ids_data: &HashMap<String, HintReference>,
+    _ap_tracking: &ApTracking,
+    _constants: &HashMap<String, Felt>,
+) -> HintResult {
+    todo!()
+}
+
+pub fn log_remaining_txs(
+    _vm: &mut VirtualMachine,
+    _exec_scopes: &mut ExecutionScopes,
+    _ids_data: &HashMap<String, HintReference>,
+    _ap_tracking: &ApTracking,
+    _constants: &HashMap<String, Felt>,
+) -> HintResult {
+    todo!()
+}
+
+pub fn fill_holes_in_rc96_segment(
+    _vm: &mut VirtualMachine,
+    _exec_scopes: &mut ExecutionScopes,
+    _ids_data: &HashMap<String, HintReference>,
+    _ap_tracking: &ApTracking,
+    _constants: &HashMap<String, Felt>,
+) -> HintResult {
+    todo!()
+}
+
+pub fn set_component_hashes(
+    _vm: &mut VirtualMachine,
+    _exec_scopes: &mut ExecutionScopes,
+    _ids_data: &HashMap<String, HintReference>,
+    _ap_tracking: &ApTracking,
+    _constants: &HashMap<String, Felt>,
+) -> HintResult {
+    todo!()
+}
+
+pub fn sha2_finalize(
+    _vm: &mut VirtualMachine,
+    _exec_scopes: &mut ExecutionScopes,
+    _ids_data: &HashMap<String, HintReference>,
+    _ap_tracking: &ApTracking,
+    _constants: &HashMap<String, Felt>,
+) -> HintResult {
+    todo!()
+}