nullnet-libfireparse 0.3.4

A library for parsing and translating firewall configurations across multiple NullNet targets
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
81
82

use roxmltree::Document;

/// A parser for extracting the hostname from a pfSense XML configuration.
pub struct PfSenseHostnameParser {}

impl PfSenseHostnameParser {
    /// Parses a given XML document and extracts the hostname and domain details.
    ///
    /// # Arguments
    ///
    /// * `document` - A reference to a `roxmltree::Document` containing pfSense configuration.
    ///
    /// # Returns
    ///
    /// A `String` representing the full hostname with domain if available.
    pub fn parse(document: &Document) -> String {
        if let Some(system_node) = document
            .descendants()
            .find(|e| e.has_tag_name("pfsense"))
            .and_then(|e| e.children().find(|ce| ce.has_tag_name("system")))
        {
            let hostname = system_node
                .children()
                .find(|c| c.has_tag_name("hostname"))
                .and_then(|c| c.text());

            let domain = system_node
                .children()
                .find(|c| c.has_tag_name("domain"))
                .and_then(|c| c.text());

            if hostname.is_some() && domain.is_some() {
                return format!("{}.{}", hostname.unwrap(), domain.unwrap());
            } else if hostname.is_some() {
                return hostname.unwrap().to_string();
            } else {
                return domain.unwrap().to_string();
            }
        }

        String::from("none")
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use roxmltree::Document;

    #[test]
    fn test_parse_hostname() {
        let xml = r#"<pfsense>
                        <system>
                            <hostname>router</hostname>
                            <domain>local</domain>
                        </system>
                    </pfsense>"#;
        let doc = Document::parse(xml).expect("Failed to parse XML");
        let hostname = PfSenseHostnameParser::parse(&doc);
        assert_eq!(hostname, "router.local");
    }

    #[test]
    fn test_parse_hostname_missing_domain() {
        let xml = r#"<pfsense>
                        <system>
                            <hostname>router</hostname>
                        </system>
                    </pfsense>"#;
        let doc = Document::parse(xml).expect("Failed to parse XML");
        let hostname = PfSenseHostnameParser::parse(&doc);
        assert_eq!(hostname, "router");
    }

    #[test]
    fn test_parse_hostname_empty_xml() {
        let xml = "<pfsense></pfsense>";
        let doc = Document::parse(xml).expect("Failed to parse XML");
        let hostname = PfSenseHostnameParser::parse(&doc);
        assert_eq!(hostname, "none");
    }
}