Add test to check compatibility between witness and zero trace

core/state/database_test.go

@@ -20,6 +20,7 @@ import (
 	"bytes"
 	"context"
 	"encoding/hex"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"math/big"
@@ -1339,6 +1340,78 @@ func TestGetStateFromWitness(t *testing.T) {
 	fmt.Printf("Balance: %s\n", balance)
 }
 
+func TestValidateZeroTracer(t *testing.T) {
+	// Load a block trace from a file
+	d, err := os.ReadFile("testdata/blockZeroTrace.json")
+	if err != nil {
+		t.Fatalf("error reading trace: %v", err)
+	}
+
+	// Decode data into json
+	var trace types.BlockTrace
+	err = json.Unmarshal(d, &trace)
+	if err != nil {
+		t.Fatalf("error decoding trace: %v", err)
+	}
+
+	witness, err := trie.NewWitnessFromReader(bytes.NewReader(trace.TriePreImage.Combined.Compact), false)
+
+	if err != nil {
+		t.Fatalf("error reading witness: %v", err)
+	}
+
+	s, err := state.NewStateless(libcommon.Hash{}, witness, 0, false, true)
+
+	if err != nil {
+		t.Fatalf("error creating state: %v", err)
+	}
+
+	s.SetStrictHash(true)
+
+	ibs := state.New(s)
+
+	for _, txnInfo := range trace.TxnInfo {
+		for addr, accTrace := range txnInfo.Traces {
+			ibs.GetBalance(addr)
+
+			if accTrace.Balance != nil {
+				ibs.SetBalance(addr, accTrace.Balance)
+			}
+
+			if accTrace.Nonce != nil {
+				ibs.SetNonce(addr, accTrace.Nonce.Uint64())
+			}
+
+			if accTrace.CodeUsage != nil {
+				if accTrace.CodeUsage.Read != nil {
+					ibs.GetCode(addr)
+				}
+
+				if accTrace.CodeUsage.Write != nil {
+					ibs.SetCode(addr, accTrace.CodeUsage.Write)
+				}
+			}
+
+			if accTrace.SelfDestructed != nil && *accTrace.SelfDestructed {
+				ibs.Selfdestruct(addr)
+			}
+
+			for _, r := range accTrace.StorageRead {
+				var value uint256.Int
+				ibs.GetState(addr, &r, &value)
+			}
+
+			for k, v := range accTrace.StorageWritten {
+				ibs.SetState(addr, &k, *v)
+			}
+		}
+
+		ibs.FinalizeTx(&chain.Rules{}, s)
+	}
+
+	s.Finalize()
+}
+
 func TestTDSWitness(t *testing.T) {
 	contract := libcommon.HexToAddress("0x71dd1027069078091B3ca48093B00E4735B20624")
 	sKey := libcommon.HexToHash("0x4321")

core/state/stateless.go

@@ -85,6 +85,10 @@ func (s *Stateless) SetBlockNr(blockNr uint64) {
 	s.blockNr = blockNr
 }
 
+func (s *Stateless) SetStrictHash(strict bool) {
+	s.t.SetStrictHash(strict)
+}
+
 // ReadAccountData is a part of the StateReader interface
 // This implementation attempts to look up account data in the state trie, and fails if it is not found
 func (s *Stateless) ReadAccountData(address common.Address) (*accounts.Account, error) {