mirror of
https://github.com/pezkuwichain/wasm-instrument.git
synced 2026-04-22 05:38:00 +00:00
Alexander Theißen
a0b548b37d
* Add rustfmt.toml from substrate repo * Apply rustfmt to code base * Fix formatting * Move rustfmt job to the top
43 lines
1.2 KiB
Rust
43 lines
1.2 KiB
Rust
use clap::{App, Arg};
|
|
use pwasm_utils::{self as utils, logger};
|
|
|
|
fn main() {
|
|
logger::init();
|
|
|
|
let target_runtime = utils::TargetRuntime::pwasm();
|
|
|
|
let matches = App::new("wasm-prune")
|
|
.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(&format!(
|
|
"Comma-separated list of exported functions to keep. Default: '{}'",
|
|
target_runtime.symbols().call
|
|
)),
|
|
)
|
|
.get_matches();
|
|
|
|
let exports = matches
|
|
.value_of("exports")
|
|
.unwrap_or(target_runtime.symbols().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
|
|
utils::optimize(&mut module, exports).expect("Optimizer failed");
|
|
|
|
parity_wasm::serialize_to_file(&output, module).expect("Serialization failed");
|
|
}
|