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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
#![deny(missing_docs)]
#![deny(missing_debug_implementations)]
use std::ffi::OsStr;
use std::fs;
use std::io::Read;
use std::path;
use twiggy_ir as ir;
use twiggy_traits as traits;
#[cfg(feature = "dwarf")]
mod object_parse;
mod wasm_parse;
const WASM_MAGIC_NUMBER: [u8; 4] = [0x00, 0x61, 0x73, 0x6D];
pub fn read_and_parse<P: AsRef<path::Path>>(
path: P,
mode: traits::ParseMode,
) -> Result<ir::Items, traits::Error> {
let path = path.as_ref();
let mut file = fs::File::open(path)?;
let mut data = vec![];
file.read_to_end(&mut data)?;
match mode {
traits::ParseMode::Wasm => parse_wasm(&data),
#[cfg(feature = "dwarf")]
traits::ParseMode::Dwarf => parse_other(&data),
traits::ParseMode::Auto => match sniff_wasm(path.extension(), &data[..]) {
true => parse_wasm(&data),
#[cfg(feature = "dwarf")]
_ => parse_other(&data),
#[cfg(not(feature = "dwarf"))]
_ => parse_fallback(&data),
},
}
}
pub fn parse(data: &[u8]) -> Result<ir::Items, traits::Error> {
parse_fallback(data)
}
pub(crate) trait Parse<'a> {
type ItemsExtra;
fn parse_items(
&mut self,
items: &mut ir::ItemsBuilder,
extra: Self::ItemsExtra,
) -> Result<(), traits::Error>;
type EdgesExtra;
fn parse_edges(
&mut self,
items: &mut ir::ItemsBuilder,
extra: Self::EdgesExtra,
) -> Result<(), traits::Error>;
}
fn sniff_wasm(extension: Option<&OsStr>, data: &[u8]) -> bool {
match extension.and_then(|s| s.to_str()) {
Some("wasm") => true,
_ => data.get(0..4) == Some(&WASM_MAGIC_NUMBER),
}
}
fn parse_wasm(data: &[u8]) -> Result<ir::Items, traits::Error> {
let mut items = ir::ItemsBuilder::new(data.len() as u32);
let mut module1 = wasmparser::ModuleReader::new(data)?;
module1.parse_items(&mut items, ())?;
let mut module2 = wasmparser::ModuleReader::new(data)?;
module2.parse_edges(&mut items, ())?;
Ok(items.finish())
}
#[cfg(feature = "dwarf")]
fn parse_other(data: &[u8]) -> Result<ir::Items, traits::Error> {
let mut items = ir::ItemsBuilder::new(data.len() as u32);
object_parse::parse(&mut items, data)?;
Ok(items.finish())
}
fn parse_fallback(data: &[u8]) -> Result<ir::Items, traits::Error> {
parse_wasm(data)
}