Merge pull request ethereum#26 from OffchainLabs/hostio-tracer

Hostio tracer
streamingfast · Sep 14, 2023 · 4d61d86 · 4d61d86
2 parents 69c097f + 27113b8
commit 4d61d86
Show file tree

Hide file tree

Showing 5 changed files with 37 additions and 0 deletions.
diff --git a/core/vm/logger.go b/core/vm/logger.go
@@ -33,6 +33,9 @@ type EVMLogger interface {
 	CaptureArbitrumStorageGet(key common.Hash, depth int, before bool)
 	CaptureArbitrumStorageSet(key, value common.Hash, depth int, before bool)
 
+	// Stylus: capture hostio invocation
+	CaptureStylusHostio(name string, args, outs []byte, ink uint64)
+
 	// Transaction level
 	CaptureTxStart(gasLimit uint64)
 	CaptureTxEnd(restGas uint64)

diff --git a/eth/tracers/js/tracer_arbitrum.go b/eth/tracers/js/tracer_arbitrum.go
@@ -55,3 +55,20 @@ func (jst *jsTracer) CaptureArbitrumTransfer(
 
 func (*jsTracer) CaptureArbitrumStorageGet(key common.Hash, depth int, before bool)        {}
 func (*jsTracer) CaptureArbitrumStorageSet(key, value common.Hash, depth int, before bool) {}
+
+func (jst *jsTracer) CaptureStylusHostio(name string, args, outs []byte, ink uint64) {
+	hostio, ok := goja.AssertFunction(jst.obj.Get("hostio"))
+	if !ok {
+		return
+	}
+
+	info := jst.vm.NewObject()
+	info.Set("name", name)
+	info.Set("args", args)
+	info.Set("outs", outs)
+	info.Set("ink", ink)
+
+	if _, err := hostio(jst.obj, info); err != nil {
+		jst.err = wrapError("hostio", err)
+	}
+}
diff --git a/eth/tracers/logger/logger_arbitrum.go b/eth/tracers/logger/logger_arbitrum.go
@@ -41,3 +41,8 @@ func (*AccessListTracer) CaptureArbitrumStorageSet(key, value common.Hash, depth
 func (*JSONLogger) CaptureArbitrumStorageSet(key, value common.Hash, depth int, before bool)       {}
 func (*StructLogger) CaptureArbitrumStorageSet(key, value common.Hash, depth int, before bool)     {}
 func (*mdLogger) CaptureArbitrumStorageSet(key, value common.Hash, depth int, before bool)         {}
+
+func (*AccessListTracer) CaptureStylusHostio(name string, args, outs []byte, ink uint64) {}
+func (*JSONLogger) CaptureStylusHostio(name string, args, outs []byte, ink uint64)       {}
+func (*StructLogger) CaptureStylusHostio(name string, args, outs []byte, ink uint64)     {}
+func (*mdLogger) CaptureStylusHostio(name string, args, outs []byte, ink uint64)         {}
diff --git a/eth/tracers/native/mux.go b/eth/tracers/native/mux.go
@@ -131,6 +131,12 @@ func (t *muxTracer) CaptureArbitrumTransfer(env *vm.EVM, from, to *common.Addres
 	}
 }
 
+func (t *muxTracer) CaptureStylusHostio(name string, args, outs []byte, ink uint64) {
+	for _, t := range t.tracers {
+		t.CaptureStylusHostio(name, args, outs, ink)
+	}
+}
+
 // GetResult returns an empty json object.
 func (t *muxTracer) GetResult() (json.RawMessage, error) {
 	resObject := make(map[string]json.RawMessage)

diff --git a/eth/tracers/native/tracer_arbitrum.go b/eth/tracers/native/tracer_arbitrum.go
@@ -90,6 +90,12 @@ func (*noopTracer) CaptureArbitrumStorageSet(key, value common.Hash, depth int,
 func (*prestateTracer) CaptureArbitrumStorageSet(key, value common.Hash, depth int, before bool) {}
 func (*flatCallTracer) CaptureArbitrumStorageSet(key, value common.Hash, depth int, before bool) {}
 
+func (*callTracer) CaptureStylusHostio(name string, args, outs []byte, ink uint64)     {}
+func (*fourByteTracer) CaptureStylusHostio(name string, args, outs []byte, ink uint64) {}
+func (*noopTracer) CaptureStylusHostio(name string, args, outs []byte, ink uint64)     {}
+func (*prestateTracer) CaptureStylusHostio(name string, args, outs []byte, ink uint64) {}
+func (*flatCallTracer) CaptureStylusHostio(name string, args, outs []byte, ink uint64) {}
+
 func bigToHex(n *big.Int) string {
 	if n == nil {
 		return ""