Merge pull request #6 from thewh1teagle/feat/windows-support

feat: add windows support

Author: Saluki

.gitignore

@@ -1,3 +1,4 @@
 /target
 /builds
 **/*.rs.bk
+*.lib

README.md

@@ -224,6 +224,16 @@ The features below will be shipped in the next releases of the project.
 - Change verbose options (for debug, network details, quiet mode, ...)
 - Avoid packet copy in userspace for faster scans (BPF filtering)
 
+## Building
+#### Linux / Mac
+Simply run `cargo build`
+
+#### Windows
+See [github.com/libpnet/libpnet#windows](https://github.com/libpnet/libpnet#windows)
+In additional for what they described there, 
+for linking `Packet.lib` you can just place it in the root of this project.
+
+
 ## Contributing
 
 Feel free to suggest an improvement, report a bug, or ask something: https://github.com/saluki/arp-scan-rs/issues

src/args.rs

@@ -61,10 +61,15 @@ pub fn build_args() -> Command {
                 ])
                 .help("Scan profile - a preset of ARP scan options")
         )
+        .arg(
+            Arg::new("index").long("index")
+                .value_name("INTERFACE_INDEX")
+                .help("Network interface index (defaults to first 'up' interface with IPv4)")
+        )
         .arg(
             Arg::new("interface").short('i').long("interface")
                 .value_name("INTERFACE_NAME")
-                .help("Network interface (defaults to first 'up' interface with IPv4)")
+                .help("Network interface name (defaults to first 'up' interface with IPv4)")
         )
         .arg(
             Arg::new("network").short('n').long("network")
@@ -207,6 +212,7 @@ pub enum ScanTiming {
 pub struct ScanOptions {
     pub profile: ProfileType,
     pub interface_name: Option<String>,
+    pub interface_index: Option<u32>,
     pub network_range: Option<Vec<ipnetwork::IpNetwork>>,
     pub timeout_ms: u64,
     pub resolve_hostname: bool,
@@ -330,6 +336,9 @@ impl ScanOptions {
         };
 
         let interface_name = matches.get_one::<String>("interface").cloned();
+        let interface_index_str = matches.get_one::<String>("index").cloned();
+        let interface_index = interface_index_str.unwrap_or_default().parse::<u32>().ok();
+
 
         let file_option = matches.get_one::<String>("file");
         let network_option = matches.get_one::<String>("network");
@@ -528,6 +537,7 @@ impl ScanOptions {
         Arc::new(ScanOptions {
             profile,
             interface_name,
+            interface_index,
             network_range,
             timeout_ms,
             resolve_hostname,

src/main.rs

@@ -45,7 +45,7 @@ fn main() {
         process::exit(0);
     }
 
-    if !utils::is_root_user() {
+    if !cfg!(windows) && !utils::is_root_user() {
         eprintln!("Should run this binary as root or use --help for options");
         process::exit(1);
     }

src/network.rs

@@ -69,29 +69,23 @@ pub struct TargetDetails {
  * specific network on a network interfaces.
  */
 pub fn compute_network_configuration<'a>(interfaces: &'a [NetworkInterface], scan_options: &'a Arc<ScanOptions>) -> (&'a NetworkInterface, Vec<&'a IpNetwork>) {
-
-    let interface_name = match &scan_options.interface_name {
-        Some(name) => String::from(name),
-        None => {
-
-            let name = utils::select_default_interface(interfaces).map(|interface| interface.name);
-
-            match name {
-                Some(name) => name,
-                None => {
+    let selected_interface = match (&scan_options.interface_name, &scan_options.interface_index) {
+        (Some(interface_name), _) => {
+            find_interface_by_name(interfaces, interface_name)
+        },
+        (None, Some(interface_index)) => {
+            find_interface_by_index(interfaces, *interface_index)
+        },
+        _ => {
                     eprintln!("Could not find a default network interface");
                     eprintln!("Use 'arp scan -l' to list available interfaces");
                     process::exit(1);
                 }
-            }
-        }
     };
 
-    let selected_interface: &NetworkInterface = interfaces.iter()
-        .find(|interface| { interface.name == interface_name && interface.is_up() && !interface.is_loopback() })
-        .unwrap_or_else(|| {
-            eprintln!("Could not find interface with name {}", interface_name);
-            eprintln!("Make sure the interface is up, not loopback and has a valid IPv4");
+    let selected_interface = selected_interface.unwrap_or_else(|| {
+        eprintln!("Could not find the specified interface");
+        eprintln!("Make sure the interface is up, not loopback, and has a valid IPv4");
             process::exit(1);
         });
 
@@ -103,6 +97,17 @@ pub fn compute_network_configuration<'a>(interfaces: &'a [NetworkInterface], sca
     (selected_interface, ip_networks)
 }
 
+fn find_interface_by_name<'a>(interfaces: &'a [NetworkInterface], interface_name: &str) -> Option<&'a NetworkInterface> {
+    interfaces.iter()
+        .find(|interface| interface.name == interface_name && (cfg!(windows) || interface.is_up()) && !interface.is_loopback())
+}
+
+fn find_interface_by_index<'a>(interfaces: &'a [NetworkInterface], interface_index: u32) -> Option<&'a NetworkInterface> {
+    interfaces.iter()
+        .find(|interface| interface.index == interface_index && (cfg!(windows) || interface.is_up()) && !interface.is_loopback())
+}
+
+
 /**
  * Based on the network size and given scan options, this function performs an
  * estimation of the scan impact (timing, bandwidth, ...). Keep in mind that

src/utils.rs

@@ -30,7 +30,6 @@ pub fn show_interfaces(interfaces: &[NetworkInterface]) {
 
     println!();
     for interface in interfaces.iter() {
-
         let up_text = match interface.is_up() {
             true => format!("{} UP", Green.paint("✔")),
             false => format!("{} DOWN", Red.paint("✖"))
@@ -44,7 +43,8 @@ pub fn show_interfaces(interfaces: &[NetworkInterface]) {
             None => "".to_string()
         };
 
-        println!("{: <20} {: <18} {: <20} {}", interface.name, up_text, mac_text, first_ip);
+        let index_text = format!("{} ", interface.index);
+        println!("{: <3} {: <20} {: <18} {: <20} {}", index_text ,interface.name, up_text, mac_text, first_ip);
 
         interface_count += 1;
         if interface.is_up() && !interface.is_loopback() && !interface.ips.is_empty() {