mirror of
https://github.com/fluencelabs/wasm-utils
synced 2025-03-16 11:30:51 +00:00
45 lines
1.6 KiB
Rust
45 lines
1.6 KiB
Rust
|
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();
|
||
|
|
}
|