detect_wasi/
lib.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
use std::{
    fs::File,
    io::{Result, Write},
    process::Command,
};
#[cfg(unix)]
use std::{fs::Permissions, os::unix::fs::PermissionsExt};

use tempfile::tempdir;

const WASI_PROGRAM: &[u8] = include_bytes!("miniwasi.wasm");

/// Detect the ability to run WASI
///
/// This attempts to run a small embedded WASI program, and returns true if no errors happened.
/// Errors returned by the `Result` are I/O errors from the establishment of the context, not
/// errors from the run attempt.
///
/// On Linux, you can configure your system to run WASI programs using a binfmt directive. Under
/// systemd, write the below to `/etc/binfmt.d/wasi.conf`, with `/usr/bin/wasmtime` optionally
/// replaced with the path to your WASI runtime of choice:
///
/// ```plain
/// :wasi:M::\x00asm::/usr/bin/wasmtime:
/// ```
pub fn detect_wasi_runability() -> Result<bool> {
    let progdir = tempdir()?;
    let prog = progdir.path().join("miniwasi.wasm");

    {
        let mut progfile = File::create(&prog)?;
        progfile.write_all(WASI_PROGRAM)?;

        #[cfg(unix)]
        progfile.set_permissions(Permissions::from_mode(0o777))?;
    }

    match Command::new(prog).output() {
        Ok(out) => Ok(out.status.success() && out.stdout.is_empty() && out.stderr.is_empty()),
        Err(_) => Ok(false),
    }
}