flex_dns/rdata/
opt.rs

1use crate::{Buffer, DnsMessage, DnsMessageError, MutBuffer};
2use crate::characters::Characters;
3use crate::parse::Parse;
4use crate::rdata::{RData, RDataParse};
5use crate::write::WriteBytes;
6
7/// # Option record
8/// This record is used to store options for the DNS protocol.
9#[derive(Copy, Clone, Debug, PartialEq)]
10pub struct Opt<'a> {
11    /// The code for the option.
12    pub code: u16,
13    /// The data for the option.
14    pub data: Characters<'a>,
15}
16
17impl<'a> RDataParse<'a> for Opt<'a> {
18    #[inline]
19    fn parse(rdata: &RData<'a>, i: &mut usize) -> Result<Self, DnsMessageError> {
20        let code = u16::parse(rdata.buffer, i)?;
21        let data = Characters::parse(rdata.buffer, i)?;
22
23        Ok(Self {
24            code,
25            data,
26        })
27    }
28}
29
30impl<'a> WriteBytes for Opt<'a> {
31    #[inline]
32    fn write<
33        const PTR_STORAGE: usize,
34        const DNS_SECTION: usize,
35        B: MutBuffer + Buffer,
36    >(&self, message: &mut DnsMessage<PTR_STORAGE, DNS_SECTION, B>) -> Result<usize, DnsMessageError> {
37        let mut bytes = 0;
38
39        bytes += self.code.write(message)?;
40        bytes += self.data.write(message)?;
41
42        Ok(bytes)
43    }
44}
45
46#[cfg(test)]
47mod test {
48    use crate::rdata::testutils::parse_write_test;
49
50    use super::*;
51
52    parse_write_test!(
53        6,
54        [
55            0x00, 0x0e, // code
56            0x03, // length
57            b'w', b'w', b'w', // data
58        ],
59        Opt {
60            code: 14,
61            data: unsafe { Characters::new_unchecked(b"www") },
62        },
63    );
64}