From 326a38551861a9cb6177d7a84aa6b0ce3d85622d Mon Sep 17 00:00:00 2001
From: "E. Almqvist" <elalmqvist@gmail.com>
Date: Wed, 10 Aug 2022 00:20:22 +0200
Subject: [PATCH] Revert & cleanup

---
 src/scanner.rs | 83 +++++++++++++++++++++++++-------------------------
 1 file changed, 42 insertions(+), 41 deletions(-)

diff --git a/src/scanner.rs b/src/scanner.rs
index f2a6404..523ba66 100644
--- a/src/scanner.rs
+++ b/src/scanner.rs
@@ -1,7 +1,8 @@
 use crate::ipv4;
 use anyhow::Result;
 use log::info;
-//use std::thread;
+use std::thread;
+use std::thread::JoinHandle;
 use std::net::{SocketAddr, TcpStream};
 
 fn tcp_scan(dest: &SocketAddr) -> bool {
@@ -14,49 +15,49 @@ fn tcp_scan(dest: &SocketAddr) -> bool {
 
 // TODO: do thread optimization
 
-// pub fn scan(target: &ipv4::IPv4) -> bool {
-//     true
-// }
-//
-// fn create_scan_thread(ip_list: Vec<ipv4::IPv4>, thread_id: u32, ips_per_thread: u32, results_list: &Vec<bool>) -> JoinHandle<Vec<bool>> {
-//     thread::spawn(move || {
-//         let results: Vec<bool> = Vec::new();
-//         // do the scan thing
-//         for i in 0..ips_per_thread {
-//             let id = (thread_id * ips_per_thread) + i;
-//             let ref target = ip_list[id as usize];
-//             let result = scan(&target);
-//         }
-//
-//         results
-//     })
-// }
-//
+pub fn scan(target: &ipv4::IPv4) -> bool {
+    true
+}
+
+fn create_scan_thread(
+    ip_list: Vec<ipv4::IPv4>,
+    thread_id: u32,
+    ips_per_thread: u32,
+) -> JoinHandle<Vec<bool>> {
+    thread::spawn(move || {
+        let results: Vec<bool> = Vec::new();
+
+        // do the scan thing
+        for i in 0..ips_per_thread {
+            let id = (thread_id * ips_per_thread) + i;
+            let ref target = ip_list[id as usize];
+            let result = scan(&target);
+        }
+
+        results
+    })
+}
+
 pub fn start_scan(target_port: u16, num_threads: u32, ignorelist: Option<Vec<u64>>) -> Result<()> {
     let ip_list = ipv4::get_all(ignorelist)?;
 
-    // casting hell
-    //     let ips_per_thread = ((ip_list.len() as f32) / num_threads as f32) as u32;
-    //     let ips_left = num_threads * ips_per_thread; // how many IPs we have left after the first threads
-    //
-    //     // Container for all of our threads
-    //     let mut threads: Vec<JoinHandle<()>> = Vec::new();
-    //
-    //     // Create all of our threads
-    //     for thread_id in 0..num_threads {
-    //         threads.push( create_scan_thread(ip_list.borrow().clone(), thread_id, ips_per_thread) );
-    //     }
-    //
-    //     // Create the last thread to do the job
-
-    let mut results: Vec<bool> = Vec::new();
-
-    ip_list.iter().for_each(|ip| {
-        if !ip.ignore {
-            let target = ip.to_socketaddr(target_port).unwrap();
-            results.push(tcp_scan(&target));
-        }
-    });
+    let ips_per_thread = ((ip_list.len() as f32) / num_threads as f32) as u32;
+    let ips_left = num_threads * ips_per_thread; // how many ips we have left after the first threads
+
+    // container for all of our threads
+    let mut threads: Vec<JoinHandle<Vec<bool>>> = Vec::new();
+    let result_list: Vec<bool> = Vec::new();
+
+    // create all of our threads
+    for thread_id in 0..num_threads {
+        threads.push(create_scan_thread(
+            ip_list,
+            thread_id,
+            ips_per_thread,
+        ));
+    }
+
+    // create the last thread to do the job
 
     Ok(())
 }