Support delegation clearing in StateDiff API

ethereum · Nov 22, 2024 · 5c8549e · 5c8549e
1 parent d487ec5
commit 5c8549e
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/test/state/state.cpp b/test/state/state.cpp
@@ -125,6 +125,8 @@ StateDiff State::build_diff(evmc_revision rev) const
         // TODO: Output also the code hash. It will be needed for DB update and MPT hash.
         if ((m.just_created && !m.code.empty()) || m.code_changed)
             a.code = m.code;
+        if (m.code_changed && m.code.empty())
+            a.code_cleared = true;
 
         for (const auto& [k, v] : m.storage)
         {

diff --git a/test/state/state_diff.hpp b/test/state/state_diff.hpp
@@ -33,6 +33,9 @@ struct StateDiff
         /// New account code. If empty, it means the code has not changed.
         bytes code;
 
+        /// If true, account code was emptied due to resetting EIP-7702 delegation.
+        bool code_cleared = false;
+
         /// The list of the account's storage modifications: key => new value.
         /// The value 0 means the storage entry is deleted.
         std::vector<std::pair<bytes32, bytes32>> modified_storage;

diff --git a/test/state/test_state.cpp b/test/state/test_state.cpp
@@ -36,6 +36,8 @@ void TestState::apply(const state::StateDiff& diff)
         a.balance = m.balance;
         if (!m.code.empty())
             a.code = m.code;  // TODO: Consider taking rvalue ref to avoid code copy.
+        if (m.code_cleared)
+            a.code.clear();
         for (const auto& [k, v] : m.modified_storage)
         {
             if (v)