fluencelabs/wasm-utils
1
0
Fork 0
mirror of https://github.com/fluencelabs/wasm-utils synced 2025-06-26 04:51:38 +00:00
Code Issues Projects Releases Wiki Activity

Merge pull request #24 from paritytech/inject-rules

Browse Source 
Inject gas using rules
This commit is contained in:
Nikolay Volf
2017-10-04 16:04:07 +02:00
committed by GitHub
parent d08ff169b6 95beaa1508
commit 6a39db802e
2 changed files with 18 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
gas/src/main.rs
View File

@ -16,7 +16,7 @@ fn main() {
// Loading module
let module = parity_wasm::deserialize_file(&args[1]).expect("Module deserialization to succeed");
let result = wasm_utils::inject_gas_counter(module);
let result = wasm_utils::inject_gas_counter(module, &Default::default());
parity_wasm::serialize_to_file(&args[2], result).expect("Module serialization to succeed")
parity_wasm::serialize_to_file(&args[2], result).expect("Module serialization to succeed")
}

31
src/gas.rs
View File

@ -1,4 +1,5 @@
use parity_wasm::{elements, builder};
use rules;
pub fn update_call_index(opcodes: &mut elements::Opcodes, inserted_index: u32) {
use parity_wasm::elements::Opcode::*;
@ -13,13 +14,13 @@ pub fn update_call_index(opcodes: &mut elements::Opcodes, inserted_index: u32) {
}
enum InjectAction {
Spawn,
Continue,
Spawn(u32),
Continue(u32),
Increment,
IncrementSpawn,
}
pub fn inject_counter(opcodes: &mut elements::Opcodes, gas_func: u32) {
pub fn inject_counter(opcodes: &mut elements::Opcodes, rules: &rules::Set, gas_func: u32) {
use parity_wasm::elements::Opcode::*;
let mut stack: Vec<(usize, usize)> = Vec::new();
@ -37,7 +38,7 @@ pub fn inject_counter(opcodes: &mut elements::Opcodes, gas_func: u32) {
let opcode = &opcodes.elements()[cursor];
match *opcode {
Block(_) | If(_) | Loop(_) => {
InjectAction::Spawn
InjectAction::Spawn(rules.process(opcode))
},
Else => {
InjectAction::IncrementSpawn
@ -46,7 +47,7 @@ pub fn inject_counter(opcodes: &mut elements::Opcodes, gas_func: u32) {
InjectAction::Increment
},
_ => {
InjectAction::Continue
InjectAction::Continue(rules.process(opcode))
}
}
};
@ -65,14 +66,14 @@ pub fn inject_counter(opcodes: &mut elements::Opcodes, gas_func: u32) {
cursor += 3;
stack.push((cursor, 1));
},
InjectAction::Continue => {
InjectAction::Continue(val) => {
cursor += 1;
let (pos, ops) = last_entry;
stack.push((pos, ops+1));
stack.push((pos, ops + val as usize));
},
InjectAction::Spawn => {
InjectAction::Spawn(val) => {
let (pos, ops) = last_entry;
stack.push((pos, ops+1));
stack.push((pos, ops + val as usize));
cursor += 1;
stack.push((cursor, 1));
@ -81,7 +82,7 @@ pub fn inject_counter(opcodes: &mut elements::Opcodes, gas_func: u32) {
}
}
pub fn inject_gas_counter(module: elements::Module) -> elements::Module {
pub fn inject_gas_counter(module: elements::Module, rules: &rules::Set) -> elements::Module {
// Injecting gas counting external
let mut mbuilder = builder::from_module(module);
let import_sig = mbuilder.push_signature(
@ -117,7 +118,7 @@ pub fn inject_gas_counter(module: elements::Module) -> elements::Module {
&mut elements::Section::Code(ref mut code_section) => {
for ref mut func_body in code_section.bodies_mut() {
update_call_index(func_body.code_mut(), gas_func);
inject_counter(func_body.code_mut(), gas_func);
inject_counter(func_body.code_mut(), rules, gas_func);
}
},
&mut elements::Section::Export(ref mut export_section) => {
@ -172,7 +173,7 @@ mod tests {
.build()
.build();
let injected_module = inject_gas_counter(module);
let injected_module = inject_gas_counter(module, &Default::default());
assert_eq!(
&vec![
@ -214,7 +215,7 @@ mod tests {
.build()
.build();
let injected_module = inject_gas_counter(module);
let injected_module = inject_gas_counter(module, &Default::default());
assert_eq!(
&vec![
@ -267,7 +268,7 @@ mod tests {
.build()
.build();
let injected_module = inject_gas_counter(module);
let injected_module = inject_gas_counter(module, &Default::default());
assert_eq!(
&vec![
@ -329,7 +330,7 @@ mod tests {
.build()
.build();
let injected_module = inject_gas_counter(module);
let injected_module = inject_gas_counter(module, &Default::default());
assert_eq!(
&vec![