Validate EOF code before execution in evm fixture

ethereum · Apr 13, 2023 · d7eb5d5 · d7eb5d5
1 parent 4caa839
commit d7eb5d5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/test/unittests/evm_fixture.hpp b/test/unittests/evm_fixture.hpp
@@ -3,6 +3,7 @@
 // SPDX-License-Identifier: Apache-2.0
 #pragma once
 
+#include "evmone/eof.hpp"
 #include <evmc/mocked_host.hpp>
 #include <gtest/gtest.h>
 #include <intx/intx.hpp>
@@ -79,6 +80,12 @@ class evm : public testing::TestWithParam<evmc::VM*>
             host.access_account(msg.recipient);
         }
 
+        if (rev >= EVMC_CANCUN && is_eof_container(code))
+        {
+            ASSERT_EQ(get_error_message(validate_eof(rev, code)),
+                get_error_message(EOFValidationError::success));
+        }
+
         result = vm.execute(host, rev, msg, code.data(), code.size());
         output = {result.output_data, result.output_size};
         gas_used = msg.gas - result.gas_left;