Files
wasmtime/crates/fuzzing/wasm-spec-interpreter/build.rs
Andrew Brown 42acb72c54 fuzz: retrieve the WebAssembly spec repository in build.rs
To avoid the large download size of the spec repository mentioned
[here](https://github.com/bytecodealliance/wasmtime/pull/3124#discussion_r684605984),
this change removes it as a submodule and instead clones it shallowly
when the directory is empty (or not present) when `build.rs` is run.
2021-08-10 11:56:07 -07:00

86 lines
3.1 KiB
Rust

/// Build the OCaml code and statically link it into the Rust library; see the
/// [ocaml-interop
/// example](https://github.com/tezedge/ocaml-interop/blob/master/testing/rust-caller/build.rs)
/// for more details. After playing with this a bit, I discovered that the best
/// approach to avoid missing symbols was to imitate `dune`: I observed `rm -rf
/// _build && dune build ./ocaml/interpret.exe.o --display=verbose` and used
/// that as a pattern, now encoded in `ocaml/Makefile` for easier debugging.
use std::{env, path::PathBuf, process::Command};
const LIB_NAME: &'static str = "interpret";
const OCAML_DIR: &'static str = "ocaml";
const SPEC_DIR: &'static str = "ocaml/spec";
const SPEC_REPOSITORY: &'static str = "https://github.com/bytecodealliance/wasm-spec-mirror";
fn main() {
if cfg!(feature = "build-libinterpret") {
build();
}
}
fn build() {
let out_dir = &env::var("OUT_DIR").unwrap();
// Re-run if changed.
println!("cargo:rerun-if-changed={}/{}.ml", OCAML_DIR, LIB_NAME);
println!("cargo:rerun-if-changed={}/Makefile", OCAML_DIR);
if let Some(other_dir) = env::var_os("FFI_LIB_DIR") {
// Link with a library provided in the `FFI_LIB_DIR`.
println!("cargo:rustc-link-search={}", other_dir.to_str().unwrap());
println!("cargo:rustc-link-lib=static={}", LIB_NAME);
} else {
// Ensure the spec repository is present.
if is_spec_repository_empty(SPEC_DIR) {
retrieve_spec_repository(SPEC_REPOSITORY, SPEC_DIR)
}
// Build the library to link to.
build_lib(out_dir, OCAML_DIR);
println!("cargo:rustc-link-search={}", out_dir);
println!("cargo:rustc-link-lib=static={}", LIB_NAME);
}
// Enabling this feature alerts the compiler to use the `with_library`
// module.
println!("cargo:rustc-cfg=feature=\"has-libinterpret\"");
}
// Build the OCaml library into Cargo's `out` directory.
fn build_lib(out_dir: &str, ocaml_dir: &str) {
let status = Command::new("make")
.arg(format!("BUILD_DIR={}", out_dir))
.current_dir(ocaml_dir)
.status()
.expect("Failed to execute 'make' command to build OCaml library");
assert!(
status.success(),
"Failed to build the OCaml library using 'make'."
)
}
// Check if the spec repository directory contains any files.
fn is_spec_repository_empty(destination: &str) -> bool {
PathBuf::from(destination)
.read_dir()
.map(|mut i| i.next().is_none())
.unwrap_or(true)
}
// Clone the spec repository into `destination`. This exists due to the large
// size of the dependencies (e.g. KaTeX) that are pulled if this were cloned
// recursively as a submodule.
fn retrieve_spec_repository(repository: &str, destination: &str) {
let status = Command::new("git")
.arg("clone")
.arg("--depth")
.arg("1")
.arg(repository)
.arg(destination)
.status()
.expect("Failed to execute 'git' command to clone spec repository.");
assert!(status.success(), "Failed to retrieve the spec repository.")
}