mirror of
https://github.com/pezkuwichain/wasm-instrument.git
synced 2026-04-22 02:07:58 +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
22 lines
596 B
Rust
22 lines
596 B
Rust
use pwasm_utils::{self as utils, logger};
|
|
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
|
|
}
|
|
|
|
// Loading module
|
|
let module =
|
|
parity_wasm::deserialize_file(&args[1]).expect("Module deserialization to succeed");
|
|
|
|
let result = utils::inject_gas_counter(module, &utils::rules::Set::default(), "env")
|
|
.expect("Failed to inject gas. Some forbidden opcodes?");
|
|
|
|
parity_wasm::serialize_to_file(&args[2], result).expect("Module serialization to succeed")
|
|
}
|