Hook up wasi exit code to wasmer cli

examples/exit.wat

@@ -0,0 +1,11 @@
+(module
+  (import "wasi_unstable" "proc_exit"      (func $proc_exit (param i32)))
+  (export "_start" (func $_start))
+
+  (memory 10)
+  (export "memory" (memory 0))
+
+  (func $_start
+    (call $proc_exit (i32.const 7))
+  )
+)

src/bin/wasmer.rs

@@ -11,12 +11,14 @@ use std::str::FromStr;
 use hashbrown::HashMap;
 use structopt::StructOpt;
 
-use wasmer::webassembly::InstanceABI;
 use wasmer::*;
 use wasmer_clif_backend::CraneliftCompiler;
 #[cfg(feature = "backend:llvm")]
 use wasmer_llvm_backend::LLVMCompiler;
-use wasmer_runtime::cache::{Cache as BaseCache, FileSystemCache, WasmHash, WASMER_VERSION_HASH};
+use wasmer_runtime::{
+    cache::{Cache as BaseCache, FileSystemCache, WasmHash, WASMER_VERSION_HASH},
+    Func, Value, error::RuntimeError,
+};
 use wasmer_runtime_core::{
     self,
     backend::{Compiler, CompilerConfig},
@@ -295,18 +297,27 @@ fn execute_wasm(options: &Run) -> Result<(), String> {
     };
 
     // TODO: refactor this
-    let (abi, import_object, _em_globals) = if wasmer_emscripten::is_emscripten_module(&module) {
+    if wasmer_emscripten::is_emscripten_module(&module) {
         let mut emscripten_globals = wasmer_emscripten::EmscriptenGlobals::new(&module);
-        (
+        let import_object = wasmer_emscripten::generate_emscripten_env(&mut emscripten_globals);
-            InstanceABI::Emscripten,
+        let mut instance = module
-            wasmer_emscripten::generate_emscripten_env(&mut emscripten_globals),
+            .instantiate(&import_object)
-            Some(emscripten_globals), // TODO Em Globals is here to extend, lifetime, find better solution
+            .map_err(|e| format!("Can't instantiate module: {:?}", e))?;
+
+        wasmer_emscripten::run_emscripten_instance(
+            &module,
+            &mut instance,
+            if let Some(cn) = &options.command_name {
+                cn
+            } else {
+                options.path.to_str().unwrap()
+            },
+            options.args.iter().map(|arg| arg.as_str()).collect(),
         )
+        .map_err(|e| format!("{:?}", e))?;
     } else {
         if cfg!(feature = "wasi") && wasmer_wasi::is_wasi_module(&module) {
-            (
+            let import_object = wasmer_wasi::generate_import_object(
-                InstanceABI::WASI,
-                wasmer_wasi::generate_import_object(
                 if let Some(cn) = &options.command_name {
                     [cn.clone()]
                 } else {
@@ -321,34 +332,46 @@ fn execute_wasm(options: &Run) -> Result<(), String> {
                     .map(|(k, v)| format!("{}={}", k, v).into_bytes())
                     .collect(),
                 options.pre_opened_directories.clone(),
-                ),
+            );
-                None,
-            )
-        } else {
-            (
-                InstanceABI::None,
-                wasmer_runtime_core::import::ImportObject::new(),
-                None,
-            )
-        }
-    };
 
-    let mut instance = module
+            let instance = module
                 .instantiate(&import_object)
                 .map_err(|e| format!("Can't instantiate module: {:?}", e))?;
 
-    webassembly::run_instance(
+            let start: Func<(), ()> = instance.func("_start").map_err(|e| format!("{:?}", e))?;
-        &module,
+
-        &mut instance,
+            let result = start.call();
-        abi,
+
-        if let Some(cn) = &options.command_name {
+            if let Err(ref err) = result {
-            cn
+                match err {
+                    RuntimeError::Trap { msg } => panic!("wasm trap occured: {}", msg),
+                    RuntimeError::Error { data } => {
+                        if let Some(error_code) = data.downcast_ref::<wasmer_wasi::ExitCode>() {
+                            std::process::exit(error_code.code as i32)
+                        }
+                    }
+                }
+                panic!("error: {:?}", err)
+            }
         } else {
-            options.path.to_str().unwrap()
+            let import_object = wasmer_runtime_core::import::ImportObject::new();
-        },
+            let instance = module
-        options.args.iter().map(|arg| arg.as_str()).collect(),
+                .instantiate(&import_object)
-    )
+                .map_err(|e| format!("Can't instantiate module: {:?}", e))?;
+
+            let args: Vec<Value> = options
+                .args
+                .iter()
+                .map(|arg| arg.as_str())
+                .map(|x| Value::I32(x.parse().unwrap()))
+                .collect();
+            instance
+                .dyn_func("main")
+                .map_err(|e| format!("{:?}", e))?
+                .call(&args)
                 .map_err(|e| format!("{:?}", e))?;
+        }
+    }
 
     Ok(())
 }