TransactionExecutionInfo: add serialize

crates/blockifier/src/execution/call_info.rs

@@ -1,6 +1,8 @@
 use std::collections::HashSet;
 
 use cairo_vm::vm::runners::cairo_runner::ExecutionResources as VmExecutionResources;
+use serde::ser::SerializeStruct;
+use serde::{Serialize, Serializer};
 use starknet_api::core::{ClassHash, EthAddress};
 use starknet_api::hash::StarkFelt;
 use starknet_api::state::StorageKey;
@@ -11,7 +13,7 @@ use crate::state::cached_state::StorageEntry;
 use crate::transaction::errors::TransactionExecutionError;
 use crate::transaction::objects::TransactionExecutionResult;
 
-#[derive(Clone, Debug, Default, Eq, PartialEq)]
+#[derive(Clone, Debug, Default, Eq, PartialEq, Serialize)]
 pub struct Retdata(pub Vec<StarkFelt>);
 
 #[macro_export]
@@ -22,29 +24,29 @@ macro_rules! retdata {
 }
 
 #[cfg_attr(test, derive(Clone))]
-#[derive(Debug, Default, Eq, PartialEq)]
+#[derive(Debug, Default, Eq, PartialEq, Serialize)]
 pub struct OrderedEvent {
     pub order: usize,
     pub event: EventContent,
 }
 
 #[cfg_attr(test, derive(Clone))]
-#[derive(Debug, Default, Eq, PartialEq)]
+#[derive(Debug, Default, Eq, PartialEq, Serialize)]
 pub struct MessageToL1 {
     pub to_address: EthAddress,
     pub payload: L2ToL1Payload,
 }
 
 #[cfg_attr(test, derive(Clone))]
-#[derive(Debug, Default, Eq, PartialEq)]
+#[derive(Debug, Default, Eq, PartialEq, Serialize)]
 pub struct OrderedL2ToL1Message {
     pub order: usize,
     pub message: MessageToL1,
 }
 
 /// Represents the effects of executing a single entry point.
 #[cfg_attr(test, derive(Clone))]
-#[derive(Debug, Default, Eq, PartialEq)]
+#[derive(Debug, Default, Eq, PartialEq, Serialize)]
 pub struct CallExecution {
     pub retdata: Retdata,
     pub events: Vec<OrderedEvent>,
@@ -53,12 +55,39 @@ pub struct CallExecution {
     pub gas_consumed: u64,
 }
 
+#[derive(Debug, Default, Eq, PartialEq, derive_more::Deref, derive_more::From)]
+pub struct VmExecutionResourcesWrapper(pub VmExecutionResources);
+
+impl VmExecutionResourcesWrapper {
+    pub fn new(vm_execution_resources: VmExecutionResources) -> Self {
+        VmExecutionResourcesWrapper(vm_execution_resources)
+    }
+}
+
+impl Serialize for VmExecutionResourcesWrapper {
+    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+    where
+        S: Serializer,
+    {
+        // Create a serialize struct with the appropriate number of fields.
+        let mut state = serializer.serialize_struct("ExecutionResourcesWrapper", 3)?;
+
+        // Serialize each field individually.
+        state.serialize_field("builtin_instance_counter", &self.0.builtin_instance_counter)?;
+        state.serialize_field("n_memory_holes", &self.0.n_memory_holes)?;
+        state.serialize_field("n_steps", &self.0.n_steps)?;
+
+        // Finish serializing the struct.
+        state.end()
+    }
+}
+
 /// Represents the full effects of executing an entry point, including the inner calls it invoked.
-#[derive(Debug, Default, Eq, PartialEq)]
+#[derive(Debug, Default, Eq, PartialEq, Serialize)]
 pub struct CallInfo {
     pub call: CallEntryPoint,
     pub execution: CallExecution,
-    pub vm_resources: VmExecutionResources,
+    pub vm_resources: VmExecutionResourcesWrapper,
     pub inner_calls: Vec<CallInfo>,
 
     // Additional information gathered during execution.

crates/blockifier/src/execution/deprecated_entry_point_execution.rs

@@ -253,7 +253,7 @@ pub fn finalize_execution(
             failed: false,
             gas_consumed: 0,
         },
-        vm_resources: full_call_vm_resources.filter_unused_builtins(),
+        vm_resources: full_call_vm_resources.filter_unused_builtins().into(),
         inner_calls: syscall_handler.inner_calls,
         storage_read_values: syscall_handler.read_values,
         accessed_storage_keys: syscall_handler.accessed_keys,

crates/blockifier/src/execution/deprecated_syscalls/deprecated_syscalls_test.rs

@@ -135,7 +135,7 @@ fn test_nested_library_call() {
     let nested_storage_call_info = CallInfo {
         call: nested_storage_entry_point,
         execution: CallExecution::from_retdata(retdata![stark_felt!(value + 1)]),
-        vm_resources: storage_entry_point_vm_resources.clone(),
+        vm_resources: storage_entry_point_vm_resources.clone().into(),
         storage_read_values: vec![stark_felt!(0_u8), stark_felt!(value + 1)],
         accessed_storage_keys: HashSet::from([StorageKey(patricia_key!(key + 1))]),
         ..Default::default()
@@ -149,14 +149,14 @@ fn test_nested_library_call() {
     let library_call_info = CallInfo {
         call: library_entry_point,
         execution: CallExecution::from_retdata(retdata![stark_felt!(value + 1)]),
-        vm_resources: library_call_vm_resources.clone(),
+        vm_resources: library_call_vm_resources.clone().into(),
         inner_calls: vec![nested_storage_call_info],
         ..Default::default()
     };
     let storage_call_info = CallInfo {
         call: storage_entry_point,
         execution: CallExecution::from_retdata(retdata![stark_felt!(value)]),
-        vm_resources: storage_entry_point_vm_resources.clone(),
+        vm_resources: storage_entry_point_vm_resources.clone().into(),
         storage_read_values: vec![stark_felt!(0_u8), stark_felt!(value)],
         accessed_storage_keys: HashSet::from([StorageKey(patricia_key!(key))]),
         ..Default::default()
@@ -168,7 +168,7 @@ fn test_nested_library_call() {
     let expected_call_info = CallInfo {
         call: main_entry_point.clone(),
         execution: CallExecution::from_retdata(retdata![stark_felt!(0_u8)]),
-        vm_resources: main_call_vm_resources,
+        vm_resources: main_call_vm_resources.into(),
         inner_calls: vec![library_call_info, storage_call_info],
         ..Default::default()
     };
@@ -209,7 +209,7 @@ fn test_call_contract() {
             ..trivial_external_entry_point()
         },
         execution: expected_execution.clone(),
-        vm_resources: VmExecutionResources { n_steps: 42, ..Default::default() },
+        vm_resources: VmExecutionResources { n_steps: 42, ..Default::default() }.into(),
         storage_read_values: vec![StarkFelt::ZERO, stark_felt!(value)],
         accessed_storage_keys: HashSet::from([StorageKey(patricia_key!(key))]),
         ..Default::default()
@@ -229,7 +229,8 @@ fn test_call_contract() {
             n_steps: 81,
             builtin_instance_counter: HashMap::from([(RANGE_CHECK_BUILTIN_NAME.to_string(), 1)]),
             ..Default::default()
-        },
+        }
+        .into(),
         ..Default::default()
     };
 

crates/blockifier/src/execution/entry_point.rs

@@ -5,6 +5,7 @@ use std::sync::Arc;
 use cairo_vm::vm::runners::cairo_runner::{
     ExecutionResources as VmExecutionResources, ResourceTracker, RunResources,
 };
+use serde::Serialize;
 use starknet_api::core::{ClassHash, ContractAddress, EntryPointSelector};
 use starknet_api::deprecated_contract_class::EntryPointType;
 use starknet_api::hash::StarkFelt;
@@ -35,14 +36,14 @@ pub const FAULTY_CLASS_HASH: &str =
 pub type EntryPointExecutionResult<T> = Result<T, EntryPointExecutionError>;
 
 /// Represents a the type of the call (used for debugging).
-#[derive(Clone, Copy, Debug, Default, Eq, Hash, PartialEq)]
+#[derive(Clone, Copy, Debug, Default, Eq, Hash, PartialEq, Serialize)]
 pub enum CallType {
     #[default]
     Call = 0,
     Delegate = 1,
 }
 /// Represents a call to an entry point of a Starknet contract.
-#[derive(Clone, Debug, Default, Eq, PartialEq)]
+#[derive(Clone, Debug, Default, Eq, PartialEq, Serialize)]
 pub struct CallEntryPoint {
     // The class hash is not given if it can be deduced from the storage address.
     pub class_hash: Option<ClassHash>,
@@ -281,7 +282,7 @@ impl EntryPointExecutionContext {
     ) -> usize {
         let validate_steps = validate_call_info
             .as_ref()
-            .map(|call_info| call_info.vm_resources.n_steps)
+            .map(|call_info| call_info.vm_resources.0.n_steps)
             .unwrap_or_default();
 
         let overhead_steps = OS_RESOURCES.resources_for_tx_type(tx_type).n_steps;

crates/blockifier/src/execution/entry_point_execution.rs

@@ -298,7 +298,7 @@ pub fn finalize_execution(
             failed: call_result.failed,
             gas_consumed: call_result.gas_consumed,
         },
-        vm_resources: full_call_vm_resources.filter_unused_builtins(),
+        vm_resources: full_call_vm_resources.filter_unused_builtins().into(),
         inner_calls: syscall_handler.inner_calls,
         storage_read_values: syscall_handler.read_values,
         accessed_storage_keys: syscall_handler.accessed_keys,

crates/blockifier/src/execution/entry_point_test.rs

@@ -531,6 +531,7 @@ fn test_cairo1_entry_point_segment_arena() {
             .execute_directly(&mut state)
             .unwrap()
             .vm_resources
+            .0
             .builtin_instance_counter
             .contains_key(BuiltinName::segment_arena.name())
     );

crates/blockifier/src/execution/syscalls/syscalls_test.rs

@@ -554,7 +554,7 @@ fn test_nested_library_call() {
             gas_consumed: REQUIRED_GAS_STORAGE_READ_WRITE_TEST,
             ..CallExecution::default()
         },
-        vm_resources: storage_entry_point_vm_resources.clone(),
+        vm_resources: storage_entry_point_vm_resources.clone().into(),
         storage_read_values: vec![stark_felt!(value + 1)],
         accessed_storage_keys: HashSet::from([StorageKey(patricia_key!(key + 1))]),
         ..Default::default()
@@ -571,7 +571,7 @@ fn test_nested_library_call() {
             gas_consumed: REQUIRED_GAS_LIBRARY_CALL_TEST,
             ..CallExecution::default()
         },
-        vm_resources: library_call_vm_resources,
+        vm_resources: library_call_vm_resources.into(),
         inner_calls: vec![nested_storage_call_info],
         ..Default::default()
     };
@@ -582,7 +582,7 @@ fn test_nested_library_call() {
             gas_consumed: REQUIRED_GAS_STORAGE_READ_WRITE_TEST,
             ..CallExecution::default()
         },
-        vm_resources: storage_entry_point_vm_resources,
+        vm_resources: storage_entry_point_vm_resources.into(),
         storage_read_values: vec![stark_felt!(value)],
         accessed_storage_keys: HashSet::from([StorageKey(patricia_key!(key))]),
         ..Default::default()
@@ -600,7 +600,7 @@ fn test_nested_library_call() {
             gas_consumed: 316180,
             ..CallExecution::default()
         },
-        vm_resources: main_call_vm_resources,
+        vm_resources: main_call_vm_resources.into(),
         inner_calls: vec![library_call_info, storage_call_info],
         ..Default::default()
     };

crates/blockifier/src/test_utils/prices.rs

@@ -81,4 +81,5 @@ fn fee_transfer_resources(
         )
         .unwrap()
         .vm_resources
+        .0
 }

crates/blockifier/src/transaction/objects.rs

@@ -3,6 +3,7 @@ use std::collections::{HashMap, HashSet};
 use cairo_felt::Felt252;
 use itertools::concat;
 use num_traits::Pow;
+use serde::Serialize;
 use starknet_api::core::{ClassHash, ContractAddress, Nonce};
 use starknet_api::data_availability::DataAvailabilityMode;
 use starknet_api::transaction::{
@@ -139,7 +140,7 @@ pub struct CommonAccountFields {
 }
 
 /// Contains the information gathered by the execution of a transaction.
-#[derive(Debug, Default, Eq, PartialEq)]
+#[derive(Debug, Default, Eq, PartialEq, Serialize)]
 pub struct TransactionExecutionInfo {
     /// Transaction validation call info; [None] for `L1Handler`.
     pub validate_call_info: Option<CallInfo>,
@@ -187,7 +188,7 @@ impl TransactionExecutionInfo {
 
 /// A mapping from a transaction execution resource to its actual usage.
 #[cfg_attr(test, derive(Clone))]
-#[derive(Debug, Default, Eq, PartialEq)]
+#[derive(Debug, Default, Eq, PartialEq, Serialize)]
 pub struct ResourcesMapping(pub HashMap<String, usize>);
 
 impl ResourcesMapping {

crates/blockifier/src/transaction/transactions_test.rs

@@ -29,6 +29,7 @@ use crate::abi::sierra_types::next_storage_key;
 use crate::block_context::BlockContext;
 use crate::execution::call_info::{
     CallExecution, CallInfo, MessageToL1, OrderedEvent, OrderedL2ToL1Message, Retdata,
+    VmExecutionResourcesWrapper,
 };
 use crate::execution::entry_point::{CallEntryPoint, CallType};
 use crate::execution::errors::{EntryPointExecutionError, VirtualMachineExecutionError};
@@ -146,7 +147,7 @@ fn expected_validate_call_info(
             initial_gas: Transaction::initial_gas(),
         },
         // The account contract we use for testing has trivial `validate` functions.
-        vm_resources,
+        vm_resources: vm_resources.into(),
         execution: CallExecution { retdata, gas_consumed, ..Default::default() },
         ..Default::default()
     })
@@ -207,7 +208,9 @@ fn expected_fee_transfer_call_info(
             events: vec![expected_fee_transfer_event],
             ..Default::default()
         },
-        vm_resources: Prices::FeeTransfer(account_address, *fee_type).into(),
+        vm_resources: VmExecutionResourcesWrapper(
+            Prices::FeeTransfer(account_address, *fee_type).into(),
+        ),
         // We read sender balance, write (which starts with read) sender balance, then the same for
         // recipient. We read Uint256(BALANCE, 0) twice, then Uint256(0, 0) twice.
         storage_read_values: vec![
@@ -381,15 +384,16 @@ fn test_invoke_tx(
             gas_consumed: expected_arguments.execute_gas_consumed,
             ..Default::default()
         },
-        vm_resources: expected_arguments.vm_resources,
+        vm_resources: expected_arguments.vm_resources.into(),
         inner_calls: vec![CallInfo {
             call: expected_return_result_call,
             execution: CallExecution::from_retdata(expected_return_result_retdata),
             vm_resources: VmExecutionResources {
                 n_steps: 23,
                 n_memory_holes: 0,
                 ..Default::default()
-            },
+            }
+            .into(),
             ..Default::default()
         }],
         ..Default::default()
@@ -1626,7 +1630,8 @@ fn test_l1_handler() {
             n_steps: 143,
             n_memory_holes: 1,
             builtin_instance_counter: HashMap::from([(RANGE_CHECK_BUILTIN_NAME.to_string(), 5)]),
-        },
+        }
+        .into(),
         accessed_storage_keys: HashSet::from_iter(vec![accessed_storage_key]),
         ..Default::default()
     };

crates/native_blockifier/src/py_transaction_execution_info.rs

@@ -101,7 +101,7 @@ impl From<CallInfo> for PyCallInfo {
             gas_consumed: execution.gas_consumed,
             failure_flag: PyFelt::from(execution.failed as u8),
             retdata: to_py_vec(execution.retdata.0, PyFelt),
-            execution_resources: PyVmExecutionResources::from(call_info.vm_resources),
+            execution_resources: PyVmExecutionResources::from(call_info.vm_resources.0),
             events: to_py_vec(execution.events, PyOrderedEvent::from),
             l2_to_l1_messages: to_py_vec(execution.l2_to_l1_messages, PyOrderedL2ToL1Message::from),
             internal_calls: to_py_vec(call_info.inner_calls, PyCallInfo::from),