gnolang/gno: prototype Machine.RunContext

gnovm/pkg/gnolang/gno_test.go

@@ -2,6 +2,7 @@
 
 import (
 	"bytes"
+	"context"
 	"fmt"
 	"io"
 	"os"
@@ -10,6 +11,7 @@
 	"strings"
 	"testing"
 	"text/template"
+	"time"
 	"unsafe"
 
 	// "github.com/davecgh/go-spew/spew"
@@ -291,6 +293,31 @@
 	m.RunMain()
 }
 
+func TestRunMemoryAllocationBuster(t *testing.T) {
+	t.Parallel()
+
+	m := NewMachine("test", nil)
+	c := `package test
+func main() {
+    var x interface{}    
+    for i := 0; i < 10000; i++ {
+        x = [1]interface{}{x}    
+    }
+    for i := 0; i < 10000; i++ {
+        println(x)   
+    }
+}
+`
+	n := MustParseFile("main.go", c)
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+	m.SetRunContext(ctx)
+
+	m.RunFiles(n)
+	m.RunMain()
+
+}
+
 func BenchmarkPreprocessForLoop(b *testing.B) {
 	m := NewMachine("test", nil)
 	c := `package test

gnovm/pkg/gnolang/machine.go

@@ -3,6 +3,7 @@ package gnolang
 // XXX rename file to machine.go.
 
 import (
+	"context"
 	"fmt"
 	"io"
 	"reflect"
@@ -79,6 +80,13 @@ type Machine struct {
 	// it is executed. It is reset to zero after the defer functions in the current
 	// scope have finished executing.
 	DeferPanicScope uint
+
+	runCtx context.Context
+}
+
+func (m *Machine) RunContext(ctx context.Context) {
+	m.runCtx = ctx
+	m.Run()
 }
 
 // NewMachine initializes a new gno virtual machine, acting as a shorthand
@@ -98,6 +106,12 @@ func NewMachine(pkgPath string, store Store) *Machine {
 		})
 }
 
+func (m *Machine) SetRunContext(ctx context.Context) {
+	if m != nil {
+		m.runCtx = ctx
+	}
+}
+
 // MachineOptions is used to pass options to [NewMachineWithOptions].
 type MachineOptions struct {
 	// Active package of the given machine; must be set before execution.
@@ -810,7 +824,11 @@ func (m *Machine) RunStatement(s Stmt) {
 	m.PushOp(OpHalt)
 	m.PushStmt(s)
 	m.PushOp(OpExec)
-	m.Run()
+	ctx := m.runCtx
+	if ctx == nil {
+		ctx = context.Background()
+	}
+	m.RunContext(ctx)
 }
 
 // Runs a declaration after preprocessing d.  If d was already
@@ -1168,6 +1186,12 @@ func (m *Machine) Run() {
 		if m.Debugger.enabled {
 			m.Debug()
 		}
+		if m.runCtx != nil {
+			if err := m.runCtx.Err(); err != nil {
+				panic(err)
+			}
+		}
+
 		op := m.PopOp()
 		if bm.OpsEnabled {
 			// benchmark the operation.