Fix the worker logic

This commit is contained in:
2025-04-06 22:22:01 +02:00
parent 9fa4dad52b
commit 9593af7b66

View File

@ -2,7 +2,6 @@ use std::{env, net::IpAddr};
use chrono::{Duration, NaiveDateTime, Utc}; use chrono::{Duration, NaiveDateTime, Utc};
use cidr::IpCidr; use cidr::IpCidr;
use detection::detect_scanner;
use dns_ptr_resolver::{get_ptr, ResolvedResult}; use dns_ptr_resolver::{get_ptr, ResolvedResult};
use log2::*; use log2::*;
use scanners::Scanners; use scanners::Scanners;
@ -160,14 +159,16 @@ impl Worker {
for addr in addresses { for addr in addresses {
let client = get_dns_client(&get_dns_server_config(&rr_dns_servers.next().unwrap())); let client = get_dns_client(&get_dns_server_config(&rr_dns_servers.next().unwrap()));
match get_ptr(addr, client) { match get_ptr(addr, client) {
Ok(result) => match detect_scanner(&result) { Ok(result) => {
Ok(Some(scanner_name)) => { let scanner: Result<Scanners, String> = result.query.clone().try_into();
match scanner {
Ok(scanner_name) => {
self.report_detection(scanner_name, addr, result); self.report_detection(scanner_name, addr, result);
} }
Ok(None) => {}
Err(err) => error!("Error detecting for {addr}: {:?}", err), Err(err) => error!("Error detecting for {addr}: {:?}", err),
}, }
}
Err(_) => { Err(_) => {
//debug!("Error processing {addr}: {err}") //debug!("Error processing {addr}: {err}")
} }