fluencelabs/wasm-utils
1
0
Fork 0
mirror of https://github.com/fluencelabs/wasm-utils synced 2025-05-29 23:51:21 +00:00
Code Issues Projects Releases Wiki Activity

working gas counter

Browse Source
This commit is contained in:
NikVolf 2017-04-12 19:41:36 +03:00
parent 9419842d51
commit fe575b4eaa
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
gas/src/main.rs
View File

@ -18,6 +18,22 @@ pub fn update_call_index(opcodes: &mut elements::Opcodes, inserted_index: u32) {
} }
} }
pub fn inject_counter(opcodes: &mut elements::Opcodes, gas_func: u32) {
use parity_wasm::elements::Opcode::*;
for opcode in opcodes.elements_mut().iter_mut() {
match opcode {
&mut Block(_, ref mut block) | &mut If(_, ref mut block) => {
inject_counter(block, gas_func)
},
_ => { }
}
}
let ops = opcodes.elements_mut().len() as u32;
opcodes.elements_mut().insert(0, I32Const(ops));
opcodes.elements_mut().insert(1, Call(gas_func));
}
fn main() { fn main() {
let args = env::args().collect::<Vec<_>>(); let args = env::args().collect::<Vec<_>>();
@ -66,6 +82,7 @@ fn main() {
&mut elements::Section::Code(ref mut code_section) => { &mut elements::Section::Code(ref mut code_section) => {
for ref mut func_body in code_section.bodies_mut() { for ref mut func_body in code_section.bodies_mut() {
update_call_index(func_body.code_mut(), gas_func); update_call_index(func_body.code_mut(), gas_func);
inject_counter(func_body.code_mut(), gas_func);
} }
}, },
_ => { } _ => { }