dns-message-parser 0.9.0

Libary to encode and decode DNS packets
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

use crate::rr::Class;
use crate::DomainName;
use hex::encode;
use std::fmt::{Display, Formatter, Result as FmtResult};

/// The bit at offset 7 of the DNSKEY flags field is the [Zone Key flag].
///
/// [Zone Key flag]: https://tools.ietf.org/html/rfc4034#section-2.1.1
pub const ZONE_KEY_FLAG: u16 = 0b0000_0001_0000_0000;
/// The  bit at offset 15 of the DNSKEY flags field is the [Secure Entry Point flag].
///
/// [Secure Entry Point flag]: https://tools.ietf.org/html/rfc4034#section-2.1.1
pub const SECURE_ENTRY_POINT_FLAG: u16 = 0b0000_0000_0000_0001;
pub const DNSKEY_ZERO_MASK: u16 = 0b1111_1110_1111_1110;

try_from_enum_to_integer! {
    #[repr(u8)]
    #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
    /// https://www.iana.org/assignments/dns-sec-alg-numbers/dns-sec-alg-numbers.xhtml
    pub enum AlgorithmType {
        Reserved = 0x00,
        RsaMd5 = 0x01,
        DiffiHellman = 0x02,
        DsaSha1 = 0x03,
        EllipticCurve = 0x04,
        RsaSha1 = 0x05,
        DsaNsec3 = 0x06,
        RsaSha1Nsec3Sha1 = 0x07,
        RsaSha256 = 0x08,
        GostR = 0x0c,
        EcDsaP256 = 0x0d,
        EcDsaP386 = 0x0e,
        Ed25519 = 0x0f,
        Ed448 = 0x10,
        Indirect = 0xfc,
        PrivateDns = 0xfd,
        PrivateOid = 0xfe,
    }
}

try_from_enum_to_integer! {
    #[repr(u8)]
    #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
    /// https://www.iana.org/assignments/ds-rr-types/ds-rr-types.xhtml#ds-rr-types-1
    pub enum DigestType {
        Reserved = 0x00,
        Sha1 = 0x01,
        Sha256 = 0x02,
        GostR = 0x03,
        Sha384 = 0x04,
    }
}

#[derive(Debug, PartialEq, Clone, Eq, Hash)]
pub struct DNSKEY {
    pub domain_name: DomainName,
    pub ttl: u32,
    pub class: Class,
    pub zone_key_flag: bool,
    pub secure_entry_point_flag: bool,
    pub algorithm_type: AlgorithmType,
    pub public_key: Vec<u8>,
}

impl DNSKEY {
    pub fn get_flags(&self) -> u16 {
        let mut flags: u16 = 0;
        if self.zone_key_flag {
            flags |= ZONE_KEY_FLAG;
        }
        if self.secure_entry_point_flag {
            flags |= SECURE_ENTRY_POINT_FLAG;
        }
        flags
    }
}

impl Display for DNSKEY {
    fn fmt(&self, f: &mut Formatter<'_>) -> FmtResult {
        write!(
            f,
            "{} {} {} DNSKEY {} 3 {} {}",
            self.domain_name,
            self.ttl,
            self.class,
            self.get_flags(),
            self.algorithm_type as u8,
            encode(&self.public_key),
        )
    }
}

#[derive(Debug, PartialEq, Clone, Eq, Hash)]
pub struct DS {
    pub domain_name: DomainName,
    pub ttl: u32,
    pub class: Class,
    pub key_tag: u16,
    pub algorithm_type: AlgorithmType,
    pub digest_type: DigestType,
    pub digest: Vec<u8>,
}

impl Display for DS {
    fn fmt(&self, f: &mut Formatter<'_>) -> FmtResult {
        write!(
            f,
            "{} {} {} DS {} {} {} {}",
            self.domain_name,
            self.ttl,
            self.class,
            self.key_tag,
            self.algorithm_type as u8,
            self.digest_type as u8,
            encode(&self.digest),
        )
    }
}