mirror of
https://git.funtimes909.xyz/ServerSeekerV2/ServerSeekerV2.git
synced 2026-10-10 04:03:09 +08:00
run scanner on start
This commit is contained in:
+8
-1
@@ -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,
|
||||
|
||||
@@ -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])
|
||||
|
||||
Reference in New Issue
Block a user