run scanner on start

This commit is contained in:
funtimes909
2026-04-05 18:10:41 +12:00
parent 8219292672
commit 1d88e59bb3
2 changed files with 10 additions and 3 deletions
+8 -1
View File
@@ -17,6 +17,7 @@ use tracing::{error, info};
use crate::config::Config;
use crate::database::{Database, country_tracking};
use crate::scanning::discovery::DiscoveryScanner;
use crate::scanning::rescanner::Rescanner;
#[derive(Parser, Debug)]
@@ -126,7 +127,13 @@ async fn main() {
}
match arguments.mode {
Mode::Scanning => todo!(),
Mode::Scanning => {
let scanner = DiscoveryScanner {
database: Database { connection: pool },
};
scanner.scan().await;
}
Mode::Rescanner => {
let rescanner = Rescanner {
is_active: true,
+2 -2
View File
@@ -10,11 +10,11 @@ use crate::database::ServerUpdateOperation;
use crate::database::Database;
pub struct DiscoveryScanner {
database: Database,
pub database: Database,
}
impl DiscoveryScanner {
async fn scan(&self) {
pub async fn scan(&self) {
// Spawn masscan
let mut command = Command::new("sudo")
.args(["masscan", "-c", &CONFIG.masscan.config_file])