use crate::{Buffer, DnsMessage, DnsMessageError};
use crate::name::DnsName;
use crate::parse::Parse;
use crate::rdata::{RData, RDataParse};
use crate::write::WriteBytes;
#[derive(Copy, Clone, Debug, PartialEq)]
pub struct Soa<'a> {
pub mname: DnsName<'a>,
pub rname: DnsName<'a>,
pub serial: u32,
pub refresh: u32,
pub retry: u32,
pub expire: u32,
pub minimum: u32,
}
impl<'a> RDataParse<'a> for Soa<'a> {
#[inline]
fn parse(rdata: &RData<'a>, i: &mut usize) -> Result<Self, DnsMessageError> {
let mname = DnsName::parse(rdata.buffer, i)?;
let rname = DnsName::parse(rdata.buffer, i)?;
let serial = u32::parse(rdata.buffer, i)?;
let refresh = u32::parse(rdata.buffer, i)?;
let retry = u32::parse(rdata.buffer, i)?;
let expire = u32::parse(rdata.buffer, i)?;
let minimum = u32::parse(rdata.buffer, i)?;
Ok(Self {
mname,
rname,
serial,
refresh,
retry,
expire,
minimum,
})
}
}
impl<'a> WriteBytes for Soa<'a> {
#[inline]
fn write<
const PTR_STORAGE: usize,
const DNS_SECTION: usize,
B: Buffer,
>(&self, message: &mut DnsMessage<PTR_STORAGE, DNS_SECTION, B>) -> Result<usize, DnsMessageError> {
let mut bytes = 0;
bytes += self.mname.write(message)?;
bytes += self.rname.write(message)?;
bytes += self.serial.write(message)?;
bytes += self.refresh.write(message)?;
bytes += self.retry.write(message)?;
bytes += self.expire.write(message)?;
bytes += self.minimum.write(message)?;
Ok(bytes)
}
}
#[cfg(test)]
mod test {
use crate::rdata::testutils::parse_write_test;
use super::*;
parse_write_test!(
53,
[
0x03, b'w', b'w', b'w',
0x07, b'e', b'x', b'a', b'm', b'p', b'l', b'e',
0x03, b'c', b'o', b'm',
0x00, 0x03, b'w', b'w', b'w',
0x06, b'g', b'o', b'o', b'g', b'l', b'e',
0x03, b'c', b'o', b'm',
0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x0f, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x11, 0x00, 0x00, 0x00, 0x12, ],
Soa {
mname: unsafe { DnsName::new_unchecked(b"\x03www\x07example\x03com\x00") },
rname: unsafe { DnsName::new_unchecked(b"\x03www\x06google\x03com\x00") },
serial: 14,
refresh: 15,
retry: 16,
expire: 17,
minimum: 18,
},
);
}