#![allow(unknown_lints)]
extern crate cpp_demangle;
#[macro_use]
extern crate clap;
use clap::{App, Arg};
use cpp_demangle::{BorrowedSymbol, DemangleOptions};
use std::io::{self, BufRead, Cursor, Write};
use std::process;
fn find_mangled(haystack: &[u8]) -> Option<usize> {
if haystack.is_empty() {
return None;
}
for i in 0..haystack.len() - 1 {
if haystack[i] == b'_' {
match (haystack[i + 1], haystack.get(i + 2), haystack.get(i + 3), haystack.get(i + 4)) {
(b'Z', _, _, _)
| (b'_', Some(b'Z'), _, _)
| (b'_', Some(b'_'), Some(b'Z'), _) => return Some(i),
| (b'_', Some(b'_'), Some(b'_'), Some(b'Z')) => return Some(i),
_ => (),
}
}
}
None
}
fn demangle_line<W>(out: &mut W, line: &[u8], options: DemangleOptions) -> io::Result<()>
where
W: Write,
{
let mut line = line;
while let Some(idx) = find_mangled(line) {
write!(out, "{}", String::from_utf8_lossy(&line[..idx]))?;
if let Ok((sym, tail)) = BorrowedSymbol::with_tail(&line[idx..]) {
let demangled = sym.demangle(&options).map_err(|e| io::Error::new(io::ErrorKind::Other, e))?;
write!(out, "{}", demangled)?;
line = tail;
} else {
write!(out, "_Z")?;
line = &line[2..];
}
}
write!(out, "{}", String::from_utf8_lossy(line))
}
fn demangle_all<R, W>(input: &mut R, out: &mut W, options: DemangleOptions) -> io::Result<()>
where
R: BufRead,
W: Write,
{
let mut buf = vec![];
while input.read_until(b'\n', &mut buf)? > 0 {
demangle_line(out, &buf[..], options)?;
buf.clear();
}
Ok(())
}
fn main() {
let matches = App::new("cppfilt")
.version(crate_version!())
.author(crate_authors!())
.about("A c++filt clone as an example of how to use the cpp_demangle crate!")
.arg(
Arg::with_name("noparams")
.short("p")
.long("no-params")
.help("Do not display function arguments"),
)
.arg(
Arg::with_name("mangled_names")
.multiple(true)
.value_delimiter(" "),
)
.get_matches();
let stdin = io::stdin();
let mut stdin = stdin.lock();
let stdout = io::stdout();
let mut stdout = stdout.lock();
let stderr = io::stderr();
let mut stderr = stderr.lock();
let options = DemangleOptions {
no_params: matches.is_present("noparams"),
};
let demangle_result = if let Some(names) = matches.values_of("mangled_names") {
let mut input = Cursor::new(names.fold(String::new(), |mut accumulated, name| {
accumulated.push_str(&name);
accumulated.push_str("\n");
accumulated
}));
demangle_all(&mut input, &mut stdout, options)
} else {
demangle_all(&mut stdin, &mut stdout, options)
};
let code = match demangle_result {
Ok(_) => 0,
Err(e) => {
let _ = writeln!(&mut stderr, "error: {}", e);
1
}
};
process::exit(code);
}