feat: add ICMP fallback and handle unspecified netnetlink

1 files changed, 7 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 6f60514..529c1ed 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -15,6 +15,8 @@ use tokio::{
 
 mod config;
 
+const PAYLOAD: [u8; 8] = [0; 8];
+
 #[tokio::main]
 async fn main() -> Result<(), anyhow::Error> {
     let config = Config::parse();
@@ -53,7 +55,9 @@ async fn main() -> Result<(), anyhow::Error> {
     let cache = Arc::new(Mutex::new(cache));
     let names = Arc::new(Mutex::new(names));
 
-    join_set.spawn(sshr::netlink::neighbours(tx.clone()));
+    if config.neigh {
+        join_set.spawn(sshr::netlink::neighbours(tx.clone()));
+    }
 
     if let Some(ip_net) = config.scan {
         join_set.spawn(sshr::scanner::scan(tx.clone(), ip_net));
@@ -76,6 +80,8 @@ async fn main() -> Result<(), anyhow::Error> {
                 if block_in_place(|| TcpStream::connect_timeout(&addr, timeout)).is_err() {
                     return Ok(());
                 }
+            } else if surge_ping::ping(ip_addr, &PAYLOAD).await.is_err() {
+                return Ok(());
             }
 
             if let Some(s) = config.resolve.then(|| lookup_addr(&ip_addr).ok()).flatten() {