feat: trace state change only if store is nil (#11)

DeBankDeFi · Jul 5, 2022 · 4a41150 · 4a41150
1 parent 3e9f6a2
commit 4a41150
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/txtracev1/trace_logger.go b/pkg/txtracev1/trace_logger.go
@@ -286,7 +286,7 @@ func (ot *OeTracer) CaptureState(pc uint64, op vm.OpCode, gas, cost uint64, scop
 		fromTrace.childTraces = append(fromTrace.childTraces, trace)
 	case vm.SSTORE:
 		stackLen := len(stack.Data())
-		if stackLen >= 2 {
+		if stackLen >= 2 && ot.store == nil {
 			accountAddress := contract.Address()
 			if ot.stateDiff[accountAddress] == nil {
 				ot.stateDiff[accountAddress] = make(AccountDiff)

diff --git a/pkg/txtracev2/trace_logger.go b/pkg/txtracev2/trace_logger.go
@@ -300,7 +300,7 @@ func (ot *OeTracer) CaptureState(pc uint64, op vm.OpCode, gas, cost uint64, scop
 		ot.traceStack[len(ot.traceStack)-1].Error = "execution reverted"
 	case vm.SSTORE:
 		stackLen := len(scope.Stack.Data())
-		if stackLen >= 2 {
+		if stackLen >= 2 && ot.store == nil {
 			accountAddress := scope.Contract.Address()
 			if ot.stateDiff[accountAddress] == nil {
 				ot.stateDiff[accountAddress] = make(AccountDiff)