Node wrapper: support for using alternative Wasm engines

Author: vouillon

tools/ci_setup.ml

@@ -64,6 +64,7 @@ let node_wrapper =
  (name node_wrapper)
        (libraries unix))|} )
   ; "node_wrapper/node_wrapper_per_profile.ml", {|let args = []|}
+  ; "node_wrapper/node_wrapper_per_engine.ml", {|let engine = "node"|}
   ; "node_wrapper/dune-project", "(lang dune 3.17)"
   ; "node_wrapper/node_wrapper.opam", ""
   ]

tools/dune

@@ -1,8 +1,17 @@
 (executable
  (name node_wrapper)
- (modules node_wrapper)
+ (link_deps
+  (env_var WASM_ENGINE))
+ (modules node_wrapper node_wrapper_per_engine)
  (libraries unix))
 
+(rule
+ (target node_wrapper_per_engine.ml)
+ (action
+  (with-stdout-to
+   %{target}
+   (run echo "let engine = \"%{env:WASM_ENGINE=node}\""))))
+
 (executable
  (name ci_setup)
  (modules ci_setup)

tools/node_wrapper.ml

@@ -1,6 +1,20 @@
+let wizard_args = [ "-ext:stack-switching"; "-stack-size=2M"; "--dir=."; "--dir=/tmp" ]
+
+let wasmtime_args =
+  [ (* "-C"; "collector=null"; *) "-W=all-proposals=y"; "--dir=."; "--dir=/tmp" ]
+
+let wasmedge_args =
+  [ "--enable-gc"
+  ; "--enable-exception-handling"
+  ; "--enable-tail-call"
+  ; "--dir=."
+  ; "--dir=/tmp"
+  ]
+
 let extra_args_for_wasoo =
   [ "--experimental-wasm-imported-strings"
   ; "--experimental-wasm-stack-switching"
+  ; "--experimental-wasm-exnref"
   ; "--stack-size=10000"
   ]
 
@@ -23,16 +37,31 @@ let env =
       else e)
     env
 
-let args =
+let environment_args () =
+  List.filter
+    (fun e -> not (String.contains e ','))
+    (Array.to_list (Array.map (fun e -> "--env=" ^ e) env))
+
+let wasm_file file =
+  Filename.concat (Filename.chop_extension file ^ ".assets") "code.wasm"
+
+let common_args file argv = environment_args () @ (wasm_file file :: List.tl argv)
+
+let exe, args =
   match Array.to_list Sys.argv with
   | exe :: argv ->
-      let argv =
+      let exe', argv =
         match argv with
-        | file :: _ when Filename.check_suffix file ".wasm.js" ->
-            extra_args_for_wasoo @ argv
-        | _ -> extra_args_for_jsoo @ argv
+        | file :: _ when Filename.check_suffix file ".wasm.js" -> (
+            match Node_wrapper_per_engine.engine with
+            | "wizard" -> "wizeng.x86-linux", wizard_args @ common_args file argv
+            | "wizard-fast" -> "wizeng.x86-64-linux", wizard_args @ common_args file argv
+            | "wasmtime" -> "wasmtime", wasmtime_args @ common_args file argv
+            | "wasmedge" -> "wasmedge", wasmedge_args @ common_args file argv
+            | _ -> "node", extra_args_for_wasoo @ argv)
+        | _ -> "node", extra_args_for_jsoo @ argv
       in
-      Array.of_list (exe :: argv)
+      exe', Array.of_list (exe :: argv)
   | [] -> assert false
 
 let () =
@@ -45,4 +74,4 @@ let () =
     | _, WEXITED n -> exit n
     | _, WSIGNALED _ -> exit 9
     | _, WSTOPPED _ -> exit 9
-  else Unix.execvpe "node" args env
+  else Unix.execvpe exe args env