rlibdns 0.1.211

Rust DNS Library
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

use std::path::PathBuf;
use crate::messages::inter::rr_classes::RRClasses;
use crate::utils::fqdn_utils::{encode_fqdn, decode_fqdn};
use crate::utils::trie::trie::Trie;
use crate::zone::inter::zone_types::ZoneTypes;
use crate::utils::name_list::NameList;
use crate::zone::zone::Zone;
use crate::zone::zone_reader::{ZoneReader, ZoneReaderError};

#[derive(Debug, Clone)]
pub struct ZoneStore {
    trie: Trie<Vec<Zone>>,
    names: NameList
}

impl ZoneStore {

    pub fn new() -> Self {
        Self {
            trie: Trie::new(),
            names: NameList::new()
        }
    }

    pub fn open<P: Into<PathBuf>>(&mut self, file_path: P, fqdn: &str, class: RRClasses) -> Result<Option<Zone>, ZoneReaderError> {
        let mut zone = Zone::new(ZoneTypes::Master, class);

        let mut reader = ZoneReader::open(file_path, fqdn, class)?;

        for record in reader.records() {
            match record {
                Ok((query, ttl, data)) => {
                    zone.add_record(&query, ttl, data);
                }
                Err(e) => {
                    println!("{}", e);
                }
            }
        }

        Ok(self.add_zone(reader.get_origin(), zone))
    }

    pub fn open_with_jnl<P: Into<PathBuf>>(&mut self, file_path: P, fqdn: &str, class: RRClasses, journal_path: P) -> Result<Option<Zone>, ZoneReaderError> {
        let mut zone = Zone::new_with_jnl(ZoneTypes::Master, class, journal_path);

        let mut reader = ZoneReader::open(file_path, fqdn, class)?;
        for record in reader.records() {
            match record {
                Ok((query, ttl, data)) => {
                    zone.add_record(&query, ttl, data);
                }
                Err(e) => {
                    println!("{}", e);
                }
            }
        }

        Ok(self.add_zone(reader.get_origin(), zone))
    }

    pub fn add_zone(&mut self, fqdn: &str, zone: Zone) -> Option<Zone> {
        let key = encode_fqdn(fqdn);
        match self.trie.get_mut(&key) {
            Some(zones) => {
                let class = zone.get_class();

                if let Some(index) = zones.iter().position(|z| z.get_class().eq(&class)) {
                    return Some(std::mem::replace(&mut zones[index], zone));
                }

                zones.push(zone);
            }
            None => {
                self.trie.insert(key, vec![zone]);
            }
        }
        None
    }

    pub fn remove_zone(&mut self, fqdn: &str, class: RRClasses) {
        //NEEDS REWORK NOW THAT CLASSES ARE INTRODUCED...
        //self.trie.remove(&encode_fqdn(fqdn));
    }

    pub fn get_zone_exact(&self, apex: &str, class: &RRClasses) -> Option<&Zone> {
        self.trie.get(&encode_fqdn(apex))?.iter().find(|z| z.get_class().eq(class))
    }

    pub fn get_zone_exact_mut(&mut self, apex: &str, class: &RRClasses) -> Option<&mut Zone> {
        self.trie.get_mut(&encode_fqdn(apex))?.iter_mut().find(|z| z.get_class().eq(class))
    }

    pub fn get_deepest_zone(&self, name: &str, class: &RRClasses) -> Option<(String, &Zone)> {
        let (key, zones) = self.trie.get_deepest(&encode_fqdn(name))?;
        Some((decode_fqdn(&key), zones.iter().find(|z| z.get_class().eq(class))?))
    }

    pub fn get_deepest_zone_mut(&mut self, name: &str, class: &RRClasses) -> Option<(String, &mut Zone)> {
        let (key, zones) = self.trie.get_deepest_mut(&encode_fqdn(name))?;
        Some((decode_fqdn(&key), zones.iter_mut().find(|z| z.get_class().eq(class))?))
    }
}