OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "net/dns/dns_config_service_win.h" | 5 #include "net/dns/dns_config_service_win.h" |
6 | 6 |
7 #include "base/basictypes.h" | 7 #include "base/basictypes.h" |
8 #include "base/logging.h" | 8 #include "base/logging.h" |
9 #include "base/win/windows_version.h" | 9 #include "base/win/windows_version.h" |
10 #include "net/dns/dns_protocol.h" | 10 #include "net/dns/dns_protocol.h" |
(...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
48 EXPECT_FALSE(result) << "Unexpected parse success on " << t.input; | 48 EXPECT_FALSE(result) << "Unexpected parse success on " << t.input; |
49 } | 49 } |
50 } | 50 } |
51 } | 51 } |
52 | 52 |
53 struct AdapterInfo { | 53 struct AdapterInfo { |
54 IFTYPE if_type; | 54 IFTYPE if_type; |
55 IF_OPER_STATUS oper_status; | 55 IF_OPER_STATUS oper_status; |
56 const WCHAR* dns_suffix; | 56 const WCHAR* dns_suffix; |
57 std::string dns_server_addresses[4]; // Empty string indicates end. | 57 std::string dns_server_addresses[4]; // Empty string indicates end. |
58 int ports[4]; | 58 uint16 ports[4]; |
59 }; | 59 }; |
60 | 60 |
61 scoped_ptr<IP_ADAPTER_ADDRESSES, base::FreeDeleter> CreateAdapterAddresses( | 61 scoped_ptr<IP_ADAPTER_ADDRESSES, base::FreeDeleter> CreateAdapterAddresses( |
62 const AdapterInfo* infos) { | 62 const AdapterInfo* infos) { |
63 size_t num_adapters = 0; | 63 size_t num_adapters = 0; |
64 size_t num_addresses = 0; | 64 size_t num_addresses = 0; |
65 for (size_t i = 0; infos[i].if_type; ++i) { | 65 for (size_t i = 0; infos[i].if_type; ++i) { |
66 ++num_adapters; | 66 ++num_adapters; |
67 for (size_t j = 0; !infos[i].dns_server_addresses[j].empty(); ++j) { | 67 for (size_t j = 0; !infos[i].dns_server_addresses[j].empty(); ++j) { |
68 ++num_addresses; | 68 ++num_addresses; |
(...skipping 26 matching lines...) Expand all Loading... |
95 for (size_t j = 0; !info.dns_server_addresses[j].empty(); ++j) { | 95 for (size_t j = 0; !info.dns_server_addresses[j].empty(); ++j) { |
96 --num_addresses; | 96 --num_addresses; |
97 if (j == 0) { | 97 if (j == 0) { |
98 address = adapter->FirstDnsServerAddress = addresses + num_addresses; | 98 address = adapter->FirstDnsServerAddress = addresses + num_addresses; |
99 } else { | 99 } else { |
100 // Note that |address| is moving backwards. | 100 // Note that |address| is moving backwards. |
101 address = address->Next = address - 1; | 101 address = address->Next = address - 1; |
102 } | 102 } |
103 IPAddressNumber ip; | 103 IPAddressNumber ip; |
104 CHECK(ParseIPLiteralToNumber(info.dns_server_addresses[j], &ip)); | 104 CHECK(ParseIPLiteralToNumber(info.dns_server_addresses[j], &ip)); |
105 IPEndPoint ipe(ip, info.ports[j]); | 105 IPEndPoint ipe = IPEndPoint(ip, info.ports[j]); |
106 address->Address.lpSockaddr = | 106 address->Address.lpSockaddr = |
107 reinterpret_cast<LPSOCKADDR>(storage + num_addresses); | 107 reinterpret_cast<LPSOCKADDR>(storage + num_addresses); |
108 socklen_t length = sizeof(struct sockaddr_storage); | 108 socklen_t length = sizeof(struct sockaddr_storage); |
109 CHECK(ipe.ToSockAddr(address->Address.lpSockaddr, &length)); | 109 CHECK(ipe.ToSockAddr(address->Address.lpSockaddr, &length)); |
110 address->Address.iSockaddrLength = static_cast<int>(length); | 110 address->Address.iSockaddrLength = static_cast<int>(length); |
111 } | 111 } |
112 } | 112 } |
113 | 113 |
114 return heap.Pass(); | 114 return heap.Pass(); |
115 } | 115 } |
116 | 116 |
117 TEST(DnsConfigServiceWinTest, ConvertAdapterAddresses) { | 117 TEST(DnsConfigServiceWinTest, ConvertAdapterAddresses) { |
118 // Check nameservers and connection-specific suffix. | 118 // Check nameservers and connection-specific suffix. |
119 const struct TestCase { | 119 const struct TestCase { |
120 AdapterInfo input_adapters[4]; // |if_type| == 0 indicates end. | 120 AdapterInfo input_adapters[4]; // |if_type| == 0 indicates end. |
121 std::string expected_nameservers[4]; // Empty string indicates end. | 121 std::string expected_nameservers[4]; // Empty string indicates end. |
122 std::string expected_suffix; | 122 std::string expected_suffix; |
123 int expected_ports[4]; | 123 uint16 expected_ports[4]; |
124 } cases[] = { | 124 } cases[] = { |
125 { // Ignore loopback and inactive adapters. | 125 { // Ignore loopback and inactive adapters. |
126 { | 126 { |
127 { IF_TYPE_SOFTWARE_LOOPBACK, IfOperStatusUp, L"funnyloop", | 127 { IF_TYPE_SOFTWARE_LOOPBACK, IfOperStatusUp, L"funnyloop", |
128 { "2.0.0.2" } }, | 128 { "2.0.0.2" } }, |
129 { IF_TYPE_FASTETHER, IfOperStatusDormant, L"example.com", | 129 { IF_TYPE_FASTETHER, IfOperStatusDormant, L"example.com", |
130 { "1.0.0.1" } }, | 130 { "1.0.0.1" } }, |
131 { IF_TYPE_USB, IfOperStatusUp, L"chromium.org", | 131 { IF_TYPE_USB, IfOperStatusUp, L"chromium.org", |
132 { "10.0.0.10", "2001:FFFF::1111" } }, | 132 { "10.0.0.10", "2001:FFFF::1111" } }, |
133 { 0 }, | 133 { 0 }, |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
174 for (size_t i = 0; i < arraysize(cases); ++i) { | 174 for (size_t i = 0; i < arraysize(cases); ++i) { |
175 const TestCase& t = cases[i]; | 175 const TestCase& t = cases[i]; |
176 internal::DnsSystemSettings settings = { | 176 internal::DnsSystemSettings settings = { |
177 CreateAdapterAddresses(t.input_adapters), | 177 CreateAdapterAddresses(t.input_adapters), |
178 // Default settings for the rest. | 178 // Default settings for the rest. |
179 }; | 179 }; |
180 std::vector<IPEndPoint> expected_nameservers; | 180 std::vector<IPEndPoint> expected_nameservers; |
181 for (size_t j = 0; !t.expected_nameservers[j].empty(); ++j) { | 181 for (size_t j = 0; !t.expected_nameservers[j].empty(); ++j) { |
182 IPAddressNumber ip; | 182 IPAddressNumber ip; |
183 ASSERT_TRUE(ParseIPLiteralToNumber(t.expected_nameservers[j], &ip)); | 183 ASSERT_TRUE(ParseIPLiteralToNumber(t.expected_nameservers[j], &ip)); |
184 int port = t.expected_ports[j]; | 184 uint16 port = t.expected_ports[j]; |
185 if (!port) | 185 if (!port) |
186 port = dns_protocol::kDefaultPort; | 186 port = dns_protocol::kDefaultPort; |
187 expected_nameservers.push_back(IPEndPoint(ip, port)); | 187 expected_nameservers.push_back(IPEndPoint(ip, port)); |
188 } | 188 } |
189 | 189 |
190 DnsConfig config; | 190 DnsConfig config; |
191 internal::ConfigParseWinResult result = | 191 internal::ConfigParseWinResult result = |
192 internal::ConvertSettingsToDnsConfig(settings, &config); | 192 internal::ConvertSettingsToDnsConfig(settings, &config); |
193 internal::ConfigParseWinResult expected_result = | 193 internal::ConfigParseWinResult expected_result = |
194 expected_nameservers.empty() ? internal::CONFIG_PARSE_WIN_NO_NAMESERVERS | 194 expected_nameservers.empty() ? internal::CONFIG_PARSE_WIN_NO_NAMESERVERS |
(...skipping 263 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
458 EXPECT_EQ(t.unhandled_options, config.unhandled_options); | 458 EXPECT_EQ(t.unhandled_options, config.unhandled_options); |
459 EXPECT_EQ(t.have_nrpt, config.use_local_ipv6); | 459 EXPECT_EQ(t.have_nrpt, config.use_local_ipv6); |
460 } | 460 } |
461 } | 461 } |
462 | 462 |
463 | 463 |
464 } // namespace | 464 } // namespace |
465 | 465 |
466 } // namespace net | 466 } // namespace net |
467 | 467 |
OLD | NEW |