Add state.reward processing

ethereum · Feb 8, 2023 · 390393f · 390393f
1 parent 886174b
commit 390393f
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/test/t8n/t8n.cpp b/test/t8n/t8n.cpp
@@ -30,6 +30,7 @@ int main(int argc, const char* argv[])
     fs::path output_dir;
     fs::path output_result_file;
     fs::path output_alloc_file;
+    intx::uint256 state_reward;
 
     for (int i = 0; i < argc; ++i)
     {
@@ -54,6 +55,14 @@ int main(int argc, const char* argv[])
             output_result_file = argv[i];
         else if (arg == "--output.alloc" && ++i < argc)
             output_alloc_file = argv[i];
+        else if (arg == "--state.reward" && ++i < argc)
+        {
+            if (std::string(argv[i]) == "-1")
+                state_reward = std::numeric_limits<intx::uint256>::max();
+            else
+                state_reward = intx::from_string<intx::uint256>(argv[i]);
+        }
+
     }
 
     state::BlockInfo block;
@@ -128,6 +137,16 @@ int main(int argc, const char* argv[])
                 }
             }
 
+            if (state_reward != 0 && state_reward != std::numeric_limits<intx::uint256>::max())
+                state.get_or_insert(block.coinbase).balance += state_reward;
+            else if (state_reward == 0)
+                state.touch(block.coinbase);
+            else if (state_reward == std::numeric_limits<intx::uint256>::max())
+            {
+                if (state.get_or_insert(block.coinbase).is_empty())
+                    state.get_accounts().erase(block.coinbase);
+            }
+
             j_result["logsHash"] = hex0x(logs_hash(txs_logs));
             j_result["stateRoot"] = hex0x(state::mpt_hash(state.get_accounts()));
         }