simple_dns/dns/rdata/
rp.rs1use crate::{
2 bytes_buffer::BytesBuffer,
3 dns::{Name, WireFormat},
4 lib::{Seek, Write},
5};
6
7use super::RR;
8
9#[derive(Debug, PartialEq, Eq, Hash, Clone)]
11pub struct RP<'a> {
12 pub mbox: Name<'a>,
14 pub txt: Name<'a>,
16}
17
18impl RR for RP<'_> {
19 const TYPE_CODE: u16 = 17;
20}
21
22impl RP<'_> {
23 pub fn into_owned<'b>(self) -> RP<'b> {
25 RP {
26 mbox: self.mbox.into_owned(),
27 txt: self.txt.into_owned(),
28 }
29 }
30}
31
32impl<'a> WireFormat<'a> for RP<'a> {
33 const MINIMUM_LEN: usize = 0;
34 fn parse(data: &mut BytesBuffer<'a>) -> crate::Result<Self>
35 where
36 Self: Sized,
37 {
38 let mbox = Name::parse(data)?;
39 let txt = Name::parse(data)?;
40
41 Ok(RP { mbox, txt })
42 }
43
44 fn write_to<T: Write>(&self, out: &mut T) -> crate::Result<()> {
45 self.mbox.write_to(out)?;
46 self.txt.write_to(out)
47 }
48
49 fn write_compressed_to<T: Write + Seek>(
50 &'a self,
51 out: &mut T,
52 name_refs: &mut crate::lib::BTreeMap<&[crate::Label<'a>], u16>,
53 ) -> crate::Result<()> {
54 self.mbox.write_compressed_to(out, name_refs)?;
55 self.txt.write_compressed_to(out, name_refs)
56 }
57
58 fn len(&self) -> usize {
59 self.txt.len() + self.mbox.len()
60 }
61}
62
63#[cfg(test)]
64mod tests {
65
66 use super::*;
67 use crate::lib::{ToString, Vec};
68
69 #[test]
70 fn parse_and_write_rp() {
71 let rp = RP {
72 mbox: Name::new("mbox.rp.com").unwrap(),
73 txt: Name::new("txt.rp.com").unwrap(),
74 };
75
76 let mut data = Vec::new();
77 assert!(rp.write_to(&mut data).is_ok());
78
79 let rp = RP::parse(&mut data[..].into());
80 assert!(rp.is_ok());
81 let rp = rp.unwrap();
82
83 assert_eq!(data.len(), rp.len());
84 assert_eq!("mbox.rp.com", rp.mbox.to_string());
85 assert_eq!("txt.rp.com", rp.txt.to_string());
86 }
87
88 #[test]
89 #[cfg(feature = "std")]
90 fn parse_sample() -> Result<(), Box<dyn std::error::Error>> {
91 use crate::{rdata::RData, ResourceRecord};
92 let sample_file = std::fs::read("samples/zonefile/RP.sample")?;
93
94 let sample_rdata = match ResourceRecord::parse(&mut sample_file[..].into())?.rdata {
95 RData::RP(rdata) => rdata,
96 _ => unreachable!(),
97 };
98
99 assert_eq!(sample_rdata.mbox, "mbox-dname.sample".try_into()?);
100 assert_eq!(sample_rdata.txt, "txt-dname.sample".try_into()?);
101 Ok(())
102 }
103}