From e16cdaec850b395071001688d83ef1160407c284 Mon Sep 17 00:00:00 2001
From: deadprogram <ron@hybridgroup.com>
Date: Sat, 30 Mar 2024 00:06:40 +0100
Subject: [PATCH] interp/wazero: add memory debugging and correct Halt()
 implementation

Signed-off-by: deadprogram <ron@hybridgroup.com>
---
 interp/wazero/interp.go | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/interp/wazero/interp.go b/interp/wazero/interp.go
index ab47809..a304d01 100644
--- a/interp/wazero/interp.go
+++ b/interp/wazero/interp.go
@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"io"
 
+	"github.com/hybridgroup/mechanoid"
 	"github.com/hybridgroup/mechanoid/engine"
 	"github.com/orsinium-labs/wypes"
 	"github.com/tetratelabs/wazero"
@@ -23,6 +24,8 @@ func (i *Interpreter) Name() string {
 }
 
 func (i *Interpreter) Init() error {
+	mechanoid.DebugMemory("Interpreter Init")
+
 	ctx := context.Background()
 	conf := wazero.NewRuntimeConfigInterpreter()
 	conf = conf.WithDebugInfoEnabled(false)
@@ -32,6 +35,8 @@ func (i *Interpreter) Init() error {
 }
 
 func (i *Interpreter) SetModules(modules wypes.Modules) error {
+	mechanoid.Log("Registering host modules...")
+
 	if i.modules == nil {
 		i.modules = modules
 		return nil
@@ -50,6 +55,8 @@ func (i *Interpreter) SetModules(modules wypes.Modules) error {
 }
 
 func (i *Interpreter) Load(code engine.Reader) error {
+	mechanoid.DebugMemory("Interpreter Load")
+
 	err := i.defineModules()
 	if err != nil {
 		return fmt.Errorf("register wazero host modules: %v", err)
@@ -115,6 +122,8 @@ func wazeroAdaptHostFunc(hf wypes.HostFunc, refs wypes.Refs) api.GoModuleFunctio
 }
 
 func (i *Interpreter) Run() (engine.Instance, error) {
+	mechanoid.DebugMemory("Interpreter Run")
+
 	var err error
 	ctx := context.Background()
 	init := i.module.ExportedFunction("_initialize")
@@ -128,9 +137,11 @@ func (i *Interpreter) Run() (engine.Instance, error) {
 }
 
 func (i *Interpreter) Halt() error {
+	mechanoid.DebugMemory("Interpreter Halt")
+
 	ctx := context.Background()
-	err := i.runtime.Close(ctx)
-	i.runtime = nil
+	err := i.module.Close(ctx)
+	i.module = nil
 	return err
 }