zond-cli 0.6.2

Command-Line Interface (CLI) for the Zond Engine
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

// Copyright (c) 2026 Erik Lening (hollowpointer) and Contributors
//
// This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0.
// If a copy of the MPL was not distributed with this file, You can obtain one at
// https://mozilla.org/MPL/2.0/.

use crate::terminal::colors;
use colored::*;
use std::net::{IpAddr, Ipv6Addr};
use zond_engine::core::models::host::Host;
use zond_engine::core::models::ip;
use zond_engine::core::redact;

// Logic moved from network/ip.rs
pub fn ipv6_to_type_str(ipv6_addr: &Ipv6Addr) -> &'static str {
    if is_global_unicast(&IpAddr::V6(*ipv6_addr)) {
        return "GUA";
    }
    if ipv6_addr.is_unique_local() {
        return "ULA";
    }
    if ipv6_addr.is_unicast_link_local() {
        return "LLA";
    }
    "IPv6"
}

pub fn ip_to_detail(host: &Host, redact: bool) -> Vec<(String, ColoredString)> {
    host.ips()
        .iter()
        .filter(|&&ip| ip != host.primary_ip())
        .map(|ip| match ip {
            IpAddr::V4(ipv4_addr) => {
                let value = ipv4_addr.to_string().color(colors::IPV4_ADDR);
                (String::from("IPv4"), value)
            }
            IpAddr::V6(ipv6_addr) => {
                let ipv6_type: &str = ipv6_to_type_str(ipv6_addr);
                let ipv6_addr: ColoredString = if redact {
                    let ip_str: String = match ip::get_ipv6_type(ipv6_addr) {
                        ip::Ipv6AddressType::GlobalUnicast => redact::global_unicast(ipv6_addr),
                        ip::Ipv6AddressType::UniqueLocal => redact::unique_local(ipv6_addr),
                        ip::Ipv6AddressType::LinkLocal => redact::link_local(ipv6_addr),
                        _ => ipv6_addr.to_string(),
                    };
                    ip_str.color(colors::IPV6_ADDR)
                } else {
                    ipv6_addr.to_string().color(colors::IPV6_ADDR)
                };
                (String::from(ipv6_type), ipv6_addr)
            }
        })
        .collect()
}

fn is_global_unicast(ip_addr: &IpAddr) -> bool {
    match ip_addr {
        IpAddr::V6(ipv6_addr) => {
            let first_byte = ipv6_addr.octets()[0];
            (0x20..=0x3F).contains(&first_byte)
        }
        _ => false,
    }
}

pub fn hostname_to_detail(
    hostname_opt: Option<&str>,
    redact: bool,
) -> Option<(String, ColoredString)> {
    let mut result: Option<(String, ColoredString)> = None;

    if let Some(hostname) = hostname_opt {
        let hostname_str: String = if redact {
            redact::hostname(hostname)
        } else {
            hostname.to_string()
        };
        result = Some((
            String::from("Hostname"),
            hostname_str.color(colors::HOSTNAME),
        ))
    }

    result
}

pub fn mac_to_detail(
    mac_opt: Option<zond_engine::core::models::mac::MacAddr>,
    redact: bool,
) -> Option<(String, ColoredString)> {
    let mut result: Option<(String, ColoredString)> = None;

    if let Some(mac) = mac_opt {
        let mac_str: String = if redact {
            redact::mac_addr(&mac)
        } else {
            mac.to_string()
        };
        result = Some(("MAC".to_string(), mac_str.color(colors::MAC_ADDR)))
    }

    result
}

pub fn vendor_to_detail(vendor_opt: Option<&str>) -> Option<(String, ColoredString)> {
    vendor_opt.map(|vendor| {
        (
            "Vendor".to_string(),
            vendor.to_string().color(colors::MAC_ADDR),
        )
    })
}