whamm 0.1.0

A framework for 'Wasm Application Monitoring and Manipulation'
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

pub mod io;
pub mod maps;
pub mod utils;

use crate::common::error::ErrorGen;
use crate::emitter::memory_allocator::MemoryAllocator;
use crate::generator::ast::AstVisitor;
use crate::lang_features::libraries::core::utils::utils_adapter::UtilsAdapter;
use std::collections::HashMap;
use wirm::ir::id::FunctionID;
use wirm::Module;

pub const WHAMM_CORE_LIB_NAME: &str = "whamm_core";
// TODO: refactor to be more generic!
pub const ASSUMED_LIB_MEM_NAME: &str = "memory";

// A lib package needs to be able to visit the AST and determine if it's needed (should be linked)
pub trait LibPackage: AstVisitor<bool> {
    fn is_used(&self) -> bool;
    fn is_used_in_global_scope(&self) -> bool;
    fn import_memory(&self) -> bool;
    fn set_lib_mem_id(&mut self, mem_id: i32);
    fn set_instr_mem_id(&mut self, mem_id: i32);
    fn get_adapter(&self) -> &dyn LibAdapter;
    fn get_adapter_mut(&mut self) -> &mut dyn LibAdapter;
    fn get_fn_names(&self) -> Vec<String> {
        self.get_adapter().get_fn_names()
    }
    fn add_fid_to_adapter(&mut self, fname: &str, fid: u32) {
        self.get_adapter_mut().add_fid(fname, fid);
    }
    fn set_adapter_usage(&mut self, is_used: bool);
    fn set_global_adapter_usage(&mut self, is_used: bool);
    fn define_helper_funcs(
        &mut self,
        utils: &UtilsAdapter,
        mem_allocator: &mut MemoryAllocator,
        app_wasm: &mut Module,
        err: &mut ErrorGen,
    ) -> Vec<FunctionID>;
}
pub trait LibAdapter {
    fn get_funcs(&self) -> &HashMap<String, u32>;
    fn get_funcs_mut(&mut self) -> &mut HashMap<String, u32>;
    fn define_helper_funcs(
        &mut self,
        utils: &UtilsAdapter,
        mem_allocator: &mut MemoryAllocator,
        app_wasm: &mut Module,
        err: &mut ErrorGen,
    ) -> Vec<FunctionID>;
    fn get_fn_names(&self) -> Vec<String> {
        self.get_funcs().keys().cloned().collect()
    }
    fn get_fid(&self, fname: &str, err: &mut ErrorGen) -> u32 {
        if let Some(fid) = self.get_funcs().get(fname) {
            *fid
        } else {
            // Just return a fake value, the error generator will handle this
            err.add_internal_error(&format!("Function {fname} not found"), &None);
            0
        }
    }

    fn add_fid(&mut self, fname: &str, fid: u32) {
        self.get_funcs_mut().insert(fname.to_string(), fid);
    }
}