winstacks 0.1.0-alpha2

Collect CPU call stack samples from a windows process
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

// Binary to run cargo-flamegraph using winstacks, which pretends to be dtrace

fn main() -> Result<(), winstacks::Error> {
    let args = std::env::args_os().skip(1).collect::<Vec<_>>();
    let dash_c_idx = args
        .iter()
        .enumerate()
        .filter(|(_, arg)| arg.to_str().unwrap() == "-c")
        .next()
        .unwrap()
        .0;
    let args = &args[dash_c_idx + 1..];
    let mut args_v = vec![];
    let mut arg0 = args[0].clone();
    let mut other_args = &args[1..];
    if other_args.is_empty() {
        let mut it = arg0.to_str().unwrap().split_whitespace();
        let arg0_str = std::ffi::OsStr::new(it.next().unwrap());
        args_v = it
            .map(|s| std::ffi::OsStr::new(s).to_os_string())
            .collect::<Vec<_>>();
        other_args = &args_v[..];
        arg0 = arg0_str.to_os_string();
    }
    let trace_ctx = winstacks::trace_command(arg0.clone(), other_args)?;

    let f = std::fs::File::create("./cargo-flamegraph.stacks").expect("Unable to create file");
    let mut f = std::io::BufWriter::new(f);
    trace_ctx.write_dtrace(&mut f)?;

    Ok(())
}