nsproxy-hickory-resolver 0.25.4

Hickory DNS is a safe and secure DNS library. This Resolver library uses the Client library to perform all DNS queries. The Resolver is intended to be a high-level library for any DNS record resolution. See Resolver for supported resolution types. The Client can be used for other queries. 
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

use std::collections::HashMap;

use serde::Deserialize;

use hickory_proto::rr::RecordType;

use super::{TtlBounds, TtlConfig};

#[derive(Deserialize)]
pub(super) struct TtlConfigMap(HashMap<TtlConfigField, TtlBounds>);

impl From<TtlConfigMap> for TtlConfig {
    fn from(value: TtlConfigMap) -> Self {
        let mut default = TtlBounds::default();
        let mut by_query_type = HashMap::new();
        for (field, bounds) in value.0.into_iter() {
            match field {
                TtlConfigField::RecordType(record_type) => {
                    by_query_type.insert(record_type, bounds);
                }
                TtlConfigField::Default => default = bounds,
            }
        }
        Self {
            default,
            by_query_type,
        }
    }
}

#[derive(PartialEq, Eq, Hash, Deserialize)]
enum TtlConfigField {
    #[serde(rename = "default")]
    Default,
    #[serde(untagged)]
    RecordType(RecordType),
}

#[cfg(test)]
mod tests {
    use serde::Deserialize;

    use crate::dns_lru::TtlConfig;

    #[test]
    fn error_cases() {
        // Duplicate of "default"
        let input = r#"[default]
positive_max_ttl = 3600
[default]
positive_max_ttl = 3599"#;
        let error = toml::from_str::<TtlConfig>(input).unwrap_err();
        assert!(
            error.message().contains("duplicate key `default`"),
            "wrong error message: {error}"
        );

        // Duplicate of a record type
        let input = r#"[default]
positive_max_ttl = 86400
[OPENPGPKEY]
positive_max_ttl = 3600
[OPENPGPKEY]
negative_min_ttl = 60"#;
        let error = toml::from_str::<TtlConfig>(input).unwrap_err();
        assert!(
            error.message().contains("duplicate key `OPENPGPKEY`"),
            "wrong error message: {error}"
        );

        // Neither "default" nor a record type
        let input = r#"[not_a_record_type]
positive_max_ttl = 3600"#;
        let error = toml::from_str::<TtlConfig>(input).unwrap_err();
        assert!(
            error.message().contains("data did not match any variant"),
            "wrong error message: {error}"
        );

        // Array instead of table
        #[derive(Debug, Deserialize)]
        struct Wrapper {
            #[allow(unused)]
            cache_policy: TtlConfig,
        }
        let input = r#"cache_policy = []"#;
        let error = toml::from_str::<Wrapper>(input).unwrap_err();
        assert!(
            error.message().contains("invalid type: sequence"),
            "wrong error message: {error}"
        );

        // String instead of table
        let input = r#"cache_policy = "yes""#;
        let error = toml::from_str::<Wrapper>(input).unwrap_err();
        assert!(
            error.message().contains("invalid type: string"),
            "wrong error message: {error}"
        );
    }
}