br0kej / faser Goto Github PK

    /// Generates a file containing every instruction within each of the functions
    /// within the AGFJFile.
    ///
    /// This ignores control flow and simple iterates the JSON objects from the top to
    /// the bottom.
    pub fn generate_linear_bb_walk(mut self, esil: bool) {
        //let fname_string: PathBuf = get_save_file_path(&self.filename, &self.output_path, None, None);
        //let fname_string = if esil {format!("{:?}-esil-singles.txt", fname_string)} else {format!("{:?}-dis-singles.txt", fname_string)};

        let fname_string: PathBuf = get_save_file_path(&self.filename, &self.output_path, None, None);
        let fname_string = if esil {format!("{}-esil-singles.txt", fname_string.display())} else {format!("{}-dis-singles.txt", fname_string.display())};

        if !Path::new(&fname_string).exists() {
            self.load_and_deserialize()
                .expect("Unable to load and desearlize JSON");

            let (sender, receiver) = channel();

            self.functions.unwrap().par_iter_mut().for_each_with(
                sender,
                |s, func: &mut Vec<AGFJFunc>| {
                    s.send(func[0].get_function_instructions(esil, &self.min_blocks, self.reg_norm))
                        .unwrap()
                },
            );

            let res: Vec<Vec<String>> = receiver.iter().filter(|x| x.is_some()).flatten().collect();

            //let write_file = File::create(fname_string).unwrap();
            let write_file = match File::create(&fname_string) {
                Ok(file) => file,
                Err(e) => {
                    //eprintln!("Error creating file: {}", e);
                    eprintln!("Error creating file {}: {}", fname_string, e);
                    return;
                }
            };

            let mut writer = BufWriter::new(&write_file);

            for func in res {
                for bb in func {
                    writer
                        .write_all(bb.as_bytes())
                        .expect("Unable to write bytes.");
                    writer.write_all(b"\n").expect("Unable to write bytes.");
                }
            }
        }
    }