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/mock_host_resolver.h" | 5 #include "net/dns/mock_host_resolver.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
11 #include "base/memory/ref_counted.h" | 11 #include "base/memory/ref_counted.h" |
12 #include "base/message_loop/message_loop.h" | 12 #include "base/message_loop/message_loop.h" |
13 #include "base/stl_util.h" | 13 #include "base/stl_util.h" |
14 #include "base/strings/string_split.h" | 14 #include "base/strings/string_split.h" |
15 #include "base/strings/string_util.h" | 15 #include "base/strings/string_util.h" |
16 #include "base/threading/platform_thread.h" | 16 #include "base/threading/platform_thread.h" |
| 17 #include "net/base/ip_endpoint.h" |
17 #include "net/base/net_errors.h" | 18 #include "net/base/net_errors.h" |
18 #include "net/base/net_util.h" | 19 #include "net/base/net_util.h" |
19 #include "net/base/test_completion_callback.h" | 20 #include "net/base/test_completion_callback.h" |
20 #include "net/dns/host_cache.h" | 21 #include "net/dns/host_cache.h" |
21 | 22 |
22 #if defined(OS_WIN) | 23 #if defined(OS_WIN) |
23 #include "net/base/winsock_init.h" | 24 #include "net/base/winsock_init.h" |
24 #endif | 25 #endif |
25 | 26 |
26 namespace net { | 27 namespace net { |
(...skipping 120 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
147 | 148 |
148 if (use_caching) { | 149 if (use_caching) { |
149 cache_.reset(new HostCache(kMaxCacheEntries)); | 150 cache_.reset(new HostCache(kMaxCacheEntries)); |
150 } | 151 } |
151 } | 152 } |
152 | 153 |
153 int MockHostResolverBase::ResolveFromIPLiteralOrCache(const RequestInfo& info, | 154 int MockHostResolverBase::ResolveFromIPLiteralOrCache(const RequestInfo& info, |
154 AddressList* addresses) { | 155 AddressList* addresses) { |
155 IPAddressNumber ip; | 156 IPAddressNumber ip; |
156 if (ParseIPLiteralToNumber(info.hostname(), &ip)) { | 157 if (ParseIPLiteralToNumber(info.hostname(), &ip)) { |
| 158 // This matches the behavior HostResolverImpl. |
| 159 if (info.address_family() != ADDRESS_FAMILY_UNSPECIFIED && |
| 160 info.address_family() != GetAddressFamily(ip)) { |
| 161 return ERR_NAME_NOT_RESOLVED; |
| 162 } |
| 163 |
157 *addresses = AddressList::CreateFromIPAddress(ip, info.port()); | 164 *addresses = AddressList::CreateFromIPAddress(ip, info.port()); |
158 if (info.host_resolver_flags() & HOST_RESOLVER_CANONNAME) | 165 if (info.host_resolver_flags() & HOST_RESOLVER_CANONNAME) |
159 addresses->SetDefaultCanonicalName(); | 166 addresses->SetDefaultCanonicalName(); |
160 return OK; | 167 return OK; |
161 } | 168 } |
162 int rv = ERR_DNS_CACHE_MISS; | 169 int rv = ERR_DNS_CACHE_MISS; |
163 if (cache_.get() && info.allow_cached_response()) { | 170 if (cache_.get() && info.allow_cached_response()) { |
164 HostCache::Key key(info.hostname(), | 171 HostCache::Key key(info.hostname(), |
165 info.address_family(), | 172 info.address_family(), |
166 info.host_resolver_flags()); | 173 info.host_resolver_flags()); |
(...skipping 269 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
436 CHECK_EQ(old_proc, current_proc_); | 443 CHECK_EQ(old_proc, current_proc_); |
437 } | 444 } |
438 | 445 |
439 void ScopedDefaultHostResolverProc::Init(HostResolverProc* proc) { | 446 void ScopedDefaultHostResolverProc::Init(HostResolverProc* proc) { |
440 current_proc_ = proc; | 447 current_proc_ = proc; |
441 previous_proc_ = HostResolverProc::SetDefault(current_proc_.get()); | 448 previous_proc_ = HostResolverProc::SetDefault(current_proc_.get()); |
442 current_proc_->SetLastProc(previous_proc_.get()); | 449 current_proc_->SetLastProc(previous_proc_.get()); |
443 } | 450 } |
444 | 451 |
445 } // namespace net | 452 } // namespace net |
OLD | NEW |