make it atomic

src/main.rs

@@ -1,13 +1,15 @@
 use std::{
     io::{self, Write},
     net::{SocketAddr, TcpStream},
+    sync::atomic::{AtomicU64, Ordering},
+    sync::Arc,
 };
 
 use clap::Parser;
 
 #[derive(Debug, Parser)]
 struct Args {
-    #[clap(default_value = "172.0.0.1:113")]
+    #[clap(default_value = "127.0.0.1:113")]
     ip: SocketAddr,
 
     #[clap(default_value = "20")]
@@ -19,28 +21,30 @@ fn main() {
 
     let threads = cli.threads;
 
-    static mut SEND_COUNT: u64 = 0;
+    println!("Target: {}", cli.ip);
+
+    let send_count = Arc::new(AtomicU64::new(0));
 
     for _ in 0..threads {
+        let send_count = Arc::clone(&send_count);
         std::thread::spawn(move || loop {
             match TcpStream::connect_timeout(&cli.ip, std::time::Duration::from_nanos(100)) {
-                Ok(_) => unsafe {
+                Ok(_) => send_count.fetch_add(1, Ordering::Relaxed),
-                    SEND_COUNT += 1;
+                Err(_) => send_count.fetch_add(1, Ordering::Relaxed),
-                },
+            };
-                Err(_) => unsafe {
-                    SEND_COUNT += 1;
-                },
-            }
         });
     }
 
-    std::thread::spawn(|| {
+    std::thread::spawn(move || {
         let mut last_count = 0;
         loop {
-            print!("{} Packets sent. {}/sec", unsafe { SEND_COUNT }, unsafe {
+            let send_count = send_count.load(Ordering::Relaxed);
-                SEND_COUNT - last_count
+            print!(
-            });
+                "{} Packets sent. {}/sec",
-            last_count = unsafe { SEND_COUNT };
+                send_count,
+                send_count - last_count
+            );
+            last_count = send_count;
             io::stdout().flush().unwrap();
             std::thread::sleep(std::time::Duration::from_secs(1));
             print!("\r");
@@ -49,5 +53,7 @@ fn main() {
 
     println!("{} Threads created.", threads);
 
-    loop {}
+    loop {
+        std::thread::sleep(std::time::Duration::from_secs(1));
+    }
 }