fix store bug, added testcases for the fix

.vscode/launch.json

@@ -27,7 +27,7 @@
             "args": [
                 "node",
                 "-t",
-                "exported-called-params-grow-mem",
+                "mem-exp-host-mod-pingpong",
             ]
         },
         {

src/replay-generator.cts

@@ -110,14 +110,14 @@ export default class Generator {
                         this.state.lastFuncReturn = true
                         r.slice(-1)[0].reps += 1
                         this.state.importCallStack.pop()
-                        this.state.importCallStackFunction.pop()
+                        this.state.lastFunc = this.state.importCallStackFunction.pop()
                         break
                     }
                 }
                 this.state.lastFuncReturn = true
                 r.push({ results: event.results, reps: 1 })
                 this.state.importCallStack.pop()
-                this.state.importCallStackFunction.pop()
+                this.state.lastFunc = this.state.importCallStackFunction.pop()
                 break
             case "Load":
                 this.pushEvent({

tests/node/mem-exp-host-mod-store-lastfunc-return/index.wat

@@ -0,0 +1,33 @@
+(module
+  (import "env" "changeMemAfterReentry" (func $changeMemAfterReentry (result i32)))
+  (import "env" "foo" (func $foo))
+  (import "env" "bar" (func $bar (result i32)))
+  (func $main (export "main")
+    call $changeMemAfterReentry
+    drop
+    i32.const 1
+    i32.load
+    drop
+    call $changeMemAfterReentry
+    drop
+    i32.const 1
+    i32.load
+    drop
+    call $changeMemAfterReentry
+    drop
+    i32.const 1
+    i32.load
+    drop
+  )
+  (func (export "reentry")
+    call $foo
+    i32.const 1
+    i32.load
+    drop
+  )
+  (func (export "reentry2")
+    call $bar
+    drop
+  )
+  (memory (export "memory") 1)
+)

tests/node/mem-exp-host-mod-store-lastfunc-return/test.js

@@ -0,0 +1,22 @@
+export default async function test(wasmBinary) {
+    let instance
+    let imports = {
+        env: {
+            changeMemAfterReentry: () => {
+                instance.exports.reentry()
+                new Uint8Array(instance.exports.memory.buffer)[1] = 1
+                return 69
+            },
+            foo: () => {
+                instance.exports.reentry2()
+            },
+            bar: () => {
+                new Uint8Array(instance.exports.memory.buffer)[1] = 2
+                return 420
+            }
+        }
+    }
+    let wasm = await WebAssembly.instantiate(wasmBinary, imports)
+    instance = wasm.instance
+    instance.exports.main()
+}

tests/node/mem-exp-host-mod-store-lastfunc/index.wat

@@ -0,0 +1,21 @@
+(module
+  (import "env" "changeMemAfterReentry" (func $changeMemAfterReentry))
+  (import "env" "foo" (func $foo))
+  (import "env" "bar" (func $bar))
+  (func $main (export "main")
+    call $changeMemAfterReentry
+    i32.const 1
+    i32.load
+    drop
+  )
+  (func (export "reentry")
+    call $foo
+    i32.const 1
+    i32.load
+    drop
+  )
+  (func (export "reentry2")
+    call $bar
+  )
+  (memory (export "memory") 1)
+)

tests/node/mem-exp-host-mod-store-lastfunc/test.js

@@ -0,0 +1,20 @@
+export default async function test(wasmBinary) {
+    let instance
+    let imports = {
+        env: {
+            changeMemAfterReentry: () => {
+                instance.exports.reentry()
+                new Uint8Array(instance.exports.memory.buffer)[1] = 1
+            },
+            foo: () => {
+                instance.exports.reentry2()
+            },
+            bar: () => {
+                new Uint8Array(instance.exports.memory.buffer)[1] = 2
+            }
+        }
+    }
+    let wasm = await WebAssembly.instantiate(wasmBinary, imports)
+    instance = wasm.instance
+    instance.exports.main()
+}