diff --git a/prune/.gitignore b/prune/.gitignore new file mode 100644 index 0000000..f2f9e58 --- /dev/null +++ b/prune/.gitignore @@ -0,0 +1,2 @@ +target +Cargo.lock \ No newline at end of file diff --git a/prune/Cargo.toml b/prune/Cargo.toml new file mode 100644 index 0000000..5270544 --- /dev/null +++ b/prune/Cargo.toml @@ -0,0 +1,9 @@ +[package] +name = "wasm-opt" +version = "0.1.0" +authors = ["NikVolf "] + +[dependencies] +parity-wasm = "0.12" +wasm-utils = { path = "../" } +clap = "2.24" diff --git a/prune/src/main.rs b/prune/src/main.rs new file mode 100644 index 0000000..6175ad9 --- /dev/null +++ b/prune/src/main.rs @@ -0,0 +1,44 @@ +extern crate parity_wasm; +extern crate wasm_utils; +extern crate clap; + +use clap::{App, Arg}; + +fn main() { + wasm_utils::init_log(); + + let matches = App::new("wasm-opt") + .arg(Arg::with_name("input") + .index(1) + .required(true) + .help("Input WASM file")) + .arg(Arg::with_name("output") + .index(2) + .required(true) + .help("Output WASM file")) + .arg(Arg::with_name("exports") + .long("exports") + .short("e") + .takes_value(true) + .value_name("functions") + .help("Comma-separated list of exported functions to keep. Default: _call")) + .get_matches(); + + let exports = matches + .value_of("exports") + .unwrap_or("_call") + .split(',') + .collect(); + + let input = matches.value_of("input").expect("is required; qed"); + let output = matches.value_of("output").expect("is required; qed"); + + let mut module = parity_wasm::deserialize_file(&input).unwrap(); + + // Invoke optimizer + // Contract is supposed to have only these functions as public api + // All other symbols not usable by this list is optimized away + wasm_utils::optimize(&mut module, exports).expect("Optimizer to finish without errors"); + + parity_wasm::serialize_to_file(&output, module).unwrap(); +}