Merge pull request #1320 from apepkuss/feat-wasmedge-executor

[libcontainer] Integrate WasmEdge Runtime

.github/workflows/main.yml

@@ -67,7 +67,9 @@ jobs:
       - run: sudo apt-get -y update
       - run: sudo apt-get install -y pkg-config libsystemd-dev libdbus-glib-1-dev libelf-dev libseccomp-dev
       - name: Run tests
-        run: cd ./crates && cargo test --all --all-features --no-fail-fast
+        run: |
+          export LD_LIBRARY_PATH=$HOME/.wasmedge/lib
+          cd ./crates && cargo test --all --all-features --no-fail-fast
   coverage:
     runs-on: ubuntu-latest
     timeout-minutes: 15

crates/libcontainer/Cargo.toml

@@ -18,6 +18,7 @@ systemd = ["libcgroups/systemd", "v2"]
 v2 = ["libcgroups/v2"]
 v1 = ["libcgroups/v1"]
 cgroupsv2_devices = ["libcgroups/cgroupsv2_devices"]
+wasm-wasmedge = ["wasmedge-sdk/standalone"]
 
 [dependencies]
 anyhow = "1.0"
@@ -43,6 +44,7 @@ syscalls = "0.6.7"
 rust-criu = "0.2.0"
 wasmer = { version = "2.2.0", optional = true }
 wasmer-wasi = { version = "2.3.0", optional = true }
+wasmedge-sdk = { version = "0.7.1", optional = true }
 
 [dev-dependencies]
 oci-spec = { version = "0.5.8", features = ["proptests", "runtime"] }

crates/libcontainer/src/workload/mod.rs

@@ -2,10 +2,14 @@ use anyhow::{Context, Result};
 use oci_spec::runtime::Spec;
 
 use self::default::DefaultExecutor;
+#[cfg(feature = "wasm-wasmedge")]
+use self::wasmedge::WasmEdgeExecutor;
 #[cfg(feature = "wasm-wasmer")]
 use self::wasmer::WasmerExecutor;
 
 pub mod default;
+#[cfg(feature = "wasm-wasmedge")]
+pub mod wasmedge;
 #[cfg(feature = "wasm-wasmer")]
 pub mod wasmer;
 
@@ -28,6 +32,11 @@ impl ExecutorManager {
             return WasmerExecutor::exec(spec).context("wasmer execution failed");
         }
 
+        #[cfg(feature = "wasm-wasmedge")]
+        if WasmEdgeExecutor::can_handle(spec)? {
+            return WasmEdgeExecutor::exec(spec).context("wasmedge execution failed");
+        }
+
         DefaultExecutor::exec(spec).context("default execution failed")
     }
 }

crates/libcontainer/src/workload/wasmedge.rs

@@ -0,0 +1,90 @@
+use anyhow::Result;
+use oci_spec::runtime::Spec;
+use wasmedge_sdk::{
+    config::{CommonConfigOptions, ConfigBuilder, HostRegistrationConfigOptions},
+    params, Vm,
+};
+
+use super::Executor;
+
+const EXECUTOR_NAME: &str = "wasmedge";
+
+pub struct WasmEdgeExecutor {}
+impl Executor for WasmEdgeExecutor {
+    fn exec(spec: &Spec) -> Result<()> {
+        // parse wasi parameters
+        let args = get_args(spec);
+        let mut cmd = args[0].clone();
+        if let Some(stripped) = args[0].strip_prefix(std::path::MAIN_SEPARATOR) {
+            cmd = stripped.to_string();
+        }
+        let envs = env_to_wasi(spec);
+
+        // create configuration with `wasi` option enabled
+        let config = ConfigBuilder::new(CommonConfigOptions::default())
+            .with_host_registration_config(HostRegistrationConfigOptions::default().wasi(true))
+            .build()?;
+
+        // create a vm with the config settings
+        let mut vm = Vm::new(Some(config))?;
+
+        // initialize the wasi module with the parsed parameters
+        let mut wasi_instance = vm.wasi_module()?;
+        wasi_instance.initialize(
+            Some(args.iter().map(|s| s as &str).collect()),
+            Some(envs.iter().map(|s| s as &str).collect()),
+            None,
+        );
+
+        let mut vm = vm.register_module_from_file("main", &cmd)?;
+
+        let ins = vm.named_module("main")?;
+        ins.func("_start")
+            .expect("Not found '_start' func in the 'main' module instance")
+            .call(&mut vm, params!())?;
+
+        Ok(())
+    }
+
+    fn can_handle(spec: &Spec) -> Result<bool> {
+        if let Some(annotations) = spec.annotations() {
+            if let Some(handler) = annotations.get("run.oci.handler") {
+                return Ok(handler == "wasm");
+            }
+
+            if let Some(variant) = annotations.get("module.wasm.image/variant") {
+                return Ok(variant == "compat");
+            }
+        }
+
+        Ok(false)
+    }
+
+    fn name() -> &'static str {
+        EXECUTOR_NAME
+    }
+}
+
+fn get_args(spec: &Spec) -> &[String] {
+    let p = match spec.process() {
+        None => return &[],
+        Some(p) => p,
+    };
+
+    match p.args() {
+        None => &[],
+        Some(args) => args.as_slice(),
+    }
+}
+
+fn env_to_wasi(spec: &Spec) -> Vec<String> {
+    let default = vec![];
+    let env = spec
+        .process()
+        .as_ref()
+        .unwrap()
+        .env()
+        .as_ref()
+        .unwrap_or(&default);
+    env.to_vec()
+}