use pwasm_utils::{logger, stack_height};
use std::env;
fn main() {
logger::init();
let args = env::args().collect::<Vec<_>>();
if args.len() != 3 {
println!("Usage: {} input_file.wasm output_file.wasm", args[0]);
return
}
let input_file = &args[1];
let output_file = &args[2];
let module =
parity_wasm::deserialize_file(&input_file).expect("Module deserialization to succeed");
let result =
stack_height::inject_limiter(module, 1024).expect("Failed to inject stack height counter");
parity_wasm::serialize_to_file(&output_file, result).expect("Module serialization to succeed")
}