simple_dns/dns/rdata/
route_through.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 RouteThrough<'a> {
12 pub preference: u16,
15
16 pub intermediate_host: Name<'a>,
18}
19
20impl RR for RouteThrough<'_> {
21 const TYPE_CODE: u16 = 21;
22}
23
24impl RouteThrough<'_> {
25 pub fn into_owned<'b>(self) -> RouteThrough<'b> {
27 RouteThrough {
28 preference: self.preference,
29 intermediate_host: self.intermediate_host.into_owned(),
30 }
31 }
32}
33
34impl<'a> WireFormat<'a> for RouteThrough<'a> {
35 const MINIMUM_LEN: usize = 2;
36 fn parse(data: &mut BytesBuffer<'a>) -> crate::Result<Self>
37 where
38 Self: Sized,
39 {
40 let preference = data.get_u16()?;
41 let intermediate_host = Name::parse(data)?;
42
43 Ok(Self {
44 preference,
45 intermediate_host,
46 })
47 }
48
49 fn write_to<T: Write>(&self, out: &mut T) -> crate::Result<()> {
50 out.write_all(&self.preference.to_be_bytes())?;
51 self.intermediate_host.write_to(out)
52 }
53
54 fn write_compressed_to<T: Write + Seek>(
55 &'a self,
56 out: &mut T,
57 name_refs: &mut crate::lib::BTreeMap<&[crate::Label<'a>], u16>,
58 ) -> crate::Result<()> {
59 out.write_all(&self.preference.to_be_bytes())?;
60 self.intermediate_host.write_compressed_to(out, name_refs)
61 }
62
63 fn len(&self) -> usize {
64 self.intermediate_host.len() + Self::MINIMUM_LEN
65 }
66}
67
68#[cfg(test)]
69mod tests {
70
71 use super::*;
72 use crate::lib::{ToString, Vec};
73
74 #[test]
75 fn parse_and_write_route_through() {
76 let rt = RouteThrough {
77 preference: 10,
78 intermediate_host: Name::new("e.exchange.com").unwrap(),
79 };
80
81 let mut data = Vec::new();
82 assert!(rt.write_to(&mut data).is_ok());
83
84 let rt = RouteThrough::parse(&mut data[..].into());
85 assert!(rt.is_ok());
86 let rt = rt.unwrap();
87
88 assert_eq!(data.len(), rt.len());
89 assert_eq!(10, rt.preference);
90 assert_eq!("e.exchange.com", rt.intermediate_host.to_string());
91 }
92
93 #[test]
94 #[cfg(feature = "std")]
95 fn parse_sample() -> Result<(), Box<dyn std::error::Error>> {
96 use crate::{rdata::RData, ResourceRecord};
97 let sample_file = std::fs::read("samples/zonefile/RT.sample")?;
98
99 let sample_rdata = match ResourceRecord::parse(&mut sample_file[..].into())?.rdata {
100 RData::RouteThrough(rdata) => rdata,
101 _ => unreachable!(),
102 };
103
104 assert_eq!(sample_rdata.preference, 0);
105 assert_eq!(
106 sample_rdata.intermediate_host,
107 "intermediate-host.sample".try_into()?
108 );
109 Ok(())
110 }
111}