Add wasi p2 example using rustc native p2 target

Signed-off-by: Ludvig Liljenberg <4257730+ludfjig@users.noreply.github.com>

Author: ludfjig

.gitignore

@@ -478,5 +478,8 @@ target/
 
 # MSVC Windows builds of rustc generate these, which store debugging information
 *.pdb
-src/component_sample/**/*.wasm
+
+# WebAssembly build artifacts
+*.wasm
+
 src/wasmsamples/components/bindings/

Cargo.lock

@@ -5,6 +5,7 @@ mkdir-arg := if os() == "windows" { "-Force" } else { "-p" }
 latest-release:= if os() == "windows" {"$(git tag -l --sort=v:refname | select -last 2 | select -first 1)"} else {`git tag -l --sort=v:refname | tail -n 2 | head -n 1`}
 wit-world := if os() == "windows" { "$env:WIT_WORLD=\"" + justfile_directory() + "\\src\\component_sample\\wit\\component-world.wasm" + "\";" } else { "WIT_WORLD=" + justfile_directory() + "/src/component_sample/wit/component-world.wasm" }
 wit-world-c := if os() == "windows" { "$env:WIT_WORLD=\"" + justfile_directory() + "\\src\\wasmsamples\\components\\runcomponent-world.wasm" + "\";" } else { "WIT_WORLD=" + justfile_directory() + "/src/wasmsamples/components/runcomponent-world.wasm" }
+wit-world-monte-carlo := if os() == "windows" { "$env:WIT_WORLD=\"" + justfile_directory() + "\\src\\wasip2_guest\\monte-carlo-world.wasm" + "\";" } else { "WIT_WORLD=" + justfile_directory() + "/src/wasip2_guest/monte-carlo-world.wasm" }
 
 set windows-shell := ["pwsh.exe", "-NoLogo", "-Command"]
 
@@ -32,6 +33,7 @@ mkdir-redist target=default-target:
 compile-wit:
     wasm-tools component wit ./src/wasmsamples/components/runcomponent.wit -w -o ./src/wasmsamples/components/runcomponent-world.wasm
     wasm-tools component wit ./src/component_sample/wit/example.wit -w -o ./src/component_sample/wit/component-world.wasm
+    wasm-tools component wit ./src/wasip2_guest/wit/monte-carlo.wit -w -o ./src/wasip2_guest/monte-carlo-world.wasm
 
 build-examples target=default-target features="": (build-wasm-examples target features) (build-rust-wasm-examples target features) (build-rust-component-examples target features)
 
@@ -50,6 +52,13 @@ build-rust-component-examples target=default-target features="": (compile-wit)
     cd ./src/component_sample && cargo component build --target wasm32-unknown-unknown --profile={{ if target == "debug" {"dev"} else { target } }}
     cargo run {{ if features =="" {''} else if features=="no-default-features" {"--no-default-features" } else {"--features " + features } }} -p hyperlight-wasm-aot compile {{ if features =~ "gdb" {"--debug"} else {""} }} --component ./src/component_sample/target/wasm32-unknown-unknown/{{ target }}/component_sample.wasm ./x64/{{ target }}/component_sample.aot
 
+build-monte-carlo-example target=default-target features="": (compile-wit) (mkdir-redist target)
+    # Monte Carlo Pi example using native wasm32-wasip2 + wit-bindgen
+    # Always build in release mode to avoid WASI dependencies (debug mode pulls in entire WASI for some reason)
+    rustup target add wasm32-wasip2
+    cd ./src/wasip2_guest && cargo build --lib --target wasm32-wasip2 --release
+    cargo run {{ if features =="" {''} else if features=="no-default-features" {"--no-default-features" } else {"--features " + features } }} -p hyperlight-wasm-aot compile {{ if features =~ "gdb" {"--debug"} else {""} }} --component ./src/wasip2_guest/target/wasm32-wasip2/release/monte_carlo.wasm ./x64/{{ target }}/monte_carlo.aot
+
 check target=default-target:
     cargo check --profile={{ if target == "debug" {"dev"} else { target } }}
     cd src/rust_wasm_samples  && cargo check --profile={{ if target == "debug" {"dev"} else { target } }}
@@ -99,9 +108,10 @@ examples-ci target=default-target features="": (build-rust-wasm-examples target)
     cargo run {{ if features =="" {''} else {"--no-default-features -F function_call_metrics," + features } }} --profile={{ if target == "debug" {"dev"} else { target } }} --example metrics
     cargo run {{ if features =="" {"--no-default-features --features kvm,mshv3"} else {"--no-default-features -F function_call_metrics," + features } }} --profile={{ if target == "debug" {"dev"} else { target } }} --example metrics 
 
-examples-components target=default-target features="": (build-rust-component-examples target) 
+examples-components target=default-target features="": (build-rust-component-examples target) (build-monte-carlo-example target)
     {{ wit-world }} cargo run {{ if features =="" {''} else {"--no-default-features -F kvm -F " + features } }} --profile={{ if target == "debug" {"dev"} else { target } }} --example component_example
     {{ wit-world-c }} cargo run {{ if features =="" {''} else {"--no-default-features -F kvm -F " + features } }} --profile={{ if target == "debug" {"dev"} else { target } }} --example c-component
+    {{ wit-world-monte-carlo }} cargo run {{ if features =="" {''} else {"--no-default-features -F kvm -F " + features } }} --profile={{ if target == "debug" {"dev"} else { target } }} --example monte_carlo_example
 
 # warning, compares to and then OVERWRITES the given baseline
 bench-ci baseline target="release" features="":

Justfile

@@ -59,6 +59,11 @@ name = "interruption"
 path = "examples/interruption/main.rs"
 test = true
 
+[[example]]
+name = "monte_carlo_example"
+path = "examples/monte_carlo_example/main.rs"
+test = true
+
 [dependencies]
 hyperlight-host = { workspace = true }
 libc = { version = "0.2.180" }
@@ -92,6 +97,7 @@ tracing = "0.1.44"
 tracing-subscriber = {version = "0.3.22", features = ["std", "env-filter"]}
 tracing-opentelemetry = "0.32.1"
 uuid = { version = "1.19.0", features = ["v4"] }
+rand = "0.8"
 
 [build-dependencies]
 cfg_aliases = "0.2.1"

src/hyperlight_wasm/Cargo.toml

@@ -0,0 +1,97 @@
+/*
+Copyright 2024 The Hyperlight Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+//! Monte Carlo Pi Estimator - Host Example
+//!
+//! Demonstrates running a WASI P2 component built with native `wasm32-wasip2`
+//! target and `wit-bindgen`. The host provides a random number generator
+//! that the guest component imports to estimate Pi.
+//!
+//! Build the guest first: `just build-monte-carlo-example`
+//! Run: `just examples-components`
+
+extern crate alloc;
+
+use examples_common::get_wasm_module_path;
+use rand::rngs::StdRng;
+use rand::{Rng, SeedableRng};
+
+mod bindings {
+    hyperlight_component_macro::host_bindgen!("../wasip2_guest/monte-carlo-world.wasm");
+}
+
+/// Host-provided RNG state for the guest component.
+///
+/// The guest cannot generate randomness in `no_std` mode, so the host
+/// provides random numbers via the imported `random` interface.
+struct State {
+    rng: StdRng,
+}
+
+impl State {
+    fn new() -> Self {
+        State {
+            rng: StdRng::from_entropy(),
+        }
+    }
+}
+
+impl bindings::my::monte_carlo::Random for State {
+    fn random(&mut self) -> f32 {
+        self.rng.r#gen::<f32>()
+    }
+}
+
+#[allow(refining_impl_trait)]
+impl bindings::my::monte_carlo::EstimatorImports for State {
+    type Random = State;
+
+    fn random(&mut self) -> &mut Self::Random {
+        self
+    }
+}
+
+fn main() {
+    let state = State::new();
+
+    let mut sb: hyperlight_wasm::ProtoWasmSandbox = hyperlight_wasm::SandboxBuilder::new()
+        .with_guest_input_buffer_size(1000000)
+        .with_guest_heap_size(10000000)
+        .with_guest_stack_size(1000000)
+        .build()
+        .unwrap();
+
+    let rt = bindings::register_host_functions(&mut sb, state);
+
+    let sb = sb.load_runtime().unwrap();
+
+    let mod_path = get_wasm_module_path("monte_carlo.aot").unwrap();
+    let sb = sb.load_module(mod_path).unwrap();
+
+    let mut wrapped = bindings::EstimatorSandbox { sb, rt };
+
+    // Estimate pi with different sample sizes
+    for samples in [100, 1000, 10000] {
+        let pi_estimate =
+            bindings::my::monte_carlo::EstimatorExports::estimate_pi(&mut wrapped, samples);
+        println!(
+            "Pi estimate with {} samples: {:.6} (error: {:.6})",
+            samples,
+            pi_estimate,
+            (pi_estimate - std::f32::consts::PI).abs()
+        );
+    }
+}