Add option to exclude object exportation

src/main.rs

@@ -59,7 +59,8 @@ fn main() {
  Some(filter_string) => filter = Regex::new(filter_string).unwrap(),
  None => filter = Regex::new(".*").unwrap(),
  }
- match subcommands::run::init(ingress, egress, filter) {
+ let export = cliopts.subcommand_matches(subcommand_string).unwrap().is_present("export-objects");
+ match subcommands::run::init(ingress, egress, filter, export) {
  Ok(_) => {}
  Err(_) => {}
  }

src/options.rs

@@ -2,7 +2,7 @@ use clap::{App, Arg, ArgMatches, SubCommand};
 
 pub fn get_options_from_cli() -> ArgMatches<'static> {
  App::new("Monet: Monitor a Network")
- .version("0.1.5-rc8")
+ .version("0.1.5-rc9")
  .author("André Nascimento and Rui Lopes")
  .about("Tool to get some insights on a network or interfaces")
  .subcommand(
@@ -25,6 +25,14 @@ pub fn get_options_from_cli() -> ArgMatches<'static> {
  .default_value("")
  .help("Set of egress interfaces, between quotes: i.e. \"enp0s1,wlan0\"")
  )
+ .arg(
+ Arg::with_name("export-objects")
+ .takes_value(false)
+ .long("export")
+ .short("O")
+ .required(false)
+ .help("Flag if you want to export captures as monet's files.")
+ )
  .arg(
  Arg::with_name("filter")
  .takes_value(true)

src/subcommands/run.rs

@@ -17,7 +17,7 @@ use crate::capture::results::CaptureResult;
 use crate::structs::node::status::NodeStatus;
 use crate::structs::node::Node;
 
-pub fn init(ingress: String, egress: String, filter: Regex) -> Result<(), Error> {
+pub fn init(ingress: String, egress: String, filter: Regex, export: bool) -> Result<(), Error> {
  // Start both ingress and egress captures.
  thread::scope(|scope| {
  let home = match env::var_os("HOME") {
@@ -85,25 +85,26 @@ pub fn init(ingress: String, egress: String, filter: Regex) -> Result<(), Error>
  }
  }
  }
-
- let mut file = OpenOptions::new()
- .create_new(true)
- .write(true)
- .open(if *is_ingress {
- println!(
- "\nTrying to create {}",
+ if export {
+ let mut file = OpenOptions::new()
+ .create_new(true)
+ .write(true)
+ .open(if *is_ingress {
+ println!(
+ "\nTrying to create {}",
+ format!("{}/.monet/ingress-{}.monet", home.display(), interface)
+ );
  format!("{}/.monet/ingress-{}.monet", home.display(), interface)
- );
- format!("{}/.monet/ingress-{}.monet", home.display(), interface)
- } else {
- println!(
- "\nTrying to create {}",
+ } else {
+  println!(
+  "\nTrying to create {}",
+  format!("{}/.monet/egress-{}.monet", home.display(), interface)
+ );
  format!("{}/.monet/egress-{}.monet", home.display(), interface)
- );
- format!("{}/.monet/egress-{}.monet", home.display(), interface)
- })
- .unwrap();
- bincode::serialize_into(&mut file, &capture).unwrap();
+ })
+ .unwrap();
+ bincode::serialize_into(&mut file, &capture).unwrap();
+ }
  });
  std::process::exit(0);
  });