use super::*;
use crate::error::Error;
use crate::globals::*;
use regex::Regex;
use std::process::Command;
pub fn update_handshakes() -> Result<(), Error> {
let handshakes = get_handshakes([
&(LIVE_SCAN_PATH.to_string() + "-01.cap"),
&(OLD_SCAN_PATH.to_string() + "-01.cap"),
])?;
let mut aps = get_aps();
for (bssid, _) in handshakes {
if let Some(ap) = aps.get_mut(&bssid) {
ap.handshake = true;
}
}
log::trace!("handshakes updated");
Ok(())
}
pub fn get_handshakes<I, S>(args: I) -> Result<Vec<(String, String)>, Error>
where
I: IntoIterator<Item = S>,
S: AsRef<std::ffi::OsStr>,
{
let capture_output = Command::new("aircrack-ng").args(args).output()?;
let output = String::from_utf8_lossy(&capture_output.stdout).to_string();
let re = Regex::new(r"\s+(\d+)\s+([\w:]+)\s+(.*)\s+WPA \((\d+)\s+handshake.*\)")?;
let mut handshakes = vec![];
for line in output.lines() {
let caps = match re.captures(line) {
Some(caps) => caps,
None => continue,
};
let bssid = caps[2].to_string();
let essid = caps[3].trim().to_string();
let handshake_count = caps[4].to_string();
if handshake_count.parse::<i32>().unwrap_or(0) > 0 {
handshakes.push((
bssid,
match essid.is_empty() {
true => "hidden".to_string(),
false => essid,
},
));
}
}
Ok(handshakes)
}
pub fn save_capture(path: &str) -> Result<(), Error> {
std::fs::copy(OLD_SCAN_PATH.to_string() + "-01.cap", path)?;
log::info!("capture saved to \"{}\"", path);
Ok(())
}