Merge branch 'master' of github.com:NikVolf/wasm-tools

This commit is contained in:
NikVolf 2017-05-25 22:16:23 +03:00
commit e06b5b9972
2 changed files with 35 additions and 14 deletions

View File

@ -6,3 +6,4 @@ authors = ["NikVolf <nikvolf@gmail.com>"]
[dependencies] [dependencies]
parity-wasm = { git="https://github.com/nikvolf/parity-wasm" } parity-wasm = { git="https://github.com/nikvolf/parity-wasm" }
wasm-utils = { path = "../" } wasm-utils = { path = "../" }
clap = "2.24"

View File

@ -1,24 +1,44 @@
extern crate parity_wasm; extern crate parity_wasm;
extern crate wasm_utils; extern crate wasm_utils;
extern crate clap;
use std::env; use clap::{App, Arg};
fn main() { fn main() {
wasm_utils::init_log(); wasm_utils::init_log();
let args = env::args().collect::<Vec<_>>(); let matches = App::new("wasm-opt")
if args.len() < 3 { .arg(Arg::with_name("input")
println!("Usage: {} input_file.wasm output_file.wasm", args[0]); .index(1)
return; .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 mut module = parity_wasm::deserialize_file(&args[1]).unwrap(); 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 // Invoke optimizer
// Contract is supposed to have only these functions as public api // Contract is supposed to have only these functions as public api
// All other symbols not usable by this list is optimized away // All other symbols not usable by this list is optimized away
wasm_utils::optimize(&mut module, vec!["_call"]).expect("Optimizer to finish without errors"); wasm_utils::optimize(&mut module, exports).expect("Optimizer to finish without errors");
parity_wasm::serialize_to_file(&args[2], module).unwrap(); parity_wasm::serialize_to_file(&output, module).unwrap();
} }