1 #include "ares-test.h"
2 #include "dns-proto.h"
3
4 #include <sstream>
5 #include <vector>
6
7 namespace ares {
8 namespace test {
9
TEST_F(LibraryTest,ParseRootName)10 TEST_F(LibraryTest, ParseRootName) {
11 DNSPacket pkt;
12 pkt.set_qid(0x1234).set_response().set_aa()
13 .add_question(new DNSQuestion(".", ns_t_a))
14 .add_answer(new DNSARR(".", 100, {0x02, 0x03, 0x04, 0x05}));
15 std::vector<byte> data = pkt.data();
16
17 struct hostent *host = nullptr;
18 struct ares_addrttl info[2];
19 int count = 2;
20 EXPECT_EQ(ARES_SUCCESS, ares_parse_a_reply(data.data(), data.size(),
21 &host, info, &count));
22 EXPECT_EQ(1, count);
23 std::stringstream ss;
24 ss << HostEnt(host);
25 EXPECT_EQ("{'' aliases=[] addrs=[2.3.4.5]}", ss.str());
26 ares_free_hostent(host);
27 }
28
TEST_F(LibraryTest,ParseIndirectRootName)29 TEST_F(LibraryTest, ParseIndirectRootName) {
30 std::vector<byte> data = {
31 0x12, 0x34, // qid
32 0x84, // response + query + AA + not-TC + not-RD
33 0x00, // not-RA + not-Z + not-AD + not-CD + rc=NoError
34 0x00, 0x01, // num questions
35 0x00, 0x01, // num answer RRs
36 0x00, 0x00, // num authority RRs
37 0x00, 0x00, // num additional RRs
38 // Question
39 0xC0, 0x04, // weird: pointer to a random zero earlier in the message
40 0x00, 0x01, // type A
41 0x00, 0x01, // class IN
42 // Answer 1
43 0xC0, 0x04,
44 0x00, 0x01, // RR type
45 0x00, 0x01, // class IN
46 0x01, 0x02, 0x03, 0x04, // TTL
47 0x00, 0x04, // rdata length
48 0x02, 0x03, 0x04, 0x05,
49 };
50
51 struct hostent *host = nullptr;
52 struct ares_addrttl info[2];
53 int count = 2;
54 EXPECT_EQ(ARES_SUCCESS, ares_parse_a_reply(data.data(), data.size(),
55 &host, info, &count));
56 EXPECT_EQ(1, count);
57 std::stringstream ss;
58 ss << HostEnt(host);
59 EXPECT_EQ("{'' aliases=[] addrs=[2.3.4.5]}", ss.str());
60 ares_free_hostent(host);
61 }
62
TEST_F(LibraryTest,ParseEscapedName)63 TEST_F(LibraryTest, ParseEscapedName) {
64 std::vector<byte> data = {
65 0x12, 0x34, // qid
66 0x84, // response + query + AA + not-TC + not-RD
67 0x00, // not-RA + not-Z + not-AD + not-CD + rc=NoError
68 0x00, 0x01, // num questions
69 0x00, 0x01, // num answer RRs
70 0x00, 0x00, // num authority RRs
71 0x00, 0x00, // num additional RRs
72 // Question
73 0x05, 'a', '\\', 'b', '.', 'c',
74 0x03, 'c', 'o', 'm',
75 0x00,
76 0x00, 0x01, // type A
77 0x00, 0x01, // class IN
78 // Answer 1
79 0x05, 'a', '\\', 'b', '.', 'c',
80 0x03, 'c', 'o', 'm',
81 0x00,
82 0x00, 0x01, // RR type
83 0x00, 0x01, // class IN
84 0x01, 0x02, 0x03, 0x04, // TTL
85 0x00, 0x04, // rdata length
86 0x02, 0x03, 0x04, 0x05,
87 };
88 struct hostent *host = nullptr;
89 struct ares_addrttl info[2];
90 int count = 2;
91 EXPECT_EQ(ARES_SUCCESS, ares_parse_a_reply(data.data(), data.size(),
92 &host, info, &count));
93 EXPECT_EQ(1, count);
94 HostEnt hent(host);
95 std::stringstream ss;
96 ss << hent;
97 // The printable name is expanded with escapes.
98 EXPECT_EQ(11, hent.name_.size());
99 EXPECT_EQ('a', hent.name_[0]);
100 EXPECT_EQ('\\', hent.name_[1]);
101 EXPECT_EQ('\\', hent.name_[2]);
102 EXPECT_EQ('b', hent.name_[3]);
103 EXPECT_EQ('\\', hent.name_[4]);
104 EXPECT_EQ('.', hent.name_[5]);
105 EXPECT_EQ('c', hent.name_[6]);
106 ares_free_hostent(host);
107 }
108
TEST_F(LibraryTest,ParsePartialCompressedName)109 TEST_F(LibraryTest, ParsePartialCompressedName) {
110 std::vector<byte> data = {
111 0x12, 0x34, // qid
112 0x84, // response + query + AA + not-TC + not-RD
113 0x00, // not-RA + not-Z + not-AD + not-CD + rc=NoError
114 0x00, 0x01, // num questions
115 0x00, 0x01, // num answer RRs
116 0x00, 0x00, // num authority RRs
117 0x00, 0x00, // num additional RRs
118 // Question
119 0x03, 'w', 'w', 'w',
120 0x07, 'e', 'x', 'a', 'm', 'p', 'l', 'e',
121 0x03, 'c', 'o', 'm',
122 0x00,
123 0x00, 0x01, // type A
124 0x00, 0x01, // class IN
125 // Answer 1
126 0x03, 'w', 'w', 'w',
127 0xc0, 0x10, // offset 16
128 0x00, 0x01, // RR type
129 0x00, 0x01, // class IN
130 0x01, 0x02, 0x03, 0x04, // TTL
131 0x00, 0x04, // rdata length
132 0x02, 0x03, 0x04, 0x05,
133 };
134 struct hostent *host = nullptr;
135 struct ares_addrttl info[2];
136 int count = 2;
137 EXPECT_EQ(ARES_SUCCESS, ares_parse_a_reply(data.data(), data.size(),
138 &host, info, &count));
139 ASSERT_NE(nullptr, host);
140 std::stringstream ss;
141 ss << HostEnt(host);
142 EXPECT_EQ("{'www.example.com' aliases=[] addrs=[2.3.4.5]}", ss.str());
143 ares_free_hostent(host);
144 }
145
TEST_F(LibraryTest,ParseFullyCompressedName)146 TEST_F(LibraryTest, ParseFullyCompressedName) {
147 std::vector<byte> data = {
148 0x12, 0x34, // qid
149 0x84, // response + query + AA + not-TC + not-RD
150 0x00, // not-RA + not-Z + not-AD + not-CD + rc=NoError
151 0x00, 0x01, // num questions
152 0x00, 0x01, // num answer RRs
153 0x00, 0x00, // num authority RRs
154 0x00, 0x00, // num additional RRs
155 // Question
156 0x03, 'w', 'w', 'w',
157 0x07, 'e', 'x', 'a', 'm', 'p', 'l', 'e',
158 0x03, 'c', 'o', 'm',
159 0x00,
160 0x00, 0x01, // type A
161 0x00, 0x01, // class IN
162 // Answer 1
163 0xc0, 0x0c, // offset 12
164 0x00, 0x01, // RR type
165 0x00, 0x01, // class IN
166 0x01, 0x02, 0x03, 0x04, // TTL
167 0x00, 0x04, // rdata length
168 0x02, 0x03, 0x04, 0x05,
169 };
170 struct hostent *host = nullptr;
171 struct ares_addrttl info[2];
172 int count = 2;
173 EXPECT_EQ(ARES_SUCCESS, ares_parse_a_reply(data.data(), data.size(),
174 &host, info, &count));
175 ASSERT_NE(nullptr, host);
176 std::stringstream ss;
177 ss << HostEnt(host);
178 EXPECT_EQ("{'www.example.com' aliases=[] addrs=[2.3.4.5]}", ss.str());
179 ares_free_hostent(host);
180 }
181
TEST_F(LibraryTest,ParseFullyCompressedName2)182 TEST_F(LibraryTest, ParseFullyCompressedName2) {
183 std::vector<byte> data = {
184 0x12, 0x34, // qid
185 0x84, // response + query + AA + not-TC + not-RD
186 0x00, // not-RA + not-Z + not-AD + not-CD + rc=NoError
187 0x00, 0x01, // num questions
188 0x00, 0x01, // num answer RRs
189 0x00, 0x00, // num authority RRs
190 0x00, 0x00, // num additional RRs
191 // Question
192 0xC0, 0x12, // pointer to later in message
193 0x00, 0x01, // type A
194 0x00, 0x01, // class IN
195 // Answer 1
196 0x03, 'w', 'w', 'w',
197 0x07, 'e', 'x', 'a', 'm', 'p', 'l', 'e',
198 0x03, 'c', 'o', 'm',
199 0x00,
200 0x00, 0x01, // RR type
201 0x00, 0x01, // class IN
202 0x01, 0x02, 0x03, 0x04, // TTL
203 0x00, 0x04, // rdata length
204 0x02, 0x03, 0x04, 0x05,
205 };
206 struct hostent *host = nullptr;
207 struct ares_addrttl info[2];
208 int count = 2;
209 EXPECT_EQ(ARES_SUCCESS, ares_parse_a_reply(data.data(), data.size(),
210 &host, info, &count));
211 ASSERT_NE(nullptr, host);
212 std::stringstream ss;
213 ss << HostEnt(host);
214 EXPECT_EQ("{'www.example.com' aliases=[] addrs=[2.3.4.5]}", ss.str());
215 ares_free_hostent(host);
216 }
217
218 } // namespace test
219 } // namespace ares
220