simple_dns/dns/rdata/
dhcid.rs

1use crate::{bytes_buffer::BytesBuffer, dns::WireFormat, lib::Cow, lib::Write};
2
3use super::RR;
4
5/// A DHCID record see [rfc4701](https://datatracker.ietf.org/doc/html/rfc4701)
6#[derive(Debug, PartialEq, Eq, Hash, Clone)]
7pub struct DHCID<'a> {
8    /// Identifier type code
9    pub identifier: u16,
10    /// Digest type code
11    pub digest_type: u8,
12    /// Digest (length depends on digest type)
13    pub digest: Cow<'a, [u8]>,
14}
15
16impl RR for DHCID<'_> {
17    const TYPE_CODE: u16 = 49;
18}
19
20impl<'a> WireFormat<'a> for DHCID<'a> {
21    const MINIMUM_LEN: usize = 3;
22
23    fn parse(data: &mut BytesBuffer<'a>) -> crate::Result<Self>
24    where
25        Self: Sized,
26    {
27        let identifier = data.get_u16()?;
28        let digest_type = data.get_u8()?;
29        let digest = Cow::Borrowed(data.get_remaining());
30
31        Ok(Self {
32            identifier,
33            digest_type,
34            digest,
35        })
36    }
37
38    fn write_to<T: Write>(&self, out: &mut T) -> crate::Result<()> {
39        out.write_all(&self.identifier.to_be_bytes())?;
40        out.write_all(&[self.digest_type])?;
41        out.write_all(&self.digest)?;
42
43        Ok(())
44    }
45
46    fn len(&self) -> usize {
47        self.digest.len() + Self::MINIMUM_LEN
48    }
49}
50
51impl DHCID<'_> {
52    /// Transforms the inner data into its owned type
53    pub fn into_owned<'b>(self) -> DHCID<'b> {
54        DHCID {
55            identifier: self.identifier,
56            digest_type: self.digest_type,
57            digest: Cow::Owned(self.digest.into_owned()),
58        }
59    }
60}
61
62#[cfg(test)]
63mod tests {
64    use crate::lib::Vec;
65
66    use super::*;
67
68    #[test]
69    fn parse_and_write_dhcid() {
70        let ds = DHCID {
71            identifier: 0,
72            digest_type: 0,
73            digest: Cow::Borrowed(&[0, 0, 0, 0]),
74        };
75
76        let mut data = Vec::new();
77        ds.write_to(&mut data).unwrap();
78
79        let ds = DHCID::parse(&mut (&data[..]).into()).unwrap();
80        assert_eq!(ds.identifier, 0);
81        assert_eq!(ds.digest_type, 0);
82        assert_eq!(ds.digest, Cow::Borrowed(&[0, 0, 0, 0]));
83    }
84
85    #[test]
86    #[cfg(feature = "std")]
87    fn parse_sample() -> Result<(), Box<dyn std::error::Error>> {
88        use crate::{rdata::RData, ResourceRecord};
89        let sample_file = std::fs::read("samples/zonefile/DHCID.sample")?;
90
91        let sample_rdata = match ResourceRecord::parse(&mut (&sample_file[..]).into())?.rdata {
92            RData::DHCID(rdata) => rdata,
93            _ => unreachable!(),
94        };
95
96        assert_eq!(sample_rdata.identifier, 0x0002);
97        assert_eq!(sample_rdata.digest_type, 0x01);
98        assert_eq!(*sample_rdata.digest, *b"\x63\x6f\xc0\xb8\x27\x1c\x82\x82\x5b\xb1\xac\x5c\x41\xcf\x53\x51\xaa\x69\xb4\xfe\xbd\x94\xe8\xf1\x7c\xdb\x95\x00\x0d\xa4\x8c\x40");
99
100        Ok(())
101    }
102}
simple_dns/dns/rdata/dhcid.rs

simple_dns/dns/rdata/
dhcid.rs
