Function extract_llfns

xtask/src/extract_llfns.rs:24–64

(file: &Path, dir: &Path)

// ------------------------------------------------

pub(crate) fn extract_llfns(file: &Path, dir: &Path) {
    let contents = std::fs::read_to_string(file).unwrap();
    let re = Regex::new(r#"define .*(_Z.*?)(\(|")"#).unwrap();
    let names = re
        .captures_iter(&contents)
        .map(|x| x.get(1).unwrap().as_str())
        .collect::<Vec<_>>();

    let mut contents = names
        .par_iter()
        .filter_map(|name| {
            let out_file = format!("{}/{}.ll", dir.display(), name);
            let _ = Command::new("llvm-extract")
                .arg(file)
                .arg(&format!("--func={}", name))
                .arg("-S")
                .arg("--recursive")
                .arg("-o")
                .arg(&out_file)
                .spawn();

            let read = std::fs::read_to_string(format!("{}/{}.ll", dir.display(), name)).ok()?;
            Some((name, read, true))
        })
        .collect::<Vec<_>>();

    // sort things by length so that we try the shortest functions first.
    contents.sort_by_key(|x| x.1.len());

    for (name, content, failed) in &mut contents {
        if PRINT_EVERY_EXECUTION {
            println!("Running command over `{}.ll`", name);
        }

        *failed = !run_command_for_each_fn(name, content);
    }

    for (name, _, _) in contents.into_iter().filter(|x| !x.2).take(30) {
        println!("Err: {}", name);
    }
}