103 lines
3.0 KiB
Rust
103 lines
3.0 KiB
Rust
//! The `print-cfg` sub-command.
|
|
//!
|
|
//! Read a series of Cretonne IL files and print their control flow graphs
|
|
//! in graphviz format.
|
|
use std::fmt::{Result, Write, Display, Formatter};
|
|
|
|
use CommandResult;
|
|
use utils::read_to_string;
|
|
use cretonne::ir::Function;
|
|
use cretonne::cfg::ControlFlowGraph;
|
|
use cretonne::ir::instructions::BranchInfo;
|
|
use cton_reader::parse_functions;
|
|
|
|
pub fn run(files: Vec<String>) -> CommandResult {
|
|
for (i, f) in files.into_iter().enumerate() {
|
|
if i != 0 {
|
|
println!("");
|
|
}
|
|
try!(print_cfg(f))
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
struct CFGPrinter<'a> {
|
|
func: &'a Function,
|
|
cfg: ControlFlowGraph,
|
|
}
|
|
|
|
impl<'a> CFGPrinter<'a> {
|
|
pub fn new(func: &'a Function) -> CFGPrinter<'a> {
|
|
CFGPrinter {
|
|
func: func,
|
|
cfg: ControlFlowGraph::new(func),
|
|
}
|
|
}
|
|
|
|
/// Write the CFG for this function to `w`.
|
|
pub fn write(&self, w: &mut Write) -> Result {
|
|
try!(self.header(w));
|
|
try!(self.ebb_nodes(w));
|
|
try!(self.cfg_connections(w));
|
|
writeln!(w, "}}")
|
|
}
|
|
|
|
fn header(&self, w: &mut Write) -> Result {
|
|
try!(writeln!(w, "digraph {} {{", self.func.name));
|
|
if let Some(entry) = self.func.layout.entry_block() {
|
|
try!(writeln!(w, " {{rank=min; {}}}", entry));
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
fn ebb_nodes(&self, w: &mut Write) -> Result {
|
|
for ebb in &self.func.layout {
|
|
try!(write!(w, " {} [shape=record, label=\"{{{}", ebb, ebb));
|
|
// Add all outgoing branch instructions to the label.
|
|
for inst in self.func.layout.ebb_insts(ebb) {
|
|
let idata = &self.func.dfg[inst];
|
|
match idata.analyze_branch() {
|
|
BranchInfo::SingleDest(dest, _) => {
|
|
try!(write!(w, " | <{}>{} {}", inst, idata.opcode(), dest))
|
|
}
|
|
BranchInfo::Table(table) => {
|
|
try!(write!(w, " | <{}>{} {}", inst, idata.opcode(), table))
|
|
}
|
|
BranchInfo::NotABranch => {}
|
|
}
|
|
}
|
|
try!(writeln!(w, "}}\"]"))
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
fn cfg_connections(&self, w: &mut Write) -> Result {
|
|
for ebb in &self.func.layout {
|
|
for &(parent, inst) in self.cfg.get_predecessors(ebb) {
|
|
try!(writeln!(w, " {}:{} -> {}", parent, inst, ebb));
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl<'a> Display for CFGPrinter<'a> {
|
|
fn fmt(&self, f: &mut Formatter) -> Result {
|
|
self.write(f)
|
|
}
|
|
}
|
|
|
|
fn print_cfg(filename: String) -> CommandResult {
|
|
let buffer = try!(read_to_string(&filename).map_err(|e| format!("{}: {}", filename, e)));
|
|
let items = try!(parse_functions(&buffer).map_err(|e| format!("{}: {}", filename, e)));
|
|
|
|
for (idx, func) in items.into_iter().enumerate() {
|
|
if idx != 0 {
|
|
println!("");
|
|
}
|
|
print!("{}", CFGPrinter::new(&func));
|
|
}
|
|
|
|
Ok(())
|
|
}
|