OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef NET_DNS_DNS_TEST_UTIL_H_ | |
6 #define NET_DNS_DNS_TEST_UTIL_H_ | |
7 | |
8 #include <string> | |
9 #include <vector> | |
10 | |
11 #include "base/basictypes.h" | |
12 #include "base/memory/scoped_ptr.h" | |
13 #include "net/dns/dns_client.h" | |
14 #include "net/dns/dns_config_service.h" | |
15 #include "net/dns/dns_protocol.h" | |
16 | |
17 namespace net { | |
18 | |
19 //----------------------------------------------------------------------------- | |
20 // Query/response set for www.google.com, ID is fixed to 0. | |
21 static const char kT0HostName[] = "www.google.com"; | |
22 static const uint16 kT0Qtype = dns_protocol::kTypeA; | |
23 static const char kT0DnsName[] = { | |
24 0x03, 'w', 'w', 'w', | |
25 0x06, 'g', 'o', 'o', 'g', 'l', 'e', | |
26 0x03, 'c', 'o', 'm', | |
27 0x00 | |
28 }; | |
29 static const size_t kT0QuerySize = 32; | |
30 static const uint8 kT0ResponseDatagram[] = { | |
31 // response contains one CNAME for www.l.google.com and the following | |
32 // IP addresses: 74.125.226.{179,180,176,177,178} | |
33 0x00, 0x00, 0x81, 0x80, 0x00, 0x01, 0x00, 0x06, | |
34 0x00, 0x00, 0x00, 0x00, 0x03, 0x77, 0x77, 0x77, | |
35 0x06, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x03, | |
36 0x63, 0x6f, 0x6d, 0x00, 0x00, 0x01, 0x00, 0x01, | |
37 0xc0, 0x0c, 0x00, 0x05, 0x00, 0x01, 0x00, 0x01, | |
38 0x4d, 0x13, 0x00, 0x08, 0x03, 0x77, 0x77, 0x77, | |
39 0x01, 0x6c, 0xc0, 0x10, 0xc0, 0x2c, 0x00, 0x01, | |
40 0x00, 0x01, 0x00, 0x00, 0x00, 0xe4, 0x00, 0x04, | |
41 0x4a, 0x7d, 0xe2, 0xb3, 0xc0, 0x2c, 0x00, 0x01, | |
42 0x00, 0x01, 0x00, 0x00, 0x00, 0xe4, 0x00, 0x04, | |
43 0x4a, 0x7d, 0xe2, 0xb4, 0xc0, 0x2c, 0x00, 0x01, | |
44 0x00, 0x01, 0x00, 0x00, 0x00, 0xe4, 0x00, 0x04, | |
45 0x4a, 0x7d, 0xe2, 0xb0, 0xc0, 0x2c, 0x00, 0x01, | |
46 0x00, 0x01, 0x00, 0x00, 0x00, 0xe4, 0x00, 0x04, | |
47 0x4a, 0x7d, 0xe2, 0xb1, 0xc0, 0x2c, 0x00, 0x01, | |
48 0x00, 0x01, 0x00, 0x00, 0x00, 0xe4, 0x00, 0x04, | |
49 0x4a, 0x7d, 0xe2, 0xb2 | |
50 }; | |
51 static const char* const kT0IpAddresses[] = { | |
52 "74.125.226.179", "74.125.226.180", "74.125.226.176", | |
53 "74.125.226.177", "74.125.226.178" | |
54 }; | |
55 static const char kT0CanonName[] = "www.l.google.com"; | |
56 static const int kT0TTL = 0x000000e4; | |
57 // +1 for the CNAME record. | |
58 static const unsigned kT0RecordCount = arraysize(kT0IpAddresses) + 1; | |
59 | |
60 //----------------------------------------------------------------------------- | |
61 // Query/response set for codereview.chromium.org, ID is fixed to 1. | |
62 static const char kT1HostName[] = "codereview.chromium.org"; | |
63 static const uint16 kT1Qtype = dns_protocol::kTypeA; | |
64 static const char kT1DnsName[] = { | |
65 0x0a, 'c', 'o', 'd', 'e', 'r', 'e', 'v', 'i', 'e', 'w', | |
66 0x08, 'c', 'h', 'r', 'o', 'm', 'i', 'u', 'm', | |
67 0x03, 'o', 'r', 'g', | |
68 0x00 | |
69 }; | |
70 static const size_t kT1QuerySize = 41; | |
71 static const uint8 kT1ResponseDatagram[] = { | |
72 // response contains one CNAME for ghs.l.google.com and the following | |
73 // IP address: 64.233.169.121 | |
74 0x00, 0x01, 0x81, 0x80, 0x00, 0x01, 0x00, 0x02, | |
75 0x00, 0x00, 0x00, 0x00, 0x0a, 0x63, 0x6f, 0x64, | |
76 0x65, 0x72, 0x65, 0x76, 0x69, 0x65, 0x77, 0x08, | |
77 0x63, 0x68, 0x72, 0x6f, 0x6d, 0x69, 0x75, 0x6d, | |
78 0x03, 0x6f, 0x72, 0x67, 0x00, 0x00, 0x01, 0x00, | |
79 0x01, 0xc0, 0x0c, 0x00, 0x05, 0x00, 0x01, 0x00, | |
80 0x01, 0x41, 0x75, 0x00, 0x12, 0x03, 0x67, 0x68, | |
81 0x73, 0x01, 0x6c, 0x06, 0x67, 0x6f, 0x6f, 0x67, | |
82 0x6c, 0x65, 0x03, 0x63, 0x6f, 0x6d, 0x00, 0xc0, | |
83 0x35, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x01, | |
84 0x0b, 0x00, 0x04, 0x40, 0xe9, 0xa9, 0x79 | |
85 }; | |
86 static const char* const kT1IpAddresses[] = { | |
87 "64.233.169.121" | |
88 }; | |
89 static const char kT1CanonName[] = "ghs.l.google.com"; | |
90 static const int kT1TTL = 0x0000010b; | |
91 // +1 for the CNAME record. | |
92 static const unsigned kT1RecordCount = arraysize(kT1IpAddresses) + 1; | |
93 | |
94 //----------------------------------------------------------------------------- | |
95 // Query/response set for www.ccs.neu.edu, ID is fixed to 2. | |
96 static const char kT2HostName[] = "www.ccs.neu.edu"; | |
97 static const uint16 kT2Qtype = dns_protocol::kTypeA; | |
98 static const char kT2DnsName[] = { | |
99 0x03, 'w', 'w', 'w', | |
100 0x03, 'c', 'c', 's', | |
101 0x03, 'n', 'e', 'u', | |
102 0x03, 'e', 'd', 'u', | |
103 0x00 | |
104 }; | |
105 static const size_t kT2QuerySize = 33; | |
106 static const uint8 kT2ResponseDatagram[] = { | |
107 // response contains one CNAME for vulcan.ccs.neu.edu and the following | |
108 // IP address: 129.10.116.81 | |
109 0x00, 0x02, 0x81, 0x80, 0x00, 0x01, 0x00, 0x02, | |
110 0x00, 0x00, 0x00, 0x00, 0x03, 0x77, 0x77, 0x77, | |
111 0x03, 0x63, 0x63, 0x73, 0x03, 0x6e, 0x65, 0x75, | |
112 0x03, 0x65, 0x64, 0x75, 0x00, 0x00, 0x01, 0x00, | |
113 0x01, 0xc0, 0x0c, 0x00, 0x05, 0x00, 0x01, 0x00, | |
114 0x00, 0x01, 0x2c, 0x00, 0x09, 0x06, 0x76, 0x75, | |
115 0x6c, 0x63, 0x61, 0x6e, 0xc0, 0x10, 0xc0, 0x2d, | |
116 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x01, 0x2c, | |
117 0x00, 0x04, 0x81, 0x0a, 0x74, 0x51 | |
118 }; | |
119 static const char* const kT2IpAddresses[] = { | |
120 "129.10.116.81" | |
121 }; | |
122 static const char kT2CanonName[] = "vulcan.ccs.neu.edu"; | |
123 static const int kT2TTL = 0x0000012c; | |
124 // +1 for the CNAME record. | |
125 static const unsigned kT2RecordCount = arraysize(kT2IpAddresses) + 1; | |
126 | |
127 //----------------------------------------------------------------------------- | |
128 // Query/response set for www.google.az, ID is fixed to 3. | |
129 static const char kT3HostName[] = "www.google.az"; | |
130 static const uint16 kT3Qtype = dns_protocol::kTypeA; | |
131 static const char kT3DnsName[] = { | |
132 0x03, 'w', 'w', 'w', | |
133 0x06, 'g', 'o', 'o', 'g', 'l', 'e', | |
134 0x02, 'a', 'z', | |
135 0x00 | |
136 }; | |
137 static const size_t kT3QuerySize = 31; | |
138 static const uint8 kT3ResponseDatagram[] = { | |
139 // response contains www.google.com as CNAME for www.google.az and | |
140 // www.l.google.com as CNAME for www.google.com and the following | |
141 // IP addresses: 74.125.226.{178,179,180,176,177} | |
142 // The TTLs on the records are: 0x00015099, 0x00025099, 0x00000415, | |
143 // 0x00003015, 0x00002015, 0x00000015, 0x00001015. | |
144 // The last record is an imaginary TXT record for t.google.com. | |
145 0x00, 0x03, 0x81, 0x80, 0x00, 0x01, 0x00, 0x08, | |
146 0x00, 0x00, 0x00, 0x00, 0x03, 0x77, 0x77, 0x77, | |
147 0x06, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x02, | |
148 0x61, 0x7a, 0x00, 0x00, 0x01, 0x00, 0x01, 0xc0, | |
149 0x0c, 0x00, 0x05, 0x00, 0x01, 0x00, 0x01, 0x50, | |
150 0x99, 0x00, 0x10, 0x03, 0x77, 0x77, 0x77, 0x06, | |
151 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x03, 0x63, | |
152 0x6f, 0x6d, 0x00, 0xc0, 0x2b, 0x00, 0x05, 0x00, | |
153 0x01, 0x00, 0x02, 0x50, 0x99, 0x00, 0x08, 0x03, | |
154 0x77, 0x77, 0x77, 0x01, 0x6c, 0xc0, 0x2f, 0xc0, | |
155 0x47, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x04, | |
156 0x15, 0x00, 0x04, 0x4a, 0x7d, 0xe2, 0xb2, 0xc0, | |
157 0x47, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x30, | |
158 0x15, 0x00, 0x04, 0x4a, 0x7d, 0xe2, 0xb3, 0xc0, | |
159 0x47, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x20, | |
160 0x15, 0x00, 0x04, 0x4a, 0x7d, 0xe2, 0xb4, 0xc0, | |
161 0x47, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x00, | |
162 0x15, 0x00, 0x04, 0x4a, 0x7d, 0xe2, 0xb0, 0xc0, | |
163 0x47, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x10, | |
164 0x15, 0x00, 0x04, 0x4a, 0x7d, 0xe2, 0xb1, 0x01, | |
165 0x74, 0xc0, 0x2f, 0x00, 0x10, 0x00, 0x01, 0x00, | |
166 0x00, 0x00, 0x01, 0x00, 0x04, 0xde, 0xad, 0xfe, | |
167 0xed | |
168 }; | |
169 static const char* const kT3IpAddresses[] = { | |
170 "74.125.226.178", "74.125.226.179", "74.125.226.180", | |
171 "74.125.226.176", "74.125.226.177" | |
172 }; | |
173 static const char kT3CanonName[] = "www.l.google.com"; | |
174 static const int kT3TTL = 0x00000015; | |
175 // +2 for the CNAME records, +1 for TXT record. | |
176 static const unsigned kT3RecordCount = arraysize(kT3IpAddresses) + 3; | |
177 | |
178 class AddressSorter; | |
179 class DnsClient; | |
180 class MockTransactionFactory; | |
181 | |
182 struct MockDnsClientRule { | |
183 enum Result { | |
184 FAIL, // Fail asynchronously with ERR_NAME_NOT_RESOLVED. | |
185 TIMEOUT, // Fail asynchronously with ERR_DNS_TIMEOUT. | |
186 EMPTY, // Return an empty response. | |
187 OK, // Return a response with loopback address. | |
188 }; | |
189 | |
190 // If |delay| is true, matching transactions will be delayed until triggered | |
191 // by the consumer. | |
192 MockDnsClientRule(const std::string& prefix_arg, | |
193 uint16 qtype_arg, | |
194 Result result_arg, | |
195 bool delay) | |
196 : result(result_arg), prefix(prefix_arg), qtype(qtype_arg), | |
197 delay(delay) {} | |
198 | |
199 Result result; | |
200 std::string prefix; | |
201 uint16 qtype; | |
202 bool delay; | |
203 }; | |
204 | |
205 typedef std::vector<MockDnsClientRule> MockDnsClientRuleList; | |
206 | |
207 // MockDnsClient provides MockTransactionFactory. | |
208 class MockDnsClient : public DnsClient { | |
209 public: | |
210 MockDnsClient(const DnsConfig& config, const MockDnsClientRuleList& rules); | |
211 ~MockDnsClient() override; | |
212 | |
213 // DnsClient interface: | |
214 void SetConfig(const DnsConfig& config) override; | |
215 const DnsConfig* GetConfig() const override; | |
216 DnsTransactionFactory* GetTransactionFactory() override; | |
217 AddressSorter* GetAddressSorter() override; | |
218 | |
219 // Completes all DnsTransactions that were delayed by a rule. | |
220 void CompleteDelayedTransactions(); | |
221 | |
222 private: | |
223 DnsConfig config_; | |
224 scoped_ptr<MockTransactionFactory> factory_; | |
225 scoped_ptr<AddressSorter> address_sorter_; | |
226 }; | |
227 | |
228 } // namespace net | |
229 | |
230 #endif // NET_DNS_DNS_TEST_UTIL_H_ | |
OLD | NEW |