Optionally compile wasmtime-bench-api with wasi-nn and wasi-crypto (#…

…2677)

This adds the ability to add feature flags (e.g. `--features wasi-nn`) when compiling `wasmtime-bench-api` to allow benchmarking Wasmtime with WASI proposals included. Note that due to rust-lang/cargo#5364, these features are only available:
 - in the `crates/bench-api` directory, e.g. `pushd crates/bench-api; cargo build --features wasi-crypto`
 - or from the top-level project directory using `-Zpackage-features`, e.g. `OPENVINO_INSTALL_DIR=/opt/intel/openvino cargo +nightly build -p wasmtime-bench-api -Zpackage-features --features wasi-nn`

crates/bench-api/Cargo.toml

@@ -19,6 +19,8 @@ anyhow = "1.0"
 shuffling-allocator = { version = "1.1.1", optional = true }
 wasmtime = { path = "../wasmtime", default-features = false }
 wasmtime-wasi = { path = "../wasi" }
+wasmtime-wasi-crypto = { path = "../wasi-crypto", optional = true }
+wasmtime-wasi-nn = { path = "../wasi-nn", optional = true }
 wasi-cap-std-sync = { path = "../wasi-common/cap-std-sync" }
 cap-std = "0.13"
 
@@ -27,3 +29,6 @@ wat = "1.0"
 
 [features]
 default = ["shuffling-allocator"]
+wasi-crypto = ["wasmtime-wasi-crypto"]
+wasi-nn = ["wasmtime-wasi-nn"]
+

crates/bench-api/src/lib.rs

@@ -223,9 +223,34 @@ impl BenchState {
             cx = cx.env("WASM_BENCH_USE_SMALL_WORKLOAD", &val)?;
         }
 
-        let cx = cx.build()?;
-        let wasi = Wasi::new(&store, cx);
-        wasi.add_to_linker(&mut linker)?;
+        Wasi::new(linker.store(), cx.build()?).add_to_linker(&mut linker)?;
+
+        #[cfg(feature = "wasi-nn")]
+        {
+            use std::cell::RefCell;
+            use std::rc::Rc;
+            use wasmtime_wasi_nn::{WasiNn, WasiNnCtx};
+
+            let wasi_nn = WasiNn::new(linker.store(), Rc::new(RefCell::new(WasiNnCtx::new()?)));
+            wasi_nn.add_to_linker(&mut linker)?;
+        }
+
+        #[cfg(feature = "wasi-crypto")]
+        {
+            use std::cell::RefCell;
+            use std::rc::Rc;
+            use wasmtime_wasi_crypto::{
+                WasiCryptoAsymmetricCommon, WasiCryptoCommon, WasiCryptoCtx, WasiCryptoSignatures,
+                WasiCryptoSymmetric,
+            };
+
+            let cx_crypto = Rc::new(RefCell::new(WasiCryptoCtx::new()));
+            WasiCryptoCommon::new(linker.store(), cx_crypto.clone()).add_to_linker(linker)?;
+            WasiCryptoAsymmetricCommon::new(linker.store(), cx_crypto.clone())
+                .add_to_linker(linker)?;
+            WasiCryptoSignatures::new(linker.store(), cx_crypto.clone()).add_to_linker(linker)?;
+            WasiCryptoSymmetric::new(linker.store(), cx_crypto).add_to_linker(linker)?;
+        }
 
         Ok(Self {
             engine,