invoke/
invoke.rs

1extern crate sophon_wasm;
2
3use std::env::args;
4
5use sophon_wasm::{interpreter, ModuleInstanceInterface, RuntimeValue};
6use sophon_wasm::elements::{Internal, External, Type, FunctionType, ValueType};
7
8
9fn main() {
10    let args: Vec<_> = args().collect();
11    if args.len() < 3 {
12        println!("Usage: {} <wasm file> <exported func> [<arg>...]", args[0]);
13        return;
14    }
15    let func_name = &args[2];
16    let (_, program_args) = args.split_at(3);
17
18    // Intrepreter initialization.
19    let program = sophon_wasm::ProgramInstance::new();
20
21    let module = sophon_wasm::deserialize_file(&args[1]).expect("File to be deserialized");
22
23    // Extracts call arguments from command-line arguments
24    let execution_params = {
25        // Export section has an entry with a func_name with an index inside a module
26        let export_section = module.export_section().expect("No export section found");
27        // It's a section with function declarations (which are references to the type section entries)
28        let function_section = module.function_section().expect("No function section found");
29        // Type section stores function types which are referenced by function_section entries
30        let type_section = module.type_section().expect("No type section found");
31
32        // Given function name used to find export section entry which contains
33        // an `internal` field which points to the index in the function index space
34        let found_entry = export_section.entries().iter()
35            .find(|entry| func_name == entry.field()).expect(&format!("No export with name {} found", func_name));
36
37        // Function index in the function index space (internally-defined + imported)
38        let function_index: usize = match found_entry.internal() {
39            &Internal::Function(index) => index as usize,
40            _ => panic!("Founded export is not a function"),
41        };
42
43        // We need to count import section entries (functions only!) to subtract it from function_index
44        // and obtain the index within the function section
45        let import_section_len: usize = match module.import_section() {
46            Some(import) =>
47                import.entries().iter().filter(|entry| match entry.external() {
48                    &External::Function(_) => true,
49                    _ => false,
50                    }).count(),
51            None => 0,
52        };
53
54        // Calculates a function index within module's function section
55        let function_index_in_section = function_index - import_section_len;
56
57        // Getting a type reference from a function section entry
58        let func_type_ref: usize = function_section.entries()[function_index_in_section].type_ref() as usize;
59
60        // Use the reference to get an actual function type
61        let function_type: &FunctionType = match &type_section.types()[func_type_ref] {
62            &Type::Function(ref func_type) => func_type,
63        };
64
65        // Parses arguments and constructs runtime values in correspondence of their types
66        let args: Vec<RuntimeValue> = function_type.params().iter().enumerate().map(|(i, value)| match value {
67            &ValueType::I32 => RuntimeValue::I32(program_args[i].parse::<i32>().expect(&format!("Can't parse arg #{} as i32", program_args[i]))),
68            &ValueType::I64 => RuntimeValue::I64(program_args[i].parse::<i64>().expect(&format!("Can't parse arg #{} as i64", program_args[i]))),
69            &ValueType::F32 => RuntimeValue::F32(program_args[i].parse::<f32>().expect(&format!("Can't parse arg #{} as f32", program_args[i]))),
70            &ValueType::F64 => RuntimeValue::F64(program_args[i].parse::<f64>().expect(&format!("Can't parse arg #{} as f64", program_args[i]))),
71        }).collect();
72
73        interpreter::ExecutionParams::from(args)
74    };
75
76    // Intialize deserialized module. It adds module into It expects 3 parameters:
77    // - a name for the module
78    // - a module declaration
79    // - "main" module doesn't import native module(s) this is why we don't need to provide external native modules here
80    // This test shows how to implement native module https://github.com/NikVolf/sophon-wasm/blob/master/src/interpreter/tests/basics.rs#L197
81    let module = program.add_module("main", module, None).expect("Failed to initialize module");
82
83    println!("Result: {:?}", module.execute_export(func_name, execution_params).expect(""));
84}
invoke/invoke.rs

invoke/
invoke.rs
