fluencelabs/parity-wasm
1
0
Fork 0
mirror of https://github.com/fluencelabs/parity-wasm synced 2025-06-27 05:32:07 +00:00
Code Issues Projects Releases Wiki Activity

spaces to tabs

Browse Source
This commit is contained in:
NikVolf
2018-01-23 22:47:20 +03:00
parent bb24f2a11a
commit f915a89b9d
36 changed files with 5875 additions and 5876 deletions
Download Patch File Download Diff File Expand all files Collapse all files

92
examples/inject.rs
View File

@ -6,58 +6,58 @@ use parity_wasm::elements;
use parity_wasm::builder;
pub fn inject_nop(opcodes: &mut elements::Opcodes) {
use parity_wasm::elements::Opcode::*;
let opcodes = opcodes.elements_mut();
let mut position = 0;
loop {
let need_inject = match &opcodes[position] {
&Block(_) | &If(_) => true,
_ => false,
};
if need_inject {
opcodes.insert(position + 1, Nop);
}
use parity_wasm::elements::Opcode::*;
let opcodes = opcodes.elements_mut();
let mut position = 0;
loop {
let need_inject = match &opcodes[position] {
&Block(_) | &If(_) => true,
_ => false,
};
if need_inject {
opcodes.insert(position + 1, Nop);
}
position += 1;
if position >= opcodes.len() {
break;
}
}
position += 1;
if position >= opcodes.len() {
break;
}
}
}
fn main() {
let args = env::args().collect::<Vec<_>>();
if args.len() != 3 {
println!("Usage: {} input_file.wasm output_file.wasm", args[0]);
return;
}
let args = env::args().collect::<Vec<_>>();
if args.len() != 3 {
println!("Usage: {} input_file.wasm output_file.wasm", args[0]);
return;
}
let mut module = parity_wasm::deserialize_file(&args[1]).unwrap();
let mut module = parity_wasm::deserialize_file(&args[1]).unwrap();
for section in module.sections_mut() {
match section {
&mut elements::Section::Code(ref mut code_section) => {
for ref mut func_body in code_section.bodies_mut() {
inject_nop(func_body.code_mut());
}
},
_ => { }
}
}
for section in module.sections_mut() {
match section {
&mut elements::Section::Code(ref mut code_section) => {
for ref mut func_body in code_section.bodies_mut() {
inject_nop(func_body.code_mut());
}
},
_ => { }
}
}
let mut build = builder::from_module(module);
let import_sig = build.push_signature(
builder::signature()
.param().i32()
.param().i32()
.return_type().i32()
.build_sig()
);
let build = build.import()
.module("env")
.field("log")
.external().func(import_sig)
.build();
let mut build = builder::from_module(module);
let import_sig = build.push_signature(
builder::signature()
.param().i32()
.param().i32()
.return_type().i32()
.build_sig()
);
let build = build.import()
.module("env")
.field("log")
.external().func(import_sig)
.build();
parity_wasm::serialize_to_file(&args[2], build.build()).unwrap();
parity_wasm::serialize_to_file(&args[2], build.build()).unwrap();
}