From c495bca4adef24811e92f9dfa5149b32ac81ef65 Mon Sep 17 00:00:00 2001
From: cdetrio <cdetrio@gmail.com>
Date: Tue, 2 Jan 2018 06:49:17 -0500
Subject: [PATCH] eth: enable preimage recording when tracing (#15787)

---
 eth/api_tracer.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/eth/api_tracer.go b/eth/api_tracer.go
index 0d0e2a73c..8ad664e06 100644
--- a/eth/api_tracer.go
+++ b/eth/api_tracer.go
@@ -667,12 +667,15 @@ func (api *PrivateDebugAPI) traceTx(ctx context.Context, message core.Message, v
 		tracer = vm.NewStructLogger(config.LogConfig)
 	}
 	// Run the transaction with tracing enabled.
-	vmenv := vm.NewEVM(vmctx, statedb, api.config, vm.Config{Debug: true, Tracer: tracer})
+	vmenv := vm.NewEVM(vmctx, statedb, api.config, vm.Config{Debug: true, Tracer: tracer, EnablePreimageRecording: true})
 
 	ret, gas, failed, err := core.ApplyMessage(vmenv, message, new(core.GasPool).AddGas(message.Gas()))
 	if err != nil {
 		return nil, fmt.Errorf("tracing failed: %v", err)
 	}
+	if err := core.WritePreimages(api.eth.ChainDb(), vmctx.BlockNumber.Uint64(), statedb.Preimages()); err != nil {
+		return nil, fmt.Errorf("Error writing preimage from trace: %v", err)
+	}
 	// Depending on the tracer type, format and return the output
 	switch tracer := tracer.(type) {
 	case *vm.StructLogger:
-- 
GitLab