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

optimize virtual calls

Browse Source
This commit is contained in:
NikVolf 2017-12-28 15:56:58 +03:00
parent f260bb8bb5
commit d45f0d51d5
2 changed files with 46 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/optimizer.rs
View File

@ -1,7 +1,7 @@
use std::collections::HashSet;
use parity_wasm::elements;
use symbols::{Symbol, expand_symbols, push_code_symbols, resolve_function};
use symbols::{Symbol, expand_symbols, push_code_symbols};
#[derive(Debug)]
pub enum Error {
@ -33,14 +33,14 @@ pub fn optimize(
push_code_symbols(&module, segment.offset().code(), &mut init_symbols);
}
}
if let Some(elements_section) = module.elements_section() {
for segment in elements_section.entries() {
push_code_symbols(&module, segment.offset().code(), &mut init_symbols);
for func_index in segment.members() {
stay.insert(resolve_function(&module, *func_index));
}
}
}
// if let Some(elements_section) = module.elements_section() {
// for segment in elements_section.entries() {
// push_code_symbols(&module, segment.offset().code(), &mut init_symbols);
// for func_index in segment.members() {
// stay.insert(resolve_function(&module, *func_index));
// }
// }
// }
for symbol in init_symbols.drain(..) { stay.insert(symbol); }
// Call function which will traverse the list recursively, filling stay with all symbols

13
src/symbols.rs
View File

@ -8,6 +8,7 @@ pub enum Symbol {
Global(usize),
Function(usize),
Export(usize),
IndirectCall,
}
pub fn resolve_function(module: &elements::Module, index: u32) -> Symbol {
@ -58,6 +59,7 @@ pub fn push_code_symbols(module: &elements::Module, opcodes: &[elements::Opcode]
},
&CallIndirect(idx, _) => {
dest.push(Symbol::Type(idx as usize));
dest.push(Symbol::IndirectCall);
},
&GetGlobal(idx) | &SetGlobal(idx) => {
dest.push(resolve_global(module, idx))
@ -143,7 +145,16 @@ pub fn expand_symbols(module: &elements::Module, set: &mut HashSet<Symbol>) {
}
set.insert(symbol);
}
}
},
IndirectCall => {
for entry in module.elements_section().map(|s| s.entries()).unwrap_or(&[]) {
for func_index in entry.members() {
let symbol = resolve_function(&module, *func_index);
if !stop.contains(&symbol) { fringe.push(symbol); }
set.insert(symbol);
}
}
},
_ => {}
}