From 2455b29b82ca72a3bb6a5972007c98298b54e4b6 Mon Sep 17 00:00:00 2001
From: JukLee0ira <meebookMonkey@163.com>
Date: Tue, 22 Oct 2024 16:05:27 +0800
Subject: [PATCH] eth/tracer:compare opcode in CaptureState

---
 eth/tracers/native/contract.go | 57 ++++++++++++++--------------------
 1 file changed, 24 insertions(+), 33 deletions(-)

diff --git a/eth/tracers/native/contract.go b/eth/tracers/native/contract.go
index 361a84e84219..b9321942b6e4 100644
--- a/eth/tracers/native/contract.go
+++ b/eth/tracers/native/contract.go
@@ -66,8 +66,8 @@ func NewContractTracer(cfg json.RawMessage) (tracers.Tracer, error) {
 }
 
 func (t *contractTracer) CaptureStart(env *vm.EVM, from common.Address, to common.Address, create bool, input []byte, gas uint64, value *big.Int) {
-	if create {
-		validateAndStoreOpCode(t, input, to)
+	if create || t.config.OpCode == "CALL" {
+		storeOpCode(t, env, to, input)
 	}
 }
 
@@ -75,22 +75,28 @@ func (t *contractTracer) CaptureEnd(output []byte, gasUsed uint64, _ time.Durati
 }
 
 func (t *contractTracer) CaptureState(env *vm.EVM, pc uint64, op vm.OpCode, gas, cost uint64, scope *vm.ScopeContext, rData []byte, depth int, err error) {
-}
-
-func (t *contractTracer) CaptureFault(env *vm.EVM, pc uint64, op vm.OpCode, gas, cost uint64, _ *vm.ScopeContext, depth int, err error) {
-}
-
-func (t *contractTracer) CaptureEnter(typ vm.OpCode, from common.Address, to common.Address, input []byte, gas uint64, value *big.Int) {
 	// Skip if tracing was interrupted
 	if atomic.LoadUint32(&t.interrupt) > 0 {
 		// TODO: env.Cancel()
 		return
 	}
-	if typ == vm.CREATE || typ == vm.CREATE2 {
-		validateAndStoreOpCode(t, input, to)
+	// If the OpCode is empty , exit early.
+	if t.config.OpCode == "" {
+		return
+	}
+	if op.String() == t.config.OpCode {
+		addr := scope.Contract.Address()
+		bytecode := env.StateDB.GetCode(addr)
+		storeOpCode(t, nil, addr, bytecode)
 	}
 }
 
+func (t *contractTracer) CaptureFault(env *vm.EVM, pc uint64, op vm.OpCode, gas, cost uint64, _ *vm.ScopeContext, depth int, err error) {
+}
+
+func (t *contractTracer) CaptureEnter(typ vm.OpCode, from common.Address, to common.Address, input []byte, gas uint64, value *big.Int) {
+}
+
 func (t *contractTracer) CaptureExit(output []byte, gasUsed uint64, err error) {
 }
 
@@ -107,30 +113,15 @@ func (t *contractTracer) Stop(err error) {
 	atomic.StoreUint32(&t.interrupt, 1)
 }
 
-func validateAndStoreOpCode(t *contractTracer, input []byte, to common.Address) {
-	// If the OpCode is not empty and doesn't match the input, exit early.
-	if t.config.OpCode != "" && !findOpcodes(input, vm.StringToOp(t.config.OpCode)) {
-		return
-	}
-	// If WithByteCode is true, store the input in the address mapping as hex.
+func storeOpCode(t *contractTracer, env *vm.EVM, addr common.Address, code []byte) {
+	t.Addrs[addrToHex(addr)] = ""
 	if t.config.WithByteCode {
-		t.Addrs[addrToHex(to)] = bytesToHex(input)
-	} else {
-		t.Addrs[addrToHex(to)] = ""
-	}
-}
-
-// Compare bytecode with the given opcode, skipping PUSH instructions.
-func findOpcodes(bytecode []byte, opcode vm.OpCode) bool {
-	for i := 0; i < len(bytecode); {
-		op := vm.OpCode(bytecode[i])
-		// Skip PUSH opcodes and their arguments
-		if op.IsPush() {
-			i += int(op - 95) // Directly calculate the number of bytes to skip
-		} else if op == opcode {
-			return true
+		if env != nil {
+			//SetCode with input code
+			env.StateDB.SetCode(addr, code)
+			code = env.StateDB.GetCode(addr)
 		}
-		i++
+		t.Addrs[addrToHex(addr)] = bytesToHex(code)
 	}
-	return false
+
 }