WIP difficulty check

Author: gumb0

test/blockchaintest/blockchaintest_runner.cpp

@@ -2,6 +2,7 @@
 // Copyright 2023 The evmone Authors.
 // SPDX-License-Identifier: Apache-2.0
 
+#include "../state/ethash_difficulty.hpp"
 #include "../state/mpt_hash.hpp"
 #include "../state/requests.hpp"
 #include "../state/rlp.hpp"
@@ -107,8 +108,8 @@ TransitionResult apply_block(const TestState& state, evmc::VM& vm, const state::
         bloom, blob_gas_left, std::move(block_state)};
 }
 
-bool validate_block(
-    evmc_revision rev, const TestBlock& test_block, const BlockHeader* parent_header) noexcept
+bool validate_block(evmc_revision rev, const TestBlock& test_block,
+    const BlockHeader* parent_header, bool parent_has_ommers) noexcept
 {
     // NOTE: includes only block validity unrelated to individual txs. See `apply_block`.
 
@@ -136,6 +137,12 @@ bool validate_block(
     if (test_block.block_info.gas_limit < 5000)
         return false;
 
+    if (test_block.block_info.difficulty != state::calculate_difficulty(parent_header->difficulty,
+                                                parent_has_ommers, parent_header->timestamp,
+                                                test_block.block_info.timestamp,
+                                                test_block.block_info.number, rev))
+        return false;
+
     if (rev >= EVMC_PARIS && !test_block.block_info.ommers.empty())
         return false;
 
@@ -257,11 +264,12 @@ void run_blockchain_tests(std::span<const BlockchainTest> tests, evmc::VM& vm)
         struct BlockData
         {
             const BlockHeader* header;
+            bool has_ommers = false;
             TestState post_state;
             intx::uint256 total_difficulty;
         };
         std::unordered_map<hash256, BlockData> block_data{{{c.genesis_block_header.hash,
-            {&c.genesis_block_header, c.pre_state, c.genesis_block_header.difficulty}}}};
+            {&c.genesis_block_header, false, c.pre_state, c.genesis_block_header.difficulty}}}};
         const auto* canonical_state = &c.pre_state;
         intx::uint256 max_total_difficulty = c.genesis_block_header.difficulty;
 
@@ -273,6 +281,8 @@ void run_blockchain_tests(std::span<const BlockchainTest> tests, evmc::VM& vm)
             const auto parent_data_it = block_data.find(test_block.block_info.parent_hash);
             const auto* parent_header =
                 parent_data_it != block_data.end() ? parent_data_it->second.header : nullptr;
+            const auto parent_has_ommers =
+                parent_data_it != block_data.end() && parent_data_it->second.has_ommers;
 
             const auto rev = c.rev.get_revision(bi.timestamp);
 
@@ -281,7 +291,7 @@ void run_blockchain_tests(std::span<const BlockchainTest> tests, evmc::VM& vm)
 
             if (test_block.valid)
             {
-                ASSERT_TRUE(validate_block(rev, test_block, parent_header))
+                ASSERT_TRUE(validate_block(rev, test_block, parent_header, parent_has_ommers))
                     << "Expected block to be valid (validate_block)";
 
                 // Block being valid guarantees its parent was found.
@@ -296,7 +306,8 @@ void run_blockchain_tests(std::span<const BlockchainTest> tests, evmc::VM& vm)
                 block_hashes[test_block.expected_block_header.block_number] =
                     test_block.expected_block_header.hash;
                 const auto [inserted_it, _] = block_data.insert({test_block.block_info.hash,
-                    {&test_block.expected_block_header, std::move(res.block_state),
+                    {&test_block.expected_block_header, !test_block.block_info.ommers.empty(),
+                        std::move(res.block_state),
                         parent_data_it->second.total_difficulty +
                             test_block.block_info.difficulty}});
                 if (inserted_it->second.total_difficulty >= max_total_difficulty)
@@ -334,7 +345,7 @@ void run_blockchain_tests(std::span<const BlockchainTest> tests, evmc::VM& vm)
             }
             else
             {
-                if (!validate_block(rev, test_block, parent_header))
+                if (!validate_block(rev, test_block, parent_header, parent_has_ommers))
                     continue;
 
                 // Block being valid guarantees its parent was found.
@@ -385,7 +396,6 @@ void run_blockchain_tests(std::span<const BlockchainTest> tests, evmc::VM& vm)
                            print_state(std::get<TestState>(c.expectation.post_state)) :
                        "");
     }
-    // TODO: Add difficulty calculation verification.
 }
 
 }  // namespace evmone::test