nd300 3.0.9

Cross-platform network diagnostic tool
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

use serde::Serialize;

use super::shared_cache::SharedCache;

#[derive(Debug, Clone, Serialize)]
pub struct ConnectionStates {
    pub established: u32,
    pub time_wait: u32,
    pub close_wait: u32,
    pub fin_wait_1: u32,
    pub fin_wait_2: u32,
    pub syn_sent: u32,
    pub syn_received: u32,
    pub closing: u32,
    pub last_ack: u32,
    pub listen: u32,
}

pub async fn collect_with_cache(cache: &SharedCache) -> Option<ConnectionStates> {
    if let Some(ref nc) = cache.netstat {
        return Some(parse_connection_states(&nc.lines));
    }
    collect().await
}

fn parse_connection_states(lines: &[String]) -> ConnectionStates {
    let mut states = ConnectionStates {
        established: 0,
        time_wait: 0,
        close_wait: 0,
        fin_wait_1: 0,
        fin_wait_2: 0,
        syn_sent: 0,
        syn_received: 0,
        closing: 0,
        last_ack: 0,
        listen: 0,
    };

    for line in lines {
        let line = line.trim().to_uppercase();
        if line.contains("ESTABLISHED") {
            states.established += 1;
        } else if line.contains("TIME_WAIT") {
            states.time_wait += 1;
        } else if line.contains("CLOSE_WAIT") {
            states.close_wait += 1;
        } else if line.contains("FIN_WAIT_1") || line.contains("FIN_WAIT1") {
            states.fin_wait_1 += 1;
        } else if line.contains("FIN_WAIT_2") || line.contains("FIN_WAIT2") {
            states.fin_wait_2 += 1;
        } else if line.contains("SYN_SENT") {
            states.syn_sent += 1;
        } else if line.contains("SYN_RECV") || line.contains("SYN_RECEIVED") {
            states.syn_received += 1;
        } else if line.contains("CLOSING") {
            states.closing += 1;
        } else if line.contains("LAST_ACK") {
            states.last_ack += 1;
        } else if line.contains("LISTEN") || line.contains("LISTENING") {
            states.listen += 1;
        }
    }

    states
}

pub async fn collect() -> Option<ConnectionStates> {
    #[cfg(windows)]
    let output = {
        let mut cmd = tokio::process::Command::new("netstat");
        cmd.args(["-an"]);
        super::util::run_with_timeout(cmd, super::util::QUICK).await
    };

    #[cfg(unix)]
    let output = {
        let mut cmd = tokio::process::Command::new("netstat");
        cmd.args(["-an"]);
        super::util::run_with_timeout(cmd, super::util::QUICK).await
    };

    let output = output?;
    let text = String::from_utf8_lossy(&output.stdout);

    let mut states = ConnectionStates {
        established: 0,
        time_wait: 0,
        close_wait: 0,
        fin_wait_1: 0,
        fin_wait_2: 0,
        syn_sent: 0,
        syn_received: 0,
        closing: 0,
        last_ack: 0,
        listen: 0,
    };

    for line in text.lines() {
        let line = line.trim().to_uppercase();
        if line.contains("ESTABLISHED") {
            states.established += 1;
        } else if line.contains("TIME_WAIT") {
            states.time_wait += 1;
        } else if line.contains("CLOSE_WAIT") {
            states.close_wait += 1;
        } else if line.contains("FIN_WAIT_1") || line.contains("FIN_WAIT1") {
            states.fin_wait_1 += 1;
        } else if line.contains("FIN_WAIT_2") || line.contains("FIN_WAIT2") {
            states.fin_wait_2 += 1;
        } else if line.contains("SYN_SENT") {
            states.syn_sent += 1;
        } else if line.contains("SYN_RECV") || line.contains("SYN_RECEIVED") {
            states.syn_received += 1;
        } else if line.contains("CLOSING") {
            states.closing += 1;
        } else if line.contains("LAST_ACK") {
            states.last_ack += 1;
        } else if line.contains("LISTEN") || line.contains("LISTENING") {
            states.listen += 1;
        }
    }

    Some(states)
}