[NIT-2640] Add EVM tracing for Stylus programs

arbitrator/wasm-libraries/user-host-trait/src/lib.rs

@@ -534,7 +534,7 @@ pub trait UserHost<DR: DataReader>: GasMeteredMachine {
     fn return_data_size(&mut self) -> Result<u32, Self::Err> {
         self.buy_ink(HOSTIO_INK)?;
         let len = *self.evm_return_data_len();
-        trace!("return_data_size", self, be!(len), &[], len)
+        trace!("return_data_size", self, &[], be!(len), len)
     }
 
     /// Emits an EVM log with the given number of topics and data, the first bytes of which should
@@ -629,7 +629,8 @@ pub trait UserHost<DR: DataReader>: GasMeteredMachine {
         self.buy_gas(gas_cost)?;
 
         let code = code.slice();
-        trace!("account_code_size", self, address, &[], code.len() as u32)
+        let len = code.len() as u32;
+        trace!("account_code_size", self, address, be!(len), len)
     }
 
     /// Gets the code hash of the account at the given address. The semantics are equivalent
@@ -735,7 +736,7 @@ pub trait UserHost<DR: DataReader>: GasMeteredMachine {
     fn evm_gas_left(&mut self) -> Result<u64, Self::Err> {
         self.buy_ink(HOSTIO_INK)?;
         let gas = self.gas_left()?;
-        trace!("evm_gas_left", self, be!(gas), &[], gas)
+        trace!("evm_gas_left", self, &[], be!(gas), gas)
     }
 
     /// Gets the amount of ink remaining after paying for the cost of this hostio. The semantics
@@ -747,7 +748,7 @@ pub trait UserHost<DR: DataReader>: GasMeteredMachine {
     fn evm_ink_left(&mut self) -> Result<u64, Self::Err> {
         self.buy_ink(HOSTIO_INK)?;
         let ink = self.ink_ready()?;
-        trace!("evm_ink_left", self, be!(ink), &[], ink)
+        trace!("evm_ink_left", self, &[], be!(ink), ink)
     }
 
     /// Computes `value ÷ exponent` using 256-bit math, writing the result to the first.

arbos/programs/api.go

@@ -65,14 +65,10 @@ func newApiClosures(
 	actingAddress := contract.Address() // not necessarily WASM
 	readOnly := interpreter.ReadOnly()
 	evm := interpreter.Evm()
-	depth := evm.Depth()
 	db := evm.StateDB
 	chainConfig := evm.ChainConfig()
 
 	getBytes32 := func(key common.Hash) (common.Hash, uint64) {
-		if tracingInfo != nil {
-			tracingInfo.RecordStorageGet(key)
-		}
 		cost := vm.WasmStateLoadCost(db, actingAddress, key)
 		return db.GetState(actingAddress, key), cost
 	}
@@ -82,9 +78,6 @@ func newApiClosures(
 			value := common.BytesToHash(data[32:64])
 			data = data[64:]
 
-			if tracingInfo != nil {
-				tracingInfo.RecordStorageSet(key, value)
-			}
 			if readOnly {
 				return WriteProtection
 			}
@@ -144,22 +137,7 @@ func newApiClosures(
 
 		// Tracing: emit the call (value transfer is done later in evm.Call)
 		if tracingInfo != nil {
-			var args []uint256.Int
-			args = append(args, *uint256.NewInt(gas))                          // gas
-			args = append(args, *uint256.NewInt(0).SetBytes(contract.Bytes())) // to address
-			if opcode == vm.CALL {
-				args = append(args, *uint256.NewInt(0).SetBytes(value.Bytes())) // call value
-			}
-			args = append(args, *uint256.NewInt(0))                  // memory offset
-			args = append(args, *uint256.NewInt(uint64(len(input)))) // memory length
-			args = append(args, *uint256.NewInt(0))                  // return offset
-			args = append(args, *uint256.NewInt(0))                  // return size
-			s := &vm.ScopeContext{
-				Memory:   util.TracingMemoryFromBytes(input),
-				Stack:    util.TracingStackFromArgs(args...),
-				Contract: scope.Contract,
-			}
-			tracingInfo.Tracer.CaptureState(0, opcode, startGas, baseCost+gas, s, []byte{}, depth, nil)
+			tracingInfo.CaptureStylusCall(opcode, contract, value, input, gas, startGas, baseCost)
 		}
 
 		var ret []byte
@@ -217,11 +195,6 @@ func newApiClosures(
 		one64th := gas / 64
 		gas -= one64th
 
-		// Tracing: emit the create
-		if tracingInfo != nil {
-			tracingInfo.Tracer.CaptureState(0, opcode, startGas, baseCost+gas, scope, []byte{}, depth, nil)
-		}
-
 		var res []byte
 		var addr common.Address // zero on failure
 		var returnGas uint64
@@ -243,9 +216,6 @@ func newApiClosures(
 		return addr, res, cost, nil
 	}
 	emitLog := func(topics []common.Hash, data []byte) error {
-		if tracingInfo != nil {
-			tracingInfo.RecordEmitLog(topics, data)
-		}
 		if readOnly {
 			return vm.ErrWriteProtection
 		}
@@ -284,10 +254,7 @@ func newApiClosures(
 	}
 	captureHostio := func(name string, args, outs []byte, startInk, endInk uint64) {
 		tracingInfo.Tracer.CaptureStylusHostio(name, args, outs, startInk, endInk)
-		if name == "evm_gas_left" || name == "evm_ink_left" {
-			tracingInfo.Tracer.CaptureState(0, vm.GAS, 0, 0, scope, []byte{}, depth, nil)
-			tracingInfo.Tracer.CaptureState(0, vm.POP, 0, 0, scope, []byte{}, depth, nil)
-		}
+		tracingInfo.CaptureEVMTraceForHostio(name, args, outs, startInk, endInk)
 	}
 
 	return func(req RequestType, input []byte) ([]byte, []byte, uint64) {

arbos/programs/native.go

@@ -206,6 +206,9 @@ func callProgram(
 	if status == userFailure && debug {
 		log.Warn("program failure", "err", err, "msg", msg, "program", address, "depth", depth)
 	}
+	if tracingInfo != nil {
+		tracingInfo.CaptureStylusExit(uint8(status), data, err, scope.Contract.Gas)
+	}
 	return data, err
 }