baseline: VM-level ExecutionStates

ethereum · Sep 12, 2024 · b2e5484 · b2e5484
1 parent 19a527e
commit b2e5484
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/lib/evmone/vm.cpp b/lib/evmone/vm.cpp
@@ -72,7 +72,7 @@ evmc_set_option_result set_option(evmc_vm* c_vm, char const* c_name, char const*
 }  // namespace
 
 
-inline constexpr VM::VM() noexcept
+VM::VM() noexcept
   : evmc_vm{
         EVMC_ABI_VERSION,
         "evmone",
@@ -82,7 +82,21 @@ inline constexpr VM::VM() noexcept
         evmone::get_capabilities,
         evmone::set_option,
     }
-{}
+{
+    m_execution_states.reserve(1025);
+}
+
+ExecutionState& VM::get_execution_state(size_t depth) noexcept
+{
+    // Vector already has the capacity for all possible depths,
+    // so reallocation never happens (therefore: noexcept).
+    // TODO: Maybe just use std::array?
+    // FIXME: unit test where max depth is reached is needed.
+    assert(depth < m_execution_states.capacity());
+    if (m_execution_states.size() <= depth)
+        m_execution_states.resize(depth + 1);
+    return m_execution_states[depth];
+}
 
 }  // namespace evmone
 

diff --git a/lib/evmone/vm.hpp b/lib/evmone/vm.hpp
@@ -3,8 +3,10 @@
 // SPDX-License-Identifier: Apache-2.0
 #pragma once
 
+#include "execution_state.hpp"
 #include "tracing.hpp"
 #include <evmc/evmc.h>
+#include <vector>
 
 #if defined(_MSC_VER) && !defined(__clang__)
 #define EVMONE_CGOTO_SUPPORTED 0
@@ -22,10 +24,13 @@ class VM : public evmc_vm
     bool validate_eof = false;
 
 private:
+    std::vector<ExecutionState> m_execution_states;
     std::unique_ptr<Tracer> m_first_tracer;
 
 public:
-    inline constexpr VM() noexcept;
+    VM() noexcept;
+
+    [[nodiscard]] ExecutionState& get_execution_state(size_t depth) noexcept;
 
     void add_tracer(std::unique_ptr<Tracer> tracer) noexcept
     {