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 #include "net/dns/dns_hosts.h" | |
6 | |
7 #include "testing/gtest/include/gtest/gtest.h" | |
8 | |
9 namespace net { | |
10 | |
11 namespace { | |
12 | |
13 struct ExpectedHostsEntry { | |
14 const char* host; | |
15 AddressFamily family; | |
16 const char* ip; | |
17 }; | |
18 | |
19 void PopulateExpectedHosts(const ExpectedHostsEntry* entries, | |
20 size_t num_entries, | |
21 DnsHosts* expected_hosts_out) { | |
22 for (size_t i = 0; i < num_entries; ++i) { | |
23 DnsHostsKey key(entries[i].host, entries[i].family); | |
24 IPAddressNumber& ip_ref = (*expected_hosts_out)[key]; | |
25 ASSERT_TRUE(ip_ref.empty()); | |
26 ASSERT_TRUE(ParseIPLiteralToNumber(entries[i].ip, &ip_ref)); | |
27 ASSERT_EQ(ip_ref.size(), | |
28 (entries[i].family == ADDRESS_FAMILY_IPV4) ? 4u : 16u); | |
29 } | |
30 } | |
31 | |
32 TEST(DnsHostsTest, ParseHosts) { | |
33 const std::string kContents = | |
34 "127.0.0.1 localhost\tlocalhost.localdomain # standard\n" | |
35 "\n" | |
36 "1.0.0.1 localhost # ignored, first hit above\n" | |
37 "fe00::x example company # ignored, malformed IPv6\n" | |
38 "1.0.0.300 company # ignored, malformed IPv4\n" | |
39 "1.0.0.1 # ignored, missing hostname\n" | |
40 "1.0.0.1\t CoMpANy # normalized to 'company' \n" | |
41 "::1\tlocalhost ip6-localhost ip6-loopback # comment # within a comment\n" | |
42 "\t fe00::0 ip6-localnet\r\n" | |
43 "2048::2 example\n" | |
44 "2048::1 company example # ignored for 'example' \n" | |
45 "127.0.0.1 cache1\n" | |
46 "127.0.0.1 cache2 # should reuse parsed IP\n" | |
47 "256.0.0.0 cache3 # bogus IP should not clear parsed IP cache\n" | |
48 "127.0.0.1 cache4 # should still be reused\n" | |
49 "127.0.0.2 cache5\n" | |
50 "gibberish"; | |
51 | |
52 const ExpectedHostsEntry kEntries[] = { | |
53 { "localhost", ADDRESS_FAMILY_IPV4, "127.0.0.1" }, | |
54 { "localhost.localdomain", ADDRESS_FAMILY_IPV4, "127.0.0.1" }, | |
55 { "company", ADDRESS_FAMILY_IPV4, "1.0.0.1" }, | |
56 { "localhost", ADDRESS_FAMILY_IPV6, "::1" }, | |
57 { "ip6-localhost", ADDRESS_FAMILY_IPV6, "::1" }, | |
58 { "ip6-loopback", ADDRESS_FAMILY_IPV6, "::1" }, | |
59 { "ip6-localnet", ADDRESS_FAMILY_IPV6, "fe00::0" }, | |
60 { "company", ADDRESS_FAMILY_IPV6, "2048::1" }, | |
61 { "example", ADDRESS_FAMILY_IPV6, "2048::2" }, | |
62 { "cache1", ADDRESS_FAMILY_IPV4, "127.0.0.1" }, | |
63 { "cache2", ADDRESS_FAMILY_IPV4, "127.0.0.1" }, | |
64 { "cache4", ADDRESS_FAMILY_IPV4, "127.0.0.1" }, | |
65 { "cache5", ADDRESS_FAMILY_IPV4, "127.0.0.2" }, | |
66 }; | |
67 | |
68 DnsHosts expected_hosts, actual_hosts; | |
69 PopulateExpectedHosts(kEntries, arraysize(kEntries), &expected_hosts); | |
70 ParseHosts(kContents, &actual_hosts); | |
71 ASSERT_EQ(expected_hosts, actual_hosts); | |
72 } | |
73 | |
74 TEST(DnsHostsTest, ParseHosts_CommaIsToken) { | |
75 const std::string kContents = "127.0.0.1 comma1,comma2"; | |
76 | |
77 const ExpectedHostsEntry kEntries[] = { | |
78 { "comma1,comma2", ADDRESS_FAMILY_IPV4, "127.0.0.1" }, | |
79 }; | |
80 | |
81 DnsHosts expected_hosts, actual_hosts; | |
82 PopulateExpectedHosts(kEntries, arraysize(kEntries), &expected_hosts); | |
83 ParseHostsWithCommaModeForTesting( | |
84 kContents, &actual_hosts, PARSE_HOSTS_COMMA_IS_TOKEN); | |
85 ASSERT_EQ(expected_hosts, actual_hosts); | |
86 } | |
87 | |
88 TEST(DnsHostsTest, ParseHosts_CommaIsWhitespace) { | |
89 std::string kContents = "127.0.0.1 comma1,comma2"; | |
90 | |
91 const ExpectedHostsEntry kEntries[] = { | |
92 { "comma1", ADDRESS_FAMILY_IPV4, "127.0.0.1" }, | |
93 { "comma2", ADDRESS_FAMILY_IPV4, "127.0.0.1" }, | |
94 }; | |
95 | |
96 DnsHosts expected_hosts, actual_hosts; | |
97 PopulateExpectedHosts(kEntries, arraysize(kEntries), &expected_hosts); | |
98 ParseHostsWithCommaModeForTesting( | |
99 kContents, &actual_hosts, PARSE_HOSTS_COMMA_IS_WHITESPACE); | |
100 ASSERT_EQ(expected_hosts, actual_hosts); | |
101 } | |
102 | |
103 // Test that the right comma mode is used on each platform. | |
104 TEST(DnsHostsTest, ParseHosts_CommaModeByPlatform) { | |
105 std::string kContents = "127.0.0.1 comma1,comma2"; | |
106 | |
107 #if defined(OS_MACOSX) | |
108 const ExpectedHostsEntry kEntries[] = { | |
109 { "comma1", ADDRESS_FAMILY_IPV4, "127.0.0.1" }, | |
110 { "comma2", ADDRESS_FAMILY_IPV4, "127.0.0.1" }, | |
111 }; | |
112 #else | |
113 const ExpectedHostsEntry kEntries[] = { | |
114 { "comma1,comma2", ADDRESS_FAMILY_IPV4, "127.0.0.1" }, | |
115 }; | |
116 #endif | |
117 | |
118 DnsHosts expected_hosts, actual_hosts; | |
119 PopulateExpectedHosts(kEntries, arraysize(kEntries), &expected_hosts); | |
120 ParseHosts(kContents, &actual_hosts); | |
121 ASSERT_EQ(expected_hosts, actual_hosts); | |
122 } | |
123 | |
124 TEST(DnsHostsTest, HostsParser_Empty) { | |
125 DnsHosts hosts; | |
126 ParseHosts("", &hosts); | |
127 EXPECT_EQ(0u, hosts.size()); | |
128 } | |
129 | |
130 TEST(DnsHostsTest, HostsParser_OnlyWhitespace) { | |
131 DnsHosts hosts; | |
132 ParseHosts(" ", &hosts); | |
133 EXPECT_EQ(0u, hosts.size()); | |
134 } | |
135 | |
136 TEST(DnsHostsTest, HostsParser_EndsWithNothing) { | |
137 DnsHosts hosts; | |
138 ParseHosts("127.0.0.1 localhost", &hosts); | |
139 EXPECT_EQ(1u, hosts.size()); | |
140 } | |
141 | |
142 TEST(DnsHostsTest, HostsParser_EndsWithWhitespace) { | |
143 DnsHosts hosts; | |
144 ParseHosts("127.0.0.1 localhost ", &hosts); | |
145 EXPECT_EQ(1u, hosts.size()); | |
146 } | |
147 | |
148 TEST(DnsHostsTest, HostsParser_EndsWithComment) { | |
149 DnsHosts hosts; | |
150 ParseHosts("127.0.0.1 localhost # comment", &hosts); | |
151 EXPECT_EQ(1u, hosts.size()); | |
152 } | |
153 | |
154 TEST(DnsHostsTest, HostsParser_EndsWithNewline) { | |
155 DnsHosts hosts; | |
156 ParseHosts("127.0.0.1 localhost\n", &hosts); | |
157 EXPECT_EQ(1u, hosts.size()); | |
158 } | |
159 | |
160 TEST(DnsHostsTest, HostsParser_EndsWithTwoNewlines) { | |
161 DnsHosts hosts; | |
162 ParseHosts("127.0.0.1 localhost\n\n", &hosts); | |
163 EXPECT_EQ(1u, hosts.size()); | |
164 } | |
165 | |
166 TEST(DnsHostsTest, HostsParser_EndsWithNewlineAndWhitespace) { | |
167 DnsHosts hosts; | |
168 ParseHosts("127.0.0.1 localhost\n ", &hosts); | |
169 EXPECT_EQ(1u, hosts.size()); | |
170 } | |
171 | |
172 TEST(DnsHostsTest, HostsParser_EndsWithNewlineAndToken) { | |
173 DnsHosts hosts; | |
174 ParseHosts("127.0.0.1 localhost\ntoken", &hosts); | |
175 EXPECT_EQ(1u, hosts.size()); | |
176 } | |
177 | |
178 } // namespace | |
179 | |
180 } // namespace net | |
181 | |
OLD | NEW |