#![deny(missing_docs)]
#![deny(missing_debug_implementations)]
#[cfg(feature = "dwarf")]
extern crate fallible_iterator;
#[cfg(feature = "dwarf")]
extern crate gimli;
#[cfg(feature = "dwarf")]
extern crate object;
extern crate parity_wasm;
#[cfg(feature = "dwarf")]
extern crate typed_arena;
extern crate twiggy_ir as ir;
extern crate twiggy_traits as traits;
#[cfg(feature = "dwarf")]
mod object_parse;
mod wasm_parse;
use parity_wasm::elements;
use std::fs;
use std::io::Read;
use std::path;
pub fn read_and_parse<P: AsRef<path::Path>>(path: P) -> 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 path.extension().and_then(|s| s.to_str()) {
Some("wasm") => 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(
&self,
items: &mut ir::ItemsBuilder,
extra: Self::ItemsExtra,
) -> Result<(), traits::Error>;
type EdgesExtra;
fn parse_edges(
&self,
items: &mut ir::ItemsBuilder,
extra: Self::EdgesExtra,
) -> Result<(), traits::Error>;
}
fn parse_wasm(data: &[u8]) -> Result<ir::Items, traits::Error> {
let mut items = ir::ItemsBuilder::new(data.len() as u32);
let module: elements::Module = elements::deserialize_buffer(data)?;
let module = match module.parse_names() {
Ok(m) | Err((_, m)) => m,
};
module.parse_items(&mut items, ())?;
module.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);
let file: object::File = object::File::parse(data)?;
file.parse_items(&mut items, ())?;
file.parse_edges(&mut items, ())?;
Ok(items.finish())
}
fn parse_fallback(data: &[u8]) -> Result<ir::Items, traits::Error> {
parse_wasm(data)
}