From 228f9631886ef03aabae7ef26ab4032e109c4110 Mon Sep 17 00:00:00 2001
From: Diego Ximenes <dxmendes1@gmail.com>
Date: Wed, 11 Dec 2024 13:28:31 -0300
Subject: [PATCH] Adds benchmark to user-host Program

---
 arbitrator/wasm-libraries/user-host/src/program.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/arbitrator/wasm-libraries/user-host/src/program.rs b/arbitrator/wasm-libraries/user-host/src/program.rs
index 7b3782b2e5..fb11efa447 100644
--- a/arbitrator/wasm-libraries/user-host/src/program.rs
+++ b/arbitrator/wasm-libraries/user-host/src/program.rs
@@ -2,6 +2,7 @@
 // For license information, see https://github.com/OffchainLabs/nitro/blob/master/LICENSE
 
 use arbutil::{
+    benchmark::Benchmark,
     evm::{
         api::{EvmApiMethod, Gas, Ink, VecReader, EVM_API_METHOD_REQ_OFFSET},
         req::{EvmApiRequestor, RequestHandler},
@@ -75,6 +76,8 @@ pub(crate) struct Program {
     pub evm_api: EvmApiRequestor<VecReader, UserHostRequester>,
     /// EVM Context info.
     pub evm_data: EvmData,
+    // Used to benchmark execution blocks of code
+    pub benchmark: Option<Benchmark>,
     /// WAVM module index.
     pub module: u32,
     /// Call configuration.
@@ -167,6 +170,7 @@ impl Program {
             outs: vec![],
             evm_api: EvmApiRequestor::new(UserHostRequester::default()),
             evm_data,
+            benchmark: None,
             module,
             config,
             early_exit: None,
@@ -237,6 +241,10 @@ impl UserHost<VecReader> for Program {
         &self.evm_data
     }
 
+    fn benchmark(&mut self) -> &mut Option<Benchmark> {
+        &mut self.benchmark
+    }
+
     fn evm_return_data_len(&mut self) -> &mut u32 {
         &mut self.evm_data.return_data_len
     }