Merge pull request #24 from shellrow/dev

Dev

Author: shellrow

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "nrev"
-version = "0.2.0"
+version = "0.3.0"
 edition = "2021"
 authors = ["shellrow <[email protected]>"]
 description = "Simple and Fast Network Revealer/Mapper."

src/app.rs

@@ -1,6 +1,6 @@
-use std::sync::{Mutex, OnceLock};
-use clap::{crate_name, crate_version, crate_description};
 use crate::sys;
+use clap::{crate_description, crate_name, crate_version};
+use std::sync::{Mutex, OnceLock};
 
 // APP information
 pub const CRATE_BIN_NAME: &str = "nrev";
@@ -13,11 +13,9 @@ pub static QUIET_MODE: OnceLock<Mutex<bool>> = OnceLock::new();
 /// Check if quiet mode is enabled
 pub fn is_quiet_mode() -> bool {
     match QUIET_MODE.get() {
-        Some(mutex) => {
-            match mutex.try_lock() {
-                Ok(guard) => *guard,
-                Err(_) => false,
-            }
+        Some(mutex) => match mutex.try_lock() {
+            Ok(guard) => *guard,
+            Err(_) => false,
         },
         None => false,
     }
@@ -29,7 +27,7 @@ pub fn set_quiet_mode(enabled: bool) -> Result<(), String> {
         Ok(mut guard) => {
             *guard = enabled;
             Ok(())
-        },
+        }
         Err(_) => Err("Failed to lock mutex".to_string()),
     }
 }
@@ -58,7 +56,7 @@ impl AppCommands {
             "interfaces" => Some(AppCommands::Interfaces),
             "interface" => Some(AppCommands::Interface),
             "check" => Some(AppCommands::CheckDependencies),
-            _ => None
+            _ => None,
         }
     }
 }

src/config/mod.rs

@@ -11,7 +11,8 @@ pub const PCAP_WAIT_TIME_MILLIS: u64 = 10;
 pub const DEFAULT_PORTS_BIN: &[u8] = include_bytes!("../../resources/ndb-default-ports.bin");
 pub const HTTP_PORTS_BIN: &[u8] = include_bytes!("../../resources/ndb-http-ports.bin");
 pub const HTTPS_PORTS_BIN: &[u8] = include_bytes!("../../resources/ndb-https-ports.bin");
-pub const OS_FAMILY_FINGERPRINT_BIN: &[u8] = include_bytes!("../../resources/ndb-os-family-fingerprint.bin");
+pub const OS_FAMILY_FINGERPRINT_BIN: &[u8] =
+    include_bytes!("../../resources/ndb-os-family-fingerprint.bin");
 pub const OS_TTL_BIN: &[u8] = include_bytes!("../../resources/ndb-os-ttl.bin");
 pub const OS_FAMILY_BIN: &[u8] = include_bytes!("../../resources/ndb-os-family.bin");
 pub const OUI_BIN: &[u8] = include_bytes!("../../resources/ndb-oui.bin");

src/db/mod.rs

@@ -28,7 +28,8 @@ pub fn get_vm_oui_map() -> HashMap<String, String> {
 
 pub fn get_tcp_map() -> HashMap<u16, String> {
     let mut tcp_map: HashMap<u16, String> = HashMap::new();
-    let ds_tcp_service: Vec<model::TcpService> = bincode::deserialize(config::TCP_SERVICE_BIN).unwrap_or(vec![]);
+    let ds_tcp_service: Vec<model::TcpService> =
+        bincode::deserialize(config::TCP_SERVICE_BIN).unwrap_or(vec![]);
     for port in ds_tcp_service {
         tcp_map.insert(port.port, port.service_name);
     }
@@ -41,7 +42,8 @@ pub fn get_default_ports() -> Vec<u16> {
 }
 
 pub fn get_wellknown_ports() -> Vec<u16> {
-    let wellknown_ports: Vec<u16> = bincode::deserialize(config::WELLKNOWN_PORTS_BIN).unwrap_or(vec![]);
+    let wellknown_ports: Vec<u16> =
+        bincode::deserialize(config::WELLKNOWN_PORTS_BIN).unwrap_or(vec![]);
     wellknown_ports
 }
 
@@ -75,7 +77,8 @@ pub fn get_subdomain() -> Vec<String> {
 }
 
 pub fn get_os_family_fingerprints() -> Vec<model::OsFamilyFingerprint> {
-    let ds_os_fingerprints: Vec<model::OsFamilyFingerprint> = bincode::deserialize(config::OS_FAMILY_FINGERPRINT_BIN).unwrap_or(vec![]);
+    let ds_os_fingerprints: Vec<model::OsFamilyFingerprint> =
+        bincode::deserialize(config::OS_FAMILY_FINGERPRINT_BIN).unwrap_or(vec![]);
     ds_os_fingerprints
 }
 
@@ -85,7 +88,9 @@ pub fn get_os_family_list() -> Vec<String> {
 }
 
 pub fn is_vm_fingerprint(fingerprint: &model::OsFingerprint) -> bool {
-    if fingerprint.os_family == "Player".to_string() && fingerprint.device_type == "specialized".to_string() {
+    if fingerprint.os_family == "Player".to_string()
+        && fingerprint.device_type == "specialized".to_string()
+    {
         return true;
     }
     false
@@ -239,7 +244,11 @@ pub fn verify_os_family_fingerprint(fingerprint: &PacketFrame) -> model::OsFamil
     };
 }
 
-pub fn get_os_family(fingerprint: &PacketFrame, os_family_list: &Vec<String>, os_family_fingerprints: &Vec<model::OsFamilyFingerprint>) -> model::OsFamilyFingerprint {
+pub fn get_os_family(
+    fingerprint: &PacketFrame,
+    os_family_list: &Vec<String>,
+    os_family_fingerprints: &Vec<model::OsFamilyFingerprint>,
+) -> model::OsFamilyFingerprint {
     let in_vm: bool = if let Some(ether_header) = &fingerprint.ethernet_header {
         in_vm_network(ether_header.clone())
     } else {
@@ -383,7 +392,7 @@ pub fn get_fingerprint_map(fingerprints: &Vec<PacketFrame>) -> HashMap<IpAddr, S
         let os_fingerprint = get_os_family(&f, &os_family_list, &os_family_fingerprints);
         if let Some(ipv4_header) = &f.ipv4_header {
             fingerprint_map.insert(IpAddr::V4(ipv4_header.source), os_fingerprint.os_family);
-        }else{
+        } else {
             if let Some(ipv6_header) = &f.ipv6_header {
                 fingerprint_map.insert(IpAddr::V6(ipv6_header.source), os_fingerprint.os_family);
             }

src/db/tcp_service.rs

@@ -5844,4 +5844,4 @@ pub(crate) static PORT_SERVICE_MAP: Map<u16, &'static str> = phf_map! {
     49000u16 => "matahari",
     49001u16 => "nusrp",
     49150u16 => "inspider",
-};
+};

src/dns/mod.rs

@@ -16,7 +16,6 @@ use std::collections::HashMap;
 use std::str::FromStr;
 use std::thread;
 
-
 #[cfg(not(target_os = "windows"))]
 const DEFAULT_TIMEOUT: Duration = Duration::from_millis(200);
 #[cfg(not(target_os = "windows"))]

src/dns/result.rs

@@ -1,8 +1,8 @@
 use crate::scan::result::ScanStatus;
 
 use super::domain::Domain;
-use std::time::Duration;
 use serde::{Deserialize, Serialize};
+use std::time::Duration;
 
 /// Result of domain scan  
 #[derive(Clone, Debug, Serialize, Deserialize)]

src/dns/scanner.rs

@@ -143,7 +143,8 @@ impl DomainScanner {
     /// Results are stored in DomainScanner::scan_result
     pub async fn run_scan(&mut self) {
         if self.passive && cfg!(not(feature = "passive")) {
-            self.scan_result.scan_status = ScanStatus::Error(String::from("Passive scan not supported"));
+            self.scan_result.scan_status =
+                ScanStatus::Error(String::from("Passive scan not supported"));
             return;
         }
         let start_time = Instant::now();
@@ -318,8 +319,8 @@ async fn scan_subdomain_passive(
             if r.status().is_success() {
                 match r.text().await {
                     Ok(res_text) => {
-                        let certs_json: serde_json::Value =
-                            serde_json::from_str(res_text.as_str()).unwrap_or(serde_json::json!({}));
+                        let certs_json: serde_json::Value = serde_json::from_str(res_text.as_str())
+                            .unwrap_or(serde_json::json!({}));
                         if certs_json.is_array() {
                             let cert_array = certs_json.as_array().unwrap();
                             for cert in cert_array {
@@ -398,4 +399,4 @@ async fn scan_subdomain_passive(
         result.push(domain.to_owned());
     }
     result
-}
+}

src/fp/setting.rs

@@ -1,9 +1,9 @@
-use std::net::IpAddr;
-use serde::{Deserialize, Serialize};
 use crate::protocol::Protocol;
+use serde::{Deserialize, Serialize};
+use std::net::IpAddr;
 
 #[derive(Deserialize, Serialize, Clone, Copy, Debug, PartialEq, Eq)]
-pub enum  FingerprintType {
+pub enum FingerprintType {
     IcmpEcho,
     IcmpTimestamp,
     IcmpAddressMask,

src/handler/check.rs

@@ -1,12 +1,12 @@
-use clap::ArgMatches;
 use crate::dep;
+use clap::ArgMatches;
 
 pub fn check_dependencies(_arg: &ArgMatches) {
     match dep::check_dependencies() {
         Ok(_) => {
             println!("All dependencies are installed.");
             std::process::exit(0);
-        },
+        }
         Err(e) => {
             println!("Error: {}", e);
             std::process::exit(1);