wamex-cli 0.1.0

Command line interface for wamex splitter
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

use anyhow::{Result, bail, ensure};
pub use wasmparser::FunctionBody;

use super::Ind;
use crate::index::{FuncTypeId, IdVec, InputFuncId};

#[derive(Debug)]
pub enum InputFunction<'a> {
    Import {},
    Defined(FunctionBody<'a>),
}
#[derive(Debug, Clone)]
pub struct FunctionWithBody<'a> {
    pub type_id: FuncTypeId,
    pub body: FunctionBody<'a>,
}

#[derive(Debug, Default)]
pub struct CodeSection<'a> {
    pub start_func: Option<InputFuncId>,
    // function (CodeSectionEntry)
    pub defined_funcs: IdVec<FunctionWithBody<'a>>,
}
impl<'a> CodeSection<'a> {
    pub fn new(
        start: Option<InputFuncId>,
        funcs: Vec<FunctionBody<'a>>,
        func_types: Vec<FuncTypeId>,
        code_header: Option<(usize, usize, u32)>,
    ) -> Result<Ind<Self>> {
        let Some((code_start, section_index, count)) = code_header else {
            bail!("No code section start");
        };
        ensure!(
            count as usize == funcs.len(),
            "Function count mismatch: {} != {}",
            count,
            funcs.len()
        );
        ensure!(
            count as usize == func_types.len(),
            "Function types count mismatch: {} != {}",
            count,
            func_types.len()
        );
        Ok(Ind {
            starting_offset: code_start,
            section_index,
            section_payload: CodeSection {
                start_func: start,
                defined_funcs: funcs
                    .into_iter()
                    .zip(&func_types)
                    .map(|(body, ty)| FunctionWithBody { type_id: *ty, body })
                    .collect(),
            },
        })
    }
}