ntrace 0.1.4

A fast and secure network port scanner and protocol analyzer
Documentation 
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
68
69
70
71
72
73
74
75
76
77
78
79
80

use clap::Parser;
use ntrace::cli::Cli;

#[test]
fn test_cli_parsing() {
    // Test basic CLI parsing
    let args = vec!["ntrace", "-H", "192.168.1.1"];
    let cli = Cli::parse_from(args);

    assert_eq!(cli.host, Some("192.168.1.1".to_string()));
    assert_eq!(cli.ports, "1-1000");
    assert_eq!(cli.protocol, "tcp");
    assert_eq!(cli.timeout, 2.0);
    assert_eq!(cli.batch_size, 100);
    assert_eq!(cli.service_detection, true);
    assert_eq!(cli.verbose, false);
    assert_eq!(cli.rate_limit, 1000);
    assert_eq!(cli.skip_discovery, false);
    assert_eq!(cli.aggressive, false);
}

#[test]
fn test_cli_with_options() {
    // Test CLI parsing with options
    let args = vec![
        "ntrace",
        "-H",
        "example.com",
        "-p",
        "80,443",
        "-P",
        "tcp",
        "--timeout",
        "5.0",
        "--batch-size",
        "50",
        "-v",
        "--skip-discovery",
    ];
    let cli = Cli::parse_from(args);

    assert_eq!(cli.host, Some("example.com".to_string()));
    assert_eq!(cli.ports, "80,443");
    assert_eq!(cli.protocol, "tcp");
    assert_eq!(cli.timeout, 5.0);
    assert_eq!(cli.batch_size, 50);
    assert_eq!(cli.verbose, true);
    assert_eq!(cli.skip_discovery, true);
}

#[test]
fn test_port_parsing() {
    // Test port range parsing
    let args = vec!["ntrace", "-H", "192.168.1.1", "-p", "1-100"];
    let cli = Cli::parse_from(args);
    let ports = cli.parse_ports().unwrap();

    assert_eq!(ports.len(), 100);
    assert_eq!(ports[0], 1);
    assert_eq!(ports[99], 100);

    // Test comma separated ports
    let args = vec!["ntrace", "-H", "192.168.1.1", "-p", "22,80,443"];
    let cli = Cli::parse_from(args);
    let ports = cli.parse_ports().unwrap();

    assert_eq!(ports.len(), 3);
    assert_eq!(ports[0], 22);
    assert_eq!(ports[1], 80);
    assert_eq!(ports[2], 443);

    // Test special keywords
    let args = vec!["ntrace", "-H", "192.168.1.1", "-p", "common"];
    let cli = Cli::parse_from(args);
    let ports = cli.parse_ports().unwrap();

    assert!(ports.contains(&22));
    assert!(ports.contains(&80));
    assert!(ports.contains(&443));
}