havn 0.1.3

A fast configurable port scanner with reasonable defaults
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

#![forbid(unsafe_code)]
#![warn(
    clippy::expect_used,
    clippy::nursery,
    clippy::pedantic,
    clippy::todo,
    clippy::unused_async,
    clippy::unwrap_used
)]
#![allow(
    clippy::module_name_repetitions,
    clippy::doc_markdown,
    clippy::similar_names
)]
// Only allow when debugging
// #![allow(unused)]

use scanner::{host_info, AllPortStatus};
use terminal::{print, spinner::Spinner};

mod parse_arg;
mod port_descriptions;
mod scanner;
mod terminal;

/// On exit re-show the cursor
fn exit(code: i32) {
    Spinner::show_cursor();
    std::process::exit(code);
}

/// Spawn a thread to watch for exit signals, so can show cursor correctly
fn tokio_signal() {
    tokio::spawn(async move {
        tokio::signal::ctrl_c().await.ok();
        exit(1);
    });
}

#[tokio::main]
async fn main() {
    let cli_args = parse_arg::CliArgs::new();
    terminal::text_color(&cli_args);
    let exit_error = || print::address_error(&cli_args);

    tokio_signal();

    if let Ok(host_info) = host_info::HostInfo::try_from(&cli_args.address).await {
        if let Some(ip) = host_info.get_ip(&cli_args) {
            print::name_and_target(&cli_args, ip);
            print::extra_ips(&host_info, ip);

            let spinner = Spinner::start();
            let now = std::time::Instant::now();
            let scan_output = AllPortStatus::scan_ports(&cli_args, ip).await;
            spinner.stop();
            let done = now.elapsed();

            print::scan_time(&scan_output, done);
            print::result_table(&scan_output);
        } else {
            exit_error();
        }
    } else {
        exit_error();
    }
}