OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/base/host_resolver.h" | 5 #include "net/base/host_resolver.h" |
6 | 6 |
7 #include "base/compiler_specific.h" | |
8 #include "base/logging.h" | |
9 #include "net/base/net_errors.h" | |
10 | |
11 namespace net { | 7 namespace net { |
12 | 8 |
13 HostResolver::RequestInfo::RequestInfo(const HostPortPair& host_port_pair) | 9 HostResolver::RequestInfo::RequestInfo(const HostPortPair& host_port_pair) |
14 : host_port_pair_(host_port_pair), | 10 : host_port_pair_(host_port_pair), |
15 address_family_(ADDRESS_FAMILY_UNSPECIFIED), | 11 address_family_(ADDRESS_FAMILY_UNSPECIFIED), |
16 host_resolver_flags_(0), | 12 host_resolver_flags_(0), |
17 allow_cached_response_(true), | 13 allow_cached_response_(true), |
18 only_use_cached_response_(false), | 14 only_use_cached_response_(false), |
19 is_speculative_(false), | 15 is_speculative_(false), |
20 priority_(MEDIUM) { | 16 priority_(MEDIUM) { |
21 } | 17 } |
22 | 18 |
23 HostResolver::~HostResolver() { | 19 HostResolver::~HostResolver() { |
24 } | 20 } |
25 | 21 |
26 AddressFamily HostResolver::GetDefaultAddressFamily() const { | 22 AddressFamily HostResolver::GetDefaultAddressFamily() const { |
27 return ADDRESS_FAMILY_UNSPECIFIED; | 23 return ADDRESS_FAMILY_UNSPECIFIED; |
28 } | 24 } |
29 | 25 |
30 HostResolverImpl* HostResolver::GetAsHostResolverImpl() { | 26 HostResolverImpl* HostResolver::GetAsHostResolverImpl() { |
31 return NULL; | 27 return NULL; |
32 } | 28 } |
33 | 29 |
34 HostResolver::HostResolver() { | 30 HostResolver::HostResolver() { |
35 } | 31 } |
36 | 32 |
37 SingleRequestHostResolver::SingleRequestHostResolver(HostResolver* resolver) | |
38 : resolver_(resolver), | |
39 cur_request_(NULL), | |
40 cur_request_callback_(NULL), | |
41 ALLOW_THIS_IN_INITIALIZER_LIST( | |
42 callback_(this, &SingleRequestHostResolver::OnResolveCompletion)) { | |
43 DCHECK(resolver_ != NULL); | |
44 } | |
45 | |
46 SingleRequestHostResolver::~SingleRequestHostResolver() { | |
47 Cancel(); | |
48 } | |
49 | |
50 int SingleRequestHostResolver::Resolve(const HostResolver::RequestInfo& info, | |
51 AddressList* addresses, | |
52 CompletionCallback* callback, | |
53 const BoundNetLog& net_log) { | |
54 DCHECK(!cur_request_ && !cur_request_callback_) << "resolver already in use"; | |
55 | |
56 HostResolver::RequestHandle request = NULL; | |
57 | |
58 // We need to be notified of completion before |callback| is called, so that | |
59 // we can clear out |cur_request_*|. | |
60 CompletionCallback* transient_callback = callback ? &callback_ : NULL; | |
61 | |
62 int rv = resolver_->Resolve( | |
63 info, addresses, transient_callback, &request, net_log); | |
64 | |
65 if (rv == ERR_IO_PENDING) { | |
66 // Cleared in OnResolveCompletion(). | |
67 cur_request_ = request; | |
68 cur_request_callback_ = callback; | |
69 } | |
70 | |
71 return rv; | |
72 } | |
73 | |
74 void SingleRequestHostResolver::Cancel() { | |
75 if (cur_request_) { | |
76 resolver_->CancelRequest(cur_request_); | |
77 cur_request_ = NULL; | |
78 } | |
79 } | |
80 | |
81 void SingleRequestHostResolver::OnResolveCompletion(int result) { | |
82 DCHECK(cur_request_ && cur_request_callback_); | |
83 | |
84 CompletionCallback* callback = cur_request_callback_; | |
85 | |
86 // Clear the outstanding request information. | |
87 cur_request_ = NULL; | |
88 cur_request_callback_ = NULL; | |
89 | |
90 // Call the user's original callback. | |
91 callback->Run(result); | |
92 } | |
93 | |
94 } // namespace net | 33 } // namespace net |
OLD | NEW |