support executeTestFunction

Author: XMadrid

.gitignore

@@ -1,4 +1,5 @@
 .vscode
+.cache
 
 /node_modules
 /dist

instrumentation/MockInstrumentationWalker.cpp

@@ -1,6 +1,8 @@
 #include "MockInstrumentationWalker.hpp"
+#include <binaryen-c.h>
 #include <string_view>
 #include <support/index.h>
+#include <wasm-type.h>
 #include <wasm.h>
 // mock test will be tested with wasm-testing-framework project, escape this class
 // LCOV_EXCL_START
@@ -86,13 +88,28 @@ bool MockInstrumentationWalker::mockFunctionDuplicateImportedCheck() const noexc
   return checkRepeat;
 }
 
+void MockInstrumentationWalker::addExecuteTestFunction() noexcept {
+  std::vector<BinaryenExpressionRef> operands{};
+  BinaryenExpressionRef body = moduleBuilder.makeCallIndirect(
+    module->tables[0]->name, 
+    BinaryenLocalGet(module, 0, wasm::Type::i32), 
+    operands,
+    wasm::Signature(wasm::Type::none, wasm::Type::none)
+  );
+
+  body->finalize();
+  BinaryenAddFunction(module, "executeTestFunction", BinaryenTypeInt32(), BinaryenTypeNone(), {}, 0, body);
+  BinaryenAddFunctionExport(module, "executeTestFunction", "executeTestFunction");
+}
+
 uint32_t MockInstrumentationWalker::mockWalk() noexcept {
   if (mockFunctionDuplicateImportedCheck()) {
     return 1U; // failed
   } else {
     wasm::ModuleUtils::iterDefinedFunctions(*module, [this](wasm::Function *const func) noexcept {
       walkFunctionInModule(func, this->module);
     });
+    addExecuteTestFunction();
     return 0U;
   }
 }

instrumentation/MockInstrumentationWalker.hpp

@@ -130,6 +130,11 @@ class MockInstrumentationWalker final : public wasm::PostWalker<MockInstrumentat
   /// @brief bool, true when the mock function is duplicated imported
   bool mockFunctionDuplicateImportedCheck() const noexcept;
 
+  ///
+  /// @brief Add export function executeTestFunction()
+  ///
+  void addExecuteTestFunction() noexcept;
+
   ///
   /// @brief Main API for mock instrumentation
   ///

src/core/execute.ts

@@ -45,6 +45,14 @@ async function nodeExecutor(wasm: string, outFolder: string, imports?: Imports):
   importsArg.exports = ins.exports;
   try {
     wasi.start(ins);
+    const execTestFunction = ins.exports["executeTestFunction"];
+    if (typeof execTestFunction === "function") {
+      for (const fncs of executionRecorder.registerFunctions) {
+        const functions = fncs[1];
+        (execTestFunction as (a: number) => void)(functions);
+        mockInstrumentFunc["mockFunctionStatus.clear"]();
+      }
+    }
   } catch (error) {
     if (error instanceof Error) {
       console.error(error.stack);