dns_parser_revived/rdata/
srv.rs

1use crate::{Error, Name};
2use byteorder::{BigEndian, ByteOrder};
3
4#[derive(Debug, Clone, Copy)]
5pub struct Record<'a> {
6    pub priority: u16,
7    pub weight: u16,
8    pub port: u16,
9    pub target: Name<'a>,
10}
11
12impl<'a> super::Record<'a> for Record<'a> {
13    const TYPE: isize = 33;
14
15    fn parse(rdata: &'a [u8], original: &'a [u8]) -> super::RDataResult<'a> {
16        if rdata.len() < 7 {
17            return Err(Error::WrongRdataLength);
18        }
19        let record = Record {
20            priority: BigEndian::read_u16(&rdata[..2]),
21            weight: BigEndian::read_u16(&rdata[2..4]),
22            port: BigEndian::read_u16(&rdata[4..6]),
23            target: Name::scan(&rdata[6..], original)?,
24        };
25        Ok(super::RData::SRV(record))
26    }
27}
28
29#[cfg(test)]
30mod test {
31
32    use super::*;
33    use crate::Class as C;
34    use crate::Opcode::*;
35    use crate::QueryClass as QC;
36    use crate::QueryType as QT;
37    use crate::RData;
38    use crate::ResponseCode::NoError;
39    use crate::{Header, Packet};
40
41    #[test]
42    fn parse_response() {
43        let response = b"[\xd9\x81\x80\x00\x01\x00\x05\x00\x00\x00\x00\
44            \x0c_xmpp-server\x04_tcp\x05gmail\x03com\x00\x00!\x00\x01\
45            \xc0\x0c\x00!\x00\x01\x00\x00\x03\x84\x00 \x00\x05\x00\x00\
46            \x14\x95\x0bxmpp-server\x01l\x06google\x03com\x00\xc0\x0c\x00!\
47            \x00\x01\x00\x00\x03\x84\x00%\x00\x14\x00\x00\x14\x95\
48            \x04alt3\x0bxmpp-server\x01l\x06google\x03com\x00\
49            \xc0\x0c\x00!\x00\x01\x00\x00\x03\x84\x00%\x00\x14\x00\x00\
50            \x14\x95\x04alt1\x0bxmpp-server\x01l\x06google\x03com\x00\
51            \xc0\x0c\x00!\x00\x01\x00\x00\x03\x84\x00%\x00\x14\x00\x00\
52            \x14\x95\x04alt2\x0bxmpp-server\x01l\x06google\x03com\x00\
53            \xc0\x0c\x00!\x00\x01\x00\x00\x03\x84\x00%\x00\x14\x00\x00\
54            \x14\x95\x04alt4\x0bxmpp-server\x01l\x06google\x03com\x00";
55        let packet = Packet::parse(response).unwrap();
56        assert_eq!(
57            packet.header,
58            Header {
59                id: 23513,
60                query: false,
61                opcode: StandardQuery,
62                authoritative: false,
63                truncated: false,
64                recursion_desired: true,
65                recursion_available: true,
66                authenticated_data: false,
67                checking_disabled: false,
68                response_code: NoError,
69                questions: 1,
70                answers: 5,
71                nameservers: 0,
72                additional: 0,
73            }
74        );
75        assert_eq!(packet.questions.len(), 1);
76        assert_eq!(packet.questions[0].qtype, QT::SRV);
77        assert_eq!(packet.questions[0].qclass, QC::IN);
78        assert_eq!(
79            &packet.questions[0].qname.to_string()[..],
80            "_xmpp-server._tcp.gmail.com"
81        );
82        assert_eq!(packet.answers.len(), 5);
83        let items = [
84            (5, 0, 5269, "xmpp-server.l.google.com"),
85            (20, 0, 5269, "alt3.xmpp-server.l.google.com"),
86            (20, 0, 5269, "alt1.xmpp-server.l.google.com"),
87            (20, 0, 5269, "alt2.xmpp-server.l.google.com"),
88            (20, 0, 5269, "alt4.xmpp-server.l.google.com"),
89        ];
90        for (i, item) in items.iter().enumerate() {
91            assert_eq!(
92                &packet.answers[i].name.to_string()[..],
93                "_xmpp-server._tcp.gmail.com"
94            );
95            assert_eq!(packet.answers[i].cls, C::IN);
96            assert_eq!(packet.answers[i].ttl, 900);
97            match packet.answers[i].data {
98                RData::SRV(Record {
99                    priority,
100                    weight,
101                    port,
102                    target,
103                }) => {
104                    assert_eq!(priority, item.0);
105                    assert_eq!(weight, item.1);
106                    assert_eq!(port, item.2);
107                    assert_eq!(target.to_string(), (item.3).to_string());
108                }
109                ref x => panic!("Wrong rdata {:?}", x),
110            }
111        }
112    }
113}