Merge pull request #12 from BugenZhao/bz-dump-targets

tr: allow dumping targets

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "flashroute_rs"
-version = "0.1.0"
+version = "0.1.2"
 authors = [
     "Bugen Zhao <[email protected]>",
     "Zhengdong Wang <[email protected]>",

src/opt.rs

@@ -55,6 +55,8 @@ pub struct Opt {
     pub salt: u16,
     #[structopt(long)]
     pub dry_run: bool,
+    #[structopt(long)]
+    pub dump_targets: Option<PathBuf>,
     #[structopt(short = "D", long)]
     pub debug: bool,
 

src/tracerouter.rs

@@ -1,5 +1,7 @@
 use std::{
+    io::Write,
     net::Ipv4Addr,
+    path::PathBuf,
     sync::{
         atomic::{AtomicBool, AtomicU64, Ordering},
         Arc,
@@ -56,6 +58,11 @@ impl Tracerouter {
         log::info!("Initializing targets...");
         let targets = Self::generate_targets()?;
 
+        if let Some(path) = OPT.dump_targets.clone() {
+            log::info!("Dumping targets...");
+            Self::dump_targets(&targets, &path)?;
+        }
+
         Ok(Self {
             targets: Arc::new(targets),
             ..Self::default()
@@ -129,6 +136,15 @@ impl Tracerouter {
             }
         }
     }
+
+    fn dump_targets(targets: &DcbMap, path: &PathBuf) -> Result<()> {
+        let mut file = std::fs::File::create(path)?;
+        for DstCtrlBlock { addr, .. } in targets.values() {
+            file.write_fmt(format_args!("{}\n", addr.to_string()))?;
+        }
+
+        Ok(())
+    }
 }
 
 impl Tracerouter {