61 lines
1.5 KiB
Rust
61 lines
1.5 KiB
Rust
extern crate cretonne;
|
|
extern crate cton_reader;
|
|
|
|
use self::cton_reader::parser::Parser;
|
|
use self::cretonne::ir::Ebb;
|
|
use self::cretonne::cfg::ControlFlowGraph;
|
|
use self::cretonne::dominator_tree::DominatorTree;
|
|
|
|
fn test_dominator_tree(function_source: &str, idoms: Vec<u32>) {
|
|
let func = &Parser::parse(function_source).unwrap()[0];
|
|
let cfg = ControlFlowGraph::new(&func);
|
|
let dtree = DominatorTree::new(&cfg);
|
|
assert_eq!(dtree.ebbs().collect::<Vec<_>>().len(), idoms.len());
|
|
for (i, j) in idoms.iter().enumerate() {
|
|
let ebb = Ebb::with_number(i.clone() as u32);
|
|
let idom = Ebb::with_number(*j);
|
|
assert_eq!(dtree.idom(ebb.unwrap()), idom);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn basic() {
|
|
test_dominator_tree("
|
|
function test(i32) {
|
|
ebb0(v0: i32):
|
|
jump ebb1
|
|
ebb1:
|
|
brz v0, ebb3
|
|
jump ebb2
|
|
ebb2:
|
|
jump ebb3
|
|
ebb3:
|
|
return
|
|
}
|
|
", vec![0, 0, 1, 1]);
|
|
}
|
|
|
|
#[test]
|
|
fn loops() {
|
|
test_dominator_tree("
|
|
function test(i32) {
|
|
ebb0(v0: i32):
|
|
brz v0, ebb1
|
|
jump ebb2
|
|
ebb1:
|
|
jump ebb3
|
|
ebb2:
|
|
brz v0, ebb4
|
|
jump ebb5
|
|
ebb3:
|
|
jump ebb4
|
|
ebb4:
|
|
brz v0, ebb3
|
|
jump ebb5
|
|
ebb5:
|
|
brz v0, ebb4
|
|
return
|
|
}
|
|
", vec![0, 0, 0, 0, 0, 0]);
|
|
}
|